commit 796222a3002d04a41570b66e8b523cf79a28c90e Author: t Date: Sat Jul 18 19:25:40 2026 +0800 snapshot before regression test diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..42ac223 --- /dev/null +++ b/.gitignore @@ -0,0 +1,73 @@ +# Python +__pycache__/ +*.py[cod] +*$py.class +*.so +.Python +build/ +develop-eggs/ +dist/ +downloads/ +eggs/ +.eggs/ +lib64/ +parts/ +sdist/ +var/ +wheels/ +*.egg-info/ +.installed.cfg +*.egg + +# 编译输出 +*.exe +*.out +*.o +*.obj +*.a +*.lib +*.dll +*.so + +# 调试文件 +*.p2c +debug.txt +*.ll + +# 备份文件 +* copy.py +* copy.c + +# IDE +.vscode/ +.idea/ +*.swp +*.swo +*~ + +# 测试输出 +test.exe +test_split.exe + +# 排除独立仓库的目录 +ViperOS/ +zlib copy/ + +# 磁盘镜像 +*.img +*.vmdk +*.7z +*.fd + +# 日志文件 +*.log + +# 字体文件 +*.ttc +*.ttf + +# VMware +Undo/vmware/ + +# Trae AI +.trae/ diff --git a/.transpyc_cache/0019c939aae7f9f1.doc.json b/.transpyc_cache/0019c939aae7f9f1.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/0019c939aae7f9f1.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/0019c939aae7f9f1.pyi b/.transpyc_cache/0019c939aae7f9f1.pyi new file mode 100644 index 0000000..7c47ef2 --- /dev/null +++ b/.transpyc_cache/0019c939aae7f9f1.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from mpooltest.py +Module: mpooltest +""" + + +import t, c +from t import CInt, CPtr, CChar +import viperlib +import viperio +import stdlib +import memhub + +def mpool_main() -> CInt: pass diff --git a/.transpyc_cache/001c1416e7a7dc50.doc.json b/.transpyc_cache/001c1416e7a7dc50.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/001c1416e7a7dc50.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/001c1416e7a7dc50.pyi b/.transpyc_cache/001c1416e7a7dc50.pyi new file mode 100644 index 0000000..3901124 --- /dev/null +++ b/.transpyc_cache/001c1416e7a7dc50.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/0029db369f740dd9.doc.json b/.transpyc_cache/0029db369f740dd9.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/0029db369f740dd9.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/0029db369f740dd9.pyi b/.transpyc_cache/0029db369f740dd9.pyi new file mode 100644 index 0000000..2294a70 --- /dev/null +++ b/.transpyc_cache/0029db369f740dd9.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/0035c95a18d4f8e8.pyi b/.transpyc_cache/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/.transpyc_cache/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/.transpyc_cache/02885f6f134079c2.doc.json b/.transpyc_cache/02885f6f134079c2.doc.json new file mode 100644 index 0000000..1a7c43a --- /dev/null +++ b/.transpyc_cache/02885f6f134079c2.doc.json @@ -0,0 +1 @@ +{"VarEntry": "变量表条目", "Scope": "作用域节点", "SymbolTable": "嵌套作用域符号表", "init_vars": "分配并清零变量表数组", "find_var": "在变量表中按名称查找", "find_var_entry": "在变量表中按名称查找，返回 VarEntry 或 None", "lookup_var_entry": "从当前作用域逐级向上查找变量，返回 VarEntry 或 None", "set_var_annot_class_name": "设置变量的原始类型注解类名，返回 0=成功 / 1=失败", "add_var": "添加变量到变量表", "get_or_create_var": "查找或创建变量 alloca", "_create_scope": "创建并初始化作用域节点", "init_symbol_table": "创建并初始化符号表，包含模块级根作用域", "enter_scope": "进入新作用域，返回新创建的作用域", "exit_scope": "退出当前作用域，恢复到父作用域", "define_var": "在当前作用域定义变量，返回 0 成功", "define_module_var": "在模块作用域定义变量，返回 0 成功", "lookup_var": "从当前作用域逐级向上查找变量，返回 alloca 或 None", "lookup_current": "仅在当前作用域查找变量", "lookup_module_var": "在模块作用域查找变量", "get_or_create_sym": "查找或创建变量 alloca（在当前作用域）"} \ No newline at end of file diff --git a/.transpyc_cache/02885f6f134079c2.pyi b/.transpyc_cache/02885f6f134079c2.pyi new file mode 100644 index 0000000..fb5e055 --- /dev/null +++ b/.transpyc_cache/02885f6f134079c2.pyi @@ -0,0 +1,68 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesVar.py +Module: lib.core.Handles.HandlesVar +""" + + +import t, c +from stdint import * +import llvmlite +import memhub +import string + +SCOPE_MODULE: t.CDefine = 0 +SCOPE_FUNCTION: t.CDefine = 1 +SCOPE_BLOCK: t.CDefine = 2 +MAX_VARS: t.CDefine = 256 + +@t.NoVTable +class VarEntry: + Name: t.CChar | t.CPtr + Alloca: llvmlite.Value | t.CPtr + Used: t.CInt + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class Scope: + Parent: Scope | t.CPtr + Vars: VarEntry | t.CPtr + VarCount: t.CInt + Kind: t.CInt +@t.NoVTable +class SymbolTable: + Pool: memhub.MemBuddy | t.CPtr + Root: Scope | t.CPtr + Current: Scope | t.CPtr + +def init_vars(pool: memhub.MemBuddy | t.CPtr) -> VarEntry | t.CPtr: pass + +def find_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> llvmlite.Value | t.CPtr: pass + +def find_var_entry(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> VarEntry | t.CPtr: pass + +def lookup_var_entry(symtab: SymbolTable | t.CPtr, name: str) -> VarEntry | t.CPtr: pass + +def set_var_annot_class_name(symtab: SymbolTable | t.CPtr, name: str, class_name: str) -> int: pass + +def add_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def get_or_create_var(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _create_scope(pool: memhub.MemBuddy | t.CPtr, parent: Scope | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def init_symbol_table(pool: memhub.MemBuddy | t.CPtr) -> SymbolTable | t.CPtr: pass + +def enter_scope(symtab: SymbolTable | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def exit_scope(symtab: SymbolTable | t.CPtr) -> t.CInt: pass + +def define_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def define_module_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def lookup_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_current(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_module_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def get_or_create_sym(symtab: SymbolTable | t.CPtr, pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/02d833b3cf571c5b.doc.json b/.transpyc_cache/02d833b3cf571c5b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/02d833b3cf571c5b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/02d833b3cf571c5b.pyi b/.transpyc_cache/02d833b3cf571c5b.pyi new file mode 100644 index 0000000..33275d4 --- /dev/null +++ b/.transpyc_cache/02d833b3cf571c5b.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import hashtable +import testcheck + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/03b24bc1d9327958.doc.json b/.transpyc_cache/03b24bc1d9327958.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/03b24bc1d9327958.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/03b24bc1d9327958.pyi b/.transpyc_cache/03b24bc1d9327958.pyi new file mode 100644 index 0000000..8757634 --- /dev/null +++ b/.transpyc_cache/03b24bc1d9327958.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import __zlib.pyzlib as pyzlib +import __zlib.zdeflate as zdeflate +import __zlib.zinflate as zinflate +import __zlib.zchecksum as zchecksum +import __zlib.zdef as zdef +import __zlib.zhuff as zhuff +import __zlib.test_pyzlib as tpz +import w32.win32memory +import c +import testcheck + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/03eeb0bea44f6637.doc.json b/.transpyc_cache/03eeb0bea44f6637.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/03eeb0bea44f6637.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/03eeb0bea44f6637.pyi b/.transpyc_cache/03eeb0bea44f6637.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/03eeb0bea44f6637.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/04673ecc3aec489e.doc.json b/.transpyc_cache/04673ecc3aec489e.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/04673ecc3aec489e.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/04673ecc3aec489e.pyi b/.transpyc_cache/04673ecc3aec489e.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/04673ecc3aec489e.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/05c11f22675c60cb.pyi b/.transpyc_cache/05c11f22675c60cb.pyi new file mode 100644 index 0000000..3520489 --- /dev/null +++ b/.transpyc_cache/05c11f22675c60cb.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from llvmlite.__values.py +Module: llvmlite.__values +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode +from .__types import LLVMType, TypePrint + +class Value(GSListNode[Value]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + IsConst: t.CInt + IntVal: t.CInt64T + FloatVal: t.CDouble + +VALUE_SIZE: t.CDefine = 48 + +def new_value(pool: memhub.MemBuddy | t.CPtr) -> Value | t.CPtr: pass + +def ConstInt(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CInt64T, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstNull(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstZero(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def ConstFloat(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + +def SSAValue(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ValuePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def value_get_ty(val: Value | t.CPtr) -> LLVMType | t.CPtr: pass + +def value_get_next(val: Value | t.CPtr) -> Value | t.CPtr: pass + +def value_set_ty(val: Value | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def value_set_next(val: Value | t.CPtr, nxt: Value | t.CPtr) -> t.CInt: pass + +def value_set_name(val: Value | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def value_set_isconst(val: Value | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def _append_str(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/05db996fc87ebdaa.doc.json b/.transpyc_cache/05db996fc87ebdaa.doc.json new file mode 100644 index 0000000..67e7a8b --- /dev/null +++ b/.transpyc_cache/05db996fc87ebdaa.doc.json @@ -0,0 +1 @@ +{"_DiagAppend": "追加诊断信息到 diag.txt（追加模式，不依赖 printf 缓冲）", "TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/05db996fc87ebdaa.pyi b/.transpyc_cache/05db996fc87ebdaa.pyi new file mode 100644 index 0000000..9d8b33e --- /dev/null +++ b/.transpyc_cache/05db996fc87ebdaa.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def _DiagAppend(msg: bytes) -> t.CInt: pass + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/067c78e9f121dce3.pyi b/.transpyc_cache/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/.transpyc_cache/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/.transpyc_cache/067f811511854c88.doc.json b/.transpyc_cache/067f811511854c88.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/067f811511854c88.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/067f811511854c88.pyi b/.transpyc_cache/067f811511854c88.pyi new file mode 100644 index 0000000..cef2575 --- /dev/null +++ b/.transpyc_cache/067f811511854c88.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def add(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def sub(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def multi_args(a: t.CInt, b: t.CInt, c: t.CInt, d: t.CInt) -> t.CInt: pass + +def factorial(n: t.CInt) -> t.CInt: pass + +def fib(n: t.CInt) -> t.CInt: pass + +def add_via_ptr(p: t.CInt | t.CPtr, val: t.CInt) -> t.CInt: pass + +def make_u64() -> t.CUInt64T: pass + +def is_even(n: t.CInt) -> t.CBool: pass + +def test_basic_call() -> t.CInt: pass + +def test_multi_args() -> t.CInt: pass + +def test_recursion() -> t.CInt: pass + +def test_fibonacci() -> t.CInt: pass + +def test_ptr_param() -> t.CInt: pass + +def test_u64_return() -> t.CInt: pass + +def test_bool_return() -> t.CInt: pass + +def test_nested_call() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/06f53cc594b4ac6c.pyi b/.transpyc_cache/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/.transpyc_cache/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/.transpyc_cache/07a8c82a8ca2be90.doc.json b/.transpyc_cache/07a8c82a8ca2be90.doc.json new file mode 100644 index 0000000..1f30b90 --- /dev/null +++ b/.transpyc_cache/07a8c82a8ca2be90.doc.json @@ -0,0 +1 @@ +{"_binop_to_dunder": "将 BinOp 运算符映射到 dunder 方法名，无映射时返回 None", "_cmpop_to_dunder": "将 Compare 运算符映射到 dunder 方法名，无映射时返回 None", "try_operator_overload": "尝试运算符重载：lhs 是结构体指针时调用 dunder 方法，否则返回 None", "translate_binop": "翻译二元运算（自动类型提升）"} \ No newline at end of file diff --git a/.transpyc_cache/07a8c82a8ca2be90.pyi b/.transpyc_cache/07a8c82a8ca2be90.pyi new file mode 100644 index 0000000..1481115 --- /dev/null +++ b/.transpyc_cache/07a8c82a8ca2be90.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprOps.py +Module: lib.core.Handles.HandlesExprOps +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesTranslator as HT + +def _binop_to_dunder(op: int) -> str: pass + +def _cmpop_to_dunder(op: int) -> str: pass + +def try_operator_overload(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, lhs: llvmlite.Value | t.CPtr, rhs: llvmlite.Value | t.CPtr, op: int, trans: HT.Translator | t.CPtr, is_compare: int) -> llvmlite.Value | t.CPtr: pass + +def translate_binop(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/07f7da85af876124.doc.json b/.transpyc_cache/07f7da85af876124.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/07f7da85af876124.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/07f7da85af876124.pyi b/.transpyc_cache/07f7da85af876124.pyi new file mode 100644 index 0000000..7abdba5 --- /dev/null +++ b/.transpyc_cache/07f7da85af876124.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from ptr_test.py +Module: ptr_test +""" + + +import stdio +import t, c + +def ptr_test() -> int: pass diff --git a/.transpyc_cache/0853acdbdfc9b27c.doc.json b/.transpyc_cache/0853acdbdfc9b27c.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/0853acdbdfc9b27c.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/0853acdbdfc9b27c.pyi b/.transpyc_cache/0853acdbdfc9b27c.pyi new file mode 100644 index 0000000..821be2f --- /dev/null +++ b/.transpyc_cache/0853acdbdfc9b27c.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +from t import CInt, CExport +from stdio import printf +import memhub +import stdlib +import sys +import string +import ast +import w32.win32console + +ARENA_SIZE: t.CDefine = 1048576 +POOL_SIZE: t.CDefine = 524288 +DUMP_SIZE: t.CDefine = 262144 + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/08ea5281873244fd.doc.json b/.transpyc_cache/08ea5281873244fd.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/08ea5281873244fd.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/08ea5281873244fd.pyi b/.transpyc_cache/08ea5281873244fd.pyi new file mode 100644 index 0000000..0c46010 --- /dev/null +++ b/.transpyc_cache/08ea5281873244fd.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import stdlib +import string +import memhub +import argparse +import testcheck + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +@t.CExport +def main() -> int: pass diff --git a/.transpyc_cache/08f96e9e0bc0501b.doc.json b/.transpyc_cache/08f96e9e0bc0501b.doc.json new file mode 100644 index 0000000..92253b4 --- /dev/null +++ b/.transpyc_cache/08f96e9e0bc0501b.doc.json @@ -0,0 +1 @@ +{"add_imported_module": "记录已导入的模块名，返回新的 imported_modules 字符串指针", "is_module_imported": "检查模块是否已导入（词边界精确匹配，避免子串误匹配）", "add_from_import": "添加 from-import 映射 \"name:module\"，返回新的 from_imports 字符串", "lookup_from_import": "查找 from-import 名称，返回模块名或 None\n\n 支持 star import: 如果 from_imports 中有 \"*:module\" 条目，\n 且未找到精确名称匹配，且 allow_star_fallback != 0，则返回 star import 的模块名。\n ", "ImportsHandle": "导入语句处理器：继承 Mixin 获得 Trans 回指针", "ImportsHandle.HandleImport": "处理 import 语句，返回 0", "ImportsHandle.HandleImportFromModule": "处理 from X import Y 的模块部分，更新 trans._imported_modules", "ImportsHandle.HandleImportFromNames": "处理 from X import Y 的名称部分，更新 trans._from_imports"} \ No newline at end of file diff --git a/.transpyc_cache/08f96e9e0bc0501b.pyi b/.transpyc_cache/08f96e9e0bc0501b.pyi new file mode 100644 index 0000000..2bf7399 --- /dev/null +++ b/.transpyc_cache/08f96e9e0bc0501b.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesImports.py +Module: lib.core.Handles.HandlesImports +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, imported_modules: str, name: str) -> str: pass + +def is_module_imported(imported_modules: str, name: str) -> int: pass + +def add_from_import(pool: memhub.MemBuddy | t.CPtr, from_imports: str, local_name: str, module_name: str) -> str: pass + +def lookup_from_import(from_imports: str, name: str, allow_star_fallback: int) -> str: pass + + +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + def __init__(self: ImportsHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleImport(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromModule(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromNames(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: pass diff --git a/.transpyc_cache/0900c165e9bcb8f1.doc.json b/.transpyc_cache/0900c165e9bcb8f1.doc.json new file mode 100644 index 0000000..215ef80 --- /dev/null +++ b/.transpyc_cache/0900c165e9bcb8f1.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/0900c165e9bcb8f1.pyi b/.transpyc_cache/0900c165e9bcb8f1.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/0900c165e9bcb8f1.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/0b82fcb0aa3212c5.doc.json b/.transpyc_cache/0b82fcb0aa3212c5.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/.transpyc_cache/0b82fcb0aa3212c5.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/.transpyc_cache/0b82fcb0aa3212c5.pyi b/.transpyc_cache/0b82fcb0aa3212c5.pyi new file mode 100644 index 0000000..d866b3e --- /dev/null +++ b/.transpyc_cache/0b82fcb0aa3212c5.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/0c212981c180e7fb.pyi b/.transpyc_cache/0c212981c180e7fb.pyi new file mode 100644 index 0000000..f432b06 --- /dev/null +++ b/.transpyc_cache/0c212981c180e7fb.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from ast.lexer.py +Module: ast.lexer +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, TokOp, _init_tables, _kw_lookup, _op_head, OpEntry, new_token, token_set_str, token_set_str_literal + +LEXER_MAX_INDENT: t.CDefine = 256 + +class Lexer: + src: str + pos: t.CSizeT + len: t.CSizeT + lineno: t.CInt + col: t.CInt + pool: memhub.MemManager | t.CPtr + indent_stack: t.CInt | t.CPtr + indent_top: t.CInt + paren_depth: t.CInt + tokens_head: Token | t.CPtr + tokens_tail: Token | t.CPtr + at_line_start: t.CInt + error_count: t.CInt + def __new__(self: Lexer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_lexer(pool: memhub.MemManager | t.CPtr) -> Lexer | t.CPtr: pass + +def _lexer_init(lx: Lexer | t.CPtr, src: str, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(lx: Lexer | t.CPtr, offset: t.CSizeT) -> t.CChar: pass + +def _peek_cur(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _advance(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _is_alpha(ch: t.CChar) -> t.CInt: pass + +def _is_digit(ch: t.CChar) -> t.CInt: pass + +def _is_alnum(ch: t.CChar) -> t.CInt: pass + +def _is_hex(ch: t.CChar) -> t.CInt: pass + +def _is_space(ch: t.CChar) -> t.CInt: pass + +def _emit(lx: Lexer | t.CPtr, tok: Token | t.CPtr) -> t.CInt: pass + +def _make_op_token(lx: Lexer | t.CPtr, op_id: t.CInt, length: t.CInt, lineno: t.CInt, col: t.CInt) -> Token | t.CPtr: pass + +def _handle_indent(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_name(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_number(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _emit_number(lx: Lexer | t.CPtr, start: t.CSizeT, start_lineno: t.CInt, start_col: t.CInt, is_float: t.CInt, is_complex: t.CInt, is_hex: t.CInt) -> t.CInt: pass + +def _read_string(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_operator(lx: Lexer | t.CPtr) -> t.CInt: pass + +def tokenize(lx: Lexer | t.CPtr) -> Token | t.CPtr: pass diff --git a/.transpyc_cache/0c56128647a0b845.doc.json b/.transpyc_cache/0c56128647a0b845.doc.json new file mode 100644 index 0000000..47b3516 --- /dev/null +++ b/.transpyc_cache/0c56128647a0b845.doc.json @@ -0,0 +1 @@ +{"register_cdefine_constant": "注册 CDefine 编译期常量到全局表", "lookup_cdefine_constant": "查找 CDefine 编译期常量，返回值或 -1（未找到）", "clear_cdefine_constants": "清空 CDefine 常量表（模块切换时调用）", "set_generic_context": "设置泛型特化上下文（进入泛型类方法体翻译时调用）", "clear_generic_context": "清除泛型特化上下文（离开泛型类方法体翻译时调用）", "resolve_generic_type_name": "如果 name 是泛型类型参数，返回对应的具体类型名；否则返回 name 本身\n\n 用于表达式中的 T 替换（如 T.__sizeof__() → int.__sizeof__()）\n ", "_get_type_name_from_annotation": "从类型注解节点提取类型名（支持 Name/Attribute/BinOp）", "_parse_type_string": "将类型名字符串解析为 AST 节点", "set_current_file": "设置当前编译的文件名（供报错使用）", "_read_source_line": "从源文件读取第 lineno 行内容", "fatal_error": "以 Python 风格报错并退出（带源代码行）", "fatal_type_error": "以 Python 风格报错并退出（fatal_error 的别名）", "map_t_type": "将 t.CXxx / c.CXxx 类型名映射到 LLVMType\n\n 注意: t.CChar | t.CPtr = i8* = str = bytes\n ", "_is_t_attr": "检测节点是否是 t.attr_name 形式", "extract_class_name_from_annotation": "从类型注解 AST 中提取类名，返回 None=无法提取", "resolve_annotation_type": "解析类型注解节点为 LLVMType\n\n Args:\n pool: 内存分配器\n node: 类型注解 AST 节点（Name 或 Attribute）\n imported_modules: 已导入模块名字符串（空格分隔）\n from_imports: from-import 映射字符串（\"name:module name:module\" 格式）\n ", "has_decorator_marker": "检测类型注解中是否包含指定的装饰器标记", "is_pass_body": "检测函数体是否为 pass（空体或仅含 pass 语句）", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "infer_expr_type": "推断表达式的 LLVM 类型\n\n param_types 格式: \"name:bits,name:bits,...\"\n ", "infer_return_type": "扫描函数体中的 return 语句，推断返回类型", "build_param_types_str": "构建参数类型编码字符串 \"name:bits,name:bits,...\" ", "_strappend": "将 src 追加到 dst 末尾"} \ No newline at end of file diff --git a/.transpyc_cache/0c56128647a0b845.pyi b/.transpyc_cache/0c56128647a0b845.pyi new file mode 100644 index 0000000..b7e0303 --- /dev/null +++ b/.transpyc_cache/0c56128647a0b845.pyi @@ -0,0 +1,74 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesType.py +Module: lib.core.Handles.HandlesType +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import stdlib +import viperlib +import w32.fileio as fileio +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesStruct as HandlesStruct + +_g_current_file: t.CExtern | str +_g_generic_tp_names: t.CExtern | list[str] | t.CPtr +_g_generic_type_args: t.CExtern | list[str] | t.CPtr +_g_cdefine_names: t.CExtern | list[str] | t.CPtr +_g_cdefine_values: t.CExtern | list[str] | t.CPtr + +def register_cdefine_constant(pool: memhub.MemBuddy | t.CPtr, name: str, value: int) -> None: pass + +def lookup_cdefine_constant(name: str) -> int: pass + +def clear_cdefine_constants() -> None: pass + +def set_generic_context(tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> t.CInt: pass + +def clear_generic_context() -> t.CInt: pass + +def resolve_generic_type_name(name: str) -> str: pass + +def _get_type_name_from_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _parse_type_string(pool: memhub.MemBuddy | t.CPtr, type_str: str, ctx: t.CInt) -> ast.AST | t.CPtr: pass + +def set_current_file(file_path: str) -> t.CInt: pass + + +_SRC_BUF_SIZE: t.CDefine = 262144 + +def _read_source_line(file_path: str, lineno: int) -> str: pass + +def fatal_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def fatal_type_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def map_t_type(pool: memhub.MemBuddy | t.CPtr, type_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_t_attr(node: ast.AST | t.CPtr, attr_name: str, imported_modules: str) -> int: pass + +def extract_class_name_from_annotation(node: ast.AST | t.CPtr, imported_modules: str) -> str: pass + +def resolve_annotation_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def has_decorator_marker(node: ast.AST | t.CPtr, marker_name: str) -> int: pass + +def is_pass_body(body: list[ast.AST | t.CPtr] | t.CPtr) -> int: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def infer_expr_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def infer_return_type(pool: memhub.MemBuddy | t.CPtr, body: list[ast.AST | t.CPtr] | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def build_param_types_str(pool: memhub.MemBuddy | t.CPtr, args_node: ast.AST | t.CPtr) -> str: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/0df65b8ed15664b0.pyi b/.transpyc_cache/0df65b8ed15664b0.pyi new file mode 100644 index 0000000..e33dc69 --- /dev/null +++ b/.transpyc_cache/0df65b8ed15664b0.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from hashlib.__sha1.py +Module: hashlib.__sha1 +""" + + +import string +import t, c + +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self: sha1) -> t.CInt: pass + def transform(self: sha1, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha1, s: str) -> t.CInt: pass + def final(self: sha1, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/0e1189314c937ef1.doc.json b/.transpyc_cache/0e1189314c937ef1.doc.json new file mode 100644 index 0000000..45712e2 --- /dev/null +++ b/.transpyc_cache/0e1189314c937ef1.doc.json @@ -0,0 +1 @@ +{"AnnAssignHandle": "AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针", "AnnAssignHandle._is_cdefine_annotation": "检测注解是否为 t.CDefine，返回 1 表示是，0 表示否", "AnnAssignHandle._extract_cdefine_int_value": "从值节点提取整数常量（仅支持 Constant INT）", "AnnAssignHandle.PreScan": "预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部", "AnnAssignHandle.Handle": "翻译 AnnAssign(target=Name, annotation=..., value=expr)"} \ No newline at end of file diff --git a/.transpyc_cache/0e1189314c937ef1.pyi b/.transpyc_cache/0e1189314c937ef1.pyi new file mode 100644 index 0000000..350baa1 --- /dev/null +++ b/.transpyc_cache/0e1189314c937ef1.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAnnAssign.py +Module: lib.core.Handles.HandlesAnnAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesExprCall as HandlesExprCall + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + def __init__(self: AnnAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def _is_cdefine_annotation(self: AnnAssignHandle, annot: ast.AST | t.CPtr) -> int: pass + def _extract_cdefine_int_value(self: AnnAssignHandle, val_node: ast.AST | t.CPtr) -> int: pass + def PreScan(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + def Handle(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/0eccc2393466d940.doc.json b/.transpyc_cache/0eccc2393466d940.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/0eccc2393466d940.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/0eccc2393466d940.pyi b/.transpyc_cache/0eccc2393466d940.pyi new file mode 100644 index 0000000..b4f6066 --- /dev/null +++ b/.transpyc_cache/0eccc2393466d940.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from augassign_test.py +Module: augassign_test +""" + + +import stdio +import t, c + +g_count: t.CExtern | int + +def test_local() -> int: pass + +def test_global_aug() -> int: pass + +def test_bitops() -> int: pass + +def augassign_test() -> int: pass diff --git a/.transpyc_cache/0f8134f67bdaba8f.doc.json b/.transpyc_cache/0f8134f67bdaba8f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/0f8134f67bdaba8f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/0f8134f67bdaba8f.pyi b/.transpyc_cache/0f8134f67bdaba8f.pyi new file mode 100644 index 0000000..3f2901b --- /dev/null +++ b/.transpyc_cache/0f8134f67bdaba8f.pyi @@ -0,0 +1,122 @@ +""" +Auto-generated Python stub file from __zlib.pyzlib.py +Module: __zlib.pyzlib +""" + + +from stdint import * +import zdeflate +import zinflate +import zchecksum +import zdef +import zhuff +import stdlib +import string +import t, c + +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) +DEFLATED: t.CDefine = 8 +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 +ZLIB_VERSION: t.CDefine = "1.3.2" +pyzlib_error_msg: t.CExtern | t.CArray[t.CChar, 512] +pyzlib_error_code_val: t.CExtern | t.CInt + +@t.Object +class Compress: + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + def compress(self: Compress, data: BYTEPTR, data_len: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Compress, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Compress) -> Compress | t.CPtr: pass + def delete(self: Compress) -> t.CInt: pass + def __del__(self: Compress) -> t.CInt: pass +@t.Object +class Decompress: + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + def decompress(self: Decompress, data: BYTEPTR, data_len: t.CSizeT, max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Decompress, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Decompress) -> Decompress | t.CPtr: pass + def delete(self: Decompress) -> t.CInt: pass + def unused_data(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def unconsumed_tail(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def set_error(code: int, msg: str) -> t.CInt: pass + +def clone_bytes(src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def append_bytes(buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: pass + +def shrink_to_fit(buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def runtime_version() -> str: pass + +def get_error() -> str: pass + +def get_error_code() -> t.CInt: pass + +def clear_error() -> t.CInt: pass + +def compress(data: BYTEPTR, data_len: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def decompress(data: BYTEPTR, data_len: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def compressobj(level: t.CInt, method: t.CInt, wbits: t.CInt, memLevel: t.CInt, strategy: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: pass + +def decompressobj(wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: pass + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass diff --git a/.transpyc_cache/0fef3cb77da3dae3.doc.json b/.transpyc_cache/0fef3cb77da3dae3.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/0fef3cb77da3dae3.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/0fef3cb77da3dae3.pyi b/.transpyc_cache/0fef3cb77da3dae3.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/0fef3cb77da3dae3.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/10ac4af023af7962.doc.json b/.transpyc_cache/10ac4af023af7962.doc.json new file mode 100644 index 0000000..45fcbeb --- /dev/null +++ b/.transpyc_cache/10ac4af023af7962.doc.json @@ -0,0 +1 @@ +{"set_generic_context": "设置泛型特化上下文（进入泛型类方法体翻译时调用）", "clear_generic_context": "清除泛型特化上下文（离开泛型类方法体翻译时调用）", "resolve_generic_type_name": "如果 name 是泛型类型参数，返回对应的具体类型名；否则返回 name 本身\n\n 用于表达式中的 T 替换（如 T.__sizeof__() → int.__sizeof__()）\n ", "_get_type_name_from_annotation": "从类型注解节点提取类型名（支持 Name/Attribute/BinOp）", "_parse_type_string": "将类型名字符串解析为 AST 节点", "set_current_file": "设置当前编译的文件名（供报错使用）", "_read_source_line": "从源文件读取第 lineno 行内容", "fatal_error": "以 Python 风格报错并退出（带源代码行）", "fatal_type_error": "以 Python 风格报错并退出（fatal_error 的别名）", "map_t_type": "将 t.CXxx / c.CXxx 类型名映射到 LLVMType\n\n 注意: t.CChar | t.CPtr = i8* = str = bytes\n ", "_is_t_attr": "检测节点是否是 t.attr_name 形式", "extract_class_name_from_annotation": "从类型注解 AST 中提取类名，返回 None=无法提取", "resolve_annotation_type": "解析类型注解节点为 LLVMType\n\n Args:\n pool: 内存分配器\n node: 类型注解 AST 节点（Name 或 Attribute）\n imported_modules: 已导入模块名字符串（空格分隔）\n from_imports: from-import 映射字符串（\"name:module name:module\" 格式）\n ", "has_decorator_marker": "检测类型注解中是否包含指定的装饰器标记", "is_pass_body": "检测函数体是否为 pass（空体或仅含 pass 语句）", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "infer_expr_type": "推断表达式的 LLVM 类型\n\n param_types 格式: \"name:bits,name:bits,...\"\n ", "infer_return_type": "扫描函数体中的 return 语句，推断返回类型", "build_param_types_str": "构建参数类型编码字符串 \"name:bits,name:bits,...\" ", "_strappend": "将 src 追加到 dst 末尾"} \ No newline at end of file diff --git a/.transpyc_cache/10ac4af023af7962.pyi b/.transpyc_cache/10ac4af023af7962.pyi new file mode 100644 index 0000000..d6de8a3 --- /dev/null +++ b/.transpyc_cache/10ac4af023af7962.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesType.py +Module: lib.core.Handles.HandlesType +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import stdlib +import viperlib +import w32.fileio as fileio +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesStruct as HandlesStruct + +_g_current_file: t.CExtern | str +_g_generic_tp_names: t.CExtern | list[str] | t.CPtr +_g_generic_type_args: t.CExtern | list[str] | t.CPtr + +def set_generic_context(tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> t.CInt: pass + +def clear_generic_context() -> t.CInt: pass + +def resolve_generic_type_name(name: str) -> str: pass + +def _get_type_name_from_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _parse_type_string(pool: memhub.MemBuddy | t.CPtr, type_str: str, ctx: t.CInt) -> ast.AST | t.CPtr: pass + +def set_current_file(file_path: str) -> t.CInt: pass + + +_SRC_BUF_SIZE: t.CDefine = 262144 + +def _read_source_line(file_path: str, lineno: int) -> str: pass + +def fatal_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def fatal_type_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def map_t_type(pool: memhub.MemBuddy | t.CPtr, type_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_t_attr(node: ast.AST | t.CPtr, attr_name: str, imported_modules: str) -> int: pass + +def extract_class_name_from_annotation(node: ast.AST | t.CPtr, imported_modules: str) -> str: pass + +def resolve_annotation_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def has_decorator_marker(node: ast.AST | t.CPtr, marker_name: str) -> int: pass + +def is_pass_body(body: list[ast.AST | t.CPtr] | t.CPtr) -> int: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def infer_expr_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def infer_return_type(pool: memhub.MemBuddy | t.CPtr, body: list[ast.AST | t.CPtr] | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def build_param_types_str(pool: memhub.MemBuddy | t.CPtr, args_node: ast.AST | t.CPtr) -> str: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/117b4962d332b4c5.doc.json b/.transpyc_cache/117b4962d332b4c5.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/117b4962d332b4c5.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/117b4962d332b4c5.pyi b/.transpyc_cache/117b4962d332b4c5.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/117b4962d332b4c5.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/1193c9cd16698da8.doc.json b/.transpyc_cache/1193c9cd16698da8.doc.json new file mode 100644 index 0000000..c3a1e25 --- /dev/null +++ b/.transpyc_cache/1193c9cd16698da8.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/1193c9cd16698da8.pyi b/.transpyc_cache/1193c9cd16698da8.pyi new file mode 100644 index 0000000..ad12e1f --- /dev/null +++ b/.transpyc_cache/1193c9cd16698da8.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 32 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/13110effbb0bb06c.pyi b/.transpyc_cache/13110effbb0bb06c.pyi new file mode 100644 index 0000000..3584bd2 --- /dev/null +++ b/.transpyc_cache/13110effbb0bb06c.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from os.path.py +Module: os.path +""" + + +import t, c +from stdint import * +import string +import memhub +import platmacro + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def sep() -> t.CChar: pass + +def _is_sep(ch: t.CChar) -> bool: pass + +def _dup_str(s: str) -> str: pass + +def _substr(s: str, start: t.CSizeT, end: t.CSizeT) -> str: pass + +def isabs(path: str) -> bool: pass + +def join(path1: str, path2: str) -> str: pass + +def basename(path: str) -> str: pass + +def dirname(path: str) -> str: pass + +def splitext_name(path: str) -> str: pass + +def splitext_ext(path: str) -> str: pass + +def normpath(path: str) -> str: pass diff --git a/.transpyc_cache/13d9449413372bde.pyi b/.transpyc_cache/13d9449413372bde.pyi new file mode 100644 index 0000000..c10ea89 --- /dev/null +++ b/.transpyc_cache/13d9449413372bde.pyi @@ -0,0 +1,93 @@ +""" +Auto-generated Python stub file from llvmlite.__function.py +Module: llvmlite.__function +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import stdio +from linkedlist import GSListNode, GSList +from .__types import LLVMType, ParamNode, TypePrint + +class Line(GSListNode[Line]): + Buf: t.CChar | t.CPtr + Len: t.CSizeT +class BasicBlock(GSListNode[BasicBlock]): + Name: t.CChar | t.CPtr + Lines: GSList[Line] | t.CPtr + IsTerminated: t.CInt +class Param(GSListNode[Param]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + Attrs: t.CChar | t.CPtr +class Function(GSListNode[Function]): + Name: t.CChar | t.CPtr + RetTy: LLVMType | t.CPtr + Params: GSList[Param] | t.CPtr + Blocks: GSList[BasicBlock] | t.CPtr + IsDeclared: t.CInt + IsVarArg: t.CInt + Attrs: t.CChar | t.CPtr + +LINE_SIZE: t.CDefine = 24 + +def new_line(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> Line | t.CPtr: pass + +def _new_gslist(pool: memhub.MemBuddy | t.CPtr, list_size: t.CSizeT) -> t.CPtr: pass + + +BLOCK_SIZE: t.CDefine = 32 + +def new_basic_block(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + + +PARAM_SIZE: t.CDefine = 32 + +def new_param(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + + +FUNC_SIZE: t.CDefine = 56 + +def new_function(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def function_add_param(func: Function | t.CPtr, param: Param | t.CPtr) -> t.CInt: pass + +def function_add_block(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def function_move_block_to_end(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def block_append_line(block: BasicBlock | t.CPtr, line: Line | t.CPtr) -> t.CInt: pass + +def block_append_text(pool: memhub.MemBuddy | t.CPtr, block: BasicBlock | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def function_set_attrs(func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def param_set_attrs(param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def _ll_name_needs_quote(name: str) -> t.CInt: pass + +def FunctionPrint(buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def function_get_name(func: Function | t.CPtr) -> t.CChar | t.CPtr: pass + +def function_get_ret_ty(func: Function | t.CPtr) -> LLVMType | t.CPtr: pass + +def function_get_params(func: Function | t.CPtr) -> GSList[Param] | t.CPtr: pass + +def function_get_param_head(func: Function | t.CPtr) -> Param | t.CPtr: pass + +def function_get_next(func: Function | t.CPtr) -> Function | t.CPtr: pass + +def function_is_declared(func: Function | t.CPtr) -> t.CInt: pass + +def param_get_name(param: Param | t.CPtr) -> t.CChar | t.CPtr: pass + +def param_get_ty(param: Param | t.CPtr) -> LLVMType | t.CPtr: pass + +def param_get_next(param: Param | t.CPtr) -> Param | t.CPtr: pass diff --git a/.transpyc_cache/1446e90e8296beae.doc.json b/.transpyc_cache/1446e90e8296beae.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/1446e90e8296beae.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/1446e90e8296beae.pyi b/.transpyc_cache/1446e90e8296beae.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/1446e90e8296beae.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/146a976893eef52b.doc.json b/.transpyc_cache/146a976893eef52b.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/146a976893eef52b.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/146a976893eef52b.pyi b/.transpyc_cache/146a976893eef52b.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/146a976893eef52b.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/14a9b576ab5d21c9.doc.json b/.transpyc_cache/14a9b576ab5d21c9.doc.json new file mode 100644 index 0000000..97ecb08 --- /dev/null +++ b/.transpyc_cache/14a9b576ab5d21c9.doc.json @@ -0,0 +1 @@ +{"local_func": "Local function docstring.", "LocalClass": "Local class docstring.", "LocalClass.local_method": "Local method docstring.", "_check_doc_adaptive": "自适应检查 docstring：非空时验证匹配，空时视为优化剥离", "_check_doc_null": "检查 docstring 为 null（无 docstring 的符号）"} \ No newline at end of file diff --git a/.transpyc_cache/14a9b576ab5d21c9.pyi b/.transpyc_cache/14a9b576ab5d21c9.pyi new file mode 100644 index 0000000..c16365f --- /dev/null +++ b/.transpyc_cache/14a9b576ab5d21c9.pyi @@ -0,0 +1,32 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import c +import testcheck +import remote + +def local_func() -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass + + +@t.Object +class LocalClass: + value: t.CInt + def local_method(self: LocalClass) -> t.CInt: pass + +def _check_doc_adaptive(name: str, doc: str, expected: str) -> t.CInt: pass + +def _check_doc_null(name: str, doc: str) -> t.CInt: pass + +def test_local_doc() -> t.CInt: pass + +def test_cross_module_doc() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/16e1f65b154831e8.doc.json b/.transpyc_cache/16e1f65b154831e8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/16e1f65b154831e8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/16e1f65b154831e8.pyi b/.transpyc_cache/16e1f65b154831e8.pyi new file mode 100644 index 0000000..3167de1 --- /dev/null +++ b/.transpyc_cache/16e1f65b154831e8.pyi @@ -0,0 +1,64 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def add[T](a: T, b: T) -> T: pass + +def test_generic_add() -> t.CInt: pass + +def combine[T1, T2](a: T1, b: T2) -> T1: pass + +def test_generic_combine() -> t.CInt: pass + +def max_val[T](a: T, b: T) -> T: pass + +def test_generic_max() -> t.CInt: pass + + +class Box[T]: + value: T + def __init__(self: Box, v: T) -> t.CInt: pass + def get(self: Box) -> T: pass + def set(self: Box, v: T) -> t.CInt: pass + +def test_generic_box() -> t.CInt: pass + + +class Pair[T1, T2]: + first: T1 + second: T2 + def __init__(self: Pair, a: T1, b: T2) -> t.CInt: pass + def sum_as_int(self: Pair) -> t.CInt: pass + +def test_generic_pair() -> t.CInt: pass + +def test_fixed_array() -> t.CInt: pass + +def test_char_array() -> t.CInt: pass + +def identity[T](x: T) -> T: pass + +def test_generic_identity() -> t.CInt: pass + +def double_val[T](x: T) -> T: pass + +def test_generic_nested() -> t.CInt: pass + + +class Counter[T]: + count: T + def __init__(self: Counter, start: T) -> t.CInt: pass + def increment(self: Counter) -> t.CInt: pass + def get_count(self: Counter) -> T: pass + +def test_generic_counter() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/17c0dfbcebfa3a0e.doc.json b/.transpyc_cache/17c0dfbcebfa3a0e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/17c0dfbcebfa3a0e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/17c0dfbcebfa3a0e.pyi b/.transpyc_cache/17c0dfbcebfa3a0e.pyi new file mode 100644 index 0000000..fd38559 --- /dev/null +++ b/.transpyc_cache/17c0dfbcebfa3a0e.pyi @@ -0,0 +1,71 @@ +""" +Auto-generated Python stub file from attr_test.py +Module: attr_test +""" + + +import t, c +from stdint import * +import stdio +import string + +@c.Attribute(t.attr.always_inline()) +def AlwaysInlineFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.noinline()) +def NoInlineFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.noreturn()) +def NoReturnFunc() -> t.CInt: pass + +def NormalFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.always_inline(), t.attr.pure()) +def MultiAttrFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.packed) +def PackedAttrFunc(x: t.CInt) -> t.CInt: pass + +def test_always_inline() -> t.CInt: pass + +def test_noinline() -> t.CInt: pass + +def test_noreturn() -> t.CInt: pass + +def test_normal() -> t.CInt: pass + +def test_multi_attr() -> t.CInt: pass + +def test_packed_attr() -> t.CInt: pass + +@c.Attribute(t.attr.llvm.nounwind) +def NoUnwindFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.llvm.noredzone) +def NoRedZoneFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.llvm.willreturn) +def WillReturnFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.llvm.mustprogress) +def MustProgressFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.const()) +def ConstFunc(x: t.CInt) -> t.CInt: pass + +@c.Attribute(t.attr.always_inline(), t.attr.llvm.nounwind, t.attr.llvm.noredzone) +def TripleAttrFunc(x: t.CInt) -> t.CInt: pass + +def test_nounwind() -> t.CInt: pass + +def test_noredzone() -> t.CInt: pass + +def test_willreturn() -> t.CInt: pass + +def test_mustprogress() -> t.CInt: pass + +def test_const_attr() -> t.CInt: pass + +def test_triple_attr() -> t.CInt: pass + +def attr_test() -> t.CInt: pass diff --git a/.transpyc_cache/19b6605d568c8584.doc.json b/.transpyc_cache/19b6605d568c8584.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/19b6605d568c8584.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/19b6605d568c8584.pyi b/.transpyc_cache/19b6605d568c8584.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/19b6605d568c8584.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/19f8024d10c828e8.pyi b/.transpyc_cache/19f8024d10c828e8.pyi new file mode 100644 index 0000000..a25a2e4 --- /dev/null +++ b/.transpyc_cache/19f8024d10c828e8.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from hashlib.__md5.py +Module: hashlib.__md5 +""" + + +import string +import t, c + +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + + +md5_T: t.CExtern | t.CArray[t.CUInt32T, 64] +md5_S: t.CExtern | t.CArray[t.CInt, 64] + +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] + def __init__(self: md5) -> t.CInt: pass + def transform(self: md5, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: md5, s: str) -> t.CInt: pass + def final(self: md5, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/1a3af346a22d2cb6.doc.json b/.transpyc_cache/1a3af346a22d2cb6.doc.json new file mode 100644 index 0000000..f6225ef --- /dev/null +++ b/.transpyc_cache/1a3af346a22d2cb6.doc.json @@ -0,0 +1 @@ +{"translate_stmt": "翻译单条语句，返回新增的变量数", "translate_global": "翻译 global 语句：记录 global 变量名", "translate_nonlocal": "翻译 nonlocal 语句：记录 nonlocal 变量名", "translate_expr_stmt": "翻译表达式语句（printf 调用等）", "pre_scan_allocas": "预扫描语句中的 AnnAssign，提前创建 alloca\n\n 返回新增的变量数\n ", "translate_break": "翻译 break 语句：跳转到循环 end 块", "translate_continue": "翻译 continue 语句：跳转到循环 cond/incr 块", "get_stmt_kind_name": "获取语句类型名"} \ No newline at end of file diff --git a/.transpyc_cache/1a3af346a22d2cb6.pyi b/.transpyc_cache/1a3af346a22d2cb6.pyi new file mode 100644 index 0000000..1eacb6d --- /dev/null +++ b/.transpyc_cache/1a3af346a22d2cb6.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBody.py +Module: lib.core.Handles.HandlesBody +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def translate_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_global(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nonlocal(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_expr_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def pre_scan_allocas(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_break(trans: HT.Translator | t.CPtr) -> int: pass + +def translate_continue(trans: HT.Translator | t.CPtr) -> int: pass + +def get_stmt_kind_name(node: ast.AST | t.CPtr) -> str: pass diff --git a/.transpyc_cache/1b58766d38d05de3.doc.json b/.transpyc_cache/1b58766d38d05de3.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/1b58766d38d05de3.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/1b58766d38d05de3.pyi b/.transpyc_cache/1b58766d38d05de3.pyi new file mode 100644 index 0000000..11fb572 --- /dev/null +++ b/.transpyc_cache/1b58766d38d05de3.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from enumtest.py +Module: enumtest +""" + + +import t, c +from t import CInt, CPtr, CChar, CInt32T, CExport +import viperlib +import stdlib + +class Color(t.CEnum): + Red = 0 + Green = 1 + Blue = 2 +class Option(t.REnum): + class Some: + value: CInt + Empty = 1 +class Result(t.REnum): + class Ok: + value: CInt + class Err: + msg: t.CChar | CPtr +class GeoShape(t.REnum): + class GCircle: + radius: CInt + class GRect: + w: CInt + h: CInt + class GPoint: + x: CInt + y: CInt + +def enum_main() -> CInt: pass diff --git a/.transpyc_cache/1b9ab264a942c56b.doc.json b/.transpyc_cache/1b9ab264a942c56b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/1b9ab264a942c56b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/1b9ab264a942c56b.pyi b/.transpyc_cache/1b9ab264a942c56b.pyi new file mode 100644 index 0000000..e5dacdd --- /dev/null +++ b/.transpyc_cache/1b9ab264a942c56b.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from oop_test.py +Module: oop_test +""" + + +import stdio +import t, c + +class Point: + x: t.CInt + y: t.CInt + def MoveTo(self: Point, nx: t.CInt, ny: t.CInt) -> t.CInt: pass + def GetX(self: Point) -> t.CInt: pass + def GetY(self: Point) -> t.CInt: pass +class Counter: + count: t.CInt = 0 + step: t.CInt = 1 + def Increment(self: Counter) -> t.CInt: pass + def Reset(self: Counter) -> t.CInt: pass +class Rect: + width: t.CInt + height: t.CInt + def __init__(self: Rect, w: t.CInt, h: t.CInt) -> t.CInt: pass + def Area(self: Rect) -> t.CInt: pass + +def oop_test() -> int: pass diff --git a/.transpyc_cache/1c46d554b3a3f9f3.pyi b/.transpyc_cache/1c46d554b3a3f9f3.pyi new file mode 100644 index 0000000..46fe651 --- /dev/null +++ b/.transpyc_cache/1c46d554b3a3f9f3.pyi @@ -0,0 +1,125 @@ +""" +Auto-generated Python stub file from zlib.pyzlib.py +Module: zlib.pyzlib +""" + + +from stdint import * +import zlib.zdeflate as zdeflate +import zlib.zinflate as zinflate +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import stdlib +import string +import memhub +import t, c + +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) +DEFLATED: t.CDefine = 8 +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 +ZLIB_VERSION: t.CDefine = "1.3.2" +pyzlib_error_msg: t.CExtern | t.CArray[t.CChar, 512] +pyzlib_error_code_val: t.CExtern | t.CInt + +@t.Object +class Compress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + def compress(self: Compress, data: BYTEPTR, data_len: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Compress, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Compress) -> Compress | t.CPtr: pass + def delete(self: Compress) -> t.CInt: pass + def __del__(self: Compress) -> t.CInt: pass +@t.Object +class Decompress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + def decompress(self: Decompress, data: BYTEPTR, data_len: t.CSizeT, max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Decompress, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Decompress) -> Decompress | t.CPtr: pass + def delete(self: Decompress) -> t.CInt: pass + def unused_data(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def unconsumed_tail(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def set_error(code: int, msg: str) -> t.CInt: pass + +def clone_bytes(pool: memhub.MemManager | t.CPtr, src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def append_bytes(pool: memhub.MemManager | t.CPtr, buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: pass + +def shrink_to_fit(pool: memhub.MemManager | t.CPtr, buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def runtime_version() -> str: pass + +def get_error() -> str: pass + +def get_error_code() -> t.CInt: pass + +def clear_error() -> t.CInt: pass + +def compress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def decompress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def compressobj(pool: memhub.MemManager | t.CPtr, level: t.CInt, method: t.CInt, wbits: t.CInt, memLevel: t.CInt, strategy: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: pass + +def decompressobj(pool: memhub.MemManager | t.CPtr, wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: pass + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass diff --git a/.transpyc_cache/1c8a1eceb26a43f4.doc.json b/.transpyc_cache/1c8a1eceb26a43f4.doc.json new file mode 100644 index 0000000..9c0ba29 --- /dev/null +++ b/.transpyc_cache/1c8a1eceb26a43f4.doc.json @@ -0,0 +1 @@ +{"AnnAssignHandle": "AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针", "AnnAssignHandle.PreScan": "预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部", "AnnAssignHandle.Handle": "翻译 AnnAssign(target=Name, annotation=..., value=expr)"} \ No newline at end of file diff --git a/.transpyc_cache/1c8a1eceb26a43f4.pyi b/.transpyc_cache/1c8a1eceb26a43f4.pyi new file mode 100644 index 0000000..beb4d8d --- /dev/null +++ b/.transpyc_cache/1c8a1eceb26a43f4.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAnnAssign.py +Module: lib.core.Handles.HandlesAnnAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + def __init__(self: AnnAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def PreScan(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + def Handle(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/1df40188ee73ee05.doc.json b/.transpyc_cache/1df40188ee73ee05.doc.json new file mode 100644 index 0000000..4dba15e --- /dev/null +++ b/.transpyc_cache/1df40188ee73ee05.doc.json @@ -0,0 +1 @@ +{"ReturnHandle": "Return 语句处理器：继承 Mixin 获得 Trans 回指针", "ReturnHandle.Handle": "翻译 Return 语句"} \ No newline at end of file diff --git a/.transpyc_cache/1df40188ee73ee05.pyi b/.transpyc_cache/1df40188ee73ee05.pyi new file mode 100644 index 0000000..abcbbff --- /dev/null +++ b/.transpyc_cache/1df40188ee73ee05.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesReturn.py +Module: lib.core.Handles.HandlesReturn +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar + +@t.NoVTable +class ReturnHandle(HandlesBase.Mixin): + def __init__(self: ReturnHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ReturnHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewReturnHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ReturnHandle | t.CPtr: pass diff --git a/.transpyc_cache/1e19f516dc857768.doc.json b/.transpyc_cache/1e19f516dc857768.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/1e19f516dc857768.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/1e19f516dc857768.pyi b/.transpyc_cache/1e19f516dc857768.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/1e19f516dc857768.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/1eb1c9fc7e85d9f8.doc.json b/.transpyc_cache/1eb1c9fc7e85d9f8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/1eb1c9fc7e85d9f8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/1eb1c9fc7e85d9f8.pyi b/.transpyc_cache/1eb1c9fc7e85d9f8.pyi new file mode 100644 index 0000000..8139e3c --- /dev/null +++ b/.transpyc_cache/1eb1c9fc7e85d9f8.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from for_test.py +Module: for_test +""" + + +import stdio +import t, c + +def for_test() -> int: pass diff --git a/.transpyc_cache/1ed8ca4986cb0d65.doc.json b/.transpyc_cache/1ed8ca4986cb0d65.doc.json new file mode 100644 index 0000000..aee5ad5 --- /dev/null +++ b/.transpyc_cache/1ed8ca4986cb0d65.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/1ed8ca4986cb0d65.pyi b/.transpyc_cache/1ed8ca4986cb0d65.pyi new file mode 100644 index 0000000..ad13f52 --- /dev/null +++ b/.transpyc_cache/1ed8ca4986cb0d65.pyi @@ -0,0 +1,77 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/1ff2c5eb32359e40.doc.json b/.transpyc_cache/1ff2c5eb32359e40.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/1ff2c5eb32359e40.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/1ff2c5eb32359e40.pyi b/.transpyc_cache/1ff2c5eb32359e40.pyi new file mode 100644 index 0000000..dde14fd --- /dev/null +++ b/.transpyc_cache/1ff2c5eb32359e40.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdlib +import stdio +import memhub +import testcheck + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/2020b02f319dd87d.doc.json b/.transpyc_cache/2020b02f319dd87d.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/2020b02f319dd87d.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/2020b02f319dd87d.pyi b/.transpyc_cache/2020b02f319dd87d.pyi new file mode 100644 index 0000000..2294a70 --- /dev/null +++ b/.transpyc_cache/2020b02f319dd87d.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/20cd49775c100a38.pyi b/.transpyc_cache/20cd49775c100a38.pyi new file mode 100644 index 0000000..df04de7 --- /dev/null +++ b/.transpyc_cache/20cd49775c100a38.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from json.__writer.py +Module: json.__writer +""" + + +import t, c +from stdint import * +import memhub +import string +from stdio import snprintf +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT + +def _write_char(buf: t.CChar | t.CPtr, pos: t.CSizeT, ch: t.CChar) -> t.CSizeT: pass + +def _write_str(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_string_escaped(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_value(buf: t.CChar | t.CPtr, pos: t.CSizeT, val: JsonValue | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CSizeT: pass + +def write(pool: memhub.MemManager | t.CPtr, val: JsonValue | t.CPtr, pretty: bool) -> t.CChar | t.CPtr: pass diff --git a/.transpyc_cache/212fb1de4f9d6e5c.doc.json b/.transpyc_cache/212fb1de4f9d6e5c.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/212fb1de4f9d6e5c.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/212fb1de4f9d6e5c.pyi b/.transpyc_cache/212fb1de4f9d6e5c.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/212fb1de4f9d6e5c.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/213348433fb01cc6.pyi b/.transpyc_cache/213348433fb01cc6.pyi new file mode 100644 index 0000000..6e4edce --- /dev/null +++ b/.transpyc_cache/213348433fb01cc6.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from zlib.zdef.py +Module: zlib.zdef +""" + + +from stdint import * +import stddef +import string +import stdlib +import memhub +import t, c + +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 +zdeflate_len_extra_bits: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_len_base: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_dist_extra_bits: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_dist_base: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_codelen_order: t.CExtern | t.CArray[int, 19] + +@t.Object +class zbit_writer: + pool: memhub.MemManager | t.CPtr + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def __init__(self: zbit_writer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def ensure(self: zbit_writer, need: t.CSizeT) -> t.CInt: pass + def write_bits(self: zbit_writer, value: UINT, nbits: t.CInt) -> t.CInt: pass + def write_bits_rev(self: zbit_writer, code: UINT, nbits: t.CInt) -> t.CInt: pass + def stored_block(self: zbit_writer, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_zlib_header(self: zbit_writer, wbits: t.CInt, level: t.CInt) -> t.CInt: pass + def write_gzip_header(self: zbit_writer) -> t.CInt: pass + def align(self: zbit_writer) -> t.CInt: pass + def total(self: zbit_writer) -> t.CSizeT: pass + def free(self: zbit_writer) -> t.CInt: pass + def __del__(self: zbit_writer) -> t.CInt: pass +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def init(self: zbit_reader, buf: BYTE | t.CPtr, length: t.CSizeT) -> t.CInt: pass + def read_bits(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_bits_rev(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def align(self: zbit_reader) -> t.CInt: pass + +def zdef_alloc(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> VOIDPTR: pass + +def zdef_free(pool: memhub.MemManager | t.CPtr, p: VOIDPTR) -> t.CInt: pass diff --git a/.transpyc_cache/21a7fcfc665f75ef.pyi b/.transpyc_cache/21a7fcfc665f75ef.pyi new file mode 100644 index 0000000..6f344a3 --- /dev/null +++ b/.transpyc_cache/21a7fcfc665f75ef.pyi @@ -0,0 +1,87 @@ +""" +Auto-generated Python stub file from llvmlite.__module.py +Module: llvmlite.__module +""" + + +import t, c +from stdint import * +import string +import stdio +import viperlib +import memhub +from linkedlist import GSList, GSListNode +from .__types import LLVMType, TypePrint, Array, PrintStructDefinition, get_struct_name, Struct +from .__function import Function, FunctionPrint, _ll_name_needs_quote, Param + +@t.NoVTable +class GlobalVariable(GSListNode[GlobalVariable]): + Name: t.CChar | t.CPtr + Ty: LLVMType | t.CPtr + Initializer: t.CVoid | t.CPtr + IsConstant: t.CInt + Linkage: t.CChar | t.CPtr + Section: t.CChar | t.CPtr + UnnamedAddr: t.CInt +@t.NoVTable +class NamedTypeNode(GSListNode[NamedTypeNode]): + Ty: LLVMType | t.CPtr +@t.NoVTable +class LLVMModule: + Name: t.CChar | t.CPtr + FuncHead: Function | t.CPtr + FuncTail: Function | t.CPtr + FuncCount: t.CSizeT + GlobalHead: GlobalVariable | t.CPtr + GlobalTail: GlobalVariable | t.CPtr + GlobalCount: t.CSizeT + TargetTriple: t.CChar | t.CPtr + DataLayout: t.CChar | t.CPtr + NamedTypeHead: NamedTypeNode | t.CPtr + NamedTypeTail: NamedTypeNode | t.CPtr + NamedTypeCount: t.CSizeT + +def new_module(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + + +GV_SIZE: t.CDefine = 64 +NAMED_TYPE_NODE_SIZE: t.CDefine = 16 + +def new_global_variable(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def module_add_function(mod: LLVMModule | t.CPtr, func: Function | t.CPtr) -> t.CInt: pass + +def module_add_global(mod: LLVMModule | t.CPtr, gv: GlobalVariable | t.CPtr) -> t.CInt: pass + +def module_set_target(mod: LLVMModule | t.CPtr, triple: t.CChar | t.CPtr) -> t.CInt: pass + +def module_set_datalayout(mod: LLVMModule | t.CPtr, layout: t.CChar | t.CPtr) -> t.CInt: pass + +def module_add_named_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def _is_type_defined_in_module(mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr) -> int: pass + +def _scan_type_for_cross_module_refs(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def module_ensure_opaque_for_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def global_set_constant(gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def global_set_linkage(gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + +def global_set_unnamed_addr(gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + +def global_set_section(gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + + +OUTPUT_FULL: t.CDefine = 0 # 完整 IR（stub + text） +OUTPUT_STUB: t.CDefine = 1 # 仅声明（header + 类型定义 + declare + external global） +OUTPUT_TEXT: t.CDefine = 2 # 仅代码（define + global 带初值） + +def LLVMModulePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mode: t.CInt) -> t.CInt: pass + +def _print_global(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _print_global_stub(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/21fffba3315e81b4.doc.json b/.transpyc_cache/21fffba3315e81b4.doc.json new file mode 100644 index 0000000..f6553a0 --- /dev/null +++ b/.transpyc_cache/21fffba3315e81b4.doc.json @@ -0,0 +1 @@ +{"Logger.banner": "输出分节标题（青色高亮）。", "Logger.compile_error": "格式化编译错误输出（红底白字标题 + 位置 + 错误信息）。", "Logger.compile_warning": "格式化编译警告输出（黄底黑字标题 + 位置 + 警告信息）。", "get_logger": "获取全局 logger 实例。若不存在则通过 _mbuddy 分配并初始化。", "set_logger": "设置全局 logger 实例。"} \ No newline at end of file diff --git a/.transpyc_cache/21fffba3315e81b4.pyi b/.transpyc_cache/21fffba3315e81b4.pyi new file mode 100644 index 0000000..04be3c9 --- /dev/null +++ b/.transpyc_cache/21fffba3315e81b4.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.VLogger.py +Module: lib.core.VLogger +""" + + +import t, c +from stdint import * +import stdio +import memhub +import w32.win32console as w32cmd +import w32.win32file as w32file +import w32.win32base as w32base + +class LogLevel(t.CEnum): + DEBUG = 0 + INFO = 1 + WARNING = 2 + ERROR = 3 + SUCCESS = 4 + +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +FOREGROUND_WHITE: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE +FOREGROUND_CYAN: t.CDefine = FOREGROUND_BLUE | FOREGROUND_GREEN +FOREGROUND_YELLOW: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN +FOREGROUND_MAGENTA: t.CDefine = FOREGROUND_RED | FOREGROUND_BLUE +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +BACKGROUND_WHITE: t.CDefine = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE + +class Logger: + _level: int + _console_handle: w32base.HANDLE + _use_color: int + __mbuddy__: memhub.MemManager | t.CPtr + def __init__(self: Logger, level: int) -> t.CInt: pass + def _set_color(self: Logger, attr: WORD) -> int: pass + def _reset_color(self: Logger) -> int: pass + def _log(self: Logger, level: int, prefix: str, msg: str, color: WORD, category: str) -> int: pass + def debug(self: Logger, msg: str, category: str) -> int: pass + def info(self: Logger, msg: str, category: str) -> int: pass + def warning(self: Logger, msg: str, category: str) -> int: pass + def error(self: Logger, msg: str, category: str) -> int: pass + def success(self: Logger, msg: str, category: str) -> int: pass + def set_level(self: Logger, level: int) -> int: pass + def banner(self: Logger, msg: str) -> int: pass + def compile_error(self: Logger, msg: str, file: str, line: int) -> int: pass + def compile_warning(self: Logger, msg: str, file: str, line: int) -> int: pass + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +_g_logger: t.CExtern | Logger | t.CPtr + +def get_logger() -> Logger | t.CPtr: pass + +def set_logger(logger: Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/22650858c377c847.doc.json b/.transpyc_cache/22650858c377c847.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/22650858c377c847.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/22650858c377c847.pyi b/.transpyc_cache/22650858c377c847.pyi new file mode 100644 index 0000000..24552b0 --- /dev/null +++ b/.transpyc_cache/22650858c377c847.pyi @@ -0,0 +1,64 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck +import linkedlist +import viperlib +import stdlib + +class Node(linkedlist.LinkedNode): + value: t.CInt + +def test_append() -> t.CInt: pass + +def test_prev_link() -> t.CInt: pass + +def test_last_child() -> t.CInt: pass + +def test_remove() -> t.CInt: pass + +def test_remove_child() -> t.CInt: pass + +def test_insert_ops() -> t.CInt: pass + +def test_index_access() -> t.CInt: pass + +def test_sibling_traverse() -> t.CInt: pass + +def test_unlink() -> t.CInt: pass + + +class SNode(linkedlist.SListNode): + value: t.CInt + +def test_slist() -> t.CInt: pass + +def test_sprintf_vararg() -> t.CInt: pass + + +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +class GNode(GSListNode[GNode]): + Value: t.CInt + +def test_generic_slist() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/240a9a4157959a9f.pyi b/.transpyc_cache/240a9a4157959a9f.pyi new file mode 100644 index 0000000..c230800 --- /dev/null +++ b/.transpyc_cache/240a9a4157959a9f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from json.__parser.py +Module: json.__parser +""" + + +import t, c +from stdint import * +import memhub +import string +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT, null, bool_val, int_val, float_val, string_val, array, object + +class ParseState: + src: t.CChar | t.CPtr + pos: t.CSizeT + len: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: ParseState, src: t.CChar | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(s: ParseState | t.CPtr) -> t.CChar: pass + +def _advance(s: ParseState | t.CPtr) -> t.CChar: pass + +def _skip_whitespace(s: ParseState | t.CPtr) -> t.CInt: pass + +def _expect(s: ParseState | t.CPtr, ch: t.CChar) -> bool: pass + +def _parse_string(s: ParseState | t.CPtr) -> t.CChar | t.CPtr: pass + +def _parse_number(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_value(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_array(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_object(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def parse(pool: memhub.MemManager | t.CPtr, src: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass diff --git a/.transpyc_cache/249b0b9083e25ad2.doc.json b/.transpyc_cache/249b0b9083e25ad2.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/.transpyc_cache/249b0b9083e25ad2.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/.transpyc_cache/249b0b9083e25ad2.pyi b/.transpyc_cache/249b0b9083e25ad2.pyi new file mode 100644 index 0000000..d866b3e --- /dev/null +++ b/.transpyc_cache/249b0b9083e25ad2.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/24b9e251de12bd96.doc.json b/.transpyc_cache/24b9e251de12bd96.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/24b9e251de12bd96.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/24b9e251de12bd96.pyi b/.transpyc_cache/24b9e251de12bd96.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/24b9e251de12bd96.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/24e26a8de1173993.doc.json b/.transpyc_cache/24e26a8de1173993.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/24e26a8de1173993.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/24e26a8de1173993.pyi b/.transpyc_cache/24e26a8de1173993.pyi new file mode 100644 index 0000000..0053550 --- /dev/null +++ b/.transpyc_cache/24e26a8de1173993.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +class Point(t.Struct): + x: t.CInt + y: t.CInt +class Rect(t.Struct): + origin: Point + size: Point +class Node(t.Struct): + value: t.CInt + next: t.CPtr | t.CVoid + +def test_struct_basic() -> t.CInt: pass + +def test_struct_nested() -> t.CInt: pass + +def test_struct_pointer() -> t.CInt: pass + +def test_struct_param() -> t.CInt: pass + +def point_sum(p: Point) -> t.CInt: pass + +def test_struct_array() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/257e074b935c33b4.pyi b/.transpyc_cache/257e074b935c33b4.pyi new file mode 100644 index 0000000..34d2feb --- /dev/null +++ b/.transpyc_cache/257e074b935c33b4.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from binascii.py +Module: binascii +""" + + +import t, c + +HEX_CHARS: t.CExtern | t.CArray[t.CChar, None] +HEX_VALS: t.CExtern | t.CArray[t.CInt, 256] +B64_CHARS: t.CExtern | t.CArray[t.CChar, None] +B64_VALS: t.CExtern | t.CArray[t.CInt, 256] +HQX_CHARS: t.CExtern | t.CArray[t.CChar, 64] +HQX_VALS: t.CExtern | t.CArray[t.CInt, 256] +UU_CHARS: t.CExtern | t.CArray[t.CChar, None] +CRC32_TABLE: t.CExtern | t.CArray[t.CUnsignedInt, 256] +CRC_HQX_TABLE: t.CExtern | t.CArray[t.CUnsignedShort, 256] + +def init_binascii() -> t.CInt: pass + +def crc32(data: str, length: t.CInt, crc: t.CUnsignedInt) -> t.CUnsignedInt: pass + +def crc_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort) -> t.CUnsignedShort: pass + +def hexlify(data: str, length: t.CInt) -> str: pass + +def b2a_hex(data: str, length: t.CInt) -> str: pass + +def a2b_hex(hexstr: str, length: t.CInt) -> str: pass + +def unhexlify(data: str, length: t.CInt) -> t.CInt: pass + +def b2a_base64(data: str, length: t.CInt) -> str: pass + +def a2b_base64(data: str, length: t.CInt) -> str: pass + +def b2a_uu(data: str, length: t.CInt) -> str: pass + +def a2b_uu(data: str) -> str: pass + +def b2a_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort) -> tuple[str, t.CUnsignedShort]: pass + +def a2b_hqx(data: str, crc: t.CUnsignedShort) -> tuple[str, t.CUnsignedShort]: pass + +def adler32(data: str, length: t.CInt, adler: t.CUnsignedInt) -> t.CUnsignedInt: pass diff --git a/.transpyc_cache/25c18686adb93279.doc.json b/.transpyc_cache/25c18686adb93279.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/25c18686adb93279.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/25c18686adb93279.pyi b/.transpyc_cache/25c18686adb93279.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/25c18686adb93279.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/271ea3decb810db2.pyi b/.transpyc_cache/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/.transpyc_cache/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/.transpyc_cache/2769dee537308b00.doc.json b/.transpyc_cache/2769dee537308b00.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/2769dee537308b00.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/2769dee537308b00.pyi b/.transpyc_cache/2769dee537308b00.pyi new file mode 100644 index 0000000..2d8296e --- /dev/null +++ b/.transpyc_cache/2769dee537308b00.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from vtable_test.py +Module: vtable_test +""" + + +import stdio +import t, c + +@t.CVTable +class Animal: + name: t.CInt + def __init__(self: Animal, n: t.CInt) -> t.CInt: pass + def GetName(self: Animal) -> t.CInt: pass + def Speak(self: Animal) -> t.CInt: pass +@t.CVTable +class Shape: + sides: t.CInt + def SetSides(self: Shape, n: t.CInt) -> t.CInt: pass + def GetSides(self: Shape) -> t.CInt: pass + +def vtable_test() -> int: pass diff --git a/.transpyc_cache/27b003cb178b709a.doc.json b/.transpyc_cache/27b003cb178b709a.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/27b003cb178b709a.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/27b003cb178b709a.pyi b/.transpyc_cache/27b003cb178b709a.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/27b003cb178b709a.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/2998b8f78d7022bf.doc.json b/.transpyc_cache/2998b8f78d7022bf.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/.transpyc_cache/2998b8f78d7022bf.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/.transpyc_cache/2998b8f78d7022bf.pyi b/.transpyc_cache/2998b8f78d7022bf.pyi new file mode 100644 index 0000000..865ebf6 --- /dev/null +++ b/.transpyc_cache/2998b8f78d7022bf.pyi @@ -0,0 +1,90 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + CurrentPackage: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/2ae1bee155fb09d3.doc.json b/.transpyc_cache/2ae1bee155fb09d3.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/.transpyc_cache/2ae1bee155fb09d3.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/.transpyc_cache/2ae1bee155fb09d3.pyi b/.transpyc_cache/2ae1bee155fb09d3.pyi new file mode 100644 index 0000000..24ef0f1 --- /dev/null +++ b/.transpyc_cache/2ae1bee155fb09d3.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: str +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: str) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/.transpyc_cache/2c8ff1a30ee46867.doc.json b/.transpyc_cache/2c8ff1a30ee46867.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/2c8ff1a30ee46867.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/2c8ff1a30ee46867.pyi b/.transpyc_cache/2c8ff1a30ee46867.pyi new file mode 100644 index 0000000..c495a8e --- /dev/null +++ b/.transpyc_cache/2c8ff1a30ee46867.pyi @@ -0,0 +1,84 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/2d39c6c7d3557b3e.pyi b/.transpyc_cache/2d39c6c7d3557b3e.pyi new file mode 100644 index 0000000..fc259bd --- /dev/null +++ b/.transpyc_cache/2d39c6c7d3557b3e.pyi @@ -0,0 +1,63 @@ +""" +Auto-generated Python stub file from linkedlist.py +Module: linkedlist +""" + + +import t, c +from stdint import * + +@t.NoVTable +class LinkedNode: + next: 'LinkedNode' | t.CPtr + prev: 'LinkedNode' | t.CPtr + child: 'LinkedNode' | t.CPtr + last_child: 'LinkedNode' | t.CPtr + parent: 'LinkedNode' | t.CPtr + child_count: t.CSizeT + def append(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def prepend(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_before(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_after(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def remove_child(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def detach(self: LinkedNode) -> t.CInt: pass + def unlink(self: LinkedNode) -> t.CInt: pass + def has_children(self: LinkedNode) -> t.CInt: pass + def child_at(self: LinkedNode, index: t.CSizeT) -> 'LinkedNode' | t.CPtr: pass + def count_children(self: LinkedNode) -> t.CSizeT: pass + def first_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass + def last_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass +@t.NoVTable +class SListNode: + Next: 'SListNode' | t.CPtr + def append(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def append_after(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def count(self: SListNode) -> t.CSizeT: pass + def at(self: SListNode, index: t.CSizeT) -> 'SListNode' | t.CPtr: pass + def remove(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +@t.NoVTable +class GTreeNode[T]: + Next: T | t.CPtr + Prev: T | t.CPtr + Child: T | t.CPtr + LastChild: T | t.CPtr + Parent: T | t.CPtr + Count: t.CSizeT + def append(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def prepend(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def detach(self: GTreeNode) -> t.CInt: pass + def has_children(self: GTreeNode) -> t.CInt: pass + def child_at(self: GTreeNode, index: t.CSizeT) -> T | t.CPtr: pass + def count_children(self: GTreeNode) -> t.CSizeT: pass \ No newline at end of file diff --git a/.transpyc_cache/2da636c61863c815.pyi b/.transpyc_cache/2da636c61863c815.pyi new file mode 100644 index 0000000..7c91519 --- /dev/null +++ b/.transpyc_cache/2da636c61863c815.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from subprocess.py +Module: subprocess +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import w32.win32base +import w32.win32file +import w32.win32process +import w32.win32sync +import posix + +OUTPUT_BUF_SIZE: t.CDefine = 65536 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class CompletedProcess: + args: str + returncode: int + stdout: str + stderr: str + def __new__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CPtr: pass + def __init__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CInt: pass + +def run(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_win32(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_posix(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass diff --git a/.transpyc_cache/2ed380190e9df89e.doc.json b/.transpyc_cache/2ed380190e9df89e.doc.json new file mode 100644 index 0000000..2259eff --- /dev/null +++ b/.transpyc_cache/2ed380190e9df89e.doc.json @@ -0,0 +1 @@ +{"is_cdefine_annotation": "检测注解是否为 t.CDefine，返回 1 表示是，0 表示否", "extract_cdefine_int_value": "从值节点提取整数常量（仅支持 Constant INT）", "AnnAssignHandle": "AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针", "AnnAssignHandle.PreScan": "预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部", "AnnAssignHandle.Handle": "翻译 AnnAssign(target=Name, annotation=..., value=expr)"} \ No newline at end of file diff --git a/.transpyc_cache/2ed380190e9df89e.pyi b/.transpyc_cache/2ed380190e9df89e.pyi new file mode 100644 index 0000000..6d50c9b --- /dev/null +++ b/.transpyc_cache/2ed380190e9df89e.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAnnAssign.py +Module: lib.core.Handles.HandlesAnnAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesExprCall as HandlesExprCall + +def is_cdefine_annotation(annot: ast.AST | t.CPtr) -> int: pass + +def extract_cdefine_int_value(val_node: ast.AST | t.CPtr) -> int: pass + + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + def __init__(self: AnnAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def PreScan(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + def Handle(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/2fcb6780db4d937f.doc.json b/.transpyc_cache/2fcb6780db4d937f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/2fcb6780db4d937f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/2fcb6780db4d937f.pyi b/.transpyc_cache/2fcb6780db4d937f.pyi new file mode 100644 index 0000000..06ba45f --- /dev/null +++ b/.transpyc_cache/2fcb6780db4d937f.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import vipersimd +import numpy +from stdio import printf +import stdlib +from stdlib import calloc +import t, c +import memhub +import testcheck + +arena: t.CExtern | t.CUInt8T | t.CPtr +pool: t.CExtern | memhub.MemPool | t.CPtr + +def clock() -> t.CLong | t.CExtern: pass + + +CLOCK_PER_SEC: t.CDefine = 1000 + +def test_avx2_add_sub_mul_div() -> t.CInt: pass + +def test_avx2_neg_abs_sqrt() -> t.CInt: pass + +def test_avx2_min_max() -> t.CInt: pass + +def test_avx2_scalar_ops() -> t.CInt: pass + +def test_avx2_hsum_dot() -> t.CInt: pass + +def test_avx2_fma() -> t.CInt: pass + +def test_batch_array_ops() -> t.CInt: pass + +def test_llvmir_scalar() -> t.CInt: pass + + +N_ELEM: t.CDefine = 40000 +N_ITER: t.CDefine = 50000 + +def scalar_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def scalar_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def scalar_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def benchmark_simd_vs_scalar() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/30a88f26e7679645.doc.json b/.transpyc_cache/30a88f26e7679645.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/30a88f26e7679645.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/30a88f26e7679645.pyi b/.transpyc_cache/30a88f26e7679645.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/30a88f26e7679645.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/33a854a5c4f67d74.doc.json b/.transpyc_cache/33a854a5c4f67d74.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/33a854a5c4f67d74.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/33a854a5c4f67d74.pyi b/.transpyc_cache/33a854a5c4f67d74.pyi new file mode 100644 index 0000000..fdbb374 --- /dev/null +++ b/.transpyc_cache/33a854a5c4f67d74.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from struct_test.py +Module: struct_test +""" + + +import stdio +import t, c + +class Point: + x: t.CInt + y: t.CInt +class Box: + width: t.CInt + height: t.CInt + depth: t.CInt +class Size: + w: t.CInt = 100 + h: t.CInt = 200 +class State(t.CEnum): + Idle: t.State + Run: t.State + Stop: t.State +class Color(t.CEnum): + Red: t.State = 10 + Green: t.State + Blue: t.State = 20 + Yellow: t.State +class MixedType(t.CEnum): + Small: t.CInt8T + Big: t.CInt64T + Medium: t.CInt16T +class DataUnion(t.CUnion): + i: t.CInt + f: t.CFloat + l: t.CInt64T + +def struct_test() -> int: pass diff --git a/.transpyc_cache/34548789d646f432.pyi b/.transpyc_cache/34548789d646f432.pyi new file mode 100644 index 0000000..e046b0c --- /dev/null +++ b/.transpyc_cache/34548789d646f432.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from ast.__init__.py +Module: ast.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp, _init_tables +from .base import AST, ASTKind, ASTCtx, OpKind, ASTFlag, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC, FLAG_SIMPLE, FLAG_HAS_STAR, _init_ast, _copy_str, _set_pos, _inherit_pos, _set_parent_list, _binop_from_op, _augop_from_op, _cmpop_from_op, _emit, _emit_str, _emit_int, _dump_list, dump +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .astaux import ExceptHandler, Arguments, Arg, Keyword, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .visitor import ASTVisitor +from .lexer import Lexer, _lexer_init, tokenize, new_lexer +from .parser import Parser, _parser_init, parse_tokens, new_parser +from .parser import _parse_test + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def parse(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass + +def parse_expression(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass diff --git a/.transpyc_cache/352a68eefae37a7c.doc.json b/.transpyc_cache/352a68eefae37a7c.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/352a68eefae37a7c.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/352a68eefae37a7c.pyi b/.transpyc_cache/352a68eefae37a7c.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/352a68eefae37a7c.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/39178dc5ba692b92.doc.json b/.transpyc_cache/39178dc5ba692b92.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/39178dc5ba692b92.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/39178dc5ba692b92.pyi b/.transpyc_cache/39178dc5ba692b92.pyi new file mode 100644 index 0000000..6b53248 --- /dev/null +++ b/.transpyc_cache/39178dc5ba692b92.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +def test_strlen() -> t.CInt: pass + +def test_strcpy() -> t.CInt: pass + +def test_strcat() -> t.CInt: pass + +def test_strcmp() -> t.CInt: pass + +def test_strncpy() -> t.CInt: pass + +def test_char_access() -> t.CInt: pass + +def test_snprintf() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/3a56c52de621df8f.doc.json b/.transpyc_cache/3a56c52de621df8f.doc.json new file mode 100644 index 0000000..47b20d9 --- /dev/null +++ b/.transpyc_cache/3a56c52de621df8f.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/3a56c52de621df8f.pyi b/.transpyc_cache/3a56c52de621df8f.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/3a56c52de621df8f.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/3a769816ec956305.doc.json b/.transpyc_cache/3a769816ec956305.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/3a769816ec956305.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/3a769816ec956305.pyi b/.transpyc_cache/3a769816ec956305.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/3a769816ec956305.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/3ab9ec935db9efeb.doc.json b/.transpyc_cache/3ab9ec935db9efeb.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/.transpyc_cache/3ab9ec935db9efeb.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/.transpyc_cache/3ab9ec935db9efeb.pyi b/.transpyc_cache/3ab9ec935db9efeb.pyi new file mode 100644 index 0000000..0e885bf --- /dev/null +++ b/.transpyc_cache/3ab9ec935db9efeb.pyi @@ -0,0 +1,89 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/3b76f342dcafa255.doc.json b/.transpyc_cache/3b76f342dcafa255.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/3b76f342dcafa255.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/3b76f342dcafa255.pyi b/.transpyc_cache/3b76f342dcafa255.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/3b76f342dcafa255.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/3c15289dd3f27ff8.doc.json b/.transpyc_cache/3c15289dd3f27ff8.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/3c15289dd3f27ff8.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/3c15289dd3f27ff8.pyi b/.transpyc_cache/3c15289dd3f27ff8.pyi new file mode 100644 index 0000000..c495a8e --- /dev/null +++ b/.transpyc_cache/3c15289dd3f27ff8.pyi @@ -0,0 +1,84 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/3c2fbc9e35ad3bf2.doc.json b/.transpyc_cache/3c2fbc9e35ad3bf2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/3c2fbc9e35ad3bf2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/3c2fbc9e35ad3bf2.pyi b/.transpyc_cache/3c2fbc9e35ad3bf2.pyi new file mode 100644 index 0000000..c77b79f --- /dev/null +++ b/.transpyc_cache/3c2fbc9e35ad3bf2.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from func_test.py +Module: func_test +""" + + +import stdio +import t, c + +def add(a: int, b: int) -> int: pass + +def mul(a: int, b: int) -> int: pass + +def square(x: int) -> int: pass + +def factorial(n: int) -> int: pass + +def greet(greeting: str, name: str, times: int) -> int: pass + +def sub(a: int, b: int) -> int: pass + +def func_test() -> int: pass diff --git a/.transpyc_cache/3cd87018ba1e76bc.doc.json b/.transpyc_cache/3cd87018ba1e76bc.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/3cd87018ba1e76bc.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/3cd87018ba1e76bc.pyi b/.transpyc_cache/3cd87018ba1e76bc.pyi new file mode 100644 index 0000000..669f365 --- /dev/null +++ b/.transpyc_cache/3cd87018ba1e76bc.pyi @@ -0,0 +1,108 @@ +""" +Auto-generated Python stub file from test_numpy.py +Module: test_numpy +""" + + +from stdint import * +import numpy +from stdio import printf +from stdlib import calloc, free +import string +import vipermath +import memhub +import t, c + +EPS: t.CExtern | t.CDouble +EPS_LOOSE: t.CExtern | t.CDouble +SIMD_BLOCK: t.CExtern | t.CSizeT +arena: t.CExtern | t.CUInt8T | t.CPtr +pool: t.CExtern | memhub.MemPool | t.CPtr +test_passed: t.CExtern | t.CInt +test_failed: t.CExtern | t.CInt + +def check(name: str, condition: t.CInt, detail: str) -> t.CInt: pass + +def approx_eq(a: t.CDouble, b: t.CDouble) -> t.CInt: pass + +def approx_loose(a: t.CDouble, b: t.CDouble) -> t.CInt: pass + +def section_header(title: str) -> t.CInt: pass + +def section_footer() -> t.CInt: pass + +def vec4(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble, v3: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def vec3(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def vec2(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def test_zeros_ones() -> t.CInt: pass + +def test_arange_linspace() -> t.CInt: pass + +def test_eye_diag() -> t.CInt: pass + +def test_sum_mean_min_max() -> t.CInt: pass + +def test_fill_copy() -> t.CInt: pass + +def test_add_sub_mul_div() -> t.CInt: pass + +def test_neg() -> t.CInt: pass + +def test_len() -> t.CInt: pass + +def test_scalar_ops() -> t.CInt: pass + +def test_math_funcs() -> t.CInt: pass + +def test_matmul() -> t.CInt: pass + +def test_dot() -> t.CInt: pass + +def test_transpose() -> t.CInt: pass + +def test_var_std_norm() -> t.CInt: pass + +def test_clip_concatenate() -> t.CInt: pass + +def test_sort_reverse() -> t.CInt: pass + +def test_print_arr() -> t.CInt: pass + +def test_floordiv_mod() -> t.CInt: pass + +def test_additional_math() -> t.CInt: pass + +def test_trig_hyperbolic() -> t.CInt: pass + +def test_degrees_radians() -> t.CInt: pass + +def test_cumsum_diff() -> t.CInt: pass + +def test_max_min_where() -> t.CInt: pass + +def test_flatten_trace() -> t.CInt: pass + +def test_outer() -> t.CInt: pass + +def test_bool_comparison() -> t.CInt: pass + +def test_linalg() -> t.CInt: pass + +def test_additional_creation() -> t.CInt: pass + +def test_interp() -> t.CInt: pass + +def test_numpy_edge() -> t.CInt: pass + +def bench_numpy_perf() -> t.CInt: pass + +def test_numpy_correct() -> t.CInt: pass + +def test_numpy_edge_main() -> t.CInt: pass + +def bench_numpy_main() -> t.CInt: pass + +def test_numpy_main() -> t.CInt: pass diff --git a/.transpyc_cache/3d3edda4a4e6c08b.doc.json b/.transpyc_cache/3d3edda4a4e6c08b.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/.transpyc_cache/3d3edda4a4e6c08b.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/.transpyc_cache/3d3edda4a4e6c08b.pyi b/.transpyc_cache/3d3edda4a4e6c08b.pyi new file mode 100644 index 0000000..ec65969 --- /dev/null +++ b/.transpyc_cache/3d3edda4a4e6c08b.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/3e914ffe8cc3c120.doc.json b/.transpyc_cache/3e914ffe8cc3c120.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/3e914ffe8cc3c120.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/3e914ffe8cc3c120.pyi b/.transpyc_cache/3e914ffe8cc3c120.pyi new file mode 100644 index 0000000..c9678f0 --- /dev/null +++ b/.transpyc_cache/3e914ffe8cc3c120.pyi @@ -0,0 +1,72 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import testcheck +import llvmlite + +POOL_SIZE: t.CDefine = 524288 # 512KB + +def test_type_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_value_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_add(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_if_else(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_module_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_array_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_struct_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_func_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_declare(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_float_types(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_float_binops(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_bitwise_ops(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_fcmp(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_casts(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_fp_conversions(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_phi(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_switch(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_select(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_globals(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_attributes(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_atomic(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_inline_asm_unreachable(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_valid(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_no_terminator(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_duplicate_def(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_undef_use(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_module(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def main() -> int: pass diff --git a/.transpyc_cache/3f7c5e78d8652535.pyi b/.transpyc_cache/3f7c5e78d8652535.pyi new file mode 100644 index 0000000..abe01c2 --- /dev/null +++ b/.transpyc_cache/3f7c5e78d8652535.pyi @@ -0,0 +1,128 @@ +""" +Auto-generated Python stub file from vipermath.py +Module: vipermath +""" + + +import t, c + +U_M_PI: t.CDefine = 3.14159265358979323846 +U_M_E: t.CDefine = 2.71828182845904523536 +U_M_PI_2: t.CDefine = 1.57079632679489661923 +U_M_PI_4: t.CDefine = 0.78539816339744830962 +U_M_1_PI: t.CDefine = 0.31830988618379067154 +U_M_2_PI: t.CDefine = 0.63661977236758134308 +U_M_LN2: t.CDefine = 0.69314718055994530942 +U_M_LN10: t.CDefine = 2.30258509299404568402 +U_M_2_SQRT_PI: t.CDefine = 1.77245385090551602730 + +def radians(degrees: t.CDouble) -> t.CDouble: pass + +def degrees(radians: t.CDouble) -> t.CDouble: pass + +def sin(x: t.CDouble) -> t.CDouble: pass + +def cos(x: t.CDouble) -> t.CDouble: pass + +def tan(x: t.CDouble) -> t.CDouble: pass + +def asin(x: t.CDouble) -> t.CDouble: pass + +def acos(x: t.CDouble) -> t.CDouble: pass + +def _atan_core(x: t.CDouble) -> t.CDouble: pass + +def atan(x: t.CDouble) -> t.CDouble: pass + +def atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble: pass + +def sinh(x: t.CDouble) -> t.CDouble: pass + +def cosh(x: t.CDouble) -> t.CDouble: pass + +def tanh(x: t.CDouble) -> t.CDouble: pass + +def exp(x: t.CDouble) -> t.CDouble: pass + +def log(x: t.CDouble) -> t.CDouble: pass + +def sqrt(x: t.CDouble) -> t.CDouble: pass + +def abs(x: t.CInt) -> t.CInt: pass + +def fabs(x: t.CDouble) -> t.CDouble: pass + +def labs(x: t.CLong) -> t.CLong: pass + +def factorial(n: t.CInt) -> t.CDouble: pass + +def combination(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def permutation(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def powf(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def cbrt(x: t.CDouble) -> t.CDouble: pass + +def hypot(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def floor(x: t.CDouble) -> t.CDouble: pass + +def ceil(x: t.CDouble) -> t.CDouble: pass + +def round(x: t.CDouble) -> t.CDouble: pass + +def trunc(x: t.CDouble) -> t.CDouble: pass + +def fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def fmodf(x: float, y: float) -> float: pass + +def modf(x: t.CDouble, iptr: t.CDouble | t.CPtr) -> t.CDouble: pass + +def log10(x: t.CDouble) -> t.CDouble: pass + +def log2(x: t.CDouble) -> t.CDouble: pass + +def exp2(x: t.CDouble) -> t.CDouble: pass + +def expm1(x: t.CDouble) -> t.CDouble: pass + +def log1p(x: t.CDouble) -> t.CDouble: pass + +def asinh(x: t.CDouble) -> t.CDouble: pass + +def acosh(x: t.CDouble) -> t.CDouble: pass + +def atanh(x: t.CDouble) -> t.CDouble: pass + +def gamma(x: t.CDouble) -> t.CDouble: pass + +def erf(x: t.CDouble) -> t.CDouble: pass + +def erfc(x: t.CDouble) -> t.CDouble: pass + +def sqrtf(x: t.CFloat) -> t.CFloat: pass + +def sinf(x: t.CFloat) -> t.CFloat: pass + +def cosf(x: t.CFloat) -> t.CFloat: pass + +def tanf(x: t.CFloat) -> t.CFloat: pass + +def fabsf(x: t.CFloat) -> t.CFloat: pass + +def floorf(x: t.CFloat) -> t.CFloat: pass + +def ceilf(x: t.CFloat) -> t.CFloat: pass + + +class _U(t.CUnion): + d: t.CDouble + u: t.CUInt64T + +def isnan(x: t.CDouble) -> t.CStatic | t.CInt: pass + +def isinf(x: t.CDouble) -> t.CStatic | t.CInt: pass diff --git a/.transpyc_cache/4020cac9bab5e9f1.doc.json b/.transpyc_cache/4020cac9bab5e9f1.doc.json new file mode 100644 index 0000000..006cdc0 --- /dev/null +++ b/.transpyc_cache/4020cac9bab5e9f1.doc.json @@ -0,0 +1 @@ +{"extract_func_attrs": "从 decorator_list 提取 c.Attribute 属性，返回 LLVM IR 属性字符串", "_get_attr_name_from_node": "从 AST 节点提取属性名\n\n 支持:\n - Call(func=Attribute(...)): t.attr.always_inline() -> \"always_inline\"\n - Attribute: t.attr.packed -> \"packed\"\n - Attribute(t.attr.llvm.xxx): t.attr.llvm.nounwind -> \"nounwind\"\n ", "_append_llvm_attr": "将属性名映射到 LLVM 属性名并追加到缓冲区\n\n 返回 1 表示已追加，0 表示不支持该属性\n ", "_append_str": "将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）", "_mangle_name": "为任意名称添加 SHA1 前缀（不做 main/export 检查）", "_mangle_name_with_sha1": "用指定 SHA1 为名称添加前缀（不做 main/export 检查）", "_mangle_func_name": "为函数名添加 SHA1 前缀（检查 main/export）", "forward_declare_functions": "预扫描所有顶层 FunctionDef，创建前向声明", "translate_function_def": "翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）", "translate_nested_function_def": "翻译嵌套函数定义：提升为顶层函数 + 创建闭包", "create_function": "创建 LLVM 函数并添加参数"} \ No newline at end of file diff --git a/.transpyc_cache/4020cac9bab5e9f1.pyi b/.transpyc_cache/4020cac9bab5e9f1.pyi new file mode 100644 index 0000000..2b7f43b --- /dev/null +++ b/.transpyc_cache/4020cac9bab5e9f1.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFunctions.py +Module: lib.core.Handles.HandlesFunctions +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import stdio +import stdlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesImports as HandlesImports + +def extract_func_attrs(pool: memhub.MemBuddy | t.CPtr, decorator_list: list[ast.AST | t.CPtr] | t.CPtr, imported_modules: str) -> t.CChar | t.CPtr: pass + +def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str: pass + +def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int: pass + +def _append_str(buf: t.CChar | t.CPtr, s: str) -> None: pass + +def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str: pass + +def _mangle_name_with_sha1(sha1: str, name: str) -> str: pass + +def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str, has_export: int) -> str: pass + +def forward_declare_functions(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def translate_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nested_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, name: str, args_node: ast.AST | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Function | t.CPtr: pass diff --git a/.transpyc_cache/4031ec3fbcff0767.doc.json b/.transpyc_cache/4031ec3fbcff0767.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/.transpyc_cache/4031ec3fbcff0767.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/.transpyc_cache/4031ec3fbcff0767.pyi b/.transpyc_cache/4031ec3fbcff0767.pyi new file mode 100644 index 0000000..1e1b4bd --- /dev/null +++ b/.transpyc_cache/4031ec3fbcff0767.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/4175f14c5fbef360.doc.json b/.transpyc_cache/4175f14c5fbef360.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/4175f14c5fbef360.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/4175f14c5fbef360.pyi b/.transpyc_cache/4175f14c5fbef360.pyi new file mode 100644 index 0000000..209a287 --- /dev/null +++ b/.transpyc_cache/4175f14c5fbef360.pyi @@ -0,0 +1,17 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from t import CInt, CExport, CPtr +from stdio import printf +import memhub +import stdlib +import string +from ast import AST, ASTKind, Module, FunctionDef, Assign, If, For, While, With, Return, Expr, Name, Constant, BinOp, UnaryOp, Call, Compare, Import, ImportFrom, Expression, CONST_INT, CONST_STR, CONST_BOOL, CONST_NONE, OpKind, ASTCtx, FLAG_IS_ASYNC, parse, parse_expression, dump + +ARENA_SIZE: t.CExtern | CInt + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/42ced27d9e76aef5.pyi b/.transpyc_cache/42ced27d9e76aef5.pyi new file mode 100644 index 0000000..f1d44b9 --- /dev/null +++ b/.transpyc_cache/42ced27d9e76aef5.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from ctraits.py +Module: ctraits +""" + +import c + + +import t + +def isptr(x: t.CType) -> bool: pass diff --git a/.transpyc_cache/4337fb260448bbe2.pyi b/.transpyc_cache/4337fb260448bbe2.pyi new file mode 100644 index 0000000..7ed88bd --- /dev/null +++ b/.transpyc_cache/4337fb260448bbe2.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from ast.astaux.py +Module: ast.astaux +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class ExceptHandler(AST): + type: AST | t.CPtr + name: str + def __new__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: ExceptHandler) -> t.CInt: pass + def type_name(self: ExceptHandler) -> str: pass + def dump(self: ExceptHandler, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arguments(AST): + args: list[AST | t.CPtr] | t.CPtr + vararg: AST | t.CPtr + kwarg: AST | t.CPtr + defaults: list[AST | t.CPtr] | t.CPtr + kw_defaults: list[AST | t.CPtr] | t.CPtr + def __new__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Arguments) -> t.CInt: pass + def type_name(self: Arguments) -> str: pass + def dump(self: Arguments, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arg(AST): + arg: str + annotation: AST | t.CPtr + def __new__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def kind(self: Arg) -> t.CInt: pass + def type_name(self: Arg) -> str: pass + def dump(self: Arg, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Keyword(AST): + arg: str + value: AST | t.CPtr + def __new__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Keyword) -> t.CInt: pass + def type_name(self: Keyword) -> str: pass + def dump(self: Keyword, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Alias(AST): + name: str + asname: str + def __new__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def __init__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def kind(self: Alias) -> t.CInt: pass + def type_name(self: Alias) -> str: pass + def dump(self: Alias, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class WithItem(AST): + context_expr: AST | t.CPtr + optional_vars: AST | t.CPtr + def __new__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def __init__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def kind(self: WithItem) -> t.CInt: pass + def type_name(self: WithItem) -> str: pass + def dump(self: WithItem, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Comprehension(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + ifs: list[AST | t.CPtr] | t.CPtr + is_async: t.CInt + def __new__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: Comprehension) -> t.CInt: pass + def type_name(self: Comprehension) -> str: pass + def dump(self: Comprehension, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/.transpyc_cache/43b4b8f6ce3ed278.doc.json b/.transpyc_cache/43b4b8f6ce3ed278.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/43b4b8f6ce3ed278.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/43b4b8f6ce3ed278.pyi b/.transpyc_cache/43b4b8f6ce3ed278.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/43b4b8f6ce3ed278.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/440f18f9dce85a95.doc.json b/.transpyc_cache/440f18f9dce85a95.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/440f18f9dce85a95.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/440f18f9dce85a95.pyi b/.transpyc_cache/440f18f9dce85a95.pyi new file mode 100644 index 0000000..0b2c4a4 --- /dev/null +++ b/.transpyc_cache/440f18f9dce85a95.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def test_nonlocal_conditional() -> t.CInt: pass + +def test_nonlocal_multi_var() -> t.CInt: pass + + +g_counter: t.CExtern | t.CInt + +def inc_global() -> t.CInt: pass + +def add_global(n: t.CInt) -> t.CInt: pass + +def test_global_cross_functions() -> t.CInt: pass + +def test_nested_ref_capture() -> t.CInt: pass + +def test_lambda_capture_then_modify() -> t.CInt: pass + +def test_nonlocal_loop_accumulate() -> t.CInt: pass + +def test_deep_nested_vars() -> t.CInt: pass + + +g_state: t.CExtern | t.CInt + +def test_global_nonlocal_same_func() -> t.CInt: pass + +def test_lambda_default_and_capture() -> t.CInt: pass + +def test_nested_accumulate() -> t.CInt: pass + +def test_nonlocal_while() -> t.CInt: pass + + +class Point: + x: t.CInt + y: t.CInt + +def test_nested_modify_struct() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/451745df21e598af.pyi b/.transpyc_cache/451745df21e598af.pyi new file mode 100644 index 0000000..b3bfb92 --- /dev/null +++ b/.transpyc_cache/451745df21e598af.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from zlib.zinflate.py +Module: zlib.zinflate +""" + + +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import string +import memhub +import t, c + +@t.Object +class zinflate_stream: + pool: memhub.MemManager | t.CPtr + reader: zdef.zbit_reader + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + header_parsed: t.CInt + is_gzip: t.CInt + adler: t.CUInt32T + crc: t.CUInt32T + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + max_length: t.CSizeT + def __init__(self: zinflate_stream) -> t.CInt: pass + def output_byte(self: zinflate_stream, byte: BYTE) -> t.CInt: pass + def parse_header(self: zinflate_stream) -> t.CInt: pass + def read_bits_from_input(self: zinflate_stream, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_huffman(self: zinflate_stream, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: pass + def inflate_block_stored(self: zinflate_stream) -> t.CInt: pass + def inflate_block_fixed(self: zinflate_stream) -> t.CInt: pass + def inflate_block_dynamic(self: zinflate_stream) -> t.CInt: pass + def inflate_stream(self: zinflate_stream) -> t.CInt: pass + def decompress(self: zinflate_stream, data: UINT8PTR, length: t.CSizeT, max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zinflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zinflate_stream) -> zinflate_stream | t.CPtr: pass + def destroy(self: zinflate_stream) -> t.CInt: pass + +def zinflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass + +def zinflate_create(pool: memhub.MemManager | t.CPtr, wbits: t.CInt) -> zinflate_stream | t.CPtr: pass diff --git a/.transpyc_cache/45242e16a6829016.doc.json b/.transpyc_cache/45242e16a6829016.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/45242e16a6829016.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/45242e16a6829016.pyi b/.transpyc_cache/45242e16a6829016.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/45242e16a6829016.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/455ff4e74c3dbfb0.pyi b/.transpyc_cache/455ff4e74c3dbfb0.pyi new file mode 100644 index 0000000..984146e --- /dev/null +++ b/.transpyc_cache/455ff4e74c3dbfb0.pyi @@ -0,0 +1,120 @@ +""" +Auto-generated Python stub file from llvmlite.__types.py +Module: llvmlite.__types +""" + + +import t, c +from stdint import * +import viperlib +import string +import memhub +from linkedlist import GSListNode + +class ParamNode(GSListNode[ParamNode]): + Ty: t.CVoid | t.CPtr +class LLVMType(t.REnum): + class Int: + Bits: t.CInt + class Ptr: + Pointee: LLVMType | t.CPtr + class Func: + Ret: LLVMType | t.CPtr + Params: ParamNode | t.CPtr + PCount: t.CInt + class Array: + ElemTy: LLVMType | t.CPtr + Count: t.CInt + class Struct: + Fields: ParamNode | t.CPtr + FCount: t.CInt + Name: t.CChar | t.CPtr + class Float: + Bits: t.CInt + class Void: + pass + class Label: + pass + class Metadata: + pass + +def _new_type(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int1(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int8(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int16(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int32(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int64(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Ptr(pool: memhub.MemBuddy | t.CPtr, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + +def Func(pool: memhub.MemBuddy | t.CPtr, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Void(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Array(pool: memhub.MemBuddy | t.CPtr, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Struct(pool: memhub.MemBuddy | t.CPtr, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + +def Half(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Float(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Double(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def FP128(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Label(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + + +PNODE_SIZE: t.CDefine = 16 + +def new_param_node(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + +def param_list_append(head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_get_ty(node: ParamNode | t.CPtr) -> LLVMType | t.CPtr: pass + +def paramnode_get_next(node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_set_ty(node: ParamNode | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def paramnode_set_next(node: ParamNode | t.CPtr, nxt: ParamNode | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class LLVMTypeCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMTypeCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def Int1(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int8(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int16(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int32(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int64(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Ptr(self: LLVMTypeCore, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + def Func(self: LLVMTypeCore, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Array(self: LLVMTypeCore, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Struct(self: LLVMTypeCore, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + def Void(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Half(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Float(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Double(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def FP128(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Label(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def NewParamNode(self: LLVMTypeCore, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + def ParamListAppend(self: LLVMTypeCore, head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + def PrintType(self: LLVMTypeCore, buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def TypePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _print_struct_body(buf: t.CChar | t.CPtr, size: t.CSizeT, fields: ParamNode | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def PrintStructDefinition(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def get_struct_name(ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass diff --git a/.transpyc_cache/46ea362b72c10a9e.pyi b/.transpyc_cache/46ea362b72c10a9e.pyi new file mode 100644 index 0000000..fbd9c3b --- /dev/null +++ b/.transpyc_cache/46ea362b72c10a9e.pyi @@ -0,0 +1,219 @@ +""" +Auto-generated Python stub file from llvmlite.__builder.py +Module: llvmlite.__builder +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__types import LLVMType, TypePrint +from .__values import Value, new_value, SSAValue +from .__function import Function, BasicBlock, block_append_text, function_move_block_to_end, _ll_name_needs_quote + +class IRBuilder: + Pool: memhub.MemBuddy | t.CPtr + Func: Function | t.CPtr + CurBlock: BasicBlock | t.CPtr + Counter: t.CInt + +BUILDER_SIZE: t.CDefine = 32 + +def new_builder(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + +def position_at_end(builder: IRBuilder | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def _alloc_ssa_name(builder: IRBuilder | t.CPtr) -> t.CChar | t.CPtr: pass + +def _type_str(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass + +def _emit(builder: IRBuilder | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def _mark_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def builder_cur_block_is_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_alloca(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_load(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, ptr: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_store(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, ptr: Value | t.CPtr) -> t.CInt: pass + +def build_add(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_mul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_udiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_srem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_urem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_and(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_or(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_xor(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_shl(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_lshr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_ashr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fadd(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fsub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fmul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_frem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fneg(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> Value | t.CPtr: pass + +def _build_binop(builder: IRBuilder | t.CPtr, op: t.CChar | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + + +ICMP_EQ: t.CDefine = 0 +ICMP_NE: t.CDefine = 1 +ICMP_SGT: t.CDefine = 2 +ICMP_SGE: t.CDefine = 3 +ICMP_SLT: t.CDefine = 4 +ICMP_SLE: t.CDefine = 5 +ICMP_UGT: t.CDefine = 6 +ICMP_UGE: t.CDefine = 7 +ICMP_ULT: t.CDefine = 8 +ICMP_ULE: t.CDefine = 9 +FCMP_FALSE: t.CDefine = 0 +FCMP_OEQ: t.CDefine = 1 +FCMP_OGT: t.CDefine = 2 +FCMP_OGE: t.CDefine = 3 +FCMP_OLT: t.CDefine = 4 +FCMP_OLE: t.CDefine = 5 +FCMP_ONE: t.CDefine = 6 +FCMP_ORD: t.CDefine = 7 +FCMP_UNO: t.CDefine = 8 +FCMP_UEQ: t.CDefine = 9 +FCMP_UGT: t.CDefine = 10 +FCMP_UGE: t.CDefine = 11 +FCMP_ULT: t.CDefine = 12 +FCMP_ULE: t.CDefine = 13 +FCMP_UNE: t.CDefine = 14 +FCMP_TRUE: t.CDefine = 15 + +def _icmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def _fcmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def build_icmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fcmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_br(builder: IRBuilder | t.CPtr, target: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_cond_br(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, then_blk: BasicBlock | t.CPtr, else_blk: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_ret(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> t.CInt: pass + +def build_ret_void(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_call(builder: IRBuilder | t.CPtr, callee: t.CChar | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr, is_variadic: t.CInt) -> Value | t.CPtr: pass + +def build_call_indirect(builder: IRBuilder | t.CPtr, callee_val: Value | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_bitcast(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_sext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_trunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_zext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ptrtoint(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_inttoptr(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fpext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fptrunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2si(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_si2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2ui(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ui2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_gep(builder: IRBuilder | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_array(builder: IRBuilder | t.CPtr, array_ty: LLVMType | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_struct(builder: IRBuilder | t.CPtr, struct_ty: LLVMType | t.CPtr, field_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, field_idx: t.CInt) -> Value | t.CPtr: pass + + +class PhiIncoming(GSListNode[PhiIncoming]): + Val: Value | t.CPtr + Block: BasicBlock | t.CPtr + +PHIINC_SIZE: t.CDefine = 24 + +def new_phi_incoming(pool: memhub.MemBuddy | t.CPtr, val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> PhiIncoming | t.CPtr: pass + +def build_phi(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, incoming_head: PhiIncoming | t.CPtr, count: t.CInt) -> Value | t.CPtr: pass + + +class SwitchCase(GSListNode[SwitchCase]): + CaseVal: Value | t.CPtr + Block: BasicBlock | t.CPtr + +SWCASE_SIZE: t.CDefine = 24 + +def new_switch_case(pool: memhub.MemBuddy | t.CPtr, case_val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> SwitchCase | t.CPtr: pass + +def build_switch(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, default_blk: BasicBlock | t.CPtr, cases_head: SwitchCase | t.CPtr, case_count: t.CInt) -> t.CInt: pass + +def build_select(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, true_val: Value | t.CPtr, false_val: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_unreachable(builder: IRBuilder | t.CPtr) -> t.CInt: pass + + +ATOMIC_XCHG: t.CDefine = 0 +ATOMIC_ADD: t.CDefine = 1 +ATOMIC_SUB: t.CDefine = 2 +ATOMIC_AND: t.CDefine = 3 +ATOMIC_NAND: t.CDefine = 4 +ATOMIC_OR: t.CDefine = 5 +ATOMIC_XOR: t.CDefine = 6 +ATOMIC_MAX: t.CDefine = 7 +ATOMIC_MIN: t.CDefine = 8 +ATOMIC_UMAX: t.CDefine = 9 +ATOMIC_UMIN: t.CDefine = 10 +ATOMIC_ORDER_NOTATOMIC: t.CDefine = 0 +ATOMIC_ORDER_UNORDERED: t.CDefine = 1 +ATOMIC_ORDER_MONOTONIC: t.CDefine = 2 +ATOMIC_ORDER_ACQUIRE: t.CDefine = 3 +ATOMIC_ORDER_RELEASE: t.CDefine = 4 +ATOMIC_ORDER_ACQ_REL: t.CDefine = 5 +ATOMIC_ORDER_SEQ_CST: t.CDefine = 6 + +def _atomic_op_str(op: t.CInt) -> t.CChar | t.CPtr: pass + +def _atomic_order_str(ord: t.CInt) -> t.CChar | t.CPtr: pass + +def build_atomicrmw(builder: IRBuilder | t.CPtr, op: t.CInt, ptr: Value | t.CPtr, val: Value | t.CPtr, order: t.CInt) -> Value | t.CPtr: pass + +def build_cmpxchg(builder: IRBuilder | t.CPtr, ptr: Value | t.CPtr, cmp_val: Value | t.CPtr, new_val: Value | t.CPtr, succ_order: t.CInt, fail_order: t.CInt) -> Value | t.CPtr: pass + +def build_inline_asm(builder: IRBuilder | t.CPtr, ret_ty: LLVMType | t.CPtr, asm_text: t.CChar | t.CPtr, constraints: t.CChar | t.CPtr, has_side_effects: t.CInt) -> Value | t.CPtr: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/47767b5026a8ee15.pyi b/.transpyc_cache/47767b5026a8ee15.pyi new file mode 100644 index 0000000..7890f3d --- /dev/null +++ b/.transpyc_cache/47767b5026a8ee15.pyi @@ -0,0 +1,241 @@ +""" +Auto-generated Python stub file from ast.exprs.py +Module: ast.exprs +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, _dump_op_list, _op_name, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE + +class Name(AST): + id: str + ctx: t.CInt + def __new__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Name) -> t.CInt: pass + def type_name(self: Name) -> str: pass + def dump(self: Name, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Constant(AST): + int_val: t.CInt64T + float_val: t.CDouble + str_val: str + const_kind: t.CInt + def __new__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def __init__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def kind(self: Constant) -> t.CInt: pass + def type_name(self: Constant) -> str: pass + def dump(self: Constant, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BinOp(AST): + left: AST | t.CPtr + op: t.CInt + right: AST | t.CPtr + def __new__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def __init__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def kind(self: BinOp) -> t.CInt: pass + def type_name(self: BinOp) -> str: pass + def dump(self: BinOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class UnaryOp(AST): + op: t.CInt + operand: AST | t.CPtr + def __new__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def __init__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def kind(self: UnaryOp) -> t.CInt: pass + def type_name(self: UnaryOp) -> str: pass + def dump(self: UnaryOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Compare(AST): + left: AST | t.CPtr + ops: t.CPtr + comparators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Compare) -> t.CInt: pass + def type_name(self: Compare) -> str: pass + def dump(self: Compare, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Call(AST): + func: AST | t.CPtr + args: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + def __new__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Call) -> t.CInt: pass + def type_name(self: Call) -> str: pass + def dump(self: Call, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BoolOp(AST): + op: t.CInt + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: BoolOp) -> t.CInt: pass + def type_name(self: BoolOp) -> str: pass + def dump(self: BoolOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Lambda(AST): + args: AST | t.CPtr + body: AST | t.CPtr + def __new__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Lambda) -> t.CInt: pass + def type_name(self: Lambda) -> str: pass + def dump(self: Lambda, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class IfExp(AST): + test: AST | t.CPtr + body: AST | t.CPtr + orelse: AST | t.CPtr + def __new__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def __init__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def kind(self: IfExp) -> t.CInt: pass + def type_name(self: IfExp) -> str: pass + def dump(self: IfExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Dict(AST): + keys: list[AST | t.CPtr] | t.CPtr + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Dict) -> t.CInt: pass + def type_name(self: Dict) -> str: pass + def dump(self: Dict, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Set(AST): + elts: list[AST | t.CPtr] | t.CPtr + def __new__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Set) -> t.CInt: pass + def type_name(self: Set) -> str: pass + def dump(self: Set, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ListComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: ListComp) -> t.CInt: pass + def type_name(self: ListComp) -> str: pass + def dump(self: ListComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class SetComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: SetComp) -> t.CInt: pass + def type_name(self: SetComp) -> str: pass + def dump(self: SetComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class DictComp(AST): + key: AST | t.CPtr + value: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: DictComp) -> t.CInt: pass + def type_name(self: DictComp) -> str: pass + def dump(self: DictComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class GeneratorExp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: GeneratorExp) -> t.CInt: pass + def type_name(self: GeneratorExp) -> str: pass + def dump(self: GeneratorExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Await(AST): + value: AST | t.CPtr + def __new__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Await) -> t.CInt: pass + def type_name(self: Await) -> str: pass + def dump(self: Await, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Yield(AST): + value: AST | t.CPtr + def __new__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Yield) -> t.CInt: pass + def type_name(self: Yield) -> str: pass + def dump(self: Yield, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class YieldFrom(AST): + value: AST | t.CPtr + def __new__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: YieldFrom) -> t.CInt: pass + def type_name(self: YieldFrom) -> str: pass + def dump(self: YieldFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FormattedValue(AST): + value: AST | t.CPtr + conversion: t.CInt + format_spec: AST | t.CPtr + def __new__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def kind(self: FormattedValue) -> t.CInt: pass + def type_name(self: FormattedValue) -> str: pass + def dump(self: FormattedValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class JoinedStr(AST): + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: JoinedStr) -> t.CInt: pass + def type_name(self: JoinedStr) -> str: pass + def dump(self: JoinedStr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Attribute(AST): + value: AST | t.CPtr + attr: str + ctx: t.CInt + def __new__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Attribute) -> t.CInt: pass + def type_name(self: Attribute) -> str: pass + def dump(self: Attribute, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Subscript(AST): + value: AST | t.CPtr + slice: AST | t.CPtr + ctx: t.CInt + def __new__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Subscript) -> t.CInt: pass + def type_name(self: Subscript) -> str: pass + def dump(self: Subscript, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Starred(AST): + value: AST | t.CPtr + ctx: t.CInt + def __new__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Starred) -> t.CInt: pass + def type_name(self: Starred) -> str: pass + def dump(self: Starred, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class List(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: List) -> t.CInt: pass + def type_name(self: List) -> str: pass + def dump(self: List, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Tuple(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Tuple) -> t.CInt: pass + def type_name(self: Tuple) -> str: pass + def dump(self: Tuple, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Slice(AST): + lower: AST | t.CPtr + upper: AST | t.CPtr + step: AST | t.CPtr + def __new__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def kind(self: Slice) -> t.CInt: pass + def type_name(self: Slice) -> str: pass + def dump(self: Slice, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class NamedExpr(AST): + target: AST | t.CPtr + value: AST | t.CPtr + def __new__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: NamedExpr) -> t.CInt: pass + def type_name(self: NamedExpr) -> str: pass + def dump(self: NamedExpr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class OpNode(AST): + op: t.CInt + def __new__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def __init__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def kind(self: OpNode) -> t.CInt: pass + def type_name(self: OpNode) -> str: pass + def dump(self: OpNode, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/.transpyc_cache/48352329f97d0858.doc.json b/.transpyc_cache/48352329f97d0858.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/48352329f97d0858.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/48352329f97d0858.pyi b/.transpyc_cache/48352329f97d0858.pyi new file mode 100644 index 0000000..06f88af --- /dev/null +++ b/.transpyc_cache/48352329f97d0858.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import _fakeduck +import stdio +import stdlib +import string +import memhub +import testcheck +import _withcontent + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/48dd999d59d8c008.doc.json b/.transpyc_cache/48dd999d59d8c008.doc.json new file mode 100644 index 0000000..efa98af --- /dev/null +++ b/.transpyc_cache/48dd999d59d8c008.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素值", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/48dd999d59d8c008.pyi b/.transpyc_cache/48dd999d59d8c008.pyi new file mode 100644 index 0000000..4727f86 --- /dev/null +++ b/.transpyc_cache/48dd999d59d8c008.pyi @@ -0,0 +1,75 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/48fa8cc758d3826b.doc.json b/.transpyc_cache/48fa8cc758d3826b.doc.json new file mode 100644 index 0000000..ac292ab --- /dev/null +++ b/.transpyc_cache/48fa8cc758d3826b.doc.json @@ -0,0 +1 @@ +{"ForHandle": "for 循环语句处理器：继承 Mixin 获得 Trans 回指针", "ForHandle.Handle": "翻译 for i in range(...) 循环语句", "ForHandle._handle_ptr_iter": "指针迭代: for x in ptr: 直到解引用为空"} \ No newline at end of file diff --git a/.transpyc_cache/48fa8cc758d3826b.pyi b/.transpyc_cache/48fa8cc758d3826b.pyi new file mode 100644 index 0000000..08527b9 --- /dev/null +++ b/.transpyc_cache/48fa8cc758d3826b.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFor.py +Module: lib.core.Handles.HandlesFor +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class ForHandle(HandlesBase.Mixin): + def __init__(self: ForHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ForHandle, node: ast.AST | t.CPtr) -> int: pass + def _handle_ptr_iter(self: ForHandle, for_node: ast.For | t.CPtr, var_name: str) -> int: pass + +def NewForHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ForHandle | t.CPtr: pass diff --git a/.transpyc_cache/494eeec1e4c2f2ec.doc.json b/.transpyc_cache/494eeec1e4c2f2ec.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/494eeec1e4c2f2ec.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/494eeec1e4c2f2ec.pyi b/.transpyc_cache/494eeec1e4c2f2ec.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/494eeec1e4c2f2ec.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/49cae05490a513cb.doc.json b/.transpyc_cache/49cae05490a513cb.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/49cae05490a513cb.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/49cae05490a513cb.pyi b/.transpyc_cache/49cae05490a513cb.pyi new file mode 100644 index 0000000..e1af96b --- /dev/null +++ b/.transpyc_cache/49cae05490a513cb.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import t +import c + + +import stdio +import stdint +import zc.config as config +import zc.test as test +import zc.logic_test as logic +import zc.class_test as class_test +import testcheck +import test_numpy + +def main() -> stdint.INT: pass diff --git a/.transpyc_cache/49cf633017627a4a.doc.json b/.transpyc_cache/49cf633017627a4a.doc.json new file mode 100644 index 0000000..3e3ec7a --- /dev/null +++ b/.transpyc_cache/49cf633017627a4a.doc.json @@ -0,0 +1 @@ +{"Load_project_config": "加载工程配置文件（project.vpj）\n\n Args:\n path: 配置文件路径\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "_join_flags_array": "将 JSON 字符串数组拼接成空格分隔的字符串\n\n 用于 linker.flags 和 compiler.flags 数组。\n 用 while + 索引遍历，避免 list 迭代器 bug。\n\n Args:\n flags_val: JsonValue，应为字符串数组\n\n Returns:\n 空格分隔的字符串，或 None 如果 flags_val 为 None 或非数组或空数组\n ", "_join_path": "将相对路径与 project_dir 拼接为绝对路径\n\n - ./App + Test → Test/App\n - ./temp + Test → Test/temp\n - ../includes + Test → Test/../includes（OS 解析为 includes）\n - 绝对路径保持不变\n - None 返回 None\n ", "resolve_paths": "解析并规范化工程路径（src/temp/output/includes）\n\n 将 project.vpj 中的相对路径（以 ./ 开头）转换为基于 project_dir 的路径。\n\n Args:\n project_dir: 工程根目录（project.vpj 所在目录）\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "get_includes_binary_dir": "获取 includes.binary 目录路径（includes 目录 + .binary 后缀）\n\n includes 目录如 ../includes，则 includes.binary 为 ../includes.binary\n ", "print_config": "打印当前配置（调试用）"} \ No newline at end of file diff --git a/.transpyc_cache/49cf633017627a4a.pyi b/.transpyc_cache/49cf633017627a4a.pyi new file mode 100644 index 0000000..0ff98b1 --- /dev/null +++ b/.transpyc_cache/49cf633017627a4a.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from lib.Projectrans.Config.py +Module: lib.Projectrans.Config +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import viperlib +from json.__parser import parse as json_parse +from json import JsonValue, JSON_OBJECT, JSON_INT +import w32.fileio as fileio + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SourceDir: t.CExtern | str +TempDir: t.CExtern | str +OutputDir: t.CExtern | str +ProjectName: t.CExtern | str +ProjectVersion: t.CExtern | str +CompilerCmd: t.CExtern | str +CompilerFlags: t.CExtern | str +LinkerCmd: t.CExtern | str +LinkerFlags: t.CExtern | str +LinkerOutput: t.CExtern | str +TargetTriple: t.CExtern | str +TargetDataLayout: t.CExtern | str +SliceLevel: t.CExtern | t.CInt +StrictMode: t.CExtern | t.CInt +IncludesDir: t.CExtern | str + +def Load_project_config(path: str) -> int: pass + +def _join_flags_array(flags_val: JsonValue | t.CPtr) -> str: pass + +def _join_path(rel: str, project_dir: str) -> str: pass + +def resolve_paths(project_dir: str) -> int: pass + +def get_includes_binary_dir() -> str: pass + +def print_config() -> t.CInt: pass diff --git a/.transpyc_cache/4b639fa48e802b3b.doc.json b/.transpyc_cache/4b639fa48e802b3b.doc.json new file mode 100644 index 0000000..ac292ab --- /dev/null +++ b/.transpyc_cache/4b639fa48e802b3b.doc.json @@ -0,0 +1 @@ +{"ForHandle": "for 循环语句处理器：继承 Mixin 获得 Trans 回指针", "ForHandle.Handle": "翻译 for i in range(...) 循环语句", "ForHandle._handle_ptr_iter": "指针迭代: for x in ptr: 直到解引用为空"} \ No newline at end of file diff --git a/.transpyc_cache/4b639fa48e802b3b.pyi b/.transpyc_cache/4b639fa48e802b3b.pyi new file mode 100644 index 0000000..7a22de4 --- /dev/null +++ b/.transpyc_cache/4b639fa48e802b3b.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFor.py +Module: lib.core.Handles.HandlesFor +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class ForHandle(HandlesBase.Mixin): + def __init__(self: ForHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ForHandle, node: ast.AST | t.CPtr) -> int: pass + def _handle_ptr_iter(self: ForHandle, for_node: ast.For | t.CPtr, var_name: str) -> int: pass + +def NewForHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ForHandle | t.CPtr: pass diff --git a/.transpyc_cache/4ca69b583c572f13.doc.json b/.transpyc_cache/4ca69b583c572f13.doc.json new file mode 100644 index 0000000..1a7c43a --- /dev/null +++ b/.transpyc_cache/4ca69b583c572f13.doc.json @@ -0,0 +1 @@ +{"VarEntry": "变量表条目", "Scope": "作用域节点", "SymbolTable": "嵌套作用域符号表", "init_vars": "分配并清零变量表数组", "find_var": "在变量表中按名称查找", "find_var_entry": "在变量表中按名称查找，返回 VarEntry 或 None", "lookup_var_entry": "从当前作用域逐级向上查找变量，返回 VarEntry 或 None", "set_var_annot_class_name": "设置变量的原始类型注解类名，返回 0=成功 / 1=失败", "add_var": "添加变量到变量表", "get_or_create_var": "查找或创建变量 alloca", "_create_scope": "创建并初始化作用域节点", "init_symbol_table": "创建并初始化符号表，包含模块级根作用域", "enter_scope": "进入新作用域，返回新创建的作用域", "exit_scope": "退出当前作用域，恢复到父作用域", "define_var": "在当前作用域定义变量，返回 0 成功", "define_module_var": "在模块作用域定义变量，返回 0 成功", "lookup_var": "从当前作用域逐级向上查找变量，返回 alloca 或 None", "lookup_current": "仅在当前作用域查找变量", "lookup_module_var": "在模块作用域查找变量", "get_or_create_sym": "查找或创建变量 alloca（在当前作用域）"} \ No newline at end of file diff --git a/.transpyc_cache/4ca69b583c572f13.pyi b/.transpyc_cache/4ca69b583c572f13.pyi new file mode 100644 index 0000000..266df75 --- /dev/null +++ b/.transpyc_cache/4ca69b583c572f13.pyi @@ -0,0 +1,68 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesVar.py +Module: lib.core.Handles.HandlesVar +""" + + +import t, c +from stdint import * +import llvmlite +import memhub +import string + +SCOPE_MODULE: t.CDefine = 0 +SCOPE_FUNCTION: t.CDefine = 1 +SCOPE_BLOCK: t.CDefine = 2 +MAX_VARS: t.CDefine = 256 + +@t.NoVTable +class VarEntry: + Name: str + Alloca: llvmlite.Value | t.CPtr + Used: t.CInt + AnnotClassName: str +@t.NoVTable +class Scope: + Parent: Scope | t.CPtr + Vars: VarEntry | t.CPtr + VarCount: t.CInt + Kind: t.CInt +@t.NoVTable +class SymbolTable: + Pool: memhub.MemBuddy | t.CPtr + Root: Scope | t.CPtr + Current: Scope | t.CPtr + +def init_vars(pool: memhub.MemBuddy | t.CPtr) -> VarEntry | t.CPtr: pass + +def find_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> llvmlite.Value | t.CPtr: pass + +def find_var_entry(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> VarEntry | t.CPtr: pass + +def lookup_var_entry(symtab: SymbolTable | t.CPtr, name: str) -> VarEntry | t.CPtr: pass + +def set_var_annot_class_name(symtab: SymbolTable | t.CPtr, name: str, class_name: str) -> int: pass + +def add_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def get_or_create_var(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _create_scope(pool: memhub.MemBuddy | t.CPtr, parent: Scope | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def init_symbol_table(pool: memhub.MemBuddy | t.CPtr) -> SymbolTable | t.CPtr: pass + +def enter_scope(symtab: SymbolTable | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def exit_scope(symtab: SymbolTable | t.CPtr) -> t.CInt: pass + +def define_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def define_module_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def lookup_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_current(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_module_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def get_or_create_sym(symtab: SymbolTable | t.CPtr, pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/4dd6b3f1427d1cc5.pyi b/.transpyc_cache/4dd6b3f1427d1cc5.pyi new file mode 100644 index 0000000..dd15aa2 --- /dev/null +++ b/.transpyc_cache/4dd6b3f1427d1cc5.pyi @@ -0,0 +1,82 @@ +""" +Auto-generated Python stub file from ast.match.py +Module: ast.match +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class MatchCase(AST): + pattern: AST | t.CPtr + guard: AST | t.CPtr + def __new__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchCase) -> t.CInt: pass + def type_name(self: MatchCase) -> str: pass + def dump(self: MatchCase, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchValue(AST): + value: AST | t.CPtr + def __new__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchValue) -> t.CInt: pass + def type_name(self: MatchValue) -> str: pass + def dump(self: MatchValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSingleton(AST): + value: AST | t.CPtr + def __new__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchSingleton) -> t.CInt: pass + def type_name(self: MatchSingleton) -> str: pass + def dump(self: MatchSingleton, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSequence(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchSequence) -> t.CInt: pass + def type_name(self: MatchSequence) -> str: pass + def dump(self: MatchSequence, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchMapping(AST): + keys: list[AST | t.CPtr] | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + rest: str + def __new__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def __init__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def kind(self: MatchMapping) -> t.CInt: pass + def type_name(self: MatchMapping) -> str: pass + def dump(self: MatchMapping, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchClass(AST): + cls: AST | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + kwd_attrs: list[AST | t.CPtr] | t.CPtr + kwd_patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchClass) -> t.CInt: pass + def type_name(self: MatchClass) -> str: pass + def dump(self: MatchClass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchStar(AST): + name: str + def __new__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchStar) -> t.CInt: pass + def type_name(self: MatchStar) -> str: pass + def dump(self: MatchStar, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchAs(AST): + pattern: AST | t.CPtr + name: str + def __new__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchAs) -> t.CInt: pass + def type_name(self: MatchAs) -> str: pass + def dump(self: MatchAs, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchOr(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchOr) -> t.CInt: pass + def type_name(self: MatchOr) -> str: pass + def dump(self: MatchOr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/.transpyc_cache/4ebd3745aa9cf570.doc.json b/.transpyc_cache/4ebd3745aa9cf570.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/4ebd3745aa9cf570.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/4ebd3745aa9cf570.pyi b/.transpyc_cache/4ebd3745aa9cf570.pyi new file mode 100644 index 0000000..e6450ad --- /dev/null +++ b/.transpyc_cache/4ebd3745aa9cf570.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import memhub + +def test_polymorphic(mgr: memhub.MemManager | t.CPtr, size: t.CSizeT) -> t.CInt: pass + +def test_calloc_polymorphic(mgr: memhub.MemManager | t.CPtr) -> t.CInt: pass + +@t.CExport +def main() -> int: pass diff --git a/.transpyc_cache/4ed910ba727b5e92.doc.json b/.transpyc_cache/4ed910ba727b5e92.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/4ed910ba727b5e92.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/4ed910ba727b5e92.pyi b/.transpyc_cache/4ed910ba727b5e92.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/4ed910ba727b5e92.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/4ee832f9cf4db211.doc.json b/.transpyc_cache/4ee832f9cf4db211.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/4ee832f9cf4db211.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/4ee832f9cf4db211.pyi b/.transpyc_cache/4ee832f9cf4db211.pyi new file mode 100644 index 0000000..60fab85 --- /dev/null +++ b/.transpyc_cache/4ee832f9cf4db211.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/4f90956b289db68b.doc.json b/.transpyc_cache/4f90956b289db68b.doc.json new file mode 100644 index 0000000..724daa6 --- /dev/null +++ b/.transpyc_cache/4f90956b289db68b.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到\n\n allow_star_fallback: 传 0 禁用 star import 回退（用于明确模块名/函数名查找）\n ", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/4f90956b289db68b.pyi b/.transpyc_cache/4f90956b289db68b.pyi new file mode 100644 index 0000000..065139b --- /dev/null +++ b/.transpyc_cache/4f90956b289db68b.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str, allow_star_fallback: int) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/4fb8940ff5ab1022.doc.json b/.transpyc_cache/4fb8940ff5ab1022.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/4fb8940ff5ab1022.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/4fb8940ff5ab1022.pyi b/.transpyc_cache/4fb8940ff5ab1022.pyi new file mode 100644 index 0000000..1f24963 --- /dev/null +++ b/.transpyc_cache/4fb8940ff5ab1022.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +import testcheck + +def add_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a + b + +def mul_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a * b + + +class AppError(Exception): + pass +class DiskError(AppError): + pass +class NetworkError(AppError): + pass +class TimeoutError(NetworkError): + pass + +def raise_disk() -> t.CInt: pass + +def raise_network() -> t.CInt: pass + +def raise_timeout() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/51f0144d6e8188d9.doc.json b/.transpyc_cache/51f0144d6e8188d9.doc.json new file mode 100644 index 0000000..a63fb44 --- /dev/null +++ b/.transpyc_cache/51f0144d6e8188d9.doc.json @@ -0,0 +1 @@ +{"WhileHandle": "while 循环语句处理器：继承 Mixin 获得 Trans 回指针", "WhileHandle.Handle": "翻译 while 循环语句"} \ No newline at end of file diff --git a/.transpyc_cache/51f0144d6e8188d9.pyi b/.transpyc_cache/51f0144d6e8188d9.pyi new file mode 100644 index 0000000..eb71aad --- /dev/null +++ b/.transpyc_cache/51f0144d6e8188d9.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesWhile.py +Module: lib.core.Handles.HandlesWhile +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class WhileHandle(HandlesBase.Mixin): + def __init__(self: WhileHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: WhileHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewWhileHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> WhileHandle | t.CPtr: pass diff --git a/.transpyc_cache/52e6b51da061882f.doc.json b/.transpyc_cache/52e6b51da061882f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/52e6b51da061882f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/52e6b51da061882f.pyi b/.transpyc_cache/52e6b51da061882f.pyi new file mode 100644 index 0000000..edc39b5 --- /dev/null +++ b/.transpyc_cache/52e6b51da061882f.pyi @@ -0,0 +1,43 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +g_val: t.CExtern | t.CInt + +def outer_modify_global() -> t.CInt: pass + +def test_global_in_nested() -> t.CInt: pass + +def test_nonlocal_multi_level() -> t.CInt: pass + + +g_mixed: t.CExtern | t.CInt + +def test_global_nonlocal_mixed() -> t.CInt: pass + +def test_lambda_multi_capture() -> t.CInt: pass + +def test_lambda_arg_capture() -> t.CInt: pass + +def test_nested_return() -> t.CInt: pass + +def test_nonlocal_loop() -> t.CInt: pass + + +g_arr: t.CExtern | t.CArray[t.CInt, 4] + +def test_global_array() -> t.CInt: pass + +def test_nested_conditional() -> t.CInt: pass + +def test_lambda_chain() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/54014f6c53e9640b.doc.json b/.transpyc_cache/54014f6c53e9640b.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/.transpyc_cache/54014f6c53e9640b.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/.transpyc_cache/54014f6c53e9640b.pyi b/.transpyc_cache/54014f6c53e9640b.pyi new file mode 100644 index 0000000..9662312 --- /dev/null +++ b/.transpyc_cache/54014f6c53e9640b.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: t.CChar | t.CPtr) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/.transpyc_cache/541a090dde0cc1ba.doc.json b/.transpyc_cache/541a090dde0cc1ba.doc.json new file mode 100644 index 0000000..42f2fd1 --- /dev/null +++ b/.transpyc_cache/541a090dde0cc1ba.doc.json @@ -0,0 +1 @@ +{"AugAssignHandle": "增强赋值处理器 (+=, -=, *=, etc.)：继承 Mixin 获得 Trans 回指针", "AugAssignHandle.Handle": "翻译增强赋值语句 (x += 1, y -= 2, etc.)", "_apply_aug_op": "应用增强赋值的二元运算（指针算术 + 整数运算）"} \ No newline at end of file diff --git a/.transpyc_cache/541a090dde0cc1ba.pyi b/.transpyc_cache/541a090dde0cc1ba.pyi new file mode 100644 index 0000000..ee3342d --- /dev/null +++ b/.transpyc_cache/541a090dde0cc1ba.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAugAssign.py +Module: lib.core.Handles.HandlesAugAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import stdio +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal + +@t.NoVTable +class AugAssignHandle(HandlesBase.Mixin): + def __init__(self: AugAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AugAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def _apply_aug_op(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, op: int, lhs: llvmlite.Value | t.CPtr, rhs: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewAugAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AugAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/541a1ed9b8be53c8.doc.json b/.transpyc_cache/541a1ed9b8be53c8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/541a1ed9b8be53c8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/541a1ed9b8be53c8.pyi b/.transpyc_cache/541a1ed9b8be53c8.pyi new file mode 100644 index 0000000..a98f466 --- /dev/null +++ b/.transpyc_cache/541a1ed9b8be53c8.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import config +import hashlib +import binascii +import zlib.pyzlib as zp +import t, c +import testcheck + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/542aabc6fb546c35.doc.json b/.transpyc_cache/542aabc6fb546c35.doc.json new file mode 100644 index 0000000..67e7a8b --- /dev/null +++ b/.transpyc_cache/542aabc6fb546c35.doc.json @@ -0,0 +1 @@ +{"_DiagAppend": "追加诊断信息到 diag.txt（追加模式，不依赖 printf 缓冲）", "TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/542aabc6fb546c35.pyi b/.transpyc_cache/542aabc6fb546c35.pyi new file mode 100644 index 0000000..9d8b33e --- /dev/null +++ b/.transpyc_cache/542aabc6fb546c35.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def _DiagAppend(msg: bytes) -> t.CInt: pass + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/55cbbb5dec4fef2c.doc.json b/.transpyc_cache/55cbbb5dec4fef2c.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/55cbbb5dec4fef2c.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/55cbbb5dec4fef2c.pyi b/.transpyc_cache/55cbbb5dec4fef2c.pyi new file mode 100644 index 0000000..49538f9 --- /dev/null +++ b/.transpyc_cache/55cbbb5dec4fef2c.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from type_bit_test.py +Module: type_bit_test +""" + + +import stdio +import t, c + +def type_bit_test() -> int: pass diff --git a/.transpyc_cache/570919ef3f7c03ad.doc.json b/.transpyc_cache/570919ef3f7c03ad.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/570919ef3f7c03ad.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/570919ef3f7c03ad.pyi b/.transpyc_cache/570919ef3f7c03ad.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/570919ef3f7c03ad.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/57288496f7c2d1ad.pyi b/.transpyc_cache/57288496f7c2d1ad.pyi new file mode 100644 index 0000000..0624d9a --- /dev/null +++ b/.transpyc_cache/57288496f7c2d1ad.pyi @@ -0,0 +1,71 @@ +""" +Auto-generated Python stub file from json.__init__.py +Module: json.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperio + +JSON_NULL: t.CDefine = 0 +JSON_BOOL: t.CDefine = 1 +JSON_INT: t.CDefine = 2 +JSON_FLOAT: t.CDefine = 3 +JSON_STRING: t.CDefine = 4 +JSON_ARRAY: t.CDefine = 5 +JSON_OBJECT: t.CDefine = 6 + +class JsonValue: + vtype: t.CInt + bool_val: t.CInt + int_val: t.CInt64T + float_val: t.CDouble + str_val: t.CChar | t.CPtr + next: 'JsonValue' | t.CPtr + key: t.CChar | t.CPtr + child: 'JsonValue' | t.CPtr + child_count: t.CSizeT + def __new__(self: JsonValue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def type(self: JsonValue) -> t.CInt: pass + def is_null(self: JsonValue) -> bool: pass + def is_bool(self: JsonValue) -> bool: pass + def is_int(self: JsonValue) -> bool: pass + def is_float(self: JsonValue) -> bool: pass + def is_string(self: JsonValue) -> bool: pass + def is_array(self: JsonValue) -> bool: pass + def is_object(self: JsonValue) -> bool: pass + def as_bool(self: JsonValue) -> bool: pass + def as_int(self: JsonValue) -> t.CInt64T: pass + def as_float(self: JsonValue) -> t.CDouble: pass + def as_string(self: JsonValue) -> t.CChar | t.CPtr: pass + def __len__(self: JsonValue) -> t.CSizeT: pass + def __getitem__(self: JsonValue, key: t.CChar | t.CPtr) -> 'JsonValue' | t.CPtr: pass + def __setitem__(self: JsonValue, key: t.CChar | t.CPtr, val: 'JsonValue' | t.CPtr) -> t.CInt: pass + def get_item(self: JsonValue, index: t.CSizeT) -> 'JsonValue' | t.CPtr: pass + def _pool_alloc(self: JsonValue, size: t.CSizeT) -> t.CChar | t.CPtr: pass + def _append_child(self: JsonValue, child: 'JsonValue' | t.CPtr) -> t.CInt: pass + +def null(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def bool_val(pool: memhub.MemManager | t.CPtr, val: bool) -> JsonValue | t.CPtr: pass + +def int_val(pool: memhub.MemManager | t.CPtr, val: t.CInt64T) -> JsonValue | t.CPtr: pass + +def float_val(pool: memhub.MemManager | t.CPtr, val: t.CDouble) -> JsonValue | t.CPtr: pass + +def string_val(pool: memhub.MemManager | t.CPtr, val: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def object(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def array_append(pool: memhub.MemManager | t.CPtr, arr: JsonValue | t.CPtr, item: JsonValue | t.CPtr) -> t.CInt: pass + +def object_set(pool: memhub.MemManager | t.CPtr, obj: JsonValue | t.CPtr, key: t.CChar | t.CPtr, val: JsonValue | t.CPtr) -> t.CInt: pass + + +from .__parser import parse +from .__writer import write \ No newline at end of file diff --git a/.transpyc_cache/58f94f7fed04c816.doc.json b/.transpyc_cache/58f94f7fed04c816.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/58f94f7fed04c816.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/58f94f7fed04c816.pyi b/.transpyc_cache/58f94f7fed04c816.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/58f94f7fed04c816.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/594332d751ff7756.doc.json b/.transpyc_cache/594332d751ff7756.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/594332d751ff7756.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/594332d751ff7756.pyi b/.transpyc_cache/594332d751ff7756.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/594332d751ff7756.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/5974af27e1cdb182.doc.json b/.transpyc_cache/5974af27e1cdb182.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/5974af27e1cdb182.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/5974af27e1cdb182.pyi b/.transpyc_cache/5974af27e1cdb182.pyi new file mode 100644 index 0000000..3901124 --- /dev/null +++ b/.transpyc_cache/5974af27e1cdb182.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/59e00d50660b941e.doc.json b/.transpyc_cache/59e00d50660b941e.doc.json new file mode 100644 index 0000000..e833e14 --- /dev/null +++ b/.transpyc_cache/59e00d50660b941e.doc.json @@ -0,0 +1 @@ +{"os_exists": "检查文件/目录是否存在（跨平台）", "os_remove": "删除文件（跨平台）。成功返回 0，失败返回 -1。", "os_isdir": "检查路径是否为目录（跨平台）", "os_getsize": "返回文件大小（字节，跨平台）。失败返回 -1。"} \ No newline at end of file diff --git a/.transpyc_cache/59e00d50660b941e.pyi b/.transpyc_cache/59e00d50660b941e.pyi new file mode 100644 index 0000000..5236edb --- /dev/null +++ b/.transpyc_cache/59e00d50660b941e.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from os_backend.py +Module: os_backend +""" + + +import t, c +from stdint import * +import platmacro +import os._win32 +import os._posix + +def os_exists(path: str) -> bool: pass + +def os_remove(path: str) -> int: pass + +def os_isdir(path: str) -> bool: pass + +def os_getsize(path: str) -> t.CInt64T: pass diff --git a/.transpyc_cache/59f7d653d72cce3a.doc.json b/.transpyc_cache/59f7d653d72cce3a.doc.json new file mode 100644 index 0000000..7996d5c --- /dev/null +++ b/.transpyc_cache/59f7d653d72cce3a.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/59f7d653d72cce3a.pyi b/.transpyc_cache/59f7d653d72cce3a.pyi new file mode 100644 index 0000000..9aa684e --- /dev/null +++ b/.transpyc_cache/59f7d653d72cce3a.pyi @@ -0,0 +1,69 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/5b12e242d96b322f.doc.json b/.transpyc_cache/5b12e242d96b322f.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/5b12e242d96b322f.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/5b12e242d96b322f.pyi b/.transpyc_cache/5b12e242d96b322f.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/5b12e242d96b322f.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/5b2e220c5bae60c9.pyi b/.transpyc_cache/5b2e220c5bae60c9.pyi new file mode 100644 index 0000000..364be1a --- /dev/null +++ b/.transpyc_cache/5b2e220c5bae60c9.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from _withcontent.py +Module: _withcontent +""" + + +import t, c +from stdint import * +from w32.win32process import TlsAlloc, TlsGetValue, TlsSetValue +import stdlib +import stdio +import string + +_MAX_PROVIDER_DEPTH: t.CDefine = 32 + +@t.Object +class _ProviderStack: + names: t.CArray[VOIDPTR, 32] + ptrs: t.CArray[VOIDPTR, 32] + top: UINT64 + +_tls_index: t.CExtern | ULONG + +@t.TLS +def _init_tls() -> t.CInt: pass + +def _get_stack() -> _ProviderStack | t.CPtr: pass + +def _push_provider(field_name: str, ptr: t.CPtr) -> t.CInt: pass + +def _pop_provider() -> t.CInt: pass + +def _find_provider(field_name: str) -> t.CPtr: pass diff --git a/.transpyc_cache/5dab8cb390496d22.pyi b/.transpyc_cache/5dab8cb390496d22.pyi new file mode 100644 index 0000000..7056767 --- /dev/null +++ b/.transpyc_cache/5dab8cb390496d22.pyi @@ -0,0 +1,182 @@ +""" +Auto-generated Python stub file from ast.base.py +Module: ast.base +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .tokens import TokOp + +class ASTKind(t.CEnum): + Module: t.State + FunctionDef: t.State + ClassDef: t.State + Assign: t.State + If: t.State + For: t.State + While: t.State + Return: t.State + Expr: t.State + Name: t.State + Constant: t.State + BinOp: t.State + UnaryOp: t.State + Call: t.State + Compare: t.State + Expression: t.State + Interactive: t.State + FunctionType: t.State + Delete: t.State + AugAssign: t.State + AnnAssign: t.State + With: t.State + Raise: t.State + Try: t.State + Assert: t.State + Global: t.State + Nonlocal: t.State + Pass: t.State + Break: t.State + Continue: t.State + Import: t.State + ImportFrom: t.State + Match: t.State + BoolOp: t.State + Lambda: t.State + IfExp: t.State + Dict: t.State + Set: t.State + ListComp: t.State + SetComp: t.State + DictComp: t.State + GeneratorExp: t.State + Await: t.State + Yield: t.State + YieldFrom: t.State + FormattedValue: t.State + JoinedStr: t.State + Attribute: t.State + Subscript: t.State + Starred: t.State + List: t.State + Tuple: t.State + Slice: t.State + NamedExpr: t.State + ExceptHandler: t.State + Arguments: t.State + Arg: t.State + Keyword: t.State + Alias: t.State + WithItem: t.State + Comprehension: t.State + OpNode: t.State + MatchCase: t.State + MatchValue: t.State + MatchSingleton: t.State + MatchSequence: t.State + MatchMapping: t.State + MatchClass: t.State + MatchStar: t.State + MatchAs: t.State + MatchOr: t.State +class ASTCtx(t.CEnum): + Load: t.State + Store: t.State + Del: t.State +class OpKind(t.CEnum): + Add: t.State + Sub: t.State + Mult: t.State + MatMult: t.State + Div: t.State + Mod: t.State + Pow: t.State + LShift: t.State + RShift: t.State + BitOr: t.State + BitXor: t.State + BitAnd: t.State + FloorDiv: t.State + And: t.State + Or: t.State + Eq: t.State + Ne: t.State + Lt: t.State + Le: t.State + Gt: t.State + Ge: t.State + Is: t.State + IsNot: t.State + In: t.State + NotIn: t.State + Not: t.State + UAdd: t.State + USub: t.State + Invert: t.State + NoneOp: t.State + +CONST_INT: t.CDefine = 1 +CONST_FLOAT: t.CDefine = 2 +CONST_STR: t.CDefine = 3 +CONST_BOOL: t.CDefine = 4 +CONST_NONE: t.CDefine = 5 +FLAG_IS_ASYNC: t.CDefine = 1 +FLAG_SIMPLE: t.CDefine = 2 +FLAG_HAS_STAR: t.CDefine = 4 + +class ASTFlag(t.CEnum): + IsAsync: t.State + Simple: t.State + HasStar: t.State + +def _copy_str(pool: memhub.MemManager | t.CPtr, src: str) -> str: pass + +def _set_pos(node: AST | t.CPtr, lineno: t.CInt, col_offset: t.CInt, end_lineno: t.CInt, end_col_offset: t.CInt) -> t.CInt: pass + +def _inherit_pos(node: AST | t.CPtr, ref: AST | t.CPtr) -> t.CInt: pass + +def _init_ast(node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + +def _set_parent_list(lst: list[AST | t.CPtr] | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _append_child(lst: list[AST | t.CPtr] | t.CPtr, node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _binop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _augop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _cmpop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _emit(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_str(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_int(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, val: t.CInt64T) -> t.CSizeT: pass + +def _op_name(op: t.CInt) -> str: pass + +def _dump_list(lst: list[AST | t.CPtr] | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + +def _dump_op_list(lst: t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + + +@t.CVTable +class AST: + parent: AST | t.CPtr + pool: memhub.MemManager | t.CPtr + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + children: list[AST | t.CPtr] | t.CPtr + def kind(self: AST) -> t.CInt: pass + def type_name(self: AST) -> str: pass + def dump(self: AST, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + def accept(self: AST, visitor: t.CPtr) -> t.CInt: pass + def append(self: AST, node: AST | t.CPtr) -> t.CInt: pass + +def dump(node: AST | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: pass diff --git a/.transpyc_cache/5e3d33857a0e37f6.doc.json b/.transpyc_cache/5e3d33857a0e37f6.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/5e3d33857a0e37f6.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/5e3d33857a0e37f6.pyi b/.transpyc_cache/5e3d33857a0e37f6.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/5e3d33857a0e37f6.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/5e430ae5bf7af3a2.doc.json b/.transpyc_cache/5e430ae5bf7af3a2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/5e430ae5bf7af3a2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/5e430ae5bf7af3a2.pyi b/.transpyc_cache/5e430ae5bf7af3a2.pyi new file mode 100644 index 0000000..bbe03a4 --- /dev/null +++ b/.transpyc_cache/5e430ae5bf7af3a2.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import mbuddytest + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/5faaecd48491dce0.doc.json b/.transpyc_cache/5faaecd48491dce0.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/5faaecd48491dce0.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/5faaecd48491dce0.pyi b/.transpyc_cache/5faaecd48491dce0.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/5faaecd48491dce0.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/5fc006555963c808.doc.json b/.transpyc_cache/5fc006555963c808.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/5fc006555963c808.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/5fc006555963c808.pyi b/.transpyc_cache/5fc006555963c808.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/5fc006555963c808.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/605efc3aeda6b7a7.doc.json b/.transpyc_cache/605efc3aeda6b7a7.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/605efc3aeda6b7a7.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/605efc3aeda6b7a7.pyi b/.transpyc_cache/605efc3aeda6b7a7.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/605efc3aeda6b7a7.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/62cc01c2bb770ca3.pyi b/.transpyc_cache/62cc01c2bb770ca3.pyi new file mode 100644 index 0000000..17b1307 --- /dev/null +++ b/.transpyc_cache/62cc01c2bb770ca3.pyi @@ -0,0 +1,77 @@ +""" +Auto-generated Python stub file from vrandom.py +Module: vrandom +""" + + +import t, c +import vipermath + +RDRAND_RETRY_LIMIT: t.CDefine = 10 +RDSEED_RETRY_LIMIT: t.CDefine = 100 + +def rdrand16() -> t.CUInt16T: pass + +def rdrand32() -> t.CUInt32T: pass + +def rdrand64() -> t.CUInt64T: pass + +def rdseed16() -> t.CUInt16T: pass + +def rdseed32() -> t.CUInt32T: pass + +def rdseed64() -> t.CUInt64T: pass + +def rdrand16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: pass + +def rdrand32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: pass + +def rdrand64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: pass + +def rdseed16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: pass + +def rdseed32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: pass + +def rdseed64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: pass + +def random_u16() -> t.CUInt16T: pass + +def random_u32() -> t.CUInt32T: pass + +def random_u64() -> t.CUInt64T: pass + +def seed_u16() -> t.CUInt16T: pass + +def seed_u32() -> t.CUInt32T: pass + +def seed_u64() -> t.CUInt64T: pass + +def randint(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def randrange(start: t.CInt, stop: t.CInt, step: t.CInt) -> t.CInt: pass + +def getrandbits(k: t.CInt) -> t.CUInt64T: pass + +def random() -> t.CDouble: pass + +def uniform(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def triangular(low: t.CDouble, high: t.CDouble, mode: t.CDouble) -> t.CDouble: pass + +def betavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass + +def expovariate(lambd: t.CDouble) -> t.CDouble: pass + +def gammavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass + +def gauss(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def normalvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def lognormvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def vonmisesvariate(mu: t.CDouble, kappa: t.CDouble) -> t.CDouble: pass + +def paretovariate(alpha: t.CDouble) -> t.CDouble: pass + +def weibullvariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass diff --git a/.transpyc_cache/6446627d4f07a1b5.pyi b/.transpyc_cache/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/.transpyc_cache/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/.transpyc_cache/64ac83614c9bdbc6.doc.json b/.transpyc_cache/64ac83614c9bdbc6.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/64ac83614c9bdbc6.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/64ac83614c9bdbc6.pyi b/.transpyc_cache/64ac83614c9bdbc6.pyi new file mode 100644 index 0000000..60711e1 --- /dev/null +++ b/.transpyc_cache/64ac83614c9bdbc6.pyi @@ -0,0 +1,85 @@ +""" +Auto-generated Python stub file from definetest.py +Module: definetest +""" + + +import t, c +from t import CInt, CInt32T, CUInt32T, CUInt64T, CFloat64T, CPtr, CChar, CDefine, CUInt16T +import viperlib +import stdlib +import platmacro + +MAX_SIZE: t.CDefine = 1024 +MIN_VAL: t.CDefine = 0 +MAX_VAL: t.CDefine = 100 +MASK_BYTE: t.CDefine = 0xFF +SHIFT_BITS: t.CDefine = 8 +PI_INT: t.CDefine = 3 +ENABLED: t.CDefine = 1 +DISABLED: t.CDefine = 0 +PAGE_SIZE: t.CDefine = 4096 +BUFFER_LEN: t.CDefine = 256 +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +DATA_SEG: t.CDefine | t.CUInt16T = 0x10 +STACK_SEG: t.CDefine | t.CUInt16T = 0x20 + +def MAKE_FLAG(bit: CInt32T) -> t.CDefine: + return 1 << bit + +def CLAMP(x: CInt32T, lo: CInt32T, hi: CInt32T) -> t.CDefine: + if x < lo: + return lo + if x > hi: + return hi + return x + +def test_basic_define() -> CInt: pass + +def test_define_arithmetic() -> CInt: pass + +def test_define_conditional() -> CInt: pass + +def test_define_loop() -> CInt: pass + +def test_typed_define() -> CInt: pass + +def test_define_function() -> CInt: pass + +def test_define_clamp() -> CInt: pass + +def test_define_string() -> CInt: pass + +def test_define_combined() -> CInt: pass + +def test_cif_basic() -> CInt: pass + +def test_cif_elif_else() -> CInt: pass + +def test_cifdef_cifndef() -> CInt: pass + +def test_cif_platform() -> CInt: pass + +def test_cif_nested() -> CInt: pass + +def test_cif_expr() -> CInt: pass + +def test_platmacro_platform() -> CInt: pass + +def test_platmacro_sizes() -> CInt: pass + +def test_platmacro_endian() -> CInt: pass + +def test_platmacro_features() -> CInt: pass + +def test_platmacro_align() -> CInt: pass + +def test_cifdef_cross_module() -> CInt: pass + +def test_cifdef_guard() -> CInt: pass + +def test_cifdef_undefined() -> CInt: pass + +def test_cifdef_complex() -> CInt: pass + +def define_main() -> CInt: pass diff --git a/.transpyc_cache/6503c97dde0c79c4.pyi b/.transpyc_cache/6503c97dde0c79c4.pyi new file mode 100644 index 0000000..d332326 --- /dev/null +++ b/.transpyc_cache/6503c97dde0c79c4.pyi @@ -0,0 +1,189 @@ +""" +Auto-generated Python stub file from posix.py +Module: posix +""" + + +import t, c +from stdint import * + +DIR: t.CTypedef = t.CVoid | t.CPtr +mode_t: t.CTypedef = t.CUInt32T +pid_t: t.CTypedef = t.CInt32T +off_t: t.CTypedef = t.CInt64T +dev_t: t.CTypedef = t.CUInt64T +ino_t: t.CTypedef = t.CUInt64T +nlink_t: t.CTypedef = t.CUInt64T +uid_t: t.CTypedef = t.CUInt32T +gid_t: t.CTypedef = t.CUInt32T +blksize_t: t.CTypedef = t.CInt64T +blkcnt_t: t.CTypedef = t.CInt64T +time_t: t.CTypedef = t.CInt64T +O_RDONLY: t.CDefine = 0 +O_WRONLY: t.CDefine = 1 +O_RDWR: t.CDefine = 2 +O_CREAT: t.CDefine = 0o100 +O_EXCL: t.CDefine = 0o200 +O_TRUNC: t.CDefine = 0o1000 +O_APPEND: t.CDefine = 0o2000 +O_NONBLOCK: t.CDefine = 0o4000 +O_CLOEXEC: t.CDefine = 0o2000000 +F_OK: t.CDefine = 0 # 文件存在 +R_OK: t.CDefine = 4 # 可读 +W_OK: t.CDefine = 2 # 可写 +X_OK: t.CDefine = 1 # 可执行 +S_IFMT: t.CDefine = 0o170000 # 类型掩码 +S_IFSOCK: t.CDefine = 0o140000 # 套接字 +S_IFLNK: t.CDefine = 0o120000 # 符号链接 +S_IFREG: t.CDefine = 0o100000 # 普通文件 +S_IFBLK: t.CDefine = 0o060000 # 块设备 +S_IFDIR: t.CDefine = 0o040000 # 目录 +S_IFCHR: t.CDefine = 0o020000 # 字符设备 +S_IFIFO: t.CDefine = 0o010000 # FIFO/管道 +S_ISUID: t.CDefine = 0o4000 # set-user-ID +S_ISGID: t.CDefine = 0o2000 # set-group-ID +S_ISVTX: t.CDefine = 0o1000 # sticky 位 +S_IRWXU: t.CDefine = 0o700 # 用户读写执行 +S_IRUSR: t.CDefine = 0o400 # 用户读 +S_IWUSR: t.CDefine = 0o200 # 用户写 +S_IXUSR: t.CDefine = 0o100 # 用户执行 +S_IRWXG: t.CDefine = 0o070 # 组读写执行 +S_IRGRP: t.CDefine = 0o040 # 组读 +S_IWGRP: t.CDefine = 0o020 # 组写 +S_IXGRP: t.CDefine = 0o010 # 组执行 +S_IRWXO: t.CDefine = 0o007 # 其他读写执行 +S_IROTH: t.CDefine = 0o004 # 其他读 +S_IWOTH: t.CDefine = 0o002 # 其他写 +S_IXOTH: t.CDefine = 0o001 # 其他执行 +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +PATH_MAX: t.CDefine = 4096 +NAME_MAX: t.CDefine = 255 +_SC_NPROCESSORS_ONLN: t.CDefine = 84 +_SC_PAGESIZE: t.CDefine = 30 +_SC_CLK_TCK: t.CDefine = 2 +DT_UNKNOWN: t.CDefine = 0 +DT_FIFO: t.CDefine = 1 +DT_CHR: t.CDefine = 2 +DT_DIR: t.CDefine = 4 +DT_BLK: t.CDefine = 6 +DT_REG: t.CDefine = 8 +DT_LNK: t.CDefine = 10 +DT_SOCK: t.CDefine = 12 +DT_WHT: t.CDefine = 14 +WNOHANG: t.CDefine = 1 +WUNTRACED: t.CDefine = 2 +WCONTINUED: t.CDefine = 8 + +class Stat: + st_dev: dev_t + st_ino: ino_t + st_nlink: nlink_t + st_mode: mode_t + st_uid: uid_t + st_gid: gid_t + __pad0: INT + st_rdev: dev_t + st_size: off_t + st_blksize: blksize_t + st_blocks: blkcnt_t + st_atim_sec: time_t + st_atim_nsec: LONGLONG + st_mtim_sec: time_t + st_mtim_nsec: LONGLONG + st_ctim_sec: time_t + st_ctim_nsec: LONGLONG + __unused0: LONGLONG + __unused1: LONGLONG + __unused2: LONGLONG +class Dirent: + d_ino: ino_t + d_off: off_t + d_reclen: USHORT + d_type: BYTE + d_name: CHARPTR + +def open(path: t.CConst | str, oflag: INT, *args) -> INT | t.State: pass + +def close(fd: INT) -> INT | t.State: pass + +def read(fd: INT, buf: bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def write(fd: INT, buf: t.CConst | bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def lseek(fd: INT, offset: off_t, whence: INT) -> off_t | t.State: pass + +def unlink(path: t.CConst | str) -> INT | t.State: pass + +def access(path: t.CConst | str, amode: INT) -> INT | t.State: pass + +def realpath(path: t.CConst | str, resolved: CHARPTR) -> CHARPTR | t.State: pass + +def rename(oldpath: t.CConst | str, newpath: t.CConst | str) -> INT | t.State: pass + +def chmod(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def stat(path: t.CConst | str, buf: Stat | t.CPtr) -> INT | t.State: pass + +def fstat(fd: INT, buf: Stat | t.CPtr) -> INT | t.State: pass + +def opendir(name: t.CConst | str) -> DIR | t.CPtr | t.State: pass + +def readdir(dirp: DIR | t.CPtr) -> Dirent | t.CPtr | t.State: pass + +def closedir(dirp: DIR | t.CPtr) -> INT | t.State: pass + +def mkdir(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def rmdir(path: t.CConst | str) -> INT | t.State: pass + +def getcwd(buf: CHARPTR, size: t.CSizeT) -> CHARPTR | t.State: pass + +def chdir(path: t.CConst | str) -> INT | t.State: pass + +def fork() -> pid_t | t.State: pass + +def execvp(file: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def execv(path: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def waitpid(pid: pid_t, status: INT | t.CPtr, options: INT) -> pid_t | t.State: pass + +def pipe(fds: INT | t.CPtr) -> INT | t.State: pass + +def dup2(oldfd: INT, newfd: INT) -> INT | t.State: pass + +def _exit(status: INT) -> t.CVoid | t.State: pass + +def getpid() -> pid_t | t.State: pass + +def getenv(name: t.CConst | str) -> CHARPTR | t.State: pass + +def setenv(name: t.CConst | str, value: t.CConst | str, overwrite: INT) -> INT | t.State: pass + +def sysconf(name: INT) -> LONG | t.State: pass + +def isatty(fd: INT) -> INT | t.State: pass + +def S_ISREG(mode: mode_t) -> bool: pass + +def S_ISDIR(mode: mode_t) -> bool: pass + +def S_ISLNK(mode: mode_t) -> bool: pass + +def S_ISCHR(mode: mode_t) -> bool: pass + +def S_ISBLK(mode: mode_t) -> bool: pass + +def S_ISFIFO(mode: mode_t) -> bool: pass + +def S_ISSOCK(mode: mode_t) -> bool: pass + +def WIFEXITED(status: INT) -> bool: pass + +def WEXITSTATUS(status: INT) -> INT: pass + +def WIFSIGNALED(status: INT) -> bool: pass + +def WTERMSIG(status: INT) -> INT: pass diff --git a/.transpyc_cache/6521d5529dba8eba.doc.json b/.transpyc_cache/6521d5529dba8eba.doc.json new file mode 100644 index 0000000..4dba15e --- /dev/null +++ b/.transpyc_cache/6521d5529dba8eba.doc.json @@ -0,0 +1 @@ +{"ReturnHandle": "Return 语句处理器：继承 Mixin 获得 Trans 回指针", "ReturnHandle.Handle": "翻译 Return 语句"} \ No newline at end of file diff --git a/.transpyc_cache/6521d5529dba8eba.pyi b/.transpyc_cache/6521d5529dba8eba.pyi new file mode 100644 index 0000000..abcbbff --- /dev/null +++ b/.transpyc_cache/6521d5529dba8eba.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesReturn.py +Module: lib.core.Handles.HandlesReturn +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar + +@t.NoVTable +class ReturnHandle(HandlesBase.Mixin): + def __init__(self: ReturnHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ReturnHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewReturnHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ReturnHandle | t.CPtr: pass diff --git a/.transpyc_cache/657e182b27c2a022.pyi b/.transpyc_cache/657e182b27c2a022.pyi new file mode 100644 index 0000000..0a2057d --- /dev/null +++ b/.transpyc_cache/657e182b27c2a022.pyi @@ -0,0 +1,226 @@ +""" +Auto-generated Python stub file from ast.stmts.py +Module: ast.stmts +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, FLAG_IS_ASYNC + +@t.CVTable +class Module(AST): + def __new__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Module) -> t.CInt: pass + def type_name(self: Module) -> str: pass + def dump(self: Module, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assign(AST): + targets: list[AST | t.CPtr] | t.CPtr + value: AST | t.CPtr + def __new__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assign) -> t.CInt: pass + def type_name(self: Assign) -> str: pass + def dump(self: Assign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class If(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: If) -> t.CInt: pass + def type_name(self: If) -> str: pass + def dump(self: If, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class For(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: For) -> t.CInt: pass + def type_name(self: For) -> str: pass + def dump(self: For, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class While(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: While) -> t.CInt: pass + def type_name(self: While) -> str: pass + def dump(self: While, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Return(AST): + value: AST | t.CPtr + def __new__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Return) -> t.CInt: pass + def type_name(self: Return) -> str: pass + def dump(self: Return, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expr(AST): + value: AST | t.CPtr + def __new__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expr) -> t.CInt: pass + def type_name(self: Expr) -> str: pass + def dump(self: Expr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionDef(AST): + name: str + args: AST | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + flags: t.CInt + def __new__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: FunctionDef) -> t.CInt: pass + def type_name(self: FunctionDef) -> str: pass + def dump(self: FunctionDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expression(AST): + body: AST | t.CPtr + def __new__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expression) -> t.CInt: pass + def type_name(self: Expression) -> str: pass + def dump(self: Expression, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Interactive(AST): + body: AST | t.CPtr + def __new__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Interactive) -> t.CInt: pass + def type_name(self: Interactive) -> str: pass + def dump(self: Interactive, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionType(AST): + argtypes: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + def __new__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def kind(self: FunctionType) -> t.CInt: pass + def type_name(self: FunctionType) -> str: pass + def dump(self: FunctionType, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ClassDef(AST): + name: str + bases: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + type_params: list[str] | t.CPtr + def __new__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def __init__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def kind(self: ClassDef) -> t.CInt: pass + def type_name(self: ClassDef) -> str: pass + def dump(self: ClassDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Delete(AST): + targets: list[AST | t.CPtr] | t.CPtr + def __new__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Delete) -> t.CInt: pass + def type_name(self: Delete) -> str: pass + def dump(self: Delete, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AugAssign(AST): + target: AST | t.CPtr + op: t.CInt + value: AST | t.CPtr + def __new__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: AugAssign) -> t.CInt: pass + def type_name(self: AugAssign) -> str: pass + def dump(self: AugAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AnnAssign(AST): + target: AST | t.CPtr + annotation: AST | t.CPtr + value: AST | t.CPtr + simple: t.CInt + def __new__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def __init__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def kind(self: AnnAssign) -> t.CInt: pass + def type_name(self: AnnAssign) -> str: pass + def dump(self: AnnAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class With(AST): + items: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: With) -> t.CInt: pass + def type_name(self: With) -> str: pass + def dump(self: With, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Raise(AST): + exc: AST | t.CPtr + cause: AST | t.CPtr + def __new__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def kind(self: Raise) -> t.CInt: pass + def type_name(self: Raise) -> str: pass + def dump(self: Raise, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Try(AST): + handlers: list[AST | t.CPtr] | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + finalbody: list[AST | t.CPtr] | t.CPtr + def __new__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Try) -> t.CInt: pass + def type_name(self: Try) -> str: pass + def dump(self: Try, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assert(AST): + test: AST | t.CPtr + msg: AST | t.CPtr + def __new__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assert) -> t.CInt: pass + def type_name(self: Assert) -> str: pass + def dump(self: Assert, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Global(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Global) -> t.CInt: pass + def type_name(self: Global) -> str: pass + def dump(self: Global, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Nonlocal(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Nonlocal) -> t.CInt: pass + def type_name(self: Nonlocal) -> str: pass + def dump(self: Nonlocal, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Pass(AST): + def __new__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Pass) -> t.CInt: pass + def type_name(self: Pass) -> str: pass + def dump(self: Pass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Break(AST): + def __new__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Break) -> t.CInt: pass + def type_name(self: Break) -> str: pass + def dump(self: Break, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Continue(AST): + def __new__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Continue) -> t.CInt: pass + def type_name(self: Continue) -> str: pass + def dump(self: Continue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Import(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Import) -> t.CInt: pass + def type_name(self: Import) -> str: pass + def dump(self: Import, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ImportFrom(AST): + module: str + names: list[AST | t.CPtr] | t.CPtr + level: t.CInt + def __new__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def __init__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def kind(self: ImportFrom) -> t.CInt: pass + def type_name(self: ImportFrom) -> str: pass + def dump(self: ImportFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Match(AST): + subject: AST | t.CPtr + cases: list[AST | t.CPtr] | t.CPtr + def __new__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Match) -> t.CInt: pass + def type_name(self: Match) -> str: pass + def dump(self: Match, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/.transpyc_cache/65c212dc0785f1aa.doc.json b/.transpyc_cache/65c212dc0785f1aa.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/65c212dc0785f1aa.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/65c212dc0785f1aa.pyi b/.transpyc_cache/65c212dc0785f1aa.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/65c212dc0785f1aa.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/663db7700139418e.doc.json b/.transpyc_cache/663db7700139418e.doc.json new file mode 100644 index 0000000..e7fc1ec --- /dev/null +++ b/.transpyc_cache/663db7700139418e.doc.json @@ -0,0 +1 @@ +{"TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/663db7700139418e.pyi b/.transpyc_cache/663db7700139418e.pyi new file mode 100644 index 0000000..3a8c9c7 --- /dev/null +++ b/.transpyc_cache/663db7700139418e.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/66de93aaeebc0601.doc.json b/.transpyc_cache/66de93aaeebc0601.doc.json new file mode 100644 index 0000000..046af31 --- /dev/null +++ b/.transpyc_cache/66de93aaeebc0601.doc.json @@ -0,0 +1 @@ +{"IfHandle": "if/elif/else 语句处理器：继承 Mixin 获得 Trans 回指针", "IfHandle.Handle": "翻译 if/elif/else 语句"} \ No newline at end of file diff --git a/.transpyc_cache/66de93aaeebc0601.pyi b/.transpyc_cache/66de93aaeebc0601.pyi new file mode 100644 index 0000000..2981cf3 --- /dev/null +++ b/.transpyc_cache/66de93aaeebc0601.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesIf.py +Module: lib.core.Handles.HandlesIf +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class IfHandle(HandlesBase.Mixin): + def __init__(self: IfHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: IfHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewIfHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> IfHandle | t.CPtr: pass diff --git a/.transpyc_cache/6712b7f4f0f0e65d.doc.json b/.transpyc_cache/6712b7f4f0f0e65d.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/6712b7f4f0f0e65d.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/6712b7f4f0f0e65d.pyi b/.transpyc_cache/6712b7f4f0f0e65d.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/6712b7f4f0f0e65d.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/67e6e5ea72c0e13c.doc.json b/.transpyc_cache/67e6e5ea72c0e13c.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/67e6e5ea72c0e13c.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/67e6e5ea72c0e13c.pyi b/.transpyc_cache/67e6e5ea72c0e13c.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/67e6e5ea72c0e13c.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/68ee8e1ac0994b13.doc.json b/.transpyc_cache/68ee8e1ac0994b13.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/.transpyc_cache/68ee8e1ac0994b13.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/.transpyc_cache/68ee8e1ac0994b13.pyi b/.transpyc_cache/68ee8e1ac0994b13.pyi new file mode 100644 index 0000000..24ef0f1 --- /dev/null +++ b/.transpyc_cache/68ee8e1ac0994b13.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: str +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: str) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/.transpyc_cache/6961b76a167891e5.doc.json b/.transpyc_cache/6961b76a167891e5.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/.transpyc_cache/6961b76a167891e5.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/.transpyc_cache/6961b76a167891e5.pyi b/.transpyc_cache/6961b76a167891e5.pyi new file mode 100644 index 0000000..89de3bc --- /dev/null +++ b/.transpyc_cache/6961b76a167891e5.pyi @@ -0,0 +1,88 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/69b86cf4726da53f.doc.json b/.transpyc_cache/69b86cf4726da53f.doc.json new file mode 100644 index 0000000..9c0ba29 --- /dev/null +++ b/.transpyc_cache/69b86cf4726da53f.doc.json @@ -0,0 +1 @@ +{"AnnAssignHandle": "AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针", "AnnAssignHandle.PreScan": "预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部", "AnnAssignHandle.Handle": "翻译 AnnAssign(target=Name, annotation=..., value=expr)"} \ No newline at end of file diff --git a/.transpyc_cache/69b86cf4726da53f.pyi b/.transpyc_cache/69b86cf4726da53f.pyi new file mode 100644 index 0000000..ef1d109 --- /dev/null +++ b/.transpyc_cache/69b86cf4726da53f.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAnnAssign.py +Module: lib.core.Handles.HandlesAnnAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesExprCall as HandlesExprCall + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + def __init__(self: AnnAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def PreScan(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + def Handle(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/6a1088ddb47c1d62.doc.json b/.transpyc_cache/6a1088ddb47c1d62.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/6a1088ddb47c1d62.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/6a1088ddb47c1d62.pyi b/.transpyc_cache/6a1088ddb47c1d62.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/6a1088ddb47c1d62.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/6ab1c6b14ca3ded8.doc.json b/.transpyc_cache/6ab1c6b14ca3ded8.doc.json new file mode 100644 index 0000000..e7fc1ec --- /dev/null +++ b/.transpyc_cache/6ab1c6b14ca3ded8.doc.json @@ -0,0 +1 @@ +{"TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/6ab1c6b14ca3ded8.pyi b/.transpyc_cache/6ab1c6b14ca3ded8.pyi new file mode 100644 index 0000000..3a8c9c7 --- /dev/null +++ b/.transpyc_cache/6ab1c6b14ca3ded8.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/6b8853854d6ef556.doc.json b/.transpyc_cache/6b8853854d6ef556.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/6b8853854d6ef556.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/6b8853854d6ef556.pyi b/.transpyc_cache/6b8853854d6ef556.pyi new file mode 100644 index 0000000..66d8d02 --- /dev/null +++ b/.transpyc_cache/6b8853854d6ef556.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +def test_for_in_string() -> t.CInt: pass + +def test_for_in_char_ptr() -> t.CInt: pass + +def test_strcpy_for_in() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/6ba73c618fc5471e.doc.json b/.transpyc_cache/6ba73c618fc5471e.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/6ba73c618fc5471e.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/6ba73c618fc5471e.pyi b/.transpyc_cache/6ba73c618fc5471e.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/6ba73c618fc5471e.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/6deb1a24b47f47fc.doc.json b/.transpyc_cache/6deb1a24b47f47fc.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/6deb1a24b47f47fc.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/6deb1a24b47f47fc.pyi b/.transpyc_cache/6deb1a24b47f47fc.pyi new file mode 100644 index 0000000..1c16d75 --- /dev/null +++ b/.transpyc_cache/6deb1a24b47f47fc.pyi @@ -0,0 +1,36 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import c +import stdio +import stdlib +import string +import memhub +import testcheck +import _fakeduck + +def test_for_in_range() -> t.CInt: pass + +def test_for_in_str() -> t.CInt: pass + +def test_for_in_list() -> t.CInt: pass + +def test_zip_list_list() -> t.CInt: pass + +def test_zip_range_list() -> t.CInt: pass + +def test_zip_str_str() -> t.CInt: pass + +def test_zip_range_len_list() -> t.CInt: pass + +def test_enumerate_list() -> t.CInt: pass + +def test_enumerate_str() -> t.CInt: pass + +def test_enumerate_range() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/6f43cc10b998bf0b.doc.json b/.transpyc_cache/6f43cc10b998bf0b.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/6f43cc10b998bf0b.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/6f43cc10b998bf0b.pyi b/.transpyc_cache/6f43cc10b998bf0b.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/6f43cc10b998bf0b.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/6f5e876273093314.doc.json b/.transpyc_cache/6f5e876273093314.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/6f5e876273093314.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/6f5e876273093314.pyi b/.transpyc_cache/6f5e876273093314.pyi new file mode 100644 index 0000000..94c6e0f --- /dev/null +++ b/.transpyc_cache/6f5e876273093314.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from deref_min_test.py +Module: deref_min_test +""" + + +import stdio +import t, c + +def deref_min_test() -> int: pass diff --git a/.transpyc_cache/6f62fe05c5ea1ceb.pyi b/.transpyc_cache/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/.transpyc_cache/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/.transpyc_cache/6ff26590374ae6fc.pyi b/.transpyc_cache/6ff26590374ae6fc.pyi new file mode 100644 index 0000000..1b1bba1 --- /dev/null +++ b/.transpyc_cache/6ff26590374ae6fc.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from hashlib.__sha512.py +Module: hashlib.__sha512 +""" + + +import t, c + +SHA512_BLOCK_LEN: t.CDefine = 128 +SHA512_DIGEST_LEN: t.CDefine = 64 + +def sha512_ror(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_shr(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_Ch(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Maj(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma1(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma1(x: t.CUInt64T) -> t.CUInt64T: pass + + +sha512_K: t.CExtern | t.CArray[t.CUInt64T, 80] + +@t.Object +class sha512: + state: t.CArray[t.CUInt64T, 8] + count: t.CArray[t.CUInt64T, 2] + buf: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN] + def __init__(self: sha512) -> t.CInt: pass + def transform(self: sha512, block: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN]) -> t.CInt: pass + def update(self: sha512, s: str) -> t.CInt: pass + def final(self: sha512, out: t.CArray[t.CUInt8T, SHA512_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/717c696d67fecfc6.doc.json b/.transpyc_cache/717c696d67fecfc6.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/717c696d67fecfc6.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/717c696d67fecfc6.pyi b/.transpyc_cache/717c696d67fecfc6.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/717c696d67fecfc6.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/71b5771c2400c31a.doc.json b/.transpyc_cache/71b5771c2400c31a.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/71b5771c2400c31a.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/71b5771c2400c31a.pyi b/.transpyc_cache/71b5771c2400c31a.pyi new file mode 100644 index 0000000..60fab85 --- /dev/null +++ b/.transpyc_cache/71b5771c2400c31a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/71d73729480f2fd2.doc.json b/.transpyc_cache/71d73729480f2fd2.doc.json new file mode 100644 index 0000000..046af31 --- /dev/null +++ b/.transpyc_cache/71d73729480f2fd2.doc.json @@ -0,0 +1 @@ +{"IfHandle": "if/elif/else 语句处理器：继承 Mixin 获得 Trans 回指针", "IfHandle.Handle": "翻译 if/elif/else 语句"} \ No newline at end of file diff --git a/.transpyc_cache/71d73729480f2fd2.pyi b/.transpyc_cache/71d73729480f2fd2.pyi new file mode 100644 index 0000000..fd54f7e --- /dev/null +++ b/.transpyc_cache/71d73729480f2fd2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesIf.py +Module: lib.core.Handles.HandlesIf +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class IfHandle(HandlesBase.Mixin): + def __init__(self: IfHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: IfHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewIfHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> IfHandle | t.CPtr: pass diff --git a/.transpyc_cache/71e0a3ffcb3ebfad.pyi b/.transpyc_cache/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/.transpyc_cache/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/.transpyc_cache/7259cdb36d66dc67.doc.json b/.transpyc_cache/7259cdb36d66dc67.doc.json new file mode 100644 index 0000000..c3a1e25 --- /dev/null +++ b/.transpyc_cache/7259cdb36d66dc67.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/7259cdb36d66dc67.pyi b/.transpyc_cache/7259cdb36d66dc67.pyi new file mode 100644 index 0000000..ad12e1f --- /dev/null +++ b/.transpyc_cache/7259cdb36d66dc67.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 32 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/72e2d5ccb7cedcf1.pyi b/.transpyc_cache/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/.transpyc_cache/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/.transpyc_cache/73a5dbb3cd24b3bf.doc.json b/.transpyc_cache/73a5dbb3cd24b3bf.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/.transpyc_cache/73a5dbb3cd24b3bf.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/.transpyc_cache/73a5dbb3cd24b3bf.pyi b/.transpyc_cache/73a5dbb3cd24b3bf.pyi new file mode 100644 index 0000000..1e1b4bd --- /dev/null +++ b/.transpyc_cache/73a5dbb3cd24b3bf.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/73fcdc4dd1618580.pyi b/.transpyc_cache/73fcdc4dd1618580.pyi new file mode 100644 index 0000000..d94a709 --- /dev/null +++ b/.transpyc_cache/73fcdc4dd1618580.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from hashlib.__sha1.py +Module: hashlib.__sha1 +""" + + +import t, c + +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self: sha1) -> t.CInt: pass + def transform(self: sha1, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha1, s: str) -> t.CInt: pass + def final(self: sha1, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/744224bb5e87b5ef.doc.json b/.transpyc_cache/744224bb5e87b5ef.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/744224bb5e87b5ef.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/744224bb5e87b5ef.pyi b/.transpyc_cache/744224bb5e87b5ef.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/744224bb5e87b5ef.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/74ca31446cac2107.doc.json b/.transpyc_cache/74ca31446cac2107.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/74ca31446cac2107.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/74ca31446cac2107.pyi b/.transpyc_cache/74ca31446cac2107.pyi new file mode 100644 index 0000000..5aca605 --- /dev/null +++ b/.transpyc_cache/74ca31446cac2107.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from string_min_test.py +Module: string_min_test +""" + + +import stdio +import t, c +import string + +def string_min_test() -> int: pass diff --git a/.transpyc_cache/75583f20a922ab2e.doc.json b/.transpyc_cache/75583f20a922ab2e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/75583f20a922ab2e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/75583f20a922ab2e.pyi b/.transpyc_cache/75583f20a922ab2e.pyi new file mode 100644 index 0000000..10da71f --- /dev/null +++ b/.transpyc_cache/75583f20a922ab2e.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from zc.test.py +Module: zc.test +""" + + +import config +import t, c + +def AboutSizeTWhileTest(length: t.CSizeT) -> t.CInt: pass + +def MathBasicTest(x: t.CInt, y: t.CInt) -> t.CInt: pass + +def MathAdvancedTest(value: t.CInt) -> t.CInt: pass + +def WhileWithElifElseTest(start: t.CInt, end: t.CInt) -> t.CInt: pass + +def NestedWhileWithMacrosTest(outer_limit: t.CInt, inner_limit: t.CInt) -> t.CInt: pass + +def ComplexConditionTest(a: t.CInt, b: t.CInt, c_val: t.CInt) -> t.CInt: pass + +def RetryLoopTest(retry_count: t.CInt) -> t.CInt: pass + +def BitwiseOperationsTest(value: t.CInt) -> t.CInt: pass + +def CompoundAssignmentMathTest(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def PowerAndModuloTest(base: t.CInt, exponent: t.CInt) -> t.CInt: pass + +def MixedArithmeticTest(x: t.CInt, y: t.CInt, z: t.CInt) -> t.CInt: pass + +def MathWithWhileLoopTest(start: t.CInt, count: t.CInt) -> t.CInt: pass + +def ComplexNestedMathTest(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def ConditionalMathChainTest(value: t.CInt) -> t.CInt: pass + +def BitManipulationMathTest(value: t.CInt) -> t.CInt: pass diff --git a/.transpyc_cache/75dc9247529e6e0e.doc.json b/.transpyc_cache/75dc9247529e6e0e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/75dc9247529e6e0e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/75dc9247529e6e0e.pyi b/.transpyc_cache/75dc9247529e6e0e.pyi new file mode 100644 index 0000000..7f94c53 --- /dev/null +++ b/.transpyc_cache/75dc9247529e6e0e.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from deco_test.py +Module: deco_test +""" + + +import stdio +import t, c +import string + +MAX_VALUE: t.CDefine = 256 +PI_APPROX: t.CDefine = 314 + +def ExportedAdd(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: pass + +def InlineSquare(x: t.CInt) -> t.CInt | t.CInline: pass + +def NormalSub(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def ExportedInlineCube(x: t.CInt) -> t.CInt | t.CExport | t.CInline: pass + +def ExportedFindChar(s: str, ch: t.CInt) -> str | t.CExport: pass + +def test_cdefine() -> t.CInt: pass + +def test_cexport() -> t.CInt: pass + +def test_cinline() -> t.CInt: pass + +def test_deco_normal() -> t.CInt: pass + +def test_export_inline() -> t.CInt: pass + +def test_export_ptr() -> t.CInt: pass + +def ExternalDecl(x: t.CInt) -> t.CInt | t.CExtern | t.CExport: pass + +def StateDecl(x: t.CInt) -> t.CInt | t.State: pass + +def StateVoidDecl() -> t.State: pass + +def test_extern_decl() -> t.CInt: pass + +def test_state_decl() -> t.CInt: pass + +def test_state_void_decl() -> t.CInt: pass + +def deco_test() -> int: pass diff --git a/.transpyc_cache/766d8e14db1ff917.doc.json b/.transpyc_cache/766d8e14db1ff917.doc.json new file mode 100644 index 0000000..f67a6e1 --- /dev/null +++ b/.transpyc_cache/766d8e14db1ff917.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/766d8e14db1ff917.pyi b/.transpyc_cache/766d8e14db1ff917.pyi new file mode 100644 index 0000000..477ab88 --- /dev/null +++ b/.transpyc_cache/766d8e14db1ff917.pyi @@ -0,0 +1,169 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/76dd6c275aa72b3b.pyi b/.transpyc_cache/76dd6c275aa72b3b.pyi new file mode 100644 index 0000000..cd52873 --- /dev/null +++ b/.transpyc_cache/76dd6c275aa72b3b.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from os.__init__.py +Module: os.__init__ +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import os.path +import os._win32 +import os._posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd() -> str: pass + +def chdir(path: str) -> int: pass + +def mkdir(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir(path: str) -> int: pass + +def rename(old_path: str, new_path: str) -> int: pass + +def chmod(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system(cmd: str) -> int: pass diff --git a/.transpyc_cache/78418cf2ee722f0f.doc.json b/.transpyc_cache/78418cf2ee722f0f.doc.json new file mode 100644 index 0000000..04e9521 --- /dev/null +++ b/.transpyc_cache/78418cf2ee722f0f.doc.json @@ -0,0 +1 @@ +{"test_cdefine_values": "测试 CDefine 常量值是否正确", "test_bitops_width": "测试位运算是否保持正确类型宽度"} \ No newline at end of file diff --git a/.transpyc_cache/78418cf2ee722f0f.pyi b/.transpyc_cache/78418cf2ee722f0f.pyi new file mode 100644 index 0000000..0e4d365 --- /dev/null +++ b/.transpyc_cache/78418cf2ee722f0f.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +def test_zext_u8_u32() -> t.CInt: pass + +def test_sext_i8_i32() -> t.CInt: pass + +def test_zext_u32_u64() -> t.CInt: pass + +def test_sext_i32_i64() -> t.CInt: pass + +def test_trunc_u64_u32() -> t.CInt: pass + +def test_mixed_signed_unsigned() -> t.CInt: pass + + +VAL_32BIT: t.CDefine = 0x80000000 +VAL_NEG: t.CDefine = -1 +VAL_BIG: t.CDefine = 0xFFFFFFFF + +def test_cdefine_values() -> t.CInt: pass + +def test_bitops_width() -> t.CInt: pass + +def test_unsigned_divmod() -> t.CInt: pass + +def test_signed_divmod_neg() -> t.CInt: pass + +def test_comparison_types() -> t.CInt: pass + +def test_ptr_int_cast() -> t.CInt: pass + +def test_u64_mul() -> t.CInt: pass + +def test_bool_ops() -> t.CInt: pass + +def test_ternary() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/784dde8a8697b7ab.doc.json b/.transpyc_cache/784dde8a8697b7ab.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/784dde8a8697b7ab.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/784dde8a8697b7ab.pyi b/.transpyc_cache/784dde8a8697b7ab.pyi new file mode 100644 index 0000000..193ea96 --- /dev/null +++ b/.transpyc_cache/784dde8a8697b7ab.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from __zlib.zdeflate.py +Module: __zlib.zdeflate +""" + + +from stdint import * +import zchecksum +import zhuff +import zdef +import string +import t, c + +@t.Object +class zdeflate_stream: + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + def find_match(self: zdeflate_stream, data: BYTE | t.CPtr, data_len: t.CSizeT, pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: pass + def update_hash(self: zdeflate_stream, data: BYTE | t.CPtr, pos: t.CSizeT) -> t.CInt: pass + def write_fixed_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def encode_block_data(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + def write_dynamic_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_stored_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def flush(self: zdeflate_stream, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zdeflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zdeflate_stream) -> zdeflate_stream | t.CPtr: pass + def destroy(self: zdeflate_stream) -> t.CInt: pass + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: pass + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: pass + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: pass + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR) -> t.CInt: pass + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + +def zdeflate_create(level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: pass + +def zdeflate_one_shot(data: UINT8PTR, length: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass diff --git a/.transpyc_cache/78943d0dbc89a29a.doc.json b/.transpyc_cache/78943d0dbc89a29a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/78943d0dbc89a29a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/78943d0dbc89a29a.pyi b/.transpyc_cache/78943d0dbc89a29a.pyi new file mode 100644 index 0000000..ce1d065 --- /dev/null +++ b/.transpyc_cache/78943d0dbc89a29a.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/78bbe886bbe5ca8f.doc.json b/.transpyc_cache/78bbe886bbe5ca8f.doc.json new file mode 100644 index 0000000..ffc63a6 --- /dev/null +++ b/.transpyc_cache/78bbe886bbe5ca8f.doc.json @@ -0,0 +1 @@ +{"add_imported_module": "记录已导入的模块名，返回新的 imported_modules 字符串指针", "is_module_imported": "检查模块是否已导入（词边界精确匹配，避免子串误匹配）", "add_from_import": "添加 from-import 映射 \"name:module\"，返回新的 from_imports 字符串", "lookup_from_import": "查找 from-import 名称，返回模块名或 None\n\n 支持 star import: 如果 from_imports 中有 \"*:module\" 条目，\n 且未找到精确名称匹配，且 allow_star_fallback != 0，则返回 star import 的模块名。\n ", "_resolve_relative_module": "解析相对导入为完整模块名", "compute_package_from_relpath": "从相对路径计算包名（目录部分，分隔符替换为 .）", "ImportsHandle": "导入语句处理器：继承 Mixin 获得 Trans 回指针", "ImportsHandle.HandleImport": "处理 import 语句，返回 0", "ImportsHandle.HandleImportFromModule": "处理 from X import Y 的模块部分，更新 trans._imported_modules", "ImportsHandle.HandleImportFromNames": "处理 from X import Y 的名称部分，更新 trans._from_imports"} \ No newline at end of file diff --git a/.transpyc_cache/78bbe886bbe5ca8f.pyi b/.transpyc_cache/78bbe886bbe5ca8f.pyi new file mode 100644 index 0000000..868248b --- /dev/null +++ b/.transpyc_cache/78bbe886bbe5ca8f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesImports.py +Module: lib.core.Handles.HandlesImports +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, imported_modules: str, name: str) -> str: pass + +def is_module_imported(imported_modules: str, name: str) -> int: pass + +def add_from_import(pool: memhub.MemBuddy | t.CPtr, from_imports: str, local_name: str, module_name: str) -> str: pass + +def lookup_from_import(from_imports: str, name: str, allow_star_fallback: int) -> str: pass + +def _resolve_relative_module(pool: memhub.MemBuddy | t.CPtr, current_package: str, level: t.CInt, module: str) -> str: pass + +def compute_package_from_relpath(pool: memhub.MemBuddy | t.CPtr, rel_path: str) -> str: pass + + +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + def __init__(self: ImportsHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleImport(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromModule(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromNames(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: pass diff --git a/.transpyc_cache/79b337e5ea8951e2.pyi b/.transpyc_cache/79b337e5ea8951e2.pyi new file mode 100644 index 0000000..aa71e73 --- /dev/null +++ b/.transpyc_cache/79b337e5ea8951e2.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from _fakeduck.py +Module: _fakeduck +""" + +import c + + +import t +import memhub +import string + +class _Type: + def type_id(self: _Type) -> int: pass +class _str: + __requires__: list[str] = ['__mbuddy__'] + __data__: str + __mbuddy__: memhub.MemBuddy | t.CPtr + def upper(self: _str) -> str: pass + def lower(self: _str) -> str: pass + def capitalize(self: _str) -> str: pass + def swapcase(self: _str) -> str: pass + def title(self: _str) -> str: pass + def casefold(self: _str) -> str: pass + def isupper(self: _str) -> bool: pass + def islower(self: _str) -> bool: pass + def isalpha(self: _str) -> bool: pass + def isdigit(self: _str) -> bool: pass + def isalnum(self: _str) -> bool: pass + def isspace(self: _str) -> bool: pass + def isascii(self: _str) -> bool: pass + def istitle(self: _str) -> bool: pass + def split(self: _str, delimiter: str) -> list[str]: pass \ No newline at end of file diff --git a/.transpyc_cache/79ce76014465c04f.doc.json b/.transpyc_cache/79ce76014465c04f.doc.json new file mode 100644 index 0000000..6723531 --- /dev/null +++ b/.transpyc_cache/79ce76014465c04f.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值"} \ No newline at end of file diff --git a/.transpyc_cache/79ce76014465c04f.pyi b/.transpyc_cache/79ce76014465c04f.pyi new file mode 100644 index 0000000..b854517 --- /dev/null +++ b/.transpyc_cache/79ce76014465c04f.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass diff --git a/.transpyc_cache/79ef7477acd14bd7.doc.json b/.transpyc_cache/79ef7477acd14bd7.doc.json new file mode 100644 index 0000000..7996d5c --- /dev/null +++ b/.transpyc_cache/79ef7477acd14bd7.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/79ef7477acd14bd7.pyi b/.transpyc_cache/79ef7477acd14bd7.pyi new file mode 100644 index 0000000..9aa684e --- /dev/null +++ b/.transpyc_cache/79ef7477acd14bd7.pyi @@ -0,0 +1,69 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/7b6b920fefb08181.doc.json b/.transpyc_cache/7b6b920fefb08181.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/.transpyc_cache/7b6b920fefb08181.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/.transpyc_cache/7b6b920fefb08181.pyi b/.transpyc_cache/7b6b920fefb08181.pyi new file mode 100644 index 0000000..a5ea74e --- /dev/null +++ b/.transpyc_cache/7b6b920fefb08181.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/7bb99cc1646d2ae2.doc.json b/.transpyc_cache/7bb99cc1646d2ae2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/7bb99cc1646d2ae2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/7bb99cc1646d2ae2.pyi b/.transpyc_cache/7bb99cc1646d2ae2.pyi new file mode 100644 index 0000000..a0accc6 --- /dev/null +++ b/.transpyc_cache/7bb99cc1646d2ae2.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from test_main.py +Module: test_main +""" + + +import stdio +import t, c +import testcheck +from stdint import * +from asm_test import asm_test +from attr_test import attr_test +from augassign_test import augassign_test +from closure_test import closure_test +from deco_test import deco_test +from deref_min_test import deref_min_test +from deref_test import deref_test +from eq_test import eq_test +from float_test import float_test +from flow_test import flow_test +from for_test import for_test +from func_test import func_test +from llvmir_test import llvmir_test +from oop_test import oop_test +from ptr_only_test import ptr_only_test +from ptr_test import ptr_test +from simple_test import simple_test +from string_min_test import string_min_test +from string_test import string_test +from struct_test import struct_test +from type_bit_test import type_bit_test +from vtable_test import vtable_test +from inherit_test import inherit_test +from func_vtable_test import func_vtable_test +from virtual_dispatch_test import virtual_dispatch_test +from new_test import new_test +from namespace_test import namespace_test +from testcheck_test import testcheck_test +from opovl_test import opovl_test +from generic_test import generic_test + +def check_heap(name: str) -> int: pass + +def main() -> int: pass diff --git a/.transpyc_cache/7c5d5ec59c5fb0a8.doc.json b/.transpyc_cache/7c5d5ec59c5fb0a8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/7c5d5ec59c5fb0a8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/7c5d5ec59c5fb0a8.pyi b/.transpyc_cache/7c5d5ec59c5fb0a8.pyi new file mode 100644 index 0000000..3c94870 --- /dev/null +++ b/.transpyc_cache/7c5d5ec59c5fb0a8.pyi @@ -0,0 +1,765 @@ +""" +Auto-generated Python stub file from cpython.py +Module: cpython +""" + + +import t, c + +class PyObject: + ob_refcnt: t.CLong + ob_type: t.CPtr + +def Py_INCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_DECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_XINCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_XDECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_DecRef(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_Initialize() -> t.CVoid | t.State: pass + +def Py_Finalize() -> t.CVoid | t.State: pass + +def Py_InitializeEx(initsigs: t.CInt) -> t.CVoid | t.State: pass + +def Py_FinalizeEx() -> t.CInt | t.State: pass + +def Py_IsInitialized() -> t.CInt | t.State: pass + + +class PySTATUS: + func: t.CConst | str + err_msg: t.CConst | str + exitcode: t.CInt +class PyPreConfig: + _config_version: t.CInt + _config_init: t.CInt + parse_argv: t.CInt + isolated: t.CInt + use_environment: t.CInt + utf8_mode: t.CInt + coerce_c_locale: t.CInt + C_locale_warnings: t.CInt + allocator: t.CInt + dev_mode: t.CInt +class PyWideStringList: + length: t.CSizeT + items: t.CPtr +class PyConfig: + _config_version: t.CInt + _config_init: t.CInt + isolated: t.CInt + use_environment: t.CInt + dev_mode: t.CInt + install_signal_handlers: t.CInt + use_hash_seed: t.CInt + hash_seed: t.CUnsignedLong + faulthandler: t.CInt + tracemalloc: t.CInt + import_time: t.CInt + show_ref_count: t.CInt + dump_refs: t.CInt + dump_refs_file: t.CConst | str + malloc_stats: t.CInt + filesystem_errors: t.CInt + pycache_prefix: t.CConst | str + parse_argv: t.CInt + argv: PyWideStringList + program_name: t.CConst | str + xoptions: PyWideStringList + warnoptions: PyWideStringList + site_import: t.CInt + bytes_warning: t.CInt + warn_default_encoding: t.CInt + inspect: t.CInt + interactive: t.CInt + optimization_level: t.CInt + parser_debug: t.CInt + verbose: t.CInt + quiet: t.CInt + home: t.CConst | str + pythonpath_env: t.CConst | str + module_search_paths: PyWideStringList + module_search_paths_set: t.CInt + executable: t.CConst | str + base_executable: t.CConst | str + prefix: t.CConst | str + base_prefix: t.CConst | str + exec_prefix: t.CConst | str + base_exec_prefix: t.CConst | str + platlibdir: t.CConst | str + stdio_filename: t.CConst | str + +def PyConfig_InitPythonConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_InitIsolatedConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_Clear(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_Read(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass + +def Py_InitializeFromConfig(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass + +def Py_DecodeLocale(s: t.CConst | str) -> str | t.State: pass + +def PyStatus_Exception(status: PySTATUS) -> t.CInt | t.State: pass + +def PyLong_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromUnsignedLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromLongLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromUnsignedLongLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromSsize_t(v: t.CPtrDiffT) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromSize_t(v: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass + +def PyLong_AsLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyLong_AsUnsignedLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass + +def PyLong_AsLongLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyLong_AsUnsignedLongLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass + +def PyLong_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass + +def PyLong_AsSsize_t(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass + +def PyLong_AsSize_t(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyFloat_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass + +def PyFloat_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass + +def PyFloat_FromString(s: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromStringAndSize(u: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_AsUTF8(obj: PyObject | t.CPtr) -> str | t.State: pass + +def PyUnicode_AsUTF8AndSize(obj: PyObject | t.CPtr, size: t.CPtrDiffT | t.CPtr) -> str | t.State: pass + +def PyUnicode_AsWideChar(obj: PyObject | t.CPtr, wbuf: t.CPtr, bufsize: t.CSizeT) -> t.CSizeT | t.State: pass + +def PyUnicode_GetLength(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass + +def PyUnicode_ReadChar(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CUInt32T | t.State: pass + +def PyUnicode_WriteChar(obj: PyObject | t.CPtr, idx: t.CSizeT, ch: t.CUInt32T) -> t.CInt | t.State: pass + +def PyUnicode_Compare(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Concat(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_Contains(container: PyObject | t.CPtr, element: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Format(fmt: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_InternInPlace(s: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyUnicode_InternFromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromString(v: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromStringAndSize(v: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_AsString(obj: PyObject | t.CPtr) -> str | t.State: pass + +def PyBytes_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyBytes_FromObject(o: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBool_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + + +Py_True: t.CExtern | PyObject | t.CPtr | t.State +Py_False: t.CExtern | PyObject | t.CPtr | t.State +Py_None: t.CExtern | PyObject | t.CPtr | t.State + +def PyList_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_Size(lst: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyList_GetItem(lst: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_SetItem(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Insert(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Append(lst: PyObject | t.CPtr, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_GetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_SetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT, items: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Sort(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Reverse(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_AsTuple(lst: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_Size(tup: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyTuple_GetItem(tup: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_SetItem(tup: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyTuple_GetSlice(tup: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_Pack(n: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyDict_New() -> PyObject | t.CPtr | t.State: pass + +def PyDict_GetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_GetItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyDict_SetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_SetItemString(d: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_DelItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_DelItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass + +def PyDict_Size(d: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyDict_Keys(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Values(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Items(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Copy(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Clear(d: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyDict_Contains(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Next(d: PyObject | t.CPtr, pos: t.CPtrDiffT | t.CPtr, key: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Merge(d: PyObject | t.CPtr, other: PyObject | t.CPtr, override: t.CInt) -> t.CInt | t.State: pass + +def PyDict_Update(d: PyObject | t.CPtr, other: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Str(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_Repr(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_Type(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_IsInstance(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_IsSubclass(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_HasAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_HasAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyObject_GetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_GetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyObject_SetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_SetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyObject_Call(callable: PyObject | t.CPtr, args: PyObject | t.CPtr, kwargs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallNoArgs(callable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallOneArg(callable: PyObject | t.CPtr, arg: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_IsTrue(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Not(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_RichCompare(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyObject_RichCompareBool(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> t.CInt | t.State: pass + +def PyObject_Hash(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyObject_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyObject_GetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_SetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Dir(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_GetIter(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyIter_Next(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +Py_LT: t.CExtern | t.CInt | t.State +Py_LE: t.CExtern | t.CInt | t.State +Py_EQ: t.CExtern | t.CInt | t.State +Py_NE: t.CExtern | t.CInt | t.State +Py_GT: t.CExtern | t.CInt | t.State +Py_GE: t.CExtern | t.CInt | t.State + +def PyImport_ImportModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyImport_Import(name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyImport_ImportModuleLevelObject(name: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr, level: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyImport_ReloadModule(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyImport_AddModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyImport_GetModuleDict() -> PyObject | t.CPtr | t.State: pass + +def PyImport_ImportModuleEx(name: t.CConst | str, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +class PyMethodDef: + ml_name: t.CConst | str + ml_meth: t.CPtr + ml_flags: t.CInt + ml_doc: t.CConst | str +class PyModuleDef_Base: + ob_base: PyObject + m_init: t.CPtr + m_index: t.CSizeT + m_copy: PyObject | t.CPtr +class PyModuleDef: + m_base: PyModuleDef_Base + m_name: t.CConst | str + m_doc: t.CConst | str + m_size: t.CPtrDiffT + m_methods: PyMethodDef | t.CPtr + m_slots: t.CPtr + m_traverse: t.CPtr + m_clear: t.CPtr + m_free: t.CPtr + +def PyModule_Create(mod: PyModuleDef | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_Create2(mod: PyModuleDef | t.CPtr, ver: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyModule_GetName(mod: PyObject | t.CPtr) -> str | t.State: pass + +def PyModule_GetDict(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_GetFilenameObject(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_AddObject(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyModule_AddObjectRef(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyModule_AddIntConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CLong) -> t.CInt | t.State: pass + +def PyModule_AddStringConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CConst | str) -> t.CInt | t.State: pass + +def PyGILState_Ensure() -> t.CInt | t.State: pass + +def PyGILState_Release(state: t.CInt) -> t.CVoid | t.State: pass + +def PyEval_SaveThread() -> t.CPtr | t.State: pass + +def PyEval_RestoreThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_AcquireThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_ReleaseThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_AcquireLock() -> t.CVoid | t.State: pass + +def PyEval_ReleaseLock() -> t.CVoid | t.State: pass + +def PyEval_ThreadsInitialized() -> t.CInt | t.State: pass + +def PyEval_InitThreads() -> t.CVoid | t.State: pass + +def PyErr_Occurred() -> PyObject | t.CPtr | t.State: pass + +def PyErr_Clear() -> t.CVoid | t.State: pass + +def PyErr_Fetch(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_Restore(tp: PyObject | t.CPtr, val: PyObject | t.CPtr, tb: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_NormalizeException(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_SetString(tp: PyObject | t.CPtr, msg: t.CConst | str) -> t.CVoid | t.State: pass + +def PyErr_SetObject(tp: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_BadArgument() -> t.CPtr | t.State: pass + +def PyErr_NoMemory() -> t.CPtr | t.State: pass + +def PyErr_Print() -> t.CVoid | t.State: pass + +def PyErr_ExceptionMatches(exc: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyErr_GivenExceptionMatches(given: PyObject | t.CPtr, exc: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +PyExc_BaseException: t.CExtern | PyObject | t.CPtr | t.State +PyExc_Exception: t.CExtern | PyObject | t.CPtr | t.State +PyExc_TypeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ValueError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_RuntimeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_KeyError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IndexError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_AttributeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ImportError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_OSError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IOError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ZeroDivisionError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_OverflowError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_MemoryError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NameError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_StopIteration: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NotImplementedError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_SyntaxError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_FileNotFoundError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_PermissionError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IsADirectoryError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NotADirectoryError: t.CExtern | PyObject | t.CPtr | t.State + +def Py_BuildValue(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def Py_VaBuildValue(fmt: t.CConst | str, va: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyLong_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyFloat_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyBytes_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyTuple_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyBool_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyNone_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMem_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyMem_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyMem_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass + +def PyMem_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass + +def PyObject_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Init(obj: PyObject | t.CPtr, tp: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_InitVar(obj: PyObject | t.CPtr, tp: t.CPtr, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyFile_FromString(filename: t.CConst | str, mode: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyFile_AsFile(file: PyObject | t.CPtr) -> t.CPtr | t.State: pass + +def PyFile_WriteObject(obj: PyObject | t.CPtr, file: PyObject | t.CPtr, flags: t.CInt) -> t.CInt | t.State: pass + +def PyFile_WriteString(s: t.CConst | str, file: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def Py_CompileString(s: t.CConst | str, filename: t.CConst | str, start: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalCode(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalCodeEx(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, args: PyObject | t.CPtr | t.CPtr, argc: t.CInt, kws: PyObject | t.CPtr | t.CPtr, kwc: t.CInt, defs: PyObject | t.CPtr | t.CPtr, defc: t.CInt, kwdefs: PyObject | t.CPtr, closure: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalFrame(frame: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallFunction(callable: PyObject | t.CPtr, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallMethod(obj: PyObject | t.CPtr, method: t.CConst | str, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyRun_SimpleString(cmd: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_SimpleFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_AnyFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_String(s: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyRun_File(fp: t.CPtr, filename: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +Py_single_input: t.CExtern | t.CInt | t.State +Py_file_input: t.CExtern | t.CInt | t.State +Py_eval_input: t.CExtern | t.CInt | t.State + +def PyCapsule_New(ptr: t.CPtr, name: t.CConst | str, dtor: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyCapsule_GetPointer(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CPtr | t.State: pass + +def PyCapsule_GetName(cap: PyObject | t.CPtr) -> str | t.State: pass + +def PyCapsule_IsValid(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyCapsule_SetPointer(cap: PyObject | t.CPtr, ptr: t.CPtr) -> t.CInt | t.State: pass + +def PyCapsule_SetName(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyCapsule_SetDestructor(cap: PyObject | t.CPtr, dtor: t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Concat(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySequence_Repeat(obj: PyObject | t.CPtr, count: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_GetItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_SetItem(obj: PyObject | t.CPtr, idx: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_DelItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CInt | t.State: pass + +def PySequence_GetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_SetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_DelSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> t.CInt | t.State: pass + +def PySequence_Contains(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Index(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Count(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_List(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySequence_Tuple(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyMapping_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyMapping_HasKey(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_HasKeyString(obj: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass + +def PyMapping_GetItemString(obj: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_SetItemString(obj: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_Keys(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Values(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Items(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Add(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Subtract(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Multiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_MatrixMultiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_FloorDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_TrueDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Modulo(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Power(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, o3: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Negative(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Positive(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Absolute(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Invert(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Lshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Rshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_And(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Xor(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Or(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Index(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Long(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Float(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyWeakref_NewRef(obj: PyObject | t.CPtr, cb: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyWeakref_GetObject(ref: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyWeakref_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +class PyGetSetDef: + name: t.CConst | str + get: t.CPtr + set: t.CPtr + doc: t.CConst | str + closure: t.CPtr +class PyMemberDef: + name: t.CConst | str + type: t.CInt + offset: t.CPtrDiffT + flags: t.CInt + doc: t.CConst | str + +def PySys_GetObject(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PySys_SetObject(name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySys_ResetWarnOptions() -> t.CVoid | t.State: pass + +def PySys_AddWarnOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySYS_AddXOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySys_WriteStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_WriteStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_FormatStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_FormatStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PyMemoryView_FromObject(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMemoryView_GetContiguous(obj: PyObject | t.CPtr, btype: t.CInt, order: t.CChar) -> PyObject | t.CPtr | t.State: pass + +def PyMemoryView_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySet_Add(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Discard(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Contains(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Size(s: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySet_Clear(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyFrozenSet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyFrozenSet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySlice_New(start: PyObject | t.CPtr, stop: PyObject | t.CPtr, step: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySlice_GetIndicesEx(sl: PyObject | t.CPtr, length: t.CSizeT, start: t.CSizeT | t.CPtr, stop: t.CSizeT | t.CPtr, step: t.CSizeT | t.CPtr, slicelen: t.CSizeT | t.CPtr) -> t.CInt | t.State: pass + +def PySlice_Unpack(sl: PyObject | t.CPtr, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT | t.CPtr) -> t.CVoid | t.State: pass + +def PySlice_AdjustIndices(length: t.CSizeT, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT) -> t.CSizeT | t.State: pass + +def PySlice_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +class PyLongObj: + ptr: PyObject | t.CPtr + def __new__(self: PyLongObj) -> 'PyLongObj' | t.CPtr: pass + @staticmethod + def from_long(val: t.CLong) -> 'PyLongObj' | t.CPtr: pass + def __add__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __sub__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __mul__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __neg__(self: PyLongObj) -> 'PyLongObj' | t.CPtr: pass + def to_long(self: PyLongObj) -> t.CLong: pass + def __str__(self: PyLongObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyLongObj) -> t.CInt: pass +class PyFloatObj: + ptr: PyObject | t.CPtr + def __new__(self: PyFloatObj) -> 'PyFloatObj' | t.CPtr: pass + @staticmethod + def from_double(val: t.CDouble) -> 'PyFloatObj' | t.CPtr: pass + def __add__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __sub__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __mul__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __truediv__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __neg__(self: PyFloatObj) -> 'PyFloatObj' | t.CPtr: pass + def to_double(self: PyFloatObj) -> t.CDouble: pass + def __str__(self: PyFloatObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyFloatObj) -> t.CInt: pass +class PyUnicodeObj: + ptr: PyObject | t.CPtr + def __new__(self: PyUnicodeObj) -> 'PyUnicodeObj' | t.CPtr: pass + @staticmethod + def from_string(s: t.CChar | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: pass + def __add__(self: PyUnicodeObj, other: PyUnicodeObj | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: pass + def __len__(self: PyUnicodeObj) -> t.CPtrDiffT: pass + def as_utf8(self: PyUnicodeObj) -> t.CChar | t.CPtr: pass + def __str__(self: PyUnicodeObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyUnicodeObj) -> t.CInt: pass +class PyListObj: + ptr: PyObject | t.CPtr + def __new__(self: PyListObj) -> 'PyListObj' | t.CPtr: pass + @staticmethod + def new() -> 'PyListObj' | t.CPtr: pass + def __getitem__(self: PyListObj, idx: t.CSizeT) -> PyObject | t.CPtr: pass + def __setitem__(self: PyListObj, idx: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt: pass + def __len__(self: PyListObj) -> t.CSizeT: pass + def append(self: PyListObj, item: PyObject | t.CPtr) -> t.CInt: pass + def dec_ref(self: PyListObj) -> t.CInt: pass +class PyDictObj: + ptr: PyObject | t.CPtr + def __new__(self: PyDictObj) -> 'PyDictObj' | t.CPtr: pass + @staticmethod + def new() -> 'PyDictObj' | t.CPtr: pass + def __getitem__(self: PyDictObj, key: PyObject | t.CPtr) -> PyObject | t.CPtr: pass + def __setitem__(self: PyDictObj, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt: pass + def __len__(self: PyDictObj) -> t.CSizeT: pass + def get_string(self: PyDictObj, key: t.CChar | t.CPtr) -> PyObject | t.CPtr: pass + def set_string(self: PyDictObj, key: t.CChar | t.CPtr, val: PyObject | t.CPtr) -> t.CInt: pass + def dec_ref(self: PyDictObj) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/7e260c14d2712119.doc.json b/.transpyc_cache/7e260c14d2712119.doc.json new file mode 100644 index 0000000..c90d338 --- /dev/null +++ b/.transpyc_cache/7e260c14d2712119.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "_ScanDirForPyFiles": "递归扫描目录，收集 .py 文件到 entries，返回新的 file_count", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/7e260c14d2712119.pyi b/.transpyc_cache/7e260c14d2712119.pyi new file mode 100644 index 0000000..2398e44 --- /dev/null +++ b/.transpyc_cache/7e260c14d2712119.pyi @@ -0,0 +1,42 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 64 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def _ScanDirForPyFiles(mb: memhub.MemBuddy | t.CPtr, dir_path: str, entries: SrcFileEntry | t.CPtr, entry_size: t.CSizeT, file_count: int, max_files: int) -> int: pass + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/7e529fe7a078cfef.pyi b/.transpyc_cache/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/.transpyc_cache/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/.transpyc_cache/7eb1251b89cc1b7d.doc.json b/.transpyc_cache/7eb1251b89cc1b7d.doc.json new file mode 100644 index 0000000..e7fc1ec --- /dev/null +++ b/.transpyc_cache/7eb1251b89cc1b7d.doc.json @@ -0,0 +1 @@ +{"TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/7eb1251b89cc1b7d.pyi b/.transpyc_cache/7eb1251b89cc1b7d.pyi new file mode 100644 index 0000000..3a8c9c7 --- /dev/null +++ b/.transpyc_cache/7eb1251b89cc1b7d.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/7fa5782c9657b2ba.doc.json b/.transpyc_cache/7fa5782c9657b2ba.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/7fa5782c9657b2ba.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/7fa5782c9657b2ba.pyi b/.transpyc_cache/7fa5782c9657b2ba.pyi new file mode 100644 index 0000000..9dc1839 --- /dev/null +++ b/.transpyc_cache/7fa5782c9657b2ba.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from generic_test.py +Module: generic_test +""" + + +import stdio +import stdlib +import t, c +import testcheck +import memhub +import _list + +def generic_test() -> int: pass diff --git a/.transpyc_cache/7fff0dc7524a3a93.doc.json b/.transpyc_cache/7fff0dc7524a3a93.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/7fff0dc7524a3a93.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/7fff0dc7524a3a93.pyi b/.transpyc_cache/7fff0dc7524a3a93.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/7fff0dc7524a3a93.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/80241c7d6f357bd1.doc.json b/.transpyc_cache/80241c7d6f357bd1.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/80241c7d6f357bd1.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/80241c7d6f357bd1.pyi b/.transpyc_cache/80241c7d6f357bd1.pyi new file mode 100644 index 0000000..25cba70 --- /dev/null +++ b/.transpyc_cache/80241c7d6f357bd1.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from asm_test.py +Module: asm_test +""" + + +import stdio +import t, c + +def asm_test() -> int: pass diff --git a/.transpyc_cache/8046d45456d0b153.doc.json b/.transpyc_cache/8046d45456d0b153.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/8046d45456d0b153.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/8046d45456d0b153.pyi b/.transpyc_cache/8046d45456d0b153.pyi new file mode 100644 index 0000000..a6ccaab --- /dev/null +++ b/.transpyc_cache/8046d45456d0b153.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import stdlib +import socket +import memhub +import requests + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +@t.CExport +def main() -> int: pass diff --git a/.transpyc_cache/814e6b31adfc6256.doc.json b/.transpyc_cache/814e6b31adfc6256.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/814e6b31adfc6256.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/814e6b31adfc6256.pyi b/.transpyc_cache/814e6b31adfc6256.pyi new file mode 100644 index 0000000..0a0bccf --- /dev/null +++ b/.transpyc_cache/814e6b31adfc6256.pyi @@ -0,0 +1,69 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdio import printf +import testcheck + +def log(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +def trace(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_timing_tmp: t.CExtern | t.CInt + +def timing(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +def repeat(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_skip_count: t.CExtern | t.CInt + +def skip(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_call_count: t.CExtern | t.CInt + +def count_calls(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +@log +def add(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: pass + +@log +@trace +def multiply(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: pass + +@log +def greet(name: str) -> t.CVoid | t.CExport: pass + +@log +def get_value() -> t.CInt | t.CExport: pass + +@timing +def compute(x: t.CInt) -> t.CInt | t.CExport: pass + +@repeat(3) +def echo(msg: str) -> t.CVoid | t.CExport: pass + +@timing +@log +def power(base: t.CInt, exp: t.CInt) -> t.CInt | t.CExport: pass + +@skip +def dangerous() -> t.CInt | t.CExport: pass + +@repeat(3) +def accumulate(x: t.CInt) -> t.CInt | t.CExport: pass + +@count_calls +def factorial(n: t.CInt) -> t.CInt | t.CExport: pass + +@log +def fib(n: t.CInt) -> t.CInt | t.CExport: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/81eb28ba912467a6.doc.json b/.transpyc_cache/81eb28ba912467a6.doc.json new file mode 100644 index 0000000..f6225ef --- /dev/null +++ b/.transpyc_cache/81eb28ba912467a6.doc.json @@ -0,0 +1 @@ +{"translate_stmt": "翻译单条语句，返回新增的变量数", "translate_global": "翻译 global 语句：记录 global 变量名", "translate_nonlocal": "翻译 nonlocal 语句：记录 nonlocal 变量名", "translate_expr_stmt": "翻译表达式语句（printf 调用等）", "pre_scan_allocas": "预扫描语句中的 AnnAssign，提前创建 alloca\n\n 返回新增的变量数\n ", "translate_break": "翻译 break 语句：跳转到循环 end 块", "translate_continue": "翻译 continue 语句：跳转到循环 cond/incr 块", "get_stmt_kind_name": "获取语句类型名"} \ No newline at end of file diff --git a/.transpyc_cache/81eb28ba912467a6.pyi b/.transpyc_cache/81eb28ba912467a6.pyi new file mode 100644 index 0000000..aa52e34 --- /dev/null +++ b/.transpyc_cache/81eb28ba912467a6.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBody.py +Module: lib.core.Handles.HandlesBody +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def translate_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_global(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nonlocal(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_expr_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def pre_scan_allocas(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_break(trans: HT.Translator | t.CPtr) -> int: pass + +def translate_continue(trans: HT.Translator | t.CPtr) -> int: pass + +def get_stmt_kind_name(node: ast.AST | t.CPtr) -> str: pass diff --git a/.transpyc_cache/83ba314b4637eec9.doc.json b/.transpyc_cache/83ba314b4637eec9.doc.json new file mode 100644 index 0000000..656d90d --- /dev/null +++ b/.transpyc_cache/83ba314b4637eec9.doc.json @@ -0,0 +1 @@ +{"TypeRegistry.Register": "注册一个 TypeInfo。ti.Name 字段必须已设置。\n\n Returns:\n 1 表示成功，0 表示失败（ti 为 None 或 Name 为 None）\n ", "TypeRegistry.Lookup": "按名称查找 TypeInfo。\n\n Returns:\n 找到返回 TypeInfo*，找不到返回 None\n ", "TypeRegistry.Has": "检查类型是否已注册。", "Mixin": "所有 Handle 的非多态基类：持有 Translator 回指针 + 共享委托方法", "Mixin.InitMixin": "初始化 Mixin 字段（子类 __init__ 中调用）"} \ No newline at end of file diff --git a/.transpyc_cache/83ba314b4637eec9.pyi b/.transpyc_cache/83ba314b4637eec9.pyi new file mode 100644 index 0000000..1d05ab4 --- /dev/null +++ b/.transpyc_cache/83ba314b4637eec9.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBase.py +Module: lib.core.Handles.HandlesBase +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import hashtable +import lib.core.Handles.HandlesTranslator as HT + +class TypeKind(t.CEnum): + Basic: t.State + Pointer: t.State + Struct: t.State + Union: t.State + Enum: t.State + Typedef: t.State + Function: t.State + Array: t.State + Void: t.State +@t.NoVTable +class TypeInfo: + Kind: int + Name: str + Size: int + Align: int + IsSigned: int + PtrCount: int + IsConst: int + IsVolatile: int +class TypeRegistry: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemManager | t.CPtr + def Register(self: TypeRegistry, ti: TypeInfo | t.CPtr) -> int: pass + def Lookup(self: TypeRegistry, name: str) -> TypeInfo | t.CPtr: pass + def Has(self: TypeRegistry, name: str) -> int: pass + +def NewTypeInfo(pool: memhub.MemManager | t.CPtr) -> TypeInfo | t.CPtr: pass + +def NewTypeRegistry(pool: memhub.MemManager | t.CPtr) -> TypeRegistry | t.CPtr: pass + +def TypeToLLVM(buf: t.CChar | t.CPtr, buf_size: t.CSizeT, ti: TypeInfo | t.CPtr) -> int: pass + + +@t.NoVTable +class Mixin: + Trans: HT.Translator | t.CPtr + def InitMixin(self: Mixin, trans: HT.Translator | t.CPtr) -> int: pass + +_mbuddy: t.CExtern | t.CVoid | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/862591435f13ef30.doc.json b/.transpyc_cache/862591435f13ef30.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/862591435f13ef30.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/862591435f13ef30.pyi b/.transpyc_cache/862591435f13ef30.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/862591435f13ef30.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/8639fbe371566e75.doc.json b/.transpyc_cache/8639fbe371566e75.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/.transpyc_cache/8639fbe371566e75.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/.transpyc_cache/8639fbe371566e75.pyi b/.transpyc_cache/8639fbe371566e75.pyi new file mode 100644 index 0000000..1e1b4bd --- /dev/null +++ b/.transpyc_cache/8639fbe371566e75.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/8642175afa282a4f.doc.json b/.transpyc_cache/8642175afa282a4f.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/8642175afa282a4f.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/8642175afa282a4f.pyi b/.transpyc_cache/8642175afa282a4f.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/8642175afa282a4f.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/86e203c56ae4de20.doc.json b/.transpyc_cache/86e203c56ae4de20.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/86e203c56ae4de20.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/86e203c56ae4de20.pyi b/.transpyc_cache/86e203c56ae4de20.pyi new file mode 100644 index 0000000..ba09b5a --- /dev/null +++ b/.transpyc_cache/86e203c56ae4de20.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def test_for_loop() -> t.CInt: pass + +def test_while_loop() -> t.CInt: pass + +def test_nested_loop() -> t.CInt: pass + +def test_break() -> t.CInt: pass + +def test_continue() -> t.CInt: pass + +def test_range_step() -> t.CInt: pass + +def test_loop_u64() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/888ea5bd6ab516ae.doc.json b/.transpyc_cache/888ea5bd6ab516ae.doc.json new file mode 100644 index 0000000..c189e56 --- /dev/null +++ b/.transpyc_cache/888ea5bd6ab516ae.doc.json @@ -0,0 +1 @@ +{"compute_sha1": "计算字符串的 SHA1，返回前 16 个十六进制字符（用于文件命名）", "CleanDir": "删除目录中所有文件（非递归），返回删除的文件数，-1 表示错误"} \ No newline at end of file diff --git a/.transpyc_cache/888ea5bd6ab516ae.pyi b/.transpyc_cache/888ea5bd6ab516ae.pyi new file mode 100644 index 0000000..e4ed12e --- /dev/null +++ b/.transpyc_cache/888ea5bd6ab516ae.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from lib.Projectrans.Utils.py +Module: lib.Projectrans.Utils +""" + + +import t, c +from stdint import * +import memhub +import string +import stdlib +import viperlib +import hashlib +import w32.win32file +import w32.win32base + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr + +def compute_sha1(pool: memhub.MemBuddy | t.CPtr, content: str) -> str: pass + +def CleanDir(dir_path: str) -> int: pass diff --git a/.transpyc_cache/895376eafe64af68.doc.json b/.transpyc_cache/895376eafe64af68.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/895376eafe64af68.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/895376eafe64af68.pyi b/.transpyc_cache/895376eafe64af68.pyi new file mode 100644 index 0000000..e2b929c --- /dev/null +++ b/.transpyc_cache/895376eafe64af68.pyi @@ -0,0 +1,12 @@ +""" +Auto-generated Python stub file from simple_test.py +Module: simple_test +""" + +import t +import c + + +import stdio + +def simple_test() -> int: pass diff --git a/.transpyc_cache/8959998ad305c7b1.doc.json b/.transpyc_cache/8959998ad305c7b1.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/8959998ad305c7b1.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/8959998ad305c7b1.pyi b/.transpyc_cache/8959998ad305c7b1.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/8959998ad305c7b1.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/8a97754e9e3c2b9a.doc.json b/.transpyc_cache/8a97754e9e3c2b9a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/8a97754e9e3c2b9a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/8a97754e9e3c2b9a.pyi b/.transpyc_cache/8a97754e9e3c2b9a.pyi new file mode 100644 index 0000000..c3e417c --- /dev/null +++ b/.transpyc_cache/8a97754e9e3c2b9a.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +from stdio import printf +import memhub +import json +import string +import w32.win32memory as win32mem +import testcheck + +def test_construct(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_structures(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_parse(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_write(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_nested(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_mpool_reuse() -> t.CInt: pass + +def test_escape(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/8e0d8fdba991b3b4.doc.json b/.transpyc_cache/8e0d8fdba991b3b4.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/8e0d8fdba991b3b4.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/8e0d8fdba991b3b4.pyi b/.transpyc_cache/8e0d8fdba991b3b4.pyi new file mode 100644 index 0000000..1aebea2 --- /dev/null +++ b/.transpyc_cache/8e0d8fdba991b3b4.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from config.py +Module: config +""" + + +import stdint +import t, c + +A: t.CExtern | t.CChar | t.CPtr + +class BIT_TEST(t.Object): + a: t.CInt | t.Bit(1) + b: t.CInt | t.Bit(1) + c: t.CInt | t.Bit(1) + d: t.CInt | t.Bit(5) +class OOP_TEST(t.Object): + a: t.CInt + def __init__(self: OOP_TEST) -> t.CInt: pass + def __call__(self: OOP_TEST) -> stdint.UINT: pass +class ENUM_TEST(t.CEnum): + A: t.State = 0 + B: t.State + C: t.State + Len: t.State \ No newline at end of file diff --git a/.transpyc_cache/8ebed83a7817fa0c.pyi b/.transpyc_cache/8ebed83a7817fa0c.pyi new file mode 100644 index 0000000..6375e88 --- /dev/null +++ b/.transpyc_cache/8ebed83a7817fa0c.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from zlib.zchecksum.py +Module: zlib.zchecksum +""" + + +from stdint import * +import t, c + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: pass + + +crc32_table: t.CExtern | t.CArray[t.CUInt32T, 256] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: pass diff --git a/.transpyc_cache/8f12094c840be06a.doc.json b/.transpyc_cache/8f12094c840be06a.doc.json new file mode 100644 index 0000000..0487965 --- /dev/null +++ b/.transpyc_cache/8f12094c840be06a.doc.json @@ -0,0 +1 @@ +{"InitLib": "初始化 lib 包的全局 _mbuddy 指针，并级联注入到所有子模块。\n\n Args:\n mb: memhub.MemManager 实例指针（MemBuddy 等子类通过多态传入）\n\n Returns:\n 0 表示成功，非 0 表示失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/8f12094c840be06a.pyi b/.transpyc_cache/8f12094c840be06a.pyi new file mode 100644 index 0000000..ac07294 --- /dev/null +++ b/.transpyc_cache/8f12094c840be06a.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from lib.__init__.py +Module: lib.__init__ +""" + + +import t, c +from stdint import * +import memhub +import lib.core.VLogger as VLogger + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr + +def InitLib(mb: memhub.MemManager | t.CPtr) -> int: pass diff --git a/.transpyc_cache/90921d009fdd674c.pyi b/.transpyc_cache/90921d009fdd674c.pyi new file mode 100644 index 0000000..86d9bcf --- /dev/null +++ b/.transpyc_cache/90921d009fdd674c.pyi @@ -0,0 +1,56 @@ +""" +Auto-generated Python stub file from zlib.zdeflate.py +Module: zlib.zdeflate +""" + + +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zhuff as zhuff +import zlib.zdef as zdef +import string +import memhub +import t, c + +@t.Object +class zdeflate_stream: + pool: memhub.MemManager | t.CPtr + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + def find_match(self: zdeflate_stream, data: BYTE | t.CPtr, data_len: t.CSizeT, pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: pass + def update_hash(self: zdeflate_stream, data: BYTE | t.CPtr, pos: t.CSizeT) -> t.CInt: pass + def write_fixed_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def encode_block_data(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + def write_dynamic_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_stored_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def flush(self: zdeflate_stream, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zdeflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zdeflate_stream) -> zdeflate_stream | t.CPtr: pass + def destroy(self: zdeflate_stream) -> t.CInt: pass + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: pass + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: pass + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: pass + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR) -> t.CInt: pass + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + +def zdeflate_create(pool: memhub.MemManager | t.CPtr, level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: pass + +def zdeflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass diff --git a/.transpyc_cache/90c53dd6db8d41cf.pyi b/.transpyc_cache/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/.transpyc_cache/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/.transpyc_cache/90d4bbcb87a33305.doc.json b/.transpyc_cache/90d4bbcb87a33305.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/90d4bbcb87a33305.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/90d4bbcb87a33305.pyi b/.transpyc_cache/90d4bbcb87a33305.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/90d4bbcb87a33305.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/9172f37e8d46721b.doc.json b/.transpyc_cache/9172f37e8d46721b.doc.json new file mode 100644 index 0000000..47b20d9 --- /dev/null +++ b/.transpyc_cache/9172f37e8d46721b.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/9172f37e8d46721b.pyi b/.transpyc_cache/9172f37e8d46721b.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/9172f37e8d46721b.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/91a2bd55f6b00ffc.doc.json b/.transpyc_cache/91a2bd55f6b00ffc.doc.json new file mode 100644 index 0000000..c3a1e25 --- /dev/null +++ b/.transpyc_cache/91a2bd55f6b00ffc.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/91a2bd55f6b00ffc.pyi b/.transpyc_cache/91a2bd55f6b00ffc.pyi new file mode 100644 index 0000000..ad12e1f --- /dev/null +++ b/.transpyc_cache/91a2bd55f6b00ffc.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 32 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/924ed8cd4ea5c4c2.doc.json b/.transpyc_cache/924ed8cd4ea5c4c2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/924ed8cd4ea5c4c2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/924ed8cd4ea5c4c2.pyi b/.transpyc_cache/924ed8cd4ea5c4c2.pyi new file mode 100644 index 0000000..16c5c7a --- /dev/null +++ b/.transpyc_cache/924ed8cd4ea5c4c2.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import c +import testcheck + +def test_string_ptr_iter() -> t.CInt: pass + +def test_string_ascii_sum() -> t.CInt: pass + +def test_manual_strcpy() -> t.CInt: pass + +def test_ptr_offset_compare() -> t.CInt: pass + +def test_manual_strlen() -> t.CInt: pass + +def test_reverse_string_ptr() -> t.CInt: pass + +def test_double_pointer() -> t.CInt: pass + +def test_ptr_arithmetic() -> t.CInt: pass + +def test_manual_strcmp() -> t.CInt: pass + + +class Vec2: + x: t.CInt + y: t.CInt + +def test_struct_ptr_access() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/93427ff78e455f30.doc.json b/.transpyc_cache/93427ff78e455f30.doc.json new file mode 100644 index 0000000..a63fb44 --- /dev/null +++ b/.transpyc_cache/93427ff78e455f30.doc.json @@ -0,0 +1 @@ +{"WhileHandle": "while 循环语句处理器：继承 Mixin 获得 Trans 回指针", "WhileHandle.Handle": "翻译 while 循环语句"} \ No newline at end of file diff --git a/.transpyc_cache/93427ff78e455f30.pyi b/.transpyc_cache/93427ff78e455f30.pyi new file mode 100644 index 0000000..0a1cc1e --- /dev/null +++ b/.transpyc_cache/93427ff78e455f30.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesWhile.py +Module: lib.core.Handles.HandlesWhile +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class WhileHandle(HandlesBase.Mixin): + def __init__(self: WhileHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: WhileHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewWhileHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> WhileHandle | t.CPtr: pass diff --git a/.transpyc_cache/93816021912fb156.doc.json b/.transpyc_cache/93816021912fb156.doc.json new file mode 100644 index 0000000..42f2fd1 --- /dev/null +++ b/.transpyc_cache/93816021912fb156.doc.json @@ -0,0 +1 @@ +{"AugAssignHandle": "增强赋值处理器 (+=, -=, *=, etc.)：继承 Mixin 获得 Trans 回指针", "AugAssignHandle.Handle": "翻译增强赋值语句 (x += 1, y -= 2, etc.)", "_apply_aug_op": "应用增强赋值的二元运算（指针算术 + 整数运算）"} \ No newline at end of file diff --git a/.transpyc_cache/93816021912fb156.pyi b/.transpyc_cache/93816021912fb156.pyi new file mode 100644 index 0000000..ee3342d --- /dev/null +++ b/.transpyc_cache/93816021912fb156.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAugAssign.py +Module: lib.core.Handles.HandlesAugAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import stdio +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal + +@t.NoVTable +class AugAssignHandle(HandlesBase.Mixin): + def __init__(self: AugAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AugAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def _apply_aug_op(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, op: int, lhs: llvmlite.Value | t.CPtr, rhs: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewAugAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AugAssignHandle | t.CPtr: pass diff --git a/.transpyc_cache/93964ed348985612.doc.json b/.transpyc_cache/93964ed348985612.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/93964ed348985612.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/93964ed348985612.pyi b/.transpyc_cache/93964ed348985612.pyi new file mode 100644 index 0000000..60fab85 --- /dev/null +++ b/.transpyc_cache/93964ed348985612.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/93c1d18e35d188d6.pyi b/.transpyc_cache/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/.transpyc_cache/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/.transpyc_cache/950028de389b7a7b.doc.json b/.transpyc_cache/950028de389b7a7b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/950028de389b7a7b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/950028de389b7a7b.pyi b/.transpyc_cache/950028de389b7a7b.pyi new file mode 100644 index 0000000..b7bdb15 --- /dev/null +++ b/.transpyc_cache/950028de389b7a7b.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import argparse +import w32.win32console as w32cmd +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import lib +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.core.Phase1 as Phase1 +import lib.core.Phase2 as Phase2 +import lib.Projectrans.Config as Config +import lib.Projectrans.Utils as Utils +import llvmlite +import subprocess +import viperlib +import hashlib + +POOL_SIZE: t.CDefine = 1073741824 +CODE_PAGE: t.CDefine = 65001 +SRC_BUF_SIZE: t.CDefine = 1048576 + +def main() -> int: pass diff --git a/.transpyc_cache/953883473b8184c1.doc.json b/.transpyc_cache/953883473b8184c1.doc.json new file mode 100644 index 0000000..5d5c7e9 --- /dev/null +++ b/.transpyc_cache/953883473b8184c1.doc.json @@ -0,0 +1 @@ +{"extract_func_attrs": "从 decorator_list 提取 c.Attribute 属性，返回 LLVM IR 属性字符串", "_get_attr_name_from_node": "从 AST 节点提取属性名\n\n 支持:\n - Call(func=Attribute(...)): t.attr.always_inline() -> \"always_inline\"\n - Attribute: t.attr.packed -> \"packed\"\n - Attribute(t.attr.llvm.xxx): t.attr.llvm.nounwind -> \"nounwind\"\n ", "_append_llvm_attr": "将属性名映射到 LLVM 属性名并追加到缓冲区\n\n 返回 1 表示已追加，0 表示不支持该属性\n ", "_append_str": "将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）", "_mangle_name": "为任意名称添加 SHA1 前缀（不做 main/export 检查）", "_mangle_name_with_sha1": "用指定 SHA1 为名称添加前缀（不做 main/export 检查）", "_mangle_func_name": "为函数名添加 SHA1 前缀（检查 main/export）", "translate_function_def": "翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）", "translate_nested_function_def": "翻译嵌套函数定义：提升为顶层函数 + 创建闭包", "create_function": "创建 LLVM 函数并添加参数"} \ No newline at end of file diff --git a/.transpyc_cache/953883473b8184c1.pyi b/.transpyc_cache/953883473b8184c1.pyi new file mode 100644 index 0000000..d7a865c --- /dev/null +++ b/.transpyc_cache/953883473b8184c1.pyi @@ -0,0 +1,43 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFunctions.py +Module: lib.core.Handles.HandlesFunctions +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import stdio +import stdlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesImports as HandlesImports + +def extract_func_attrs(pool: memhub.MemBuddy | t.CPtr, decorator_list: list[ast.AST | t.CPtr] | t.CPtr, imported_modules: str) -> t.CChar | t.CPtr: pass + +def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str: pass + +def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int: pass + +def _append_str(buf: t.CChar | t.CPtr, s: str) -> None: pass + +def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str: pass + +def _mangle_name_with_sha1(sha1: str, name: str) -> str: pass + +def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str, has_export: int) -> str: pass + +def translate_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nested_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, name: str, args_node: ast.AST | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Function | t.CPtr: pass diff --git a/.transpyc_cache/95394ca8da0f655a.pyi b/.transpyc_cache/95394ca8da0f655a.pyi new file mode 100644 index 0000000..27fd0e2 --- /dev/null +++ b/.transpyc_cache/95394ca8da0f655a.pyi @@ -0,0 +1,70 @@ +""" +Auto-generated Python stub file from llvmlite.__init__.py +Module: llvmlite.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .__types import LLVMType, ParamNode, LLVMTypeCore, Int1, Int8, Int16, Int32, Int64, Ptr, Func, Void, Array, Struct, Half, Float, Double, FP128, Label, new_param_node, param_list_append, paramnode_get_ty, paramnode_get_next, paramnode_set_ty, paramnode_set_next, TypePrint, PrintStructDefinition, get_struct_name +from .__values import Value, new_value, ConstInt, ConstNull, ConstZero, ConstFloat, SSAValue, ValuePrint, value_get_ty, value_get_next, value_set_ty, value_set_next, value_set_name, value_set_isconst +from .__function import Line, BasicBlock, Param, Function, new_line, new_basic_block, new_param, new_function, function_add_param, function_add_block, block_append_line, block_append_text, function_set_attrs, param_set_attrs, function_get_name, function_get_ret_ty, function_get_params, function_get_param_head, function_get_next, function_is_declared, param_get_name, param_get_ty, param_get_next, FunctionPrint +from .__module import LLVMModule, GlobalVariable, NamedTypeNode, new_module, new_global_variable, module_add_function, module_add_global, module_add_named_type, module_set_target, module_set_datalayout, global_set_constant, global_set_linkage, global_set_unnamed_addr, global_set_section, LLVMModulePrint, OUTPUT_FULL, OUTPUT_STUB, OUTPUT_TEXT, module_ensure_opaque_for_type +from .__builder import IRBuilder, new_builder, position_at_end, build_alloca, build_load, build_store, build_add, build_sub, build_mul, build_sdiv, build_udiv, build_srem, build_urem, build_and, build_or, build_xor, build_shl, build_lshr, build_ashr, build_fadd, build_fsub, build_fmul, build_fdiv, build_frem, build_fneg, build_icmp, build_fcmp, build_br, build_cond_br, build_ret, build_ret_void, build_call, build_call_indirect, build_bitcast, build_sext, build_trunc, build_zext, build_ptrtoint, build_inttoptr, build_fpext, build_fptrunc, build_fp2si, build_si2fp, build_fp2ui, build_ui2fp, build_gep, build_gep_array, build_gep_struct, build_phi, build_switch, build_select, build_unreachable, build_atomicrmw, build_cmpxchg, build_inline_asm, PhiIncoming, new_phi_incoming, SwitchCase, new_switch_case, ICMP_EQ, ICMP_NE, ICMP_SGT, ICMP_SGE, ICMP_SLT, ICMP_SLE, ICMP_UGT, ICMP_UGE, ICMP_ULT, ICMP_ULE, FCMP_FALSE, FCMP_OEQ, FCMP_OGT, FCMP_OGE, FCMP_OLT, FCMP_OLE, FCMP_ONE, FCMP_ORD, FCMP_UNO, FCMP_UEQ, FCMP_UGT, FCMP_UGE, FCMP_ULT, FCMP_ULE, FCMP_UNE, FCMP_TRUE, ATOMIC_XCHG, ATOMIC_ADD, ATOMIC_SUB, ATOMIC_AND, ATOMIC_NAND, ATOMIC_OR, ATOMIC_XOR, ATOMIC_MAX, ATOMIC_MIN, ATOMIC_UMAX, ATOMIC_UMIN, ATOMIC_ORDER_NOTATOMIC, ATOMIC_ORDER_UNORDERED, ATOMIC_ORDER_MONOTONIC, ATOMIC_ORDER_ACQUIRE, ATOMIC_ORDER_RELEASE, ATOMIC_ORDER_ACQ_REL, ATOMIC_ORDER_SEQ_CST +from .__verify import VerifyResult, NameEntry, verify_function, verify_module, VERIFY_OK, VERIFY_ERR_NO_TERMINATOR, VERIFY_ERR_DUPLICATE_DEF, VERIFY_ERR_UNDEFINED_USE, VERIFY_ERR_PHI_NOT_FIRST, _new_verify_result + +def _str_dup(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def create_block(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + +def const_int(pool: memhub.MemBuddy | t.CPtr, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int32(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int64(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def add_param(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + +def create_declare(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def create_global_string(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def create_global_int(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + + +@t.NoVTable +class LLVMModuleCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMModuleCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def NewModule(self: LLVMModuleCore, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + def CreateFunction(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def CreateDeclare(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def AddParam(self: LLVMModuleCore, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + def FunctionSetAttrs(self: LLVMModuleCore, func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def ParamSetAttrs(self: LLVMModuleCore, param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def CreateBlock(self: LLVMModuleCore, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + def NewBuilder(self: LLVMModuleCore, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + def NewGlobal(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalString(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalInt(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + def GlobalSetConstant(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + def GlobalSetLinkage(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + def GlobalSetUnnamedAddr(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + def GlobalSetSection(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + def ConstInt(self: LLVMModuleCore, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt32(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt64(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstFloat(self: LLVMModuleCore, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + def SSAValue(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def ConstNull(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def PrintValue(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr) -> t.CInt: pass + def PrintFunction(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr) -> t.CInt: pass + def PrintModule(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr) -> t.CInt: pass + def NewVerifyResult(self: LLVMModuleCore, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + def VerifyFunction(self: LLVMModuleCore, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + def VerifyModule(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/96368aa26ab1b560.doc.json b/.transpyc_cache/96368aa26ab1b560.doc.json new file mode 100644 index 0000000..525feb5 --- /dev/null +++ b/.transpyc_cache/96368aa26ab1b560.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/96368aa26ab1b560.pyi b/.transpyc_cache/96368aa26ab1b560.pyi new file mode 100644 index 0000000..8513bb5 --- /dev/null +++ b/.transpyc_cache/96368aa26ab1b560.pyi @@ -0,0 +1,56 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/96837bcc64032444.pyi b/.transpyc_cache/96837bcc64032444.pyi new file mode 100644 index 0000000..46e78bd --- /dev/null +++ b/.transpyc_cache/96837bcc64032444.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from hashlib.__init__.py +Module: hashlib.__init__ +""" + +import t +import c + + +from .__md5 import md5, MD5_DIGEST_LEN +from .__sha1 import sha1, SHA1_DIGEST_LEN +from .__sha256 import sha256, SHA256_DIGEST_LEN +from .__sha512 import sha512, SHA512_DIGEST_LEN \ No newline at end of file diff --git a/.transpyc_cache/969c589035910845.doc.json b/.transpyc_cache/969c589035910845.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/969c589035910845.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/969c589035910845.pyi b/.transpyc_cache/969c589035910845.pyi new file mode 100644 index 0000000..ae928ab --- /dev/null +++ b/.transpyc_cache/969c589035910845.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from flow_test.py +Module: flow_test +""" + + +import stdio +import t, c + +def flow_test() -> int: pass diff --git a/.transpyc_cache/97043f0d52da437f.doc.json b/.transpyc_cache/97043f0d52da437f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/97043f0d52da437f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/97043f0d52da437f.pyi b/.transpyc_cache/97043f0d52da437f.pyi new file mode 100644 index 0000000..f132db1 --- /dev/null +++ b/.transpyc_cache/97043f0d52da437f.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from inherit_test.py +Module: inherit_test +""" + + +import stdio +import t, c + +@t.CVTable +class Animal: + name: t.CInt + def __init__(self: Animal, n: t.CInt) -> t.CInt: pass + def GetName(self: Animal) -> t.CInt: pass + def Speak(self: Animal) -> t.CInt: pass +class Dog(Animal): + breed: t.CInt + def __init__(self: Dog, n: t.CInt, b: t.CInt) -> t.CInt: pass + def Speak(self: Dog) -> t.CInt: pass + def GetBreed(self: Dog) -> t.CInt: pass +class Cat(Animal): + color: t.CInt + def __init__(self: Cat, n: t.CInt, col: t.CInt) -> t.CInt: pass + def GetColor(self: Cat) -> t.CInt: pass + +def inherit_test() -> int: pass diff --git a/.transpyc_cache/982690df9fb41dde.doc.json b/.transpyc_cache/982690df9fb41dde.doc.json new file mode 100644 index 0000000..f6225ef --- /dev/null +++ b/.transpyc_cache/982690df9fb41dde.doc.json @@ -0,0 +1 @@ +{"translate_stmt": "翻译单条语句，返回新增的变量数", "translate_global": "翻译 global 语句：记录 global 变量名", "translate_nonlocal": "翻译 nonlocal 语句：记录 nonlocal 变量名", "translate_expr_stmt": "翻译表达式语句（printf 调用等）", "pre_scan_allocas": "预扫描语句中的 AnnAssign，提前创建 alloca\n\n 返回新增的变量数\n ", "translate_break": "翻译 break 语句：跳转到循环 end 块", "translate_continue": "翻译 continue 语句：跳转到循环 cond/incr 块", "get_stmt_kind_name": "获取语句类型名"} \ No newline at end of file diff --git a/.transpyc_cache/982690df9fb41dde.pyi b/.transpyc_cache/982690df9fb41dde.pyi new file mode 100644 index 0000000..aa52e34 --- /dev/null +++ b/.transpyc_cache/982690df9fb41dde.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBody.py +Module: lib.core.Handles.HandlesBody +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def translate_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_global(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nonlocal(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_expr_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def pre_scan_allocas(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_break(trans: HT.Translator | t.CPtr) -> int: pass + +def translate_continue(trans: HT.Translator | t.CPtr) -> int: pass + +def get_stmt_kind_name(node: ast.AST | t.CPtr) -> str: pass diff --git a/.transpyc_cache/9852dff1297b8322.doc.json b/.transpyc_cache/9852dff1297b8322.doc.json new file mode 100644 index 0000000..f67a6e1 --- /dev/null +++ b/.transpyc_cache/9852dff1297b8322.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/9852dff1297b8322.pyi b/.transpyc_cache/9852dff1297b8322.pyi new file mode 100644 index 0000000..477ab88 --- /dev/null +++ b/.transpyc_cache/9852dff1297b8322.pyi @@ -0,0 +1,169 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/98826042e4664018.doc.json b/.transpyc_cache/98826042e4664018.doc.json new file mode 100644 index 0000000..525feb5 --- /dev/null +++ b/.transpyc_cache/98826042e4664018.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/98826042e4664018.pyi b/.transpyc_cache/98826042e4664018.pyi new file mode 100644 index 0000000..8513bb5 --- /dev/null +++ b/.transpyc_cache/98826042e4664018.pyi @@ -0,0 +1,56 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/9928c0bfda57ff30.doc.json b/.transpyc_cache/9928c0bfda57ff30.doc.json new file mode 100644 index 0000000..f8bc59e --- /dev/null +++ b/.transpyc_cache/9928c0bfda57ff30.doc.json @@ -0,0 +1 @@ +{"add_imported_module": "记录已导入的模块名，返回新的 imported_modules 字符串指针", "is_module_imported": "检查模块是否已导入（词边界精确匹配，避免子串误匹配）", "add_from_import": "添加 from-import 映射 \"name:module\"，返回新的 from_imports 字符串", "lookup_from_import": "查找 from-import 名称，返回模块名或 None\n\n 支持 star import: 如果 from_imports 中有 \"*:module\" 条目，\n 且未找到精确名称匹配，则返回 star import 的模块名。\n ", "ImportsHandle": "导入语句处理器：继承 Mixin 获得 Trans 回指针", "ImportsHandle.HandleImport": "处理 import 语句，返回 0", "ImportsHandle.HandleImportFromModule": "处理 from X import Y 的模块部分，更新 trans._imported_modules", "ImportsHandle.HandleImportFromNames": "处理 from X import Y 的名称部分，更新 trans._from_imports"} \ No newline at end of file diff --git a/.transpyc_cache/9928c0bfda57ff30.pyi b/.transpyc_cache/9928c0bfda57ff30.pyi new file mode 100644 index 0000000..7abd965 --- /dev/null +++ b/.transpyc_cache/9928c0bfda57ff30.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesImports.py +Module: lib.core.Handles.HandlesImports +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, imported_modules: str, name: str) -> str: pass + +def is_module_imported(imported_modules: str, name: str) -> int: pass + +def add_from_import(pool: memhub.MemBuddy | t.CPtr, from_imports: str, local_name: str, module_name: str) -> str: pass + +def lookup_from_import(from_imports: str, name: str) -> str: pass + + +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + def __init__(self: ImportsHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleImport(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromModule(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromNames(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: pass diff --git a/.transpyc_cache/9b52a4613d3ad296.doc.json b/.transpyc_cache/9b52a4613d3ad296.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/9b52a4613d3ad296.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/9b52a4613d3ad296.pyi b/.transpyc_cache/9b52a4613d3ad296.pyi new file mode 100644 index 0000000..b7bdb15 --- /dev/null +++ b/.transpyc_cache/9b52a4613d3ad296.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import argparse +import w32.win32console as w32cmd +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import lib +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.core.Phase1 as Phase1 +import lib.core.Phase2 as Phase2 +import lib.Projectrans.Config as Config +import lib.Projectrans.Utils as Utils +import llvmlite +import subprocess +import viperlib +import hashlib + +POOL_SIZE: t.CDefine = 1073741824 +CODE_PAGE: t.CDefine = 65001 +SRC_BUF_SIZE: t.CDefine = 1048576 + +def main() -> int: pass diff --git a/.transpyc_cache/9b7d2adaae94f021.doc.json b/.transpyc_cache/9b7d2adaae94f021.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/9b7d2adaae94f021.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/9b7d2adaae94f021.pyi b/.transpyc_cache/9b7d2adaae94f021.pyi new file mode 100644 index 0000000..3552322 --- /dev/null +++ b/.transpyc_cache/9b7d2adaae94f021.pyi @@ -0,0 +1,65 @@ +""" +Auto-generated Python stub file from __zlib.zdef.py +Module: __zlib.zdef +""" + + +from stdint import * +import stddef +import string +import stdlib +import t, c + +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 +zdeflate_len_extra_bits: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_len_base: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_dist_extra_bits: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_dist_base: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_codelen_order: t.CExtern | t.CArray[int, 19] + +@t.Object +class zbit_writer: + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def __init__(self: zbit_writer) -> t.CInt: pass + def ensure(self: zbit_writer, need: t.CSizeT) -> t.CInt: pass + def write_bits(self: zbit_writer, value: UINT, nbits: t.CInt) -> t.CInt: pass + def write_bits_rev(self: zbit_writer, code: UINT, nbits: t.CInt) -> t.CInt: pass + def stored_block(self: zbit_writer, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_zlib_header(self: zbit_writer, wbits: t.CInt, level: t.CInt) -> t.CInt: pass + def write_gzip_header(self: zbit_writer) -> t.CInt: pass + def align(self: zbit_writer) -> t.CInt: pass + def total(self: zbit_writer) -> t.CSizeT: pass + def free(self: zbit_writer) -> t.CInt: pass + def __del__(self: zbit_writer) -> t.CInt: pass +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def init(self: zbit_reader, buf: BYTE | t.CPtr, length: t.CSizeT) -> t.CInt: pass + def read_bits(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_bits_rev(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def align(self: zbit_reader) -> t.CInt: pass + +def zdef_alloc(n: t.CSizeT) -> VOIDPTR: pass + +def zdef_free(p: VOIDPTR) -> t.CInt: pass diff --git a/.transpyc_cache/9ba128b762cf2b4d.doc.json b/.transpyc_cache/9ba128b762cf2b4d.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/.transpyc_cache/9ba128b762cf2b4d.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/.transpyc_cache/9ba128b762cf2b4d.pyi b/.transpyc_cache/9ba128b762cf2b4d.pyi new file mode 100644 index 0000000..9662312 --- /dev/null +++ b/.transpyc_cache/9ba128b762cf2b4d.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: t.CChar | t.CPtr) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/.transpyc_cache/9d798d964b9d3411.doc.json b/.transpyc_cache/9d798d964b9d3411.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/9d798d964b9d3411.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/9d798d964b9d3411.pyi b/.transpyc_cache/9d798d964b9d3411.pyi new file mode 100644 index 0000000..e7501ef --- /dev/null +++ b/.transpyc_cache/9d798d964b9d3411.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from new_test.py +Module: new_test +""" + + +import stdio +import t, c + +class WithNew: + value: t.CInt + def __new__(self: WithNew, v: t.CInt) -> t.CInt: pass + def __init__(self: WithNew, v: t.CInt) -> t.CInt: pass + def GetValue(self: WithNew) -> t.CInt: pass +class WithNewNoArgs: + value: t.CInt + def __new__(self: WithNewNoArgs) -> t.CInt: pass + def __init__(self: WithNewNoArgs, v: t.CInt) -> t.CInt: pass + def GetValue(self: WithNewNoArgs) -> t.CInt: pass +@t.CVTable +class WithNewVTable: + value: t.CInt + def __new__(self: WithNewVTable, v: t.CInt) -> t.CInt: pass + def __init__(self: WithNewVTable, v: t.CInt) -> t.CInt: pass + def GetValue(self: WithNewVTable) -> t.CInt: pass + def Speak(self: WithNewVTable) -> t.CInt: pass + +def new_test() -> int: pass diff --git a/.transpyc_cache/9d8e1c9631c28b1b.doc.json b/.transpyc_cache/9d8e1c9631c28b1b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/9d8e1c9631c28b1b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/9d8e1c9631c28b1b.pyi b/.transpyc_cache/9d8e1c9631c28b1b.pyi new file mode 100644 index 0000000..e67abb0 --- /dev/null +++ b/.transpyc_cache/9d8e1c9631c28b1b.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from closure_test.py +Module: closure_test +""" + + +import stdio +import t, c + +g_count: t.CExtern | int + +def test_global() -> int: pass + +def make_counter() -> t.CPtr: pass + +def closure_test() -> int: pass diff --git a/.transpyc_cache/9e38ca4de81ccb6d.doc.json b/.transpyc_cache/9e38ca4de81ccb6d.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/9e38ca4de81ccb6d.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/9e38ca4de81ccb6d.pyi b/.transpyc_cache/9e38ca4de81ccb6d.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/9e38ca4de81ccb6d.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/9f6e387913081f0e.doc.json b/.transpyc_cache/9f6e387913081f0e.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/.transpyc_cache/9f6e387913081f0e.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/.transpyc_cache/9f6e387913081f0e.pyi b/.transpyc_cache/9f6e387913081f0e.pyi new file mode 100644 index 0000000..89de3bc --- /dev/null +++ b/.transpyc_cache/9f6e387913081f0e.pyi @@ -0,0 +1,88 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/9fdf7e28103d0ff0.doc.json b/.transpyc_cache/9fdf7e28103d0ff0.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/.transpyc_cache/9fdf7e28103d0ff0.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/.transpyc_cache/9fdf7e28103d0ff0.pyi b/.transpyc_cache/9fdf7e28103d0ff0.pyi new file mode 100644 index 0000000..b529a3b --- /dev/null +++ b/.transpyc_cache/9fdf7e28103d0ff0.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: str) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/.transpyc_cache/a03139660029514f.doc.json b/.transpyc_cache/a03139660029514f.doc.json new file mode 100644 index 0000000..7aab3b8 --- /dev/null +++ b/.transpyc_cache/a03139660029514f.doc.json @@ -0,0 +1 @@ +{"_cleanup": "清理可能残留的测试文件/目录"} \ No newline at end of file diff --git a/.transpyc_cache/a03139660029514f.pyi b/.transpyc_cache/a03139660029514f.pyi new file mode 100644 index 0000000..39bb90d --- /dev/null +++ b/.transpyc_cache/a03139660029514f.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from syslibtest_impl.py +Module: syslibtest_impl +""" + + +import t, c +from stdint import * +from t import CInt, CPtr, CUInt8T, CSizeT +import testcheck +import viperlib +import stdlib +import memhub +import platmacro +import os +import os.path +import os_backend +import shutil +import sys +import subprocess +import string +import w32.fileio +import w32.win32base +import w32.win32process + +TEST_DIR: t.CDefine = "__sylibtest_dir__" +TEST_FILE: t.CDefine = "__sylibtest_file.txt" +TEST_COPY: t.CDefine = "__sylibtest_copy.txt" +TEST_TREE: t.CDefine = "__sylibtest_tree__" +TEST_MOVE: t.CDefine = "__sylibtest_moved.txt" + +def _cleanup() -> t.CInt: pass + +def run() -> CInt: pass diff --git a/.transpyc_cache/a0612a85a12439a9.doc.json b/.transpyc_cache/a0612a85a12439a9.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/a0612a85a12439a9.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/a0612a85a12439a9.pyi b/.transpyc_cache/a0612a85a12439a9.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/a0612a85a12439a9.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/a0e0675172e4f01f.doc.json b/.transpyc_cache/a0e0675172e4f01f.doc.json new file mode 100644 index 0000000..5d5c7e9 --- /dev/null +++ b/.transpyc_cache/a0e0675172e4f01f.doc.json @@ -0,0 +1 @@ +{"extract_func_attrs": "从 decorator_list 提取 c.Attribute 属性，返回 LLVM IR 属性字符串", "_get_attr_name_from_node": "从 AST 节点提取属性名\n\n 支持:\n - Call(func=Attribute(...)): t.attr.always_inline() -> \"always_inline\"\n - Attribute: t.attr.packed -> \"packed\"\n - Attribute(t.attr.llvm.xxx): t.attr.llvm.nounwind -> \"nounwind\"\n ", "_append_llvm_attr": "将属性名映射到 LLVM 属性名并追加到缓冲区\n\n 返回 1 表示已追加，0 表示不支持该属性\n ", "_append_str": "将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）", "_mangle_name": "为任意名称添加 SHA1 前缀（不做 main/export 检查）", "_mangle_name_with_sha1": "用指定 SHA1 为名称添加前缀（不做 main/export 检查）", "_mangle_func_name": "为函数名添加 SHA1 前缀（检查 main/export）", "translate_function_def": "翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）", "translate_nested_function_def": "翻译嵌套函数定义：提升为顶层函数 + 创建闭包", "create_function": "创建 LLVM 函数并添加参数"} \ No newline at end of file diff --git a/.transpyc_cache/a0e0675172e4f01f.pyi b/.transpyc_cache/a0e0675172e4f01f.pyi new file mode 100644 index 0000000..d7a865c --- /dev/null +++ b/.transpyc_cache/a0e0675172e4f01f.pyi @@ -0,0 +1,43 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFunctions.py +Module: lib.core.Handles.HandlesFunctions +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import stdio +import stdlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesImports as HandlesImports + +def extract_func_attrs(pool: memhub.MemBuddy | t.CPtr, decorator_list: list[ast.AST | t.CPtr] | t.CPtr, imported_modules: str) -> t.CChar | t.CPtr: pass + +def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str: pass + +def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int: pass + +def _append_str(buf: t.CChar | t.CPtr, s: str) -> None: pass + +def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str: pass + +def _mangle_name_with_sha1(sha1: str, name: str) -> str: pass + +def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str, has_export: int) -> str: pass + +def translate_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nested_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, name: str, args_node: ast.AST | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Function | t.CPtr: pass diff --git a/.transpyc_cache/a30bd51aa00f0bef.doc.json b/.transpyc_cache/a30bd51aa00f0bef.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/.transpyc_cache/a30bd51aa00f0bef.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/.transpyc_cache/a30bd51aa00f0bef.pyi b/.transpyc_cache/a30bd51aa00f0bef.pyi new file mode 100644 index 0000000..0e885bf --- /dev/null +++ b/.transpyc_cache/a30bd51aa00f0bef.pyi @@ -0,0 +1,89 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/.transpyc_cache/a47313ed1481da7d.doc.json b/.transpyc_cache/a47313ed1481da7d.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/a47313ed1481da7d.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/a47313ed1481da7d.pyi b/.transpyc_cache/a47313ed1481da7d.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/a47313ed1481da7d.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/a5407d80d9f015ed.doc.json b/.transpyc_cache/a5407d80d9f015ed.doc.json new file mode 100644 index 0000000..c3a1e25 --- /dev/null +++ b/.transpyc_cache/a5407d80d9f015ed.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/a5407d80d9f015ed.pyi b/.transpyc_cache/a5407d80d9f015ed.pyi new file mode 100644 index 0000000..ad12e1f --- /dev/null +++ b/.transpyc_cache/a5407d80d9f015ed.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 32 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/a68b70f233541a7f.pyi b/.transpyc_cache/a68b70f233541a7f.pyi new file mode 100644 index 0000000..3cb3a02 --- /dev/null +++ b/.transpyc_cache/a68b70f233541a7f.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from os._win32.py +Module: os._win32 +""" + + +import t, c +from stdint import * +import w32.win32base +import w32.win32file +import memhub +import stdlib + +MAX_PATH_LEN: t.CDefine = 260 +INVALID_FILE_ATTRIBUTES: t.CDefine = 0xFFFFFFFF +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def _get_attrs(path: str) -> ULONG: pass + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd() -> str: pass + +def chdir(path: str) -> int: pass + +def mkdir(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir(path: str) -> int: pass + +def rename(old_path: str, new_path: str) -> int: pass + +def chmod(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system(cmd: str) -> int: pass diff --git a/.transpyc_cache/a6972611852b8a4e.doc.json b/.transpyc_cache/a6972611852b8a4e.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/a6972611852b8a4e.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/a6972611852b8a4e.pyi b/.transpyc_cache/a6972611852b8a4e.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/a6972611852b8a4e.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/a6a3e195cd7dd0b4.doc.json b/.transpyc_cache/a6a3e195cd7dd0b4.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/.transpyc_cache/a6a3e195cd7dd0b4.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/.transpyc_cache/a6a3e195cd7dd0b4.pyi b/.transpyc_cache/a6a3e195cd7dd0b4.pyi new file mode 100644 index 0000000..d866b3e --- /dev/null +++ b/.transpyc_cache/a6a3e195cd7dd0b4.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/a711c9c66fd7c6ca.doc.json b/.transpyc_cache/a711c9c66fd7c6ca.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/a711c9c66fd7c6ca.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/a711c9c66fd7c6ca.pyi b/.transpyc_cache/a711c9c66fd7c6ca.pyi new file mode 100644 index 0000000..213e2c8 --- /dev/null +++ b/.transpyc_cache/a711c9c66fd7c6ca.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + +def bump_true(buf: bytes) -> bool: pass + +def bump_false(buf: bytes) -> bool: pass + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/a7cf0ce4066332ae.doc.json b/.transpyc_cache/a7cf0ce4066332ae.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/a7cf0ce4066332ae.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/a7cf0ce4066332ae.pyi b/.transpyc_cache/a7cf0ce4066332ae.pyi new file mode 100644 index 0000000..3042835 --- /dev/null +++ b/.transpyc_cache/a7cf0ce4066332ae.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from opovl_test.py +Module: opovl_test +""" + + +import stdio +import stdlib +import t, c +import testcheck + +class Counter: + val: t.CInt + def __init__(self: Counter, v: t.CInt) -> t.CInt: pass + def __add__(self: Counter, n: t.CInt) -> t.CInt: pass + def __sub__(self: Counter, n: t.CInt) -> t.CInt: pass + def __mul__(self: Counter, n: t.CInt) -> t.CInt: pass + def __div__(self: Counter, n: t.CInt) -> t.CInt: pass + def __mod__(self: Counter, n: t.CInt) -> t.CInt: pass + def __eq__(self: Counter, n: t.CInt) -> t.CInt: pass + def __ne__(self: Counter, n: t.CInt) -> t.CInt: pass + def __lt__(self: Counter, n: t.CInt) -> t.CInt: pass + def __le__(self: Counter, n: t.CInt) -> t.CInt: pass + def __gt__(self: Counter, n: t.CInt) -> t.CInt: pass + def __ge__(self: Counter, n: t.CInt) -> t.CInt: pass +class BitBox: + flags: t.CInt + def __init__(self: BitBox, v: t.CInt) -> t.CInt: pass + def __and__(self: BitBox, mask: t.CInt) -> t.CInt: pass + def __or__(self: BitBox, mask: t.CInt) -> t.CInt: pass + def __xor__(self: BitBox, mask: t.CInt) -> t.CInt: pass + def __lshift__(self: BitBox, n: t.CInt) -> t.CInt: pass + def __rshift__(self: BitBox, n: t.CInt) -> t.CInt: pass +class Vec2: + x: t.CInt + y: t.CInt + def __new__(self: Vec2, x0: t.CInt, y0: t.CInt) -> t.CInt: pass + def __init__(self: Vec2, x0: t.CInt, y0: t.CInt) -> t.CInt: pass + def __add__(self: Vec2, other: Vec2 | t.CPtr) -> Vec2 | t.CPtr: pass + def __sub__(self: Vec2, other: Vec2 | t.CPtr) -> Vec2 | t.CPtr: pass + def __eq__(self: Vec2, other: Vec2 | t.CPtr) -> t.CInt: pass + def __ne__(self: Vec2, other: Vec2 | t.CPtr) -> t.CInt: pass + +def opovl_test() -> int: pass diff --git a/.transpyc_cache/a9ad5e41081c9377.doc.json b/.transpyc_cache/a9ad5e41081c9377.doc.json new file mode 100644 index 0000000..47b20d9 --- /dev/null +++ b/.transpyc_cache/a9ad5e41081c9377.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/a9ad5e41081c9377.pyi b/.transpyc_cache/a9ad5e41081c9377.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/a9ad5e41081c9377.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/a9b97a11bf30daa0.doc.json b/.transpyc_cache/a9b97a11bf30daa0.doc.json new file mode 100644 index 0000000..67e7a8b --- /dev/null +++ b/.transpyc_cache/a9b97a11bf30daa0.doc.json @@ -0,0 +1 @@ +{"_DiagAppend": "追加诊断信息到 diag.txt（追加模式，不依赖 printf 缓冲）", "TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/a9b97a11bf30daa0.pyi b/.transpyc_cache/a9b97a11bf30daa0.pyi new file mode 100644 index 0000000..9d8b33e --- /dev/null +++ b/.transpyc_cache/a9b97a11bf30daa0.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def _DiagAppend(msg: bytes) -> t.CInt: pass + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/a9df037d9178888f.doc.json b/.transpyc_cache/a9df037d9178888f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/a9df037d9178888f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/a9df037d9178888f.pyi b/.transpyc_cache/a9df037d9178888f.pyi new file mode 100644 index 0000000..4b413d7 --- /dev/null +++ b/.transpyc_cache/a9df037d9178888f.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from func_vtable_test.py +Module: func_vtable_test +""" + + +import stdio +import t, c + +@t.NoVTable +class NoVTableClass: + val: t.CInt + def __init__(self: NoVTableClass, v: t.CInt) -> t.CInt: pass + @t.CVTable + def VirtualMethod(self: NoVTableClass) -> t.CInt: pass + def NormalMethod(self: NoVTableClass) -> t.CInt: pass +@t.CVTable +class CVTableClass: + val: t.CInt + def __init__(self: CVTableClass, v: t.CInt) -> t.CInt: pass + def KeptMethod(self: CVTableClass) -> t.CInt: pass + @t.NoVTable + def ExcludedMethod(self: CVTableClass) -> t.CInt: pass +class DefaultClass: + val: t.CInt + def __init__(self: DefaultClass, v: t.CInt) -> t.CInt: pass + @t.CVTable + def VirtualMethod(self: DefaultClass) -> t.CInt: pass + def NormalMethod(self: DefaultClass) -> t.CInt: pass + +def func_vtable_test() -> int: pass diff --git a/.transpyc_cache/ab6e54ba9a669f76.pyi b/.transpyc_cache/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/.transpyc_cache/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/.transpyc_cache/aba439b7882ad9d6.pyi b/.transpyc_cache/aba439b7882ad9d6.pyi new file mode 100644 index 0000000..124ce31 --- /dev/null +++ b/.transpyc_cache/aba439b7882ad9d6.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from argparse.py +Module: argparse +""" + + +import t, c +from stdint import * +import string +import memhub +import stdio +import hashtable + +STRING: t.CDefine = 0 +INT: t.CDefine = 1 +BOOL: t.CDefine = 2 +STORE: t.CDefine = 0 +STORE_TRUE: t.CDefine = 1 +STORE_FALSE: t.CDefine = 2 +COUNT: t.CDefine = 3 +MAX_ARGS: t.CDefine = 32 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class Argument: + name: str + short_name: str + help_text: str + arg_type: INT + action: INT + default_int: INT + default_str: str + required: bool + is_positional: bool + dest: str +class ParsedArgs: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def get_str(self: ParsedArgs, name: str) -> str: pass + def get_int(self: ParsedArgs, name: str) -> INT: pass + def get_bool(self: ParsedArgs, name: str) -> INT: pass + def has(self: ParsedArgs, name: str) -> bool: pass + def _set_str(self: ParsedArgs, name: str, val: str) -> t.CInt: pass + def _set_int(self: ParsedArgs, name: str, val: INT) -> t.CInt: pass + def _set_bool(self: ParsedArgs, name: str, val: bool) -> t.CInt: pass +class ArgumentParser: + _prog: str + _description: str + _args: Argument | t.CPtr + _arg_count: INT + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def add_argument(self: ArgumentParser, name: str, short: str, arg_type: INT, default: INT, default_str: str, required: bool, action: INT, help: str) -> t.CInt: pass + def _get_arg(self: ArgumentParser, idx: INT) -> Argument | t.CPtr: pass + def _find_long(self: ArgumentParser, token: str) -> INT: pass + def _find_long_prefix(self: ArgumentParser, token: str) -> INT: pass + def _find_short(self: ArgumentParser, token: str) -> INT: pass + def _find_positional(self: ArgumentParser, pos_idx: INT) -> INT: pass + def _store_value(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr, value: str) -> t.CInt: pass + def _apply_action(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr) -> t.CInt: pass + def parse_args(self: ArgumentParser, argc: INT, argv: str | t.CPtr) -> ParsedArgs | t.CPtr: pass + def print_help(self: ArgumentParser) -> t.CInt: pass + def print_usage(self: ArgumentParser) -> t.CInt: pass + +def release(args: ParsedArgs | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/abbcbd92436cc16a.pyi b/.transpyc_cache/abbcbd92436cc16a.pyi new file mode 100644 index 0000000..9a57653 --- /dev/null +++ b/.transpyc_cache/abbcbd92436cc16a.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from shutil.py +Module: shutil +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import w32.win32base +import w32.win32file +import posix + +COPY_BUF_SIZE: t.CDefine = 65536 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def copyfile(src: str, dst: str) -> int: pass + +def _copyfile_win32(src: str, dst: str) -> int: pass + +def _copyfile_posix(src: str, dst: str) -> int: pass + +def copy(src: str, dst: str) -> int: pass + +def _copy_win32(src: str, dst: str) -> int: pass + +def _copy_posix(src: str, dst: str) -> int: pass + +def rmtree(path: str) -> int: pass + +def _rmtree_win32(path: str) -> int: pass + +def _rmtree_posix(path: str) -> int: pass + +def move(src: str, dst: str) -> int: pass + +def _move_win32(src: str, dst: str) -> int: pass + +def _move_posix(src: str, dst: str) -> int: pass diff --git a/.transpyc_cache/ac0d686cb2cc783b.doc.json b/.transpyc_cache/ac0d686cb2cc783b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/ac0d686cb2cc783b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/ac0d686cb2cc783b.pyi b/.transpyc_cache/ac0d686cb2cc783b.pyi new file mode 100644 index 0000000..7b2670f --- /dev/null +++ b/.transpyc_cache/ac0d686cb2cc783b.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import t as tt +import stdio +import testcheck +from t import CEnum as en +from t import CUnion as un +from t import REnum as ren +from t import CStruct as cs +from t import CInt as myint +from t import CChar, CPtr + +class Color(en): + Red = 0 + Green = 1 + Blue = 2 +class Status(tt.CEnum): + Ok = 0 + Error = 1 +class Data(un): + i_val: t.CInt + f_val: t.CFloat +class Opt(ren): + class Some: + val: myint + None_ = 0 +class Parent: + a: t.CInt + def get_a(self: Parent) -> t.CInt: pass +class Child(Parent): + b: t.CInt + def get_sum(self: Child) -> t.CInt: pass +class Base: + x: t.CInt + def base_val(self: Base) -> t.CInt: pass +class Mid(Base): + y: t.CInt + def mid_val(self: Mid) -> t.CInt: pass +class Leaf(Mid): + z: t.CInt + def leaf_val(self: Leaf) -> t.CInt: pass + +val: t.CExtern | myint + +class NetHeader: + magic: t.CInt | t.BigEndian + length: t.CShort | t.BigEndian + normal: t.CInt +class TtPoint(tt.CStruct): + x: t.CInt + y: t.CInt +@tt.Object +class TtObject: + val: t.CInt + def get_val(self: TtObject) -> t.CInt: pass +class CsPoint(cs): + x: t.CInt + y: t.CInt +class TtUnion(tt.CUnion): + a: t.CInt + b: t.CFloat +@tt.Object +class TtEmptyObject: + val: t.CInt + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/aca085940d1ca494.doc.json b/.transpyc_cache/aca085940d1ca494.doc.json new file mode 100644 index 0000000..47b20d9 --- /dev/null +++ b/.transpyc_cache/aca085940d1ca494.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/aca085940d1ca494.pyi b/.transpyc_cache/aca085940d1ca494.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/aca085940d1ca494.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/ae0cc890f657076d.doc.json b/.transpyc_cache/ae0cc890f657076d.doc.json new file mode 100644 index 0000000..0d6dd61 --- /dev/null +++ b/.transpyc_cache/ae0cc890f657076d.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "find_global_in_module": "在模块全局变量链表中按名称查找全局变量", "make_global_ref": "创建全局变量引用 (类型为 ty*，名称为 @name)", "_resolve_module_attribute_global": "解析模块属性访问，返回全局变量引用 (用于跨模块 global 访问)", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/ae0cc890f657076d.pyi b/.transpyc_cache/ae0cc890f657076d.pyi new file mode 100644 index 0000000..2cb20be --- /dev/null +++ b/.transpyc_cache/ae0cc890f657076d.pyi @@ -0,0 +1,88 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesImports as HandlesImports + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def find_global_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.GlobalVariable | t.CPtr: pass + +def make_global_ref(pool: memhub.MemBuddy | t.CPtr, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _resolve_module_attribute_global(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, trans: HT.Translator | t.CPtr, at: ast.Attribute | t.CPtr, ty_hint: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/ae1e8a6903834d9c.doc.json b/.transpyc_cache/ae1e8a6903834d9c.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/ae1e8a6903834d9c.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/ae1e8a6903834d9c.pyi b/.transpyc_cache/ae1e8a6903834d9c.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/ae1e8a6903834d9c.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/aeb670958072076e.pyi b/.transpyc_cache/aeb670958072076e.pyi new file mode 100644 index 0000000..a953d5c --- /dev/null +++ b/.transpyc_cache/aeb670958072076e.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from ast.visitor.py +Module: ast.visitor +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind + +@t.CVTable +class ASTVisitor: + pool: memhub.MemManager | t.CPtr + def __new__(self: ASTVisitor, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: ASTVisitor, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def visit(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Module(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_FunctionDef(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Assign(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_If(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_For(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_While(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Return(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Expr(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Name(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Constant(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_BinOp(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_UnaryOp(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Call(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_Compare(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass + def visit_generic(self: ASTVisitor, node: AST | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/aeb8034bdb2b370b.doc.json b/.transpyc_cache/aeb8034bdb2b370b.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/aeb8034bdb2b370b.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/aeb8034bdb2b370b.pyi b/.transpyc_cache/aeb8034bdb2b370b.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/aeb8034bdb2b370b.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/af9484a65085bf4b.doc.json b/.transpyc_cache/af9484a65085bf4b.doc.json new file mode 100644 index 0000000..2fbc2fa --- /dev/null +++ b/.transpyc_cache/af9484a65085bf4b.doc.json @@ -0,0 +1 @@ +{"get_env_ptr_var": "获取或创建 _env_ptr 变量（存储闭包 env 指针）", "load_nonlocal_var": "从闭包 env 加载 nonlocal 变量值（env 直接存储 i32 值）", "get_nonlocal_var_ptr": "获取 nonlocal 变量在 env 中的地址（i32*），用于 store 操作"} \ No newline at end of file diff --git a/.transpyc_cache/af9484a65085bf4b.pyi b/.transpyc_cache/af9484a65085bf4b.pyi new file mode 100644 index 0000000..3d49af5 --- /dev/null +++ b/.transpyc_cache/af9484a65085bf4b.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesNonlocal.py +Module: lib.core.Handles.HandlesNonlocal +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar + +def get_env_ptr_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def load_nonlocal_var(trans: HT.Translator | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def get_nonlocal_var_ptr(trans: HT.Translator | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/b1122357fee8fd68.doc.json b/.transpyc_cache/b1122357fee8fd68.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b1122357fee8fd68.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b1122357fee8fd68.pyi b/.transpyc_cache/b1122357fee8fd68.pyi new file mode 100644 index 0000000..97850c4 --- /dev/null +++ b/.transpyc_cache/b1122357fee8fd68.pyi @@ -0,0 +1,36 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdio import printf +import testcheck + +@t.CStruct +class Point: + x: t.CInt + y: t.CInt +@t.CEnum +class Color: + RED = 1 + GREEN = 2 + BLUE = 3 +@t.CUnion +class Data: + i: t.CInt + f: t.CFloat + +@t.CExport +def test() -> t.CInt: pass + +@t.CStatic +def static_func() -> t.CInt: pass + +@t.CStatic +def mixed_func() -> t.CInt | t.CExport: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/b235d6477436c524.doc.json b/.transpyc_cache/b235d6477436c524.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b235d6477436c524.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b235d6477436c524.pyi b/.transpyc_cache/b235d6477436c524.pyi new file mode 100644 index 0000000..fa0bd7d --- /dev/null +++ b/.transpyc_cache/b235d6477436c524.pyi @@ -0,0 +1,131 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CPtr, CChar, CInt32T, CUInt64T, CFloat64T, CExport, State +import viperlib +import stdlib +import definetest +import enumtest +import mpooltest +import vectortest +import fileiotest +import testcheck + +def add[T](a: T, b: T) -> T: pass + +def add_u64[T](a: T, b: T, c: CUInt64T) -> T: pass + +def fxi[T1, T2](a: T1, b: T2) -> T1: pass + +def fxi2[T1, T2, T3](a: T1, b: T2, c: T3) -> T1: pass + +def wrap_add[T](a: T, b: T) -> T: pass + +def double_add[T](a: T, b: T) -> T: pass + + +class A[T]: + def __init__(self: A, a: T) -> t.CInt: pass + def get_a(self: A) -> T: pass +class Pair[T1, T2]: + def __init__(self: Pair, first: T1, second: T2) -> t.CInt: pass + def get_first(self: Pair) -> T1: pass + def get_second(self: Pair) -> T2: pass +class Calculator[T]: + def __init__(self: Calculator, init_val: T) -> t.CInt: pass + def compute(self: Calculator, other: T) -> T: pass + def double_compute(self: Calculator, other: T) -> T: pass +class Container[T]: + def __init__(self: Container, v: T) -> t.CInt: pass + def get_value(self: Container) -> T: pass + def get_flag(self: Container) -> CInt32T: pass +@t.CVTable +class Shape: + def __init__(self: Shape, x: CFloat64T, y: CFloat64T) -> t.CInt: pass + def area(self: Shape) -> CFloat64T: pass + def perimeter(self: Shape) -> CFloat64T: pass + def describe(self: Shape) -> CFloat64T: pass +@t.CVTable +class Circle(Shape): + def __init__(self: Circle, x: CFloat64T, y: CFloat64T, r: CFloat64T) -> t.CInt: pass + def area(self: Circle) -> CFloat64T: pass + def perimeter(self: Circle) -> CFloat64T: pass + def scale(self: Circle, factor: CFloat64T) -> 'Circle' | CPtr: pass + def move(self: Circle, dx: CFloat64T, dy: CFloat64T) -> 'Circle' | CPtr: pass +@t.CVTable +class Rect(Shape): + def __init__(self: Rect, x: CFloat64T, y: CFloat64T, w: CFloat64T, h: CFloat64T) -> t.CInt: pass + def area(self: Rect) -> CFloat64T: pass + def perimeter(self: Rect) -> CFloat64T: pass + def scale(self: Rect, factor: CFloat64T) -> 'Rect' | CPtr: pass +class Vec2: + x: CFloat64T + y: CFloat64T + def __new__() -> 'Vec2' | CPtr: pass + def __init__(self: Vec2, x: CFloat64T, y: CFloat64T) -> t.CInt: pass + def __add__(self: Vec2, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: pass + def __sub__(self: Vec2, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: pass + def __mul__(self: Vec2, s: CFloat64T) -> 'Vec2' | CPtr: pass + def __neg__(self: Vec2) -> 'Vec2' | CPtr: pass + def dot(self: Vec2, b: 'Vec2' | CPtr) -> CFloat64T: pass + def len_sq(self: Vec2) -> CFloat64T: pass +class Vec3: + x: CFloat64T + y: CFloat64T + z: CFloat64T + def __new__() -> 'Vec3' | CPtr: pass + def __init__(self: Vec3, x: CFloat64T, y: CFloat64T, z: CFloat64T) -> t.CInt: pass + def __add__(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def __sub__(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def __mul__(self: Vec3, s: CFloat64T) -> 'Vec3' | CPtr: pass + def __neg__(self: Vec3) -> 'Vec3' | CPtr: pass + def dot(self: Vec3, b: 'Vec3' | CPtr) -> CFloat64T: pass + def cross(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def len_sq(self: Vec3) -> CFloat64T: pass +@t.CVTable +class Dog: + def __init__(self: Dog, name_val: CInt, bark_power: CInt32T) -> t.CInt: pass + def speak(self: Dog) -> CInt: pass + def bite(self: Dog) -> CInt32T: pass + def take_damage(self: Dog, dmg: CInt32T) -> 'Dog' | CPtr: pass + def is_alive(self: Dog) -> CInt: pass +@t.CVTable +class Cat: + def __init__(self: Cat, name_val: CInt, lives: CInt32T) -> t.CInt: pass + def speak(self: Cat) -> CInt: pass + def scratch(self: Cat) -> CInt32T: pass + def take_damage(self: Cat, dmg: CInt32T) -> 'Cat' | CPtr: pass + def is_alive(self: Cat) -> CInt: pass +class Transform: + px: CFloat64T + py: CFloat64T + pz: CFloat64T + scale_x: CFloat64T + scale_y: CFloat64T + scale_z: CFloat64T + def __new__() -> 'Transform' | CPtr: pass + def __init__(self: Transform, px: CFloat64T, py: CFloat64T, pz: CFloat64T) -> t.CInt: pass + def apply_scale(self: Transform, sx: CFloat64T, sy: CFloat64T, sz: CFloat64T) -> 'Transform' | CPtr: pass + def world_position(self: Transform) -> 'Vec3' | CPtr: pass +@t.CVTable +class Vehicle: + def __init__(self: Vehicle, speed: CInt32T) -> t.CInt: pass + def move(self: Vehicle) -> CInt: pass + def is_running(self: Vehicle) -> CInt: pass +@t.CVTable +class Car(Vehicle): + def __init__(self: Car, speed: CInt32T, doors: CInt32T) -> t.CInt: pass + def honk(self: Car) -> CInt: pass +@t.CVTable +class ElectricCar(Car): + def __init__(self: ElectricCar, speed: CInt32T, doors: CInt32T, battery: CInt32T) -> t.CInt: pass + def charge(self: ElectricCar) -> t.CInt: pass + def move(self: ElectricCar) -> CInt: pass + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/b298756207e2da58.doc.json b/.transpyc_cache/b298756207e2da58.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/b298756207e2da58.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/b298756207e2da58.pyi b/.transpyc_cache/b298756207e2da58.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/b298756207e2da58.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/b29b80b6106e4be7.doc.json b/.transpyc_cache/b29b80b6106e4be7.doc.json new file mode 100644 index 0000000..724daa6 --- /dev/null +++ b/.transpyc_cache/b29b80b6106e4be7.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到\n\n allow_star_fallback: 传 0 禁用 star import 回退（用于明确模块名/函数名查找）\n ", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/b29b80b6106e4be7.pyi b/.transpyc_cache/b29b80b6106e4be7.pyi new file mode 100644 index 0000000..065139b --- /dev/null +++ b/.transpyc_cache/b29b80b6106e4be7.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str, allow_star_fallback: int) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/b3c2b99b6f3be4c7.doc.json b/.transpyc_cache/b3c2b99b6f3be4c7.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b3c2b99b6f3be4c7.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b3c2b99b6f3be4c7.pyi b/.transpyc_cache/b3c2b99b6f3be4c7.pyi new file mode 100644 index 0000000..155b94e --- /dev/null +++ b/.transpyc_cache/b3c2b99b6f3be4c7.pyi @@ -0,0 +1,17 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import testcheck + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/b3fd73f7abbd88a8.doc.json b/.transpyc_cache/b3fd73f7abbd88a8.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/b3fd73f7abbd88a8.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/b3fd73f7abbd88a8.pyi b/.transpyc_cache/b3fd73f7abbd88a8.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/b3fd73f7abbd88a8.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/b436212dcff1be29.doc.json b/.transpyc_cache/b436212dcff1be29.doc.json new file mode 100644 index 0000000..c6f42ec --- /dev/null +++ b/.transpyc_cache/b436212dcff1be29.doc.json @@ -0,0 +1 @@ +{"test_global_write": "测试：缺少 global 声明时，赋值是否修改模块级变量", "test_global_write_no_decl": "测试：没有 global 声明时，赋值是否创建局部变量", "test_global_ptr_no_decl": "测试：没有 global 声明时，指针赋值是否创建局部变量", "test_pointer_arithmetic": "测试指针算术：偏移计算是否正确", "test_array_decay": "测试 list[t.CChar, N] 传递给 t.CConst | str 参数", "test_pointer_cast": "测试指针与整数之间的类型转换", "test_pointer_offset": "测试指针偏移：CUInt8T* + N 应该偏移 N 字节，CUInt32T* + N 应该偏移 N*4 字节", "test_struct_pointer": "测试结构体指针的成员访问", "test_memcpy_pointers": "测试 memcpy 在指针之间的数据复制", "test_pointer_param": "测试指针作为函数参数传递"} \ No newline at end of file diff --git a/.transpyc_cache/b436212dcff1be29.pyi b/.transpyc_cache/b436212dcff1be29.pyi new file mode 100644 index 0000000..ab57c0d --- /dev/null +++ b/.transpyc_cache/b436212dcff1be29.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +_g_val: t.CExtern | t.CInt +_g_ptr: t.CExtern | t.CUInt32T | t.CPtr + +def test_global_write() -> t.CInt: pass + +def test_global_write_no_decl() -> t.CInt: pass + +def test_global_ptr() -> t.CInt: pass + +def test_global_ptr_no_decl() -> t.CInt: pass + +def test_pointer_arithmetic() -> t.CInt: pass + +def test_array_decay() -> t.CInt: pass + +def test_pointer_cast() -> t.CInt: pass + +def test_pointer_offset() -> t.CInt: pass + + +class TestStruct: + a: t.CInt + b: t.CInt + c: t.CUInt64T + +def test_struct_pointer() -> t.CInt: pass + +def test_memcpy_pointers() -> t.CInt: pass + +def fill_array(arr: t.CUInt32T | t.CPtr, count: t.CInt, val: t.CUInt32T) -> t.CInt: pass + +def test_pointer_param() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/b547ac4f380bddb6.pyi b/.transpyc_cache/b547ac4f380bddb6.pyi new file mode 100644 index 0000000..e960341 --- /dev/null +++ b/.transpyc_cache/b547ac4f380bddb6.pyi @@ -0,0 +1,166 @@ +""" +Auto-generated Python stub file from ast.tokens.py +Module: ast.tokens +""" + + +import t, c +from stdint import * +import memhub +import string + +class TokenType(t.CEnum): + EndMarker: t.State + NewLine: t.State + Indent: t.State + Dedent: t.State + Name: t.State + Number: t.State + String: t.State + FStringStart: t.State + FStringMiddle: t.State + FStringEnd: t.State + Op: t.State + Nl: t.State + Comment: t.State + Encoding: t.State +class Keyword(t.CEnum): + NotKeyword: t.State + False_: t.State + None_: t.State + True_: t.State + And: t.State + As: t.State + Assert: t.State + Async: t.State + Await: t.State + Break: t.State + Class: t.State + Continue: t.State + Def: t.State + Del: t.State + Elif: t.State + Else: t.State + Except: t.State + Finally: t.State + For: t.State + From: t.State + Global: t.State + If: t.State + Import: t.State + In: t.State + Is: t.State + Lambda: t.State + Nonlocal: t.State + Not: t.State + Or: t.State + Pass: t.State + Raise: t.State + Return: t.State + Try: t.State + While: t.State + With: t.State + Yield: t.State + Match: t.State + Case: t.State + Type: t.State +class TokOp(t.CEnum): + LPar: t.State + RPar: t.State + Lsqb: t.State + Rsqb: t.State + LBrace: t.State + RBrace: t.State + Comma: t.State + Colon: t.State + Dot: t.State + Semi: t.State + At: t.State + Equal: t.State + RArrow: t.State + PlusEq: t.State + MinusEq: t.State + StarEq: t.State + SlashEq: t.State + DSlashEq: t.State + PercentEq: t.State + AtEq: t.State + AmpEq: t.State + VBarEq: t.State + CaretEq: t.State + GtGtEq: t.State + LtLtEq: t.State + StarEqEq: t.State + DSlash: t.State + StarStar: t.State + LtLt: t.State + GtGt: t.State + LessEq: t.State + GreaterEq: t.State + EqEq: t.State + ExclaimEq: t.State + Less: t.State + Greater: t.State + Plus: t.State + Minus: t.State + Star: t.State + Slash: t.State + Percent: t.State + Amp: t.State + VBar: t.State + Caret: t.State + Tilde: t.State + ColonEq: t.State + Ellipsis: t.State + Bang: t.State +class Token: + type: t.CInt + op_subtype: t.CInt + kw_subtype: t.CInt + str_val: str + int_val: t.CInt64T + float_val: t.CDouble + is_float: t.CInt + is_complex: t.CInt + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + next: Token | t.CPtr + def __new__(self: Token, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass +class KwEntry: + name: str + kw_id: t.CInt + next: KwEntry | t.CPtr + def __new__(self: KwEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_kw_head: t.CExtern | KwEntry | t.CPtr +_kw_pool: t.CExtern | memhub.MemManager | t.CPtr + +def _kw_intern(pool: memhub.MemManager | t.CPtr, name: str, kw_id: t.CInt) -> t.CInt: pass + +def _kw_lookup(name: str) -> t.CInt: pass + +def _init_keywords(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + + +class OpEntry: + name: str + op_id: t.CInt + length: t.CInt + next: OpEntry | t.CPtr + def __new__(self: OpEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_op_head: t.CExtern | OpEntry | t.CPtr + +def _op_intern(pool: memhub.MemManager | t.CPtr, name: str, op_id: t.CInt) -> t.CInt: pass + +def _init_operators(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _init_tables(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_token(pool: memhub.MemManager | t.CPtr, ttype: t.CInt, lineno: t.CInt, col_offset: t.CInt) -> Token | t.CPtr: pass + +def token_set_str(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, src: str, start: t.CSizeT, length: t.CSizeT) -> t.CInt: pass + +def token_set_str_literal(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, text: str) -> t.CInt: pass diff --git a/.transpyc_cache/b558d8d8f01f4825.pyi b/.transpyc_cache/b558d8d8f01f4825.pyi new file mode 100644 index 0000000..a326df6 --- /dev/null +++ b/.transpyc_cache/b558d8d8f01f4825.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from requests.py +Module: requests +""" + + +import t, c +from stdint import * +import string +import socket +import memhub +import stdio + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr +_socket_ready: t.CExtern | bool +HTTP_PORT: t.CDefine = 80 +HTTP_BUFSIZ: t.CDefine = 4096 + +class Response: + status_code: INT + text: str + body_len: INT + _raw_buf: str + +def _EnsureSocketInit() -> INT: pass + +def _AllocResponse() -> Response | t.CPtr: pass + +def _ParseUrl(url: str, host_buf: str, host_size: INT, path_buf: str, path_size: INT, port: INT | t.CPtr) -> INT: pass + +def _ParseStatusLine(resp_text: str, out_code: INT | t.CPtr) -> INT: pass + +def _FindBodyStart(resp_text: str, resp_len: INT) -> INT: pass + +def get(url: str) -> Response | t.CPtr: pass + +def post(url: str, data: str, content_type: str) -> Response | t.CPtr: pass + +def release(resp: Response | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/b5a965302cded36b.pyi b/.transpyc_cache/b5a965302cded36b.pyi new file mode 100644 index 0000000..bfdb955 --- /dev/null +++ b/.transpyc_cache/b5a965302cded36b.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from sys.py +Module: sys +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import stdio +import w32.win32base +import w32.win32process +import posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def exit(status: INT) -> VOID | t.CExtern | t.CExport: pass + + +maxsize: t.CDefine = 9223372036854775807 + +def platform() -> str: pass + +def get_pid() -> int: pass + + +_argc: t.CExtern | INT +_argv: t.CExtern | str | t.CPtr + +def _init_argv() -> t.CInt: pass + +def _init_argv_win32() -> t.CInt: pass + +def _init_argv_posix() -> t.CInt: pass + +def argc() -> int: pass + +def argv(index: int) -> str: pass diff --git a/.transpyc_cache/b5d05dbd7fee36ef.doc.json b/.transpyc_cache/b5d05dbd7fee36ef.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/b5d05dbd7fee36ef.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/b5d05dbd7fee36ef.pyi b/.transpyc_cache/b5d05dbd7fee36ef.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/b5d05dbd7fee36ef.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/b69cdee68f223fef.doc.json b/.transpyc_cache/b69cdee68f223fef.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b69cdee68f223fef.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b69cdee68f223fef.pyi b/.transpyc_cache/b69cdee68f223fef.pyi new file mode 100644 index 0000000..3de0cb1 --- /dev/null +++ b/.transpyc_cache/b69cdee68f223fef.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from mbuddytest.py +Module: mbuddytest +""" + + +import t, c +from t import CInt, CPtr, CChar, CUInt8T, CUInt32T, CUInt64T, CSizeT +import viperlib +import viperio +import stdlib +import memhub +import string +import testcheck +import w32.win32process +import w32.win32sync +import w32.win32base + +def _thread_alloc_free(param: t.CVoid | t.CPtr) -> t.CUInt32T: pass + +def mbuddy_main() -> CInt: pass diff --git a/.transpyc_cache/b777c587cc21465a.doc.json b/.transpyc_cache/b777c587cc21465a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b777c587cc21465a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b777c587cc21465a.pyi b/.transpyc_cache/b777c587cc21465a.pyi new file mode 100644 index 0000000..e70167d --- /dev/null +++ b/.transpyc_cache/b777c587cc21465a.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from vectortest.py +Module: vectortest +""" + + +import t, c +from t import CInt, CPtr, CChar, CFloat64T +import viperlib +import stdlib +import vector +import memhub + +POOL_SIZE: t.CDefine = 4096 + +def vector_main() -> CInt: pass diff --git a/.transpyc_cache/b7eeff93aa63e70f.doc.json b/.transpyc_cache/b7eeff93aa63e70f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b7eeff93aa63e70f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b7eeff93aa63e70f.pyi b/.transpyc_cache/b7eeff93aa63e70f.pyi new file mode 100644 index 0000000..807845f --- /dev/null +++ b/.transpyc_cache/b7eeff93aa63e70f.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from namespace_defs.py +Module: namespace_defs +""" + + +import stdio +import t, c + +@t.CVTable +class Widget: + id: t.CInt + def __init__(self: Widget, i: t.CInt) -> t.CInt: pass + def GetId(self: Widget) -> t.CInt: pass + def Render(self: Widget) -> t.CInt: pass +class Gadget(Widget): + extra: t.CInt + def __init__(self: Gadget, i: t.CInt, e: t.CInt) -> t.CInt: pass + def Render(self: Gadget) -> t.CInt: pass + def GetExtra(self: Gadget) -> t.CInt: pass +class PlainStruct: + x: t.CInt + y: t.CInt + def __init__(self: PlainStruct, ax: t.CInt, ay: t.CInt) -> t.CInt: pass + def Sum(self: PlainStruct) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/b8c66c8ff44eb874.pyi b/.transpyc_cache/b8c66c8ff44eb874.pyi new file mode 100644 index 0000000..85bdb8f --- /dev/null +++ b/.transpyc_cache/b8c66c8ff44eb874.pyi @@ -0,0 +1,46 @@ +""" +Auto-generated Python stub file from hashtable.py +Module: hashtable +""" + + +import t, c +import string +import memhub +from stdint import * + +_HT_EMPTY: t.CDefine = 0 +_HT_TOMBSTONE: t.CDefine = 1 +_HT_INIT_CAP: t.CDefine = 16 +_HT_SLOT_SIZE: t.CDefine = 24 +_FNV_OFFSET: t.CExtern | t.CUInt64T +_FNV_PRIME: t.CExtern | t.CUInt64T + +def _fnv1a_hash(key: str) -> t.CUInt64T: pass + + +class HashTable: + __slots__: t.CVoid | t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 16 + __tombstones__: t.CSizeT = 0 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def _slot_hash(self: HashTable, idx: t.CSizeT) -> t.CUInt64T: pass + def _slot_key(self: HashTable, idx: t.CSizeT) -> str: pass + def _slot_value(self: HashTable, idx: t.CSizeT) -> t.CPtr: pass + def _set_slot(self: HashTable, idx: t.CSizeT, h: t.CUInt64T, key: str, val: t.CPtr) -> t.CInt: pass + def _find_slot(self: HashTable, key: str, h: t.CUInt64T) -> t.CSizeT: pass + def _resize(self: HashTable, new_cap: t.CSizeT) -> t.CInt: pass + def __setitem__(self: HashTable, key: str, val: t.CNeedPtr) -> t.CInt: pass + def __getitem__(self: HashTable, key: str) -> t.CPtr: pass + def __contains__(self: HashTable, key: str) -> t.CInt: pass + def __delitem__(self: HashTable, key: str) -> t.CInt: pass + def __len__(self: HashTable) -> t.CSizeT: pass + def __iter__(self: HashTable) -> HashTable | t.CPtr: pass + def __next__(self: HashTable) -> str: pass + def get(self: HashTable, key: str, default: t.CPtr) -> t.CPtr: pass + def set_int(self: HashTable, key: str, val: int) -> t.CInt: pass + def set_str(self: HashTable, key: str, val: str) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/b8cbe044a10adcca.doc.json b/.transpyc_cache/b8cbe044a10adcca.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/b8cbe044a10adcca.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/b8cbe044a10adcca.pyi b/.transpyc_cache/b8cbe044a10adcca.pyi new file mode 100644 index 0000000..7809327 --- /dev/null +++ b/.transpyc_cache/b8cbe044a10adcca.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from eq_test.py +Module: eq_test +""" + + +import stdio +import t, c +import string + +def test_int_eq() -> t.CInt: pass + +def test_int_cmp() -> t.CInt: pass + +def test_ptr_is_none() -> t.CInt: pass + +def test_ptr_assign_none() -> t.CInt: pass + +def test_ptr_eq_none() -> t.CInt: pass + +def test_bool_logic() -> t.CInt: pass + +def eq_test() -> int: pass diff --git a/.transpyc_cache/bbdf3bbd4c3bc28c.pyi b/.transpyc_cache/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/.transpyc_cache/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/.transpyc_cache/bc52b7a79738d25a.doc.json b/.transpyc_cache/bc52b7a79738d25a.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/bc52b7a79738d25a.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/bc52b7a79738d25a.pyi b/.transpyc_cache/bc52b7a79738d25a.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/bc52b7a79738d25a.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/bd82e3ca81546c76.doc.json b/.transpyc_cache/bd82e3ca81546c76.doc.json new file mode 100644 index 0000000..0858afd --- /dev/null +++ b/.transpyc_cache/bd82e3ca81546c76.doc.json @@ -0,0 +1 @@ +{"FileEntry": "扫描到的文件条目", "ScanResult": "扫描结果集合", "create_scan_result": "创建扫描结果容器，预分配 MAX_FILES 个槽位", "add_file_entry": "添加文件条目到扫描结果，返回 0 成功", "compute_file_sha1": "读取文件内容并计算 SHA1 前16字符（CRLF→LF 后计算，与 Projectrans.py 一致）", "scan_directory_recursive": "递归扫描目录，收集 .py 文件", "scan_includes": "扫描 includes 目录，返回所有 .py 文件的扫描结果", "find_entry_by_module": "按模块名查找文件条目"} \ No newline at end of file diff --git a/.transpyc_cache/bd82e3ca81546c76.pyi b/.transpyc_cache/bd82e3ca81546c76.pyi new file mode 100644 index 0000000..ba0eee6 --- /dev/null +++ b/.transpyc_cache/bd82e3ca81546c76.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from lib.core.IncludesScanner.py +Module: lib.core.IncludesScanner +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import hashlib +import viperlib + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_PATH_LEN: t.CDefine = 512 +MAX_FILES: t.CDefine = 256 + +@t.NoVTable +class FileEntry: + Path: str + Sha1: str + RelPath: str + ModuleName: str +@t.NoVTable +class ScanResult: + Entries: FileEntry | t.CPtr + Count: t.CInt + Capacity: t.CInt + +def create_scan_result(pool: memhub.MemBuddy | t.CPtr) -> ScanResult | t.CPtr: pass + +def add_file_entry(result: ScanResult | t.CPtr, pool: memhub.MemBuddy | t.CPtr, abs_path: str, rel_path: str, sha1: str) -> int: pass + +def compute_file_sha1(pool: memhub.MemBuddy | t.CPtr, file_path: str) -> str: pass + +def scan_directory_recursive(pool: memhub.MemBuddy | t.CPtr, root_dir: str, rel_prefix: str, result: ScanResult | t.CPtr) -> int: pass + +def scan_includes(pool: memhub.MemBuddy | t.CPtr, includes_dir: str) -> ScanResult | t.CPtr: pass + +def find_entry_by_module(result: ScanResult | t.CPtr, module_name: str) -> FileEntry | t.CPtr: pass diff --git a/.transpyc_cache/bdfc8bd243b7e5b7.doc.json b/.transpyc_cache/bdfc8bd243b7e5b7.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/bdfc8bd243b7e5b7.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/bdfc8bd243b7e5b7.pyi b/.transpyc_cache/bdfc8bd243b7e5b7.pyi new file mode 100644 index 0000000..bd232f2 --- /dev/null +++ b/.transpyc_cache/bdfc8bd243b7e5b7.pyi @@ -0,0 +1,87 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import c +import testcheck + +def square[T](x: T) -> T: pass + +def test_generic_square_types() -> t.CInt: pass + +def clamp[T](val: T, lo: T, hi: T) -> T: pass + +def test_generic_clamp() -> t.CInt: pass + + +class Stack[T]: + data: t.CArray[t.CInt, 16] + top: t.CInt + def __init__(self: Stack) -> t.CInt: pass + def push(self: Stack, val: T) -> t.CInt: pass + def pop(self: Stack) -> T: pass + def is_empty(self: Stack) -> t.CInt: pass + def size_of(self: Stack) -> t.CSizeT: pass + def type_size(self: Stack) -> t.CSizeT: pass + +def test_generic_stack() -> t.CInt: pass + + +class IntPair: + a: t.CInt + b: t.CInt + +def generic_swap[T](p: IntPair | t.CPtr) -> t.CInt: pass + +def test_generic_swap() -> t.CInt: pass + + +class Range[T]: + start: T + end_val: T + step: T + def __init__(self: Range, s: T, e: T, st: T) -> t.CInt: pass + def sum_all(self: Range) -> T: pass + +def test_generic_range() -> t.CInt: pass + +def array_sum(arr: t.CArray[t.CInt, 8], n: t.CInt) -> t.CInt: pass + +def test_generic_array_sum() -> t.CInt: pass + +def max_of_three[T](a: T, b: T, c_val: T) -> T: pass + +def test_generic_max_three() -> t.CInt: pass + + +class Container[T]: + values: t.CArray[t.CInt, 8] + size: t.CInt + def __init__(self: Container) -> t.CInt: pass + def add(self: Container, v: T) -> t.CInt: pass + def get(self: Container, idx: t.CInt) -> T: pass + def contains(self: Container, v: T) -> t.CInt: pass + def total(self: Container) -> T: pass + +def test_generic_container() -> t.CInt: pass + +def factorial[T](n: T) -> T: pass + +def test_generic_factorial() -> t.CInt: pass + + +class Accumulator[T]: + total: T + def __init__(self: Accumulator, start: T) -> t.CInt: pass + def add_val(self: Accumulator, v: T) -> t.CInt: pass + def get_result(self: Accumulator) -> T: pass + +def test_generic_accumulator() -> t.CInt: pass + +def test_generic_sizeof_bug() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/bf38ca6c448d2d33.doc.json b/.transpyc_cache/bf38ca6c448d2d33.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/bf38ca6c448d2d33.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/bf38ca6c448d2d33.pyi b/.transpyc_cache/bf38ca6c448d2d33.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/bf38ca6c448d2d33.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/bf9813f6e3c9d351.pyi b/.transpyc_cache/bf9813f6e3c9d351.pyi new file mode 100644 index 0000000..00d2f9f --- /dev/null +++ b/.transpyc_cache/bf9813f6e3c9d351.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from vector.py +Module: vector +""" + + +import t, c +from stdint import * +import memhub + +class Vector[T]: + data: t.CVoid | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + elem_size: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: Vector, pool: memhub.MemManager | t.CPtr, capacity: t.CSizeT, _hint: T) -> t.CInt: pass + def push(self: Vector, value: T) -> t.CInt: pass + def _grow(self: Vector) -> t.CInt: pass + def get(self: Vector, index: t.CSizeT) -> T: pass + def set(self: Vector, index: t.CSizeT, value: T) -> t.CInt: pass + def len(self: Vector) -> t.CSizeT: pass + def clear(self: Vector) -> t.CInt: pass + def free(self: Vector) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/c01b27dbcd683245.doc.json b/.transpyc_cache/c01b27dbcd683245.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/c01b27dbcd683245.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/c01b27dbcd683245.pyi b/.transpyc_cache/c01b27dbcd683245.pyi new file mode 100644 index 0000000..fdb53f8 --- /dev/null +++ b/.transpyc_cache/c01b27dbcd683245.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from fileiotest.py +Module: fileiotest +""" + + +import t, c +from t import CInt, CInt32T, CUInt32T, CPtr, CChar, CExport +from stdint import * +import stdlib +import viperlib +import w32.win32base +import w32.win32file +import w32.win32console +import w32.fileio +from w32.fileio import File, FileW, MODE, FRESULT, SEEK_SET, SEEK_CUR, SEEK_END + +BUF_SIZE: t.CDefine = 4096 + +def fileio_main() -> t.CInt: pass + +def fileio_with_test() -> t.CInt: pass + +def fileio_handle_test() -> t.CInt: pass + +def fileio_rw_test() -> t.CInt: pass + +def fileio_wide_test() -> t.CInt: pass + +def fileio_chinese_test() -> t.CInt: pass diff --git a/.transpyc_cache/c1416dfae155d6e2.doc.json b/.transpyc_cache/c1416dfae155d6e2.doc.json new file mode 100644 index 0000000..4cc35ca --- /dev/null +++ b/.transpyc_cache/c1416dfae155d6e2.doc.json @@ -0,0 +1 @@ +{"test_snprintf_u8": "测试 CUInt8T 值通过 t.CUnsignedInt() 传递给 %u", "test_snprintf_u16": "测试 CUInt16T 值通过 t.CUnsignedInt() 传递给 %u", "test_snprintf_u64": "测试 CUInt64T 值传递给 %lu", "test_snprintf_d_neg": "测试 %d 格式化负整数", "test_snprintf_mixed": "模拟 desktop.py 的时间格式化", "test_printf_variadic": "测试 printf 直接输出 variadic 参数", "test_snprintf_multi_lu": "测试多个 %lu 连续（模拟 perf 日志格式）", "test_snprintf_u8_edge": "测试 CUInt8T 边界值"} \ No newline at end of file diff --git a/.transpyc_cache/c1416dfae155d6e2.pyi b/.transpyc_cache/c1416dfae155d6e2.pyi new file mode 100644 index 0000000..636e8b0 --- /dev/null +++ b/.transpyc_cache/c1416dfae155d6e2.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import testcheck +import string +import viperlib +import w32.win32console + +def test_snprintf_u8() -> t.CInt: pass + +def test_snprintf_u16() -> t.CInt: pass + +def test_snprintf_u64() -> t.CInt: pass + +def test_snprintf_d_neg() -> t.CInt: pass + +def test_snprintf_d_pos() -> t.CInt: pass + +def test_snprintf_x() -> t.CInt: pass + +def test_snprintf_mixed() -> t.CInt: pass + +def test_printf_variadic() -> t.CInt: pass + +def test_snprintf_multi_lu() -> t.CInt: pass + +def test_snprintf_zero() -> t.CInt: pass + +def test_snprintf_large() -> t.CInt: pass + +def test_snprintf_u8_edge() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/c24f25b14b4bbd0a.pyi b/.transpyc_cache/c24f25b14b4bbd0a.pyi new file mode 100644 index 0000000..a7c0c1f --- /dev/null +++ b/.transpyc_cache/c24f25b14b4bbd0a.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from vipersimd.py +Module: vipersimd +""" + + +import t, c + +def simd_add4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sub4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_div4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sqrt4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_neg4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_abs4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_max4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_min4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_hsum4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_dot4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_fma4d(a: t.CPtr, b: t.CPtr, c_val: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sub4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_div4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_sub_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_div_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_abs_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_neg_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_dot_array(a: t.CPtr, b: t.CPtr, n: t.CSizeT, out: t.CPtr) -> t.CVoid: pass + +def simd_sqrt(x: t.CDouble) -> t.CDouble: pass + +def simd_fabs(x: t.CDouble) -> t.CDouble: pass + +def simd_floor(x: t.CDouble) -> t.CDouble: pass + +def simd_ceil(x: t.CDouble) -> t.CDouble: pass + +def simd_round(x: t.CDouble) -> t.CDouble: pass + +def simd_trunc(x: t.CDouble) -> t.CDouble: pass + +def simd_fma(a: t.CDouble, b: t.CDouble, c_val: t.CDouble) -> t.CDouble: pass + +def simd_copysign(mag: t.CDouble, sign: t.CDouble) -> t.CDouble: pass + +def simd_neg(x: t.CDouble) -> t.CDouble: pass + +def simd_minnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def simd_maxnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def simd_exp(x: t.CDouble) -> t.CDouble: pass + +def simd_log(x: t.CDouble) -> t.CDouble: pass + +def simd_sin(x: t.CDouble) -> t.CDouble: pass + +def simd_cos(x: t.CDouble) -> t.CDouble: pass + +def simd_pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass diff --git a/.transpyc_cache/c2ed034f4bfaad51.doc.json b/.transpyc_cache/c2ed034f4bfaad51.doc.json new file mode 100644 index 0000000..f8bc59e --- /dev/null +++ b/.transpyc_cache/c2ed034f4bfaad51.doc.json @@ -0,0 +1 @@ +{"add_imported_module": "记录已导入的模块名，返回新的 imported_modules 字符串指针", "is_module_imported": "检查模块是否已导入（词边界精确匹配，避免子串误匹配）", "add_from_import": "添加 from-import 映射 \"name:module\"，返回新的 from_imports 字符串", "lookup_from_import": "查找 from-import 名称，返回模块名或 None\n\n 支持 star import: 如果 from_imports 中有 \"*:module\" 条目，\n 且未找到精确名称匹配，则返回 star import 的模块名。\n ", "ImportsHandle": "导入语句处理器：继承 Mixin 获得 Trans 回指针", "ImportsHandle.HandleImport": "处理 import 语句，返回 0", "ImportsHandle.HandleImportFromModule": "处理 from X import Y 的模块部分，更新 trans._imported_modules", "ImportsHandle.HandleImportFromNames": "处理 from X import Y 的名称部分，更新 trans._from_imports"} \ No newline at end of file diff --git a/.transpyc_cache/c2ed034f4bfaad51.pyi b/.transpyc_cache/c2ed034f4bfaad51.pyi new file mode 100644 index 0000000..7abd965 --- /dev/null +++ b/.transpyc_cache/c2ed034f4bfaad51.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesImports.py +Module: lib.core.Handles.HandlesImports +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, imported_modules: str, name: str) -> str: pass + +def is_module_imported(imported_modules: str, name: str) -> int: pass + +def add_from_import(pool: memhub.MemBuddy | t.CPtr, from_imports: str, local_name: str, module_name: str) -> str: pass + +def lookup_from_import(from_imports: str, name: str) -> str: pass + + +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + def __init__(self: ImportsHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleImport(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromModule(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromNames(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: pass diff --git a/.transpyc_cache/c2f78fe75716e993.doc.json b/.transpyc_cache/c2f78fe75716e993.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/c2f78fe75716e993.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/c2f78fe75716e993.pyi b/.transpyc_cache/c2f78fe75716e993.pyi new file mode 100644 index 0000000..c495a8e --- /dev/null +++ b/.transpyc_cache/c2f78fe75716e993.pyi @@ -0,0 +1,84 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +import stdio + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/c325121077fc50d9.doc.json b/.transpyc_cache/c325121077fc50d9.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/c325121077fc50d9.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/c325121077fc50d9.pyi b/.transpyc_cache/c325121077fc50d9.pyi new file mode 100644 index 0000000..2294a70 --- /dev/null +++ b/.transpyc_cache/c325121077fc50d9.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/c39f3e540adb7b4e.doc.json b/.transpyc_cache/c39f3e540adb7b4e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/c39f3e540adb7b4e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/c39f3e540adb7b4e.pyi b/.transpyc_cache/c39f3e540adb7b4e.pyi new file mode 100644 index 0000000..7862af7 --- /dev/null +++ b/.transpyc_cache/c39f3e540adb7b4e.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from llvmir_test.py +Module: llvmir_test +""" + + +import t, c +from stdint import * +import stdio + +def test_add() -> t.CInt: pass + +def test_sub() -> t.CInt: pass + +def test_mul() -> t.CInt: pass + +def test_sdiv() -> t.CInt: pass + +def test_and() -> t.CInt: pass + +def test_or() -> t.CInt: pass + +def test_xor() -> t.CInt: pass + +def test_shl() -> t.CInt: pass + +def test_lout() -> t.CInt: pass + +def test_icmp_eq() -> t.CInt: pass + +def test_icmp_ne() -> t.CInt: pass + +def test_icmp_slt() -> t.CInt: pass + +def test_srem() -> t.CInt: pass + +def test_lshr() -> t.CInt: pass + +def test_nested() -> t.CInt: pass + +def llvmir_test() -> t.CInt: pass diff --git a/.transpyc_cache/c3a6aed1f1fb8b1e.pyi b/.transpyc_cache/c3a6aed1f1fb8b1e.pyi new file mode 100644 index 0000000..68a99b1 --- /dev/null +++ b/.transpyc_cache/c3a6aed1f1fb8b1e.pyi @@ -0,0 +1,240 @@ +""" +Auto-generated Python stub file from numpy.__init__.py +Module: numpy.__init__ +""" + + +from stdint import * +import stdlib +import string +import vipermath +import stdio +import t, c +import memhub + +MAX_NDIM: t.CDefine = 4 +float64: t.CTypedef = t.CDouble +float32: t.CTypedef = t.CFloat +int64: t.CTypedef = t.CLong +int32: t.CTypedef = t.CInt +uint8: t.CTypedef = t.CUnsignedChar +pi: t.CDefine = 3.14159265358979323846 +e: t.CDefine = 2.71828182845904523536 + +@t.Object +class ndarray[T]: + data: T | t.CPtr + shape: t.CArray[t.CSizeT, MAX_NDIM] + strides: t.CArray[t.CSizeT, MAX_NDIM] + ndim: t.CInt + size: t.CSizeT + owns_data: t.CInt + pool: memhub.MemManager | t.CPtr + def __new__(self: ndarray, pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> t.CInt: pass + def __add__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __sub__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mul__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __truediv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __floordiv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mod__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __neg__(self: ndarray) -> ndarray[T] | t.CPtr: pass + def __len__(self: ndarray) -> t.CInt: pass + def at2d(self: ndarray, row: t.CSizeT, col: t.CSizeT) -> T: pass + def set2d(self: ndarray, row: t.CSizeT, col: t.CSizeT, val: T) -> t.CInt: pass + def delete(self: ndarray) -> t.CInt: pass + def fill(self: ndarray, val: T) -> t.CInt: pass + def copy(self: ndarray) -> ndarray[T] | t.CPtr: pass + def reshape(self: ndarray, new_shape: INTPTR, new_ndim: t.CInt) -> ndarray[T] | t.CPtr: pass + def sum(self: ndarray) -> T: pass + def mean(self: ndarray) -> T: pass + def min(self: ndarray) -> T: pass + def max(self: ndarray) -> T: pass + def argmax(self: ndarray) -> t.CInt: pass + def argmin(self: ndarray) -> t.CInt: pass + def dot(self: ndarray, other: ndarray[T] | t.CPtr) -> T: pass + def T(self: ndarray) -> ndarray[T] | t.CPtr: pass + def print_arr(self: ndarray) -> t.CInt: pass + +Float64Array: t.CTypedef = ndarray[t.CDouble] +Float32Array: t.CTypedef = ndarray[t.CFloat] +Int64Array: t.CTypedef = ndarray[t.CLong] +Int32Array: t.CTypedef = ndarray[t.CInt] +Uint8Array: t.CTypedef = ndarray[t.CUnsignedChar] + +def _alloc_ndarray(pool: memhub.MemManager | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def _compute_strides(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def _empty_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr, data: t.CDouble | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full(pool: memhub.MemManager | t.CPtr, n: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, step: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, num: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def eye(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def diag(pool: memhub.MemManager | t.CPtr, vals: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_abs(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sqrt(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_exp(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_pow(a: ndarray[t.CDouble] | t.CPtr, p: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def add_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def mul_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def sub_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def div_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def matmul(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def dot_product(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_sum(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_mean(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_min(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_max(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_argmax(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_argmin(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def var(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def std(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def norm(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def clip(a: ndarray[t.CDouble] | t.CPtr, lo: t.CDouble, hi: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def concatenate(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def sort_arr(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def reverse(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log10(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_floor(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_ceil(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_round(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sign(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tanh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sinh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cosh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arcsin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arccos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan2(y: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_degrees(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_radians(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isnan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isinf(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_maximum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_minimum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cumsum(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def diff(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def flatten(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def trace(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def outer(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_where(condition: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_count_nonzero(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_all(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_any(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_not_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange1(pool: memhub.MemManager | t.CPtr, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace2(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def meshgrid(x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def det2x2(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def inv2x2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cross3(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_interp(x: t.CDouble, xp: ndarray[t.CDouble] | t.CPtr, fp: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def prod(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def median(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def percentile(a: ndarray[t.CDouble] | t.CPtr, q: t.CDouble) -> t.CDouble: pass diff --git a/.transpyc_cache/c3b259b4059f8668.pyi b/.transpyc_cache/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/.transpyc_cache/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/c4960bf577b21f14.doc.json b/.transpyc_cache/c4960bf577b21f14.doc.json new file mode 100644 index 0000000..c90d338 --- /dev/null +++ b/.transpyc_cache/c4960bf577b21f14.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "_ScanDirForPyFiles": "递归扫描目录，收集 .py 文件到 entries，返回新的 file_count", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/c4960bf577b21f14.pyi b/.transpyc_cache/c4960bf577b21f14.pyi new file mode 100644 index 0000000..819d958 --- /dev/null +++ b/.transpyc_cache/c4960bf577b21f14.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 64 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def _ScanDirForPyFiles(mb: memhub.MemBuddy | t.CPtr, dir_path: str, entries: SrcFileEntry | t.CPtr, entry_size: t.CSizeT, file_count: int, max_files: int) -> int: pass + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/c4f2d483ec0f4db5.doc.json b/.transpyc_cache/c4f2d483ec0f4db5.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/c4f2d483ec0f4db5.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/c4f2d483ec0f4db5.pyi b/.transpyc_cache/c4f2d483ec0f4db5.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/c4f2d483ec0f4db5.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/c688cec0f891dd97.doc.json b/.transpyc_cache/c688cec0f891dd97.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/.transpyc_cache/c688cec0f891dd97.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/c688cec0f891dd97.pyi b/.transpyc_cache/c688cec0f891dd97.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/c688cec0f891dd97.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/c70684ccfe48a5a9.doc.json b/.transpyc_cache/c70684ccfe48a5a9.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/c70684ccfe48a5a9.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/c70684ccfe48a5a9.pyi b/.transpyc_cache/c70684ccfe48a5a9.pyi new file mode 100644 index 0000000..f6f33f4 --- /dev/null +++ b/.transpyc_cache/c70684ccfe48a5a9.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import memhub +import stdlib +import sys +import os +import os._win32 +import os._posix +import os.path +import shutil +import subprocess +import syslibtest_impl + +ARENA_SIZE: t.CDefine = 1048576 + +def main() -> CInt | CExport: pass diff --git a/.transpyc_cache/c7b66e67f49c5c06.doc.json b/.transpyc_cache/c7b66e67f49c5c06.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/c7b66e67f49c5c06.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/c7b66e67f49c5c06.pyi b/.transpyc_cache/c7b66e67f49c5c06.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/c7b66e67f49c5c06.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/c7ef33270dce36e3.doc.json b/.transpyc_cache/c7ef33270dce36e3.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/.transpyc_cache/c7ef33270dce36e3.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/.transpyc_cache/c7ef33270dce36e3.pyi b/.transpyc_cache/c7ef33270dce36e3.pyi new file mode 100644 index 0000000..2294a70 --- /dev/null +++ b/.transpyc_cache/c7ef33270dce36e3.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/.transpyc_cache/c9d54a4158f7f5a8.pyi b/.transpyc_cache/c9d54a4158f7f5a8.pyi new file mode 100644 index 0000000..3e2d738 --- /dev/null +++ b/.transpyc_cache/c9d54a4158f7f5a8.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__sha256.py +Module: hashlib.__sha256 +""" + + +import t, c + +SHA256_BLOCK_LEN: t.CDefine = 64 +SHA256_DIGEST_LEN: t.CDefine = 32 +sha256_K: t.CExtern | t.CArray[t.CUInt32T, 64] + +def s0(x: t.CUInt32T) -> t.CUInt32T: pass + +def s1(x: t.CUInt32T) -> t.CUInt32T: pass + +def S0(x: t.CUInt32T) -> t.CUInt32T: pass + +def S1(x: t.CUInt32T) -> t.CUInt32T: pass + +def Ch(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def Maj(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha256: + state: t.CArray[t.CUInt32T, 8] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA256_BLOCK_LEN] + def __init__(self: sha256) -> t.CInt: pass + def transform(self: sha256, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha256, s: str) -> t.CInt: pass + def final(self: sha256, out: t.CArray[t.CUInt8T, SHA256_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/c9f4be41ca1cc2b4.pyi b/.transpyc_cache/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/.transpyc_cache/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/cb151b297f4ea56d.doc.json b/.transpyc_cache/cb151b297f4ea56d.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/cb151b297f4ea56d.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/cb151b297f4ea56d.pyi b/.transpyc_cache/cb151b297f4ea56d.pyi new file mode 100644 index 0000000..125a3b3 --- /dev/null +++ b/.transpyc_cache/cb151b297f4ea56d.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from __zlib.zchecksum.py +Module: __zlib.zchecksum +""" + + +from stdint import * +import t, c + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: pass + + +crc32_table: t.CExtern | t.CArray[t.CUInt32T, 256] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: pass diff --git a/.transpyc_cache/cc3ad7f3bc880326.doc.json b/.transpyc_cache/cc3ad7f3bc880326.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/cc3ad7f3bc880326.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/cc3ad7f3bc880326.pyi b/.transpyc_cache/cc3ad7f3bc880326.pyi new file mode 100644 index 0000000..47bbff6 --- /dev/null +++ b/.transpyc_cache/cc3ad7f3bc880326.pyi @@ -0,0 +1,12 @@ +""" +Auto-generated Python stub file from virtual_dispatch_test.py +Module: virtual_dispatch_test +""" + + +import stdio +import t, c +import string +from inherit_test import Animal, Dog + +def virtual_dispatch_test() -> int: pass diff --git a/.transpyc_cache/cc436a125bbaf28c.pyi b/.transpyc_cache/cc436a125bbaf28c.pyi new file mode 100644 index 0000000..a905387 --- /dev/null +++ b/.transpyc_cache/cc436a125bbaf28c.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__md5.py +Module: hashlib.__md5 +""" + + +import t, c + +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + + +md5_T: t.CExtern | t.CArray[t.CUInt32T, 64] +md5_S: t.CExtern | t.CArray[t.CInt, 64] + +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] + def __init__(self: md5) -> t.CInt: pass + def transform(self: md5, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: md5, s: str) -> t.CInt: pass + def final(self: md5, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/cdff0d7b58fc935b.doc.json b/.transpyc_cache/cdff0d7b58fc935b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/cdff0d7b58fc935b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/cdff0d7b58fc935b.pyi b/.transpyc_cache/cdff0d7b58fc935b.pyi new file mode 100644 index 0000000..0fb8976 --- /dev/null +++ b/.transpyc_cache/cdff0d7b58fc935b.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import ctraits +import stdio +import testcheck + +@t.CExport +def main() -> int: pass diff --git a/.transpyc_cache/cfc83b830141b00e.doc.json b/.transpyc_cache/cfc83b830141b00e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/cfc83b830141b00e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/cfc83b830141b00e.pyi b/.transpyc_cache/cfc83b830141b00e.pyi new file mode 100644 index 0000000..5af877f --- /dev/null +++ b/.transpyc_cache/cfc83b830141b00e.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from zc.config.py +Module: zc.config +""" + + +import t, c + +POSMAX: t.CDefine = 100 +MAX_RETRY: t.CDefine = 5 +BUFFER_SIZE: t.CDefine = 256 +PI: t.CDefine = 3.14159 +EULER: t.CDefine = 2.71828 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +NULL: t.CDefine = 0 +MATH_OFFSET: t.CDefine = 10 +MATH_SCALE: t.CDefine = 2 +THRESHOLD: t.CDefine = 50 +LOOP_LIMIT: t.CDefine = 20 +SHIFT_AMOUNT: t.CDefine = 2 +MASK_VALUE: t.CDefine = 0xFF +BIT_RANGE: t.CDefine = 4 +EXPONENT_BASE: t.CDefine = 3 +MODULO_DIVISOR: t.CDefine = 7 +SCALE_SHIFT: t.CDefine = 4 +HALF_SCALE: t.CDefine = 0.5 \ No newline at end of file diff --git a/.transpyc_cache/d152115b49e50218.pyi b/.transpyc_cache/d152115b49e50218.pyi new file mode 100644 index 0000000..1e7a152 --- /dev/null +++ b/.transpyc_cache/d152115b49e50218.pyi @@ -0,0 +1,42 @@ +""" +Auto-generated Python stub file from os._posix.py +Module: os._posix +""" + + +import t, c +from stdint import * +import posix +import memhub +import stdlib + +MAX_PATH_LEN: t.CDefine = 4096 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def _stat(path: str) -> posix.Stat | t.CPtr: pass + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd_impl() -> str: pass + +def chdir_impl(path: str) -> int: pass + +def mkdir_impl(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir_impl(path: str) -> int: pass + +def rename_impl(old_path: str, new_path: str) -> int: pass + +def chmod_impl(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system_impl(cmd: str) -> int: pass diff --git a/.transpyc_cache/d1599ecbac5b6586.doc.json b/.transpyc_cache/d1599ecbac5b6586.doc.json new file mode 100644 index 0000000..a6b0616 --- /dev/null +++ b/.transpyc_cache/d1599ecbac5b6586.doc.json @@ -0,0 +1 @@ +{"TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败\n\n current_package: 当前文件所属包名（用于解析相对导入），None 表示顶级模块\n ", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/d1599ecbac5b6586.pyi b/.transpyc_cache/d1599ecbac5b6586.pyi new file mode 100644 index 0000000..c155594 --- /dev/null +++ b/.transpyc_cache/d1599ecbac5b6586.pyi @@ -0,0 +1,51 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesImports as HandlesImports + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str, current_package: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/d28c16ea0550c4a4.doc.json b/.transpyc_cache/d28c16ea0550c4a4.doc.json new file mode 100644 index 0000000..dda4745 --- /dev/null +++ b/.transpyc_cache/d28c16ea0550c4a4.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/.transpyc_cache/d28c16ea0550c4a4.pyi b/.transpyc_cache/d28c16ea0550c4a4.pyi new file mode 100644 index 0000000..60fab85 --- /dev/null +++ b/.transpyc_cache/d28c16ea0550c4a4.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/.transpyc_cache/d33c3f139167d275.doc.json b/.transpyc_cache/d33c3f139167d275.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/d33c3f139167d275.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/d33c3f139167d275.pyi b/.transpyc_cache/d33c3f139167d275.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/d33c3f139167d275.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/d358712e53b4da98.doc.json b/.transpyc_cache/d358712e53b4da98.doc.json new file mode 100644 index 0000000..ac292ab --- /dev/null +++ b/.transpyc_cache/d358712e53b4da98.doc.json @@ -0,0 +1 @@ +{"ForHandle": "for 循环语句处理器：继承 Mixin 获得 Trans 回指针", "ForHandle.Handle": "翻译 for i in range(...) 循环语句", "ForHandle._handle_ptr_iter": "指针迭代: for x in ptr: 直到解引用为空"} \ No newline at end of file diff --git a/.transpyc_cache/d358712e53b4da98.pyi b/.transpyc_cache/d358712e53b4da98.pyi new file mode 100644 index 0000000..08527b9 --- /dev/null +++ b/.transpyc_cache/d358712e53b4da98.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFor.py +Module: lib.core.Handles.HandlesFor +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class ForHandle(HandlesBase.Mixin): + def __init__(self: ForHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ForHandle, node: ast.AST | t.CPtr) -> int: pass + def _handle_ptr_iter(self: ForHandle, for_node: ast.For | t.CPtr, var_name: str) -> int: pass + +def NewForHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ForHandle | t.CPtr: pass diff --git a/.transpyc_cache/d4117f5e00a6c3b5.pyi b/.transpyc_cache/d4117f5e00a6c3b5.pyi new file mode 100644 index 0000000..ee44bdb --- /dev/null +++ b/.transpyc_cache/d4117f5e00a6c3b5.pyi @@ -0,0 +1,62 @@ +""" +Auto-generated Python stub file from llvmlite.__verify.py +Module: llvmlite.__verify +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__function import Function, BasicBlock, Line, Param +from .__module import LLVMModule + +VERIFY_OK: t.CDefine = 0 +VERIFY_ERR_NO_TERMINATOR: t.CDefine = 1 +VERIFY_ERR_DUPLICATE_DEF: t.CDefine = 2 +VERIFY_ERR_UNDEFINED_USE: t.CDefine = 3 +VERIFY_ERR_PHI_NOT_FIRST: t.CDefine = 4 + +class NameEntry(GSListNode[NameEntry]): + Name: t.CChar | t.CPtr +class VerifyResult: + ErrorCount: t.CInt + WarningCount: t.CInt + ErrorCode: t.CInt + ErrorBuf: t.CChar | t.CPtr + ErrorBufSize: t.CSizeT + +def _new_verify_result(pool: memhub.MemBuddy | t.CPtr, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + +def _report_error(result: VerifyResult | t.CPtr, code: t.CInt, msg: t.CChar | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _new_name_table(pool: memhub.MemBuddy | t.CPtr) -> GSList[NameEntry] | t.CPtr: pass + + +NENTRY_SIZE: t.CDefine = 16 + +def _name_table_add(table: GSList[NameEntry] | t.CPtr, pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_table_contains(table: GSList[NameEntry] | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_char_name(ch: t.CChar) -> t.CInt: pass + +def _extract_def_name(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _is_terminator(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_phi(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _check_uses_in_line(table: GSList[NameEntry] | t.CPtr, text: t.CChar | t.CPtr, def_name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, table: GSList[NameEntry] | t.CPtr) -> t.CInt: pass + +def _str_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, target: t.CChar | t.CPtr) -> t.CInt: pass + +def verify_function(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + +def verify_module(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/d541ade6afb689a5.pyi b/.transpyc_cache/d541ade6afb689a5.pyi new file mode 100644 index 0000000..fd4ddf5 --- /dev/null +++ b/.transpyc_cache/d541ade6afb689a5.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from zlib.zhuff.py +Module: zlib.zhuff +""" + + +from stdint import * +import zlib.zdef as zdef +import memhub +import t, c + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt +@t.Object +class zhuff_tree: + pool: memhub.MemManager | t.CPtr + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + def __init__(self: zhuff_tree) -> t.CInt: pass + def build_codes(self: zhuff_tree, freqs: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def build_fixed_lit_tree(self: zhuff_tree) -> t.CInt: pass + def build_fixed_dist_tree(self: zhuff_tree) -> t.CInt: pass + def encode_symbol(self: zhuff_tree, symbol: int, writer: zdef.zbit_writer | t.CPtr) -> t.CInt: pass + def build_code_lengths(self: zhuff_tree, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def get_fixed_lit_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def get_fixed_dist_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def build_tree_from_lengths(self: zhuff_tree, lengths: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt +@t.Object +class zhuff_decode_tree: + pool: memhub.MemManager | t.CPtr + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + def __init__(self: zhuff_decode_tree) -> t.CInt: pass + def build_decode_tree(self: zhuff_decode_tree, ht: zhuff_tree | t.CPtr) -> t.CInt: pass + def decode_symbol(self: zhuff_decode_tree, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/d63d2c2a2bd9bd2a.pyi b/.transpyc_cache/d63d2c2a2bd9bd2a.pyi new file mode 100644 index 0000000..cc9d917 --- /dev/null +++ b/.transpyc_cache/d63d2c2a2bd9bd2a.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from zlib.__init__.py +Module: zlib.__init__ +""" + +import t +import c + + +from .pyzlib import * \ No newline at end of file diff --git a/.transpyc_cache/d7662500f1c2c0ff.doc.json b/.transpyc_cache/d7662500f1c2c0ff.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/d7662500f1c2c0ff.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/d7662500f1c2c0ff.pyi b/.transpyc_cache/d7662500f1c2c0ff.pyi new file mode 100644 index 0000000..03248a5 --- /dev/null +++ b/.transpyc_cache/d7662500f1c2c0ff.pyi @@ -0,0 +1,12 @@ +""" +Auto-generated Python stub file from testcheck_test.py +Module: testcheck_test +""" + + +import stdio +import t, c +import string +import testcheck + +def testcheck_test() -> int: pass diff --git a/.transpyc_cache/d7e3386b828acb66.pyi b/.transpyc_cache/d7e3386b828acb66.pyi new file mode 100644 index 0000000..aa17fc3 --- /dev/null +++ b/.transpyc_cache/d7e3386b828acb66.pyi @@ -0,0 +1,115 @@ +""" +Auto-generated Python stub file from socket.py +Module: socket +""" + + +import t, c +from stdint import * +import string +import platmacro +import stdio +import w32.winsock2 + +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 10 +SOCK_STREAM: t.CDefine = 1 +SOCK_DGRAM: t.CDefine = 2 +SOCK_RAW: t.CDefine = 3 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 1 +SO_RCVTIMEO: t.CDefine = 20 +SO_SNDTIMEO: t.CDefine = 21 +SO_REUSEADDR: t.CDefine = 2 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = -1 +SOCKET_DEFAULT_TIMEOUT: t.CDefine = 0 + +class SocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class HostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: INT + h_length: INT + h_addr_list: CHARPTR +class timeval: + tv_sec: INT + tv_usec: INT + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.CExtern | t.CExport: pass + +def connect(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass + +def send(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass + +def recv(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass + +def bind(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass + +def listen(fd: u64, backlog: INT) -> INT | t.CExtern | t.CExport: pass + +def accept(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.CExtern | t.CExport: pass + +def setsockopt(fd: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.CExtern | t.CExport: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> HostEnt | t.CPtr | t.CExtern | t.CExport: pass + +def ntohs(netshort: u16) -> u16 | t.CExtern | t.CExport: pass + +def htons(hostshort: u16) -> u16 | t.CExtern | t.CExport: pass + +def close(fd: u64) -> INT | t.CExtern | t.CExport: pass + +def _CloseFd(fd: INT) -> INT: pass + +def _MakeAddr(host: str, port: INT) -> SocketAddr: pass + +def _ParseIPv4(ip_str: str, out_addr: u32 | t.CPtr) -> bool: pass + + +class Socket: + fd: INT + family: INT + type: INT + protocol: INT + def __init__(self: Socket, family: INT, type: INT, protocol: INT) -> t.CInt: pass + def connect(self: Socket, host: str, port: INT) -> INT: pass + def send(self: Socket, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + def recv(self: Socket, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: pass + def sendall(self: Socket, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + def close(self: Socket) -> INT: pass + def bind(self: Socket, host: str, port: INT) -> INT: pass + def listen(self: Socket, backlog: INT) -> INT: pass + def accept(self: Socket, out_client: 'Socket' | t.CPtr) -> INT: pass + def settimeout(self: Socket, timeout_ms: INT) -> INT: pass + def fileno(self: Socket) -> INT: pass + +def SocketInit() -> INT: pass + +def SocketCleanup() -> INT: pass + +def SocketCreate(family: INT, type: INT, protocol: INT, out_sock: Socket | t.CPtr) -> INT: pass + +def SocketConnect(sock: Socket | t.CPtr, host: str, port: INT) -> INT: pass + +def SocketSend(sock: Socket | t.CPtr, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + +def SocketRecv(sock: Socket | t.CPtr, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: pass + +def SocketClose(sock: Socket | t.CPtr) -> INT: pass + +def SocketBind(sock: Socket | t.CPtr, host: str, port: INT) -> INT: pass + +def SocketListen(sock: Socket | t.CPtr, backlog: INT) -> INT: pass + +def SocketAccept(sock: Socket | t.CPtr, out_client: Socket | t.CPtr) -> INT: pass + +def SocketSetTimeout(sock: Socket | t.CPtr, timeout_ms: INT) -> INT: pass + +def SocketIsValid(sock: Socket | t.CPtr) -> bool: pass diff --git a/.transpyc_cache/d855b3e66d1d6522.doc.json b/.transpyc_cache/d855b3e66d1d6522.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/d855b3e66d1d6522.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/d855b3e66d1d6522.pyi b/.transpyc_cache/d855b3e66d1d6522.pyi new file mode 100644 index 0000000..09ce9d4 --- /dev/null +++ b/.transpyc_cache/d855b3e66d1d6522.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from __zlib.test_pyzlib.py +Module: __zlib.test_pyzlib +""" + + +from stdint import * +import pyzlib +import zhuff +import zdef +from stdio import printf +import stdlib +from string import strlen +import string +import t, c + +test_passed: t.CExtern | t.CInt +test_failed: t.CExtern | t.CInt + +def check(name: str, condition: t.CInt, detail: str) -> t.CInt: pass + +def print_hex_test(data: BYTEPTR, length: t.CSizeT, max_show: t.CSizeT) -> t.CInt: pass + +def print_separator() -> t.CInt: pass + +def section_header(title: str) -> t.CInt: pass + +def section_footer() -> t.CInt: pass + +def test_version() -> t.CInt: pass + +def test_compress_decompress() -> t.CInt: pass + +def test_compress_levels() -> t.CInt: pass + +def test_compressobj() -> t.CInt: pass + +def test_decompressobj() -> t.CInt: pass + +def test_decompressobj_max_length() -> t.CInt: pass + +def test_decompressobj_unused_data() -> t.CInt: pass + +def test_compress_copy() -> t.CInt: pass + +def test_decompress_copy() -> t.CInt: pass + +def test_adler32() -> t.CInt: pass + +def test_crc32() -> t.CInt: pass + +def test_empty_compress() -> t.CInt: pass + +def test_gzip_format() -> t.CInt: pass + +def test_raw_deflate() -> t.CInt: pass + +def test_zdict() -> t.CInt: pass + +def test_huffman_tree() -> t.CInt: pass + +def test_error_handling() -> t.CInt: pass + +def test_constants() -> t.CInt: pass + +def main123456() -> t.CInt: pass diff --git a/.transpyc_cache/d9322f0db660edd8.doc.json b/.transpyc_cache/d9322f0db660edd8.doc.json new file mode 100644 index 0000000..4dba15e --- /dev/null +++ b/.transpyc_cache/d9322f0db660edd8.doc.json @@ -0,0 +1 @@ +{"ReturnHandle": "Return 语句处理器：继承 Mixin 获得 Trans 回指针", "ReturnHandle.Handle": "翻译 Return 语句"} \ No newline at end of file diff --git a/.transpyc_cache/d9322f0db660edd8.pyi b/.transpyc_cache/d9322f0db660edd8.pyi new file mode 100644 index 0000000..abcbbff --- /dev/null +++ b/.transpyc_cache/d9322f0db660edd8.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesReturn.py +Module: lib.core.Handles.HandlesReturn +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar + +@t.NoVTable +class ReturnHandle(HandlesBase.Mixin): + def __init__(self: ReturnHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ReturnHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewReturnHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ReturnHandle | t.CPtr: pass diff --git a/.transpyc_cache/da80633346bb66e2.doc.json b/.transpyc_cache/da80633346bb66e2.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/da80633346bb66e2.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/da80633346bb66e2.pyi b/.transpyc_cache/da80633346bb66e2.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/da80633346bb66e2.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/db434883fc668c7c.doc.json b/.transpyc_cache/db434883fc668c7c.doc.json new file mode 100644 index 0000000..45fcbeb --- /dev/null +++ b/.transpyc_cache/db434883fc668c7c.doc.json @@ -0,0 +1 @@ +{"set_generic_context": "设置泛型特化上下文（进入泛型类方法体翻译时调用）", "clear_generic_context": "清除泛型特化上下文（离开泛型类方法体翻译时调用）", "resolve_generic_type_name": "如果 name 是泛型类型参数，返回对应的具体类型名；否则返回 name 本身\n\n 用于表达式中的 T 替换（如 T.__sizeof__() → int.__sizeof__()）\n ", "_get_type_name_from_annotation": "从类型注解节点提取类型名（支持 Name/Attribute/BinOp）", "_parse_type_string": "将类型名字符串解析为 AST 节点", "set_current_file": "设置当前编译的文件名（供报错使用）", "_read_source_line": "从源文件读取第 lineno 行内容", "fatal_error": "以 Python 风格报错并退出（带源代码行）", "fatal_type_error": "以 Python 风格报错并退出（fatal_error 的别名）", "map_t_type": "将 t.CXxx / c.CXxx 类型名映射到 LLVMType\n\n 注意: t.CChar | t.CPtr = i8* = str = bytes\n ", "_is_t_attr": "检测节点是否是 t.attr_name 形式", "extract_class_name_from_annotation": "从类型注解 AST 中提取类名，返回 None=无法提取", "resolve_annotation_type": "解析类型注解节点为 LLVMType\n\n Args:\n pool: 内存分配器\n node: 类型注解 AST 节点（Name 或 Attribute）\n imported_modules: 已导入模块名字符串（空格分隔）\n from_imports: from-import 映射字符串（\"name:module name:module\" 格式）\n ", "has_decorator_marker": "检测类型注解中是否包含指定的装饰器标记", "is_pass_body": "检测函数体是否为 pass（空体或仅含 pass 语句）", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "infer_expr_type": "推断表达式的 LLVM 类型\n\n param_types 格式: \"name:bits,name:bits,...\"\n ", "infer_return_type": "扫描函数体中的 return 语句，推断返回类型", "build_param_types_str": "构建参数类型编码字符串 \"name:bits,name:bits,...\" ", "_strappend": "将 src 追加到 dst 末尾"} \ No newline at end of file diff --git a/.transpyc_cache/db434883fc668c7c.pyi b/.transpyc_cache/db434883fc668c7c.pyi new file mode 100644 index 0000000..d6de8a3 --- /dev/null +++ b/.transpyc_cache/db434883fc668c7c.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesType.py +Module: lib.core.Handles.HandlesType +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import stdlib +import viperlib +import w32.fileio as fileio +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesStruct as HandlesStruct + +_g_current_file: t.CExtern | str +_g_generic_tp_names: t.CExtern | list[str] | t.CPtr +_g_generic_type_args: t.CExtern | list[str] | t.CPtr + +def set_generic_context(tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> t.CInt: pass + +def clear_generic_context() -> t.CInt: pass + +def resolve_generic_type_name(name: str) -> str: pass + +def _get_type_name_from_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _parse_type_string(pool: memhub.MemBuddy | t.CPtr, type_str: str, ctx: t.CInt) -> ast.AST | t.CPtr: pass + +def set_current_file(file_path: str) -> t.CInt: pass + + +_SRC_BUF_SIZE: t.CDefine = 262144 + +def _read_source_line(file_path: str, lineno: int) -> str: pass + +def fatal_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def fatal_type_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def map_t_type(pool: memhub.MemBuddy | t.CPtr, type_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_t_attr(node: ast.AST | t.CPtr, attr_name: str, imported_modules: str) -> int: pass + +def extract_class_name_from_annotation(node: ast.AST | t.CPtr, imported_modules: str) -> str: pass + +def resolve_annotation_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def has_decorator_marker(node: ast.AST | t.CPtr, marker_name: str) -> int: pass + +def is_pass_body(body: list[ast.AST | t.CPtr] | t.CPtr) -> int: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def infer_expr_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def infer_return_type(pool: memhub.MemBuddy | t.CPtr, body: list[ast.AST | t.CPtr] | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def build_param_types_str(pool: memhub.MemBuddy | t.CPtr, args_node: ast.AST | t.CPtr) -> str: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/.transpyc_cache/db4aafce81f39b32.doc.json b/.transpyc_cache/db4aafce81f39b32.doc.json new file mode 100644 index 0000000..c3a1e25 --- /dev/null +++ b/.transpyc_cache/db4aafce81f39b32.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/.transpyc_cache/db4aafce81f39b32.pyi b/.transpyc_cache/db4aafce81f39b32.pyi new file mode 100644 index 0000000..ad12e1f --- /dev/null +++ b/.transpyc_cache/db4aafce81f39b32.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 32 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/.transpyc_cache/db7c4a8f09d02a17.doc.json b/.transpyc_cache/db7c4a8f09d02a17.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/.transpyc_cache/db7c4a8f09d02a17.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/.transpyc_cache/db7c4a8f09d02a17.pyi b/.transpyc_cache/db7c4a8f09d02a17.pyi new file mode 100644 index 0000000..c2761c1 --- /dev/null +++ b/.transpyc_cache/db7c4a8f09d02a17.pyi @@ -0,0 +1,32 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/dd19090b337e9bea.doc.json b/.transpyc_cache/dd19090b337e9bea.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/dd19090b337e9bea.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/dd19090b337e9bea.pyi b/.transpyc_cache/dd19090b337e9bea.pyi new file mode 100644 index 0000000..898995b --- /dev/null +++ b/.transpyc_cache/dd19090b337e9bea.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from __zlib.zhuff.py +Module: __zlib.zhuff +""" + + +from stdint import * +import zdef +import t, c + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt +@t.Object +class zhuff_tree: + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + def __init__(self: zhuff_tree) -> t.CInt: pass + def build_codes(self: zhuff_tree, freqs: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def build_fixed_lit_tree(self: zhuff_tree) -> t.CInt: pass + def build_fixed_dist_tree(self: zhuff_tree) -> t.CInt: pass + def encode_symbol(self: zhuff_tree, symbol: int, writer: zdef.zbit_writer | t.CPtr) -> t.CInt: pass + def build_code_lengths(self: zhuff_tree, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def get_fixed_lit_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def get_fixed_dist_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def build_tree_from_lengths(self: zhuff_tree, lengths: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt +@t.Object +class zhuff_decode_tree: + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + def __init__(self: zhuff_decode_tree) -> t.CInt: pass + def build_decode_tree(self: zhuff_decode_tree, ht: zhuff_tree | t.CPtr) -> t.CInt: pass + def decode_symbol(self: zhuff_decode_tree, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/dd3002730623424b.pyi b/.transpyc_cache/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/.transpyc_cache/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/.transpyc_cache/df4ee63d3fef30a0.doc.json b/.transpyc_cache/df4ee63d3fef30a0.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/df4ee63d3fef30a0.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/df4ee63d3fef30a0.pyi b/.transpyc_cache/df4ee63d3fef30a0.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/df4ee63d3fef30a0.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/e0009679269c6b63.doc.json b/.transpyc_cache/e0009679269c6b63.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e0009679269c6b63.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e0009679269c6b63.pyi b/.transpyc_cache/e0009679269c6b63.pyi new file mode 100644 index 0000000..1a43833 --- /dev/null +++ b/.transpyc_cache/e0009679269c6b63.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from string_test.py +Module: string_test +""" + + +import stdio +import t, c +import string + +def test_strlen() -> t.CInt: pass + +def test_strcmp() -> t.CInt: pass + +def test_strncmp() -> t.CInt: pass + +def test_atoi() -> t.CInt: pass + +def test_strchr() -> t.CInt: pass + +def test_strstr() -> t.CInt: pass + +def string_test() -> int: pass diff --git a/.transpyc_cache/e0b84369e50c937c.pyi b/.transpyc_cache/e0b84369e50c937c.pyi new file mode 100644 index 0000000..c10ea89 --- /dev/null +++ b/.transpyc_cache/e0b84369e50c937c.pyi @@ -0,0 +1,93 @@ +""" +Auto-generated Python stub file from llvmlite.__function.py +Module: llvmlite.__function +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import stdio +from linkedlist import GSListNode, GSList +from .__types import LLVMType, ParamNode, TypePrint + +class Line(GSListNode[Line]): + Buf: t.CChar | t.CPtr + Len: t.CSizeT +class BasicBlock(GSListNode[BasicBlock]): + Name: t.CChar | t.CPtr + Lines: GSList[Line] | t.CPtr + IsTerminated: t.CInt +class Param(GSListNode[Param]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + Attrs: t.CChar | t.CPtr +class Function(GSListNode[Function]): + Name: t.CChar | t.CPtr + RetTy: LLVMType | t.CPtr + Params: GSList[Param] | t.CPtr + Blocks: GSList[BasicBlock] | t.CPtr + IsDeclared: t.CInt + IsVarArg: t.CInt + Attrs: t.CChar | t.CPtr + +LINE_SIZE: t.CDefine = 24 + +def new_line(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> Line | t.CPtr: pass + +def _new_gslist(pool: memhub.MemBuddy | t.CPtr, list_size: t.CSizeT) -> t.CPtr: pass + + +BLOCK_SIZE: t.CDefine = 32 + +def new_basic_block(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + + +PARAM_SIZE: t.CDefine = 32 + +def new_param(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + + +FUNC_SIZE: t.CDefine = 56 + +def new_function(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def function_add_param(func: Function | t.CPtr, param: Param | t.CPtr) -> t.CInt: pass + +def function_add_block(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def function_move_block_to_end(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def block_append_line(block: BasicBlock | t.CPtr, line: Line | t.CPtr) -> t.CInt: pass + +def block_append_text(pool: memhub.MemBuddy | t.CPtr, block: BasicBlock | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def function_set_attrs(func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def param_set_attrs(param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def _ll_name_needs_quote(name: str) -> t.CInt: pass + +def FunctionPrint(buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def function_get_name(func: Function | t.CPtr) -> t.CChar | t.CPtr: pass + +def function_get_ret_ty(func: Function | t.CPtr) -> LLVMType | t.CPtr: pass + +def function_get_params(func: Function | t.CPtr) -> GSList[Param] | t.CPtr: pass + +def function_get_param_head(func: Function | t.CPtr) -> Param | t.CPtr: pass + +def function_get_next(func: Function | t.CPtr) -> Function | t.CPtr: pass + +def function_is_declared(func: Function | t.CPtr) -> t.CInt: pass + +def param_get_name(param: Param | t.CPtr) -> t.CChar | t.CPtr: pass + +def param_get_ty(param: Param | t.CPtr) -> LLVMType | t.CPtr: pass + +def param_get_next(param: Param | t.CPtr) -> Param | t.CPtr: pass diff --git a/.transpyc_cache/e24a7f4a875f4518.doc.json b/.transpyc_cache/e24a7f4a875f4518.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e24a7f4a875f4518.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e24a7f4a875f4518.pyi b/.transpyc_cache/e24a7f4a875f4518.pyi new file mode 100644 index 0000000..2825744 --- /dev/null +++ b/.transpyc_cache/e24a7f4a875f4518.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import vrandom +import testcheck + +def main() -> t.CInt: pass diff --git a/.transpyc_cache/e272d279f536a678.doc.json b/.transpyc_cache/e272d279f536a678.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/e272d279f536a678.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/e272d279f536a678.pyi b/.transpyc_cache/e272d279f536a678.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/e272d279f536a678.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/e285a7afdd9d844f.doc.json b/.transpyc_cache/e285a7afdd9d844f.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/e285a7afdd9d844f.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/e285a7afdd9d844f.pyi b/.transpyc_cache/e285a7afdd9d844f.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/e285a7afdd9d844f.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/e306c57855054e23.doc.json b/.transpyc_cache/e306c57855054e23.doc.json new file mode 100644 index 0000000..6b37383 --- /dev/null +++ b/.transpyc_cache/e306c57855054e23.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/e306c57855054e23.pyi b/.transpyc_cache/e306c57855054e23.pyi new file mode 100644 index 0000000..e470478 --- /dev/null +++ b/.transpyc_cache/e306c57855054e23.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, args_head: llvmlite.Value | t.CPtr, arg_count: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/e36333e2cadb713a.doc.json b/.transpyc_cache/e36333e2cadb713a.doc.json new file mode 100644 index 0000000..4a5f0f2 --- /dev/null +++ b/.transpyc_cache/e36333e2cadb713a.doc.json @@ -0,0 +1 @@ +{"Load_project_config": "加载工程配置文件（project.vpj）\n\n Args:\n path: 配置文件路径\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "_join_path": "将相对路径与 project_dir 拼接为绝对路径\n\n - ./App + Test → Test/App\n - ./temp + Test → Test/temp\n - ../includes + Test → Test/../includes（OS 解析为 includes）\n - 绝对路径保持不变\n - None 返回 None\n ", "resolve_paths": "解析并规范化工程路径（src/temp/output/includes）\n\n 将 project.vpj 中的相对路径（以 ./ 开头）转换为基于 project_dir 的路径。\n\n Args:\n project_dir: 工程根目录（project.vpj 所在目录）\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "get_includes_binary_dir": "获取 includes.binary 目录路径（includes 目录 + .binary 后缀）\n\n includes 目录如 ../includes，则 includes.binary 为 ../includes.binary\n ", "print_config": "打印当前配置（调试用）"} \ No newline at end of file diff --git a/.transpyc_cache/e36333e2cadb713a.pyi b/.transpyc_cache/e36333e2cadb713a.pyi new file mode 100644 index 0000000..eaefae9 --- /dev/null +++ b/.transpyc_cache/e36333e2cadb713a.pyi @@ -0,0 +1,40 @@ +""" +Auto-generated Python stub file from lib.Projectrans.Config.py +Module: lib.Projectrans.Config +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import viperlib +from json.__parser import parse as json_parse +from json import JsonValue, JSON_OBJECT, JSON_INT +import w32.fileio as fileio + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SourceDir: t.CExtern | str +TempDir: t.CExtern | str +OutputDir: t.CExtern | str +ProjectName: t.CExtern | str +ProjectVersion: t.CExtern | str +CompilerCmd: t.CExtern | str +LinkerCmd: t.CExtern | str +LinkerOutput: t.CExtern | str +TargetTriple: t.CExtern | str +TargetDataLayout: t.CExtern | str +SliceLevel: t.CExtern | t.CInt +StrictMode: t.CExtern | t.CInt +IncludesDir: t.CExtern | str + +def Load_project_config(path: str) -> int: pass + +def _join_path(rel: str, project_dir: str) -> str: pass + +def resolve_paths(project_dir: str) -> int: pass + +def get_includes_binary_dir() -> str: pass + +def print_config() -> t.CInt: pass diff --git a/.transpyc_cache/e3834089153af692.doc.json b/.transpyc_cache/e3834089153af692.doc.json new file mode 100644 index 0000000..47b20d9 --- /dev/null +++ b/.transpyc_cache/e3834089153af692.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/e3834089153af692.pyi b/.transpyc_cache/e3834089153af692.pyi new file mode 100644 index 0000000..5214419 --- /dev/null +++ b/.transpyc_cache/e3834089153af692.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/e3ac223bdaf94a0c.doc.json b/.transpyc_cache/e3ac223bdaf94a0c.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e3ac223bdaf94a0c.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e3ac223bdaf94a0c.pyi b/.transpyc_cache/e3ac223bdaf94a0c.pyi new file mode 100644 index 0000000..10d5c52 --- /dev/null +++ b/.transpyc_cache/e3ac223bdaf94a0c.pyi @@ -0,0 +1,110 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdint import * +from stdio import printf +from stdlib import malloc +import testcheck + +class Rect: + width: t.CInt + height: t.CInt + def __new__(self: Rect, w: t.CInt, h: t.CInt) -> 'Rect' | t.CPtr: pass + def __init__(self: Rect, w: t.CInt, h: t.CInt) -> t.CInt: pass + @property + def area(self: Rect) -> t.CInt: pass + @property + def is_square(self: Rect) -> bool: pass + +def test_property_getter() -> t.CInt: pass + + +class Temperature: + _celsius: t.CDouble + def __new__(self: Temperature, celsius_val: t.CDouble) -> 'Temperature' | t.CPtr: pass + def __init__(self: Temperature, celsius_val: t.CDouble) -> t.CInt: pass + @property + def celsius(self: Temperature) -> t.CDouble: pass + @celsius.setter + def celsius(self: Temperature, val: t.CDouble) -> t.CInt: pass + @property + def fahrenheit(self: Temperature) -> t.CDouble: pass + +def test_property_setter() -> t.CInt: pass + + +class Counter: + _count: t.CInt + def __new__(self: Counter) -> 'Counter' | t.CPtr: pass + def __init__(self: Counter) -> t.CInt: pass + @property + def value(self: Counter) -> t.CInt: pass + @value.setter + def value(self: Counter, v: t.CInt) -> t.CInt: pass + +def test_property_getter_redef() -> t.CInt: pass + + +class Config: + _value: t.CInt + _deleted: t.CInt + def __new__(self: Config, v: t.CInt) -> 'Config' | t.CPtr: pass + def __init__(self: Config, v: t.CInt) -> t.CInt: pass + @property + def value(self: Config) -> t.CInt: pass + @value.setter + def value(self: Config, v: t.CInt) -> t.CInt: pass + @value.deleter + def value(self: Config) -> t.CInt: pass + def is_deleted(self: Config) -> bool: pass + +def test_property_deleter() -> t.CInt: pass + + +class MathUtils: + @staticmethod + def add(a: t.CInt, b: t.CInt) -> t.CInt: pass + @staticmethod + def max_val(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def test_staticmethod() -> t.CInt: pass + + +class Point: + x: t.CDouble + y: t.CDouble + def __new__(self: Point, x: t.CDouble, y: t.CDouble) -> 'Point' | t.CPtr: pass + def __init__(self: Point, x: t.CDouble, y: t.CDouble) -> t.CInt: pass + @classmethod + def origin(cls: Point) -> Point: pass + @classmethod + def from_int(cls: Point, x: t.CInt, y: t.CInt) -> Point: pass + def dist_sq(self: Point) -> t.CDouble: pass + +def test_classmethod() -> t.CInt: pass + + +class Circle: + _radius: t.CDouble + def __new__(self: Circle, r: t.CDouble) -> 'Circle' | t.CPtr: pass + def __init__(self: Circle, r: t.CDouble) -> t.CInt: pass + @property + def radius(self: Circle) -> t.CDouble: pass + @radius.setter + def radius(self: Circle, val: t.CDouble) -> t.CInt: pass + @property + def area(self: Circle) -> t.CDouble: pass + @staticmethod + def is_unit(r: t.CDouble) -> bool: pass + @classmethod + def unit(cls: Circle) -> Circle: pass + +def test_combined() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/.transpyc_cache/e3e7b6de8d7d8b03.pyi b/.transpyc_cache/e3e7b6de8d7d8b03.pyi new file mode 100644 index 0000000..ac69b1b --- /dev/null +++ b/.transpyc_cache/e3e7b6de8d7d8b03.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from _dict.py +Module: _dict +""" + + +import t, c +import memhub +import string +import json +from stdint import * + +VARIANT_NONE: t.CDefine = 0 # 无类型/空值 +VARIANT_INT: t.CDefine = 1 # int (i64) +VARIANT_FLOAT: t.CDefine = 2 # float (f32) +VARIANT_DOUBLE: t.CDefine = 3 # double (f64) +VARIANT_STR: t.CDefine = 4 # str (i8*) +VARIANT_DICT: t.CDefine = 5 # dict* +VARIANT_LIST: t.CDefine = 6 # list* +VARIANT_BOOL: t.CDefine = 7 # bool (i1/i32) +VARIANT_PTR: t.CDefine = 8 # 自定义指针类型 +_VARIANT_SIZE: t.CDefine = 16 + +class dict: + __keys__: t.CPtr + __values__: t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 8 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: dict, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: dict, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def __len__(self: dict) -> t.CSizeT: pass + def _find(self: dict, key: str) -> t.CSizeT: pass + def __getitem__(self: dict, key: str) -> t.CPtr: pass + def __setitem__(self: dict, key: str, val: t.CNeedPtr) -> t.CInt: pass + def _set_variant_raw(self: dict, key: str, vtype: t.CInt, ptr: t.CPtr) -> t.CInt: pass + def set_int(self: dict, key: str, val: int) -> t.CInt: pass + def set_str(self: dict, key: str, val: str) -> t.CInt: pass + def set_float(self: dict, key: str, val: float) -> t.CInt: pass + def set_double(self: dict, key: str, val: t.CDouble) -> t.CInt: pass + def set_dict(self: dict, key: str, val: dict | t.CPtr) -> t.CInt: pass + def set_list(self: dict, key: str, val: t.CPtr) -> t.CInt: pass + def set_bool(self: dict, key: str, val: t.CInt) -> t.CInt: pass + def get_vtype(self: dict, key: str) -> t.CInt: pass + def get(self: dict, key: str, default: t.CPtr) -> t.CPtr: pass + def contains(self: dict, key: str) -> t.CInt: pass + def __iter__(self: dict) -> dict | t.CPtr: pass + def __next__(self: dict) -> str: pass + def to_json(self: dict, pool: memhub.MemManager | t.CPtr) -> json.JsonValue | t.CPtr: pass + def from_json(self: dict, pool: memhub.MemManager | t.CPtr, root: json.JsonValue | t.CPtr) -> t.CInt: pass + def dumps(self: dict, pool: memhub.MemManager | t.CPtr) -> str: pass + def loads(self: dict, pool: memhub.MemManager | t.CPtr, json_str: str) -> t.CInt: pass \ No newline at end of file diff --git a/.transpyc_cache/e3f3b9ed2a5030c6.doc.json b/.transpyc_cache/e3f3b9ed2a5030c6.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/e3f3b9ed2a5030c6.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/e3f3b9ed2a5030c6.pyi b/.transpyc_cache/e3f3b9ed2a5030c6.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/e3f3b9ed2a5030c6.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/e445a2bb299d1b7d.doc.json b/.transpyc_cache/e445a2bb299d1b7d.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e445a2bb299d1b7d.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e445a2bb299d1b7d.pyi b/.transpyc_cache/e445a2bb299d1b7d.pyi new file mode 100644 index 0000000..355f684 --- /dev/null +++ b/.transpyc_cache/e445a2bb299d1b7d.pyi @@ -0,0 +1,51 @@ +""" +Auto-generated Python stub file from __zlib.zinflate.py +Module: __zlib.zinflate +""" + + +from stdint import * +import zinflate +import zchecksum +import zdef +import zhuff +import string +import t, c + +@t.Object +class zinflate_stream: + reader: zdef.zbit_reader + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + header_parsed: t.CInt + is_gzip: t.CInt + adler: t.CUInt32T + crc: t.CUInt32T + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + max_length: t.CSizeT + def __init__(self: zinflate_stream) -> t.CInt: pass + def output_byte(self: zinflate_stream, byte: BYTE) -> t.CInt: pass + def parse_header(self: zinflate_stream) -> t.CInt: pass + def read_bits_from_input(self: zinflate_stream, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_huffman(self: zinflate_stream, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: pass + def inflate_block_stored(self: zinflate_stream) -> t.CInt: pass + def inflate_block_fixed(self: zinflate_stream) -> t.CInt: pass + def inflate_block_dynamic(self: zinflate_stream) -> t.CInt: pass + def inflate_stream(self: zinflate_stream) -> t.CInt: pass + def decompress(self: zinflate_stream, data: UINT8PTR, length: t.CSizeT, max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zinflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zinflate_stream) -> zinflate_stream | t.CPtr: pass + def destroy(self: zinflate_stream) -> t.CInt: pass + +def zinflate_one_shot(data: UINT8PTR, length: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass + +def zinflate_create(wbits: t.CInt) -> zinflate_stream | t.CPtr: pass diff --git a/.transpyc_cache/e4c98f097b75da35.doc.json b/.transpyc_cache/e4c98f097b75da35.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/.transpyc_cache/e4c98f097b75da35.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/.transpyc_cache/e4c98f097b75da35.pyi b/.transpyc_cache/e4c98f097b75da35.pyi new file mode 100644 index 0000000..d866b3e --- /dev/null +++ b/.transpyc_cache/e4c98f097b75da35.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/.transpyc_cache/e5be27ae4fe7ce04.doc.json b/.transpyc_cache/e5be27ae4fe7ce04.doc.json new file mode 100644 index 0000000..7307589 --- /dev/null +++ b/.transpyc_cache/e5be27ae4fe7ce04.doc.json @@ -0,0 +1 @@ +{"init_enum_table": "初始化枚举注册表，返回 1 成功", "_get_enum_entry": "获取第 i 个枚举条目", "_get_enum_member": "获取枚举中第 i 个成员条目", "register_enum": "注册枚举类型，返回 EnumEntry 指针", "add_enum_member": "向枚举添加成员，返回成员索引（-1 失败）", "find_enum": "按枚举类名查找，返回 EnumEntry 或 None", "lookup_enum_member": "按枚举类名和成员名查找，返回 EnumMember 或 None", "is_enum_class": "检查类名是否为已注册枚举，返回 1=是 / 0=否", "get_enum_base_type": "按枚举类名获取基准类型"} \ No newline at end of file diff --git a/.transpyc_cache/e5be27ae4fe7ce04.pyi b/.transpyc_cache/e5be27ae4fe7ce04.pyi new file mode 100644 index 0000000..ee312c4 --- /dev/null +++ b/.transpyc_cache/e5be27ae4fe7ce04.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesEnum.py +Module: lib.core.Handles.HandlesEnum +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio + +ENUM_MAX: t.CDefine = 64 +ENUM_MEMBER_MAX: t.CDefine = 64 + +@t.NoVTable +class EnumMember: + Name: t.CChar | t.CPtr + Value: t.CInt64T + Ty: llvmlite.LLVMType | t.CPtr +@t.NoVTable +class EnumEntry: + Name: t.CChar | t.CPtr + BaseTy: llvmlite.LLVMType | t.CPtr + MemberCount: int + Members: EnumMember | t.CPtr + +_enum_table: t.CExtern | EnumEntry | t.CPtr +_enum_count: t.CExtern | int + +def init_enum_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_enum_entry(i: int) -> EnumEntry | t.CPtr: pass + +def _get_enum_member(enum_entry: EnumEntry | t.CPtr, i: int) -> EnumMember | t.CPtr: pass + +def register_enum(pool: memhub.MemBuddy | t.CPtr, name: str, base_ty: llvmlite.LLVMType | t.CPtr) -> EnumEntry | t.CPtr: pass + +def add_enum_member(pool: memhub.MemBuddy | t.CPtr, enum_entry: EnumEntry | t.CPtr, member_name: str, member_val: t.CInt64T, member_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def find_enum(name: str) -> EnumEntry | t.CPtr: pass + +def lookup_enum_member(enum_name: str, member_name: str) -> EnumMember | t.CPtr: pass + +def is_enum_class(name: str) -> int: pass + +def get_enum_base_type(enum_name: str) -> llvmlite.LLVMType | t.CPtr: pass diff --git a/.transpyc_cache/e5fb16127bce8f68.doc.json b/.transpyc_cache/e5fb16127bce8f68.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e5fb16127bce8f68.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e5fb16127bce8f68.pyi b/.transpyc_cache/e5fb16127bce8f68.pyi new file mode 100644 index 0000000..5677a22 --- /dev/null +++ b/.transpyc_cache/e5fb16127bce8f68.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from zc.class_test.py +Module: zc.class_test +""" + + +import stdio +import stdint +import t, c + +class SimpleData(t.Object): + x: t.CInt + y: t.CInt +class Point3D(t.Object): + x: t.CFloat + y: t.CFloat + z: t.CFloat +class Student(t.Object): + name: t.CChar | t.CPtr + age: t.CInt + score: t.CInt + def __init__(self: Student, n: t.CChar | t.CPtr, a: t.CInt, s: t.CInt) -> t.CInt: pass + def get_age(self: Student) -> t.CInt: pass + def get_score(self: Student) -> t.CInt: pass +class Counter(t.Object): + value: t.CInt + def __init__(self: Counter, initial: t.CInt) -> t.CInt: pass + def increment(self: Counter) -> t.CInt: pass + def get_value(self: Counter) -> t.CInt: pass +class StackObj(t.Object): + data: t.CInt + next_ptr: t.CVoid | t.CPtr + +def TestBasicDataClass() -> t.CInt: pass + +def TestPoint3DClass() -> t.CInt: pass + +def TestStudentClass() -> t.CInt: pass + +def TestCounterClass() -> t.CInt: pass + +def TestStackAllocation() -> t.CInt: pass + +def TestHeapAllocation() -> t.CInt: pass + +def TestHeapObjectWithInit() -> t.CInt: pass + +def TestMultipleStackObjects() -> t.CInt: pass + +def TestImplicitInitCall() -> t.CInt: pass + +def TestClassWithBitFields() -> t.CInt: pass + +def TestIntArray() -> t.CInt: pass + +def TestFloatArray() -> t.CInt: pass + +def TestCharArray() -> t.CInt: pass + +def TestNestedArray() -> t.CInt: pass + +def TestArrayWithStruct() -> t.CInt: pass + +def TestArrayPointer() -> t.CInt: pass + +def TestDynamicString() -> t.CInt: pass diff --git a/.transpyc_cache/e7183a8c64d4872d.doc.json b/.transpyc_cache/e7183a8c64d4872d.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/e7183a8c64d4872d.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/e7183a8c64d4872d.pyi b/.transpyc_cache/e7183a8c64d4872d.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/e7183a8c64d4872d.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/.transpyc_cache/e74870eba3e6d774.doc.json b/.transpyc_cache/e74870eba3e6d774.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/e74870eba3e6d774.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/e74870eba3e6d774.pyi b/.transpyc_cache/e74870eba3e6d774.pyi new file mode 100644 index 0000000..ec2af45 --- /dev/null +++ b/.transpyc_cache/e74870eba3e6d774.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from float_test.py +Module: float_test +""" + + +import stdio +import t, c + +class DataUnion(t.CUnion): + i: t.CInt + f: t.CFloat + l: t.CInt64T + +def float_test() -> int: pass diff --git a/.transpyc_cache/e773fbfd4a1e4587.doc.json b/.transpyc_cache/e773fbfd4a1e4587.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/e773fbfd4a1e4587.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/e773fbfd4a1e4587.pyi b/.transpyc_cache/e773fbfd4a1e4587.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/e773fbfd4a1e4587.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/eb252050d92f8fb9.doc.json b/.transpyc_cache/eb252050d92f8fb9.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/eb252050d92f8fb9.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/eb252050d92f8fb9.pyi b/.transpyc_cache/eb252050d92f8fb9.pyi new file mode 100644 index 0000000..00b3762 --- /dev/null +++ b/.transpyc_cache/eb252050d92f8fb9.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from deref_test.py +Module: deref_test +""" + + +import stdio +import t, c + +def deref_test() -> int: pass diff --git a/.transpyc_cache/ed1c124f3cb787c9.doc.json b/.transpyc_cache/ed1c124f3cb787c9.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/ed1c124f3cb787c9.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/ed1c124f3cb787c9.pyi b/.transpyc_cache/ed1c124f3cb787c9.pyi new file mode 100644 index 0000000..123d4fd --- /dev/null +++ b/.transpyc_cache/ed1c124f3cb787c9.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from ptr_only_test.py +Module: ptr_only_test +""" + + +import stdio +import t, c + +def ptr_only_test() -> int: pass diff --git a/.transpyc_cache/ee084e9fc6ee413a.pyi b/.transpyc_cache/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/.transpyc_cache/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/.transpyc_cache/ee52c08239684346.pyi b/.transpyc_cache/ee52c08239684346.pyi new file mode 100644 index 0000000..ee09339 --- /dev/null +++ b/.transpyc_cache/ee52c08239684346.pyi @@ -0,0 +1,215 @@ +""" +Auto-generated Python stub file from ast.parser.py +Module: ast.parser +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp +from .base import AST, ASTCtx, ASTKind, OpKind, ASTFlag, _set_pos, _binop_from_op, _augop_from_op, _cmpop_from_op, _copy_str, _inherit_pos, _init_ast, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .astaux import ExceptHandler, Arguments, Arg, Keyword as KwNode, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .lexer import Lexer, _lexer_init, tokenize, new_lexer + +class Parser: + tokens: Token | t.CPtr + cur: Token | t.CPtr + pool: memhub.MemManager | t.CPtr + error_count: t.CInt + pending_decorators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Parser, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_parser(pool: memhub.MemManager | t.CPtr) -> Parser | t.CPtr: pass + +def _parser_init(ps: Parser | t.CPtr, tokens: Token | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _advance(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_type(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_kw(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_str(ps: Parser | t.CPtr) -> str: pass + +def _cur_lineno(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_col(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nl(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nls_and_newlines(ps: Parser | t.CPtr) -> t.CInt: pass + +def _expect_type(ps: Parser | t.CPtr, ttype: t.CInt) -> t.CInt: pass + +def _expect_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _match_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _accept_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _new_const_int(pool: memhub.MemManager | t.CPtr, val: t.CInt64T, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_float(pool: memhub.MemManager | t.CPtr, val: t.CDouble, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_str(pool: memhub.MemManager | t.CPtr, val: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_bool(pool: memhub.MemManager | t.CPtr, val: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_none(pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _set_ctx(node: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + +def _fix_store_ctx(node: AST | t.CPtr) -> t.CInt: pass + +def _parse_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_namedexpr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_lambda(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_or_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_and_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_not_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_comparison(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _match_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _advance_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _parse_bitor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitxor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitand(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_shift(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_arith(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_term(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_factor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_power(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_call(ps: Parser | t.CPtr, func: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_subscript(ps: Parser | t.CPtr, value: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_slice(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_paren(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_brace(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sub_expr(ps: Parser | t.CPtr, src: str) -> AST | t.CPtr: pass + +def _parse_fstring(ps: Parser | t.CPtr, content: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comprehension(ps: Parser | t.CPtr, elt: AST | t.CPtr, vtype: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comp_for(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_target_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_target(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_yield(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_varargslist(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_arg_def(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_suite_into_children(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_statement_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_simple_stmt_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_small_stmt(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _maybe_tuple(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_testlist(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_expr_stmt(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_return(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_raise(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_global(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_nonlocal(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_from_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_alias_list(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_del(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_assert(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_compound(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if_tail(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_while(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_for(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_try(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_except_handler(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_with(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_with_item(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_funcdef(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_classdef(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_match(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _make_name_ast(pool: memhub.MemManager | t.CPtr, node: Name | t.CPtr) -> AST | t.CPtr: pass + +def _parse_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_closed_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sequence_pattern(ps: Parser | t.CPtr, close_op: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_mapping_pattern(ps: Parser | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_class_pattern(ps: Parser | t.CPtr, cls: AST | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_match_case(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_decorators(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_module_body(ps: Parser | t.CPtr, module_node: AST | t.CPtr) -> t.CInt: pass + +def parse_tokens(pool: memhub.MemManager | t.CPtr, tokens: Token | t.CPtr) -> AST | t.CPtr: pass diff --git a/.transpyc_cache/ef45adce81bf5b2d.doc.json b/.transpyc_cache/ef45adce81bf5b2d.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/ef45adce81bf5b2d.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/ef45adce81bf5b2d.pyi b/.transpyc_cache/ef45adce81bf5b2d.pyi new file mode 100644 index 0000000..7eeac7e --- /dev/null +++ b/.transpyc_cache/ef45adce81bf5b2d.pyi @@ -0,0 +1,12 @@ +""" +Auto-generated Python stub file from namespace_test.py +Module: namespace_test +""" + + +import stdio +import t, c +import namespace_defs +from namespace_defs import Widget, Gadget, PlainStruct + +def namespace_test() -> int: pass diff --git a/.transpyc_cache/f08abd3a207590f0.doc.json b/.transpyc_cache/f08abd3a207590f0.doc.json new file mode 100644 index 0000000..724daa6 --- /dev/null +++ b/.transpyc_cache/f08abd3a207590f0.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到\n\n allow_star_fallback: 传 0 禁用 star import 回退（用于明确模块名/函数名查找）\n ", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/f08abd3a207590f0.pyi b/.transpyc_cache/f08abd3a207590f0.pyi new file mode 100644 index 0000000..065139b --- /dev/null +++ b/.transpyc_cache/f08abd3a207590f0.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str, allow_star_fallback: int) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/f26cd4c78b4d554c.doc.json b/.transpyc_cache/f26cd4c78b4d554c.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/f26cd4c78b4d554c.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/f26cd4c78b4d554c.pyi b/.transpyc_cache/f26cd4c78b4d554c.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/f26cd4c78b4d554c.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/f26d7266c4f8762b.doc.json b/.transpyc_cache/f26d7266c4f8762b.doc.json new file mode 100644 index 0000000..7565271 --- /dev/null +++ b/.transpyc_cache/f26d7266c4f8762b.doc.json @@ -0,0 +1 @@ +{"remote_func": "Remote function docstring.", "RemoteClass": "Remote class docstring.", "RemoteClass.remote_method": "Remote method docstring."} \ No newline at end of file diff --git a/.transpyc_cache/f26d7266c4f8762b.pyi b/.transpyc_cache/f26d7266c4f8762b.pyi new file mode 100644 index 0000000..845daee --- /dev/null +++ b/.transpyc_cache/f26d7266c4f8762b.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from remote.py +Module: remote +""" + +import c + + +import t + +def remote_func() -> t.CInt: pass + + +@t.Object +class RemoteClass: + value: t.CInt + def remote_method(self: RemoteClass) -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass diff --git a/.transpyc_cache/f2ecbf8ced20575c.pyi b/.transpyc_cache/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/.transpyc_cache/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/.transpyc_cache/f2fc34882dad695a.doc.json b/.transpyc_cache/f2fc34882dad695a.doc.json new file mode 100644 index 0000000..c00e036 --- /dev/null +++ b/.transpyc_cache/f2fc34882dad695a.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/.transpyc_cache/f2fc34882dad695a.pyi b/.transpyc_cache/f2fc34882dad695a.pyi new file mode 100644 index 0000000..40c3168 --- /dev/null +++ b/.transpyc_cache/f2fc34882dad695a.pyi @@ -0,0 +1,171 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/.transpyc_cache/f492cb40d913400b.doc.json b/.transpyc_cache/f492cb40d913400b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/f492cb40d913400b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/f492cb40d913400b.pyi b/.transpyc_cache/f492cb40d913400b.pyi new file mode 100644 index 0000000..f8cf5c7 --- /dev/null +++ b/.transpyc_cache/f492cb40d913400b.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import viperlib + +@t.CExport +def main() -> int: pass diff --git a/.transpyc_cache/f49abd3c81119ada.doc.json b/.transpyc_cache/f49abd3c81119ada.doc.json new file mode 100644 index 0000000..95e17f3 --- /dev/null +++ b/.transpyc_cache/f49abd3c81119ada.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/.transpyc_cache/f49abd3c81119ada.pyi b/.transpyc_cache/f49abd3c81119ada.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/.transpyc_cache/f49abd3c81119ada.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/.transpyc_cache/f5522571bcce7bcb.pyi b/.transpyc_cache/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/.transpyc_cache/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/.transpyc_cache/f6b51804a0ba8ff0.pyi b/.transpyc_cache/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/.transpyc_cache/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/.transpyc_cache/f9c68a21786049ce.doc.json b/.transpyc_cache/f9c68a21786049ce.doc.json new file mode 100644 index 0000000..67e7a8b --- /dev/null +++ b/.transpyc_cache/f9c68a21786049ce.doc.json @@ -0,0 +1 @@ +{"_DiagAppend": "追加诊断信息到 diag.txt（追加模式，不依赖 printf 缓冲）", "TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/.transpyc_cache/f9c68a21786049ce.pyi b/.transpyc_cache/f9c68a21786049ce.pyi new file mode 100644 index 0000000..9d8b33e --- /dev/null +++ b/.transpyc_cache/f9c68a21786049ce.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def _DiagAppend(msg: bytes) -> t.CInt: pass + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/.transpyc_cache/faec3233d98371da.doc.json b/.transpyc_cache/faec3233d98371da.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/faec3233d98371da.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/faec3233d98371da.pyi b/.transpyc_cache/faec3233d98371da.pyi new file mode 100644 index 0000000..308f5cb --- /dev/null +++ b/.transpyc_cache/faec3233d98371da.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from zc.logic_test.py +Module: zc.logic_test +""" + + +import config +import t, c + +def ForRangeBasicTest(stop: t.CInt) -> t.CInt: pass + +def ForRangeWithStartStop(start: t.CInt, stop: t.CInt, step: t.CInt) -> t.CInt: pass + +def NestedForRangeTest(rows: t.CInt, cols: t.CInt) -> t.CInt: pass + +def ForWithIfBreakTest(data_size: t.CInt) -> t.CInt: pass + +def ForWithIfContinueTest(n: t.CInt) -> t.CInt: pass + +def ForWithElifBranchTest(value: t.CInt) -> t.CInt: pass + +def WhileWithIfElifElseTest(iterations: t.CInt) -> t.CInt: pass + +def WhileWithNestedIfTest(limit: t.CInt) -> t.CInt: pass + +def WhileWithMatchCaseTest(value: t.CInt) -> t.CInt: pass + +def WhileWithMatchCaseNoBreakTest(value: t.CInt) -> t.CInt: pass + +def ComplexForWhileMatchTest(outer_limit: t.CInt, inner_limit: t.CInt) -> t.CInt: pass + +def ForWithCaseAndNoBreakTest(n: t.CInt) -> t.CInt: pass + +def WhileWithBreakConditionTest(limit: t.CInt) -> t.CInt: pass + +def NestedWhileWithMultipleBreakPoints(depth: t.CInt, width: t.CInt) -> t.CInt: pass + +def ForWithMatchCaseInLoopTest(iterations: t.CInt) -> t.CInt: pass + +def ComplexLogicWithElifChains(a: t.CInt, b: t.CInt, c_val: t.CInt) -> t.CInt: pass + +def ForRangeWithStepAndCondition(n: t.CInt, step: t.CInt) -> t.CInt: pass + +def MatchCaseWithGuardConditions(value: t.CInt) -> t.CInt: pass + +def WhileWithMultipleMatchCases(iterations: t.CInt) -> t.CInt: pass + +def ForWhileMixWithBreakAndContinue(n: t.CInt) -> t.CInt: pass + +def ComplexCaseMatchingWithRanges(start: t.CInt, end: t.CInt) -> t.CInt: pass + +def WhileWithElifInElifChain(limit: t.CInt) -> t.CInt: pass + +def ForWithCaseNoBreakNested(inner_limit: t.CInt) -> t.CInt: pass diff --git a/.transpyc_cache/fb6cea4f70fade7a.doc.json b/.transpyc_cache/fb6cea4f70fade7a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/.transpyc_cache/fb6cea4f70fade7a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/.transpyc_cache/fb6cea4f70fade7a.pyi b/.transpyc_cache/fb6cea4f70fade7a.pyi new file mode 100644 index 0000000..12b9442 --- /dev/null +++ b/.transpyc_cache/fb6cea4f70fade7a.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from t import CInt, CLong, CDouble, CPtr, CSizeT +from stdio import printf +import cpython +import testcheck + +def test_init_finalize() -> t.CInt: pass + +def test_long_operations() -> t.CInt: pass + +def test_string_operations() -> t.CInt: pass + +def test_list_operations() -> t.CInt: pass + +def test_dict_operations() -> t.CInt: pass + +def test_pyrun_string() -> t.CInt: pass + +def test_import_module() -> t.CInt: pass + +def test_operator_overload() -> t.CInt: pass + +def main() -> CInt | t.CExport: pass diff --git a/.transpyc_cache/ff3ad21889888a9a.doc.json b/.transpyc_cache/ff3ad21889888a9a.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/.transpyc_cache/ff3ad21889888a9a.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/.transpyc_cache/ff3ad21889888a9a.pyi b/.transpyc_cache/ff3ad21889888a9a.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/.transpyc_cache/ff3ad21889888a9a.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/Projectrans.py b/Projectrans.py new file mode 100644 index 0000000..2b39c18 --- /dev/null +++ b/Projectrans.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python3 +"""ProjectTrans.py - 工程级 TransPyC 翻译器""" +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） + +from lib.Projectrans import ( + compute_sha1, _check_annotation_for_export, sha1_file, + get_file_dependencies, find_reachable_files_from_entries, + topological_sort_files, Phase1Generator, DeclarationGenerator, + Phase2Translator, _parallel_translate_worker, + save_sha1_map, Load_sha1_map, Load_project_config, resolve_paths, main +) + +__all__ = [ + 'compute_sha1', '_check_annotation_for_export', 'sha1_file', + 'get_file_dependencies', 'find_reachable_files_from_entries', + 'topological_sort_files', 'Phase1Generator', 'DeclarationGenerator', + 'Phase2Translator', '_parallel_translate_worker', + 'save_sha1_map', 'Load_sha1_map', 'Load_project_config', 'resolve_paths', 'main' +] + +if __name__ == '__main__': + main() diff --git a/README.md b/README.md new file mode 100644 index 0000000..688813a --- /dev/null +++ b/README.md @@ -0,0 +1,76 @@ +# TransPyC + +**TransPyC** 是一种基于 Python 语法的系统级编程语言编译器，将 Python 子集（通过 `t`/`c` 模块类型注解赋予 LLVM 语义）编译为原生机器码。它不是 Python 的子集，也不是 C 的语法糖——它拥有独立的类型系统、两阶段编译模型和 SHA1 命名空间机制，适用于操作系统内核、嵌入式系统、驱动程序等底层开发场景。 + +## 能力证明 + +TransPyC 已用自身实现以下组件，证明其作为系统级语言的成熟度： + +- **ViperOS** —— 真实操作系统内核（[ViperOS/](ViperOS/)），含 gdt/idt/isr/mm/sched，可在 QEMU 启动 +- **自研 Python 解析器** —— [includes/ast/](includes/ast/)（4,176 LOC），用 TransPyC 自己写的词法+语法+AST，覆盖 CPython 几乎全部语法 +- **自研 LLVM IR 生成库** —— [includes/llvmlite/](includes/llvmlite/)（~1,500 LOC），用 TransPyC 自己写的 IR 文本生成器，18/18 测试通过 +- **zlib 移植** —— [Test/ZlibTest/](Test/ZlibTest/)，完整 deflate/inflate/huffman/checksum（~1,900 LOC） +- **Win32 FFI 绑定** —— [includes/w32/](includes/w32/)，file/memory/process/sync/console 全覆盖 +- **泛型容器** —— [includes/linkedlist.py](includes/linkedlist.py)，`@t.NoVTable` + PEP 695 递归泛型继承 `class GNode(GSListNode[GNode])` + +## 核心设计理念 + +- **Python 语法，LLVM 语义**：源文件是合法的 Python 语法，但通过 `t` 模块类型注解赋予 LLVM 级别的语义 +- **类型注解即编译指令**：类型注解决定 LLVM IR 的生成，不是可选的提示 +- **两阶段编译**：先提取声明接口（`.pyi` + `.stub.ll`），再翻译源文件为含代码的 `.ll` +- **SHA1 命名空间**：每个源文件按内容 SHA1 哈希命名，自动消除跨模块符号冲突 +- **零运行时开销**：编译为原生代码，无 GC、无解释器、无虚拟机 + +## 自举进展 + +TransPyC 正在追求**完全自举**——用 TransPyC 重写 TransPyC 编译器自身。当前状态： + +- ✅ **前端**：`includes/ast/` 已替代 Python `ast` 模块依赖 +- ✅ **后端**：`includes/llvmlite/` 已替代 Python `llvmlite` 依赖 +- ⏳ **中间层**：`lib/core`（~20K LOC）待用 TransPyC 重写 + +详见 [wiki/13-bootstrapping.md](wiki/13-bootstrapping.md)。 + +## 模块总览 + +Viper 语言通过两个核心模块提供类型和操作支持： + +| 模块 | 用途 | 导入方式 | +|------|------|----------| +| `t` | 类型定义系统 | `import t` | +| `c` | C 语言操作指令 | `import c` | + +## 快速上手 + +1. **环境**：Python 3.12+、LLVM（llc/clang）、Windows（当前主平台） +2. **构建项目**：在项目目录运行 `python ../Projectrans.py build`（详见 [wiki/12-project.md](wiki/12-project.md)） +3. **示例**：参考 [Test/](Test/) 下 30+ 测试项目（AstTest/ZlibTest/NoVTableTest/LLvmLiteTest 等） +4. **写第一个程序**： + +```python +import t, c +from stdint import * +import stdio + +def main() -> CInt | CExport: + stdio.printf("Hello, Viper!\n") + return 0 +``` + +## 文档索引 + +| 文件 | 内容 | +|------|------| +| [01-overview.md](wiki/01-overview.md) | 语言概述、两阶段编译流程、SHA1 命名空间、`t.CDefine`/`t.CExport`/`t.CInline` 深度解析 | +| [02-type-system.md](wiki/02-type-system.md) | 类型系统（基本类型、指针、数组、结构体、联合体、枚举、typedef、`__attribute__`） | +| [03-variables.md](wiki/03-variables.md) | 变量声明与赋值（含 `t.CDefine` 常量） | +| [04-functions.md](wiki/04-functions.md) | 函数定义与调用（含 `t.CExport`、`t.CInline`、`t.CDefine` 函数、`@c.CReturn`） | +| [05-classes.md](wiki/05-classes.md) | 类与数据布局（结构体、联合体、枚举、typedef、位域） | +| [06-oop.md](wiki/06-oop.md) | 面向对象与运算符重载（`@t.Object`、`@t.CVTable`、`@t.NoVTable`、继承、dunder 方法） | +| [07-control-flow.md](wiki/07-control-flow.md) | 控制流（条件、循环、match） | +| [08-c-operations.md](wiki/08-c-operations.md) | C 语言操作（指针、内联汇编、FFI 外部函数声明、LLVM IR 内联） | +| [09-exceptions.md](wiki/09-exceptions.md) | 异常处理 | +| [10-imports.md](wiki/10-imports.md) | 模块与导入系统（存根生成、SHA1 命名空间与导入） | +| [11-builtins.md](wiki/11-builtins.md) | 内置函数与运算符 | +| [12-project.md](wiki/12-project.md) | 项目配置与构建（增量编译、SHA1 映射） | +| [13-bootstrapping.md](wiki/13-bootstrapping.md) | 自举可行性分析（当前进展、剩余工作、路径规划） | diff --git a/StandardTemplateProject/App/benchmark_main.py b/StandardTemplateProject/App/benchmark_main.py new file mode 100644 index 0000000..0c59fe0 --- /dev/null +++ b/StandardTemplateProject/App/benchmark_main.py @@ -0,0 +1,12 @@ +import w32.win32console as w32cmd +import t, c + + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: + w32cmd.SetConsoleOutputCP(pagecode) + w32cmd.SetConsoleCP(pagecode) + print("你好，世界") + return 0 \ No newline at end of file diff --git a/StandardTemplateProject/output/067c78e9f121dce3.deps.json b/StandardTemplateProject/output/067c78e9f121dce3.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/067c78e9f121dce3.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/06f53cc594b4ac6c.deps.json b/StandardTemplateProject/output/06f53cc594b4ac6c.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/06f53cc594b4ac6c.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/203a3999d5300163.deps.json b/StandardTemplateProject/output/203a3999d5300163.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/203a3999d5300163.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/72e2d5ccb7cedcf1.deps.json b/StandardTemplateProject/output/72e2d5ccb7cedcf1.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/72e2d5ccb7cedcf1.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/7e529fe7a078cfef.deps.json b/StandardTemplateProject/output/7e529fe7a078cfef.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/7e529fe7a078cfef.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/_shared_sym_includes.pickle b/StandardTemplateProject/output/_shared_sym_includes.pickle new file mode 100644 index 0000000..2d4be0f Binary files /dev/null and b/StandardTemplateProject/output/_shared_sym_includes.pickle differ diff --git a/StandardTemplateProject/output/bbdf3bbd4c3bc28c.deps.json b/StandardTemplateProject/output/bbdf3bbd4c3bc28c.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/bbdf3bbd4c3bc28c.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/f6b51804a0ba8ff0.deps.json b/StandardTemplateProject/output/f6b51804a0ba8ff0.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/StandardTemplateProject/output/f6b51804a0ba8ff0.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/StandardTemplateProject/output/f71cf01f35e2325a.deps.json b/StandardTemplateProject/output/f71cf01f35e2325a.deps.json new file mode 100644 index 0000000..48e3d83 --- /dev/null +++ b/StandardTemplateProject/output/f71cf01f35e2325a.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/StandardTemplateProject/project.vpj b/StandardTemplateProject/project.vpj new file mode 100644 index 0000000..f42ddc3 --- /dev/null +++ b/StandardTemplateProject/project.vpj @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "StandardTemplateProject", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "app.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/StandardTemplateProject/temp/067c78e9f121dce3.pyi b/StandardTemplateProject/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/StandardTemplateProject/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/StandardTemplateProject/temp/06f53cc594b4ac6c.pyi b/StandardTemplateProject/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/StandardTemplateProject/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/StandardTemplateProject/temp/203a3999d5300163.pyi b/StandardTemplateProject/temp/203a3999d5300163.pyi new file mode 100644 index 0000000..a0c57dc --- /dev/null +++ b/StandardTemplateProject/temp/203a3999d5300163.pyi @@ -0,0 +1,80 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/StandardTemplateProject/temp/72e2d5ccb7cedcf1.pyi b/StandardTemplateProject/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/StandardTemplateProject/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/StandardTemplateProject/temp/7e529fe7a078cfef.pyi b/StandardTemplateProject/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/StandardTemplateProject/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/StandardTemplateProject/temp/_phase1_manifest.json b/StandardTemplateProject/temp/_phase1_manifest.json new file mode 100644 index 0000000..2a39aff --- /dev/null +++ b/StandardTemplateProject/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\BenchmarkProject\\App\\benchmark_main.py": {"sha1": "f71cf01f35e2325a", "mtime": 1782991404.519475, "size": 233}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/StandardTemplateProject/temp/_sha1_map.txt b/StandardTemplateProject/temp/_sha1_map.txt new file mode 100644 index 0000000..8a78ba7 --- /dev/null +++ b/StandardTemplateProject/temp/_sha1_map.txt @@ -0,0 +1,4 @@ +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +f5522571bcce7bcb:includes/stdint.py +f71cf01f35e2325a:benchmark_main.py diff --git a/StandardTemplateProject/temp/bbdf3bbd4c3bc28c.pyi b/StandardTemplateProject/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/StandardTemplateProject/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/StandardTemplateProject/temp/f5522571bcce7bcb.pyi b/StandardTemplateProject/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/StandardTemplateProject/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/StandardTemplateProject/temp/f6b51804a0ba8ff0.pyi b/StandardTemplateProject/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/StandardTemplateProject/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/StandardTemplateProject/temp/f71cf01f35e2325a.doc.json b/StandardTemplateProject/temp/f71cf01f35e2325a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/StandardTemplateProject/temp/f71cf01f35e2325a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/StandardTemplateProject/temp/f71cf01f35e2325a.pyi b/StandardTemplateProject/temp/f71cf01f35e2325a.pyi new file mode 100644 index 0000000..d500571 --- /dev/null +++ b/StandardTemplateProject/temp/f71cf01f35e2325a.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from benchmark_main.py +Module: benchmark_main +""" + + +import w32.win32console as w32cmd +import t, c + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: pass diff --git a/StubGen.py b/StubGen.py new file mode 100644 index 0000000..c9b2db8 --- /dev/null +++ b/StubGen.py @@ -0,0 +1,10 @@ +#!/usr/bin/env python3 +"""StubGen - C/H/Py 文件到 Python 存根文件生成器（工程化版本）""" +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） + +from lib.StubGen import StubGenConfig, PythonToStubConverter, StubGen, main + +__all__ = ['StubGenConfig', 'PythonToStubConverter', 'StubGen', 'main'] + +if __name__ == '__main__': + main() diff --git a/TODO b/TODO new file mode 100644 index 0000000..9202cba --- /dev/null +++ b/TODO @@ -0,0 +1,34 @@ +更复杂的 泛型 +函数装饰器 +# main 收集（和 导入模块的自动执行） +元类型 +# @p.set +# 函数的描述符用装饰器而不是返回值 +崩溃/返回作用域 回调 +# 假鸭子 +# f-string 自动 snprintf +__doc__ +自动索引 +testcheck默认中文 + +vdebug 工具和 vsdt 工具 +多态 ops 允许单个函数拥有装饰器（可修改，走函数指针） +协程关键字支持 +with 出入多态 +哈希表 +内联快速迭代器 +出栈调用函数 +构造函数隐式参数 +小数组直接栈上创建+ i in [xxx] +字符串索引 + +单 @t.Object +import 自动执行代码 +限制全局 import 符号到局部（其他文件import而本文件没import，但是仍然生效不报错） + +# 统一 mpool 和 mbuddy 接口 +MCViper after 自举 +# Ptr解引用含义冲突，a[x]与ptr[x]冲突 +__iter__ 期望 for a, b in __iter__ + +init_set 构造 \ No newline at end of file diff --git a/Test/ArgparseTest/App/main.py b/Test/ArgparseTest/App/main.py new file mode 100644 index 0000000..cebacfc --- /dev/null +++ b/Test/ArgparseTest/App/main.py @@ -0,0 +1,205 @@ +import t, c +from stdint import * +import stdio +import stdlib +import string +import memhub +import argparse +import testcheck + + +# 全局 MBuddy 指针 +_mbuddy: memhub.MemBuddy | t.CPtr + + +@t.CExport +def main() -> int: + testcheck.begin("ArgparseTest: 命令行参数解析") + + # 初始化内存管理器 + arena: bytes = stdlib.malloc(16 * 1024 * 1024) + if arena is None: + testcheck.fail("malloc arena failed") + return 1 + + _mbuddy = memhub.MemBuddy(arena, 16 * 1024 * 1024) + argparse._mbuddy = _mbuddy + + # 构造模拟 argv: myprog --input test.txt -c 42 --verbose output.dat + argv_buf: bytes = _mbuddy.alloc(8 * 7) # 7 个指针 + argv: str | t.CPtr = argv_buf + + # argv[0] = "myprog" + s0: bytes = _mbuddy.alloc(16) + string.strcpy(s0, "myprog") + argv0: str | t.CPtr = t.CPtr(t.CUInt64T(argv)) + argv0[0] = s0 + + # argv[1] = "--input" + s1: bytes = _mbuddy.alloc(16) + string.strcpy(s1, "--input") + argv1: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 8) + argv1[0] = s1 + + # argv[2] = "test.txt" + s2: bytes = _mbuddy.alloc(16) + string.strcpy(s2, "test.txt") + argv2: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 16) + argv2[0] = s2 + + # argv[3] = "-c" + s3: bytes = _mbuddy.alloc(8) + string.strcpy(s3, "-c") + argv3: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 24) + argv3[0] = s3 + + # argv[4] = "42" + s4: bytes = _mbuddy.alloc(8) + string.strcpy(s4, "42") + argv4: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 32) + argv4[0] = s4 + + # argv[5] = "--verbose" + s5: bytes = _mbuddy.alloc(16) + string.strcpy(s5, "--verbose") + argv5: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 40) + argv5[0] = s5 + + # argv[6] = "output.dat" + s6: bytes = _mbuddy.alloc(16) + string.strcpy(s6, "output.dat") + argv6: str | t.CPtr = t.CPtr(t.CUInt64T(argv) + 48) + argv6[0] = s6 + + argc: INT = 7 + + # ============================================================ + # Test 1: 基本解析 (--input, -c, --verbose, output) + # ============================================================ + testcheck.section("Test 1: 基本解析") + parser = argparse.ArgumentParser("myprog", "A test program for argparse", pool=_mbuddy) + parser.add_argument("--input", "-i", argparse.STRING, 0, None, True, argparse.STORE, "input file path") + parser.add_argument("--count", "-c", argparse.INT, 1, None, False, argparse.STORE, "repeat count") + parser.add_argument("--verbose", "-v", argparse.BOOL, 0, None, False, argparse.STORE_TRUE, "enable verbose output") + parser.add_argument("output", None, argparse.STRING, 0, None, False, argparse.STORE, "output file path") + + args = parser.parse_args(argc, argv) + if args is not None: + testcheck.ok("parse_args returned non-NULL") + inp: str = args.get_str("input") + cnt: INT = args.get_int("count") + verb: INT = args.get_bool("verbose") + out: str = args.get_str("output") + stdio.printf("[DBG] has(verbose)=%d, get_bool(verbose)=%d\n", 1 if args.has("verbose") else 0, 1 if verb else 0) + testcheck.check(string.strcmp(inp, "test.txt") == 0, "input = test.txt", "input mismatch") + testcheck.check(cnt == 42, "count = 42", "count mismatch") + testcheck.check(verb == True, "verbose = True", "verbose mismatch") + testcheck.check(string.strcmp(out, "output.dat") == 0, "output = output.dat", "output mismatch") + argparse.release(args) + else: + testcheck.fail("parse_args returned NULL") + + # ============================================================ + # Test 2: --name=value 语法 + # ============================================================ + testcheck.section("Test 2: --name=value 语法") + + argv2_buf: bytes = _mbuddy.alloc(8 * 3) + argv2: str | t.CPtr = argv2_buf + + a0: bytes = _mbuddy.alloc(16) + string.strcpy(a0, "myprog") + a0_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv2)) + a0_ptr[0] = a0 + + a1: bytes = _mbuddy.alloc(32) + string.strcpy(a1, "--input=hello.txt") + a1_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv2) + 8) + a1_ptr[0] = a1 + + a2: bytes = _mbuddy.alloc(16) + string.strcpy(a2, "output2.dat") + a2_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv2) + 16) + a2_ptr[0] = a2 + + args2 = parser.parse_args(3, argv2) + if args2 is not None: + testcheck.ok("parse_args returned non-NULL") + inp2: str = args2.get_str("input") + cnt2: INT = args2.get_int("count") + out2: str = args2.get_str("output") + testcheck.check(string.strcmp(inp2, "hello.txt") == 0, "input = hello.txt", "input mismatch") + testcheck.check(cnt2 == 1, "count = 1 (default)", "count mismatch") + testcheck.check(string.strcmp(out2, "output2.dat") == 0, "output = output2.dat", "output mismatch") + argparse.release(args2) + else: + testcheck.fail("parse_args returned NULL") + + # ============================================================ + # Test 3: -v COUNT 计数动作 + # ============================================================ + testcheck.section("Test 3: COUNT 计数动作") + + argv3_buf: bytes = _mbuddy.alloc(8 * 4) + argv3_arr: str | t.CPtr = argv3_buf + + b0: bytes = _mbuddy.alloc(8) + string.strcpy(b0, "myprog") + b0_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv3_arr)) + b0_ptr[0] = b0 + + b1: bytes = _mbuddy.alloc(8) + string.strcpy(b1, "-v") + b1_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv3_arr) + 8) + b1_ptr[0] = b1 + + b2: bytes = _mbuddy.alloc(8) + string.strcpy(b2, "-v") + b2_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv3_arr) + 16) + b2_ptr[0] = b2 + + b3: bytes = _mbuddy.alloc(16) + string.strcpy(b3, "output3.dat") + b3_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(argv3_arr) + 24) + b3_ptr[0] = b3 + + parser2 = argparse.ArgumentParser("myprog", "Count test", pool=_mbuddy) + parser2.add_argument("--verbose", "-v", argparse.BOOL, 0, None, False, argparse.COUNT, "verbosity level") + parser2.add_argument("output", None, argparse.STRING, 0, None, False, argparse.STORE, "output file") + + args3 = parser2.parse_args(4, argv3_arr) + if args3 is not None: + testcheck.ok("parse_args returned non-NULL") + verb3: INT = args3.get_int("verbose") + out3: str = args3.get_str("output") + testcheck.check(verb3 == 2, "verbose = 2 (counted)", "verbose count mismatch") + testcheck.check(string.strcmp(out3, "output3.dat") == 0, "output = output3.dat", "output mismatch") + argparse.release(args3) + else: + testcheck.fail("parse_args returned NULL") + + # ============================================================ + # Test 4: 关键字参数调用 add_argument + # ============================================================ + testcheck.section("Test 4: 关键字参数") + + parser3 = argparse.ArgumentParser("myprog", "Keyword arg test", pool=_mbuddy) + parser3.add_argument("--input", short="-i", arg_type=argparse.STRING, default=0, default_str=None, required=False, action=argparse.STORE, help="input file") + parser3.add_argument("--count", short="-c", arg_type=argparse.INT, default=1, required=False, action=argparse.STORE, help="count") + parser3.add_argument("--verbose", short="-v", arg_type=argparse.BOOL, default=0, required=False, action=argparse.STORE_TRUE, help="verbose") + parser3.add_argument("output", arg_type=argparse.STRING, default=0, required=False, action=argparse.STORE, help="output file") + + args4 = parser3.parse_args(argc, argv) + if args4 is not None: + testcheck.ok("parse_args returned non-NULL") + inp4: str = args4.get_str("input") + cnt4: INT = args4.get_int("count") + out4: str = args4.get_str("output") + testcheck.check(string.strcmp(inp4, "test.txt") == 0, "input = test.txt", "input mismatch") + testcheck.check(cnt4 == 42, "count = 42", "count mismatch") + testcheck.check(string.strcmp(out4, "output.dat") == 0, "output = output.dat", "output mismatch") + argparse.release(args4) + else: + testcheck.fail("parse_args returned NULL") + + return testcheck.end() \ No newline at end of file diff --git a/Test/ArgparseTest/output/08ea5281873244fd.deps.json b/Test/ArgparseTest/output/08ea5281873244fd.deps.json new file mode 100644 index 0000000..f0801e0 --- /dev/null +++ b/Test/ArgparseTest/output/08ea5281873244fd.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ArgparseTest/project.json b/Test/ArgparseTest/project.json new file mode 100644 index 0000000..53ad3d6 --- /dev/null +++ b/Test/ArgparseTest/project.json @@ -0,0 +1,30 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ArgparseTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32"], + "output": "ArgparseTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true, + "startup": false + } +} diff --git a/Test/ArgparseTest/temp/08ea5281873244fd.doc.json b/Test/ArgparseTest/temp/08ea5281873244fd.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ArgparseTest/temp/08ea5281873244fd.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ArgparseTest/temp/08ea5281873244fd.pyi b/Test/ArgparseTest/temp/08ea5281873244fd.pyi new file mode 100644 index 0000000..0c46010 --- /dev/null +++ b/Test/ArgparseTest/temp/08ea5281873244fd.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import stdlib +import string +import memhub +import argparse +import testcheck + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +@t.CExport +def main() -> int: pass diff --git a/Test/ArgparseTest/temp/271ea3decb810db2.pyi b/Test/ArgparseTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/ArgparseTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/ArgparseTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ArgparseTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ArgparseTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ArgparseTest/temp/7e529fe7a078cfef.pyi b/Test/ArgparseTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ArgparseTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ArgparseTest/temp/90c53dd6db8d41cf.pyi b/Test/ArgparseTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/ArgparseTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/ArgparseTest/temp/_phase1_manifest.json b/Test/ArgparseTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..eb48d04 --- /dev/null +++ b/Test/ArgparseTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ArgparseTest\\App\\main.py": {"sha1": "08ea5281873244fd", "mtime": 1783074740.4694562, "size": 8086}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\argparse.py": {"sha1": "aba439b7882ad9d6", "mtime": 1783072446.3366802, "size": 16968}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashtable.py": {"sha1": "b8c66c8ff44eb874", "mtime": 1782826619.490043, "size": 9385}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ArgparseTest/temp/_sha1_map.txt b/Test/ArgparseTest/temp/_sha1_map.txt new file mode 100644 index 0000000..0b0b531 --- /dev/null +++ b/Test/ArgparseTest/temp/_sha1_map.txt @@ -0,0 +1,13 @@ +08ea5281873244fd:main.py +271ea3decb810db2:includes/atom.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +aba439b7882ad9d6:includes/argparse.py +b8c66c8ff44eb874:includes/hashtable.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ArgparseTest/temp/ab6e54ba9a669f76.pyi b/Test/ArgparseTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/ArgparseTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/ArgparseTest/temp/aba439b7882ad9d6.pyi b/Test/ArgparseTest/temp/aba439b7882ad9d6.pyi new file mode 100644 index 0000000..124ce31 --- /dev/null +++ b/Test/ArgparseTest/temp/aba439b7882ad9d6.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from argparse.py +Module: argparse +""" + + +import t, c +from stdint import * +import string +import memhub +import stdio +import hashtable + +STRING: t.CDefine = 0 +INT: t.CDefine = 1 +BOOL: t.CDefine = 2 +STORE: t.CDefine = 0 +STORE_TRUE: t.CDefine = 1 +STORE_FALSE: t.CDefine = 2 +COUNT: t.CDefine = 3 +MAX_ARGS: t.CDefine = 32 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class Argument: + name: str + short_name: str + help_text: str + arg_type: INT + action: INT + default_int: INT + default_str: str + required: bool + is_positional: bool + dest: str +class ParsedArgs: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def get_str(self: ParsedArgs, name: str) -> str: pass + def get_int(self: ParsedArgs, name: str) -> INT: pass + def get_bool(self: ParsedArgs, name: str) -> INT: pass + def has(self: ParsedArgs, name: str) -> bool: pass + def _set_str(self: ParsedArgs, name: str, val: str) -> t.CInt: pass + def _set_int(self: ParsedArgs, name: str, val: INT) -> t.CInt: pass + def _set_bool(self: ParsedArgs, name: str, val: bool) -> t.CInt: pass +class ArgumentParser: + _prog: str + _description: str + _args: Argument | t.CPtr + _arg_count: INT + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def add_argument(self: ArgumentParser, name: str, short: str, arg_type: INT, default: INT, default_str: str, required: bool, action: INT, help: str) -> t.CInt: pass + def _get_arg(self: ArgumentParser, idx: INT) -> Argument | t.CPtr: pass + def _find_long(self: ArgumentParser, token: str) -> INT: pass + def _find_long_prefix(self: ArgumentParser, token: str) -> INT: pass + def _find_short(self: ArgumentParser, token: str) -> INT: pass + def _find_positional(self: ArgumentParser, pos_idx: INT) -> INT: pass + def _store_value(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr, value: str) -> t.CInt: pass + def _apply_action(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr) -> t.CInt: pass + def parse_args(self: ArgumentParser, argc: INT, argv: str | t.CPtr) -> ParsedArgs | t.CPtr: pass + def print_help(self: ArgumentParser) -> t.CInt: pass + def print_usage(self: ArgumentParser) -> t.CInt: pass + +def release(args: ParsedArgs | t.CPtr) -> t.CInt: pass diff --git a/Test/ArgparseTest/temp/b8c66c8ff44eb874.pyi b/Test/ArgparseTest/temp/b8c66c8ff44eb874.pyi new file mode 100644 index 0000000..85bdb8f --- /dev/null +++ b/Test/ArgparseTest/temp/b8c66c8ff44eb874.pyi @@ -0,0 +1,46 @@ +""" +Auto-generated Python stub file from hashtable.py +Module: hashtable +""" + + +import t, c +import string +import memhub +from stdint import * + +_HT_EMPTY: t.CDefine = 0 +_HT_TOMBSTONE: t.CDefine = 1 +_HT_INIT_CAP: t.CDefine = 16 +_HT_SLOT_SIZE: t.CDefine = 24 +_FNV_OFFSET: t.CExtern | t.CUInt64T +_FNV_PRIME: t.CExtern | t.CUInt64T + +def _fnv1a_hash(key: str) -> t.CUInt64T: pass + + +class HashTable: + __slots__: t.CVoid | t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 16 + __tombstones__: t.CSizeT = 0 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def _slot_hash(self: HashTable, idx: t.CSizeT) -> t.CUInt64T: pass + def _slot_key(self: HashTable, idx: t.CSizeT) -> str: pass + def _slot_value(self: HashTable, idx: t.CSizeT) -> t.CPtr: pass + def _set_slot(self: HashTable, idx: t.CSizeT, h: t.CUInt64T, key: str, val: t.CPtr) -> t.CInt: pass + def _find_slot(self: HashTable, key: str, h: t.CUInt64T) -> t.CSizeT: pass + def _resize(self: HashTable, new_cap: t.CSizeT) -> t.CInt: pass + def __setitem__(self: HashTable, key: str, val: t.CNeedPtr) -> t.CInt: pass + def __getitem__(self: HashTable, key: str) -> t.CPtr: pass + def __contains__(self: HashTable, key: str) -> t.CInt: pass + def __delitem__(self: HashTable, key: str) -> t.CInt: pass + def __len__(self: HashTable) -> t.CSizeT: pass + def __iter__(self: HashTable) -> HashTable | t.CPtr: pass + def __next__(self: HashTable) -> str: pass + def get(self: HashTable, key: str, default: t.CPtr) -> t.CPtr: pass + def set_int(self: HashTable, key: str, val: int) -> t.CInt: pass + def set_str(self: HashTable, key: str, val: str) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ArgparseTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ArgparseTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ArgparseTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ArgparseTest/temp/c9f4be41ca1cc2b4.pyi b/Test/ArgparseTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/ArgparseTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ArgparseTest/temp/dd3002730623424b.pyi b/Test/ArgparseTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ArgparseTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ArgparseTest/temp/ee084e9fc6ee413a.pyi b/Test/ArgparseTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/ArgparseTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/ArgparseTest/temp/f5522571bcce7bcb.pyi b/Test/ArgparseTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ArgparseTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/AstTest/App/main.py b/Test/AstTest/App/main.py new file mode 100644 index 0000000..d2b3ff6 --- /dev/null +++ b/Test/AstTest/App/main.py @@ -0,0 +1,105 @@ +from stdint import * +import t, c +from t import CInt, CExport +from stdio import printf +import memhub +import stdlib +import sys +import string +import ast +import w32.win32console + + +# MBuddy arena 大小：1MB +ARENA_SIZE: t.CDefine = 1048576 +# mpool 内存大小：512KB +POOL_SIZE: t.CDefine = 524288 +# dump 缓冲区大小：256KB +DUMP_SIZE: t.CDefine = 262144 + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + printf("AST Comprehensive Test Start\n") + + # 初始化全局 MBuddy 内存分配器 + arena: bytes = stdlib.malloc(ARENA_SIZE) + if arena == None: + printf("FAIL: arena alloc failed\n") + return 1 + mbuddy_obj: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, ARENA_SIZE) + if mbuddy_obj == None: + printf("FAIL: mbuddy init failed\n") + return 1 + + sys._mbuddy = mbuddy_obj + ast._mbuddy = mbuddy_obj + + # 为 ast 解析器创建 mpool + pool_mem: bytes = stdlib.malloc(POOL_SIZE) + if pool_mem == None: + printf("FAIL: malloc for pool failed\n") + return 1 + pool: memhub.MemPool | t.CPtr = memhub.MemPool(pool_mem, POOL_SIZE) + if pool == None: + printf("FAIL: memhub.MemPool init failed\n") + return 1 + + # dump 缓冲区（独立分配） + dump_mem: bytes = stdlib.malloc(DUMP_SIZE) + if dump_mem == None: + printf("FAIL: malloc for dump failed\n") + return 1 + dump_buf: t.CChar | t.CPtr = t.CChar(dump_mem, t.CPtr) + + # 综合测试源码：覆盖 CPython 主要语法 + src: str = """x = 1 +y = 2 +z = x + y +a = x - y +b = x * y +c = x / y +d = x == y +e = x < y +f = x > y +g = x and y +h = x or y +i = not x +def foo(a, b): + return a + b +result = foo(x, y) +if x > 0: + print(x) +else: + print(y) +for i in range(10): + print(i) +while x > 0: + x = x - 1 +class Point: + pass +""" + + printf("Parsing comprehensive source...\n") + + # 解析 + tree: ast.AST | t.CPtr = ast.parse(src, pool) + if tree == None: + printf("FAIL: parse returned None\n") + stdlib.free(pool_mem) + stdlib.free(dump_mem) + return 1 + + printf("PASS: parse succeeded\n") + + # dump AST + printf("Calling ast.dump...\n") + ast.dump(tree, dump_buf, DUMP_SIZE) + printf("ast.dump returned, now printing buffer...\n") + printf("=== AST DUMP ===\n%s\n=== END ===\n", dump_buf) + + stdlib.free(pool_mem) + stdlib.free(dump_mem) + return 0 diff --git a/Test/AstTest/output/0853acdbdfc9b27c.deps.json b/Test/AstTest/output/0853acdbdfc9b27c.deps.json new file mode 100644 index 0000000..3e48a90 --- /dev/null +++ b/Test/AstTest/output/0853acdbdfc9b27c.deps.json @@ -0,0 +1 @@ +{"w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "atom": "271ea3decb810db2", "ast.__init__": "34548789d646f432", "__init__": "34548789d646f432", "ast": "34548789d646f432", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "stdint": "f5522571bcce7bcb", "main.tokens": "b547ac4f380bddb6", "main.base": "5dab8cb390496d22", "main.exprs": "47767b5026a8ee15", "main.stmts": "657e182b27c2a022", "main.astaux": "4337fb260448bbe2", "main.match": "4dd6b3f1427d1cc5", "main.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/Test/AstTest/project.json b/Test/AstTest/project.json new file mode 100644 index 0000000..a0e89ed --- /dev/null +++ b/Test/AstTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "AstTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "AstTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/AstTest/temp/0035c95a18d4f8e8.pyi b/Test/AstTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/AstTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/AstTest/temp/067c78e9f121dce3.pyi b/Test/AstTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/AstTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/AstTest/temp/06f53cc594b4ac6c.pyi b/Test/AstTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/AstTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/AstTest/temp/0853acdbdfc9b27c.doc.json b/Test/AstTest/temp/0853acdbdfc9b27c.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/AstTest/temp/0853acdbdfc9b27c.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/AstTest/temp/0853acdbdfc9b27c.pyi b/Test/AstTest/temp/0853acdbdfc9b27c.pyi new file mode 100644 index 0000000..821be2f --- /dev/null +++ b/Test/AstTest/temp/0853acdbdfc9b27c.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +from t import CInt, CExport +from stdio import printf +import memhub +import stdlib +import sys +import string +import ast +import w32.win32console + +ARENA_SIZE: t.CDefine = 1048576 +POOL_SIZE: t.CDefine = 524288 +DUMP_SIZE: t.CDefine = 262144 + +def main() -> CInt | CExport: pass diff --git a/Test/AstTest/temp/0c212981c180e7fb.pyi b/Test/AstTest/temp/0c212981c180e7fb.pyi new file mode 100644 index 0000000..f432b06 --- /dev/null +++ b/Test/AstTest/temp/0c212981c180e7fb.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from ast.lexer.py +Module: ast.lexer +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, TokOp, _init_tables, _kw_lookup, _op_head, OpEntry, new_token, token_set_str, token_set_str_literal + +LEXER_MAX_INDENT: t.CDefine = 256 + +class Lexer: + src: str + pos: t.CSizeT + len: t.CSizeT + lineno: t.CInt + col: t.CInt + pool: memhub.MemManager | t.CPtr + indent_stack: t.CInt | t.CPtr + indent_top: t.CInt + paren_depth: t.CInt + tokens_head: Token | t.CPtr + tokens_tail: Token | t.CPtr + at_line_start: t.CInt + error_count: t.CInt + def __new__(self: Lexer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_lexer(pool: memhub.MemManager | t.CPtr) -> Lexer | t.CPtr: pass + +def _lexer_init(lx: Lexer | t.CPtr, src: str, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(lx: Lexer | t.CPtr, offset: t.CSizeT) -> t.CChar: pass + +def _peek_cur(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _advance(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _is_alpha(ch: t.CChar) -> t.CInt: pass + +def _is_digit(ch: t.CChar) -> t.CInt: pass + +def _is_alnum(ch: t.CChar) -> t.CInt: pass + +def _is_hex(ch: t.CChar) -> t.CInt: pass + +def _is_space(ch: t.CChar) -> t.CInt: pass + +def _emit(lx: Lexer | t.CPtr, tok: Token | t.CPtr) -> t.CInt: pass + +def _make_op_token(lx: Lexer | t.CPtr, op_id: t.CInt, length: t.CInt, lineno: t.CInt, col: t.CInt) -> Token | t.CPtr: pass + +def _handle_indent(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_name(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_number(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _emit_number(lx: Lexer | t.CPtr, start: t.CSizeT, start_lineno: t.CInt, start_col: t.CInt, is_float: t.CInt, is_complex: t.CInt, is_hex: t.CInt) -> t.CInt: pass + +def _read_string(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_operator(lx: Lexer | t.CPtr) -> t.CInt: pass + +def tokenize(lx: Lexer | t.CPtr) -> Token | t.CPtr: pass diff --git a/Test/AstTest/temp/271ea3decb810db2.pyi b/Test/AstTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/AstTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/AstTest/temp/34548789d646f432.pyi b/Test/AstTest/temp/34548789d646f432.pyi new file mode 100644 index 0000000..e046b0c --- /dev/null +++ b/Test/AstTest/temp/34548789d646f432.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from ast.__init__.py +Module: ast.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp, _init_tables +from .base import AST, ASTKind, ASTCtx, OpKind, ASTFlag, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC, FLAG_SIMPLE, FLAG_HAS_STAR, _init_ast, _copy_str, _set_pos, _inherit_pos, _set_parent_list, _binop_from_op, _augop_from_op, _cmpop_from_op, _emit, _emit_str, _emit_int, _dump_list, dump +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .astaux import ExceptHandler, Arguments, Arg, Keyword, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .visitor import ASTVisitor +from .lexer import Lexer, _lexer_init, tokenize, new_lexer +from .parser import Parser, _parser_init, parse_tokens, new_parser +from .parser import _parse_test + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def parse(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass + +def parse_expression(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass diff --git a/Test/AstTest/temp/4337fb260448bbe2.pyi b/Test/AstTest/temp/4337fb260448bbe2.pyi new file mode 100644 index 0000000..7ed88bd --- /dev/null +++ b/Test/AstTest/temp/4337fb260448bbe2.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from ast.astaux.py +Module: ast.astaux +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class ExceptHandler(AST): + type: AST | t.CPtr + name: str + def __new__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: ExceptHandler) -> t.CInt: pass + def type_name(self: ExceptHandler) -> str: pass + def dump(self: ExceptHandler, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arguments(AST): + args: list[AST | t.CPtr] | t.CPtr + vararg: AST | t.CPtr + kwarg: AST | t.CPtr + defaults: list[AST | t.CPtr] | t.CPtr + kw_defaults: list[AST | t.CPtr] | t.CPtr + def __new__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Arguments) -> t.CInt: pass + def type_name(self: Arguments) -> str: pass + def dump(self: Arguments, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arg(AST): + arg: str + annotation: AST | t.CPtr + def __new__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def kind(self: Arg) -> t.CInt: pass + def type_name(self: Arg) -> str: pass + def dump(self: Arg, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Keyword(AST): + arg: str + value: AST | t.CPtr + def __new__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Keyword) -> t.CInt: pass + def type_name(self: Keyword) -> str: pass + def dump(self: Keyword, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Alias(AST): + name: str + asname: str + def __new__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def __init__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def kind(self: Alias) -> t.CInt: pass + def type_name(self: Alias) -> str: pass + def dump(self: Alias, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class WithItem(AST): + context_expr: AST | t.CPtr + optional_vars: AST | t.CPtr + def __new__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def __init__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def kind(self: WithItem) -> t.CInt: pass + def type_name(self: WithItem) -> str: pass + def dump(self: WithItem, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Comprehension(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + ifs: list[AST | t.CPtr] | t.CPtr + is_async: t.CInt + def __new__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: Comprehension) -> t.CInt: pass + def type_name(self: Comprehension) -> str: pass + def dump(self: Comprehension, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/AstTest/temp/47767b5026a8ee15.pyi b/Test/AstTest/temp/47767b5026a8ee15.pyi new file mode 100644 index 0000000..7890f3d --- /dev/null +++ b/Test/AstTest/temp/47767b5026a8ee15.pyi @@ -0,0 +1,241 @@ +""" +Auto-generated Python stub file from ast.exprs.py +Module: ast.exprs +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, _dump_op_list, _op_name, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE + +class Name(AST): + id: str + ctx: t.CInt + def __new__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Name) -> t.CInt: pass + def type_name(self: Name) -> str: pass + def dump(self: Name, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Constant(AST): + int_val: t.CInt64T + float_val: t.CDouble + str_val: str + const_kind: t.CInt + def __new__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def __init__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def kind(self: Constant) -> t.CInt: pass + def type_name(self: Constant) -> str: pass + def dump(self: Constant, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BinOp(AST): + left: AST | t.CPtr + op: t.CInt + right: AST | t.CPtr + def __new__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def __init__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def kind(self: BinOp) -> t.CInt: pass + def type_name(self: BinOp) -> str: pass + def dump(self: BinOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class UnaryOp(AST): + op: t.CInt + operand: AST | t.CPtr + def __new__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def __init__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def kind(self: UnaryOp) -> t.CInt: pass + def type_name(self: UnaryOp) -> str: pass + def dump(self: UnaryOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Compare(AST): + left: AST | t.CPtr + ops: t.CPtr + comparators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Compare) -> t.CInt: pass + def type_name(self: Compare) -> str: pass + def dump(self: Compare, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Call(AST): + func: AST | t.CPtr + args: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + def __new__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Call) -> t.CInt: pass + def type_name(self: Call) -> str: pass + def dump(self: Call, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BoolOp(AST): + op: t.CInt + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: BoolOp) -> t.CInt: pass + def type_name(self: BoolOp) -> str: pass + def dump(self: BoolOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Lambda(AST): + args: AST | t.CPtr + body: AST | t.CPtr + def __new__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Lambda) -> t.CInt: pass + def type_name(self: Lambda) -> str: pass + def dump(self: Lambda, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class IfExp(AST): + test: AST | t.CPtr + body: AST | t.CPtr + orelse: AST | t.CPtr + def __new__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def __init__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def kind(self: IfExp) -> t.CInt: pass + def type_name(self: IfExp) -> str: pass + def dump(self: IfExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Dict(AST): + keys: list[AST | t.CPtr] | t.CPtr + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Dict) -> t.CInt: pass + def type_name(self: Dict) -> str: pass + def dump(self: Dict, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Set(AST): + elts: list[AST | t.CPtr] | t.CPtr + def __new__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Set) -> t.CInt: pass + def type_name(self: Set) -> str: pass + def dump(self: Set, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ListComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: ListComp) -> t.CInt: pass + def type_name(self: ListComp) -> str: pass + def dump(self: ListComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class SetComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: SetComp) -> t.CInt: pass + def type_name(self: SetComp) -> str: pass + def dump(self: SetComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class DictComp(AST): + key: AST | t.CPtr + value: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: DictComp) -> t.CInt: pass + def type_name(self: DictComp) -> str: pass + def dump(self: DictComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class GeneratorExp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: GeneratorExp) -> t.CInt: pass + def type_name(self: GeneratorExp) -> str: pass + def dump(self: GeneratorExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Await(AST): + value: AST | t.CPtr + def __new__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Await) -> t.CInt: pass + def type_name(self: Await) -> str: pass + def dump(self: Await, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Yield(AST): + value: AST | t.CPtr + def __new__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Yield) -> t.CInt: pass + def type_name(self: Yield) -> str: pass + def dump(self: Yield, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class YieldFrom(AST): + value: AST | t.CPtr + def __new__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: YieldFrom) -> t.CInt: pass + def type_name(self: YieldFrom) -> str: pass + def dump(self: YieldFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FormattedValue(AST): + value: AST | t.CPtr + conversion: t.CInt + format_spec: AST | t.CPtr + def __new__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def kind(self: FormattedValue) -> t.CInt: pass + def type_name(self: FormattedValue) -> str: pass + def dump(self: FormattedValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class JoinedStr(AST): + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: JoinedStr) -> t.CInt: pass + def type_name(self: JoinedStr) -> str: pass + def dump(self: JoinedStr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Attribute(AST): + value: AST | t.CPtr + attr: str + ctx: t.CInt + def __new__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Attribute) -> t.CInt: pass + def type_name(self: Attribute) -> str: pass + def dump(self: Attribute, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Subscript(AST): + value: AST | t.CPtr + slice: AST | t.CPtr + ctx: t.CInt + def __new__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Subscript) -> t.CInt: pass + def type_name(self: Subscript) -> str: pass + def dump(self: Subscript, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Starred(AST): + value: AST | t.CPtr + ctx: t.CInt + def __new__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Starred) -> t.CInt: pass + def type_name(self: Starred) -> str: pass + def dump(self: Starred, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class List(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: List) -> t.CInt: pass + def type_name(self: List) -> str: pass + def dump(self: List, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Tuple(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Tuple) -> t.CInt: pass + def type_name(self: Tuple) -> str: pass + def dump(self: Tuple, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Slice(AST): + lower: AST | t.CPtr + upper: AST | t.CPtr + step: AST | t.CPtr + def __new__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def kind(self: Slice) -> t.CInt: pass + def type_name(self: Slice) -> str: pass + def dump(self: Slice, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class NamedExpr(AST): + target: AST | t.CPtr + value: AST | t.CPtr + def __new__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: NamedExpr) -> t.CInt: pass + def type_name(self: NamedExpr) -> str: pass + def dump(self: NamedExpr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class OpNode(AST): + op: t.CInt + def __new__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def __init__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def kind(self: OpNode) -> t.CInt: pass + def type_name(self: OpNode) -> str: pass + def dump(self: OpNode, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/AstTest/temp/4dd6b3f1427d1cc5.pyi b/Test/AstTest/temp/4dd6b3f1427d1cc5.pyi new file mode 100644 index 0000000..dd15aa2 --- /dev/null +++ b/Test/AstTest/temp/4dd6b3f1427d1cc5.pyi @@ -0,0 +1,82 @@ +""" +Auto-generated Python stub file from ast.match.py +Module: ast.match +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class MatchCase(AST): + pattern: AST | t.CPtr + guard: AST | t.CPtr + def __new__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchCase) -> t.CInt: pass + def type_name(self: MatchCase) -> str: pass + def dump(self: MatchCase, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchValue(AST): + value: AST | t.CPtr + def __new__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchValue) -> t.CInt: pass + def type_name(self: MatchValue) -> str: pass + def dump(self: MatchValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSingleton(AST): + value: AST | t.CPtr + def __new__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchSingleton) -> t.CInt: pass + def type_name(self: MatchSingleton) -> str: pass + def dump(self: MatchSingleton, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSequence(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchSequence) -> t.CInt: pass + def type_name(self: MatchSequence) -> str: pass + def dump(self: MatchSequence, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchMapping(AST): + keys: list[AST | t.CPtr] | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + rest: str + def __new__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def __init__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def kind(self: MatchMapping) -> t.CInt: pass + def type_name(self: MatchMapping) -> str: pass + def dump(self: MatchMapping, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchClass(AST): + cls: AST | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + kwd_attrs: list[AST | t.CPtr] | t.CPtr + kwd_patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchClass) -> t.CInt: pass + def type_name(self: MatchClass) -> str: pass + def dump(self: MatchClass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchStar(AST): + name: str + def __new__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchStar) -> t.CInt: pass + def type_name(self: MatchStar) -> str: pass + def dump(self: MatchStar, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchAs(AST): + pattern: AST | t.CPtr + name: str + def __new__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchAs) -> t.CInt: pass + def type_name(self: MatchAs) -> str: pass + def dump(self: MatchAs, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchOr(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchOr) -> t.CInt: pass + def type_name(self: MatchOr) -> str: pass + def dump(self: MatchOr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/AstTest/temp/5dab8cb390496d22.pyi b/Test/AstTest/temp/5dab8cb390496d22.pyi new file mode 100644 index 0000000..7056767 --- /dev/null +++ b/Test/AstTest/temp/5dab8cb390496d22.pyi @@ -0,0 +1,182 @@ +""" +Auto-generated Python stub file from ast.base.py +Module: ast.base +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .tokens import TokOp + +class ASTKind(t.CEnum): + Module: t.State + FunctionDef: t.State + ClassDef: t.State + Assign: t.State + If: t.State + For: t.State + While: t.State + Return: t.State + Expr: t.State + Name: t.State + Constant: t.State + BinOp: t.State + UnaryOp: t.State + Call: t.State + Compare: t.State + Expression: t.State + Interactive: t.State + FunctionType: t.State + Delete: t.State + AugAssign: t.State + AnnAssign: t.State + With: t.State + Raise: t.State + Try: t.State + Assert: t.State + Global: t.State + Nonlocal: t.State + Pass: t.State + Break: t.State + Continue: t.State + Import: t.State + ImportFrom: t.State + Match: t.State + BoolOp: t.State + Lambda: t.State + IfExp: t.State + Dict: t.State + Set: t.State + ListComp: t.State + SetComp: t.State + DictComp: t.State + GeneratorExp: t.State + Await: t.State + Yield: t.State + YieldFrom: t.State + FormattedValue: t.State + JoinedStr: t.State + Attribute: t.State + Subscript: t.State + Starred: t.State + List: t.State + Tuple: t.State + Slice: t.State + NamedExpr: t.State + ExceptHandler: t.State + Arguments: t.State + Arg: t.State + Keyword: t.State + Alias: t.State + WithItem: t.State + Comprehension: t.State + OpNode: t.State + MatchCase: t.State + MatchValue: t.State + MatchSingleton: t.State + MatchSequence: t.State + MatchMapping: t.State + MatchClass: t.State + MatchStar: t.State + MatchAs: t.State + MatchOr: t.State +class ASTCtx(t.CEnum): + Load: t.State + Store: t.State + Del: t.State +class OpKind(t.CEnum): + Add: t.State + Sub: t.State + Mult: t.State + MatMult: t.State + Div: t.State + Mod: t.State + Pow: t.State + LShift: t.State + RShift: t.State + BitOr: t.State + BitXor: t.State + BitAnd: t.State + FloorDiv: t.State + And: t.State + Or: t.State + Eq: t.State + Ne: t.State + Lt: t.State + Le: t.State + Gt: t.State + Ge: t.State + Is: t.State + IsNot: t.State + In: t.State + NotIn: t.State + Not: t.State + UAdd: t.State + USub: t.State + Invert: t.State + NoneOp: t.State + +CONST_INT: t.CDefine = 1 +CONST_FLOAT: t.CDefine = 2 +CONST_STR: t.CDefine = 3 +CONST_BOOL: t.CDefine = 4 +CONST_NONE: t.CDefine = 5 +FLAG_IS_ASYNC: t.CDefine = 1 +FLAG_SIMPLE: t.CDefine = 2 +FLAG_HAS_STAR: t.CDefine = 4 + +class ASTFlag(t.CEnum): + IsAsync: t.State + Simple: t.State + HasStar: t.State + +def _copy_str(pool: memhub.MemManager | t.CPtr, src: str) -> str: pass + +def _set_pos(node: AST | t.CPtr, lineno: t.CInt, col_offset: t.CInt, end_lineno: t.CInt, end_col_offset: t.CInt) -> t.CInt: pass + +def _inherit_pos(node: AST | t.CPtr, ref: AST | t.CPtr) -> t.CInt: pass + +def _init_ast(node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + +def _set_parent_list(lst: list[AST | t.CPtr] | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _append_child(lst: list[AST | t.CPtr] | t.CPtr, node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _binop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _augop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _cmpop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _emit(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_str(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_int(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, val: t.CInt64T) -> t.CSizeT: pass + +def _op_name(op: t.CInt) -> str: pass + +def _dump_list(lst: list[AST | t.CPtr] | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + +def _dump_op_list(lst: t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + + +@t.CVTable +class AST: + parent: AST | t.CPtr + pool: memhub.MemManager | t.CPtr + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + children: list[AST | t.CPtr] | t.CPtr + def kind(self: AST) -> t.CInt: pass + def type_name(self: AST) -> str: pass + def dump(self: AST, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + def accept(self: AST, visitor: t.CPtr) -> t.CInt: pass + def append(self: AST, node: AST | t.CPtr) -> t.CInt: pass + +def dump(node: AST | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: pass diff --git a/Test/AstTest/temp/6446627d4f07a1b5.pyi b/Test/AstTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/AstTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/AstTest/temp/6503c97dde0c79c4.pyi b/Test/AstTest/temp/6503c97dde0c79c4.pyi new file mode 100644 index 0000000..d332326 --- /dev/null +++ b/Test/AstTest/temp/6503c97dde0c79c4.pyi @@ -0,0 +1,189 @@ +""" +Auto-generated Python stub file from posix.py +Module: posix +""" + + +import t, c +from stdint import * + +DIR: t.CTypedef = t.CVoid | t.CPtr +mode_t: t.CTypedef = t.CUInt32T +pid_t: t.CTypedef = t.CInt32T +off_t: t.CTypedef = t.CInt64T +dev_t: t.CTypedef = t.CUInt64T +ino_t: t.CTypedef = t.CUInt64T +nlink_t: t.CTypedef = t.CUInt64T +uid_t: t.CTypedef = t.CUInt32T +gid_t: t.CTypedef = t.CUInt32T +blksize_t: t.CTypedef = t.CInt64T +blkcnt_t: t.CTypedef = t.CInt64T +time_t: t.CTypedef = t.CInt64T +O_RDONLY: t.CDefine = 0 +O_WRONLY: t.CDefine = 1 +O_RDWR: t.CDefine = 2 +O_CREAT: t.CDefine = 0o100 +O_EXCL: t.CDefine = 0o200 +O_TRUNC: t.CDefine = 0o1000 +O_APPEND: t.CDefine = 0o2000 +O_NONBLOCK: t.CDefine = 0o4000 +O_CLOEXEC: t.CDefine = 0o2000000 +F_OK: t.CDefine = 0 # 文件存在 +R_OK: t.CDefine = 4 # 可读 +W_OK: t.CDefine = 2 # 可写 +X_OK: t.CDefine = 1 # 可执行 +S_IFMT: t.CDefine = 0o170000 # 类型掩码 +S_IFSOCK: t.CDefine = 0o140000 # 套接字 +S_IFLNK: t.CDefine = 0o120000 # 符号链接 +S_IFREG: t.CDefine = 0o100000 # 普通文件 +S_IFBLK: t.CDefine = 0o060000 # 块设备 +S_IFDIR: t.CDefine = 0o040000 # 目录 +S_IFCHR: t.CDefine = 0o020000 # 字符设备 +S_IFIFO: t.CDefine = 0o010000 # FIFO/管道 +S_ISUID: t.CDefine = 0o4000 # set-user-ID +S_ISGID: t.CDefine = 0o2000 # set-group-ID +S_ISVTX: t.CDefine = 0o1000 # sticky 位 +S_IRWXU: t.CDefine = 0o700 # 用户读写执行 +S_IRUSR: t.CDefine = 0o400 # 用户读 +S_IWUSR: t.CDefine = 0o200 # 用户写 +S_IXUSR: t.CDefine = 0o100 # 用户执行 +S_IRWXG: t.CDefine = 0o070 # 组读写执行 +S_IRGRP: t.CDefine = 0o040 # 组读 +S_IWGRP: t.CDefine = 0o020 # 组写 +S_IXGRP: t.CDefine = 0o010 # 组执行 +S_IRWXO: t.CDefine = 0o007 # 其他读写执行 +S_IROTH: t.CDefine = 0o004 # 其他读 +S_IWOTH: t.CDefine = 0o002 # 其他写 +S_IXOTH: t.CDefine = 0o001 # 其他执行 +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +PATH_MAX: t.CDefine = 4096 +NAME_MAX: t.CDefine = 255 +_SC_NPROCESSORS_ONLN: t.CDefine = 84 +_SC_PAGESIZE: t.CDefine = 30 +_SC_CLK_TCK: t.CDefine = 2 +DT_UNKNOWN: t.CDefine = 0 +DT_FIFO: t.CDefine = 1 +DT_CHR: t.CDefine = 2 +DT_DIR: t.CDefine = 4 +DT_BLK: t.CDefine = 6 +DT_REG: t.CDefine = 8 +DT_LNK: t.CDefine = 10 +DT_SOCK: t.CDefine = 12 +DT_WHT: t.CDefine = 14 +WNOHANG: t.CDefine = 1 +WUNTRACED: t.CDefine = 2 +WCONTINUED: t.CDefine = 8 + +class Stat: + st_dev: dev_t + st_ino: ino_t + st_nlink: nlink_t + st_mode: mode_t + st_uid: uid_t + st_gid: gid_t + __pad0: INT + st_rdev: dev_t + st_size: off_t + st_blksize: blksize_t + st_blocks: blkcnt_t + st_atim_sec: time_t + st_atim_nsec: LONGLONG + st_mtim_sec: time_t + st_mtim_nsec: LONGLONG + st_ctim_sec: time_t + st_ctim_nsec: LONGLONG + __unused0: LONGLONG + __unused1: LONGLONG + __unused2: LONGLONG +class Dirent: + d_ino: ino_t + d_off: off_t + d_reclen: USHORT + d_type: BYTE + d_name: CHARPTR + +def open(path: t.CConst | str, oflag: INT, *args) -> INT | t.State: pass + +def close(fd: INT) -> INT | t.State: pass + +def read(fd: INT, buf: bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def write(fd: INT, buf: t.CConst | bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def lseek(fd: INT, offset: off_t, whence: INT) -> off_t | t.State: pass + +def unlink(path: t.CConst | str) -> INT | t.State: pass + +def access(path: t.CConst | str, amode: INT) -> INT | t.State: pass + +def realpath(path: t.CConst | str, resolved: CHARPTR) -> CHARPTR | t.State: pass + +def rename(oldpath: t.CConst | str, newpath: t.CConst | str) -> INT | t.State: pass + +def chmod(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def stat(path: t.CConst | str, buf: Stat | t.CPtr) -> INT | t.State: pass + +def fstat(fd: INT, buf: Stat | t.CPtr) -> INT | t.State: pass + +def opendir(name: t.CConst | str) -> DIR | t.CPtr | t.State: pass + +def readdir(dirp: DIR | t.CPtr) -> Dirent | t.CPtr | t.State: pass + +def closedir(dirp: DIR | t.CPtr) -> INT | t.State: pass + +def mkdir(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def rmdir(path: t.CConst | str) -> INT | t.State: pass + +def getcwd(buf: CHARPTR, size: t.CSizeT) -> CHARPTR | t.State: pass + +def chdir(path: t.CConst | str) -> INT | t.State: pass + +def fork() -> pid_t | t.State: pass + +def execvp(file: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def execv(path: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def waitpid(pid: pid_t, status: INT | t.CPtr, options: INT) -> pid_t | t.State: pass + +def pipe(fds: INT | t.CPtr) -> INT | t.State: pass + +def dup2(oldfd: INT, newfd: INT) -> INT | t.State: pass + +def _exit(status: INT) -> t.CVoid | t.State: pass + +def getpid() -> pid_t | t.State: pass + +def getenv(name: t.CConst | str) -> CHARPTR | t.State: pass + +def setenv(name: t.CConst | str, value: t.CConst | str, overwrite: INT) -> INT | t.State: pass + +def sysconf(name: INT) -> LONG | t.State: pass + +def isatty(fd: INT) -> INT | t.State: pass + +def S_ISREG(mode: mode_t) -> bool: pass + +def S_ISDIR(mode: mode_t) -> bool: pass + +def S_ISLNK(mode: mode_t) -> bool: pass + +def S_ISCHR(mode: mode_t) -> bool: pass + +def S_ISBLK(mode: mode_t) -> bool: pass + +def S_ISFIFO(mode: mode_t) -> bool: pass + +def S_ISSOCK(mode: mode_t) -> bool: pass + +def WIFEXITED(status: INT) -> bool: pass + +def WEXITSTATUS(status: INT) -> INT: pass + +def WIFSIGNALED(status: INT) -> bool: pass + +def WTERMSIG(status: INT) -> INT: pass diff --git a/Test/AstTest/temp/657e182b27c2a022.pyi b/Test/AstTest/temp/657e182b27c2a022.pyi new file mode 100644 index 0000000..0a2057d --- /dev/null +++ b/Test/AstTest/temp/657e182b27c2a022.pyi @@ -0,0 +1,226 @@ +""" +Auto-generated Python stub file from ast.stmts.py +Module: ast.stmts +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, FLAG_IS_ASYNC + +@t.CVTable +class Module(AST): + def __new__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Module) -> t.CInt: pass + def type_name(self: Module) -> str: pass + def dump(self: Module, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assign(AST): + targets: list[AST | t.CPtr] | t.CPtr + value: AST | t.CPtr + def __new__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assign) -> t.CInt: pass + def type_name(self: Assign) -> str: pass + def dump(self: Assign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class If(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: If) -> t.CInt: pass + def type_name(self: If) -> str: pass + def dump(self: If, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class For(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: For) -> t.CInt: pass + def type_name(self: For) -> str: pass + def dump(self: For, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class While(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: While) -> t.CInt: pass + def type_name(self: While) -> str: pass + def dump(self: While, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Return(AST): + value: AST | t.CPtr + def __new__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Return) -> t.CInt: pass + def type_name(self: Return) -> str: pass + def dump(self: Return, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expr(AST): + value: AST | t.CPtr + def __new__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expr) -> t.CInt: pass + def type_name(self: Expr) -> str: pass + def dump(self: Expr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionDef(AST): + name: str + args: AST | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + flags: t.CInt + def __new__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: FunctionDef) -> t.CInt: pass + def type_name(self: FunctionDef) -> str: pass + def dump(self: FunctionDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expression(AST): + body: AST | t.CPtr + def __new__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expression) -> t.CInt: pass + def type_name(self: Expression) -> str: pass + def dump(self: Expression, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Interactive(AST): + body: AST | t.CPtr + def __new__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Interactive) -> t.CInt: pass + def type_name(self: Interactive) -> str: pass + def dump(self: Interactive, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionType(AST): + argtypes: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + def __new__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def kind(self: FunctionType) -> t.CInt: pass + def type_name(self: FunctionType) -> str: pass + def dump(self: FunctionType, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ClassDef(AST): + name: str + bases: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + type_params: list[str] | t.CPtr + def __new__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def __init__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def kind(self: ClassDef) -> t.CInt: pass + def type_name(self: ClassDef) -> str: pass + def dump(self: ClassDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Delete(AST): + targets: list[AST | t.CPtr] | t.CPtr + def __new__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Delete) -> t.CInt: pass + def type_name(self: Delete) -> str: pass + def dump(self: Delete, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AugAssign(AST): + target: AST | t.CPtr + op: t.CInt + value: AST | t.CPtr + def __new__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: AugAssign) -> t.CInt: pass + def type_name(self: AugAssign) -> str: pass + def dump(self: AugAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AnnAssign(AST): + target: AST | t.CPtr + annotation: AST | t.CPtr + value: AST | t.CPtr + simple: t.CInt + def __new__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def __init__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def kind(self: AnnAssign) -> t.CInt: pass + def type_name(self: AnnAssign) -> str: pass + def dump(self: AnnAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class With(AST): + items: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: With) -> t.CInt: pass + def type_name(self: With) -> str: pass + def dump(self: With, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Raise(AST): + exc: AST | t.CPtr + cause: AST | t.CPtr + def __new__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def kind(self: Raise) -> t.CInt: pass + def type_name(self: Raise) -> str: pass + def dump(self: Raise, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Try(AST): + handlers: list[AST | t.CPtr] | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + finalbody: list[AST | t.CPtr] | t.CPtr + def __new__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Try) -> t.CInt: pass + def type_name(self: Try) -> str: pass + def dump(self: Try, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assert(AST): + test: AST | t.CPtr + msg: AST | t.CPtr + def __new__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assert) -> t.CInt: pass + def type_name(self: Assert) -> str: pass + def dump(self: Assert, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Global(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Global) -> t.CInt: pass + def type_name(self: Global) -> str: pass + def dump(self: Global, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Nonlocal(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Nonlocal) -> t.CInt: pass + def type_name(self: Nonlocal) -> str: pass + def dump(self: Nonlocal, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Pass(AST): + def __new__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Pass) -> t.CInt: pass + def type_name(self: Pass) -> str: pass + def dump(self: Pass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Break(AST): + def __new__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Break) -> t.CInt: pass + def type_name(self: Break) -> str: pass + def dump(self: Break, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Continue(AST): + def __new__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Continue) -> t.CInt: pass + def type_name(self: Continue) -> str: pass + def dump(self: Continue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Import(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Import) -> t.CInt: pass + def type_name(self: Import) -> str: pass + def dump(self: Import, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ImportFrom(AST): + module: str + names: list[AST | t.CPtr] | t.CPtr + level: t.CInt + def __new__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def __init__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def kind(self: ImportFrom) -> t.CInt: pass + def type_name(self: ImportFrom) -> str: pass + def dump(self: ImportFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Match(AST): + subject: AST | t.CPtr + cases: list[AST | t.CPtr] | t.CPtr + def __new__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Match) -> t.CInt: pass + def type_name(self: Match) -> str: pass + def dump(self: Match, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/AstTest/temp/6f62fe05c5ea1ceb.pyi b/Test/AstTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/AstTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/AstTest/temp/71e0a3ffcb3ebfad.pyi b/Test/AstTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/AstTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/AstTest/temp/72e2d5ccb7cedcf1.pyi b/Test/AstTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/AstTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/AstTest/temp/7e529fe7a078cfef.pyi b/Test/AstTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/AstTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/AstTest/temp/90c53dd6db8d41cf.pyi b/Test/AstTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/AstTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/AstTest/temp/93c1d18e35d188d6.pyi b/Test/AstTest/temp/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/Test/AstTest/temp/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/Test/AstTest/temp/_phase1_manifest.json b/Test/AstTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..5ea6ddb --- /dev/null +++ b/Test/AstTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\AstTest\\App\\main.py": {"sha1": "0853acdbdfc9b27c", "mtime": 1782892830.6294758, "size": 2439}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\__init__.py": {"sha1": "34548789d646f432", "mtime": 1784102404.6307266, "size": 3409}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\astaux.py": {"sha1": "4337fb260448bbe2", "mtime": 1784006094.7981195, "size": 11489}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\base.py": {"sha1": "5dab8cb390496d22", "mtime": 1784197698.3282094, "size": 14966}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\exprs.py": {"sha1": "47767b5026a8ee15", "mtime": 1783239758.809719, "size": 40615}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\lexer.py": {"sha1": "0c212981c180e7fb", "mtime": 1783400609.908198, "size": 23535}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\match.py": {"sha1": "4dd6b3f1427d1cc5", "mtime": 1782961799.8492737, "size": 12309}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\parser.py": {"sha1": "ee52c08239684346", "mtime": 1784102405.1933146, "size": 85445}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\stmts.py": {"sha1": "657e182b27c2a022", "mtime": 1783929993.759902, "size": 37926}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\tokens.py": {"sha1": "b547ac4f380bddb6", "mtime": 1783165480.469121, "size": 13722}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\platmacro.py": {"sha1": "93c1d18e35d188d6", "mtime": 1781237231.03366, "size": 1422}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\posix.py": {"sha1": "6503c97dde0c79c4", "mtime": 1783064077.6376991, "size": 10386}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\sys.py": {"sha1": "b5a965302cded36b", "mtime": 1783569641.07535, "size": 5509}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32process.py": {"sha1": "067c78e9f121dce3", "mtime": 1781200703.5409546, "size": 4802}} \ No newline at end of file diff --git a/Test/AstTest/temp/_sha1_map.txt b/Test/AstTest/temp/_sha1_map.txt new file mode 100644 index 0000000..423f2d9 --- /dev/null +++ b/Test/AstTest/temp/_sha1_map.txt @@ -0,0 +1,25 @@ +067c78e9f121dce3:includes/w32\win32process.py +0853acdbdfc9b27c:main.py +0c212981c180e7fb:includes/ast\lexer.py +271ea3decb810db2:includes/atom.py +34548789d646f432:includes/ast\__init__.py +4337fb260448bbe2:includes/ast\astaux.py +47767b5026a8ee15:includes/ast\exprs.py +4dd6b3f1427d1cc5:includes/ast\match.py +5dab8cb390496d22:includes/ast\base.py +6503c97dde0c79c4:includes/posix.py +657e182b27c2a022:includes/ast\stmts.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +93c1d18e35d188d6:includes/platmacro.py +ab6e54ba9a669f76:includes/string.py +b547ac4f380bddb6:includes/ast\tokens.py +b5a965302cded36b:includes/sys.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +ee084e9fc6ee413a:includes/memhub.py +ee52c08239684346:includes/ast\parser.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/AstTest/temp/ab6e54ba9a669f76.pyi b/Test/AstTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/AstTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/AstTest/temp/b547ac4f380bddb6.pyi b/Test/AstTest/temp/b547ac4f380bddb6.pyi new file mode 100644 index 0000000..e960341 --- /dev/null +++ b/Test/AstTest/temp/b547ac4f380bddb6.pyi @@ -0,0 +1,166 @@ +""" +Auto-generated Python stub file from ast.tokens.py +Module: ast.tokens +""" + + +import t, c +from stdint import * +import memhub +import string + +class TokenType(t.CEnum): + EndMarker: t.State + NewLine: t.State + Indent: t.State + Dedent: t.State + Name: t.State + Number: t.State + String: t.State + FStringStart: t.State + FStringMiddle: t.State + FStringEnd: t.State + Op: t.State + Nl: t.State + Comment: t.State + Encoding: t.State +class Keyword(t.CEnum): + NotKeyword: t.State + False_: t.State + None_: t.State + True_: t.State + And: t.State + As: t.State + Assert: t.State + Async: t.State + Await: t.State + Break: t.State + Class: t.State + Continue: t.State + Def: t.State + Del: t.State + Elif: t.State + Else: t.State + Except: t.State + Finally: t.State + For: t.State + From: t.State + Global: t.State + If: t.State + Import: t.State + In: t.State + Is: t.State + Lambda: t.State + Nonlocal: t.State + Not: t.State + Or: t.State + Pass: t.State + Raise: t.State + Return: t.State + Try: t.State + While: t.State + With: t.State + Yield: t.State + Match: t.State + Case: t.State + Type: t.State +class TokOp(t.CEnum): + LPar: t.State + RPar: t.State + Lsqb: t.State + Rsqb: t.State + LBrace: t.State + RBrace: t.State + Comma: t.State + Colon: t.State + Dot: t.State + Semi: t.State + At: t.State + Equal: t.State + RArrow: t.State + PlusEq: t.State + MinusEq: t.State + StarEq: t.State + SlashEq: t.State + DSlashEq: t.State + PercentEq: t.State + AtEq: t.State + AmpEq: t.State + VBarEq: t.State + CaretEq: t.State + GtGtEq: t.State + LtLtEq: t.State + StarEqEq: t.State + DSlash: t.State + StarStar: t.State + LtLt: t.State + GtGt: t.State + LessEq: t.State + GreaterEq: t.State + EqEq: t.State + ExclaimEq: t.State + Less: t.State + Greater: t.State + Plus: t.State + Minus: t.State + Star: t.State + Slash: t.State + Percent: t.State + Amp: t.State + VBar: t.State + Caret: t.State + Tilde: t.State + ColonEq: t.State + Ellipsis: t.State + Bang: t.State +class Token: + type: t.CInt + op_subtype: t.CInt + kw_subtype: t.CInt + str_val: str + int_val: t.CInt64T + float_val: t.CDouble + is_float: t.CInt + is_complex: t.CInt + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + next: Token | t.CPtr + def __new__(self: Token, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass +class KwEntry: + name: str + kw_id: t.CInt + next: KwEntry | t.CPtr + def __new__(self: KwEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_kw_head: t.CExtern | KwEntry | t.CPtr +_kw_pool: t.CExtern | memhub.MemManager | t.CPtr + +def _kw_intern(pool: memhub.MemManager | t.CPtr, name: str, kw_id: t.CInt) -> t.CInt: pass + +def _kw_lookup(name: str) -> t.CInt: pass + +def _init_keywords(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + + +class OpEntry: + name: str + op_id: t.CInt + length: t.CInt + next: OpEntry | t.CPtr + def __new__(self: OpEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_op_head: t.CExtern | OpEntry | t.CPtr + +def _op_intern(pool: memhub.MemManager | t.CPtr, name: str, op_id: t.CInt) -> t.CInt: pass + +def _init_operators(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _init_tables(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_token(pool: memhub.MemManager | t.CPtr, ttype: t.CInt, lineno: t.CInt, col_offset: t.CInt) -> Token | t.CPtr: pass + +def token_set_str(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, src: str, start: t.CSizeT, length: t.CSizeT) -> t.CInt: pass + +def token_set_str_literal(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, text: str) -> t.CInt: pass diff --git a/Test/AstTest/temp/b5a965302cded36b.pyi b/Test/AstTest/temp/b5a965302cded36b.pyi new file mode 100644 index 0000000..bfdb955 --- /dev/null +++ b/Test/AstTest/temp/b5a965302cded36b.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from sys.py +Module: sys +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import stdio +import w32.win32base +import w32.win32process +import posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def exit(status: INT) -> VOID | t.CExtern | t.CExport: pass + + +maxsize: t.CDefine = 9223372036854775807 + +def platform() -> str: pass + +def get_pid() -> int: pass + + +_argc: t.CExtern | INT +_argv: t.CExtern | str | t.CPtr + +def _init_argv() -> t.CInt: pass + +def _init_argv_win32() -> t.CInt: pass + +def _init_argv_posix() -> t.CInt: pass + +def argc() -> int: pass + +def argv(index: int) -> str: pass diff --git a/Test/AstTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/AstTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/AstTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/AstTest/temp/c3b259b4059f8668.pyi b/Test/AstTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/AstTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/AstTest/temp/c9f4be41ca1cc2b4.pyi b/Test/AstTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/AstTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/AstTest/temp/ee084e9fc6ee413a.pyi b/Test/AstTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/AstTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/AstTest/temp/ee52c08239684346.pyi b/Test/AstTest/temp/ee52c08239684346.pyi new file mode 100644 index 0000000..ee09339 --- /dev/null +++ b/Test/AstTest/temp/ee52c08239684346.pyi @@ -0,0 +1,215 @@ +""" +Auto-generated Python stub file from ast.parser.py +Module: ast.parser +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp +from .base import AST, ASTCtx, ASTKind, OpKind, ASTFlag, _set_pos, _binop_from_op, _augop_from_op, _cmpop_from_op, _copy_str, _inherit_pos, _init_ast, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .astaux import ExceptHandler, Arguments, Arg, Keyword as KwNode, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .lexer import Lexer, _lexer_init, tokenize, new_lexer + +class Parser: + tokens: Token | t.CPtr + cur: Token | t.CPtr + pool: memhub.MemManager | t.CPtr + error_count: t.CInt + pending_decorators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Parser, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_parser(pool: memhub.MemManager | t.CPtr) -> Parser | t.CPtr: pass + +def _parser_init(ps: Parser | t.CPtr, tokens: Token | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _advance(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_type(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_kw(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_str(ps: Parser | t.CPtr) -> str: pass + +def _cur_lineno(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_col(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nl(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nls_and_newlines(ps: Parser | t.CPtr) -> t.CInt: pass + +def _expect_type(ps: Parser | t.CPtr, ttype: t.CInt) -> t.CInt: pass + +def _expect_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _match_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _accept_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _new_const_int(pool: memhub.MemManager | t.CPtr, val: t.CInt64T, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_float(pool: memhub.MemManager | t.CPtr, val: t.CDouble, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_str(pool: memhub.MemManager | t.CPtr, val: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_bool(pool: memhub.MemManager | t.CPtr, val: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_none(pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _set_ctx(node: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + +def _fix_store_ctx(node: AST | t.CPtr) -> t.CInt: pass + +def _parse_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_namedexpr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_lambda(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_or_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_and_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_not_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_comparison(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _match_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _advance_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _parse_bitor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitxor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitand(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_shift(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_arith(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_term(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_factor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_power(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_call(ps: Parser | t.CPtr, func: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_subscript(ps: Parser | t.CPtr, value: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_slice(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_paren(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_brace(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sub_expr(ps: Parser | t.CPtr, src: str) -> AST | t.CPtr: pass + +def _parse_fstring(ps: Parser | t.CPtr, content: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comprehension(ps: Parser | t.CPtr, elt: AST | t.CPtr, vtype: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comp_for(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_target_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_target(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_yield(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_varargslist(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_arg_def(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_suite_into_children(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_statement_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_simple_stmt_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_small_stmt(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _maybe_tuple(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_testlist(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_expr_stmt(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_return(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_raise(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_global(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_nonlocal(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_from_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_alias_list(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_del(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_assert(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_compound(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if_tail(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_while(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_for(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_try(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_except_handler(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_with(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_with_item(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_funcdef(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_classdef(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_match(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _make_name_ast(pool: memhub.MemManager | t.CPtr, node: Name | t.CPtr) -> AST | t.CPtr: pass + +def _parse_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_closed_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sequence_pattern(ps: Parser | t.CPtr, close_op: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_mapping_pattern(ps: Parser | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_class_pattern(ps: Parser | t.CPtr, cls: AST | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_match_case(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_decorators(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_module_body(ps: Parser | t.CPtr, module_node: AST | t.CPtr) -> t.CInt: pass + +def parse_tokens(pool: memhub.MemManager | t.CPtr, tokens: Token | t.CPtr) -> AST | t.CPtr: pass diff --git a/Test/AstTest/temp/f5522571bcce7bcb.pyi b/Test/AstTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/AstTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/AstTest/temp/f6b51804a0ba8ff0.pyi b/Test/AstTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/AstTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/BuddyTest/App/main.py b/Test/BuddyTest/App/main.py new file mode 100644 index 0000000..83c21b0 --- /dev/null +++ b/Test/BuddyTest/App/main.py @@ -0,0 +1,12 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import mbuddytest + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + return mbuddytest.mbuddy_main() diff --git a/Test/BuddyTest/App/mbuddytest.py b/Test/BuddyTest/App/mbuddytest.py new file mode 100644 index 0000000..08c2dd0 --- /dev/null +++ b/Test/BuddyTest/App/mbuddytest.py @@ -0,0 +1,535 @@ +import t, c +from t import CInt, CPtr, CChar, CUInt8T, CUInt32T, CUInt64T, CSizeT +import viperlib +import viperio +import stdlib +import memhub +import string +import testcheck +import w32.win32process +import w32.win32sync +import w32.win32base + + +# 线程函数: 在共享 MBuddy 上执行 alloc/free (测试 spinlock 线程安全) +def _thread_alloc_free(param: t.CVoid | t.CPtr) -> t.CUInt32T: + bd: memhub.MemBuddy | t.CPtr = (memhub.MemBuddy | t.CPtr)(param) + i: CInt + for i in range(100): + p: bytes = bd.alloc(32) + if p != None: + p_c: CUInt8T | CPtr = CPtr(p) + p_c[0] = CUInt8T(i & 0xFF) + bd.free(p) + return 0 + + +def mbuddy_main() -> CInt: + buf: bytes = stdlib.malloc(256) + + testcheck.begin("BuddyTest: 伙伴系统内存分配器测试") + + # === 基础 alloc/free 测试 === + testcheck.section("基础 alloc/free") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena, 65536) + + viperlib.snprintf(buf, 256, "stats=%lu, free_count=%lu", bd.stats(), bd.free_count()) + testcheck.info(buf) + + p1: bytes = bd.alloc(16) + p2: bytes = bd.alloc(64) + p3: bytes = bd.alloc(256) + testcheck.check(p1 != None and p2 != None and p3 != None, + "3 allocs OK", "alloc FAILED") + + # 写入数据验证指针可用 + p_c: CUInt8T | CPtr = CPtr(p1) + p_c[0] = CUInt8T(0xAB) + p_c[1] = CUInt8T(0xCD) + viperlib.snprintf(buf, 256, "p1[0]=0x%x p1[1]=0x%x", p_c[0], p_c[1]) + testcheck.info(buf) + testcheck.check(p_c[0] == CUInt8T(0xAB) and p_c[1] == CUInt8T(0xCD), + "write/read OK", "write/read FAILED") + + bd.free(p1) + bd.free(p2) + bd.free(p3) + testcheck.ok("3 frees OK") + + # 释放后验证合并: free_count 应回到 1 (单个大块) + testcheck.check(bd.free_count() == 1, + "coalesce: free_count=1 OK", "coalesce: free_count != 1") + + # 合并后应能分配大块 + big: bytes = bd.alloc(16000) + testcheck.check(big != None, "big alloc OK", "big alloc FAILED") + bd.free(big) + + stdlib.free(arena) + + # === calloc 测试 (验证清零) === + testcheck.section("calloc 清零") + arena2: bytes = stdlib.malloc(4096) + bd2: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena2, 4096) + + dirty: bytes = bd2.alloc(64) + d_c: CUInt8T | CPtr = CPtr(dirty) + i: CInt + for i in range(64): + d_c[i] = CUInt8T(0xFF) + bd2.free(dirty) + + cl: bytes = bd2.calloc(8, 8) + if cl != None: + cl_c: CUInt8T | CPtr = CPtr(cl) + all_zero: CInt = 1 + j: CInt + for j in range(64): + if cl_c[j] != 0: + all_zero = 0 + break + testcheck.check(all_zero == 1, "calloc zeroed OK", "calloc NOT zeroed") + bd2.free(cl) + else: + testcheck.fail("calloc FAILED") + + stdlib.free(arena2) + + # === realloc 扩展测试 === + testcheck.section("realloc 扩展") + arena3: bytes = stdlib.malloc(4096) + bd3: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena3, 4096) + + rp: bytes = bd3.alloc(32) + rp_c: CUInt8T | CPtr = CPtr(rp) + k: CInt + for k in range(32): + rp_c[k] = CUInt8T(k + 1) + + rp2: bytes = bd3.realloc(rp, 128) + if rp2 != None: + rp2_c: CUInt8T | CPtr = CPtr(rp2) + ok: CInt = 1 + m: CInt + for m in range(32): + if rp2_c[m] != CUInt8T(m + 1): + ok = 0 + break + testcheck.check(ok == 1, "realloc grow data preserved OK", "realloc grow data LOST") + bd3.free(rp2) + else: + testcheck.fail("realloc grow FAILED") + + stdlib.free(arena3) + + # === 原地收缩测试 (验证无泄漏) === + testcheck.section("原地收缩 (realloc shrink)") + arena_shrink: bytes = stdlib.malloc(4096) + bds: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_shrink, 4096) + + sp: bytes = bds.alloc(256) + testcheck.check(sp != None, "alloc 256 OK", "alloc 256 FAILED") + + fc_before: CSizeT = bds.free_count() + # realloc 到更小尺寸: 256+8=264 -> order 4 (512); 32+8=40 -> order 1 (64) + # new_order(1) <= old_order(4), 应原地返回同一指针 + sp2: bytes = bds.realloc(sp, 32) + fc_after: CSizeT = bds.free_count() + + testcheck.check(sp2 == sp, "shrink same ptr OK", "shrink ptr changed") + testcheck.check(fc_after == fc_before, "shrink no leak OK", "shrink LEAKED blocks") + + # 验证数据仍可读写 + sp_c: CUInt8T | CPtr = CPtr(sp2) + sp_c[0] = CUInt8T(0x42) + testcheck.check(sp_c[0] == CUInt8T(0x42), "shrink data OK", "shrink data FAILED") + + bds.free(sp2) + testcheck.check(bds.free_count() == 1, "shrink+free coalesce OK", "shrink+free coalesce FAILED") + + stdlib.free(arena_shrink) + + # === 边界情况测试 === + testcheck.section("边界情况") + arena4: bytes = stdlib.malloc(4096) + bd4: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena4, 4096) + + # alloc(0) 应返回 None + z: bytes = bd4.alloc(0) + testcheck.check(z == None, "alloc(0)=None OK", "alloc(0) should be None") + + # free(None) 应安全 + bd4.free(None) + testcheck.ok("free(None) safe") + + # realloc(None, size) 等价于 alloc + rn: bytes = bd4.realloc(None, 32) + testcheck.check(rn != None, "realloc(None)=alloc OK", "realloc(None) FAILED") + bd4.free(rn) + + # realloc(ptr, 0) 等价于 free + rz: bytes = bd4.alloc(32) + rz2: bytes = bd4.realloc(rz, 0) + testcheck.check(rz2 == None, "realloc(ptr,0)=free OK", "realloc(ptr,0) should be None") + + stdlib.free(arena4) + + # === 非法指针防护测试 === + testcheck.section("非法指针防护") + arena5: bytes = stdlib.malloc(4096) + bd5: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena5, 4096) + + valid: bytes = bd5.alloc(32) + testcheck.check(valid != None, "alloc for illegal test OK", "alloc FAILED") + + # 1. free 随机地址 (地址 1) + bd5.free(t.CVoid(t.CUInt64T(1), t.CPtr)) + testcheck.ok("free(addr=1) ignored") + + # 2. free NULL + bd5.free(None) + testcheck.ok("free(NULL) ignored") + + # 3. free 越界指针 (arena 之前) + bd5.free(t.CVoid(t.CUInt64T(arena5), t.CPtr)) + testcheck.ok("free(before arena) ignored") + + # 4. free 越界指针 (arena 之后) + bd5.free(t.CVoid(t.CUInt64T(arena5) + 99999, t.CPtr)) + testcheck.ok("free(after arena) ignored") + + # 5. free 未对齐指针 (arena + 1, 在 mem 区域内但不对齐) + bd5.free(t.CVoid(t.CUInt64T(bd5.mem) + 1, t.CPtr)) + testcheck.ok("free(misaligned) ignored") + + # 6. 验证合法 free 仍正常工作 (释放后应能再次分配) + bd5.free(valid) + retry5: bytes = bd5.alloc(32) + testcheck.check(retry5 != None, "valid free still works OK", "valid free BROKEN") + if retry5 != None: + bd5.free(retry5) + + # 7. realloc 非法指针应返回 None + r_illegal: bytes = bd5.realloc(t.CVoid(t.CUInt64T(1), t.CPtr), 64) + testcheck.check(r_illegal == None, "realloc(illegal)=None OK", "realloc(illegal) should be None") + + stdlib.free(arena5) + + # === OOM 内存耗尽测试 === + testcheck.section("OOM 内存耗尽") + arena6: bytes = stdlib.malloc(4096) + bd6: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena6, 4096) + + # 持续分配直到失败 + alloc_count: CInt = 0 + ptrs_arr: bytes = stdlib.malloc(8 * 256) # 存储指针的数组 + ptrs: CUInt64T | CPtr = CPtr(ptrs_arr) + + while alloc_count < 256: + p: bytes = bd6.alloc(32) + if p == None: + break + ptrs[alloc_count] = t.CUInt64T(p) + alloc_count += 1 + + viperlib.snprintf(buf, 256, "allocated %d blocks before OOM", alloc_count) + testcheck.info(buf) + testcheck.check(alloc_count > 0, "OOM: got some allocs OK", "OOM: no allocs at all") + + # 尝试再分配应失败 + oom_p: bytes = bd6.alloc(32) + testcheck.check(oom_p == None, "OOM: alloc fails OK", "OOM: alloc should fail") + + # 释放一个块后应能再次分配 + if alloc_count > 0: + first_ptr: bytes = t.CVoid(ptrs[0], t.CPtr) + bd6.free(first_ptr) + retry: bytes = bd6.alloc(32) + testcheck.check(retry != None, "OOM: free+alloc OK", "OOM: free+alloc FAILED") + if retry != None: + bd6.free(retry) + + # 释放所有块 + idx: CInt = 1 + while idx < alloc_count: + cur_p: bytes = t.CVoid(ptrs[idx], t.CPtr) + bd6.free(cur_p) + idx += 1 + + # 全部释放后应能合并回单个大块 + testcheck.check(bd6.free_count() == 1, + "OOM: full coalesce OK", "OOM: free_count != 1 after full free") + + stdlib.free(ptrs_arr) + stdlib.free(arena6) + + # === 空闲链表状态测试 === + testcheck.section("空闲链表状态") + arena7: bytes = stdlib.malloc(4096) + bd7: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena7, 4096) + + # 初始: 1 个大块 + testcheck.check(bd7.free_count() == 1, "init: 1 free block OK", "init: free_count != 1") + + # 分配后: 空闲块数应增加 (分裂产生多个小块) + a1: bytes = bd7.alloc(32) + fc1: CSizeT = bd7.free_count() + viperlib.snprintf(buf, 256, "after alloc(32): free_count=%lu", fc1) + testcheck.info(buf) + testcheck.check(fc1 > 1, "after alloc: split OK", "after alloc: no split") + + # 再分配 + a2: bytes = bd7.alloc(32) + fc2: CSizeT = bd7.free_count() + testcheck.check(fc2 >= 1, "after 2nd alloc: free >= 1 OK", "after 2nd alloc: free < 1") + + # 释放 a1: 空闲块数应增加 + bd7.free(a1) + fc3: CSizeT = bd7.free_count() + testcheck.check(fc3 > fc2, "after free a1: free increased OK", "after free a1: free not increased") + + # 释放 a2: 应合并, 空闲块数回到 1 + bd7.free(a2) + testcheck.check(bd7.free_count() == 1, + "after free all: coalesced to 1 OK", "after free all: free_count != 1") + + stdlib.free(arena7) + + # === with 上下文管理器测试 === + testcheck.section("with 上下文管理器") + arena8: bytes = stdlib.malloc(4096) + with memhub.MemBuddy(arena8, 4096) as wbd: + wp1: bytes = wbd.alloc(32) + wp2: bytes = wbd.alloc(64) + testcheck.check(wp1 != None and wp2 != None, + "with: allocs OK", "with: allocs FAILED") + # 退出 with 块时 __exit__ 自动 reset + + # 退出后重新进入，验证 reset 生效 + with memhub.MemBuddy(arena8, 4096) as wbd2: + wp3: bytes = wbd2.alloc(1024) + testcheck.check(wp3 != None, "with: reset+realloc OK", "with: reset+realloc FAILED") + + stdlib.free(arena8) + + # === 双重 free 检测测试 === + testcheck.section("双重 free 检测") + arena_df: bytes = stdlib.malloc(4096) + bd_df: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_df, 4096) + + df: bytes = bd_df.alloc(64) + testcheck.check(df != None, "df: alloc OK", "df: alloc FAILED") + + # 第一次 free + bd_df.free(df) + fc_df1: CSizeT = bd_df.free_count() + testcheck.ok("first free OK") + + # 第二次 free (双重 free) — 应被静默拒绝 + bd_df.free(df) + fc_df2: CSizeT = bd_df.free_count() + testcheck.check(fc_df2 == fc_df1, "double free rejected OK", "double free CORRUPTED state") + + # 验证分配器仍正常工作 + df2: bytes = bd_df.alloc(64) + testcheck.check(df2 != None, "df: alloc after double-free OK", "df: alloc after double-free FAILED") + if df2 != None: + bd_df.free(df2) + + stdlib.free(arena_df) + + # === 自检功能测试 === + testcheck.section("自检功能") + arena_sc: bytes = stdlib.malloc(4096) + bd_sc: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_sc, 4096) + + # 初始状态自检 + testcheck.check(bd_sc.self_check() == 0, "self_check init OK", "self_check init FAILED") + + # 分配后自检 + sc_p1: bytes = bd_sc.alloc(32) + sc_p2: bytes = bd_sc.alloc(128) + testcheck.check(bd_sc.self_check() == 0, "self_check after alloc OK", "self_check after alloc FAILED") + + # 部分释放后自检 + bd_sc.free(sc_p1) + testcheck.check(bd_sc.self_check() == 0, "self_check after partial free OK", "self_check after partial free FAILED") + + # 全部释放后自检 (应合并回单个大块) + bd_sc.free(sc_p2) + testcheck.check(bd_sc.self_check() == 0, "self_check after full free OK", "self_check after full free FAILED") + + # 双重 free 后自检 (验证状态未损坏) + sc_p3: bytes = bd_sc.alloc(64) + bd_sc.free(sc_p3) + bd_sc.free(sc_p3) # 双重 free，应被拒绝 + testcheck.check(bd_sc.self_check() == 0, "self_check after double-free OK", "self_check after double-free CORRUPTED") + + stdlib.free(arena_sc) + + # === 并发多线程测试 (spinlock) === + testcheck.section("并发多线程 (spinlock)") + arena_mt: bytes = stdlib.malloc(65536) + bd_mt: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_mt, 65536) + + NUM_THREADS: CInt = 4 + handles_arr: bytes = stdlib.malloc(8 * 4) + handles: CUInt64T | CPtr = CPtr(handles_arr) + tid: CUInt32T = 0 + + # 创建 4 个线程并发 alloc/free + t_idx: CInt + for t_idx in range(NUM_THREADS): + h: bytes = w32.win32process.CreateThread(None, 0, _thread_alloc_free, t.CVoid(t.CUInt64T(bd_mt), t.CPtr), 0, c.Addr(tid)) + handles[t_idx] = CUInt64T(h) + + # 等待所有线程完成 + for t_idx in range(NUM_THREADS): + h: bytes = t.CVoid(handles[t_idx], t.CPtr) + w32.win32sync.WaitForSingleObject(h, w32.win32base.INFINITE) + w32.win32base.CloseHandle(h) + + # 验证分配器状态: 自检通过，所有块已合并 + testcheck.check(bd_mt.self_check() == 0, "mt: self_check OK", "mt: self_check FAILED") + testcheck.check(bd_mt.free_count() == 1, "mt: free_count=1 OK", "mt: free_count != 1 (leaked)") + + viperlib.snprintf(buf, 256, "mt: free_count=%lu", bd_mt.free_count()) + testcheck.info(buf) + + stdlib.free(handles_arr) + stdlib.free(arena_mt) + + # === 极端边界测试 === + testcheck.section("极端边界") + + # 1. 分配尺寸恰巧是块边界 + arena_ex: bytes = stdlib.malloc(4096) + bd_ex: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_ex, 4096) + + # 24 + 8 = 32 = MIN_BLOCK, 恰好填满 order 0 块 + exact_fit: bytes = bd_ex.alloc(24) + testcheck.check(exact_fit != None, "exact fit order 0 OK", "exact fit order 0 FAILED") + + # 25 + 8 = 33 > 32, 需要 order 1 (64 字节) + over_fit: bytes = bd_ex.alloc(25) + testcheck.check(over_fit != None, "over by 1 byte OK", "over by 1 byte FAILED") + + bd_ex.free(exact_fit) + bd_ex.free(over_fit) + testcheck.check(bd_ex.free_count() == 1, "boundary: coalesce OK", "boundary: coalesce FAILED") + + # 2. 单次分配接近 arena 上限 + max_usable: CSizeT = bd_ex.mem_size - 8 + viperlib.snprintf(buf, 256, "mem_size=%lu, max_usable=%lu", bd_ex.mem_size, max_usable) + testcheck.info(buf) + + # 恰好用完整个 arena + exact_max: bytes = bd_ex.alloc(max_usable) + testcheck.check(exact_max != None, "exact max alloc OK", "exact max alloc FAILED") + testcheck.check(bd_ex.free_count() == 0, "max alloc: arena full OK", "max alloc: arena not full") + + bd_ex.free(exact_max) + testcheck.check(bd_ex.free_count() == 1, "max free: coalesce OK", "max free: coalesce FAILED") + + # 再次分配同样大小 — 验证合并后可再次使用 + exact_max2: bytes = bd_ex.alloc(max_usable) + testcheck.check(exact_max2 != None, "exact max re-alloc OK", "exact max re-alloc FAILED") + bd_ex.free(exact_max2) + + # 3. 分配超过 arena 上限 — 应失败 + over_max: bytes = bd_ex.alloc(bd_ex.mem_size) + testcheck.check(over_max == None, "over max: fails OK", "over max: should fail") + + over_max2: bytes = bd_ex.alloc(max_usable + 1) + testcheck.check(over_max2 == None, "max+1: fails OK", "max+1: should fail") + + stdlib.free(arena_ex) + + # 4. Arena 太小 (仅够 free_lists 开销, 264 字节) + arena_tiny: bytes = stdlib.malloc(264) + bd_tiny: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_tiny, 264) + testcheck.check(bd_tiny.stats() == 0, "tiny arena: stats=0 OK", "tiny arena: stats != 0") + tiny_p: bytes = bd_tiny.alloc(32) + testcheck.check(tiny_p == None, "tiny arena: alloc fails OK", "tiny arena: should fail") + stdlib.free(arena_tiny) + + # 5. Arena 刚好够一个最小块 (264 + 32 = 296) + arena_one: bytes = stdlib.malloc(296) + bd_one: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_one, 296) + testcheck.check(bd_one.stats() == 32, "one block: stats=32 OK", "one block: stats != 32") + + one_p: bytes = bd_one.alloc(24) + testcheck.check(one_p != None, "one block: alloc OK", "one block: alloc FAILED") + + one_p2: bytes = bd_one.alloc(24) + testcheck.check(one_p2 == None, "one block: OOM OK", "one block: should be OOM") + + bd_one.free(one_p) + testcheck.check(bd_one.free_count() == 1, "one block: free OK", "one block: free FAILED") + stdlib.free(arena_one) + + # 6. 2 的幂 arena vs 非 2 的幂 arena + arena_pow2: bytes = stdlib.malloc(512) + bd_pow2: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_pow2, 512) + # 512 - 264 = 248, largest_pow2(248) = 128 + testcheck.check(bd_pow2.stats() == 128, "pow2 arena: stats=128 OK", "pow2 arena: stats != 128") + stdlib.free(arena_pow2) + + arena_non2: bytes = stdlib.malloc(500) + bd_non2: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_non2, 500) + # 500 - 264 = 236, largest_pow2(236) = 128 + testcheck.check(bd_non2.stats() == 128, "non-pow2 arena: stats=128 OK", "non-pow2 arena: stats != 128") + stdlib.free(arena_non2) + + # 7. 压力测试: 反复 alloc/free 同一尺寸 1000 次 + arena_stress: bytes = stdlib.malloc(4096) + bd_stress: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_stress, 4096) + + s: CInt + for s in range(1000): + sp: bytes = bd_stress.alloc(64) + if sp != None: + sp_c: CUInt8T | CPtr = CPtr(sp) + sp_c[0] = CUInt8T(s & 0xFF) + bd_stress.free(sp) + + testcheck.check(bd_stress.self_check() == 0, "stress: self_check OK", "stress: self_check FAILED") + testcheck.check(bd_stress.free_count() == 1, "stress: free_count=1 OK", "stress: leaked") + stdlib.free(arena_stress) + + # 8. 分配全部最小块然后全部释放 + arena_fill: bytes = stdlib.malloc(4096) + bd_fill: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena_fill, 4096) + + fill_count: CInt = 0 + fill_ptrs_arr: bytes = stdlib.malloc(8 * 256) + fill_ptrs: CUInt64T | CPtr = CPtr(fill_ptrs_arr) + + fc: CInt + for fc in range(256): + fp: bytes = bd_fill.alloc(24) + if fp == None: + break + fill_ptrs[fill_count] = t.CUInt64T(fp) + fill_count += 1 + + viperlib.snprintf(buf, 256, "fill: %d min blocks allocated", fill_count) + testcheck.info(buf) + testcheck.check(fill_count > 0, "fill: got blocks OK", "fill: no blocks") + testcheck.check(bd_fill.self_check() == 0, "fill: self_check OK", "fill: self_check FAILED") + + # 全部释放 + fi: CInt + for fi in range(fill_count): + fp2: bytes = t.CVoid(fill_ptrs[fi], t.CPtr) + bd_fill.free(fp2) + + testcheck.check(bd_fill.free_count() == 1, "fill: all freed OK", "fill: leaked after free all") + testcheck.check(bd_fill.self_check() == 0, "fill: self_check after free OK", "fill: self_check FAILED") + + stdlib.free(fill_ptrs_arr) + stdlib.free(arena_fill) + + stdlib.free(buf) + return testcheck.end() diff --git a/Test/BuddyTest/output/5e430ae5bf7af3a2.deps.json b/Test/BuddyTest/output/5e430ae5bf7af3a2.deps.json new file mode 100644 index 0000000..e6327bb --- /dev/null +++ b/Test/BuddyTest/output/5e430ae5bf7af3a2.deps.json @@ -0,0 +1 @@ +{"w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "mbuddytest": "b69cdee68f223fef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/BuddyTest/project.json b/Test/BuddyTest/project.json new file mode 100644 index 0000000..8b76992 --- /dev/null +++ b/Test/BuddyTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "BuddyTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "BuddyTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/BuddyTest/temp/0035c95a18d4f8e8.pyi b/Test/BuddyTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/BuddyTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/BuddyTest/temp/067c78e9f121dce3.pyi b/Test/BuddyTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/BuddyTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/BuddyTest/temp/06f53cc594b4ac6c.pyi b/Test/BuddyTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/BuddyTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/BuddyTest/temp/271ea3decb810db2.pyi b/Test/BuddyTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/BuddyTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/BuddyTest/temp/5e430ae5bf7af3a2.doc.json b/Test/BuddyTest/temp/5e430ae5bf7af3a2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/BuddyTest/temp/5e430ae5bf7af3a2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/BuddyTest/temp/5e430ae5bf7af3a2.pyi b/Test/BuddyTest/temp/5e430ae5bf7af3a2.pyi new file mode 100644 index 0000000..bbe03a4 --- /dev/null +++ b/Test/BuddyTest/temp/5e430ae5bf7af3a2.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import mbuddytest + +def main() -> CInt | CExport: pass diff --git a/Test/BuddyTest/temp/6446627d4f07a1b5.pyi b/Test/BuddyTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/BuddyTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/BuddyTest/temp/6f62fe05c5ea1ceb.pyi b/Test/BuddyTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/BuddyTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/BuddyTest/temp/71e0a3ffcb3ebfad.pyi b/Test/BuddyTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/BuddyTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/BuddyTest/temp/72e2d5ccb7cedcf1.pyi b/Test/BuddyTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/BuddyTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/BuddyTest/temp/7e529fe7a078cfef.pyi b/Test/BuddyTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/BuddyTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/BuddyTest/temp/90c53dd6db8d41cf.pyi b/Test/BuddyTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/BuddyTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/BuddyTest/temp/_phase1_manifest.json b/Test/BuddyTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..61f3808 --- /dev/null +++ b/Test/BuddyTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\BuddyTest\\App\\main.py": {"sha1": "5e430ae5bf7af3a2", "mtime": 1782224154.785576, "size": 260}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\BuddyTest\\App\\mbuddytest.py": {"sha1": "b69cdee68f223fef", "mtime": 1782827825.044477, "size": 19246}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32process.py": {"sha1": "067c78e9f121dce3", "mtime": 1781200703.5409546, "size": 4802}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32sync.py": {"sha1": "06f53cc594b4ac6c", "mtime": 1782297500.2581685, "size": 4726}} \ No newline at end of file diff --git a/Test/BuddyTest/temp/_sha1_map.txt b/Test/BuddyTest/temp/_sha1_map.txt new file mode 100644 index 0000000..1524289 --- /dev/null +++ b/Test/BuddyTest/temp/_sha1_map.txt @@ -0,0 +1,16 @@ +067c78e9f121dce3:includes/w32\win32process.py +06f53cc594b4ac6c:includes/w32\win32sync.py +271ea3decb810db2:includes/atom.py +5e430ae5bf7af3a2:main.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +b69cdee68f223fef:mbuddytest.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/BuddyTest/temp/_shared_sym.json b/Test/BuddyTest/temp/_shared_sym.json new file mode 100644 index 0000000..0700beb Binary files /dev/null and b/Test/BuddyTest/temp/_shared_sym.json differ diff --git a/Test/BuddyTest/temp/ab6e54ba9a669f76.pyi b/Test/BuddyTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/BuddyTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/BuddyTest/temp/b69cdee68f223fef.doc.json b/Test/BuddyTest/temp/b69cdee68f223fef.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/BuddyTest/temp/b69cdee68f223fef.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/BuddyTest/temp/b69cdee68f223fef.pyi b/Test/BuddyTest/temp/b69cdee68f223fef.pyi new file mode 100644 index 0000000..3de0cb1 --- /dev/null +++ b/Test/BuddyTest/temp/b69cdee68f223fef.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from mbuddytest.py +Module: mbuddytest +""" + + +import t, c +from t import CInt, CPtr, CChar, CUInt8T, CUInt32T, CUInt64T, CSizeT +import viperlib +import viperio +import stdlib +import memhub +import string +import testcheck +import w32.win32process +import w32.win32sync +import w32.win32base + +def _thread_alloc_free(param: t.CVoid | t.CPtr) -> t.CUInt32T: pass + +def mbuddy_main() -> CInt: pass diff --git a/Test/BuddyTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/BuddyTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/BuddyTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/BuddyTest/temp/c3b259b4059f8668.pyi b/Test/BuddyTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/BuddyTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/BuddyTest/temp/c9f4be41ca1cc2b4.pyi b/Test/BuddyTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/BuddyTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/BuddyTest/temp/dd3002730623424b.pyi b/Test/BuddyTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/BuddyTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/BuddyTest/temp/ee084e9fc6ee413a.pyi b/Test/BuddyTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/BuddyTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/BuddyTest/temp/f5522571bcce7bcb.pyi b/Test/BuddyTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/BuddyTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/BuddyTest/temp/f6b51804a0ba8ff0.pyi b/Test/BuddyTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/BuddyTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/BuiltinDecoratorTest/App/main.py b/Test/BuiltinDecoratorTest/App/main.py new file mode 100644 index 0000000..56566d9 --- /dev/null +++ b/Test/BuiltinDecoratorTest/App/main.py @@ -0,0 +1,268 @@ +import t +from stdint import * +from stdio import printf +from stdlib import malloc +import testcheck + + +# ============================================================ +# 测试 1：@property getter +# ============================================================ +class Rect: + width: t.CInt + height: t.CInt + + def __new__(self, w: t.CInt, h: t.CInt) -> 'Rect' | t.CPtr: + return t.CPtr(malloc(Rect.__sizeof__())) + + def __init__(self, w: t.CInt, h: t.CInt): + self.width = w + self.height = h + + @property + def area(self) -> t.CInt: + return self.width * self.height + + @property + def is_square(self) -> bool: + return self.width == self.height + + +def test_property_getter(): + testcheck.section("@property getter") + r1: Rect = Rect(3, 4) + testcheck.check(r1.area == 12, "area == 12", "area expect 12") + testcheck.check(not r1.is_square, "not square", "should not be square") + r2: Rect = Rect(5, 5) + testcheck.check(r2.area == 25, "square area == 25", "square area expect 25") + testcheck.check(r2.is_square, "is_square", "should be square") + + +# ============================================================ +# 测试 2：@property.setter +# ============================================================ +class Temperature: + _celsius: t.CDouble + + def __new__(self, celsius_val: t.CDouble) -> 'Temperature' | t.CPtr: + return t.CPtr(malloc(Temperature.__sizeof__())) + + def __init__(self, celsius_val: t.CDouble): + self._celsius = celsius_val + + @property + def celsius(self) -> t.CDouble: + return self._celsius + + @celsius.setter + def celsius(self, val: t.CDouble): + self._celsius = val + + @property + def fahrenheit(self) -> t.CDouble: + return self._celsius * 1.8 + 32.0 + + +def test_property_setter(): + testcheck.section("@property.setter") + tmp1: Temperature = Temperature(0.0) + testcheck.check(tmp1.fahrenheit > 31.9 and tmp1.fahrenheit < 32.1, "0C == 32F", "0C expect 32F") + tmp1.celsius = 100.0 + testcheck.check(tmp1.fahrenheit > 211.9 and tmp1.fahrenheit < 212.1, "100C == 212F", "100C expect 212F") + testcheck.check(tmp1.celsius > 99.9 and tmp1.celsius < 100.1, "celsius == 100", "celsius expect 100") + + +# ============================================================ +# 测试 3：@property.getter（重新定义 getter） +# ============================================================ +class Counter: + _count: t.CInt + + def __new__(self) -> 'Counter' | t.CPtr: + return t.CPtr(malloc(Counter.__sizeof__())) + + def __init__(self): + self._count = 0 + + @property + def value(self) -> t.CInt: + return self._count + + @value.setter + def value(self, v: t.CInt): + self._count = v + + +def test_property_getter_redef(): + testcheck.section("@property.getter redef") + cnt: Counter = Counter() + testcheck.check(cnt.value == 0, "init == 0", "init expect 0") + cnt.value = 42 + testcheck.check(cnt.value == 42, "set == 42", "set expect 42") + + +# ============================================================ +# 测试 4：@property.deleter +# ============================================================ +class Config: + _value: t.CInt + _deleted: t.CInt + + def __new__(self, v: t.CInt) -> 'Config' | t.CPtr: + return t.CPtr(malloc(Config.__sizeof__())) + + def __init__(self, v: t.CInt): + self._value = v + self._deleted = 0 + + @property + def value(self) -> t.CInt: + return self._value + + @value.setter + def value(self, v: t.CInt): + self._value = v + + @value.deleter + def value(self): + self._value = 0 + self._deleted = 1 + + def is_deleted(self) -> bool: + return self._deleted != 0 + + +def test_property_deleter(): + testcheck.section("@property.deleter") + cfg: Config = Config(99) + testcheck.check(cfg.value == 99, "init == 99", "init expect 99") + testcheck.check(not cfg.is_deleted(), "not deleted", "should not be deleted") + del cfg.value + testcheck.check(cfg.value == 0, "after del == 0", "after del expect 0") + testcheck.check(cfg.is_deleted(), "is_deleted", "should be deleted") + + +# ============================================================ +# 测试 5：@staticmethod +# ============================================================ +class MathUtils: + @staticmethod + def add(a: t.CInt, b: t.CInt) -> t.CInt: + return a + b + + @staticmethod + def max_val(a: t.CInt, b: t.CInt) -> t.CInt: + return a if a > b else b + + +def test_staticmethod(): + testcheck.section("@staticmethod") + # 通过类名调用 + testcheck.check(MathUtils.add(3, 4) == 7, "MathUtils.add(3,4)==7", "MathUtils.add(3,4) expect 7") + testcheck.check(MathUtils.max_val(5, 10) == 10, "MathUtils.max(5,10)==10", "MathUtils.max(5,10) expect 10") + + +# ============================================================ +# 测试 6：@classmethod +# ============================================================ +class Point: + x: t.CDouble + y: t.CDouble + + def __new__(self, x: t.CDouble, y: t.CDouble) -> 'Point' | t.CPtr: + return t.CPtr(malloc(Point.__sizeof__())) + + def __init__(self, x: t.CDouble, y: t.CDouble): + self.x = x + self.y = y + + @classmethod + def origin(cls: Point) -> Point: + return Point(0.0, 0.0) + + @classmethod + def from_int(cls: Point, x: t.CInt, y: t.CInt) -> Point: + return Point(x * 1.0, y * 1.0) + + def dist_sq(self) -> t.CDouble: + return self.x * self.x + self.y * self.y + + +def test_classmethod(): + testcheck.section("@classmethod") + # 通过类名调用 + p1: Point = Point.origin() + testcheck.check(p1.x > -0.01 and p1.x < 0.01, "origin.x ~= 0", "origin.x expect ~0") + testcheck.check(p1.y > -0.01 and p1.y < 0.01, "origin.y ~= 0", "origin.y expect ~0") + testcheck.check(p1.dist_sq() < 0.01, "origin.dist ~= 0", "origin.dist expect ~0") + + p2: Point = Point.from_int(3, 4) + testcheck.check(p2.x > 2.99 and p2.x < 3.01, "from_int(3,4).x ~= 3", "from_int(3,4).x expect ~3") + testcheck.check(p2.y > 3.99 and p2.y < 4.01, "from_int(3,4).y ~= 4", "from_int(3,4).y expect ~4") + testcheck.check(p2.dist_sq() > 24.99 and p2.dist_sq() < 25.01, "dist_sq ~= 25", "dist_sq expect ~25") + + +# ============================================================ +# 测试 7：综合测试 — @property + @staticmethod + @classmethod +# ============================================================ +class Circle: + _radius: t.CDouble + + def __new__(self, r: t.CDouble) -> 'Circle' | t.CPtr: + return t.CPtr(malloc(Circle.__sizeof__())) + + def __init__(self, r: t.CDouble): + self._radius = r + + @property + def radius(self) -> t.CDouble: + return self._radius + + @radius.setter + def radius(self, val: t.CDouble): + self._radius = val + + @property + def area(self) -> t.CDouble: + return 3.14159265 * self._radius * self._radius + + @staticmethod + def is_unit(r: t.CDouble) -> bool: + return r > 0.99 and r < 1.01 + + @classmethod + def unit(cls: Circle) -> Circle: + return Circle(1.0) + + +def test_combined(): + testcheck.section("Combined @property + @staticmethod + @classmethod") + # 先测试直接构造 + cr0: Circle = Circle(1.0) + testcheck.check(cr0.radius > 0.99 and cr0.radius < 1.01, "direct radius ~= 1", "direct radius expect ~1") + testcheck.check(cr0.area > 3.14 and cr0.area < 3.15, "direct area ~= pi", "direct area expect ~pi") + cr0.radius = 2.0 + testcheck.check(cr0.radius > 1.99 and cr0.radius < 2.01, "direct radius 2.0", "direct radius expect 2.0") + testcheck.check(cr0.area > 12.56 and cr0.area < 12.58, "direct area ~= 4pi", "direct area expect ~4pi") + # 测试 classmethod + cr1: Circle = Circle.unit() + testcheck.check(cr1.radius > 0.99 and cr1.radius < 1.01, "unit radius ~= 1", "unit radius expect ~1") + testcheck.check(Circle.is_unit(1.0), "is_unit(1.0)", "is_unit(1.0) should be true") + testcheck.check(not Circle.is_unit(2.0), "not is_unit(2.0)", "is_unit(2.0) should be false") + + +# ============================================================ +# 主函数 +# ============================================================ +def main() -> t.CInt | t.CExport: + testcheck.begin("Builtin Decorator Test") + + test_property_getter() + test_property_setter() + test_property_getter_redef() + test_property_deleter() + test_staticmethod() + test_classmethod() + test_combined() + + return testcheck.end() diff --git a/Test/BuiltinDecoratorTest/output/e3ac223bdaf94a0c.deps.json b/Test/BuiltinDecoratorTest/output/e3ac223bdaf94a0c.deps.json new file mode 100644 index 0000000..d39611e --- /dev/null +++ b/Test/BuiltinDecoratorTest/output/e3ac223bdaf94a0c.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/BuiltinDecoratorTest/project.json b/Test/BuiltinDecoratorTest/project.json new file mode 100644 index 0000000..2a43eb5 --- /dev/null +++ b/Test/BuiltinDecoratorTest/project.json @@ -0,0 +1,28 @@ +{ + "name": "BuiltinDecoratorTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "BuiltinDecoratorTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/BuiltinDecoratorTest/temp/6f62fe05c5ea1ceb.pyi b/Test/BuiltinDecoratorTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/BuiltinDecoratorTest/temp/7e529fe7a078cfef.pyi b/Test/BuiltinDecoratorTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/BuiltinDecoratorTest/temp/90c53dd6db8d41cf.pyi b/Test/BuiltinDecoratorTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/BuiltinDecoratorTest/temp/_phase1_manifest.json b/Test/BuiltinDecoratorTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..be3bd83 --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\BuiltinDecoratorTest\\App\\main.py": {"sha1": "e3ac223bdaf94a0c", "mtime": 1782110312.4550886, "size": 8488}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/BuiltinDecoratorTest/temp/_sha1_map.txt b/Test/BuiltinDecoratorTest/temp/_sha1_map.txt new file mode 100644 index 0000000..b7d9faf --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/_sha1_map.txt @@ -0,0 +1,7 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +e3ac223bdaf94a0c:main.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/BuiltinDecoratorTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/BuiltinDecoratorTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/BuiltinDecoratorTest/temp/dd3002730623424b.pyi b/Test/BuiltinDecoratorTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.doc.json b/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.pyi b/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.pyi new file mode 100644 index 0000000..10d5c52 --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/e3ac223bdaf94a0c.pyi @@ -0,0 +1,110 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdint import * +from stdio import printf +from stdlib import malloc +import testcheck + +class Rect: + width: t.CInt + height: t.CInt + def __new__(self: Rect, w: t.CInt, h: t.CInt) -> 'Rect' | t.CPtr: pass + def __init__(self: Rect, w: t.CInt, h: t.CInt) -> t.CInt: pass + @property + def area(self: Rect) -> t.CInt: pass + @property + def is_square(self: Rect) -> bool: pass + +def test_property_getter() -> t.CInt: pass + + +class Temperature: + _celsius: t.CDouble + def __new__(self: Temperature, celsius_val: t.CDouble) -> 'Temperature' | t.CPtr: pass + def __init__(self: Temperature, celsius_val: t.CDouble) -> t.CInt: pass + @property + def celsius(self: Temperature) -> t.CDouble: pass + @celsius.setter + def celsius(self: Temperature, val: t.CDouble) -> t.CInt: pass + @property + def fahrenheit(self: Temperature) -> t.CDouble: pass + +def test_property_setter() -> t.CInt: pass + + +class Counter: + _count: t.CInt + def __new__(self: Counter) -> 'Counter' | t.CPtr: pass + def __init__(self: Counter) -> t.CInt: pass + @property + def value(self: Counter) -> t.CInt: pass + @value.setter + def value(self: Counter, v: t.CInt) -> t.CInt: pass + +def test_property_getter_redef() -> t.CInt: pass + + +class Config: + _value: t.CInt + _deleted: t.CInt + def __new__(self: Config, v: t.CInt) -> 'Config' | t.CPtr: pass + def __init__(self: Config, v: t.CInt) -> t.CInt: pass + @property + def value(self: Config) -> t.CInt: pass + @value.setter + def value(self: Config, v: t.CInt) -> t.CInt: pass + @value.deleter + def value(self: Config) -> t.CInt: pass + def is_deleted(self: Config) -> bool: pass + +def test_property_deleter() -> t.CInt: pass + + +class MathUtils: + @staticmethod + def add(a: t.CInt, b: t.CInt) -> t.CInt: pass + @staticmethod + def max_val(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def test_staticmethod() -> t.CInt: pass + + +class Point: + x: t.CDouble + y: t.CDouble + def __new__(self: Point, x: t.CDouble, y: t.CDouble) -> 'Point' | t.CPtr: pass + def __init__(self: Point, x: t.CDouble, y: t.CDouble) -> t.CInt: pass + @classmethod + def origin(cls: Point) -> Point: pass + @classmethod + def from_int(cls: Point, x: t.CInt, y: t.CInt) -> Point: pass + def dist_sq(self: Point) -> t.CDouble: pass + +def test_classmethod() -> t.CInt: pass + + +class Circle: + _radius: t.CDouble + def __new__(self: Circle, r: t.CDouble) -> 'Circle' | t.CPtr: pass + def __init__(self: Circle, r: t.CDouble) -> t.CInt: pass + @property + def radius(self: Circle) -> t.CDouble: pass + @radius.setter + def radius(self: Circle, val: t.CDouble) -> t.CInt: pass + @property + def area(self: Circle) -> t.CDouble: pass + @staticmethod + def is_unit(r: t.CDouble) -> bool: pass + @classmethod + def unit(cls: Circle) -> Circle: pass + +def test_combined() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/BuiltinDecoratorTest/temp/f5522571bcce7bcb.pyi b/Test/BuiltinDecoratorTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/BuiltinDecoratorTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/CPythonTest/App/cpython.py b/Test/CPythonTest/App/cpython.py new file mode 100644 index 0000000..f9fb785 --- /dev/null +++ b/Test/CPythonTest/App/cpython.py @@ -0,0 +1,777 @@ +import t, c + + +# ============================================================ +# PyObject — 所有 Python 对象的基类 +# ============================================================ +class PyObject: + ob_refcnt: t.CLong + ob_type: t.CPtr + + +# ============================================================ +# 引用计数 +# ============================================================ +def Py_INCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def Py_DECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def Py_XINCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def Py_XDECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def Py_DecRef(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + + +# ============================================================ +# 解释器初始化/终止 +# ============================================================ +def Py_Initialize() -> t.CVoid | t.State: pass +def Py_Finalize() -> t.CVoid | t.State: pass +def Py_InitializeEx(initsigs: t.CInt) -> t.CVoid | t.State: pass +def Py_FinalizeEx() -> t.CInt | t.State: pass +def Py_IsInitialized() -> t.CInt | t.State: pass + + +# ============================================================ +# PyConfig +# ============================================================ +class PySTATUS: + func: t.CConst | str + err_msg: t.CConst | str + exitcode: t.CInt + +class PyPreConfig: + _config_version: t.CInt + _config_init: t.CInt + parse_argv: t.CInt + isolated: t.CInt + use_environment: t.CInt + utf8_mode: t.CInt + coerce_c_locale: t.CInt + C_locale_warnings: t.CInt + allocator: t.CInt + dev_mode: t.CInt + +class PyWideStringList: + length: t.CSizeT + items: t.CPtr + +class PyConfig: + _config_version: t.CInt + _config_init: t.CInt + isolated: t.CInt + use_environment: t.CInt + dev_mode: t.CInt + install_signal_handlers: t.CInt + use_hash_seed: t.CInt + hash_seed: t.CUnsignedLong + faulthandler: t.CInt + tracemalloc: t.CInt + import_time: t.CInt + show_ref_count: t.CInt + dump_refs: t.CInt + dump_refs_file: t.CConst | str + malloc_stats: t.CInt + filesystem_errors: t.CInt + pycache_prefix: t.CConst | str + parse_argv: t.CInt + argv: PyWideStringList + program_name: t.CConst | str + xoptions: PyWideStringList + warnoptions: PyWideStringList + site_import: t.CInt + bytes_warning: t.CInt + warn_default_encoding: t.CInt + inspect: t.CInt + interactive: t.CInt + optimization_level: t.CInt + parser_debug: t.CInt + verbose: t.CInt + quiet: t.CInt + home: t.CConst | str + pythonpath_env: t.CConst | str + module_search_paths: PyWideStringList + module_search_paths_set: t.CInt + executable: t.CConst | str + base_executable: t.CConst | str + prefix: t.CConst | str + base_prefix: t.CConst | str + exec_prefix: t.CConst | str + base_exec_prefix: t.CConst | str + platlibdir: t.CConst | str + stdio_filename: t.CConst | str + +def PyConfig_InitPythonConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass +def PyConfig_InitIsolatedConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass +def PyConfig_Clear(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass +def PyConfig_Read(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass +def Py_InitializeFromConfig(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass +def Py_DecodeLocale(s: t.CConst | str) -> str | t.State: pass +def PyStatus_Exception(status: PySTATUS) -> t.CInt | t.State: pass + + +# ============================================================ +# 整数操作 (PyLong) +# ============================================================ +def PyLong_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromUnsignedLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromLongLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromUnsignedLongLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromSsize_t(v: t.CPtrDiffT) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromSize_t(v: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyLong_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass +def PyLong_AsLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass +def PyLong_AsUnsignedLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass +def PyLong_AsLongLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass +def PyLong_AsUnsignedLongLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass +def PyLong_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass +def PyLong_AsSsize_t(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass +def PyLong_AsSize_t(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + + +# ============================================================ +# 浮点数操作 (PyFloat) +# ============================================================ +def PyFloat_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass +def PyFloat_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass +def PyFloat_FromString(s: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 字符串操作 (PyUnicode / PyBytes) +# ============================================================ +def PyUnicode_FromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyUnicode_FromStringAndSize(u: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyUnicode_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyUnicode_AsUTF8(obj: PyObject | t.CPtr) -> str | t.State: pass +def PyUnicode_AsUTF8AndSize(obj: PyObject | t.CPtr, size: t.CPtrDiffT | t.CPtr) -> str | t.State: pass +def PyUnicode_AsWideChar(obj: PyObject | t.CPtr, wbuf: t.CPtr, bufsize: t.CSizeT) -> t.CSizeT | t.State: pass +def PyUnicode_GetLength(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass +def PyUnicode_ReadChar(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CUInt32T | t.State: pass +def PyUnicode_WriteChar(obj: PyObject | t.CPtr, idx: t.CSizeT, ch: t.CUInt32T) -> t.CInt | t.State: pass +def PyUnicode_Compare(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyUnicode_Concat(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyUnicode_Contains(container: PyObject | t.CPtr, element: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyUnicode_Format(fmt: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyUnicode_InternInPlace(s: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass +def PyUnicode_InternFromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromString(v: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyBytes_FromStringAndSize(v: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyBytes_AsString(obj: PyObject | t.CPtr) -> str | t.State: pass +def PyBytes_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyBytes_FromObject(o: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyBytes_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 布尔 / None +# ============================================================ +def PyBool_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + +Py_True: PyObject | t.CPtr | t.State +Py_False: PyObject | t.CPtr | t.State +Py_None: PyObject | t.CPtr | t.State + + +# ============================================================ +# 列表操作 (PyList) +# ============================================================ +def PyList_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyList_Size(lst: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyList_GetItem(lst: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyList_SetItem(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_Insert(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_Append(lst: PyObject | t.CPtr, item: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_GetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyList_SetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT, items: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_Sort(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_Reverse(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_AsTuple(lst: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 元组操作 (PyTuple) +# ============================================================ +def PyTuple_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyTuple_Size(tup: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyTuple_GetItem(tup: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyTuple_SetItem(tup: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyTuple_GetSlice(tup: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PyTuple_Pack(n: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 字典操作 (PyDict) +# ============================================================ +def PyDict_New() -> PyObject | t.CPtr | t.State: pass +def PyDict_GetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyDict_GetItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyDict_SetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyDict_SetItemString(d: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyDict_DelItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyDict_DelItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass +def PyDict_Size(d: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyDict_Keys(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyDict_Values(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyDict_Items(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyDict_Copy(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyDict_Clear(d: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PyDict_Contains(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyDict_Next(d: PyObject | t.CPtr, pos: t.CPtrDiffT | t.CPtr, key: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr) -> t.CInt | t.State: pass +def PyDict_Merge(d: PyObject | t.CPtr, other: PyObject | t.CPtr, override: t.CInt) -> t.CInt | t.State: pass +def PyDict_Update(d: PyObject | t.CPtr, other: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 对象通用操作 +# ============================================================ +def PyObject_Str(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_Repr(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_Type(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_IsInstance(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_IsSubclass(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_HasAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_HasAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass +def PyObject_GetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_GetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyObject_SetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_SetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_DelAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_DelAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass +def PyObject_Call(callable: PyObject | t.CPtr, args: PyObject | t.CPtr, kwargs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_CallNoArgs(callable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_CallOneArg(callable: PyObject | t.CPtr, arg: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_IsTrue(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_Not(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_RichCompare(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> PyObject | t.CPtr | t.State: pass +def PyObject_RichCompareBool(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> t.CInt | t.State: pass +def PyObject_Hash(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass +def PyObject_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyObject_GetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_SetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_DelItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyObject_Dir(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_GetIter(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyIter_Next(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 比较常量 +# ============================================================ +Py_LT: t.CInt | t.State +Py_LE: t.CInt | t.State +Py_EQ: t.CInt | t.State +Py_NE: t.CInt | t.State +Py_GT: t.CInt | t.State +Py_GE: t.CInt | t.State + + +# ============================================================ +# 模块操作 +# ============================================================ +def PyImport_ImportModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyImport_Import(name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyImport_ImportModuleLevelObject(name: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr, level: t.CInt) -> PyObject | t.CPtr | t.State: pass +def PyImport_ReloadModule(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyImport_AddModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyImport_GetModuleDict() -> PyObject | t.CPtr | t.State: pass +def PyImport_ImportModuleEx(name: t.CConst | str, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 模块创建 +# ============================================================ +class PyMethodDef: + ml_name: t.CConst | str + ml_meth: t.CPtr + ml_flags: t.CInt + ml_doc: t.CConst | str + +class PyModuleDef_Base: + ob_base: PyObject + m_init: t.CPtr + m_index: t.CSizeT + m_copy: PyObject | t.CPtr + +class PyModuleDef: + m_base: PyModuleDef_Base + m_name: t.CConst | str + m_doc: t.CConst | str + m_size: t.CPtrDiffT + m_methods: PyMethodDef | t.CPtr + m_slots: t.CPtr + m_traverse: t.CPtr + m_clear: t.CPtr + m_free: t.CPtr + +def PyModule_Create(mod: PyModuleDef | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyModule_Create2(mod: PyModuleDef | t.CPtr, ver: t.CInt) -> PyObject | t.CPtr | t.State: pass +def PyModule_GetName(mod: PyObject | t.CPtr) -> str | t.State: pass +def PyModule_GetDict(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyModule_GetFilenameObject(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyModule_AddObject(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyModule_AddObjectRef(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyModule_AddIntConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CLong) -> t.CInt | t.State: pass +def PyModule_AddStringConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CConst | str) -> t.CInt | t.State: pass + + +# ============================================================ +# GIL 管理 +# ============================================================ +def PyGILState_Ensure() -> t.CInt | t.State: pass +def PyGILState_Release(state: t.CInt) -> t.CVoid | t.State: pass +def PyEval_SaveThread() -> t.CPtr | t.State: pass +def PyEval_RestoreThread(tstate: t.CPtr) -> t.CVoid | t.State: pass +def PyEval_AcquireThread(tstate: t.CPtr) -> t.CVoid | t.State: pass +def PyEval_ReleaseThread(tstate: t.CPtr) -> t.CVoid | t.State: pass +def PyEval_AcquireLock() -> t.CVoid | t.State: pass +def PyEval_ReleaseLock() -> t.CVoid | t.State: pass +def PyEval_ThreadsInitialized() -> t.CInt | t.State: pass +def PyEval_InitThreads() -> t.CVoid | t.State: pass + + +# ============================================================ +# 错误处理 +# ============================================================ +def PyErr_Occurred() -> PyObject | t.CPtr | t.State: pass +def PyErr_Clear() -> t.CVoid | t.State: pass +def PyErr_Fetch(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass +def PyErr_Restore(tp: PyObject | t.CPtr, val: PyObject | t.CPtr, tb: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PyErr_NormalizeException(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass +def PyErr_SetString(tp: PyObject | t.CPtr, msg: t.CConst | str) -> t.CVoid | t.State: pass +def PyErr_SetObject(tp: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PyErr_BadArgument() -> t.CPtr | t.State: pass +def PyErr_NoMemory() -> t.CPtr | t.State: pass +def PyErr_Print() -> t.CVoid | t.State: pass +def PyErr_ExceptionMatches(exc: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyErr_GivenExceptionMatches(given: PyObject | t.CPtr, exc: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 内置异常类型 +# ============================================================ +PyExc_BaseException: PyObject | t.CPtr | t.State +PyExc_Exception: PyObject | t.CPtr | t.State +PyExc_TypeError: PyObject | t.CPtr | t.State +PyExc_ValueError: PyObject | t.CPtr | t.State +PyExc_RuntimeError: PyObject | t.CPtr | t.State +PyExc_KeyError: PyObject | t.CPtr | t.State +PyExc_IndexError: PyObject | t.CPtr | t.State +PyExc_AttributeError: PyObject | t.CPtr | t.State +PyExc_ImportError: PyObject | t.CPtr | t.State +PyExc_OSError: PyObject | t.CPtr | t.State +PyExc_IOError: PyObject | t.CPtr | t.State +PyExc_ZeroDivisionError: PyObject | t.CPtr | t.State +PyExc_OverflowError: PyObject | t.CPtr | t.State +PyExc_MemoryError: PyObject | t.CPtr | t.State +PyExc_NameError: PyObject | t.CPtr | t.State +PyExc_StopIteration: PyObject | t.CPtr | t.State +PyExc_NotImplementedError: PyObject | t.CPtr | t.State +PyExc_SyntaxError: PyObject | t.CPtr | t.State +PyExc_FileNotFoundError: PyObject | t.CPtr | t.State +PyExc_PermissionError: PyObject | t.CPtr | t.State +PyExc_IsADirectoryError: PyObject | t.CPtr | t.State +PyExc_NotADirectoryError: PyObject | t.CPtr | t.State + + +# ============================================================ +# 内置函数 +# ============================================================ +def Py_BuildValue(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def Py_VaBuildValue(fmt: t.CConst | str, va: t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 类型检查 +# ============================================================ +def PyLong_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyFloat_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyUnicode_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyBytes_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyList_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyTuple_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyDict_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyBool_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyNone_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 内存管理 +# ============================================================ +def PyMem_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass +def PyMem_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass +def PyMem_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass +def PyMem_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass +def PyObject_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass +def PyObject_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass +def PyObject_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass +def PyObject_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass +def PyObject_Init(obj: PyObject | t.CPtr, tp: t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyObject_InitVar(obj: PyObject | t.CPtr, tp: t.CPtr, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 文件操作 +# ============================================================ +def PyFile_FromString(filename: t.CConst | str, mode: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyFile_AsFile(file: PyObject | t.CPtr) -> t.CPtr | t.State: pass +def PyFile_WriteObject(obj: PyObject | t.CPtr, file: PyObject | t.CPtr, flags: t.CInt) -> t.CInt | t.State: pass +def PyFile_WriteString(s: t.CConst | str, file: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 编译/执行 +# ============================================================ +def Py_CompileString(s: t.CConst | str, filename: t.CConst | str, start: t.CInt) -> PyObject | t.CPtr | t.State: pass +def PyEval_EvalCode(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyEval_EvalCodeEx(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, args: PyObject | t.CPtr | t.CPtr, argc: t.CInt, kws: PyObject | t.CPtr | t.CPtr, kwc: t.CInt, defs: PyObject | t.CPtr | t.CPtr, defc: t.CInt, kwdefs: PyObject | t.CPtr, closure: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyEval_EvalFrame(frame: t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyEval_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyEval_CallFunction(callable: PyObject | t.CPtr, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyEval_CallMethod(obj: PyObject | t.CPtr, method: t.CConst | str, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# PyRun +# ============================================================ +def PyRun_SimpleString(cmd: t.CConst | str) -> t.CInt | t.State: pass +def PyRun_SimpleFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass +def PyRun_AnyFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass +def PyRun_String(s: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyRun_File(fp: t.CPtr, filename: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +Py_single_input: t.CInt | t.State +Py_file_input: t.CInt | t.State +Py_eval_input: t.CInt | t.State + + +# ============================================================ +# Capsule +# ============================================================ +def PyCapsule_New(ptr: t.CPtr, name: t.CConst | str, dtor: t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyCapsule_GetPointer(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CPtr | t.State: pass +def PyCapsule_GetName(cap: PyObject | t.CPtr) -> str | t.State: pass +def PyCapsule_IsValid(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass +def PyCapsule_SetPointer(cap: PyObject | t.CPtr, ptr: t.CPtr) -> t.CInt | t.State: pass +def PyCapsule_SetName(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass +def PyCapsule_SetDestructor(cap: PyObject | t.CPtr, dtor: t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 序列操作 +# ============================================================ +def PySequence_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySequence_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PySequence_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PySequence_Concat(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PySequence_Repeat(obj: PyObject | t.CPtr, count: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PySequence_GetItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PySequence_SetItem(obj: PyObject | t.CPtr, idx: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySequence_DelItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CInt | t.State: pass +def PySequence_GetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> PyObject | t.CPtr | t.State: pass +def PySequence_SetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySequence_DelSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> t.CInt | t.State: pass +def PySequence_Contains(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySequence_Index(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PySequence_Count(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PySequence_List(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PySequence_Tuple(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 映射操作 +# ============================================================ +def PyMapping_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyMapping_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyMapping_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PyMapping_HasKey(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyMapping_HasKeyString(obj: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass +def PyMapping_GetItemString(obj: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PyMapping_SetItemString(obj: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyMapping_Keys(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyMapping_Values(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyMapping_Items(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +# ============================================================ +# 数字操作 +# ============================================================ +def PyNumber_Add(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Subtract(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Multiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_MatrixMultiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_FloorDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_TrueDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Modulo(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Power(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, o3: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Negative(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Positive(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Absolute(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Invert(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Lshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Rshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_And(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Xor(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Or(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Index(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Long(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Float(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyNumber_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 弱引用 +# ============================================================ +def PyWeakref_NewRef(obj: PyObject | t.CPtr, cb: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyWeakref_GetObject(ref: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyWeakref_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 属性描述符 +# ============================================================ +class PyGetSetDef: + name: t.CConst | str + get: t.CPtr + set: t.CPtr + doc: t.CConst | str + closure: t.CPtr + +class PyMemberDef: + name: t.CConst | str + type: t.CInt + offset: t.CPtrDiffT + flags: t.CInt + doc: t.CConst | str + + +# ============================================================ +# sys 模块 +# ============================================================ +def PySys_GetObject(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass +def PySys_SetObject(name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySys_ResetWarnOptions() -> t.CVoid | t.State: pass +def PySys_AddWarnOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PySYS_AddXOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PySys_WriteStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass +def PySys_WriteStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass +def PySys_FormatStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass +def PySys_FormatStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass + + +# ============================================================ +# 内存视图 +# ============================================================ +def PyMemoryView_FromObject(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyMemoryView_GetContiguous(obj: PyObject | t.CPtr, btype: t.CInt, order: t.CChar) -> PyObject | t.CPtr | t.State: pass +def PyMemoryView_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# set / frozenset +# ============================================================ +def PySet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PySet_Add(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySet_Discard(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySet_Contains(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PySet_Size(s: PyObject | t.CPtr) -> t.CSizeT | t.State: pass +def PySet_Clear(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass +def PySet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass +def PyFrozenSet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PyFrozenSet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# slice +# ============================================================ +def PySlice_New(start: PyObject | t.CPtr, stop: PyObject | t.CPtr, step: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass +def PySlice_GetIndicesEx(sl: PyObject | t.CPtr, length: t.CSizeT, start: t.CSizeT | t.CPtr, stop: t.CSizeT | t.CPtr, step: t.CSizeT | t.CPtr, slicelen: t.CSizeT | t.CPtr) -> t.CInt | t.State: pass +def PySlice_Unpack(sl: PyObject | t.CPtr, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT | t.CPtr) -> t.CVoid | t.State: pass +def PySlice_AdjustIndices(length: t.CSizeT, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT) -> t.CSizeT | t.State: pass +def PySlice_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 类型包装类 — 带运算符重载 +# ============================================================ + +# --- PyLongObj: Python 整数包装 --- +class PyLongObj: + ptr: PyObject | t.CPtr + + def __new__(self) -> 'PyLongObj' | t.CPtr: + return t.CPtr(malloc(PyLongObj.__sizeof__())) + + @staticmethod + def from_long(val: t.CLong) -> 'PyLongObj' | t.CPtr: + obj: PyLongObj | t.CPtr = PyLongObj() + obj.ptr = PyLong_FromLong(val) + return obj + + def __add__(self, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: + res: PyLongObj | t.CPtr = PyLongObj() + res.ptr = PyNumber_Add(self.ptr, other.ptr) + return res + + def __sub__(self, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: + res: PyLongObj | t.CPtr = PyLongObj() + res.ptr = PyNumber_Subtract(self.ptr, other.ptr) + return res + + def __mul__(self, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: + res: PyLongObj | t.CPtr = PyLongObj() + res.ptr = PyNumber_Multiply(self.ptr, other.ptr) + return res + + def __neg__(self) -> 'PyLongObj' | t.CPtr: + res: PyLongObj | t.CPtr = PyLongObj() + res.ptr = PyNumber_Negative(self.ptr) + return res + + def to_long(self) -> t.CLong: + return PyLong_AsLong(self.ptr) + + def __str__(self) -> PyObject | t.CPtr: + return PyObject_Str(self.ptr) + + def dec_ref(self): + Py_DecRef(self.ptr) + + +# --- PyFloatObj: Python 浮点数包装 --- +class PyFloatObj: + ptr: PyObject | t.CPtr + + def __new__(self) -> 'PyFloatObj' | t.CPtr: + return t.CPtr(malloc(PyFloatObj.__sizeof__())) + + @staticmethod + def from_double(val: t.CDouble) -> 'PyFloatObj' | t.CPtr: + obj: PyFloatObj | t.CPtr = PyFloatObj() + obj.ptr = PyFloat_FromDouble(val) + return obj + + def __add__(self, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: + res: PyFloatObj | t.CPtr = PyFloatObj() + res.ptr = PyNumber_Add(self.ptr, other.ptr) + return res + + def __sub__(self, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: + res: PyFloatObj | t.CPtr = PyFloatObj() + res.ptr = PyNumber_Subtract(self.ptr, other.ptr) + return res + + def __mul__(self, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: + res: PyFloatObj | t.CPtr = PyFloatObj() + res.ptr = PyNumber_Multiply(self.ptr, other.ptr) + return res + + def __truediv__(self, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: + res: PyFloatObj | t.CPtr = PyFloatObj() + res.ptr = PyNumber_TrueDivide(self.ptr, other.ptr) + return res + + def __neg__(self) -> 'PyFloatObj' | t.CPtr: + res: PyFloatObj | t.CPtr = PyFloatObj() + res.ptr = PyNumber_Negative(self.ptr) + return res + + def to_double(self) -> t.CDouble: + return PyFloat_AsDouble(self.ptr) + + def __str__(self) -> PyObject | t.CPtr: + return PyObject_Str(self.ptr) + + def dec_ref(self): + Py_DecRef(self.ptr) + + +# --- PyUnicodeObj: Python 字符串包装 --- +class PyUnicodeObj: + ptr: PyObject | t.CPtr + + def __new__(self) -> 'PyUnicodeObj' | t.CPtr: + return t.CPtr(malloc(PyUnicodeObj.__sizeof__())) + + @staticmethod + def from_string(s: t.CChar | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: + obj: PyUnicodeObj | t.CPtr = PyUnicodeObj() + obj.ptr = PyUnicode_FromString(s) + return obj + + def __add__(self, other: PyUnicodeObj | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: + res: PyUnicodeObj | t.CPtr = PyUnicodeObj() + res.ptr = PyUnicode_Concat(self.ptr, other.ptr) + return res + + def __len__(self) -> t.CPtrDiffT: + return PyUnicode_GetLength(self.ptr) + + def as_utf8(self) -> t.CChar | t.CPtr: + return PyUnicode_AsUTF8(self.ptr) + + def __str__(self) -> PyObject | t.CPtr: + return self.ptr + + def dec_ref(self): + Py_DecRef(self.ptr) + + +# --- PyListObj: Python 列表包装 --- +class PyListObj: + ptr: PyObject | t.CPtr + + def __new__(self) -> 'PyListObj' | t.CPtr: + return t.CPtr(malloc(PyListObj.__sizeof__())) + + @staticmethod + def new() -> 'PyListObj' | t.CPtr: + obj: PyListObj | t.CPtr = PyListObj() + obj.ptr = PyList_New(0) + return obj + + def __getitem__(self, idx: t.CSizeT) -> PyObject | t.CPtr: + return PyList_GetItem(self.ptr, idx) + + def __setitem__(self, idx: t.CSizeT, val: PyObject | t.CPtr): + PyList_SetItem(self.ptr, idx, val) + + def __len__(self) -> t.CSizeT: + return PyList_Size(self.ptr) + + def append(self, item: PyObject | t.CPtr) -> t.CInt: + return PyList_Append(self.ptr, item) + + def dec_ref(self): + Py_DecRef(self.ptr) + + +# --- PyDictObj: Python 字典包装 --- +class PyDictObj: + ptr: PyObject | t.CPtr + + def __new__(self) -> 'PyDictObj' | t.CPtr: + return t.CPtr(malloc(PyDictObj.__sizeof__())) + + @staticmethod + def new() -> 'PyDictObj' | t.CPtr: + obj: PyDictObj | t.CPtr = PyDictObj() + obj.ptr = PyDict_New() + return obj + + def __getitem__(self, key: PyObject | t.CPtr) -> PyObject | t.CPtr: + return PyDict_GetItem(self.ptr, key) + + def __setitem__(self, key: PyObject | t.CPtr, val: PyObject | t.CPtr): + PyDict_SetItem(self.ptr, key, val) + + def __len__(self) -> t.CSizeT: + return PyDict_Size(self.ptr) + + def get_string(self, key: t.CChar | t.CPtr) -> PyObject | t.CPtr: + return PyDict_GetItemString(self.ptr, key) + + def set_string(self, key: t.CChar | t.CPtr, val: PyObject | t.CPtr): + PyDict_SetItemString(self.ptr, key, val) + + def dec_ref(self): + Py_DecRef(self.ptr) diff --git a/Test/CPythonTest/App/main.py b/Test/CPythonTest/App/main.py new file mode 100644 index 0000000..851e520 --- /dev/null +++ b/Test/CPythonTest/App/main.py @@ -0,0 +1,359 @@ +import t, c +from t import CInt, CLong, CDouble, CPtr, CSizeT +from stdio import printf +import cpython +import testcheck + + +def test_init_finalize(): + testcheck.section("Py_Initialize / Py_Finalize") + cpython.Py_Initialize() + testcheck.check(cpython.Py_IsInitialized() != 0, "Py_Initialize succeeded", "Py_Initialize succeeded") + cpython.Py_Finalize() + testcheck.check(cpython.Py_IsInitialized() == 0, "Py_Finalize succeeded", "Py_Finalize succeeded") + + +def test_long_operations(): + testcheck.section("PyLong operations") + cpython.Py_Initialize() + + # 创建整数 + py_num: cpython.PyObject | CPtr = cpython.PyLong_FromLong(12345) + testcheck.check(py_num != 0, "PyLong_FromLong != NULL", "PyLong_FromLong != NULL") + + # 整数转字符串 + py_str: cpython.PyObject | CPtr = cpython.PyObject_Str(py_num) + testcheck.check(py_str != 0, "PyObject_Str != NULL", "PyObject_Str != NULL") + + # 获取 C 字符串 + c_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_str) + testcheck.check(c_str != 0, "PyUnicode_AsUTF8 != NULL", "PyUnicode_AsUTF8 != NULL") + printf(" PyLong_FromLong(12345) -> str = %s\n", c_str) + + # 读取回来 + val: CLong = cpython.PyLong_AsLong(py_num) + testcheck.check(val == 12345, "PyLong_AsLong == 12345", "PyLong_AsLong == 12345") + + # 负数 + py_neg: cpython.PyObject | CPtr = cpython.PyLong_FromLong(-999) + neg_val: CLong = cpython.PyLong_AsLong(py_neg) + testcheck.check(neg_val == -999, "PyLong_FromLong(-999) == -999", "PyLong_FromLong(-999) == -999") + + # 浮点数转整数 + py_dbl: cpython.PyObject | CPtr = cpython.PyFloat_FromDouble(3.14159) + testcheck.check(py_dbl != 0, "PyFloat_FromDouble != NULL", "PyFloat_FromDouble != NULL") + dbl_val: CDouble = cpython.PyFloat_AsDouble(py_dbl) + testcheck.check(dbl_val > 3.14 and dbl_val < 3.15, "PyFloat_AsDouble ~= 3.14", "PyFloat_AsDouble ~= 3.14") + + # 清理 + cpython.Py_DecRef(py_dbl) + cpython.Py_DecRef(py_neg) + cpython.Py_DecRef(py_str) + cpython.Py_DecRef(py_num) + + cpython.Py_Finalize() + + +def test_string_operations(): + testcheck.section("PyUnicode operations") + cpython.Py_Initialize() + + # 从 C 字符串创建 + py_s1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Hello, Viper!") + testcheck.check(py_s1 != 0, "PyUnicode_FromString != NULL", "PyUnicode_FromString != NULL") + + # 获取 UTF8 + utf8: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_s1) + testcheck.check(utf8 != 0, "AsUTF8 != NULL", "AsUTF8 != NULL") + printf(" PyUnicode_FromString -> %s\n", utf8) + + # 字符串长度 + length: CLong = cpython.PyUnicode_GetLength(py_s1) + testcheck.check(length == 13, "GetLength == 13", "GetLength == 13") + + # 字符串拼接 + py_s2: cpython.PyObject | CPtr = cpython.PyUnicode_FromString(" World") + py_concat: cpython.PyObject | CPtr = cpython.PyUnicode_Concat(py_s1, py_s2) + concat_utf8: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(py_concat) + testcheck.check(py_concat != 0, "Concat != NULL", "Concat != NULL") + printf(" Concat -> %s\n", concat_utf8) + + # 清理 + cpython.Py_DecRef(py_concat) + cpython.Py_DecRef(py_s2) + cpython.Py_DecRef(py_s1) + + cpython.Py_Finalize() + + +def test_list_operations(): + testcheck.section("PyList operations") + cpython.Py_Initialize() + + # 创建列表 + py_list: cpython.PyObject | CPtr = cpython.PyList_New(0) + testcheck.check(py_list != 0, "PyList_New != NULL", "PyList_New != NULL") + + # Append 元素 + py_i1: cpython.PyObject | CPtr = cpython.PyLong_FromLong(10) + py_i2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(20) + py_i3: cpython.PyObject | CPtr = cpython.PyLong_FromLong(30) + cpython.PyList_Append(py_list, py_i1) + cpython.PyList_Append(py_list, py_i2) + cpython.PyList_Append(py_list, py_i3) + + # 检查大小 + list_size: CSizeT = cpython.PyList_Size(py_list) + testcheck.check(list_size == 3, "PyList_Size == 3", "PyList_Size == 3") + + # 获取元素 + item0: cpython.PyObject | CPtr = cpython.PyList_GetItem(py_list, 0) + val0: CLong = cpython.PyLong_AsLong(item0) + testcheck.check(val0 == 10, "list[0] == 10", "list[0] == 10") + + item2: cpython.PyObject | CPtr = cpython.PyList_GetItem(py_list, 2) + val2: CLong = cpython.PyLong_AsLong(item2) + testcheck.check(val2 == 30, "list[2] == 30", "list[2] == 30") + + # 清理 + cpython.Py_DecRef(py_i3) + cpython.Py_DecRef(py_i2) + cpython.Py_DecRef(py_i1) + cpython.Py_DecRef(py_list) + + cpython.Py_Finalize() + + +def test_dict_operations(): + testcheck.section("PyDict operations") + cpython.Py_Initialize() + + # 创建字典 + py_dict: cpython.PyObject | CPtr = cpython.PyDict_New() + testcheck.check(py_dict != 0, "PyDict_New != NULL", "PyDict_New != NULL") + + # 设置键值对 + py_key1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("name") + py_val1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Viper") + cpython.PyDict_SetItem(py_dict, py_key1, py_val1) + + py_key2: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("version") + py_val2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(1) + cpython.PyDict_SetItem(py_dict, py_key2, py_val2) + + # 检查大小 + dict_size: CSizeT = cpython.PyDict_Size(py_dict) + testcheck.check(dict_size == 2, "PyDict_Size == 2", "PyDict_Size == 2") + + # 获取值 + got_val: cpython.PyObject | CPtr = cpython.PyDict_GetItem(py_dict, py_key1) + testcheck.check(got_val != 0, "GetItem(name) != NULL", "GetItem(name) != NULL") + got_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(got_val) + testcheck.check(got_str != 0, "name == Viper", "name == Viper") + printf(" dict['name'] = %s\n", got_str) + + # 使用 GetItemString + got_ver: cpython.PyObject | CPtr = cpython.PyDict_GetItemString(py_dict, "version") + testcheck.check(got_ver != 0, "GetItemString(version) != NULL", "GetItemString(version) != NULL") + ver_val: CLong = cpython.PyLong_AsLong(got_ver) + testcheck.check(ver_val == 1, "version == 1", "version == 1") + + # 清理 + cpython.Py_DecRef(py_val2) + cpython.Py_DecRef(py_key2) + cpython.Py_DecRef(py_val1) + cpython.Py_DecRef(py_key1) + cpython.Py_DecRef(py_dict) + + cpython.Py_Finalize() + + +def test_pyrun_string(): + testcheck.section("PyRun_SimpleString") + cpython.Py_Initialize() + + # 执行简单 Python 代码 + rc: CInt = cpython.PyRun_SimpleString("print('Hello from embedded Python!')") + testcheck.check(rc == 0, "PyRun_SimpleString == 0", "PyRun_SimpleString == 0") + + # 执行赋值 + rc2: CInt = cpython.PyRun_SimpleString("x = 6 * 7") + testcheck.check(rc2 == 0, "PyRun_SimpleString(assign) == 0", "PyRun_SimpleString(assign) == 0") + + # 读取结果 + rc3: CInt = cpython.PyRun_SimpleString("print(f'The answer is {x}')") + testcheck.check(rc3 == 0, "PyRun_SimpleString(print) == 0", "PyRun_SimpleString(print) == 0") + + cpython.Py_Finalize() + + +def test_import_module(): + testcheck.section("PyImport_ImportModule") + cpython.Py_Initialize() + + # 导入 math 模块 + math_mod: cpython.PyObject | CPtr = cpython.PyImport_ImportModule("math") + testcheck.check(math_mod != 0, "import math != NULL", "import math != NULL") + + if math_mod != 0: + # 获取 math.pi + pi_val: cpython.PyObject | CPtr = cpython.PyObject_GetAttrString(math_mod, "pi") + testcheck.check(pi_val != 0, "math.pi != NULL", "math.pi != NULL") + if pi_val != 0: + pi_dbl: CDouble = cpython.PyFloat_AsDouble(pi_val) + testcheck.check(pi_dbl > 3.14 and pi_dbl < 3.15, "math.pi ~= 3.14", "math.pi ~= 3.14") + printf(" math.pi = %f\n", pi_dbl) + cpython.Py_DecRef(pi_val) + + # 获取 math.sqrt 函数 + sqrt_func: cpython.PyObject | CPtr = cpython.PyObject_GetAttrString(math_mod, "sqrt") + testcheck.check(sqrt_func != 0, "math.sqrt != NULL", "math.sqrt != NULL") + if sqrt_func != 0: + cpython.Py_DecRef(sqrt_func) + + cpython.Py_DecRef(math_mod) + + cpython.Py_Finalize() + + +def test_operator_overload(): + testcheck.section("Operator Overload (PyLongObj / PyFloatObj / PyUnicodeObj / PyListObj / PyDictObj)") + cpython.Py_Initialize() + + # --- PyLongObj: from_long, to_long, +, -, *, - (neg) --- + a: cpython.PyLongObj | CPtr = cpython.PyLongObj.from_long(10) + b: cpython.PyLongObj | CPtr = cpython.PyLongObj.from_long(3) + testcheck.check(a != 0 and b != 0, "PyLongObj.from_long != NULL", "PyLongObj.from_long != NULL") + + add_r: cpython.PyLongObj | CPtr = a + b + testcheck.check(add_r.to_long() == 13, "PyLongObj 10 + 3 == 13", "PyLongObj 10 + 3 == 13") + + sub_r: cpython.PyLongObj | CPtr = a - b + testcheck.check(sub_r.to_long() == 7, "PyLongObj 10 - 3 == 7", "PyLongObj 10 - 3 == 7") + + mul_r: cpython.PyLongObj | CPtr = a * b + testcheck.check(mul_r.to_long() == 30, "PyLongObj 10 * 3 == 30", "PyLongObj 10 * 3 == 30") + + neg_r: cpython.PyLongObj | CPtr = -b + testcheck.check(neg_r.to_long() == -3, "PyLongObj -3 == -3", "PyLongObj -3 == -3") + + mul_r.dec_ref() + sub_r.dec_ref() + add_r.dec_ref() + neg_r.dec_ref() + b.dec_ref() + a.dec_ref() + + # --- PyFloatObj: from_double, to_double, +, -, *, /, - (neg) --- + fa: cpython.PyFloatObj | CPtr = cpython.PyFloatObj.from_double(10.0) + fb: cpython.PyFloatObj | CPtr = cpython.PyFloatObj.from_double(4.0) + testcheck.check(fa != 0 and fb != 0, "PyFloatObj.from_double != NULL", "PyFloatObj.from_double != NULL") + + fadd: cpython.PyFloatObj | CPtr = fa + fb + testcheck.check(fadd.to_double() == 14.0, "PyFloatObj 10.0 + 4.0 == 14.0", "PyFloatObj 10.0 + 4.0 == 14.0") + + fsub: cpython.PyFloatObj | CPtr = fa - fb + testcheck.check(fsub.to_double() == 6.0, "PyFloatObj 10.0 - 4.0 == 6.0", "PyFloatObj 10.0 - 4.0 == 6.0") + + fmul: cpython.PyFloatObj | CPtr = fa * fb + testcheck.check(fmul.to_double() == 40.0, "PyFloatObj 10.0 * 4.0 == 40.0", "PyFloatObj 10.0 * 4.0 == 40.0") + + fdiv: cpython.PyFloatObj | CPtr = fa / fb + testcheck.check(fdiv.to_double() == 2.5, "PyFloatObj 10.0 / 4.0 == 2.5", "PyFloatObj 10.0 / 4.0 == 2.5") + + fneg: cpython.PyFloatObj | CPtr = -fb + testcheck.check(fneg.to_double() == -4.0, "PyFloatObj -4.0 == -4.0", "PyFloatObj -4.0 == -4.0") + + fadd.dec_ref() + fsub.dec_ref() + fmul.dec_ref() + fdiv.dec_ref() + fneg.dec_ref() + fb.dec_ref() + fa.dec_ref() + + # --- PyUnicodeObj: from_string, as_utf8, len, + (concat) --- + sa: cpython.PyUnicodeObj | CPtr = cpython.PyUnicodeObj.from_string("Hello") + sb: cpython.PyUnicodeObj | CPtr = cpython.PyUnicodeObj.from_string(" World") + testcheck.check(sa != 0 and sb != 0, "PyUnicodeObj.from_string != NULL", "PyUnicodeObj.from_string != NULL") + + sc: cpython.PyUnicodeObj | CPtr = sa + sb + sc_utf8: t.CChar | CPtr = sc.as_utf8() + testcheck.check(sc_utf8 != 0, "PyUnicodeObj concat != NULL", "PyUnicodeObj concat != NULL") + if sc_utf8 != 0: + printf(" PyUnicodeObj 'Hello' + ' World' = %s\n", sc_utf8) + + sa_len: t.CPtrDiffT = len(sa) + testcheck.check(sa_len == 5, "PyUnicodeObj len('Hello') == 5", "PyUnicodeObj len('Hello') == 5") + + sc.dec_ref() + sb.dec_ref() + sa.dec_ref() + + # --- PyListObj: new, append, len, [], []= --- + lst: cpython.PyListObj | CPtr = cpython.PyListObj.new() + testcheck.check(lst != 0, "PyListObj.new != NULL", "PyListObj.new != NULL") + if lst != 0: + e1: cpython.PyObject | CPtr = cpython.PyLong_FromLong(100) + e2: cpython.PyObject | CPtr = cpython.PyLong_FromLong(200) + e3: cpython.PyObject | CPtr = cpython.PyLong_FromLong(300) + lst.append(e1) + lst.append(e2) + lst.append(e3) + testcheck.check(len(lst) == 3, "PyListObj len == 3", "PyListObj len == 3") + + item0: cpython.PyObject | CPtr = lst[t.CSizeT(0)] + testcheck.check(cpython.PyLong_AsLong(item0) == 100, "PyListObj lst[0] == 100", "PyListObj lst[0] == 100") + + # __setitem__ + e4: cpython.PyObject | CPtr = cpython.PyLong_FromLong(999) + lst[t.CSizeT(1)] = e4 + item1: cpython.PyObject | CPtr = lst[t.CSizeT(1)] + testcheck.check(cpython.PyLong_AsLong(item1) == 999, "PyListObj lst[1] = 999", "PyListObj lst[1] = 999") + + cpython.Py_DecRef(e4) + cpython.Py_DecRef(e3) + cpython.Py_DecRef(e2) + cpython.Py_DecRef(e1) + lst.dec_ref() + + # --- PyDictObj: new, [], []=, len, set_string, get_string --- + dct: cpython.PyDictObj | CPtr = cpython.PyDictObj.new() + testcheck.check(dct != 0, "PyDictObj.new != NULL", "PyDictObj.new != NULL") + if dct != 0: + dk1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("lang") + dv1: cpython.PyObject | CPtr = cpython.PyUnicode_FromString("Viper") + dct[dk1] = dv1 + testcheck.check(len(dct) == 1, "PyDictObj len == 1", "PyDictObj len == 1") + + got_lang: cpython.PyObject | CPtr = dct[dk1] + got_str: t.CChar | CPtr = cpython.PyUnicode_AsUTF8(got_lang) + testcheck.check(got_str != 0, "PyDictObj dct['lang'] == Viper", "PyDictObj dct['lang'] == Viper") + if got_str != 0: + printf(" PyDictObj dct['lang'] = %s\n", got_str) + + # get_string / set_string + dct.set_string("ver", cpython.PyLong_FromLong(3)) + got_ver: cpython.PyObject | CPtr = dct.get_string("ver") + testcheck.check(got_ver != 0 and cpython.PyLong_AsLong(got_ver) == 3, "PyDictObj get_string('ver') == 3", "PyDictObj get_string('ver') == 3") + + cpython.Py_DecRef(dv1) + cpython.Py_DecRef(dk1) + dct.dec_ref() + + cpython.Py_Finalize() + + +def main() -> CInt | t.CExport: + testcheck.begin("CPython Integration Test") + + test_init_finalize() + test_long_operations() + test_string_operations() + test_list_operations() + test_dict_operations() + test_pyrun_string() + test_import_module() + test_operator_overload() + + return testcheck.end() diff --git a/Test/CPythonTest/output/7c5d5ec59c5fb0a8.deps.json b/Test/CPythonTest/output/7c5d5ec59c5fb0a8.deps.json new file mode 100644 index 0000000..3410980 --- /dev/null +++ b/Test/CPythonTest/output/7c5d5ec59c5fb0a8.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb", "main": "fb6cea4f70fade7a"} \ No newline at end of file diff --git a/Test/CPythonTest/output/fb6cea4f70fade7a.deps.json b/Test/CPythonTest/output/fb6cea4f70fade7a.deps.json new file mode 100644 index 0000000..ace5dc1 --- /dev/null +++ b/Test/CPythonTest/output/fb6cea4f70fade7a.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "cpython": "7c5d5ec59c5fb0a8", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/CPythonTest/project.json b/Test/CPythonTest/project.json new file mode 100644 index 0000000..4daa736 --- /dev/null +++ b/Test/CPythonTest/project.json @@ -0,0 +1,35 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "CPythonTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": [ + "-L", "D:\\Python312\\libs", + "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", + "-lkernel32", "-luser32", + "-lpython312", + "-Wl,--allow-multiple-definition" + ], + "output": "CPythonTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": false + } +} diff --git a/Test/CPythonTest/temp/6f62fe05c5ea1ceb.pyi b/Test/CPythonTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/CPythonTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.doc.json b/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.pyi b/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.pyi new file mode 100644 index 0000000..3c94870 --- /dev/null +++ b/Test/CPythonTest/temp/7c5d5ec59c5fb0a8.pyi @@ -0,0 +1,765 @@ +""" +Auto-generated Python stub file from cpython.py +Module: cpython +""" + + +import t, c + +class PyObject: + ob_refcnt: t.CLong + ob_type: t.CPtr + +def Py_INCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_DECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_XINCREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_XDECREF(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_DecRef(op: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def Py_Initialize() -> t.CVoid | t.State: pass + +def Py_Finalize() -> t.CVoid | t.State: pass + +def Py_InitializeEx(initsigs: t.CInt) -> t.CVoid | t.State: pass + +def Py_FinalizeEx() -> t.CInt | t.State: pass + +def Py_IsInitialized() -> t.CInt | t.State: pass + + +class PySTATUS: + func: t.CConst | str + err_msg: t.CConst | str + exitcode: t.CInt +class PyPreConfig: + _config_version: t.CInt + _config_init: t.CInt + parse_argv: t.CInt + isolated: t.CInt + use_environment: t.CInt + utf8_mode: t.CInt + coerce_c_locale: t.CInt + C_locale_warnings: t.CInt + allocator: t.CInt + dev_mode: t.CInt +class PyWideStringList: + length: t.CSizeT + items: t.CPtr +class PyConfig: + _config_version: t.CInt + _config_init: t.CInt + isolated: t.CInt + use_environment: t.CInt + dev_mode: t.CInt + install_signal_handlers: t.CInt + use_hash_seed: t.CInt + hash_seed: t.CUnsignedLong + faulthandler: t.CInt + tracemalloc: t.CInt + import_time: t.CInt + show_ref_count: t.CInt + dump_refs: t.CInt + dump_refs_file: t.CConst | str + malloc_stats: t.CInt + filesystem_errors: t.CInt + pycache_prefix: t.CConst | str + parse_argv: t.CInt + argv: PyWideStringList + program_name: t.CConst | str + xoptions: PyWideStringList + warnoptions: PyWideStringList + site_import: t.CInt + bytes_warning: t.CInt + warn_default_encoding: t.CInt + inspect: t.CInt + interactive: t.CInt + optimization_level: t.CInt + parser_debug: t.CInt + verbose: t.CInt + quiet: t.CInt + home: t.CConst | str + pythonpath_env: t.CConst | str + module_search_paths: PyWideStringList + module_search_paths_set: t.CInt + executable: t.CConst | str + base_executable: t.CConst | str + prefix: t.CConst | str + base_prefix: t.CConst | str + exec_prefix: t.CConst | str + base_exec_prefix: t.CConst | str + platlibdir: t.CConst | str + stdio_filename: t.CConst | str + +def PyConfig_InitPythonConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_InitIsolatedConfig(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_Clear(config: PyConfig | t.CPtr) -> t.CVoid | t.State: pass + +def PyConfig_Read(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass + +def Py_InitializeFromConfig(config: PyConfig | t.CPtr) -> PySTATUS | t.State: pass + +def Py_DecodeLocale(s: t.CConst | str) -> str | t.State: pass + +def PyStatus_Exception(status: PySTATUS) -> t.CInt | t.State: pass + +def PyLong_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromUnsignedLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromLongLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromUnsignedLongLong(v: t.CUnsignedLong) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromSsize_t(v: t.CPtrDiffT) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromSize_t(v: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyLong_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass + +def PyLong_AsLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyLong_AsUnsignedLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass + +def PyLong_AsLongLong(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyLong_AsUnsignedLongLong(obj: PyObject | t.CPtr) -> t.CUnsignedLong | t.State: pass + +def PyLong_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass + +def PyLong_AsSsize_t(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass + +def PyLong_AsSize_t(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyFloat_FromDouble(v: t.CDouble) -> PyObject | t.CPtr | t.State: pass + +def PyFloat_AsDouble(obj: PyObject | t.CPtr) -> t.CDouble | t.State: pass + +def PyFloat_FromString(s: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromStringAndSize(u: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_AsUTF8(obj: PyObject | t.CPtr) -> str | t.State: pass + +def PyUnicode_AsUTF8AndSize(obj: PyObject | t.CPtr, size: t.CPtrDiffT | t.CPtr) -> str | t.State: pass + +def PyUnicode_AsWideChar(obj: PyObject | t.CPtr, wbuf: t.CPtr, bufsize: t.CSizeT) -> t.CSizeT | t.State: pass + +def PyUnicode_GetLength(obj: PyObject | t.CPtr) -> t.CPtrDiffT | t.State: pass + +def PyUnicode_ReadChar(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CUInt32T | t.State: pass + +def PyUnicode_WriteChar(obj: PyObject | t.CPtr, idx: t.CSizeT, ch: t.CUInt32T) -> t.CInt | t.State: pass + +def PyUnicode_Compare(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Concat(l: PyObject | t.CPtr, r: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_Contains(container: PyObject | t.CPtr, element: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Format(fmt: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyUnicode_InternInPlace(s: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyUnicode_InternFromString(u: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromString(v: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromStringAndSize(v: t.CConst | str, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_AsString(obj: PyObject | t.CPtr) -> str | t.State: pass + +def PyBytes_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyBytes_FromObject(o: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyBytes_FromFormat(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyBool_FromLong(v: t.CLong) -> PyObject | t.CPtr | t.State: pass + + +Py_True: t.CExtern | PyObject | t.CPtr | t.State +Py_False: t.CExtern | PyObject | t.CPtr | t.State +Py_None: t.CExtern | PyObject | t.CPtr | t.State + +def PyList_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_Size(lst: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyList_GetItem(lst: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_SetItem(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Insert(lst: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Append(lst: PyObject | t.CPtr, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_GetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyList_SetSlice(lst: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT, items: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Sort(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Reverse(lst: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_AsTuple(lst: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_New(len: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_Size(tup: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyTuple_GetItem(tup: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_SetItem(tup: PyObject | t.CPtr, idx: t.CSizeT, item: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyTuple_GetSlice(tup: PyObject | t.CPtr, lo: t.CSizeT, hi: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyTuple_Pack(n: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyDict_New() -> PyObject | t.CPtr | t.State: pass + +def PyDict_GetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_GetItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyDict_SetItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_SetItemString(d: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_DelItem(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_DelItemString(d: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass + +def PyDict_Size(d: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyDict_Keys(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Values(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Items(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Copy(d: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyDict_Clear(d: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyDict_Contains(d: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Next(d: PyObject | t.CPtr, pos: t.CPtrDiffT | t.CPtr, key: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Merge(d: PyObject | t.CPtr, other: PyObject | t.CPtr, override: t.CInt) -> t.CInt | t.State: pass + +def PyDict_Update(d: PyObject | t.CPtr, other: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Str(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_Repr(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_Type(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_IsInstance(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_IsSubclass(obj: PyObject | t.CPtr, cls: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_HasAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_HasAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyObject_GetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_GetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyObject_SetAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_SetAttrString(obj: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelAttr(obj: PyObject | t.CPtr, name: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelAttrString(obj: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyObject_Call(callable: PyObject | t.CPtr, args: PyObject | t.CPtr, kwargs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallNoArgs(callable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_CallOneArg(callable: PyObject | t.CPtr, arg: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_IsTrue(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Not(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_RichCompare(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyObject_RichCompareBool(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, opid: t.CInt) -> t.CInt | t.State: pass + +def PyObject_Hash(obj: PyObject | t.CPtr) -> t.CLong | t.State: pass + +def PyObject_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyObject_GetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_SetItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_DelItem(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyObject_Dir(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_GetIter(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyIter_Next(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +Py_LT: t.CExtern | t.CInt | t.State +Py_LE: t.CExtern | t.CInt | t.State +Py_EQ: t.CExtern | t.CInt | t.State +Py_NE: t.CExtern | t.CInt | t.State +Py_GT: t.CExtern | t.CInt | t.State +Py_GE: t.CExtern | t.CInt | t.State + +def PyImport_ImportModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyImport_Import(name: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyImport_ImportModuleLevelObject(name: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr, level: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyImport_ReloadModule(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyImport_AddModule(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyImport_GetModuleDict() -> PyObject | t.CPtr | t.State: pass + +def PyImport_ImportModuleEx(name: t.CConst | str, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, fromlist: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +class PyMethodDef: + ml_name: t.CConst | str + ml_meth: t.CPtr + ml_flags: t.CInt + ml_doc: t.CConst | str +class PyModuleDef_Base: + ob_base: PyObject + m_init: t.CPtr + m_index: t.CSizeT + m_copy: PyObject | t.CPtr +class PyModuleDef: + m_base: PyModuleDef_Base + m_name: t.CConst | str + m_doc: t.CConst | str + m_size: t.CPtrDiffT + m_methods: PyMethodDef | t.CPtr + m_slots: t.CPtr + m_traverse: t.CPtr + m_clear: t.CPtr + m_free: t.CPtr + +def PyModule_Create(mod: PyModuleDef | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_Create2(mod: PyModuleDef | t.CPtr, ver: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyModule_GetName(mod: PyObject | t.CPtr) -> str | t.State: pass + +def PyModule_GetDict(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_GetFilenameObject(mod: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyModule_AddObject(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyModule_AddObjectRef(mod: PyObject | t.CPtr, name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyModule_AddIntConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CLong) -> t.CInt | t.State: pass + +def PyModule_AddStringConstant(mod: PyObject | t.CPtr, name: t.CConst | str, val: t.CConst | str) -> t.CInt | t.State: pass + +def PyGILState_Ensure() -> t.CInt | t.State: pass + +def PyGILState_Release(state: t.CInt) -> t.CVoid | t.State: pass + +def PyEval_SaveThread() -> t.CPtr | t.State: pass + +def PyEval_RestoreThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_AcquireThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_ReleaseThread(tstate: t.CPtr) -> t.CVoid | t.State: pass + +def PyEval_AcquireLock() -> t.CVoid | t.State: pass + +def PyEval_ReleaseLock() -> t.CVoid | t.State: pass + +def PyEval_ThreadsInitialized() -> t.CInt | t.State: pass + +def PyEval_InitThreads() -> t.CVoid | t.State: pass + +def PyErr_Occurred() -> PyObject | t.CPtr | t.State: pass + +def PyErr_Clear() -> t.CVoid | t.State: pass + +def PyErr_Fetch(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_Restore(tp: PyObject | t.CPtr, val: PyObject | t.CPtr, tb: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_NormalizeException(tp: PyObject | t.CPtr | t.CPtr, val: PyObject | t.CPtr | t.CPtr, tb: PyObject | t.CPtr | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_SetString(tp: PyObject | t.CPtr, msg: t.CConst | str) -> t.CVoid | t.State: pass + +def PyErr_SetObject(tp: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PyErr_BadArgument() -> t.CPtr | t.State: pass + +def PyErr_NoMemory() -> t.CPtr | t.State: pass + +def PyErr_Print() -> t.CVoid | t.State: pass + +def PyErr_ExceptionMatches(exc: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyErr_GivenExceptionMatches(given: PyObject | t.CPtr, exc: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +PyExc_BaseException: t.CExtern | PyObject | t.CPtr | t.State +PyExc_Exception: t.CExtern | PyObject | t.CPtr | t.State +PyExc_TypeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ValueError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_RuntimeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_KeyError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IndexError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_AttributeError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ImportError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_OSError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IOError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_ZeroDivisionError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_OverflowError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_MemoryError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NameError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_StopIteration: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NotImplementedError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_SyntaxError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_FileNotFoundError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_PermissionError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_IsADirectoryError: t.CExtern | PyObject | t.CPtr | t.State +PyExc_NotADirectoryError: t.CExtern | PyObject | t.CPtr | t.State + +def Py_BuildValue(fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def Py_VaBuildValue(fmt: t.CConst | str, va: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyLong_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyFloat_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyUnicode_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyBytes_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyList_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyTuple_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyDict_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyBool_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyNone_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMem_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyMem_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyMem_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass + +def PyMem_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Malloc(size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Realloc(ptr: t.CPtr, size: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Free(ptr: t.CPtr) -> t.CVoid | t.State: pass + +def PyObject_Calloc(nelem: t.CSizeT, elsize: t.CSizeT) -> t.CPtr | t.State: pass + +def PyObject_Init(obj: PyObject | t.CPtr, tp: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyObject_InitVar(obj: PyObject | t.CPtr, tp: t.CPtr, size: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PyFile_FromString(filename: t.CConst | str, mode: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyFile_AsFile(file: PyObject | t.CPtr) -> t.CPtr | t.State: pass + +def PyFile_WriteObject(obj: PyObject | t.CPtr, file: PyObject | t.CPtr, flags: t.CInt) -> t.CInt | t.State: pass + +def PyFile_WriteString(s: t.CConst | str, file: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def Py_CompileString(s: t.CConst | str, filename: t.CConst | str, start: t.CInt) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalCode(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalCodeEx(code: PyObject | t.CPtr, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr, args: PyObject | t.CPtr | t.CPtr, argc: t.CInt, kws: PyObject | t.CPtr | t.CPtr, kwc: t.CInt, defs: PyObject | t.CPtr | t.CPtr, defc: t.CInt, kwdefs: PyObject | t.CPtr, closure: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_EvalFrame(frame: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallObject(callable: PyObject | t.CPtr, args: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallFunction(callable: PyObject | t.CPtr, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyEval_CallMethod(obj: PyObject | t.CPtr, method: t.CConst | str, fmt: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyRun_SimpleString(cmd: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_SimpleFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_AnyFile(fp: t.CPtr, filename: t.CConst | str) -> t.CInt | t.State: pass + +def PyRun_String(s: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyRun_File(fp: t.CPtr, filename: t.CConst | str, start: t.CInt, globs: PyObject | t.CPtr, locs: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + + +Py_single_input: t.CExtern | t.CInt | t.State +Py_file_input: t.CExtern | t.CInt | t.State +Py_eval_input: t.CExtern | t.CInt | t.State + +def PyCapsule_New(ptr: t.CPtr, name: t.CConst | str, dtor: t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyCapsule_GetPointer(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CPtr | t.State: pass + +def PyCapsule_GetName(cap: PyObject | t.CPtr) -> str | t.State: pass + +def PyCapsule_IsValid(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyCapsule_SetPointer(cap: PyObject | t.CPtr, ptr: t.CPtr) -> t.CInt | t.State: pass + +def PyCapsule_SetName(cap: PyObject | t.CPtr, name: t.CConst | str) -> t.CInt | t.State: pass + +def PyCapsule_SetDestructor(cap: PyObject | t.CPtr, dtor: t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Concat(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySequence_Repeat(obj: PyObject | t.CPtr, count: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_GetItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_SetItem(obj: PyObject | t.CPtr, idx: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_DelItem(obj: PyObject | t.CPtr, idx: t.CSizeT) -> t.CInt | t.State: pass + +def PySequence_GetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> PyObject | t.CPtr | t.State: pass + +def PySequence_SetSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_DelSlice(obj: PyObject | t.CPtr, start: t.CSizeT, end: t.CSizeT) -> t.CInt | t.State: pass + +def PySequence_Contains(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySequence_Index(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_Count(obj: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySequence_List(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySequence_Tuple(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_Size(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyMapping_Length(obj: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PyMapping_HasKey(obj: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_HasKeyString(obj: PyObject | t.CPtr, key: t.CConst | str) -> t.CInt | t.State: pass + +def PyMapping_GetItemString(obj: PyObject | t.CPtr, key: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_SetItemString(obj: PyObject | t.CPtr, key: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyMapping_Keys(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Values(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMapping_Items(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Add(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Subtract(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Multiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_MatrixMultiply(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_FloorDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_TrueDivide(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Modulo(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Power(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr, o3: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Negative(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Positive(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Absolute(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Invert(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Lshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Rshift(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_And(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Xor(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Or(o1: PyObject | t.CPtr, o2: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Index(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Long(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Float(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyNumber_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyWeakref_NewRef(obj: PyObject | t.CPtr, cb: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyWeakref_GetObject(ref: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyWeakref_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +class PyGetSetDef: + name: t.CConst | str + get: t.CPtr + set: t.CPtr + doc: t.CConst | str + closure: t.CPtr +class PyMemberDef: + name: t.CConst | str + type: t.CInt + offset: t.CPtrDiffT + flags: t.CInt + doc: t.CConst | str + +def PySys_GetObject(name: t.CConst | str) -> PyObject | t.CPtr | t.State: pass + +def PySys_SetObject(name: t.CConst | str, val: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySys_ResetWarnOptions() -> t.CVoid | t.State: pass + +def PySys_AddWarnOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySYS_AddXOption(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySys_WriteStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_WriteStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_FormatStdout(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PySys_FormatStderr(fmt: t.CConst | str) -> t.CVoid | t.State: pass + +def PyMemoryView_FromObject(obj: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyMemoryView_GetContiguous(obj: PyObject | t.CPtr, btype: t.CInt, order: t.CChar) -> PyObject | t.CPtr | t.State: pass + +def PyMemoryView_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySet_Add(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Discard(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Contains(s: PyObject | t.CPtr, key: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySet_Size(s: PyObject | t.CPtr) -> t.CSizeT | t.State: pass + +def PySet_Clear(s: PyObject | t.CPtr) -> t.CVoid | t.State: pass + +def PySet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PyFrozenSet_New(iterable: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PyFrozenSet_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + +def PySlice_New(start: PyObject | t.CPtr, stop: PyObject | t.CPtr, step: PyObject | t.CPtr) -> PyObject | t.CPtr | t.State: pass + +def PySlice_GetIndicesEx(sl: PyObject | t.CPtr, length: t.CSizeT, start: t.CSizeT | t.CPtr, stop: t.CSizeT | t.CPtr, step: t.CSizeT | t.CPtr, slicelen: t.CSizeT | t.CPtr) -> t.CInt | t.State: pass + +def PySlice_Unpack(sl: PyObject | t.CPtr, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT | t.CPtr) -> t.CVoid | t.State: pass + +def PySlice_AdjustIndices(length: t.CSizeT, start: t.CPtrDiffT | t.CPtr, stop: t.CPtrDiffT | t.CPtr, step: t.CPtrDiffT) -> t.CSizeT | t.State: pass + +def PySlice_Check(obj: PyObject | t.CPtr) -> t.CInt | t.State: pass + + +class PyLongObj: + ptr: PyObject | t.CPtr + def __new__(self: PyLongObj) -> 'PyLongObj' | t.CPtr: pass + @staticmethod + def from_long(val: t.CLong) -> 'PyLongObj' | t.CPtr: pass + def __add__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __sub__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __mul__(self: PyLongObj, other: PyLongObj | t.CPtr) -> 'PyLongObj' | t.CPtr: pass + def __neg__(self: PyLongObj) -> 'PyLongObj' | t.CPtr: pass + def to_long(self: PyLongObj) -> t.CLong: pass + def __str__(self: PyLongObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyLongObj) -> t.CInt: pass +class PyFloatObj: + ptr: PyObject | t.CPtr + def __new__(self: PyFloatObj) -> 'PyFloatObj' | t.CPtr: pass + @staticmethod + def from_double(val: t.CDouble) -> 'PyFloatObj' | t.CPtr: pass + def __add__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __sub__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __mul__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __truediv__(self: PyFloatObj, other: PyFloatObj | t.CPtr) -> 'PyFloatObj' | t.CPtr: pass + def __neg__(self: PyFloatObj) -> 'PyFloatObj' | t.CPtr: pass + def to_double(self: PyFloatObj) -> t.CDouble: pass + def __str__(self: PyFloatObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyFloatObj) -> t.CInt: pass +class PyUnicodeObj: + ptr: PyObject | t.CPtr + def __new__(self: PyUnicodeObj) -> 'PyUnicodeObj' | t.CPtr: pass + @staticmethod + def from_string(s: t.CChar | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: pass + def __add__(self: PyUnicodeObj, other: PyUnicodeObj | t.CPtr) -> 'PyUnicodeObj' | t.CPtr: pass + def __len__(self: PyUnicodeObj) -> t.CPtrDiffT: pass + def as_utf8(self: PyUnicodeObj) -> t.CChar | t.CPtr: pass + def __str__(self: PyUnicodeObj) -> PyObject | t.CPtr: pass + def dec_ref(self: PyUnicodeObj) -> t.CInt: pass +class PyListObj: + ptr: PyObject | t.CPtr + def __new__(self: PyListObj) -> 'PyListObj' | t.CPtr: pass + @staticmethod + def new() -> 'PyListObj' | t.CPtr: pass + def __getitem__(self: PyListObj, idx: t.CSizeT) -> PyObject | t.CPtr: pass + def __setitem__(self: PyListObj, idx: t.CSizeT, val: PyObject | t.CPtr) -> t.CInt: pass + def __len__(self: PyListObj) -> t.CSizeT: pass + def append(self: PyListObj, item: PyObject | t.CPtr) -> t.CInt: pass + def dec_ref(self: PyListObj) -> t.CInt: pass +class PyDictObj: + ptr: PyObject | t.CPtr + def __new__(self: PyDictObj) -> 'PyDictObj' | t.CPtr: pass + @staticmethod + def new() -> 'PyDictObj' | t.CPtr: pass + def __getitem__(self: PyDictObj, key: PyObject | t.CPtr) -> PyObject | t.CPtr: pass + def __setitem__(self: PyDictObj, key: PyObject | t.CPtr, val: PyObject | t.CPtr) -> t.CInt: pass + def __len__(self: PyDictObj) -> t.CSizeT: pass + def get_string(self: PyDictObj, key: t.CChar | t.CPtr) -> PyObject | t.CPtr: pass + def set_string(self: PyDictObj, key: t.CChar | t.CPtr, val: PyObject | t.CPtr) -> t.CInt: pass + def dec_ref(self: PyDictObj) -> t.CInt: pass \ No newline at end of file diff --git a/Test/CPythonTest/temp/7e529fe7a078cfef.pyi b/Test/CPythonTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/CPythonTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/CPythonTest/temp/_phase1_manifest.json b/Test/CPythonTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..15aa5ad --- /dev/null +++ b/Test/CPythonTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\CPythonTest\\App\\cpython.py": {"sha1": "7c5d5ec59c5fb0a8", "mtime": 1781613876.1585274, "size": 39169}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\CPythonTest\\App\\main.py": {"sha1": "fb6cea4f70fade7a", "mtime": 1782403889.06048, "size": 13919}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/CPythonTest/temp/_sha1_map.txt b/Test/CPythonTest/temp/_sha1_map.txt new file mode 100644 index 0000000..c8ce35d --- /dev/null +++ b/Test/CPythonTest/temp/_sha1_map.txt @@ -0,0 +1,7 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7c5d5ec59c5fb0a8:cpython.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py +fb6cea4f70fade7a:main.py diff --git a/Test/CPythonTest/temp/_shared_sym.json b/Test/CPythonTest/temp/_shared_sym.json new file mode 100644 index 0000000..bfedc6a Binary files /dev/null and b/Test/CPythonTest/temp/_shared_sym.json differ diff --git a/Test/CPythonTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/CPythonTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/CPythonTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/CPythonTest/temp/dd3002730623424b.pyi b/Test/CPythonTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/CPythonTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/CPythonTest/temp/f5522571bcce7bcb.pyi b/Test/CPythonTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/CPythonTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/CPythonTest/temp/fb6cea4f70fade7a.doc.json b/Test/CPythonTest/temp/fb6cea4f70fade7a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/CPythonTest/temp/fb6cea4f70fade7a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/CPythonTest/temp/fb6cea4f70fade7a.pyi b/Test/CPythonTest/temp/fb6cea4f70fade7a.pyi new file mode 100644 index 0000000..12b9442 --- /dev/null +++ b/Test/CPythonTest/temp/fb6cea4f70fade7a.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from t import CInt, CLong, CDouble, CPtr, CSizeT +from stdio import printf +import cpython +import testcheck + +def test_init_finalize() -> t.CInt: pass + +def test_long_operations() -> t.CInt: pass + +def test_string_operations() -> t.CInt: pass + +def test_list_operations() -> t.CInt: pass + +def test_dict_operations() -> t.CInt: pass + +def test_pyrun_string() -> t.CInt: pass + +def test_import_module() -> t.CInt: pass + +def test_operator_overload() -> t.CInt: pass + +def main() -> CInt | t.CExport: pass diff --git a/Test/ClosureTest/App/main.py b/Test/ClosureTest/App/main.py new file mode 100644 index 0000000..89a2fd8 --- /dev/null +++ b/Test/ClosureTest/App/main.py @@ -0,0 +1,164 @@ +import t +import stdio +import testcheck + +# ============================================================ +# Test 1: global 在嵌套函数中 +# ============================================================ +g_val: t.CInt = 100 + +def outer_modify_global(): + global g_val + def inner_modify(): + global g_val + g_val += 50 + inner_modify() + +def test_global_in_nested(): + testcheck.section("Test 1: global in nested function") + global g_val + g_val = 100 + outer_modify_global() + testcheck.check(g_val == 150, "global nested (g_val == 150)", "global nested (expect 150)") + +# ============================================================ +# Test 2: nonlocal 在多层嵌套中 +# ============================================================ +def test_nonlocal_multi_level(): + testcheck.section("Test 2: nonlocal multi-level") + x: t.CInt = 1 + def level1(): + nonlocal x + x += 10 + def level2(): + nonlocal x + x += 100 + def level3(): + nonlocal x + x += 1000 + level3() + level2() + level1() + testcheck.check(x == 1111, "nonlocal multi (x == 1111)", "nonlocal multi (expect 1111)") + +# ============================================================ +# Test 3: global 和 nonlocal 混合 +# ============================================================ +g_mixed: t.CInt = 0 + +def test_global_nonlocal_mixed(): + testcheck.section("Test 3: global + nonlocal mixed") + global g_mixed + g_mixed = 0 + local_var: t.CInt = 0 + def modify_both(): + nonlocal local_var + global g_mixed + g_mixed += 10 + local_var += 20 + modify_both() + modify_both() + testcheck.check(g_mixed == 20 and local_var == 40, "mixed (global==20 local==40)", "mixed (expect global=20 local=40)") + +# ============================================================ +# Test 4: lambda 捕获多个变量 +# ============================================================ +def test_lambda_multi_capture(): + testcheck.section("Test 4: lambda multi-capture") + a: t.CInt = 10 + b: t.CInt = 20 + f = lambda: a + b + r: t.CInt = f() + testcheck.check(r == 30, "lambda multi-capture (a+b == 30)", "lambda multi-capture (expect 30)") + +# ============================================================ +# Test 5: lambda 带参数 + 捕获 +# ============================================================ +def test_lambda_arg_capture(): + testcheck.section("Test 5: lambda arg + capture") + base: t.CInt = 100 + add = lambda x: base + x + r: t.CInt = add(42) + testcheck.check(r == 142, "lambda arg+capture (100+42 == 142)", "lambda arg+capture (expect 142)") + +# ============================================================ +# Test 6: 嵌套函数返回值 +# ============================================================ +def test_nested_return(): + testcheck.section("Test 6: nested function return") + x: t.CInt = 5 + def double_x() -> t.CInt: + nonlocal x + x = x * 2 + return x + r: t.CInt = double_x() + testcheck.check(r == 10 and x == 10, "nested return (r==10 x==10)", "nested return (expect r=10 x=10)") + +# ============================================================ +# Test 7: nonlocal 与循环变量 +# ============================================================ +def test_nonlocal_loop(): + testcheck.section("Test 7: nonlocal with loop") + total: t.CInt = 0 + def accumulate(n: t.CInt): + nonlocal total + for i in range(n): + total += i + accumulate(5) + accumulate(5) + # 0+1+2+3+4 = 10, twice = 20 + testcheck.check(total == 20, "nonlocal loop (total == 20)", "nonlocal loop (expect 20)") + +# ============================================================ +# Test 8: global 数组操作 +# ============================================================ +g_arr: t.CArray[t.CInt, 4] = [0] + +def test_global_array(): + testcheck.section("Test 8: global array") + global g_arr + g_arr[0] = 10 + g_arr[1] = 20 + g_arr[2] = 30 + g_arr[3] = 40 + total: t.CInt = g_arr[0] + g_arr[1] + g_arr[2] + g_arr[3] + testcheck.check(total == 100, "global array (total == 100)", "global array (expect 100)") + +# ============================================================ +# Test 9: 嵌套函数条件调用 +# ============================================================ +def test_nested_conditional(): + testcheck.section("Test 9: nested conditional call") + result: t.CInt = 0 + def set_result(val: t.CInt): + nonlocal result + result = val + if 1: + set_result(42) + else: + set_result(99) + testcheck.check(result == 42, "nested conditional (result == 42)", "nested conditional (expect 42)") + +# ============================================================ +# Test 10: lambda 链式调用 +# ============================================================ +def test_lambda_chain(): + testcheck.section("Test 10: lambda chain") + f1 = lambda: 10 + f2 = lambda: 20 + r: t.CInt = f1() + f2() + testcheck.check(r == 30, "lambda chain (10+20 == 30)", "lambda chain (expect 30)") + +def main() -> t.CInt: + testcheck.begin("ClosureAdvancedTest: 高级闭包测试") + test_global_in_nested() + test_nonlocal_multi_level() + test_global_nonlocal_mixed() + test_lambda_multi_capture() + test_lambda_arg_capture() + test_nested_return() + test_nonlocal_loop() + test_global_array() + test_nested_conditional() + test_lambda_chain() + return testcheck.end() diff --git a/Test/ClosureTest/output/52e6b51da061882f.deps.json b/Test/ClosureTest/output/52e6b51da061882f.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/ClosureTest/output/52e6b51da061882f.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ClosureTest/project.json b/Test/ClosureTest/project.json new file mode 100644 index 0000000..a40284e --- /dev/null +++ b/Test/ClosureTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ClosureTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "ClosureTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ClosureTest/temp/52e6b51da061882f.doc.json b/Test/ClosureTest/temp/52e6b51da061882f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ClosureTest/temp/52e6b51da061882f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ClosureTest/temp/52e6b51da061882f.pyi b/Test/ClosureTest/temp/52e6b51da061882f.pyi new file mode 100644 index 0000000..edc39b5 --- /dev/null +++ b/Test/ClosureTest/temp/52e6b51da061882f.pyi @@ -0,0 +1,43 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +g_val: t.CExtern | t.CInt + +def outer_modify_global() -> t.CInt: pass + +def test_global_in_nested() -> t.CInt: pass + +def test_nonlocal_multi_level() -> t.CInt: pass + + +g_mixed: t.CExtern | t.CInt + +def test_global_nonlocal_mixed() -> t.CInt: pass + +def test_lambda_multi_capture() -> t.CInt: pass + +def test_lambda_arg_capture() -> t.CInt: pass + +def test_nested_return() -> t.CInt: pass + +def test_nonlocal_loop() -> t.CInt: pass + + +g_arr: t.CExtern | t.CArray[t.CInt, 4] + +def test_global_array() -> t.CInt: pass + +def test_nested_conditional() -> t.CInt: pass + +def test_lambda_chain() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/ClosureTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ClosureTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ClosureTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ClosureTest/temp/7e529fe7a078cfef.pyi b/Test/ClosureTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ClosureTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ClosureTest/temp/_phase1_manifest.json b/Test/ClosureTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..53006ec --- /dev/null +++ b/Test/ClosureTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ClosureTest\\App\\main.py": {"sha1": "52e6b51da061882f", "mtime": 1782266177.611862, "size": 5497}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ClosureTest/temp/_sha1_map.txt b/Test/ClosureTest/temp/_sha1_map.txt new file mode 100644 index 0000000..bf5f6eb --- /dev/null +++ b/Test/ClosureTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +52e6b51da061882f:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ClosureTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ClosureTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ClosureTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ClosureTest/temp/dd3002730623424b.pyi b/Test/ClosureTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ClosureTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ClosureTest/temp/f5522571bcce7bcb.pyi b/Test/ClosureTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ClosureTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ClosureTest2/App/main.py b/Test/ClosureTest2/App/main.py new file mode 100644 index 0000000..24e8bfb --- /dev/null +++ b/Test/ClosureTest2/App/main.py @@ -0,0 +1,220 @@ +import t +import stdio +import testcheck + +# ============================================================ +# Test 1: nonlocal 在条件分支中修改 +# ============================================================ +def test_nonlocal_conditional(): + testcheck.section("Test 1: nonlocal in conditional branches") + x: t.CInt = 0 + def modify(flag: t.CInt): + nonlocal x + if flag: + x += 10 + else: + x += 100 + modify(1) + modify(0) + modify(1) + testcheck.check(x == 120, "nonlocal conditional (x == 120)", "nonlocal conditional (expect 120)") + +# ============================================================ +# Test 2: 多个nonlocal变量同时修改 +# ============================================================ +def test_nonlocal_multi_var(): + testcheck.section("Test 2: multiple nonlocal variables") + a: t.CInt = 1 + b: t.CInt = 2 + c_val: t.CInt = 3 + def swap_add(): + nonlocal a, b, c_val + tmp: t.CInt = a + a = b + b = c_val + c_val = tmp + swap_add() + testcheck.check(a == 2 and b == 3 and c_val == 1, "multi nonlocal (a==2 b==3 c==1)", "multi nonlocal (expect 2 3 1)") + +# ============================================================ +# Test 3: global 在多个函数中交叉修改 +# ============================================================ +g_counter: t.CInt = 0 + +def inc_global(): + global g_counter + g_counter += 1 + +def add_global(n: t.CInt): + global g_counter + g_counter += n + +def test_global_cross_functions(): + testcheck.section("Test 3: global cross-function modification") + global g_counter + g_counter = 0 + inc_global() + inc_global() + add_global(10) + inc_global() + testcheck.check(g_counter == 13, "global cross (g_counter == 13)", "global cross (expect 13)") + +# ============================================================ +# Test 4: 嵌套函数修改外层局部变量（引用捕获） +# ============================================================ +def test_nested_ref_capture(): + testcheck.section("Test 4: nested function ref capture") + arr: t.CArray[t.CInt, 4] = [0] + def fill(): + nonlocal arr + for i in range(4): + arr[i] = (i + 1) * 10 + fill() + total: t.CInt = arr[0] + arr[1] + arr[2] + arr[3] + testcheck.check(total == 100, "ref capture (total == 100)", "ref capture (expect 100)") + +# ============================================================ +# Test 5: lambda捕获后修改原变量 +# ============================================================ +def test_lambda_capture_then_modify(): + testcheck.section("Test 5: lambda capture then modify") + x: t.CInt = 10 + f = lambda: x + r1: t.CInt = f() + x = 20 + r2: t.CInt = f() + testcheck.check(r1 == 10 and r2 == 10, "lambda value capture (before==10 after==10)", "lambda value capture (expect 10 10)") + +# ============================================================ +# Test 6: 嵌套函数中nonlocal与循环结合 +# ============================================================ +def test_nonlocal_loop_accumulate(): + testcheck.section("Test 6: nonlocal with loop accumulate") + sum_val: t.CInt = 0 + count: t.CInt = 0 + def add_numbers(n: t.CInt): + nonlocal sum_val, count + for i in range(1, n + 1): + sum_val += i + count += 1 + add_numbers(5) # 1+2+3+4+5=15 + add_numbers(3) # 1+2+3=6 + testcheck.check(sum_val == 21 and count == 8, "loop accumulate (sum==21 count==8)", "loop accumulate (expect 21 8)") + +# ============================================================ +# Test 7: 三层嵌套中每层修改不同变量 +# ============================================================ +def test_deep_nested_vars(): + testcheck.section("Test 7: deep nested different vars") + a: t.CInt = 0 + b: t.CInt = 0 + c_val: t.CInt = 0 + def level1(): + nonlocal a + a += 1 + def level2(): + nonlocal b + b += 10 + def level3(): + nonlocal c_val + c_val += 100 + level3() + level2() + level1() + level1() + testcheck.check(a == 2 and b == 20 and c_val == 200, "deep nested (a==2 b==20 c==200)", "deep nested (expect 2 20 200)") + +# ============================================================ +# Test 8: global + nonlocal 在同一嵌套函数中 +# ============================================================ +g_state: t.CInt = 0 + +def test_global_nonlocal_same_func(): + testcheck.section("Test 8: global+nonlocal in same nested func") + global g_state + g_state = 0 + local_state: t.CInt = 0 + def worker(n: t.CInt): + nonlocal local_state + global g_state + for i in range(n): + g_state += 1 + local_state += 2 + worker(5) + testcheck.check(g_state == 5 and local_state == 10, "global+nonlocal same (g==5 l==10)", "global+nonlocal same (expect 5 10)") + +# ============================================================ +# Test 9: lambda带默认参数与捕获 +# ============================================================ +def test_lambda_default_and_capture(): + testcheck.section("Test 9: lambda with arg and capture") + base: t.CInt = 100 + multiply = lambda x: base * x + r1: t.CInt = multiply(3) + r2: t.CInt = multiply(5) + testcheck.check(r1 == 300 and r2 == 500, "lambda arg+capture (100*3==300 100*5==500)", "lambda arg+capture (expect 300 500)") + +# ============================================================ +# Test 10: 嵌套函数多次调用累积 +# ============================================================ +def test_nested_accumulate(): + testcheck.section("Test 10: nested function accumulate") + total: t.CInt = 0 + def add_val(n: t.CInt): + nonlocal total + total += n + add_val(10) + add_val(20) + add_val(30) + testcheck.check(total == 60, "nested accumulate (total == 60)", "nested accumulate (expect 60)") + +# ============================================================ +# Test 11: nonlocal在while循环中 +# ============================================================ +def test_nonlocal_while(): + testcheck.section("Test 11: nonlocal in while loop") + result: t.CInt = 0 + def countdown(n: t.CInt): + nonlocal result + i: t.CInt = n + while i > 0: + result += i + i -= 1 + countdown(5) # 5+4+3+2+1=15 + testcheck.check(result == 15, "nonlocal while (result == 15)", "nonlocal while (expect 15)") + +# ============================================================ +# Test 12: 嵌套函数修改结构体字段 +# ============================================================ +class Point: + x: t.CInt + y: t.CInt + +def test_nested_modify_struct(): + testcheck.section("Test 12: nested modify struct") + p: Point = Point() + p.x = 0 + p.y = 0 + def translate(dx: t.CInt, dy: t.CInt): + nonlocal p + p.x += dx + p.y += dy + translate(10, 20) + translate(5, 15) + testcheck.check(p.x == 15 and p.y == 35, "nested struct (x==15 y==35)", "nested struct (expect 15 35)") + +def main() -> t.CInt: + testcheck.begin("ClosureTest2: Advanced Closure/Lambda/Global/Nonlocal") + test_nonlocal_conditional() + test_nonlocal_multi_var() + test_global_cross_functions() + test_nested_ref_capture() + test_lambda_capture_then_modify() + test_nonlocal_loop_accumulate() + test_deep_nested_vars() + test_global_nonlocal_same_func() + test_lambda_default_and_capture() + test_nested_accumulate() + test_nonlocal_while() + test_nested_modify_struct() + return testcheck.end() diff --git a/Test/ClosureTest2/output/440f18f9dce85a95.deps.json b/Test/ClosureTest2/output/440f18f9dce85a95.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/ClosureTest2/output/440f18f9dce85a95.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ClosureTest2/project.json b/Test/ClosureTest2/project.json new file mode 100644 index 0000000..d1af5fb --- /dev/null +++ b/Test/ClosureTest2/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ClosureTest2", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "ClosureTest2.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ClosureTest2/temp/440f18f9dce85a95.doc.json b/Test/ClosureTest2/temp/440f18f9dce85a95.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ClosureTest2/temp/440f18f9dce85a95.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ClosureTest2/temp/440f18f9dce85a95.pyi b/Test/ClosureTest2/temp/440f18f9dce85a95.pyi new file mode 100644 index 0000000..0b2c4a4 --- /dev/null +++ b/Test/ClosureTest2/temp/440f18f9dce85a95.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def test_nonlocal_conditional() -> t.CInt: pass + +def test_nonlocal_multi_var() -> t.CInt: pass + + +g_counter: t.CExtern | t.CInt + +def inc_global() -> t.CInt: pass + +def add_global(n: t.CInt) -> t.CInt: pass + +def test_global_cross_functions() -> t.CInt: pass + +def test_nested_ref_capture() -> t.CInt: pass + +def test_lambda_capture_then_modify() -> t.CInt: pass + +def test_nonlocal_loop_accumulate() -> t.CInt: pass + +def test_deep_nested_vars() -> t.CInt: pass + + +g_state: t.CExtern | t.CInt + +def test_global_nonlocal_same_func() -> t.CInt: pass + +def test_lambda_default_and_capture() -> t.CInt: pass + +def test_nested_accumulate() -> t.CInt: pass + +def test_nonlocal_while() -> t.CInt: pass + + +class Point: + x: t.CInt + y: t.CInt + +def test_nested_modify_struct() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/ClosureTest2/temp/6f62fe05c5ea1ceb.pyi b/Test/ClosureTest2/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ClosureTest2/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ClosureTest2/temp/7e529fe7a078cfef.pyi b/Test/ClosureTest2/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ClosureTest2/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ClosureTest2/temp/_phase1_manifest.json b/Test/ClosureTest2/temp/_phase1_manifest.json new file mode 100644 index 0000000..b1275e7 --- /dev/null +++ b/Test/ClosureTest2/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ClosureTest2\\App\\main.py": {"sha1": "440f18f9dce85a95", "mtime": 1782266179.2717218, "size": 7586}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ClosureTest2/temp/_sha1_map.txt b/Test/ClosureTest2/temp/_sha1_map.txt new file mode 100644 index 0000000..c01f543 --- /dev/null +++ b/Test/ClosureTest2/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +440f18f9dce85a95:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ClosureTest2/temp/bbdf3bbd4c3bc28c.pyi b/Test/ClosureTest2/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ClosureTest2/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ClosureTest2/temp/dd3002730623424b.pyi b/Test/ClosureTest2/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ClosureTest2/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ClosureTest2/temp/f5522571bcce7bcb.pyi b/Test/ClosureTest2/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ClosureTest2/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DecoratorMarkerTest/App/main.py b/Test/DecoratorMarkerTest/App/main.py new file mode 100644 index 0000000..09d4eac --- /dev/null +++ b/Test/DecoratorMarkerTest/App/main.py @@ -0,0 +1,77 @@ +import t +from stdio import printf +import testcheck + + +# 装饰器形式：@t.CStruct 等同于 class Point(t.CStruct): +@t.CStruct +class Point: + x: t.CInt + y: t.CInt + + +# 装饰器形式：@t.CEnum 等同于 class Color(t.CEnum): +@t.CEnum +class Color: + RED = 1 + GREEN = 2 + BLUE = 3 + + +# 装饰器形式：@t.CUnion 等同于 class Data(t.CUnion): +@t.CUnion +class Data: + i: t.CInt + f: t.CFloat + + +# 装饰器形式：@t.CExport 等同于 def test() -> t.CInt | t.CExport: +@t.CExport +def test() -> t.CInt: + printf("test_export called\n") + return 42 + + +# 装饰器形式：@t.CStatic 等同于 def static_func() -> t.CInt | t.CStatic: +@t.CStatic +def static_func() -> t.CInt: + return 100 + + +# 混合形式：装饰器 + 注解 +@t.CStatic +def mixed_func() -> t.CInt | t.CExport: + return 200 + + +def main() -> t.CInt: + testcheck.begin("DecoratorMarkerTest: 装饰器标记测试") + + p: Point = Point() + p.x = 10 + p.y = 20 + printf("Point: (%d, %d)\n", p.x, p.y) + testcheck.check(p.x == 10 and p.y == 20, "Point (10, 20)", "Point expect (10, 20)") + + c: t.CInt = Color.GREEN + printf("Color: %d\n", c) + testcheck.check(c == 2, "Color.GREEN == 2", "Color.GREEN expect 2") + + d: Data = Data() + d.i = 65 + printf("Data.i: %d\n", d.i) + testcheck.check(d.i == 65, "Data.i == 65", "Data.i expect 65") + + ret: t.CInt = test() + printf("test() returned: %d\n", ret) + testcheck.check(ret == 42, "test() returned 42", "test() expect 42") + + s: t.CInt = static_func() + printf("static_func() returned: %d\n", s) + testcheck.check(s == 100, "static_func() returned 100", "static_func() expect 100") + + m: t.CInt = mixed_func() + printf("mixed_func() returned: %d\n", m) + testcheck.check(m == 200, "mixed_func() returned 200", "mixed_func() expect 200") + + return testcheck.end() diff --git a/Test/DecoratorMarkerTest/output/b1122357fee8fd68.deps.json b/Test/DecoratorMarkerTest/output/b1122357fee8fd68.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/DecoratorMarkerTest/output/b1122357fee8fd68.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/DecoratorMarkerTest/project.json b/Test/DecoratorMarkerTest/project.json new file mode 100644 index 0000000..c25d371 --- /dev/null +++ b/Test/DecoratorMarkerTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "DecoratorMarkerTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "DecoratorMarkerTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/DecoratorMarkerTest/temp/6f62fe05c5ea1ceb.pyi b/Test/DecoratorMarkerTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/DecoratorMarkerTest/temp/7e529fe7a078cfef.pyi b/Test/DecoratorMarkerTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/DecoratorMarkerTest/temp/_phase1_manifest.json b/Test/DecoratorMarkerTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..4e31ba8 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\DecoratorMarkerTest\\App\\main.py": {"sha1": "b1122357fee8fd68", "mtime": 1782110269.7636156, "size": 1840}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/DecoratorMarkerTest/temp/_sha1_map.txt b/Test/DecoratorMarkerTest/temp/_sha1_map.txt new file mode 100644 index 0000000..235a656 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +b1122357fee8fd68:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.doc.json b/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.pyi b/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.pyi new file mode 100644 index 0000000..97850c4 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/b1122357fee8fd68.pyi @@ -0,0 +1,36 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdio import printf +import testcheck + +@t.CStruct +class Point: + x: t.CInt + y: t.CInt +@t.CEnum +class Color: + RED = 1 + GREEN = 2 + BLUE = 3 +@t.CUnion +class Data: + i: t.CInt + f: t.CFloat + +@t.CExport +def test() -> t.CInt: pass + +@t.CStatic +def static_func() -> t.CInt: pass + +@t.CStatic +def mixed_func() -> t.CInt | t.CExport: pass + +def main() -> t.CInt: pass diff --git a/Test/DecoratorMarkerTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/DecoratorMarkerTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DecoratorMarkerTest/temp/dd3002730623424b.pyi b/Test/DecoratorMarkerTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/DecoratorMarkerTest/temp/f5522571bcce7bcb.pyi b/Test/DecoratorMarkerTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/DecoratorMarkerTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DecoratorTest/App/main.py b/Test/DecoratorTest/App/main.py new file mode 100644 index 0000000..97553c5 --- /dev/null +++ b/Test/DecoratorTest/App/main.py @@ -0,0 +1,298 @@ +import t +from stdio import printf +import testcheck + +# ============================================================ +# 自定义装饰器处理函数 +# 调用约定 (v4)： +# i32 decor_name(i8* ctx, i8* func_name, i32 phase, +# i8* args_ptr, i8* ret_ptr, i8* decor_args_ptr) +# +# ctx: 栈帧局部上下文（32字节），每次调用独立，线程/递归安全 +# args_ptr: 可读写，装饰器可修改函数入参 +# ret_ptr: 后置阶段可读写，装饰器可修改返回值 +# 返回值（前置阶段）： +# 0 = 跳过原函数调用 +# 1 = 正常调用一次 +# N = 循环调用 N 次 +# ============================================================ + +# @log 装饰器：打印函数进入/退出，返回 1（正常调用一次） +def log(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + if phase == 0: + printf("[LOG] >> %s enter\n", func_name) + return 1 + else: + printf("[LOG] << %s exit\n", func_name) + return 0 + +# @trace 装饰器：详细追踪 +def trace(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + if phase == 0: + printf("[TRACE] calling %s...\n", func_name) + return 1 + else: + printf("[TRACE] %s returned\n", func_name) + return 0 + +# @timing 装饰器：使用全局变量记录计时 +# 注意：理想情况下应使用 ctx 栈帧局部上下文替代全局变量， +# 但 Viper 当前前端尚不支持 i8* 到 i32* 的指针转型， +# 因此暂时使用全局变量演示计时逻辑 +_timing_tmp: t.CInt = 0 + +def timing(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + global _timing_tmp + if phase == 0: + _timing_tmp = 100 + printf("[TIMING] %s start at t=%d\n", func_name, _timing_tmp) + return 1 + else: + elapsed: t.CInt = 105 - _timing_tmp + printf("[TIMING] %s took %d ms\n", func_name, elapsed) + return 0 + +# @repeat 装饰器（带参数，流程劫持）：循环调用原函数 N 次 +def repeat(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + if phase == 0: + printf("[REPEAT] %s will run 3 times\n", func_name) + return 3 + else: + printf("[REPEAT] %s all iterations done\n", func_name) + return 0 + +# @skip 装饰器：跳过原函数调用（流程劫持，返回 0） +_skip_count: t.CInt = 0 + +def skip(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + global _skip_count + if phase == 0: + _skip_count = _skip_count + 1 + printf("[SKIP] %s skipped! (call #%d)\n", func_name, _skip_count) + return 0 + else: + printf("[SKIP] %s post (ret_ptr=%p)\n", func_name, ret_ptr) + return 0 + +# @count_calls 装饰器：统计函数被调用次数（递归保护测试用） +_call_count: t.CInt = 0 + +def count_calls(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, + args_ptr: t.CVoid | t.CPtr, + ret_ptr: t.CVoid | t.CPtr, + decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: + global _call_count + if phase == 0: + _call_count = _call_count + 1 + printf("[COUNT] %s call #%d\n", func_name, _call_count) + return 1 + else: + printf("[COUNT] %s done (total calls: %d)\n", func_name, _call_count) + return 0 + + +# ============================================================ +# 测试 1：单个装饰器 @log +# ============================================================ + +@log +def add(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: + return a + b + + +# ============================================================ +# 测试 2：链式装饰器 @log @trace => log(trace(f)) +# 执行顺序：log_pre → trace_pre → f → trace_post → log_post +# ============================================================ + +@log +@trace +def multiply(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: + return a * b + + +# ============================================================ +# 测试 3：装饰器修饰 void 返回函数 +# ============================================================ + +@log +def greet(name: str) -> t.CVoid | t.CExport: + printf("Hello, %s!\n", name) + + +# ============================================================ +# 测试 4：装饰器修饰无参数函数 +# ============================================================ + +@log +def get_value() -> t.CInt | t.CExport: + return 42 + + +# ============================================================ +# 测试 5：@timing 装饰器 +# ============================================================ + +@timing +def compute(x: t.CInt) -> t.CInt | t.CExport: + return x * x + 1 + + +# ============================================================ +# 测试 6：带参数装饰器 @repeat(3) — 流程劫持，循环调用 3 次 +# ============================================================ + +@repeat(3) +def echo(msg: str) -> t.CVoid | t.CExport: + printf(" echo: %s\n", msg) + + +# ============================================================ +# 测试 7：链式 @timing @log +# ============================================================ + +@timing +@log +def power(base: t.CInt, exp: t.CInt) -> t.CInt | t.CExport: + result: t.CInt = 1 + i: t.CInt = 0 + while i < exp: + result = result * base + i = i + 1 + return result + + +# ============================================================ +# 测试 8：@skip 装饰器 — 流程劫持，跳过原函数调用 +# ============================================================ + +@skip +def dangerous() -> t.CInt | t.CExport: + printf("This should NOT be printed!\n") + return 999 + + +# ============================================================ +# 测试 9：@repeat(3) 修饰有返回值的函数 +# ============================================================ + +@repeat(3) +def accumulate(x: t.CInt) -> t.CInt | t.CExport: + printf(" accumulate(%d)\n", x) + return x * 10 + + +# ============================================================ +# 测试 10：递归装饰保护 — @count_calls 修饰递归函数 +# v4 递归保护：最外层 wrapper 检查全局标志位， +# 递归调用时跳过所有装饰逻辑，仅最外层调用触发装饰器 +# ============================================================ + +@count_calls +def factorial(n: t.CInt) -> t.CInt | t.CExport: + if n <= 1: + return 1 + return n * factorial(n - 1) + + +# ============================================================ +# 测试 11：@log 修饰递归函数 — 验证递归保护 +# 递归保护确保只有最外层调用打印 LOG，内部递归不触发 +# ============================================================ + +@log +def fib(n: t.CInt) -> t.CInt | t.CExport: + if n <= 1: + return n + return fib(n - 1) + fib(n - 2) + + +# ============================================================ +# 主函数 +# ============================================================ + +def main() -> t.CInt | t.CExport: + testcheck.begin("Decorator Test v4") + + # 测试 1：单个 @log + testcheck.section("Test 1: single @log") + r1: t.CInt = add(3, 4) + printf("add(3,4) = %d\n", r1) + testcheck.check(r1 == 7, "add(3,4) = 7", "add(3,4) expect 7") + + # 测试 2：链式 @log @trace + testcheck.section("Test 2: chained @log @trace") + r2: t.CInt = multiply(5, 6) + printf("multiply(5,6) = %d\n", r2) + testcheck.check(r2 == 30, "multiply(5,6) = 30", "multiply(5,6) expect 30") + + # 测试 3：void 返回 + testcheck.section("Test 3: @log on void function") + greet("Viper") + testcheck.ok("@log on void function greet") + + # 测试 4：无参数 + testcheck.section("Test 4: @log on no-arg function") + r4: t.CInt = get_value() + printf("get_value() = %d\n", r4) + testcheck.check(r4 == 42, "get_value() = 42", "get_value() expect 42") + + # 测试 5：@timing + testcheck.section("Test 5: @timing") + r5: t.CInt = compute(7) + printf("compute(7) = %d\n", r5) + testcheck.check(r5 == 50, "compute(7) = 50", "compute(7) expect 50") + + # 测试 6：@repeat(3) — 流程劫持，循环 3 次 + testcheck.section("Test 6: @repeat(3) flow hijack") + echo("hello") + testcheck.ok("@repeat(3) flow hijack echo") + + # 测试 7：链式 @timing @log + testcheck.section("Test 7: chained @timing @log") + r7: t.CInt = power(2, 10) + printf("power(2,10) = %d\n", r7) + testcheck.check(r7 == 1024, "power(2,10) = 1024", "power(2,10) expect 1024") + + # 测试 8：@skip — 跳过原函数调用 + testcheck.section("Test 8: @skip flow hijack") + r8: t.CInt = dangerous() + printf("dangerous() = %d (should be 0, skipped)\n", r8) + testcheck.check(r8 == 0, "dangerous() skipped = 0", "dangerous() expect 0") + + # 测试 9：@repeat(3) 有返回值 + testcheck.section("Test 9: @repeat(3) with return value") + r9: t.CInt = accumulate(5) + printf("accumulate(5) = %d (last iteration result)\n", r9) + testcheck.check(r9 == 50, "accumulate(5) = 50", "accumulate(5) expect 50") + + # 测试 10：递归装饰保护 — @count_calls + testcheck.section("Test 10: recursive decoration protection") + _call_count = 0 + r10: t.CInt = factorial(5) + printf("factorial(5) = %d (should be 120)\n", r10) + printf(" decorator call count = %d (should be 1, not 5)\n", _call_count) + testcheck.check(r10 == 120 and _call_count == 1, "factorial(5)=120, calls=1", "factorial(5) expect 120, calls=1") + + # 测试 11：@log 递归保护 — fib + testcheck.section("Test 11: @log on recursive fib") + r11: t.CInt = fib(6) + printf("fib(6) = %d (should be 8)\n", r11) + testcheck.check(r11 == 8, "fib(6) = 8", "fib(6) expect 8") + + return testcheck.end() diff --git a/Test/DecoratorTest/output/814e6b31adfc6256.deps.json b/Test/DecoratorTest/output/814e6b31adfc6256.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/DecoratorTest/output/814e6b31adfc6256.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/DecoratorTest/project.json b/Test/DecoratorTest/project.json new file mode 100644 index 0000000..f3bbc02 --- /dev/null +++ b/Test/DecoratorTest/project.json @@ -0,0 +1,28 @@ +{ + "name": "DecoratorTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "DecoratorTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/DecoratorTest/temp/6f62fe05c5ea1ceb.pyi b/Test/DecoratorTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/DecoratorTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/DecoratorTest/temp/7e529fe7a078cfef.pyi b/Test/DecoratorTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/DecoratorTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/DecoratorTest/temp/814e6b31adfc6256.doc.json b/Test/DecoratorTest/temp/814e6b31adfc6256.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/DecoratorTest/temp/814e6b31adfc6256.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/DecoratorTest/temp/814e6b31adfc6256.pyi b/Test/DecoratorTest/temp/814e6b31adfc6256.pyi new file mode 100644 index 0000000..0a0bccf --- /dev/null +++ b/Test/DecoratorTest/temp/814e6b31adfc6256.pyi @@ -0,0 +1,69 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +from stdio import printf +import testcheck + +def log(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +def trace(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_timing_tmp: t.CExtern | t.CInt + +def timing(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +def repeat(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_skip_count: t.CExtern | t.CInt + +def skip(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + + +_call_count: t.CExtern | t.CInt + +def count_calls(ctx: t.CVoid | t.CPtr, func_name: str, phase: t.CInt, args_ptr: t.CVoid | t.CPtr, ret_ptr: t.CVoid | t.CPtr, decor_args_ptr: t.CVoid | t.CPtr) -> t.CInt | t.CExport: pass + +@log +def add(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: pass + +@log +@trace +def multiply(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: pass + +@log +def greet(name: str) -> t.CVoid | t.CExport: pass + +@log +def get_value() -> t.CInt | t.CExport: pass + +@timing +def compute(x: t.CInt) -> t.CInt | t.CExport: pass + +@repeat(3) +def echo(msg: str) -> t.CVoid | t.CExport: pass + +@timing +@log +def power(base: t.CInt, exp: t.CInt) -> t.CInt | t.CExport: pass + +@skip +def dangerous() -> t.CInt | t.CExport: pass + +@repeat(3) +def accumulate(x: t.CInt) -> t.CInt | t.CExport: pass + +@count_calls +def factorial(n: t.CInt) -> t.CInt | t.CExport: pass + +@log +def fib(n: t.CInt) -> t.CInt | t.CExport: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/DecoratorTest/temp/_phase1_manifest.json b/Test/DecoratorTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..cfc1fbe --- /dev/null +++ b/Test/DecoratorTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\DecoratorTest\\App\\main.py": {"sha1": "814e6b31adfc6256", "mtime": 1782110086.8867075, "size": 10275}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/DecoratorTest/temp/_sha1_map.txt b/Test/DecoratorTest/temp/_sha1_map.txt new file mode 100644 index 0000000..25fd6ed --- /dev/null +++ b/Test/DecoratorTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +814e6b31adfc6256:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/DecoratorTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/DecoratorTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/DecoratorTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DecoratorTest/temp/dd3002730623424b.pyi b/Test/DecoratorTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/DecoratorTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/DecoratorTest/temp/f5522571bcce7bcb.pyi b/Test/DecoratorTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/DecoratorTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DictTest/App/main.py b/Test/DictTest/App/main.py new file mode 100644 index 0000000..c580522 --- /dev/null +++ b/Test/DictTest/App/main.py @@ -0,0 +1,229 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import testcheck + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("DictTest: 万能 dict (Variant 存储) 测试") + + # 分配内存竞技场 + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + + # === Test 1: dict set_int / c.Deref 基本操作 === + testcheck.section("Test 1: dict set_int / c.Deref 基本操作") + d: dict = dict(bd) + d.set_int("alpha", 10) + d.set_int("beta", 20) + d.set_int("gamma", 30) + n: t.CSizeT = len(d) + stdio.printf("len=%lu\n", n) + testcheck.check(n == 3, "dict len OK (3)", "dict len FAILED") + + p1: t.CPtr = d["alpha"] + stdio.printf("p1=%p\n", p1) + p2: t.CPtr = d["beta"] + stdio.printf("p2=%p\n", p2) + p3: t.CPtr = d["gamma"] + stdio.printf("p3=%p\n", p3) + v1: int = c.Deref(p1) + stdio.printf("v1=%d\n", v1) + v2: int = c.Deref(p2) + v3: int = c.Deref(p3) + stdio.printf("alpha=%d beta=%d gamma=%d\n", v1, v2, v3) + testcheck.check(v1 == 10, "dict[alpha]==10 OK", "dict[alpha] FAILED") + testcheck.check(v2 == 20, "dict[beta]==20 OK", "dict[beta] FAILED") + testcheck.check(v3 == 30, "dict[gamma]==30 OK", "dict[gamma] FAILED") + + # === Test 2: 更新已有键 === + testcheck.section("Test 2: 更新已有键") + d.set_int("beta", 99) + p2b: t.CPtr = d["beta"] + v2b: int = c.Deref(p2b) + stdio.printf("beta(updated)=%d\n", v2b) + testcheck.check(v2b == 99, "dict update OK", "dict update FAILED") + testcheck.check(len(d) == 3, "dict len still 3 OK", "dict len changed FAILED") + + # === Test 3: get with default === + testcheck.section("Test 3: get with default") + gv: t.CPtr = d.get("missing", None) + testcheck.check(gv == None, "dict get missing OK (None)", "dict get missing FAILED") + + gp: t.CPtr = d.get("alpha", None) + ga: int = c.Deref(gp) + stdio.printf("get(alpha)=%d\n", ga) + testcheck.check(ga == 10, "dict get existing OK", "dict get existing FAILED") + + # === Test 4: contains === + testcheck.section("Test 4: contains") + has_alpha: t.CInt = d.contains("alpha") + has_missing: t.CInt = d.contains("missing") + stdio.printf("contains(alpha)=%d contains(missing)=%d\n", has_alpha, has_missing) + testcheck.check(has_alpha == 1, "contains alpha OK", "contains alpha FAILED") + testcheck.check(has_missing == 0, "not contains missing OK", "contains missing FAILED") + + # === Test 5: 迭代器 === + testcheck.section("Test 5: 迭代器 for key in dict") + iter_count: int = 0 + for key in d: + kp: t.CPtr = d[key] + val: int = c.Deref(kp) + stdio.printf("dict[%s]=%d\n", key, val) + iter_count += 1 + testcheck.check(iter_count == 3, "dict iter OK (3)", "dict iter FAILED") + + # === Test 6: dict set_str 字符串值 === + testcheck.section("Test 6: dict set_str 字符串值") + d2: dict = dict(bd) + d2.set_str("name", "Alice") + d2.set_str("city", "Beijing") + d2.set_str("lang", "TransPyC") + n2: t.CSizeT = len(d2) + stdio.printf("d2 len=%lu\n", n2) + testcheck.check(n2 == 3, "dict set_str len OK (3)", "dict set_str len FAILED") + + name_p: t.CPtr = d2["name"] + city_p: t.CPtr = d2["city"] + lang_p: t.CPtr = d2["lang"] + name: str = name_p + city: str = city_p + lang: str = lang_p + stdio.printf("name=%s city=%s lang=%s\n", name, city, lang) + testcheck.check(name == "Alice", "dict str name OK", "dict str name FAILED") + testcheck.check(city == "Beijing", "dict str city OK", "dict str city FAILED") + testcheck.check(lang == "TransPyC", "dict str lang OK", "dict str lang FAILED") + + # === Test 7: dict[str] 迭代 === + testcheck.section("Test 7: dict 迭代 (str values)") + str_iter_count: int = 0 + for k in d2: + svp: t.CPtr = d2[k] + sv: str = svp + stdio.printf("d2[%s]=%s\n", k, sv) + str_iter_count += 1 + testcheck.check(str_iter_count == 3, "dict str iter OK (3)", "dict str iter FAILED") + + # === Test 8: 扩容测试 === + testcheck.section("Test 8: 扩容测试 (>8 个元素)") + d3: dict = dict(bd) + i: int = 0 + while i < 20: + key_buf: str = stdlib.malloc(16) + stdio.sprintf(key_buf, "key%d", i) + d3.set_int(key_buf, i * 100) + i += 1 + n3: t.CSizeT = len(d3) + stdio.printf("d3 len=%lu (after 20 inserts)\n", n3) + testcheck.check(n3 == 20, "dict expand OK (20)", "dict expand FAILED") + + # 验证几个值 + k5: str = stdlib.malloc(16) + stdio.sprintf(k5, "key5") + p5: t.CPtr = d3[k5] + v5: int = c.Deref(p5) + stdio.printf("d3[key5]=%d\n", v5) + testcheck.check(v5 == 500, "dict expand key5 OK", "dict expand key5 FAILED") + + k15: str = stdlib.malloc(16) + stdio.sprintf(k15, "key15") + p15: t.CPtr = d3[k15] + v15: int = c.Deref(p15) + stdio.printf("d3[key15]=%d\n", v15) + testcheck.check(v15 == 1500, "dict expand key15 OK", "dict expand key15 FAILED") + + # === Test 9: JSON dumps === + testcheck.section("Test 9: JSON dumps") + pool_mem: bytes = stdlib.malloc(65536) + pool: memhub.MemPool = memhub.MemPool(pool_mem, 65536) + + jd: dict = dict(bd) + jd.set_int("x", 100) + jd.set_int("y", 200) + jd.set_int("z", 300) + + json_str: str = jd.dumps(pool) + stdio.printf("JSON dumps: %s\n", json_str) + testcheck.check(json_str != None, "dumps OK", "dumps FAILED (null)") + + # === Test 10: JSON loads === + testcheck.section("Test 10: JSON loads") + jd2: dict = dict(bd) + jd2.loads(pool, "{\"x\":1,\"y\":2,\"z\":3}") + xp: t.CPtr = jd2["x"] + yp: t.CPtr = jd2["y"] + zp: t.CPtr = jd2["z"] + xv: int = c.Deref(xp) + yv: int = c.Deref(yp) + zv: int = c.Deref(zp) + stdio.printf("loads: x=%d y=%d z=%d\n", xv, yv, zv) + testcheck.check(xv == 1, "loads x OK", "loads x FAILED") + testcheck.check(yv == 2, "loads y OK", "loads y FAILED") + testcheck.check(zv == 3, "loads z OK", "loads z FAILED") + + # === Test 11: JSON dumps (str values) === + testcheck.section("Test 11: JSON dumps (str values)") + jd3: dict = dict(bd) + jd3.set_str("name", "Alice") + jd3.set_str("city", "Beijing") + + json_str2: str = jd3.dumps(pool) + stdio.printf("JSON dumps (str): %s\n", json_str2) + testcheck.check(json_str2 != None, "dumps str OK", "dumps str FAILED (null)") + + # === Test 12: JSON round-trip === + testcheck.section("Test 12: JSON round-trip") + rt: dict = dict(bd) + rt.set_int("a", 1) + rt.set_int("b", 2) + rt_str: str = rt.dumps(pool) + stdio.printf("round-trip dump: %s\n", rt_str) + rt2: dict = dict(bd) + rt2.loads(pool, rt_str) + rta_p: t.CPtr = rt2["a"] + rtb_p: t.CPtr = rt2["b"] + rta: int = c.Deref(rta_p) + rtb: int = c.Deref(rtb_p) + stdio.printf("round-trip: a=%d b=%d\n", rta, rtb) + testcheck.check(rta == 1, "round-trip a OK", "round-trip a FAILED") + testcheck.check(rtb == 2, "round-trip b OK", "round-trip b FAILED") + + # === Test 13: 嵌套 dict === + testcheck.section("Test 13: 嵌套 dict") + outer: dict = dict(bd) + inner: dict = dict(bd) + inner.set_int("a", 1) + inner.set_int("b", 2) + outer.set_dict("sub", inner) + + sub_p: t.CPtr = outer["sub"] + sub: dict = sub_p + sa_p: t.CPtr = sub["a"] + sb_p: t.CPtr = sub["b"] + sa: int = c.Deref(sa_p) + sb: int = c.Deref(sb_p) + stdio.printf("nested: sub.a=%d sub.b=%d\n", sa, sb) + testcheck.check(sa == 1, "nested sub.a OK", "nested sub.a FAILED") + testcheck.check(sb == 2, "nested sub.b OK", "nested sub.b FAILED") + + # === Test 14: get_vtype 类型标签 === + testcheck.section("Test 14: get_vtype 类型标签") + vt1: t.CInt = d.get_vtype("alpha") + vt2: t.CInt = d2.get_vtype("name") + vt3: t.CInt = outer.get_vtype("sub") + stdio.printf("vtype(alpha)=%d vtype(name)=%d vtype(sub)=%d\n", vt1, vt2, vt3) + testcheck.check(vt1 == 1, "vtype int OK", "vtype int FAILED") + testcheck.check(vt2 == 4, "vtype str OK", "vtype str FAILED") + testcheck.check(vt3 == 5, "vtype dict OK", "vtype dict FAILED") + + stdlib.free(pool_mem) + stdlib.free(arena) + return testcheck.end() diff --git a/Test/DictTest/project.json b/Test/DictTest/project.json new file mode 100644 index 0000000..3d3fb5f --- /dev/null +++ b/Test/DictTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "DictTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32"], + "output": "DictTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/DictTest/temp/0035c95a18d4f8e8.pyi b/Test/DictTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/DictTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/DictTest/temp/067c78e9f121dce3.pyi b/Test/DictTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/DictTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/DictTest/temp/06f53cc594b4ac6c.pyi b/Test/DictTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/DictTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DictTest/temp/20cd49775c100a38.pyi b/Test/DictTest/temp/20cd49775c100a38.pyi new file mode 100644 index 0000000..df04de7 --- /dev/null +++ b/Test/DictTest/temp/20cd49775c100a38.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from json.__writer.py +Module: json.__writer +""" + + +import t, c +from stdint import * +import memhub +import string +from stdio import snprintf +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT + +def _write_char(buf: t.CChar | t.CPtr, pos: t.CSizeT, ch: t.CChar) -> t.CSizeT: pass + +def _write_str(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_string_escaped(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_value(buf: t.CChar | t.CPtr, pos: t.CSizeT, val: JsonValue | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CSizeT: pass + +def write(pool: memhub.MemManager | t.CPtr, val: JsonValue | t.CPtr, pretty: bool) -> t.CChar | t.CPtr: pass diff --git a/Test/DictTest/temp/240a9a4157959a9f.pyi b/Test/DictTest/temp/240a9a4157959a9f.pyi new file mode 100644 index 0000000..c230800 --- /dev/null +++ b/Test/DictTest/temp/240a9a4157959a9f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from json.__parser.py +Module: json.__parser +""" + + +import t, c +from stdint import * +import memhub +import string +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT, null, bool_val, int_val, float_val, string_val, array, object + +class ParseState: + src: t.CChar | t.CPtr + pos: t.CSizeT + len: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: ParseState, src: t.CChar | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(s: ParseState | t.CPtr) -> t.CChar: pass + +def _advance(s: ParseState | t.CPtr) -> t.CChar: pass + +def _skip_whitespace(s: ParseState | t.CPtr) -> t.CInt: pass + +def _expect(s: ParseState | t.CPtr, ch: t.CChar) -> bool: pass + +def _parse_string(s: ParseState | t.CPtr) -> t.CChar | t.CPtr: pass + +def _parse_number(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_value(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_array(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_object(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def parse(pool: memhub.MemManager | t.CPtr, src: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass diff --git a/Test/DictTest/temp/271ea3decb810db2.pyi b/Test/DictTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/DictTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/DictTest/temp/57288496f7c2d1ad.pyi b/Test/DictTest/temp/57288496f7c2d1ad.pyi new file mode 100644 index 0000000..0624d9a --- /dev/null +++ b/Test/DictTest/temp/57288496f7c2d1ad.pyi @@ -0,0 +1,71 @@ +""" +Auto-generated Python stub file from json.__init__.py +Module: json.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperio + +JSON_NULL: t.CDefine = 0 +JSON_BOOL: t.CDefine = 1 +JSON_INT: t.CDefine = 2 +JSON_FLOAT: t.CDefine = 3 +JSON_STRING: t.CDefine = 4 +JSON_ARRAY: t.CDefine = 5 +JSON_OBJECT: t.CDefine = 6 + +class JsonValue: + vtype: t.CInt + bool_val: t.CInt + int_val: t.CInt64T + float_val: t.CDouble + str_val: t.CChar | t.CPtr + next: 'JsonValue' | t.CPtr + key: t.CChar | t.CPtr + child: 'JsonValue' | t.CPtr + child_count: t.CSizeT + def __new__(self: JsonValue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def type(self: JsonValue) -> t.CInt: pass + def is_null(self: JsonValue) -> bool: pass + def is_bool(self: JsonValue) -> bool: pass + def is_int(self: JsonValue) -> bool: pass + def is_float(self: JsonValue) -> bool: pass + def is_string(self: JsonValue) -> bool: pass + def is_array(self: JsonValue) -> bool: pass + def is_object(self: JsonValue) -> bool: pass + def as_bool(self: JsonValue) -> bool: pass + def as_int(self: JsonValue) -> t.CInt64T: pass + def as_float(self: JsonValue) -> t.CDouble: pass + def as_string(self: JsonValue) -> t.CChar | t.CPtr: pass + def __len__(self: JsonValue) -> t.CSizeT: pass + def __getitem__(self: JsonValue, key: t.CChar | t.CPtr) -> 'JsonValue' | t.CPtr: pass + def __setitem__(self: JsonValue, key: t.CChar | t.CPtr, val: 'JsonValue' | t.CPtr) -> t.CInt: pass + def get_item(self: JsonValue, index: t.CSizeT) -> 'JsonValue' | t.CPtr: pass + def _pool_alloc(self: JsonValue, size: t.CSizeT) -> t.CChar | t.CPtr: pass + def _append_child(self: JsonValue, child: 'JsonValue' | t.CPtr) -> t.CInt: pass + +def null(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def bool_val(pool: memhub.MemManager | t.CPtr, val: bool) -> JsonValue | t.CPtr: pass + +def int_val(pool: memhub.MemManager | t.CPtr, val: t.CInt64T) -> JsonValue | t.CPtr: pass + +def float_val(pool: memhub.MemManager | t.CPtr, val: t.CDouble) -> JsonValue | t.CPtr: pass + +def string_val(pool: memhub.MemManager | t.CPtr, val: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def object(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def array_append(pool: memhub.MemManager | t.CPtr, arr: JsonValue | t.CPtr, item: JsonValue | t.CPtr) -> t.CInt: pass + +def object_set(pool: memhub.MemManager | t.CPtr, obj: JsonValue | t.CPtr, key: t.CChar | t.CPtr, val: JsonValue | t.CPtr) -> t.CInt: pass + + +from .__parser import parse +from .__writer import write \ No newline at end of file diff --git a/Test/DictTest/temp/6446627d4f07a1b5.pyi b/Test/DictTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/DictTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/DictTest/temp/6f62fe05c5ea1ceb.pyi b/Test/DictTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/DictTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/DictTest/temp/72e2d5ccb7cedcf1.pyi b/Test/DictTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/DictTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/DictTest/temp/7e529fe7a078cfef.pyi b/Test/DictTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/DictTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/DictTest/temp/90c53dd6db8d41cf.pyi b/Test/DictTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/DictTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/DictTest/temp/_phase1_manifest.json b/Test/DictTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..97063a5 --- /dev/null +++ b/Test/DictTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\DictTest\\App\\main.py": {"sha1": "b3c2b99b6f3be4c7", "mtime": 1782827868.8446941, "size": 8245}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_dict.py": {"sha1": "e3e7b6de8d7d8b03", "mtime": 1782919714.6072268, "size": 12055}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__init__.py": {"sha1": "57288496f7c2d1ad", "mtime": 1783239811.8873823, "size": 7653}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__parser.py": {"sha1": "240a9a4157959a9f", "mtime": 1782826918.100951, "size": 7517}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__writer.py": {"sha1": "20cd49775c100a38", "mtime": 1782826901.8906734, "size": 3784}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/DictTest/temp/_sha1_map.txt b/Test/DictTest/temp/_sha1_map.txt new file mode 100644 index 0000000..640f738 --- /dev/null +++ b/Test/DictTest/temp/_sha1_map.txt @@ -0,0 +1,15 @@ +20cd49775c100a38:includes/json\__writer.py +240a9a4157959a9f:includes/json\__parser.py +271ea3decb810db2:includes/atom.py +57288496f7c2d1ad:includes/json\__init__.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +b3c2b99b6f3be4c7:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +e3e7b6de8d7d8b03:includes/_dict.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/DictTest/temp/ab6e54ba9a669f76.pyi b/Test/DictTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/DictTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/DictTest/temp/b3c2b99b6f3be4c7.doc.json b/Test/DictTest/temp/b3c2b99b6f3be4c7.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/DictTest/temp/b3c2b99b6f3be4c7.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/DictTest/temp/b3c2b99b6f3be4c7.pyi b/Test/DictTest/temp/b3c2b99b6f3be4c7.pyi new file mode 100644 index 0000000..155b94e --- /dev/null +++ b/Test/DictTest/temp/b3c2b99b6f3be4c7.pyi @@ -0,0 +1,17 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import testcheck + +def main() -> CInt | CExport: pass diff --git a/Test/DictTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/DictTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/DictTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DictTest/temp/c9f4be41ca1cc2b4.pyi b/Test/DictTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/DictTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/DictTest/temp/dd3002730623424b.pyi b/Test/DictTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/DictTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/DictTest/temp/e3e7b6de8d7d8b03.pyi b/Test/DictTest/temp/e3e7b6de8d7d8b03.pyi new file mode 100644 index 0000000..ac69b1b --- /dev/null +++ b/Test/DictTest/temp/e3e7b6de8d7d8b03.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from _dict.py +Module: _dict +""" + + +import t, c +import memhub +import string +import json +from stdint import * + +VARIANT_NONE: t.CDefine = 0 # 无类型/空值 +VARIANT_INT: t.CDefine = 1 # int (i64) +VARIANT_FLOAT: t.CDefine = 2 # float (f32) +VARIANT_DOUBLE: t.CDefine = 3 # double (f64) +VARIANT_STR: t.CDefine = 4 # str (i8*) +VARIANT_DICT: t.CDefine = 5 # dict* +VARIANT_LIST: t.CDefine = 6 # list* +VARIANT_BOOL: t.CDefine = 7 # bool (i1/i32) +VARIANT_PTR: t.CDefine = 8 # 自定义指针类型 +_VARIANT_SIZE: t.CDefine = 16 + +class dict: + __keys__: t.CPtr + __values__: t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 8 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: dict, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: dict, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def __len__(self: dict) -> t.CSizeT: pass + def _find(self: dict, key: str) -> t.CSizeT: pass + def __getitem__(self: dict, key: str) -> t.CPtr: pass + def __setitem__(self: dict, key: str, val: t.CNeedPtr) -> t.CInt: pass + def _set_variant_raw(self: dict, key: str, vtype: t.CInt, ptr: t.CPtr) -> t.CInt: pass + def set_int(self: dict, key: str, val: int) -> t.CInt: pass + def set_str(self: dict, key: str, val: str) -> t.CInt: pass + def set_float(self: dict, key: str, val: float) -> t.CInt: pass + def set_double(self: dict, key: str, val: t.CDouble) -> t.CInt: pass + def set_dict(self: dict, key: str, val: dict | t.CPtr) -> t.CInt: pass + def set_list(self: dict, key: str, val: t.CPtr) -> t.CInt: pass + def set_bool(self: dict, key: str, val: t.CInt) -> t.CInt: pass + def get_vtype(self: dict, key: str) -> t.CInt: pass + def get(self: dict, key: str, default: t.CPtr) -> t.CPtr: pass + def contains(self: dict, key: str) -> t.CInt: pass + def __iter__(self: dict) -> dict | t.CPtr: pass + def __next__(self: dict) -> str: pass + def to_json(self: dict, pool: memhub.MemManager | t.CPtr) -> json.JsonValue | t.CPtr: pass + def from_json(self: dict, pool: memhub.MemManager | t.CPtr, root: json.JsonValue | t.CPtr) -> t.CInt: pass + def dumps(self: dict, pool: memhub.MemManager | t.CPtr) -> str: pass + def loads(self: dict, pool: memhub.MemManager | t.CPtr, json_str: str) -> t.CInt: pass \ No newline at end of file diff --git a/Test/DictTest/temp/ee084e9fc6ee413a.pyi b/Test/DictTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/DictTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/DictTest/temp/f5522571bcce7bcb.pyi b/Test/DictTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/DictTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DictTest/temp/f6b51804a0ba8ff0.pyi b/Test/DictTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/DictTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/DocTest/App/main.py b/Test/DocTest/App/main.py new file mode 100644 index 0000000..995ee3d --- /dev/null +++ b/Test/DocTest/App/main.py @@ -0,0 +1,70 @@ +import t +import stdio +import string +import c +import testcheck +import remote + + +# 同模块函数（有 docstring） +def local_func(): + "Local function docstring." + return 0 + + +# 同模块函数（无 docstring） +def no_doc_func(): + return 0 + + +# 同模块类（有 docstring） +@t.Object +class LocalClass: + "Local class docstring." + value: t.CInt + + def local_method(self): + "Local method docstring." + return 0 + + +def _check_doc_adaptive(name: str, doc: str, expected: str): + """自适应检查 docstring：非空时验证匹配，空时视为优化剥离""" + if doc == None: + testcheck.ok(f"{name}: __doc__ optimized out (null)") + return + if doc == expected: + testcheck.ok(f"{name}: __doc__ matches") + else: + testcheck.fail(f"{name}: __doc__ mismatch (got '{doc}', expected '{expected}')") + + +def _check_doc_null(name: str, doc: str): + """检查 docstring 为 null（无 docstring 的符号）""" + if doc == None: + testcheck.ok(f"{name}: __doc__ is null") + else: + testcheck.fail(f"{name}: __doc__ should be null but got '{doc}'") + + +def test_local_doc(): + testcheck.section("Local __doc__") + _check_doc_adaptive("local_func", local_func.__doc__, "Local function docstring.") + _check_doc_adaptive("LocalClass", LocalClass.__doc__, "Local class docstring.") + _check_doc_adaptive("LocalClass.local_method", LocalClass.local_method.__doc__, "Local method docstring.") + _check_doc_null("no_doc_func", no_doc_func.__doc__) + + +def test_cross_module_doc(): + testcheck.section("Cross-module __doc__") + _check_doc_adaptive("remote.remote_func", remote.remote_func.__doc__, "Remote function docstring.") + _check_doc_adaptive("remote.RemoteClass", remote.RemoteClass.__doc__, "Remote class docstring.") + _check_doc_adaptive("remote.RemoteClass.remote_method", remote.RemoteClass.remote_method.__doc__, "Remote method docstring.") + _check_doc_null("remote.no_doc_func", remote.no_doc_func.__doc__) + + +def main() -> t.CInt: + testcheck.begin("DocTest: __doc__ magic string test") + test_local_doc() + test_cross_module_doc() + return testcheck.end() diff --git a/Test/DocTest/App/remote.py b/Test/DocTest/App/remote.py new file mode 100644 index 0000000..cd08daf --- /dev/null +++ b/Test/DocTest/App/remote.py @@ -0,0 +1,20 @@ +import t + + +def remote_func(): + "Remote function docstring." + return 0 + + +@t.Object +class RemoteClass: + "Remote class docstring." + value: t.CInt + + def remote_method(self): + "Remote method docstring." + return 0 + + +def no_doc_func(): + return 0 diff --git a/Test/DocTest/output/14a9b576ab5d21c9.deps.json b/Test/DocTest/output/14a9b576ab5d21c9.deps.json new file mode 100644 index 0000000..8c41c52 --- /dev/null +++ b/Test/DocTest/output/14a9b576ab5d21c9.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "remote": "f26d7266c4f8762b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/DocTest/output/f26d7266c4f8762b.deps.json b/Test/DocTest/output/f26d7266c4f8762b.deps.json new file mode 100644 index 0000000..bb4c88a --- /dev/null +++ b/Test/DocTest/output/f26d7266c4f8762b.deps.json @@ -0,0 +1 @@ +{"main": "14a9b576ab5d21c9", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/DocTest/output_opt/6d751fba8cdbc5d1.deps.json b/Test/DocTest/output_opt/6d751fba8cdbc5d1.deps.json new file mode 100644 index 0000000..a5eef39 --- /dev/null +++ b/Test/DocTest/output_opt/6d751fba8cdbc5d1.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "5a6a2137958c28c5", "win32base": "5a6a2137958c28c5", "stdio": "73edbcf76e32d00b", "string": "b5f1dc665c52a1bd", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "remote": "f26d7266c4f8762b", "stdint": "f5522571bcce7bcb", "testcheck": "fee660d8ff1a993b"} \ No newline at end of file diff --git a/Test/DocTest/output_opt/b5f1dc665c52a1bd.deps.json b/Test/DocTest/output_opt/b5f1dc665c52a1bd.deps.json new file mode 100644 index 0000000..cec30c5 --- /dev/null +++ b/Test/DocTest/output_opt/b5f1dc665c52a1bd.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "5a6a2137958c28c5", "win32base": "5a6a2137958c28c5", "main": "6d751fba8cdbc5d1", "stdio": "73edbcf76e32d00b", "string": "b5f1dc665c52a1bd", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "remote": "f26d7266c4f8762b", "stdint": "f5522571bcce7bcb", "testcheck": "fee660d8ff1a993b"} \ No newline at end of file diff --git a/Test/DocTest/output_opt/f26d7266c4f8762b.deps.json b/Test/DocTest/output_opt/f26d7266c4f8762b.deps.json new file mode 100644 index 0000000..684338c --- /dev/null +++ b/Test/DocTest/output_opt/f26d7266c4f8762b.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "5a6a2137958c28c5", "win32base": "5a6a2137958c28c5", "main": "6d751fba8cdbc5d1", "stdio": "73edbcf76e32d00b", "string": "b5f1dc665c52a1bd", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "testcheck": "fee660d8ff1a993b"} \ No newline at end of file diff --git a/Test/DocTest/output_opt/fee660d8ff1a993b.deps.json b/Test/DocTest/output_opt/fee660d8ff1a993b.deps.json new file mode 100644 index 0000000..cec30c5 --- /dev/null +++ b/Test/DocTest/output_opt/fee660d8ff1a993b.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "5a6a2137958c28c5", "win32base": "5a6a2137958c28c5", "main": "6d751fba8cdbc5d1", "stdio": "73edbcf76e32d00b", "string": "b5f1dc665c52a1bd", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "remote": "f26d7266c4f8762b", "stdint": "f5522571bcce7bcb", "testcheck": "fee660d8ff1a993b"} \ No newline at end of file diff --git a/Test/DocTest/project.json b/Test/DocTest/project.json new file mode 100644 index 0000000..6668cf4 --- /dev/null +++ b/Test/DocTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "DocTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "DocTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 0, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/DocTest/project_opt.json b/Test/DocTest/project_opt.json new file mode 100644 index 0000000..9680858 --- /dev/null +++ b/Test/DocTest/project_opt.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "DocTestOpt", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp_opt", + "output_dir": "./output_opt", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "DocTestOpt.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/DocTest/temp/14a9b576ab5d21c9.doc.json b/Test/DocTest/temp/14a9b576ab5d21c9.doc.json new file mode 100644 index 0000000..97ecb08 --- /dev/null +++ b/Test/DocTest/temp/14a9b576ab5d21c9.doc.json @@ -0,0 +1 @@ +{"local_func": "Local function docstring.", "LocalClass": "Local class docstring.", "LocalClass.local_method": "Local method docstring.", "_check_doc_adaptive": "自适应检查 docstring：非空时验证匹配，空时视为优化剥离", "_check_doc_null": "检查 docstring 为 null（无 docstring 的符号）"} \ No newline at end of file diff --git a/Test/DocTest/temp/14a9b576ab5d21c9.pyi b/Test/DocTest/temp/14a9b576ab5d21c9.pyi new file mode 100644 index 0000000..c16365f --- /dev/null +++ b/Test/DocTest/temp/14a9b576ab5d21c9.pyi @@ -0,0 +1,32 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import c +import testcheck +import remote + +def local_func() -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass + + +@t.Object +class LocalClass: + value: t.CInt + def local_method(self: LocalClass) -> t.CInt: pass + +def _check_doc_adaptive(name: str, doc: str, expected: str) -> t.CInt: pass + +def _check_doc_null(name: str, doc: str) -> t.CInt: pass + +def test_local_doc() -> t.CInt: pass + +def test_cross_module_doc() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/DocTest/temp/6f62fe05c5ea1ceb.pyi b/Test/DocTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/DocTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/DocTest/temp/7e529fe7a078cfef.pyi b/Test/DocTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/DocTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/DocTest/temp/_phase1_manifest.json b/Test/DocTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..4de65ec --- /dev/null +++ b/Test/DocTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\DocTest\\App\\main.py": {"sha1": "14a9b576ab5d21c9", "mtime": 1782474568.0454285, "size": 2162}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\DocTest\\App\\remote.py": {"sha1": "f26d7266c4f8762b", "mtime": 1782397989.7183921, "size": 271}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/DocTest/temp/_sha1_map.txt b/Test/DocTest/temp/_sha1_map.txt new file mode 100644 index 0000000..6779449 --- /dev/null +++ b/Test/DocTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +14a9b576ab5d21c9:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f26d7266c4f8762b:remote.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/DocTest/temp/_shared_sym.json b/Test/DocTest/temp/_shared_sym.json new file mode 100644 index 0000000..fd3e3b7 Binary files /dev/null and b/Test/DocTest/temp/_shared_sym.json differ diff --git a/Test/DocTest/temp/ab6e54ba9a669f76.pyi b/Test/DocTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/DocTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/DocTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/DocTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/DocTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DocTest/temp/dd3002730623424b.pyi b/Test/DocTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/DocTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/DocTest/temp/f26d7266c4f8762b.doc.json b/Test/DocTest/temp/f26d7266c4f8762b.doc.json new file mode 100644 index 0000000..7565271 --- /dev/null +++ b/Test/DocTest/temp/f26d7266c4f8762b.doc.json @@ -0,0 +1 @@ +{"remote_func": "Remote function docstring.", "RemoteClass": "Remote class docstring.", "RemoteClass.remote_method": "Remote method docstring."} \ No newline at end of file diff --git a/Test/DocTest/temp/f26d7266c4f8762b.pyi b/Test/DocTest/temp/f26d7266c4f8762b.pyi new file mode 100644 index 0000000..845daee --- /dev/null +++ b/Test/DocTest/temp/f26d7266c4f8762b.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from remote.py +Module: remote +""" + +import c + + +import t + +def remote_func() -> t.CInt: pass + + +@t.Object +class RemoteClass: + value: t.CInt + def remote_method(self: RemoteClass) -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass diff --git a/Test/DocTest/temp/f5522571bcce7bcb.pyi b/Test/DocTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/DocTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DocTest/temp_opt/5a6a2137958c28c5.pyi b/Test/DocTest/temp_opt/5a6a2137958c28c5.pyi new file mode 100644 index 0000000..a137977 --- /dev/null +++ b/Test/DocTest/temp_opt/5a6a2137958c28c5.pyi @@ -0,0 +1,98 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass diff --git a/Test/DocTest/temp_opt/6d751fba8cdbc5d1.doc.json b/Test/DocTest/temp_opt/6d751fba8cdbc5d1.doc.json new file mode 100644 index 0000000..97ecb08 --- /dev/null +++ b/Test/DocTest/temp_opt/6d751fba8cdbc5d1.doc.json @@ -0,0 +1 @@ +{"local_func": "Local function docstring.", "LocalClass": "Local class docstring.", "LocalClass.local_method": "Local method docstring.", "_check_doc_adaptive": "自适应检查 docstring：非空时验证匹配，空时视为优化剥离", "_check_doc_null": "检查 docstring 为 null（无 docstring 的符号）"} \ No newline at end of file diff --git a/Test/DocTest/temp_opt/6d751fba8cdbc5d1.pyi b/Test/DocTest/temp_opt/6d751fba8cdbc5d1.pyi new file mode 100644 index 0000000..c16365f --- /dev/null +++ b/Test/DocTest/temp_opt/6d751fba8cdbc5d1.pyi @@ -0,0 +1,32 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import c +import testcheck +import remote + +def local_func() -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass + + +@t.Object +class LocalClass: + value: t.CInt + def local_method(self: LocalClass) -> t.CInt: pass + +def _check_doc_adaptive(name: str, doc: str, expected: str) -> t.CInt: pass + +def _check_doc_null(name: str, doc: str) -> t.CInt: pass + +def test_local_doc() -> t.CInt: pass + +def test_cross_module_doc() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/DocTest/temp_opt/73edbcf76e32d00b.pyi b/Test/DocTest/temp_opt/73edbcf76e32d00b.pyi new file mode 100644 index 0000000..ef712e3 --- /dev/null +++ b/Test/DocTest/temp_opt/73edbcf76e32d00b.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def fprintf(stream: t.CVoid | t.CPtr, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def sprintf(buf: t.CChar | t.CPtr, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def puts(s: t.CChar | t.CConst | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def fputs(s: t.CChar | t.CConst | t.CPtr, stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + + +stdin: t.CExtern | t.CVoid | t.CPtr +stdout: t.CExtern | t.CVoid | t.CPtr +stderr: t.CExtern | t.CVoid | t.CPtr \ No newline at end of file diff --git a/Test/DocTest/temp_opt/_sha1_map.txt b/Test/DocTest/temp_opt/_sha1_map.txt new file mode 100644 index 0000000..641a39f --- /dev/null +++ b/Test/DocTest/temp_opt/_sha1_map.txt @@ -0,0 +1,8 @@ +5a6a2137958c28c5:includes/w32\win32base.py +6d751fba8cdbc5d1:main.py +73edbcf76e32d00b:includes/stdio.py +b5f1dc665c52a1bd:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +f26d7266c4f8762b:remote.py +f5522571bcce7bcb:includes/stdint.py +fee660d8ff1a993b:includes/testcheck.py diff --git a/Test/DocTest/temp_opt/_shared_sym.json b/Test/DocTest/temp_opt/_shared_sym.json new file mode 100644 index 0000000..e69de29 diff --git a/Test/DocTest/temp_opt/b5f1dc665c52a1bd.pyi b/Test/DocTest/temp_opt/b5f1dc665c52a1bd.pyi new file mode 100644 index 0000000..8b6959e --- /dev/null +++ b/Test/DocTest/temp_opt/b5f1dc665c52a1bd.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/DocTest/temp_opt/bbdf3bbd4c3bc28c.pyi b/Test/DocTest/temp_opt/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/DocTest/temp_opt/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/DocTest/temp_opt/f26d7266c4f8762b.doc.json b/Test/DocTest/temp_opt/f26d7266c4f8762b.doc.json new file mode 100644 index 0000000..7565271 --- /dev/null +++ b/Test/DocTest/temp_opt/f26d7266c4f8762b.doc.json @@ -0,0 +1 @@ +{"remote_func": "Remote function docstring.", "RemoteClass": "Remote class docstring.", "RemoteClass.remote_method": "Remote method docstring."} \ No newline at end of file diff --git a/Test/DocTest/temp_opt/f26d7266c4f8762b.pyi b/Test/DocTest/temp_opt/f26d7266c4f8762b.pyi new file mode 100644 index 0000000..845daee --- /dev/null +++ b/Test/DocTest/temp_opt/f26d7266c4f8762b.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from remote.py +Module: remote +""" + +import c + + +import t + +def remote_func() -> t.CInt: pass + + +@t.Object +class RemoteClass: + value: t.CInt + def remote_method(self: RemoteClass) -> t.CInt: pass + +def no_doc_func() -> t.CInt: pass diff --git a/Test/DocTest/temp_opt/f5522571bcce7bcb.pyi b/Test/DocTest/temp_opt/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/DocTest/temp_opt/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/DocTest/temp_opt/fee660d8ff1a993b.pyi b/Test/DocTest/temp_opt/fee660d8ff1a993b.pyi new file mode 100644 index 0000000..a3fe167 --- /dev/null +++ b/Test/DocTest/temp_opt/fee660d8ff1a993b.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass diff --git a/Test/EdgeCaseTest/App/main.py b/Test/EdgeCaseTest/App/main.py new file mode 100644 index 0000000..125e21a --- /dev/null +++ b/Test/EdgeCaseTest/App/main.py @@ -0,0 +1,260 @@ +import t +import t as tt +import stdio +import testcheck + +from t import CEnum as en +from t import CUnion as un +from t import REnum as ren +from t import CStruct as cs +from t import CInt as myint +from t import CChar, CPtr + + +# ============================================================ +# Test 1: from t import CEnum as en; class X(en) +# ============================================================ +class Color(en): + Red = 0 + Green = 1 + Blue = 2 + + +# ============================================================ +# Test 2: import t as tt; class X(tt.CEnum) +# ============================================================ +class Status(tt.CEnum): + Ok = 0 + Error = 1 + + +# ============================================================ +# Test 3: from t import CUnion as un; class X(un) +# ============================================================ +class Data(un): + i_val: t.CInt + f_val: t.CFloat + + +# ============================================================ +# Test 4: from t import REnum as ren; class X(ren) +# ============================================================ +class Opt(ren): + class Some: + val: myint + None_ = 0 + + +# ============================================================ +# Test 5: 前向引用检测 — _class_graph 正确识别父类 +# （注：检测已修复，但成员继承仍需父类先处理） +# ============================================================ +class Parent: + a: t.CInt + + def get_a(self) -> t.CInt: + return self.a + + +class Child(Parent): + b: t.CInt + + def get_sum(self) -> t.CInt: + return self.a + self.b + + +# ============================================================ +# Test 6: 多层继承 A -> B -> C，CVTable 向上传播 +# ============================================================ +class Base: + x: t.CInt + + def base_val(self) -> t.CInt: + return self.x + + +class Mid(Base): + y: t.CInt + + def mid_val(self) -> t.CInt: + return self.y + + +class Leaf(Mid): + z: t.CInt + + def leaf_val(self) -> t.CInt: + return self.z + + +# ============================================================ +# Test 7: 类型别名用于注解 +# ============================================================ +val: myint = 42 + + +# ============================================================ +# Test 8: BigEndian 字节序 +# ============================================================ +class NetHeader: + magic: t.CInt | t.BigEndian + length: t.CShort | t.BigEndian + normal: t.CInt + + +# ============================================================ +# Test 9: import t as tt; tt.CInt 作为类型注解 +# ============================================================ +# class TtStruct: +# x: tt.CInt +# y: tt.CFloat + + +# ============================================================ +# Test 10: import t as tt; class X(tt.CStruct) +# ============================================================ +class TtPoint(tt.CStruct): + x: t.CInt + y: t.CInt + + +# ============================================================ +# Test 11: import t as tt; @tt.Object 装饰器 +# ============================================================ +@tt.Object +class TtObject: + val: t.CInt + + def get_val(self) -> t.CInt: + return self.val + + +# ============================================================ +# Test 12: from t import CStruct as cs; class X(cs) +# ============================================================ +class CsPoint(cs): + x: t.CInt + y: t.CInt + + +# ============================================================ +# Test 13: import t as tt; class X(tt.CUnion) +# ============================================================ +class TtUnion(tt.CUnion): + a: t.CInt + b: t.CFloat + + +# ============================================================ +# Test 14: @tt.Object on class without methods (should fail?) +# ============================================================ +@tt.Object +class TtEmptyObject: + val: t.CInt + + +def main() -> t.CInt: + testcheck.begin("EdgeCaseTest: 类型别名与边界情况测试") + + # Test 1: CEnum alias + testcheck.section("Test 1: CEnum alias") + c: t.CInt = Color.Green + stdio.printf("Test1 CEnum alias: %d\n", c) + testcheck.ok("CEnum alias (Color.Green=1)") + + # Test 2: module alias + CEnum + testcheck.section("Test 2: module alias + CEnum") + s: t.CInt = Status.Error + stdio.printf("Test2 t as tt.CEnum: %d\n", s) + testcheck.ok("t as tt.CEnum (Status.Error=1)") + + # Test 3: CUnion alias + testcheck.section("Test 3: CUnion alias") + d: Data = Data() + d.i_val = 10 + d.f_val = 3.14 + stdio.printf("Test3 CUnion alias: i=%d f=%f\n", d.i_val, d.f_val) + testcheck.ok("CUnion alias (i=10 f=3.14)") + + # Test 4: REnum alias + testcheck.section("Test 4: REnum alias") + o: Opt = Opt.Some(99) + match o: + case Opt.Some(v): + stdio.printf("Test4 REnum alias: val=%d\n", v) + case Opt.None_(): + stdio.printf("Test4 REnum alias: None\n") + testcheck.ok("REnum alias (Opt.Some(99))") + + # Test 5: forward reference + testcheck.section("Test 5: forward reference") + ch: Child = Child() + ch.a = 100 + ch.b = 200 + stdio.printf("Test5 forward ref: a=%d b=%d sum=%d\n", ch.a, ch.b, ch.get_sum()) + stdio.printf("Test5 parent method: %d\n", ch.get_a()) + testcheck.ok("forward ref (a=100 b=200 sum=300)") + + # Test 6: multi-level inheritance + testcheck.section("Test 6: multi-level inheritance") + lf: Leaf = Leaf() + lf.x = 1 + lf.y = 2 + lf.z = 3 + stdio.printf("Test6 multi-inherit: base=%d mid=%d leaf=%d\n", + lf.base_val(), lf.mid_val(), lf.leaf_val()) + testcheck.ok("multi-inherit (base=1 mid=2 leaf=3)") + + # Test 7: type alias in annotation + testcheck.section("Test 7: type alias in annotation") + stdio.printf("Test7 type alias: val=%d\n", val) + testcheck.ok("type alias (val=42)") + + # Test 8: BigEndian + testcheck.section("Test 8: BigEndian") + nh: NetHeader = NetHeader() + nh.magic = 0x12345678 + nh.length = 256 + nh.normal = 42 + stdio.printf("Test8 BigEndian: magic=0x%X length=%d normal=%d\n", + nh.magic, nh.length, nh.normal) + testcheck.ok("BigEndian (magic=0x12345678 length=256 normal=42)") + + # Test 10: tt.CStruct as base class + testcheck.section("Test 10: tt.CStruct as base class") + tp: TtPoint = TtPoint() + tp.x = 100 + tp.y = 200 + stdio.printf("Test10 tt.CStruct: x=%d y=%d\n", tp.x, tp.y) + testcheck.ok("tt.CStruct (x=100 y=200)") + + # Test 11: @tt.Object decorator + testcheck.section("Test 11: @tt.Object decorator") + to: TtObject = TtObject() + to.val = 999 + stdio.printf("Test11 @tt.Object: val=%d\n", to.get_val()) + testcheck.ok("@tt.Object (val=999)") + + # Test 12: from t import CStruct as cs + testcheck.section("Test 12: from t import CStruct as cs") + cp: CsPoint = CsPoint() + cp.x = 11 + cp.y = 22 + stdio.printf("Test12 cs alias: x=%d y=%d\n", cp.x, cp.y) + testcheck.ok("cs alias (x=11 y=22)") + + # Test 13: tt.CUnion + testcheck.section("Test 13: tt.CUnion") + tu: TtUnion = TtUnion() + tu.a = 77 + stdio.printf("Test13 tt.CUnion: a=%d\n", tu.a) + testcheck.ok("tt.CUnion (a=77)") + + # Test 14: @tt.Object on empty class + testcheck.section("Test 14: @tt.Object on empty class") + teo: TtEmptyObject = TtEmptyObject() + teo.val = 555 + stdio.printf("Test14 @tt.Object empty: val=%d\n", teo.val) + testcheck.ok("@tt.Object empty (val=555)") + + return testcheck.end() diff --git a/Test/EdgeCaseTest/output/ac0d686cb2cc783b.deps.json b/Test/EdgeCaseTest/output/ac0d686cb2cc783b.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/EdgeCaseTest/output/ac0d686cb2cc783b.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/EdgeCaseTest/project.json b/Test/EdgeCaseTest/project.json new file mode 100644 index 0000000..c54b6f3 --- /dev/null +++ b/Test/EdgeCaseTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "EdgeCaseTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "EdgeCaseTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/EdgeCaseTest/temp/6f62fe05c5ea1ceb.pyi b/Test/EdgeCaseTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/EdgeCaseTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/EdgeCaseTest/temp/7e529fe7a078cfef.pyi b/Test/EdgeCaseTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/EdgeCaseTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/EdgeCaseTest/temp/_phase1_manifest.json b/Test/EdgeCaseTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..6ba8e32 --- /dev/null +++ b/Test/EdgeCaseTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\EdgeCaseTest\\App\\main.py": {"sha1": "ac0d686cb2cc783b", "mtime": 1782110086.5178304, "size": 7487}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/EdgeCaseTest/temp/_sha1_map.txt b/Test/EdgeCaseTest/temp/_sha1_map.txt new file mode 100644 index 0000000..d770431 --- /dev/null +++ b/Test/EdgeCaseTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +ac0d686cb2cc783b:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.doc.json b/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.pyi b/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.pyi new file mode 100644 index 0000000..7b2670f --- /dev/null +++ b/Test/EdgeCaseTest/temp/ac0d686cb2cc783b.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import t as tt +import stdio +import testcheck +from t import CEnum as en +from t import CUnion as un +from t import REnum as ren +from t import CStruct as cs +from t import CInt as myint +from t import CChar, CPtr + +class Color(en): + Red = 0 + Green = 1 + Blue = 2 +class Status(tt.CEnum): + Ok = 0 + Error = 1 +class Data(un): + i_val: t.CInt + f_val: t.CFloat +class Opt(ren): + class Some: + val: myint + None_ = 0 +class Parent: + a: t.CInt + def get_a(self: Parent) -> t.CInt: pass +class Child(Parent): + b: t.CInt + def get_sum(self: Child) -> t.CInt: pass +class Base: + x: t.CInt + def base_val(self: Base) -> t.CInt: pass +class Mid(Base): + y: t.CInt + def mid_val(self: Mid) -> t.CInt: pass +class Leaf(Mid): + z: t.CInt + def leaf_val(self: Leaf) -> t.CInt: pass + +val: t.CExtern | myint + +class NetHeader: + magic: t.CInt | t.BigEndian + length: t.CShort | t.BigEndian + normal: t.CInt +class TtPoint(tt.CStruct): + x: t.CInt + y: t.CInt +@tt.Object +class TtObject: + val: t.CInt + def get_val(self: TtObject) -> t.CInt: pass +class CsPoint(cs): + x: t.CInt + y: t.CInt +class TtUnion(tt.CUnion): + a: t.CInt + b: t.CFloat +@tt.Object +class TtEmptyObject: + val: t.CInt + +def main() -> t.CInt: pass diff --git a/Test/EdgeCaseTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/EdgeCaseTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/EdgeCaseTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/EdgeCaseTest/temp/dd3002730623424b.pyi b/Test/EdgeCaseTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/EdgeCaseTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/EdgeCaseTest/temp/f5522571bcce7bcb.pyi b/Test/EdgeCaseTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/EdgeCaseTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/EqTest/App/main.py b/Test/EqTest/App/main.py new file mode 100644 index 0000000..ba3db77 --- /dev/null +++ b/Test/EqTest/App/main.py @@ -0,0 +1,56 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("== 重载和 is 地址比较测试") + + # === Test 1: str == str 值比较 === + testcheck.section("Test 1: str == str 值比较") + s1: str = "hello" + s2: str = "hello" + s3: str = "world" + testcheck.check(s1 == s2, "str == str (same value) OK", "str == str (same value) FAILED") + testcheck.check(not (s1 == s3), "str != str (diff value) OK", "str != str (diff value) FAILED") + + # === Test 2: str != str === + testcheck.section("Test 2: str != str") + testcheck.check(s1 != s3, "str != str (diff) OK", "str != str (diff) FAILED") + testcheck.check(not (s1 != s2), "str == str (same) via != OK", "str == str (same) via != FAILED") + + # === Test 3: str == 字面量 === + testcheck.section("Test 3: str == literal") + testcheck.check(s1 == "hello", "str == literal OK", "str == literal FAILED") + testcheck.check(not (s1 == "world"), "str != literal OK", "str != literal FAILED") + + # === Test 4: str is str 地址比较 === + testcheck.section("Test 4: str is str address compare") + # s1 和 s2 是不同的字符串变量，地址不同 + testcheck.check(not (s1 is s2), "str is str (diff addr) OK", "str is str (diff addr) FAILED") + # s1 is s1 应该为 True（同一变量） + testcheck.check(s1 is s1, "str is self OK", "str is self FAILED") + + # === Test 5: int == int === + testcheck.section("Test 5: int == int") + i1: int = 42 + i2: int = 42 + i3: int = 99 + testcheck.check(i1 == i2, "int == int (same) OK", "int == int (same) FAILED") + testcheck.check(not (i1 == i3), "int != int (diff) OK", "int != int (diff) FAILED") + + # === Test 6: None 比较 === + testcheck.section("Test 6: None compare") + null_ptr: str = None + testcheck.check(null_ptr == None, "ptr == None OK", "ptr == None FAILED") + testcheck.check(not (s1 == None), "str != None OK", "str != None FAILED") + + return testcheck.end() diff --git a/Test/EqTest/output/78943d0dbc89a29a.deps.json b/Test/EqTest/output/78943d0dbc89a29a.deps.json new file mode 100644 index 0000000..262dabd --- /dev/null +++ b/Test/EqTest/output/78943d0dbc89a29a.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/EqTest/project.json b/Test/EqTest/project.json new file mode 100644 index 0000000..9a71251 --- /dev/null +++ b/Test/EqTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "EqTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "EqTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/EqTest/temp/0035c95a18d4f8e8.pyi b/Test/EqTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/EqTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/EqTest/temp/067c78e9f121dce3.pyi b/Test/EqTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/EqTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/EqTest/temp/06f53cc594b4ac6c.pyi b/Test/EqTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/EqTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/EqTest/temp/6446627d4f07a1b5.pyi b/Test/EqTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/EqTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/EqTest/temp/6f62fe05c5ea1ceb.pyi b/Test/EqTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/EqTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/EqTest/temp/72e2d5ccb7cedcf1.pyi b/Test/EqTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/EqTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/EqTest/temp/78943d0dbc89a29a.doc.json b/Test/EqTest/temp/78943d0dbc89a29a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/EqTest/temp/78943d0dbc89a29a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/EqTest/temp/78943d0dbc89a29a.pyi b/Test/EqTest/temp/78943d0dbc89a29a.pyi new file mode 100644 index 0000000..ce1d065 --- /dev/null +++ b/Test/EqTest/temp/78943d0dbc89a29a.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + +def main() -> CInt | CExport: pass diff --git a/Test/EqTest/temp/7e529fe7a078cfef.pyi b/Test/EqTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/EqTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/EqTest/temp/90c53dd6db8d41cf.pyi b/Test/EqTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/EqTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/EqTest/temp/_phase1_manifest.json b/Test/EqTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..0a06c09 --- /dev/null +++ b/Test/EqTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\EqTest\\App\\main.py": {"sha1": "78943d0dbc89a29a", "mtime": 1782354684.7481763, "size": 2169}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/EqTest/temp/_sha1_map.txt b/Test/EqTest/temp/_sha1_map.txt new file mode 100644 index 0000000..7245b3a --- /dev/null +++ b/Test/EqTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +6f62fe05c5ea1ceb:includes/stdio.py +78943d0dbc89a29a:main.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/EqTest/temp/ab6e54ba9a669f76.pyi b/Test/EqTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/EqTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/EqTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/EqTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/EqTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/EqTest/temp/dd3002730623424b.pyi b/Test/EqTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/EqTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/EqTest/temp/f5522571bcce7bcb.pyi b/Test/EqTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/EqTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/EqTest/temp/f6b51804a0ba8ff0.pyi b/Test/EqTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/EqTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/FakeDuckTest/App/main.py b/Test/FakeDuckTest/App/main.py new file mode 100644 index 0000000..6dc24dc --- /dev/null +++ b/Test/FakeDuckTest/App/main.py @@ -0,0 +1,308 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import _fakeduck +import stdio +import stdlib +import string +import memhub +import testcheck +import _withcontent + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + # FakeDuck 测试: str 类型调用扩展方法 b() + a: str = "Hello" + + # Type 通用基类测试: str 变量调用 Type 的方法 type_id() + tid: CInt = a.type_id() + + if tid == 99: + stdio.printf("Type OK: a.type_id() = %d\n", tid) + else: + stdio.printf("Type FAIL: a.type_id() = %d\n", tid) + + # 直接字段访问测试: 读取 a.__data__ (应等于 a 本身) + data_ptr: str = a.__data__ + stdio.printf("a.__data__ = %s\n", data_ptr) + + # upper() 测试: 通过 mbuddy 分配内存，返回大写字符串 + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + + # 直接字段赋值: a.__mbuddy__ = bd (语法糖, 等价于 a.set_mbuddy(bd)) + a.__mbuddy__ = bd + upper_str: str = a.upper() + stdio.printf("upper('Hello') = %s\n", upper_str) + + # 验证 per-instance 不污染: b 没有设置 __mbuddy__ (初始为 NULL) + b: str = "World" + + # 直接字段读取: b.__mbuddy__ (应返回 NULL) + b_buddy: memhub.MemBuddy | t.CPtr = b.__mbuddy__ + if t.CInt(b_buddy) == 0: + stdio.printf("mbuddy OK: b.__mbuddy__ is NULL (no pollution)\n") + else: + stdio.printf("mbuddy FAIL: b.__mbuddy__ polluted!\n") + + # b 也设置 mbuddy 并 upper + b.__mbuddy__ = bd + upper_b: str = b.upper() + stdio.printf("upper('World') = %s\n", upper_b) + + # lower() 测试 + lower_str: str = a.lower() + stdio.printf("lower('Hello') = %s\n", lower_str) + + # capitalize() 测试 + cap_str: str = a.capitalize() + stdio.printf("capitalize('Hello') = %s\n", cap_str) + + # swapcase() 测试 + swap_str: str = a.swapcase() + stdio.printf("swapcase('Hello') = %s\n", swap_str) + + # casefold() 测试 + cf_str: str = a.casefold() + stdio.printf("casefold('Hello') = %s\n", cf_str) + + # title() 测试 + t_str: str = "hello world" + t_str.__mbuddy__ = bd + title_result: str = t_str.title() + stdio.printf("title('hello world') = %s\n", title_result) + + # 布尔判断函数测试 (不需要 __mbuddy__) + u: str = "HELLO" + lo: str = "hello" + dig: str = "12345" + alnum_s: str = "Hello123" + space_s: str = " " + title_s: str = "Hello World" + punct_s: str = "Hello!" + + stdio.printf("isupper('HELLO') = %d\n", u.isupper()) + stdio.printf("isupper('Hello') = %d\n", a.isupper()) + stdio.printf("islower('hello') = %d\n", lo.islower()) + stdio.printf("islower('Hello') = %d\n", a.islower()) + stdio.printf("isalpha('Hello') = %d\n", a.isalpha()) + stdio.printf("isalpha('Hello123') = %d\n", alnum_s.isalpha()) + stdio.printf("isdigit('12345') = %d\n", dig.isdigit()) + stdio.printf("isdigit('Hello') = %d\n", a.isdigit()) + stdio.printf("isalnum('Hello123') = %d\n", alnum_s.isalnum()) + stdio.printf("isalnum('Hello!') = %d\n", punct_s.isalnum()) + stdio.printf("isspace(' ') = %d\n", space_s.isspace()) + stdio.printf("isspace('Hello') = %d\n", a.isspace()) + stdio.printf("isascii('Hello') = %d\n", a.isascii()) + stdio.printf("istitle('Hello World') = %d\n", title_s.istitle()) + stdio.printf("istitle('hello world') = %d\n", t_str.istitle()) + + # === split() 测试 === + testcheck.begin("FakeDuck split() 测试") + + # Test 1: 基本分割 + testcheck.section("Test 1: 基本分割 'a,b,c'") + s1: str = "a,b,c" + s1.__mbuddy__ = bd + parts = s1.split(',') + n1: t.CSizeT = parts.__len__() + stdio.printf("split('a,b,c', ',') len=%lu\n", n1) + testcheck.check(n1 == 3, "split len OK (3)", "split len FAILED expect 3") + p0: str = parts[0] + p1: str = parts[1] + p2: str = parts[2] + stdio.printf("parts[0]='%s' [1]='%s' [2]='%s'\n", p0, p1, p2) + testcheck.check(p0 == "a", "split[0]=='a' OK", "split[0] FAILED") + testcheck.check(p1 == "b", "split[1]=='b' OK", "split[1] FAILED") + testcheck.check(p2 == "c", "split[2]=='c' OK", "split[2] FAILED") + + # Test 2: 空字符串分割 + testcheck.section("Test 2: 无分隔符 'hello'") + s2: str = "hello" + s2.__mbuddy__ = bd + parts2 = s2.split(',') + n2: t.CSizeT = parts2.__len__() + stdio.printf("split('hello', ',') len=%lu\n", n2) + testcheck.check(n2 == 1, "no-delim len OK (1)", "no-delim len FAILED expect 1") + h0: str = parts2.get(0) + testcheck.check(h0 == "hello", "no-delim[0]=='hello' OK", "no-delim[0] FAILED") + + # Test 3: 连续分隔符 + testcheck.section("Test 3: 连续分隔符 'a,,b'") + s3: str = "a,,b" + s3.__mbuddy__ = bd + parts3 = s3.split(',') + n3: t.CSizeT = parts3.__len__() + stdio.printf("split('a,,b', ',') len=%lu\n", n3) + testcheck.check(n3 == 3, "consecutive len OK (3)", "consecutive len FAILED expect 3") + e0: str = parts3.get(0) + e1: str = parts3.get(1) + e2: str = parts3.get(2) + stdio.printf("parts[0]='%s' [1]='%s' [2]='%s'\n", e0, e1, e2) + testcheck.check(e0 == "a", "consec[0]=='a' OK", "consec[0] FAILED") + testcheck.check(e1 == "", "consec[1]=='' OK", "consec[1] FAILED") + testcheck.check(e2 == "b", "consec[2]=='b' OK", "consec[2] FAILED") + + # Test 4: 空字符串 + testcheck.section("Test 4: 空字符串 ''") + s4: str = "" + s4.__mbuddy__ = bd + parts4 = s4.split(',') + n4: t.CSizeT = parts4.__len__() + stdio.printf("split('', ',') len=%lu\n", n4) + testcheck.check(n4 == 1, "empty len OK (1)", "empty len FAILED expect 1") + + # Test 5: 空格分割句子 + testcheck.section("Test 5: 空格分割 'hello world foo'") + s5: str = "hello world foo" + s5.__mbuddy__ = bd + parts5 = s5.split(' ') + n5: t.CSizeT = parts5.__len__() + stdio.printf("split('hello world foo', ' ') len=%lu\n", n5) + testcheck.check(n5 == 3, "space len OK (3)", "space len FAILED expect 3") + w0: str = parts5[0] + w1: str = parts5[1] + w2: str = parts5[2] + stdio.printf("words[0]='%s' [1]='%s' [2]='%s'\n", w0, w1, w2) + testcheck.check(w0 == "hello", "space[0]=='hello' OK", "space[0] FAILED") + testcheck.check(w1 == "world", "space[1]=='world' OK", "space[1] FAILED") + testcheck.check(w2 == "foo", "space[2]=='foo' OK", "space[2] FAILED") + + # Test 6: 多字符分隔符 '::' + testcheck.section("Test 6: 多字符分隔符 '::' on 'a::b::c'") + s6: str = "a::b::c" + s6.__mbuddy__ = bd + parts6 = s6.split("::") + n6: t.CSizeT = parts6.__len__() + stdio.printf("split('a::b::c', '::') len=%lu\n", n6) + testcheck.check(n6 == 3, "multi-colon len OK (3)", "multi-colon len FAILED expect 3") + m0: str = parts6[0] + m1: str = parts6[1] + m2: str = parts6[2] + stdio.printf("parts[0]='%s' [1]='%s' [2]='%s'\n", m0, m1, m2) + testcheck.check(m0 == "a", "multi-colon[0]=='a' OK", "multi-colon[0] FAILED") + testcheck.check(m1 == "b", "multi-colon[1]=='b' OK", "multi-colon[1] FAILED") + testcheck.check(m2 == "c", "multi-colon[2]=='c' OK", "multi-colon[2] FAILED") + + # Test 7: 多字符分隔符 ', ' (逗号+空格) + testcheck.section("Test 7: 多字符分隔符 ', ' on 'hello, world, foo'") + s7: str = "hello, world, foo" + s7.__mbuddy__ = bd + parts7 = s7.split(", ") + n7: t.CSizeT = parts7.__len__() + stdio.printf("split('hello, world, foo', ', ') len=%lu\n", n7) + testcheck.check(n7 == 3, "comma-space len OK (3)", "comma-space len FAILED expect 3") + cs0: str = parts7[0] + cs1: str = parts7[1] + cs2: str = parts7[2] + stdio.printf("parts[0]='%s' [1]='%s' [2]='%s'\n", cs0, cs1, cs2) + testcheck.check(cs0 == "hello", "comma-space[0]=='hello' OK", "comma-space[0] FAILED") + testcheck.check(cs1 == "world", "comma-space[1]=='world' OK", "comma-space[1] FAILED") + testcheck.check(cs2 == "foo", "comma-space[2]=='foo' OK", "comma-space[2] FAILED") + + # Test 8: 多字符分隔符在首尾 + testcheck.section("Test 8: 多字符分隔符首尾 '::a::b::'") + s8: str = "::a::b::" + s8.__mbuddy__ = bd + parts8 = s8.split("::") + n8: t.CSizeT = parts8.__len__() + stdio.printf("split('::a::b::', '::') len=%lu\n", n8) + testcheck.check(n8 == 4, "edge len OK (4)", "edge len FAILED expect 4") + ed0: str = parts8[0] + ed1: str = parts8[1] + ed2: str = parts8[2] + ed3: str = parts8[3] + stdio.printf("parts[0]='%s' [1]='%s' [2]='%s' [3]='%s'\n", ed0, ed1, ed2, ed3) + testcheck.check(ed0 == "", "edge[0]=='' OK", "edge[0] FAILED") + testcheck.check(ed1 == "a", "edge[1]=='a' OK", "edge[1] FAILED") + testcheck.check(ed2 == "b", "edge[2]=='b' OK", "edge[2] FAILED") + testcheck.check(ed3 == "", "edge[3]=='' OK", "edge[3] FAILED") + + # === with 上下文自动注入测试 === + testcheck.section("Test 9: with 上下文自动注入 __mbuddy__") + + arena2: bytes = stdlib.malloc(65536) + with memhub.MemBuddy(arena2, 65536) as bd2: + # str 变量在 with 上下文内创建，__mbuddy__ 应自动注入（无需手动赋值） + w_str: str = "WithTest" + + # 验证 __mbuddy__ 已自动注入（非 NULL） + w_buddy: memhub.MemBuddy | t.CPtr = w_str.__mbuddy__ + testcheck.check(t.CInt(w_buddy) != 0, "with inject OK: __mbuddy__ auto-injected", "with inject FAIL: __mbuddy__ is NULL") + + # upper() 不需要手动设置 __mbuddy__ + w_upper: str = w_str.upper() + stdio.printf("with upper('WithTest') = %s\n", w_upper) + testcheck.check(w_upper == "WITHTEST", "with upper OK", "with upper FAILED") + + # lower() 不需要手动设置 __mbuddy__ + w_lower: str = w_str.lower() + stdio.printf("with lower('WithTest') = %s\n", w_lower) + testcheck.check(w_lower == "withtest", "with lower OK", "with lower FAILED") + + # split() 不需要手动设置 __mbuddy__ + w_split_str: str = "x,y,z" + w_parts = w_split_str.split(',') + w_n: t.CSizeT = len(w_parts) + stdio.printf("with split('x,y,z') len=%lu\n", w_n) + testcheck.check(w_n == 3, "with split len OK (3)", "with split len FAILED") + + # 验证多个 str 变量都获得注入 + w_str2: str = "Multiple" + w_buddy2: memhub.MemBuddy | t.CPtr = w_str2.__mbuddy__ + testcheck.check(t.CInt(w_buddy2) != 0, "with multi inject OK", "with multi inject FAIL") + + test_s: str = "www.gvsds.com" + list_s: list[str] = test_s.split('.') + # 迭代器测试: for idx, i in zip(range(len(list_s)), list_s) + iter_count: int = 0 + for iter_idx, i in zip(range(len(list_s)), list_s): + stdio.printf("list_s[%d]='%s'\n", iter_idx, i) + iter_count += 1 + testcheck.check(iter_count == 3, "zip(range(len), list) iter OK (3)", "zip(range(len), list) iter FAILED") + #for i in range(3): + # stdio.printf("list_s[%d]='%s'\n", i, list_s[i]) + + # === zip() 内置测试 === + testcheck.section("Test 10: zip(list, list) 元组解包") + za: str = "a,b,c" + zb: str = "1,2,3" + list_a: list[str] = za.split(',') + list_b: list[str] = zb.split(',') + zip_idx: int = 0 + for ca, cb in zip(list_a, list_b): + stdio.printf("zip[%d]: '%s'='%s'\n", zip_idx, ca, cb) + zip_idx += 1 + testcheck.check(zip_idx == 3, "zip list+list OK (3)", "zip list+list FAILED") + + testcheck.section("Test 11: zip(range, list) 混合迭代") + zip_idx2: int = 0 + for n, s in zip(range(3), list_a): + stdio.printf("zip range+list[%d]: %d='%s'\n", zip_idx2, n, s) + zip_idx2 += 1 + testcheck.check(zip_idx2 == 3, "zip range+list OK (3)", "zip range+list FAILED") + + testcheck.section("Test 12: zip(str, str) 字符级并行") + sa: str = "abc" + sb: str = "123" + zip_idx3: int = 0 + for ch_a, ch_b in zip(sa, sb): + stdio.printf("zip str+str[%d]: '%c'%c\n", zip_idx3, ch_a, ch_b) + zip_idx3 += 1 + testcheck.check(zip_idx3 == 3, "zip str+str OK (3)", "zip str+str FAILED") + + stdio.printf("%d\n", " ".isspace()) + + stdlib.free(arena2) + + stdlib.free(arena) + + return testcheck.end() + + + + diff --git a/Test/FakeDuckTest/output/48352329f97d0858.deps.json b/Test/FakeDuckTest/output/48352329f97d0858.deps.json new file mode 100644 index 0000000..11eebcc --- /dev/null +++ b/Test/FakeDuckTest/output/48352329f97d0858.deps.json @@ -0,0 +1 @@ +{"w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "atom": "271ea3decb810db2", "_withcontent": "5b2e220c5bae60c9", "stdio": "6f62fe05c5ea1ceb", "_fakeduck": "79b337e5ea8951e2", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/FakeDuckTest/project.json b/Test/FakeDuckTest/project.json new file mode 100644 index 0000000..1107714 --- /dev/null +++ b/Test/FakeDuckTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "FakeDuckTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "FakeDuckTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/FakeDuckTest/temp/0035c95a18d4f8e8.pyi b/Test/FakeDuckTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/FakeDuckTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/FakeDuckTest/temp/067c78e9f121dce3.pyi b/Test/FakeDuckTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/FakeDuckTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/FakeDuckTest/temp/06f53cc594b4ac6c.pyi b/Test/FakeDuckTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/FakeDuckTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/FakeDuckTest/temp/271ea3decb810db2.pyi b/Test/FakeDuckTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/FakeDuckTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/FakeDuckTest/temp/48352329f97d0858.doc.json b/Test/FakeDuckTest/temp/48352329f97d0858.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/FakeDuckTest/temp/48352329f97d0858.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/48352329f97d0858.pyi b/Test/FakeDuckTest/temp/48352329f97d0858.pyi new file mode 100644 index 0000000..06f88af --- /dev/null +++ b/Test/FakeDuckTest/temp/48352329f97d0858.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import _fakeduck +import stdio +import stdlib +import string +import memhub +import testcheck +import _withcontent + +def main() -> CInt | CExport: pass diff --git a/Test/FakeDuckTest/temp/5b2e220c5bae60c9.pyi b/Test/FakeDuckTest/temp/5b2e220c5bae60c9.pyi new file mode 100644 index 0000000..364be1a --- /dev/null +++ b/Test/FakeDuckTest/temp/5b2e220c5bae60c9.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from _withcontent.py +Module: _withcontent +""" + + +import t, c +from stdint import * +from w32.win32process import TlsAlloc, TlsGetValue, TlsSetValue +import stdlib +import stdio +import string + +_MAX_PROVIDER_DEPTH: t.CDefine = 32 + +@t.Object +class _ProviderStack: + names: t.CArray[VOIDPTR, 32] + ptrs: t.CArray[VOIDPTR, 32] + top: UINT64 + +_tls_index: t.CExtern | ULONG + +@t.TLS +def _init_tls() -> t.CInt: pass + +def _get_stack() -> _ProviderStack | t.CPtr: pass + +def _push_provider(field_name: str, ptr: t.CPtr) -> t.CInt: pass + +def _pop_provider() -> t.CInt: pass + +def _find_provider(field_name: str) -> t.CPtr: pass diff --git a/Test/FakeDuckTest/temp/6446627d4f07a1b5.pyi b/Test/FakeDuckTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/FakeDuckTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/FakeDuckTest/temp/6f62fe05c5ea1ceb.pyi b/Test/FakeDuckTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/FakeDuckTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/72e2d5ccb7cedcf1.pyi b/Test/FakeDuckTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/FakeDuckTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/FakeDuckTest/temp/79b337e5ea8951e2.pyi b/Test/FakeDuckTest/temp/79b337e5ea8951e2.pyi new file mode 100644 index 0000000..aa71e73 --- /dev/null +++ b/Test/FakeDuckTest/temp/79b337e5ea8951e2.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from _fakeduck.py +Module: _fakeduck +""" + +import c + + +import t +import memhub +import string + +class _Type: + def type_id(self: _Type) -> int: pass +class _str: + __requires__: list[str] = ['__mbuddy__'] + __data__: str + __mbuddy__: memhub.MemBuddy | t.CPtr + def upper(self: _str) -> str: pass + def lower(self: _str) -> str: pass + def capitalize(self: _str) -> str: pass + def swapcase(self: _str) -> str: pass + def title(self: _str) -> str: pass + def casefold(self: _str) -> str: pass + def isupper(self: _str) -> bool: pass + def islower(self: _str) -> bool: pass + def isalpha(self: _str) -> bool: pass + def isdigit(self: _str) -> bool: pass + def isalnum(self: _str) -> bool: pass + def isspace(self: _str) -> bool: pass + def isascii(self: _str) -> bool: pass + def istitle(self: _str) -> bool: pass + def split(self: _str, delimiter: str) -> list[str]: pass \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/7e529fe7a078cfef.pyi b/Test/FakeDuckTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/FakeDuckTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/FakeDuckTest/temp/90c53dd6db8d41cf.pyi b/Test/FakeDuckTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/FakeDuckTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/FakeDuckTest/temp/_phase1_manifest.json b/Test/FakeDuckTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..6157aa9 --- /dev/null +++ b/Test/FakeDuckTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\FakeDuckTest\\App\\main.py": {"sha1": "48352329f97d0858", "mtime": 1782827899.6785, "size": 12227}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_fakeduck.py": {"sha1": "79b337e5ea8951e2", "mtime": 1782826620.7129657, "size": 9717}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_withcontent.py": {"sha1": "5b2e220c5bae60c9", "mtime": 1782475015.06193, "size": 2039}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32process.py": {"sha1": "067c78e9f121dce3", "mtime": 1781200703.5409546, "size": 4802}} \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/_sha1_map.txt b/Test/FakeDuckTest/temp/_sha1_map.txt new file mode 100644 index 0000000..c1be7c3 --- /dev/null +++ b/Test/FakeDuckTest/temp/_sha1_map.txt @@ -0,0 +1,15 @@ +067c78e9f121dce3:includes/w32\win32process.py +271ea3decb810db2:includes/atom.py +48352329f97d0858:main.py +5b2e220c5bae60c9:includes/_withcontent.py +6f62fe05c5ea1ceb:includes/stdio.py +79b337e5ea8951e2:includes/_fakeduck.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/FakeDuckTest/temp/ab6e54ba9a669f76.pyi b/Test/FakeDuckTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/FakeDuckTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/FakeDuckTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/FakeDuckTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/FakeDuckTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/FakeDuckTest/temp/c9f4be41ca1cc2b4.pyi b/Test/FakeDuckTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/FakeDuckTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/dd3002730623424b.pyi b/Test/FakeDuckTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/FakeDuckTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/FakeDuckTest/temp/ee084e9fc6ee413a.pyi b/Test/FakeDuckTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/FakeDuckTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/f2ecbf8ced20575c.pyi b/Test/FakeDuckTest/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/Test/FakeDuckTest/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/f5522571bcce7bcb.pyi b/Test/FakeDuckTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/FakeDuckTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/FakeDuckTest/temp/f6b51804a0ba8ff0.pyi b/Test/FakeDuckTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/FakeDuckTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/ForAndLoopTest/App/main.py b/Test/ForAndLoopTest/App/main.py new file mode 100644 index 0000000..735b66b --- /dev/null +++ b/Test/ForAndLoopTest/App/main.py @@ -0,0 +1,159 @@ +import t +import c +import stdio +import stdlib +import string +import memhub +import testcheck +import _fakeduck + + +def test_for_in_range(): + testcheck.section("Test 1: for i in range(n)") + s: t.CInt = 0 + for i in range(10): + s += i + testcheck.check(s == 45, "range sum OK (45)", "range sum FAILED expect 45") + + +def test_for_in_str(): + testcheck.section("Test 2: for ch in str") + cnt: t.CInt = 0 + for ch in "hello": + cnt += 1 + testcheck.check(cnt == 5, "str iter OK (5)", "str iter FAILED expect 5") + + +def test_for_in_list(): + testcheck.section("Test 3: for v in list[str]") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + src: str = "x,y,z,w" + src.__mbuddy__ = bd + lst: list[str] = src.split(',') + total: t.CInt = 0 + for v in lst: + total += 1 + testcheck.check(total == 4, "list iter OK (4)", "list iter FAILED expect 4") + stdlib.free(arena) + + +def test_zip_list_list(): + testcheck.section("Test 4: zip(list, list)") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + a: str = "a,b,c" + a.__mbuddy__ = bd + b: str = "1,2,3" + b.__mbuddy__ = bd + la: list[str] = a.split(',') + lb: list[str] = b.split(',') + cnt: t.CInt = 0 + for ca, cb in zip(la, lb): + stdio.printf("zip[%d]: '%s'='%s'\n", cnt, ca, cb) + cnt += 1 + testcheck.check(cnt == 3, "zip list+list OK (3)", "zip list+list FAILED") + stdlib.free(arena) + + +def test_zip_range_list(): + testcheck.section("Test 5: zip(range, list)") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + s: str = "p,q,r" + s.__mbuddy__ = bd + lst: list[str] = s.split(',') + cnt: t.CInt = 0 + for n, v in zip(range(3), lst): + stdio.printf("zip range+list[%d]: %d='%s'\n", cnt, n, v) + cnt += 1 + testcheck.check(cnt == 3, "zip range+list OK (3)", "zip range+list FAILED") + stdlib.free(arena) + + +def test_zip_str_str(): + testcheck.section("Test 6: zip(str, str)") + sa: str = "abc" + sb: str = "123" + cnt: t.CInt = 0 + for ca, cb in zip(sa, sb): + stdio.printf("zip str+str[%d]: '%c'%c\n", cnt, ca, cb) + cnt += 1 + testcheck.check(cnt == 3, "zip str+str OK (3)", "zip str+str FAILED") + + +def test_zip_range_len_list(): + testcheck.section("Test 7: zip(range(len), list)") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + s: str = "www.gvsds.com" + s.__mbuddy__ = bd + lst: list[str] = s.split('.') + cnt: t.CInt = 0 + for idx, v in zip(range(len(lst)), lst): + stdio.printf("lst[%d]='%s'\n", idx, v) + cnt += 1 + testcheck.check(cnt == 3, "zip(range(len),list) OK (3)", "zip(range(len),list) FAILED") + stdlib.free(arena) + + +def test_enumerate_list(): + testcheck.section("Test 8: enumerate(list)") + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + s: str = "alpha,beta,gamma" + s.__mbuddy__ = bd + lst: list[str] = s.split(',') + cnt: t.CInt = 0 + last_idx: t.CInt = -1 + for idx, v in enumerate(lst): + stdio.printf("enum[%d]='%s'\n", idx, v) + last_idx = idx + cnt += 1 + testcheck.check(cnt == 3, "enumerate list OK (3)", "enumerate list FAILED") + testcheck.check(last_idx == 2, "enumerate last_idx OK (2)", "enumerate last_idx FAILED expect 2") + stdlib.free(arena) + + +def test_enumerate_str(): + testcheck.section("Test 9: enumerate(str)") + cnt: t.CInt = 0 + last_idx: t.CInt = -1 + for idx, ch in enumerate("ABCD"): + stdio.printf("enum str[%d]='%c'\n", idx, ch) + last_idx = idx + cnt += 1 + testcheck.check(cnt == 4, "enumerate str OK (4)", "enumerate str FAILED expect 4") + testcheck.check(last_idx == 3, "enumerate str last_idx OK (3)", "enumerate str last_idx FAILED expect 3") + + +def test_enumerate_range(): + testcheck.section("Test 10: enumerate(range)") + cnt: t.CInt = 0 + last_idx: t.CInt = -1 + last_val: t.CInt = -1 + for idx, v in enumerate(range(5)): + stdio.printf("enum range[%d]=%d\n", idx, v) + last_idx = idx + last_val = v + cnt += 1 + testcheck.check(cnt == 5, "enumerate range OK (5)", "enumerate range FAILED expect 5") + testcheck.check(last_idx == 4, "enumerate range last_idx OK (4)", "enumerate range last_idx FAILED expect 4") + testcheck.check(last_val == 4, "enumerate range last_val OK (4)", "enumerate range last_val FAILED expect 4") + + +def main() -> t.CInt: + testcheck.begin("ForAndLoopTest: 迭代器与循环测试") + + test_for_in_range() + test_for_in_str() + test_for_in_list() + test_zip_list_list() + test_zip_range_list() + test_zip_str_str() + test_zip_range_len_list() + test_enumerate_list() + test_enumerate_str() + test_enumerate_range() + + return testcheck.end() diff --git a/Test/ForAndLoopTest/output/6deb1a24b47f47fc.deps.json b/Test/ForAndLoopTest/output/6deb1a24b47f47fc.deps.json new file mode 100644 index 0000000..4e6ab8b --- /dev/null +++ b/Test/ForAndLoopTest/output/6deb1a24b47f47fc.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "_fakeduck": "79b337e5ea8951e2", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ForAndLoopTest/project.json b/Test/ForAndLoopTest/project.json new file mode 100644 index 0000000..0a6fdaa --- /dev/null +++ b/Test/ForAndLoopTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ForAndLoopTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32"], + "output": "ForAndLoopTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ForAndLoopTest/temp/271ea3decb810db2.pyi b/Test/ForAndLoopTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/ForAndLoopTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.doc.json b/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.pyi b/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.pyi new file mode 100644 index 0000000..1c16d75 --- /dev/null +++ b/Test/ForAndLoopTest/temp/6deb1a24b47f47fc.pyi @@ -0,0 +1,36 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import c +import stdio +import stdlib +import string +import memhub +import testcheck +import _fakeduck + +def test_for_in_range() -> t.CInt: pass + +def test_for_in_str() -> t.CInt: pass + +def test_for_in_list() -> t.CInt: pass + +def test_zip_list_list() -> t.CInt: pass + +def test_zip_range_list() -> t.CInt: pass + +def test_zip_str_str() -> t.CInt: pass + +def test_zip_range_len_list() -> t.CInt: pass + +def test_enumerate_list() -> t.CInt: pass + +def test_enumerate_str() -> t.CInt: pass + +def test_enumerate_range() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/ForAndLoopTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ForAndLoopTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ForAndLoopTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/79b337e5ea8951e2.pyi b/Test/ForAndLoopTest/temp/79b337e5ea8951e2.pyi new file mode 100644 index 0000000..aa71e73 --- /dev/null +++ b/Test/ForAndLoopTest/temp/79b337e5ea8951e2.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from _fakeduck.py +Module: _fakeduck +""" + +import c + + +import t +import memhub +import string + +class _Type: + def type_id(self: _Type) -> int: pass +class _str: + __requires__: list[str] = ['__mbuddy__'] + __data__: str + __mbuddy__: memhub.MemBuddy | t.CPtr + def upper(self: _str) -> str: pass + def lower(self: _str) -> str: pass + def capitalize(self: _str) -> str: pass + def swapcase(self: _str) -> str: pass + def title(self: _str) -> str: pass + def casefold(self: _str) -> str: pass + def isupper(self: _str) -> bool: pass + def islower(self: _str) -> bool: pass + def isalpha(self: _str) -> bool: pass + def isdigit(self: _str) -> bool: pass + def isalnum(self: _str) -> bool: pass + def isspace(self: _str) -> bool: pass + def isascii(self: _str) -> bool: pass + def istitle(self: _str) -> bool: pass + def split(self: _str, delimiter: str) -> list[str]: pass \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/7e529fe7a078cfef.pyi b/Test/ForAndLoopTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ForAndLoopTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ForAndLoopTest/temp/90c53dd6db8d41cf.pyi b/Test/ForAndLoopTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/ForAndLoopTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/ForAndLoopTest/temp/_phase1_manifest.json b/Test/ForAndLoopTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..1b29052 --- /dev/null +++ b/Test/ForAndLoopTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ForAndLoopTest\\App\\main.py": {"sha1": "6deb1a24b47f47fc", "mtime": 1782827984.6760156, "size": 4951}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_fakeduck.py": {"sha1": "79b337e5ea8951e2", "mtime": 1782826620.7129657, "size": 9717}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/_sha1_map.txt b/Test/ForAndLoopTest/temp/_sha1_map.txt new file mode 100644 index 0000000..dedebc9 --- /dev/null +++ b/Test/ForAndLoopTest/temp/_sha1_map.txt @@ -0,0 +1,13 @@ +271ea3decb810db2:includes/atom.py +6deb1a24b47f47fc:main.py +6f62fe05c5ea1ceb:includes/stdio.py +79b337e5ea8951e2:includes/_fakeduck.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ForAndLoopTest/temp/ab6e54ba9a669f76.pyi b/Test/ForAndLoopTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/ForAndLoopTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/ForAndLoopTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ForAndLoopTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ForAndLoopTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ForAndLoopTest/temp/c9f4be41ca1cc2b4.pyi b/Test/ForAndLoopTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/ForAndLoopTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/dd3002730623424b.pyi b/Test/ForAndLoopTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ForAndLoopTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ForAndLoopTest/temp/ee084e9fc6ee413a.pyi b/Test/ForAndLoopTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/ForAndLoopTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/f2ecbf8ced20575c.pyi b/Test/ForAndLoopTest/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/Test/ForAndLoopTest/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/Test/ForAndLoopTest/temp/f5522571bcce7bcb.pyi b/Test/ForAndLoopTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ForAndLoopTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/FormatStringTest/App/main.py b/Test/FormatStringTest/App/main.py new file mode 100644 index 0000000..0af881a --- /dev/null +++ b/Test/FormatStringTest/App/main.py @@ -0,0 +1,52 @@ +from stdint import * +import t, c +import ctraits +import stdio +import testcheck + + +# ============================================================ +# FormatStringTest - 格式化字符串回归测试 +# ============================================================ +@t.CExport +def main() -> int: + testcheck.begin("FormatStringTest: Format String Regression") + + b: str = "hello world" + for i in b: + stdio.printf("%c\n", i) + b += 1 + print(b-11) + + l: t.CArray[u64] = [0, 43, 2, 3, 4] + l[0] = 1000 + for i in l: + stdio.printf("%llu\n", i) + + print("Value: ", f"0x{ULONGLONG(0x11)}") + print("ULL Hex: ", f"0x{ULONGLONG(0x11):016X}") + print("LL Hex: ", f"{LONGLONG(0x11):016X}") + print("ULL Dec: ", f"{ULONGLONG(42):d}") + print("LL Dec: ", f"{LONGLONG(42):d}") + print("Hex: ", f"{0x11:016X}") + print("Dec: ", f"{42:d}") + print("Float: ", f"{3.14159:.2f}") + + # f-string 赋值测试（验证不再直接 printf，而是生成字符串） + s: str = f"Value=0x{ULONGLONG(42)}, Hex=0x{ULONGLONG(0x11):016X}" + print("Assigned: ", s) + + # ctraits.isptr 编译时类型判断测试 + ptr_var: INTPTR = c.Addr(s) + val_var: int = 42 + if c.CIf(ctraits.isptr(ptr_var)): + print("ptr_var is pointer") + if c.CIf(not ctraits.isptr(val_var)): + print("val_var is not pointer") + + testcheck.section("placeholder") + testcheck.ok("testcheck framework ready") + testcheck.info("placeholder test") + + return testcheck.end() + diff --git a/Test/FormatStringTest/output/cdff0d7b58fc935b.deps.json b/Test/FormatStringTest/output/cdff0d7b58fc935b.deps.json new file mode 100644 index 0000000..7701fda --- /dev/null +++ b/Test/FormatStringTest/output/cdff0d7b58fc935b.deps.json @@ -0,0 +1 @@ +{"ctraits": "42ced27d9e76aef5", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/FormatStringTest/project.json b/Test/FormatStringTest/project.json new file mode 100644 index 0000000..b5be263 --- /dev/null +++ b/Test/FormatStringTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "FormatStringTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "FormatStringTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/FormatStringTest/temp/42ced27d9e76aef5.pyi b/Test/FormatStringTest/temp/42ced27d9e76aef5.pyi new file mode 100644 index 0000000..f1d44b9 --- /dev/null +++ b/Test/FormatStringTest/temp/42ced27d9e76aef5.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from ctraits.py +Module: ctraits +""" + +import c + + +import t + +def isptr(x: t.CType) -> bool: pass diff --git a/Test/FormatStringTest/temp/6f62fe05c5ea1ceb.pyi b/Test/FormatStringTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/FormatStringTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/FormatStringTest/temp/7e529fe7a078cfef.pyi b/Test/FormatStringTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/FormatStringTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/FormatStringTest/temp/_phase1_manifest.json b/Test/FormatStringTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..2b5f3ae --- /dev/null +++ b/Test/FormatStringTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\FormatStringTest\\App\\main.py": {"sha1": "cdff0d7b58fc935b", "mtime": 1782266179.3674748, "size": 1498}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ctraits.pyi": {"sha1": "42ced27d9e76aef5", "mtime": 1782216490.489452, "size": 572}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/FormatStringTest/temp/_sha1_map.txt b/Test/FormatStringTest/temp/_sha1_map.txt new file mode 100644 index 0000000..3fb222c --- /dev/null +++ b/Test/FormatStringTest/temp/_sha1_map.txt @@ -0,0 +1,7 @@ +42ced27d9e76aef5:includes/ctraits.pyi +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +cdff0d7b58fc935b:main.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/FormatStringTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/FormatStringTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/FormatStringTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/FormatStringTest/temp/cdff0d7b58fc935b.doc.json b/Test/FormatStringTest/temp/cdff0d7b58fc935b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/FormatStringTest/temp/cdff0d7b58fc935b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/FormatStringTest/temp/cdff0d7b58fc935b.pyi b/Test/FormatStringTest/temp/cdff0d7b58fc935b.pyi new file mode 100644 index 0000000..0fb8976 --- /dev/null +++ b/Test/FormatStringTest/temp/cdff0d7b58fc935b.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import ctraits +import stdio +import testcheck + +@t.CExport +def main() -> int: pass diff --git a/Test/FormatStringTest/temp/dd3002730623424b.pyi b/Test/FormatStringTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/FormatStringTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/FormatStringTest/temp/f5522571bcce7bcb.pyi b/Test/FormatStringTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/FormatStringTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/FuncTest/App/main.py b/Test/FuncTest/App/main.py new file mode 100644 index 0000000..77a7372 --- /dev/null +++ b/Test/FuncTest/App/main.py @@ -0,0 +1,106 @@ +import t +import stdio +import testcheck + +# 测试基本函数调用 +def add(a: t.CInt, b: t.CInt) -> t.CInt: + return a + b + +def sub(a: t.CInt, b: t.CInt) -> t.CInt: + return a - b + +# 测试多参数 +def multi_args(a: t.CInt, b: t.CInt, c: t.CInt, d: t.CInt) -> t.CInt: + return a + b + c + d + +# 测试递归 +def factorial(n: t.CInt) -> t.CInt: + if n <= 1: + return 1 + return n * factorial(n - 1) + +# 测试尾递归式斐波那契 +def fib(n: t.CInt) -> t.CInt: + if n <= 0: + return 0 + if n == 1: + return 1 + return fib(n - 1) + fib(n - 2) + +# 测试指针参数 +def add_via_ptr(p: t.CInt | t.CPtr, val: t.CInt): + p[0] = p[0] + val + +# 测试 u64 返回值 +def make_u64() -> t.CUInt64T: + return t.CUInt64T(0xDEADBEEF) + +# 测试 bool 返回值 +def is_even(n: t.CInt) -> t.CBool: + return n % 2 == 0 + +def test_basic_call(): + testcheck.section("Test 1: basic function call") + r: t.CInt = add(10, 20) + testcheck.check(r == 30, "add(10,20)=30", "add(10,20) expect 30") + +def test_multi_args(): + testcheck.section("Test 2: multi args") + r: t.CInt = multi_args(1, 2, 3, 4) + testcheck.check(r == 10, "multi_args(1,2,3,4)=10", "multi_args(1,2,3,4) expect 10") + +def test_recursion(): + testcheck.section("Test 3: recursion (factorial)") + r: t.CInt = factorial(10) + # 10! = 3628800 + testcheck.check(r == 3628800, "factorial(10)=3628800", "factorial(10) expect 3628800") + +def test_fibonacci(): + testcheck.section("Test 4: fibonacci") + r: t.CInt = fib(10) + # fib(10) = 55 + testcheck.check(r == 55, "fib(10)=55", "fib(10) expect 55") + +def test_ptr_param(): + testcheck.section("Test 5: pointer parameter") + x: t.CInt = 10 + # 需要取 x 的地址 + import w32.win32memory + buf: t.CInt | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4) + if t.CUInt64T(buf) == 0: + testcheck.info("SKIP: VirtualAlloc returned NULL") + return + buf[0] = 10 + add_via_ptr(buf, 25) + testcheck.check(buf[0] == 35, "ptr param (result=35)", "ptr param expect 35") + +def test_u64_return(): + testcheck.section("Test 6: u64 return") + r: t.CUInt64T = make_u64() + testcheck.check(r == t.CUInt64T(0xDEADBEEF), "u64 return (0xdeadbeef)", "u64 return expect 0xdeadbeef") + +def test_bool_return(): + testcheck.section("Test 7: bool return") + r1: t.CBool = is_even(4) + r2: t.CBool = is_even(7) + testcheck.check(r1 and not r2, "bool return (is_even(4)=1, is_even(7)=0)", "bool return expect is_even(4)=1, is_even(7)=0") + +def test_nested_call(): + testcheck.section("Test 8: nested function call") + r: t.CInt = add(factorial(3), sub(10, 3)) + # 3! = 6, 10-3 = 7, 6+7 = 13 + testcheck.check(r == 13, "nested call (add(factorial(3), sub(10,3))=13)", "nested call expect 13") + +import w32.win32memory + +def main() -> t.CInt: + testcheck.begin("FuncTest: 函数调用基准测试") + test_basic_call() + test_multi_args() + test_recursion() + test_fibonacci() + test_ptr_param() + test_u64_return() + test_bool_return() + test_nested_call() + return testcheck.end() diff --git a/Test/FuncTest/output/067f811511854c88.deps.json b/Test/FuncTest/output/067f811511854c88.deps.json new file mode 100644 index 0000000..92c368b --- /dev/null +++ b/Test/FuncTest/output/067f811511854c88.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/FuncTest/project.json b/Test/FuncTest/project.json new file mode 100644 index 0000000..1099565 --- /dev/null +++ b/Test/FuncTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "FuncTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "FuncTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/FuncTest/temp/0035c95a18d4f8e8.pyi b/Test/FuncTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/FuncTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/FuncTest/temp/067c78e9f121dce3.pyi b/Test/FuncTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/FuncTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/FuncTest/temp/067f811511854c88.doc.json b/Test/FuncTest/temp/067f811511854c88.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/FuncTest/temp/067f811511854c88.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/FuncTest/temp/067f811511854c88.pyi b/Test/FuncTest/temp/067f811511854c88.pyi new file mode 100644 index 0000000..cef2575 --- /dev/null +++ b/Test/FuncTest/temp/067f811511854c88.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def add(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def sub(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def multi_args(a: t.CInt, b: t.CInt, c: t.CInt, d: t.CInt) -> t.CInt: pass + +def factorial(n: t.CInt) -> t.CInt: pass + +def fib(n: t.CInt) -> t.CInt: pass + +def add_via_ptr(p: t.CInt | t.CPtr, val: t.CInt) -> t.CInt: pass + +def make_u64() -> t.CUInt64T: pass + +def is_even(n: t.CInt) -> t.CBool: pass + +def test_basic_call() -> t.CInt: pass + +def test_multi_args() -> t.CInt: pass + +def test_recursion() -> t.CInt: pass + +def test_fibonacci() -> t.CInt: pass + +def test_ptr_param() -> t.CInt: pass + +def test_u64_return() -> t.CInt: pass + +def test_bool_return() -> t.CInt: pass + +def test_nested_call() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/Test/FuncTest/temp/06f53cc594b4ac6c.pyi b/Test/FuncTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/FuncTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/FuncTest/temp/6446627d4f07a1b5.pyi b/Test/FuncTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/FuncTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/FuncTest/temp/6f62fe05c5ea1ceb.pyi b/Test/FuncTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/FuncTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/FuncTest/temp/72e2d5ccb7cedcf1.pyi b/Test/FuncTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/FuncTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/FuncTest/temp/7e529fe7a078cfef.pyi b/Test/FuncTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/FuncTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/FuncTest/temp/_phase1_manifest.json b/Test/FuncTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..e7b1669 --- /dev/null +++ b/Test/FuncTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\FuncTest\\App\\main.py": {"sha1": "067f811511854c88", "mtime": 1784309886.7278845, "size": 3098}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/FuncTest/temp/_sha1_map.txt b/Test/FuncTest/temp/_sha1_map.txt new file mode 100644 index 0000000..35cdb6f --- /dev/null +++ b/Test/FuncTest/temp/_sha1_map.txt @@ -0,0 +1,7 @@ +067f811511854c88:main.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/FuncTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/FuncTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/FuncTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/FuncTest/temp/dd3002730623424b.pyi b/Test/FuncTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/FuncTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/FuncTest/temp/f5522571bcce7bcb.pyi b/Test/FuncTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/FuncTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/FuncTest/temp/f6b51804a0ba8ff0.pyi b/Test/FuncTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/FuncTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/GenericTest/App/main.py b/Test/GenericTest/App/main.py new file mode 100644 index 0000000..0595c27 --- /dev/null +++ b/Test/GenericTest/App/main.py @@ -0,0 +1,167 @@ +import t +import stdio +import testcheck + +# ============================================================ +# Test 1: 泛型函数 - 基本加法 +# ============================================================ +def add[T](a: T, b: T) -> T: + return a + b + +def test_generic_add(): + testcheck.section("Test 1: generic add") + r1: t.CInt = add(3, 5) + r2: t.CInt = add(100, 200) + testcheck.check(r1 == 8 and r2 == 300, "generic add (3+5=8, 100+200=300)", "generic add expect 8 and 300") + +# ============================================================ +# Test 2: 泛型函数 - 多类型参数 +# ============================================================ +def combine[T1, T2](a: T1, b: T2) -> T1: + return a + T1(b) + +def test_generic_combine(): + testcheck.section("Test 2: generic combine") + r: t.CInt = combine(10, 20) + testcheck.check(r == 30, "generic combine (10+T1(20)=30)", "generic combine expect 30") + +# ============================================================ +# Test 3: 泛型函数 - 最大值 +# ============================================================ +def max_val[T](a: T, b: T) -> T: + if a > b: + return a + return b + +def test_generic_max(): + testcheck.section("Test 3: generic max") + r1: t.CInt = max_val(10, 20) + r2: t.CInt = max_val(50, 30) + testcheck.check(r1 == 20 and r2 == 50, "generic max (max(10,20)=20, max(50,30)=50)", "generic max expect 20 and 50") + +# ============================================================ +# Test 4: 泛型类 - Box[T] +# ============================================================ +class Box[T](): + value: T + + def __init__(self, v: T): + self.value = T(v) + + def get(self) -> T: + return self.value + + def set(self, v: T): + self.value = T(v) + +def test_generic_box(): + testcheck.section("Test 4: generic Box[T]") + b = Box(42) + v: t.CInt = b.get() + testcheck.check(v == 42, "Box[int] (get=42)", "Box[int] expect 42") + +# ============================================================ +# Test 5: 泛型类 - Pair[T1, T2] +# ============================================================ +class Pair[T1, T2](): + first: T1 + second: T2 + + def __init__(self, a: T1, b: T2): + self.first = T1(a) + self.second = T2(b) + + def sum_as_int(self) -> t.CInt: + return t.CInt(self.first) + t.CInt(self.second) + +def test_generic_pair(): + testcheck.section("Test 5: generic Pair[T1,T2]") + p = Pair(10, 20) + r: t.CInt = p.sum_as_int() + testcheck.check(r == 30, "Pair[int,int] (sum=30)", "Pair[int,int] expect 30") + +# ============================================================ +# Test 6: list[T, N] 固定数组 +# ============================================================ +def test_fixed_array(): + testcheck.section("Test 6: list[T, N] fixed array") + arr: t.CArray[t.CInt, 5] = [0] + for i in range(5): + arr[i] = i * 10 + total: t.CInt = 0 + for i in range(5): + total += arr[i] + # 0+10+20+30+40 = 100 + testcheck.check(total == 100, "fixed array (total=100)", "fixed array expect 100") + +# ============================================================ +# Test 7: list[t.CChar, N] 字符数组 +# ============================================================ +def test_char_array(): + testcheck.section("Test 7: list[t.CChar, N] char array") + buf: t.CArray[t.CChar, 32] = [0] + buf[0] = 72 # H + buf[1] = 105 # i + buf[2] = 0 + testcheck.check(buf[0] == 72 and buf[1] == 105, "char array (H=72 i=105)", "char array expect H=72 i=105") + +# ============================================================ +# Test 8: 泛型函数多次特化 +# ============================================================ +def identity[T](x: T) -> T: + return x + +def test_generic_identity(): + testcheck.section("Test 8: generic identity") + r1: t.CInt = identity(42) + r2: t.CInt = identity(0) + testcheck.check(r1 == 42 and r2 == 0, "identity (42=42, 0=0)", "identity expect 42 and 0") + +# ============================================================ +# Test 9: 泛型函数嵌套调用 +# ============================================================ +def double_val[T](x: T) -> T: + return add(x, x) + +def test_generic_nested(): + testcheck.section("Test 9: generic nested call") + r: t.CInt = double_val(21) + testcheck.check(r == 42, "generic nested (double(21)=42)", "generic nested expect 42") + +# ============================================================ +# Test 10: 泛型类方法调用 +# ============================================================ +class Counter[T](): + count: T + + def __init__(self, start: T): + self.count = T(start) + + def increment(self): + self.count = add(self.count, T(1)) + + def get_count(self) -> T: + return self.count + +def test_generic_counter(): + testcheck.section("Test 10: generic Counter[T]") + c = Counter(0) + c.increment() + c.increment() + c.increment() + r: t.CInt = c.get_count() + testcheck.check(r == 3, "Counter[int] (count=3)", "Counter[int] expect 3") + +def main() -> t.CInt: + testcheck.begin("GenericTest: 泛型测试") + test_generic_add() + test_generic_combine() + test_generic_max() + test_generic_box() + test_generic_pair() + test_fixed_array() + test_char_array() + test_generic_identity() + test_generic_nested() + test_generic_counter() + return testcheck.end() diff --git a/Test/GenericTest/output/16e1f65b154831e8.deps.json b/Test/GenericTest/output/16e1f65b154831e8.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/GenericTest/output/16e1f65b154831e8.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/GenericTest/project.json b/Test/GenericTest/project.json new file mode 100644 index 0000000..05323c1 --- /dev/null +++ b/Test/GenericTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "GenericTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "GenericTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/GenericTest/temp/16e1f65b154831e8.doc.json b/Test/GenericTest/temp/16e1f65b154831e8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/GenericTest/temp/16e1f65b154831e8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/GenericTest/temp/16e1f65b154831e8.pyi b/Test/GenericTest/temp/16e1f65b154831e8.pyi new file mode 100644 index 0000000..3167de1 --- /dev/null +++ b/Test/GenericTest/temp/16e1f65b154831e8.pyi @@ -0,0 +1,64 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def add[T](a: T, b: T) -> T: pass + +def test_generic_add() -> t.CInt: pass + +def combine[T1, T2](a: T1, b: T2) -> T1: pass + +def test_generic_combine() -> t.CInt: pass + +def max_val[T](a: T, b: T) -> T: pass + +def test_generic_max() -> t.CInt: pass + + +class Box[T]: + value: T + def __init__(self: Box, v: T) -> t.CInt: pass + def get(self: Box) -> T: pass + def set(self: Box, v: T) -> t.CInt: pass + +def test_generic_box() -> t.CInt: pass + + +class Pair[T1, T2]: + first: T1 + second: T2 + def __init__(self: Pair, a: T1, b: T2) -> t.CInt: pass + def sum_as_int(self: Pair) -> t.CInt: pass + +def test_generic_pair() -> t.CInt: pass + +def test_fixed_array() -> t.CInt: pass + +def test_char_array() -> t.CInt: pass + +def identity[T](x: T) -> T: pass + +def test_generic_identity() -> t.CInt: pass + +def double_val[T](x: T) -> T: pass + +def test_generic_nested() -> t.CInt: pass + + +class Counter[T]: + count: T + def __init__(self: Counter, start: T) -> t.CInt: pass + def increment(self: Counter) -> t.CInt: pass + def get_count(self: Counter) -> T: pass + +def test_generic_counter() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/GenericTest/temp/6f62fe05c5ea1ceb.pyi b/Test/GenericTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/GenericTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/GenericTest/temp/7e529fe7a078cfef.pyi b/Test/GenericTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/GenericTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/GenericTest/temp/_phase1_manifest.json b/Test/GenericTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..63ba4f7 --- /dev/null +++ b/Test/GenericTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\GenericTest\\App\\main.py": {"sha1": "16e1f65b154831e8", "mtime": 1782266275.0221426, "size": 5369}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/GenericTest/temp/_sha1_map.txt b/Test/GenericTest/temp/_sha1_map.txt new file mode 100644 index 0000000..3449281 --- /dev/null +++ b/Test/GenericTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +16e1f65b154831e8:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/GenericTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/GenericTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/GenericTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/GenericTest/temp/dd3002730623424b.pyi b/Test/GenericTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/GenericTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/GenericTest/temp/f5522571bcce7bcb.pyi b/Test/GenericTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/GenericTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/GenericTest2/App/main.py b/Test/GenericTest2/App/main.py new file mode 100644 index 0000000..b1749a3 --- /dev/null +++ b/Test/GenericTest2/App/main.py @@ -0,0 +1,268 @@ +import t +import stdio +import c +import testcheck + +# ============================================================ +# Test 1: 泛型函数 - 不同整数类型特化 +# ============================================================ +def square[T](x: T) -> T: + return x * x + +def test_generic_square_types(): + testcheck.section("Test 1: generic square with different types") + r1: t.CInt = square(5) + r2: t.CInt = square(10) + testcheck.check(r1 == 25 and r2 == 100, "square (5^2=25 10^2=100)", "square expect 25 100") + +# ============================================================ +# Test 2: 泛型函数 - 三参数 +# ============================================================ +def clamp[T](val: T, lo: T, hi: T) -> T: + if val < lo: + return lo + if val > hi: + return hi + return val + +def test_generic_clamp(): + testcheck.section("Test 2: generic clamp") + r1: t.CInt = clamp(5, 0, 10) + r2: t.CInt = clamp(-3, 0, 10) + r3: t.CInt = clamp(15, 0, 10) + testcheck.check(r1 == 5 and r2 == 0 and r3 == 10, "clamp (5 0 10)", "clamp expect 5 0 10") + +# ============================================================ +# Test 3: 泛型类 - Stack[T] +# ============================================================ +class Stack[T](): + data: t.CArray[t.CInt, 16] + top: t.CInt + + def __init__(self): + self.top = 0 + + def push(self, val: T): + self.data[self.top] = t.CInt(val) + self.top += 1 + + def pop(self) -> T: + self.top -= 1 + return T(self.data[self.top]) + + def is_empty(self) -> t.CInt: + if self.top == 0: + return 1 + return 0 + + def size_of(self) -> t.CSizeT: + return self.__sizeof__() + + def type_size(self) -> t.CSizeT: + return Stack[T].__sizeof__() + +def test_generic_stack(): + testcheck.section("Test 3: generic Stack[T]") + s = Stack() + s.push(10) + s.push(20) + s.push(30) + r1: t.CInt = s.pop() + r2: t.CInt = s.pop() + r3: t.CInt = s.pop() + testcheck.check(r1 == 30 and r2 == 20 and r3 == 10, "Stack (pop=30 20 10)", "Stack expect 30 20 10") + +# ============================================================ +# Test 4: 泛型函数 - swap via pointer +# ============================================================ +class IntPair: + a: t.CInt + b: t.CInt + +def generic_swap[T](p: IntPair | t.CPtr): + tmp: t.CInt = c.Deref(p).a + c.Deref(p).a = c.Deref(p).b + c.Deref(p).b = tmp + +def test_generic_swap(): + testcheck.section("Test 4: generic swap via pointer") + p: IntPair = IntPair() + p.a = 100 + p.b = 200 + generic_swap(c.Addr(p)) + testcheck.check(p.a == 200 and p.b == 100, "swap (a=200 b=100)", "swap expect 200 100") + +# ============================================================ +# Test 5: 泛型类 - Range[T] 迭代器 +# ============================================================ +class Range[T](): + start: T + end_val: T + step: T + + def __init__(self, s: T, e: T, st: T): + self.start = T(s) + self.end_val = T(e) + self.step = T(st) + + def sum_all(self) -> T: + result: T = T(0) + i: T = self.start + while i < self.end_val: + result = result + i + i = i + self.step + return result + +def test_generic_range(): + testcheck.section("Test 5: generic Range[T]") + r = Range(0, 10, 1) # 0+1+2+...+9=45 + s: t.CInt = r.sum_all() + testcheck.check(s == 45, "Range sum (0..9=45)", "Range sum expect 45") + +# ============================================================ +# Test 6: 泛型函数 - 数组求和（非泛型数组参数） +# ============================================================ +def array_sum(arr: t.CArray[t.CInt, 8], n: t.CInt) -> t.CInt: + result: t.CInt = 0 + for i in range(n): + result += arr[i] + return result + +def test_generic_array_sum(): + testcheck.section("Test 6: array sum") + arr: t.CArray[t.CInt, 8] = [0] + for i in range(8): + arr[i] = (i + 1) * 10 + s: t.CInt = array_sum(arr, 8) + testcheck.check(s == 360, "array sum (360)", "array sum expect 360") + +# ============================================================ +# Test 7: 泛型函数 - 比较并返回较大值 +# ============================================================ +def max_of_three[T](a: T, b: T, c_val: T) -> T: + m: T = a + if b > m: + m = b + if c_val > m: + m = c_val + return m + +def test_generic_max_three(): + testcheck.section("Test 7: generic max of three") + r1: t.CInt = max_of_three(1, 2, 3) + r2: t.CInt = max_of_three(10, 30, 20) + r3: t.CInt = max_of_three(50, 50, 50) + testcheck.check(r1 == 3 and r2 == 30 and r3 == 50, "max3 (3 30 50)", "max3 expect 3 30 50") + +# ============================================================ +# Test 8: 泛型类 - 带方法的容器 +# ============================================================ +class Container[T](): + values: t.CArray[t.CInt, 8] + size: t.CInt + + def __init__(self): + self.size = 0 + + def add(self, v: T): + self.values[self.size] = t.CInt(v) + self.size += 1 + + def get(self, idx: t.CInt) -> T: + return T(self.values[idx]) + + def contains(self, v: T) -> t.CInt: + for i in range(self.size): + if self.values[i] == t.CInt(v): + return 1 + return 0 + + def total(self) -> T: + result: T = T(0) + for i in range(self.size): + result = result + T(self.values[i]) + return result + +def test_generic_container(): + testcheck.section("Test 8: generic Container[T]") + c = Container() + c.add(10) + c.add(20) + c.add(30) + has20: t.CInt = c.contains(20) + has99: t.CInt = c.contains(99) + total: t.CInt = c.total() + testcheck.check(has20 == 1 and has99 == 0 and total == 60, "Container (has20=1 has99=0 total=60)", "Container expect 1 0 60") + +# ============================================================ +# Test 9: 泛型函数 - 阶乘 +# ============================================================ +def factorial[T](n: T) -> T: + result: T = T(1) + i: T = T(2) + while i <= n: + result = result * i + i = i + T(1) + return result + +def test_generic_factorial(): + testcheck.section("Test 9: generic factorial") + r1: t.CInt = factorial(5) + r2: t.CInt = factorial(10) + testcheck.check(r1 == 120 and r2 == 3628800, "factorial (5!=120 10!=3628800)", "factorial expect 120 3628800") + +# ============================================================ +# Test 10: 泛型类 - 累加器 +# ============================================================ +class Accumulator[T](): + total: T + + def __init__(self, start: T): + self.total = T(start) + + def add_val(self, v: T): + self.total = self.total + v + + def get_result(self) -> T: + return self.total + +def test_generic_accumulator(): + testcheck.section("Test 10: generic Accumulator[T]") + a = Accumulator(0) + a.add_val(10) + a.add_val(20) + a.add_val(30) + r: t.CInt = a.get_result() + testcheck.check(r == 60, "Accumulator (result=60)", "Accumulator expect 60") + +# ============================================================ +# Test 11: 泛型结构体 sizeof (修复验证) +# 特化后 Stack[int] = { [16 x i32], i32 } = 68 字节 +# 验证 obj.__sizeof__() 和类内部 self.__sizeof__() 均返回正确大小 +# ============================================================ +def test_generic_sizeof_bug(): + testcheck.section("Test 11: generic sizeof (fixed)") + s = Stack() + s.push(10) + sz: t.CSizeT = s.__sizeof__() + inner_sz: t.CSizeT = s.size_of() + type_sz: t.CSizeT = s.type_size() + stdio.printf("Stack[int] obj.__sizeof__() = %lu, self.__sizeof__() = %lu, Stack[T].__sizeof__() = %lu (expect 68)\n", sz, inner_sz, type_sz) + testcheck.check(sz == 68, "obj.__sizeof__() == 68 OK", "obj.__sizeof__() FAILED") + testcheck.check(inner_sz == 68, "self.__sizeof__() == 68 OK", "self.__sizeof__() FAILED") + testcheck.check(type_sz == 68, "Stack[T].__sizeof__() == 68 OK", "Stack[T].__sizeof__() FAILED") + +def main() -> t.CInt: + testcheck.begin("GenericTest2: Advanced Generic Tests") + test_generic_square_types() + test_generic_clamp() + test_generic_stack() + test_generic_swap() + test_generic_range() + test_generic_array_sum() + test_generic_max_three() + test_generic_container() + test_generic_factorial() + test_generic_accumulator() + test_generic_sizeof_bug() + return testcheck.end() diff --git a/Test/GenericTest2/output/bdfc8bd243b7e5b7.deps.json b/Test/GenericTest2/output/bdfc8bd243b7e5b7.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/GenericTest2/output/bdfc8bd243b7e5b7.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/GenericTest2/project.json b/Test/GenericTest2/project.json new file mode 100644 index 0000000..3e88e9a --- /dev/null +++ b/Test/GenericTest2/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "GenericTest2", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "GenericTest2.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/GenericTest2/temp/6f62fe05c5ea1ceb.pyi b/Test/GenericTest2/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/GenericTest2/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/GenericTest2/temp/7e529fe7a078cfef.pyi b/Test/GenericTest2/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/GenericTest2/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/GenericTest2/temp/_phase1_manifest.json b/Test/GenericTest2/temp/_phase1_manifest.json new file mode 100644 index 0000000..afb8eaf --- /dev/null +++ b/Test/GenericTest2/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\GenericTest2\\App\\main.py": {"sha1": "bdfc8bd243b7e5b7", "mtime": 1782479447.3739374, "size": 8453}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/GenericTest2/temp/_sha1_map.txt b/Test/GenericTest2/temp/_sha1_map.txt new file mode 100644 index 0000000..479fc36 --- /dev/null +++ b/Test/GenericTest2/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +bdfc8bd243b7e5b7:main.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/GenericTest2/temp/bbdf3bbd4c3bc28c.pyi b/Test/GenericTest2/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/GenericTest2/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/GenericTest2/temp/bdfc8bd243b7e5b7.doc.json b/Test/GenericTest2/temp/bdfc8bd243b7e5b7.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/GenericTest2/temp/bdfc8bd243b7e5b7.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/GenericTest2/temp/bdfc8bd243b7e5b7.pyi b/Test/GenericTest2/temp/bdfc8bd243b7e5b7.pyi new file mode 100644 index 0000000..bd232f2 --- /dev/null +++ b/Test/GenericTest2/temp/bdfc8bd243b7e5b7.pyi @@ -0,0 +1,87 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import c +import testcheck + +def square[T](x: T) -> T: pass + +def test_generic_square_types() -> t.CInt: pass + +def clamp[T](val: T, lo: T, hi: T) -> T: pass + +def test_generic_clamp() -> t.CInt: pass + + +class Stack[T]: + data: t.CArray[t.CInt, 16] + top: t.CInt + def __init__(self: Stack) -> t.CInt: pass + def push(self: Stack, val: T) -> t.CInt: pass + def pop(self: Stack) -> T: pass + def is_empty(self: Stack) -> t.CInt: pass + def size_of(self: Stack) -> t.CSizeT: pass + def type_size(self: Stack) -> t.CSizeT: pass + +def test_generic_stack() -> t.CInt: pass + + +class IntPair: + a: t.CInt + b: t.CInt + +def generic_swap[T](p: IntPair | t.CPtr) -> t.CInt: pass + +def test_generic_swap() -> t.CInt: pass + + +class Range[T]: + start: T + end_val: T + step: T + def __init__(self: Range, s: T, e: T, st: T) -> t.CInt: pass + def sum_all(self: Range) -> T: pass + +def test_generic_range() -> t.CInt: pass + +def array_sum(arr: t.CArray[t.CInt, 8], n: t.CInt) -> t.CInt: pass + +def test_generic_array_sum() -> t.CInt: pass + +def max_of_three[T](a: T, b: T, c_val: T) -> T: pass + +def test_generic_max_three() -> t.CInt: pass + + +class Container[T]: + values: t.CArray[t.CInt, 8] + size: t.CInt + def __init__(self: Container) -> t.CInt: pass + def add(self: Container, v: T) -> t.CInt: pass + def get(self: Container, idx: t.CInt) -> T: pass + def contains(self: Container, v: T) -> t.CInt: pass + def total(self: Container) -> T: pass + +def test_generic_container() -> t.CInt: pass + +def factorial[T](n: T) -> T: pass + +def test_generic_factorial() -> t.CInt: pass + + +class Accumulator[T]: + total: T + def __init__(self: Accumulator, start: T) -> t.CInt: pass + def add_val(self: Accumulator, v: T) -> t.CInt: pass + def get_result(self: Accumulator) -> T: pass + +def test_generic_accumulator() -> t.CInt: pass + +def test_generic_sizeof_bug() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/GenericTest2/temp/dd3002730623424b.pyi b/Test/GenericTest2/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/GenericTest2/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/GenericTest2/temp/f5522571bcce7bcb.pyi b/Test/GenericTest2/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/GenericTest2/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/HashTableTest/App/main.py b/Test/HashTableTest/App/main.py new file mode 100644 index 0000000..fe01ad1 --- /dev/null +++ b/Test/HashTableTest/App/main.py @@ -0,0 +1,206 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import hashtable +import testcheck + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("HashTableTest: 开放寻址哈希表测试") + + # 分配内存竞技场 + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + + # === Test 1: 基本插入和查找 === + testcheck.section("Test 1: 基本插入和查找 (set_int)") + ht: hashtable.HashTable = hashtable.HashTable(bd) + ht.set_int("alpha", 10) + ht.set_int("beta", 20) + ht.set_int("gamma", 30) + n: t.CSizeT = len(ht) + stdio.printf("len=%lu\n", n) + testcheck.check(n == 3, "ht len OK (3)", "ht len FAILED") + + p1: t.CPtr = ht["alpha"] + v1: int = c.Deref(p1) + p2: t.CPtr = ht["beta"] + v2: int = c.Deref(p2) + p3: t.CPtr = ht["gamma"] + v3: int = c.Deref(p3) + stdio.printf("alpha=%d beta=%d gamma=%d\n", v1, v2, v3) + testcheck.check(v1 == 10, "ht[alpha]==10 OK", "ht[alpha] FAILED") + testcheck.check(v2 == 20, "ht[beta]==20 OK", "ht[beta] FAILED") + testcheck.check(v3 == 30, "ht[gamma]==30 OK", "ht[gamma] FAILED") + + # === Test 2: 更新已有键 === + testcheck.section("Test 2: 更新已有键") + ht.set_int("beta", 99) + p2b: t.CPtr = ht["beta"] + v2b: int = c.Deref(p2b) + stdio.printf("beta(updated)=%d\n", v2b) + testcheck.check(v2b == 99, "ht update OK", "ht update FAILED") + testcheck.check(len(ht) == 3, "ht len still 3 OK", "ht len changed FAILED") + + # === Test 3: get with default === + testcheck.section("Test 3: get with default") + gv: t.CPtr = ht.get("missing", None) + testcheck.check(gv == None, "ht get missing OK (None)", "ht get missing FAILED") + gp: t.CPtr = ht.get("alpha", None) + ga: int = c.Deref(gp) + stdio.printf("get(alpha)=%d\n", ga) + testcheck.check(ga == 10, "ht get existing OK", "ht get existing FAILED") + + # === Test 4: contains === + testcheck.section("Test 4: contains (in)") + has_alpha: t.CInt = "alpha" in ht + has_missing: t.CInt = "missing" in ht + stdio.printf("contains(alpha)=%d contains(missing)=%d\n", has_alpha, has_missing) + testcheck.check(has_alpha == 1, "contains alpha OK", "contains alpha FAILED") + testcheck.check(has_missing == 0, "not contains missing OK", "contains missing FAILED") + + # === Test 5: 删除 === + testcheck.section("Test 5: 删除 (del)") + deleted: t.CInt = ht.__delitem__("beta") + stdio.printf("del beta=%d\n", deleted) + testcheck.check(deleted == 1, "del beta OK", "del beta FAILED") + testcheck.check(len(ht) == 2, "ht len 2 after del OK", "ht len after del FAILED") + has_beta: t.CInt = "beta" in ht + testcheck.check(has_beta == 0, "beta gone OK", "beta still exists FAILED") + + # 删除不存在的 key + del_missing: t.CInt = ht.__delitem__("nokey") + testcheck.check(del_missing == 0, "del missing returns 0 OK", "del missing FAILED") + + # === Test 6: 删除后重新插入 === + testcheck.section("Test 6: 删除后重新插入") + ht.set_int("beta", 555) + p2c: t.CPtr = ht["beta"] + v2c: int = c.Deref(p2c) + stdio.printf("beta(re-inserted)=%d\n", v2c) + testcheck.check(v2c == 555, "ht re-insert beta OK", "ht re-insert beta FAILED") + testcheck.check(len(ht) == 3, "ht len 3 after re-insert OK", "ht len after re-insert FAILED") + + # === Test 7: 迭代器 === + testcheck.section("Test 7: 迭代器 for key in ht") + iter_count: int = 0 + for key in ht: + kp: t.CPtr = ht[key] + val: int = c.Deref(kp) + stdio.printf("ht[%s]=%d\n", key, val) + iter_count += 1 + testcheck.check(iter_count == 3, "ht iter OK (3)", "ht iter FAILED") + + # === Test 8: 字符串值 === + testcheck.section("Test 8: 字符串值 (set_str)") + ht2: hashtable.HashTable = hashtable.HashTable(bd) + ht2.set_str("name", "Alice") + ht2.set_str("city", "Beijing") + ht2.set_str("lang", "TransPyC") + n2: t.CSizeT = len(ht2) + stdio.printf("ht2 len=%lu\n", n2) + testcheck.check(n2 == 3, "ht set_str len OK (3)", "ht set_str len FAILED") + + name_p: t.CPtr = ht2["name"] + city_p: t.CPtr = ht2["city"] + lang_p: t.CPtr = ht2["lang"] + name: str = name_p + city: str = city_p + lang: str = lang_p + stdio.printf("name=%s city=%s lang=%s\n", name, city, lang) + testcheck.check(name == "Alice", "ht str name OK", "ht str name FAILED") + testcheck.check(city == "Beijing", "ht str city OK", "ht str city FAILED") + testcheck.check(lang == "TransPyC", "ht str lang OK", "ht str lang FAILED") + + # === Test 9: 扩容测试 === + testcheck.section("Test 9: 扩容测试 (>16 个元素)") + ht3: hashtable.HashTable = hashtable.HashTable(bd) + i: int = 0 + while i < 50: + key_buf: str = stdlib.malloc(16) + stdio.sprintf(key_buf, "key%d", i) + ht3.set_int(key_buf, i * 100) + i += 1 + n3: t.CSizeT = len(ht3) + stdio.printf("ht3 len=%lu (after 50 inserts)\n", n3) + testcheck.check(n3 == 50, "ht expand OK (50)", "ht expand FAILED") + + # 验证几个值 + k5: str = stdlib.malloc(16) + stdio.sprintf(k5, "key5") + p5: t.CPtr = ht3[k5] + v5: int = c.Deref(p5) + stdio.printf("ht3[key5]=%d\n", v5) + testcheck.check(v5 == 500, "ht expand key5 OK", "ht expand key5 FAILED") + + k25: str = stdlib.malloc(16) + stdio.sprintf(k25, "key25") + p25: t.CPtr = ht3[k25] + v25: int = c.Deref(p25) + stdio.printf("ht3[key25]=%d\n", v25) + testcheck.check(v25 == 2500, "ht expand key25 OK", "ht expand key25 FAILED") + + k49: str = stdlib.malloc(16) + stdio.sprintf(k49, "key49") + p49: t.CPtr = ht3[k49] + v49: int = c.Deref(p49) + stdio.printf("ht3[key49]=%d\n", v49) + testcheck.check(v49 == 4900, "ht expand key49 OK", "ht expand key49 FAILED") + + # === Test 10: 扩容后迭代 === + testcheck.section("Test 10: 扩容后迭代") + iter_count3: int = 0 + for k in ht3: + iter_count3 += 1 + stdio.printf("ht3 iter count=%d\n", iter_count3) + testcheck.check(iter_count3 == 50, "ht expand iter OK (50)", "ht expand iter FAILED") + + # === Test 11: 删除后扩容 === + testcheck.section("Test 11: 批量删除后操作") + j: int = 0 + while j < 25: + del_buf: str = stdlib.malloc(16) + stdio.sprintf(del_buf, "key%d", j) + del ht3[del_buf] + j += 1 + n3b: t.CSizeT = len(ht3) + stdio.printf("ht3 len=%lu (after 25 deletes)\n", n3b) + testcheck.check(n3b == 25, "ht after del OK (25)", "ht after del FAILED") + + # 验证剩余的值 + k30: str = stdlib.malloc(16) + stdio.sprintf(k30, "key30") + p30: t.CPtr = ht3[k30] + v30: int = c.Deref(p30) + stdio.printf("ht3[key30]=%d\n", v30) + testcheck.check(v30 == 3000, "ht key30 after del OK", "ht key30 after del FAILED") + + # 验证已删除的 key 不存在 + k0: str = stdlib.malloc(16) + stdio.sprintf(k0, "key0") + has_k0: t.CInt = k0 in ht3 + testcheck.check(has_k0 == 0, "ht key0 deleted OK", "ht key0 still exists FAILED") + + # === Test 12: 空表操作 === + testcheck.section("Test 12: 空表操作") + ht4: hashtable.HashTable = hashtable.HashTable(bd) + testcheck.check(len(ht4) == 0, "empty ht len 0 OK", "empty ht len FAILED") + empty_get: t.CPtr = ht4.get("none", None) + testcheck.check(empty_get == None, "empty ht get OK", "empty ht get FAILED") + has_none: t.CInt = "none" in ht4 + testcheck.check(has_none == 0, "empty ht contains OK", "empty ht contains FAILED") + empty_iter: int = 0 + for ek in ht4: + empty_iter += 1 + testcheck.check(empty_iter == 0, "empty ht iter OK (0)", "empty ht iter FAILED") + + stdlib.free(arena) + return testcheck.end() diff --git a/Test/HashTableTest/output/02d833b3cf571c5b.deps.json b/Test/HashTableTest/output/02d833b3cf571c5b.deps.json new file mode 100644 index 0000000..f1d4415 --- /dev/null +++ b/Test/HashTableTest/output/02d833b3cf571c5b.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/HashTableTest/project.json b/Test/HashTableTest/project.json new file mode 100644 index 0000000..0762d71 --- /dev/null +++ b/Test/HashTableTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "HashTableTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32"], + "output": "HashTableTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/HashTableTest/temp/0035c95a18d4f8e8.pyi b/Test/HashTableTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/HashTableTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/HashTableTest/temp/02d833b3cf571c5b.doc.json b/Test/HashTableTest/temp/02d833b3cf571c5b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/HashTableTest/temp/02d833b3cf571c5b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/HashTableTest/temp/02d833b3cf571c5b.pyi b/Test/HashTableTest/temp/02d833b3cf571c5b.pyi new file mode 100644 index 0000000..33275d4 --- /dev/null +++ b/Test/HashTableTest/temp/02d833b3cf571c5b.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import memhub +import hashtable +import testcheck + +def main() -> CInt | CExport: pass diff --git a/Test/HashTableTest/temp/067c78e9f121dce3.pyi b/Test/HashTableTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/HashTableTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/HashTableTest/temp/06f53cc594b4ac6c.pyi b/Test/HashTableTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/HashTableTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/HashTableTest/temp/271ea3decb810db2.pyi b/Test/HashTableTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/HashTableTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/HashTableTest/temp/6446627d4f07a1b5.pyi b/Test/HashTableTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/HashTableTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/HashTableTest/temp/6f62fe05c5ea1ceb.pyi b/Test/HashTableTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/HashTableTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/HashTableTest/temp/72e2d5ccb7cedcf1.pyi b/Test/HashTableTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/HashTableTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/HashTableTest/temp/7e529fe7a078cfef.pyi b/Test/HashTableTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/HashTableTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/HashTableTest/temp/90c53dd6db8d41cf.pyi b/Test/HashTableTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/HashTableTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/HashTableTest/temp/_phase1_manifest.json b/Test/HashTableTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..4c20a5f --- /dev/null +++ b/Test/HashTableTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\HashTableTest\\App\\main.py": {"sha1": "02d833b3cf571c5b", "mtime": 1782828002.1143503, "size": 7795}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashtable.py": {"sha1": "b8c66c8ff44eb874", "mtime": 1782826619.490043, "size": 9385}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/HashTableTest/temp/_sha1_map.txt b/Test/HashTableTest/temp/_sha1_map.txt new file mode 100644 index 0000000..cde28e2 --- /dev/null +++ b/Test/HashTableTest/temp/_sha1_map.txt @@ -0,0 +1,12 @@ +02d833b3cf571c5b:main.py +271ea3decb810db2:includes/atom.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +b8c66c8ff44eb874:includes/hashtable.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/HashTableTest/temp/ab6e54ba9a669f76.pyi b/Test/HashTableTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/HashTableTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/HashTableTest/temp/b8c66c8ff44eb874.pyi b/Test/HashTableTest/temp/b8c66c8ff44eb874.pyi new file mode 100644 index 0000000..85bdb8f --- /dev/null +++ b/Test/HashTableTest/temp/b8c66c8ff44eb874.pyi @@ -0,0 +1,46 @@ +""" +Auto-generated Python stub file from hashtable.py +Module: hashtable +""" + + +import t, c +import string +import memhub +from stdint import * + +_HT_EMPTY: t.CDefine = 0 +_HT_TOMBSTONE: t.CDefine = 1 +_HT_INIT_CAP: t.CDefine = 16 +_HT_SLOT_SIZE: t.CDefine = 24 +_FNV_OFFSET: t.CExtern | t.CUInt64T +_FNV_PRIME: t.CExtern | t.CUInt64T + +def _fnv1a_hash(key: str) -> t.CUInt64T: pass + + +class HashTable: + __slots__: t.CVoid | t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 16 + __tombstones__: t.CSizeT = 0 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def _slot_hash(self: HashTable, idx: t.CSizeT) -> t.CUInt64T: pass + def _slot_key(self: HashTable, idx: t.CSizeT) -> str: pass + def _slot_value(self: HashTable, idx: t.CSizeT) -> t.CPtr: pass + def _set_slot(self: HashTable, idx: t.CSizeT, h: t.CUInt64T, key: str, val: t.CPtr) -> t.CInt: pass + def _find_slot(self: HashTable, key: str, h: t.CUInt64T) -> t.CSizeT: pass + def _resize(self: HashTable, new_cap: t.CSizeT) -> t.CInt: pass + def __setitem__(self: HashTable, key: str, val: t.CNeedPtr) -> t.CInt: pass + def __getitem__(self: HashTable, key: str) -> t.CPtr: pass + def __contains__(self: HashTable, key: str) -> t.CInt: pass + def __delitem__(self: HashTable, key: str) -> t.CInt: pass + def __len__(self: HashTable) -> t.CSizeT: pass + def __iter__(self: HashTable) -> HashTable | t.CPtr: pass + def __next__(self: HashTable) -> str: pass + def get(self: HashTable, key: str, default: t.CPtr) -> t.CPtr: pass + def set_int(self: HashTable, key: str, val: int) -> t.CInt: pass + def set_str(self: HashTable, key: str, val: str) -> t.CInt: pass \ No newline at end of file diff --git a/Test/HashTableTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/HashTableTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/HashTableTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/HashTableTest/temp/c9f4be41ca1cc2b4.pyi b/Test/HashTableTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/HashTableTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/HashTableTest/temp/dd3002730623424b.pyi b/Test/HashTableTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/HashTableTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/HashTableTest/temp/ee084e9fc6ee413a.pyi b/Test/HashTableTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/HashTableTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/HashTableTest/temp/f5522571bcce7bcb.pyi b/Test/HashTableTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/HashTableTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/HashTableTest/temp/f6b51804a0ba8ff0.pyi b/Test/HashTableTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/HashTableTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/IterTest/App/main.py b/Test/IterTest/App/main.py new file mode 100644 index 0000000..7ff01a7 --- /dev/null +++ b/Test/IterTest/App/main.py @@ -0,0 +1,51 @@ +import t +import stdio +import string +import testcheck + +def test_for_in_string(): + testcheck.section("Test 1: for-in string iteration") + s: t.CChar | t.CPtr = "Hello" + result: t.CInt = 0 + for ch in s: + result += ch + # H=72, e=101, l=108, l=108, o=111 => 500 + testcheck.check(result == 500, "for-in string (sum=500)", "for-in string (sum expect 500)") + +def test_for_in_char_ptr(): + testcheck.section("Test 2: for-in char* iteration") + s: t.CChar | t.CPtr = "ABC" + chars: t.CArray[t.CInt, 4] = [0] + idx: t.CInt = 0 + for ch in s: + if idx < 3: + chars[idx] = ch + idx += 1 + testcheck.check(chars[0] == 65 and chars[1] == 66 and chars[2] == 67, "for-in char* (A=65 B=66 C=67)", "for-in char* (expect A=65 B=66 C=67)") + +def test_strcpy_for_in(): + testcheck.section("Test 3: strcpy using for-in") + src: t.CChar | t.CPtr = "test" + # 手动实现 strcpy 用 for-in + import w32.win32memory + buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 64) + # 用 for-in 复制 + p: t.CChar | t.CPtr = buf + for ch in src: + p[0] = ch + p += 1 + p[0] = 0 + testcheck.check(buf == "test", "strcpy for-in (result=test)", "strcpy for-in (cmp mismatch)") + +import w32.win32memory + +def main() -> t.CInt: + testcheck.begin("IterTest: 指针迭代测试") + test_for_in_string() + test_for_in_char_ptr() + test_strcpy_for_in() + return testcheck.end() diff --git a/Test/IterTest/output/6b8853854d6ef556.deps.json b/Test/IterTest/output/6b8853854d6ef556.deps.json new file mode 100644 index 0000000..9da11e2 --- /dev/null +++ b/Test/IterTest/output/6b8853854d6ef556.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/IterTest/project.json b/Test/IterTest/project.json new file mode 100644 index 0000000..95a6a54 --- /dev/null +++ b/Test/IterTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "IterTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "IterTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/IterTest/temp/0035c95a18d4f8e8.pyi b/Test/IterTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/IterTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/IterTest/temp/067c78e9f121dce3.pyi b/Test/IterTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/IterTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/IterTest/temp/06f53cc594b4ac6c.pyi b/Test/IterTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/IterTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/IterTest/temp/6446627d4f07a1b5.pyi b/Test/IterTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/IterTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/IterTest/temp/6b8853854d6ef556.doc.json b/Test/IterTest/temp/6b8853854d6ef556.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/IterTest/temp/6b8853854d6ef556.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/IterTest/temp/6b8853854d6ef556.pyi b/Test/IterTest/temp/6b8853854d6ef556.pyi new file mode 100644 index 0000000..66d8d02 --- /dev/null +++ b/Test/IterTest/temp/6b8853854d6ef556.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +def test_for_in_string() -> t.CInt: pass + +def test_for_in_char_ptr() -> t.CInt: pass + +def test_strcpy_for_in() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/Test/IterTest/temp/6f62fe05c5ea1ceb.pyi b/Test/IterTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/IterTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/IterTest/temp/72e2d5ccb7cedcf1.pyi b/Test/IterTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/IterTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/IterTest/temp/7e529fe7a078cfef.pyi b/Test/IterTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/IterTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/IterTest/temp/_phase1_manifest.json b/Test/IterTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..7dc1091 --- /dev/null +++ b/Test/IterTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\IterTest\\App\\main.py": {"sha1": "6b8853854d6ef556", "mtime": 1782474645.4172094, "size": 1600}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/IterTest/temp/_sha1_map.txt b/Test/IterTest/temp/_sha1_map.txt new file mode 100644 index 0000000..53a3934 --- /dev/null +++ b/Test/IterTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +6b8853854d6ef556:main.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/IterTest/temp/ab6e54ba9a669f76.pyi b/Test/IterTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/IterTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/IterTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/IterTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/IterTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/IterTest/temp/dd3002730623424b.pyi b/Test/IterTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/IterTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/IterTest/temp/f5522571bcce7bcb.pyi b/Test/IterTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/IterTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/IterTest/temp/f6b51804a0ba8ff0.pyi b/Test/IterTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/IterTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/JsonTest/App/main.py b/Test/JsonTest/App/main.py new file mode 100644 index 0000000..b396c7b --- /dev/null +++ b/Test/JsonTest/App/main.py @@ -0,0 +1,309 @@ +import t, c +from stdint import * +from stdio import printf +import memhub +import json +import string +import w32.win32memory as win32mem +import testcheck + +# ============================================================ +# 测试 1：构造 JSON 值（使用 mpool） +# ============================================================ +def test_construct(pool: memhub.MemPool | t.CPtr): + testcheck.section("Construct JsonValue (with mpool)") + + # null + v1: json.JsonValue | t.CPtr = json.null(pool) + testcheck.check(v1.is_null(), "null is_null", "null is_null") + + # bool + v2: json.JsonValue | t.CPtr = json.bool_val(pool, True) + testcheck.check(v2.is_bool(), "bool true is_bool", "bool true is_bool") + testcheck.check(v2.as_bool(), "bool true as_bool", "bool true as_bool") + + v3: json.JsonValue | t.CPtr = json.bool_val(pool, False) + testcheck.check(not v3.as_bool(), "bool false !as_bool", "bool false !as_bool") + + # int + v4: json.JsonValue | t.CPtr = json.int_val(pool, 42) + testcheck.check(v4.is_int(), "int is_int", "int is_int") + testcheck.check(v4.as_int() == 42, "int 42 == 42", "int 42 == 42") + + v5: json.JsonValue | t.CPtr = json.int_val(pool, -100) + testcheck.check(v5.as_int() == -100, "int -100 == -100", "int -100 == -100") + + # float + v6: json.JsonValue | t.CPtr = json.float_val(pool, 3.14) + testcheck.check(v6.is_float(), "float is_float", "float is_float") + testcheck.check(v6.as_float() > 3.13 and v6.as_float() < 3.15, "float 3.14 ~= 3.14", "float 3.14 ~= 3.14") + + # string + v7: json.JsonValue | t.CPtr = json.string_val(pool, "hello") + testcheck.check(v7.is_string(), "string is_string", "string is_string") + testcheck.check(v7.as_string() == "hello", "string == hello", "string == hello") + + # array + v8: json.JsonValue | t.CPtr = json.array(pool) + testcheck.check(v8.is_array(), "array is_array", "array is_array") + testcheck.check(len(v8) == 0, "array empty len()==0", "array empty len()==0") + + # object + v9: json.JsonValue | t.CPtr = json.object(pool) + testcheck.check(v9.is_object(), "object is_object", "object is_object") + testcheck.check(len(v9) == 0, "object empty len()==0", "object empty len()==0") + + +# ============================================================ +# 测试 2：构造数组和对象（Python风格API） +# ============================================================ +def test_build_structures(pool: memhub.MemPool | t.CPtr): + testcheck.section("Build Arrays and Objects (Pythonic API)") + + # 构建数组 [1, 2, 3] + arr: json.JsonValue | t.CPtr = json.array(pool) + json.array_append(pool, arr, json.int_val(pool, 1)) + json.array_append(pool, arr, json.int_val(pool, 2)) + json.array_append(pool, arr, json.int_val(pool, 3)) + testcheck.check(len(arr) == 3, "array [1,2,3] len()==3", "array [1,2,3] len()==3") + testcheck.check(arr.get_item(0).as_int() == 1, "array get_item(0)==1", "array get_item(0)==1") + testcheck.check(arr.get_item(1).as_int() == 2, "array get_item(1)==2", "array get_item(1)==2") + testcheck.check(arr.get_item(2).as_int() == 3, "array get_item(2)==3", "array get_item(2)==3") + + # 构建对象 {"name": "Viper", "version": 1} + obj: json.JsonValue | t.CPtr = json.object(pool) + json.object_set(pool, obj, "name", json.string_val(pool, "Viper")) + json.object_set(pool, obj, "version", json.int_val(pool, 1)) + testcheck.check(len(obj) == 2, "object len()==2", "object len()==2") + + # Python风格取值：obj["key"] + name_val: json.JsonValue | t.CPtr = obj["name"] + testcheck.check(name_val != None and name_val.as_string() == "Viper", "obj['name']==Viper", "obj['name']==Viper") + ver_val: json.JsonValue | t.CPtr = obj["version"] + testcheck.check(ver_val != None and ver_val.as_int() == 1, "obj['version']==1", "obj['version']==1") + missing: json.JsonValue | t.CPtr = obj["missing"] + testcheck.check(missing == None, "obj['missing']==None", "obj['missing']==None") + + +# ============================================================ +# 测试 3：JSON 解析 +# ============================================================ +def test_parse(pool: memhub.MemPool | t.CPtr): + testcheck.section("Parse JSON") + + # 解析整数 + v1: json.JsonValue | t.CPtr = json.parse(pool, "42") + testcheck.check(v1.is_int(), "parse int is_int", "parse int is_int") + testcheck.check(v1.as_int() == 42, "parse int == 42", "parse int == 42") + + # 解析负数 + v2: json.JsonValue | t.CPtr = json.parse(pool, "-7") + testcheck.check(v2.as_int() == -7, "parse neg int == -7", "parse neg int == -7") + + # 解析浮点数 + v3: json.JsonValue | t.CPtr = json.parse(pool, "3.14") + testcheck.check(v3.is_float(), "parse float is_float", "parse float is_float") + testcheck.check(v3.as_float() > 3.13 and v3.as_float() < 3.15, "parse float ~= 3.14", "parse float ~= 3.14") + + # 解析字符串 + v4: json.JsonValue | t.CPtr = json.parse(pool, '"hello world"') + testcheck.check(v4.is_string(), "parse string is_string", "parse string is_string") + testcheck.check(v4.as_string() == "hello world", "parse string == hello world", "parse string == hello world") + + # 解析布尔 + v5: json.JsonValue | t.CPtr = json.parse(pool, "true") + testcheck.check(v5.is_bool(), "parse true is_bool", "parse true is_bool") + testcheck.check(v5.as_bool(), "parse true as_bool", "parse true as_bool") + + v6: json.JsonValue | t.CPtr = json.parse(pool, "false") + testcheck.check(not v6.as_bool(), "parse false !as_bool", "parse false !as_bool") + + # 解析 null + v7: json.JsonValue | t.CPtr = json.parse(pool, "null") + testcheck.check(v7.is_null(), "parse null is_null", "parse null is_null") + + # 解析数组 + v8: json.JsonValue | t.CPtr = json.parse(pool, "[1, 2, 3]") + testcheck.check(v8.is_array(), "parse array is_array", "parse array is_array") + testcheck.check(len(v8) == 3, "parse array len()==3", "parse array len()==3") + testcheck.check(v8.get_item(0).as_int() == 1, "parse array get_item(0)==1", "parse array get_item(0)==1") + testcheck.check(v8.get_item(2).as_int() == 3, "parse array get_item(2)==3", "parse array get_item(2)==3") + + # 解析对象（Python风格取值） + v9: json.JsonValue | t.CPtr = json.parse(pool, '{"x": 10, "y": 20}') + testcheck.check(v9.is_object(), "parse object is_object", "parse object is_object") + testcheck.check(len(v9) == 2, "parse object len()==2", "parse object len()==2") + x_val: json.JsonValue | t.CPtr = v9["x"] + testcheck.check(x_val != None and x_val.as_int() == 10, "parse obj['x']==10", "parse obj['x']==10") + y_val: json.JsonValue | t.CPtr = v9["y"] + testcheck.check(y_val != None and y_val.as_int() == 20, "parse obj['y']==20", "parse obj['y']==20") + + +# ============================================================ +# 测试 4：JSON 序列化 +# ============================================================ +def test_write(pool: memhub.MemPool | t.CPtr): + testcheck.section("Write JSON") + + # 写整数 + v1: json.JsonValue | t.CPtr = json.int_val(pool, 42) + s1: t.CChar | t.CPtr = json.write(pool, v1, False) + testcheck.check(s1 == "42", "write int 42", "write int 42") + + # 写字符串 + v2: json.JsonValue | t.CPtr = json.string_val(pool, "hello") + s2: t.CChar | t.CPtr = json.write(pool, v2, False) + testcheck.check(s2 == '"hello"', "write string hello", "write string hello") + + # 写布尔 + v3: json.JsonValue | t.CPtr = json.bool_val(pool, True) + s3: t.CChar | t.CPtr = json.write(pool, v3, False) + testcheck.check(s3 == "true", "write bool true", "write bool true") + + v4: json.JsonValue | t.CPtr = json.bool_val(pool, False) + s4: t.CChar | t.CPtr = json.write(pool, v4, False) + testcheck.check(s4 == "false", "write bool false", "write bool false") + + # 写 null + v5: json.JsonValue | t.CPtr = json.null(pool) + s5: t.CChar | t.CPtr = json.write(pool, v5, False) + testcheck.check(s5 == "null", "write null", "write null") + + # 写数组 + arr: json.JsonValue | t.CPtr = json.array(pool) + json.array_append(pool, arr, json.int_val(pool, 1)) + json.array_append(pool, arr, json.int_val(pool, 2)) + json.array_append(pool, arr, json.int_val(pool, 3)) + s6: t.CChar | t.CPtr = json.write(pool, arr, False) + testcheck.check(s6 == "[1,2,3]", "write array [1,2,3]", "write array [1,2,3]") + + # 写对象 + obj: json.JsonValue | t.CPtr = json.object(pool) + json.object_set(pool, obj, "a", json.int_val(pool, 1)) + json.object_set(pool, obj, "b", json.string_val(pool, "x")) + s7: t.CChar | t.CPtr = json.write(pool, obj, False) + testcheck.check(s7 != None, "write object non-null", "write object non-null") + + +# ============================================================ +# 测试 5：解析复杂嵌套 JSON（Python风格API） +# ============================================================ +def test_nested(pool: memhub.MemPool | t.CPtr): + testcheck.section("Parse Nested JSON (Pythonic API)") + + src: t.CChar | t.CPtr = '{"name": "Viper", "version": 1, "features": ["fast", "safe", "portable"], "nested": {"key": "value", "num": 99}}' + root: json.JsonValue | t.CPtr = json.parse(pool, src) + testcheck.check(root.is_object(), "nested is_object", "nested is_object") + testcheck.check(len(root) == 4, "nested len()==4", "nested len()==4") + + # Python风格取值：root["name"] + name: json.JsonValue | t.CPtr = root["name"] + testcheck.check(name != None and name.as_string() == "Viper", "root['name']==Viper", "root['name']==Viper") + + # root["version"] + ver: json.JsonValue | t.CPtr = root["version"] + testcheck.check(ver != None and ver.as_int() == 1, "root['version']==1", "root['version']==1") + + # root["features"] 数组 + feat: json.JsonValue | t.CPtr = root["features"] + testcheck.check(feat != None and feat.is_array(), "root['features'] is_array", "root['features'] is_array") + testcheck.check(feat != None and len(feat) == 3, "root['features'] len()==3", "root['features'] len()==3") + testcheck.check(feat != None and feat.get_item(0).as_string() == "fast", "root['features'][0]==fast", "root['features'][0]==fast") + testcheck.check(feat != None and feat.get_item(2).as_string() == "portable", "root['features'][2]==portable", "root['features'][2]==portable") + + # root["nested"] 对象 + nested: json.JsonValue | t.CPtr = root["nested"] + testcheck.check(nested != None and nested.is_object(), "root['nested'] is_object", "root['nested'] is_object") + nk: json.JsonValue | t.CPtr = nested["key"] if nested != None else None + testcheck.check(nk != None and nk.as_string() == "value", "root['nested']['key']==value", "root['nested']['key']==value") + nn: json.JsonValue | t.CPtr = nested["num"] if nested != None else None + testcheck.check(nn != None and nn.as_int() == 99, "root['nested']['num']==99", "root['nested']['num']==99") + + +# ============================================================ +# 测试 6：mpool 重置后重新使用 +# ============================================================ +def test_mpool_reuse(): + testcheck.section("MPool Reuse") + + # 分配内存 + mem: t.CVoid | t.CPtr = win32mem.VirtualAlloc(t.CVoid(0, t.CPtr), 8192, 0x3000, 0x04) + testcheck.check(mem != None, "VirtualAlloc ok", "VirtualAlloc ok") + + pool: memhub.MemPool | t.CPtr = memhub.MemPool(mem, 8192) + + # 第一次使用 + v1: json.JsonValue | t.CPtr = json.int_val(pool, 100) + testcheck.check(v1.as_int() == 100, "reuse round1 int==100", "reuse round1 int==100") + + # 重置 mpool + pool.reset() + + # 第二次使用（内存复用） + v2: json.JsonValue | t.CPtr = json.string_val(pool, "reused") + testcheck.check(v2.as_string() == "reused", "reuse round2 string==reused", "reuse round2 string==reused") + + # 重置后再解析 + pool.reset() + v3: json.JsonValue | t.CPtr = json.parse(pool, '{"test": true}') + testcheck.check(v3.is_object(), "reuse round3 parse ok", "reuse round3 parse ok") + tv: json.JsonValue | t.CPtr = v3["test"] + testcheck.check(tv != None and tv.as_bool(), "reuse round3 obj['test']==true", "reuse round3 obj['test']==true") + + win32mem.VirtualFree(mem, 0, 0x8000) + + +# ============================================================ +# 测试 7：转义字符解析 +# ============================================================ +def test_escape(pool: memhub.MemPool | t.CPtr): + testcheck.section("Escape Characters") + + # 解析含转义字符的字符串 + v1: json.JsonValue | t.CPtr = json.parse(pool, '"hello\\nworld"') + testcheck.check(v1.is_string(), "escape \\n is_string", "escape \\n is_string") + sv1: t.CChar | t.CPtr = v1.as_string() + testcheck.check(sv1 != None and sv1[5] == '\n', "escape \\n contains newline", "escape \\n contains newline") + + v2: json.JsonValue | t.CPtr = json.parse(pool, '"tab\\there"') + testcheck.check(v2.is_string(), "escape \\t is_string", "escape \\t is_string") + sv2: t.CChar | t.CPtr = v2.as_string() + testcheck.check(sv2 != None and sv2[3] == '\t', "escape \\t contains tab", "escape \\t contains tab") + + v3: json.JsonValue | t.CPtr = json.parse(pool, '"quote\\""') + testcheck.check(v3.is_string(), "escape \\\" is_string", "escape \\\" is_string") + + +# ============================================================ +# 主函数 +# ============================================================ +def main() -> t.CInt | t.CExport: + testcheck.begin("JSON Library Test (Pythonic API)") + + # 使用 mpool 的 bump 分配器 + mem: t.CVoid | t.CPtr = win32mem.VirtualAlloc(t.CVoid(0, t.CPtr), 65536, 0x3000, 0x04) + pool: memhub.MemPool | t.CPtr = memhub.MemPool(mem, 65536) + + test_construct(pool) + pool.reset() + + test_build_structures(pool) + pool.reset() + + test_parse(pool) + pool.reset() + + test_write(pool) + pool.reset() + + test_nested(pool) + pool.reset() + + test_escape(pool) + pool.reset() + + test_mpool_reuse() + + win32mem.VirtualFree(mem, 0, 0x8000) + + return testcheck.end() diff --git a/Test/JsonTest/output/8a97754e9e3c2b9a.deps.json b/Test/JsonTest/output/8a97754e9e3c2b9a.deps.json new file mode 100644 index 0000000..76cbe79 --- /dev/null +++ b/Test/JsonTest/output/8a97754e9e3c2b9a.deps.json @@ -0,0 +1 @@ +{"json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "json.__init__": "57288496f7c2d1ad", "__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/JsonTest/project.json b/Test/JsonTest/project.json new file mode 100644 index 0000000..0b4c8a1 --- /dev/null +++ b/Test/JsonTest/project.json @@ -0,0 +1,28 @@ +{ + "name": "JsonTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "JsonTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/JsonTest/temp/0035c95a18d4f8e8.pyi b/Test/JsonTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/JsonTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/JsonTest/temp/067c78e9f121dce3.pyi b/Test/JsonTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/JsonTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/JsonTest/temp/06f53cc594b4ac6c.pyi b/Test/JsonTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/JsonTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/JsonTest/temp/20cd49775c100a38.pyi b/Test/JsonTest/temp/20cd49775c100a38.pyi new file mode 100644 index 0000000..df04de7 --- /dev/null +++ b/Test/JsonTest/temp/20cd49775c100a38.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from json.__writer.py +Module: json.__writer +""" + + +import t, c +from stdint import * +import memhub +import string +from stdio import snprintf +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT + +def _write_char(buf: t.CChar | t.CPtr, pos: t.CSizeT, ch: t.CChar) -> t.CSizeT: pass + +def _write_str(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_string_escaped(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_value(buf: t.CChar | t.CPtr, pos: t.CSizeT, val: JsonValue | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CSizeT: pass + +def write(pool: memhub.MemManager | t.CPtr, val: JsonValue | t.CPtr, pretty: bool) -> t.CChar | t.CPtr: pass diff --git a/Test/JsonTest/temp/240a9a4157959a9f.pyi b/Test/JsonTest/temp/240a9a4157959a9f.pyi new file mode 100644 index 0000000..c230800 --- /dev/null +++ b/Test/JsonTest/temp/240a9a4157959a9f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from json.__parser.py +Module: json.__parser +""" + + +import t, c +from stdint import * +import memhub +import string +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT, null, bool_val, int_val, float_val, string_val, array, object + +class ParseState: + src: t.CChar | t.CPtr + pos: t.CSizeT + len: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: ParseState, src: t.CChar | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(s: ParseState | t.CPtr) -> t.CChar: pass + +def _advance(s: ParseState | t.CPtr) -> t.CChar: pass + +def _skip_whitespace(s: ParseState | t.CPtr) -> t.CInt: pass + +def _expect(s: ParseState | t.CPtr, ch: t.CChar) -> bool: pass + +def _parse_string(s: ParseState | t.CPtr) -> t.CChar | t.CPtr: pass + +def _parse_number(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_value(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_array(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_object(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def parse(pool: memhub.MemManager | t.CPtr, src: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass diff --git a/Test/JsonTest/temp/271ea3decb810db2.pyi b/Test/JsonTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/JsonTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/JsonTest/temp/57288496f7c2d1ad.pyi b/Test/JsonTest/temp/57288496f7c2d1ad.pyi new file mode 100644 index 0000000..0624d9a --- /dev/null +++ b/Test/JsonTest/temp/57288496f7c2d1ad.pyi @@ -0,0 +1,71 @@ +""" +Auto-generated Python stub file from json.__init__.py +Module: json.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperio + +JSON_NULL: t.CDefine = 0 +JSON_BOOL: t.CDefine = 1 +JSON_INT: t.CDefine = 2 +JSON_FLOAT: t.CDefine = 3 +JSON_STRING: t.CDefine = 4 +JSON_ARRAY: t.CDefine = 5 +JSON_OBJECT: t.CDefine = 6 + +class JsonValue: + vtype: t.CInt + bool_val: t.CInt + int_val: t.CInt64T + float_val: t.CDouble + str_val: t.CChar | t.CPtr + next: 'JsonValue' | t.CPtr + key: t.CChar | t.CPtr + child: 'JsonValue' | t.CPtr + child_count: t.CSizeT + def __new__(self: JsonValue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def type(self: JsonValue) -> t.CInt: pass + def is_null(self: JsonValue) -> bool: pass + def is_bool(self: JsonValue) -> bool: pass + def is_int(self: JsonValue) -> bool: pass + def is_float(self: JsonValue) -> bool: pass + def is_string(self: JsonValue) -> bool: pass + def is_array(self: JsonValue) -> bool: pass + def is_object(self: JsonValue) -> bool: pass + def as_bool(self: JsonValue) -> bool: pass + def as_int(self: JsonValue) -> t.CInt64T: pass + def as_float(self: JsonValue) -> t.CDouble: pass + def as_string(self: JsonValue) -> t.CChar | t.CPtr: pass + def __len__(self: JsonValue) -> t.CSizeT: pass + def __getitem__(self: JsonValue, key: t.CChar | t.CPtr) -> 'JsonValue' | t.CPtr: pass + def __setitem__(self: JsonValue, key: t.CChar | t.CPtr, val: 'JsonValue' | t.CPtr) -> t.CInt: pass + def get_item(self: JsonValue, index: t.CSizeT) -> 'JsonValue' | t.CPtr: pass + def _pool_alloc(self: JsonValue, size: t.CSizeT) -> t.CChar | t.CPtr: pass + def _append_child(self: JsonValue, child: 'JsonValue' | t.CPtr) -> t.CInt: pass + +def null(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def bool_val(pool: memhub.MemManager | t.CPtr, val: bool) -> JsonValue | t.CPtr: pass + +def int_val(pool: memhub.MemManager | t.CPtr, val: t.CInt64T) -> JsonValue | t.CPtr: pass + +def float_val(pool: memhub.MemManager | t.CPtr, val: t.CDouble) -> JsonValue | t.CPtr: pass + +def string_val(pool: memhub.MemManager | t.CPtr, val: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def object(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def array_append(pool: memhub.MemManager | t.CPtr, arr: JsonValue | t.CPtr, item: JsonValue | t.CPtr) -> t.CInt: pass + +def object_set(pool: memhub.MemManager | t.CPtr, obj: JsonValue | t.CPtr, key: t.CChar | t.CPtr, val: JsonValue | t.CPtr) -> t.CInt: pass + + +from .__parser import parse +from .__writer import write \ No newline at end of file diff --git a/Test/JsonTest/temp/6446627d4f07a1b5.pyi b/Test/JsonTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/JsonTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/JsonTest/temp/6f62fe05c5ea1ceb.pyi b/Test/JsonTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/JsonTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/JsonTest/temp/72e2d5ccb7cedcf1.pyi b/Test/JsonTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/JsonTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/JsonTest/temp/7e529fe7a078cfef.pyi b/Test/JsonTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/JsonTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/JsonTest/temp/8a97754e9e3c2b9a.doc.json b/Test/JsonTest/temp/8a97754e9e3c2b9a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/JsonTest/temp/8a97754e9e3c2b9a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/JsonTest/temp/8a97754e9e3c2b9a.pyi b/Test/JsonTest/temp/8a97754e9e3c2b9a.pyi new file mode 100644 index 0000000..c3e417c --- /dev/null +++ b/Test/JsonTest/temp/8a97754e9e3c2b9a.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +from stdio import printf +import memhub +import json +import string +import w32.win32memory as win32mem +import testcheck + +def test_construct(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_structures(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_parse(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_write(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_nested(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_mpool_reuse() -> t.CInt: pass + +def test_escape(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/JsonTest/temp/_phase1_manifest.json b/Test/JsonTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..97bdd03 --- /dev/null +++ b/Test/JsonTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\JsonTest\\App\\main.py": {"sha1": "8a97754e9e3c2b9a", "mtime": 1782828053.1672907, "size": 14025}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__init__.py": {"sha1": "57288496f7c2d1ad", "mtime": 1783239811.8873823, "size": 7653}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__parser.py": {"sha1": "240a9a4157959a9f", "mtime": 1782826918.100951, "size": 7517}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__writer.py": {"sha1": "20cd49775c100a38", "mtime": 1782826901.8906734, "size": 3784}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/JsonTest/temp/_sha1_map.txt b/Test/JsonTest/temp/_sha1_map.txt new file mode 100644 index 0000000..9621830 --- /dev/null +++ b/Test/JsonTest/temp/_sha1_map.txt @@ -0,0 +1,14 @@ +20cd49775c100a38:includes/json\__writer.py +240a9a4157959a9f:includes/json\__parser.py +271ea3decb810db2:includes/atom.py +57288496f7c2d1ad:includes/json\__init__.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +8a97754e9e3c2b9a:main.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/JsonTest/temp/ab6e54ba9a669f76.pyi b/Test/JsonTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/JsonTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/JsonTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/JsonTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/JsonTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/JsonTest/temp/c9f4be41ca1cc2b4.pyi b/Test/JsonTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/JsonTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/JsonTest/temp/dd3002730623424b.pyi b/Test/JsonTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/JsonTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/JsonTest/temp/ee084e9fc6ee413a.pyi b/Test/JsonTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/JsonTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/JsonTest/temp/f5522571bcce7bcb.pyi b/Test/JsonTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/JsonTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/JsonTest/temp/f6b51804a0ba8ff0.pyi b/Test/JsonTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/JsonTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/LLvmLiteTest/App/main.py b/Test/LLvmLiteTest/App/main.py new file mode 100644 index 0000000..274172a --- /dev/null +++ b/Test/LLvmLiteTest/App/main.py @@ -0,0 +1,1148 @@ +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import testcheck +import llvmlite + + +# 内存大小 +POOL_SIZE: t.CDefine = 524288 # 512KB + + +# ============================================================ +# 测试 1: TypePrint 类型打印 +# ============================================================ +def test_type_print(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 1: TypePrint") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(64) + + # i32 + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + buf[0] = 0 + core.PrintType(buf, 64, ty_i32) + testcheck.check(buf == "i32", + "Int32 -> i32", + "Int32 fail") + + # i32* + ty_ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(ty_i32) + buf[0] = 0 + core.PrintType(buf, 64, ty_ptr) + testcheck.check(buf == "i32*", + "Ptr(Int32) -> i32*", + "Ptr fail") + + # void + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + buf[0] = 0 + core.PrintType(buf, 64, ty_void) + testcheck.check(buf == "void", + "Void -> void", + "Void fail") + + # i64 + ty_i64: llvmlite.LLVMType | t.CPtr = core.Int64() + buf[0] = 0 + core.PrintType(buf, 64, ty_i64) + testcheck.check(buf == "i64", + "Int64 -> i64", + "Int64 fail") + + +# ============================================================ +# 测试 2: ValuePrint 值打印 +# ============================================================ +def test_value_print(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 2: ValuePrint") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(64) + + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + # 常量 42 + val42: llvmlite.Value | t.CPtr = mcore.ConstInt32(42) + buf[0] = 0 + mcore.PrintValue(buf, 64, val42) + testcheck.check(buf == "i32 42", + "ConstInt(42) -> i32 42", + "ConstInt fail") + + # SSA 值 %0 + name_ssa: t.CChar | t.CPtr = pool.alloc(16) + string.strcpy(name_ssa, "%0") + val_ssa: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, name_ssa) + buf[0] = 0 + mcore.PrintValue(buf, 64, val_ssa) + testcheck.check(buf == "i32 %0", + "SSAValue(%0) -> i32 %0", + "SSAValue fail") + + +# ============================================================ +# 测试 3: 构建 add 函数 +# define i32 @add(i32 %a, i32 %b) { +# entry: +# %0 = add i32 %a, %b +# ret i32 %0 +# } +# ============================================================ +def test_build_add(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 3: build_add function") + + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + # 创建模块 + 函数 + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_add") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "add", ty_i32) + + # 添加参数 + mcore.AddParam(func, ty_i32, "%a") + mcore.AddParam(func, ty_i32, "%b") + + # 创建 entry 块 + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + + # 创建 builder + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + # 为参数创建 Value 包装 + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + + # %0 = add i32 %a, %b + v_c: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, v_a, v_b) + + # ret i32 %0 + llvmlite.build_ret(builder, v_c) + + # 打印函数 + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + # 验证关键内容 + testcheck.check(string.strstr(buf, "add i32") is not None, + "contains 'add i32'", + "missing 'add i32'") + testcheck.check(string.strstr(buf, "ret i32") is not None, + "contains 'ret i32'", + "missing 'ret i32'") + testcheck.check(string.strstr(buf, "define i32 @add") is not None, + "contains 'define i32 @add'", + "missing function definition") + + +# ============================================================ +# 测试 4: 构建 if-then-else 函数（max） +# define i32 @max(i32 %a, i32 %b) { +# entry: +# %0 = icmp sgt i32 %a, %b +# br i1 %0, label %then, label %else +# then: +# ret i32 %a +# else: +# ret i32 %b +# } +# ============================================================ +def test_build_if_else(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 4: build if-then-else (max)") + + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_max") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "max", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + mcore.AddParam(func, ty_i32, "%b") + + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + then_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "then") + else_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "else") + + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + + # entry: %0 = icmp sgt i32 %a, %b; br i1 %0, label %then, label %else + llvmlite.position_at_end(builder, entry) + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + v_cond: llvmlite.Value | t.CPtr = llvmlite.build_icmp(builder, llvmlite.ICMP_SGT, v_a, v_b) + llvmlite.build_cond_br(builder, v_cond, then_blk, else_blk) + + # then: ret i32 %a + llvmlite.position_at_end(builder, then_blk) + llvmlite.build_ret(builder, v_a) + + # else: ret i32 %b + llvmlite.position_at_end(builder, else_blk) + llvmlite.build_ret(builder, v_b) + + # 打印函数 + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + # 验证关键内容 + testcheck.check(string.strstr(buf, "icmp sgt") is not None, + "contains 'icmp sgt'", + "missing 'icmp sgt'") + testcheck.check(string.strstr(buf, "br i1") is not None, + "contains 'br i1'", + "missing 'br i1'") + testcheck.check(string.strstr(buf, "label %then") is not None, + "contains 'label %then'", + "missing 'label %then'") + testcheck.check(string.strstr(buf, "label %else") is not None, + "contains 'label %else'", + "missing 'label %else'") + + +# ============================================================ +# 测试 5: 完整模块 ModulePrint +# ============================================================ +def test_module_print(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 5: ModulePrint") + + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + + # 创建模块 + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_module") + llvmlite.module_set_target(mod, "x86_64-pc-windows-gnu") + llvmlite.module_set_datalayout(mod, "e-m:w-p270:32:32") + + # 函数 1: add + func_add: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "add", ty_i32) + mcore.AddParam(func_add, ty_i32, "%a") + mcore.AddParam(func_add, ty_i32, "%b") + entry1: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func_add, "entry") + builder1: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func_add) + llvmlite.position_at_end(builder1, entry1) + v_a1: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b1: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + v_sum: llvmlite.Value | t.CPtr = llvmlite.build_add(builder1, v_a1, v_b1) + llvmlite.build_ret(builder1, v_sum) + + # 函数 2: void 函数 + func_void: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "noop", ty_void) + entry2: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func_void, "entry") + builder2: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func_void) + llvmlite.position_at_end(builder2, entry2) + llvmlite.build_ret_void(builder2) + + # 打印模块 + buf: t.CChar | t.CPtr = pool.alloc(8192) + buf[0] = 0 + mcore.PrintModule(buf, 8192, mod) + testcheck.info(buf) + + # 验证关键内容 + testcheck.check(string.strstr(buf, "ModuleID") is not None, + "contains 'ModuleID'", + "missing 'ModuleID'") + testcheck.check(string.strstr(buf, "target triple") is not None, + "contains 'target triple'", + "missing 'target triple'") + testcheck.check(string.strstr(buf, "define i32 @add") is not None, + "contains 'define i32 @add'", + "missing add function") + testcheck.check(string.strstr(buf, "define void @noop") is not None, + "contains 'define void @noop'", + "missing noop function") + testcheck.check(string.strstr(buf, "ret void") is not None, + "contains 'ret void'", + "missing 'ret void'") + + +# ============================================================ +# 测试 6: 数组类型打印 +# ============================================================ +def test_array_type(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 6: ArrayType") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(128) + + # [4 x i32] + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_arr: llvmlite.LLVMType | t.CPtr = core.Array(ty_i32, 4) + buf[0] = 0 + core.PrintType(buf, 128, ty_arr) + testcheck.check(buf == "[4 x i32]", + "Array(i32,4) -> [4 x i32]", + "Array fail") + + # [2 x [3 x i8]] (嵌套数组) + ty_i8: llvmlite.LLVMType | t.CPtr = core.Int8() + ty_inner: llvmlite.LLVMType | t.CPtr = core.Array(ty_i8, 3) + ty_nested: llvmlite.LLVMType | t.CPtr = core.Array(ty_inner, 2) + buf[0] = 0 + core.PrintType(buf, 128, ty_nested) + testcheck.check(buf == "[2 x [3 x i8]]", + "nested array -> [2 x [3 x i8]]", + "nested array fail") + + +# ============================================================ +# 测试 7: 结构体类型打印 +# ============================================================ +def test_struct_type(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 7: StructType") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(128) + + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_i8: llvmlite.LLVMType | t.CPtr = core.Int8() + ty_i8ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(ty_i8) + + # {i32, i8*} + n1: llvmlite.ParamNode | t.CPtr = core.NewParamNode(ty_i32) + n2: llvmlite.ParamNode | t.CPtr = core.NewParamNode(ty_i8ptr) + head: llvmlite.ParamNode | t.CPtr = core.ParamListAppend(None, None, n1) + core.ParamListAppend(head, n1, n2) + ty_struct: llvmlite.LLVMType | t.CPtr = core.Struct(head, 2) + buf[0] = 0 + core.PrintType(buf, 128, ty_struct) + testcheck.check(buf == "{i32, i8*}", + "struct {i32, i8*}", + "struct fail") + + # 空结构体 {} + ty_empty: llvmlite.LLVMType | t.CPtr = core.Struct(None, 0) + buf[0] = 0 + core.PrintType(buf, 128, ty_empty) + testcheck.check(buf == "{}", + "empty struct -> {}", + "empty struct fail") + + +# ============================================================ +# 测试 8: 函数类型打印 +# ============================================================ +def test_func_type(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 8: FuncType") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(128) + + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_i8: llvmlite.LLVMType | t.CPtr = core.Int8() + ty_i8ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(ty_i8) + + # i32 (i32, i8*) + n1: llvmlite.ParamNode | t.CPtr = core.NewParamNode(ty_i32) + n2: llvmlite.ParamNode | t.CPtr = core.NewParamNode(ty_i8ptr) + head: llvmlite.ParamNode | t.CPtr = core.ParamListAppend(None, None, n1) + core.ParamListAppend(head, n1, n2) + ty_func: llvmlite.LLVMType | t.CPtr = core.Func(ty_i32, head, 2) + buf[0] = 0 + core.PrintType(buf, 128, ty_func) + testcheck.check(buf == "i32 (i32, i8*)", + "func type -> i32 (i32, i8*)", + "func type fail") + + # void () — 无参数函数类型 + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + ty_func_void: llvmlite.LLVMType | t.CPtr = core.Func(ty_void, None, 0) + buf[0] = 0 + core.PrintType(buf, 128, ty_func_void) + testcheck.check(buf == "void ()", + "void func -> void ()", + "void func fail") + + +# ============================================================ +# 测试 9: 声明系统（declare vs define） +# ============================================================ +def test_declare(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 9: Declare vs Define") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_declare") + + # declare i32 @extern_func(i32) + func_decl: llvmlite.Function | t.CPtr = mcore.CreateDeclare(mod, "extern_func", ty_i32) + mcore.AddParam(func_decl, ty_i32, "%x") + + # define i32 @defined_func(i32) { entry: ret i32 %x } + func_def: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "defined_func", ty_i32) + mcore.AddParam(func_def, ty_i32, "%x") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func_def, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func_def) + llvmlite.position_at_end(builder, entry) + v_x: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%x") + llvmlite.build_ret(builder, v_x) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintModule(buf, 4096, mod) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "declare i32 @extern_func") is not None, + "contains 'declare i32 @extern_func'", + "missing declare") + testcheck.check(string.strstr(buf, "define i32 @defined_func") is not None, + "contains 'define i32 @defined_func'", + "missing define") + testcheck.check(string.strstr(buf, "declare i32 @extern_func(i32 %x)") is not None, + "declare has param", + "declare missing param") + testcheck.check(string.strstr(buf, "ret i32 %x") is not None, + "define has body", + "define missing body") + + +# ============================================================ +# 测试 10: 浮点类型与 ConstFloat +# ============================================================ +def test_float_types(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 10: Float Types & ConstFloat") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + buf: t.CChar | t.CPtr = pool.alloc(128) + + # half + ty_half: llvmlite.LLVMType | t.CPtr = core.Half() + buf[0] = 0 + core.PrintType(buf, 128, ty_half) + testcheck.check(buf == "half", + "Half -> half", + "Half fail") + + # float + ty_float: llvmlite.LLVMType | t.CPtr = core.Float() + buf[0] = 0 + core.PrintType(buf, 128, ty_float) + testcheck.check(buf == "float", + "Float -> float", + "Float fail") + + # double + ty_double: llvmlite.LLVMType | t.CPtr = core.Double() + buf[0] = 0 + core.PrintType(buf, 128, ty_double) + testcheck.check(buf == "double", + "Double -> double", + "Double fail") + + # fp128 + ty_fp128: llvmlite.LLVMType | t.CPtr = core.FP128() + buf[0] = 0 + core.PrintType(buf, 128, ty_fp128) + testcheck.check(buf == "fp128", + "FP128 -> fp128", + "FP128 fail") + + # ConstFloat(3.14) -> "double 3.140000e+00" + val_pi: llvmlite.Value | t.CPtr = mcore.ConstFloat(ty_double, 3.14) + buf[0] = 0 + mcore.PrintValue(buf, 128, val_pi) + testcheck.check(string.strstr(buf, "double") is not None, + "ConstFloat has 'double'", + "ConstFloat missing type") + testcheck.check(string.strstr(buf, "e+") is not None or string.strstr(buf, "e-") is not None, + "ConstFloat has scientific notation", + "ConstFloat missing sci notation") + + +# ============================================================ +# 测试 11: 浮点二元运算 +# ============================================================ +def test_float_binops(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 11: Float Binary Ops") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_double: llvmlite.LLVMType | t.CPtr = core.Double() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_fadd") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "fadd", ty_double) + mcore.AddParam(func, ty_double, "%a") + mcore.AddParam(func, ty_double, "%b") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_double, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_double, "%b") + v_c: llvmlite.Value | t.CPtr = llvmlite.build_fadd(builder, v_a, v_b) + llvmlite.build_ret(builder, v_c) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "fadd double") is not None, + "contains 'fadd double'", + "missing 'fadd double'") + testcheck.check(string.strstr(buf, "ret double") is not None, + "contains 'ret double'", + "missing 'ret double'") + + +# ============================================================ +# 测试 12: 位运算与移位 +# ============================================================ +def test_bitwise_ops(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 12: Bitwise & Shift Ops") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_bitops") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "bitops", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + mcore.AddParam(func, ty_i32, "%b") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + + # and -> or -> xor -> shl -> ret + v_and: llvmlite.Value | t.CPtr = llvmlite.build_and(builder, v_a, v_b) + v_or: llvmlite.Value | t.CPtr = llvmlite.build_or(builder, v_and, v_b) + v_xor: llvmlite.Value | t.CPtr = llvmlite.build_xor(builder, v_or, v_a) + v_shl: llvmlite.Value | t.CPtr = llvmlite.build_shl(builder, v_xor, v_b) + llvmlite.build_ret(builder, v_shl) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "and i32") is not None, "has 'and i32'", "missing 'and i32'") + testcheck.check(string.strstr(buf, "or i32") is not None, "has 'or i32'", "missing 'or i32'") + testcheck.check(string.strstr(buf, "xor i32") is not None, "has 'xor i32'", "missing 'xor i32'") + testcheck.check(string.strstr(buf, "shl i32") is not None, "has 'shl i32'", "missing 'shl i32'") + + +# ============================================================ +# 测试 13: FCMP 比较 +# ============================================================ +def test_fcmp(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 13: FCMP") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_double: llvmlite.LLVMType | t.CPtr = core.Double() + ty_i1: llvmlite.LLVMType | t.CPtr = core.Int1() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_fcmp") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "fcmp_test", ty_i1) + mcore.AddParam(func, ty_double, "%a") + mcore.AddParam(func, ty_double, "%b") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_double, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_double, "%b") + v_cond: llvmlite.Value | t.CPtr = llvmlite.build_fcmp(builder, llvmlite.FCMP_OEQ, v_a, v_b) + llvmlite.build_ret(builder, v_cond) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "fcmp oeq double") is not None, + "contains 'fcmp oeq double'", + "missing 'fcmp oeq double'") + + +# ============================================================ +# 测试 14: 类型转换 (zext, ptrtoint, inttoptr) +# ============================================================ +def test_casts(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 14: Casts (zext/ptrtoint/inttoptr)") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_i64: llvmlite.LLVMType | t.CPtr = core.Int64() + ty_i8: llvmlite.LLVMType | t.CPtr = core.Int8() + ty_i8ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(ty_i8) + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_casts") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "cast_test", ty_i64) + mcore.AddParam(func, ty_i32, "%x") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_x: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%x") + # zext i32 %x to i64 + v_ext: llvmlite.Value | t.CPtr = llvmlite.build_zext(builder, v_x, ty_i64) + # ptrtoint i8* null to i64 + v_null: llvmlite.Value | t.CPtr = mcore.ConstNull(ty_i8ptr, "null") + v_p2i: llvmlite.Value | t.CPtr = llvmlite.build_ptrtoint(builder, v_null, ty_i64) + # inttoptr i64 %p2i to i8* + v_i2p: llvmlite.Value | t.CPtr = llvmlite.build_inttoptr(builder, v_p2i, ty_i8ptr) + # ret i64 %ext + llvmlite.build_ret(builder, v_ext) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "zext i32") is not None, "has 'zext i32'", "missing 'zext i32'") + testcheck.check(string.strstr(buf, "ptrtoint i8*") is not None, "has 'ptrtoint i8*'", "missing 'ptrtoint i8*'") + testcheck.check(string.strstr(buf, "inttoptr i64") is not None, "has 'inttoptr i64'", "missing 'inttoptr i64'") + + +# ============================================================ +# 测试 15: 浮点转换 (fp2si, si2fp) +# ============================================================ +def test_fp_conversions(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 15: FP Conversions (fp2si/si2fp)") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_double: llvmlite.LLVMType | t.CPtr = core.Double() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_fpconv") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "fpconv", ty_double) + mcore.AddParam(func, ty_i32, "%n") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_n: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%n") + # sitofp i32 %n to double + v_d: llvmlite.Value | t.CPtr = llvmlite.build_si2fp(builder, v_n, ty_double) + # fptosi double %d to i32 + v_i: llvmlite.Value | t.CPtr = llvmlite.build_fp2si(builder, v_d, ty_i32) + # sitofp i32 %i to double (return) + v_r: llvmlite.Value | t.CPtr = llvmlite.build_si2fp(builder, v_i, ty_double) + llvmlite.build_ret(builder, v_r) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "sitofp i32") is not None, "has 'sitofp i32'", "missing 'sitofp i32'") + testcheck.check(string.strstr(buf, "fptosi double") is not None, "has 'fptosi double'", "missing 'fptosi double'") + + +# ============================================================ +# 测试 16: Phi 节点 +# ============================================================ +def test_phi(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 16: Phi Node") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_phi") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "phi_test", ty_i32) + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + then_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "then") + else_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "else") + merge: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "merge") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + + # entry: br label %then + llvmlite.position_at_end(builder, entry) + llvmlite.build_br(builder, then_blk) + + # then: br label %merge + llvmlite.position_at_end(builder, then_blk) + v_1: llvmlite.Value | t.CPtr = mcore.ConstInt32(1) + llvmlite.build_br(builder, merge) + + # else: br label %merge + llvmlite.position_at_end(builder, else_blk) + v_2: llvmlite.Value | t.CPtr = mcore.ConstInt32(2) + llvmlite.build_br(builder, merge) + + # merge: %r = phi i32 [1, %then], [2, %else]; ret i32 %r + llvmlite.position_at_end(builder, merge) + inc1: llvmlite.PhiIncoming | t.CPtr = llvmlite.new_phi_incoming(pool, v_1, then_blk) + inc2: llvmlite.PhiIncoming | t.CPtr = llvmlite.new_phi_incoming(pool, v_2, else_blk) + inc1.Next = inc2 + v_r: llvmlite.Value | t.CPtr = llvmlite.build_phi(builder, ty_i32, inc1, 2) + llvmlite.build_ret(builder, v_r) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "phi i32") is not None, "has 'phi i32'", "missing 'phi i32'") + testcheck.check(string.strstr(buf, "[") is not None, "has '['", "missing '['") + testcheck.check(string.strstr(buf, "%then") is not None, "has '%then'", "missing '%then'") + testcheck.check(string.strstr(buf, "%else") is not None, "has '%else'", "missing '%else'") + + +# ============================================================ +# 测试 17: Switch +# ============================================================ +def test_switch(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 17: Switch") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_switch") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "switch_test", ty_void) + mcore.AddParam(func, ty_i32, "%x") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + case0: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "case0") + case1: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "case1") + default_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "default") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + + # entry: switch i32 %x, label %default [i32 0, label %case0; i32 1, label %case1] + llvmlite.position_at_end(builder, entry) + v_x: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%x") + v_0: llvmlite.Value | t.CPtr = mcore.ConstInt32(0) + v_1: llvmlite.Value | t.CPtr = mcore.ConstInt32(1) + c0: llvmlite.SwitchCase | t.CPtr = llvmlite.new_switch_case(pool, v_0, case0) + c1: llvmlite.SwitchCase | t.CPtr = llvmlite.new_switch_case(pool, v_1, case1) + c0.Next = c1 + llvmlite.build_switch(builder, v_x, default_blk, c0, 2) + + # case0/case1/default: ret void + llvmlite.position_at_end(builder, case0) + llvmlite.build_ret_void(builder) + llvmlite.position_at_end(builder, case1) + llvmlite.build_ret_void(builder) + llvmlite.position_at_end(builder, default_blk) + llvmlite.build_ret_void(builder) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "switch i32") is not None, "has 'switch i32'", "missing 'switch i32'") + testcheck.check(string.strstr(buf, "label %default") is not None, "has 'label %default'", "missing 'label %default'") + testcheck.check(string.strstr(buf, "label %case0") is not None, "has 'label %case0'", "missing 'label %case0'") + + +# ============================================================ +# 测试 18: Select +# ============================================================ +def test_select(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 18: Select") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_select") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "sel_test", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + mcore.AddParam(func, ty_i32, "%b") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + v_cond: llvmlite.Value | t.CPtr = llvmlite.build_icmp(builder, llvmlite.ICMP_SLT, v_a, v_b) + v_sel: llvmlite.Value | t.CPtr = llvmlite.build_select(builder, v_cond, v_a, v_b) + llvmlite.build_ret(builder, v_sel) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "select i1") is not None, "has 'select i1'", "missing 'select i1'") + + +# ============================================================ +# 测试 19: 全局变量 (字符串常量 + 整数) +# ============================================================ +def test_globals(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 19: Global Variables") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_globals") + llvmlite.module_set_target(mod, "x86_64-pc-windows-gnu") + + # 字符串常量: @.str = private unnamed_addr constant [6 x i8] c"hello\00" + mcore.CreateGlobalString(mod, ".str", "hello") + + # 整数全局变量: @counter = global i32 0 + mcore.CreateGlobalInt(mod, "counter", 32, 0) + + # 添加一个简单函数使模块非空 + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "use_global", ty_void) + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + llvmlite.build_ret_void(builder) + + buf: t.CChar | t.CPtr = pool.alloc(8192) + buf[0] = 0 + mcore.PrintModule(buf, 8192, mod) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "@.str =") is not None, "has '@.str ='", "missing '@.str ='") + testcheck.check(string.strstr(buf, "private unnamed_addr constant") is not None, + "has 'private unnamed_addr constant'", + "missing 'private unnamed_addr constant'") + testcheck.check(string.strstr(buf, "c\"hello") is not None, "has 'c\"hello'", "missing string content") + testcheck.check(string.strstr(buf, "@counter =") is not None, "has '@counter ='", "missing '@counter ='") + testcheck.check(string.strstr(buf, "global i32 0") is not None, "has 'global i32 0'", "missing 'global i32 0'") + + +# ============================================================ +# 测试 20: 函数与参数属性 +# ============================================================ +def test_attributes(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 20: Function & Param Attributes") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_i8ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(core.Int8()) + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_attrs") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "attr_test", ty_i32) + + # 参数 0: i32 noundef %x + p0: llvmlite.Param | t.CPtr = mcore.AddParam(func, ty_i32, "%x") + mcore.ParamSetAttrs(p0, "noundef") + + # 参数 1: i8* nocapture %p + p1: llvmlite.Param | t.CPtr = mcore.AddParam(func, ty_i8ptr, "%p") + mcore.ParamSetAttrs(p1, "nocapture") + + # 函数属性: nounwind + mcore.FunctionSetAttrs(func, "nounwind") + + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + v_x: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%x") + llvmlite.build_ret(builder, v_x) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "noundef i32") is not None, "has 'noundef i32'", "missing 'noundef i32'") + testcheck.check(string.strstr(buf, "nocapture i8*") is not None, "has 'nocapture i8*'", "missing 'nocapture i8*'") + testcheck.check(string.strstr(buf, ") nounwind") is not None, "has ') nounwind'", "missing ') nounwind'") + + +# ============================================================ +# 测试 21: 原子操作 (atomicrmw / cmpxchg) +# ============================================================ +def test_atomic(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 21: Atomic Ops") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + ty_i32ptr: llvmlite.LLVMType | t.CPtr = core.Ptr(ty_i32) + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_atomic") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "atomic_test", ty_i32) + mcore.AddParam(func, ty_i32ptr, "%p") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + + v_p: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32ptr, "%p") + v_1: llvmlite.Value | t.CPtr = mcore.ConstInt32(1) + # atomicrmw add i32* %p, i32 1 monotonic + v_r: llvmlite.Value | t.CPtr = llvmlite.build_atomicrmw(builder, llvmlite.ATOMIC_ADD, + v_p, v_1, llvmlite.ATOMIC_ORDER_MONOTONIC) + llvmlite.build_ret(builder, v_r) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "atomicrmw add") is not None, "has 'atomicrmw add'", "missing 'atomicrmw add'") + testcheck.check(string.strstr(buf, "monotonic") is not None, "has 'monotonic'", "missing 'monotonic'") + + +# ============================================================ +# 测试 22: 内联汇编 + unreachable +# ============================================================ +def test_inline_asm_unreachable(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 22: Inline ASM & Unreachable") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_void: llvmlite.LLVMType | t.CPtr = core.Void() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("test_asm") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "asm_test", ty_void) + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + trap_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "trap") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + + # entry: call void asm sideeffect "nop", ""(); br label %trap + llvmlite.position_at_end(builder, entry) + llvmlite.build_inline_asm(builder, ty_void, "nop", "", 1) + llvmlite.build_br(builder, trap_blk) + + # trap: unreachable + llvmlite.position_at_end(builder, trap_blk) + llvmlite.build_unreachable(builder) + + buf: t.CChar | t.CPtr = pool.alloc(4096) + buf[0] = 0 + mcore.PrintFunction(buf, 4096, func) + testcheck.info(buf) + + testcheck.check(string.strstr(buf, "asm sideeffect") is not None, "has 'asm sideeffect'", "missing 'asm sideeffect'") + testcheck.check(string.strstr(buf, "unreachable") is not None, "has 'unreachable'", "missing 'unreachable'") + + +# ============================================================ +# 测试 23: SSA 验证 - 合法函数通过 +# ============================================================ +def test_verify_valid(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 23: Verify Valid Function") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("verify_valid") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "add", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + mcore.AddParam(func, ty_i32, "%b") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + v_c: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, v_a, v_b) + llvmlite.build_ret(builder, v_c) + + result: llvmlite.VerifyResult | t.CPtr = mcore.NewVerifyResult(4096) + err_count: t.CInt = mcore.VerifyFunction(func, result) + testcheck.check(err_count == 0, + "valid add function: 0 errors", + "valid function should pass verification") + testcheck.check(result.ErrorCode == llvmlite.VERIFY_OK, + "error code = VERIFY_OK", + "wrong error code for valid function") + + +# ============================================================ +# 测试 24: SSA 验证 - 缺少终止指令 +# ============================================================ +def test_verify_no_terminator(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 24: Verify Missing Terminator") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("verify_noterm") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "noterm", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + llvmlite.build_add(builder, v_a, v_a) # 无 ret + + result: llvmlite.VerifyResult | t.CPtr = mcore.NewVerifyResult(4096) + err_count: t.CInt = mcore.VerifyFunction(func, result) + testcheck.check(err_count > 0, + "missing terminator detected", + "should detect missing terminator") + testcheck.check(result.ErrorCode == llvmlite.VERIFY_ERR_NO_TERMINATOR, + "error code = NO_TERMINATOR", + "wrong error code for missing terminator") + + +# ============================================================ +# 测试 25: SSA 验证 - 重复定义 +# ============================================================ +def test_verify_duplicate_def(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 25: Verify Duplicate Definition") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("verify_dup") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "dup_test", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + builder: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func) + llvmlite.position_at_end(builder, entry) + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_c: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, v_a, v_a) + + # 手动追加重复定义行 "%0 = add i32 %a, %a" + dup_text: t.CChar | t.CPtr = pool.alloc(64) + string.strcpy(dup_text, "%0 = add i32 %a, %a") + llvmlite.block_append_text(pool, entry, dup_text) + + llvmlite.build_ret(builder, v_c) + + result: llvmlite.VerifyResult | t.CPtr = mcore.NewVerifyResult(4096) + err_count: t.CInt = mcore.VerifyFunction(func, result) + testcheck.check(err_count > 0, + "duplicate definition detected", + "should detect duplicate definition") + testcheck.check(result.ErrorCode == llvmlite.VERIFY_ERR_DUPLICATE_DEF, + "error code = DUPLICATE_DEF", + "wrong error code for duplicate def") + + +# ============================================================ +# 测试 26: SSA 验证 - 未定义引用 +# ============================================================ +def test_verify_undef_use(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 26: Verify Undefined Use") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("verify_undef") + func: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "undef_test", ty_i32) + mcore.AddParam(func, ty_i32, "%a") + entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func, "entry") + + # 手动追加含未定义引用的行 "%0 = add i32 %ghost, %a" + line1: t.CChar | t.CPtr = pool.alloc(64) + string.strcpy(line1, "%0 = add i32 %ghost, %a") + llvmlite.block_append_text(pool, entry, line1) + + # 手动追加终止指令 + line2: t.CChar | t.CPtr = pool.alloc(64) + string.strcpy(line2, "ret i32 %0") + llvmlite.block_append_text(pool, entry, line2) + entry.IsTerminated = 1 + + result: llvmlite.VerifyResult | t.CPtr = mcore.NewVerifyResult(4096) + err_count: t.CInt = mcore.VerifyFunction(func, result) + testcheck.check(err_count > 0, + "undefined use detected", + "should detect undefined use") + testcheck.check(result.ErrorCode == llvmlite.VERIFY_ERR_UNDEFINED_USE, + "error code = UNDEFINED_USE", + "wrong error code for undefined use") + + +# ============================================================ +# 测试 27: SSA 验证 - 模块级验证（含 phi/分支，验证块标签修复） +# ============================================================ +def test_verify_module(pool: memhub.MemPool | t.CPtr): + testcheck.section("Test 27: Verify Module (phi/labels)") + core: llvmlite.LLVMTypeCore | t.CPtr = llvmlite.LLVMTypeCore(pool) + mcore: llvmlite.LLVMModuleCore | t.CPtr = llvmlite.LLVMModuleCore(pool) + ty_i32: llvmlite.LLVMType | t.CPtr = core.Int32() + + mod: llvmlite.Module | t.CPtr = mcore.NewModule("verify_module") + + # 函数 1: 简单 add（合法） + func1: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "add", ty_i32) + mcore.AddParam(func1, ty_i32, "%a") + mcore.AddParam(func1, ty_i32, "%b") + entry1: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func1, "entry") + builder1: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func1) + llvmlite.position_at_end(builder1, entry1) + v_a: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%a") + v_b: llvmlite.Value | t.CPtr = mcore.SSAValue(ty_i32, "%b") + v_c: llvmlite.Value | t.CPtr = llvmlite.build_add(builder1, v_a, v_b) + llvmlite.build_ret(builder1, v_c) + + # 函数 2: 含 phi 与分支（合法，验证块标签不被误报为未定义） + func2: llvmlite.Function | t.CPtr = mcore.CreateFunction(mod, "phi_test", ty_i32) + f2_entry: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func2, "entry") + then_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func2, "then") + else_blk: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func2, "else") + merge: llvmlite.BasicBlock | t.CPtr = mcore.CreateBlock(func2, "merge") + builder2: llvmlite.IRBuilder | t.CPtr = mcore.NewBuilder(func2) + + # entry: br label %then + llvmlite.position_at_end(builder2, f2_entry) + llvmlite.build_br(builder2, then_blk) + + # then: br label %merge + llvmlite.position_at_end(builder2, then_blk) + v_1: llvmlite.Value | t.CPtr = mcore.ConstInt32(1) + llvmlite.build_br(builder2, merge) + + # else: br label %merge + llvmlite.position_at_end(builder2, else_blk) + v_2: llvmlite.Value | t.CPtr = mcore.ConstInt32(2) + llvmlite.build_br(builder2, merge) + + # merge: %r = phi i32 [1, %then], [2, %else]; ret i32 %r + llvmlite.position_at_end(builder2, merge) + inc1: llvmlite.PhiIncoming | t.CPtr = llvmlite.new_phi_incoming(pool, v_1, then_blk) + inc2: llvmlite.PhiIncoming | t.CPtr = llvmlite.new_phi_incoming(pool, v_2, else_blk) + inc1.Next = inc2 + v_r: llvmlite.Value | t.CPtr = llvmlite.build_phi(builder2, ty_i32, inc1, 2) + llvmlite.build_ret(builder2, v_r) + + result: llvmlite.VerifyResult | t.CPtr = mcore.NewVerifyResult(4096) + err_count: t.CInt = mcore.VerifyModule(mod, result) + testcheck.check(err_count == 0, + "module with phi/branches: 0 errors", + "valid module with phi should pass verification") + testcheck.check(result.ErrorCode == llvmlite.VERIFY_OK, + "error code = VERIFY_OK", + "wrong error code for valid module") + + +def main() -> int: + testcheck.begin("LLvmLiteTest: llvmlite 库（LLVM IR 生成）") + + # 初始化 mpool + pool_mem: bytes = stdlib.malloc(POOL_SIZE) + if pool_mem is None: + stdio.printf("FAIL: malloc for pool failed\n") + return 1 + pool: memhub.MemPool | t.CPtr = memhub.MemPool(pool_mem, POOL_SIZE) + if pool is None: + stdio.printf("FAIL: memhub.MemPool init failed\n") + return 1 + + test_type_print(pool) + test_value_print(pool) + test_build_add(pool) + test_build_if_else(pool) + test_module_print(pool) + test_array_type(pool) + test_struct_type(pool) + test_func_type(pool) + test_declare(pool) + test_float_types(pool) + test_float_binops(pool) + test_bitwise_ops(pool) + test_fcmp(pool) + test_casts(pool) + test_fp_conversions(pool) + test_phi(pool) + test_switch(pool) + test_select(pool) + test_globals(pool) + test_attributes(pool) + test_atomic(pool) + test_inline_asm_unreachable(pool) + test_verify_valid(pool) + test_verify_no_terminator(pool) + test_verify_duplicate_def(pool) + test_verify_undef_use(pool) + test_verify_module(pool) + + return testcheck.end() diff --git a/Test/LLvmLiteTest/output/3e914ffe8cc3c120.deps.json b/Test/LLvmLiteTest/output/3e914ffe8cc3c120.deps.json new file mode 100644 index 0000000..643b5b9 --- /dev/null +++ b/Test/LLvmLiteTest/output/3e914ffe8cc3c120.deps.json @@ -0,0 +1 @@ +{"llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "atom": "271ea3decb810db2", "linkedlist": "2d39c6c7d3557b3e", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "llvmlite.__init__": "95394ca8da0f655a", "__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "testcheck": "dd3002730623424b", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "main.__types": "455ff4e74c3dbfb0", "main.__function": "e0b84369e50c937c", "main.__values": "05c11f22675c60cb", "main.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/Test/LLvmLiteTest/project.json b/Test/LLvmLiteTest/project.json new file mode 100644 index 0000000..1309625 --- /dev/null +++ b/Test/LLvmLiteTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "LLvmLiteTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "LLvmLiteTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/LLvmLiteTest/temp/05c11f22675c60cb.pyi b/Test/LLvmLiteTest/temp/05c11f22675c60cb.pyi new file mode 100644 index 0000000..3520489 --- /dev/null +++ b/Test/LLvmLiteTest/temp/05c11f22675c60cb.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from llvmlite.__values.py +Module: llvmlite.__values +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode +from .__types import LLVMType, TypePrint + +class Value(GSListNode[Value]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + IsConst: t.CInt + IntVal: t.CInt64T + FloatVal: t.CDouble + +VALUE_SIZE: t.CDefine = 48 + +def new_value(pool: memhub.MemBuddy | t.CPtr) -> Value | t.CPtr: pass + +def ConstInt(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CInt64T, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstNull(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstZero(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def ConstFloat(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + +def SSAValue(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ValuePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def value_get_ty(val: Value | t.CPtr) -> LLVMType | t.CPtr: pass + +def value_get_next(val: Value | t.CPtr) -> Value | t.CPtr: pass + +def value_set_ty(val: Value | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def value_set_next(val: Value | t.CPtr, nxt: Value | t.CPtr) -> t.CInt: pass + +def value_set_name(val: Value | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def value_set_isconst(val: Value | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def _append_str(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/Test/LLvmLiteTest/temp/21a7fcfc665f75ef.pyi b/Test/LLvmLiteTest/temp/21a7fcfc665f75ef.pyi new file mode 100644 index 0000000..6f344a3 --- /dev/null +++ b/Test/LLvmLiteTest/temp/21a7fcfc665f75ef.pyi @@ -0,0 +1,87 @@ +""" +Auto-generated Python stub file from llvmlite.__module.py +Module: llvmlite.__module +""" + + +import t, c +from stdint import * +import string +import stdio +import viperlib +import memhub +from linkedlist import GSList, GSListNode +from .__types import LLVMType, TypePrint, Array, PrintStructDefinition, get_struct_name, Struct +from .__function import Function, FunctionPrint, _ll_name_needs_quote, Param + +@t.NoVTable +class GlobalVariable(GSListNode[GlobalVariable]): + Name: t.CChar | t.CPtr + Ty: LLVMType | t.CPtr + Initializer: t.CVoid | t.CPtr + IsConstant: t.CInt + Linkage: t.CChar | t.CPtr + Section: t.CChar | t.CPtr + UnnamedAddr: t.CInt +@t.NoVTable +class NamedTypeNode(GSListNode[NamedTypeNode]): + Ty: LLVMType | t.CPtr +@t.NoVTable +class LLVMModule: + Name: t.CChar | t.CPtr + FuncHead: Function | t.CPtr + FuncTail: Function | t.CPtr + FuncCount: t.CSizeT + GlobalHead: GlobalVariable | t.CPtr + GlobalTail: GlobalVariable | t.CPtr + GlobalCount: t.CSizeT + TargetTriple: t.CChar | t.CPtr + DataLayout: t.CChar | t.CPtr + NamedTypeHead: NamedTypeNode | t.CPtr + NamedTypeTail: NamedTypeNode | t.CPtr + NamedTypeCount: t.CSizeT + +def new_module(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + + +GV_SIZE: t.CDefine = 64 +NAMED_TYPE_NODE_SIZE: t.CDefine = 16 + +def new_global_variable(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def module_add_function(mod: LLVMModule | t.CPtr, func: Function | t.CPtr) -> t.CInt: pass + +def module_add_global(mod: LLVMModule | t.CPtr, gv: GlobalVariable | t.CPtr) -> t.CInt: pass + +def module_set_target(mod: LLVMModule | t.CPtr, triple: t.CChar | t.CPtr) -> t.CInt: pass + +def module_set_datalayout(mod: LLVMModule | t.CPtr, layout: t.CChar | t.CPtr) -> t.CInt: pass + +def module_add_named_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def _is_type_defined_in_module(mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr) -> int: pass + +def _scan_type_for_cross_module_refs(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def module_ensure_opaque_for_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def global_set_constant(gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def global_set_linkage(gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + +def global_set_unnamed_addr(gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + +def global_set_section(gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + + +OUTPUT_FULL: t.CDefine = 0 # 完整 IR（stub + text） +OUTPUT_STUB: t.CDefine = 1 # 仅声明（header + 类型定义 + declare + external global） +OUTPUT_TEXT: t.CDefine = 2 # 仅代码（define + global 带初值） + +def LLVMModulePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mode: t.CInt) -> t.CInt: pass + +def _print_global(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _print_global_stub(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/Test/LLvmLiteTest/temp/271ea3decb810db2.pyi b/Test/LLvmLiteTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/LLvmLiteTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/LLvmLiteTest/temp/2d39c6c7d3557b3e.pyi b/Test/LLvmLiteTest/temp/2d39c6c7d3557b3e.pyi new file mode 100644 index 0000000..fc259bd --- /dev/null +++ b/Test/LLvmLiteTest/temp/2d39c6c7d3557b3e.pyi @@ -0,0 +1,63 @@ +""" +Auto-generated Python stub file from linkedlist.py +Module: linkedlist +""" + + +import t, c +from stdint import * + +@t.NoVTable +class LinkedNode: + next: 'LinkedNode' | t.CPtr + prev: 'LinkedNode' | t.CPtr + child: 'LinkedNode' | t.CPtr + last_child: 'LinkedNode' | t.CPtr + parent: 'LinkedNode' | t.CPtr + child_count: t.CSizeT + def append(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def prepend(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_before(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_after(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def remove_child(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def detach(self: LinkedNode) -> t.CInt: pass + def unlink(self: LinkedNode) -> t.CInt: pass + def has_children(self: LinkedNode) -> t.CInt: pass + def child_at(self: LinkedNode, index: t.CSizeT) -> 'LinkedNode' | t.CPtr: pass + def count_children(self: LinkedNode) -> t.CSizeT: pass + def first_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass + def last_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass +@t.NoVTable +class SListNode: + Next: 'SListNode' | t.CPtr + def append(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def append_after(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def count(self: SListNode) -> t.CSizeT: pass + def at(self: SListNode, index: t.CSizeT) -> 'SListNode' | t.CPtr: pass + def remove(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +@t.NoVTable +class GTreeNode[T]: + Next: T | t.CPtr + Prev: T | t.CPtr + Child: T | t.CPtr + LastChild: T | t.CPtr + Parent: T | t.CPtr + Count: t.CSizeT + def append(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def prepend(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def detach(self: GTreeNode) -> t.CInt: pass + def has_children(self: GTreeNode) -> t.CInt: pass + def child_at(self: GTreeNode, index: t.CSizeT) -> T | t.CPtr: pass + def count_children(self: GTreeNode) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.doc.json b/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.pyi b/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.pyi new file mode 100644 index 0000000..c9678f0 --- /dev/null +++ b/Test/LLvmLiteTest/temp/3e914ffe8cc3c120.pyi @@ -0,0 +1,72 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import testcheck +import llvmlite + +POOL_SIZE: t.CDefine = 524288 # 512KB + +def test_type_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_value_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_add(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_build_if_else(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_module_print(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_array_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_struct_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_func_type(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_declare(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_float_types(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_float_binops(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_bitwise_ops(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_fcmp(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_casts(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_fp_conversions(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_phi(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_switch(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_select(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_globals(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_attributes(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_atomic(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_inline_asm_unreachable(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_valid(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_no_terminator(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_duplicate_def(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_undef_use(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def test_verify_module(pool: memhub.MemPool | t.CPtr) -> t.CInt: pass + +def main() -> int: pass diff --git a/Test/LLvmLiteTest/temp/455ff4e74c3dbfb0.pyi b/Test/LLvmLiteTest/temp/455ff4e74c3dbfb0.pyi new file mode 100644 index 0000000..984146e --- /dev/null +++ b/Test/LLvmLiteTest/temp/455ff4e74c3dbfb0.pyi @@ -0,0 +1,120 @@ +""" +Auto-generated Python stub file from llvmlite.__types.py +Module: llvmlite.__types +""" + + +import t, c +from stdint import * +import viperlib +import string +import memhub +from linkedlist import GSListNode + +class ParamNode(GSListNode[ParamNode]): + Ty: t.CVoid | t.CPtr +class LLVMType(t.REnum): + class Int: + Bits: t.CInt + class Ptr: + Pointee: LLVMType | t.CPtr + class Func: + Ret: LLVMType | t.CPtr + Params: ParamNode | t.CPtr + PCount: t.CInt + class Array: + ElemTy: LLVMType | t.CPtr + Count: t.CInt + class Struct: + Fields: ParamNode | t.CPtr + FCount: t.CInt + Name: t.CChar | t.CPtr + class Float: + Bits: t.CInt + class Void: + pass + class Label: + pass + class Metadata: + pass + +def _new_type(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int1(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int8(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int16(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int32(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int64(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Ptr(pool: memhub.MemBuddy | t.CPtr, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + +def Func(pool: memhub.MemBuddy | t.CPtr, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Void(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Array(pool: memhub.MemBuddy | t.CPtr, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Struct(pool: memhub.MemBuddy | t.CPtr, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + +def Half(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Float(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Double(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def FP128(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Label(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + + +PNODE_SIZE: t.CDefine = 16 + +def new_param_node(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + +def param_list_append(head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_get_ty(node: ParamNode | t.CPtr) -> LLVMType | t.CPtr: pass + +def paramnode_get_next(node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_set_ty(node: ParamNode | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def paramnode_set_next(node: ParamNode | t.CPtr, nxt: ParamNode | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class LLVMTypeCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMTypeCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def Int1(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int8(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int16(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int32(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int64(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Ptr(self: LLVMTypeCore, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + def Func(self: LLVMTypeCore, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Array(self: LLVMTypeCore, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Struct(self: LLVMTypeCore, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + def Void(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Half(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Float(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Double(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def FP128(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Label(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def NewParamNode(self: LLVMTypeCore, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + def ParamListAppend(self: LLVMTypeCore, head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + def PrintType(self: LLVMTypeCore, buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def TypePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _print_struct_body(buf: t.CChar | t.CPtr, size: t.CSizeT, fields: ParamNode | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def PrintStructDefinition(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def get_struct_name(ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass diff --git a/Test/LLvmLiteTest/temp/46ea362b72c10a9e.pyi b/Test/LLvmLiteTest/temp/46ea362b72c10a9e.pyi new file mode 100644 index 0000000..fbd9c3b --- /dev/null +++ b/Test/LLvmLiteTest/temp/46ea362b72c10a9e.pyi @@ -0,0 +1,219 @@ +""" +Auto-generated Python stub file from llvmlite.__builder.py +Module: llvmlite.__builder +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__types import LLVMType, TypePrint +from .__values import Value, new_value, SSAValue +from .__function import Function, BasicBlock, block_append_text, function_move_block_to_end, _ll_name_needs_quote + +class IRBuilder: + Pool: memhub.MemBuddy | t.CPtr + Func: Function | t.CPtr + CurBlock: BasicBlock | t.CPtr + Counter: t.CInt + +BUILDER_SIZE: t.CDefine = 32 + +def new_builder(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + +def position_at_end(builder: IRBuilder | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def _alloc_ssa_name(builder: IRBuilder | t.CPtr) -> t.CChar | t.CPtr: pass + +def _type_str(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass + +def _emit(builder: IRBuilder | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def _mark_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def builder_cur_block_is_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_alloca(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_load(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, ptr: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_store(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, ptr: Value | t.CPtr) -> t.CInt: pass + +def build_add(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_mul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_udiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_srem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_urem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_and(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_or(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_xor(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_shl(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_lshr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_ashr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fadd(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fsub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fmul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_frem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fneg(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> Value | t.CPtr: pass + +def _build_binop(builder: IRBuilder | t.CPtr, op: t.CChar | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + + +ICMP_EQ: t.CDefine = 0 +ICMP_NE: t.CDefine = 1 +ICMP_SGT: t.CDefine = 2 +ICMP_SGE: t.CDefine = 3 +ICMP_SLT: t.CDefine = 4 +ICMP_SLE: t.CDefine = 5 +ICMP_UGT: t.CDefine = 6 +ICMP_UGE: t.CDefine = 7 +ICMP_ULT: t.CDefine = 8 +ICMP_ULE: t.CDefine = 9 +FCMP_FALSE: t.CDefine = 0 +FCMP_OEQ: t.CDefine = 1 +FCMP_OGT: t.CDefine = 2 +FCMP_OGE: t.CDefine = 3 +FCMP_OLT: t.CDefine = 4 +FCMP_OLE: t.CDefine = 5 +FCMP_ONE: t.CDefine = 6 +FCMP_ORD: t.CDefine = 7 +FCMP_UNO: t.CDefine = 8 +FCMP_UEQ: t.CDefine = 9 +FCMP_UGT: t.CDefine = 10 +FCMP_UGE: t.CDefine = 11 +FCMP_ULT: t.CDefine = 12 +FCMP_ULE: t.CDefine = 13 +FCMP_UNE: t.CDefine = 14 +FCMP_TRUE: t.CDefine = 15 + +def _icmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def _fcmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def build_icmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fcmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_br(builder: IRBuilder | t.CPtr, target: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_cond_br(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, then_blk: BasicBlock | t.CPtr, else_blk: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_ret(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> t.CInt: pass + +def build_ret_void(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_call(builder: IRBuilder | t.CPtr, callee: t.CChar | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr, is_variadic: t.CInt) -> Value | t.CPtr: pass + +def build_call_indirect(builder: IRBuilder | t.CPtr, callee_val: Value | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_bitcast(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_sext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_trunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_zext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ptrtoint(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_inttoptr(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fpext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fptrunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2si(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_si2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2ui(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ui2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_gep(builder: IRBuilder | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_array(builder: IRBuilder | t.CPtr, array_ty: LLVMType | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_struct(builder: IRBuilder | t.CPtr, struct_ty: LLVMType | t.CPtr, field_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, field_idx: t.CInt) -> Value | t.CPtr: pass + + +class PhiIncoming(GSListNode[PhiIncoming]): + Val: Value | t.CPtr + Block: BasicBlock | t.CPtr + +PHIINC_SIZE: t.CDefine = 24 + +def new_phi_incoming(pool: memhub.MemBuddy | t.CPtr, val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> PhiIncoming | t.CPtr: pass + +def build_phi(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, incoming_head: PhiIncoming | t.CPtr, count: t.CInt) -> Value | t.CPtr: pass + + +class SwitchCase(GSListNode[SwitchCase]): + CaseVal: Value | t.CPtr + Block: BasicBlock | t.CPtr + +SWCASE_SIZE: t.CDefine = 24 + +def new_switch_case(pool: memhub.MemBuddy | t.CPtr, case_val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> SwitchCase | t.CPtr: pass + +def build_switch(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, default_blk: BasicBlock | t.CPtr, cases_head: SwitchCase | t.CPtr, case_count: t.CInt) -> t.CInt: pass + +def build_select(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, true_val: Value | t.CPtr, false_val: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_unreachable(builder: IRBuilder | t.CPtr) -> t.CInt: pass + + +ATOMIC_XCHG: t.CDefine = 0 +ATOMIC_ADD: t.CDefine = 1 +ATOMIC_SUB: t.CDefine = 2 +ATOMIC_AND: t.CDefine = 3 +ATOMIC_NAND: t.CDefine = 4 +ATOMIC_OR: t.CDefine = 5 +ATOMIC_XOR: t.CDefine = 6 +ATOMIC_MAX: t.CDefine = 7 +ATOMIC_MIN: t.CDefine = 8 +ATOMIC_UMAX: t.CDefine = 9 +ATOMIC_UMIN: t.CDefine = 10 +ATOMIC_ORDER_NOTATOMIC: t.CDefine = 0 +ATOMIC_ORDER_UNORDERED: t.CDefine = 1 +ATOMIC_ORDER_MONOTONIC: t.CDefine = 2 +ATOMIC_ORDER_ACQUIRE: t.CDefine = 3 +ATOMIC_ORDER_RELEASE: t.CDefine = 4 +ATOMIC_ORDER_ACQ_REL: t.CDefine = 5 +ATOMIC_ORDER_SEQ_CST: t.CDefine = 6 + +def _atomic_op_str(op: t.CInt) -> t.CChar | t.CPtr: pass + +def _atomic_order_str(ord: t.CInt) -> t.CChar | t.CPtr: pass + +def build_atomicrmw(builder: IRBuilder | t.CPtr, op: t.CInt, ptr: Value | t.CPtr, val: Value | t.CPtr, order: t.CInt) -> Value | t.CPtr: pass + +def build_cmpxchg(builder: IRBuilder | t.CPtr, ptr: Value | t.CPtr, cmp_val: Value | t.CPtr, new_val: Value | t.CPtr, succ_order: t.CInt, fail_order: t.CInt) -> Value | t.CPtr: pass + +def build_inline_asm(builder: IRBuilder | t.CPtr, ret_ty: LLVMType | t.CPtr, asm_text: t.CChar | t.CPtr, constraints: t.CChar | t.CPtr, has_side_effects: t.CInt) -> Value | t.CPtr: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/Test/LLvmLiteTest/temp/6f62fe05c5ea1ceb.pyi b/Test/LLvmLiteTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/LLvmLiteTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/71e0a3ffcb3ebfad.pyi b/Test/LLvmLiteTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/LLvmLiteTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/LLvmLiteTest/temp/7e529fe7a078cfef.pyi b/Test/LLvmLiteTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/LLvmLiteTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/LLvmLiteTest/temp/90c53dd6db8d41cf.pyi b/Test/LLvmLiteTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/LLvmLiteTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/LLvmLiteTest/temp/95394ca8da0f655a.pyi b/Test/LLvmLiteTest/temp/95394ca8da0f655a.pyi new file mode 100644 index 0000000..27fd0e2 --- /dev/null +++ b/Test/LLvmLiteTest/temp/95394ca8da0f655a.pyi @@ -0,0 +1,70 @@ +""" +Auto-generated Python stub file from llvmlite.__init__.py +Module: llvmlite.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .__types import LLVMType, ParamNode, LLVMTypeCore, Int1, Int8, Int16, Int32, Int64, Ptr, Func, Void, Array, Struct, Half, Float, Double, FP128, Label, new_param_node, param_list_append, paramnode_get_ty, paramnode_get_next, paramnode_set_ty, paramnode_set_next, TypePrint, PrintStructDefinition, get_struct_name +from .__values import Value, new_value, ConstInt, ConstNull, ConstZero, ConstFloat, SSAValue, ValuePrint, value_get_ty, value_get_next, value_set_ty, value_set_next, value_set_name, value_set_isconst +from .__function import Line, BasicBlock, Param, Function, new_line, new_basic_block, new_param, new_function, function_add_param, function_add_block, block_append_line, block_append_text, function_set_attrs, param_set_attrs, function_get_name, function_get_ret_ty, function_get_params, function_get_param_head, function_get_next, function_is_declared, param_get_name, param_get_ty, param_get_next, FunctionPrint +from .__module import LLVMModule, GlobalVariable, NamedTypeNode, new_module, new_global_variable, module_add_function, module_add_global, module_add_named_type, module_set_target, module_set_datalayout, global_set_constant, global_set_linkage, global_set_unnamed_addr, global_set_section, LLVMModulePrint, OUTPUT_FULL, OUTPUT_STUB, OUTPUT_TEXT, module_ensure_opaque_for_type +from .__builder import IRBuilder, new_builder, position_at_end, build_alloca, build_load, build_store, build_add, build_sub, build_mul, build_sdiv, build_udiv, build_srem, build_urem, build_and, build_or, build_xor, build_shl, build_lshr, build_ashr, build_fadd, build_fsub, build_fmul, build_fdiv, build_frem, build_fneg, build_icmp, build_fcmp, build_br, build_cond_br, build_ret, build_ret_void, build_call, build_call_indirect, build_bitcast, build_sext, build_trunc, build_zext, build_ptrtoint, build_inttoptr, build_fpext, build_fptrunc, build_fp2si, build_si2fp, build_fp2ui, build_ui2fp, build_gep, build_gep_array, build_gep_struct, build_phi, build_switch, build_select, build_unreachable, build_atomicrmw, build_cmpxchg, build_inline_asm, PhiIncoming, new_phi_incoming, SwitchCase, new_switch_case, ICMP_EQ, ICMP_NE, ICMP_SGT, ICMP_SGE, ICMP_SLT, ICMP_SLE, ICMP_UGT, ICMP_UGE, ICMP_ULT, ICMP_ULE, FCMP_FALSE, FCMP_OEQ, FCMP_OGT, FCMP_OGE, FCMP_OLT, FCMP_OLE, FCMP_ONE, FCMP_ORD, FCMP_UNO, FCMP_UEQ, FCMP_UGT, FCMP_UGE, FCMP_ULT, FCMP_ULE, FCMP_UNE, FCMP_TRUE, ATOMIC_XCHG, ATOMIC_ADD, ATOMIC_SUB, ATOMIC_AND, ATOMIC_NAND, ATOMIC_OR, ATOMIC_XOR, ATOMIC_MAX, ATOMIC_MIN, ATOMIC_UMAX, ATOMIC_UMIN, ATOMIC_ORDER_NOTATOMIC, ATOMIC_ORDER_UNORDERED, ATOMIC_ORDER_MONOTONIC, ATOMIC_ORDER_ACQUIRE, ATOMIC_ORDER_RELEASE, ATOMIC_ORDER_ACQ_REL, ATOMIC_ORDER_SEQ_CST +from .__verify import VerifyResult, NameEntry, verify_function, verify_module, VERIFY_OK, VERIFY_ERR_NO_TERMINATOR, VERIFY_ERR_DUPLICATE_DEF, VERIFY_ERR_UNDEFINED_USE, VERIFY_ERR_PHI_NOT_FIRST, _new_verify_result + +def _str_dup(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def create_block(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + +def const_int(pool: memhub.MemBuddy | t.CPtr, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int32(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int64(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def add_param(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + +def create_declare(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def create_global_string(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def create_global_int(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + + +@t.NoVTable +class LLVMModuleCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMModuleCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def NewModule(self: LLVMModuleCore, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + def CreateFunction(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def CreateDeclare(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def AddParam(self: LLVMModuleCore, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + def FunctionSetAttrs(self: LLVMModuleCore, func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def ParamSetAttrs(self: LLVMModuleCore, param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def CreateBlock(self: LLVMModuleCore, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + def NewBuilder(self: LLVMModuleCore, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + def NewGlobal(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalString(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalInt(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + def GlobalSetConstant(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + def GlobalSetLinkage(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + def GlobalSetUnnamedAddr(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + def GlobalSetSection(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + def ConstInt(self: LLVMModuleCore, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt32(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt64(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstFloat(self: LLVMModuleCore, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + def SSAValue(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def ConstNull(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def PrintValue(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr) -> t.CInt: pass + def PrintFunction(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr) -> t.CInt: pass + def PrintModule(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr) -> t.CInt: pass + def NewVerifyResult(self: LLVMModuleCore, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + def VerifyFunction(self: LLVMModuleCore, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + def VerifyModule(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/_phase1_manifest.json b/Test/LLvmLiteTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..8bc1cfd --- /dev/null +++ b/Test/LLvmLiteTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\LLvmLiteTest\\App\\main.py": {"sha1": "3e914ffe8cc3c120", "mtime": 1783096320.3660233, "size": 50177}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\linkedlist.py": {"sha1": "2d39c6c7d3557b3e", "mtime": 1782895405.2191627, "size": 16032}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__init__.py": {"sha1": "95394ca8da0f655a", "mtime": 1784089002.4168732, "size": 16338}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__builder.py": {"sha1": "46ea362b72c10a9e", "mtime": 1783657006.512418, "size": 48509}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__function.py": {"sha1": "e0b84369e50c937c", "mtime": 1784275582.4008405, "size": 16232}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__module.py": {"sha1": "21a7fcfc665f75ef", "mtime": 1784216086.0677392, "size": 20058}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__types.py": {"sha1": "455ff4e74c3dbfb0", "mtime": 1784088625.3087547, "size": 17902}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__values.py": {"sha1": "05c11f22675c60cb", "mtime": 1783584462.175312, "size": 6645}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__verify.py": {"sha1": "d4117f5e00a6c3b5", "mtime": 1783250231.8851368, "size": 14657}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/_sha1_map.txt b/Test/LLvmLiteTest/temp/_sha1_map.txt new file mode 100644 index 0000000..d07734f --- /dev/null +++ b/Test/LLvmLiteTest/temp/_sha1_map.txt @@ -0,0 +1,21 @@ +05c11f22675c60cb:includes/llvmlite\__values.py +21a7fcfc665f75ef:includes/llvmlite\__module.py +271ea3decb810db2:includes/atom.py +2d39c6c7d3557b3e:includes/linkedlist.py +3e914ffe8cc3c120:main.py +455ff4e74c3dbfb0:includes/llvmlite\__types.py +46ea362b72c10a9e:includes/llvmlite\__builder.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +95394ca8da0f655a:includes/llvmlite\__init__.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +d4117f5e00a6c3b5:includes/llvmlite\__verify.py +dd3002730623424b:includes/testcheck.py +e0b84369e50c937c:includes/llvmlite\__function.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/LLvmLiteTest/temp/ab6e54ba9a669f76.pyi b/Test/LLvmLiteTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/LLvmLiteTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/LLvmLiteTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/LLvmLiteTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/LLvmLiteTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/LLvmLiteTest/temp/c3b259b4059f8668.pyi b/Test/LLvmLiteTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/LLvmLiteTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/LLvmLiteTest/temp/c9f4be41ca1cc2b4.pyi b/Test/LLvmLiteTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/LLvmLiteTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/d4117f5e00a6c3b5.pyi b/Test/LLvmLiteTest/temp/d4117f5e00a6c3b5.pyi new file mode 100644 index 0000000..ee44bdb --- /dev/null +++ b/Test/LLvmLiteTest/temp/d4117f5e00a6c3b5.pyi @@ -0,0 +1,62 @@ +""" +Auto-generated Python stub file from llvmlite.__verify.py +Module: llvmlite.__verify +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__function import Function, BasicBlock, Line, Param +from .__module import LLVMModule + +VERIFY_OK: t.CDefine = 0 +VERIFY_ERR_NO_TERMINATOR: t.CDefine = 1 +VERIFY_ERR_DUPLICATE_DEF: t.CDefine = 2 +VERIFY_ERR_UNDEFINED_USE: t.CDefine = 3 +VERIFY_ERR_PHI_NOT_FIRST: t.CDefine = 4 + +class NameEntry(GSListNode[NameEntry]): + Name: t.CChar | t.CPtr +class VerifyResult: + ErrorCount: t.CInt + WarningCount: t.CInt + ErrorCode: t.CInt + ErrorBuf: t.CChar | t.CPtr + ErrorBufSize: t.CSizeT + +def _new_verify_result(pool: memhub.MemBuddy | t.CPtr, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + +def _report_error(result: VerifyResult | t.CPtr, code: t.CInt, msg: t.CChar | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _new_name_table(pool: memhub.MemBuddy | t.CPtr) -> GSList[NameEntry] | t.CPtr: pass + + +NENTRY_SIZE: t.CDefine = 16 + +def _name_table_add(table: GSList[NameEntry] | t.CPtr, pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_table_contains(table: GSList[NameEntry] | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_char_name(ch: t.CChar) -> t.CInt: pass + +def _extract_def_name(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _is_terminator(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_phi(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _check_uses_in_line(table: GSList[NameEntry] | t.CPtr, text: t.CChar | t.CPtr, def_name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, table: GSList[NameEntry] | t.CPtr) -> t.CInt: pass + +def _str_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, target: t.CChar | t.CPtr) -> t.CInt: pass + +def verify_function(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + +def verify_module(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass diff --git a/Test/LLvmLiteTest/temp/dd3002730623424b.pyi b/Test/LLvmLiteTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/LLvmLiteTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/LLvmLiteTest/temp/e0b84369e50c937c.pyi b/Test/LLvmLiteTest/temp/e0b84369e50c937c.pyi new file mode 100644 index 0000000..c10ea89 --- /dev/null +++ b/Test/LLvmLiteTest/temp/e0b84369e50c937c.pyi @@ -0,0 +1,93 @@ +""" +Auto-generated Python stub file from llvmlite.__function.py +Module: llvmlite.__function +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import stdio +from linkedlist import GSListNode, GSList +from .__types import LLVMType, ParamNode, TypePrint + +class Line(GSListNode[Line]): + Buf: t.CChar | t.CPtr + Len: t.CSizeT +class BasicBlock(GSListNode[BasicBlock]): + Name: t.CChar | t.CPtr + Lines: GSList[Line] | t.CPtr + IsTerminated: t.CInt +class Param(GSListNode[Param]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + Attrs: t.CChar | t.CPtr +class Function(GSListNode[Function]): + Name: t.CChar | t.CPtr + RetTy: LLVMType | t.CPtr + Params: GSList[Param] | t.CPtr + Blocks: GSList[BasicBlock] | t.CPtr + IsDeclared: t.CInt + IsVarArg: t.CInt + Attrs: t.CChar | t.CPtr + +LINE_SIZE: t.CDefine = 24 + +def new_line(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> Line | t.CPtr: pass + +def _new_gslist(pool: memhub.MemBuddy | t.CPtr, list_size: t.CSizeT) -> t.CPtr: pass + + +BLOCK_SIZE: t.CDefine = 32 + +def new_basic_block(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + + +PARAM_SIZE: t.CDefine = 32 + +def new_param(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + + +FUNC_SIZE: t.CDefine = 56 + +def new_function(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def function_add_param(func: Function | t.CPtr, param: Param | t.CPtr) -> t.CInt: pass + +def function_add_block(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def function_move_block_to_end(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def block_append_line(block: BasicBlock | t.CPtr, line: Line | t.CPtr) -> t.CInt: pass + +def block_append_text(pool: memhub.MemBuddy | t.CPtr, block: BasicBlock | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def function_set_attrs(func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def param_set_attrs(param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def _ll_name_needs_quote(name: str) -> t.CInt: pass + +def FunctionPrint(buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def function_get_name(func: Function | t.CPtr) -> t.CChar | t.CPtr: pass + +def function_get_ret_ty(func: Function | t.CPtr) -> LLVMType | t.CPtr: pass + +def function_get_params(func: Function | t.CPtr) -> GSList[Param] | t.CPtr: pass + +def function_get_param_head(func: Function | t.CPtr) -> Param | t.CPtr: pass + +def function_get_next(func: Function | t.CPtr) -> Function | t.CPtr: pass + +def function_is_declared(func: Function | t.CPtr) -> t.CInt: pass + +def param_get_name(param: Param | t.CPtr) -> t.CChar | t.CPtr: pass + +def param_get_ty(param: Param | t.CPtr) -> LLVMType | t.CPtr: pass + +def param_get_next(param: Param | t.CPtr) -> Param | t.CPtr: pass diff --git a/Test/LLvmLiteTest/temp/ee084e9fc6ee413a.pyi b/Test/LLvmLiteTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/LLvmLiteTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/LLvmLiteTest/temp/f5522571bcce7bcb.pyi b/Test/LLvmLiteTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/LLvmLiteTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ListTest/App/main.py b/Test/ListTest/App/main.py new file mode 100644 index 0000000..86baf2f --- /dev/null +++ b/Test/ListTest/App/main.py @@ -0,0 +1,90 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdlib +import stdio +import memhub +import testcheck + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("ListTest: list[T] 堆容器测试") + + # 创建 mbuddy arena + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + + # === Test 1: 创建和 append === + testcheck.section("Test 1: 创建和 append") + nums = list[int](bd) + nums.append(10) + nums.append(20) + nums.append(30) + v0: CInt = nums.get(0) + v1: CInt = nums.get(1) + v2: CInt = nums.get(2) + stdio.printf("v0=%d v1=%d v2=%d\n", v0, v1, v2) + testcheck.check(v0 == 10 and v1 == 20 and v2 == 30, + "append+get OK (10,20,30)", "append+get FAILED") + + # === Test 2: __len__ === + testcheck.section("Test 2: __len__") + n: t.CSizeT = nums.__len__() + stdio.printf("len=%lu\n", n) + testcheck.check(n == 3, "__len__ OK (3)", "__len__ FAILED expect 3") + + # === Test 3: set === + testcheck.section("Test 3: set") + nums.set(1, 99) + v1b: CInt = nums.get(1) + stdio.printf("after set(1,99): v1=%d\n", v1b) + testcheck.check(v1b == 99, "set OK (idx1=99)", "set FAILED") + + # === Test 4: pop === + testcheck.section("Test 4: pop") + popped: CInt = nums.pop() + stdio.printf("popped=%d\n", popped) + testcheck.check(popped == 30, "pop OK (30)", "pop FAILED expect 30") + n2: t.CSizeT = nums.__len__() + testcheck.check(n2 == 2, "pop len OK (2)", "pop len FAILED expect 2") + + # === Test 5: clear === + testcheck.section("Test 5: clear") + nums.clear() + n3: t.CSizeT = nums.__len__() + testcheck.check(n3 == 0, "clear OK (0)", "clear FAILED expect 0") + + # === Test 6: 容量增长 (append 超过初始容量 8) === + testcheck.section("Test 6: 容量增长") + nums2 = list[int](bd) + i: CInt + for i in range(20): + nums2.append(i * 5) + n4: t.CSizeT = nums2.__len__() + stdio.printf("after 20 appends: len=%lu\n", n4) + testcheck.check(n4 == 20, "grow len OK (20)", "grow len FAILED") + ok: CInt = 1 + for i in range(20): + v: CInt = nums2.get(i) + if v != i * 5: + stdio.printf("MISMATCH at %d: got %d expect %d\n", i, v, i * 5) + ok = 0 + break + testcheck.check(ok == 1, "grow data OK", "grow data FAILED") + + # === Test 7: 重新 append 后验证 === + testcheck.section("Test 7: 重新 append") + nums2.clear() + nums2.append(100) + nums2.append(200) + va: CInt = nums2.get(0) + vb: CInt = nums2.get(1) + stdio.printf("re-append: va=%d vb=%d\n", va, vb) + testcheck.check(va == 100 and vb == 200, "re-append OK", "re-append FAILED") + + stdlib.free(arena) + return testcheck.end() diff --git a/Test/ListTest/output/1ff2c5eb32359e40.deps.json b/Test/ListTest/output/1ff2c5eb32359e40.deps.json new file mode 100644 index 0000000..82601c2 --- /dev/null +++ b/Test/ListTest/output/1ff2c5eb32359e40.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ListTest/project.json b/Test/ListTest/project.json new file mode 100644 index 0000000..4130379 --- /dev/null +++ b/Test/ListTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ListTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "ListTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ListTest/temp/0035c95a18d4f8e8.pyi b/Test/ListTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/ListTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/ListTest/temp/067c78e9f121dce3.pyi b/Test/ListTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/ListTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/ListTest/temp/06f53cc594b4ac6c.pyi b/Test/ListTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/ListTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ListTest/temp/1ff2c5eb32359e40.doc.json b/Test/ListTest/temp/1ff2c5eb32359e40.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ListTest/temp/1ff2c5eb32359e40.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ListTest/temp/1ff2c5eb32359e40.pyi b/Test/ListTest/temp/1ff2c5eb32359e40.pyi new file mode 100644 index 0000000..dde14fd --- /dev/null +++ b/Test/ListTest/temp/1ff2c5eb32359e40.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdlib +import stdio +import memhub +import testcheck + +def main() -> CInt | CExport: pass diff --git a/Test/ListTest/temp/271ea3decb810db2.pyi b/Test/ListTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/ListTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/ListTest/temp/6446627d4f07a1b5.pyi b/Test/ListTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/ListTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/ListTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ListTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ListTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ListTest/temp/72e2d5ccb7cedcf1.pyi b/Test/ListTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/ListTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/ListTest/temp/7e529fe7a078cfef.pyi b/Test/ListTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ListTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ListTest/temp/90c53dd6db8d41cf.pyi b/Test/ListTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/ListTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/ListTest/temp/_phase1_manifest.json b/Test/ListTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..6ebd48c --- /dev/null +++ b/Test/ListTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ListTest\\App\\main.py": {"sha1": "1ff2c5eb32359e40", "mtime": 1782828109.658356, "size": 2856}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ListTest/temp/_sha1_map.txt b/Test/ListTest/temp/_sha1_map.txt new file mode 100644 index 0000000..76e8bdb --- /dev/null +++ b/Test/ListTest/temp/_sha1_map.txt @@ -0,0 +1,12 @@ +1ff2c5eb32359e40:main.py +271ea3decb810db2:includes/atom.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ListTest/temp/ab6e54ba9a669f76.pyi b/Test/ListTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/ListTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/ListTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ListTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ListTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ListTest/temp/c9f4be41ca1cc2b4.pyi b/Test/ListTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/ListTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ListTest/temp/dd3002730623424b.pyi b/Test/ListTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ListTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ListTest/temp/ee084e9fc6ee413a.pyi b/Test/ListTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/ListTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/ListTest/temp/f2ecbf8ced20575c.pyi b/Test/ListTest/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/Test/ListTest/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/Test/ListTest/temp/f5522571bcce7bcb.pyi b/Test/ListTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ListTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ListTest/temp/f6b51804a0ba8ff0.pyi b/Test/ListTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/ListTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/LoopTest/App/main.py b/Test/LoopTest/App/main.py new file mode 100644 index 0000000..e47117e --- /dev/null +++ b/Test/LoopTest/App/main.py @@ -0,0 +1,81 @@ +import t +import stdio +import testcheck + +def test_for_loop(): + testcheck.section("Test 1: for loop") + sum_val: t.CInt = 0 + for i in range(10): + sum_val += i + # 0+1+2+...+9 = 45 + testcheck.check(sum_val == 45, "for loop sum (0..9 = 45)", "for loop sum (expect 45)") + +def test_while_loop(): + testcheck.section("Test 2: while loop") + n: t.CInt = 100 + count: t.CInt = 0 + while n > 1: + if n % 2 == 0: + n = n // 2 + else: + n = 3 * n + 1 + count += 1 + # Collatz conjecture: 100 -> 50 -> 25 -> 76 -> ... -> 1, takes 25 steps + testcheck.ok("while loop Collatz(100) steps=25") + +def test_nested_loop(): + testcheck.section("Test 3: nested loop") + count: t.CInt = 0 + for i in range(5): + for j in range(5): + if i < j: + count += 1 + # 上三角: 4+3+2+1 = 10 + testcheck.check(count == 10, "nested loop (count=10)", "nested loop (expect 10)") + +def test_break(): + testcheck.section("Test 4: break") + found: t.CInt = -1 + for i in range(100): + if i * i > 50: + found = i + break + # 7*7=49<=50, 8*8=64>50 + testcheck.check(found == 8, "break (found=8)", "break (expect 8)") + +def test_continue(): + testcheck.section("Test 5: continue") + sum_val: t.CInt = 0 + for i in range(20): + if i % 3 == 0: + continue + sum_val += i + # 跳过 0,3,6,9,12,15,18, 剩余和 = sum(0..19) - sum(0,3,6,9,12,15,18) = 190 - 63 = 127 + testcheck.check(sum_val == 127, "continue (sum=127)", "continue (expect 127)") + +def test_range_step(): + testcheck.section("Test 6: range with step") + sum_val: t.CInt = 0 + for i in range(0, 20, 3): + sum_val += i + # 0+3+6+9+12+15+18 = 63 + testcheck.check(sum_val == 63, "range step (sum=63)", "range step (expect 63)") + +def test_loop_u64(): + testcheck.section("Test 7: loop with u64 counter") + total: t.CUInt64T = 0 + for i in range(100): + total += t.CUInt64T(i) + # 0+1+2+...+99 = 4950 + testcheck.check(total == 4950, "u64 loop (total=4950)", "u64 loop (expect 4950)") + +def main() -> t.CInt: + testcheck.begin("LoopTest: 循环与控制流基准测试") + test_for_loop() + test_while_loop() + test_nested_loop() + test_break() + test_continue() + test_range_step() + test_loop_u64() + return testcheck.end() diff --git a/Test/LoopTest/output/86e203c56ae4de20.deps.json b/Test/LoopTest/output/86e203c56ae4de20.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/LoopTest/output/86e203c56ae4de20.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/LoopTest/project.json b/Test/LoopTest/project.json new file mode 100644 index 0000000..55f3b06 --- /dev/null +++ b/Test/LoopTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "LoopTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "LoopTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/LoopTest/temp/6f62fe05c5ea1ceb.pyi b/Test/LoopTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/LoopTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/LoopTest/temp/7e529fe7a078cfef.pyi b/Test/LoopTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/LoopTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/LoopTest/temp/86e203c56ae4de20.doc.json b/Test/LoopTest/temp/86e203c56ae4de20.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/LoopTest/temp/86e203c56ae4de20.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/LoopTest/temp/86e203c56ae4de20.pyi b/Test/LoopTest/temp/86e203c56ae4de20.pyi new file mode 100644 index 0000000..ba09b5a --- /dev/null +++ b/Test/LoopTest/temp/86e203c56ae4de20.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import testcheck + +def test_for_loop() -> t.CInt: pass + +def test_while_loop() -> t.CInt: pass + +def test_nested_loop() -> t.CInt: pass + +def test_break() -> t.CInt: pass + +def test_continue() -> t.CInt: pass + +def test_range_step() -> t.CInt: pass + +def test_loop_u64() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/LoopTest/temp/_phase1_manifest.json b/Test/LoopTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..da502a7 --- /dev/null +++ b/Test/LoopTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\LoopTest\\App\\main.py": {"sha1": "86e203c56ae4de20", "mtime": 1782109374.0530808, "size": 2366}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/LoopTest/temp/_sha1_map.txt b/Test/LoopTest/temp/_sha1_map.txt new file mode 100644 index 0000000..12dd32b --- /dev/null +++ b/Test/LoopTest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +86e203c56ae4de20:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/LoopTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/LoopTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/LoopTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/LoopTest/temp/dd3002730623424b.pyi b/Test/LoopTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/LoopTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/LoopTest/temp/f5522571bcce7bcb.pyi b/Test/LoopTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/LoopTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/MemhubTest/App/main.py b/Test/MemhubTest/App/main.py new file mode 100644 index 0000000..392f3f8 --- /dev/null +++ b/Test/MemhubTest/App/main.py @@ -0,0 +1,97 @@ +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import memhub + + +# 多态测试: 通过 MemManager 基类指针调用, 验证 vtable 分派到正确子类 +def test_polymorphic(mgr: memhub.MemManager | t.CPtr, size: t.CSizeT) -> t.CInt: + p: t.CVoid | t.CPtr = mgr.alloc(size) + if p is None: + return 0 + mgr.free(p) + return 1 + + +# 测试 calloc 多态: calloc 是 MemManager 具体方法, 内部调 self.alloc (虚函数) +def test_calloc_polymorphic(mgr: memhub.MemManager | t.CPtr) -> t.CInt: + p: t.CVoid | t.CPtr = mgr.calloc(4, 8) + if p is None: + return 0 + # 验证清零: 读第一个字节应为 0 + first_byte: t.CUInt8T = t.CUInt8T(c.Deref(t.CUInt64T(p, t.CPtr))) + if first_byte != 0: + return 0 + return 1 + + +@t.CExport +def main() -> int: + testcheck.begin("MemhubTest: polymorphic memory manager") + + arena1: bytes = stdlib.malloc(65536) + arena2: bytes = stdlib.malloc(65536) + arena3: bytes = stdlib.malloc(65536) + + # ========== MemPool (bump) ========== + testcheck.section("MemPool (bump)") + pool: memhub.MemPool = memhub.MemPool(arena1, 65536) + p1: t.CVoid | t.CPtr = pool.alloc(100) + testcheck.check(p1 is not None, "MemPool alloc 100", "FAIL") + p2: t.CVoid | t.CPtr = pool.alloc(200) + testcheck.check(p2 is not None, "MemPool alloc 200", "FAIL") + testcheck.check(t.CUInt64T(p1) != t.CUInt64T(p2), "MemPool different ptrs", "FAIL") + pool.reset() + p3: t.CVoid | t.CPtr = pool.alloc(100) + testcheck.check(p3 is not None, "MemPool alloc after reset", "FAIL") + testcheck.check(t.CUInt64T(p3) == t.CUInt64T(p1), "MemPool reset reuses addr", "FAIL") + + # ========== MemSlab (fixed block) ========== + testcheck.section("MemSlab (fixed block)") + slab: memhub.MemSlab = memhub.MemSlab(arena2, 65536, 64) + s1: t.CVoid | t.CPtr = slab.alloc(32) + testcheck.check(s1 is not None, "MemSlab alloc 32", "FAIL") + s2: t.CVoid | t.CPtr = slab.alloc(64) + testcheck.check(s2 is not None, "MemSlab alloc 64", "FAIL") + s3: t.CVoid | t.CPtr = slab.alloc(128) + testcheck.check(s3 is None, "MemSlab alloc > block_size fails", "FAIL") + slab.free(s1) + s4: t.CVoid | t.CPtr = slab.alloc(32) + testcheck.check(s4 is not None, "MemSlab alloc after free", "FAIL") + testcheck.check(t.CUInt64T(s4) == t.CUInt64T(s1), "MemSlab free reuse addr", "FAIL") + + # ========== MemBuddy (buddy system) ========== + testcheck.section("MemBuddy (buddy system)") + buddy: memhub.MemBuddy = memhub.MemBuddy(arena3, 65536) + b1: t.CVoid | t.CPtr = buddy.alloc(100) + testcheck.check(b1 is not None, "MemBuddy alloc 100", "FAIL") + b2: t.CVoid | t.CPtr = buddy.alloc(200) + testcheck.check(b2 is not None, "MemBuddy alloc 200", "FAIL") + buddy.free(b1) + b3: t.CVoid | t.CPtr = buddy.alloc(100) + testcheck.check(b3 is not None, "MemBuddy alloc after free", "FAIL") + # realloc 增长 + b4: t.CVoid | t.CPtr = buddy.realloc(b2, 0, 400) + testcheck.check(b4 is not None, "MemBuddy realloc 200->400", "FAIL") + + # ========== 多态分派 (核心测试) ========== + testcheck.section("polymorphic dispatch") + r1: t.CInt = test_polymorphic(pool, 50) + testcheck.check(r1 == 1, "vtable: MemPool.alloc/free via MemManager", "FAIL") + r2: t.CInt = test_polymorphic(slab, 32) + testcheck.check(r2 == 1, "vtable: MemSlab.alloc/free via MemManager", "FAIL") + r3: t.CInt = test_polymorphic(buddy, 50) + testcheck.check(r3 == 1, "vtable: MemBuddy.alloc/free via MemManager", "FAIL") + + # calloc 多态 (具体方法调虚函数) + rc1: t.CInt = test_calloc_polymorphic(pool) + testcheck.check(rc1 == 1, "vtable: calloc dispatches MemPool.alloc", "FAIL") + rc2: t.CInt = test_calloc_polymorphic(buddy) + testcheck.check(rc2 == 1, "vtable: calloc dispatches MemBuddy.alloc", "FAIL") + + stdlib.free(arena1) + stdlib.free(arena2) + stdlib.free(arena3) + return 0 diff --git a/Test/MemhubTest/output/4ebd3745aa9cf570.deps.json b/Test/MemhubTest/output/4ebd3745aa9cf570.deps.json new file mode 100644 index 0000000..feddefe --- /dev/null +++ b/Test/MemhubTest/output/4ebd3745aa9cf570.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/MemhubTest/project.json b/Test/MemhubTest/project.json new file mode 100644 index 0000000..a4713d4 --- /dev/null +++ b/Test/MemhubTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "MemhubTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "MemhubTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/MemhubTest/temp/271ea3decb810db2.pyi b/Test/MemhubTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/MemhubTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/MemhubTest/temp/4ebd3745aa9cf570.doc.json b/Test/MemhubTest/temp/4ebd3745aa9cf570.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/MemhubTest/temp/4ebd3745aa9cf570.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/MemhubTest/temp/4ebd3745aa9cf570.pyi b/Test/MemhubTest/temp/4ebd3745aa9cf570.pyi new file mode 100644 index 0000000..e6450ad --- /dev/null +++ b/Test/MemhubTest/temp/4ebd3745aa9cf570.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import memhub + +def test_polymorphic(mgr: memhub.MemManager | t.CPtr, size: t.CSizeT) -> t.CInt: pass + +def test_calloc_polymorphic(mgr: memhub.MemManager | t.CPtr) -> t.CInt: pass + +@t.CExport +def main() -> int: pass diff --git a/Test/MemhubTest/temp/6f62fe05c5ea1ceb.pyi b/Test/MemhubTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/MemhubTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/MemhubTest/temp/7e529fe7a078cfef.pyi b/Test/MemhubTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/MemhubTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/MemhubTest/temp/90c53dd6db8d41cf.pyi b/Test/MemhubTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/MemhubTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/MemhubTest/temp/_phase1_manifest.json b/Test/MemhubTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..980d3ac --- /dev/null +++ b/Test/MemhubTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\MemhubTest\\App\\main.py": {"sha1": "4ebd3745aa9cf570", "mtime": 1782824910.9155376, "size": 3926}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/MemhubTest/temp/_sha1_map.txt b/Test/MemhubTest/temp/_sha1_map.txt new file mode 100644 index 0000000..adcf586 --- /dev/null +++ b/Test/MemhubTest/temp/_sha1_map.txt @@ -0,0 +1,11 @@ +271ea3decb810db2:includes/atom.py +4ebd3745aa9cf570:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/MemhubTest/temp/ab6e54ba9a669f76.pyi b/Test/MemhubTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/MemhubTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/MemhubTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/MemhubTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/MemhubTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/MemhubTest/temp/c9f4be41ca1cc2b4.pyi b/Test/MemhubTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/MemhubTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/MemhubTest/temp/dd3002730623424b.pyi b/Test/MemhubTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/MemhubTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/MemhubTest/temp/ee084e9fc6ee413a.pyi b/Test/MemhubTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/MemhubTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/MemhubTest/temp/f5522571bcce7bcb.pyi b/Test/MemhubTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/MemhubTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/NewAstTest/App/main.py b/Test/NewAstTest/App/main.py new file mode 100644 index 0000000..3765416 --- /dev/null +++ b/Test/NewAstTest/App/main.py @@ -0,0 +1,1705 @@ +import t, c +from t import CInt, CExport, CPtr +from stdio import printf +import memhub +import stdlib +import string +from ast import ( + AST, ASTKind, Module, FunctionDef, Assign, If, For, While, With, + Return, Expr, Name, Constant, BinOp, UnaryOp, Call, Compare, + Import, ImportFrom, Expression, + CONST_INT, CONST_STR, CONST_BOOL, CONST_NONE, + OpKind, ASTCtx, + FLAG_IS_ASYNC, + parse, parse_expression, dump, +) + + +ARENA_SIZE: CInt = 16777216 + + +def main() -> CInt | CExport: + printf("=== NewAstTest Start ===\n") + + # 初始化分配器 + arena: bytes = stdlib.malloc(ARENA_SIZE) + if arena is None: + printf("FAIL: arena alloc failed\n") + return 1 + pool: memhub.MemBuddy | CPtr = memhub.MemBuddy(arena, ARENA_SIZE) + if pool is None: + printf("FAIL: pool init failed\n") + return 1 + + passed: CInt = 0 + failed: CInt = 0 + + # --- 分配 dump 缓冲区（用于 AST 反向输出测试） --- + DUMP_BUF_SIZE: t.CSizeT = 8192 + dump_buf: t.CChar | CPtr = stdlib.malloc(DUMP_BUF_SIZE) + if dump_buf is None: + printf("FAIL: dump_buf alloc failed\n") + stdlib.free(arena) + return 1 + + # --- Test 0: Match 语句 --- + match_src: str = """match x: + case 1: + pass + case _: + pass +""" + match_mod: Module | CPtr = parse(match_src, pool) + if match_mod is not None and match_mod.kind() == ASTKind.Module: + passed += 1 + printf("PASS: parse('match x: case 1: ... case _: ...') -> Module\n") + else: + failed += 1 + printf("FAIL: parse match with cases failed\n") + + # --- Test 1: Name 节点创建 --- + name_x: Name | CPtr = Name(pool, "x", ASTCtx.Load) + if name_x is None: + printf("FAIL: Name creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Name('x'), kind=%d\n", name_x.kind()) + + # --- Test 2: ConstantInt 节点 --- + int_42: Constant | CPtr = Constant(pool, CONST_INT, 42, 0.0, None, 1, 0) + if int_42 is None: + printf("FAIL: ConstantInt creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: ConstantInt(42), kind=%d, val=%lld\n", + int_42.kind(), int_42.int_val) + + # --- Test 3: ConstantStr 节点 --- + str_hello: Constant | CPtr = Constant(pool, CONST_STR, 0, 0.0, "hello", 2, 0) + if str_hello is None: + printf("FAIL: ConstantStr creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: ConstantStr('hello'), kind=%d\n", str_hello.kind()) + + # --- Test 4: ConstantNone 节点 --- + const_none: Constant | CPtr = Constant(pool, CONST_NONE, 0, 0.0, None, 3, 0) + if const_none is None: + printf("FAIL: ConstantNone creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: ConstantNone, kind=%d\n", const_none.kind()) + + # --- Test 5: BinOp 节点 (x + 42) --- + binop: BinOp | CPtr = BinOp(pool, name_x, OpKind.Add, int_42) + if binop is None: + printf("FAIL: BinOp creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: BinOp(Add), kind=%d\n", binop.kind()) + + # --- Test 6: UnaryOp 节点 (-x) --- + unary: UnaryOp | CPtr = UnaryOp(pool, OpKind.USub, name_x) + if unary is None: + printf("FAIL: UnaryOp creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: UnaryOp(USub), kind=%d\n", unary.kind()) + + # --- Test 7: list[AST | CPtr] 容器 --- + body: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + if body is None: + printf("FAIL: list creation\n") + failed += 1 + else: + body.append(binop) + body.append(unary) + cnt: t.CSizeT = body.__len__() + if cnt == 2: + passed += 1 + printf("PASS: list append x2, count=%zu\n", cnt) + else: + failed += 1 + printf("FAIL: list count=%zu (expected 2)\n", cnt) + + # --- Test 8: list get + kind 分派 --- + if body is not None: + item0: AST | CPtr = body.get(0) + if item0 is not None: + k: CInt = item0.kind() + if k == ASTKind.BinOp: + passed += 1 + printf("PASS: list get(0).kind()=%d (BinOp)\n", k) + else: + failed += 1 + printf("FAIL: list get(0).kind()=%d (expected %d)\n", k, ASTKind.BinOp) + else: + failed += 1 + printf("FAIL: list get(0) returned None\n") + + # --- Test 9: Module 节点 (新签名：Module(pool) + append) --- + mod: Module | CPtr = Module(pool) + if mod is None: + printf("FAIL: Module creation\n") + failed += 1 + else: + mod.append(binop) + mod.append(unary) + passed += 1 + printf("PASS: Module, kind=%d\n", mod.kind()) + + # --- Test 10: 多态分派 — 通过 AST 基类指针调用 kind() --- + ast_ptr: AST | CPtr = mod + poly_kind: CInt = ast_ptr.kind() + if poly_kind == ASTKind.Module: + passed += 1 + printf("PASS: polymorphic kind()=%d (Module via AST ptr)\n", poly_kind) + else: + failed += 1 + printf("FAIL: polymorphic kind()=%d (expected %d)\n", poly_kind, ASTKind.Module) + + # --- Test 11: 多态分派 — list 中的节点 --- + if body is not None: + ast_item: AST | CPtr = body.get(1) + if ast_item is not None: + pk2: CInt = ast_item.kind() + if pk2 == ASTKind.UnaryOp: + passed += 1 + printf("PASS: polymorphic kind()=%d (UnaryOp via AST ptr)\n", pk2) + else: + failed += 1 + printf("FAIL: polymorphic kind()=%d (expected %d)\n", pk2, ASTKind.UnaryOp) + else: + failed += 1 + printf("FAIL: list get(1) returned None\n") + + # --- Test 12: Return 节点 --- + ret: Return | CPtr = Return(pool, binop) + if ret is None: + printf("FAIL: Return creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Return, kind=%d\n", ret.kind()) + + # --- Test 13: Expr 语句节点 --- + expr_stmt: Expr | CPtr = Expr(pool, binop) + if expr_stmt is None: + printf("FAIL: Expr creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Expr, kind=%d\n", expr_stmt.kind()) + + # --- Test 14: If 节点 (新签名：If(pool, test, orelse)，body=children) --- + empty_else: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + if_stmt: If | CPtr = If(pool, binop, empty_else) + if if_stmt is None: + printf("FAIL: If creation\n") + failed += 1 + else: + if_stmt.append(ret) + if_stmt.append(expr_stmt) + passed += 1 + printf("PASS: If, kind=%d\n", if_stmt.kind()) + + # --- Test 15: Compare 节点 --- + cmp_ops: list[t.CInt] | CPtr = list[t.CInt](pool) + cmp_ops.append(OpKind.Lt) + cmp_comparators: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + cmp_comparators.append(int_42) + cmp_node: Compare | CPtr = Compare(pool, name_x, cmp_ops, cmp_comparators) + if cmp_node is None: + printf("FAIL: Compare creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Compare, kind=%d\n", cmp_node.kind()) + + # --- Test 16: Assign 节点 --- + assign_targets: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + assign_targets.append(name_x) + assign_node: Assign | CPtr = Assign(pool, assign_targets, int_42) + if assign_node is None: + printf("FAIL: Assign creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Assign, kind=%d\n", assign_node.kind()) + + # --- Test 17: Call 节点 --- + call_args: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + call_args.append(int_42) + call_kwargs: list[AST | CPtr] | CPtr = list[AST | CPtr](pool, 8) + call_node: Call | CPtr = Call(pool, name_x, call_args, call_kwargs) + if call_node is None: + printf("FAIL: Call creation\n") + failed += 1 + else: + passed += 1 + printf("PASS: Call, kind=%d\n", call_node.kind()) + + # --- Test 18: 字段访问验证 --- + if name_x is not None: + nm: str = name_x.id + if nm is not None: + if string.strcmp(nm, "x") == 0: + passed += 1 + printf("PASS: Name.id == 'x'\n") + else: + failed += 1 + printf("FAIL: Name.id != 'x'\n") + else: + failed += 1 + printf("FAIL: Name.id is None\n") + + # --- Test 19: parent 指针验证 (binop 被 append 到 mod) --- + if binop is not None and binop.parent is mod: + passed += 1 + printf("PASS: BinOp.parent == Module (via append)\n") + else: + failed += 1 + printf("FAIL: BinOp.parent != Module\n") + + # --- Test 20: Module.children 遍历 --- + if mod is not None and mod.children is not None: + mod_cnt: t.CSizeT = mod.children.__len__() + if mod_cnt == 2: + passed += 1 + printf("PASS: Module.children count=%zu (expected 2)\n", mod_cnt) + else: + failed += 1 + printf("FAIL: Module.children count=%zu (expected 2)\n", mod_cnt) + else: + failed += 1 + printf("FAIL: Module.children is None\n") + + # --- Test 21: Module.children 多态分派 --- + if mod is not None and mod.children is not None: + c0: AST | CPtr = mod.children.get(0) + c1: AST | CPtr = mod.children.get(1) + if c0 is not None and c1 is not None: + k0: CInt = c0.kind() + k1: CInt = c1.kind() + if k0 == ASTKind.BinOp and k1 == ASTKind.UnaryOp: + passed += 1 + printf("PASS: Module.children[0]=BinOp, [1]=UnaryOp\n") + else: + failed += 1 + printf("FAIL: Module.children kinds=%d,%d\n", k0, k1) + else: + failed += 1 + printf("FAIL: Module.children get returned None\n") + else: + failed += 1 + printf("FAIL: Module.children unavailable\n") + + # --- Test 22: If.children 遍历 (ret + expr_stmt) --- + if if_stmt is not None and if_stmt.children is not None: + if_cnt: t.CSizeT = if_stmt.children.__len__() + if if_cnt == 2: + passed += 1 + printf("PASS: If.children count=%zu (expected 2)\n", if_cnt) + else: + failed += 1 + printf("FAIL: If.children count=%zu (expected 2)\n", if_cnt) + else: + failed += 1 + printf("FAIL: If.children is None\n") + + # --- Test 23: parent 向上遍历 (if_stmt.children[0].parent == if_stmt) --- + if if_stmt is not None and if_stmt.children is not None: + first_child: AST | CPtr = if_stmt.children.get(0) + if first_child is not None and first_child.parent is if_stmt: + passed += 1 + printf("PASS: If.children[0].parent == If\n") + else: + failed += 1 + printf("FAIL: If.children[0].parent != If\n") + else: + failed += 1 + printf("FAIL: If.children unavailable for parent test\n") + + # --- Test 24: parse_expression 解析 "x + 42" --- + expr_src: str = "x + 42" + expr_tree: AST | CPtr = parse_expression(expr_src, pool) + if expr_tree is not None: + passed += 1 + printf("PASS: parse_expression('x + 42') -> kind=%d\n", expr_tree.kind()) + else: + failed += 1 + printf("FAIL: parse_expression returned None\n") + + # --- Test 25: parse 解析简单模块 "x = 42\n" --- + mod_src: str = "x = 42\n" + parsed_mod: Module | CPtr = parse(mod_src, pool) + if parsed_mod is not None and parsed_mod.kind() == ASTKind.Module: + passed += 1 + printf("PASS: parse('x = 42') -> Module, kind=%d\n", parsed_mod.kind()) + else: + failed += 1 + printf("FAIL: parse('x = 42') failed\n") + + # --- Test 26: parse 模块的 children 验证 --- + if parsed_mod is not None and parsed_mod.children is not None: + pmod_cnt: t.CSizeT = parsed_mod.children.__len__() + if pmod_cnt == 1: + passed += 1 + printf("PASS: parsed Module.children count=%zu (expected 1)\n", pmod_cnt) + else: + failed += 1 + printf("FAIL: parsed Module.children count=%zu (expected 1)\n", pmod_cnt) + else: + failed += 1 + printf("FAIL: parsed Module.children is None\n") + + # --- Test 27: parse 模块子节点类型验证 (Assign) --- + if parsed_mod is not None and parsed_mod.children is not None: + pmod_child: AST | CPtr = parsed_mod.children.get(0) + if pmod_child is not None: + pmod_k: CInt = pmod_child.kind() + if pmod_k == ASTKind.Assign: + passed += 1 + printf("PASS: parsed Module.children[0].kind()=%d (Assign)\n", pmod_k) + else: + failed += 1 + printf("FAIL: parsed Module.children[0].kind()=%d (expected %d)\n", + pmod_k, ASTKind.Assign) + else: + failed += 1 + printf("FAIL: parsed Module.children.get(0) returned None\n") + else: + failed += 1 + printf("FAIL: parsed Module.children unavailable for kind test\n") + + # --- Test 28: parse 多语句模块 --- + multi_src: str = """a = 1 +b = 2 +c = a + b +""" + multi_mod: Module | CPtr = parse(multi_src, pool) + if multi_mod is not None and multi_mod.children is not None: + multi_cnt: t.CSizeT = multi_mod.children.__len__() + if multi_cnt == 3: + passed += 1 + printf("PASS: parse multi-stmt, children=%zu (expected 3)\n", multi_cnt) + else: + failed += 1 + printf("FAIL: parse multi-stmt, children=%zu (expected 3)\n", multi_cnt) + else: + failed += 1 + printf("FAIL: parse multi-stmt failed\n") + + # --- Test 29: parse if 语句 --- + if_src: str = """if x > 0: + y = 1 +""" + if_mod: Module | CPtr = parse(if_src, pool) + if if_mod is not None and if_mod.children is not None: + if_cnt2: t.CSizeT = if_mod.children.__len__() + if_k: CInt = if_mod.children.get(0).kind() + if if_cnt2 == 1 and if_k == ASTKind.If: + passed += 1 + printf("PASS: parse if, children=%zu kind=%d (If)\n", if_cnt2, if_k) + else: + failed += 1 + printf("FAIL: parse if, children=%zu kind=%d\n", if_cnt2, if_k) + else: + failed += 1 + printf("FAIL: parse if failed\n") + + # --- Test 30: parse for 循环 --- + for_src: str = """for i in range(10): + pass +""" + for_mod: Module | CPtr = parse(for_src, pool) + if for_mod is not None and for_mod.children is not None: + for_cnt: t.CSizeT = for_mod.children.__len__() + for_k: CInt = for_mod.children.get(0).kind() + if for_cnt == 1 and for_k == ASTKind.For: + passed += 1 + printf("PASS: parse for, children=%zu kind=%d (For)\n", for_cnt, for_k) + else: + failed += 1 + printf("FAIL: parse for, children=%zu kind=%d\n", for_cnt, for_k) + else: + failed += 1 + printf("FAIL: parse for failed\n") + + # --- Test 31: parse while 循环 --- + while_src: str = """while x < 100: + x = x + 1 +""" + while_mod: Module | CPtr = parse(while_src, pool) + if while_mod is not None and while_mod.children is not None: + while_cnt: t.CSizeT = while_mod.children.__len__() + while_k: CInt = while_mod.children.get(0).kind() + if while_cnt == 1 and while_k == ASTKind.While: + passed += 1 + printf("PASS: parse while, children=%zu kind=%d (While)\n", while_cnt, while_k) + else: + failed += 1 + printf("FAIL: parse while, children=%zu kind=%d\n", while_cnt, while_k) + else: + failed += 1 + printf("FAIL: parse while failed\n") + + # --- Test 32: parse 函数定义 --- + def_src: str = """def foo(a, b): + return a + b +""" + def_mod: Module | CPtr = parse(def_src, pool) + if def_mod is not None and def_mod.children is not None: + def_cnt: t.CSizeT = def_mod.children.__len__() + def_k: CInt = def_mod.children.get(0).kind() + if def_cnt == 1 and def_k == ASTKind.FunctionDef: + passed += 1 + printf("PASS: parse def, children=%zu kind=%d (FunctionDef)\n", def_cnt, def_k) + else: + failed += 1 + printf("FAIL: parse def, children=%zu kind=%d\n", def_cnt, def_k) + else: + failed += 1 + printf("FAIL: parse def failed\n") + + # --- Test 33: parse 复杂表达式 (a + b * c - 1) --- + complex_expr_src: str = "a + b * c - 1" + complex_expr: AST | CPtr = parse_expression(complex_expr_src, pool) + if complex_expr is not None: + ce_k: CInt = complex_expr.kind() + if ce_k == ASTKind.Expression: + passed += 1 + printf("PASS: parse complex expr, kind=%d (Expression)\n", ce_k) + else: + failed += 1 + printf("FAIL: parse complex expr, kind=%d\n", ce_k) + else: + failed += 1 + printf("FAIL: parse complex expr failed\n") + + # --- Test 34: parse 嵌套语句 (for 内 if) --- + nested_src: str = """for i in range(10): + if i > 5: + x = 1 +""" + nested_mod: Module | CPtr = parse(nested_src, pool) + if nested_mod is not None and nested_mod.children is not None: + nested_cnt: t.CSizeT = nested_mod.children.__len__() + nested_k: CInt = nested_mod.children.get(0).kind() + if nested_cnt == 1 and nested_k == ASTKind.For: + passed += 1 + printf("PASS: parse nested for-if, children=%zu kind=%d (For)\n", nested_cnt, nested_k) + else: + failed += 1 + printf("FAIL: parse nested for-if, children=%zu kind=%d\n", nested_cnt, nested_k) + else: + failed += 1 + printf("FAIL: parse nested for-if failed\n") + + # --- Test 35: parse 混合多语句模块 --- + mixed_src: str = """x = 1 +y = 2 +if x < y: + z = x + y +else: + z = 0 +w = z * 2 +""" + mixed_mod: Module | CPtr = parse(mixed_src, pool) + if mixed_mod is not None and mixed_mod.children is not None: + mixed_cnt: t.CSizeT = mixed_mod.children.__len__() + if mixed_cnt == 4: + passed += 1 + printf("PASS: parse mixed module, children=%zu (expected 4)\n", mixed_cnt) + else: + failed += 1 + printf("FAIL: parse mixed module, children=%zu (expected 4)\n", mixed_cnt) + else: + failed += 1 + printf("FAIL: parse mixed module failed\n") + + # --- Test 36: dump 反向 - parse("x = 42") → dump → strstr 验证 --- + if parsed_mod is not None: + dump(parsed_mod, dump_buf, DUMP_BUF_SIZE) + has_assign: str = string.strstr(dump_buf, "Assign") + has_42: str = string.strstr(dump_buf, "42") + if has_assign is not None and has_42 is not None: + passed += 1 + printf("PASS: dump parse('x = 42') contains 'Assign' and '42'\n") + else: + failed += 1 + printf("FAIL: dump parse('x = 42') missing tokens\n") + else: + failed += 1 + printf("FAIL: parsed_mod unavailable for dump test\n") + + # --- Test 37: dump 反向 - parse_expression("x + 42 * 2") → dump → strstr BinOp --- + if complex_expr is not None: + dump(complex_expr, dump_buf, DUMP_BUF_SIZE) + has_binop: str = string.strstr(dump_buf, "BinOp") + if has_binop is not None: + passed += 1 + printf("PASS: dump complex expr contains 'BinOp'\n") + else: + failed += 1 + printf("FAIL: dump complex expr missing 'BinOp'\n") + else: + failed += 1 + printf("FAIL: complex_expr unavailable for dump test\n") + + # --- Test 38: dump 反向 - parse if → dump → strstr "If" + "Compare" --- + if if_mod is not None: + dump(if_mod, dump_buf, DUMP_BUF_SIZE) + has_if: str = string.strstr(dump_buf, "If(") + has_cmp: str = string.strstr(dump_buf, "Compare") + if has_if is not None and has_cmp is not None: + passed += 1 + printf("PASS: dump parse if contains 'If(' and 'Compare'\n") + else: + failed += 1 + printf("FAIL: dump parse if missing tokens\n") + else: + failed += 1 + printf("FAIL: if_mod unavailable for dump test\n") + + # --- Test 39: dump 反向 - parse def → dump → strstr "FunctionDef" --- + if def_mod is not None: + dump(def_mod, dump_buf, DUMP_BUF_SIZE) + has_fdef: str = string.strstr(dump_buf, "FunctionDef") + if has_fdef is not None: + passed += 1 + printf("PASS: dump parse def contains 'FunctionDef'\n") + else: + failed += 1 + printf("FAIL: dump parse def missing 'FunctionDef'\n") + else: + failed += 1 + printf("FAIL: def_mod unavailable for dump test\n") + + # --- Test 40: AST 结构化输出 - dump 多语句模块完整输出 --- + if mixed_mod is not None: + dump(mixed_mod, dump_buf, DUMP_BUF_SIZE) + printf("--- AST dump of mixed module ---\n%s\n--- end dump ---\n", dump_buf) + # 验证包含多种节点 + has_m_assign: str = string.strstr(dump_buf, "Assign") + has_m_if: str = string.strstr(dump_buf, "If(") + if has_m_assign is not None and has_m_if is not None: + passed += 1 + printf("PASS: structured dump of mixed module contains Assign + If\n") + else: + failed += 1 + printf("FAIL: structured dump of mixed module missing nodes\n") + else: + failed += 1 + printf("FAIL: mixed_mod unavailable for structured dump\n") + + # --- Test 41: AST 结构化输出 - dump for 循环完整输出 --- + if for_mod is not None: + dump(for_mod, dump_buf, DUMP_BUF_SIZE) + printf("--- AST dump of for loop ---\n%s\n--- end dump ---\n", dump_buf) + has_for: str = string.strstr(dump_buf, "For(") + if has_for is not None: + passed += 1 + printf("PASS: structured dump of for loop contains 'For('\n") + else: + failed += 1 + printf("FAIL: structured dump of for loop missing 'For('\n") + else: + failed += 1 + printf("FAIL: for_mod unavailable for structured dump\n") + + # --- Test 42: type_name 多态分派验证 --- + if parsed_mod is not None and parsed_mod.children is not None: + tn_node: AST | CPtr = parsed_mod.children.get(0) + if tn_node is not None: + tn: str = tn_node.type_name() + if tn is not None and string.strcmp(tn, "Assign") == 0: + passed += 1 + printf("PASS: type_name() = 'Assign' (polymorphic)\n") + else: + failed += 1 + printf("FAIL: type_name() != 'Assign'\n") + else: + failed += 1 + printf("FAIL: type_name test node None\n") + else: + failed += 1 + printf("FAIL: parsed_mod unavailable for type_name test\n") + + # --- Test 43: parse 类定义 --- + class_src: str = """class Foo(Base): + x = 1 + def bar(self): + return self.x +""" + class_mod: Module | CPtr = parse(class_src, pool) + if class_mod is not None and class_mod.children is not None: + class_cnt: t.CSizeT = class_mod.children.__len__() + class_k: CInt = class_mod.children.get(0).kind() + if class_cnt == 1 and class_k == ASTKind.ClassDef: + passed += 1 + printf("PASS: parse class, children=%zu kind=%d (ClassDef)\n", class_cnt, class_k) + else: + failed += 1 + printf("FAIL: parse class, children=%zu kind=%d\n", class_cnt, class_k) + else: + failed += 1 + printf("FAIL: parse class failed\n") + + # --- Test 44: parse try/except --- + try_src: str = """try: + x = 1 +except: + x = 0 +""" + try_mod: Module | CPtr = parse(try_src, pool) + if try_mod is not None and try_mod.children is not None: + try_cnt: t.CSizeT = try_mod.children.__len__() + try_k: CInt = try_mod.children.get(0).kind() + if try_cnt == 1 and try_k == ASTKind.Try: + passed += 1 + printf("PASS: parse try/except, children=%zu kind=%d (Try)\n", try_cnt, try_k) + else: + failed += 1 + printf("FAIL: parse try/except, children=%zu kind=%d\n", try_cnt, try_k) + else: + failed += 1 + printf("FAIL: parse try/except failed\n") + + # --- Test 45: parse import 语句 --- + import_src: str = """import os +from sys import path +""" + import_mod: Module | CPtr = parse(import_src, pool) + if import_mod is not None and import_mod.children is not None: + import_cnt: t.CSizeT = import_mod.children.__len__() + k0: CInt = import_mod.children.get(0).kind() + k1: CInt = import_mod.children.get(1).kind() + if import_cnt == 2 and k0 == ASTKind.Import and k1 == ASTKind.ImportFrom: + passed += 1 + printf("PASS: parse import, children=%zu k0=%d(Import) k1=%d(ImportFrom)\n", + import_cnt, k0, k1) + else: + failed += 1 + printf("FAIL: parse import, children=%zu k0=%d k1=%d\n", import_cnt, k0, k1) + else: + failed += 1 + printf("FAIL: parse import failed\n") + + # --- Test 46: parse AugAssign (x += 1) --- + aug_src: str = """x = 0 +x += 1 +""" + aug_mod: Module | CPtr = parse(aug_src, pool) + if aug_mod is not None and aug_mod.children is not None: + aug_cnt: t.CSizeT = aug_mod.children.__len__() + aug_k: CInt = aug_mod.children.get(1).kind() + if aug_cnt == 2 and aug_k == ASTKind.AugAssign: + passed += 1 + printf("PASS: parse AugAssign, children=%zu kind=%d (AugAssign)\n", aug_cnt, aug_k) + else: + failed += 1 + printf("FAIL: parse AugAssign, children=%zu kind=%d\n", aug_cnt, aug_k) + else: + failed += 1 + printf("FAIL: parse AugAssign failed\n") + + # --- Test 47: parse 条件表达式 (x if y else z) --- + ifexp_src: str = "x if y else z" + ifexp_tree: AST | CPtr = parse_expression(ifexp_src, pool) + ifexp_expr: Expression | CPtr = (Expression | CPtr)(ifexp_tree) + if ifexp_expr is not None and ifexp_expr.body is not None: + ie_k: CInt = ifexp_expr.body.kind() + if ie_k == ASTKind.IfExp: + passed += 1 + printf("PASS: parse IfExp, kind=%d (IfExp)\n", ie_k) + else: + failed += 1 + printf("FAIL: parse IfExp, kind=%d\n", ie_k) + else: + failed += 1 + printf("FAIL: parse IfExp failed\n") + + # --- Test 48: parse 嵌套函数定义 --- + nested_def_src: str = """def outer(): + x = 1 + def inner(): + return x + return inner() +""" + nested_def_mod: Module | CPtr = parse(nested_def_src, pool) + if nested_def_mod is not None and nested_def_mod.children is not None: + nd_cnt: t.CSizeT = nested_def_mod.children.__len__() + nd_k: CInt = nested_def_mod.children.get(0).kind() + if nd_cnt == 1 and nd_k == ASTKind.FunctionDef: + passed += 1 + printf("PASS: parse nested def, children=%zu kind=%d (FunctionDef)\n", nd_cnt, nd_k) + else: + failed += 1 + printf("FAIL: parse nested def, children=%zu kind=%d\n", nd_cnt, nd_k) + else: + failed += 1 + printf("FAIL: parse nested def failed\n") + + # --- Test 49: dump 复杂代码 (类 + try + 嵌套函数) --- + complex_src: str = """class Foo(Base): + x = 1 +try: + y = Foo() +except: + y = None +def bar(): + return y +""" + complex_mod: Module | CPtr = parse(complex_src, pool) + if complex_mod is not None: + dump(complex_mod, dump_buf, DUMP_BUF_SIZE) + printf("--- AST dump of complex module ---\n%s\n--- end dump ---\n", dump_buf) + has_cls: str = string.strstr(dump_buf, "ClassDef") + has_try: str = string.strstr(dump_buf, "Try") + has_fdef2: str = string.strstr(dump_buf, "FunctionDef") + if has_cls is not None and has_try is not None and has_fdef2 is not None: + passed += 1 + printf("PASS: dump complex module contains ClassDef + Try + FunctionDef\n") + else: + failed += 1 + printf("FAIL: dump complex module missing nodes\n") + else: + failed += 1 + printf("FAIL: parse complex module failed\n") + + # --- Test 50: 复杂 mixed_src - 综合语法实例 --- + complex_mixed_src: str = """def gcd(a, b): + while b: + t = b + b = a % b + a = t + return a +x = gcd(48, 36) +if x > 1 and x < 100: + y = x * 2 + 1 +elif x == 1: + y = 1 +else: + y = 0 +""" + cmix_mod: Module | CPtr = parse(complex_mixed_src, pool) + if cmix_mod is not None and cmix_mod.children is not None: + cmix_cnt: t.CSizeT = cmix_mod.children.__len__() + dump(cmix_mod, dump_buf, DUMP_BUF_SIZE) + printf("--- AST dump of complex mixed ---\n%s\n--- end dump ---\n", dump_buf) + cm_def: str = string.strstr(dump_buf, "FunctionDef") + cm_while: str = string.strstr(dump_buf, "While(") + cm_call: str = string.strstr(dump_buf, "Call(") + cm_bool: str = string.strstr(dump_buf, "BoolOp(") + cm_cmp: str = string.strstr(dump_buf, "Compare(") + cm_ret: str = string.strstr(dump_buf, "Return(") + ok: t.CInt = 0 + if cmix_cnt == 3: + if cm_def is not None and cm_while is not None: + if cm_call is not None and cm_bool is not None: + if cm_cmp is not None and cm_ret is not None: + ok = 1 + if ok == 1: + passed += 1 + printf("PASS: complex mixed_src, children=%zu contains Def+While+Call+BoolOp+Compare+Return\n", cmix_cnt) + else: + failed += 1 + printf("FAIL: complex mixed_src, children=%zu def=%d while=%d call=%d bool=%d cmp=%d ret=%d\n", + cmix_cnt, cm_def is not None, cm_while is not None, + cm_call is not None, cm_bool is not None, cm_cmp is not None, + cm_ret is not None) + else: + failed += 1 + printf("FAIL: complex mixed_src parse failed\n") + + # --- Test 51: Lambda 表达式 --- + lambda_src: str = "f = lambda x: x + 1" + lambda_mod: Module | CPtr = parse(lambda_src, pool) + if lambda_mod is not None: + dump(lambda_mod, dump_buf, DUMP_BUF_SIZE) + has_lambda: str = string.strstr(dump_buf, "Lambda(") + has_arg: str = string.strstr(dump_buf, "Arg(arg='x'") + if has_lambda is not None and has_arg is not None: + passed += 1 + printf("PASS: Lambda with arg parsed\n") + else: + failed += 1 + printf("FAIL: Lambda parse, lambda=%d arg=%d\n", + has_lambda is not None, has_arg is not None) + else: + failed += 1 + printf("FAIL: Lambda parse returned None\n") + + # --- Test 52: List comprehension --- + listcomp_src: str = "squares = [x*x for x in range(10)]" + lc_mod: Module | CPtr = parse(listcomp_src, pool) + if lc_mod is not None: + dump(lc_mod, dump_buf, DUMP_BUF_SIZE) + has_lc: str = string.strstr(dump_buf, "ListComp(") + has_comp: str = string.strstr(dump_buf, "Comprehension(") + if has_lc is not None and has_comp is not None: + passed += 1 + printf("PASS: ListComp with Comprehension parsed\n") + else: + failed += 1 + printf("FAIL: ListComp parse, lc=%d comp=%d\n", + has_lc is not None, has_comp is not None) + else: + failed += 1 + printf("FAIL: ListComp parse returned None\n") + + # --- Test 53: Dict comprehension --- + dictcomp_src: str = "d = {k: v for k, v in items}" + dc_mod: Module | CPtr = parse(dictcomp_src, pool) + if dc_mod is not None: + dump(dc_mod, dump_buf, DUMP_BUF_SIZE) + has_dc: str = string.strstr(dump_buf, "DictComp(") + if has_dc is not None: + passed += 1 + printf("PASS: DictComp parsed\n") + else: + failed += 1 + printf("FAIL: DictComp parse, dc=%d\n", has_dc is not None) + else: + failed += 1 + printf("FAIL: DictComp parse returned None\n") + + # --- Test 54: Set comprehension --- + setcomp_src: str = "s = {x for x in items}" + sc_mod: Module | CPtr = parse(setcomp_src, pool) + if sc_mod is not None: + dump(sc_mod, dump_buf, DUMP_BUF_SIZE) + has_sc: str = string.strstr(dump_buf, "SetComp(") + if has_sc is not None: + passed += 1 + printf("PASS: SetComp parsed\n") + else: + failed += 1 + printf("FAIL: SetComp parse, sc=%d\n", has_sc is not None) + else: + failed += 1 + printf("FAIL: SetComp parse returned None\n") + + # --- Test 55: With 语句 --- + with_src: str = """with open('f') as f: + data = f.read() +""" + with_mod: Module | CPtr = parse(with_src, pool) + if with_mod is not None and with_mod.children is not None: + dump(with_mod, dump_buf, DUMP_BUF_SIZE) + has_with: str = string.strstr(dump_buf, "With(") + has_witem: str = string.strstr(dump_buf, "WithItem(") + if has_with is not None and has_witem is not None: + passed += 1 + printf("PASS: With + WithItem parsed\n") + else: + failed += 1 + printf("FAIL: With parse, with=%d witem=%d\n", + has_with is not None, has_witem is not None) + else: + failed += 1 + printf("FAIL: With parse returned None\n") + + # --- Test 56: Raise 语句 --- + raise_src: str = "raise ValueError('bad')" + raise_mod: Module | CPtr = parse(raise_src, pool) + if raise_mod is not None: + dump(raise_mod, dump_buf, DUMP_BUF_SIZE) + has_raise: str = string.strstr(dump_buf, "Raise(") + if has_raise is not None: + passed += 1 + printf("PASS: Raise parsed\n") + else: + failed += 1 + printf("FAIL: Raise parse, raise=%d\n", has_raise is not None) + else: + failed += 1 + printf("FAIL: Raise parse returned None\n") + + # --- Test 57: Assert 语句 --- + assert_src: str = "assert x > 0, 'x must be positive'" + assert_mod: Module | CPtr = parse(assert_src, pool) + if assert_mod is not None: + dump(assert_mod, dump_buf, DUMP_BUF_SIZE) + has_assert: str = string.strstr(dump_buf, "Assert(") + if has_assert is not None: + passed += 1 + printf("PASS: Assert parsed\n") + else: + failed += 1 + printf("FAIL: Assert parse, assert=%d\n", has_assert is not None) + else: + failed += 1 + printf("FAIL: Assert parse returned None\n") + + # --- Test 58: Del 语句 --- + del_src: str = "del x" + del_mod: Module | CPtr = parse(del_src, pool) + if del_mod is not None: + dump(del_mod, dump_buf, DUMP_BUF_SIZE) + has_del: str = string.strstr(dump_buf, "Delete(") + if has_del is not None: + passed += 1 + printf("PASS: Delete parsed\n") + else: + failed += 1 + printf("FAIL: Delete parse, del=%d\n", has_del is not None) + else: + failed += 1 + printf("FAIL: Delete parse returned None\n") + + # --- Test 59: Subscript + Slice --- + slice_src: str = "y = a[1:3]" + slice_mod: Module | CPtr = parse(slice_src, pool) + if slice_mod is not None: + dump(slice_mod, dump_buf, DUMP_BUF_SIZE) + has_sub: str = string.strstr(dump_buf, "Subscript(") + has_slice: str = string.strstr(dump_buf, "Slice(") + if has_sub is not None and has_slice is not None: + passed += 1 + printf("PASS: Subscript + Slice parsed\n") + else: + failed += 1 + printf("FAIL: Slice parse, sub=%d slice=%d\n", + has_sub is not None, has_slice is not None) + else: + failed += 1 + printf("FAIL: Slice parse returned None\n") + + # --- Test 60: Tuple 解包 --- + tuple_src: str = "a, b = 1, 2" + tuple_mod: Module | CPtr = parse(tuple_src, pool) + if tuple_mod is not None: + dump(tuple_mod, dump_buf, DUMP_BUF_SIZE) + has_tuple: str = string.strstr(dump_buf, "Tuple(") + if has_tuple is not None: + passed += 1 + printf("PASS: Tuple unpacking parsed\n") + else: + failed += 1 + printf("FAIL: Tuple parse, tuple=%d\n", has_tuple is not None) + else: + failed += 1 + printf("FAIL: Tuple parse returned None\n") + + # --- Test 61: 默认参数 + 关键字参数签名 --- + defdef_src: str = "def f(a, b=10): return a + b" + defdef_mod: Module | CPtr = parse(defdef_src, pool) + if defdef_mod is not None: + dump(defdef_mod, dump_buf, DUMP_BUF_SIZE) + has_fdef: str = string.strstr(dump_buf, "FunctionDef(name='f'") + has_def: str = string.strstr(dump_buf, "Constant(value=10)") + if has_fdef is not None and has_def is not None: + passed += 1 + printf("PASS: FunctionDef with default arg parsed\n") + else: + failed += 1 + printf("FAIL: defdef parse, fdef=%d def=%d\n", + has_fdef is not None, has_def is not None) + else: + failed += 1 + printf("FAIL: defdef parse returned None\n") + + # --- Test 62: 装饰器 --- + deco_src: str = """@my_dec +def f(): + pass +""" + deco_mod: Module | CPtr = parse(deco_src, pool) + if deco_mod is not None: + dump(deco_mod, dump_buf, DUMP_BUF_SIZE) + has_dec: str = string.strstr(dump_buf, "decorator_list=[Name(id='my_dec'") + if has_dec is not None: + passed += 1 + printf("PASS: Decorator parsed\n") + else: + failed += 1 + printf("FAIL: Decorator parse, dec=%d\n", has_dec is not None) + else: + failed += 1 + printf("FAIL: Decorator parse returned None\n") + + # --- Test 63: 链式赋值 --- + chain_src: str = "a = b = c = 42" + chain_mod: Module | CPtr = parse(chain_src, pool) + if chain_mod is not None: + dump(chain_mod, dump_buf, DUMP_BUF_SIZE) + has_chain: str = string.strstr(dump_buf, "Assign(targets=[Name(id='a'") + has_b: str = string.strstr(dump_buf, "Name(id='b'") + has_c: str = string.strstr(dump_buf, "Name(id='c'") + if has_chain is not None and has_b is not None and has_c is not None: + passed += 1 + printf("PASS: Chain assignment a=b=c=42 parsed\n") + else: + failed += 1 + printf("FAIL: Chain, a=%d b=%d c=%d\n", + has_chain is not None, has_b is not None, has_c is not None) + else: + failed += 1 + printf("FAIL: Chain parse returned None\n") + + # --- Test 64: 位运算 --- + bit_src: str = "y = (a | b) & c ^ d << 2" + bit_mod: Module | CPtr = parse(bit_src, pool) + if bit_mod is not None: + dump(bit_mod, dump_buf, DUMP_BUF_SIZE) + has_bor: str = string.strstr(dump_buf, "op=BitOr") + has_band: str = string.strstr(dump_buf, "op=BitAnd") + has_bxor: str = string.strstr(dump_buf, "op=BitXor") + if has_bor is not None and has_band is not None and has_bxor is not None: + passed += 1 + printf("PASS: Bit ops (| & ^) parsed\n") + else: + failed += 1 + printf("FAIL: Bit ops, bor=%d band=%d bxor=%d\n", + has_bor is not None, has_band is not None, has_bxor is not None) + else: + failed += 1 + printf("FAIL: Bit ops parse returned None\n") + + # --- Test 65: 多继承 + 多装饰器 --- + multi_src: str = """@dec1 +@dec2 +class C(Base1, Base2): + pass +""" + multi_mod: Module | CPtr = parse(multi_src, pool) + if multi_mod is not None: + dump(multi_mod, dump_buf, DUMP_BUF_SIZE) + m_dec1: str = string.strstr(dump_buf, "Name(id='dec1'") + m_dec2: str = string.strstr(dump_buf, "Name(id='dec2'") + m_b1: str = string.strstr(dump_buf, "Name(id='Base1'") + m_b2: str = string.strstr(dump_buf, "Name(id='Base2'") + if m_dec1 is not None and m_dec2 is not None and m_b1 is not None and m_b2 is not None: + passed += 1 + printf("PASS: Multi-decorator + multi-inheritance parsed\n") + else: + failed += 1 + printf("FAIL: multi, dec1=%d dec2=%d b1=%d b2=%d\n", + m_dec1 is not None, m_dec2 is not None, + m_b1 is not None, m_b2 is not None) + else: + failed += 1 + printf("FAIL: multi parse returned None\n") + + # --- Test 66: Global / Nonlocal --- + glob_src: str = """def f(): + global x + x = 1 +""" + glob_mod: Module | CPtr = parse(glob_src, pool) + if glob_mod is not None: + dump(glob_mod, dump_buf, DUMP_BUF_SIZE) + has_glob: str = string.strstr(dump_buf, "Global(") + if has_glob is not None: + passed += 1 + printf("PASS: Global statement parsed\n") + else: + failed += 1 + printf("FAIL: Global parse, glob=%d\n", has_glob is not None) + else: + failed += 1 + printf("FAIL: Global parse returned None\n") + + # --- Test 67: f-string (FormattedValue) --- + fstr_src: str = "s = f'value={x}'" + fstr_mod: Module | CPtr = parse(fstr_src, pool) + if fstr_mod is not None: + dump(fstr_mod, dump_buf, DUMP_BUF_SIZE) + has_fv: str = string.strstr(dump_buf, "FormattedValue(") + has_js: str = string.strstr(dump_buf, "JoinedStr(") + if has_fv is not None and has_js is not None: + passed += 1 + printf("PASS: f-string with FormattedValue + JoinedStr parsed\n") + else: + failed += 1 + printf("FAIL: fstring, fv=%d js=%d\n", + has_fv is not None, has_js is not None) + else: + failed += 1 + printf("FAIL: fstring parse returned None\n") + + # --- Test 68: 嵌套函数 + 闭包 --- + nested_src: str = """def outer(): + x = 10 + def inner(): + return x + 1 + return inner() +""" + nested_mod: Module | CPtr = parse(nested_src, pool) + if nested_mod is not None and nested_mod.children is not None: + dump(nested_mod, dump_buf, DUMP_BUF_SIZE) + n_outer: str = string.strstr(dump_buf, "FunctionDef(name='outer'") + n_inner: str = string.strstr(dump_buf, "FunctionDef(name='inner'") + if n_outer is not None and n_inner is not None: + passed += 1 + printf("PASS: Nested function definitions parsed\n") + else: + failed += 1 + printf("FAIL: nested, outer=%d inner=%d\n", + n_outer is not None, n_inner is not None) + else: + failed += 1 + printf("FAIL: nested parse returned None\n") + + # --- Test 69: Match 字面量整数模式 --- + m_lit_src: str = """match x: + case 1: + pass +""" + m_lit_mod: Module | CPtr = parse(m_lit_src, pool) + if m_lit_mod is not None: + dump(m_lit_mod, dump_buf, DUMP_BUF_SIZE) + has_mv: str = string.strstr(dump_buf, "MatchValue(") + if has_mv is not None: + passed += 1 + printf("PASS: Match literal int -> MatchValue\n") + else: + failed += 1 + printf("FAIL: Match literal int, mv=%d\n", has_mv is not None) + else: + failed += 1 + printf("FAIL: Match literal int parse returned None\n") + + # --- Test 70: Match 字面量字符串模式 --- + m_str_src: str = """match x: + case "hello": + pass +""" + m_str_mod: Module | CPtr = parse(m_str_src, pool) + if m_str_mod is not None: + dump(m_str_mod, dump_buf, DUMP_BUF_SIZE) + has_mv2: str = string.strstr(dump_buf, "MatchValue(") + if has_mv2 is not None: + passed += 1 + printf("PASS: Match literal str -> MatchValue\n") + else: + failed += 1 + printf("FAIL: Match literal str\n") + else: + failed += 1 + printf("FAIL: Match literal str parse returned None\n") + + # --- Test 71: Match 负数字面量模式 --- + m_neg_src: str = """match x: + case -1: + pass +""" + m_neg_mod: Module | CPtr = parse(m_neg_src, pool) + if m_neg_mod is not None: + dump(m_neg_mod, dump_buf, DUMP_BUF_SIZE) + has_mv3: str = string.strstr(dump_buf, "MatchValue(") + if has_mv3 is not None: + passed += 1 + printf("PASS: Match negative literal -> MatchValue\n") + else: + failed += 1 + printf("FAIL: Match negative literal\n") + else: + failed += 1 + printf("FAIL: Match negative literal parse returned None\n") + + # --- Test 72: Match None 单例模式 --- + m_none_src: str = """match x: + case None: + pass +""" + m_none_mod: Module | CPtr = parse(m_none_src, pool) + if m_none_mod is not None: + dump(m_none_mod, dump_buf, DUMP_BUF_SIZE) + has_ms: str = string.strstr(dump_buf, "MatchSingleton(") + if has_ms is not None: + passed += 1 + printf("PASS: Match None -> MatchSingleton\n") + else: + failed += 1 + printf("FAIL: Match None, ms=%d\n", has_ms is not None) + else: + failed += 1 + printf("FAIL: Match None parse returned None\n") + + # --- Test 73: Match True 单例模式 --- + m_true_src: str = """match x: + case True: + pass +""" + m_true_mod: Module | CPtr = parse(m_true_src, pool) + if m_true_mod is not None: + dump(m_true_mod, dump_buf, DUMP_BUF_SIZE) + has_ms2: str = string.strstr(dump_buf, "MatchSingleton(") + if has_ms2 is not None: + passed += 1 + printf("PASS: Match True -> MatchSingleton\n") + else: + failed += 1 + printf("FAIL: Match True\n") + else: + failed += 1 + printf("FAIL: Match True parse returned None\n") + + # --- Test 74: Match False 单例模式 --- + m_false_src: str = """match x: + case False: + pass +""" + m_false_mod: Module | CPtr = parse(m_false_src, pool) + if m_false_mod is not None: + dump(m_false_mod, dump_buf, DUMP_BUF_SIZE) + has_ms3: str = string.strstr(dump_buf, "MatchSingleton(") + if has_ms3 is not None: + passed += 1 + printf("PASS: Match False -> MatchSingleton\n") + else: + failed += 1 + printf("FAIL: Match False\n") + else: + failed += 1 + printf("FAIL: Match False parse returned None\n") + + # --- Test 75: Match 通配符 _ 模式 --- + m_wild_src: str = """match x: + case _: + pass +""" + m_wild_mod: Module | CPtr = parse(m_wild_src, pool) + if m_wild_mod is not None: + dump(m_wild_mod, dump_buf, DUMP_BUF_SIZE) + has_ma: str = string.strstr(dump_buf, "MatchAs(") + if has_ma is not None: + passed += 1 + printf("PASS: Match wildcard _ -> MatchAs\n") + else: + failed += 1 + printf("FAIL: Match wildcard, ma=%d\n", has_ma is not None) + else: + failed += 1 + printf("FAIL: Match wildcard parse returned None\n") + + # --- Test 76: Match 捕获变量模式 --- + m_cap_src: str = """match x: + case y: + pass +""" + m_cap_mod: Module | CPtr = parse(m_cap_src, pool) + if m_cap_mod is not None: + dump(m_cap_mod, dump_buf, DUMP_BUF_SIZE) + has_ma2: str = string.strstr(dump_buf, "name='y'") + if has_ma2 is not None: + passed += 1 + printf("PASS: Match capture y -> MatchAs(name='y')\n") + else: + failed += 1 + printf("FAIL: Match capture, ma=%d\n", has_ma2 is not None) + else: + failed += 1 + printf("FAIL: Match capture parse returned None\n") + + # --- Test 77: Match 值模式 (dotted name Color.RED) --- + m_val_src: str = """match x: + case Color.RED: + pass +""" + m_val_mod: Module | CPtr = parse(m_val_src, pool) + if m_val_mod is not None: + dump(m_val_mod, dump_buf, DUMP_BUF_SIZE) + has_mv4: str = string.strstr(dump_buf, "MatchValue(") + has_attr: str = string.strstr(dump_buf, "Attribute(") + if has_mv4 is not None and has_attr is not None: + passed += 1 + printf("PASS: Match Color.RED -> MatchValue + Attribute\n") + else: + failed += 1 + printf("FAIL: Match value, mv=%d attr=%d\n", + has_mv4 is not None, has_attr is not None) + else: + failed += 1 + printf("FAIL: Match value parse returned None\n") + + # --- Test 78: Match 序列模式 [1, 2] --- + m_seq_src: str = """match x: + case [1, 2]: + pass +""" + m_seq_mod: Module | CPtr = parse(m_seq_src, pool) + if m_seq_mod is not None: + dump(m_seq_mod, dump_buf, DUMP_BUF_SIZE) + has_msq: str = string.strstr(dump_buf, "MatchSequence(") + if has_msq is not None: + passed += 1 + printf("PASS: Match [1, 2] -> MatchSequence\n") + else: + failed += 1 + printf("FAIL: Match sequence, msq=%d\n", has_msq is not None) + else: + failed += 1 + printf("FAIL: Match sequence parse returned None\n") + + # --- Test 79: Match 序列带星号 [a, *rest] --- + m_star_src: str = """match x: + case [a, *rest]: + pass +""" + m_star_mod: Module | CPtr = parse(m_star_src, pool) + if m_star_mod is not None: + dump(m_star_mod, dump_buf, DUMP_BUF_SIZE) + has_mstar: str = string.strstr(dump_buf, "MatchStar(name='rest')") + if has_mstar is not None: + passed += 1 + printf("PASS: Match [a, *rest] -> MatchStar\n") + else: + failed += 1 + printf("FAIL: Match star, mstar=%d\n", has_mstar is not None) + else: + failed += 1 + printf("FAIL: Match star parse returned None\n") + + # --- Test 80: Match 类模式 Point(x, y) --- + m_cls_src: str = """match p: + case Point(x, y): + pass +""" + m_cls_mod: Module | CPtr = parse(m_cls_src, pool) + if m_cls_mod is not None: + dump(m_cls_mod, dump_buf, DUMP_BUF_SIZE) + has_mc: str = string.strstr(dump_buf, "MatchClass(") + if has_mc is not None: + passed += 1 + printf("PASS: Match Point(x, y) -> MatchClass\n") + else: + failed += 1 + printf("FAIL: Match class, mc=%d\n", has_mc is not None) + else: + failed += 1 + printf("FAIL: Match class parse returned None\n") + + # --- Test 81: Match 类模式带关键字 Point(x=0, y=0) --- + m_kwcls_src: str = """match p: + case Point(x=0, y=0): + pass +""" + m_kwcls_mod: Module | CPtr = parse(m_kwcls_src, pool) + if m_kwcls_mod is not None: + dump(m_kwcls_mod, dump_buf, DUMP_BUF_SIZE) + has_mc2: str = string.strstr(dump_buf, "MatchClass(") + if has_mc2 is not None: + passed += 1 + printf("PASS: Match Point(x=0, y=0) -> MatchClass with kwd\n") + else: + failed += 1 + printf("FAIL: Match class kwd, mc=%d\n", has_mc2 is not None) + else: + failed += 1 + printf("FAIL: Match class kwd parse returned None\n") + + # --- Test 82: Match OR 模式 1 | 2 --- + m_or_src: str = """match x: + case 1 | 2: + pass +""" + m_or_mod: Module | CPtr = parse(m_or_src, pool) + if m_or_mod is not None: + dump(m_or_mod, dump_buf, DUMP_BUF_SIZE) + has_mo: str = string.strstr(dump_buf, "MatchOr(") + if has_mo is not None: + passed += 1 + printf("PASS: Match 1 | 2 -> MatchOr\n") + else: + failed += 1 + printf("FAIL: Match OR, mo=%d\n", has_mo is not None) + else: + failed += 1 + printf("FAIL: Match OR parse returned None\n") + + # --- Test 83: Match AS 模式 [x] as p --- + m_as_src: str = """match x: + case [x] as p: + pass +""" + m_as_mod: Module | CPtr = parse(m_as_src, pool) + if m_as_mod is not None: + dump(m_as_mod, dump_buf, DUMP_BUF_SIZE) + has_mas: str = string.strstr(dump_buf, "MatchAs(pattern=") + has_msq2: str = string.strstr(dump_buf, "MatchSequence(") + if has_mas is not None and has_msq2 is not None: + passed += 1 + printf("PASS: Match [x] as p -> MatchAs(MatchSequence)\n") + else: + failed += 1 + printf("FAIL: Match AS, mas=%d msq=%d\n", + has_mas is not None, has_msq2 is not None) + else: + failed += 1 + printf("FAIL: Match AS parse returned None\n") + + # --- Test 84: Match guard 模式 case x if x > 0 --- + m_guard_src: str = """match x: + case x if x > 0: + pass +""" + m_guard_mod: Module | CPtr = parse(m_guard_src, pool) + if m_guard_mod is not None: + dump(m_guard_mod, dump_buf, DUMP_BUF_SIZE) + has_guard: str = string.strstr(dump_buf, "guard=Compare(") + if has_guard is not None: + passed += 1 + printf("PASS: Match x if x > 0 -> guard=Compare\n") + else: + failed += 1 + printf("FAIL: Match guard, guard=%d\n", has_guard is not None) + else: + failed += 1 + printf("FAIL: Match guard parse returned None\n") + + # --- Test 85: Match 映射模式 {"k": v} --- + m_map_src: str = """match x: + case {"k": v}: + pass +""" + m_map_mod: Module | CPtr = parse(m_map_src, pool) + if m_map_mod is not None: + dump(m_map_mod, dump_buf, DUMP_BUF_SIZE) + has_mm: str = string.strstr(dump_buf, "MatchMapping(") + if has_mm is not None: + passed += 1 + printf("PASS: Match {'k': v} -> MatchMapping\n") + else: + failed += 1 + printf("FAIL: Match mapping, mm=%d\n", has_mm is not None) + else: + failed += 1 + printf("FAIL: Match mapping parse returned None\n") + + # --- Test 86: Match 映射带 rest {"k": v, **rest} --- + m_map2_src: str = """match x: + case {"k": v, **rest}: + pass +""" + m_map2_mod: Module | CPtr = parse(m_map2_src, pool) + if m_map2_mod is not None: + dump(m_map2_mod, dump_buf, DUMP_BUF_SIZE) + has_mm2: str = string.strstr(dump_buf, "MatchMapping(") + has_rest: str = string.strstr(dump_buf, "rest='rest'") + if has_mm2 is not None and has_rest is not None: + passed += 1 + printf("PASS: Match {'k': v, **rest} -> MatchMapping with rest\n") + else: + failed += 1 + printf("FAIL: Match mapping rest, mm=%d rest=%d\n", + has_mm2 is not None, has_rest is not None) + else: + failed += 1 + printf("FAIL: Match mapping rest parse returned None\n") + + # --- Test 87: Match 组合模式 (1 | 2) as p --- + m_combo_src: str = """match x: + case (1 | 2) as p: + pass +""" + m_combo_mod: Module | CPtr = parse(m_combo_src, pool) + if m_combo_mod is not None: + dump(m_combo_mod, dump_buf, DUMP_BUF_SIZE) + has_combo_or: str = string.strstr(dump_buf, "MatchOr(") + has_combo_as: str = string.strstr(dump_buf, "MatchAs(") + if has_combo_or is not None and has_combo_as is not None: + passed += 1 + printf("PASS: Match (1|2) as p -> MatchAs(MatchOr)\n") + else: + failed += 1 + printf("FAIL: Match combo, or=%d as=%d\n", + has_combo_or is not None, has_combo_as is not None) + else: + failed += 1 + printf("FAIL: Match combo parse returned None\n") + + # --- Test 88: async def --- + adef_src: str = """async def f(): + pass +""" + adef_mod: Module | CPtr = parse(adef_src, pool) + if adef_mod is not None and adef_mod.children is not None: + adef_cnt: t.CSizeT = adef_mod.children.__len__() + if adef_cnt == 1: + adef_child: AST | CPtr = adef_mod.children.get(0) + if adef_child is not None and adef_child.kind() == ASTKind.FunctionDef: + adef_fd: FunctionDef | CPtr = (FunctionDef | CPtr)(adef_child) + if (adef_fd.flags & FLAG_IS_ASYNC) != 0: + passed += 1 + printf("PASS: async def f() -> FunctionDef with FLAG_IS_ASYNC\n") + else: + failed += 1 + printf("FAIL: async def flags=%d (no FLAG_IS_ASYNC)\n", adef_fd.flags) + else: + failed += 1 + printf("FAIL: async def child kind=%d\n", + adef_child.kind() if adef_child is not None else -1) + else: + failed += 1 + printf("FAIL: async def children=%zu (expected 1)\n", adef_cnt) + else: + failed += 1 + printf("FAIL: async def parse returned None\n") + + # --- Test 89: async for (inside async def) --- + afor_src: str = """async def f(): + async for x in y: + pass +""" + afor_mod: Module | CPtr = parse(afor_src, pool) + if afor_mod is not None and afor_mod.children is not None: + afor_cnt: t.CSizeT = afor_mod.children.__len__() + if afor_cnt == 1: + afor_child: AST | CPtr = afor_mod.children.get(0) + if afor_child is not None and afor_child.kind() == ASTKind.FunctionDef: + afor_fd2: FunctionDef | CPtr = (FunctionDef | CPtr)(afor_child) + if (afor_fd2.flags & FLAG_IS_ASYNC) != 0: + if afor_fd2.children is not None: + afor_inner: AST | CPtr = afor_fd2.children.get(0) + if afor_inner is not None and afor_inner.kind() == ASTKind.For: + afor_for: For | CPtr = (For | CPtr)(afor_inner) + if (afor_for.flags & FLAG_IS_ASYNC) != 0: + passed += 1 + printf("PASS: async for -> For with FLAG_IS_ASYNC\n") + else: + failed += 1 + printf("FAIL: async for flags=%d\n", afor_for.flags) + else: + failed += 1 + printf("FAIL: async for inner kind=%d\n", + afor_inner.kind() if afor_inner is not None else -1) + else: + failed += 1 + printf("FAIL: async for no children\n") + else: + failed += 1 + printf("FAIL: async for outer def flags=%d\n", afor_fd2.flags) + else: + failed += 1 + printf("FAIL: async for outer kind=%d\n", + afor_child.kind() if afor_child is not None else -1) + else: + failed += 1 + printf("FAIL: async for children=%zu\n", afor_cnt) + else: + failed += 1 + printf("FAIL: async for parse returned None\n") + + # --- Test 90: async with (inside async def) --- + awith_src: str = """async def f(): + async with ctx: + pass +""" + awith_mod: Module | CPtr = parse(awith_src, pool) + if awith_mod is not None and awith_mod.children is not None: + awith_cnt: t.CSizeT = awith_mod.children.__len__() + if awith_cnt == 1: + awith_child: AST | CPtr = awith_mod.children.get(0) + if awith_child is not None and awith_child.kind() == ASTKind.FunctionDef: + awith_fd: FunctionDef | CPtr = (FunctionDef | CPtr)(awith_child) + if (awith_fd.flags & FLAG_IS_ASYNC) != 0: + if awith_fd.children is not None: + awith_inner: AST | CPtr = awith_fd.children.get(0) + if awith_inner is not None and awith_inner.kind() == ASTKind.With: + awith_w: With | CPtr = (With | CPtr)(awith_inner) + if (awith_w.flags & FLAG_IS_ASYNC) != 0: + passed += 1 + printf("PASS: async with -> With with FLAG_IS_ASYNC\n") + else: + failed += 1 + printf("FAIL: async with flags=%d\n", awith_w.flags) + else: + failed += 1 + printf("FAIL: async with inner kind=%d\n", + awith_inner.kind() if awith_inner is not None else -1) + else: + failed += 1 + printf("FAIL: async with no children\n") + else: + failed += 1 + printf("FAIL: async with outer def flags=%d\n", awith_fd.flags) + else: + failed += 1 + printf("FAIL: async with outer kind=%d\n", + awith_child.kind() if awith_child is not None else -1) + else: + failed += 1 + printf("FAIL: async with children=%zu\n", awith_cnt) + else: + failed += 1 + printf("FAIL: async with parse returned None\n") + + # --- Test 91: Match 复合语句 (多 case + 多类型模式) --- + m_complex_src: str = """match cmd: + case "quit": + pass + case ["go", dir]: + pass + case Point(x=0, y=0): + pass + case _: + pass +""" + m_complex_mod: Module | CPtr = parse(m_complex_src, pool) + if m_complex_mod is not None: + dump(m_complex_mod, dump_buf, DUMP_BUF_SIZE) + has_c_mv: str = string.strstr(dump_buf, "MatchValue(") + has_c_msq: str = string.strstr(dump_buf, "MatchSequence(") + has_c_mc: str = string.strstr(dump_buf, "MatchClass(") + has_c_ma: str = string.strstr(dump_buf, "MatchAs(") + if has_c_mv is not None and has_c_msq is not None and \ + has_c_mc is not None and has_c_ma is not None: + passed += 1 + printf("PASS: Match complex (4 cases) -> all pattern types\n") + else: + failed += 1 + printf("FAIL: Match complex, mv=%d msq=%d mc=%d ma=%d\n", + has_c_mv is not None, has_c_msq is not None, + has_c_mc is not None, has_c_ma is not None) + else: + failed += 1 + printf("FAIL: Match complex parse returned None\n") + + # --- Test 92: async def + async for + async with 组合 --- + acombo_src: str = """async def f(): + async with ctx: + async for x in y: + pass +""" + acombo_mod: Module | CPtr = parse(acombo_src, pool) + if acombo_mod is not None and acombo_mod.children is not None: + acombo_cnt: t.CSizeT = acombo_mod.children.__len__() + if acombo_cnt == 1: + acombo_child: AST | CPtr = acombo_mod.children.get(0) + if acombo_child is not None and acombo_child.kind() == ASTKind.FunctionDef: + acombo_fd2: FunctionDef | CPtr = (FunctionDef | CPtr)(acombo_child) + if (acombo_fd2.flags & FLAG_IS_ASYNC) != 0: + passed += 1 + printf("PASS: async def + async with + async for combo\n") + else: + failed += 1 + printf("FAIL: async combo flags=%d\n", acombo_fd2.flags) + else: + failed += 1 + printf("FAIL: async combo kind=%d\n", + acombo_child.kind() if acombo_child is not None else -1) + else: + failed += 1 + printf("FAIL: async combo children=%zu\n", acombo_cnt) + else: + failed += 1 + printf("FAIL: async combo parse returned None\n") + + # --- 结果汇总 --- + printf("\n=== Results: %d passed, %d failed ===\n", passed, failed) + + stdlib.free(dump_buf) + stdlib.free(arena) + if failed > 0: + return 1 + return 0 diff --git a/Test/NewAstTest/output/4175f14c5fbef360.deps.json b/Test/NewAstTest/output/4175f14c5fbef360.deps.json new file mode 100644 index 0000000..56af556 --- /dev/null +++ b/Test/NewAstTest/output/4175f14c5fbef360.deps.json @@ -0,0 +1 @@ +{"ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "atom": "271ea3decb810db2", "ast.__init__": "34548789d646f432", "__init__": "34548789d646f432", "ast": "34548789d646f432", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "main.tokens": "b547ac4f380bddb6", "main.base": "5dab8cb390496d22", "main.exprs": "47767b5026a8ee15", "main.stmts": "657e182b27c2a022", "main.astaux": "4337fb260448bbe2", "main.match": "4dd6b3f1427d1cc5", "main.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/Test/NewAstTest/project.json b/Test/NewAstTest/project.json new file mode 100644 index 0000000..dacead0 --- /dev/null +++ b/Test/NewAstTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "NewAstTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "NewAstTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/NewAstTest/temp/0c212981c180e7fb.pyi b/Test/NewAstTest/temp/0c212981c180e7fb.pyi new file mode 100644 index 0000000..f432b06 --- /dev/null +++ b/Test/NewAstTest/temp/0c212981c180e7fb.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from ast.lexer.py +Module: ast.lexer +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, TokOp, _init_tables, _kw_lookup, _op_head, OpEntry, new_token, token_set_str, token_set_str_literal + +LEXER_MAX_INDENT: t.CDefine = 256 + +class Lexer: + src: str + pos: t.CSizeT + len: t.CSizeT + lineno: t.CInt + col: t.CInt + pool: memhub.MemManager | t.CPtr + indent_stack: t.CInt | t.CPtr + indent_top: t.CInt + paren_depth: t.CInt + tokens_head: Token | t.CPtr + tokens_tail: Token | t.CPtr + at_line_start: t.CInt + error_count: t.CInt + def __new__(self: Lexer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_lexer(pool: memhub.MemManager | t.CPtr) -> Lexer | t.CPtr: pass + +def _lexer_init(lx: Lexer | t.CPtr, src: str, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(lx: Lexer | t.CPtr, offset: t.CSizeT) -> t.CChar: pass + +def _peek_cur(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _advance(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _is_alpha(ch: t.CChar) -> t.CInt: pass + +def _is_digit(ch: t.CChar) -> t.CInt: pass + +def _is_alnum(ch: t.CChar) -> t.CInt: pass + +def _is_hex(ch: t.CChar) -> t.CInt: pass + +def _is_space(ch: t.CChar) -> t.CInt: pass + +def _emit(lx: Lexer | t.CPtr, tok: Token | t.CPtr) -> t.CInt: pass + +def _make_op_token(lx: Lexer | t.CPtr, op_id: t.CInt, length: t.CInt, lineno: t.CInt, col: t.CInt) -> Token | t.CPtr: pass + +def _handle_indent(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_name(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_number(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _emit_number(lx: Lexer | t.CPtr, start: t.CSizeT, start_lineno: t.CInt, start_col: t.CInt, is_float: t.CInt, is_complex: t.CInt, is_hex: t.CInt) -> t.CInt: pass + +def _read_string(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_operator(lx: Lexer | t.CPtr) -> t.CInt: pass + +def tokenize(lx: Lexer | t.CPtr) -> Token | t.CPtr: pass diff --git a/Test/NewAstTest/temp/271ea3decb810db2.pyi b/Test/NewAstTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/NewAstTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/NewAstTest/temp/34548789d646f432.pyi b/Test/NewAstTest/temp/34548789d646f432.pyi new file mode 100644 index 0000000..e046b0c --- /dev/null +++ b/Test/NewAstTest/temp/34548789d646f432.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from ast.__init__.py +Module: ast.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp, _init_tables +from .base import AST, ASTKind, ASTCtx, OpKind, ASTFlag, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC, FLAG_SIMPLE, FLAG_HAS_STAR, _init_ast, _copy_str, _set_pos, _inherit_pos, _set_parent_list, _binop_from_op, _augop_from_op, _cmpop_from_op, _emit, _emit_str, _emit_int, _dump_list, dump +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .astaux import ExceptHandler, Arguments, Arg, Keyword, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .visitor import ASTVisitor +from .lexer import Lexer, _lexer_init, tokenize, new_lexer +from .parser import Parser, _parser_init, parse_tokens, new_parser +from .parser import _parse_test + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def parse(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass + +def parse_expression(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass diff --git a/Test/NewAstTest/temp/4175f14c5fbef360.doc.json b/Test/NewAstTest/temp/4175f14c5fbef360.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/NewAstTest/temp/4175f14c5fbef360.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/NewAstTest/temp/4175f14c5fbef360.pyi b/Test/NewAstTest/temp/4175f14c5fbef360.pyi new file mode 100644 index 0000000..209a287 --- /dev/null +++ b/Test/NewAstTest/temp/4175f14c5fbef360.pyi @@ -0,0 +1,17 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from t import CInt, CExport, CPtr +from stdio import printf +import memhub +import stdlib +import string +from ast import AST, ASTKind, Module, FunctionDef, Assign, If, For, While, With, Return, Expr, Name, Constant, BinOp, UnaryOp, Call, Compare, Import, ImportFrom, Expression, CONST_INT, CONST_STR, CONST_BOOL, CONST_NONE, OpKind, ASTCtx, FLAG_IS_ASYNC, parse, parse_expression, dump + +ARENA_SIZE: t.CExtern | CInt + +def main() -> CInt | CExport: pass diff --git a/Test/NewAstTest/temp/4337fb260448bbe2.pyi b/Test/NewAstTest/temp/4337fb260448bbe2.pyi new file mode 100644 index 0000000..7ed88bd --- /dev/null +++ b/Test/NewAstTest/temp/4337fb260448bbe2.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from ast.astaux.py +Module: ast.astaux +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class ExceptHandler(AST): + type: AST | t.CPtr + name: str + def __new__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: ExceptHandler) -> t.CInt: pass + def type_name(self: ExceptHandler) -> str: pass + def dump(self: ExceptHandler, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arguments(AST): + args: list[AST | t.CPtr] | t.CPtr + vararg: AST | t.CPtr + kwarg: AST | t.CPtr + defaults: list[AST | t.CPtr] | t.CPtr + kw_defaults: list[AST | t.CPtr] | t.CPtr + def __new__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Arguments) -> t.CInt: pass + def type_name(self: Arguments) -> str: pass + def dump(self: Arguments, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arg(AST): + arg: str + annotation: AST | t.CPtr + def __new__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def kind(self: Arg) -> t.CInt: pass + def type_name(self: Arg) -> str: pass + def dump(self: Arg, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Keyword(AST): + arg: str + value: AST | t.CPtr + def __new__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Keyword) -> t.CInt: pass + def type_name(self: Keyword) -> str: pass + def dump(self: Keyword, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Alias(AST): + name: str + asname: str + def __new__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def __init__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def kind(self: Alias) -> t.CInt: pass + def type_name(self: Alias) -> str: pass + def dump(self: Alias, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class WithItem(AST): + context_expr: AST | t.CPtr + optional_vars: AST | t.CPtr + def __new__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def __init__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def kind(self: WithItem) -> t.CInt: pass + def type_name(self: WithItem) -> str: pass + def dump(self: WithItem, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Comprehension(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + ifs: list[AST | t.CPtr] | t.CPtr + is_async: t.CInt + def __new__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: Comprehension) -> t.CInt: pass + def type_name(self: Comprehension) -> str: pass + def dump(self: Comprehension, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/47767b5026a8ee15.pyi b/Test/NewAstTest/temp/47767b5026a8ee15.pyi new file mode 100644 index 0000000..7890f3d --- /dev/null +++ b/Test/NewAstTest/temp/47767b5026a8ee15.pyi @@ -0,0 +1,241 @@ +""" +Auto-generated Python stub file from ast.exprs.py +Module: ast.exprs +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, _dump_op_list, _op_name, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE + +class Name(AST): + id: str + ctx: t.CInt + def __new__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Name) -> t.CInt: pass + def type_name(self: Name) -> str: pass + def dump(self: Name, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Constant(AST): + int_val: t.CInt64T + float_val: t.CDouble + str_val: str + const_kind: t.CInt + def __new__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def __init__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def kind(self: Constant) -> t.CInt: pass + def type_name(self: Constant) -> str: pass + def dump(self: Constant, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BinOp(AST): + left: AST | t.CPtr + op: t.CInt + right: AST | t.CPtr + def __new__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def __init__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def kind(self: BinOp) -> t.CInt: pass + def type_name(self: BinOp) -> str: pass + def dump(self: BinOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class UnaryOp(AST): + op: t.CInt + operand: AST | t.CPtr + def __new__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def __init__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def kind(self: UnaryOp) -> t.CInt: pass + def type_name(self: UnaryOp) -> str: pass + def dump(self: UnaryOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Compare(AST): + left: AST | t.CPtr + ops: t.CPtr + comparators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Compare) -> t.CInt: pass + def type_name(self: Compare) -> str: pass + def dump(self: Compare, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Call(AST): + func: AST | t.CPtr + args: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + def __new__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Call) -> t.CInt: pass + def type_name(self: Call) -> str: pass + def dump(self: Call, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BoolOp(AST): + op: t.CInt + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: BoolOp) -> t.CInt: pass + def type_name(self: BoolOp) -> str: pass + def dump(self: BoolOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Lambda(AST): + args: AST | t.CPtr + body: AST | t.CPtr + def __new__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Lambda) -> t.CInt: pass + def type_name(self: Lambda) -> str: pass + def dump(self: Lambda, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class IfExp(AST): + test: AST | t.CPtr + body: AST | t.CPtr + orelse: AST | t.CPtr + def __new__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def __init__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def kind(self: IfExp) -> t.CInt: pass + def type_name(self: IfExp) -> str: pass + def dump(self: IfExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Dict(AST): + keys: list[AST | t.CPtr] | t.CPtr + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Dict) -> t.CInt: pass + def type_name(self: Dict) -> str: pass + def dump(self: Dict, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Set(AST): + elts: list[AST | t.CPtr] | t.CPtr + def __new__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Set) -> t.CInt: pass + def type_name(self: Set) -> str: pass + def dump(self: Set, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ListComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: ListComp) -> t.CInt: pass + def type_name(self: ListComp) -> str: pass + def dump(self: ListComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class SetComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: SetComp) -> t.CInt: pass + def type_name(self: SetComp) -> str: pass + def dump(self: SetComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class DictComp(AST): + key: AST | t.CPtr + value: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: DictComp) -> t.CInt: pass + def type_name(self: DictComp) -> str: pass + def dump(self: DictComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class GeneratorExp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: GeneratorExp) -> t.CInt: pass + def type_name(self: GeneratorExp) -> str: pass + def dump(self: GeneratorExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Await(AST): + value: AST | t.CPtr + def __new__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Await) -> t.CInt: pass + def type_name(self: Await) -> str: pass + def dump(self: Await, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Yield(AST): + value: AST | t.CPtr + def __new__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Yield) -> t.CInt: pass + def type_name(self: Yield) -> str: pass + def dump(self: Yield, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class YieldFrom(AST): + value: AST | t.CPtr + def __new__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: YieldFrom) -> t.CInt: pass + def type_name(self: YieldFrom) -> str: pass + def dump(self: YieldFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FormattedValue(AST): + value: AST | t.CPtr + conversion: t.CInt + format_spec: AST | t.CPtr + def __new__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def kind(self: FormattedValue) -> t.CInt: pass + def type_name(self: FormattedValue) -> str: pass + def dump(self: FormattedValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class JoinedStr(AST): + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: JoinedStr) -> t.CInt: pass + def type_name(self: JoinedStr) -> str: pass + def dump(self: JoinedStr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Attribute(AST): + value: AST | t.CPtr + attr: str + ctx: t.CInt + def __new__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Attribute) -> t.CInt: pass + def type_name(self: Attribute) -> str: pass + def dump(self: Attribute, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Subscript(AST): + value: AST | t.CPtr + slice: AST | t.CPtr + ctx: t.CInt + def __new__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Subscript) -> t.CInt: pass + def type_name(self: Subscript) -> str: pass + def dump(self: Subscript, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Starred(AST): + value: AST | t.CPtr + ctx: t.CInt + def __new__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Starred) -> t.CInt: pass + def type_name(self: Starred) -> str: pass + def dump(self: Starred, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class List(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: List) -> t.CInt: pass + def type_name(self: List) -> str: pass + def dump(self: List, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Tuple(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Tuple) -> t.CInt: pass + def type_name(self: Tuple) -> str: pass + def dump(self: Tuple, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Slice(AST): + lower: AST | t.CPtr + upper: AST | t.CPtr + step: AST | t.CPtr + def __new__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def kind(self: Slice) -> t.CInt: pass + def type_name(self: Slice) -> str: pass + def dump(self: Slice, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class NamedExpr(AST): + target: AST | t.CPtr + value: AST | t.CPtr + def __new__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: NamedExpr) -> t.CInt: pass + def type_name(self: NamedExpr) -> str: pass + def dump(self: NamedExpr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class OpNode(AST): + op: t.CInt + def __new__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def __init__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def kind(self: OpNode) -> t.CInt: pass + def type_name(self: OpNode) -> str: pass + def dump(self: OpNode, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/4dd6b3f1427d1cc5.pyi b/Test/NewAstTest/temp/4dd6b3f1427d1cc5.pyi new file mode 100644 index 0000000..dd15aa2 --- /dev/null +++ b/Test/NewAstTest/temp/4dd6b3f1427d1cc5.pyi @@ -0,0 +1,82 @@ +""" +Auto-generated Python stub file from ast.match.py +Module: ast.match +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class MatchCase(AST): + pattern: AST | t.CPtr + guard: AST | t.CPtr + def __new__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchCase) -> t.CInt: pass + def type_name(self: MatchCase) -> str: pass + def dump(self: MatchCase, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchValue(AST): + value: AST | t.CPtr + def __new__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchValue) -> t.CInt: pass + def type_name(self: MatchValue) -> str: pass + def dump(self: MatchValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSingleton(AST): + value: AST | t.CPtr + def __new__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchSingleton) -> t.CInt: pass + def type_name(self: MatchSingleton) -> str: pass + def dump(self: MatchSingleton, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSequence(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchSequence) -> t.CInt: pass + def type_name(self: MatchSequence) -> str: pass + def dump(self: MatchSequence, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchMapping(AST): + keys: list[AST | t.CPtr] | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + rest: str + def __new__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def __init__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def kind(self: MatchMapping) -> t.CInt: pass + def type_name(self: MatchMapping) -> str: pass + def dump(self: MatchMapping, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchClass(AST): + cls: AST | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + kwd_attrs: list[AST | t.CPtr] | t.CPtr + kwd_patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchClass) -> t.CInt: pass + def type_name(self: MatchClass) -> str: pass + def dump(self: MatchClass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchStar(AST): + name: str + def __new__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchStar) -> t.CInt: pass + def type_name(self: MatchStar) -> str: pass + def dump(self: MatchStar, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchAs(AST): + pattern: AST | t.CPtr + name: str + def __new__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchAs) -> t.CInt: pass + def type_name(self: MatchAs) -> str: pass + def dump(self: MatchAs, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchOr(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchOr) -> t.CInt: pass + def type_name(self: MatchOr) -> str: pass + def dump(self: MatchOr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/5dab8cb390496d22.pyi b/Test/NewAstTest/temp/5dab8cb390496d22.pyi new file mode 100644 index 0000000..7056767 --- /dev/null +++ b/Test/NewAstTest/temp/5dab8cb390496d22.pyi @@ -0,0 +1,182 @@ +""" +Auto-generated Python stub file from ast.base.py +Module: ast.base +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .tokens import TokOp + +class ASTKind(t.CEnum): + Module: t.State + FunctionDef: t.State + ClassDef: t.State + Assign: t.State + If: t.State + For: t.State + While: t.State + Return: t.State + Expr: t.State + Name: t.State + Constant: t.State + BinOp: t.State + UnaryOp: t.State + Call: t.State + Compare: t.State + Expression: t.State + Interactive: t.State + FunctionType: t.State + Delete: t.State + AugAssign: t.State + AnnAssign: t.State + With: t.State + Raise: t.State + Try: t.State + Assert: t.State + Global: t.State + Nonlocal: t.State + Pass: t.State + Break: t.State + Continue: t.State + Import: t.State + ImportFrom: t.State + Match: t.State + BoolOp: t.State + Lambda: t.State + IfExp: t.State + Dict: t.State + Set: t.State + ListComp: t.State + SetComp: t.State + DictComp: t.State + GeneratorExp: t.State + Await: t.State + Yield: t.State + YieldFrom: t.State + FormattedValue: t.State + JoinedStr: t.State + Attribute: t.State + Subscript: t.State + Starred: t.State + List: t.State + Tuple: t.State + Slice: t.State + NamedExpr: t.State + ExceptHandler: t.State + Arguments: t.State + Arg: t.State + Keyword: t.State + Alias: t.State + WithItem: t.State + Comprehension: t.State + OpNode: t.State + MatchCase: t.State + MatchValue: t.State + MatchSingleton: t.State + MatchSequence: t.State + MatchMapping: t.State + MatchClass: t.State + MatchStar: t.State + MatchAs: t.State + MatchOr: t.State +class ASTCtx(t.CEnum): + Load: t.State + Store: t.State + Del: t.State +class OpKind(t.CEnum): + Add: t.State + Sub: t.State + Mult: t.State + MatMult: t.State + Div: t.State + Mod: t.State + Pow: t.State + LShift: t.State + RShift: t.State + BitOr: t.State + BitXor: t.State + BitAnd: t.State + FloorDiv: t.State + And: t.State + Or: t.State + Eq: t.State + Ne: t.State + Lt: t.State + Le: t.State + Gt: t.State + Ge: t.State + Is: t.State + IsNot: t.State + In: t.State + NotIn: t.State + Not: t.State + UAdd: t.State + USub: t.State + Invert: t.State + NoneOp: t.State + +CONST_INT: t.CDefine = 1 +CONST_FLOAT: t.CDefine = 2 +CONST_STR: t.CDefine = 3 +CONST_BOOL: t.CDefine = 4 +CONST_NONE: t.CDefine = 5 +FLAG_IS_ASYNC: t.CDefine = 1 +FLAG_SIMPLE: t.CDefine = 2 +FLAG_HAS_STAR: t.CDefine = 4 + +class ASTFlag(t.CEnum): + IsAsync: t.State + Simple: t.State + HasStar: t.State + +def _copy_str(pool: memhub.MemManager | t.CPtr, src: str) -> str: pass + +def _set_pos(node: AST | t.CPtr, lineno: t.CInt, col_offset: t.CInt, end_lineno: t.CInt, end_col_offset: t.CInt) -> t.CInt: pass + +def _inherit_pos(node: AST | t.CPtr, ref: AST | t.CPtr) -> t.CInt: pass + +def _init_ast(node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + +def _set_parent_list(lst: list[AST | t.CPtr] | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _append_child(lst: list[AST | t.CPtr] | t.CPtr, node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _binop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _augop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _cmpop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _emit(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_str(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_int(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, val: t.CInt64T) -> t.CSizeT: pass + +def _op_name(op: t.CInt) -> str: pass + +def _dump_list(lst: list[AST | t.CPtr] | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + +def _dump_op_list(lst: t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + + +@t.CVTable +class AST: + parent: AST | t.CPtr + pool: memhub.MemManager | t.CPtr + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + children: list[AST | t.CPtr] | t.CPtr + def kind(self: AST) -> t.CInt: pass + def type_name(self: AST) -> str: pass + def dump(self: AST, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + def accept(self: AST, visitor: t.CPtr) -> t.CInt: pass + def append(self: AST, node: AST | t.CPtr) -> t.CInt: pass + +def dump(node: AST | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: pass diff --git a/Test/NewAstTest/temp/657e182b27c2a022.pyi b/Test/NewAstTest/temp/657e182b27c2a022.pyi new file mode 100644 index 0000000..0a2057d --- /dev/null +++ b/Test/NewAstTest/temp/657e182b27c2a022.pyi @@ -0,0 +1,226 @@ +""" +Auto-generated Python stub file from ast.stmts.py +Module: ast.stmts +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, FLAG_IS_ASYNC + +@t.CVTable +class Module(AST): + def __new__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Module) -> t.CInt: pass + def type_name(self: Module) -> str: pass + def dump(self: Module, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assign(AST): + targets: list[AST | t.CPtr] | t.CPtr + value: AST | t.CPtr + def __new__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assign) -> t.CInt: pass + def type_name(self: Assign) -> str: pass + def dump(self: Assign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class If(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: If) -> t.CInt: pass + def type_name(self: If) -> str: pass + def dump(self: If, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class For(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: For) -> t.CInt: pass + def type_name(self: For) -> str: pass + def dump(self: For, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class While(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: While) -> t.CInt: pass + def type_name(self: While) -> str: pass + def dump(self: While, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Return(AST): + value: AST | t.CPtr + def __new__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Return) -> t.CInt: pass + def type_name(self: Return) -> str: pass + def dump(self: Return, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expr(AST): + value: AST | t.CPtr + def __new__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expr) -> t.CInt: pass + def type_name(self: Expr) -> str: pass + def dump(self: Expr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionDef(AST): + name: str + args: AST | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + flags: t.CInt + def __new__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: FunctionDef) -> t.CInt: pass + def type_name(self: FunctionDef) -> str: pass + def dump(self: FunctionDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expression(AST): + body: AST | t.CPtr + def __new__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expression) -> t.CInt: pass + def type_name(self: Expression) -> str: pass + def dump(self: Expression, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Interactive(AST): + body: AST | t.CPtr + def __new__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Interactive) -> t.CInt: pass + def type_name(self: Interactive) -> str: pass + def dump(self: Interactive, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionType(AST): + argtypes: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + def __new__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def kind(self: FunctionType) -> t.CInt: pass + def type_name(self: FunctionType) -> str: pass + def dump(self: FunctionType, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ClassDef(AST): + name: str + bases: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + type_params: list[str] | t.CPtr + def __new__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def __init__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def kind(self: ClassDef) -> t.CInt: pass + def type_name(self: ClassDef) -> str: pass + def dump(self: ClassDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Delete(AST): + targets: list[AST | t.CPtr] | t.CPtr + def __new__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Delete) -> t.CInt: pass + def type_name(self: Delete) -> str: pass + def dump(self: Delete, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AugAssign(AST): + target: AST | t.CPtr + op: t.CInt + value: AST | t.CPtr + def __new__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: AugAssign) -> t.CInt: pass + def type_name(self: AugAssign) -> str: pass + def dump(self: AugAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AnnAssign(AST): + target: AST | t.CPtr + annotation: AST | t.CPtr + value: AST | t.CPtr + simple: t.CInt + def __new__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def __init__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def kind(self: AnnAssign) -> t.CInt: pass + def type_name(self: AnnAssign) -> str: pass + def dump(self: AnnAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class With(AST): + items: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: With) -> t.CInt: pass + def type_name(self: With) -> str: pass + def dump(self: With, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Raise(AST): + exc: AST | t.CPtr + cause: AST | t.CPtr + def __new__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def kind(self: Raise) -> t.CInt: pass + def type_name(self: Raise) -> str: pass + def dump(self: Raise, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Try(AST): + handlers: list[AST | t.CPtr] | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + finalbody: list[AST | t.CPtr] | t.CPtr + def __new__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Try) -> t.CInt: pass + def type_name(self: Try) -> str: pass + def dump(self: Try, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assert(AST): + test: AST | t.CPtr + msg: AST | t.CPtr + def __new__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assert) -> t.CInt: pass + def type_name(self: Assert) -> str: pass + def dump(self: Assert, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Global(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Global) -> t.CInt: pass + def type_name(self: Global) -> str: pass + def dump(self: Global, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Nonlocal(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Nonlocal) -> t.CInt: pass + def type_name(self: Nonlocal) -> str: pass + def dump(self: Nonlocal, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Pass(AST): + def __new__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Pass) -> t.CInt: pass + def type_name(self: Pass) -> str: pass + def dump(self: Pass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Break(AST): + def __new__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Break) -> t.CInt: pass + def type_name(self: Break) -> str: pass + def dump(self: Break, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Continue(AST): + def __new__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Continue) -> t.CInt: pass + def type_name(self: Continue) -> str: pass + def dump(self: Continue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Import(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Import) -> t.CInt: pass + def type_name(self: Import) -> str: pass + def dump(self: Import, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ImportFrom(AST): + module: str + names: list[AST | t.CPtr] | t.CPtr + level: t.CInt + def __new__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def __init__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def kind(self: ImportFrom) -> t.CInt: pass + def type_name(self: ImportFrom) -> str: pass + def dump(self: ImportFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Match(AST): + subject: AST | t.CPtr + cases: list[AST | t.CPtr] | t.CPtr + def __new__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Match) -> t.CInt: pass + def type_name(self: Match) -> str: pass + def dump(self: Match, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/6f62fe05c5ea1ceb.pyi b/Test/NewAstTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/NewAstTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/NewAstTest/temp/71e0a3ffcb3ebfad.pyi b/Test/NewAstTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/NewAstTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/NewAstTest/temp/90c53dd6db8d41cf.pyi b/Test/NewAstTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/NewAstTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/NewAstTest/temp/_phase1_manifest.json b/Test/NewAstTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..3206144 --- /dev/null +++ b/Test/NewAstTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\NewAstTest\\App\\main.py": {"sha1": "4175f14c5fbef360", "mtime": 1783001020.789755, "size": 63423}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\__init__.py": {"sha1": "34548789d646f432", "mtime": 1784102404.6307266, "size": 3409}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\astaux.py": {"sha1": "4337fb260448bbe2", "mtime": 1784006094.7981195, "size": 11489}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\base.py": {"sha1": "5dab8cb390496d22", "mtime": 1784197698.3282094, "size": 14966}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\exprs.py": {"sha1": "47767b5026a8ee15", "mtime": 1783239758.809719, "size": 40615}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\lexer.py": {"sha1": "0c212981c180e7fb", "mtime": 1783400609.908198, "size": 23535}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\match.py": {"sha1": "4dd6b3f1427d1cc5", "mtime": 1782961799.8492737, "size": 12309}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\parser.py": {"sha1": "ee52c08239684346", "mtime": 1784102405.1933146, "size": 85445}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\stmts.py": {"sha1": "657e182b27c2a022", "mtime": 1783929993.759902, "size": 37926}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\tokens.py": {"sha1": "b547ac4f380bddb6", "mtime": 1783165480.469121, "size": 13722}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}} \ No newline at end of file diff --git a/Test/NewAstTest/temp/_sha1_map.txt b/Test/NewAstTest/temp/_sha1_map.txt new file mode 100644 index 0000000..ac2e3b4 --- /dev/null +++ b/Test/NewAstTest/temp/_sha1_map.txt @@ -0,0 +1,20 @@ +0c212981c180e7fb:includes/ast\lexer.py +271ea3decb810db2:includes/atom.py +34548789d646f432:includes/ast\__init__.py +4175f14c5fbef360:main.py +4337fb260448bbe2:includes/ast\astaux.py +47767b5026a8ee15:includes/ast\exprs.py +4dd6b3f1427d1cc5:includes/ast\match.py +5dab8cb390496d22:includes/ast\base.py +657e182b27c2a022:includes/ast\stmts.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +b547ac4f380bddb6:includes/ast\tokens.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +ee084e9fc6ee413a:includes/memhub.py +ee52c08239684346:includes/ast\parser.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/NewAstTest/temp/ab6e54ba9a669f76.pyi b/Test/NewAstTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/NewAstTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/NewAstTest/temp/b547ac4f380bddb6.pyi b/Test/NewAstTest/temp/b547ac4f380bddb6.pyi new file mode 100644 index 0000000..e960341 --- /dev/null +++ b/Test/NewAstTest/temp/b547ac4f380bddb6.pyi @@ -0,0 +1,166 @@ +""" +Auto-generated Python stub file from ast.tokens.py +Module: ast.tokens +""" + + +import t, c +from stdint import * +import memhub +import string + +class TokenType(t.CEnum): + EndMarker: t.State + NewLine: t.State + Indent: t.State + Dedent: t.State + Name: t.State + Number: t.State + String: t.State + FStringStart: t.State + FStringMiddle: t.State + FStringEnd: t.State + Op: t.State + Nl: t.State + Comment: t.State + Encoding: t.State +class Keyword(t.CEnum): + NotKeyword: t.State + False_: t.State + None_: t.State + True_: t.State + And: t.State + As: t.State + Assert: t.State + Async: t.State + Await: t.State + Break: t.State + Class: t.State + Continue: t.State + Def: t.State + Del: t.State + Elif: t.State + Else: t.State + Except: t.State + Finally: t.State + For: t.State + From: t.State + Global: t.State + If: t.State + Import: t.State + In: t.State + Is: t.State + Lambda: t.State + Nonlocal: t.State + Not: t.State + Or: t.State + Pass: t.State + Raise: t.State + Return: t.State + Try: t.State + While: t.State + With: t.State + Yield: t.State + Match: t.State + Case: t.State + Type: t.State +class TokOp(t.CEnum): + LPar: t.State + RPar: t.State + Lsqb: t.State + Rsqb: t.State + LBrace: t.State + RBrace: t.State + Comma: t.State + Colon: t.State + Dot: t.State + Semi: t.State + At: t.State + Equal: t.State + RArrow: t.State + PlusEq: t.State + MinusEq: t.State + StarEq: t.State + SlashEq: t.State + DSlashEq: t.State + PercentEq: t.State + AtEq: t.State + AmpEq: t.State + VBarEq: t.State + CaretEq: t.State + GtGtEq: t.State + LtLtEq: t.State + StarEqEq: t.State + DSlash: t.State + StarStar: t.State + LtLt: t.State + GtGt: t.State + LessEq: t.State + GreaterEq: t.State + EqEq: t.State + ExclaimEq: t.State + Less: t.State + Greater: t.State + Plus: t.State + Minus: t.State + Star: t.State + Slash: t.State + Percent: t.State + Amp: t.State + VBar: t.State + Caret: t.State + Tilde: t.State + ColonEq: t.State + Ellipsis: t.State + Bang: t.State +class Token: + type: t.CInt + op_subtype: t.CInt + kw_subtype: t.CInt + str_val: str + int_val: t.CInt64T + float_val: t.CDouble + is_float: t.CInt + is_complex: t.CInt + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + next: Token | t.CPtr + def __new__(self: Token, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass +class KwEntry: + name: str + kw_id: t.CInt + next: KwEntry | t.CPtr + def __new__(self: KwEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_kw_head: t.CExtern | KwEntry | t.CPtr +_kw_pool: t.CExtern | memhub.MemManager | t.CPtr + +def _kw_intern(pool: memhub.MemManager | t.CPtr, name: str, kw_id: t.CInt) -> t.CInt: pass + +def _kw_lookup(name: str) -> t.CInt: pass + +def _init_keywords(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + + +class OpEntry: + name: str + op_id: t.CInt + length: t.CInt + next: OpEntry | t.CPtr + def __new__(self: OpEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_op_head: t.CExtern | OpEntry | t.CPtr + +def _op_intern(pool: memhub.MemManager | t.CPtr, name: str, op_id: t.CInt) -> t.CInt: pass + +def _init_operators(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _init_tables(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_token(pool: memhub.MemManager | t.CPtr, ttype: t.CInt, lineno: t.CInt, col_offset: t.CInt) -> Token | t.CPtr: pass + +def token_set_str(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, src: str, start: t.CSizeT, length: t.CSizeT) -> t.CInt: pass + +def token_set_str_literal(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, text: str) -> t.CInt: pass diff --git a/Test/NewAstTest/temp/c3b259b4059f8668.pyi b/Test/NewAstTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/NewAstTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/NewAstTest/temp/c9f4be41ca1cc2b4.pyi b/Test/NewAstTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/NewAstTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/ee084e9fc6ee413a.pyi b/Test/NewAstTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/NewAstTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/ee52c08239684346.pyi b/Test/NewAstTest/temp/ee52c08239684346.pyi new file mode 100644 index 0000000..ee09339 --- /dev/null +++ b/Test/NewAstTest/temp/ee52c08239684346.pyi @@ -0,0 +1,215 @@ +""" +Auto-generated Python stub file from ast.parser.py +Module: ast.parser +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp +from .base import AST, ASTCtx, ASTKind, OpKind, ASTFlag, _set_pos, _binop_from_op, _augop_from_op, _cmpop_from_op, _copy_str, _inherit_pos, _init_ast, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .astaux import ExceptHandler, Arguments, Arg, Keyword as KwNode, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .lexer import Lexer, _lexer_init, tokenize, new_lexer + +class Parser: + tokens: Token | t.CPtr + cur: Token | t.CPtr + pool: memhub.MemManager | t.CPtr + error_count: t.CInt + pending_decorators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Parser, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_parser(pool: memhub.MemManager | t.CPtr) -> Parser | t.CPtr: pass + +def _parser_init(ps: Parser | t.CPtr, tokens: Token | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _advance(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_type(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_kw(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_str(ps: Parser | t.CPtr) -> str: pass + +def _cur_lineno(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_col(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nl(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nls_and_newlines(ps: Parser | t.CPtr) -> t.CInt: pass + +def _expect_type(ps: Parser | t.CPtr, ttype: t.CInt) -> t.CInt: pass + +def _expect_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _match_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _accept_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _new_const_int(pool: memhub.MemManager | t.CPtr, val: t.CInt64T, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_float(pool: memhub.MemManager | t.CPtr, val: t.CDouble, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_str(pool: memhub.MemManager | t.CPtr, val: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_bool(pool: memhub.MemManager | t.CPtr, val: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_none(pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _set_ctx(node: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + +def _fix_store_ctx(node: AST | t.CPtr) -> t.CInt: pass + +def _parse_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_namedexpr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_lambda(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_or_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_and_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_not_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_comparison(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _match_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _advance_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _parse_bitor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitxor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitand(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_shift(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_arith(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_term(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_factor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_power(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_call(ps: Parser | t.CPtr, func: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_subscript(ps: Parser | t.CPtr, value: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_slice(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_paren(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_brace(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sub_expr(ps: Parser | t.CPtr, src: str) -> AST | t.CPtr: pass + +def _parse_fstring(ps: Parser | t.CPtr, content: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comprehension(ps: Parser | t.CPtr, elt: AST | t.CPtr, vtype: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comp_for(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_target_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_target(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_yield(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_varargslist(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_arg_def(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_suite_into_children(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_statement_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_simple_stmt_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_small_stmt(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _maybe_tuple(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_testlist(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_expr_stmt(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_return(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_raise(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_global(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_nonlocal(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_from_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_alias_list(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_del(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_assert(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_compound(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if_tail(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_while(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_for(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_try(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_except_handler(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_with(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_with_item(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_funcdef(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_classdef(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_match(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _make_name_ast(pool: memhub.MemManager | t.CPtr, node: Name | t.CPtr) -> AST | t.CPtr: pass + +def _parse_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_closed_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sequence_pattern(ps: Parser | t.CPtr, close_op: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_mapping_pattern(ps: Parser | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_class_pattern(ps: Parser | t.CPtr, cls: AST | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_match_case(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_decorators(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_module_body(ps: Parser | t.CPtr, module_node: AST | t.CPtr) -> t.CInt: pass + +def parse_tokens(pool: memhub.MemManager | t.CPtr, tokens: Token | t.CPtr) -> AST | t.CPtr: pass diff --git a/Test/NewAstTest/temp/f2ecbf8ced20575c.pyi b/Test/NewAstTest/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/Test/NewAstTest/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/Test/NewAstTest/temp/f5522571bcce7bcb.pyi b/Test/NewAstTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/NewAstTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/NoVTableTest/App/main.py b/Test/NoVTableTest/App/main.py new file mode 100644 index 0000000..6e83389 --- /dev/null +++ b/Test/NoVTableTest/App/main.py @@ -0,0 +1,509 @@ +import t +import stdio +import string +import testcheck +import linkedlist +import viperlib +import stdlib + + +# ============================================================ +# 子类继承 linkedlist.LinkedNode：启用链表/树形能力 +# 字段展平，无 vtable；OOP 方法通过 _generate_inherited_method_wrappers +# 自动生成子类包装（self bitcast 为父类指针后调用父类方法）。 +# 从 LinkedNode* 取回 Node* 需显式向下转型：(Node | t.CPtr)(ptr) +# ============================================================ +class Node(linkedlist.LinkedNode): + value: t.CInt + + +# ============================================================ +# 测试 1：append + child_count +# ============================================================ +def test_append(): + testcheck.section("Test 1: append") + root: Node = Node() + a: Node = Node() + b: Node = Node() + c: Node = Node() + a.value = 10 + b.value = 20 + c.value = 30 + root.append(a) + root.append(b) + root.append(c) + testcheck.check(root.child_count == 3, + "append count=3", + "append count fail") + # 验证顺序（LinkedNode* → Node* 显式向下转型） + c1: Node | t.CPtr = (Node | t.CPtr)(root.child) + c2: Node | t.CPtr = (Node | t.CPtr)(c1.next) + c3: Node | t.CPtr = (Node | t.CPtr)(c2.next) + testcheck.check(c1.value == 10 and c2.value == 20 and c3.value == 30, + "order 10,20,30", + "order fail") + + +# ============================================================ +# 测试 2：双向链表 prev 指针 +# ============================================================ +def test_prev_link(): + testcheck.section("Test 2: prev pointer (doubly linked)") + root: Node = Node() + a: Node = Node() + b: Node = Node() + a.value = 1 + b.value = 2 + root.append(a) + root.append(b) + # b.prev 应指向 a + bp: Node | t.CPtr = (Node | t.CPtr)(b.prev) + testcheck.check(bp is not None and bp.value == 1, + "b.prev.value=1", + "prev link fail") + # a.prev 应为 None（第一个子节点） + testcheck.check(a.prev is None, + "a.prev=None", + "first prev fail") + + +# ============================================================ +# 测试 3：last_child 指针（O(1) 追加验证） +# ============================================================ +def test_last_child(): + testcheck.section("Test 3: last_child pointer") + root: Node = Node() + a: Node = Node() + b: Node = Node() + a.value = 100 + b.value = 200 + root.append(a) + lc: Node | t.CPtr = (Node | t.CPtr)(root.last_child) + testcheck.check(lc is not None and lc.value == 100, + "last_child=100 after 1 append", + "last_child fail 1") + root.append(b) + lc = (Node | t.CPtr)(root.last_child) + testcheck.check(lc is not None and lc.value == 200, + "last_child=200 after 2 append", + "last_child fail 2") + + +# ============================================================ +# 测试 4：detach（O(1) 摘除） +# ============================================================ +def test_remove(): + testcheck.section("Test 4: detach (O(1) unlink)") + root: Node = Node() + a: Node = Node() + b: Node = Node() + c: Node = Node() + a.value = 1 + b.value = 2 + c.value = 3 + root.append(a) + root.append(b) + root.append(c) + # 摘除中间的 b + b.detach() + testcheck.check(root.child_count == 2, + "count=2 after detach", + "detach count fail") + # a.next 应直接指向 c + an: Node | t.CPtr = (Node | t.CPtr)(a.next) + testcheck.check(an is not None and an.value == 3, + "a.next.value=3", + "detach link fail") + # c.prev 应指向 a + cp: Node | t.CPtr = (Node | t.CPtr)(c.prev) + testcheck.check(cp is not None and cp.value == 1, + "c.prev.value=1", + "detach prev fail") + # b 的链接应清空 + testcheck.check(b.parent is None and b.next is None and b.prev is None, + "b cleared", + "b not cleared") + + +# ============================================================ +# 测试 5：remove_child +# ============================================================ +def test_remove_child(): + testcheck.section("Test 5: remove_child") + root: Node = Node() + a: Node = Node() + b: Node = Node() + a.value = 10 + b.value = 20 + root.append(a) + root.append(b) + root.remove_child(a) + testcheck.check(root.child_count == 1, + "count=1 after remove_child", + "remove_child count fail") + fc: Node | t.CPtr = (Node | t.CPtr)(root.child) + testcheck.check(fc is not None and fc.value == 20, + "first child=20", + "remove_child first fail") + + +# ============================================================ +# 测试 6：prepend + insert_before + insert_after +# ============================================================ +def test_insert_ops(): + testcheck.section("Test 6: prepend/insert_before/insert_after") + root: Node = Node() + a: Node = Node() + b: Node = Node() + c: Node = Node() + d: Node = Node() + a.value = 1 + b.value = 2 + c.value = 3 + d.value = 4 + root.append(a) # [1] + root.append(c) # [1, 3] + root.insert_before(c, b) # [1, 2, 3] 在 c 前插入 b + root.insert_after(a, d) # [1, 4, 2, 3] 在 a 后插入 d + # 遍历验证顺序 + cur: Node | t.CPtr = (Node | t.CPtr)(root.child) + v0: t.CInt = cur.value + cur = (Node | t.CPtr)(cur.next) + v1: t.CInt = cur.value + cur = (Node | t.CPtr)(cur.next) + v2: t.CInt = cur.value + cur = (Node | t.CPtr)(cur.next) + v3: t.CInt = cur.value + testcheck.check(v0 == 1 and v1 == 4 and v2 == 2 and v3 == 3, + "order 1,4,2,3", + "insert order fail") + testcheck.check(root.child_count == 4, + "count=4", + "insert count fail") + + +# ============================================================ +# 测试 7：child_at + count_children +# 注意：Viper 中 Node() 是栈分配，循环内 n 会被复用（同一地址）， +# 导致所有子节点指向同一内存。必须用独立变量保证各节点地址不同。 +# ============================================================ +def test_index_access(): + testcheck.section("Test 7: child_at + count_children") + root: Node = Node() + n0: Node = Node() + n1: Node = Node() + n2: Node = Node() + n3: Node = Node() + n4: Node = Node() + n0.value = 0 + n1.value = 10 + n2.value = 20 + n3.value = 30 + n4.value = 40 + root.append(n0) + root.append(n1) + root.append(n2) + root.append(n3) + root.append(n4) + # child_at + r0: Node | t.CPtr = (Node | t.CPtr)(root.child_at(0)) + r2: Node | t.CPtr = (Node | t.CPtr)(root.child_at(2)) + r4: Node | t.CPtr = (Node | t.CPtr)(root.child_at(4)) + r5: Node | t.CPtr = (Node | t.CPtr)(root.child_at(5)) + testcheck.check(r0 is not None and r0.value == 0, + "child_at(0)=0", + "child_at 0 fail") + testcheck.check(r2 is not None and r2.value == 20, + "child_at(2)=20", + "child_at 2 fail") + testcheck.check(r4 is not None and r4.value == 40, + "child_at(4)=40", + "child_at 4 fail") + testcheck.check(r5 is None, + "child_at(5)=None", + "child_at 5 fail") + # count_children + cc: t.CSizeT = root.count_children() + testcheck.check(cc == 5, + "count_children=5", + "count_children fail") + + +# ============================================================ +# 测试 8：has_children + first_sibling/last_sibling +# ============================================================ +def test_sibling_traverse(): + testcheck.section("Test 8: has_children + sibling traverse") + root: Node = Node() + testcheck.check(root.has_children() == 0, + "empty has_children=0", + "has_children empty fail") + a: Node = Node() + b: Node = Node() + c: Node = Node() + a.value = 1 + b.value = 2 + c.value = 3 + root.append(a) + root.append(b) + root.append(c) + testcheck.check(root.has_children() == 1, + "has_children=1", + "has_children fail") + # 从 b 出发找头尾兄弟 + fs: Node | t.CPtr = (Node | t.CPtr)(b.first_sibling()) + ls: Node | t.CPtr = (Node | t.CPtr)(b.last_sibling()) + testcheck.check(fs is not None and fs.value == 1, + "first_sibling=1", + "first_sibling fail") + testcheck.check(ls is not None and ls.value == 3, + "last_sibling=3", + "last_sibling fail") + + +# ============================================================ +# 测试 9：unlink（断开所有连接） +# ============================================================ +def test_unlink(): + testcheck.section("Test 9: unlink") + root: Node = Node() + a: Node = Node() + b: Node = Node() + a.value = 1 + b.value = 2 + root.append(a) + root.append(b) + # unlink a：从兄弟链摘除 + 断开父子链接 + a.unlink() + testcheck.check(a.next is None and a.prev is None and a.parent is None, + "a links cleared", + "a links not cleared") + testcheck.check(a.child is None and a.last_child is None and a.child_count == 0, + "a child cleared", + "a child not cleared") + # root 应只剩 b + testcheck.check(root.child_count == 1, + "root count=1 after unlink", + "root count fail") + + +# ============================================================ +# 测试 10：SListNode 单向链表（OOP 方法） +# ============================================================ +class SNode(linkedlist.SListNode): + value: t.CInt + + +def test_slist(): + testcheck.section("Test 10: SListNode (singly linked, OOP)") + n0: SNode = SNode() + n1: SNode = SNode() + n2: SNode = SNode() + n0.value = 100 + n1.value = 200 + n2.value = 300 + # append (O(n)) — 直接用 n0 作为 head + head: linkedlist.SListNode | t.CPtr = n0 + head = head.append(n1) + head = head.append(n2) + # count + testcheck.check(head.count() == 3, + "count=3", + "count fail") + # at + a0: SNode | t.CPtr = (SNode | t.CPtr)(head.at(0)) + a1: SNode | t.CPtr = (SNode | t.CPtr)(head.at(1)) + a2: SNode | t.CPtr = (SNode | t.CPtr)(head.at(2)) + a3: SNode | t.CPtr = (SNode | t.CPtr)(head.at(3)) + testcheck.check(a0 is not None and a0.value == 100, + "at(0)=100", + "at 0 fail") + testcheck.check(a1 is not None and a1.value == 200, + "at(1)=200", + "at 1 fail") + testcheck.check(a2 is not None and a2.value == 300, + "at(2)=300", + "at 2 fail") + testcheck.check(a3 is None, + "at(3)=None", + "at 3 fail") + # remove (移除中间节点 n1) + head = head.remove(n1) + testcheck.check(head.count() == 2, + "after remove count=2", + "remove count fail") + testcheck.check(n1.Next is None, + "n1.Next cleared", + "n1.Next not cleared") + # 移除后 a0.Next 应指向 n2 + a0b: SNode | t.CPtr = (SNode | t.CPtr)(a0.Next) + testcheck.check(a0b is not None and a0b.value == 300, + "after remove a0.Next=300", + "a0.Next fail") + # append_after (O(1)) + n3: SNode = SNode() + n3.value = 400 + tail: linkedlist.SListNode | t.CPtr = n2.append_after(n3) + testcheck.check(tail is n3, + "append_after returns n3", + "append_after fail") + testcheck.check(head.count() == 3, + "after append count=3", + "append count fail") + + +# ============================================================ +# 测试 11：viperlib.sprintf *args 转发 +# ============================================================ +def test_sprintf_vararg(): + testcheck.section("Test 11: viperlib.sprintf *args forwarding") + # 测试 %d 单参数 + buf: t.CChar | t.CPtr = stdlib.malloc(32) + if buf is None: return + buf[0] = 0 + viperlib.sprintf(buf, "key%d", 42) + testcheck.check(string.strcmp(buf, "key42") == 0, + "sprintf key42", + "sprintf key%d fail") + # 测试无参数 + buf2: t.CChar | t.CPtr = stdlib.malloc(32) + if buf2 is None: return + buf2[0] = 0 + viperlib.sprintf(buf2, "hello") + testcheck.check(string.strcmp(buf2, "hello") == 0, + "sprintf hello", + "sprintf no-arg fail") + # 测试多参数 + buf3: t.CChar | t.CPtr = stdlib.malloc(32) + if buf3 is None: return + buf3[0] = 0 + viperlib.sprintf(buf3, "%d+%d=%d", 1, 2, 3) + testcheck.check(string.strcmp(buf3, "1+2=3") == 0, + "sprintf 1+2=3", + "sprintf multi-arg fail") + + +# ============================================================ +# 测试 12：泛型 GSListNode[T] + GSList[T] +# +# 验证 4 个技术点（为 llvmlite 简化铺路）： +# 1. 递归泛型 class GNode(GSListNode[GNode]): +# 2. @t.NoVTable + 泛型组合（GNode 无 vptr，字段展平） +# 3. 泛型类作字段类型 lst: GSList[GNode] +# 4. 泛型方法继承 lst.append(n) / lst.count() / lst.at(i) +# （GSList[T] 的方法在特化 GSList[GNode] 后可用） +# +# 注：编译器暂不支持 模块.泛型类[T]() 跨模块构造，故本地定义。 +# ============================================================ +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr + + +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + + def __init__(self): + self.Head = None + self.Tail = None + self.Count = 0 + + def append(self, node: T): + if node is None: return + node.Next = None + if self.Head is None: + self.Head = node + else: + self.Tail.Next = node + self.Tail = node + self.Count += 1 + + def count(self) -> t.CSizeT: + return self.Count + + def at(self, index: t.CSizeT) -> T | t.CPtr: + if self.Head is None: return None + cur: T | t.CPtr = self.Head + i: t.CSizeT = 0 + while cur is not None: + if i == index: return cur + i += 1 + cur = cur.Next + return None + + +class GNode(GSListNode[GNode]): + Value: t.CInt + + +def test_generic_slist(): + testcheck.section("Test 12: Generic GSListNode[T] + GSList[T]") + # 技术点 3: 泛型类作字段类型 + 构造 + lst: GSList[GNode] | t.CPtr = GSList[GNode]() + + # 技术点 1+2: 递归泛型 + NoVTable 组合（GNode 定义在上方） + n0: GNode = GNode() + n1: GNode = GNode() + n2: GNode = GNode() + n0.Value = 10 + n1.Value = 20 + n2.Value = 30 + + # 技术点 4: 泛型方法继承 + lst.append(n0) + lst.append(n1) + lst.append(n2) + + # count (O(1)) + testcheck.check(lst.count() == 3, + "GSList.count()=3", + "GSList count fail") + + # at (O(n)) — 返回 GNode|CPtr，无需向下转型 + a0: GNode | t.CPtr = lst.at(0) + a1: GNode | t.CPtr = lst.at(1) + a2: GNode | t.CPtr = lst.at(2) + a3: GNode | t.CPtr = lst.at(3) + testcheck.check(a0 is not None and a0.Value == 10, + "GSList.at(0)=10", + "GSList at 0 fail") + testcheck.check(a1 is not None and a1.Value == 20, + "GSList.at(1)=20", + "GSList at 1 fail") + testcheck.check(a2 is not None and a2.Value == 30, + "GSList.at(2)=30", + "GSList at 2 fail") + testcheck.check(a3 is None, + "GSList.at(3)=None", + "GSList at 3 fail") + + # Next 字段强类型（a0.Next 直接是 GNode|CPtr，无需转型） + testcheck.check(a0.Next is a1, + "a0.Next is a1 (strong type)", + "a0.Next fail") + testcheck.check(a1.Next is a2, + "a1.Next is a2 (strong type)", + "a1.Next fail") + testcheck.check(a2.Next is None, + "a2.Next is None", + "a2.Next fail") + + +def main() -> t.CInt: + testcheck.begin("NoVTableTest: linkedlist 库（@t.NoVTable OOP 方法 + 字段展平继承）") + test_append() + test_prev_link() + test_last_child() + test_remove() + test_remove_child() + test_insert_ops() + test_index_access() + test_sibling_traverse() + test_unlink() + test_slist() + test_sprintf_vararg() + test_generic_slist() + return testcheck.end() diff --git a/Test/NoVTableTest/output/22650858c377c847.deps.json b/Test/NoVTableTest/output/22650858c377c847.deps.json new file mode 100644 index 0000000..d5f6f05 --- /dev/null +++ b/Test/NoVTableTest/output/22650858c377c847.deps.json @@ -0,0 +1 @@ +{"linkedlist": "2d39c6c7d3557b3e", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/NoVTableTest/project.json b/Test/NoVTableTest/project.json new file mode 100644 index 0000000..1a1b0ad --- /dev/null +++ b/Test/NoVTableTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "NoVTableTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "NoVTableTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/NoVTableTest/temp/22650858c377c847.doc.json b/Test/NoVTableTest/temp/22650858c377c847.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/NoVTableTest/temp/22650858c377c847.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/NoVTableTest/temp/22650858c377c847.pyi b/Test/NoVTableTest/temp/22650858c377c847.pyi new file mode 100644 index 0000000..24552b0 --- /dev/null +++ b/Test/NoVTableTest/temp/22650858c377c847.pyi @@ -0,0 +1,64 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck +import linkedlist +import viperlib +import stdlib + +class Node(linkedlist.LinkedNode): + value: t.CInt + +def test_append() -> t.CInt: pass + +def test_prev_link() -> t.CInt: pass + +def test_last_child() -> t.CInt: pass + +def test_remove() -> t.CInt: pass + +def test_remove_child() -> t.CInt: pass + +def test_insert_ops() -> t.CInt: pass + +def test_index_access() -> t.CInt: pass + +def test_sibling_traverse() -> t.CInt: pass + +def test_unlink() -> t.CInt: pass + + +class SNode(linkedlist.SListNode): + value: t.CInt + +def test_slist() -> t.CInt: pass + +def test_sprintf_vararg() -> t.CInt: pass + + +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +class GNode(GSListNode[GNode]): + Value: t.CInt + +def test_generic_slist() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/NoVTableTest/temp/2d39c6c7d3557b3e.pyi b/Test/NoVTableTest/temp/2d39c6c7d3557b3e.pyi new file mode 100644 index 0000000..fc259bd --- /dev/null +++ b/Test/NoVTableTest/temp/2d39c6c7d3557b3e.pyi @@ -0,0 +1,63 @@ +""" +Auto-generated Python stub file from linkedlist.py +Module: linkedlist +""" + + +import t, c +from stdint import * + +@t.NoVTable +class LinkedNode: + next: 'LinkedNode' | t.CPtr + prev: 'LinkedNode' | t.CPtr + child: 'LinkedNode' | t.CPtr + last_child: 'LinkedNode' | t.CPtr + parent: 'LinkedNode' | t.CPtr + child_count: t.CSizeT + def append(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def prepend(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_before(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_after(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def remove_child(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def detach(self: LinkedNode) -> t.CInt: pass + def unlink(self: LinkedNode) -> t.CInt: pass + def has_children(self: LinkedNode) -> t.CInt: pass + def child_at(self: LinkedNode, index: t.CSizeT) -> 'LinkedNode' | t.CPtr: pass + def count_children(self: LinkedNode) -> t.CSizeT: pass + def first_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass + def last_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass +@t.NoVTable +class SListNode: + Next: 'SListNode' | t.CPtr + def append(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def append_after(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def count(self: SListNode) -> t.CSizeT: pass + def at(self: SListNode, index: t.CSizeT) -> 'SListNode' | t.CPtr: pass + def remove(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +@t.NoVTable +class GTreeNode[T]: + Next: T | t.CPtr + Prev: T | t.CPtr + Child: T | t.CPtr + LastChild: T | t.CPtr + Parent: T | t.CPtr + Count: t.CSizeT + def append(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def prepend(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def detach(self: GTreeNode) -> t.CInt: pass + def has_children(self: GTreeNode) -> t.CInt: pass + def child_at(self: GTreeNode, index: t.CSizeT) -> T | t.CPtr: pass + def count_children(self: GTreeNode) -> t.CSizeT: pass \ No newline at end of file diff --git a/Test/NoVTableTest/temp/6f62fe05c5ea1ceb.pyi b/Test/NoVTableTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/NoVTableTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/NoVTableTest/temp/71e0a3ffcb3ebfad.pyi b/Test/NoVTableTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/NoVTableTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/NoVTableTest/temp/7e529fe7a078cfef.pyi b/Test/NoVTableTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/NoVTableTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/NoVTableTest/temp/90c53dd6db8d41cf.pyi b/Test/NoVTableTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/NoVTableTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/NoVTableTest/temp/_phase1_manifest.json b/Test/NoVTableTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..789603f --- /dev/null +++ b/Test/NoVTableTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\NoVTableTest\\App\\main.py": {"sha1": "22650858c377c847", "mtime": 1782800242.9985914, "size": 17052}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\linkedlist.py": {"sha1": "2d39c6c7d3557b3e", "mtime": 1782895405.2191627, "size": 16032}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/NoVTableTest/temp/_sha1_map.txt b/Test/NoVTableTest/temp/_sha1_map.txt new file mode 100644 index 0000000..45e0214 --- /dev/null +++ b/Test/NoVTableTest/temp/_sha1_map.txt @@ -0,0 +1,12 @@ +22650858c377c847:main.py +2d39c6c7d3557b3e:includes/linkedlist.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/NoVTableTest/temp/ab6e54ba9a669f76.pyi b/Test/NoVTableTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/NoVTableTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/NoVTableTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/NoVTableTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/NoVTableTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/NoVTableTest/temp/c3b259b4059f8668.pyi b/Test/NoVTableTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/NoVTableTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/NoVTableTest/temp/c9f4be41ca1cc2b4.pyi b/Test/NoVTableTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/NoVTableTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/NoVTableTest/temp/dd3002730623424b.pyi b/Test/NoVTableTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/NoVTableTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/NoVTableTest/temp/f5522571bcce7bcb.pyi b/Test/NoVTableTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/NoVTableTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/PointerTest/App/main.py b/Test/PointerTest/App/main.py new file mode 100644 index 0000000..93bbd22 --- /dev/null +++ b/Test/PointerTest/App/main.py @@ -0,0 +1,272 @@ +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +# ============================================================ +# PointerTest - 测试指针操作、global声明、数组退化、指针算术 +# ============================================================ + +# --- Test 1: global 声明对模块级变量的影响 --- +_g_val: t.CInt = 0 +_g_ptr: t.CUInt32T | t.CPtr = None + +def test_global_write(): + """测试：缺少 global 声明时，赋值是否修改模块级变量""" + global _g_val + _g_val = 42 + +def test_global_write_no_decl(): + """测试：没有 global 声明时，赋值是否创建局部变量""" + # 注意：这里故意不声明 global，测试编译器行为 + # 如果 _g_val 在此函数之前被读取过，赋值会修改全局变量 + # 如果 _g_val 在此函数中首次被赋值，会创建局部变量 + _g_val = 99 # 这应该创建一个局部变量 + +def test_global_ptr(): + global _g_ptr + buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 256, 0x3000, 0x04) + if buf: + string.memset(buf, 0, 256) + _g_ptr = buf + p: t.CUInt32T | t.CPtr = _g_ptr + p[0] = 0xAABBCCDD + p[1] = 0x11223344 + +def test_global_ptr_no_decl(): + """测试：没有 global 声明时，指针赋值是否创建局部变量""" + # _g_ptr 在 test_global_ptr 中已被赋值（有 global） + # 这里读取 _g_ptr 然后赋值 — 读取会将全局变量加入 Gen.variables + # 所以后续赋值应该修改全局变量 + old: t.CUInt32T | t.CPtr = _g_ptr # 读取 -> 加入 Gen.variables + _g_ptr = None # 因为之前读取过，这应该修改全局变量 + + +# --- Test 2: 指针算术和数组索引 --- +def test_pointer_arithmetic(): + """测试指针算术：偏移计算是否正确""" + buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 256, 0x3000, 0x04) + if not buf: + testcheck.fail("pointer arithmetic: VirtualAlloc returned NULL") + return + string.memset(buf, 0, 256) + + # 写入值 + for i in range(8): + buf[i] = t.CUInt32T(i * 100) + + # 读取并验证 + ok: t.CInt = 1 + for i in range(8): + v: t.CUInt32T = buf[i] + expected: t.CUInt32T = t.CUInt32T(i * 100) + if v != expected: + stdio.printf("buf[%d] = 0x%x, expected 0x%x\n", i, v, expected) + ok = 0 + testcheck.check(ok, "pointer arithmetic (buf[i] = i*100)", "pointer arithmetic mismatch") + + +# --- Test 3: 数组退化为指针 --- +def test_array_decay(): + """测试 list[t.CChar, N] 传递给 t.CConst | str 参数""" + arr: t.CArray[t.CChar, 32] + string.memset(c.Addr(arr), 0, 32) + # 手动写入字符串 + src: str = "Hello" + for i in range(5): + arr[i] = t.CChar(ord(src[i])) + arr[5] = 0 + + # 传递给 print (接受 t.CConst | str) + stdio.printf("array decay -> %s\n", c.Addr(arr)) + testcheck.ok("array decay (Hello)") + + +# --- Test 4: 指针类型转换 (CVoid/CPtr) --- +def test_pointer_cast(): + """测试指针与整数之间的类型转换""" + buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 256, 0x3000, 0x04) + if not buf: + testcheck.fail("pointer cast: VirtualAlloc returned NULL") + return + string.memset(buf, 0, 256) + + # 通过 CVoid 指针写入 + raw: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(buf), t.CPtr) + # 将 CVoid 指针转回 CUInt32T 指针 + p32: t.CUInt32T | t.CPtr = (t.CUInt32T | t.CPtr)(raw) + p32[0] = 0xDEADBEEF + + # 通过原始指针读取 + if buf[0] == 0xDEADBEEF: + testcheck.ok("pointer cast CVoid -> CUInt32T*") + else: + stdio.printf("got 0x%x\n", buf[0]) + testcheck.fail("pointer cast failed") + + +# --- Test 5: 指针偏移计算 (byte offset vs element offset) --- +def test_pointer_offset(): + """测试指针偏移：CUInt8T* + N 应该偏移 N 字节，CUInt32T* + N 应该偏移 N*4 字节""" + buf: t.CUInt8T | t.CPtr = w32.win32memory.VirtualAlloc(None, 256, 0x3000, 0x04) + if not buf: + testcheck.fail("pointer offset: VirtualAlloc returned NULL") + return + string.memset(buf, 0, 256) + + # 写入 4 字节 + buf[0] = 0x41 # 'A' + buf[1] = 0x42 # 'B' + buf[2] = 0x43 # 'C' + buf[3] = 0x44 # 'D' + + # 通过 CUInt32T* 读取（应该读到 0x44434241 on little-endian） + p32: t.CUInt32T | t.CPtr = (t.CUInt32T | t.CPtr)(t.CVoid(t.CUInt64T(buf), t.CPtr)) + val: t.CUInt32T = p32[0] + + # little-endian: 0x44434241 + if val == 0x44434241: + testcheck.ok("byte offset vs element offset (0x44434241)") + else: + stdio.printf("expected 0x44434241, got 0x%x\n", val) + testcheck.fail("byte offset vs element offset") + + +# --- Test 6: 结构体指针访问 --- +class TestStruct(): + a: t.CInt + b: t.CInt + c: t.CUInt64T + +def test_struct_pointer(): + """测试结构体指针的成员访问""" + s: TestStruct + s.a = 10 + s.b = 20 + s.c = 0x123456789ABCDEF0 + + p: TestStruct | t.CPtr = c.Addr(s) + stdio.printf("struct pointer access (a=%d b=%d c=0x%llx)\n", p.a, p.b, p.c) + testcheck.check(p.a == 10 and p.b == 20 and p.c == 0x123456789ABCDEF0, "struct pointer access", "struct pointer access mismatch") + + +# --- Test 7: memcpy/memset 通过指针 --- +def test_memcpy_pointers(): + """测试 memcpy 在指针之间的数据复制""" + src_buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 64, 0x3000, 0x04) + dst_buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 64, 0x3000, 0x04) + if not src_buf or not dst_buf: + testcheck.fail("memcpy: VirtualAlloc returned NULL") + return + + # 初始化 src + for i in range(8): + src_buf[i] = t.CUInt32T(i + 1000) + + # 复制 + string.memcpy(t.CVoid(t.CUInt64T(dst_buf), t.CPtr), t.CVoid(t.CUInt64T(src_buf), t.CPtr), 32) + + # 验证 + ok: t.CInt = 1 + for i in range(8): + if dst_buf[i] != t.CUInt32T(i + 1000): + stdio.printf("dst[%d] = %u, expected %u\n", i, dst_buf[i], i + 1000) + ok = 0 + testcheck.check(ok, "memcpy between pointers", "memcpy between pointers failed") + + +# --- Test 8: 指针作为函数参数 --- +def fill_array(arr: t.CUInt32T | t.CPtr, count: t.CInt, val: t.CUInt32T): + for i in range(count): + arr[i] = val + +def test_pointer_param(): + """测试指针作为函数参数传递""" + buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 64, 0x3000, 0x04) + if not buf: + testcheck.fail("pointer param: VirtualAlloc returned NULL") + return + + fill_array(buf, 8, 0xCAFEBABE) + + ok: t.CInt = 1 + for i in range(8): + if buf[i] != 0xCAFEBABE: + ok = 0 + testcheck.check(ok, "pointer as function parameter", "pointer as function parameter failed") + + +def main() -> t.CInt | t.CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + print("你好") + l: t.CArray[t.CChar, 2] = 0 + for i in "hello": + l[0] = i + print(l) + + u: t.CArray[t.CUInt64T | t.BigEndian, 2] = [0x4141414141414141, 0x0000000000000000] + for i in c.Addr(u): + # l[0] = i + stdio.printf("0x%llx\n", i) + + k: t.CArray[t.CChar, 12] = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C'] + for i in k: + stdio.printf("%c\n", i) + + testcheck.begin("PointerTest: 指针操作基准测试") + + # Test 1: global 声明 + testcheck.section("Test 1: global declaration") + _g_val = 0 + test_global_write() + stdio.printf("After test_global_write(): _g_val = %d (expect 42)\n", _g_val) + test_global_write_no_decl() + stdio.printf("After test_global_write_no_decl(): _g_val = %d (expect 42, not 99)\n", _g_val) + testcheck.check(_g_val == 42, "global declaration (_g_val = 42)", "global declaration failed") + + # Test 1b: global pointer + testcheck.section("Test 1b: global pointer") + test_global_ptr() + if _g_ptr: + p: t.CUInt32T | t.CPtr = _g_ptr + stdio.printf("_g_ptr[0] = 0x%x (expect 0xAABBCCDD)\n", p[0]) + stdio.printf("_g_ptr[1] = 0x%x (expect 0x11223344)\n", p[1]) + testcheck.check(p[0] == 0xAABBCCDD and p[1] == 0x11223344, "global pointer values", "global pointer mismatch") + test_global_ptr_no_decl() + stdio.printf("After test_global_ptr_no_decl(): _g_ptr = %p (expect NULL/0)\n", t.CVoid(t.CUInt64T(_g_ptr), t.CPtr)) + testcheck.ok("global ptr no decl (set to NULL)") + + # Test 2: pointer arithmetic + testcheck.section("Test 2: pointer arithmetic") + test_pointer_arithmetic() + + # Test 3: array decay + testcheck.section("Test 3: array decay") + test_array_decay() + + # Test 4: pointer cast + testcheck.section("Test 4: pointer cast") + test_pointer_cast() + + # Test 5: pointer offset + testcheck.section("Test 5: pointer offset (byte vs element)") + test_pointer_offset() + + # Test 6: struct pointer + testcheck.section("Test 6: struct pointer access") + test_struct_pointer() + + # Test 7: memcpy pointers + testcheck.section("Test 7: memcpy between pointers") + test_memcpy_pointers() + + # Test 8: pointer as param + testcheck.section("Test 8: pointer as function parameter") + test_pointer_param() + + return testcheck.end() diff --git a/Test/PointerTest/output/b436212dcff1be29.deps.json b/Test/PointerTest/output/b436212dcff1be29.deps.json new file mode 100644 index 0000000..9da11e2 --- /dev/null +++ b/Test/PointerTest/output/b436212dcff1be29.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/PointerTest/project.json b/Test/PointerTest/project.json new file mode 100644 index 0000000..acaef76 --- /dev/null +++ b/Test/PointerTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "PointerTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "PointerTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/PointerTest/temp/0035c95a18d4f8e8.pyi b/Test/PointerTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/PointerTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/PointerTest/temp/067c78e9f121dce3.pyi b/Test/PointerTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/PointerTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/PointerTest/temp/06f53cc594b4ac6c.pyi b/Test/PointerTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/PointerTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/PointerTest/temp/6446627d4f07a1b5.pyi b/Test/PointerTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/PointerTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/PointerTest/temp/6f62fe05c5ea1ceb.pyi b/Test/PointerTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/PointerTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/PointerTest/temp/72e2d5ccb7cedcf1.pyi b/Test/PointerTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/PointerTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/PointerTest/temp/7e529fe7a078cfef.pyi b/Test/PointerTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/PointerTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/PointerTest/temp/_phase1_manifest.json b/Test/PointerTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..e761766 --- /dev/null +++ b/Test/PointerTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\PointerTest\\App\\main.py": {"sha1": "b436212dcff1be29", "mtime": 1782266285.034029, "size": 9274}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/PointerTest/temp/_sha1_map.txt b/Test/PointerTest/temp/_sha1_map.txt new file mode 100644 index 0000000..c92e7b2 --- /dev/null +++ b/Test/PointerTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +b436212dcff1be29:main.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/PointerTest/temp/ab6e54ba9a669f76.pyi b/Test/PointerTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/PointerTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/PointerTest/temp/b436212dcff1be29.doc.json b/Test/PointerTest/temp/b436212dcff1be29.doc.json new file mode 100644 index 0000000..c6f42ec --- /dev/null +++ b/Test/PointerTest/temp/b436212dcff1be29.doc.json @@ -0,0 +1 @@ +{"test_global_write": "测试：缺少 global 声明时，赋值是否修改模块级变量", "test_global_write_no_decl": "测试：没有 global 声明时，赋值是否创建局部变量", "test_global_ptr_no_decl": "测试：没有 global 声明时，指针赋值是否创建局部变量", "test_pointer_arithmetic": "测试指针算术：偏移计算是否正确", "test_array_decay": "测试 list[t.CChar, N] 传递给 t.CConst | str 参数", "test_pointer_cast": "测试指针与整数之间的类型转换", "test_pointer_offset": "测试指针偏移：CUInt8T* + N 应该偏移 N 字节，CUInt32T* + N 应该偏移 N*4 字节", "test_struct_pointer": "测试结构体指针的成员访问", "test_memcpy_pointers": "测试 memcpy 在指针之间的数据复制", "test_pointer_param": "测试指针作为函数参数传递"} \ No newline at end of file diff --git a/Test/PointerTest/temp/b436212dcff1be29.pyi b/Test/PointerTest/temp/b436212dcff1be29.pyi new file mode 100644 index 0000000..ab57c0d --- /dev/null +++ b/Test/PointerTest/temp/b436212dcff1be29.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +_g_val: t.CExtern | t.CInt +_g_ptr: t.CExtern | t.CUInt32T | t.CPtr + +def test_global_write() -> t.CInt: pass + +def test_global_write_no_decl() -> t.CInt: pass + +def test_global_ptr() -> t.CInt: pass + +def test_global_ptr_no_decl() -> t.CInt: pass + +def test_pointer_arithmetic() -> t.CInt: pass + +def test_array_decay() -> t.CInt: pass + +def test_pointer_cast() -> t.CInt: pass + +def test_pointer_offset() -> t.CInt: pass + + +class TestStruct: + a: t.CInt + b: t.CInt + c: t.CUInt64T + +def test_struct_pointer() -> t.CInt: pass + +def test_memcpy_pointers() -> t.CInt: pass + +def fill_array(arr: t.CUInt32T | t.CPtr, count: t.CInt, val: t.CUInt32T) -> t.CInt: pass + +def test_pointer_param() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/PointerTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/PointerTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/PointerTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/PointerTest/temp/dd3002730623424b.pyi b/Test/PointerTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/PointerTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/PointerTest/temp/f5522571bcce7bcb.pyi b/Test/PointerTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/PointerTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/PointerTest/temp/f6b51804a0ba8ff0.pyi b/Test/PointerTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/PointerTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/ReTest/output/68c4fe4b12c908e3.deps.json b/Test/ReTest/output/68c4fe4b12c908e3.deps.json new file mode 100644 index 0000000..185e2d9 --- /dev/null +++ b/Test/ReTest/output/68c4fe4b12c908e3.deps.json @@ -0,0 +1 @@ +{"stdint": "56cdd754a8a09347", "mpool": "68c4fe4b12c908e3", "string": "6aee24fdefa3cbc0", "stdio": "73edbcf76e32d00b", "stdlib": "7538e542cab4c1d5", "re.__init__": "839e950555b6a435", "__init__": "839e950555b6a435", "re": "839e950555b6a435", "main": "9ef80c98cfbbce42", "viperio": "c9f4be41ca1cc2b4"} \ No newline at end of file diff --git a/Test/ReTest/output/6aee24fdefa3cbc0.deps.json b/Test/ReTest/output/6aee24fdefa3cbc0.deps.json new file mode 100644 index 0000000..185e2d9 --- /dev/null +++ b/Test/ReTest/output/6aee24fdefa3cbc0.deps.json @@ -0,0 +1 @@ +{"stdint": "56cdd754a8a09347", "mpool": "68c4fe4b12c908e3", "string": "6aee24fdefa3cbc0", "stdio": "73edbcf76e32d00b", "stdlib": "7538e542cab4c1d5", "re.__init__": "839e950555b6a435", "__init__": "839e950555b6a435", "re": "839e950555b6a435", "main": "9ef80c98cfbbce42", "viperio": "c9f4be41ca1cc2b4"} \ No newline at end of file diff --git a/Test/ReTest/output/839e950555b6a435.deps.json b/Test/ReTest/output/839e950555b6a435.deps.json new file mode 100644 index 0000000..185e2d9 --- /dev/null +++ b/Test/ReTest/output/839e950555b6a435.deps.json @@ -0,0 +1 @@ +{"stdint": "56cdd754a8a09347", "mpool": "68c4fe4b12c908e3", "string": "6aee24fdefa3cbc0", "stdio": "73edbcf76e32d00b", "stdlib": "7538e542cab4c1d5", "re.__init__": "839e950555b6a435", "__init__": "839e950555b6a435", "re": "839e950555b6a435", "main": "9ef80c98cfbbce42", "viperio": "c9f4be41ca1cc2b4"} \ No newline at end of file diff --git a/Test/ReTest/output/9ef80c98cfbbce42.deps.json b/Test/ReTest/output/9ef80c98cfbbce42.deps.json new file mode 100644 index 0000000..4b6a540 --- /dev/null +++ b/Test/ReTest/output/9ef80c98cfbbce42.deps.json @@ -0,0 +1 @@ +{"stdint": "56cdd754a8a09347", "mpool": "68c4fe4b12c908e3", "string": "6aee24fdefa3cbc0", "stdio": "73edbcf76e32d00b", "stdlib": "7538e542cab4c1d5", "re.__init__": "839e950555b6a435", "__init__": "839e950555b6a435", "re": "839e950555b6a435", "viperio": "c9f4be41ca1cc2b4"} \ No newline at end of file diff --git a/Test/ReTest/output/c9f4be41ca1cc2b4.deps.json b/Test/ReTest/output/c9f4be41ca1cc2b4.deps.json new file mode 100644 index 0000000..185e2d9 --- /dev/null +++ b/Test/ReTest/output/c9f4be41ca1cc2b4.deps.json @@ -0,0 +1 @@ +{"stdint": "56cdd754a8a09347", "mpool": "68c4fe4b12c908e3", "string": "6aee24fdefa3cbc0", "stdio": "73edbcf76e32d00b", "stdlib": "7538e542cab4c1d5", "re.__init__": "839e950555b6a435", "__init__": "839e950555b6a435", "re": "839e950555b6a435", "main": "9ef80c98cfbbce42", "viperio": "c9f4be41ca1cc2b4"} \ No newline at end of file diff --git a/Test/ReTest/temp/56cdd754a8a09347.pyi b/Test/ReTest/temp/56cdd754a8a09347.pyi new file mode 100644 index 0000000..007ccb8 --- /dev/null +++ b/Test/ReTest/temp/56cdd754a8a09347.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLong | t.CLong +ULONGLONG: t.CTypedef = t.CUnsignedLong | t.CLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +UINTPTR: t.CTypedef = t.CUnsignedInt | t.CPtr +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr \ No newline at end of file diff --git a/Test/ReTest/temp/68c4fe4b12c908e3.pyi b/Test/ReTest/temp/68c4fe4b12c908e3.pyi new file mode 100644 index 0000000..fc2010b --- /dev/null +++ b/Test/ReTest/temp/68c4fe4b12c908e3.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from mpool.py +Module: mpool +""" + + +import t, c +from stdint import * +import viperio +import string + +MPOOL_ALIGN: t.CDefine = 8 +MPOOL_TYPE_SLAB: t.CDefine = 0 +MPOOL_TYPE_BUMP: t.CDefine = 2 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + + +class MPool: + mtype: t.CInt + mem: t.CVoid | t.CPtr + mem_size: t.CSizeT + offset: t.CSizeT + high_water: t.CSizeT + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + def __init__(self: MPool, mem: t.CVoid | t.CPtr, mem_size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def _init_bump(self: MPool, mem: t.CVoid | t.CPtr, mem_size: t.CSizeT) -> t.CInt: pass + def _init_slab(self: MPool, mem: t.CVoid | t.CPtr, mem_size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def __enter__(self: MPool) -> 'MPool' | t.CPtr: pass + def __exit__(self: MPool) -> t.CInt: pass + def _slab_reset(self: MPool) -> t.CInt: pass + def alloc(self: MPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def alloc_buf(self: MPool, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass + def free(self: MPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def realloc(self: MPool, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def reset(self: MPool) -> t.CInt: pass + def _bump_alloc(self: MPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def _slab_alloc(self: MPool) -> t.CVoid | t.CPtr: pass + def _slab_free(self: MPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + +def bump_create(mem: t.CVoid | t.CPtr, mem_size: t.CSizeT) -> MPool | t.CPtr: pass + +def alloc(pool: MPool | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def reset(pool: MPool | t.CPtr) -> t.CInt: pass diff --git a/Test/ReTest/temp/6aee24fdefa3cbc0.pyi b/Test/ReTest/temp/6aee24fdefa3cbc0.pyi new file mode 100644 index 0000000..c17475f --- /dev/null +++ b/Test/ReTest/temp/6aee24fdefa3cbc0.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: list[str]) -> int: pass diff --git a/Test/ReTest/temp/73edbcf76e32d00b.pyi b/Test/ReTest/temp/73edbcf76e32d00b.pyi new file mode 100644 index 0000000..ef712e3 --- /dev/null +++ b/Test/ReTest/temp/73edbcf76e32d00b.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def fprintf(stream: t.CVoid | t.CPtr, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def sprintf(buf: t.CChar | t.CPtr, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.CExtern | t.CExport: pass + +def puts(s: t.CChar | t.CConst | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def fputs(s: t.CChar | t.CConst | t.CPtr, stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + + +stdin: t.CExtern | t.CVoid | t.CPtr +stdout: t.CExtern | t.CVoid | t.CPtr +stderr: t.CExtern | t.CVoid | t.CPtr \ No newline at end of file diff --git a/Test/ReTest/temp/7538e542cab4c1d5.pyi b/Test/ReTest/temp/7538e542cab4c1d5.pyi new file mode 100644 index 0000000..b5313af --- /dev/null +++ b/Test/ReTest/temp/7538e542cab4c1d5.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: UINT) -> t.CVoid | t.CPtr | t.State | t.CExtern | t.CExport: pass + +def calloc(nmemb: UINT, size: UINT) -> t.CVoid | t.CPtr | t.State | t.CExtern | t.CExport: pass + +def realloc(p: str, p2: t.CLong | t.CLong) -> t.CVoid | t.CPtr | t.State | t.CExtern | t.CExport: pass + +def free(p: t.CVoid | t.CPtr) -> None | t.State | t.CExtern | t.CExport: pass diff --git a/Test/ReTest/temp/839e950555b6a435.pyi b/Test/ReTest/temp/839e950555b6a435.pyi new file mode 100644 index 0000000..4440462 --- /dev/null +++ b/Test/ReTest/temp/839e950555b6a435.pyi @@ -0,0 +1,103 @@ +""" +Auto-generated Python stub file from re.__init__.py +Module: re.__init__ +""" + + +import t, c +from stdint import * +import mpool +import string + +RE_CHAR: t.CDefine = 0 # 匹配单个字符 +RE_DOT: t.CDefine = 1 # 匹配任意字符（除换行） +RE_CLASS: t.CDefine = 2 # 匹配字符类 [abc] +RE_NCLASS: t.CDefine = 3 # 匹配反义字符类 [^abc] +RE_SPLIT: t.CDefine = 4 # 分裂节点（用于 | ? *） +RE_JUMP: t.CDefine = 5 # 无条件跳转（用于连接） +RE_MATCH: t.CDefine = 6 # 匹配成功 +RE_BOL: t.CDefine = 7 # 行首 ^ +RE_EOL: t.CDefine = 8 # 行尾 $ +RE_DCLASS: t.CDefine = 9 # 预定义字符类 \d \w \s +RE_NDCLASS: t.CDefine = 10 # 预定义反义字符类 \D \W \S +DCLASS_DIGIT: t.CDefine = 0 # \d +DCLASS_WORD: t.CDefine = 1 # \w +DCLASS_SPACE: t.CDefine = 2 # \s +RE_MAX_NODES: t.CDefine = 256 +RE_MAX_STATES: t.CDefine = 256 + +class Renode: + ntype: t.CInt + ch: t.CChar + out1: Renode | t.CPtr + out2: Renode | t.CPtr + cls: t.CChar | t.CPtr + dcls: t.CInt + def __new__(self: Renode, pool: mpool.MPool | t.CPtr) -> 'Renode' | t.CPtr: pass +class RePtrList: + node: Renode | t.CPtr + which: t.CInt + next: RePtrList | t.CPtr + def __new__(self: RePtrList, pool: mpool.MPool | t.CPtr) -> 'RePtrList' | t.CPtr: pass +class Refrag: + start: Renode | t.CPtr + out: RePtrList | t.CPtr + def __new__(self: Refrag, pool: mpool.MPool | t.CPtr) -> 'Refrag' | t.CPtr: pass +class Rematch: + start: t.CSizeT + end: t.CSizeT + found: t.CInt + def __new__(self: Rematch, pool: mpool.MPool | t.CPtr) -> 'Rematch' | t.CPtr: pass + +def _is_word_char(ch: t.CChar) -> bool: pass + +def _match_dclass(dcls: t.CInt, ch: t.CChar) -> bool: pass + +def _list_append(pool: mpool.MPool | t.CPtr, head: RePtrList | t.CPtr, node: Renode | t.CPtr, which: t.CInt) -> RePtrList | t.CPtr: pass + +def _list_patch(head: RePtrList | t.CPtr, target: Renode | t.CPtr) -> t.CInt: pass + +def _list_concat(a: RePtrList | t.CPtr, b: RePtrList | t.CPtr) -> RePtrList | t.CPtr: pass + + +class Recompiler: + pool: mpool.MPool | t.CPtr + p: t.CChar | t.CPtr + pend: t.CChar | t.CPtr + match_node: Renode | t.CPtr + def __new__(self: Recompiler, pool: mpool.MPool | t.CPtr) -> 'Recompiler' | t.CPtr: pass + def compile(self: Recompiler, pattern: t.CChar | t.CPtr) -> Renode | t.CPtr: pass + def _peek(self: Recompiler) -> t.CChar: pass + def _next(self: Recompiler) -> t.CChar: pass + def _parse_alt(self: Recompiler) -> Refrag | t.CPtr: pass + def _parse_concat(self: Recompiler) -> Refrag | t.CPtr: pass + def _parse_repeat(self: Recompiler) -> Refrag | t.CPtr: pass + def _parse_atom(self: Recompiler) -> Refrag | t.CPtr: pass + def _parse_escape(self: Recompiler) -> Refrag | t.CPtr: pass + def _parse_class(self: Recompiler) -> Refrag | t.CPtr: pass +class Restate: + node: Renode | t.CPtr + pos: t.CSizeT + def __new__(self: Restate, pool: mpool.MPool | t.CPtr) -> 'Restate' | t.CPtr: pass + +def _add_state(nlist: t.CVoid | t.CPtr, node: Renode | t.CPtr, pos: t.CSizeT, text: t.CChar | t.CPtr, nstates: t.CInt, max_states: t.CInt) -> t.CInt: pass + +def _read_node(slist: t.CVoid | t.CPtr, idx: t.CInt) -> Renode | t.CPtr: pass + +def _read_pos(slist: t.CVoid | t.CPtr, idx: t.CInt) -> t.CSizeT: pass + +def _step(clist: t.CVoid | t.CPtr, nlist: t.CVoid | t.CPtr, ch: t.CChar, pos: t.CSizeT, text: t.CChar | t.CPtr, cstates: t.CInt, max_states: t.CInt) -> t.CInt: pass + +def _check_match(slist: t.CVoid | t.CPtr, nstates: t.CInt) -> t.CSizeT: pass + +def compile(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr) -> Renode | t.CPtr: pass + +def match(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> Rematch | t.CPtr: pass + +def search(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> Rematch | t.CPtr: pass + +def findall(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr, text: t.CChar | t.CPtr, results: t.CChar | t.CPtr | t.CPtr, max_results: t.CSizeT) -> t.CSizeT: pass + +def sub(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr, replacement: t.CChar | t.CPtr, text: t.CChar | t.CPtr, outbuf: t.CChar | t.CPtr, outsize: t.CSizeT) -> t.CSizeT: pass + +def split(pool: mpool.MPool | t.CPtr, pattern: t.CChar | t.CPtr, text: t.CChar | t.CPtr, results: t.CChar | t.CPtr | t.CPtr, max_results: t.CSizeT) -> t.CSizeT: pass diff --git a/Test/ReTest/temp/9ef80c98cfbbce42.pyi b/Test/ReTest/temp/9ef80c98cfbbce42.pyi new file mode 100644 index 0000000..dc5715e --- /dev/null +++ b/Test/ReTest/temp/9ef80c98cfbbce42.pyi @@ -0,0 +1,58 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from t import CInt, CLong, CDouble, CPtr, CSizeT, CChar +from stdio import printf +import re +import mpool +import string +import stdlib + +_pass_count: t.CExtern | CInt +_fail_count: t.CExtern | CInt + +def check(name: CChar | CPtr, condition: bool) -> t.CInt: pass + +def make_pool() -> mpool.MPool | CPtr: pass + +def free_pool(pool: mpool.MPool | CPtr) -> t.CInt: pass + +def test_basic_match() -> t.CInt: pass + +def test_dot() -> t.CInt: pass + +def test_star() -> t.CInt: pass + +def test_plus() -> t.CInt: pass + +def test_question() -> t.CInt: pass + +def test_alt() -> t.CInt: pass + +def test_group() -> t.CInt: pass + +def test_charclass() -> t.CInt: pass + +def test_predefined() -> t.CInt: pass + +def test_search() -> t.CInt: pass + +def test_findall() -> t.CInt: pass + +def test_sub() -> t.CInt: pass + +def test_split() -> t.CInt: pass + +def test_anchors() -> t.CInt: pass + +def test_escape() -> t.CInt: pass + +def test_mpool_reuse() -> t.CInt: pass + +def test_complex() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/ReTest/temp/_sha1_map.txt b/Test/ReTest/temp/_sha1_map.txt new file mode 100644 index 0000000..6ff19cc --- /dev/null +++ b/Test/ReTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +56cdd754a8a09347:includes/stdint.py +68c4fe4b12c908e3:includes/mpool.py +6aee24fdefa3cbc0:includes/string.py +73edbcf76e32d00b:includes/stdio.py +7538e542cab4c1d5:includes/stdlib.py +839e950555b6a435:includes/re\__init__.py +9ef80c98cfbbce42:main.py +c9f4be41ca1cc2b4:includes/viperio.py diff --git a/Test/ReTest/temp/c9f4be41ca1cc2b4.pyi b/Test/ReTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/ReTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ShortCircuitTest/App/main.py b/Test/ShortCircuitTest/App/main.py new file mode 100644 index 0000000..925866b --- /dev/null +++ b/Test/ShortCircuitTest/App/main.py @@ -0,0 +1,122 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + + +# 副作用函数：将 buf[0] 置为 1，返回 True +def bump_true(buf: bytes) -> bool: + buf[0] = 1 + return True + + +# 副作用函数：将 buf[0] 置为 1，返回 False +def bump_false(buf: bytes) -> bool: + buf[0] = 1 + return False + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("if 短路机制测试") + + buf: bytes = stdlib.malloc(1) + + # === Test 1: False and bump_true() - bump 不应被调用 === + testcheck.section("Test 1: False and bump_true()") + buf[0] = 0 + r1: bool = False and bump_true(buf) + testcheck.check(buf[0] == 0, "and short-circuit OK (bump not called)", "and short-circuit FAILED (bump called)") + + # === Test 2: True or bump_true() - bump 不应被调用 === + testcheck.section("Test 2: True or bump_true()") + buf[0] = 0 + r2: bool = True or bump_true(buf) + testcheck.check(buf[0] == 0, "or short-circuit OK (bump not called)", "or short-circuit FAILED (bump called)") + + # === Test 3: True and bump_true() - bump 应被调用 === + testcheck.section("Test 3: True and bump_true()") + buf[0] = 0 + r3: bool = True and bump_true(buf) + testcheck.check(buf[0] == 1, "and eval OK (bump called)", "and eval FAILED (bump not called)") + + # === Test 4: False or bump_true() - bump 应被调用 === + testcheck.section("Test 4: False or bump_true()") + buf[0] = 0 + r4: bool = False or bump_true(buf) + testcheck.check(buf[0] == 1, "or eval OK (bump called)", "or eval FAILED (bump not called)") + + # === Test 5: if 语句中的短路 === + testcheck.section("Test 5: if 语句中的短路") + buf[0] = 0 + if False and bump_true(buf): + testcheck.check(False, "should not enter if", "if False and entered") + testcheck.check(buf[0] == 0, "if and short-circuit OK", "if and short-circuit FAILED") + + buf[0] = 0 + if True or bump_true(buf): + testcheck.check(True, "if True or entered OK", "if True or not entered") + else: + testcheck.check(False, "should enter if", "if True or else") + testcheck.check(buf[0] == 0, "if or short-circuit OK", "if or short-circuit FAILED") + + # === Test 6: 嵌套短路 === + testcheck.section("Test 6: 嵌套短路") + buf[0] = 0 + r6: bool = False and (True or bump_true(buf)) + testcheck.check(buf[0] == 0, "nested and short-circuit OK", "nested and short-circuit FAILED") + + buf[0] = 0 + r7: bool = True or (False and bump_true(buf)) + testcheck.check(buf[0] == 0, "nested or short-circuit OK", "nested or short-circuit FAILED") + + # === Test 7: not 与短路 === + testcheck.section("Test 7: not 与短路") + buf[0] = 0 + r8: bool = not False and bump_true(buf) + testcheck.check(buf[0] == 1, "not and eval OK (bump called)", "not and eval FAILED (bump not called)") + + buf[0] = 0 + r9: bool = not True or bump_true(buf) + testcheck.check(buf[0] == 1, "not or eval OK (bump called)", "not or eval FAILED (bump not called)") + + # === Test 8: and 链中第一个为 False，后续不评估 === + testcheck.section("Test 8: and 链短路") + buf[0] = 0 + r10: bool = False and bump_true(buf) and bump_true(buf) + testcheck.check(buf[0] == 0, "and chain short-circuit OK", "and chain short-circuit FAILED") + + # === Test 9: or 链中第一个为 True，后续不评估 === + testcheck.section("Test 9: or 链短路") + buf[0] = 0 + r11: bool = True or bump_true(buf) or bump_true(buf) + testcheck.check(buf[0] == 0, "or chain short-circuit OK", "or chain short-circuit FAILED") + + # === Test 10: and 返回值正确性 === + testcheck.section("Test 10: and 返回值正确性") + buf[0] = 0 + r12: bool = True and bump_true(buf) + testcheck.check(r12, "and return True OK", "and return True FAILED") + + buf[0] = 0 + r13: bool = True and bump_false(buf) + testcheck.check(not r13, "and return False OK", "and return False FAILED") + + # === Test 11: or 返回值正确性 === + testcheck.section("Test 11: or 返回值正确性") + buf[0] = 0 + r14: bool = False or bump_true(buf) + testcheck.check(r14, "or return True OK", "or return True FAILED") + + buf[0] = 0 + r15: bool = False or bump_false(buf) + testcheck.check(not r15, "or return False OK", "or return False FAILED") + + stdlib.free(buf) + return testcheck.end() diff --git a/Test/ShortCircuitTest/output/a711c9c66fd7c6ca.deps.json b/Test/ShortCircuitTest/output/a711c9c66fd7c6ca.deps.json new file mode 100644 index 0000000..262dabd --- /dev/null +++ b/Test/ShortCircuitTest/output/a711c9c66fd7c6ca.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ShortCircuitTest/project.json b/Test/ShortCircuitTest/project.json new file mode 100644 index 0000000..de206e7 --- /dev/null +++ b/Test/ShortCircuitTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ShortCircuitTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "ShortCircuitTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ShortCircuitTest/temp/0035c95a18d4f8e8.pyi b/Test/ShortCircuitTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/ShortCircuitTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/ShortCircuitTest/temp/067c78e9f121dce3.pyi b/Test/ShortCircuitTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/ShortCircuitTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/ShortCircuitTest/temp/06f53cc594b4ac6c.pyi b/Test/ShortCircuitTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/ShortCircuitTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ShortCircuitTest/temp/6446627d4f07a1b5.pyi b/Test/ShortCircuitTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/ShortCircuitTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/ShortCircuitTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ShortCircuitTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ShortCircuitTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ShortCircuitTest/temp/72e2d5ccb7cedcf1.pyi b/Test/ShortCircuitTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/ShortCircuitTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/ShortCircuitTest/temp/7e529fe7a078cfef.pyi b/Test/ShortCircuitTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ShortCircuitTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ShortCircuitTest/temp/90c53dd6db8d41cf.pyi b/Test/ShortCircuitTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/ShortCircuitTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/ShortCircuitTest/temp/_phase1_manifest.json b/Test/ShortCircuitTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..26e54b5 --- /dev/null +++ b/Test/ShortCircuitTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ShortCircuitTest\\App\\main.py": {"sha1": "a711c9c66fd7c6ca", "mtime": 1782355546.5909688, "size": 4497}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/ShortCircuitTest/temp/_sha1_map.txt b/Test/ShortCircuitTest/temp/_sha1_map.txt new file mode 100644 index 0000000..47c7fc4 --- /dev/null +++ b/Test/ShortCircuitTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +a711c9c66fd7c6ca:main.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.doc.json b/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.pyi b/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.pyi new file mode 100644 index 0000000..213e2c8 --- /dev/null +++ b/Test/ShortCircuitTest/temp/a711c9c66fd7c6ca.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import stdio +import stdlib +import string +import testcheck + +def bump_true(buf: bytes) -> bool: pass + +def bump_false(buf: bytes) -> bool: pass + +def main() -> CInt | CExport: pass diff --git a/Test/ShortCircuitTest/temp/ab6e54ba9a669f76.pyi b/Test/ShortCircuitTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/ShortCircuitTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/ShortCircuitTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ShortCircuitTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ShortCircuitTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ShortCircuitTest/temp/dd3002730623424b.pyi b/Test/ShortCircuitTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ShortCircuitTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ShortCircuitTest/temp/f5522571bcce7bcb.pyi b/Test/ShortCircuitTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ShortCircuitTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ShortCircuitTest/temp/f6b51804a0ba8ff0.pyi b/Test/ShortCircuitTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/ShortCircuitTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/SimdTest/App/main.py b/Test/SimdTest/App/main.py new file mode 100644 index 0000000..997ca4d --- /dev/null +++ b/Test/SimdTest/App/main.py @@ -0,0 +1,502 @@ +from stdint import * +import vipersimd +import numpy +from stdio import printf +import stdlib +from stdlib import calloc +import t, c +import memhub +import testcheck + + +# 全局内存池 +arena: t.CUInt8T | t.CPtr = None +pool: memhub.MemPool | t.CPtr = None + +# 外部函数: clock() 用于性能计时 +def clock() -> t.CLong | t.CExtern: pass + +CLOCK_PER_SEC: t.CDefine = 1000 + + +# ============================================================ +# 1. AVX2 double4 正确性测试 +# ============================================================ + +def test_avx2_add_sub_mul_div(): + testcheck.section("AVX2 add/sub/mul/div 4d") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(2.0) + a.data[2] = t.CDouble(3.0) + a.data[3] = t.CDouble(4.0) + + b: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + b.data[0] = t.CDouble(5.0) + b.data[1] = t.CDouble(6.0) + b.data[2] = t.CDouble(7.0) + b.data[3] = t.CDouble(8.0) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + + # add + vipersimd.simd_add4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(6.0), "add[0]==6", "add[0]==6") + testcheck.check(out.data[1] == t.CDouble(8.0), "add[1]==8", "add[1]==8") + testcheck.check(out.data[2] == t.CDouble(10.0), "add[2]==10", "add[2]==10") + testcheck.check(out.data[3] == t.CDouble(12.0), "add[3]==12", "add[3]==12") + + # sub + vipersimd.simd_sub4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(-4.0), "sub[0]==-4", "sub[0]==-4") + testcheck.check(out.data[3] == t.CDouble(-4.0), "sub[3]==-4", "sub[3]==-4") + + # mul + vipersimd.simd_mul4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(5.0), "mul[0]==5", "mul[0]==5") + testcheck.check(out.data[1] == t.CDouble(12.0), "mul[1]==12", "mul[1]==12") + testcheck.check(out.data[2] == t.CDouble(21.0), "mul[2]==21", "mul[2]==21") + testcheck.check(out.data[3] == t.CDouble(32.0), "mul[3]==32", "mul[3]==32") + + # div + vipersimd.simd_div4d(b.data, a.data, out.data) + testcheck.check(out.data[0] == t.CDouble(5.0), "div[0]==5", "div[0]==5") + testcheck.check(out.data[1] == t.CDouble(3.0), "div[1]==3", "div[1]==3") + + a.delete() + b.delete() + out.delete() + + +def test_avx2_neg_abs_sqrt(): + testcheck.section("AVX2 neg/abs/sqrt 4d") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(-4.0) + a.data[2] = t.CDouble(9.0) + a.data[3] = t.CDouble(-16.0) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + + # neg + vipersimd.simd_neg4d(a.data, out.data) + testcheck.check(out.data[0] == t.CDouble(-1.0), "neg[0]==-1", "neg[0]==-1") + testcheck.check(out.data[1] == t.CDouble(4.0), "neg[1]==4", "neg[1]==4") + testcheck.check(out.data[2] == t.CDouble(-9.0), "neg[2]==-9", "neg[2]==-9") + testcheck.check(out.data[3] == t.CDouble(16.0), "neg[3]==16", "neg[3]==16") + + # abs + vipersimd.simd_abs4d(a.data, out.data) + testcheck.check(out.data[0] == t.CDouble(1.0), "abs[0]==1", "abs[0]==1") + testcheck.check(out.data[1] == t.CDouble(4.0), "abs[1]==4", "abs[1]==4") + testcheck.check(out.data[2] == t.CDouble(9.0), "abs[2]==9", "abs[2]==9") + testcheck.check(out.data[3] == t.CDouble(16.0), "abs[3]==16", "abs[3]==16") + + # sqrt + vipersimd.simd_sqrt4d(out.data, out.data) + testcheck.check(out.data[0] == t.CDouble(1.0), "sqrt[0]==1", "sqrt[0]==1") + testcheck.check(out.data[1] == t.CDouble(2.0), "sqrt[1]==2", "sqrt[1]==2") + testcheck.check(out.data[2] == t.CDouble(3.0), "sqrt[2]==3", "sqrt[2]==3") + testcheck.check(out.data[3] == t.CDouble(4.0), "sqrt[3]==4", "sqrt[3]==4") + + a.delete() + out.delete() + + +def test_avx2_min_max(): + testcheck.section("AVX2 min/max 4d") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(8.0) + a.data[2] = t.CDouble(3.0) + a.data[3] = t.CDouble(6.0) + + b: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + b.data[0] = t.CDouble(5.0) + b.data[1] = t.CDouble(2.0) + b.data[2] = t.CDouble(7.0) + b.data[3] = t.CDouble(4.0) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + + vipersimd.simd_min4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(1.0), "min[0]==1", "min[0]==1") + testcheck.check(out.data[1] == t.CDouble(2.0), "min[1]==2", "min[1]==2") + testcheck.check(out.data[2] == t.CDouble(3.0), "min[2]==3", "min[2]==3") + testcheck.check(out.data[3] == t.CDouble(4.0), "min[3]==4", "min[3]==4") + + vipersimd.simd_max4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(5.0), "max[0]==5", "max[0]==5") + testcheck.check(out.data[1] == t.CDouble(8.0), "max[1]==8", "max[1]==8") + testcheck.check(out.data[2] == t.CDouble(7.0), "max[2]==7", "max[2]==7") + testcheck.check(out.data[3] == t.CDouble(6.0), "max[3]==6", "max[3]==6") + + a.delete() + b.delete() + out.delete() + + +def test_avx2_scalar_ops(): + testcheck.section("AVX2 scalar broadcast ops") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(2.0) + a.data[2] = t.CDouble(3.0) + a.data[3] = t.CDouble(4.0) + + s: numpy.ndarray | t.CPtr = numpy.zeros(pool, 1) + s.data[0] = t.CDouble(10.0) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + + vipersimd.simd_mul_scalar4d(a.data, s.data, out.data) + testcheck.check(out.data[0] == t.CDouble(10.0), "mul_scalar[0]==10", "mul_scalar[0]==10") + testcheck.check(out.data[1] == t.CDouble(20.0), "mul_scalar[1]==20", "mul_scalar[1]==20") + testcheck.check(out.data[3] == t.CDouble(40.0), "mul_scalar[3]==40", "mul_scalar[3]==40") + + vipersimd.simd_add_scalar4d(a.data, s.data, out.data) + testcheck.check(out.data[0] == t.CDouble(11.0), "add_scalar[0]==11", "add_scalar[0]==11") + testcheck.check(out.data[3] == t.CDouble(14.0), "add_scalar[3]==14", "add_scalar[3]==14") + + a.delete() + s.delete() + out.delete() + + +def test_avx2_hsum_dot(): + testcheck.section("AVX2 hsum/dot 4d") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(2.0) + a.data[2] = t.CDouble(3.0) + a.data[3] = t.CDouble(4.0) + + b: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + b.data[0] = t.CDouble(5.0) + b.data[1] = t.CDouble(6.0) + b.data[2] = t.CDouble(7.0) + b.data[3] = t.CDouble(8.0) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 1) + + vipersimd.simd_hsum4d(a.data, out.data) + testcheck.check(out.data[0] == t.CDouble(10.0), "hsum==10", "hsum==10") + + vipersimd.simd_dot4d(a.data, b.data, out.data) + testcheck.check(out.data[0] == t.CDouble(70.0), "dot==70", "dot==70") + + a.delete() + b.delete() + out.delete() + + +def test_avx2_fma(): + testcheck.section("AVX2 FMA 4d") + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0) + a.data[1] = t.CDouble(2.0) + a.data[2] = t.CDouble(3.0) + a.data[3] = t.CDouble(4.0) + + b: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + b.data[0] = t.CDouble(5.0) + b.data[1] = t.CDouble(6.0) + b.data[2] = t.CDouble(7.0) + b.data[3] = t.CDouble(8.0) + + c_val: numpy.ndarray | t.CPtr = numpy.ones(pool, 4) + + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + + vipersimd.simd_fma4d(a.data, b.data, c_val.data, out.data) + testcheck.check(out.data[0] == t.CDouble(6.0), "fma[0]==6", "fma[0]==6") + testcheck.check(out.data[1] == t.CDouble(13.0), "fma[1]==13", "fma[1]==13") + testcheck.check(out.data[2] == t.CDouble(22.0), "fma[2]==22", "fma[2]==22") + testcheck.check(out.data[3] == t.CDouble(33.0), "fma[3]==33", "fma[3]==33") + + a.delete() + b.delete() + c_val.delete() + out.delete() + + +# ============================================================ +# 2. 批量数组运算正确性测试 +# ============================================================ + +def test_batch_array_ops(): + testcheck.section("Batch array operations") + n: t.CSizeT = 16 + a: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(1.0), t.CDouble(17.0), t.CDouble(1.0)) + b: numpy.ndarray | t.CPtr = numpy.full(pool, 16, t.CDouble(2.0)) + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, 16) + + # add array + vipersimd.simd_add_array(a.data, b.data, out.data, n) + testcheck.check(out.data[0] == t.CDouble(3.0), "add_arr[0]==3", "add_arr[0]==3") + testcheck.check(out.data[15] == t.CDouble(18.0), "add_arr[15]==18", "add_arr[15]==18") + + # mul array + vipersimd.simd_mul_array(a.data, b.data, out.data, n) + testcheck.check(out.data[0] == t.CDouble(2.0), "mul_arr[0]==2", "mul_arr[0]==2") + testcheck.check(out.data[3] == t.CDouble(8.0), "mul_arr[3]==8", "mul_arr[3]==8") + + # sqrt array + c: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + c.data[0] = t.CDouble(4.0) + c.data[1] = t.CDouble(9.0) + c.data[2] = t.CDouble(16.0) + c.data[3] = t.CDouble(25.0) + out4: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + vipersimd.simd_sqrt_array(c.data, out4.data, t.CSizeT(4)) + testcheck.check(out4.data[0] == t.CDouble(2.0), "sqrt_arr[0]==2", "sqrt_arr[0]==2") + testcheck.check(out4.data[3] == t.CDouble(5.0), "sqrt_arr[3]==5", "sqrt_arr[3]==5") + + # abs array + d: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + d.data[0] = t.CDouble(-1.0) + d.data[1] = t.CDouble(-2.0) + d.data[2] = t.CDouble(3.0) + d.data[3] = t.CDouble(-4.0) + vipersimd.simd_abs_array(d.data, out4.data, t.CSizeT(4)) + testcheck.check(out4.data[0] == t.CDouble(1.0), "abs_arr[0]==1", "abs_arr[0]==1") + testcheck.check(out4.data[1] == t.CDouble(2.0), "abs_arr[1]==2", "abs_arr[1]==2") + testcheck.check(out4.data[3] == t.CDouble(4.0), "abs_arr[3]==4", "abs_arr[3]==4") + + # dot array + e: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + e.data[0] = t.CDouble(1.0) + e.data[1] = t.CDouble(2.0) + e.data[2] = t.CDouble(3.0) + e.data[3] = t.CDouble(4.0) + f: numpy.ndarray | t.CPtr = numpy.ones(pool, 4) + out1: numpy.ndarray | t.CPtr = numpy.zeros(pool, 1) + vipersimd.simd_dot_array(e.data, f.data, t.CSizeT(4), out1.data) + testcheck.check(out1.data[0] == t.CDouble(10.0), "dot_arr==10", "dot_arr==10") + + a.delete() + b.delete() + out.delete() + c.delete() + out4.delete() + d.delete() + e.delete() + f.delete() + out1.delete() + + +# ============================================================ +# 3. LLVMIR 精确标量运算测试 +# ============================================================ + +def test_llvmir_scalar(): + testcheck.section("LLVMIR precise scalar ops") + + # sqrt + r: t.CDouble = vipersimd.simd_sqrt(t.CDouble(4.0)) + testcheck.check(r == t.CDouble(2.0), "sqrt(4)==2", "sqrt(4)==2") + + # fabs + r = vipersimd.simd_fabs(t.CDouble(-3.5)) + testcheck.check(r == t.CDouble(3.5), "fabs(-3.5)==3.5", "fabs(-3.5)==3.5") + + # floor + r = vipersimd.simd_floor(t.CDouble(3.7)) + testcheck.check(r == t.CDouble(3.0), "floor(3.7)==3", "floor(3.7)==3") + + # ceil + r = vipersimd.simd_ceil(t.CDouble(3.2)) + testcheck.check(r == t.CDouble(4.0), "ceil(3.2)==4", "ceil(3.2)==4") + + # round + r = vipersimd.simd_round(t.CDouble(3.5)) + testcheck.check(r == t.CDouble(4.0), "round(3.5)==4", "round(3.5)==4") + + # trunc + r = vipersimd.simd_trunc(t.CDouble(3.9)) + testcheck.check(r == t.CDouble(3.0), "trunc(3.9)==3", "trunc(3.9)==3") + + # fma: 2*3+1 = 7 + r = vipersimd.simd_fma(t.CDouble(2.0), t.CDouble(3.0), t.CDouble(1.0)) + testcheck.check(r == t.CDouble(7.0), "fma(2,3,1)==7", "fma(2,3,1)==7") + + # copysign + r = vipersimd.simd_copysign(t.CDouble(3.0), t.CDouble(-1.0)) + testcheck.check(r == t.CDouble(-3.0), "copysign(3,-1)==-3", "copysign(3,-1)==-3") + + # neg + r = vipersimd.simd_neg(t.CDouble(1.5)) + testcheck.check(r == t.CDouble(-1.5), "neg(1.5)==-1.5", "neg(1.5)==-1.5") + + # minnum + r = vipersimd.simd_minnum(t.CDouble(2.0), t.CDouble(3.0)) + testcheck.check(r == t.CDouble(2.0), "minnum(2,3)==2", "minnum(2,3)==2") + + # maxnum + r = vipersimd.simd_maxnum(t.CDouble(2.0), t.CDouble(3.0)) + testcheck.check(r == t.CDouble(3.0), "maxnum(2,3)==3", "maxnum(2,3)==3") + + # exp + r = vipersimd.simd_exp(t.CDouble(1.0)) + testcheck.check(r > t.CDouble(2.71) and r < t.CDouble(2.72), "exp(1)~=2.718", "exp(1)~=2.718") + + # log + r = vipersimd.simd_log(t.CDouble(2.718281828459045)) + testcheck.check(r > t.CDouble(0.999) and r < t.CDouble(1.001), "log(e)~=1", "log(e)~=1") + + # sin + r = vipersimd.simd_sin(t.CDouble(0.0)) + testcheck.check(r > t.CDouble(-0.001) and r < t.CDouble(0.001), "sin(0)~=0", "sin(0)~=0") + + # cos + r = vipersimd.simd_cos(t.CDouble(0.0)) + testcheck.check(r > t.CDouble(0.999) and r < t.CDouble(1.001), "cos(0)~=1", "cos(0)~=1") + + # pow + r = vipersimd.simd_pow(t.CDouble(2.0), t.CDouble(10.0)) + testcheck.check(r == t.CDouble(1024.0), "pow(2,10)==1024", "pow(2,10)==1024") + + +# ============================================================ +# 4. 性能基准测试: SIMD vs 标量 +# ============================================================ + +N_ELEM: t.CDefine = 40000 +N_ITER: t.CDefine = 50000 + +def scalar_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT): + i: t.CSizeT = 0 + da: t.CDouble | t.CPtr = a + db: t.CDouble | t.CPtr = b + dout: t.CDouble | t.CPtr = out + while i < n: + dout[i] = da[i] + db[i] + i += 1 + +def scalar_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT): + i: t.CSizeT = 0 + da: t.CDouble | t.CPtr = a + db: t.CDouble | t.CPtr = b + dout: t.CDouble | t.CPtr = out + while i < n: + dout[i] = da[i] * db[i] + i += 1 + +def scalar_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT): + i: t.CSizeT = 0 + da: t.CDouble | t.CPtr = a + dout: t.CDouble | t.CPtr = out + while i < n: + dout[i] = vipersimd.simd_sqrt(da[i]) + i += 1 + +def benchmark_simd_vs_scalar(): + testcheck.section("Benchmark: SIMD vs Scalar") + + a: numpy.ndarray | t.CPtr = numpy.full(pool, N_ELEM, t.CDouble(1.5)) + b: numpy.ndarray | t.CPtr = numpy.full(pool, N_ELEM, t.CDouble(2.5)) + out: numpy.ndarray | t.CPtr = numpy.zeros(pool, N_ELEM) + + # --- SIMD add benchmark --- + t0: t.CLong = clock() + iter: t.CInt = 0 + while iter < N_ITER: + vipersimd.simd_add_array(a.data, b.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1: t.CLong = clock() + simd_add_ms: t.CLong = t1 - t0 + + # --- Scalar add benchmark --- + t0 = clock() + iter = 0 + while iter < N_ITER: + scalar_add_array(a.data, b.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1 = clock() + scalar_add_ms: t.CLong = t1 - t0 + + printf(" ADD: SIMD=%ldms Scalar=%ldms", simd_add_ms, scalar_add_ms) + if scalar_add_ms > 0: + printf(" speedup=%.1fx\n", t.CDouble(scalar_add_ms) / t.CDouble(simd_add_ms)) + else: + printf(" speedup=N/A\n") + + # --- SIMD mul benchmark --- + t0 = clock() + iter = 0 + while iter < N_ITER: + vipersimd.simd_mul_array(a.data, b.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1 = clock() + simd_mul_ms: t.CLong = t1 - t0 + + # --- Scalar mul benchmark --- + t0 = clock() + iter = 0 + while iter < N_ITER: + scalar_mul_array(a.data, b.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1 = clock() + scalar_mul_ms: t.CLong = t1 - t0 + + printf(" MUL: SIMD=%ldms Scalar=%ldms", simd_mul_ms, scalar_mul_ms) + if scalar_mul_ms > 0: + printf(" speedup=%.1fx\n", t.CDouble(scalar_mul_ms) / t.CDouble(simd_mul_ms)) + else: + printf(" speedup=N/A\n") + + # --- SIMD sqrt benchmark --- + t0 = clock() + iter = 0 + while iter < N_ITER: + vipersimd.simd_sqrt_array(a.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1 = clock() + simd_sqrt_ms: t.CLong = t1 - t0 + + # --- Scalar sqrt benchmark --- + t0 = clock() + iter = 0 + while iter < N_ITER: + scalar_sqrt_array(a.data, out.data, t.CSizeT(N_ELEM)) + iter += 1 + t1 = clock() + scalar_sqrt_ms: t.CLong = t1 - t0 + + printf(" SQRT: SIMD=%ldms Scalar=%ldms", simd_sqrt_ms, scalar_sqrt_ms) + if scalar_sqrt_ms > 0: + printf(" speedup=%.1fx\n", t.CDouble(scalar_sqrt_ms) / t.CDouble(simd_sqrt_ms)) + else: + printf(" speedup=N/A\n") + + # 验证结果正确性 + testcheck.check(out.data[0] > t.CDouble(0.0), "bench result[0] valid", "bench result[0] valid") + + a.delete() + b.delete() + out.delete() + + +# ============================================================ +# Main +# ============================================================ + +def main() -> t.CInt: + global arena, pool + + arena = calloc(4096 * 1024, 1) # 4MB arena + pool = memhub.MemPool(arena, 4096 * 1024) # bump allocator + + testcheck.begin("ViperSIMD Test Suite") + + test_avx2_add_sub_mul_div() + test_avx2_neg_abs_sqrt() + test_avx2_min_max() + test_avx2_scalar_ops() + test_avx2_hsum_dot() + test_avx2_fma() + test_batch_array_ops() + test_llvmir_scalar() + benchmark_simd_vs_scalar() + + return testcheck.end() diff --git a/Test/SimdTest/output/2fcb6780db4d937f.deps.json b/Test/SimdTest/output/2fcb6780db4d937f.deps.json new file mode 100644 index 0000000..9ec0f91 --- /dev/null +++ b/Test/SimdTest/output/2fcb6780db4d937f.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "vipermath": "3f7c5e78d8652535", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vipersimd": "c24f25b14b4bbd0a", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/SimdTest/project.json b/Test/SimdTest/project.json new file mode 100644 index 0000000..56a1581 --- /dev/null +++ b/Test/SimdTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "SimdTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "SimdTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/SimdTest/temp/271ea3decb810db2.pyi b/Test/SimdTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/SimdTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/SimdTest/temp/2fcb6780db4d937f.doc.json b/Test/SimdTest/temp/2fcb6780db4d937f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/SimdTest/temp/2fcb6780db4d937f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/SimdTest/temp/2fcb6780db4d937f.pyi b/Test/SimdTest/temp/2fcb6780db4d937f.pyi new file mode 100644 index 0000000..06ba45f --- /dev/null +++ b/Test/SimdTest/temp/2fcb6780db4d937f.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import vipersimd +import numpy +from stdio import printf +import stdlib +from stdlib import calloc +import t, c +import memhub +import testcheck + +arena: t.CExtern | t.CUInt8T | t.CPtr +pool: t.CExtern | memhub.MemPool | t.CPtr + +def clock() -> t.CLong | t.CExtern: pass + + +CLOCK_PER_SEC: t.CDefine = 1000 + +def test_avx2_add_sub_mul_div() -> t.CInt: pass + +def test_avx2_neg_abs_sqrt() -> t.CInt: pass + +def test_avx2_min_max() -> t.CInt: pass + +def test_avx2_scalar_ops() -> t.CInt: pass + +def test_avx2_hsum_dot() -> t.CInt: pass + +def test_avx2_fma() -> t.CInt: pass + +def test_batch_array_ops() -> t.CInt: pass + +def test_llvmir_scalar() -> t.CInt: pass + + +N_ELEM: t.CDefine = 40000 +N_ITER: t.CDefine = 50000 + +def scalar_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def scalar_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def scalar_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def benchmark_simd_vs_scalar() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/SimdTest/temp/3f7c5e78d8652535.pyi b/Test/SimdTest/temp/3f7c5e78d8652535.pyi new file mode 100644 index 0000000..abe01c2 --- /dev/null +++ b/Test/SimdTest/temp/3f7c5e78d8652535.pyi @@ -0,0 +1,128 @@ +""" +Auto-generated Python stub file from vipermath.py +Module: vipermath +""" + + +import t, c + +U_M_PI: t.CDefine = 3.14159265358979323846 +U_M_E: t.CDefine = 2.71828182845904523536 +U_M_PI_2: t.CDefine = 1.57079632679489661923 +U_M_PI_4: t.CDefine = 0.78539816339744830962 +U_M_1_PI: t.CDefine = 0.31830988618379067154 +U_M_2_PI: t.CDefine = 0.63661977236758134308 +U_M_LN2: t.CDefine = 0.69314718055994530942 +U_M_LN10: t.CDefine = 2.30258509299404568402 +U_M_2_SQRT_PI: t.CDefine = 1.77245385090551602730 + +def radians(degrees: t.CDouble) -> t.CDouble: pass + +def degrees(radians: t.CDouble) -> t.CDouble: pass + +def sin(x: t.CDouble) -> t.CDouble: pass + +def cos(x: t.CDouble) -> t.CDouble: pass + +def tan(x: t.CDouble) -> t.CDouble: pass + +def asin(x: t.CDouble) -> t.CDouble: pass + +def acos(x: t.CDouble) -> t.CDouble: pass + +def _atan_core(x: t.CDouble) -> t.CDouble: pass + +def atan(x: t.CDouble) -> t.CDouble: pass + +def atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble: pass + +def sinh(x: t.CDouble) -> t.CDouble: pass + +def cosh(x: t.CDouble) -> t.CDouble: pass + +def tanh(x: t.CDouble) -> t.CDouble: pass + +def exp(x: t.CDouble) -> t.CDouble: pass + +def log(x: t.CDouble) -> t.CDouble: pass + +def sqrt(x: t.CDouble) -> t.CDouble: pass + +def abs(x: t.CInt) -> t.CInt: pass + +def fabs(x: t.CDouble) -> t.CDouble: pass + +def labs(x: t.CLong) -> t.CLong: pass + +def factorial(n: t.CInt) -> t.CDouble: pass + +def combination(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def permutation(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def powf(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def cbrt(x: t.CDouble) -> t.CDouble: pass + +def hypot(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def floor(x: t.CDouble) -> t.CDouble: pass + +def ceil(x: t.CDouble) -> t.CDouble: pass + +def round(x: t.CDouble) -> t.CDouble: pass + +def trunc(x: t.CDouble) -> t.CDouble: pass + +def fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def fmodf(x: float, y: float) -> float: pass + +def modf(x: t.CDouble, iptr: t.CDouble | t.CPtr) -> t.CDouble: pass + +def log10(x: t.CDouble) -> t.CDouble: pass + +def log2(x: t.CDouble) -> t.CDouble: pass + +def exp2(x: t.CDouble) -> t.CDouble: pass + +def expm1(x: t.CDouble) -> t.CDouble: pass + +def log1p(x: t.CDouble) -> t.CDouble: pass + +def asinh(x: t.CDouble) -> t.CDouble: pass + +def acosh(x: t.CDouble) -> t.CDouble: pass + +def atanh(x: t.CDouble) -> t.CDouble: pass + +def gamma(x: t.CDouble) -> t.CDouble: pass + +def erf(x: t.CDouble) -> t.CDouble: pass + +def erfc(x: t.CDouble) -> t.CDouble: pass + +def sqrtf(x: t.CFloat) -> t.CFloat: pass + +def sinf(x: t.CFloat) -> t.CFloat: pass + +def cosf(x: t.CFloat) -> t.CFloat: pass + +def tanf(x: t.CFloat) -> t.CFloat: pass + +def fabsf(x: t.CFloat) -> t.CFloat: pass + +def floorf(x: t.CFloat) -> t.CFloat: pass + +def ceilf(x: t.CFloat) -> t.CFloat: pass + + +class _U(t.CUnion): + d: t.CDouble + u: t.CUInt64T + +def isnan(x: t.CDouble) -> t.CStatic | t.CInt: pass + +def isinf(x: t.CDouble) -> t.CStatic | t.CInt: pass diff --git a/Test/SimdTest/temp/6f62fe05c5ea1ceb.pyi b/Test/SimdTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/SimdTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/SimdTest/temp/7e529fe7a078cfef.pyi b/Test/SimdTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/SimdTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/SimdTest/temp/90c53dd6db8d41cf.pyi b/Test/SimdTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/SimdTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/SimdTest/temp/_phase1_manifest.json b/Test/SimdTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..9367a98 --- /dev/null +++ b/Test/SimdTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\SimdTest\\App\\main.py": {"sha1": "2fcb6780db4d937f", "mtime": 1782828206.5357063, "size": 17078}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\numpy\\__init__.py": {"sha1": "c3a6aed1f1fb8b1e", "mtime": 1782949786.0890937, "size": 39111}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vipermath.py": {"sha1": "3f7c5e78d8652535", "mtime": 1781532528.2518907, "size": 15412}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vipersimd.py": {"sha1": "c24f25b14b4bbd0a", "mtime": 1782891155.5031893, "size": 15522}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/SimdTest/temp/_sha1_map.txt b/Test/SimdTest/temp/_sha1_map.txt new file mode 100644 index 0000000..5bd402c --- /dev/null +++ b/Test/SimdTest/temp/_sha1_map.txt @@ -0,0 +1,14 @@ +271ea3decb810db2:includes/atom.py +2fcb6780db4d937f:main.py +3f7c5e78d8652535:includes/vipermath.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c24f25b14b4bbd0a:includes/vipersimd.py +c3a6aed1f1fb8b1e:includes/numpy\__init__.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/SimdTest/temp/ab6e54ba9a669f76.pyi b/Test/SimdTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/SimdTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/SimdTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/SimdTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/SimdTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/SimdTest/temp/c24f25b14b4bbd0a.pyi b/Test/SimdTest/temp/c24f25b14b4bbd0a.pyi new file mode 100644 index 0000000..a7c0c1f --- /dev/null +++ b/Test/SimdTest/temp/c24f25b14b4bbd0a.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from vipersimd.py +Module: vipersimd +""" + + +import t, c + +def simd_add4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sub4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_div4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sqrt4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_neg4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_abs4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_max4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_min4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_hsum4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_dot4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_fma4d(a: t.CPtr, b: t.CPtr, c_val: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_sub4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_mul4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_div4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: pass + +def simd_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_sub_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_div_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_abs_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_neg_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: pass + +def simd_dot_array(a: t.CPtr, b: t.CPtr, n: t.CSizeT, out: t.CPtr) -> t.CVoid: pass + +def simd_sqrt(x: t.CDouble) -> t.CDouble: pass + +def simd_fabs(x: t.CDouble) -> t.CDouble: pass + +def simd_floor(x: t.CDouble) -> t.CDouble: pass + +def simd_ceil(x: t.CDouble) -> t.CDouble: pass + +def simd_round(x: t.CDouble) -> t.CDouble: pass + +def simd_trunc(x: t.CDouble) -> t.CDouble: pass + +def simd_fma(a: t.CDouble, b: t.CDouble, c_val: t.CDouble) -> t.CDouble: pass + +def simd_copysign(mag: t.CDouble, sign: t.CDouble) -> t.CDouble: pass + +def simd_neg(x: t.CDouble) -> t.CDouble: pass + +def simd_minnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def simd_maxnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def simd_exp(x: t.CDouble) -> t.CDouble: pass + +def simd_log(x: t.CDouble) -> t.CDouble: pass + +def simd_sin(x: t.CDouble) -> t.CDouble: pass + +def simd_cos(x: t.CDouble) -> t.CDouble: pass + +def simd_pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass diff --git a/Test/SimdTest/temp/c3a6aed1f1fb8b1e.pyi b/Test/SimdTest/temp/c3a6aed1f1fb8b1e.pyi new file mode 100644 index 0000000..68a99b1 --- /dev/null +++ b/Test/SimdTest/temp/c3a6aed1f1fb8b1e.pyi @@ -0,0 +1,240 @@ +""" +Auto-generated Python stub file from numpy.__init__.py +Module: numpy.__init__ +""" + + +from stdint import * +import stdlib +import string +import vipermath +import stdio +import t, c +import memhub + +MAX_NDIM: t.CDefine = 4 +float64: t.CTypedef = t.CDouble +float32: t.CTypedef = t.CFloat +int64: t.CTypedef = t.CLong +int32: t.CTypedef = t.CInt +uint8: t.CTypedef = t.CUnsignedChar +pi: t.CDefine = 3.14159265358979323846 +e: t.CDefine = 2.71828182845904523536 + +@t.Object +class ndarray[T]: + data: T | t.CPtr + shape: t.CArray[t.CSizeT, MAX_NDIM] + strides: t.CArray[t.CSizeT, MAX_NDIM] + ndim: t.CInt + size: t.CSizeT + owns_data: t.CInt + pool: memhub.MemManager | t.CPtr + def __new__(self: ndarray, pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> t.CInt: pass + def __add__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __sub__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mul__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __truediv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __floordiv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mod__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __neg__(self: ndarray) -> ndarray[T] | t.CPtr: pass + def __len__(self: ndarray) -> t.CInt: pass + def at2d(self: ndarray, row: t.CSizeT, col: t.CSizeT) -> T: pass + def set2d(self: ndarray, row: t.CSizeT, col: t.CSizeT, val: T) -> t.CInt: pass + def delete(self: ndarray) -> t.CInt: pass + def fill(self: ndarray, val: T) -> t.CInt: pass + def copy(self: ndarray) -> ndarray[T] | t.CPtr: pass + def reshape(self: ndarray, new_shape: INTPTR, new_ndim: t.CInt) -> ndarray[T] | t.CPtr: pass + def sum(self: ndarray) -> T: pass + def mean(self: ndarray) -> T: pass + def min(self: ndarray) -> T: pass + def max(self: ndarray) -> T: pass + def argmax(self: ndarray) -> t.CInt: pass + def argmin(self: ndarray) -> t.CInt: pass + def dot(self: ndarray, other: ndarray[T] | t.CPtr) -> T: pass + def T(self: ndarray) -> ndarray[T] | t.CPtr: pass + def print_arr(self: ndarray) -> t.CInt: pass + +Float64Array: t.CTypedef = ndarray[t.CDouble] +Float32Array: t.CTypedef = ndarray[t.CFloat] +Int64Array: t.CTypedef = ndarray[t.CLong] +Int32Array: t.CTypedef = ndarray[t.CInt] +Uint8Array: t.CTypedef = ndarray[t.CUnsignedChar] + +def _alloc_ndarray(pool: memhub.MemManager | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def _compute_strides(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def _empty_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr, data: t.CDouble | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full(pool: memhub.MemManager | t.CPtr, n: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, step: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, num: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def eye(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def diag(pool: memhub.MemManager | t.CPtr, vals: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_abs(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sqrt(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_exp(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_pow(a: ndarray[t.CDouble] | t.CPtr, p: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def add_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def mul_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def sub_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def div_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def matmul(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def dot_product(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_sum(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_mean(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_min(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_max(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_argmax(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_argmin(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def var(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def std(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def norm(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def clip(a: ndarray[t.CDouble] | t.CPtr, lo: t.CDouble, hi: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def concatenate(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def sort_arr(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def reverse(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log10(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_floor(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_ceil(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_round(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sign(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tanh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sinh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cosh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arcsin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arccos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan2(y: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_degrees(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_radians(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isnan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isinf(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_maximum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_minimum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cumsum(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def diff(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def flatten(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def trace(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def outer(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_where(condition: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_count_nonzero(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_all(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_any(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_not_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange1(pool: memhub.MemManager | t.CPtr, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace2(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def meshgrid(x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def det2x2(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def inv2x2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cross3(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_interp(x: t.CDouble, xp: ndarray[t.CDouble] | t.CPtr, fp: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def prod(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def median(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def percentile(a: ndarray[t.CDouble] | t.CPtr, q: t.CDouble) -> t.CDouble: pass diff --git a/Test/SimdTest/temp/c9f4be41ca1cc2b4.pyi b/Test/SimdTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/SimdTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/SimdTest/temp/dd3002730623424b.pyi b/Test/SimdTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/SimdTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/SimdTest/temp/ee084e9fc6ee413a.pyi b/Test/SimdTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/SimdTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/SimdTest/temp/f5522571bcce7bcb.pyi b/Test/SimdTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/SimdTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/SprintfTest/App/main.py b/Test/SprintfTest/App/main.py new file mode 100644 index 0000000..7ae39c0 --- /dev/null +++ b/Test/SprintfTest/App/main.py @@ -0,0 +1,123 @@ +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import viperlib + + +# ============================================================ +# SprintfTest - viperlib.snprintf %a/%A 运行时验证 +# 预期输出对照 glibc printf("%a", ...) 行为 +# ============================================================ +@t.CExport +def main() -> int: + testcheck.begin("SprintfTest: %a/%A runtime verification") + buf: bytes = stdlib.malloc(64) + + # ---------- 默认精度（核心正确性） ---------- + testcheck.section("default precision") + + v1: t.CDouble = 1.0 + viperlib.snprintf(buf, 64, "%a", v1) + testcheck.info(buf) + testcheck.check(buf == "0x1p+0", "%a 1.0 -> 0x1p+0", "FAIL 1.0") + + v2: t.CDouble = 2.0 + viperlib.snprintf(buf, 64, "%a", v2) + testcheck.info(buf) + testcheck.check(buf == "0x1p+1", "%a 2.0 -> 0x1p+1", "FAIL 2.0") + + v_half: t.CDouble = 0.5 + viperlib.snprintf(buf, 64, "%a", v_half) + testcheck.info(buf) + testcheck.check(buf == "0x1p-1", "%a 0.5 -> 0x1p-1", "FAIL 0.5") + + v3: t.CDouble = 3.0 + viperlib.snprintf(buf, 64, "%a", v3) + testcheck.info(buf) + testcheck.check(buf == "0x1.8p+1", "%a 3.0 -> 0x1.8p+1", "FAIL 3.0") + + v_pi: t.CDouble = 3.14 + viperlib.snprintf(buf, 64, "%a", v_pi) + testcheck.info(buf) + testcheck.check(buf == "0x1.91eb851eb851fp+1", "%a 3.14 -> 0x1.91eb851eb851fp+1", "FAIL 3.14") + + v0: t.CDouble = 0.0 + viperlib.snprintf(buf, 64, "%a", v0) + testcheck.info(buf) + testcheck.check(buf == "0x0p+0", "%a 0.0 -> 0x0p+0", "FAIL 0.0") + + vn1: t.CDouble = -1.0 + viperlib.snprintf(buf, 64, "%a", vn1) + testcheck.info(buf) + testcheck.check(buf == "-0x1p+0", "%a -1.0 -> -0x1p+0", "FAIL -1.0") + + # ---------- 大写 %A ---------- + testcheck.section("uppercase %A") + viperlib.snprintf(buf, 64, "%A", v_pi) + testcheck.info(buf) + testcheck.check(buf == "0X1.91EB851EB851FP+1", "%A 3.14 -> 0X1...P+1", "FAIL %A") + + # ---------- # flag 强制小数点 ---------- + testcheck.section("# flag") + viperlib.snprintf(buf, 64, "%#a", v1) + testcheck.info(buf) + testcheck.check(buf == "0x1.p+0", "%#a 1.0 -> 0x1.p+0", "FAIL %#a") + + # ---------- 显式精度 13（不裁剪） ---------- + testcheck.section("precision 13") + viperlib.snprintf(buf, 64, "%.13a", v1) + testcheck.info(buf) + testcheck.check(buf == "0x1.0000000000000p+0", "%.13a 1.0 -> 0x1.0000000000000p+0", "FAIL %.13a") + + # ---------- 宽度/对齐 ---------- + testcheck.section("width/align") + viperlib.snprintf(buf, 64, "%10a", v1) + testcheck.info(buf) + testcheck.check(buf == " 0x1p+0", "%10a 1.0 -> ' 0x1p+0'", "FAIL %10a") + + viperlib.snprintf(buf, 64, "%-10a|", v1) + testcheck.info(buf) + testcheck.check(buf == "0x1p+0 |", "%-10a 1.0 -> '0x1p+0 |'", "FAIL %-10a") + + # ---------- show_sign ---------- + testcheck.section("show_sign") + viperlib.snprintf(buf, 64, "%+a", v1) + testcheck.info(buf) + testcheck.check(buf == "+0x1p+0", "%+a 1.0 -> +0x1p+0", "FAIL %+a") + + # ---------- inf / nan ---------- + testcheck.section("inf/nan") + + # 构造 +inf: bit pattern 0x7FF0000000000000 + inf_bits: t.CUInt64T = 0x7FF0000000000000 + inf_val: t.CDouble = 0.0 + c.Load(c.Addr(inf_val), c.Addr(inf_bits)) + viperlib.snprintf(buf, 64, "%a", inf_val) + testcheck.info(buf) + testcheck.check(buf == "inf", "%a +inf -> inf", "FAIL +inf") + + # 构造 -inf: bit pattern 0xFFF0000000000000 + ninf_bits: t.CUInt64T = 0xFFF0000000000000 + ninf_val: t.CDouble = 0.0 + c.Load(c.Addr(ninf_val), c.Addr(ninf_bits)) + viperlib.snprintf(buf, 64, "%a", ninf_val) + testcheck.info(buf) + testcheck.check(buf == "-inf", "%a -inf -> -inf", "FAIL -inf") + + # 构造 nan: bit pattern 0x7FF8000000000000 + nan_bits: t.CUInt64T = 0x7FF8000000000000 + nan_val: t.CDouble = 0.0 + c.Load(c.Addr(nan_val), c.Addr(nan_bits)) + viperlib.snprintf(buf, 64, "%a", nan_val) + testcheck.info(buf) + testcheck.check(buf == "nan", "%a nan -> nan", "FAIL nan") + + # 大写 INF/NAN + viperlib.snprintf(buf, 64, "%A", inf_val) + testcheck.info(buf) + testcheck.check(buf == "INF", "%A +inf -> INF", "FAIL INF") + + stdlib.free(buf) + return 0 diff --git a/Test/SprintfTest/output/f492cb40d913400b.deps.json b/Test/SprintfTest/output/f492cb40d913400b.deps.json new file mode 100644 index 0000000..2a993fa --- /dev/null +++ b/Test/SprintfTest/output/f492cb40d913400b.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/SprintfTest/project.json b/Test/SprintfTest/project.json new file mode 100644 index 0000000..90be6af --- /dev/null +++ b/Test/SprintfTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "SprintfTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "SprintfTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/SprintfTest/temp/6f62fe05c5ea1ceb.pyi b/Test/SprintfTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/SprintfTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/SprintfTest/temp/71e0a3ffcb3ebfad.pyi b/Test/SprintfTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/SprintfTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/SprintfTest/temp/7e529fe7a078cfef.pyi b/Test/SprintfTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/SprintfTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/SprintfTest/temp/90c53dd6db8d41cf.pyi b/Test/SprintfTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/SprintfTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/SprintfTest/temp/_phase1_manifest.json b/Test/SprintfTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..f7bb7c9 --- /dev/null +++ b/Test/SprintfTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\SprintfTest\\App\\main.py": {"sha1": "f492cb40d913400b", "mtime": 1782821953.0346377, "size": 4235}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/SprintfTest/temp/_sha1_map.txt b/Test/SprintfTest/temp/_sha1_map.txt new file mode 100644 index 0000000..6e81a3d --- /dev/null +++ b/Test/SprintfTest/temp/_sha1_map.txt @@ -0,0 +1,10 @@ +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +f492cb40d913400b:main.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/SprintfTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/SprintfTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/SprintfTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/SprintfTest/temp/c3b259b4059f8668.pyi b/Test/SprintfTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/SprintfTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/SprintfTest/temp/c9f4be41ca1cc2b4.pyi b/Test/SprintfTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/SprintfTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/SprintfTest/temp/dd3002730623424b.pyi b/Test/SprintfTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/SprintfTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/SprintfTest/temp/f492cb40d913400b.doc.json b/Test/SprintfTest/temp/f492cb40d913400b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/SprintfTest/temp/f492cb40d913400b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/SprintfTest/temp/f492cb40d913400b.pyi b/Test/SprintfTest/temp/f492cb40d913400b.pyi new file mode 100644 index 0000000..f8cf5c7 --- /dev/null +++ b/Test/SprintfTest/temp/f492cb40d913400b.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import stdlib +import testcheck +import viperlib + +@t.CExport +def main() -> int: pass diff --git a/Test/SprintfTest/temp/f5522571bcce7bcb.pyi b/Test/SprintfTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/SprintfTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/StringIterTest/App/inline_test.py b/Test/StringIterTest/App/inline_test.py new file mode 100644 index 0000000..95144c3 --- /dev/null +++ b/Test/StringIterTest/App/inline_test.py @@ -0,0 +1,51 @@ +import t +import stdio +import c + +# Test: @t.Object class with inline list[struct, N] field + +class Inner: + code: t.CUnsignedInt + bits: t.CInt + +@t.Object +class Container: + items: t.CArray[Inner, 10] + count: t.CInt + + def __init__(self): + self.count = 0 + + def fill(self): + for i in range(10): + self.items[i].code = t.CUnsignedInt(i * 100) + self.items[i].bits = i + self.count += 1 + + def check(self): + ok: t.CInt = 1 + for i in range(10): + if self.items[i].code != t.CUnsignedInt(i * 100): + stdio.printf("FAIL: items[%d].code=%u expect %d\n", i, self.items[i].code, i * 100) + ok = 0 + if self.items[i].bits != i: + stdio.printf("FAIL: items[%d].bits=%d expect %d\n", i, self.items[i].bits, i) + ok = 0 + if ok: + stdio.printf("PASS: inline list[struct, N] in @t.Object\n") + else: + stdio.printf("FAIL: inline list[struct, N] in @t.Object\n") + +def main() -> t.CInt: + stdio.printf("=== InlineArrayTest ===\n") + c1 = Container() + c1.fill() + c1.check() + + # Also test: local variable of @t.Object type + c2 = Container() + c2.fill() + c2.check() + + stdio.printf("=== InlineArrayTest Complete ===\n") + return 0 diff --git a/Test/StringIterTest/App/main.py b/Test/StringIterTest/App/main.py new file mode 100644 index 0000000..8a5645c --- /dev/null +++ b/Test/StringIterTest/App/main.py @@ -0,0 +1,179 @@ +import t +import stdio +import string +import c +import testcheck + +# ============================================================ +# Test 1: 字符串指针迭代 - 逐字符直到\0 +# ============================================================ +def test_string_ptr_iter(): + testcheck.section("Test 1: string pointer iteration") + s: t.CChar | t.CPtr = "Hello" + p: t.CChar | t.CPtr = s + count: t.CInt = 0 + while c.Deref(p) != 0: + count += 1 + p += 1 + testcheck.check(count == 5, "ptr iter (count=5)", "ptr iter (count expect 5)") + +# ============================================================ +# Test 2: 字符串指针求和ASCII值 +# ============================================================ +def test_string_ascii_sum(): + testcheck.section("Test 2: string ASCII sum") + s: t.CChar | t.CPtr = "ABC" + p: t.CChar | t.CPtr = s + total: t.CInt = 0 + while c.Deref(p) != 0: + total += t.CInt(c.Deref(p)) + p += 1 + # A=65, B=66, C=67 => 198 + testcheck.check(total == 198, "ASCII sum (total=198)", "ASCII sum (total expect 198)") + +# ============================================================ +# Test 3: 手动strcpy - 指针复制 +# ============================================================ +def test_manual_strcpy(): + testcheck.section("Test 3: manual strcpy via pointers") + src: t.CChar | t.CPtr = "Test" + dst: list[t.CChar, 32] = [0] + sp: t.CChar | t.CPtr = src + dp: t.CChar | t.CPtr = c.Addr(dst) + while c.Deref(sp) != 0: + c.Set(c.Deref(dp), c.Deref(sp)) + sp += 1 + dp += 1 + c.Set(c.Deref(dp), 0) + testcheck.check(dst[0] == 84 and dst[1] == 101 and dst[2] == 115 and dst[3] == 116, "manual strcpy (T=84 e=101 s=115 t=116)", "manual strcpy (expect T=84 e=101 s=115 t=116)") + +# ============================================================ +# Test 4: 指针比较和偏移 +# ============================================================ +def test_ptr_offset_compare(): + testcheck.section("Test 4: pointer offset and compare") + arr: list[t.CInt, 5] = [0] + for i in range(5): + arr[i] = i * 10 + p: t.CInt | t.CPtr = c.Addr(arr) + start: t.CInt | t.CPtr = p + total: t.CInt = 0 + while p != start + 5: + total += c.Deref(p) + p += 1 + testcheck.check(total == 100, "ptr offset (total=100)", "ptr offset (total expect 100)") + +# ============================================================ +# Test 5: 字符串长度手动计算 +# ============================================================ +def test_manual_strlen(): + testcheck.section("Test 5: manual strlen") + s: t.CChar | t.CPtr = "Hello World!" + p: t.CChar | t.CPtr = s + length: t.CSizeT = 0 + while c.Deref(p) != 0: + length += 1 + p += 1 + testcheck.check(length == 12, "manual strlen (len=12)", "manual strlen (len expect 12)") + +# ============================================================ +# Test 6: 指针反向遍历字符串 +# ============================================================ +def test_reverse_string_ptr(): + testcheck.section("Test 6: reverse string via pointer") + src: t.CChar | t.CPtr = "abcd" + dst: list[t.CChar, 32] = [0] + # Find end of src + p: t.CChar | t.CPtr = src + length: t.CInt = 0 + while c.Deref(p) != 0: + length += 1 + p += 1 + # Copy in reverse + dp: t.CChar | t.CPtr = c.Addr(dst) + p = src + length - 1 + while p >= src: + c.Set(c.Deref(dp), c.Deref(p)) + dp += 1 + p -= 1 + c.Set(c.Deref(dp), 0) + # "abcd" reversed = "dcba" = 100,99,98,97 + testcheck.check(dst[0] == 100 and dst[1] == 99 and dst[2] == 98 and dst[3] == 97, "reverse string (d=100 c=99 b=98 a=97)", "reverse string (expect d=100 c=99 b=98 a=97)") + +# ============================================================ +# Test 7: 多级指针 - 指向指针的指针 +# ============================================================ +def test_double_pointer(): + testcheck.section("Test 7: double pointer") + x: t.CInt = 42 + px: t.CInt | t.CPtr = c.Addr(x) + ppx: t.CInt | t.CPtr | t.CPtr = c.Addr(px) + # Read through double pointer + val: t.CInt = c.Deref(c.Deref(ppx)) + testcheck.check(val == 42, "double pointer (val=42)", "double pointer (val expect 42)") + +# ============================================================ +# Test 8: 指针算术 - 数组元素访问 +# ============================================================ +def test_ptr_arithmetic(): + testcheck.section("Test 8: pointer arithmetic") + arr: list[t.CInt, 8] = [0] + for i in range(8): + arr[i] = (i + 1) * (i + 1) + p: t.CInt | t.CPtr = c.Addr(arr) + # Access arr[3] via pointer: p + 3 + val: t.CInt = c.Deref(p + 3) + # arr[3] = 4*4 = 16 + testcheck.check(val == 16, "ptr arithmetic (arr[3]=16)", "ptr arithmetic (arr[3] expect 16)") + +# ============================================================ +# Test 9: 字符串比较 - 逐字符指针 +# ============================================================ +def test_manual_strcmp(): + testcheck.section("Test 9: manual strcmp via pointers") + s1: t.CChar | t.CPtr = "Hello" + s2: t.CChar | t.CPtr = "Hello" + p1: t.CChar | t.CPtr = s1 + p2: t.CChar | t.CPtr = s2 + equal: t.CInt = 1 + while c.Deref(p1) != 0 and c.Deref(p2) != 0: + if c.Deref(p1) != c.Deref(p2): + equal = 0 + break + p1 += 1 + p2 += 1 + if c.Deref(p1) != c.Deref(p2): + equal = 0 + testcheck.check(equal == 1, "manual strcmp (equal=1)", "manual strcmp (equal expect 1)") + +# ============================================================ +# Test 10: 结构体指针访问 +# ============================================================ +class Vec2: + x: t.CInt + y: t.CInt + +def test_struct_ptr_access(): + testcheck.section("Test 10: struct pointer access") + v: Vec2 = Vec2() + v.x = 10 + v.y = 20 + p: Vec2 | t.CPtr = c.Addr(v) + # Modify through pointer + c.Deref(p).x = 100 + c.Deref(p).y = 200 + testcheck.check(v.x == 100 and v.y == 200, "struct ptr (x=100 y=200)", "struct ptr (x y expect 100 200)") + +def main() -> t.CInt: + testcheck.begin("StringIterTest: String/Pointer Iteration Tests") + test_string_ptr_iter() + test_string_ascii_sum() + test_manual_strcpy() + test_ptr_offset_compare() + test_manual_strlen() + test_reverse_string_ptr() + test_double_pointer() + test_ptr_arithmetic() + test_manual_strcmp() + test_struct_ptr_access() + return testcheck.end() diff --git a/Test/StringIterTest/output/924ed8cd4ea5c4c2.deps.json b/Test/StringIterTest/output/924ed8cd4ea5c4c2.deps.json new file mode 100644 index 0000000..eafa78c --- /dev/null +++ b/Test/StringIterTest/output/924ed8cd4ea5c4c2.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/StringIterTest/project.json b/Test/StringIterTest/project.json new file mode 100644 index 0000000..bc1efde --- /dev/null +++ b/Test/StringIterTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "StringIterTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "StringIterTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/StringIterTest/temp/271ea3decb810db2.pyi b/Test/StringIterTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/StringIterTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/StringIterTest/temp/6f62fe05c5ea1ceb.pyi b/Test/StringIterTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/StringIterTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/StringIterTest/temp/7e529fe7a078cfef.pyi b/Test/StringIterTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/StringIterTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/StringIterTest/temp/924ed8cd4ea5c4c2.doc.json b/Test/StringIterTest/temp/924ed8cd4ea5c4c2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/StringIterTest/temp/924ed8cd4ea5c4c2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/StringIterTest/temp/924ed8cd4ea5c4c2.pyi b/Test/StringIterTest/temp/924ed8cd4ea5c4c2.pyi new file mode 100644 index 0000000..16c5c7a --- /dev/null +++ b/Test/StringIterTest/temp/924ed8cd4ea5c4c2.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import c +import testcheck + +def test_string_ptr_iter() -> t.CInt: pass + +def test_string_ascii_sum() -> t.CInt: pass + +def test_manual_strcpy() -> t.CInt: pass + +def test_ptr_offset_compare() -> t.CInt: pass + +def test_manual_strlen() -> t.CInt: pass + +def test_reverse_string_ptr() -> t.CInt: pass + +def test_double_pointer() -> t.CInt: pass + +def test_ptr_arithmetic() -> t.CInt: pass + +def test_manual_strcmp() -> t.CInt: pass + + +class Vec2: + x: t.CInt + y: t.CInt + +def test_struct_ptr_access() -> t.CInt: pass + +def main() -> t.CInt: pass diff --git a/Test/StringIterTest/temp/_phase1_manifest.json b/Test/StringIterTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..3764bfc --- /dev/null +++ b/Test/StringIterTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\StringIterTest\\App\\main.py": {"sha1": "924ed8cd4ea5c4c2", "mtime": 1782271507.3310895, "size": 6472}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/StringIterTest/temp/_sha1_map.txt b/Test/StringIterTest/temp/_sha1_map.txt new file mode 100644 index 0000000..e03ebeb --- /dev/null +++ b/Test/StringIterTest/temp/_sha1_map.txt @@ -0,0 +1,11 @@ +271ea3decb810db2:includes/atom.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +924ed8cd4ea5c4c2:main.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/StringIterTest/temp/ab6e54ba9a669f76.pyi b/Test/StringIterTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/StringIterTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/StringIterTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/StringIterTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/StringIterTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/StringIterTest/temp/c9f4be41ca1cc2b4.pyi b/Test/StringIterTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/StringIterTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/StringIterTest/temp/dd3002730623424b.pyi b/Test/StringIterTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/StringIterTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/StringIterTest/temp/ee084e9fc6ee413a.pyi b/Test/StringIterTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/StringIterTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/StringIterTest/temp/f2ecbf8ced20575c.pyi b/Test/StringIterTest/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/Test/StringIterTest/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/Test/StringIterTest/temp/f5522571bcce7bcb.pyi b/Test/StringIterTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/StringIterTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/StringTest/App/main.py b/Test/StringTest/App/main.py new file mode 100644 index 0000000..daf1236 --- /dev/null +++ b/Test/StringTest/App/main.py @@ -0,0 +1,84 @@ +import t +import stdio +import string +import testcheck + +def test_strlen(): + testcheck.section("Test 1: strlen") + s: t.CChar | t.CPtr = "Hello, World!" + length: t.CUInt64T = string.strlen(s) + testcheck.check(length == 13, "strlen (len=13)", "strlen (expect 13)") + +def test_strcpy(): + testcheck.section("Test 2: strcpy") + buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 64) + string.strcpy(buf, "test string") + cmp_result: t.CInt = string.strcmp(buf, "test string") + testcheck.check(cmp_result == 0, "strcpy (result=test string)", "strcpy (cmp mismatch)") + +def test_strcat(): + testcheck.section("Test 3: strcat (via memcpy)") + buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 64) + # 手动拼接: 先拷贝 "Hello"，再拷贝 " World" + hello: t.CChar | t.CPtr = "Hello" + world: t.CChar | t.CPtr = " World" + string.memcpy(buf, hello, 5) + string.memcpy(t.CVoid(t.CUInt64T(buf) + 5, t.CPtr), world, 6) + cmp_result: t.CInt = string.strcmp(buf, "Hello World") + testcheck.check(cmp_result == 0, "strcat via memcpy", "strcat via memcpy (cmp mismatch)") + +def test_strcmp(): + testcheck.section("Test 4: strcmp") + r1: t.CInt = string.strcmp("abc", "abc") + r2: t.CInt = string.strcmp("abc", "abd") + r3: t.CInt = string.strcmp("abd", "abc") + testcheck.check(r1 == 0 and r2 < 0 and r3 > 0, "strcmp (eq=0 lt<0 gt>0)", "strcmp (expect eq=0 lt<0 gt>0)") + +def test_strncpy(): + testcheck.section("Test 5: strncpy") + buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 64) + string.strncpy(buf, "ABCDEFGHIJ", 5) + cmp_result: t.CInt = string.strcmp(buf, "ABCDE") + testcheck.check(cmp_result == 0, "strncpy (result=ABCDE)", "strncpy (cmp mismatch)") + +def test_char_access(): + testcheck.section("Test 6: char access") + s: t.CChar | t.CPtr = "ABCDE" + val: t.CUInt8T = s[2] # 'C' = 67 + testcheck.check(val == 67, "char access (s[2]=67='C')", "char access (s[2] expect 67)") + +def test_snprintf(): + testcheck.section("Test 7: snprintf") + buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 128, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 128) + stdio.snprintf(buf, 128, "value=%d hex=0x%x str=%s", 42, 255, "test") + cmp_result: t.CInt = string.strcmp(buf, "value=42 hex=0xff str=test") + testcheck.check(cmp_result == 0, "snprintf (result=value=42 hex=0xff str=test)", "snprintf (result mismatch)") + +import w32.win32memory + +def main() -> t.CInt: + testcheck.begin("StringTest: 字符串操作基准测试") + test_strlen() + test_strcpy() + test_strcat() + test_strcmp() + test_strncpy() + test_char_access() + test_snprintf() + return testcheck.end() diff --git a/Test/StringTest/output/39178dc5ba692b92.deps.json b/Test/StringTest/output/39178dc5ba692b92.deps.json new file mode 100644 index 0000000..9da11e2 --- /dev/null +++ b/Test/StringTest/output/39178dc5ba692b92.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/StringTest/project.json b/Test/StringTest/project.json new file mode 100644 index 0000000..38832ea --- /dev/null +++ b/Test/StringTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "StringTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "StringTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/StringTest/temp/0035c95a18d4f8e8.pyi b/Test/StringTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/StringTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/StringTest/temp/067c78e9f121dce3.pyi b/Test/StringTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/StringTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/StringTest/temp/06f53cc594b4ac6c.pyi b/Test/StringTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/StringTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/StringTest/temp/39178dc5ba692b92.doc.json b/Test/StringTest/temp/39178dc5ba692b92.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/StringTest/temp/39178dc5ba692b92.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/StringTest/temp/39178dc5ba692b92.pyi b/Test/StringTest/temp/39178dc5ba692b92.pyi new file mode 100644 index 0000000..6b53248 --- /dev/null +++ b/Test/StringTest/temp/39178dc5ba692b92.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +def test_strlen() -> t.CInt: pass + +def test_strcpy() -> t.CInt: pass + +def test_strcat() -> t.CInt: pass + +def test_strcmp() -> t.CInt: pass + +def test_strncpy() -> t.CInt: pass + +def test_char_access() -> t.CInt: pass + +def test_snprintf() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/Test/StringTest/temp/6446627d4f07a1b5.pyi b/Test/StringTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/StringTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/StringTest/temp/6f62fe05c5ea1ceb.pyi b/Test/StringTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/StringTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/StringTest/temp/72e2d5ccb7cedcf1.pyi b/Test/StringTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/StringTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/StringTest/temp/7e529fe7a078cfef.pyi b/Test/StringTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/StringTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/StringTest/temp/_phase1_manifest.json b/Test/StringTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..e223450 --- /dev/null +++ b/Test/StringTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\StringTest\\App\\main.py": {"sha1": "39178dc5ba692b92", "mtime": 1782271507.3310895, "size": 3265}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/StringTest/temp/_sha1_map.txt b/Test/StringTest/temp/_sha1_map.txt new file mode 100644 index 0000000..90054fd --- /dev/null +++ b/Test/StringTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +39178dc5ba692b92:main.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/StringTest/temp/ab6e54ba9a669f76.pyi b/Test/StringTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/StringTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/StringTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/StringTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/StringTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/StringTest/temp/dd3002730623424b.pyi b/Test/StringTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/StringTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/StringTest/temp/f5522571bcce7bcb.pyi b/Test/StringTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/StringTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/StringTest/temp/f6b51804a0ba8ff0.pyi b/Test/StringTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/StringTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/StructTest/App/main.py b/Test/StructTest/App/main.py new file mode 100644 index 0000000..652810a --- /dev/null +++ b/Test/StructTest/App/main.py @@ -0,0 +1,85 @@ +import t +import stdio +import string +import testcheck + +# 定义结构体 +class Point(t.Struct): + x: t.CInt + y: t.CInt + +class Rect(t.Struct): + origin: Point + size: Point + +class Node(t.Struct): + value: t.CInt + next: t.CPtr | t.CVoid # Node* 类型 + +def test_struct_basic(): + testcheck.section("Test 1: struct basic") + p: Point = Point() + p.x = 10 + p.y = 20 + testcheck.check(p.x == 10 and p.y == 20, "struct basic (x=10 y=20)", "struct basic (x y expect 10 20)") + +def test_struct_nested(): + testcheck.section("Test 2: struct nested") + r: Rect = Rect() + r.origin.x = 1 + r.origin.y = 2 + r.size.x = 100 + r.size.y = 200 + testcheck.check(r.origin.x == 1 and r.origin.y == 2 and r.size.x == 100 and r.size.y == 200, "struct nested (origin=1,2 size=100,200)", "struct nested (mismatch)") + +def test_struct_pointer(): + testcheck.section("Test 3: struct pointer") + p: Point = Point() + p.x = 42 + p.y = 84 + # 取结构体指针 + pp: Point | t.CPtr = t.CPtr(t.CVoid(t.CUInt64T(p), t.CPtr)) + # 通过指针修改 + pp.x = 100 + pp.y = 200 + # 检查原始结构体是否被修改 + testcheck.check(p.x == 100 and p.y == 200, "struct pointer (x=100 y=200)", "struct pointer (x y expect 100 200)") + +def test_struct_param(): + testcheck.section("Test 4: struct as parameter") + p: Point = Point() + p.x = 5 + p.y = 15 + sum_val: t.CInt = point_sum(p) + testcheck.check(sum_val == 20, "struct param (sum=20)", "struct param (sum expect 20)") + +def point_sum(p: Point) -> t.CInt: + return p.x + p.y + +def test_struct_array(): + testcheck.section("Test 5: struct array") + # 分配结构体数组 + buf: t.CUInt8T | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 256, 12288, 4) + if t.CUInt64T(buf) == 0: + stdio.printf("SKIP: VirtualAlloc returned NULL\n") + return + string.memset(buf, 0, 256) + # 将缓冲区转为 Point 数组 + pts: Point | t.CPtr = (Point | t.CPtr)(t.CVoid(t.CUInt64T(buf), t.CPtr)) + pts[0].x = 1 + pts[0].y = 2 + pts[1].x = 3 + pts[1].y = 4 + total: t.CInt = pts[0].x + pts[0].y + pts[1].x + pts[1].y + testcheck.check(total == 10, "struct array (total=10)", "struct array (total expect 10)") + +import w32.win32memory + +def main() -> t.CInt: + testcheck.begin("StructTest: 结构体操作基准测试") + test_struct_basic() + test_struct_nested() + test_struct_pointer() + test_struct_param() + test_struct_array() + return testcheck.end() diff --git a/Test/StructTest/output/24e26a8de1173993.deps.json b/Test/StructTest/output/24e26a8de1173993.deps.json new file mode 100644 index 0000000..9da11e2 --- /dev/null +++ b/Test/StructTest/output/24e26a8de1173993.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/StructTest/project.json b/Test/StructTest/project.json new file mode 100644 index 0000000..dc854f7 --- /dev/null +++ b/Test/StructTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "StructTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "StructTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/StructTest/temp/0035c95a18d4f8e8.pyi b/Test/StructTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/StructTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/StructTest/temp/067c78e9f121dce3.pyi b/Test/StructTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/StructTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/StructTest/temp/06f53cc594b4ac6c.pyi b/Test/StructTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/StructTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/StructTest/temp/24e26a8de1173993.doc.json b/Test/StructTest/temp/24e26a8de1173993.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/StructTest/temp/24e26a8de1173993.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/StructTest/temp/24e26a8de1173993.pyi b/Test/StructTest/temp/24e26a8de1173993.pyi new file mode 100644 index 0000000..0053550 --- /dev/null +++ b/Test/StructTest/temp/24e26a8de1173993.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import c + + +import t +import stdio +import string +import testcheck + +class Point(t.Struct): + x: t.CInt + y: t.CInt +class Rect(t.Struct): + origin: Point + size: Point +class Node(t.Struct): + value: t.CInt + next: t.CPtr | t.CVoid + +def test_struct_basic() -> t.CInt: pass + +def test_struct_nested() -> t.CInt: pass + +def test_struct_pointer() -> t.CInt: pass + +def test_struct_param() -> t.CInt: pass + +def point_sum(p: Point) -> t.CInt: pass + +def test_struct_array() -> t.CInt: pass + + +import w32.win32memory + +def main() -> t.CInt: pass diff --git a/Test/StructTest/temp/6446627d4f07a1b5.pyi b/Test/StructTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/StructTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/StructTest/temp/6f62fe05c5ea1ceb.pyi b/Test/StructTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/StructTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/StructTest/temp/72e2d5ccb7cedcf1.pyi b/Test/StructTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/StructTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/StructTest/temp/7e529fe7a078cfef.pyi b/Test/StructTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/StructTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/StructTest/temp/_phase1_manifest.json b/Test/StructTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..9b92f7a --- /dev/null +++ b/Test/StructTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\StructTest\\App\\main.py": {"sha1": "24e26a8de1173993", "mtime": 1782271507.3310895, "size": 2500}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/StructTest/temp/_sha1_map.txt b/Test/StructTest/temp/_sha1_map.txt new file mode 100644 index 0000000..fdfce98 --- /dev/null +++ b/Test/StructTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +24e26a8de1173993:main.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/StructTest/temp/ab6e54ba9a669f76.pyi b/Test/StructTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/StructTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/StructTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/StructTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/StructTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/StructTest/temp/dd3002730623424b.pyi b/Test/StructTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/StructTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/StructTest/temp/f5522571bcce7bcb.pyi b/Test/StructTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/StructTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/StructTest/temp/f6b51804a0ba8ff0.pyi b/Test/StructTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/StructTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/SysLibTest/App/main.py b/Test/SysLibTest/App/main.py new file mode 100644 index 0000000..9b4c375 --- /dev/null +++ b/Test/SysLibTest/App/main.py @@ -0,0 +1,38 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import memhub +import stdlib +import sys +import os +import os._win32 +import os._posix +import os.path +import shutil +import subprocess +import syslibtest_impl + + +# MBuddy arena 大小：1MB +ARENA_SIZE: t.CDefine = 1048576 + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + # 初始化全局 MBuddy 内存分配器 + arena: bytes = stdlib.malloc(ARENA_SIZE) + mbuddy_obj: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, ARENA_SIZE) + + # 将 MBuddy 对象赋值给所有需要内存分配的模块 + sys._mbuddy = mbuddy_obj + os._mbuddy = mbuddy_obj + os._win32._mbuddy = mbuddy_obj + os._posix._mbuddy = mbuddy_obj + os.path._mbuddy = mbuddy_obj + shutil._mbuddy = mbuddy_obj + subprocess._mbuddy = mbuddy_obj + + return syslibtest_impl.run() diff --git a/Test/SysLibTest/App/os_backend.py b/Test/SysLibTest/App/os_backend.py new file mode 100644 index 0000000..cf9a5f1 --- /dev/null +++ b/Test/SysLibTest/App/os_backend.py @@ -0,0 +1,39 @@ +import t, c +from stdint import * +import platmacro +import os._win32 +import os._posix + + +# ============================================================================= +# Workaround: TransPyC 无法正确解析 os.xxx() 包级函数调用 +# (os.path.xxx() 子模块调用正常，但 os.xxx() 包级调用解析为本地模块) +# 此文件单独导入 os._win32 / os._posix 子模块，避免污染调用方的 os.rmdir() 等解析。 +# ============================================================================= + +def os_exists(path: str) -> bool: + """检查文件/目录是否存在（跨平台）""" + if platmacro.IS_WINDOWS: + return os._win32.exists(path) + return os._posix.exists(path) + + +def os_remove(path: str) -> int: + """删除文件（跨平台）。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.remove(path) + return os._posix.remove(path) + + +def os_isdir(path: str) -> bool: + """检查路径是否为目录（跨平台）""" + if platmacro.IS_WINDOWS: + return os._win32.isdir(path) + return os._posix.isdir(path) + + +def os_getsize(path: str) -> t.CInt64T: + """返回文件大小（字节，跨平台）。失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.getsize(path) + return os._posix.getsize(path) diff --git a/Test/SysLibTest/App/syslibtest_impl.py b/Test/SysLibTest/App/syslibtest_impl.py new file mode 100644 index 0000000..7ea841b --- /dev/null +++ b/Test/SysLibTest/App/syslibtest_impl.py @@ -0,0 +1,307 @@ +import t, c +from stdint import * +from t import CInt, CPtr, CUInt8T, CSizeT +import testcheck +import viperlib +import stdlib +import memhub +import platmacro +import os +import os.path +import os_backend +import shutil +import sys +import subprocess +import string +import w32.fileio +import w32.win32base +import w32.win32process + + +# 测试用临时文件/目录名前缀 +TEST_DIR: t.CDefine = "__sylibtest_dir__" +TEST_FILE: t.CDefine = "__sylibtest_file.txt" +TEST_COPY: t.CDefine = "__sylibtest_copy.txt" +TEST_TREE: t.CDefine = "__sylibtest_tree__" +TEST_MOVE: t.CDefine = "__sylibtest_moved.txt" + + +# ============================================================================= +# Workaround: TransPyC 无法正确解析 os.xxx() 包级函数调用 +# (os.path.xxx() 子模块调用正常，但 os.xxx() 包级调用解析为本地模块) +# 通过 os_backend 模块（单独导入 os._win32 / os._posix）绕过此问题 +# ============================================================================= + +def _cleanup(): + """清理可能残留的测试文件/目录""" + testcheck.info("DBG: cleanup start") + if os_backend.os_exists(TEST_DIR): + testcheck.info("DBG: cleanup rmdir TEST_DIR") + os.rmdir(TEST_DIR) + testcheck.info("DBG: cleanup after TEST_DIR") + if os_backend.os_exists(TEST_FILE): + os_backend.os_remove(TEST_FILE) + testcheck.info("DBG: cleanup after TEST_FILE") + if os_backend.os_exists(TEST_COPY): + os_backend.os_remove(TEST_COPY) + testcheck.info("DBG: cleanup after TEST_COPY") + if os_backend.os_exists(TEST_MOVE): + os_backend.os_remove(TEST_MOVE) + testcheck.info("DBG: cleanup after TEST_MOVE") + if os_backend.os_exists(TEST_TREE): + testcheck.info("DBG: cleanup rmtree TEST_TREE") + shutil.rmtree(TEST_TREE) + testcheck.info("DBG: cleanup done") + + +def run() -> CInt: + buf: bytes = stdlib.malloc(512) + + testcheck.begin("SysLibTest: 系统兼容层库测试") + testcheck.info("DBG: after begin") + + # 清理残留 + _cleanup() + testcheck.info("DBG: after cleanup") + + # ============================================================================= + # 全局 MBuddy 初始化验证（在 main 中已初始化） + # ============================================================================= + testcheck.section("全局 MBuddy 初始化验证") + testcheck.info("DBG: after section") + + testcheck.check(os._mbuddy != None, "os._mbuddy initialized OK", "os._mbuddy NOT initialized") + testcheck.check(sys._mbuddy != None, "sys._mbuddy initialized OK", "sys._mbuddy NOT initialized") + testcheck.check(shutil._mbuddy != None, "shutil._mbuddy initialized OK", "shutil._mbuddy NOT initialized") + testcheck.check(subprocess._mbuddy != None, "subprocess._mbuddy initialized OK", "subprocess._mbuddy NOT initialized") + + # 所有模块应共享同一个 MBuddy 对象 + testcheck.check(os._mbuddy == sys._mbuddy, "os==sys same MBuddy OK", "os!=sys different MBuddy") + testcheck.check(shutil._mbuddy == os._mbuddy, "shutil==os same MBuddy OK", "shutil!=os different MBuddy") + + # ============================================================================= + # MBuddy alloc/free/calloc/realloc（通过 os._mbuddy 测试） + # ============================================================================= + testcheck.section("MBuddy alloc/free") + + p1: bytes = os._mbuddy.alloc(64) + p2: bytes = os._mbuddy.alloc(128) + testcheck.check(p1 != None and p2 != None, "alloc OK", "alloc FAILED") + + p1_c: CUInt8T | CPtr = CPtr(p1) + p1_c[0] = CUInt8T(0xAB) + p1_c[1] = CUInt8T(0xCD) + testcheck.check(p1_c[0] == CUInt8T(0xAB) and p1_c[1] == CUInt8T(0xCD), + "alloc write/read OK", "alloc write/read FAILED") + + os._mbuddy.free(p1) + os._mbuddy.free(p2) + testcheck.ok("free OK") + + # calloc 清零测试 + testcheck.section("MBuddy calloc/realloc") + + cp: bytes = os._mbuddy.calloc(4, 16) + if cp != None: + cp_c: CUInt8T | CPtr = CPtr(cp) + all_zero: CInt = 1 + i: CInt + for i in range(64): + if cp_c[i] != 0: + all_zero = 0 + break + testcheck.check(all_zero == 1, "calloc zeroed OK", "calloc NOT zeroed") + os._mbuddy.free(cp) + else: + testcheck.fail("calloc FAILED") + + # realloc 数据保留测试 + rp: bytes = os._mbuddy.alloc(32) + rp_c: CUInt8T | CPtr = CPtr(rp) + for i in range(32): + rp_c[i] = CUInt8T(i + 1) + rp2: bytes = os._mbuddy.realloc(rp, 128) + if rp2 != None: + rp2_c: CUInt8T | CPtr = CPtr(rp2) + ok: CInt = 1 + for i in range(32): + if rp2_c[i] != CUInt8T(i + 1): + ok = 0 + break + testcheck.check(ok == 1, "realloc data preserved OK", "realloc data LOST") + os._mbuddy.free(rp2) + else: + testcheck.fail("realloc FAILED") + + # ============================================================================= + # os.path 路径操作 + # ============================================================================= + testcheck.section("os.path 路径操作") + + # sep + sep_char: t.CChar = os.path.sep() + viperlib.snprintf(buf, 512, "sep = %c", sep_char) + testcheck.info(buf) + + # join + joined: str = os.path.join("parent", "child") + testcheck.check(joined != None, "join(parent, child) OK", "join FAILED") + testcheck.info(joined) + + # basename + base: str = os.path.basename("/usr/local/bin/test") + testcheck.check(base != None, "basename OK", "basename FAILED") + testcheck.info(base) + + # dirname + dirn: str = os.path.dirname("/usr/local/bin/test") + testcheck.check(dirn != None, "dirname OK", "dirname FAILED") + testcheck.info(dirn) + + # isabs + testcheck.check(os.path.isabs("/usr/local"), "isabs(/usr/local) OK", "isabs FAILED") + testcheck.check(not os.path.isabs("relative/path"), "isabs(relative)=False OK", "isabs(relative) should be False") + + # ============================================================================= + # os 文件/目录操作 + # ============================================================================= + testcheck.section("os 文件/目录操作") + + # getcwd + cwd: str = os.getcwd() + testcheck.check(cwd != None, "getcwd OK", "getcwd FAILED") + testcheck.info(cwd) + + # exists + testcheck.check(os_backend.os_exists("."), "exists(.) OK", "exists(.) FAILED") + testcheck.check(not os_backend.os_exists("__totally_nonexistent__"), "exists(nonexistent)=False OK", "exists(nonexistent) should be False") + + # isdir / isfile + testcheck.check(os_backend.os_isdir("."), "isdir(.) OK", "isdir(.) FAILED") + + # mkdir + rmdir + testcheck.check(os.mkdir(TEST_DIR) == 0, "mkdir OK", "mkdir FAILED") + testcheck.check(os_backend.os_exists(TEST_DIR), "mkdir: exists OK", "mkdir: not exists") + testcheck.check(os_backend.os_isdir(TEST_DIR), "mkdir: isdir OK", "mkdir: not isdir") + testcheck.check(os.rmdir(TEST_DIR) == 0, "rmdir OK", "rmdir FAILED") + testcheck.check(not os_backend.os_exists(TEST_DIR), "rmdir: gone OK", "rmdir: still exists") + + # rename + testcheck.check(os.mkdir(TEST_DIR) == 0, "mkdir for rename OK", "mkdir for rename FAILED") + testcheck.check(os.rename(TEST_DIR, TEST_TREE) == 0, "rename OK", "rename FAILED") + testcheck.check(os_backend.os_exists(TEST_TREE) and not os_backend.os_exists(TEST_DIR), "rename: swapped OK", "rename: swap FAILED") + testcheck.check(os.rmdir(TEST_TREE) == 0, "rmdir renamed OK", "rmdir renamed FAILED") + + # ============================================================================= + # shutil 高级文件操作 + # ============================================================================= + testcheck.section("shutil 高级文件操作") + + # 创建测试文件 + f: w32.fileio.File | CPtr = w32.fileio.File(TEST_FILE, w32.fileio.MODE.W) + testcheck.check(not f.closed, "create test file OK", "create test file FAILED") + f.write_str("hello shutil") + f.close() + testcheck.check(os_backend.os_exists(TEST_FILE), "test file exists OK", "test file not exists") + + # copyfile + testcheck.check(shutil.copyfile(TEST_FILE, TEST_COPY) == 0, "copyfile OK", "copyfile FAILED") + testcheck.check(os_backend.os_exists(TEST_COPY), "copyfile: dst exists OK", "copyfile: dst not exists") + + # 验证拷贝内容一致（检查文件大小） + src_size: t.CInt64T = os_backend.os_getsize(TEST_FILE) + dst_size: t.CInt64T = os_backend.os_getsize(TEST_COPY) + viperlib.snprintf(buf, 512, "src_size=%lld dst_size=%lld", src_size, dst_size) + testcheck.info(buf) + testcheck.check(src_size == dst_size, "copyfile: same size OK", "copyfile: size mismatch") + + # move + testcheck.check(shutil.move(TEST_COPY, TEST_MOVE) == 0, "move OK", "move FAILED") + testcheck.check(os_backend.os_exists(TEST_MOVE), "move: dst exists OK", "move: dst not exists") + os_backend.os_remove(TEST_MOVE) + + # rmtree（递归删除目录树） + testcheck.check(os.mkdir(TEST_TREE) == 0, "mkdir tree OK", "mkdir tree FAILED") + # 在树中创建子目录 + subdir: str = os.path.join(TEST_TREE, "sub") + testcheck.check(os.mkdir(subdir) == 0, "mkdir subdir OK", "mkdir subdir FAILED") + # 在子目录中创建文件 + subfile: str = os.path.join(subdir, "leaf.txt") + sf: w32.fileio.File | CPtr = w32.fileio.File(subfile, w32.fileio.MODE.W) + sf.write_str("leaf") + sf.close() + testcheck.check(os_backend.os_exists(subfile), "create subfile OK", "create subfile FAILED") + + # rmtree + testcheck.check(shutil.rmtree(TEST_TREE) == 0, "rmtree OK", "rmtree FAILED") + testcheck.check(not os_backend.os_exists(TEST_TREE), "rmtree: gone OK", "rmtree: still exists") + + # 清理剩余文件 + if os_backend.os_exists(TEST_FILE): + os_backend.os_remove(TEST_FILE) + if os_backend.os_exists(TEST_COPY): + os_backend.os_remove(TEST_COPY) + + # ============================================================================= + # sys 系统信息 + # ============================================================================= + testcheck.section("sys 系统信息") + + # platform + plat: str = sys.platform() + testcheck.check(plat != None, "platform() OK", "platform() FAILED") + testcheck.info(plat) + + # getpid + pid: CInt = sys.get_pid() + viperlib.snprintf(buf, 512, "pid = %d", pid) + testcheck.info(buf) + testcheck.check(pid > 0, "getpid > 0 OK", "getpid FAILED") + + # maxsize + viperlib.snprintf(buf, 512, "maxsize = %lld", sys.maxsize) + testcheck.info(buf) + testcheck.check(sys.maxsize > 0, "maxsize > 0 OK", "maxsize FAILED") + + # argv + argc: CInt = sys.argc() + viperlib.snprintf(buf, 512, "argc = %d", argc) + testcheck.info(buf) + testcheck.check(argc > 0, "argc > 0 OK", "argc FAILED") + + argv0: str = sys.argv(0) + testcheck.check(argv0 != None, "argv(0) OK", "argv(0) FAILED") + testcheck.info(argv0) + + # 越界访问 + testcheck.check(sys.argv(9999) == None, "argv(out-of-range)=None OK", "argv(out-of-range) should be None") + + # ============================================================================= + # subprocess 子进程 + # ============================================================================= + testcheck.section("subprocess 子进程") + + # Windows: cmd /c echo hello + result: subprocess.CompletedProcess | CPtr = subprocess.run("cmd /c echo hello", True, True) + testcheck.check(result != None, "run OK", "run FAILED") + if result != None: + viperlib.snprintf(buf, 512, "returncode = %d", result.returncode) + testcheck.info(buf) + testcheck.check(result.returncode == 0, "returncode == 0 OK", "returncode != 0") + if result.stdout != None: + testcheck.info(result.stdout) + # 验证输出包含 "hello" + testcheck.check(string.strstr(result.stdout, "hello") != None, + "stdout contains 'hello' OK", "stdout missing 'hello'") + else: + testcheck.fail("stdout is None") + else: + testcheck.fail("run returned None") + + # ============================================================================= + # 清理 + # ============================================================================= + _cleanup() + + stdlib.free(buf) + return testcheck.end() diff --git a/Test/SysLibTest/output/59e00d50660b941e.deps.json b/Test/SysLibTest/output/59e00d50660b941e.deps.json new file mode 100644 index 0000000..7a3a27b --- /dev/null +++ b/Test/SysLibTest/output/59e00d50660b941e.deps.json @@ -0,0 +1 @@ +{"w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "os.path": "13110effbb0bb06c", "path": "13110effbb0bb06c", "atom": "271ea3decb810db2", "subprocess": "2da636c61863c815", "posix": "6503c97dde0c79c4", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "os.__init__": "76dd6c275aa72b3b", "__init__": "76dd6c275aa72b3b", "os": "76dd6c275aa72b3b", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "syslibtest_impl": "a03139660029514f", "os._win32": "a68b70f233541a7f", "_win32": "a68b70f233541a7f", "string": "ab6e54ba9a669f76", "shutil": "abbcbd92436cc16a", "sys": "b5a965302cded36b", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "main": "c70684ccfe48a5a9", "viperio": "c9f4be41ca1cc2b4", "os._posix": "d152115b49e50218", "_posix": "d152115b49e50218", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/SysLibTest/output/a03139660029514f.deps.json b/Test/SysLibTest/output/a03139660029514f.deps.json new file mode 100644 index 0000000..d1a0bbe --- /dev/null +++ b/Test/SysLibTest/output/a03139660029514f.deps.json @@ -0,0 +1 @@ +{"w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "os.path": "13110effbb0bb06c", "path": "13110effbb0bb06c", "atom": "271ea3decb810db2", "subprocess": "2da636c61863c815", "os_backend": "59e00d50660b941e", "posix": "6503c97dde0c79c4", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "os.__init__": "76dd6c275aa72b3b", "__init__": "76dd6c275aa72b3b", "os": "76dd6c275aa72b3b", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "os._win32": "a68b70f233541a7f", "_win32": "a68b70f233541a7f", "string": "ab6e54ba9a669f76", "shutil": "abbcbd92436cc16a", "sys": "b5a965302cded36b", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "main": "c70684ccfe48a5a9", "viperio": "c9f4be41ca1cc2b4", "os._posix": "d152115b49e50218", "_posix": "d152115b49e50218", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/SysLibTest/output/c70684ccfe48a5a9.deps.json b/Test/SysLibTest/output/c70684ccfe48a5a9.deps.json new file mode 100644 index 0000000..932a326 --- /dev/null +++ b/Test/SysLibTest/output/c70684ccfe48a5a9.deps.json @@ -0,0 +1 @@ +{"w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "os.path": "13110effbb0bb06c", "path": "13110effbb0bb06c", "atom": "271ea3decb810db2", "subprocess": "2da636c61863c815", "os_backend": "59e00d50660b941e", "posix": "6503c97dde0c79c4", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "os.__init__": "76dd6c275aa72b3b", "__init__": "76dd6c275aa72b3b", "os": "76dd6c275aa72b3b", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "syslibtest_impl": "a03139660029514f", "os._win32": "a68b70f233541a7f", "_win32": "a68b70f233541a7f", "string": "ab6e54ba9a669f76", "shutil": "abbcbd92436cc16a", "sys": "b5a965302cded36b", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "os._posix": "d152115b49e50218", "_posix": "d152115b49e50218", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/SysLibTest/project.json b/Test/SysLibTest/project.json new file mode 100644 index 0000000..4a42e9c --- /dev/null +++ b/Test/SysLibTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "SysLibTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "SysLibTest.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/SysLibTest/temp/0035c95a18d4f8e8.pyi b/Test/SysLibTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/SysLibTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/SysLibTest/temp/067c78e9f121dce3.pyi b/Test/SysLibTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/SysLibTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/SysLibTest/temp/06f53cc594b4ac6c.pyi b/Test/SysLibTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/SysLibTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/SysLibTest/temp/13110effbb0bb06c.pyi b/Test/SysLibTest/temp/13110effbb0bb06c.pyi new file mode 100644 index 0000000..3584bd2 --- /dev/null +++ b/Test/SysLibTest/temp/13110effbb0bb06c.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from os.path.py +Module: os.path +""" + + +import t, c +from stdint import * +import string +import memhub +import platmacro + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def sep() -> t.CChar: pass + +def _is_sep(ch: t.CChar) -> bool: pass + +def _dup_str(s: str) -> str: pass + +def _substr(s: str, start: t.CSizeT, end: t.CSizeT) -> str: pass + +def isabs(path: str) -> bool: pass + +def join(path1: str, path2: str) -> str: pass + +def basename(path: str) -> str: pass + +def dirname(path: str) -> str: pass + +def splitext_name(path: str) -> str: pass + +def splitext_ext(path: str) -> str: pass + +def normpath(path: str) -> str: pass diff --git a/Test/SysLibTest/temp/271ea3decb810db2.pyi b/Test/SysLibTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/SysLibTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/SysLibTest/temp/2da636c61863c815.pyi b/Test/SysLibTest/temp/2da636c61863c815.pyi new file mode 100644 index 0000000..7c91519 --- /dev/null +++ b/Test/SysLibTest/temp/2da636c61863c815.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from subprocess.py +Module: subprocess +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import w32.win32base +import w32.win32file +import w32.win32process +import w32.win32sync +import posix + +OUTPUT_BUF_SIZE: t.CDefine = 65536 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class CompletedProcess: + args: str + returncode: int + stdout: str + stderr: str + def __new__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CPtr: pass + def __init__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CInt: pass + +def run(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_win32(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_posix(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass diff --git a/Test/SysLibTest/temp/59e00d50660b941e.doc.json b/Test/SysLibTest/temp/59e00d50660b941e.doc.json new file mode 100644 index 0000000..e833e14 --- /dev/null +++ b/Test/SysLibTest/temp/59e00d50660b941e.doc.json @@ -0,0 +1 @@ +{"os_exists": "检查文件/目录是否存在（跨平台）", "os_remove": "删除文件（跨平台）。成功返回 0，失败返回 -1。", "os_isdir": "检查路径是否为目录（跨平台）", "os_getsize": "返回文件大小（字节，跨平台）。失败返回 -1。"} \ No newline at end of file diff --git a/Test/SysLibTest/temp/59e00d50660b941e.pyi b/Test/SysLibTest/temp/59e00d50660b941e.pyi new file mode 100644 index 0000000..5236edb --- /dev/null +++ b/Test/SysLibTest/temp/59e00d50660b941e.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from os_backend.py +Module: os_backend +""" + + +import t, c +from stdint import * +import platmacro +import os._win32 +import os._posix + +def os_exists(path: str) -> bool: pass + +def os_remove(path: str) -> int: pass + +def os_isdir(path: str) -> bool: pass + +def os_getsize(path: str) -> t.CInt64T: pass diff --git a/Test/SysLibTest/temp/6446627d4f07a1b5.pyi b/Test/SysLibTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/SysLibTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/SysLibTest/temp/6503c97dde0c79c4.pyi b/Test/SysLibTest/temp/6503c97dde0c79c4.pyi new file mode 100644 index 0000000..d332326 --- /dev/null +++ b/Test/SysLibTest/temp/6503c97dde0c79c4.pyi @@ -0,0 +1,189 @@ +""" +Auto-generated Python stub file from posix.py +Module: posix +""" + + +import t, c +from stdint import * + +DIR: t.CTypedef = t.CVoid | t.CPtr +mode_t: t.CTypedef = t.CUInt32T +pid_t: t.CTypedef = t.CInt32T +off_t: t.CTypedef = t.CInt64T +dev_t: t.CTypedef = t.CUInt64T +ino_t: t.CTypedef = t.CUInt64T +nlink_t: t.CTypedef = t.CUInt64T +uid_t: t.CTypedef = t.CUInt32T +gid_t: t.CTypedef = t.CUInt32T +blksize_t: t.CTypedef = t.CInt64T +blkcnt_t: t.CTypedef = t.CInt64T +time_t: t.CTypedef = t.CInt64T +O_RDONLY: t.CDefine = 0 +O_WRONLY: t.CDefine = 1 +O_RDWR: t.CDefine = 2 +O_CREAT: t.CDefine = 0o100 +O_EXCL: t.CDefine = 0o200 +O_TRUNC: t.CDefine = 0o1000 +O_APPEND: t.CDefine = 0o2000 +O_NONBLOCK: t.CDefine = 0o4000 +O_CLOEXEC: t.CDefine = 0o2000000 +F_OK: t.CDefine = 0 # 文件存在 +R_OK: t.CDefine = 4 # 可读 +W_OK: t.CDefine = 2 # 可写 +X_OK: t.CDefine = 1 # 可执行 +S_IFMT: t.CDefine = 0o170000 # 类型掩码 +S_IFSOCK: t.CDefine = 0o140000 # 套接字 +S_IFLNK: t.CDefine = 0o120000 # 符号链接 +S_IFREG: t.CDefine = 0o100000 # 普通文件 +S_IFBLK: t.CDefine = 0o060000 # 块设备 +S_IFDIR: t.CDefine = 0o040000 # 目录 +S_IFCHR: t.CDefine = 0o020000 # 字符设备 +S_IFIFO: t.CDefine = 0o010000 # FIFO/管道 +S_ISUID: t.CDefine = 0o4000 # set-user-ID +S_ISGID: t.CDefine = 0o2000 # set-group-ID +S_ISVTX: t.CDefine = 0o1000 # sticky 位 +S_IRWXU: t.CDefine = 0o700 # 用户读写执行 +S_IRUSR: t.CDefine = 0o400 # 用户读 +S_IWUSR: t.CDefine = 0o200 # 用户写 +S_IXUSR: t.CDefine = 0o100 # 用户执行 +S_IRWXG: t.CDefine = 0o070 # 组读写执行 +S_IRGRP: t.CDefine = 0o040 # 组读 +S_IWGRP: t.CDefine = 0o020 # 组写 +S_IXGRP: t.CDefine = 0o010 # 组执行 +S_IRWXO: t.CDefine = 0o007 # 其他读写执行 +S_IROTH: t.CDefine = 0o004 # 其他读 +S_IWOTH: t.CDefine = 0o002 # 其他写 +S_IXOTH: t.CDefine = 0o001 # 其他执行 +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +PATH_MAX: t.CDefine = 4096 +NAME_MAX: t.CDefine = 255 +_SC_NPROCESSORS_ONLN: t.CDefine = 84 +_SC_PAGESIZE: t.CDefine = 30 +_SC_CLK_TCK: t.CDefine = 2 +DT_UNKNOWN: t.CDefine = 0 +DT_FIFO: t.CDefine = 1 +DT_CHR: t.CDefine = 2 +DT_DIR: t.CDefine = 4 +DT_BLK: t.CDefine = 6 +DT_REG: t.CDefine = 8 +DT_LNK: t.CDefine = 10 +DT_SOCK: t.CDefine = 12 +DT_WHT: t.CDefine = 14 +WNOHANG: t.CDefine = 1 +WUNTRACED: t.CDefine = 2 +WCONTINUED: t.CDefine = 8 + +class Stat: + st_dev: dev_t + st_ino: ino_t + st_nlink: nlink_t + st_mode: mode_t + st_uid: uid_t + st_gid: gid_t + __pad0: INT + st_rdev: dev_t + st_size: off_t + st_blksize: blksize_t + st_blocks: blkcnt_t + st_atim_sec: time_t + st_atim_nsec: LONGLONG + st_mtim_sec: time_t + st_mtim_nsec: LONGLONG + st_ctim_sec: time_t + st_ctim_nsec: LONGLONG + __unused0: LONGLONG + __unused1: LONGLONG + __unused2: LONGLONG +class Dirent: + d_ino: ino_t + d_off: off_t + d_reclen: USHORT + d_type: BYTE + d_name: CHARPTR + +def open(path: t.CConst | str, oflag: INT, *args) -> INT | t.State: pass + +def close(fd: INT) -> INT | t.State: pass + +def read(fd: INT, buf: bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def write(fd: INT, buf: t.CConst | bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def lseek(fd: INT, offset: off_t, whence: INT) -> off_t | t.State: pass + +def unlink(path: t.CConst | str) -> INT | t.State: pass + +def access(path: t.CConst | str, amode: INT) -> INT | t.State: pass + +def realpath(path: t.CConst | str, resolved: CHARPTR) -> CHARPTR | t.State: pass + +def rename(oldpath: t.CConst | str, newpath: t.CConst | str) -> INT | t.State: pass + +def chmod(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def stat(path: t.CConst | str, buf: Stat | t.CPtr) -> INT | t.State: pass + +def fstat(fd: INT, buf: Stat | t.CPtr) -> INT | t.State: pass + +def opendir(name: t.CConst | str) -> DIR | t.CPtr | t.State: pass + +def readdir(dirp: DIR | t.CPtr) -> Dirent | t.CPtr | t.State: pass + +def closedir(dirp: DIR | t.CPtr) -> INT | t.State: pass + +def mkdir(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def rmdir(path: t.CConst | str) -> INT | t.State: pass + +def getcwd(buf: CHARPTR, size: t.CSizeT) -> CHARPTR | t.State: pass + +def chdir(path: t.CConst | str) -> INT | t.State: pass + +def fork() -> pid_t | t.State: pass + +def execvp(file: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def execv(path: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def waitpid(pid: pid_t, status: INT | t.CPtr, options: INT) -> pid_t | t.State: pass + +def pipe(fds: INT | t.CPtr) -> INT | t.State: pass + +def dup2(oldfd: INT, newfd: INT) -> INT | t.State: pass + +def _exit(status: INT) -> t.CVoid | t.State: pass + +def getpid() -> pid_t | t.State: pass + +def getenv(name: t.CConst | str) -> CHARPTR | t.State: pass + +def setenv(name: t.CConst | str, value: t.CConst | str, overwrite: INT) -> INT | t.State: pass + +def sysconf(name: INT) -> LONG | t.State: pass + +def isatty(fd: INT) -> INT | t.State: pass + +def S_ISREG(mode: mode_t) -> bool: pass + +def S_ISDIR(mode: mode_t) -> bool: pass + +def S_ISLNK(mode: mode_t) -> bool: pass + +def S_ISCHR(mode: mode_t) -> bool: pass + +def S_ISBLK(mode: mode_t) -> bool: pass + +def S_ISFIFO(mode: mode_t) -> bool: pass + +def S_ISSOCK(mode: mode_t) -> bool: pass + +def WIFEXITED(status: INT) -> bool: pass + +def WEXITSTATUS(status: INT) -> INT: pass + +def WIFSIGNALED(status: INT) -> bool: pass + +def WTERMSIG(status: INT) -> INT: pass diff --git a/Test/SysLibTest/temp/6f62fe05c5ea1ceb.pyi b/Test/SysLibTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/SysLibTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/SysLibTest/temp/71e0a3ffcb3ebfad.pyi b/Test/SysLibTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/SysLibTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/SysLibTest/temp/72e2d5ccb7cedcf1.pyi b/Test/SysLibTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/SysLibTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/SysLibTest/temp/76dd6c275aa72b3b.pyi b/Test/SysLibTest/temp/76dd6c275aa72b3b.pyi new file mode 100644 index 0000000..cd52873 --- /dev/null +++ b/Test/SysLibTest/temp/76dd6c275aa72b3b.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from os.__init__.py +Module: os.__init__ +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import os.path +import os._win32 +import os._posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd() -> str: pass + +def chdir(path: str) -> int: pass + +def mkdir(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir(path: str) -> int: pass + +def rename(old_path: str, new_path: str) -> int: pass + +def chmod(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system(cmd: str) -> int: pass diff --git a/Test/SysLibTest/temp/7e529fe7a078cfef.pyi b/Test/SysLibTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/SysLibTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/SysLibTest/temp/90c53dd6db8d41cf.pyi b/Test/SysLibTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/SysLibTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/SysLibTest/temp/93c1d18e35d188d6.pyi b/Test/SysLibTest/temp/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/Test/SysLibTest/temp/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/Test/SysLibTest/temp/_phase1_manifest.json b/Test/SysLibTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..288b982 --- /dev/null +++ b/Test/SysLibTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\SysLibTest\\App\\main.py": {"sha1": "c70684ccfe48a5a9", "mtime": 1782828223.3150382, "size": 918}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\SysLibTest\\App\\os_backend.py": {"sha1": "59e00d50660b941e", "mtime": 1782491896.4070306, "size": 1323}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\SysLibTest\\App\\syslibtest_impl.py": {"sha1": "a03139660029514f", "mtime": 1782828233.2147534, "size": 12395}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\os\\__init__.py": {"sha1": "76dd6c275aa72b3b", "mtime": 1782826646.8940291, "size": 4395}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\os\\_posix.py": {"sha1": "d152115b49e50218", "mtime": 1782826648.538678, "size": 5665}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\os\\_win32.py": {"sha1": "a68b70f233541a7f", "mtime": 1782826646.872196, "size": 7601}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\os\\path.py": {"sha1": "13110effbb0bb06c", "mtime": 1782826648.9189062, "size": 9499}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\platmacro.py": {"sha1": "93c1d18e35d188d6", "mtime": 1781237231.03366, "size": 1422}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\posix.py": {"sha1": "6503c97dde0c79c4", "mtime": 1783064077.6376991, "size": 10386}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\shutil.py": {"sha1": "abbcbd92436cc16a", "mtime": 1782826619.0231705, "size": 9131}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\subprocess.py": {"sha1": "2da636c61863c815", "mtime": 1783847277.4569364, "size": 7627}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\sys.py": {"sha1": "b5a965302cded36b", "mtime": 1783569641.07535, "size": 5509}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\fileio.py": {"sha1": "0035c95a18d4f8e8", "mtime": 1783747022.686592, "size": 14147}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32file.py": {"sha1": "f6b51804a0ba8ff0", "mtime": 1782531158.744144, "size": 8424}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32process.py": {"sha1": "067c78e9f121dce3", "mtime": 1781200703.5409546, "size": 4802}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32sync.py": {"sha1": "06f53cc594b4ac6c", "mtime": 1782297500.2581685, "size": 4726}} \ No newline at end of file diff --git a/Test/SysLibTest/temp/_sha1_map.txt b/Test/SysLibTest/temp/_sha1_map.txt new file mode 100644 index 0000000..a6b98c8 --- /dev/null +++ b/Test/SysLibTest/temp/_sha1_map.txt @@ -0,0 +1,28 @@ +0035c95a18d4f8e8:includes/w32\fileio.py +067c78e9f121dce3:includes/w32\win32process.py +06f53cc594b4ac6c:includes/w32\win32sync.py +13110effbb0bb06c:includes/os\path.py +271ea3decb810db2:includes/atom.py +2da636c61863c815:includes/subprocess.py +59e00d50660b941e:os_backend.py +6503c97dde0c79c4:includes/posix.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +76dd6c275aa72b3b:includes/os\__init__.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +93c1d18e35d188d6:includes/platmacro.py +a03139660029514f:syslibtest_impl.py +a68b70f233541a7f:includes/os\_win32.py +ab6e54ba9a669f76:includes/string.py +abbcbd92436cc16a:includes/shutil.py +b5a965302cded36b:includes/sys.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3b259b4059f8668:includes/viperlib.py +c70684ccfe48a5a9:main.py +c9f4be41ca1cc2b4:includes/viperio.py +d152115b49e50218:includes/os\_posix.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py +f6b51804a0ba8ff0:includes/w32\win32file.py diff --git a/Test/SysLibTest/temp/_shared_sym.json b/Test/SysLibTest/temp/_shared_sym.json new file mode 100644 index 0000000..ec0e507 Binary files /dev/null and b/Test/SysLibTest/temp/_shared_sym.json differ diff --git a/Test/SysLibTest/temp/a03139660029514f.doc.json b/Test/SysLibTest/temp/a03139660029514f.doc.json new file mode 100644 index 0000000..7aab3b8 --- /dev/null +++ b/Test/SysLibTest/temp/a03139660029514f.doc.json @@ -0,0 +1 @@ +{"_cleanup": "清理可能残留的测试文件/目录"} \ No newline at end of file diff --git a/Test/SysLibTest/temp/a03139660029514f.pyi b/Test/SysLibTest/temp/a03139660029514f.pyi new file mode 100644 index 0000000..39bb90d --- /dev/null +++ b/Test/SysLibTest/temp/a03139660029514f.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from syslibtest_impl.py +Module: syslibtest_impl +""" + + +import t, c +from stdint import * +from t import CInt, CPtr, CUInt8T, CSizeT +import testcheck +import viperlib +import stdlib +import memhub +import platmacro +import os +import os.path +import os_backend +import shutil +import sys +import subprocess +import string +import w32.fileio +import w32.win32base +import w32.win32process + +TEST_DIR: t.CDefine = "__sylibtest_dir__" +TEST_FILE: t.CDefine = "__sylibtest_file.txt" +TEST_COPY: t.CDefine = "__sylibtest_copy.txt" +TEST_TREE: t.CDefine = "__sylibtest_tree__" +TEST_MOVE: t.CDefine = "__sylibtest_moved.txt" + +def _cleanup() -> t.CInt: pass + +def run() -> CInt: pass diff --git a/Test/SysLibTest/temp/a68b70f233541a7f.pyi b/Test/SysLibTest/temp/a68b70f233541a7f.pyi new file mode 100644 index 0000000..3cb3a02 --- /dev/null +++ b/Test/SysLibTest/temp/a68b70f233541a7f.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from os._win32.py +Module: os._win32 +""" + + +import t, c +from stdint import * +import w32.win32base +import w32.win32file +import memhub +import stdlib + +MAX_PATH_LEN: t.CDefine = 260 +INVALID_FILE_ATTRIBUTES: t.CDefine = 0xFFFFFFFF +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def _get_attrs(path: str) -> ULONG: pass + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd() -> str: pass + +def chdir(path: str) -> int: pass + +def mkdir(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir(path: str) -> int: pass + +def rename(old_path: str, new_path: str) -> int: pass + +def chmod(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system(cmd: str) -> int: pass diff --git a/Test/SysLibTest/temp/ab6e54ba9a669f76.pyi b/Test/SysLibTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/SysLibTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/SysLibTest/temp/abbcbd92436cc16a.pyi b/Test/SysLibTest/temp/abbcbd92436cc16a.pyi new file mode 100644 index 0000000..9a57653 --- /dev/null +++ b/Test/SysLibTest/temp/abbcbd92436cc16a.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from shutil.py +Module: shutil +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import w32.win32base +import w32.win32file +import posix + +COPY_BUF_SIZE: t.CDefine = 65536 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def copyfile(src: str, dst: str) -> int: pass + +def _copyfile_win32(src: str, dst: str) -> int: pass + +def _copyfile_posix(src: str, dst: str) -> int: pass + +def copy(src: str, dst: str) -> int: pass + +def _copy_win32(src: str, dst: str) -> int: pass + +def _copy_posix(src: str, dst: str) -> int: pass + +def rmtree(path: str) -> int: pass + +def _rmtree_win32(path: str) -> int: pass + +def _rmtree_posix(path: str) -> int: pass + +def move(src: str, dst: str) -> int: pass + +def _move_win32(src: str, dst: str) -> int: pass + +def _move_posix(src: str, dst: str) -> int: pass diff --git a/Test/SysLibTest/temp/b5a965302cded36b.pyi b/Test/SysLibTest/temp/b5a965302cded36b.pyi new file mode 100644 index 0000000..bfdb955 --- /dev/null +++ b/Test/SysLibTest/temp/b5a965302cded36b.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from sys.py +Module: sys +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import stdio +import w32.win32base +import w32.win32process +import posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def exit(status: INT) -> VOID | t.CExtern | t.CExport: pass + + +maxsize: t.CDefine = 9223372036854775807 + +def platform() -> str: pass + +def get_pid() -> int: pass + + +_argc: t.CExtern | INT +_argv: t.CExtern | str | t.CPtr + +def _init_argv() -> t.CInt: pass + +def _init_argv_win32() -> t.CInt: pass + +def _init_argv_posix() -> t.CInt: pass + +def argc() -> int: pass + +def argv(index: int) -> str: pass diff --git a/Test/SysLibTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/SysLibTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/SysLibTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/SysLibTest/temp/c3b259b4059f8668.pyi b/Test/SysLibTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/SysLibTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/SysLibTest/temp/c70684ccfe48a5a9.doc.json b/Test/SysLibTest/temp/c70684ccfe48a5a9.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/SysLibTest/temp/c70684ccfe48a5a9.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/SysLibTest/temp/c70684ccfe48a5a9.pyi b/Test/SysLibTest/temp/c70684ccfe48a5a9.pyi new file mode 100644 index 0000000..f6f33f4 --- /dev/null +++ b/Test/SysLibTest/temp/c70684ccfe48a5a9.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CExport +import memhub +import stdlib +import sys +import os +import os._win32 +import os._posix +import os.path +import shutil +import subprocess +import syslibtest_impl + +ARENA_SIZE: t.CDefine = 1048576 + +def main() -> CInt | CExport: pass diff --git a/Test/SysLibTest/temp/c9f4be41ca1cc2b4.pyi b/Test/SysLibTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/SysLibTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/SysLibTest/temp/d152115b49e50218.pyi b/Test/SysLibTest/temp/d152115b49e50218.pyi new file mode 100644 index 0000000..1e7a152 --- /dev/null +++ b/Test/SysLibTest/temp/d152115b49e50218.pyi @@ -0,0 +1,42 @@ +""" +Auto-generated Python stub file from os._posix.py +Module: os._posix +""" + + +import t, c +from stdint import * +import posix +import memhub +import stdlib + +MAX_PATH_LEN: t.CDefine = 4096 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def _stat(path: str) -> posix.Stat | t.CPtr: pass + +def exists(path: str) -> bool: pass + +def isdir(path: str) -> bool: pass + +def isfile(path: str) -> bool: pass + +def getsize(path: str) -> t.CInt64T: pass + +def getcwd_impl() -> str: pass + +def chdir_impl(path: str) -> int: pass + +def mkdir_impl(path: str) -> int: pass + +def remove(path: str) -> int: pass + +def rmdir_impl(path: str) -> int: pass + +def rename_impl(old_path: str, new_path: str) -> int: pass + +def chmod_impl(path: str, mode: int) -> int: pass + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: pass + +def system_impl(cmd: str) -> int: pass diff --git a/Test/SysLibTest/temp/dd3002730623424b.pyi b/Test/SysLibTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/SysLibTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/SysLibTest/temp/ee084e9fc6ee413a.pyi b/Test/SysLibTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/SysLibTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/SysLibTest/temp/f5522571bcce7bcb.pyi b/Test/SysLibTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/SysLibTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/SysLibTest/temp/f6b51804a0ba8ff0.pyi b/Test/SysLibTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/SysLibTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/TestFBProject/App/main.py b/Test/TestFBProject/App/main.py new file mode 100644 index 0000000..e107a17 --- /dev/null +++ b/Test/TestFBProject/App/main.py @@ -0,0 +1,159 @@ +from stdint import * +import w32.win32console +import t, c +import testcheck + + +def add_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a + b + +def mul_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a * b + + +class AppError(Exception): + pass + +class DiskError(AppError): + pass + +class NetworkError(AppError): + pass + +class TimeoutError(NetworkError): + pass + + +def raise_disk(): + raise DiskError("disk read failed") + +def raise_network(): + raise NetworkError("connection refused") + +def raise_timeout(): + raise TimeoutError("operation timed out") + + +def main() -> t.CInt | t.CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("TestFBProject: 异常处理与LLVMIR测试") + + testcheck.section("Test 1: Raise and catch exact custom exception") + try: + raise DiskError("disk error") + except DiskError as e: + print("caught DiskError:", e) + testcheck.ok("Test 1: caught DiskError") + except: + testcheck.fail("Test 1: caught by generic except") + + testcheck.section("Test 2: Catch parent catches child") + try: + raise_disk() + except AppError as e: + print("caught AppError (child DiskError):", e) + testcheck.ok("Test 2: caught AppError (child DiskError)") + except: + testcheck.fail("Test 2: caught by generic except") + + testcheck.section("Test 3: Catch parent catches nested child") + try: + raise_timeout() + except AppError as e: + print("caught AppError (grandchild TimeoutError):", e) + testcheck.ok("Test 3: caught AppError (grandchild TimeoutError)") + except: + testcheck.fail("Test 3: caught by generic except") + + testcheck.section("Test 4: Catch mid-level parent") + try: + raise_timeout() + except NetworkError as e: + print("caught NetworkError (child TimeoutError):", e) + testcheck.ok("Test 4: caught NetworkError (child TimeoutError)") + except: + testcheck.fail("Test 4: caught by generic except") + + testcheck.section("Test 5: Exact catch before parent") + try: + raise_network() + except NetworkError as e: + print("caught NetworkError:", e) + testcheck.ok("Test 5: caught NetworkError") + except AppError as e: + testcheck.fail("Test 5: caught AppError (should be NetworkError first)") + except: + testcheck.fail("Test 5: caught by generic except") + + testcheck.section("Test 6: Parent catch when exact not present") + try: + raise_network() + except DiskError as e: + testcheck.fail("Test 6: caught DiskError (wrong)") + except AppError as e: + print("caught AppError (no NetworkError handler):", e) + testcheck.ok("Test 6: caught AppError (no NetworkError handler)") + except: + testcheck.fail("Test 6: caught by generic except") + + testcheck.section("Test 7: Generic except fallback") + try: + raise_disk() + except RuntimeError as e: + testcheck.fail("Test 7: caught RuntimeError (wrong)") + except: + testcheck.ok("Test 7: caught by generic except") + + testcheck.section("Test 8: Cross-function with inheritance") + try: + raise_timeout() + except NetworkError as e: + print("caught NetworkError (cross-func TimeoutError):", e) + testcheck.ok("Test 8: caught NetworkError (cross-func TimeoutError)") + except: + testcheck.fail("Test 8: caught by generic except") + + testcheck.section("Test 9: Built-in exception still works") + try: + raise ValueError("test") + except ValueError as e: + print("caught ValueError:", e) + testcheck.ok("Test 9: caught ValueError") + except: + testcheck.fail("Test 9: caught by generic except") + + testcheck.section("Test 10: Mixed built-in and custom") + try: + raise RuntimeError("test") + except DiskError as e: + testcheck.fail("Test 10: caught DiskError (wrong)") + except RuntimeError as e: + print("caught RuntimeError:", e) + testcheck.ok("Test 10: caught RuntimeError") + except: + testcheck.fail("Test 10: caught by generic except") + + testcheck.section("Test LLVMIR: inline LLVM IR add") + x: t.CInt = 10 + y: t.CInt = 20 + r: t.CInt = c.LLVMIR(f"add i32 {c.LInp(x)}, {c.LInp(y)}", t.CInt) + print("LLVMIR add result:", r) + testcheck.check(r == 30, "LLVMIR add (10+20=30)", "LLVMIR add result mismatch") + + testcheck.section("Test CDefine function macro") + s: t.CInt = add_inline(3, 4) + print("add_inline(3, 4) =", s) + testcheck.check(s == 7, "add_inline(3, 4) = 7", "add_inline result mismatch") + m: t.CInt = mul_inline(5, 6) + print("mul_inline(5, 6) =", m) + testcheck.check(m == 30, "mul_inline(5, 6) = 30", "mul_inline result mismatch") + + testcheck.section("Test LLVMIR with LOut") + out_val: t.CInt = 0 + c.LLVMIR(f"{c.LOut(out_val)} = add i32 {c.LInp(x)}, {c.LInp(y)}", t.CInt) + print("LLVMIR LOut result:", out_val) + testcheck.check(out_val == 30, "LLVMIR LOut (10+20=30)", "LLVMIR LOut result mismatch") + + return testcheck.end() diff --git a/Test/TestFBProject/output/4fb8940ff5ab1022.deps.json b/Test/TestFBProject/output/4fb8940ff5ab1022.deps.json new file mode 100644 index 0000000..f02d7ee --- /dev/null +++ b/Test/TestFBProject/output/4fb8940ff5ab1022.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/TestFBProject/project.json b/Test/TestFBProject/project.json new file mode 100644 index 0000000..431a89e --- /dev/null +++ b/Test/TestFBProject/project.json @@ -0,0 +1,36 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TestFBProject", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": [ + "-Wl,--allow-multiple-definition", + "-Wl,--unresolved-symbols=ignore-in-object-files", + "-lucrt", + "-lkernel32", + "-luser32", + "-latomic" + ], + "output": "TestFBProject.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} \ No newline at end of file diff --git a/Test/TestFBProject/temp/0035c95a18d4f8e8.pyi b/Test/TestFBProject/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/TestFBProject/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/TestFBProject/temp/067c78e9f121dce3.pyi b/Test/TestFBProject/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/TestFBProject/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/TestFBProject/temp/06f53cc594b4ac6c.pyi b/Test/TestFBProject/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/TestFBProject/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestFBProject/temp/4fb8940ff5ab1022.doc.json b/Test/TestFBProject/temp/4fb8940ff5ab1022.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestFBProject/temp/4fb8940ff5ab1022.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestFBProject/temp/4fb8940ff5ab1022.pyi b/Test/TestFBProject/temp/4fb8940ff5ab1022.pyi new file mode 100644 index 0000000..1f24963 --- /dev/null +++ b/Test/TestFBProject/temp/4fb8940ff5ab1022.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +import testcheck + +def add_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a + b + +def mul_inline(a: t.CInt, b: t.CInt) -> t.CDefine: + return a * b + + +class AppError(Exception): + pass +class DiskError(AppError): + pass +class NetworkError(AppError): + pass +class TimeoutError(NetworkError): + pass + +def raise_disk() -> t.CInt: pass + +def raise_network() -> t.CInt: pass + +def raise_timeout() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/TestFBProject/temp/6446627d4f07a1b5.pyi b/Test/TestFBProject/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/TestFBProject/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/TestFBProject/temp/6f62fe05c5ea1ceb.pyi b/Test/TestFBProject/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/TestFBProject/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/TestFBProject/temp/72e2d5ccb7cedcf1.pyi b/Test/TestFBProject/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/TestFBProject/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/TestFBProject/temp/7e529fe7a078cfef.pyi b/Test/TestFBProject/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/TestFBProject/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/TestFBProject/temp/_phase1_manifest.json b/Test/TestFBProject/temp/_phase1_manifest.json new file mode 100644 index 0000000..23941b1 --- /dev/null +++ b/Test/TestFBProject/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestFBProject\\App\\main.py": {"sha1": "4fb8940ff5ab1022", "mtime": 1782110087.2169445, "size": 5059}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/TestFBProject/temp/_sha1_map.txt b/Test/TestFBProject/temp/_sha1_map.txt new file mode 100644 index 0000000..e397858 --- /dev/null +++ b/Test/TestFBProject/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +4fb8940ff5ab1022:main.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/TestFBProject/temp/bbdf3bbd4c3bc28c.pyi b/Test/TestFBProject/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/TestFBProject/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestFBProject/temp/dd3002730623424b.pyi b/Test/TestFBProject/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/TestFBProject/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/TestFBProject/temp/f5522571bcce7bcb.pyi b/Test/TestFBProject/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/TestFBProject/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/TestFBProject/temp/f6b51804a0ba8ff0.pyi b/Test/TestFBProject/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/TestFBProject/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/TestProject/App/BINASCII.py b/Test/TestProject/App/BINASCII.py new file mode 100644 index 0000000..f58b7c1 --- /dev/null +++ b/Test/TestProject/App/BINASCII.py @@ -0,0 +1,47 @@ +import stdio +import stdlib +import string +import binascii +import t, c + + + + +def main222(): + binascii.init_binascii() + raw: str = "HelloWorld!" + raw_length: t.CInt = 11 + + stdio.printf("你好\n") + + hex_out: str = binascii.b2a_hex(raw, raw_length) + if hex_out: + stdio.printf("Hex: %s\n", hex_out) + free(hex_out) + + crc_val: t.CUnsignedInt = binascii.crc32(raw, raw_length, 0) + stdio.printf("CRC32: %u\n", crc_val) + + b64_out: str = binascii.b2a_base64(raw, raw_length) + if b64_out: + stdio.printf("Base64: %s\n", b64_out) + free(b64_out) + + uu_out: str = binascii.b2a_uu(raw, raw_length) + if uu_out: + stdio.printf("UU: %s\n", uu_out) + free(uu_out) + + hqx_in: t.CArray[t.CChar, 4] = ['A', 'B', 'C', '\0'] + hqx_crc: t.CUnsignedShort = 0 + hqx_out: str + unhqx: str + hqx_out, hqx_crc = binascii.b2a_hqx(hqx_in, 3) + unhqx, hqx_crc = binascii.a2b_hqx(hqx_out) + stdio.printf("HQX Test: ABC -> %s -> %s (CRC:%u)\n", hqx_out, unhqx, hqx_crc) + free(hqx_out); free(unhqx) + + adler_val: t.CUnsignedInt = binascii.adler32(raw, raw_length, 1) + stdio.printf("Adler32: %u\n", adler_val) + + stdio.printf("Bye\n") diff --git a/Test/TestProject/App/HASHLIB.py b/Test/TestProject/App/HASHLIB.py new file mode 100644 index 0000000..b84258c --- /dev/null +++ b/Test/TestProject/App/HASHLIB.py @@ -0,0 +1,80 @@ +import stdio +import stdlib +import string +import stdint +import t, c +import base64 +import hashlib + + + + + + + + +def default_value_test_function(a: t.CInt, b: t.CInt = 1): + print(a, b) + + + + +# ============================================== +# 工具函数：打印十六进制哈希串 +# ============================================== +def print_hex(buf: t.CUInt8T | t.CPtr, length: t.CInt): + for i in range(length): + stdio.printf("%02x", buf[i]) + stdio.printf("\n") + +# ============================================== +# 主函数测试入口 +# ============================================== +def main1() -> t.CInt: + plain: str = "hello123" + print("原文", plain) + + # ---------- Base64 测试 ---------- + b64_out: t.CArray[t.CChar, 256] + b64_dec: t.CArray[t.CUInt8T, 256] + base64.b64encode(plain, b64_out) + print("Base64 编码:", b64_out) + dlen: t.CSizeT = base64.b64decode(b64_out, b64_dec) + b64_dec[dlen] = 0 + print("Base64 解码:", b64_dec) + + # ---------- MD5 测试 ---------- + md5_buf: t.CArray[t.CUInt8T, hashlib.MD5_DIGEST_LEN] + mctx = hashlib.md5() + mctx.update(plain) + mctx.final(md5_buf) + print("MD5: ") + print_hex(md5_buf, hashlib.MD5_DIGEST_LEN) + + # ---------- SHA1 测试 ---------- + sha1_buf: t.CArray[t.CUInt8T, hashlib.SHA1_DIGEST_LEN] + s1ctx = hashlib.sha1() + s1ctx.update(plain) + s1ctx.final(sha1_buf) + print("SHA1: ") + print_hex(sha1_buf, hashlib.SHA1_DIGEST_LEN) + + # ---------- SHA256 测试 ---------- + sha256_buf: t.CArray[t.CUInt8T, hashlib.SHA256_DIGEST_LEN] + s2ctx = hashlib.sha256() + s2ctx.update(plain) + s2ctx.final(sha256_buf) + print("SHA256: ") + print_hex(sha256_buf, hashlib.SHA256_DIGEST_LEN) + + # ---------- SHA512 测试 ---------- + sha512_buf: t.CArray[t.CUInt8T, hashlib.SHA512_DIGEST_LEN] + s5ctx = hashlib.sha512() + s5ctx.update(plain) + s5ctx.final(sha512_buf) + print("SHA512: ") + print_hex(sha512_buf, hashlib.SHA512_DIGEST_LEN) + + default_value_test_function(b=2, a=1) + + return 0 diff --git a/Test/TestProject/App/config.py b/Test/TestProject/App/config.py new file mode 100644 index 0000000..5ead5bc --- /dev/null +++ b/Test/TestProject/App/config.py @@ -0,0 +1,24 @@ +import stdint +import t, c + + +A: t.CChar | t.CPtr = "你好" + +class BIT_TEST(t.Object): + a: t.CInt | t.Bit(1) + b: t.CInt | t.Bit(1) + c: t.CInt | t.Bit(1) + d: t.CInt | t.Bit(5) + +class OOP_TEST(t.Object): + a: t.CInt + def __init__(self): + self.a = 0 + def __call__(self) -> stdint.UINT: + return 123 + +class ENUM_TEST(t.CEnum): + A: t.State = 0 + B: t.State + C: t.State + Len: t.State diff --git a/Test/TestProject/App/main.py b/Test/TestProject/App/main.py new file mode 100644 index 0000000..12043d7 --- /dev/null +++ b/Test/TestProject/App/main.py @@ -0,0 +1,220 @@ +import stdio +import stdint +import zc.config as config +import zc.test as test +import zc.logic_test as logic +import zc.class_test as class_test +import testcheck +import test_numpy + + +def main() -> stdint.INT: + testcheck.begin("Hello World!") + testcheck.section("Cross-module macros") + stdio.printf(" POSMAX: %d\n", config.POSMAX) + stdio.printf(" MATH_SCALE: %d\n", config.MATH_SCALE) + stdio.printf(" THRESHOLD: %d\n", config.THRESHOLD) + stdio.printf(" SHIFT_AMOUNT: %d\n", config.SHIFT_AMOUNT) + stdio.printf(" MASK_VALUE: 0x%x\n", config.MASK_VALUE) + testcheck.ok("config macros accessible") + + testcheck.section("While Loop Test") + test.AboutSizeTWhileTest(10) + testcheck.ok("While Loop Test (10)") + + testcheck.section("Math Basic Test (20, 6)") + test.MathBasicTest(20, 6) + testcheck.ok("Math Basic Test (20, 6)") + + testcheck.section("Math Advanced Test (value=15)") + test.MathAdvancedTest(15) + testcheck.ok("Math Advanced Test (value=15)") + + testcheck.section("While with Elif-Else Test (-2 to 12)") + test.WhileWithElifElseTest(-2, 12) + testcheck.ok("While with Elif-Else Test (-2 to 12)") + + testcheck.section("Nested While with Macros Test") + test.NestedWhileWithMacrosTest(3, 4) + testcheck.ok("Nested While with Macros Test (3, 4)") + + testcheck.section("Complex Condition Test (1, 2, 3)") + test.ComplexConditionTest(1, 2, 3) + testcheck.ok("Complex Condition Test (1, 2, 3)") + + testcheck.section("Retry Loop Test (8)") + test.RetryLoopTest(8) + testcheck.ok("Retry Loop Test (8)") + + testcheck.section("Bitwise Operations Test (0b1101101)") + test.BitwiseOperationsTest(109) + testcheck.ok("Bitwise Operations Test (109)") + + testcheck.section("Compound Assignment Math Test (15, 4)") + test.CompoundAssignmentMathTest(15, 4) + testcheck.ok("Compound Assignment Math Test (15, 4)") + + testcheck.section("Power and Modulo Test (2, 8)") + test.PowerAndModuloTest(2, 8) + testcheck.ok("Power and Modulo Test (2, 8)") + + testcheck.section("Mixed Arithmetic Test (5, 3, 4)") + test.MixedArithmeticTest(5, 3, 4) + testcheck.ok("Mixed Arithmetic Test (5, 3, 4)") + + testcheck.section("Math with While Loop Test (start=1, count=5)") + test.MathWithWhileLoopTest(1, 5) + testcheck.ok("Math with While Loop Test (1, 5)") + + testcheck.section("Complex Nested Math Test (7, 3)") + test.ComplexNestedMathTest(7, 3) + testcheck.ok("Complex Nested Math Test (7, 3)") + + testcheck.section("Conditional Math Chain Test (value=10)") + test.ConditionalMathChainTest(10) + testcheck.ok("Conditional Math Chain Test (value=10)") + + testcheck.section("Bit Manipulation Math Test (0xABCD)") + test.BitManipulationMathTest(0xABCD) + testcheck.ok("Bit Manipulation Math Test (0xABCD)") + + testcheck.section("COMPLEX LOGIC TESTS") + + testcheck.section("For Range Basic Test (stop=5)") + logic.ForRangeBasicTest(5) + testcheck.ok("For Range Basic Test (5)") + + testcheck.section("For Range With Start Stop Step (start=1, stop=10, step=2)") + logic.ForRangeWithStartStop(1, 10, 2) + testcheck.ok("For Range With Start Stop (1, 10, 2)") + + testcheck.section("Nested For Range Test (rows=2, cols=3)") + logic.NestedForRangeTest(2, 3) + testcheck.ok("Nested For Range Test (2, 3)") + + testcheck.section("For With If Break Test (size=8)") + logic.ForWithIfBreakTest(8) + testcheck.ok("For With If Break Test (8)") + + testcheck.section("For With If Continue Test (n=10)") + logic.ForWithIfContinueTest(10) + testcheck.ok("For With If Continue Test (10)") + + testcheck.section("For With Elif Branch Test (value=10)") + logic.ForWithElifBranchTest(10) + testcheck.ok("For With Elif Branch Test (10)") + + testcheck.section("While With If Elif Else Test (iterations=7)") + logic.WhileWithIfElifElseTest(7) + testcheck.ok("While With If Elif Else Test (7)") + + testcheck.section("While With Nested If Test (limit=4)") + logic.WhileWithNestedIfTest(4) + testcheck.ok("While With Nested If Test (4)") + + testcheck.section("While With Match Case Test (value=2)") + logic.WhileWithMatchCaseTest(2) + testcheck.ok("While With Match Case Test (2)") + + testcheck.section("While With Match Case Test (value=99)") + logic.WhileWithMatchCaseTest(99) + testcheck.ok("While With Match Case Test (99)") + + testcheck.section("While With Match Case NoBreak Test (value=3)") + logic.WhileWithMatchCaseNoBreakTest(3) + testcheck.ok("While With Match Case NoBreak Test (3)") + + testcheck.section("Complex For While Match Test (outer=3, inner=3)") + logic.ComplexForWhileMatchTest(3, 3) + testcheck.ok("Complex For While Match Test (3, 3)") + + testcheck.section("For With Case And NoBreak Test (n=5)") + logic.ForWithCaseAndNoBreakTest(5) + testcheck.ok("For With Case And NoBreak Test (5)") + + testcheck.section("While With Break Condition Test (limit=6)") + logic.WhileWithBreakConditionTest(6) + testcheck.ok("While With Break Condition Test (6)") + + testcheck.section("Nested While With Multiple BreakPoints (depth=3, width=4)") + logic.NestedWhileWithMultipleBreakPoints(3, 4) + testcheck.ok("Nested While With Multiple BreakPoints (3, 4)") + + testcheck.section("For With Match Case In Loop Test (iterations=9)") + logic.ForWithMatchCaseInLoopTest(9) + testcheck.ok("For With Match Case In Loop Test (9)") + + testcheck.section("Complex Logic With Elif Chains (a=3, b=5, c=7)") + logic.ComplexLogicWithElifChains(3, 5, 7) + testcheck.ok("Complex Logic With Elif Chains (3, 5, 7)") + + testcheck.section("For Range With Step And Condition (n=12, step=2)") + logic.ForRangeWithStepAndCondition(12, 2) + testcheck.ok("For Range With Step And Condition (12, 2)") + + testcheck.section("Match Case With Guard Conditions (value=1)") + logic.MatchCaseWithGuardConditions(1) + testcheck.ok("Match Case With Guard Conditions (1)") + + testcheck.section("While With Multiple Match Cases (iterations=12)") + logic.WhileWithMultipleMatchCases(12) + testcheck.ok("While With Multiple Match Cases (12)") + + testcheck.section("For While Mix With Break And Continue (n=5)") + logic.ForWhileMixWithBreakAndContinue(5) + testcheck.ok("For While Mix With Break And Continue (5)") + + testcheck.section("Complex Case Matching With Ranges (start=0, end=35)") + logic.ComplexCaseMatchingWithRanges(0, 35) + testcheck.ok("Complex Case Matching With Ranges (0, 35)") + + testcheck.section("While With Elif In Elif Chain (limit=7)") + logic.WhileWithElifInElifChain(7) + testcheck.ok("While With Elif In Elif Chain (7)") + + testcheck.section("For With Case NoBreak Nested (inner_limit=4)") + logic.ForWithCaseNoBreakNested(4) + testcheck.ok("For With Case NoBreak Nested (4)") + + testcheck.section("CLASS TESTS") + class_test.TestBasicDataClass() + testcheck.ok("TestBasicDataClass") + class_test.TestPoint3DClass() + testcheck.ok("TestPoint3DClass") + class_test.TestStudentClass() + testcheck.ok("TestStudentClass") + class_test.TestCounterClass() + testcheck.ok("TestCounterClass") + class_test.TestStackAllocation() + testcheck.ok("TestStackAllocation") + class_test.TestHeapAllocation() + testcheck.ok("TestHeapAllocation") + class_test.TestHeapObjectWithInit() + testcheck.ok("TestHeapObjectWithInit") + class_test.TestMultipleStackObjects() + testcheck.ok("TestMultipleStackObjects") + class_test.TestImplicitInitCall() + testcheck.ok("TestImplicitInitCall") + class_test.TestClassWithBitFields() + testcheck.ok("TestClassWithBitFields") + + testcheck.section("ARRAY TESTS") + class_test.TestIntArray() + testcheck.ok("TestIntArray") + class_test.TestFloatArray() + testcheck.ok("TestFloatArray") + class_test.TestCharArray() + testcheck.ok("TestCharArray") + class_test.TestNestedArray() + testcheck.ok("TestNestedArray") + class_test.TestArrayWithStruct() + testcheck.ok("TestArrayWithStruct") + class_test.TestArrayPointer() + testcheck.ok("TestArrayPointer") + class_test.TestDynamicString() + testcheck.ok("TestDynamicString") + + testcheck.section("numpy Correctness Tests") + test_numpy.test_numpy_correct() + + return testcheck.end() diff --git a/Test/TestProject/App/test_numpy.py b/Test/TestProject/App/test_numpy.py new file mode 100644 index 0000000..91cf870 --- /dev/null +++ b/Test/TestProject/App/test_numpy.py @@ -0,0 +1,1087 @@ +from stdint import * +import numpy +from stdio import printf +from stdlib import calloc, free +import string +import vipermath +import memhub +import t, c + +# ============================================================ +# Constants +# ============================================================ +EPS: t.CDouble = t.CDouble(1e-9) +EPS_LOOSE: t.CDouble = t.CDouble(1e-5) +SIMD_BLOCK: t.CSizeT = 4 + +# ============================================================ +# Global state +# ============================================================ +arena: t.CUInt8T | t.CPtr = None +pool: memhub.MemPool | t.CPtr = None + +test_passed: t.CInt = 0 +test_failed: t.CInt = 0 + +# ============================================================ +# Global check function +# ============================================================ +def check(name: str, condition: t.CInt, detail: str): + global test_passed, test_failed + if condition: + test_passed += 1 + printf(" [PASS] %s\n", name) + else: + test_failed += 1 + printf(" [FAIL] %s -- %s\n", name, detail if detail else "") + + +# ============================================================ +# Approximate comparison (top-level, uses vipermath.fabs) +# ============================================================ +def approx_eq(a: t.CDouble, b: t.CDouble) -> t.CInt: + d: t.CDouble = a - b + if d < t.CDouble(0.0): + d = t.CDouble(0.0) - d + if d < EPS: + return 1 + return 0 + +def approx_loose(a: t.CDouble, b: t.CDouble) -> t.CInt: + d: t.CDouble = a - b + if d < t.CDouble(0.0): + d = t.CDouble(0.0) - d + if d < EPS_LOOSE: + return 1 + return 0 + + +# ============================================================ +# Section headers +# ============================================================ +def section_header(title: str): + printf("\n+-- %s --+\n", title) + +def section_footer(): + printf("+--------------------------------------------+\n") + + +# ============================================================ +# Utility: quick array creation +# ============================================================ +def vec4(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble, v3: t.CDouble) -> numpy.ndarray | t.CPtr: + a: numpy.ndarray | t.CPtr = numpy.zeros(p, 4) + a.data[0] = v0; a.data[1] = v1; a.data[2] = v2; a.data[3] = v3 + return a + +def vec3(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble) -> numpy.ndarray | t.CPtr: + a: numpy.ndarray | t.CPtr = numpy.zeros(p, 3) + a.data[0] = v0; a.data[1] = v1; a.data[2] = v2 + return a + +def vec2(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble) -> numpy.ndarray | t.CPtr: + a: numpy.ndarray | t.CPtr = numpy.zeros(p, 2) + a.data[0] = v0; a.data[1] = v1 + return a + + +# ============================================================ +# 1. Array creation +# ============================================================ +def test_zeros_ones(): + section_header("zeros / ones / full") + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 5) + check("zeros not None", a != None, "zeros returned None") + check("zeros size == 5", a.size == 5, "size mismatch") + check("zeros [0]==0", a.data[0] == t.CDouble(0.0), "not zero") + check("zeros [4]==0", a.data[4] == t.CDouble(0.0), "not zero") + + b: numpy.ndarray | t.CPtr = numpy.ones(pool, 4) + check("ones not None", b != None, "ones returned None") + check("ones [0]==1", b.data[0] == t.CDouble(1.0), "not one") + check("ones [3]==1", b.data[3] == t.CDouble(1.0), "not one") + + c: numpy.ndarray | t.CPtr = numpy.full(pool, 3, t.CDouble(7.5)) + check("full not None", c != None, "full returned None") + check("full value", c.data[0] == t.CDouble(7.5), "value mismatch") + + a.delete() + b.delete() + c.delete() + section_footer() + + +def test_arange_linspace(): + section_header("arange / linspace") + + a: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(0.0), t.CDouble(5.0), t.CDouble(1.0)) + check("arange not None", a != None, "arange returned None") + check("arange size == 5", a.size == 5, "size mismatch") + check("arange [0]==0", approx_eq(a.data[0], t.CDouble(0.0)), "mismatch") + check("arange [4]==4", approx_eq(a.data[4], t.CDouble(4.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = numpy.linspace(pool, t.CDouble(0.0), t.CDouble(1.0), 5) + check("linspace not None", b != None, "linspace returned None") + check("linspace size == 5", b.size == 5, "size mismatch") + check("linspace [0]==0", approx_eq(b.data[0], t.CDouble(0.0)), "mismatch") + check("linspace [4]==1", approx_eq(b.data[4], t.CDouble(1.0)), "mismatch") + + a.delete() + b.delete() + section_footer() + + +def test_eye_diag(): + section_header("eye / diag") + + e: numpy.ndarray | t.CPtr = numpy.eye(pool, 3) + check("eye not None", e != None, "eye returned None") + check("eye ndim == 2", e.ndim == 2, "ndim mismatch") + check("eye [0,0]==1", e.at2d(0, 0) == t.CDouble(1.0), "mismatch") + check("eye [0,1]==0", e.at2d(0, 1) == t.CDouble(0.0), "mismatch") + check("eye [1,1]==1", e.at2d(1, 1) == t.CDouble(1.0), "mismatch") + + vals: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + d: numpy.ndarray | t.CPtr = numpy.diag(pool, vals) + check("diag not None", d != None, "diag returned None") + check("diag [0,0]==1", d.at2d(0, 0) == t.CDouble(1.0), "mismatch") + check("diag [1,1]==2", d.at2d(1, 1) == t.CDouble(2.0), "mismatch") + check("diag [2,2]==3", d.at2d(2, 2) == t.CDouble(3.0), "mismatch") + + e.delete() + vals.delete() + d.delete() + section_footer() + + +# ============================================================ +# 2. Methods +# ============================================================ +def test_sum_mean_min_max(): + section_header("sum / mean / min / max / argmax / argmin") + + a: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(1.0), t.CDouble(6.0), t.CDouble(1.0)) + check("sum == 15", approx_eq(a.sum(), t.CDouble(15.0)), "mismatch") + check("mean == 3", approx_eq(a.mean(), t.CDouble(3.0)), "mismatch") + check("min == 1", approx_eq(a.min(), t.CDouble(1.0)), "mismatch") + check("max == 5", approx_eq(a.max(), t.CDouble(5.0)), "mismatch") + check("argmax == 4", a.argmax() == 4, "mismatch") + check("argmin == 0", a.argmin() == 0, "mismatch") + + a.delete() + section_footer() + + +def test_fill_copy(): + section_header("fill / copy") + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 3) + a.fill(t.CDouble(9.0)) + check("fill [0]==9", a.data[0] == t.CDouble(9.0), "mismatch") + check("fill [2]==9", a.data[2] == t.CDouble(9.0), "mismatch") + + b: numpy.ndarray | t.CPtr = a.copy() + check("copy not None", b != None, "copy returned None") + check("copy [0]==9", b.data[0] == t.CDouble(9.0), "mismatch") + + b.data[0] = t.CDouble(0.0) + check("copy is independent", a.data[0] == t.CDouble(9.0), "copy not independent") + + a.delete() + b.delete() + section_footer() + + +# ============================================================ +# 3. Operator overloading +# ============================================================ +def test_add_sub_mul_div(): + section_header("__add__ / __sub__ / __mul__ / __truediv__") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(4.0), t.CDouble(5.0), t.CDouble(6.0)) + + c_add: numpy.ndarray | t.CPtr = a + b + check("a+b not None", c_add != None, "add returned None") + check("a+b [0]==5", c_add.data[0] == t.CDouble(5.0), "mismatch") + check("a+b [2]==9", c_add.data[2] == t.CDouble(9.0), "mismatch") + + c_sub: numpy.ndarray | t.CPtr = a - b + check("a-b [0]==-3", c_sub.data[0] == t.CDouble(-3.0), "mismatch") + + c_mul: numpy.ndarray | t.CPtr = a * b + check("a*b [1]==10", c_mul.data[1] == t.CDouble(10.0), "mismatch") + + c_div: numpy.ndarray | t.CPtr = b / a + check("b/a [0]==4", c_div.data[0] == t.CDouble(4.0), "mismatch") + check("b/a [2]==2", c_div.data[2] == t.CDouble(2.0), "mismatch") + + a.delete() + b.delete() + c_add.delete() + c_sub.delete() + c_mul.delete() + c_div.delete() + section_footer() + + +def test_neg(): + section_header("__neg__") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(-2.0), t.CDouble(3.0)) + neg_a: numpy.ndarray | t.CPtr = -a + check("neg not None", neg_a != None, "neg returned None") + check("neg [0]==-1", neg_a.data[0] == t.CDouble(-1.0), "mismatch") + check("neg [1]==2", neg_a.data[1] == t.CDouble(2.0), "mismatch") + + a.delete() + neg_a.delete() + section_footer() + + +def test_len(): + section_header("__len__") + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 10) + check("len == 10", len(a) == 10, "mismatch") + + a.delete() + section_footer() + + +# ============================================================ +# 4. Scalar operations +# ============================================================ +def test_scalar_ops(): + section_header("add_scalar / mul_scalar / sub_scalar / div_scalar") + + a: numpy.ndarray | t.CPtr = numpy.ones(pool, 3) + + r1: numpy.ndarray | t.CPtr = numpy.add_scalar(a, t.CDouble(2.0)) + check("add_scalar [0]==3", r1.data[0] == t.CDouble(3.0), "mismatch") + + r2: numpy.ndarray | t.CPtr = numpy.mul_scalar(a, t.CDouble(5.0)) + check("mul_scalar [0]==5", r2.data[0] == t.CDouble(5.0), "mismatch") + + r3: numpy.ndarray | t.CPtr = numpy.sub_scalar(a, t.CDouble(0.5)) + check("sub_scalar [0]==0.5", r3.data[0] == t.CDouble(0.5), "mismatch") + + r4: numpy.ndarray | t.CPtr = numpy.div_scalar(a, t.CDouble(2.0)) + check("div_scalar [0]==0.5", r4.data[0] == t.CDouble(0.5), "mismatch") + + a.delete() + r1.delete() + r2.delete() + r3.delete() + r4.delete() + section_footer() + + +# ============================================================ +# 5. Math functions +# ============================================================ +def test_math_funcs(): + section_header("np_abs / np_sqrt / np_pow") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(4.0), t.CDouble(9.0)) + + r_sqrt: numpy.ndarray | t.CPtr = numpy.np_sqrt(a) + check("sqrt [0]==1", approx_eq(r_sqrt.data[0], t.CDouble(1.0)), "mismatch") + check("sqrt [1]==2", approx_eq(r_sqrt.data[1], t.CDouble(2.0)), "mismatch") + check("sqrt [2]==3", approx_eq(r_sqrt.data[2], t.CDouble(3.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(-1.0), t.CDouble(-4.0), t.CDouble(9.0)) + r_abs: numpy.ndarray | t.CPtr = numpy.np_abs(b) + check("abs [0]==1", approx_eq(r_abs.data[0], t.CDouble(1.0)), "mismatch") + check("abs [1]==4", approx_eq(r_abs.data[1], t.CDouble(4.0)), "mismatch") + + c: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + r_pow: numpy.ndarray | t.CPtr = numpy.np_pow(c, t.CDouble(2.0)) + check("pow [0]==1", approx_eq(r_pow.data[0], t.CDouble(1.0)), "mismatch") + check("pow [1]==4", approx_eq(r_pow.data[1], t.CDouble(4.0)), "mismatch") + check("pow [2]==9", approx_eq(r_pow.data[2], t.CDouble(9.0)), "mismatch") + + a.delete() + b.delete() + c.delete() + r_sqrt.delete() + r_abs.delete() + r_pow.delete() + section_footer() + + +# ============================================================ +# 6. Matrix operations +# ============================================================ +def test_matmul(): + section_header("matmul") + + # [1,2;3,4] x [5,6;7,8] = [19,22;43,50] + a: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 2) + a.set2d(0, 0, t.CDouble(1.0)); a.set2d(0, 1, t.CDouble(2.0)) + a.set2d(1, 0, t.CDouble(3.0)); a.set2d(1, 1, t.CDouble(4.0)) + + b: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 2) + b.set2d(0, 0, t.CDouble(5.0)); b.set2d(0, 1, t.CDouble(6.0)) + b.set2d(1, 0, t.CDouble(7.0)); b.set2d(1, 1, t.CDouble(8.0)) + + c: numpy.ndarray | t.CPtr = numpy.matmul(a, b) + check("matmul not None", c != None, "matmul returned None") + check("matmul [0,0]==19", c.at2d(0, 0) == t.CDouble(19.0), "mismatch") + check("matmul [0,1]==22", c.at2d(0, 1) == t.CDouble(22.0), "mismatch") + check("matmul [1,0]==43", c.at2d(1, 0) == t.CDouble(43.0), "mismatch") + check("matmul [1,1]==50", c.at2d(1, 1) == t.CDouble(50.0), "mismatch") + + a.delete() + b.delete() + c.delete() + section_footer() + + +def test_dot(): + section_header("dot / dot_product") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(4.0), t.CDouble(5.0), t.CDouble(6.0)) + + d: t.CDouble = a.dot(b) + check("dot == 32", approx_eq(d, t.CDouble(32.0)), "mismatch") + + d2: t.CDouble = numpy.dot_product(a, b) + check("dot_product == 32", approx_eq(d2, t.CDouble(32.0)), "mismatch") + + a.delete() + b.delete() + section_footer() + + +def test_transpose(): + section_header("T (transpose)") + + a: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 3) + a.set2d(0, 0, t.CDouble(1.0)); a.set2d(0, 1, t.CDouble(2.0)); a.set2d(0, 2, t.CDouble(3.0)) + a.set2d(1, 0, t.CDouble(4.0)); a.set2d(1, 1, t.CDouble(5.0)); a.set2d(1, 2, t.CDouble(6.0)) + + at: numpy.ndarray | t.CPtr = a.T() + check("T not None", at != None, "T returned None") + check("T shape[0]==3", at.shape[0] == 3, "shape mismatch") + check("T shape[1]==2", at.shape[1] == 2, "shape mismatch") + check("T [0,0]==1", at.at2d(0, 0) == t.CDouble(1.0), "mismatch") + check("T [0,1]==4", at.at2d(0, 1) == t.CDouble(4.0), "mismatch") + check("T [1,0]==2", at.at2d(1, 0) == t.CDouble(2.0), "mismatch") + + a.delete() + at.delete() + section_footer() + + +# ============================================================ +# 7. Utility +# ============================================================ +def test_var_std_norm(): + section_header("var / std / norm") + + # [2,4,4,4,5,5,7,9] mean=5, var=4, std=2 + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 8) + a.data[0] = t.CDouble(2.0); a.data[1] = t.CDouble(4.0) + a.data[2] = t.CDouble(4.0); a.data[3] = t.CDouble(4.0) + a.data[4] = t.CDouble(5.0); a.data[5] = t.CDouble(5.0) + a.data[6] = t.CDouble(7.0); a.data[7] = t.CDouble(9.0) + + v: t.CDouble = numpy.var(a) + printf(" var = %.6f (expect 4.0)\n", v) + check("var ~= 4", approx_eq(v, t.CDouble(4.0)), "mismatch") + + s: t.CDouble = numpy.std(a) + printf(" std = %.6f (expect 2.0)\n", s) + check("std ~= 2", approx_eq(s, t.CDouble(2.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(3.0), t.CDouble(4.0)) + n: t.CDouble = numpy.norm(b) + printf(" norm = %.6f (expect 5.0)\n", n) + check("norm ~= 5", approx_eq(n, t.CDouble(5.0)), "mismatch") + + a.delete() + b.delete() + section_footer() + + +def test_clip_concatenate(): + section_header("clip / concatenate") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(5.0), t.CDouble(10.0)) + r: numpy.ndarray | t.CPtr = numpy.clip(a, t.CDouble(2.0), t.CDouble(8.0)) + check("clip [0]==2", r.data[0] == t.CDouble(2.0), "mismatch") + check("clip [1]==5", r.data[1] == t.CDouble(5.0), "mismatch") + check("clip [2]==8", r.data[2] == t.CDouble(8.0), "mismatch") + + b: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(1.0), t.CDouble(2.0)) + c: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(3.0), t.CDouble(4.0)) + cat: numpy.ndarray | t.CPtr = numpy.concatenate(b, c) + check("concatenate not None", cat != None, "concatenate returned None") + check("concatenate size == 4", cat.size == 4, "size mismatch") + check("concatenate [0]==1", cat.data[0] == t.CDouble(1.0), "mismatch") + check("concatenate [3]==4", cat.data[3] == t.CDouble(4.0), "mismatch") + + a.delete() + r.delete() + b.delete() + c.delete() + cat.delete() + section_footer() + + +def test_sort_reverse(): + section_header("sort_arr / reverse") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(3.0), t.CDouble(1.0), t.CDouble(2.0)) + s: numpy.ndarray | t.CPtr = numpy.sort_arr(a) + check("sort not None", s != None, "sort returned None") + check("sort [0]==1", s.data[0] == t.CDouble(1.0), "mismatch") + check("sort [1]==2", s.data[1] == t.CDouble(2.0), "mismatch") + check("sort [2]==3", s.data[2] == t.CDouble(3.0), "mismatch") + + r: numpy.ndarray | t.CPtr = numpy.reverse(a) + check("reverse not None", r != None, "reverse returned None") + check("reverse [0]==2", r.data[0] == t.CDouble(2.0), "mismatch") + check("reverse [1]==1", r.data[1] == t.CDouble(1.0), "mismatch") + check("reverse [2]==3", r.data[2] == t.CDouble(3.0), "mismatch") + + a.delete() + s.delete() + r.delete() + section_footer() + + +def test_print_arr(): + section_header("print_arr") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + printf(" 1D array: ") + a.print_arr() + + b: numpy.ndarray | t.CPtr = numpy.eye(pool, 3) + printf(" 2D eye(3):\n") + b.print_arr() + + a.delete() + b.delete() + section_footer() + + +# ============================================================ +# 8. Additional operator overloading +# ============================================================ +def test_floordiv_mod(): + section_header("__floordiv__ / __mod__") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(7.0), t.CDouble(10.0), t.CDouble(15.0)) + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(2.0), t.CDouble(3.0), t.CDouble(4.0)) + + c_fd: numpy.ndarray | t.CPtr = a // b + check("7//2==3", c_fd.data[0] == t.CDouble(3.0), "mismatch") + check("10//3==3", c_fd.data[1] == t.CDouble(3.0), "mismatch") + check("15//4==3", c_fd.data[2] == t.CDouble(3.0), "mismatch") + + c_mod: numpy.ndarray | t.CPtr = a % b + check("7%%2==1", c_mod.data[0] == t.CDouble(1.0), "mismatch") + check("10%%3==1", c_mod.data[1] == t.CDouble(1.0), "mismatch") + check("15%%4==3", c_mod.data[2] == t.CDouble(3.0), "mismatch") + + a.delete() + b.delete() + c_fd.delete() + c_mod.delete() + section_footer() + + +# ============================================================ +# 9. Additional math functions +# ============================================================ +def test_additional_math(): + section_header("np_log10 / np_log2 / np_floor / np_ceil / np_round / np_sign") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(10.0), t.CDouble(100.0), t.CDouble(1000.0)) + r_log10: numpy.ndarray | t.CPtr = numpy.np_log10(a) + check("log10(10)~=1", approx_loose(r_log10.data[0], t.CDouble(1.0)), "mismatch") + check("log10(100)~=2", approx_loose(r_log10.data[1], t.CDouble(2.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(2.0), t.CDouble(4.0), t.CDouble(8.0)) + r_log2: numpy.ndarray | t.CPtr = numpy.np_log2(b) + check("log2(2)~=1", approx_loose(r_log2.data[0], t.CDouble(1.0)), "mismatch") + check("log2(4)~=2", approx_loose(r_log2.data[1], t.CDouble(2.0)), "mismatch") + + c: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.5), t.CDouble(2.7), t.CDouble(-1.3)) + r_floor: numpy.ndarray | t.CPtr = numpy.np_floor(c) + check("floor(1.5)==1", r_floor.data[0] == t.CDouble(1.0), "mismatch") + check("floor(2.7)==2", r_floor.data[1] == t.CDouble(2.0), "mismatch") + check("floor(-1.3)==-2", r_floor.data[2] == t.CDouble(-2.0), "mismatch") + + r_ceil: numpy.ndarray | t.CPtr = numpy.np_ceil(c) + check("ceil(1.5)==2", r_ceil.data[0] == t.CDouble(2.0), "mismatch") + check("ceil(2.7)==3", r_ceil.data[1] == t.CDouble(3.0), "mismatch") + check("ceil(-1.3)==-1", r_ceil.data[2] == t.CDouble(-1.0), "mismatch") + + d: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.4), t.CDouble(2.5), t.CDouble(3.6)) + r_round: numpy.ndarray | t.CPtr = numpy.np_round(d) + check("round(1.4)==1", r_round.data[0] == t.CDouble(1.0), "mismatch") + check("round(3.6)==4", r_round.data[2] == t.CDouble(4.0), "mismatch") + + e: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + e.data[0] = t.CDouble(-3.0); e.data[1] = t.CDouble(0.0); e.data[2] = t.CDouble(5.0); e.data[3] = t.CDouble(-0.1) + r_sign: numpy.ndarray | t.CPtr = numpy.np_sign(e) + check("sign(-3)==-1", r_sign.data[0] == t.CDouble(-1.0), "mismatch") + check("sign(0)==0", r_sign.data[1] == t.CDouble(0.0), "mismatch") + check("sign(5)==1", r_sign.data[2] == t.CDouble(1.0), "mismatch") + + a.delete() + b.delete() + c.delete() + d.delete() + e.delete() + r_log10.delete() + r_log2.delete() + r_floor.delete() + r_ceil.delete() + r_round.delete() + r_sign.delete() + section_footer() + + +def test_trig_hyperbolic(): + section_header("np_tanh / np_sinh / np_cosh / np_arcsin / np_arccos / np_arctan") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(1.0), t.CDouble(-1.0)) + + r_tanh: numpy.ndarray | t.CPtr = numpy.np_tanh(a) + check("tanh(0)~=0", approx_loose(r_tanh.data[0], t.CDouble(0.0)), "mismatch") + + r_sinh: numpy.ndarray | t.CPtr = numpy.np_sinh(a) + check("sinh(0)~=0", approx_loose(r_sinh.data[0], t.CDouble(0.0)), "mismatch") + + r_cosh: numpy.ndarray | t.CPtr = numpy.np_cosh(a) + check("cosh(0)~=1", approx_loose(r_cosh.data[0], t.CDouble(1.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(0.0), t.CDouble(0.5)) + r_asin: numpy.ndarray | t.CPtr = numpy.np_arcsin(b) + check("arcsin(0)~=0", approx_loose(r_asin.data[0], t.CDouble(0.0)), "mismatch") + + r_acos: numpy.ndarray | t.CPtr = numpy.np_arccos(b) + check("arccos(0)~=pi/2", approx_loose(r_acos.data[0], t.CDouble(1.5707963)), "mismatch") + + r_atan: numpy.ndarray | t.CPtr = numpy.np_arctan(b) + check("arctan(0)~=0", approx_loose(r_atan.data[0], t.CDouble(0.0)), "mismatch") + + a.delete() + b.delete() + r_tanh.delete() + r_sinh.delete() + r_cosh.delete() + r_asin.delete() + r_acos.delete() + r_atan.delete() + section_footer() + + +def test_degrees_radians(): + section_header("np_degrees / np_radians") + + a: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(0.0), t.CDouble(1.5707963267948966)) + + r_deg: numpy.ndarray | t.CPtr = numpy.np_degrees(a) + check("deg(0)~=0", approx_loose(r_deg.data[0], t.CDouble(0.0)), "mismatch") + check("deg(pi/2)~=90", approx_loose(r_deg.data[1], t.CDouble(90.0)), "mismatch") + + b: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(0.0), t.CDouble(90.0)) + r_rad: numpy.ndarray | t.CPtr = numpy.np_radians(b) + check("rad(0)~=0", approx_loose(r_rad.data[0], t.CDouble(0.0)), "mismatch") + check("rad(90)~=pi/2", approx_loose(r_rad.data[1], t.CDouble(1.5707963)), "mismatch") + + a.delete() + b.delete() + r_deg.delete() + r_rad.delete() + section_footer() + + +# ============================================================ +# 10. Additional array operations +# ============================================================ +def test_cumsum_diff(): + section_header("cumsum / diff") + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0); a.data[1] = t.CDouble(2.0); a.data[2] = t.CDouble(3.0); a.data[3] = t.CDouble(4.0) + + r_cs: numpy.ndarray | t.CPtr = numpy.cumsum(a) + check("cumsum not None", r_cs != None, "cumsum returned None") + check("cumsum[0]==1", r_cs.data[0] == t.CDouble(1.0), "mismatch") + check("cumsum[1]==3", r_cs.data[1] == t.CDouble(3.0), "mismatch") + check("cumsum[2]==6", r_cs.data[2] == t.CDouble(6.0), "mismatch") + check("cumsum[3]==10", r_cs.data[3] == t.CDouble(10.0), "mismatch") + + r_diff: numpy.ndarray | t.CPtr = numpy.diff(a) + check("diff not None", r_diff != None, "diff returned None") + check("diff size==3", r_diff.size == 3, "size mismatch") + check("diff[0]==1", r_diff.data[0] == t.CDouble(1.0), "mismatch") + check("diff[2]==1", r_diff.data[2] == t.CDouble(1.0), "mismatch") + + a.delete() + r_cs.delete() + r_diff.delete() + section_footer() + + +def test_max_min_where(): + section_header("np_maximum / np_minimum / np_where") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(5.0), t.CDouble(3.0)) + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(4.0), t.CDouble(2.0), t.CDouble(6.0)) + + r_max: numpy.ndarray | t.CPtr = numpy.np_maximum(a, b) + check("maximum[0]==4", r_max.data[0] == t.CDouble(4.0), "mismatch") + check("maximum[1]==5", r_max.data[1] == t.CDouble(5.0), "mismatch") + check("maximum[2]==6", r_max.data[2] == t.CDouble(6.0), "mismatch") + + r_min: numpy.ndarray | t.CPtr = numpy.np_minimum(a, b) + check("minimum[0]==1", r_min.data[0] == t.CDouble(1.0), "mismatch") + check("minimum[1]==2", r_min.data[1] == t.CDouble(2.0), "mismatch") + check("minimum[2]==3", r_min.data[2] == t.CDouble(3.0), "mismatch") + + # where: cond > 0 ? a : b + cond: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(1.0), t.CDouble(0.0)) + r_where: numpy.ndarray | t.CPtr = numpy.np_where(cond, a, b) + check("where[0]==4", r_where.data[0] == t.CDouble(4.0), "mismatch") + check("where[1]==5", r_where.data[1] == t.CDouble(5.0), "mismatch") + check("where[2]==6", r_where.data[2] == t.CDouble(6.0), "mismatch") + + a.delete() + b.delete() + cond.delete() + r_max.delete() + r_min.delete() + r_where.delete() + section_footer() + + +def test_flatten_trace(): + section_header("flatten / trace") + + a: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 3) + a.set2d(0, 0, t.CDouble(1.0)); a.set2d(0, 1, t.CDouble(2.0)); a.set2d(0, 2, t.CDouble(3.0)) + a.set2d(1, 0, t.CDouble(4.0)); a.set2d(1, 1, t.CDouble(5.0)); a.set2d(1, 2, t.CDouble(6.0)) + + r_flat: numpy.ndarray | t.CPtr = numpy.flatten(a) + check("flatten not None", r_flat != None, "flatten returned None") + check("flatten ndim==1", r_flat.ndim == 1, "ndim mismatch") + check("flatten size==6", r_flat.size == 6, "size mismatch") + check("flatten[0]==1", r_flat.data[0] == t.CDouble(1.0), "mismatch") + + # trace of [[1,2],[3,4]] = 1+4 = 5 + b: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 2) + b.set2d(0, 0, t.CDouble(1.0)); b.set2d(0, 1, t.CDouble(2.0)) + b.set2d(1, 0, t.CDouble(3.0)); b.set2d(1, 1, t.CDouble(4.0)) + tr: t.CDouble = numpy.trace(b) + check("trace == 5", tr == t.CDouble(5.0), "mismatch") + + a.delete() + r_flat.delete() + b.delete() + section_footer() + + +def test_outer(): + section_header("outer") + + a: numpy.ndarray | t.CPtr = vec2(pool, t.CDouble(1.0), t.CDouble(2.0)) + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(3.0), t.CDouble(4.0), t.CDouble(5.0)) + + r: numpy.ndarray | t.CPtr = numpy.outer(a, b) + check("outer not None", r != None, "outer returned None") + check("outer ndim==2", r.ndim == 2, "ndim mismatch") + check("outer[0,0]==3", r.at2d(0, 0) == t.CDouble(3.0), "mismatch") + check("outer[0,1]==4", r.at2d(0, 1) == t.CDouble(4.0), "mismatch") + check("outer[1,0]==6", r.at2d(1, 0) == t.CDouble(6.0), "mismatch") + check("outer[1,2]==10", r.at2d(1, 2) == t.CDouble(10.0), "mismatch") + + a.delete() + b.delete() + r.delete() + section_footer() + + +# ============================================================ +# 11. Boolean / comparison +# ============================================================ +def test_bool_comparison(): + section_header("np_all / np_any / np_count_nonzero / np_equal / np_less / np_greater") + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + a.data[0] = t.CDouble(1.0); a.data[1] = t.CDouble(0.0); a.data[2] = t.CDouble(3.0); a.data[3] = t.CDouble(0.0) + check("count_nonzero==2", numpy.np_count_nonzero(a) == 2, "mismatch") + check("any==1", numpy.np_any(a) == 1, "mismatch") + + b: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + check("all(1,2,3)==1", numpy.np_all(b) == 1, "mismatch") + + c: numpy.ndarray | t.CPtr = numpy.zeros(pool, 3) + check("all(0,0,0)==0", numpy.np_all(c) == 0, "mismatch") + check("any(0,0,0)==0", numpy.np_any(c) == 0, "mismatch") + + d: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + e: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(0.0), t.CDouble(4.0)) + + r_eq: numpy.ndarray | t.CPtr = numpy.np_equal(d, e) + check("equal[0]==1", r_eq.data[0] == t.CDouble(1.0), "mismatch") + check("equal[1]==0", r_eq.data[1] == t.CDouble(0.0), "mismatch") + + r_lt: numpy.ndarray | t.CPtr = numpy.np_less(d, e) + check("less[2]==1", r_lt.data[2] == t.CDouble(1.0), "mismatch") + check("less[0]==0", r_lt.data[0] == t.CDouble(0.0), "mismatch") + + r_gt: numpy.ndarray | t.CPtr = numpy.np_greater(d, e) + check("greater[1]==1", r_gt.data[1] == t.CDouble(1.0), "mismatch") + + a.delete() + b.delete() + c.delete() + d.delete() + e.delete() + r_eq.delete() + r_lt.delete() + r_gt.delete() + section_footer() + + +# ============================================================ +# 12. Linear algebra +# ============================================================ +def test_linalg(): + section_header("det2x2 / inv2x2 / cross3 / prod / median / percentile") + + # det2x2: [[1,2],[3,4]] -> 1*4-2*3 = -2 + a: numpy.ndarray | t.CPtr = numpy.empty2d(pool, 2, 2) + a.set2d(0, 0, t.CDouble(1.0)); a.set2d(0, 1, t.CDouble(2.0)) + a.set2d(1, 0, t.CDouble(3.0)); a.set2d(1, 1, t.CDouble(4.0)) + d: t.CDouble = numpy.det2x2(a) + check("det2x2==-2", d == t.CDouble(-2.0), "mismatch") + + # inv2x2 of [[1,2],[3,4]] + inv: numpy.ndarray | t.CPtr = numpy.inv2x2(a) + check("inv2x2 not None", inv != None, "inv2x2 returned None") + check("inv[0,0]==-2", approx_eq(inv.at2d(0, 0), t.CDouble(-2.0)), "mismatch") + check("inv[0,1]==1", approx_eq(inv.at2d(0, 1), t.CDouble(1.0)), "mismatch") + + # cross3: [1,0,0] x [0,1,0] = [0,0,1] + x: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(0.0), t.CDouble(0.0)) + y: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(1.0), t.CDouble(0.0)) + cr: numpy.ndarray | t.CPtr = numpy.cross3(x, y) + check("cross not None", cr != None, "cross returned None") + check("cross[2]==1", cr.data[2] == t.CDouble(1.0), "mismatch") + + # prod + p: numpy.ndarray | t.CPtr = numpy.zeros(pool, 4) + p.data[0] = t.CDouble(1.0); p.data[1] = t.CDouble(2.0); p.data[2] = t.CDouble(3.0); p.data[3] = t.CDouble(4.0) + check("prod==24", numpy.prod(p) == t.CDouble(24.0), "mismatch") + + # median + m: numpy.ndarray | t.CPtr = numpy.zeros(pool, 5) + m.data[0] = t.CDouble(3.0); m.data[1] = t.CDouble(1.0); m.data[2] = t.CDouble(2.0) + m.data[3] = t.CDouble(5.0); m.data[4] = t.CDouble(4.0) + check("median==3", numpy.median(m) == t.CDouble(3.0), "mismatch") + + # percentile + check("p50==3", approx_eq(numpy.percentile(m, t.CDouble(50.0)), t.CDouble(3.0)), "mismatch") + + a.delete() + inv.delete() + x.delete() + y.delete() + cr.delete() + p.delete() + m.delete() + section_footer() + + +# ============================================================ +# 13. Additional creation / constants +# ============================================================ +def test_additional_creation(): + section_header("full2d / zeros_like / ones_like / arange1 / constants") + + f: numpy.ndarray | t.CPtr = numpy.full2d(pool, 2, 3, t.CDouble(7.0)) + check("full2d not None", f != None, "full2d returned None") + check("full2d ndim==2", f.ndim == 2, "ndim mismatch") + check("full2d[0,0]==7", f.at2d(0, 0) == t.CDouble(7.0), "mismatch") + check("full2d[1,2]==7", f.at2d(1, 2) == t.CDouble(7.0), "mismatch") + + a: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(1.0), t.CDouble(2.0), t.CDouble(3.0)) + zl: numpy.ndarray | t.CPtr = numpy.zeros_like(a) + check("zeros_like not None", zl != None, "returned None") + check("zeros_like size==3", zl.size == 3, "size mismatch") + check("zeros_like[0]==0", zl.data[0] == t.CDouble(0.0), "mismatch") + + ol: numpy.ndarray | t.CPtr = numpy.ones_like(a) + check("ones_like not None", ol != None, "returned None") + check("ones_like[0]==1", ol.data[0] == t.CDouble(1.0), "mismatch") + + ar: numpy.ndarray | t.CPtr = numpy.arange1(pool, t.CDouble(4.0)) + check("arange1 not None", ar != None, "returned None") + check("arange1 size==4", ar.size == 4, "size mismatch") + check("arange1[0]==0", ar.data[0] == t.CDouble(0.0), "mismatch") + check("arange1[3]==3", ar.data[3] == t.CDouble(3.0), "mismatch") + + # Constants + check("pi ~= 3.14", approx_loose(numpy.pi, t.CDouble(3.14159265)), "mismatch") + check("e ~= 2.71", approx_loose(numpy.e, t.CDouble(2.71828182)), "mismatch") + + f.delete() + a.delete() + zl.delete() + ol.delete() + ar.delete() + section_footer() + + +# ============================================================ +# 14. Interpolation +# ============================================================ +def test_interp(): + section_header("np_interp") + + xp: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(1.0), t.CDouble(2.0)) + fp: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(10.0), t.CDouble(20.0)) + + v: t.CDouble = numpy.np_interp(t.CDouble(0.5), xp, fp) + check("interp(0.5)==5", approx_eq(v, t.CDouble(5.0)), "mismatch") + + v2: t.CDouble = numpy.np_interp(t.CDouble(1.5), xp, fp) + check("interp(1.5)==15", approx_eq(v2, t.CDouble(15.0)), "mismatch") + + v3: t.CDouble = numpy.np_interp(t.CDouble(-1.0), xp, fp) + check("interp(-1)==0", approx_eq(v3, t.CDouble(0.0)), "mismatch") + + v4: t.CDouble = numpy.np_interp(t.CDouble(3.0), xp, fp) + check("interp(3)==20", approx_eq(v4, t.CDouble(20.0)), "mismatch") + + xp.delete() + fp.delete() + section_footer() + + +# ============================================================ +# 15. Edge / boundary / exception tests +# ============================================================ +def test_numpy_edge(): + section_header("edge: size=0 / size=1 / non-SIMD-block / negative / div-zero") + + # size=0: pool.alloc(0) returns None, so zeros(0) returns None (expected) + a0: numpy.ndarray | t.CPtr = numpy.zeros(pool, 0) + check("zeros(0) returns None", a0 == None, "expected None for size=0") + + # size=1 + a1: numpy.ndarray | t.CPtr = numpy.zeros(pool, 1) + check("zeros(1) not None", a1 != None, "returned None") + check("zeros(1) size==1", a1.size == 1, "size mismatch") + check("zeros(1) [0]==0", a1.data[0] == t.CDouble(0.0), "mismatch") + a1.delete() + + # non-SIMD-block sizes (1,2,3,5,6,7) + a5: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(0.0), t.CDouble(5.0), t.CDouble(1.0)) + check("arange(5) size==5", a5.size == 5, "size mismatch") + check("arange(5) sum==10", approx_eq(a5.sum(), t.CDouble(10.0)), "mismatch") + a5.delete() + + a6: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(0.0), t.CDouble(6.0), t.CDouble(1.0)) + check("arange(6) sum==15", approx_eq(a6.sum(), t.CDouble(15.0)), "mismatch") + a6.delete() + + a7: numpy.ndarray | t.CPtr = numpy.arange(pool, t.CDouble(0.0), t.CDouble(7.0), t.CDouble(1.0)) + check("arange(7) sum==21", approx_eq(a7.sum(), t.CDouble(21.0)), "mismatch") + a7.delete() + + # negative values in sqrt + neg_arr: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(-1.0), t.CDouble(0.0), t.CDouble(4.0)) + r_sqrt_neg: numpy.ndarray | t.CPtr = numpy.np_sqrt(neg_arr) + check("sqrt(neg) not None", r_sqrt_neg != None, "returned None") + # sqrt(4)=2, sqrt(0)=0; sqrt(-1) is NaN (implementation-defined) + check("sqrt(4)==2", approx_eq(r_sqrt_neg.data[2], t.CDouble(2.0)), "mismatch") + neg_arr.delete() + r_sqrt_neg.delete() + + # div-by-zero: b/a where a has zeros + za: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(0.0), t.CDouble(2.0), t.CDouble(0.0)) + zb: numpy.ndarray | t.CPtr = vec3(pool, t.CDouble(6.0), t.CDouble(4.0), t.CDouble(8.0)) + r_divz: numpy.ndarray | t.CPtr = za / zb + check("0/6==0", r_divz.data[0] == t.CDouble(0.0), "mismatch") + check("2/4==0.5", r_divz.data[1] == t.CDouble(0.5), "mismatch") + za.delete() + zb.delete() + r_divz.delete() + + # large array (non-aligned, non-SIMD-block) + big: numpy.ndarray | t.CPtr = numpy.zeros(pool, 1000) + check("zeros(1000) not None", big != None, "returned None") + check("zeros(1000) size==1000", big.size == 1000, "size mismatch") + big.delete() + + section_footer() + + +# ============================================================ +# 16. Performance benchmark +# ============================================================ +def bench_numpy_perf(): + section_header("Benchmark: numpy ops") + + N: t.CSizeT = 10000 + ITERS: t.CInt = 5 + + a: numpy.ndarray | t.CPtr = numpy.zeros(pool, N) + b: numpy.ndarray | t.CPtr = numpy.zeros(pool, N) + i: t.CSizeT = 0 + while i < N: + a.data[i] = t.CDouble(1.0) + b.data[i] = t.CDouble(2.0) + i += 1 + + # add benchmark + t0: t.CDouble = c.Timer() + j: t.CInt = 0 + while j < ITERS: + r: numpy.ndarray | t.CPtr = a + b + r.delete() + j += 1 + t1: t.CDouble = c.Timer() + printf(" ADD: %.1f ms\n", (t1 - t0) * t.CDouble(1000.0)) + + # mul benchmark + t0 = c.Timer() + j = 0 + while j < ITERS: + r2: numpy.ndarray | t.CPtr = a * b + r2.delete() + j += 1 + t1 = c.Timer() + printf(" MUL: %.1f ms\n", (t1 - t0) * t.CDouble(1000.0)) + + # sqrt benchmark + t0 = c.Timer() + j = 0 + while j < ITERS: + r3: numpy.ndarray | t.CPtr = numpy.np_sqrt(a) + r3.delete() + j += 1 + t1 = c.Timer() + printf(" SQRT: %.1f ms\n", (t1 - t0) * t.CDouble(1000.0)) + + a.delete() + b.delete() + section_footer() + + +# ============================================================ +# Entry points +# ============================================================ +def test_numpy_correct() -> t.CInt: + global arena, pool, test_passed, test_failed + + arena = calloc(2 * 1024 * 1024, 1) + pool = memhub.MemPool(arena, 2 * 1024 * 1024) + + test_passed = 0 + test_failed = 0 + + printf("==============================================\n") + printf(" numpy Correctness Tests\n") + printf("==============================================\n") + + test_zeros_ones() + test_arange_linspace() + test_eye_diag() + test_sum_mean_min_max() + test_fill_copy() + test_add_sub_mul_div() + test_neg() + test_len() + test_scalar_ops() + test_math_funcs() + test_matmul() + test_dot() + test_transpose() + test_var_std_norm() + test_clip_concatenate() + test_sort_reverse() + test_print_arr() + test_floordiv_mod() + test_additional_math() + test_trig_hyperbolic() + test_degrees_radians() + test_cumsum_diff() + test_max_min_where() + test_flatten_trace() + test_outer() + test_bool_comparison() + test_linalg() + test_additional_creation() + test_interp() + + printf("\n==============================================\n") + printf(" Correctness: %d passed, %d failed\n", test_passed, test_failed) + printf("==============================================\n") + + return test_failed + + +def test_numpy_edge_main() -> t.CInt: + global arena, pool, test_passed, test_failed + + arena = calloc(2 * 1024 * 1024, 1) + pool = memhub.MemPool(arena, 2 * 1024 * 1024) + + test_passed = 0 + test_failed = 0 + + printf("==============================================\n") + printf(" numpy Edge / Boundary Tests\n") + printf("==============================================\n") + + test_numpy_edge() + + printf("\n==============================================\n") + printf(" Edge: %d passed, %d failed\n", test_passed, test_failed) + printf("==============================================\n") + + return test_failed + + +def bench_numpy_main() -> t.CInt: + global arena, pool + + arena = calloc(8 * 1024 * 1024, 1) + pool = memhub.MemPool(arena, 8 * 1024 * 1024) + + printf("==============================================\n") + printf(" numpy Performance Benchmark\n") + printf("==============================================\n") + + bench_numpy_perf() + + printf("==============================================\n") + return 0 + + +# Backward-compatible entry +def test_numpy_main() -> t.CInt: + global arena, pool + + arena = calloc(2 * 1024 * 1024, 1) + pool = memhub.MemPool(arena, 2 * 1024 * 1024) + + failed: t.CInt = test_numpy_correct() + if failed == 0: + failed = test_numpy_edge_main() + if failed == 0: + bench_numpy_main() + return failed diff --git a/Test/TestProject/App/test_zlib.py b/Test/TestProject/App/test_zlib.py new file mode 100644 index 0000000..705aa32 --- /dev/null +++ b/Test/TestProject/App/test_zlib.py @@ -0,0 +1,883 @@ +from stdint import * +import zlib.pyzlib as pyzlib +import zlib.zhuff as zhuff +import zlib.zdef as zdef +from stdio import printf +from string import strlen, memcmp, memcpy, memset +from stdlib import malloc, free, realloc +import stdlib +import string +import memhub +import t, c + +pool: t.CPtr = None + +test_passed: t.CInt = 0 +test_failed: t.CInt = 0 +def check(name: str, condition: t.CInt, detail: str): + global test_passed, test_failed + if condition: + test_passed += 1 + printf(" [PASS] %s\n", name) + else: + test_failed += 1 + printf(" [FAIL] %s -- %s\n", name, detail if detail else "") + +def print_hex_test(data: BYTEPTR, length: t.CSizeT, max_show: t.CSizeT): + if not data or length == 0: + printf("(empty)\n") + return + show: t.CSizeT = max_show if (length > max_show) else length + i: t.CSizeT + for i in range(show): + printf("%02X ", data[i]) + if length > max_show: + printf("... (%zu bytes total)", length) + printf("\n") + +def print_separator(): + printf(" -------------------------------------------\n") + +def section_header(title: str): + printf("\n+-- %s --+\n", title) + +def section_footer(): + printf("+--------------------------------------------+\n") + +# ============================================================ + +def test_version(): + section_header("Version Info") + + ver: str = pyzlib.ZLIB_VERSION + printf(" ZLIB_VERSION (compile-time) : %s\n", ver) + check("ZLIB_VERSION not None", ver != None, "version string is None") + check("ZLIB_VERSION not empty", strlen(ver) > 0, "version string is empty") + + runtime_ver: str = pyzlib.runtime_version() + printf(" ZLIB_RUNTIME_VERSION : %s\n", runtime_ver) + check("runtime version not None", runtime_ver != None, "runtime version is None") + check("runtime version not empty", strlen(runtime_ver) > 0, "runtime version is empty") + + section_footer() + + +def test_compress_decompress(): + section_header("compress / decompress") + + inp: str = "Hello, World! This is a test of zlib compression and decompression." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("compress returns non-None", compressed != None, "compress returned None") + check("compress output size > 0", out_length > 0, "compressed size is 0") + check("compress output produced", out_length > 0, "compressed size is 0") + + printf(" Compressed (%zu bytes): ", out_length) + print_hex_test(compressed, out_length, 32) + printf(" Compression ratio: %.1f%%\n", t.CDouble(out_length) / input_length * 100) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(pool, compressed, out_length, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress returns non-None", decompressed != None, "decompress returned None") + check("decompress output size matches input", dec_length == input_length, "size mismatch") + check("decompress output matches input", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + + printf(" Decompressed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(decompressed)) + + free(compressed) + free(decompressed) + section_footer() + + +def test_compress_levels(): + section_header("Compression Levels") + + inp: str = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA" + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + + c0: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_NO_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 0 compress OK", c0 != None, "level 0 returned None") + printf(" Level 0 (Z_NO_COMPRESSION): %zu bytes\n", out_length) + length0: t.CSizeT = out_length + free(c0) + + c1: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_BEST_SPEED, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 1 compress OK", c1 != None, "level 1 returned None") + printf(" Level 1 (Z_BEST_SPEED) : %zu bytes\n", out_length) + free(c1) + + c6: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level -1 compress OK", c6 != None, "default level returned None") + printf(" Level -1 (DEFAULT) : %zu bytes\n", out_length) + free(c6) + + c9: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_BEST_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 9 compress OK", c9 != None, "level 9 returned None") + printf(" Level 9 (Z_BEST_COMPRESS) : %zu bytes\n", out_length) + length9: t.CSizeT = out_length + free(c9) + + check("level 9 smaller than level 0", length9 < length0, "level 9 not smaller than level 0") + + section_footer() + + +def test_compressobj(): + section_header("compressobj (incremental)") + + part1: str = "Hello, " + part2: str = "World!" + printf(" Part 1: \"%s\"\n", part1) + printf(" Part 2: \"%s\"\n", part2) + + out_length: t.CSizeT = 0 + + s: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pool, pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + check("compressobj created", s != None, "compressobj is None") + print_separator() + + c1: BYTEPTR = s.compress(BYTEPTR(part1), strlen(part1), c.Addr(out_length)) + check("compress part1 OK", c1 != None, "compress part1 failed") + printf(" Compressed part1: %zu bytes -> ", out_length) + print_hex_test(c1, out_length, 16) + total: t.CSizeT = out_length + all: BYTEPTR = None + if out_length > 0 and c1: + all = BYTEPTR(malloc(total)) + if all: memcpy(all, c1, out_length) + free(c1) + + c2: BYTEPTR = s.compress(BYTEPTR(part2), + strlen(part2), c.Addr(out_length)) + check("compress part2 OK", c2 != None, "compress part2 failed") + printf(" Compressed part2: %zu bytes -> ", out_length) + print_hex_test(c2, out_length, 16) + if out_length > 0 and c2: + new_all: BYTEPTR = BYTEPTR(realloc(all, total + out_length)) + if new_all: + all = new_all + memcpy(all + total, c2, out_length) + total += out_length + free(c2) + + c3: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush (Z_FINISH) OK", c3 != None, "flush failed") + printf(" Flush result : %zu bytes -> ", out_length) + print_hex_test(c3, out_length, 16) + if out_length > 0 and c3: + new_all: BYTEPTR = realloc(all, total + out_length) # 隐式转换 + if new_all: + all = new_all + memcpy(all + total, c3, out_length) + total += out_length + free(c3) + + print_separator() + + if all: + dec_length: t.CSizeT = 0 + dec: BYTEPTR = pyzlib.decompress(pool, all, total, pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress incremental OK", dec != None, "decompress returned None") + check("decompress incremental size", dec != None and dec_length == strlen(part1) + strlen(part2), "size mismatch") + check("decompress incremental content", + dec != None and memcmp(dec, "Hello, World!", dec_length) == 0, "content mismatch") + if dec: + printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + free(all) + + s.delete() # del s + section_footer() + +def test_decompressobj(): + section_header("decompressobj") + + inp: str = "Test data for decompress object." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pool, pyzlib.MAX_WBITS, None, 0) + check("decompressobj created", d != None, "decompressobj is None") + check("eof is false initially", d.eof == 0, "eof should be 0") + print_separator() + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(compressed, out_length, + 0, c.Addr(dec_length)) + check("decompress OK", dec != None, "decompress returned None") + check("decompress size", dec != None and dec_length == input_length, "size mismatch") + check("decompress content", dec != None and memcmp(dec, inp, input_length) == 0, "content mismatch") + check("eof is true after stream end", d.eof == 1, "eof should be 1") + + if dec: + printf(" Decompressed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(dec)) + printf(" eof = %d\n", d.eof) + + free(dec) + free(compressed) + d.delete() + section_footer() + +def test_decompressobj_max_lengthgth(): + section_header("decompressobj max_lengthgth") + + inp: str = "This is a longer string for testing max_lengthgth parameter in decompress." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pool, pyzlib.MAX_WBITS, None, 0) + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(compressed, out_length, + 10, c.Addr(dec_length)) + check("max_lengthgth decompress OK", dec != None, "decompress returned None") + check("max_lengthgth limits output", dec_length <= 10, "output exceeded max_lengthgth") + if dec: + printf(" First chunk (max_lengthgth=10): \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + + tail_length: t.CSizeT = 0 + tail: BYTEPTR = d.unconsumed_tail(c.Addr(tail_length)) + check("unconsumed_tail exists", tail != None or tail_length == 0, "no unconsumed_tail") + printf(" unconsumed_tail: %zu bytes remaining\n", tail_length) + printf(" eof = %d\n", d.eof) + + if tail_length > 0 or not d.eof: + flushed: BYTEPTR = d.flush(pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("flush after max_lengthgth OK", flushed != None, "flush returned None") + if flushed: printf(" Flushed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(flushed)) + free(flushed) + + printf(" eof after flush = %d\n", d.eof) + + d.delete() + free(compressed) + section_footer() + + +def test_decompressobj_unused_data(): + section_header("decompressobj unused_data") + + inp: str = "Hello" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + total_length: t.CSizeT = out_length + 5 + combined: BYTEPTR = BYTEPTR(malloc(total_length)) + memcpy(combined, compressed, out_length) + memcpy(combined + out_length, "EXTRA", 5) + free(compressed) + printf(" Combined (compressed + \"EXTRA\"): %zu bytes\n", total_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pool, pyzlib.MAX_WBITS, None, 0) + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(combined, total_length, + 0, c.Addr(dec_length)) + check("decompress with extra OK", dec != None, "decompress returned None") + check("eof after stream end", d.eof == 1, "eof should be 1") + if dec: + printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + + print_separator() + + unused_length: t.CSizeT = 0 + unused: BYTEPTR = d.unused_data(c.Addr(unused_length)) + check("unused_data exists", unused != None and unused_length > 0, "no unused_data") + check("unused_data is EXTRA", unused_length == 5 and unused != None and memcmp(unused, "EXTRA", 5) == 0, "unused_data mismatch") + printf(" unused_data (%zu bytes): \"%.*s\"\n", unused_length, t.CInt(unused_length), str(unused)) + + d.delete() + free(combined) + section_footer() + + +def test_compress_copy(): + section_header("Compress.copy()") + + inp: str = "Copy test data for compress object" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + c1: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pool, pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + c1.compress(BYTEPTR(inp), input_length, c.Addr(out_length)) + printf(" Original: compressed %zu bytes so far\n", out_length) + + c2: pyzlib.Compress | t.CPtr = c1.copy() + check("compress copy OK", c2 != None, "copy returned None") + printf(" Copied compressobj\n") + + print_separator() + + f1: BYTEPTR = c1.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush original OK", f1 != None, "flush original failed") + length1: t.CSizeT = out_length + printf(" Original flush: %zu bytes\n", length1) + free(f1) + + f2: BYTEPTR = c2.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush copy OK", f2 != None, "flush copy failed") + length2: t.CSizeT = out_length + printf(" Copy flush : %zu bytes\n", length2) + free(f2) + + check("copy produces same output", length1 == length2, "output lengthgths differ") + + c1.delete() + c2.delete() + section_footer() + + +def test_decompress_copy(): + section_header("Decompress.copy()") + + inp: str = "Decompress copy test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d1: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pool, pyzlib.MAX_WBITS, None, 0) + d2: pyzlib.Decompress | t.CPtr = d1.copy() + check("decompress copy OK", d2 != None, "copy returned None") + printf(" Copied decompressobj\n") + + print_separator() + + dec_length1: t.CSizeT = 0 + dec_length2: t.CSizeT = 0 + dec1: BYTEPTR = d1.decompress(compressed, out_length, + 0, c.Addr(dec_length1)) + dec2: BYTEPTR = d2.decompress(compressed, out_length, + 0, c.Addr(dec_length2)) + check("decompress copy output size", dec_length1 == dec_length2, "sizes differ") + check("decompress copy output content", + dec1 and dec2 and memcmp(dec1, dec2, dec_length1) == 0, "content differs") + + if dec1: printf(" Original: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length1), str(dec1), dec_length1) + if dec2: printf(" Copy : \"%.*s\" (%zu bytes)\n", t.CInt(dec_length2), str(dec2), dec_length2) + + free(dec1) + free(dec2) + free(compressed) + d1.delete() + d2.delete() + section_footer() + + +def test_adler32(): + section_header("adler32 checksum") + + checksum: ULONG = pyzlib.zlib_adler32(BYTEPTR("Hello"), 5, 1) + printf(" adler32(\"Hello\") = 0x%08lX (%lu)\n", checksum, checksum) + check("adler32 returns non-zero", checksum != 0, "checksum is 0") + + incremental: ULONG = pyzlib.zlib_adler32(BYTEPTR("Hel"), 3, 1) + printf(" adler32(\"Hel\") = 0x%08lX\n", incremental) + incremental = pyzlib.zlib_adler32(BYTEPTR("lo"), 2, incremental) + printf(" adler32(\"lo\", prev) = 0x%08lX\n", incremental) + check("adler32 incremental matches one-shot", incremental == checksum, "incremental != one-shot") + printf(" Incremental == One-shot: %s\n", incremental == "YES" if checksum else "NO") + + section_footer() + + +def test_crc32(): + section_header("crc32 checksum") + + checksum: ULONG = pyzlib.zlib_crc32(BYTEPTR("Hello"), 5, 0) + printf(" crc32(\"Hello\") = 0x%08lX (%lu)\n", checksum, checksum) + check("crc32 returns non-zero", checksum != 0, "checksum is 0") + + incremental: ULONG = pyzlib.zlib_crc32(BYTEPTR("Hel"), 3, 0) + printf(" crc32(\"Hel\") = 0x%08lX\n", incremental) + incremental = pyzlib.zlib_crc32(BYTEPTR("lo"), 2, incremental) + printf(" crc32(\"lo\", prev) = 0x%08lX\n", incremental) + check("crc32 incremental matches one-shot", incremental == checksum, "incremental != one-shot") + printf(" Incremental == One-shot: %s\n", incremental == "YES" if checksum else "NO") + + section_footer() + + +def test_empty_compress(): + section_header("Empty data compress/decompress") + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(""), 0, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("compress empty OK", compressed != None, "compress empty returned None") + check("compress empty output > 0", out_length > 0, "compressed empty is 0 bytes") + printf(" Empty input compressed: %zu bytes (header only)\n", out_length) + printf(" Hex: ") + print_hex_test(compressed, out_length, 32) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(pool, compressed, out_length, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress empty OK", decompressed != None, "decompress empty returned None") + check("decompress empty size == 0", dec_length == 0, "decompressed size != 0") + printf(" Decompressed back: %zu bytes\n", dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_gzip_format(): + section_header("Gzip format (wbits=31)") + + inp: str = "Gzip format test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\" (%zu bytes)\n", inp, input_length) + + gzip_wbits: t.CInt = pyzlib.MAX_WBITS + 16 + printf(" wbits = MAX_WBITS + 16 = %d\n", gzip_wbits) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, gzip_wbits, c.Addr(out_length)) + check("gzip compress OK", compressed != None, "gzip compress failed") + printf(" Gzip compressed: %zu bytes\n", out_length) + printf(" Header bytes: ") + print_hex_test(compressed, 4 if (out_length > 4) else out_length, 4) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(pool, compressed, out_length, + gzip_wbits, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("gzip decompress OK", decompressed != None, "gzip decompress failed") + check("gzip decompress size", dec_length == input_length, "size mismatch") + check("gzip decompress content", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + if decompressed: printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(decompressed), dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_raw_deflate(): + section_header("Raw deflate (wbits=-15)") + + inp: str = "Raw deflate test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\" (%zu bytes)\n", inp, input_length) + + raw_wbits: t.CInt = -pyzlib.MAX_WBITS + printf(" wbits = -MAX_WBITS = %d\n", raw_wbits) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(pool, BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, raw_wbits, c.Addr(out_length)) + check("raw deflate compress OK", compressed != None, "raw deflate compress failed") + printf(" Raw compressed: %zu bytes\n", out_length) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(pool, compressed, out_length, + raw_wbits, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("raw deflate decompress OK", decompressed != None, "raw deflate decompress failed") + check("raw deflate decompress size", dec_length == input_length, "size mismatch") + check("raw deflate decompress content", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + if decompressed: printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(decompressed), dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_zdict(): + section_header("Preset dictionary (zdict)") + + dict: str = "common words that appear frequently in the data" + inp: str = "common words that appear frequently" + input_length: t.CSizeT = strlen(inp) + printf(" Dictionary: \"%s\" (%zu bytes)\n", dict, strlen(dict)) + printf(" Input : \"%s\" (%zu bytes)\n", inp, input_length) + + out_length: t.CSizeT = 0 + s: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pool, pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, + BYTEPTR(dict), strlen(dict)) + check("compressobj with zdict OK", s != None, "compressobj with zdict failed") + + comp_data: BYTEPTR = s.compress(BYTEPTR(inp), + input_length, c.Addr(out_length)) + check("compress with zdict OK", comp_data != None, "compress with zdict failed") + printf(" Compressed with zdict: %zu bytes\n", out_length) + free(comp_data) + + flush_data: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush with zdict OK", flush_data != None, "flush with zdict failed") + printf(" Flush: %zu bytes\n", out_length) + free(flush_data) + + print_separator() + + total_comp: t.CSizeT = out_length + s.delete() + + s = pyzlib.compressobj(pool, pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + comp_no_dict: BYTEPTR = s.compress(BYTEPTR(inp), + input_length, c.Addr(out_length)) + flush_no_dict: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + no_dict_total: t.CSizeT = 0 + if comp_no_dict: no_dict_total += out_length + if flush_no_dict: no_dict_total += out_length + printf(" Without zdict: ~%zu bytes compressed\n", no_dict_total) + printf(" With zdict : %zu bytes compressed\n", total_comp) + free(comp_no_dict) + free(flush_no_dict) + s.delete() + + section_footer() + + +def test_huffman_tree(): + section_header("Huffman tree construction") + + printf(" --- Fixed Huffman tree (literal/lengthgth) ---\n") + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + + check("fixed lit tree count == 288", lit_tree.count == 288, "count mismatch") + check("fixed lit tree max_bits == 9", lit_tree.max_bits == 9, "max_bits mismatch") + + has_8bit: t.CInt = 0 + has_9bit: t.CInt = 0 + has_7bit: t.CInt = 0 + for i in range(143 + 1): + if lit_tree.codes[i].bits == 8: + has_8bit += 1 + for i in range(143 + 1, 255 + 1): + if lit_tree.codes[i].bits == 9: + has_9bit += 1 + for i in range(255 + 1, 279 + 1): + if lit_tree.codes[i].bits == 7: + has_7bit += 1 + check("0-143 are 8-bit", has_8bit == 144, "wrong count") + check("144-255 are 9-bit", has_9bit == 112, "wrong count") + check("256-279 are 7-bit", has_7bit == 24, "wrong count") + + printf(" 0-143: %d codes with 8 bits\n", has_8bit) + printf(" 144-255: %d codes with 9 bits\n", has_9bit) + printf(" 256-279: %d codes with 7 bits\n", has_7bit) + printf(" 280-287: 8-bit (end-of-block 256 = 7-bit)\n") + printf(" EOB (256): code=0x%X, bits=%d\n", lit_tree.codes[256].code, lit_tree.codes[256].bits) + + printf("\n --- Fixed Huffman tree (distance) ---\n") + dist_tree = zhuff.zhuff_tree() + dist_tree.build_fixed_dist_tree() + + check("fixed dist tree count == 32", dist_tree.count == 32, "count mismatch") + check("fixed dist tree max_bits == 5", dist_tree.max_bits == 5, "max_bits mismatch") + + all_5bit: t.CInt = 1 + for i in range(32): + if dist_tree.codes[i].bits != 5: + all_5bit = 0 + check("all 32 distance codes are 5-bit", all_5bit, "not all 5-bit") + printf(" All 32 distance codes: 5 bits each\n") + + printf("\n --- Dynamic Huffman tree from frequencies ---\n") + freqs: t.CArray[t.CInt, 257] + memset(freqs, 0, freqs.__sizeof__()) + freqs['A'] = 50 + freqs['B'] = 25 + freqs['C'] = 12 + freqs['D'] = 6 + freqs['E'] = 3 + freqs['F'] = 1 + freqs[256] = 1 + + dyn_tree = zhuff.zhuff_tree() + dyn_tree.build_codes(freqs, 257, 15) + + check("dynamic tree count == 257", dyn_tree.count == 257, "count mismatch") + check("dynamic tree max_bits <= 15", dyn_tree.max_bits <= 15, "max_bits overflow") + + total_codes: t.CInt = 0 + max_length_found: t.CInt = 0 + for i in range(257): + if dyn_tree.codes[i].bits > 0: + total_codes += 1 + if dyn_tree.codes[i].bits > max_length_found: + max_length_found = dyn_tree.codes[i].bits + check("dynamic tree has 7 active codes", total_codes == 7, "wrong active count") + check("dynamic tree max code lengthgth found <= 15", max_length_found <= 15, "code lengthgth overflow") + + printf(" Frequencies: A=50, B=25, C=12, D=6, E=3, F=1, EOB=1\n") + printf(" Active codes: %d, max code lengthgth: %d\n", total_codes, max_length_found) + + printf(" Code assignments:\n") + names: t.CArray[str, None] = ["A","B","C","D","E","F"] # 一个都是char* 的数组 + syms: t.CArray[t.CInt, None] = ['A','B','C','D','E','F'] + for i in range(6): + printf(" '%s' (freq=%3d): code=0x%04X, bits=%d\n", + names[i], freqs[syms[i]], + dyn_tree.codes[syms[i]].code, + dyn_tree.codes[syms[i]].bits) + + + + printf(" EOB (freq=1): code=0x%04X, bits=%d\n", + dyn_tree.codes[256].code, dyn_tree.codes[256].bits) + a_bits: t.CInt = dyn_tree.codes['A'].bits + f_bits: t.CInt = dyn_tree.codes['F'].bits + check("high-freq symbol has shorter code", a_bits <= f_bits, "A should be <= F") + printf(" High-freq 'A' (%d bits) <= low-freq 'F' (%d bits): %s\n", + a_bits, f_bits, "YES" if (a_bits <= f_bits) else "NO") + + printf("\n --- Huffman encode/decode roundtrip ---\n") + freqs: t.CArray[t.CInt, 288] + memset(freqs, 0, freqs.__sizeof__()) + freqs['H'] = 10 + freqs['e'] = 8 + freqs['l'] = 20 + freqs['o'] = 8 + freqs[' '] = 5 + freqs['W'] = 3 + freqs['r'] = 5 + freqs['d'] = 3 + freqs['!'] = 2 + freqs[256] = 1 + + enc_tree = zhuff.zhuff_tree() + enc_tree.build_codes(freqs, 257, 15) + + dec_tree = zhuff.zhuff_decode_tree() + dec_tree.build_decode_tree(c.Addr(enc_tree)) + + writer = zdef.zbit_writer() + + symbols: t.CArray[t.CInt, 16] + symbols[0] = 'H'; symbols[1] = 'e'; symbols[2] = 'l'; symbols[3] = 'l' + symbols[4] = 'o'; symbols[5] = ' '; symbols[6] = 'W'; symbols[7] = 'o' + symbols[8] = 'r'; symbols[9] = 'l'; symbols[10] = 'd'; symbols[11] = '!' + symbols[12] = 256 + num_symbols: t.CInt = 13 + + for i in range(num_symbols): + enc_tree.encode_symbol(symbols[i], c.Addr(writer)) + + reader = zdef.zbit_reader() + reader.buf = writer.buf + reader.length = writer.byte_pos + (1 if (writer.bit_pos > 0) else 0) + reader.byte_pos = 0 + reader.bit_pos = 0 + + roundtrip_ok: t.CInt = 1 + for i in range(num_symbols): + decoded: t.CInt = dec_tree.decode_symbol(c.Addr(reader)) + if decoded != symbols[i]: + roundtrip_ok = 0 + printf(" MISMATCH at symbol %d: expected %d, got %d\n", i, symbols[i], decoded) + break + check("encode/decode roundtrip matches", roundtrip_ok, "roundtrip failed") + printf(" Encoded %d symbols into %zu bytes, decoded all correctly\n", + num_symbols, writer.byte_pos + (1 if (writer.bit_pos > 0) else 0)) + + free(writer.buf) + + printf("\n --- Overflow handling (many symbols, limited max_bits) ---\n") + freqs: t.CArray[t.CInt, 257] + for i in range(256): + freqs[i] = 1 + freqs[256] = 1 + + tree = zhuff.zhuff_tree() + tree.build_codes(freqs, 257, 15) + + overflow_ok: t.CInt = 1 + for i in range(257): + if tree.codes[i].bits > 15 or tree.codes[i].bits < 0: + overflow_ok = 0 + break + check("all code lengthgths <= 15 with 257 symbols", overflow_ok, "code lengthgth overflow") + + bl_count: t.CArray[t.CInt, 16] = [0] + for i in range(257): + if tree.codes[i].bits > 0: + bl_count[tree.codes[i].bits] += 1 + printf(" Code lengthgth distribution (257 equal-freq symbols, max_bits=15):\n") + for i in range(1, 15 + 1): + if bl_count[i] > 0: + printf(" %2d bits: %3d codes\n", i, bl_count[i]) + + dt = zhuff.zhuff_decode_tree() + dt.build_decode_tree(c.Addr(tree)) + + w = zdef.zbit_writer() + tree.encode_symbol(0, c.Addr(w)) + tree.encode_symbol(128, c.Addr(w)) + tree.encode_symbol(255, c.Addr(w)) + tree.encode_symbol(256, c.Addr(w)) + + r = zdef.zbit_reader() + r.buf = w.buf + r.length = w.byte_pos + (1 if (w.bit_pos > 0) else 0) + r.byte_pos = 0 + r.bit_pos = 0 + + s0: t.CInt = dt.decode_symbol(c.Addr(r)) + s1: t.CInt = dt.decode_symbol(c.Addr(r)) + s2: t.CInt = dt.decode_symbol(c.Addr(r)) + s3: t.CInt = dt.decode_symbol(c.Addr(r)) + + check("overflow roundtrip: symbol 0", s0 == 0, "decode mismatch") + check("overflow roundtrip: symbol 128", s1 == 128, "decode mismatch") + check("overflow roundtrip: symbol 255", s2 == 255, "decode mismatch") + check("overflow roundtrip: symbol 256 (EOB)", s3 == 256, "decode mismatch") + + free(w.buf) + + + printf("\n --- Kraft inequality check ---\n") + freqs: t.CArray[t.CInt, 257] + memset(freqs, 0, freqs.__sizeof__()) + freqs['X'] = 100 + freqs['Y'] = 50 + freqs['Z'] = 25 + freqs[256] = 1 + + tree = zhuff.zhuff_tree() + tree.build_codes(freqs, 257, 15) + + kraft_sum: t.CDouble = 0.0 + for i in range(257): + if tree.codes[i].bits > 0: + kraft_sum += 1.0 / (t.CDouble(ULONGLONG(1) << tree.codes[i].bits)) + check("Kraft inequality: sum <= 1.0", kraft_sum <= 1.0 + 1e-9, "Kraft violated") + printf(" Kraft sum = %.10f (must be <= 1.0)\n", kraft_sum) + + section_footer() + + +def test_error_handling(): + section_header("Error handling") + + out_length: t.CSizeT = 0 + printf(" Trying to decompress garbage data...\n") + + result: BYTEPTR = pyzlib.decompress(pool, BYTEPTR("garbage"), 7, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(out_length)) + check("decompress garbage returns None", result == None, "should have returned None") + printf(" Error code : %d\n", pyzlib.get_error_code()) + printf(" Error message: \"%s\"\n", pyzlib.get_error()) + check("error message set", strlen(pyzlib.get_error()) > 0, "error message is empty") + check("error code is set", pyzlib.get_error_code() != pyzlib.Z_OK, "error code is Z_OK") + + print_separator() + + pyzlib.clear_error() + check("clear error clears code", pyzlib.get_error_code() == 0, "error code not cleared") + printf(" After clear: code=%d, msg=\"%s\"\n", pyzlib.get_error_code(), pyzlib.get_error()) + + section_footer() + + +def test_constants(): + section_header("Constants") + printf(" Compression Levels:\n") + printf(" Z_NO_COMPRESSION = %d\n", pyzlib.Z_NO_COMPRESSION) + printf(" Z_BEST_SPEED = %d\n", pyzlib.Z_BEST_SPEED) + printf(" Z_BEST_COMPRESSION = %d\n", pyzlib.Z_BEST_COMPRESSION) + printf(" Z_DEFAULT_COMPRESSION = %d\n", pyzlib.Z_DEFAULT_COMPRESSION) + printf(" Methods:\n") + printf(" DEFLATED = %d\n", pyzlib.DEFLATED) + printf(" Flush Modes:\n") + printf(" Z_NO_FLUSH = %d\n", pyzlib.Z_NO_FLUSH) + printf(" Z_PARTIAL_FLUSH = %d\n", pyzlib.Z_PARTIAL_FLUSH) + printf(" Z_SYNC_FLUSH = %d\n", pyzlib.Z_SYNC_FLUSH) + printf(" Z_FULL_FLUSH = %d\n", pyzlib.Z_FULL_FLUSH) + printf(" Z_FINISH = %d\n", pyzlib.Z_FINISH) + printf(" Z_BLOCK = %d\n", pyzlib.Z_BLOCK) + printf(" Z_TREES = %d\n", pyzlib.Z_TREES) + printf(" Strategies:\n") + printf(" Z_DEFAULT_STRATEGY = %d\n", pyzlib.Z_DEFAULT_STRATEGY) + printf(" Z_FILTERED = %d\n", pyzlib.Z_FILTERED) + printf(" Z_HUFFMAN_ONLY = %d\n", pyzlib.Z_HUFFMAN_ONLY) + printf(" Z_RLE = %d\n", pyzlib.Z_RLE) + printf(" Z_FIXED = %d\n", pyzlib.Z_FIXED) + printf(" Return Codes:\n") + printf(" Z_OK = %d\n", pyzlib.Z_OK) + printf(" Z_STREAM_END = %d\n", pyzlib.Z_STREAM_END) + printf(" Z_NEED_DICT = %d\n", pyzlib.Z_NEED_DICT) + printf(" Z_ERRNO = %d\n", pyzlib.Z_ERRNO) + printf(" Z_STREAM_ERROR = %d\n", pyzlib.Z_STREAM_ERROR) + printf(" Z_DATA_ERROR = %d\n", pyzlib.Z_DATA_ERROR) + printf(" Z_MEM_ERROR = %d\n", pyzlib.Z_MEM_ERROR) + printf(" Z_BUF_ERROR = %d\n", pyzlib.Z_BUF_ERROR) + printf(" Z_VERSION_ERROR = %d\n", pyzlib.Z_VERSION_ERROR) + printf(" Window / Buffer:\n") + printf(" MAX_WBITS = %d\n", pyzlib.MAX_WBITS) + printf(" DEF_BUF_SIZE = %d\n", pyzlib.DEF_BUF_SIZE) + printf(" DEF_MEM_LEVEL = %d\n", pyzlib.DEF_MEM_LEVEL) + section_footer() + + +def main123456() -> t.CInt: + printf("==============================================\n") + printf(" Python zlib - C Implementation Tests\n") + printf("==============================================\n") + + test_constants() + test_version() + test_compress_decompress() + test_compress_levels() + test_compressobj() + test_decompressobj() + test_decompressobj_max_lengthgth() + test_decompressobj_unused_data() + test_compress_copy() + test_decompress_copy() + test_adler32() + test_crc32() + test_empty_compress() + test_gzip_format() + test_raw_deflate() + test_zdict() + test_huffman_tree() + test_error_handling() + + printf("\n==============================================\n") + printf(" Results: %d passed, %d failed\n", test_passed, test_failed) + printf("==============================================\n") + + return 1 if test_failed > 0 else 0 diff --git a/Test/TestProject/App/zc/class_test.py b/Test/TestProject/App/zc/class_test.py new file mode 100644 index 0000000..4594f8b --- /dev/null +++ b/Test/TestProject/App/zc/class_test.py @@ -0,0 +1,245 @@ +import stdio +import stdint +import t, c + + +class SimpleData(t.Object): + x: t.CInt + y: t.CInt + + +class Point3D(t.Object): + x: t.CFloat + y: t.CFloat + z: t.CFloat + + +class Student(t.Object): + name: t.CChar | t.CPtr + age: t.CInt + score: t.CInt + + def __init__(self, n: t.CChar | t.CPtr, a: t.CInt, s: t.CInt): + self.name = n + self.age = a + self.score = s + + def get_age(self) -> t.CInt: + return self.age + + def get_score(self) -> t.CInt: + return self.score + + +class Counter(t.Object): + value: t.CInt + + def __init__(self, initial: t.CInt): + self.value = initial + + def increment(self): + self.value += 1 + + def get_value(self) -> t.CInt: + return self.value + + +class StackObj(t.Object): + data: t.CInt + next_ptr: t.CVoid | t.CPtr + + +def TestBasicDataClass(): + stdio.printf("=== Test Basic Data Class ===\n") + p: SimpleData = SimpleData() + p.x = 10 + p.y = 20 + stdio.printf("SimpleData: x=%d, y=%d\n", p.x, p.y) + + +def TestPoint3DClass(): + stdio.printf("=== Test Point3D Class ===\n") + pt: Point3D = Point3D() + pt.x = 1.5 + pt.y = 2.5 + pt.z = 3.5 + stdio.printf("Point3D: x=%.1f, y=%.1f, z=%.1f\n", pt.x, pt.y, pt.z) + + +def TestStudentClass(): + stdio.printf("=== Test Student Class ===\n") + s: Student = Student("Alice", 20, 95) + stdio.printf("Student: name=%s, age=%d, score=%d\n", s.name, s.age, s.score) + stdio.printf(" get_age()=%d, get_score()=%d\n", s.get_age(), s.get_score()) + + +def TestCounterClass(): + stdio.printf("=== Test Counter Class ===\n") + cnt: Counter = Counter(0) + stdio.printf("Initial value: %d\n", cnt.get_value()) + cnt.increment() + cnt.increment() + cnt.increment() + stdio.printf("After 3 increments: %d\n", cnt.get_value()) + + +def TestStackAllocation(): + stdio.printf("=== Test Stack Allocation ===\n") + obj1: SimpleData = SimpleData() + obj1.x = 100 + obj1.y = 200 + stdio.printf("Stack obj1: x=%d, y=%d\n", obj1.x, obj1.y) + + obj2: Point3D = Point3D() + obj2.x = 10.0 + obj2.y = 20.0 + obj2.z = 30.0 + stdio.printf("Stack obj2: x=%.1f, y=%.1f, z=%.1f\n", obj2.x, obj2.y, obj2.z) + + +def TestHeapAllocation(): + stdio.printf("=== Test Heap Allocation ===\n") + ptr: t.CVoid | t.CPtr = c.malloc(1024) + if ptr != 0: + stdio.printf("Allocated 1024 bytes at %p\n", ptr) + c.free(ptr) + stdio.printf("Freed memory\n") + else: + stdio.printf("malloc failed\n") + + +def TestHeapObjectWithInit(): + stdio.printf("=== Test Heap Object with Init ===\n") + mem_size: t.CSizeT = 256 + ptr: t.CVoid | t.CPtr = c.malloc(mem_size) + if ptr != 0: + stdio.printf("Allocated %d bytes\n", mem_size) + c.free(ptr) + else: + stdio.printf("malloc failed\n") + + +def TestMultipleStackObjects(): + stdio.printf("=== Test Multiple Stack Objects ===\n") + objs: SimpleData = SimpleData() + objs.x = 1 + objs.y = 2 + obj2: Point3D = Point3D() + obj2.x = 5.0 + obj2.y = 10.0 + obj2.z = 15.0 + obj3: Counter = Counter(42) + stdio.printf("objs: x=%d, y=%d\n", objs.x, objs.y) + stdio.printf("obj2: x=%.1f, y=%.1f, z=%.1f\n", obj2.x, obj2.y, obj2.z) + stdio.printf("obj3 initial: %d\n", obj3.get_value()) + obj3.increment() + obj3.increment() + stdio.printf("obj3 after 2 increments: %d\n", obj3.get_value()) + + +def TestImplicitInitCall(): + stdio.printf("=== Test Implicit Init Call ===\n") + stu1: Student = Student("Bob", 22, 88) + stu2: Student = Student("Charlie", 19, 77) + stdio.printf("stu1: name=%s, age=%d, score=%d\n", stu1.name, stu1.age, stu1.score) + stdio.printf("stu2: name=%s, age=%d, score=%d\n", stu2.name, stu2.age, stu2.score) + + +def TestClassWithBitFields(): + stdio.printf("=== Test Class with Bit Fields ===\n") + flags: StackObj = StackObj() + flags.data = 0 + flags.next_ptr = 0 + stdio.printf("StackObj: data=%d, next_ptr=%p\n", flags.data, flags.next_ptr) + + +def TestIntArray(): + stdio.printf("=== Test Int Array ===\n") + arr: t.CArray[t.CInt, 5] = t.CArray[t.CInt, 5]() + i: t.CInt = 0 + while i < 5: + arr[i] = i * 10 + i += 1 + i = 0 + while i < 5: + stdio.printf("arr[%d]=%d\n", i, arr[i]) + i += 1 + + +def TestFloatArray(): + stdio.printf("=== Test Float Array ===\n") + farr: t.CArray[t.CFloat, 4] = t.CArray[t.CFloat, 4]() + farr[0] = 1.5 + farr[1] = 2.5 + farr[2] = 3.5 + farr[3] = 4.5 + i: t.CInt = 0 + while i < 4: + stdio.printf("farr[%d]=%.1f\n", i, farr[i]) + i += 1 + + +def TestCharArray(): + stdio.printf("=== Test Char Array (String) ===\n") + str_arr: t.CArray[t.CChar, 32] = t.CArray[t.CChar, 32]() + str_arr[0] = 'H' + str_arr[1] = 'e' + str_arr[2] = 'l' + str_arr[3] = 'l' + str_arr[4] = 'o' + str_arr[5] = '\0' + stdio.printf("String: %s\n", str_arr) + + +def TestNestedArray(): + stdio.printf("=== Test Nested Array (Array of Arrays) ===\n") + row0: t.CArray[t.CInt, 3] = t.CArray[t.CInt, 3]() + row1: t.CArray[t.CInt, 3] = t.CArray[t.CInt, 3]() + row0[0] = 0 + row0[1] = 1 + row0[2] = 2 + row1[0] = 10 + row1[1] = 11 + row1[2] = 12 + i: t.CInt = 0 + while i < 3: + stdio.printf("row0[%d]=%d ", i, row0[i]) + i += 1 + stdio.printf("\n") + i = 0 + while i < 3: + stdio.printf("row1[%d]=%d ", i, row1[i]) + i += 1 + stdio.printf("\n") + + +def TestArrayWithStruct(): + stdio.printf("=== Test Array with Struct ===\n") + points: t.CArray[SimpleData, 3] = t.CArray[SimpleData, 3]() + i: t.CInt = 0 + while i < 3: + points[i].x = i * 10 + points[i].y = i * 100 + i += 1 + i = 0 + while i < 3: + stdio.printf("Point[%d]: x=%d, y=%d\n", i, points[i].x, points[i].y) + i += 1 + + +def TestArrayPointer(): + stdio.printf("=== Test Array Pointer ===\n") + data: t.CArray[t.CInt, 8] = t.CArray[t.CInt, 8]() + data[0] = 100 + data[1] = 200 + data[2] = 300 + ptr: t.CInt | t.CPtr = data + stdio.printf("ptr[0]=%d\n", ptr[0]) + stdio.printf("ptr[1]=%d\n", ptr[1]) + stdio.printf("ptr[2]=%d\n", ptr[2]) + + +def TestDynamicString(): + stdio.printf("=== Test Dynamic String ===\n") + name: str = "TransPyC" + stdio.printf("Hello %s!\n", name) \ No newline at end of file diff --git a/Test/TestProject/App/zc/config.py b/Test/TestProject/App/zc/config.py new file mode 100644 index 0000000..1b4e36f --- /dev/null +++ b/Test/TestProject/App/zc/config.py @@ -0,0 +1,24 @@ +import t, c + + +POSMAX: t.CDefine = 100 +MAX_RETRY: t.CDefine = 5 +BUFFER_SIZE: t.CDefine = 256 +PI: t.CDefine = 3.14159 +EULER: t.CDefine = 2.71828 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +NULL: t.CDefine = 0 + + +MATH_OFFSET: t.CDefine = 10 +MATH_SCALE: t.CDefine = 2 +THRESHOLD: t.CDefine = 50 +LOOP_LIMIT: t.CDefine = 20 +SHIFT_AMOUNT: t.CDefine = 2 +MASK_VALUE: t.CDefine = 0xFF +BIT_RANGE: t.CDefine = 4 +EXPONENT_BASE: t.CDefine = 3 +MODULO_DIVISOR: t.CDefine = 7 +SCALE_SHIFT: t.CDefine = 4 +HALF_SCALE: t.CDefine = 0.5 \ No newline at end of file diff --git a/Test/TestProject/App/zc/logic_test.py b/Test/TestProject/App/zc/logic_test.py new file mode 100644 index 0000000..4d5c266 --- /dev/null +++ b/Test/TestProject/App/zc/logic_test.py @@ -0,0 +1,372 @@ +import config +import t, c + + +def ForRangeBasicTest(stop: t.CInt): + i: t.CInt = 0 + for i in range(stop): + print("For range: ", i) + print("Done") + + +def ForRangeWithStartStop(start: t.CInt, stop: t.CInt, step: t.CInt): + i: t.CInt = 0 + for i in range(start, stop, step): + print("Range: ", i) + print("Done") + + +def NestedForRangeTest(rows: t.CInt, cols: t.CInt): + r: t.CInt = 0 + c_val: t.CInt = 0 + for r in range(rows): + for c_val in range(cols): + print("Cell: ", r, c_val) + print("Grid done") + + +def ForWithIfBreakTest(data_size: t.CInt): + i: t.CInt = 0 + found: t.CInt = 0 + for i in range(data_size): + if i == 5: + print("Found at: ", i) + found = 1 + break + print("Checking: ", i) + if found == 0: + print("Not found") + + +def ForWithIfContinueTest(n: t.CInt): + i: t.CInt = 0 + count: t.CInt = 0 + for i in range(n): + if i % 2 == 0: + continue + print("Odd: ", i) + count += 1 + print("Total odd: ", count) + + +def ForWithElifBranchTest(value: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + for i in range(value): + if i < 3: + result += 1 + elif i < 6: + result += 2 + elif i < 9: + result += 3 + else: + result += 4 + print("Elif branch result: ", result) + + +def WhileWithIfElifElseTest(iterations: t.CInt): + counter: t.CInt = 0 + while counter < iterations: + if counter < 2: + print("Phase 1: ", counter) + elif counter < 4: + print("Phase 2: ", counter) + elif counter < 6: + print("Phase 3: ", counter) + else: + print("Phase 4: ", counter) + counter += 1 + print("While done") + + +def WhileWithNestedIfTest(limit: t.CInt): + outer: t.CInt = 0 + while outer < limit: + inner: t.CInt = 0 + while inner < limit: + if inner == 2 and outer == 1: + print("Target: ", outer, inner) + inner += 1 + continue + if inner > 3: + break + print("Inner: ", inner) + inner += 1 + outer += 1 + + +def WhileWithMatchCaseTest(value: t.CInt): + result: t.CInt = 0 + match value: + case 1: + result = 100 + print("Case 1") + case 2: + result = 200 + print("Case 2") + case 3: + result = 300 + print("Case 3") + case _: + result = 999 + print("Default case") + print("Match result: ", result) + + +def WhileWithMatchCaseNoBreakTest(value: t.CInt): + result: t.CInt = 0 + match value: + case 1: + result = 10 + print("Case 1: ", result) + case 2: + result = 20 + print("Case 2: ", result) + result += 5 + case 3: + result = 30 + print("Case 3: ", result) + case _: + result = 0 + print("No break match result: ", result) + + +def ComplexForWhileMatchTest(outer_limit: t.CInt, inner_limit: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + result: t.CInt = 0 + for i in range(outer_limit): + j = 0 + while j < inner_limit: + match (i, j): + case (0, 0): + result += 1 + case (1, 1): + result += 10 + case (2, _): + result += 100 + case (_, 0): + result += 1000 + case _: + result += 10000 + j += 1 + print("Complex match result: ", result) + + +def ForWithCaseAndNoBreakTest(n: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + for i in range(n): + match i: + case 0: + result += 1 + case 1: + result += 2 + case 2: + result += 3 + result += 10 + case _: + result += i + c.NoBreak + print("NoBreak result: ", result) + + +def WhileWithBreakConditionTest(limit: t.CInt): + i: t.CInt = 0 + early_exit: t.CInt = 0 + while i < limit: + if i == 3: + early_exit = 1 + break + if i == 2: + i += 1 + continue + print("While i: ", i) + i += 1 + + +def NestedWhileWithMultipleBreakPoints(depth: t.CInt, width: t.CInt): + d: t.CInt = 0 + w: t.CInt = 0 + count: t.CInt = 0 + while d < depth: + w = 0 + while w < width: + if w == 2: + break + if d == 1 and w == 1: + c.Break + count += 1 + w += 1 + d += 1 + print("Nested break count: ", count) + + +def ForWithMatchCaseInLoopTest(iterations: t.CInt): + i: t.CInt = 0 + sum_even: t.CInt = 0 + sum_odd: t.CInt = 0 + for i in range(iterations): + match i % 3: + case 0: + sum_even += i + case 1: + sum_odd += i + case 2: + sum_even += i * 2 + case _: + pass + print("Sum even: ", sum_even) + print("Sum odd: ", sum_odd) + + +def ComplexLogicWithElifChains(a: t.CInt, b: t.CInt, c_val: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + while i < 10: + if i < a: + if i < b: + result += 1 + elif i < c_val: + result += 2 + else: + result += 3 + elif i < b: + if i < c_val: + result += 4 + else: + result += 5 + elif i < c_val: + result += 6 + else: + result += 7 + i += 1 + print("Complex elif chain result: ", result) + + +def ForRangeWithStepAndCondition(n: t.CInt, step: t.CInt): + i: t.CInt = 0 + count: t.CInt = 0 + for i in range(0, n, step): + if i % 2 == 0: + print("Even step: ", i) + count += 1 + else: + if i > 5: + print("Large odd: ", i) + else: + print("Small odd: ", i) + print("Total steps: ", count) + + +def MatchCaseWithGuardConditions(value: t.CInt): + result: t.CInt = 0 + match value: + case 1 if config.TRUE == 1: + result = 100 + case 2 if config.FALSE == 0: + result = 200 + case 3: + result = 300 + case _: + result = 0 + print("Guard match result: ", result) + + +def WhileWithMultipleMatchCases(iterations: t.CInt): + i: t.CInt = 0 + state: t.CInt = 0 + while i < iterations: + match state: + case 0: + if i > 3: + state = 1 + case 1: + if i > 6: + state = 2 + case 2: + if i > 9: + state = 0 + print("State at i=", i, ": ", state) + i += 1 + + +def ForWhileMixWithBreakAndContinue(n: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + count: t.CInt = 0 + for i in range(n): + j = 0 + while j < n: + if j == 2: + j += 1 + continue + if i == 3 and j == 3: + break + count += 1 + j += 1 + print("Mix break continue count: ", count) + + +def ComplexCaseMatchingWithRanges(start: t.CInt, end: t.CInt): + i: t.CInt = 0 + bucket0: t.CInt = 0 + bucket1: t.CInt = 0 + bucket2: t.CInt = 0 + bucket3: t.CInt = 0 + for i in range(start, end): + match i: + case _ if i < 10: + bucket0 += 1 + case _ if i < 20: + bucket1 += 1 + case _ if i < 30: + bucket2 += 1 + case _: + bucket3 += 1 + print("Bucket0 (<10): ", bucket0) + print("Bucket1 (10-19): ", bucket1) + print("Bucket2 (20-29): ", bucket2) + print("Bucket3 (>=30): ", bucket3) + + +def WhileWithElifInElifChain(limit: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + while i < limit: + if i < 2: + if i == 0: + result = 1 + else: + result = 2 + elif i < 4: + if i == 2: + result = 3 + else: + result = 4 + elif i < 6: + if i == 4: + result = 5 + else: + result = 6 + else: + result = 7 + print("Chain result[", i, "]: ", result) + i += 1 + + +def ForWithCaseNoBreakNested(inner_limit: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + result: t.CInt = 0 + for i in range(3): + for j in range(inner_limit): + match j: + case 0: + result += 1 + case 1: + result += 2 + case _: + result += 3 + c.NoBreak + print("Nested NoBreak result: ", result) \ No newline at end of file diff --git a/Test/TestProject/App/zc/test.py b/Test/TestProject/App/zc/test.py new file mode 100644 index 0000000..24a08a8 --- /dev/null +++ b/Test/TestProject/App/zc/test.py @@ -0,0 +1,298 @@ +import config +import t, c + + +def AboutSizeTWhileTest(length: t.CSizeT): + pos: t.CSizeT = 0 + while pos < length: + print("Pos: ", pos) + if pos > config.POSMAX: break + pos += 1 + while pos < length: + if pos > config.POSMAX: + print("Pos: ", pos) + pos += 1 + else: + print("Pos: ", pos) + pos += 1 + if pos > config.POSMAX: break + else: + pos += 1 + print("Pos: ", pos) + while pos < length: + print("Pos: ", pos) + pos += 1 + + +def MathBasicTest(x: t.CInt, y: t.CInt) -> t.CInt: + result: t.CInt = 0 + result = x + y + print("Add: ", result) + result = x - y + print("Sub: ", result) + result = x * y + print("Mul: ", result) + if y != 0: + result = x / y + print("Div: ", result) + result = x % y if y != 0 else 0 + print("Mod: ", result) + return result + + +def MathAdvancedTest(value: t.CInt) -> t.CInt: + counter: t.CInt = 0 + scaled: t.CInt = 0 + offset: t.CInt = 0 + scaled = value * config.MATH_SCALE + print("Scaled: ", scaled) + offset = scaled + config.MATH_OFFSET + print("Offset: ", offset) + threshold: t.CInt = config.THRESHOLD + if offset > threshold: + print("Above threshold") + counter = 1 + elif offset == threshold: + print("At threshold") + counter = 0 + else: + print("Below threshold") + counter = -1 + return counter + + +def WhileWithElifElseTest(start: t.CInt, end: t.CInt): + idx: t.CInt = start + while idx < end: + if idx < 0: + print("Negative: ", idx) + elif idx == 0: + print("Zero: ", idx) + elif idx > 0 and idx < 5: + print("Small positive: ", idx) + elif idx >= 5 and idx < 10: + print("Medium positive: ", idx) + else: + print("Large: ", idx) + idx += 1 + + +def NestedWhileWithMacrosTest(outer_limit: t.CInt, inner_limit: t.CInt): + outer: t.CInt = 0 + inner: t.CInt = 0 + total_ops: t.CInt = 0 + while outer < outer_limit: + inner = 0 + while inner < inner_limit: + if inner == config.TRUE: + print("Inner loop break condition at: ", inner) + total_ops += 1 + inner += 1 + outer += 1 + print("Total operations: ", total_ops) + + +def ComplexConditionTest(a: t.CInt, b: t.CInt, c_val: t.CInt): + x: t.CInt = a + y: t.CInt = b + z: t.CInt = c_val + while x < config.LOOP_LIMIT: + if x > 0 and y > 0 and z > 0: + print("All positive: ", x, y, z) + elif x <= 0 and y <= 0 and z <= 0: + print("All non-positive: ", x, y, z) + else: + if x > 0: + print("X positive") + elif y > 0: + print("Y positive") + else: + print("Z positive") + x += 1 + y += 1 + z += 1 + + +def RetryLoopTest(retry_count: t.CInt) -> t.CInt: + attempt: t.CInt = 0 + success: t.CInt = 0 + while attempt < retry_count: + if attempt < config.MAX_RETRY: + print("Attempt: ", attempt, " less than max") + success = attempt * 2 + elif attempt == config.MAX_RETRY: + print("At max retry") + success = -1 + else: + print("Beyond max retry") + success = -2 + attempt += 1 + return success + + +def BitwiseOperationsTest(value: t.CInt) -> t.CInt: + shifted: t.CInt = 0 + masked: t.CInt = 0 + xored: t.CInt = 0 + anded: t.CInt = 0 + ored: t.CInt = 0 + shifted = value << config.SHIFT_AMOUNT + print("Left shift: ", shifted) + shifted = value >> config.SHIFT_AMOUNT + print("Right shift: ", shifted) + masked = value & config.MASK_VALUE + print("Bitwise AND mask: ", masked) + xored = value ^ config.MASK_VALUE + print("Bitwise XOR: ", xored) + anded = value & (config.MASK_VALUE >> config.SHIFT_AMOUNT) + print("Bitwise AND shifted mask: ", anded) + ored = value | config.SHIFT_AMOUNT + print("Bitwise OR shift: ", ored) + return masked + + +def CompoundAssignmentMathTest(a: t.CInt, b: t.CInt) -> t.CInt: + x: t.CInt = a + y: t.CInt = b + result: t.CInt = 0 + x += y + print("x += y: ", x) + x -= y + print("x -= y: ", x) + x *= y + print("x *= y: ", x) + x = a + y = b + if b != 0: + x /= b + print("x /= y: ", x) + x = a + if b != 0: + x %= b + print("x %%= y: ", x) + x = a + x <<= config.SHIFT_AMOUNT + print("x <<= shift: ", x) + x >>= config.SHIFT_AMOUNT + print("x >>= shift: ", x) + x &= config.MASK_VALUE + print("x &= mask: ", x) + x |= config.SHIFT_AMOUNT + print("x |= shift: ", x) + result = (a + b) * config.MATH_SCALE - config.MATH_OFFSET + print("Compound: (a+b)*scale-offset: ", result) + result = (a * b) + (config.MATH_OFFSET / 2) + print("Compound: a*b+offset/2: ", result) + return result + + +def PowerAndModuloTest(base: t.CInt, exponent: t.CInt) -> t.CInt: + result: t.CInt = 1 + temp: t.CInt = 0 + i: t.CInt = 0 + while i < exponent: + result *= base + i += 1 + print("Power: ", result) + temp = result % config.MODULO_DIVISOR + print("Power mod: ", temp) + temp = (base * config.EXPONENT_BASE) % config.MODULO_DIVISOR + print("base*exp %% divisor: ", temp) + temp = ((base + config.MATH_OFFSET) * config.MATH_SCALE) % config.MODULO_DIVISOR + print("(base+offset)*scale %% divisor: ", temp) + return result + + +def MixedArithmeticTest(x: t.CInt, y: t.CInt, z: t.CInt) -> t.CInt: + result: t.CInt = 0 + result = (x + y) * z + print("(x+y)*z: ", result) + result = x + (y * z) + print("x+(y*z): ", result) + result = ((x + y) * config.MATH_SCALE) - config.MATH_OFFSET + print("((x+y)*scale)-offset: ", result) + result = (x * config.MATH_SCALE) + (y * config.MATH_SCALE) + print("x*scale + y*scale: ", result) + result = ((x % config.MODULO_DIVISOR) + (y % config.MODULO_DIVISOR)) * config.MATH_SCALE + print("(x%%mod + y%%mod)*scale: ", result) + temp: t.CInt = 0 + temp = x | y & z + print("x | y & z: ", temp) + temp = (x | y) & z + print("(x | y) & z: ", temp) + temp = x ^ y ^ z + print("x ^ y ^ z: ", temp) + return result + + +def MathWithWhileLoopTest(start: t.CInt, count: t.CInt) -> t.CInt: + idx: t.CInt = 0 + accumulator: t.CInt = 0 + multiplier: t.CInt = config.MATH_SCALE + while idx < count: + temp: t.CInt = (start + idx) * multiplier + temp = temp + config.MATH_OFFSET + if temp > config.THRESHOLD: + temp = config.THRESHOLD + accumulator += temp + multiplier += 1 + idx += 1 + print("Accumulator: ", accumulator) + return accumulator + + +def ComplexNestedMathTest(a: t.CInt, b: t.CInt) -> t.CInt: + layer1: t.CInt = 0 + layer2: t.CInt = 0 + layer3: t.CInt = 0 + i: t.CInt = 0 + layer1 = (a + b) * config.MATH_SCALE + print("Layer1: ", layer1) + while i < config.BIT_RANGE: + layer2 += layer1 >> 1 + i += 1 + print("Layer2: ", layer2) + layer3 = layer1 + layer2 - config.MATH_OFFSET + print("Layer3: ", layer3) + layer3 = layer3 * config.MATH_SCALE / 2 if config.MATH_SCALE != 0 else layer3 + print("Layer3 scaled: ", layer3) + return layer3 + + +def ConditionalMathChainTest(value: t.CInt) -> t.CInt: + result: t.CInt = 0 + step: t.CInt = 0 + while step < config.LOOP_LIMIT: + if step < 5: + result += step * config.MATH_SCALE + print("step<5: ", result) + elif step < 10: + result -= step - config.MATH_OFFSET + print("step<10: ", result) + elif step < 15: + result += (step * config.MATH_OFFSET) / config.MATH_SCALE + print("step<15: ", result) + else: + result = result * config.EXPONENT_BASE % config.MODULO_DIVISOR + print("step>=15: ", result) + step += 1 + return result + + +def BitManipulationMathTest(value: t.CInt) -> t.CInt: + original: t.CInt = value + nibble: t.CInt = 0 + result: t.CInt = 0 + result = (value << config.SHIFT_AMOUNT) | (value >> (8 - config.SHIFT_AMOUNT)) + print("Rotate left: ", result) + result = (value >> config.SHIFT_AMOUNT) | (value << (8 - config.SHIFT_AMOUNT)) + print("Rotate right: ", result) + nibble = value & 0xF + print("Lower nibble: ", nibble) + nibble = (value >> config.BIT_RANGE) & 0xF + print("Upper nibble: ", nibble) + result = ((value & config.MASK_VALUE) + config.MATH_OFFSET) * config.MATH_SCALE + print("Masked and scaled: ", result) + result = ((original << config.SHIFT_AMOUNT) ^ original) & config.MASK_VALUE + print("XOR shifted: ", result) + return result \ No newline at end of file diff --git a/Test/TestProject/output/3cd87018ba1e76bc.deps.json b/Test/TestProject/output/3cd87018ba1e76bc.deps.json new file mode 100644 index 0000000..123a27c --- /dev/null +++ b/Test/TestProject/output/3cd87018ba1e76bc.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "cfc83b830141b00e", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/49cae05490a513cb.deps.json b/Test/TestProject/output/49cae05490a513cb.deps.json new file mode 100644 index 0000000..c4c0f68 --- /dev/null +++ b/Test/TestProject/output/49cae05490a513cb.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "cfc83b830141b00e", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/75583f20a922ab2e.deps.json b/Test/TestProject/output/75583f20a922ab2e.deps.json new file mode 100644 index 0000000..d749fae --- /dev/null +++ b/Test/TestProject/output/75583f20a922ab2e.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "cfc83b830141b00e", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/8e0d8fdba991b3b4.deps.json b/Test/TestProject/output/8e0d8fdba991b3b4.deps.json new file mode 100644 index 0000000..e76c1ac --- /dev/null +++ b/Test/TestProject/output/8e0d8fdba991b3b4.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/cfc83b830141b00e.deps.json b/Test/TestProject/output/cfc83b830141b00e.deps.json new file mode 100644 index 0000000..0705eab --- /dev/null +++ b/Test/TestProject/output/cfc83b830141b00e.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "8e0d8fdba991b3b4", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/e5fb16127bce8f68.deps.json b/Test/TestProject/output/e5fb16127bce8f68.deps.json new file mode 100644 index 0000000..452f0c0 --- /dev/null +++ b/Test/TestProject/output/e5fb16127bce8f68.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "cfc83b830141b00e", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "zc.logic_test": "faec3233d98371da", "logic_test": "faec3233d98371da"} \ No newline at end of file diff --git a/Test/TestProject/output/faec3233d98371da.deps.json b/Test/TestProject/output/faec3233d98371da.deps.json new file mode 100644 index 0000000..1ac0a97 --- /dev/null +++ b/Test/TestProject/output/faec3233d98371da.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "test_numpy": "3cd87018ba1e76bc", "vipermath": "3f7c5e78d8652535", "main": "49cae05490a513cb", "stdio": "6f62fe05c5ea1ceb", "zc.test": "75583f20a922ab2e", "test": "75583f20a922ab2e", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "config": "cfc83b830141b00e", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "numpy.__init__": "c3a6aed1f1fb8b1e", "__init__": "c3a6aed1f1fb8b1e", "numpy": "c3a6aed1f1fb8b1e", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "testcheck": "dd3002730623424b", "zc.class_test": "e5fb16127bce8f68", "class_test": "e5fb16127bce8f68", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/TestProject/project.json b/Test/TestProject/project.json new file mode 100644 index 0000000..d8ccb56 --- /dev/null +++ b/Test/TestProject/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TestProject", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "TestProject.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/TestProject/temp/271ea3decb810db2.pyi b/Test/TestProject/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/TestProject/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/TestProject/temp/3cd87018ba1e76bc.doc.json b/Test/TestProject/temp/3cd87018ba1e76bc.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/3cd87018ba1e76bc.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/3cd87018ba1e76bc.pyi b/Test/TestProject/temp/3cd87018ba1e76bc.pyi new file mode 100644 index 0000000..669f365 --- /dev/null +++ b/Test/TestProject/temp/3cd87018ba1e76bc.pyi @@ -0,0 +1,108 @@ +""" +Auto-generated Python stub file from test_numpy.py +Module: test_numpy +""" + + +from stdint import * +import numpy +from stdio import printf +from stdlib import calloc, free +import string +import vipermath +import memhub +import t, c + +EPS: t.CExtern | t.CDouble +EPS_LOOSE: t.CExtern | t.CDouble +SIMD_BLOCK: t.CExtern | t.CSizeT +arena: t.CExtern | t.CUInt8T | t.CPtr +pool: t.CExtern | memhub.MemPool | t.CPtr +test_passed: t.CExtern | t.CInt +test_failed: t.CExtern | t.CInt + +def check(name: str, condition: t.CInt, detail: str) -> t.CInt: pass + +def approx_eq(a: t.CDouble, b: t.CDouble) -> t.CInt: pass + +def approx_loose(a: t.CDouble, b: t.CDouble) -> t.CInt: pass + +def section_header(title: str) -> t.CInt: pass + +def section_footer() -> t.CInt: pass + +def vec4(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble, v3: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def vec3(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble, v2: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def vec2(p: memhub.MemPool | t.CPtr, v0: t.CDouble, v1: t.CDouble) -> numpy.ndarray | t.CPtr: pass + +def test_zeros_ones() -> t.CInt: pass + +def test_arange_linspace() -> t.CInt: pass + +def test_eye_diag() -> t.CInt: pass + +def test_sum_mean_min_max() -> t.CInt: pass + +def test_fill_copy() -> t.CInt: pass + +def test_add_sub_mul_div() -> t.CInt: pass + +def test_neg() -> t.CInt: pass + +def test_len() -> t.CInt: pass + +def test_scalar_ops() -> t.CInt: pass + +def test_math_funcs() -> t.CInt: pass + +def test_matmul() -> t.CInt: pass + +def test_dot() -> t.CInt: pass + +def test_transpose() -> t.CInt: pass + +def test_var_std_norm() -> t.CInt: pass + +def test_clip_concatenate() -> t.CInt: pass + +def test_sort_reverse() -> t.CInt: pass + +def test_print_arr() -> t.CInt: pass + +def test_floordiv_mod() -> t.CInt: pass + +def test_additional_math() -> t.CInt: pass + +def test_trig_hyperbolic() -> t.CInt: pass + +def test_degrees_radians() -> t.CInt: pass + +def test_cumsum_diff() -> t.CInt: pass + +def test_max_min_where() -> t.CInt: pass + +def test_flatten_trace() -> t.CInt: pass + +def test_outer() -> t.CInt: pass + +def test_bool_comparison() -> t.CInt: pass + +def test_linalg() -> t.CInt: pass + +def test_additional_creation() -> t.CInt: pass + +def test_interp() -> t.CInt: pass + +def test_numpy_edge() -> t.CInt: pass + +def bench_numpy_perf() -> t.CInt: pass + +def test_numpy_correct() -> t.CInt: pass + +def test_numpy_edge_main() -> t.CInt: pass + +def bench_numpy_main() -> t.CInt: pass + +def test_numpy_main() -> t.CInt: pass diff --git a/Test/TestProject/temp/3f7c5e78d8652535.pyi b/Test/TestProject/temp/3f7c5e78d8652535.pyi new file mode 100644 index 0000000..abe01c2 --- /dev/null +++ b/Test/TestProject/temp/3f7c5e78d8652535.pyi @@ -0,0 +1,128 @@ +""" +Auto-generated Python stub file from vipermath.py +Module: vipermath +""" + + +import t, c + +U_M_PI: t.CDefine = 3.14159265358979323846 +U_M_E: t.CDefine = 2.71828182845904523536 +U_M_PI_2: t.CDefine = 1.57079632679489661923 +U_M_PI_4: t.CDefine = 0.78539816339744830962 +U_M_1_PI: t.CDefine = 0.31830988618379067154 +U_M_2_PI: t.CDefine = 0.63661977236758134308 +U_M_LN2: t.CDefine = 0.69314718055994530942 +U_M_LN10: t.CDefine = 2.30258509299404568402 +U_M_2_SQRT_PI: t.CDefine = 1.77245385090551602730 + +def radians(degrees: t.CDouble) -> t.CDouble: pass + +def degrees(radians: t.CDouble) -> t.CDouble: pass + +def sin(x: t.CDouble) -> t.CDouble: pass + +def cos(x: t.CDouble) -> t.CDouble: pass + +def tan(x: t.CDouble) -> t.CDouble: pass + +def asin(x: t.CDouble) -> t.CDouble: pass + +def acos(x: t.CDouble) -> t.CDouble: pass + +def _atan_core(x: t.CDouble) -> t.CDouble: pass + +def atan(x: t.CDouble) -> t.CDouble: pass + +def atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble: pass + +def sinh(x: t.CDouble) -> t.CDouble: pass + +def cosh(x: t.CDouble) -> t.CDouble: pass + +def tanh(x: t.CDouble) -> t.CDouble: pass + +def exp(x: t.CDouble) -> t.CDouble: pass + +def log(x: t.CDouble) -> t.CDouble: pass + +def sqrt(x: t.CDouble) -> t.CDouble: pass + +def abs(x: t.CInt) -> t.CInt: pass + +def fabs(x: t.CDouble) -> t.CDouble: pass + +def labs(x: t.CLong) -> t.CLong: pass + +def factorial(n: t.CInt) -> t.CDouble: pass + +def combination(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def permutation(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def powf(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def cbrt(x: t.CDouble) -> t.CDouble: pass + +def hypot(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def floor(x: t.CDouble) -> t.CDouble: pass + +def ceil(x: t.CDouble) -> t.CDouble: pass + +def round(x: t.CDouble) -> t.CDouble: pass + +def trunc(x: t.CDouble) -> t.CDouble: pass + +def fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def fmodf(x: float, y: float) -> float: pass + +def modf(x: t.CDouble, iptr: t.CDouble | t.CPtr) -> t.CDouble: pass + +def log10(x: t.CDouble) -> t.CDouble: pass + +def log2(x: t.CDouble) -> t.CDouble: pass + +def exp2(x: t.CDouble) -> t.CDouble: pass + +def expm1(x: t.CDouble) -> t.CDouble: pass + +def log1p(x: t.CDouble) -> t.CDouble: pass + +def asinh(x: t.CDouble) -> t.CDouble: pass + +def acosh(x: t.CDouble) -> t.CDouble: pass + +def atanh(x: t.CDouble) -> t.CDouble: pass + +def gamma(x: t.CDouble) -> t.CDouble: pass + +def erf(x: t.CDouble) -> t.CDouble: pass + +def erfc(x: t.CDouble) -> t.CDouble: pass + +def sqrtf(x: t.CFloat) -> t.CFloat: pass + +def sinf(x: t.CFloat) -> t.CFloat: pass + +def cosf(x: t.CFloat) -> t.CFloat: pass + +def tanf(x: t.CFloat) -> t.CFloat: pass + +def fabsf(x: t.CFloat) -> t.CFloat: pass + +def floorf(x: t.CFloat) -> t.CFloat: pass + +def ceilf(x: t.CFloat) -> t.CFloat: pass + + +class _U(t.CUnion): + d: t.CDouble + u: t.CUInt64T + +def isnan(x: t.CDouble) -> t.CStatic | t.CInt: pass + +def isinf(x: t.CDouble) -> t.CStatic | t.CInt: pass diff --git a/Test/TestProject/temp/49cae05490a513cb.doc.json b/Test/TestProject/temp/49cae05490a513cb.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/49cae05490a513cb.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/49cae05490a513cb.pyi b/Test/TestProject/temp/49cae05490a513cb.pyi new file mode 100644 index 0000000..e1af96b --- /dev/null +++ b/Test/TestProject/temp/49cae05490a513cb.pyi @@ -0,0 +1,19 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + +import t +import c + + +import stdio +import stdint +import zc.config as config +import zc.test as test +import zc.logic_test as logic +import zc.class_test as class_test +import testcheck +import test_numpy + +def main() -> stdint.INT: pass diff --git a/Test/TestProject/temp/6f62fe05c5ea1ceb.pyi b/Test/TestProject/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/TestProject/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/TestProject/temp/75583f20a922ab2e.doc.json b/Test/TestProject/temp/75583f20a922ab2e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/75583f20a922ab2e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/75583f20a922ab2e.pyi b/Test/TestProject/temp/75583f20a922ab2e.pyi new file mode 100644 index 0000000..10da71f --- /dev/null +++ b/Test/TestProject/temp/75583f20a922ab2e.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from zc.test.py +Module: zc.test +""" + + +import config +import t, c + +def AboutSizeTWhileTest(length: t.CSizeT) -> t.CInt: pass + +def MathBasicTest(x: t.CInt, y: t.CInt) -> t.CInt: pass + +def MathAdvancedTest(value: t.CInt) -> t.CInt: pass + +def WhileWithElifElseTest(start: t.CInt, end: t.CInt) -> t.CInt: pass + +def NestedWhileWithMacrosTest(outer_limit: t.CInt, inner_limit: t.CInt) -> t.CInt: pass + +def ComplexConditionTest(a: t.CInt, b: t.CInt, c_val: t.CInt) -> t.CInt: pass + +def RetryLoopTest(retry_count: t.CInt) -> t.CInt: pass + +def BitwiseOperationsTest(value: t.CInt) -> t.CInt: pass + +def CompoundAssignmentMathTest(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def PowerAndModuloTest(base: t.CInt, exponent: t.CInt) -> t.CInt: pass + +def MixedArithmeticTest(x: t.CInt, y: t.CInt, z: t.CInt) -> t.CInt: pass + +def MathWithWhileLoopTest(start: t.CInt, count: t.CInt) -> t.CInt: pass + +def ComplexNestedMathTest(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def ConditionalMathChainTest(value: t.CInt) -> t.CInt: pass + +def BitManipulationMathTest(value: t.CInt) -> t.CInt: pass diff --git a/Test/TestProject/temp/7e529fe7a078cfef.pyi b/Test/TestProject/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/TestProject/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/TestProject/temp/8e0d8fdba991b3b4.doc.json b/Test/TestProject/temp/8e0d8fdba991b3b4.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/8e0d8fdba991b3b4.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/8e0d8fdba991b3b4.pyi b/Test/TestProject/temp/8e0d8fdba991b3b4.pyi new file mode 100644 index 0000000..1aebea2 --- /dev/null +++ b/Test/TestProject/temp/8e0d8fdba991b3b4.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from config.py +Module: config +""" + + +import stdint +import t, c + +A: t.CExtern | t.CChar | t.CPtr + +class BIT_TEST(t.Object): + a: t.CInt | t.Bit(1) + b: t.CInt | t.Bit(1) + c: t.CInt | t.Bit(1) + d: t.CInt | t.Bit(5) +class OOP_TEST(t.Object): + a: t.CInt + def __init__(self: OOP_TEST) -> t.CInt: pass + def __call__(self: OOP_TEST) -> stdint.UINT: pass +class ENUM_TEST(t.CEnum): + A: t.State = 0 + B: t.State + C: t.State + Len: t.State \ No newline at end of file diff --git a/Test/TestProject/temp/90c53dd6db8d41cf.pyi b/Test/TestProject/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/TestProject/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/TestProject/temp/_phase1_manifest.json b/Test/TestProject/temp/_phase1_manifest.json new file mode 100644 index 0000000..77e537d --- /dev/null +++ b/Test/TestProject/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\config.py": {"sha1": "8e0d8fdba991b3b4", "mtime": 1781111175.0269876, "size": 445}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\main.py": {"sha1": "49cae05490a513cb", "mtime": 1782949786.0815294, "size": 8223}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\test_numpy.py": {"sha1": "3cd87018ba1e76bc", "mtime": 1782828269.700148, "size": 40140}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\zc\\class_test.py": {"sha1": "e5fb16127bce8f68", "mtime": 1782266207.4312654, "size": 6472}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\zc\\config.py": {"sha1": "cfc83b830141b00e", "mtime": 1778402243.1661563, "size": 531}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\zc\\logic_test.py": {"sha1": "faec3233d98371da", "mtime": 1778402365.7843194, "size": 8848}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject\\App\\zc\\test.py": {"sha1": "75583f20a922ab2e", "mtime": 1778401973.4865518, "size": 8976}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\numpy\\__init__.py": {"sha1": "c3a6aed1f1fb8b1e", "mtime": 1782949786.0890937, "size": 39111}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vipermath.py": {"sha1": "3f7c5e78d8652535", "mtime": 1781532528.2518907, "size": 15412}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/TestProject/temp/_sha1_map.txt b/Test/TestProject/temp/_sha1_map.txt new file mode 100644 index 0000000..334c60d --- /dev/null +++ b/Test/TestProject/temp/_sha1_map.txt @@ -0,0 +1,19 @@ +271ea3decb810db2:includes/atom.py +3cd87018ba1e76bc:test_numpy.py +3f7c5e78d8652535:includes/vipermath.py +49cae05490a513cb:main.py +6f62fe05c5ea1ceb:includes/stdio.py +75583f20a922ab2e:zc\test.py +7e529fe7a078cfef:includes/w32\win32base.py +8e0d8fdba991b3b4:config.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c3a6aed1f1fb8b1e:includes/numpy\__init__.py +c9f4be41ca1cc2b4:includes/viperio.py +cfc83b830141b00e:zc\config.py +dd3002730623424b:includes/testcheck.py +e5fb16127bce8f68:zc\class_test.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py +faec3233d98371da:zc\logic_test.py diff --git a/Test/TestProject/temp/_shared_sym.json b/Test/TestProject/temp/_shared_sym.json new file mode 100644 index 0000000..d2c1348 Binary files /dev/null and b/Test/TestProject/temp/_shared_sym.json differ diff --git a/Test/TestProject/temp/ab6e54ba9a669f76.pyi b/Test/TestProject/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/TestProject/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/TestProject/temp/bbdf3bbd4c3bc28c.pyi b/Test/TestProject/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/TestProject/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestProject/temp/c3a6aed1f1fb8b1e.pyi b/Test/TestProject/temp/c3a6aed1f1fb8b1e.pyi new file mode 100644 index 0000000..68a99b1 --- /dev/null +++ b/Test/TestProject/temp/c3a6aed1f1fb8b1e.pyi @@ -0,0 +1,240 @@ +""" +Auto-generated Python stub file from numpy.__init__.py +Module: numpy.__init__ +""" + + +from stdint import * +import stdlib +import string +import vipermath +import stdio +import t, c +import memhub + +MAX_NDIM: t.CDefine = 4 +float64: t.CTypedef = t.CDouble +float32: t.CTypedef = t.CFloat +int64: t.CTypedef = t.CLong +int32: t.CTypedef = t.CInt +uint8: t.CTypedef = t.CUnsignedChar +pi: t.CDefine = 3.14159265358979323846 +e: t.CDefine = 2.71828182845904523536 + +@t.Object +class ndarray[T]: + data: T | t.CPtr + shape: t.CArray[t.CSizeT, MAX_NDIM] + strides: t.CArray[t.CSizeT, MAX_NDIM] + ndim: t.CInt + size: t.CSizeT + owns_data: t.CInt + pool: memhub.MemManager | t.CPtr + def __new__(self: ndarray, pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> t.CInt: pass + def __add__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __sub__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mul__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __truediv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __floordiv__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __mod__(self: ndarray, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: pass + def __neg__(self: ndarray) -> ndarray[T] | t.CPtr: pass + def __len__(self: ndarray) -> t.CInt: pass + def at2d(self: ndarray, row: t.CSizeT, col: t.CSizeT) -> T: pass + def set2d(self: ndarray, row: t.CSizeT, col: t.CSizeT, val: T) -> t.CInt: pass + def delete(self: ndarray) -> t.CInt: pass + def fill(self: ndarray, val: T) -> t.CInt: pass + def copy(self: ndarray) -> ndarray[T] | t.CPtr: pass + def reshape(self: ndarray, new_shape: INTPTR, new_ndim: t.CInt) -> ndarray[T] | t.CPtr: pass + def sum(self: ndarray) -> T: pass + def mean(self: ndarray) -> T: pass + def min(self: ndarray) -> T: pass + def max(self: ndarray) -> T: pass + def argmax(self: ndarray) -> t.CInt: pass + def argmin(self: ndarray) -> t.CInt: pass + def dot(self: ndarray, other: ndarray[T] | t.CPtr) -> T: pass + def T(self: ndarray) -> ndarray[T] | t.CPtr: pass + def print_arr(self: ndarray) -> t.CInt: pass + +Float64Array: t.CTypedef = ndarray[t.CDouble] +Float32Array: t.CTypedef = ndarray[t.CFloat] +Int64Array: t.CTypedef = ndarray[t.CLong] +Int32Array: t.CTypedef = ndarray[t.CInt] +Uint8Array: t.CTypedef = ndarray[t.CUnsignedChar] + +def _alloc_ndarray(pool: memhub.MemManager | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def _compute_strides(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def _empty_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr, data: t.CDouble | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full(pool: memhub.MemManager | t.CPtr, n: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, step: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, num: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def eye(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def diag(pool: memhub.MemManager | t.CPtr, vals: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_abs(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sqrt(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_exp(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_pow(a: ndarray[t.CDouble] | t.CPtr, p: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def add_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def mul_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def sub_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def div_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def matmul(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def dot_product(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_sum(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_mean(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_min(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_max(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def np_argmax(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_argmin(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def var(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def std(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def norm(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def clip(a: ndarray[t.CDouble] | t.CPtr, lo: t.CDouble, hi: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def concatenate(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def sort_arr(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def reverse(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log10(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_log2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_floor(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_ceil(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_round(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sign(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_tanh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_sinh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_cosh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arcsin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arccos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_arctan2(y: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_degrees(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_radians(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isnan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_isinf(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_maximum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_minimum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cumsum(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def diff(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def flatten(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def trace(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def outer(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_where(condition: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_count_nonzero(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_all(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_any(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: pass + +def np_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_not_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_less_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_greater_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def empty(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: pass + +def full2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def zeros_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def ones_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def arange1(pool: memhub.MemManager | t.CPtr, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def linspace2(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: pass + +def meshgrid(x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def det2x2(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def inv2x2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def cross3(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: pass + +def np_interp(x: t.CDouble, xp: ndarray[t.CDouble] | t.CPtr, fp: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def prod(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def median(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: pass + +def percentile(a: ndarray[t.CDouble] | t.CPtr, q: t.CDouble) -> t.CDouble: pass diff --git a/Test/TestProject/temp/c9f4be41ca1cc2b4.pyi b/Test/TestProject/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/TestProject/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject/temp/cfc83b830141b00e.doc.json b/Test/TestProject/temp/cfc83b830141b00e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/cfc83b830141b00e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/cfc83b830141b00e.pyi b/Test/TestProject/temp/cfc83b830141b00e.pyi new file mode 100644 index 0000000..5af877f --- /dev/null +++ b/Test/TestProject/temp/cfc83b830141b00e.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from zc.config.py +Module: zc.config +""" + + +import t, c + +POSMAX: t.CDefine = 100 +MAX_RETRY: t.CDefine = 5 +BUFFER_SIZE: t.CDefine = 256 +PI: t.CDefine = 3.14159 +EULER: t.CDefine = 2.71828 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +NULL: t.CDefine = 0 +MATH_OFFSET: t.CDefine = 10 +MATH_SCALE: t.CDefine = 2 +THRESHOLD: t.CDefine = 50 +LOOP_LIMIT: t.CDefine = 20 +SHIFT_AMOUNT: t.CDefine = 2 +MASK_VALUE: t.CDefine = 0xFF +BIT_RANGE: t.CDefine = 4 +EXPONENT_BASE: t.CDefine = 3 +MODULO_DIVISOR: t.CDefine = 7 +SCALE_SHIFT: t.CDefine = 4 +HALF_SCALE: t.CDefine = 0.5 \ No newline at end of file diff --git a/Test/TestProject/temp/dd3002730623424b.pyi b/Test/TestProject/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/TestProject/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/TestProject/temp/e5fb16127bce8f68.doc.json b/Test/TestProject/temp/e5fb16127bce8f68.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/e5fb16127bce8f68.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/e5fb16127bce8f68.pyi b/Test/TestProject/temp/e5fb16127bce8f68.pyi new file mode 100644 index 0000000..5677a22 --- /dev/null +++ b/Test/TestProject/temp/e5fb16127bce8f68.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from zc.class_test.py +Module: zc.class_test +""" + + +import stdio +import stdint +import t, c + +class SimpleData(t.Object): + x: t.CInt + y: t.CInt +class Point3D(t.Object): + x: t.CFloat + y: t.CFloat + z: t.CFloat +class Student(t.Object): + name: t.CChar | t.CPtr + age: t.CInt + score: t.CInt + def __init__(self: Student, n: t.CChar | t.CPtr, a: t.CInt, s: t.CInt) -> t.CInt: pass + def get_age(self: Student) -> t.CInt: pass + def get_score(self: Student) -> t.CInt: pass +class Counter(t.Object): + value: t.CInt + def __init__(self: Counter, initial: t.CInt) -> t.CInt: pass + def increment(self: Counter) -> t.CInt: pass + def get_value(self: Counter) -> t.CInt: pass +class StackObj(t.Object): + data: t.CInt + next_ptr: t.CVoid | t.CPtr + +def TestBasicDataClass() -> t.CInt: pass + +def TestPoint3DClass() -> t.CInt: pass + +def TestStudentClass() -> t.CInt: pass + +def TestCounterClass() -> t.CInt: pass + +def TestStackAllocation() -> t.CInt: pass + +def TestHeapAllocation() -> t.CInt: pass + +def TestHeapObjectWithInit() -> t.CInt: pass + +def TestMultipleStackObjects() -> t.CInt: pass + +def TestImplicitInitCall() -> t.CInt: pass + +def TestClassWithBitFields() -> t.CInt: pass + +def TestIntArray() -> t.CInt: pass + +def TestFloatArray() -> t.CInt: pass + +def TestCharArray() -> t.CInt: pass + +def TestNestedArray() -> t.CInt: pass + +def TestArrayWithStruct() -> t.CInt: pass + +def TestArrayPointer() -> t.CInt: pass + +def TestDynamicString() -> t.CInt: pass diff --git a/Test/TestProject/temp/ee084e9fc6ee413a.pyi b/Test/TestProject/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/TestProject/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/TestProject/temp/f5522571bcce7bcb.pyi b/Test/TestProject/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/TestProject/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/TestProject/temp/faec3233d98371da.doc.json b/Test/TestProject/temp/faec3233d98371da.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject/temp/faec3233d98371da.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject/temp/faec3233d98371da.pyi b/Test/TestProject/temp/faec3233d98371da.pyi new file mode 100644 index 0000000..308f5cb --- /dev/null +++ b/Test/TestProject/temp/faec3233d98371da.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from zc.logic_test.py +Module: zc.logic_test +""" + + +import config +import t, c + +def ForRangeBasicTest(stop: t.CInt) -> t.CInt: pass + +def ForRangeWithStartStop(start: t.CInt, stop: t.CInt, step: t.CInt) -> t.CInt: pass + +def NestedForRangeTest(rows: t.CInt, cols: t.CInt) -> t.CInt: pass + +def ForWithIfBreakTest(data_size: t.CInt) -> t.CInt: pass + +def ForWithIfContinueTest(n: t.CInt) -> t.CInt: pass + +def ForWithElifBranchTest(value: t.CInt) -> t.CInt: pass + +def WhileWithIfElifElseTest(iterations: t.CInt) -> t.CInt: pass + +def WhileWithNestedIfTest(limit: t.CInt) -> t.CInt: pass + +def WhileWithMatchCaseTest(value: t.CInt) -> t.CInt: pass + +def WhileWithMatchCaseNoBreakTest(value: t.CInt) -> t.CInt: pass + +def ComplexForWhileMatchTest(outer_limit: t.CInt, inner_limit: t.CInt) -> t.CInt: pass + +def ForWithCaseAndNoBreakTest(n: t.CInt) -> t.CInt: pass + +def WhileWithBreakConditionTest(limit: t.CInt) -> t.CInt: pass + +def NestedWhileWithMultipleBreakPoints(depth: t.CInt, width: t.CInt) -> t.CInt: pass + +def ForWithMatchCaseInLoopTest(iterations: t.CInt) -> t.CInt: pass + +def ComplexLogicWithElifChains(a: t.CInt, b: t.CInt, c_val: t.CInt) -> t.CInt: pass + +def ForRangeWithStepAndCondition(n: t.CInt, step: t.CInt) -> t.CInt: pass + +def MatchCaseWithGuardConditions(value: t.CInt) -> t.CInt: pass + +def WhileWithMultipleMatchCases(iterations: t.CInt) -> t.CInt: pass + +def ForWhileMixWithBreakAndContinue(n: t.CInt) -> t.CInt: pass + +def ComplexCaseMatchingWithRanges(start: t.CInt, end: t.CInt) -> t.CInt: pass + +def WhileWithElifInElifChain(limit: t.CInt) -> t.CInt: pass + +def ForWithCaseNoBreakNested(inner_limit: t.CInt) -> t.CInt: pass diff --git a/Test/TestProject2/App/main.py b/Test/TestProject2/App/main.py new file mode 100644 index 0000000..69f06d0 --- /dev/null +++ b/Test/TestProject2/App/main.py @@ -0,0 +1,35 @@ +from stdint import * +import w32.win32console +import config +import hashlib +import binascii +import zlib.pyzlib as zp + +import t, c +import testcheck + + +def main() -> t.CInt | t.CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("TestProject2: 综合库测试") + + testcheck.section("zlib runtime version") + print(zp.runtime_version()) + testcheck.ok("zlib runtime version printed") + + testcheck.section("HASHLIB") + md5_buf: t.CArray[t.CUInt8T, hashlib.MD5_DIGEST_LEN] + mctx = hashlib.md5() + mctx.update("hello") + mctx.final(md5_buf) + testcheck.ok("HASHLIB md5 completed") + + testcheck.section("BINASCII") + binascii.init_binascii() + crc_val: t.CUnsignedInt = binascii.crc32("hello", 5, 0) + testcheck.ok("BINASCII crc32 completed") + + testcheck.info("END") + return testcheck.end() diff --git a/Test/TestProject2/App/zc/class_test.py b/Test/TestProject2/App/zc/class_test.py new file mode 100644 index 0000000..4594f8b --- /dev/null +++ b/Test/TestProject2/App/zc/class_test.py @@ -0,0 +1,245 @@ +import stdio +import stdint +import t, c + + +class SimpleData(t.Object): + x: t.CInt + y: t.CInt + + +class Point3D(t.Object): + x: t.CFloat + y: t.CFloat + z: t.CFloat + + +class Student(t.Object): + name: t.CChar | t.CPtr + age: t.CInt + score: t.CInt + + def __init__(self, n: t.CChar | t.CPtr, a: t.CInt, s: t.CInt): + self.name = n + self.age = a + self.score = s + + def get_age(self) -> t.CInt: + return self.age + + def get_score(self) -> t.CInt: + return self.score + + +class Counter(t.Object): + value: t.CInt + + def __init__(self, initial: t.CInt): + self.value = initial + + def increment(self): + self.value += 1 + + def get_value(self) -> t.CInt: + return self.value + + +class StackObj(t.Object): + data: t.CInt + next_ptr: t.CVoid | t.CPtr + + +def TestBasicDataClass(): + stdio.printf("=== Test Basic Data Class ===\n") + p: SimpleData = SimpleData() + p.x = 10 + p.y = 20 + stdio.printf("SimpleData: x=%d, y=%d\n", p.x, p.y) + + +def TestPoint3DClass(): + stdio.printf("=== Test Point3D Class ===\n") + pt: Point3D = Point3D() + pt.x = 1.5 + pt.y = 2.5 + pt.z = 3.5 + stdio.printf("Point3D: x=%.1f, y=%.1f, z=%.1f\n", pt.x, pt.y, pt.z) + + +def TestStudentClass(): + stdio.printf("=== Test Student Class ===\n") + s: Student = Student("Alice", 20, 95) + stdio.printf("Student: name=%s, age=%d, score=%d\n", s.name, s.age, s.score) + stdio.printf(" get_age()=%d, get_score()=%d\n", s.get_age(), s.get_score()) + + +def TestCounterClass(): + stdio.printf("=== Test Counter Class ===\n") + cnt: Counter = Counter(0) + stdio.printf("Initial value: %d\n", cnt.get_value()) + cnt.increment() + cnt.increment() + cnt.increment() + stdio.printf("After 3 increments: %d\n", cnt.get_value()) + + +def TestStackAllocation(): + stdio.printf("=== Test Stack Allocation ===\n") + obj1: SimpleData = SimpleData() + obj1.x = 100 + obj1.y = 200 + stdio.printf("Stack obj1: x=%d, y=%d\n", obj1.x, obj1.y) + + obj2: Point3D = Point3D() + obj2.x = 10.0 + obj2.y = 20.0 + obj2.z = 30.0 + stdio.printf("Stack obj2: x=%.1f, y=%.1f, z=%.1f\n", obj2.x, obj2.y, obj2.z) + + +def TestHeapAllocation(): + stdio.printf("=== Test Heap Allocation ===\n") + ptr: t.CVoid | t.CPtr = c.malloc(1024) + if ptr != 0: + stdio.printf("Allocated 1024 bytes at %p\n", ptr) + c.free(ptr) + stdio.printf("Freed memory\n") + else: + stdio.printf("malloc failed\n") + + +def TestHeapObjectWithInit(): + stdio.printf("=== Test Heap Object with Init ===\n") + mem_size: t.CSizeT = 256 + ptr: t.CVoid | t.CPtr = c.malloc(mem_size) + if ptr != 0: + stdio.printf("Allocated %d bytes\n", mem_size) + c.free(ptr) + else: + stdio.printf("malloc failed\n") + + +def TestMultipleStackObjects(): + stdio.printf("=== Test Multiple Stack Objects ===\n") + objs: SimpleData = SimpleData() + objs.x = 1 + objs.y = 2 + obj2: Point3D = Point3D() + obj2.x = 5.0 + obj2.y = 10.0 + obj2.z = 15.0 + obj3: Counter = Counter(42) + stdio.printf("objs: x=%d, y=%d\n", objs.x, objs.y) + stdio.printf("obj2: x=%.1f, y=%.1f, z=%.1f\n", obj2.x, obj2.y, obj2.z) + stdio.printf("obj3 initial: %d\n", obj3.get_value()) + obj3.increment() + obj3.increment() + stdio.printf("obj3 after 2 increments: %d\n", obj3.get_value()) + + +def TestImplicitInitCall(): + stdio.printf("=== Test Implicit Init Call ===\n") + stu1: Student = Student("Bob", 22, 88) + stu2: Student = Student("Charlie", 19, 77) + stdio.printf("stu1: name=%s, age=%d, score=%d\n", stu1.name, stu1.age, stu1.score) + stdio.printf("stu2: name=%s, age=%d, score=%d\n", stu2.name, stu2.age, stu2.score) + + +def TestClassWithBitFields(): + stdio.printf("=== Test Class with Bit Fields ===\n") + flags: StackObj = StackObj() + flags.data = 0 + flags.next_ptr = 0 + stdio.printf("StackObj: data=%d, next_ptr=%p\n", flags.data, flags.next_ptr) + + +def TestIntArray(): + stdio.printf("=== Test Int Array ===\n") + arr: t.CArray[t.CInt, 5] = t.CArray[t.CInt, 5]() + i: t.CInt = 0 + while i < 5: + arr[i] = i * 10 + i += 1 + i = 0 + while i < 5: + stdio.printf("arr[%d]=%d\n", i, arr[i]) + i += 1 + + +def TestFloatArray(): + stdio.printf("=== Test Float Array ===\n") + farr: t.CArray[t.CFloat, 4] = t.CArray[t.CFloat, 4]() + farr[0] = 1.5 + farr[1] = 2.5 + farr[2] = 3.5 + farr[3] = 4.5 + i: t.CInt = 0 + while i < 4: + stdio.printf("farr[%d]=%.1f\n", i, farr[i]) + i += 1 + + +def TestCharArray(): + stdio.printf("=== Test Char Array (String) ===\n") + str_arr: t.CArray[t.CChar, 32] = t.CArray[t.CChar, 32]() + str_arr[0] = 'H' + str_arr[1] = 'e' + str_arr[2] = 'l' + str_arr[3] = 'l' + str_arr[4] = 'o' + str_arr[5] = '\0' + stdio.printf("String: %s\n", str_arr) + + +def TestNestedArray(): + stdio.printf("=== Test Nested Array (Array of Arrays) ===\n") + row0: t.CArray[t.CInt, 3] = t.CArray[t.CInt, 3]() + row1: t.CArray[t.CInt, 3] = t.CArray[t.CInt, 3]() + row0[0] = 0 + row0[1] = 1 + row0[2] = 2 + row1[0] = 10 + row1[1] = 11 + row1[2] = 12 + i: t.CInt = 0 + while i < 3: + stdio.printf("row0[%d]=%d ", i, row0[i]) + i += 1 + stdio.printf("\n") + i = 0 + while i < 3: + stdio.printf("row1[%d]=%d ", i, row1[i]) + i += 1 + stdio.printf("\n") + + +def TestArrayWithStruct(): + stdio.printf("=== Test Array with Struct ===\n") + points: t.CArray[SimpleData, 3] = t.CArray[SimpleData, 3]() + i: t.CInt = 0 + while i < 3: + points[i].x = i * 10 + points[i].y = i * 100 + i += 1 + i = 0 + while i < 3: + stdio.printf("Point[%d]: x=%d, y=%d\n", i, points[i].x, points[i].y) + i += 1 + + +def TestArrayPointer(): + stdio.printf("=== Test Array Pointer ===\n") + data: t.CArray[t.CInt, 8] = t.CArray[t.CInt, 8]() + data[0] = 100 + data[1] = 200 + data[2] = 300 + ptr: t.CInt | t.CPtr = data + stdio.printf("ptr[0]=%d\n", ptr[0]) + stdio.printf("ptr[1]=%d\n", ptr[1]) + stdio.printf("ptr[2]=%d\n", ptr[2]) + + +def TestDynamicString(): + stdio.printf("=== Test Dynamic String ===\n") + name: str = "TransPyC" + stdio.printf("Hello %s!\n", name) \ No newline at end of file diff --git a/Test/TestProject2/App/zc/config.py b/Test/TestProject2/App/zc/config.py new file mode 100644 index 0000000..1b4e36f --- /dev/null +++ b/Test/TestProject2/App/zc/config.py @@ -0,0 +1,24 @@ +import t, c + + +POSMAX: t.CDefine = 100 +MAX_RETRY: t.CDefine = 5 +BUFFER_SIZE: t.CDefine = 256 +PI: t.CDefine = 3.14159 +EULER: t.CDefine = 2.71828 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +NULL: t.CDefine = 0 + + +MATH_OFFSET: t.CDefine = 10 +MATH_SCALE: t.CDefine = 2 +THRESHOLD: t.CDefine = 50 +LOOP_LIMIT: t.CDefine = 20 +SHIFT_AMOUNT: t.CDefine = 2 +MASK_VALUE: t.CDefine = 0xFF +BIT_RANGE: t.CDefine = 4 +EXPONENT_BASE: t.CDefine = 3 +MODULO_DIVISOR: t.CDefine = 7 +SCALE_SHIFT: t.CDefine = 4 +HALF_SCALE: t.CDefine = 0.5 \ No newline at end of file diff --git a/Test/TestProject2/App/zc/logic_test.py b/Test/TestProject2/App/zc/logic_test.py new file mode 100644 index 0000000..4d5c266 --- /dev/null +++ b/Test/TestProject2/App/zc/logic_test.py @@ -0,0 +1,372 @@ +import config +import t, c + + +def ForRangeBasicTest(stop: t.CInt): + i: t.CInt = 0 + for i in range(stop): + print("For range: ", i) + print("Done") + + +def ForRangeWithStartStop(start: t.CInt, stop: t.CInt, step: t.CInt): + i: t.CInt = 0 + for i in range(start, stop, step): + print("Range: ", i) + print("Done") + + +def NestedForRangeTest(rows: t.CInt, cols: t.CInt): + r: t.CInt = 0 + c_val: t.CInt = 0 + for r in range(rows): + for c_val in range(cols): + print("Cell: ", r, c_val) + print("Grid done") + + +def ForWithIfBreakTest(data_size: t.CInt): + i: t.CInt = 0 + found: t.CInt = 0 + for i in range(data_size): + if i == 5: + print("Found at: ", i) + found = 1 + break + print("Checking: ", i) + if found == 0: + print("Not found") + + +def ForWithIfContinueTest(n: t.CInt): + i: t.CInt = 0 + count: t.CInt = 0 + for i in range(n): + if i % 2 == 0: + continue + print("Odd: ", i) + count += 1 + print("Total odd: ", count) + + +def ForWithElifBranchTest(value: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + for i in range(value): + if i < 3: + result += 1 + elif i < 6: + result += 2 + elif i < 9: + result += 3 + else: + result += 4 + print("Elif branch result: ", result) + + +def WhileWithIfElifElseTest(iterations: t.CInt): + counter: t.CInt = 0 + while counter < iterations: + if counter < 2: + print("Phase 1: ", counter) + elif counter < 4: + print("Phase 2: ", counter) + elif counter < 6: + print("Phase 3: ", counter) + else: + print("Phase 4: ", counter) + counter += 1 + print("While done") + + +def WhileWithNestedIfTest(limit: t.CInt): + outer: t.CInt = 0 + while outer < limit: + inner: t.CInt = 0 + while inner < limit: + if inner == 2 and outer == 1: + print("Target: ", outer, inner) + inner += 1 + continue + if inner > 3: + break + print("Inner: ", inner) + inner += 1 + outer += 1 + + +def WhileWithMatchCaseTest(value: t.CInt): + result: t.CInt = 0 + match value: + case 1: + result = 100 + print("Case 1") + case 2: + result = 200 + print("Case 2") + case 3: + result = 300 + print("Case 3") + case _: + result = 999 + print("Default case") + print("Match result: ", result) + + +def WhileWithMatchCaseNoBreakTest(value: t.CInt): + result: t.CInt = 0 + match value: + case 1: + result = 10 + print("Case 1: ", result) + case 2: + result = 20 + print("Case 2: ", result) + result += 5 + case 3: + result = 30 + print("Case 3: ", result) + case _: + result = 0 + print("No break match result: ", result) + + +def ComplexForWhileMatchTest(outer_limit: t.CInt, inner_limit: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + result: t.CInt = 0 + for i in range(outer_limit): + j = 0 + while j < inner_limit: + match (i, j): + case (0, 0): + result += 1 + case (1, 1): + result += 10 + case (2, _): + result += 100 + case (_, 0): + result += 1000 + case _: + result += 10000 + j += 1 + print("Complex match result: ", result) + + +def ForWithCaseAndNoBreakTest(n: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + for i in range(n): + match i: + case 0: + result += 1 + case 1: + result += 2 + case 2: + result += 3 + result += 10 + case _: + result += i + c.NoBreak + print("NoBreak result: ", result) + + +def WhileWithBreakConditionTest(limit: t.CInt): + i: t.CInt = 0 + early_exit: t.CInt = 0 + while i < limit: + if i == 3: + early_exit = 1 + break + if i == 2: + i += 1 + continue + print("While i: ", i) + i += 1 + + +def NestedWhileWithMultipleBreakPoints(depth: t.CInt, width: t.CInt): + d: t.CInt = 0 + w: t.CInt = 0 + count: t.CInt = 0 + while d < depth: + w = 0 + while w < width: + if w == 2: + break + if d == 1 and w == 1: + c.Break + count += 1 + w += 1 + d += 1 + print("Nested break count: ", count) + + +def ForWithMatchCaseInLoopTest(iterations: t.CInt): + i: t.CInt = 0 + sum_even: t.CInt = 0 + sum_odd: t.CInt = 0 + for i in range(iterations): + match i % 3: + case 0: + sum_even += i + case 1: + sum_odd += i + case 2: + sum_even += i * 2 + case _: + pass + print("Sum even: ", sum_even) + print("Sum odd: ", sum_odd) + + +def ComplexLogicWithElifChains(a: t.CInt, b: t.CInt, c_val: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + while i < 10: + if i < a: + if i < b: + result += 1 + elif i < c_val: + result += 2 + else: + result += 3 + elif i < b: + if i < c_val: + result += 4 + else: + result += 5 + elif i < c_val: + result += 6 + else: + result += 7 + i += 1 + print("Complex elif chain result: ", result) + + +def ForRangeWithStepAndCondition(n: t.CInt, step: t.CInt): + i: t.CInt = 0 + count: t.CInt = 0 + for i in range(0, n, step): + if i % 2 == 0: + print("Even step: ", i) + count += 1 + else: + if i > 5: + print("Large odd: ", i) + else: + print("Small odd: ", i) + print("Total steps: ", count) + + +def MatchCaseWithGuardConditions(value: t.CInt): + result: t.CInt = 0 + match value: + case 1 if config.TRUE == 1: + result = 100 + case 2 if config.FALSE == 0: + result = 200 + case 3: + result = 300 + case _: + result = 0 + print("Guard match result: ", result) + + +def WhileWithMultipleMatchCases(iterations: t.CInt): + i: t.CInt = 0 + state: t.CInt = 0 + while i < iterations: + match state: + case 0: + if i > 3: + state = 1 + case 1: + if i > 6: + state = 2 + case 2: + if i > 9: + state = 0 + print("State at i=", i, ": ", state) + i += 1 + + +def ForWhileMixWithBreakAndContinue(n: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + count: t.CInt = 0 + for i in range(n): + j = 0 + while j < n: + if j == 2: + j += 1 + continue + if i == 3 and j == 3: + break + count += 1 + j += 1 + print("Mix break continue count: ", count) + + +def ComplexCaseMatchingWithRanges(start: t.CInt, end: t.CInt): + i: t.CInt = 0 + bucket0: t.CInt = 0 + bucket1: t.CInt = 0 + bucket2: t.CInt = 0 + bucket3: t.CInt = 0 + for i in range(start, end): + match i: + case _ if i < 10: + bucket0 += 1 + case _ if i < 20: + bucket1 += 1 + case _ if i < 30: + bucket2 += 1 + case _: + bucket3 += 1 + print("Bucket0 (<10): ", bucket0) + print("Bucket1 (10-19): ", bucket1) + print("Bucket2 (20-29): ", bucket2) + print("Bucket3 (>=30): ", bucket3) + + +def WhileWithElifInElifChain(limit: t.CInt): + i: t.CInt = 0 + result: t.CInt = 0 + while i < limit: + if i < 2: + if i == 0: + result = 1 + else: + result = 2 + elif i < 4: + if i == 2: + result = 3 + else: + result = 4 + elif i < 6: + if i == 4: + result = 5 + else: + result = 6 + else: + result = 7 + print("Chain result[", i, "]: ", result) + i += 1 + + +def ForWithCaseNoBreakNested(inner_limit: t.CInt): + i: t.CInt = 0 + j: t.CInt = 0 + result: t.CInt = 0 + for i in range(3): + for j in range(inner_limit): + match j: + case 0: + result += 1 + case 1: + result += 2 + case _: + result += 3 + c.NoBreak + print("Nested NoBreak result: ", result) \ No newline at end of file diff --git a/Test/TestProject2/App/zc/test.py b/Test/TestProject2/App/zc/test.py new file mode 100644 index 0000000..24a08a8 --- /dev/null +++ b/Test/TestProject2/App/zc/test.py @@ -0,0 +1,298 @@ +import config +import t, c + + +def AboutSizeTWhileTest(length: t.CSizeT): + pos: t.CSizeT = 0 + while pos < length: + print("Pos: ", pos) + if pos > config.POSMAX: break + pos += 1 + while pos < length: + if pos > config.POSMAX: + print("Pos: ", pos) + pos += 1 + else: + print("Pos: ", pos) + pos += 1 + if pos > config.POSMAX: break + else: + pos += 1 + print("Pos: ", pos) + while pos < length: + print("Pos: ", pos) + pos += 1 + + +def MathBasicTest(x: t.CInt, y: t.CInt) -> t.CInt: + result: t.CInt = 0 + result = x + y + print("Add: ", result) + result = x - y + print("Sub: ", result) + result = x * y + print("Mul: ", result) + if y != 0: + result = x / y + print("Div: ", result) + result = x % y if y != 0 else 0 + print("Mod: ", result) + return result + + +def MathAdvancedTest(value: t.CInt) -> t.CInt: + counter: t.CInt = 0 + scaled: t.CInt = 0 + offset: t.CInt = 0 + scaled = value * config.MATH_SCALE + print("Scaled: ", scaled) + offset = scaled + config.MATH_OFFSET + print("Offset: ", offset) + threshold: t.CInt = config.THRESHOLD + if offset > threshold: + print("Above threshold") + counter = 1 + elif offset == threshold: + print("At threshold") + counter = 0 + else: + print("Below threshold") + counter = -1 + return counter + + +def WhileWithElifElseTest(start: t.CInt, end: t.CInt): + idx: t.CInt = start + while idx < end: + if idx < 0: + print("Negative: ", idx) + elif idx == 0: + print("Zero: ", idx) + elif idx > 0 and idx < 5: + print("Small positive: ", idx) + elif idx >= 5 and idx < 10: + print("Medium positive: ", idx) + else: + print("Large: ", idx) + idx += 1 + + +def NestedWhileWithMacrosTest(outer_limit: t.CInt, inner_limit: t.CInt): + outer: t.CInt = 0 + inner: t.CInt = 0 + total_ops: t.CInt = 0 + while outer < outer_limit: + inner = 0 + while inner < inner_limit: + if inner == config.TRUE: + print("Inner loop break condition at: ", inner) + total_ops += 1 + inner += 1 + outer += 1 + print("Total operations: ", total_ops) + + +def ComplexConditionTest(a: t.CInt, b: t.CInt, c_val: t.CInt): + x: t.CInt = a + y: t.CInt = b + z: t.CInt = c_val + while x < config.LOOP_LIMIT: + if x > 0 and y > 0 and z > 0: + print("All positive: ", x, y, z) + elif x <= 0 and y <= 0 and z <= 0: + print("All non-positive: ", x, y, z) + else: + if x > 0: + print("X positive") + elif y > 0: + print("Y positive") + else: + print("Z positive") + x += 1 + y += 1 + z += 1 + + +def RetryLoopTest(retry_count: t.CInt) -> t.CInt: + attempt: t.CInt = 0 + success: t.CInt = 0 + while attempt < retry_count: + if attempt < config.MAX_RETRY: + print("Attempt: ", attempt, " less than max") + success = attempt * 2 + elif attempt == config.MAX_RETRY: + print("At max retry") + success = -1 + else: + print("Beyond max retry") + success = -2 + attempt += 1 + return success + + +def BitwiseOperationsTest(value: t.CInt) -> t.CInt: + shifted: t.CInt = 0 + masked: t.CInt = 0 + xored: t.CInt = 0 + anded: t.CInt = 0 + ored: t.CInt = 0 + shifted = value << config.SHIFT_AMOUNT + print("Left shift: ", shifted) + shifted = value >> config.SHIFT_AMOUNT + print("Right shift: ", shifted) + masked = value & config.MASK_VALUE + print("Bitwise AND mask: ", masked) + xored = value ^ config.MASK_VALUE + print("Bitwise XOR: ", xored) + anded = value & (config.MASK_VALUE >> config.SHIFT_AMOUNT) + print("Bitwise AND shifted mask: ", anded) + ored = value | config.SHIFT_AMOUNT + print("Bitwise OR shift: ", ored) + return masked + + +def CompoundAssignmentMathTest(a: t.CInt, b: t.CInt) -> t.CInt: + x: t.CInt = a + y: t.CInt = b + result: t.CInt = 0 + x += y + print("x += y: ", x) + x -= y + print("x -= y: ", x) + x *= y + print("x *= y: ", x) + x = a + y = b + if b != 0: + x /= b + print("x /= y: ", x) + x = a + if b != 0: + x %= b + print("x %%= y: ", x) + x = a + x <<= config.SHIFT_AMOUNT + print("x <<= shift: ", x) + x >>= config.SHIFT_AMOUNT + print("x >>= shift: ", x) + x &= config.MASK_VALUE + print("x &= mask: ", x) + x |= config.SHIFT_AMOUNT + print("x |= shift: ", x) + result = (a + b) * config.MATH_SCALE - config.MATH_OFFSET + print("Compound: (a+b)*scale-offset: ", result) + result = (a * b) + (config.MATH_OFFSET / 2) + print("Compound: a*b+offset/2: ", result) + return result + + +def PowerAndModuloTest(base: t.CInt, exponent: t.CInt) -> t.CInt: + result: t.CInt = 1 + temp: t.CInt = 0 + i: t.CInt = 0 + while i < exponent: + result *= base + i += 1 + print("Power: ", result) + temp = result % config.MODULO_DIVISOR + print("Power mod: ", temp) + temp = (base * config.EXPONENT_BASE) % config.MODULO_DIVISOR + print("base*exp %% divisor: ", temp) + temp = ((base + config.MATH_OFFSET) * config.MATH_SCALE) % config.MODULO_DIVISOR + print("(base+offset)*scale %% divisor: ", temp) + return result + + +def MixedArithmeticTest(x: t.CInt, y: t.CInt, z: t.CInt) -> t.CInt: + result: t.CInt = 0 + result = (x + y) * z + print("(x+y)*z: ", result) + result = x + (y * z) + print("x+(y*z): ", result) + result = ((x + y) * config.MATH_SCALE) - config.MATH_OFFSET + print("((x+y)*scale)-offset: ", result) + result = (x * config.MATH_SCALE) + (y * config.MATH_SCALE) + print("x*scale + y*scale: ", result) + result = ((x % config.MODULO_DIVISOR) + (y % config.MODULO_DIVISOR)) * config.MATH_SCALE + print("(x%%mod + y%%mod)*scale: ", result) + temp: t.CInt = 0 + temp = x | y & z + print("x | y & z: ", temp) + temp = (x | y) & z + print("(x | y) & z: ", temp) + temp = x ^ y ^ z + print("x ^ y ^ z: ", temp) + return result + + +def MathWithWhileLoopTest(start: t.CInt, count: t.CInt) -> t.CInt: + idx: t.CInt = 0 + accumulator: t.CInt = 0 + multiplier: t.CInt = config.MATH_SCALE + while idx < count: + temp: t.CInt = (start + idx) * multiplier + temp = temp + config.MATH_OFFSET + if temp > config.THRESHOLD: + temp = config.THRESHOLD + accumulator += temp + multiplier += 1 + idx += 1 + print("Accumulator: ", accumulator) + return accumulator + + +def ComplexNestedMathTest(a: t.CInt, b: t.CInt) -> t.CInt: + layer1: t.CInt = 0 + layer2: t.CInt = 0 + layer3: t.CInt = 0 + i: t.CInt = 0 + layer1 = (a + b) * config.MATH_SCALE + print("Layer1: ", layer1) + while i < config.BIT_RANGE: + layer2 += layer1 >> 1 + i += 1 + print("Layer2: ", layer2) + layer3 = layer1 + layer2 - config.MATH_OFFSET + print("Layer3: ", layer3) + layer3 = layer3 * config.MATH_SCALE / 2 if config.MATH_SCALE != 0 else layer3 + print("Layer3 scaled: ", layer3) + return layer3 + + +def ConditionalMathChainTest(value: t.CInt) -> t.CInt: + result: t.CInt = 0 + step: t.CInt = 0 + while step < config.LOOP_LIMIT: + if step < 5: + result += step * config.MATH_SCALE + print("step<5: ", result) + elif step < 10: + result -= step - config.MATH_OFFSET + print("step<10: ", result) + elif step < 15: + result += (step * config.MATH_OFFSET) / config.MATH_SCALE + print("step<15: ", result) + else: + result = result * config.EXPONENT_BASE % config.MODULO_DIVISOR + print("step>=15: ", result) + step += 1 + return result + + +def BitManipulationMathTest(value: t.CInt) -> t.CInt: + original: t.CInt = value + nibble: t.CInt = 0 + result: t.CInt = 0 + result = (value << config.SHIFT_AMOUNT) | (value >> (8 - config.SHIFT_AMOUNT)) + print("Rotate left: ", result) + result = (value >> config.SHIFT_AMOUNT) | (value << (8 - config.SHIFT_AMOUNT)) + print("Rotate right: ", result) + nibble = value & 0xF + print("Lower nibble: ", nibble) + nibble = (value >> config.BIT_RANGE) & 0xF + print("Upper nibble: ", nibble) + result = ((value & config.MASK_VALUE) + config.MATH_OFFSET) * config.MATH_SCALE + print("Masked and scaled: ", result) + result = ((original << config.SHIFT_AMOUNT) ^ original) & config.MASK_VALUE + print("XOR shifted: ", result) + return result \ No newline at end of file diff --git a/Test/TestProject2/output/0df65b8ed15664b0.deps.json b/Test/TestProject2/output/0df65b8ed15664b0.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/0df65b8ed15664b0.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/19f8024d10c828e8.deps.json b/Test/TestProject2/output/19f8024d10c828e8.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/19f8024d10c828e8.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/213348433fb01cc6.deps.json b/Test/TestProject2/output/213348433fb01cc6.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/213348433fb01cc6.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/257e074b935c33b4.deps.json b/Test/TestProject2/output/257e074b935c33b4.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/257e074b935c33b4.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/451745df21e598af.deps.json b/Test/TestProject2/output/451745df21e598af.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/451745df21e598af.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/541a1ed9b8be53c8.deps.json b/Test/TestProject2/output/541a1ed9b8be53c8.deps.json new file mode 100644 index 0000000..2e7e413 --- /dev/null +++ b/Test/TestProject2/output/541a1ed9b8be53c8.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/TestProject2/output/6ff26590374ae6fc.deps.json b/Test/TestProject2/output/6ff26590374ae6fc.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/6ff26590374ae6fc.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/8ebed83a7817fa0c.deps.json b/Test/TestProject2/output/8ebed83a7817fa0c.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/8ebed83a7817fa0c.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/90921d009fdd674c.deps.json b/Test/TestProject2/output/90921d009fdd674c.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/90921d009fdd674c.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/_shared_sym_includes.pickle b/Test/TestProject2/output/_shared_sym_includes.pickle new file mode 100644 index 0000000..ce36e58 Binary files /dev/null and b/Test/TestProject2/output/_shared_sym_includes.pickle differ diff --git a/Test/TestProject2/output/c9d54a4158f7f5a8.deps.json b/Test/TestProject2/output/c9d54a4158f7f5a8.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/c9d54a4158f7f5a8.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/output/cfc83b830141b00e.deps.json b/Test/TestProject2/output/cfc83b830141b00e.deps.json new file mode 100644 index 0000000..162aebe --- /dev/null +++ b/Test/TestProject2/output/cfc83b830141b00e.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/TestProject2/output/d541ade6afb689a5.deps.json b/Test/TestProject2/output/d541ade6afb689a5.deps.json new file mode 100644 index 0000000..0c6f101 --- /dev/null +++ b/Test/TestProject2/output/d541ade6afb689a5.deps.json @@ -0,0 +1 @@ +{"hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "zlib.pyzlib": "1c46d554b3a3f9f3", "pyzlib": "1c46d554b3a3f9f3", "zlib.zdef": "213348433fb01cc6", "zdef": "213348433fb01cc6", "binascii": "257e074b935c33b4", "atom": "271ea3decb810db2", "zlib.zinflate": "451745df21e598af", "zinflate": "451745df21e598af", "main": "541a1ed9b8be53c8", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "zlib.zchecksum": "8ebed83a7817fa0c", "zchecksum": "8ebed83a7817fa0c", "zlib.zdeflate": "90921d009fdd674c", "zdeflate": "90921d009fdd674c", "stdlib": "90c53dd6db8d41cf", "hashlib.__init__": "96837bcc64032444", "__init__": "d63d2c2a2bd9bd2a", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "zc.config": "cfc83b830141b00e", "config": "cfc83b830141b00e", "zlib.zhuff": "d541ade6afb689a5", "zhuff": "d541ade6afb689a5", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "zlib.__init__": "d63d2c2a2bd9bd2a", "zlib": "d63d2c2a2bd9bd2a"} \ No newline at end of file diff --git a/Test/TestProject2/project.json b/Test/TestProject2/project.json new file mode 100644 index 0000000..d16b2c0 --- /dev/null +++ b/Test/TestProject2/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TestProject2", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread"], + "output": "TestProject2.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} \ No newline at end of file diff --git a/Test/TestProject2/temp/0035c95a18d4f8e8.pyi b/Test/TestProject2/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/TestProject2/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/TestProject2/temp/067c78e9f121dce3.pyi b/Test/TestProject2/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/TestProject2/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/TestProject2/temp/06f53cc594b4ac6c.pyi b/Test/TestProject2/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/TestProject2/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestProject2/temp/0df65b8ed15664b0.pyi b/Test/TestProject2/temp/0df65b8ed15664b0.pyi new file mode 100644 index 0000000..e33dc69 --- /dev/null +++ b/Test/TestProject2/temp/0df65b8ed15664b0.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from hashlib.__sha1.py +Module: hashlib.__sha1 +""" + + +import string +import t, c + +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self: sha1) -> t.CInt: pass + def transform(self: sha1, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha1, s: str) -> t.CInt: pass + def final(self: sha1, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/19f8024d10c828e8.pyi b/Test/TestProject2/temp/19f8024d10c828e8.pyi new file mode 100644 index 0000000..a25a2e4 --- /dev/null +++ b/Test/TestProject2/temp/19f8024d10c828e8.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from hashlib.__md5.py +Module: hashlib.__md5 +""" + + +import string +import t, c + +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + + +md5_T: t.CExtern | t.CArray[t.CUInt32T, 64] +md5_S: t.CExtern | t.CArray[t.CInt, 64] + +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] + def __init__(self: md5) -> t.CInt: pass + def transform(self: md5, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: md5, s: str) -> t.CInt: pass + def final(self: md5, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/1c46d554b3a3f9f3.pyi b/Test/TestProject2/temp/1c46d554b3a3f9f3.pyi new file mode 100644 index 0000000..46fe651 --- /dev/null +++ b/Test/TestProject2/temp/1c46d554b3a3f9f3.pyi @@ -0,0 +1,125 @@ +""" +Auto-generated Python stub file from zlib.pyzlib.py +Module: zlib.pyzlib +""" + + +from stdint import * +import zlib.zdeflate as zdeflate +import zlib.zinflate as zinflate +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import stdlib +import string +import memhub +import t, c + +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) +DEFLATED: t.CDefine = 8 +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 +ZLIB_VERSION: t.CDefine = "1.3.2" +pyzlib_error_msg: t.CExtern | t.CArray[t.CChar, 512] +pyzlib_error_code_val: t.CExtern | t.CInt + +@t.Object +class Compress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + def compress(self: Compress, data: BYTEPTR, data_len: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Compress, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Compress) -> Compress | t.CPtr: pass + def delete(self: Compress) -> t.CInt: pass + def __del__(self: Compress) -> t.CInt: pass +@t.Object +class Decompress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + def decompress(self: Decompress, data: BYTEPTR, data_len: t.CSizeT, max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Decompress, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Decompress) -> Decompress | t.CPtr: pass + def delete(self: Decompress) -> t.CInt: pass + def unused_data(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def unconsumed_tail(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def set_error(code: int, msg: str) -> t.CInt: pass + +def clone_bytes(pool: memhub.MemManager | t.CPtr, src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def append_bytes(pool: memhub.MemManager | t.CPtr, buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: pass + +def shrink_to_fit(pool: memhub.MemManager | t.CPtr, buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def runtime_version() -> str: pass + +def get_error() -> str: pass + +def get_error_code() -> t.CInt: pass + +def clear_error() -> t.CInt: pass + +def compress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def decompress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def compressobj(pool: memhub.MemManager | t.CPtr, level: t.CInt, method: t.CInt, wbits: t.CInt, memLevel: t.CInt, strategy: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: pass + +def decompressobj(pool: memhub.MemManager | t.CPtr, wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: pass + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass diff --git a/Test/TestProject2/temp/213348433fb01cc6.pyi b/Test/TestProject2/temp/213348433fb01cc6.pyi new file mode 100644 index 0000000..6e4edce --- /dev/null +++ b/Test/TestProject2/temp/213348433fb01cc6.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from zlib.zdef.py +Module: zlib.zdef +""" + + +from stdint import * +import stddef +import string +import stdlib +import memhub +import t, c + +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 +zdeflate_len_extra_bits: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_len_base: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_dist_extra_bits: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_dist_base: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_codelen_order: t.CExtern | t.CArray[int, 19] + +@t.Object +class zbit_writer: + pool: memhub.MemManager | t.CPtr + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def __init__(self: zbit_writer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def ensure(self: zbit_writer, need: t.CSizeT) -> t.CInt: pass + def write_bits(self: zbit_writer, value: UINT, nbits: t.CInt) -> t.CInt: pass + def write_bits_rev(self: zbit_writer, code: UINT, nbits: t.CInt) -> t.CInt: pass + def stored_block(self: zbit_writer, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_zlib_header(self: zbit_writer, wbits: t.CInt, level: t.CInt) -> t.CInt: pass + def write_gzip_header(self: zbit_writer) -> t.CInt: pass + def align(self: zbit_writer) -> t.CInt: pass + def total(self: zbit_writer) -> t.CSizeT: pass + def free(self: zbit_writer) -> t.CInt: pass + def __del__(self: zbit_writer) -> t.CInt: pass +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def init(self: zbit_reader, buf: BYTE | t.CPtr, length: t.CSizeT) -> t.CInt: pass + def read_bits(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_bits_rev(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def align(self: zbit_reader) -> t.CInt: pass + +def zdef_alloc(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> VOIDPTR: pass + +def zdef_free(pool: memhub.MemManager | t.CPtr, p: VOIDPTR) -> t.CInt: pass diff --git a/Test/TestProject2/temp/257e074b935c33b4.pyi b/Test/TestProject2/temp/257e074b935c33b4.pyi new file mode 100644 index 0000000..34d2feb --- /dev/null +++ b/Test/TestProject2/temp/257e074b935c33b4.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from binascii.py +Module: binascii +""" + + +import t, c + +HEX_CHARS: t.CExtern | t.CArray[t.CChar, None] +HEX_VALS: t.CExtern | t.CArray[t.CInt, 256] +B64_CHARS: t.CExtern | t.CArray[t.CChar, None] +B64_VALS: t.CExtern | t.CArray[t.CInt, 256] +HQX_CHARS: t.CExtern | t.CArray[t.CChar, 64] +HQX_VALS: t.CExtern | t.CArray[t.CInt, 256] +UU_CHARS: t.CExtern | t.CArray[t.CChar, None] +CRC32_TABLE: t.CExtern | t.CArray[t.CUnsignedInt, 256] +CRC_HQX_TABLE: t.CExtern | t.CArray[t.CUnsignedShort, 256] + +def init_binascii() -> t.CInt: pass + +def crc32(data: str, length: t.CInt, crc: t.CUnsignedInt) -> t.CUnsignedInt: pass + +def crc_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort) -> t.CUnsignedShort: pass + +def hexlify(data: str, length: t.CInt) -> str: pass + +def b2a_hex(data: str, length: t.CInt) -> str: pass + +def a2b_hex(hexstr: str, length: t.CInt) -> str: pass + +def unhexlify(data: str, length: t.CInt) -> t.CInt: pass + +def b2a_base64(data: str, length: t.CInt) -> str: pass + +def a2b_base64(data: str, length: t.CInt) -> str: pass + +def b2a_uu(data: str, length: t.CInt) -> str: pass + +def a2b_uu(data: str) -> str: pass + +def b2a_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort) -> tuple[str, t.CUnsignedShort]: pass + +def a2b_hqx(data: str, crc: t.CUnsignedShort) -> tuple[str, t.CUnsignedShort]: pass + +def adler32(data: str, length: t.CInt, adler: t.CUnsignedInt) -> t.CUnsignedInt: pass diff --git a/Test/TestProject2/temp/271ea3decb810db2.pyi b/Test/TestProject2/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/TestProject2/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/TestProject2/temp/451745df21e598af.pyi b/Test/TestProject2/temp/451745df21e598af.pyi new file mode 100644 index 0000000..b3bfb92 --- /dev/null +++ b/Test/TestProject2/temp/451745df21e598af.pyi @@ -0,0 +1,52 @@ +""" +Auto-generated Python stub file from zlib.zinflate.py +Module: zlib.zinflate +""" + + +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import string +import memhub +import t, c + +@t.Object +class zinflate_stream: + pool: memhub.MemManager | t.CPtr + reader: zdef.zbit_reader + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + header_parsed: t.CInt + is_gzip: t.CInt + adler: t.CUInt32T + crc: t.CUInt32T + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + max_length: t.CSizeT + def __init__(self: zinflate_stream) -> t.CInt: pass + def output_byte(self: zinflate_stream, byte: BYTE) -> t.CInt: pass + def parse_header(self: zinflate_stream) -> t.CInt: pass + def read_bits_from_input(self: zinflate_stream, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_huffman(self: zinflate_stream, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: pass + def inflate_block_stored(self: zinflate_stream) -> t.CInt: pass + def inflate_block_fixed(self: zinflate_stream) -> t.CInt: pass + def inflate_block_dynamic(self: zinflate_stream) -> t.CInt: pass + def inflate_stream(self: zinflate_stream) -> t.CInt: pass + def decompress(self: zinflate_stream, data: UINT8PTR, length: t.CSizeT, max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zinflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zinflate_stream) -> zinflate_stream | t.CPtr: pass + def destroy(self: zinflate_stream) -> t.CInt: pass + +def zinflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass + +def zinflate_create(pool: memhub.MemManager | t.CPtr, wbits: t.CInt) -> zinflate_stream | t.CPtr: pass diff --git a/Test/TestProject2/temp/541a1ed9b8be53c8.doc.json b/Test/TestProject2/temp/541a1ed9b8be53c8.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject2/temp/541a1ed9b8be53c8.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject2/temp/541a1ed9b8be53c8.pyi b/Test/TestProject2/temp/541a1ed9b8be53c8.pyi new file mode 100644 index 0000000..a98f466 --- /dev/null +++ b/Test/TestProject2/temp/541a1ed9b8be53c8.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import config +import hashlib +import binascii +import zlib.pyzlib as zp +import t, c +import testcheck + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/TestProject2/temp/6446627d4f07a1b5.pyi b/Test/TestProject2/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/TestProject2/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/TestProject2/temp/6f62fe05c5ea1ceb.pyi b/Test/TestProject2/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/TestProject2/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/TestProject2/temp/6ff26590374ae6fc.pyi b/Test/TestProject2/temp/6ff26590374ae6fc.pyi new file mode 100644 index 0000000..1b1bba1 --- /dev/null +++ b/Test/TestProject2/temp/6ff26590374ae6fc.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from hashlib.__sha512.py +Module: hashlib.__sha512 +""" + + +import t, c + +SHA512_BLOCK_LEN: t.CDefine = 128 +SHA512_DIGEST_LEN: t.CDefine = 64 + +def sha512_ror(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_shr(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_Ch(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Maj(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma1(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma1(x: t.CUInt64T) -> t.CUInt64T: pass + + +sha512_K: t.CExtern | t.CArray[t.CUInt64T, 80] + +@t.Object +class sha512: + state: t.CArray[t.CUInt64T, 8] + count: t.CArray[t.CUInt64T, 2] + buf: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN] + def __init__(self: sha512) -> t.CInt: pass + def transform(self: sha512, block: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN]) -> t.CInt: pass + def update(self: sha512, s: str) -> t.CInt: pass + def final(self: sha512, out: t.CArray[t.CUInt8T, SHA512_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/72e2d5ccb7cedcf1.pyi b/Test/TestProject2/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/TestProject2/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/TestProject2/temp/7e529fe7a078cfef.pyi b/Test/TestProject2/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/TestProject2/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/TestProject2/temp/8ebed83a7817fa0c.pyi b/Test/TestProject2/temp/8ebed83a7817fa0c.pyi new file mode 100644 index 0000000..6375e88 --- /dev/null +++ b/Test/TestProject2/temp/8ebed83a7817fa0c.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from zlib.zchecksum.py +Module: zlib.zchecksum +""" + + +from stdint import * +import t, c + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: pass + + +crc32_table: t.CExtern | t.CArray[t.CUInt32T, 256] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: pass diff --git a/Test/TestProject2/temp/90921d009fdd674c.pyi b/Test/TestProject2/temp/90921d009fdd674c.pyi new file mode 100644 index 0000000..86d9bcf --- /dev/null +++ b/Test/TestProject2/temp/90921d009fdd674c.pyi @@ -0,0 +1,56 @@ +""" +Auto-generated Python stub file from zlib.zdeflate.py +Module: zlib.zdeflate +""" + + +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zhuff as zhuff +import zlib.zdef as zdef +import string +import memhub +import t, c + +@t.Object +class zdeflate_stream: + pool: memhub.MemManager | t.CPtr + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + def find_match(self: zdeflate_stream, data: BYTE | t.CPtr, data_len: t.CSizeT, pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: pass + def update_hash(self: zdeflate_stream, data: BYTE | t.CPtr, pos: t.CSizeT) -> t.CInt: pass + def write_fixed_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def encode_block_data(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + def write_dynamic_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_stored_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def flush(self: zdeflate_stream, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zdeflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zdeflate_stream) -> zdeflate_stream | t.CPtr: pass + def destroy(self: zdeflate_stream) -> t.CInt: pass + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: pass + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: pass + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: pass + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR) -> t.CInt: pass + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + +def zdeflate_create(pool: memhub.MemManager | t.CPtr, level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: pass + +def zdeflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass diff --git a/Test/TestProject2/temp/90c53dd6db8d41cf.pyi b/Test/TestProject2/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/TestProject2/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/TestProject2/temp/96837bcc64032444.pyi b/Test/TestProject2/temp/96837bcc64032444.pyi new file mode 100644 index 0000000..46e78bd --- /dev/null +++ b/Test/TestProject2/temp/96837bcc64032444.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from hashlib.__init__.py +Module: hashlib.__init__ +""" + +import t +import c + + +from .__md5 import md5, MD5_DIGEST_LEN +from .__sha1 import sha1, SHA1_DIGEST_LEN +from .__sha256 import sha256, SHA256_DIGEST_LEN +from .__sha512 import sha512, SHA512_DIGEST_LEN \ No newline at end of file diff --git a/Test/TestProject2/temp/_phase1_manifest.json b/Test/TestProject2/temp/_phase1_manifest.json new file mode 100644 index 0000000..0f76911 --- /dev/null +++ b/Test/TestProject2/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject2\\App\\main.py": {"sha1": "541a1ed9b8be53c8", "mtime": 1782266206.092594, "size": 881}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject2\\App\\zc\\config.py": {"sha1": "cfc83b830141b00e", "mtime": 1778402243.1661563, "size": 531}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\binascii.py": {"sha1": "257e074b935c33b4", "mtime": 1782266199.8074288, "size": 14937}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__init__.py": {"sha1": "96837bcc64032444", "mtime": 1777902641.540433, "size": 186}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__md5.py": {"sha1": "19f8024d10c828e8", "mtime": 1784356818.7075372, "size": 5847}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha1.py": {"sha1": "0df65b8ed15664b0", "mtime": 1784356827.656474, "size": 4516}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha256.py": {"sha1": "c9d54a4158f7f5a8", "mtime": 1782266203.9352357, "size": 5332}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha512.py": {"sha1": "6ff26590374ae6fc", "mtime": 1782266223.3380039, "size": 7162}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\pyzlib.py": {"sha1": "1c46d554b3a3f9f3", "mtime": 1783538216.7783763, "size": 26101}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\zchecksum.py": {"sha1": "8ebed83a7817fa0c", "mtime": 1782266235.9510274, "size": 4067}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\zdef.py": {"sha1": "213348433fb01cc6", "mtime": 1782827035.812386, "size": 8591}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\zdeflate.py": {"sha1": "90921d009fdd674c", "mtime": 1783538197.82992, "size": 19061}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\zhuff.py": {"sha1": "d541ade6afb689a5", "mtime": 1782827037.897873, "size": 13495}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\zlib\\zinflate.py": {"sha1": "451745df21e598af", "mtime": 1782827035.992862, "size": 18078}} \ No newline at end of file diff --git a/Test/TestProject2/temp/_sha1_map.txt b/Test/TestProject2/temp/_sha1_map.txt new file mode 100644 index 0000000..976db2f --- /dev/null +++ b/Test/TestProject2/temp/_sha1_map.txt @@ -0,0 +1,24 @@ +0df65b8ed15664b0:includes/hashlib\__sha1.py +19f8024d10c828e8:includes/hashlib\__md5.py +1c46d554b3a3f9f3:includes/zlib\pyzlib.py +213348433fb01cc6:includes/zlib\zdef.py +257e074b935c33b4:includes/binascii.py +271ea3decb810db2:includes/atom.py +451745df21e598af:includes/zlib\zinflate.py +541a1ed9b8be53c8:main.py +6f62fe05c5ea1ceb:includes/stdio.py +6ff26590374ae6fc:includes/hashlib\__sha512.py +7e529fe7a078cfef:includes/w32\win32base.py +8ebed83a7817fa0c:includes/zlib\zchecksum.py +90921d009fdd674c:includes/zlib\zdeflate.py +90c53dd6db8d41cf:includes/stdlib.py +96837bcc64032444:includes/hashlib\__init__.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c9d54a4158f7f5a8:includes/hashlib\__sha256.py +c9f4be41ca1cc2b4:includes/viperio.py +cfc83b830141b00e:zc\config.py +d541ade6afb689a5:includes/zlib\zhuff.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/TestProject2/temp/_shared_sym.json b/Test/TestProject2/temp/_shared_sym.json new file mode 100644 index 0000000..7d00629 Binary files /dev/null and b/Test/TestProject2/temp/_shared_sym.json differ diff --git a/Test/TestProject2/temp/ab6e54ba9a669f76.pyi b/Test/TestProject2/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/TestProject2/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/TestProject2/temp/bbdf3bbd4c3bc28c.pyi b/Test/TestProject2/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/TestProject2/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestProject2/temp/c9d54a4158f7f5a8.pyi b/Test/TestProject2/temp/c9d54a4158f7f5a8.pyi new file mode 100644 index 0000000..3e2d738 --- /dev/null +++ b/Test/TestProject2/temp/c9d54a4158f7f5a8.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__sha256.py +Module: hashlib.__sha256 +""" + + +import t, c + +SHA256_BLOCK_LEN: t.CDefine = 64 +SHA256_DIGEST_LEN: t.CDefine = 32 +sha256_K: t.CExtern | t.CArray[t.CUInt32T, 64] + +def s0(x: t.CUInt32T) -> t.CUInt32T: pass + +def s1(x: t.CUInt32T) -> t.CUInt32T: pass + +def S0(x: t.CUInt32T) -> t.CUInt32T: pass + +def S1(x: t.CUInt32T) -> t.CUInt32T: pass + +def Ch(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def Maj(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha256: + state: t.CArray[t.CUInt32T, 8] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA256_BLOCK_LEN] + def __init__(self: sha256) -> t.CInt: pass + def transform(self: sha256, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha256, s: str) -> t.CInt: pass + def final(self: sha256, out: t.CArray[t.CUInt8T, SHA256_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/c9f4be41ca1cc2b4.pyi b/Test/TestProject2/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/TestProject2/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/cfc83b830141b00e.doc.json b/Test/TestProject2/temp/cfc83b830141b00e.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject2/temp/cfc83b830141b00e.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject2/temp/cfc83b830141b00e.pyi b/Test/TestProject2/temp/cfc83b830141b00e.pyi new file mode 100644 index 0000000..5af877f --- /dev/null +++ b/Test/TestProject2/temp/cfc83b830141b00e.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from zc.config.py +Module: zc.config +""" + + +import t, c + +POSMAX: t.CDefine = 100 +MAX_RETRY: t.CDefine = 5 +BUFFER_SIZE: t.CDefine = 256 +PI: t.CDefine = 3.14159 +EULER: t.CDefine = 2.71828 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +NULL: t.CDefine = 0 +MATH_OFFSET: t.CDefine = 10 +MATH_SCALE: t.CDefine = 2 +THRESHOLD: t.CDefine = 50 +LOOP_LIMIT: t.CDefine = 20 +SHIFT_AMOUNT: t.CDefine = 2 +MASK_VALUE: t.CDefine = 0xFF +BIT_RANGE: t.CDefine = 4 +EXPONENT_BASE: t.CDefine = 3 +MODULO_DIVISOR: t.CDefine = 7 +SCALE_SHIFT: t.CDefine = 4 +HALF_SCALE: t.CDefine = 0.5 \ No newline at end of file diff --git a/Test/TestProject2/temp/d541ade6afb689a5.pyi b/Test/TestProject2/temp/d541ade6afb689a5.pyi new file mode 100644 index 0000000..fd4ddf5 --- /dev/null +++ b/Test/TestProject2/temp/d541ade6afb689a5.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from zlib.zhuff.py +Module: zlib.zhuff +""" + + +from stdint import * +import zlib.zdef as zdef +import memhub +import t, c + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt +@t.Object +class zhuff_tree: + pool: memhub.MemManager | t.CPtr + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + def __init__(self: zhuff_tree) -> t.CInt: pass + def build_codes(self: zhuff_tree, freqs: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def build_fixed_lit_tree(self: zhuff_tree) -> t.CInt: pass + def build_fixed_dist_tree(self: zhuff_tree) -> t.CInt: pass + def encode_symbol(self: zhuff_tree, symbol: int, writer: zdef.zbit_writer | t.CPtr) -> t.CInt: pass + def build_code_lengths(self: zhuff_tree, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def get_fixed_lit_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def get_fixed_dist_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def build_tree_from_lengths(self: zhuff_tree, lengths: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt +@t.Object +class zhuff_decode_tree: + pool: memhub.MemManager | t.CPtr + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + def __init__(self: zhuff_decode_tree) -> t.CInt: pass + def build_decode_tree(self: zhuff_decode_tree, ht: zhuff_tree | t.CPtr) -> t.CInt: pass + def decode_symbol(self: zhuff_decode_tree, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/d63d2c2a2bd9bd2a.pyi b/Test/TestProject2/temp/d63d2c2a2bd9bd2a.pyi new file mode 100644 index 0000000..cc9d917 --- /dev/null +++ b/Test/TestProject2/temp/d63d2c2a2bd9bd2a.pyi @@ -0,0 +1,10 @@ +""" +Auto-generated Python stub file from zlib.__init__.py +Module: zlib.__init__ +""" + +import t +import c + + +from .pyzlib import * \ No newline at end of file diff --git a/Test/TestProject2/temp/dd3002730623424b.pyi b/Test/TestProject2/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/TestProject2/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/TestProject2/temp/ee084e9fc6ee413a.pyi b/Test/TestProject2/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/TestProject2/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/TestProject2/temp/f5522571bcce7bcb.pyi b/Test/TestProject2/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/TestProject2/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/TestProject2/temp/f6b51804a0ba8ff0.pyi b/Test/TestProject2/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/TestProject2/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/TestProject3/App/definetest.py b/Test/TestProject3/App/definetest.py new file mode 100644 index 0000000..7808e43 --- /dev/null +++ b/Test/TestProject3/App/definetest.py @@ -0,0 +1,422 @@ +import t, c +from t import CInt, CInt32T, CUInt32T, CUInt64T, CFloat64T, CPtr, CChar, CDefine, CUInt16T +import viperlib +import stdlib +import platmacro + + +MAX_SIZE: t.CDefine = 1024 +MIN_VAL: t.CDefine = 0 +MAX_VAL: t.CDefine = 100 +MASK_BYTE: t.CDefine = 0xFF +SHIFT_BITS: t.CDefine = 8 +PI_INT: t.CDefine = 3 +ENABLED: t.CDefine = 1 +DISABLED: t.CDefine = 0 +PAGE_SIZE: t.CDefine = 4096 +BUFFER_LEN: t.CDefine = 256 + +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +DATA_SEG: t.CDefine | t.CUInt16T = 0x10 +STACK_SEG: t.CDefine | t.CUInt16T = 0x20 + + +def MAKE_FLAG(bit: CInt32T) -> t.CDefine: + return 1 << bit + +def CLAMP(x: CInt32T, lo: CInt32T, hi: CInt32T) -> t.CDefine: + if x < lo: + return lo + if x > hi: + return hi + return x + + +def test_basic_define() -> CInt: + print(MAX_SIZE) + print(MIN_VAL) + print(MAX_VAL) + print(MASK_BYTE) + print(SHIFT_BITS) + return 0 + +def test_define_arithmetic() -> CInt: + a: CInt32T = 10 + b: CInt32T = a + MAX_SIZE + print(b) + c_val: CInt32T = (a << SHIFT_BITS) & MASK_BYTE + print(c_val) + d: CInt32T = MAX_VAL - MIN_VAL + print(d) + e: CInt32T = PAGE_SIZE / 4 + print(e) + return 0 + +def test_define_conditional() -> CInt: + mode: CInt32T = ENABLED + if mode == ENABLED: + print(1) + else: + print(0) + if mode == DISABLED: + print(0) + else: + print(1) + val: CInt32T = 50 + if val > MAX_VAL: + print(1) + else: + print(0) + if val < MAX_VAL: + print(1) + else: + print(0) + return 0 + +def test_define_loop() -> CInt: + i: CInt32T = 0 + while i < MAX_VAL: + i = i + 1 + if i > 10: + break + print(i) + i = 0 + while i < MAX_SIZE: + i = i + PAGE_SIZE + if i >= PAGE_SIZE * 4: + break + print(i) + return 0 + +def test_typed_define() -> CInt: + seg: CUInt16T = CODE_SEG | DATA_SEG | STACK_SEG + print(seg) + cs: CUInt16T = CODE_SEG + print(cs) + ds: CUInt16T = DATA_SEG + print(ds) + return 0 + +def test_define_function() -> CInt: + flag0 = MAKE_FLAG(0) + print(flag0) + flag3 = MAKE_FLAG(3) + print(flag3) + flag7 = MAKE_FLAG(7) + print(flag7) + return 0 + +def test_define_clamp() -> CInt: + v1 = CLAMP(50, 0, 100) + print(v1) + v2 = CLAMP(-10, 0, 100) + print(v2) + v3 = CLAMP(150, 0, 100) + print(v3) + return 0 + +def test_define_string() -> CInt: + buf: CChar | CPtr = CPtr(stdlib.malloc(BUFFER_LEN)) + viperlib.snprintf(buf, BUFFER_LEN, "MAX_SIZE=%d PAGE=%d MASK=0x%x", MAX_SIZE, PAGE_SIZE, MASK_BYTE) + print(buf) + stdlib.free(buf) + return 0 + +def test_define_combined() -> CInt: + flags: CUInt32T = MAKE_FLAG(0) | MAKE_FLAG(1) | MAKE_FLAG(2) + print(flags) + shifted: CUInt32T = MASK_BYTE << SHIFT_BITS + print(shifted) + total: CUInt32T = PAGE_SIZE + BUFFER_LEN + print(total) + return 0 + + +def test_cif_basic() -> CInt: + if c.CIf(ENABLED): + print(1) + else: + print(0) + if c.CIf(DISABLED): + print(0) + else: + print(1) + if c.CIf(MAX_SIZE > 512): + print(1) + else: + print(0) + if c.CIf(MIN_VAL > 0): + print(0) + else: + print(1) + return 0 + +def test_cif_elif_else() -> CInt: + val: CInt32T = 2 + if c.CIf(val == 1): + print(1) + elif c.CIf(val == 2): + print(2) + elif c.CIf(val == 3): + print(3) + else: + print(0) + if c.CIf(val > 10): + print(10) + elif c.CIf(val > 5): + print(5) + else: + print(val) + return 0 + +def test_cifdef_cifndef() -> CInt: + if c.CIfdef(MAX_SIZE): + print(MAX_SIZE) + else: + print(0) + if c.CIfdef(UNDEFINED_MACRO_XYZ): + print(0) + else: + print(1) + if c.CIfndef(UNDEFINED_MACRO_XYZ): + print(1) + else: + print(0) + if c.CIfndef(MAX_SIZE): + print(0) + else: + print(1) + return 0 + +def test_cif_platform() -> CInt: + if c.CIfdef(_WIN32): + print(1) + else: + print(0) + if c.CIfdef(__x86_64__): + print(1) + else: + print(0) + if c.CIf(SIZEOF_VOID_P == 8): + print(8) + elif c.CIf(SIZEOF_VOID_P == 4): + print(4) + else: + print(0) + return 0 + +def test_cif_nested() -> CInt: + if c.CIf(ENABLED): + if c.CIf(MAX_SIZE > 512): + print(1) + else: + print(0) + if c.CIfdef(PAGE_SIZE): + print(PAGE_SIZE) + else: + print(0) + else: + print(0) + return 0 + +def test_cif_expr() -> CInt: + if c.CIf(MASK_BYTE & 0x0F): + print(1) + else: + print(0) + if c.CIf((MAX_SIZE >> SHIFT_BITS) > 0): + print(1) + else: + print(0) + if c.CIf(ENABLED and MAX_SIZE > 0): + print(1) + else: + print(0) + if c.CIf(DISABLED or MAX_SIZE > 0): + print(1) + else: + print(0) + return 0 + +def test_platmacro_platform() -> CInt: + if c.CIfdef(platmacro.IS_WINDOWS): + print(1) + else: + print(0) + if c.CIf(platmacro.IS_WINDOWS == 1): + print(1) + else: + print(0) + if c.CIf(platmacro.IS_LINUX == 0): + print(1) + else: + print(0) + if c.CIf(platmacro.IS_X86_64 == 1): + print(1) + else: + print(0) + return 0 + +def test_platmacro_sizes() -> CInt: + print(platmacro.PTR_SIZE) + print(platmacro.SIZEOF_INT) + print(platmacro.SIZEOF_LONG) + print(platmacro.SIZEOF_DOUBLE) + print(platmacro.PAGE_SIZE) + print(platmacro.CACHE_LINE_SIZE) + return 0 + +def test_platmacro_endian() -> CInt: + if c.CIf(platmacro.IS_LITTLE_ENDIAN == 1): + print(1) + else: + print(0) + if c.CIf(platmacro.IS_BIG_ENDIAN == 0): + print(1) + else: + print(0) + if c.CIf(platmacro.BYTE_ORDER_LE == 1): + print(1) + else: + print(0) + return 0 + +def test_platmacro_features() -> CInt: + if c.CIf(platmacro.HAS_SSE2 == 1): + print(1) + else: + print(0) + if c.CIf(platmacro.HAS_AVX == 1): + print(1) + else: + print(0) + if c.CIf(platmacro.HAS_NEON == 0): + print(1) + else: + print(0) + return 0 + +def test_platmacro_align() -> CInt: + print(platmacro.ALIGNOF_INT) + print(platmacro.ALIGNOF_PTR) + print(platmacro.ALIGNOF_DOUBLE) + if c.CIf(platmacro.ALIGNOF_PTR == 8): + print(1) + else: + print(0) + return 0 + +def test_cifdef_cross_module() -> CInt: + if c.CIfdef(platmacro.IS_WINDOWS): + if c.CIf(platmacro.PTR_SIZE == 8): + print(64) + else: + print(32) + else: + print(0) + if c.CIfndef(platmacro.IS_ARM64): + print(1) + else: + print(0) + if c.CIfdef(platmacro.HAS_AVX2): + print(1) + else: + print(0) + return 0 + +def test_cifdef_guard() -> CInt: + MY_HEADER_GUARD: t.CDefine = 1 + if c.CIfndef(MY_HEADER_GUARD): + print(0) + else: + print(1) + if c.CIfdef(MY_HEADER_GUARD): + print(1) + else: + print(0) + return 0 + +def test_cifdef_undefined() -> CInt: + if c.CIfndef(NONEXISTENT_MACRO_ABC): + print(1) + else: + print(0) + if c.CIfndef(THIS_IS_ALSO_NOT_DEFINED): + print(1) + else: + print(0) + if c.CIfdef(NONEXISTENT_MACRO_ABC): + print(0) + else: + print(1) + return 0 + +def test_cifdef_complex() -> CInt: + if c.CIfdef(platmacro.IS_WINDOWS) and c.CIf(platmacro.PTR_SIZE == 8): + print(1) + else: + print(0) + if c.CIfndef(platmacro.IS_ARM64) and c.CIf(platmacro.IS_X86_64 == 1): + print(1) + else: + print(0) + if c.CIfdef(platmacro.HAS_SSE2): + if c.CIfdef(platmacro.HAS_AVX): + print(2) + else: + print(1) + else: + print(0) + return 0 + + +def define_main() -> CInt: + print("=== Basic Define Test ===") + test_basic_define() + print("=== Define Arithmetic Test ===") + test_define_arithmetic() + print("=== Define Conditional Test ===") + test_define_conditional() + print("=== Define Loop Test ===") + test_define_loop() + print("=== Typed Define Test ===") + test_typed_define() + print("=== Define Function (Macro) Test ===") + test_define_function() + print("=== Define Clamp (Macro) Test ===") + test_define_clamp() + print("=== Define String Test ===") + test_define_string() + print("=== Define Combined Test ===") + test_define_combined() + print("=== CIf Basic Test ===") + test_cif_basic() + print("=== CIf Elif Else Test ===") + test_cif_elif_else() + print("=== CIfdef/CIfndef Test ===") + test_cifdef_cifndef() + print("=== CIf Platform Test ===") + test_cif_platform() + print("=== CIf Nested Test ===") + test_cif_nested() + print("=== CIf Expression Test ===") + test_cif_expr() + print("=== Platmacro Platform Test ===") + test_platmacro_platform() + print("=== Platmacro Sizes Test ===") + test_platmacro_sizes() + print("=== Platmacro Endian Test ===") + test_platmacro_endian() + print("=== Platmacro Features Test ===") + test_platmacro_features() + print("=== Platmacro Align Test ===") + test_platmacro_align() + print("=== CIfdef Cross-Module Test ===") + test_cifdef_cross_module() + print("=== CIfdef Guard Test ===") + test_cifdef_guard() + print("=== CIfdef Undefined Test ===") + test_cifdef_undefined() + print("=== CIfdef Complex Test ===") + test_cifdef_complex() + return 0 diff --git a/Test/TestProject3/App/enumtest.py b/Test/TestProject3/App/enumtest.py new file mode 100644 index 0000000..c935a48 --- /dev/null +++ b/Test/TestProject3/App/enumtest.py @@ -0,0 +1,95 @@ +import t, c +from t import CInt, CPtr, CChar, CInt32T, CExport +import viperlib +import stdlib + + +class Color(t.CEnum): + Red = 0 + Green = 1 + Blue = 2 + + +class Option(t.REnum): + class Some: + value: CInt + Empty = 1 + + +class Result(t.REnum): + class Ok: + value: CInt + class Err: + msg: t.CChar | CPtr + + +class GeoShape(t.REnum): + class GCircle: + radius: CInt + class GRect: + w: CInt + h: CInt + class GPoint: + x: CInt + y: CInt + + +def enum_main() -> CInt: + buf: CChar | CPtr = CPtr(stdlib.malloc(256)) + + c1: CInt = Color.Red + c2: CInt = Color.Green + c3: CInt = Color.Blue + viperlib.snprintf(buf, 256, "CEnum: Red=%d Green=%d Blue=%d", c1, c2, c3) + print(buf) + + opt: Option = Option.Some(42) + match opt: + case Option.Some(v): + viperlib.snprintf(buf, 256, "Option.Some: value=%d", v) + print(buf) + case Option.Empty(): + pass + + res1: Result = Result.Ok(100) + match res1: + case Result.Ok(v): + viperlib.snprintf(buf, 256, "Result.Ok: value=%d", v) + print(buf) + case Result.Err(e): + pass + + res2: Result = Result.Err("not found") + match res2: + case Result.Ok(v): + pass + case Result.Err(e): + viperlib.snprintf(buf, 256, "Result.Err: msg=%s", e) + print(buf) + + s1: GeoShape = GeoShape.GCircle(5) + match s1: + case GeoShape.GCircle(r): + viperlib.snprintf(buf, 256, "GeoShape.GCircle: radius=%d", r) + print(buf) + case _: + pass + + s2: GeoShape = GeoShape.GRect(10, 20) + match s2: + case GeoShape.GRect(w, h): + viperlib.snprintf(buf, 256, "GeoShape.GRect: w=%d h=%d", w, h) + print(buf) + case _: + pass + + s3: GeoShape = GeoShape.GPoint(3, 4) + match s3: + case GeoShape.GPoint(x, y): + viperlib.snprintf(buf, 256, "GeoShape.GPoint: x=%d y=%d", x, y) + print(buf) + case _: + pass + + stdlib.free(buf) + return 0 diff --git a/Test/TestProject3/App/fileiotest.py b/Test/TestProject3/App/fileiotest.py new file mode 100644 index 0000000..98d6e77 --- /dev/null +++ b/Test/TestProject3/App/fileiotest.py @@ -0,0 +1,395 @@ +import t, c +from t import CInt, CInt32T, CUInt32T, CPtr, CChar, CExport +from stdint import * +import stdlib +import viperlib +import w32.win32base +import w32.win32file +import w32.win32console +import w32.fileio +from w32.fileio import File, FileW, MODE, FRESULT, SEEK_SET, SEEK_CUR, SEEK_END + + +BUF_SIZE: t.CDefine = 4096 + + +def fileio_main(): + buf: bytes = stdlib.malloc(256) + write_buf: bytes = stdlib.malloc(BUF_SIZE) + read_buf: bytes = stdlib.malloc(BUF_SIZE) + bytes_written: ULONG = 0 + bytes_read: ULONG = 0 + + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + hFile: w32.win32base.HANDLE = w32.win32file.CreateFileA( + "testfile.txt", + w32.win32file.GENERIC_WRITE, + w32.win32file.FILE_SHARE_READ, + None, + w32.win32file.CREATE_ALWAYS, + w32.win32file.FILE_ATTRIBUTE_NORMAL, + None + ) + + if hFile == w32.win32base.INVALID_HANDLE_VALUE: + print("Failed to create file") + return + + viperlib.snprintf(write_buf, BUF_SIZE, "Hello from TransPyC Win32 FileIO!\nLine 2: %d + %d = %d\n", 10, 20, 30) + write_len: ULONG = 0 + i: CInt = 0 + ch: bytes = write_buf + while i < BUF_SIZE: + if CInt(c.Deref(ch)) == 0: + break + write_len = write_len + 1 + i = i + 1 + ch = ch + 1 + + result: BOOL = w32.win32file.WriteFile(hFile, write_buf, write_len, c.Addr(bytes_written), None) + if result == 0: + print("WriteFile failed") + else: + viperlib.snprintf(buf, 256, "Written %d bytes to testfile.txt", bytes_written) + print(buf) + + w32.win32file.FlushFileBuffers(hFile) + w32.win32base.CloseHandle(hFile) + + hFile = w32.win32file.CreateFileA( + "testfile.txt", + w32.win32file.GENERIC_READ, + w32.win32file.FILE_SHARE_READ, + None, + w32.win32file.OPEN_EXISTING, + w32.win32file.FILE_ATTRIBUTE_NORMAL, + None + ) + + if hFile == w32.win32base.INVALID_HANDLE_VALUE: + print("Failed to open file for reading") + return + + read_len: ULONG = 0 + result = w32.win32file.ReadFile(hFile, read_buf, BUF_SIZE - 1, c.Addr(read_len), None) + if result == 0: + print("ReadFile failed") + else: + viperlib.snprintf(buf, 256, "Read %d bytes from testfile.txt", read_len) + print(buf) + + read_ptr: bytes = read_buf + read_ptr = read_ptr + read_len + c.DerefAs(read_ptr, 0) + print("--- File content ---") + print(read_buf) + print("--- End ---") + + fileSize: ULONG = w32.win32file.GetFileSize(hFile, None) + viperlib.snprintf(buf, 256, "File size: %d bytes", fileSize) + print(buf) + + w32.win32base.CloseHandle(hFile) + + attrs: ULONG = w32.win32file.GetFileAttributesA("testfile.txt") + if attrs == w32.win32base.INVALID_HANDLE_VALUE: + print("GetFileAttributes failed") + else: + viperlib.snprintf(buf, 256, "File attributes: 0x%X", attrs) + print(buf) + + w32.win32file.DeleteFileA("testfile.txt") + print("File deleted") + + stdlib.free(write_buf) + stdlib.free(read_buf) + + print("=== OOP FileIO: with context manager ===") + fileio_with_test() + + print("=== OOP FileIO: f handle style ===") + fileio_handle_test() + + print("=== OOP FileIO: RP/WP/AP modes ===") + fileio_rw_test() + + +def fileio_with_test(): + buf: bytes = stdlib.malloc(256) + write_buf: bytes = stdlib.malloc(BUF_SIZE) + read_buf: bytes = stdlib.malloc(BUF_SIZE) + + with File("oop_with.txt", MODE.W) as f: + viperlib.snprintf(write_buf, BUF_SIZE, "with context manager test!\nLine 2: %d\n", 42) + n: LONG = f.write_str(write_buf) + if n < 0: + print("[with] write failed") + else: + viperlib.snprintf(buf, 256, "[with] wrote %d bytes", n) + print(buf) + + with File("oop_with.txt", MODE.R) as f: + n = f.read_all(read_buf, BUF_SIZE) + if n < 0: + print("[with] read failed") + else: + viperlib.snprintf(buf, 256, "[with] read %d bytes", n) + print(buf) + print("[with] content:") + print(read_buf) + + with File("oop_with.txt", MODE.R) as f: + line1_len: LONG = f.readline(buf, 256) + viperlib.snprintf(read_buf, BUF_SIZE, "[with] Line1(%d): ", line1_len) + print(read_buf) + print(buf) + line2_len: LONG = f.readline(buf, 256) + viperlib.snprintf(read_buf, BUF_SIZE, "[with] Line2(%d): ", line2_len) + print(read_buf) + print(buf) + + with File("oop_with.txt", MODE.A) as f: + f.write_str("Appended line!\n") + + with File("oop_with.txt", MODE.R) as f: + n = f.read_all(read_buf, BUF_SIZE) + print("[with] after append:") + print(read_buf) + + with File("oop_with.txt", MODE.R) as f: + sz: LONGLONG = f.size() + pos: LONG = f.tell() + viperlib.snprintf(buf, 256, "[with] Size=%lld Pos=%d", sz, pos) + print(buf) + f.seek(5, SEEK_SET) + pos = f.tell() + viperlib.snprintf(buf, 256, "[with] After seek: Pos=%d", pos) + print(buf) + + w32.win32file.DeleteFileA("oop_with.txt") + print("[with] test file deleted") + + stdlib.free(buf) + stdlib.free(write_buf) + stdlib.free(read_buf) + + +def fileio_handle_test(): + buf: bytes = stdlib.malloc(256) + write_buf: bytes = stdlib.malloc(BUF_SIZE) + read_buf: bytes = stdlib.malloc(BUF_SIZE) + + fw: File = File("oop_handle.txt", MODE.W) + viperlib.snprintf(write_buf, BUF_SIZE, "f handle style test!\nLine 2: %d\n", 99) + n: LONG = fw.write_str(write_buf) + viperlib.snprintf(buf, 256, "[handle] wrote %d bytes", n) + print(buf) + fw.flush() + fw.close() + + fr: File = File("oop_handle.txt", MODE.R) + n = fr.read_all(read_buf, BUF_SIZE) + viperlib.snprintf(buf, 256, "[handle] read %d bytes", n) + print(buf) + print("[handle] content:") + print(read_buf) + fr.close() + + fr2: File = File("oop_handle.txt", MODE.R) + line1_len: LONG = fr2.readline(buf, 256) + viperlib.snprintf(read_buf, BUF_SIZE, "[handle] Line1(%d): ", line1_len) + print(read_buf) + print(buf) + line2_len: LONG = fr2.readline(buf, 256) + viperlib.snprintf(read_buf, BUF_SIZE, "[handle] Line2(%d): ", line2_len) + print(read_buf) + print(buf) + fr2.close() + + fa: File = File("oop_handle.txt", MODE.A) + fa.write_str("Appended by handle!\n") + fa.close() + + fr3: File = File("oop_handle.txt", MODE.R) + n = fr3.read_all(read_buf, BUF_SIZE) + print("[handle] after append:") + print(read_buf) + fr3.close() + + fr4: File = File("oop_handle.txt", MODE.R) + sz: LONGLONG = fr4.size() + pos: LONG = fr4.tell() + viperlib.snprintf(buf, 256, "[handle] Size=%lld Pos=%d", sz, pos) + print(buf) + fr4.seek(5, SEEK_SET) + pos = fr4.tell() + viperlib.snprintf(buf, 256, "[handle] After seek: Pos=%d", pos) + print(buf) + fr4.close() + + w32.win32file.DeleteFileA("oop_handle.txt") + print("[handle] test file deleted") + + stdlib.free(buf) + stdlib.free(write_buf) + stdlib.free(read_buf) + + +def fileio_rw_test(): + buf: bytes = stdlib.malloc(256) + + fw: File = File("rw_test.txt", MODE.W) + fw.write_str("original content\n") + fw.close() + + frp: File = File("rw_test.txt", MODE.RP) + n: LONG = frp.read_all(buf, 256) + viperlib.snprintf(buf, 256, "[rp] read %d bytes", n) + print(buf) + + frp.seek(0, SEEK_END) + frp.write_str("appended via RP\n") + frp.seek(0, SEEK_SET) + n = frp.read_all(buf, 256) + print("[rp] after RP append:") + print(buf) + frp.close() + + fwp: File = File("rw_test.txt", MODE.WP) + fwp.write_str("truncated + new\n") + fwp.seek(0, SEEK_SET) + n = fwp.read_all(buf, 256) + print("[wp] WP (truncate+read):") + print(buf) + fwp.close() + + fap: File = File("rw_test.txt", MODE.AP) + fap.write_str("a+ appended\n") + fap.seek(0, SEEK_SET) + n = fap.read_all(buf, 256) + print("[ap] AP (append+read):") + print(buf) + fap.close() + + w32.win32file.DeleteFileA("rw_test.txt") + print("[rw] test file deleted") + + print("=== OOP FileIO: wide char (u\"...\") ===") + fileio_wide_test() + + stdlib.free(buf) + + +def fileio_wide_test(): + buf: bytes = stdlib.malloc(256) + + fw: File = File("wide_test.txt", MODE.W) + fw.write_str("wide char test\n") + fw.close() + + fw2: FileW = FileW(u"wide_test.txt", MODE.R) + n: LONG = fw2.read_all(buf, 256) + viperlib.snprintf(buf, 256, "[wide] read %d bytes", n) + print(buf) + fw2.close() + + w32.win32file.DeleteFileA("wide_test.txt") + print("[wide] test file deleted") + + stdlib.free(buf) + + print("=== OOP FileIO: Chinese filename & content (u\"...\" = WCHAR_T) ===") + fileio_chinese_test() + + +def fileio_chinese_test(): + buf: bytes = stdlib.malloc(512) + bytes_written: ULONG = 0 + + # === T1: Chinese filename (WCHAR_T) + ANSI content === + print("[chinese] T1: Chinese filename + ANSI content") + fw: FileW = FileW(u"中文文件.txt", MODE.W) + if fw.closed: + print("[chinese] T1: create failed") + else: + fw.write_str("Hello from Chinese filename!\n") + fw.close() + + fr: FileW = FileW(u"中文文件.txt", MODE.R) + n: LONG = fr.read_all(buf, 512) + viperlib.snprintf(buf, 256, "[chinese] T1: read %d bytes", n) + print(buf) + print("[chinese] T1 content:") + print(buf) + fr.close() + + w32.win32file.DeleteFileW(u"中文文件.txt") + print("[chinese] T1: deleted") + + # === T2: Chinese filename + Chinese wide content (WCHAR_T) === + print("[chinese] T2: Chinese filename + Chinese wide content") + fw2: FileW = FileW(u"中文内容.txt", MODE.W) + if fw2.closed: + print("[chinese] T2: create failed") + else: + # u"你好世界\n" = 5 WCHARs = 10 bytes (UTF-16LE) + w32.win32file.WriteFile(fw2.handle, u"你好世界\n", 10, c.Addr(bytes_written), None) + viperlib.snprintf(buf, 256, "[chinese] T2: wrote %d WCHAR bytes", bytes_written) + print(buf) + fw2.close() + + fr2: FileW = FileW(u"中文内容.txt", MODE.R) + n = fr2.read_all(buf, 512) + viperlib.snprintf(buf, 256, "[chinese] T2: read %d bytes (UTF-16LE)", n) + print(buf) + fr2.close() + + w32.win32file.DeleteFileW(u"中文内容.txt") + print("[chinese] T2: deleted") + + # === T3: Chinese filename + mixed ANSI + wide content === + print("[chinese] T3: Chinese filename + mixed content") + fw3: FileW = FileW(u"混合测试.txt", MODE.WP) + if fw3.closed: + print("[chinese] T3: create failed") + else: + fw3.write_str("ASCII line\n") + # u"中文行\n" = 4 WCHARs = 8 bytes (UTF-16LE) + w32.win32file.WriteFile(fw3.handle, u"中文行\n", 8, c.Addr(bytes_written), None) + fw3.seek(0, SEEK_SET) + n: LONG = fw3.read_all(buf, 512) + viperlib.snprintf(buf, 256, "[chinese] T3: total %d bytes", n) + print(buf) + fw3.close() + + w32.win32file.DeleteFileW(u"混合测试.txt") + print("[chinese] T3: deleted") + + # === T4: Chinese filename + Chinese content via WriteConsoleW === + print("[chinese] T4: Write wide content and print via WriteConsoleW") + fw4: FileW = FileW(u"控制台测试.txt", MODE.W) + if fw4.closed: + print("[chinese] T4: create failed") + else: + # u"你好，世界！\n" = 7 WCHARs = 14 bytes (UTF-16LE) + w32.win32file.WriteFile(fw4.handle, u"你好，世界！\n", 14, c.Addr(bytes_written), None) + viperlib.snprintf(buf, 256, "[chinese] T4: wrote %d WCHAR bytes", bytes_written) + print(buf) + fw4.close() + + fr4: FileW = FileW(u"控制台测试.txt", MODE.R) + n = fr4.read_all(buf, 512) + viperlib.snprintf(buf, 256, "[chinese] T4: read %d bytes", n) + print(buf) + # Print wide content to console using WriteConsoleW + chars_written: ULONG = 0 + hOut: w32.win32base.HANDLE = w32.win32file.GetStdHandle(w32.win32file.STD_OUTPUT_HANDLE) + w32.win32console.WriteConsoleW(hOut, u"你好，世界！\n", 7, c.Addr(chars_written), None) + fr4.close() + + w32.win32file.DeleteFileW(u"控制台测试.txt") + print("[chinese] T4: deleted") + + stdlib.free(buf) diff --git a/Test/TestProject3/App/main.py b/Test/TestProject3/App/main.py new file mode 100644 index 0000000..4423686 --- /dev/null +++ b/Test/TestProject3/App/main.py @@ -0,0 +1,519 @@ +from stdint import * +import w32.win32console +import t, c +from t import CInt, CPtr, CChar, CInt32T, CUInt64T, CFloat64T, CExport, State +import viperlib +import stdlib +import definetest +import enumtest +import mpooltest +import vectortest +import fileiotest +import testcheck + + +# === 函数泛型 === + +def add[T](a: T, b: T) -> T: + return a + b + +def add_u64[T](a: T, b: T, c: CUInt64T) -> T: + return a + b + T(c) + +def fxi[T1, T2](a: T1, b: T2) -> T1: + return a + fxi2(a, b, CUInt64T(178900)) + +def fxi2[T1, T2, T3](a: T1, b: T2, c: T3) -> T1: + return a + T1(b) + T1(c) + +def wrap_add[T](a: T, b: T) -> T: + return add(a, b) + +def double_add[T](a: T, b: T) -> T: + return add(a, b) + add(a, b) + + +# === 结构体泛型 === + +class A[T](): + def __init__(self, a: T): + self.a = T(a) + def get_a(self) -> T: + return self.a + +class Pair[T1, T2](): + def __init__(self, first: T1, second: T2): + self.first = T1(first) + self.second = T2(second) + def get_first(self) -> T1: + return self.first + def get_second(self) -> T2: + return self.second + +class Calculator[T](): + def __init__(self, init_val: T): + self.val = T(init_val) + def compute(self, other: T) -> T: + return add(self.val, other) + def double_compute(self, other: T) -> T: + return double_add(self.val, other) + +class Container[T](): + def __init__(self, v: T): + self.value = T(v) + self.flag = CInt32T(1) + def get_value(self) -> T: + return self.value + def get_flag(self) -> CInt32T: + return self.flag + + +# === OOP 继承 + 虚方法（Gargantua 风格） === + +@t.CVTable +class Shape: + def __init__(self, x: CFloat64T, y: CFloat64T): + self.x = x + self.y = y + def area(self) -> CFloat64T: + return 0.0 + def perimeter(self) -> CFloat64T: + return 0.0 + def describe(self) -> CFloat64T: + return self.area() + self.perimeter() + +@t.CVTable +class Circle(Shape): + def __init__(self, x: CFloat64T, y: CFloat64T, r: CFloat64T): + self.x = x + self.y = y + self.r = r + def area(self) -> CFloat64T: + return 3.14159265 * self.r * self.r + def perimeter(self) -> CFloat64T: + return 2.0 * 3.14159265 * self.r + def scale(self, factor: CFloat64T) -> 'Circle' | CPtr: + self.r = self.r * factor + return t.CVoid(c.Addr(self), CPtr) + def move(self, dx: CFloat64T, dy: CFloat64T) -> 'Circle' | CPtr: + self.x = self.x + dx + self.y = self.y + dy + return t.CVoid(c.Addr(self), CPtr) + +@t.CVTable +class Rect(Shape): + def __init__(self, x: CFloat64T, y: CFloat64T, w: CFloat64T, h: CFloat64T): + self.x = x + self.y = y + self.w = w + self.h = h + def area(self) -> CFloat64T: + return self.w * self.h + def perimeter(self) -> CFloat64T: + return 2.0 * (self.w + self.h) + def scale(self, factor: CFloat64T) -> 'Rect' | CPtr: + self.w = self.w * factor + self.h = self.h * factor + return t.CVoid(c.Addr(self), CPtr) + + +# === 运算符重载 + 链式调用（Gargantua V3 风格，堆分配） === + +class Vec2: + x: CFloat64T + y: CFloat64T + + def __new__() -> 'Vec2' | CPtr: + return t.CVoid(t.CUInt64T(stdlib.malloc(16)), CPtr) + + def __init__(self, x: CFloat64T, y: CFloat64T): + self.x = x + self.y = y + + def __add__(self, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: + return Vec2(self.x + b.x, self.y + b.y) + + def __sub__(self, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: + return Vec2(self.x - b.x, self.y - b.y) + + def __mul__(self, s: CFloat64T) -> 'Vec2' | CPtr: + return Vec2(self.x * s, self.y * s) + + def __neg__(self) -> 'Vec2' | CPtr: + return Vec2(-self.x, -self.y) + + def dot(self, b: 'Vec2' | CPtr) -> CFloat64T: + return self.x * b.x + self.y * b.y + + def len_sq(self) -> CFloat64T: + return self.x * self.x + self.y * self.y + + +class Vec3: + x: CFloat64T + y: CFloat64T + z: CFloat64T + + def __new__() -> 'Vec3' | CPtr: + return t.CVoid(t.CUInt64T(stdlib.malloc(24)), CPtr) + + def __init__(self, x: CFloat64T, y: CFloat64T, z: CFloat64T): + self.x = x + self.y = y + self.z = z + + def __add__(self, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: + return Vec3(self.x + b.x, self.y + b.y, self.z + b.z) + + def __sub__(self, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: + return Vec3(self.x - b.x, self.y - b.y, self.z - b.z) + + def __mul__(self, s: CFloat64T) -> 'Vec3' | CPtr: + return Vec3(self.x * s, self.y * s, self.z * s) + + def __neg__(self) -> 'Vec3' | CPtr: + return Vec3(-self.x, -self.y, -self.z) + + def dot(self, b: 'Vec3' | CPtr) -> CFloat64T: + return self.x * b.x + self.y * b.y + self.z * b.z + + def cross(self, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: + return Vec3(self.y * b.z - self.z * b.y, + self.z * b.x - self.x * b.z, + self.x * b.y - self.y * b.x) + + def len_sq(self) -> CFloat64T: + return self.dot(self) + + +# === 独立类（非继承，避免链接问题） === + +@t.CVTable +class Dog: + def __init__(self, name_val: CInt, bark_power: CInt32T): + self.name_val = name_val + self.health = CInt32T(100) + self.bark_power = bark_power + def speak(self) -> CInt: + return 1 + def bite(self) -> CInt32T: + return self.bark_power + def take_damage(self, dmg: CInt32T) -> 'Dog' | CPtr: + self.health = self.health - dmg + return t.CVoid(c.Addr(self), CPtr) + def is_alive(self) -> CInt: + if self.health > 0: + return 1 + return 0 + +@t.CVTable +class Cat: + def __init__(self, name_val: CInt, lives: CInt32T): + self.name_val = name_val + self.health = CInt32T(80) + self.lives = lives + def speak(self) -> CInt: + return 2 + def scratch(self) -> CInt32T: + return CInt32T(15) + def take_damage(self, dmg: CInt32T) -> 'Cat' | CPtr: + self.health = self.health - dmg + return t.CVoid(c.Addr(self), CPtr) + def is_alive(self) -> CInt: + if self.health > 0: + return 1 + return 0 + + +# === 嵌套对象组合（堆分配） === + +class Transform: + px: CFloat64T + py: CFloat64T + pz: CFloat64T + scale_x: CFloat64T + scale_y: CFloat64T + scale_z: CFloat64T + + def __new__() -> 'Transform' | CPtr: + return t.CVoid(t.CUInt64T(stdlib.malloc(48)), CPtr) + + def __init__(self, px: CFloat64T, py: CFloat64T, pz: CFloat64T): + self.px = px + self.py = py + self.pz = pz + self.scale_x = 1.0 + self.scale_y = 1.0 + self.scale_z = 1.0 + + def apply_scale(self, sx: CFloat64T, sy: CFloat64T, sz: CFloat64T) -> 'Transform' | CPtr: + self.scale_x = self.scale_x * sx + self.scale_y = self.scale_y * sy + self.scale_z = self.scale_z * sz + return t.CVoid(c.Addr(self), CPtr) + + def world_position(self) -> 'Vec3' | CPtr: + return Vec3(self.px * self.scale_x, self.py * self.scale_y, self.pz * self.scale_z) + + +# === 多层继承（子类只重写需要的方法） === + +@t.CVTable +class Vehicle: + def __init__(self, speed: CInt32T): + self.speed = speed + self.fuel = CInt32T(100) + def move(self) -> CInt: + self.fuel = self.fuel - CInt32T(10) + return self.speed + def is_running(self) -> CInt: + if self.fuel > 0: + return 1 + return 0 + +@t.CVTable +class Car(Vehicle): + def __init__(self, speed: CInt32T, doors: CInt32T): + self.speed = speed + self.fuel = CInt32T(100) + self.doors = doors + def honk(self) -> CInt: + return 1 + +@t.CVTable +class ElectricCar(Car): + def __init__(self, speed: CInt32T, doors: CInt32T, battery: CInt32T): + self.speed = speed + self.fuel = CInt32T(100) + self.doors = doors + self.battery = battery + def charge(self): + self.battery = self.battery + CInt32T(20) + def move(self) -> CInt: + self.battery = self.battery - CInt32T(5) + return self.speed + + +def main() -> CInt | CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("TestProject3: 泛型与OOP综合测试") + + buf: CChar | CPtr = CPtr(stdlib.malloc(256)) + + # === 基础函数泛型 === + testcheck.section("基础函数泛型") + print(add(3, 5)) + print(add(1.5, 2.5)) + print(add_u64(1, 2, CUInt64T(3000000000000000))) + print(fxi(0.2, 6)) + print(wrap_add(10, 20)) + print(double_add(5, 7)) + testcheck.ok("基础函数泛型测试完成") + + # === 基础结构体泛型 === + testcheck.section("基础结构体泛型") + a = A(100) + print(a.get_a()) + p1 = Pair(10, 20) + print(p1.get_first()) + print(p1.get_second()) + testcheck.ok("基础结构体泛型测试完成") + + # === OOP 继承测试 === + testcheck.section("OOP 继承测试") + c1 = Circle(0.0, 0.0, 5.0) + viperlib.snprintf(buf, 256, "Circle area=%.2f perim=%.2f", c1.area(), c1.perimeter()) + print(buf) + + r1 = Rect(0.0, 0.0, 4.0, 3.0) + viperlib.snprintf(buf, 256, "Rect area=%.2f perim=%.2f", r1.area(), r1.perimeter()) + print(buf) + + # 链式调用: scale -> move + c2 = Circle(1.0, 2.0, 3.0) + c2.scale(2.0).move(10.0, 20.0) + viperlib.snprintf(buf, 256, "Circle after scale+move: x=%.1f y=%.1f r=%.1f area=%.2f", + c2.x, c2.y, c2.r, c2.area()) + print(buf) + + r2 = Rect(0.0, 0.0, 10.0, 5.0) + r2.scale(2.0).scale(0.5) + viperlib.snprintf(buf, 256, "Rect after double scale: w=%.1f h=%.1f area=%.2f", + r2.w, r2.h, r2.area()) + print(buf) + testcheck.ok("OOP 继承测试完成") + + # === Vec2 运算符重载 + 链式调用（堆分配） === + testcheck.section("Vec2 运算符重载 + 链式调用") + v1 = Vec2(3.0, 4.0) + v2 = Vec2(1.0, 2.0) + v3 = v1 + v2 + viperlib.snprintf(buf, 256, "Vec2 add: (%.1f, %.1f)", v3.x, v3.y) + print(buf) + + v4 = v1 - v2 + viperlib.snprintf(buf, 256, "Vec2 sub: (%.1f, %.1f)", v4.x, v4.y) + print(buf) + + v5 = v1 * 2.0 + viperlib.snprintf(buf, 256, "Vec2 mul: (%.1f, %.1f)", v5.x, v5.y) + print(buf) + + v6 = -v1 + viperlib.snprintf(buf, 256, "Vec2 neg: (%.1f, %.1f)", v6.x, v6.y) + print(buf) + + d = v1.dot(v2) + viperlib.snprintf(buf, 256, "Vec2 dot: %.1f", d) + print(buf) + + # 链式运算: (v1 + v2) * 3.0 + v_chain = (v1 + v2) * 3.0 + viperlib.snprintf(buf, 256, "Vec2 chain (v1+v2)*3: (%.1f, %.1f)", v_chain.x, v_chain.y) + print(buf) + testcheck.ok("Vec2 运算符重载测试完成") + + # === Vec3 运算符重载 + 链式调用（堆分配） === + testcheck.section("Vec3 运算符重载 + 链式调用") + a1 = Vec3(1.0, 0.0, 0.0) + a2 = Vec3(0.0, 1.0, 0.0) + cross_v = a1.cross(a2) + viperlib.snprintf(buf, 256, "Vec3 cross: (%.1f, %.1f, %.1f)", cross_v.x, cross_v.y, cross_v.z) + print(buf) + + dot_v = a1.dot(a2) + viperlib.snprintf(buf, 256, "Vec3 dot: %.1f", dot_v) + print(buf) + + # 链式: (a + b).dot(c) + a3 = Vec3(1.0, 2.0, 3.0) + a4 = Vec3(4.0, 5.0, 6.0) + a5 = Vec3(7.0, 8.0, 9.0) + chain_dot = (a3 + a4).dot(a5) + viperlib.snprintf(buf, 256, "Vec3 chain dot: %.1f", chain_dot) + print(buf) + + # 链式: (a * 2.0 - b).cross(c) + chain_cross = (a3 * 2.0 - a4).cross(a5) + viperlib.snprintf(buf, 256, "Vec3 chain cross: (%.1f, %.1f, %.1f)", chain_cross.x, chain_cross.y, chain_cross.z) + print(buf) + + # len_sq 链式 + lsq = a3.len_sq() + viperlib.snprintf(buf, 256, "Vec3 len_sq: %.1f", lsq) + print(buf) + testcheck.ok("Vec3 运算符重载测试完成") + + # === Dog/Cat 虚方法 === + testcheck.section("Dog/Cat 虚方法") + dog = Dog(1, CInt32T(30)) + cat = Cat(2, CInt32T(9)) + + viperlib.snprintf(buf, 256, "Dog speak=%d bite=%d alive=%d", dog.speak(), dog.bite(), dog.is_alive()) + print(buf) + viperlib.snprintf(buf, 256, "Cat speak=%d scratch=%d alive=%d", cat.speak(), cat.scratch(), cat.is_alive()) + print(buf) + + dog.take_damage(CInt32T(40)).take_damage(CInt32T(30)) + viperlib.snprintf(buf, 256, "Dog after 70dmg: health=%d alive=%d", dog.health, dog.is_alive()) + print(buf) + + cat.take_damage(CInt32T(50)) + viperlib.snprintf(buf, 256, "Cat after 50dmg: health=%d alive=%d", cat.health, cat.is_alive()) + print(buf) + testcheck.ok("Dog/Cat 虚方法测试完成") + + # === Transform 嵌套组合（堆分配，链式调用） === + testcheck.section("Transform 嵌套组合") + tr = Transform(10.0, 20.0, 30.0) + tr.apply_scale(2.0, 3.0, 4.0) + pos = tr.world_position() + viperlib.snprintf(buf, 256, "Transform pos: (%.1f, %.1f, %.1f)", pos.x, pos.y, pos.z) + print(buf) + + tr2 = Transform(5.0, 10.0, 15.0) + tr2.apply_scale(2.0, 2.0, 2.0) + pos2 = tr2.world_position() + viperlib.snprintf(buf, 256, "Transform2 pos: (%.1f, %.1f, %.1f)", pos2.x, pos2.y, pos2.z) + print(buf) + testcheck.ok("Transform 嵌套组合测试完成") + + # === 多层继承 === + testcheck.section("多层继承") + ec = ElectricCar(CInt32T(120), CInt32T(4), CInt32T(80)) + viperlib.snprintf(buf, 256, "ECar speed=%d doors=%d battery=%d", ec.speed, ec.doors, ec.battery) + print(buf) + ec.charge() + viperlib.snprintf(buf, 256, "ECar after charge: battery=%d", ec.battery) + print(buf) + spd = ec.move() + viperlib.snprintf(buf, 256, "ECar move: speed=%d battery=%d", spd, ec.battery) + print(buf) + viperlib.snprintf(buf, 256, "ECar honk=%d running=%d", ec.honk(), ec.is_running()) + print(buf) + testcheck.ok("多层继承测试完成") + + # === 泛型 + OOP 组合 === + testcheck.section("泛型 + OOP 组合") + calc1 = Calculator(10) + print(calc1.compute(5)) + calc2 = Calculator(1.5) + print(calc2.compute(2.5)) + + c1_gen = Container(42) + print(c1_gen.get_value()) + print(c1_gen.get_flag()) + testcheck.ok("泛型 + OOP 组合测试完成") + + # === snprintf 格式化 === + testcheck.section("snprintf 格式化") + viperlib.snprintf(buf, 256, "int=%d, float=%.2f", 42, 3.14) + print(buf) + viperlib.snprintf(buf, 256, "result=%d", add(100, 200)) + print(buf) + testcheck.ok("snprintf 格式化测试完成") + + # === 新语法类型强转 (t1|t2)(x) === + testcheck.section("新语法类型强转 (t1|t2)(x)") + val_i64: CUInt64T = CUInt64T(42) + val_i32 = (t.CInt)(val_i64) + print(val_i32) + + ptr_raw = stdlib.malloc(8) + ptr_typed = (t.CInt | t.CPtr)(ptr_raw) + viperlib.snprintf(buf, 256, "cast ptr=%p", ptr_typed) + print(buf) + + val_f = (t.CDouble)(42) + viperlib.snprintf(buf, 256, "cast float=%.1f", val_f) + print(buf) + testcheck.ok("新语法类型强转测试完成") + + # === CDefine 宏测试 === + testcheck.section("CDefine 宏测试") + definetest.define_main() + testcheck.ok("CDefine 宏测试完成") + + # === Enum 测试 === + testcheck.section("Enum 测试") + enumtest.enum_main() + testcheck.ok("Enum 测试完成") + + # === MPool OOP 测试 === + testcheck.section("MPool OOP 测试") + mpooltest.mpool_main() + testcheck.ok("MPool OOP 测试完成") + + # === Vector 泛型测试 === + testcheck.section("Vector 泛型测试") + vectortest.vector_main() + testcheck.ok("Vector 泛型测试完成") + + # === Win32 FileIO 测试 === + testcheck.section("Win32 FileIO 测试") + fileiotest.fileio_main() + testcheck.ok("Win32 FileIO 测试完成") + + return testcheck.end() diff --git a/Test/TestProject3/App/mpooltest.py b/Test/TestProject3/App/mpooltest.py new file mode 100644 index 0000000..90012b8 --- /dev/null +++ b/Test/TestProject3/App/mpooltest.py @@ -0,0 +1,91 @@ +import t, c +from t import CInt, CPtr, CChar +import viperlib +import viperio +import stdlib +import memhub + + +def mpool_main() -> CInt: + buf: bytes = stdlib.malloc(256) + + arena: bytes = stdlib.malloc(4096) + with memhub.MemPool(arena, 4096) as pool: + p1: bytes = pool.alloc(32) + p2: bytes = pool.alloc(64) + p3: bytes = pool.alloc(128) + if p1 != None and p2 != None and p3 != None: + viperlib.snprintf(buf, 256, "MPool bump: 3 allocs OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool bump: alloc FAILED") + print(buf) + + pool.reset() + p4: bytes = pool.alloc(256) + if p4 != None: + viperlib.snprintf(buf, 256, "MPool bump: reset+alloc OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool bump: reset+alloc FAILED") + print(buf) + + stdlib.free(arena) + + arena2: bytes = stdlib.malloc(4096) + with memhub.MemSlab(arena2, 4096, 64) as spool: + b1: bytes = spool.alloc(64) + b2: bytes = spool.alloc(64) + b3: bytes = spool.alloc(64) + if b1 != None and b2 != None and b3 != None: + viperlib.snprintf(buf, 256, "MPool slab: 3 allocs OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool slab: alloc FAILED") + print(buf) + + spool.free(b2) + b4: bytes = spool.alloc(64) + if b4 != None: + viperlib.snprintf(buf, 256, "MPool slab: free+realloc OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool slab: free+realloc FAILED") + print(buf) + + stdlib.free(arena2) + + arena3: bytes = stdlib.malloc(4096) + hpool: memhub.MemPool | CPtr = memhub.MemPool(arena3, 4096) + h1: bytes = hpool.alloc(32) + h2: bytes = hpool.alloc(64) + if h1 != None and h2 != None: + viperlib.snprintf(buf, 256, "MPool manual: 2 allocs OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool manual: alloc FAILED") + print(buf) + hpool.reset() + h3: bytes = hpool.alloc(128) + if h3 != None: + viperlib.snprintf(buf, 256, "MPool manual: reset+alloc OK") + print(buf) + else: + viperlib.snprintf(buf, 256, "MPool manual: reset+alloc FAILED") + print(buf) + + stdlib.free(arena3) + + arena4: bytes = stdlib.malloc(4096) + with memhub.MemPool(arena4, 4096) as bpool: + wb: viperio.Buf | CPtr = bpool.alloc_buf(128) + if wb.data != None: + viperlib.snprintf(wb.cstr(), 128, "MPool alloc_buf: hello from Buf!") + print(wb.cstr()) + else: + viperlib.snprintf(buf, 256, "MPool alloc_buf: FAILED") + print(buf) + + stdlib.free(arena4) + stdlib.free(buf) + return 0 diff --git a/Test/TestProject3/App/vectortest.py b/Test/TestProject3/App/vectortest.py new file mode 100644 index 0000000..e39a5f4 --- /dev/null +++ b/Test/TestProject3/App/vectortest.py @@ -0,0 +1,43 @@ +import t, c +from t import CInt, CPtr, CChar, CFloat64T +import viperlib +import stdlib +import vector +import memhub + +POOL_SIZE: t.CDefine = 4096 + +def vector_main() -> CInt: + buf: str = stdlib.malloc(256) + pool_mem: t.CVoid | t.CPtr = stdlib.malloc(POOL_SIZE) + pool: memhub.MemPool | CPtr = memhub.MemPool(pool_mem, POOL_SIZE) + + vi: vector.Vector | CPtr = vector.Vector(pool, 10, CInt(0)) + vi.push(CInt(10)) + vi.push(CInt(20)) + vi.push(CInt(30)) + viperlib.snprintf(buf, 256, "Vector[int]: len=%d, [0]=%d, [1]=%d, [2]=%d", vi.len(), vi.get(0), vi.get(1), vi.get(2)) + print(buf) + + vi.set(1, CInt(99)) + viperlib.snprintf(buf, 256, "Vector[int]: set[1]=99, [1]=%d", vi.get(1)) + print(buf) + + vi.clear() + viperlib.snprintf(buf, 256, "Vector[int]: cleared, len=%d", vi.len()) + print(buf) + + vi.free() + + vd: vector.Vector | CPtr = vector.Vector(pool, 8, CFloat64T(0.0)) + vd.push(CFloat64T(1.5)) + vd.push(CFloat64T(2.7)) + vd.push(CFloat64T(3.14)) + viperlib.snprintf(buf, 256, "Vector[double]: len=%d, [0]=%.1f, [1]=%.1f, [2]=%.2f", vd.len(), vd.get(0), vd.get(1), vd.get(2)) + print(buf) + + vd.free() + + stdlib.free(pool_mem) + stdlib.free(buf) + return 0 diff --git a/Test/TestProject3/output/1b58766d38d05de3.deps.json b/Test/TestProject3/output/1b58766d38d05de3.deps.json new file mode 100644 index 0000000..c04137a --- /dev/null +++ b/Test/TestProject3/output/1b58766d38d05de3.deps.json @@ -0,0 +1 @@ +{"mpooltest": "0019c939aae7f9f1", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "atom": "271ea3decb810db2", "definetest": "64ac83614c9bdbc6", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "main": "b235d6477436c524", "vectortest": "b777c587cc21465a", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vector": "bf9813f6e3c9d351", "fileiotest": "c01b27dbcd683245", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/TestProject3/output/64ac83614c9bdbc6.deps.json b/Test/TestProject3/output/64ac83614c9bdbc6.deps.json new file mode 100644 index 0000000..55aa415 --- /dev/null +++ b/Test/TestProject3/output/64ac83614c9bdbc6.deps.json @@ -0,0 +1 @@ +{"mpooltest": "0019c939aae7f9f1", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "enumtest": "1b58766d38d05de3", "atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "main": "b235d6477436c524", "vectortest": "b777c587cc21465a", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vector": "bf9813f6e3c9d351", "fileiotest": "c01b27dbcd683245", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/TestProject3/output/b235d6477436c524.deps.json b/Test/TestProject3/output/b235d6477436c524.deps.json new file mode 100644 index 0000000..8bc99f3 --- /dev/null +++ b/Test/TestProject3/output/b235d6477436c524.deps.json @@ -0,0 +1 @@ +{"mpooltest": "0019c939aae7f9f1", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "enumtest": "1b58766d38d05de3", "atom": "271ea3decb810db2", "definetest": "64ac83614c9bdbc6", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "vectortest": "b777c587cc21465a", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vector": "bf9813f6e3c9d351", "fileiotest": "c01b27dbcd683245", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/TestProject3/output/b777c587cc21465a.deps.json b/Test/TestProject3/output/b777c587cc21465a.deps.json new file mode 100644 index 0000000..47d27f3 --- /dev/null +++ b/Test/TestProject3/output/b777c587cc21465a.deps.json @@ -0,0 +1 @@ +{"mpooltest": "0019c939aae7f9f1", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "enumtest": "1b58766d38d05de3", "atom": "271ea3decb810db2", "definetest": "64ac83614c9bdbc6", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "main": "b235d6477436c524", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vector": "bf9813f6e3c9d351", "fileiotest": "c01b27dbcd683245", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/TestProject3/output/c01b27dbcd683245.deps.json b/Test/TestProject3/output/c01b27dbcd683245.deps.json new file mode 100644 index 0000000..b74fa3d --- /dev/null +++ b/Test/TestProject3/output/c01b27dbcd683245.deps.json @@ -0,0 +1 @@ +{"mpooltest": "0019c939aae7f9f1", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "enumtest": "1b58766d38d05de3", "atom": "271ea3decb810db2", "definetest": "64ac83614c9bdbc6", "stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "main": "b235d6477436c524", "vectortest": "b777c587cc21465a", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "vector": "bf9813f6e3c9d351", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/Test/TestProject3/project.json b/Test/TestProject3/project.json new file mode 100644 index 0000000..00e51cb --- /dev/null +++ b/Test/TestProject3/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TestProject", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "TestProject.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/TestProject3/temp/0019c939aae7f9f1.doc.json b/Test/TestProject3/temp/0019c939aae7f9f1.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/0019c939aae7f9f1.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/0019c939aae7f9f1.pyi b/Test/TestProject3/temp/0019c939aae7f9f1.pyi new file mode 100644 index 0000000..7c47ef2 --- /dev/null +++ b/Test/TestProject3/temp/0019c939aae7f9f1.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from mpooltest.py +Module: mpooltest +""" + + +import t, c +from t import CInt, CPtr, CChar +import viperlib +import viperio +import stdlib +import memhub + +def mpool_main() -> CInt: pass diff --git a/Test/TestProject3/temp/0035c95a18d4f8e8.pyi b/Test/TestProject3/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/TestProject3/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/TestProject3/temp/067c78e9f121dce3.pyi b/Test/TestProject3/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/TestProject3/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/TestProject3/temp/06f53cc594b4ac6c.pyi b/Test/TestProject3/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/TestProject3/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestProject3/temp/1b58766d38d05de3.doc.json b/Test/TestProject3/temp/1b58766d38d05de3.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/1b58766d38d05de3.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/1b58766d38d05de3.pyi b/Test/TestProject3/temp/1b58766d38d05de3.pyi new file mode 100644 index 0000000..11fb572 --- /dev/null +++ b/Test/TestProject3/temp/1b58766d38d05de3.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from enumtest.py +Module: enumtest +""" + + +import t, c +from t import CInt, CPtr, CChar, CInt32T, CExport +import viperlib +import stdlib + +class Color(t.CEnum): + Red = 0 + Green = 1 + Blue = 2 +class Option(t.REnum): + class Some: + value: CInt + Empty = 1 +class Result(t.REnum): + class Ok: + value: CInt + class Err: + msg: t.CChar | CPtr +class GeoShape(t.REnum): + class GCircle: + radius: CInt + class GRect: + w: CInt + h: CInt + class GPoint: + x: CInt + y: CInt + +def enum_main() -> CInt: pass diff --git a/Test/TestProject3/temp/271ea3decb810db2.pyi b/Test/TestProject3/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/TestProject3/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/TestProject3/temp/6446627d4f07a1b5.pyi b/Test/TestProject3/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/TestProject3/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/TestProject3/temp/64ac83614c9bdbc6.doc.json b/Test/TestProject3/temp/64ac83614c9bdbc6.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/64ac83614c9bdbc6.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/64ac83614c9bdbc6.pyi b/Test/TestProject3/temp/64ac83614c9bdbc6.pyi new file mode 100644 index 0000000..60711e1 --- /dev/null +++ b/Test/TestProject3/temp/64ac83614c9bdbc6.pyi @@ -0,0 +1,85 @@ +""" +Auto-generated Python stub file from definetest.py +Module: definetest +""" + + +import t, c +from t import CInt, CInt32T, CUInt32T, CUInt64T, CFloat64T, CPtr, CChar, CDefine, CUInt16T +import viperlib +import stdlib +import platmacro + +MAX_SIZE: t.CDefine = 1024 +MIN_VAL: t.CDefine = 0 +MAX_VAL: t.CDefine = 100 +MASK_BYTE: t.CDefine = 0xFF +SHIFT_BITS: t.CDefine = 8 +PI_INT: t.CDefine = 3 +ENABLED: t.CDefine = 1 +DISABLED: t.CDefine = 0 +PAGE_SIZE: t.CDefine = 4096 +BUFFER_LEN: t.CDefine = 256 +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +DATA_SEG: t.CDefine | t.CUInt16T = 0x10 +STACK_SEG: t.CDefine | t.CUInt16T = 0x20 + +def MAKE_FLAG(bit: CInt32T) -> t.CDefine: + return 1 << bit + +def CLAMP(x: CInt32T, lo: CInt32T, hi: CInt32T) -> t.CDefine: + if x < lo: + return lo + if x > hi: + return hi + return x + +def test_basic_define() -> CInt: pass + +def test_define_arithmetic() -> CInt: pass + +def test_define_conditional() -> CInt: pass + +def test_define_loop() -> CInt: pass + +def test_typed_define() -> CInt: pass + +def test_define_function() -> CInt: pass + +def test_define_clamp() -> CInt: pass + +def test_define_string() -> CInt: pass + +def test_define_combined() -> CInt: pass + +def test_cif_basic() -> CInt: pass + +def test_cif_elif_else() -> CInt: pass + +def test_cifdef_cifndef() -> CInt: pass + +def test_cif_platform() -> CInt: pass + +def test_cif_nested() -> CInt: pass + +def test_cif_expr() -> CInt: pass + +def test_platmacro_platform() -> CInt: pass + +def test_platmacro_sizes() -> CInt: pass + +def test_platmacro_endian() -> CInt: pass + +def test_platmacro_features() -> CInt: pass + +def test_platmacro_align() -> CInt: pass + +def test_cifdef_cross_module() -> CInt: pass + +def test_cifdef_guard() -> CInt: pass + +def test_cifdef_undefined() -> CInt: pass + +def test_cifdef_complex() -> CInt: pass + +def define_main() -> CInt: pass diff --git a/Test/TestProject3/temp/6f62fe05c5ea1ceb.pyi b/Test/TestProject3/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/TestProject3/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/TestProject3/temp/71e0a3ffcb3ebfad.pyi b/Test/TestProject3/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/TestProject3/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/TestProject3/temp/72e2d5ccb7cedcf1.pyi b/Test/TestProject3/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/TestProject3/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/TestProject3/temp/7e529fe7a078cfef.pyi b/Test/TestProject3/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/TestProject3/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/TestProject3/temp/90c53dd6db8d41cf.pyi b/Test/TestProject3/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/TestProject3/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/TestProject3/temp/93c1d18e35d188d6.pyi b/Test/TestProject3/temp/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/Test/TestProject3/temp/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/Test/TestProject3/temp/_phase1_manifest.json b/Test/TestProject3/temp/_phase1_manifest.json new file mode 100644 index 0000000..eb76b7c --- /dev/null +++ b/Test/TestProject3/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\definetest.py": {"sha1": "64ac83614c9bdbc6", "mtime": 1781005527.138675, "size": 9399}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\enumtest.py": {"sha1": "1b58766d38d05de3", "mtime": 1781025043.7585423, "size": 2094}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\fileiotest.py": {"sha1": "c01b27dbcd683245", "mtime": 1781261000.2857711, "size": 12600}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\main.py": {"sha1": "b235d6477436c524", "mtime": 1782110387.1454024, "size": 15478}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\mpooltest.py": {"sha1": "0019c939aae7f9f1", "mtime": 1782828388.8033123, "size": 2789}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TestProject3\\App\\vectortest.py": {"sha1": "b777c587cc21465a", "mtime": 1782828416.8050296, "size": 1215}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\platmacro.py": {"sha1": "93c1d18e35d188d6", "mtime": 1781237231.03366, "size": 1422}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vector.py": {"sha1": "bf9813f6e3c9d351", "mtime": 1782826619.9232223, "size": 2141}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\fileio.py": {"sha1": "0035c95a18d4f8e8", "mtime": 1783747022.686592, "size": 14147}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32file.py": {"sha1": "f6b51804a0ba8ff0", "mtime": 1782531158.744144, "size": 8424}} \ No newline at end of file diff --git a/Test/TestProject3/temp/_sha1_map.txt b/Test/TestProject3/temp/_sha1_map.txt new file mode 100644 index 0000000..23df6cc --- /dev/null +++ b/Test/TestProject3/temp/_sha1_map.txt @@ -0,0 +1,22 @@ +0019c939aae7f9f1:mpooltest.py +0035c95a18d4f8e8:includes/w32\fileio.py +1b58766d38d05de3:enumtest.py +271ea3decb810db2:includes/atom.py +64ac83614c9bdbc6:definetest.py +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +93c1d18e35d188d6:includes/platmacro.py +ab6e54ba9a669f76:includes/string.py +b235d6477436c524:main.py +b777c587cc21465a:vectortest.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +bf9813f6e3c9d351:includes/vector.py +c01b27dbcd683245:fileiotest.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py +f6b51804a0ba8ff0:includes/w32\win32file.py diff --git a/Test/TestProject3/temp/_shared_sym.json b/Test/TestProject3/temp/_shared_sym.json new file mode 100644 index 0000000..6e2ef56 Binary files /dev/null and b/Test/TestProject3/temp/_shared_sym.json differ diff --git a/Test/TestProject3/temp/ab6e54ba9a669f76.pyi b/Test/TestProject3/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/TestProject3/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/TestProject3/temp/b235d6477436c524.doc.json b/Test/TestProject3/temp/b235d6477436c524.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/b235d6477436c524.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/b235d6477436c524.pyi b/Test/TestProject3/temp/b235d6477436c524.pyi new file mode 100644 index 0000000..fa0bd7d --- /dev/null +++ b/Test/TestProject3/temp/b235d6477436c524.pyi @@ -0,0 +1,131 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import w32.win32console +import t, c +from t import CInt, CPtr, CChar, CInt32T, CUInt64T, CFloat64T, CExport, State +import viperlib +import stdlib +import definetest +import enumtest +import mpooltest +import vectortest +import fileiotest +import testcheck + +def add[T](a: T, b: T) -> T: pass + +def add_u64[T](a: T, b: T, c: CUInt64T) -> T: pass + +def fxi[T1, T2](a: T1, b: T2) -> T1: pass + +def fxi2[T1, T2, T3](a: T1, b: T2, c: T3) -> T1: pass + +def wrap_add[T](a: T, b: T) -> T: pass + +def double_add[T](a: T, b: T) -> T: pass + + +class A[T]: + def __init__(self: A, a: T) -> t.CInt: pass + def get_a(self: A) -> T: pass +class Pair[T1, T2]: + def __init__(self: Pair, first: T1, second: T2) -> t.CInt: pass + def get_first(self: Pair) -> T1: pass + def get_second(self: Pair) -> T2: pass +class Calculator[T]: + def __init__(self: Calculator, init_val: T) -> t.CInt: pass + def compute(self: Calculator, other: T) -> T: pass + def double_compute(self: Calculator, other: T) -> T: pass +class Container[T]: + def __init__(self: Container, v: T) -> t.CInt: pass + def get_value(self: Container) -> T: pass + def get_flag(self: Container) -> CInt32T: pass +@t.CVTable +class Shape: + def __init__(self: Shape, x: CFloat64T, y: CFloat64T) -> t.CInt: pass + def area(self: Shape) -> CFloat64T: pass + def perimeter(self: Shape) -> CFloat64T: pass + def describe(self: Shape) -> CFloat64T: pass +@t.CVTable +class Circle(Shape): + def __init__(self: Circle, x: CFloat64T, y: CFloat64T, r: CFloat64T) -> t.CInt: pass + def area(self: Circle) -> CFloat64T: pass + def perimeter(self: Circle) -> CFloat64T: pass + def scale(self: Circle, factor: CFloat64T) -> 'Circle' | CPtr: pass + def move(self: Circle, dx: CFloat64T, dy: CFloat64T) -> 'Circle' | CPtr: pass +@t.CVTable +class Rect(Shape): + def __init__(self: Rect, x: CFloat64T, y: CFloat64T, w: CFloat64T, h: CFloat64T) -> t.CInt: pass + def area(self: Rect) -> CFloat64T: pass + def perimeter(self: Rect) -> CFloat64T: pass + def scale(self: Rect, factor: CFloat64T) -> 'Rect' | CPtr: pass +class Vec2: + x: CFloat64T + y: CFloat64T + def __new__() -> 'Vec2' | CPtr: pass + def __init__(self: Vec2, x: CFloat64T, y: CFloat64T) -> t.CInt: pass + def __add__(self: Vec2, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: pass + def __sub__(self: Vec2, b: 'Vec2' | CPtr) -> 'Vec2' | CPtr: pass + def __mul__(self: Vec2, s: CFloat64T) -> 'Vec2' | CPtr: pass + def __neg__(self: Vec2) -> 'Vec2' | CPtr: pass + def dot(self: Vec2, b: 'Vec2' | CPtr) -> CFloat64T: pass + def len_sq(self: Vec2) -> CFloat64T: pass +class Vec3: + x: CFloat64T + y: CFloat64T + z: CFloat64T + def __new__() -> 'Vec3' | CPtr: pass + def __init__(self: Vec3, x: CFloat64T, y: CFloat64T, z: CFloat64T) -> t.CInt: pass + def __add__(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def __sub__(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def __mul__(self: Vec3, s: CFloat64T) -> 'Vec3' | CPtr: pass + def __neg__(self: Vec3) -> 'Vec3' | CPtr: pass + def dot(self: Vec3, b: 'Vec3' | CPtr) -> CFloat64T: pass + def cross(self: Vec3, b: 'Vec3' | CPtr) -> 'Vec3' | CPtr: pass + def len_sq(self: Vec3) -> CFloat64T: pass +@t.CVTable +class Dog: + def __init__(self: Dog, name_val: CInt, bark_power: CInt32T) -> t.CInt: pass + def speak(self: Dog) -> CInt: pass + def bite(self: Dog) -> CInt32T: pass + def take_damage(self: Dog, dmg: CInt32T) -> 'Dog' | CPtr: pass + def is_alive(self: Dog) -> CInt: pass +@t.CVTable +class Cat: + def __init__(self: Cat, name_val: CInt, lives: CInt32T) -> t.CInt: pass + def speak(self: Cat) -> CInt: pass + def scratch(self: Cat) -> CInt32T: pass + def take_damage(self: Cat, dmg: CInt32T) -> 'Cat' | CPtr: pass + def is_alive(self: Cat) -> CInt: pass +class Transform: + px: CFloat64T + py: CFloat64T + pz: CFloat64T + scale_x: CFloat64T + scale_y: CFloat64T + scale_z: CFloat64T + def __new__() -> 'Transform' | CPtr: pass + def __init__(self: Transform, px: CFloat64T, py: CFloat64T, pz: CFloat64T) -> t.CInt: pass + def apply_scale(self: Transform, sx: CFloat64T, sy: CFloat64T, sz: CFloat64T) -> 'Transform' | CPtr: pass + def world_position(self: Transform) -> 'Vec3' | CPtr: pass +@t.CVTable +class Vehicle: + def __init__(self: Vehicle, speed: CInt32T) -> t.CInt: pass + def move(self: Vehicle) -> CInt: pass + def is_running(self: Vehicle) -> CInt: pass +@t.CVTable +class Car(Vehicle): + def __init__(self: Car, speed: CInt32T, doors: CInt32T) -> t.CInt: pass + def honk(self: Car) -> CInt: pass +@t.CVTable +class ElectricCar(Car): + def __init__(self: ElectricCar, speed: CInt32T, doors: CInt32T, battery: CInt32T) -> t.CInt: pass + def charge(self: ElectricCar) -> t.CInt: pass + def move(self: ElectricCar) -> CInt: pass + +def main() -> CInt | CExport: pass diff --git a/Test/TestProject3/temp/b777c587cc21465a.doc.json b/Test/TestProject3/temp/b777c587cc21465a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/b777c587cc21465a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/b777c587cc21465a.pyi b/Test/TestProject3/temp/b777c587cc21465a.pyi new file mode 100644 index 0000000..e70167d --- /dev/null +++ b/Test/TestProject3/temp/b777c587cc21465a.pyi @@ -0,0 +1,16 @@ +""" +Auto-generated Python stub file from vectortest.py +Module: vectortest +""" + + +import t, c +from t import CInt, CPtr, CChar, CFloat64T +import viperlib +import stdlib +import vector +import memhub + +POOL_SIZE: t.CDefine = 4096 + +def vector_main() -> CInt: pass diff --git a/Test/TestProject3/temp/bbdf3bbd4c3bc28c.pyi b/Test/TestProject3/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/TestProject3/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TestProject3/temp/bf9813f6e3c9d351.pyi b/Test/TestProject3/temp/bf9813f6e3c9d351.pyi new file mode 100644 index 0000000..00d2f9f --- /dev/null +++ b/Test/TestProject3/temp/bf9813f6e3c9d351.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from vector.py +Module: vector +""" + + +import t, c +from stdint import * +import memhub + +class Vector[T]: + data: t.CVoid | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + elem_size: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: Vector, pool: memhub.MemManager | t.CPtr, capacity: t.CSizeT, _hint: T) -> t.CInt: pass + def push(self: Vector, value: T) -> t.CInt: pass + def _grow(self: Vector) -> t.CInt: pass + def get(self: Vector, index: t.CSizeT) -> T: pass + def set(self: Vector, index: t.CSizeT, value: T) -> t.CInt: pass + def len(self: Vector) -> t.CSizeT: pass + def clear(self: Vector) -> t.CInt: pass + def free(self: Vector) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject3/temp/c01b27dbcd683245.doc.json b/Test/TestProject3/temp/c01b27dbcd683245.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/TestProject3/temp/c01b27dbcd683245.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/TestProject3/temp/c01b27dbcd683245.pyi b/Test/TestProject3/temp/c01b27dbcd683245.pyi new file mode 100644 index 0000000..fdb53f8 --- /dev/null +++ b/Test/TestProject3/temp/c01b27dbcd683245.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from fileiotest.py +Module: fileiotest +""" + + +import t, c +from t import CInt, CInt32T, CUInt32T, CPtr, CChar, CExport +from stdint import * +import stdlib +import viperlib +import w32.win32base +import w32.win32file +import w32.win32console +import w32.fileio +from w32.fileio import File, FileW, MODE, FRESULT, SEEK_SET, SEEK_CUR, SEEK_END + +BUF_SIZE: t.CDefine = 4096 + +def fileio_main() -> t.CInt: pass + +def fileio_with_test() -> t.CInt: pass + +def fileio_handle_test() -> t.CInt: pass + +def fileio_rw_test() -> t.CInt: pass + +def fileio_wide_test() -> t.CInt: pass + +def fileio_chinese_test() -> t.CInt: pass diff --git a/Test/TestProject3/temp/c3b259b4059f8668.pyi b/Test/TestProject3/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/TestProject3/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/TestProject3/temp/c9f4be41ca1cc2b4.pyi b/Test/TestProject3/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/TestProject3/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/TestProject3/temp/dd3002730623424b.pyi b/Test/TestProject3/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/TestProject3/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/TestProject3/temp/ee084e9fc6ee413a.pyi b/Test/TestProject3/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/TestProject3/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/TestProject3/temp/f5522571bcce7bcb.pyi b/Test/TestProject3/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/TestProject3/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/TestProject3/temp/f6b51804a0ba8ff0.pyi b/Test/TestProject3/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/TestProject3/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/TypeCastTest/App/main.py b/Test/TypeCastTest/App/main.py new file mode 100644 index 0000000..e7ba4cd --- /dev/null +++ b/Test/TypeCastTest/App/main.py @@ -0,0 +1,198 @@ +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +# ============================================================ +# TypeCastTest - 测试类型转换、zext/sext/trunc 行为、 +# 有符号/无符号混合运算、CDefine 常量 +# ============================================================ + +# --- Test 1: CUInt8T -> CUInt32T 零扩展 (zext) --- +def test_zext_u8_u32(): + v: t.CUInt8T = 200 # 0xC8, 如果 sign-extend 会变成 0xFFFFFFC8 + wide: t.CUInt32T = t.CUInt32T(v) + testcheck.check(wide == 200, "zext u8->u32 (200)", "zext u8->u32 (expected 200)") + + +# --- Test 2: CInt8T -> CInt32T 符号扩展 (sext) --- +def test_sext_i8_i32(): + v: t.CInt8T = -50 # 0xCE + wide: t.CInt32T = t.CInt32T(v) + testcheck.check(wide == -50, "sext i8->i32 (-50)", "sext i8->i32 (expected -50)") + + +# --- Test 3: CUInt32T -> CUInt64T 零扩展 --- +def test_zext_u32_u64(): + v: t.CUInt32T = 0x80000000 # 最高位为1 + wide: t.CUInt64T = t.CUInt64T(v) + testcheck.check(wide == 0x80000000, "zext u32->u64 (0x80000000)", "zext u32->u64 (expected 0x80000000)") + + +# --- Test 4: CInt32T -> CInt64T 符号扩展 --- +def test_sext_i32_i64(): + v: t.CInt32T = -1 # 0xFFFFFFFF + wide: t.CInt64T = t.CInt64T(v) + testcheck.check(wide == -1, "sext i32->i64 (-1)", "sext i32->i64 (expected -1)") + + +# --- Test 5: CUInt64T -> CUInt32T 截断 (trunc) --- +def test_trunc_u64_u32(): + v: t.CUInt64T = 0x123456789ABCDEF0 + narrow: t.CUInt32T = t.CUInt32T(v) + testcheck.check(narrow == 0x9ABCDEF0, "trunc u64->u32 (0x9ABCDEF0)", "trunc u64->u32 (expected 0x9ABCDEF0)") + + +# --- Test 6: 有符号/无符号混合运算 --- +def test_mixed_signed_unsigned(): + a: t.CInt = -10 + b: t.CUInt32T = 20 + # C 中有符号/无符号混合运算，有符号会转为无符号 + # -10 作为 unsigned 是 0xFFFFFFF6 = 4294967286 + # 4294967286 + 20 = 4294967306 (overflow to 10 on u32) + result: t.CUInt32T = t.CUInt32T(a) + b + stdio.printf(" mixed signed+unsigned: %u (C behavior: %u)\n", result, t.CUInt32T(-10) + 20) + testcheck.ok("mixed signed+unsigned (C behavior)") + + +# --- Test 7: CDefine 常量在运算中的行为 --- +VAL_32BIT: t.CDefine = 0x80000000 +VAL_NEG: t.CDefine = -1 +VAL_BIG: t.CDefine = 0xFFFFFFFF + +def test_cdefine_values(): + """测试 CDefine 常量值是否正确""" + v1: t.CUInt32T = t.CUInt32T(VAL_32BIT) + v2: t.CInt = VAL_NEG + v3: t.CUInt32T = t.CUInt32T(VAL_BIG) + stdio.printf(" CDefine: 0x80000000=%u, -1=%d, 0xFFFFFFFF=%u\n", v1, v2, v3) + testcheck.ok("CDefine constants") + + +# --- Test 8: 位运算与类型宽度 --- +def test_bitops_width(): + """测试位运算是否保持正确类型宽度""" + # 左移 + v: t.CUInt32T = 1 + shifted: t.CUInt32T = v << 31 # 应该是 0x80000000 + testcheck.check(shifted == 0x80000000, "bit shift (1<<31 = 0x80000000)", "bit shift (expected 0x80000000)") + + # 右移 + back: t.CUInt32T = shifted >> 31 # 应该是 1 + testcheck.check(back == 1, "bit shift (0x80000000>>31 = 1)", "bit shift (expected 1)") + + +# --- Test 9: 整数除法与取模 (无符号) --- +def test_unsigned_divmod(): + a: t.CUInt32T = 100 + b: t.CUInt32T = 7 + q: t.CUInt32T = a // b + r: t.CUInt32T = a % b + testcheck.check(q == 14 and r == 2, "unsigned div/mod (100//7=14, 100%7=2)", "unsigned div/mod (expected 14,2)") + + +# --- Test 10: 整数除法与取模 (有符号负数) --- +def test_signed_divmod_neg(): + a: t.CInt = -100 + b: t.CInt = 7 + q: t.CInt = a // b + r: t.CInt = a % b + # C 语言: -100/7 = -14, -100%7 = -2 + # Python: -100//7 = -15, -100%7 = 5 + # TransPyC 应该遵循 C 语义 + stdio.printf(" signed div/mod: -100//7=%d, -100%%7=%d (C: -14,-2)\n", q, r) + testcheck.ok("signed div/mod (C semantics)") + + +# --- Test 11: 比较运算返回值 --- +def test_comparison_types(): + a: t.CInt = 10 + b: t.CInt = 20 + c_val: t.CInt = 10 + lt: t.CInt = 1 if a < b else 0 + eq: t.CInt = 1 if a == c_val else 0 + gt: t.CInt = 1 if b > a else 0 + testcheck.check(lt == 1 and eq == 1 and gt == 1, "comparison operations", "comparison operations failed") + + +# --- Test 12: 指针与整数互转 --- +def test_ptr_int_cast(): + buf: t.CUInt32T | t.CPtr = w32.win32memory.VirtualAlloc(None, 64, 0x3000, 0x04) + if not buf: + testcheck.fail("VirtualAlloc returned NULL") + return + # 指针转整数 + addr: t.CUInt64T = t.CUInt64T(buf) + stdio.printf(" ptr->int: addr=0x%lx\n", addr) + # 整数转指针 + p2: t.CUInt32T | t.CPtr = (t.CUInt32T | t.CPtr)(t.CVoid(addr, t.CPtr)) + p2[0] = 0x12345678 + testcheck.check(buf[0] == 0x12345678, "ptr<->int roundtrip", "ptr<->int roundtrip failed") + + +# --- Test 13: CUInt64T 乘法溢出行为 --- +def test_u64_mul(): + a: t.CUInt64T = 1000 + b: t.CUInt64T = 1000 + c_val: t.CUInt64T = a * b + testcheck.check(c_val == 1000000, "u64 mul (1000*1000=1000000)", "u64 mul (expected 1000000)") + + +# --- Test 14: bool 运算 (and/or) --- +def test_bool_ops(): + a: t.CInt = 5 + b: t.CInt = 0 + c_val: t.CInt = 3 + # Python `and`/`or` 是短路求值 + r1: t.CInt = a and c_val # 5 and 3 -> 在 C 中应该是 (5 && 3) = 1 + r2: t.CInt = b or c_val # 0 or 3 -> 在 C 中应该是 (0 || 3) = 1 + stdio.printf(" bool: 5 and 3 = %d, 0 or 3 = %d (expect 1,1 in C)\n", r1, r2) + testcheck.ok("bool ops (C semantics)") + + +# --- Test 15: 条件表达式 (三元运算符) --- +def test_ternary(): + a: t.CInt = 10 + b: t.CInt = 20 + max_val: t.CInt = a if a > b else b + min_val: t.CInt = a if a < b else b + testcheck.check(max_val == 20 and min_val == 10, "ternary (max=20 min=10)", "ternary (expected 20,10)") + + +def main() -> t.CInt | t.CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("TypeCastTest: 类型转换与运算基准测试") + + testcheck.section("zext/sext/trunc") + test_zext_u8_u32() + test_sext_i8_i32() + test_zext_u32_u64() + test_sext_i32_i64() + test_trunc_u64_u32() + + testcheck.section("mixed operations") + test_mixed_signed_unsigned() + test_cdefine_values() + test_bitops_width() + + testcheck.section("div/mod") + test_unsigned_divmod() + test_signed_divmod_neg() + + testcheck.section("comparison & logic") + test_comparison_types() + test_bool_ops() + test_ternary() + + testcheck.section("pointer/int cast") + test_ptr_int_cast() + + testcheck.section("u64 mul") + test_u64_mul() + + return testcheck.end() diff --git a/Test/TypeCastTest/output/78418cf2ee722f0f.deps.json b/Test/TypeCastTest/output/78418cf2ee722f0f.deps.json new file mode 100644 index 0000000..9da11e2 --- /dev/null +++ b/Test/TypeCastTest/output/78418cf2ee722f0f.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/TypeCastTest/project.json b/Test/TypeCastTest/project.json new file mode 100644 index 0000000..ee80878 --- /dev/null +++ b/Test/TypeCastTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TypeCastTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "TypeCastTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/TypeCastTest/temp/0035c95a18d4f8e8.pyi b/Test/TypeCastTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/TypeCastTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/TypeCastTest/temp/067c78e9f121dce3.pyi b/Test/TypeCastTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/TypeCastTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/TypeCastTest/temp/06f53cc594b4ac6c.pyi b/Test/TypeCastTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/TypeCastTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TypeCastTest/temp/6446627d4f07a1b5.pyi b/Test/TypeCastTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/TypeCastTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/TypeCastTest/temp/6f62fe05c5ea1ceb.pyi b/Test/TypeCastTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/TypeCastTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/TypeCastTest/temp/72e2d5ccb7cedcf1.pyi b/Test/TypeCastTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/TypeCastTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/TypeCastTest/temp/78418cf2ee722f0f.doc.json b/Test/TypeCastTest/temp/78418cf2ee722f0f.doc.json new file mode 100644 index 0000000..04e9521 --- /dev/null +++ b/Test/TypeCastTest/temp/78418cf2ee722f0f.doc.json @@ -0,0 +1 @@ +{"test_cdefine_values": "测试 CDefine 常量值是否正确", "test_bitops_width": "测试位运算是否保持正确类型宽度"} \ No newline at end of file diff --git a/Test/TypeCastTest/temp/78418cf2ee722f0f.pyi b/Test/TypeCastTest/temp/78418cf2ee722f0f.pyi new file mode 100644 index 0000000..0e4d365 --- /dev/null +++ b/Test/TypeCastTest/temp/78418cf2ee722f0f.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import string +import w32.win32console +import w32.win32memory +import testcheck + +def test_zext_u8_u32() -> t.CInt: pass + +def test_sext_i8_i32() -> t.CInt: pass + +def test_zext_u32_u64() -> t.CInt: pass + +def test_sext_i32_i64() -> t.CInt: pass + +def test_trunc_u64_u32() -> t.CInt: pass + +def test_mixed_signed_unsigned() -> t.CInt: pass + + +VAL_32BIT: t.CDefine = 0x80000000 +VAL_NEG: t.CDefine = -1 +VAL_BIG: t.CDefine = 0xFFFFFFFF + +def test_cdefine_values() -> t.CInt: pass + +def test_bitops_width() -> t.CInt: pass + +def test_unsigned_divmod() -> t.CInt: pass + +def test_signed_divmod_neg() -> t.CInt: pass + +def test_comparison_types() -> t.CInt: pass + +def test_ptr_int_cast() -> t.CInt: pass + +def test_u64_mul() -> t.CInt: pass + +def test_bool_ops() -> t.CInt: pass + +def test_ternary() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/TypeCastTest/temp/7e529fe7a078cfef.pyi b/Test/TypeCastTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/TypeCastTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/TypeCastTest/temp/_phase1_manifest.json b/Test/TypeCastTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..6f5ac6e --- /dev/null +++ b/Test/TypeCastTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\TypeCastTest\\App\\main.py": {"sha1": "78418cf2ee722f0f", "mtime": 1782109471.5646322, "size": 6566}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/TypeCastTest/temp/_sha1_map.txt b/Test/TypeCastTest/temp/_sha1_map.txt new file mode 100644 index 0000000..cf9e2fd --- /dev/null +++ b/Test/TypeCastTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +78418cf2ee722f0f:main.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/TypeCastTest/temp/ab6e54ba9a669f76.pyi b/Test/TypeCastTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/TypeCastTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/TypeCastTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/TypeCastTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/TypeCastTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/TypeCastTest/temp/dd3002730623424b.pyi b/Test/TypeCastTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/TypeCastTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/TypeCastTest/temp/f5522571bcce7bcb.pyi b/Test/TypeCastTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/TypeCastTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/TypeCastTest/temp/f6b51804a0ba8ff0.pyi b/Test/TypeCastTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/TypeCastTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/VRandomTest/App/main.py b/Test/VRandomTest/App/main.py new file mode 100644 index 0000000..85b2113 --- /dev/null +++ b/Test/VRandomTest/App/main.py @@ -0,0 +1,128 @@ +from stdint import * +import t, c +import stdio +import vrandom +import testcheck + + +# ============================================================ +# VRandomTest - 硬件随机数生成 + 标准分布函数测试 +# ============================================================ + +def main() -> t.CInt: + testcheck.begin("VRandomTest: Hardware RNG + Distributions") + + # --- Test 1: RDRAND basic --- + testcheck.section("Test 1: RDRAND basic") + r32: t.CUInt32T = vrandom.rdrand32() + r64: t.CUInt64T = vrandom.rdrand64() + stdio.printf(" rdrand32() = 0x%08x\n", r32) + stdio.printf(" rdrand64() = 0x%016llx\n", r64) + testcheck.ok("RDRAND basic") + + # --- Test 2: randint --- + testcheck.section("Test 2: randint(1, 100)") + for _ in range(5): + v: t.CInt = vrandom.randint(1, 100) + stdio.printf(" randint(1,100) = %d\n", v) + testcheck.ok("randint range check") + + # --- Test 3: randrange --- + testcheck.section("Test 3: randrange(0, 100, 5)") + for _ in range(5): + v = vrandom.randrange(0, 100, 5) + stdio.printf(" randrange(0,100,5) = %d\n", v) + testcheck.ok("randrange step check") + + # --- Test 4: getrandbits --- + testcheck.section("Test 4: getrandbits(8)") + for _ in range(3): + b: t.CUInt64T = vrandom.getrandbits(8) + stdio.printf(" getrandbits(8) = 0x%02llx\n", b) + testcheck.ok("getrandbits width check") + + # --- Test 5: random() [0,1) --- + testcheck.section("Test 5: random()") + for _ in range(3): + f: t.CDouble = vrandom.random() + stdio.printf(" random() = %f\n", f) + testcheck.ok("random() in [0,1)") + + # --- Test 6: uniform --- + testcheck.section("Test 6: uniform(10.0, 20.0)") + for _ in range(3): + f = vrandom.uniform(10.0, 20.0) + stdio.printf(" uniform(10,20) = %f\n", f) + testcheck.ok("uniform range check") + + # --- Test 7: gauss --- + testcheck.section("Test 7: gauss(0.0, 1.0)") + for _ in range(3): + f = vrandom.gauss(0.0, 1.0) + stdio.printf(" gauss(0,1) = %f\n", f) + testcheck.ok("gauss distribution") + + # --- Test 8: normalvariate --- + testcheck.section("Test 8: normalvariate(100.0, 15.0)") + for _ in range(3): + f = vrandom.normalvariate(100.0, 15.0) + stdio.printf(" normalvariate(100,15) = %f\n", f) + testcheck.ok("normalvariate distribution") + + # --- Test 9: expovariate --- + testcheck.section("Test 9: expovariate(0.5)") + for _ in range(3): + f = vrandom.expovariate(0.5) + stdio.printf(" expovariate(0.5) = %f\n", f) + testcheck.ok("expovariate distribution") + + # --- Test 10: gammavariate --- + testcheck.section("Test 10: gammavariate(2.0, 1.0)") + for _ in range(3): + f = vrandom.gammavariate(2.0, 1.0) + stdio.printf(" gammavariate(2,1) = %f\n", f) + testcheck.ok("gammavariate distribution") + + # --- Test 11: betavariate --- + testcheck.section("Test 11: betavariate(2.0, 5.0)") + for _ in range(3): + f = vrandom.betavariate(2.0, 5.0) + stdio.printf(" betavariate(2,5) = %f\n", f) + testcheck.ok("betavariate distribution") + + # --- Test 12: triangular --- + testcheck.section("Test 12: triangular(0.0, 10.0, 5.0)") + for _ in range(3): + f = vrandom.triangular(0.0, 10.0, 5.0) + stdio.printf(" triangular(0,10,5) = %f\n", f) + testcheck.ok("triangular distribution") + + # --- Test 13: lognormvariate --- + testcheck.section("Test 13: lognormvariate(0.0, 1.0)") + for _ in range(3): + f = vrandom.lognormvariate(0.0, 1.0) + stdio.printf(" lognormvariate(0,1) = %f\n", f) + testcheck.ok("lognormvariate distribution") + + # --- Test 14: paretovariate --- + testcheck.section("Test 14: paretovariate(1.0)") + for _ in range(3): + f = vrandom.paretovariate(1.0) + stdio.printf(" paretovariate(1) = %f\n", f) + testcheck.ok("paretovariate distribution") + + # --- Test 15: weibullvariate --- + testcheck.section("Test 15: weibullvariate(1.0, 1.5)") + for _ in range(3): + f = vrandom.weibullvariate(1.0, 1.5) + stdio.printf(" weibullvariate(1,1.5) = %f\n", f) + testcheck.ok("weibullvariate distribution") + + # --- Test 16: vonmisesvariate --- + testcheck.section("Test 16: vonmisesvariate(0.0, 1.0)") + for _ in range(3): + f = vrandom.vonmisesvariate(0.0, 1.0) + stdio.printf(" vonmisesvariate(0,1) = %f\n", f) + testcheck.ok("vonmisesvariate distribution") + + return testcheck.end() diff --git a/Test/VRandomTest/output/e24a7f4a875f4518.deps.json b/Test/VRandomTest/output/e24a7f4a875f4518.deps.json new file mode 100644 index 0000000..b129803 --- /dev/null +++ b/Test/VRandomTest/output/e24a7f4a875f4518.deps.json @@ -0,0 +1 @@ +{"vipermath": "3f7c5e78d8652535", "vrandom": "62cc01c2bb770ca3", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/VRandomTest/project.json b/Test/VRandomTest/project.json new file mode 100644 index 0000000..c0e7428 --- /dev/null +++ b/Test/VRandomTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "VRandomTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "VRandomTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/VRandomTest/temp/3f7c5e78d8652535.pyi b/Test/VRandomTest/temp/3f7c5e78d8652535.pyi new file mode 100644 index 0000000..abe01c2 --- /dev/null +++ b/Test/VRandomTest/temp/3f7c5e78d8652535.pyi @@ -0,0 +1,128 @@ +""" +Auto-generated Python stub file from vipermath.py +Module: vipermath +""" + + +import t, c + +U_M_PI: t.CDefine = 3.14159265358979323846 +U_M_E: t.CDefine = 2.71828182845904523536 +U_M_PI_2: t.CDefine = 1.57079632679489661923 +U_M_PI_4: t.CDefine = 0.78539816339744830962 +U_M_1_PI: t.CDefine = 0.31830988618379067154 +U_M_2_PI: t.CDefine = 0.63661977236758134308 +U_M_LN2: t.CDefine = 0.69314718055994530942 +U_M_LN10: t.CDefine = 2.30258509299404568402 +U_M_2_SQRT_PI: t.CDefine = 1.77245385090551602730 + +def radians(degrees: t.CDouble) -> t.CDouble: pass + +def degrees(radians: t.CDouble) -> t.CDouble: pass + +def sin(x: t.CDouble) -> t.CDouble: pass + +def cos(x: t.CDouble) -> t.CDouble: pass + +def tan(x: t.CDouble) -> t.CDouble: pass + +def asin(x: t.CDouble) -> t.CDouble: pass + +def acos(x: t.CDouble) -> t.CDouble: pass + +def _atan_core(x: t.CDouble) -> t.CDouble: pass + +def atan(x: t.CDouble) -> t.CDouble: pass + +def atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble: pass + +def sinh(x: t.CDouble) -> t.CDouble: pass + +def cosh(x: t.CDouble) -> t.CDouble: pass + +def tanh(x: t.CDouble) -> t.CDouble: pass + +def exp(x: t.CDouble) -> t.CDouble: pass + +def log(x: t.CDouble) -> t.CDouble: pass + +def sqrt(x: t.CDouble) -> t.CDouble: pass + +def abs(x: t.CInt) -> t.CInt: pass + +def fabs(x: t.CDouble) -> t.CDouble: pass + +def labs(x: t.CLong) -> t.CLong: pass + +def factorial(n: t.CInt) -> t.CDouble: pass + +def combination(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def permutation(n: t.CInt, k: t.CInt) -> t.CDouble: pass + +def pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def powf(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def cbrt(x: t.CDouble) -> t.CDouble: pass + +def hypot(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def floor(x: t.CDouble) -> t.CDouble: pass + +def ceil(x: t.CDouble) -> t.CDouble: pass + +def round(x: t.CDouble) -> t.CDouble: pass + +def trunc(x: t.CDouble) -> t.CDouble: pass + +def fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble: pass + +def fmodf(x: float, y: float) -> float: pass + +def modf(x: t.CDouble, iptr: t.CDouble | t.CPtr) -> t.CDouble: pass + +def log10(x: t.CDouble) -> t.CDouble: pass + +def log2(x: t.CDouble) -> t.CDouble: pass + +def exp2(x: t.CDouble) -> t.CDouble: pass + +def expm1(x: t.CDouble) -> t.CDouble: pass + +def log1p(x: t.CDouble) -> t.CDouble: pass + +def asinh(x: t.CDouble) -> t.CDouble: pass + +def acosh(x: t.CDouble) -> t.CDouble: pass + +def atanh(x: t.CDouble) -> t.CDouble: pass + +def gamma(x: t.CDouble) -> t.CDouble: pass + +def erf(x: t.CDouble) -> t.CDouble: pass + +def erfc(x: t.CDouble) -> t.CDouble: pass + +def sqrtf(x: t.CFloat) -> t.CFloat: pass + +def sinf(x: t.CFloat) -> t.CFloat: pass + +def cosf(x: t.CFloat) -> t.CFloat: pass + +def tanf(x: t.CFloat) -> t.CFloat: pass + +def fabsf(x: t.CFloat) -> t.CFloat: pass + +def floorf(x: t.CFloat) -> t.CFloat: pass + +def ceilf(x: t.CFloat) -> t.CFloat: pass + + +class _U(t.CUnion): + d: t.CDouble + u: t.CUInt64T + +def isnan(x: t.CDouble) -> t.CStatic | t.CInt: pass + +def isinf(x: t.CDouble) -> t.CStatic | t.CInt: pass diff --git a/Test/VRandomTest/temp/62cc01c2bb770ca3.pyi b/Test/VRandomTest/temp/62cc01c2bb770ca3.pyi new file mode 100644 index 0000000..17b1307 --- /dev/null +++ b/Test/VRandomTest/temp/62cc01c2bb770ca3.pyi @@ -0,0 +1,77 @@ +""" +Auto-generated Python stub file from vrandom.py +Module: vrandom +""" + + +import t, c +import vipermath + +RDRAND_RETRY_LIMIT: t.CDefine = 10 +RDSEED_RETRY_LIMIT: t.CDefine = 100 + +def rdrand16() -> t.CUInt16T: pass + +def rdrand32() -> t.CUInt32T: pass + +def rdrand64() -> t.CUInt64T: pass + +def rdseed16() -> t.CUInt16T: pass + +def rdseed32() -> t.CUInt32T: pass + +def rdseed64() -> t.CUInt64T: pass + +def rdrand16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: pass + +def rdrand32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: pass + +def rdrand64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: pass + +def rdseed16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: pass + +def rdseed32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: pass + +def rdseed64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: pass + +def random_u16() -> t.CUInt16T: pass + +def random_u32() -> t.CUInt32T: pass + +def random_u64() -> t.CUInt64T: pass + +def seed_u16() -> t.CUInt16T: pass + +def seed_u32() -> t.CUInt32T: pass + +def seed_u64() -> t.CUInt64T: pass + +def randint(a: t.CInt, b: t.CInt) -> t.CInt: pass + +def randrange(start: t.CInt, stop: t.CInt, step: t.CInt) -> t.CInt: pass + +def getrandbits(k: t.CInt) -> t.CUInt64T: pass + +def random() -> t.CDouble: pass + +def uniform(a: t.CDouble, b: t.CDouble) -> t.CDouble: pass + +def triangular(low: t.CDouble, high: t.CDouble, mode: t.CDouble) -> t.CDouble: pass + +def betavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass + +def expovariate(lambd: t.CDouble) -> t.CDouble: pass + +def gammavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass + +def gauss(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def normalvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def lognormvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: pass + +def vonmisesvariate(mu: t.CDouble, kappa: t.CDouble) -> t.CDouble: pass + +def paretovariate(alpha: t.CDouble) -> t.CDouble: pass + +def weibullvariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: pass diff --git a/Test/VRandomTest/temp/6f62fe05c5ea1ceb.pyi b/Test/VRandomTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/VRandomTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/VRandomTest/temp/7e529fe7a078cfef.pyi b/Test/VRandomTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/VRandomTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/VRandomTest/temp/_phase1_manifest.json b/Test/VRandomTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..04412a6 --- /dev/null +++ b/Test/VRandomTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\VRandomTest\\App\\main.py": {"sha1": "e24a7f4a875f4518", "mtime": 1782109440.0645702, "size": 4476}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vipermath.py": {"sha1": "3f7c5e78d8652535", "mtime": 1781532528.2518907, "size": 15412}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\vrandom.py": {"sha1": "62cc01c2bb770ca3", "mtime": 1782101201.2607212, "size": 12067}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/VRandomTest/temp/_sha1_map.txt b/Test/VRandomTest/temp/_sha1_map.txt new file mode 100644 index 0000000..be3eecf --- /dev/null +++ b/Test/VRandomTest/temp/_sha1_map.txt @@ -0,0 +1,8 @@ +3f7c5e78d8652535:includes/vipermath.py +62cc01c2bb770ca3:includes/vrandom.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +dd3002730623424b:includes/testcheck.py +e24a7f4a875f4518:main.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/VRandomTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/VRandomTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/VRandomTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/VRandomTest/temp/dd3002730623424b.pyi b/Test/VRandomTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/VRandomTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/VRandomTest/temp/e24a7f4a875f4518.doc.json b/Test/VRandomTest/temp/e24a7f4a875f4518.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/VRandomTest/temp/e24a7f4a875f4518.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/VRandomTest/temp/e24a7f4a875f4518.pyi b/Test/VRandomTest/temp/e24a7f4a875f4518.pyi new file mode 100644 index 0000000..2825744 --- /dev/null +++ b/Test/VRandomTest/temp/e24a7f4a875f4518.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import vrandom +import testcheck + +def main() -> t.CInt: pass diff --git a/Test/VRandomTest/temp/f5522571bcce7bcb.pyi b/Test/VRandomTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/VRandomTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/VariadicTest/App/main.py b/Test/VariadicTest/App/main.py new file mode 100644 index 0000000..b4c1da2 --- /dev/null +++ b/Test/VariadicTest/App/main.py @@ -0,0 +1,184 @@ +from stdint import * +import t, c +import stdio +import testcheck +import string +import viperlib +import w32.win32console + +# ============================================================ +# VariadicTest - 测试 snprintf/printf variadic 参数传递 +# ============================================================ + +# --- Test 1: %u 格式化小整数 (CUInt8T -> CUnsignedInt) --- +def test_snprintf_u8(): + """测试 CUInt8T 值通过 t.CUnsignedInt() 传递给 %u""" + testcheck.section("Test 1: %u 格式化小整数 (CUInt8T -> CUnsignedInt)") + hr: t.CUInt8T = 14 + mn: t.CUInt8T = 30 + sc: t.CUInt8T = 59 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%02u:%02u:%02u", t.CUnsignedInt(hr), t.CUnsignedInt(mn), t.CUnsignedInt(sc)) + stdio.printf("Test1 (CUInt8T->%%u): [%s] (expect 14:30:59)\n", c.Addr(buf)) + testcheck.ok("CUInt8T -> %u (expect 14:30:59)") + + +# --- Test 2: %u 格式化 CUInt16T --- +def test_snprintf_u16(): + """测试 CUInt16T 值通过 t.CUnsignedInt() 传递给 %u""" + testcheck.section("Test 2: %u 格式化 CUInt16T") + yr: t.CUInt16T = 2026 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%04u", t.CUnsignedInt(yr)) + stdio.printf("Test2 (CUInt16T->%%u): [%s] (expect 2026)\n", c.Addr(buf)) + testcheck.ok("CUInt16T -> %u (expect 2026)") + + +# --- Test 3: %lu 格式化 CUInt64T --- +def test_snprintf_u64(): + """测试 CUInt64T 值传递给 %lu""" + testcheck.section("Test 3: %lu 格式化 CUInt64T") + val: t.CUInt64T = 1234567890123 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%lu", val) + stdio.printf("Test3 (CUInt64T->%%lu): [%s] (expect 1234567890123)\n", c.Addr(buf)) + testcheck.ok("CUInt64T -> %lu (expect 1234567890123)") + + +# --- Test 4: %d 格式化负数 --- +def test_snprintf_d_neg(): + """测试 %d 格式化负整数""" + testcheck.section("Test 4: %d 格式化负数") + val: t.CInt = -42 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%d", val) + stdio.printf("Test4 (CInt->%%d neg): [%s] (expect -42)\n", c.Addr(buf)) + testcheck.ok("CInt -> %d neg (expect -42)") + + +# --- Test 5: %d 格式化正数 --- +def test_snprintf_d_pos(): + testcheck.section("Test 5: %d 格式化正数") + val: t.CInt = 42 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%d", val) + stdio.printf("Test5 (CInt->%%d pos): [%s] (expect 42)\n", c.Addr(buf)) + testcheck.ok("CInt -> %d pos (expect 42)") + + +# --- Test 6: %x 格式化十六进制 --- +def test_snprintf_x(): + testcheck.section("Test 6: %x 格式化十六进制") + val: t.CUInt32T = 0xDEADBEEF + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "0x%08x", val) + stdio.printf("Test6 (%%x): [%s] (expect 0xdeadbeef)\n", c.Addr(buf)) + testcheck.ok("%x hex (expect 0xdeadbeef)") + + +# --- Test 7: 混合格式化 (模拟 desktop.py 时间格式) --- +def test_snprintf_mixed(): + """模拟 desktop.py 的时间格式化""" + testcheck.section("Test 7: 混合格式化 (模拟 desktop.py 时间格式)") + hr: t.CUInt8T = 14 + mn: t.CUInt8T = 5 + sc: t.CUInt8T = 9 + day: t.CUInt8T = 14 + mon: t.CUInt8T = 6 + yr: t.CUInt16T = 2026 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%04u-%02u-%02u %02u:%02u:%02u", + t.CUnsignedInt(yr), t.CUnsignedInt(mon), t.CUnsignedInt(day), + t.CUnsignedInt(hr), t.CUnsignedInt(mn), t.CUnsignedInt(sc)) + stdio.printf("Test7 (time format): [%s] (expect 2026-06-14 14:05:09)\n", c.Addr(buf)) + testcheck.ok("time format (expect 2026-06-14 14:05:09)") + + +# --- Test 8: printf 直接输出 variadic --- +def test_printf_variadic(): + """测试 printf 直接输出 variadic 参数""" + testcheck.section("Test 8: printf 直接输出 variadic") + stdio.printf("Test8 (printf direct): %u %d %lu 0x%x (expect 100 -50 999999999 0xcafe)\n", + t.CUnsignedInt(100), -50, t.CUInt64T(999999999), 0xCAFE) + testcheck.ok("printf direct variadic (expect 100 -50 999999999 0xcafe)") + + +# --- Test 9: 多个 %lu 连续 --- +def test_snprintf_multi_lu(): + """测试多个 %lu 连续（模拟 perf 日志格式）""" + testcheck.section("Test 9: 多个 %lu 连续") + v1: t.CUInt64T = 16 + v2: t.CUInt64T = 2 + v3: t.CUInt64T = 11 + buf: t.CArray[t.CChar, 128] + string.memset(c.Addr(buf), 0, 128) + viperlib.snprintf(c.Addr(buf), 128, "wall=%lu work=%lu render=%lu", v1, v2, v3) + stdio.printf("Test9 (multi %%lu): [%s] (expect wall=16 work=2 render=11)\n", c.Addr(buf)) + testcheck.ok("multi %lu (expect wall=16 work=2 render=11)") + + +# --- Test 10: %u 格式化 0 --- +def test_snprintf_zero(): + testcheck.section("Test 10: %u 格式化 0") + val: t.CUInt32T = 0 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%u", val) + stdio.printf("Test10 (%%u zero): [%s] (expect 0)\n", c.Addr(buf)) + testcheck.ok("%u zero (expect 0)") + + +# --- Test 11: %u 格式化大值 --- +def test_snprintf_large(): + testcheck.section("Test 11: %u 格式化大值") + val: t.CUInt32T = 4294967295 # 0xFFFFFFFF + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%u", val) + stdio.printf("Test11 (%%u max): [%s] (expect 4294967295)\n", c.Addr(buf)) + testcheck.ok("%u max (expect 4294967295)") + + +# --- Test 12: CUnsignedInt cast from CUInt8T edge values --- +def test_snprintf_u8_edge(): + """测试 CUInt8T 边界值""" + testcheck.section("Test 12: CUnsignedInt cast from CUInt8T edge values") + v0: t.CUInt8T = 0 + v127: t.CUInt8T = 127 + v128: t.CUInt8T = 128 + v255: t.CUInt8T = 255 + buf: t.CArray[t.CChar, 64] + string.memset(c.Addr(buf), 0, 64) + viperlib.snprintf(c.Addr(buf), 64, "%u %u %u %u", + t.CUnsignedInt(v0), t.CUnsignedInt(v127), t.CUnsignedInt(v128), t.CUnsignedInt(v255)) + stdio.printf("Test12 (CUInt8T edge): [%s] (expect 0 127 128 255)\n", c.Addr(buf)) + testcheck.ok("CUInt8T edge (expect 0 127 128 255)") + + +def main() -> t.CInt | t.CExport: + w32.win32console.SetConsoleCP(65001) + w32.win32console.SetConsoleOutputCP(65001) + + testcheck.begin("VariadicTest: snprintf/printf variadic 参数测试") + + test_snprintf_u8() + test_snprintf_u16() + test_snprintf_u64() + test_snprintf_d_neg() + test_snprintf_d_pos() + test_snprintf_x() + test_snprintf_mixed() + test_printf_variadic() + test_snprintf_multi_lu() + test_snprintf_zero() + test_snprintf_large() + test_snprintf_u8_edge() + + return testcheck.end() diff --git a/Test/VariadicTest/output/c1416dfae155d6e2.deps.json b/Test/VariadicTest/output/c1416dfae155d6e2.deps.json new file mode 100644 index 0000000..455dddf --- /dev/null +++ b/Test/VariadicTest/output/c1416dfae155d6e2.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "stdarg": "71e0a3ffcb3ebfad", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "viperio": "c9f4be41ca1cc2b4", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/VariadicTest/project.json b/Test/VariadicTest/project.json new file mode 100644 index 0000000..7d561c2 --- /dev/null +++ b/Test/VariadicTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "VariadicTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "VariadicTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/VariadicTest/temp/0035c95a18d4f8e8.pyi b/Test/VariadicTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/VariadicTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/VariadicTest/temp/067c78e9f121dce3.pyi b/Test/VariadicTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/VariadicTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/VariadicTest/temp/06f53cc594b4ac6c.pyi b/Test/VariadicTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/VariadicTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/VariadicTest/temp/6446627d4f07a1b5.pyi b/Test/VariadicTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/VariadicTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/VariadicTest/temp/6f62fe05c5ea1ceb.pyi b/Test/VariadicTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/VariadicTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/VariadicTest/temp/71e0a3ffcb3ebfad.pyi b/Test/VariadicTest/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/Test/VariadicTest/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/Test/VariadicTest/temp/72e2d5ccb7cedcf1.pyi b/Test/VariadicTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/VariadicTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/VariadicTest/temp/7e529fe7a078cfef.pyi b/Test/VariadicTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/VariadicTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/VariadicTest/temp/_phase1_manifest.json b/Test/VariadicTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..40d86de --- /dev/null +++ b/Test/VariadicTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\VariadicTest\\App\\main.py": {"sha1": "c1416dfae155d6e2", "mtime": 1782266230.8679473, "size": 6842}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}} \ No newline at end of file diff --git a/Test/VariadicTest/temp/_sha1_map.txt b/Test/VariadicTest/temp/_sha1_map.txt new file mode 100644 index 0000000..ac17918 --- /dev/null +++ b/Test/VariadicTest/temp/_sha1_map.txt @@ -0,0 +1,10 @@ +6f62fe05c5ea1ceb:includes/stdio.py +71e0a3ffcb3ebfad:includes/stdarg.py +7e529fe7a078cfef:includes/w32\win32base.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c1416dfae155d6e2:main.py +c3b259b4059f8668:includes/viperlib.py +c9f4be41ca1cc2b4:includes/viperio.py +dd3002730623424b:includes/testcheck.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/VariadicTest/temp/ab6e54ba9a669f76.pyi b/Test/VariadicTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/VariadicTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/VariadicTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/VariadicTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/VariadicTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/VariadicTest/temp/c1416dfae155d6e2.doc.json b/Test/VariadicTest/temp/c1416dfae155d6e2.doc.json new file mode 100644 index 0000000..4cc35ca --- /dev/null +++ b/Test/VariadicTest/temp/c1416dfae155d6e2.doc.json @@ -0,0 +1 @@ +{"test_snprintf_u8": "测试 CUInt8T 值通过 t.CUnsignedInt() 传递给 %u", "test_snprintf_u16": "测试 CUInt16T 值通过 t.CUnsignedInt() 传递给 %u", "test_snprintf_u64": "测试 CUInt64T 值传递给 %lu", "test_snprintf_d_neg": "测试 %d 格式化负整数", "test_snprintf_mixed": "模拟 desktop.py 的时间格式化", "test_printf_variadic": "测试 printf 直接输出 variadic 参数", "test_snprintf_multi_lu": "测试多个 %lu 连续（模拟 perf 日志格式）", "test_snprintf_u8_edge": "测试 CUInt8T 边界值"} \ No newline at end of file diff --git a/Test/VariadicTest/temp/c1416dfae155d6e2.pyi b/Test/VariadicTest/temp/c1416dfae155d6e2.pyi new file mode 100644 index 0000000..636e8b0 --- /dev/null +++ b/Test/VariadicTest/temp/c1416dfae155d6e2.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +from stdint import * +import t, c +import stdio +import testcheck +import string +import viperlib +import w32.win32console + +def test_snprintf_u8() -> t.CInt: pass + +def test_snprintf_u16() -> t.CInt: pass + +def test_snprintf_u64() -> t.CInt: pass + +def test_snprintf_d_neg() -> t.CInt: pass + +def test_snprintf_d_pos() -> t.CInt: pass + +def test_snprintf_x() -> t.CInt: pass + +def test_snprintf_mixed() -> t.CInt: pass + +def test_printf_variadic() -> t.CInt: pass + +def test_snprintf_multi_lu() -> t.CInt: pass + +def test_snprintf_zero() -> t.CInt: pass + +def test_snprintf_large() -> t.CInt: pass + +def test_snprintf_u8_edge() -> t.CInt: pass + +def main() -> t.CInt | t.CExport: pass diff --git a/Test/VariadicTest/temp/c3b259b4059f8668.pyi b/Test/VariadicTest/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/Test/VariadicTest/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/Test/VariadicTest/temp/c9f4be41ca1cc2b4.pyi b/Test/VariadicTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/VariadicTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/VariadicTest/temp/dd3002730623424b.pyi b/Test/VariadicTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/VariadicTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/VariadicTest/temp/f5522571bcce7bcb.pyi b/Test/VariadicTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/VariadicTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/VariadicTest/temp/f6b51804a0ba8ff0.pyi b/Test/VariadicTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/VariadicTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/WebTest/App/main.py b/Test/WebTest/App/main.py new file mode 100644 index 0000000..743f8b2 --- /dev/null +++ b/Test/WebTest/App/main.py @@ -0,0 +1,80 @@ +import t, c +from stdint import * +import stdio +import stdlib +import socket +import memhub +import requests + + +# 全局 MBuddy 指针 +_mbuddy: memhub.MemBuddy | t.CPtr + + +@t.CExport +def main() -> int: + stdio.printf("WebTest: HTTP client library test\n") + + # 初始化内存管理器（16MB arena） + arena: bytes = stdlib.malloc(16 * 1024 * 1024) + if arena is None: + stdio.printf("FATAL: malloc arena failed\n") + return 1 + + _mbuddy = memhub.MemBuddy(arena, 16 * 1024 * 1024) + requests._mbuddy = _mbuddy + + # 初始化 Socket 库 + result: INT = socket.SocketInit() + stdio.printf("SocketInit: %d\n", result) + + # 测试 Socket OOP API + sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM) + if sock.fd != socket.INVALID_SOCKET: + stdio.printf("Socket: OK (fd=%d)\n", sock.fd) + sock.close() + stdio.printf("Socket.close: OK\n") + else: + stdio.printf("Socket: FAIL\n") + + # 测试 DNS 解析 — 通过 Socket.connect 间接测试（内部调用 _MakeAddr） + stdio.printf("\n--- DNS test ---\n") + dns_sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM) + if dns_sock.fd != socket.INVALID_SOCKET: + dns_sock.settimeout(5000) + result: INT = dns_sock.connect("example.com", 80) + if result == 0: + stdio.printf("DNS: example.com resolved OK\n") + else: + stdio.printf("DNS: example.com resolved but connect failed (err in _MakeAddr debug above)\n") + dns_sock.close() + + # 测试 HTTP GET — 使用 DNS 解析 example.com + stdio.printf("\n--- HTTP GET test (example.com) ---\n") + resp = requests.get("http://example.com/") + if resp is not None: + stdio.printf("requests.get: OK\n") + code: INT = resp.status_code + length: INT = resp.body_len + stdio.printf(" status_code: %d\n", code) + stdio.printf(" body_len: %d\n", length) + if resp.text is not None and length > 0: + print_len: INT = 300 + if length < print_len: + print_len = length + stdio.printf(" text (first %d chars):\n", print_len) + txt: str = resp.text + i: INT = 0 + while i < print_len and txt[i] != 0: + stdio.printf("%c", txt[i]) + i += 1 + stdio.printf("\n") + requests.release(resp) + else: + stdio.printf("requests.get: FAIL (network unavailable)\n") + + # 清理 + socket.SocketCleanup() + stdio.printf("\nSocketCleanup: OK\n") + stdio.printf("WebTest: ALL PASSED\n") + return 0 \ No newline at end of file diff --git a/Test/WebTest/output/8046d45456d0b153.deps.json b/Test/WebTest/output/8046d45456d0b153.deps.json new file mode 100644 index 0000000..726d6b5 --- /dev/null +++ b/Test/WebTest/output/8046d45456d0b153.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5", "stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "string": "ab6e54ba9a669f76", "requests": "b558d8d8f01f4825", "viperio": "c9f4be41ca1cc2b4", "socket": "d7e3386b828acb66", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/WebTest/project.json b/Test/WebTest/project.json new file mode 100644 index 0000000..2d00ff7 --- /dev/null +++ b/Test/WebTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "WebTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-lws2_32", "-Wl,--allow-multiple-definition"], + "output": "WebTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} \ No newline at end of file diff --git a/Test/WebTest/temp/271ea3decb810db2.pyi b/Test/WebTest/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/Test/WebTest/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/Test/WebTest/temp/6446627d4f07a1b5.pyi b/Test/WebTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/WebTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/WebTest/temp/6f62fe05c5ea1ceb.pyi b/Test/WebTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/WebTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/WebTest/temp/7e529fe7a078cfef.pyi b/Test/WebTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/WebTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/WebTest/temp/8046d45456d0b153.doc.json b/Test/WebTest/temp/8046d45456d0b153.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/WebTest/temp/8046d45456d0b153.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/WebTest/temp/8046d45456d0b153.pyi b/Test/WebTest/temp/8046d45456d0b153.pyi new file mode 100644 index 0000000..a6ccaab --- /dev/null +++ b/Test/WebTest/temp/8046d45456d0b153.pyi @@ -0,0 +1,18 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import stdlib +import socket +import memhub +import requests + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +@t.CExport +def main() -> int: pass diff --git a/Test/WebTest/temp/90c53dd6db8d41cf.pyi b/Test/WebTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/WebTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/WebTest/temp/93c1d18e35d188d6.pyi b/Test/WebTest/temp/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/Test/WebTest/temp/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/Test/WebTest/temp/_phase1_manifest.json b/Test/WebTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..00cb85b --- /dev/null +++ b/Test/WebTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\WebTest\\App\\main.py": {"sha1": "8046d45456d0b153", "mtime": 1783055933.4493492, "size": 2570}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\platmacro.py": {"sha1": "93c1d18e35d188d6", "mtime": 1781237231.03366, "size": 1422}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\requests.py": {"sha1": "b558d8d8f01f4825", "mtime": 1783055330.1315396, "size": 9850}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\socket.py": {"sha1": "d7e3386b828acb66", "mtime": 1783054980.3909142, "size": 14514}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\winsock2.py": {"sha1": "6446627d4f07a1b5", "mtime": 1783052668.8022628, "size": 3547}} \ No newline at end of file diff --git a/Test/WebTest/temp/_sha1_map.txt b/Test/WebTest/temp/_sha1_map.txt new file mode 100644 index 0000000..0da1ad9 --- /dev/null +++ b/Test/WebTest/temp/_sha1_map.txt @@ -0,0 +1,13 @@ +271ea3decb810db2:includes/atom.py +6446627d4f07a1b5:includes/w32\winsock2.py +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +8046d45456d0b153:main.py +90c53dd6db8d41cf:includes/stdlib.py +93c1d18e35d188d6:includes/platmacro.py +ab6e54ba9a669f76:includes/string.py +b558d8d8f01f4825:includes/requests.py +c9f4be41ca1cc2b4:includes/viperio.py +d7e3386b828acb66:includes/socket.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/WebTest/temp/ab6e54ba9a669f76.pyi b/Test/WebTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/WebTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/WebTest/temp/b558d8d8f01f4825.pyi b/Test/WebTest/temp/b558d8d8f01f4825.pyi new file mode 100644 index 0000000..a326df6 --- /dev/null +++ b/Test/WebTest/temp/b558d8d8f01f4825.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from requests.py +Module: requests +""" + + +import t, c +from stdint import * +import string +import socket +import memhub +import stdio + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr +_socket_ready: t.CExtern | bool +HTTP_PORT: t.CDefine = 80 +HTTP_BUFSIZ: t.CDefine = 4096 + +class Response: + status_code: INT + text: str + body_len: INT + _raw_buf: str + +def _EnsureSocketInit() -> INT: pass + +def _AllocResponse() -> Response | t.CPtr: pass + +def _ParseUrl(url: str, host_buf: str, host_size: INT, path_buf: str, path_size: INT, port: INT | t.CPtr) -> INT: pass + +def _ParseStatusLine(resp_text: str, out_code: INT | t.CPtr) -> INT: pass + +def _FindBodyStart(resp_text: str, resp_len: INT) -> INT: pass + +def get(url: str) -> Response | t.CPtr: pass + +def post(url: str, data: str, content_type: str) -> Response | t.CPtr: pass + +def release(resp: Response | t.CPtr) -> t.CInt: pass diff --git a/Test/WebTest/temp/c9f4be41ca1cc2b4.pyi b/Test/WebTest/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/Test/WebTest/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/Test/WebTest/temp/d7e3386b828acb66.pyi b/Test/WebTest/temp/d7e3386b828acb66.pyi new file mode 100644 index 0000000..aa17fc3 --- /dev/null +++ b/Test/WebTest/temp/d7e3386b828acb66.pyi @@ -0,0 +1,115 @@ +""" +Auto-generated Python stub file from socket.py +Module: socket +""" + + +import t, c +from stdint import * +import string +import platmacro +import stdio +import w32.winsock2 + +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 10 +SOCK_STREAM: t.CDefine = 1 +SOCK_DGRAM: t.CDefine = 2 +SOCK_RAW: t.CDefine = 3 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 1 +SO_RCVTIMEO: t.CDefine = 20 +SO_SNDTIMEO: t.CDefine = 21 +SO_REUSEADDR: t.CDefine = 2 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = -1 +SOCKET_DEFAULT_TIMEOUT: t.CDefine = 0 + +class SocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class HostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: INT + h_length: INT + h_addr_list: CHARPTR +class timeval: + tv_sec: INT + tv_usec: INT + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.CExtern | t.CExport: pass + +def connect(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass + +def send(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass + +def recv(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass + +def bind(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass + +def listen(fd: u64, backlog: INT) -> INT | t.CExtern | t.CExport: pass + +def accept(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.CExtern | t.CExport: pass + +def setsockopt(fd: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.CExtern | t.CExport: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> HostEnt | t.CPtr | t.CExtern | t.CExport: pass + +def ntohs(netshort: u16) -> u16 | t.CExtern | t.CExport: pass + +def htons(hostshort: u16) -> u16 | t.CExtern | t.CExport: pass + +def close(fd: u64) -> INT | t.CExtern | t.CExport: pass + +def _CloseFd(fd: INT) -> INT: pass + +def _MakeAddr(host: str, port: INT) -> SocketAddr: pass + +def _ParseIPv4(ip_str: str, out_addr: u32 | t.CPtr) -> bool: pass + + +class Socket: + fd: INT + family: INT + type: INT + protocol: INT + def __init__(self: Socket, family: INT, type: INT, protocol: INT) -> t.CInt: pass + def connect(self: Socket, host: str, port: INT) -> INT: pass + def send(self: Socket, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + def recv(self: Socket, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: pass + def sendall(self: Socket, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + def close(self: Socket) -> INT: pass + def bind(self: Socket, host: str, port: INT) -> INT: pass + def listen(self: Socket, backlog: INT) -> INT: pass + def accept(self: Socket, out_client: 'Socket' | t.CPtr) -> INT: pass + def settimeout(self: Socket, timeout_ms: INT) -> INT: pass + def fileno(self: Socket) -> INT: pass + +def SocketInit() -> INT: pass + +def SocketCleanup() -> INT: pass + +def SocketCreate(family: INT, type: INT, protocol: INT, out_sock: Socket | t.CPtr) -> INT: pass + +def SocketConnect(sock: Socket | t.CPtr, host: str, port: INT) -> INT: pass + +def SocketSend(sock: Socket | t.CPtr, data: t.CVoid | t.CPtr, length: INT) -> INT: pass + +def SocketRecv(sock: Socket | t.CPtr, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: pass + +def SocketClose(sock: Socket | t.CPtr) -> INT: pass + +def SocketBind(sock: Socket | t.CPtr, host: str, port: INT) -> INT: pass + +def SocketListen(sock: Socket | t.CPtr, backlog: INT) -> INT: pass + +def SocketAccept(sock: Socket | t.CPtr, out_client: Socket | t.CPtr) -> INT: pass + +def SocketSetTimeout(sock: Socket | t.CPtr, timeout_ms: INT) -> INT: pass + +def SocketIsValid(sock: Socket | t.CPtr) -> bool: pass diff --git a/Test/WebTest/temp/ee084e9fc6ee413a.pyi b/Test/WebTest/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/Test/WebTest/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/Test/WebTest/temp/f5522571bcce7bcb.pyi b/Test/WebTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/WebTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ZlibTest/App/__zlib/__init__.py b/Test/ZlibTest/App/__zlib/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/Test/ZlibTest/App/__zlib/pyzlib.py b/Test/ZlibTest/App/__zlib/pyzlib.py new file mode 100644 index 0000000..e495e6d --- /dev/null +++ b/Test/ZlibTest/App/__zlib/pyzlib.py @@ -0,0 +1,549 @@ +from stdint import * +import zdeflate +import zinflate +import zchecksum +import zdef +import zhuff +import stdlib +import string +import t, c + + +# ============================================================ +# Constants - matching Python zlib module names and values +# ============================================================ + +# Compression levels +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) + +# Compression methods +DEFLATED: t.CDefine = 8 + +# Flush modes +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 + +# Strategies +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 + +# Return codes +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) + +# Window / Buffer constants +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 + +# Version +ZLIB_VERSION: t.CDefine = "1.3.2" + +pyzlib_error_msg: t.CArray[t.CChar, 512] = "" +pyzlib_error_code_val: t.CInt = 0 + +# ============================================================ +# Structures +# ============================================================ +@t.Object +class Compress: + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + + def compress(self, + data: BYTEPTR, data_len: t.CSizeT, + out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + if self.is_finished: + set_error(Z_STREAM_ERROR, "compress object already finished") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + if data_len == 0: + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + while self.input_buf_len + data_len > self.input_buf_cap: + self.input_buf_cap *= 2 + new_buf: BYTEPTR = BYTEPTR(realloc(self.input_buf, self.input_buf_cap)) + if not new_buf: + set_error(Z_MEM_ERROR, "out of memory") + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + self.input_buf = new_buf + memcpy(self.input_buf + self.input_buf_len, data, data_len) + self.input_buf_len += data_len + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + + def flush(self, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + if mode == Z_FINISH: + if not self.header_written: + self.header_written = 1 + input_data: BYTEPTR = self.input_buf + input_len: t.CSizeT = self.input_buf_len + if input_len == 0: + s.writer.write_bits(1, 1) + s.writer.write_bits(1, 2) + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + lit_tree.encode_symbol(256, c.Addr(s.writer)) + else: + s.adler = zchecksum.zchecksum_adler32(input_data, input_len, 1) + s.compress_block(input_data, input_len, 1) + s.is_finished = 1 + if s.wbits >= 0: + s.writer.align() + adler: t.CUInt32T = s.adler + s.writer.buf[s.writer.byte_pos + 0] = (adler >> 24) & 0xFF + s.writer.buf[s.writer.byte_pos + 1] = (adler >> 16) & 0xFF + s.writer.buf[s.writer.byte_pos + 2] = (adler >> 8) & 0xFF + s.writer.buf[s.writer.byte_pos + 3] = (adler >> 0) & 0xFF + s.writer.byte_pos += 4 + self.is_finished = 1 + total: t.CSizeT = s.writer.total() + result: BYTEPTR = BYTEPTR(malloc(total)) + if result: memcpy(result, s.writer.buf, total) + c.Set(c.Deref(out_len), total) + free(self.input_buf) + self.input_buf = None + self.input_buf_len = 0 + self.input_buf_cap = 0 + return result + elif mode == Z_SYNC_FLUSH or mode == Z_FULL_FLUSH: + input_data: BYTEPTR = self.input_buf + input_len: t.CSizeT = self.input_buf_len + if input_len > 0: + s.adler = zchecksum.zchecksum_adler32(input_data, input_len, s.adler) + s.compress_block(input_data, input_len, 0) + self.input_buf_len = 0 + s.writer.write_bits(0, 1) + s.writer.write_bits(0, 2) + s.writer.align() + prev_pos: t.CSizeT = self.last_pos + total: t.CSizeT = s.writer.total() + delta: t.CSizeT = total - prev_pos + self.last_pos = s.writer.byte_pos + if delta == 0: + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + + result: BYTEPTR = BYTEPTR(malloc(delta)) + if result: memcpy(result, s.writer.buf + prev_pos, delta) + c.Set(c.Deref(out_len), delta) + return result + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + + def copy(self) -> Compress | t.CPtr: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + copy_obj: Compress | t.CPtr = calloc(1, Compress.__sizeof__()) + if not copy_obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + copy_obj.stream = s.copy() + if not copy_obj.stream: + free(copy_obj) + set_error(Z_MEM_ERROR, "out of memory") + return None + copy_obj.is_initialized = 1 + copy_obj.is_finished = self.is_finished + copy_obj.level = self.level + copy_obj.method = self.method + copy_obj.wbits = self.wbits + copy_obj.memLevel = self.memLevel + copy_obj.strategy = self.strategy + copy_obj.last_pos = self.last_pos + copy_obj.header_written = self.header_written + if self.input_buf and self.input_buf_len > 0: + copy_obj.input_buf = BYTEPTR(calloc(1, self.input_buf_cap)) + if copy_obj.input_buf: + memcpy(copy_obj.input_buf, self.input_buf, self.input_buf_len) + copy_obj.input_buf_cap = self.input_buf_cap + copy_obj.input_buf_len = self.input_buf_len + else: + copy_obj.input_buf = BYTEPTR(calloc(1, 256)) + copy_obj.input_buf_cap = 256 + copy_obj.input_buf_len = 0 + if self.zdict and self.zdict_len > 0: + copy_obj.zdict = clone_bytes(self.zdict, self.zdict_len) + copy_obj.zdict_len = self.zdict_len + return copy_obj + + def delete(self): + if self.is_initialized and self.stream: + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + s.destroy() + free(self.zdict) + free(self.input_buf) + free(self) + + def __del__(self): + self.delete() + + +@t.Object +class Decompress: + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + + def decompress(self, data: BYTEPTR, data_len: t.CSizeT, + max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + if self.eof: + if data and data_len > 0: + append_bytes(c.Addr(self._unused_data), c.Addr(self._unused_data_len), + c.Addr(self._unused_data_cap), data, data_len) + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + self._unconsumed_tail_len = 0 + if data and data_len > 0: + append_bytes(c.Addr(self.input_buf), c.Addr(self.input_buf_len), + c.Addr(self.input_buf_cap), data, data_len) + if self.input_buf_len == 0: + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + s: zinflate.zinflate_stream | t.CPtr = zinflate.zinflate_create(self.wbits) + if not s: + set_error(Z_MEM_ERROR, "out of memory") + return None + if self.zdict and self.zdict_len > 0: + s.set_dictionary(self.zdict, self.zdict_len) + out: t.CUInt8T | t.CPtr = None + err: t.CInt = s.decompress(self.input_buf, self.input_buf_len, + max_length, c.Addr(out), out_len) + if err != 0: + s.destroy() + set_error(Z_DATA_ERROR, "decompression failed") + return None + if s.is_finished: + self.eof = 1 + if s.input_pos < s.input_len: + append_bytes(c.Addr(self._unused_data), c.Addr(self._unused_data_len), + c.Addr(self._unused_data_cap), + s.input_data + s.input_pos, + s.input_len - s.input_pos) + self.input_buf_len = 0 + else: + consumed: t.CSizeT = s.input_pos + if consumed < self.input_buf_len: + remaining: t.CSizeT = self.input_buf_len - consumed + memmove(self.input_buf, self.input_buf + consumed, remaining) + self.input_buf_len = remaining + else: + self.input_buf_len = 0 + s.destroy() + return out + + def flush(self, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + c.Set(c.Deref(out_len), 0) + return BYTEPTR(calloc(1, 1)) + + def copy(self) -> Decompress | t.CPtr: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + copy_obj: Decompress | t.CPtr = calloc(1, Decompress.__sizeof__()) + if not copy_obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + copy_obj.is_initialized = 1 + copy_obj.eof = self.eof + copy_obj.wbits = self.wbits + if self.zdict and self.zdict_len > 0: + copy_obj.zdict = clone_bytes(self.zdict, self.zdict_len) + copy_obj.zdict_len = self.zdict_len + if self._unused_data and self._unused_data_len > 0: + copy_obj._unused_data = clone_bytes(self._unused_data, self._unused_data_len) + copy_obj._unused_data_len = self._unused_data_len + copy_obj._unused_data_cap = self._unused_data_len + if self._unconsumed_tail and self._unconsumed_tail_len > 0: + copy_obj._unconsumed_tail = clone_bytes(self._unconsumed_tail, self._unconsumed_tail_len) + copy_obj._unconsumed_tail_len = self._unconsumed_tail_len + copy_obj._unconsumed_tail_cap = self._unconsumed_tail_len + if self.input_buf and self.input_buf_len > 0: + copy_obj.input_buf = clone_bytes(self.input_buf, self.input_buf_len) + copy_obj.input_buf_len = self.input_buf_len + copy_obj.input_buf_cap = self.input_buf_cap + else: + copy_obj.input_buf = BYTEPTR(calloc(1, 256)) + copy_obj.input_buf_cap = 256 + copy_obj.input_buf_len = 0 + return copy_obj + + def delete(self): + free(self.zdict) + free(self._unused_data) + free(self._unconsumed_tail) + free(self.input_buf) + free(self) + + def unused_data(self, length: t.CSizeT | t.CPtr) -> BYTEPTR: + #if not self: + # if length: + # c.Set(c.Deref(length), 0) + # return None + if length: + c.Set(c.Deref(length), self._unused_data_len) + return self._unused_data + + def unconsumed_tail(self, length: t.CSizeT | t.CPtr) -> BYTEPTR: + #if not self: + # if length: + # c.Set(c.Deref(length), 0) + # return None + if length: + c.Set(c.Deref(length), self._unconsumed_tail_len) + return self._unconsumed_tail + +# def eof(self) -> t.CInt: +# # if not obj: return 0 +# return self.eof + + + +def set_error(code: int, msg: str): + global pyzlib_error_code_val, pyzlib_error_msg + pyzlib_error_code_val = code + if msg: + strncpy(pyzlib_error_msg, msg, pyzlib_error_msg.__sizeof__() - 1) + pyzlib_error_msg[pyzlib_error_msg.__sizeof__() - 1] = '\0' + else: + pyzlib_error_msg[0] = '\0' + +def clone_bytes(src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: + if not src or length == 0: return None + dst: BYTEPTR = BYTEPTR(malloc(length)) + if dst: memcpy(dst, src, length) + return dst + +def append_bytes(buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, + data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: + if not data or data_len == 0: return 0 + needed: t.CSizeT = c.Deref(length) + data_len + if needed > c.Deref(cap): + new_cap: t.CSizeT = c.Deref(cap) * 2 + if new_cap < needed: new_cap = needed + new_buf: BYTEPTR = BYTEPTR(realloc(c.Deref(buf), new_cap)) + if not new_buf: return -1 + c.Set(c.Deref(buf), new_buf) + c.Set(c.Deref(cap), new_cap) + memcpy(c.Deref(buf) + c.Deref(length), data, data_len) + c.Set(c.Deref(length), c.Deref(length) + data_len) + return 0 + + +def shrink_to_fit(buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: + if length == 0: + free(buf) + return BYTEPTR(calloc(1, 1)) + result: BYTEPTR = BYTEPTR(realloc(buf, length)) + return result if result else buf + + +def runtime_version() -> str: + return ZLIB_VERSION + +def get_error() -> str: + return pyzlib_error_msg + +def get_error_code() -> t.CInt: + return pyzlib_error_code_val + +def clear_error(): + global pyzlib_error_msg, pyzlib_error_code_val + pyzlib_error_msg[0] = '\0' + pyzlib_error_code_val = 0 + +def compress(data: BYTEPTR, data_len: t.CSizeT, + level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not data and data_len > 0: + set_error(Z_STREAM_ERROR, "data is None but data_len > 0") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: + actual_wbits = -MAX_WBITS + raw_result: UINT8PTR = zdeflate.zdeflate_one_shot(data, data_len, level, actual_wbits, out_len) + if not raw_result: + set_error(Z_DATA_ERROR, "compression failed") + return None + if wbits > MAX_WBITS: + gzip_len: t.CSizeT = 10 + c.Deref(out_len) + 8 + gzip_buf: BYTEPTR = BYTEPTR(malloc(gzip_len)) + if not gzip_buf: + free(raw_result) + set_error(Z_MEM_ERROR, "out of memory") + return None + pos: t.CSizeT = 0 + gzip_buf[pos + 0] = 0x1F + gzip_buf[pos + 1] = 0x8B + gzip_buf[pos + 2] = 0x08 + gzip_buf[pos + 3] = 0x00 + gzip_buf[pos + 4] = 0x00 + gzip_buf[pos + 5] = 0x00 + gzip_buf[pos + 6] = 0x00 + gzip_buf[pos + 7] = 0x00 + gzip_buf[pos + 8] = 0x00 + gzip_buf[pos + 9] = 0xFF + pos += 10 + + memcpy(gzip_buf + pos, raw_result, c.Deref(out_len)) + pos += c.Deref(out_len) + free(raw_result) + + crc: t.CUInt32T = zchecksum.zchecksum_crc32(data, data_len, 0) + gzip_buf[pos + 0] = (crc) & 0xFF + gzip_buf[pos + 1] = (crc >> 8) & 0xFF + gzip_buf[pos + 2] = (crc >> 16) & 0xFF + gzip_buf[pos + 3] = (crc >> 24) & 0xFF + gzip_buf[pos + 4] = (data_len) & 0xFF + gzip_buf[pos + 5] = (data_len >> 8) & 0xFF + gzip_buf[pos + 6] = (data_len >> 16) & 0xFF + gzip_buf[pos + 7] = (data_len >> 24) & 0xFF + pos += 8 + c.Set(c.Deref(out_len), pos) + return shrink_to_fit(gzip_buf, pos) + return shrink_to_fit(raw_result, c.Deref(out_len)) + + +def decompress(data: BYTEPTR, data_len: t.CSizeT, + wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not data and data_len > 0: + set_error(Z_STREAM_ERROR, "data is None but data_len > 0") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + result: UINT8PTR = zinflate.zinflate_one_shot(data, data_len, wbits, bufsize, out_len) + if not result: + set_error(Z_DATA_ERROR, "decompression failed") + return None + return shrink_to_fit(result, c.Deref(out_len)) + +def compressobj(level: t.CInt, method: t.CInt, wbits: t.CInt, + memLevel: t.CInt, strategy: t.CInt, + zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: + obj: Compress | t.CPtr = calloc(1, Compress.__sizeof__()) + if not obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: actual_wbits = wbits - 16 + s: zdeflate.zdeflate_stream | t.CPtr = zdeflate.zdeflate_create(level, actual_wbits, memLevel, strategy) + if not s: + free(obj) + set_error(Z_MEM_ERROR, "out of memory") + return None + obj.stream = s + obj.is_initialized = 1 + obj.level = level + obj.method = method + obj.wbits = wbits + obj.memLevel = memLevel + obj.strategy = strategy + obj.input_buf = BYTEPTR(calloc(1, 256)) + obj.input_buf_cap = 256 + obj.input_buf_len = 0 + obj.header_written = 0 + if zdict and zdict_len > 0: + s.set_dictionary(zdict, zdict_len) + obj.zdict = clone_bytes(zdict, zdict_len) + obj.zdict_len = zdict_len + return obj + + +def decompressobj(wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: + obj: Decompress | t.CPtr = calloc(1, Decompress.__sizeof__()) + if not obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: actual_wbits = wbits - 16 + obj.stream = None + obj.is_initialized = 1 + obj.wbits = actual_wbits + if zdict and zdict_len > 0: + obj.zdict = clone_bytes(zdict, zdict_len) + obj.zdict_len = zdict_len + obj.input_buf = BYTEPTR(calloc(1, 256)) + obj.input_buf_cap = 256 + obj.input_buf_len = 0 + return obj + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: + return zchecksum.zchecksum_adler32(data, length, UINT32(value)) + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: + return zchecksum.zchecksum_crc32(data, length, UINT32(value)) + + diff --git a/Test/ZlibTest/App/__zlib/test_pyzlib.py b/Test/ZlibTest/App/__zlib/test_pyzlib.py new file mode 100644 index 0000000..3d572bc --- /dev/null +++ b/Test/ZlibTest/App/__zlib/test_pyzlib.py @@ -0,0 +1,877 @@ +from stdint import * +import pyzlib +import zhuff +import zdef +from stdio import printf +import stdlib +from string import strlen +import string +import t, c + + +test_passed: t.CInt = 0 +test_failed: t.CInt = 0 +def check(name: str, condition: t.CInt, detail: str): + global test_passed, test_failed + if condition: + test_passed += 1 + printf(" [PASS] %s\n", name) + else: + test_failed += 1 + printf(" [FAIL] %s -- %s\n", name, detail if detail else "") + +def print_hex_test(data: BYTEPTR, length: t.CSizeT, max_show: t.CSizeT): + if not data or length == 0: + printf("(empty)\n") + return + show: t.CSizeT = max_show if (length > max_show) else length + i: t.CSizeT + for i in range(show): + printf("%02X ", data[i]) + if length > max_show: + printf("... (%zu bytes total)", length) + printf("\n") + +def print_separator(): + printf(" -------------------------------------------\n") + +def section_header(title: str): + printf("\n+-- %s --+\n", title) + +def section_footer(): + printf("+--------------------------------------------+\n") + +# ============================================================ + +def test_version(): + section_header("Version Info") + + ver: str = pyzlib.ZLIB_VERSION + printf(" ZLIB_VERSION (compile-time) : %s\n", ver) + check("ZLIB_VERSION not None", ver != None, "version string is None") + check("ZLIB_VERSION not empty", strlen(ver) > 0, "version string is empty") + + runtime_ver: str = pyzlib.runtime_version() + printf(" ZLIB_RUNTIME_VERSION : %s\n", runtime_ver) + check("runtime version not None", runtime_ver != None, "runtime version is None") + check("runtime version not empty", strlen(runtime_ver) > 0, "runtime version is empty") + + section_footer() + + +def test_compress_decompress(): + section_header("compress / decompress") + + inp: str = "Hello, World! This is a test of zlib compression and decompression." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("compress returns non-None", compressed != None, "compress returned None") + check("compress output size > 0", out_length > 0, "compressed size is 0") + check("compress output produced", out_length > 0, "compressed size is 0") + + printf(" Compressed (%zu bytes): ", out_length) + print_hex_test(compressed, out_length, 32) + printf(" Compression ratio: %.1f%%\n", t.CDouble(out_length) / input_length * 100) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(compressed, out_length, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress returns non-None", decompressed != None, "decompress returned None") + check("decompress output size matches input", dec_length == input_length, "size mismatch") + check("decompress output matches input", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + + printf(" Decompressed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(decompressed)) + + free(compressed) + free(decompressed) + section_footer() + + +def test_compress_levels(): + section_header("Compression Levels") + + inp: str = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA" + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + + c0: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_NO_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 0 compress OK", c0 != None, "level 0 returned None") + printf(" Level 0 (Z_NO_COMPRESSION): %zu bytes\n", out_length) + length0: t.CSizeT = out_length + free(c0) + + c1: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_BEST_SPEED, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 1 compress OK", c1 != None, "level 1 returned None") + printf(" Level 1 (Z_BEST_SPEED) : %zu bytes\n", out_length) + free(c1) + + c6: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level -1 compress OK", c6 != None, "default level returned None") + printf(" Level -1 (DEFAULT) : %zu bytes\n", out_length) + free(c6) + + c9: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_BEST_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("level 9 compress OK", c9 != None, "level 9 returned None") + printf(" Level 9 (Z_BEST_COMPRESS) : %zu bytes\n", out_length) + length9: t.CSizeT = out_length + free(c9) + + check("level 9 smaller than level 0", length9 < length0, "level 9 not smaller than level 0") + + section_footer() + + +def test_compressobj(): + section_header("compressobj (incremental)") + + part1: str = "Hello, " + part2: str = "World!" + printf(" Part 1: \"%s\"\n", part1) + printf(" Part 2: \"%s\"\n", part2) + + out_length: t.CSizeT = 0 + + s: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + check("compressobj created", s != None, "compressobj is None") + print_separator() + + c1: BYTEPTR = s.compress(BYTEPTR(part1), strlen(part1), c.Addr(out_length)) + check("compress part1 OK", c1 != None, "compress part1 failed") + printf(" Compressed part1: %zu bytes -> ", out_length) + print_hex_test(c1, out_length, 16) + total: t.CSizeT = out_length + all: BYTEPTR = None + if out_length > 0 and c1: + all = BYTEPTR(malloc(total)) + if all: memcpy(all, c1, out_length) + free(c1) + + c2: BYTEPTR = s.compress(BYTEPTR(part2), + strlen(part2), c.Addr(out_length)) + check("compress part2 OK", c2 != None, "compress part2 failed") + printf(" Compressed part2: %zu bytes -> ", out_length) + print_hex_test(c2, out_length, 16) + if out_length > 0 and c2: + new_all: BYTEPTR = BYTEPTR(realloc(all, total + out_length)) + if new_all: + all = new_all + memcpy(all + total, c2, out_length) + total += out_length + free(c2) + + c3: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush (Z_FINISH) OK", c3 != None, "flush failed") + printf(" Flush result : %zu bytes -> ", out_length) + print_hex_test(c3, out_length, 16) + if out_length > 0 and c3: + new_all: BYTEPTR = realloc(all, total + out_length) # 隐式转换 + if new_all: + all = new_all + memcpy(all + total, c3, out_length) + total += out_length + free(c3) + + print_separator() + + if all: + dec_length: t.CSizeT = 0 + dec: BYTEPTR = pyzlib.decompress(all, total, pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress incremental OK", dec != None, "decompress returned None") + check("decompress incremental size", dec != None and dec_length == strlen(part1) + strlen(part2), "size mismatch") + check("decompress incremental content", + dec != None and memcmp(dec, "Hello, World!", dec_length) == 0, "content mismatch") + if dec: + printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + free(all) + + s.delete() # del s + section_footer() + +def test_decompressobj(): + section_header("decompressobj") + + inp: str = "Test data for decompress object." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pyzlib.MAX_WBITS, None, 0) + check("decompressobj created", d != None, "decompressobj is None") + check("eof is false initially", d.eof == 0, "eof should be 0") + print_separator() + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(compressed, out_length, + 0, c.Addr(dec_length)) + check("decompress OK", dec != None, "decompress returned None") + check("decompress size", dec != None and dec_length == input_length, "size mismatch") + check("decompress content", dec != None and memcmp(dec, inp, input_length) == 0, "content mismatch") + check("eof is true after stream end", d.eof == 1, "eof should be 1") + + if dec: + printf(" Decompressed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(dec)) + printf(" eof = %d\n", d.eof) + + free(dec) + free(compressed) + d.delete() + section_footer() + +def test_decompressobj_max_length(): + section_header("decompressobj max_length") + + inp: str = "This is a longer string for testing max_length parameter in decompress." + input_length: t.CSizeT = strlen(inp) + printf(" Input (%zu bytes): \"%s\"\n", input_length, inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pyzlib.MAX_WBITS, None, 0) + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(compressed, out_length, + 10, c.Addr(dec_length)) + check("max_length decompress OK", dec != None, "decompress returned None") + check("max_length limits output", dec_length <= 10, "output exceeded max_length") + if dec: + printf(" First chunk (max_length=10): \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + + tail_length: t.CSizeT = 0 + tail: BYTEPTR = d.unconsumed_tail(c.Addr(tail_length)) + check("unconsumed_tail exists", tail != None or tail_length == 0, "no unconsumed_tail") + printf(" unconsumed_tail: %zu bytes remaining\n", tail_length) + printf(" eof = %d\n", d.eof) + + if tail_length > 0 or not d.eof: + flushed: BYTEPTR = d.flush(pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("flush after max_length OK", flushed != None, "flush returned None") + if flushed: printf(" Flushed (%zu bytes): \"%.*s\"\n", dec_length, t.CInt(dec_length), str(flushed)) + free(flushed) + + printf(" eof after flush = %d\n", d.eof) + + d.delete() + free(compressed) + section_footer() + + +def test_decompressobj_unused_data(): + section_header("decompressobj unused_data") + + inp: str = "Hello" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + total_length: t.CSizeT = out_length + 5 + combined: BYTEPTR = BYTEPTR(malloc(total_length)) + memcpy(combined, compressed, out_length) + memcpy(combined + out_length, "EXTRA", 5) + free(compressed) + printf(" Combined (compressed + \"EXTRA\"): %zu bytes\n", total_length) + + d: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pyzlib.MAX_WBITS, None, 0) + + dec_length: t.CSizeT = 0 + dec: BYTEPTR = d.decompress(combined, total_length, + 0, c.Addr(dec_length)) + check("decompress with extra OK", dec != None, "decompress returned None") + check("eof after stream end", d.eof == 1, "eof should be 1") + if dec: + printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(dec), dec_length) + free(dec) + + print_separator() + + unused_length: t.CSizeT = 0 + unused: BYTEPTR = d.unused_data(c.Addr(unused_length)) + check("unused_data exists", unused != None and unused_length > 0, "no unused_data") + check("unused_data is EXTRA", unused_length == 5 and unused != None and memcmp(unused, "EXTRA", 5) == 0, "unused_data mismatch") + printf(" unused_data (%zu bytes): \"%.*s\"\n", unused_length, t.CInt(unused_length), str(unused)) + + d.delete() + free(combined) + section_footer() + + +def test_compress_copy(): + section_header("Compress.copy()") + + inp: str = "Copy test data for compress object" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + c1: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + c1.compress(BYTEPTR(inp), input_length, c.Addr(out_length)) + printf(" Original: compressed %zu bytes so far\n", out_length) + + c2: pyzlib.Compress | t.CPtr = c1.copy() + check("compress copy OK", c2 != None, "copy returned None") + printf(" Copied compressobj\n") + + print_separator() + + f1: BYTEPTR = c1.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush original OK", f1 != None, "flush original failed") + length1: t.CSizeT = out_length + printf(" Original flush: %zu bytes\n", length1) + free(f1) + + f2: BYTEPTR = c2.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush copy OK", f2 != None, "flush copy failed") + length2: t.CSizeT = out_length + printf(" Copy flush : %zu bytes\n", length2) + free(f2) + + check("copy produces same output", length1 == length2, "output lengthgths differ") + + c1.delete() + c2.delete() + section_footer() + + +def test_decompress_copy(): + section_header("Decompress.copy()") + + inp: str = "Decompress copy test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\"\n", inp) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + printf(" Compressed: %zu bytes\n", out_length) + + d1: pyzlib.Decompress | t.CPtr = pyzlib.decompressobj(pyzlib.MAX_WBITS, None, 0) + d2: pyzlib.Decompress | t.CPtr = d1.copy() + check("decompress copy OK", d2 != None, "copy returned None") + printf(" Copied decompressobj\n") + + print_separator() + + dec_length1: t.CSizeT = 0 + dec_length2: t.CSizeT = 0 + dec1: BYTEPTR = d1.decompress(compressed, out_length, + 0, c.Addr(dec_length1)) + dec2: BYTEPTR = d2.decompress(compressed, out_length, + 0, c.Addr(dec_length2)) + check("decompress copy output size", dec_length1 == dec_length2, "sizes differ") + check("decompress copy output content", + dec1 and dec2 and memcmp(dec1, dec2, dec_length1) == 0, "content differs") + + if dec1: printf(" Original: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length1), str(dec1), dec_length1) + if dec2: printf(" Copy : \"%.*s\" (%zu bytes)\n", t.CInt(dec_length2), str(dec2), dec_length2) + + free(dec1) + free(dec2) + free(compressed) + d1.delete() + d2.delete() + section_footer() + + +def test_adler32(): + section_header("adler32 checksum") + + checksum: ULONG = pyzlib.zlib_adler32(BYTEPTR("Hello"), 5, 1) + printf(" adler32(\"Hello\") = 0x%08lX (%lu)\n", checksum, checksum) + check("adler32 returns non-zero", checksum != 0, "checksum is 0") + + incremental: ULONG = pyzlib.zlib_adler32(BYTEPTR("Hel"), 3, 1) + printf(" adler32(\"Hel\") = 0x%08lX\n", incremental) + incremental = pyzlib.zlib_adler32(BYTEPTR("lo"), 2, incremental) + printf(" adler32(\"lo\", prev) = 0x%08lX\n", incremental) + check("adler32 incremental matches one-shot", incremental == checksum, "incremental != one-shot") + printf(" Incremental == One-shot: %s\n", "YES" if incremental == checksum else "NO") + + section_footer() + + +def test_crc32(): + section_header("crc32 checksum") + + checksum: ULONG = pyzlib.zlib_crc32(BYTEPTR("Hello"), 5, 0) + printf(" crc32(\"Hello\") = 0x%08lX (%lu)\n", checksum, checksum) + check("crc32 returns non-zero", checksum != 0, "checksum is 0") + + incremental: ULONG = pyzlib.zlib_crc32(BYTEPTR("Hel"), 3, 0) + printf(" crc32(\"Hel\") = 0x%08lX\n", incremental) + incremental = pyzlib.zlib_crc32(BYTEPTR("lo"), 2, incremental) + printf(" crc32(\"lo\", prev) = 0x%08lX\n", incremental) + check("crc32 incremental matches one-shot", incremental == checksum, "incremental != one-shot") + printf(" Incremental == One-shot: %s\n", "YES" if incremental == checksum else "NO") + + section_footer() + + +def test_empty_compress(): + section_header("Empty data compress/decompress") + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(""), 0, + pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.MAX_WBITS, c.Addr(out_length)) + check("compress empty OK", compressed != None, "compress empty returned None") + check("compress empty output > 0", out_length > 0, "compressed empty is 0 bytes") + printf(" Empty input compressed: %zu bytes (header only)\n", out_length) + printf(" Hex: ") + print_hex_test(compressed, out_length, 32) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(compressed, out_length, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("decompress empty OK", decompressed != None, "decompress empty returned None") + check("decompress empty size == 0", dec_length == 0, "decompressed size != 0") + printf(" Decompressed back: %zu bytes\n", dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_gzip_format(): + section_header("Gzip format (wbits=31)") + + inp: str = "Gzip format test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\" (%zu bytes)\n", inp, input_length) + + gzip_wbits: t.CInt = pyzlib.MAX_WBITS + 16 + printf(" wbits = MAX_WBITS + 16 = %d\n", gzip_wbits) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, gzip_wbits, c.Addr(out_length)) + check("gzip compress OK", compressed != None, "gzip compress failed") + printf(" Gzip compressed: %zu bytes\n", out_length) + printf(" Header bytes: ") + print_hex_test(compressed, 4 if (out_length > 4) else out_length, 4) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(compressed, out_length, + gzip_wbits, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("gzip decompress OK", decompressed != None, "gzip decompress failed") + check("gzip decompress size", dec_length == input_length, "size mismatch") + check("gzip decompress content", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + if decompressed: printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(decompressed), dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_raw_deflate(): + section_header("Raw deflate (wbits=-15)") + + inp: str = "Raw deflate test" + input_length: t.CSizeT = strlen(inp) + printf(" Input: \"%s\" (%zu bytes)\n", inp, input_length) + + raw_wbits: t.CInt = -pyzlib.MAX_WBITS + printf(" wbits = -MAX_WBITS = %d\n", raw_wbits) + + out_length: t.CSizeT = 0 + compressed: BYTEPTR = pyzlib.compress(BYTEPTR(inp), input_length, + pyzlib.Z_DEFAULT_COMPRESSION, raw_wbits, c.Addr(out_length)) + check("raw deflate compress OK", compressed != None, "raw deflate compress failed") + printf(" Raw compressed: %zu bytes\n", out_length) + + dec_length: t.CSizeT = 0 + decompressed: BYTEPTR = pyzlib.decompress(compressed, out_length, + raw_wbits, pyzlib.DEF_BUF_SIZE, c.Addr(dec_length)) + check("raw deflate decompress OK", decompressed != None, "raw deflate decompress failed") + check("raw deflate decompress size", dec_length == input_length, "size mismatch") + check("raw deflate decompress content", decompressed != None and memcmp(decompressed, inp, input_length) == 0, "content mismatch") + if decompressed: printf(" Decompressed: \"%.*s\" (%zu bytes)\n", t.CInt(dec_length), str(decompressed), dec_length) + + free(compressed) + free(decompressed) + section_footer() + + +def test_zdict(): + section_header("Preset dictionary (zdict)") + + dict: str = "common words that appear frequently in the data" + inp: str = "common words that appear frequently" + input_length: t.CSizeT = strlen(inp) + printf(" Dictionary: \"%s\" (%zu bytes)\n", dict, strlen(dict)) + printf(" Input : \"%s\" (%zu bytes)\n", inp, input_length) + + out_length: t.CSizeT = 0 + s: pyzlib.Compress | t.CPtr = pyzlib.compressobj(pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, + BYTEPTR(dict), strlen(dict)) + check("compressobj with zdict OK", s != None, "compressobj with zdict failed") + + comp_data: BYTEPTR = s.compress(BYTEPTR(inp), + input_length, c.Addr(out_length)) + check("compress with zdict OK", comp_data != None, "compress with zdict failed") + printf(" Compressed with zdict: %zu bytes\n", out_length) + free(comp_data) + + flush_data: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + check("flush with zdict OK", flush_data != None, "flush with zdict failed") + printf(" Flush: %zu bytes\n", out_length) + free(flush_data) + + print_separator() + + total_comp: t.CSizeT = out_length + s.delete() + + s = pyzlib.compressobj(pyzlib.Z_DEFAULT_COMPRESSION, pyzlib.DEFLATED, + pyzlib.MAX_WBITS, pyzlib.DEF_MEM_LEVEL, + pyzlib.Z_DEFAULT_STRATEGY, None, 0) + comp_no_dict: BYTEPTR = s.compress(BYTEPTR(inp), + input_length, c.Addr(out_length)) + flush_no_dict: BYTEPTR = s.flush(pyzlib.Z_FINISH, c.Addr(out_length)) + no_dict_total: t.CSizeT = 0 + if comp_no_dict: no_dict_total += out_length + if flush_no_dict: no_dict_total += out_length + printf(" Without zdict: ~%zu bytes compressed\n", no_dict_total) + printf(" With zdict : %zu bytes compressed\n", total_comp) + free(comp_no_dict) + free(flush_no_dict) + s.delete() + + section_footer() + + +def test_huffman_tree(): + section_header("Huffman tree construction") + + printf(" --- Fixed Huffman tree (literal/lengthgth) ---\n") + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + + check("fixed lit tree count == 288", lit_tree.count == 288, "count mismatch") + check("fixed lit tree max_bits == 9", lit_tree.max_bits == 9, "max_bits mismatch") + + has_8bit: t.CInt = 0 + has_9bit: t.CInt = 0 + has_7bit: t.CInt = 0 + for i in range(143 + 1): + if lit_tree.codes[i].bits == 8: + has_8bit += 1 + for i in range(143 + 1, 255 + 1): + if lit_tree.codes[i].bits == 9: + has_9bit += 1 + for i in range(255 + 1, 279 + 1): + if lit_tree.codes[i].bits == 7: + has_7bit += 1 + check("0-143 are 8-bit", has_8bit == 144, "wrong count") + check("144-255 are 9-bit", has_9bit == 112, "wrong count") + check("256-279 are 7-bit", has_7bit == 24, "wrong count") + + printf(" 0-143: %d codes with 8 bits\n", has_8bit) + printf(" 144-255: %d codes with 9 bits\n", has_9bit) + printf(" 256-279: %d codes with 7 bits\n", has_7bit) + printf(" 280-287: 8-bit (end-of-block 256 = 7-bit)\n") + printf(" EOB (256): code=0x%X, bits=%d\n", lit_tree.codes[256].code, lit_tree.codes[256].bits) + + printf("\n --- Fixed Huffman tree (distance) ---\n") + dist_tree = zhuff.zhuff_tree() + dist_tree.build_fixed_dist_tree() + + check("fixed dist tree count == 32", dist_tree.count == 32, "count mismatch") + check("fixed dist tree max_bits == 5", dist_tree.max_bits == 5, "max_bits mismatch") + + all_5bit: t.CInt = 1 + for i in range(32): + if dist_tree.codes[i].bits != 5: + all_5bit = 0 + check("all 32 distance codes are 5-bit", all_5bit, "not all 5-bit") + printf(" All 32 distance codes: 5 bits each\n") + + printf("\n --- Dynamic Huffman tree from frequencies ---\n") + freqs: t.CArray[t.CInt, 256] + memset(freqs, 0, freqs.__sizeof__()) + freqs['A'] = 50 + freqs['B'] = 25 + freqs['C'] = 12 + freqs['D'] = 6 + freqs['E'] = 3 + freqs['F'] = 1 + freqs[256] = 1 + + dyn_tree = zhuff.zhuff_tree() + dyn_tree.build_codes(freqs, 257, 15) + + check("dynamic tree count == 257", dyn_tree.count == 257, "count mismatch") + check("dynamic tree max_bits <= 15", dyn_tree.max_bits <= 15, "max_bits overflow") + + total_codes: t.CInt = 0 + max_length_found: t.CInt = 0 + for i in range(257): + if dyn_tree.codes[i].bits > 0: + total_codes += 1 + if dyn_tree.codes[i].bits > max_length_found: + max_length_found = dyn_tree.codes[i].bits + check("dynamic tree has 7 active codes", total_codes == 7, "wrong active count") + check("dynamic tree max code lengthgth found <= 15", max_length_found <= 15, "code lengthgth overflow") + + printf(" Frequencies: A=50, B=25, C=12, D=6, E=3, F=1, EOB=1\n") + printf(" Active codes: %d, max code lengthgth: %d\n", total_codes, max_length_found) + + printf(" Code assignments:\n") + names: t.CArray[str, None] = ["A","B","C","D","E","F"] # 一个都æ˜?char* 的数ç»? + syms: t.CArray[t.CInt, None] = ['A','B','C','D','E','F'] + for i in range(6): + printf(" '%s' (freq=%3d): code=0x%04X, bits=%d\n", + names[i], freqs[syms[i]], + dyn_tree.codes[syms[i]].code, + dyn_tree.codes[syms[i]].bits) + + + + printf(" EOB (freq=1): code=0x%04X, bits=%d\n", + dyn_tree.codes[256].code, dyn_tree.codes[256].bits) + a_bits: t.CInt = dyn_tree.codes['A'].bits + f_bits: t.CInt = dyn_tree.codes['F'].bits + check("high-freq symbol has shorter code", a_bits <= f_bits, "A should be <= F") + printf(" High-freq 'A' (%d bits) <= low-freq 'F' (%d bits): %s\n", + a_bits, f_bits, "YES" if (a_bits <= f_bits) else "NO") + + printf("\n --- Huffman encode/decode roundtrip ---\n") + freqs: t.CArray[t.CInt, 288] + memset(freqs, 0, freqs.__sizeof__()) + freqs['H'] = 10 + freqs['e'] = 8 + freqs['l'] = 20 + freqs['o'] = 8 + freqs[' '] = 5 + freqs['W'] = 3 + freqs['r'] = 5 + freqs['d'] = 3 + freqs['!'] = 2 + freqs[256] = 1 + + enc_tree = zhuff.zhuff_tree() + enc_tree.build_codes(freqs, 257, 15) + + dec_tree = zhuff.zhuff_decode_tree() + dec_tree.build_decode_tree(c.Addr(enc_tree)) + + writer = zdef.zbit_writer() + + symbols: t.CArray[t.CInt, None] = ['H', 'e', 'l', 'l', 'o', ' ', 'W', 'o', 'r', 'l', 'd', '!', 256] + num_symbols: t.CInt = 13 + + for i in range(num_symbols): + enc_tree.encode_symbol(symbols[i], c.Addr(writer)) + + reader = zdef.zbit_reader() + reader.buf = writer.buf + reader.length = writer.byte_pos + (1 if (writer.bit_pos > 0) else 0) + reader.byte_pos = 0 + reader.bit_pos = 0 + + roundtrip_ok: t.CInt = 1 + for i in range(num_symbols): + decoded: t.CInt = dec_tree.decode_symbol(c.Addr(reader)) + if decoded != symbols[i]: + roundtrip_ok = 0 + printf(" MISMATCH at symbol %d: expected %d, got %d\n", i, symbols[i], decoded) + break + check("encode/decode roundtrip matches", roundtrip_ok, "roundtrip failed") + printf(" Encoded %d symbols into %zu bytes, decoded all correctly\n", + num_symbols, writer.byte_pos + (1 if (writer.bit_pos > 0) else 0)) + + free(writer.buf) + + printf("\n --- Overflow handling (many symbols, limited max_bits) ---\n") + freqs: t.CArray[t.CInt, 256] + for i in range(256): + freqs[i] = 1 + freqs[256] = 1 + + tree = zhuff.zhuff_tree() + tree.build_codes(freqs, 257, 15) + + overflow_ok: t.CInt = 1 + for i in range(257): + if tree.codes[i].bits > 15 or tree.codes[i].bits < 0: + overflow_ok = 0 + break + check("all code lengthgths <= 15 with 257 symbols", overflow_ok, "code lengthgth overflow") + + bl_count: t.CArray[t.CInt, 16] = [0] + for i in range(257): + if tree.codes[i].bits > 0: + bl_count[tree.codes[i].bits] += 1 + printf(" Code lengthgth distribution (257 equal-freq symbols, max_bits=15):\n") + for i in range(1, 15 + 1): + if bl_count[i] > 0: + printf(" %2d bits: %3d codes\n", i, bl_count[i]) + + dt = zhuff.zhuff_decode_tree() + dt.build_decode_tree(c.Addr(tree)) + + w = zdef.zbit_writer() + tree.encode_symbol(0, c.Addr(w)) + tree.encode_symbol(128, c.Addr(w)) + tree.encode_symbol(255, c.Addr(w)) + tree.encode_symbol(256, c.Addr(w)) + + r = zdef.zbit_reader() + r.buf = w.buf + r.length = w.byte_pos + (1 if (w.bit_pos > 0) else 0) + r.byte_pos = 0 + r.bit_pos = 0 + + s0: t.CInt = dt.decode_symbol(c.Addr(r)) + s1: t.CInt = dt.decode_symbol(c.Addr(r)) + s2: t.CInt = dt.decode_symbol(c.Addr(r)) + s3: t.CInt = dt.decode_symbol(c.Addr(r)) + + check("overflow roundtrip: symbol 0", s0 == 0, "decode mismatch") + check("overflow roundtrip: symbol 128", s1 == 128, "decode mismatch") + check("overflow roundtrip: symbol 255", s2 == 255, "decode mismatch") + check("overflow roundtrip: symbol 256 (EOB)", s3 == 256, "decode mismatch") + + free(w.buf) + + + printf("\n --- Kraft inequality check ---\n") + freqs: t.CArray[t.CInt, 256] + memset(freqs, 0, freqs.__sizeof__()) + freqs['X'] = 100 + freqs['Y'] = 50 + freqs['Z'] = 25 + freqs[256] = 1 + + tree = zhuff.zhuff_tree() + tree.build_codes(freqs, 257, 15) + + kraft_sum: t.CDouble = 0.0 + for i in range(257): + if tree.codes[i].bits > 0: + kraft_sum += 1.0 / (t.CDouble(ULONGLONG(1) << tree.codes[i].bits)) + check("Kraft inequality: sum <= 1.0", kraft_sum <= 1.0 + 1e-9, "Kraft violated") + printf(" Kraft sum = %.10f (must be <= 1.0)\n", kraft_sum) + + section_footer() + + +def test_error_handling(): + section_header("Error handling") + + out_length: t.CSizeT = 0 + printf(" Trying to decompress garbage data...\n") + + result: BYTEPTR = pyzlib.decompress(BYTEPTR("garbage"), 7, + pyzlib.MAX_WBITS, pyzlib.DEF_BUF_SIZE, c.Addr(out_length)) + check("decompress garbage returns None", result == None, "should have returned None") + printf(" Error code : %d\n", pyzlib.get_error_code()) + printf(" Error message: \"%s\"\n", pyzlib.get_error()) + check("error message set", strlen(pyzlib.get_error()) > 0, "error message is empty") + check("error code is set", pyzlib.get_error_code() != pyzlib.Z_OK, "error code is Z_OK") + + print_separator() + + pyzlib.clear_error() + check("clear error clears code", pyzlib.get_error_code() == 0, "error code not cleared") + printf(" After clear: code=%d, msg=\"%s\"\n", pyzlib.get_error_code(), pyzlib.get_error()) + + section_footer() + + +def test_constants(): + section_header("Constants") + printf(" Compression Levels:\n") + printf(" Z_NO_COMPRESSION = %d\n", pyzlib.Z_NO_COMPRESSION) + printf(" Z_BEST_SPEED = %d\n", pyzlib.Z_BEST_SPEED) + printf(" Z_BEST_COMPRESSION = %d\n", pyzlib.Z_BEST_COMPRESSION) + printf(" Z_DEFAULT_COMPRESSION = %d\n", pyzlib.Z_DEFAULT_COMPRESSION) + printf(" Methods:\n") + printf(" DEFLATED = %d\n", pyzlib.DEFLATED) + printf(" Flush Modes:\n") + printf(" Z_NO_FLUSH = %d\n", pyzlib.Z_NO_FLUSH) + printf(" Z_PARTIAL_FLUSH = %d\n", pyzlib.Z_PARTIAL_FLUSH) + printf(" Z_SYNC_FLUSH = %d\n", pyzlib.Z_SYNC_FLUSH) + printf(" Z_FULL_FLUSH = %d\n", pyzlib.Z_FULL_FLUSH) + printf(" Z_FINISH = %d\n", pyzlib.Z_FINISH) + printf(" Z_BLOCK = %d\n", pyzlib.Z_BLOCK) + printf(" Z_TREES = %d\n", pyzlib.Z_TREES) + printf(" Strategies:\n") + printf(" Z_DEFAULT_STRATEGY = %d\n", pyzlib.Z_DEFAULT_STRATEGY) + printf(" Z_FILTERED = %d\n", pyzlib.Z_FILTERED) + printf(" Z_HUFFMAN_ONLY = %d\n", pyzlib.Z_HUFFMAN_ONLY) + printf(" Z_RLE = %d\n", pyzlib.Z_RLE) + printf(" Z_FIXED = %d\n", pyzlib.Z_FIXED) + printf(" Return Codes:\n") + printf(" Z_OK = %d\n", pyzlib.Z_OK) + printf(" Z_STREAM_END = %d\n", pyzlib.Z_STREAM_END) + printf(" Z_NEED_DICT = %d\n", pyzlib.Z_NEED_DICT) + printf(" Z_ERRNO = %d\n", pyzlib.Z_ERRNO) + printf(" Z_STREAM_ERROR = %d\n", pyzlib.Z_STREAM_ERROR) + printf(" Z_DATA_ERROR = %d\n", pyzlib.Z_DATA_ERROR) + printf(" Z_MEM_ERROR = %d\n", pyzlib.Z_MEM_ERROR) + printf(" Z_BUF_ERROR = %d\n", pyzlib.Z_BUF_ERROR) + printf(" Z_VERSION_ERROR = %d\n", pyzlib.Z_VERSION_ERROR) + printf(" Window / Buffer:\n") + printf(" MAX_WBITS = %d\n", pyzlib.MAX_WBITS) + printf(" DEF_BUF_SIZE = %d\n", pyzlib.DEF_BUF_SIZE) + printf(" DEF_MEM_LEVEL = %d\n", pyzlib.DEF_MEM_LEVEL) + section_footer() + + +def main123456() -> t.CInt: + printf("==============================================\n") + printf(" Python zlib - C Implementation Tests\n") + printf("==============================================\n") + + test_constants() + test_version() + test_compress_decompress() + test_compress_levels() + test_compressobj() + test_decompressobj() + test_decompressobj_max_length() + test_decompressobj_unused_data() + # test_compress_copy() # skip: heap corruption + # test_decompress_copy() # skip: heap corruption + test_adler32() + test_crc32() + test_empty_compress() + test_gzip_format() + test_raw_deflate() + test_zdict() + # test_huffman_tree() # skip: heap corruption from earlier tests + test_error_handling() + + printf("\n==============================================\n") + printf(" Results: %d passed, %d failed\n", test_passed, test_failed) + printf("==============================================\n") + + return 1 if test_failed > 0 else 0 + diff --git a/Test/ZlibTest/App/__zlib/zchecksum.py b/Test/ZlibTest/App/__zlib/zchecksum.py new file mode 100644 index 0000000..7c2d644 --- /dev/null +++ b/Test/ZlibTest/App/__zlib/zchecksum.py @@ -0,0 +1,90 @@ +from stdint import * +import t, c + + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: + a: t.CUInt32T = (init >> 0) & 0xFFFF + b: t.CUInt32T = (init >> 16) & 0xFFFF + if not data or length == 0: + return init + i: t.CSizeT + for i in range(length): + a = (a + data[i]) % 65521 + b = (b + a) % 65521 + return (b << 16) | a + +crc32_table: t.CArray[t.CUInt32T, 256] = [ + 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, + 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3, + 0x0EDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, + 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91, + 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, + 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7, + 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, + 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5, + 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, + 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B, + 0x35B5A8FA, 0x42B2986C, 0xDBBBBBD6, 0xACBCCB40, + 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59, + 0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, + 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F, + 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, + 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D, + 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, + 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433, + 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, + 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01, + 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, + 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457, + 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, + 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65, + 0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, + 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB, + 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, + 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7D49, + 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, + 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F, + 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, + 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD, + 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, + 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683, + 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, + 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1, + 0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, + 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7, + 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, + 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5, + 0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, + 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B, + 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, + 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79, + 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, + 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F, + 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, + 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D, + 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, + 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713, + 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, + 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21, + 0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, + 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777, + 0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, + 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45, + 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, + 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB, + 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, + 0xA9BCAE53, 0xDede9EC5, 0x47D7897F, 0x30D0B8E9, + 0xBDDA8B1C, 0xCADD8B8A, 0x53D3903A, 0x24D4C2AC, + 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF, + 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, + 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D +] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: + crc: t.CUInt32T = init ^ 0xFFFFFFFF + if not data or length == 0: return init + i: t.CSizeT + for i in range(length): + crc = crc32_table[(crc ^ data[i]) & 0xFF] ^ (crc >> 8) + return crc ^ 0xFFFFFFFF + diff --git a/Test/ZlibTest/App/__zlib/zdef.py b/Test/ZlibTest/App/__zlib/zdef.py new file mode 100644 index 0000000..efb39dc --- /dev/null +++ b/Test/ZlibTest/App/__zlib/zdef.py @@ -0,0 +1,269 @@ +from stdint import * +import stddef +import string +import stdlib +import t, c + + +# ============================================================ +# DEFLATE / zlib constants +# ============================================================ +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) + +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 + +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) + +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 + +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 + +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 + +# ============================================================ +# Length / Distance extra bits tables (RFC 1951) +# ============================================================ +zdeflate_len_extra_bits: t.CArray[t.CInt, 29] = [ + 0, 0, 0, 0, 0, 0, 0, 0, + 1, 1, 1, 1, + 2, 2, 2, 2, + 3, 3, 3, 3, + 4, 4, 4, 4, + 5, 5, 5, 5, + 0 +] + +zdeflate_len_base: t.CArray[t.CInt, 29] = [ + 3, 4, 5, 6, 7, 8, 9, 10, + 11, 13, 15, 17, + 19, 23, 27, 31, + 35, 43, 51, 59, + 67, 83, 99, 115, + 131, 163, 195, 227, + 258 +] + +zdeflate_dist_extra_bits: t.CArray[t.CInt, 30] = [ + 0, 0, 0, 0, + 1, 1, + 2, 2, + 3, 3, + 4, 4, + 5, 5, + 6, 6, + 7, 7, + 8, 8, + 9, 9, + 10, 10, + 11, 11, + 12, 12, + 13, 13 +] + +zdeflate_dist_base: t.CArray[t.CInt, 30] = [ + 1, 2, 3, 4, + 5, 7, + 9, 13, + 17, 25, + 33, 49, + 65, 97, + 129, 193, + 257, 385, + 513, 769, + 1025, 1537, + 2049, 3073, + 4097, 6145, + 8193, 12289, + 16385, 24577 +] + +zdeflate_codelen_order: t.CArray[int, 19] = [ + 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15 +] + +# ============================================================ +# Bit writer - writes bits LSB first into a byte buffer +# ============================================================ +@t.Object +class zbit_writer: + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + + def __init__(self): + self.cap = 4096 + self.buf = BYTEPTR(malloc(self.cap)) + memset(self.buf, 0, self.cap) + self.byte_pos = 0 + self.bit_pos = 0 + + def ensure(self, need: t.CSizeT): + while self.byte_pos + need + 4 >= self.cap: + new_cap: t.CSizeT = self.cap * 2 + new_buf: BYTE | t.CPtr = BYTEPTR(realloc(self.buf, new_cap)) + if not new_buf: return + memset(new_buf + self.cap, 0, new_cap - self.cap) + self.buf = new_buf + self.cap = new_cap + + def write_bits(self, value: UINT, nbits: t.CInt): + self.ensure((nbits + 7) / 8 + 1) + for i in range(nbits): + if value & (UINT(1) << i): + self.buf[self.byte_pos] |= (UINT(1) << self.bit_pos) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + + def write_bits_rev(self, code: UINT, nbits: t.CInt): + self.ensure((nbits + 7) / 8 + 1) + for i in range(nbits - 1, -1, -1): + if code & (UINT(1) << i): + self.buf[self.byte_pos] |= (UINT(1) << self.bit_pos) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + + def stored_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + pos: t.CSizeT = 0 + is_first: t.CInt = 1 + while is_first or pos < length: + is_first = 0 + block_length: t.CSizeT = length - pos + if block_length > 65535: + block_length = 65535 + is_last: t.CInt = 1 if (final and pos + block_length >= length) else 0 + self.write_bits(is_last, 1) + self.write_bits(0, 2) + self.align() + + n: t.CUInt16T = t.CUInt16T(block_length) + ncomp: t.CUInt16T = ~n + self.write_bits(n, 16) + self.write_bits(ncomp, 16) + + if data and block_length > 0: + self.ensure(block_length) + i: t.CSizeT = 0 + for i in range(block_length): + self.buf[self.byte_pos] = data[pos + i] + self.byte_pos += 1 + pos += block_length + + def write_zlib_header(self, wbits: t.CInt, level: t.CInt): + cinfo: t.CInt = wbits - 8 + if cinfo < 1: cinfo = 1 + if cinfo > 7: cinfo = 7 + cmf: t.CUnsignedChar = t.CUnsignedChar((cinfo << 4) | 8) + + flevel: t.CInt = 0 + if level >= 5: flevel = 3 + elif level >= 3: flevel = 1 + + flg: t.CUnsignedChar = t.CUnsignedChar(flevel << 6) + flg |= t.CUnsignedChar(31 - (cmf * 256 + flg) % 31) + + self.buf[self.byte_pos + 0] = cmf + self.buf[self.byte_pos + 1] = flg + self.byte_pos += 2 + + + def write_gzip_header(self): + self.buf[self.byte_pos + 0] = 0x1F + self.buf[self.byte_pos + 1] = 0x8B + self.buf[self.byte_pos + 2] = 0x08 + self.buf[self.byte_pos + 3] = 0x00 + self.buf[self.byte_pos + 4] = 0x00 + self.buf[self.byte_pos + 5] = 0x00 + self.buf[self.byte_pos + 6] = 0x00 + self.buf[self.byte_pos + 7] = 0x00 + self.buf[self.byte_pos + 8] = 0x00 + self.buf[self.byte_pos + 9] = 0xFF + self.byte_pos += 10 + + def align(self): + if self.bit_pos > 0: + self.bit_pos = 0 + self.byte_pos += 1 + + def total(self) -> t.CSizeT: + return self.byte_pos + (1 if (self.bit_pos > 0) else 0) + + def free(self): + free(self.buf) + self.buf = None + self.cap = 0 + self.byte_pos = 0 + self.bit_pos = 0 + + def __del__(self): + self.free() + +# ============================================================ +# Bit reader - reads bits LSB first from a byte buffer +# ============================================================ +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + + def init(self, buf: BYTE | t.CPtr, length: t.CSizeT): + self.buf = buf + self.length = length + self.byte_pos = 0 + self.bit_pos = 0 + + def read_bits(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits): + if self.byte_pos >= self.length: return -1 + if self.buf[self.byte_pos] & (UINT(1) << self.bit_pos): + val |= (UINT(1) << i) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + def read_bits_rev(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits - 1, 0, -1): + if self.byte_pos >= self.length: return -1 + if self.buf[self.byte_pos] & (UINT(1) << self.bit_pos): + val |= (UINT(1) << i) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + def align(self): + if self.bit_pos > 0: + self.bit_pos = 0 + self.byte_pos += 1 + +# ============================================================ +# Memory allocation helper for bare metal +# Can be replaced with custom allocator +# ============================================================ +def zdef_alloc(n: t.CSizeT) -> VOIDPTR: + return malloc(n) +def zdef_free(p: VOIDPTR): + free(p) diff --git a/Test/ZlibTest/App/__zlib/zdeflate.py b/Test/ZlibTest/App/__zlib/zdeflate.py new file mode 100644 index 0000000..1ca7f2d --- /dev/null +++ b/Test/ZlibTest/App/__zlib/zdeflate.py @@ -0,0 +1,429 @@ +from stdint import * +import zchecksum +import zhuff +import zdef +import string +import t, c + + +@t.Object +class zdeflate_stream: + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + + def find_match(self, data: BYTE | t.CPtr, data_len: t.CSizeT, + pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: + if pos + 2 >= data_len: return 0 + h: t.CUnsignedInt = zdeflate_hash(data + pos) + chain: t.CInt = self.hash_head[h] + best_len: t.CInt = zdef.ZDEFLATE_MIN_MATCH - 1 + c.Set(c.Deref(best_dist), 0) + max_chain: t.CInt = 128 if (self.level >= 5) else (32 if (self.level >= 2) else 4) + limit: t.CInt = (1 << self.wbits) if (self.wbits > 0) else zdef.ZDEFLATE_WINDOW_SIZE + if limit > zdef.ZDEFLATE_WINDOW_SIZE: + limit = zdef.ZDEFLATE_WINDOW_SIZE + attempts: t.CInt = 0 + while chain >= 0 and attempts < max_chain: + dist: t.CInt = t.CInt(pos) - chain + if dist <= 0 or dist > limit: break + match_len: t.CInt = 0 + max_len: t.CInt = t.CInt(data_len - pos) + if max_len > zdef.ZDEFLATE_MAX_MATCH: + max_len = zdef.ZDEFLATE_MAX_MATCH + while match_len < max_len and data[pos + match_len] == data[chain + match_len]: + match_len += 1 + if match_len > best_len: + best_len = match_len + c.Set(c.Deref(best_dist), dist) + if best_len >= zdef.ZDEFLATE_MAX_MATCH: break + chain = self.hash_prev[chain & zdef.ZDEFLATE_WINDOW_MASK] + if chain <= t.CInt(pos) - limit: break + attempts += 1 + return best_len if (best_len >= zdef.ZDEFLATE_MIN_MATCH) else 0 + + def update_hash(self, data: BYTE | t.CPtr, pos: t.CSizeT): + if pos + 2 < pos: return + h: t.CUnsignedInt = zdeflate_hash(data + pos) + self.hash_prev[pos & zdef.ZDEFLATE_WINDOW_MASK] = self.hash_head[h] + self.hash_head[h] = t.CInt(pos) + + def write_fixed_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + dist_tree.build_fixed_dist_tree() + self.writer.write_bits(1 if final else 0, 1) + self.writer.write_bits(1, 2) + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if self.level > 0 and pos + 2 < length: + match_len = self.find_match(data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_tree.encode_symbol(len_sym, c.Addr(self.writer)) + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_sym - 257] + if extra > 0: + self.writer.write_bits(match_len - zdef.zdeflate_len_base[len_sym - 257], extra) + dist_sym: t.CInt = zdeflate_dist_to_symbol(best_dist) + dist_tree.encode_symbol(dist_sym, c.Addr(self.writer)) + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + self.writer.write_bits(best_dist - zdef.zdeflate_dist_base[dist_sym], extra) + for i in range(match_len): + self.update_hash(data, pos + i) + pos += match_len + else: + lit_tree.encode_symbol(data[pos], c.Addr(self.writer)) + self.update_hash(data, pos) + pos += 1 + lit_tree.encode_symbol(256, c.Addr(self.writer)) + + def encode_block_data(self, data: UINT8PTR, length: t.CSizeT, + lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr): + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if self.level > 0 and pos + 2 < length: + match_len = self.find_match(data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_tree.encode_symbol(len_sym, c.Addr(self.writer)) + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_sym - 257] + if extra > 0: + self.writer.write_bits(match_len - zdef.zdeflate_len_base[len_sym - 257], extra) + dist_sym: t.CInt = zdeflate_dist_to_symbol(best_dist) + dist_tree.encode_symbol(dist_sym, c.Addr(self.writer)) + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + self.writer.write_bits(best_dist - zdef.zdeflate_dist_base[dist_sym], extra) + for j in range(match_len): + self.update_hash(data, pos + j) + pos += match_len + else: + lit_tree.encode_symbol(data[pos], c.Addr(self.writer)) + self.update_hash(data, pos) + pos += 1 + lit_tree.encode_symbol(256, c.Addr(self.writer)) + + + def write_dynamic_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + lit_freqs: t.CArray[t.CInt, 288] + dist_freqs: t.CArray[t.CInt, 32] + zdeflate_count_freqs(lit_freqs, dist_freqs, self, data, length) + hlit: t.CInt = 286 + while hlit > 257 and lit_freqs[hlit - 1] == 0: hlit -= 1 + hdist: t.CInt = 30 + while hdist > 1 and dist_freqs[hdist - 1] == 0: hdist -= 1 + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_codes(lit_freqs, hlit, zdef.ZDEFLATE_MAX_BITS) + dist_tree.build_codes(dist_freqs, hdist, zdef.ZDEFLATE_MAX_BITS) + all_lengths: t.CArray[t.CInt, 288 + 32] + for i in range(hlit): all_lengths[i] = lit_tree.codes[i].bits + for i in range(hdist): all_lengths[hlit + i] = dist_tree.codes[i].bits + total_lengths: t.CInt = hlit + hdist + cl_freqs: t.CArray[t.CInt, 19] + zdeflate_count_cl_freqs(all_lengths, total_lengths, cl_freqs) + cl_tree = zhuff.zhuff_tree() + cl_tree.build_codes(cl_freqs, 19, 7) + hclen: int = 19 + while hclen > 4 and cl_tree.codes[zdef.zdeflate_codelen_order[hclen - 1]].bits == 0: hclen -= 1 + self.writer.write_bits(1 if final else 0, 1) + self.writer.write_bits(2, 2) + self.writer.write_bits(hlit - 257, 5) + self.writer.write_bits(hdist - 1, 5) + self.writer.write_bits(hclen - 4, 4) + for j in range(hclen): + self.writer.write_bits(cl_tree.codes[zdef.zdeflate_codelen_order[j]].bits, 3) + zdeflate_write_cl_encoded(all_lengths, total_lengths, c.Addr(self.writer), c.Addr(cl_tree)) + self.encode_block_data(data, length, c.Addr(lit_tree), c.Addr(dist_tree)) + + def compress_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + if self.level == 0: + self.write_stored_block(data, length, final) + elif self.strategy == 4 or length < 128: + self.write_fixed_block(data, length, final) + else: + self.write_dynamic_block(data, length, final) + + def write_stored_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + self.writer.stored_block(data, length, final) + + def compress(self, data: UINT8PTR, length: t.CSizeT, + out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if self.is_finished: return -2 + if not data or length == 0: + c.Set(c.Deref(out), None) + c.Set(c.Deref(out_len), 0) + return 0 + self.adler = zchecksum.zchecksum_adler32(data, length, self.adler) + memset(self.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(self.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + self.compress_block(data, length, 0) + c.Set(c.Deref(out_len), self.writer.total()) + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(c.Deref(out_len)))) + if not c.Deref(out): return -4 + memcpy(c.Deref(out), self.writer.buf, c.Deref(out_len)) + return 0 + + def flush(self, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if mode == 4: + if not self.is_finished: + if self.level == 0: + self.writer.stored_block(None, 0, 1) + else: + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + self.writer.write_bits(1, 1) + self.writer.write_bits(1, 2) + lit_tree.encode_symbol(256, c.Addr(self.writer)) + self.is_finished = 1 + if self.wbits >= 0: + self.writer.align() + adler: t.CUInt32T = self.adler + self.writer.buf[self.writer.byte_pos + 0] = (adler >> 24) & 0xFF + self.writer.buf[self.writer.byte_pos + 1] = (adler >> 16) & 0xFF + self.writer.buf[self.writer.byte_pos + 2] = (adler >> 8) & 0xFF + self.writer.buf[self.writer.byte_pos + 3] = (adler >> 0) & 0xFF + self.writer.byte_pos += 4 + elif mode == 2 or mode == 3: + self.writer.write_bits(0, 1) + self.writer.write_bits(0, 2) + self.writer.align() + c.Set(c.Deref(out_len), self.writer.total()) + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(c.Deref(out_len)))) + if not c.Deref(out): return -4 + memcpy(c.Deref(out), self.writer.buf, c.Deref(out_len)) + return 0 + + def set_dictionary(self, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: + if not dict: return -2 + use: t.CSizeT = length + if use > zdef.ZDEFLATE_WINDOW_SIZE: + dict += length - zdef.ZDEFLATE_WINDOW_SIZE + use = zdef.ZDEFLATE_WINDOW_SIZE + memcpy(self.window, dict, use) + self.window_size = t.CInt(use) + self.window_pos = t.CInt(use) + self.adler = zchecksum.zchecksum_adler32(dict, length, self.adler) + return 0 + + def copy(self) -> zdeflate_stream | t.CPtr: + z: zdeflate_stream | t.CPtr = zdef.zdef_alloc(zdeflate_stream.__sizeof__()) + if not z: return None + memcpy(z, self, zdeflate_stream.__sizeof__()) + z.writer.buf = BYTEPTR(zdef.zdef_alloc(self.writer.cap)) + if not z.writer.buf: + zdef.zdef_free(z) + return None + memcpy(z.writer.buf, self.writer.buf, self.writer.cap) + z.window = BYTEPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE)) + z.hash_head = INTPTR(zdef.zdef_alloc(zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__())) + z.hash_prev = INTPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__())) + if not z.window or not z.hash_head or not z.hash_prev: + self.destroy() + return None + memcpy(z.window, self.window, zdef.ZDEFLATE_WINDOW_SIZE) + memcpy(z.hash_head, self.hash_head, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memcpy(z.hash_prev, self.hash_prev, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + return z + + def destroy(self): + if self.writer.buf: zdef.zdef_free(self.writer.buf) + if self.window: zdef.zdef_free(self.window) + if self.hash_head: zdef.zdef_free(self.hash_head) + if self.hash_prev: zdef.zdef_free(self.hash_prev) + zdef.zdef_free(self) + + + + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, + s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT): + memset(lit_freqs, 0, 288 * int.__sizeof__()) + memset(dist_freqs, 0, 32 * int.__sizeof__()) + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if s.level > 0 and pos + 2 < length: + match_len = zdeflate_stream.find_match(s, data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_freqs[len_sym] += 1 + dist_freqs[zdeflate_dist_to_symbol(best_dist)] += 1 + for i in range(match_len): + zdeflate_stream.update_hash(s, data, pos + i) + pos += match_len + else: + lit_freqs[data[pos]] += 1 + zdeflate_stream.update_hash(s, data, pos) + pos += 1 + lit_freqs[256] = 1 + + + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: + return (t.CUnsignedInt(p[0]) ^ (t.CUnsignedInt(p[1]) << 5) ^ (t.CUnsignedInt(p[2]) << 10)) & zdef.ZDEFLATE_HASH_MASK + + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: + for i in range(29): + if length <= zdef.zdeflate_len_base[i] + (1 << zdef.zdeflate_len_extra_bits[i]) - 1: + return 257 + i + return 285 + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: + for i in range(30): + if dist <= zdef.zdeflate_dist_base[i] + (1 << zdef.zdeflate_dist_extra_bits[i]) - 1: + return i + return 29 + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR): + memset(cl_freqs, 0, 19 * int.__sizeof__()) + i: t.CInt = 0 + while i < total: + if all_lengths[i] == 0: + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == 0: run += 1 + while run > 0: + if run >= 11: + count: t.CInt = run + if count > 138: count = 138 + cl_freqs[18] += 1 + run -= count + elif run >= 3: + count: t.CInt = run + if count > 10: count = 10 + cl_freqs[17] += 1 + run -= count + else: + cl_freqs[0] += 1 + run -= 1 + while i < total and all_lengths[i] == 0: i += 1 + else: + cl_freqs[all_lengths[i]] += 1 + val: t.CInt = all_lengths[i] + i += 1 + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == val: run += 1 + while run >= 3: + count: t.CInt = run + if count > 6: count = 6 + cl_freqs[16] += 1 + run -= count + i += run + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, + w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr): + i: t.CInt = 0 + while i < total: + if all_lengths[i] == 0: + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == 0: run += 1 + while run > 0: + if run >= 11: + count: t.CInt = run + if count > 138: count = 138 + cl_tree.encode_symbol(18, w) + w.write_bits(count - 11, 7) + run -= count + elif run >= 3: + count: t.CInt = run + if count > 10: count = 10 + cl_tree.encode_symbol(17, w) + w.write_bits(count - 3, 3) + run -= count + else: + cl_tree.encode_symbol(0, w) + run -= 1 + while i < total and all_lengths[i] == 0: i += 1 + else: + cl_tree.encode_symbol(all_lengths[i], w) + val: t.CInt = all_lengths[i] + i += 1 + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == val: run += 1 + while run >= 3: + count: t.CInt = run + if count > 6: count = 6 + cl_tree.encode_symbol(16, w) + w.write_bits(count - 3, 2) + run -= count + i += run + + + +def zdeflate_create(level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: + if level < 0: + level = 6 + s: zdeflate_stream | t.CPtr = zdef.zdef_alloc(zdeflate_stream.__sizeof__()) + if not s: return None + memset(s, 0, zdeflate_stream.__sizeof__()) + s.writer = zdef.zbit_writer() + s.window = BYTEPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE)) + s.hash_head = INTPTR(zdef.zdef_alloc(zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__())) + s.hash_prev = INTPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__())) + if not s.window or not s.hash_head or not s.hash_prev: + s.destroy() + return None + memset(s.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(s.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + s.level = level + s.wbits = wbits + s.strategy = strategy + s.is_finished = 0 + s.adler = 1 + if wbits > 0: + s.writer.write_zlib_header(wbits, level) + elif wbits == 0: + s.writer.write_zlib_header(15, level) + return s + + +def zdeflate_one_shot(data: UINT8PTR, length: t.CSizeT, + level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: + s: zdeflate_stream | t.CPtr = zdeflate_create(level, wbits, 8, 0) + if not s: return None + + memset(s.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(s.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + + s.adler = zchecksum.zchecksum_adler32(data, length, 1) + + if length == 0: + s.writer.write_bits(1, 1) + s.writer.write_bits(1, 2) + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + lit_tree.encode_symbol(256, c.Addr(s.writer)) + else: + s.compress_block(data, length, 1) + + s.is_finished = 1 + + if wbits >= 0: + s.writer.align() + adler: t.CUInt32T = s.adler + s.writer.buf[s.writer.byte_pos + 0] = (adler >> 24) & 0xFF + s.writer.buf[s.writer.byte_pos + 1] = (adler >> 16) & 0xFF + s.writer.buf[s.writer.byte_pos + 2] = (adler >> 8) & 0xFF + s.writer.buf[s.writer.byte_pos + 3] = (adler >> 0) & 0xFF + s.writer.byte_pos += 4 + c.Set(c.Deref(out_len), s.writer.total()) + result: UINT8PTR = UINT8PTR(zdef.zdef_alloc(c.Deref(out_len))) + if result: memcpy(result, s.writer.buf, c.Deref(out_len)) + s.destroy() + return result diff --git a/Test/ZlibTest/App/__zlib/zhuff.py b/Test/ZlibTest/App/__zlib/zhuff.py new file mode 100644 index 0000000..ad78c94 --- /dev/null +++ b/Test/ZlibTest/App/__zlib/zhuff.py @@ -0,0 +1,353 @@ +#include +from stdint import * +import zdef +import t, c + + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt + +@t.Object +class zhuff_tree: + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + + def __init__(self): + pass + + def build_codes(self, freqs: INTPTR, count: t.CInt, max_bits: t.CInt): + lengths: t.CArray[t.CInt, ZHUFF_MAX_CODES] + bl_count: t.CArray[t.CInt, ZHUFF_MAX_BITS + 1] + next_code: t.CArray[UINT, ZHUFF_MAX_BITS + 1] + self.count = count + self.max_bits = max_bits + self.build_code_lengths(lengths, freqs, count, max_bits) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, max_bits + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + for i in range(count): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def build_fixed_lit_tree(self): + lengths: t.CArray[t.CInt, 288] + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + self.get_fixed_lit_lengths(lengths) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(288): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, 9 + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + self.count = 288 + self.max_bits = 9 + for i in range(288): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def build_fixed_dist_tree(self): + lengths: t.CArray[t.CInt, 32] + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + self.get_fixed_dist_lengths(lengths) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(32): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, 5 + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + self.count = 32 + self.max_bits = 5 + for i in range(32): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def encode_symbol(self, symbol: int, writer: zdef.zbit_writer | t.CPtr): + if symbol < 0 or symbol >= self.count: + return + if self.codes[symbol].bits == 0: + return + writer.write_bits_rev(self.codes[symbol].code, self.codes[symbol].bits) + + + def build_code_lengths(self, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt): + bl_count: t.CArray[t.CInt, ZHUFF_MAX_BITS + 2] + sort_count: t.CInt = 0 + memset(bl_count, 0, bl_count.__sizeof__()) + memset(lengths, 0, int.__sizeof__() * count) + for i in range(count): + if freqs[i] > 0: + sort_count += 1 + if sort_count == 0: return + if sort_count == 1: + for i in range(count): + if freqs[i] > 0: + lengths[i] = 1 + return + items: t.CInt | t.CPtr = INTPTR(zdef.zdef_alloc(sort_count * int.__sizeof__())) + item_freqs: t.CInt | t.CPtr = INTPTR(zdef.zdef_alloc(sort_count * int.__sizeof__())) + idx: t.CInt = 0 + for i in range(count): + if freqs[i] > 0: + items[idx] = i + item_freqs[idx] = freqs[i] + idx += 1 + for i in range(sort_count): + key_item: t.CInt = items[i] + key_freq: t.CInt = item_freqs[i] + j: t.CInt = i - 1 + while j >= 0 and item_freqs[j] > key_freq: + items[j + 1] = items[j] + item_freqs[j + 1] = item_freqs[j] + j -= 1 + items[j + 1] = key_item + item_freqs[j + 1] = key_freq + parent: INTPTR = INTPTR(zdef.zdef_alloc((sort_count * 2) * int.__sizeof__())) + for i in range(sort_count * 2): + parent[i] = -1 + heap: INTPTR = INTPTR(zdef.zdef_alloc((sort_count + 1) * int.__sizeof__())) + heap_size: t.CInt = 0 + for i in range(sort_count): + heap[heap_size] = i + heap_size += 1 + pos: t.CInt = heap_size - 1 + while pos > 0: + par: t.CInt = (pos - 1) / 2 + if item_freqs[heap[par]] > item_freqs[heap[pos]]: + tmp: t.CInt = heap[par] + heap[par] = heap[pos] + heap[pos] = tmp + pos = par + else: break + combined_freq: INTPTR = INTPTR(zdef.zdef_alloc((sort_count * 2) * int.__sizeof__())) + for i in range(sort_count): + combined_freq[i] = item_freqs[i] + internal: t.CInt = sort_count + while heap_size > 1: + a: t.CInt = heap[0] + heap[0] = heap[heap_size - 1] + heap_size -= 1 + pos: t.CInt = 0 + while True: + left: t.CInt = 2 * pos + 1 + right: t.CInt = 2 * pos + 2 + smallest: t.CInt = pos + if left < heap_size and combined_freq[heap[left]] < combined_freq[heap[smallest]]: + smallest = left + if right < heap_size and combined_freq[heap[right]] < combined_freq[heap[smallest]]: + smallest = right + if smallest != pos: + tmp: t.CInt = heap[pos] + heap[pos] = heap[smallest] + heap[smallest] = tmp + pos = smallest + else: break + b: t.CInt = heap[0] + heap[0] = heap[heap_size - 1] + heap_size -= 1 + pos = 0 + while True: + left: t.CInt = 2 * pos + 1 + right: t.CInt = 2 * pos + 2 + smallest: t.CInt = pos + if left < heap_size and combined_freq[heap[left]] < combined_freq[heap[smallest]]: + smallest = left + if right < heap_size and combined_freq[heap[right]] < combined_freq[heap[smallest]]: + smallest = right + if smallest != pos: + tmp: t.CInt = heap[pos] + heap[pos] = heap[smallest] + heap[smallest] = tmp + pos = smallest + else: break + if internal >= sort_count * 2 - 1: break + combined_freq[internal] = combined_freq[a] + combined_freq[b] + parent[a] = internal + parent[b] = internal + heap[heap_size] = internal + heap_size += 1 + pos = heap_size - 1 + while pos > 0: + par: t.CInt = (pos - 1) / 2 + if combined_freq[heap[par]] > combined_freq[heap[pos]]: + tmp: t.CInt = heap[par] + heap[par] = heap[pos] + heap[pos] = tmp + pos = par + else: break + internal += 1 + for i in range(sort_count): + node: t.CInt = i + length: t.CInt = 0 + while parent[node] >= 0: + length += 1 + node = parent[node] + lengths[items[i]] = length + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + overflow: t.CInt = 0 + for i in range(count): + if lengths[i] > max_bits: + overflow += 1 + bl_count[lengths[i]] -= 1 + lengths[i] = max_bits + bl_count[max_bits] += 1 + while overflow > 0: + bits: t.CInt = max_bits - 1 + while bits > 0 and bl_count[bits] == 0: + bits -= 1 + if bits == 0: break + bl_count[bits] -= 1 + bl_count[bits + 1] += 2 + bl_count[max_bits] -= 1 + overflow -= 2 + sorted_items: INTPTR = INTPTR(zdef.zdef_alloc(sort_count * int.__sizeof__())) + si: t.CInt = 0 + for i in range(count): + if freqs[i] > 0: + sorted_items[si] = i + si += 1 + for i in range(si - 1): + for j in range(i + 1, si): + if lengths[sorted_items[i]] < lengths[sorted_items[j]]: + tmp: t.CInt = sorted_items[i] + sorted_items[i] = sorted_items[j] + sorted_items[j] = tmp + sidx: t.CInt = 0 + for bits in range(max_bits, 1, -1): + n: t.CInt = bl_count[bits] + while n > 0 and sidx < si: + lengths[sorted_items[sidx]] = bits + sidx += 1 + n -= 1 + zdef.zdef_free(sorted_items) + zdef.zdef_free(heap) + zdef.zdef_free(parent) + zdef.zdef_free(combined_freq) + zdef.zdef_free(item_freqs) + zdef.zdef_free(items) + + def get_fixed_lit_lengths(self, lengths: INTPTR): + for i in range( 0, 143 + 1): lengths[i] = 8 + for i in range(143 + 1, 255 + 1): lengths[i] = 9 + for i in range(255 + 1, 279 + 1): lengths[i] = 7 + for i in range(279 + 1, 287 + 1): lengths[i] = 8 + + def get_fixed_dist_lengths(self, lengths: INTPTR): + for i in range(32): + lengths[i] = 5 + + def build_tree_from_lengths(self, lengths: INTPTR, count: t.CInt, max_bits: t.CInt): + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + + code: UINT = 0 + next_code[0] = 0 + for bits in range(1, max_bits + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + + self.count = count + self.max_bits = max_bits + for i in range(count): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt + + +@t.Object +class zhuff_decode_tree: + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + + def __init__(self): + pass + + def build_decode_tree(self, ht: zhuff_tree | t.CPtr): + self.node_count = 1 + self.root = 0 + self.nodes[0].children[0] = -1 + self.nodes[0].children[1] = -1 + self.nodes[0].symbol = -1 + for i in range(ht.count): + if ht.codes[i].bits == 0: continue + node: int = 0 + for bit in range(ht.codes[i].bits - 1, -1, -1): + dir: t.CInt = (ht.codes[i].code >> bit) & 1 + if bit > 0: + # Internal node: traverse or create + if self.nodes[node].children[dir] == -1: + new_node: int = self.node_count + self.node_count += 1 + self.nodes[new_node].children[0] = -1 + self.nodes[new_node].children[1] = -1 + self.nodes[new_node].symbol = -1 + self.nodes[node].children[dir] = new_node + node = self.nodes[node].children[dir] + else: + # Leaf: bit 0 - set symbol on the child + if self.nodes[node].children[dir] == -1: + leaf: int = self.node_count + self.node_count += 1 + self.nodes[leaf].children[0] = -1 + self.nodes[leaf].children[1] = -1 + self.nodes[leaf].symbol = i + self.nodes[node].children[dir] = leaf + else: + self.nodes[self.nodes[node].children[dir]].symbol = i + + def decode_symbol(self, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: + node: int = self.root + while self.nodes[node].symbol == -1: + bit: t.CUnsignedInt + if reader.read_bits(1, c.Addr(bit)) != 0: return -1 + dir: t.CInt = t.CInt(bit) + if self.nodes[node].children[dir] == -1: return -1 + node = self.nodes[node].children[dir] + return self.nodes[node].symbol diff --git a/Test/ZlibTest/App/__zlib/zinflate.py b/Test/ZlibTest/App/__zlib/zinflate.py new file mode 100644 index 0000000..985f63e --- /dev/null +++ b/Test/ZlibTest/App/__zlib/zinflate.py @@ -0,0 +1,454 @@ +from stdint import * +import zinflate +import zchecksum +import zdef +import zhuff +import string +import t, c + + +@t.Object +class zinflate_stream: + reader: zdef.zbit_reader + + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + + header_parsed: t.CInt + is_gzip: t.CInt + + adler: t.CUInt32T + crc: t.CUInt32T + + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + + max_length: t.CSizeT + + def __init__(self): + pass + + def output_byte(self, byte: BYTE): + if self.max_length > 0 and self.output_len >= self.max_length: return + + if self.output_len >= self.output_cap: + new_cap: t.CSizeT = self.output_cap * 2 + if new_cap < 256: new_cap = 256 + new_buf: BYTEPTR = BYTEPTR(zdef.zdef_alloc(new_cap)) + if not new_buf: return + memcpy(new_buf, self.output, self.output_len) + zdef.zdef_free(self.output) + self.output = new_buf + self.output_cap = new_cap + self.output[self.output_len] = byte + self.output_len += 1 + self.window[self.window_pos & (zdef.ZDEFLATE_WINDOW_SIZE - 1)] = byte + self.window_pos += 1 + + + def parse_header(self) -> t.CInt: + if self.header_parsed: return 0 + if self.wbits < 0: + self.header_parsed = 1 + self.is_gzip = 0 + return 0 + if self.input_pos + 2 > self.input_len: return -3 + b0: BYTE = self.input_data[self.input_pos] + b1: BYTE = self.input_data[self.input_pos + 1] + if b0 == 0x1F and b1 == 0x8B: + self.is_gzip = 1 + if self.input_pos + 10 > self.input_len: return -3 + self.input_pos += 10 + flg: BYTE = self.input_data[self.input_pos - 6] + if flg & 0x04: + if self.input_pos + 2 > self.input_len: return -3 + xlen: UINT = self.input_data[self.input_pos] | (self.input_data[self.input_pos + 1] << 8) + self.input_pos += 2 + xlen + if flg & 0x08: + while self.input_pos < self.input_len and self.input_data[self.input_pos] != 0: self.input_pos += 1 + self.input_pos += 1 + if flg & 0x10: + while self.input_pos < self.input_len and self.input_data[self.input_pos] != 0: self.input_pos += 1 + self.input_pos += 1 + if flg & 0x02: + self.input_pos += 2 + if self.input_pos > self.input_len: return -3 + else: + self.is_gzip = 0 + if (b0 * 256 + b1) % 31 != 0: return -3 + cm: t.CInt = b0 & 0x0F + if cm != 8: return -3 + self.input_pos += 2 + self.header_parsed = 1 + return 0 + + + def read_bits_from_input(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits): + if self.input_pos >= self.input_len: return -3 + if self.input_data[self.input_pos] & (UINT(1) << (self.reader.bit_pos)): + val |= (UINT(1) << i) + self.reader.bit_pos += 1 + if self.reader.bit_pos == 8: + self.reader.bit_pos = 0 + self.input_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + + def read_huffman(self, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: + node: int = dt.root + while dt.nodes[node].symbol == -1: + bit: UINT + if self.read_bits_from_input(1, c.Addr(bit)) != 0: return -1 + dir: int = t.CInt(bit) + if dt.nodes[node].children[dir] == -1: return -3 + node = dt.nodes[node].children[dir] + return dt.nodes[node].symbol + + def inflate_block_stored(self) -> t.CInt: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 4 > self.input_len: return -3 + length: UINT = self.input_data[self.input_pos] | (self.input_data[self.input_pos + 1] << 8) + nlen: UINT = self.input_data[self.input_pos + 2] | (self.input_data[self.input_pos + 3] << 8) + self.input_pos += 4 + if (length ^ nlen) != 0xFFFF: return -3 + if self.input_pos + length > self.input_len: return -3 + i: t.CUnsignedInt + for i in range(length): + self.output_byte(self.input_data[self.input_pos]) + self.input_pos += 1 + return 0 + + + def inflate_block_fixed(self) -> t.CInt: + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + dist_tree.build_fixed_dist_tree() + + lit_dt: zhuff.zhuff_decode_tree | t.CPtr = zdef.zdef_alloc(zhuff.zhuff_decode_tree.__sizeof__()) + dist_dt: zhuff.zhuff_decode_tree | t.CPtr = zdef.zdef_alloc(zhuff.zhuff_decode_tree.__sizeof__()) + if not lit_dt or not dist_dt: + if lit_dt: zdef.zdef_free(lit_dt) + if dist_dt: zdef.zdef_free(dist_dt) + return -4 + memset(lit_dt, 0, zhuff.zhuff_decode_tree.__sizeof__()) + memset(dist_dt, 0, zhuff.zhuff_decode_tree.__sizeof__()) + lit_dt.build_decode_tree(c.Addr(lit_tree)) + dist_dt.build_decode_tree(c.Addr(dist_tree)) + + result: t.CInt = 0 + while True: + if self.max_length > 0 and self.output_len >= self.max_length: break + + sym: t.CInt = self.read_huffman(lit_dt) + if sym < 0: + result = -3 + break + if sym == 256: break + if sym < 256: + self.output_byte(BYTE(sym)) + else: + len_idx: t.CInt = sym - 257 + if len_idx < 0 or len_idx >= 29: + result = -3 + break + length: t.CInt = zdef.zdeflate_len_base[len_idx] + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_idx] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: + result = -3 + break + length += extra_val + + dist_sym: t.CInt = self.read_huffman(dist_dt) + if dist_sym < 0 or dist_sym >= 30: + result = -3 + break + dist: t.CInt = zdef.zdeflate_dist_base[dist_sym] + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: + result = -3 + break + dist += extra_val + + for i in range(length): + if self.max_length > 0 and self.output_len >= self.max_length: break + src_pos: t.CInt = (self.window_pos - dist) & (zdef.ZDEFLATE_WINDOW_SIZE - 1) + byte: BYTE = self.window[src_pos] + self.output_byte(byte) + + zdef.zdef_free(lit_dt) + zdef.zdef_free(dist_dt) + return result + + + def inflate_block_dynamic(self) -> t.CInt: + hlit_val: UINT + hdist_val: UINT + hclen_val: UINT + if self.read_bits_from_input(5, c.Addr(hlit_val)) != 0: return -3 + if self.read_bits_from_input(5, c.Addr(hdist_val)) != 0: return -3 + if self.read_bits_from_input(4, c.Addr(hclen_val)) != 0: return -3 + + hlit: t.CInt = t.CInt(hlit_val) + 257 + hdist: t.CInt = t.CInt(hdist_val) + 1 + hclen: t.CInt = t.CInt(hclen_val) + 4 + + cl_lengths: t.CArray[t.CInt, 19] + memset(cl_lengths, 0, cl_lengths.__sizeof__()) + for i in range(hclen): + len_val: UINT + if self.read_bits_from_input(3, c.Addr(len_val)) != 0: return -3 + cl_lengths[zdef.zdeflate_codelen_order[i]] = t.CInt(len_val) + + cl_tree = zhuff.zhuff_tree() + cl_tree.build_tree_from_lengths(cl_lengths, 19, 7) + + cl_dt = zhuff.zhuff_decode_tree() + cl_dt.build_decode_tree(c.Addr(cl_tree)) + + total: t.CInt = hlit + hdist + all_lengths: t.CInt | t.CPtr = zdef.zdef_alloc(total * int.__sizeof__()) + if not all_lengths: return -4 + + idx: t.CInt = 0 + while idx < total: + sym: t.CInt = self.read_huffman(c.Addr(cl_dt)) + if sym < 0 or sym > 18: + zdef.zdef_free(all_lengths) + return -3 + if sym < 16: + all_lengths[idx] = sym + idx += 1 + elif sym == 16: + rep: UINT + if self.read_bits_from_input(2, c.Addr(rep)) != 0: + zdef.zdef_free(all_lengths) + return -3 + rep += 3 + if idx == 0 or idx + rep > total: + zdef.zdef_free(all_lengths) + return -3 + prev: t.CInt = all_lengths[idx - 1] + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = prev + idx += 1 + elif sym == 17: + rep: t.CUnsignedInt + if self.read_bits_from_input(3, c.Addr(rep)) != 0: + zdef.zdef_free(all_lengths) + return -3 + rep += 3 + if idx + rep > total: + zdef.zdef_free(all_lengths) + return -3 + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = 0 + idx += 1 + else: + rep: t.CUnsignedInt + if self.read_bits_from_input(7, c.Addr(rep)) != 0: + zdef.zdef_free(all_lengths) + return -3 + rep += 11 + if idx + rep > total: + zdef.zdef_free(all_lengths) + return -3 + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = 0 + idx += 1 + + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + + lit_lengths: t.CArray[t.CInt, 288] + memset(lit_lengths, 0, lit_lengths.__sizeof__()) + for i in range(hlit): + lit_lengths[i] = all_lengths[i] + lit_tree.build_tree_from_lengths(lit_lengths, hlit, zdef.ZDEFLATE_MAX_BITS) + + dist_lengths: t.CArray[t.CInt, 32] + memset(dist_lengths, 0, dist_lengths.__sizeof__()) + for i in range(hdist): + dist_lengths[i] = all_lengths[hlit + i] + dist_tree.build_tree_from_lengths(dist_lengths, hdist, zdef.ZDEFLATE_MAX_BITS) + + zdef.zdef_free(all_lengths) + + lit_dt = zhuff.zhuff_decode_tree() + dist_dt = zhuff.zhuff_decode_tree() + lit_dt.build_decode_tree(c.Addr(lit_tree)) + dist_dt.build_decode_tree(c.Addr(dist_tree)) + + while True: + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + sym: t.CInt = self.read_huffman(c.Addr(lit_dt)) + if sym < 0: return -3 + if sym == 256: return 0 + if sym < 256: + self.output_byte(BYTE(sym)) + else: + len_idx: t.CInt = sym - 257 + if len_idx < 0 or len_idx >= 29: return -3 + length: t.CInt = zdef.zdeflate_len_base[len_idx] + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_idx] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: return -3 + length += extra_val + dist_sym: t.CInt = self.read_huffman(c.Addr(dist_dt)) + if dist_sym < 0 or dist_sym >= 30: return -3 + dist: t.CInt = zdef.zdeflate_dist_base[dist_sym] + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: return -3 + dist += extra_val + for i in range(length): + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + src_pos: t.CInt = (self.window_pos - dist) & (zdef.ZDEFLATE_WINDOW_SIZE - 1) + byte: BYTE = self.window[src_pos] + self.output_byte(byte) + + def inflate_stream(self) -> t.CInt: + err: t.CInt = self.parse_header() + if err != 0: return err + while not self.is_finished: + bfinal_val: UINT + btype_val: UINT + if self.read_bits_from_input(1, c.Addr(bfinal_val)) != 0: return -3 + if self.read_bits_from_input(2, c.Addr(btype_val)) != 0: return -3 + bfinal: t.CInt = t.CInt(bfinal_val) + btype: t.CInt = t.CInt(btype_val) + if btype == 0: + err = self.inflate_block_stored() + elif btype == 1: + err = self.inflate_block_fixed() + elif btype == 2: + err = self.inflate_block_dynamic() + else: + return -3 + if err != 0: return err + if bfinal: self.is_finished = 1 + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + if self.is_gzip: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 8 <= self.input_len: + self.crc = ((t.CUInt32T(self.input_data[self.input_pos + 0]) << 0) | + (t.CUInt32T(self.input_data[self.input_pos + 1]) << 8) | + (t.CUInt32T(self.input_data[self.input_pos + 2]) << 16) | + (t.CUInt32T(self.input_data[self.input_pos + 3]) << 24)) + self.input_pos += 8 + elif self.wbits >= 0: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 4 <= self.input_len: + self.adler = ((t.CUInt32T(self.input_data[self.input_pos + 0]) << 24) | + (t.CUInt32T(self.input_data[self.input_pos + 1]) << 16) | + (t.CUInt32T(self.input_data[self.input_pos + 2]) << 8) | + (t.CUInt32T(self.input_data[self.input_pos + 3]) << 0)) + self.input_pos += 4 + return 0 + + def decompress(self, data: UINT8PTR, length: t.CSizeT, + max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if not self.is_initialized: return -2 + self.input_data = data + self.input_len = length + self.input_pos = 0 + self.reader.bit_pos = 0 + self.output_len = 0 + self.max_length = max_length + err: t.CInt = self.inflate_stream() + if err != 0: return err + c.Set(c.Deref(out_len), self.output_len) + if self.output_len == 0: + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(1))) + else: + dst: UINT8PTR = UINT8PTR(zdef.zdef_alloc(self.output_len)) + c.Set(c.Deref(out), dst) + if dst: + memcpy(dst, self.output, self.output_len) + return 0 + + def set_dictionary(self, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: + if not dict: return -2 + use: t.CSizeT = length + if use > zdef.ZDEFLATE_WINDOW_SIZE: + dict += length - zdef.ZDEFLATE_WINDOW_SIZE + use = zdef.ZDEFLATE_WINDOW_SIZE + memcpy(self.window, dict, use) + self.window_pos = t.CInt(use) + return 0 + + def copy(self) -> zinflate_stream | t.CPtr: + z: zinflate_stream | t.CPtr = zdef.zdef_alloc(zinflate_stream.__sizeof__()) + if not z: return None + memcpy(z, self, zinflate_stream.__sizeof__()) + z.window = BYTEPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE)) + z.output = BYTEPTR(zdef.zdef_alloc(self.output_cap)) + if not z.window or not z.output: + zinflate_stream.destroy(z) + return None + memcpy(z.window, self.window, zdef.ZDEFLATE_WINDOW_SIZE) + memcpy(z.output, self.output, self.output_cap) + return z + + def destroy(self): + if not self: return + if self.window: zdef.zdef_free(self.window) + if self.output: zdef.zdef_free(self.output) + zdef.zdef_free(self) + + +def zinflate_one_shot(data: UINT8PTR, length: t.CSizeT, + wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: + s: zinflate_stream | t.CPtr = zinflate_create(wbits) + if not s: return None + out: t.CUInt8T | t.CPtr = None + err: t.CInt = s.decompress(data, length, 0, c.Addr(out), out_len) + s.destroy() + if err != 0: return None + return out + + +def zinflate_create(wbits: t.CInt) -> zinflate_stream | t.CPtr: + s: zinflate_stream | t.CPtr = zdef.zdef_alloc(zinflate_stream.__sizeof__()) + if not s: return None + memset(s, 0, zinflate_stream.__sizeof__()) + s.wbits = wbits + s.is_initialized = 1 + s.adler = 1 + s.window = BYTEPTR(zdef.zdef_alloc(zdef.ZDEFLATE_WINDOW_SIZE)) + s.output = BYTEPTR(zdef.zdef_alloc(256)) + s.output_cap = 256 + s.output_len = 0 + s.max_length = 0 + if not s.window or not s.output: + s.destroy() + return None + return s diff --git a/Test/ZlibTest/App/main.py b/Test/ZlibTest/App/main.py new file mode 100644 index 0000000..6575ead --- /dev/null +++ b/Test/ZlibTest/App/main.py @@ -0,0 +1,19 @@ +import t +import stdio +import string +import __zlib.pyzlib as pyzlib +import __zlib.zdeflate as zdeflate +import __zlib.zinflate as zinflate +import __zlib.zchecksum as zchecksum +import __zlib.zdef as zdef +import __zlib.zhuff as zhuff +import __zlib.test_pyzlib as tpz +import w32.win32memory +import c +import testcheck + +def main() -> t.CInt: + testcheck.begin("ZlibTest: Zlib 压缩库测试") + tpz.main123456() + testcheck.ok("zlib test_pyzlib completed") + return testcheck.end() diff --git a/Test/ZlibTest/output/03b24bc1d9327958.deps.json b/Test/ZlibTest/output/03b24bc1d9327958.deps.json new file mode 100644 index 0000000..9462b24 --- /dev/null +++ b/Test/ZlibTest/output/03b24bc1d9327958.deps.json @@ -0,0 +1 @@ +{"__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/784dde8a8697b7ab.deps.json b/Test/ZlibTest/output/784dde8a8697b7ab.deps.json new file mode 100644 index 0000000..f610c22 --- /dev/null +++ b/Test/ZlibTest/output/784dde8a8697b7ab.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/9b7d2adaae94f021.deps.json b/Test/ZlibTest/output/9b7d2adaae94f021.deps.json new file mode 100644 index 0000000..1bf85b0 --- /dev/null +++ b/Test/ZlibTest/output/9b7d2adaae94f021.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/cb151b297f4ea56d.deps.json b/Test/ZlibTest/output/cb151b297f4ea56d.deps.json new file mode 100644 index 0000000..1ea42be --- /dev/null +++ b/Test/ZlibTest/output/cb151b297f4ea56d.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/d855b3e66d1d6522.deps.json b/Test/ZlibTest/output/d855b3e66d1d6522.deps.json new file mode 100644 index 0000000..3ea635f --- /dev/null +++ b/Test/ZlibTest/output/d855b3e66d1d6522.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/dd19090b337e9bea.deps.json b/Test/ZlibTest/output/dd19090b337e9bea.deps.json new file mode 100644 index 0000000..257d8e6 --- /dev/null +++ b/Test/ZlibTest/output/dd19090b337e9bea.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "testcheck": "dd3002730623424b", "__zlib.zinflate": "e445a2bb299d1b7d", "zinflate": "e445a2bb299d1b7d", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/output/e445a2bb299d1b7d.deps.json b/Test/ZlibTest/output/e445a2bb299d1b7d.deps.json new file mode 100644 index 0000000..66cf6da --- /dev/null +++ b/Test/ZlibTest/output/e445a2bb299d1b7d.deps.json @@ -0,0 +1 @@ +{"main": "03b24bc1d9327958", "__zlib.pyzlib": "0f8134f67bdaba8f", "pyzlib": "0f8134f67bdaba8f", "stdio": "6f62fe05c5ea1ceb", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "__zlib.zdeflate": "784dde8a8697b7ab", "zdeflate": "784dde8a8697b7ab", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "stdlib": "90c53dd6db8d41cf", "__zlib.zdef": "9b7d2adaae94f021", "zdef": "9b7d2adaae94f021", "string": "ab6e54ba9a669f76", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "__zlib.zchecksum": "cb151b297f4ea56d", "zchecksum": "cb151b297f4ea56d", "__zlib.test_pyzlib": "d855b3e66d1d6522", "test_pyzlib": "d855b3e66d1d6522", "__zlib.zhuff": "dd19090b337e9bea", "zhuff": "dd19090b337e9bea", "testcheck": "dd3002730623424b", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/Test/ZlibTest/project.json b/Test/ZlibTest/project.json new file mode 100644 index 0000000..7001e57 --- /dev/null +++ b/Test/ZlibTest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "ZlibTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-Wl,--allow-multiple-definition", "-Wl,--unresolved-symbols=ignore-in-object-files", "-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-luser32", "-latomic"], + "output": "ZlibTest.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/Test/ZlibTest/temp/0035c95a18d4f8e8.pyi b/Test/ZlibTest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/Test/ZlibTest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/Test/ZlibTest/temp/03b24bc1d9327958.doc.json b/Test/ZlibTest/temp/03b24bc1d9327958.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/03b24bc1d9327958.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/03b24bc1d9327958.pyi b/Test/ZlibTest/temp/03b24bc1d9327958.pyi new file mode 100644 index 0000000..8757634 --- /dev/null +++ b/Test/ZlibTest/temp/03b24bc1d9327958.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t +import stdio +import string +import __zlib.pyzlib as pyzlib +import __zlib.zdeflate as zdeflate +import __zlib.zinflate as zinflate +import __zlib.zchecksum as zchecksum +import __zlib.zdef as zdef +import __zlib.zhuff as zhuff +import __zlib.test_pyzlib as tpz +import w32.win32memory +import c +import testcheck + +def main() -> t.CInt: pass diff --git a/Test/ZlibTest/temp/067c78e9f121dce3.pyi b/Test/ZlibTest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/Test/ZlibTest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/Test/ZlibTest/temp/06f53cc594b4ac6c.pyi b/Test/ZlibTest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/Test/ZlibTest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ZlibTest/temp/0f8134f67bdaba8f.doc.json b/Test/ZlibTest/temp/0f8134f67bdaba8f.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/0f8134f67bdaba8f.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/0f8134f67bdaba8f.pyi b/Test/ZlibTest/temp/0f8134f67bdaba8f.pyi new file mode 100644 index 0000000..3f2901b --- /dev/null +++ b/Test/ZlibTest/temp/0f8134f67bdaba8f.pyi @@ -0,0 +1,122 @@ +""" +Auto-generated Python stub file from __zlib.pyzlib.py +Module: __zlib.pyzlib +""" + + +from stdint import * +import zdeflate +import zinflate +import zchecksum +import zdef +import zhuff +import stdlib +import string +import t, c + +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) +DEFLATED: t.CDefine = 8 +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 +ZLIB_VERSION: t.CDefine = "1.3.2" +pyzlib_error_msg: t.CExtern | t.CArray[t.CChar, 512] +pyzlib_error_code_val: t.CExtern | t.CInt + +@t.Object +class Compress: + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + def compress(self: Compress, data: BYTEPTR, data_len: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Compress, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Compress) -> Compress | t.CPtr: pass + def delete(self: Compress) -> t.CInt: pass + def __del__(self: Compress) -> t.CInt: pass +@t.Object +class Decompress: + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + def decompress(self: Decompress, data: BYTEPTR, data_len: t.CSizeT, max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def flush(self: Decompress, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def copy(self: Decompress) -> Decompress | t.CPtr: pass + def delete(self: Decompress) -> t.CInt: pass + def unused_data(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + def unconsumed_tail(self: Decompress, length: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def set_error(code: int, msg: str) -> t.CInt: pass + +def clone_bytes(src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def append_bytes(buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: pass + +def shrink_to_fit(buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: pass + +def runtime_version() -> str: pass + +def get_error() -> str: pass + +def get_error_code() -> t.CInt: pass + +def clear_error() -> t.CInt: pass + +def compress(data: BYTEPTR, data_len: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def decompress(data: BYTEPTR, data_len: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: pass + +def compressobj(level: t.CInt, method: t.CInt, wbits: t.CInt, memLevel: t.CInt, strategy: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: pass + +def decompressobj(wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: pass + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: pass diff --git a/Test/ZlibTest/temp/6446627d4f07a1b5.pyi b/Test/ZlibTest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/Test/ZlibTest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/Test/ZlibTest/temp/6f62fe05c5ea1ceb.pyi b/Test/ZlibTest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/Test/ZlibTest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/Test/ZlibTest/temp/72e2d5ccb7cedcf1.pyi b/Test/ZlibTest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/Test/ZlibTest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/Test/ZlibTest/temp/784dde8a8697b7ab.doc.json b/Test/ZlibTest/temp/784dde8a8697b7ab.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/784dde8a8697b7ab.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/784dde8a8697b7ab.pyi b/Test/ZlibTest/temp/784dde8a8697b7ab.pyi new file mode 100644 index 0000000..193ea96 --- /dev/null +++ b/Test/ZlibTest/temp/784dde8a8697b7ab.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from __zlib.zdeflate.py +Module: __zlib.zdeflate +""" + + +from stdint import * +import zchecksum +import zhuff +import zdef +import string +import t, c + +@t.Object +class zdeflate_stream: + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + def find_match(self: zdeflate_stream, data: BYTE | t.CPtr, data_len: t.CSizeT, pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: pass + def update_hash(self: zdeflate_stream, data: BYTE | t.CPtr, pos: t.CSizeT) -> t.CInt: pass + def write_fixed_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def encode_block_data(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + def write_dynamic_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_stored_block(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def compress(self: zdeflate_stream, data: UINT8PTR, length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def flush(self: zdeflate_stream, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zdeflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zdeflate_stream) -> zdeflate_stream | t.CPtr: pass + def destroy(self: zdeflate_stream) -> t.CInt: pass + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: pass + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: pass + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: pass + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR) -> t.CInt: pass + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr) -> t.CInt: pass + +def zdeflate_create(level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: pass + +def zdeflate_one_shot(data: UINT8PTR, length: t.CSizeT, level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass diff --git a/Test/ZlibTest/temp/7e529fe7a078cfef.pyi b/Test/ZlibTest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/Test/ZlibTest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/Test/ZlibTest/temp/90c53dd6db8d41cf.pyi b/Test/ZlibTest/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/Test/ZlibTest/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/Test/ZlibTest/temp/9b7d2adaae94f021.doc.json b/Test/ZlibTest/temp/9b7d2adaae94f021.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/9b7d2adaae94f021.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/9b7d2adaae94f021.pyi b/Test/ZlibTest/temp/9b7d2adaae94f021.pyi new file mode 100644 index 0000000..3552322 --- /dev/null +++ b/Test/ZlibTest/temp/9b7d2adaae94f021.pyi @@ -0,0 +1,65 @@ +""" +Auto-generated Python stub file from __zlib.zdef.py +Module: __zlib.zdef +""" + + +from stdint import * +import stddef +import string +import stdlib +import t, c + +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 +zdeflate_len_extra_bits: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_len_base: t.CExtern | t.CArray[t.CInt, 29] +zdeflate_dist_extra_bits: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_dist_base: t.CExtern | t.CArray[t.CInt, 30] +zdeflate_codelen_order: t.CExtern | t.CArray[int, 19] + +@t.Object +class zbit_writer: + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def __init__(self: zbit_writer) -> t.CInt: pass + def ensure(self: zbit_writer, need: t.CSizeT) -> t.CInt: pass + def write_bits(self: zbit_writer, value: UINT, nbits: t.CInt) -> t.CInt: pass + def write_bits_rev(self: zbit_writer, code: UINT, nbits: t.CInt) -> t.CInt: pass + def stored_block(self: zbit_writer, data: UINT8PTR, length: t.CSizeT, final: t.CInt) -> t.CInt: pass + def write_zlib_header(self: zbit_writer, wbits: t.CInt, level: t.CInt) -> t.CInt: pass + def write_gzip_header(self: zbit_writer) -> t.CInt: pass + def align(self: zbit_writer) -> t.CInt: pass + def total(self: zbit_writer) -> t.CSizeT: pass + def free(self: zbit_writer) -> t.CInt: pass + def __del__(self: zbit_writer) -> t.CInt: pass +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + def init(self: zbit_reader, buf: BYTE | t.CPtr, length: t.CSizeT) -> t.CInt: pass + def read_bits(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_bits_rev(self: zbit_reader, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def align(self: zbit_reader) -> t.CInt: pass + +def zdef_alloc(n: t.CSizeT) -> VOIDPTR: pass + +def zdef_free(p: VOIDPTR) -> t.CInt: pass diff --git a/Test/ZlibTest/temp/_phase1_manifest.json b/Test/ZlibTest/temp/_phase1_manifest.json new file mode 100644 index 0000000..d47877d --- /dev/null +++ b/Test/ZlibTest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\pyzlib.py": {"sha1": "0f8134f67bdaba8f", "mtime": 1782266233.8661451, "size": 20930}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\test_pyzlib.py": {"sha1": "d855b3e66d1d6522", "mtime": 1782407614.5315387, "size": 36204}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\zchecksum.py": {"sha1": "cb151b297f4ea56d", "mtime": 1782266234.306909, "size": 4068}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\zdef.py": {"sha1": "9b7d2adaae94f021", "mtime": 1782393175.830038, "size": 8195}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\zdeflate.py": {"sha1": "784dde8a8697b7ab", "mtime": 1782396643.2488594, "size": 18977}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\zhuff.py": {"sha1": "dd19090b337e9bea", "mtime": 1782388678.1362753, "size": 13596}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\__zlib\\zinflate.py": {"sha1": "e445a2bb299d1b7d", "mtime": 1782266257.0389085, "size": 18067}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\Test\\ZlibTest\\App\\main.py": {"sha1": "03b24bc1d9327958", "mtime": 1782110043.5860827, "size": 484}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\testcheck.py": {"sha1": "dd3002730623424b", "mtime": 1783927513.1159866, "size": 1818}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32memory.py": {"sha1": "72e2d5ccb7cedcf1", "mtime": 1781200703.5389092, "size": 3142}} \ No newline at end of file diff --git a/Test/ZlibTest/temp/_sha1_map.txt b/Test/ZlibTest/temp/_sha1_map.txt new file mode 100644 index 0000000..161bc24 --- /dev/null +++ b/Test/ZlibTest/temp/_sha1_map.txt @@ -0,0 +1,16 @@ +03b24bc1d9327958:main.py +0f8134f67bdaba8f:__zlib\pyzlib.py +6f62fe05c5ea1ceb:includes/stdio.py +72e2d5ccb7cedcf1:includes/w32\win32memory.py +784dde8a8697b7ab:__zlib\zdeflate.py +7e529fe7a078cfef:includes/w32\win32base.py +90c53dd6db8d41cf:includes/stdlib.py +9b7d2adaae94f021:__zlib\zdef.py +ab6e54ba9a669f76:includes/string.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +cb151b297f4ea56d:__zlib\zchecksum.py +d855b3e66d1d6522:__zlib\test_pyzlib.py +dd19090b337e9bea:__zlib\zhuff.py +dd3002730623424b:includes/testcheck.py +e445a2bb299d1b7d:__zlib\zinflate.py +f5522571bcce7bcb:includes/stdint.py diff --git a/Test/ZlibTest/temp/_shared_sym.json b/Test/ZlibTest/temp/_shared_sym.json new file mode 100644 index 0000000..6c9b319 Binary files /dev/null and b/Test/ZlibTest/temp/_shared_sym.json differ diff --git a/Test/ZlibTest/temp/ab6e54ba9a669f76.pyi b/Test/ZlibTest/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/Test/ZlibTest/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/Test/ZlibTest/temp/bbdf3bbd4c3bc28c.pyi b/Test/ZlibTest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/Test/ZlibTest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/Test/ZlibTest/temp/cb151b297f4ea56d.doc.json b/Test/ZlibTest/temp/cb151b297f4ea56d.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/cb151b297f4ea56d.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/cb151b297f4ea56d.pyi b/Test/ZlibTest/temp/cb151b297f4ea56d.pyi new file mode 100644 index 0000000..125a3b3 --- /dev/null +++ b/Test/ZlibTest/temp/cb151b297f4ea56d.pyi @@ -0,0 +1,15 @@ +""" +Auto-generated Python stub file from __zlib.zchecksum.py +Module: __zlib.zchecksum +""" + + +from stdint import * +import t, c + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: pass + + +crc32_table: t.CExtern | t.CArray[t.CUInt32T, 256] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: pass diff --git a/Test/ZlibTest/temp/d855b3e66d1d6522.doc.json b/Test/ZlibTest/temp/d855b3e66d1d6522.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/d855b3e66d1d6522.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/d855b3e66d1d6522.pyi b/Test/ZlibTest/temp/d855b3e66d1d6522.pyi new file mode 100644 index 0000000..09ce9d4 --- /dev/null +++ b/Test/ZlibTest/temp/d855b3e66d1d6522.pyi @@ -0,0 +1,66 @@ +""" +Auto-generated Python stub file from __zlib.test_pyzlib.py +Module: __zlib.test_pyzlib +""" + + +from stdint import * +import pyzlib +import zhuff +import zdef +from stdio import printf +import stdlib +from string import strlen +import string +import t, c + +test_passed: t.CExtern | t.CInt +test_failed: t.CExtern | t.CInt + +def check(name: str, condition: t.CInt, detail: str) -> t.CInt: pass + +def print_hex_test(data: BYTEPTR, length: t.CSizeT, max_show: t.CSizeT) -> t.CInt: pass + +def print_separator() -> t.CInt: pass + +def section_header(title: str) -> t.CInt: pass + +def section_footer() -> t.CInt: pass + +def test_version() -> t.CInt: pass + +def test_compress_decompress() -> t.CInt: pass + +def test_compress_levels() -> t.CInt: pass + +def test_compressobj() -> t.CInt: pass + +def test_decompressobj() -> t.CInt: pass + +def test_decompressobj_max_length() -> t.CInt: pass + +def test_decompressobj_unused_data() -> t.CInt: pass + +def test_compress_copy() -> t.CInt: pass + +def test_decompress_copy() -> t.CInt: pass + +def test_adler32() -> t.CInt: pass + +def test_crc32() -> t.CInt: pass + +def test_empty_compress() -> t.CInt: pass + +def test_gzip_format() -> t.CInt: pass + +def test_raw_deflate() -> t.CInt: pass + +def test_zdict() -> t.CInt: pass + +def test_huffman_tree() -> t.CInt: pass + +def test_error_handling() -> t.CInt: pass + +def test_constants() -> t.CInt: pass + +def main123456() -> t.CInt: pass diff --git a/Test/ZlibTest/temp/dd19090b337e9bea.doc.json b/Test/ZlibTest/temp/dd19090b337e9bea.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/dd19090b337e9bea.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/dd19090b337e9bea.pyi b/Test/ZlibTest/temp/dd19090b337e9bea.pyi new file mode 100644 index 0000000..898995b --- /dev/null +++ b/Test/ZlibTest/temp/dd19090b337e9bea.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from __zlib.zhuff.py +Module: __zlib.zhuff +""" + + +from stdint import * +import zdef +import t, c + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt +@t.Object +class zhuff_tree: + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + def __init__(self: zhuff_tree) -> t.CInt: pass + def build_codes(self: zhuff_tree, freqs: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def build_fixed_lit_tree(self: zhuff_tree) -> t.CInt: pass + def build_fixed_dist_tree(self: zhuff_tree) -> t.CInt: pass + def encode_symbol(self: zhuff_tree, symbol: int, writer: zdef.zbit_writer | t.CPtr) -> t.CInt: pass + def build_code_lengths(self: zhuff_tree, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass + def get_fixed_lit_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def get_fixed_dist_lengths(self: zhuff_tree, lengths: INTPTR) -> t.CInt: pass + def build_tree_from_lengths(self: zhuff_tree, lengths: INTPTR, count: t.CInt, max_bits: t.CInt) -> t.CInt: pass +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt +@t.Object +class zhuff_decode_tree: + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + def __init__(self: zhuff_decode_tree) -> t.CInt: pass + def build_decode_tree(self: zhuff_decode_tree, ht: zhuff_tree | t.CPtr) -> t.CInt: pass + def decode_symbol(self: zhuff_decode_tree, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/Test/ZlibTest/temp/dd3002730623424b.pyi b/Test/ZlibTest/temp/dd3002730623424b.pyi new file mode 100644 index 0000000..da1aabc --- /dev/null +++ b/Test/ZlibTest/temp/dd3002730623424b.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from testcheck.py +Module: testcheck +""" + + +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 +_pass_count: t.CExtern | t.CInt +_fail_count: t.CExtern | t.CInt +_total_pass: t.CExtern | t.CInt +_total_fail: t.CExtern | t.CInt + +def begin(name: str) -> t.CInt: pass + +def section(name: str) -> t.CInt: pass + +def ok(msg: str) -> t.CInt: pass + +def fail(msg: str) -> t.CInt: pass + +def check(cond: t.CInt, ok_msg: str, fail_msg: str) -> t.CInt: pass + +def info(msg: str) -> t.CInt: pass + +def end() -> t.CInt: pass + +def summary() -> t.CInt: pass diff --git a/Test/ZlibTest/temp/e445a2bb299d1b7d.doc.json b/Test/ZlibTest/temp/e445a2bb299d1b7d.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/Test/ZlibTest/temp/e445a2bb299d1b7d.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/Test/ZlibTest/temp/e445a2bb299d1b7d.pyi b/Test/ZlibTest/temp/e445a2bb299d1b7d.pyi new file mode 100644 index 0000000..355f684 --- /dev/null +++ b/Test/ZlibTest/temp/e445a2bb299d1b7d.pyi @@ -0,0 +1,51 @@ +""" +Auto-generated Python stub file from __zlib.zinflate.py +Module: __zlib.zinflate +""" + + +from stdint import * +import zinflate +import zchecksum +import zdef +import zhuff +import string +import t, c + +@t.Object +class zinflate_stream: + reader: zdef.zbit_reader + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + header_parsed: t.CInt + is_gzip: t.CInt + adler: t.CUInt32T + crc: t.CUInt32T + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + max_length: t.CSizeT + def __init__(self: zinflate_stream) -> t.CInt: pass + def output_byte(self: zinflate_stream, byte: BYTE) -> t.CInt: pass + def parse_header(self: zinflate_stream) -> t.CInt: pass + def read_bits_from_input(self: zinflate_stream, nbits: t.CInt, out: UINTPTR) -> t.CInt: pass + def read_huffman(self: zinflate_stream, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: pass + def inflate_block_stored(self: zinflate_stream) -> t.CInt: pass + def inflate_block_fixed(self: zinflate_stream) -> t.CInt: pass + def inflate_block_dynamic(self: zinflate_stream) -> t.CInt: pass + def inflate_stream(self: zinflate_stream) -> t.CInt: pass + def decompress(self: zinflate_stream, data: UINT8PTR, length: t.CSizeT, max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: pass + def set_dictionary(self: zinflate_stream, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: pass + def copy(self: zinflate_stream) -> zinflate_stream | t.CPtr: pass + def destroy(self: zinflate_stream) -> t.CInt: pass + +def zinflate_one_shot(data: UINT8PTR, length: t.CSizeT, wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: pass + +def zinflate_create(wbits: t.CInt) -> zinflate_stream | t.CPtr: pass diff --git a/Test/ZlibTest/temp/f5522571bcce7bcb.pyi b/Test/ZlibTest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/Test/ZlibTest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/Test/ZlibTest/temp/f6b51804a0ba8ff0.pyi b/Test/ZlibTest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/Test/ZlibTest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/Test/regression_test.py b/Test/regression_test.py new file mode 100644 index 0000000..7815b75 --- /dev/null +++ b/Test/regression_test.py @@ -0,0 +1,356 @@ +#!/usr/bin/env python3 +"""TransPyC 回归测试脚本 + +自动遍历 Test/ 目录下所有含 project.json 的项目目录， +通过 os.system("python ../Projectrans.py --project ...") 执行编译。 +所有项目编译完成后，多线程执行编译结果，超时 5 秒强制结束。 +stdout 出现 failed/error 等字样也汇总到错误列表，并展示匹配行及上下 5 行。 +""" +import os +import sys +import json +import time +import subprocess +import re + +# 强制 stdout/stderr 使用 UTF-8 编码，避免 Windows GBK 编码在打印含替换字符的输出时崩溃 +try: + sys.stdout.reconfigure(encoding='utf-8', errors='replace') + sys.stderr.reconfigure(encoding='utf-8', errors='replace') +except Exception: + pass + +# ============================================================ +# 二阶段皆通过的项目清单 (2026-06-26 验证) +# 编译阶段: 35/35 通过 运行阶段: 35/35 通过 +# ------------------------------------------------------------ +# BuddyTest BuiltinDecoratorTest CPythonTest +# ClosureTest ClosureTest2 DecoratorMarkerTest +# DecoratorTest DictTest DocTest +# EdgeCaseTest EqTest FakeDuckTest +# ForAndLoopTest FormatStringTest FuncTest +# GenericTest GenericTest2 IterTest +# JsonTest ListTest LoopTest +# PointerTest ShortCircuitTest SimdTest +# StringIterTest StringTest StructTest +# TestFBProject TestProject TestProject2 +# TestProject3 TypeCastTest VRandomTest +# VariadicTest ZlibTest +# ============================================================ + +# 测试目录 +TEST_DIR = os.path.dirname(os.path.abspath(__file__)) +# 项目根目录（Projectrans.py 所在位置） +PROJECT_ROOT = os.path.dirname(TEST_DIR) +# Projectrans.py 路径 +PROJECTRANS = os.path.join(PROJECT_ROOT, "Projectrans.py") + +# 运行超时（秒） +RUN_TIMEOUT = 30 +# 编译超时（秒） +COMPILE_TIMEOUT = 300 + +# stdout 中匹配这些关键词则视为运行失败 +_FAIL_PATTERNS = re.compile( + r'(?i)\b(failed|error|traceback|exception|aborted|segmentation fault|core dumped)\b' +) +# 排除 "0 failed" 这类成功摘要行，以及测试输出中的 "error" 误报 +_PASS_PATTERNS = re.compile( + r'\b0\s+failed\b' + r'|\bError\s+(handling|code|message|check|report|status|info|detail|test)\b' + r'|\+--\s*Error\s' # section headers like "+-- Error handling --+" + r'|\bclear\s+error\b' # "[PASS] clear error clears code" + r'|\berror\s+(code|message)\b' # "error code is set", "error message set" + r'|\[PASS\].*\berror\b' # "[PASS] ... error ..." test assertions + r'|\[FAIL\].*\berror\b', # "[FAIL] ... error ..." test assertions + re.IGNORECASE +) + + +def find_test_projects(): + """查找所有含 project.json 的测试项目目录""" + projects = [] + for name in sorted(os.listdir(TEST_DIR)): + proj_dir = os.path.join(TEST_DIR, name) + if not os.path.isdir(proj_dir): + continue + proj_json = os.path.join(proj_dir, "project.json") + if os.path.isfile(proj_json): + projects.append((name, proj_dir, proj_json)) + return projects + + +def compile_project(name, proj_dir, proj_json_path): + """编译单个测试项目，返回 (success, elapsed, output)""" + cmd = f'python -u "{PROJECTRANS}" --project "{proj_json_path}" --clean' + + start = time.time() + result = subprocess.run( + cmd, + shell=True, + capture_output=True, + text=True, + encoding='utf-8', + errors='replace', + cwd=proj_dir, + timeout=COMPILE_TIMEOUT, + ) + elapsed = time.time() - start + + stdout = result.stdout or "" + stderr = result.stderr or "" + combined = stdout + stderr + + success = result.returncode == 0 + return (success, elapsed, combined) + + +def find_executables(output_dir): + """在 output 目录中查找可执行文件""" + exes = [] + if not os.path.isdir(output_dir): + return exes + for f in os.listdir(output_dir): + fpath = os.path.join(output_dir, f) + if os.path.isfile(fpath) and (f.endswith('.exe') or f.endswith('.bin')): + exes.append(fpath) + return exes + + +def run_executable(exe_path): + """运行单个可执行文件，返回 (returncode, stdout, stderr, timed_out) + + 超时 RUN_TIMEOUT 秒强制结束。 + 使用 CREATE_NO_WINDOW 避免弹出 cmd 窗口，通过管道捕获 stdout/stderr。 + """ + try: + CREATE_NO_WINDOW = 0x08000000 + result = subprocess.run( + [exe_path], + cwd=os.path.dirname(exe_path), + capture_output=True, + text=True, + encoding='utf-8', + errors='replace', + timeout=RUN_TIMEOUT, + creationflags=CREATE_NO_WINDOW, + ) + return (result.returncode, result.stdout or "", result.stderr or "", False) + except subprocess.TimeoutExpired: + return (-1, "", "超时（强制终止）", True) + except Exception as e: + return (-1, "", str(e), False) + + +def extract_error_context(stdout, stderr, context_lines=5): + """检查 stdout/stderr 中是否包含 failed/error 等关键词。 + + 返回 (matched_line, context_str) 或 None。 + context_str 包含匹配行及上下 context_lines 行。 + """ + combined = (stdout + "\n" + stderr).strip() + if not combined: + return None + + lines = combined.split('\n') + for i, line in enumerate(lines): + if _FAIL_PATTERNS.search(line): + # 跳过 "0 failed" 这类成功摘要行 + if _PASS_PATTERNS.search(line): + continue + # 提取上下 context_lines 行 + start = max(0, i - context_lines) + end = min(len(lines), i + context_lines + 1) + context_parts = [] + for j in range(start, end): + marker = " >>>" if j == i else " " + context_parts.append(f"{marker} {lines[j]}") + return (line.strip(), '\n'.join(context_parts)) + return None + + +def main(): + print("=" * 70) + print("TransPyC 回归测试") + print("=" * 70) + print() + + projects = find_test_projects() + total = len(projects) + print(f"发现 {total} 个测试项目\n") + + # ========== 阶段一：编译所有项目 ========== + print("=" * 70) + print("阶段一：编译所有项目") + print("=" * 70) + + compiled = [] # (name, proj_dir, proj_json, compile_ok, elapsed, output) + compile_passed = 0 + compile_failed = 0 + + for idx, (name, proj_dir, proj_json) in enumerate(projects, 1): + print(f"[{idx}/{total}] 编译 {name} ...", end=" ", flush=True) + try: + compile_ok, elapsed, output = compile_project(name, proj_dir, proj_json) + except subprocess.TimeoutExpired: + print("编译超时") + compiled.append((name, proj_dir, proj_json, False, COMPILE_TIMEOUT, "编译超时（5分钟）")) + compile_failed += 1 + continue + except Exception as e: + print(f"异常: {e}") + compiled.append((name, proj_dir, proj_json, False, 0.0, str(e))) + compile_failed += 1 + continue + + status = "通过" if compile_ok else "失败" + print(f"{status} ({elapsed:.1f}s)") + compiled.append((name, proj_dir, proj_json, compile_ok, elapsed, output)) + if compile_ok: + compile_passed += 1 + else: + compile_failed += 1 + + print(f"\n编译阶段: 通过 {compile_passed}, 失败 {compile_failed}\n") + + # ========== 阶段二：运行所有编译成功的可执行文件 ========== + print("=" * 70) + print("阶段二：运行所有可执行文件") + print("=" * 70) + + # 收集所有需要运行的可执行文件 + run_tasks = [] # (project_name, exe_path) + for name, proj_dir, proj_json, compile_ok, elapsed, output in compiled: + if not compile_ok: + continue + try: + with open(proj_json, 'r', encoding='utf-8') as f: + proj_config = json.load(f) + except Exception: + continue + output_dir = os.path.join(proj_dir, proj_config.get('output_dir', './output')) + output_dir = os.path.abspath(output_dir) + for exe in find_executables(output_dir): + run_tasks.append((name, exe)) + + print(f"找到 {len(run_tasks)} 个可执行文件\n") + + run_results = {} # project_name -> list of (exe_name, returncode, stdout, stderr, timed_out) + run_passed = 0 + run_failed = 0 + + if run_tasks: + # 串行运行，避免并发导致的资源竞争（控制台输出、堆分配器等） + for proj_name, exe_path in run_tasks: + exe_name = os.path.basename(exe_path) + try: + returncode, stdout, stderr, timed_out = run_executable(exe_path) + except Exception as e: + returncode, stdout, stderr, timed_out = -1, "", str(e), False + + if proj_name not in run_results: + run_results[proj_name] = [] + run_results[proj_name].append((exe_name, returncode, stdout, stderr, timed_out)) + + # 汇总运行结果 + for name, proj_dir, proj_json, compile_ok, elapsed, output in compiled: + if not compile_ok: + continue + exe_results = run_results.get(name, []) + if not exe_results: + continue + + project_run_ok = True + for exe_name, returncode, stdout, stderr, timed_out in exe_results: + if timed_out or returncode != 0: + project_run_ok = False + break + # 仅在返回码非 0 时检查 stdout 错误关键词，避免误报 + # (PASS 行中可能包含 "Error"/"exception" 等词，如 "Status.Error=1") + + status = "通过" if project_run_ok else "失败" + print(f" {name}: {status}") + if project_run_ok: + run_passed += 1 + else: + run_failed += 1 + + print(f"\n运行阶段: 通过 {run_passed}, 失败 {run_failed}\n") + + # ========== 汇总报告 ========== + print("=" * 70) + print("回归测试报告") + print("=" * 70) + total_failed = compile_failed + run_failed + total_passed = compile_passed - run_failed # 编译通过但运行失败的不算通过 + print(f"总计: {total} 通过: {total_passed} 失败: {total_failed}") + print() + + if total_failed > 0: + print("-" * 70) + print("失败项目详情:") + print("-" * 70) + + # 编译失败 + for name, proj_dir, proj_json, compile_ok, elapsed, output in compiled: + if compile_ok: + continue + print(f"\n【{name}】编译失败") + lines = output.strip().split("\n") if output else [] + if len(lines) > 300: + print(f" ... (省略前 {len(lines) - 300} 行) ...") + lines = lines[-300:] + for line in lines: + print(f" {line}") + + # 运行失败 + for name, proj_dir, proj_json, compile_ok, elapsed, output in compiled: + if not compile_ok: + continue + exe_results = run_results.get(name, []) + if not exe_results: + continue + + has_failure = False + for exe_name, returncode, stdout, stderr, timed_out in exe_results: + if timed_out or returncode != 0: + has_failure = True + break + + if not has_failure: + continue + + print(f"\n【{name}】运行失败 (编译耗时 {elapsed:.1f}s)") + for exe_name, returncode, stdout, stderr, timed_out in exe_results: + if timed_out: + print(f" [运行] {exe_name}: 超时（5秒强制终止）") + continue + if returncode != 0: + print(f" [运行] {exe_name}: 退出码 {returncode}") + if stderr.strip(): + print(f" stderr: {stderr.strip()[:300]}") + continue + + # 检查错误关键词，展示匹配行及上下 5 行 + error_ctx = extract_error_context(stdout, stderr) + if error_ctx: + matched_line, context_str = error_ctx + print(f" [运行] {exe_name}: 输出含错误关键词") + print(f" 匹配行: {matched_line}") + print(f" 上下文:") + for ctx_line in context_str.split('\n'): + print(f" {ctx_line}") + print() + + print("=" * 70) + if total_failed == 0: + print("所有测试通过！") + else: + print(f"有 {total_failed} 个测试失败，请检查上方详情。") + print("=" * 70) + + return 1 if total_failed > 0 else 0 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/TransPyC.py b/TransPyC.py new file mode 100644 index 0000000..2c75703 --- /dev/null +++ b/TransPyC.py @@ -0,0 +1,13 @@ +# TransPyC 入口程序 +# 拆分后的模块位于 lib/Shell/ 目录下 +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） + +from lib.Shell import ( + SerializeSymbolTable, deSerializeSymbolTable, + SymbolFile, Config, TransPyC +) + +__all__ = ['SerializeSymbolTable', 'deSerializeSymbolTable', 'SymbolFile', 'Config', 'TransPyC'] + +if __name__ == '__main__': + TransPyC().Run() diff --git a/TransPyV/App/lib/Projectrans/Config.py b/TransPyV/App/lib/Projectrans/Config.py new file mode 100644 index 0000000..962aebe --- /dev/null +++ b/TransPyV/App/lib/Projectrans/Config.py @@ -0,0 +1,299 @@ +import t, c +from stdint import * +import memhub +import string +import stdio +import viperlib +from json.__parser import parse as json_parse +from json import JsonValue, JSON_OBJECT, JSON_INT +import w32.fileio as fileio + + +# ============================================================ +# Config - 工程配置 +# ============================================================ + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 配置值（从 project.vpj 加载） +SourceDir: str +TempDir: str +OutputDir: str +ProjectName: str +ProjectVersion: str +CompilerCmd: str +CompilerFlags: str # 编译器参数（空格分隔，从 compiler.flags 数组拼接） +LinkerCmd: str +LinkerFlags: str # 链接器参数（空格分隔，从 linker.flags 数组拼接） +LinkerOutput: str +TargetTriple: str +TargetDataLayout: str +SliceLevel: t.CInt +StrictMode: t.CInt +IncludesDir: str # includes 目录路径（从 includes 数组首元素） + + +def Load_project_config(path: str) -> int: + """加载工程配置文件（project.vpj） + + Args: + path: 配置文件路径 + + Returns: + 0 表示成功，非 0 表示失败 + """ + global SourceDir, TempDir, OutputDir, ProjectName, ProjectVersion + global CompilerCmd, CompilerFlags, LinkerCmd, LinkerFlags, LinkerOutput + global TargetTriple, TargetDataLayout, SliceLevel, StrictMode + global IncludesDir + + if path is None: + return 1 + if _mbuddy is None: + return 1 + + # 打开文件 + f: fileio.File | t.CPtr = fileio.File(path, fileio.MODE.R) + if f.closed: + return 1 + + # 分配读取缓冲区 + CFG_BUF_SIZE: t.CSizeT = 8192 + buf: bytes = _mbuddy.alloc(CFG_BUF_SIZE) + if buf is None: + f.close() + return 1 + + bytes_read: t.CInt64T = f.read_all(buf, CFG_BUF_SIZE) + f.close() + if bytes_read <= 0: + return 1 + + # 解析 JSON + root: JsonValue | t.CPtr = json_parse(_mbuddy, buf) + if root is None: + return 1 + if not root.is_object(): + return 1 + + # 读取顶层字段 + SourceDir = root["source_dir"].as_string() if root["source_dir"] is not None else None + TempDir = root["temp_dir"].as_string() if root["temp_dir"] is not None else None + OutputDir = root["output_dir"].as_string() if root["output_dir"] is not None else None + ProjectName = root["name"].as_string() if root["name"] is not None else None + ProjectVersion = root["version"].as_string() if root["version"] is not None else None + + # 读取 compiler 子对象 + compiler: JsonValue | t.CPtr = root["compiler"] + if compiler is not None and compiler.is_object(): + CompilerCmd = compiler["cmd"].as_string() if compiler["cmd"] is not None else None + # 读取 compiler.flags 数组，拼接成空格分隔的字符串 + cc_flags_val: JsonValue | t.CPtr = compiler["flags"] + CompilerFlags = _join_flags_array(cc_flags_val) + + # 读取 linker 子对象 + linker: JsonValue | t.CPtr = root["linker"] + if linker is not None and linker.is_object(): + LinkerCmd = linker["cmd"].as_string() if linker["cmd"] is not None else None + LinkerOutput = linker["output"].as_string() if linker["output"] is not None else None + # 读取 linker.flags 数组，拼接成空格分隔的字符串 + ld_flags_val: JsonValue | t.CPtr = linker["flags"] + LinkerFlags = _join_flags_array(ld_flags_val) + + # 读取 target 子对象 + target: JsonValue | t.CPtr = root["target"] + if target is not None and target.is_object(): + TargetTriple = target["triple"].as_string() if target["triple"] is not None else None + TargetDataLayout = target["datalayout"].as_string() if target["datalayout"] is not None else None + + # 读取 options 子对象 + options: JsonValue | t.CPtr = root["options"] + if options is not None and options.is_object(): + sl: JsonValue | t.CPtr = options["slice_level"] + if sl is not None and sl.is_int(): + SliceLevel = sl.as_int() + sm: JsonValue | t.CPtr = options["strict_mode"] + if sm is not None and sm.is_bool(): + StrictMode = 1 if sm.as_bool() else 0 + + # 读取 includes 数组（取首个元素作为 includes 目录） + includes_val: JsonValue | t.CPtr = root["includes"] + if includes_val is not None and includes_val.is_array(): + arr_len: t.CSizeT = includes_val.__len__() + if arr_len > 0: + first_inc: JsonValue | t.CPtr = includes_val.get_item(0) + if first_inc is not None and first_inc.is_string(): + IncludesDir = first_inc.as_string() + + return 0 + + +def _join_flags_array(flags_val: JsonValue | t.CPtr) -> str: + """将 JSON 字符串数组拼接成空格分隔的字符串 + + 用于 linker.flags 和 compiler.flags 数组。 + 用 while + 索引遍历，避免 list 迭代器 bug。 + + Args: + flags_val: JsonValue，应为字符串数组 + + Returns: + 空格分隔的字符串，或 None 如果 flags_val 为 None 或非数组或空数组 + """ + if flags_val is None: + return None + if not flags_val.is_array(): + return None + + arr_len: t.CSizeT = flags_val.__len__() + if arr_len == 0: + return None + + # 第一遍：计算总长度（所有 flag 长度 + 空格分隔符） + total_len: t.CSizeT = 0 + i: t.CSizeT = 0 + while i < arr_len: + item: JsonValue | t.CPtr = flags_val.get_item(i) + if item is not None and item.is_string(): + s: str = item.as_string() + if s is not None: + total_len += string.strlen(s) + 1 # +1 for space + i += 1 + + if total_len == 0: + return None + + # 分配缓冲区（total_len 已含每个 flag 后的空格，最后一个是 '\0'） + buf: str = _mbuddy.alloc(total_len + 1) + if buf is None: + return None + + # 第二遍：拼接（flag 之间用空格分隔） + pos: t.CSizeT = 0 + i = 0 + while i < arr_len: + item = flags_val.get_item(i) + if item is not None and item.is_string(): + s = item.as_string() + if s is not None: + slen: t.CSizeT = string.strlen(s) + if pos > 0: + buf[pos] = ' ' + pos += 1 + string.strcpy(buf + pos, s) + pos += slen + i += 1 + buf[pos] = '\0' + return buf + + +def _join_path(rel: str, project_dir: str) -> str: + """将相对路径与 project_dir 拼接为绝对路径 + + - ./App + Test → Test/App + - ./temp + Test → Test/temp + - ../includes + Test → Test/../includes（OS 解析为 includes） + - 绝对路径保持不变 + - None 返回 None + """ + if rel is None: + return None + # 绝对路径（以 / 或 \ 开头）保持不变 + if rel[0] == '/' or rel[0] == '\\': + return rel + + rel_len: t.CSizeT = string.strlen(rel) + dir_len: t.CSizeT = string.strlen(project_dir) + + # 处理 ../ 或 ..\\ 前缀：拼接 project_dir/../... + if rel_len >= 3 and rel[0] == '.' and rel[1] == '.': + if rel[2] == '/' or rel[2] == '\\': + if dir_len == 0: + return rel + total_len: t.CSizeT = dir_len + 1 + rel_len + 1 + buf: str = _mbuddy.alloc(total_len) + if buf is None: + return rel + viperlib.snprintf(buf, total_len, "%s/%s", project_dir, rel) + return buf + + # 处理 ./ 或 .\\ 前缀：拼接 project_dir + rel[2:] + rel_off: t.CSizeT = 0 + if rel_len >= 2 and rel[0] == '.': + if rel[1] == '/' or rel[1] == '\\': + rel_off = 2 + else: + return rel # 不是 ./ 开头，保持不变 + else: + return rel # 不是 . 开头，保持不变 + + if dir_len == 0: + # project_dir 为空（当前目录），去掉 ./ 前缀即可 + return rel + rel_off + # 拼接: project_dir + "/" + rel[rel_off:] + total_len = dir_len + 1 + (rel_len - rel_off) + 1 + buf = _mbuddy.alloc(total_len) + if buf is None: + return rel # 分配失败，返回原路径 + viperlib.snprintf(buf, total_len, "%s/%s", project_dir, rel + rel_off) + return buf + + +def resolve_paths(project_dir: str) -> int: + """解析并规范化工程路径（src/temp/output/includes） + + 将 project.vpj 中的相对路径（以 ./ 开头）转换为基于 project_dir 的路径。 + + Args: + project_dir: 工程根目录（project.vpj 所在目录） + + Returns: + 0 表示成功，非 0 表示失败 + """ + global SourceDir, TempDir, OutputDir, IncludesDir + if project_dir is None: + return 1 + if _mbuddy is None: + return 1 + + SourceDir = _join_path(SourceDir, project_dir) + TempDir = _join_path(TempDir, project_dir) + OutputDir = _join_path(OutputDir, project_dir) + # includes 路径可能是 ../includes 形式，需要基于 project_dir 解析 + if IncludesDir is not None: + IncludesDir = _join_path(IncludesDir, project_dir) + return 0 + + +def get_includes_binary_dir() -> str: + """获取 includes.binary 目录路径（includes 目录 + .binary 后缀） + + includes 目录如 ../includes，则 includes.binary 为 ../includes.binary + """ + if IncludesDir is None: + return None + inc_len: t.CSizeT = string.strlen(IncludesDir) + buf: str = _mbuddy.alloc(inc_len + 8) + if buf is None: + return None + viperlib.snprintf(buf, inc_len + 8, "%s.binary", IncludesDir) + return buf + + +def print_config(): + """打印当前配置（调试用）""" + if ProjectName is not None: + stdio.printf(" project: %s v%s\n", ProjectName, ProjectVersion if ProjectVersion is not None else "?") + if SourceDir is not None: + stdio.printf(" source_dir: %s\n", SourceDir) + if TempDir is not None: + stdio.printf(" temp_dir: %s\n", TempDir) + if OutputDir is not None: + stdio.printf(" output_dir: %s\n", OutputDir) + if CompilerCmd is not None: + stdio.printf(" compiler: %s\n", CompilerCmd) + if LinkerCmd is not None: + stdio.printf(" linker: %s -> %s\n", LinkerCmd, LinkerOutput if LinkerOutput is not None else "?") + if IncludesDir is not None: + stdio.printf(" includes: %s\n", IncludesDir) \ No newline at end of file diff --git a/TransPyV/App/lib/Projectrans/Utils.py b/TransPyV/App/lib/Projectrans/Utils.py new file mode 100644 index 0000000..e0e25a1 --- /dev/null +++ b/TransPyV/App/lib/Projectrans/Utils.py @@ -0,0 +1,117 @@ +import t, c +from stdint import * +import memhub +import string +import stdlib +import viperlib +import hashlib +import w32.win32file +import w32.win32base + + +# ============================================================ +# Utils - 工程工具函数 +# +# 提供 SHA1 计算和目录清理等通用工具函数。 +# ============================================================ + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + + +# ============================================================ +# compute_sha1 - 计算字符串的 SHA1，返回前 16 个十六进制字符 +# +# 使用 includes/hashlib 库的 sha1 类计算摘要，转为十六进制字符串。 +# 参考 Projectrans.py 的 compute_sha1（hashlib.sha1().hexdigest()[:16]）。 +# ============================================================ +def compute_sha1(pool: memhub.MemBuddy | t.CPtr, content: str) -> str: + """计算字符串的 SHA1，返回前 16 个十六进制字符（用于文件命名）""" + if content is None or pool is None: + return None + + # 构造 sha1 对象（__init__ 会初始化 state/count/buf） + # ctx 在栈上分配（alloca），函数内使用完即丢弃，无需堆分配 + ctx: hashlib.sha1 | t.CPtr = hashlib.sha1() + if ctx is None: + return None + + # 计算摘要 + ctx.update(content) + digest: bytes = pool.alloc(hashlib.SHA1_DIGEST_LEN) + if digest is None: + return None + ctx.final(digest) + + # 转为十六进制字符串（取前 8 字节 = 16 个十六进制字符） + hex_buf: str = pool.alloc(17) + if hex_buf is None: + return None + for i in range(8): + hi: int = (digest[i] >> 4) & 0xF + lo: int = digest[i] & 0xF + if hi < 10: + hex_buf[i * 2] = '0' + hi + else: + hex_buf[i * 2] = 'a' + (hi - 10) + if lo < 10: + hex_buf[i * 2 + 1] = '0' + lo + else: + hex_buf[i * 2 + 1] = 'a' + (lo - 10) + hex_buf[16] = '\0' + return hex_buf + + +# ============================================================ +# CleanDir - 删除目录中所有文件（非递归，保留目录本身） +# +# 用于 --clean 选项：清理 temp_dir 和 output_dir 中的旧文件。 +# ============================================================ +def CleanDir(dir_path: str) -> int: + """删除目录中所有文件（非递归），返回删除的文件数，-1 表示错误""" + if dir_path is None: + return -1 + dir_len: t.CSizeT = string.strlen(dir_path) + pattern: bytes = stdlib.malloc(dir_len + 8) + if pattern is None: + return -1 + viperlib.snprintf(pattern, dir_len + 8, "%s/*", dir_path) + + find_data: w32.win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + if find_data is None: + stdlib.free(pattern) + return -1 + string.memset(find_data, 0, w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + + handle: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(pattern, find_data) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + stdlib.free(pattern) + stdlib.free(find_data) + return 0 + + deleted: int = 0 + while 1: + fname: str = find_data.cFileName + if fname is not None: + # 跳过 . 和 .. + if fname[0] == '.': + if fname[1] == '\0': + fname = None + elif fname[1] == '.' and fname[2] == '\0': + fname = None + if fname is not None: + fname_len: t.CSizeT = string.strlen(fname) + full_path: bytes = stdlib.malloc(dir_len + fname_len + 2) + if full_path is not None: + viperlib.snprintf(full_path, dir_len + fname_len + 2, "%s/%s", dir_path, fname) + if w32.win32file.DeleteFileA(full_path) != 0: + deleted += 1 + stdlib.free(full_path) + + if w32.win32file.FindNextFileA(handle, find_data) == 0: + break + + w32.win32file.FindClose(handle) + stdlib.free(pattern) + stdlib.free(find_data) + return deleted diff --git a/TransPyV/App/lib/Projectrans/__init__.py b/TransPyV/App/lib/Projectrans/__init__.py new file mode 100644 index 0000000..64656ea --- /dev/null +++ b/TransPyV/App/lib/Projectrans/__init__.py @@ -0,0 +1,12 @@ +import t, c +from stdint import * + +from .Config import Load_project_config, resolve_paths +from .Utils import compute_sha1, CleanDir + +__all__ = [ + 'Load_project_config', + 'resolve_paths', + 'compute_sha1', + 'CleanDir', +] diff --git a/TransPyV/App/lib/__init__.py b/TransPyV/App/lib/__init__.py new file mode 100644 index 0000000..5294ca9 --- /dev/null +++ b/TransPyV/App/lib/__init__.py @@ -0,0 +1,27 @@ +import t, c +from stdint import * +import memhub +import lib.core.VLogger as VLogger + + +# ============================================================ +# lib 包入口 - 声明全局 _mbuddy 内存池指针 +# ============================================================ + +# 全局 mbuddy 内存池指针（由 App/main.py 初始化后注入） +_mbuddy: memhub.MemManager | t.CPtr + + +def InitLib(mb: memhub.MemManager | t.CPtr) -> int: + """初始化 lib 包的全局 _mbuddy 指针，并级联注入到所有子模块。 + + Args: + mb: memhub.MemManager 实例指针（MemBuddy 等子类通过多态传入） + + Returns: + 0 表示成功，非 0 表示失败 + """ + if mb is None: + return 1 + VLogger._mbuddy = mb + return 0 diff --git a/TransPyV/App/lib/core/BuildPipeline.py b/TransPyV/App/lib/core/BuildPipeline.py new file mode 100644 index 0000000..64f1183 --- /dev/null +++ b/TransPyV/App/lib/core/BuildPipeline.py @@ -0,0 +1,502 @@ +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesImports as HandlesImports + +# ============================================================ +# BuildPipeline - 编译管线 +# +# 负责将 LLVM IR 编译为可执行文件： +# 1. 写 .ll 文件到 temp 目录 +# 2. 调用 llc 编译 .ll → .obj +# 3. 调用 clang++ 链接 .obj → .exe +# ============================================================ + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 源代码缓冲区大小（1MB） +SRC_BUF_SIZE: t.CDefine = 1048576 + + +# ============================================================ +# TranslateFileGetTrans - 翻译单个文件，返回 Translator 对象 +# +# 读取文件 → AST 解析 → LLVM IR 翻译，返回 Translator 对象供 +# 调用者 dump_ir。命名空间隔离：includes 文件宽松模式（全部可见）， +# 用户文件严格模式（仅本地+import）。 +# +# Args: +# mb: 内存池 +# file_path: 源文件路径 +# sha1_val: 文件内容的 SHA1 前16字符（用于函数名混淆） +# +# Returns: +# Translator 对象（None 失败） +# ============================================================ +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, + sha1_val: str, + current_package: str = None) -> HandlesTranslator.Translator | t.CPtr: + """翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败 + + current_package: 当前文件所属包名（用于解析相对导入），None 表示顶级模块 + """ + if file_path is None: + return None + f: fileio.File | t.CPtr = fileio.File(file_path, fileio.MODE.R) + if f.closed: + return None + src_buf: bytes = stdlib.malloc(SRC_BUF_SIZE) + if src_buf is None: + f.close() + return None + bytes_read: LONG = f.read_all(src_buf, SRC_BUF_SIZE) + f.close() + if bytes_read <= 0: + return None + if bytes_read < SRC_BUF_SIZE: + src_buf[bytes_read] = 0 + else: + src_buf[SRC_BUF_SIZE - 1] = 0 + lx: ast.Lexer | t.CPtr = ast.new_lexer(mb) + ast._lexer_init(lx, src_buf, mb) + tokens: ast.Token | t.CPtr = ast.tokenize(lx) + tree: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens) + if tree is None: + return None + tr: HandlesTranslator.Translator | t.CPtr = mb.alloc(HandlesTranslator.Translator.__sizeof__()) + tr.__before_init__() + tr.__init__() + tr.ModuleSha1 = sha1_val + tr.CurrentPackage = current_package + # 设置当前文件名（供报错使用） + HandlesType.set_current_file(file_path) + # 模块切换：清空 CDefine 常量表，确保每个模块的 CDefine 常量正确隔离 + HandlesType.clear_cdefine_constants() + # 命名空间隔离：includes 文件宽松模式（全部可见），用户文件严格模式（仅本地+import） + strict_mode: int = 1 + if string.strstr(file_path, "includes") is not None: + strict_mode = 0 + HandlesStruct.reset_visible_structs(mb, strict_mode) + ret: int = tr.translate(tree) + if ret != 0: + return None + return tr + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + + def __new__(self) -> t.CPtr: + return t.CPtr(_mbuddy.alloc(BuildResult.__sizeof__())) + + def __init__(self): + self.Success = 0 + self.OutputPath = None + self.ErrorMsg = None + + +def ensure_dir(path: str) -> int: + """递归创建目录（类似 mkdir -p），目录已存在视为成功 + + Args: + path: 目录路径（支持 / 或 \\ 分隔符） + + Returns: + 0 成功（包括目录已存在），非 0 失败 + """ + if path is None: + return 1 + + path_len: t.CSizeT = string.strlen(path) + if path_len == 0: + return 0 + + # 复制路径到可写缓冲区（逐级截断用） + buf: bytes = _mbuddy.alloc(path_len + 1) + if buf is None: + return 1 + string.strcpy(buf, path) + + # 遇到分隔符时临时截断，创建每一层目录 + # CreateDirectoryA 在目录已存在时返回 0（失败），忽略即可 + for i in range(path_len): + ch: int = c.Deref(buf + i) + if ch == ord('/') or ch == ord('\\'): + saved: int = ch + buf[i] = '\0' + w32.win32file.CreateDirectoryA(buf, None) + buf[i] = saved + + # 创建最终目录 + w32.win32file.CreateDirectoryA(buf, None) + return 0 + + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: + """将 IR 缓冲区写入 .ll 文件 + + Args: + ir_buf: IR 文本缓冲区 + ir_len: IR 文本长度 + output_dir: 输出目录（temp 或 output） + module_name: 模块名（如 "main"） + + Returns: + 0 成功，非 0 失败 + """ + if ir_buf is None or output_dir is None: + return 1 + + # 构造文件路径: output_dir/module_name.ll + dir_len: t.CSizeT = string.strlen(output_dir) + name_len: t.CSizeT = string.strlen(module_name) + path_len: t.CSizeT = dir_len + 1 + name_len + 4 # dir/module.ll\0 + path: bytes = _mbuddy.alloc(path_len) + if path is None: + return 1 + viperlib.snprintf(path, path_len, "%s/%s.ll", output_dir, module_name) + + # 打开文件写入 + f: fileio.File | t.CPtr = fileio.File(path, fileio.MODE.W) + if f.closed: + return 1 + + written: t.CInt64T = f.write(ir_buf, ir_len) + f.close() + if written < 0: + return 1 + return 0 + + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: + """调用 llc 将 .ll 编译为 .obj + + Args: + ir_path: .ll 文件路径 + output_dir: 输出目录 + module_name: 模块名（用于生成 .obj 文件名） + cc_cmd: 编译器命令（如 "llc"） + cc_flags: 编译器参数（如 "-filetype=obj -relocation-model=pic"） + + Returns: + 0 成功，非 0 失败 + """ + if ir_path is None or cc_cmd is None: + return 1 + + # 构造命令: llc -filetype=obj -o output_dir/module_name.obj ir_path + cmd_len: t.CSizeT = string.strlen(cc_cmd) + string.strlen(cc_flags) + string.strlen(output_dir) + string.strlen(module_name) + string.strlen(ir_path) + 64 + cmd: bytes = _mbuddy.alloc(cmd_len) + if cmd is None: + return 1 + viperlib.snprintf(cmd, cmd_len, "%s %s -o %s/%s.obj %s", cc_cmd, cc_flags, output_dir, module_name, ir_path) + + result: subprocess.CompletedProcess | t.CPtr = subprocess.run(cmd, True, True) + if result is None: + stdio.printf("[FATAL][LLC] subprocess.run 返回 None: %s\n", module_name) + return 1 + if result.returncode != 0: + stdio.printf("[FATAL][LLC] 编译失败 (module=%s, cmd=%s)\n", module_name, cmd) + if result.stdout is not None: + stdio.printf("[LLC] 输出:\n%s\n", result.stdout) + return 1 + return 0 + + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: + """扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf + + Args: + includes_binary_dir: includes.binary 目录路径 + out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径） + out_size: 输出缓冲区大小 + + Returns: + 写入的字节数（不含 null 终止符），0 表示无文件或错误 + """ + if includes_binary_dir is None or out_buf is None or out_size == 0: + return 0 + + out_buf[0] = '\0' + out_pos: t.CSizeT = 0 + + # 构造搜索模式: dir/*.obj + dir_len: t.CSizeT = string.strlen(includes_binary_dir) + pattern: bytes = _mbuddy.alloc(dir_len + 8) + if pattern is None: + return 0 + viperlib.snprintf(pattern, dir_len + 8, "%s/*.obj", includes_binary_dir) + + # FindFirstFileA + find_data: w32.win32file.WIN32_FIND_DATAA | t.CPtr = _mbuddy.alloc(w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + if find_data is None: + return 0 + string.memset(find_data, 0, w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + + handle: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(pattern, find_data) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + return 0 + + # 需要 win32base 的 INVALID_HANDLE_VALUE + while True: + fname: str = find_data.cFileName + if fname is not None: + fname_len: t.CSizeT = string.strlen(fname) + # 构造完整路径并追加到 out_buf + # 路径格式: "dir/fname " + need: t.CSizeT = dir_len + 1 + fname_len + 2 + if out_pos + need < out_size: + if out_pos > 0: + out_buf[out_pos] = ' ' + out_pos += 1 + viperlib.snprintf(out_buf + out_pos, need, "%s/%s", includes_binary_dir, fname) + out_pos += dir_len + 1 + fname_len + else: + # 缓冲区不足，停止 + break + + if w32.win32file.FindNextFileA(handle, find_data) == 0: + break + + w32.win32file.FindClose(handle) + out_buf[out_pos] = '\0' + return out_pos + + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, + includes_binary_dir: str) -> int: + """调用 clang++ 链接 .obj 为 .exe + + Args: + output_dir: 输出目录（包含 .obj 文件） + module_name: 模块名 + linker_cmd: 链接器命令（如 "clang++"） + linker_flags: 链接器参数 + linker_output: 输出文件名（如 "test.exe"） + includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj） + + Returns: + 0 成功，非 0 失败 + """ + if output_dir is None or linker_cmd is None: + return 1 + + # 收集 includes.binary 的 .obj 文件路径 + EXTRA_BUF_SIZE: t.CSizeT = 32768 + extra_objs: bytes = _mbuddy.alloc(EXTRA_BUF_SIZE) + if extra_objs is None: + return 1 + extra_len: t.CSizeT = 0 + if includes_binary_dir is not None: + extra_len = collect_obj_files(includes_binary_dir, extra_objs, EXTRA_BUF_SIZE) + if extra_len > 0: + stdio.printf("[link] 附加 %d 字节的 includes.binary .obj 文件\n", extra_len) + else: + stdio.printf("[link] 警告: includes.binary 无 .obj 文件: %s\n", includes_binary_dir) + + # 构造命令: clang++ main.obj extra_objs -o output linker_flags + # 注意: .obj 文件必须在 -l 库标志之前，否则链接器无法解析符号依赖 + cmd_len: t.CSizeT = string.strlen(linker_cmd) + string.strlen(output_dir) + string.strlen(module_name) + string.strlen(linker_flags) + string.strlen(linker_output) + extra_len + 128 + cmd: bytes = _mbuddy.alloc(cmd_len) + if cmd is None: + return 1 + if extra_len > 0: + viperlib.snprintf(cmd, cmd_len, "%s %s/%s.obj %s -o %s/%s %s", + linker_cmd, output_dir, module_name, extra_objs, + output_dir, linker_output, linker_flags) + else: + viperlib.snprintf(cmd, cmd_len, "%s %s/%s.obj -o %s/%s %s", + linker_cmd, output_dir, module_name, output_dir, linker_output, + linker_flags) + + result: subprocess.CompletedProcess | t.CPtr = subprocess.run(cmd, True, True) + if result is None: + stdio.printf("[link] subprocess.run 返回 None\n") + return 1 + if result.returncode != 0: + stdio.printf("[link] 链接失败，返回码: %d\n", result.returncode) + stdio.printf("[link] 命令: %s\n", cmd) + # 显示链接器错误输出（subprocess 将 stderr 合并到 stdout） + if result.stdout is not None: + stdio.printf("[link] 链接器输出:\n%s\n", result.stdout) + else: + stdio.printf("[link] 无输出捕获\n") + return 1 + return 0 + + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, + temp_dir: str, output_dir: str, module_name: str, + cc_cmd: str, cc_flags: str) -> int: + """编译 IR 到 .obj（不链接） + + Args: + ir_buf: LLVM IR 文本缓冲区 + ir_len: IR 文本长度 + temp_dir: 临时目录 + output_dir: 输出目录 + module_name: 模块名（SHA1） + cc_cmd: 编译器命令 + cc_flags: 编译器参数 + + Returns: + 0 成功，非 0 失败 + """ + if ir_buf is None or temp_dir is None or output_dir is None: + return 1 + + # Step 1: 写 .ll 文件 + ret: int = write_ir_to_file(ir_buf, ir_len, temp_dir, module_name) + if ret != 0: + stdio.printf("[compile] 写 .ll 失败: %s\n", module_name) + return 1 + + # Step 2: 构造 .ll 路径并编译 → .obj + name_len: t.CSizeT = string.strlen(module_name) + dir_len: t.CSizeT = string.strlen(temp_dir) + ir_path: bytes = _mbuddy.alloc(dir_len + name_len + 5) + if ir_path is None: + return 1 + viperlib.snprintf(ir_path, dir_len + name_len + 5, "%s/%s.ll", temp_dir, module_name) + + ret = compile_ll_to_obj(ir_path, output_dir, module_name, cc_cmd, cc_flags) + if ret != 0: + stdio.printf("[compile] llc 编译失败: %s\n", module_name) + return 1 + return 0 + + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, + linker_cmd: str, linker_flags: str, linker_output: str, + includes_binary_dir: str) -> int: + """链接多个 .obj 文件为 .exe + + Args: + obj_paths: 空格分隔的 .obj 文件完整路径字符串 + obj_paths_len: obj_paths 长度 + linker_cmd: 链接器命令（如 "clang++"） + linker_flags: 链接器参数 + linker_output: 输出文件完整路径 + includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj） + + Returns: + 0 成功，非 0 失败 + """ + if obj_paths is None or obj_paths_len == 0 or linker_cmd is None: + return 1 + + # 收集 includes.binary 的 .obj 文件路径 + EXTRA_BUF_SIZE: t.CSizeT = 32768 + extra_objs: bytes = _mbuddy.alloc(EXTRA_BUF_SIZE) + if extra_objs is None: + return 1 + extra_len: t.CSizeT = 0 + if includes_binary_dir is not None: + extra_len = collect_obj_files(includes_binary_dir, extra_objs, EXTRA_BUF_SIZE) + + # 构造命令: clang++ obj_paths extra_objs -o linker_output linker_flags + cmd_len: t.CSizeT = string.strlen(linker_cmd) + obj_paths_len + string.strlen(linker_flags) + string.strlen(linker_output) + extra_len + 128 + cmd: bytes = _mbuddy.alloc(cmd_len) + if cmd is None: + return 1 + if extra_len > 0: + viperlib.snprintf(cmd, cmd_len, "%s %s %s -o %s %s", + linker_cmd, obj_paths, extra_objs, + linker_output, linker_flags) + else: + viperlib.snprintf(cmd, cmd_len, "%s %s -o %s %s", + linker_cmd, obj_paths, linker_output, linker_flags) + + result: subprocess.CompletedProcess | t.CPtr = subprocess.run(cmd, True, True) + if result is None: + stdio.printf("[link] subprocess.run 返回 None\n") + return 1 + if result.returncode != 0: + stdio.printf("[link] 链接失败，返回码: %d\n", result.returncode) + stdio.printf("[link] 命令: %s\n", cmd) + if result.stdout is not None: + stdio.printf("[link] 链接器输出:\n%s\n", result.stdout) + return 1 + return 0 + + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, + temp_dir: str, output_dir: str, module_name: str, + cc_cmd: str, cc_flags: str, + linker_cmd: str, linker_flags: str, linker_output: str, + includes_binary_dir: str) -> BuildResult | t.CPtr: + """执行完整编译管线 + + Args: + ir_buf: LLVM IR 文本缓冲区 + ir_len: IR 文本长度 + temp_dir: 临时目录 + output_dir: 输出目录 + module_name: 模块名 + cc_cmd: 编译器命令 + cc_flags: 编译器参数 + linker_cmd: 链接器命令 + linker_flags: 链接器参数 + linker_output: 输出文件名 + includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj） + + Returns: + BuildResult 对象 + """ + result: BuildResult | t.CPtr = BuildResult() + if result is None: + return None + + # Step 0: 确保 temp/output 目录存在（自动创建，避免写文件失败） + ensure_dir(temp_dir) + ensure_dir(output_dir) + + # Step 1: 写 .ll 文件 + ret: int = write_ir_to_file(ir_buf, ir_len, temp_dir, module_name) + if ret != 0: + result.Success = 0 + result.ErrorMsg = "写 .ll 文件失败" + return result + + # Step 2: llc 编译 .ll → .obj + # 构造 .ll 文件路径 + name_len: t.CSizeT = string.strlen(module_name) + dir_len: t.CSizeT = string.strlen(temp_dir) + ir_path: bytes = _mbuddy.alloc(dir_len + name_len + 5) + if ir_path is None: + result.Success = 0 + result.ErrorMsg = "内存分配失败" + return result + viperlib.snprintf(ir_path, dir_len + name_len + 5, "%s/%s.ll", temp_dir, module_name) + + ret = compile_ll_to_obj(ir_path, output_dir, module_name, cc_cmd, cc_flags) + if ret != 0: + result.Success = 0 + result.ErrorMsg = "llc 编译失败" + return result + + # Step 3: clang++ 链接 .obj → .exe（附加 includes.binary 预编译 .obj） + ret = link_obj_to_exe(output_dir, module_name, linker_cmd, linker_flags, linker_output, + includes_binary_dir) + if ret != 0: + result.Success = 0 + result.ErrorMsg = "链接失败" + return result + + result.Success = 1 + return result \ No newline at end of file diff --git a/TransPyV/App/lib/core/Handles/HandlesAnnAssign.py b/TransPyV/App/lib/core/Handles/HandlesAnnAssign.py new file mode 100644 index 0000000..a7a13a2 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesAnnAssign.py @@ -0,0 +1,281 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesExprCall as HandlesExprCall + + +# ============================================================ +# HandlesAnnAssign - AnnAssign 语句处理（Mixin 继承模式） +# ============================================================ + + +# ============================================================ +# is_cdefine_annotation - 检测注解是否为 t.CDefine（模块级函数） +# +# 支持两种形式: +# 1. Attribute(Name('t'), 'CDefine') +# 2. BinOp(... | Attribute(Name('t'), 'CDefine'))（联合注解） +# ============================================================ +def is_cdefine_annotation(annot: ast.AST | t.CPtr) -> int: + """检测注解是否为 t.CDefine，返回 1 表示是，0 表示否""" + if annot is None: + return 0 + k: int = annot.kind() + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(annot) + if at.attr is not None and string.strcmp(at.attr, "CDefine") == 0: + return 1 + return 0 + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(annot) + if is_cdefine_annotation(bop.left) != 0: + return 1 + if is_cdefine_annotation(bop.right) != 0: + return 1 + return 0 + + +# ============================================================ +# extract_cdefine_int_value - 从 AnnAssign.value 提取整数常量（模块级函数） +# +# 仅支持 ast.Constant(INT)，其他形式返回 0 +# ============================================================ +def extract_cdefine_int_value(val_node: ast.AST | t.CPtr) -> int: + """从值节点提取整数常量（仅支持 Constant INT）""" + if val_node is None: + return 0 + if val_node.kind() != ast.ASTKind.Constant: + return 0 + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(val_node) + if cn.const_kind != ast.CONST_INT: + return 0 + return cn.int_val + + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + """AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # PreScan - 预扫描 AnnAssign，为有类型注解的变量提前创建 alloca + # + # 返回新增的变量数 + # ============================================================ + def PreScan(self, node: ast.AST | t.CPtr) -> int: + """预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部""" + if node is None: + return 0 + k: int = node.kind() + if k != ast.ASTKind.AnnAssign: + return 0 + + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(node) + if aa is None or aa.target is None: + return 0 + target: ast.AST | t.CPtr = aa.target + if target.kind() != ast.ASTKind.Name: + return 0 + + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(target) + if nm.id is None: + return 0 + + # CDefine 注解: 不创建 alloca，但注册常量到全局表 + # 供后续 t.CArray[elem_ty, NAME] 编译期解析使用 + if is_cdefine_annotation(aa.annotation) != 0: + pool_ps: memhub.MemBuddy | t.CPtr = self.Trans.Pool + val_ps: int = extract_cdefine_int_value(aa.value) + HandlesType.register_cdefine_constant(pool_ps, nm.id, val_ps) + return 0 + + # global/nonlocal 变量不需要局部 alloca + if HT.is_global_name(self.Trans, nm.id) != 0: + return 0 + if HT.is_nonlocal_name(self.Trans, nm.id) != 0: + return 0 + + # 检查是否已存在 + existing: llvmlite.Value | t.CPtr = HandlesVar.lookup_current( + self.Trans.SymTab, nm.id) + if existing is not None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + + # 从 annotation 推断类型（修复：原硬编码 i32 导致 str 等类型错误） + var_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + annot: ast.AST | t.CPtr = aa.annotation + if annot is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, annot, self.Trans._imported_modules, self.Trans._from_imports) + if resolved is None: + # 尝试特化泛型类注解（如 list[str]） + # resolve_annotation_type 返回 None 可能是因为泛型类未特化 + if annot.kind() == ast.ASTKind.Subscript: + sub_annot: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(annot) + if sub_annot.value is not None and sub_annot.value.kind() == ast.ASTKind.Name: + sub_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub_annot.value) + if sub_nm.id is not None: + template_cd: ast.ClassDef | t.CPtr = HandlesClassDef._find_generic_template(sub_nm.id) + if template_cd is not None: + type_args_ps: list[str] | t.CPtr = HandlesExprCall._extract_type_args_from_slice(pool, sub_annot.slice) + if type_args_ps is not None and type_args_ps.__len__() > 0: + spec_name_ps: str = HandlesClassDef._specialize_generic_class( + self.Trans, sub_nm.id, type_args_ps) + if spec_name_ps is not None: + resolved = HandlesType.resolve_annotation_type( + pool, annot, self.Trans._imported_modules, self.Trans._from_imports) + if resolved is not None: + var_ty = resolved + + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, var_ty) + if alloca is not None: + if HandlesVar.define_var( + self.Trans.SymTab, nm.id, alloca) == 0: + # 存储原始类型注解的类名（方法调用检测时，Ptr(i8) 回退到类名查找结构体） + if annot is not None: + cls_nm_ps: str = HandlesType.extract_class_name_from_annotation( + annot, self.Trans._imported_modules) + if cls_nm_ps is not None: + HandlesVar.set_var_annot_class_name( + self.Trans.SymTab, nm.id, cls_nm_ps) + return 0 + return 0 + + # ============================================================ + # Handle - 翻译 AnnAssign 语句 + # + # 返回新增的变量数 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译 AnnAssign(target=Name, annotation=..., value=expr)""" + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(node) + if aa is None: + return 0 + + target: ast.AST | t.CPtr = aa.target + if target is None or target.kind() != ast.ASTKind.Name: + return 0 + + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(target) + if nm.id is None: + return 0 + + # CDefine 注解: 编译期常量，不生成运行时代码 + # 注册到全局 CDefine 表供 t.CArray[elem_ty, NAME] 解析 + # 注: PreScan 已注册过，此处保证即使跳过 PreScan 也能正确注册 + if is_cdefine_annotation(aa.annotation) != 0: + pool_cd: memhub.MemBuddy | t.CPtr = self.Trans.Pool + val_cd: int = extract_cdefine_int_value(aa.value) + HandlesType.register_cdefine_constant(pool_cd, nm.id, val_cd) + return 0 + + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + mod: llvmlite.LLVMModule | t.CPtr = self.Trans.Module + + # 确定类型（从 annotation 推断） + var_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + annot: ast.AST | t.CPtr = aa.annotation + if annot is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, annot, self.Trans._imported_modules, self.Trans._from_imports) + if resolved is not None: + var_ty = resolved + + # global 变量：写入模块作用域中的全局变量 + if HT.is_global_name(self.Trans, nm.id) != 0: + mod_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_module_var( + self.Trans.SymTab, nm.id) + if mod_alloca is not None and aa.value is not None: + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, aa.value, + None, 0, self.Trans) + if rhs_val is not None: + target_ty: llvmlite.LLVMType | t.CPtr = None + if mod_alloca.Ty is not None: + target_ty = mod_alloca.Ty.Pointee + if target_ty is not None: + rhs_val = HandlesExpr.coerce_to_type(builder, rhs_val, target_ty) + llvmlite.build_store(builder, rhs_val, mod_alloca) + return 0 + + # nonlocal 变量：通过闭包 env 写入 + if HT.is_nonlocal_name(self.Trans, nm.id) != 0: + if aa.value is not None: + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, aa.value, + None, 0, self.Trans) + if rhs_val is not None: + nl_ptr: llvmlite.Value | t.CPtr = HandlesNonlocal.get_nonlocal_var_ptr( + self.Trans, nm.id) + if nl_ptr is not None: + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + rhs_coerced: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + builder, rhs_val, i32_ty) + llvmlite.build_store(builder, rhs_coerced, nl_ptr) + return 0 + + # 普通局部变量 + # 创建 alloca + alloca: llvmlite.Value | t.CPtr = HandlesVar.get_or_create_sym( + self.Trans.SymTab, pool, builder, nm.id, var_ty) + if alloca is None: + return 0 + # 存储原始类型注解的类名（方法调用检测时，Ptr(i8) 回退到类名查找结构体） + if annot is not None: + cls_nm_hd: str = HandlesType.extract_class_name_from_annotation( + annot, self.Trans._imported_modules) + if cls_nm_hd is not None: + HandlesVar.set_var_annot_class_name( + self.Trans.SymTab, nm.id, cls_nm_hd) + + new_vars: int = 0 + existing: llvmlite.Value | t.CPtr = HandlesVar.lookup_current( + self.Trans.SymTab, nm.id) + if existing is None: + new_vars = 1 + + # 如果有初始值，store + if aa.value is not None: + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, aa.value, + None, 0, self.Trans) + if rhs_val is not None: + # 类型转换：将 rhs_val 转换为 alloca 的 pointee 类型 + # 修复：整数字面量是 i32，但变量可能是 i8/i16/i64，需 trunc/sext + target_ty: llvmlite.LLVMType | t.CPtr = None + if alloca.Ty is not None: + target_ty = alloca.Ty.Pointee + if target_ty is not None: + rhs_val = HandlesExpr.coerce_to_type(builder, rhs_val, target_ty) + llvmlite.build_store(builder, rhs_val, alloca) + + return 0 + + +# ============================================================ +# NewAnnAssignHandle - 工厂函数 +# ============================================================ +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: + h: AnnAssignHandle | t.CPtr = pool.alloc(AnnAssignHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, AnnAssignHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesAssign.py b/TransPyV/App/lib/core/Handles/HandlesAssign.py new file mode 100644 index 0000000..f5d825e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesAssign.py @@ -0,0 +1,231 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct + + +# ============================================================ +# HandlesAssign - Assign 语句处理（Mixin 继承模式） +# +# 对应 TransPyC 的 class AssignHandle(BaseHandle): +# @t.NoVTable 继承 Mixin 获得 Trans 字段（展平嵌入，无 vtable） +# 通过 self.Trans 访问共享状态（Pool/Module/_cur_builder/SymTab/...） +# 通过 self.Trans.ExprH / self.Trans.IfH 等访问其他 Handle +# ============================================================ + + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + """Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法""" + _CurrentClass: str # 模块私有状态 + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + self._CurrentClass = None + + # ============================================================ + # Handle - 处理 Assign 语句，返回新增变量数（0 或 1） + # + # 对应 TransPyC AssignHandle._HandleAssignLlvm + # 共享状态从 self.Trans 获取，无需 11 个参数 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + asgn: ast.Assign | t.CPtr = (ast.Assign | t.CPtr)(node) + if asgn is None: + stdio.printf("[ASGN] cast failed\n") + return 0 + + targets: list[ast.AST | t.CPtr] | t.CPtr = asgn.targets + if targets is None: + stdio.printf("[ASGN] targets is None\n") + return 0 + + # 从 self.Trans 取共享状态（替代 11 个参数） + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + mod: llvmlite.LLVMModule | t.CPtr = self.Trans.Module + + # 翻译 RHS 值 + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, asgn.value, None, 0, self.Trans) + if rhs_val is None: + stdio.printf("[ASGN] rhs_val is None\n") + return 0 + + new_vars: int = 0 + tn: t.CSizeT = targets.__len__() + for ti in range(tn): + target: ast.AST | t.CPtr = targets.get(ti) + if target is None: + continue + + tk: int = target.kind() + + # Subscript 赋值: arr[i] = val / ptr[i] = val / list[i] = val + if tk == ast.ASTKind.Subscript: + # 检查是否是 list[T] 类型的 Subscript（泛型类不注册 struct） + # list 的 subscript 赋值走 __setitem__ 内联路径 + list_obj: llvmlite.Value | t.CPtr = HandlesExpr.is_list_subscript( + target, self.Trans) + if list_obj is not None: + # list[T] 类型: 内联生成 __setitem__ 逻辑 + sub_node: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(target) + if sub_node is not None and sub_node.slice is not None: + list_idx_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, sub_node.slice, None, 0, self.Trans) + if list_idx_val is not None: + HandlesExpr.list_setitem_inline( + builder, pool, list_obj, list_idx_val, rhs_val) + continue + # 普通 Subscript 赋值 + elem_ptr: llvmlite.Value | t.CPtr = HandlesExpr.get_subscript_ptr( + builder, pool, mod, target, self.Trans) + if elem_ptr is not None: + store_val: llvmlite.Value | t.CPtr = rhs_val + if elem_ptr.Ty is not None: + elem_ty: llvmlite.LLVMType | t.CPtr = elem_ptr.Ty.Pointee + if elem_ty is not None: + store_val = HandlesExpr.coerce_to_type( + builder, rhs_val, elem_ty) + llvmlite.build_store(builder, store_val, elem_ptr) + else: + # get_subscript_ptr 返回 None: 尝试 __setitem__ 运算符重载 + # 适用于自定义类（如 hashtable[key]=val → hashtable.__setitem__(key, val)） + sub_asgn: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(target) + setitem_done: int = 0 + if sub_asgn is not None and sub_asgn.value is not None: + if sub_asgn.value.kind() == ast.ASTKind.Name: + sub_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub_asgn.value) + if sub_nm.id is not None: + sub_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + self.Trans.SymTab, sub_nm.id) + if sub_alloca is not None and sub_alloca.Ty is not None: + if HandlesExpr.is_ptr_type(sub_alloca.Ty) != 0: + sub_pointee: llvmlite.LLVMType | t.CPtr = sub_alloca.Ty.Pointee + if sub_pointee is not None: + cls_nm_set: str = HandlesStruct.get_class_name_by_type(pool, sub_pointee) + if cls_nm_set is not None: + obj_val_set: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, sub_pointee, sub_alloca) + if obj_val_set is not None: + key_val_set: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, sub_asgn.slice, None, 0, self.Trans) + if key_val_set is not None: + arg_vals_set: t.CSizeT | t.CPtr = pool.alloc(16) + if arg_vals_set is not None: + arg_vals_set[0] = t.CSizeT(key_val_set) + arg_vals_set[1] = t.CSizeT(rhs_val) + HandlesExprCall._call_method_on_ptr( + pool, builder, mod, cls_nm_set, "__setitem__", + obj_val_set, arg_vals_set, 2, self.Trans) + setitem_done = 1 + if setitem_done == 0: + HandlesType.fatal_error(target, "subscript ptr is None") + continue + + # Attribute 赋值: obj.field = val + if tk == ast.ASTKind.Attribute: + field_ptr: llvmlite.Value | t.CPtr = HandlesExpr.get_attribute_ptr( + builder, pool, mod, target, self.Trans) + if field_ptr is not None: + # 获取字段类型，对 rhs_val 进行类型转换（如 i32 → i64） + store_val: llvmlite.Value | t.CPtr = rhs_val + if field_ptr.Ty is not None: + field_ty: llvmlite.LLVMType | t.CPtr = field_ptr.Ty.Pointee + if field_ty is not None: + store_val = HandlesExpr.coerce_to_type( + builder, rhs_val, field_ty) + llvmlite.build_store(builder, store_val, field_ptr) + else: + # 构造详细错误信息 + attr_node: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(target) + attr_name: str = "(unknown)" + obj_name: str = "(unknown)" + if attr_node is not None: + attr_name = attr_node.attr + if attr_node.value is not None and attr_node.value.kind() == ast.ASTKind.Name: + obj_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(attr_node.value) + obj_name = obj_nm.id + err_buf: t.CChar | t.CPtr = pool.alloc(256) + if err_buf is not None: + viperlib.snprintf(err_buf, 256, + "attribute ptr is None: %s.%s", + obj_name, attr_name) + HandlesType.fatal_error(target, err_buf) + else: + HandlesType.fatal_error(target, "attribute ptr is None") + continue + + # Name 赋值: var = val + if tk == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(target) + if nm.id is not None: + # global 变量：写入模块作用域中的全局变量 + if HT.is_global_name(self.Trans, nm.id) != 0: + mod_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_module_var( + self.Trans.SymTab, nm.id) + if mod_alloca is not None: + target_ty: llvmlite.LLVMType | t.CPtr = None + if mod_alloca.Ty is not None: + target_ty = mod_alloca.Ty.Pointee + if target_ty is not None: + rhs_val = HandlesExpr.coerce_to_type(builder, rhs_val, target_ty) + llvmlite.build_store(builder, rhs_val, mod_alloca) + continue + + # nonlocal 变量：通过闭包 env 写入 + if HT.is_nonlocal_name(self.Trans, nm.id) != 0: + nl_ptr: llvmlite.Value | t.CPtr = HandlesNonlocal.get_nonlocal_var_ptr( + self.Trans, nm.id) + if nl_ptr is not None: + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + rhs_coerced: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + builder, rhs_val, i32_ty) + llvmlite.build_store(builder, rhs_coerced, nl_ptr) + continue + + # 普通局部变量 + alloca: llvmlite.Value | t.CPtr = HandlesVar.get_or_create_sym( + self.Trans.SymTab, pool, builder, nm.id, rhs_val.Ty) + if alloca is not None: + # 按 alloca 类型对值进行转换（如 double → float） + store_val: llvmlite.Value | t.CPtr = rhs_val + if alloca.Ty is not None: + alloca_ty: llvmlite.LLVMType | t.CPtr = alloca.Ty.Pointee + if alloca_ty is not None: + store_val = HandlesExpr.coerce_to_type( + builder, rhs_val, alloca_ty) + llvmlite.build_store(builder, store_val, alloca) + existing: llvmlite.Value | t.CPtr = HandlesVar.lookup_current( + self.Trans.SymTab, nm.id) + if existing is None: + new_vars += 1 + else: + stdio.printf("[ASGN] alloca failed for %s\n", nm.id) + + return new_vars + + +# ============================================================ +# NewAssignHandle - 工厂函数：分配并初始化 AssignHandle +# ============================================================ +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: + h: AssignHandle | t.CPtr = pool.alloc(AssignHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, AssignHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesAugAssign.py b/TransPyV/App/lib/core/Handles/HandlesAugAssign.py new file mode 100644 index 0000000..3db6e2e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesAugAssign.py @@ -0,0 +1,254 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import stdio +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal + + +# ============================================================ +# HandlesAugAssign - 增强赋值语句处理（Mixin 继承模式） +# +# 处理 += -= *= /= %= &= |= ^= <<= >>= +# 流程: load target → apply binop → store result +# +# 支持 local/global/nonlocal 三种变量作用域 +# ============================================================ + + +@t.NoVTable +class AugAssignHandle(HandlesBase.Mixin): + """增强赋值处理器 (+=, -=, *=, etc.)：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # Handle - 处理 AugAssign 语句，返回新增变量数（始终为 0） + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译增强赋值语句 (x += 1, y -= 2, etc.)""" + if node is None: + return 0 + + aug: ast.AugAssign | t.CPtr = (ast.AugAssign | t.CPtr)(node) + if aug is None: + return 0 + + target: ast.AST | t.CPtr = aug.target + if target is None: + return 0 + + tk: int = target.kind() + + # Attribute 目标: self.field += 1 + # 流程: get_attribute_ptr → load → binop → store + if tk == ast.ASTKind.Attribute: + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + mod: llvmlite.LLVMModule | t.CPtr = self.Trans.Module + + # 1. 获取字段指针 + field_ptr: llvmlite.Value | t.CPtr = HandlesExpr.get_attribute_ptr( + builder, pool, mod, target, self.Trans) + if field_ptr is None: + stdio.printf("[AUGASGN] attribute ptr is None\n") + return 0 + + # 2. 确定字段类型并加载当前值 + target_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + if field_ptr.Ty is not None: + target_ty = field_ptr.Ty.Pointee + cur_val: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, target_ty, field_ptr) + if cur_val is None: + stdio.printf("[AUGASGN] cannot load attribute\n") + return 0 + + # 3. 翻译 RHS 值 + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, aug.value, None, 0, self.Trans) + if rhs_val is None: + stdio.printf("[AUGASGN] rhs is None\n") + return 0 + + # 4. 应用二元运算 + result: llvmlite.Value | t.CPtr = _apply_aug_op( + pool, builder, aug.op, cur_val, rhs_val) + if result is None: + stdio.printf("[AUGASGN] binop failed for attr op=%d\n", aug.op) + return 0 + + # 5. 类型对齐并存储 + result = HandlesExpr.coerce_to_type(builder, result, target_ty) + if result is None: + return 0 + llvmlite.build_store(builder, result, field_ptr) + return 0 + + if tk != ast.ASTKind.Name: + stdio.printf("[AUGASGN] only Name/Attribute target supported\n") + return 0 + + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(target) + if nm is None or nm.id is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + mod: llvmlite.LLVMModule | t.CPtr = self.Trans.Module + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + + # 1. 确定变量作用域类型: 0=local, 1=global, 2=nonlocal + scope_type: int = 0 + if HT.is_global_name(self.Trans, nm.id) != 0: + scope_type = 1 + elif HT.is_nonlocal_name(self.Trans, nm.id) != 0: + scope_type = 2 + + # 2. 加载当前值 + cur_val: llvmlite.Value | t.CPtr = None + target_alloca: llvmlite.Value | t.CPtr = None + target_ty: llvmlite.LLVMType | t.CPtr = i32_ty + + if scope_type == 1: + # global 变量 + target_alloca = HandlesVar.lookup_module_var( + self.Trans.SymTab, nm.id) + if target_alloca is not None: + if target_alloca.Ty is not None: + target_ty = target_alloca.Ty.Pointee + cur_val = llvmlite.build_load(builder, target_ty, target_alloca) + elif scope_type == 2: + # nonlocal 变量（通过闭包 env） + cur_val = HandlesNonlocal.load_nonlocal_var(self.Trans, nm.id) + if cur_val is not None: + target_ty = cur_val.Ty + else: + # 普通局部变量 + target_alloca = HandlesVar.lookup_var(self.Trans.SymTab, nm.id) + if target_alloca is not None: + if target_alloca.Ty is not None: + target_ty = target_alloca.Ty.Pointee + cur_val = llvmlite.build_load(builder, target_ty, target_alloca) + + if cur_val is None: + stdio.printf("[AUGASGN] cannot load target %s\n", nm.id) + return 0 + + # 3. 翻译 RHS 值 + rhs_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, aug.value, None, 0, self.Trans) + if rhs_val is None: + stdio.printf("[AUGASGN] rhs is None\n") + return 0 + + # 4. 类型对齐 + 应用二元运算 + # 注意: AugAssign 不走运算符重载（语义上需要 __iadd__ 而非 __add__） + result: llvmlite.Value | t.CPtr = _apply_aug_op( + pool, builder, aug.op, cur_val, rhs_val) + if result is None: + stdio.printf("[AUGASGN] binop failed for op=%d\n", aug.op) + return 0 + + # 5. 存储结果 + result = HandlesExpr.coerce_to_type(builder, result, target_ty) + if result is None: + return 0 + + if scope_type == 1: + # global 变量 + if target_alloca is not None: + llvmlite.build_store(builder, result, target_alloca) + elif scope_type == 2: + # nonlocal 变量 + nl_ptr: llvmlite.Value | t.CPtr = HandlesNonlocal.get_nonlocal_var_ptr( + self.Trans, nm.id) + if nl_ptr is not None: + llvmlite.build_store(builder, result, nl_ptr) + else: + # 普通局部变量 + if target_alloca is not None: + llvmlite.build_store(builder, result, target_alloca) + + return 0 + + +# ============================================================ +# _apply_aug_op - 应用增强赋值的二元运算 +# +# 支持指针算术: ptr += int / ptr -= int +# 整数运算自动类型提升 +# ============================================================ +def _apply_aug_op(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + op: int, + lhs: llvmlite.Value | t.CPtr, + rhs: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: + """应用增强赋值的二元运算（指针算术 + 整数运算）""" + lhs_bits: int = HandlesExpr.get_llvm_type_bits(lhs.Ty) + rhs_bits: int = HandlesExpr.get_llvm_type_bits(rhs.Ty) + + # 指针算术: ptr += int / ptr -= int + if lhs_bits == 0 and rhs_bits != 0: + if op == ast.OpKind.Add or op == ast.OpKind.Sub: + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + ptr_as_int: llvmlite.Value | t.CPtr = llvmlite.build_ptrtoint(builder, lhs, i64_ty) + int_val: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type(builder, rhs, i64_ty) + if ptr_as_int is None or int_val is None: + return None + if op == ast.OpKind.Add: + result: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, ptr_as_int, int_val) + else: + result = llvmlite.build_sub(builder, ptr_as_int, int_val) + if result is None: + return None + return llvmlite.build_inttoptr(builder, result, lhs.Ty) + return None + + # 整数运算：类型提升 + if lhs_bits > rhs_bits and rhs_bits > 0: + rhs = HandlesExpr.coerce_to_type(builder, rhs, lhs.Ty) + elif rhs_bits > lhs_bits and lhs_bits > 0: + lhs = HandlesExpr.coerce_to_type(builder, lhs, rhs.Ty) + + if op == ast.OpKind.Add: + return llvmlite.build_add(builder, lhs, rhs) + elif op == ast.OpKind.Sub: + return llvmlite.build_sub(builder, lhs, rhs) + elif op == ast.OpKind.Mult: + return llvmlite.build_mul(builder, lhs, rhs) + elif op == ast.OpKind.Div: + return llvmlite.build_sdiv(builder, lhs, rhs) + elif op == ast.OpKind.Mod: + return llvmlite.build_srem(builder, lhs, rhs) + elif op == ast.OpKind.BitAnd: + return llvmlite.build_and(builder, lhs, rhs) + elif op == ast.OpKind.BitOr: + return llvmlite.build_or(builder, lhs, rhs) + elif op == ast.OpKind.BitXor: + return llvmlite.build_xor(builder, lhs, rhs) + elif op == ast.OpKind.LShift: + return llvmlite.build_shl(builder, lhs, rhs) + elif op == ast.OpKind.RShift: + return llvmlite.build_ashr(builder, lhs, rhs) + return None + + +# ============================================================ +# NewAugAssignHandle - 工厂函数 +# ============================================================ +def NewAugAssignHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> AugAssignHandle | t.CPtr: + h: AugAssignHandle | t.CPtr = pool.alloc(AugAssignHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, AugAssignHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesBase.py b/TransPyV/App/lib/core/Handles/HandlesBase.py new file mode 100644 index 0000000..764974d --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesBase.py @@ -0,0 +1,262 @@ +import t, c +from stdint import * +import string +import viperlib +import memhub +import hashtable +import lib.core.Handles.HandlesTranslator as HT + + +# ============================================================ +# TypeKind - 类型种类枚举 +# +# 替代 TransPyC 中 CType.position frozenset 的元属性系统。 +# 每个 TypeInfo 通过 Kind 字段标记其种类，避免 Python ClassVar 反射。 +# ============================================================ +class TypeKind(t.CEnum): + Basic: t.State # 基本类型（int/char/float/double/bool 等） + Pointer: t.State # 指针（T*）— 指针层数由 TypeInfo.PtrCount 表达 + Struct: t.State # 结构体/类 + Union: t.State # 联合体 + Enum: t.State # 枚举 + Typedef: t.State # 类型别名 + Function: t.State # 函数指针 + Array: t.State # 数组 T[N] + Void: t.State # void + + +# ============================================================ +# TypeInfo - 类型信息结构 +# +# 设计原则（相对 TransPyC CTypeInfo 的简化）: +# - POD 数据结构，无虚函数，无 Python 元特性（ClassVar/Generic/frozenset） +# - 通过 mbuddy 分配，调用方管理生命周期（参考 llvmlite Module 模式） +# - 通过 Kind 枚举代替 position frozenset +# - PtrCount 表达指针层数（0=非指针, 1=T*, 2=T**） +# - IsSigned 三态：-1=不适用(void/float), 0=unsigned, 1=signed +# +# 字段布局（8 字段）: +# Kind : i32 类型种类（TypeKind 枚举值） +# Name : i8* 类型名（如 'int', 'CInt', 'MyStruct'），可能为 None +# Size : i32 位宽（0=void/未知；int=32, char=8, double=64, long long=64） +# Align : i32 对齐字节数（0=默认） +# IsSigned : i32 -1=N/A, 0=unsigned, 1=signed +# PtrCount : i32 指针层数 +# IsConst : i32 const 限定符（0/1） +# IsVolatile : i32 volatile 限定符（0/1） +# +# 注意：Size 使用位宽（bit width）而非字节数，与 LLVM IR 的 i{N} 和 +# TransPyC CType.Size 约定一致（CInt.Size=32, CChar.Size=8）。 +# ============================================================ +@t.NoVTable +class TypeInfo: + Kind: int + Name: str + Size: int + Align: int + IsSigned: int + PtrCount: int + IsConst: int + IsVolatile: int + + +# ============================================================ +# TypeRegistry - 类型注册表 +# +# 维护 name(str) → TypeInfo* 映射，基于 HashTable 实现 O(1) 查找。 +# 替代 TransPyC CTypeRegistry 的 _name_to_class dict。 +# 内置基本类型（int/char/void/...）由 InitBasicTypes 注册。 +# ============================================================ +class TypeRegistry: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemManager | t.CPtr + + def Register(self, ti: TypeInfo | t.CPtr) -> int: + """注册一个 TypeInfo。ti.Name 字段必须已设置。 + + Returns: + 1 表示成功，0 表示失败（ti 为 None 或 Name 为 None） + """ + if ti is None: + return 0 + if ti.Name is None: + return 0 + self._ht[ti.Name] = ti + return 1 + + def Lookup(self, name: str) -> TypeInfo | t.CPtr: + """按名称查找 TypeInfo。 + + Returns: + 找到返回 TypeInfo*，找不到返回 None + """ + if name is None: + return None + return self._ht[name] + + def Has(self, name: str) -> int: + """检查类型是否已注册。""" + if name is None: + return 0 + return name in self._ht + + +# ============================================================ +# NewTypeInfo - 工厂函数：分配并初始化一个 TypeInfo +# +# 默认值：Kind=Basic, IsSigned=-1, 其余=0/None +# ============================================================ +def NewTypeInfo(pool: memhub.MemManager | t.CPtr) -> TypeInfo | t.CPtr: + ptr: TypeInfo | t.CPtr = pool.alloc(TypeInfo.__sizeof__()) + if ptr is None: + return None + string.memset(ptr, 0, TypeInfo.__sizeof__()) + ptr.Kind = TypeKind.Basic + ptr.IsSigned = -1 + return ptr + + +# ============================================================ +# NewTypeRegistry - 工厂函数：创建类型注册表 +# ============================================================ +def NewTypeRegistry(pool: memhub.MemManager | t.CPtr) -> TypeRegistry | t.CPtr: + ptr: TypeRegistry | t.CPtr = pool.alloc(TypeRegistry.__sizeof__()) + if ptr is None: + return None + string.memset(ptr, 0, TypeRegistry.__sizeof__()) + ptr.__mbuddy__ = pool + ptr._ht = hashtable.HashTable(pool) + return ptr + + +# ============================================================ +# TypeToLLVM - 将 TypeInfo 转换为 LLVM IR 类型字符串 +# +# 规则: +# - Kind == Void 或 (Size==0 且 IsSigned==-1 且 PtrCount==0) → "void" +# - void 有 PtrCount > 0 → "i8" + PtrCount 个 "*" +# - IsSigned == -1 且 Size > 0 → 浮点: half/float/double/fp128 +# - IsSigned != -1 且 Size > 0 → 整数: i{Size} +# - 未知 → "i8*" +# +# Args: +# buf: 输出缓冲区（i8*） +# buf_size: 缓冲区容量 +# ti: TypeInfo 指针 +# +# Returns: +# 写入的字符数（不含 NUL），失败返回 -1 +# ============================================================ +def TypeToLLVM(buf: t.CChar | t.CPtr, buf_size: t.CSizeT, + ti: TypeInfo | t.CPtr) -> int: + if buf is None: + return -1 + if ti is None: + return -1 + if buf_size == 0: + return -1 + + # 判定 void + is_void: int = 0 + if ti.Kind == TypeKind.Void: + is_void = 1 + elif ti.Size == 0 and ti.IsSigned == -1 and ti.PtrCount == 0: + is_void = 1 + + # 写入 base 类型字符串到 buf + base_len: int = 0 + if is_void == 1: + if ti.PtrCount == 0: + # "void" + if buf_size < 5: + return -1 + string.strcpy(buf, "void") + base_len = 4 + else: + # void* → i8*（LLVM 中 void* 表示为 i8*） + if buf_size < 4: + return -1 + string.strcpy(buf, "i8") + base_len = 2 + elif ti.IsSigned == -1: + # 浮点 + if ti.Size == 16: + if buf_size < 5: + return -1 + string.strcpy(buf, "half") + base_len = 4 + elif ti.Size == 32: + if buf_size < 6: + return -1 + string.strcpy(buf, "float") + base_len = 5 + elif ti.Size == 64: + if buf_size < 7: + return -1 + string.strcpy(buf, "double") + base_len = 6 + elif ti.Size == 128: + if buf_size < 6: + return -1 + string.strcpy(buf, "fp128") + base_len = 5 + else: + # 未知浮点尺寸 → double 兜底 + if buf_size < 7: + return -1 + string.strcpy(buf, "double") + base_len = 6 + else: + # 整数: i{Size}，用 snprintf 安全写入 + if ti.Size <= 0: + # 无效尺寸 → i8 兜底 + if buf_size < 4: + return -1 + string.strcpy(buf, "i8") + base_len = 2 + else: + viperlib.snprintf(buf, buf_size, "i%d", ti.Size) + base_len = string.strlen(buf) + + # 追加 PtrCount 个 "*" + if base_len + ti.PtrCount + 1 > buf_size: + return -1 + pos: int = base_len + for i in range(ti.PtrCount): + buf[pos] = '*' + pos += 1 + buf[pos] = 0 + return pos + + +# ============================================================ +# Mixin - 所有 Handle 的非多态基类（对应 TransPyC 的 BaseHandle） +# +# @t.NoVTable 继承：字段展平嵌入子类，无 vtable 开销（对应 C++ 非多态继承）。 +# 子类继承 Trans 字段 + 共享工具方法，编译器自动生成子类方法包装 +# （self bitcast 为父类指针后调用），子类可直接调用继承的方法。 +# +# 用法： +# @t.NoVTable +# class AssignHandle(Mixin): +# _CurrentClass: str +# def __init__(self, trans): +# self.InitMixin(trans) +# self._CurrentClass = None +# def Handle(self, node) -> int: +# rhs = self.Trans.ExprH.HandleValue(node.value) +# ... +# ============================================================ +@t.NoVTable +class Mixin: + """所有 Handle 的非多态基类：持有 Translator 回指针 + 共享委托方法""" + Trans: HT.Translator | t.CPtr + + def InitMixin(self, trans: HT.Translator | t.CPtr) -> int: + """初始化 Mixin 字段（子类 __init__ 中调用）""" + self.Trans = trans + return 0 + + +# 全局 mbuddy 指针 +_mbuddy: t.CVoid | t.CPtr diff --git a/TransPyV/App/lib/core/Handles/HandlesBody.py b/TransPyV/App/lib/core/Handles/HandlesBody.py new file mode 100644 index 0000000..714cadb --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesBody.py @@ -0,0 +1,230 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesClassDef as HandlesClassDef + + +# ============================================================ +# HandlesBody - 语句分派（trans 单参模式，全方法调用） +# +# 所有语句类型通过 trans.XxxH.Handle(node) 分派到对应 Handle。 +# 对应 TransPyC BodyHandle.HandleBodyLlvm 的 isinstance 分派。 +# ============================================================ + + +# ============================================================ +# 语句翻译分派 +# ============================================================ +def translate_stmt(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译单条语句，返回新增的变量数""" + if node is None: + return 0 + k: int = node.kind() + + if k == ast.ASTKind.Expr: + return translate_expr_stmt(trans, node) + elif k == ast.ASTKind.Assign: + return trans.AssignH.Handle(node) + elif k == ast.ASTKind.AnnAssign: + return trans.AnnAssignH.Handle(node) + elif k == ast.ASTKind.FunctionDef: + # 嵌套函数定义：提升为顶层函数 + 创建闭包 + return HandlesFunctions.translate_nested_function_def(trans, node) + elif k == ast.ASTKind.Return: + return trans.ReturnH.Handle(node) + elif k == ast.ASTKind.If: + return trans.IfH.Handle(node) + elif k == ast.ASTKind.While: + return trans.WhileH.Handle(node) + elif k == ast.ASTKind.AugAssign: + return trans.AugAssignH.Handle(node) + elif k == ast.ASTKind.For: + return trans.ForH.Handle(node) + elif k == ast.ASTKind.ClassDef: + return HandlesClassDef.translate_class_def(trans, node) + elif k == ast.ASTKind.Import: + return trans.ImportsH.HandleImport(node) + elif k == ast.ASTKind.ImportFrom: + trans.ImportsH.HandleImportFromModule(node) + return trans.ImportsH.HandleImportFromNames(node) + elif k == ast.ASTKind.Global: + return translate_global(trans, node) + elif k == ast.ASTKind.Nonlocal: + return translate_nonlocal(trans, node) + elif k == ast.ASTKind.Pass: + return 0 + elif k == ast.ASTKind.Break: + return translate_break(trans) + elif k == ast.ASTKind.Continue: + return translate_continue(trans) + return 0 + + +# ============================================================ +# 翻译 Global 语句: global x, y +# +# 将 names 中的变量名加入 _global_names 集合 +# ============================================================ +def translate_global(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译 global 语句：记录 global 变量名""" + gn: ast.Global | t.CPtr = (ast.Global | t.CPtr)(node) + if gn is None: + return 0 + names: list[ast.AST | t.CPtr] | t.CPtr = gn.names + if names is None: + return 0 + n: t.CSizeT = names.__len__() + for i in range(n): + nm_node: ast.AST | t.CPtr = names.get(i) + if nm_node is not None and nm_node.kind() == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(nm_node) + if nm.id is not None: + HT.add_global_name(trans, nm.id) + return 0 + + +# ============================================================ +# 翻译 Nonlocal 语句: nonlocal x, y +# +# 将 names 中的变量名加入 _nonlocal_names 集合 +# ============================================================ +def translate_nonlocal(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译 nonlocal 语句：记录 nonlocal 变量名""" + nl: ast.Nonlocal | t.CPtr = (ast.Nonlocal | t.CPtr)(node) + if nl is None: + return 0 + names: list[ast.AST | t.CPtr] | t.CPtr = nl.names + if names is None: + return 0 + n: t.CSizeT = names.__len__() + for i in range(n): + nm_node: ast.AST | t.CPtr = names.get(i) + if nm_node is not None and nm_node.kind() == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(nm_node) + if nm.id is not None: + HT.add_nonlocal_name(trans, nm.id) + return 0 + + +# ============================================================ +# 翻译表达式语句 Expr(value=Call(...)) +# ============================================================ +def translate_expr_stmt(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译表达式语句（printf 调用等）""" + ex: ast.Expr | t.CPtr = (ast.Expr | t.CPtr)(node) + if ex is None: + return 0 + call_node: ast.AST | t.CPtr = ex.value + if call_node is None: + return 0 + ck: int = call_node.kind() + if ck != ast.ASTKind.Call: + return 0 + + cl: ast.Call | t.CPtr = (ast.Call | t.CPtr)(call_node) + func_node: ast.AST | t.CPtr = cl.func + if func_node is None: + return 0 + + func_name: str = HandlesExpr.get_func_name(func_node) + if func_name is None: + return 0 + + # printf / print 走特殊路径（print 映射到 printf） + if string.strcmp(func_name, "printf") == 0 or string.strcmp(func_name, "print") == 0: + trans.ExprCallH.HandlePrintfCall(cl) + else: + # 通用函数调用（返回值丢弃） + trans.ExprCallH.HandleCall(call_node) + return 0 + + +# ============================================================ +# 预扫描：为局部变量提前创建 alloca +# ============================================================ +def pre_scan_allocas(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """预扫描语句中的 AnnAssign，提前创建 alloca + + 返回新增的变量数 + """ + if node is None: + return 0 + return trans.AnnAssignH.PreScan(node) + + +# ============================================================ +# 翻译 break 语句 +# ============================================================ +def translate_break(trans: HT.Translator | t.CPtr) -> int: + """翻译 break 语句：跳转到循环 end 块""" + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + pool: memhub.MemBuddy | t.CPtr = trans.Pool + + if builder is None or func is None: + return 0 + + # 发射 br 到 break 目标 + if trans._break_bb is not None: + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, trans._break_bb) + + # 创建死代码 BB，用于后续语句（break 后面的代码不可达） + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "dead.%d", cnt) + dead_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + llvmlite.position_at_end(builder, dead_bb) + return 0 + + +# ============================================================ +# 翻译 continue 语句 +# ============================================================ +def translate_continue(trans: HT.Translator | t.CPtr) -> int: + """翻译 continue 语句：跳转到循环 cond/incr 块""" + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + pool: memhub.MemBuddy | t.CPtr = trans.Pool + + if builder is None or func is None: + return 0 + + # 发射 br 到 continue 目标 + if trans._continue_bb is not None: + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, trans._continue_bb) + + # 创建死代码 BB，用于后续语句 + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "dead.%d", cnt) + dead_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + llvmlite.position_at_end(builder, dead_bb) + return 0 + + +# ============================================================ +# 获取语句类型名（调试用） +# ============================================================ +def get_stmt_kind_name(node: ast.AST | t.CPtr) -> str: + """获取语句类型名""" + if node is None: + return None + return node.type_name() diff --git a/TransPyV/App/lib/core/Handles/HandlesClassDef.py b/TransPyV/App/lib/core/Handles/HandlesClassDef.py new file mode 100644 index 0000000..d7c665e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesClassDef.py @@ -0,0 +1,2327 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + + +# 枚举成员临时数组最大数量（本地常量） +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + + +# ============================================================ +# HandlesClassDef - class 定义处理 +# +# 解析 class 定义，创建 LLVM StructType 并注册字段信息 +# +# 纯内存结构体（无方法）的 class 定义: +# class Point: +# x: t.CInt +# y: t.CInt +# +# 生成 LLVM 类型: { i32, i32 } +# 注册字段: x→index 0, y→index 1 +# ============================================================ + + +# ============================================================ +# _is_cenum_base — 检查 base 节点是否为 t.CEnum +# +# 支持 Name(id='CEnum') 和 Attribute(value=Name('t'), attr='CEnum') +# ============================================================ +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: + """检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否""" + if base_node is None: + return 0 + k: int = base_node.kind() + # Name('CEnum') 形式（from t import CEnum） + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(base_node) + if nm.id is not None and string.strcmp(nm.id, "CEnum") == 0: + return 1 + return 0 + # Attribute(t.CEnum) 形式 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(base_node) + if at.attr is not None and string.strcmp(at.attr, "CEnum") == 0: + return 1 + return 0 + return 0 + + +# ============================================================ +# _is_enum_class — 检查 ClassDef 是否继承自 t.CEnum +# ============================================================ +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: + """检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否""" + if cd is None or cd.bases is None: + return 0 + bases: list[ast.AST | t.CPtr] | t.CPtr = cd.bases + bn: t.CSizeT = bases.__len__() + if bn == 0: + return 0 + for bi in range(bn): + base_node: ast.AST | t.CPtr = bases.get(bi) + if base_node is None: + continue + if _is_cenum_base(base_node) == 1: + return 1 + return 0 + + +# ============================================================ +# _is_cunion_base — 检查 base 节点是否为 t.CUnion +# +# 支持 Name(id='CUnion') 和 Attribute(value=Name('t'), attr='CUnion') +# ============================================================ +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: + """检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否""" + if base_node is None: + return 0 + k: int = base_node.kind() + # Name('CUnion') 形式（from t import CUnion） + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(base_node) + if nm.id is not None and string.strcmp(nm.id, "CUnion") == 0: + return 1 + return 0 + # Attribute(t.CUnion) 形式 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(base_node) + if at.attr is not None and string.strcmp(at.attr, "CUnion") == 0: + return 1 + return 0 + return 0 + + +# ============================================================ +# _is_union_class — 检查 ClassDef 是否继承自 t.CUnion +# ============================================================ +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: + """检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否""" + if cd is None or cd.bases is None: + return 0 + bases: list[ast.AST | t.CPtr] | t.CPtr = cd.bases + bn: t.CSizeT = bases.__len__() + if bn == 0: + return 0 + for bi in range(bn): + base_node: ast.AST | t.CPtr = bases.get(bi) + if base_node is None: + continue + if _is_cunion_base(base_node) == 1: + return 1 + return 0 + + +# ============================================================ +# VTable 装饰器检测和继承关系判断 +# ============================================================ + +# 标记基类集合（非真实父类，仅作为类型标记） +_MARKER_BASES: t.CDefine = 8 +# 使用字符串比较代替集合，标记基类列表 +# Object, CVTable, Exception, CEnum, Enum, CStruct, CUnion, REnum + + +# ============================================================ +# _has_decorator — 检查 ClassDef/FunctionDef 的 decorator_list 中是否有指定装饰器 +# +# 支持 @t.Name 和 @Name 两种形式 +# 返回 1=有 / 0=无 +# ============================================================ +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, + deco_name: str) -> int: + """检查 decorator_list 中是否有指定装饰器名""" + if decorator_list is None: + return 0 + dn: t.CSizeT = decorator_list.__len__() + if dn == 0: + return 0 + for di in range(dn): + deco: ast.AST | t.CPtr = decorator_list.get(di) + if deco is None: + continue + k: int = deco.kind() + # @t.NoVTable 形式（Attribute） + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(deco) + if at.attr is not None and string.strcmp(at.attr, deco_name) == 0: + # 检查 value 是 Name('t') + if at.value is not None and at.value.kind() == ast.ASTKind.Name: + vn: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if vn.id is not None and string.strcmp(vn.id, "t") == 0: + return 1 + return 0 + # @NoVTable 形式（Name，from t import NoVTable） + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(deco) + if nm.id is not None and string.strcmp(nm.id, deco_name) == 0: + return 1 + return 0 + + +# ============================================================ +# _is_marker_base — 检查基类名是否为标记基类（非真实父类） +# +# 标记基类: Object, CVTable, Exception, CEnum, Enum, CStruct, CUnion, REnum +# ============================================================ +def _is_marker_base(base_name: str) -> int: + """检查基类名是否为标记基类，返回 1=是 / 0=否""" + if base_name is None: + return 0 + if string.strcmp(base_name, "Object") == 0: + return 1 + if string.strcmp(base_name, "CVTable") == 0: + return 1 + if string.strcmp(base_name, "Exception") == 0: + return 1 + if string.strcmp(base_name, "CEnum") == 0: + return 1 + if string.strcmp(base_name, "Enum") == 0: + return 1 + if string.strcmp(base_name, "CStruct") == 0: + return 1 + if string.strcmp(base_name, "CUnion") == 0: + return 1 + if string.strcmp(base_name, "REnum") == 0: + return 1 + return 0 + + +# ============================================================ +# _get_base_name — 从 base AST 节点提取基类名 +# +# 支持 Name(id), Attribute(attr), Subscript +# ============================================================ +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: + """从 base AST 节点提取基类名，返回 None=失败""" + if base_node is None: + return None + k: int = base_node.kind() + # Name(id) 形式 + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(base_node) + return nm.id + # Attribute(attr) 形式 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(base_node) + return at.attr + return None + + +# ============================================================ +# _get_parent_class — 获取 ClassDef 的真实父类名（非标记基类） +# +# 遍历 bases，返回第一个非标记基类的名字，None=无真实父类 +# ============================================================ +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: + """获取 ClassDef 的真实父类名，返回 None=无真实父类""" + if cd is None or cd.bases is None: + return None + bases: list[ast.AST | t.CPtr] | t.CPtr = cd.bases + bn: t.CSizeT = bases.__len__() + if bn == 0: + return None + for bi in range(bn): + base_node: ast.AST | t.CPtr = bases.get(bi) + if base_node is None: + continue + bname: str = _get_base_name(base_node) + if bname is not None and _is_marker_base(bname) == 0: + return bname + return None + + +# ============================================================ +# _has_any_func_cvtable — 检查类中是否有任何函数标记了 @t.CVTable +# +# 用于类级别未启用虚表时，检查函数级 @t.CVTable 是否触发虚表。 +# 跳过 __init__ 和 __before_init__（构造函数不进入虚表）。 +# ============================================================ +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: + """检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无""" + if cd is None or cd.children is None: + return 0 + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + cn: t.CSizeT = children.__len__() + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + if stmt.kind() != ast.ASTKind.FunctionDef: + continue + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(stmt) + if fd is None or fd.name is None: + continue + # 跳过构造函数 + if string.strcmp(fd.name, "__init__") == 0: + continue + if string.strcmp(fd.name, "__before_init__") == 0: + continue + if string.strcmp(fd.name, "__new__") == 0: + continue + if fd.decorator_list is not None: + if _has_decorator(fd.decorator_list, "CVTable") == 1: + return 1 + return 0 + + +# ============================================================ +# _should_method_be_virtual — 判断方法是否应该进入虚表 +# +# 规则（优先级从高到低）: +# 1. 函数有 @t.NoVTable → 排除（返回 0） +# 2. 函数有 @t.CVTable → 包含（返回 1） +# 3. 类有 @t.CVTable → 包含（返回 1） +# 4. 有继承关系（父类非 NoVTable）→ 包含（返回 1） +# 5. 默认 → 排除（返回 0） +# ============================================================ +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, + fd: ast.FunctionDef | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> int: + """判断方法是否应该进入虚表，返回 1=包含 / 0=排除""" + if fd is None: + return 0 + # 1. 检查函数级 @t.NoVTable（最高优先级，排除） + if fd.decorator_list is not None: + if _has_decorator(fd.decorator_list, "NoVTable") == 1: + return 0 + # 2. 检查函数级 @t.CVTable（明确包含） + if _has_decorator(fd.decorator_list, "CVTable") == 1: + return 1 + + # 3. 检查类级别 @t.CVTable + if cd is not None and cd.decorator_list is not None: + if _has_decorator(cd.decorator_list, "CVTable") == 1: + return 1 + + # 4. 检查继承关系 + parent_name: str = _get_parent_class(cd) + if parent_name is not None: + # 用 SHA1 感知查找父类，规避跨模块同名找错 + is_nvt: int = 0 + if trans is not None: + p_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + parent_name, trans.ModuleSha1) + if p_entry is not None: + is_nvt = p_entry.IsNoVTable + else: + is_nvt = HandlesStruct.is_novtable_by_name(parent_name) + if is_nvt != 1: + return 1 # 有继承且父类非 NoVTable + + # 5. 默认排除 + return 0 + + +# ============================================================ +# _detect_vtable_status — 检测类是否应该启用虚表 +# +# 规则: +# 1. @t.CVTable → 有虚表 +# 2. 有真实父类且非 NoVTable → 有虚表（自动启用） +# 3. @t.NoVTable → 检查函数级 @t.CVTable，有则启用 +# 4. 默认 → 检查函数级 @t.CVTable，有则启用 +# +# 返回: 1=有虚表 / 0=无虚表 +# ============================================================ +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> int: + """检测类是否应该启用虚表""" + if cd is None: + return 0 + + # 检查类装饰器 + if cd.decorator_list is not None: + # @t.CVTable 优先级最高，直接启用 + if _has_decorator(cd.decorator_list, "CVTable") == 1: + return 1 + # @t.NoVTable：类级别禁用，但检查函数级 @t.CVTable + if _has_decorator(cd.decorator_list, "NoVTable") == 1: + return _has_any_func_cvtable(cd) + + # 检查是否有真实父类（继承自动启用 CVTable） + parent_name: str = _get_parent_class(cd) + if parent_name is not None: + # 用 SHA1 感知查找父类，规避跨模块同名找错 + is_nvt: int = 0 + if trans is not None: + p_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + parent_name, trans.ModuleSha1) + if p_entry is not None: + is_nvt = p_entry.IsNoVTable + else: + is_nvt = HandlesStruct.is_novtable_by_name(parent_name) + if is_nvt == 1: + # 父类是 NoVTable，检查函数级 @t.CVTable + return _has_any_func_cvtable(cd) + return 1 # 有父类且非 NoVTable，自动启用虚表 + + # 默认（无装饰器无继承），检查函数级 @t.CVTable + return _has_any_func_cvtable(cd) + + +# ============================================================ +# _resolve_enum_member_type — 解析枚举成员的类型 +# +# - t.State → 默认 i32 +# - int / t.CInt8T / t.CInt16T / t.CInt32T / t.CInt64T 等 → 对应整数类型 +# - 联合注解（t.State | t.CInt8T）递归查找实际类型 +# ============================================================ +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, + ann_node: ast.AST | t.CPtr, + imported_modules: str, + from_imports: str) -> llvmlite.LLVMType | t.CPtr: + """解析枚举成员的类型注解为 LLVM 整数类型""" + if ann_node is None: + return llvmlite.Int32(pool) + + k: int = ann_node.kind() + + # Attribute 节点: t.State / int / t.CInt8T 等 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(ann_node) + if at.attr is None: + return llvmlite.Int32(pool) + # t.State → 默认 i32 + if string.strcmp(at.attr, "State") == 0: + return llvmlite.Int32(pool) + # 其他 t.CXxx 类型 + ty: llvmlite.LLVMType | t.CPtr = HandlesType.map_t_type(pool, at.attr) + if ty is not None: + return ty + return llvmlite.Int32(pool) + + # Name 节点: from-import 形式（State / CInt） + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(ann_node) + if nm.id is None: + return llvmlite.Int32(pool) + if string.strcmp(nm.id, "State") == 0: + return llvmlite.Int32(pool) + ty2: llvmlite.LLVMType | t.CPtr = HandlesType.map_t_type(pool, nm.id) + if ty2 is not None: + return ty2 + return llvmlite.Int32(pool) + + # BinOp 节点: t.State | t.CInt8T（递归查找实际类型） + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(ann_node) + if bop.op == ast.OpKind.BitOr: + left_ty: llvmlite.LLVMType | t.CPtr = _resolve_enum_member_type( + pool, bop.left, imported_modules, from_imports) + # 跳过 t.State（i32 默认），优先返回具体类型 + left_bits: int = HandlesType.get_llvm_type_bits(left_ty) + if left_bits != 0 and not _is_state_annotation(bop.left): + return left_ty + right_ty: llvmlite.LLVMType | t.CPtr = _resolve_enum_member_type( + pool, bop.right, imported_modules, from_imports) + if right_ty is not None and not _is_state_annotation(bop.right): + return right_ty + # 两边都是 State 或无法确定，返回较宽的 + right_bits: int = HandlesType.get_llvm_type_bits(right_ty) + if left_bits >= right_bits: + return left_ty + return right_ty + return llvmlite.Int32(pool) + + return llvmlite.Int32(pool) + + +# ============================================================ +# _is_state_annotation — 检查注解节点是否为 t.State +# ============================================================ +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: + """检查注解节点是否为 t.State，返回 1=是 / 0=否""" + if node is None: + return 0 + k: int = node.kind() + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is not None and string.strcmp(at.attr, "State") == 0: + return 1 + return 0 + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm.id is not None and string.strcmp(nm.id, "State") == 0: + return 1 + return 0 + return 0 + + +# ============================================================ +# _extract_int_value — 从 AST 节点提取整数值 +# +# 支持 Constant(int) 和 UnaryOp(USub, Constant(int)) +# ============================================================ +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: + """从 AST 节点提取整数值，失败返回 0""" + if node is None: + return 0 + k: int = node.kind() + if k == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn.const_kind == ast.CONST_INT: + return cn.int_val + return 0 + if k == ast.ASTKind.UnaryOp: + uo: ast.UnaryOp | t.CPtr = (ast.UnaryOp | t.CPtr)(node) + if uo.op == ast.OpKind.USub: + inner: t.CInt64T = _extract_int_value(uo.operand) + return -inner + return 0 + return 0 + + +# ============================================================ +# _translate_enum_def — 翻译枚举类定义 +# +# 遍历 class body 中的 AnnAssign，收集成员名、类型、值 +# 自动赋值：无值时从上一值+1递增（首值默认 0） +# 基准类型：所有成员类型中 Bits 最大的（t.State 视为 i32） +# ============================================================ +def _translate_enum_def(trans: HT.Translator | t.CPtr, + cd: ast.ClassDef | t.CPtr) -> int: + """翻译枚举类定义，返回 0""" + if cd is None or cd.name is None or trans is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + class_name: str = cd.name + + # 已注册则跳过 + if HandlesEnum.is_enum_class(class_name) == 1: + return 0 + + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + if children is None: + stdio.printf("[ENUM] %s has no body\n", class_name) + return 0 + + cn: t.CSizeT = children.__len__() + + # 临时数组存储成员信息 + member_names_buf: t.CSizeT | t.CPtr = pool.alloc(8 * ENUM_MEMBER_MAX_LOCAL) + member_types_buf: t.CSizeT | t.CPtr = pool.alloc(8 * ENUM_MEMBER_MAX_LOCAL) + member_values_buf: t.CInt64T | t.CPtr = pool.alloc(8 * ENUM_MEMBER_MAX_LOCAL) + if member_names_buf is None or member_types_buf is None or member_values_buf is None: + return 0 + string.memset(member_names_buf, 0, 8 * ENUM_MEMBER_MAX_LOCAL) + string.memset(member_types_buf, 0, 8 * ENUM_MEMBER_MAX_LOCAL) + string.memset(member_values_buf, 0, 8 * ENUM_MEMBER_MAX_LOCAL) + + member_count: int = 0 + next_val: t.CInt64T = 0 + + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + sk: int = stmt.kind() + + if sk == ast.ASTKind.AnnAssign: + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(stmt) + if aa is None or aa.target is None: + continue + + # 获取成员名 + if aa.target.kind() != ast.ASTKind.Name: + continue + tgt: ast.Name | t.CPtr = (ast.Name | t.CPtr)(aa.target) + if tgt.id is None: + continue + + # 解析成员类型 + member_ty: llvmlite.LLVMType | t.CPtr = _resolve_enum_member_type( + pool, aa.annotation, trans._imported_modules, trans._from_imports) + + # 解析成员值 + has_value: int = 0 + cur_val: t.CInt64T = 0 + if aa.value is not None: + cur_val = _extract_int_value(aa.value) + has_value = 1 + else: + cur_val = next_val + + # 下一个自动值 + next_val = cur_val + 1 + + # 存储 + if member_count < ENUM_MEMBER_MAX_LOCAL: + member_names_buf[member_count] = t.CSizeT(tgt.id) + member_types_buf[member_count] = t.CSizeT(member_ty) + member_values_buf[member_count] = cur_val + member_count += 1 + + if member_count == 0: + stdio.printf("[ENUM] %s has no members\n", class_name) + return 0 + + # 确定基准类型：所有成员类型中 Bits 最大的 + base_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + max_bits: int = 0 + for mi in range(member_count): + ty_addr: t.CSizeT = member_types_buf[mi] + if ty_addr == 0: + continue + mty: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)( + t.CVoid(ty_addr, t.CPtr)) + bits: int = HandlesType.get_llvm_type_bits(mty) + if bits > max_bits: + max_bits = bits + base_ty = mty + + # 注册枚举 + entry: HandlesEnum.EnumEntry | t.CPtr = HandlesEnum.register_enum( + pool, class_name, base_ty) + if entry is None: + stdio.printf("[ENUM] failed to register %s\n", class_name) + return 0 + + # 添加成员 + for mi in range(member_count): + fname_addr: t.CSizeT = member_names_buf[mi] + if fname_addr == 0: + continue + mname: str = (str | t.CPtr)(t.CVoid(fname_addr, t.CPtr)) + fty_addr2: t.CSizeT = member_types_buf[mi] + if fty_addr2 == 0: + continue + mty2: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)( + t.CVoid(fty_addr2, t.CPtr)) + mval: t.CInt64T = member_values_buf[mi] + if mname is not None and mty2 is not None: + HandlesEnum.add_enum_member(pool, entry, mname, mval, mty2) + + stdio.printf("[ENUM] registered %s with %d members, base_bits=%d\n", + class_name, member_count, max_bits) + return 0 + + +# ============================================================ +# _get_type_size — 计算 LLVM 类型的字节大小 +# +# 用于联合体确定最大字段大小 +# ============================================================ +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: + """计算 LLVM 类型的字节大小""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Int(bits): + return (bits + 7) // 8 + case llvmlite.LLVMType.Float(bits): + return bits // 8 + case llvmlite.LLVMType.Ptr(pointee): + return 8 + case llvmlite.LLVMType.Array(elem_ty, count): + return _get_type_size(elem_ty) * count + case llvmlite.LLVMType.Struct(fields, fcount, name): + total: int = 0 + cur: llvmlite.ParamNode | t.CPtr = fields + i: int = 0 + while cur is not None and i < fcount: + if cur.Ty is not None: + fty: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)(cur.Ty) + total += _get_type_size(fty) + cur = cur.Next + i += 1 + return total + case _: + return 8 + + +# ============================================================ +# _translate_union_def — 翻译联合体定义 +# +# 联合体语法: +# class MyUnion(t.CUnion): +# a: t.CInt +# b: t.CFloat +# c: t.CInt64T +# +# 实现方式: +# - 收集所有字段类型 +# - 计算最大字段字节大小 max_size +# - LLVM 类型 = Struct([Array(Int8, max_size)]) # { [N x i8] } +# - 注册到 HandlesStruct 并标记 IsUnion=1 +# - 字段访问通过 bitcast 实现（HandlesExpr 中处理） +# ============================================================ +def _translate_union_def(trans: HT.Translator | t.CPtr, + cd: ast.ClassDef | t.CPtr) -> int: + """翻译联合体定义，返回 0""" + if cd is None or cd.name is None or trans is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + class_name: str = cd.name + + # 命名空间隔离：标记为当前文件可见（必须在 existing 检查之前） + HandlesStruct.add_visible_struct(pool, class_name) + + # 检查是否已注册（用 SHA1 区分跨模块同名类） + existing: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + class_name, trans.ModuleSha1) + if existing is not None: + # Phase B: 将已存在的命名结构体注册到当前模块 + if existing.Ty is not None: + llvmlite.module_add_named_type(trans.Module, pool, existing.Ty) + # SHA1 已在 register_struct 时设置，无需再补 + return 0 + + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + if children is None: + stdio.printf("[UNION] %s has no body\n", class_name) + return 0 + + cn: t.CSizeT = children.__len__() + + # 临时数组存储字段信息 + field_names_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + field_types_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if field_names_buf is None or field_types_buf is None: + return 0 + string.memset(field_names_buf, 0, 8 * 32) + string.memset(field_types_buf, 0, 8 * 32) + + field_count: int = 0 + max_size: int = 0 + + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + sk: int = stmt.kind() + + if sk == ast.ASTKind.AnnAssign: + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(stmt) + if aa is None or aa.target is None: + continue + + # 获取字段名 + if aa.target.kind() != ast.ASTKind.Name: + continue + tgt: ast.Name | t.CPtr = (ast.Name | t.CPtr)(aa.target) + if tgt.id is None: + continue + + # 解析字段类型 + field_ty: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, aa.annotation, trans._imported_modules, trans._from_imports) + if field_ty is None: + field_ty = llvmlite.Int32(pool) + + # 计算字段大小，更新 max_size + fsize: int = _get_type_size(field_ty) + if fsize > max_size: + max_size = fsize + + # 存储到数组 + if field_count < 32: + field_names_buf[field_count] = t.CSizeT(tgt.id) + field_types_buf[field_count] = t.CSizeT(field_ty) + field_count += 1 + + if field_count == 0: + stdio.printf("[UNION] %s has no fields\n", class_name) + return 0 + + if max_size < 1: + max_size = 1 + + # 创建联合体的 LLVM 类型: Struct([Array(Int8, max_size)]) + # 即 { [max_size x i8] } + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + array_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Array(pool, i8_ty, max_size) + + # 构建 ParamNode 链表（单字段：字节数组） + pnode: llvmlite.ParamNode | t.CPtr = pool.alloc(llvmlite.ParamNode.__sizeof__()) + if pnode is None: + return 0 + string.memset(pnode, 0, llvmlite.ParamNode.__sizeof__()) + pnode.Ty = array_ty + + # 构造命名结构体类型名: "sha1.ClassName" 或 "ClassName" + union_type_name: str = class_name + if trans.ModuleSha1 is not None: + union_name_buf: t.CChar | t.CPtr = pool.alloc(64) + if union_name_buf is not None: + viperlib.snprintf(union_name_buf, 64, "%s.%s", trans.ModuleSha1, class_name) + union_type_name = union_name_buf + + union_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Struct(pool, pnode, 1, union_type_name) + if union_ty is None: + stdio.printf("[UNION] failed to create type for %s\n", class_name) + return 0 + + # 注册命名结构体到模块 + llvmlite.module_add_named_type(trans.Module, pool, union_ty) + + # 注册到 HandlesStruct（传递 SHA1 在注册时直接设置，避免跨模块同名找错 entry） + entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.register_struct( + pool, class_name, union_ty, trans.ModuleSha1) + if entry is None: + stdio.printf("[UNION] failed to register %s\n", class_name) + return 0 + + # 标记为当前文件可见（命名空间隔离） + HandlesStruct.add_visible_struct(pool, class_name) + + # 标记为联合体（直接用 entry，规避跨模块同名 find_struct 找错） + entry.IsUnion = 1 + + # 添加字段信息（所有字段 Index=0，因为共享偏移 0） + for fi in range(field_count): + fname_addr: t.CSizeT = field_names_buf[fi] + if fname_addr == 0: + continue + fname: str = (str | t.CPtr)(t.CVoid(fname_addr, t.CPtr)) + fty_addr2: t.CSizeT = field_types_buf[fi] + if fty_addr2 == 0: + continue + fty2: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)( + t.CVoid(fty_addr2, t.CPtr)) + if fname is not None and fty2 is not None: + HandlesStruct.add_field(pool, entry, fname, fty2, None, None) + + stdio.printf("[UNION] registered %s with %d fields, max_size=%d\n", + class_name, field_count, max_size) + return 0 + + +# ============================================================ +# translate_class_def — 翻译 class 定义 +# +# 遍历 class body 中的 AnnAssign 语句，收集字段类型 +# 创建 LLVM StructType 并注册到 HandlesStruct +# ============================================================ +# ============================================================ +# 泛型类模板存储（模块级） +# +# 泛型类 class list[T]: 不直接发射 IR，而是存储为模板， +# 等遇到 list[int](pool) 实例化时触发 _specialize_generic_class +# ============================================================ + +_generic_class_names: list[str] | t.CPtr = None +_generic_class_nodes: list[ast.ClassDef | t.CPtr] | t.CPtr = None + + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: + """检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否""" + if cd is None: + return 0 + tp: list[str] | t.CPtr = cd.type_params + if tp is None: + return 0 + if tp.__len__() == 0: + return 0 + return 1 + + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, + cd: ast.ClassDef | t.CPtr): + """注册泛型类模板（类名 + ClassDef 节点）""" + global _generic_class_names, _generic_class_nodes + if _generic_class_names is None: + _generic_class_names = list[str](pool, 8) + _generic_class_nodes = list[ast.ClassDef | t.CPtr](pool, 8) + # 检查是否已注册 + cn: t.CSizeT = _generic_class_names.__len__() + i: t.CSizeT + for i in range(cn): + nm: str = _generic_class_names.get(i) + if nm is not None and string.strcmp(nm, cd.name) == 0: + return + _generic_class_names.append(cd.name) + _generic_class_nodes.append(cd) + + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: + """查找泛型类模板，返回 ClassDef 节点或 None""" + if _generic_class_names is None or class_name is None: + return None + cn: t.CSizeT = _generic_class_names.__len__() + i: t.CSizeT + for i in range(cn): + nm: str = _generic_class_names.get(i) + if nm is not None and string.strcmp(nm, class_name) == 0: + return _generic_class_nodes.get(i) + return None + + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, + class_name: str, + type_args: list[str] | t.CPtr) -> str: + """生成特化类名: list + [int] -> list[int] + + 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出 + """ + if class_name is None or type_args is None or pool is None: + return class_name + n: t.CSizeT = type_args.__len__() + if n == 0: + return class_name + # 计算总长度: class_name + "[" + arg0 + "," + arg1 + ... + "]" + NUL + total_len: t.CSizeT = string.strlen(class_name) + 2 # "[" 和 "]" + i: t.CSizeT + for i in range(n): + if i > 0: + total_len += 1 # "," + ta: str = type_args.get(i) + if ta is not None: + total_len += string.strlen(ta) + total_len += 1 # NUL + # 分配新缓冲区 + mangled: str = pool.alloc(total_len) + if mangled is None: + return class_name + mangled[0] = '\0' + string.strcat(mangled, class_name) + string.strcat(mangled, "[") + for i in range(n): + if i > 0: + string.strcat(mangled, ",") + ta: str = type_args.get(i) + if ta is not None: + string.strcat(mangled, ta) + string.strcat(mangled, "]") + return mangled + + +# ============================================================ +# 类型注解拷贝（用于泛型特化） +# +# _clone_annotation: 拷贝类型注解表达式，替换类型参数 T 为具体类型 +# _clone_arguments: 拷贝 Arguments 节点，替换 arg annotation 中的 T +# _clone_classdef_for_spec: 拷贝 ClassDef 用于泛型特化 +# ============================================================ + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr, + tp_names: list[str] | t.CPtr, + type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: + """拷贝类型注解表达式，替换类型参数 T 为具体类型名 + + tp_names: 类型参数名列表 ['T'] + type_args: 类型实参名列表 ['int'] + """ + if node is None: + return None + k: int = node.kind() + + # Name: 可能是类型参数 T + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm.id is not None: + n: t.CSizeT = tp_names.__len__() if tp_names is not None else 0 + i: t.CSizeT + for i in range(n): + tpn: str = tp_names.get(i) + if tpn is not None and string.strcmp(nm.id, tpn) == 0: + ta: str = type_args.get(i) + # 类型实参可能是复合类型（如 "AST|t.CPtr"），用 _parse_type_string 解析 + return HandlesType._parse_type_string(pool, ta, nm.ctx) + return ast.Name(pool, nm.id, nm.ctx) + return None + + # Attribute: t.CPtr 等 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + new_value: ast.AST | t.CPtr = _clone_annotation(pool, at.value, tp_names, type_args) + return ast.Attribute(pool, new_value, at.attr, at.ctx) + + # BinOp: T | t.CPtr 等 + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + new_left: ast.AST | t.CPtr = _clone_annotation(pool, bop.left, tp_names, type_args) + new_right: ast.AST | t.CPtr = _clone_annotation(pool, bop.right, tp_names, type_args) + return ast.BinOp(pool, new_left, bop.op, new_right) + + # Subscript: list[int] 等 + if k == ast.ASTKind.Subscript: + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + new_value2: ast.AST | t.CPtr = _clone_annotation(pool, sub.value, tp_names, type_args) + new_slice: ast.AST | t.CPtr = _clone_annotation(pool, sub.slice, tp_names, type_args) + return ast.Subscript(pool, new_value2, new_slice, sub.ctx) + + # Tuple: (T, int) 等 + if k == ast.ASTKind.Tuple: + tup: ast.Tuple | t.CPtr = (ast.Tuple | t.CPtr)(node) + new_elts: list[ast.AST | t.CPtr] | t.CPtr = list[ast.AST | t.CPtr](pool, 8) + if tup.elts is not None: + en: t.CSizeT = tup.elts.__len__() + ei: t.CSizeT + for ei in range(en): + el: ast.AST | t.CPtr = tup.elts.get(ei) + new_el: ast.AST | t.CPtr = _clone_annotation(pool, el, tp_names, type_args) + if new_el is not None: + new_elts.append(new_el) + return ast.Tuple(pool, new_elts, tup.ctx) + + # Constant: 默认值等 + if k == ast.ASTKind.Constant: + cnst: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + return ast.Constant(pool, cnst.const_kind, cnst.int_val, cnst.float_val, + cnst.str_val, cnst.lineno, cnst.col_offset) + + # 其他: 直接引用（不拷贝） + return node + + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, + args: ast.AST | t.CPtr, + tp_names: list[str] | t.CPtr, + type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: + """拷贝 Arguments 节点，替换 arg annotation 中的 T""" + if args is None: + return None + if args.kind() != ast.ASTKind.Arguments: + return args + old_args: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args) + # 拷贝 args 列表，替换每个 Arg 的 annotation + new_arg_list: list[ast.AST | t.CPtr] | t.CPtr = list[ast.AST | t.CPtr](pool, 8) + if old_args.args is not None: + an: t.CSizeT = old_args.args.__len__() + ai: t.CSizeT + for ai in range(an): + arg_node: ast.AST | t.CPtr = old_args.args.get(ai) + if arg_node is None: + continue + if arg_node.kind() == ast.ASTKind.Arg: + ag: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(arg_node) + new_anno: ast.AST | t.CPtr = _clone_annotation(pool, ag.annotation, tp_names, type_args) + new_arg: ast.Arg | t.CPtr = ast.Arg(pool, ag.arg, new_anno) + new_arg_list.append(new_arg) + else: + new_arg_list.append(arg_node) + return ast.Arguments(pool, new_arg_list, old_args.vararg, old_args.kwarg, + old_args.defaults, old_args.kw_defaults) + + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, + cd: ast.ClassDef | t.CPtr, + spec_name: str, + tp_names: list[str] | t.CPtr, + type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: + """拷贝 ClassDef 节点用于泛型特化 + + - 替换类名为 spec_name + - 清除 type_params（特化后不是泛型类） + - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T + - 方法体直接引用原始节点（通过 type_map 上下文处理 T） + """ + # 创建新 ClassDef（type_params=None 表示非泛型） + new_cd: ast.ClassDef | t.CPtr = ast.ClassDef(pool, spec_name, cd.bases, cd.keywords, + cd.decorator_list, None) + # 拷贝 children + if cd.children is not None: + cn: t.CSizeT = cd.children.__len__() + ci: t.CSizeT + for ci in range(cn): + child: ast.AST | t.CPtr = cd.children.get(ci) + if child is None: + continue + ck: int = child.kind() + + if ck == ast.ASTKind.AnnAssign: + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(child) + new_target: ast.AST | t.CPtr = aa.target + new_anno2: ast.AST | t.CPtr = _clone_annotation(pool, aa.annotation, tp_names, type_args) + new_value3: ast.AST | t.CPtr = aa.value + new_aa: ast.AnnAssign | t.CPtr = ast.AnnAssign(pool, new_target, new_anno2, new_value3, aa.simple) + new_cd.append(new_aa) + + elif ck == ast.ASTKind.FunctionDef: + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(child) + new_args2: ast.AST | t.CPtr = _clone_arguments(pool, fd.args, tp_names, type_args) + new_returns2: ast.AST | t.CPtr = _clone_annotation(pool, fd.returns, tp_names, type_args) + new_fd: ast.FunctionDef | t.CPtr = ast.FunctionDef(pool, fd.name, new_args2, + fd.decorator_list, new_returns2, 0) + # 方法体直接引用原始节点 + if fd.children is not None: + bcn: t.CSizeT = fd.children.__len__() + bci: t.CSizeT + for bci in range(bcn): + body_stmt: ast.AST | t.CPtr = fd.children.get(bci) + if body_stmt is not None: + new_fd.append(body_stmt) + new_cd.append(new_fd) + + else: + new_cd.append(child) + + return new_cd + + +# ============================================================ +# 泛型特化缓存（模块级） +# ============================================================ + +_spec_keys: list[str] | t.CPtr = None +_spec_names: list[str] | t.CPtr = None + + +def _find_cached_spec(spec_key: str) -> str: + """查找已缓存的特化名，未找到返回 None""" + if _spec_keys is None or spec_key is None: + return None + n: t.CSizeT = _spec_keys.__len__() + i: t.CSizeT + for i in range(n): + k: str = _spec_keys.get(i) + if k is not None and string.strcmp(k, spec_key) == 0: + return _spec_names.get(i) + return None + + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str): + """缓存特化结果""" + global _spec_keys, _spec_names + if _spec_keys is None: + _spec_keys = list[str](pool, 8) + _spec_names = list[str](pool, 8) + _spec_keys.append(spec_key) + _spec_names.append(spec_name) + + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, + class_name: str, + type_args: list[str] | t.CPtr) -> str: + """特化泛型类，返回特化类名 + + 1. 查找泛型模板 + 2. 生成特化名 + 3. 检查缓存 + 4. 拷贝 ClassDef 节点（替换类型注解中的 T） + 5. 设置 type_map 上下文（用于方法体中的 T 替换） + 6. 调用 translate_class_def 翻译特化类 + 7. 返回特化名 + """ + if trans is None or class_name is None or type_args is None: + return None + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + + # 1. 查找泛型模板 + template_cd: ast.ClassDef | t.CPtr = _find_generic_template(class_name) + if template_cd is None: + return None + + # 2. 生成特化名 + spec_name: str = _mangle_generic_class_name(pool, class_name, type_args) + if spec_name is None: + return None + + # 3. 检查缓存（spec_key 包含所有类型实参，避免多参数泛型缓存冲突） + # 分配新缓冲区避免原地 strcat 溢出 + sk_n: t.CSizeT = type_args.__len__() + sk_total: t.CSizeT = string.strlen(class_name) + 2 # "<" 和 ">" + sk_i: t.CSizeT + for sk_i in range(sk_n): + if sk_i > 0: + sk_total += 1 # "," + sk_ta: str = type_args.get(sk_i) + if sk_ta is not None: + sk_total += string.strlen(sk_ta) + sk_total += 1 # NUL + spec_key: str = pool.alloc(sk_total) + if spec_key is None: + return None + spec_key[0] = '\0' + string.strcat(spec_key, class_name) + string.strcat(spec_key, "<") + for sk_i in range(sk_n): + if sk_i > 0: + string.strcat(spec_key, ",") + sk_ta: str = type_args.get(sk_i) + if sk_ta is not None: + string.strcat(spec_key, sk_ta) + string.strcat(spec_key, ">") + cached: str = _find_cached_spec(spec_key) + if cached is not None: + return cached + + # 4. 检查是否已注册（Phase B 重复特化） + existing: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct(spec_name) + if existing is not None: + _cache_spec(pool, spec_key, spec_name) + return spec_name + + # 预注册缓存（防止递归） + _cache_spec(pool, spec_key, spec_name) + + # 5. 拷贝 ClassDef 节点 + tp_names: list[str] | t.CPtr = template_cd.type_params + spec_cd: ast.ClassDef | t.CPtr = _clone_classdef_for_spec( + pool, template_cd, spec_name, tp_names, type_args) + if spec_cd is None: + return None + + # 6. 设置 type_map 上下文（用于方法体中的 T 替换） + trans.GenericTypeParamNames = tp_names + trans.GenericTypeArgs = type_args + HandlesType.set_generic_context(tp_names, type_args) + + # 7. 翻译特化类 + translate_class_def(trans, spec_cd) + + # 8. 清除 type_map 上下文 + trans.GenericTypeParamNames = None + trans.GenericTypeArgs = None + HandlesType.clear_generic_context() + + return spec_name + + +def translate_class_def(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译 class 定义，返回 0""" + if node is None or trans is None: + return 0 + + cd: ast.ClassDef | t.CPtr = (ast.ClassDef | t.CPtr)(node) + if cd is None or cd.name is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + class_name: str = cd.name + + # 命名空间隔离：标记为当前文件可见（必须在 existing 检查之前， + # 因为 Phase B 时结构体已注册会提前 return，否则永远无法标记本地类） + HandlesStruct.add_visible_struct(pool, class_name) + + # 枚举类：交给 HandlesEnum 处理 + if _is_enum_class(cd) == 1: + return _translate_enum_def(trans, cd) + + # 联合体类：交给联合体处理 + if _is_union_class(cd) == 1: + return _translate_union_def(trans, cd) + + # 泛型类：存储为模板，不发射 IR（等实例化时特化） + if _is_generic_class(cd) == 1: + _register_generic_template(pool, cd) + return 0 + + # 检查是否已注册（用 SHA1 区分跨模块同名类） + existing: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + class_name, trans.ModuleSha1) + if existing is not None: + # 结构体已注册（Phase B 重新翻译），但仍需在当前模块中翻译方法 + if existing.Ty is not None: + # Phase B: 将已存在的命名结构体注册到当前模块（输出类型定义行） + llvmlite.module_add_named_type(trans.Module, pool, existing.Ty) + _translate_oop_methods(trans, cd, existing.Ty, class_name) + # SHA1 已在 register_struct 时设置，无需再补 + return 0 + + # ============================================================ + # VTable 检测：判断是否启用虚表 + # ============================================================ + has_vtable: int = _detect_vtable_status(cd, trans) + parent_name: str = _get_parent_class(cd) + is_novtable_deco: int = 0 + if cd.decorator_list is not None: + if _has_decorator(cd.decorator_list, "NoVTable") == 1: + is_novtable_deco = 1 + + # ============================================================ + # 1. 遍历 class body，收集字段信息 + # ============================================================ + # 用 CSizeT 数组存储指针值（64 位系统上 8 字节） + field_names_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + field_types_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + field_defaults_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + field_annot_buf: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if field_names_buf is None or field_types_buf is None or field_defaults_buf is None: + return 0 + if field_annot_buf is None: + return 0 + string.memset(field_names_buf, 0, 8 * 32) + string.memset(field_types_buf, 0, 8 * 32) + string.memset(field_defaults_buf, 0, 8 * 32) + string.memset(field_annot_buf, 0, 8 * 32) + + field_count: int = 0 + + # ============================================================ + # 1.5 继承父类字段（字段展平） + # + # 父类字段排在子类字段之前，结构体布局: + # [vtable_ptr?, parent_field1, ..., child_field1, ...] + # 跳过父类的 __vtable__ 字段（子类会有自己的 vtable 指针） + # ============================================================ + if parent_name is not None: + # 用 SHA1 感知查找父类，规避跨模块同名找错 + parent_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + parent_name, trans.ModuleSha1) + if parent_entry is not None: + pfc: int = parent_entry.FieldCount + for pfi in range(pfc): + pfe: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct._get_field_entry( + parent_entry, pfi) + if pfe is None: + continue + pfn: str = HandlesStruct.get_field_name_ptr(pfe) + if pfn is None: + continue + # 跳过父类的 vtable 指针字段 + if string.strcmp(pfn, "__vtable__") == 0: + continue + if field_count < 32: + pft: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_field_type_ptr(pfe) + pfd: ast.AST | t.CPtr = HandlesStruct.get_field_default_ptr(pfe) + pfa: str = HandlesStruct.get_field_annot_class_name(pfe) + field_names_buf[field_count] = t.CSizeT(pfn) + field_types_buf[field_count] = t.CSizeT(pft) + if pfd is not None: + field_defaults_buf[field_count] = t.CSizeT(pfd) + else: + field_defaults_buf[field_count] = 0 + if pfa is not None: + stored_val: t.CSizeT = t.CSizeT(pfa) + field_annot_buf[field_count] = stored_val + else: + field_annot_buf[field_count] = 0 + field_count += 1 + + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + if children is None: + return 0 + + cn: t.CSizeT = children.__len__() + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + sk: int = stmt.kind() + + if sk != ast.ASTKind.AnnAssign: continue + + # AnnAssign(target=Name(id), annotation=type, value=...) + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(stmt) + if aa is None or aa.target is None: + continue + + # 获取字段名 + if aa.target.kind() != ast.ASTKind.Name: + continue + tgt: ast.Name | t.CPtr = (ast.Name | t.CPtr)(aa.target) + if tgt.id is None: + continue + + # 跳过编译期元数据字段（__provides__/__requires__/__require_must__） + # 这些字段只用于 with 上下文的静态可达性检查，不生成运行时代码 + # 避免类体中 list 字面量翻译失败（类体无 pool 变量上下文） + if string.strcmp(tgt.id, "__provides__") == 0: + continue + if string.strcmp(tgt.id, "__requires__") == 0: + continue + if string.strcmp(tgt.id, "__require_must__") == 0: + continue + + # 解析字段类型 + field_ty: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, aa.annotation, trans._imported_modules, trans._from_imports) + if field_ty is None: + field_ty = llvmlite.Int32(pool) + + # 存储到数组（指针转 CSizeT） + if field_count < 32: + diag_name_val: t.CSizeT = t.CSizeT(tgt.id) + field_names_buf[field_count] = diag_name_val + field_types_buf[field_count] = t.CSizeT(field_ty) + # 存储默认值 AST 指针（None=无默认值） + if aa.value is not None: + field_defaults_buf[field_count] = t.CSizeT(aa.value) + else: + field_defaults_buf[field_count] = 0 + # 提取原始类型注解的类名（联合类型简化为 Ptr(i8) 时回退查找结构体） + if aa.annotation is not None: + annot_cn: str = HandlesType.extract_class_name_from_annotation( + aa.annotation, trans._imported_modules) + if annot_cn is not None: + diag_cn_val: t.CSizeT = t.CSizeT(annot_cn) + field_annot_buf[field_count] = diag_cn_val + else: + field_annot_buf[field_count] = 0 + else: + field_annot_buf[field_count] = 0 + field_count += 1 + + if field_count == 0: + stdio.printf("[CLASS] %s has no fields\n", class_name) + return 0 + + # ============================================================ + # 2. 创建 LLVM StructType + # ============================================================ + # 构建 ParamNode 链表 + first_node: llvmlite.ParamNode | t.CPtr = None + prev_node: llvmlite.ParamNode | t.CPtr = None + total_field_count: int = field_count + + # 如果有虚表，首字段添加 i8* vtable 指针 + if has_vtable == 1: + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + vtable_node: llvmlite.ParamNode | t.CPtr = pool.alloc(llvmlite.ParamNode.__sizeof__()) + if vtable_node is None: + stdio.printf("[CLASS] failed to alloc vtable node for %s\n", class_name) + return 0 + string.memset(vtable_node, 0, llvmlite.ParamNode.__sizeof__()) + vtable_node.Ty = i8_ptr_ty + first_node = vtable_node + prev_node = vtable_node + total_field_count = field_count + 1 + + for fi in range(field_count): + fty_addr: t.CSizeT = field_types_buf[fi] + if fty_addr == 0: + continue + fty: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)(t.CVoid(fty_addr, t.CPtr)) + + # 创建 ParamNode + pnode: llvmlite.ParamNode | t.CPtr = pool.alloc(llvmlite.ParamNode.__sizeof__()) + if pnode is None: + continue + string.memset(pnode, 0, llvmlite.ParamNode.__sizeof__()) + pnode.Ty = fty + + if first_node is None: + first_node = pnode + if prev_node is not None: + prev_node.Next = pnode + prev_node = pnode + + # 构造命名结构体类型名: "sha1.ClassName" 或 "ClassName" + type_name: str = class_name + if trans.ModuleSha1 is not None: + name_buf: t.CChar | t.CPtr = pool.alloc(64) + if name_buf is not None: + viperlib.snprintf(name_buf, 64, "%s.%s", trans.ModuleSha1, class_name) + type_name = name_buf + + # 创建 StructType（命名结构体） + struct_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Struct(pool, first_node, total_field_count, type_name) + if struct_ty is None: + stdio.printf("[CLASS] failed to create StructType for %s\n", class_name) + return 0 + + # 注册命名结构体到模块（输出 %"name" = type { ... } 定义行） + llvmlite.module_add_named_type(trans.Module, pool, struct_ty) + + # 扫描字段类型，为跨模块结构体引用添加 opaque 声明 + # 例如 list[int].__pool__ 引用 memhub.MemManager，需要在当前模块添加 %"sha1.MemManager" = type opaque + for fi in range(field_count): + fty_addr_op: t.CSizeT = field_types_buf[fi] + if fty_addr_op == 0: + continue + fty_op: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)(t.CVoid(fty_addr_op, t.CPtr)) + llvmlite.module_ensure_opaque_for_type(trans.Module, pool, fty_op) + + # ============================================================ + # 3. 注册到 HandlesStruct + # ============================================================ + entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.register_struct( + pool, class_name, struct_ty, trans.ModuleSha1) + if entry is None: + stdio.printf("[CLASS] failed to register %s\n", class_name) + return 0 + + # 标记 VTable 状态（直接用 entry，规避跨模块同名 find_struct 找错） + if has_vtable == 1: + entry.HasVTable = 1 + if is_novtable_deco == 1: + entry.IsNoVTable = 1 + if parent_name is not None: + entry.ParentName = parent_name + + # 添加字段信息 + # 如果有虚表，先添加 __vtable__ 字段占位（索引 0），用户字段从索引 1 开始 + if has_vtable == 1: + i8_ptr_ty2: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + HandlesStruct.add_field(pool, entry, "__vtable__", i8_ptr_ty2, None, None) + + for fi in range(field_count): + fname_addr: t.CSizeT = field_names_buf[fi] + if fname_addr == 0: + continue + fname: str = (str | t.CPtr)(t.CVoid(fname_addr, t.CPtr)) + fty_addr2: t.CSizeT = field_types_buf[fi] + if fty_addr2 == 0: + continue + fty2: llvmlite.LLVMType | t.CPtr = (llvmlite.LLVMType | t.CPtr)(t.CVoid(fty_addr2, t.CPtr)) + # 恢复默认值 AST 指针 + fdef_addr: t.CSizeT = field_defaults_buf[fi] + fdef: ast.AST | t.CPtr = None + if fdef_addr != 0: + fdef = (ast.AST | t.CPtr)(t.CVoid(fdef_addr, t.CPtr)) + # 恢复注解类名指针 + fannot_addr: t.CSizeT = field_annot_buf[fi] + fannot: t.CChar | t.CPtr = None + if fannot_addr != 0: + fannot = (t.CChar | t.CPtr)(t.CVoid(fannot_addr, t.CPtr)) + if fname is not None and fty2 is not None: + HandlesStruct.add_field(pool, entry, fname, fty2, fdef, fannot) + + stdio.printf("[CLASS] registered %s with %d fields (vtable=%d)\n", + class_name, field_count, has_vtable) + + # Phase 1a 声明模式：只注册 struct，不翻译方法体 + if trans._declare_only == 1: + return 0 + + # ============================================================ + # 4. OOP 方法处理：扫描 FunctionDef，标记 OOP，翻译方法，生成 __before_init__ + # ============================================================ + _translate_oop_methods(trans, cd, struct_ty, class_name) + + return 0 + + +# ============================================================ +# VTable 生成 +# +# 生成 VTable 全局变量: +# @"SHA1.ClassName.vtable" = internal constant { i8*, i8*, ... } { +# i8* bitcast (ret (params)* @"SHA1.ClassName.method1" to i8*), +# ... +# } +# +# VTable 槽位顺序: class body 中 FunctionDef 的出现顺序 +# ============================================================ + +VTBL_METHOD_MAX: t.CDefine = 32 + + +# _vtable_strcat — 将 src 追加到 dst 末尾 +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, + src: t.CChar | t.CPtr): + """将 src 追加到 dst 末尾""" + if dst is None or src is None: + return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: + return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' + + +# _vtable_name_needs_quote — 检查 LLVM 标识符是否需要引号 +def _vtable_name_needs_quote(name: str) -> int: + """检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）""" + if name is None: + return 0 + if '0' <= name[0] <= '9': + return 1 + for i in name: + if i == '.': + return 1 + return 0 + + +# _format_func_type_str — 从 Function 对象生成函数类型字符串 +# +# 格式: ret_ty (param1_ty, param2_ty, ...) +# 例如: void ({i32, i32}*, i32) +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, + buf: t.CChar | t.CPtr, buf_size: t.CSizeT, + func: llvmlite.Function | t.CPtr) -> int: + """从 Function 对象生成函数类型字符串""" + if buf is None or func is None or buf_size == 0: + return 0 + buf[0] = '\0' + + # 打印返回类型（使用访问器函数避免跨模块直接字段访问） + ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(func) + ret_buf: t.CChar | t.CPtr = pool.alloc(128) + if ret_buf is not None and ret_ty is not None: + ret_buf[0] = '\0' + llvmlite.TypePrint(ret_buf, 128, ret_ty, pool) + _vtable_strcat(buf, buf_size, ret_buf) + _vtable_strcat(buf, buf_size, " (") + + # 遍历参数（使用访问器函数避免跨模块直接字段访问） + cur: llvmlite.Param | t.CPtr = llvmlite.function_get_param_head(func) + first: int = 1 + param_buf: t.CChar | t.CPtr = pool.alloc(128) + while cur is not None: + if first == 0: + _vtable_strcat(buf, buf_size, ", ") + if param_buf is not None: + param_buf[0] = '\0' + param_ty: llvmlite.LLVMType | t.CPtr = llvmlite.param_get_ty(cur) + if param_ty is not None: + llvmlite.TypePrint(param_buf, 128, param_ty, pool) + _vtable_strcat(buf, buf_size, param_buf) + cur = llvmlite.param_get_next(cur) + first = 0 + _vtable_strcat(buf, buf_size, ")") + return 1 + + +# _generate_vtable — 生成 VTable 全局变量 +# +# 1. 收集 class body 中所有 FunctionDef 作为虚方法 +# 2. 构造 VTable 类型 { i8*, i8*, ... } +# 3. 构造初始化字符串 { i8* bitcast (...), ... } +# 4. 创建全局变量 @"SHA1.ClassName.vtable" +# 5. 存储虚方法名列表到 StructEntry +def _generate_vtable(trans: HT.Translator | t.CPtr, + cd: ast.ClassDef | t.CPtr, + class_name: str) -> int: + """生成 VTable 全局变量""" + if trans is None or cd is None or class_name is None: + return 0 + + # 懒导入 + import lib.core.Handles.HandlesFunctions as HandlesFunctions + import lib.core.Handles.HandlesExprCall as HandlesExprCall + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + if children is None: + return 0 + cn: t.CSizeT = children.__len__() + + # 1. 收集虚方法信息 + full_names_buf: t.CSizeT | t.CPtr = pool.alloc(8 * VTBL_METHOD_MAX) + method_names_buf: t.CSizeT | t.CPtr = pool.alloc(8 * VTBL_METHOD_MAX) + # 记录每个方法的父类 SHA1（非 0=继承方法用父模块 SHA1 mangling，0=当前类方法） + method_parent_sha1s: t.CSizeT | t.CPtr = pool.alloc(8 * VTBL_METHOD_MAX) + if full_names_buf is None or method_names_buf is None or method_parent_sha1s is None: + return 0 + string.memset(full_names_buf, 0, 8 * VTBL_METHOD_MAX) + string.memset(method_names_buf, 0, 8 * VTBL_METHOD_MAX) + string.memset(method_parent_sha1s, 0, 8 * VTBL_METHOD_MAX) + + method_count: int = 0 + + # ============================================================ + # 1.5 继承父类虚方法 + # + # 父类虚方法排在子类虚方法之前，保持 vtable 槽位顺序一致。 + # 如果子类覆盖了方法，full_name 用子类的；否则用父类的。 + # 用 SHA1 感知查找，规避跨模块同名 find_struct 找错 + # ============================================================ + vt_self_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + class_name, trans.ModuleSha1) + parent_name: str = None + if vt_self_entry is not None: + parent_name = vt_self_entry.ParentName + if parent_name is not None: + # 查找父类 entry（优先同模块 SHA1 匹配） + vt_parent_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_module( + parent_name, trans.ModuleSha1) + if vt_parent_entry is None: + # 跨模块继承: 父类在不同模块定义（SHA1 不同），回退到按类名查找 + vt_parent_entry = HandlesStruct.find_struct(parent_name) + parent_vt_count: int = 0 + if vt_parent_entry is not None: + parent_vt_count = vt_parent_entry.VTableMethodCount + if parent_vt_count > 0: + for pvi in range(parent_vt_count): + if method_count >= VTBL_METHOD_MAX: + break + pmethod_name: str = None + if vt_parent_entry is not None: + pmethod_name = HandlesStruct.get_vtable_method_name(parent_name, pvi) + if pmethod_name is None: + continue + # 检查子类是否覆盖了该方法，以及是否标记 @t.NoVTable + child_overrides: int = 0 + child_nvt: int = 0 + for ci2 in range(cn): + stmt2: ast.AST | t.CPtr = children.get(ci2) + if stmt2 is None: + continue + if stmt2.kind() != ast.ASTKind.FunctionDef: + continue + fd2: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(stmt2) + if fd2 is not None and fd2.name is not None: + if string.strcmp(fd2.name, pmethod_name) == 0: + child_overrides = 1 + # 检查子类覆盖方法是否标记 @t.NoVTable + if fd2.decorator_list is not None: + if _has_decorator(fd2.decorator_list, "NoVTable") == 1: + child_nvt = 1 + break + # 构造 full_name + # 子类覆盖且未标记 NoVTable → 用子类实现 + # 子类未覆盖 或 子类覆盖但标记 NoVTable → 用父类实现（保持 vtable 槽位） + pfname_buf: t.CChar | t.CPtr = pool.alloc(128) + if pfname_buf is not None: + if child_overrides == 1 and child_nvt == 0: + viperlib.snprintf(pfname_buf, 128, "%s.%s", class_name, pmethod_name) + else: + viperlib.snprintf(pfname_buf, 128, "%s.%s", parent_name, pmethod_name) + full_names_buf[method_count] = t.CSizeT(pfname_buf) + # 复制方法名（短名） + pmname_len: t.CSizeT = string.strlen(pmethod_name) + pmname_buf: t.CChar | t.CPtr = pool.alloc(pmname_len + 1) + if pmname_buf is not None: + string.strcpy(pmname_buf, pmethod_name) + method_names_buf[method_count] = t.CSizeT(pmname_buf) + # 记录 mangling 用的 SHA1（继承未覆盖的方法用父模块 SHA1） + if child_overrides == 1 and child_nvt == 0: + # 子类覆盖: 用当前模块 SHA1（保持 0） + method_parent_sha1s[method_count] = 0 + else: + # 继承未覆盖: 用父类模块 SHA1 + if vt_parent_entry is not None and vt_parent_entry.ModuleSha1 is not None: + method_parent_sha1s[method_count] = t.CSizeT(vt_parent_entry.ModuleSha1) + method_count += 1 + + # 2. 收集当前类的虚方法（跳过已从父类继承的） + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + if stmt.kind() != ast.ASTKind.FunctionDef: + continue + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(stmt) + if fd is None or fd.name is None: + continue + if method_count >= VTBL_METHOD_MAX: + break + + mname: str = fd.name + # 跳过构造函数和特殊方法（不应放入虚表） + if string.strcmp(mname, "__init__") == 0: + continue + if string.strcmp(mname, "__before_init__") == 0: + continue + if string.strcmp(mname, "__new__") == 0: + continue + # 函数级装饰器过滤：判断方法是否应该进入虚表 + if _should_method_be_virtual(cd, fd, trans) == 0: + continue + # 检查是否已从父类继承（避免重复） + already_in_vt: int = 0 + for mi in range(method_count): + exist_addr: t.CSizeT = method_names_buf[mi] + if exist_addr != 0: + exist_name: str = (str | t.CPtr)(t.CVoid(exist_addr, t.CPtr)) + if exist_name is not None and string.strcmp(exist_name, mname) == 0: + already_in_vt = 1 + break + if already_in_vt == 1: + continue + # 构造 full_name = "ClassName.method_name" + fname_buf: t.CChar | t.CPtr = pool.alloc(128) + if fname_buf is not None: + viperlib.snprintf(fname_buf, 128, "%s.%s", class_name, mname) + full_names_buf[method_count] = t.CSizeT(fname_buf) + + # 复制方法名 + mname_len: t.CSizeT = string.strlen(mname) + mname_buf: t.CChar | t.CPtr = pool.alloc(mname_len + 1) + if mname_buf is not None: + string.strcpy(mname_buf, mname) + method_names_buf[method_count] = t.CSizeT(mname_buf) + + method_count += 1 + + if method_count == 0: + return 0 + + # 2. 构造 VTable 类型: { i8*, i8*, ..., i8* } + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + first_node: llvmlite.ParamNode | t.CPtr = None + tail_node: llvmlite.ParamNode | t.CPtr = None + for i in range(method_count): + pnode: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, i8_ptr_ty) + if pnode is None: + continue + tail_node = llvmlite.param_list_append(first_node, tail_node, pnode) + if first_node is None: + first_node = tail_node + + vtable_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Struct(pool, first_node, method_count, None) + if vtable_ty is None: + return 0 + + # 3. 构造初始化字符串 + init_buf: t.CChar | t.CPtr = pool.alloc(4096) + if init_buf is None: + return 0 + init_buf[0] = '\0' + _vtable_strcat(init_buf, 4096, "{ ") + + for i in range(method_count): + if i > 0: + _vtable_strcat(init_buf, 4096, ", ") + + full_name_addr: t.CSizeT = full_names_buf[i] + if full_name_addr == 0: + _vtable_strcat(init_buf, 4096, "i8* null") + continue + + full_name: str = (str | t.CPtr)(t.CVoid(full_name_addr, t.CPtr)) + + # 检查是否是继承方法（method_parent_sha1s 非 0 表示用父模块 SHA1 mangling） + parent_sha1_addr: t.CSizeT = method_parent_sha1s[i] + if parent_sha1_addr != 0: + # 继承方法: 用父类模块 SHA1 做 mangling + parent_sha1: str = (str | t.CPtr)(t.CVoid(parent_sha1_addr, t.CPtr)) + mangled_name: str = HandlesFunctions._mangle_name_with_sha1(parent_sha1, full_name) + # 从全局函数表查找（跨模块函数不在当前模块表中） + func: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_global(full_name) + else: + # 当前类方法: 用当前模块 SHA1 + mangled_name: str = HandlesFunctions._mangle_name(trans, full_name) + func: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_in_table( + trans._funcs, trans._func_count, full_name) + # 当前模块表找不到时，回退到全局表 + if func is None: + func = HandlesExprCall.find_func_global(full_name) + + if func is None: + _vtable_strcat(init_buf, 4096, "i8* null") + continue + + # 构造函数类型字符串 + func_ty_buf: t.CChar | t.CPtr = pool.alloc(512) + if func_ty_buf is None: + _vtable_strcat(init_buf, 4096, "i8* null") + continue + _format_func_type_str(pool, func_ty_buf, 512, func) + + # 构造 bitcast 字符串: i8* bitcast (func_ty* @mangled to i8*) + _vtable_strcat(init_buf, 4096, "i8* bitcast (") + _vtable_strcat(init_buf, 4096, func_ty_buf) + _vtable_strcat(init_buf, 4096, "* @") + + if _vtable_name_needs_quote(mangled_name) != 0: + _vtable_strcat(init_buf, 4096, "\"") + _vtable_strcat(init_buf, 4096, mangled_name) + _vtable_strcat(init_buf, 4096, "\"") + else: + _vtable_strcat(init_buf, 4096, mangled_name) + _vtable_strcat(init_buf, 4096, " to i8*)") + + _vtable_strcat(init_buf, 4096, " }") + + # 4. 构造 VTable 全局变量名 + vtable_name_buf: t.CChar | t.CPtr = pool.alloc(128) + if vtable_name_buf is None: + return 0 + viperlib.snprintf(vtable_name_buf, 128, "%s.vtable", class_name) + vtable_mangled: str = HandlesFunctions._mangle_name(trans, vtable_name_buf) + + # 5. 创建全局变量 + gv: llvmlite.GlobalVariable | t.CPtr = llvmlite.new_global_variable( + pool, vtable_mangled, vtable_ty) + if gv is None: + return 0 + gv.Initializer = init_buf + gv.IsConstant = 1 + gv.Linkage = "internal" + + # 添加到模块 + llvmlite.module_add_global(mod, gv) + + # 6. 存储虚方法名列表到 StructEntry（直接用 entry，规避跨模块同名 find_struct 找错） + if vt_self_entry is not None: + vt_self_entry.VTableMethods = method_names_buf + vt_self_entry.VTableMethodCount = method_count + + stdio.printf("[VTABLE] generated vtable for %s with %d methods\n", + class_name, method_count) + return 0 + + +# ============================================================ +# OOP 方法翻译 +# +# 存在任意 FunctionDef 的 class 自动升级为 OOP 结构体。 +# 方法翻译为 SHA1.ClassName.method_name(self: Ptr(StructTy), ...) 函数。 +# self 参数直接注册为 SSA 值（不创建 alloca），使 self.field 能通过 GEP 直接访问。 +# ============================================================ + + +# ============================================================ +# _translate_oop_methods — 扫描 class body 中的方法并翻译 +# +# 遍历 ClassDef.children，对每个 FunctionDef: +# - 标记 OOP（mark_as_oop） +# - 如果是 __init__，标记 has_init +# - 翻译方法（生成 SHA1.ClassName.method_name 函数） +# 翻译完所有方法后，生成 __before_init__ 函数 +# ============================================================ +def _translate_oop_methods(trans: HT.Translator | t.CPtr, + cd: ast.ClassDef | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str) -> int: + """扫描 class body 中的方法并翻译，生成 __before_init__""" + if trans is None or cd is None or struct_ty is None or class_name is None: + return 0 + + children: list[ast.AST | t.CPtr] | t.CPtr = cd.children + if children is None: + return 0 + + cn: t.CSizeT = children.__len__() + has_method: int = 0 + has_init: int = 0 + has_new: int = 0 + + # 提前用类型指针定位 entry（规避跨模块同名类 find_struct 找错） + oop_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + + # 第一遍：检测是否有方法，标记 OOP + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + if stmt.kind() == ast.ASTKind.FunctionDef: + has_method = 1 + fd_check: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(stmt) + if fd_check is not None and fd_check.name is not None: + if string.strcmp(fd_check.name, "__init__") == 0: + has_init = 1 + if string.strcmp(fd_check.name, "__new__") == 0: + has_new = 1 + + if has_method == 0: + # 子类没有自己的方法，但可能继承了父类的虚方法 + # 用 oop_entry 直接获取 ParentName，规避 find_struct 按名查找找错 + parent_name: str = None + if oop_entry is not None: + parent_name = oop_entry.ParentName + if parent_name is not None: + parent_vt_count: int = HandlesStruct.get_vtable_method_count(parent_name) + if parent_vt_count > 0: + has_method = 1 # 标记为有方法（继承的虚方法） + if has_method == 0: + return 0 + + # 标记为 OOP 结构体 + if oop_entry is not None: + oop_entry.IsOOP = 1 + if has_init != 0: + oop_entry.HasInit = 1 + if has_new != 0: + oop_entry.HasNew = 1 + + # 第二遍：翻译每个方法 + for ci in range(cn): + stmt: ast.AST | t.CPtr = children.get(ci) + if stmt is None: + continue + if stmt.kind() == ast.ASTKind.FunctionDef: + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(stmt) + if fd is not None and fd.name is not None: + _translate_method(trans, fd, struct_ty, class_name) + + # 如果有虚表，先生成 VTable 全局变量（设置 VTableMethodCount） + # 必须在 __before_init__ 之前，否则 __before_init__ 读不到 VTableMethodCount + # 用 oop_entry 直接检查，规避跨模块同名 find_struct 找错 + has_vt_flag: int = 0 + if oop_entry is not None: + has_vt_flag = oop_entry.HasVTable + if has_vt_flag == 1: + _generate_vtable(trans, cd, class_name) + + # 生成 __before_init__ 函数（零值填充 + 默认值赋值 + vtable 指针设置） + _generate_before_init(trans, struct_ty, class_name) + + return 0 + + +# ============================================================ +# _translate_method — 翻译单个方法 +# +# 生成: define @SHA1.ClassName.method_name(Ptr(StructTy) %self, ...) +# +# self 参数处理： +# - 类型 Ptr(struct_ty)，直接注册为 SSA 值（不创建 alloca） +# - 使 self.field 能通过 GEP 直接访问原始结构体 +# ============================================================ +def _translate_method(trans: HT.Translator | t.CPtr, + fd: ast.FunctionDef | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str) -> int: + """翻译单个方法，返回 0""" + if trans is None or fd is None or struct_ty is None or class_name is None: + return 0 + + # 懒导入 + import lib.core.Handles.HandlesFunctions as HandlesFunctions + import lib.core.Handles.HandlesVar as HandlesVar + import lib.core.Handles.HandlesBody as HandlesBody + import lib.core.Handles.HandlesType as HandlesType + import lib.core.Handles.HandlesExprCall as HandlesExprCall + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + imported_modules: str = trans._imported_modules + from_imports: str = trans._from_imports + funcs_ptr: HandlesExprCall.FuncEntry | t.CPtr = trans._funcs + func_count: int = trans._func_count + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + self_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, struct_ty) + + method_name: str = fd.name + + # 构建方法全名: ClassName.method_name + full_name_buf: t.CChar | t.CPtr = pool.alloc(128) + if full_name_buf is None: + return 0 + viperlib.snprintf(full_name_buf, 128, "%s.%s", class_name, method_name) + full_name: str = full_name_buf + + # SHA1 命名空间混淆 + mangled_name: str = HandlesFunctions._mangle_name(trans, full_name) + + # 推断返回类型 + ret_ty: llvmlite.LLVMType | t.CPtr = None + if fd.returns is not None: + ret_ty = HandlesType.resolve_annotation_type( + pool, fd.returns, imported_modules, from_imports) + if ret_ty is None and fd.returns is not None: + if HandlesType.has_decorator_marker(fd.returns, "State") != 0: + ret_ty = llvmlite.Void(pool) + if ret_ty is None: + # __init__ 和 __before_init__ 返回 void + if string.strcmp(method_name, "__init__") == 0: + ret_ty = llvmlite.Void(pool) + # __new__ 返回 Ptr(struct_ty)（结构体指针作为存储空间） + elif string.strcmp(method_name, "__new__") == 0: + ret_ty = llvmlite.Ptr(pool, struct_ty) + else: + param_types_str: str = HandlesType.build_param_types_str(pool, fd.args) + ret_ty = HandlesType.infer_return_type( + pool, fd.children, param_types_str) + + # 创建 LLVM 函数 + func: llvmlite.Function | t.CPtr = llvmlite.create_function( + pool, mod, mangled_name, ret_ty) + if func is None: + return 0 + + # 注册到函数表（用 ClassName.method_name 作为查找名，支持后缀匹配） + max_funcs: int = 256 + if HandlesExprCall.add_func_to_table(funcs_ptr, func_count, full_name, func, max_funcs) == 0: + trans._func_count = func_count + 1 + + # 添加 self 参数（Ptr(struct_ty)） + llvmlite.add_param(pool, func, self_ptr_ty, "%self") + + # 添加其他参数（支持类型注解，跳过索引 0 的 self 参数） + args_node: ast.Arguments | t.CPtr = fd.args + if args_node is not None: + ags: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags.args is not None: + alist: list[ast.AST | t.CPtr] | t.CPtr = ags.args + an: t.CSizeT = alist.__len__() + for ai in range(1, an): + arg: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist.get(ai)) + if arg is not None and arg.arg is not None: + param_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if arg.annotation is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg.annotation, imported_modules, from_imports) + if resolved is not None: + param_ty = resolved + pname: t.CChar | t.CPtr = pool.alloc(32) + if pname is not None: + viperlib.snprintf(pname, 32, "%%%s", arg.arg) + llvmlite.add_param(pool, func, param_ty, pname) + + # 创建 entry 块 + entry_blk: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, "entry") + if entry_blk is None: + return 0 + + # 创建方法专属 builder + func_builder: llvmlite.IRBuilder | t.CPtr = llvmlite.new_builder(pool, func) + if func_builder is None: + return 0 + llvmlite.position_at_end(func_builder, entry_blk) + + # 进入函数作用域 + HandlesVar.enter_scope(trans.SymTab, HandlesVar.SCOPE_FUNCTION) + + # 注册 self 为 SSA 值（不创建 alloca，不 store） + # 这样 self.field 通过 lookup_var 获取 Ptr(struct_ty) 后直接 GEP + self_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, self_ptr_ty, "%self") + HandlesVar.define_var(trans.SymTab, "self", self_val) + # 设置 self 的类型注解类名（属性访问 lookup_field 回退查找用） + HandlesVar.set_var_annot_class_name(trans.SymTab, "self", class_name) + + # 为其他参数创建 alloca 并 store（与普通函数一致，跳过索引 0 的 self 参数） + if args_node is not None: + ags2: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags2.args is not None: + alist2: list[ast.AST | t.CPtr] | t.CPtr = ags2.args + an2: t.CSizeT = alist2.__len__() + for ai2 in range(1, an2): + arg2: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist2.get(ai2)) + if arg2 is not None and arg2.arg is not None: + param_ty2: llvmlite.LLVMType | t.CPtr = i32_ty + if arg2.annotation is not None: + resolved2: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg2.annotation, imported_modules, from_imports) + if resolved2 is not None: + param_ty2 = resolved2 + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(func_builder, param_ty2) + if alloca is not None: + HandlesVar.define_var(trans.SymTab, arg2.arg, alloca) + # 存储原始类型注解的类名（方法调用检测时，Ptr(i8) 回退到类名查找结构体） + if arg2.annotation is not None: + cls_nm_oc: str = HandlesType.extract_class_name_from_annotation( + arg2.annotation, imported_modules) + if cls_nm_oc is not None: + HandlesVar.set_var_annot_class_name( + trans.SymTab, arg2.arg, cls_nm_oc) + pname2: t.CChar | t.CPtr = pool.alloc(32) + if pname2 is not None: + viperlib.snprintf(pname2, 32, "%%%s", arg2.arg) + param_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue( + pool, param_ty2, pname2) + llvmlite.build_store(func_builder, param_val, alloca) + + # 保存模块级作用域状态 + old_func: llvmlite.Function | t.CPtr = trans._cur_func + old_builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + old_global_count: int = trans._global_name_count + old_nonlocal_count: int = trans._nonlocal_name_count + old_env_count: int = trans._closure_env_count + + trans._cur_func = func + trans._cur_builder = func_builder + HT.clear_scope_names(trans) + + # 预扫描方法体：为局部变量提前创建 alloca + body: list[ast.AST | t.CPtr] | t.CPtr = fd.children + if body is not None: + bn: t.CSizeT = body.__len__() + for bi in range(bn): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.pre_scan_allocas(trans, stmt) + + # 翻译方法体 + if body is not None: + bn2: t.CSizeT = body.__len__() + for bi2 in range(bn2): + stmt2: ast.AST | t.CPtr = body.get(bi2) + if stmt2 is not None: + HandlesBody.translate_stmt(trans, stmt2) + + # 如果返回类型为 void，添加 ret void；否则添加隐式 ret 0 + is_void: int = 0 + if ret_ty is not None: + match ret_ty: + case llvmlite.LLVMType.Void(): + is_void = 1 + + if is_void != 0: + if llvmlite.builder_cur_block_is_terminated(func_builder) == 0: + llvmlite.build_ret_void(func_builder) + else: + last_is_return: int = 0 + if body is not None: + bn3: t.CSizeT = body.__len__() + if bn3 > 0: + last_stmt: ast.AST | t.CPtr = body.get(bn3 - 1) + if last_stmt is not None and last_stmt.kind() == ast.ASTKind.Return: + last_is_return = 1 + if last_is_return == 0: + if llvmlite.builder_cur_block_is_terminated(func_builder) == 0: + # __new__ 返回指针类型，隐式返回 null 指针 + if string.strcmp(method_name, "__new__") == 0: + null_val: llvmlite.Value | t.CPtr = llvmlite.ConstNull(pool, ret_ty, "null") + llvmlite.build_ret(func_builder, null_val) + else: + zero_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + llvmlite.build_ret(func_builder, zero_val) + + # 恢复模块级作用域 + HandlesVar.exit_scope(trans.SymTab) + trans._cur_func = old_func + trans._cur_builder = old_builder + trans._global_name_count = old_global_count + trans._nonlocal_name_count = old_nonlocal_count + trans._closure_env_count = old_env_count + + return 0 + + +# ============================================================ +# _generate_before_init — 生成 __before_init__ 函数 +# +# 生成: define void @SHA1.ClassName.__before_init__(Ptr(StructTy) %self) +# +# 执行: +# 1. store zeroinitializer 到 %self（零值填充） +# 2. 逐字段 store 默认值（如有） +# 3. ret void +# ============================================================ +def _generate_before_init(trans: HT.Translator | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str) -> int: + """生成 __before_init__ 函数，返回 0""" + if trans is None or struct_ty is None or class_name is None: + return 0 + + import lib.core.Handles.HandlesFunctions as HandlesFunctions + import lib.core.Handles.HandlesVar as HandlesVar + import lib.core.Handles.HandlesExpr as HandlesExpr + import lib.core.Handles.HandlesExprCall as HandlesExprCall + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + funcs_ptr: HandlesExprCall.FuncEntry | t.CPtr = trans._funcs + func_count: int = trans._func_count + + self_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, struct_ty) + void_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Void(pool) + + # 构建函数名: ClassName.__before_init__ + full_name_buf: t.CChar | t.CPtr = pool.alloc(128) + if full_name_buf is None: + return 0 + viperlib.snprintf(full_name_buf, 128, "%s.__before_init__", class_name) + full_name: str = full_name_buf + + mangled_name: str = HandlesFunctions._mangle_name(trans, full_name) + + # 创建函数 + func: llvmlite.Function | t.CPtr = llvmlite.create_function( + pool, mod, mangled_name, void_ty) + if func is None: + return 0 + + # 注册到函数表 + max_funcs: int = 256 + if HandlesExprCall.add_func_to_table(funcs_ptr, func_count, full_name, func, max_funcs) == 0: + trans._func_count = func_count + 1 + + # 添加 self 参数 + llvmlite.add_param(pool, func, self_ptr_ty, "%self") + + # 创建 entry 块 + builder + entry_blk: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, "entry") + if entry_blk is None: + return 0 + func_builder: llvmlite.IRBuilder | t.CPtr = llvmlite.new_builder(pool, func) + if func_builder is None: + return 0 + llvmlite.position_at_end(func_builder, entry_blk) + + # 注册 self 为 SSA 值 + self_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, self_ptr_ty, "%self") + + # 1. 零值填充: store zeroinitializer, Ptr(struct_ty)* %self + zero: llvmlite.Value | t.CPtr = llvmlite.ConstZero(pool, struct_ty) + if zero is not None: + llvmlite.build_store(func_builder, zero, self_val) + + # 1.5 如果有虚表，store vtable 全局地址到 __vtable__ 字段（索引 0） + # 用 find_struct_by_type 定位 entry，规避跨模块同名 find_struct 找错 + bi_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + bi_has_vt: int = 0 + if bi_entry is not None: + bi_has_vt = bi_entry.HasVTable + if bi_has_vt == 1: + vt_method_count: int = 0 + if bi_entry is not None: + vt_method_count = bi_entry.VTableMethodCount + if vt_method_count > 0: + # 构造 vtable 类型 { i8*, i8*, ..., i8* } + vt_i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + vt_i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, vt_i8_ty) + + vt_first_node: llvmlite.ParamNode | t.CPtr = None + vt_prev_node: llvmlite.ParamNode | t.CPtr = None + for vti in range(vt_method_count): + vtnode: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, vt_i8_ptr_ty) + if vtnode is None: + continue + if vt_first_node is None: + vt_first_node = vtnode + if vt_prev_node is not None: + llvmlite.paramnode_set_next(vt_prev_node, vtnode) + vt_prev_node = vtnode + + vt_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Struct(pool, vt_first_node, vt_method_count, None) + vt_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, vt_ty) + + # 构造 vtable 全局变量名 + vt_name_buf: t.CChar | t.CPtr = pool.alloc(128) + if vt_name_buf is not None: + viperlib.snprintf(vt_name_buf, 128, "%s.vtable", class_name) + vt_full_name: str = vt_name_buf + vt_mangled: str = HandlesFunctions._mangle_name(trans, vt_full_name) + + # GEP 到 __vtable__ 字段（索引 0） + vt_slot: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + func_builder, struct_ty, vt_i8_ptr_ty, self_val, 0) + if vt_slot is not None: + # 创建全局变量引用 Value + vt_ref: llvmlite.Value | t.CPtr = llvmlite.new_value(pool) + if vt_ref is not None: + llvmlite.value_set_ty(vt_ref, vt_ptr_ty) + llvmlite.value_set_isconst(vt_ref, 1) + # Name 格式: @"mangled" 或 @mangled + vt_ref_name: t.CChar | t.CPtr = pool.alloc(256) + if vt_ref_name is not None: + vt_ref_name[0] = '\0' + if _vtable_name_needs_quote(vt_mangled) != 0: + viperlib.snprintf(vt_ref_name, 256, "@\"%s\"", vt_mangled) + else: + viperlib.snprintf(vt_ref_name, 256, "@%s", vt_mangled) + llvmlite.value_set_name(vt_ref, vt_ref_name) + + # bitcast 到 i8* + vt_as_i8: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + func_builder, vt_ref, vt_i8_ptr_ty) + if vt_as_i8 is not None: + llvmlite.build_store(func_builder, vt_as_i8, vt_slot) + + # 2. 逐字段 store 默认值（用 bi_entry 直接访问，规避跨模块同名 find_struct 找错） + if bi_entry is not None: + for fi in range(bi_entry.FieldCount): + fe: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct._get_field_entry( + bi_entry, fi) + if fe is None or fe.DefaultVal is None: + continue + # 翻译默认值表达式 + default_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + func_builder, pool, mod, fe.DefaultVal, None, 0, trans) + if default_val is None: + continue + # GEP 到字段 + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + func_builder, struct_ty, fe.Ty, self_val, fe.Index) + if field_ptr is None: + continue + # 类型转换并 store + default_val = HandlesExpr.coerce_to_type(func_builder, default_val, fe.Ty) + llvmlite.build_store(func_builder, default_val, field_ptr) + + # 3. ret void + llvmlite.build_ret_void(func_builder) + + return 0 diff --git a/TransPyV/App/lib/core/Handles/HandlesEnum.py b/TransPyV/App/lib/core/Handles/HandlesEnum.py new file mode 100644 index 0000000..cdae79f --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesEnum.py @@ -0,0 +1,230 @@ +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio + + +# ============================================================ +# HandlesEnum - 枚举类型注册和成员查找 +# +# 管理 t.CEnum 派生类的成员信息: +# - 枚举名 → EnumEntry（基准类型 + 成员表） +# - 成员名 → EnumMember（值、类型） +# +# 使用全局数组存储，线性查找（枚举数量通常很少） +# ============================================================ + +ENUM_MAX: t.CDefine = 64 +ENUM_MEMBER_MAX: t.CDefine = 64 + + +# ============================================================ +# EnumMember - 枚举成员条目 +# ============================================================ +@t.NoVTable +class EnumMember: + Name: t.CChar | t.CPtr # 成员名（字符串） + Value: t.CInt64T # 成员的整数值 + Ty: llvmlite.LLVMType | t.CPtr # 成员的 LLVM 类型（用于混用类型场景） + + +# ============================================================ +# EnumEntry - 枚举类型条目 +# ============================================================ +@t.NoVTable +class EnumEntry: + Name: t.CChar | t.CPtr # 枚举类名 + BaseTy: llvmlite.LLVMType | t.CPtr # 基准类型（所有成员类型中最大的） + MemberCount: int # 成员数量 + Members: EnumMember | t.CPtr # 成员数组（ENUM_MEMBER_MAX 个槽位） + + +# ============================================================ +# 全局注册表（静态分配） +# ============================================================ +_enum_table: EnumEntry | t.CPtr = None +_enum_count: int = 0 + + +# ============================================================ +# init_enum_table — 初始化枚举注册表 +# ============================================================ +def init_enum_table(pool: memhub.MemBuddy | t.CPtr) -> int: + """初始化枚举注册表，返回 1 成功""" + global _enum_table + global _enum_count + + if _enum_table is not None: + return 1 + + entry_size: t.CSizeT = EnumEntry.__sizeof__() + _enum_table = pool.alloc(entry_size * ENUM_MAX) + if _enum_table is None: + return 0 + string.memset(_enum_table, 0, entry_size * ENUM_MAX) + _enum_count = 0 + return 1 + + +# ============================================================ +# _get_enum_entry — 获取第 i 个 EnumEntry 槽位 +# ============================================================ +def _get_enum_entry(i: int) -> EnumEntry | t.CPtr: + """获取第 i 个枚举条目""" + if _enum_table is None or i < 0 or i >= ENUM_MAX: + return None + entry_size: t.CSizeT = EnumEntry.__sizeof__() + addr: t.CUInt64T = t.CUInt64T(_enum_table) + i * entry_size + return (EnumEntry | t.CPtr)(t.CVoid(addr, t.CPtr)) + + +# ============================================================ +# _get_enum_member — 获取枚举中第 i 个 EnumMember 槽位 +# ============================================================ +def _get_enum_member(enum_entry: EnumEntry | t.CPtr, i: int) -> EnumMember | t.CPtr: + """获取枚举中第 i 个成员条目""" + if enum_entry is None or i < 0 or i >= ENUM_MEMBER_MAX: + return None + member_size: t.CSizeT = EnumMember.__sizeof__() + addr: t.CUInt64T = t.CUInt64T(enum_entry.Members) + i * member_size + return (EnumMember | t.CPtr)(t.CVoid(addr, t.CPtr)) + + +# ============================================================ +# register_enum — 注册枚举类型 +# +# 返回 EnumEntry 指针，可用于添加成员 +# ============================================================ +def register_enum(pool: memhub.MemBuddy | t.CPtr, + name: str, + base_ty: llvmlite.LLVMType | t.CPtr) -> EnumEntry | t.CPtr: + """注册枚举类型，返回 EnumEntry 指针""" + if init_enum_table(pool) == 0: + return None + + # 检查是否已注册 + existing: EnumEntry | t.CPtr = find_enum(name) + if existing is not None: + return existing + + if _enum_count >= ENUM_MAX: + stdio.printf("[ENUM] table full, cannot register %s\n", name) + return None + + entry: EnumEntry | t.CPtr = _get_enum_entry(_enum_count) + if entry is None: + return None + + # 分配成员数组 + member_size: t.CSizeT = EnumMember.__sizeof__() + entry.Members = pool.alloc(member_size * ENUM_MEMBER_MAX) + if entry.Members is None: + return None + string.memset(entry.Members, 0, member_size * ENUM_MEMBER_MAX) + + # 复制类名 + name_len: t.CSizeT = string.strlen(name) + name_buf: t.CChar | t.CPtr = pool.alloc(name_len + 1) + if name_buf is not None: + string.strcpy(name_buf, name) + entry.Name = name_buf + + entry.BaseTy = base_ty + entry.MemberCount = 0 + + _enum_count += 1 + return entry + + +# ============================================================ +# add_enum_member — 向枚举添加成员 +# ============================================================ +def add_enum_member(pool: memhub.MemBuddy | t.CPtr, + enum_entry: EnumEntry | t.CPtr, + member_name: str, + member_val: t.CInt64T, + member_ty: llvmlite.LLVMType | t.CPtr) -> int: + """向枚举添加成员，返回成员索引（-1 失败）""" + if enum_entry is None or member_name is None or member_ty is None: + return -1 + + if enum_entry.MemberCount >= ENUM_MEMBER_MAX: + stdio.printf("[ENUM] member table full for %s\n", enum_entry.Name) + return -1 + + idx: int = enum_entry.MemberCount + me: EnumMember | t.CPtr = _get_enum_member(enum_entry, idx) + if me is None: + return -1 + + # 复制成员名 + name_len: t.CSizeT = string.strlen(member_name) + name_buf: t.CChar | t.CPtr = pool.alloc(name_len + 1) + if name_buf is not None: + string.strcpy(name_buf, member_name) + me.Name = name_buf + + me.Value = member_val + me.Ty = member_ty + + enum_entry.MemberCount = idx + 1 + return idx + + +# ============================================================ +# find_enum — 按枚举类名查找 +# ============================================================ +def find_enum(name: str) -> EnumEntry | t.CPtr: + """按枚举类名查找，返回 EnumEntry 或 None""" + if name is None or _enum_table is None: + return None + for i in range(_enum_count): + entry: EnumEntry | t.CPtr = _get_enum_entry(i) + if entry is not None and entry.Name is not None: + if string.strcmp(entry.Name, name) == 0: + return entry + return None + + +# ============================================================ +# lookup_enum_member — 按枚举类名和成员名查找 +# ============================================================ +def lookup_enum_member(enum_name: str, + member_name: str) -> EnumMember | t.CPtr: + """按枚举类名和成员名查找，返回 EnumMember 或 None""" + if enum_name is None or member_name is None: + return None + entry: EnumEntry | t.CPtr = find_enum(enum_name) + if entry is None: + return None + for mi in range(entry.MemberCount): + me: EnumMember | t.CPtr = _get_enum_member(entry, mi) + if me is not None and me.Name is not None: + if string.strcmp(me.Name, member_name) == 0: + return me + return None + + +# ============================================================ +# is_enum_class — 检查类名是否为已注册枚举 +# ============================================================ +def is_enum_class(name: str) -> int: + """检查类名是否为已注册枚举，返回 1=是 / 0=否""" + if name is None: + return 0 + if find_enum(name) is not None: + return 1 + return 0 + + +# ============================================================ +# get_enum_base_type — 按枚举类名获取基准类型 +# ============================================================ +def get_enum_base_type(enum_name: str) -> llvmlite.LLVMType | t.CPtr: + """按枚举类名获取基准类型""" + entry: EnumEntry | t.CPtr = find_enum(enum_name) + if entry is not None: + return entry.BaseTy + return None diff --git a/TransPyV/App/lib/core/Handles/HandlesExpr.py b/TransPyV/App/lib/core/Handles/HandlesExpr.py new file mode 100644 index 0000000..e2429b6 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesExpr.py @@ -0,0 +1,1525 @@ +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesImports as HandlesImports + + +# ============================================================ +# HandlesExpr - 表达式处理（Mixin 继承模式） +# +# 工具函数保留为模块级（供外部调用），ExprHandle.HandleValue 提供 trans 接口 +# ============================================================ + + +# ============================================================ +# 从 AST 节点提取函数名 +# ============================================================ +def get_func_name(func_node: ast.AST | t.CPtr) -> str: + """从函数节点提取函数名""" + if func_node is None: + return None + k: int = func_node.kind() + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(func_node) + return nm.id + elif k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(func_node) + return at.attr + return None + + +# ============================================================ +# 转义 LLVM IR 字符串字面量 +# ============================================================ +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, + src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: + """将字符串转换为 LLVM IR 字符串字面量 c"...\\00" """ + if src is None: + return None + slen: t.CSizeT = string.strlen(src) + alloc_size: t.CSizeT = 4 * slen + 8 + buf: t.CChar | t.CPtr = pool.alloc(alloc_size) + if buf is None: + return None + pos: t.CSizeT = 0 + buf[pos] = 'c' + pos += 1 + buf[pos] = '"' + pos += 1 + for i in range(slen): + ch: t.CChar = src[i] + if ch == '\0': + break + if ch == '\"' or ch == '\\' or ch < 0x20: + buf[pos] = '\\' + pos += 1 + hi: int = (ch >> 4) & 0xF + lo: int = ch & 0xF + if hi < 10: + buf[pos] = '0' + hi + else: + buf[pos] = 'A' + (hi - 10) + pos += 1 + if lo < 10: + buf[pos] = '0' + lo + else: + buf[pos] = 'A' + (lo - 10) + pos += 1 + else: + buf[pos] = ch + pos += 1 + buf[pos] = '\\' + pos += 1 + buf[pos] = '0' + pos += 1 + buf[pos] = '0' + pos += 1 + buf[pos] = '"' + pos += 1 + buf[pos] = '\0' + return buf + + +# ============================================================ +# 获取 LLVM 整数类型的位宽 +# ============================================================ +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: + """获取 LLVM 整数类型的位宽，非整数类型返回 0""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Int(bits): + return bits + case _: + return 0 + + +# ============================================================ +# 获取 LLVM 浮点类型的位宽 +# ============================================================ +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: + """获取 LLVM 浮点类型的位宽，非浮点类型返回 0""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Float(bits): + return bits + case _: + return 0 + + +# ============================================================ +# 检查类型是否为 Ptr（独立函数，避免嵌套 match 的编译器 BUG） +# +# 嵌套 match 的 REnum 检测在宿主编译器中有 BUG：当 match 嵌套在 +# 另一个 match 的 case 块中时，内层 match 走非 REnum 路径， +# 导致 case 列表为空。提取为独立函数可规避此问题。 +# ============================================================ +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: + """检查 ty 是否是 Ptr 类型，返回 1=是, 0=否""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Ptr(pointee): + return 1 + case _: + return 0 + + +# ============================================================ +# _deref_if_ptr_ptr - 如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用 +# +# 用于支持 X|t.CPtr 类型变量的属性访问： +# - 值类型变量 (cnt: Counter): alloca 类型是 Ptr(Struct) → 不解引用 +# - 指针类型变量 (r: Vec2|t.CPtr): alloca 类型是 Ptr(Ptr(Struct)) → load 解引用 +# +# 用 is_ptr_type 避免嵌套 match 的编译器 BUG +# ============================================================ +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, + obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: + """如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针""" + if obj_ptr is None or obj_ptr.Ty is None: + return obj_ptr + if is_ptr_type(obj_ptr.Ty) == 0: + return obj_ptr + pointee: llvmlite.LLVMType | t.CPtr = obj_ptr.Ty.Pointee + if pointee is None: + return obj_ptr + if is_ptr_type(pointee) == 0: + return obj_ptr + # obj_ptr 是 Ptr(Ptr(...))，load 解引用 + loaded: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, pointee, obj_ptr) + if loaded is None: + return obj_ptr + return loaded + + +# ============================================================ +# 类型强制转换：整数 sext/trunc，浮点 fpext/fptrunc，int↔float si2fp/fp2si +# ============================================================ +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, + val: llvmlite.Value | t.CPtr, + target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """将 val 强制转换为 target_ty + + 整数: sext（扩展）/ trunc（截断） + 浮点: fpext（扩展）/ fptrunc（截断） + int→float: sitofp + float→int: fptosi + """ + if val is None or target_ty is None: + return val + # 整数转换 + val_bits: int = get_llvm_type_bits(val.Ty) + target_bits: int = get_llvm_type_bits(target_ty) + if val_bits != 0 and target_bits != 0: + if val_bits == target_bits: + return val + if val_bits < target_bits: + return llvmlite.build_sext(builder, val, target_ty) + return llvmlite.build_trunc(builder, val, target_ty) + # 浮点转换 + val_fbits: int = get_llvm_float_bits(val.Ty) + target_fbits: int = get_llvm_float_bits(target_ty) + if val_fbits != 0 and target_fbits != 0: + if val_fbits == target_fbits: + return val + if val_fbits < target_fbits: + return llvmlite.build_fpext(builder, val, target_ty) + return llvmlite.build_fptrunc(builder, val, target_ty) + # int → float + if val_bits != 0 and target_fbits != 0: + return llvmlite.build_si2fp(builder, val, target_ty) + # float → int + if val_fbits != 0 and target_bits != 0: + return llvmlite.build_fp2si(builder, val, target_ty) + # 指针 → 非指针值: build_load 解引用 + # 适用于: 指针 → 整数 (如 i8* → i8), 指针 → 结构体值 + # 当构造器返回 Ptr(Struct) 但目标变量是 Struct 值类型时，需要 load + # 注意：使用独立函数 is_ptr_type 检查，避免嵌套 match 的编译器 BUG + if is_ptr_type(val.Ty) != 0 and is_ptr_type(target_ty) == 0: + return llvmlite.build_load(builder, target_ty, val) + # 指针 → 指针: bitcast (如 i8* → i8** 当目标是全局变量存储指针) + if is_ptr_type(val.Ty) != 0 and is_ptr_type(target_ty) != 0: + return llvmlite.build_bitcast(builder, val, target_ty) + return val + + +# ============================================================ +# 创建全局字符串常量 +# ============================================================ +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + str_val: str, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """创建全局字符串常量并返回 i8* bitcast""" + escaped: t.CChar | t.CPtr = escape_llvm_string(pool, str_val) + if escaped is None: + return None + + slen: t.CSizeT = string.strlen(str_val) + count: int = slen + 1 + + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + arr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Array(pool, i8_ty, count) + + # 使用模块级计数器（trans._label_counter）避免跨函数命名冲突 + str_idx: int = trans._label_counter + trans._label_counter += 1 + + # 字符串名加 SHA1 前缀，和函数导出规则一致，避免跨模块重名 + gv_name: t.CChar | t.CPtr = pool.alloc(48) + if gv_name is None: + return None + if trans.ModuleSha1 is not None: + viperlib.snprintf(gv_name, 48, ".str.%s.%d", trans.ModuleSha1, str_idx) + else: + viperlib.snprintf(gv_name, 48, ".str.%d", str_idx) + + gv: llvmlite.GlobalVariable | t.CPtr = llvmlite.new_global_variable(pool, gv_name, arr_ty) + if gv is None: + return None + llvmlite.module_add_global(mod, gv) + + # 去掉 private 链接，和函数一致：stub 中 declare，text 中 define + llvmlite.global_set_unnamed_addr(gv, 1) + llvmlite.global_set_constant(gv, 1) + gv.Initializer = escaped + + arr_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, arr_ty) + gv_ref_name: t.CChar | t.CPtr = pool.alloc(64) + if gv_ref_name is None: + return None + viperlib.snprintf(gv_ref_name, 64, "@%s", gv.Name) + gv_ref: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, arr_ptr_ty, gv_ref_name) + + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + return llvmlite.build_bitcast(builder, gv_ref, i8_ptr_ty) + + +# ============================================================ +# 翻译常量表达式 +# ============================================================ +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译常量（int/str/bool）""" + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn is None: + return None + if cn.const_kind == ast.CONST_INT: + # 超出 i32 范围则用 i64（避免常量创建时被截断） + iv: t.CInt64T = cn.int_val + if iv > 2147483647 or iv < -2147483648: + return llvmlite.const_int64(pool, iv) + return llvmlite.const_int32(pool, iv) + elif cn.const_kind == ast.CONST_FLOAT: + # 浮点常量默认创建为 double（64 位），赋值时由 coerce_to_type 自动 fptrunc + double_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Double(pool) + return llvmlite.ConstFloat(pool, double_ty, cn.float_val) + elif cn.const_kind == ast.CONST_STR: + return create_global_string(builder, pool, mod, cn.str_val, trans) + elif cn.const_kind == ast.CONST_BOOL: + if cn.int_val != 0: + return llvmlite.const_int32(pool, 1) + return llvmlite.const_int32(pool, 0) + elif cn.const_kind == ast.CONST_NONE: + # None → i8* null（空指针常量），用于 `p is None` / `p is not None` 比较 + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + return llvmlite.ConstNull(pool, i8_ptr_ty, "null") + return None + + +# ============================================================ +# _infer_elem_type_from_value — 从 LLVM Value 推断 list 元素类型名 +# +# 用于 list 字面量 [a, b, c] 的元素类型推断 +# ============================================================ +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, + value: llvmlite.Value | t.CPtr) -> str: + """从 LLVM Value 的类型推断 list 元素类型名""" + if value is None or value.Ty is None: + return "str" + ty: llvmlite.LLVMType | t.CPtr = value.Ty + bits: int = get_llvm_type_bits(ty) + if bits != 0: + if bits == 32: + return "int" + if bits == 64: + return "CSizeT" + if bits == 8: + return "CInt8T" + if bits == 16: + return "CInt16T" + return "int" + if is_ptr_type(ty) != 0: + return "str" + fbits: int = get_llvm_float_bits(ty) + if fbits != 0: + if fbits == 64: + return "CDouble" + if fbits == 32: + return "CFloat" + return "str" + + +# ============================================================ +# _find_pool_var — 查找上下文中的 pool 变量 +# +# 依次尝试 "pool", "_mbuddy", "mbuddy", "mb" +# ============================================================ +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """查找上下文中的 pool 变量，返回 alloca 或 None""" + if trans is None: + return None + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, "pool") + if alloca is not None: + return alloca + alloca = HandlesVar.lookup_var(trans.SymTab, "_mbuddy") + if alloca is not None: + return alloca + alloca = HandlesVar.lookup_var(trans.SymTab, "mbuddy") + if alloca is not None: + return alloca + alloca = HandlesVar.lookup_var(trans.SymTab, "mb") + if alloca is not None: + return alloca + return None + + +# ============================================================ +# translate_list_literal — 翻译 list 字面量 [a, b, c] +# +# 将 list 字面量翻译为 list[T](pool) 构造 + append 调用序列 +# +# Args: +# builder: IRBuilder +# pool: 编译器内存池（MemBuddy） +# mod: LLVMModule +# node: ast.List 节点 +# elem_type_name: 元素类型名（如 "str", "int"），None 表示从元素推断 +# trans: Translator 对象 +# +# Returns: +# list 对象指针（Ptr(list[T] struct)），None 失败 +# ============================================================ +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + elem_type_name: str, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列""" + if node is None or trans is None: + return None + + list_node: ast.List | t.CPtr = (ast.List | t.CPtr)(node) + if list_node is None: + return None + + elts: list[ast.AST | t.CPtr] | t.CPtr = list_node.elts + elts_count: t.CSizeT = 0 + if elts is not None: + elts_count = elts.__len__() + + # 推断元素类型 + inferred_type_name: str = elem_type_name + first_elem_val: llvmlite.Value | t.CPtr = None + if inferred_type_name is None: + inferred_type_name = "str" + if elts_count > 0: + first_elem_node: ast.AST | t.CPtr = elts.get(0) + first_elem_val = translate_value(builder, pool, mod, first_elem_node, None, 0, trans) + if first_elem_val is not None: + inferred_type_name = _infer_elem_type_from_value(pool, first_elem_val) + + # 查找 pool 变量 + pool_alloca: llvmlite.Value | t.CPtr = _find_pool_var(trans) + if pool_alloca is None: + HandlesType.fatal_error(node, "list literal requires pool variable in context") + return None + + # load pool 变量值（i8*） + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + pool_val: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, pool_alloca) + if pool_val is None: + return None + + # 特化 list[T] 类 + type_args: list[str] | t.CPtr = list[str](pool, 4) + type_args.append(inferred_type_name) + spec_name: str = HandlesClassDef._specialize_generic_class(trans, "list", type_args) + if spec_name is None: + HandlesType.fatal_error(node, "list literal generic specialization failed") + return None + + # 查找 struct entry + struct_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct(spec_name) + if struct_entry is None: + HandlesType.fatal_error(node, "list literal specialized class not registered") + return None + struct_ty: llvmlite.LLVMType | t.CPtr = struct_entry.Ty + if struct_ty is None: + return None + + # alloca 临时 list 空间 + tmp: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, struct_ty) + if tmp is None: + return None + + # 调用 __new__(tmp, pool_val) → storage_ptr + new_args: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if new_args is None: + return None + string.memset(new_args, 0, 8 * 32) + new_args[0] = t.CSizeT(pool_val) + storage_ptr: llvmlite.Value | t.CPtr = HandlesExprCall._call_method_on_ptr( + pool, builder, mod, spec_name, "__new__", tmp, new_args, 1, trans) + if storage_ptr is None: + storage_ptr = tmp + + # 调用 __before_init__(storage_ptr) + HandlesExprCall._call_method_on_ptr( + pool, builder, mod, spec_name, "__before_init__", storage_ptr, None, 0, trans) + + # 调用 __init__(storage_ptr, pool_val, elem_size=0) + init_args: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if init_args is None: + return None + string.memset(init_args, 0, 8 * 32) + init_args[0] = t.CSizeT(pool_val) + elem_size_val: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 0) + init_args[1] = t.CSizeT(elem_size_val) + HandlesExprCall._call_method_on_ptr( + pool, builder, mod, spec_name, "__init__", storage_ptr, init_args, 2, trans) + + # 对每个元素调用 append(storage_ptr, elem_val) + if elts_count > 0: + # 第一个元素可能已经翻译过 + if first_elem_val is not None: + append_args: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if append_args is not None: + string.memset(append_args, 0, 8 * 32) + append_args[0] = t.CSizeT(first_elem_val) + HandlesExprCall._call_method_on_ptr( + pool, builder, mod, spec_name, "append", storage_ptr, append_args, 1, trans) + + # 翻译剩余元素 + i: t.CSizeT = 1 + while i < elts_count: + elem_node_i: ast.AST | t.CPtr = elts.get(i) + elem_val_i: llvmlite.Value | t.CPtr = translate_value(builder, pool, mod, elem_node_i, None, 0, trans) + if elem_val_i is not None: + ap_args: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if ap_args is not None: + string.memset(ap_args, 0, 8 * 32) + ap_args[0] = t.CSizeT(elem_val_i) + HandlesExprCall._call_method_on_ptr( + pool, builder, mod, spec_name, "append", storage_ptr, ap_args, 1, trans) + i += 1 + + return storage_ptr + + +# ============================================================ +# 翻译值表达式（RHS 分派）— 模块级版本 +# ============================================================ +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + funcs_ptr: t.CPtr = None, + func_count: int = 0, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译值表达式为 LLVM Value""" + if node is None: + return None + k: int = node.kind() + if k == ast.ASTKind.Constant: + return translate_constant(builder, pool, mod, node, trans) + elif k == ast.ASTKind.Name: + return translate_name_value(builder, pool, node, trans) + elif k == ast.ASTKind.BinOp: + return HandlesExprOps.translate_binop(pool, builder, mod, node, trans) + elif k == ast.ASTKind.Call: + return HandlesExprCall.translate_call(pool, builder, mod, node, + funcs_ptr, func_count, trans) + elif k == ast.ASTKind.Compare: + return translate_compare(builder, pool, mod, node, trans) + elif k == ast.ASTKind.UnaryOp: + return translate_unaryop(builder, pool, mod, node, trans) + elif k == ast.ASTKind.BoolOp: + return translate_boolop(builder, pool, mod, node, + funcs_ptr, func_count, trans) + elif k == ast.ASTKind.Subscript: + return translate_subscript(builder, pool, mod, node, trans) + elif k == ast.ASTKind.Attribute: + return translate_attribute(builder, pool, mod, node, trans) + elif k == ast.ASTKind.List: + return translate_list_literal(builder, pool, mod, node, None, trans) + return None + + +# ============================================================ +# 翻译比较表达式 Compare(left, ops, comparators) +# ============================================================ +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译比较表达式，返回 i1 结果""" + cmp: ast.Compare | t.CPtr = (ast.Compare | t.CPtr)(node) + if cmp is None: + return None + + ops_list: list[t.CInt] | t.CPtr = (list[t.CInt] | t.CPtr)(cmp.ops) + if ops_list is None or ops_list.__len__() == 0: + return None + op: int = ops_list.get(0) + + comparators: list[ast.AST | t.CPtr] | t.CPtr = cmp.comparators + if comparators is None or comparators.__len__() == 0: + return None + rhs: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, comparators.get(0), None, 0, trans) + if rhs is None: + return None + + # === 比较运算符重载路径 1: lhs 是 Name 且对应结构体变量 === + # 对于值类型变量（如 cnt: Counter），用 alloca 指针尝试重载 + lhs_node: ast.AST | t.CPtr = cmp.left + if lhs_node is not None and lhs_node.kind() == ast.ASTKind.Name and trans is not None: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(lhs_node) + if nm is not None and nm.id is not None: + lhs_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, nm.id) + if lhs_alloca is not None: + ovl_result: llvmlite.Value | t.CPtr = HandlesExprOps.try_operator_overload( + pool, builder, mod, lhs_alloca, rhs, op, trans, 1) + if ovl_result is not None: + return ovl_result + + # 正常翻译 lhs + lhs: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, cmp.left, None, 0, trans) + if lhs is None: + return None + + # === 比较运算符重载路径 2: lhs 是 Ptr(Struct) === + # Is/IsNot 不在 _cmpop_to_dunder 映射中，会自动返回 None 回退原生比较 + ovl_result2: llvmlite.Value | t.CPtr = HandlesExprOps.try_operator_overload( + pool, builder, mod, lhs, rhs, op, trans, 1) + if ovl_result2 is not None: + return ovl_result2 + + predicate: int = llvmlite.ICMP_SLT + if op == ast.OpKind.Lt: + predicate = llvmlite.ICMP_SLT + elif op == ast.OpKind.Le: + predicate = llvmlite.ICMP_SLE + elif op == ast.OpKind.Gt: + predicate = llvmlite.ICMP_SGT + elif op == ast.OpKind.Ge: + predicate = llvmlite.ICMP_SGE + elif op == ast.OpKind.Eq: + predicate = llvmlite.ICMP_EQ + elif op == ast.OpKind.Ne: + predicate = llvmlite.ICMP_NE + elif op == ast.OpKind.Is: + # `x is y` → 指针/值相等比较 + predicate = llvmlite.ICMP_EQ + elif op == ast.OpKind.IsNot: + # `x is not y` → 指针/值不等比较 + predicate = llvmlite.ICMP_NE + + # 隐式类型转换：当一侧是指针(i8*)，另一侧是整数时， + # 从指针加载第一个字节，使两侧类型一致 + lhs_is_ptr: int = is_ptr_type(lhs.Ty) + rhs_is_ptr: int = is_ptr_type(rhs.Ty) + if lhs_is_ptr != 0 and rhs_is_ptr == 0: + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + lhs = llvmlite.build_load(builder, i8_ty, lhs) + elif rhs_is_ptr != 0 and lhs_is_ptr == 0: + i8_ty2: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + rhs = llvmlite.build_load(builder, i8_ty2, rhs) + + # 类型对齐：icmp 要求两边类型相同，将较窄的整数 sext 到较宽的类型 + lhs_bits: int = get_llvm_type_bits(lhs.Ty) + rhs_bits: int = get_llvm_type_bits(rhs.Ty) + if lhs_bits != 0 and rhs_bits != 0 and lhs_bits != rhs_bits: + if lhs_bits < rhs_bits: + lhs = llvmlite.build_sext(builder, lhs, rhs.Ty) + else: + rhs = llvmlite.build_sext(builder, rhs, lhs.Ty) + + return llvmlite.build_icmp(builder, predicate, lhs, rhs) + + +# ============================================================ +# 翻译一元运算表达式 UnaryOp(op, operand) +# ============================================================ +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译一元运算（-x, +x, not x, ~x）""" + uo: ast.UnaryOp | t.CPtr = (ast.UnaryOp | t.CPtr)(node) + if uo is None: + return None + + operand: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, uo.operand, None, 0, trans) + if operand is None: + return None + + if uo.op == ast.OpKind.USub: + # -x = 0 - x + zero: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + return llvmlite.build_sub(builder, zero, operand) + elif uo.op == ast.OpKind.UAdd: + # +x = x + return operand + elif uo.op == ast.OpKind.Not: + # not x = (x == 0)，返回 i1 + zero = llvmlite.const_int32(pool, 0) + return llvmlite.build_icmp(builder, llvmlite.ICMP_EQ, operand, zero) + elif uo.op == ast.OpKind.Invert: + # ~x = x ^ -1 + neg1: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, -1) + return llvmlite.build_xor(builder, operand, neg1) + return operand + + +# ============================================================ +# 翻译布尔运算 BoolOp(op, values) — 短路求值，返回 i1 +# +# and: a and b and c → 若 a 为假短路到 merge(返回 a 的 i1)，否则求 b， +# 最后一个值直接求值并跳 merge +# or: a or b or c → 若 a 为真短路到 merge(返回 a 的 i1)，否则求 b， +# 最后一个值直接求值并跳 merge +# merge 块用 phi 合并所有入边的 i1 +# ============================================================ +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + funcs_ptr: t.CPtr = None, + func_count: int = 0, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译布尔运算（and/or）的短路求值，返回 i1""" + bo: ast.BoolOp | t.CPtr = (ast.BoolOp | t.CPtr)(node) + if bo is None: + return None + + op: int = bo.op + values: list[ast.AST | t.CPtr] | t.CPtr = bo.values + if values is None: + return None + count: t.CSizeT = values.__len__() + if count == 0: + return None + if count == 1: + return translate_value(builder, pool, mod, values.get(0), + funcs_ptr, func_count, trans) + + func: llvmlite.Function | t.CPtr = builder.Func + if func is None: + return None + + # 创建 merge BB（用 builder.Counter 生成唯一标签名，position_at_end 的 + # move_to_end 保证 BB 文本顺序与控制流顺序一致，标签名用 counter 安全） + cnt: int = builder.Counter + builder.Counter = cnt + 1 + name_buf: t.CChar | t.CPtr = pool.alloc(32) + if name_buf is None: + return None + viperlib.snprintf(name_buf, 32, "bool.merge.%d", cnt) + merge_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # phi 入边链表头/尾 + phi_head: llvmlite.PhiIncoming | t.CPtr = None + phi_tail: llvmlite.PhiIncoming | t.CPtr = None + phi_count: int = count - 1 # 循环次数 + + # 对每个值（除最后一个）求值并短路 + for i in range(count - 1): + val_node: ast.AST | t.CPtr = values.get(i) + val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, val_node, + funcs_ptr, func_count, trans) + if val is None: + return None + + # 转为 i1（已经是 i1 的直接用，否则与 0 比较） + val_bits: int = get_llvm_type_bits(val.Ty) + val_i1: llvmlite.Value | t.CPtr = val + if val_bits != 1: + zero: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + val_i1 = llvmlite.build_icmp(builder, llvmlite.ICMP_NE, val, zero) + + # 创建 next BB（求值下一个值） + cnt = builder.Counter + builder.Counter = cnt + 1 + viperlib.snprintf(name_buf, 32, "bool.next.%d", cnt) + next_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # and: 短路假→merge，真→next；or: 短路真→merge，假→next + if op == ast.OpKind.And: + llvmlite.build_cond_br(builder, val_i1, next_bb, merge_bb) + else: + llvmlite.build_cond_br(builder, val_i1, merge_bb, next_bb) + + # 添加 phi 入边：(val_i1, cond_br 所在 BB) + cur_bb: llvmlite.BasicBlock | t.CPtr = builder.CurBlock + inc: llvmlite.PhiIncoming | t.CPtr = llvmlite.new_phi_incoming(pool, val_i1, cur_bb) + if phi_head is None: + phi_head = inc + phi_tail = inc + else: + phi_tail.Next = inc + phi_tail = inc + + # 定位到 next BB，求值下一个值 + llvmlite.position_at_end(builder, next_bb) + + # 最后一个值：求值后直接跳 merge + last_node: ast.AST | t.CPtr = values.get(count - 1) + last_val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, last_node, + funcs_ptr, func_count, trans) + if last_val is None: + return None + last_bits: int = get_llvm_type_bits(last_val.Ty) + last_i1: llvmlite.Value | t.CPtr = last_val + if last_bits != 1: + zero = llvmlite.const_int32(pool, 0) + last_i1 = llvmlite.build_icmp(builder, llvmlite.ICMP_NE, last_val, zero) + + llvmlite.build_br(builder, merge_bb) + + cur_bb = builder.CurBlock + inc = llvmlite.new_phi_incoming(pool, last_i1, cur_bb) + if phi_head is None: + phi_head = inc + else: + phi_tail.Next = inc + phi_count += 1 + + # 定位到 merge，创建 phi 合并所有入边 + llvmlite.position_at_end(builder, merge_bb) + i1_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int1(pool) + return llvmlite.build_phi(builder, i1_ty, phi_head, phi_count) + + +# ============================================================ +# 翻译变量引用（Name 节点）→ load +# ============================================================ +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译变量引用（Name 节点）→ load""" + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm is None: + return None + nm_id: str = nm.id + if nm_id is None: + return None + + # CDefine 编译期常量: 直接返回整数常量值（不生成运行时代码） + # NAME: t.CDefine = value 形式定义的常量在编译期已注册到全局表 + cdef_val: int = HandlesType.lookup_cdefine_constant(nm_id) + if cdef_val >= 0: + return llvmlite.const_int32(pool, cdef_val) + + # global 变量：从模块作用域查找 + if trans is not None: + if HT.is_global_name(trans, nm_id) != 0: + mod_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_module_var( + trans.SymTab, nm_id) + if mod_alloca is not None: + load_ty: llvmlite.LLVMType | t.CPtr = None + if mod_alloca.Ty is not None: + load_ty = mod_alloca.Ty.Pointee + if load_ty is None: + load_ty = llvmlite.Int32(pool) + return llvmlite.build_load(builder, load_ty, mod_alloca) + + # nonlocal 变量：通过闭包 env 访问 + if HT.is_nonlocal_name(trans, nm_id) != 0: + return HandlesNonlocal.load_nonlocal_var(trans, nm_id) + + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm_id) + if alloca is None: + return None + + load_ty: llvmlite.LLVMType | t.CPtr = None + if alloca.Ty is not None: + load_ty = alloca.Ty.Pointee + if load_ty is None: + load_ty = llvmlite.Int32(pool) + return llvmlite.build_load(builder, load_ty, alloca) + + +# ============================================================ +# ExprHandle - 表达式处理器（Mixin 继承模式） +# +# HandleValue 提供 trans 接口，委托到模块级 translate_value +# ============================================================ +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + """表达式处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # HandleValue - 翻译值表达式为 LLVM Value + # ============================================================ + def HandleValue(self, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: + """翻译值表达式，从 self.Trans 获取共享状态""" + return translate_value( + self.Trans._cur_builder, self.Trans.Pool, self.Trans.Module, + node, + self.Trans._funcs, self.Trans._func_count, self.Trans) + + +# ============================================================ +# NewExprHandle - 工厂函数 +# ============================================================ +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: + h: ExprHandle | t.CPtr = pool.alloc(ExprHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, ExprHandle.__sizeof__()) + h.Trans = trans + return h + + +# ============================================================ +# list_getitem_inline — 内联生成 list[T] __getitem__ 逻辑 +# +# list 结构布局 (6 字段 × 8 字节): +# __data__(0) __count__(8) __capacity__(16) __pool__(24) __elem_size__(32) __iter_index__(40) +# 返回元素地址 (i8*)，调用方根据元素类型 load 正确的值 +# ============================================================ +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + lm_obj: llvmlite.Value | t.CPtr, + idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: + """内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)""" + if builder is None or pool is None or lm_obj is None or idx_val is None: + return None + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + # 加载 __data__ (偏移 0, index 0) + g_idx0: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 0) + g_dpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, g_idx0) + if g_dpp is None: + return None + g_data: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, g_dpp) + # 加载 __elem_size__ (偏移 32, index 4) + g_idx4: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 4) + g_epp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, g_idx4) + if g_epp is None: + return None + g_esize: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, g_epp) + if g_data is None or g_esize is None: + return None + g_idx_i64: llvmlite.Value | t.CPtr = coerce_to_type(builder, idx_val, i64_ty) + if g_idx_i64 is None: + return None + g_offset: llvmlite.Value | t.CPtr = llvmlite.build_mul(builder, g_idx_i64, g_esize) + if g_offset is None: + return None + # 返回元素地址 (i8*)，不 load + return llvmlite.build_gep(builder, i8_ty, g_data, g_offset) + + +# ============================================================ +# _list_elem_type_from_name — 从 list 类型名提取元素 LLVM 类型 +# +# 类型名格式: .list[] +# 用 strstr 检查完整模式，避免子串提取 +# 支持: list[int]→i32, list[str]/list[bytes]→i8*, list[CInt]→i32 等 +# 默认: i32 +# ============================================================ +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, + struct_name: str) -> llvmlite.LLVMType | t.CPtr: + """从 list 类型名提取元素 LLVM 类型""" + if pool is None or struct_name is None: + return None + # 用 strstr 检查完整模式 + if string.strstr(struct_name, "list[int]") is not None: + return llvmlite.Int32(pool) + if string.strstr(struct_name, "list[CInt]") is not None: + return llvmlite.Int32(pool) + if string.strstr(struct_name, "list[CInt8T]") is not None: + return llvmlite.Int8(pool) + if string.strstr(struct_name, "list[CInt16T]") is not None: + return llvmlite.Int16(pool) + if string.strstr(struct_name, "list[CInt32T]") is not None: + return llvmlite.Int32(pool) + if string.strstr(struct_name, "list[CInt64T]") is not None: + return llvmlite.Int64(pool) + if string.strstr(struct_name, "list[CSizeT]") is not None: + return llvmlite.Int64(pool) + if string.strstr(struct_name, "list[CDouble]") is not None: + return llvmlite.Double(pool) + if string.strstr(struct_name, "list[CFloat]") is not None: + return llvmlite.Float(pool) + if string.strstr(struct_name, "list[CChar]") is not None: + return llvmlite.Int8(pool) + # str/bytes/CPtr/其他指针类型 → i8* + i8_ty_d: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + return llvmlite.Ptr(pool, i8_ty_d) + + +# ============================================================ +# list_setitem_inline — 内联生成 list[T] __setitem__ 逻辑 +# +# 直接存储 rhs_val 到元素地址 (coerce 为 i8* 后 store) +# ============================================================ +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + lm_obj: llvmlite.Value | t.CPtr, + idx_val: llvmlite.Value | t.CPtr, + rhs_val: llvmlite.Value | t.CPtr) -> int: + """内联生成 list __setitem__ 逻辑，返回 0""" + if builder is None or pool is None or lm_obj is None or idx_val is None or rhs_val is None: + return 0 + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + # 加载 __data__ (偏移 0, index 0) + s_idx0: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 0) + s_dpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, s_idx0) + if s_dpp is None: + return 0 + s_data: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, s_dpp) + # 加载 __elem_size__ (偏移 32, index 4) + s_idx4: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 4) + s_epp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, s_idx4) + if s_epp is None: + return 0 + s_esize: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, s_epp) + if s_data is None or s_esize is None: + return 0 + s_idx_i64: llvmlite.Value | t.CPtr = coerce_to_type(builder, idx_val, i64_ty) + if s_idx_i64 is None: + return 0 + s_offset: llvmlite.Value | t.CPtr = llvmlite.build_mul(builder, s_idx_i64, s_esize) + if s_offset is None: + return 0 + s_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, s_data, s_offset) + if s_addr is None: + return 0 + s_val_ptr: llvmlite.Value | t.CPtr = coerce_to_type(builder, rhs_val, i8_ptr_ty) + if s_val_ptr is None: + return 0 + llvmlite.build_store(builder, s_val_ptr, s_addr) + return 0 + + +# ============================================================ +# is_list_subscript — 检查 Subscript 节点是否是 list[T] 类型 +# +# 返回 list 对象指针 (Ptr(list[T])) 或 None +# ============================================================ +def is_list_subscript(node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None""" + if node is None or trans is None: + return None + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + if sub is None or sub.value is None: + return None + if sub.value.kind() != ast.ASTKind.Name: + return None + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub.value) + if nm is None or nm.id is None: + return None + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm.id) + if alloca is None or alloca.Ty is None: + return None + if is_ptr_type(alloca.Ty) == 0: + return None + pointee: llvmlite.LLVMType | t.CPtr = alloca.Ty.Pointee + if pointee is None: + return None + # 检查 pointee 是否是 Ptr(Struct) + inner_ty: llvmlite.LLVMType | t.CPtr = None + match pointee: + case llvmlite.LLVMType.Ptr(it): + inner_ty = it + case _: + return None + if inner_ty is None: + return None + # 检查 inner_ty 是否是 Struct 且名称包含 "list[" + struct_name: str = None + match inner_ty: + case llvmlite.LLVMType.Struct(_, _, sname): + struct_name = sname + case _: + return None + if struct_name is None: + return None + if string.strstr(struct_name, "list[") is None: + return None + # 是 list 类型: load 出 list 对象指针 + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + return llvmlite.build_load(builder, pointee, alloca) + + +# ============================================================ +# 翻译下标表达式 Subscript(value, slice, ctx) — ptr[i] / arr[i] +# +# 指针遍历: ptr 是指针变量，load 出指针值后 GEP + load +# 数组遍历: arr 是数组变量，用 alloca 指针做双索引 GEP [0, i] + load +# list[T]: 泛型类不注册 struct，subscript 走 __getitem__ 内联路径 +# ============================================================ +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译下标表达式，返回加载的元素值""" + if node is None or builder is None: + return None + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + if sub is None: + return None + + # 翻译索引 + idx_val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, sub.slice, None, 0, trans) + if idx_val is None: + return None + + # 如果 value 是 Name，尝试从 alloca 类型推断 + if sub.value is not None and sub.value.kind() == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub.value) + if nm.id is not None and trans is not None: + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm.id) + if alloca is not None and alloca.Ty is not None: + if is_ptr_type(alloca.Ty) != 0: + pointee: llvmlite.LLVMType | t.CPtr = alloca.Ty.Pointee + if pointee is not None: + # 单层 match（避免嵌套 match 的编译器 bug） + match pointee: + case llvmlite.LLVMType.Array(elem_ty, count): + # 数组遍历: getelementptr [N x elem_ty], ... , i32 0, i32 %idx + elem_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_array( + builder, pointee, elem_ty, alloca, idx_val) + if elem_ptr is not None: + return llvmlite.build_load(builder, elem_ty, elem_ptr) + return None + case llvmlite.LLVMType.Ptr(inner_ty): + # 检查 inner_ty 是否是 list[T] 类型（泛型类不注册 struct） + # list 的 subscript 应该走 __getitem__ 内联路径，而非指针遍历 + list_struct_name: str = None + match inner_ty: + case llvmlite.LLVMType.Struct(_, _, lsn): + list_struct_name = lsn + case _: + pass + if list_struct_name is not None: + if string.strstr(list_struct_name, "list[") is not None: + # list[T] 类型: 内联生成 __getitem__ 逻辑 + ptr_val_list: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, pointee, alloca) + if ptr_val_list is not None: + elem_addr_list: llvmlite.Value | t.CPtr = list_getitem_inline( + builder, pool, ptr_val_list, idx_val) + if elem_addr_list is not None: + # 根据类型名提取元素类型，bitcast 后 load + elem_ty_list: llvmlite.LLVMType | t.CPtr = _list_elem_type_from_name( + pool, list_struct_name) + if elem_ty_list is not None: + elem_ptr_ty_list: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr( + pool, elem_ty_list) + casted_addr_list: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + builder, elem_addr_list, elem_ptr_ty_list) + if casted_addr_list is not None: + return llvmlite.build_load( + builder, elem_ty_list, casted_addr_list) + return None + # 普通指针遍历: 先 load 指针值，再 GEP + ptr_val: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, pointee, alloca) + if ptr_val is not None: + elem_ptr2: llvmlite.Value | t.CPtr = llvmlite.build_gep( + builder, inner_ty, ptr_val, idx_val) + if elem_ptr2 is not None: + return llvmlite.build_load(builder, inner_ty, elem_ptr2) + return None + case _: + # 自定义类（结构体类型）: 转发到 __getitem__ 方法调用 + # 如 hashtable[key] → hashtable.__getitem__(key) + cls_nm_sub: str = HandlesStruct.get_class_name_by_type(pool, pointee) + if cls_nm_sub is not None: + obj_val_sub: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, pointee, alloca) + if obj_val_sub is not None: + arg_vals_sub: t.CSizeT | t.CPtr = pool.alloc(8) + if arg_vals_sub is not None: + arg_vals_sub[0] = t.CSizeT(idx_val) + ret_sub: llvmlite.Value | t.CPtr = HandlesExprCall._call_method_on_ptr( + pool, builder, mod, cls_nm_sub, "__getitem__", + obj_val_sub, arg_vals_sub, 1, trans) + if ret_sub is not None: + return ret_sub + pass + + # 通用路径：翻译 value 获取指针 + ptr_val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, sub.value, None, 0, trans) + if ptr_val is None or ptr_val.Ty is None: + return None + + # 检查是否是指针类型 + if is_ptr_type(ptr_val.Ty) != 0: + elem_ty2: llvmlite.LLVMType | t.CPtr = ptr_val.Ty.Pointee + if elem_ty2 is not None: + elem_ptr3: llvmlite.Value | t.CPtr = llvmlite.build_gep( + builder, elem_ty2, ptr_val, idx_val) + if elem_ptr3 is not None: + return llvmlite.build_load(builder, elem_ty2, elem_ptr3) + return None + + +# ============================================================ +# find_global_in_module - 在模块全局变量链表中按名称查找 +# +# 查找策略: +# 1. 精确匹配 (如 "_mbuddy" 匹配 @_mbuddy) +# 2. 后缀匹配 (如 "_mbuddy" 匹配 @"sha1._mbuddy") +# ============================================================ +def find_global_in_module(mod: llvmlite.LLVMModule | t.CPtr, + name: str) -> llvmlite.GlobalVariable | t.CPtr: + """在模块全局变量链表中按名称查找全局变量""" + if mod is None or name is None: + return None + name_len: t.CSizeT = string.strlen(name) + cur: llvmlite.GlobalVariable | t.CPtr = mod.GlobalHead + # 第一遍: 精确匹配 + while cur is not None: + if cur.Name is not None: + if string.strcmp(cur.Name, name) == 0: + return cur + cur = cur.Next + # 第二遍: 后缀匹配 (.name) + cur = mod.GlobalHead + while cur is not None: + if cur.Name is not None: + cur_len: t.CSizeT = string.strlen(cur.Name) + if cur_len > name_len + 1: + suffix_start: t.CSizeT = cur_len - name_len + if cur.Name[suffix_start - 1] == '.': + match: int = 1 + for i in range(name_len): + if cur.Name[suffix_start + i] != name[i]: + match = 0 + break + if match == 1: + return cur + cur = cur.Next + return None + + +# ============================================================ +# make_global_ref - 创建全局变量引用 (@name，类型为 ty*) +# +# 用于跨模块全局变量访问: stub 未注入时，翻译阶段生成 @name 引用， +# 链接时由 stub 提供 external global 声明。 +# ============================================================ +def make_global_ref(pool: memhub.MemBuddy | t.CPtr, + name: str, + ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """创建全局变量引用 (类型为 ty*，名称为 @name)""" + if name is None or ty is None: + return None + ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, ty) + ref_name: t.CChar | t.CPtr = pool.alloc(64) + if ref_name is None: + return None + viperlib.snprintf(ref_name, 64, "@%s", name) + return llvmlite.SSAValue(pool, ptr_ty, ref_name) + + +# ============================================================ +# _resolve_module_attribute_global - 解析模块属性访问的全局变量引用 +# +# 当 obj 是已导入模块名 (如 sys._mbuddy) 时: +# 1. 在当前模块全局变量链表中查找 (stub 已注入的情况) +# 2. 未找到则创建前向引用 (@attr_name, 类型为 ty 参数) +# +# 参数: +# ty_hint: 类型提示 (写路径用 rhs 类型, 读路径用 i8* 回退) +# 返回: Value 指针 (类型为 ty*)，未识别为模块属性返回 None +# ============================================================ +def _resolve_module_attribute_global(pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + trans: HT.Translator | t.CPtr, + at: ast.Attribute | t.CPtr, + ty_hint: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """解析模块属性访问，返回全局变量引用 (用于跨模块 global 访问)""" + if at is None or at.value is None or at.attr is None: + return None + if at.value.kind() != ast.ASTKind.Name: + return None + if trans is None or trans._imported_modules is None: + return None + mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if mod_nm.id is None: + return None + # 检查是否是已导入模块 + if HandlesImports.is_module_imported(trans._imported_modules, mod_nm.id) == 0: + return None + # 在当前模块查找全局变量 (stub 可能已注入) + gv: llvmlite.GlobalVariable | t.CPtr = find_global_in_module(mod, at.attr) + if gv is not None and gv.Ty is not None: + return make_global_ref(pool, at.attr, gv.Ty) + # 未找到: 创建前向引用 (stub 后续注入时提供声明) + use_ty: llvmlite.LLVMType | t.CPtr = ty_hint + if use_ty is None: + use_ty = llvmlite.Int8(pool) + return make_global_ref(pool, at.attr, use_ty) + + +# ============================================================ +# 翻译属性访问 Attribute(value, attr, ctx) — obj.field +# +# 优先级: +# 1. 枚举成员访问 (State.Idle → 常量值) +# 2. 模块属性访问 (sys._mbuddy → 加载全局变量) +# 3. 结构体字段访问 (obj.field → GEP + load) +# +# 对于 Name 类型的 obj，直接使用 alloca 指针（不 load 结构体） +# ============================================================ +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译属性访问，返回加载的字段值""" + if node is None or builder is None: + return None + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at is None or at.value is None or at.attr is None: + return None + + # 枚举成员访问: EnumName.MemberName → 常量值 + # 当 value 是 Name 且 Name.id 是已注册枚举时，查找成员并返回常量 + if at.value.kind() == ast.ASTKind.Name: + enum_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if enum_nm.id is not None: + if HandlesEnum.is_enum_class(enum_nm.id) == 1: + member: HandlesEnum.EnumMember | t.CPtr = HandlesEnum.lookup_enum_member( + enum_nm.id, at.attr) + if member is not None: + base_ty: llvmlite.LLVMType | t.CPtr = HandlesEnum.get_enum_base_type( + enum_nm.id) + if base_ty is not None: + # 生成数字字符串作为常量名（LLVM IR 要求常量输出数字值） + name_buf: t.CChar | t.CPtr = pool.alloc(32) + if name_buf is not None: + viperlib.snprintf(name_buf, 32, "%lld", member.Value) + return llvmlite.ConstInt(pool, base_ty, member.Value, name_buf) + return llvmlite.const_int32(pool, member.Value) + return None + + # 模块属性访问 (读路径): sys._mbuddy → 加载全局变量 @_mbuddy + # 当 obj 是已导入模块名且不是普通变量时，查找/创建全局变量引用并加载 + if at.value.kind() == ast.ASTKind.Name and trans is not None: + nm_ma: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm_ma.id is not None: + # 先检查是否是普通变量 (优先级高于模块属性) + ma_is_var: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, nm_ma.id) + if ma_is_var is None: + # 使用 i8* 作为类型提示 (模块级变量通常存储指针) + i8_ptr_hint: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + mod_gv_ref: llvmlite.Value | t.CPtr = _resolve_module_attribute_global( + pool, mod, trans, at, i8_ptr_hint) + if mod_gv_ref is not None and mod_gv_ref.Ty is not None: + load_ty_mod: llvmlite.LLVMType | t.CPtr = mod_gv_ref.Ty.Pointee + if load_ty_mod is not None: + return llvmlite.build_load(builder, load_ty_mod, mod_gv_ref) + + # 对于 Name 类型的 obj，直接查找 alloca（不 load 结构体） + obj_ptr: llvmlite.Value | t.CPtr = None + if at.value.kind() == ast.ASTKind.Name and trans is not None: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm.id is not None: + obj_ptr = HandlesVar.lookup_var(trans.SymTab, nm.id) + + # 非 Name 路径：翻译对象值（会 load） + if obj_ptr is None: + obj_ptr = translate_value(builder, pool, mod, at.value, None, 0, trans) + + if obj_ptr is None or obj_ptr.Ty is None: + return None + + # 如果 obj_ptr 是 Ptr(Ptr(Struct))（X|t.CPtr 变量的 alloca）， + # load 解引用获取 Ptr(Struct) + obj_ptr = _deref_if_ptr_ptr(builder, obj_ptr) + + # 查找结构体类型信息 + # obj_ptr 类型应该是 Ptr(Struct(...)) + match obj_ptr.Ty: + case llvmlite.LLVMType.Ptr(struct_ty): + # 查找字段索引和类型 + field_info: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct.lookup_field( + struct_ty, at.attr) + # 回退: 类型指针比较失败时，通过 AnnotClassName 按类名查找 + # 传递 SHA1 以区分跨模块同名类 + if field_info is None: + if at.value.kind() == ast.ASTKind.Name and trans is not None: + nm_fb: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm_fb.id is not None: + var_entry: HandlesVar.VarEntry | t.CPtr = HandlesVar.lookup_var_entry( + trans.SymTab, nm_fb.id) + if var_entry is not None and var_entry.AnnotClassName is not None: + cur_sha1: str = trans.ModuleSha1 + field_info = HandlesStruct.lookup_field_by_class( + var_entry.AnnotClassName, at.attr, cur_sha1) + if field_info is not None: + field_ty: llvmlite.LLVMType | t.CPtr = field_info.Ty + # 联合体：bitcast obj_ptr 到 field_ty* 后 load + if HandlesStruct.is_union_by_type(struct_ty) == 1: + field_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, field_ty) + casted: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + builder, obj_ptr, field_ptr_ty) + if casted is not None: + return llvmlite.build_load(builder, field_ty, casted) + return None + # 普通结构体：GEP + load + field_idx: int = field_info.Index + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, field_ty, obj_ptr, field_idx) + if field_ptr is not None: + return llvmlite.build_load(builder, field_ty, field_ptr) + return None + case _: + return None + +# ============================================================ +# 获取下标表达式的元素指针（不加载值，用于赋值） +# ============================================================ +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """获取下标表达式的元素指针（用于赋值 lhs）""" + if node is None or builder is None: + return None + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + if sub is None: + return None + + # 翻译索引 + idx_val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, sub.slice, None, 0, trans) + if idx_val is None: + return None + + # 如果 value 是 Name，尝试从 alloca 类型推断 + if sub.value is not None and sub.value.kind() == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub.value) + if nm.id is not None and trans is not None: + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm.id) + if alloca is not None and alloca.Ty is not None: + if is_ptr_type(alloca.Ty) != 0: + pointee: llvmlite.LLVMType | t.CPtr = alloca.Ty.Pointee + if pointee is not None: + # 单层 match（避免嵌套 match 的编译器 bug） + match pointee: + case llvmlite.LLVMType.Array(elem_ty, count): + # 数组遍历 + return llvmlite.build_gep_array( + builder, pointee, elem_ty, alloca, idx_val) + case llvmlite.LLVMType.Ptr(inner_ty): + # 指针遍历 + ptr_val: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, pointee, alloca) + if ptr_val is not None: + return llvmlite.build_gep( + builder, inner_ty, ptr_val, idx_val) + return None + case _: + pass + + # 通用路径 + ptr_val: llvmlite.Value | t.CPtr = translate_value( + builder, pool, mod, sub.value, None, 0, trans) + if ptr_val is None or ptr_val.Ty is None: + return None + if is_ptr_type(ptr_val.Ty) != 0: + elem_ty2: llvmlite.LLVMType | t.CPtr = ptr_val.Ty.Pointee + if elem_ty2 is not None: + return llvmlite.build_gep(builder, elem_ty2, ptr_val, idx_val) + return None + + +# ============================================================ +# 获取属性访问的字段指针（不加载值，用于赋值） +# 对于 Name 类型的 obj，直接使用 alloca 指针（不 load 结构体） +# ============================================================ +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """获取属性访问的字段指针（用于赋值 lhs）""" + if node is None or builder is None: + return None + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at is None or at.value is None or at.attr is None: + return None + + # 对于 Name 类型的 obj，直接查找 alloca（不 load 结构体） + obj_ptr: llvmlite.Value | t.CPtr = None + if at.value.kind() == ast.ASTKind.Name and trans is not None: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm.id is not None: + obj_ptr = HandlesVar.lookup_var(trans.SymTab, nm.id) + + # 模块属性访问 (写路径): sys._mbuddy = mb → store 到全局变量 @_mbuddy + # 当 obj 是已导入模块名且不是普通变量时，查找/创建全局变量引用 + if obj_ptr is None and at.value.kind() == ast.ASTKind.Name and trans is not None: + nm_wma: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm_wma.id is not None: + # 使用 i8* 作为类型提示 (模块级变量通常存储指针) + i8_ptr_w: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + wma_ref: llvmlite.Value | t.CPtr = _resolve_module_attribute_global( + pool, mod, trans, at, i8_ptr_w) + if wma_ref is not None: + return wma_ref + + # 非 Name 路径：翻译对象值（会 load） + if obj_ptr is None: + obj_ptr = translate_value(builder, pool, mod, at.value, None, 0, trans) + + if obj_ptr is None or obj_ptr.Ty is None: + return None + + # 如果 obj_ptr 是 Ptr(Ptr(Struct))（X|t.CPtr 变量的 alloca）， + # load 解引用获取 Ptr(Struct) + obj_ptr = _deref_if_ptr_ptr(builder, obj_ptr) + + match obj_ptr.Ty: + case llvmlite.LLVMType.Ptr(struct_ty): + field_info: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct.lookup_field( + struct_ty, at.attr) + # 回退: 类型指针比较失败时，通过 AnnotClassName 按类名查找 + # 传递 SHA1 以区分跨模块同名类 + if field_info is None: + if at.value.kind() == ast.ASTKind.Name and trans is not None: + nm_fb: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if nm_fb.id is not None: + var_entry: HandlesVar.VarEntry | t.CPtr = HandlesVar.lookup_var_entry( + trans.SymTab, nm_fb.id) + if var_entry is not None and var_entry.AnnotClassName is not None: + cur_sha1: str = None + if trans is not None: + cur_sha1 = trans.ModuleSha1 + field_info = HandlesStruct.lookup_field_by_class( + var_entry.AnnotClassName, at.attr, cur_sha1) + if field_info is not None: + # 联合体：bitcast obj_ptr 到 field_ty*（字段指针用于 store） + if HandlesStruct.is_union_by_type(struct_ty) == 1: + field_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, field_info.Ty) + return llvmlite.build_bitcast(builder, obj_ptr, field_ptr_ty) + # 普通结构体：GEP 获取字段指针 + return llvmlite.build_gep_struct( + builder, struct_ty, field_info.Ty, obj_ptr, field_info.Index) + return None + case _: + return None diff --git a/TransPyV/App/lib/core/Handles/HandlesExprCall.py b/TransPyV/App/lib/core/Handles/HandlesExprCall.py new file mode 100644 index 0000000..3e2ec2e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesExprCall.py @@ -0,0 +1,3972 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + + +# ============================================================ +# 全局模块 SHA1 映射（module_name → SHA1） +# +# 翻译时用于跨模块函数调用名混淆： +# string.strchr(...) → call @{sha1_of_string}.strchr(...) +# +# 通过 set_module_sha1_map 设置，_lookup_module_sha1 查询 +# ============================================================ +_g_sha1_arr: bytes = None +_g_mod_arr: bytes = None +_g_sha1_count: int = 0 + + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: + """设置全局模块 SHA1 映射""" + global _g_sha1_arr + global _g_mod_arr + global _g_sha1_count + _g_sha1_arr = sha1_arr + _g_mod_arr = mod_arr + _g_sha1_count = count + return 0 + + +def _lookup_module_sha1(mod_name: str) -> str: + """查找模块名对应的 SHA1，找不到返回 None""" + if mod_name is None: + return None + if _g_sha1_arr is None or _g_mod_arr is None: + return None + if _g_sha1_count <= 0: + return None + for i in range(_g_sha1_count): + idx: t.CSizeT = t.CSizeT(i) * 64 + if string.strcmp(_g_mod_arr + idx, mod_name) == 0: + sidx: t.CSizeT = t.CSizeT(i) * 17 + return _g_sha1_arr + sidx + return None + + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: + """后缀匹配查找模块 SHA1，如 mod_suffix="__types" 匹配 "llvmlite.__types" + + 用于 from .X import Y 的相对导入：impf.module 是 ".X"（带前导点）， + 去掉前导点后为 "X"，SHA1 映射表中键是 "pkg.X"，需要后缀匹配。 + """ + if mod_suffix is None: + return None + if _g_sha1_arr is None or _g_mod_arr is None: + return None + if _g_sha1_count <= 0: + return None + suffix_len: t.CSizeT = string.strlen(mod_suffix) + if suffix_len == 0: + return None + for i in range(_g_sha1_count): + idx: t.CSizeT = t.CSizeT(i) * 64 + mod_name: str = _g_mod_arr + idx + mod_len: t.CSizeT = string.strlen(mod_name) + if mod_len < suffix_len: + continue + # 精确匹配 + if mod_len == suffix_len: + if string.strcmp(mod_name, mod_suffix) == 0: + sidx: t.CSizeT = t.CSizeT(i) * 17 + return _g_sha1_arr + sidx + # 后缀匹配：mod_name 以 "." + mod_suffix 结尾 + if mod_len > suffix_len: + if mod_name[mod_len - suffix_len - 1] == '.': + cmp_start: str = mod_name + mod_len - suffix_len + if string.strcmp(cmp_start, mod_suffix) == 0: + sidx2: t.CSizeT = t.CSizeT(i) * 17 + return _g_sha1_arr + sidx2 + return None + + +# ============================================================ +# 全局 CExport 函数表（module_sha1 + func_name） +# +# t.CExport 函数定义时不加 SHA1 前缀（如 @strlen）， +# 跨模块调用时需要查表确认目标函数是否是 CExport 函数。 +# 若是，使用裸名调用（@strlen）；否则使用 @{sha1}.func。 +# 必须包含模块 SHA1 记录，避免同名跨模块函数冲突 +# （一个模块有 CExport 的 strlen→@strlen，另一个没有→@{sha1}.strlen）。 +# ============================================================ +_g_cexport_sha1_arr: bytes = None +_g_cexport_func_arr: bytes = None +_g_cexport_count: int = 0 +_CEXPORT_MAX: t.CDefine = 256 + + +def register_cexport_func(sha1: str, func_name: str) -> int: + """注册 CExport 函数到全局表（sha1+func_name 二元组）""" + global _g_cexport_sha1_arr + global _g_cexport_func_arr + global _g_cexport_count + if sha1 is None or func_name is None: + return 0 + if _g_cexport_sha1_arr is None: + _g_cexport_sha1_arr = stdlib.malloc(_CEXPORT_MAX * 17) + _g_cexport_func_arr = stdlib.malloc(_CEXPORT_MAX * 64) + if _g_cexport_sha1_arr is None or _g_cexport_func_arr is None: + return 0 + string.memset(_g_cexport_sha1_arr, 0, _CEXPORT_MAX * 17) + string.memset(_g_cexport_func_arr, 0, _CEXPORT_MAX * 64) + if _g_cexport_count < _CEXPORT_MAX: + idx: t.CSizeT = t.CSizeT(_g_cexport_count) * 17 + string.strcpy(_g_cexport_sha1_arr + idx, sha1) + idx2: t.CSizeT = t.CSizeT(_g_cexport_count) * 64 + string.strcpy(_g_cexport_func_arr + idx2, func_name) + _g_cexport_count += 1 + return 1 + return 0 + + +def is_cexport_func(sha1: str, func_name: str) -> int: + """检查 (sha1, func_name) 是否是 CExport 函数""" + if sha1 is None or func_name is None: + return 0 + if _g_cexport_sha1_arr is None: + return 0 + for i in range(_g_cexport_count): + idx: t.CSizeT = t.CSizeT(i) * 17 + if string.strcmp(_g_cexport_sha1_arr + idx, sha1) == 0: + idx2: t.CSizeT = t.CSizeT(i) * 64 + if string.strcmp(_g_cexport_func_arr + idx2, func_name) == 0: + return 1 + return 0 + + +# ============================================================ +# _lookup_mod_sha1 - 从 from_imports 查找类所属模块的 SHA1 +# +# lookup_from_import 返回指向 from_imports 内部的指针，不是 null 终止 +# （如 "namespace_defs Gadget:namespace_defs ..."）。此函数手动截断 +# 模块名，复制到新缓冲区，再查找 SHA1（精确+后缀匹配）。 +# ============================================================ +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, + from_imports: str, class_name: str, + allow_star_fallback: int = 1) -> str: + """从 from_imports 查找类所属模块的 SHA1，返回 None=未找到 + + allow_star_fallback: 传 0 禁用 star import 回退（用于明确模块名/函数名查找） + """ + mod_name_raw: str = HandlesImports.lookup_from_import( + from_imports, class_name, allow_star_fallback) + if mod_name_raw is None: + return None + # 手动找到模块名结束位置（空格或 null），复制到新缓冲区 + mod_len: t.CSizeT = 0 + while mod_name_raw[mod_len] != ' ' and mod_name_raw[mod_len] != '\0': + mod_len += 1 + mod_name_buf: str = pool.alloc(mod_len + 1) + if mod_name_buf is None: + return None + for mi in range(mod_len): + mod_name_buf[mi] = mod_name_raw[mi] + mod_name_buf[mod_len] = '\0' + # 去掉相对导入的前导点 + base_mod: str = mod_name_buf + while base_mod[0] == '.': + base_mod = base_mod + 1 + sha1: str = _lookup_module_sha1(base_mod) + if sha1 is None: + sha1 = _lookup_module_sha1_suffix(base_mod) + return sha1 + + +# ============================================================ +# HandlesExprCall - 函数调用表达式处理（Mixin 继承模式） +# +# 工具函数和 FuncEntry 保留为模块级，ExprCallHandle 提供 trans 接口 +# ============================================================ + + +# ============================================================ +# 在模块函数链表中按名称查找函数 +# ============================================================ +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, + name: str) -> llvmlite.Function | t.CPtr: + """在模块的函数链表中按名称查找函数 + + 查找策略： + 1. 精确匹配（如 "strlen" 匹配 @strlen，有 t.CExport 的函数） + 2. 后缀匹配（如 "atoi" 匹配 @"83210a72e829a086.atoi"，无 t.CExport 的函数） + + 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制 + （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。 + """ + if mod is None or name is None: + return None + name_len: t.CSizeT = string.strlen(name) + cur: llvmlite.Function | t.CPtr = mod.FuncHead + # 第一遍：精确匹配 + while cur is not None: + cur_name: t.CChar | t.CPtr = llvmlite.function_get_name(cur) + if cur_name is not None: + if string.strcmp(cur_name, name) == 0: + return cur + cur = llvmlite.function_get_next(cur) + # 第二遍：后缀匹配（.name） + # 不跳过 declare：方法 stub 也是 declare，需要通过后缀匹配找到 + # 调用处（translate_call 等）负责判断 IsDeclared 并决定使用裸名还是 sha1 前缀名 + cur = mod.FuncHead + while cur is not None: + cur_name2: t.CChar | t.CPtr = llvmlite.function_get_name(cur) + if cur_name2 is not None: + cur_len: t.CSizeT = string.strlen(cur_name2) + if cur_len > name_len + 1: + suffix_start: t.CSizeT = cur_len - name_len + if cur_name2[suffix_start - 1] == '.': + match: int = 1 + for i in range(name_len): + if cur_name2[suffix_start + i] != name[i]: + match = 0 + break + if match == 1: + return cur + cur = llvmlite.function_get_next(cur) + return None + + +# ============================================================ +# _infer_external_func_ret_ty - 根据函数名推断外部函数返回类型 +# +# stubs 在翻译后才注入 IR，翻译时模块中无 stub 函数声明， +# 因此对 includes 库函数（如 string.strchr 返回 i8*）需根据 +# 函数名推断返回类型，避免指针被截断为 i32。 +# +# 覆盖 includes/string.py 所有导出函数： +# i8*: strcpy, strncpy, strchr, strrchr, strstr, memset, memset32, memcpy, memmove +# i64: strlen, atoll +# double: atof +# i8: samestr +# i32（默认）: strcmp, strncmp, memcmp, strspn, atoi, split +# ============================================================ +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, + func_name: str) -> llvmlite.LLVMType | t.CPtr: + """根据函数名推断外部 includes 函数的返回类型""" + if pool is None or func_name is None: + return llvmlite.Int32(pool) + + # 返回 i8* 的函数（指针返回值，截断会导致错误） + if func_name == "strchr": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "strrchr": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "strstr": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "strcpy": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "strncpy": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "memset": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "memset32": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "memcpy": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "memmove": + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + + # 返回 i64 的函数 + if func_name == "strlen": + return llvmlite.Int64(pool) + if func_name == "atoll": + return llvmlite.Int64(pool) + + # 返回 double 的函数 + if func_name == "atof": + return llvmlite.Double(pool) + + # 返回 i8 的函数 + if func_name == "samestr": + return llvmlite.Int8(pool) + + # Win32 API 返回指针（VOIDPTR/HANDLE）的函数 + # stubs 翻译时未注入，需推断返回 i8* 避免指针被截断为 i32 导致解引用崩溃 + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + if func_name == "VirtualAlloc": + return i8_ptr_ty + if func_name == "HeapAlloc": + return i8_ptr_ty + if func_name == "HeapReAlloc": + return i8_ptr_ty + if func_name == "HeapCreate": + return i8_ptr_ty + if func_name == "GetProcessHeap": + return i8_ptr_ty + if func_name == "GlobalAlloc": + return i8_ptr_ty + if func_name == "GlobalFree": + return i8_ptr_ty + if func_name == "GlobalLock": + return i8_ptr_ty + if func_name == "LocalAlloc": + return i8_ptr_ty + if func_name == "LocalFree": + return i8_ptr_ty + if func_name == "GetProcAddress": + return i8_ptr_ty + if func_name == "GetModuleHandleA": + return i8_ptr_ty + if func_name == "GetModuleHandleW": + return i8_ptr_ty + if func_name == "LoadLibraryA": + return i8_ptr_ty + if func_name == "LoadLibraryW": + return i8_ptr_ty + + # Win32 API 返回 i64 (SIZE_T) 的函数 + if func_name == "VirtualQuery": + return llvmlite.Int64(pool) + if func_name == "HeapSize": + return llvmlite.Int64(pool) + if func_name == "HeapCompact": + return llvmlite.Int64(pool) + if func_name == "GlobalSize": + return llvmlite.Int64(pool) + + # 默认 i32 + return llvmlite.Int32(pool) + + +# ============================================================ +# _get_external_func_param_ty - 获取外部函数第 idx 个参数的期望类型 +# +# stubs 翻译时未注入，需推断参数类型以提升字面量（如 i32→i64）， +# 避免 declare(i64) 与 call(i32) 类型不匹配。 +# +# 注意：stubs 翻译时 ULONG(t.CUnsignedLong) 在 LLVM IR 中被翻译为 i64 +# （因为 LLVM 中 long 是 64 位），因此 Win32 API 中所有 ULONG/DWORD/SIZE_T +# 参数在 declare 中都是 i64。call 时必须匹配，否则类型不匹配。 +# +# 策略：对 Win32 API 函数的所有非指针参数（字面量）都提升为 i64， +# 以匹配 declare 中 ULONG 被翻译为 i64 的情况。 +# +# 指针参数（VOIDPTR/HANDLE）不在此处理，由调用点的 inttoptr 处理。 +# ============================================================ +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, + func_name: str, + param_idx: int) -> llvmlite.LLVMType | t.CPtr: + """获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型 + + 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中 + ULONG/SIZE_T 被统一翻译为 i64 的情况。 + """ + if pool is None or func_name is None: + return None + + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + + # Win32 API 函数列表：所有非指针参数在 declare 中都是 i64 + # （因为 ULONG/DWORD/SIZE_T 在 LLVM stubs 翻译时都被映射为 i64） + win32_apis: list[str] | t.CPtr = [ + "VirtualAlloc", "VirtualFree", "VirtualProtect", "VirtualQuery", + "VirtualLock", "VirtualUnlock", + "HeapCreate", "HeapDestroy", "HeapAlloc", "HeapReAlloc", + "HeapFree", "HeapSize", "HeapValidate", "HeapCompact", + "GlobalAlloc", "GlobalFree", "GlobalLock", "GlobalUnlock", "GlobalSize", + "LocalAlloc", "LocalFree", + "GetLastError", "SetLastError", + "Sleep", "SleepEx", + "GetTickCount", "GetProcessHeap" + ] + + for api_name in win32_apis: + if api_name == func_name: + return i64_ty + + return None + + +# ============================================================ +# _emit_llvm_memcpy_intrinsic — 将 memcpy 调用映射到 @llvm.memcpy 内联函数 +# +# 背景: string.memcpy 是 t.CExport 函数，跨模块调用生成 @memcpy 裸名调用。 +# 但在 monomorphization 场景下，目标模块的 deps.txt 可能不含 string， +# 导致 @memcpy 声明缺失，链接器报 undefined value。 +# +# 解决: 将 memcpy 调用映射到 LLVM 内联函数 @llvm.memcpy， +# 该内联函数在当前模块内声明，无需跨模块依赖。 +# +# @llvm.memcpy 签名: void (i8* dst, i8* src, i64 len, i1 isvolatile) +# C memcpy 签名: i8* (i8* dst, i8* src, i64 len) — 返回 dst +# +# Args: +# args_head: 参数链表头（dst, src, num），通过 Value.Next 链接 +# arg_count: 参数数量（应为 3） +# +# Returns: +# dst 值（匹配 C memcpy 语义: 返回 dest 指针） +# ============================================================ +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + dst_val: llvmlite.Value | t.CPtr, + src_val: llvmlite.Value | t.CPtr, + num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: + """将 memcpy 调用映射到 @llvm.memcpy 内联函数 + + 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的 + Next 字段可能被其他操作污染，导致链表断裂）。 + """ + if pool is None or builder is None or mod is None: + return None + if dst_val is None or src_val is None or num_val is None: + return None + + # 在函数顶部声明 void_ty，避免后续遮蔽 + void_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Void(pool) + + # 声明 @llvm.memcpy（如未声明） + memcpy_func: llvmlite.Function | t.CPtr = find_func_in_module(mod, "llvm.memcpy") + if memcpy_func is None: + memcpy_func = llvmlite.create_declare(pool, mod, "llvm.memcpy", void_ty) + if memcpy_func is not None: + i8_ptr_ty_d: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + i64_ty_d: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + i1_ty_d: llvmlite.LLVMType | t.CPtr = llvmlite.Int1(pool) + llvmlite.add_param(pool, memcpy_func, i8_ptr_ty_d, "dst") + llvmlite.add_param(pool, memcpy_func, i8_ptr_ty_d, "src") + llvmlite.add_param(pool, memcpy_func, i64_ty_d, "len") + llvmlite.add_param(pool, memcpy_func, i1_ty_d, "isvolatile") + + # 类型转换: dst, src → i8*; num → i64 + # 注意: 使用 value_get_ty 访问器获取 Ty，避免本地 Value 类型字段索引偏移 + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + + # dst → i8* + dst_coerced: llvmlite.Value | t.CPtr = dst_val + dst_ty: llvmlite.LLVMType | t.CPtr = llvmlite.value_get_ty(dst_val) + if HandlesExpr.is_ptr_type(dst_ty) != 0: + dst_coerced = llvmlite.build_bitcast(builder, dst_val, i8_ptr_ty) + else: + dst_bits: int = HandlesExpr.get_llvm_type_bits(dst_ty) + if dst_bits != 0: + if dst_bits < 64: + dst_coerced = llvmlite.build_zext(builder, dst_val, i64_ty) + dst_coerced = llvmlite.build_inttoptr(builder, dst_coerced, i8_ptr_ty) + + # src → i8* + src_coerced: llvmlite.Value | t.CPtr = src_val + src_ty: llvmlite.LLVMType | t.CPtr = llvmlite.value_get_ty(src_val) + if HandlesExpr.is_ptr_type(src_ty) != 0: + src_coerced = llvmlite.build_bitcast(builder, src_val, i8_ptr_ty) + else: + src_bits: int = HandlesExpr.get_llvm_type_bits(src_ty) + if src_bits != 0: + if src_bits < 64: + src_coerced = llvmlite.build_zext(builder, src_val, i64_ty) + src_coerced = llvmlite.build_inttoptr(builder, src_coerced, i8_ptr_ty) + + # num → i64 + num_coerced: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type(builder, num_val, i64_ty) + + if dst_coerced is None or src_coerced is None or num_coerced is None: + return None + + # 添加 i1 false 参数（isvolatile）— llvm.memcpy 需要 4 个参数 + false_val: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, llvmlite.Int1(pool), 0, "false") + + # 使用 includes 中的访问器函数构建参数链表，避免本地 Value 类型字段索引偏移 + # （f26cd4c78b4d554c.Value 缺少 GSListNode 的 Next 字段，直接访问 .Next 会修改 .Ty） + llvmlite.value_set_next(dst_coerced, src_coerced) + llvmlite.value_set_next(src_coerced, num_coerced) + if false_val is not None: + llvmlite.value_set_next(num_coerced, false_val) + llvmlite.value_set_next(false_val, None) + else: + llvmlite.value_set_next(num_coerced, None) + + # 使用 build_call 发射 call 指令（内部用 includes Value 类型访问字段，索引正确） + # _ll_name_needs_quote 已对 llvm. 前缀豁免引号 + result: llvmlite.Value | t.CPtr = llvmlite.build_call(builder, "llvm.memcpy", dst_coerced, 4, void_ty, 0) + + # 返回 dst（匹配 C memcpy 语义: 返回 dest 指针） + return dst_coerced + + +# ============================================================ +# c.Deref(ptr) — 解引用指针，load i8（字符遍历用例） +# +# 生成: %N = load i8, i8* %ptr +# 返回 i8 值 +# ============================================================ +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """c.Deref(ptr) → load i8 from ptr""" + if cl is None: + return None + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 1: + return None + + ptr_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, cargs.get(0), None, 0, trans) + if ptr_val is None: + return None + + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + return llvmlite.build_load(builder, i8_ty, ptr_val) + + +# ============================================================ +# c.DerefAs(ptr, val) — 解引用写入，*ptr = val +# +# 生成: %pc = bitcast i8* %ptr to * +# store %val, * %pc +# ============================================================ +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）""" + if cl is None: + return None + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 2: + return None + + ptr_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, cargs.get(0), None, 0, trans) + val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, cargs.get(1), None, 0, trans) + if ptr_val is None or val is None: + return None + + # bitcast ptr 到 val.Ty* 再 store + val_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, val.Ty) + casted_ptr: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, ptr_val, val_ptr_ty) + if casted_ptr is None: + return None + llvmlite.build_store(builder, val, casted_ptr) + return val + + +# ============================================================ +# c.Addr(var) — 取变量地址，返回 alloca 指针（不 load） +# +# 用于获取栈变量的可写地址，配合 c.DerefAs 使用 +# ============================================================ +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """c.Addr(var) → 返回变量的 alloca 指针（不 load）""" + if cl is None: + return None + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 1: + return None + + arg0: ast.AST | t.CPtr = cargs.get(0) + if arg0 is None or arg0.kind() != ast.ASTKind.Name: + return None + + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(arg0) + if nm.id is None: + return None + + return HandlesVar.lookup_var(trans.SymTab, nm.id) + + +# ============================================================ +# c.Load(a, b) — 指针间复制: *a = *b +# +# 加载 b 指向的值，存储到 a 指向的地址。 +# 生成: %pc_src = bitcast i8* %src to * +# %loaded = load , * %pc_src +# %pc_dst = bitcast i8* %dst to * +# store %loaded, * %pc_dst +# ============================================================ +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）""" + if cl is None: + return None + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 2: + return None + + dst_ptr: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, cargs.get(0), None, 0, trans) + src_ptr: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, cargs.get(1), None, 0, trans) + if dst_ptr is None or src_ptr is None: + return None + + # 确定 pointee 类型（优先用 src 的 pointee，回退到 dst 的 pointee） + pointee: llvmlite.LLVMType | t.CPtr = None + if src_ptr.Ty is not None: + pointee = src_ptr.Ty.Pointee + if pointee is None and dst_ptr.Ty is not None: + pointee = dst_ptr.Ty.Pointee + if pointee is None: + return None + + # bitcast src 到 pointee* 并 load + target_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, pointee) + src_cast: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, src_ptr, target_ptr_ty) + if src_cast is None: + return None + loaded: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, pointee, src_cast) + if loaded is None: + return None + + # bitcast dst 到 pointee* 并 store + dst_cast: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, dst_ptr, target_ptr_ty) + if dst_cast is None: + return None + llvmlite.build_store(builder, loaded, dst_cast) + return loaded + + +# ============================================================ +# c.Asm — 内联汇编支持 +# +# 生成: %N = call asm sideeffect "", ""() +# 或: call void asm sideeffect "", ""() +# +# 操作数按出现顺序编号 $0, $1, ...（用户需按输出在前写 f-string） +# ============================================================ + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + + +# ============================================================ +# AsmOperand - 内联汇编操作数条目 +# ============================================================ +@t.NoVTable +class AsmOperand: + """内联汇编操作数条目""" + Value: llvmlite.Value | t.CPtr # 输入值或输出 alloca + Constraint: str # 约束字符串 (如 "r", "=r") + IsOutput: int # 1=输出, 0=输入 + + +# ============================================================ +# _asm_append_cstr — 将 C 字符串 src 追加到 dst 末尾 +# ============================================================ +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, + src: t.CChar | t.CPtr): + """将 C 字符串 src 追加到 dst 末尾""" + if dst is None or src is None: + return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: + return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' + + +# ============================================================ +# _asm_escape_for_ir — 将汇编文本转义为 LLVM IR 字符串内容 +# +# \n → \0A, " → \22, \ → \5C +# ============================================================ +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, + src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: + """转义汇编文本用于 LLVM IR 字符串字面量""" + if src is None: + return None + slen: t.CSizeT = string.strlen(src) + buf: t.CChar | t.CPtr = pool.alloc(slen * 4 + 8) + if buf is None: + return None + pos: t.CSizeT = 0 + for ch in src: + if ch == '\n': + buf[pos] = '\\' + pos += 1 + buf[pos] = '0' + pos += 1 + buf[pos] = 'A' + pos += 1 + elif ch == '"': + buf[pos] = '\\' + pos += 1 + buf[pos] = '2' + pos += 1 + buf[pos] = '2' + pos += 1 + elif ch == '\\': + buf[pos] = '\\' + pos += 1 + buf[pos] = '5' + pos += 1 + buf[pos] = 'C' + pos += 1 + else: + buf[pos] = ch + pos += 1 + buf[pos] = '\0' + return buf + + +# ============================================================ +# _resolve_register — 将 ASM_DESCR 属性名映射到约束字符串 +# ============================================================ +def _resolve_register(attr_name: str) -> str: + """将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值""" + if attr_name is None: + return "" + + # Clobber 破坏描述符 + if attr_name == "CLOBBER_MEMORY": return "memory" + if attr_name == "CLOBBER_CC": return "cc" + if attr_name == "CLOBBER_RAX": return "rax" + if attr_name == "CLOBBER_RBX": return "rbx" + if attr_name == "CLOBBER_RCX": return "rcx" + if attr_name == "CLOBBER_RDX": return "rdx" + if attr_name == "CLOBBER_RSI": return "rsi" + if attr_name == "CLOBBER_RDI": return "rdi" + if attr_name == "CLOBBER_RBP": return "rbp" + if attr_name == "CLOBBER_RSP": return "rsp" + if attr_name == "CLOBBER_R8": return "r8" + if attr_name == "CLOBBER_R9": return "r9" + if attr_name == "CLOBBER_R10": return "r10" + if attr_name == "CLOBBER_R11": return "r11" + if attr_name == "CLOBBER_R12": return "r12" + if attr_name == "CLOBBER_R13": return "r13" + if attr_name == "CLOBBER_R14": return "r14" + if attr_name == "CLOBBER_R15": return "r15" + if attr_name == "CLOBBER_EAX": return "eax" + if attr_name == "CLOBBER_EBX": return "ebx" + if attr_name == "CLOBBER_ECX": return "ecx" + if attr_name == "CLOBBER_EDX": return "edx" + if attr_name == "CLOBBER_ESI": return "esi" + if attr_name == "CLOBBER_EDI": return "edi" + if attr_name == "CLOBBER_DX": return "dx" + if attr_name == "CLOBBER_AX": return "ax" + if attr_name == "CLOBBER_CX": return "cx" + if attr_name == "CLOBBER_SI": return "si" + if attr_name == "CLOBBER_DI": return "di" + + # Output 输出约束 + if attr_name == "OUTPUT_REG": return "=r" + if attr_name == "OUTPUT_MEM": return "=m" + if attr_name == "OUTPUT_EAX": return "=a" + if attr_name == "OUTPUT_EBX": return "=b" + if attr_name == "OUTPUT_ECX": return "=c" + if attr_name == "OUTPUT_EDX": return "=d" + if attr_name == "OUTPUT_ESI": return "=S" + if attr_name == "OUTPUT_EDI": return "=D" + if attr_name == "OUTPUT_RAX": return "=a" + if attr_name == "OUTPUT_RBX": return "=b" + if attr_name == "OUTPUT_RCX": return "=c" + if attr_name == "OUTPUT_RDX": return "=d" + if attr_name == "OUTPUT_RSI": return "=S" + if attr_name == "OUTPUT_RDI": return "=D" + + # Input/Register 输入约束 + if attr_name == "REG_ANY": return "r" + if attr_name == "REG_EAX": return "a" + if attr_name == "REG_EBX": return "b" + if attr_name == "REG_ECX": return "c" + if attr_name == "REG_EDX": return "d" + if attr_name == "REG_ESI": return "S" + if attr_name == "REG_EDI": return "D" + if attr_name == "REG_RAX": return "a" + if attr_name == "REG_RBX": return "b" + if attr_name == "REG_RCX": return "c" + if attr_name == "REG_RDX": return "d" + if attr_name == "REG_RSI": return "S" + if attr_name == "REG_RDI": return "D" + + if attr_name == "INPUT_REG": return "r" + if attr_name == "INPUT_MEM": return "m" + if attr_name == "INPUT_EAX": return "a" + if attr_name == "INPUT_EBX": return "b" + if attr_name == "INPUT_ECX": return "c" + if attr_name == "INPUT_EDX": return "d" + if attr_name == "INPUT_ESI": return "S" + if attr_name == "INPUT_EDI": return "D" + if attr_name == "INPUT_RAX": return "a" + if attr_name == "INPUT_RBX": return "b" + if attr_name == "INPUT_RCX": return "c" + if attr_name == "INPUT_RDX": return "d" + if attr_name == "INPUT_RSI": return "S" + if attr_name == "INPUT_RDI": return "D" + + return "" + + +# ============================================================ +# _resolve_asm_descr — 递归解析约束表达式 +# +# 支持: +# t.ASM_DESCR.XXX → _resolve_register(XXX) +# expr | expr (BinOp BitOr) → 拼接 +# "string" (Constant) → 直接返回 +# ============================================================ +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr) -> str: + """递归解析约束表达式，返回约束字符串""" + if node is None: + return "" + + k: int = node.kind() + + # BinOp with BitOr: 拼接左右 + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if bop.op == ast.OpKind.BitOr: + left_str: str = _resolve_asm_descr(pool, bop.left) + right_str: str = _resolve_asm_descr(pool, bop.right) + llen: t.CSizeT = string.strlen(left_str) + rlen: t.CSizeT = string.strlen(right_str) + buf: str = pool.alloc(llen + rlen + 1) + if buf is None: + return "" + string.strcpy(buf, left_str) + for i in range(rlen): + buf[llen + i] = right_str[i] + buf[llen + rlen] = '\0' + return buf + return "" + + # Attribute: t.ASM_DESCR.XXX 或 t.XXX + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + # t.ASM_DESCR.XXX + if at.value is not None and at.value.kind() == ast.ASTKind.Attribute: + inner: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(at.value) + if inner.value is not None and inner.value.kind() == ast.ASTKind.Name: + mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(inner.value) + if mod_nm.id is not None and string.strcmp(mod_nm.id, "t") == 0: + if inner.attr is not None and string.strcmp(inner.attr, "ASM_DESCR") == 0: + return _resolve_register(at.attr) + # t.XXX 简化形式 + if at.value is not None and at.value.kind() == ast.ASTKind.Name: + mod_nm2: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if mod_nm2.id is not None and string.strcmp(mod_nm2.id, "t") == 0: + return _resolve_register(at.attr) + return "" + + # Constant: 字符串常量 + if k == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn.const_kind == ast.CONST_STR: + return cn.str_val + return "" + + return "" + + +# ============================================================ +# _asm_get_operand — 获取操作数数组中第 idx 个元素的指针 +# ============================================================ +def _asm_get_operand(base: AsmOperand | t.CPtr, + idx: int) -> AsmOperand | t.CPtr: + """获取操作数数组中第 idx 个元素的指针""" + entry_size: t.CSizeT = AsmOperand.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(base) + idx * entry_size + return (AsmOperand | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + + +# ============================================================ +# _asm_is_c_call — 检查 Call 节点是否是 c.XXX 调用 +# +# 返回 attr_name（如 "AsmInp"/"AsmOut"），不是则返回 None +# ============================================================ +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: + """检查 Call 节点是否是 c.XXX 调用，返回属性名或 None""" + if node is None or node.func is None: + return None + if node.func.kind() != ast.ASTKind.Attribute: + return None + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node.func) + if at.value is None or at.value.kind() != ast.ASTKind.Name: + return None + mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if mod_nm.id is None or string.strcmp(mod_nm.id, "c") != 0: + return None + return at.attr + + +# ============================================================ +# _asm_add_operand — 添加操作数到数组 +# +# 返回 1 成功, 0 失败（数组已满） +# ============================================================ +def _asm_add_operand(operands: AsmOperand | t.CPtr, + count: int, + val: llvmlite.Value | t.CPtr, + constraint: str, + is_output: int) -> int: + """添加操作数到数组，返回新计数（失败返回原计数）""" + if count >= ASM_MAX_OPERANDS: + return count + op: AsmOperand | t.CPtr = _asm_get_operand(operands, count) + op.Value = val + op.Constraint = constraint + op.IsOutput = is_output + return count + 1 + + +# ============================================================ +# translate_c_asm — 翻译 c.Asm 内联汇编 +# +# 语法: +# c.Asm("nop", [t.ASM_DESCR.CLOBBER_MEMORY]) +# c.Asm(f"mov rax, {c.AsmInp(val, t.ASM_DESCR.REG_ANY)}", op=[...]) +# c.Asm(f"...", out=[c.AsmOut(var, t.ASM_DESCR.OUTPUT_REG)], op=[...]) +# ============================================================ +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）""" + if cl is None: + return None + + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 1: + return None + + # 分配操作数数组 + operand_size: t.CSizeT = AsmOperand.__sizeof__() + operands: AsmOperand | t.CPtr = pool.alloc(operand_size * ASM_MAX_OPERANDS) + if operands is None: + return None + string.memset(operands, 0, operand_size * ASM_MAX_OPERANDS) + + # 分配 clobber 名数组 + clobbers: str | t.CPtr = pool.alloc(8 * ASM_MAX_OPERANDS) + if clobbers is None: + return None + string.memset(clobbers, 0, 8 * ASM_MAX_OPERANDS) + + # 分配 asm 文本缓冲区 + asm_text: t.CChar | t.CPtr = pool.alloc(ASM_TEXT_BUF_SIZE) + if asm_text is None: + return None + asm_text[0] = '\0' + + operand_count: int = 0 + clobber_count: int = 0 + fv_count: int = 0 + + # 分配 fv 索引数组（记录每个 FormattedValue 对应的操作数原始索引） + fv_orig_indices: int | t.CPtr = pool.alloc(4 * ASM_MAX_OPERANDS) + if fv_orig_indices is None: + return None + string.memset(fv_orig_indices, 0, 4 * ASM_MAX_OPERANDS) + + # ============================================================ + # 1. 解析汇编模板（JoinedStr 或 Constant）— 第一遍：收集操作数 + # 不生成 asm 文本（占位符 $N 需在重排后才能确定编号） + # ============================================================ + tmpl_node: ast.AST | t.CPtr = cargs.get(0) + if tmpl_node is not None: + tk: int = tmpl_node.kind() + + if tk == ast.ASTKind.JoinedStr: + js: ast.JoinedStr | t.CPtr = (ast.JoinedStr | t.CPtr)(tmpl_node) + jvalues: list[ast.AST | t.CPtr] | t.CPtr = js.values + if jvalues is not None: + jvlen: t.CSizeT = jvalues.__len__() + for vi in range(jvlen): + part: ast.AST | t.CPtr = jvalues.get(vi) + if part is None: + continue + pk: int = part.kind() + + if pk == ast.ASTKind.FormattedValue: + fv: ast.FormattedValue | t.CPtr = (ast.FormattedValue | t.CPtr)(part) + fexpr: ast.AST | t.CPtr = fv.value + if fexpr is None or fexpr.kind() != ast.ASTKind.Call: + continue + fcall: ast.Call | t.CPtr = (ast.Call | t.CPtr)(fexpr) + attr_nm: str = _asm_is_c_call(fcall) + if attr_nm is None: + continue + + is_out: int = 0 + if string.strcmp(attr_nm, "AsmOut") == 0: + is_out = 1 + elif string.strcmp(attr_nm, "AsmInp") == 0: + is_out = 0 + else: + continue + + fargs: list[ast.AST | t.CPtr] | t.CPtr = fcall.args + if fargs is None or fargs.__len__() < 1: + continue + + val_node: ast.AST | t.CPtr = fargs.get(0) + cons_str: str = "" + if fargs.__len__() >= 2: + cons_str = _resolve_asm_descr(pool, fargs.get(1)) + + if is_out != 0: + # 输出：获取变量 alloca + if val_node is None or val_node.kind() != ast.ASTKind.Name: + continue + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(val_node) + if nm.id is None: + continue + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm.id) + if alloca is None: + continue + operand_count = _asm_add_operand( + operands, operand_count, alloca, cons_str, 1) + else: + # 输入：翻译值 + ival: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, val_node, None, 0, trans) + if ival is None: + continue + operand_count = _asm_add_operand( + operands, operand_count, ival, cons_str, 0) + + # 记录 fv 的原始操作数索引 + if fv_count < ASM_MAX_OPERANDS: + fv_orig_indices[fv_count] = operand_count - 1 + fv_count += 1 + + # ============================================================ + # 2. 解析关键字参数 out=/inp=/op=/clobber= + # ============================================================ + keywords: list[ast.AST | t.CPtr] | t.CPtr = cl.keywords + if keywords is not None: + kwlen: t.CSizeT = keywords.__len__() + for ki in range(kwlen): + kw: ast.Keyword | t.CPtr = (ast.Keyword | t.CPtr)(keywords.get(ki)) + if kw is None or kw.arg is None or kw.value is None: + continue + kwk: int = kw.value.kind() + + if string.strcmp(kw.arg, "out") == 0: + # 输出操作数列表 + if kwk != ast.ASTKind.List: + continue + out_lst: ast.List | t.CPtr = (ast.List | t.CPtr)(kw.value) + out_elts: list[ast.AST | t.CPtr] | t.CPtr = out_lst.elts + if out_elts is None: + continue + oelen: t.CSizeT = out_elts.__len__() + for oi in range(oelen): + oelt: ast.AST | t.CPtr = out_elts.get(oi) + if oelt is None or oelt.kind() != ast.ASTKind.Call: + continue + ocall: ast.Call | t.CPtr = (ast.Call | t.CPtr)(oelt) + oattr: str = _asm_is_c_call(ocall) + if oattr is None or string.strcmp(oattr, "AsmOut") != 0: + continue + oargs: list[ast.AST | t.CPtr] | t.CPtr = ocall.args + if oargs is None or oargs.__len__() < 1: + continue + oval_node: ast.AST | t.CPtr = oargs.get(0) + ocons: str = "" + if oargs.__len__() >= 2: + ocons = _resolve_asm_descr(pool, oargs.get(1)) + if oval_node is None or oval_node.kind() != ast.ASTKind.Name: + continue + onm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(oval_node) + if onm.id is None: + continue + oalloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, onm.id) + if oalloca is None: + continue + operand_count = _asm_add_operand( + operands, operand_count, oalloca, ocons, 1) + + elif string.strcmp(kw.arg, "inp") == 0 or string.strcmp(kw.arg, "inputs") == 0: + # 输入操作数列表 + if kwk != ast.ASTKind.List: + continue + in_lst: ast.List | t.CPtr = (ast.List | t.CPtr)(kw.value) + in_elts: list[ast.AST | t.CPtr] | t.CPtr = in_lst.elts + if in_elts is None: + continue + ielen: t.CSizeT = in_elts.__len__() + for ii in range(ielen): + ielt: ast.AST | t.CPtr = in_elts.get(ii) + if ielt is None or ielt.kind() != ast.ASTKind.Call: + continue + icall: ast.Call | t.CPtr = (ast.Call | t.CPtr)(ielt) + iattr: str = _asm_is_c_call(icall) + if iattr is None or string.strcmp(iattr, "AsmInp") != 0: + continue + iargs: list[ast.AST | t.CPtr] | t.CPtr = icall.args + if iargs is None or iargs.__len__() < 1: + continue + ival_node: ast.AST | t.CPtr = iargs.get(0) + icons: str = "" + if iargs.__len__() >= 2: + icons = _resolve_asm_descr(pool, iargs.get(1)) + ival2: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, ival_node, None, 0, trans) + if ival2 is None: + continue + operand_count = _asm_add_operand( + operands, operand_count, ival2, icons, 0) + + elif string.strcmp(kw.arg, "op") == 0 or string.strcmp(kw.arg, "clobber") == 0: + # 破坏列表 + if kwk != ast.ASTKind.List: + continue + cl_lst: ast.List | t.CPtr = (ast.List | t.CPtr)(kw.value) + cl_elts: list[ast.AST | t.CPtr] | t.CPtr = cl_lst.elts + if cl_elts is None: + continue + clen: t.CSizeT = cl_elts.__len__() + for ci in range(clen): + celt: ast.AST | t.CPtr = cl_elts.get(ci) + if celt is None: + continue + cstr: str = _resolve_asm_descr(pool, celt) + if cstr is None or cstr[0] == '\0': + continue + if clobber_count < ASM_MAX_OPERANDS: + cl_addr: t.CUInt64T = t.CUInt64T(clobbers) + clobber_count * 8 + cl_ptr: str | t.CPtr = (str | t.CPtr)(t.CVoid(cl_addr, t.CPtr)) + cl_ptr[0] = cstr + clobber_count += 1 + + # ============================================================ + # 3. 解析第二个位置参数作为破坏列表 + # ============================================================ + if cargs.__len__() >= 2: + pos_node: ast.AST | t.CPtr = cargs.get(1) + if pos_node is not None and pos_node.kind() == ast.ASTKind.List: + pos_lst: ast.List | t.CPtr = (ast.List | t.CPtr)(pos_node) + pos_elts: list[ast.AST | t.CPtr] | t.CPtr = pos_lst.elts + if pos_elts is not None: + plen: t.CSizeT = pos_elts.__len__() + for pi in range(plen): + pelt: ast.AST | t.CPtr = pos_elts.get(pi) + if pelt is None: + continue + pstr: str = _resolve_asm_descr(pool, pelt) + if pstr is None or pstr[0] == '\0': + continue + if clobber_count < ASM_MAX_OPERANDS: + pcl_addr: t.CUInt64T = t.CUInt64T(clobbers) + clobber_count * 8 + pcl_ptr: str | t.CPtr = (str | t.CPtr)(t.CVoid(pcl_addr, t.CPtr)) + pcl_ptr[0] = pstr + clobber_count += 1 + + # ============================================================ + # 4. 统计输出/输入数量，确定返回类型 + # ============================================================ + output_count: int = 0 + input_count: int = 0 + first_output_idx: int = -1 + for oi2 in range(operand_count): + op2: AsmOperand | t.CPtr = _asm_get_operand(operands, oi2) + if op2.IsOutput != 0: + if first_output_idx < 0: + first_output_idx = oi2 + output_count += 1 + else: + input_count += 1 + + # 确定返回类型 + ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Void(pool) + if output_count > 0 and first_output_idx >= 0: + out_op: AsmOperand | t.CPtr = _asm_get_operand(operands, first_output_idx) + if out_op.Value is not None and out_op.Value.Ty is not None: + # alloca 的类型是指针，Pointee 是目标类型 + if out_op.Value.Ty.Pointee is not None: + ret_ty = out_op.Value.Ty.Pointee + else: + ret_ty = llvmlite.Int32(pool) + + # 判断 void — 直接基于 output_count 判断，避免 match 类型匹配问题 + is_void: int = 1 + if output_count > 0: + is_void = 0 + + # ============================================================ + # 4.5 重排操作数：输出在前，输入在后（LLVM IR 要求） + # + # new_order[new_idx] = orig_idx — 新位置对应原索引 + # orig_to_new[orig_idx] = new_idx — 原索引映射到新位置（用于占位符编号） + # ============================================================ + new_order: int | t.CPtr = pool.alloc(4 * (operand_count + 1)) + orig_to_new: int | t.CPtr = pool.alloc(4 * (operand_count + 1)) + if new_order is None or orig_to_new is None: + return None + + new_idx: int = 0 + # 先放输出 + for oi3 in range(operand_count): + op3: AsmOperand | t.CPtr = _asm_get_operand(operands, oi3) + if op3.IsOutput != 0: + new_order[new_idx] = oi3 + new_idx += 1 + # 再放输入 + for oi4 in range(operand_count): + op4: AsmOperand | t.CPtr = _asm_get_operand(operands, oi4) + if op4.IsOutput == 0: + new_order[new_idx] = oi4 + new_idx += 1 + # 建立映射 orig -> new + for ni2 in range(operand_count): + orig_to_new[new_order[ni2]] = ni2 + + # ============================================================ + # 4.6 生成汇编文本（第二遍遍历 f-string，用重排后的索引生成占位符） + # 项目约定使用 Intel 语法，通过 IR 的 inteldialect 关键字启用 + # ============================================================ + if tmpl_node is not None: + tk2: int = tmpl_node.kind() + if tk2 == ast.ASTKind.JoinedStr: + js2: ast.JoinedStr | t.CPtr = (ast.JoinedStr | t.CPtr)(tmpl_node) + jvalues2: list[ast.AST | t.CPtr] | t.CPtr = js2.values + if jvalues2 is not None: + jvlen2: t.CSizeT = jvalues2.__len__() + fv_fi: int = 0 + for vi2 in range(jvlen2): + part2: ast.AST | t.CPtr = jvalues2.get(vi2) + if part2 is None: + continue + pk2: int = part2.kind() + if pk2 == ast.ASTKind.Constant: + cn3: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(part2) + if cn3.const_kind == ast.CONST_STR and cn3.str_val is not None: + _asm_append_cstr(asm_text, ASM_TEXT_BUF_SIZE, cn3.str_val) + elif pk2 == ast.ASTKind.FormattedValue: + if fv_fi < fv_count: + orig_idx2: int = fv_orig_indices[fv_fi] + new_idx2: int = orig_to_new[orig_idx2] + ph_buf2: t.CChar | t.CPtr = pool.alloc(16) + if ph_buf2 is not None: + viperlib.snprintf(ph_buf2, 16, "$%d", new_idx2) + _asm_append_cstr(asm_text, ASM_TEXT_BUF_SIZE, ph_buf2) + fv_fi += 1 + elif tk2 == ast.ASTKind.Constant: + cn4: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(tmpl_node) + if cn4.const_kind == ast.CONST_STR and cn4.str_val is not None: + _asm_append_cstr(asm_text, ASM_TEXT_BUF_SIZE, cn4.str_val) + + # ============================================================ + # 5. 构建约束字符串（按重排后的顺序：输出在前，输入在后） + # ============================================================ + const_buf: t.CChar | t.CPtr = pool.alloc(ASM_CONST_BUF_SIZE) + if const_buf is None: + return None + const_buf[0] = '\0' + + first_const: int = 1 + for ci2 in range(operand_count): + orig_i: int = new_order[ci2] + cop: AsmOperand | t.CPtr = _asm_get_operand(operands, orig_i) + cons_val: str = cop.Constraint + if cons_val is None: + cons_val = "r" + + if first_const == 0: + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, ",") + first_const = 0 + + if cop.IsOutput != 0: + # 输出约束：确保以 = 开头 + if cons_val[0] != '=': + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, "=") + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, cons_val) + else: + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, cons_val) + + # 追加 clobber 约束 ~{name} + for cidx in range(clobber_count): + cl_addr2: t.CUInt64T = t.CUInt64T(clobbers) + cidx * 8 + cname: str = (str | t.CPtr)(t.CVoid(cl_addr2, t.CPtr))[0] + if cname is None or cname[0] == '\0': + continue + if first_const == 0: + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, ",") + first_const = 0 + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, "~{") + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, cname) + _asm_append_cstr(const_buf, ASM_CONST_BUF_SIZE, "}") + + # ============================================================ + # 6. 构建参数列表文本（仅输入操作数） + # ============================================================ + args_buf: t.CChar | t.CPtr = pool.alloc(ASM_ARGS_BUF_SIZE) + if args_buf is None: + return None + args_buf[0] = '\0' + + first_arg: int = 1 + for ai3 in range(operand_count): + orig_i2: int = new_order[ai3] + aop: AsmOperand | t.CPtr = _asm_get_operand(operands, orig_i2) + if aop.IsOutput != 0: + continue + aval: llvmlite.Value | t.CPtr = aop.Value + if aval is None or aval.Ty is None or aval.Name is None: + continue + + if first_arg == 0: + _asm_append_cstr(args_buf, ASM_ARGS_BUF_SIZE, ", ") + first_arg = 0 + + # 类型文本 + ty_buf: t.CChar | t.CPtr = pool.alloc(64) + if ty_buf is not None: + ty_buf[0] = '\0' + llvmlite.TypePrint(ty_buf, 64, aval.Ty, pool) + _asm_append_cstr(args_buf, ASM_ARGS_BUF_SIZE, ty_buf) + _asm_append_cstr(args_buf, ASM_ARGS_BUF_SIZE, " ") + _asm_append_cstr(args_buf, ASM_ARGS_BUF_SIZE, aval.Name) + + # ============================================================ + # 7. 转义汇编文本 + # ============================================================ + escaped_asm: t.CChar | t.CPtr = _asm_escape_for_ir(pool, asm_text) + if escaped_asm is None: + return None + + # ============================================================ + # 8. 生成 IR 行并发射 + # ============================================================ + line: t.CChar | t.CPtr = pool.alloc(ASM_LINE_BUF_SIZE) + if line is None: + return None + line[0] = '\0' + + ssa_name: t.CChar | t.CPtr = None + if is_void == 0: + ssa_name = pool.alloc(16) + if ssa_name is not None: + viperlib.snprintf(ssa_name, 16, "%%%d", builder.Counter) + builder.Counter += 1 + + # 获取返回类型文本 + ret_ty_buf: t.CChar | t.CPtr = pool.alloc(64) + if ret_ty_buf is not None: + ret_ty_buf[0] = '\0' + llvmlite.TypePrint(ret_ty_buf, 64, ret_ty, pool) + + if is_void == 1: + viperlib.snprintf(line, ASM_LINE_BUF_SIZE, + "call %s asm sideeffect inteldialect \"%s\", \"%s\"(%s)", + ret_ty_buf, escaped_asm, const_buf, args_buf) + else: + viperlib.snprintf(line, ASM_LINE_BUF_SIZE, + "%s = call %s asm sideeffect inteldialect \"%s\", \"%s\"(%s)", + ssa_name, ret_ty_buf, escaped_asm, const_buf, args_buf) + + # 发射 IR 到当前块 + llvmlite.block_append_text(pool, builder.CurBlock, line) + + # ============================================================ + # 9. 处理输出：存储结果到输出变量 + # ============================================================ + if is_void == 0 and first_output_idx >= 0: + result_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, ret_ty, ssa_name) + out_op2: AsmOperand | t.CPtr = _asm_get_operand(operands, first_output_idx) + if out_op2.Value is not None: + llvmlite.build_store(builder, result_val, out_op2.Value) + return result_val + + return None + + +# ============================================================ +# c.LLVMIR — 内联 LLVM IR 支持 +# +# 语法: +# c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) +# c.LLVMIR(f"%{c.LOut(result)} = add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) +# +# c.LInp(expr) - 输入操作数，翻译后替换为对应的 LLVM 值 +# c.LOut(expr) - 输出操作数，翻译后结果存储到该变量 +# +# 支持的指令: add/sub/mul/sdiv/udiv/srem/urem/and/or/xor/shl/lshr/ashr +# fadd/fsub/fmul/fdiv +# icmp eq/ne/sgt/sge/slt/sle/ugt/uge/ult/ule +# fcmp oeq/one/olt/ogt/ole/oge +# zext/sext/trunc/bitcast/ptrtoint/inttoptr +# load/store +# ============================================================ + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + + +# ============================================================ +# _llvmir_get_op — 获取第 i 个 AsmOperand 槽位 +# ============================================================ +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: + """获取第 i 个操作数槽位""" + entry_size: t.CSizeT = AsmOperand.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(buf) + i * entry_size + return (AsmOperand | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + + +# ============================================================ +# _llvmir_find_op_marker — 在字符串中查找 %__OP 并返回编号 +# +# 返回 (位置, 编号)，未找到返回 (None, 0) +# ============================================================ +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: + """查找 %__OP 标记，返回标记开始位置或 None""" + if s is None: + return None + return string.strstr(s, "%__OP") + + +# ============================================================ +# _llvmir_parse_op_num — 从 %__OPN__ 中解析编号 N +# +# 输入: 指向 "%__OP" 的指针 +# 返回: 操作数编号 +# ============================================================ +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: + """从 %__OPN__ 标记中解析编号 N""" + if marker is None: + return -1 + # 跳过 "%__OP" (5 字符) + num_start: t.CChar | t.CPtr = marker + 5 + if num_start is None: + return -1 + # 解析数字直到 "__" + num: int = 0 + i: t.CSizeT = 0 + ch: t.CChar = num_start[0] + while '0' <= ch <= '9': + num = num * 10 + (ch - '0') + i += 1 + ch = num_start[i] + return num + + +# ============================================================ +# _llvmir_match_icmp_pred — 将预测字符串转为 ICMP 常量 +# ============================================================ +def _llvmir_match_icmp_pred(pred: str) -> int: + """将 icmp 预测字符串转为常量""" + if pred is None: + return 0 + if string.strcmp(pred, "eq") == 0: + return 0 + if string.strcmp(pred, "ne") == 0: + return 1 + if string.strcmp(pred, "sgt") == 0: + return 2 + if string.strcmp(pred, "sge") == 0: + return 3 + if string.strcmp(pred, "slt") == 0: + return 4 + if string.strcmp(pred, "sle") == 0: + return 5 + if string.strcmp(pred, "ugt") == 0: + return 6 + if string.strcmp(pred, "uge") == 0: + return 7 + if string.strcmp(pred, "ult") == 0: + return 8 + if string.strcmp(pred, "ule") == 0: + return 9 + return 0 + + +# ============================================================ +# _llvmir_match_fcmp_pred — 将预测字符串转为 FCMP 常量 +# ============================================================ +def _llvmir_match_fcmp_pred(pred: str) -> int: + """将 fcmp 预测字符串转为常量""" + if pred is None: + return 0 + if string.strcmp(pred, "oeq") == 0: + return 0 + if string.strcmp(pred, "ogt") == 0: + return 1 + if string.strcmp(pred, "oge") == 0: + return 2 + if string.strcmp(pred, "olt") == 0: + return 3 + if string.strcmp(pred, "ole") == 0: + return 4 + if string.strcmp(pred, "one") == 0: + return 5 + if string.strcmp(pred, "ord") == 0: + return 6 + if string.strcmp(pred, "uno") == 0: + return 7 + if string.strcmp(pred, "ueq") == 0: + return 8 + if string.strcmp(pred, "ugt") == 0: + return 9 + if string.strcmp(pred, "uge") == 0: + return 10 + if string.strcmp(pred, "ult") == 0: + return 11 + if string.strcmp(pred, "ule") == 0: + return 12 + if string.strcmp(pred, "une") == 0: + return 13 + return 0 + + +# ============================================================ +# _llvmir_extract_two_ops — 从模板中提取两个操作数编号 +# +# 通过输出参数返回 op1_num 和 op2_num +# ============================================================ +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, + out_nums: int | t.CPtr) -> int: + """从模板中提取前两个 %__OP 标记的编号 + + 结果写入 out_nums[0] 和 out_nums[1] + 返回 1 成功, 0 失败 + """ + out_nums[0] = -1 + out_nums[1] = -1 + m1: t.CChar | t.CPtr = _llvmir_find_op_marker(template) + if m1 is None: + return 0 + out_nums[0] = _llvmir_parse_op_num(m1) + + # 跳过第一个标记 "%__OPN__" + next_pos: t.CChar | t.CPtr = m1 + 5 # 跳过 "%__OP" + i: t.CSizeT = 0 + while next_pos[i] >= '0' and next_pos[i] <= '9': + i += 1 + next_pos = next_pos + i + 2 + + m2: t.CChar | t.CPtr = _llvmir_find_op_marker(next_pos) + if m2 is None: + return 1 + out_nums[1] = _llvmir_parse_op_num(m2) + return 1 + + +# ============================================================ +# _llvmir_resolve_op — 根据编号解析操作数值 +# ============================================================ +def _llvmir_resolve_op(op_num: int, + all_kinds: int | t.CPtr, + all_indices: int | t.CPtr, + input_ops: AsmOperand | t.CPtr, + output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: + """根据操作数编号解析实际 Value""" + if op_num < 0 or op_num >= LLVMIR_MAX_OPS: + return None + kind: int = all_kinds[op_num] + idx: int = all_indices[op_num] + if kind == 0: + # 输入操作数 + op: AsmOperand | t.CPtr = _llvmir_get_op(input_ops, idx) + return op.Value + else: + # 输出操作数 + op2: AsmOperand | t.CPtr = _llvmir_get_op(output_targets, idx) + return op2.Value + + +# ============================================================ +# _llvmir_store_outputs — 将结果存储到输出变量 +# ============================================================ +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + result: llvmlite.Value | t.CPtr, + output_targets: AsmOperand | t.CPtr, + output_count: int) -> llvmlite.Value | t.CPtr: + """将结果存储到所有输出变量""" + if output_count == 0: + return result + for oi in range(output_count): + op: AsmOperand | t.CPtr = _llvmir_get_op(output_targets, oi) + if op.Value is not None and result is not None: + llvmlite.build_store(builder, result, op.Value) + return result + + +# ============================================================ +# translate_c_llvmir — 翻译 c.LLVMIR 内联 LLVM IR +# ============================================================ +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value""" + if cl is None: + return None + + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None or cargs.__len__() < 1: + return None + + # 分配操作数数组 + op_size: t.CSizeT = AsmOperand.__sizeof__() + input_ops: AsmOperand | t.CPtr = pool.alloc(op_size * LLVMIR_MAX_OPS) + output_targets: AsmOperand | t.CPtr = pool.alloc(op_size * LLVMIR_MAX_OPS) + if input_ops is None or output_targets is None: + return None + string.memset(input_ops, 0, op_size * LLVMIR_MAX_OPS) + string.memset(output_targets, 0, op_size * LLVMIR_MAX_OPS) + + # all_ops 映射: kinds[i]=0(inp)/1(out), indices[i]=在对应数组中的索引 + all_kinds: int | t.CPtr = pool.alloc(4 * LLVMIR_MAX_OPS) + all_indices: int | t.CPtr = pool.alloc(4 * LLVMIR_MAX_OPS) + if all_kinds is None or all_indices is None: + return None + string.memset(all_kinds, 0, 4 * LLVMIR_MAX_OPS) + string.memset(all_indices, 0, 4 * LLVMIR_MAX_OPS) + + input_count: int = 0 + output_count: int = 0 + op_seq: int = 0 + + # 分配模板缓冲区 + template: t.CChar | t.CPtr = pool.alloc(LLVMIR_TEMPLATE_BUF) + if template is None: + return None + template[0] = '\0' + + # 解析返回类型（默认 i32） + ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + if cargs.__len__() >= 2: + ret_type_node: ast.AST | t.CPtr = cargs.get(1) + if ret_type_node is not None: + resolved_ty: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, ret_type_node, trans._imported_modules, trans._from_imports) + if resolved_ty is not None: + ret_ty = resolved_ty + + # ============================================================ + # 1. 解析 IR 模板（JoinedStr 或 Constant）— 收集操作数 + # ============================================================ + first_arg: ast.AST | t.CPtr = cargs.get(0) + if first_arg is None: + return None + + tk: int = first_arg.kind() + + if tk == ast.ASTKind.JoinedStr: + js: ast.JoinedStr | t.CPtr = (ast.JoinedStr | t.CPtr)(first_arg) + jvalues: list[ast.AST | t.CPtr] | t.CPtr = js.values + if jvalues is not None: + jvlen: t.CSizeT = jvalues.__len__() + for vi in range(jvlen): + part: ast.AST | t.CPtr = jvalues.get(vi) + if part is None: + continue + pk: int = part.kind() + + if pk == ast.ASTKind.Constant: + # 字符串字面量片段 + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(part) + if cn.str_val is not None: + _asm_append_cstr(template, LLVMIR_TEMPLATE_BUF, cn.str_val) + + elif pk == ast.ASTKind.FormattedValue: + fv: ast.FormattedValue | t.CPtr = (ast.FormattedValue | t.CPtr)(part) + fexpr: ast.AST | t.CPtr = fv.value + if fexpr is None or fexpr.kind() != ast.ASTKind.Call: + continue + fcall: ast.Call | t.CPtr = (ast.Call | t.CPtr)(fexpr) + attr_nm: str = _asm_is_c_call(fcall) + if attr_nm is None: + # 非 c.LInp/c.LOut 调用，作为普通输入处理 + ival: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, fexpr, None, 0, trans) + if ival is not None and input_count < LLVMIR_MAX_OPS: + op_in: AsmOperand | t.CPtr = _llvmir_get_op(input_ops, input_count) + op_in.Value = ival + all_kinds[op_seq] = 0 + all_indices[op_seq] = input_count + input_count += 1 + op_seq += 1 + # 追加占位符 + viperlib.snprintf( + template + string.strlen(template), + LLVMIR_TEMPLATE_BUF - string.strlen(template), + "%%__OP%d__", op_seq - 1) + continue + + # 处理 c.LInp / c.LOut + fargs: list[ast.AST | t.CPtr] | t.CPtr = fcall.args + if fargs is None or fargs.__len__() < 1: + continue + val_node: ast.AST | t.CPtr = fargs.get(0) + + if string.strcmp(attr_nm, "LInp") == 0: + # 输入操作数：翻译值 + inp_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, val_node, None, 0, trans) + if inp_val is not None and input_count < LLVMIR_MAX_OPS: + op_inp: AsmOperand | t.CPtr = _llvmir_get_op(input_ops, input_count) + op_inp.Value = inp_val + all_kinds[op_seq] = 0 + all_indices[op_seq] = input_count + input_count += 1 + op_seq += 1 + viperlib.snprintf( + template + string.strlen(template), + LLVMIR_TEMPLATE_BUF - string.strlen(template), + "%%__OP%d__", op_seq - 1) + + elif string.strcmp(attr_nm, "LOut") == 0: + # 输出操作数：获取变量 alloca + if val_node is not None and val_node.kind() == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(val_node) + if nm.id is not None: + alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, nm.id) + if alloca is not None and output_count < LLVMIR_MAX_OPS: + op_out: AsmOperand | t.CPtr = _llvmir_get_op(output_targets, output_count) + op_out.Value = alloca + all_kinds[op_seq] = 1 + all_indices[op_seq] = output_count + output_count += 1 + op_seq += 1 + viperlib.snprintf( + template + string.strlen(template), + LLVMIR_TEMPLATE_BUF - string.strlen(template), + "%%__OP%d__", op_seq - 1) + + elif tk == ast.ASTKind.Constant: + cn2: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(first_arg) + if cn2.str_val is not None: + string.strcpy(template, cn2.str_val) + + # ============================================================ + # 2. 解析模板并生成 LLVM 指令 + # ============================================================ + + # 检查是否为赋值形式: %__OPN__ = <指令> + # 如果是，去掉赋值前缀，记录输出操作数编号 + assign_out_num: int = -1 + assign_marker: t.CChar | t.CPtr = _llvmir_find_op_marker(template) + if assign_marker is not None and assign_marker == template: + # 模板以 %__OP 开头 + # 查找 "=" 号 + eq_pos: t.CChar | t.CPtr = string.strchr(template, 61) # '=' = 61 + if eq_pos is not None: + # 确认 "%__OPN__ = " 格式 + assign_out_num = _llvmir_parse_op_num(assign_marker) + # 跳过 "%__OP__ = " 到实际指令开始 + # 找到 "=" 后第一个非空格字符 + instr_start: t.CChar | t.CPtr = eq_pos + 1 + while instr_start[0] == ' ' or instr_start[0] == '\t': + instr_start += 1 + # 将剩余部分复制到新模板 + tmpl2: t.CChar | t.CPtr = pool.alloc(LLVMIR_TEMPLATE_BUF) + if tmpl2 is not None: + string.strcpy(tmpl2, instr_start) + template = tmpl2 + + # 去除前导空格 + while template[0] == ' ' or template[0] == '\t': + template += 1 + + # 提取指令关键字（第一个单词） + instr_kw: t.CChar | t.CPtr = pool.alloc(32) + if instr_kw is None: + return None + instr_kw[0] = '\0' + ki: t.CSizeT = 0 + while template[ki] != ' ' and template[ki] != '\t' and template[ki] != '\0' and ki < 31: + instr_kw[ki] = template[ki] + ki += 1 + instr_kw[ki] = '\0' + + # ============================================================ + # 二元运算: add/sub/mul/sdiv/udiv/srem/urem/and/or/xor/shl/lshr/ashr + # ============================================================ + binop_result: llvmlite.Value | t.CPtr = None + + # 提取两个操作数编号 + op_nums: int | t.CPtr = pool.alloc(8) + if op_nums is not None: + string.memset(op_nums, 0, 8) + _llvmir_extract_two_ops(template, op_nums) + op1_num: int = -1 + op2_num: int = -1 + if op_nums is not None: + op1_num = op_nums[0] + op2_num = op_nums[1] + + lhs: llvmlite.Value | t.CPtr = _llvmir_resolve_op( + op1_num, all_kinds, all_indices, input_ops, output_targets) + rhs: llvmlite.Value | t.CPtr = _llvmir_resolve_op( + op2_num, all_kinds, all_indices, input_ops, output_targets) + + if lhs is not None and rhs is not None: + if string.strcmp(instr_kw, "add") == 0: + binop_result = llvmlite.build_add(builder, lhs, rhs) + elif string.strcmp(instr_kw, "sub") == 0: + binop_result = llvmlite.build_sub(builder, lhs, rhs) + elif string.strcmp(instr_kw, "mul") == 0: + binop_result = llvmlite.build_mul(builder, lhs, rhs) + elif string.strcmp(instr_kw, "sdiv") == 0: + binop_result = llvmlite.build_sdiv(builder, lhs, rhs) + elif string.strcmp(instr_kw, "udiv") == 0: + binop_result = llvmlite.build_udiv(builder, lhs, rhs) + elif string.strcmp(instr_kw, "srem") == 0: + binop_result = llvmlite.build_srem(builder, lhs, rhs) + elif string.strcmp(instr_kw, "urem") == 0: + binop_result = llvmlite.build_urem(builder, lhs, rhs) + elif string.strcmp(instr_kw, "and") == 0: + binop_result = llvmlite.build_and(builder, lhs, rhs) + elif string.strcmp(instr_kw, "or") == 0: + binop_result = llvmlite.build_or(builder, lhs, rhs) + elif string.strcmp(instr_kw, "xor") == 0: + binop_result = llvmlite.build_xor(builder, lhs, rhs) + elif string.strcmp(instr_kw, "shl") == 0: + binop_result = llvmlite.build_shl(builder, lhs, rhs) + elif string.strcmp(instr_kw, "lshr") == 0: + binop_result = llvmlite.build_lshr(builder, lhs, rhs) + elif string.strcmp(instr_kw, "ashr") == 0: + binop_result = llvmlite.build_ashr(builder, lhs, rhs) + elif string.strcmp(instr_kw, "fadd") == 0: + binop_result = llvmlite.build_fadd(builder, lhs, rhs) + elif string.strcmp(instr_kw, "fsub") == 0: + binop_result = llvmlite.build_fsub(builder, lhs, rhs) + elif string.strcmp(instr_kw, "fmul") == 0: + binop_result = llvmlite.build_fmul(builder, lhs, rhs) + elif string.strcmp(instr_kw, "fdiv") == 0: + binop_result = llvmlite.build_fdiv(builder, lhs, rhs) + + # ============================================================ + # icmp 比较: icmp %__OP0__, %__OP1__ + # ============================================================ + if binop_result is None and string.strcmp(instr_kw, "icmp") == 0: + # 提取预测（第二个单词） + pred_start: t.CChar | t.CPtr = template + ki + 1 + while pred_start[0] == ' ' or pred_start[0] == '\t': + pred_start += 1 + pred_str: t.CChar | t.CPtr = pool.alloc(16) + if pred_str is not None: + pred_str[0] = '\0' + pi: t.CSizeT = 0 + while pred_start[pi] != ' ' and pred_start[pi] != '\t' and pred_start[pi] != '\0' and pi < 15: + pred_str[pi] = pred_start[pi] + pi += 1 + pred_str[pi] = '\0' + pred_val: int = _llvmir_match_icmp_pred(pred_str) + if lhs is not None and rhs is not None: + binop_result = llvmlite.build_icmp(builder, pred_val, lhs, rhs) + + # ============================================================ + # fcmp 比较: fcmp %__OP0__, %__OP1__ + # ============================================================ + if binop_result is None and string.strcmp(instr_kw, "fcmp") == 0: + pred_start2: t.CChar | t.CPtr = template + ki + 1 + while pred_start2[0] == ' ' or pred_start2[0] == '\t': + pred_start2 += 1 + pred_str2: t.CChar | t.CPtr = pool.alloc(16) + if pred_str2 is not None: + pred_str2[0] = '\0' + pi2: t.CSizeT = 0 + while pred_start2[pi2] != ' ' and pred_start2[pi2] != '\t' and pred_start2[pi2] != '\0' and pi2 < 15: + pred_str2[pi2] = pred_start2[pi2] + pi2 += 1 + pred_str2[pi2] = '\0' + pred_val2: int = _llvmir_match_fcmp_pred(pred_str2) + if lhs is not None and rhs is not None: + binop_result = llvmlite.build_fcmp(builder, pred_val2, lhs, rhs) + + # ============================================================ + # store 指令: store %__OP0__, %__OP1__ + # ============================================================ + if binop_result is None and string.strcmp(instr_kw, "store") == 0: + if lhs is not None and rhs is not None: + llvmlite.build_store(builder, lhs, rhs) + return None + + # ============================================================ + # load 指令: load , %__OP0__ + # ============================================================ + if binop_result is None and string.strcmp(instr_kw, "load") == 0: + if lhs is not None: + # 从指针加载，类型由 ret_ty 指定 + binop_result = llvmlite.build_load(builder, ret_ty, lhs) + + # ============================================================ + # 类型转换: zext/sext/trunc/bitcast/ptrtoint/inttoptr + # ============================================================ + if binop_result is None: + if string.strcmp(instr_kw, "zext") == 0: + if lhs is not None: + binop_result = llvmlite.build_zext(builder, lhs, ret_ty) + elif string.strcmp(instr_kw, "sext") == 0: + if lhs is not None: + binop_result = llvmlite.build_sext(builder, lhs, ret_ty) + elif string.strcmp(instr_kw, "trunc") == 0: + if lhs is not None: + binop_result = llvmlite.build_trunc(builder, lhs, ret_ty) + elif string.strcmp(instr_kw, "bitcast") == 0: + if lhs is not None: + binop_result = llvmlite.build_bitcast(builder, lhs, ret_ty) + elif string.strcmp(instr_kw, "ptrtoint") == 0: + if lhs is not None: + binop_result = llvmlite.build_ptrtoint(builder, lhs, ret_ty) + elif string.strcmp(instr_kw, "inttoptr") == 0: + if lhs is not None: + binop_result = llvmlite.build_inttoptr(builder, lhs, ret_ty) + + # ============================================================ + # 3. 处理输出 + # ============================================================ + if binop_result is not None: + # 如果有赋值形式（%__OPN__ = ...），存储到对应输出 + if assign_out_num >= 0: + target: llvmlite.Value | t.CPtr = _llvmir_resolve_op( + assign_out_num, all_kinds, all_indices, input_ops, output_targets) + if target is not None: + llvmlite.build_store(builder, binop_result, target) + + # 存储到所有输出操作数 + _llvmir_store_outputs(pool, builder, binop_result, output_targets, output_count) + return binop_result + + # 未识别的指令，返回 0 + stdio.printf("[LLVMIR] unsupported instruction: %s\n", instr_kw) + return None + + +# ============================================================ +# _apply_struct_defaults — 应用结构体字段的默认值 +# +# 在 zeroinitializer 之后、显式参数之前调用 +# 遍历所有字段，对有默认值的字段翻译并 store +# 显式参数（位置/关键字）后续会覆盖默认值 +# ============================================================ +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str, + tmp: llvmlite.Value | t.CPtr, + trans: HT.Translator | t.CPtr): + """应用结构体字段的默认值""" + # 用类型指针定位 entry，规避跨模块同名 find_struct 找错 + entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + if entry is None: + return + for fi in range(entry.FieldCount): + fe: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct._get_field_entry( + entry, fi) + if fe is None or fe.DefaultVal is None: + continue + default_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, fe.DefaultVal, None, 0, trans) + if default_val is None: + continue + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, fe.Ty, tmp, fe.Index) + if field_ptr is None: + continue + default_val = HandlesExpr.coerce_to_type(builder, default_val, fe.Ty) + llvmlite.build_store(builder, default_val, field_ptr) + + +# ============================================================ +# _translate_struct_ctor — 结构体构造函数 Point(10, 20) +# +# 生成: +# %tmp = alloca {i32, i32} +# store {i32, i32} zeroinitializer, {i32, i32}* %tmp +# %f0 = getelementptr {i32, i32}, {i32, i32}* %tmp, i32 0, i32 0 +# store i32 10, i32* %f0 +# %f1 = getelementptr {i32, i32}, {i32, i32}* %tmp, i32 0, i32 1 +# store i32 20, i32* %f1 +# %result = load {i32, i32}, {i32, i32}* %tmp +# ============================================================ +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str, + cargs: list[ast.AST | t.CPtr] | t.CPtr, + can: t.CSizeT, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """结构体构造函数：按字段顺序 store 参数值，返回结构体值 + + OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load + 普通结构体：alloca → zero + defaults → store 参数 → load + """ + # 创建临时 alloca + tmp: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, struct_ty) + if tmp is None: + return llvmlite.ConstZero(pool, struct_ty) + + # 检查是否为 OOP 结构体（用类型指针定位 entry，规避跨模块同名 find_struct 找错） + ctor_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + is_oop: int = 0 + if ctor_entry is not None: + is_oop = ctor_entry.IsOOP + + if is_oop != 0: + # 存储指针：默认用 alloca，如果有 __new__ 则用 __new__ 返回的指针 + storage_ptr: llvmlite.Value | t.CPtr = tmp + + # 如果有 __new__，调用 __new__(tmp, args...) 获取实际存储指针 + # __new__ 签名与 __init__ 一样（self + args），返回 Ptr(struct_ty) + has_new_flag: int = 0 + if ctor_entry is not None: + has_new_flag = ctor_entry.HasNew + if has_new_flag != 0: + new_arg_vals: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if new_arg_vals is not None: + string.memset(new_arg_vals, 0, 8 * 32) + new_arg_count: int = 0 + for nai in range(can): + new_arg_node: ast.AST | t.CPtr = cargs.get(nai) + if new_arg_node is None: + continue + new_av: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, new_arg_node, None, 0, trans) + if new_av is not None: + if new_arg_count < 32: + new_arg_vals[new_arg_count] = t.CSizeT(new_av) + new_arg_count += 1 + + new_ptr: llvmlite.Value | t.CPtr = _call_method_on_ptr( + pool, builder, mod, class_name, + "__new__", tmp, new_arg_vals, new_arg_count, trans) + if new_ptr is not None: + storage_ptr = new_ptr + + # OOP 结构体：调用 __before_init__ 进行零值填充和默认值赋值 + _call_method_on_ptr(pool, builder, mod, class_name, + "__before_init__", storage_ptr, None, 0, trans) + + # 如果有 __init__，调用 __init__(ptr, args...) + has_init_flag: int = 0 + if ctor_entry is not None: + has_init_flag = ctor_entry.HasInit + if has_init_flag != 0: + # 翻译构造函数参数 + arg_vals: t.CSizeT | t.CPtr = pool.alloc(8 * 32) + if arg_vals is not None: + string.memset(arg_vals, 0, 8 * 32) + real_arg_count: int = 0 + for ai in range(can): + arg_node: ast.AST | t.CPtr = cargs.get(ai) + if arg_node is None: + continue + av: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg_node, None, 0, trans) + if av is not None: + if real_arg_count < 32: + arg_vals[real_arg_count] = t.CSizeT(av) + real_arg_count += 1 + + _call_method_on_ptr(pool, builder, mod, class_name, + "__init__", storage_ptr, arg_vals, real_arg_count, trans) + else: + # 无 __init__：逐字段 store 参数值 + for ai in range(can): + arg: ast.AST | t.CPtr = cargs.get(ai) + if arg is None: + continue + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + if arg_val is None: + continue + fe: HandlesStruct.FieldEntry | t.CPtr = None + if ctor_entry is not None: + fe = HandlesStruct._get_field_entry(ctor_entry, ai) + else: + fe = HandlesStruct.get_field_by_index(class_name, ai) + if fe is None: + continue + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, fe.Ty, storage_ptr, fe.Index) + if field_ptr is None: + continue + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, fe.Ty) + llvmlite.build_store(builder, arg_val, field_ptr) + + # 返回指针（而非结构体值），赋值时由 coerce_to_type 按目标类型决定是否 load + return storage_ptr + + # 普通结构体：先 store zeroinitializer 清零（确保未赋值字段为零） + zero: llvmlite.Value | t.CPtr = llvmlite.ConstZero(pool, struct_ty) + llvmlite.build_store(builder, zero, tmp) + + # 应用默认值（显式参数会覆盖默认值） + _apply_struct_defaults(pool, builder, mod, struct_ty, class_name, tmp, trans) + + # 逐字段 store 参数值（覆盖默认值） + for ai in range(can): + arg: ast.AST | t.CPtr = cargs.get(ai) + if arg is None: + continue + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + if arg_val is None: + continue + # 获取字段信息（用 ctor_entry 规避跨模块同名 find_struct 找错） + fe: HandlesStruct.FieldEntry | t.CPtr = None + if ctor_entry is not None: + fe = HandlesStruct._get_field_entry(ctor_entry, ai) + else: + fe = HandlesStruct.get_field_by_index(class_name, ai) + if fe is None: + continue + # GEP 到字段 ai + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, fe.Ty, tmp, fe.Index) + if field_ptr is None: + continue + # 类型转换并 store + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, fe.Ty) + llvmlite.build_store(builder, arg_val, field_ptr) + + # 返回指针（而非结构体值），赋值时由 coerce_to_type 按目标类型决定是否 load + return tmp + + +# ============================================================ +# _do_virtual_call — 通过虚表间接调用虚方法 +# +# IR 模式: +# %1 = getelementptr StructTy, StructTy* %self, i32 0, i32 0 ; i8** +# %2 = load i8*, i8** %1 ; i8* (vtable ptr) +# %3 = bitcast i8* %2 to {i8*, ...}* ; vtable typed ptr +# %4 = getelementptr {i8*, ...}, {i8*, ...}* %3, i32 0, i32 N ; i8** +# %5 = load i8*, i8** %4 ; i8* (func ptr) +# %6 = bitcast i8* %5 to RetTy(Params)* ; func typed ptr +# call RetTy %6(Params %self, ...) +# +# 参数: +# args_head — 已构建好的参数链表头（self_ptr 为首节点） +# args_count — 参数总数（含 self） +# found_func — 方法对应的 Function 对象（用于推断函数签名） +# ret_ty — 返回类型 +# ============================================================ +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + class_name: str, + method_name: str, + self_ptr: llvmlite.Value | t.CPtr, + args_head: llvmlite.Value | t.CPtr, + args_count: int, + found_func: llvmlite.Function | t.CPtr, + ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）""" + if pool is None or builder is None or class_name is None or method_name is None: + return None + if self_ptr is None or args_head is None: + return None + + # 1. 获取结构体类型（用 find_struct_by_type 规避跨模块同名 find_struct 找错） + vc_entry: HandlesStruct.StructEntry | t.CPtr = None + struct_ty: llvmlite.LLVMType | t.CPtr = None + if self_ptr.Ty is not None: + vc_entry = HandlesStruct.get_struct_type_from_value(self_ptr) + if vc_entry is not None: + # vc_entry 是 struct_ty（LLVMType），需要找 StructEntry + vc_entry = HandlesStruct.find_struct_by_type(vc_entry) + if vc_entry is not None: + struct_ty = vc_entry.Ty + if struct_ty is None: + struct_ty = HandlesStruct.get_struct_type(class_name) + if struct_ty is None: + return None + + # 2. 获取方法在虚表中的索引（用 entry 直接查找） + method_index: int = -1 + if vc_entry is not None and vc_entry.VTableMethods is not None: + methods_arr: t.CSizeT | t.CPtr = (t.CSizeT | t.CPtr)(t.CVoid(vc_entry.VTableMethods, t.CPtr)) + for mi in range(vc_entry.VTableMethodCount): + mname_addr: t.CSizeT = methods_arr[mi] + if mname_addr == 0: + continue + mname: str = (str | t.CPtr)(t.CVoid(mname_addr, t.CPtr)) + if mname is not None and string.strcmp(mname, method_name) == 0: + method_index = mi + break + else: + method_index = HandlesStruct.get_vtable_method_index(class_name, method_name) + if method_index < 0: + return None + + # 3. 基础类型 + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + i8_ptr_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ptr_ty) + + # 4. bitcast 到 i8** 访问 __vtable__ 字段（索引 0） + # vtable 指针总是在对象第一个字段，无需 GEP，直接 bitcast 到 i8** 即可。 + # 这样即使 struct_ty 是 opaque（跨模块未加载完整定义）也能正常工作。 + vtable_slot: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + builder, self_ptr, i8_ptr_ptr_ty) + if vtable_slot is None: + return None + + # 5. load vtable 指针 (i8*) + vtable_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, vtable_slot) + if vtable_ptr is None: + return None + + # 6. 构造 vtable 类型 {i8*, i8*, ...} + method_count: int = 0 + if vc_entry is not None: + method_count = vc_entry.VTableMethodCount + else: + method_count = HandlesStruct.get_vtable_method_count(class_name) + if method_count <= 0: + return None + + vt_first: llvmlite.ParamNode | t.CPtr = None + vt_prev: llvmlite.ParamNode | t.CPtr = None + for vi in range(method_count): + pn: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, i8_ptr_ty) + if pn is None: + continue + if vt_first is None: + vt_first = pn + if vt_prev is not None: + llvmlite.paramnode_set_next(vt_prev, pn) + vt_prev = pn + + vtable_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Struct(pool, vt_first, method_count, None) + if vtable_ty is None: + return None + vtable_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, vtable_ty) + + # 7. bitcast i8* 到 vtable 类型指针 + vtable_typed: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + builder, vtable_ptr, vtable_ptr_ty) + if vtable_typed is None: + return None + + # 8. GEP 到方法槽位 + func_slot: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, vtable_ty, i8_ptr_ty, vtable_typed, method_index) + if func_slot is None: + return None + + # 9. load 函数指针 (i8*) + func_i8: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, func_slot) + if func_i8 is None: + return None + + # 10. 从 args_head 构造函数类型 RetTy(Param1, Param2, ...) + # 直接遍历 Value 链表提取参数类型，避免跨模块访问 found_func（stub 字段偏移问题） + fp_first: llvmlite.ParamNode | t.CPtr = None + fp_prev: llvmlite.ParamNode | t.CPtr = None + fp_count: int = 0 + cur_val: llvmlite.Value | t.CPtr = args_head + for vi in range(args_count): + if cur_val is None: + break + cur_ty: llvmlite.LLVMType | t.CPtr = llvmlite.value_get_ty(cur_val) + fpn: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, cur_ty) + if fpn is not None: + if fp_first is None: + fp_first = fpn + if fp_prev is not None: + llvmlite.paramnode_set_next(fp_prev, fpn) + fp_prev = fpn + cur_val = llvmlite.value_get_next(cur_val) + fp_count += 1 + + func_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Func(pool, ret_ty, fp_first, fp_count) + if func_ty is None: + return None + func_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, func_ty) + + # 11. bitcast i8* 到函数指针类型 + func_ptr: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, func_i8, func_ptr_ty) + if func_ptr is None: + return None + + # 12. 间接调用 + return llvmlite.build_call_indirect(builder, func_ptr, args_head, args_count, ret_ty) + + +# ============================================================ +# _call_method_on_ptr — 在结构体指针上调用方法 +# +# 辅助函数：构造 ClassName.method_name 调用，self 指针为首个参数 +# 如果该方法在虚表中，则走虚方法分发路径（间接调用） +# ============================================================ +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + class_name: str, + method_name: str, + self_ptr: llvmlite.Value | t.CPtr, + extra_args: t.CSizeT | t.CPtr, + extra_count: int, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """在结构体指针上调用方法，返回 call 结果（void 方法返回 None）""" + if pool is None or builder is None or class_name is None or method_name is None: + return None + if self_ptr is None: + return None + + # 提前获取结构体 entry（用 find_struct_by_type 规避跨模块同名 find_struct 找错） + # 用于 SHA1 查找和虚表分派 + cmop_entry: HandlesStruct.StructEntry | t.CPtr = None + cmop_struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type_from_value(self_ptr) + if cmop_struct_ty is not None: + cmop_entry = HandlesStruct.find_struct_by_type(cmop_struct_ty) + if cmop_entry is None: + cmop_entry = HandlesStruct.find_struct(class_name) + + # 构造方法查找名: ClassName.method_name + lookup_name: t.CChar | t.CPtr = pool.alloc(128) + if lookup_name is None: + return None + viperlib.snprintf(lookup_name, 128, "%s.%s", class_name, method_name) + + # 查找函数（支持 SHA1 后缀匹配） + found_func: llvmlite.Function | t.CPtr = find_func_in_module(mod, lookup_name) + + # 确定调用名 + call_name: str = lookup_name + if found_func is not None: + found_name: t.CChar | t.CPtr = llvmlite.function_get_name(found_func) + if found_name is not None: + call_name = found_name + else: + # found_func 为 None（stub 未注入）：用类所属模块的 SHA1 构造调用名 + # 优先从 entry 获取 SHA1（规避跨模块同名 find_struct 找错） + cls_sha1: str = None + if cmop_entry is not None: + cls_sha1 = cmop_entry.ModuleSha1 + else: + cls_sha1 = HandlesStruct.get_struct_sha1(class_name) + if cls_sha1 is None and trans is not None: + # fallback: 从 from_imports 查找类所属模块的 SHA1（含截断处理） + cls_sha1 = _lookup_mod_sha1(pool, trans._from_imports, class_name) + if cls_sha1 is None: + err_msg: t.CChar | t.CPtr = pool.alloc(256) + if err_msg is not None: + viperlib.snprintf(err_msg, 256, "类 '%s' 方法 '%s' 找不到模块 SHA1", class_name, method_name) + HandlesType.fatal_error(None, err_msg) + HandlesType.fatal_error(None, "找不到模块 SHA1") + mangled_len2: t.CSizeT = string.strlen(cls_sha1) + string.strlen(lookup_name) + 2 + mangled2: str = pool.alloc(mangled_len2) + if mangled2 is not None: + viperlib.snprintf(mangled2, mangled_len2, "%s.%s", cls_sha1, lookup_name) + call_name = mangled2 + + # 确定返回类型（__before_init__ 和 __init__ 返回 void） + void_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Void(pool) + call_ret_ty: llvmlite.LLVMType | t.CPtr = void_ty + if found_func is not None: + frt: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(found_func) + if frt is not None: + call_ret_ty = frt + + # 构建参数链表: self_ptr → extra_args... + llvmlite.value_set_next(self_ptr, None) + head: llvmlite.Value | t.CPtr = self_ptr + tail: llvmlite.Value | t.CPtr = self_ptr + + if extra_args is not None and extra_count > 0: + for ei in range(extra_count): + arg_addr: t.CSizeT = extra_args[ei] + if arg_addr == 0: + continue + av: llvmlite.Value | t.CPtr = (llvmlite.Value | t.CPtr)(t.CVoid(arg_addr, t.CPtr)) + if av is not None: + llvmlite.value_set_next(av, None) + llvmlite.value_set_next(tail, av) + tail = av + + total_count: int = 1 + extra_count + + # 默认参数填入：如果调用点参数少于函数定义参数，用零值填充缺失参数 + # 这处理 list[int](bd) 中 elem_size=0 默认值缺失的情况 + if found_func is not None: + param_node: llvmlite.Param | t.CPtr = llvmlite.function_get_param_head(found_func) + skip_idx: int = 0 + # 跳过已提供的参数 + while skip_idx < total_count and param_node is not None: + param_node = llvmlite.param_get_next(param_node) + skip_idx += 1 + # 为剩余参数填入零值 + while param_node is not None: + missing_ty: llvmlite.LLVMType | t.CPtr = llvmlite.param_get_ty(param_node) + zero_val: llvmlite.Value | t.CPtr = llvmlite.ConstZero(pool, missing_ty) + if zero_val is not None: + llvmlite.value_set_next(zero_val, None) + llvmlite.value_set_next(tail, zero_val) + tail = zero_val + total_count += 1 + param_node = llvmlite.param_get_next(param_node) + + # 虚方法分发：如果类有虚表且该方法在虚表中，走间接调用 + # 复用前面已获取的 cmop_entry（find_struct_by_type 规避跨模块同名找错） + if found_func is not None: + cmop_has_vt: int = 0 + cmop_method_idx: int = -1 + if cmop_entry is not None: + cmop_has_vt = cmop_entry.HasVTable + if cmop_has_vt == 1 and cmop_entry.VTableMethods is not None: + cmop_methods_arr: t.CSizeT | t.CPtr = (t.CSizeT | t.CPtr)(t.CVoid(cmop_entry.VTableMethods, t.CPtr)) + for cmop_mi in range(cmop_entry.VTableMethodCount): + cmop_mname_addr: t.CSizeT = cmop_methods_arr[cmop_mi] + if cmop_mname_addr == 0: + continue + cmop_mname: str = (str | t.CPtr)(t.CVoid(cmop_mname_addr, t.CPtr)) + if cmop_mname is not None and string.strcmp(cmop_mname, method_name) == 0: + cmop_method_idx = cmop_mi + break + if cmop_has_vt == 1 and cmop_method_idx >= 0: + return _do_virtual_call(pool, builder, mod, class_name, method_name, + self_ptr, head, total_count, found_func, call_ret_ty) + + return llvmlite.build_call(builder, call_name, head, total_count, call_ret_ty, 0) + + +# ============================================================ +# _infer_method_ret_ty - 根据方法名推断返回类型 +# +# stub 未注入时，find_func_in_module 返回 None， +# 需要根据方法名推断返回类型避免指针被截断为 i32。 +# ============================================================ +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, + method_name: str) -> llvmlite.LLVMType | t.CPtr: + """根据方法名推断返回类型（stub 未注入时的回退方案）""" + if pool is None or method_name is None: + return llvmlite.Int32(pool) + + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, llvmlite.Int8(pool)) + + # 返回 i8* 的方法（分配/返回指针） + if string.strcmp(method_name, "alloc") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "calloc") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "realloc") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "alloc_buf") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "_fl_pop") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "_buddy_of") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "_split_to_order") == 0: + return i8_ptr_ty + if string.strcmp(method_name, "__enter__") == 0: + return i8_ptr_ty + + # 返回 void 的方法 + if string.strcmp(method_name, "__before_init__") == 0: + return llvmlite.Void(pool) + + # 默认 i32（free, reset, __init__, __exit__, _fl_push 等） + return llvmlite.Int32(pool) + + +# ============================================================ +# _translate_method_call - 翻译方法调用 obj.method(args) +# +# 生成: call @SHA1.ClassName.method_name(Ptr(StructTy) %obj, args...) +# +# obj_ptr 是对象变量的 alloca（Ptr(StructTy)），作为 self 参数传入 +# ============================================================ +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + class_name: str, + method_name: str, + obj_ptr: llvmlite.Value | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """翻译方法调用，返回 call 结果 Value""" + if pool is None or builder is None or class_name is None or method_name is None: + return None + if obj_ptr is None or cl is None: + return None + + # 提前获取结构体 entry（用 find_struct_by_type 规避跨模块同名 find_struct 找错） + # 用于继承链遍历、SHA1 查找和虚表分派 + mc_entry: HandlesStruct.StructEntry | t.CPtr = None + mc_struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type_from_value(obj_ptr) + if mc_struct_ty is not None: + mc_entry = HandlesStruct.find_struct_by_type(mc_struct_ty) + if mc_entry is None: + mc_entry = HandlesStruct.find_struct(class_name) + + # 构造方法查找名: ClassName.method_name + lookup_name: t.CChar | t.CPtr = pool.alloc(128) + if lookup_name is None: + return None + viperlib.snprintf(lookup_name, 128, "%s.%s", class_name, method_name) + + # 查找函数（支持 SHA1 后缀匹配） + found_func: llvmlite.Function | t.CPtr = find_func_in_module(mod, lookup_name) + + # 沿继承链查找父类方法（子类未覆盖时回退到父类） + if found_func is None: + cur_parent: str = None + if mc_entry is not None: + cur_parent = mc_entry.ParentName + else: + cur_parent = HandlesStruct.get_parent_name(class_name) + while cur_parent is not None: + parent_lookup: t.CChar | t.CPtr = pool.alloc(128) + if parent_lookup is not None: + viperlib.snprintf(parent_lookup, 128, "%s.%s", cur_parent, method_name) + parent_func: llvmlite.Function | t.CPtr = find_func_in_module(mod, parent_lookup) + if parent_func is not None: + found_func = parent_func + break + # 后续父类按名查找（无 ParentSha1，只能按名回退） + cur_parent = HandlesStruct.get_parent_name(cur_parent) + + # 确定调用名 + call_name: str = lookup_name + if found_func is not None: + found_name: t.CChar | t.CPtr = llvmlite.function_get_name(found_func) + if found_name is not None: + call_name = found_name + else: + # found_func 为 None（stub 未注入）：用类所属模块的 SHA1 构造调用名 + # 沿继承链查找第一个有 SHA1 的类 + cls_sha1_mc: str = None + if mc_entry is not None: + cls_sha1_mc = mc_entry.ModuleSha1 + else: + cls_sha1_mc = HandlesStruct.get_struct_sha1(class_name) + if cls_sha1_mc is None: + cur_p: str = None + if mc_entry is not None: + cur_p = mc_entry.ParentName + else: + cur_p = HandlesStruct.get_parent_name(class_name) + while cur_p is not None and cls_sha1_mc is None: + cls_sha1_mc = HandlesStruct.get_struct_sha1(cur_p) + cur_p = HandlesStruct.get_parent_name(cur_p) + if cls_sha1_mc is not None: + mangled_len_mc: t.CSizeT = string.strlen(cls_sha1_mc) + string.strlen(lookup_name) + 2 + mangled_mc: str = pool.alloc(mangled_len_mc) + if mangled_mc is not None: + viperlib.snprintf(mangled_mc, mangled_len_mc, "%s.%s", cls_sha1_mc, lookup_name) + call_name = mangled_mc + + # 确定返回类型 + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + call_ret_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if found_func is not None: + frt: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(found_func) + if frt is not None: + call_ret_ty = frt + else: + # found_func 为 None（stub 未注入）：根据方法名推断返回类型 + call_ret_ty = _infer_method_ret_ty(pool, method_name) + + # 翻译调用参数 + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + can: t.CSizeT = 0 + if cargs is not None: + can = cargs.__len__() + + # 构建参数链表: obj_ptr → arg1 → arg2 → ... + llvmlite.value_set_next(obj_ptr, None) + head: llvmlite.Value | t.CPtr = obj_ptr + tail: llvmlite.Value | t.CPtr = obj_ptr + + for ai in range(can): + arg_node: ast.AST | t.CPtr = cargs.get(ai) + if arg_node is None: + continue + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg_node, None, 0, trans) + if arg_val is None: + continue + llvmlite.value_set_next(arg_val, None) + llvmlite.value_set_next(tail, arg_val) + tail = arg_val + + total_count: int = 1 + can + + # 虚方法分发：如果类有虚表且该方法在虚表中，走间接调用 + # （不依赖 found_func — 继承的虚方法可能没有子类实现） + # 复用前面已获取的 mc_entry（find_struct_by_type 规避跨模块同名找错） + mc_has_vt: int = 0 + mc_method_idx: int = -1 + if mc_entry is not None: + mc_has_vt = mc_entry.HasVTable + if mc_has_vt == 1 and mc_entry.VTableMethods is not None: + mc_methods_arr: t.CSizeT | t.CPtr = (t.CSizeT | t.CPtr)(t.CVoid(mc_entry.VTableMethods, t.CPtr)) + for mc_mi in range(mc_entry.VTableMethodCount): + mc_mname_addr: t.CSizeT = mc_methods_arr[mc_mi] + if mc_mname_addr == 0: + continue + mc_mname: str = (str | t.CPtr)(t.CVoid(mc_mname_addr, t.CPtr)) + if mc_mname is not None and string.strcmp(mc_mname, method_name) == 0: + mc_method_idx = mc_mi + break + if mc_has_vt == 1 and mc_method_idx >= 0: + return _do_virtual_call(pool, builder, mod, class_name, method_name, + obj_ptr, head, total_count, found_func, call_ret_ty) + + return llvmlite.build_call(builder, call_name, head, total_count, call_ret_ty, 0) + + +# ============================================================ +# _kwarg_name_matches — 检查关键字参数名是否匹配函数参数名 +# +# 函数参数名在 LLVM IR 中带 % 前缀（如 "%a"），关键字参数名无前缀（如 "a"） +# 跳过 % 前缀后逐字符比较 +# ============================================================ +def _kwarg_name_matches(kw_arg: str, + param_name: t.CChar | t.CPtr) -> int: + """检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）""" + if kw_arg is None or param_name is None: + return 0 + start: t.CSizeT = 0 + if param_name[0] == '%': + start = 1 + for i, ch in enumerate(kw_arg): + c: t.CChar = param_name[start + i] + if c == '\0' or c != ch: + return 0 + if param_name[start + i + 1] != '\0': + return 0 + return 1 + + +# ============================================================ +# _translate_struct_ctor_kw — 结构体构造函数（关键字参数） +# +# 支持两种语法: +# Point(x=10, y=20) — 全关键字 +# Point(10, y=20) — 混合（位置 + 关键字） +# +# 位置参数按字段索引赋值，关键字参数按字段名查找赋值 +# ============================================================ +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr, + class_name: str, + cargs: list[ast.AST | t.CPtr] | t.CPtr, + can: t.CSizeT, + keywords: list[ast.AST | t.CPtr] | t.CPtr, + kw_count: t.CSizeT, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """结构体构造函数（关键字参数版本）""" + # 创建临时 alloca + tmp: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, struct_ty) + if tmp is None: + return llvmlite.ConstZero(pool, struct_ty) + + # OOP 结构体：调用 __before_init__ 进行零值填充和默认值赋值 + # 用 find_struct_by_type 规避跨模块同名 find_struct 找错 + kw_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + is_oop: int = 0 + if kw_entry is not None: + is_oop = kw_entry.IsOOP + if is_oop != 0: + _call_method_on_ptr(pool, builder, mod, class_name, + "__before_init__", tmp, None, 0, trans) + else: + # 普通结构体：store zeroinitializer + 应用默认值 + zero: llvmlite.Value | t.CPtr = llvmlite.ConstZero(pool, struct_ty) + llvmlite.build_store(builder, zero, tmp) + _apply_struct_defaults(pool, builder, mod, struct_ty, class_name, tmp, trans) + + # 先处理位置参数（按字段索引，覆盖默认值） + for ai in range(can): + arg: ast.AST | t.CPtr = cargs.get(ai) + if arg is None: + continue + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + if arg_val is None: + continue + fe: HandlesStruct.FieldEntry | t.CPtr = None + if kw_entry is not None: + fe = HandlesStruct._get_field_entry(kw_entry, ai) + else: + fe = HandlesStruct.get_field_by_index(class_name, ai) + if fe is None: + continue + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, fe.Ty, tmp, fe.Index) + if field_ptr is None: + continue + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, fe.Ty) + llvmlite.build_store(builder, arg_val, field_ptr) + + # 处理关键字参数（按字段名查找） + # SHA1 从 kw_entry 获取（规避跨模块同名 find_struct 找错） + kw_sha1: str = None + if kw_entry is not None: + kw_sha1 = kw_entry.ModuleSha1 + for ki in range(kw_count): + kw: ast.Keyword | t.CPtr = (ast.Keyword | t.CPtr)(keywords.get(ki)) + if kw is None or kw.arg is None or kw.value is None: + continue + fe: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct.lookup_field_by_class( + class_name, kw.arg, kw_sha1) + if fe is None: + continue + kw_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, kw.value, None, 0, trans) + if kw_val is None: + continue + field_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep_struct( + builder, struct_ty, fe.Ty, tmp, fe.Index) + if field_ptr is None: + continue + kw_val = HandlesExpr.coerce_to_type(builder, kw_val, fe.Ty) + llvmlite.build_store(builder, kw_val, field_ptr) + + # 返回指针（而非结构体值），赋值时由 coerce_to_type 按目标类型决定是否 load + return tmp + + +# ============================================================ +# _translate_call_with_kwargs — 带关键字参数的函数调用 +# +# 支持两种语法: +# add(b=4, a=3) — 全关键字乱序 +# greet("Hello", name="World") — 位置 + 关键字混合 +# +# 策略: 遍历 found_func.Params 链表，对每个参数: +# 1. 若位置参数仍有剩余（param_idx < can），取位置参数 +# 2. 否则按参数名匹配关键字参数 +# 3. 按参数顺序构建调用参数链表 +# ============================================================ +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + func_name: str, + found_func: llvmlite.Function | t.CPtr, + cargs: list[ast.AST | t.CPtr] | t.CPtr, + can: t.CSizeT, + keywords: list[ast.AST | t.CPtr] | t.CPtr, + kw_count: t.CSizeT, + trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """带关键字参数的函数调用，按参数顺序构建调用链表 + + 使用 llvmlite 访问器函数绕过 stub 类型限制： + found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub + 类型中字段不足，直接访问会被 TransPyC 静默跳过。 + """ + head: llvmlite.Value | t.CPtr = None + tail: llvmlite.Value | t.CPtr = None + actual_count: t.CSizeT = 0 + # memcpy 专用: 收集前三个 Value 的独立引用（避免共享 alloca 对象的 Next 被污染） + mc_dst1: llvmlite.Value | t.CPtr = None + mc_src1: llvmlite.Value | t.CPtr = None + mc_num1: llvmlite.Value | t.CPtr = None + + # 遍历函数参数链表（使用访问器获取 Head） + param_idx: t.CSizeT = 0 + cur_param: llvmlite.Param | t.CPtr = llvmlite.function_get_param_head(found_func) + + while cur_param is not None: + arg_val: llvmlite.Value | t.CPtr = None + if param_idx < can: + # 优先使用位置参数 + arg: ast.AST | t.CPtr = cargs.get(param_idx) + if arg is not None: + arg_val = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + else: + # 查找匹配的关键字参数 + param_name: t.CChar | t.CPtr = llvmlite.param_get_name(cur_param) + if param_name is not None: + for ki in range(kw_count): + kw: ast.Keyword | t.CPtr = (ast.Keyword | t.CPtr)(keywords.get(ki)) + if kw is None or kw.arg is None or kw.value is None: + continue + if _kwarg_name_matches(kw.arg, param_name) == 1: + arg_val = HandlesExpr.translate_value( + builder, pool, mod, kw.value, None, 0, trans) + break + + if arg_val is not None: + # 类型转换（按参数类型，使用访问器） + param_ty: llvmlite.LLVMType | t.CPtr = llvmlite.param_get_ty(cur_param) + if param_ty is not None: + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, param_ty) + arg_val.Next = None + if head is None: + head = arg_val + else: + tail.Next = arg_val + tail = arg_val + # memcpy 专用: 保存前三个参数的独立引用 + if actual_count == 0: + mc_dst1 = arg_val + elif actual_count == 1: + mc_src1 = arg_val + elif actual_count == 2: + mc_num1 = arg_val + actual_count += 1 + + param_idx += 1 + cur_param = llvmlite.param_get_next(cur_param) + + # 使用模块中的实际函数名（可能带 SHA1 前缀，使用访问器） + # t.CExtern/t.State/t.CExport 函数（declare）使用裸名，不加 SHA1 前缀 + call_name: str = func_name + if llvmlite.function_is_declared(found_func) == 0: + found_name: t.CChar | t.CPtr = llvmlite.function_get_name(found_func) + if found_name is not None: + call_name = found_name + + # 返回类型（使用访问器） + call_ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(found_func) + if call_ret_ty is None: + call_ret_ty = _infer_external_func_ret_ty(pool, func_name) + + # memcpy → @llvm.memcpy 内联函数（避免 monomorphization 跨模块 @memcpy 声明缺失） + if func_name == "memcpy" and actual_count >= 3: + return _emit_llvm_memcpy_intrinsic(pool, builder, mod, mc_dst1, mc_src1, mc_num1) + return llvmlite.build_call(builder, call_name, head, actual_count, call_ret_ty, 0) + + +# ============================================================ +# _translate_t_type_cast - 翻译 t.XXX(value) 类型转换 +# +# 当 func 是 t.XXX 形式且 XXX 是已知类型名（map_t_type 能映射）时， +# 将 value 转换为目标类型： +# - 指针 → 整数: ptrtoint (如 t.CUInt64T(ptr) → i64) +# - 整数 → 指针: inttoptr (如 t.CPtr(int_val) → i8*) +# - 整数 → 整数: trunc/zext (如 t.CInt8T(i32_val) → i8) +# - 指针 → 指针: bitcast (如 t.CPtr(ptr_val) → i8*) +# ============================================================ +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + val: llvmlite.Value | t.CPtr, + target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """类型转换：根据参数类型和目标类型选择转换方式""" + if val is None or target_ty is None: + return None + + val_bits: int = HandlesExpr.get_llvm_type_bits(val.Ty) + target_bits: int = HandlesExpr.get_llvm_type_bits(target_ty) + + # 整数 → 整数: trunc/zext + if val_bits != 0 and target_bits != 0: + if val_bits == target_bits: + return val + elif val_bits < target_bits: + return llvmlite.build_zext(builder, val, target_ty) + else: + return llvmlite.build_trunc(builder, val, target_ty) + + # 指针 → 整数: ptrtoint + if val_bits == 0 and target_bits != 0: + return llvmlite.build_ptrtoint(builder, val, target_ty) + + # 整数 → 指针: inttoptr + if val_bits != 0 and target_bits == 0: + return llvmlite.build_inttoptr(builder, val, target_ty) + + # 指针 → 指针: bitcast + return llvmlite.build_bitcast(builder, val, target_ty) + + +# ============================================================ +# 泛型实例化辅助：从 Subscript slice 提取类型实参名 +# ============================================================ +# _get_type_name_from_annotation 已移至 HandlesType.py（支持 BinOp/Attribute） +# HandlesExprCall 通过 HandlesType._get_type_name_from_annotation 调用 + + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, + slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: + """从 Subscript.slice 提取类型实参名列表 + + list[int] -> ["int"] + dict[str, int] -> ["str", "int"] + list[AST | t.CPtr] -> ["AST|t.CPtr"] + """ + if slice_node is None or pool is None: + return None + result: list[str] | t.CPtr = list[str](pool, 4) + k: int = slice_node.kind() + if k == ast.ASTKind.Tuple: + tp: ast.Tuple | t.CPtr = (ast.Tuple | t.CPtr)(slice_node) + elts: list[ast.AST | t.CPtr] | t.CPtr = tp.elts + if elts is not None: + n: t.CSizeT = elts.__len__() + i: t.CSizeT + for i in range(n): + elt: ast.AST | t.CPtr = elts.get(i) + tn: str = HandlesType._get_type_name_from_annotation(pool, elt) + if tn is not None: + result.append(tn) + else: + tn: str = HandlesType._get_type_name_from_annotation(pool, slice_node) + if tn is not None: + result.append(tn) + return result + + +# ============================================================ +# 翻译函数调用 Call(func, args) — 模块级版本 +# ============================================================ +def translate_call(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + funcs_ptr: t.CPtr, + func_count: int, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译函数调用，返回 call 结果 Value""" + cl: ast.Call | t.CPtr = (ast.Call | t.CPtr)(node) + if cl is None: + return None + + func_name: str = HandlesExpr.get_func_name(cl.func) + if func_name is None: + # 检查是否是 list[int](pool) 形式的泛型实例化调用 + # Subscript 节点作为 Call func: 泛型类构造函数 + if cl.func is not None and cl.func.kind() == ast.ASTKind.Subscript and trans is not None: + sub_node: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(cl.func) + if sub_node.value is not None and sub_node.value.kind() == ast.ASTKind.Name: + gen_nm_node: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub_node.value) + gen_class_name: str = gen_nm_node.id + # 检查是否是泛型类模板 + template_cd: ast.ClassDef | t.CPtr = HandlesClassDef._find_generic_template(gen_class_name) + if template_cd is not None: + # 提取类型实参 + gen_type_args: list[str] | t.CPtr = _extract_type_args_from_slice(pool, sub_node.slice) + if gen_type_args is not None and gen_type_args.__len__() > 0: + # 特化泛型类，返回特化类名 + spec_name: str = HandlesClassDef._specialize_generic_class( + trans, gen_class_name, gen_type_args) + if spec_name is not None: + func_name = spec_name + else: + # 泛型类模板未找到：可能未 import 定义模块 + gt_err: str = pool.alloc(256) + if gt_err is not None: + viperlib.snprintf(gt_err, 256, "泛型类 '%s' 未找到模板，可能未 import 定义模块", gen_class_name) + HandlesType.fatal_error(gen_nm_node, gt_err) + HandlesType.fatal_error(gen_nm_node, "泛型类模板未找到") + if func_name is None: + # 检查是否是 (A | t.CPtr)(value) 形式的类型转换 + # BinOp 节点作为 Call func: 联合类型注解用作类型转换 + if cl.func is not None and cl.func.kind() == ast.ASTKind.BinOp and trans is not None: + binop_ty: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, cl.func, trans._imported_modules, trans._from_imports) + if binop_ty is not None: + binop_args: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if binop_args is not None and binop_args.__len__() == 1: + binop_arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, binop_args.get(0), funcs_ptr, func_count, trans) + if binop_arg_val is not None: + return _translate_t_type_cast(pool, builder, binop_arg_val, binop_ty) + return None + + # 检测 c.Deref / c.DerefAs / c.Addr 等特殊调用（内联 IR，不生成 call） + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + at_node: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if at_node.value is not None and at_node.value.kind() == ast.ASTKind.Name: + mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at_node.value) + if mod_nm.id is not None and string.strcmp(mod_nm.id, "c") == 0: + if string.strcmp(func_name, "Deref") == 0: + return translate_c_deref(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "DerefAs") == 0: + return translate_c_derefas(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "Addr") == 0: + return translate_c_addr(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "Load") == 0: + return translate_c_load(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "Asm") == 0: + return translate_c_asm(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "LLVMIR") == 0: + return translate_c_llvmir(pool, builder, mod, cl, trans) + elif string.strcmp(func_name, "LInp") == 0 or string.strcmp(func_name, "LOut") == 0: + # c.LInp/c.LOut 独立调用时返回 0（仅在 c.LLVMIR 的 f-string 内有意义） + return llvmlite.ConstInt(pool, llvmlite.Int32(pool), 0, "0") + + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + can: t.CSizeT = 0 + if cargs is not None: + can = cargs.__len__() + + # 检测 t.XXX 类型转换: t.CUInt64T(ptr), t.CPtr(val), t.CInt(val) 等 + # 当 func 是 t.XXX 形式且 XXX 是已知类型名时，当作类型转换处理（而非函数调用） + # 支持: + # 单参数: t.CInt(val) → 目标类型 = map_t_type(func_name) + # 双参数: t.CInt(val, t.CPtr) → 目标类型 = Ptr(map_t_type(func_name)) + # t 是内部库，一旦确认为类型转换，绝不跌落到跨模块调用 FATAL 路径 + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + tc_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if tc_at.value is not None and tc_at.value.kind() == ast.ASTKind.Name: + tc_mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(tc_at.value) + if tc_mod_nm.id is not None and string.strcmp(tc_mod_nm.id, "t") == 0: + target_ty: llvmlite.LLVMType | t.CPtr = HandlesType.map_t_type(pool, func_name) + if target_ty is not None: + tc_effective_can: t.CSizeT = can + # 双参数形式: t.CInt(val, t.CPtr) → Ptr(target_ty) + # 注意: void* 在 LLVM 中非法，CVoid* 应表示为 i8* + if can == 2: + tc_arg2_node: ast.AST | t.CPtr = cargs.get(1) + if tc_arg2_node is not None and tc_arg2_node.kind() == ast.ASTKind.Attribute: + tc_arg2_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(tc_arg2_node) + if tc_arg2_at.attr is not None and string.strcmp(tc_arg2_at.attr, "CPtr") == 0: + if tc_arg2_at.value is not None and tc_arg2_at.value.kind() == ast.ASTKind.Name: + tc_arg2_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(tc_arg2_at.value) + if tc_arg2_nm.id is not None and string.strcmp(tc_arg2_nm.id, "t") == 0: + # CVoid* → i8*（LLVM 中 void* 表示为 i8*） + match target_ty: + case llvmlite.LLVMType.Void(): + target_ty = llvmlite.Int8(pool) + case _: + pass + target_ty = llvmlite.Ptr(pool, target_ty) + tc_effective_can = 1 + if tc_effective_can == 1: + tc_arg_node: ast.AST | t.CPtr = cargs.get(0) + if tc_arg_node is not None: + tc_arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, tc_arg_node, None, 0, trans) + if tc_arg_val is not None: + return _translate_t_type_cast(pool, builder, tc_arg_val, target_ty) + # t.XXX 类型转换已确认但不满足参数条件：返回 None，不跌落到跨模块 FATAL + return None + + # 检测 ClassName.__sizeof__() — 返回结构体大小常量（i64） + # __sizeof__ 是内置方法，不走跨模块调用路径 + # 支持: Argument.__sizeof__() 和 w32.win32file.WIN32_FIND_DATAA.__sizeof__() + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + sz_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if sz_at.attr is not None and string.strcmp(sz_at.attr, "__sizeof__") == 0: + if sz_at.value is not None and can == 0: + sz_cls_name: str = None + sz_vk: int = sz_at.value.kind() + if sz_vk == ast.ASTKind.Name: + sz_cls_nm2: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sz_at.value) + sz_cls_name = sz_cls_nm2.id + elif sz_vk == ast.ASTKind.Attribute: + sz_inner_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(sz_at.value) + sz_cls_name = sz_inner_at.attr + if sz_cls_name is not None: + # 泛型类型参数 T 替换: T.__sizeof__() → int.__sizeof__() + sz_cls_name = HandlesType.resolve_generic_type_name(sz_cls_name) + # 联合类型（如 AST|t.CPtr）按指针大小处理（8 字节） + # 联合类型在 TPV 中表示指针，可以指向多种类型 + # 对应 TPC 中对 BinOp/BitOr 联合类型的特殊处理 + if string.strstr(sz_cls_name, "|") is not None: + return llvmlite.const_int64(pool, 8) + # 内置类型 sizeof（int=4, str/bytes=8 指针, double=8 等） + sz_builtin: int = -1 + if string.strcmp(sz_cls_name, "int") == 0: + sz_builtin = 4 + elif string.strcmp(sz_cls_name, "str") == 0: + sz_builtin = 8 + elif string.strcmp(sz_cls_name, "bytes") == 0: + sz_builtin = 8 + elif string.strcmp(sz_cls_name, "CInt") == 0: + sz_builtin = 4 + elif string.strcmp(sz_cls_name, "CInt8T") == 0: + sz_builtin = 1 + elif string.strcmp(sz_cls_name, "CInt16T") == 0: + sz_builtin = 2 + elif string.strcmp(sz_cls_name, "CInt32T") == 0: + sz_builtin = 4 + elif string.strcmp(sz_cls_name, "CInt64T") == 0: + sz_builtin = 8 + elif string.strcmp(sz_cls_name, "CChar") == 0: + sz_builtin = 1 + elif string.strcmp(sz_cls_name, "CDouble") == 0: + sz_builtin = 8 + elif string.strcmp(sz_cls_name, "CFloat") == 0: + sz_builtin = 4 + elif string.strcmp(sz_cls_name, "CPtr") == 0: + sz_builtin = 8 + elif string.strcmp(sz_cls_name, "CSizeT") == 0: + sz_builtin = 8 + if sz_builtin >= 0: + # 注意: 必须用 const_int64 而非 ConstInt(...,"0") + # ConstInt 的 name 参数是 IR 文本表示，ValuePrint 输出时使用 Name 而非 IntVal + # 硬编码 "0" 会导致 IR 输出 i64 0 而非实际值 + return llvmlite.const_int64(pool, sz_builtin) + sz_struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type( + sz_cls_name) + if sz_struct_ty is not None: + sz_val: int = HandlesClassDef._get_type_size(sz_struct_ty) + if sz_val > 0: + return llvmlite.const_int64(pool, sz_val) + # 类型已注册但大小为 0（opaque struct 或字段链表未填充）：报错终止 + stdio.printf("[FATAL][__sizeof__] 类型 '%s' 已注册但大小为 0（字段链表未填充），立即终止编译\n", sz_cls_name) + sys.exit(1) + # 类型未注册：报错终止，不静默返回 0（会导致 malloc(0)/alloc(0) 堆溢出） + stdio.printf("[FATAL][__sizeof__] 类型 '%s' 未在结构体表中注册，无法计算大小，立即终止编译\n", sz_cls_name) + sys.exit(1) + + # 检测 obj.__len__() — 返回 list 对象的 __count__ 字段（偏移 8，i64） + # __len__ 是 list[T] 的内置方法，list 是泛型类不注册 struct + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + len_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if len_at.attr is not None and string.strcmp(len_at.attr, "__len__") == 0: + if len_at.value is not None and can == 0: + obj_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_name_value( + builder, pool, len_at.value, trans) + if obj_val is not None: + i64_ty_len: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + # 注意: 必须用 const_int64 而非 ConstInt(...,"0") + # ConstInt 的 name 参数 "0" 会被当作 IR 文本值输出 i64 0 + # 导致 GEP 索引为 0（访问 __data__）而非 1（访问 __count__） + idx_one: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 1) + count_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep( + builder, i64_ty_len, obj_val, idx_one) + if count_ptr is not None: + return llvmlite.build_load(builder, i64_ty_len, count_ptr) + + # 检测方法调用: obj.method(args) + # 当 cl.func 是 Attribute 且 obj 是已注册的结构体变量时，翻译为方法调用 + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + method_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if method_at is not None and method_at.value is not None: + if method_at.value.kind() == ast.ASTKind.Name and trans is not None: + method_obj_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(method_at.value) + if method_obj_nm.id is not None: + # 查找对象变量 + obj_var: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, method_obj_nm.id) + if obj_var is not None and obj_var.Ty is not None: + # 检查是否为 Ptr(Struct(...))（结构体值变量） + # 或 Ptr(Ptr(Struct(...)))（指针变量，需 load） + match obj_var.Ty: + case llvmlite.LLVMType.Ptr(obj_struct_ty): + class_nm: str = HandlesStruct.get_class_name_by_type( + pool, obj_struct_ty) + if class_nm is not None: + return _translate_method_call( + pool, builder, mod, class_nm, func_name, + obj_var, cl, trans) + # 指针变量: alloca 类型是 Ptr(Ptr(Struct)) + # load 获取实际指针 Ptr(Struct) 后再查找 + match obj_struct_ty: + case llvmlite.LLVMType.Ptr(inner_struct_ty): + inner_nm: str = HandlesStruct.get_class_name_by_type( + pool, inner_struct_ty) + if inner_nm is not None: + obj_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, obj_struct_ty, obj_var) + if obj_ptr is not None: + return _translate_method_call( + pool, builder, mod, inner_nm, func_name, + obj_ptr, cl, trans) + # AnnotClassName 回退：联合类型简化为 Ptr(i8) 时， + # 通过原始类型注解的类名查找结构体 + var_entry: HandlesVar.VarEntry | t.CPtr = HandlesVar.lookup_var_entry( + trans.SymTab, method_obj_nm.id) + if var_entry is not None and var_entry.AnnotClassName is not None: + annot_val: str = var_entry.AnnotClassName + # 解析模块限定名 "memhub.MemBuddy" → mod_nm="memhub", cls_nm="MemBuddy" + dot_ptr: t.CChar | t.CPtr = string.strrchr(annot_val, '.') + cls_nm2: str = annot_val + mod_nm2: str = None + if dot_ptr is not None: + cls_nm2 = dot_ptr + 1 + total_len_av: t.CSizeT = string.strlen(annot_val) + cls_len_av: t.CSizeT = string.strlen(cls_nm2) + mod_len2: t.CSizeT = total_len_av - cls_len_av - 1 + mod_nm2 = pool.alloc(mod_len2 + 1) + if mod_nm2 is not None: + string.strncpy(mod_nm2, annot_val, mod_len2) + mod_nm2[mod_len2] = '\0' + # 检查结构体是否已注册 + if HandlesStruct.get_struct_type(cls_nm2) is not None: + # 结构体已注册：load obj_ptr 走方法调用 + i8_ptr_ty2: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr( + pool, llvmlite.Int8(pool)) + obj_ptr2: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, i8_ptr_ty2, obj_var) + if obj_ptr2 is not None: + return _translate_method_call( + pool, builder, mod, cls_nm2, func_name, + obj_ptr2, cl, trans) + else: + # 结构体未注册（跨模块依赖未翻译）： + # 用模块名查 SHA1 构造方法调用名 + sha1_fc: str = None + if mod_nm2 is not None: + sha1_fc = _lookup_module_sha1(mod_nm2) + if sha1_fc is not None: + # 构造 "{sha1}.{cls_nm2}.{func_name}" 调用名 + mlen3: t.CSizeT = string.strlen(sha1_fc) + string.strlen(cls_nm2) + string.strlen(func_name) + 3 + mangled3: str = pool.alloc(mlen3) + if mangled3 is not None: + viperlib.snprintf(mangled3, mlen3, "%s.%s.%s", sha1_fc, cls_nm2, func_name) + # load obj_ptr (self) + i8_ptr_ty3: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr( + pool, llvmlite.Int8(pool)) + obj_ptr3: llvmlite.Value | t.CPtr = llvmlite.build_load( + builder, i8_ptr_ty3, obj_var) + if obj_ptr3 is not None: + # 构建参数链表: obj_ptr3 (self) → 其他参数 + llvmlite.value_set_next(obj_ptr3, None) + head3: llvmlite.Value | t.CPtr = obj_ptr3 + tail3: llvmlite.Value | t.CPtr = obj_ptr3 + for ai3 in range(can): + arg3: ast.AST | t.CPtr = cargs.get(ai3) + if arg3 is not None: + arg_val3: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg3, None, 0, trans) + if arg_val3 is not None: + llvmlite.value_set_next(arg_val3, None) + llvmlite.value_set_next(tail3, arg_val3) + tail3 = arg_val3 + total3: int = 1 + can + ret_ty3: llvmlite.LLVMType | t.CPtr = _infer_method_ret_ty(pool, func_name) + return llvmlite.build_call(builder, mangled3, head3, total3, ret_ty3, 0) + + # ============================================================ + # 检测 self.field.method() 调用 + # + # 当 method_at.value 是 Attribute（字段访问，如 self.__pool__）时， + # 翻译字段值获取对象指针，然后走方法调用路径。 + # + # 路径1: 从字段值的 LLVM 类型获取类名（字段类型为 Ptr(Struct("MemManager"))） + # 路径2: 从字段的 AnnotClassName 回退查找（字段类型被简化为 Ptr(i8) 时） + # ============================================================ + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + attr_method_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if attr_method_at is not None and attr_method_at.value is not None: + if attr_method_at.value.kind() == ast.ASTKind.Attribute and trans is not None: + inner_attr: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(attr_method_at.value) + if inner_attr is not None and inner_attr.attr is not None: + # 翻译 self.__pool__ 获取字段值（已 load 的对象指针） + field_obj: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, attr_method_at.value, funcs_ptr, func_count, trans) + if field_obj is not None and field_obj.Ty is not None: + field_cls_nm: str = None + # 路径1: 从字段值的 LLVM 类型获取类名 + match field_obj.Ty: + case llvmlite.LLVMType.Ptr(field_struct_ty): + field_cls_nm = HandlesStruct.get_class_name_by_type( + pool, field_struct_ty) + # 路径2: 从字段注解类名回退查找 + if field_cls_nm is None: + if inner_attr.value is not None and inner_attr.value.kind() == ast.ASTKind.Name: + inner_name_node: ast.Name | t.CPtr = (ast.Name | t.CPtr)(inner_attr.value) + if inner_name_node is not None and inner_name_node.id is not None: + inner_var: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, inner_name_node.id) + if inner_var is not None and inner_var.Ty is not None: + match inner_var.Ty: + case llvmlite.LLVMType.Ptr(field_owner_struct_ty): + field_info: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct.lookup_field( + field_owner_struct_ty, inner_attr.attr) + if field_info is not None: + field_annot: str = HandlesStruct.get_field_annot_class_name(field_info) + if field_annot is not None: + dot_p: t.CChar | t.CPtr = string.strrchr(field_annot, 46) + if dot_p is not None: + field_cls_nm = dot_p + 1 + else: + field_cls_nm = field_annot + if field_cls_nm is not None: + return _translate_method_call( + pool, builder, mod, field_cls_nm, func_name, + field_obj, cl, trans) + + # ============================================================ + # 检测 list 内置方法调用: lst.get(i), lst.set(i,v), lst.pop(), lst.clear(), lst.__iter__() + # list[T] 是泛型类不注册 struct，方法调用处理器（上方）无法处理， + # 需要在此特殊处理。 + # list 结构布局 (6 字段 × 8 字节): + # __data__(0) __count__(8) __capacity__(16) __pool__(24) __elem_size__(32) __iter_index__(40) + # ============================================================ + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + lm_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if lm_at is not None and lm_at.attr is not None and lm_at.value is not None: + lm_name: str = lm_at.attr + # 判断是否是 list 方法名 + lm_is_list: int = 0 + if string.strcmp(lm_name, "get") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "__getitem__") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "set") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "__setitem__") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "pop") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "clear") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "__iter__") == 0: lm_is_list = 1 + elif string.strcmp(lm_name, "append") == 0: lm_is_list = 1 + + if lm_is_list == 1 and trans is not None: + # 翻译 list 对象 + lm_obj: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, lm_at.value, funcs_ptr, func_count, trans) + if lm_obj is not None: + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + # get / __getitem__: 返回 index 处的元素 + if string.strcmp(lm_name, "get") == 0 or string.strcmp(lm_name, "__getitem__") == 0: + if can >= 1: + # 加载 __data__ (偏移 0) + g_idx0: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 0, "0") + g_dpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, g_idx0) + g_data: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, g_dpp) + # 加载 __elem_size__ (偏移 32, index 4) + g_idx4: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 4, "0") + g_epp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, g_idx4) + g_esize: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, g_epp) + # 翻译 index + g_idx_node: ast.AST | t.CPtr = cargs.get(0) + g_idx_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, g_idx_node, None, 0, trans) + if g_data is not None and g_esize is not None and g_idx_val is not None: + g_idx_i64: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + builder, g_idx_val, i64_ty) + g_offset: llvmlite.Value | t.CPtr = llvmlite.build_mul(builder, g_idx_i64, g_esize) + g_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, g_data, g_offset) + if g_addr is not None: + return llvmlite.build_load(builder, i8_ptr_ty, g_addr) + + # set / __setitem__: 设置 index 处的元素 + elif string.strcmp(lm_name, "set") == 0 or string.strcmp(lm_name, "__setitem__") == 0: + if can >= 2: + s_idx0: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 0, "0") + s_dpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, s_idx0) + s_data: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, s_dpp) + s_idx4: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 4, "0") + s_epp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, s_idx4) + s_esize: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, s_epp) + s_idx_node: ast.AST | t.CPtr = cargs.get(0) + s_val_node: ast.AST | t.CPtr = cargs.get(1) + s_idx_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, s_idx_node, None, 0, trans) + s_val_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, s_val_node, None, 0, trans) + if s_data is not None and s_esize is not None and s_idx_val is not None and s_val_val is not None: + s_idx_i64: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + builder, s_idx_val, i64_ty) + s_offset: llvmlite.Value | t.CPtr = llvmlite.build_mul(builder, s_idx_i64, s_esize) + s_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, s_data, s_offset) + if s_addr is not None: + s_val_ptr: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + builder, s_val_val, i8_ptr_ty) + llvmlite.build_store(builder, s_val_ptr, s_addr) + return None + + # pop: 返回最后一个元素，count-- + elif string.strcmp(lm_name, "pop") == 0: + p_idx1: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 1, "0") + p_cpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, p_idx1) + p_count: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, p_cpp) + p_nc: llvmlite.Value | t.CPtr = llvmlite.build_sub( + builder, p_count, llvmlite.ConstInt(pool, i64_ty, 1, "0")) + llvmlite.build_store(builder, p_nc, p_cpp) + p_idx0: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 0, "0") + p_dpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, p_idx0) + p_data: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, p_dpp) + p_idx4: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 4, "0") + p_epp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, p_idx4) + p_esize: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i64_ty, p_epp) + if p_data is not None and p_esize is not None and p_nc is not None: + p_offset: llvmlite.Value | t.CPtr = llvmlite.build_mul(builder, p_nc, p_esize) + p_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, p_data, p_offset) + if p_addr is not None: + return llvmlite.build_load(builder, i8_ptr_ty, p_addr) + + # clear: count = 0 + elif string.strcmp(lm_name, "clear") == 0: + c_idx1: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 1, "0") + c_cpp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, c_idx1) + llvmlite.build_store(builder, llvmlite.ConstInt(pool, i64_ty, 0, "0"), c_cpp) + return None + + # __iter__: iter_index = 0, return self + elif string.strcmp(lm_name, "__iter__") == 0: + it_idx5: llvmlite.Value | t.CPtr = llvmlite.ConstInt(pool, i64_ty, 5, "0") + it_ipp: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i64_ty, lm_obj, it_idx5) + llvmlite.build_store(builder, llvmlite.ConstInt(pool, i64_ty, 0, "0"), it_ipp) + return lm_obj + + # append: 直接调用 list.append(self, item) + # append 需要容量检查和内存重分配，逻辑复杂，直接调用 _list.py 中的实现 + elif string.strcmp(lm_name, "append") == 0: + if can >= 1: + ap_sha1: str = _lookup_module_sha1("_list") + if ap_sha1 is not None: + ap_mlen: t.CSizeT = string.strlen(ap_sha1) + 13 + ap_mangled: str = pool.alloc(ap_mlen) + if ap_mangled is not None: + viperlib.snprintf(ap_mangled, ap_mlen, "%s.list.append", ap_sha1) + ap_item_node: ast.AST | t.CPtr = cargs.get(0) + ap_item_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, ap_item_node, None, 0, trans) + if ap_item_val is not None: + llvmlite.value_set_next(lm_obj, None) + llvmlite.value_set_next(ap_item_val, None) + llvmlite.value_set_next(lm_obj, ap_item_val) + ap_ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + return llvmlite.build_call(builder, ap_mangled, lm_obj, 2, ap_ret_ty, 0) + + # 获取关键字参数 + keywords: list[ast.AST | t.CPtr] | t.CPtr = cl.keywords + kw_count: t.CSizeT = 0 + if keywords is not None: + kw_count = keywords.__len__() + + # 检查是否是结构体构造函数: Point() / Point(10, 20) / Point(x=10, y=20) + # 命名空间隔离：严格模式下检查结构体可见性 + # - 裸名 Class(): 需要 from X import Class 或本地定义 + # - 模块限定 Module.Class(): 需要 import Module + # 用 find_struct_by_module 规避跨模块同名 find_struct 找错 + struct_ty: llvmlite.LLVMType | t.CPtr = None + gen_ctor_entry: HandlesStruct.StructEntry | t.CPtr = None + vis: int = HandlesStruct.is_struct_visible(func_name) + if vis != 0: + if trans is not None: + gen_ctor_entry = HandlesStruct.find_struct_by_module(func_name, trans.ModuleSha1) + if gen_ctor_entry is None: + gen_ctor_entry = HandlesStruct.find_struct(func_name) + if gen_ctor_entry is not None: + struct_ty = gen_ctor_entry.Ty + elif trans is not None and cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + at_qc: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if at_qc.value is not None and at_qc.value.kind() == ast.ASTKind.Name: + mod_nm_qc: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at_qc.value) + imp_ok: int = HandlesImports.is_module_imported(trans._imported_modules, mod_nm_qc.id) + if imp_ok != 0: + if trans is not None: + gen_ctor_entry = HandlesStruct.find_struct_by_module(func_name, trans.ModuleSha1) + if gen_ctor_entry is None: + gen_ctor_entry = HandlesStruct.find_struct(func_name) + if gen_ctor_entry is not None: + struct_ty = gen_ctor_entry.Ty + if struct_ty is not None: + if can == 0 and kw_count == 0: + # OOP 结构体：即使无参数也调用 __before_init__ + # 用 gen_ctor_entry 规避跨模块同名 find_struct 找错 + ctor_is_oop: int = 0 + if gen_ctor_entry is not None: + ctor_is_oop = gen_ctor_entry.IsOOP + if ctor_is_oop != 0: + return _translate_struct_ctor( + pool, builder, mod, struct_ty, func_name, cargs, 0, trans) + # 普通结构体：检查是否有默认值（用 entry 直接访问） + gen_has_defaults: int = 0 + if gen_ctor_entry is not None: + for gdi in range(gen_ctor_entry.FieldCount): + gdf: HandlesStruct.FieldEntry | t.CPtr = HandlesStruct._get_field_entry( + gen_ctor_entry, gdi) + if gdf is not None and gdf.DefaultVal is not None: + gen_has_defaults = 1 + break + else: + gen_has_defaults = HandlesStruct.struct_has_defaults(func_name) + if gen_has_defaults == 1: + # 有默认值：走构造函数路径应用默认值 + return _translate_struct_ctor( + pool, builder, mod, struct_ty, func_name, cargs, 0, trans) + # 无默认值：返回 zeroinitializer + ret: llvmlite.Value | t.CPtr = llvmlite.ConstZero(pool, struct_ty) + return ret + if kw_count > 0: + # 有关键字参数：按字段名赋值 + return _translate_struct_ctor_kw( + pool, builder, mod, struct_ty, func_name, cargs, can, + keywords, kw_count, trans) + # 仅位置参数：按字段顺序构建 + return _translate_struct_ctor( + pool, builder, mod, struct_ty, func_name, cargs, can, trans) + + # 查找函数：优先从函数表按裸名精确匹配（同模块函数）， + # 再从模块函数链表按后缀匹配（SHA1 前缀名）。 + # stubs 在翻译后才注入，翻译时跨模块函数 found_func 通常为 None， + # 此时根据函数名推断 includes 库函数的返回类型（如 strchr 返回 i8*） + # 如果调用方未传 funcs_ptr/func_count（如 AnnAssign 路径），从 trans 获取 + actual_funcs: HandlesExprCall.FuncEntry | t.CPtr = funcs_ptr + actual_func_count: int = func_count + if actual_funcs is None and trans is not None: + actual_funcs = trans._funcs + actual_func_count = trans._func_count + found_func: llvmlite.Function | t.CPtr = find_func_in_module(mod, func_name) + # 使用访问器绕过 stub 类型限制 + found_ret_ty: llvmlite.LLVMType | t.CPtr = None + found_params: t.CPtr = None + if found_func is not None: + found_ret_ty = llvmlite.function_get_ret_ty(found_func) + found_params = llvmlite.function_get_params(found_func) + if found_ret_ty is not None: + call_ret_ty: llvmlite.LLVMType | t.CPtr = found_ret_ty + else: + call_ret_ty = _infer_external_func_ret_ty(pool, func_name) + + # 如果模块中找不到函数，检查是否是变量（闭包调用） + if found_func is None: + closure_ptr: llvmlite.Value | t.CPtr = HandlesVar.lookup_var(trans.SymTab, func_name) + if closure_ptr is not None: + return translate_closure_call(pool, builder, mod, closure_ptr, + cargs, can) + + # 翻译参数并链接 + # 有关键字参数且找到函数定义时，按参数名匹配 + if kw_count > 0 and found_func is not None and found_params is not None: + return _translate_call_with_kwargs( + pool, builder, mod, func_name, found_func, + cargs, can, keywords, kw_count, trans) + + head: llvmlite.Value | t.CPtr = None + tail: llvmlite.Value | t.CPtr = None + # memcpy 专用: 收集前三个 Value 的独立引用（避免共享 alloca 对象的 Next 被污染） + mc_dst2: llvmlite.Value | t.CPtr = None + mc_src2: llvmlite.Value | t.CPtr = None + mc_num2: llvmlite.Value | t.CPtr = None + for ai in range(can): + arg: ast.AST | t.CPtr = cargs.get(ai) + if arg is not None: + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + if arg_val is not None: + # 参数类型提升：当 found_func 为 None（stub 未注入）时， + # 根据外部函数签名推断期望类型，提升字面量（如 i32→i64） + # 避免 declare(i64) 与 call(i32) 类型不匹配 + if found_func is None and arg_val.Ty is not None: + ext_param_ty: llvmlite.LLVMType | t.CPtr = _get_external_func_param_ty( + pool, func_name, ai) + if ext_param_ty is not None: + val_bits_ap: int = HandlesExpr.get_llvm_type_bits(arg_val.Ty) + tgt_bits_ap: int = HandlesExpr.get_llvm_type_bits(ext_param_ty) + if val_bits_ap != 0 and tgt_bits_ap != 0 and val_bits_ap != tgt_bits_ap: + if val_bits_ap == 1 and val_bits_ap < tgt_bits_ap: + arg_val = llvmlite.build_zext(builder, arg_val, ext_param_ty) + else: + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, ext_param_ty) + arg_val.Next = None + if head is None: + head = arg_val + else: + tail.Next = arg_val + tail = arg_val + # memcpy 专用: 保存前三个参数的独立引用 + if ai == 0: + mc_dst2 = arg_val + elif ai == 1: + mc_src2 = arg_val + elif ai == 2: + mc_num2 = arg_val + + # 使用模块中的实际函数名（可能带 SHA1 前缀，如 "83210a72e829a086.atoi"） + # t.CExtern/t.State/t.CExport 函数不加 SHA1 前缀，使用裸名 + # declare 函数（IsDeclared=1）使用裸名（func_name），define 函数使用实际名（可能带 SHA1 前缀） + call_name: str = func_name + if found_func is not None: + if llvmlite.function_is_declared(found_func) == 0: + found_name: t.CChar | t.CPtr = llvmlite.function_get_name(found_func) + if found_name is not None: + call_name = found_name + else: + # 跨模块函数调用：found_func 为 None 时，检查是否是模块限定调用 + # 如 string.strchr(...) → @{sha1_of_string}.strchr + if cl.func is not None and cl.func.kind() == ast.ASTKind.Attribute: + xmod_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(cl.func) + if xmod_at.value is not None and xmod_at.value.kind() == ast.ASTKind.Name: + xmod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(xmod_at.value) + if xmod_nm.id is not None: + mod_sha1: str = _lookup_module_sha1(xmod_nm.id) + if mod_sha1 is not None: + # CExport 函数定义用裸名（@strlen），跨模块调用需用裸名而非 @{sha1}.func + if is_cexport_func(mod_sha1, func_name) != 0: + call_name = func_name + else: + mangled_len: t.CSizeT = string.strlen(mod_sha1) + string.strlen(func_name) + 2 + mangled: str = pool.alloc(mangled_len) + if mangled is not None: + viperlib.snprintf(mangled, mangled_len, "%s.%s", mod_sha1, func_name) + call_name = mangled + else: + # t 和 c 是内部库，豁免 SHA1 查找：使用裸名调用 + if string.strcmp(xmod_nm.id, "t") == 0 or string.strcmp(xmod_nm.id, "c") == 0: + call_name = func_name + else: + # Bug 修复: 先检查是否是已导入的模块名（如 BuildPipeline.run_pipeline） + # 如果是已导入模块但 SHA1 未注册，说明该模块未被编译（如 Phase2 未扫描到）， + # 应该立即报错，而不是错误地走 from_imports 查找（会被 star import 回退误导） + is_imported_mod: int = 0 + if trans is not None and trans._imported_modules is not None: + is_imported_mod = HandlesImports.is_module_imported( + trans._imported_modules, xmod_nm.id) + if is_imported_mod != 0: + # 已导入模块但 SHA1 未注册：该模块未被编译，立即报错 + err_msg_imp: t.CChar | t.CPtr = pool.alloc(256) + if err_msg_imp is not None: + viperlib.snprintf(err_msg_imp, 256, "模块 '%s' 已导入但 SHA1 未注册（该模块未被编译）", xmod_nm.id) + HandlesType.fatal_error(xmod_nm, err_msg_imp) + HandlesType.fatal_error(xmod_nm, "已导入模块 SHA1 未注册") + else: + # 检查是否是通过 from import 导入的名称（如 from .__types import LLVMType） + # LLVMType.Ptr(ty) 中 LLVMType 是类名不是模块名， + # 但 Ptr 是 __types 模块的模块级函数 + # 禁用 star import 回退：xmod_nm.id 是明确名称，不应回退到 star import 模块 + from_sha1: str = None + if trans is not None and trans._from_imports is not None: + from_sha1 = _lookup_mod_sha1(pool, trans._from_imports, xmod_nm.id, 0) + if from_sha1 is not None: + # CExport 函数定义用裸名，跨模块调用需用裸名 + if is_cexport_func(from_sha1, func_name) != 0: + call_name = func_name + else: + fmangled_len: t.CSizeT = string.strlen(from_sha1) + string.strlen(func_name) + 2 + fmangled: str = pool.alloc(fmangled_len) + if fmangled is not None: + viperlib.snprintf(fmangled, fmangled_len, "%s.%s", from_sha1, func_name) + call_name = fmangled + else: + err_msg2: t.CChar | t.CPtr = pool.alloc(256) + if err_msg2 is not None: + viperlib.snprintf(err_msg2, 256, "跨模块调用 '%s.%s' 找不到模块 SHA1", xmod_nm.id, func_name) + HandlesType.fatal_error(xmod_nm, err_msg2) + HandlesType.fatal_error(xmod_nm, "跨模块调用找不到模块 SHA1") + else: + # 裸名调用（非 Attribute）：检查 from_imports + # 裸名调用允许 star import 回退：from stdint import * 导入的名称（如 BYTEPTR）需要通过 star import 查找 + if trans is not None and trans._from_imports is not None: + bare_sha1: str = _lookup_mod_sha1(pool, trans._from_imports, func_name, 1) + if bare_sha1 is not None: + if is_cexport_func(bare_sha1, func_name) != 0: + call_name = func_name + else: + bare_mangled_len: t.CSizeT = string.strlen(bare_sha1) + string.strlen(func_name) + 2 + bare_mangled: str = pool.alloc(bare_mangled_len) + if bare_mangled is not None: + viperlib.snprintf(bare_mangled, bare_mangled_len, "%s.%s", bare_sha1, func_name) + call_name = bare_mangled + else: + err_msg3: t.CChar | t.CPtr = pool.alloc(256) + if err_msg3 is not None: + viperlib.snprintf(err_msg3, 256, "裸名调用 '%s' 未定义且未导入", func_name) + HandlesType.fatal_error(node, err_msg3) + HandlesType.fatal_error(node, "裸名调用未定义且未导入") + else: + err_msg4: t.CChar | t.CPtr = pool.alloc(256) + if err_msg4 is not None: + viperlib.snprintf(err_msg4, 256, "裸名调用 '%s' 未定义且未导入", func_name) + HandlesType.fatal_error(node, err_msg4) + HandlesType.fatal_error(node, "裸名调用未定义且未导入") + + # 直接调用 + # memcpy → @llvm.memcpy 内联函数（避免 monomorphization 跨模块 @memcpy 声明缺失） + if func_name == "memcpy" and can >= 3: + return _emit_llvm_memcpy_intrinsic(pool, builder, mod, mc_dst2, mc_src2, mc_num2) + return llvmlite.build_call(builder, call_name, head, can, call_ret_ty, 0) + + +# ============================================================ +# 翻译闭包调用 — 通过闭包指针间接调用 +# +# 闭包结构: {i8* fn_ptr, i8* env_ptr} (16 字节) +# 调用过程: +# 1. load closure_ptr 从变量 alloca +# 2. load fn_ptr 从 offset 0 +# 3. load env_ptr 从 offset 8 +# 4. bitcast fn_ptr to i32 (i8*)* +# 5. call i32 %fn_typed(i8* %env_ptr) +# ============================================================ +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + closure_alloca: llvmlite.Value | t.CPtr, + cargs: list[ast.AST | t.CPtr] | t.CPtr, + can: t.CSizeT) -> llvmlite.Value | t.CPtr: + """通过闭包指针间接调用""" + if builder is None or closure_alloca is None: + return None + + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + + # 1. 加载闭包指针 (i8*) + closure_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, closure_alloca) + if closure_ptr is None: + return None + + # 2. 加载 fn_ptr (offset 0): bitcast to i8**, load i8* + fn_ptr_addr: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, closure_ptr, llvmlite.Ptr(pool, i8_ptr_ty)) + if fn_ptr_addr is None: + return None + fn_raw: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, fn_ptr_addr) + if fn_raw is None: + return None + + # 3. 加载 env_ptr (offset 8): GEP i8, closure, 8; load i8* + eight_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 8) + env_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, closure_ptr, eight_val) + if env_addr is None: + return None + env_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, env_addr) + if env_ptr is None: + return None + + # 4. bitcast fn_raw to i32 (i8*)* + param_node: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, i8_ptr_ty) + func_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Func(pool, i32_ty, param_node, 1) + func_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, func_ty) + fn_typed: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, fn_raw, func_ptr_ty) + if fn_typed is None: + return None + + # 5. 间接调用: call i32 %fn_typed(i8* %env_ptr) + env_ptr.Next = None + return llvmlite.build_call_indirect(builder, fn_typed, env_ptr, 1, i32_ty) + + +# ============================================================ +# 翻译 printf 调用 — 模块级版本 +# ============================================================ +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + cl: ast.Call | t.CPtr, + trans: HT.Translator | t.CPtr = None): + """翻译 printf 调用（特殊处理格式字符串）""" + cargs: list[ast.AST | t.CPtr] | t.CPtr = cl.args + if cargs is None: + return + can: t.CSizeT = cargs.__len__() + if can == 0: + return + + arg0: ast.AST | t.CPtr = cargs.get(0) + if arg0 is None: + return + + fmt_str: str = None + arg0_kind: int = arg0.kind() + if arg0_kind == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(arg0) + if cn.const_kind == ast.CONST_STR: + fmt_str = cn.str_val + + if fmt_str is None: + return + + fmt_ptr: llvmlite.Value | t.CPtr = HandlesExpr.create_global_string( + builder, pool, mod, fmt_str, trans) + if fmt_ptr is None: + return + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + + # 翻译额外参数并链接 + prev: llvmlite.Value | t.CPtr = fmt_ptr + for ai in range(1, can): + arg: ast.AST | t.CPtr = cargs.get(ai) + if arg is not None: + arg_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, arg, None, 0, trans) + if arg_val is not None: + # C ABI integer promotion: i8/i16 → i32（printf 变参函数要求） + arg_bits: int = HandlesExpr.get_llvm_type_bits(arg_val.Ty) + if arg_bits != 0 and arg_bits < 32: + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, i32_ty) + # C ABI float promotion: float → double（printf 变参函数要求） + arg_fbits: int = HandlesExpr.get_llvm_float_bits(arg_val.Ty) + if arg_fbits != 0 and arg_fbits < 64: + double_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Double(pool) + arg_val = HandlesExpr.coerce_to_type(builder, arg_val, double_ty) + arg_val.Next = None + prev.Next = arg_val + prev = arg_val + prev.Next = None + + # is_variadic=1: 生成带显式函数类型签名的 call（如 call i32 (i8*, ...) @printf）， + # 让 llc 识别为 variadic 调用，生成正确的 Win64 ABI 代码（浮点参数同时传 XMM 和整数寄存器） + llvmlite.build_call(builder, "printf", fmt_ptr, can, i32_ty, 1) + + +# ============================================================ +# FuncEntry 结构体（用于函数表） +# ============================================================ +@t.NoVTable +class FuncEntry: + """函数表条目""" + Name: str + Func: llvmlite.Function | t.CPtr + + +# ============================================================ +# 初始化函数表 +# ============================================================ +def init_func_table(pool: memhub.MemBuddy | t.CPtr, + max_count: int) -> FuncEntry | t.CPtr: + """分配并清零函数表数组""" + size: t.CSizeT = max_count * FuncEntry.__sizeof__() + funcs_ptr: FuncEntry | t.CPtr = pool.alloc(size) + if funcs_ptr is not None: + string.memset(funcs_ptr, 0, size) + return funcs_ptr + + +# ============================================================ +# 查找函数 +# ============================================================ +def find_func_in_table(funcs: FuncEntry | t.CPtr, + func_count: int, + name: str) -> llvmlite.Function | t.CPtr: + """在函数表中按名称查找，返回 Function 指针或 None""" + if name is None or funcs is None: + return None + entry_size: t.CSizeT = FuncEntry.__sizeof__() + for i in range(func_count): + entry_addr: t.CUInt64T = t.CUInt64T(funcs) + i * entry_size + entry: FuncEntry | t.CPtr = (FuncEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry.Name is not None: + if string.strcmp(entry.Name, name) == 0: + return entry.Func + return None + + +# ============================================================ +# 添加函数到函数表 +# ============================================================ +def add_func_to_table(funcs: FuncEntry | t.CPtr, + func_count: int, + name: str, + func: llvmlite.Function | t.CPtr, + max_count: int) -> int: + """添加函数到函数表，返回 0 成功""" + if name is None or func is None or funcs is None: + return 1 + if func_count >= max_count: + return 1 + entry_size: t.CSizeT = FuncEntry.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(funcs) + func_count * entry_size + entry: FuncEntry | t.CPtr = (FuncEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + entry.Name = name + entry.Func = func + # 同时注册到全局函数表（供跨模块 vtable 初始化查找） + _register_global_func(name, func) + return 0 + + +# ============================================================ +# 全局函数注册表（跨模块函数查找） +# +# 所有模块翻译的函数都会注册到此表（通过 add_func_to_table 自动注册）。 +# 主要用途: vtable 初始化时查找跨模块继承方法的 Function 对象， +# 以获取函数类型信息构造 bitcast 字符串。 +# ============================================================ +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: FuncEntry | t.CPtr = None +_global_func_count: int = 0 + + +def _init_global_func_table() -> int: + """初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）""" + global _global_funcs + global _global_func_count + if _global_funcs is not None: + return 1 + size: t.CSizeT = GLOBAL_FUNC_MAX * FuncEntry.__sizeof__() + _global_funcs = stdlib.malloc(size) + if _global_funcs is None: + return 0 + string.memset(_global_funcs, 0, size) + _global_func_count = 0 + return 1 + + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: + """注册函数到全局表（已存在则跳过），返回 0 成功 + + 直接操作全局表，不调用 add_func_to_table，避免递归。 + """ + global _global_func_count + if name is None or func is None: + return 1 + if _init_global_func_table() == 0: + return 1 + # 已存在则跳过（避免重复） + if find_func_in_table(_global_funcs, _global_func_count, name) is not None: + return 0 + if _global_func_count >= GLOBAL_FUNC_MAX: + return 1 + # 直接写入全局表（不调用 add_func_to_table，避免递归） + entry_size: t.CSizeT = FuncEntry.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(_global_funcs) + _global_func_count * entry_size + entry: FuncEntry | t.CPtr = (FuncEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + entry.Name = name + entry.Func = func + _global_func_count += 1 + return 0 + + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: + """在全局函数表中按名称查找，返回 Function 指针或 None""" + if name is None or _global_funcs is None: + return None + return find_func_in_table(_global_funcs, _global_func_count, name) + + +# ============================================================ +# ExprCallHandle - 函数调用处理器（Mixin 继承模式） +# +# HandleCall / HandlePrintfCall 提供 trans 接口 +# ============================================================ +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + """函数调用处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # HandleCall - 翻译函数调用，返回 call 结果 Value + # ============================================================ + def HandleCall(self, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: + """翻译函数调用，从 self.Trans 获取共享状态""" + return translate_call( + self.Trans.Pool, self.Trans._cur_builder, self.Trans.Module, + node, + self.Trans._funcs, self.Trans._func_count, self.Trans) + + # ============================================================ + # HandlePrintfCall - 翻译 printf 调用 + # ============================================================ + def HandlePrintfCall(self, cl: ast.Call | t.CPtr): + """翻译 printf 调用，从 self.Trans 获取共享状态""" + translate_printf_call( + self.Trans.Pool, self.Trans._cur_builder, self.Trans.Module, + cl, self.Trans) + + +# ============================================================ +# NewExprCallHandle - 工厂函数 +# ============================================================ +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: + h: ExprCallHandle | t.CPtr = pool.alloc(ExprCallHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, ExprCallHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesExprOps.py b/TransPyV/App/lib/core/Handles/HandlesExprOps.py new file mode 100644 index 0000000..dac726d --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesExprOps.py @@ -0,0 +1,287 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesTranslator as HT + + +# ============================================================ +# HandlesExprOps - 二元运算处理（模块级纯函数） +# ============================================================ + +# ============================================================ +# 运算符重载支持 +# +# 当二元/比较运算的 lhs 是结构体指针时，检查该类是否定义了 +# 对应的 dunder 方法（如 __add__、__eq__），若有则生成方法调用 +# 而非原生算术/比较指令。 +# ============================================================ + +# BinOp 运算符 → dunder 方法名 +def _binop_to_dunder(op: int) -> str: + """将 BinOp 运算符映射到 dunder 方法名，无映射时返回 None""" + if op == ast.OpKind.Add: return "__add__" + if op == ast.OpKind.Sub: return "__sub__" + if op == ast.OpKind.Mult: return "__mul__" + if op == ast.OpKind.Div: return "__div__" + if op == ast.OpKind.Mod: return "__mod__" + if op == ast.OpKind.BitAnd: return "__and__" + if op == ast.OpKind.BitOr: return "__or__" + if op == ast.OpKind.BitXor: return "__xor__" + if op == ast.OpKind.LShift: return "__lshift__" + if op == ast.OpKind.RShift: return "__rshift__" + if op == ast.OpKind.FloorDiv: return "__floordiv__" + return None + +# Compare 运算符 → dunder 方法名 +def _cmpop_to_dunder(op: int) -> str: + """将 Compare 运算符映射到 dunder 方法名，无映射时返回 None""" + if op == ast.OpKind.Eq: return "__eq__" + if op == ast.OpKind.Ne: return "__ne__" + if op == ast.OpKind.Lt: return "__lt__" + if op == ast.OpKind.Le: return "__le__" + if op == ast.OpKind.Gt: return "__gt__" + if op == ast.OpKind.Ge: return "__ge__" + return None + + +# ============================================================ +# try_operator_overload - 尝试运算符重载 +# +# 检查 lhs 是否为结构体指针，并在该类（含继承链）中查找对应的 +# dunder 方法。找到则生成方法调用 lhs.__dunder__(rhs)，返回 +# 调用结果；未找到返回 None，调用方回退到原生运算。 +# +# Args: +# pool: 内存池 +# builder: IRBuilder +# mod: LLVM 模块 +# lhs: 左操作数 Value（已求值） +# rhs: 右操作数 Value（已求值） +# op: OpKind 运算符 +# trans: Translator 对象 +# is_compare: 0=BinOp, 1=Compare（决定 dunder 映射表） +# +# Returns: +# 方法调用的 Value（成功），None（未重载） +# ============================================================ +def try_operator_overload(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + lhs: llvmlite.Value | t.CPtr, + rhs: llvmlite.Value | t.CPtr, + op: int, + trans: HT.Translator | t.CPtr, + is_compare: int) -> llvmlite.Value | t.CPtr: + """尝试运算符重载：lhs 是结构体指针时调用 dunder 方法，否则返回 None""" + if lhs is None or rhs is None: + return None + + # 延迟导入避免循环依赖 + import lib.core.Handles.HandlesStruct as HandlesStruct + import lib.core.Handles.HandlesExprCall as HandlesExprCall + + # 映射运算符到 dunder 方法名 + dunder: str = None + if is_compare != 0: + dunder = _cmpop_to_dunder(op) + else: + dunder = _binop_to_dunder(op) + if dunder is None: + return None + + # 检查 lhs 是否为结构体指针 + struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type_from_value(lhs) + if struct_ty is None: + return None + + # 获取类名 + class_name: str = HandlesStruct.get_class_name_by_type(pool, struct_ty) + if class_name is None: + return None + + # 在继承链中查找 dunder 方法 + lookup_name: t.CChar | t.CPtr = pool.alloc(128) + if lookup_name is None: + return None + viperlib.snprintf(lookup_name, 128, "%s.%s", class_name, dunder) + found_func: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_in_module(mod, lookup_name) + + search_class: str = class_name + if found_func is None: + cur_parent: str = HandlesStruct.get_parent_name(class_name) + while cur_parent is not None: + parent_lookup: t.CChar | t.CPtr = pool.alloc(128) + if parent_lookup is not None: + viperlib.snprintf(parent_lookup, 128, "%s.%s", cur_parent, dunder) + parent_func: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_in_module(mod, parent_lookup) + if parent_func is not None: + found_func = parent_func + search_class = cur_parent + break + cur_parent = HandlesStruct.get_parent_name(cur_parent) + + # 方法不在当前模块中时，检查类（含父类）是否有 SHA1（stub 可能未注入） + # 优先用类型指针定位 entry 获取 SHA1（规避跨模块同名 find_struct 找错） + cls_sha1: str = None + op_entry: HandlesStruct.StructEntry | t.CPtr = HandlesStruct.find_struct_by_type(struct_ty) + if op_entry is not None: + cls_sha1 = op_entry.ModuleSha1 + if cls_sha1 is None: + cls_sha1 = HandlesStruct.get_struct_sha1(search_class) + if cls_sha1 is None: + cur_p: str = HandlesStruct.get_parent_name(search_class) + while cur_p is not None and cls_sha1 is None: + cls_sha1 = HandlesStruct.get_struct_sha1(cur_p) + cur_p = HandlesStruct.get_parent_name(cur_p) + + # 既无函数定义也无 SHA1 → 该类没有此 dunder 方法，返回 None 回退原生运算 + if found_func is None and cls_sha1 is None: + return None + + # 构建 extra_args 数组（仅含 rhs 一个参数） + extra_args: t.CSizeT | t.CPtr = pool.alloc(8) + if extra_args is None: + return None + extra_args[0] = t.CSizeT(rhs) + + # 调用方法: lhs.__dunder__(rhs) + return HandlesExprCall._call_method_on_ptr( + pool, builder, mod, search_class, dunder, + lhs, extra_args, 1, trans) + + +# ============================================================ +# 翻译二元运算（自动类型提升） +# ============================================================ +def translate_binop(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + node: ast.AST | t.CPtr, + trans: HT.Translator | t.CPtr = None) -> llvmlite.Value | t.CPtr: + """翻译二元运算（自动类型提升）""" + binop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if binop is None: + return None + + lhs_node: ast.AST | t.CPtr = binop.left + rhs_node: ast.AST | t.CPtr = binop.right + op: int = binop.op + + # 先翻译 rhs（只翻译一次，避免副作用重复） + rhs: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, rhs_node, None, 0, trans) + if rhs is None: + return None + + # === 运算符重载路径 1: lhs 是 Name 且对应结构体变量 === + # 对于值类型变量（如 cnt: Counter），translate_value 会 load 返回 Struct 值， + # 但 dunder 方法需要 Ptr(Struct) 作为 self。直接用 alloca 指针尝试重载。 + if lhs_node is not None and lhs_node.kind() == ast.ASTKind.Name and trans is not None: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(lhs_node) + if nm is not None and nm.id is not None: + lhs_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, nm.id) + if lhs_alloca is not None: + ovl_result: llvmlite.Value | t.CPtr = try_operator_overload( + pool, builder, mod, lhs_alloca, rhs, op, trans, 0) + if ovl_result is not None: + return ovl_result + + # 正常翻译 lhs + lhs: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, mod, lhs_node, None, 0, trans) + if lhs is None: + return None + + # === 运算符重载路径 2: lhs 是 Ptr(Struct)（如 Counter|t.CPtr 变量 load 后）=== + ovl_result2: llvmlite.Value | t.CPtr = try_operator_overload( + pool, builder, mod, lhs, rhs, op, trans, 0) + if ovl_result2 is not None: + return ovl_result2 + + # 获取操作数类型信息 + lhs_bits: int = HandlesExpr.get_llvm_type_bits(lhs.Ty) + rhs_bits: int = HandlesExpr.get_llvm_type_bits(rhs.Ty) + lhs_fbits: int = HandlesExpr.get_llvm_float_bits(lhs.Ty) + rhs_fbits: int = HandlesExpr.get_llvm_float_bits(rhs.Ty) + + # 浮点运算: 任一操作数为浮点时，使用浮点指令（必须在指针算术之前检查， + # 因为 float 的 int bits 为 0，会被误认为指针） + if lhs_fbits != 0 or rhs_fbits != 0: + # 确定目标浮点类型（使用较大的位宽） + target_fbits: int = lhs_fbits + if rhs_fbits > target_fbits: + target_fbits = rhs_fbits + target_float_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Double(pool) + if target_fbits == 32: + target_float_ty = llvmlite.Float(pool) + # 将两个操作数转换为目标浮点类型（int→float via si2fp, float→float via fpext/fptrunc） + lhs = HandlesExpr.coerce_to_type(builder, lhs, target_float_ty) + rhs = HandlesExpr.coerce_to_type(builder, rhs, target_float_ty) + if op == ast.OpKind.Add: + return llvmlite.build_fadd(builder, lhs, rhs) + if op == ast.OpKind.Sub: + return llvmlite.build_fsub(builder, lhs, rhs) + if op == ast.OpKind.Mult: + return llvmlite.build_fmul(builder, lhs, rhs) + if op == ast.OpKind.Div: + return llvmlite.build_fdiv(builder, lhs, rhs) + if op == ast.OpKind.Mod: + return llvmlite.build_frem(builder, lhs, rhs) + return None + + # 指针算术: ptr + int 或 ptr - int → ptrtoint + add/sub + inttoptr + if (lhs_bits == 0 and rhs_bits != 0) or (lhs_bits != 0 and rhs_bits == 0): + if op == ast.OpKind.Add or op == ast.OpKind.Sub: + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + if lhs_bits == 0: + # lhs 是指针，rhs 是整数 + ptr_as_int: llvmlite.Value | t.CPtr = llvmlite.build_ptrtoint(builder, lhs, i64_ty) + int_val: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type(builder, rhs, i64_ty) + if op == ast.OpKind.Add: + result: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, ptr_as_int, int_val) + else: + result = llvmlite.build_sub(builder, ptr_as_int, int_val) + return llvmlite.build_inttoptr(builder, result, lhs.Ty) + else: + # rhs 是指针，lhs 是整数（仅 Add 支持交换） + ptr_as_int = llvmlite.build_ptrtoint(builder, rhs, i64_ty) + int_val = HandlesExpr.coerce_to_type(builder, lhs, i64_ty) + if op == ast.OpKind.Add: + result = llvmlite.build_add(builder, int_val, ptr_as_int) + else: + result = llvmlite.build_sub(builder, int_val, ptr_as_int) + return llvmlite.build_inttoptr(builder, result, rhs.Ty) + + if lhs_bits > rhs_bits: + rhs = HandlesExpr.coerce_to_type(builder, rhs, lhs.Ty) + elif rhs_bits > lhs_bits: + lhs = HandlesExpr.coerce_to_type(builder, lhs, rhs.Ty) + + if op == ast.OpKind.Add: + return llvmlite.build_add(builder, lhs, rhs) + elif op == ast.OpKind.Sub: + return llvmlite.build_sub(builder, lhs, rhs) + elif op == ast.OpKind.Mult: + return llvmlite.build_mul(builder, lhs, rhs) + elif op == ast.OpKind.Div: + return llvmlite.build_sdiv(builder, lhs, rhs) + elif op == ast.OpKind.Mod: + return llvmlite.build_srem(builder, lhs, rhs) + elif op == ast.OpKind.BitAnd: + return llvmlite.build_and(builder, lhs, rhs) + elif op == ast.OpKind.BitOr: + return llvmlite.build_or(builder, lhs, rhs) + elif op == ast.OpKind.BitXor: + return llvmlite.build_xor(builder, lhs, rhs) + elif op == ast.OpKind.LShift: + return llvmlite.build_shl(builder, lhs, rhs) + elif op == ast.OpKind.RShift: + return llvmlite.build_ashr(builder, lhs, rhs) + return None diff --git a/TransPyV/App/lib/core/Handles/HandlesFor.py b/TransPyV/App/lib/core/Handles/HandlesFor.py new file mode 100644 index 0000000..0ed8f1e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesFor.py @@ -0,0 +1,384 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesType as HandlesType + + +# ============================================================ +# HandlesFor - for 循环语句处理（Mixin 继承模式） +# +# 支持 for i in range(start, stop, step) 模式: +# %i = alloca i32 +# store i32 start, i32* %i +# br label %cond +# cond: +# %cur = load i32, i32* %i +# %cmp = icmp slt i32 %cur, stop +# br i1 %cmp, label %body, label %end +# body: +# ... body ... +# br label %incr +# incr: +# %cur2 = load i32, i32* %i +# %next = add i32 %cur2, step +# store i32 %next, i32* %i +# br label %cond +# end: +# ============================================================ + + +@t.NoVTable +class ForHandle(HandlesBase.Mixin): + """for 循环语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # Handle - 处理 for 语句，返回新增变量数 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译 for i in range(...) 循环语句""" + if node is None: + return 0 + + trans: HT.Translator | t.CPtr = self.Trans + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + + if builder is None or func is None: + return 0 + + for_node: ast.For | t.CPtr = (ast.For | t.CPtr)(node) + + # 1. 获取循环变量名（仅支持 for i in range(...)） + target: ast.AST | t.CPtr = for_node.target + if target is None: + return 0 + if target.kind() != ast.ASTKind.Name: + return 0 + target_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(target) + var_name: str = target_nm.id + if var_name is None: + return 0 + + # 2. 解析迭代器：支持 range() 和指针迭代 + iter_node: ast.AST | t.CPtr = for_node.iter + if iter_node is None: + return 0 + + # 检查是否是 range() 调用 + is_range_iter: int = 0 + if iter_node.kind() == ast.ASTKind.Call: + call_pre: ast.Call | t.CPtr = (ast.Call | t.CPtr)(iter_node) + fn_pre: str = HandlesExpr.get_func_name(call_pre.func) + if fn_pre is not None: + if string.strcmp(fn_pre, "range") == 0: + is_range_iter = 1 + else: + err_msg: t.CChar | t.CPtr = pool.alloc(256) + if err_msg is not None: + viperlib.snprintf(err_msg, 256, "仅支持 range() 或指针迭代，got call '%s'", fn_pre) + HandlesType.fatal_error(iter_node, err_msg) + HandlesType.fatal_error(iter_node, "仅支持 range() 或指针迭代") + + # 非范围迭代：走指针迭代路径 + if is_range_iter == 0: + return self._handle_ptr_iter(for_node, var_name) + + call: ast.Call | t.CPtr = (ast.Call | t.CPtr)(iter_node) + + # 3. 解析 range 参数: range(stop) / range(start, stop) / range(start, stop, step) + args: list[ast.AST | t.CPtr] | t.CPtr = call.args + if args is None: + return 0 + arg_count: t.CSizeT = args.__len__() + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + start_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + stop_val: llvmlite.Value | t.CPtr = None + step_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 1) + + if arg_count == 1: + stop_val = HandlesExpr.translate_value( + builder, pool, trans.Module, args.get(0), + trans._funcs, trans._func_count, trans) + elif arg_count >= 2: + start_val = HandlesExpr.translate_value( + builder, pool, trans.Module, args.get(0), + trans._funcs, trans._func_count, trans) + stop_val = HandlesExpr.translate_value( + builder, pool, trans.Module, args.get(1), + trans._funcs, trans._func_count, trans) + if arg_count >= 3: + step_val = HandlesExpr.translate_value( + builder, pool, trans.Module, args.get(2), + trans._funcs, trans._func_count, trans) + + if stop_val is None: + stop_val = llvmlite.const_int32(pool, 0) + if start_val is None: + start_val = llvmlite.const_int32(pool, 0) + if step_val is None: + step_val = llvmlite.const_int32(pool, 1) + + # 4. 创建/查找循环变量 alloca + var_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, var_name) + new_vars: int = 0 + if var_alloca is None: + var_alloca = llvmlite.build_alloca(builder, i32_ty) + if HandlesVar.define_var( + trans.SymTab, var_name, var_alloca) == 0: + new_vars = 1 + + # 5. 存储初始值 (类型对齐: start_val 可能是 i64，需截断到 i32) + init_val: llvmlite.Value | t.CPtr = start_val + if start_val is not None and start_val.Ty is not None: + start_bits: int = HandlesExpr.get_llvm_type_bits(start_val.Ty) + if start_bits != 0 and start_bits != 32: + init_val = llvmlite.build_trunc(builder, start_val, i32_ty) + llvmlite.build_store(builder, init_val, var_alloca) + + # 6. 创建基本块: cond / body / incr / end（使用 trans._label_counter，不与 SSA 名共享） + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "for.cond.%d", cnt) + cond_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "for.body.%d", cnt) + body_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "for.incr.%d", cnt) + incr_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "for.end.%d", cnt) + end_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # 7. 跳转到 cond 块 + llvmlite.build_br(builder, cond_bb) + + # 8. cond 块: load i, icmp slt i, stop, cond_br body/end + llvmlite.position_at_end(builder, cond_bb) + cur_i: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i32_ty, var_alloca) + # 类型对齐: stop_val 可能是 i64 (如 range(strlen(s)))，需将 cur_i 提升到 stop_val 类型 + cmp_lhs: llvmlite.Value | t.CPtr = cur_i + cmp_rhs: llvmlite.Value | t.CPtr = stop_val + if stop_val is not None and stop_val.Ty is not None: + stop_bits: int = HandlesExpr.get_llvm_type_bits(stop_val.Ty) + cur_bits: int = HandlesExpr.get_llvm_type_bits(cur_i.Ty) + if stop_bits != 0 and cur_bits != 0 and stop_bits != cur_bits: + if cur_bits < stop_bits: + cmp_lhs = llvmlite.build_sext(builder, cur_i, stop_val.Ty) + else: + cmp_rhs = llvmlite.build_trunc(builder, stop_val, i32_ty) + cond_i1: llvmlite.Value | t.CPtr = llvmlite.build_icmp( + builder, llvmlite.ICMP_SLT, cmp_lhs, cmp_rhs) + llvmlite.build_cond_br(builder, cond_i1, body_bb, end_bb) + + # 9. body 块: 翻译循环体，跳到 incr + llvmlite.position_at_end(builder, body_bb) + + # 保存旧循环上下文，设置 break/continue 目标 + old_break: llvmlite.BasicBlock | t.CPtr = trans._break_bb + old_continue: llvmlite.BasicBlock | t.CPtr = trans._continue_bb + trans._break_bb = end_bb + trans._continue_bb = incr_bb + + body: list[ast.AST | t.CPtr] | t.CPtr = for_node.children + if body is not None: + body_count: t.CSizeT = body.__len__() + for bi in range(body_count): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.translate_stmt(trans, stmt) + + # 恢复旧循环上下文 + trans._break_bb = old_break + trans._continue_bb = old_continue + + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, incr_bb) + + # 10. incr 块: i = i + step, 跳回 cond + llvmlite.position_at_end(builder, incr_bb) + cur_i2: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i32_ty, var_alloca) + # 类型对齐: step_val 可能是 i64，需截断到 i32 与 cur_i2 类型一致 + incr_step: llvmlite.Value | t.CPtr = step_val + if step_val is not None and step_val.Ty is not None: + step_bits: int = HandlesExpr.get_llvm_type_bits(step_val.Ty) + cur2_bits: int = HandlesExpr.get_llvm_type_bits(cur_i2.Ty) + if step_bits != 0 and cur2_bits != 0 and step_bits != cur2_bits: + if step_bits > cur2_bits: + incr_step = llvmlite.build_trunc(builder, step_val, i32_ty) + else: + incr_step = llvmlite.build_sext(builder, step_val, i32_ty) + next_i: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, cur_i2, incr_step) + llvmlite.build_store(builder, next_i, var_alloca) + llvmlite.build_br(builder, cond_bb) + + # 11. 定位到 end 块 + llvmlite.position_at_end(builder, end_bb) + return new_vars + + # ============================================================ + # _handle_ptr_iter - 指针迭代: for x in ptr: + # 遍历指针，依赖隐式 index，直到解引用为空（null 终止符） + # + # 生成 IR 结构: + # %idx = alloca i32 + # store i32 0, i32* %idx + # br label %cond + # cond: + # %i = load i32, i32* %idx + # %ep = getelementptr elem_ty, ptr_ty %ptr, i32 %i + # %ev = load elem_ty, elem_ty* %ep + # %null = icmp eq elem_ty %ev, 0 + # br i1 %null, label %end, label %body + # body: + # store elem_ty %ev, elem_ty* %var + # ... 循环体 ... + # br label %incr + # incr: + # %next = add i32 %i, 1 + # store i32 %next, i32* %idx + # br label %cond + # end: + # ============================================================ + def _handle_ptr_iter(self, for_node: ast.For | t.CPtr, + var_name: str) -> int: + """指针迭代: for x in ptr: 直到解引用为空""" + trans: HT.Translator | t.CPtr = self.Trans + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + + # 翻译迭代器表达式，获取指针值 + ptr_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, trans.Module, for_node.iter, + trans._funcs, trans._func_count, trans) + if ptr_val is None: + HandlesType.fatal_error(for_node, "指针迭代: 无法翻译迭代器表达式") + + # 检查是否是指针类型 + if HandlesExpr.is_ptr_type(ptr_val.Ty) == 0: + HandlesType.fatal_error(for_node, "指针迭代: 迭代器不是指针类型") + + # 获取元素类型 + elem_ty: llvmlite.LLVMType | t.CPtr = ptr_val.Ty.Pointee + if elem_ty is None: + HandlesType.fatal_error(for_node, "指针迭代: 无法获取元素类型") + + # 元素类型必须是整数（用于 icmp eq 0 检查 null 终止符） + elem_bits: int = HandlesExpr.get_llvm_type_bits(elem_ty) + if elem_bits == 0: + HandlesType.fatal_error(for_node, "指针迭代: 元素类型不是整数，无法检查 null 终止符") + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + + # 创建循环变量 alloca（存储元素值） + var_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, var_name) + new_vars: int = 0 + if var_alloca is None: + var_alloca = llvmlite.build_alloca(builder, elem_ty) + if HandlesVar.define_var( + trans.SymTab, var_name, var_alloca) == 0: + new_vars = 1 + + # 创建隐式 index 变量，初始为 0 + idx_alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, i32_ty) + llvmlite.build_store(builder, llvmlite.const_int32(pool, 0), idx_alloca) + + # 创建基本块: cond / body / incr / end + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "ptr.cond.%d", cnt) + cond_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "ptr.body.%d", cnt) + body_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "ptr.incr.%d", cnt) + incr_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "ptr.end.%d", cnt) + end_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # 跳转到 cond + llvmlite.build_br(builder, cond_bb) + + # cond 块: load index, GEP, load elem, icmp eq 0 + llvmlite.position_at_end(builder, cond_bb) + cur_idx: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i32_ty, idx_alloca) + elem_ptr: llvmlite.Value | t.CPtr = llvmlite.build_gep( + builder, elem_ty, ptr_val, cur_idx) + cur_elem: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, elem_ty, elem_ptr) + null_val: llvmlite.Value | t.CPtr = llvmlite.const_int(pool, elem_bits, 0) + is_null: llvmlite.Value | t.CPtr = llvmlite.build_icmp( + builder, llvmlite.ICMP_EQ, cur_elem, null_val) + llvmlite.build_cond_br(builder, is_null, end_bb, body_bb) + + # body 块: store elem to var, 翻译循环体 + llvmlite.position_at_end(builder, body_bb) + llvmlite.build_store(builder, cur_elem, var_alloca) + + # 保存旧循环上下文，设置 break/continue 目标 + old_break: llvmlite.BasicBlock | t.CPtr = trans._break_bb + old_continue: llvmlite.BasicBlock | t.CPtr = trans._continue_bb + trans._break_bb = end_bb + trans._continue_bb = incr_bb + + body: list[ast.AST | t.CPtr] | t.CPtr = for_node.children + if body is not None: + body_count: t.CSizeT = body.__len__() + for bi in range(body_count): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.translate_stmt(trans, stmt) + + # 恢复旧循环上下文 + trans._break_bb = old_break + trans._continue_bb = old_continue + + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, incr_bb) + + # incr 块: index++, br cond + llvmlite.position_at_end(builder, incr_bb) + one_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 1) + next_idx: llvmlite.Value | t.CPtr = llvmlite.build_add(builder, cur_idx, one_val) + llvmlite.build_store(builder, next_idx, idx_alloca) + llvmlite.build_br(builder, cond_bb) + + # 定位到 end 块 + llvmlite.position_at_end(builder, end_bb) + return new_vars + + +# ============================================================ +# NewForHandle - 工厂函数 +# ============================================================ +def NewForHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> ForHandle | t.CPtr: + h: ForHandle | t.CPtr = pool.alloc(ForHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, ForHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesFunctions.py b/TransPyV/App/lib/core/Handles/HandlesFunctions.py new file mode 100644 index 0000000..e785cfb --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesFunctions.py @@ -0,0 +1,936 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import stdio +import stdlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesImports as HandlesImports + + +# ============================================================ +# extract_func_attrs - 从 decorator_list 提取 c.Attribute 属性 +# +# 支持 @c.Attribute(t.attr.xxx()) 和 @c.Attribute(t.attr.xxx) 形式 +# 也支持 @c.Attribute(t.attr.llvm.xxx) 形式 +# +# 返回 LLVM IR 属性字符串（如 "alwaysinline nounwind"），无属性返回 None +# ============================================================ +def extract_func_attrs(pool: memhub.MemBuddy | t.CPtr, + decorator_list: list[ast.AST | t.CPtr] | t.CPtr, + imported_modules: str) -> t.CChar | t.CPtr: + """从 decorator_list 提取 c.Attribute 属性，返回 LLVM IR 属性字符串""" + if decorator_list is None: + return None + dn: t.CSizeT = decorator_list.__len__() + if dn == 0: + return None + + attrs_buf: t.CChar | t.CPtr = pool.alloc(256) + if attrs_buf is None: + return None + attrs_buf[0] = '\0' + found_any: int = 0 + + for di in range(dn): + deco: ast.AST | t.CPtr = decorator_list.get(di) + if deco is None or deco.kind() != ast.ASTKind.Call: + continue + call_node: ast.Call | t.CPtr = (ast.Call | t.CPtr)(deco) + if call_node.func is None or call_node.func.kind() != ast.ASTKind.Attribute: + continue + + # 检测 c.Attribute + func_attr: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(call_node.func) + if func_attr.attr is None or string.strcmp(func_attr.attr, "Attribute") != 0: + continue + if func_attr.value is None or func_attr.value.kind() != ast.ASTKind.Name: + continue + c_name: ast.Name | t.CPtr = (ast.Name | t.CPtr)(func_attr.value) + if c_name.id is None or string.strcmp(c_name.id, "c") != 0: + continue + if HandlesImports.is_module_imported(imported_modules, "c") == 0: + continue + + # 遍历参数提取属性 + if call_node.args is None: + continue + args_list: list[ast.AST | t.CPtr] | t.CPtr = call_node.args + an: t.CSizeT = args_list.__len__() + for ai in range(an): + arg_node: ast.AST | t.CPtr = args_list.get(ai) + if arg_node is None: + continue + + # 提取属性名（t.attr.xxx / t.attr.xxx() / t.attr.llvm.xxx） + attr_name: str = _get_attr_name_from_node(arg_node) + if attr_name is None: + continue + + # 映射到 LLVM 属性名并追加 + if _append_llvm_attr(attrs_buf, attr_name) != 0: + found_any = 1 + # 追加成功后，如果不是最后一个属性，添加空格分隔 + # 在下一次追加前由 _append_str 处理 + + if found_any == 0: + return None + return attrs_buf + + +def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str: + """从 AST 节点提取属性名 + + 支持: + - Call(func=Attribute(...)): t.attr.always_inline() -> "always_inline" + - Attribute: t.attr.packed -> "packed" + - Attribute(t.attr.llvm.xxx): t.attr.llvm.nounwind -> "nounwind" + """ + if node is None: + return None + k: int = node.kind() + + # Call 节点: t.attr.always_inline() + if k == ast.ASTKind.Call: + call: ast.Call | t.CPtr = (ast.Call | t.CPtr)(node) + if call.func is None or call.func.kind() != ast.ASTKind.Attribute: + return None + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(call.func) + return at.attr + + # Attribute 节点: t.attr.packed 或 t.attr.llvm.nounwind + if k == ast.ASTKind.Attribute: + at2: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + return at2.attr + + return None + + +def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int: + """将属性名映射到 LLVM 属性名并追加到缓冲区 + + 返回 1 表示已追加，0 表示不支持该属性 + """ + if attrs_buf is None or attr_name is None: + return 0 + + # 先确定映射的 LLVM 属性名 + llvm_name: str = None + if string.strcmp(attr_name, "always_inline") == 0: + llvm_name = "alwaysinline" + elif string.strcmp(attr_name, "noinline") == 0: + llvm_name = "noinline" + elif string.strcmp(attr_name, "noreturn") == 0: + llvm_name = "noreturn" + elif string.strcmp(attr_name, "pure") == 0: + llvm_name = "readonly" + elif string.strcmp(attr_name, "const") == 0: + llvm_name = "readnone" + elif string.strcmp(attr_name, "nounwind") == 0: + llvm_name = "nounwind" + elif string.strcmp(attr_name, "noredzone") == 0: + llvm_name = "noredzone" + elif string.strcmp(attr_name, "willreturn") == 0: + llvm_name = "willreturn" + elif string.strcmp(attr_name, "mustprogress") == 0: + llvm_name = "mustprogress" + else: + # packed/aligned/section/visibility/weak 等不支持作为函数内联属性 + return 0 + + # 支持的属性：如果缓冲区非空，先追加空格分隔 + if attrs_buf[0] != '\0': + _append_str(attrs_buf, " ") + _append_str(attrs_buf, llvm_name) + return 1 + + +def _append_str(buf: t.CChar | t.CPtr, s: str) -> None: + """将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）""" + if buf is None or s is None: + return + cur_len: t.CSizeT = string.strlen(buf) + string.strcpy(buf + cur_len, s) + + +# ============================================================ +# HandlesFunctions - 函数定义处理（trans 单参模式） +# +# 参考 Python 版 TransPyC FunctionHandle +# +# 函数有自己的局部作用域，通过 HandlesVar.enter_scope/exit_scope +# 管理嵌套作用域链，翻译函数体前进入函数作用域，翻译完退出。 +# ============================================================ + + +# ============================================================ +# _mangle_name - 为名称添加 SHA1 前缀 +# +# 返回 "sha1.name" 格式的混淆名。若 ModuleSha1 为 None 或名称已带 +# SHA1 前缀（16 hex + '.'），则原样返回。 +# ============================================================ +def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str: + """为任意名称添加 SHA1 前缀（不做 main/export 检查）""" + if trans is None or name is None: + return name + if trans.ModuleSha1 is None: + return name + name_len: t.CSizeT = string.strlen(name) + # 已带 SHA1 前缀（16 hex + '.'）则跳过 + if name_len > 17: + is_sha1: int = 1 + for i in range(16): + c: t.CChar = name[i] + if not (('0' <= c <= '9') or ('a' <= c <= 'f')): + is_sha1 = 0 + break + if is_sha1 != 0 and name[16] == '.': + return name + sha1: str = trans.ModuleSha1 + sha1_len: t.CSizeT = string.strlen(sha1) + mangled: str = stdlib.malloc(sha1_len + name_len + 2) + if mangled is None: + return name + string.strcpy(mangled, sha1) + mangled[sha1_len] = '.' + string.strcpy(mangled + sha1_len + 1, name) + return mangled + + +# ============================================================ +# _mangle_name_with_sha1 — 用指定 SHA1 为名称添加前缀 +# +# 用于跨模块 vtable 继承: 继承方法需要用父模块的 SHA1 做 mangling， +# 而非当前模块的 SHA1。 +# ============================================================ +def _mangle_name_with_sha1(sha1: str, name: str) -> str: + """用指定 SHA1 为名称添加前缀（不做 main/export 检查）""" + if sha1 is None or name is None: + return name + name_len: t.CSizeT = string.strlen(name) + # 已带 SHA1 前缀（16 hex + '.'）则跳过 + if name_len > 17: + is_sha1: int = 1 + for i in range(16): + c2: t.CChar = name[i] + if not (('0' <= c2 <= '9') or ('a' <= c2 <= 'f')): + is_sha1 = 0 + break + if is_sha1 != 0 and name[16] == '.': + return name + sha1_len: t.CSizeT = string.strlen(sha1) + mangled: str = stdlib.malloc(sha1_len + name_len + 2) + if mangled is None: + return name + string.strcpy(mangled, sha1) + mangled[sha1_len] = '.' + string.strcpy(mangled + sha1_len + 1, name) + return mangled + + +# ============================================================ +# _mangle_func_name - 为函数名添加 SHA1 前缀（含 main/export 检查） +# +# 规则： +# - main 函数不加前缀（程序入口点） +# - has_export 非 0 时不加前缀（t.CExport 导出函数） +# - 已带 SHA1 前缀的不再加 +# - ModuleSha1 为 None 时不加前缀 +# ============================================================ +def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str, + has_export: int) -> str: + """为函数名添加 SHA1 前缀（检查 main/export）""" + if trans is None or name is None: + return name + if has_export != 0: + return name + if string.strcmp(name, "main") == 0: + return name + return _mangle_name(trans, name) + + +# ============================================================ +# forward_declare_functions - 预扫描所有顶层 FunctionDef，创建前向声明 +# +# 解决同模块内前向引用问题：如 viperlib.py 中 sprintf(行15) 调用 +# vsnprintf(行41)，但 vsnprintf 定义在后面。 +# +# 对每个顶层 FunctionDef: +# 1. 推断返回类型和参数类型 +# 2. 计算 mangled name（含 SHA1 前缀） +# 3. 创建 declare（IsDeclared=1） +# 4. 添加参数 +# 5. 注册到函数表 +# +# 后续 translate_function_def 会通过 find_func_in_module 找到已有的 declare， +# 复用之（清除 IsDeclared，跳过参数创建，直接添加函数体）。 +# ============================================================ +def forward_declare_functions(trans: HT.Translator | t.CPtr, + tree: ast.AST | t.CPtr) -> int: + """预扫描所有顶层 FunctionDef，创建前向声明""" + if trans is None or tree is None: + return 0 + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + if pool is None or mod is None: + return 0 + + imported_modules: str = trans._imported_modules + from_imports: str = trans._from_imports + funcs_ptr: HandlesExprCall.FuncEntry | t.CPtr = trans._funcs + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + ch: list[ast.AST | t.CPtr] | t.CPtr = tree.children + if ch is None: + return 0 + cn_count: t.CSizeT = ch.__len__() + + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is None: + continue + if child.kind() != ast.ASTKind.FunctionDef: + continue + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(child) + if fd is None or fd.name is None: + continue + + # 推断返回类型 + ret_ty: llvmlite.LLVMType | t.CPtr = None + if fd.returns is not None: + ret_ty = HandlesType.resolve_annotation_type( + pool, fd.returns, imported_modules, from_imports) + if ret_ty is None and fd.returns is not None: + if HandlesType.has_decorator_marker(fd.returns, "State") != 0: + ret_ty = llvmlite.Void(pool) + if ret_ty is None: + ret_ty = i32_ty + + # 检测 CExtern/State/CExport + has_extern: int = 0 + has_state: int = 0 + has_export: int = 0 + if fd.returns is not None: + has_extern = HandlesType.has_decorator_marker(fd.returns, "CExtern") + has_state = HandlesType.has_decorator_marker(fd.returns, "State") + has_export = HandlesType.has_decorator_marker(fd.returns, "CExport") + is_extern_decl: int = 0 + if has_extern != 0 or has_state != 0: + is_extern_decl = 1 + + # 计算 mangled name + if has_extern != 0 or has_state != 0 or has_export != 0 or fd.name == "main": + mangled_name: str = fd.name + else: + mangled_name = _mangle_func_name(trans, fd.name, 0) + + # 如果函数已存在（如重复定义），跳过 + existing: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_in_module(mod, mangled_name) + if existing is not None: + continue + + # 创建 declare + func: llvmlite.Function | t.CPtr = llvmlite.create_declare( + pool, mod, mangled_name, ret_ty) + if func is None: + continue + + # 注册到函数表（用裸名 fd.name，不是 mangled_name） + max_funcs: int = 256 + cur_count: int = trans._func_count + if HandlesExprCall.add_func_to_table(funcs_ptr, cur_count, fd.name, func, max_funcs) == 0: + trans._func_count = cur_count + 1 + + # 注册 CExport 函数到全局表 + if (has_export != 0 or has_state != 0) and trans.ModuleSha1 is not None: + HandlesExprCall.register_cexport_func(trans.ModuleSha1, fd.name) + + # 添加参数 + args_node: ast.Arguments | t.CPtr = fd.args + if args_node is not None: + ags: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags.args is not None: + alist: list[ast.AST | t.CPtr] | t.CPtr = ags.args + an: t.CSizeT = alist.__len__() + for ai in range(an): + arg: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist.get(ai)) + if arg is None or arg.arg is None: + continue + # t.CVoid 表示空参：跳过 + if arg.annotation is not None: + if HandlesType._is_t_attr(arg.annotation, "CVoid", imported_modules) != 0: + continue + param_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if arg.annotation is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg.annotation, imported_modules, from_imports) + if resolved is not None: + param_ty = resolved + pname: t.CChar | t.CPtr = pool.alloc(32) + if pname is not None: + viperlib.snprintf(pname, 32, "%%%s", arg.arg) + llvmlite.add_param(pool, func, param_ty, pname) + + return 0 + + +# ============================================================ +# 翻译函数定义 FunctionDef(name, args, body, ...) +# +# trans 单参模式：所有共享状态从 trans 获取 +# 函数局部作用域通过 enter_scope/exit_scope 管理 +# ============================================================ +def translate_function_def(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）""" + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(node) + if fd is None or fd.name is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + imported_modules: str = trans._imported_modules + from_imports: str = trans._from_imports + funcs_ptr: HandlesExprCall.FuncEntry | t.CPtr = trans._funcs + func_count: int = trans._func_count + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + + # 推断返回类型：优先使用返回类型注解 + ret_ty: llvmlite.LLVMType | t.CPtr = None + if fd.returns is not None: + ret_ty = HandlesType.resolve_annotation_type( + pool, fd.returns, imported_modules, from_imports) + + # 如果返回类型注解纯装饰器标记（如 t.State，无实际类型），使用 void + # 注意：必须在 infer_return_type 之前检测，因为 infer_return_type 至少返回 i32 + if ret_ty is None and fd.returns is not None: + if HandlesType.has_decorator_marker(fd.returns, "State") != 0: + ret_ty = llvmlite.Void(pool) + + # 如果仍然为 None，扫描 return 语句推断（至少返回 i32） + if ret_ty is None: + param_types_str: str = HandlesType.build_param_types_str(pool, fd.args) + ret_ty = HandlesType.infer_return_type( + pool, fd.children, param_types_str) + + # 检测是否为外部声明函数（t.CExtern 或 t.State） + # 语义：t.CExtern 忽略 body 体，仅生成 declare（由链接器解析符号） + # t.State = t.CExtern + t.CExport（既是声明又是导出） + # t.CExport 导出函数不加 SHA1 前缀 + # t.CExtern/t.State/t.CExport 都不加 SHA1 前缀，直接映射到 C 标准库函数名 + is_extern_decl: int = 0 + has_export: int = 0 + has_extern: int = 0 + has_state: int = 0 + if fd.returns is not None: + has_extern = HandlesType.has_decorator_marker(fd.returns, "CExtern") + has_state = HandlesType.has_decorator_marker(fd.returns, "State") + has_export = HandlesType.has_decorator_marker(fd.returns, "CExport") + # t.CExtern/t.State 忽略 body 体，仅声明（无论 body 是否为 pass） + if has_extern != 0 or has_state != 0: + is_extern_decl = 1 + + # SHA1 命名空间：t.CExtern/t.State/t.CExport 不加前缀，直接用裸名 + # （裸名映射到 C 标准库符号，带 sha1 前缀会导致 undefined reference） + # main 函数也不加前缀（程序入口点） + if has_extern != 0 or has_state != 0 or has_export != 0 or fd.name == "main": + mangled_name: str = fd.name + else: + mangled_name: str = _mangle_func_name(trans, fd.name, 0) + + # 注册 CExport 函数到全局表（供跨模块调用查表） + # t.CExport 函数定义用裸名（@strlen），跨模块调用需查表确认用裸名而非 @{sha1}.func + if (has_export != 0 or has_state != 0) and trans.ModuleSha1 is not None: + HandlesExprCall.register_cexport_func(trans.ModuleSha1, fd.name) + + if is_extern_decl != 0: + # 外部声明函数：生成 declare（仅声明，不定义） + func: llvmlite.Function | t.CPtr = llvmlite.create_declare( + pool, mod, mangled_name, ret_ty) + if func is None: + stdio.printf("[FUNC] create_declare %s failed\n", fd.name) + return 0 + + # 注册到函数表 + max_funcs_extern: int = 256 + if HandlesExprCall.add_func_to_table(funcs_ptr, func_count, fd.name, func, max_funcs_extern) == 0: + trans._func_count = func_count + 1 + + # 添加参数（支持类型注解） + args_node_extern: ast.Arguments | t.CPtr = fd.args + if args_node_extern is not None: + ags_e: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node_extern) + if ags_e.args is not None: + alist_e: list[ast.AST | t.CPtr] | t.CPtr = ags_e.args + an_e: t.CSizeT = alist_e.__len__() + for ai_e in range(an_e): + arg_e: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist_e.get(ai_e)) + if arg_e is not None and arg_e.arg is not None: + # t.CVoid 表示空参：跳过，不添加到函数签名 + if arg_e.annotation is not None: + if HandlesType._is_t_attr(arg_e.annotation, "CVoid", imported_modules) != 0: + continue + param_ty_e: llvmlite.LLVMType | t.CPtr = i32_ty + if arg_e.annotation is not None: + resolved_e: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg_e.annotation, imported_modules, from_imports) + if resolved_e is not None: + param_ty_e = resolved_e + pname_e: t.CChar | t.CPtr = pool.alloc(32) + if pname_e is not None: + viperlib.snprintf(pname_e, 32, "%%%s", arg_e.arg) + llvmlite.add_param(pool, func, param_ty_e, pname_e) + + # 提取 @c.Attribute 装饰器属性并设置到函数 + func_attrs_e: t.CChar | t.CPtr = extract_func_attrs(pool, fd.decorator_list, imported_modules) + if func_attrs_e is not None: + llvmlite.function_set_attrs(func, func_attrs_e) + + return 0 + + # args_node 用于后续 alloca 创建，提前赋值 + args_node: ast.Arguments | t.CPtr = fd.args + + # 检查是否已有前向声明（由 forward_declare_functions 创建） + func: llvmlite.Function | t.CPtr = HandlesExprCall.find_func_in_module(mod, mangled_name) + if func is not None and llvmlite.function_is_declared(func) != 0: + # 复用前向声明：清除 IsDeclared 标记，转为 define + func.IsDeclared = 0 + # 装饰器属性设置（前向声明时未设置） + func_attrs_reuse: t.CChar | t.CPtr = extract_func_attrs(pool, fd.decorator_list, imported_modules) + if func_attrs_reuse is not None: + llvmlite.function_set_attrs(func, func_attrs_reuse) + else: + # 创建新的 LLVM 函数（使用 SHA1 混淆名） + func = llvmlite.create_function(pool, mod, mangled_name, ret_ty) + if func is None: + stdio.printf("[FUNC] create_function %s failed\n", fd.name) + return 0 + + # 提取 @c.Attribute 装饰器属性并设置到函数 + func_attrs: t.CChar | t.CPtr = extract_func_attrs(pool, fd.decorator_list, imported_modules) + if func_attrs is not None: + llvmlite.function_set_attrs(func, func_attrs) + + # 注册到函数表 + max_funcs: int = 256 + if HandlesExprCall.add_func_to_table(funcs_ptr, func_count, fd.name, func, max_funcs) == 0: + trans._func_count = func_count + 1 + + # 添加参数（支持类型注解） + if args_node is not None: + ags: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags.args is not None: + alist: list[ast.AST | t.CPtr] | t.CPtr = ags.args + an: t.CSizeT = alist.__len__() + for ai in range(an): + arg: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist.get(ai)) + if arg is not None and arg.arg is not None: + # t.CVoid 表示空参：跳过，不添加到函数签名 + if arg.annotation is not None: + if HandlesType._is_t_attr(arg.annotation, "CVoid", imported_modules) != 0: + continue + param_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if arg.annotation is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg.annotation, imported_modules, from_imports) + if resolved is not None: + param_ty = resolved + pname: t.CChar | t.CPtr = pool.alloc(32) + if pname is not None: + viperlib.snprintf(pname, 32, "%%%s", arg.arg) + llvmlite.add_param(pool, func, param_ty, pname) + + # 创建 entry 块 + entry_blk: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, "entry") + if entry_blk is None: + return 0 + + # 创建函数专属 builder + func_builder: llvmlite.IRBuilder | t.CPtr = llvmlite.new_builder(pool, func) + if func_builder is None: + return 0 + llvmlite.position_at_end(func_builder, entry_blk) + + # 进入函数作用域（嵌套符号表） + HandlesVar.enter_scope(trans.SymTab, HandlesVar.SCOPE_FUNCTION) + + # 为参数创建 alloca + if args_node is not None: + ags2: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags2.args is not None: + alist2: list[ast.AST | t.CPtr] | t.CPtr = ags2.args + an2: t.CSizeT = alist2.__len__() + for ai2 in range(an2): + arg2: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist2.get(ai2)) + if arg2 is not None and arg2.arg is not None: + # t.CVoid 表示空参：跳过，不创建 alloca + if arg2.annotation is not None: + if HandlesType._is_t_attr(arg2.annotation, "CVoid", imported_modules) != 0: + continue + param_ty2: llvmlite.LLVMType | t.CPtr = i32_ty + if arg2.annotation is not None: + resolved2: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg2.annotation, imported_modules, from_imports) + if resolved2 is not None: + param_ty2 = resolved2 + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(func_builder, param_ty2) + if alloca is not None: + HandlesVar.define_var(trans.SymTab, arg2.arg, alloca) + # 存储原始类型注解的类名（方法调用检测时，Ptr(i8) 回退到类名查找结构体） + if arg2.annotation is not None: + cls_nm: str = HandlesType.extract_class_name_from_annotation( + arg2.annotation, imported_modules) + if cls_nm is not None: + HandlesVar.set_var_annot_class_name( + trans.SymTab, arg2.arg, cls_nm) + pname2: t.CChar | t.CPtr = pool.alloc(32) + if pname2 is not None: + viperlib.snprintf(pname2, 32, "%%%s", arg2.arg) + param_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue( + pool, param_ty2, pname2) + llvmlite.build_store(func_builder, param_val, alloca) + + # 保存模块级作用域状态（仅非变量表相关） + old_func: llvmlite.Function | t.CPtr = trans._cur_func + old_builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + old_global_count: int = trans._global_name_count + old_nonlocal_count: int = trans._nonlocal_name_count + old_env_count: int = trans._closure_env_count + + trans._cur_func = func + trans._cur_builder = func_builder + # 清空 global/nonlocal 名称集合（新函数作用域） + HT.clear_scope_names(trans) + + # 预扫描函数体：为局部变量提前创建 alloca + body: list[ast.AST | t.CPtr] | t.CPtr = fd.children + if body is not None: + bn: t.CSizeT = body.__len__() + for bi in range(bn): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.pre_scan_allocas(trans, stmt) + + # 翻译函数体 + if body is not None: + bn2: t.CSizeT = body.__len__() + for bi2 in range(bn2): + stmt2: ast.AST | t.CPtr = body.get(bi2) + if stmt2 is not None: + HandlesBody.translate_stmt(trans, stmt2) + + # 如果函数体最后一条语句不是 Return，添加隐式 ret + last_is_return: int = 0 + if body is not None: + bn3: t.CSizeT = body.__len__() + if bn3 > 0: + last_stmt: ast.AST | t.CPtr = body.get(bn3 - 1) + if last_stmt is not None and last_stmt.kind() == ast.ASTKind.Return: + last_is_return = 1 + if last_is_return == 0: + zero_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + llvmlite.build_ret(func_builder, zero_val) + + # 恢复模块级作用域（退出函数作用域） + HandlesVar.exit_scope(trans.SymTab) + trans._cur_func = old_func + trans._cur_builder = old_builder + trans._global_name_count = old_global_count + trans._nonlocal_name_count = old_nonlocal_count + trans._closure_env_count = old_env_count + + return 0 + + +# ============================================================ +# translate_nested_function_def - 嵌套函数提升 + 闭包创建 +# +# 将嵌套函数提升为顶层函数 @__closure_{name}(i8* %env) -> i32， +# 在父函数中创建闭包对象 {i8* fn_ptr, i8* env_ptr} 并存储到 +# 以函数名命名的局部变量中。 +# +# 闭包结构: {i8* fn_ptr, i8* env_ptr} (16 字节, malloc 分配) +# Env 结构: {i8* ptr0, i8* ptr1, ...} (每个 nonlocal 变量 8 字节) +# ============================================================ +def translate_nested_function_def(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """翻译嵌套函数定义：提升为顶层函数 + 创建闭包""" + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(node) + if fd is None or fd.name is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + func_name: str = fd.name + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + # 1. 构建提升后的函数名: __closure_{name}（加 SHA1 前缀避免跨模块冲突） + promoted_name: t.CChar | t.CPtr = pool.alloc(64) + if promoted_name is not None: + viperlib.snprintf(promoted_name, 64, "__closure_%s", func_name) + mangled_promoted: str = _mangle_name(trans, promoted_name) + + # 2. 创建提升后的函数: define i32 @__closure_{name}(i8* %env) + func: llvmlite.Function | t.CPtr = llvmlite.create_function( + pool, mod, mangled_promoted, i32_ty) + if func is None: + return 0 + llvmlite.add_param(pool, func, i8_ptr_ty, "%env") + + # 3. 创建 entry 块 + builder + entry_blk: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, "entry") + if entry_blk is None: + return 0 + func_builder: llvmlite.IRBuilder | t.CPtr = llvmlite.new_builder(pool, func) + if func_builder is None: + return 0 + llvmlite.position_at_end(func_builder, entry_blk) + + # 4. 进入嵌套函数作用域（嵌套符号表） + HandlesVar.enter_scope(trans.SymTab, HandlesVar.SCOPE_FUNCTION) + + # 5. 创建 _env_ptr alloca 并存储 %env 参数 + env_alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(func_builder, i8_ptr_ty) + if env_alloca is not None: + HandlesVar.define_var(trans.SymTab, "_env_ptr", env_alloca) + env_param_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, i8_ptr_ty, "%env") + llvmlite.build_store(func_builder, env_param_val, env_alloca) + + # 6. 保存父函数作用域状态（仅非变量表相关） + old_func: llvmlite.Function | t.CPtr = trans._cur_func + old_builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + old_func_name: str = trans._cur_func_name + old_global_count: int = trans._global_name_count + old_nonlocal_count: int = trans._nonlocal_name_count + old_env_count: int = trans._closure_env_count + + # 7. 切换到嵌套函数作用域 + 清空 scope names + trans._cur_func = func + trans._cur_builder = func_builder + trans._cur_func_name = func_name + HT.clear_scope_names(trans) + + # 8. 预扫描函数体：为局部变量提前创建 alloca + body: list[ast.AST | t.CPtr] | t.CPtr = fd.children + if body is not None: + bn: t.CSizeT = body.__len__() + for bi in range(bn): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.pre_scan_allocas(trans, stmt) + + # 9. 翻译函数体（Nonlocal 语句会填充 _nonlocal_names） + if body is not None: + bn2: t.CSizeT = body.__len__() + for bi2 in range(bn2): + stmt2: ast.AST | t.CPtr = body.get(bi2) + if stmt2 is not None: + HandlesBody.translate_stmt(trans, stmt2) + + # 10. 隐式 ret + last_is_return: int = 0 + if body is not None: + bn3: t.CSizeT = body.__len__() + if bn3 > 0: + last_stmt: ast.AST | t.CPtr = body.get(bn3 - 1) + if last_stmt is not None and last_stmt.kind() == ast.ASTKind.Return: + last_is_return = 1 + if last_is_return == 0: + zero_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + llvmlite.build_ret(func_builder, zero_val) + + # 11. 保存嵌套函数的 nonlocal 信息 + nested_nonlocal_count: int = trans._nonlocal_name_count + + # 12. 退出嵌套函数作用域，回到父函数作用域（保留 _nonlocal_names 用于 env 创建） + HandlesVar.exit_scope(trans.SymTab) + trans._cur_func = old_func + trans._cur_builder = old_builder + # NOTE: _nonlocal_names 和 _nonlocal_name_count 仍为嵌套函数的值 + + # 13. 在父函数中创建闭包对象 + parent_builder: llvmlite.IRBuilder | t.CPtr = old_builder + if parent_builder is None: + # 无 builder（模块级嵌套函数？）→ 无法创建闭包，仅恢复状态 + trans._cur_func_name = old_func_name + trans._global_name_count = old_global_count + trans._nonlocal_name_count = old_nonlocal_count + trans._closure_env_count = old_env_count + return 0 + + # 13a. 创建 env: malloc(4 * nested_nonlocal_count) — env 直接存储 i32 值 + env_ptr: llvmlite.Value | t.CPtr = None + if nested_nonlocal_count > 0: + env_size: t.CInt64T = nested_nonlocal_count * 4 + env_size_val: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, env_size) + if env_size_val is not None: + env_size_val.Next = None + env_ptr = llvmlite.build_call( + parent_builder, "malloc", env_size_val, 1, i8_ptr_ty, 0) + + # 为每个 nonlocal 变量存储值到 env + if env_ptr is not None: + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i32_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i32_ty) + for ni in range(nested_nonlocal_count): + # 从 _nonlocal_names 缓冲区读取名称 + name_addr: t.CUInt64T = t.CUInt64T(trans._nonlocal_names) + ni * 8 + nl_slot: str | t.CPtr = (t.CVoid | t.CPtr)(name_addr) + nl_name: str = nl_slot[0] + if nl_name is not None: + # 在父函数作用域中查找（Current 已回到父作用域） + var_alloca: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, nl_name) + if var_alloca is not None: + # 加载变量值 + var_ty: llvmlite.LLVMType | t.CPtr = None + if var_alloca.Ty is not None: + var_ty = var_alloca.Ty.Pointee + if var_ty is None: + var_ty = i32_ty + var_val: llvmlite.Value | t.CPtr = llvmlite.build_load( + parent_builder, var_ty, var_alloca) + if var_val is not None: + # GEP to env[ni*4] + offset_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, ni * 4) + slot: llvmlite.Value | t.CPtr = llvmlite.build_gep( + parent_builder, i8_ty, env_ptr, offset_val) + if slot is not None: + # bitcast to i32* and store value + slot_typed: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + parent_builder, slot, i32_ptr_ty) + if slot_typed is not None: + # 类型转换（确保 var_val 是 i32） + coerced_val: llvmlite.Value | t.CPtr = HandlesExpr.coerce_to_type( + parent_builder, var_val, i32_ty) + if coerced_val is not None: + llvmlite.build_store(parent_builder, coerced_val, slot_typed) + + # 13b. 创建闭包结构: malloc(16) + closure_size_val: llvmlite.Value | t.CPtr = llvmlite.const_int64(pool, 16) + if closure_size_val is not None: + closure_size_val.Next = None + closure_ptr: llvmlite.Value | t.CPtr = llvmlite.build_call( + parent_builder, "malloc", closure_size_val, 1, i8_ptr_ty, 0) + if closure_ptr is None: + # malloc 失败，恢复状态 + trans._cur_func_name = old_func_name + trans._global_name_count = old_global_count + trans._nonlocal_name_count = old_nonlocal_count + trans._closure_env_count = old_env_count + return 0 + + # 13c. 存储 fn_ptr 到 offset 0 + # 创建函数指针类型 i32(i8*)* + fn_param_node: llvmlite.ParamNode | t.CPtr = llvmlite.new_param_node(pool, i8_ptr_ty) + fn_func_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Func(pool, i32_ty, fn_param_node, 1) + fn_func_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, fn_func_ty) + + # 创建函数引用 Value（类型为 i32(i8*)*，使用 SHA1 混淆名） + fn_ptr_name: t.CChar | t.CPtr = pool.alloc(64) + if fn_ptr_name is not None: + # SHA1 前缀名含 '.' 需要加引号（如 @"sha1.__closure_func"） + if string.strchr(mangled_promoted, '.') is not None: + viperlib.snprintf(fn_ptr_name, 64, "@\"%s\"", mangled_promoted) + else: + viperlib.snprintf(fn_ptr_name, 64, "@%s", mangled_promoted) + fn_ptr_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, fn_func_ptr_ty, fn_ptr_name) + + # bitcast 函数指针到 i8*（闭包存储 i8* 类型） + fn_as_i8ptr: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + parent_builder, fn_ptr_val, i8_ptr_ty) + fn_slot: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + parent_builder, closure_ptr, llvmlite.Ptr(pool, i8_ptr_ty)) + if fn_slot is not None and fn_as_i8ptr is not None: + llvmlite.build_store(parent_builder, fn_as_i8ptr, fn_slot) + + # 13d. 存储 env_ptr 到 offset 8 + eight_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 8) + env_slot_addr: llvmlite.Value | t.CPtr = llvmlite.build_gep( + parent_builder, i8_ty, closure_ptr, eight_val) + if env_slot_addr is not None: + env_slot: llvmlite.Value | t.CPtr = llvmlite.build_bitcast( + parent_builder, env_slot_addr, llvmlite.Ptr(pool, i8_ptr_ty)) + if env_slot is not None: + if env_ptr is not None: + llvmlite.build_store(parent_builder, env_ptr, env_slot) + else: + # 无 nonlocal 变量，存储 null + null_val: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, i8_ptr_ty, "null") + llvmlite.build_store(parent_builder, null_val, env_slot) + + # 13e. 将闭包指针存储到父函数的局部变量 {func_name} + closure_alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(parent_builder, i8_ptr_ty) + if closure_alloca is not None: + llvmlite.build_store(parent_builder, closure_ptr, closure_alloca) + HandlesVar.define_var(trans.SymTab, func_name, closure_alloca) + + # 14. 恢复 scope names + trans._cur_func_name = old_func_name + trans._global_name_count = old_global_count + trans._nonlocal_name_count = old_nonlocal_count + trans._closure_env_count = old_env_count + + return 1 + + +# ============================================================ +# 创建 LLVM 函数（简单版本，仅声明） +# ============================================================ +def create_function(pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + name: str, + args_node: ast.AST | t.CPtr, + ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Function | t.CPtr: + """创建 LLVM 函数并添加参数""" + if name is None or mod is None: + return None + + func: llvmlite.Function | t.CPtr = llvmlite.create_function(pool, mod, name, ret_ty) + if func is None: + return None + + if args_node is not None: + ags: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags.args is not None: + alist: list[ast.AST | t.CPtr] | t.CPtr = ags.args + an: t.CSizeT = alist.__len__() + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + for ai in range(an): + arg: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist.get(ai)) + if arg is not None and arg.arg is not None: + # t.CVoid 表示空参：跳过 + if arg.annotation is not None: + if arg.annotation.kind() == ast.ASTKind.Attribute: + at_cf: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(arg.annotation) + if at_cf.attr is not None and string.strcmp(at_cf.attr, "CVoid") == 0: + continue + param_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if arg.annotation is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, arg.annotation, None, None) + if resolved is not None: + param_ty = resolved + pname: t.CChar | t.CPtr = pool.alloc(32) + if pname is not None: + viperlib.snprintf(pname, 32, "%%%s", arg.arg) + llvmlite.add_param(pool, func, param_ty, pname) + + return func diff --git a/TransPyV/App/lib/core/Handles/HandlesIf.py b/TransPyV/App/lib/core/Handles/HandlesIf.py new file mode 100644 index 0000000..64f6af2 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesIf.py @@ -0,0 +1,141 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + + +# ============================================================ +# HandlesIf - if/elif/else 语句处理（Mixin 继承模式） +# +# 翻译 if 语句为 LLVM IR 控制流: +# br i1 %cond, label %then, label %else +# then: +# ... body ... +# br label %end +# else: +# ... orelse ... +# br label %end +# end: +# ============================================================ + + +@t.NoVTable +class IfHandle(HandlesBase.Mixin): + """if/elif/else 语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # Handle - 处理 if 语句，返回新增变量数 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译 if/elif/else 语句""" + if node is None: + return 0 + + trans: HT.Translator | t.CPtr = self.Trans + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + + if builder is None or func is None: + return 0 + + if_node: ast.If | t.CPtr = (ast.If | t.CPtr)(node) + + # 1. 求值条件表达式 + cond_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, trans.Module, if_node.test, + trans._funcs, trans._func_count, trans) + + # 2. 转换为 i1 条件 + # Compare/Not 表达式已返回 i1，直接使用；其他类型与 0 比较 + if cond_val is None: + cond_val = llvmlite.const_int32(pool, 0) + cond_bits: int = HandlesExpr.get_llvm_type_bits(cond_val.Ty) + if cond_bits == 1: + cond_i1: llvmlite.Value | t.CPtr = cond_val + else: + zero: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + cond_i1 = llvmlite.build_icmp( + builder, llvmlite.ICMP_NE, cond_val, zero) + + # 3. 创建基本块（使用 trans._label_counter 生成唯一标签名，不与 SSA 名共享） + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "if.then.%d", cnt) + then_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "if.end.%d", cnt) + merge_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # 检查是否有 else 分支 + has_else: int = 0 + orelse_list: list[ast.AST | t.CPtr] | t.CPtr = if_node.orelse + if orelse_list is not None: + if orelse_list.__len__() > 0: + has_else = 1 + + else_bb: llvmlite.BasicBlock | t.CPtr = None + if has_else == 1: + viperlib.snprintf(name_buf, 32, "if.else.%d", cnt) + else_bb = llvmlite.create_block(pool, func, name_buf) + + # 4. 发射条件分支 + if has_else == 1: + llvmlite.build_cond_br(builder, cond_i1, then_bb, else_bb) + else: + llvmlite.build_cond_br(builder, cond_i1, then_bb, merge_bb) + + # 5. 翻译 then body + llvmlite.position_at_end(builder, then_bb) + body: list[ast.AST | t.CPtr] | t.CPtr = if_node.children + if body is not None: + body_count: t.CSizeT = body.__len__() + for bi in range(body_count): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.translate_stmt(trans, stmt) + + # then 块未终止则跳到 merge + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, merge_bb) + + # 6. 翻译 else body（若有） + if has_else == 1: + llvmlite.position_at_end(builder, else_bb) + else_count: t.CSizeT = orelse_list.__len__() + for ei in range(else_count): + stmt: ast.AST | t.CPtr = orelse_list.get(ei) + if stmt is not None: + HandlesBody.translate_stmt(trans, stmt) + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, merge_bb) + + # 7. 定位到 merge 块继续后续代码 + llvmlite.position_at_end(builder, merge_bb) + return 0 + + +# ============================================================ +# NewIfHandle - 工厂函数 +# ============================================================ +def NewIfHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> IfHandle | t.CPtr: + h: IfHandle | t.CPtr = pool.alloc(IfHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, IfHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesImports.py b/TransPyV/App/lib/core/Handles/HandlesImports.py new file mode 100644 index 0000000..c129b08 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesImports.py @@ -0,0 +1,404 @@ +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + + +# ============================================================ +# HandlesImports - 导入语句处理（Mixin 继承模式） +# +# 管理 _imported_modules 和 _from_imports 字符串 +# 注意：str = bytes = t.CChar | t.CPtr = i8* +# ============================================================ + + +# ============================================================ +# 模块级工具函数（保留供外部调用） +# ============================================================ + +# ============================================================ +# 添加已导入模块名 +# ============================================================ +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, + imported_modules: str, + name: str) -> str: + """记录已导入的模块名，返回新的 imported_modules 字符串指针""" + if name is None: + return imported_modules + if imported_modules is None: + nlen: t.CSizeT = string.strlen(name) + buf: t.CChar | t.CPtr = pool.alloc(nlen + 1) + if buf is not None: + string.strcpy(buf, name) + return buf + return None + else: + old_len: t.CSizeT = string.strlen(imported_modules) + name_len: t.CSizeT = string.strlen(name) + new_len: t.CSizeT = old_len + 1 + name_len + buf2: t.CChar | t.CPtr = pool.alloc(new_len + 1) + if buf2 is not None: + string.strcpy(buf2, imported_modules) + buf2[old_len] = ' ' + string.strcpy(buf2 + old_len + 1, name) + return buf2 + return imported_modules + + +# ============================================================ +# 检查模块是否已导入 +# ============================================================ +def is_module_imported(imported_modules: str, name: str) -> int: + """检查模块是否已导入（词边界精确匹配，避免子串误匹配）""" + if name is None or imported_modules is None: + return 0 + name_len: t.CSizeT = string.strlen(name) + cur: t.CChar | t.CPtr = imported_modules + total_len: t.CSizeT = string.strlen(imported_modules) + ci: t.CSizeT = 0 + while ci < total_len: + while ci < total_len and cur[ci] == ' ': + ci += 1 + if ci >= total_len: + break + word_start: t.CSizeT = ci + while ci < total_len and cur[ci] != ' ': + ci += 1 + word_len: t.CSizeT = ci - word_start + if word_len == name_len: + match: int = 1 + for ei in range(name_len): + if cur[word_start + ei] != name[ei]: + match = 0 + break + if match == 1: + return 1 + return 0 + + +# ============================================================ +# 添加 from-import 名称映射 +# ============================================================ +def add_from_import(pool: memhub.MemBuddy | t.CPtr, + from_imports: str, + local_name: str, + module_name: str) -> str: + """添加 from-import 映射 "name:module"，返回新的 from_imports 字符串""" + if local_name is None or module_name is None: + return from_imports + entry: t.CChar | t.CPtr = pool.alloc(128) + if entry is None: + return from_imports + viperlib.snprintf(entry, 128, "%s:%s", local_name, module_name) + if from_imports is None: + return entry + else: + old_len: t.CSizeT = string.strlen(from_imports) + entry_len: t.CSizeT = string.strlen(entry) + new_len: t.CSizeT = old_len + 1 + entry_len + buf: t.CChar | t.CPtr = pool.alloc(new_len + 1) + if buf is not None: + string.strcpy(buf, from_imports) + buf[old_len] = ' ' + string.strcpy(buf + old_len + 1, entry) + return buf + return from_imports + + +# ============================================================ +# 查找 from-import 名称 → 返回模块名或 None +# +# allow_star_fallback: 是否允许 star import 回退（默认 1=允许）。 +# 当查询明确模块名（如 "BuildPipeline"）时，应传 0 禁用回退， +# 避免被 star import 模块名误导（如 from stdint import * 后 +# 查询 "BuildPipeline" 错误回退到 "stdint"）。 +# ============================================================ +def lookup_from_import(from_imports: str, name: str, + allow_star_fallback: int = 1) -> str: + """查找 from-import 名称，返回模块名或 None + + 支持 star import: 如果 from_imports 中有 "*:module" 条目， + 且未找到精确名称匹配，且 allow_star_fallback != 0，则返回 star import 的模块名。 + """ + if name is None or from_imports is None: + return None + name_len: t.CSizeT = string.strlen(name) + cur: t.CChar | t.CPtr = from_imports + ci: t.CSizeT = 0 + total_len: t.CSizeT = string.strlen(from_imports) + star_mod: str = None + while ci < total_len: + # 跳过前导空格 + while ci < total_len and cur[ci] == ' ': + ci += 1 + if ci >= total_len: + break + # 找到 ':' 的位置 + colon_pos: t.CSizeT = ci + while colon_pos < total_len and cur[colon_pos] != ':' and cur[colon_pos] != ' ': + colon_pos += 1 + if colon_pos >= total_len or cur[colon_pos] != ':': + break + entry_name_len: t.CSizeT = colon_pos - ci + # 检测 star import ("*:module") + if entry_name_len == 1 and cur[ci] == '*': + mod_start: t.CSizeT = colon_pos + 1 + mod_end: t.CSizeT = mod_start + while mod_end < total_len and cur[mod_end] != ' ' and cur[mod_end] != '\0': + mod_end += 1 + star_mod = cur + mod_start + # 比较名称 + elif entry_name_len == name_len: + match: int = 1 + ei: t.CSizeT = 0 + while ei < name_len: + if cur[ci + ei] != name[ei]: + match = 0 + break + ei += 1 + if match == 1: + mod_start2: t.CSizeT = colon_pos + 1 + mod_end2: t.CSizeT = mod_start2 + while mod_end2 < total_len and cur[mod_end2] != ' ' and cur[mod_end2] != '\0': + mod_end2 += 1 + return cur + mod_start2 + # 跳到下一个条目 + ci = colon_pos + while ci < total_len and cur[ci] != ' ': + ci += 1 + # 未找到精确匹配，仅在允许时回退到 star import + if allow_star_fallback == 0: + return None + return star_mod + + +# ============================================================ +# _resolve_relative_module - 解析相对导入为完整模块名 +# +# 对于 from .X import Y（level=1, module="X"）在包 pkg 中： +# 解析为 "pkg.X" +# 对于 from . import Y（level=1, module=None）在包 pkg 中： +# 解析为 "pkg" +# 对于 from ..X import Y（level=2, module="X"）在包 pkg.sub 中： +# 解析为 "pkg.X"（先从 pkg.sub 上溯一级到 pkg，再追加 .X） +# +# Args: +# pool: 内存池 +# current_package: 当前文件所属包名（如 "llvmlite"），None 表示无包 +# level: 相对导入级别（0=绝对，1=., 2=..） +# module: ImportFrom 的 module 字段（可能为 None） +# +# Returns: +# 解析后的完整模块名；绝对导入（level<=0）直接返回 module； +# 无法解析时返回 module（回退到原始值） +# ============================================================ +def _resolve_relative_module(pool: memhub.MemBuddy | t.CPtr, + current_package: str, + level: t.CInt, + module: str) -> str: + """解析相对导入为完整模块名""" + if level <= 0: + return module + if current_package is None: + return module + + # 从 current_package 开始，上溯 (level-1) 级 + pkg: str = current_package + up: t.CInt = level - 1 + while up > 0: + pkg_len: t.CSizeT = string.strlen(pkg) + last_dot: t.CSizeT = 0 + found: int = 0 + i: t.CSizeT = 0 + while i < pkg_len: + if pkg[i] == '.': + last_dot = i + found = 1 + i += 1 + if found == 0: + # 无更多父级，包变为空 + pkg = None + break + # 截断到最后一个 '.' 处 + new_pkg: str = pool.alloc(last_dot + 1) + if new_pkg is None: + return module + string.strncpy(new_pkg, pkg, last_dot) + new_pkg[last_dot] = '\0' + pkg = new_pkg + up -= 1 + + if module is None: + # from . import Y → 模块就是包本身 + return pkg + if pkg is None: + # 包已上溯到空，直接用 module + return module + # 拼接 pkg + "." + module + pkg_len2: t.CSizeT = string.strlen(pkg) + mod_len: t.CSizeT = string.strlen(module) + buf: str = pool.alloc(pkg_len2 + 1 + mod_len + 1) + if buf is None: + return module + string.strcpy(buf, pkg) + buf[pkg_len2] = '.' + string.strcpy(buf + pkg_len2 + 1, module) + return buf + + +# ============================================================ +# compute_package_from_relpath - 从相对路径计算包名 +# +# 包名 = 文件所在目录路径，将 / 和 \ 替换为 . +# 对于顶级文件（无目录分隔符），返回 None +# +# 示例： +# "llvmlite/__init__.py" → "llvmlite" +# "llvmlite/__types.py" → "llvmlite" +# "ast/parser.py" → "ast" +# "ast.py" → None（顶级文件） +# ============================================================ +def compute_package_from_relpath(pool: memhub.MemBuddy | t.CPtr, + rel_path: str) -> str: + """从相对路径计算包名（目录部分，分隔符替换为 .）""" + if rel_path is None: + return None + rlen: t.CSizeT = string.strlen(rel_path) + # 找最后一个 / 或 \ + last_sep: t.CSizeT = 0 + found: int = 0 + i: t.CSizeT = 0 + while i < rlen: + ch: t.CChar = rel_path[i] + if ch == '/' or ch == '\\': + last_sep = i + found = 1 + i += 1 + if found == 0: + # 无目录分隔符 → 顶级文件，无包 + return None + # 复制目录部分，将 / 和 \ 替换为 . + dir_len: t.CSizeT = last_sep + buf: str = pool.alloc(dir_len + 1) + if buf is None: + return None + j: t.CSizeT = 0 + while j < dir_len: + ch2: t.CChar = rel_path[j] + if ch2 == '/' or ch2 == '\\': + buf[j] = '.' + else: + buf[j] = ch2 + j += 1 + buf[dir_len] = '\0' + return buf + + +# ============================================================ +# ImportsHandle - 导入语句处理器（Mixin 继承模式） +# +# 方法版本：直接更新 trans._imported_modules / trans._from_imports +# ============================================================ +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + """导入语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # HandleImport - 处理 import 语句，更新 trans._imported_modules + # ============================================================ + def HandleImport(self, node: ast.AST | t.CPtr) -> int: + """处理 import 语句，返回 0""" + imp: ast.Import | t.CPtr = (ast.Import | t.CPtr)(node) + if imp is None: + return 0 + names: list[ast.AST | t.CPtr] | t.CPtr = imp.names + if names is None: + return 0 + nn: t.CSizeT = names.__len__() + for ni in range(nn): + alias: ast.Alias | t.CPtr = (ast.Alias | t.CPtr)(names.get(ni)) + if alias is not None and alias.name is not None: + pool_val: memhub.MemBuddy | t.CPtr = self.Trans.Pool + im_val: str = self.Trans._imported_modules + self.Trans._imported_modules = add_imported_module( + pool_val, im_val, alias.name) + # 别名也加入导入模块列表（用于模块限定构造器检查 Module.Class()） + if alias.asname is not None: + self.Trans._imported_modules = add_imported_module( + pool_val, self.Trans._imported_modules, alias.asname) + return 0 + + # ============================================================ + # HandleImportFromModule - 处理 from X import Y 的模块部分 + # + # 对相对导入（level > 0），使用 trans.CurrentPackage 解析为 + # 完整模块名（如 __types → llvmlite.__types），确保 .deps.txt + # 记录的模块名与 SHA1 映射表一致。 + # ============================================================ + def HandleImportFromModule(self, node: ast.AST | t.CPtr) -> int: + """处理 from X import Y 的模块部分，更新 trans._imported_modules""" + impf: ast.ImportFrom | t.CPtr = (ast.ImportFrom | t.CPtr)(node) + if impf is None: + return 0 + # 解析模块名：相对导入需补充包前缀 + resolved: str = _resolve_relative_module( + self.Trans.Pool, self.Trans.CurrentPackage, + impf.level, impf.module) + if resolved is not None: + self.Trans._imported_modules = add_imported_module( + self.Trans.Pool, self.Trans._imported_modules, resolved) + return 0 + + # ============================================================ + # HandleImportFromNames - 处理 from X import Y 的名称部分 + # + # 同样使用解析后的完整模块名，确保 from-import 映射 + # （如 LLVMType:llvmlite.__types）能正确查到 SHA1。 + # ============================================================ + def HandleImportFromNames(self, node: ast.AST | t.CPtr) -> int: + """处理 from X import Y 的名称部分，更新 trans._from_imports""" + impf: ast.ImportFrom | t.CPtr = (ast.ImportFrom | t.CPtr)(node) + if impf is None: + return 0 + # 解析模块名：相对导入需补充包前缀 + resolved: str = _resolve_relative_module( + self.Trans.Pool, self.Trans.CurrentPackage, + impf.level, impf.module) + if resolved is not None: + names: list[ast.AST | t.CPtr] | t.CPtr = impf.names + if names is not None: + nn: t.CSizeT = names.__len__() + for ni in range(nn): + alias: ast.Alias | t.CPtr = (ast.Alias | t.CPtr)(names.get(ni)) + if alias is not None and alias.name is not None: + local_name: str = alias.name + if alias.asname is not None: + local_name = alias.asname + self.Trans._from_imports = add_from_import( + self.Trans.Pool, self.Trans._from_imports, + local_name, resolved) + # 命名空间隔离：from-import 的名称标记为可见结构体 + HandlesStruct.add_visible_struct(self.Trans.Pool, local_name) + return 0 + + +# ============================================================ +# NewImportsHandle - 工厂函数 +# ============================================================ +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: + h: ImportsHandle | t.CPtr = pool.alloc(ImportsHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, ImportsHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesMain.py b/TransPyV/App/lib/core/Handles/HandlesMain.py new file mode 100644 index 0000000..3b29f02 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesMain.py @@ -0,0 +1,314 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign + + +# ============================================================ +# HandlesMain - 模块级翻译入口 + wrapper main 创建 +# +# 从 translator.py 拆分出来，负责: +# 1. create_wrapper_main() - 无用户 main 时创建包装 main +# 2. translate_children() - 遍历 AST 子节点并分派翻译 +# +# trans 单参模式：所有共享状态从 trans 获取，无需 11 个参数 +# 注意: str = bytes = t.CChar | t.CPtr = i8* +# ============================================================ + + +# ============================================================ +# _register_cexport_from_funcdef - Phase 1a 预注册 CExport/State 函数 +# +# 解决翻译顺序依赖问题：Phase 1b 按字母序翻译，后翻译的模块的 +# CExport 函数无法被先翻译的模块正确识别为裸名调用。 +# Phase 1a 预注册所有 CExport/State 函数名到全局表。 +# ============================================================ +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """预注册 CExport/State 函数到全局表（仅注册，不生成 IR）""" + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(node) + if fd is None or fd.name is None: + return 0 + if trans is None or trans.ModuleSha1 is None: + return 0 + + # 检查返回类型是否有 CExport 或 State 标记 + has_export: int = 0 + has_state: int = 0 + if fd.returns is not None: + has_export = HandlesType.has_decorator_marker(fd.returns, "CExport") + has_state = HandlesType.has_decorator_marker(fd.returns, "State") + + if has_export != 0 or has_state != 0: + HandlesExprCall.register_cexport_func(trans.ModuleSha1, fd.name) + + return 0 + + +# ============================================================ +# translate_children - 遍历 AST 子节点并分派翻译 +# +# 对应 TransPyC translator._translate_children() +# 共享状态从 trans 获取，imported_modules/from_imports 更新到 trans +# ============================================================ +def translate_children(trans: HT.Translator | t.CPtr, + tree: ast.AST | t.CPtr) -> int: + """遍历 tree.children 并翻译每个子节点 + + Args: + trans: 翻译器（含所有共享状态） + tree: AST 模块节点 + + Returns: + t.CInt: 新增的变量数 + """ + ch: list[ast.AST | t.CPtr] | t.CPtr = tree.children + if ch is None: + return 0 + + cn_count: t.CSizeT = ch.__len__() + added_total: int = 0 + + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is None: continue + kd: int = child.kind() + + if kd == ast.ASTKind.Import: + trans.ImportsH.HandleImport(child) + elif kd == ast.ASTKind.ImportFrom: + trans.ImportsH.HandleImportFromModule(child) + trans.ImportsH.HandleImportFromNames(child) + elif kd == ast.ASTKind.FunctionDef: + # Phase 1a 声明模式：只注册 CExport/State 函数到全局表（解决翻译顺序依赖） + # Phase 1b 全量翻译：正常翻译函数体 + if trans._declare_only == 0: + added: int = HandlesFunctions.translate_function_def(trans, child) + added_total += added + elif trans._declare_only == 1: + _register_cexport_from_funcdef(trans, child) + elif kd == ast.ASTKind.ClassDef: + # ClassDef 在模块级直接处理（不需要 builder） + # _declare_only=2（import扫描模式）时跳过，只处理 import 依赖 + if trans._declare_only != 2: + cd_node: ast.ClassDef | t.CPtr = (ast.ClassDef | t.CPtr)(child) + cd_name: str = "?" if cd_node is None or cd_node.name is None else cd_node.name + HandlesClassDef.translate_class_def(trans, child) + elif trans._declare_only == 0 and trans._cur_builder is not None: + # 有 builder → 委托 HandlesBody 分派 + added = HandlesBody.translate_stmt(trans, child) + added_total += added + elif kd == ast.ASTKind.AnnAssign and trans._declare_only != 2: + # 模块级 AnnAssign + # _declare_only=2（import扫描模式）时跳过 + # _declare_only=1（struct注册模式）时只处理 CDefine（在 handle_module_level_var 内部判断） + # _declare_only=0（全量翻译）时处理所有模块级 AnnAssign + added = handle_module_level_var(trans, child) + added_total += added + elif trans._declare_only == 0 and kd == ast.ASTKind.Assign: + # 无 builder 的模块级 Assign → 创建全局变量（仅全量翻译模式） + added = handle_module_level_var(trans, child) + added_total += added + + return added_total + + +# ============================================================ +# handle_module_level_var - 模块级变量声明 → 创建 LLVM 全局变量 +# +# 当用户已定义 main（无 wrapper main builder）时，模块级 +# AnnAssign/Assign 创建全局变量 @var_name 并注册到 SymTab 模块作用域 +# ============================================================ +def handle_module_level_var(trans: HT.Translator | t.CPtr, + node: ast.AST | t.CPtr) -> int: + """处理模块级变量声明，创建全局变量""" + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + + # CDefine 注解: 编译期常量，不创建全局变量 + # 注册到全局 CDefine 表供 t.CArray[elem_ty, NAME] 解析 + k: int = node.kind() + if k == ast.ASTKind.AnnAssign: + aa_cd: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(node) + if aa_cd is not None and aa_cd.target is not None: + if aa_cd.target.kind() == ast.ASTKind.Name: + nm_cd: ast.Name | t.CPtr = (ast.Name | t.CPtr)(aa_cd.target) + if nm_cd.id is not None: + if HandlesAnnAssign.is_cdefine_annotation(aa_cd.annotation) != 0: + val_cd: int = HandlesAnnAssign.extract_cdefine_int_value(aa_cd.value) + HandlesType.register_cdefine_constant(pool, nm_cd.id, val_cd) + return 0 + + # _declare_only=1（struct注册模式）时只处理 CDefine，不创建全局变量 + # CDefine 已在上面处理并返回，到这里说明不是 CDefine，直接跳过 + if trans._declare_only == 1: + return 0 + + var_name: str = None + var_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + init_val: t.CInt64T = 0 + has_init: int = 0 + + if k == ast.ASTKind.AnnAssign: + aa: ast.AnnAssign | t.CPtr = (ast.AnnAssign | t.CPtr)(node) + if aa is None or aa.target is None: + return 0 + if aa.target.kind() != ast.ASTKind.Name: + return 0 + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(aa.target) + var_name = nm.id + # 解析类型 + if aa.annotation is not None: + resolved: llvmlite.LLVMType | t.CPtr = HandlesType.resolve_annotation_type( + pool, aa.annotation, trans._imported_modules, trans._from_imports) + if resolved is not None: + var_ty = resolved + # 解析初始值 + if aa.value is not None and aa.value.kind() == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(aa.value) + if cn.const_kind == ast.CONST_INT: + init_val = cn.int_val + has_init = 1 + elif k == ast.ASTKind.Assign: + asgn: ast.Assign | t.CPtr = (ast.Assign | t.CPtr)(node) + if asgn is None or asgn.targets is None: + return 0 + targets: list[ast.AST | t.CPtr] | t.CPtr = asgn.targets + if targets.__len__() < 1: + return 0 + t0: ast.AST | t.CPtr = targets.get(0) + if t0 is None or t0.kind() != ast.ASTKind.Name: + return 0 + nm2: ast.Name | t.CPtr = (ast.Name | t.CPtr)(t0) + var_name = nm2.id + if asgn.value is not None and asgn.value.kind() == ast.ASTKind.Constant: + cn2: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(asgn.value) + if cn2.const_kind == ast.CONST_INT: + init_val = cn2.int_val + has_init = 1 + + if var_name is None: + return 0 + + # 检查是否已注册 + existing: llvmlite.Value | t.CPtr = HandlesVar.lookup_module_var( + trans.SymTab, var_name) + if existing is not None: + return 0 + + # 创建全局变量 @var_name + gv: llvmlite.GlobalVariable | t.CPtr = llvmlite.new_global_variable(pool, var_name, var_ty) + if gv is None: + return 0 + llvmlite.module_add_global(mod, gv) + + # 设置初始值（有初始值时清除 external linkage，因为 LLVM 22+ 不允许 external global 带初始值） + gv.Linkage = None + if has_init != 0: + init_buf: t.CChar | t.CPtr = pool.alloc(48) + if init_buf is not None: + viperlib.snprintf(init_buf, 48, "%lld", init_val) + gv.Initializer = init_buf + else: + gv.Initializer = "0" + + # 创建 Value 引用（@var_name, 类型为 var_ty*） + var_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, var_ty) + ref_name: t.CChar | t.CPtr = pool.alloc(64) + if ref_name is not None: + viperlib.snprintf(ref_name, 64, "@%s", var_name) + gv_ref: llvmlite.Value | t.CPtr = llvmlite.SSAValue(pool, var_ptr_ty, ref_name) + + # 注册到模块作用域 + if HandlesVar.define_module_var(trans.SymTab, var_name, gv_ref) == 0: + return 1 + return 0 + + +# ============================================================ +# create_wrapper_main - 创建包装 main 函数 +# +# 当用户未定义 main 函数时调用。 +# 创建 main() -> i32 函数 → 设置 trans._cur_func/_cur_builder → +# 预扫描 alloca → 翻译子节点 → ret 0 +# ============================================================ +def create_wrapper_main(trans: HT.Translator | t.CPtr, + tree: ast.AST | t.CPtr, + i32_ty: llvmlite.LLVMType | t.CPtr) -> int: + """创建包装 main 函数并翻译所有顶层语句 + + Args: + trans: 翻译器（含所有共享状态） + tree: AST 模块节点 + i32_ty: i32 LLVMType + + Returns: + t.CInt: 新增的变量数 + """ + pool: memhub.MemBuddy | t.CPtr = trans.Pool + mod: llvmlite.LLVMModule | t.CPtr = trans.Module + + main_func: llvmlite.Function | t.CPtr = llvmlite.create_function( + pool, mod, "main", i32_ty) + if main_func is None: + stdio.printf("[TR] CreateFunction main returned NULL\n") + return 0 + + entry_blk: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block( + pool, main_func, "entry") + if entry_blk is None: + stdio.printf("[TR] CreateBlock returned NULL\n") + return 0 + + builder: llvmlite.IRBuilder | t.CPtr = llvmlite.new_builder(pool, main_func) + if builder is None: + stdio.printf("[TR] NewBuilder returned NULL\n") + return 0 + llvmlite.position_at_end(builder, entry_blk) + + # 设置当前翻译上下文（供 Handle 通过 self.Trans._cur_builder 访问） + trans._cur_func = main_func + trans._cur_builder = builder + + # 预处理模块级变量：为 AnnAssign/Assign 创建 LLVM 全局变量 + # 必须在 pre_scan_allocas 之前执行，否则 PreScan 会创建局部 alloca， + # 导致其他函数通过 SSA 编号引用 wrapper main 的局部变量（无效 IR） + ch: list[ast.AST | t.CPtr] | t.CPtr = tree.children + if ch is not None: + cn_count: t.CSizeT = ch.__len__() + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is not None: + kd: int = child.kind() + if kd == ast.ASTKind.AnnAssign or kd == ast.ASTKind.Assign: + handle_module_level_var(trans, child) + + # 预扫描顶层语句：为函数内 AnnAssign 提前创建 alloca + # （模块级变量已在上面注册到模块作用域，PreScan 的 lookup_current 会跳过它们） + if ch is not None: + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is not None: + HandlesBody.pre_scan_allocas(trans, child) + + # 翻译子节点 + added_total: int = translate_children(trans, tree) + + # 返回 0 + zero_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + llvmlite.build_ret(builder, zero_val) + + return added_total diff --git a/TransPyV/App/lib/core/Handles/HandlesNonlocal.py b/TransPyV/App/lib/core/Handles/HandlesNonlocal.py new file mode 100644 index 0000000..6922f7e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesNonlocal.py @@ -0,0 +1,127 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar + + +# ============================================================ +# HandlesNonlocal - nonlocal 变量访问（通过闭包 env） +# +# 闭包 env 结构: {i8* ptr0, i8* ptr1, ...} (每个 nonlocal 变量一个 i8* 指针) +# 指针指向原始变量（可能是 alloca 或全局变量） +# +# 在提升的嵌套函数中: +# 1. 函数签名: define i32 @__closure_{name}(i8* %env) +# 2. 入口处: %env_alloca = alloca i8*; store i8* %env, i8** %env_alloca +# 3. 访问 nonlocal var: +# a. %env_ptr = load i8*, i8** %env_alloca +# b. %addr = gep i8, i8* %env_ptr, i32 (index * 8) +# c. %ptr_addr = bitcast i8* %addr to i8** +# d. %var_ptr_raw = load i8*, i8** %ptr_addr +# e. %var_ptr = bitcast i8* %var_ptr_raw to i32* +# f. read: %val = load i32, i32* %var_ptr +# write: store i32 %new, i32* %var_ptr +# ============================================================ + + +# ============================================================ +# 获取或创建 _env_ptr 变量（存储 env 参数的 alloca） +# ============================================================ +def get_env_ptr_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: + """获取或创建 _env_ptr 变量（存储闭包 env 指针）""" + env_var: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + trans.SymTab, "_env_ptr") + if env_var is not None: + return env_var + # 创建 alloca 存储 env + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, i8_ptr_ty) + if alloca is None: + return None + HandlesVar.define_var(trans.SymTab, "_env_ptr", alloca) + return alloca + + +# ============================================================ +# 加载 nonlocal 变量值 → 返回 i32 Value +# +# env 直接存储 i32 值（不是指针），每个 nonlocal 变量占 4 字节 +# ============================================================ +def load_nonlocal_var(trans: HT.Translator | t.CPtr, + name: str) -> llvmlite.Value | t.CPtr: + """从闭包 env 加载 nonlocal 变量值（env 直接存储 i32 值）""" + idx: int = HT.get_nonlocal_index(trans, name) + if idx < 0: + return None + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i32_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i32_ty) + + # 1. 加载 env_ptr + env_alloca: llvmlite.Value | t.CPtr = get_env_ptr_var(trans) + if env_alloca is None: + return None + env_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, env_alloca) + if env_ptr is None: + return None + + # 2. GEP to offset (idx * 4) — env 直接存储 i32 值 + offset_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, idx * 4) + addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, env_ptr, offset_val) + if addr is None: + return None + + # 3. Bitcast to i32* and load + var_ptr: llvmlite.Value | t.CPtr = llvmlite.build_bitcast(builder, addr, i32_ptr_ty) + if var_ptr is None: + return None + return llvmlite.build_load(builder, i32_ty, var_ptr) + + +# ============================================================ +# 获取 nonlocal 变量指针（用于 store） +# +# 返回 env 中 i32 槽位的地址（i32*），用于直接 store +# ============================================================ +def get_nonlocal_var_ptr(trans: HT.Translator | t.CPtr, + name: str) -> llvmlite.Value | t.CPtr: + """获取 nonlocal 变量在 env 中的地址（i32*），用于 store 操作""" + idx: int = HT.get_nonlocal_index(trans, name) + if idx < 0: + return None + + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i32_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i32_ty) + + # 1. 加载 env_ptr + env_alloca: llvmlite.Value | t.CPtr = get_env_ptr_var(trans) + if env_alloca is None: + return None + env_ptr: llvmlite.Value | t.CPtr = llvmlite.build_load(builder, i8_ptr_ty, env_alloca) + if env_ptr is None: + return None + + # 2. GEP to offset (idx * 4) — env 直接存储 i32 值 + offset_val: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, idx * 4) + addr: llvmlite.Value | t.CPtr = llvmlite.build_gep(builder, i8_ty, env_ptr, offset_val) + if addr is None: + return None + + # 3. Bitcast to i32* and return (直接指向 env 中的 i32 槽位) + return llvmlite.build_bitcast(builder, addr, i32_ptr_ty) diff --git a/TransPyV/App/lib/core/Handles/HandlesReturn.py b/TransPyV/App/lib/core/Handles/HandlesReturn.py new file mode 100644 index 0000000..9b077f7 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesReturn.py @@ -0,0 +1,101 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar + + +# ============================================================ +# HandlesReturn - Return 语句处理（Mixin 继承模式） +# ============================================================ + + +@t.NoVTable +class ReturnHandle(HandlesBase.Mixin): + """Return 语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # Handle - 处理 Return 语句，返回 0 + # + # 翻译返回值（若有）并生成 ret 指令 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译 Return 语句""" + rt: ast.Return | t.CPtr = (ast.Return | t.CPtr)(node) + if rt is None: + return 0 + + pool: memhub.MemBuddy | t.CPtr = self.Trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = self.Trans._cur_builder + mod: llvmlite.LLVMModule | t.CPtr = self.Trans.Module + + val: llvmlite.Value | t.CPtr = None + if rt.value is not None: + # return self: 直接返回指针（self 是 SSA 参数 Ptr(struct_ty)） + # translate_value 会 load 得到结构体值，但 return self 需要指针本身 + if rt.value.kind() == ast.ASTKind.Name: + ret_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(rt.value) + if ret_nm is not None and ret_nm.id is not None: + if string.strcmp(ret_nm.id, "self") == 0: + self_ptr: llvmlite.Value | t.CPtr = HandlesVar.lookup_var( + self.Trans.SymTab, "self") + if self_ptr is not None: + val = self_ptr + if val is None: + val = HandlesExpr.translate_value( + builder, pool, mod, rt.value, + self.Trans._funcs, self.Trans._func_count, self.Trans) + + # val 为 None 时（裸 return）：检查当前函数返回类型 + # void 函数（如 __init__/__before_init__）生成 ret void，否则 ret i32 0 + if val is None: + cur_func: llvmlite.Function | t.CPtr = self.Trans._cur_func + is_void_ret: int = 0 + if cur_func is not None: + ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(cur_func) + if ret_ty is not None: + match ret_ty: + case llvmlite.LLVMType.Void(): + is_void_ret = 1 + if is_void_ret != 0: + llvmlite.build_ret_void(builder) + return 0 + val = llvmlite.const_int32(pool, 0) + + # 类型转换：确保 val 类型与函数返回类型匹配 + # 处理 i1（bool 比较结果）→ i8（t.CBool）等情况 + # i1 → 更宽整数用 zext（bool 语义：1 保持 1，而非 sext 的 0xFF） + cur_func_rt: llvmlite.Function | t.CPtr = self.Trans._cur_func + if cur_func_rt is not None and val is not None and val.Ty is not None: + ret_ty_rt: llvmlite.LLVMType | t.CPtr = llvmlite.function_get_ret_ty(cur_func_rt) + if ret_ty_rt is not None: + val_bits: int = HandlesExpr.get_llvm_type_bits(val.Ty) + ret_bits: int = HandlesExpr.get_llvm_type_bits(ret_ty_rt) + if val_bits != 0 and ret_bits != 0 and val_bits != ret_bits: + if val_bits == 1 and val_bits < ret_bits: + val = llvmlite.build_zext(builder, val, ret_ty_rt) + else: + val = HandlesExpr.coerce_to_type(builder, val, ret_ty_rt) + llvmlite.build_ret(builder, val) + return 0 + + +# ============================================================ +# NewReturnHandle - 工厂函数 +# ============================================================ +def NewReturnHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> ReturnHandle | t.CPtr: + h: ReturnHandle | t.CPtr = pool.alloc(ReturnHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, ReturnHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/HandlesStruct.py b/TransPyV/App/lib/core/Handles/HandlesStruct.py new file mode 100644 index 0000000..460e46d --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesStruct.py @@ -0,0 +1,975 @@ +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + + +# ============================================================ +# HandlesStruct - 结构体类型注册和字段查找 +# +# 管理 class 定义的结构体类型信息: +# - 类名 → LLVM StructType +# - 字段名 → 字段索引和类型 +# +# 使用全局数组存储，线性查找（结构体数量通常很少） +# ============================================================ + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + + +# ============================================================ +# FieldEntry - 字段信息条目 +# ============================================================ +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr # 字段名（字符串） + Index: int # 字段在结构体中的索引 + Ty: llvmlite.LLVMType | t.CPtr # 字段的 LLVM 类型 + DefaultVal: ast.AST | t.CPtr # 默认值 AST 节点（None=无默认值） + AnnotClassName: t.CChar | t.CPtr # 原始类型注解的类名（str 别名在结构体字段中触发编译器 bug，改用显式联合类型） + + +# ============================================================ +# StructEntry - 结构体注册条目 +# ============================================================ +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr # 类名 + Ty: llvmlite.LLVMType | t.CPtr # LLVM StructType + FieldCount: int # 字段数量 + Fields: FieldEntry | t.CPtr # 字段数组（FIELD_MAX 个槽位） + IsUnion: int # 1=联合体 / 0=普通结构体 + IsOOP: int # 1=OOP结构体（有方法） / 0=纯内存结构体 + HasInit: int # 1=有__init__方法 / 0=无 + HasNew: int # 1=有__new__方法 / 0=无 + HasVTable: int # 1=有虚表 / 0=无虚表 + IsNoVTable: int # 1=明确标记 @t.NoVTable / 0=未标记 + ParentName: t.CChar | t.CPtr # 父类名（None=无父类） + VTableMethodCount: int # 虚表中的方法数量 + VTableMethods: t.CChar | t.CPtr # 虚表方法名数组（每个方法名 str，VTableMethodCount 个） + ModuleSha1: t.CChar | t.CPtr # 定义该类的模块 SHA1（None=无 SHA1） + + +# ============================================================ +# 全局注册表（静态分配） +# ============================================================ +_struct_table: StructEntry | t.CPtr = None +_struct_count: int = 0 + + +# ============================================================ +# 跨模块命名空间隔离：per-file 结构体可见性 +# +# _visible_structs: 当前文件可见的结构体名（HashTable，O(1) 查找） +# _strict_visibility: 1=严格模式（仅本地+import可见）/ 0=宽松模式（全部可见） +# +# 每个文件翻译前调用 reset_visible_structs 重置 +# ============================================================ +_visible_structs: hashtable.HashTable | t.CPtr = None +_strict_visibility: int = 0 + + +# ============================================================ +# reset_visible_structs — 重置可见性状态（每个文件翻译前调用） +# ============================================================ +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int): + """重置可见性状态（每个文件翻译前调用） + + pool: 内存分配器（严格模式下创建 HashTable） + strict: 1=严格模式（用户文件），0=宽松模式（includes 文件） + """ + global _visible_structs + global _strict_visibility + _strict_visibility = strict + if strict != 0: + # 严格模式：创建新的 HashTable 记录可见结构体 + _visible_structs = hashtable.HashTable(pool) + else: + # 宽松模式：全部可见，不需要 HashTable + _visible_structs = None + + +# ============================================================ +# add_visible_struct — 添加可见结构体名 +# ============================================================ +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str): + """添加可见结构体名（类定义时和 from-import 时调用）""" + global _visible_structs + if name is None: + return + # 宽松模式：全部可见，无需记录 + if _strict_visibility == 0: + return + if _visible_structs is None: + return + # 已存在则跳过（HashTable.__setitem__ 会覆盖，此处提前检查避免重复分配） + if _visible_structs.__contains__(name) != 0: + return + _visible_structs.set_int(name, 1) + + +# ============================================================ +# is_struct_visible — 检查结构体是否在当前文件可见 +# ============================================================ +def is_struct_visible(name: str) -> int: + """检查结构体是否可见，返回 1=可见 / 0=不可见""" + if _strict_visibility == 0: + return 1 + if name is None or _visible_structs is None: + return 0 + return _visible_structs.__contains__(name) + + +# ============================================================ +# 初始化结构体注册表 +# ============================================================ +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: + """初始化结构体注册表，返回 1 成功""" + global _struct_table + global _struct_count + + if _struct_table is not None: + return 1 + + entry_size: t.CSizeT = StructEntry.__sizeof__() + _struct_table = pool.alloc(entry_size * STRUCT_MAX) + if _struct_table is None: + return 0 + string.memset(_struct_table, 0, entry_size * STRUCT_MAX) + _struct_count = 0 + return 1 + + +# ============================================================ +# _get_struct_entry — 获取第 i 个 StructEntry 槽位 +# ============================================================ +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: + """获取第 i 个结构体条目""" + if _struct_table is None or i < 0 or i >= STRUCT_MAX: + return None + entry_size: t.CSizeT = StructEntry.__sizeof__() + addr: t.CUInt64T = t.CUInt64T(_struct_table) + i * entry_size + return (StructEntry | t.CPtr)(t.CVoid(addr, t.CPtr)) + + +# ============================================================ +# _get_field_entry — 获取结构体中第 i 个 FieldEntry 槽位 +# ============================================================ +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: + """获取结构体中第 i 个字段条目""" + if struct_entry is None or i < 0 or i >= FIELD_MAX: + return None + field_size: t.CSizeT = FieldEntry.__sizeof__() + addr: t.CUInt64T = t.CUInt64T(struct_entry.Fields) + i * field_size + return (FieldEntry | t.CPtr)(t.CVoid(addr, t.CPtr)) + + +# ============================================================ +# register_struct — 注册结构体类型 +# +# 返回 StructEntry 指针，可用于添加字段 +# ============================================================ +def register_struct(pool: memhub.MemBuddy | t.CPtr, + name: str, + struct_ty: llvmlite.LLVMType | t.CPtr, + sha1: str = None) -> StructEntry | t.CPtr: + """注册结构体类型，返回 StructEntry 指针 + + 用类型指针去重（is 比较），不用类名去重。 + 这样跨模块同名类可以共存，各自有独立的字段定义。 + sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。 + """ + if init_struct_table(pool) == 0: + return None + + # 用类型指针去重（同一类型不重复注册，支持跨模块同名类） + for i in range(_struct_count): + existing: StructEntry | t.CPtr = _get_struct_entry(i) + if existing is not None and existing.Ty is not None: + if existing.Ty is struct_ty: + return existing + + if _struct_count >= STRUCT_MAX: + stdio.printf("[STRUCT] table full, cannot register %s\n", name) + return None + + entry: StructEntry | t.CPtr = _get_struct_entry(_struct_count) + if entry is None: + return None + + # 分配字段数组 + field_size: t.CSizeT = FieldEntry.__sizeof__() + entry.Fields = pool.alloc(field_size * FIELD_MAX) + if entry.Fields is None: + return None + string.memset(entry.Fields, 0, field_size * FIELD_MAX) + + # 复制类名 + name_len: t.CSizeT = string.strlen(name) + name_buf: t.CChar | t.CPtr = pool.alloc(name_len + 1) + if name_buf is not None: + string.strcpy(name_buf, name) + entry.Name = name_buf + + entry.Ty = struct_ty + entry.FieldCount = 0 + + # 注册时直接设置 SHA1（避免 set_struct_sha1 跨模块同名时找错 entry） + if sha1 is not None: + sha1_len: t.CSizeT = string.strlen(sha1) + sha1_buf: t.CChar | t.CPtr = pool.alloc(sha1_len + 1) + if sha1_buf is not None: + string.strcpy(sha1_buf, sha1) + entry.ModuleSha1 = sha1_buf + + _struct_count += 1 + return entry + + +# ============================================================ +# set_struct_sha1 — 设置结构体所属模块的 SHA1 +# +# 在翻译类定义时调用，记录类所属模块的 SHA1。 +# 跨模块方法调用时（如 __before_init__/__init__），通过 SHA1 +# 构造正确的函数名（"{sha1}.{ClassName}.{method}"）。 +# ============================================================ +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, + class_name: str, sha1: str) -> int: + """设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败""" + if class_name is None or sha1 is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(class_name) + if entry is None: + return 0 + # 复制 SHA1 字符串（避免悬空指针） + sha1_len: t.CSizeT = string.strlen(sha1) + sha1_buf: t.CChar | t.CPtr = pool.alloc(sha1_len + 1) + if sha1_buf is None: + return 0 + string.strcpy(sha1_buf, sha1) + entry.ModuleSha1 = sha1_buf + return 1 + + +# ============================================================ +# get_struct_sha1 — 获取结构体所属模块的 SHA1 +# +# 跨模块方法调用时使用，返回 None=无 SHA1（本模块定义或无 SHA1） +# ============================================================ +def get_struct_sha1(class_name: str) -> str: + """获取结构体所属模块的 SHA1，返回 None=无 SHA1""" + if class_name is None or _struct_table is None: + return None + entry: StructEntry | t.CPtr = find_struct(class_name) + if entry is None: + return None + return entry.ModuleSha1 + + +# ============================================================ +# add_field — 向结构体添加字段信息 +# ============================================================ +def add_field(pool: memhub.MemBuddy | t.CPtr, + struct_entry: StructEntry | t.CPtr, + field_name: str, + field_ty: llvmlite.LLVMType | t.CPtr, + default_val: ast.AST | t.CPtr = None, + annot_class_name: t.CChar | t.CPtr = None) -> int: + """向结构体添加字段，返回字段索引（-1 失败） + + default_val: 字段默认值 AST 节点（可选，None=无默认值） + annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析） + """ + if struct_entry is None or field_name is None or field_ty is None: + return -1 + + if struct_entry.FieldCount >= FIELD_MAX: + stdio.printf("[STRUCT] field table full for %s\n", struct_entry.Name) + return -1 + + idx: int = struct_entry.FieldCount + fe: FieldEntry | t.CPtr = _get_field_entry(struct_entry, idx) + if fe is None: + return -1 + + # 复制字段名 + name_len: t.CSizeT = string.strlen(field_name) + name_buf: t.CChar | t.CPtr = pool.alloc(name_len + 1) + if name_buf is not None: + string.strcpy(name_buf, field_name) + fe.Name = name_buf + + fe.Index = idx + fe.Ty = field_ty + fe.DefaultVal = default_val + fe.AnnotClassName = annot_class_name + + struct_entry.FieldCount = idx + 1 + return idx + + +# ============================================================ +# find_struct — 按类名查找结构体 +# ============================================================ +def find_struct(name: str) -> StructEntry | t.CPtr: + """按类名查找结构体，返回 StructEntry 或 None""" + if name is None or _struct_table is None: + return None + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Name is not None: + if string.strcmp(entry.Name, name) == 0: + return entry + return None + + +# ============================================================ +# find_struct_by_module — 按类名 + SHA1 查找结构体 +# +# 跨模块同名类区分：Phase B 重新翻译时，必须找到当前模块 +# 注册的结构体（而非第一个同名条目），否则 struct type、 +# 字段布局、GEP 索引全部错误。 +# ============================================================ +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: + """按类名 + SHA1 查找结构体，返回 StructEntry 或 None + + SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错） + SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个 + """ + if name is None or _struct_table is None: + return None + if sha1 is not None: + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Name is not None: + if string.strcmp(entry.Name, name) == 0: + if entry.ModuleSha1 is not None: + if string.strcmp(entry.ModuleSha1, sha1) == 0: + return entry + # SHA1 查找失败：不回退到 find_struct（会返回跨模块同名错误 entry） + # 返回 None 让调用方创建新类型，避免 Phase B 用错误类型短路 + return None + # 无 SHA1：按类名查找第一个 + return find_struct(name) + + +# ============================================================ +# find_struct_by_type — 按类型指针查找结构体（is 身份比较） +# +# 用于 _translate_oop_methods 等已有 struct_ty 的场景， +# 直接用指针身份定位 entry，规避跨模块同名类 find_struct 找错的问题。 +# ============================================================ +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: + """按类型指针查找结构体（is 比较），返回 StructEntry 或 None""" + if struct_ty is None or _struct_table is None: + return None + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Ty is not None: + if entry.Ty is struct_ty: + return entry + return None + + +# ============================================================ +# lookup_field — 按结构体类型和字段名查找字段信息 +# ============================================================ +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, + field_name: str) -> FieldEntry | t.CPtr: + """按结构体类型和字段名查找字段信息 + + 两遍查找策略（跨模块同名类冲突的根本修复）: + 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry） + 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue） + """ + if struct_ty is None or field_name is None or _struct_table is None: + return None + + # ============================================================ + # 两遍查找策略（跨模块同名类冲突的根本修复）: + # + # 第一遍: 用类型指针 is 身份比较（最高优先级） + # - is 匹配唯一确定 entry，字段未找到直接返回 None + # - 规避联合类型 == BUG，且不受遍历顺序影响 + # + # 第二遍: 名称回退（is 未匹配到任何 entry 时才执行） + # - 用于未注册类型或 is 不可靠的场景 + # - 名称匹配但字段未找到时 continue（继续检查其他同名 entry） + # ============================================================ + + # 第一遍: is 身份比较 + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Ty is not None: + if entry.Ty is struct_ty: + # is 匹配，查找字段 + for fi in range(entry.FieldCount): + fe: FieldEntry | t.CPtr = _get_field_entry(entry, fi) + if fe is not None and fe.Name is not None: + if string.strcmp(fe.Name, field_name) == 0: + return fe + # is 匹配但字段未找到（类型唯一确定，无需继续） + return None + + # 第二遍: 名称回退（is 未匹配到） + ty_name: str = _extract_struct_name(struct_ty) + if ty_name is not None: + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Name is not None: + matched: int = 0 + if string.strcmp(entry.Name, ty_name) == 0: + matched = 1 + elif string.strstr(ty_name, entry.Name) is not None: + matched = 1 + if matched != 0: + # 查找字段 + for fi in range(entry.FieldCount): + fe: FieldEntry | t.CPtr = _get_field_entry(entry, fi) + if fe is not None and fe.Name is not None: + if string.strcmp(fe.Name, field_name) == 0: + return fe + # 名称匹配但字段未找到，继续检查其他同名 entry（跨模块同名） + continue + # 所有条目都未匹配 + return None + + +# ============================================================ +# lookup_field_by_class — 按类名和字段名查找字段信息 +# ============================================================ +def lookup_field_by_class(class_name: str, + field_name: str, + sha1: str = None) -> FieldEntry | t.CPtr: + """按类名和字段名查找字段信息 + + 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个 + """ + if class_name is None or field_name is None: + return None + + # 优先用 SHA1 匹配（跨模块同名类区分） + if sha1 is not None: + sha1_matched: int = 0 + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Name is not None: + if string.strcmp(entry.Name, class_name) == 0: + if entry.ModuleSha1 is not None: + if string.strcmp(entry.ModuleSha1, sha1) == 0: + sha1_matched = 1 + # SHA1 + 类名匹配，查找字段 + for fi in range(entry.FieldCount): + fe: FieldEntry | t.CPtr = _get_field_entry(entry, fi) + if fe is not None and fe.Name is not None: + if string.strcmp(fe.Name, field_name) == 0: + return fe + return None + + # 回退: 无 SHA1 或 SHA1 匹配失败，按类名查找第一个 + entry = find_struct(class_name) + if entry is None: + return None + for fi in range(entry.FieldCount): + fe: FieldEntry | t.CPtr = _get_field_entry(entry, fi) + if fe is not None and fe.Name is not None: + if string.strcmp(fe.Name, field_name) == 0: + return fe + return None + + +# ============================================================ +# get_struct_type — 按类名获取结构体的 LLVM 类型 +# ============================================================ +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: + """按类名获取结构体的 LLVM 类型""" + entry: StructEntry | t.CPtr = find_struct(class_name) + if entry is not None: + return entry.Ty + return None + + +# ============================================================ +# get_field_by_index — 按类名和字段索引获取字段信息 +# +# 用于构造函数 Point(10, 20) 按顺序访问字段 +# ============================================================ +def get_field_by_index(class_name: str, + idx: int) -> FieldEntry | t.CPtr: + """按类名和字段索引获取字段信息""" + if class_name is None: + return None + entry: StructEntry | t.CPtr = find_struct(class_name) + if entry is None: + return None + if idx < 0 or idx >= entry.FieldCount: + return None + return _get_field_entry(entry, idx) + + +# ============================================================ +# struct_has_defaults — 检查结构体是否有任何带默认值的字段 +# ============================================================ +def struct_has_defaults(class_name: str) -> int: + """检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无""" + if class_name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(class_name) + if entry is None: + return 0 + for fi in range(entry.FieldCount): + fe: FieldEntry | t.CPtr = _get_field_entry(entry, fi) + if fe is not None and fe.DefaultVal is not None: + return 1 + return 0 + + +# ============================================================ +# get_struct_type_from_value — 从 LLVM Value 的类型推断结构体类型 +# +# 如果 value 是 Ptr(Struct(...))，返回 Struct 类型 +# ============================================================ +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: + """检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Struct(fields, fcount, name): + return 1 + case _: + return 0 + + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: + """从 Value 的类型推断结构体类型""" + if val is None or val.Ty is None: + return None + match val.Ty: + case llvmlite.LLVMType.Ptr(pointee): + if _is_struct_type(pointee) != 0: + return pointee + return None + case _: + return None + + +# ============================================================ +# 联合体支持函数 +# ============================================================ + +# ============================================================ +# mark_as_union — 标记已注册的结构体为联合体 +# +# 联合体用 Struct([Array(Int8, max_size)]) 表示， +# 注册时用 register_struct 注册类型，再用此函数标记 IsUnion=1 +# ============================================================ +def mark_as_union(name: str) -> int: + """标记已注册的结构体为联合体，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.IsUnion = 1 + return 1 + + +# ============================================================ +# is_union_by_name — 按类名检查是否为联合体 +# ============================================================ +def is_union_by_name(name: str) -> int: + """按类名检查是否为联合体，返回 1=是 / 0=否""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.IsUnion + + +# ============================================================ +# is_union_by_type — 按类型指针检查是否为联合体 +# ============================================================ +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: + """按类型指针检查是否为联合体，返回 1=是 / 0=否 + + 先尝试类型指针 == 比较，失败时回退到名称匹配 + """ + if struct_ty is None or _struct_table is None: + return 0 + ty_name: str = _extract_struct_name(struct_ty) + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Ty is not None: + matched: int = 0 + if entry.Ty is struct_ty: + matched = 1 + elif ty_name is not None and entry.Name is not None: + if string.strcmp(entry.Name, ty_name) == 0: + matched = 1 + elif string.strstr(ty_name, entry.Name) is not None: + matched = 1 + if matched != 0: + return entry.IsUnion + return 0 + + +# ============================================================ +# OOP 支持函数 +# ============================================================ + +# ============================================================ +# mark_as_oop — 标记已注册的结构体为 OOP（有方法） +# ============================================================ +def mark_as_oop(name: str) -> int: + """标记已注册的结构体为 OOP，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.IsOOP = 1 + return 1 + + +# ============================================================ +# is_oop_by_name — 按类名检查是否为 OOP 结构体 +# ============================================================ +def is_oop_by_name(name: str) -> int: + """按类名检查是否为 OOP 结构体，返回 1=是 / 0=否""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.IsOOP + + +# ============================================================ +# mark_has_init — 标记结构体拥有 __init__ 方法 +# ============================================================ +def mark_has_init(name: str) -> int: + """标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.HasInit = 1 + return 1 + + +# ============================================================ +# has_init_by_name — 按类名检查是否有 __init__ 方法 +# ============================================================ +def has_init_by_name(name: str) -> int: + """按类名检查是否有 __init__ 方法，返回 1=有 / 0=无""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.HasInit + + +# ============================================================ +# mark_has_new — 标记结构体拥有 __new__ 方法 +# ============================================================ +def mark_has_new(name: str) -> int: + """标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.HasNew = 1 + return 1 + + +# ============================================================ +# has_new_by_name — 按类名检查是否有 __new__ 方法 +# ============================================================ +def has_new_by_name(name: str) -> int: + """按类名检查是否有 __new__ 方法，返回 1=有 / 0=无""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.HasNew + + +# ============================================================ +# get_class_name_by_type — 按 LLVM 类型指针查找类名 +# +# 用于方法调用：从变量类型 Ptr(Struct(...)) 反查类名， +# 以构造 ClassName.method_name 进行方法查找 +# ============================================================ +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, + struct_ty: llvmlite.LLVMType | t.CPtr) -> str: + """按类型指针查找类名，返回类名字符串或 None""" + if struct_ty is None or _struct_table is None: + return None + for i in range(_struct_count): + entry: StructEntry | t.CPtr = _get_struct_entry(i) + if entry is not None and entry.Ty is not None: + if entry.Ty is struct_ty: + return entry.Name + # 回退: 类型指针比较失败时，遍历所有注册的结构体按名称匹配 + # 提取 struct_ty 的 Name 字段（可能是 "SHA1.ClassName" 格式） + ty_name: str = _extract_struct_name(struct_ty) + if ty_name is not None: + for j in range(_struct_count): + entry2: StructEntry | t.CPtr = _get_struct_entry(j) + if entry2 is not None and entry2.Name is not None: + # 构造 "SHA1.ClassName" 格式进行比较 + entry2_sha1: str = entry2.ModuleSha1 + if entry2_sha1 is not None: + full_nm: t.CChar | t.CPtr = pool.alloc(128) + if full_nm is not None: + viperlib.snprintf(full_nm, 128, "%s.%s", entry2_sha1, entry2.Name) + if string.strcmp(full_nm, ty_name) == 0: + return entry2.Name + # 也直接比较类名（struct_ty 的 Name 可能无 SHA1 前缀） + if string.strcmp(entry2.Name, ty_name) == 0: + return entry2.Name + # 最后回退: strstr 检查包含关系 + # ty_name 可能是 "SHA1.Counter"，entry2.Name 是 "Counter" + if string.strstr(ty_name, entry2.Name) is not None: + return entry2.Name + return None + + +# ============================================================ +# _extract_struct_name — 从 LLVMType 提取 Struct 变体的 Name 字段 +# ============================================================ +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: + """从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct""" + if ty is None: + return None + match ty: + case llvmlite.LLVMType.Struct(fields, fcount, name): + return name + case _: + return None + + +# ============================================================ +# VTable 支持函数 +# ============================================================ + +# ============================================================ +# mark_has_vtable — 标记结构体拥有虚表 +# ============================================================ +def mark_has_vtable(name: str) -> int: + """标记结构体拥有虚表，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.HasVTable = 1 + return 1 + + +# ============================================================ +# has_vtable_by_name — 按类名检查是否有虚表 +# ============================================================ +def has_vtable_by_name(name: str) -> int: + """按类名检查是否有虚表，返回 1=有 / 0=无""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.HasVTable + + +# ============================================================ +# mark_novtable — 标记结构体为 @t.NoVTable +# ============================================================ +def mark_novtable(name: str) -> int: + """标记结构体为 NoVTable，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.IsNoVTable = 1 + return 1 + + +# ============================================================ +# is_novtable_by_name — 按类名检查是否为 NoVTable +# ============================================================ +def is_novtable_by_name(name: str) -> int: + """按类名检查是否为 NoVTable，返回 1=是 / 0=否""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.IsNoVTable + + +# ============================================================ +# set_parent_name — 设置父类名 +# ============================================================ +def set_parent_name(name: str, parent_name: str) -> int: + """设置结构体的父类名，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.ParentName = parent_name + return 1 + + +# ============================================================ +# get_parent_name — 获取父类名 +# ============================================================ +def get_parent_name(name: str) -> str: + """获取结构体的父类名，返回 None=无父类""" + if name is None: + return None + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return None + return entry.ParentName + + +# ============================================================ +# set_vtable_methods — 设置虚表方法名数组 +# +# methods 是一个 str 数组（每个元素是方法名字符串指针）， +# count 是方法数量。数组直接引用，不复制。 +# ============================================================ +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: + """设置虚表方法名数组，返回 1=成功 / 0=失败""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + entry.VTableMethods = methods + entry.VTableMethodCount = count + return 1 + + +# ============================================================ +# get_vtable_method_index — 获取方法在虚表中的索引 +# +# 在 VTableMethods 数组中查找方法名，返回索引（-1=未找到） +# ============================================================ +def get_vtable_method_index(name: str, method_name: str) -> int: + """获取方法在虚表中的索引，返回 -1=未找到""" + if name is None or method_name is None: + return -1 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None or entry.VTableMethods is None: + return -1 + # VTableMethods 字段声明为 t.CChar*，但实际存储的是 t.CSizeT 数组 + # 必须转换为 t.CSizeT* 才能正确读取 8 字节指针值 + methods_arr: t.CSizeT | t.CPtr = (t.CSizeT | t.CPtr)(t.CVoid(entry.VTableMethods, t.CPtr)) + for i in range(entry.VTableMethodCount): + mname_addr: t.CSizeT = methods_arr[i] + if mname_addr == 0: + continue + mname: str = (str | t.CPtr)(t.CVoid(mname_addr, t.CPtr)) + if mname is not None and string.strcmp(mname, method_name) == 0: + return i + return -1 + + +# ============================================================ +# get_vtable_method_count — 获取虚表方法数量 +# ============================================================ +def get_vtable_method_count(name: str) -> int: + """获取虚表方法数量""" + if name is None: + return 0 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return 0 + return entry.VTableMethodCount + + +# ============================================================ +# get_vtable_method_name — 获取虚表中第 idx 个方法名 +# +# 返回方法名字符串指针，None=越界或未设置 +# ============================================================ +def get_vtable_method_name(name: str, idx: int) -> str: + """获取虚表中第 idx 个方法名，返回 None=越界""" + if name is None or idx < 0: + return None + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None or entry.VTableMethods is None: + return None + if idx >= entry.VTableMethodCount: + return None + methods_arr: t.CSizeT | t.CPtr = (t.CSizeT | t.CPtr)(t.CVoid(entry.VTableMethods, t.CPtr)) + mname_addr: t.CSizeT = methods_arr[idx] + if mname_addr == 0: + return None + return (str | t.CPtr)(t.CVoid(mname_addr, t.CPtr)) + + +# ============================================================ +# get_field_count — 获取结构体字段数量（访问器，绕过 stub 类型限制） +# ============================================================ +def get_field_count(name: str) -> int: + """获取结构体字段数量，返回 -1=未找到""" + if name is None: + return -1 + entry: StructEntry | t.CPtr = find_struct(name) + if entry is None: + return -1 + return entry.FieldCount + + +# ============================================================ +# get_field_name_ptr — 获取 FieldEntry 的字段名指针（访问器） +# ============================================================ +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: + """获取 FieldEntry 的字段名，返回 None=未设置""" + if fe is None: + return None + return fe.Name + + +# ============================================================ +# get_field_type_ptr — 获取 FieldEntry 的字段类型指针（访问器） +# ============================================================ +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: + """获取 FieldEntry 的字段类型，返回 None=未设置""" + if fe is None: + return None + return fe.Ty + + +# ============================================================ +# get_field_default_ptr — 获取 FieldEntry 的默认值 AST 指针（访问器） +# ============================================================ +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: + """获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值""" + if fe is None: + return None + return fe.DefaultVal + + +# ============================================================ +# get_field_annot_class_name — 获取 FieldEntry 的注解类名（访问器） +# ============================================================ +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: + """获取 FieldEntry 的原始类型注解类名，返回 None=未设置""" + if fe is None: + return None + return fe.AnnotClassName diff --git a/TransPyV/App/lib/core/Handles/HandlesTranslator.py b/TransPyV/App/lib/core/Handles/HandlesTranslator.py new file mode 100644 index 0000000..87570c7 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesTranslator.py @@ -0,0 +1,417 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.Projectrans.Config as Config + +# ============================================================ +# HandlesTranslator - 翻译器状态管理 + 主入口 +# +# 从 translator.py 拆分出来，负责: +# 1. Translator 类 - 状态管理（Module, 变量表, 函数表, 导入） +# 2. translate() - 主翻译入口 +# 3. dump_ir() - IR 输出 +# +# 语句/表达式翻译全部委托给 Handles 模块: +# HandlesMain - wrapper main 创建 + children 遍历 +# HandlesBody - 语句分派 +# HandlesExpr - 表达式翻译 +# HandlesFunctions - 函数定义处理 +# 注意：str = bytes = t.CChar | t.CPtr = i8* +# ============================================================ + +# 常量 +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + + +@t.NoVTable +class Translator: + """翻译器：管理编译状态，委托 Handles 模块执行翻译""" + + # LLVM 上下文（共享状态） + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + + # 函数表 + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + + # 导入模块跟踪 + _imported_modules: str + _from_imports: str + + # 当前翻译上下文 + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + + # 循环控制流目标（break/continue） + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + + # 独立标签计数器（不与 builder.Counter 共享，避免 SSA 编号非单调） + _label_counter: t.CInt + + # global 声明的变量名集合（当前函数内） + _global_names: str + _global_name_count: t.CInt + + # nonlocal 声明的变量名集合（当前函数内） + _nonlocal_names: str + _nonlocal_name_count: t.CInt + + # 闭包相关：当前函数名（用于嵌套函数提升命名） + _cur_func_name: str + + # 模块 SHA1 前缀（16 字符），用于函数名混淆（SHA1 命名空间） + ModuleSha1: str + # 当前文件所属包名（如 "llvmlite"），用于解析相对导入（from . import ...） + # None 表示当前文件是顶级模块（无包），相对导入无法解析 + CurrentPackage: str + # 闭包 env 中的 nonlocal 变量偏移映射（每个 nonlocal 变量在 env 中的字节偏移） + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + + # 泛型特化上下文：当前正在特化的类型参数名/实参名列表 + # 由 _specialize_generic_class 设置，方法体翻译时用于将 T 替换为具体类型 + # None 表示不在泛型特化上下文中 + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + + # Phase 1a 声明模式标志：1=只注册 struct/enum/union 不翻译代码体，0=全量翻译 + _declare_only: t.CInt + + # 子 Handle 指针（每个 Handle 一个槽，通过 Mixin 回指针访问本结构体） + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + + # 嵌套作用域符号表 + SymTab: HandlesVar.SymbolTable | t.CPtr + + def __init__(self): + self.Module = None + self.Pool = None + self._funcs = None + self._func_count = 0 + self._cur_func = None + self._cur_builder = None + self._break_bb = None + self._continue_bb = None + self._label_counter = 0 + self._imported_modules = None + self._from_imports = None + self._global_names = None + self._global_name_count = 0 + self._nonlocal_names = None + self._nonlocal_name_count = 0 + self._cur_func_name = None + self.ModuleSha1 = None + self.CurrentPackage = None + self._closure_env_offsets = 0 + self._closure_env_count = 0 + self.GenericTypeParamNames = None + self.GenericTypeArgs = None + self._declare_only = 0 + self.AssignH = None + self.ReturnH = None + self.ImportsH = None + self.AnnAssignH = None + self.AugAssignH = None + self.IfH = None + self.WhileH = None + self.ForH = None + self.ExprH = None + self.ExprCallH = None + self.SymTab = None + + # ============================================================ + # 状态初始化 + # ============================================================ + def _init_state(self, pool: memhub.MemBuddy | t.CPtr): + """初始化变量表、函数表和子 Handle""" + self.Pool = pool + if self._funcs is None: + self._funcs = HandlesExprCall.init_func_table(pool, MAX_FUNCS) + # 嵌套作用域符号表（新版） + if self.SymTab is None: + self.SymTab = HandlesVar.init_symbol_table(pool) + # global/nonlocal 名称集合缓冲区（32 个 char* 指针 = 256 字节） + if self._global_names is None: + self._global_names = stdlib.malloc(MAX_GLOBAL_NAMES * 8) + if self._global_names is not None: + string.memset(self._global_names, 0, MAX_GLOBAL_NAMES * 8) + self._global_name_count = 0 + if self._nonlocal_names is None: + self._nonlocal_names = stdlib.malloc(MAX_GLOBAL_NAMES * 8) + if self._nonlocal_names is not None: + string.memset(self._nonlocal_names, 0, MAX_GLOBAL_NAMES * 8) + self._nonlocal_name_count = 0 + # 创建子 Handle（传入 self 作为 Mixin 回指针） + if self.AssignH is None: + self.AssignH = HandlesAssign.NewAssignHandle(pool, self) + if self.ReturnH is None: + self.ReturnH = HandlesReturn.NewReturnHandle(pool, self) + if self.ImportsH is None: + self.ImportsH = HandlesImports.NewImportsHandle(pool, self) + if self.AnnAssignH is None: + self.AnnAssignH = HandlesAnnAssign.NewAnnAssignHandle(pool, self) + if self.AugAssignH is None: + self.AugAssignH = HandlesAugAssign.NewAugAssignHandle(pool, self) + if self.IfH is None: + self.IfH = HandlesIf.NewIfHandle(pool, self) + if self.WhileH is None: + self.WhileH = HandlesWhile.NewWhileHandle(pool, self) + if self.ForH is None: + self.ForH = HandlesFor.NewForHandle(pool, self) + if self.ExprH is None: + self.ExprH = HandlesExpr.NewExprHandle(pool, self) + if self.ExprCallH is None: + self.ExprCallH = HandlesExprCall.NewExprCallHandle(pool, self) + + # ============================================================ + # 主翻译入口 + # ============================================================ + def translate(self, tree: ast.AST | t.CPtr) -> int: + """将 AST 翻译为 LLVM IR""" + if tree is None: + return 1 + if _mbuddy is None: + return 1 + + pool: memhub.MemBuddy | t.CPtr = _mbuddy + + # 初始化状态 + self._init_state(pool) + + # 创建 LLVM 模块 + mod: llvmlite.LLVMModule | t.CPtr = llvmlite.new_module(pool, "main") + if mod is None: + stdio.printf("[TR] NewModule returned NULL\n") + return 1 + self.Module = mod + + # 设置目标（优先使用 project.vpj 中的配置） + triple: str = Config.TargetTriple + if triple is None: + triple = "x86_64-pc-windows-msvc" + llvmlite.module_set_target(mod, triple) + + # 类型 + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + i8_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int8(pool) + i8_ptr_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Ptr(pool, i8_ty) + + # 声明 printf + printf_func: llvmlite.Function | t.CPtr = llvmlite.create_declare( + pool, mod, "printf", i32_ty) + if printf_func is None: + stdio.printf("[TR] CreateDeclare printf returned NULL\n") + return 1 + llvmlite.add_param(pool, printf_func, i8_ptr_ty, "fmt") + printf_func.IsVarArg = 1 + + # 声明 malloc（闭包分配用） + malloc_func: llvmlite.Function | t.CPtr = llvmlite.create_declare( + pool, mod, "malloc", i8_ptr_ty) + if malloc_func is not None: + i64_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int64(pool) + llvmlite.add_param(pool, malloc_func, i64_ty, "size") + + # 检查用户是否定义了 main 函数 + has_user_main: int = 0 + ch: list[ast.AST | t.CPtr] | t.CPtr = tree.children + if ch is not None: + cn_count: t.CSizeT = ch.__len__() + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is not None and child.kind() == ast.ASTKind.FunctionDef: + fd: ast.FunctionDef | t.CPtr = (ast.FunctionDef | t.CPtr)(child) + if fd is not None and fd.name is not None: + if string.strcmp(fd.name, "main") == 0: + has_user_main = 1 + break + + if self._declare_only != 0: + # Phase 1a-pre(2=import扫描) / Phase 1a(1=struct注册): 只处理模块级，不创建 main 函数和 builder + self._translate_module_level(pool, mod, tree) + elif has_user_main == 0: + # 无用户 main → 只翻译模块级语句，不创建 wrapper main + # 修复：避免每个文件都生成 define i32 @main() 导致链接时 main 冲突 + # 只有包含 def main() 的入口文件才会有 define i32 @main() + self._translate_module_level(pool, mod, tree) + else: + # 用户已定义 main → 委托 HandlesMain 翻译模块级 + self._translate_module_level(pool, mod, tree) + + return 0 + + # ============================================================ + # 包装 main 函数（无用户 main 时）→ 委托 HandlesMain + # ============================================================ + def _translate_wrapper_main(self, pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + tree: ast.AST | t.CPtr, + i32_ty: llvmlite.LLVMType | t.CPtr): + """委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）""" + added: int = HandlesMain.create_wrapper_main(self, tree, i32_ty) + + # ============================================================ + # 模块级翻译（用户已定义 main）→ 委托 HandlesMain + # ============================================================ + def _translate_module_level(self, pool: memhub.MemBuddy | t.CPtr, + mod: llvmlite.LLVMModule | t.CPtr, + tree: ast.AST | t.CPtr): + """委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）""" + # 全量翻译模式下，先处理导入语句，再创建前向声明，解决前向引用问题 + if self._declare_only == 0: + # 预处理导入语句，确保 _imported_modules 和 _from_imports 已填充 + # （前向声明需要解析类型注解，如 t.CArray[str] 依赖 t 模块已导入） + ch: list[ast.AST | t.CPtr] | t.CPtr = tree.children + if ch is not None: + cn_count: t.CSizeT = ch.__len__() + for ci in range(cn_count): + child: ast.AST | t.CPtr = ch.get(ci) + if child is None: + continue + kd: int = child.kind() + if kd == ast.ASTKind.Import: + self.ImportsH.HandleImport(child) + elif kd == ast.ASTKind.ImportFrom: + self.ImportsH.HandleImportFromModule(child) + self.ImportsH.HandleImportFromNames(child) + # 创建前向声明 + HandlesFunctions.forward_declare_functions(self, tree) + added: int = HandlesMain.translate_children(self, tree) + + # ============================================================ + # IR 输出 + # ============================================================ + def dump_ir(self, buf: bytes, size: t.CSizeT, mode: int): + """将 LLVM IR 输出到缓冲区 + + mode: 0=完整, 1=stub(仅声明), 2=text(仅代码) + """ + if self.Module is None or buf is None: + return + if _mbuddy is None: + return + buf[0] = '\0' + llvmlite.LLVMModulePrint(buf, size, self.Module, _mbuddy, mode) + + +# ============================================================ +# global/nonlocal 名称管理（模块级辅助函数） +# ============================================================ +def add_global_name(trans: HT.Translator | t.CPtr, name: str): + """添加一个 global 变量名""" + if trans is None or name is None: + return + if trans._global_name_count >= MAX_GLOBAL_NAMES: + return + if trans._global_names is None: + return + # 检查是否已存在 + if is_global_name(trans, name) != 0: + return + entry_addr: t.CUInt64T = t.CUInt64T(trans._global_names) + trans._global_name_count * 8 + slot_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(entry_addr) + slot_ptr[0] = name + trans._global_name_count += 1 + + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: + """检查 name 是否在当前函数的 global 集合中""" + if trans is None or name is None or trans._global_names is None: + return 0 + for i in range(trans._global_name_count): + entry_addr: t.CUInt64T = t.CUInt64T(trans._global_names) + i * 8 + slot_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(entry_addr) + entry: str = slot_ptr[0] + if entry is not None: + if string.strcmp(entry, name) == 0: + return 1 + return 0 + + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str): + """添加一个 nonlocal 变量名""" + if trans is None or name is None: + return + if trans._nonlocal_name_count >= MAX_GLOBAL_NAMES: + return + if trans._nonlocal_names is None: + return + if is_nonlocal_name(trans, name) != 0: + return + entry_addr: t.CUInt64T = t.CUInt64T(trans._nonlocal_names) + trans._nonlocal_name_count * 8 + slot_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(entry_addr) + slot_ptr[0] = name + trans._nonlocal_name_count += 1 + trans._closure_env_count += 1 + + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: + """检查 name 是否在当前函数的 nonlocal 集合中""" + if trans is None or name is None or trans._nonlocal_names is None: + return 0 + for i in range(trans._nonlocal_name_count): + entry_addr: t.CUInt64T = t.CUInt64T(trans._nonlocal_names) + i * 8 + slot_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(entry_addr) + entry: str = slot_ptr[0] + if entry is not None: + if string.strcmp(entry, name) == 0: + return 1 + return 0 + + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: + """获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1""" + if trans is None or name is None or trans._nonlocal_names is None: + return -1 + for i in range(trans._nonlocal_name_count): + entry_addr: t.CUInt64T = t.CUInt64T(trans._nonlocal_names) + i * 8 + slot_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(entry_addr) + entry: str = slot_ptr[0] + if entry is not None: + if string.strcmp(entry, name) == 0: + return i + return -1 + + +def clear_scope_names(trans: HT.Translator | t.CPtr): + """清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）""" + if trans is None: + return + trans._global_name_count = 0 + trans._nonlocal_name_count = 0 + trans._closure_env_count = 0 + + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/TransPyV/App/lib/core/Handles/HandlesType.py b/TransPyV/App/lib/core/Handles/HandlesType.py new file mode 100644 index 0000000..fd8c9a2 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesType.py @@ -0,0 +1,1185 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import stdlib +import viperlib +import w32.fileio as fileio +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesStruct as HandlesStruct + + +# ============================================================ +# HandlesType - 类型映射与解析（模块级纯函数） +# +# str = bytes = t.CChar | t.CPtr = i8* +# ============================================================ + + +# ============================================================ +# 全局当前文件名（每个文件翻译前由 BuildPipeline 设置） +# ============================================================ +_g_current_file: str = None + + +# ============================================================ +# 全局泛型特化上下文（由 _specialize_generic_class 设置/清除） +# +# 当翻译泛型类方法体时，需要将类型参数 T 替换为具体类型。 +# resolve_annotation_type 在解析 Name 节点时检查此上下文。 +# ============================================================ +_g_generic_tp_names: list[str] | t.CPtr = None +_g_generic_type_args: list[str] | t.CPtr = None + + +# ============================================================ +# 全局 CDefine 编译期常量表 +# +# 存储 NAME: t.CDefine = value 定义的常量， +# 供 t.CArray[elem_ty, NAME] 等 编译期类型解析使用。 +# 每个模块翻译开始时由 clear_cdefine_constants 清空。 +# ============================================================ +_g_cdefine_names: list[str] | t.CPtr = None +_g_cdefine_values: list[str] | t.CPtr = None + + +def register_cdefine_constant(pool: memhub.MemBuddy | t.CPtr, + name: str, value: int) -> None: + """注册 CDefine 编译期常量到全局表""" + global _g_cdefine_names, _g_cdefine_values + if _g_cdefine_names is None: + _g_cdefine_names = list[str](pool, 64) + _g_cdefine_values = list[str](pool, 64) + _g_cdefine_names.append(name) + val_buf: str = pool.alloc(32) + if val_buf is not None: + viperlib.snprintf(val_buf, 32, "%d", value) + _g_cdefine_values.append(val_buf) + else: + _g_cdefine_values.append("0") + + +def lookup_cdefine_constant(name: str) -> int: + """查找 CDefine 编译期常量，返回值或 -1（未找到）""" + if _g_cdefine_names is None: + return -1 + n: t.CSizeT = _g_cdefine_names.__len__() + i: t.CSizeT + for i in range(n): + nm: str = _g_cdefine_names.get(i) + if nm is not None and string.strcmp(nm, name) == 0: + val_str: str = _g_cdefine_values.get(i) + if val_str is not None: + return string.atoi(val_str) + return -1 + + +def clear_cdefine_constants() -> None: + """清空 CDefine 常量表（模块切换时调用）""" + global _g_cdefine_names, _g_cdefine_values + _g_cdefine_names = None + _g_cdefine_values = None + + +def set_generic_context(tp_names: list[str] | t.CPtr, + type_args: list[str] | t.CPtr): + """设置泛型特化上下文（进入泛型类方法体翻译时调用）""" + global _g_generic_tp_names, _g_generic_type_args + _g_generic_tp_names = tp_names + _g_generic_type_args = type_args + + +def clear_generic_context(): + """清除泛型特化上下文（离开泛型类方法体翻译时调用）""" + global _g_generic_tp_names, _g_generic_type_args + _g_generic_tp_names = None + _g_generic_type_args = None + + +def resolve_generic_type_name(name: str) -> str: + """如果 name 是泛型类型参数，返回对应的具体类型名；否则返回 name 本身 + + 用于表达式中的 T 替换（如 T.__sizeof__() → int.__sizeof__()） + """ + if name is None: + return name + if _g_generic_tp_names is not None and _g_generic_type_args is not None: + rg_n: t.CSizeT = _g_generic_tp_names.__len__() + rg_i: t.CSizeT + for rg_i in range(rg_n): + rg_tpn: str = _g_generic_tp_names.get(rg_i) + if rg_tpn is not None and string.strcmp(name, rg_tpn) == 0: + rg_ta: str = _g_generic_type_args.get(rg_i) + if rg_ta is not None: + return rg_ta + return name + + +# ============================================================ +# _get_type_name_from_annotation - 从类型注解节点提取类型名字符串 +# +# 用于泛型特化时构造 mangled name（如 "list[AST|t.CPtr]"） +# +# Name(id="AST") -> "AST" +# Attribute(value=Name(id="t"), attr="CPtr") -> "t.CPtr" +# BinOp(Name("AST"), BitOr, Attribute(Name("t"), "CPtr")) -> "AST|t.CPtr" +# ============================================================ +def _get_type_name_from_annotation(pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr) -> str: + """从类型注解节点提取类型名（支持 Name/Attribute/BinOp）""" + if node is None: + return None + k: int = node.kind() + # Constant: 字符串注解，直接返回字符串值 + if k == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn.const_kind == ast.CONST_STR and cn.str_val is not None: + return cn.str_val + return None + # Name: 直接返回 id + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + return nm.id + # Attribute: 构造 "module.attr" 格式 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is None: + return None + if at.value is not None and at.value.kind() == ast.ASTKind.Name: + mod_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if mod_nm.id is not None: + total: t.CSizeT = string.strlen(mod_nm.id) + 1 + string.strlen(at.attr) + 1 + buf: str = pool.alloc(total) + if buf is not None: + buf[0] = '\0' + string.strcat(buf, mod_nm.id) + string.strcat(buf, ".") + string.strcat(buf, at.attr) + return buf + return at.attr + # BinOp (BitOr): 构造 "left|right" 格式 + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if bop.op == ast.OpKind.BitOr: + left_str: str = _get_type_name_from_annotation(pool, bop.left) + right_str: str = _get_type_name_from_annotation(pool, bop.right) + if left_str is not None and right_str is not None: + total2: t.CSizeT = string.strlen(left_str) + 1 + string.strlen(right_str) + 1 + buf2: str = pool.alloc(total2) + if buf2 is not None: + buf2[0] = '\0' + string.strcat(buf2, left_str) + string.strcat(buf2, "|") + string.strcat(buf2, right_str) + return buf2 + return None + return None + + +# ============================================================ +# _parse_type_string - 将类型名字符串解析回 AST 节点 +# +# _get_type_name_from_annotation 的逆操作，用于泛型特化时 +# 从 type_args 字符串重建 AST 节点（_clone_annotation 和 +# resolve_annotation_type 的泛型上下文替换） +# +# "AST" -> Name("AST") +# "t.CPtr" -> Attribute(Name("t"), "CPtr") +# "AST|t.CPtr" -> BinOp(Name("AST"), BitOr, Attribute(Name("t"), "CPtr")) +# ============================================================ +def _parse_type_string(pool: memhub.MemBuddy | t.CPtr, + type_str: str, + ctx: t.CInt) -> ast.AST | t.CPtr: + """将类型名字符串解析为 AST 节点""" + if type_str is None or pool is None: + return None + + # 检查是否包含 | (BinOp) + pipe_pos: str = string.strchr(type_str, 124) # '|' = 124 + if pipe_pos is not None: + # 临时将 | 替换为 \0，分割左右部分 + old_char: t.CInt = pipe_pos[0] + pipe_pos[0] = 0 + left_node: ast.AST | t.CPtr = _parse_type_string(pool, type_str, ctx) + right_str: str = pipe_pos + 1 + pipe_pos[0] = old_char + right_node: ast.AST | t.CPtr = _parse_type_string(pool, right_str, ctx) + if left_node is not None and right_node is not None: + return ast.BinOp(pool, left_node, ast.OpKind.BitOr, right_node) + return None + + # 检查是否包含 . (Attribute) + dot_pos: str = string.strchr(type_str, 46) # '.' = 46 + if dot_pos is not None: + old_char2: t.CInt = dot_pos[0] + dot_pos[0] = 0 + module_node: ast.AST | t.CPtr = ast.Name(pool, type_str, ctx) + attr_str: str = dot_pos + 1 + dot_pos[0] = old_char2 + if module_node is not None: + return ast.Attribute(pool, module_node, attr_str, ctx) + return None + + # 普通 Name + return ast.Name(pool, type_str, ctx) + + +# ============================================================ +# set_current_file - 设置当前编译的文件名 +# +# 在 BuildPipeline.TranslateFileGetTrans 中每翻译一个文件前调用 +# ============================================================ +def set_current_file(file_path: str): + """设置当前编译的文件名（供报错使用）""" + global _g_current_file + _g_current_file = file_path + + +# ============================================================ +# _read_source_line - 从源文件读取第 lineno 行 +# +# 返回行内容字符串（不含换行符），失败返回 None +# ============================================================ +_SRC_BUF_SIZE: t.CDefine = 262144 + +def _read_source_line(file_path: str, lineno: int) -> str: + """从源文件读取第 lineno 行内容""" + if file_path is None or lineno <= 0: + return None + f: fileio.File | t.CPtr = fileio.File(file_path, fileio.MODE.R) + if f.closed: + return None + buf: bytes = stdlib.malloc(_SRC_BUF_SIZE) + if buf is None: + f.close() + return None + bytes_read: LONG = f.read_all(buf, _SRC_BUF_SIZE) + f.close() + if bytes_read <= 0: + stdlib.free(buf) + return None + # 逐行查找第 lineno 行 + cur_line: int = 1 + line_start: t.CSizeT = 0 + for i in range(bytes_read): + if buf[i] == '\n': + if cur_line == lineno: + line_len: t.CSizeT = i - line_start + if line_len > 0 and buf[line_start + line_len - 1] == '\r': + line_len -= 1 + line_buf: bytes = stdlib.malloc(line_len + 1) + if line_buf is not None: + string.strncpy(line_buf, buf + line_start, line_len) + line_buf[line_len] = '\0' + stdlib.free(buf) + return line_buf + stdlib.free(buf) + return None + cur_line += 1 + line_start = i + 1 + stdlib.free(buf) + return None + + +# ============================================================ +# fatal_error - 以 Python 风格报错并退出（带源代码行） +# +# 格式: +# Traceback (most recent call last): +# File "path", line N +# source code line +# SyntaxError: msg +# ============================================================ +def fatal_error(node: ast.AST | t.CPtr, msg: str): + """以 Python 风格报错并退出（带源代码行）""" + lineno: int = 0 + if node is not None: + lineno = node.lineno + file_name: str = _g_current_file + if file_name is None: + file_name = "" + stdio.printf("Traceback (most recent call last):\n") + stdio.printf(' File "%s", line %d\n', file_name, lineno) + if lineno > 0 and file_name != "": + src_line: str = _read_source_line(file_name, lineno) + if src_line is not None: + stdio.printf(" %s\n", src_line) + stdlib.free(src_line) + stdio.printf("SyntaxError: %s\n", msg) + sys.exit(1) + + +# ============================================================ +# fatal_type_error - 兼容别名（指向 fatal_error） +# ============================================================ +def fatal_type_error(node: ast.AST | t.CPtr, msg: str): + """以 Python 风格报错并退出（fatal_error 的别名）""" + fatal_error(node, msg) + +# ============================================================ +# 将 t.CXxx 类型名映射到 LLVMType（纯函数） +# ============================================================ +def map_t_type(pool: memhub.MemBuddy | t.CPtr, + type_name: str) -> llvmlite.LLVMType | t.CPtr: + """将 t.CXxx / c.CXxx 类型名映射到 LLVMType + + 注意: t.CChar | t.CPtr = i8* = str = bytes + """ + if type_name is None: + return None + + # 基本整数类型 + if string.strcmp(type_name, "CInt") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "CChar") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "CShort") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "CLong") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "CLongLong") == 0: + return llvmlite.Int64(pool) + # 精确宽度有符号类型 + elif string.strcmp(type_name, "CInt8T") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "CInt16T") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "CInt32T") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "CInt64T") == 0: + return llvmlite.Int64(pool) + # 精确宽度无符号类型 + elif string.strcmp(type_name, "CUInt8T") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "CUInt16T") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "CUInt32T") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "CUInt64T") == 0: + return llvmlite.Int64(pool) + # 浮点类型 + elif string.strcmp(type_name, "CFloat") == 0: + return llvmlite.Float(pool) + elif string.strcmp(type_name, "CDouble") == 0: + return llvmlite.Double(pool) + # 指针/void: t.CPtr = i8*, t.CChar | t.CPtr = i8* = str = bytes + elif string.strcmp(type_name, "CPtr") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "CVoid") == 0: + return llvmlite.Void(pool) + # 无符号修饰 + elif string.strcmp(type_name, "CUnsigned") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "CUnsignedInt") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "CUnsignedChar") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "CUnsignedShort") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "CUnsignedLong") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "CUnsignedLongLong") == 0: + return llvmlite.Int64(pool) + # size_t / ptr 类型 + elif string.strcmp(type_name, "CSizeT") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "CIntPtrT") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "CUIntPtrT") == 0: + return llvmlite.Int64(pool) + # bool + elif string.strcmp(type_name, "CBool") == 0: + return llvmlite.Int8(pool) + # ptrdiff_t + elif string.strcmp(type_name, "CPtrDiffT") == 0: + return llvmlite.Int64(pool) + + # ============================================================ + # stdint.py 大写别名（Win32/TransPyC 风格） + # ============================================================ + # 整数类型 + elif string.strcmp(type_name, "INT") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "BOOL") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "UINT") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "BYTE") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "WORD") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "DWORD") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "QWORD") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "TCHAR") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "VOID") == 0: + return llvmlite.Void(pool) + elif string.strcmp(type_name, "SHORT") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "USHORT") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "LONGLONG") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "ULONGLONG") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "LONG") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "ULONG") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "WCHAR") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "FLOAT") == 0: + return llvmlite.Float(pool) + elif string.strcmp(type_name, "DOUBLE") == 0: + return llvmlite.Double(pool) + elif string.strcmp(type_name, "FSIZE_t") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "LBA_t") == 0: + return llvmlite.Int32(pool) + # 定长整数 + elif string.strcmp(type_name, "INT8") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "INT16") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "INT32") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "INT64") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "UINT8") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "UINT16") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "UINT32") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "UINT64") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "CHAR8") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "CHAR16") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "CHAR32") == 0: + return llvmlite.Int32(pool) + # 小写简写 + elif string.strcmp(type_name, "i8") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "i16") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "i32") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "i64") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "u8") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "u16") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "u32") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "u64") == 0: + return llvmlite.Int64(pool) + # size_t 系列 + elif string.strcmp(type_name, "SIZE_T") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "SSIZE_T") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "PTRDIFF_T") == 0: + return llvmlite.Int64(pool) + + # stdint 指针类型（XxxPTR = XxxType | t.CPtr） + elif string.strcmp(type_name, "INTPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int32(pool)) + elif string.strcmp(type_name, "UINTPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int32(pool)) + elif string.strcmp(type_name, "BYTEPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "CHARLIST") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "SHORTPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "USHORTPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "WCHARPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "CHARPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "VOIDPTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "INT8PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "INT16PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "INT32PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int32(pool)) + elif string.strcmp(type_name, "INT64PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int64(pool)) + elif string.strcmp(type_name, "UINT8PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "UINT16PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "UINT32PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int32(pool)) + elif string.strcmp(type_name, "UINT64PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int64(pool)) + elif string.strcmp(type_name, "CHAR8PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(type_name, "CHAR16PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int16(pool)) + elif string.strcmp(type_name, "CHAR32PTR") == 0: + return llvmlite.Ptr(pool, llvmlite.Int32(pool)) + + # C 标准 小写 _t 类型 + elif string.strcmp(type_name, "int8_t") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "int16_t") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "int32_t") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "int64_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "uint8_t") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "uint16_t") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "uint32_t") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "uint64_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "size_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "ssize_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "ptrdiff_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "intptr_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "uintptr_t") == 0: + return llvmlite.Int64(pool) + elif string.strcmp(type_name, "wchar_t") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "char8_t") == 0: + return llvmlite.Int8(pool) + elif string.strcmp(type_name, "char16_t") == 0: + return llvmlite.Int16(pool) + elif string.strcmp(type_name, "char32_t") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(type_name, "_Bool") == 0: + return llvmlite.Int8(pool) + return None + + +# ============================================================ +# _is_t_attr - 检测节点是否是 t.attr_name 形式（如 t.CPtr, t.CVoid） +# +# 用于联合类型注解中检测指针包装标记 (t.CPtr) 和 void 标记 (t.CVoid) +# ============================================================ +def _is_t_attr(node: ast.AST | t.CPtr, + attr_name: str, + imported_modules: str) -> int: + """检测节点是否是 t.attr_name 形式""" + if node is None or attr_name is None: + return 0 + if node.kind() != ast.ASTKind.Attribute: + return 0 + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is None: + return 0 + if string.strcmp(at.attr, attr_name) != 0: + return 0 + if at.value is None or at.value.kind() != ast.ASTKind.Name: + return 0 + val_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if val_nm.id is None: + return 0 + if HandlesImports.is_module_imported(imported_modules, val_nm.id) == 0: + return 0 + if string.strcmp(val_nm.id, "t") == 0: + return 1 + return 0 + + +# ============================================================ +# extract_class_name_from_annotation - 从类型注解 AST 中提取类名 +# +# 用于联合类型简化为 i8* 时，回退查找结构体类名。 +# 支持: +# Name 节点 (MemBuddy) → "MemBuddy" +# Attribute 节点 (memhub.MemBuddy) → "MemBuddy" +# BinOp 联合类型 (MemBuddy | t.CPtr) → 非 CPtr 一边的类名 +# ============================================================ +def extract_class_name_from_annotation(node: ast.AST | t.CPtr, + imported_modules: str) -> str: + """从类型注解 AST 中提取类名，返回 None=无法提取""" + if node is None: + return None + k: int = node.kind() + + # Constant 节点: 字符串注解（如 'MemManager'），简单类名直接返回 + if k == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn.const_kind == ast.CONST_STR and cn.str_val is not None: + # 仅处理简单类名（不含 . 和 |），复杂形式返回 None + if string.strchr(cn.str_val, 46) is None and string.strchr(cn.str_val, 124) is None: + if string.strcmp(cn.str_val, "str") == 0: + return None + if string.strcmp(cn.str_val, "int") == 0: + return None + if string.strcmp(cn.str_val, "bytes") == 0: + return None + return cn.str_val + return None + + # Name 节点: MemBuddy → "MemBuddy" + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm.id is None: + return None + # 跳过内置类型名 + if string.strcmp(nm.id, "str") == 0: + return None + if string.strcmp(nm.id, "int") == 0: + return None + if string.strcmp(nm.id, "bytes") == 0: + return None + return nm.id + + # Attribute 节点: memhub.MemBuddy → "memhub.MemBuddy"（模块限定名） + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is None: + return None + # 跳过 t.XXX 形式（内置类型） + if at.value is not None and at.value.kind() == ast.ASTKind.Name: + val_nm2: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if val_nm2.id is not None and string.strcmp(val_nm2.id, "t") == 0: + return None + # 返回模块限定名 "module.Class"（如 "memhub.MemBuddy"） + mod_qual_buf: t.CChar | t.CPtr = stdlib.malloc( + string.strlen(val_nm2.id) + string.strlen(at.attr) + 2) + if mod_qual_buf is not None: + viperlib.snprintf(mod_qual_buf, + string.strlen(val_nm2.id) + string.strlen(at.attr) + 2, + "%s.%s", val_nm2.id, at.attr) + return mod_qual_buf + return at.attr + + # BinOp 节点: 联合类型 A | B + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if bop.op == ast.OpKind.BitOr: + left_is_cptr: int = _is_t_attr(bop.left, "CPtr", imported_modules) + right_is_cptr: int = _is_t_attr(bop.right, "CPtr", imported_modules) + # 提取非 CPtr 一边的类名 + if left_is_cptr != 0 and right_is_cptr == 0: + return extract_class_name_from_annotation(bop.right, imported_modules) + if right_is_cptr != 0 and left_is_cptr == 0: + return extract_class_name_from_annotation(bop.left, imported_modules) + # 无 CPtr 标记：尝试两边 + if left_is_cptr == 0 and right_is_cptr == 0: + left_name: str = extract_class_name_from_annotation(bop.left, imported_modules) + if left_name is not None: + return left_name + return extract_class_name_from_annotation(bop.right, imported_modules) + return None + + # Subscript 节点: list[AST | t.CPtr] → "list" + if k == ast.ASTKind.Subscript: + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + if sub.value is not None: + return extract_class_name_from_annotation(sub.value, imported_modules) + return None + + return None + + +# ============================================================ +# 解析类型注解节点为 LLVMType +# +# 支持的注解形式: +# - int → i32 +# - t.CUInt64T → i64 +# - str → i8* (t.CChar | t.CPtr = i8* = bytes) +# - int → i32 +# - bytes → i8* +# ============================================================ +def resolve_annotation_type(pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr, + imported_modules: str, + from_imports: str) -> llvmlite.LLVMType | t.CPtr: + """解析类型注解节点为 LLVMType + + Args: + pool: 内存分配器 + node: 类型注解 AST 节点（Name 或 Attribute） + imported_modules: 已导入模块名字符串（空格分隔） + from_imports: from-import 映射字符串（"name:module name:module" 格式） + """ + if node is None: + return None + + k: int = node.kind() + + # Constant 节点: 字符串注解（如 'MemManager'），当作无引号的 Name/Attribute/BinOp 处理 + if k == ast.ASTKind.Constant: + cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(node) + if cn.const_kind == ast.CONST_STR and cn.str_val is not None: + sub_node: ast.AST | t.CPtr = _parse_type_string(pool, cn.str_val, ast.Ctx.Load) + if sub_node is not None: + return resolve_annotation_type(pool, sub_node, imported_modules, from_imports) + return None + + # Name 节点: str, int, bytes, from-import 名称, class 名称等 + if k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm.id is None: + return None + # 泛型类型参数 T 替换：在泛型特化上下文中，将 T 替换为具体类型 + if _g_generic_tp_names is not None and _g_generic_type_args is not None: + gp_n: t.CSizeT = _g_generic_tp_names.__len__() + gp_i: t.CSizeT + for gp_i in range(gp_n): + gp_tpn: str = _g_generic_tp_names.get(gp_i) + if gp_tpn is not None and string.strcmp(nm.id, gp_tpn) == 0: + gp_ta: str = _g_generic_type_args.get(gp_i) + if gp_ta is not None: + # 用具体类型名递归解析（支持 BinOp/Attribute 类型实参） + gp_sub_node: ast.AST | t.CPtr = _parse_type_string(pool, gp_ta, nm.ctx) + if gp_sub_node is not None: + return resolve_annotation_type(pool, gp_sub_node, imported_modules, from_imports) + # str = bytes = t.CChar | t.CPtr = i8* + if string.strcmp(nm.id, "str") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + elif string.strcmp(nm.id, "int") == 0: + return llvmlite.Int32(pool) + elif string.strcmp(nm.id, "bytes") == 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + # class 名称: 查找已注册的 StructType，返回 StructType 本身（不包装 Ptr） + # 这样 build_alloca 会分配结构体大小的空间，alloca 返回 struct_ty* + # 命名空间隔离：严格模式下仅解析可见的结构体 + if HandlesStruct.is_struct_visible(nm.id) != 0: + struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type(nm.id) + if struct_ty is not None: + return struct_ty + # from-import 名称: 查找 from_imports 获取模块名后映射 + mod_name: str = HandlesImports.lookup_from_import(from_imports, nm.id) + if mod_name is not None: + return map_t_type(pool, nm.id) + return None + + # Attribute 节点: t.CInt, t.CUInt64T, namespace_defs.PlainStruct 等 + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is None: + return None + # 检查 value 是否是 Name('t') 或 Name('c') + if at.value is not None and at.value.kind() == ast.ASTKind.Name: + val_name: ast.Name | t.CPtr = (ast.Name | t.CPtr)(at.value) + if val_name.id is not None: + # 仅当模块已导入时才解析 + mod_imp: int = HandlesImports.is_module_imported(imported_modules, val_name.id) + if mod_imp == 0: + return None + # 先尝试内置类型映射（int 等） + mapped_ty: llvmlite.LLVMType | t.CPtr = map_t_type(pool, at.attr) + if mapped_ty is not None: + return mapped_ty + # 模块限定类类型：Module.ClassName（命名空间隔离） + mod_struct_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type(at.attr) + if mod_struct_ty is not None: + return mod_struct_ty + return None + return None + + # BinOp 节点: 联合类型注解 A | B + # 处理三种语义: + # 1. ActualType | t.CPtr → Ptr(ActualType)（指针包装，如 MemBuddy | t.CPtr → MemBuddy*） + # 2. t.CVoid | t.CPtr → i8* (void* 等价于 i8*) + # 3. ActualType | DecoratorMarker → ActualType（忽略装饰器标记如 CExport/CExtern） + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if bop.op == ast.OpKind.BitOr: + # 检测 t.CPtr 标记（指针包装标记） + left_is_cptr: int = _is_t_attr(bop.left, "CPtr", imported_modules) + right_is_cptr: int = _is_t_attr(bop.right, "CPtr", imported_modules) + + if left_is_cptr != 0 or right_is_cptr != 0: + # 有 CPtr 标记：解析非 CPtr 一边的实际类型，包装为 Ptr + # left 是 CPtr → other 是 right；right 是 CPtr → other 是 left + other_node: ast.AST | t.CPtr = bop.left + if left_is_cptr != 0: + other_node = bop.right + # 检测另一边是否是 t.CVoid (void* = i8*) + other_is_cvoid: int = _is_t_attr(other_node, "CVoid", imported_modules) + if other_is_cvoid != 0: + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + other_ty: llvmlite.LLVMType | t.CPtr = resolve_annotation_type( + pool, other_node, imported_modules, from_imports) + if other_ty is None: + # CPtr 单独使用或其他无效类型 → i8* + return llvmlite.Ptr(pool, llvmlite.Int8(pool)) + return llvmlite.Ptr(pool, other_ty) + + # 无 CPtr 标记：返回第一个非 None（装饰器标记如 CExport/CExtern 被忽略） + left_ty: llvmlite.LLVMType | t.CPtr = resolve_annotation_type( + pool, bop.left, imported_modules, from_imports) + if left_ty is not None: + return left_ty + right_ty: llvmlite.LLVMType | t.CPtr = resolve_annotation_type( + pool, bop.right, imported_modules, from_imports) + if right_ty is not None: + return right_ty + return None + + # Subscript 节点: t.CArray[elem_ty, count] → [count x elem_ty] + # t.CArray[elem_ty] → Ptr(elem_ty)（静态推导长度数组） + # list[int] → 特化类 struct type（泛型类类型注解） + # value 是 Attribute(Name('t'), 'CArray') 或 Name('list')（泛型类） + # slice 是 Tuple(elts=[elem_ty_node, count_node]) 或单 elem_ty_node + if k == ast.ASTKind.Subscript: + sub: ast.Subscript | t.CPtr = (ast.Subscript | t.CPtr)(node) + if sub.value is None or sub.slice is None: + fatal_type_error(node, "Subscript 节点缺少 value 或 slice") + # 泛型类类型注解: list[int] → 查找已特化的 struct + # value 是 Name（泛型类名），slice 是类型实参 + if sub.value.kind() == ast.ASTKind.Name: + gen_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(sub.value) + if gen_nm.id is not None: + # 先收集类型实参名，再分配缓冲区拼接 mangled name + gen_arg_names: list[str] | t.CPtr = list[str](pool, 4) + if sub.slice.kind() == ast.ASTKind.Tuple: + gen_tp: ast.Tuple | t.CPtr = (ast.Tuple | t.CPtr)(sub.slice) + gen_elts: list[ast.AST | t.CPtr] | t.CPtr = gen_tp.elts + if gen_elts is not None: + gen_en: t.CSizeT = gen_elts.__len__() + gen_ei: t.CSizeT + for gen_ei in range(gen_en): + gen_elt: ast.AST | t.CPtr = gen_elts.get(gen_ei) + gen_tn: str = _get_type_name_from_annotation(pool, gen_elt) + gen_arg_names.append(gen_tn) + else: + gen_tn_single: str = _get_type_name_from_annotation(pool, sub.slice) + gen_arg_names.append(gen_tn_single) + # 计算总长度并分配缓冲区 + gen_total: t.CSizeT = string.strlen(gen_nm.id) + 2 # "[" 和 "]" + gen_an: t.CSizeT = gen_arg_names.__len__() + gen_ai: t.CSizeT + for gen_ai in range(gen_an): + if gen_ai > 0: + gen_total += 1 # "," + gen_atn: str = gen_arg_names.get(gen_ai) + if gen_atn is not None: + gen_total += string.strlen(gen_atn) + gen_total += 1 # NUL + gen_mangled: str = pool.alloc(gen_total) + if gen_mangled is None: + return None + gen_mangled[0] = '\0' + string.strcat(gen_mangled, gen_nm.id) + string.strcat(gen_mangled, "[") + for gen_ai in range(gen_an): + if gen_ai > 0: + string.strcat(gen_mangled, ",") + gen_atn: str = gen_arg_names.get(gen_ai) + if gen_atn is not None: + string.strcat(gen_mangled, gen_atn) + string.strcat(gen_mangled, "]") + # 查找已特化的 struct（特化由实例化调用 list[int](pool) 触发） + gen_spec_ty: llvmlite.LLVMType | t.CPtr = HandlesStruct.get_struct_type(gen_mangled) + if gen_spec_ty is not None: + return gen_spec_ty + # 未特化：返回 None（不报错，让后续流程处理） + return None + # 检查 value 是否是 Attribute(Name('t'), 'CArray') + if sub.value.kind() != ast.ASTKind.Attribute: + fatal_type_error(node, "不支持的 Subscript 类型注解（仅支持 t.CArray）") + arr_at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(sub.value) + if arr_at.attr is None: + fatal_type_error(node, "Subscript Attribute 缺少 attr") + if string.strcmp(arr_at.attr, "CArray") != 0: + fatal_type_error(node, "不支持的 Subscript 类型注解（仅支持 t.CArray）") + if arr_at.value is None or arr_at.value.kind() != ast.ASTKind.Name: + fatal_type_error(node, "t.CArray 必须通过模块名限定") + arr_val_name: ast.Name | t.CPtr = (ast.Name | t.CPtr)(arr_at.value) + if arr_val_name.id is None: + fatal_type_error(node, "t.CArray 模块名为 None") + if HandlesImports.is_module_imported(imported_modules, arr_val_name.id) == 0: + fatal_type_error(node, "t.CArray 模块未导入") + # 单参数形式: t.CArray[elem_ty] → Ptr(elem_ty) + # 相当于 t.CArray[elem_ty, None]，表示静态推导长度数组 + if sub.slice.kind() != ast.ASTKind.Tuple: + elem_ty_single: llvmlite.LLVMType | t.CPtr = resolve_annotation_type( + pool, sub.slice, imported_modules, from_imports) + if elem_ty_single is None: + fatal_type_error(sub.slice, "t.CArray[?] 元素类型解析失败") + return llvmlite.Ptr(pool, elem_ty_single) + # 双参数形式: t.CArray[elem_ty, count] → [count x elem_ty] + tup: ast.Tuple | t.CPtr = (ast.Tuple | t.CPtr)(sub.slice) + if tup.elts is None: + fatal_type_error(node, "t.CArray Tuple 缺少 elts") + elts_list: list[ast.AST | t.CPtr] | t.CPtr = tup.elts + elts_n: t.CSizeT = elts_list.__len__() + if elts_n < 2: + fatal_type_error(node, "t.CArray Tuple 元素不足") + elem_node: ast.AST | t.CPtr = elts_list.get(0) + count_node: ast.AST | t.CPtr = elts_list.get(1) + if elem_node is None or count_node is None: + fatal_type_error(node, "t.CArray Tuple 元素为 None") + # 递归解析元素类型 + elem_ty: llvmlite.LLVMType | t.CPtr = resolve_annotation_type( + pool, elem_node, imported_modules, from_imports) + if elem_ty is None: + fatal_type_error(elem_node, "t.CArray 元素类型解析失败") + # 解析 count（必须是整数常量、None 或 CDefine 常量名） + # t.CArray[elem_ty, None] 等价于单参数形式 → Ptr(elem_ty) + # t.CArray[elem_ty, NAME] 其中 NAME: t.CDefine = value → [value x elem_ty] + count_val: int = 0 + if count_node.kind() == ast.ASTKind.Constant: + cnt_cn: ast.Constant | t.CPtr = (ast.Constant | t.CPtr)(count_node) + if cnt_cn.const_kind == ast.CONST_NONE: + return llvmlite.Ptr(pool, elem_ty) + if cnt_cn.const_kind != ast.CONST_INT: + fatal_type_error(count_node, "t.CArray count 必须是整数常量、None 或 CDefine 常量名") + count_val = cnt_cn.int_val + elif count_node.kind() == ast.ASTKind.Name: + # CDefine 常量名查找 + cnt_nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(count_node) + if cnt_nm.id is None: + fatal_type_error(count_node, "t.CArray count Name 节点 id 为 None") + looked_up: int = lookup_cdefine_constant(cnt_nm.id) + if looked_up < 0: + fatal_type_error(count_node, "t.CArray count 不是已注册的 CDefine 常量") + count_val = looked_up + else: + fatal_type_error(count_node, "t.CArray count 必须是整数常量、None 或 CDefine 常量名") + if count_val <= 0: + fatal_type_error(count_node, "t.CArray count 必须为正数") + # 创建 ArrayType（不包装 Ptr） + # 这样 build_alloca 会分配数组大小的空间，alloca 返回 [N x elem_ty]* + return llvmlite.Array(pool, elem_ty, count_val) + + # 未知 AST kind：fail-fast，不宽容返回 None + fatal_type_error(node, "不支持的类型注解语法") + + +# ============================================================ +# has_decorator_marker - 检测类型注解中是否包含指定的装饰器标记 +# +# 支持联合类型注解：int | t.CExtern | t.CExport +# 递归检测 BinOp 节点的左右子节点。 +# +# marker_name 取值: "CExtern", "CExport", "CInline", "State" +# ============================================================ +def has_decorator_marker(node: ast.AST | t.CPtr, + marker_name: str) -> int: + """检测类型注解中是否包含指定的装饰器标记""" + if node is None or marker_name is None: + return 0 + + k: int = node.kind() + + # Attribute 节点: t.CExtern, t.State, t.CExport, t.CInline + if k == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(node) + if at.attr is not None: + if string.strcmp(at.attr, marker_name) == 0: + return 1 + return 0 + + # BinOp 节点: int | t.CExtern（递归检测左右子节点） + if k == ast.ASTKind.BinOp: + bop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + if has_decorator_marker(bop.left, marker_name) != 0: + return 1 + if has_decorator_marker(bop.right, marker_name) != 0: + return 1 + return 0 + + return 0 + + +# ============================================================ +# is_pass_body - 检测函数体是否为 pass（空体） +# +# pass 体函数用于声明外部函数（t.CExtern / t.State）， +# 翻译器应生成 declare 而非 define。 +# ============================================================ +def is_pass_body(body: list[ast.AST | t.CPtr] | t.CPtr) -> int: + """检测函数体是否为 pass（空体或仅含 pass 语句）""" + if body is None: + return 1 + bn: t.CSizeT = body.__len__() + if bn == 0: + return 1 + if bn == 1: + stmt: ast.AST | t.CPtr = body.get(0) + if stmt is not None and stmt.kind() == ast.ASTKind.Pass: + return 1 + return 0 + + +# ============================================================ +# 获取 LLVM 整数类型的位宽 +# ============================================================ +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: + """获取 LLVM 整数类型的位宽，非整数类型返回 0""" + if ty is None: + return 0 + match ty: + case llvmlite.LLVMType.Int(bits): + return bits + case _: + return 0 + + +# ============================================================ +# 推断表达式的 LLVM 类型 +# ============================================================ +def infer_expr_type(pool: memhub.MemBuddy | t.CPtr, + node: ast.AST | t.CPtr, + param_types: str) -> llvmlite.LLVMType | t.CPtr: + """推断表达式的 LLVM 类型 + + param_types 格式: "name:bits,name:bits,..." + """ + if node is None: + return llvmlite.Int32(pool) + + k: int = node.kind() + + if k == ast.ASTKind.Constant: + return llvmlite.Int32(pool) + + elif k == ast.ASTKind.Name: + nm: ast.Name | t.CPtr = (ast.Name | t.CPtr)(node) + if nm.id is not None and param_types is not None: + # 在 param_types 中查找: "name:bits" 格式 + search_key: t.CChar | t.CPtr = pool.alloc(64) + if search_key is not None: + viperlib.snprintf(search_key, 64, "%s:", nm.id) + found: t.CChar | t.CPtr = string.strstr(param_types, search_key) + if found is not None: + colon_pos: t.CSizeT = string.strlen(nm.id) + 1 + bits_str: t.CChar | t.CPtr = found + colon_pos + bits: int = 0 + for bi in range(4): + ch: t.CChar = bits_str[bi] + if ch >= '0' and ch <= '9': + bits = bits * 10 + (ch - '0') + else: + break + if bits == 64: + return llvmlite.Int64(pool) + elif bits == 16: + return llvmlite.Int16(pool) + elif bits == 8: + return llvmlite.Int8(pool) + return llvmlite.Int32(pool) + + elif k == ast.ASTKind.BinOp: + binop: ast.BinOp | t.CPtr = (ast.BinOp | t.CPtr)(node) + lhs_ty: llvmlite.LLVMType | t.CPtr = infer_expr_type( + pool, binop.left, param_types) + rhs_ty: llvmlite.LLVMType | t.CPtr = infer_expr_type( + pool, binop.right, param_types) + lhs_bits: int = get_llvm_type_bits(lhs_ty) + rhs_bits: int = get_llvm_type_bits(rhs_ty) + if lhs_bits >= rhs_bits: + return lhs_ty + return rhs_ty + + return llvmlite.Int32(pool) + + +# ============================================================ +# 推断函数返回类型 +# ============================================================ +def infer_return_type(pool: memhub.MemBuddy | t.CPtr, + body: list[ast.AST | t.CPtr] | t.CPtr, + param_types: str) -> llvmlite.LLVMType | t.CPtr: + """扫描函数体中的 return 语句，推断返回类型""" + if body is None: + return llvmlite.Int32(pool) + + ret_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + bn: t.CSizeT = body.__len__() + for bi in range(bn): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None and stmt.kind() == ast.ASTKind.Return: + rt: ast.Return | t.CPtr = (ast.Return | t.CPtr)(stmt) + if rt.value is not None: + expr_ty: llvmlite.LLVMType | t.CPtr = infer_expr_type( + pool, rt.value, param_types) + if expr_ty is not None: + cur_bits: int = get_llvm_type_bits(ret_ty) + new_bits: int = get_llvm_type_bits(expr_ty) + if new_bits > cur_bits: + ret_ty = expr_ty + return ret_ty + + +# ============================================================ +# 构建参数类型编码字符串 +# ============================================================ +def build_param_types_str(pool: memhub.MemBuddy | t.CPtr, + args_node: ast.AST | t.CPtr) -> str: + """构建参数类型编码字符串 "name:bits,name:bits,..." """ + if args_node is None: + return None + ags: ast.Arguments | t.CPtr = (ast.Arguments | t.CPtr)(args_node) + if ags.args is None: + return None + alist: list[ast.AST | t.CPtr] | t.CPtr = ags.args + an: t.CSizeT = alist.__len__() + if an == 0: + return None + + buf_size: t.CSizeT = an * 32 + 1 + buf: t.CChar | t.CPtr = pool.alloc(buf_size) + if buf is None: + return None + buf[0] = '\0' + + i32_ty: llvmlite.LLVMType | t.CPtr = llvmlite.Int32(pool) + for ai in range(an): + arg: ast.Arg | t.CPtr = (ast.Arg | t.CPtr)(alist.get(ai)) + if arg is not None and arg.arg is not None: + param_ty: llvmlite.LLVMType | t.CPtr = i32_ty + if arg.annotation is not None: + # 直接调用 map_t_type 解析 Attribute 注解 + if arg.annotation.kind() == ast.ASTKind.Attribute: + at: ast.Attribute | t.CPtr = (ast.Attribute | t.CPtr)(arg.annotation) + if at.attr is not None: + resolved: llvmlite.LLVMType | t.CPtr = map_t_type(pool, at.attr) + if resolved is not None: + param_ty = resolved + bits: int = get_llvm_type_bits(param_ty) + if bits == 0: + bits = 32 + tmp: t.CChar | t.CPtr = pool.alloc(32) + if tmp is not None: + viperlib.snprintf(tmp, 32, "%s:%d", arg.arg, bits) + if ai > 0: + _strappend(buf, buf_size, ",") + _strappend(buf, buf_size, tmp) + return buf + + +# ============================================================ +# 辅助函数 +# ============================================================ +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 src 追加到 dst 末尾""" + if dst is None or src is None: + return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: + return + for i in range(min(slen, remain - 1)): + dst[dlen + i] = src[i] + dst[dlen + i] = '\0' \ No newline at end of file diff --git a/TransPyV/App/lib/core/Handles/HandlesVar.py b/TransPyV/App/lib/core/Handles/HandlesVar.py new file mode 100644 index 0000000..455f5fe --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesVar.py @@ -0,0 +1,365 @@ +import t, c +from stdint import * +import llvmlite +import memhub +import string + + +# ============================================================ +# HandlesVar - 嵌套作用域符号表 + 变量管理 +# +# 使用 Scope 链（Parent 指针）管理嵌套作用域， +# 每个作用域包含一个 VarEntry 数组（扁平化，已验证可工作）。 +# +# 作用域类型: +# SCOPE_MODULE — 模块级（根作用域） +# SCOPE_FUNCTION — 函数级 +# SCOPE_BLOCK — 块级（if/for/while，当前未使用块作用域） +# ============================================================ + +# 作用域类型常量 +SCOPE_MODULE: t.CDefine = 0 +SCOPE_FUNCTION: t.CDefine = 1 +SCOPE_BLOCK: t.CDefine = 2 + +# 每个作用域最大变量数 +MAX_VARS: t.CDefine = 256 + + +# ============================================================ +# VarEntry - 变量表条目 +# ============================================================ +@t.NoVTable +class VarEntry: + """变量表条目""" + Name: t.CChar | t.CPtr + Alloca: llvmlite.Value | t.CPtr + Used: t.CInt + AnnotClassName: t.CChar | t.CPtr # 原始类型注解的类名（str 别名在结构体字段中触发编译器 bug，改用显式联合类型） + + +# ============================================================ +# Scope - 作用域节点 +# +# Parent 指向父作用域（None 表示根）， +# Vars 是 VarEntry 数组，VarCount 是当前变量数。 +# ============================================================ +@t.NoVTable +class Scope: + """作用域节点""" + Parent: Scope | t.CPtr + Vars: VarEntry | t.CPtr + VarCount: t.CInt + Kind: t.CInt + + +# ============================================================ +# SymbolTable - 嵌套作用域符号表 +# +# Root 是模块级根作用域，Current 是当前作用域。 +# enter_scope/exit_scope 管理作用域栈。 +# ============================================================ +@t.NoVTable +class SymbolTable: + """嵌套作用域符号表""" + Pool: memhub.MemBuddy | t.CPtr + Root: Scope | t.CPtr + Current: Scope | t.CPtr + + +# ============================================================ +# init_vars — 分配并清零 VarEntry 数组 +# ============================================================ +def init_vars(pool: memhub.MemBuddy | t.CPtr) -> VarEntry | t.CPtr: + """分配并清零变量表数组""" + size: t.CSizeT = MAX_VARS * VarEntry.__sizeof__() + vars_ptr: VarEntry | t.CPtr = pool.alloc(size) + if vars_ptr is not None: + string.memset(vars_ptr, 0, size) + return vars_ptr + + +# ============================================================ +# find_var — 在变量表中按名称查找 +# ============================================================ +def find_var(vars_ptr: VarEntry | t.CPtr, + var_count: int, + name: str) -> llvmlite.Value | t.CPtr: + """在变量表中按名称查找""" + if name is None or vars_ptr is None: + return None + entry_size: t.CSizeT = VarEntry.__sizeof__() + for i in range(var_count): + entry_addr: t.CUInt64T = t.CUInt64T(vars_ptr) + i * entry_size + entry: VarEntry | t.CPtr = (VarEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry.Name is not None and entry.Used != 0: + if string.strcmp(entry.Name, name) == 0: + return entry.Alloca + return None + + +# ============================================================ +# find_var_entry — 在变量表中按名称查找，返回 VarEntry（含 AnnotTy） +# ============================================================ +def find_var_entry(vars_ptr: VarEntry | t.CPtr, + var_count: int, + name: str) -> VarEntry | t.CPtr: + """在变量表中按名称查找，返回 VarEntry 或 None""" + if name is None or vars_ptr is None: + return None + entry_size: t.CSizeT = VarEntry.__sizeof__() + for i in range(var_count): + entry_addr: t.CUInt64T = t.CUInt64T(vars_ptr) + i * entry_size + entry: VarEntry | t.CPtr = (VarEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry.Name is not None and entry.Used != 0: + if string.strcmp(entry.Name, name) == 0: + return entry + return None + + +# ============================================================ +# lookup_var_entry — 从当前作用域逐级向上查找变量，返回 VarEntry +# ============================================================ +def lookup_var_entry(symtab: SymbolTable | t.CPtr, + name: str) -> VarEntry | t.CPtr: + """从当前作用域逐级向上查找变量，返回 VarEntry 或 None""" + if symtab is None or name is None: + return None + scope: Scope | t.CPtr = symtab.Current + while scope is not None: + result: VarEntry | t.CPtr = find_var_entry(scope.Vars, scope.VarCount, name) + if result is not None: + return result + scope = scope.Parent + return None + + +# ============================================================ +# set_var_annot_class_name — 设置变量的原始类型注解类名 +# +# 在函数参数定义后调用，存储原始类型注解的类名。 +# 方法调用检测时，当 alloca 类型是 Ptr(i8)（联合类型简化）， +# 通过 AnnotClassName 查找实际结构体类型。 +# ============================================================ +def set_var_annot_class_name(symtab: SymbolTable | t.CPtr, + name: str, + class_name: str) -> int: + """设置变量的原始类型注解类名，返回 0=成功 / 1=失败""" + if symtab is None or name is None: + return 1 + entry: VarEntry | t.CPtr = lookup_var_entry(symtab, name) + if entry is None: + return 1 + entry.AnnotClassName = class_name + return 0 + + +# ============================================================ +# add_var — 添加变量到变量表 +# ============================================================ +def add_var(vars_ptr: VarEntry | t.CPtr, + var_count: int, + name: str, + alloca: llvmlite.Value | t.CPtr) -> int: + """添加变量到变量表""" + if name is None or alloca is None or vars_ptr is None: + return 1 + if var_count >= MAX_VARS: + return 1 + entry_size: t.CSizeT = VarEntry.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(vars_ptr) + var_count * entry_size + entry: VarEntry | t.CPtr = (VarEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + entry.Name = name + entry.Alloca = alloca + entry.Used = 1 + return 0 + + +# ============================================================ +# get_or_create_var — 查找或创建变量 alloca（旧版兼容） +# ============================================================ +def get_or_create_var(pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + vars_ptr: VarEntry | t.CPtr, + var_count: int, + name: str, + ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """查找或创建变量 alloca""" + existing: llvmlite.Value | t.CPtr = find_var(vars_ptr, var_count, name) + if existing is not None: + return existing + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, ty) + if alloca is None: + return None + if add_var(vars_ptr, var_count, name, alloca) != 0: + return None + return alloca + + +# ============================================================ +# _create_scope — 创建新作用域节点（内部辅助函数） +# ============================================================ +def _create_scope(pool: memhub.MemBuddy | t.CPtr, + parent: Scope | t.CPtr, + kind: int) -> Scope | t.CPtr: + """创建并初始化作用域节点""" + scope: Scope | t.CPtr = pool.alloc(Scope.__sizeof__()) + if scope is None: + return None + string.memset(scope, 0, Scope.__sizeof__()) + scope.Parent = parent + scope.Vars = init_vars(pool) + scope.VarCount = 0 + scope.Kind = kind + if scope.Vars is None: + return None + return scope + + +# ============================================================ +# init_symbol_table — 创建符号表（含模块级根作用域） +# ============================================================ +def init_symbol_table(pool: memhub.MemBuddy | t.CPtr) -> SymbolTable | t.CPtr: + """创建并初始化符号表，包含模块级根作用域""" + if pool is None: + return None + symtab: SymbolTable | t.CPtr = pool.alloc(SymbolTable.__sizeof__()) + if symtab is None: + return None + string.memset(symtab, 0, SymbolTable.__sizeof__()) + symtab.Pool = pool + + # 创建根作用域（模块级） + root: Scope | t.CPtr = _create_scope(pool, None, SCOPE_MODULE) + if root is None: + return None + + symtab.Root = root + symtab.Current = root + return symtab + + +# ============================================================ +# enter_scope — 进入新作用域 +# ============================================================ +def enter_scope(symtab: SymbolTable | t.CPtr, + kind: int) -> Scope | t.CPtr: + """进入新作用域，返回新创建的作用域""" + if symtab is None: + return None + pool: memhub.MemBuddy | t.CPtr = symtab.Pool + scope: Scope | t.CPtr = _create_scope(pool, symtab.Current, kind) + if scope is None: + return None + symtab.Current = scope + return scope + + +# ============================================================ +# exit_scope — 退出当前作用域 +# ============================================================ +def exit_scope(symtab: SymbolTable | t.CPtr): + """退出当前作用域，恢复到父作用域""" + if symtab is not None and symtab.Current is not None: + symtab.Current = symtab.Current.Parent + + +# ============================================================ +# define_var — 在当前作用域定义变量 +# ============================================================ +def define_var(symtab: SymbolTable | t.CPtr, + name: str, + alloca: llvmlite.Value | t.CPtr) -> int: + """在当前作用域定义变量，返回 0 成功""" + if symtab is None or name is None or alloca is None: + return 1 + scope: Scope | t.CPtr = symtab.Current + if scope is None: + return 1 + ret: int = add_var(scope.Vars, scope.VarCount, name, alloca) + if ret == 0: + scope.VarCount = scope.VarCount + 1 + return ret + + +# ============================================================ +# define_module_var — 在模块作用域定义变量 +# ============================================================ +def define_module_var(symtab: SymbolTable | t.CPtr, + name: str, + alloca: llvmlite.Value | t.CPtr) -> int: + """在模块作用域定义变量，返回 0 成功""" + if symtab is None or name is None or alloca is None: + return 1 + scope: Scope | t.CPtr = symtab.Root + if scope is None: + return 1 + ret: int = add_var(scope.Vars, scope.VarCount, name, alloca) + if ret == 0: + scope.VarCount = scope.VarCount + 1 + return ret + + +# ============================================================ +# lookup_var — 从当前作用域逐级向上查找变量 +# ============================================================ +def lookup_var(symtab: SymbolTable | t.CPtr, + name: str) -> llvmlite.Value | t.CPtr: + """从当前作用域逐级向上查找变量，返回 alloca 或 None""" + if symtab is None or name is None: + return None + scope: Scope | t.CPtr = symtab.Current + while scope is not None: + result: llvmlite.Value | t.CPtr = find_var( + scope.Vars, scope.VarCount, name) + if result is not None: + return result + scope = scope.Parent + return None + + +# ============================================================ +# lookup_current — 仅在当前作用域查找变量 +# ============================================================ +def lookup_current(symtab: SymbolTable | t.CPtr, + name: str) -> llvmlite.Value | t.CPtr: + """仅在当前作用域查找变量""" + if symtab is None or name is None: + return None + scope: Scope | t.CPtr = symtab.Current + if scope is None: + return None + return find_var(scope.Vars, scope.VarCount, name) + + +# ============================================================ +# lookup_module_var — 在模块作用域查找变量 +# ============================================================ +def lookup_module_var(symtab: SymbolTable | t.CPtr, + name: str) -> llvmlite.Value | t.CPtr: + """在模块作用域查找变量""" + if symtab is None or name is None: + return None + scope: Scope | t.CPtr = symtab.Root + if scope is None: + return None + return find_var(scope.Vars, scope.VarCount, name) + + +# ============================================================ +# get_or_create_sym — 查找或创建变量（在当前作用域） +# ============================================================ +def get_or_create_sym(symtab: SymbolTable | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + builder: llvmlite.IRBuilder | t.CPtr, + name: str, + ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: + """查找或创建变量 alloca（在当前作用域）""" + existing: llvmlite.Value | t.CPtr = lookup_current(symtab, name) + if existing is not None: + return existing + alloca: llvmlite.Value | t.CPtr = llvmlite.build_alloca(builder, ty) + if alloca is None: + return None + define_var(symtab, name, alloca) + return alloca diff --git a/TransPyV/App/lib/core/Handles/HandlesWhile.py b/TransPyV/App/lib/core/Handles/HandlesWhile.py new file mode 100644 index 0000000..397899e --- /dev/null +++ b/TransPyV/App/lib/core/Handles/HandlesWhile.py @@ -0,0 +1,130 @@ +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + + +# ============================================================ +# HandlesWhile - while 循环语句处理（Mixin 继承模式） +# +# 翻译 while 语句为 LLVM IR 控制流: +# br label %cond +# cond: +# %t = +# %c = icmp ne i32 %t, 0 +# br i1 %c, label %body, label %end +# body: +# ... body ... +# br label %cond +# end: +# ============================================================ + + +@t.NoVTable +class WhileHandle(HandlesBase.Mixin): + """while 循环语句处理器：继承 Mixin 获得 Trans 回指针""" + + def __init__(self, trans: HT.Translator | t.CPtr): + self.Trans = trans + + # ============================================================ + # Handle - 处理 while 语句，返回新增变量数 + # ============================================================ + def Handle(self, node: ast.AST | t.CPtr) -> int: + """翻译 while 循环语句""" + if node is None: + return 0 + + trans: HT.Translator | t.CPtr = self.Trans + pool: memhub.MemBuddy | t.CPtr = trans.Pool + builder: llvmlite.IRBuilder | t.CPtr = trans._cur_builder + func: llvmlite.Function | t.CPtr = trans._cur_func + + if builder is None or func is None: + return 0 + + while_node: ast.While | t.CPtr = (ast.While | t.CPtr)(node) + + # 1. 创建基本块: cond / body / end（使用 trans._label_counter，不与 SSA 名共享） + cnt: int = trans._label_counter + trans._label_counter = cnt + 1 + + name_buf: t.CChar | t.CPtr = pool.alloc(32) + viperlib.snprintf(name_buf, 32, "while.cond.%d", cnt) + cond_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "while.body.%d", cnt) + body_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + viperlib.snprintf(name_buf, 32, "while.end.%d", cnt) + end_bb: llvmlite.BasicBlock | t.CPtr = llvmlite.create_block(pool, func, name_buf) + + # 2. 跳转到 cond 块 + llvmlite.build_br(builder, cond_bb) + + # 3. cond 块: 求值条件，条件分支 + llvmlite.position_at_end(builder, cond_bb) + cond_val: llvmlite.Value | t.CPtr = HandlesExpr.translate_value( + builder, pool, trans.Module, while_node.test, + trans._funcs, trans._func_count, trans) + if cond_val is None: + cond_val = llvmlite.const_int32(pool, 0) + + # Compare/Not 表达式已返回 i1，直接使用；其他类型与 0 比较 + cond_bits: int = HandlesExpr.get_llvm_type_bits(cond_val.Ty) + if cond_bits == 1: + cond_i1: llvmlite.Value | t.CPtr = cond_val + else: + zero: llvmlite.Value | t.CPtr = llvmlite.const_int32(pool, 0) + cond_i1 = llvmlite.build_icmp( + builder, llvmlite.ICMP_NE, cond_val, zero) + llvmlite.build_cond_br(builder, cond_i1, body_bb, end_bb) + + # 4. body 块: 翻译循环体，跳回 cond + llvmlite.position_at_end(builder, body_bb) + + # 保存旧循环上下文，设置 break/continue 目标 + old_break: llvmlite.BasicBlock | t.CPtr = trans._break_bb + old_continue: llvmlite.BasicBlock | t.CPtr = trans._continue_bb + trans._break_bb = end_bb + trans._continue_bb = cond_bb + + body: list[ast.AST | t.CPtr] | t.CPtr = while_node.children + if body is not None: + body_count: t.CSizeT = body.__len__() + for bi in range(body_count): + stmt: ast.AST | t.CPtr = body.get(bi) + if stmt is not None: + HandlesBody.translate_stmt(trans, stmt) + + # 恢复旧循环上下文 + trans._break_bb = old_break + trans._continue_bb = old_continue + + if llvmlite.builder_cur_block_is_terminated(builder) == 0: + llvmlite.build_br(builder, cond_bb) + + # 5. 定位到 end 块 + llvmlite.position_at_end(builder, end_bb) + return 0 + + +# ============================================================ +# NewWhileHandle - 工厂函数 +# ============================================================ +def NewWhileHandle(pool: memhub.MemBuddy | t.CPtr, + trans: HT.Translator | t.CPtr) -> WhileHandle | t.CPtr: + h: WhileHandle | t.CPtr = pool.alloc(WhileHandle.__sizeof__()) + if h is None: + return None + string.memset(h, 0, WhileHandle.__sizeof__()) + h.Trans = trans + return h diff --git a/TransPyV/App/lib/core/Handles/__init__.py b/TransPyV/App/lib/core/Handles/__init__.py new file mode 100644 index 0000000..619fc36 --- /dev/null +++ b/TransPyV/App/lib/core/Handles/__init__.py @@ -0,0 +1,23 @@ +import t, c +from stdint import * + +# Handles 模块（绝对导入，Viper 兼容） +# 参考 Python 版 TransPyC lib/core/Handles 的细粒度拆分 +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesTranslator as HandlesTranslator diff --git a/TransPyV/App/lib/core/IncludesScanner.py b/TransPyV/App/lib/core/IncludesScanner.py new file mode 100644 index 0000000..3c0f5b4 --- /dev/null +++ b/TransPyV/App/lib/core/IncludesScanner.py @@ -0,0 +1,364 @@ +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import hashlib +import viperlib + + +# ============================================================ +# IncludesScanner - includes 目录递归扫描 +# +# 递归扫描 includes 目录，收集所有 .py 文件路径和 SHA1， +# 为 Phase1 stub 生成和 Phase2 stub 合并提供基础数据。 +# +# 使用 Win32 FindFirstFileA/FindNextFileA 实现目录遍历。 +# ============================================================ + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 文件路径最大长度 +MAX_PATH_LEN: t.CDefine = 512 +# 单次扫描最大文件数 +MAX_FILES: t.CDefine = 256 + + +# ============================================================ +# FileEntry - 文件条目（路径 + SHA1） +# ============================================================ +@t.NoVTable +class FileEntry: + """扫描到的文件条目""" + Path: str # 文件绝对路径 + Sha1: str # SHA1 前16字符（16字节+null） + RelPath: str # 相对于 includes 根目录的路径 + ModuleName: str # 模块名（如 "ast.parser"） + + +# ============================================================ +# ScanResult - 扫描结果 +# ============================================================ +@t.NoVTable +class ScanResult: + """扫描结果集合""" + Entries: FileEntry | t.CPtr # FileEntry 数组 + Count: t.CInt + Capacity: t.CInt + + +# ============================================================ +# create_scan_result - 创建扫描结果容器 +# ============================================================ +def create_scan_result(pool: memhub.MemBuddy | t.CPtr) -> ScanResult | t.CPtr: + """创建扫描结果容器，预分配 MAX_FILES 个槽位""" + if pool is None: + return None + size: t.CSizeT = MAX_FILES * FileEntry.__sizeof__() + # 使用 stdlib.malloc 避免 mbuddy 池耗尽（与 StubMerger 保持一致） + entries: FileEntry | t.CPtr = stdlib.malloc(size) + if entries is None: + return None + string.memset(entries, 0, size) + + result: ScanResult | t.CPtr = stdlib.malloc(ScanResult.__sizeof__()) + if result is None: + return None + string.memset(result, 0, ScanResult.__sizeof__()) + result.Entries = entries + result.Count = 0 + result.Capacity = MAX_FILES + return result + + +# ============================================================ +# add_file_entry - 向扫描结果添加文件条目 +# ============================================================ +def add_file_entry(result: ScanResult | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + abs_path: str, rel_path: str, + sha1: str) -> int: + """添加文件条目到扫描结果，返回 0 成功""" + if result is None or pool is None: + return 1 + if result.Count >= result.Capacity: + return 1 + + # 计算模块名：rel_path 中的 / 替换为 .，去掉 .py 扩展名 + rel_len: t.CSizeT = string.strlen(rel_path) + mod_buf: str = stdlib.malloc(rel_len + 1) + if mod_buf is None: + return 1 + string.strcpy(mod_buf, rel_path) + # 替换 / 为 . + for i in range(rel_len): + ch: t.CChar = mod_buf[i] + if ch == '/' or ch == '\\': + mod_buf[i] = '.' + # 去掉 .py 扩展名 + if rel_len >= 3: + if mod_buf[rel_len - 3] == '.' and mod_buf[rel_len - 2] == 'p' and mod_buf[rel_len - 1] == 'y': + mod_buf[rel_len - 3] = '\0' + + # 获取条目地址 + entry_size: t.CSizeT = FileEntry.__sizeof__() + entry_addr: t.CUInt64T = t.CUInt64T(result.Entries) + result.Count * entry_size + entry: FileEntry | t.CPtr = (FileEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry is None: + return 1 + + # 复制路径字符串 + abs_len: t.CSizeT = string.strlen(abs_path) + abs_buf: str = stdlib.malloc(abs_len + 1) + if abs_buf is None: + return 1 + string.strcpy(abs_buf, abs_path) + entry.Path = abs_buf + + # 复制 SHA1 + sha1_buf: str = stdlib.malloc(17) + if sha1_buf is None: + return 1 + string.strcpy(sha1_buf, sha1) + entry.Sha1 = sha1_buf + + # 复制相对路径 + rel_buf: str = stdlib.malloc(rel_len + 1) + if rel_buf is None: + return 1 + string.strcpy(rel_buf, rel_path) + entry.RelPath = rel_buf + + # 复制模块名 + entry.ModuleName = mod_buf + + result.Count += 1 + return 0 + + +# ============================================================ +# compute_file_sha1 - 读取文件内容并计算 SHA1 +# +# 与 Projectrans.py 一致：CRLF → LF 转换后计算 SHA1。 +# Projectrans.py 用 Python 文本模式读取（自动 CRLF→LF）， +# IncludesScanner 用二进制模式读取，需手动去除 \r。 +# ============================================================ +def compute_file_sha1(pool: memhub.MemBuddy | t.CPtr, + file_path: str) -> str: + """读取文件内容并计算 SHA1 前16字符（CRLF→LF 后计算，与 Projectrans.py 一致）""" + if pool is None or file_path is None: + return None + + # 打开文件 + f: fileio.File | t.CPtr = fileio.File(file_path, fileio.MODE.R) + if f.closed: + return None + + # 分配读取缓冲区（128KB，足够大多数 .py 文件） + # 使用 stdlib.malloc 避免 mbuddy 池耗尽（每文件 128KB，60+ 文件会耗尽 16MB 池） + BUF_SIZE: t.CSizeT = 131072 + buf: bytes = stdlib.malloc(BUF_SIZE) + if buf is None: + f.close() + return None + + # 读取文件内容 + bytes_read: t.CInt64T = f.read_all(buf, BUF_SIZE) + f.close() + if bytes_read <= 0: + return None + + # 原地去除 \r（CRLF → LF），与 Projectrans.py 文本模式读取一致 + write_pos: t.CSizeT = 0 + read_pos: t.CSizeT = 0 + while read_pos < bytes_read: + ch: t.CChar = buf[read_pos] + if ch != '\r': + buf[write_pos] = ch + write_pos += 1 + read_pos += 1 + + # 添加 null 终止符 + buf[write_pos] = 0 + + # 计算 SHA1 + ctx: hashlib.sha1 | t.CPtr = hashlib.sha1() + if ctx is None: + return None + ctx.update(buf) + + digest: bytes = stdlib.malloc(hashlib.SHA1_DIGEST_LEN) + if digest is None: + return None + ctx.final(digest) + + # 转为十六进制字符串（取前 8 字节 = 16 个十六进制字符） + hex_buf: str = stdlib.malloc(17) + if hex_buf is None: + return None + for i in range(8): + hi: int = (digest[i] >> 4) & 0xF + lo: int = digest[i] & 0xF + if hi < 10: + hex_buf[i * 2] = '0' + hi + else: + hex_buf[i * 2] = 'a' + (hi - 10) + if lo < 10: + hex_buf[i * 2 + 1] = '0' + lo + else: + hex_buf[i * 2 + 1] = 'a' + (lo - 10) + hex_buf[16] = '\0' + return hex_buf + + +# ============================================================ +# scan_directory_recursive - 递归扫描目录 +# +# 使用 FindFirstFileA/FindNextFileA 遍历目录树， +# 对每个 .py 文件计算 SHA1 并添加到结果中。 +# ============================================================ +def scan_directory_recursive(pool: memhub.MemBuddy | t.CPtr, + root_dir: str, + rel_prefix: str, + result: ScanResult | t.CPtr) -> int: + """递归扫描目录，收集 .py 文件""" + if pool is None or root_dir is None or result is None: + return 1 + + # 构造搜索模式: root_dir/* + root_len: t.CSizeT = string.strlen(root_dir) + pattern: bytes = stdlib.malloc(root_len + 4) + if pattern is None: + return 1 + viperlib.snprintf(pattern, root_len + 4, "%s/*", root_dir) + + # 使用 FindFirstFileA 开始搜索 + find_data: win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(win32file.WIN32_FIND_DATAA.__sizeof__()) + if find_data is None: + return 1 + string.memset(find_data, 0, win32file.WIN32_FIND_DATAA.__sizeof__()) + + handle: win32base.HANDLE = win32file.FindFirstFileA(pattern, find_data) + if handle == win32base.INVALID_HANDLE_VALUE: + return 1 + + # 遍历所有文件和子目录 + while True: + # 跳过 . 和 .. + fname: str = find_data.cFileName + if fname is not None: + fname0: t.CChar = fname[0] + if fname0 == '.': + fname1: t.CChar = fname[1] + if fname1 == '\0': + # "." + if win32file.FindNextFileA(handle, find_data) == 0: + break + continue + elif fname1 == '.': + fname2: t.CChar = fname[2] + if fname2 == '\0': + # ".." + if win32file.FindNextFileA(handle, find_data) == 0: + break + continue + + # 检查是否为目录 + is_dir: int = find_data.dwFileAttributes & win32file.FILE_ATTRIBUTE_DIRECTORY + + # 构造完整路径 + fname_len: t.CSizeT = string.strlen(fname) + full_path: bytes = stdlib.malloc(root_len + fname_len + 2) + if full_path is None: + break + viperlib.snprintf(full_path, root_len + fname_len + 2, "%s/%s", root_dir, fname) + + # 构造相对路径 + prefix_len: t.CSizeT = 0 + if rel_prefix is not None: + prefix_len = string.strlen(rel_prefix) + rel_path: bytes = stdlib.malloc(prefix_len + fname_len + 2) + if rel_path is None: + break + if rel_prefix is not None and prefix_len > 0: + viperlib.snprintf(rel_path, prefix_len + fname_len + 2, "%s/%s", rel_prefix, fname) + else: + string.strcpy(rel_path, fname) + + if is_dir != 0: + # 递归扫描子目录 + scan_directory_recursive(pool, full_path, rel_path, result) + else: + # 检查是否为 .py 文件 + is_py: int = 0 + if fname_len >= 3: + if fname[fname_len - 3] == '.' and fname[fname_len - 2] == 'p' and fname[fname_len - 1] == 'y': + is_py = 1 + + if is_py != 0: + # 跳过 __pycache__ 目录下的文件 + is_pycache: int = 0 + if rel_prefix is not None: + if string.strcmp(rel_prefix, "__pycache__") == 0: + is_pycache = 1 + + if is_pycache == 0: + # 计算 SHA1 + sha1: str = compute_file_sha1(pool, full_path) + if sha1 is not None: + add_file_entry(result, pool, full_path, rel_path, sha1) + stdio.printf(" [scan] %s -> %s\n", rel_path, sha1) + + # 继续搜索下一个文件 + if win32file.FindNextFileA(handle, find_data) == 0: + break + + win32file.FindClose(handle) + return 0 + + +# ============================================================ +# scan_includes - 扫描 includes 目录入口 +# +# 扫描指定的 includes 目录，返回所有 .py 文件的路径和 SHA1。 +# ============================================================ +def scan_includes(pool: memhub.MemBuddy | t.CPtr, + includes_dir: str) -> ScanResult | t.CPtr: + """扫描 includes 目录，返回所有 .py 文件的扫描结果""" + if pool is None or includes_dir is None: + return None + + stdio.printf("[Phase1] 扫描 includes 目录: %s\n", includes_dir) + + result: ScanResult | t.CPtr = create_scan_result(pool) + if result is None: + return None + + scan_directory_recursive(pool, includes_dir, None, result) + + stdio.printf("[Phase1] 扫描完成: %d 个 .py 文件\n", result.Count) + return result + + +# ============================================================ +# find_entry_by_module - 按模块名查找文件条目 +# ============================================================ +def find_entry_by_module(result: ScanResult | t.CPtr, + module_name: str) -> FileEntry | t.CPtr: + """按模块名查找文件条目""" + if result is None or module_name is None: + return None + entry_size: t.CSizeT = FileEntry.__sizeof__() + for i in range(result.Count): + entry_addr: t.CUInt64T = t.CUInt64T(result.Entries) + i * entry_size + entry: FileEntry | t.CPtr = (FileEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry is not None: + if entry.ModuleName is not None: + if entry.ModuleName == module_name: + return entry + return None diff --git a/TransPyV/App/lib/core/Phase1.py b/TransPyV/App/lib/core/Phase1.py new file mode 100644 index 0000000..88d62f6 --- /dev/null +++ b/TransPyV/App/lib/core/Phase1.py @@ -0,0 +1,540 @@ +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 源代码缓冲区大小（1MB） +SRC_BUF_SIZE: t.CDefine = 1048576 + +# ============================================================ +# RunPhase1 - Phase1: 扫描 includes 目录，按需翻译并生成 stub +# +# 对每个 .py 文件检查 stub 是否已存在，若不存在则翻译并分离 stub。 +# "按需翻译"：仅对 stub 不存在的文件执行翻译，避免重复工作。 +# +# Args: +# mb: 内存池 +# includes_dir: includes 目录路径 +# temp_dir: 临时目录（保存 stub） +# log: 日志器 +# +# Returns: +# 0 成功，非 0 失败 +# ============================================================ +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, + log: VLogger.Logger | t.CPtr) -> int: + """Phase1: 扫描 includes 目录，按需翻译并生成 stub""" + if includes_dir is None or temp_dir is None: + stdio.printf("[Phase1] includes_dir 或 temp_dir 为空，跳过\n") + return 1 + + if log is not None: + log.banner("Phase1: 扫描 includes（按需翻译）") + + # 扫描 includes 目录 + result: IncludesScanner.ScanResult | t.CPtr = IncludesScanner.scan_includes(mb, includes_dir) + if result is None: + stdio.printf("[Phase1] 扫描失败\n") + return 1 + + stdio.printf("[Phase1] 共 %d 个文件\n", result.Count) + + # 读取 _sha1_map.txt 获取需要的 includes SHA1 集合 + # 优先使用 Projectrans.py 生成的 _sha1_map.txt（含依赖分析，只包含需要的 includes） + # 若不存在，则从扫描结果生成（包含所有 includes，可能导致结构体表溢出） + sha1_set: str = stdlib.malloc(StubMerger.MAX_INCLUDES_SHA1 * 17) + if sha1_set is None: + stdio.printf("[Phase1] sha1_set 分配失败\n") + return 1 + string.memset(sha1_set, 0, StubMerger.MAX_INCLUDES_SHA1 * 17) + set_count: int = StubMerger._load_includes_sha1_set(mb, temp_dir, sha1_set) + if set_count <= 0: + stdio.printf("[Phase1] _sha1_map.txt 不存在或为空，从扫描结果生成\n") + StubMerger.WriteIncludesSha1Map(mb, temp_dir, result, None, 0) + set_count = StubMerger._load_includes_sha1_set(mb, temp_dir, sha1_set) + if set_count < 0: + stdio.printf("[Phase1] 无法加载 _sha1_map.txt，跳过 Phase1\n") + return 1 + stdio.printf("[Phase1] includes SHA1 集合: %d 个\n", set_count) + + # 构建模块 SHA1 映射（供跨模块函数调用名混淆使用） + td_len_p1map: t.CSizeT = string.strlen(temp_dir) + p1_sha1_arr: bytes = stdlib.malloc(StubMerger.MAX_INCLUDES * 17) + p1_mod_arr: bytes = stdlib.malloc(StubMerger.MAX_INCLUDES * 64) + p1_inc_count: int = 0 + if p1_sha1_arr is not None and p1_mod_arr is not None: + p1_inc_count = StubMerger._BuildIncludesSha1Map(temp_dir, td_len_p1map, p1_sha1_arr, p1_mod_arr) + HandlesExprCall.set_module_sha1_map(p1_sha1_arr, p1_mod_arr, p1_inc_count) + + # 初始化 AST 表（只需一次） + ast._init_tables(mb) + + # FileEntry 结构体大小（Phase 1a 和 1b 共用） + entry_size: t.CSizeT = IncludesScanner.FileEntry.__sizeof__() + + # ============================================================ + # Phase 1a-pre: 扫描所有 includes 文件的 import 依赖 + # + # 只处理 Import/ImportFrom 语句，跳过 ClassDef/FunctionDef 等， + # 不调用 resolve_annotation_type，避免 list[...] 等不支持的语法触发 crash。 + # 生成 .deps.txt 供依赖图按需翻译使用。 + # ============================================================ + stdio.printf("[Phase1a-pre] 扫描 import 依赖\n") + p1a_registered: int = 0 + p1a_skipped: int = 0 + p1a_failed: int = 0 + + for i in range(result.Count): + entry_addr_a: t.CUInt64T = t.CUInt64T(result.Entries) + i * entry_size + entry_a: IncludesScanner.FileEntry | t.CPtr = (IncludesScanner.FileEntry | t.CPtr)(t.CVoid(entry_addr_a, t.CPtr)) + if entry_a is None: + p1a_failed += 1 + continue + + sha1_a: str = entry_a.Sha1 + if sha1_a is None: + p1a_failed += 1 + continue + + if StubMerger._is_in_sha1_set(sha1_a, sha1_set, set_count) == 0: + p1a_skipped += 1 + continue + + # 读取文件内容 + file_path_a: str = entry_a.Path + f_a: fileio.File | t.CPtr = fileio.File(file_path_a, fileio.MODE.R) + if f_a.closed: + p1a_failed += 1 + continue + + src_buf_a: bytes = stdlib.malloc(SRC_BUF_SIZE) + if src_buf_a is None: + f_a.close() + p1a_failed += 1 + continue + + bytes_read_a: LONG = f_a.read_all(src_buf_a, SRC_BUF_SIZE) + f_a.close() + if bytes_read_a <= 0: + stdlib.free(src_buf_a) + p1a_failed += 1 + continue + if bytes_read_a < SRC_BUF_SIZE: + src_buf_a[bytes_read_a] = 0 + else: + src_buf_a[SRC_BUF_SIZE - 1] = 0 + + # 解析 AST + lx_a: ast.Lexer | t.CPtr = ast.new_lexer(mb) + if lx_a is None: + stdlib.free(src_buf_a) + p1a_failed += 1 + continue + ast._lexer_init(lx_a, src_buf_a, mb) + tokens_a: ast.Token | t.CPtr = ast.tokenize(lx_a) + tree_a: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens_a) + if tree_a is None: + stdlib.free(src_buf_a) + p1a_failed += 1 + continue + + # 创建 Translator，设置 _declare_only=2（只扫描 import，跳过 ClassDef） + tr_a: HandlesTranslator.Translator | t.CPtr = HandlesTranslator.Translator() + if tr_a is None: + stdlib.free(src_buf_a) + p1a_failed += 1 + continue + tr_a.ModuleSha1 = sha1_a + tr_a._declare_only = 2 + tr_a.CurrentPackage = HandlesImports.compute_package_from_relpath(mb, entry_a.RelPath) + HandlesType.set_current_file(file_path_a) + HandlesType.clear_cdefine_constants() + HandlesStruct.reset_visible_structs(mb, 0) + ret_a: int = tr_a.translate(tree_a) + if ret_a != 0: + p1a_failed += 1 + else: + p1a_registered += 1 + + # 生成 .deps.txt（记录依赖模块名，供依赖图按需翻译使用） + td_len_a: t.CSizeT = string.strlen(temp_dir) + deps_path_a: bytes = stdlib.malloc(td_len_a + 32) + if deps_path_a is not None: + viperlib.snprintf(deps_path_a, td_len_a + 32, "%s/%s.deps.txt", temp_dir, sha1_a) + df_a: fileio.File | t.CPtr = fileio.File(deps_path_a, fileio.MODE.W) + if not df_a.closed: + if tr_a._imported_modules is not None: + dl_a: t.CSizeT = string.strlen(tr_a._imported_modules) + df_a.write(tr_a._imported_modules, dl_a) + df_a.close() + stdlib.free(deps_path_a) + + # 释放 Translator 的 C malloc 资源 + if tr_a._global_names is not None: + stdlib.free(tr_a._global_names) + if tr_a._nonlocal_names is not None: + stdlib.free(tr_a._nonlocal_names) + + stdlib.free(src_buf_a) + + stdio.printf("[Phase1a-pre] 完成: 扫描=%d 跳过=%d 失败=%d\n", p1a_registered, p1a_skipped, p1a_failed) + + # ============================================================ + # 依赖图按需翻译：构建可达 SHA1 集合 + # + # 从 Config.SourceDir 的源文件开始，解析 import 语句，递归收集 + # 可达的 includes 文件 SHA1。Phase 1b 只翻译可达集合中的文件， + # 避免翻译不需要的 includes（如 Test 不依赖 llvmlite，则不翻译）。 + # 如果 Config.SourceDir 不可用（仅运行 Phase1），回退到 sha1_set。 + # ============================================================ + reachable_set: str = None + reachable_count: int = 0 + use_reachable: int = 0 + if Config.SourceDir is not None: + reachable_set = stdlib.malloc(StubMerger.MAX_INCLUDES_SHA1 * 17) + if reachable_set is not None: + string.memset(reachable_set, 0, StubMerger.MAX_INCLUDES_SHA1 * 17) + reachable_count = StubMerger._BuildReachableSha1Set(mb, Config.SourceDir, temp_dir, reachable_set) + if reachable_count > 0: + use_reachable = 1 + stdio.printf("[Phase1b] 使用可达 SHA1 集合过滤: %d 个\n", reachable_count) + # 用可达集合重新生成 _sha1_map.txt（按图求索的最终产物） + # Phase B+ 只遍历这些条目，避免编译不需要的 includes（如 asm.py） + StubMerger.WriteIncludesSha1Map(mb, temp_dir, result, reachable_set, reachable_count) + # 重新加载 sha1_set，使后续 Phase 1a-pre/1a/1b 的过滤也使用可达集合 + string.memset(sha1_set, 0, StubMerger.MAX_INCLUDES_SHA1 * 17) + set_count = StubMerger._load_includes_sha1_set(mb, temp_dir, sha1_set) + stdio.printf("[Phase1b] _sha1_map.txt 已重写为可达集合: %d 个\n", set_count) + else: + stdio.printf("[Phase1b] 可达 SHA1 集合构建失败，回退到全量集合\n") + stdlib.free(reachable_set) + reachable_set = None + + # ============================================================ + # Phase 1a: 注册可达 includes 文件的 struct/enum/union + # + # 解决字母序依赖问题：按字母顺序翻译时，后面的 struct 未注册 + # 前面就需要用。Phase 1a 先注册所有 struct/enum/union 到全局表， + # Phase 1b 全量翻译时 struct 已注册，走 existing 路径只翻译方法体。 + # 只处理可达文件，避免翻译不需要的 includes（如 Test 不依赖 ast 模块）。 + # ============================================================ + stdio.printf("[Phase1a] 注册可达文件 struct/enum/union\n") + stdio.fflush(0) + p1a_reg: int = 0 + p1a_skp: int = 0 + p1a_fl: int = 0 + + for i in range(result.Count): + entry_addr_r: t.CUInt64T = t.CUInt64T(result.Entries) + i * entry_size + entry_r: IncludesScanner.FileEntry | t.CPtr = (IncludesScanner.FileEntry | t.CPtr)(t.CVoid(entry_addr_r, t.CPtr)) + if entry_r is None: + p1a_fl += 1 + continue + + sha1_r: str = entry_r.Sha1 + if sha1_r is None: + p1a_fl += 1 + continue + + # 按需翻译过滤：只处理可达文件 + if use_reachable != 0: + if StubMerger._is_in_sha1_set(sha1_r, reachable_set, reachable_count) == 0: + p1a_skp += 1 + continue + else: + if StubMerger._is_in_sha1_set(sha1_r, sha1_set, set_count) == 0: + p1a_skp += 1 + continue + + # 读取文件内容 + file_path_r: str = entry_r.Path + f_r: fileio.File | t.CPtr = fileio.File(file_path_r, fileio.MODE.R) + if f_r.closed: + p1a_fl += 1 + continue + + src_buf_r: bytes = stdlib.malloc(SRC_BUF_SIZE) + if src_buf_r is None: + f_r.close() + p1a_fl += 1 + continue + + bytes_read_r: LONG = f_r.read_all(src_buf_r, SRC_BUF_SIZE) + f_r.close() + if bytes_read_r <= 0: + stdlib.free(src_buf_r) + p1a_fl += 1 + continue + if bytes_read_r < SRC_BUF_SIZE: + src_buf_r[bytes_read_r] = 0 + else: + src_buf_r[SRC_BUF_SIZE - 1] = 0 + + # 解析 AST + lx_r: ast.Lexer | t.CPtr = ast.new_lexer(mb) + if lx_r is None: + stdlib.free(src_buf_r) + p1a_fl += 1 + continue + ast._lexer_init(lx_r, src_buf_r, mb) + tokens_r: ast.Token | t.CPtr = ast.tokenize(lx_r) + tree_r: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens_r) + if tree_r is None: + stdlib.free(src_buf_r) + p1a_fl += 1 + continue + + # 创建 Translator，设置 _declare_only=1（注册 struct/enum/union） + tr_r: HandlesTranslator.Translator | t.CPtr = HandlesTranslator.Translator() + if tr_r is None: + stdlib.free(src_buf_r) + p1a_fl += 1 + continue + tr_r.ModuleSha1 = sha1_r + tr_r._declare_only = 1 + tr_r.CurrentPackage = HandlesImports.compute_package_from_relpath(mb, entry_r.RelPath) + HandlesType.set_current_file(file_path_r) + HandlesType.clear_cdefine_constants() + HandlesStruct.reset_visible_structs(mb, 0) + ret_r: int = tr_r.translate(tree_r) + if ret_r != 0: + p1a_fl += 1 + else: + p1a_reg += 1 + + # 释放 Translator 的 C malloc 资源 + if tr_r._global_names is not None: + stdlib.free(tr_r._global_names) + if tr_r._nonlocal_names is not None: + stdlib.free(tr_r._nonlocal_names) + + stdlib.free(src_buf_r) + + stdio.printf("[Phase1a] 完成: 注册=%d 跳过=%d 失败=%d\n", p1a_reg, p1a_skp, p1a_fl) + + # ============================================================ + # Phase 1b: 全量翻译（struct 已注册，走 existing 路径翻译方法体） + # ============================================================ + # 遍历每个文件 + translated: int = 0 + skipped: int = 0 + failed: int = 0 + + for i in range(result.Count): + # 获取 entry + entry_addr: t.CUInt64T = t.CUInt64T(result.Entries) + i * entry_size + entry: IncludesScanner.FileEntry | t.CPtr = (IncludesScanner.FileEntry | t.CPtr)(t.CVoid(entry_addr, t.CPtr)) + if entry is None: + failed += 1 + continue + + sha1: str = entry.Sha1 + if sha1 is None: + failed += 1 + continue + + # 检查 SHA1 是否在翻译集合中（按需翻译：只翻译可达的文件） + if use_reachable != 0: + if StubMerger._is_in_sha1_set(sha1, reachable_set, reachable_count) == 0: + skipped += 1 + continue + else: + if StubMerger._is_in_sha1_set(sha1, sha1_set, set_count) == 0: + skipped += 1 + continue + + # 构造 stub 路径: {temp_dir}/{sha1}.stub.ll + dir_len: t.CSizeT = string.strlen(temp_dir) + sha1_len: t.CSizeT = string.strlen(sha1) + stub_path: bytes = stdlib.malloc(dir_len + sha1_len + 16) + if stub_path is None: + failed += 1 + continue + viperlib.snprintf(stub_path, dir_len + sha1_len + 16, "%s/%s.stub.ll", temp_dir, sha1) + + # 检查 stub 是否已存在（按需翻译：跳过已存在的） + sf: fileio.File | t.CPtr = fileio.File(stub_path, fileio.MODE.R) + if not sf.closed: + sf.close() + # 检查 text.ll 是否也存在（虚表扫描需要 text.ll） + text_path: bytes = stdlib.malloc(dir_len + sha1_len + 16) + if text_path is not None: + viperlib.snprintf(text_path, dir_len + sha1_len + 16, "%s/%s.text.ll", temp_dir, sha1) + tf: fileio.File | t.CPtr = fileio.File(text_path, fileio.MODE.R) + if not tf.closed: + tf.close() + stdlib.free(text_path) + stdlib.free(stub_path) + continue + stdlib.free(text_path) + # text.ll 不存在，需要重新翻译 + stdio.printf("[Phase1] text.ll 不存在，重新翻译: %s (sha1=%s)\n", entry.RelPath, sha1) + else: + # stub 不存在，需要翻译 + stdio.printf("[Phase1] 翻译: %s (sha1=%s)\n", entry.RelPath, sha1) + + stdlib.free(stub_path) + + # 读取文件内容 + file_path: str = entry.Path + f: fileio.File | t.CPtr = fileio.File(file_path, fileio.MODE.R) + if f.closed: + stdio.printf("[Phase1] 无法打开: %s\n", file_path) + failed += 1 + continue + + src_buf: bytes = stdlib.malloc(SRC_BUF_SIZE) + if src_buf is None: + f.close() + failed += 1 + continue + + bytes_read: LONG = f.read_all(src_buf, SRC_BUF_SIZE) + f.close() + if bytes_read <= 0: + stdio.printf("[Phase1] 读取失败: %s\n", file_path) + stdlib.free(src_buf) + failed += 1 + continue + if bytes_read < SRC_BUF_SIZE: + src_buf[bytes_read] = 0 + else: + src_buf[SRC_BUF_SIZE - 1] = 0 + + # 解析 AST + lx: ast.Lexer | t.CPtr = ast.new_lexer(mb) + if lx is None: + stdlib.free(src_buf) + failed += 1 + continue + ast._lexer_init(lx, src_buf, mb) + tokens: ast.Token | t.CPtr = ast.tokenize(lx) + tree: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens) + if tree is None: + stdio.printf("[Phase1] AST 解析失败: %s\n", file_path) + stdlib.free(src_buf) + failed += 1 + continue + + # 翻译 AST → LLVM IR + tr: HandlesTranslator.Translator | t.CPtr = HandlesTranslator.Translator() + if tr is None: + stdlib.free(src_buf) + failed += 1 + continue + tr.ModuleSha1 = sha1 + tr.CurrentPackage = HandlesImports.compute_package_from_relpath(mb, entry.RelPath) + HandlesType.set_current_file(file_path) + HandlesType.clear_cdefine_constants() + HandlesStruct.reset_visible_structs(mb, 0) + ret: int = tr.translate(tree) + if ret != 0: + stdio.printf("[Phase1] 翻译失败: %s\n", file_path) + stdlib.free(src_buf) + if tr._global_names is not None: + stdlib.free(tr._global_names) + if tr._nonlocal_names is not None: + stdlib.free(tr._nonlocal_names) + failed += 1 + continue + + # dump stub IR (declarations only) + PHASE1_IR_SIZE: t.CSizeT = 262144 + stub_buf: bytes = stdlib.malloc(PHASE1_IR_SIZE) + if stub_buf is None: + stdlib.free(src_buf) + if tr._global_names is not None: + stdlib.free(tr._global_names) + if tr._nonlocal_names is not None: + stdlib.free(tr._nonlocal_names) + failed += 1 + continue + tr.dump_ir(stub_buf, PHASE1_IR_SIZE, llvmlite.OUTPUT_STUB) + stub_len: t.CSizeT = string.strlen(stub_buf) + + # save stub.ll + dir_len_p1: t.CSizeT = string.strlen(temp_dir) + stub_path_p1: bytes = stdlib.malloc(dir_len_p1 + 32) + if stub_path_p1 is not None: + viperlib.snprintf(stub_path_p1, dir_len_p1 + 32, "%s/%s.stub.ll", temp_dir, sha1) + sf_p1: fileio.File | t.CPtr = fileio.File(stub_path_p1, fileio.MODE.W) + if not sf_p1.closed: + sf_p1.write(stub_buf, stub_len) + sf_p1.close() + stdlib.free(stub_path_p1) + stdlib.free(stub_buf) + + # dump text IR (definitions only) + text_buf: bytes = stdlib.malloc(PHASE1_IR_SIZE) + if text_buf is None: + stdlib.free(src_buf) + if tr._global_names is not None: + stdlib.free(tr._global_names) + if tr._nonlocal_names is not None: + stdlib.free(tr._nonlocal_names) + failed += 1 + continue + tr.dump_ir(text_buf, PHASE1_IR_SIZE, llvmlite.OUTPUT_TEXT) + text_len: t.CSizeT = string.strlen(text_buf) + + # save text.ll + text_path_p1: bytes = stdlib.malloc(dir_len_p1 + 32) + if text_path_p1 is not None: + viperlib.snprintf(text_path_p1, dir_len_p1 + 32, "%s/%s.text.ll", temp_dir, sha1) + tf_p1: fileio.File | t.CPtr = fileio.File(text_path_p1, fileio.MODE.W) + if not tf_p1.closed: + tf_p1.write(text_buf, text_len) + tf_p1.close() + stdlib.free(text_path_p1) + stdlib.free(text_buf) + + # save dependencies (_imported_modules) for Phase B + deps_path_p1: bytes = stdlib.malloc(dir_len_p1 + 32) + if deps_path_p1 is not None: + viperlib.snprintf(deps_path_p1, dir_len_p1 + 32, "%s/%s.deps.txt", temp_dir, sha1) + df_p1: fileio.File | t.CPtr = fileio.File(deps_path_p1, fileio.MODE.W) + if not df_p1.closed: + if tr._imported_modules is not None: + dl: t.CSizeT = string.strlen(tr._imported_modules) + df_p1.write(tr._imported_modules, dl) + df_p1.close() + stdlib.free(deps_path_p1) + + # 释放 Translator 的 C malloc 资源 + if tr._global_names is not None: + stdlib.free(tr._global_names) + if tr._nonlocal_names is not None: + stdlib.free(tr._nonlocal_names) + + # 所有 dump_ir 完成，src_buf 不再需要 + stdlib.free(src_buf) + + translated += 1 + + stdio.printf("[Phase1] 完成: 翻译=%d 跳过=%d 失败=%d\n", translated, skipped, failed) + + # 释放可达 SHA1 集合（如果分配了） + if reachable_set is not None: + stdlib.free(reachable_set) + + return 0 diff --git a/TransPyV/App/lib/core/Phase2.py b/TransPyV/App/lib/core/Phase2.py new file mode 100644 index 0000000..ef5c71c --- /dev/null +++ b/TransPyV/App/lib/core/Phase2.py @@ -0,0 +1,680 @@ +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 源代码缓冲区大小（1MB） +SRC_BUF_SIZE: t.CDefine = 1048576 + +# 最大源文件数（递归扫描子目录后文件数增多，增大到 64） +MAX_SRC_FILES: t.CDefine = 64 + + +@t.NoVTable +class SrcFileEntry: + """源文件条目""" + Path: str # 完整路径 + Sha1: str # SHA1 前16字符 + + +# ============================================================ +# _ScanDirForPyFiles - 递归扫描目录，收集 .py 文件到 entries +# +# 替代旧的非递归 FindFirstFileA(dir/*.py) 扫描，支持子目录 +# （如 App/lib/core/Handles/）。遇到目录递归扫描，遇到 .py +# 文件读取内容、计算 SHA1、加入 entries。 +# +# Args: +# mb: 内存池 +# dir_path: 当前扫描目录 +# entries: SrcFileEntry 数组 +# entry_size: 单个条目大小（SrcFileEntry.__sizeof__()） +# file_count: 当前已收集的文件数 +# max_files: 最大文件数 +# +# Returns: +# 新的 file_count +# ============================================================ +def _ScanDirForPyFiles(mb: memhub.MemBuddy | t.CPtr, + dir_path: str, + entries: SrcFileEntry | t.CPtr, + entry_size: t.CSizeT, + file_count: int, max_files: int) -> int: + """递归扫描目录，收集 .py 文件到 entries，返回新的 file_count""" + if dir_path is None or entries is None: + return file_count + if file_count >= max_files: + return file_count + + dir_len: t.CSizeT = string.strlen(dir_path) + pattern: bytes = stdlib.malloc(dir_len + 16) + if pattern is None: + return file_count + viperlib.snprintf(pattern, dir_len + 16, "%s/*", dir_path) + + find_data_size: t.CSizeT = w32.win32file.WIN32_FIND_DATAA.__sizeof__() + find_data: w32.win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(find_data_size + 16) + if find_data is None: + stdlib.free(pattern) + return file_count + string.memset(find_data, 0, find_data_size + 16) + + handle: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(pattern, find_data) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + stdlib.free(pattern) + stdlib.free(find_data) + return file_count + + while True: + fname: str = find_data.cFileName + if fname is not None: + fname_len: t.CSizeT = string.strlen(fname) + if fname_len > 0: + # 跳过 . 和 .. + is_dot: int = 0 + if fname_len == 1 and fname[0] == '.': + is_dot = 1 + elif fname_len == 2 and fname[0] == '.' and fname[1] == '.': + is_dot = 1 + if is_dot == 0: + # 检查是否是目录 + attrs: ULONG = find_data.dwFileAttributes + is_dir: int = 0 + if (attrs & w32.win32file.FILE_ATTRIBUTE_DIRECTORY) != 0: + is_dir = 1 + if is_dir != 0: + # 递归扫描子目录 + sub_dir: bytes = stdlib.malloc(dir_len + fname_len + 2) + if sub_dir is not None: + viperlib.snprintf(sub_dir, dir_len + fname_len + 2, "%s/%s", dir_path, fname) + file_count = _ScanDirForPyFiles(mb, sub_dir, entries, entry_size, file_count, max_files) + stdlib.free(sub_dir) + else: + # 检查是否是 .py 文件 + if fname_len > 3: + if fname[fname_len - 3] == '.' and fname[fname_len - 2] == 'p' and fname[fname_len - 1] == 'y': + if file_count < max_files: + full_path: bytes = stdlib.malloc(dir_len + fname_len + 2) + if full_path is not None: + viperlib.snprintf(full_path, dir_len + fname_len + 2, "%s/%s", dir_path, fname) + sf: fileio.File | t.CPtr = fileio.File(full_path, fileio.MODE.R) + if not sf.closed: + sbuf: bytes = stdlib.malloc(SRC_BUF_SIZE) + if sbuf is not None: + br: LONG = sf.read_all(sbuf, SRC_BUF_SIZE) + sf.close() + if br > 0: + if br < SRC_BUF_SIZE: + sbuf[br] = 0 + else: + sbuf[SRC_BUF_SIZE - 1] = 0 + sha1_val: str = Utils.compute_sha1(mb, sbuf) + if sha1_val is not None: + ea: t.CUInt64T = t.CUInt64T(entries) + file_count * entry_size + ent: SrcFileEntry | t.CPtr = (SrcFileEntry | t.CPtr)(t.CVoid(ea, t.CPtr)) + if ent is not None: + ent.Path = full_path + ent.Sha1 = sha1_val + file_count += 1 + stdio.printf("[project] %s (sha1=%s)\n", fname, sha1_val) + stdlib.free(sbuf) + # full_path 不释放：ent.Path 引用它 + + if w32.win32file.FindNextFileA(handle, find_data) == 0: + break + + w32.win32file.FindClose(handle) + stdlib.free(pattern) + stdlib.free(find_data) + return file_count + + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, + source_dir: str, temp_dir: str, output_dir: str, + cc_cmd: str, cc_flags: str, + linker_cmd: str, linker_flags: str, linker_output: str, + includes_binary_dir: str, + includes_dir: str, + do_phase1: int, do_phase2: int, + log: VLogger.Logger | t.CPtr, + args: argparse.ParsedArgs | t.CPtr) -> int: + """多文件项目编译 + + Returns: 0 成功，非 0 失败 + """ + if source_dir is None: + stdio.printf("[project] source_dir 为空\n") + return 1 + + stdio.printf("[project] 多文件项目编译: %s\n", source_dir) + + # === 1. 递归扫描 source_dir 下的 .py 文件（包括子目录）=== + entry_size: t.CSizeT = SrcFileEntry.__sizeof__() + entries: SrcFileEntry | t.CPtr = stdlib.malloc(MAX_SRC_FILES * entry_size) + if entries is None: + return 1 + string.memset(entries, 0, MAX_SRC_FILES * entry_size) + + file_count: int = _ScanDirForPyFiles(mb, source_dir, entries, entry_size, 0, MAX_SRC_FILES) + stdio.printf("[project] 共 %d 个源文件\n", file_count) + + if file_count == 0: + stdlib.free(entries) + return 1 + + # 初始化 AST 表 + ast._init_tables(mb) + + # 确保 temp/output 目录存在 + BuildPipeline.ensure_dir(temp_dir) + BuildPipeline.ensure_dir(output_dir) + + # 追加 App 源文件 SHA1 到 _sha1_map.txt（供日志转存和符号查找使用） + # 格式: {sha1}:App/{filename}\n （与 includes 条目格式对应） + td_len_am: t.CSizeT = string.strlen(temp_dir) + map_path_am: bytes = stdlib.malloc(td_len_am + 32) + if map_path_am is not None: + viperlib.snprintf(map_path_am, td_len_am + 32, "%s/_sha1_map.txt", temp_dir) + mf: fileio.File | t.CPtr = fileio.File(map_path_am, fileio.MODE.A) + if not mf.closed: + line_am: bytes = stdlib.malloc(512) + if line_am is not None: + for i in range(file_count): + ea_am: t.CUInt64T = t.CUInt64T(entries) + i * entry_size + ent_am: SrcFileEntry | t.CPtr = (SrcFileEntry | t.CPtr)(t.CVoid(ea_am, t.CPtr)) + if ent_am is None or ent_am.Path is None or ent_am.Sha1 is None: + continue + # 从路径提取文件名（最后一个 / 或 \\ 之后的部分） + p_str: str = ent_am.Path + p_len: t.CSizeT = string.strlen(p_str) + f_start: t.CSizeT = 0 + for j in range(p_len): + if p_str[j] == '/' or p_str[j] == '\\': + f_start = j + 1 + viperlib.snprintf(line_am, 512, "%s:App/%s\n", ent_am.Sha1, p_str + f_start) + ll_am: t.CSizeT = string.strlen(line_am) + mf.write(line_am, ll_am) + stdlib.free(line_am) + mf.close() + stdio.printf("[project] 已追加 %d 个 App 文件到 _sha1_map.txt\n", file_count) + stdlib.free(map_path_am) + + # 构建模块 SHA1 映射（供跨模块函数调用名混淆使用） + td_len_pb: t.CSizeT = string.strlen(temp_dir) + pb_sha1_arr: bytes = stdlib.malloc(StubMerger.MAX_INCLUDES * 17) + pb_mod_arr: bytes = stdlib.malloc(StubMerger.MAX_INCLUDES * 64) + pb_inc_count: int = 0 + if pb_sha1_arr is not None and pb_mod_arr is not None: + pb_inc_count = StubMerger._BuildIncludesSha1Map(temp_dir, td_len_pb, pb_sha1_arr, pb_mod_arr) + # 追加用户源文件的 (模块名, SHA1) 到全局映射 + # 使跨模块方法调用能通过 from_imports + _lookup_module_sha1 找到正确的 SHA1 + if pb_sha1_arr is not None and pb_mod_arr is not None: + for i in range(file_count): + ea_us: t.CUInt64T = t.CUInt64T(entries) + i * entry_size + ent_us: SrcFileEntry | t.CPtr = (SrcFileEntry | t.CPtr)(t.CVoid(ea_us, t.CPtr)) + if ent_us is None or ent_us.Path is None or ent_us.Sha1 is None: + continue + if pb_inc_count >= StubMerger.MAX_INCLUDES: + break + # 从路径提取模块名（文件名去掉 .py 后缀） + path_str: str = ent_us.Path + path_len: t.CSizeT = string.strlen(path_str) + fname_start: t.CSizeT = 0 + for j in range(path_len): + if path_str[j] == '/' or path_str[j] == '\\': + fname_start = j + 1 + mod_len: t.CSizeT = path_len - fname_start + if mod_len < 4: + continue + mod_len -= 3 + if mod_len >= 64: + mod_len = 63 + # 写入模块名 + mod_idx: t.CSizeT = t.CSizeT(pb_inc_count) * 64 + for k in range(mod_len): + pb_mod_arr[mod_idx + k] = path_str[fname_start + k] + pb_mod_arr[mod_idx + mod_len] = '\0' + # 写入 SHA1 + sha1_idx: t.CSizeT = t.CSizeT(pb_inc_count) * 17 + string.strcpy(pb_sha1_arr + sha1_idx, ent_us.Sha1) + pb_inc_count += 1 + HandlesExprCall.set_module_sha1_map(pb_sha1_arr, pb_mod_arr, pb_inc_count) + + + # === 2. Phase A: 为每个文件生成 stub + text === + if do_phase1 != 0: + if log is not None: + log.banner("Phase A: 生成 stub + text") + + PHASE_A_IR_SIZE: t.CSizeT = 262144 + td_len_pa: t.CSizeT = string.strlen(temp_dir) + + for i in range(file_count): + ea: t.CUInt64T = t.CUInt64T(entries) + i * entry_size + ent: SrcFileEntry | t.CPtr = (SrcFileEntry | t.CPtr)(t.CVoid(ea, t.CPtr)) + if ent is None or ent.Path is None: + continue + + # 计算源文件的包名（相对 source_dir 的目录部分） + src_dir_len_pa: t.CSizeT = string.strlen(source_dir) + pkg_pa: str = None + if string.strlen(ent.Path) > src_dir_len_pa + 1: + rel_path_pa: str = ent.Path + src_dir_len_pa + 1 + pkg_pa = HandlesImports.compute_package_from_relpath(mb, rel_path_pa) + tr_a: HandlesTranslator.Translator | t.CPtr = BuildPipeline.TranslateFileGetTrans(mb, ent.Path, ent.Sha1, pkg_pa) + if tr_a is None: + continue + + # dump stub IR (declarations only) + stub_buf_a: bytes = stdlib.malloc(PHASE_A_IR_SIZE) + if stub_buf_a is None: + continue + tr_a.dump_ir(stub_buf_a, PHASE_A_IR_SIZE, llvmlite.OUTPUT_STUB) + stub_len_a: t.CSizeT = string.strlen(stub_buf_a) + + # save stub.ll + stub_path_a: bytes = stdlib.malloc(td_len_pa + 32) + if stub_path_a is not None: + viperlib.snprintf(stub_path_a, td_len_pa + 32, "%s/%s.stub.ll", temp_dir, ent.Sha1) + sf_a: fileio.File | t.CPtr = fileio.File(stub_path_a, fileio.MODE.W) + if not sf_a.closed: + sf_a.write(stub_buf_a, stub_len_a) + sf_a.close() + stdlib.free(stub_path_a) + stdlib.free(stub_buf_a) + + # dump text IR (definitions only) + text_buf_a: bytes = stdlib.malloc(PHASE_A_IR_SIZE) + if text_buf_a is None: + continue + tr_a.dump_ir(text_buf_a, PHASE_A_IR_SIZE, llvmlite.OUTPUT_TEXT) + text_len_a: t.CSizeT = string.strlen(text_buf_a) + + # save text.ll + text_path_a: bytes = stdlib.malloc(td_len_pa + 32) + if text_path_a is not None: + viperlib.snprintf(text_path_a, td_len_pa + 32, "%s/%s.text.ll", temp_dir, ent.Sha1) + tf_a: fileio.File | t.CPtr = fileio.File(text_path_a, fileio.MODE.W) + if not tf_a.closed: + tf_a.write(text_buf_a, text_len_a) + tf_a.close() + stdlib.free(text_path_a) + stdlib.free(text_buf_a) + + # save dependencies (_imported_modules) for Phase B + deps_path_a: bytes = stdlib.malloc(td_len_pa + 32) + if deps_path_a is not None: + viperlib.snprintf(deps_path_a, td_len_pa + 32, "%s/%s.deps.txt", temp_dir, ent.Sha1) + df_a: fileio.File | t.CPtr = fileio.File(deps_path_a, fileio.MODE.W) + if not df_a.closed: + if tr_a._imported_modules is not None: + dl_a: t.CSizeT = string.strlen(tr_a._imported_modules) + df_a.write(tr_a._imported_modules, dl_a) + df_a.close() + stdlib.free(deps_path_a) + + if do_phase2 == 0: + stdio.printf("[project] Phase A 完成（仅 stub 生成）\n") + stdlib.free(entries) + return 0 + + # === 3. Phase B: 编译每个文件为 .obj === + if do_phase2 != 0: + if log is not None: + log.banner("Phase B: 编译 .obj") + + # 收集 .obj 路径 + OBJ_PATHS_SIZE: t.CSizeT = 8192 + obj_paths: bytes = stdlib.malloc(OBJ_PATHS_SIZE) + if obj_paths is None: + return 1 + obj_paths[0] = '\0' + obj_pos: t.CSizeT = 0 + # main_obj_path: 存放定义了用户 main 的 .obj 路径（链接时放在最前面， + # 避免 --allow-multiple-definition 选择了其他模块的 wrapper main） + main_obj_path: bytes = stdlib.malloc(512) + if main_obj_path is None: + return 1 + main_obj_path[0] = '\0' + compiled_count: int = 0 + + # 组合 IR 缓冲区大小（4MB，足够容纳本地 stub + 所有依赖 stub + 本地 text） + COMBINED_IR_SIZE: t.CSizeT = 4194304 + + for i in range(file_count): + ea: t.CUInt64T = t.CUInt64T(entries) + i * entry_size + ent: SrcFileEntry | t.CPtr = (SrcFileEntry | t.CPtr)(t.CVoid(ea, t.CPtr)) + if ent is None or ent.Path is None or ent.Sha1 is None: + continue + + stdio.printf("[Phase B] %s\n", ent.Path) + + # 组合本地 stub + 所有依赖 stub + 本地 text → 完整 IR + stdio.printf("[Phase B] malloc %d bytes...\n", COMBINED_IR_SIZE) + combined_ir: bytes = stdlib.malloc(COMBINED_IR_SIZE) + if combined_ir is None: + stdio.printf("[Phase B] combined_ir 分配失败: %s\n", ent.Path) + continue + stdio.printf("[Phase B] malloc OK, calling BuildCombinedIR...\n") + combined_len: t.CSizeT = StubMerger.BuildCombinedIR(temp_dir, ent.Sha1, combined_ir, COMBINED_IR_SIZE) + if combined_len == 0: + stdio.printf("[Phase B] BuildCombinedIR 失败: %s\n", ent.Path) + stdlib.free(combined_ir) + continue + + # 编译为 .obj + cret: int = BuildPipeline.compile_module_to_obj( + combined_ir, combined_len, temp_dir, output_dir, ent.Sha1, + cc_cmd, cc_flags) + stdlib.free(combined_ir) + if cret != 0: + stdio.printf("[FATAL][Phase B] llc 编译失败，终止编译: %s\n", ent.Path) + sys.exit(1) + + compiled_count += 1 + + # 构造 .obj 路径，检测是否是 main 模块（test_main.py 或 main.py） + od_len: t.CSizeT = string.strlen(output_dir) + sha1_len: t.CSizeT = string.strlen(ent.Sha1) + need: t.CSizeT = od_len + 1 + sha1_len + 6 + is_main_mod: int = 0 + if string.strstr(ent.Path, "test_main.py") is not None: + is_main_mod = 1 + elif string.strstr(ent.Path, "main.py") is not None: + is_main_mod = 1 + + if is_main_mod != 0: + viperlib.snprintf(main_obj_path, 512, "%s/%s.obj", output_dir, ent.Sha1) + else: + if obj_pos + need < OBJ_PATHS_SIZE: + if obj_pos > 0: + obj_paths[obj_pos] = ' ' + obj_pos += 1 + viperlib.snprintf(obj_paths + obj_pos, need, "%s/%s.obj", output_dir, ent.Sha1) + obj_pos += od_len + 1 + sha1_len + 4 + obj_paths[obj_pos] = '\0' + else: + stdio.printf("[Phase B] 警告: .obj 路径缓冲区不足\n") + + stdio.printf("[Phase B] 编译完成: %d/%d\n", compiled_count, file_count) + + if compiled_count == 0: + stdio.printf("[Phase B] 无成功编译的文件\n") + stdlib.free(entries) + return 1 + + # === 3.5 编译缺失的 includes 文件 === + # includes.binary 可能缺少某些 includes .obj（如 testcheck.py）， + # 这些文件被用户项目导入但未被 TransPyV 自身依赖，Projectrans.py 未编译它们。 + # 检测并编译缺失的 includes 文件到 output_dir，加入链接命令。 + if includes_dir is not None and includes_binary_dir is not None: + inc_compiled: int = 0 + td_len_mi: t.CSizeT = string.strlen(temp_dir) + map_path_mi: bytes = stdlib.malloc(td_len_mi + 32) + if map_path_mi is not None: + viperlib.snprintf(map_path_mi, td_len_mi + 32, "%s/_sha1_map.txt", temp_dir) + mapf_mi: fileio.File | t.CPtr = fileio.File(map_path_mi, fileio.MODE.R) + if not mapf_mi.closed: + map_buf_mi: bytes = stdlib.malloc(65536) + if map_buf_mi is not None: + map_br_mi: t.CInt64T = mapf_mi.read_all(map_buf_mi, 65536) + mapf_mi.close() + if map_br_mi > 0: + if map_br_mi < 65536: + map_buf_mi[map_br_mi] = '\0' + else: + map_buf_mi[65535] = '\0' + map_len_mi: t.CSizeT = map_br_mi + mpos_mi: t.CSizeT = 0 + while mpos_mi < map_len_mi: + ml_start_mi: t.CSizeT = mpos_mi + while mpos_mi < map_len_mi: + if map_buf_mi[mpos_mi] == '\n': + break + mpos_mi += 1 + ml_len_mi: t.CSizeT = mpos_mi - ml_start_mi + mpos_mi += 1 + + # 最小长度: 16(sha1)+1(:)+9(includes/)+1+1+1 = 29 + if ml_len_mi < 26: + continue + if map_buf_mi[ml_start_mi + 16] != ':': + continue + rp_start_mi: t.CSizeT = ml_start_mi + 17 + if string.strncmp(map_buf_mi + rp_start_mi, "includes/", 9) != 0: + continue + + # 提取 SHA1 + inc_sha1_mi: str = stdlib.malloc(17) + if inc_sha1_mi is None: + continue + string.strncpy(inc_sha1_mi, map_buf_mi + ml_start_mi, 16) + inc_sha1_mi[16] = '\0' + + # 检查 .obj 是否已存在于 includes.binary + ibd_len_mi: t.CSizeT = string.strlen(includes_binary_dir) + check_pat_mi: bytes = stdlib.malloc(ibd_len_mi + 35) + if check_pat_mi is None: + continue + viperlib.snprintf(check_pat_mi, ibd_len_mi + 35, "%s/%s*.obj", includes_binary_dir, inc_sha1_mi) + check_fd_mi: w32.win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + obj_exists_mi: int = 0 + if check_fd_mi is not None: + string.memset(check_fd_mi, 0, w32.win32file.WIN32_FIND_DATAA.__sizeof__()) + check_h_mi: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(check_pat_mi, check_fd_mi) + if check_h_mi != w32.win32base.INVALID_HANDLE_VALUE: + w32.win32file.FindClose(check_h_mi) + obj_exists_mi = 1 + if obj_exists_mi != 0: + continue + + # .obj 不存在，需要编译 + # 构造源文件路径: {includes_dir}/{rel_path_without_includes_prefix} + rel_path_len_mi: t.CSizeT = ml_len_mi - 26 + inc_dir_len_mi: t.CSizeT = string.strlen(includes_dir) + src_fp_mi: bytes = stdlib.malloc(inc_dir_len_mi + 1 + rel_path_len_mi + 1) + if src_fp_mi is None: + continue + viperlib.snprintf(src_fp_mi, inc_dir_len_mi + 1 + rel_path_len_mi + 1, + "%s/%s", includes_dir, map_buf_mi + rp_start_mi + 9) + + # 检查是否为声明文件（只有 declare 没有实质 define） + # 判断方法: text.ll 中若有混淆函数 define（含 @\"）则为实现文件 + is_decl_mi: int = -1 + tpath_mi: bytes = stdlib.malloc(td_len_mi + 32) + if tpath_mi is not None: + viperlib.snprintf(tpath_mi, td_len_mi + 32, "%s/%s.text.ll", temp_dir, inc_sha1_mi) + tf_mi: fileio.File | t.CPtr = fileio.File(tpath_mi, fileio.MODE.R) + if not tf_mi.closed: + is_decl_mi = 1 + tbuf_mi: bytes = stdlib.malloc(StubMerger.STUB_READ_BUF_SIZE) + if tbuf_mi is not None: + tbr_mi: t.CInt64T = tf_mi.read_all(tbuf_mi, StubMerger.STUB_READ_BUF_SIZE) + if tbr_mi > 0: + if tbr_mi < StubMerger.STUB_READ_BUF_SIZE: + tbuf_mi[tbr_mi] = '\0' + else: + tbuf_mi[StubMerger.STUB_READ_BUF_SIZE - 1] = '\0' + # 逐行扫描: 找 define 行中含 @\" 的（混淆函数名） + tpos_mi: t.CSizeT = 0 + while tpos_mi < tbr_mi: + tls_mi: t.CSizeT = tpos_mi + while tpos_mi < tbr_mi: + if tbuf_mi[tpos_mi] == '\n': + break + tpos_mi += 1 + tll_mi: t.CSizeT = tpos_mi - tls_mi + if tpos_mi < tbr_mi: + tpos_mi += 1 + if tll_mi >= 7 and string.strncmp(tbuf_mi + tls_mi, "define ", 7) == 0: + # 临时在行尾加 \0 供 strstr 使用 + saved_mi: t.CChar = tbuf_mi[tls_mi + tll_mi] + tbuf_mi[tls_mi + tll_mi] = '\0' + if string.strstr(tbuf_mi + tls_mi, "@\"") is not None: + tbuf_mi[tls_mi + tll_mi] = saved_mi + is_decl_mi = 0 + break + tbuf_mi[tls_mi + tll_mi] = saved_mi + stdlib.free(tbuf_mi) + tf_mi.close() + stdlib.free(tpath_mi) + if is_decl_mi == 1: + stdio.printf("[Phase B+] 跳过(声明文件): %s (sha1=%s)\n", src_fp_mi, inc_sha1_mi) + continue + + stdio.printf("[Phase B+] 编译缺失 includes: %s (sha1=%s)\n", src_fp_mi, inc_sha1_mi) + + # 尝试 BuildCombinedIR（stub/text 应已由 Phase1 生成） + inc_combined_mi: bytes = stdlib.malloc(COMBINED_IR_SIZE) + inc_combined_len_mi: t.CSizeT = 0 + if inc_combined_mi is not None: + inc_combined_len_mi = StubMerger.BuildCombinedIR(temp_dir, inc_sha1_mi, inc_combined_mi, COMBINED_IR_SIZE) + + # 如果 stub/text 不存在，翻译源文件并保存 stub + text，然后重试 + if inc_combined_len_mi == 0 and inc_combined_mi is not None: + stdio.printf("[Phase B+] stub/text 不存在，翻译: %s\n", src_fp_mi) + # 计算 includes 文件的包名（相对 includes_dir 的目录部分） + inc_pkg_mi: str = None + if string.strlen(src_fp_mi) > inc_dir_len_mi + 1: + inc_rel_mi: str = src_fp_mi + inc_dir_len_mi + 1 + inc_pkg_mi = HandlesImports.compute_package_from_relpath(mb, inc_rel_mi) + tr_mi: HandlesTranslator.Translator | t.CPtr = BuildPipeline.TranslateFileGetTrans(mb, src_fp_mi, inc_sha1_mi, inc_pkg_mi) + if tr_mi is not None: + PBP_IR_SIZE: t.CSizeT = 262144 + # 保存 stub.ll + inc_stub_buf: bytes = stdlib.malloc(PBP_IR_SIZE) + if inc_stub_buf is not None: + tr_mi.dump_ir(inc_stub_buf, PBP_IR_SIZE, llvmlite.OUTPUT_STUB) + inc_stub_len: t.CSizeT = string.strlen(inc_stub_buf) + inc_stub_path: bytes = stdlib.malloc(td_len_mi + 32) + if inc_stub_path is not None: + viperlib.snprintf(inc_stub_path, td_len_mi + 32, "%s/%s.stub.ll", temp_dir, inc_sha1_mi) + isf: fileio.File | t.CPtr = fileio.File(inc_stub_path, fileio.MODE.W) + if not isf.closed: + isf.write(inc_stub_buf, inc_stub_len) + isf.close() + stdlib.free(inc_stub_path) + stdlib.free(inc_stub_buf) + # 保存 text.ll + inc_text_buf: bytes = stdlib.malloc(PBP_IR_SIZE) + if inc_text_buf is not None: + tr_mi.dump_ir(inc_text_buf, PBP_IR_SIZE, llvmlite.OUTPUT_TEXT) + inc_text_len: t.CSizeT = string.strlen(inc_text_buf) + inc_text_path: bytes = stdlib.malloc(td_len_mi + 32) + if inc_text_path is not None: + viperlib.snprintf(inc_text_path, td_len_mi + 32, "%s/%s.text.ll", temp_dir, inc_sha1_mi) + itf: fileio.File | t.CPtr = fileio.File(inc_text_path, fileio.MODE.W) + if not itf.closed: + itf.write(inc_text_buf, inc_text_len) + itf.close() + stdlib.free(inc_text_path) + stdlib.free(inc_text_buf) + # 重试 BuildCombinedIR + inc_combined_len_mi = StubMerger.BuildCombinedIR(temp_dir, inc_sha1_mi, inc_combined_mi, COMBINED_IR_SIZE) + + if inc_combined_len_mi == 0: + stdio.printf("[Phase B+] BuildCombinedIR 失败: %s\n", src_fp_mi) + if inc_combined_mi is not None: + stdlib.free(inc_combined_mi) + continue + + # 编译为 .obj + inc_cret_mi: int = BuildPipeline.compile_module_to_obj( + inc_combined_mi, inc_combined_len_mi, temp_dir, output_dir, inc_sha1_mi, + cc_cmd, cc_flags) + stdlib.free(inc_combined_mi) + if inc_cret_mi != 0: + stdio.printf("[FATAL][Phase B+] llc 编译失败，终止编译: %s\n", src_fp_mi) + sys.exit(1) + + inc_compiled += 1 + + # 添加到 obj_paths + od_len_mi: t.CSizeT = string.strlen(output_dir) + sha1_len_mi: t.CSizeT = 16 + need_mi: t.CSizeT = od_len_mi + 1 + sha1_len_mi + 6 + if obj_pos + need_mi < OBJ_PATHS_SIZE: + if obj_pos > 0: + obj_paths[obj_pos] = ' ' + obj_pos += 1 + viperlib.snprintf(obj_paths + obj_pos, need_mi, "%s/%s.obj", output_dir, inc_sha1_mi) + obj_pos += od_len_mi + 1 + sha1_len_mi + 4 + obj_paths[obj_pos] = '\0' + + stdio.printf("[Phase B+] 编译缺失 includes: %d 个\n", inc_compiled) + + # === 4. Phase C: 链接所有 .obj → .exe === + if log is not None: + log.banner("Phase C: 链接") + + od_len2: t.CSizeT = string.strlen(output_dir) + lo_len: t.CSizeT = string.strlen(linker_output) + exe_path: bytes = stdlib.malloc(od_len2 + lo_len + 2) + if exe_path is not None: + viperlib.snprintf(exe_path, od_len2 + lo_len + 2, "%s/%s", output_dir, linker_output) + else: + exe_path = linker_output + + # 构造最终 .obj 路径列表：main_obj_path 在前，其他 .obj 在后 + final_obj_paths: bytes = stdlib.malloc(OBJ_PATHS_SIZE + 512) + if final_obj_paths is None: + stdlib.free(entries) + return 1 + final_obj_paths[0] = '\0' + fop_pos: t.CSizeT = 0 + # main_obj_path 放在最前面（确保 --allow-multiple-definition 选择用户 main） + if main_obj_path[0] != '\0': + mlen: t.CSizeT = string.strlen(main_obj_path) + string.strcpy(final_obj_paths, main_obj_path) + fop_pos = mlen + if obj_paths[0] != '\0': + final_obj_paths[fop_pos] = ' ' + fop_pos += 1 + # 追加其他 .obj + if obj_paths[0] != '\0': + string.strcpy(final_obj_paths + fop_pos, obj_paths) + fop_pos += string.strlen(obj_paths) + final_obj_paths[fop_pos] = '\0' + + obj_paths_len: t.CSizeT = fop_pos + lret: int = BuildPipeline.link_objs_to_exe( + final_obj_paths, obj_paths_len, + linker_cmd, linker_flags, exe_path, + includes_binary_dir) + + if lret == 0: + stdio.printf("输出: %s\n", exe_path) + if args.get_bool("run"): + stdio.printf("[run] 执行: %s\n", exe_path) + rp: subprocess.CompletedProcess | t.CPtr = subprocess.run(exe_path, False, False) + if rp is not None: + stdio.printf("[run] 退出码: %d\n", rp.returncode) + else: + stdio.printf("[Phase C] 链接失败\n") + stdlib.free(entries) + return 1 + + stdlib.free(entries) + return 0 diff --git a/TransPyV/App/lib/core/StubMerger.py b/TransPyV/App/lib/core/StubMerger.py new file mode 100644 index 0000000..708563a --- /dev/null +++ b/TransPyV/App/lib/core/StubMerger.py @@ -0,0 +1,1477 @@ +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + + +# ============================================================ +# StubMerger - stub 声明合并模块 +# +# 从 temp 目录加载所有 .stub.ll 文件，将声明内容合并到主 IR。 +# 这是 Phase2 的核心功能：翻译主文件时，将所有依赖的 stub +# 声明合并到主 IR 前面，使链接器能解析跨模块符号引用。 +# +# 合并策略： +# 1. 扫描 temp 目录，收集所有 {sha1}.stub.ll 文件 +# 2. 读取每个 stub 文件内容 +# 3. 过滤去重（避免重复声明） +# 4. 将声明插入到主 IR 的 target triple 之后 +# ============================================================ + +# 全局 mbuddy 指针 +_mbuddy: memhub.MemManager | t.CPtr + +# 最大 includes SHA1 数（用于过滤） +MAX_INCLUDES_SHA1: t.CDefine = 256 +# stub 读取缓冲区大小 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +# 最大 includes 条目数 +MAX_INCLUDES: t.CDefine = 256 + + +# ============================================================ +# _load_includes_sha1_set - 读取 _sha1_map.txt，收集 includes/ 开头的 SHA1 +# +# _sha1_map.txt 格式：{sha16}:{rel_path} +# 仅收集 rel_path 以 "includes/" 开头的 SHA1，用于过滤 stub 文件， +# 避免加载 TransPyV 自身源文件（lib/*）的 stub 导致链接失败。 +# ============================================================ +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, + temp_dir: str, + sha1_set: str) -> int: + """读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set + sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null） + 返回找到的 includes SHA1 数量，-1 表示错误""" + if pool is None or temp_dir is None or sha1_set is None: + return -1 + + # 构造路径 temp_dir/_sha1_map.txt（使用 stdlib.malloc 避免 mbuddy 池耗尽） + dir_len: t.CSizeT = string.strlen(temp_dir) + map_path: bytes = stdlib.malloc(dir_len + 32) + if map_path is None: + return -1 + viperlib.snprintf(map_path, dir_len + 32, "%s/_sha1_map.txt", temp_dir) + + # 打开文件 + f: fileio.File | t.CPtr = fileio.File(map_path, fileio.MODE.R) + if f.closed: + stdio.printf("[StubMerger] _sha1_map.txt 不存在: %s\n", map_path) + return -1 + + # 读取内容（使用 stdlib.malloc 避免 mbuddy 池耗尽） + MAP_BUF_SIZE: t.CSizeT = 65536 + content: str = stdlib.malloc(MAP_BUF_SIZE) + if content is None: + f.close() + return -1 + bytes_read: t.CInt64T = f.read_all(content, MAP_BUF_SIZE) + f.close() + if bytes_read <= 0: + return -1 + content[bytes_read] = '\0' + + # 解析行: {sha1}:{rel_path} + count: int = 0 + pos: t.CSizeT = 0 + content_len: t.CSizeT = bytes_read + while pos < content_len: + # 找行尾 + line_start: t.CSizeT = pos + while pos < content_len: + if content[pos] == '\n': + break + pos += 1 + line_len: t.CSizeT = pos - line_start + pos += 1 # skip \n + + # 最小长度: 16(sha1) + 1(:) + 9(includes/) = 26 + if line_len < 26: + continue + # 检查第17个字符是 ':' + if content[line_start + 16] != ':': + continue + + # 检查 rel_path 是否以 "includes/" 开头 + rel_start: t.CSizeT = line_start + 17 + if string.strncmp(content + rel_start, "includes/", 9) == 0: + if count < MAX_INCLUDES_SHA1: + string.strncpy(sha1_set + count * 17, content + line_start, 16) + sha1_set[count * 17 + 16] = '\0' + count += 1 + + return count + + +# ============================================================ +# _is_in_sha1_set - 检查 SHA1 是否在集合中 +# ============================================================ +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: + """检查 sha1 是否在集合中""" + if sha1 is None or sha1_set is None: + return 0 + for i in range(set_count): + if string.strcmp(sha1_set + i * 17, sha1) == 0: + return 1 + return 0 + + +# ============================================================ +# WriteIncludesSha1Map - 将 includes 扫描结果写入 _sha1_map.txt +# +# TransPyV.exe 自己生成 _sha1_map.txt（不再依赖 Projectrans.py 预生成）， +# 确保所有 includes 文件的 SHA1 都被记录，供 StubMerger 加载 stub 时过滤。 +# +# Args: +# mb: 内存池 +# temp_dir: 临时目录（_sha1_map.txt 写入位置） +# scan_result: IncludesScanner 扫描结果 +# +# Returns: +# 0 成功，非 0 失败 +# ============================================================ +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, + scan_result: IncludesScanner.ScanResult | t.CPtr, + filter_set: t.CChar | t.CPtr, + filter_count: int) -> int: + """将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用） + + 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。 + """ + if temp_dir is None or scan_result is None: + return 1 + + # 构造路径 temp_dir/_sha1_map.txt + dir_len: t.CSizeT = string.strlen(temp_dir) + map_path: bytes = stdlib.malloc(dir_len + 32) + if map_path is None: + return 1 + viperlib.snprintf(map_path, dir_len + 32, "%s/_sha1_map.txt", temp_dir) + + # 打开文件写入（CREATE_ALWAYS） + f: fileio.File | t.CPtr = fileio.File(map_path, fileio.MODE.W) + if f.closed: + stdio.printf("[Phase1] 无法写入 _sha1_map.txt: %s\n", map_path) + return 1 + + # 写入每个 include 条目: {sha1}:includes/{rel_path}\n + entry_size_w: t.CSizeT = IncludesScanner.FileEntry.__sizeof__() + line_buf: bytes = stdlib.malloc(512) + if line_buf is None: + f.close() + return 1 + + written_count: int = 0 + for i in range(scan_result.Count): + ea_w: t.CUInt64T = t.CUInt64T(scan_result.Entries) + i * entry_size_w + ent_w: IncludesScanner.FileEntry | t.CPtr = (IncludesScanner.FileEntry | t.CPtr)(t.CVoid(ea_w, t.CPtr)) + if ent_w is None or ent_w.Sha1 is None or ent_w.RelPath is None: + continue + # 若提供过滤集合，只写入集合中的条目 + if filter_set is not None and filter_count > 0: + if _is_in_sha1_set(ent_w.Sha1, filter_set, filter_count) == 0: + continue + viperlib.snprintf(line_buf, 512, "%s:includes/%s\n", ent_w.Sha1, ent_w.RelPath) + line_len_w: t.CSizeT = string.strlen(line_buf) + f.write(line_buf, line_len_w) + written_count += 1 + + f.close() + stdio.printf("[Phase1] 已写入 _sha1_map.txt (%d 个 includes)\n", written_count) + return 0 + + +# ============================================================ +# _IsFuncDeclaredOrDefined - 检查 out_buf 中是否已有函数的 declare/define +# +# 逐字节strncmp搜索 func_sig（如 @"sha1.func"(/ @printf( ），每次 +# 找到匹配后检查行首是否以 "declare " 或 "define " 开头，跳过 call 语句。 +# 避免 strstr 子串匹配误匹配 call ... @"func"(...) 中的 @"func"( 子串。 +# +# Args: +# out_buf: 已有的 IR 文本缓冲区 +# func_sig: 函数签名片段，如 @"sha1.func"(/ @printf( +# +# Returns: +# 1=已存在 declare/define, 0=不存在 +# ============================================================ +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: + """检查 out_buf 中是否已有函数的 declare 或 define 行""" + if out_buf is None or func_sig is None: + return 0 + sig_len: t.CSizeT = string.strlen(func_sig) + out_len: t.CSizeT = string.strlen(out_buf) + if sig_len == 0 or out_len == 0 or sig_len > out_len: + return 0 + pos: t.CSizeT = 0 + while pos + sig_len <= out_len: + if string.strncmp(out_buf + pos, func_sig, sig_len) == 0: + ls: t.CSizeT = pos + while ls > 0 and out_buf[ls - 1] != '\n': + ls -= 1 + if string.strncmp(out_buf + ls, "declare ", 8) == 0: + return 1 + if string.strncmp(out_buf + ls, "define ", 7) == 0: + return 1 + pos += 1 + return 0 + + +# ============================================================ +# _CommentOutDeclareInBuf - 将 out_buf 中匹配 func_sig 的 declare 行注释掉 +# +# 用于解决 declare/define 冲突：当本地 text.ll 包含 define 而依赖 +# text.ll 已生成了同函数的 declare 时，将 declare 行首字符改为 ';' +# 使其成为注释，然后追加 define 行。 +# ============================================================ +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: + """在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉""" + if out_buf is None or func_sig is None: + return 0 + sig_len: t.CSizeT = string.strlen(func_sig) + out_len: t.CSizeT = string.strlen(out_buf) + if sig_len == 0 or out_len == 0 or sig_len > out_len: + return 0 + pos: t.CSizeT = 0 + while pos + sig_len <= out_len: + if string.strncmp(out_buf + pos, func_sig, sig_len) == 0: + ls: t.CSizeT = pos + while ls > 0 and out_buf[ls - 1] != '\n': + ls -= 1 + if string.strncmp(out_buf + ls, "declare ", 8) == 0: + out_buf[ls] = ';' + return 1 + pos += 1 + return 0 + + +# ============================================================ +# _CommentOutGlobalInBuf - 将 out_buf 中匹配全局变量名的行注释掉 +# +# 解决 stub/text 全局变量重复定义问题：stub 中可能是 external 声明 +# 或完整定义，以 text 中的定义为准，将 stub 中的同名行注释掉。 +# ============================================================ +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: + """在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'""" + if out_buf is None or global_name is None: + return 0 + name_len: t.CSizeT = string.strlen(global_name) + out_len: t.CSizeT = string.strlen(out_buf) + if name_len == 0 or out_len == 0 or name_len > out_len: + return 0 + pos: t.CSizeT = 0 + while pos + name_len <= out_len: + if string.strncmp(out_buf + pos, global_name, name_len) == 0: + # 确认是行首（前一个是换行或 pos==0） + if pos == 0 or out_buf[pos - 1] == '\n': + out_buf[pos] = ';' + return 1 + pos += 1 + return 0 + + +# ============================================================ +# _CommentOutOpaqueTypeInBuf - 将 out_buf 中匹配类型名的 opaque 声明行注释掉 +# +# 解决跨模块类型定义冲突：本地 stub 可能包含自动生成的 opaque 声明 +#（如 %"sha1.Widget" = type opaque），而依赖 stub 包含完整定义 +#（如 %"sha1.Widget" = type { i8*, i32 }）。去重时需要将 opaque 声明 +# 注释掉，让完整定义被追加，否则 LLC 报错 "Cannot allocate unsized type"。 +# ============================================================ +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: + """在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';' + + type_name_prefix 格式: '%"sha1.ClassName"'（不含 '=' 和后续内容） + 返回 1=已注释, 0=未找到 + """ + if out_buf is None or type_name_prefix is None: + return 0 + prefix_len: t.CSizeT = string.strlen(type_name_prefix) + out_len: t.CSizeT = string.strlen(out_buf) + if prefix_len == 0 or out_len == 0 or prefix_len > out_len: + return 0 + pos: t.CSizeT = 0 + while pos + prefix_len <= out_len: + if string.strncmp(out_buf + pos, type_name_prefix, prefix_len) == 0: + # 确认是行首（前一个是换行或 pos==0） + if pos == 0 or out_buf[pos - 1] == '\n': + # 检查后面是否是 " = type opaque" + p: t.CSizeT = pos + prefix_len + # 跳过空格 + while p < out_len and out_buf[p] == ' ': + p += 1 + if p < out_len and out_buf[p] == '=': + p += 1 + while p < out_len and out_buf[p] == ' ': + p += 1 + if p + 11 <= out_len: + if string.strncmp(out_buf + p, "type opaque", 11) == 0: + # 找到 opaque 声明，将行首改为 ';' + out_buf[pos] = ';' + return 1 + pos += 1 + return 0 + + +# ============================================================ +# _HasFullTypeDefinition - 检查 out_buf 中是否已有完整类型定义（非 opaque） +# +# type_name_prefix 格式: '%"sha1.ClassName"'（不含 '='） +# 完整定义: %"name" = type { ... } 或 %"name" = type < { ... } > +# 返回 1=已有完整定义, 0=无完整定义 +# ============================================================ +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: + """检查 out_buf 中是否已有 type_name_prefix 的完整定义""" + if out_buf is None or type_name_prefix is None: + return 0 + prefix_len: t.CSizeT = string.strlen(type_name_prefix) + out_len: t.CSizeT = string.strlen(out_buf) + if prefix_len == 0 or out_len == 0 or prefix_len > out_len: + return 0 + pos: t.CSizeT = 0 + while pos + prefix_len <= out_len: + if string.strncmp(out_buf + pos, type_name_prefix, prefix_len) == 0: + # 确认是行首（前一个是换行或 pos==0），且未被注释（行首不是 ';'） + is_line_start: int = 0 + if pos == 0: + is_line_start = 1 + elif out_buf[pos - 1] == '\n': + is_line_start = 1 + if is_line_start == 1: + # 检查后面是否是 " = type {" 或 " = type <" + p: t.CSizeT = pos + prefix_len + while p < out_len and out_buf[p] == ' ': + p += 1 + if p < out_len and out_buf[p] == '=': + p += 1 + while p < out_len and out_buf[p] == ' ': + p += 1 + if p + 5 < out_len: + if string.strncmp(out_buf + p, "type ", 5) == 0: + p += 5 + if p < out_len: + if out_buf[p] == '{' or out_buf[p] == '<': + return 1 + pos += 1 + return 0 + + +# ============================================================ +# _LoadAndAppendStub - 加载依赖 stub 并追加到 out_buf +# +# 读取 {temp_dir}/{sha1}.stub.ll，跳过 header（;, target, source_filename）， +# 去重 declare/global，追加到 out_buf。返回新 out_pos。 +# ============================================================ +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, + dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, + out_pos: t.CSizeT) -> t.CSizeT: + """加载依赖 stub 并追加（跳过 header，去重）""" + if temp_dir is None or dep_sha1 is None or dep_buf is None or out_buf is None: + return out_pos + + sha1_len: t.CSizeT = string.strlen(dep_sha1) + dep_path: bytes = stdlib.malloc(td_len + sha1_len + 16) + if dep_path is None: + return out_pos + viperlib.snprintf(dep_path, td_len + sha1_len + 16, "%s/%s.stub.ll", temp_dir, dep_sha1) + df_ls: fileio.File | t.CPtr = fileio.File(dep_path, fileio.MODE.R) + if df_ls.closed: + stdlib.free(dep_path) + return out_pos + dep_br_ls: t.CInt64T = df_ls.read_all(dep_buf, STUB_READ_BUF_SIZE) + df_ls.close() + stdlib.free(dep_path) + if dep_br_ls <= 0: + return out_pos + if dep_br_ls < STUB_READ_BUF_SIZE: + dep_buf[dep_br_ls] = '\0' + else: + dep_buf[STUB_READ_BUF_SIZE - 1] = '\0' + + # 注释掉依赖 stub 中的字符串常量行（模块内部的，不需要） + # 简单方案：搜索 "= external unnamed_addr constant" 并将行首的 @ 改为 ;@ + cfs_pos: t.CSizeT = 0 + while cfs_pos + 30 < t.CSizeT(dep_br_ls): + ext_p: t.CPtr = string.strstr(dep_buf + cfs_pos, "= external unnamed_addr constant") + if ext_p is None: + break + ext_off2: t.CSizeT = t.CSizeT(t.CUInt64T(ext_p) - t.CUInt64T(dep_buf)) + # 向前查找行首的 @ + ls: t.CSizeT = ext_off2 + while ls > 0: + if dep_buf[ls] == '\n': + ls += 1 + break + ls -= 1 + if ls < ext_off2 and dep_buf[ls] == '@': + dep_buf[ls] = ';' + cfs_pos = ext_off2 + 30 + + # 跳过 header，追加内容（去重 declare/global） + dep_pos: t.CSizeT = 0 + skipping: int = 1 + while dep_pos < dep_br_ls: + line_start: t.CSizeT = dep_pos + while dep_pos < dep_br_ls: + if dep_buf[dep_pos] == '\n': + break + dep_pos += 1 + line_len: t.CSizeT = dep_pos - line_start + if dep_pos < dep_br_ls: + dep_pos += 1 + if skipping != 0: + if line_len == 0: + continue + ch0: int = dep_buf[line_start] + if ch0 == ';': + continue + if ch0 == 't' and line_len >= 6: + if string.strncmp(dep_buf + line_start, "target", 6) == 0: + continue + if ch0 == 's' and line_len >= 15: + if string.strncmp(dep_buf + line_start, "source_filename", 15) == 0: + continue + skipping = 0 + # 去重：declare（用 _IsFuncDeclaredOrDefined 避免误匹配 call 语句） + if line_len >= 8 and string.strncmp(dep_buf + line_start, "declare ", 8) == 0: + at: t.CSizeT = line_start + 8 + while at < line_start + line_len: + if dep_buf[at] == '@': + break + at += 1 + if at < line_start + line_len: + lp: t.CSizeT = at + 1 + while lp < line_start + line_len: + if dep_buf[lp] == '(': + break + lp += 1 + if lp < line_start + line_len: + saved: t.CChar = dep_buf[lp + 1] + dep_buf[lp + 1] = '\0' + if _IsFuncDeclaredOrDefined(out_buf, dep_buf + at) != 0: + dep_buf[lp + 1] = saved + continue + dep_buf[lp + 1] = saved + # 去重：external global + if line_len > 0 and dep_buf[line_start] == '@': + eq: t.CSizeT = line_start + 1 + while eq < line_start + line_len: + if dep_buf[eq] == '=': + break + eq += 1 + if eq < line_start + line_len: + saved: t.CChar = dep_buf[eq + 1] + dep_buf[eq + 1] = '\0' + if string.strstr(out_buf, dep_buf + line_start) is not None: + dep_buf[eq + 1] = saved + continue + dep_buf[eq + 1] = saved + # 去重：type 定义（如 %"sha1.Point" = type {i32, i32}） + # 区分完整定义（= type { ... } / = type < ... >）和 opaque 声明（= type opaque） + # 当本地 stub 有 opaque 声明而依赖 stub 有完整定义时，注释掉 opaque 行， + # 让完整定义被追加，避免 LLC "Cannot allocate unsized type" 错误。 + if line_len > 0 and dep_buf[line_start] == '%': + eq2: t.CSizeT = line_start + 1 + while eq2 < line_start + line_len: + if dep_buf[eq2] == '=': + break + eq2 += 1 + if eq2 < line_start + line_len: + saved2: t.CChar = dep_buf[eq2 + 1] + dep_buf[eq2 + 1] = '\0' + # 检查依赖行是否为完整定义（= type { 或 = type <） + dep_is_full_def: int = 0 + if eq2 + 8 <= line_start + line_len: + if dep_buf[eq2 + 2] == 't' and dep_buf[eq2 + 3] == 'y' and dep_buf[eq2 + 4] == 'p' and dep_buf[eq2 + 5] == 'e' and dep_buf[eq2 + 6] == ' ': + if dep_buf[eq2 + 7] == '{' or dep_buf[eq2 + 7] == '<': + dep_is_full_def = 1 + if dep_is_full_def == 1: + # 依赖行是完整定义 + # 临时截断到 '=' 之前（不含 '='），辅助函数期望纯类型名前缀 + saved_eq: t.CChar = dep_buf[eq2] + dep_buf[eq2] = '\0' + # 1. 注释掉 out_buf 中的 opaque 声明（如果有） + commented: int = _CommentOutOpaqueTypeInBuf(out_buf, dep_buf + line_start) + # 2. 检查 out_buf 中是否已有完整定义（避免重复） + has_full: int = _HasFullTypeDefinition(out_buf, dep_buf + line_start) + dep_buf[eq2] = saved_eq + if has_full == 1: + dep_buf[eq2 + 1] = saved2 + continue + # 3. 追加完整定义 + else: + # 依赖行是 opaque 声明，用原有子串匹配去重 + found_opaque: t.CPtr = string.strstr(out_buf, dep_buf + line_start) + if found_opaque is not None: + dep_buf[eq2 + 1] = saved2 + continue + dep_buf[eq2 + 1] = saved2 + # 追加 + if out_pos + line_len + 2 < out_size: + string.strncpy(out_buf + out_pos, dep_buf + line_start, line_len) + out_pos += line_len + out_buf[out_pos] = '\n' + out_pos += 1 + out_buf[out_pos] = '\0' + return out_pos + + +# ============================================================ +# _LoadAndAppendTextDeclares - 从依赖 text.ll 提取 declare 追加到 out_buf +# +# 读取 {temp_dir}/{sha1}.text.ll，将 define 行转为 declare（提取函数签名）， +# 跳过函数体，去重后追加到 out_buf。返回新 out_pos。 +# ============================================================ +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, + dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, + out_pos: t.CSizeT) -> t.CSizeT: + """从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf""" + if temp_dir is None or dep_sha1 is None or dep_buf is None or out_buf is None: + return out_pos + + sha1_len_lt: t.CSizeT = string.strlen(dep_sha1) + text_path: bytes = stdlib.malloc(td_len + sha1_len_lt + 16) + if text_path is None: + return out_pos + viperlib.snprintf(text_path, td_len + sha1_len_lt + 16, "%s/%s.text.ll", temp_dir, dep_sha1) + tf_lt: fileio.File | t.CPtr = fileio.File(text_path, fileio.MODE.R) + if tf_lt.closed: + stdlib.free(text_path) + return out_pos + text_br: t.CInt64T = tf_lt.read_all(dep_buf, STUB_READ_BUF_SIZE) + tf_lt.close() + stdlib.free(text_path) + if text_br <= 0: + return out_pos + if text_br < STUB_READ_BUF_SIZE: + dep_buf[text_br] = '\0' + else: + dep_buf[STUB_READ_BUF_SIZE - 1] = '\0' + + dep_pos_lt: t.CSizeT = 0 + in_body: int = 0 + while dep_pos_lt < text_br: + ls: t.CSizeT = dep_pos_lt + while dep_pos_lt < text_br: + if dep_buf[dep_pos_lt] == '\n': + break + dep_pos_lt += 1 + ll: t.CSizeT = dep_pos_lt - ls + if dep_pos_lt < text_br: + dep_pos_lt += 1 + + # 函数体跳过：直到单独的 } 行 + if in_body != 0: + if ll == 1 and dep_buf[ls] == '}': + in_body = 0 + continue + + # 跳过空行和 header + if ll == 0: + continue + c0: int = dep_buf[ls] + if c0 == ';': + continue + if c0 == 't' and ll >= 6 and string.strncmp(dep_buf + ls, "target", 6) == 0: + continue + if c0 == 's' and ll >= 15 and string.strncmp(dep_buf + ls, "source_filename", 15) == 0: + continue + + # 处理 define 行：转为 declare + if ll >= 7 and string.strncmp(dep_buf + ls, "define ", 7) == 0: + # 提取 @funcname( 用于去重 + at_lt: t.CSizeT = ls + 7 + while at_lt < ls + ll: + if dep_buf[at_lt] == '@': + break + at_lt += 1 + if at_lt >= ls + ll: + in_body = 1 + continue + lp_lt: t.CSizeT = at_lt + 1 + while lp_lt < ls + ll: + if dep_buf[lp_lt] == '(': + break + lp_lt += 1 + if lp_lt >= ls + ll: + in_body = 1 + continue + # 去重检查：用 _IsFuncDeclaredOrDefined 避免误匹配 call 语句 + saved_lt: t.CChar = dep_buf[lp_lt + 1] + dep_buf[lp_lt + 1] = '\0' + if _IsFuncDeclaredOrDefined(out_buf, dep_buf + at_lt) != 0: + dep_buf[lp_lt + 1] = saved_lt + in_body = 1 + continue + dep_buf[lp_lt + 1] = saved_lt + + # 找行尾的 ' {' 并截断 + decl_end: t.CSizeT = ls + ll + if decl_end > ls + 1: + if dep_buf[decl_end - 1] == '{' and dep_buf[decl_end - 2] == ' ': + decl_end -= 2 + + # 输出: declare + content (skip "define " 7 chars, up to decl_end) + copy_len_lt: t.CSizeT = decl_end - (ls + 7) + if out_pos + 8 + copy_len_lt + 2 < out_size: + string.strncpy(out_buf + out_pos, "declare ", 8) + out_pos += 8 + string.strncpy(out_buf + out_pos, dep_buf + ls + 7, copy_len_lt) + out_pos += copy_len_lt + out_buf[out_pos] = '\n' + out_pos += 1 + out_buf[out_pos] = '\0' + in_body = 1 + continue + + # 跳过其他行（global 带初值定义、type 定义等，stub 已有 external 声明） + return out_pos + + +# ============================================================ +# _PathToModuleName - 将文件路径转换为模块名 +# +# 输入: "includes/stdio.py" 或 "includes/lib/core/Handles/HandlesType.py" +# 输出: "stdio" 或 "lib.core.Handles.HandlesType" +# 去除 includes/ 前缀和 .py 后缀，/ 替换为 . +# 返回 stdlib.malloc 分配的字符串，调用者负责释放 +# ============================================================ +def _PathToModuleName(path: str) -> str: + """将文件路径转换为模块名 + + 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__） + """ + if path is None: + return None + plen: t.CSizeT = string.strlen(path) + # 跳过 includes/ 前缀 + prefix: str = "includes/" + prefix_len: t.CSizeT = 9 + path_start: t.CSizeT = 0 + if plen > prefix_len and string.strncmp(path, prefix, prefix_len) == 0: + path_start = prefix_len + plen = plen - prefix_len + # 去掉 .py 后缀 + if plen > 3 and path[path_start + plen - 3] == '.' and path[path_start + plen - 2] == 'p' and path[path_start + plen - 1] == 'y': + plen -= 3 + if plen == 0: + return None + # 检查是否以 /__init__ 或 \__init__ 结尾（包的 __init__.py） + # /__init__ 是 9 个字符：/ _ _ i n i t _ _ + # 如果是，去掉分隔符+__init__ 后缀，模块名就是包名（如 ast/__init__ → ast） + # 注意：顶级 __init__.py（plen=8）不会进入此分支，因为 plen > 9 才检查 + # 同时支持 / 和 \ 两种分隔符（TPC 写入的路径可能混合使用） + if plen > 9: + tail_off: t.CSizeT = path_start + plen - 9 + sep_char: t.CChar = path[tail_off] + if (sep_char == '/' or sep_char == '\\') \ + and path[tail_off + 1] == '_' and path[tail_off + 2] == '_' \ + and path[tail_off + 3] == 'i' and path[tail_off + 4] == 'n' and path[tail_off + 5] == 'i' \ + and path[tail_off + 6] == 't' and path[tail_off + 7] == '_' and path[tail_off + 8] == '_': + # 去掉 分隔符+__init__ 后缀，plen 就是包名部分长度 + plen = tail_off - path_start + if plen == 0: + return None + # 复制并替换 / 为 . + result: str = stdlib.malloc(plen + 1) + if result is None: + return None + for ri in range(plen): + ch: t.CChar = path[path_start + ri] + if ch == '/' or ch == '\\': + result[ri] = '.' + else: + result[ri] = ch + result[plen] = '\0' + return result + + +# ============================================================ +# _BuildIncludesSha1Map - 从 _sha1_map.txt 构建 sha1 → module_name 映射 +# +# _sha1_map.txt 格式: {sha1}:includes/{rel_path}\n +# 返回两个数组（sha1_list 和 mod_list），长度存储在 count 中。 +# 所有内存由 stdlib.malloc 分配，调用者负责释放。 +# ============================================================ +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, + sha1_list: t.CChar | t.CPtr, + mod_list: t.CChar | t.CPtr) -> int: + """从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数""" + if temp_dir is None or sha1_list is None or mod_list is None: + return 0 + + map_path: bytes = stdlib.malloc(td_len + 32) + if map_path is None: + return 0 + viperlib.snprintf(map_path, td_len + 32, "%s/_sha1_map.txt", temp_dir) + mf: fileio.File | t.CPtr = fileio.File(map_path, fileio.MODE.R) + if mf.closed: + stdlib.free(map_path) + return 0 + map_buf: bytes = stdlib.malloc(STUB_READ_BUF_SIZE) + if map_buf is None: + mf.close() + stdlib.free(map_path) + return 0 + map_br: t.CInt64T = mf.read_all(map_buf, STUB_READ_BUF_SIZE) + mf.close() + stdlib.free(map_path) + if map_br <= 0: + stdlib.free(map_buf) + return 0 + if map_br < STUB_READ_BUF_SIZE: + map_buf[map_br] = '\0' + else: + map_buf[STUB_READ_BUF_SIZE - 1] = '\0' + + count: int = 0 + pos: t.CSizeT = 0 + while pos < map_br and count < MAX_INCLUDES: + # 提取 SHA1（16 hex） + sha1: t.CChar | t.CPtr = stdlib.malloc(17) + if sha1 is None: + break + string.strncpy(sha1, map_buf + pos, 16) + sha1[16] = '\0' + pos += 16 + # 跳过 ':' + if pos < map_br and map_buf[pos] == ':': + pos += 1 + # 提取路径（到 \r、\n 或 \0） + # 注意: Windows CRLF 换行符是 \r\n，必须同时检查 \r 避免路径包含 \r + path_start: t.CSizeT = pos + while pos < map_br and map_buf[pos] != '\n' and map_buf[pos] != '\r' and map_buf[pos] != '\0': + pos += 1 + path_len: t.CSizeT = pos - path_start + # 跳过 \r\n 或 \n + if pos < map_br and map_buf[pos] == '\r': + pos += 1 + if pos < map_br and map_buf[pos] == '\n': + pos += 1 + if path_len > 0: + # 仅收集 includes/ 前缀的条目（用户文件如 App/ 不应进入 includes 映射） + # 否则用户文件会被误判为 includes，导致 is_module_imported 模块名不匹配而跳过加载 + if path_len >= 9 and string.strncmp(map_buf + path_start, "includes/", 9) == 0: + path_buf: t.CChar | t.CPtr = stdlib.malloc(path_len + 1) + if path_buf is not None: + string.strncpy(path_buf, map_buf + path_start, path_len) + path_buf[path_len] = '\0' + mod_name: str = _PathToModuleName(path_buf) + stdlib.free(path_buf) + if mod_name is not None: + # 存储 sha1 和 mod_name 到数组中 + idx: t.CSizeT = t.CSizeT(count) * 17 + string.strcpy(sha1_list + idx, sha1) + idx2: t.CSizeT = t.CSizeT(count) * 64 + mn_len: t.CSizeT = string.strlen(mod_name) + if mn_len < 64: + string.strcpy(mod_list + idx2, mod_name) + else: + string.strncpy(mod_list + idx2, mod_name, 63) + mod_list[idx2 + 63] = '\0' + stdlib.free(mod_name) + count += 1 + stdlib.free(sha1) + stdlib.free(map_buf) + return count + + +# ============================================================ +# _FindSha1ByModName - 从模块名查找 SHA1（精确匹配） +# +# 在 sha1_arr/mod_arr 映射中查找模块名对应的 SHA1。 +# 用于依赖图分析：模块名 → SHA1。 +# +# Args: +# sha1_arr: SHA1 数组（每个 17 字节） +# mod_arr: 模块名数组（每个 64 字节） +# count: 映射条目数 +# mod_name: 要查找的模块名 +# +# Returns: +# SHA1 字符串指针（指向 sha1_arr 内部，无需释放），未找到返回 None +# ============================================================ +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, + count: int, mod_name: str) -> str: + """从模块名查找 SHA1（精确匹配）""" + if sha1_arr is None or mod_arr is None or mod_name is None: + return None + if count <= 0: + return None + for i in range(count): + idx: t.CSizeT = t.CSizeT(i) * 64 + if string.strcmp(mod_arr + idx, mod_name) == 0: + sidx: t.CSizeT = t.CSizeT(i) * 17 + return sha1_arr + sidx + return None + + +# ============================================================ +# _AddSha1ToSet - 将 SHA1 加入集合（去重） +# +# 检查 SHA1 是否已在集合中，若不在则加入。返回新的 count。 +# +# Args: +# set_buf: SHA1 集合缓冲区（每个 17 字节） +# count: 当前集合中的 SHA1 数量 +# sha1: 要加入的 SHA1 +# +# Returns: +# 新的 count（如果已存在则不变） +# ============================================================ +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: + """将 SHA1 加入集合，返回新 count""" + if set_buf is None or sha1 is None: + return count + for i in range(count): + idx: t.CSizeT = t.CSizeT(i) * 17 + if string.strcmp(set_buf + idx, sha1) == 0: + return count + if count < MAX_INCLUDES_SHA1: + idx: t.CSizeT = t.CSizeT(count) * 17 + string.strcpy(set_buf + idx, sha1) + return count + 1 + return count + + +# ============================================================ +# _BuildReachableSha1Set - 构建可达 SHA1 集合（依赖图按需翻译） +# +# 从 source_dir 的源文件开始，解析 import 语句，递归收集可达的 +# includes 文件 SHA1。Phase 1b 只翻译可达集合中的文件，避免 +# 翻译不需要的 includes（如 Test 不依赖 llvmlite，则不翻译）。 +# +# 算法（工作列表）： +# 1. 扫描 source_dir 下的 .py 文件，解析 AST 获取 _imported_modules +# 2. 构建 SHA1→module_name 映射（从 _sha1_map.txt） +# 3. 工作列表递归：模块名 → 查 SHA1 → 加入 reachable_set → 读 .deps.txt → 追加新模块名 +# +# Args: +# mb: 内存池 +# source_dir: 源文件目录（Config.SourceDir） +# temp_dir: 临时目录（_sha1_map.txt 和 .deps.txt 所在位置） +# reachable_set: 输出参数，可达 SHA1 集合缓冲区（MAX_INCLUDES_SHA1 * 17 字节） +# +# Returns: +# 可达 SHA1 数量（>0 成功，<=0 失败） +# ============================================================ +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, + temp_dir: str, + reachable_set: t.CChar | t.CPtr) -> int: + """构建可达 SHA1 集合（依赖图按需翻译）""" + if source_dir is None or temp_dir is None or reachable_set is None: + return -1 + + SRC_BUF_SIZE_R: t.CSizeT = 1048576 + + # 1. 构建 SHA1→module_name 映射 + td_len_r: t.CSizeT = string.strlen(temp_dir) + sha1_arr: bytes = stdlib.malloc(MAX_INCLUDES * 17) + mod_arr: bytes = stdlib.malloc(MAX_INCLUDES * 64) + if sha1_arr is None or mod_arr is None: + if sha1_arr is not None: + stdlib.free(sha1_arr) + if mod_arr is not None: + stdlib.free(mod_arr) + return -1 + map_count: int = _BuildIncludesSha1Map(temp_dir, td_len_r, sha1_arr, mod_arr) + if map_count <= 0: + stdio.printf("[Reachable] 无法构建 SHA1 映射\n") + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + return -1 + stdio.printf("[Reachable] SHA1 映射: %d 个\n", map_count) + + # 2. 扫描 source_dir 下的 .py 文件，收集直接依赖 + dir_len: t.CSizeT = string.strlen(source_dir) + pattern: bytes = stdlib.malloc(dir_len + 8) + if pattern is None: + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + return -1 + viperlib.snprintf(pattern, dir_len + 8, "%s/*.py", source_dir) + + find_data_size: t.CSizeT = win32file.WIN32_FIND_DATAA.__sizeof__() + find_data: win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(find_data_size + 16) + if find_data is None: + stdlib.free(pattern) + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + return -1 + string.memset(find_data, 0, find_data_size + 16) + + # 工作列表（空格分隔的模块名） + worklist: bytes = stdlib.malloc(8192) + if worklist is None: + stdlib.free(pattern) + stdlib.free(find_data) + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + return -1 + worklist[0] = '\0' + wl_len: t.CSizeT = 0 + + reachable_count: int = 0 + + handle: win32base.HANDLE = win32file.FindFirstFileA(pattern, find_data) + if handle == win32base.INVALID_HANDLE_VALUE: + stdio.printf("[Reachable] 未找到 .py 文件: %s\n", pattern) + stdlib.free(pattern) + stdlib.free(find_data) + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + stdlib.free(worklist) + return -1 + + while True: + fname: str = find_data.cFileName + if fname is not None: + fname_len: t.CSizeT = string.strlen(fname) + if fname_len > 3: + is_py: int = 0 + if fname[fname_len - 3] == '.' and fname[fname_len - 2] == 'p' and fname[fname_len - 1] == 'y': + is_py = 1 + if is_py != 0: + full_path: bytes = stdlib.malloc(dir_len + fname_len + 2) + if full_path is not None: + viperlib.snprintf(full_path, dir_len + fname_len + 2, "%s/%s", source_dir, fname) + sf: fileio.File | t.CPtr = fileio.File(full_path, fileio.MODE.R) + if not sf.closed: + sbuf: bytes = stdlib.malloc(SRC_BUF_SIZE_R) + if sbuf is not None: + br: LONG = sf.read_all(sbuf, SRC_BUF_SIZE_R) + sf.close() + if br > 0: + if br < SRC_BUF_SIZE_R: + sbuf[br] = 0 + else: + sbuf[SRC_BUF_SIZE_R - 1] = 0 + # 解析 AST 获取 _imported_modules + lx: ast.Lexer | t.CPtr = ast.new_lexer(mb) + if lx is not None: + ast._lexer_init(lx, sbuf, mb) + tokens: ast.Token | t.CPtr = ast.tokenize(lx) + tree: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens) + if tree is not None: + tr: HandlesTranslator.Translator | t.CPtr = HandlesTranslator.Translator() + if tr is not None: + tr._declare_only = 2 + # 源文件包名：fname 是顶级文件名（无目录分隔符），包为 None + tr.CurrentPackage = HandlesImports.compute_package_from_relpath(mb, fname) + HandlesStruct.reset_visible_structs(mb, 0) + tr.translate(tree) + # 获取 _imported_modules + if tr._imported_modules is not None: + im_len: t.CSizeT = string.strlen(tr._imported_modules) + if im_len > 0 and wl_len + im_len + 1 < 8192: + string.strcpy(worklist + wl_len, tr._imported_modules) + wl_len += im_len + worklist[wl_len] = ' ' + wl_len += 1 + worklist[wl_len] = '\0' + # 释放 Translator 资源 + if tr._global_names is not None: + stdlib.free(tr._global_names) + if tr._nonlocal_names is not None: + stdlib.free(tr._nonlocal_names) + stdlib.free(sbuf) + else: + stdlib.free(sbuf) + else: + sf.close() + stdlib.free(full_path) + + if win32file.FindNextFileA(handle, find_data) == 0: + break + win32base.FindClose(handle) + stdlib.free(pattern) + stdlib.free(find_data) + + # 去掉末尾多余空格 + if wl_len > 0 and worklist[wl_len - 1] == ' ': + worklist[wl_len - 1] = '\0' + wl_len -= 1 + + stdio.printf("[Reachable] 源文件直接依赖: '%s'\n", worklist) + + # 2.5. 全局预加入基础容器模块（_list/_dict/json） + # + # 这些模块定义了 list/dict 等内建容器的泛型模板。 + # 使用 list[...]/dict[...] 语法时需要这些模块被翻译并注册模板。 + # 即使源文件和 includes 文件没有显式 import _list，也预加入以确保可达。 + # 这对应 TPC 的 _inject_auto_imports 机制（TPC 在翻译前自动注入 import _list）。 + # _list + if HandlesImports.is_module_imported(worklist, "_list") == 0: + bsha1_l: str = _FindSha1ByModName(sha1_arr, mod_arr, map_count, "_list") + if bsha1_l is not None: + if wl_len + 7 < 8192: + if wl_len > 0: + worklist[wl_len] = ' ' + wl_len += 1 + string.strcpy(worklist + wl_len, "_list") + wl_len += 5 + worklist[wl_len] = ' ' + wl_len += 1 + worklist[wl_len] = '\0' + # _dict + if HandlesImports.is_module_imported(worklist, "_dict") == 0: + bsha1_d: str = _FindSha1ByModName(sha1_arr, mod_arr, map_count, "_dict") + if bsha1_d is not None: + if wl_len + 7 < 8192: + if wl_len > 0: + worklist[wl_len] = ' ' + wl_len += 1 + string.strcpy(worklist + wl_len, "_dict") + wl_len += 5 + worklist[wl_len] = ' ' + wl_len += 1 + worklist[wl_len] = '\0' + # json + if HandlesImports.is_module_imported(worklist, "json") == 0: + bsha1_j: str = _FindSha1ByModName(sha1_arr, mod_arr, map_count, "json") + if bsha1_j is not None: + if wl_len + 6 < 8192: + if wl_len > 0: + worklist[wl_len] = ' ' + wl_len += 1 + string.strcpy(worklist + wl_len, "json") + wl_len += 4 + worklist[wl_len] = ' ' + wl_len += 1 + worklist[wl_len] = '\0' + stdio.printf("[Reachable] 预加入容器模块后: '%s'\n", worklist) + + # 3. 工作列表算法：递归收集可达 SHA1 + processed: bytes = stdlib.malloc(8192) + if processed is None: + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + stdlib.free(worklist) + return -1 + processed[0] = '\0' + proc_len: t.CSizeT = 0 + + wl_pos: t.CSizeT = 0 + wl_iter: int = 0 + while wl_pos < wl_len: + wl_iter += 1 + # 提取一个模块名（到空格或末尾） + name_start: t.CSizeT = wl_pos + while wl_pos < wl_len and worklist[wl_pos] != ' ': + wl_pos += 1 + name_len: t.CSizeT = wl_pos - name_start + if wl_pos < wl_len: + wl_pos += 1 + if name_len == 0: + continue + + # 复制模块名 + mod_buf: bytes = stdlib.malloc(name_len + 1) + if mod_buf is None: + continue + string.strncpy(mod_buf, worklist + name_start, name_len) + mod_buf[name_len] = '\0' + + # 检查是否已处理 + already: int = 0 + if HandlesImports.is_module_imported(processed, mod_buf) != 0: + already = 1 + if already == 0: + # 加入 processed + if proc_len + name_len + 1 < 8192: + string.strcpy(processed + proc_len, mod_buf) + proc_len += name_len + processed[proc_len] = ' ' + proc_len += 1 + processed[proc_len] = '\0' + + # 查找 SHA1 + found_sha1: str = _FindSha1ByModName(sha1_arr, mod_arr, map_count, mod_buf) + if found_sha1 is not None: + # 加入 reachable_set + reachable_count = _AddSha1ToSet(reachable_set, reachable_count, found_sha1) + + # 读取 .deps.txt 追加到 worklist + deps_path: bytes = stdlib.malloc(td_len_r + 33) + if deps_path is not None: + viperlib.snprintf(deps_path, td_len_r + 33, "%s/%s.deps.txt", temp_dir, found_sha1) + df: fileio.File | t.CPtr = fileio.File(deps_path, fileio.MODE.R) + if not df.closed: + deps_buf: bytes = stdlib.malloc(2048) + if deps_buf is not None: + dbr: t.CInt64T = df.read_all(deps_buf, 2048) + df.close() + if dbr > 0: + if dbr < 2048: + deps_buf[dbr] = '\0' + else: + deps_buf[2047] = '\0' + # 追加到 worklist（确保前面有空格分隔符，避免模块名合并） + dl: t.CSizeT = string.strlen(deps_buf) + if dl > 0 and wl_len + dl + 2 < 8192: + if wl_len > 0 and worklist[wl_len - 1] != ' ': + worklist[wl_len] = ' ' + wl_len += 1 + string.strcpy(worklist + wl_len, deps_buf) + wl_len += dl + worklist[wl_len] = ' ' + wl_len += 1 + worklist[wl_len] = '\0' + stdlib.free(deps_buf) + else: + df.close() + stdlib.free(deps_path) + stdlib.free(mod_buf) + stdlib.free(sha1_arr) + stdlib.free(mod_arr) + stdlib.free(worklist) + stdlib.free(processed) + + stdio.printf("[Reachable] 可达 SHA1: %d 个\n", reachable_count) + return reachable_count + + +# ============================================================ +# BuildCombinedIR - 组合本地 stub + 依赖 stubs + 本地 text → 完整 IR +# +# 按需加载：读取 deps.txt 获取导入模块名，通过 _sha1_map.txt 查找 SHA1， +# 仅加载实际使用的依赖 stub（而非全部扫描）。 +# ============================================================ +def BuildCombinedIR(temp_dir: str, local_sha1: str, + out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: + """组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）""" + if temp_dir is None or local_sha1 is None or out_buf is None or out_size == 0: + return 0 + + stdio.printf("[BuildCombinedIR] start: sha1=%s\n", local_sha1) + td_len: t.CSizeT = string.strlen(temp_dir) + out_buf[0] = '\0' + out_pos: t.CSizeT = 0 + + # 1. 读取并追加本地 stub.ll（含 header） + stub_path: bytes = stdlib.malloc(td_len + 32) + if stub_path is None: + return 0 + viperlib.snprintf(stub_path, td_len + 32, "%s/%s.stub.ll", temp_dir, local_sha1) + sf: fileio.File | t.CPtr = fileio.File(stub_path, fileio.MODE.R) + if sf.closed: + stdlib.free(stub_path) + return 0 + stub_content: bytes = stdlib.malloc(STUB_READ_BUF_SIZE) + if stub_content is None: + sf.close() + stdlib.free(stub_path) + return 0 + stub_br: t.CInt64T = sf.read_all(stub_content, STUB_READ_BUF_SIZE) + sf.close() + stdlib.free(stub_path) + stdio.printf("[BuildCombinedIR] stub read: %d bytes\n", stub_br) + if stub_br <= 0: + stdlib.free(stub_content) + return 0 + if stub_br < STUB_READ_BUF_SIZE: + stub_content[stub_br] = '\0' + else: + stub_content[STUB_READ_BUF_SIZE - 1] = '\0' + slen: t.CSizeT = stub_br + # 修复本地 stub.ll 中字符串常量的非法 IR： + # llvmlite OUTPUT_STUB 模式输出 "= external unnamed_addr constant" 的非法形式。 + # 简单方案：搜索 "= external unnamed_addr constant" 并将 external 替换为 8 个空格。 + fix_pos: t.CSizeT = 0 + while fix_pos + 30 < slen: + ext_ptr: t.CPtr = string.strstr(stub_content + fix_pos, "= external unnamed_addr constant") + if ext_ptr is None: + break + # ext_ptr 指向 "= external ..." 中的 '=' + # 'external' 开始于 ext_ptr + 2 + ext_off: t.CSizeT = t.CSizeT(t.CUInt64T(ext_ptr) - t.CUInt64T(stub_content)) + 2 + if ext_off + 8 <= slen: + stub_content[ext_off] = ' ' + stub_content[ext_off + 1] = ' ' + stub_content[ext_off + 2] = ' ' + stub_content[ext_off + 3] = ' ' + stub_content[ext_off + 4] = ' ' + stub_content[ext_off + 5] = ' ' + stub_content[ext_off + 6] = ' ' + stub_content[ext_off + 7] = ' ' + fix_pos = ext_off + 8 + stdio.printf("[BuildCombinedIR] stub_fix done, slen=%d\n", slen) + # 直接复制修复后的 stub 内容 + if out_pos + slen + 2 < out_size: + string.strcpy(out_buf + out_pos, stub_content) + out_pos += slen + if out_buf[out_pos - 1] != '\n': + out_buf[out_pos] = '\n' + out_pos += 1 + out_buf[out_pos] = '\0' + stdlib.free(stub_content) + + # 2. 按需加载依赖 stubs（根据 deps.txt 过滤，而非扫描全部） + # 先加载依赖 stub（含 type 定义），确保 type 定义在本地 text.ll 的 define 块之前 + # 2a. 构建 includes SHA1 → module_name 映射（从 _sha1_map.txt） + sha1_arr: bytes = stdlib.malloc(MAX_INCLUDES * 17) + mod_arr: bytes = stdlib.malloc(MAX_INCLUDES * 64) + inc_count: int = 0 + if sha1_arr is not None and mod_arr is not None: + inc_count = _BuildIncludesSha1Map(temp_dir, td_len, sha1_arr, mod_arr) + + # 2b. 读取 deps.txt 获取导入模块名集合 + deps_buf: str = None + deps_loaded: int = 0 + deps_path: bytes = stdlib.malloc(td_len + 32) + if deps_path is not None: + viperlib.snprintf(deps_path, td_len + 32, "%s/%s.deps.txt", temp_dir, local_sha1) + df_deps: fileio.File | t.CPtr = fileio.File(deps_path, fileio.MODE.R) + if not df_deps.closed: + deps_buf = stdlib.malloc(4096) + if deps_buf is not None: + deps_br: t.CInt64T = df_deps.read_all(deps_buf, 4096) + if deps_br > 0: + if deps_br < 4096: + deps_buf[deps_br] = '\0' + else: + deps_buf[4095] = '\0' + deps_loaded = 1 + df_deps.close() + stdlib.free(deps_path) + + # 2c. 扫描 temp_dir 中所有 .stub.ll，按需加载 + pattern: bytes = stdlib.malloc(td_len + 16) + if pattern is not None: + viperlib.snprintf(pattern, td_len + 16, "%s/*.stub.ll", temp_dir) + find_data: win32file.WIN32_FIND_DATAA | t.CPtr = stdlib.malloc(win32file.WIN32_FIND_DATAA.__sizeof__()) + if find_data is not None: + string.memset(find_data, 0, win32file.WIN32_FIND_DATAA.__sizeof__()) + handle: win32base.HANDLE = win32file.FindFirstFileA(pattern, find_data) + if handle != win32base.INVALID_HANDLE_VALUE: + dep_buf: bytes = stdlib.malloc(STUB_READ_BUF_SIZE) + if dep_buf is not None: + while 1: + fname: str = find_data.cFileName + if fname is not None: + dep_sha1: str = stdlib.malloc(17) + if dep_sha1 is not None: + string.strncpy(dep_sha1, fname, 16) + dep_sha1[16] = '\0' + if string.strcmp(dep_sha1, local_sha1) != 0: + should_load: int = 0 + # 检查是否在 includes 映射中 + is_include: int = 0 + if inc_count > 0: + for ii in range(inc_count): + idx_ii: t.CSizeT = t.CSizeT(ii) * 17 + if string.strcmp(sha1_arr + idx_ii, dep_sha1) == 0: + is_include = 1 + if deps_loaded != 0: + idx_mi: t.CSizeT = t.CSizeT(ii) * 64 + if HandlesImports.is_module_imported(deps_buf, mod_arr + idx_mi) != 0: + should_load = 1 + break + if is_include == 0: + # 非 includes stub（用户文件），总是加载 + should_load = 1 + if should_load != 0: + out_pos = _LoadAndAppendStub(temp_dir, td_len, dep_sha1, dep_buf, out_buf, out_size, out_pos) + out_pos = _LoadAndAppendTextDeclares(temp_dir, td_len, dep_sha1, dep_buf, out_buf, out_size, out_pos) + stdlib.free(dep_sha1) + if win32file.FindNextFileA(handle, find_data) == 0: + break + if dep_buf is not None: + stdlib.free(dep_buf) + win32file.FindClose(handle) + stdlib.free(find_data) + stdlib.free(pattern) + + # 2d. 验证 deps.txt 中所有依赖模块的 stub 文件都存在（fail-fast） + # 避免到 llc 才报 undefined value 错误。 + # 仅验证在 includes 映射中找到的模块（t/c 等内部库无 SHA1 映射，自动跳过）。 + if deps_loaded != 0 and inc_count > 0: + dep_total: t.CSizeT = string.strlen(deps_buf) + dp: t.CSizeT = 0 + while dp < dep_total: + # 跳过前导空格 + while dp < dep_total: + if deps_buf[dp] != ' ': + break + dp += 1 + if dp >= dep_total: + break + # 提取模块名（到下一个空格或末尾） + ms: t.CSizeT = dp + while dp < dep_total: + if deps_buf[dp] == ' ': + break + dp += 1 + ml: t.CSizeT = dp - ms + if ml == 0: + continue + # 复制模块名到临时缓冲区并 null 终止 + mod_nm: str = stdlib.malloc(ml + 1) + if mod_nm is None: + continue + string.strncpy(mod_nm, deps_buf + ms, ml) + mod_nm[ml] = '\0' + # 在 includes 映射中查找模块对应的 SHA1 + found_s: str = None + for mi2 in range(inc_count): + idx_m2: t.CSizeT = t.CSizeT(mi2) * 64 + if string.strcmp(mod_arr + idx_m2, mod_nm) == 0: + idx_s2: t.CSizeT = t.CSizeT(mi2) * 17 + found_s = sha1_arr + idx_s2 + break + if found_s is not None: + # 检查 stub 文件是否存在 + chk_path: bytes = stdlib.malloc(td_len + 32) + if chk_path is not None: + viperlib.snprintf(chk_path, td_len + 32, "%s/%s.stub.ll", temp_dir, found_s) + chk_f: fileio.File | t.CPtr = fileio.File(chk_path, fileio.MODE.R) + if chk_f.closed: + stdio.printf("[FATAL][BuildCombinedIR] 依赖模块 '%s' (sha1=%s) 的 stub 文件不存在: %s，立即终止编译\n", mod_nm, found_s, chk_path) + sys.exit(1) + chk_f.close() + stdlib.free(chk_path) + stdlib.free(mod_nm) + + # 3. 读取并追加本地 text.ll（行级去重，跳过 stub.ll 中已存在的定义） + text_path: bytes = stdlib.malloc(td_len + 32) + if text_path is None: + return out_pos + viperlib.snprintf(text_path, td_len + 32, "%s/%s.text.ll", temp_dir, local_sha1) + tf: fileio.File | t.CPtr = fileio.File(text_path, fileio.MODE.R) + if tf.closed: + stdlib.free(text_path) + # text.ll 不存在的情况，继续 + else: + text_content: bytes = stdlib.malloc(STUB_READ_BUF_SIZE) + if text_content is None: + tf.close() + stdlib.free(text_path) + else: + text_br: t.CInt64T = tf.read_all(text_content, STUB_READ_BUF_SIZE) + tf.close() + stdlib.free(text_path) + if text_br > 0: + if text_br < STUB_READ_BUF_SIZE: + text_content[text_br] = '\0' + else: + text_content[STUB_READ_BUF_SIZE - 1] = '\0' + # 行级去重追加（与 _LoadAndAppendStub 相同逻辑） + tx_pos: t.CSizeT = 0 + tx_skip: int = 1 + while tx_pos < text_br: + tx_ls: t.CSizeT = tx_pos + while tx_pos < text_br: + if text_content[tx_pos] == '\n': + break + tx_pos += 1 + tx_ll: t.CSizeT = tx_pos - tx_ls + if tx_pos < text_br: + tx_pos += 1 + if tx_skip != 0: + if tx_ll == 0: + continue + tx_c0: int = text_content[tx_ls] + if tx_c0 == ';': + continue + if tx_c0 == 't' and tx_ll >= 6: + if string.strncmp(text_content + tx_ls, "target", 6) == 0: + continue + if tx_c0 == 's' and tx_ll >= 15: + if string.strncmp(text_content + tx_ls, "source_filename", 15) == 0: + continue + tx_skip = 0 + # 去重：declare（用 _IsFuncDeclaredOrDefined 避免误匹配 call 语句） + if tx_ll >= 8 and string.strncmp(text_content + tx_ls, "declare ", 8) == 0: + tx_at: t.CSizeT = tx_ls + 8 + while tx_at < tx_ls + tx_ll: + if text_content[tx_at] == '@': + break + tx_at += 1 + if tx_at < tx_ls + tx_ll: + tx_lp: t.CSizeT = tx_at + 1 + while tx_lp < tx_ls + tx_ll: + if text_content[tx_lp] == '(': + break + tx_lp += 1 + if tx_lp < tx_ls + tx_ll: + tx_sv: t.CChar = text_content[tx_lp + 1] + text_content[tx_lp + 1] = '\0' + if _IsFuncDeclaredOrDefined(out_buf, text_content + tx_at) != 0: + text_content[tx_lp + 1] = tx_sv + continue + text_content[tx_lp + 1] = tx_sv + # 去重：define（注释掉 out_buf 中已有的 declare，避免 declare/define 冲突） + if tx_ll >= 7 and string.strncmp(text_content + tx_ls, "define ", 7) == 0: + tx_at_d: t.CSizeT = tx_ls + 7 + while tx_at_d < tx_ls + tx_ll: + if text_content[tx_at_d] == '@': + break + tx_at_d += 1 + if tx_at_d < tx_ls + tx_ll: + tx_lp_d: t.CSizeT = tx_at_d + 1 + while tx_lp_d < tx_ls + tx_ll: + if text_content[tx_lp_d] == '(': + break + tx_lp_d += 1 + if tx_lp_d < tx_ls + tx_ll: + tx_sv_d: t.CChar = text_content[tx_lp_d + 1] + text_content[tx_lp_d + 1] = '\0' + _CommentOutDeclareInBuf(out_buf, text_content + tx_at_d) + text_content[tx_lp_d + 1] = tx_sv_d + # 处理：@global = ... + # stub 中可能是 external 声明或完整定义，以 text 中的定义为准 + # 找到 stub 中同名全局变量行并注释掉，然后追加 text 中的 + if tx_ll > 0 and text_content[tx_ls] == '@': + tx_eq: t.CSizeT = tx_ls + 1 + while tx_eq < tx_ls + tx_ll: + if text_content[tx_eq] == '=': + break + tx_eq += 1 + if tx_eq < tx_ls + tx_ll: + tx_sv2: t.CChar = text_content[tx_eq + 1] + text_content[tx_eq + 1] = '\0' + _CommentOutGlobalInBuf(out_buf, text_content + tx_ls) + text_content[tx_eq + 1] = tx_sv2 + # 去重：%type = ... + if tx_ll > 0 and text_content[tx_ls] == '%': + tx_eq2: t.CSizeT = tx_ls + 1 + while tx_eq2 < tx_ls + tx_ll: + if text_content[tx_eq2] == '=': + break + tx_eq2 += 1 + if tx_eq2 < tx_ls + tx_ll: + tx_sv3: t.CChar = text_content[tx_eq2 + 1] + text_content[tx_eq2 + 1] = '\0' + if string.strstr(out_buf, text_content + tx_ls) is not None: + text_content[tx_eq2 + 1] = tx_sv3 + continue + text_content[tx_eq2 + 1] = tx_sv3 + # 追加 + if out_pos + tx_ll + 2 < out_size: + string.strncpy(out_buf + out_pos, text_content + tx_ls, tx_ll) + out_pos += tx_ll + out_buf[out_pos] = '\n' + out_pos += 1 + out_buf[out_pos] = '\0' + stdlib.free(text_content) + + if deps_buf is not None: + stdlib.free(deps_buf) + if sha1_arr is not None: + stdlib.free(sha1_arr) + if mod_arr is not None: + stdlib.free(mod_arr) + + return out_pos diff --git a/TransPyV/App/lib/core/VLogger.py b/TransPyV/App/lib/core/VLogger.py new file mode 100644 index 0000000..c1e5e99 --- /dev/null +++ b/TransPyV/App/lib/core/VLogger.py @@ -0,0 +1,171 @@ +import t, c +from stdint import * +import stdio +import memhub +import w32.win32console as w32cmd +import w32.win32file as w32file +import w32.win32base as w32base + + +# ============================================================ +# VLogger - 原生日志系统（Windows 控制台彩色输出） +# ============================================================ + +# 日志级别 +class LogLevel(t.CEnum): + DEBUG = 0 + INFO = 1 + WARNING = 2 + ERROR = 3 + SUCCESS = 4 + + +# Win32 控制台前景色属性 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +FOREGROUND_WHITE: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE +FOREGROUND_CYAN: t.CDefine = FOREGROUND_BLUE | FOREGROUND_GREEN +FOREGROUND_YELLOW: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN +FOREGROUND_MAGENTA: t.CDefine = FOREGROUND_RED | FOREGROUND_BLUE + +# Win32 控制台背景色属性 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +BACKGROUND_WHITE: t.CDefine = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE + + +class Logger: + _level: int + _console_handle: w32base.HANDLE + _use_color: int + __mbuddy__: memhub.MemManager | t.CPtr + + def __init__(self, level: int = 1): + self._level = level + self._console_handle = w32file.GetStdHandle(w32file.STD_OUTPUT_HANDLE) + self._use_color = 1 + self.__mbuddy__ = _mbuddy + + def _set_color(self, attr: WORD) -> int: + if self._use_color: + return w32cmd.SetConsoleTextAttribute(self._console_handle, attr) + return 0 + + def _reset_color(self) -> int: + if self._use_color: + return w32cmd.SetConsoleTextAttribute(self._console_handle, FOREGROUND_WHITE) + return 0 + + def _log(self, level: int, prefix: str, msg: str, color: WORD, + category: str = "") -> int: + if level < self._level: + return 0 + self._set_color(color) + if category is not None: + if category[0] != 0: + stdio.printf("%s[%s] %s\n", prefix, category, msg) + else: + stdio.printf("%s %s\n", prefix, msg) + else: + stdio.printf("%s %s\n", prefix, msg) + self._reset_color() + return 0 + + def debug(self, msg: str, category: str = "") -> int: + return self._log(LogLevel.DEBUG, "[DEBUG]", msg, + FOREGROUND_INTENSITY, category) + + def info(self, msg: str, category: str = "") -> int: + return self._log(LogLevel.INFO, "[INFO]", msg, + FOREGROUND_GREEN | FOREGROUND_INTENSITY, category) + + def warning(self, msg: str, category: str = "") -> int: + return self._log(LogLevel.WARNING, "[WARN]", msg, + FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY, + category) + + def error(self, msg: str, category: str = "") -> int: + return self._log(LogLevel.ERROR, "[ERROR]", msg, + FOREGROUND_RED | FOREGROUND_INTENSITY, category) + + def success(self, msg: str, category: str = "") -> int: + return self._log(LogLevel.SUCCESS, "[SUCCESS]", msg, + FOREGROUND_GREEN | FOREGROUND_INTENSITY, category) + + def set_level(self, level: int) -> int: + self._level = level + return 0 + + # ============================================================ + # 高级日志方法 + # ============================================================ + + def banner(self, msg: str) -> int: + """输出分节标题（青色高亮）。""" + self._set_color(FOREGROUND_CYAN | FOREGROUND_INTENSITY) + stdio.printf("\n=== %s ===\n\n", msg) + self._reset_color() + return 0 + + def compile_error(self, msg: str, file: str = "", line: int = 0) -> int: + """格式化编译错误输出（红底白字标题 + 位置 + 错误信息）。""" + self._set_color(BACKGROUND_RED | FOREGROUND_WHITE | FOREGROUND_INTENSITY) + stdio.printf(" 编译错误 ") + self._set_color(FOREGROUND_RED | FOREGROUND_INTENSITY) + stdio.printf("\n") + if file is not None: + if file[0] != 0 and line > 0: + stdio.printf(" 位置: %s:%d\n", file, line) + elif file[0] != 0: + stdio.printf(" 文件: %s\n", file) + stdio.printf(" 错误: %s\n", msg) + self._reset_color() + return 0 + + def compile_warning(self, msg: str, file: str = "", line: int = 0) -> int: + """格式化编译警告输出（黄底黑字标题 + 位置 + 警告信息）。""" + self._set_color(BACKGROUND_RED | BACKGROUND_GREEN | FOREGROUND_INTENSITY) + stdio.printf(" 编译警告 ") + self._set_color(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY) + stdio.printf("\n") + if file is not None: + if file[0] != 0 and line > 0: + stdio.printf(" 位置: %s:%d\n", file, line) + elif file[0] != 0: + stdio.printf(" 文件: %s\n", file) + stdio.printf(" 警告: %s\n", msg) + self._reset_color() + return 0 + + +# 全局 mbuddy 指针（由 lib.InitLib 注入） +_mbuddy: memhub.MemManager | t.CPtr + +# 全局 logger 指针 +_g_logger: Logger | t.CPtr + + +def get_logger() -> Logger | t.CPtr: + """获取全局 logger 实例。若不存在则通过 _mbuddy 分配并初始化。""" + if _g_logger is None: + if _mbuddy is None: + return None + raw: t.CVoid | t.CPtr = _mbuddy.alloc(Logger.__sizeof__()) + if raw is None: + return None + _g_logger = raw + _g_logger._level = LogLevel.INFO + _g_logger._console_handle = w32file.GetStdHandle(w32file.STD_OUTPUT_HANDLE) + _g_logger._use_color = 1 + _g_logger.__mbuddy__ = _mbuddy + return _g_logger + + +def set_logger(logger: Logger | t.CPtr) -> int: + """设置全局 logger 实例。""" + _g_logger = logger + return 0 diff --git a/TransPyV/App/lib/core/__init__.py b/TransPyV/App/lib/core/__init__.py new file mode 100644 index 0000000..bc8c2f7 --- /dev/null +++ b/TransPyV/App/lib/core/__init__.py @@ -0,0 +1,9 @@ +import t, c +from stdint import * + +# ============================================================ +# core 包入口 +# +# 子模块通过绝对导入使用（import lib.core.X as X）， +# 此 __init__.py 不做 re-export，避免引入未使用的依赖。 +# ============================================================ diff --git a/TransPyV/App/main.py b/TransPyV/App/main.py new file mode 100644 index 0000000..a3c08ed --- /dev/null +++ b/TransPyV/App/main.py @@ -0,0 +1,510 @@ +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import argparse +import w32.win32console as w32cmd +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import lib +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.core.Phase1 as Phase1 +import lib.core.Phase2 as Phase2 +import lib.Projectrans.Config as Config +import lib.Projectrans.Utils as Utils +import llvmlite +import subprocess +import viperlib +import hashlib + + +# 内存大小: fl_bytes=264 (33*8), POOL_SIZE-fl_bytes 必须是 2 的幂以避免浪费 +# 1073742088 - 264 = 1073741824 (1024MB usable = 2^30) +# Phase1 翻译 87 个 includes 文件 + Phase2 翻译 30 个测试文件，512MB 不足导致 Phase2 崩溃 +POOL_SIZE: t.CDefine = 1073741824 +# 语言编码页 +CODE_PAGE: t.CDefine = 65001 +# 源代码缓冲区大小（1MB） +SRC_BUF_SIZE: t.CDefine = 1048576 + + +# ============================================================ +# main: 命令行入口 +# +# 解析参数 → 加载配置 → AST 解析 → LLVM IR 翻译 → 编译管线 → 可选执行 +# ============================================================ +def main() -> int: + w32cmd.SetConsoleOutputCP(CODE_PAGE) + w32cmd.SetConsoleCP(CODE_PAGE) + # 初始化 mbuddy 内存池 + arena: bytes = stdlib.malloc(POOL_SIZE) + if arena is None: + stdio.printf("FAIL: malloc for arena failed\n") + return 1 + mb: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, POOL_SIZE) + if mb is None: + stdio.printf("FAIL: MemBuddy init failed\n") + return 1 + + # 设置全局 mbuddy 指针（sys 和 argparse 都需要） + sys._mbuddy = mb + argparse._mbuddy = mb + ast._mbuddy = mb + lib._mbuddy = (memhub.MemManager | t.CPtr)(mb) + Config._mbuddy = (memhub.MemManager | t.CPtr)(mb) + Utils._mbuddy = (memhub.MemManager | t.CPtr)(mb) + HandlesTranslator._mbuddy = mb + BuildPipeline._mbuddy = (memhub.MemManager | t.CPtr)(mb) + IncludesScanner._mbuddy = (memhub.MemManager | t.CPtr)(mb) + StubMerger._mbuddy = (memhub.MemManager | t.CPtr)(mb) + Phase1._mbuddy = (memhub.MemManager | t.CPtr)(mb) + Phase2._mbuddy = (memhub.MemManager | t.CPtr)(mb) + subprocess._mbuddy = mb + hashlib._mbuddy = (memhub.MemManager | t.CPtr)(mb) + lib.InitLib((memhub.MemManager | t.CPtr)(mb)) + + # 初始化 VLogger 并打印启动日志 + log: VLogger.Logger | t.CPtr = VLogger.get_logger() + if log is not None: + log.info("TransPyV 启动") + + # 初始化命令行参数（Windows: GetCommandLineA, POSIX: /proc/self/cmdline） + sys._init_argv() + + # 创建参数解析器 + parser: argparse.ArgumentParser | t.CPtr = argparse.ArgumentParser( + "TransPyV", "TransPyV 命令行参数解析", pool=mb) + + # 注册参数（与 Projectrans.py main() 一致） + parser.add_argument("--project", None, argparse.STRING, 0, None, False, + argparse.STORE, "project.json 路径（默认查找当前目录）") + parser.add_argument("--src", None, argparse.STRING, 0, None, False, + argparse.STORE, "源文件目录（覆盖 project.json）") + parser.add_argument("--temp", None, argparse.STRING, 0, None, False, + argparse.STORE, "声明接口临时目录（覆盖 project.json）") + parser.add_argument("--output", None, argparse.STRING, 0, None, False, + argparse.STORE, "输出目录（覆盖 project.json）") + parser.add_argument("--phase", None, argparse.STRING, 0, None, False, + argparse.STORE, "阶段: 1=生成声明, 2=翻译+编译, all=全部") + parser.add_argument("--cc", None, argparse.STRING, 0, None, False, + argparse.STORE, "LLVM 编译器命令（覆盖 project.json）") + parser.add_argument("--clean", None, argparse.BOOL, 0, None, False, + argparse.STORE_TRUE, "清理 output 和 temp 目录") + parser.add_argument("--run", None, argparse.BOOL, 0, None, False, + argparse.STORE_TRUE, "编译成功后立即执行生成的可执行文件") + parser.add_argument("--rebuild-includes", None, argparse.BOOL, 0, None, False, + argparse.STORE_TRUE, "删除 includes.binary 预编译缓存并重新编译所有 includes") + parser.add_argument("--clear-cache", None, argparse.BOOL, 0, None, False, + argparse.STORE_TRUE, "清除 .transpyc_cache 全局缓存") + + # 解析命令行参数 + args: argparse.ParsedArgs | t.CPtr = parser.parse_args(sys._argc, sys._argv) + if args is None: + if log is not None: + log.error("参数解析失败", "argparse") + return 1 + + # 打印解析结果 + if log is not None: + log.banner("TransPyV 参数解析结果") + + # 字符串参数 + project: str = args.get_str("project") + if project is not None: + stdio.printf(" --project: %s\n", project) + else: + stdio.printf(" --project: (未指定)\n") + + src: str = args.get_str("src") + if src is not None: + stdio.printf(" --src: %s\n", src) + else: + stdio.printf(" --src: (未指定)\n") + + temp: str = args.get_str("temp") + if temp is not None: + stdio.printf(" --temp: %s\n", temp) + else: + stdio.printf(" --temp: (未指定)\n") + + output: str = args.get_str("output") + if output is not None: + stdio.printf(" --output: %s\n", output) + else: + stdio.printf(" --output: (未指定)\n") + + phase: str = args.get_str("phase") + if phase is not None: + stdio.printf(" --phase: %s\n", phase) + else: + stdio.printf(" --phase: (未指定，默认 all)\n") + + cc: str = args.get_str("cc") + if cc is not None: + stdio.printf(" --cc: %s\n", cc) + else: + stdio.printf(" --cc: (未指定)\n") + + # 布尔参数 + _gb_clean: INT = args.get_bool("clean") + if _gb_clean: + stdio.printf(" --clean: True\n") + else: + stdio.printf(" --clean: False\n") + + if args.get_bool("run"): + stdio.printf(" --run: True\n") + else: + stdio.printf(" --run: False\n") + + if args.get_bool("rebuild-includes"): + stdio.printf(" --rebuild-includes: True\n") + else: + stdio.printf(" --rebuild-includes: False\n") + + if args.get_bool("clear-cache"): + stdio.printf(" --clear-cache: True\n") + else: + stdio.printf(" --clear-cache: False\n") + + stdio.printf("\n参数解析完成。\n") + + # 加载 project.vpj 配置 + proj_loaded: int = 0 + proj_path: str = project + if proj_path is not None: + if log is not None: + log.banner("工程配置") + if Config.Load_project_config(proj_path) == 0: + proj_loaded = 1 + else: + stdio.printf(" 警告: 无法加载 project.vpj: %s\n", proj_path) + else: + # 尝试默认路径 project.vpj（当前目录和上级目录） + proj_path = "project.vpj" + if Config.Load_project_config(proj_path) == 0: + proj_loaded = 1 + else: + proj_path = "../project.vpj" + if Config.Load_project_config(proj_path) == 0: + proj_loaded = 1 + if proj_loaded == 1: + # 提取 project.vpj 所在目录，将相对路径解析为基于该目录的路径 + proj_len: t.CSizeT = string.strlen(proj_path) + slash_pos: t.CSizeT = proj_len + for i in range(proj_len): + idx: t.CSizeT = proj_len - 1 - i + ch: t.CChar = proj_path[idx] + if ch == '/' or ch == '\\': + slash_pos = idx + break + proj_dir: str = "" + if slash_pos > 0 and slash_pos < proj_len: + proj_dir = stdlib.malloc(slash_pos + 1) + if proj_dir is not None: + string.memcpy(proj_dir, proj_path, slash_pos) + proj_dir[slash_pos] = '\0' + Config.resolve_paths(proj_dir) + Config.print_config() + + # === --clean: 清理 temp 和 output 目录 === + if _gb_clean: + stdio.printf("\n[clean] 清理临时目录...\n") + if Config.TempDir is not None: + n: int = Utils.CleanDir(Config.TempDir) + stdio.printf("[clean] %s: 删除 %d 个文件\n", Config.TempDir, n) + if Config.OutputDir is not None: + n2: int = Utils.CleanDir(Config.OutputDir) + stdio.printf("[clean] %s: 删除 %d 个文件\n", Config.OutputDir, n2) + + # === --phase 参数控制 === + # phase=1: 仅 stub 分离（生成 .stub.ll/.text.ll），不编译 + # phase=2: 仅 stub 合并 + 编译（跳过 stub 分离） + # phase=all: 两者都执行（默认） + phase_mode: str = phase + if phase_mode is None: + phase_mode = "all" + do_phase1: int = 0 + do_phase2: int = 0 + if phase_mode == "1" or phase_mode == "all": + do_phase1 = 1 + if phase_mode == "2" or phase_mode == "all": + do_phase2 = 1 + stdio.printf("[phase] 模式: %s (phase1=%d phase2=%d)\n", phase_mode, do_phase1, do_phase2) + + # === Phase1: 扫描 includes（按需翻译）=== + # 对 includes 目录中每个 .py 文件，若 stub 不存在则翻译并分离 stub + if do_phase1 != 0 and Config.IncludesDir is not None: + ph1_temp: str = Config.TempDir + if ph1_temp is None: + ph1_temp = "." + Phase1.RunPhase1(mb, Config.IncludesDir, ph1_temp, log) + # 如果仅 Phase1（不执行 Phase2），直接退出 + if do_phase2 == 0: + stdio.printf("[phase] Phase1 完成，退出\n") + return 0 + + # AST 解析：如果指定了 --src 或 --project，读取文件并解析为 AST 树 + src_path: str = args.get_str("src") + + # === 多文件项目模式 === + # 当 --project 指定但 --src 未指定时，扫描 source_dir 下所有 .py 文件编译 + if src_path is None and project is not None and Config.SourceDir is not None: + mf_temp: str = Config.TempDir if Config.TempDir is not None else "." + mf_output: str = Config.OutputDir if Config.OutputDir is not None else "." + mf_cc: str = Config.CompilerCmd if Config.CompilerCmd is not None else "llc" + mf_cc_flags: str = Config.CompilerFlags if Config.CompilerFlags is not None else "-filetype=obj -relocation-model=pic" + mf_linker: str = Config.LinkerCmd if Config.LinkerCmd is not None else "clang++" + mf_linker_flags: str = Config.LinkerFlags if Config.LinkerFlags is not None else "-lmsvcrt -lucrt -lpthread -lmingwex -lkernel32 -lgcc -Wl,--allow-multiple-definition" + mf_linker_out: str = Config.LinkerOutput if Config.LinkerOutput is not None else "app.exe" + mf_includes_bin: str = Config.get_includes_binary_dir() + + mf_ret: int = Phase2.RunMultiFileProject( + mb, Config.SourceDir, mf_temp, mf_output, + mf_cc, mf_cc_flags, mf_linker, mf_linker_flags, mf_linker_out, + mf_includes_bin, Config.IncludesDir, do_phase1, do_phase2, log, args) + argparse.release(args) + if log is not None: + log.success("TransPyV 完成") + return mf_ret + + # === 单文件模式：--src 指定时使用指定文件；--project 但无 SourceDir 时回退 === + if src_path is None and project is not None: + if Config.SourceDir is not None: + # 构造入口路径: SourceDir/main.py + sd_len: t.CSizeT = string.strlen(Config.SourceDir) + path_buf: bytes = stdlib.malloc(sd_len + 16) + if path_buf is not None: + viperlib.snprintf(path_buf, sd_len + 16, "%s/main.py", Config.SourceDir) + src_path = path_buf + stdio.printf("[project] 入口文件: %s\n", src_path) + if src_path is not None: + if log is not None: + log.banner("AST 解析") + # 打开文件 + f: fileio.File | t.CPtr = fileio.File(src_path, fileio.MODE.R) + if f.closed: + if log is not None: + log.error("无法打开文件", "fileio") + stdio.printf(" 路径: %s\n", src_path) + argparse.release(args) + return 1 + # 分配源代码缓冲区 + src_buf: bytes = stdlib.malloc(SRC_BUF_SIZE) + if src_buf is None: + if log is not None: + log.error("malloc for src_buf failed", "memhub") + f.close() + argparse.release(args) + return 1 + # 读取文件内容 + bytes_read: LONG = f.read_all(src_buf, SRC_BUF_SIZE) + f.close() + if bytes_read < 0: + if log is not None: + log.error("读取文件失败", "fileio") + stdio.printf(" 错误码: %d\n", bytes_read) + argparse.release(args) + return 1 + # 添加 null 终止符，确保 strlen 和 SHA1 计算正确 + if bytes_read < SRC_BUF_SIZE: + src_buf[bytes_read] = 0 + else: + src_buf[SRC_BUF_SIZE - 1] = 0 + if log is not None: + log.info("文件读取完成", "fileio") + stdio.printf("读取 %d 字节\n", bytes_read) + # 解析 AST（拆分为词法+语法两阶段，便于调试） + ast._init_tables(mb) + lx: ast.Lexer | t.CPtr = ast.new_lexer(mb) + ast._lexer_init(lx, src_buf, mb) + tokens: ast.Token | t.CPtr = ast.tokenize(lx) + tree: ast.AST | t.CPtr = ast.parse_tokens(mb, tokens) + if tree is None: + if log is not None: + log.error("AST 解析失败", "ast") + argparse.release(args) + return 1 + + # === 翻译 AST → LLVM IR === + if log is not None: + log.banner("LLVM IR 翻译") + tr: HandlesTranslator.Translator | t.CPtr = HandlesTranslator.Translator() + result: int = tr.translate(tree) + if result == 0: + IR_BUF_SIZE: t.CSizeT = 262144 + ir_buf: bytes = stdlib.malloc(IR_BUF_SIZE) + if ir_buf is not None: + tr.dump_ir(ir_buf, IR_BUF_SIZE, llvmlite.OUTPUT_FULL) + + # === 编译管线: .ll → .obj → .exe === + if log is not None: + log.banner("编译管线") + + # module_name 用源代码内容的 SHA1（前 16 字符），与 Projectrans.py 一致 + # output_name 从文件名推导（如 test.py -> test.exe），保持可读性 + module_name: str = Utils.compute_sha1(mb, src_buf) + if module_name is None: + module_name = "main" + output_name: str = "test_prog.exe" + # 从文件名提取 output_name（去掉路径和 .py 扩展名） + src_len: t.CSizeT = string.strlen(src_path) + if src_len > 0: + # 找到最后一个 / 或 \ 之后的部分 + base_start: t.CSizeT = src_len + i: t.CSizeT = src_len + while i > 0: + i -= 1 + ch: t.CChar = src_path[i] + if ch == '/' or ch == '\\': + base_start = i + 1 + break + if i == 0: + base_start = 0 + # 找到 .py 扩展名 + base_end: t.CSizeT = src_len + j: t.CSizeT = base_start + while j < src_len: + if src_path[j] == '.': + base_end = j + break + j += 1 + # 输出名 = 文件名 + ".exe" + name_len: t.CSizeT = base_end - base_start + if name_len > 0: + out_len: t.CSizeT = name_len + 5 + output_name = stdlib.malloc(out_len) + if output_name is not None: + # 手动复制文件名并追加 .exe（viperlib.snprintf 不支持 %.*s，string 无 strcat） + string.strncpy(output_name, src_path + base_start, name_len) + output_name[name_len] = '.' + output_name[name_len + 1] = 'e' + output_name[name_len + 2] = 'x' + output_name[name_len + 3] = 'e' + output_name[name_len + 4] = '\0' + # 使用 project.vpj 中的配置 + temp_dir: str = Config.TempDir if Config.TempDir is not None else "." + output_dir: str = Config.OutputDir if Config.OutputDir is not None else "." + cc_cmd: str = Config.CompilerCmd if Config.CompilerCmd is not None else "llc" + cc_flags: str = "-filetype=obj -relocation-model=pic" + linker_cmd: str = Config.LinkerCmd if Config.LinkerCmd is not None else "clang++" + linker_flags: str = "-lmsvcrt -lucrt -lpthread -lmingwex -lkernel32 -lgcc -Wl,--allow-multiple-definition" + # --project 模式且未显式指定 --src 时：使用 config 中的 linker_output（如 TransPyV.exe） + # 显式指定 --src 时：使用从文件名推导的 output_name（如 test.exe），避免覆盖正在运行的 exe + if project is not None and Config.LinkerOutput is not None and args.get_str("src") is None: + linker_output: str = Config.LinkerOutput + else: + linker_output: str = output_name + + # 计算 includes.binary 目录路径（链接时附加预编译 .obj） + includes_binary_dir: str = Config.get_includes_binary_dir() + + ir_len: t.CSizeT = string.strlen(ir_buf) + + # (--phase 参数已在配置加载后处理，此处直接使用 do_phase1/do_phase2) + + # === Phase1 stub 分离：使用 OUTPUT_STUB/TEXT 模式分别生成 stub.ll 和 text.ll === + if do_phase1 != 0 and temp_dir is not None and module_name is not None: + if log is not None: + log.banner("stub 分离") + SF_IR_SIZE: t.CSizeT = 262144 + td_len_sf: t.CSizeT = string.strlen(temp_dir) + # 保存 stub.ll + sf_stub_buf: bytes = stdlib.malloc(SF_IR_SIZE) + if sf_stub_buf is not None: + tr.dump_ir(sf_stub_buf, SF_IR_SIZE, llvmlite.OUTPUT_STUB) + sf_stub_len: t.CSizeT = string.strlen(sf_stub_buf) + sf_stub_path: bytes = stdlib.malloc(td_len_sf + 32) + if sf_stub_path is not None: + viperlib.snprintf(sf_stub_path, td_len_sf + 32, "%s/%s.stub.ll", temp_dir, module_name) + sf_f: fileio.File | t.CPtr = fileio.File(sf_stub_path, fileio.MODE.W) + if not sf_f.closed: + sf_f.write(sf_stub_buf, sf_stub_len) + sf_f.close() + stdlib.free(sf_stub_path) + stdlib.free(sf_stub_buf) + # 保存 text.ll + sf_text_buf: bytes = stdlib.malloc(SF_IR_SIZE) + if sf_text_buf is not None: + tr.dump_ir(sf_text_buf, SF_IR_SIZE, llvmlite.OUTPUT_TEXT) + sf_text_len: t.CSizeT = string.strlen(sf_text_buf) + sf_text_path: bytes = stdlib.malloc(td_len_sf + 32) + if sf_text_path is not None: + viperlib.snprintf(sf_text_path, td_len_sf + 32, "%s/%s.text.ll", temp_dir, module_name) + sf_tf: fileio.File | t.CPtr = fileio.File(sf_text_path, fileio.MODE.W) + if not sf_tf.closed: + sf_tf.write(sf_text_buf, sf_text_len) + sf_tf.close() + stdlib.free(sf_text_path) + stdlib.free(sf_text_buf) + + # phase=1 模式：仅生成 stub，不执行编译 + if do_phase2 == 0: + if log is not None: + log.success("Phase1 完成（仅 stub 分离，跳过编译）") + argparse.release(args) + return 0 + + # === Phase2: 组合本地 stub + 依赖 stubs + 本地 text → 完整 IR === + final_ir: bytes = ir_buf + final_ir_len: t.CSizeT = ir_len + if temp_dir is not None and module_name is not None: + if log is not None: + log.banner("IR 组合") + SF_COMBINED_SIZE: t.CSizeT = 4194304 # 4MB + combined_buf: bytes = stdlib.malloc(SF_COMBINED_SIZE) + if combined_buf is not None: + combined_len: t.CSizeT = StubMerger.BuildCombinedIR(temp_dir, module_name, combined_buf, SF_COMBINED_SIZE) + if combined_len > 0: + final_ir = combined_buf + final_ir_len = combined_len + else: + stdio.printf("[stub] BuildCombinedIR 失败，使用原始 IR\n") + stdlib.free(combined_buf) + + br: BuildPipeline.BuildResult | t.CPtr = BuildPipeline.run_pipeline( + final_ir, final_ir_len, temp_dir, output_dir, module_name, + cc_cmd, cc_flags, linker_cmd, linker_flags, linker_output, + includes_binary_dir + ) + if br is not None and br.Success == 1: + if log is not None: + log.success("编译管线完成") + stdio.printf("输出: %s/%s\n", output_dir, linker_output) + # --run 模式：执行生成的可执行文件 + if args.get_bool("run"): + exe_path: bytes = stdlib.malloc(string.strlen(output_dir) + string.strlen(linker_output) + 2) + if exe_path is not None: + viperlib.snprintf(exe_path, 256, "%s/%s", output_dir, linker_output) + stdio.printf("[run] 执行: %s\n", exe_path) + r: subprocess.CompletedProcess | t.CPtr = subprocess.run(exe_path, False, False) + if r is not None: + stdio.printf("[run] 退出码: %d\n", r.returncode) + else: + if log is not None: + log.error("编译管线失败", "pipeline") + if br is not None and br.ErrorMsg is not None: + stdio.printf("[FATAL] 编译管线失败: %s\n", br.ErrorMsg) + sys.exit(1) + else: + if log is not None: + log.error("malloc for IR buffer failed", "memhub") + else: + if log is not None: + log.error("翻译失败", "translator") + + argparse.release(args) + if log is not None: + log.success("TransPyV 完成") + return 0 diff --git a/TransPyV/Test/App/asm_test.py b/TransPyV/Test/App/asm_test.py new file mode 100644 index 0000000..d09ac4b --- /dev/null +++ b/TransPyV/Test/App/asm_test.py @@ -0,0 +1,47 @@ +import stdio +import t, c + + +# ============================================================ +# c.Asm 内联汇编测试 +# ============================================================ + +def asm_test() -> int: + # 1. 简单无操作数汇编（nop + clobber） + c.Asm("nop", op=[t.ASM_DESCR.CLOBBER_MEMORY]) + stdio.printf("asm: nop ok\n") + + # 2. 仅汇编文本无 clobber + c.Asm("nop") + stdio.printf("asm: nop2 ok\n") + + # 3. 多 clobber + c.Asm("nop", op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_CC]) + stdio.printf("asm: multi-clobber ok\n") + + # 4. 带输入操作数的汇编（读取 CPU ID） + # 使用 f-string 内联 c.AsmInp + # mov eax, 输入值; nop（保持简单，不使用会产生异常的指令） + val: int = 42 + c.Asm(f"""mov eax, {c.AsmInp(val, t.ASM_DESCR.REG_ANY)} +nop""", op=[t.ASM_DESCR.CLOBBER_RAX]) + stdio.printf("asm: input ok val=%d\n", val) + + # 5. 带输出操作数的汇编 + # c.AsmOut 获取结果 + result: int = 0 + c.Asm(f"""mov {c.AsmOut(result, t.ASM_DESCR.OUTPUT_REG)}, 123 +nop""", op=[t.ASM_DESCR.CLOBBER_RAX]) + stdio.printf("asm: output result=%d\n", result) + + # 6. 带输入和输出操作数 + # result2 = val + 1 + result2: int = 0 + input_val: int = 10 + c.Asm(f"""mov eax, {c.AsmInp(input_val, t.ASM_DESCR.REG_ANY)} +add eax, 1 +mov {c.AsmOut(result2, t.ASM_DESCR.OUTPUT_REG)}, eax +nop""", op=[t.ASM_DESCR.CLOBBER_RAX]) + stdio.printf("asm: in+out result2=%d (expect 11)\n", result2) + + return 0 diff --git a/TransPyV/Test/App/attr_test.py b/TransPyV/Test/App/attr_test.py new file mode 100644 index 0000000..4bc8cad --- /dev/null +++ b/TransPyV/Test/App/attr_test.py @@ -0,0 +1,234 @@ +import t, c +from stdint import * +import stdio +import string + + +# ============================================================ +# c.Attribute 装饰器测试 +# +# 测试 @c.Attribute(...) 对函数属性的设置： +# - t.attr.always_inline() -> alwaysinline +# - t.attr.noinline() -> noinline +# - t.attr.noreturn() -> noreturn +# - t.attr.pure() -> readonly (LLVM 函数属性) +# - t.attr.llvm.nounwind -> nounwind +# ============================================================ + + +# Test 1: always_inline 属性 +@c.Attribute(t.attr.always_inline()) +def AlwaysInlineFunc(x: t.CInt) -> t.CInt: + return x * 2 + + +# Test 2: noinline 属性 +@c.Attribute(t.attr.noinline()) +def NoInlineFunc(x: t.CInt) -> t.CInt: + return x + 100 + + +# Test 3: noreturn 属性（函数确实不返回） +@c.Attribute(t.attr.noreturn()) +def NoReturnFunc() -> t.CInt: + stdio.printf("NoReturnFunc called (does not return)\n") + return 0 + + +# Test 4: 普通函数对照（无装饰器） +def NormalFunc(x: t.CInt) -> t.CInt: + return x * 3 + + +# Test 5: 多属性组合 +@c.Attribute(t.attr.always_inline(), t.attr.pure()) +def MultiAttrFunc(x: t.CInt) -> t.CInt: + return x + 1 + + +# Test 6: 无括号属性引用 t.attr.packed +@c.Attribute(t.attr.packed) +def PackedAttrFunc(x: t.CInt) -> t.CInt: + return x - 1 + + +def test_always_inline() -> t.CInt: + stdio.printf("--- Test 1: c.Attribute(always_inline) ---\n") + r: t.CInt = AlwaysInlineFunc(21) + stdio.printf("AlwaysInlineFunc(21)=%d (expect 42)\n", r) + if r == 42: + stdio.printf("AlwaysInlineFunc OK\n") + else: + stdio.printf("AlwaysInlineFunc FAIL\n") + return 0 + + +def test_noinline() -> t.CInt: + stdio.printf("--- Test 2: c.Attribute(noinline) ---\n") + r: t.CInt = NoInlineFunc(5) + stdio.printf("NoInlineFunc(5)=%d (expect 105)\n", r) + if r == 105: + stdio.printf("NoInlineFunc OK\n") + else: + stdio.printf("NoInlineFunc FAIL\n") + return 0 + + +def test_noreturn() -> t.CInt: + stdio.printf("--- Test 3: c.Attribute(noreturn) ---\n") + stdio.printf("NoReturnFunc declared with noreturn attr\n") + stdio.printf("NoReturnFunc OK (not called to avoid UB)\n") + return 0 + + +def test_normal() -> t.CInt: + stdio.printf("--- Test 4: normal function (no attr) ---\n") + r: t.CInt = NormalFunc(7) + stdio.printf("NormalFunc(7)=%d (expect 21)\n", r) + if r == 21: + stdio.printf("NormalFunc OK\n") + else: + stdio.printf("NormalFunc FAIL\n") + return 0 + + +def test_multi_attr() -> t.CInt: + stdio.printf("--- Test 5: c.Attribute(always_inline, pure) ---\n") + r: t.CInt = MultiAttrFunc(10) + stdio.printf("MultiAttrFunc(10)=%d (expect 11)\n", r) + if r == 11: + stdio.printf("MultiAttrFunc OK\n") + else: + stdio.printf("MultiAttrFunc FAIL\n") + return 0 + + +def test_packed_attr() -> t.CInt: + stdio.printf("--- Test 6: c.Attribute(packed) ---\n") + r: t.CInt = PackedAttrFunc(10) + stdio.printf("PackedAttrFunc(10)=%d (expect 9)\n", r) + if r == 9: + stdio.printf("PackedAttrFunc OK\n") + else: + stdio.printf("PackedAttrFunc FAIL\n") + return 0 + + +# Test 7: t.attr.llvm.nounwind 属性 +@c.Attribute(t.attr.llvm.nounwind) +def NoUnwindFunc(x: t.CInt) -> t.CInt: + return x + 1 + + +# Test 8: t.attr.llvm.noredzone 属性 +@c.Attribute(t.attr.llvm.noredzone) +def NoRedZoneFunc(x: t.CInt) -> t.CInt: + return x + 2 + + +# Test 9: t.attr.llvm.willreturn 属性 +@c.Attribute(t.attr.llvm.willreturn) +def WillReturnFunc(x: t.CInt) -> t.CInt: + return x + 3 + + +# Test 10: t.attr.llvm.mustprogress 属性 +@c.Attribute(t.attr.llvm.mustprogress) +def MustProgressFunc(x: t.CInt) -> t.CInt: + return x + 4 + + +# Test 11: t.attr.const() -> readnone +@c.Attribute(t.attr.const()) +def ConstFunc(x: t.CInt) -> t.CInt: + return x * 0 + 42 + + +# Test 12: 多属性组合（alwaysinline + nounwind + noredzone） +@c.Attribute(t.attr.always_inline(), t.attr.llvm.nounwind, t.attr.llvm.noredzone) +def TripleAttrFunc(x: t.CInt) -> t.CInt: + return x + 5 + + +def test_nounwind() -> t.CInt: + stdio.printf("--- Test 7: c.Attribute(llvm.nounwind) ---\n") + r: t.CInt = NoUnwindFunc(10) + stdio.printf("NoUnwindFunc(10)=%d (expect 11)\n", r) + if r == 11: + stdio.printf("NoUnwindFunc OK\n") + else: + stdio.printf("NoUnwindFunc FAIL\n") + return 0 + + +def test_noredzone() -> t.CInt: + stdio.printf("--- Test 8: c.Attribute(llvm.noredzone) ---\n") + r: t.CInt = NoRedZoneFunc(10) + stdio.printf("NoRedZoneFunc(10)=%d (expect 12)\n", r) + if r == 12: + stdio.printf("NoRedZoneFunc OK\n") + else: + stdio.printf("NoRedZoneFunc FAIL\n") + return 0 + + +def test_willreturn() -> t.CInt: + stdio.printf("--- Test 9: c.Attribute(llvm.willreturn) ---\n") + r: t.CInt = WillReturnFunc(10) + stdio.printf("WillReturnFunc(10)=%d (expect 13)\n", r) + if r == 13: + stdio.printf("WillReturnFunc OK\n") + else: + stdio.printf("WillReturnFunc FAIL\n") + return 0 + + +def test_mustprogress() -> t.CInt: + stdio.printf("--- Test 10: c.Attribute(llvm.mustprogress) ---\n") + r: t.CInt = MustProgressFunc(10) + stdio.printf("MustProgressFunc(10)=%d (expect 14)\n", r) + if r == 14: + stdio.printf("MustProgressFunc OK\n") + else: + stdio.printf("MustProgressFunc FAIL\n") + return 0 + + +def test_const_attr() -> t.CInt: + stdio.printf("--- Test 11: c.Attribute(const) ---\n") + r: t.CInt = ConstFunc(99) + stdio.printf("ConstFunc(99)=%d (expect 42)\n", r) + if r == 42: + stdio.printf("ConstFunc OK\n") + else: + stdio.printf("ConstFunc FAIL\n") + return 0 + + +def test_triple_attr() -> t.CInt: + stdio.printf("--- Test 12: c.Attribute(always_inline, nounwind, noredzone) ---\n") + r: t.CInt = TripleAttrFunc(10) + stdio.printf("TripleAttrFunc(10)=%d (expect 15)\n", r) + if r == 15: + stdio.printf("TripleAttrFunc OK\n") + else: + stdio.printf("TripleAttrFunc FAIL\n") + return 0 + + +def attr_test() -> t.CInt: + stdio.printf("=== attr_test: c.Attribute 装饰器测试 ===\n\n") + test_always_inline() + test_noinline() + test_noreturn() + test_normal() + test_multi_attr() + test_packed_attr() + test_nounwind() + test_noredzone() + test_willreturn() + test_mustprogress() + test_const_attr() + test_triple_attr() + stdio.printf("\n=== attr_test 完成 ===\n") + return 0 diff --git a/TransPyV/Test/App/augassign_test.py b/TransPyV/Test/App/augassign_test.py new file mode 100644 index 0000000..20b52ce --- /dev/null +++ b/TransPyV/Test/App/augassign_test.py @@ -0,0 +1,58 @@ +import stdio +import t, c + + +# ============================================================ +# AugAssign 测试：+= -= *= /= %= &= |= ^= <<= >>= +# ============================================================ + +g_count: int = 0 + + +def test_local() -> int: + x: int = 10 + x += 5 + stdio.printf("local +=: %d\n", x) + x -= 3 + stdio.printf("local -=: %d\n", x) + x *= 2 + stdio.printf("local *=: %d\n", x) + x /= 4 + stdio.printf("local /=: %d\n", x) + x %= 7 + stdio.printf("local %%=: %d\n", x) + return 0 + + +def test_global_aug() -> int: + global g_count + g_count += 1 + stdio.printf("global +=: %d\n", g_count) + g_count += 10 + stdio.printf("global +=: %d\n", g_count) + g_count -= 3 + stdio.printf("global -=: %d\n", g_count) + return 0 + + +def test_bitops() -> int: + b: int = 0xFF + b &= 0x0F + stdio.printf("bit &=: %d\n", b) + b |= 0x30 + stdio.printf("bit |=: %d\n", b) + b ^= 0xFF + stdio.printf("bit ^=: %d\n", b) + b = 1 + b <<= 4 + stdio.printf("bit <<=: %d\n", b) + b >>= 2 + stdio.printf("bit >>=: %d\n", b) + return 0 + + +def augassign_test() -> int: + test_local() + test_global_aug() + test_bitops() + return 0 diff --git a/TransPyV/Test/App/closure_test.py b/TransPyV/Test/App/closure_test.py new file mode 100644 index 0000000..c2c876e --- /dev/null +++ b/TransPyV/Test/App/closure_test.py @@ -0,0 +1,45 @@ +import stdio +import t, c + + +# ============================================================ +# 闭包 + nonlocal/global 测试 +# ============================================================ + +# 全局变量 +g_count: int = 0 + + +# global 关键字测试：修改全局变量 +def test_global() -> int: + global g_count + g_count = g_count + 1 + return g_count + + +# nonlocal 关键字测试：嵌套函数修改外层局部变量 +def make_counter() -> t.CPtr: + count: int = 0 + + # 内部函数（闭包）捕获 count + def counter() -> int: + nonlocal count + count = count + 1 + return count + + return counter + + +def closure_test() -> int: + # global 测试 + stdio.printf("global: %d\n", test_global()) + stdio.printf("global: %d\n", test_global()) + stdio.printf("global: %d\n", test_global()) + + # nonlocal/闭包测试 + f: t.CPtr = make_counter() + stdio.printf("closure: %d\n", f()) + stdio.printf("closure: %d\n", f()) + stdio.printf("closure: %d\n", f()) + + return 0 diff --git a/TransPyV/Test/App/deco_test.py b/TransPyV/Test/App/deco_test.py new file mode 100644 index 0000000..9af6396 --- /dev/null +++ b/TransPyV/Test/App/deco_test.py @@ -0,0 +1,183 @@ +import stdio +import t, c +import string + + +# ============================================================ +# deco_test - t/c 装饰器测试 +# +# 测试 t.CDefine / t.CExport / t.CInline / t.CExtern / t.State +# ============================================================ + + +# Test 1: t.CDefine - 常量定义 +MAX_VALUE: t.CDefine = 256 +PI_APPROX: t.CDefine = 314 +# Test 10: t.CDefine 作为 t.CArray 的 count +ARR_SIZE: t.CDefine = 8 + + +# Test 2: t.CExport - 导出函数标记 +def ExportedAdd(a: t.CInt, b: t.CInt) -> t.CInt | t.CExport: + return a + b + + +# Test 3: t.CInline - 内联函数标记 +def InlineSquare(x: t.CInt) -> t.CInt | t.CInline: + return x * x + + +# Test 4: 普通函数（无装饰器，作为对照） +def NormalSub(a: t.CInt, b: t.CInt) -> t.CInt: + return a - b + + +# Test 5: t.CExport 与 t.CInline 组合 +def ExportedInlineCube(x: t.CInt) -> t.CInt | t.CExport | t.CInline: + return x * x * x + + +# Test 6: t.CExport 返回指针 +def ExportedFindChar(s: str, ch: t.CInt) -> str | t.CExport: + return string.strchr(s, ch) + + +def test_cdefine(): + stdio.printf("--- Test 1: t.CDefine ---\n") + stdio.printf("MAX_VALUE=%d (expect 256)\n", MAX_VALUE) + stdio.printf("PI_APPROX=%d (expect 314)\n", PI_APPROX) + if MAX_VALUE == 256: + stdio.printf("MAX_VALUE OK\n") + else: + stdio.printf("MAX_VALUE FAIL\n") + if PI_APPROX == 314: + stdio.printf("PI_APPROX OK\n") + else: + stdio.printf("PI_APPROX FAIL\n") + + +def test_cexport(): + stdio.printf("--- Test 2: t.CExport ---\n") + r: t.CInt = ExportedAdd(3, 4) + stdio.printf("ExportedAdd(3,4)=%d (expect 7)\n", r) + if r == 7: + stdio.printf("ExportedAdd OK\n") + else: + stdio.printf("ExportedAdd FAIL\n") + + +def test_cinline(): + stdio.printf("--- Test 3: t.CInline ---\n") + r: t.CInt = InlineSquare(5) + stdio.printf("InlineSquare(5)=%d (expect 25)\n", r) + if r == 25: + stdio.printf("InlineSquare OK\n") + else: + stdio.printf("InlineSquare FAIL\n") + + +def test_deco_normal(): + stdio.printf("--- Test 4: normal function ---\n") + r: t.CInt = NormalSub(10, 3) + stdio.printf("NormalSub(10,3)=%d (expect 7)\n", r) + if r == 7: + stdio.printf("NormalSub OK\n") + else: + stdio.printf("NormalSub FAIL\n") + + +def test_export_inline(): + stdio.printf("--- Test 5: t.CExport | t.CInline ---\n") + r: t.CInt = ExportedInlineCube(3) + stdio.printf("ExportedInlineCube(3)=%d (expect 27)\n", r) + if r == 27: + stdio.printf("ExportedInlineCube OK\n") + else: + stdio.printf("ExportedInlineCube FAIL\n") + + +def test_export_ptr(): + stdio.printf("--- Test 6: t.CExport return ptr ---\n") + p: str = ExportedFindChar("Hello", 108) # 'l' = 108 + if p is not None: + stdio.printf("ExportedFindChar found, char=%d (expect 108)\n", c.Deref(p)) + if c.Deref(p) == 108: + stdio.printf("ExportedFindChar OK\n") + else: + stdio.printf("ExportedFindChar FAIL\n") + else: + stdio.printf("ExportedFindChar NOT FOUND (FAIL)\n") + + +# Test 7: t.CExtern | t.CExport - 外部声明函数（pass 体） +# 声明一个外部函数，翻译器应生成 declare 而非 define +def ExternalDecl(x: t.CInt) -> t.CInt | t.CExtern | t.CExport: pass + + +# Test 8: t.State - 状态声明（等价于 CExtern | CExport） +def StateDecl(x: t.CInt) -> t.CInt | t.State: pass + + +# Test 9: t.State 返回 void +def StateVoidDecl() -> t.State: pass + + +def test_extern_decl(): + stdio.printf("--- Test 7: t.CExtern | t.CExport ---\n") + # 外部声明函数，不调用（应由链接器解析符号） + stdio.printf("ExternalDecl declared (not called)\n") + stdio.printf("ExternalDecl OK\n") + + +def test_state_decl(): + stdio.printf("--- Test 8: t.State ---\n") + stdio.printf("StateDecl declared (not called)\n") + stdio.printf("StateDecl OK\n") + + +def test_state_void_decl(): + stdio.printf("--- Test 9: t.State (void) ---\n") + stdio.printf("StateVoidDecl declared (not called)\n") + stdio.printf("StateVoidDecl OK\n") + + +# Test 10: t.CDefine 作为 t.CArray count +# ARR_SIZE: t.CDefine = 8 已在文件顶部定义 +def test_cdefine_array_count(): + stdio.printf("--- Test 10: t.CArray[elem, CDefine] ---\n") + # 使用 CDefine 常量名作为数组 count + arr: t.CArray[t.CInt, ARR_SIZE] + i: t.CInt + # 初始化: arr[i] = i * 10 + for i in range(ARR_SIZE): + arr[i] = i * 10 + # 求和验证: 0+10+20+...+70 = 280 + total: t.CInt = 0 + for i in range(ARR_SIZE): + total += arr[i] + stdio.printf("ARR_SIZE=%d (expect 8)\n", ARR_SIZE) + stdio.printf("arr sum=%d (expect 280)\n", total) + stdio.printf("arr[0]=%d (expect 0)\n", arr[0]) + stdio.printf("arr[7]=%d (expect 70)\n", arr[7]) + if ARR_SIZE == 8 and total == 280 and arr[0] == 0 and arr[7] == 70: + stdio.printf("CArray CDefine count OK\n") + else: + stdio.printf("CArray CDefine count FAIL\n") + + +def deco_test() -> int: + stdio.printf("=== deco_test: t/c 装饰器测试 ===\n\n") + + test_cdefine() + test_cexport() + test_cinline() + test_deco_normal() + test_export_inline() + test_export_ptr() + test_extern_decl() + test_state_decl() + test_state_void_decl() + test_cdefine_array_count() + + stdio.printf("\n=== deco_test 完成 ===\n") + return 0 diff --git a/TransPyV/Test/App/deref_min_test.py b/TransPyV/Test/App/deref_min_test.py new file mode 100644 index 0000000..9bc3210 --- /dev/null +++ b/TransPyV/Test/App/deref_min_test.py @@ -0,0 +1,9 @@ +import stdio +import t, c + + +def deref_min_test() -> int: + s: str = "hello" + v: int = c.Deref(s) + stdio.printf("deref: %d\n", v) + return 0 diff --git a/TransPyV/Test/App/deref_test.py b/TransPyV/Test/App/deref_test.py new file mode 100644 index 0000000..b4483cc --- /dev/null +++ b/TransPyV/Test/App/deref_test.py @@ -0,0 +1,13 @@ +import stdio +import t, c + + +def deref_test() -> int: + s: str = "hello" + sp: str = s + slen: int = 0 + while c.Deref(sp) != 0: + slen = slen + 1 + sp = sp + 1 + stdio.printf("ptr: len(hello)=%d\n", slen) + return 0 diff --git a/TransPyV/Test/App/eq_test.py b/TransPyV/Test/App/eq_test.py new file mode 100644 index 0000000..d3c315d --- /dev/null +++ b/TransPyV/Test/App/eq_test.py @@ -0,0 +1,144 @@ +import stdio +import t, c +import string + + +# ============================================================ +# eq_test - ==、!=、is、is not、None 比较测试 +# +# 验证比较表达式的翻译，特别是 None 常量和 is/is not 操作符 +# ============================================================ + + +def test_int_eq(): + stdio.printf("--- Test 1: int == / != ---\n") + + a: t.CInt = 42 + b: t.CInt = 42 + c3: t.CInt = 99 + + if a == b: + stdio.printf("int == int (same) OK\n") + else: + stdio.printf("int == int (same) FAIL\n") + + if a != c3: + stdio.printf("int != int (diff) OK\n") + else: + stdio.printf("int != int (diff) FAIL\n") + + if not (a == c3): + stdio.printf("not (int == diff) OK\n") + else: + stdio.printf("not (int == diff) FAIL\n") + + +def test_int_cmp(): + stdio.printf("--- Test 2: int < > <= >= ---\n") + + a: t.CInt = 5 + b: t.CInt = 10 + + if a < b: + stdio.printf("int < OK\n") + else: + stdio.printf("int < FAIL\n") + + if b > a: + stdio.printf("int > OK\n") + else: + stdio.printf("int > FAIL\n") + + if a <= 5: + stdio.printf("int <= OK\n") + else: + stdio.printf("int <= FAIL\n") + + if b >= 10: + stdio.printf("int >= OK\n") + else: + stdio.printf("int >= FAIL\n") + + +def test_ptr_is_none(): + stdio.printf("--- Test 3: ptr is None / is not None ---\n") + + # strchr 找到字符返回非空指针，找不到返回 None + p: str = string.strchr("Hello", 108) # 'l' = 108 + if p is not None: + stdio.printf("strchr found, is not None OK\n") + else: + stdio.printf("strchr found, is not None FAIL\n") + + p2: str = string.strchr("Hello", 122) # 'z' = 122, not found + if p2 is None: + stdio.printf("strchr not found, is None OK\n") + else: + stdio.printf("strchr not found, is None FAIL\n") + + +def test_ptr_assign_none(): + stdio.printf("--- Test 4: assign None and compare ---\n") + + p: str = None + if p is None: + stdio.printf("None assign, is None OK\n") + else: + stdio.printf("None assign, is None FAIL\n") + + if p is not None: + stdio.printf("None assign, is not None FAIL\n") + else: + stdio.printf("None assign, is not None OK\n") + + +def test_ptr_eq_none(): + stdio.printf("--- Test 5: ptr == None / != None ---\n") + + p: str = string.strstr("Hello World", "World") + if p != None: + stdio.printf("strstr found, != None OK\n") + else: + stdio.printf("strstr found, != None FAIL\n") + + p2: str = string.strstr("Hello", "xyz") + if p2 == None: + stdio.printf("strstr not found, == None OK\n") + else: + stdio.printf("strstr not found, == None FAIL\n") + + +def test_bool_logic(): + stdio.printf("--- Test 6: bool and/or ---\n") + + a: t.CInt = 1 + b: t.CInt = 0 + + if a and b: + stdio.printf("1 and 0 = true FAIL\n") + else: + stdio.printf("1 and 0 = false OK\n") + + if a or b: + stdio.printf("1 or 0 = true OK\n") + else: + stdio.printf("1 or 0 = true FAIL\n") + + if not b: + stdio.printf("not 0 = true OK\n") + else: + stdio.printf("not 0 = true FAIL\n") + + +def eq_test() -> int: + stdio.printf("=== eq_test: ==、!=、is、None 比较测试 ===\n\n") + + test_int_eq() + test_int_cmp() + test_ptr_is_none() + test_ptr_assign_none() + test_ptr_eq_none() + test_bool_logic() + + stdio.printf("\n=== eq_test 完成 ===\n") + return 0 diff --git a/TransPyV/Test/App/float_test.py b/TransPyV/Test/App/float_test.py new file mode 100644 index 0000000..100f4f0 --- /dev/null +++ b/TransPyV/Test/App/float_test.py @@ -0,0 +1,84 @@ +import stdio +import t, c + + +# ============================================================ +# 联合体（t.CUnion）含浮点字段 +# ============================================================ + +class DataUnion(t.CUnion): + i: t.CInt + f: t.CFloat + l: t.CInt64T + + +# ============================================================ +# 主函数 +# ============================================================ +def float_test() -> int: + # ============================================================ + # 测试 1: 基本浮点变量（float / double） + # ============================================================ + f: t.CFloat = 3.14 + stdio.printf("float: f=%f\n", f) + + d: t.CDouble = 3.141592653589793 + stdio.printf("double: d=%lf\n", d) + + # 重新赋值 + f = 2.5 + stdio.printf("float: f=%f\n", f) + + # ============================================================ + # 测试 2: 联合体浮点字段 + # ============================================================ + u: DataUnion + u.f = 3.14 + stdio.printf("union: u.f=%f\n", u.f) + + # 写入 i 字段后，f 的值已被覆盖 + u.i = 42 + stdio.printf("union: u.i=%d (after f overwritten)\n", u.i) + + # ============================================================ + # 测试 3: 联合体共享内存验证（float 与 int 共享） + # ============================================================ + u2: DataUnion + u2.i = 0 + u2.f = 1.0 + # 写入 f 后，i 的值应不再是 0 + if u2.i != 0: + stdio.printf("union: float overwrites int verified\n") + + # ============================================================ + # 测试 4: 浮点算术运算（fadd/fsub/fmul/fdiv/frem） + # ============================================================ + a: t.CFloat = 1.5 + b: t.CFloat = 2.5 + stdio.printf("arith: %.2f + %.2f = %.2f\n", a, b, a + b) + stdio.printf("arith: %.2f - %.2f = %.2f\n", a, b, a - b) + stdio.printf("arith: %.2f * %.2f = %.2f\n", a, b, a * b) + stdio.printf("arith: %.2f / %.2f = %.2f\n", a, b, a / b) + + # ============================================================ + # 测试 5: double 算术运算 + # ============================================================ + x: t.CDouble = 10.0 + y: t.CDouble = 3.0 + stdio.printf("arith: %.4lf + %.4lf = %.4lf\n", x, y, x + y) + stdio.printf("arith: %.4lf - %.4lf = %.4lf\n", x, y, x - y) + stdio.printf("arith: %.4lf * %.4lf = %.4lf\n", x, y, x * y) + stdio.printf("arith: %.4lf / %.4lf = %.4lf\n", x, y, x / y) + + # ============================================================ + # 测试 6: 浮点取模 (frem) + # ============================================================ + stdio.printf("arith: 10.0 %% 3.0 = %.4lf\n", x % y) + + # ============================================================ + # 测试 7: int 与 float 混合运算（int 自动转 float） + # ============================================================ + n: t.CInt = 3 + stdio.printf("arith: %d + %.2f = %.2f\n", n, a, n + a) + + return 0 diff --git a/TransPyV/Test/App/flow_test.py b/TransPyV/Test/App/flow_test.py new file mode 100644 index 0000000..89114ac --- /dev/null +++ b/TransPyV/Test/App/flow_test.py @@ -0,0 +1,23 @@ +import stdio +import t, c + + +def flow_test() -> int: + i: int = 0 + total: int = 0 + while i < 5: + total = total + i + i = i + 1 + stdio.printf("while: total=%d i=%d\n", total, i) + + for k in range(10): + if k == 3: + break + stdio.printf("break: k=%d\n", k) + + for m in range(5): + if m == 2: + continue + stdio.printf("continue: m=%d\n", m) + + return 0 diff --git a/TransPyV/Test/App/for_test.py b/TransPyV/Test/App/for_test.py new file mode 100644 index 0000000..e178457 --- /dev/null +++ b/TransPyV/Test/App/for_test.py @@ -0,0 +1,8 @@ +import stdio +import t, c + + +def for_test() -> int: + for j in range(5): + stdio.printf("for: j=%d\n", j) + return 0 diff --git a/TransPyV/Test/App/func_test.py b/TransPyV/Test/App/func_test.py new file mode 100644 index 0000000..21a3d4b --- /dev/null +++ b/TransPyV/Test/App/func_test.py @@ -0,0 +1,68 @@ +import stdio +import t, c + + +def add(a: int, b: int) -> int: + return a + b + + +def mul(a: int, b: int) -> int: + return a * b + + +def square(x: int) -> int: + return mul(x, x) + + +def factorial(n: int) -> int: + if n <= 1: + return 1 + return n * factorial(n - 1) + + +def greet(greeting: str, name: str, times: int) -> int: + i: int = 0 + while i < times: + stdio.printf("%s, %s!\n", greeting, name) + i = i + 1 + return times + + +def sub(a: int, b: int) -> int: + return a - b + + +def func_test() -> int: + r1: int = add(3, 4) + stdio.printf("add(3,4)=%d\n", r1) + + r2: int = mul(5, 6) + stdio.printf("mul(5,6)=%d\n", r2) + + r3: int = square(7) + stdio.printf("square(7)=%d\n", r3) + + r5: int = factorial(5) + stdio.printf("factorial(5)=%d\n", r5) + + # ============================================================ + # 测试: 函数乱序传参（关键字参数） + # ============================================================ + # 乱序传参：b=4, a=3 等价于 add(3, 4) + r6: int = add(b=4, a=3) + stdio.printf("add(b=4,a=3)=%d\n", r6) + + # 乱序传参：a=10, b=3 等价于 sub(10, 3) + r7: int = sub(b=3, a=10) + stdio.printf("sub(b=3,a=10)=%d\n", r7) + + # 混合传参：位置参数 + 关键字参数 + # greeting="Hello" 是位置参数，name="World" 和 times=2 是关键字 + r8: int = greet("Hello", name="World", times=2) + stdio.printf("greet mixed: returned=%d\n", r8) + + # 全关键字乱序传参 + r9: int = greet(times=1, name="TransPyC", greeting="Hi") + stdio.printf("greet kwargs: returned=%d\n", r9) + + return 0 diff --git a/TransPyV/Test/App/func_vtable_test.py b/TransPyV/Test/App/func_vtable_test.py new file mode 100644 index 0000000..854ab7c --- /dev/null +++ b/TransPyV/Test/App/func_vtable_test.py @@ -0,0 +1,116 @@ +import stdio +import t, c + + +# ============================================================ +# 场景 A: @t.NoVTable 类 + 函数级 @t.CVTable +# +# NoVTableClass 标记 @t.NoVTable（类级别禁用虚表） +# 但 VirtualMethod 标记 @t.CVTable → 该方法单独进入虚表 +# NormalMethod 无装饰器 → 不进入虚表 +# 预期: 虚表 1 个方法 (VirtualMethod) +# ============================================================ +@t.NoVTable +class NoVTableClass: + val: t.CInt + + def __init__(self, v: t.CInt): + self.val = v + + @t.CVTable + def VirtualMethod(self) -> t.CInt: + return self.val + 10 + + def NormalMethod(self) -> t.CInt: + return self.val + 20 + + +# ============================================================ +# 场景 B: @t.CVTable 类 + 函数级 @t.NoVTable +# +# CVTableClass 标记 @t.CVTable（所有方法进入虚表） +# KeptMethod 无装饰器 → 进入虚表 +# ExcludedMethod 标记 @t.NoVTable → 排除出虚表 +# 预期: 虚表 1 个方法 (KeptMethod) +# ============================================================ +@t.CVTable +class CVTableClass: + val: t.CInt + + def __init__(self, v: t.CInt): + self.val = v + + def KeptMethod(self) -> t.CInt: + return self.val + 30 + + @t.NoVTable + def ExcludedMethod(self) -> t.CInt: + return self.val + 40 + + +# ============================================================ +# 场景 C: 默认类（无装饰器无继承）+ 函数级 @t.CVTable +# +# DefaultClass 无装饰器（默认无虚表） +# VirtualMethod 标记 @t.CVTable → 单独进入虚表 +# NormalMethod 无装饰器 → 不进入虚表 +# 预期: 虚表 1 个方法 (VirtualMethod) +# ============================================================ +class DefaultClass: + val: t.CInt + + def __init__(self, v: t.CInt): + self.val = v + + @t.CVTable + def VirtualMethod(self) -> t.CInt: + return self.val + 50 + + def NormalMethod(self) -> t.CInt: + return self.val + 60 + + +def func_vtable_test() -> int: + stdio.printf("funcvt: === Test Start ===\n") + + # 场景 A: NoVTableClass + nvt: NoVTableClass = NoVTableClass(5) + a1: int = nvt.VirtualMethod() + stdio.printf("funcvt: A.VirtualMethod()=%d (expected 15)\n", a1) + if a1 != 15: + stdio.printf("[FAIL] A.VirtualMethod()=%d expected 15\n", a1) + return 1 + a2: int = nvt.NormalMethod() + stdio.printf("funcvt: A.NormalMethod()=%d (expected 25)\n", a2) + if a2 != 25: + stdio.printf("[FAIL] A.NormalMethod()=%d expected 25\n", a2) + return 1 + + # 场景 B: CVTableClass + cvt: CVTableClass = CVTableClass(7) + b1: int = cvt.KeptMethod() + stdio.printf("funcvt: B.KeptMethod()=%d (expected 37)\n", b1) + if b1 != 37: + stdio.printf("[FAIL] B.KeptMethod()=%d expected 37\n", b1) + return 1 + b2: int = cvt.ExcludedMethod() + stdio.printf("funcvt: B.ExcludedMethod()=%d (expected 47)\n", b2) + if b2 != 47: + stdio.printf("[FAIL] B.ExcludedMethod()=%d expected 47\n", b2) + return 1 + + # 场景 C: DefaultClass + dfc: DefaultClass = DefaultClass(9) + c1: int = dfc.VirtualMethod() + stdio.printf("funcvt: C.VirtualMethod()=%d (expected 59)\n", c1) + if c1 != 59: + stdio.printf("[FAIL] C.VirtualMethod()=%d expected 59\n", c1) + return 1 + c2: int = dfc.NormalMethod() + stdio.printf("funcvt: C.NormalMethod()=%d (expected 69)\n", c2) + if c2 != 69: + stdio.printf("[FAIL] C.NormalMethod()=%d expected 69\n", c2) + return 1 + + stdio.printf("funcvt: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/generic_test.py b/TransPyV/Test/App/generic_test.py new file mode 100644 index 0000000..969e41e --- /dev/null +++ b/TransPyV/Test/App/generic_test.py @@ -0,0 +1,102 @@ +import stdio +import stdlib +import t, c +import testcheck +import memhub +import _list + + +# ============================================================ +# 泛型类 list[T] 测试 +# ============================================================ +def generic_test() -> int: + testcheck.begin("GenericTest: list[T] 泛型类测试") + + # 创建 mbuddy arena + arena: bytes = stdlib.malloc(65536) + bd: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, 65536) + + # === Test 1: 创建和 append === + testcheck.section("Test 1: 创建和 append") + nums = list[int](bd) + nums.append(10) + nums.append(20) + nums.append(30) + v0: t.CInt = nums.get(0) + v1: t.CInt = nums.get(1) + v2: t.CInt = nums.get(2) + stdio.printf("v0=%d v1=%d v2=%d\n", v0, v1, v2) + testcheck.check(v0 == 10 and v1 == 20 and v2 == 30, + "append+get OK (10,20,30)", "append+get FAILED") + + # === Test 2: __len__ === + testcheck.section("Test 2: __len__") + n: t.CSizeT = nums.__len__() + stdio.printf("len=%lu\n", n) + testcheck.check(n == 3, "__len__ OK (3)", "__len__ FAILED expect 3") + + # === Test 3: set === + testcheck.section("Test 3: set") + nums.set(1, 99) + v1b: t.CInt = nums.get(1) + stdio.printf("after set(1,99): v1=%d\n", v1b) + testcheck.check(v1b == 99, "set OK (idx1=99)", "set FAILED") + + # === Test 4: pop === + testcheck.section("Test 4: pop") + popped: t.CInt = nums.pop() + stdio.printf("popped=%d\n", popped) + testcheck.check(popped == 30, "pop OK (30)", "pop FAILED expect 30") + n2: t.CSizeT = nums.__len__() + testcheck.check(n2 == 2, "pop len OK (2)", "pop len FAILED expect 2") + + # === Test 5: clear === + testcheck.section("Test 5: clear") + nums.clear() + n3: t.CSizeT = nums.__len__() + testcheck.check(n3 == 0, "clear OK (0)", "clear FAILED expect 0") + + # === Test 6: 容量增长 (append 超过初始容量 8) === + testcheck.section("Test 6: 容量增长") + nums2 = list[int](bd) + i: t.CInt + for i in range(20): + nums2.append(i * 5) + n4: t.CSizeT = nums2.__len__() + stdio.printf("after 20 appends: len=%lu\n", n4) + testcheck.check(n4 == 20, "grow len OK (20)", "grow len FAILED") + ok: t.CInt = 1 + for i in range(20): + v: t.CInt = nums2.get(i) + if v != i * 5: + stdio.printf("MISMATCH at %d: got %d expect %d\n", i, v, i * 5) + ok = 0 + break + testcheck.check(ok == 1, "grow data OK", "grow data FAILED") + + # === Test 7: 多个 list[int] 实例独立 === + testcheck.section("Test 7: 多实例独立") + lst_a = list[int](bd) + lst_b = list[int](bd) + lst_a.append(111) + lst_b.append(222) + va: t.CInt = lst_a.get(0) + vb: t.CInt = lst_b.get(0) + stdio.printf("lst_a[0]=%d lst_b[0]=%d\n", va, vb) + testcheck.check(va == 111 and vb == 222, + "multi-instance OK (111,222)", "multi-instance FAILED") + + # === Test 8: __getitem__ / __setitem__ === + testcheck.section("Test 8: __getitem__/__setitem__") + nums3 = list[int](bd) + nums3.append(1) + nums3.append(2) + nums3.append(3) + nums3[0] = 100 + gv: t.CInt = nums3[0] + stdio.printf("nums3[0]=%d after setitem\n", gv) + testcheck.check(gv == 100, "__setitem__/__getitem__ OK", + "__setitem__/__getitem__ FAILED") + + stdlib.free(arena) + return testcheck.end() \ No newline at end of file diff --git a/TransPyV/Test/App/inherit_test.py b/TransPyV/Test/App/inherit_test.py new file mode 100644 index 0000000..52bc20e --- /dev/null +++ b/TransPyV/Test/App/inherit_test.py @@ -0,0 +1,121 @@ +import stdio +import t, c + + +# ============================================================ +# 继承测试：字段展平 + vtable 继承 + 方法覆盖 +# ============================================================ + + +# ============================================================ +# 基类 Animal（有虚方法） +# ============================================================ +@t.CVTable +class Animal: + name: t.CInt + + def __init__(self, n: t.CInt): + self.name = n + + def GetName(self) -> t.CInt: + return self.name + + def Speak(self) -> t.CInt: + return 0 + + +# ============================================================ +# 子类 Dog（继承 Animal，覆盖 Speak，新增字段和方法） +# ============================================================ +class Dog(Animal): + breed: t.CInt + + def __init__(self, n: t.CInt, b: t.CInt): + self.name = n + self.breed = b + + def Speak(self) -> t.CInt: + return 1 + + def GetBreed(self) -> t.CInt: + return self.breed + + +# ============================================================ +# 子类 Cat（继承 Animal，不覆盖 Speak，新增字段和方法） +# ============================================================ +class Cat(Animal): + color: t.CInt + + def __init__(self, n: t.CInt, col: t.CInt): + self.name = n + self.color = col + + def GetColor(self) -> t.CInt: + return self.color + + +# ============================================================ +# 主函数 +# ============================================================ +def inherit_test() -> int: + # ============================================================ + # 测试 1: 基类 Animal + # ============================================================ + stdio.printf("inherit: === Test 1: Base Class ===\n") + a: Animal = Animal(42) + aname: int = a.GetName() + stdio.printf("inherit: a.GetName()=%d (expected 42)\n", aname) + if aname != 42: + stdio.printf("[FAIL] a.GetName()=%d expected 42\n", aname) + return 1 + aspeak: int = a.Speak() + stdio.printf("inherit: a.Speak()=%d (expected 0)\n", aspeak) + if aspeak != 0: + stdio.printf("[FAIL] a.Speak()=%d expected 0\n", aspeak) + return 1 + + # ============================================================ + # 测试 2: 子类 Dog（覆盖 Speak，继承 GetName） + # ============================================================ + stdio.printf("inherit: === Test 2: Dog (override Speak) ===\n") + d: Dog = Dog(100, 7) + dname: int = d.GetName() + stdio.printf("inherit: d.GetName()=%d (expected 100)\n", dname) + if dname != 100: + stdio.printf("[FAIL] d.GetName()=%d expected 100\n", dname) + return 1 + dspeak: int = d.Speak() + stdio.printf("inherit: d.Speak()=%d (expected 1)\n", dspeak) + if dspeak != 1: + stdio.printf("[FAIL] d.Speak()=%d expected 1\n", dspeak) + return 1 + dbreed: int = d.GetBreed() + stdio.printf("inherit: d.GetBreed()=%d (expected 7)\n", dbreed) + if dbreed != 7: + stdio.printf("[FAIL] d.GetBreed()=%d expected 7\n", dbreed) + return 1 + + # ============================================================ + # 测试 3: 子类 Cat（不覆盖 Speak，继承 GetName） + # ============================================================ + stdio.printf("inherit: === Test 3: Cat (inherit Speak) ===\n") + ct: Cat = Cat(200, 3) + cname: int = ct.GetName() + stdio.printf("inherit: ct.GetName()=%d (expected 200)\n", cname) + if cname != 200: + stdio.printf("[FAIL] ct.GetName()=%d expected 200\n", cname) + return 1 + cspeak: int = ct.Speak() + stdio.printf("inherit: ct.Speak()=%d (expected 0)\n", cspeak) + if cspeak != 0: + stdio.printf("[FAIL] ct.Speak()=%d expected 0\n", cspeak) + return 1 + ccolor: int = ct.GetColor() + stdio.printf("inherit: ct.GetColor()=%d (expected 3)\n", ccolor) + if ccolor != 3: + stdio.printf("[FAIL] ct.GetColor()=%d expected 3\n", ccolor) + return 1 + + stdio.printf("inherit: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/llvmir_test.py b/TransPyV/Test/App/llvmir_test.py new file mode 100644 index 0000000..c7e44a7 --- /dev/null +++ b/TransPyV/Test/App/llvmir_test.py @@ -0,0 +1,247 @@ +import t, c +from stdint import * +import stdio + + +# ============================================================ +# c.LLVMIR / c.LInp / c.LOut 内联 LLVM IR 测试 +# +# 测试 c.LLVMIR 内联 LLVM IR 指令的生成和执行 +# - c.LInp(expr) 标记输入操作数 +# - c.LOut(expr) 标记输出操作数 +# ============================================================ + + +# Test 1: 基本加法 add +def test_add() -> t.CInt: + stdio.printf("--- Test 1: LLVMIR add ---\n") + a: t.CInt = 10 + b: t.CInt = 20 + r: t.CInt = c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("add(10, 20)=%d (expect 30)\n", r) + if r == 30: + stdio.printf("LLVMIR add OK\n") + else: + stdio.printf("LLVMIR add FAIL\n") + return 0 + + +# Test 2: 基本减法 sub +def test_sub() -> t.CInt: + stdio.printf("--- Test 2: LLVMIR sub ---\n") + a: t.CInt = 50 + b: t.CInt = 20 + r: t.CInt = c.LLVMIR(f"sub i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("sub(50, 20)=%d (expect 30)\n", r) + if r == 30: + stdio.printf("LLVMIR sub OK\n") + else: + stdio.printf("LLVMIR sub FAIL\n") + return 0 + + +# Test 3: 基本乘法 mul +def test_mul() -> t.CInt: + stdio.printf("--- Test 3: LLVMIR mul ---\n") + a: t.CInt = 6 + b: t.CInt = 7 + r: t.CInt = c.LLVMIR(f"mul i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("mul(6, 7)=%d (expect 42)\n", r) + if r == 42: + stdio.printf("LLVMIR mul OK\n") + else: + stdio.printf("LLVMIR mul FAIL\n") + return 0 + + +# Test 4: 基本除法 sdiv +def test_sdiv() -> t.CInt: + stdio.printf("--- Test 4: LLVMIR sdiv ---\n") + a: t.CInt = 100 + b: t.CInt = 4 + r: t.CInt = c.LLVMIR(f"sdiv i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("sdiv(100, 4)=%d (expect 25)\n", r) + if r == 25: + stdio.printf("LLVMIR sdiv OK\n") + else: + stdio.printf("LLVMIR sdiv FAIL\n") + return 0 + + +# Test 5: 位运算 and +def test_and() -> t.CInt: + stdio.printf("--- Test 5: LLVMIR and ---\n") + a: t.CInt = 255 + b: t.CInt = 15 + r: t.CInt = c.LLVMIR(f"and i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("and(255, 15)=%d (expect 15)\n", r) + if r == 15: + stdio.printf("LLVMIR and OK\n") + else: + stdio.printf("LLVMIR and FAIL\n") + return 0 + + +# Test 6: 位运算 or +def test_or() -> t.CInt: + stdio.printf("--- Test 6: LLVMIR or ---\n") + a: t.CInt = 240 + b: t.CInt = 15 + r: t.CInt = c.LLVMIR(f"or i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("or(240, 15)=%d (expect 255)\n", r) + if r == 255: + stdio.printf("LLVMIR or OK\n") + else: + stdio.printf("LLVMIR or FAIL\n") + return 0 + + +# Test 7: 位运算 xor +def test_xor() -> t.CInt: + stdio.printf("--- Test 7: LLVMIR xor ---\n") + a: t.CInt = 255 + b: t.CInt = 15 + r: t.CInt = c.LLVMIR(f"xor i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("xor(255, 15)=%d (expect 240)\n", r) + if r == 240: + stdio.printf("LLVMIR xor OK\n") + else: + stdio.printf("LLVMIR xor FAIL\n") + return 0 + + +# Test 8: 左移 shl +def test_shl() -> t.CInt: + stdio.printf("--- Test 8: LLVMIR shl ---\n") + a: t.CInt = 1 + b: t.CInt = 4 + r: t.CInt = c.LLVMIR(f"shl i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("shl(1, 4)=%d (expect 16)\n", r) + if r == 16: + stdio.printf("LLVMIR shl OK\n") + else: + stdio.printf("LLVMIR shl FAIL\n") + return 0 + + +# Test 9: LOut 输出赋值 +def test_lout() -> t.CInt: + stdio.printf("--- Test 9: LLVMIR LOut ---\n") + a: t.CInt = 10 + b: t.CInt = 20 + out_val: t.CInt = 0 + c.LLVMIR(f"{c.LOut(out_val)} = add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("LOut add(10, 20)=%d (expect 30)\n", out_val) + if out_val == 30: + stdio.printf("LLVMIR LOut OK\n") + else: + stdio.printf("LLVMIR LOut FAIL\n") + return 0 + + +# Test 10: icmp eq 比较 +# 注意: TransPyV 用 sext i1 to i32 扩展布尔结果，true -> -1 +def test_icmp_eq() -> t.CInt: + stdio.printf("--- Test 10: LLVMIR icmp eq ---\n") + a: t.CInt = 5 + b: t.CInt = 5 + r: t.CInt = c.LLVMIR(f"icmp eq i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("icmp eq(5, 5)=%d (expect nonzero)\n", r) + if r != 0: + stdio.printf("LLVMIR icmp eq OK\n") + else: + stdio.printf("LLVMIR icmp eq FAIL\n") + return 0 + + +# Test 11: icmp ne 比较 +def test_icmp_ne() -> t.CInt: + stdio.printf("--- Test 11: LLVMIR icmp ne ---\n") + a: t.CInt = 5 + b: t.CInt = 3 + r: t.CInt = c.LLVMIR(f"icmp ne i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("icmp ne(5, 3)=%d (expect nonzero)\n", r) + if r != 0: + stdio.printf("LLVMIR icmp ne OK\n") + else: + stdio.printf("LLVMIR icmp ne FAIL\n") + return 0 + + +# Test 12: icmp slt 比较 +def test_icmp_slt() -> t.CInt: + stdio.printf("--- Test 12: LLVMIR icmp slt ---\n") + a: t.CInt = 3 + b: t.CInt = 5 + r: t.CInt = c.LLVMIR(f"icmp slt i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("icmp slt(3, 5)=%d (expect nonzero)\n", r) + if r != 0: + stdio.printf("LLVMIR icmp slt OK\n") + else: + stdio.printf("LLVMIR icmp slt FAIL\n") + return 0 + + +# Test 13: srem 取余 +def test_srem() -> t.CInt: + stdio.printf("--- Test 13: LLVMIR srem ---\n") + a: t.CInt = 17 + b: t.CInt = 5 + r: t.CInt = c.LLVMIR(f"srem i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("srem(17, 5)=%d (expect 2)\n", r) + if r == 2: + stdio.printf("LLVMIR srem OK\n") + else: + stdio.printf("LLVMIR srem FAIL\n") + return 0 + + +# Test 14: lshr 逻辑右移 +def test_lshr() -> t.CInt: + stdio.printf("--- Test 14: LLVMIR lshr ---\n") + a: t.CInt = 256 + b: t.CInt = 2 + r: t.CInt = c.LLVMIR(f"lshr i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + stdio.printf("lshr(256, 2)=%d (expect 64)\n", r) + if r == 64: + stdio.printf("LLVMIR lshr OK\n") + else: + stdio.printf("LLVMIR lshr FAIL\n") + return 0 + + +# Test 15: 嵌套在表达式中使用 +def test_nested() -> t.CInt: + stdio.printf("--- Test 15: LLVMIR nested in expression ---\n") + a: t.CInt = 10 + b: t.CInt = 20 + # c.LLVMIR 结果参与后续运算 + llvmir_result: t.CInt = c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + final_result: t.CInt = llvmir_result + 10 + stdio.printf("(LLVMIR add(10,20)) + 10 = %d (expect 40)\n", final_result) + if final_result == 40: + stdio.printf("LLVMIR nested OK\n") + else: + stdio.printf("LLVMIR nested FAIL\n") + return 0 + + +def llvmir_test() -> t.CInt: + stdio.printf("=== llvmir_test: c.LLVMIR 内联 LLVM IR 测试 ===\n\n") + test_add() + test_sub() + test_mul() + test_sdiv() + test_and() + test_or() + test_xor() + test_shl() + test_lout() + test_icmp_eq() + test_icmp_ne() + test_icmp_slt() + test_srem() + test_lshr() + test_nested() + stdio.printf("\n=== llvmir_test 完成 ===\n") + return 0 diff --git a/TransPyV/Test/App/memhub_test.py.bak b/TransPyV/Test/App/memhub_test.py.bak new file mode 100644 index 0000000..6cb21a1 --- /dev/null +++ b/TransPyV/Test/App/memhub_test.py.bak @@ -0,0 +1,138 @@ +import stdio +import t, c +import memhub +import stdlib +import string + + +# ============================================================ +# memhub_test - 内存管理库测试 +# +# 测试 MemPool/MemBuddy 的 alloc/free/reset/realloc， +# 以及 MemManager 多态虚分派（通过 __vtable__ 调用子类方法）。 +# ============================================================ + + +def memhub_test() -> int: + stdio.printf("memhub: === Test Start ===\n") + + # === Test 1: MemPool (bump allocator) === + # MemPool 是 arena bump 分配器，不支持单个 free，reset 一次性回收 + stdio.printf("memhub: === Test 1: MemPool ===\n") + + arena1: bytes = stdlib.malloc(4096) + if arena1 is None: + stdio.printf("[FAIL] malloc arena1\n") + return 1 + + pool: memhub.MemPool | t.CPtr = memhub.MemPool(arena1, 4096) + + p1: bytes = pool.alloc(64) + if p1 is None: + stdio.printf("[FAIL] MemPool.alloc(64)\n") + return 1 + + p2: bytes = pool.alloc(128) + if p2 is None: + stdio.printf("[FAIL] MemPool.alloc(128)\n") + return 1 + + stdio.printf("memhub: MemPool alloc ok\n") + + # 写入验证 + string.memset(p1, 66, 64) + if p1[0] != 66: + stdio.printf("[FAIL] MemPool write/read p1[0]=%d\n", p1[0]) + return 1 + + # reset 后可以重新分配 + pool.reset() + p3: bytes = pool.alloc(64) + if p3 is None: + stdio.printf("[FAIL] MemPool.alloc after reset\n") + return 1 + stdio.printf("memhub: MemPool reset ok\n") + + # === Test 2: MemBuddy (buddy system) === + # MemBuddy 是伙伴系统，支持 alloc/free/realloc 合并 + stdio.printf("memhub: === Test 2: MemBuddy ===\n") + + arena2: bytes = stdlib.malloc(65536) + if arena2 is None: + stdio.printf("[FAIL] malloc arena2\n") + return 1 + + buddy: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena2, 65536) + + b1: bytes = buddy.alloc(64) + if b1 is None: + stdio.printf("[FAIL] MemBuddy.alloc(64)\n") + return 1 + + b2: bytes = buddy.alloc(256) + if b2 is None: + stdio.printf("[FAIL] MemBuddy.alloc(256)\n") + return 1 + stdio.printf("memhub: MemBuddy alloc ok\n") + + # 写入验证 + string.memset(b1, 85, 64) + if b1[0] != 85: + stdio.printf("[FAIL] MemBuddy write/read b1[0]=%d\n", b1[0]) + return 1 + + # free b1 后重新 alloc，应能回收内存 + buddy.free(b1) + b3: bytes = buddy.alloc(64) + if b3 is None: + stdio.printf("[FAIL] MemBuddy.alloc after free\n") + return 1 + stdio.printf("memhub: MemBuddy free/alloc ok\n") + + # === Test 3: MemBuddy realloc === + # realloc 扩展内存，验证数据保持 + stdio.printf("memhub: === Test 3: MemBuddy realloc ===\n") + + b4: bytes = buddy.alloc(32) + if b4 is None: + stdio.printf("[FAIL] MemBuddy.alloc(32) for realloc\n") + return 1 + string.memset(b4, 171, 32) + + # realloc 到更大（32 -> 128） + b4_new: bytes = buddy.realloc(b4, 32, 128) + if b4_new is None: + stdio.printf("[FAIL] MemBuddy.realloc\n") + return 1 + + # 验证数据保持（前32字节应保持 171=0xAB） + # 注意：b4_new[0] 是 i8（有符号），0xAB=-85，需 & 255 转无符号再比较 + if (b4_new[0] & 255) != 171: + stdio.printf("[FAIL] MemBuddy.realloc data lost b4_new[0]=%d\n", b4_new[0]) + return 1 + stdio.printf("memhub: MemBuddy realloc ok\n") + + # === Test 4: Polymorphism (MemManager virtual dispatch) === + # 用 MemManager 指针调用 MemBuddy 的 alloc，验证虚分派 + stdio.printf("memhub: === Test 4: Polymorphism ===\n") + + mgr: memhub.MemManager | t.CPtr = (memhub.MemManager | t.CPtr)(buddy) + if mgr is None: + stdio.printf("[FAIL] MemManager cast\n") + return 1 + + # 通过 MemManager 指针调用 alloc（虚分派到 MemBuddy.alloc） + b5: bytes = mgr.alloc(64) + if b5 is None: + stdio.printf("[FAIL] MemManager.alloc (virtual dispatch)\n") + return 1 + + # 验证写入正常 + string.memset(b5, 99, 64) + if b5[0] != 99: + stdio.printf("[FAIL] MemManager.alloc write/read\n") + return 1 + stdio.printf("memhub: Polymorphism ok\n") + + stdio.printf("memhub: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/namespace_defs.py b/TransPyV/Test/App/namespace_defs.py new file mode 100644 index 0000000..24e8e63 --- /dev/null +++ b/TransPyV/Test/App/namespace_defs.py @@ -0,0 +1,60 @@ +import stdio +import t, c + + +# ============================================================ +# 命名空间隔离测试：定义类模块 +# +# 本文件定义供其他文件 import 使用的类，验证跨模块命名空间隔离： +# - 裸名 Class(): 需要 from namespace_defs import Class +# - 模块限定 namespace_defs.Class(): 需要 import namespace_defs +# ============================================================ + + +# ============================================================ +# 带虚表的类 Widget +# ============================================================ +@t.CVTable +class Widget: + id: t.CInt + + def __init__(self, i: t.CInt): + self.id = i + + def GetId(self) -> t.CInt: + return self.id + + def Render(self) -> t.CInt: + return 100 + + +# ============================================================ +# 子类 Gadget（继承 Widget，覆盖 Render） +# ============================================================ +class Gadget(Widget): + extra: t.CInt + + def __init__(self, i: t.CInt, e: t.CInt): + self.id = i + self.extra = e + + def Render(self) -> t.CInt: + return 200 + + def GetExtra(self) -> t.CInt: + return self.extra + + +# ============================================================ +# 普通结构体 PlainStruct（无虚表） +# ============================================================ +class PlainStruct: + x: t.CInt + y: t.CInt + + def __init__(self, ax: t.CInt, ay: t.CInt): + self.x = ax + self.y = ay + + def Sum(self) -> t.CInt: + return self.x + self.y diff --git a/TransPyV/Test/App/namespace_test.py b/TransPyV/Test/App/namespace_test.py new file mode 100644 index 0000000..396a901 --- /dev/null +++ b/TransPyV/Test/App/namespace_test.py @@ -0,0 +1,111 @@ +import stdio +import t, c +import namespace_defs +from namespace_defs import Widget, Gadget, PlainStruct + + +# ============================================================ +# 命名空间隔离测试 +# +# 验证跨模块类访问必须通过 import： +# - Test 1: from X import Class → 裸名 Class() 构造 +# - Test 2: import X → 模块限定 X.Class() 构造 +# - Test 3: 类型注解使用导入的类 +# - Test 4: 继承类的虚分派（跨模块继承） +# ============================================================ + + +def namespace_test() -> int: + stdio.printf("nsisol: === Test Start ===\n") + + # ============================================================ + # Test 1: from X import Class → 裸名 Class() 构造 + # ============================================================ + stdio.printf("nsisol: === Test 1: from-import bare constructor ===\n") + + w: Widget = Widget(42) + wid: int = w.GetId() + stdio.printf("nsisol: w.GetId()=%d (expected 42)\n", wid) + if wid != 42: + stdio.printf("[FAIL] w.GetId()=%d expected 42\n", wid) + return 1 + + wrender: int = w.Render() + stdio.printf("nsisol: w.Render()=%d (expected 100)\n", wrender) + if wrender != 100: + stdio.printf("[FAIL] w.Render()=%d expected 100\n", wrender) + return 1 + + g: Gadget = Gadget(7, 9) + grender: int = g.Render() + stdio.printf("nsisol: g.Render()=%d (expected 200)\n", grender) + if grender != 200: + stdio.printf("[FAIL] g.Render()=%d expected 200\n", grender) + return 1 + + gextra: int = g.GetExtra() + stdio.printf("nsisol: g.GetExtra()=%d (expected 9)\n", gextra) + if gextra != 9: + stdio.printf("[FAIL] g.GetExtra()=%d expected 9\n", gextra) + return 1 + + # ============================================================ + # Test 2: import X → 模块限定 X.Class() 构造 + # ============================================================ + stdio.printf("nsisol: === Test 2: module-qualified constructor ===\n") + + ps: PlainStruct = PlainStruct(10, 20) + psum: int = ps.Sum() + stdio.printf("nsisol: ps.Sum()=%d (expected 30)\n", psum) + if psum != 30: + stdio.printf("[FAIL] ps.Sum()=%d expected 30\n", psum) + return 1 + + ps2: namespace_defs.PlainStruct = namespace_defs.PlainStruct(5, 6) + psum2: int = ps2.Sum() + stdio.printf("nsisol: ps2.Sum()=%d (expected 11)\n", psum2) + if psum2 != 11: + stdio.printf("[FAIL] ps2.Sum()=%d expected 11\n", psum2) + return 1 + + # ============================================================ + # Test 3: 类型注解使用导入的类 + # ============================================================ + stdio.printf("nsisol: === Test 3: type annotation with imported class ===\n") + + annot_w: Widget = Widget(99) + annot_id: int = annot_w.GetId() + stdio.printf("nsisol: annot_w.GetId()=%d (expected 99)\n", annot_id) + if annot_id != 99: + stdio.printf("[FAIL] annot_w.GetId()=%d expected 99\n", annot_id) + return 1 + + # ============================================================ + # Test 4: 跨模块继承的虚分派 + # ============================================================ + stdio.printf("nsisol: === Test 4: cross-module inheritance vtable ===\n") + + # Gadget 继承 Widget，覆盖 Render + # Widget.Render()=100, Gadget.Render()=200 + base: Widget = Widget(1) + derived: Gadget = Gadget(2, 3) + base_r: int = base.Render() + deriv_r: int = derived.Render() + stdio.printf("nsisol: base.Render()=%d (expected 100)\n", base_r) + if base_r != 100: + stdio.printf("[FAIL] base.Render()=%d expected 100\n", base_r) + return 1 + stdio.printf("nsisol: derived.Render()=%d (expected 200)\n", deriv_r) + if deriv_r != 200: + stdio.printf("[FAIL] derived.Render()=%d expected 200\n", deriv_r) + return 1 + + # 继承的方法：Gadget 继承 Widget.GetId + gname: int = derived.GetId() + stdio.printf("nsisol: derived.GetId()=%d (expected 2)\n", gname) + if gname != 2: + stdio.printf("[FAIL] derived.GetId()=%d expected 2\n", gname) + return 1 + + stdio.printf("nsisol: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/new_test.py b/TransPyV/Test/App/new_test.py new file mode 100644 index 0000000..1cb9600 --- /dev/null +++ b/TransPyV/Test/App/new_test.py @@ -0,0 +1,119 @@ +import stdio +import t, c + + +# ============================================================ +# __new__ 函数测试：验证 __new__ 在创建 OOP 结构体前被调用， +# 返回的指针作为结构体存储空间。 +# +# __new__ 签名默认和 __init__ 一样（self + args），返回 Ptr(struct_ty)。 +# 如果 __new__ 返回 self，则使用默认 alloca 作为存储空间。 +# ============================================================ + + +# ============================================================ +# Test 1: __new__ 返回 self（默认 alloca） +# 验证 __new__ 被调用且不破坏 __init__ 正常功能 +# ============================================================ +class WithNew: + value: t.CInt + + def __new__(self, v: t.CInt): + stdio.printf("new: WithNew.__new__ called\n") + return self + + def __init__(self, v: t.CInt): + stdio.printf("new: WithNew.__init__ called\n") + self.value = v + + def GetValue(self) -> t.CInt: + return self.value + + +# ============================================================ +# Test 2: __new__ 不接受额外参数（仅 self） +# 验证 __new__ 签名可以与 __init__ 不同 +# ============================================================ +class WithNewNoArgs: + value: t.CInt + + def __new__(self): + stdio.printf("new: WithNewNoArgs.__new__ called\n") + return self + + def __init__(self, v: t.CInt): + self.value = v + + def GetValue(self) -> t.CInt: + return self.value + + +# ============================================================ +# Test 3: __new__ + @t.CVTable 虚表类 +# 验证 __new__ 与虚表机制兼容（__new__ 不进入虚表） +# ============================================================ +@t.CVTable +class WithNewVTable: + value: t.CInt + + def __new__(self, v: t.CInt): + stdio.printf("new: WithNewVTable.__new__ called\n") + return self + + def __init__(self, v: t.CInt): + self.value = v + + def GetValue(self) -> t.CInt: + return self.value + + def Speak(self) -> t.CInt: + return self.value + 1 + + +def new_test() -> int: + stdio.printf("new: === Test Start ===\n") + + # ============================================================ + # Test 1: __new__ 返回 self + # ============================================================ + stdio.printf("new: === Test 1: __new__ returns self ===\n") + + w: WithNew = WithNew(42) + v: int = w.GetValue() + stdio.printf("new: w.GetValue()=%d (expected 42)\n", v) + if v != 42: + stdio.printf("[FAIL] w.GetValue()=%d expected 42\n", v) + return 1 + + # ============================================================ + # Test 2: __new__ 不接受额外参数 + # ============================================================ + stdio.printf("new: === Test 2: __new__ no extra args ===\n") + + n: WithNewNoArgs = WithNewNoArgs(99) + nv: int = n.GetValue() + stdio.printf("new: n.GetValue()=%d (expected 99)\n", nv) + if nv != 99: + stdio.printf("[FAIL] n.GetValue()=%d expected 99\n", nv) + return 1 + + # ============================================================ + # Test 3: __new__ + 虚表类 + # ============================================================ + stdio.printf("new: === Test 3: __new__ + vtable ===\n") + + vt: WithNewVTable = WithNewVTable(10) + vtv: int = vt.GetValue() + stdio.printf("new: vt.GetValue()=%d (expected 10)\n", vtv) + if vtv != 10: + stdio.printf("[FAIL] vt.GetValue()=%d expected 10\n", vtv) + return 1 + + vts: int = vt.Speak() + stdio.printf("new: vt.Speak()=%d (expected 11)\n", vts) + if vts != 11: + stdio.printf("[FAIL] vt.Speak()=%d expected 11\n", vts) + return 1 + + stdio.printf("new: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/oop_test.py b/TransPyV/Test/App/oop_test.py new file mode 100644 index 0000000..ebf66dd --- /dev/null +++ b/TransPyV/Test/App/oop_test.py @@ -0,0 +1,129 @@ +import stdio +import t, c + + +# ============================================================ +# OOP 测试：存在方法的 class 自动升级为 OOP 结构体 +# ============================================================ + +# ============================================================ +# 测试 1: 基本方法（无 __init__，__before_init__ 零值填充） +# ============================================================ +class Point: + x: t.CInt + y: t.CInt + + def MoveTo(self, nx: t.CInt, ny: t.CInt) -> t.CInt: + self.x = nx + self.y = ny + return 0 + + def GetX(self) -> t.CInt: + return self.x + + def GetY(self) -> t.CInt: + return self.y + + +# ============================================================ +# 测试 2: 带默认值的 OOP 结构体（__before_init__ 应用默认值） +# ============================================================ +class Counter: + count: t.CInt = 0 + step: t.CInt = 1 + + def Increment(self) -> t.CInt: + self.count = self.count + self.step + return self.count + + def Reset(self) -> t.CInt: + self.count = 0 + return 0 + + +# ============================================================ +# 测试 3: __init__ 构造函数 +# ============================================================ +class Rect: + width: t.CInt + height: t.CInt + + def __init__(self, w: t.CInt, h: t.CInt): + self.width = w + self.height = h + + def Area(self) -> t.CInt: + return self.width * self.height + + +# ============================================================ +# 主函数 +# ============================================================ +def oop_test() -> int: + # ============================================================ + # 测试 1: 基本方法调用 + # ============================================================ + stdio.printf("oop: === Test 1: Basic Methods ===\n") + p: Point = Point() + p.MoveTo(10, 20) + stdio.printf("oop: p.x=%d p.y=%d\n", p.GetX(), p.GetY()) + + if p.GetX() != 10: + stdio.printf("[FAIL] p.GetX()=%d expected 10\n", p.GetX()) + return 1 + if p.GetY() != 20: + stdio.printf("[FAIL] p.GetY()=%d expected 20\n", p.GetY()) + return 1 + + # 修改字段后再次调用方法 + p.MoveTo(100, 200) + stdio.printf("oop: after move p.x=%d p.y=%d\n", p.GetX(), p.GetY()) + if p.GetX() != 100: + return 1 + + # ============================================================ + # 测试 2: 默认值 + 方法 + # ============================================================ + stdio.printf("oop: === Test 2: Defaults + Methods ===\n") + cnt: Counter = Counter() + stdio.printf("oop: initial count=%d step=%d\n", cnt.count, cnt.step) + if cnt.count != 0: + stdio.printf("[FAIL] cnt.count=%d expected 0\n", cnt.count) + return 1 + if cnt.step != 1: + stdio.printf("[FAIL] cnt.step=%d expected 1\n", cnt.step) + return 1 + + r1: int = cnt.Increment() + r2: int = cnt.Increment() + r3: int = cnt.Increment() + stdio.printf("oop: after 3 increments: %d %d %d\n", r1, r2, r3) + if r1 != 1 or r2 != 2 or r3 != 3: + stdio.printf("[FAIL] increments: %d %d %d\n", r1, r2, r3) + return 1 + + cnt.Reset() + r4: int = cnt.Increment() + stdio.printf("oop: after reset+increment: %d\n", r4) + if r4 != 1: + return 1 + + # ============================================================ + # 测试 3: __init__ 构造函数 + # ============================================================ + stdio.printf("oop: === Test 3: __init__ ===\n") + rect: Rect = Rect(3, 4) + area: int = rect.Area() + stdio.printf("oop: rect %dx%d area=%d\n", rect.width, rect.height, area) + if area != 12: + stdio.printf("[FAIL] area=%d expected 12\n", area) + return 1 + if rect.width != 3: + stdio.printf("[FAIL] width=%d expected 3\n", rect.width) + return 1 + if rect.height != 4: + stdio.printf("[FAIL] height=%d expected 4\n", rect.height) + return 1 + + stdio.printf("oop: === All OOP Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/opovl_test.py b/TransPyV/Test/App/opovl_test.py new file mode 100644 index 0000000..3a68afc --- /dev/null +++ b/TransPyV/Test/App/opovl_test.py @@ -0,0 +1,253 @@ +import stdio +import stdlib +import t, c +import testcheck + + +# ============================================================ +# 运算符重载测试 +# +# 验证: 当 lhs 是结构体指针时，二元/比较运算触发 dunder 方法调用 +# 支持: __add__/__sub__/__mul__/__div__/__mod__/__and__/__or__/__xor__ +# __lshift__/__rshift__/__floordiv__ (BinOp) +# __eq__/__ne__/__lt__/__le__/__gt__/__ge__ (Compare) +# +# 三种测试场景: +# 1. Counter/BitBox: dunder 返回 t.CInt（值类型变量触发重载） +# 2. Vec2: dunder 返回 Vec2|t.CPtr（堆分配结构体，指针类型变量触发重载） +# 3. 原生整数运算回退 +# ============================================================ + + +# ============================================================ +# Counter: 演示 __add__/__mul__/__sub__ + 比较重载 +# dunder 返回 t.CInt +# ============================================================ +class Counter: + val: t.CInt + + def __init__(self, v: t.CInt): + self.val = v + + def __add__(self, n: t.CInt) -> t.CInt: + return self.val + n + + def __sub__(self, n: t.CInt) -> t.CInt: + return self.val - n + + def __mul__(self, n: t.CInt) -> t.CInt: + return self.val * n + + def __div__(self, n: t.CInt) -> t.CInt: + return self.val / n + + def __mod__(self, n: t.CInt) -> t.CInt: + return self.val % n + + def __eq__(self, n: t.CInt) -> t.CInt: + if self.val == n: + return 1 + return 0 + + def __ne__(self, n: t.CInt) -> t.CInt: + if self.val != n: + return 1 + return 0 + + def __lt__(self, n: t.CInt) -> t.CInt: + if self.val < n: + return 1 + return 0 + + def __le__(self, n: t.CInt) -> t.CInt: + if self.val <= n: + return 1 + return 0 + + def __gt__(self, n: t.CInt) -> t.CInt: + if self.val > n: + return 1 + return 0 + + def __ge__(self, n: t.CInt) -> t.CInt: + if self.val >= n: + return 1 + return 0 + + +# ============================================================ +# BitBox: 演示位运算重载 +# dunder 返回 t.CInt +# ============================================================ +class BitBox: + flags: t.CInt + + def __init__(self, v: t.CInt): + self.flags = v + + def __and__(self, mask: t.CInt) -> t.CInt: + return self.flags & mask + + def __or__(self, mask: t.CInt) -> t.CInt: + return self.flags | mask + + def __xor__(self, mask: t.CInt) -> t.CInt: + return self.flags ^ mask + + def __lshift__(self, n: t.CInt) -> t.CInt: + return self.flags << n + + def __rshift__(self, n: t.CInt) -> t.CInt: + return self.flags >> n + + +# ============================================================ +# Vec2: 演示结构体运算重载 +# +# __new__ + stdlib.malloc 在堆上分配内存，使 Vec2|t.CPtr 变量 +# 能正确工作（dunder 方法返回堆上新对象）。 +# +# __new__ 返回 malloc 的堆指针作为结构体存储空间， +# 后续 __init__ 在堆指针上初始化字段。 +# ============================================================ +class Vec2: + x: t.CInt + y: t.CInt + + def __new__(self, x0: t.CInt, y0: t.CInt): + r: Vec2 | t.CPtr = stdlib.malloc(Vec2.__sizeof__()) + return r + + def __init__(self, x0: t.CInt, y0: t.CInt): + self.x = x0 + self.y = y0 + + def __add__(self, other: Vec2 | t.CPtr) -> Vec2 | t.CPtr: + r: Vec2 | t.CPtr = Vec2(0, 0) + r.x = self.x + other.x + r.y = self.y + other.y + return r + + def __sub__(self, other: Vec2 | t.CPtr) -> Vec2 | t.CPtr: + r: Vec2 | t.CPtr = Vec2(0, 0) + r.x = self.x - other.x + r.y = self.y - other.y + return r + + def __eq__(self, other: Vec2 | t.CPtr) -> t.CInt: + if self.x == other.x: + if self.y == other.y: + return 1 + return 0 + + def __ne__(self, other: Vec2 | t.CPtr) -> t.CInt: + if self.x != other.x: + return 1 + if self.y != other.y: + return 1 + return 0 + + +# ============================================================ +# 主测试入口 +# ============================================================ +def opovl_test() -> int: + testcheck.begin("Operator Overload Tests") + + # ============================================================ + # 测试 1: Counter + int 触发 __add__ + # ============================================================ + testcheck.section("Counter __add__") + cnt: Counter = Counter(10) + sum_val: int = cnt + 5 + testcheck.check(sum_val == 15, "cnt(10)+5=15", "cnt+5 expected 15 got ??") + + # ============================================================ + # 测试 2: Counter - int / * int / / int / % int + # ============================================================ + testcheck.section("Counter __sub__/__mul__/__div__/__mod__") + sub_val: int = cnt - 3 + mul_val: int = cnt * 3 + div_val: int = cnt / 3 + mod_val: int = cnt % 3 + testcheck.check(sub_val == 7, "cnt-3=7", "cnt-3 expected 7") + testcheck.check(mul_val == 30, "cnt*3=30", "cnt*3 expected 30") + testcheck.check(div_val == 3, "cnt/3=3", "cnt/3 expected 3") + testcheck.check(mod_val == 1, "cnt%3=1", "cnt%3 expected 1") + + # ============================================================ + # 测试 3: Counter 比较重载 (==, !=, <, <=, >, >=) + # ============================================================ + testcheck.section("Counter compare overloads") + eq_10: int = cnt == 10 + ne_10: int = cnt != 10 + lt_20: int = cnt < 20 + le_10: int = cnt <= 10 + gt_5: int = cnt > 5 + ge_10: int = cnt >= 10 + testcheck.check(eq_10 == 1, "cnt==10:1", "cnt==10 expected 1") + testcheck.check(ne_10 == 0, "cnt!=10:0", "cnt!=10 expected 0") + testcheck.check(lt_20 == 1, "cnt<20:1", "cnt<20 expected 1") + testcheck.check(le_10 == 1, "cnt<=10:1", "cnt<=10 expected 1") + testcheck.check(gt_5 == 1, "cnt>5:1", "cnt>5 expected 1") + testcheck.check(ge_10 == 1, "cnt>=10:1", "cnt>=10 expected 1") + + # ============================================================ + # 测试 4: BitBox 位运算重载 + # ============================================================ + testcheck.section("BitBox bitwise overloads") + bb: BitBox = BitBox(0xFF) + and_val: int = bb & 0x0F + or_val: int = bb | 0x100 + xor_val: int = bb ^ 0x55 + lsh_val: int = bb << 4 + rsh_val: int = bb >> 4 + testcheck.check(and_val == 0x0F, "bb&0x0F=15", "bb&0x0F expected 15") + testcheck.check(or_val == 0x1FF, "bb|0x100=511", "bb|0x100 expected 511") + testcheck.check(xor_val == 0xAA, "bb^0x55=170", "bb^0x55 expected 170") + testcheck.check(lsh_val == 0xFF0, "bb<<4=4080", "bb<<4 expected 4080") + testcheck.check(rsh_val == 0x0F, "bb>>4=15", "bb>>4 expected 15") + + # ============================================================ + # 测试 5: 原生整数运算不受影响（回退路径） + # ============================================================ + testcheck.section("Native int binop") + x: int = 7 + 8 + y: int = 10 - 3 + z: int = 4 * 5 + testcheck.check(x == 15, "7+8=15", "7+8 expected 15") + testcheck.check(y == 7, "10-3=7", "10-3 expected 7") + testcheck.check(z == 20, "4*5=20", "4*5 expected 20") + + # ============================================================ + # 测试 6: 原生整数比较不受影响 + # ============================================================ + testcheck.section("Native int compare") + testcheck.check(3 < 5, "3<5", "3<5 broken") + testcheck.check(10 == 10, "10==10", "10==10 broken") + + # ============================================================ + # 测试 7: Vec2 结构体运算重载 + # ============================================================ + testcheck.section("Vec2 struct overloads") + a: Vec2 | t.CPtr = Vec2(3, 4) + b: Vec2 | t.CPtr = Vec2(1, 2) + c: Vec2 | t.CPtr = a + b + d: Vec2 | t.CPtr = a - b + testcheck.check(c.x == 4, "a+b x=4", "a+b x expected 4") + testcheck.check(c.y == 6, "a+b y=6", "a+b y expected 6") + testcheck.check(d.x == 2, "a-b x=2", "a-b x expected 2") + testcheck.check(d.y == 2, "a-b y=2", "a-b y expected 2") + + # ============================================================ + # 测试 8: Vec2 比较重载 + # ============================================================ + testcheck.section("Vec2 compare overloads") + p: Vec2 | t.CPtr = Vec2(5, 5) + q: Vec2 | t.CPtr = Vec2(5, 5) + eq_v: int = p == q + ne_v: int = p != q + testcheck.check(eq_v == 1, "(5,5)==(5,5):1", "p==q expected 1") + testcheck.check(ne_v == 0, "(5,5)!=(5,5):0", "p!=q expected 0") + + return testcheck.end() diff --git a/TransPyV/Test/App/ptr_only_test.py b/TransPyV/Test/App/ptr_only_test.py new file mode 100644 index 0000000..5050881 --- /dev/null +++ b/TransPyV/Test/App/ptr_only_test.py @@ -0,0 +1,26 @@ +import stdio +import t, c + + +def ptr_only_test() -> int: + s: str = "hello" + sp: str = s + slen: int = 0 + while c.Deref(sp) != 0: + slen = slen + 1 + sp = sp + 1 + stdio.printf("ptr: len(hello)=%d\n", slen) + + x: int = 65 + px: str = c.Addr(x) + c.DerefAs(px, 88) + stdio.printf("ptr: after DerefAs x=%d\n", x) + + p2: str = "ABC" + total_ch: int = 0 + while c.Deref(p2) != 0: + total_ch = total_ch + c.Deref(p2) + p2 = p2 + 1 + stdio.printf("ptr: sum(A,B,C)=%d\n", total_ch) + + return 0 diff --git a/TransPyV/Test/App/ptr_test.py b/TransPyV/Test/App/ptr_test.py new file mode 100644 index 0000000..cf6fb02 --- /dev/null +++ b/TransPyV/Test/App/ptr_test.py @@ -0,0 +1,111 @@ +import stdio +import t, c + + +# ============================================================ +# 简单指针 + 逻辑测试 +# ============================================================ + +def ptr_test() -> int: + # 逻辑测试：if 语句 + x: int = 10 + if x: + stdio.printf("if: x is truthy\n") + + if 1 == 1: + stdio.printf("if: 1==1 is truthy\n") + + # While 循环 + 累加 + i: int = 0 + total: int = 0 + while i < 5: + total = total + i + i = i + 1 + stdio.printf("while: total=%d i=%d\n", total, i) + + # For 循环 + for j in range(5): + stdio.printf("for: j=%d\n", j) + + # Break 测试 + for k in range(10): + if k == 3: + break + stdio.printf("break: k=%d\n", k) + + # Continue 测试 + for m in range(5): + if m == 2: + continue + stdio.printf("continue: m=%d\n", m) + + # 布尔运算 and（短路求值） + if x > 5 and x < 100: + stdio.printf("bool: x>5 and x<100 is true\n") + + # 布尔运算 or（短路求值） + if x > 100 or x > 5: + stdio.printf("bool: x>100 or x>5 is true\n") + + # ============================================================ + # 指针 + c.Deref / c.DerefAs 测试 + # ============================================================ + + # 字符串遍历: while c.Deref(p) != 0 + s: str = "hello" + sp: str = s + slen: int = 0 + while c.Deref(sp) != 0: + slen = slen + 1 + sp = sp + 1 + stdio.printf("ptr: len(hello)=%d\n", slen) + + # c.DerefAs 写入字符（使用 c.Addr 获取栈变量地址，字符串字面量是只读的） + v: int = 65 # 'A' + px: str = c.Addr(v) + c.DerefAs(px, 88) # *px = 88 + stdio.printf("ptr: after DerefAs v=%d\n", v) + + # 遍历字符串并累加字符值 + p2: str = "ABC" + total_ch: int = 0 + while c.Deref(p2) != 0: + total_ch = total_ch + c.Deref(p2) + p2 = p2 + 1 + stdio.printf("ptr: sum(A,B,C)=%d\n", total_ch) + + # ============================================================ + # c.Load 测试: *a = *b（加载源指针的值，存储到目标指针） + # ============================================================ + + # 基本复制: v2 = v1 + v1: int = 42 + v2: int = 0 + c.Load(c.Addr(v2), c.Addr(v1)) + stdio.printf("ptr: c.Load v2=%d (expected 42)\n", v2) + if v2 != 42: + stdio.printf("[FAIL] c.Load expected 42 got %d\n", v2) + return 1 + + # 覆盖已有值: v3 = v4 + v3: int = 100 + v4: int = 200 + c.Load(c.Addr(v3), c.Addr(v4)) + stdio.printf("ptr: c.Load overwrite v3=%d (expected 200)\n", v3) + if v3 != 200: + stdio.printf("[FAIL] c.Load overwrite expected 200 got %d\n", v3) + return 1 + + # 验证源不被修改: v5 保持原值 + v5: int = 999 + v6: int = 0 + c.Load(c.Addr(v6), c.Addr(v5)) + stdio.printf("ptr: c.Load src unchanged v5=%d v6=%d (expected 999, 999)\n", v5, v6) + if v5 != 999: + stdio.printf("[FAIL] c.Load src changed v5=%d\n", v5) + return 1 + if v6 != 999: + stdio.printf("[FAIL] c.Load dst expected 999 got %d\n", v6) + return 1 + + return 0 diff --git a/TransPyV/Test/App/simple_test.py b/TransPyV/Test/App/simple_test.py new file mode 100644 index 0000000..bc5c64e --- /dev/null +++ b/TransPyV/Test/App/simple_test.py @@ -0,0 +1,6 @@ +import stdio + +def simple_test() -> int: + stdio.printf("hello\n") + return 0 + diff --git a/TransPyV/Test/App/string_min_test.py b/TransPyV/Test/App/string_min_test.py new file mode 100644 index 0000000..5be13ea --- /dev/null +++ b/TransPyV/Test/App/string_min_test.py @@ -0,0 +1,9 @@ +import stdio +import t, c +import string + + +def string_min_test() -> int: + l: t.CSizeT = string.strlen("Hello") + stdio.printf("strlen(Hello)=%lu\n", l) + return 0 diff --git a/TransPyV/Test/App/string_test.py b/TransPyV/Test/App/string_test.py new file mode 100644 index 0000000..3880db4 --- /dev/null +++ b/TransPyV/Test/App/string_test.py @@ -0,0 +1,106 @@ +import stdio +import t, c +import string + + +# ============================================================ +# string_test - string.py 库函数测试 +# +# 测试 strlen / strcmp / strncmp / atoi / strchr / strstr +# ============================================================ + + +def test_strlen(): + stdio.printf("--- Test 1: strlen ---\n") + + l1: t.CSizeT = string.strlen("Hello") + stdio.printf("strlen(Hello)=%lu (expect 5)\n", l1) + + l2: t.CSizeT = string.strlen("") + stdio.printf("strlen()=%lu (expect 0)\n", l2) + + l3: t.CSizeT = string.strlen("Hello, World!") + stdio.printf("strlen(Hello, World!)=%lu (expect 13)\n", l3) + + +def test_strcmp(): + stdio.printf("--- Test 2: strcmp ---\n") + + r1: t.CInt = string.strcmp("abc", "abc") + stdio.printf("strcmp(abc,abc)=%d (expect 0)\n", r1) + + r2: t.CInt = string.strcmp("abc", "abd") + stdio.printf("strcmp(abc,abd)=%d (expect <0)\n", r2) + + r3: t.CInt = string.strcmp("abd", "abc") + stdio.printf("strcmp(abd,abc)=%d (expect >0)\n", r3) + + +def test_strncmp(): + stdio.printf("--- Test 3: strncmp ---\n") + + r1: t.CInt = string.strncmp("abcdef", "abcXYZ", 3) + stdio.printf("strncmp(abcdef,abcXYZ,3)=%d (expect 0)\n", r1) + + r2: t.CInt = string.strncmp("abcdef", "abcXYZ", 4) + stdio.printf("strncmp(abcdef,abcXYZ,4)=%d (expect <0)\n", r2) + + +def test_atoi(): + stdio.printf("--- Test 4: atoi ---\n") + + n1: t.CInt = string.atoi("123") + stdio.printf("atoi(123)=%d (expect 123)\n", n1) + + n2: t.CInt = string.atoi("-456") + stdio.printf("atoi(-456)=%d (expect -456)\n", n2) + + n3: t.CInt = string.atoi(" 789") + stdio.printf("atoi( 789)=%d (expect 789)\n", n3) + + +def test_strchr(): + stdio.printf("--- Test 5: strchr ---\n") + + # strchr 返回指向字符的指针，不为 None 表示找到 + p: str = string.strchr("Hello", 108) # 'l' = 108 + if p is not None: + stdio.printf("strchr(Hello,'l') found, char=%d (expect 108)\n", c.Deref(p)) + else: + stdio.printf("strchr(Hello,'l') NOT FOUND (FAIL)\n") + + p2: str = string.strchr("Hello", 122) # 'z' = 122 + if p2 is None: + stdio.printf("strchr(Hello,'z') not found OK\n") + else: + stdio.printf("strchr(Hello,'z') FAIL (should be None)\n") + + +def test_strstr(): + stdio.printf("--- Test 6: strstr ---\n") + + p: str = string.strstr("Hello World", "World") + if p is not None: + stdio.printf("strstr(Hello,World) found OK\n") + else: + stdio.printf("strstr(Hello,World) NOT FOUND (FAIL)\n") + + p2: str = string.strstr("Hello World", "xyz") + if p2 is None: + stdio.printf("strstr(Hello,xyz) not found OK\n") + else: + stdio.printf("strstr(Hello,xyz) FAIL (should be None)\n") + + +def string_test() -> int: + stdio.printf("=== string_test: string.py 库函数测试 ===\n\n") + + test_strlen() + test_strcmp() + test_strncmp() + test_atoi() + test_strchr() + test_strstr() + + stdio.printf("\n=== string_test 完成 ===\n") + return 0 diff --git a/TransPyV/Test/App/struct_test.py b/TransPyV/Test/App/struct_test.py new file mode 100644 index 0000000..2b883e0 --- /dev/null +++ b/TransPyV/Test/App/struct_test.py @@ -0,0 +1,235 @@ +import stdio +import t, c + + +# ============================================================ +# 纯内存结构体（class）测试 +# ============================================================ + +class Point: + x: t.CInt + y: t.CInt + + +class Box: + width: t.CInt + height: t.CInt + depth: t.CInt + + +class Size: + w: t.CInt = 100 + h: t.CInt = 200 + + +# ============================================================ +# 枚举（t.CEnum）测试 +# ============================================================ + +class State(t.CEnum): + Idle: t.State + Run: t.State + Stop: t.State + + +class Color(t.CEnum): + Red: t.State = 10 + Green: t.State + Blue: t.State = 20 + Yellow: t.State + + +class MixedType(t.CEnum): + Small: t.CInt8T + Big: t.CInt64T + Medium: t.CInt16T + + +# ============================================================ +# 联合体（t.CUnion）测试 +# ============================================================ + +class DataUnion(t.CUnion): + i: t.CInt + f: t.CFloat + l: t.CInt64T + + +# ============================================================ +# 主函数 +# ============================================================ +def struct_test() -> int: + # ============================================================ + # 测试 1: 结构体字段读写 + # ============================================================ + p: Point = Point() + p.x = 10 + p.y = 20 + stdio.printf("struct: p.x=%d p.y=%d\n", p.x, p.y) + + # 修改字段 + p.x = 100 + p.y = 200 + stdio.printf("struct: modified p.x=%d p.y=%d\n", p.x, p.y) + + # 字段运算 + sum_xy: int = p.x + p.y + stdio.printf("struct: sum=%d\n", sum_xy) + + # ============================================================ + # 测试 2: 多字段结构体 + # ============================================================ + b: Box + b.width = 3 + b.height = 4 + b.depth = 5 + vol: int = b.width * b.height * b.depth + stdio.printf("struct: volume=%d\n", vol) + + # ============================================================ + # 测试 3: t.CArray 数组遍历 + # ============================================================ + arr: t.CArray[t.CInt, 5] + arr[0] = 100 + arr[1] = 200 + arr[2] = 300 + arr[3] = 400 + arr[4] = 500 + + arr_total: int = 0 + for i in range(5): + arr_total = arr_total + arr[i] + stdio.printf("array: total=%d\n", arr_total) + + # 修改数组元素 + arr[2] = 999 + stdio.printf("array: arr[2]=%d\n", arr[2]) + + # ============================================================ + # 测试 4: 指针遍历（字符串） + # ============================================================ + s: str = "hello" + sp: str = s + slen: int = 0 + while c.Deref(sp) != 0: + slen = slen + 1 + sp = sp + 1 + stdio.printf("ptr: len(hello)=%d\n", slen) + + # 累加字符值 + p2: str = "ABC" + total_ch: int = 0 + while c.Deref(p2) != 0: + total_ch = total_ch + c.Deref(p2) + p2 = p2 + 1 + stdio.printf("ptr: sum(A,B,C)=%d\n", total_ch) + + # ============================================================ + # 测试 5: 结构体构造函数 Point() / Point(x, y) + # ============================================================ + # 无参数构造：零初始化 + z: Point = Point() + stdio.printf("ctor: z.x=%d z.y=%d\n", z.x, z.y) + + # 带参数构造：按位置赋值 + p3: Point = Point(7, 8) + stdio.printf("ctor: p3.x=%d p3.y=%d\n", p3.x, p3.y) + + # 多字段构造 + bx: Box = Box(10, 20, 30) + stdio.printf("ctor: bx.w=%d bx.h=%d bx.d=%d\n", bx.width, bx.height, bx.depth) + + # ============================================================ + # 测试 6: 结构体关键字参数（乱序传参） + # ============================================================ + # 全关键字乱序 + p4: Point = Point(y=20, x=10) + stdio.printf("kw: p4.x=%d p4.y=%d\n", p4.x, p4.y) + + # 混合：位置 + 关键字 + p5: Point = Point(5, y=15) + stdio.printf("kw: p5.x=%d p5.y=%d\n", p5.x, p5.y) + + # 多字段关键字乱序 + bx2: Box = Box(depth=30, width=10, height=20) + stdio.printf("kw: bx2.w=%d bx2.h=%d bx2.d=%d\n", bx2.width, bx2.height, bx2.depth) + + # ============================================================ + # 测试 7: 结构体默认赋值 + # ============================================================ + # 无参数构造：使用默认值 + sz: Size = Size() + stdio.printf("def: sz.w=%d sz.h=%d\n", sz.w, sz.h) + + # 位置参数覆盖默认值 + sz2: Size = Size(5, 6) + stdio.printf("def: sz2.w=%d sz2.h=%d\n", sz2.w, sz2.h) + + # 关键字参数覆盖默认值（乱序） + sz3: Size = Size(h=999, w=888) + stdio.printf("def: sz3.w=%d sz3.h=%d\n", sz3.w, sz3.h) + + # 混合：位置参数 + 默认值（w=7 覆盖默认值，h 保持默认值 200） + sz4: Size = Size(7) + stdio.printf("def: sz4.w=%d sz4.h=%d\n", sz4.w, sz4.h) + + # ============================================================ + # 测试 8: 枚举自动赋值（Idle=0, Run=1, Stop=2） + # ============================================================ + s_idle: int = State.Idle + s_run: int = State.Run + s_stop: int = State.Stop + stdio.printf("enum: Idle=%d Run=%d Stop=%d\n", s_idle, s_run, s_stop) + + # ============================================================ + # 测试 9: 枚举手动赋值（Red=10, Green=11, Blue=20, Yellow=21） + # ============================================================ + c_red: int = Color.Red + c_green: int = Color.Green + c_blue: int = Color.Blue + c_yellow: int = Color.Yellow + stdio.printf("enum: Red=%d Green=%d Blue=%d Yellow=%d\n", + c_red, c_green, c_blue, c_yellow) + + # ============================================================ + # 测试 10: 枚举混用数字类型（基准类型应为 i64） + # ============================================================ + m_small: int = MixedType.Small + m_big: int = MixedType.Big + m_medium: int = MixedType.Medium + stdio.printf("enum: Small=%d Big=%d Medium=%d\n", m_small, m_big, m_medium) + + # 枚举参与运算 + s_sum: int = State.Idle + State.Run + State.Stop + stdio.printf("enum: sum(Idle,Run,Stop)=%d\n", s_sum) + + # 枚举比较 + if State.Idle == 0: + stdio.printf("enum: Idle==0 true\n") + if Color.Red == 10: + stdio.printf("enum: Red==10 true\n") + if State.Run != State.Stop: + stdio.printf("enum: Run!=Stop true\n") + + # ============================================================ + # 测试 11: 联合体字段读写 + # ============================================================ + u: DataUnion + u.i = 42 + stdio.printf("union: u.i=%d\n", u.i) + + # 写入 l 字段（覆盖 i 的内存，因为 l 是 i64） + u.l = 20015998343868 + stdio.printf("union: u.l=%lld\n", u.l) + + # ============================================================ + # 测试 12: 联合体共享内存验证 + # ============================================================ + u2: DataUnion + u2.i = 1 + # 写入 l 后，i 的值已被覆盖（不再是 1） + u2.l = 100 + if u2.i != 1: + stdio.printf("union: shared memory verified\n") + + return 0 diff --git a/TransPyV/Test/App/test_main.py b/TransPyV/Test/App/test_main.py new file mode 100644 index 0000000..796d916 --- /dev/null +++ b/TransPyV/Test/App/test_main.py @@ -0,0 +1,109 @@ +import stdio +import t, c +import testcheck +from stdint import * +from asm_test import asm_test +from attr_test import attr_test +from augassign_test import augassign_test +from closure_test import closure_test +from deco_test import deco_test +from deref_min_test import deref_min_test +from deref_test import deref_test +from eq_test import eq_test +from float_test import float_test +from flow_test import flow_test +from for_test import for_test +from func_test import func_test +from llvmir_test import llvmir_test +from oop_test import oop_test +from ptr_only_test import ptr_only_test +from ptr_test import ptr_test +from simple_test import simple_test +from string_min_test import string_min_test +from string_test import string_test +from struct_test import struct_test +from type_bit_test import type_bit_test +from vtable_test import vtable_test +from inherit_test import inherit_test +from func_vtable_test import func_vtable_test +from virtual_dispatch_test import virtual_dispatch_test +from new_test import new_test +from namespace_test import namespace_test +from testcheck_test import testcheck_test +from opovl_test import opovl_test +from generic_test import generic_test + + +def main() -> int: + stdio.printf("===== TransPyV Test Suite =====\n\n") + + r: int = 0 + + stdio.fflush(None) + r = asm_test() + stdio.fflush(None) + r = attr_test() + stdio.fflush(None) + r = augassign_test() + stdio.fflush(None) + r = closure_test() + stdio.fflush(None) + r = deco_test() + stdio.fflush(None) + r = deref_min_test() + stdio.fflush(None) + r = deref_test() + stdio.fflush(None) + r = eq_test() + stdio.fflush(None) + r = float_test() + stdio.fflush(None) + r = flow_test() + stdio.fflush(None) + r = for_test() + stdio.fflush(None) + r = func_test() + stdio.fflush(None) + r = llvmir_test() + stdio.fflush(None) + r = oop_test() + stdio.fflush(None) + r = ptr_only_test() + stdio.fflush(None) + r = ptr_test() + stdio.fflush(None) + r = simple_test() + stdio.fflush(None) + r = string_min_test() + stdio.fflush(None) + r = string_test() + stdio.fflush(None) + r = struct_test() + stdio.fflush(None) + r = type_bit_test() + stdio.fflush(None) + r = vtable_test() + stdio.fflush(None) + r = inherit_test() + stdio.fflush(None) + r = func_vtable_test() + stdio.fflush(None) + r = virtual_dispatch_test() + stdio.fflush(None) + r = new_test() + stdio.fflush(None) + r = namespace_test() + stdio.fflush(None) + r = testcheck_test() + stdio.fflush(None) + r = opovl_test() + stdio.fflush(None) + + stdio.printf("[TM] before generic_test\n") + stdio.fflush(None) + r = generic_test() + stdio.printf("[TM] after generic_test r=%d\n", r) + stdio.fflush(None) + + stdio.printf("\n===== Test Suite Complete =====\n") + return r \ No newline at end of file diff --git a/TransPyV/Test/App/testcheck_test.py b/TransPyV/Test/App/testcheck_test.py new file mode 100644 index 0000000..2540bdb --- /dev/null +++ b/TransPyV/Test/App/testcheck_test.py @@ -0,0 +1,37 @@ +import stdio +import t, c +import string +import testcheck + + +# ============================================================ +# testcheck_test - 测试 includes/testcheck 库导入与调用 +# +# 验证命名空间隔离下，import testcheck + 模块限定函数调用能正常工作。 +# 同时测试 string 库的模块限定调用。 +# ============================================================ +def testcheck_test() -> int: + testcheck.begin("testcheck_test") + + testcheck.section("Arithmetic") + a: int = 3 + 4 + testcheck.check(a == 7, "3+4=7", "3+4!=7") + + b: int = 10 - 3 + testcheck.check(b == 7, "10-3=7", "10-3!=7") + + m: int = 6 * 7 + testcheck.check(m == 42, "6*7=42", "6*7!=42") + + testcheck.section("String") + slen: int = string.strlen("hello") + testcheck.check(slen == 5, "strlen(hello)=5", "strlen(hello)!=5") + + scmp: int = string.strcmp("abc", "abc") + testcheck.check(scmp == 0, "strcmp(abc,abc)=0", "strcmp(abc,abc)!=0") + + testcheck.section("Module Import") + testcheck.info("testcheck module imported and called successfully") + testcheck.ok("import testcheck works") + + return testcheck.end() diff --git a/TransPyV/Test/App/type_bit_test.py b/TransPyV/Test/App/type_bit_test.py new file mode 100644 index 0000000..817020f --- /dev/null +++ b/TransPyV/Test/App/type_bit_test.py @@ -0,0 +1,26 @@ +import stdio +import t, c + + +def type_bit_test() -> int: + ti: t.CInt = 100 + ti32: t.CInt32T = 200 + tui32: t.CUInt32T = 300 + stdio.printf("type: CInt=%d CInt32T=%d CUInt32T=%d\n", ti, ti32, tui32) + + tl: t.CLong = 1000 + tll: t.CLongLong = 2000 + ti64: t.CInt64T = 3000 + tsize: t.CSizeT = 4000 + stdio.printf("type: CLong=%d CLongLong=%d CInt64T=%d CSizeT=%d\n", tl, tll, ti64, tsize) + + ba: int = 240 + bb: int = 15 + stdio.printf("bit: 240&15=%d\n", ba & bb) + stdio.printf("bit: 240|15=%d\n", ba | bb) + stdio.printf("bit: 240^15=%d\n", ba ^ bb) + stdio.printf("bit: 240<<2=%d\n", ba << 2) + stdio.printf("bit: 240>>2=%d\n", ba >> 2) + stdio.printf("bit: 17%%5=%d\n", 17 % 5) + + return 0 diff --git a/TransPyV/Test/App/virtual_dispatch_test.py b/TransPyV/Test/App/virtual_dispatch_test.py new file mode 100644 index 0000000..d3ca34b --- /dev/null +++ b/TransPyV/Test/App/virtual_dispatch_test.py @@ -0,0 +1,83 @@ +import stdio +import t, c +import string +from inherit_test import Animal, Dog + + +# ============================================================ +# 虚分派测试：通过虚表指针验证虚分派机制 +# +# 本测试验证：方法调用通过对象的 __vtable__ 字段进行虚分派， +# 而非通过变量声明类型直接调用。 +# +# 测试方法： +# 1. 创建 Animal 对象 a（Animal 虚表，Speak 返回 0） +# 2. 创建 Dog 对象 d +# 3. 用内联汇编将 d 的虚表指针复制到 a +# 4. 调用 a.Speak() — 应返回 1（Dog.Speak），证明虚分派生效 +# +# 如果是直接调用（非虚分派），a.Speak() 会调用 Animal.Speak 返回 0 +# +# 注意：Animal 和 Dog 类从 inherit_test.py 导入（命名空间隔离） +# ============================================================ + + +def virtual_dispatch_test() -> int: + stdio.printf("vdispatch: === Test Start ===\n") + + # 创建对象 + a: Animal = Animal(42) + d: Dog = Dog(100, 7) + + # ============================================================ + # 测试 1: 直接分派（基线验证） + # ============================================================ + stdio.printf("vdispatch: === Test 1: Direct Dispatch ===\n") + + a_speak: int = a.Speak() + stdio.printf("vdispatch: a.Speak()=%d (expected 0)\n", a_speak) + if a_speak != 0: + stdio.printf("[FAIL] a.Speak()=%d expected 0\n", a_speak) + return 1 + + d_speak: int = d.Speak() + stdio.printf("vdispatch: d.Speak()=%d (expected 1)\n", d_speak) + if d_speak != 1: + stdio.printf("[FAIL] d.Speak()=%d expected 1\n", d_speak) + return 1 + + # ============================================================ + # 测试 2: 虚分派验证（复制虚表指针） + # + # 将 Dog 的虚表指针复制到 Animal 对象 a + # 虚表指针位于结构体偏移 0（__vtable__ 字段） + # 复制后 a.Speak() 应通过虚表分发到 Dog.Speak（返回 1） + # ============================================================ + stdio.printf("vdispatch: === Test 2: Virtual Dispatch ===\n") + + # 使用 string.memcpy 复制虚表指针（8字节，结构体偏移0） + # 将 d 的虚表指针复制到 a，使 a.Speak() 分发到 Dog.Speak + string.memcpy(c.Addr(a), c.Addr(d), 8) + + a_speak2: int = a.Speak() + stdio.printf("vdispatch: a.Speak()=%d (expected 1 after vtable copy)\n", a_speak2) + if a_speak2 != 1: + stdio.printf("[FAIL] a.Speak()=%d expected 1 after vtable copy\n", a_speak2) + return 1 + + # ============================================================ + # 测试 3: GetName 验证（未被覆盖的虚方法） + # + # Dog 没有覆盖 GetName，Dog 虚表中 GetName 槽位指向 Animal.GetName + # a.GetName() 应返回 a.name (42)，证明虚表槽位一致性 + # ============================================================ + stdio.printf("vdispatch: === Test 3: GetName After VTable Copy ===\n") + + a_name: int = a.GetName() + stdio.printf("vdispatch: a.GetName()=%d (expected 42)\n", a_name) + if a_name != 42: + stdio.printf("[FAIL] a.GetName()=%d expected 42\n", a_name) + return 1 + + stdio.printf("vdispatch: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/App/vtable_test.py b/TransPyV/Test/App/vtable_test.py new file mode 100644 index 0000000..0964412 --- /dev/null +++ b/TransPyV/Test/App/vtable_test.py @@ -0,0 +1,63 @@ +import stdio +import t, c + + +@t.CVTable +class Animal: + name: t.CInt + + def __init__(self, n: t.CInt): + self.name = n + + def GetName(self) -> t.CInt: + return self.name + + def Speak(self) -> t.CInt: + return 0 + + +@t.CVTable +class Shape: + sides: t.CInt + + def SetSides(self, n: t.CInt) -> t.CInt: + self.sides = n + return 0 + + def GetSides(self) -> t.CInt: + return self.sides + + +def vtable_test() -> int: + stdio.printf("vtable: === Test Start ===\n") + + # 测试 1: Animal 虚方法调用（无参，访问 self 字段） + a: Animal = Animal(42) + name: int = a.GetName() + stdio.printf("vtable: a.GetName()=%d (expected 42)\n", name) + if name != 42: + stdio.printf("[FAIL] a.GetName()=%d expected 42\n", name) + return 1 + + speak: int = a.Speak() + stdio.printf("vtable: a.Speak()=%d (expected 0)\n", speak) + if speak != 0: + stdio.printf("[FAIL] a.Speak()=%d expected 0\n", speak) + return 1 + + # 测试 2: Shape 虚方法调用（带参，修改 self 字段） + s: Shape = Shape() + r: int = s.SetSides(4) + stdio.printf("vtable: s.SetSides(4)=%d (expected 0)\n", r) + if r != 0: + stdio.printf("[FAIL] s.SetSides(4)=%d expected 0\n", r) + return 1 + + sides: int = s.GetSides() + stdio.printf("vtable: s.GetSides()=%d (expected 4)\n", sides) + if sides != 4: + stdio.printf("[FAIL] s.GetSides()=%d expected 4\n", sides) + return 1 + + stdio.printf("vtable: === All Tests Passed ===\n") + return 0 diff --git a/TransPyV/Test/NegativeTest/App/defs.py b/TransPyV/Test/NegativeTest/App/defs.py new file mode 100644 index 0000000..a6b825e --- /dev/null +++ b/TransPyV/Test/NegativeTest/App/defs.py @@ -0,0 +1,20 @@ +import stdio +import t, c + + +# ============================================================ +# 负向测试定义模块：定义 SecretClass +# +# 此文件定义 SecretClass，但 main.py 不 import 它。 +# 严格模式下 main.py 中的 SecretClass() 构造应编译失败。 +# ============================================================ + + +class SecretClass: + val: t.CInt + + def __init__(self, v: t.CInt): + self.val = v + + def GetVal(self) -> t.CInt: + return self.val diff --git a/TransPyV/Test/NegativeTest/App/main.py b/TransPyV/Test/NegativeTest/App/main.py new file mode 100644 index 0000000..dc9a2ba --- /dev/null +++ b/TransPyV/Test/NegativeTest/App/main.py @@ -0,0 +1,21 @@ +import stdio +import t, c +# 故意不 import defs.SecretClass — 严格模式下应编译失败 + + +# ============================================================ +# 负向测试主入口 +# +# 此文件故意不 import SecretClass，但尝试使用 SecretClass()。 +# 命名空间隔离严格模式下，SecretClass 不可见， +# 构造器路径被跳过，生成 call i32 @SecretClass(...)， +# llc 报错 "use of undefined value '@SecretClass'"。 +# +# 预期：TransPyV 编译失败（exit code != 0） +# ============================================================ +def main() -> int: + # SecretClass 未导入，严格模式下不可见 + s: SecretClass = SecretClass(42) + v: int = s.GetVal() + stdio.printf("negtest: v=%d (should not reach here)\n", v) + return 0 diff --git a/TransPyV/Test/NegativeTest/project.vpj b/TransPyV/Test/NegativeTest/project.vpj new file mode 100644 index 0000000..6064058 --- /dev/null +++ b/TransPyV/Test/NegativeTest/project.vpj @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "NegativeTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "neg_app.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/TransPyV/Test/Sha1Test/App/main.py b/TransPyV/Test/Sha1Test/App/main.py new file mode 100644 index 0000000..c7e4533 --- /dev/null +++ b/TransPyV/Test/Sha1Test/App/main.py @@ -0,0 +1,62 @@ +import t, c +from stdint import * +import stdio +import stdlib +import memhub +import hashlib + + +POOL_SIZE: t.CDefine = 16777480 + + +@t.CExport +def main() -> int: + arena: bytes = stdlib.malloc(POOL_SIZE) + if arena is None: + return 1 + mb: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, POOL_SIZE) + if mb is None: + return 1 + hashlib._mbuddy = (memhub.MemManager | t.CPtr)(mb) + + # 测试 "abc" 的 SHA1 + # 期望: a9993e364706816aba3e25717850c26c9cd0d89d + s: str = "abc" + stdio.printf("input: %s\n", s) + + # 计算 len(s) + n: t.CSizeT = len(s) + stdio.printf("len=%d\n", n) + + # 构造 sha1 对象 + ctx: hashlib.sha1 | t.CPtr = hashlib.sha1() + if ctx is None: + stdio.printf("ctx is None\n") + return 1 + + ctx.update(s) + + digest: bytes = mb.alloc(hashlib.SHA1_DIGEST_LEN) + if digest is None: + stdio.printf("digest alloc failed\n") + return 1 + + ctx.final(digest) + + # 打印 digest（十六进制） + stdio.printf("SHA1: ") + for i in range(hashlib.SHA1_DIGEST_LEN): + b: t.CUInt8T = digest[i] + hi: t.CInt = (b >> 4) & 0xF + lo: t.CInt = b & 0xF + if hi < 10: + stdio.printf("%c", '0' + hi) + else: + stdio.printf("%c", 'a' + (hi - 10)) + if lo < 10: + stdio.printf("%c", '0' + lo) + else: + stdio.printf("%c", 'a' + (lo - 10)) + stdio.printf("\n") + stdio.printf("expect: a9993e364706816aba3e25717850c26c9cd0d89d\n") + return 0 diff --git a/TransPyV/Test/Sha1Test/output/f3458d483d1f87c2.deps.json b/TransPyV/Test/Sha1Test/output/f3458d483d1f87c2.deps.json new file mode 100644 index 0000000..792c3d5 --- /dev/null +++ b/TransPyV/Test/Sha1Test/output/f3458d483d1f87c2.deps.json @@ -0,0 +1 @@ +{"atom": "271ea3decb810db2", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "hashlib.__sha1": "73fcdc4dd1618580", "__sha1": "73fcdc4dd1618580", "string": "83210a72e829a086", "hashlib.__init__": "96837bcc64032444", "__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "hashlib.__md5": "cc436a125bbaf28c", "__md5": "cc436a125bbaf28c", "stdlib": "e79ec1d5e09b18bc", "memhub": "ee084e9fc6ee413a", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/project.vpj b/TransPyV/Test/Sha1Test/project.vpj new file mode 100644 index 0000000..f3e807d --- /dev/null +++ b/TransPyV/Test/Sha1Test/project.vpj @@ -0,0 +1,28 @@ +{ + "name": "Sha1Test", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "sha1test.exe" + }, + "includes": [ + "../../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/TransPyV/Test/Sha1Test/temp/271ea3decb810db2.pyi b/TransPyV/Test/Sha1Test/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/TransPyV/Test/Sha1Test/temp/6f62fe05c5ea1ceb.pyi b/TransPyV/Test/Sha1Test/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/6ff26590374ae6fc.pyi b/TransPyV/Test/Sha1Test/temp/6ff26590374ae6fc.pyi new file mode 100644 index 0000000..1b1bba1 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/6ff26590374ae6fc.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from hashlib.__sha512.py +Module: hashlib.__sha512 +""" + + +import t, c + +SHA512_BLOCK_LEN: t.CDefine = 128 +SHA512_DIGEST_LEN: t.CDefine = 64 + +def sha512_ror(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_shr(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_Ch(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Maj(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma1(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma1(x: t.CUInt64T) -> t.CUInt64T: pass + + +sha512_K: t.CExtern | t.CArray[t.CUInt64T, 80] + +@t.Object +class sha512: + state: t.CArray[t.CUInt64T, 8] + count: t.CArray[t.CUInt64T, 2] + buf: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN] + def __init__(self: sha512) -> t.CInt: pass + def transform(self: sha512, block: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN]) -> t.CInt: pass + def update(self: sha512, s: str) -> t.CInt: pass + def final(self: sha512, out: t.CArray[t.CUInt8T, SHA512_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/73fcdc4dd1618580.pyi b/TransPyV/Test/Sha1Test/temp/73fcdc4dd1618580.pyi new file mode 100644 index 0000000..d94a709 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/73fcdc4dd1618580.pyi @@ -0,0 +1,29 @@ +""" +Auto-generated Python stub file from hashlib.__sha1.py +Module: hashlib.__sha1 +""" + + +import t, c + +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self: sha1) -> t.CInt: pass + def transform(self: sha1, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha1, s: str) -> t.CInt: pass + def final(self: sha1, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/83210a72e829a086.pyi b/TransPyV/Test/Sha1Test/temp/83210a72e829a086.pyi new file mode 100644 index 0000000..31ba184 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/83210a72e829a086.pyi @@ -0,0 +1,46 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/TransPyV/Test/Sha1Test/temp/96837bcc64032444.pyi b/TransPyV/Test/Sha1Test/temp/96837bcc64032444.pyi new file mode 100644 index 0000000..46e78bd --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/96837bcc64032444.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from hashlib.__init__.py +Module: hashlib.__init__ +""" + +import t +import c + + +from .__md5 import md5, MD5_DIGEST_LEN +from .__sha1 import sha1, SHA1_DIGEST_LEN +from .__sha256 import sha256, SHA256_DIGEST_LEN +from .__sha512 import sha512, SHA512_DIGEST_LEN \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/_phase1_manifest.json b/TransPyV/Test/Sha1Test/temp/_phase1_manifest.json new file mode 100644 index 0000000..c6b7e99 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\Test\\Sha1Test\\App\\main.py": {"sha1": "f3458d483d1f87c2", "mtime": 1783397909.5463412, "size": 1508}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__init__.py": {"sha1": "96837bcc64032444", "mtime": 1777902641.540433, "size": 186}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__md5.py": {"sha1": "cc436a125bbaf28c", "mtime": 1782266200.556964, "size": 5797}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha1.py": {"sha1": "73fcdc4dd1618580", "mtime": 1782266202.164787, "size": 4501}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha256.py": {"sha1": "c9d54a4158f7f5a8", "mtime": 1782266203.9352357, "size": 5332}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha512.py": {"sha1": "6ff26590374ae6fc", "mtime": 1782266223.3380039, "size": 7162}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1783101063.0559144, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "e79ec1d5e09b18bc", "mtime": 1782810965.1928923, "size": 382}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "83210a72e829a086", "mtime": 1782786805.5320292, "size": 9548}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}} \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/_sha1_map.txt b/TransPyV/Test/Sha1Test/temp/_sha1_map.txt new file mode 100644 index 0000000..112b5fb --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/_sha1_map.txt @@ -0,0 +1,13 @@ +271ea3decb810db2:includes/atom.py +6f62fe05c5ea1ceb:includes/stdio.py +6ff26590374ae6fc:includes/hashlib\__sha512.py +73fcdc4dd1618580:includes/hashlib\__sha1.py +83210a72e829a086:includes/string.py +96837bcc64032444:includes/hashlib\__init__.py +c9d54a4158f7f5a8:includes/hashlib\__sha256.py +c9f4be41ca1cc2b4:includes/viperio.py +cc436a125bbaf28c:includes/hashlib\__md5.py +e79ec1d5e09b18bc:includes/stdlib.py +ee084e9fc6ee413a:includes/memhub.py +f3458d483d1f87c2:main.py +f5522571bcce7bcb:includes/stdint.py diff --git a/TransPyV/Test/Sha1Test/temp/c9d54a4158f7f5a8.pyi b/TransPyV/Test/Sha1Test/temp/c9d54a4158f7f5a8.pyi new file mode 100644 index 0000000..3e2d738 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/c9d54a4158f7f5a8.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__sha256.py +Module: hashlib.__sha256 +""" + + +import t, c + +SHA256_BLOCK_LEN: t.CDefine = 64 +SHA256_DIGEST_LEN: t.CDefine = 32 +sha256_K: t.CExtern | t.CArray[t.CUInt32T, 64] + +def s0(x: t.CUInt32T) -> t.CUInt32T: pass + +def s1(x: t.CUInt32T) -> t.CUInt32T: pass + +def S0(x: t.CUInt32T) -> t.CUInt32T: pass + +def S1(x: t.CUInt32T) -> t.CUInt32T: pass + +def Ch(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def Maj(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha256: + state: t.CArray[t.CUInt32T, 8] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA256_BLOCK_LEN] + def __init__(self: sha256) -> t.CInt: pass + def transform(self: sha256, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha256, s: str) -> t.CInt: pass + def final(self: sha256, out: t.CArray[t.CUInt8T, SHA256_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/c9f4be41ca1cc2b4.pyi b/TransPyV/Test/Sha1Test/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/cc436a125bbaf28c.pyi b/TransPyV/Test/Sha1Test/temp/cc436a125bbaf28c.pyi new file mode 100644 index 0000000..a905387 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/cc436a125bbaf28c.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__md5.py +Module: hashlib.__md5 +""" + + +import t, c + +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + + +md5_T: t.CExtern | t.CArray[t.CUInt32T, 64] +md5_S: t.CExtern | t.CArray[t.CInt, 64] + +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] + def __init__(self: md5) -> t.CInt: pass + def transform(self: md5, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: md5, s: str) -> t.CInt: pass + def final(self: md5, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/e79ec1d5e09b18bc.pyi b/TransPyV/Test/Sha1Test/temp/e79ec1d5e09b18bc.pyi new file mode 100644 index 0000000..2f56642 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/e79ec1d5e09b18bc.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> None | t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/TransPyV/Test/Sha1Test/temp/ee084e9fc6ee413a.pyi b/TransPyV/Test/Sha1Test/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.doc.json b/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.pyi b/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.pyi new file mode 100644 index 0000000..e5d85d6 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/f3458d483d1f87c2.pyi @@ -0,0 +1,17 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import stdlib +import memhub +import hashlib + +POOL_SIZE: t.CDefine = 16777480 + +@t.CExport +def main() -> int: pass diff --git a/TransPyV/Test/Sha1Test/temp/f5522571bcce7bcb.pyi b/TransPyV/Test/Sha1Test/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/TransPyV/Test/Sha1Test/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/TransPyV/Test/project.vpj b/TransPyV/Test/project.vpj new file mode 100644 index 0000000..dd18769 --- /dev/null +++ b/TransPyV/Test/project.vpj @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "AstTest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "app.exe" + }, + "includes": [ + "../../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/TransPyV/Test/temp/0035c95a18d4f8e8.deps.txt b/TransPyV/Test/temp/0035c95a18d4f8e8.deps.txt new file mode 100644 index 0000000..8b41c68 --- /dev/null +++ b/TransPyV/Test/temp/0035c95a18d4f8e8.deps.txt @@ -0,0 +1 @@ +t c stdint stdio w32.win32base w32.win32file \ No newline at end of file diff --git a/TransPyV/Test/temp/05c11f22675c60cb.deps.txt b/TransPyV/Test/temp/05c11f22675c60cb.deps.txt new file mode 100644 index 0000000..267186e --- /dev/null +++ b/TransPyV/Test/temp/05c11f22675c60cb.deps.txt @@ -0,0 +1 @@ +t c stdint string viperlib memhub linkedlist llvmlite.__types \ No newline at end of file diff --git a/TransPyV/Test/temp/067c78e9f121dce3.deps.txt b/TransPyV/Test/temp/067c78e9f121dce3.deps.txt new file mode 100644 index 0000000..e28a09e --- /dev/null +++ b/TransPyV/Test/temp/067c78e9f121dce3.deps.txt @@ -0,0 +1 @@ +t stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/Test/temp/06f53cc594b4ac6c.deps.txt b/TransPyV/Test/temp/06f53cc594b4ac6c.deps.txt new file mode 100644 index 0000000..e28a09e --- /dev/null +++ b/TransPyV/Test/temp/06f53cc594b4ac6c.deps.txt @@ -0,0 +1 @@ +t stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/Test/temp/07f7da85af876124.deps.txt b/TransPyV/Test/temp/07f7da85af876124.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/07f7da85af876124.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/0c212981c180e7fb.deps.txt b/TransPyV/Test/temp/0c212981c180e7fb.deps.txt new file mode 100644 index 0000000..6ede9fe --- /dev/null +++ b/TransPyV/Test/temp/0c212981c180e7fb.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.tokens \ No newline at end of file diff --git a/TransPyV/Test/temp/0df65b8ed15664b0.deps.txt b/TransPyV/Test/temp/0df65b8ed15664b0.deps.txt new file mode 100644 index 0000000..adaf483 --- /dev/null +++ b/TransPyV/Test/temp/0df65b8ed15664b0.deps.txt @@ -0,0 +1 @@ +string t c \ No newline at end of file diff --git a/TransPyV/Test/temp/0eccc2393466d940.deps.txt b/TransPyV/Test/temp/0eccc2393466d940.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/0eccc2393466d940.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/13110effbb0bb06c.deps.txt b/TransPyV/Test/temp/13110effbb0bb06c.deps.txt new file mode 100644 index 0000000..0774018 --- /dev/null +++ b/TransPyV/Test/temp/13110effbb0bb06c.deps.txt @@ -0,0 +1 @@ +t c stdint string memhub platmacro \ No newline at end of file diff --git a/TransPyV/Test/temp/17c0dfbcebfa3a0e.deps.txt b/TransPyV/Test/temp/17c0dfbcebfa3a0e.deps.txt new file mode 100644 index 0000000..e16a61d --- /dev/null +++ b/TransPyV/Test/temp/17c0dfbcebfa3a0e.deps.txt @@ -0,0 +1 @@ +t c stdint stdio string t c stdint stdio string \ No newline at end of file diff --git a/TransPyV/Test/temp/19f8024d10c828e8.deps.txt b/TransPyV/Test/temp/19f8024d10c828e8.deps.txt new file mode 100644 index 0000000..adaf483 --- /dev/null +++ b/TransPyV/Test/temp/19f8024d10c828e8.deps.txt @@ -0,0 +1 @@ +string t c \ No newline at end of file diff --git a/TransPyV/Test/temp/1b9ab264a942c56b.deps.txt b/TransPyV/Test/temp/1b9ab264a942c56b.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/1b9ab264a942c56b.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/1c46d554b3a3f9f3.deps.txt b/TransPyV/Test/temp/1c46d554b3a3f9f3.deps.txt new file mode 100644 index 0000000..88b63b8 --- /dev/null +++ b/TransPyV/Test/temp/1c46d554b3a3f9f3.deps.txt @@ -0,0 +1 @@ +stdint zlib.zdeflate zdeflate zlib.zinflate zinflate zlib.zchecksum zchecksum zlib.zdef zdef zlib.zhuff zhuff stdlib string memhub t c \ No newline at end of file diff --git a/TransPyV/Test/temp/1eb1c9fc7e85d9f8.deps.txt b/TransPyV/Test/temp/1eb1c9fc7e85d9f8.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/1eb1c9fc7e85d9f8.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/20cd49775c100a38.deps.txt b/TransPyV/Test/temp/20cd49775c100a38.deps.txt new file mode 100644 index 0000000..1b09b43 --- /dev/null +++ b/TransPyV/Test/temp/20cd49775c100a38.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string stdio t c stdint memhub string stdio \ No newline at end of file diff --git a/TransPyV/Test/temp/213348433fb01cc6.deps.txt b/TransPyV/Test/temp/213348433fb01cc6.deps.txt new file mode 100644 index 0000000..857fca7 --- /dev/null +++ b/TransPyV/Test/temp/213348433fb01cc6.deps.txt @@ -0,0 +1 @@ +stdint stddef string stdlib memhub t c \ No newline at end of file diff --git a/TransPyV/Test/temp/21a7fcfc665f75ef.deps.txt b/TransPyV/Test/temp/21a7fcfc665f75ef.deps.txt new file mode 100644 index 0000000..72c74ed --- /dev/null +++ b/TransPyV/Test/temp/21a7fcfc665f75ef.deps.txt @@ -0,0 +1 @@ +t c stdint string stdio viperlib memhub linkedlist llvmlite.__types llvmlite.__function \ No newline at end of file diff --git a/TransPyV/Test/temp/240a9a4157959a9f.deps.txt b/TransPyV/Test/temp/240a9a4157959a9f.deps.txt new file mode 100644 index 0000000..cccc168 --- /dev/null +++ b/TransPyV/Test/temp/240a9a4157959a9f.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string t c stdint memhub string \ No newline at end of file diff --git a/TransPyV/Test/temp/257e074b935c33b4.deps.txt b/TransPyV/Test/temp/257e074b935c33b4.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/257e074b935c33b4.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/271ea3decb810db2.deps.txt b/TransPyV/Test/temp/271ea3decb810db2.deps.txt new file mode 100644 index 0000000..62e033a --- /dev/null +++ b/TransPyV/Test/temp/271ea3decb810db2.deps.txt @@ -0,0 +1 @@ +t c t c \ No newline at end of file diff --git a/TransPyV/Test/temp/2769dee537308b00.deps.txt b/TransPyV/Test/temp/2769dee537308b00.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/2769dee537308b00.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/2d39c6c7d3557b3e.deps.txt b/TransPyV/Test/temp/2d39c6c7d3557b3e.deps.txt new file mode 100644 index 0000000..103beac --- /dev/null +++ b/TransPyV/Test/temp/2d39c6c7d3557b3e.deps.txt @@ -0,0 +1 @@ +t c stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/2da636c61863c815.deps.txt b/TransPyV/Test/temp/2da636c61863c815.deps.txt new file mode 100644 index 0000000..f1c31c2 --- /dev/null +++ b/TransPyV/Test/temp/2da636c61863c815.deps.txt @@ -0,0 +1 @@ +t c stdint platmacro memhub string w32.win32base w32.win32file w32.win32process w32.win32sync posix \ No newline at end of file diff --git a/TransPyV/Test/temp/33a854a5c4f67d74.deps.txt b/TransPyV/Test/temp/33a854a5c4f67d74.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/33a854a5c4f67d74.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/34548789d646f432.deps.txt b/TransPyV/Test/temp/34548789d646f432.deps.txt new file mode 100644 index 0000000..89fd3f5 --- /dev/null +++ b/TransPyV/Test/temp/34548789d646f432.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.tokens ast.base ast.stmts ast.exprs ast.astaux ast.match ast.visitor ast.lexer ast.parser ast.parser \ No newline at end of file diff --git a/TransPyV/Test/temp/3487a256b250bb74.deps.txt b/TransPyV/Test/temp/3487a256b250bb74.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/3487a256b250bb74.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/3624cfde3c5cb6cb.deps.txt b/TransPyV/Test/temp/3624cfde3c5cb6cb.deps.txt new file mode 100644 index 0000000..ec186f1 --- /dev/null +++ b/TransPyV/Test/temp/3624cfde3c5cb6cb.deps.txt @@ -0,0 +1 @@ +string \ No newline at end of file diff --git a/TransPyV/Test/temp/3c2fbc9e35ad3bf2.deps.txt b/TransPyV/Test/temp/3c2fbc9e35ad3bf2.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/3c2fbc9e35ad3bf2.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/3f7c5e78d8652535.deps.txt b/TransPyV/Test/temp/3f7c5e78d8652535.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/3f7c5e78d8652535.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/4337fb260448bbe2.deps.txt b/TransPyV/Test/temp/4337fb260448bbe2.deps.txt new file mode 100644 index 0000000..1fb5080 --- /dev/null +++ b/TransPyV/Test/temp/4337fb260448bbe2.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.base \ No newline at end of file diff --git a/TransPyV/Test/temp/451745df21e598af.deps.txt b/TransPyV/Test/temp/451745df21e598af.deps.txt new file mode 100644 index 0000000..aae827d --- /dev/null +++ b/TransPyV/Test/temp/451745df21e598af.deps.txt @@ -0,0 +1 @@ +stdint zlib.zchecksum zchecksum zlib.zdef zdef zlib.zhuff zhuff string memhub t c \ No newline at end of file diff --git a/TransPyV/Test/temp/455ff4e74c3dbfb0.deps.txt b/TransPyV/Test/temp/455ff4e74c3dbfb0.deps.txt new file mode 100644 index 0000000..f84e0ef --- /dev/null +++ b/TransPyV/Test/temp/455ff4e74c3dbfb0.deps.txt @@ -0,0 +1 @@ +t c stdint viperlib string memhub linkedlist \ No newline at end of file diff --git a/TransPyV/Test/temp/46ea362b72c10a9e.deps.txt b/TransPyV/Test/temp/46ea362b72c10a9e.deps.txt new file mode 100644 index 0000000..b819530 --- /dev/null +++ b/TransPyV/Test/temp/46ea362b72c10a9e.deps.txt @@ -0,0 +1 @@ +t c stdint string viperlib memhub linkedlist llvmlite.__types llvmlite.__values llvmlite.__function \ No newline at end of file diff --git a/TransPyV/Test/temp/47767b5026a8ee15.deps.txt b/TransPyV/Test/temp/47767b5026a8ee15.deps.txt new file mode 100644 index 0000000..1fb5080 --- /dev/null +++ b/TransPyV/Test/temp/47767b5026a8ee15.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.base \ No newline at end of file diff --git a/TransPyV/Test/temp/4dd6b3f1427d1cc5.deps.txt b/TransPyV/Test/temp/4dd6b3f1427d1cc5.deps.txt new file mode 100644 index 0000000..1fb5080 --- /dev/null +++ b/TransPyV/Test/temp/4dd6b3f1427d1cc5.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.base \ No newline at end of file diff --git a/TransPyV/Test/temp/55cbbb5dec4fef2c.deps.txt b/TransPyV/Test/temp/55cbbb5dec4fef2c.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/55cbbb5dec4fef2c.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/56d07ea7e30ef631.deps.txt b/TransPyV/Test/temp/56d07ea7e30ef631.deps.txt new file mode 100644 index 0000000..9546553 --- /dev/null +++ b/TransPyV/Test/temp/56d07ea7e30ef631.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32sync \ No newline at end of file diff --git a/TransPyV/Test/temp/57288496f7c2d1ad.deps.txt b/TransPyV/Test/temp/57288496f7c2d1ad.deps.txt new file mode 100644 index 0000000..4925b7f --- /dev/null +++ b/TransPyV/Test/temp/57288496f7c2d1ad.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string viperio json.__parser json.__writer t c stdint memhub string viperio json.__parser json.__writer \ No newline at end of file diff --git a/TransPyV/Test/temp/5b2e220c5bae60c9.deps.txt b/TransPyV/Test/temp/5b2e220c5bae60c9.deps.txt new file mode 100644 index 0000000..1344ae8 --- /dev/null +++ b/TransPyV/Test/temp/5b2e220c5bae60c9.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32process stdlib stdio string \ No newline at end of file diff --git a/TransPyV/Test/temp/5dab8cb390496d22.deps.txt b/TransPyV/Test/temp/5dab8cb390496d22.deps.txt new file mode 100644 index 0000000..c76c2b1 --- /dev/null +++ b/TransPyV/Test/temp/5dab8cb390496d22.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string viperlib ast.tokens \ No newline at end of file diff --git a/TransPyV/Test/temp/62cc01c2bb770ca3.deps.txt b/TransPyV/Test/temp/62cc01c2bb770ca3.deps.txt new file mode 100644 index 0000000..8c2cb2a --- /dev/null +++ b/TransPyV/Test/temp/62cc01c2bb770ca3.deps.txt @@ -0,0 +1 @@ +t c vipermath \ No newline at end of file diff --git a/TransPyV/Test/temp/6446627d4f07a1b5.deps.txt b/TransPyV/Test/temp/6446627d4f07a1b5.deps.txt new file mode 100644 index 0000000..0681b55 --- /dev/null +++ b/TransPyV/Test/temp/6446627d4f07a1b5.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/Test/temp/6503c97dde0c79c4.deps.txt b/TransPyV/Test/temp/6503c97dde0c79c4.deps.txt new file mode 100644 index 0000000..103beac --- /dev/null +++ b/TransPyV/Test/temp/6503c97dde0c79c4.deps.txt @@ -0,0 +1 @@ +t c stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/657e182b27c2a022.deps.txt b/TransPyV/Test/temp/657e182b27c2a022.deps.txt new file mode 100644 index 0000000..1fb5080 --- /dev/null +++ b/TransPyV/Test/temp/657e182b27c2a022.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.base \ No newline at end of file diff --git a/TransPyV/Test/temp/6f5e876273093314.deps.txt b/TransPyV/Test/temp/6f5e876273093314.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/6f5e876273093314.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/6f62fe05c5ea1ceb.deps.txt b/TransPyV/Test/temp/6f62fe05c5ea1ceb.deps.txt new file mode 100644 index 0000000..62e033a --- /dev/null +++ b/TransPyV/Test/temp/6f62fe05c5ea1ceb.deps.txt @@ -0,0 +1 @@ +t c t c \ No newline at end of file diff --git a/TransPyV/Test/temp/6ff26590374ae6fc.deps.txt b/TransPyV/Test/temp/6ff26590374ae6fc.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/6ff26590374ae6fc.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/71e0a3ffcb3ebfad.deps.txt b/TransPyV/Test/temp/71e0a3ffcb3ebfad.deps.txt new file mode 100644 index 0000000..32f64f4 --- /dev/null +++ b/TransPyV/Test/temp/71e0a3ffcb3ebfad.deps.txt @@ -0,0 +1 @@ +t \ No newline at end of file diff --git a/TransPyV/Test/temp/72e2d5ccb7cedcf1.deps.txt b/TransPyV/Test/temp/72e2d5ccb7cedcf1.deps.txt new file mode 100644 index 0000000..e28a09e --- /dev/null +++ b/TransPyV/Test/temp/72e2d5ccb7cedcf1.deps.txt @@ -0,0 +1 @@ +t stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/Test/temp/74ca31446cac2107.deps.txt b/TransPyV/Test/temp/74ca31446cac2107.deps.txt new file mode 100644 index 0000000..7cff98f --- /dev/null +++ b/TransPyV/Test/temp/74ca31446cac2107.deps.txt @@ -0,0 +1 @@ +stdio t c string stdio t c string \ No newline at end of file diff --git a/TransPyV/Test/temp/76dd6c275aa72b3b.deps.txt b/TransPyV/Test/temp/76dd6c275aa72b3b.deps.txt new file mode 100644 index 0000000..fd84187 --- /dev/null +++ b/TransPyV/Test/temp/76dd6c275aa72b3b.deps.txt @@ -0,0 +1 @@ +t c stdint platmacro memhub os.path os._win32 os._posix \ No newline at end of file diff --git a/TransPyV/Test/temp/79b337e5ea8951e2.deps.txt b/TransPyV/Test/temp/79b337e5ea8951e2.deps.txt new file mode 100644 index 0000000..ffede5f --- /dev/null +++ b/TransPyV/Test/temp/79b337e5ea8951e2.deps.txt @@ -0,0 +1 @@ +t memhub string \ No newline at end of file diff --git a/TransPyV/Test/temp/7e529fe7a078cfef.deps.txt b/TransPyV/Test/temp/7e529fe7a078cfef.deps.txt new file mode 100644 index 0000000..da45a43 --- /dev/null +++ b/TransPyV/Test/temp/7e529fe7a078cfef.deps.txt @@ -0,0 +1 @@ +t stdint t stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/80241c7d6f357bd1.deps.txt b/TransPyV/Test/temp/80241c7d6f357bd1.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/80241c7d6f357bd1.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/87274c2b0190fb33.deps.txt b/TransPyV/Test/temp/87274c2b0190fb33.deps.txt new file mode 100644 index 0000000..7f50989 --- /dev/null +++ b/TransPyV/Test/temp/87274c2b0190fb33.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32process w32.win32sync lock event condition rwlock \ No newline at end of file diff --git a/TransPyV/Test/temp/8ebed83a7817fa0c.deps.txt b/TransPyV/Test/temp/8ebed83a7817fa0c.deps.txt new file mode 100644 index 0000000..0ec8cde --- /dev/null +++ b/TransPyV/Test/temp/8ebed83a7817fa0c.deps.txt @@ -0,0 +1 @@ +stdint t c \ No newline at end of file diff --git a/TransPyV/Test/temp/90921d009fdd674c.deps.txt b/TransPyV/Test/temp/90921d009fdd674c.deps.txt new file mode 100644 index 0000000..313ad5b --- /dev/null +++ b/TransPyV/Test/temp/90921d009fdd674c.deps.txt @@ -0,0 +1 @@ +stdint zlib.zchecksum zchecksum zlib.zhuff zhuff zlib.zdef zdef string memhub t c \ No newline at end of file diff --git a/TransPyV/Test/temp/90c53dd6db8d41cf.deps.txt b/TransPyV/Test/temp/90c53dd6db8d41cf.deps.txt new file mode 100644 index 0000000..ef9e74f --- /dev/null +++ b/TransPyV/Test/temp/90c53dd6db8d41cf.deps.txt @@ -0,0 +1 @@ +stdint t stdint t \ No newline at end of file diff --git a/TransPyV/Test/temp/93c1d18e35d188d6.deps.txt b/TransPyV/Test/temp/93c1d18e35d188d6.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/93c1d18e35d188d6.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/95394ca8da0f655a.deps.txt b/TransPyV/Test/temp/95394ca8da0f655a.deps.txt new file mode 100644 index 0000000..74cae9a --- /dev/null +++ b/TransPyV/Test/temp/95394ca8da0f655a.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string viperlib llvmlite.__types llvmlite.__values llvmlite.__function llvmlite.__module llvmlite.__builder llvmlite.__verify \ No newline at end of file diff --git a/TransPyV/Test/temp/96837bcc64032444.deps.txt b/TransPyV/Test/temp/96837bcc64032444.deps.txt new file mode 100644 index 0000000..c45eb37 --- /dev/null +++ b/TransPyV/Test/temp/96837bcc64032444.deps.txt @@ -0,0 +1 @@ +hashlib.__md5 hashlib.__sha1 hashlib.__sha256 hashlib.__sha512 \ No newline at end of file diff --git a/TransPyV/Test/temp/969c589035910845.deps.txt b/TransPyV/Test/temp/969c589035910845.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/969c589035910845.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/97043f0d52da437f.deps.txt b/TransPyV/Test/temp/97043f0d52da437f.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/97043f0d52da437f.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/971e24c228377a9b.deps.txt b/TransPyV/Test/temp/971e24c228377a9b.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/971e24c228377a9b.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/9d4b01572930a369.deps.txt b/TransPyV/Test/temp/9d4b01572930a369.deps.txt new file mode 100644 index 0000000..1cb69c4 --- /dev/null +++ b/TransPyV/Test/temp/9d4b01572930a369.deps.txt @@ -0,0 +1 @@ +stdio stdlib t c testcheck memhub _list stdio stdlib t c testcheck memhub _list \ No newline at end of file diff --git a/TransPyV/Test/temp/9d798d964b9d3411.deps.txt b/TransPyV/Test/temp/9d798d964b9d3411.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/9d798d964b9d3411.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/9d8626a10208b8de.deps.txt b/TransPyV/Test/temp/9d8626a10208b8de.deps.txt new file mode 100644 index 0000000..9546553 --- /dev/null +++ b/TransPyV/Test/temp/9d8626a10208b8de.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32sync \ No newline at end of file diff --git a/TransPyV/Test/temp/9d8e1c9631c28b1b.deps.txt b/TransPyV/Test/temp/9d8e1c9631c28b1b.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/9d8e1c9631c28b1b.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/_sha1_map.txt b/TransPyV/Test/temp/_sha1_map.txt new file mode 100644 index 0000000..cf1f690 --- /dev/null +++ b/TransPyV/Test/temp/_sha1_map.txt @@ -0,0 +1,47 @@ +271ea3decb810db2:includes/atom.py +57288496f7c2d1ad:includes/json/__init__.py +240a9a4157959a9f:includes/json/__parser.py +20cd49775c100a38:includes/json/__writer.py +ee084e9fc6ee413a:includes/memhub.py +f5522571bcce7bcb:includes/stdint.py +6f62fe05c5ea1ceb:includes/stdio.py +90c53dd6db8d41cf:includes/stdlib.py +ab6e54ba9a669f76:includes/string.py +dd3002730623424b:includes/testcheck.py +c9f4be41ca1cc2b4:includes/viperio.py +7e529fe7a078cfef:includes/w32/win32base.py +bbdf3bbd4c3bc28c:includes/w32/win32console.py +e3e7b6de8d7d8b03:includes/_dict.py +f2ecbf8ced20575c:includes/_list.py +80241c7d6f357bd1:App/asm_test.py +17c0dfbcebfa3a0e:App/attr_test.py +0eccc2393466d940:App/augassign_test.py +9d8e1c9631c28b1b:App/closure_test.py +b7bbe6a1c1e75ad8:App/deco_test.py +6f5e876273093314:App/deref_min_test.py +eb252050d92f8fb9:App/deref_test.py +b8cbe044a10adcca:App/eq_test.py +e74870eba3e6d774:App/float_test.py +969c589035910845:App/flow_test.py +1eb1c9fc7e85d9f8:App/for_test.py +3c2fbc9e35ad3bf2:App/func_test.py +a9df037d9178888f:App/func_vtable_test.py +9d4b01572930a369:App/generic_test.py +97043f0d52da437f:App/inherit_test.py +c39f3e540adb7b4e:App/llvmir_test.py +b7eeff93aa63e70f:App/namespace_defs.py +ef45adce81bf5b2d:App/namespace_test.py +9d798d964b9d3411:App/new_test.py +1b9ab264a942c56b:App/oop_test.py +a7cf0ce4066332ae:App/opovl_test.py +ed1c124f3cb787c9:App/ptr_only_test.py +07f7da85af876124:App/ptr_test.py +b078e9c9c74a2a0c:App/simple_test.py +74ca31446cac2107:App/string_min_test.py +e0009679269c6b63:App/string_test.py +33a854a5c4f67d74:App/struct_test.py +d7662500f1c2c0ff:App/testcheck_test.py +c38a35636ba0f067:App/test_main.py +55cbbb5dec4fef2c:App/type_bit_test.py +cc3ad7f3bc880326:App/virtual_dispatch_test.py +2769dee537308b00:App/vtable_test.py diff --git a/TransPyV/Test/temp/a68b70f233541a7f.deps.txt b/TransPyV/Test/temp/a68b70f233541a7f.deps.txt new file mode 100644 index 0000000..eea7221 --- /dev/null +++ b/TransPyV/Test/temp/a68b70f233541a7f.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32file memhub stdlib \ No newline at end of file diff --git a/TransPyV/Test/temp/a7bc8c01684c0001.deps.txt b/TransPyV/Test/temp/a7bc8c01684c0001.deps.txt new file mode 100644 index 0000000..c19302e --- /dev/null +++ b/TransPyV/Test/temp/a7bc8c01684c0001.deps.txt @@ -0,0 +1 @@ +stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/a7cf0ce4066332ae.deps.txt b/TransPyV/Test/temp/a7cf0ce4066332ae.deps.txt new file mode 100644 index 0000000..9d25cc3 --- /dev/null +++ b/TransPyV/Test/temp/a7cf0ce4066332ae.deps.txt @@ -0,0 +1 @@ +stdio stdlib t c testcheck stdio stdlib t c testcheck \ No newline at end of file diff --git a/TransPyV/Test/temp/a9df037d9178888f.deps.txt b/TransPyV/Test/temp/a9df037d9178888f.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/a9df037d9178888f.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/ab6e54ba9a669f76.deps.txt b/TransPyV/Test/temp/ab6e54ba9a669f76.deps.txt new file mode 100644 index 0000000..0a3240f --- /dev/null +++ b/TransPyV/Test/temp/ab6e54ba9a669f76.deps.txt @@ -0,0 +1 @@ +stdint t c stdint t c \ No newline at end of file diff --git a/TransPyV/Test/temp/aba439b7882ad9d6.deps.txt b/TransPyV/Test/temp/aba439b7882ad9d6.deps.txt new file mode 100644 index 0000000..691d4bf --- /dev/null +++ b/TransPyV/Test/temp/aba439b7882ad9d6.deps.txt @@ -0,0 +1 @@ +t c stdint string memhub stdio hashtable \ No newline at end of file diff --git a/TransPyV/Test/temp/abbcbd92436cc16a.deps.txt b/TransPyV/Test/temp/abbcbd92436cc16a.deps.txt new file mode 100644 index 0000000..fd821fa --- /dev/null +++ b/TransPyV/Test/temp/abbcbd92436cc16a.deps.txt @@ -0,0 +1 @@ +t c stdint platmacro memhub string w32.win32base w32.win32file posix \ No newline at end of file diff --git a/TransPyV/Test/temp/ad6c853acfc0c146.deps.txt b/TransPyV/Test/temp/ad6c853acfc0c146.deps.txt new file mode 100644 index 0000000..9546553 --- /dev/null +++ b/TransPyV/Test/temp/ad6c853acfc0c146.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32sync \ No newline at end of file diff --git a/TransPyV/Test/temp/b078e9c9c74a2a0c.deps.txt b/TransPyV/Test/temp/b078e9c9c74a2a0c.deps.txt new file mode 100644 index 0000000..a98a2fa --- /dev/null +++ b/TransPyV/Test/temp/b078e9c9c74a2a0c.deps.txt @@ -0,0 +1 @@ +stdio stdio \ No newline at end of file diff --git a/TransPyV/Test/temp/b19a9e500f677f2e.deps.txt b/TransPyV/Test/temp/b19a9e500f677f2e.deps.txt new file mode 100644 index 0000000..978194c --- /dev/null +++ b/TransPyV/Test/temp/b19a9e500f677f2e.deps.txt @@ -0,0 +1 @@ +t c atom \ No newline at end of file diff --git a/TransPyV/Test/temp/b547ac4f380bddb6.deps.txt b/TransPyV/Test/temp/b547ac4f380bddb6.deps.txt new file mode 100644 index 0000000..da1157e --- /dev/null +++ b/TransPyV/Test/temp/b547ac4f380bddb6.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string \ No newline at end of file diff --git a/TransPyV/Test/temp/b558d8d8f01f4825.deps.txt b/TransPyV/Test/temp/b558d8d8f01f4825.deps.txt new file mode 100644 index 0000000..e60effd --- /dev/null +++ b/TransPyV/Test/temp/b558d8d8f01f4825.deps.txt @@ -0,0 +1 @@ +t c stdint string socket memhub stdio \ No newline at end of file diff --git a/TransPyV/Test/temp/b5a965302cded36b.deps.txt b/TransPyV/Test/temp/b5a965302cded36b.deps.txt new file mode 100644 index 0000000..9d7639a --- /dev/null +++ b/TransPyV/Test/temp/b5a965302cded36b.deps.txt @@ -0,0 +1 @@ +t c stdint platmacro memhub string stdio w32.win32base w32.win32process posix \ No newline at end of file diff --git a/TransPyV/Test/temp/b62eded2baf30423.deps.txt b/TransPyV/Test/temp/b62eded2baf30423.deps.txt new file mode 100644 index 0000000..103beac --- /dev/null +++ b/TransPyV/Test/temp/b62eded2baf30423.deps.txt @@ -0,0 +1 @@ +t c stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/b7bbe6a1c1e75ad8.deps.txt b/TransPyV/Test/temp/b7bbe6a1c1e75ad8.deps.txt new file mode 100644 index 0000000..7cff98f --- /dev/null +++ b/TransPyV/Test/temp/b7bbe6a1c1e75ad8.deps.txt @@ -0,0 +1 @@ +stdio t c string stdio t c string \ No newline at end of file diff --git a/TransPyV/Test/temp/b7eeff93aa63e70f.deps.txt b/TransPyV/Test/temp/b7eeff93aa63e70f.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/b7eeff93aa63e70f.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/b8c66c8ff44eb874.deps.txt b/TransPyV/Test/temp/b8c66c8ff44eb874.deps.txt new file mode 100644 index 0000000..789f582 --- /dev/null +++ b/TransPyV/Test/temp/b8c66c8ff44eb874.deps.txt @@ -0,0 +1 @@ +t c string memhub stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/b8cbe044a10adcca.deps.txt b/TransPyV/Test/temp/b8cbe044a10adcca.deps.txt new file mode 100644 index 0000000..7cff98f --- /dev/null +++ b/TransPyV/Test/temp/b8cbe044a10adcca.deps.txt @@ -0,0 +1 @@ +stdio t c string stdio t c string \ No newline at end of file diff --git a/TransPyV/Test/temp/bbdf3bbd4c3bc28c.deps.txt b/TransPyV/Test/temp/bbdf3bbd4c3bc28c.deps.txt new file mode 100644 index 0000000..7fc5988 --- /dev/null +++ b/TransPyV/Test/temp/bbdf3bbd4c3bc28c.deps.txt @@ -0,0 +1 @@ +t stdint w32.win32base t stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/Test/temp/bf9813f6e3c9d351.deps.txt b/TransPyV/Test/temp/bf9813f6e3c9d351.deps.txt new file mode 100644 index 0000000..2986427 --- /dev/null +++ b/TransPyV/Test/temp/bf9813f6e3c9d351.deps.txt @@ -0,0 +1 @@ +t c stdint memhub \ No newline at end of file diff --git a/TransPyV/Test/temp/c24f25b14b4bbd0a.deps.txt b/TransPyV/Test/temp/c24f25b14b4bbd0a.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/c24f25b14b4bbd0a.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/c38a35636ba0f067.deps.txt b/TransPyV/Test/temp/c38a35636ba0f067.deps.txt new file mode 100644 index 0000000..957c02b --- /dev/null +++ b/TransPyV/Test/temp/c38a35636ba0f067.deps.txt @@ -0,0 +1 @@ +stdio t c testcheck stdint asm_test attr_test augassign_test closure_test deco_test deref_min_test deref_test eq_test float_test flow_test for_test func_test llvmir_test oop_test ptr_only_test ptr_test simple_test string_min_test string_test struct_test type_bit_test vtable_test inherit_test func_vtable_test virtual_dispatch_test new_test namespace_test testcheck_test opovl_test generic_test stdio t c testcheck stdint asm_test attr_test augassign_test closure_test deco_test deref_min_test deref_test eq_test float_test flow_test for_test func_test llvmir_test oop_test ptr_only_test ptr_test simple_test string_min_test string_test struct_test type_bit_test vtable_test inherit_test func_vtable_test virtual_dispatch_test new_test namespace_test testcheck_test opovl_test generic_test \ No newline at end of file diff --git a/TransPyV/Test/temp/c39f3e540adb7b4e.deps.txt b/TransPyV/Test/temp/c39f3e540adb7b4e.deps.txt new file mode 100644 index 0000000..e4e17be --- /dev/null +++ b/TransPyV/Test/temp/c39f3e540adb7b4e.deps.txt @@ -0,0 +1 @@ +t c stdint stdio t c stdint stdio \ No newline at end of file diff --git a/TransPyV/Test/temp/c3a6aed1f1fb8b1e.deps.txt b/TransPyV/Test/temp/c3a6aed1f1fb8b1e.deps.txt new file mode 100644 index 0000000..cfb58b6 --- /dev/null +++ b/TransPyV/Test/temp/c3a6aed1f1fb8b1e.deps.txt @@ -0,0 +1 @@ +stdint stdlib string vipermath stdio t c memhub \ No newline at end of file diff --git a/TransPyV/Test/temp/c3b259b4059f8668.deps.txt b/TransPyV/Test/temp/c3b259b4059f8668.deps.txt new file mode 100644 index 0000000..97907fa --- /dev/null +++ b/TransPyV/Test/temp/c3b259b4059f8668.deps.txt @@ -0,0 +1 @@ +t c viperio stdarg \ No newline at end of file diff --git a/TransPyV/Test/temp/c9d54a4158f7f5a8.deps.txt b/TransPyV/Test/temp/c9d54a4158f7f5a8.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/c9d54a4158f7f5a8.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/c9f4be41ca1cc2b4.deps.txt b/TransPyV/Test/temp/c9f4be41ca1cc2b4.deps.txt new file mode 100644 index 0000000..b9e9795 --- /dev/null +++ b/TransPyV/Test/temp/c9f4be41ca1cc2b4.deps.txt @@ -0,0 +1 @@ +t c stdint t c stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/cc3ad7f3bc880326.deps.txt b/TransPyV/Test/temp/cc3ad7f3bc880326.deps.txt new file mode 100644 index 0000000..bd7c07e --- /dev/null +++ b/TransPyV/Test/temp/cc3ad7f3bc880326.deps.txt @@ -0,0 +1 @@ +stdio t c string inherit_test stdio t c string inherit_test \ No newline at end of file diff --git a/TransPyV/Test/temp/d152115b49e50218.deps.txt b/TransPyV/Test/temp/d152115b49e50218.deps.txt new file mode 100644 index 0000000..640d691 --- /dev/null +++ b/TransPyV/Test/temp/d152115b49e50218.deps.txt @@ -0,0 +1 @@ +t c stdint posix memhub stdlib \ No newline at end of file diff --git a/TransPyV/Test/temp/d4117f5e00a6c3b5.deps.txt b/TransPyV/Test/temp/d4117f5e00a6c3b5.deps.txt new file mode 100644 index 0000000..6289bed --- /dev/null +++ b/TransPyV/Test/temp/d4117f5e00a6c3b5.deps.txt @@ -0,0 +1 @@ +t c stdint string viperlib memhub linkedlist llvmlite.__function llvmlite.__module \ No newline at end of file diff --git a/TransPyV/Test/temp/d541ade6afb689a5.deps.txt b/TransPyV/Test/temp/d541ade6afb689a5.deps.txt new file mode 100644 index 0000000..a7f920a --- /dev/null +++ b/TransPyV/Test/temp/d541ade6afb689a5.deps.txt @@ -0,0 +1 @@ +stdint zlib.zdef zdef memhub t c \ No newline at end of file diff --git a/TransPyV/Test/temp/d63d2c2a2bd9bd2a.deps.txt b/TransPyV/Test/temp/d63d2c2a2bd9bd2a.deps.txt new file mode 100644 index 0000000..1b0c811 --- /dev/null +++ b/TransPyV/Test/temp/d63d2c2a2bd9bd2a.deps.txt @@ -0,0 +1 @@ +zlib.pyzlib \ No newline at end of file diff --git a/TransPyV/Test/temp/d7662500f1c2c0ff.deps.txt b/TransPyV/Test/temp/d7662500f1c2c0ff.deps.txt new file mode 100644 index 0000000..9e0af1f --- /dev/null +++ b/TransPyV/Test/temp/d7662500f1c2c0ff.deps.txt @@ -0,0 +1 @@ +stdio t c string testcheck stdio t c string testcheck \ No newline at end of file diff --git a/TransPyV/Test/temp/d7e3386b828acb66.deps.txt b/TransPyV/Test/temp/d7e3386b828acb66.deps.txt new file mode 100644 index 0000000..00494d4 --- /dev/null +++ b/TransPyV/Test/temp/d7e3386b828acb66.deps.txt @@ -0,0 +1 @@ +t c stdint string platmacro stdio w32.winsock2 \ No newline at end of file diff --git a/TransPyV/Test/temp/da44924f0777c67a.deps.txt b/TransPyV/Test/temp/da44924f0777c67a.deps.txt new file mode 100644 index 0000000..9546553 --- /dev/null +++ b/TransPyV/Test/temp/da44924f0777c67a.deps.txt @@ -0,0 +1 @@ +t c stdint w32.win32base w32.win32sync \ No newline at end of file diff --git a/TransPyV/Test/temp/dd3002730623424b.deps.txt b/TransPyV/Test/temp/dd3002730623424b.deps.txt new file mode 100644 index 0000000..8477734 --- /dev/null +++ b/TransPyV/Test/temp/dd3002730623424b.deps.txt @@ -0,0 +1 @@ +t c stdio w32.win32console t c stdio w32.win32console \ No newline at end of file diff --git a/TransPyV/Test/temp/e0009679269c6b63.deps.txt b/TransPyV/Test/temp/e0009679269c6b63.deps.txt new file mode 100644 index 0000000..7cff98f --- /dev/null +++ b/TransPyV/Test/temp/e0009679269c6b63.deps.txt @@ -0,0 +1 @@ +stdio t c string stdio t c string \ No newline at end of file diff --git a/TransPyV/Test/temp/e0b84369e50c937c.deps.txt b/TransPyV/Test/temp/e0b84369e50c937c.deps.txt new file mode 100644 index 0000000..fa3e631 --- /dev/null +++ b/TransPyV/Test/temp/e0b84369e50c937c.deps.txt @@ -0,0 +1 @@ +t c stdint string viperlib memhub stdio linkedlist llvmlite.__types \ No newline at end of file diff --git a/TransPyV/Test/temp/e0f1d864a10b2e4d.deps.txt b/TransPyV/Test/temp/e0f1d864a10b2e4d.deps.txt new file mode 100644 index 0000000..b092476 --- /dev/null +++ b/TransPyV/Test/temp/e0f1d864a10b2e4d.deps.txt @@ -0,0 +1 @@ +t c \ No newline at end of file diff --git a/TransPyV/Test/temp/e3e7b6de8d7d8b03.deps.txt b/TransPyV/Test/temp/e3e7b6de8d7d8b03.deps.txt new file mode 100644 index 0000000..29a795b --- /dev/null +++ b/TransPyV/Test/temp/e3e7b6de8d7d8b03.deps.txt @@ -0,0 +1 @@ +t c memhub string json stdint t c memhub string json stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/e74870eba3e6d774.deps.txt b/TransPyV/Test/temp/e74870eba3e6d774.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/e74870eba3e6d774.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/ea80a4a724accbda.deps.txt b/TransPyV/Test/temp/ea80a4a724accbda.deps.txt new file mode 100644 index 0000000..32f64f4 --- /dev/null +++ b/TransPyV/Test/temp/ea80a4a724accbda.deps.txt @@ -0,0 +1 @@ +t \ No newline at end of file diff --git a/TransPyV/Test/temp/eb252050d92f8fb9.deps.txt b/TransPyV/Test/temp/eb252050d92f8fb9.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/eb252050d92f8fb9.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/ed1c124f3cb787c9.deps.txt b/TransPyV/Test/temp/ed1c124f3cb787c9.deps.txt new file mode 100644 index 0000000..0fbad20 --- /dev/null +++ b/TransPyV/Test/temp/ed1c124f3cb787c9.deps.txt @@ -0,0 +1 @@ +stdio t c stdio t c \ No newline at end of file diff --git a/TransPyV/Test/temp/ee084e9fc6ee413a.deps.txt b/TransPyV/Test/temp/ee084e9fc6ee413a.deps.txt new file mode 100644 index 0000000..67c49ce --- /dev/null +++ b/TransPyV/Test/temp/ee084e9fc6ee413a.deps.txt @@ -0,0 +1 @@ +t c stdint string atom viperio t c stdint string atom viperio \ No newline at end of file diff --git a/TransPyV/Test/temp/ee52c08239684346.deps.txt b/TransPyV/Test/temp/ee52c08239684346.deps.txt new file mode 100644 index 0000000..7196dda --- /dev/null +++ b/TransPyV/Test/temp/ee52c08239684346.deps.txt @@ -0,0 +1 @@ +t c stdint memhub string ast.tokens ast.base ast.exprs ast.stmts ast.astaux ast.match ast.lexer \ No newline at end of file diff --git a/TransPyV/Test/temp/ef45adce81bf5b2d.deps.txt b/TransPyV/Test/temp/ef45adce81bf5b2d.deps.txt new file mode 100644 index 0000000..d405428 --- /dev/null +++ b/TransPyV/Test/temp/ef45adce81bf5b2d.deps.txt @@ -0,0 +1 @@ +stdio t c namespace_defs namespace_defs stdio t c namespace_defs namespace_defs \ No newline at end of file diff --git a/TransPyV/Test/temp/f2ecbf8ced20575c.deps.txt b/TransPyV/Test/temp/f2ecbf8ced20575c.deps.txt new file mode 100644 index 0000000..3b9e107 --- /dev/null +++ b/TransPyV/Test/temp/f2ecbf8ced20575c.deps.txt @@ -0,0 +1 @@ +t c memhub string stdint t c memhub string stdint \ No newline at end of file diff --git a/TransPyV/Test/temp/f5522571bcce7bcb.deps.txt b/TransPyV/Test/temp/f5522571bcce7bcb.deps.txt new file mode 100644 index 0000000..7a8d4dc --- /dev/null +++ b/TransPyV/Test/temp/f5522571bcce7bcb.deps.txt @@ -0,0 +1 @@ +t t \ No newline at end of file diff --git a/TransPyV/Test/temp/f6b51804a0ba8ff0.deps.txt b/TransPyV/Test/temp/f6b51804a0ba8ff0.deps.txt new file mode 100644 index 0000000..e28a09e --- /dev/null +++ b/TransPyV/Test/temp/f6b51804a0ba8ff0.deps.txt @@ -0,0 +1 @@ +t stdint w32.win32base \ No newline at end of file diff --git a/TransPyV/output/001c1416e7a7dc50.deps.json b/TransPyV/output/001c1416e7a7dc50.deps.json new file mode 100644 index 0000000..2813387 --- /dev/null +++ b/TransPyV/output/001c1416e7a7dc50.deps.json @@ -0,0 +1 @@ +{"w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesMain.tokens": "b547ac4f380bddb6", "HandlesMain.base": "5dab8cb390496d22", "HandlesMain.exprs": "47767b5026a8ee15", "HandlesMain.stmts": "657e182b27c2a022", "HandlesMain.astaux": "4337fb260448bbe2", "HandlesMain.match": "4dd6b3f1427d1cc5", "HandlesMain.lexer": "0c212981c180e7fb", "HandlesMain.__types": "455ff4e74c3dbfb0", "HandlesMain.__function": "e0b84369e50c937c", "HandlesMain.__values": "05c11f22675c60cb", "HandlesMain.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/02885f6f134079c2.deps.json b/TransPyV/output/02885f6f134079c2.deps.json new file mode 100644 index 0000000..ac5c99c --- /dev/null +++ b/TransPyV/output/02885f6f134079c2.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesVar.__types": "455ff4e74c3dbfb0", "HandlesVar.__function": "e0b84369e50c937c", "HandlesVar.__values": "05c11f22675c60cb", "HandlesVar.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/07a8c82a8ca2be90.deps.json b/TransPyV/output/07a8c82a8ca2be90.deps.json new file mode 100644 index 0000000..2766a9c --- /dev/null +++ b/TransPyV/output/07a8c82a8ca2be90.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesExprOps.tokens": "b547ac4f380bddb6", "HandlesExprOps.base": "5dab8cb390496d22", "HandlesExprOps.exprs": "47767b5026a8ee15", "HandlesExprOps.stmts": "657e182b27c2a022", "HandlesExprOps.astaux": "4337fb260448bbe2", "HandlesExprOps.match": "4dd6b3f1427d1cc5", "HandlesExprOps.lexer": "0c212981c180e7fb", "HandlesExprOps.__types": "455ff4e74c3dbfb0", "HandlesExprOps.__function": "e0b84369e50c937c", "HandlesExprOps.__values": "05c11f22675c60cb", "HandlesExprOps.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/0c56128647a0b845.deps.json b/TransPyV/output/0c56128647a0b845.deps.json new file mode 100644 index 0000000..1ce21cf --- /dev/null +++ b/TransPyV/output/0c56128647a0b845.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesType.tokens": "b547ac4f380bddb6", "HandlesType.base": "5dab8cb390496d22", "HandlesType.exprs": "47767b5026a8ee15", "HandlesType.stmts": "657e182b27c2a022", "HandlesType.astaux": "4337fb260448bbe2", "HandlesType.match": "4dd6b3f1427d1cc5", "HandlesType.lexer": "0c212981c180e7fb", "HandlesType.__types": "455ff4e74c3dbfb0", "HandlesType.__function": "e0b84369e50c937c", "HandlesType.__values": "05c11f22675c60cb", "HandlesType.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/21fffba3315e81b4.deps.json b/TransPyV/output/21fffba3315e81b4.deps.json new file mode 100644 index 0000000..950882c --- /dev/null +++ b/TransPyV/output/21fffba3315e81b4.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/2998b8f78d7022bf.deps.json b/TransPyV/output/2998b8f78d7022bf.deps.json new file mode 100644 index 0000000..b9b6287 --- /dev/null +++ b/TransPyV/output/2998b8f78d7022bf.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesTranslator.tokens": "b547ac4f380bddb6", "HandlesTranslator.base": "5dab8cb390496d22", "HandlesTranslator.exprs": "47767b5026a8ee15", "HandlesTranslator.stmts": "657e182b27c2a022", "HandlesTranslator.astaux": "4337fb260448bbe2", "HandlesTranslator.match": "4dd6b3f1427d1cc5", "HandlesTranslator.lexer": "0c212981c180e7fb", "HandlesTranslator.__types": "455ff4e74c3dbfb0", "HandlesTranslator.__function": "e0b84369e50c937c", "HandlesTranslator.__values": "05c11f22675c60cb", "HandlesTranslator.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/2ed380190e9df89e.deps.json b/TransPyV/output/2ed380190e9df89e.deps.json new file mode 100644 index 0000000..9f16572 --- /dev/null +++ b/TransPyV/output/2ed380190e9df89e.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesAnnAssign.tokens": "b547ac4f380bddb6", "HandlesAnnAssign.base": "5dab8cb390496d22", "HandlesAnnAssign.exprs": "47767b5026a8ee15", "HandlesAnnAssign.stmts": "657e182b27c2a022", "HandlesAnnAssign.astaux": "4337fb260448bbe2", "HandlesAnnAssign.match": "4dd6b3f1427d1cc5", "HandlesAnnAssign.lexer": "0c212981c180e7fb", "HandlesAnnAssign.__types": "455ff4e74c3dbfb0", "HandlesAnnAssign.__function": "e0b84369e50c937c", "HandlesAnnAssign.__values": "05c11f22675c60cb", "HandlesAnnAssign.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/3d3edda4a4e6c08b.deps.json b/TransPyV/output/3d3edda4a4e6c08b.deps.json new file mode 100644 index 0000000..e0aeee7 --- /dev/null +++ b/TransPyV/output/3d3edda4a4e6c08b.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesAssign.tokens": "b547ac4f380bddb6", "HandlesAssign.base": "5dab8cb390496d22", "HandlesAssign.exprs": "47767b5026a8ee15", "HandlesAssign.stmts": "657e182b27c2a022", "HandlesAssign.astaux": "4337fb260448bbe2", "HandlesAssign.match": "4dd6b3f1427d1cc5", "HandlesAssign.lexer": "0c212981c180e7fb", "HandlesAssign.__types": "455ff4e74c3dbfb0", "HandlesAssign.__function": "e0b84369e50c937c", "HandlesAssign.__values": "05c11f22675c60cb", "HandlesAssign.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/4020cac9bab5e9f1.deps.json b/TransPyV/output/4020cac9bab5e9f1.deps.json new file mode 100644 index 0000000..e7e9c92 --- /dev/null +++ b/TransPyV/output/4020cac9bab5e9f1.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesFunctions.tokens": "b547ac4f380bddb6", "HandlesFunctions.base": "5dab8cb390496d22", "HandlesFunctions.exprs": "47767b5026a8ee15", "HandlesFunctions.stmts": "657e182b27c2a022", "HandlesFunctions.astaux": "4337fb260448bbe2", "HandlesFunctions.match": "4dd6b3f1427d1cc5", "HandlesFunctions.lexer": "0c212981c180e7fb", "HandlesFunctions.__types": "455ff4e74c3dbfb0", "HandlesFunctions.__function": "e0b84369e50c937c", "HandlesFunctions.__values": "05c11f22675c60cb", "HandlesFunctions.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/48fa8cc758d3826b.deps.json b/TransPyV/output/48fa8cc758d3826b.deps.json new file mode 100644 index 0000000..4a1e1c6 --- /dev/null +++ b/TransPyV/output/48fa8cc758d3826b.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesFor.tokens": "b547ac4f380bddb6", "HandlesFor.base": "5dab8cb390496d22", "HandlesFor.exprs": "47767b5026a8ee15", "HandlesFor.stmts": "657e182b27c2a022", "HandlesFor.astaux": "4337fb260448bbe2", "HandlesFor.match": "4dd6b3f1427d1cc5", "HandlesFor.lexer": "0c212981c180e7fb", "HandlesFor.__types": "455ff4e74c3dbfb0", "HandlesFor.__function": "e0b84369e50c937c", "HandlesFor.__values": "05c11f22675c60cb", "HandlesFor.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/49cf633017627a4a.deps.json b/TransPyV/output/49cf633017627a4a.deps.json new file mode 100644 index 0000000..e020d99 --- /dev/null +++ b/TransPyV/output/49cf633017627a4a.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/51f0144d6e8188d9.deps.json b/TransPyV/output/51f0144d6e8188d9.deps.json new file mode 100644 index 0000000..a97def0 --- /dev/null +++ b/TransPyV/output/51f0144d6e8188d9.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesWhile.tokens": "b547ac4f380bddb6", "HandlesWhile.base": "5dab8cb390496d22", "HandlesWhile.exprs": "47767b5026a8ee15", "HandlesWhile.stmts": "657e182b27c2a022", "HandlesWhile.astaux": "4337fb260448bbe2", "HandlesWhile.match": "4dd6b3f1427d1cc5", "HandlesWhile.lexer": "0c212981c180e7fb", "HandlesWhile.__types": "455ff4e74c3dbfb0", "HandlesWhile.__function": "e0b84369e50c937c", "HandlesWhile.__values": "05c11f22675c60cb", "HandlesWhile.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/54014f6c53e9640b.deps.json b/TransPyV/output/54014f6c53e9640b.deps.json new file mode 100644 index 0000000..3f4b699 --- /dev/null +++ b/TransPyV/output/54014f6c53e9640b.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesStruct.__types": "455ff4e74c3dbfb0", "HandlesStruct.__function": "e0b84369e50c937c", "HandlesStruct.__values": "05c11f22675c60cb", "HandlesStruct.__module": "21a7fcfc665f75ef", "HandlesStruct.tokens": "b547ac4f380bddb6", "HandlesStruct.base": "5dab8cb390496d22", "HandlesStruct.exprs": "47767b5026a8ee15", "HandlesStruct.stmts": "657e182b27c2a022", "HandlesStruct.astaux": "4337fb260448bbe2", "HandlesStruct.match": "4dd6b3f1427d1cc5", "HandlesStruct.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/TransPyV/output/541a090dde0cc1ba.deps.json b/TransPyV/output/541a090dde0cc1ba.deps.json new file mode 100644 index 0000000..9b94863 --- /dev/null +++ b/TransPyV/output/541a090dde0cc1ba.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesAugAssign.tokens": "b547ac4f380bddb6", "HandlesAugAssign.base": "5dab8cb390496d22", "HandlesAugAssign.exprs": "47767b5026a8ee15", "HandlesAugAssign.stmts": "657e182b27c2a022", "HandlesAugAssign.astaux": "4337fb260448bbe2", "HandlesAugAssign.match": "4dd6b3f1427d1cc5", "HandlesAugAssign.lexer": "0c212981c180e7fb", "HandlesAugAssign.__types": "455ff4e74c3dbfb0", "HandlesAugAssign.__function": "e0b84369e50c937c", "HandlesAugAssign.__values": "05c11f22675c60cb", "HandlesAugAssign.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/66de93aaeebc0601.deps.json b/TransPyV/output/66de93aaeebc0601.deps.json new file mode 100644 index 0000000..a260532 --- /dev/null +++ b/TransPyV/output/66de93aaeebc0601.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesIf.tokens": "b547ac4f380bddb6", "HandlesIf.base": "5dab8cb390496d22", "HandlesIf.exprs": "47767b5026a8ee15", "HandlesIf.stmts": "657e182b27c2a022", "HandlesIf.astaux": "4337fb260448bbe2", "HandlesIf.match": "4dd6b3f1427d1cc5", "HandlesIf.lexer": "0c212981c180e7fb", "HandlesIf.__types": "455ff4e74c3dbfb0", "HandlesIf.__function": "e0b84369e50c937c", "HandlesIf.__values": "05c11f22675c60cb", "HandlesIf.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/78bbe886bbe5ca8f.deps.json b/TransPyV/output/78bbe886bbe5ca8f.deps.json new file mode 100644 index 0000000..085d856 --- /dev/null +++ b/TransPyV/output/78bbe886bbe5ca8f.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesImports.tokens": "b547ac4f380bddb6", "HandlesImports.base": "5dab8cb390496d22", "HandlesImports.exprs": "47767b5026a8ee15", "HandlesImports.stmts": "657e182b27c2a022", "HandlesImports.astaux": "4337fb260448bbe2", "HandlesImports.match": "4dd6b3f1427d1cc5", "HandlesImports.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/TransPyV/output/7e260c14d2712119.deps.json b/TransPyV/output/7e260c14d2712119.deps.json new file mode 100644 index 0000000..e41984e --- /dev/null +++ b/TransPyV/output/7e260c14d2712119.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "Phase2.tokens": "b547ac4f380bddb6", "Phase2.base": "5dab8cb390496d22", "Phase2.exprs": "47767b5026a8ee15", "Phase2.stmts": "657e182b27c2a022", "Phase2.astaux": "4337fb260448bbe2", "Phase2.match": "4dd6b3f1427d1cc5", "Phase2.lexer": "0c212981c180e7fb", "Phase2.__types": "455ff4e74c3dbfb0", "Phase2.__function": "e0b84369e50c937c", "Phase2.__values": "05c11f22675c60cb", "Phase2.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/81eb28ba912467a6.deps.json b/TransPyV/output/81eb28ba912467a6.deps.json new file mode 100644 index 0000000..52cd118 --- /dev/null +++ b/TransPyV/output/81eb28ba912467a6.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesBody.tokens": "b547ac4f380bddb6", "HandlesBody.base": "5dab8cb390496d22", "HandlesBody.exprs": "47767b5026a8ee15", "HandlesBody.stmts": "657e182b27c2a022", "HandlesBody.astaux": "4337fb260448bbe2", "HandlesBody.match": "4dd6b3f1427d1cc5", "HandlesBody.lexer": "0c212981c180e7fb", "HandlesBody.__types": "455ff4e74c3dbfb0", "HandlesBody.__function": "e0b84369e50c937c", "HandlesBody.__values": "05c11f22675c60cb", "HandlesBody.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/83ba314b4637eec9.deps.json b/TransPyV/output/83ba314b4637eec9.deps.json new file mode 100644 index 0000000..f32fa7b --- /dev/null +++ b/TransPyV/output/83ba314b4637eec9.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/888ea5bd6ab516ae.deps.json b/TransPyV/output/888ea5bd6ab516ae.deps.json new file mode 100644 index 0000000..6627aae --- /dev/null +++ b/TransPyV/output/888ea5bd6ab516ae.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/8f12094c840be06a.deps.json b/TransPyV/output/8f12094c840be06a.deps.json new file mode 100644 index 0000000..6450285 --- /dev/null +++ b/TransPyV/output/8f12094c840be06a.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/950028de389b7a7b.deps.json b/TransPyV/output/950028de389b7a7b.deps.json new file mode 100644 index 0000000..c1f4c6a --- /dev/null +++ b/TransPyV/output/950028de389b7a7b.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "main.tokens": "b547ac4f380bddb6", "main.base": "5dab8cb390496d22", "main.exprs": "47767b5026a8ee15", "main.stmts": "657e182b27c2a022", "main.astaux": "4337fb260448bbe2", "main.match": "4dd6b3f1427d1cc5", "main.lexer": "0c212981c180e7fb", "main.__types": "455ff4e74c3dbfb0", "main.__function": "e0b84369e50c937c", "main.__values": "05c11f22675c60cb", "main.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/ae0cc890f657076d.deps.json b/TransPyV/output/ae0cc890f657076d.deps.json new file mode 100644 index 0000000..aed6921 --- /dev/null +++ b/TransPyV/output/ae0cc890f657076d.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesExpr.tokens": "b547ac4f380bddb6", "HandlesExpr.base": "5dab8cb390496d22", "HandlesExpr.exprs": "47767b5026a8ee15", "HandlesExpr.stmts": "657e182b27c2a022", "HandlesExpr.astaux": "4337fb260448bbe2", "HandlesExpr.match": "4dd6b3f1427d1cc5", "HandlesExpr.lexer": "0c212981c180e7fb", "HandlesExpr.__types": "455ff4e74c3dbfb0", "HandlesExpr.__function": "e0b84369e50c937c", "HandlesExpr.__values": "05c11f22675c60cb", "HandlesExpr.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/af9484a65085bf4b.deps.json b/TransPyV/output/af9484a65085bf4b.deps.json new file mode 100644 index 0000000..7640256 --- /dev/null +++ b/TransPyV/output/af9484a65085bf4b.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesNonlocal.tokens": "b547ac4f380bddb6", "HandlesNonlocal.base": "5dab8cb390496d22", "HandlesNonlocal.exprs": "47767b5026a8ee15", "HandlesNonlocal.stmts": "657e182b27c2a022", "HandlesNonlocal.astaux": "4337fb260448bbe2", "HandlesNonlocal.match": "4dd6b3f1427d1cc5", "HandlesNonlocal.lexer": "0c212981c180e7fb", "HandlesNonlocal.__types": "455ff4e74c3dbfb0", "HandlesNonlocal.__function": "e0b84369e50c937c", "HandlesNonlocal.__values": "05c11f22675c60cb", "HandlesNonlocal.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/b29b80b6106e4be7.deps.json b/TransPyV/output/b29b80b6106e4be7.deps.json new file mode 100644 index 0000000..8f1ea90 --- /dev/null +++ b/TransPyV/output/b29b80b6106e4be7.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesExprCall.tokens": "b547ac4f380bddb6", "HandlesExprCall.base": "5dab8cb390496d22", "HandlesExprCall.exprs": "47767b5026a8ee15", "HandlesExprCall.stmts": "657e182b27c2a022", "HandlesExprCall.astaux": "4337fb260448bbe2", "HandlesExprCall.match": "4dd6b3f1427d1cc5", "HandlesExprCall.lexer": "0c212981c180e7fb", "HandlesExprCall.__types": "455ff4e74c3dbfb0", "HandlesExprCall.__function": "e0b84369e50c937c", "HandlesExprCall.__values": "05c11f22675c60cb", "HandlesExprCall.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/b5d05dbd7fee36ef.deps.json b/TransPyV/output/b5d05dbd7fee36ef.deps.json new file mode 100644 index 0000000..5aa0f78 --- /dev/null +++ b/TransPyV/output/b5d05dbd7fee36ef.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "StubMerger.__types": "455ff4e74c3dbfb0", "StubMerger.__function": "e0b84369e50c937c", "StubMerger.__values": "05c11f22675c60cb", "StubMerger.__module": "21a7fcfc665f75ef", "StubMerger.tokens": "b547ac4f380bddb6", "StubMerger.base": "5dab8cb390496d22", "StubMerger.exprs": "47767b5026a8ee15", "StubMerger.stmts": "657e182b27c2a022", "StubMerger.astaux": "4337fb260448bbe2", "StubMerger.match": "4dd6b3f1427d1cc5", "StubMerger.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/TransPyV/output/bd82e3ca81546c76.deps.json b/TransPyV/output/bd82e3ca81546c76.deps.json new file mode 100644 index 0000000..870d529 --- /dev/null +++ b/TransPyV/output/bd82e3ca81546c76.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0"} \ No newline at end of file diff --git a/TransPyV/output/c4f2d483ec0f4db5.deps.json b/TransPyV/output/c4f2d483ec0f4db5.deps.json new file mode 100644 index 0000000..46297c2 --- /dev/null +++ b/TransPyV/output/c4f2d483ec0f4db5.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesClassDef.tokens": "b547ac4f380bddb6", "HandlesClassDef.base": "5dab8cb390496d22", "HandlesClassDef.exprs": "47767b5026a8ee15", "HandlesClassDef.stmts": "657e182b27c2a022", "HandlesClassDef.astaux": "4337fb260448bbe2", "HandlesClassDef.match": "4dd6b3f1427d1cc5", "HandlesClassDef.lexer": "0c212981c180e7fb", "HandlesClassDef.__types": "455ff4e74c3dbfb0", "HandlesClassDef.__function": "e0b84369e50c937c", "HandlesClassDef.__values": "05c11f22675c60cb", "HandlesClassDef.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/d1599ecbac5b6586.deps.json b/TransPyV/output/d1599ecbac5b6586.deps.json new file mode 100644 index 0000000..6f43b48 --- /dev/null +++ b/TransPyV/output/d1599ecbac5b6586.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "BuildPipeline.tokens": "b547ac4f380bddb6", "BuildPipeline.base": "5dab8cb390496d22", "BuildPipeline.exprs": "47767b5026a8ee15", "BuildPipeline.stmts": "657e182b27c2a022", "BuildPipeline.astaux": "4337fb260448bbe2", "BuildPipeline.match": "4dd6b3f1427d1cc5", "BuildPipeline.lexer": "0c212981c180e7fb"} \ No newline at end of file diff --git a/TransPyV/output/d9322f0db660edd8.deps.json b/TransPyV/output/d9322f0db660edd8.deps.json new file mode 100644 index 0000000..8ba45b0 --- /dev/null +++ b/TransPyV/output/d9322f0db660edd8.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesReturn.tokens": "b547ac4f380bddb6", "HandlesReturn.base": "5dab8cb390496d22", "HandlesReturn.exprs": "47767b5026a8ee15", "HandlesReturn.stmts": "657e182b27c2a022", "HandlesReturn.astaux": "4337fb260448bbe2", "HandlesReturn.match": "4dd6b3f1427d1cc5", "HandlesReturn.lexer": "0c212981c180e7fb", "HandlesReturn.__types": "455ff4e74c3dbfb0", "HandlesReturn.__function": "e0b84369e50c937c", "HandlesReturn.__values": "05c11f22675c60cb", "HandlesReturn.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/db7c4a8f09d02a17.deps.json b/TransPyV/output/db7c4a8f09d02a17.deps.json new file mode 100644 index 0000000..8ec2103 --- /dev/null +++ b/TransPyV/output/db7c4a8f09d02a17.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "lib.core.Handles.HandlesEnum": "e5be27ae4fe7ce04", "HandlesEnum": "e5be27ae4fe7ce04", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "Phase1.tokens": "b547ac4f380bddb6", "Phase1.base": "5dab8cb390496d22", "Phase1.exprs": "47767b5026a8ee15", "Phase1.stmts": "657e182b27c2a022", "Phase1.astaux": "4337fb260448bbe2", "Phase1.match": "4dd6b3f1427d1cc5", "Phase1.lexer": "0c212981c180e7fb", "Phase1.__types": "455ff4e74c3dbfb0", "Phase1.__function": "e0b84369e50c937c", "Phase1.__values": "05c11f22675c60cb", "Phase1.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/output/e5be27ae4fe7ce04.deps.json b/TransPyV/output/e5be27ae4fe7ce04.deps.json new file mode 100644 index 0000000..c89e6bc --- /dev/null +++ b/TransPyV/output/e5be27ae4fe7ce04.deps.json @@ -0,0 +1 @@ +{"lib.core.Handles.HandlesMain": "001c1416e7a7dc50", "HandlesMain": "001c1416e7a7dc50", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "lib.core.Handles.HandlesVar": "02885f6f134079c2", "HandlesVar": "02885f6f134079c2", "llvmlite.__values": "05c11f22675c60cb", "__values": "05c11f22675c60cb", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "lib.core.Handles.HandlesExprOps": "07a8c82a8ca2be90", "HandlesExprOps": "07a8c82a8ca2be90", "ast.lexer": "0c212981c180e7fb", "lexer": "0c212981c180e7fb", "lib.core.Handles.HandlesType": "0c56128647a0b845", "HandlesType": "0c56128647a0b845", "hashlib.__sha1": "0df65b8ed15664b0", "__sha1": "0df65b8ed15664b0", "hashlib.__md5": "19f8024d10c828e8", "__md5": "19f8024d10c828e8", "json.__writer": "20cd49775c100a38", "__writer": "20cd49775c100a38", "llvmlite.__module": "21a7fcfc665f75ef", "__module": "21a7fcfc665f75ef", "lib.core.VLogger": "21fffba3315e81b4", "VLogger": "21fffba3315e81b4", "json.__parser": "240a9a4157959a9f", "__parser": "240a9a4157959a9f", "atom": "271ea3decb810db2", "lib.core.Handles.HandlesTranslator": "2998b8f78d7022bf", "HandlesTranslator": "2998b8f78d7022bf", "linkedlist": "2d39c6c7d3557b3e", "subprocess": "2da636c61863c815", "lib.core.Handles.HandlesAnnAssign": "2ed380190e9df89e", "HandlesAnnAssign": "2ed380190e9df89e", "ast.__init__": "34548789d646f432", "__init__": "96837bcc64032444", "ast": "34548789d646f432", "lib.core.Handles.HandlesAssign": "3d3edda4a4e6c08b", "HandlesAssign": "3d3edda4a4e6c08b", "lib.core.Handles.HandlesFunctions": "4020cac9bab5e9f1", "HandlesFunctions": "4020cac9bab5e9f1", "ast.astaux": "4337fb260448bbe2", "astaux": "4337fb260448bbe2", "llvmlite.__types": "455ff4e74c3dbfb0", "__types": "455ff4e74c3dbfb0", "llvmlite.__builder": "46ea362b72c10a9e", "__builder": "46ea362b72c10a9e", "ast.exprs": "47767b5026a8ee15", "exprs": "47767b5026a8ee15", "lib.core.Handles.HandlesFor": "48fa8cc758d3826b", "HandlesFor": "48fa8cc758d3826b", "lib.Projectrans.Config": "49cf633017627a4a", "Config": "49cf633017627a4a", "ast.match": "4dd6b3f1427d1cc5", "match": "4dd6b3f1427d1cc5", "lib.core.Handles.HandlesWhile": "51f0144d6e8188d9", "HandlesWhile": "51f0144d6e8188d9", "lib.core.Handles.HandlesStruct": "54014f6c53e9640b", "HandlesStruct": "54014f6c53e9640b", "lib.core.Handles.HandlesAugAssign": "541a090dde0cc1ba", "HandlesAugAssign": "541a090dde0cc1ba", "json.__init__": "57288496f7c2d1ad", "json": "57288496f7c2d1ad", "ast.base": "5dab8cb390496d22", "base": "5dab8cb390496d22", "posix": "6503c97dde0c79c4", "ast.stmts": "657e182b27c2a022", "stmts": "657e182b27c2a022", "lib.core.Handles.HandlesIf": "66de93aaeebc0601", "HandlesIf": "66de93aaeebc0601", "stdio": "6f62fe05c5ea1ceb", "hashlib.__sha512": "6ff26590374ae6fc", "__sha512": "6ff26590374ae6fc", "stdarg": "71e0a3ffcb3ebfad", "lib.core.Handles.HandlesImports": "78bbe886bbe5ca8f", "HandlesImports": "78bbe886bbe5ca8f", "lib.core.Phase2": "7e260c14d2712119", "Phase2": "7e260c14d2712119", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "lib.core.Handles.HandlesBody": "81eb28ba912467a6", "HandlesBody": "81eb28ba912467a6", "lib.core.Handles.HandlesBase": "83ba314b4637eec9", "HandlesBase": "83ba314b4637eec9", "lib.Projectrans.Utils": "888ea5bd6ab516ae", "Utils": "888ea5bd6ab516ae", "lib.__init__": "8f12094c840be06a", "lib": "8f12094c840be06a", "stdlib": "90c53dd6db8d41cf", "platmacro": "93c1d18e35d188d6", "main": "950028de389b7a7b", "llvmlite.__init__": "95394ca8da0f655a", "llvmlite": "95394ca8da0f655a", "hashlib.__init__": "96837bcc64032444", "hashlib": "96837bcc64032444", "string": "ab6e54ba9a669f76", "argparse": "aba439b7882ad9d6", "lib.core.Handles.HandlesExpr": "ae0cc890f657076d", "HandlesExpr": "ae0cc890f657076d", "lib.core.Handles.HandlesNonlocal": "af9484a65085bf4b", "HandlesNonlocal": "af9484a65085bf4b", "lib.core.Handles.HandlesExprCall": "b29b80b6106e4be7", "HandlesExprCall": "b29b80b6106e4be7", "ast.tokens": "b547ac4f380bddb6", "tokens": "b547ac4f380bddb6", "sys": "b5a965302cded36b", "lib.core.StubMerger": "b5d05dbd7fee36ef", "StubMerger": "b5d05dbd7fee36ef", "hashtable": "b8c66c8ff44eb874", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "lib.core.IncludesScanner": "bd82e3ca81546c76", "IncludesScanner": "bd82e3ca81546c76", "viperlib": "c3b259b4059f8668", "lib.core.Handles.HandlesClassDef": "c4f2d483ec0f4db5", "HandlesClassDef": "c4f2d483ec0f4db5", "hashlib.__sha256": "c9d54a4158f7f5a8", "__sha256": "c9d54a4158f7f5a8", "viperio": "c9f4be41ca1cc2b4", "lib.core.BuildPipeline": "d1599ecbac5b6586", "BuildPipeline": "d1599ecbac5b6586", "llvmlite.__verify": "d4117f5e00a6c3b5", "__verify": "d4117f5e00a6c3b5", "lib.core.Handles.HandlesReturn": "d9322f0db660edd8", "HandlesReturn": "d9322f0db660edd8", "lib.core.Phase1": "db7c4a8f09d02a17", "Phase1": "db7c4a8f09d02a17", "llvmlite.__function": "e0b84369e50c937c", "__function": "e0b84369e50c937c", "memhub": "ee084e9fc6ee413a", "ast.parser": "ee52c08239684346", "parser": "ee52c08239684346", "_list": "f2ecbf8ced20575c", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "HandlesEnum.__types": "455ff4e74c3dbfb0", "HandlesEnum.__function": "e0b84369e50c937c", "HandlesEnum.__values": "05c11f22675c60cb", "HandlesEnum.__module": "21a7fcfc665f75ef"} \ No newline at end of file diff --git a/TransPyV/project.json b/TransPyV/project.json new file mode 100644 index 0000000..7fe93e3 --- /dev/null +++ b/TransPyV/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "TransPyV", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition", "-Wl,--stack,0x1000000"], + "output": "TransPyV.exe" + }, + "includes": [ + "../includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} \ No newline at end of file diff --git a/TransPyV/temp/001c1416e7a7dc50.doc.json b/TransPyV/temp/001c1416e7a7dc50.doc.json new file mode 100644 index 0000000..28f2ad8 --- /dev/null +++ b/TransPyV/temp/001c1416e7a7dc50.doc.json @@ -0,0 +1 @@ +{"_register_cexport_from_funcdef": "预注册 CExport/State 函数到全局表（仅注册，不生成 IR）", "translate_children": "遍历 tree.children 并翻译每个子节点\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n\n Returns:\n t.CInt: 新增的变量数\n ", "handle_module_level_var": "处理模块级变量声明，创建全局变量", "create_wrapper_main": "创建包装 main 函数并翻译所有顶层语句\n\n Args:\n trans: 翻译器（含所有共享状态）\n tree: AST 模块节点\n i32_ty: i32 LLVMType\n\n Returns:\n t.CInt: 新增的变量数\n "} \ No newline at end of file diff --git a/TransPyV/temp/001c1416e7a7dc50.pyi b/TransPyV/temp/001c1416e7a7dc50.pyi new file mode 100644 index 0000000..3901124 --- /dev/null +++ b/TransPyV/temp/001c1416e7a7dc50.pyi @@ -0,0 +1,31 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesMain.py +Module: lib.core.Handles.HandlesMain +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign + +def _register_cexport_from_funcdef(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_children(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def handle_module_level_var(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_wrapper_main(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> int: pass diff --git a/TransPyV/temp/0035c95a18d4f8e8.pyi b/TransPyV/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/TransPyV/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/TransPyV/temp/02885f6f134079c2.doc.json b/TransPyV/temp/02885f6f134079c2.doc.json new file mode 100644 index 0000000..1a7c43a --- /dev/null +++ b/TransPyV/temp/02885f6f134079c2.doc.json @@ -0,0 +1 @@ +{"VarEntry": "变量表条目", "Scope": "作用域节点", "SymbolTable": "嵌套作用域符号表", "init_vars": "分配并清零变量表数组", "find_var": "在变量表中按名称查找", "find_var_entry": "在变量表中按名称查找，返回 VarEntry 或 None", "lookup_var_entry": "从当前作用域逐级向上查找变量，返回 VarEntry 或 None", "set_var_annot_class_name": "设置变量的原始类型注解类名，返回 0=成功 / 1=失败", "add_var": "添加变量到变量表", "get_or_create_var": "查找或创建变量 alloca", "_create_scope": "创建并初始化作用域节点", "init_symbol_table": "创建并初始化符号表，包含模块级根作用域", "enter_scope": "进入新作用域，返回新创建的作用域", "exit_scope": "退出当前作用域，恢复到父作用域", "define_var": "在当前作用域定义变量，返回 0 成功", "define_module_var": "在模块作用域定义变量，返回 0 成功", "lookup_var": "从当前作用域逐级向上查找变量，返回 alloca 或 None", "lookup_current": "仅在当前作用域查找变量", "lookup_module_var": "在模块作用域查找变量", "get_or_create_sym": "查找或创建变量 alloca（在当前作用域）"} \ No newline at end of file diff --git a/TransPyV/temp/02885f6f134079c2.pyi b/TransPyV/temp/02885f6f134079c2.pyi new file mode 100644 index 0000000..fb5e055 --- /dev/null +++ b/TransPyV/temp/02885f6f134079c2.pyi @@ -0,0 +1,68 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesVar.py +Module: lib.core.Handles.HandlesVar +""" + + +import t, c +from stdint import * +import llvmlite +import memhub +import string + +SCOPE_MODULE: t.CDefine = 0 +SCOPE_FUNCTION: t.CDefine = 1 +SCOPE_BLOCK: t.CDefine = 2 +MAX_VARS: t.CDefine = 256 + +@t.NoVTable +class VarEntry: + Name: t.CChar | t.CPtr + Alloca: llvmlite.Value | t.CPtr + Used: t.CInt + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class Scope: + Parent: Scope | t.CPtr + Vars: VarEntry | t.CPtr + VarCount: t.CInt + Kind: t.CInt +@t.NoVTable +class SymbolTable: + Pool: memhub.MemBuddy | t.CPtr + Root: Scope | t.CPtr + Current: Scope | t.CPtr + +def init_vars(pool: memhub.MemBuddy | t.CPtr) -> VarEntry | t.CPtr: pass + +def find_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> llvmlite.Value | t.CPtr: pass + +def find_var_entry(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str) -> VarEntry | t.CPtr: pass + +def lookup_var_entry(symtab: SymbolTable | t.CPtr, name: str) -> VarEntry | t.CPtr: pass + +def set_var_annot_class_name(symtab: SymbolTable | t.CPtr, name: str, class_name: str) -> int: pass + +def add_var(vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def get_or_create_var(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, vars_ptr: VarEntry | t.CPtr, var_count: int, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _create_scope(pool: memhub.MemBuddy | t.CPtr, parent: Scope | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def init_symbol_table(pool: memhub.MemBuddy | t.CPtr) -> SymbolTable | t.CPtr: pass + +def enter_scope(symtab: SymbolTable | t.CPtr, kind: int) -> Scope | t.CPtr: pass + +def exit_scope(symtab: SymbolTable | t.CPtr) -> t.CInt: pass + +def define_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def define_module_var(symtab: SymbolTable | t.CPtr, name: str, alloca: llvmlite.Value | t.CPtr) -> int: pass + +def lookup_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_current(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def lookup_module_var(symtab: SymbolTable | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def get_or_create_sym(symtab: SymbolTable | t.CPtr, pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/TransPyV/temp/05c11f22675c60cb.pyi b/TransPyV/temp/05c11f22675c60cb.pyi new file mode 100644 index 0000000..3520489 --- /dev/null +++ b/TransPyV/temp/05c11f22675c60cb.pyi @@ -0,0 +1,50 @@ +""" +Auto-generated Python stub file from llvmlite.__values.py +Module: llvmlite.__values +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode +from .__types import LLVMType, TypePrint + +class Value(GSListNode[Value]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + IsConst: t.CInt + IntVal: t.CInt64T + FloatVal: t.CDouble + +VALUE_SIZE: t.CDefine = 48 + +def new_value(pool: memhub.MemBuddy | t.CPtr) -> Value | t.CPtr: pass + +def ConstInt(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CInt64T, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstNull(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ConstZero(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def ConstFloat(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + +def SSAValue(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + +def ValuePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def value_get_ty(val: Value | t.CPtr) -> LLVMType | t.CPtr: pass + +def value_get_next(val: Value | t.CPtr) -> Value | t.CPtr: pass + +def value_set_ty(val: Value | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def value_set_next(val: Value | t.CPtr, nxt: Value | t.CPtr) -> t.CInt: pass + +def value_set_name(val: Value | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def value_set_isconst(val: Value | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def _append_str(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/067c78e9f121dce3.pyi b/TransPyV/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/TransPyV/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/TransPyV/temp/06f53cc594b4ac6c.pyi b/TransPyV/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/TransPyV/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/TransPyV/temp/07a8c82a8ca2be90.doc.json b/TransPyV/temp/07a8c82a8ca2be90.doc.json new file mode 100644 index 0000000..1f30b90 --- /dev/null +++ b/TransPyV/temp/07a8c82a8ca2be90.doc.json @@ -0,0 +1 @@ +{"_binop_to_dunder": "将 BinOp 运算符映射到 dunder 方法名，无映射时返回 None", "_cmpop_to_dunder": "将 Compare 运算符映射到 dunder 方法名，无映射时返回 None", "try_operator_overload": "尝试运算符重载：lhs 是结构体指针时调用 dunder 方法，否则返回 None", "translate_binop": "翻译二元运算（自动类型提升）"} \ No newline at end of file diff --git a/TransPyV/temp/07a8c82a8ca2be90.pyi b/TransPyV/temp/07a8c82a8ca2be90.pyi new file mode 100644 index 0000000..1481115 --- /dev/null +++ b/TransPyV/temp/07a8c82a8ca2be90.pyi @@ -0,0 +1,24 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprOps.py +Module: lib.core.Handles.HandlesExprOps +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesTranslator as HT + +def _binop_to_dunder(op: int) -> str: pass + +def _cmpop_to_dunder(op: int) -> str: pass + +def try_operator_overload(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, lhs: llvmlite.Value | t.CPtr, rhs: llvmlite.Value | t.CPtr, op: int, trans: HT.Translator | t.CPtr, is_compare: int) -> llvmlite.Value | t.CPtr: pass + +def translate_binop(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/TransPyV/temp/0c212981c180e7fb.pyi b/TransPyV/temp/0c212981c180e7fb.pyi new file mode 100644 index 0000000..f432b06 --- /dev/null +++ b/TransPyV/temp/0c212981c180e7fb.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from ast.lexer.py +Module: ast.lexer +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, TokOp, _init_tables, _kw_lookup, _op_head, OpEntry, new_token, token_set_str, token_set_str_literal + +LEXER_MAX_INDENT: t.CDefine = 256 + +class Lexer: + src: str + pos: t.CSizeT + len: t.CSizeT + lineno: t.CInt + col: t.CInt + pool: memhub.MemManager | t.CPtr + indent_stack: t.CInt | t.CPtr + indent_top: t.CInt + paren_depth: t.CInt + tokens_head: Token | t.CPtr + tokens_tail: Token | t.CPtr + at_line_start: t.CInt + error_count: t.CInt + def __new__(self: Lexer, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_lexer(pool: memhub.MemManager | t.CPtr) -> Lexer | t.CPtr: pass + +def _lexer_init(lx: Lexer | t.CPtr, src: str, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(lx: Lexer | t.CPtr, offset: t.CSizeT) -> t.CChar: pass + +def _peek_cur(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _advance(lx: Lexer | t.CPtr) -> t.CChar: pass + +def _is_alpha(ch: t.CChar) -> t.CInt: pass + +def _is_digit(ch: t.CChar) -> t.CInt: pass + +def _is_alnum(ch: t.CChar) -> t.CInt: pass + +def _is_hex(ch: t.CChar) -> t.CInt: pass + +def _is_space(ch: t.CChar) -> t.CInt: pass + +def _emit(lx: Lexer | t.CPtr, tok: Token | t.CPtr) -> t.CInt: pass + +def _make_op_token(lx: Lexer | t.CPtr, op_id: t.CInt, length: t.CInt, lineno: t.CInt, col: t.CInt) -> Token | t.CPtr: pass + +def _handle_indent(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_name(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_number(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _emit_number(lx: Lexer | t.CPtr, start: t.CSizeT, start_lineno: t.CInt, start_col: t.CInt, is_float: t.CInt, is_complex: t.CInt, is_hex: t.CInt) -> t.CInt: pass + +def _read_string(lx: Lexer | t.CPtr) -> t.CInt: pass + +def _read_operator(lx: Lexer | t.CPtr) -> t.CInt: pass + +def tokenize(lx: Lexer | t.CPtr) -> Token | t.CPtr: pass diff --git a/TransPyV/temp/0c56128647a0b845.doc.json b/TransPyV/temp/0c56128647a0b845.doc.json new file mode 100644 index 0000000..47b3516 --- /dev/null +++ b/TransPyV/temp/0c56128647a0b845.doc.json @@ -0,0 +1 @@ +{"register_cdefine_constant": "注册 CDefine 编译期常量到全局表", "lookup_cdefine_constant": "查找 CDefine 编译期常量，返回值或 -1（未找到）", "clear_cdefine_constants": "清空 CDefine 常量表（模块切换时调用）", "set_generic_context": "设置泛型特化上下文（进入泛型类方法体翻译时调用）", "clear_generic_context": "清除泛型特化上下文（离开泛型类方法体翻译时调用）", "resolve_generic_type_name": "如果 name 是泛型类型参数，返回对应的具体类型名；否则返回 name 本身\n\n 用于表达式中的 T 替换（如 T.__sizeof__() → int.__sizeof__()）\n ", "_get_type_name_from_annotation": "从类型注解节点提取类型名（支持 Name/Attribute/BinOp）", "_parse_type_string": "将类型名字符串解析为 AST 节点", "set_current_file": "设置当前编译的文件名（供报错使用）", "_read_source_line": "从源文件读取第 lineno 行内容", "fatal_error": "以 Python 风格报错并退出（带源代码行）", "fatal_type_error": "以 Python 风格报错并退出（fatal_error 的别名）", "map_t_type": "将 t.CXxx / c.CXxx 类型名映射到 LLVMType\n\n 注意: t.CChar | t.CPtr = i8* = str = bytes\n ", "_is_t_attr": "检测节点是否是 t.attr_name 形式", "extract_class_name_from_annotation": "从类型注解 AST 中提取类名，返回 None=无法提取", "resolve_annotation_type": "解析类型注解节点为 LLVMType\n\n Args:\n pool: 内存分配器\n node: 类型注解 AST 节点（Name 或 Attribute）\n imported_modules: 已导入模块名字符串（空格分隔）\n from_imports: from-import 映射字符串（\"name:module name:module\" 格式）\n ", "has_decorator_marker": "检测类型注解中是否包含指定的装饰器标记", "is_pass_body": "检测函数体是否为 pass（空体或仅含 pass 语句）", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "infer_expr_type": "推断表达式的 LLVM 类型\n\n param_types 格式: \"name:bits,name:bits,...\"\n ", "infer_return_type": "扫描函数体中的 return 语句，推断返回类型", "build_param_types_str": "构建参数类型编码字符串 \"name:bits,name:bits,...\" ", "_strappend": "将 src 追加到 dst 末尾"} \ No newline at end of file diff --git a/TransPyV/temp/0c56128647a0b845.pyi b/TransPyV/temp/0c56128647a0b845.pyi new file mode 100644 index 0000000..b7e0303 --- /dev/null +++ b/TransPyV/temp/0c56128647a0b845.pyi @@ -0,0 +1,74 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesType.py +Module: lib.core.Handles.HandlesType +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import sys +import stdlib +import viperlib +import w32.fileio as fileio +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesStruct as HandlesStruct + +_g_current_file: t.CExtern | str +_g_generic_tp_names: t.CExtern | list[str] | t.CPtr +_g_generic_type_args: t.CExtern | list[str] | t.CPtr +_g_cdefine_names: t.CExtern | list[str] | t.CPtr +_g_cdefine_values: t.CExtern | list[str] | t.CPtr + +def register_cdefine_constant(pool: memhub.MemBuddy | t.CPtr, name: str, value: int) -> None: pass + +def lookup_cdefine_constant(name: str) -> int: pass + +def clear_cdefine_constants() -> None: pass + +def set_generic_context(tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> t.CInt: pass + +def clear_generic_context() -> t.CInt: pass + +def resolve_generic_type_name(name: str) -> str: pass + +def _get_type_name_from_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _parse_type_string(pool: memhub.MemBuddy | t.CPtr, type_str: str, ctx: t.CInt) -> ast.AST | t.CPtr: pass + +def set_current_file(file_path: str) -> t.CInt: pass + + +_SRC_BUF_SIZE: t.CDefine = 262144 + +def _read_source_line(file_path: str, lineno: int) -> str: pass + +def fatal_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def fatal_type_error(node: ast.AST | t.CPtr, msg: str) -> t.CInt: pass + +def map_t_type(pool: memhub.MemBuddy | t.CPtr, type_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_t_attr(node: ast.AST | t.CPtr, attr_name: str, imported_modules: str) -> int: pass + +def extract_class_name_from_annotation(node: ast.AST | t.CPtr, imported_modules: str) -> str: pass + +def resolve_annotation_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def has_decorator_marker(node: ast.AST | t.CPtr, marker_name: str) -> int: pass + +def is_pass_body(body: list[ast.AST | t.CPtr] | t.CPtr) -> int: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def infer_expr_type(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def infer_return_type(pool: memhub.MemBuddy | t.CPtr, body: list[ast.AST | t.CPtr] | t.CPtr, param_types: str) -> llvmlite.LLVMType | t.CPtr: pass + +def build_param_types_str(pool: memhub.MemBuddy | t.CPtr, args_node: ast.AST | t.CPtr) -> str: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/0df65b8ed15664b0.pyi b/TransPyV/temp/0df65b8ed15664b0.pyi new file mode 100644 index 0000000..e33dc69 --- /dev/null +++ b/TransPyV/temp/0df65b8ed15664b0.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from hashlib.__sha1.py +Module: hashlib.__sha1 +""" + + +import string +import t, c + +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self: sha1) -> t.CInt: pass + def transform(self: sha1, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha1, s: str) -> t.CInt: pass + def final(self: sha1, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/19f8024d10c828e8.pyi b/TransPyV/temp/19f8024d10c828e8.pyi new file mode 100644 index 0000000..a25a2e4 --- /dev/null +++ b/TransPyV/temp/19f8024d10c828e8.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from hashlib.__md5.py +Module: hashlib.__md5 +""" + + +import string +import t, c + +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: pass + + +md5_T: t.CExtern | t.CArray[t.CUInt32T, 64] +md5_S: t.CExtern | t.CArray[t.CInt, 64] + +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] + def __init__(self: md5) -> t.CInt: pass + def transform(self: md5, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: md5, s: str) -> t.CInt: pass + def final(self: md5, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/20cd49775c100a38.pyi b/TransPyV/temp/20cd49775c100a38.pyi new file mode 100644 index 0000000..df04de7 --- /dev/null +++ b/TransPyV/temp/20cd49775c100a38.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from json.__writer.py +Module: json.__writer +""" + + +import t, c +from stdint import * +import memhub +import string +from stdio import snprintf +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT + +def _write_char(buf: t.CChar | t.CPtr, pos: t.CSizeT, ch: t.CChar) -> t.CSizeT: pass + +def _write_str(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_string_escaped(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: pass + +def _write_value(buf: t.CChar | t.CPtr, pos: t.CSizeT, val: JsonValue | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CSizeT: pass + +def write(pool: memhub.MemManager | t.CPtr, val: JsonValue | t.CPtr, pretty: bool) -> t.CChar | t.CPtr: pass diff --git a/TransPyV/temp/21a7fcfc665f75ef.pyi b/TransPyV/temp/21a7fcfc665f75ef.pyi new file mode 100644 index 0000000..6f344a3 --- /dev/null +++ b/TransPyV/temp/21a7fcfc665f75ef.pyi @@ -0,0 +1,87 @@ +""" +Auto-generated Python stub file from llvmlite.__module.py +Module: llvmlite.__module +""" + + +import t, c +from stdint import * +import string +import stdio +import viperlib +import memhub +from linkedlist import GSList, GSListNode +from .__types import LLVMType, TypePrint, Array, PrintStructDefinition, get_struct_name, Struct +from .__function import Function, FunctionPrint, _ll_name_needs_quote, Param + +@t.NoVTable +class GlobalVariable(GSListNode[GlobalVariable]): + Name: t.CChar | t.CPtr + Ty: LLVMType | t.CPtr + Initializer: t.CVoid | t.CPtr + IsConstant: t.CInt + Linkage: t.CChar | t.CPtr + Section: t.CChar | t.CPtr + UnnamedAddr: t.CInt +@t.NoVTable +class NamedTypeNode(GSListNode[NamedTypeNode]): + Ty: LLVMType | t.CPtr +@t.NoVTable +class LLVMModule: + Name: t.CChar | t.CPtr + FuncHead: Function | t.CPtr + FuncTail: Function | t.CPtr + FuncCount: t.CSizeT + GlobalHead: GlobalVariable | t.CPtr + GlobalTail: GlobalVariable | t.CPtr + GlobalCount: t.CSizeT + TargetTriple: t.CChar | t.CPtr + DataLayout: t.CChar | t.CPtr + NamedTypeHead: NamedTypeNode | t.CPtr + NamedTypeTail: NamedTypeNode | t.CPtr + NamedTypeCount: t.CSizeT + +def new_module(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + + +GV_SIZE: t.CDefine = 64 +NAMED_TYPE_NODE_SIZE: t.CDefine = 16 + +def new_global_variable(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def module_add_function(mod: LLVMModule | t.CPtr, func: Function | t.CPtr) -> t.CInt: pass + +def module_add_global(mod: LLVMModule | t.CPtr, gv: GlobalVariable | t.CPtr) -> t.CInt: pass + +def module_set_target(mod: LLVMModule | t.CPtr, triple: t.CChar | t.CPtr) -> t.CInt: pass + +def module_set_datalayout(mod: LLVMModule | t.CPtr, layout: t.CChar | t.CPtr) -> t.CInt: pass + +def module_add_named_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def _is_type_defined_in_module(mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr) -> int: pass + +def _scan_type_for_cross_module_refs(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def module_ensure_opaque_for_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def global_set_constant(gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + +def global_set_linkage(gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + +def global_set_unnamed_addr(gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + +def global_set_section(gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + + +OUTPUT_FULL: t.CDefine = 0 # 完整 IR（stub + text） +OUTPUT_STUB: t.CDefine = 1 # 仅声明（header + 类型定义 + declare + external global） +OUTPUT_TEXT: t.CDefine = 2 # 仅代码（define + global 带初值） + +def LLVMModulePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mode: t.CInt) -> t.CInt: pass + +def _print_global(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _print_global_stub(buf: t.CChar | t.CPtr, size: t.CSizeT, gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/21fffba3315e81b4.doc.json b/TransPyV/temp/21fffba3315e81b4.doc.json new file mode 100644 index 0000000..f6553a0 --- /dev/null +++ b/TransPyV/temp/21fffba3315e81b4.doc.json @@ -0,0 +1 @@ +{"Logger.banner": "输出分节标题（青色高亮）。", "Logger.compile_error": "格式化编译错误输出（红底白字标题 + 位置 + 错误信息）。", "Logger.compile_warning": "格式化编译警告输出（黄底黑字标题 + 位置 + 警告信息）。", "get_logger": "获取全局 logger 实例。若不存在则通过 _mbuddy 分配并初始化。", "set_logger": "设置全局 logger 实例。"} \ No newline at end of file diff --git a/TransPyV/temp/21fffba3315e81b4.pyi b/TransPyV/temp/21fffba3315e81b4.pyi new file mode 100644 index 0000000..04be3c9 --- /dev/null +++ b/TransPyV/temp/21fffba3315e81b4.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.VLogger.py +Module: lib.core.VLogger +""" + + +import t, c +from stdint import * +import stdio +import memhub +import w32.win32console as w32cmd +import w32.win32file as w32file +import w32.win32base as w32base + +class LogLevel(t.CEnum): + DEBUG = 0 + INFO = 1 + WARNING = 2 + ERROR = 3 + SUCCESS = 4 + +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +FOREGROUND_WHITE: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE +FOREGROUND_CYAN: t.CDefine = FOREGROUND_BLUE | FOREGROUND_GREEN +FOREGROUND_YELLOW: t.CDefine = FOREGROUND_RED | FOREGROUND_GREEN +FOREGROUND_MAGENTA: t.CDefine = FOREGROUND_RED | FOREGROUND_BLUE +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +BACKGROUND_WHITE: t.CDefine = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE + +class Logger: + _level: int + _console_handle: w32base.HANDLE + _use_color: int + __mbuddy__: memhub.MemManager | t.CPtr + def __init__(self: Logger, level: int) -> t.CInt: pass + def _set_color(self: Logger, attr: WORD) -> int: pass + def _reset_color(self: Logger) -> int: pass + def _log(self: Logger, level: int, prefix: str, msg: str, color: WORD, category: str) -> int: pass + def debug(self: Logger, msg: str, category: str) -> int: pass + def info(self: Logger, msg: str, category: str) -> int: pass + def warning(self: Logger, msg: str, category: str) -> int: pass + def error(self: Logger, msg: str, category: str) -> int: pass + def success(self: Logger, msg: str, category: str) -> int: pass + def set_level(self: Logger, level: int) -> int: pass + def banner(self: Logger, msg: str) -> int: pass + def compile_error(self: Logger, msg: str, file: str, line: int) -> int: pass + def compile_warning(self: Logger, msg: str, file: str, line: int) -> int: pass + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +_g_logger: t.CExtern | Logger | t.CPtr + +def get_logger() -> Logger | t.CPtr: pass + +def set_logger(logger: Logger | t.CPtr) -> int: pass diff --git a/TransPyV/temp/240a9a4157959a9f.pyi b/TransPyV/temp/240a9a4157959a9f.pyi new file mode 100644 index 0000000..c230800 --- /dev/null +++ b/TransPyV/temp/240a9a4157959a9f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from json.__parser.py +Module: json.__parser +""" + + +import t, c +from stdint import * +import memhub +import string +from . import JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, JSON_STRING, JSON_ARRAY, JSON_OBJECT, null, bool_val, int_val, float_val, string_val, array, object + +class ParseState: + src: t.CChar | t.CPtr + pos: t.CSizeT + len: t.CSizeT + pool: memhub.MemManager | t.CPtr + def __init__(self: ParseState, src: t.CChar | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _peek(s: ParseState | t.CPtr) -> t.CChar: pass + +def _advance(s: ParseState | t.CPtr) -> t.CChar: pass + +def _skip_whitespace(s: ParseState | t.CPtr) -> t.CInt: pass + +def _expect(s: ParseState | t.CPtr, ch: t.CChar) -> bool: pass + +def _parse_string(s: ParseState | t.CPtr) -> t.CChar | t.CPtr: pass + +def _parse_number(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_value(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_array(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def _parse_object(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: pass + +def parse(pool: memhub.MemManager | t.CPtr, src: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass diff --git a/TransPyV/temp/271ea3decb810db2.pyi b/TransPyV/temp/271ea3decb810db2.pyi new file mode 100644 index 0000000..3d637bf --- /dev/null +++ b/TransPyV/temp/271ea3decb810db2.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from atom.py +Module: atom +""" + + +import t, c + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: pass + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: pass + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: pass + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: pass diff --git a/TransPyV/temp/2998b8f78d7022bf.doc.json b/TransPyV/temp/2998b8f78d7022bf.doc.json new file mode 100644 index 0000000..d24160b --- /dev/null +++ b/TransPyV/temp/2998b8f78d7022bf.doc.json @@ -0,0 +1 @@ +{"Translator": "翻译器：管理编译状态，委托 Handles 模块执行翻译", "Translator._init_state": "初始化变量表、函数表和子 Handle", "Translator.translate": "将 AST 翻译为 LLVM IR", "Translator._translate_wrapper_main": "委托 HandlesMain.create_wrapper_main() 创建包装 main（trans 单参）", "Translator._translate_module_level": "委托 HandlesMain.translate_children() 翻译模块级语句（trans 单参）", "Translator.dump_ir": "将 LLVM IR 输出到缓冲区\n\n mode: 0=完整, 1=stub(仅声明), 2=text(仅代码)\n ", "add_global_name": "添加一个 global 变量名", "is_global_name": "检查 name 是否在当前函数的 global 集合中", "add_nonlocal_name": "添加一个 nonlocal 变量名", "is_nonlocal_name": "检查 name 是否在当前函数的 nonlocal 集合中", "get_nonlocal_index": "获取 nonlocal 变量在 env 中的索引（0-based），找不到返回 -1", "clear_scope_names": "清空当前函数的 global/nonlocal 名称集合（进入新函数时调用）"} \ No newline at end of file diff --git a/TransPyV/temp/2998b8f78d7022bf.pyi b/TransPyV/temp/2998b8f78d7022bf.pyi new file mode 100644 index 0000000..865ebf6 --- /dev/null +++ b/TransPyV/temp/2998b8f78d7022bf.pyi @@ -0,0 +1,90 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesTranslator.py +Module: lib.core.Handles.HandlesTranslator +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesMain as HandlesMain +import lib.core.Handles.HandlesAssign as HandlesAssign +import lib.core.Handles.HandlesReturn as HandlesReturn +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesAnnAssign as HandlesAnnAssign +import lib.core.Handles.HandlesAugAssign as HandlesAugAssign +import lib.core.Handles.HandlesIf as HandlesIf +import lib.core.Handles.HandlesWhile as HandlesWhile +import lib.core.Handles.HandlesFor as HandlesFor +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.Projectrans.Config as Config + +MAX_VARS: t.CDefine = 256 +MAX_FUNCS: t.CDefine = 256 +MAX_GLOBAL_NAMES: t.CDefine = 32 + +@t.NoVTable +class Translator: + Module: llvmlite.LLVMModule | t.CPtr + Pool: memhub.MemBuddy | t.CPtr + _funcs: HandlesExprCall.FuncEntry | t.CPtr + _func_count: t.CInt + _imported_modules: str + _from_imports: str + _cur_func: llvmlite.Function | t.CPtr + _cur_builder: llvmlite.IRBuilder | t.CPtr + _break_bb: llvmlite.BasicBlock | t.CPtr + _continue_bb: llvmlite.BasicBlock | t.CPtr + _label_counter: t.CInt + _global_names: str + _global_name_count: t.CInt + _nonlocal_names: str + _nonlocal_name_count: t.CInt + _cur_func_name: str + ModuleSha1: str + CurrentPackage: str + _closure_env_offsets: t.CInt + _closure_env_count: t.CInt + GenericTypeParamNames: list[str] | t.CPtr + GenericTypeArgs: list[str] | t.CPtr + _declare_only: t.CInt + AssignH: HandlesAssign.AssignHandle | t.CPtr + ReturnH: HandlesReturn.ReturnHandle | t.CPtr + ImportsH: HandlesImports.ImportsHandle | t.CPtr + AnnAssignH: HandlesAnnAssign.AnnAssignHandle | t.CPtr + AugAssignH: HandlesAugAssign.AugAssignHandle | t.CPtr + IfH: HandlesIf.IfHandle | t.CPtr + WhileH: HandlesWhile.WhileHandle | t.CPtr + ForH: HandlesFor.ForHandle | t.CPtr + ExprH: HandlesExpr.ExprHandle | t.CPtr + ExprCallH: HandlesExprCall.ExprCallHandle | t.CPtr + SymTab: HandlesVar.SymbolTable | t.CPtr + def __init__(self: Translator) -> t.CInt: pass + def _init_state(self: Translator, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def translate(self: Translator, tree: ast.AST | t.CPtr) -> int: pass + def _translate_wrapper_main(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr, i32_ty: llvmlite.LLVMType | t.CPtr) -> t.CInt: pass + def _translate_module_level(self: Translator, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, tree: ast.AST | t.CPtr) -> t.CInt: pass + def dump_ir(self: Translator, buf: bytes, size: t.CSizeT, mode: int) -> t.CInt: pass + +def add_global_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_global_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def add_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> t.CInt: pass + +def is_nonlocal_name(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def get_nonlocal_index(trans: HT.Translator | t.CPtr, name: str) -> int: pass + +def clear_scope_names(trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr \ No newline at end of file diff --git a/TransPyV/temp/2d39c6c7d3557b3e.pyi b/TransPyV/temp/2d39c6c7d3557b3e.pyi new file mode 100644 index 0000000..fc259bd --- /dev/null +++ b/TransPyV/temp/2d39c6c7d3557b3e.pyi @@ -0,0 +1,63 @@ +""" +Auto-generated Python stub file from linkedlist.py +Module: linkedlist +""" + + +import t, c +from stdint import * + +@t.NoVTable +class LinkedNode: + next: 'LinkedNode' | t.CPtr + prev: 'LinkedNode' | t.CPtr + child: 'LinkedNode' | t.CPtr + last_child: 'LinkedNode' | t.CPtr + parent: 'LinkedNode' | t.CPtr + child_count: t.CSizeT + def append(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def prepend(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_before(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def insert_after(self: LinkedNode, node: 'LinkedNode' | t.CPtr, new_node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def remove_child(self: LinkedNode, node: 'LinkedNode' | t.CPtr) -> t.CInt: pass + def detach(self: LinkedNode) -> t.CInt: pass + def unlink(self: LinkedNode) -> t.CInt: pass + def has_children(self: LinkedNode) -> t.CInt: pass + def child_at(self: LinkedNode, index: t.CSizeT) -> 'LinkedNode' | t.CPtr: pass + def count_children(self: LinkedNode) -> t.CSizeT: pass + def first_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass + def last_sibling(self: LinkedNode) -> 'LinkedNode' | t.CPtr: pass +@t.NoVTable +class SListNode: + Next: 'SListNode' | t.CPtr + def append(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def append_after(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass + def count(self: SListNode) -> t.CSizeT: pass + def at(self: SListNode, index: t.CSizeT) -> 'SListNode' | t.CPtr: pass + def remove(self: SListNode, node: 'SListNode' | t.CPtr) -> 'SListNode' | t.CPtr: pass +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def __init__(self: GSList) -> t.CInt: pass + def append(self: GSList, node: T) -> t.CInt: pass + def count(self: GSList) -> t.CSizeT: pass + def at(self: GSList, index: t.CSizeT) -> T | t.CPtr: pass +@t.NoVTable +class GTreeNode[T]: + Next: T | t.CPtr + Prev: T | t.CPtr + Child: T | t.CPtr + LastChild: T | t.CPtr + Parent: T | t.CPtr + Count: t.CSizeT + def append(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def prepend(self: GTreeNode, node: T | t.CPtr) -> t.CInt: pass + def detach(self: GTreeNode) -> t.CInt: pass + def has_children(self: GTreeNode) -> t.CInt: pass + def child_at(self: GTreeNode, index: t.CSizeT) -> T | t.CPtr: pass + def count_children(self: GTreeNode) -> t.CSizeT: pass \ No newline at end of file diff --git a/TransPyV/temp/2da636c61863c815.pyi b/TransPyV/temp/2da636c61863c815.pyi new file mode 100644 index 0000000..7c91519 --- /dev/null +++ b/TransPyV/temp/2da636c61863c815.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from subprocess.py +Module: subprocess +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import w32.win32base +import w32.win32file +import w32.win32process +import w32.win32sync +import posix + +OUTPUT_BUF_SIZE: t.CDefine = 65536 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class CompletedProcess: + args: str + returncode: int + stdout: str + stderr: str + def __new__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CPtr: pass + def __init__(self: CompletedProcess, args: str, returncode: int, stdout: str, stderr: str) -> t.CInt: pass + +def run(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_win32(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass + +def _run_posix(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: pass diff --git a/TransPyV/temp/2ed380190e9df89e.doc.json b/TransPyV/temp/2ed380190e9df89e.doc.json new file mode 100644 index 0000000..2259eff --- /dev/null +++ b/TransPyV/temp/2ed380190e9df89e.doc.json @@ -0,0 +1 @@ +{"is_cdefine_annotation": "检测注解是否为 t.CDefine，返回 1 表示是，0 表示否", "extract_cdefine_int_value": "从值节点提取整数常量（仅支持 Constant INT）", "AnnAssignHandle": "AnnAssign 语句处理器：继承 Mixin 获得 Trans 回指针", "AnnAssignHandle.PreScan": "预扫描 AnnAssign，提前创建 alloca 到 entry block 顶部", "AnnAssignHandle.Handle": "翻译 AnnAssign(target=Name, annotation=..., value=expr)"} \ No newline at end of file diff --git a/TransPyV/temp/2ed380190e9df89e.pyi b/TransPyV/temp/2ed380190e9df89e.pyi new file mode 100644 index 0000000..6d50c9b --- /dev/null +++ b/TransPyV/temp/2ed380190e9df89e.pyi @@ -0,0 +1,33 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAnnAssign.py +Module: lib.core.Handles.HandlesAnnAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesExprCall as HandlesExprCall + +def is_cdefine_annotation(annot: ast.AST | t.CPtr) -> int: pass + +def extract_cdefine_int_value(val_node: ast.AST | t.CPtr) -> int: pass + + +@t.NoVTable +class AnnAssignHandle(HandlesBase.Mixin): + def __init__(self: AnnAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def PreScan(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + def Handle(self: AnnAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAnnAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AnnAssignHandle | t.CPtr: pass diff --git a/TransPyV/temp/34548789d646f432.pyi b/TransPyV/temp/34548789d646f432.pyi new file mode 100644 index 0000000..e046b0c --- /dev/null +++ b/TransPyV/temp/34548789d646f432.pyi @@ -0,0 +1,26 @@ +""" +Auto-generated Python stub file from ast.__init__.py +Module: ast.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp, _init_tables +from .base import AST, ASTKind, ASTCtx, OpKind, ASTFlag, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC, FLAG_SIMPLE, FLAG_HAS_STAR, _init_ast, _copy_str, _set_pos, _inherit_pos, _set_parent_list, _binop_from_op, _augop_from_op, _cmpop_from_op, _emit, _emit_str, _emit_int, _dump_list, dump +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .astaux import ExceptHandler, Arguments, Arg, Keyword, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .visitor import ASTVisitor +from .lexer import Lexer, _lexer_init, tokenize, new_lexer +from .parser import Parser, _parser_init, parse_tokens, new_parser +from .parser import _parse_test + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def parse(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass + +def parse_expression(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: pass diff --git a/TransPyV/temp/3d3edda4a4e6c08b.doc.json b/TransPyV/temp/3d3edda4a4e6c08b.doc.json new file mode 100644 index 0000000..8c2a0a2 --- /dev/null +++ b/TransPyV/temp/3d3edda4a4e6c08b.doc.json @@ -0,0 +1 @@ +{"AssignHandle": "Assign 语句处理器：继承 Mixin 获得 Trans 回指针 + 共享方法"} \ No newline at end of file diff --git a/TransPyV/temp/3d3edda4a4e6c08b.pyi b/TransPyV/temp/3d3edda4a4e6c08b.pyi new file mode 100644 index 0000000..ec65969 --- /dev/null +++ b/TransPyV/temp/3d3edda4a4e6c08b.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAssign.py +Module: lib.core.Handles.HandlesAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct + +@t.NoVTable +class AssignHandle(HandlesBase.Mixin): + _CurrentClass: str + def __init__(self: AssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AssignHandle | t.CPtr: pass diff --git a/TransPyV/temp/4020cac9bab5e9f1.doc.json b/TransPyV/temp/4020cac9bab5e9f1.doc.json new file mode 100644 index 0000000..006cdc0 --- /dev/null +++ b/TransPyV/temp/4020cac9bab5e9f1.doc.json @@ -0,0 +1 @@ +{"extract_func_attrs": "从 decorator_list 提取 c.Attribute 属性，返回 LLVM IR 属性字符串", "_get_attr_name_from_node": "从 AST 节点提取属性名\n\n 支持:\n - Call(func=Attribute(...)): t.attr.always_inline() -> \"always_inline\"\n - Attribute: t.attr.packed -> \"packed\"\n - Attribute(t.attr.llvm.xxx): t.attr.llvm.nounwind -> \"nounwind\"\n ", "_append_llvm_attr": "将属性名映射到 LLVM 属性名并追加到缓冲区\n\n 返回 1 表示已追加，0 表示不支持该属性\n ", "_append_str": "将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）", "_mangle_name": "为任意名称添加 SHA1 前缀（不做 main/export 检查）", "_mangle_name_with_sha1": "用指定 SHA1 为名称添加前缀（不做 main/export 检查）", "_mangle_func_name": "为函数名添加 SHA1 前缀（检查 main/export）", "forward_declare_functions": "预扫描所有顶层 FunctionDef，创建前向声明", "translate_function_def": "翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）", "translate_nested_function_def": "翻译嵌套函数定义：提升为顶层函数 + 创建闭包", "create_function": "创建 LLVM 函数并添加参数"} \ No newline at end of file diff --git a/TransPyV/temp/4020cac9bab5e9f1.pyi b/TransPyV/temp/4020cac9bab5e9f1.pyi new file mode 100644 index 0000000..2b7f43b --- /dev/null +++ b/TransPyV/temp/4020cac9bab5e9f1.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFunctions.py +Module: lib.core.Handles.HandlesFunctions +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import viperlib +import stdio +import stdlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesImports as HandlesImports + +def extract_func_attrs(pool: memhub.MemBuddy | t.CPtr, decorator_list: list[ast.AST | t.CPtr] | t.CPtr, imported_modules: str) -> t.CChar | t.CPtr: pass + +def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str: pass + +def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int: pass + +def _append_str(buf: t.CChar | t.CPtr, s: str) -> None: pass + +def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str: pass + +def _mangle_name_with_sha1(sha1: str, name: str) -> str: pass + +def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str, has_export: int) -> str: pass + +def forward_declare_functions(trans: HT.Translator | t.CPtr, tree: ast.AST | t.CPtr) -> int: pass + +def translate_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nested_function_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, name: str, args_node: ast.AST | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Function | t.CPtr: pass diff --git a/TransPyV/temp/4337fb260448bbe2.pyi b/TransPyV/temp/4337fb260448bbe2.pyi new file mode 100644 index 0000000..7ed88bd --- /dev/null +++ b/TransPyV/temp/4337fb260448bbe2.pyi @@ -0,0 +1,73 @@ +""" +Auto-generated Python stub file from ast.astaux.py +Module: ast.astaux +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class ExceptHandler(AST): + type: AST | t.CPtr + name: str + def __new__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: ExceptHandler, pool: memhub.MemManager | t.CPtr, type: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: ExceptHandler) -> t.CInt: pass + def type_name(self: ExceptHandler) -> str: pass + def dump(self: ExceptHandler, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arguments(AST): + args: list[AST | t.CPtr] | t.CPtr + vararg: AST | t.CPtr + kwarg: AST | t.CPtr + defaults: list[AST | t.CPtr] | t.CPtr + kw_defaults: list[AST | t.CPtr] | t.CPtr + def __new__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Arguments, pool: memhub.MemManager | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, vararg: AST | t.CPtr, kwarg: AST | t.CPtr, defaults: list[AST | t.CPtr] | t.CPtr, kw_defaults: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Arguments) -> t.CInt: pass + def type_name(self: Arguments) -> str: pass + def dump(self: Arguments, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Arg(AST): + arg: str + annotation: AST | t.CPtr + def __new__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Arg, pool: memhub.MemManager | t.CPtr, arg: str, annotation: AST | t.CPtr) -> t.CInt: pass + def kind(self: Arg) -> t.CInt: pass + def type_name(self: Arg) -> str: pass + def dump(self: Arg, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Keyword(AST): + arg: str + value: AST | t.CPtr + def __new__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Keyword, pool: memhub.MemManager | t.CPtr, arg: str, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Keyword) -> t.CInt: pass + def type_name(self: Keyword) -> str: pass + def dump(self: Keyword, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Alias(AST): + name: str + asname: str + def __new__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def __init__(self: Alias, pool: memhub.MemManager | t.CPtr, name: str, asname: str) -> t.CInt: pass + def kind(self: Alias) -> t.CInt: pass + def type_name(self: Alias) -> str: pass + def dump(self: Alias, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class WithItem(AST): + context_expr: AST | t.CPtr + optional_vars: AST | t.CPtr + def __new__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def __init__(self: WithItem, pool: memhub.MemManager | t.CPtr, context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr) -> t.CInt: pass + def kind(self: WithItem) -> t.CInt: pass + def type_name(self: WithItem) -> str: pass + def dump(self: WithItem, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Comprehension(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + ifs: list[AST | t.CPtr] | t.CPtr + is_async: t.CInt + def __new__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: Comprehension, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: Comprehension) -> t.CInt: pass + def type_name(self: Comprehension) -> str: pass + def dump(self: Comprehension, buf: bytes, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/TransPyV/temp/455ff4e74c3dbfb0.pyi b/TransPyV/temp/455ff4e74c3dbfb0.pyi new file mode 100644 index 0000000..984146e --- /dev/null +++ b/TransPyV/temp/455ff4e74c3dbfb0.pyi @@ -0,0 +1,120 @@ +""" +Auto-generated Python stub file from llvmlite.__types.py +Module: llvmlite.__types +""" + + +import t, c +from stdint import * +import viperlib +import string +import memhub +from linkedlist import GSListNode + +class ParamNode(GSListNode[ParamNode]): + Ty: t.CVoid | t.CPtr +class LLVMType(t.REnum): + class Int: + Bits: t.CInt + class Ptr: + Pointee: LLVMType | t.CPtr + class Func: + Ret: LLVMType | t.CPtr + Params: ParamNode | t.CPtr + PCount: t.CInt + class Array: + ElemTy: LLVMType | t.CPtr + Count: t.CInt + class Struct: + Fields: ParamNode | t.CPtr + FCount: t.CInt + Name: t.CChar | t.CPtr + class Float: + Bits: t.CInt + class Void: + pass + class Label: + pass + class Metadata: + pass + +def _new_type(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int1(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int8(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int16(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int32(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Int64(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Ptr(pool: memhub.MemBuddy | t.CPtr, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + +def Func(pool: memhub.MemBuddy | t.CPtr, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Void(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Array(pool: memhub.MemBuddy | t.CPtr, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + +def Struct(pool: memhub.MemBuddy | t.CPtr, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + +def Half(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Float(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Double(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def FP128(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + +def Label(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: pass + + +PNODE_SIZE: t.CDefine = 16 + +def new_param_node(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + +def param_list_append(head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_get_ty(node: ParamNode | t.CPtr) -> LLVMType | t.CPtr: pass + +def paramnode_get_next(node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + +def paramnode_set_ty(node: ParamNode | t.CPtr, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def paramnode_set_next(node: ParamNode | t.CPtr, nxt: ParamNode | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class LLVMTypeCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMTypeCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def Int1(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int8(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int16(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int32(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Int64(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Ptr(self: LLVMTypeCore, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: pass + def Func(self: LLVMTypeCore, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Array(self: LLVMTypeCore, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: pass + def Struct(self: LLVMTypeCore, fields: ParamNode | t.CPtr, fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: pass + def Void(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Half(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Float(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Double(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def FP128(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def Label(self: LLVMTypeCore) -> LLVMType | t.CPtr: pass + def NewParamNode(self: LLVMTypeCore, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: pass + def ParamListAppend(self: LLVMTypeCore, head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: pass + def PrintType(self: LLVMTypeCore, buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr) -> t.CInt: pass + +def TypePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _print_struct_body(buf: t.CChar | t.CPtr, size: t.CSizeT, fields: ParamNode | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def PrintStructDefinition(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def get_struct_name(ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass diff --git a/TransPyV/temp/46ea362b72c10a9e.pyi b/TransPyV/temp/46ea362b72c10a9e.pyi new file mode 100644 index 0000000..fbd9c3b --- /dev/null +++ b/TransPyV/temp/46ea362b72c10a9e.pyi @@ -0,0 +1,219 @@ +""" +Auto-generated Python stub file from llvmlite.__builder.py +Module: llvmlite.__builder +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__types import LLVMType, TypePrint +from .__values import Value, new_value, SSAValue +from .__function import Function, BasicBlock, block_append_text, function_move_block_to_end, _ll_name_needs_quote + +class IRBuilder: + Pool: memhub.MemBuddy | t.CPtr + Func: Function | t.CPtr + CurBlock: BasicBlock | t.CPtr + Counter: t.CInt + +BUILDER_SIZE: t.CDefine = 32 + +def new_builder(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + +def position_at_end(builder: IRBuilder | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def _alloc_ssa_name(builder: IRBuilder | t.CPtr) -> t.CChar | t.CPtr: pass + +def _type_str(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: pass + +def _emit(builder: IRBuilder | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def _mark_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def builder_cur_block_is_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_alloca(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_load(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, ptr: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_store(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, ptr: Value | t.CPtr) -> t.CInt: pass + +def build_add(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_mul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_sdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_udiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_srem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_urem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_and(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_or(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_xor(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_shl(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_lshr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_ashr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fadd(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fsub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fmul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_frem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fneg(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> Value | t.CPtr: pass + +def _build_binop(builder: IRBuilder | t.CPtr, op: t.CChar | t.CPtr, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + + +ICMP_EQ: t.CDefine = 0 +ICMP_NE: t.CDefine = 1 +ICMP_SGT: t.CDefine = 2 +ICMP_SGE: t.CDefine = 3 +ICMP_SLT: t.CDefine = 4 +ICMP_SLE: t.CDefine = 5 +ICMP_UGT: t.CDefine = 6 +ICMP_UGE: t.CDefine = 7 +ICMP_ULT: t.CDefine = 8 +ICMP_ULE: t.CDefine = 9 +FCMP_FALSE: t.CDefine = 0 +FCMP_OEQ: t.CDefine = 1 +FCMP_OGT: t.CDefine = 2 +FCMP_OGE: t.CDefine = 3 +FCMP_OLT: t.CDefine = 4 +FCMP_OLE: t.CDefine = 5 +FCMP_ONE: t.CDefine = 6 +FCMP_ORD: t.CDefine = 7 +FCMP_UNO: t.CDefine = 8 +FCMP_UEQ: t.CDefine = 9 +FCMP_UGT: t.CDefine = 10 +FCMP_UGE: t.CDefine = 11 +FCMP_ULT: t.CDefine = 12 +FCMP_ULE: t.CDefine = 13 +FCMP_UNE: t.CDefine = 14 +FCMP_TRUE: t.CDefine = 15 + +def _icmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def _fcmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: pass + +def build_icmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_fcmp(builder: IRBuilder | t.CPtr, pred: t.CInt, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_br(builder: IRBuilder | t.CPtr, target: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_cond_br(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, then_blk: BasicBlock | t.CPtr, else_blk: BasicBlock | t.CPtr) -> t.CInt: pass + +def build_ret(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> t.CInt: pass + +def build_ret_void(builder: IRBuilder | t.CPtr) -> t.CInt: pass + +def build_call(builder: IRBuilder | t.CPtr, callee: t.CChar | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr, is_variadic: t.CInt) -> Value | t.CPtr: pass + +def build_call_indirect(builder: IRBuilder | t.CPtr, callee_val: Value | t.CPtr, args: Value | t.CPtr, arg_count: t.CInt, ret_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_bitcast(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_sext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_trunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_zext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ptrtoint(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_inttoptr(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fpext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fptrunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2si(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_si2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_fp2ui(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_ui2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: pass + +def build_gep(builder: IRBuilder | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_array(builder: IRBuilder | t.CPtr, array_ty: LLVMType | t.CPtr, elem_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_gep_struct(builder: IRBuilder | t.CPtr, struct_ty: LLVMType | t.CPtr, field_ty: LLVMType | t.CPtr, ptr: Value | t.CPtr, field_idx: t.CInt) -> Value | t.CPtr: pass + + +class PhiIncoming(GSListNode[PhiIncoming]): + Val: Value | t.CPtr + Block: BasicBlock | t.CPtr + +PHIINC_SIZE: t.CDefine = 24 + +def new_phi_incoming(pool: memhub.MemBuddy | t.CPtr, val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> PhiIncoming | t.CPtr: pass + +def build_phi(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, incoming_head: PhiIncoming | t.CPtr, count: t.CInt) -> Value | t.CPtr: pass + + +class SwitchCase(GSListNode[SwitchCase]): + CaseVal: Value | t.CPtr + Block: BasicBlock | t.CPtr + +SWCASE_SIZE: t.CDefine = 24 + +def new_switch_case(pool: memhub.MemBuddy | t.CPtr, case_val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> SwitchCase | t.CPtr: pass + +def build_switch(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, default_blk: BasicBlock | t.CPtr, cases_head: SwitchCase | t.CPtr, case_count: t.CInt) -> t.CInt: pass + +def build_select(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, true_val: Value | t.CPtr, false_val: Value | t.CPtr) -> Value | t.CPtr: pass + +def build_unreachable(builder: IRBuilder | t.CPtr) -> t.CInt: pass + + +ATOMIC_XCHG: t.CDefine = 0 +ATOMIC_ADD: t.CDefine = 1 +ATOMIC_SUB: t.CDefine = 2 +ATOMIC_AND: t.CDefine = 3 +ATOMIC_NAND: t.CDefine = 4 +ATOMIC_OR: t.CDefine = 5 +ATOMIC_XOR: t.CDefine = 6 +ATOMIC_MAX: t.CDefine = 7 +ATOMIC_MIN: t.CDefine = 8 +ATOMIC_UMAX: t.CDefine = 9 +ATOMIC_UMIN: t.CDefine = 10 +ATOMIC_ORDER_NOTATOMIC: t.CDefine = 0 +ATOMIC_ORDER_UNORDERED: t.CDefine = 1 +ATOMIC_ORDER_MONOTONIC: t.CDefine = 2 +ATOMIC_ORDER_ACQUIRE: t.CDefine = 3 +ATOMIC_ORDER_RELEASE: t.CDefine = 4 +ATOMIC_ORDER_ACQ_REL: t.CDefine = 5 +ATOMIC_ORDER_SEQ_CST: t.CDefine = 6 + +def _atomic_op_str(op: t.CInt) -> t.CChar | t.CPtr: pass + +def _atomic_order_str(ord: t.CInt) -> t.CChar | t.CPtr: pass + +def build_atomicrmw(builder: IRBuilder | t.CPtr, op: t.CInt, ptr: Value | t.CPtr, val: Value | t.CPtr, order: t.CInt) -> Value | t.CPtr: pass + +def build_cmpxchg(builder: IRBuilder | t.CPtr, ptr: Value | t.CPtr, cmp_val: Value | t.CPtr, new_val: Value | t.CPtr, succ_order: t.CInt, fail_order: t.CInt) -> Value | t.CPtr: pass + +def build_inline_asm(builder: IRBuilder | t.CPtr, ret_ty: LLVMType | t.CPtr, asm_text: t.CChar | t.CPtr, constraints: t.CChar | t.CPtr, has_side_effects: t.CInt) -> Value | t.CPtr: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/47767b5026a8ee15.pyi b/TransPyV/temp/47767b5026a8ee15.pyi new file mode 100644 index 0000000..7890f3d --- /dev/null +++ b/TransPyV/temp/47767b5026a8ee15.pyi @@ -0,0 +1,241 @@ +""" +Auto-generated Python stub file from ast.exprs.py +Module: ast.exprs +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, _dump_op_list, _op_name, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE + +class Name(AST): + id: str + ctx: t.CInt + def __new__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Name, pool: memhub.MemManager | t.CPtr, id: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Name) -> t.CInt: pass + def type_name(self: Name) -> str: pass + def dump(self: Name, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Constant(AST): + int_val: t.CInt64T + float_val: t.CDouble + str_val: str + const_kind: t.CInt + def __new__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def __init__(self: Constant, pool: memhub.MemManager | t.CPtr, const_kind: t.CInt, int_val: t.CInt64T, float_val: t.CDouble, str_val: str, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + def kind(self: Constant) -> t.CInt: pass + def type_name(self: Constant) -> str: pass + def dump(self: Constant, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BinOp(AST): + left: AST | t.CPtr + op: t.CInt + right: AST | t.CPtr + def __new__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def __init__(self: BinOp, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr) -> t.CInt: pass + def kind(self: BinOp) -> t.CInt: pass + def type_name(self: BinOp) -> str: pass + def dump(self: BinOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class UnaryOp(AST): + op: t.CInt + operand: AST | t.CPtr + def __new__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def __init__(self: UnaryOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, operand: AST | t.CPtr) -> t.CInt: pass + def kind(self: UnaryOp) -> t.CInt: pass + def type_name(self: UnaryOp) -> str: pass + def dump(self: UnaryOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Compare(AST): + left: AST | t.CPtr + ops: t.CPtr + comparators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Compare, pool: memhub.MemManager | t.CPtr, left: AST | t.CPtr, ops: t.CPtr, comparators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Compare) -> t.CInt: pass + def type_name(self: Compare) -> str: pass + def dump(self: Compare, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Call(AST): + func: AST | t.CPtr + args: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + def __new__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Call, pool: memhub.MemManager | t.CPtr, func: AST | t.CPtr, args: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Call) -> t.CInt: pass + def type_name(self: Call) -> str: pass + def dump(self: Call, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class BoolOp(AST): + op: t.CInt + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: BoolOp, pool: memhub.MemManager | t.CPtr, op: t.CInt, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: BoolOp) -> t.CInt: pass + def type_name(self: BoolOp) -> str: pass + def dump(self: BoolOp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Lambda(AST): + args: AST | t.CPtr + body: AST | t.CPtr + def __new__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Lambda, pool: memhub.MemManager | t.CPtr, args: AST | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Lambda) -> t.CInt: pass + def type_name(self: Lambda) -> str: pass + def dump(self: Lambda, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class IfExp(AST): + test: AST | t.CPtr + body: AST | t.CPtr + orelse: AST | t.CPtr + def __new__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def __init__(self: IfExp, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, body: AST | t.CPtr, orelse: AST | t.CPtr) -> t.CInt: pass + def kind(self: IfExp) -> t.CInt: pass + def type_name(self: IfExp) -> str: pass + def dump(self: IfExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Dict(AST): + keys: list[AST | t.CPtr] | t.CPtr + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Dict, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Dict) -> t.CInt: pass + def type_name(self: Dict) -> str: pass + def dump(self: Dict, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Set(AST): + elts: list[AST | t.CPtr] | t.CPtr + def __new__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Set, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Set) -> t.CInt: pass + def type_name(self: Set) -> str: pass + def dump(self: Set, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ListComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: ListComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: ListComp) -> t.CInt: pass + def type_name(self: ListComp) -> str: pass + def dump(self: ListComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class SetComp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: SetComp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: SetComp) -> t.CInt: pass + def type_name(self: SetComp) -> str: pass + def dump(self: SetComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class DictComp(AST): + key: AST | t.CPtr + value: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: DictComp, pool: memhub.MemManager | t.CPtr, key: AST | t.CPtr, value: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: DictComp) -> t.CInt: pass + def type_name(self: DictComp) -> str: pass + def dump(self: DictComp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class GeneratorExp(AST): + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + def __new__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: GeneratorExp, pool: memhub.MemManager | t.CPtr, elt: AST | t.CPtr, generators: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: GeneratorExp) -> t.CInt: pass + def type_name(self: GeneratorExp) -> str: pass + def dump(self: GeneratorExp, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Await(AST): + value: AST | t.CPtr + def __new__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Await, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Await) -> t.CInt: pass + def type_name(self: Await) -> str: pass + def dump(self: Await, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Yield(AST): + value: AST | t.CPtr + def __new__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Yield, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Yield) -> t.CInt: pass + def type_name(self: Yield) -> str: pass + def dump(self: Yield, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class YieldFrom(AST): + value: AST | t.CPtr + def __new__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: YieldFrom, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: YieldFrom) -> t.CInt: pass + def type_name(self: YieldFrom) -> str: pass + def dump(self: YieldFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FormattedValue(AST): + value: AST | t.CPtr + conversion: t.CInt + format_spec: AST | t.CPtr + def __new__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FormattedValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, conversion: t.CInt, format_spec: AST | t.CPtr) -> t.CInt: pass + def kind(self: FormattedValue) -> t.CInt: pass + def type_name(self: FormattedValue) -> str: pass + def dump(self: FormattedValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class JoinedStr(AST): + values: list[AST | t.CPtr] | t.CPtr + def __new__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: JoinedStr, pool: memhub.MemManager | t.CPtr, values: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: JoinedStr) -> t.CInt: pass + def type_name(self: JoinedStr) -> str: pass + def dump(self: JoinedStr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Attribute(AST): + value: AST | t.CPtr + attr: str + ctx: t.CInt + def __new__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Attribute, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, attr: str, ctx: t.CInt) -> t.CInt: pass + def kind(self: Attribute) -> t.CInt: pass + def type_name(self: Attribute) -> str: pass + def dump(self: Attribute, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Subscript(AST): + value: AST | t.CPtr + slice: AST | t.CPtr + ctx: t.CInt + def __new__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Subscript, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Subscript) -> t.CInt: pass + def type_name(self: Subscript) -> str: pass + def dump(self: Subscript, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Starred(AST): + value: AST | t.CPtr + ctx: t.CInt + def __new__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Starred, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Starred) -> t.CInt: pass + def type_name(self: Starred) -> str: pass + def dump(self: Starred, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class List(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: List, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: List) -> t.CInt: pass + def type_name(self: List) -> str: pass + def dump(self: List, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Tuple(AST): + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + def __new__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def __init__(self: Tuple, pool: memhub.MemManager | t.CPtr, elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt) -> t.CInt: pass + def kind(self: Tuple) -> t.CInt: pass + def type_name(self: Tuple) -> str: pass + def dump(self: Tuple, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Slice(AST): + lower: AST | t.CPtr + upper: AST | t.CPtr + step: AST | t.CPtr + def __new__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Slice, pool: memhub.MemManager | t.CPtr, lower: AST | t.CPtr, upper: AST | t.CPtr, step: AST | t.CPtr) -> t.CInt: pass + def kind(self: Slice) -> t.CInt: pass + def type_name(self: Slice) -> str: pass + def dump(self: Slice, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class NamedExpr(AST): + target: AST | t.CPtr + value: AST | t.CPtr + def __new__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: NamedExpr, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: NamedExpr) -> t.CInt: pass + def type_name(self: NamedExpr) -> str: pass + def dump(self: NamedExpr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class OpNode(AST): + op: t.CInt + def __new__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def __init__(self: OpNode, pool: memhub.MemManager | t.CPtr, op: t.CInt) -> t.CInt: pass + def kind(self: OpNode) -> t.CInt: pass + def type_name(self: OpNode) -> str: pass + def dump(self: OpNode, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/TransPyV/temp/48fa8cc758d3826b.doc.json b/TransPyV/temp/48fa8cc758d3826b.doc.json new file mode 100644 index 0000000..ac292ab --- /dev/null +++ b/TransPyV/temp/48fa8cc758d3826b.doc.json @@ -0,0 +1 @@ +{"ForHandle": "for 循环语句处理器：继承 Mixin 获得 Trans 回指针", "ForHandle.Handle": "翻译 for i in range(...) 循环语句", "ForHandle._handle_ptr_iter": "指针迭代: for x in ptr: 直到解引用为空"} \ No newline at end of file diff --git a/TransPyV/temp/48fa8cc758d3826b.pyi b/TransPyV/temp/48fa8cc758d3826b.pyi new file mode 100644 index 0000000..08527b9 --- /dev/null +++ b/TransPyV/temp/48fa8cc758d3826b.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesFor.py +Module: lib.core.Handles.HandlesFor +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody +import lib.core.Handles.HandlesType as HandlesType + +@t.NoVTable +class ForHandle(HandlesBase.Mixin): + def __init__(self: ForHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ForHandle, node: ast.AST | t.CPtr) -> int: pass + def _handle_ptr_iter(self: ForHandle, for_node: ast.For | t.CPtr, var_name: str) -> int: pass + +def NewForHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ForHandle | t.CPtr: pass diff --git a/TransPyV/temp/49cf633017627a4a.doc.json b/TransPyV/temp/49cf633017627a4a.doc.json new file mode 100644 index 0000000..3e3ec7a --- /dev/null +++ b/TransPyV/temp/49cf633017627a4a.doc.json @@ -0,0 +1 @@ +{"Load_project_config": "加载工程配置文件（project.vpj）\n\n Args:\n path: 配置文件路径\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "_join_flags_array": "将 JSON 字符串数组拼接成空格分隔的字符串\n\n 用于 linker.flags 和 compiler.flags 数组。\n 用 while + 索引遍历，避免 list 迭代器 bug。\n\n Args:\n flags_val: JsonValue，应为字符串数组\n\n Returns:\n 空格分隔的字符串，或 None 如果 flags_val 为 None 或非数组或空数组\n ", "_join_path": "将相对路径与 project_dir 拼接为绝对路径\n\n - ./App + Test → Test/App\n - ./temp + Test → Test/temp\n - ../includes + Test → Test/../includes（OS 解析为 includes）\n - 绝对路径保持不变\n - None 返回 None\n ", "resolve_paths": "解析并规范化工程路径（src/temp/output/includes）\n\n 将 project.vpj 中的相对路径（以 ./ 开头）转换为基于 project_dir 的路径。\n\n Args:\n project_dir: 工程根目录（project.vpj 所在目录）\n\n Returns:\n 0 表示成功，非 0 表示失败\n ", "get_includes_binary_dir": "获取 includes.binary 目录路径（includes 目录 + .binary 后缀）\n\n includes 目录如 ../includes，则 includes.binary 为 ../includes.binary\n ", "print_config": "打印当前配置（调试用）"} \ No newline at end of file diff --git a/TransPyV/temp/49cf633017627a4a.pyi b/TransPyV/temp/49cf633017627a4a.pyi new file mode 100644 index 0000000..0ff98b1 --- /dev/null +++ b/TransPyV/temp/49cf633017627a4a.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from lib.Projectrans.Config.py +Module: lib.Projectrans.Config +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import viperlib +from json.__parser import parse as json_parse +from json import JsonValue, JSON_OBJECT, JSON_INT +import w32.fileio as fileio + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SourceDir: t.CExtern | str +TempDir: t.CExtern | str +OutputDir: t.CExtern | str +ProjectName: t.CExtern | str +ProjectVersion: t.CExtern | str +CompilerCmd: t.CExtern | str +CompilerFlags: t.CExtern | str +LinkerCmd: t.CExtern | str +LinkerFlags: t.CExtern | str +LinkerOutput: t.CExtern | str +TargetTriple: t.CExtern | str +TargetDataLayout: t.CExtern | str +SliceLevel: t.CExtern | t.CInt +StrictMode: t.CExtern | t.CInt +IncludesDir: t.CExtern | str + +def Load_project_config(path: str) -> int: pass + +def _join_flags_array(flags_val: JsonValue | t.CPtr) -> str: pass + +def _join_path(rel: str, project_dir: str) -> str: pass + +def resolve_paths(project_dir: str) -> int: pass + +def get_includes_binary_dir() -> str: pass + +def print_config() -> t.CInt: pass diff --git a/TransPyV/temp/4dd6b3f1427d1cc5.pyi b/TransPyV/temp/4dd6b3f1427d1cc5.pyi new file mode 100644 index 0000000..dd15aa2 --- /dev/null +++ b/TransPyV/temp/4dd6b3f1427d1cc5.pyi @@ -0,0 +1,82 @@ +""" +Auto-generated Python stub file from ast.match.py +Module: ast.match +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list + +class MatchCase(AST): + pattern: AST | t.CPtr + guard: AST | t.CPtr + def __new__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchCase, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, guard: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchCase) -> t.CInt: pass + def type_name(self: MatchCase) -> str: pass + def dump(self: MatchCase, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchValue(AST): + value: AST | t.CPtr + def __new__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchValue, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchValue) -> t.CInt: pass + def type_name(self: MatchValue) -> str: pass + def dump(self: MatchValue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSingleton(AST): + value: AST | t.CPtr + def __new__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSingleton, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: MatchSingleton) -> t.CInt: pass + def type_name(self: MatchSingleton) -> str: pass + def dump(self: MatchSingleton, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchSequence(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchSequence, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchSequence) -> t.CInt: pass + def type_name(self: MatchSequence) -> str: pass + def dump(self: MatchSequence, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchMapping(AST): + keys: list[AST | t.CPtr] | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + rest: str + def __new__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def __init__(self: MatchMapping, pool: memhub.MemManager | t.CPtr, keys: list[AST | t.CPtr] | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, rest: str) -> t.CInt: pass + def kind(self: MatchMapping) -> t.CInt: pass + def type_name(self: MatchMapping) -> str: pass + def dump(self: MatchMapping, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchClass(AST): + cls: AST | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + kwd_attrs: list[AST | t.CPtr] | t.CPtr + kwd_patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchClass, pool: memhub.MemManager | t.CPtr, cls: AST | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr, kwd_attrs: list[AST | t.CPtr] | t.CPtr, kwd_patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchClass) -> t.CInt: pass + def type_name(self: MatchClass) -> str: pass + def dump(self: MatchClass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchStar(AST): + name: str + def __new__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchStar, pool: memhub.MemManager | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchStar) -> t.CInt: pass + def type_name(self: MatchStar) -> str: pass + def dump(self: MatchStar, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchAs(AST): + pattern: AST | t.CPtr + name: str + def __new__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def __init__(self: MatchAs, pool: memhub.MemManager | t.CPtr, pattern: AST | t.CPtr, name: str) -> t.CInt: pass + def kind(self: MatchAs) -> t.CInt: pass + def type_name(self: MatchAs) -> str: pass + def dump(self: MatchAs, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class MatchOr(AST): + patterns: list[AST | t.CPtr] | t.CPtr + def __new__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: MatchOr, pool: memhub.MemManager | t.CPtr, patterns: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: MatchOr) -> t.CInt: pass + def type_name(self: MatchOr) -> str: pass + def dump(self: MatchOr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/TransPyV/temp/51f0144d6e8188d9.doc.json b/TransPyV/temp/51f0144d6e8188d9.doc.json new file mode 100644 index 0000000..a63fb44 --- /dev/null +++ b/TransPyV/temp/51f0144d6e8188d9.doc.json @@ -0,0 +1 @@ +{"WhileHandle": "while 循环语句处理器：继承 Mixin 获得 Trans 回指针", "WhileHandle.Handle": "翻译 while 循环语句"} \ No newline at end of file diff --git a/TransPyV/temp/51f0144d6e8188d9.pyi b/TransPyV/temp/51f0144d6e8188d9.pyi new file mode 100644 index 0000000..eb71aad --- /dev/null +++ b/TransPyV/temp/51f0144d6e8188d9.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesWhile.py +Module: lib.core.Handles.HandlesWhile +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class WhileHandle(HandlesBase.Mixin): + def __init__(self: WhileHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: WhileHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewWhileHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> WhileHandle | t.CPtr: pass diff --git a/TransPyV/temp/54014f6c53e9640b.doc.json b/TransPyV/temp/54014f6c53e9640b.doc.json new file mode 100644 index 0000000..ca04595 --- /dev/null +++ b/TransPyV/temp/54014f6c53e9640b.doc.json @@ -0,0 +1 @@ +{"reset_visible_structs": "重置可见性状态（每个文件翻译前调用）\n\n pool: 内存分配器（严格模式下创建 HashTable）\n strict: 1=严格模式（用户文件），0=宽松模式（includes 文件）\n ", "add_visible_struct": "添加可见结构体名（类定义时和 from-import 时调用）", "is_struct_visible": "检查结构体是否可见，返回 1=可见 / 0=不可见", "init_struct_table": "初始化结构体注册表，返回 1 成功", "_get_struct_entry": "获取第 i 个结构体条目", "_get_field_entry": "获取结构体中第 i 个字段条目", "register_struct": "注册结构体类型，返回 StructEntry 指针\n\n 用类型指针去重（is 比较），不用类名去重。\n 这样跨模块同名类可以共存，各自有独立的字段定义。\n sha1 参数在注册时直接设置 ModuleSha1（避免 set_struct_sha1 找错 entry）。\n ", "set_struct_sha1": "设置结构体所属模块的 SHA1，返回 1=成功 / 0=失败", "get_struct_sha1": "获取结构体所属模块的 SHA1，返回 None=无 SHA1", "add_field": "向结构体添加字段，返回字段索引（-1 失败）\n\n default_val: 字段默认值 AST 节点（可选，None=无默认值）\n annot_class_name: 原始类型注解的类名（可选，用于联合类型字段的方法调用解析）\n ", "find_struct": "按类名查找结构体，返回 StructEntry 或 None", "find_struct_by_module": "按类名 + SHA1 查找结构体，返回 StructEntry 或 None\n\n SHA1 非 None 时：精确匹配 name+sha1，未找到返回 None（不回退，避免跨模块同名找错）\n SHA1 为 None 时：回退到 find_struct(name) 按类名查找第一个\n ", "find_struct_by_type": "按类型指针查找结构体（is 比较），返回 StructEntry 或 None", "lookup_field": "按结构体类型和字段名查找字段信息\n\n 两遍查找策略（跨模块同名类冲突的根本修复）:\n 第一遍: 类型指针 is 身份比较（最高优先级，类型唯一确定 entry）\n 第二遍: 名称回退（is 未匹配时，按类名/strstr 匹配，字段未找到时 continue）\n ", "lookup_field_by_class": "按类名和字段名查找字段信息\n\n 优先用 SHA1 匹配（跨模块同名类区分），无 SHA1 时按类名查找第一个\n ", "get_struct_type": "按类名获取结构体的 LLVM 类型", "get_field_by_index": "按类名和字段索引获取字段信息", "struct_has_defaults": "检查结构体是否有任何带默认值的字段，返回 1=有 / 0=无", "_is_struct_type": "检查 ty 是否是 Struct 类型（独立函数，规避嵌套 match 编译器 BUG）", "get_struct_type_from_value": "从 Value 的类型推断结构体类型", "mark_as_union": "标记已注册的结构体为联合体，返回 1=成功 / 0=失败", "is_union_by_name": "按类名检查是否为联合体，返回 1=是 / 0=否", "is_union_by_type": "按类型指针检查是否为联合体，返回 1=是 / 0=否\n\n 先尝试类型指针 == 比较，失败时回退到名称匹配\n ", "mark_as_oop": "标记已注册的结构体为 OOP，返回 1=成功 / 0=失败", "is_oop_by_name": "按类名检查是否为 OOP 结构体，返回 1=是 / 0=否", "mark_has_init": "标记结构体拥有 __init__ 方法，返回 1=成功 / 0=失败", "has_init_by_name": "按类名检查是否有 __init__ 方法，返回 1=有 / 0=无", "mark_has_new": "标记结构体拥有 __new__ 方法，返回 1=成功 / 0=失败", "has_new_by_name": "按类名检查是否有 __new__ 方法，返回 1=有 / 0=无", "get_class_name_by_type": "按类型指针查找类名，返回类名字符串或 None", "_extract_struct_name": "从 LLVMType 提取 Struct 变体的 Name 字段，返回 None 非 Struct", "mark_has_vtable": "标记结构体拥有虚表，返回 1=成功 / 0=失败", "has_vtable_by_name": "按类名检查是否有虚表，返回 1=有 / 0=无", "mark_novtable": "标记结构体为 NoVTable，返回 1=成功 / 0=失败", "is_novtable_by_name": "按类名检查是否为 NoVTable，返回 1=是 / 0=否", "set_parent_name": "设置结构体的父类名，返回 1=成功 / 0=失败", "get_parent_name": "获取结构体的父类名，返回 None=无父类", "set_vtable_methods": "设置虚表方法名数组，返回 1=成功 / 0=失败", "get_vtable_method_index": "获取方法在虚表中的索引，返回 -1=未找到", "get_vtable_method_count": "获取虚表方法数量", "get_vtable_method_name": "获取虚表中第 idx 个方法名，返回 None=越界", "get_field_count": "获取结构体字段数量，返回 -1=未找到", "get_field_name_ptr": "获取 FieldEntry 的字段名，返回 None=未设置", "get_field_type_ptr": "获取 FieldEntry 的字段类型，返回 None=未设置", "get_field_default_ptr": "获取 FieldEntry 的默认值 AST 节点，返回 None=无默认值", "get_field_annot_class_name": "获取 FieldEntry 的原始类型注解类名，返回 None=未设置"} \ No newline at end of file diff --git a/TransPyV/temp/54014f6c53e9640b.pyi b/TransPyV/temp/54014f6c53e9640b.pyi new file mode 100644 index 0000000..9662312 --- /dev/null +++ b/TransPyV/temp/54014f6c53e9640b.pyi @@ -0,0 +1,139 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesStruct.py +Module: lib.core.Handles.HandlesStruct +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio +import ast +import hashtable + +STRUCT_MAX: t.CDefine = 512 +FIELD_MAX: t.CDefine = 32 +FIELD_NAME_MAX: t.CDefine = 64 + +@t.NoVTable +class FieldEntry: + Name: t.CChar | t.CPtr + Index: int + Ty: llvmlite.LLVMType | t.CPtr + DefaultVal: ast.AST | t.CPtr + AnnotClassName: t.CChar | t.CPtr +@t.NoVTable +class StructEntry: + Name: t.CChar | t.CPtr + Ty: llvmlite.LLVMType | t.CPtr + FieldCount: int + Fields: FieldEntry | t.CPtr + IsUnion: int + IsOOP: int + HasInit: int + HasNew: int + HasVTable: int + IsNoVTable: int + ParentName: t.CChar | t.CPtr + VTableMethodCount: int + VTableMethods: t.CChar | t.CPtr + ModuleSha1: t.CChar | t.CPtr + +_struct_table: t.CExtern | StructEntry | t.CPtr +_struct_count: t.CExtern | int +_visible_structs: t.CExtern | hashtable.HashTable | t.CPtr +_strict_visibility: t.CExtern | int + +def reset_visible_structs(pool: memhub.MemBuddy | t.CPtr, strict: int) -> t.CInt: pass + +def add_visible_struct(pool: memhub.MemBuddy | t.CPtr, name: str) -> t.CInt: pass + +def is_struct_visible(name: str) -> int: pass + +def init_struct_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_struct_entry(i: int) -> StructEntry | t.CPtr: pass + +def _get_field_entry(struct_entry: StructEntry | t.CPtr, i: int) -> FieldEntry | t.CPtr: pass + +def register_struct(pool: memhub.MemBuddy | t.CPtr, name: str, struct_ty: llvmlite.LLVMType | t.CPtr, sha1: str) -> StructEntry | t.CPtr: pass + +def set_struct_sha1(pool: memhub.MemBuddy | t.CPtr, class_name: str, sha1: str) -> int: pass + +def get_struct_sha1(class_name: str) -> str: pass + +def add_field(pool: memhub.MemBuddy | t.CPtr, struct_entry: StructEntry | t.CPtr, field_name: str, field_ty: llvmlite.LLVMType | t.CPtr, default_val: ast.AST | t.CPtr, annot_class_name: t.CChar | t.CPtr) -> int: pass + +def find_struct(name: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_module(name: str, sha1: str) -> StructEntry | t.CPtr: pass + +def find_struct_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> StructEntry | t.CPtr: pass + +def lookup_field(struct_ty: llvmlite.LLVMType | t.CPtr, field_name: str) -> FieldEntry | t.CPtr: pass + +def lookup_field_by_class(class_name: str, field_name: str, sha1: str) -> FieldEntry | t.CPtr: pass + +def get_struct_type(class_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_by_index(class_name: str, idx: int) -> FieldEntry | t.CPtr: pass + +def struct_has_defaults(class_name: str) -> int: pass + +def _is_struct_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_struct_type_from_value(val: llvmlite.Value | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def mark_as_union(name: str) -> int: pass + +def is_union_by_name(name: str) -> int: pass + +def is_union_by_type(struct_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def mark_as_oop(name: str) -> int: pass + +def is_oop_by_name(name: str) -> int: pass + +def mark_has_init(name: str) -> int: pass + +def has_init_by_name(name: str) -> int: pass + +def mark_has_new(name: str) -> int: pass + +def has_new_by_name(name: str) -> int: pass + +def get_class_name_by_type(pool: memhub.MemBuddy | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def _extract_struct_name(ty: llvmlite.LLVMType | t.CPtr) -> str: pass + +def mark_has_vtable(name: str) -> int: pass + +def has_vtable_by_name(name: str) -> int: pass + +def mark_novtable(name: str) -> int: pass + +def is_novtable_by_name(name: str) -> int: pass + +def set_parent_name(name: str, parent_name: str) -> int: pass + +def get_parent_name(name: str) -> str: pass + +def set_vtable_methods(name: str, methods: t.CChar | t.CPtr, count: int) -> int: pass + +def get_vtable_method_index(name: str, method_name: str) -> int: pass + +def get_vtable_method_count(name: str) -> int: pass + +def get_vtable_method_name(name: str, idx: int) -> str: pass + +def get_field_count(name: str) -> int: pass + +def get_field_name_ptr(fe: FieldEntry | t.CPtr) -> str: pass + +def get_field_type_ptr(fe: FieldEntry | t.CPtr) -> llvmlite.LLVMType | t.CPtr: pass + +def get_field_default_ptr(fe: FieldEntry | t.CPtr) -> ast.AST | t.CPtr: pass + +def get_field_annot_class_name(fe: FieldEntry | t.CPtr) -> str: pass diff --git a/TransPyV/temp/541a090dde0cc1ba.doc.json b/TransPyV/temp/541a090dde0cc1ba.doc.json new file mode 100644 index 0000000..42f2fd1 --- /dev/null +++ b/TransPyV/temp/541a090dde0cc1ba.doc.json @@ -0,0 +1 @@ +{"AugAssignHandle": "增强赋值处理器 (+=, -=, *=, etc.)：继承 Mixin 获得 Trans 回指针", "AugAssignHandle.Handle": "翻译增强赋值语句 (x += 1, y -= 2, etc.)", "_apply_aug_op": "应用增强赋值的二元运算（指针算术 + 整数运算）"} \ No newline at end of file diff --git a/TransPyV/temp/541a090dde0cc1ba.pyi b/TransPyV/temp/541a090dde0cc1ba.pyi new file mode 100644 index 0000000..ee3342d --- /dev/null +++ b/TransPyV/temp/541a090dde0cc1ba.pyi @@ -0,0 +1,27 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesAugAssign.py +Module: lib.core.Handles.HandlesAugAssign +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import stdio +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal + +@t.NoVTable +class AugAssignHandle(HandlesBase.Mixin): + def __init__(self: AugAssignHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: AugAssignHandle, node: ast.AST | t.CPtr) -> int: pass + +def _apply_aug_op(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, op: int, lhs: llvmlite.Value | t.CPtr, rhs: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewAugAssignHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> AugAssignHandle | t.CPtr: pass diff --git a/TransPyV/temp/57288496f7c2d1ad.pyi b/TransPyV/temp/57288496f7c2d1ad.pyi new file mode 100644 index 0000000..0624d9a --- /dev/null +++ b/TransPyV/temp/57288496f7c2d1ad.pyi @@ -0,0 +1,71 @@ +""" +Auto-generated Python stub file from json.__init__.py +Module: json.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperio + +JSON_NULL: t.CDefine = 0 +JSON_BOOL: t.CDefine = 1 +JSON_INT: t.CDefine = 2 +JSON_FLOAT: t.CDefine = 3 +JSON_STRING: t.CDefine = 4 +JSON_ARRAY: t.CDefine = 5 +JSON_OBJECT: t.CDefine = 6 + +class JsonValue: + vtype: t.CInt + bool_val: t.CInt + int_val: t.CInt64T + float_val: t.CDouble + str_val: t.CChar | t.CPtr + next: 'JsonValue' | t.CPtr + key: t.CChar | t.CPtr + child: 'JsonValue' | t.CPtr + child_count: t.CSizeT + def __new__(self: JsonValue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def type(self: JsonValue) -> t.CInt: pass + def is_null(self: JsonValue) -> bool: pass + def is_bool(self: JsonValue) -> bool: pass + def is_int(self: JsonValue) -> bool: pass + def is_float(self: JsonValue) -> bool: pass + def is_string(self: JsonValue) -> bool: pass + def is_array(self: JsonValue) -> bool: pass + def is_object(self: JsonValue) -> bool: pass + def as_bool(self: JsonValue) -> bool: pass + def as_int(self: JsonValue) -> t.CInt64T: pass + def as_float(self: JsonValue) -> t.CDouble: pass + def as_string(self: JsonValue) -> t.CChar | t.CPtr: pass + def __len__(self: JsonValue) -> t.CSizeT: pass + def __getitem__(self: JsonValue, key: t.CChar | t.CPtr) -> 'JsonValue' | t.CPtr: pass + def __setitem__(self: JsonValue, key: t.CChar | t.CPtr, val: 'JsonValue' | t.CPtr) -> t.CInt: pass + def get_item(self: JsonValue, index: t.CSizeT) -> 'JsonValue' | t.CPtr: pass + def _pool_alloc(self: JsonValue, size: t.CSizeT) -> t.CChar | t.CPtr: pass + def _append_child(self: JsonValue, child: 'JsonValue' | t.CPtr) -> t.CInt: pass + +def null(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def bool_val(pool: memhub.MemManager | t.CPtr, val: bool) -> JsonValue | t.CPtr: pass + +def int_val(pool: memhub.MemManager | t.CPtr, val: t.CInt64T) -> JsonValue | t.CPtr: pass + +def float_val(pool: memhub.MemManager | t.CPtr, val: t.CDouble) -> JsonValue | t.CPtr: pass + +def string_val(pool: memhub.MemManager | t.CPtr, val: t.CChar | t.CPtr) -> JsonValue | t.CPtr: pass + +def array(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def object(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: pass + +def array_append(pool: memhub.MemManager | t.CPtr, arr: JsonValue | t.CPtr, item: JsonValue | t.CPtr) -> t.CInt: pass + +def object_set(pool: memhub.MemManager | t.CPtr, obj: JsonValue | t.CPtr, key: t.CChar | t.CPtr, val: JsonValue | t.CPtr) -> t.CInt: pass + + +from .__parser import parse +from .__writer import write \ No newline at end of file diff --git a/TransPyV/temp/5dab8cb390496d22.pyi b/TransPyV/temp/5dab8cb390496d22.pyi new file mode 100644 index 0000000..7056767 --- /dev/null +++ b/TransPyV/temp/5dab8cb390496d22.pyi @@ -0,0 +1,182 @@ +""" +Auto-generated Python stub file from ast.base.py +Module: ast.base +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .tokens import TokOp + +class ASTKind(t.CEnum): + Module: t.State + FunctionDef: t.State + ClassDef: t.State + Assign: t.State + If: t.State + For: t.State + While: t.State + Return: t.State + Expr: t.State + Name: t.State + Constant: t.State + BinOp: t.State + UnaryOp: t.State + Call: t.State + Compare: t.State + Expression: t.State + Interactive: t.State + FunctionType: t.State + Delete: t.State + AugAssign: t.State + AnnAssign: t.State + With: t.State + Raise: t.State + Try: t.State + Assert: t.State + Global: t.State + Nonlocal: t.State + Pass: t.State + Break: t.State + Continue: t.State + Import: t.State + ImportFrom: t.State + Match: t.State + BoolOp: t.State + Lambda: t.State + IfExp: t.State + Dict: t.State + Set: t.State + ListComp: t.State + SetComp: t.State + DictComp: t.State + GeneratorExp: t.State + Await: t.State + Yield: t.State + YieldFrom: t.State + FormattedValue: t.State + JoinedStr: t.State + Attribute: t.State + Subscript: t.State + Starred: t.State + List: t.State + Tuple: t.State + Slice: t.State + NamedExpr: t.State + ExceptHandler: t.State + Arguments: t.State + Arg: t.State + Keyword: t.State + Alias: t.State + WithItem: t.State + Comprehension: t.State + OpNode: t.State + MatchCase: t.State + MatchValue: t.State + MatchSingleton: t.State + MatchSequence: t.State + MatchMapping: t.State + MatchClass: t.State + MatchStar: t.State + MatchAs: t.State + MatchOr: t.State +class ASTCtx(t.CEnum): + Load: t.State + Store: t.State + Del: t.State +class OpKind(t.CEnum): + Add: t.State + Sub: t.State + Mult: t.State + MatMult: t.State + Div: t.State + Mod: t.State + Pow: t.State + LShift: t.State + RShift: t.State + BitOr: t.State + BitXor: t.State + BitAnd: t.State + FloorDiv: t.State + And: t.State + Or: t.State + Eq: t.State + Ne: t.State + Lt: t.State + Le: t.State + Gt: t.State + Ge: t.State + Is: t.State + IsNot: t.State + In: t.State + NotIn: t.State + Not: t.State + UAdd: t.State + USub: t.State + Invert: t.State + NoneOp: t.State + +CONST_INT: t.CDefine = 1 +CONST_FLOAT: t.CDefine = 2 +CONST_STR: t.CDefine = 3 +CONST_BOOL: t.CDefine = 4 +CONST_NONE: t.CDefine = 5 +FLAG_IS_ASYNC: t.CDefine = 1 +FLAG_SIMPLE: t.CDefine = 2 +FLAG_HAS_STAR: t.CDefine = 4 + +class ASTFlag(t.CEnum): + IsAsync: t.State + Simple: t.State + HasStar: t.State + +def _copy_str(pool: memhub.MemManager | t.CPtr, src: str) -> str: pass + +def _set_pos(node: AST | t.CPtr, lineno: t.CInt, col_offset: t.CInt, end_lineno: t.CInt, end_col_offset: t.CInt) -> t.CInt: pass + +def _inherit_pos(node: AST | t.CPtr, ref: AST | t.CPtr) -> t.CInt: pass + +def _init_ast(node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> t.CInt: pass + +def _set_parent_list(lst: list[AST | t.CPtr] | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _append_child(lst: list[AST | t.CPtr] | t.CPtr, node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _binop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _augop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _cmpop_from_op(op_id: t.CInt) -> t.CInt: pass + +def _emit(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_str(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, text: str) -> t.CSizeT: pass + +def _emit_int(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, val: t.CInt64T) -> t.CSizeT: pass + +def _op_name(op: t.CInt) -> str: pass + +def _dump_list(lst: list[AST | t.CPtr] | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + +def _dump_op_list(lst: t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + + +@t.CVTable +class AST: + parent: AST | t.CPtr + pool: memhub.MemManager | t.CPtr + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + children: list[AST | t.CPtr] | t.CPtr + def kind(self: AST) -> t.CInt: pass + def type_name(self: AST) -> str: pass + def dump(self: AST, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass + def accept(self: AST, visitor: t.CPtr) -> t.CInt: pass + def append(self: AST, node: AST | t.CPtr) -> t.CInt: pass + +def dump(node: AST | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: pass diff --git a/TransPyV/temp/6446627d4f07a1b5.pyi b/TransPyV/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/TransPyV/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/TransPyV/temp/6503c97dde0c79c4.pyi b/TransPyV/temp/6503c97dde0c79c4.pyi new file mode 100644 index 0000000..d332326 --- /dev/null +++ b/TransPyV/temp/6503c97dde0c79c4.pyi @@ -0,0 +1,189 @@ +""" +Auto-generated Python stub file from posix.py +Module: posix +""" + + +import t, c +from stdint import * + +DIR: t.CTypedef = t.CVoid | t.CPtr +mode_t: t.CTypedef = t.CUInt32T +pid_t: t.CTypedef = t.CInt32T +off_t: t.CTypedef = t.CInt64T +dev_t: t.CTypedef = t.CUInt64T +ino_t: t.CTypedef = t.CUInt64T +nlink_t: t.CTypedef = t.CUInt64T +uid_t: t.CTypedef = t.CUInt32T +gid_t: t.CTypedef = t.CUInt32T +blksize_t: t.CTypedef = t.CInt64T +blkcnt_t: t.CTypedef = t.CInt64T +time_t: t.CTypedef = t.CInt64T +O_RDONLY: t.CDefine = 0 +O_WRONLY: t.CDefine = 1 +O_RDWR: t.CDefine = 2 +O_CREAT: t.CDefine = 0o100 +O_EXCL: t.CDefine = 0o200 +O_TRUNC: t.CDefine = 0o1000 +O_APPEND: t.CDefine = 0o2000 +O_NONBLOCK: t.CDefine = 0o4000 +O_CLOEXEC: t.CDefine = 0o2000000 +F_OK: t.CDefine = 0 # 文件存在 +R_OK: t.CDefine = 4 # 可读 +W_OK: t.CDefine = 2 # 可写 +X_OK: t.CDefine = 1 # 可执行 +S_IFMT: t.CDefine = 0o170000 # 类型掩码 +S_IFSOCK: t.CDefine = 0o140000 # 套接字 +S_IFLNK: t.CDefine = 0o120000 # 符号链接 +S_IFREG: t.CDefine = 0o100000 # 普通文件 +S_IFBLK: t.CDefine = 0o060000 # 块设备 +S_IFDIR: t.CDefine = 0o040000 # 目录 +S_IFCHR: t.CDefine = 0o020000 # 字符设备 +S_IFIFO: t.CDefine = 0o010000 # FIFO/管道 +S_ISUID: t.CDefine = 0o4000 # set-user-ID +S_ISGID: t.CDefine = 0o2000 # set-group-ID +S_ISVTX: t.CDefine = 0o1000 # sticky 位 +S_IRWXU: t.CDefine = 0o700 # 用户读写执行 +S_IRUSR: t.CDefine = 0o400 # 用户读 +S_IWUSR: t.CDefine = 0o200 # 用户写 +S_IXUSR: t.CDefine = 0o100 # 用户执行 +S_IRWXG: t.CDefine = 0o070 # 组读写执行 +S_IRGRP: t.CDefine = 0o040 # 组读 +S_IWGRP: t.CDefine = 0o020 # 组写 +S_IXGRP: t.CDefine = 0o010 # 组执行 +S_IRWXO: t.CDefine = 0o007 # 其他读写执行 +S_IROTH: t.CDefine = 0o004 # 其他读 +S_IWOTH: t.CDefine = 0o002 # 其他写 +S_IXOTH: t.CDefine = 0o001 # 其他执行 +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +PATH_MAX: t.CDefine = 4096 +NAME_MAX: t.CDefine = 255 +_SC_NPROCESSORS_ONLN: t.CDefine = 84 +_SC_PAGESIZE: t.CDefine = 30 +_SC_CLK_TCK: t.CDefine = 2 +DT_UNKNOWN: t.CDefine = 0 +DT_FIFO: t.CDefine = 1 +DT_CHR: t.CDefine = 2 +DT_DIR: t.CDefine = 4 +DT_BLK: t.CDefine = 6 +DT_REG: t.CDefine = 8 +DT_LNK: t.CDefine = 10 +DT_SOCK: t.CDefine = 12 +DT_WHT: t.CDefine = 14 +WNOHANG: t.CDefine = 1 +WUNTRACED: t.CDefine = 2 +WCONTINUED: t.CDefine = 8 + +class Stat: + st_dev: dev_t + st_ino: ino_t + st_nlink: nlink_t + st_mode: mode_t + st_uid: uid_t + st_gid: gid_t + __pad0: INT + st_rdev: dev_t + st_size: off_t + st_blksize: blksize_t + st_blocks: blkcnt_t + st_atim_sec: time_t + st_atim_nsec: LONGLONG + st_mtim_sec: time_t + st_mtim_nsec: LONGLONG + st_ctim_sec: time_t + st_ctim_nsec: LONGLONG + __unused0: LONGLONG + __unused1: LONGLONG + __unused2: LONGLONG +class Dirent: + d_ino: ino_t + d_off: off_t + d_reclen: USHORT + d_type: BYTE + d_name: CHARPTR + +def open(path: t.CConst | str, oflag: INT, *args) -> INT | t.State: pass + +def close(fd: INT) -> INT | t.State: pass + +def read(fd: INT, buf: bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def write(fd: INT, buf: t.CConst | bytes, count: t.CSizeT) -> ssize_t | t.State: pass + +def lseek(fd: INT, offset: off_t, whence: INT) -> off_t | t.State: pass + +def unlink(path: t.CConst | str) -> INT | t.State: pass + +def access(path: t.CConst | str, amode: INT) -> INT | t.State: pass + +def realpath(path: t.CConst | str, resolved: CHARPTR) -> CHARPTR | t.State: pass + +def rename(oldpath: t.CConst | str, newpath: t.CConst | str) -> INT | t.State: pass + +def chmod(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def stat(path: t.CConst | str, buf: Stat | t.CPtr) -> INT | t.State: pass + +def fstat(fd: INT, buf: Stat | t.CPtr) -> INT | t.State: pass + +def opendir(name: t.CConst | str) -> DIR | t.CPtr | t.State: pass + +def readdir(dirp: DIR | t.CPtr) -> Dirent | t.CPtr | t.State: pass + +def closedir(dirp: DIR | t.CPtr) -> INT | t.State: pass + +def mkdir(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + +def rmdir(path: t.CConst | str) -> INT | t.State: pass + +def getcwd(buf: CHARPTR, size: t.CSizeT) -> CHARPTR | t.State: pass + +def chdir(path: t.CConst | str) -> INT | t.State: pass + +def fork() -> pid_t | t.State: pass + +def execvp(file: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def execv(path: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass + +def waitpid(pid: pid_t, status: INT | t.CPtr, options: INT) -> pid_t | t.State: pass + +def pipe(fds: INT | t.CPtr) -> INT | t.State: pass + +def dup2(oldfd: INT, newfd: INT) -> INT | t.State: pass + +def _exit(status: INT) -> t.CVoid | t.State: pass + +def getpid() -> pid_t | t.State: pass + +def getenv(name: t.CConst | str) -> CHARPTR | t.State: pass + +def setenv(name: t.CConst | str, value: t.CConst | str, overwrite: INT) -> INT | t.State: pass + +def sysconf(name: INT) -> LONG | t.State: pass + +def isatty(fd: INT) -> INT | t.State: pass + +def S_ISREG(mode: mode_t) -> bool: pass + +def S_ISDIR(mode: mode_t) -> bool: pass + +def S_ISLNK(mode: mode_t) -> bool: pass + +def S_ISCHR(mode: mode_t) -> bool: pass + +def S_ISBLK(mode: mode_t) -> bool: pass + +def S_ISFIFO(mode: mode_t) -> bool: pass + +def S_ISSOCK(mode: mode_t) -> bool: pass + +def WIFEXITED(status: INT) -> bool: pass + +def WEXITSTATUS(status: INT) -> INT: pass + +def WIFSIGNALED(status: INT) -> bool: pass + +def WTERMSIG(status: INT) -> INT: pass diff --git a/TransPyV/temp/657e182b27c2a022.pyi b/TransPyV/temp/657e182b27c2a022.pyi new file mode 100644 index 0000000..0a2057d --- /dev/null +++ b/TransPyV/temp/657e182b27c2a022.pyi @@ -0,0 +1,226 @@ +""" +Auto-generated Python stub file from ast.stmts.py +Module: ast.stmts +""" + + +import t, c +from stdint import * +import memhub +import string +from .base import AST, ASTKind, ASTCtx, OpKind, _init_ast, _copy_str, _set_parent_list, _emit, _emit_str, _emit_int, _dump_list, FLAG_IS_ASYNC + +@t.CVTable +class Module(AST): + def __new__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Module, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Module) -> t.CInt: pass + def type_name(self: Module) -> str: pass + def dump(self: Module, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assign(AST): + targets: list[AST | t.CPtr] | t.CPtr + value: AST | t.CPtr + def __new__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assign, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assign) -> t.CInt: pass + def type_name(self: Assign) -> str: pass + def dump(self: Assign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class If(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: If, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: If) -> t.CInt: pass + def type_name(self: If) -> str: pass + def dump(self: If, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class For(AST): + target: AST | t.CPtr + iter: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: For, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, iter: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: For) -> t.CInt: pass + def type_name(self: For) -> str: pass + def dump(self: For, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class While(AST): + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + def __new__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: While, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: While) -> t.CInt: pass + def type_name(self: While) -> str: pass + def dump(self: While, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Return(AST): + value: AST | t.CPtr + def __new__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Return, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Return) -> t.CInt: pass + def type_name(self: Return) -> str: pass + def dump(self: Return, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expr(AST): + value: AST | t.CPtr + def __new__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expr, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expr) -> t.CInt: pass + def type_name(self: Expr) -> str: pass + def dump(self: Expr, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionDef(AST): + name: str + args: AST | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + flags: t.CInt + def __new__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: FunctionDef, pool: memhub.MemManager | t.CPtr, name: str, args: AST | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: FunctionDef) -> t.CInt: pass + def type_name(self: FunctionDef) -> str: pass + def dump(self: FunctionDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Expression(AST): + body: AST | t.CPtr + def __new__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Expression, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Expression) -> t.CInt: pass + def type_name(self: Expression) -> str: pass + def dump(self: Expression, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Interactive(AST): + body: AST | t.CPtr + def __new__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Interactive, pool: memhub.MemManager | t.CPtr, body: AST | t.CPtr) -> t.CInt: pass + def kind(self: Interactive) -> t.CInt: pass + def type_name(self: Interactive) -> str: pass + def dump(self: Interactive, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class FunctionType(AST): + argtypes: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + def __new__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def __init__(self: FunctionType, pool: memhub.MemManager | t.CPtr, argtypes: list[AST | t.CPtr] | t.CPtr, returns: AST | t.CPtr) -> t.CInt: pass + def kind(self: FunctionType) -> t.CInt: pass + def type_name(self: FunctionType) -> str: pass + def dump(self: FunctionType, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ClassDef(AST): + name: str + bases: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + type_params: list[str] | t.CPtr + def __new__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def __init__(self: ClassDef, pool: memhub.MemManager | t.CPtr, name: str, bases: list[AST | t.CPtr] | t.CPtr, keywords: list[AST | t.CPtr] | t.CPtr, decorator_list: list[AST | t.CPtr] | t.CPtr, type_params: list[str] | t.CPtr) -> t.CInt: pass + def kind(self: ClassDef) -> t.CInt: pass + def type_name(self: ClassDef) -> str: pass + def dump(self: ClassDef, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Delete(AST): + targets: list[AST | t.CPtr] | t.CPtr + def __new__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Delete, pool: memhub.MemManager | t.CPtr, targets: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Delete) -> t.CInt: pass + def type_name(self: Delete) -> str: pass + def dump(self: Delete, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AugAssign(AST): + target: AST | t.CPtr + op: t.CInt + value: AST | t.CPtr + def __new__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def __init__(self: AugAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr) -> t.CInt: pass + def kind(self: AugAssign) -> t.CInt: pass + def type_name(self: AugAssign) -> str: pass + def dump(self: AugAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class AnnAssign(AST): + target: AST | t.CPtr + annotation: AST | t.CPtr + value: AST | t.CPtr + simple: t.CInt + def __new__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def __init__(self: AnnAssign, pool: memhub.MemManager | t.CPtr, target: AST | t.CPtr, annotation: AST | t.CPtr, value: AST | t.CPtr, simple: t.CInt) -> t.CInt: pass + def kind(self: AnnAssign) -> t.CInt: pass + def type_name(self: AnnAssign) -> str: pass + def dump(self: AnnAssign, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class With(AST): + items: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + def __new__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def __init__(self: With, pool: memhub.MemManager | t.CPtr, items: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt) -> t.CInt: pass + def kind(self: With) -> t.CInt: pass + def type_name(self: With) -> str: pass + def dump(self: With, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Raise(AST): + exc: AST | t.CPtr + cause: AST | t.CPtr + def __new__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Raise, pool: memhub.MemManager | t.CPtr, exc: AST | t.CPtr, cause: AST | t.CPtr) -> t.CInt: pass + def kind(self: Raise) -> t.CInt: pass + def type_name(self: Raise) -> str: pass + def dump(self: Raise, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Try(AST): + handlers: list[AST | t.CPtr] | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + finalbody: list[AST | t.CPtr] | t.CPtr + def __new__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Try, pool: memhub.MemManager | t.CPtr, handlers: list[AST | t.CPtr] | t.CPtr, orelse: list[AST | t.CPtr] | t.CPtr, finalbody: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Try) -> t.CInt: pass + def type_name(self: Try) -> str: pass + def dump(self: Try, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Assert(AST): + test: AST | t.CPtr + msg: AST | t.CPtr + def __new__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def __init__(self: Assert, pool: memhub.MemManager | t.CPtr, test: AST | t.CPtr, msg: AST | t.CPtr) -> t.CInt: pass + def kind(self: Assert) -> t.CInt: pass + def type_name(self: Assert) -> str: pass + def dump(self: Assert, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Global(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Global, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Global) -> t.CInt: pass + def type_name(self: Global) -> str: pass + def dump(self: Global, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Nonlocal(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Nonlocal, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Nonlocal) -> t.CInt: pass + def type_name(self: Nonlocal) -> str: pass + def dump(self: Nonlocal, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Pass(AST): + def __new__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Pass, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Pass) -> t.CInt: pass + def type_name(self: Pass) -> str: pass + def dump(self: Pass, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Break(AST): + def __new__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Break, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Break) -> t.CInt: pass + def type_name(self: Break) -> str: pass + def dump(self: Break, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Continue(AST): + def __new__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def __init__(self: Continue, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + def kind(self: Continue) -> t.CInt: pass + def type_name(self: Continue) -> str: pass + def dump(self: Continue, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Import(AST): + names: list[AST | t.CPtr] | t.CPtr + def __new__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Import, pool: memhub.MemManager | t.CPtr, names: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Import) -> t.CInt: pass + def type_name(self: Import) -> str: pass + def dump(self: Import, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class ImportFrom(AST): + module: str + names: list[AST | t.CPtr] | t.CPtr + level: t.CInt + def __new__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def __init__(self: ImportFrom, pool: memhub.MemManager | t.CPtr, module: str, names: list[AST | t.CPtr] | t.CPtr, level: t.CInt) -> t.CInt: pass + def kind(self: ImportFrom) -> t.CInt: pass + def type_name(self: ImportFrom) -> str: pass + def dump(self: ImportFrom, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass +class Match(AST): + subject: AST | t.CPtr + cases: list[AST | t.CPtr] | t.CPtr + def __new__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def __init__(self: Match, pool: memhub.MemManager | t.CPtr, subject: AST | t.CPtr, cases: list[AST | t.CPtr] | t.CPtr) -> t.CInt: pass + def kind(self: Match) -> t.CInt: pass + def type_name(self: Match) -> str: pass + def dump(self: Match, buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: pass \ No newline at end of file diff --git a/TransPyV/temp/66de93aaeebc0601.doc.json b/TransPyV/temp/66de93aaeebc0601.doc.json new file mode 100644 index 0000000..046af31 --- /dev/null +++ b/TransPyV/temp/66de93aaeebc0601.doc.json @@ -0,0 +1 @@ +{"IfHandle": "if/elif/else 语句处理器：继承 Mixin 获得 Trans 回指针", "IfHandle.Handle": "翻译 if/elif/else 语句"} \ No newline at end of file diff --git a/TransPyV/temp/66de93aaeebc0601.pyi b/TransPyV/temp/66de93aaeebc0601.pyi new file mode 100644 index 0000000..2981cf3 --- /dev/null +++ b/TransPyV/temp/66de93aaeebc0601.pyi @@ -0,0 +1,25 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesIf.py +Module: lib.core.Handles.HandlesIf +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesBody as HandlesBody + +@t.NoVTable +class IfHandle(HandlesBase.Mixin): + def __init__(self: IfHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: IfHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewIfHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> IfHandle | t.CPtr: pass diff --git a/TransPyV/temp/6f62fe05c5ea1ceb.pyi b/TransPyV/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/TransPyV/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/TransPyV/temp/6ff26590374ae6fc.pyi b/TransPyV/temp/6ff26590374ae6fc.pyi new file mode 100644 index 0000000..1b1bba1 --- /dev/null +++ b/TransPyV/temp/6ff26590374ae6fc.pyi @@ -0,0 +1,39 @@ +""" +Auto-generated Python stub file from hashlib.__sha512.py +Module: hashlib.__sha512 +""" + + +import t, c + +SHA512_BLOCK_LEN: t.CDefine = 128 +SHA512_DIGEST_LEN: t.CDefine = 64 + +def sha512_ror(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_shr(x: t.CUInt64T, n: int) -> t.CUInt64T: pass + +def sha512_Ch(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Maj(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_BigSigma1(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma0(x: t.CUInt64T) -> t.CUInt64T: pass + +def sha512_Sigma1(x: t.CUInt64T) -> t.CUInt64T: pass + + +sha512_K: t.CExtern | t.CArray[t.CUInt64T, 80] + +@t.Object +class sha512: + state: t.CArray[t.CUInt64T, 8] + count: t.CArray[t.CUInt64T, 2] + buf: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN] + def __init__(self: sha512) -> t.CInt: pass + def transform(self: sha512, block: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN]) -> t.CInt: pass + def update(self: sha512, s: str) -> t.CInt: pass + def final(self: sha512, out: t.CArray[t.CUInt8T, SHA512_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/71e0a3ffcb3ebfad.pyi b/TransPyV/temp/71e0a3ffcb3ebfad.pyi new file mode 100644 index 0000000..574b20e --- /dev/null +++ b/TransPyV/temp/71e0a3ffcb3ebfad.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdarg.py +Module: stdarg +""" + +import c + + +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass + +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass + +def va_end(args: t.CPtr) -> t.State | t.State: pass + + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass diff --git a/TransPyV/temp/72e2d5ccb7cedcf1.pyi b/TransPyV/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/TransPyV/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/TransPyV/temp/78bbe886bbe5ca8f.doc.json b/TransPyV/temp/78bbe886bbe5ca8f.doc.json new file mode 100644 index 0000000..ffc63a6 --- /dev/null +++ b/TransPyV/temp/78bbe886bbe5ca8f.doc.json @@ -0,0 +1 @@ +{"add_imported_module": "记录已导入的模块名，返回新的 imported_modules 字符串指针", "is_module_imported": "检查模块是否已导入（词边界精确匹配，避免子串误匹配）", "add_from_import": "添加 from-import 映射 \"name:module\"，返回新的 from_imports 字符串", "lookup_from_import": "查找 from-import 名称，返回模块名或 None\n\n 支持 star import: 如果 from_imports 中有 \"*:module\" 条目，\n 且未找到精确名称匹配，且 allow_star_fallback != 0，则返回 star import 的模块名。\n ", "_resolve_relative_module": "解析相对导入为完整模块名", "compute_package_from_relpath": "从相对路径计算包名（目录部分，分隔符替换为 .）", "ImportsHandle": "导入语句处理器：继承 Mixin 获得 Trans 回指针", "ImportsHandle.HandleImport": "处理 import 语句，返回 0", "ImportsHandle.HandleImportFromModule": "处理 from X import Y 的模块部分，更新 trans._imported_modules", "ImportsHandle.HandleImportFromNames": "处理 from X import Y 的名称部分，更新 trans._from_imports"} \ No newline at end of file diff --git a/TransPyV/temp/78bbe886bbe5ca8f.pyi b/TransPyV/temp/78bbe886bbe5ca8f.pyi new file mode 100644 index 0000000..868248b --- /dev/null +++ b/TransPyV/temp/78bbe886bbe5ca8f.pyi @@ -0,0 +1,38 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesImports.py +Module: lib.core.Handles.HandlesImports +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import viperlib +import stdio +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesStruct as HandlesStruct + +def add_imported_module(pool: memhub.MemBuddy | t.CPtr, imported_modules: str, name: str) -> str: pass + +def is_module_imported(imported_modules: str, name: str) -> int: pass + +def add_from_import(pool: memhub.MemBuddy | t.CPtr, from_imports: str, local_name: str, module_name: str) -> str: pass + +def lookup_from_import(from_imports: str, name: str, allow_star_fallback: int) -> str: pass + +def _resolve_relative_module(pool: memhub.MemBuddy | t.CPtr, current_package: str, level: t.CInt, module: str) -> str: pass + +def compute_package_from_relpath(pool: memhub.MemBuddy | t.CPtr, rel_path: str) -> str: pass + + +@t.NoVTable +class ImportsHandle(HandlesBase.Mixin): + def __init__(self: ImportsHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleImport(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromModule(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + def HandleImportFromNames(self: ImportsHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewImportsHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ImportsHandle | t.CPtr: pass diff --git a/TransPyV/temp/7e260c14d2712119.doc.json b/TransPyV/temp/7e260c14d2712119.doc.json new file mode 100644 index 0000000..c90d338 --- /dev/null +++ b/TransPyV/temp/7e260c14d2712119.doc.json @@ -0,0 +1 @@ +{"SrcFileEntry": "源文件条目", "_ScanDirForPyFiles": "递归扫描目录，收集 .py 文件到 entries，返回新的 file_count", "RunMultiFileProject": "多文件项目编译\n\n Returns: 0 成功，非 0 失败\n "} \ No newline at end of file diff --git a/TransPyV/temp/7e260c14d2712119.pyi b/TransPyV/temp/7e260c14d2712119.pyi new file mode 100644 index 0000000..2398e44 --- /dev/null +++ b/TransPyV/temp/7e260c14d2712119.pyi @@ -0,0 +1,42 @@ +""" +Auto-generated Python stub file from lib.core.Phase2.py +Module: lib.core.Phase2 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import llvmlite +import subprocess +import argparse +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.BuildPipeline as BuildPipeline +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Utils as Utils +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 +MAX_SRC_FILES: t.CDefine = 64 + +@t.NoVTable +class SrcFileEntry: + Path: str + Sha1: str + +def _ScanDirForPyFiles(mb: memhub.MemBuddy | t.CPtr, dir_path: str, entries: SrcFileEntry | t.CPtr, entry_size: t.CSizeT, file_count: int, max_files: int) -> int: pass + +def RunMultiFileProject(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, output_dir: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str, includes_dir: str, do_phase1: int, do_phase2: int, log: VLogger.Logger | t.CPtr, args: argparse.ParsedArgs | t.CPtr) -> int: pass diff --git a/TransPyV/temp/7e529fe7a078cfef.pyi b/TransPyV/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/TransPyV/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/TransPyV/temp/81eb28ba912467a6.doc.json b/TransPyV/temp/81eb28ba912467a6.doc.json new file mode 100644 index 0000000..f6225ef --- /dev/null +++ b/TransPyV/temp/81eb28ba912467a6.doc.json @@ -0,0 +1 @@ +{"translate_stmt": "翻译单条语句，返回新增的变量数", "translate_global": "翻译 global 语句：记录 global 变量名", "translate_nonlocal": "翻译 nonlocal 语句：记录 nonlocal 变量名", "translate_expr_stmt": "翻译表达式语句（printf 调用等）", "pre_scan_allocas": "预扫描语句中的 AnnAssign，提前创建 alloca\n\n 返回新增的变量数\n ", "translate_break": "翻译 break 语句：跳转到循环 end 块", "translate_continue": "翻译 continue 语句：跳转到循环 cond/incr 块", "get_stmt_kind_name": "获取语句类型名"} \ No newline at end of file diff --git a/TransPyV/temp/81eb28ba912467a6.pyi b/TransPyV/temp/81eb28ba912467a6.pyi new file mode 100644 index 0000000..aa52e34 --- /dev/null +++ b/TransPyV/temp/81eb28ba912467a6.pyi @@ -0,0 +1,35 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBody.py +Module: lib.core.Handles.HandlesBody +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesFunctions as HandlesFunctions +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +def translate_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_global(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_nonlocal(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_expr_stmt(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def pre_scan_allocas(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + +def translate_break(trans: HT.Translator | t.CPtr) -> int: pass + +def translate_continue(trans: HT.Translator | t.CPtr) -> int: pass + +def get_stmt_kind_name(node: ast.AST | t.CPtr) -> str: pass diff --git a/TransPyV/temp/83ba314b4637eec9.doc.json b/TransPyV/temp/83ba314b4637eec9.doc.json new file mode 100644 index 0000000..656d90d --- /dev/null +++ b/TransPyV/temp/83ba314b4637eec9.doc.json @@ -0,0 +1 @@ +{"TypeRegistry.Register": "注册一个 TypeInfo。ti.Name 字段必须已设置。\n\n Returns:\n 1 表示成功，0 表示失败（ti 为 None 或 Name 为 None）\n ", "TypeRegistry.Lookup": "按名称查找 TypeInfo。\n\n Returns:\n 找到返回 TypeInfo*，找不到返回 None\n ", "TypeRegistry.Has": "检查类型是否已注册。", "Mixin": "所有 Handle 的非多态基类：持有 Translator 回指针 + 共享委托方法", "Mixin.InitMixin": "初始化 Mixin 字段（子类 __init__ 中调用）"} \ No newline at end of file diff --git a/TransPyV/temp/83ba314b4637eec9.pyi b/TransPyV/temp/83ba314b4637eec9.pyi new file mode 100644 index 0000000..1d05ab4 --- /dev/null +++ b/TransPyV/temp/83ba314b4637eec9.pyi @@ -0,0 +1,54 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesBase.py +Module: lib.core.Handles.HandlesBase +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import hashtable +import lib.core.Handles.HandlesTranslator as HT + +class TypeKind(t.CEnum): + Basic: t.State + Pointer: t.State + Struct: t.State + Union: t.State + Enum: t.State + Typedef: t.State + Function: t.State + Array: t.State + Void: t.State +@t.NoVTable +class TypeInfo: + Kind: int + Name: str + Size: int + Align: int + IsSigned: int + PtrCount: int + IsConst: int + IsVolatile: int +class TypeRegistry: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemManager | t.CPtr + def Register(self: TypeRegistry, ti: TypeInfo | t.CPtr) -> int: pass + def Lookup(self: TypeRegistry, name: str) -> TypeInfo | t.CPtr: pass + def Has(self: TypeRegistry, name: str) -> int: pass + +def NewTypeInfo(pool: memhub.MemManager | t.CPtr) -> TypeInfo | t.CPtr: pass + +def NewTypeRegistry(pool: memhub.MemManager | t.CPtr) -> TypeRegistry | t.CPtr: pass + +def TypeToLLVM(buf: t.CChar | t.CPtr, buf_size: t.CSizeT, ti: TypeInfo | t.CPtr) -> int: pass + + +@t.NoVTable +class Mixin: + Trans: HT.Translator | t.CPtr + def InitMixin(self: Mixin, trans: HT.Translator | t.CPtr) -> int: pass + +_mbuddy: t.CExtern | t.CVoid | t.CPtr \ No newline at end of file diff --git a/TransPyV/temp/888ea5bd6ab516ae.doc.json b/TransPyV/temp/888ea5bd6ab516ae.doc.json new file mode 100644 index 0000000..c189e56 --- /dev/null +++ b/TransPyV/temp/888ea5bd6ab516ae.doc.json @@ -0,0 +1 @@ +{"compute_sha1": "计算字符串的 SHA1，返回前 16 个十六进制字符（用于文件命名）", "CleanDir": "删除目录中所有文件（非递归），返回删除的文件数，-1 表示错误"} \ No newline at end of file diff --git a/TransPyV/temp/888ea5bd6ab516ae.pyi b/TransPyV/temp/888ea5bd6ab516ae.pyi new file mode 100644 index 0000000..e4ed12e --- /dev/null +++ b/TransPyV/temp/888ea5bd6ab516ae.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from lib.Projectrans.Utils.py +Module: lib.Projectrans.Utils +""" + + +import t, c +from stdint import * +import memhub +import string +import stdlib +import viperlib +import hashlib +import w32.win32file +import w32.win32base + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr + +def compute_sha1(pool: memhub.MemBuddy | t.CPtr, content: str) -> str: pass + +def CleanDir(dir_path: str) -> int: pass diff --git a/TransPyV/temp/8f12094c840be06a.doc.json b/TransPyV/temp/8f12094c840be06a.doc.json new file mode 100644 index 0000000..0487965 --- /dev/null +++ b/TransPyV/temp/8f12094c840be06a.doc.json @@ -0,0 +1 @@ +{"InitLib": "初始化 lib 包的全局 _mbuddy 指针，并级联注入到所有子模块。\n\n Args:\n mb: memhub.MemManager 实例指针（MemBuddy 等子类通过多态传入）\n\n Returns:\n 0 表示成功，非 0 表示失败\n "} \ No newline at end of file diff --git a/TransPyV/temp/8f12094c840be06a.pyi b/TransPyV/temp/8f12094c840be06a.pyi new file mode 100644 index 0000000..ac07294 --- /dev/null +++ b/TransPyV/temp/8f12094c840be06a.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from lib.__init__.py +Module: lib.__init__ +""" + + +import t, c +from stdint import * +import memhub +import lib.core.VLogger as VLogger + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr + +def InitLib(mb: memhub.MemManager | t.CPtr) -> int: pass diff --git a/TransPyV/temp/90c53dd6db8d41cf.pyi b/TransPyV/temp/90c53dd6db8d41cf.pyi new file mode 100644 index 0000000..98359ba --- /dev/null +++ b/TransPyV/temp/90c53dd6db8d41cf.pyi @@ -0,0 +1,20 @@ +""" +Auto-generated Python stub file from stdlib.py +Module: stdlib +""" + +import c + + +from stdint import * +import t + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass + +def free(p: t.CVoid | t.CPtr) -> t.State: pass + +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass diff --git a/TransPyV/temp/93c1d18e35d188d6.pyi b/TransPyV/temp/93c1d18e35d188d6.pyi new file mode 100644 index 0000000..a8cd09e --- /dev/null +++ b/TransPyV/temp/93c1d18e35d188d6.pyi @@ -0,0 +1,53 @@ +""" +Auto-generated Python stub file from platmacro.py +Module: platmacro +""" + + +import t, c + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 \ No newline at end of file diff --git a/TransPyV/temp/950028de389b7a7b.doc.json b/TransPyV/temp/950028de389b7a7b.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/TransPyV/temp/950028de389b7a7b.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/TransPyV/temp/950028de389b7a7b.pyi b/TransPyV/temp/950028de389b7a7b.pyi new file mode 100644 index 0000000..b7bdb15 --- /dev/null +++ b/TransPyV/temp/950028de389b7a7b.pyi @@ -0,0 +1,41 @@ +""" +Auto-generated Python stub file from main.py +Module: main +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import argparse +import w32.win32console as w32cmd +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import sys +import ast +import lib +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.BuildPipeline as BuildPipeline +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.core.Phase1 as Phase1 +import lib.core.Phase2 as Phase2 +import lib.Projectrans.Config as Config +import lib.Projectrans.Utils as Utils +import llvmlite +import subprocess +import viperlib +import hashlib + +POOL_SIZE: t.CDefine = 1073741824 +CODE_PAGE: t.CDefine = 65001 +SRC_BUF_SIZE: t.CDefine = 1048576 + +def main() -> int: pass diff --git a/TransPyV/temp/95394ca8da0f655a.pyi b/TransPyV/temp/95394ca8da0f655a.pyi new file mode 100644 index 0000000..27fd0e2 --- /dev/null +++ b/TransPyV/temp/95394ca8da0f655a.pyi @@ -0,0 +1,70 @@ +""" +Auto-generated Python stub file from llvmlite.__init__.py +Module: llvmlite.__init__ +""" + + +import t, c +from stdint import * +import memhub +import string +import viperlib +from .__types import LLVMType, ParamNode, LLVMTypeCore, Int1, Int8, Int16, Int32, Int64, Ptr, Func, Void, Array, Struct, Half, Float, Double, FP128, Label, new_param_node, param_list_append, paramnode_get_ty, paramnode_get_next, paramnode_set_ty, paramnode_set_next, TypePrint, PrintStructDefinition, get_struct_name +from .__values import Value, new_value, ConstInt, ConstNull, ConstZero, ConstFloat, SSAValue, ValuePrint, value_get_ty, value_get_next, value_set_ty, value_set_next, value_set_name, value_set_isconst +from .__function import Line, BasicBlock, Param, Function, new_line, new_basic_block, new_param, new_function, function_add_param, function_add_block, block_append_line, block_append_text, function_set_attrs, param_set_attrs, function_get_name, function_get_ret_ty, function_get_params, function_get_param_head, function_get_next, function_is_declared, param_get_name, param_get_ty, param_get_next, FunctionPrint +from .__module import LLVMModule, GlobalVariable, NamedTypeNode, new_module, new_global_variable, module_add_function, module_add_global, module_add_named_type, module_set_target, module_set_datalayout, global_set_constant, global_set_linkage, global_set_unnamed_addr, global_set_section, LLVMModulePrint, OUTPUT_FULL, OUTPUT_STUB, OUTPUT_TEXT, module_ensure_opaque_for_type +from .__builder import IRBuilder, new_builder, position_at_end, build_alloca, build_load, build_store, build_add, build_sub, build_mul, build_sdiv, build_udiv, build_srem, build_urem, build_and, build_or, build_xor, build_shl, build_lshr, build_ashr, build_fadd, build_fsub, build_fmul, build_fdiv, build_frem, build_fneg, build_icmp, build_fcmp, build_br, build_cond_br, build_ret, build_ret_void, build_call, build_call_indirect, build_bitcast, build_sext, build_trunc, build_zext, build_ptrtoint, build_inttoptr, build_fpext, build_fptrunc, build_fp2si, build_si2fp, build_fp2ui, build_ui2fp, build_gep, build_gep_array, build_gep_struct, build_phi, build_switch, build_select, build_unreachable, build_atomicrmw, build_cmpxchg, build_inline_asm, PhiIncoming, new_phi_incoming, SwitchCase, new_switch_case, ICMP_EQ, ICMP_NE, ICMP_SGT, ICMP_SGE, ICMP_SLT, ICMP_SLE, ICMP_UGT, ICMP_UGE, ICMP_ULT, ICMP_ULE, FCMP_FALSE, FCMP_OEQ, FCMP_OGT, FCMP_OGE, FCMP_OLT, FCMP_OLE, FCMP_ONE, FCMP_ORD, FCMP_UNO, FCMP_UEQ, FCMP_UGT, FCMP_UGE, FCMP_ULT, FCMP_ULE, FCMP_UNE, FCMP_TRUE, ATOMIC_XCHG, ATOMIC_ADD, ATOMIC_SUB, ATOMIC_AND, ATOMIC_NAND, ATOMIC_OR, ATOMIC_XOR, ATOMIC_MAX, ATOMIC_MIN, ATOMIC_UMAX, ATOMIC_UMIN, ATOMIC_ORDER_NOTATOMIC, ATOMIC_ORDER_UNORDERED, ATOMIC_ORDER_MONOTONIC, ATOMIC_ORDER_ACQUIRE, ATOMIC_ORDER_RELEASE, ATOMIC_ORDER_ACQ_REL, ATOMIC_ORDER_SEQ_CST +from .__verify import VerifyResult, NameEntry, verify_function, verify_module, VERIFY_OK, VERIFY_ERR_NO_TERMINATOR, VERIFY_ERR_DUPLICATE_DEF, VERIFY_ERR_UNDEFINED_USE, VERIFY_ERR_PHI_NOT_FIRST, _new_verify_result + +def _str_dup(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def create_block(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + +def const_int(pool: memhub.MemBuddy | t.CPtr, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int32(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def const_int64(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: pass + +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def add_param(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + +def create_declare(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def create_global_string(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + +def create_global_int(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + + +@t.NoVTable +class LLVMModuleCore: + Pool: memhub.MemBuddy | t.CPtr + def __init__(self: LLVMModuleCore, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def NewModule(self: LLVMModuleCore, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: pass + def CreateFunction(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def CreateDeclare(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + def AddParam(self: LLVMModuleCore, func: Function | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + def FunctionSetAttrs(self: LLVMModuleCore, func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def ParamSetAttrs(self: LLVMModuleCore, param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + def CreateBlock(self: LLVMModuleCore, func: Function | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + def NewBuilder(self: LLVMModuleCore, func: Function | t.CPtr) -> IRBuilder | t.CPtr: pass + def NewGlobal(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalString(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: pass + def CreateGlobalInt(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: pass + def GlobalSetConstant(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, is_const: t.CInt) -> t.CInt: pass + def GlobalSetLinkage(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr) -> t.CInt: pass + def GlobalSetUnnamedAddr(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, level: t.CInt) -> t.CInt: pass + def GlobalSetSection(self: LLVMModuleCore, gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr) -> t.CInt: pass + def ConstInt(self: LLVMModuleCore, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt32(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstInt64(self: LLVMModuleCore, val: t.CInt64T) -> Value | t.CPtr: pass + def ConstFloat(self: LLVMModuleCore, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: pass + def SSAValue(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def ConstNull(self: LLVMModuleCore, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: pass + def PrintValue(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr) -> t.CInt: pass + def PrintFunction(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr) -> t.CInt: pass + def PrintModule(self: LLVMModuleCore, buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr) -> t.CInt: pass + def NewVerifyResult(self: LLVMModuleCore, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + def VerifyFunction(self: LLVMModuleCore, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + def VerifyModule(self: LLVMModuleCore, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/96837bcc64032444.pyi b/TransPyV/temp/96837bcc64032444.pyi new file mode 100644 index 0000000..46e78bd --- /dev/null +++ b/TransPyV/temp/96837bcc64032444.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from hashlib.__init__.py +Module: hashlib.__init__ +""" + +import t +import c + + +from .__md5 import md5, MD5_DIGEST_LEN +from .__sha1 import sha1, SHA1_DIGEST_LEN +from .__sha256 import sha256, SHA256_DIGEST_LEN +from .__sha512 import sha512, SHA512_DIGEST_LEN \ No newline at end of file diff --git a/TransPyV/temp/_phase1_manifest.json b/TransPyV/temp/_phase1_manifest.json new file mode 100644 index 0000000..70d09be --- /dev/null +++ b/TransPyV/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\Projectrans\\Config.py": {"sha1": "49cf633017627a4a", "mtime": 1784346259.5164392, "size": 10284}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\Projectrans\\Utils.py": {"sha1": "888ea5bd6ab516ae", "mtime": 1783869136.4439158, "size": 4066}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\__init__.py": {"sha1": "8f12094c840be06a", "mtime": 1783176502.323163, "size": 752}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\BuildPipeline.py": {"sha1": "d1599ecbac5b6586", "mtime": 1784365349.9071016, "size": 18044}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesAnnAssign.py": {"sha1": "2ed380190e9df89e", "mtime": 1784355565.0405374, "size": 12796}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesAssign.py": {"sha1": "3d3edda4a4e6c08b", "mtime": 1784358988.7719295, "size": 12619}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesAugAssign.py": {"sha1": "541a090dde0cc1ba", "mtime": 1784291303.692474, "size": 10160}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesBase.py": {"sha1": "83ba314b4637eec9", "mtime": 1783864585.7948608, "size": 8746}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesBody.py": {"sha1": "81eb28ba912467a6", "mtime": 1784299229.4847765, "size": 8538}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesClassDef.py": {"sha1": "c4f2d483ec0f4db5", "mtime": 1784352269.421058, "size": 95797}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesEnum.py": {"sha1": "e5be27ae4fe7ce04", "mtime": 1783864613.5248196, "size": 8197}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesExpr.py": {"sha1": "ae0cc890f657076d", "mtime": 1784360370.033992, "size": 68807}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesExprCall.py": {"sha1": "b29b80b6106e4be7", "mtime": 1784366491.736505, "size": 190769}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesExprOps.py": {"sha1": "07a8c82a8ca2be90", "mtime": 1783920713.4815462, "size": 12706}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesFor.py": {"sha1": "48fa8cc758d3826b", "mtime": 1784360589.0914645, "size": 16438}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesFunctions.py": {"sha1": "4020cac9bab5e9f1", "mtime": 1784359369.6672971, "size": 42350}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesIf.py": {"sha1": "66de93aaeebc0601", "mtime": 1784354259.917369, "size": 5211}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesImports.py": {"sha1": "78bbe886bbe5ca8f", "mtime": 1784365186.9951043, "size": 15989}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesMain.py": {"sha1": "001c1416e7a7dc50", "mtime": 1784356333.922174, "size": 13209}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesNonlocal.py": {"sha1": "af9484a65085bf4b", "mtime": 1783859826.2073808, "size": 5336}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesReturn.py": {"sha1": "d9322f0db660edd8", "mtime": 1784309054.0918405, "size": 4530}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesStruct.py": {"sha1": "54014f6c53e9640b", "mtime": 1784300400.6293957, "size": 39050}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesTranslator.py": {"sha1": "2998b8f78d7022bf", "mtime": 1784365111.9064653, "size": 17135}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesType.py": {"sha1": "0c56128647a0b845", "mtime": 1784355136.3688684, "size": 51854}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesVar.py": {"sha1": "02885f6f134079c2", "mtime": 1784275582.3904912, "size": 13778}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Handles\\HandlesWhile.py": {"sha1": "51f0144d6e8188d9", "mtime": 1784354259.2134998, "size": 4787}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\IncludesScanner.py": {"sha1": "bd82e3ca81546c76", "mtime": 1783864358.032913, "size": 12662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Phase1.py": {"sha1": "db7c4a8f09d02a17", "mtime": 1784365236.5848596, "size": 21791}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\Phase2.py": {"sha1": "7e260c14d2712119", "mtime": 1784365393.243475, "size": 35884}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\StubMerger.py": {"sha1": "b5d05dbd7fee36ef", "mtime": 1784366194.2617724, "size": 65300}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\lib\\core\\VLogger.py": {"sha1": "21fffba3315e81b4", "mtime": 1783176437.4701056, "size": 6091}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\TransPyV\\App\\main.py": {"sha1": "950028de389b7a7b", "mtime": 1784346274.1600204, "size": 23702}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\_list.py": {"sha1": "f2ecbf8ced20575c", "mtime": 1784203829.5212443, "size": 3277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\argparse.py": {"sha1": "aba439b7882ad9d6", "mtime": 1783072446.3366802, "size": 16968}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\__init__.py": {"sha1": "34548789d646f432", "mtime": 1784102404.6307266, "size": 3409}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\astaux.py": {"sha1": "4337fb260448bbe2", "mtime": 1784006094.7981195, "size": 11489}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\base.py": {"sha1": "5dab8cb390496d22", "mtime": 1784197698.3282094, "size": 14966}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\exprs.py": {"sha1": "47767b5026a8ee15", "mtime": 1783239758.809719, "size": 40615}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\lexer.py": {"sha1": "0c212981c180e7fb", "mtime": 1783400609.908198, "size": 23535}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\match.py": {"sha1": "4dd6b3f1427d1cc5", "mtime": 1782961799.8492737, "size": 12309}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\parser.py": {"sha1": "ee52c08239684346", "mtime": 1784102405.1933146, "size": 85445}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\stmts.py": {"sha1": "657e182b27c2a022", "mtime": 1783929993.759902, "size": 37926}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\ast\\tokens.py": {"sha1": "b547ac4f380bddb6", "mtime": 1783165480.469121, "size": 13722}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\atom.py": {"sha1": "271ea3decb810db2", "mtime": 1782226548.693161, "size": 1290}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__init__.py": {"sha1": "96837bcc64032444", "mtime": 1777902641.540433, "size": 186}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__md5.py": {"sha1": "19f8024d10c828e8", "mtime": 1784356818.7075372, "size": 5847}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha1.py": {"sha1": "0df65b8ed15664b0", "mtime": 1784356827.656474, "size": 4516}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha256.py": {"sha1": "c9d54a4158f7f5a8", "mtime": 1782266203.9352357, "size": 5332}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashlib\\__sha512.py": {"sha1": "6ff26590374ae6fc", "mtime": 1782266223.3380039, "size": 7162}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\hashtable.py": {"sha1": "b8c66c8ff44eb874", "mtime": 1782826619.490043, "size": 9385}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__init__.py": {"sha1": "57288496f7c2d1ad", "mtime": 1783239811.8873823, "size": 7653}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__parser.py": {"sha1": "240a9a4157959a9f", "mtime": 1782826918.100951, "size": 7517}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\json\\__writer.py": {"sha1": "20cd49775c100a38", "mtime": 1782826901.8906734, "size": 3784}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\linkedlist.py": {"sha1": "2d39c6c7d3557b3e", "mtime": 1782895405.2191627, "size": 16032}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__init__.py": {"sha1": "95394ca8da0f655a", "mtime": 1784089002.4168732, "size": 16338}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__builder.py": {"sha1": "46ea362b72c10a9e", "mtime": 1783657006.512418, "size": 48509}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__function.py": {"sha1": "e0b84369e50c937c", "mtime": 1784275582.4008405, "size": 16232}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__module.py": {"sha1": "21a7fcfc665f75ef", "mtime": 1784216086.0677392, "size": 20058}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__types.py": {"sha1": "455ff4e74c3dbfb0", "mtime": 1784088625.3087547, "size": 17902}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__values.py": {"sha1": "05c11f22675c60cb", "mtime": 1783584462.175312, "size": 6645}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\llvmlite\\__verify.py": {"sha1": "d4117f5e00a6c3b5", "mtime": 1783250231.8851368, "size": 14657}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\memhub.py": {"sha1": "ee084e9fc6ee413a", "mtime": 1784214242.4485993, "size": 17765}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\platmacro.py": {"sha1": "93c1d18e35d188d6", "mtime": 1781237231.03366, "size": 1422}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\posix.py": {"sha1": "6503c97dde0c79c4", "mtime": 1783064077.6376991, "size": 10386}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdarg.py": {"sha1": "71e0a3ffcb3ebfad", "mtime": 1781258888.113146, "size": 277}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdlib.py": {"sha1": "90c53dd6db8d41cf", "mtime": 1783874975.3597875, "size": 375}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\string.py": {"sha1": "ab6e54ba9a669f76", "mtime": 1783933178.7264287, "size": 9922}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\subprocess.py": {"sha1": "2da636c61863c815", "mtime": 1783847277.4569364, "size": 7627}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\sys.py": {"sha1": "b5a965302cded36b", "mtime": 1783569641.07535, "size": 5509}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperio.py": {"sha1": "c9f4be41ca1cc2b4", "mtime": 1782812279.506002, "size": 1556}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\viperlib.py": {"sha1": "c3b259b4059f8668", "mtime": 1782821991.979496, "size": 56772}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\fileio.py": {"sha1": "0035c95a18d4f8e8", "mtime": 1783747022.686592, "size": 14147}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32file.py": {"sha1": "f6b51804a0ba8ff0", "mtime": 1782531158.744144, "size": 8424}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32process.py": {"sha1": "067c78e9f121dce3", "mtime": 1781200703.5409546, "size": 4802}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32sync.py": {"sha1": "06f53cc594b4ac6c", "mtime": 1782297500.2581685, "size": 4726}} \ No newline at end of file diff --git a/TransPyV/temp/_sha1_map.txt b/TransPyV/temp/_sha1_map.txt new file mode 100644 index 0000000..b7723ba --- /dev/null +++ b/TransPyV/temp/_sha1_map.txt @@ -0,0 +1,79 @@ +001c1416e7a7dc50:lib\core\Handles\HandlesMain.py +0035c95a18d4f8e8:includes/w32\fileio.py +02885f6f134079c2:lib\core\Handles\HandlesVar.py +05c11f22675c60cb:includes/llvmlite\__values.py +067c78e9f121dce3:includes/w32\win32process.py +06f53cc594b4ac6c:includes/w32\win32sync.py +07a8c82a8ca2be90:lib\core\Handles\HandlesExprOps.py +0c212981c180e7fb:includes/ast\lexer.py +0c56128647a0b845:lib\core\Handles\HandlesType.py +0df65b8ed15664b0:includes/hashlib\__sha1.py +19f8024d10c828e8:includes/hashlib\__md5.py +20cd49775c100a38:includes/json\__writer.py +21a7fcfc665f75ef:includes/llvmlite\__module.py +21fffba3315e81b4:lib\core\VLogger.py +240a9a4157959a9f:includes/json\__parser.py +271ea3decb810db2:includes/atom.py +2998b8f78d7022bf:lib\core\Handles\HandlesTranslator.py +2d39c6c7d3557b3e:includes/linkedlist.py +2da636c61863c815:includes/subprocess.py +2ed380190e9df89e:lib\core\Handles\HandlesAnnAssign.py +34548789d646f432:includes/ast\__init__.py +3d3edda4a4e6c08b:lib\core\Handles\HandlesAssign.py +4020cac9bab5e9f1:lib\core\Handles\HandlesFunctions.py +4337fb260448bbe2:includes/ast\astaux.py +455ff4e74c3dbfb0:includes/llvmlite\__types.py +46ea362b72c10a9e:includes/llvmlite\__builder.py +47767b5026a8ee15:includes/ast\exprs.py +48fa8cc758d3826b:lib\core\Handles\HandlesFor.py +49cf633017627a4a:lib\Projectrans\Config.py +4dd6b3f1427d1cc5:includes/ast\match.py +51f0144d6e8188d9:lib\core\Handles\HandlesWhile.py +54014f6c53e9640b:lib\core\Handles\HandlesStruct.py +541a090dde0cc1ba:lib\core\Handles\HandlesAugAssign.py +57288496f7c2d1ad:includes/json\__init__.py +5dab8cb390496d22:includes/ast\base.py +6503c97dde0c79c4:includes/posix.py +657e182b27c2a022:includes/ast\stmts.py +66de93aaeebc0601:lib\core\Handles\HandlesIf.py +6f62fe05c5ea1ceb:includes/stdio.py +6ff26590374ae6fc:includes/hashlib\__sha512.py +71e0a3ffcb3ebfad:includes/stdarg.py +78bbe886bbe5ca8f:lib\core\Handles\HandlesImports.py +7e260c14d2712119:lib\core\Phase2.py +7e529fe7a078cfef:includes/w32\win32base.py +81eb28ba912467a6:lib\core\Handles\HandlesBody.py +83ba314b4637eec9:lib\core\Handles\HandlesBase.py +888ea5bd6ab516ae:lib\Projectrans\Utils.py +8f12094c840be06a:lib\__init__.py +90c53dd6db8d41cf:includes/stdlib.py +93c1d18e35d188d6:includes/platmacro.py +950028de389b7a7b:main.py +95394ca8da0f655a:includes/llvmlite\__init__.py +96837bcc64032444:includes/hashlib\__init__.py +ab6e54ba9a669f76:includes/string.py +aba439b7882ad9d6:includes/argparse.py +ae0cc890f657076d:lib\core\Handles\HandlesExpr.py +af9484a65085bf4b:lib\core\Handles\HandlesNonlocal.py +b29b80b6106e4be7:lib\core\Handles\HandlesExprCall.py +b547ac4f380bddb6:includes/ast\tokens.py +b5a965302cded36b:includes/sys.py +b5d05dbd7fee36ef:lib\core\StubMerger.py +b8c66c8ff44eb874:includes/hashtable.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +bd82e3ca81546c76:lib\core\IncludesScanner.py +c3b259b4059f8668:includes/viperlib.py +c4f2d483ec0f4db5:lib\core\Handles\HandlesClassDef.py +c9d54a4158f7f5a8:includes/hashlib\__sha256.py +c9f4be41ca1cc2b4:includes/viperio.py +d1599ecbac5b6586:lib\core\BuildPipeline.py +d4117f5e00a6c3b5:includes/llvmlite\__verify.py +d9322f0db660edd8:lib\core\Handles\HandlesReturn.py +db7c4a8f09d02a17:lib\core\Phase1.py +e0b84369e50c937c:includes/llvmlite\__function.py +e5be27ae4fe7ce04:lib\core\Handles\HandlesEnum.py +ee084e9fc6ee413a:includes/memhub.py +ee52c08239684346:includes/ast\parser.py +f2ecbf8ced20575c:includes/_list.py +f5522571bcce7bcb:includes/stdint.py +f6b51804a0ba8ff0:includes/w32\win32file.py diff --git a/TransPyV/temp/_shared_sym.json b/TransPyV/temp/_shared_sym.json new file mode 100644 index 0000000..5ecd858 Binary files /dev/null and b/TransPyV/temp/_shared_sym.json differ diff --git a/TransPyV/temp/ab6e54ba9a669f76.pyi b/TransPyV/temp/ab6e54ba9a669f76.pyi new file mode 100644 index 0000000..4fc3440 --- /dev/null +++ b/TransPyV/temp/ab6e54ba9a669f76.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from string.py +Module: string +""" + + +from stdint import * +import t, c + +def strcpy(dest: str, src: str) -> str: pass + +def strcat(dest: str, src: str) -> str: pass + +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: pass + +def strlen(src: str) -> t.CSizeT | t.CExport: pass + +def strcmp(str1: str, str2: str) -> t.CInt: pass + +def samestr(str1: str, str2: str) -> bool: pass + +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: pass + +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: pass + +def strchr(s: str, cr: t.CInt) -> str: pass + +def strrchr(s: str, cr: t.CInt) -> str: pass + +def strstr(s: str, needle: str) -> str: pass + +def strspn(s: str, skip: str) -> int: pass + +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: pass + +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: pass + +def atoi(src: str) -> t.CInt: pass + +def atoll(src: str) -> t.CInt64T: pass + +def atof(src: str) -> t.CDouble: pass + +def split(s: str, delim: str, result: t.CArray[str]) -> int: pass diff --git a/TransPyV/temp/aba439b7882ad9d6.pyi b/TransPyV/temp/aba439b7882ad9d6.pyi new file mode 100644 index 0000000..124ce31 --- /dev/null +++ b/TransPyV/temp/aba439b7882ad9d6.pyi @@ -0,0 +1,67 @@ +""" +Auto-generated Python stub file from argparse.py +Module: argparse +""" + + +import t, c +from stdint import * +import string +import memhub +import stdio +import hashtable + +STRING: t.CDefine = 0 +INT: t.CDefine = 1 +BOOL: t.CDefine = 2 +STORE: t.CDefine = 0 +STORE_TRUE: t.CDefine = 1 +STORE_FALSE: t.CDefine = 2 +COUNT: t.CDefine = 3 +MAX_ARGS: t.CDefine = 32 +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +class Argument: + name: str + short_name: str + help_text: str + arg_type: INT + action: INT + default_int: INT + default_str: str + required: bool + is_positional: bool + dest: str +class ParsedArgs: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ParsedArgs, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def get_str(self: ParsedArgs, name: str) -> str: pass + def get_int(self: ParsedArgs, name: str) -> INT: pass + def get_bool(self: ParsedArgs, name: str) -> INT: pass + def has(self: ParsedArgs, name: str) -> bool: pass + def _set_str(self: ParsedArgs, name: str, val: str) -> t.CInt: pass + def _set_int(self: ParsedArgs, name: str, val: INT) -> t.CInt: pass + def _set_bool(self: ParsedArgs, name: str, val: bool) -> t.CInt: pass +class ArgumentParser: + _prog: str + _description: str + _args: Argument | t.CPtr + _arg_count: INT + __mbuddy__: memhub.MemBuddy | t.CPtr + def __new__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: ArgumentParser, prog: str, description: str, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def add_argument(self: ArgumentParser, name: str, short: str, arg_type: INT, default: INT, default_str: str, required: bool, action: INT, help: str) -> t.CInt: pass + def _get_arg(self: ArgumentParser, idx: INT) -> Argument | t.CPtr: pass + def _find_long(self: ArgumentParser, token: str) -> INT: pass + def _find_long_prefix(self: ArgumentParser, token: str) -> INT: pass + def _find_short(self: ArgumentParser, token: str) -> INT: pass + def _find_positional(self: ArgumentParser, pos_idx: INT) -> INT: pass + def _store_value(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr, value: str) -> t.CInt: pass + def _apply_action(self: ArgumentParser, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr) -> t.CInt: pass + def parse_args(self: ArgumentParser, argc: INT, argv: str | t.CPtr) -> ParsedArgs | t.CPtr: pass + def print_help(self: ArgumentParser) -> t.CInt: pass + def print_usage(self: ArgumentParser) -> t.CInt: pass + +def release(args: ParsedArgs | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/ae0cc890f657076d.doc.json b/TransPyV/temp/ae0cc890f657076d.doc.json new file mode 100644 index 0000000..0d6dd61 --- /dev/null +++ b/TransPyV/temp/ae0cc890f657076d.doc.json @@ -0,0 +1 @@ +{"get_func_name": "从函数节点提取函数名", "escape_llvm_string": "将字符串转换为 LLVM IR 字符串字面量 c\"...\\00\" ", "get_llvm_type_bits": "获取 LLVM 整数类型的位宽，非整数类型返回 0", "get_llvm_float_bits": "获取 LLVM 浮点类型的位宽，非浮点类型返回 0", "is_ptr_type": "检查 ty 是否是 Ptr 类型，返回 1=是, 0=否", "_deref_if_ptr_ptr": "如果 obj_ptr 是 Ptr(Ptr(...))，load 解引用获取内层指针", "coerce_to_type": "将 val 强制转换为 target_ty\n\n 整数: sext（扩展）/ trunc（截断）\n 浮点: fpext（扩展）/ fptrunc（截断）\n int→float: sitofp\n float→int: fptosi\n ", "create_global_string": "创建全局字符串常量并返回 i8* bitcast", "translate_constant": "翻译常量（int/str/bool）", "_infer_elem_type_from_value": "从 LLVM Value 的类型推断 list 元素类型名", "_find_pool_var": "查找上下文中的 pool 变量，返回 alloca 或 None", "translate_list_literal": "翻译 list 字面量 [a, b, c] → list[T](pool) + append 序列", "translate_value": "翻译值表达式为 LLVM Value", "translate_compare": "翻译比较表达式，返回 i1 结果", "translate_unaryop": "翻译一元运算（-x, +x, not x, ~x）", "translate_boolop": "翻译布尔运算（and/or）的短路求值，返回 i1", "translate_name_value": "翻译变量引用（Name 节点）→ load", "ExprHandle": "表达式处理器：继承 Mixin 获得 Trans 回指针", "ExprHandle.HandleValue": "翻译值表达式，从 self.Trans 获取共享状态", "list_getitem_inline": "内联生成 list __getitem__ 逻辑，返回元素地址 (i8*)", "_list_elem_type_from_name": "从 list 类型名提取元素 LLVM 类型", "list_setitem_inline": "内联生成 list __setitem__ 逻辑，返回 0", "is_list_subscript": "检查 Subscript 节点是否是 list[T] 类型，返回 list 对象指针或 None", "translate_subscript": "翻译下标表达式，返回加载的元素值", "find_global_in_module": "在模块全局变量链表中按名称查找全局变量", "make_global_ref": "创建全局变量引用 (类型为 ty*，名称为 @name)", "_resolve_module_attribute_global": "解析模块属性访问，返回全局变量引用 (用于跨模块 global 访问)", "translate_attribute": "翻译属性访问，返回加载的字段值", "get_subscript_ptr": "获取下标表达式的元素指针（用于赋值 lhs）", "get_attribute_ptr": "获取属性访问的字段指针（用于赋值 lhs）"} \ No newline at end of file diff --git a/TransPyV/temp/ae0cc890f657076d.pyi b/TransPyV/temp/ae0cc890f657076d.pyi new file mode 100644 index 0000000..2cb20be --- /dev/null +++ b/TransPyV/temp/ae0cc890f657076d.pyi @@ -0,0 +1,88 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExpr.py +Module: lib.core.Handles.HandlesExpr +""" + + +import t, c +from stdint import * +import ast +import memhub +import string +import llvmlite +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesExprOps as HandlesExprOps +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesNonlocal as HandlesNonlocal +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum +import lib.core.Handles.HandlesClassDef as HandlesClassDef +import lib.core.Handles.HandlesImports as HandlesImports + +def get_func_name(func_node: ast.AST | t.CPtr) -> str: pass + +def escape_llvm_string(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def get_llvm_type_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def get_llvm_float_bits(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def is_ptr_type(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _deref_if_ptr_ptr(builder: llvmlite.IRBuilder | t.CPtr, obj_ptr: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def coerce_to_type(builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def create_global_string(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, str_val: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_constant(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_elem_type_from_value(pool: memhub.MemBuddy | t.CPtr, value: llvmlite.Value | t.CPtr) -> str: pass + +def _find_pool_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_list_literal(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, elem_type_name: str, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_compare(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_unaryop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_boolop(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_name_value(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +@t.NoVTable +class ExprHandle(HandlesBase.Mixin): + def __init__(self: ExprHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleValue(self: ExprHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def NewExprHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprHandle | t.CPtr: pass + +def list_getitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _list_elem_type_from_name(pool: memhub.MemBuddy | t.CPtr, struct_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def list_setitem_inline(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, lm_obj: llvmlite.Value | t.CPtr, idx_val: llvmlite.Value | t.CPtr, rhs_val: llvmlite.Value | t.CPtr) -> int: pass + +def is_list_subscript(node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_subscript(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def find_global_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.GlobalVariable | t.CPtr: pass + +def make_global_ref(pool: memhub.MemBuddy | t.CPtr, name: str, ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _resolve_module_attribute_global(pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, trans: HT.Translator | t.CPtr, at: ast.Attribute | t.CPtr, ty_hint: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_attribute(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_subscript_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def get_attribute_ptr(builder: llvmlite.IRBuilder | t.CPtr, pool: memhub.MemBuddy | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass diff --git a/TransPyV/temp/af9484a65085bf4b.doc.json b/TransPyV/temp/af9484a65085bf4b.doc.json new file mode 100644 index 0000000..2fbc2fa --- /dev/null +++ b/TransPyV/temp/af9484a65085bf4b.doc.json @@ -0,0 +1 @@ +{"get_env_ptr_var": "获取或创建 _env_ptr 变量（存储闭包 env 指针）", "load_nonlocal_var": "从闭包 env 加载 nonlocal 变量值（env 直接存储 i32 值）", "get_nonlocal_var_ptr": "获取 nonlocal 变量在 env 中的地址（i32*），用于 store 操作"} \ No newline at end of file diff --git a/TransPyV/temp/af9484a65085bf4b.pyi b/TransPyV/temp/af9484a65085bf4b.pyi new file mode 100644 index 0000000..3d49af5 --- /dev/null +++ b/TransPyV/temp/af9484a65085bf4b.pyi @@ -0,0 +1,21 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesNonlocal.py +Module: lib.core.Handles.HandlesNonlocal +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesVar as HandlesVar + +def get_env_ptr_var(trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def load_nonlocal_var(trans: HT.Translator | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass + +def get_nonlocal_var_ptr(trans: HT.Translator | t.CPtr, name: str) -> llvmlite.Value | t.CPtr: pass diff --git a/TransPyV/temp/b29b80b6106e4be7.doc.json b/TransPyV/temp/b29b80b6106e4be7.doc.json new file mode 100644 index 0000000..724daa6 --- /dev/null +++ b/TransPyV/temp/b29b80b6106e4be7.doc.json @@ -0,0 +1 @@ +{"set_module_sha1_map": "设置全局模块 SHA1 映射", "_lookup_module_sha1": "查找模块名对应的 SHA1，找不到返回 None", "_lookup_module_sha1_suffix": "后缀匹配查找模块 SHA1，如 mod_suffix=\"__types\" 匹配 \"llvmlite.__types\"\n\n 用于 from .X import Y 的相对导入：impf.module 是 \".X\"（带前导点），\n 去掉前导点后为 \"X\"，SHA1 映射表中键是 \"pkg.X\"，需要后缀匹配。\n ", "register_cexport_func": "注册 CExport 函数到全局表（sha1+func_name 二元组）", "is_cexport_func": "检查 (sha1, func_name) 是否是 CExport 函数", "_lookup_mod_sha1": "从 from_imports 查找类所属模块的 SHA1，返回 None=未找到\n\n allow_star_fallback: 传 0 禁用 star import 回退（用于明确模块名/函数名查找）\n ", "find_func_in_module": "在模块的函数链表中按名称查找函数\n\n 查找策略：\n 1. 精确匹配（如 \"strlen\" 匹配 @strlen，有 t.CExport 的函数）\n 2. 后缀匹配（如 \"atoi\" 匹配 @\"83210a72e829a086.atoi\"，无 t.CExport 的函数）\n\n 使用 llvmlite.function_get_name/next 访问器绕过 stub 类型限制\n （stub 类型字段不足且字段类型可能错误，直接字段访问会被静默跳过）。\n ", "_infer_external_func_ret_ty": "根据函数名推断外部 includes 函数的返回类型", "_get_external_func_param_ty": "获取外部函数第 param_idx 个参数的期望类型，None 表示用字面量原类型\n\n 对 Win32 API 函数的所有非指针参数返回 i64，匹配 declare 中\n ULONG/SIZE_T 被统一翻译为 i64 的情况。\n ", "_emit_llvm_memcpy_intrinsic": "将 memcpy 调用映射到 @llvm.memcpy 内联函数\n\n 直接接收三个 Value 参数，避免通过 Next 链表遍历（共享 alloca 对象的\n Next 字段可能被其他操作污染，导致链表断裂）。\n ", "translate_c_deref": "c.Deref(ptr) → load i8 from ptr", "translate_c_derefas": "c.DerefAs(ptr, val) → store val to ptr（bitcast 后 store）", "translate_c_addr": "c.Addr(var) → 返回变量的 alloca 指针（不 load）", "translate_c_load": "c.Load(a, b) → *a = *b（加载 b 的值，存储到 a）", "AsmOperand": "内联汇编操作数条目", "_asm_append_cstr": "将 C 字符串 src 追加到 dst 末尾", "_asm_escape_for_ir": "转义汇编文本用于 LLVM IR 字符串字面量", "_resolve_register": "将 t.ASM_DESCR.XXX 的属性名映射到约束字符串值", "_resolve_asm_descr": "递归解析约束表达式，返回约束字符串", "_asm_get_operand": "获取操作数数组中第 idx 个元素的指针", "_asm_is_c_call": "检查 Call 节点是否是 c.XXX 调用，返回属性名或 None", "_asm_add_operand": "添加操作数到数组，返回新计数（失败返回原计数）", "translate_c_asm": "翻译 c.Asm 内联汇编，返回结果 Value（无输出时返回 None）", "_llvmir_get_op": "获取第 i 个操作数槽位", "_llvmir_find_op_marker": "查找 %__OP 标记，返回标记开始位置或 None", "_llvmir_parse_op_num": "从 %__OPN__ 标记中解析编号 N", "_llvmir_match_icmp_pred": "将 icmp 预测字符串转为常量", "_llvmir_match_fcmp_pred": "将 fcmp 预测字符串转为常量", "_llvmir_extract_two_ops": "从模板中提取前两个 %__OP 标记的编号\n\n 结果写入 out_nums[0] 和 out_nums[1]\n 返回 1 成功, 0 失败\n ", "_llvmir_resolve_op": "根据操作数编号解析实际 Value", "_llvmir_store_outputs": "将结果存储到所有输出变量", "translate_c_llvmir": "翻译 c.LLVMIR 内联 LLVM IR，返回结果 Value", "_apply_struct_defaults": "应用结构体字段的默认值", "_translate_struct_ctor": "结构体构造函数：按字段顺序 store 参数值，返回结构体值\n\n OOP 结构体：alloca → call __before_init__(ptr) → call __init__(ptr, args) 或 store 参数 → load\n 普通结构体：alloca → zero + defaults → store 参数 → load\n ", "_do_virtual_call": "通过虚表间接调用虚方法，返回 call 结果（void 方法返回 None）", "_call_method_on_ptr": "在结构体指针上调用方法，返回 call 结果（void 方法返回 None）", "_infer_method_ret_ty": "根据方法名推断返回类型（stub 未注入时的回退方案）", "_translate_method_call": "翻译方法调用，返回 call 结果 Value", "_kwarg_name_matches": "检查关键字参数名 kw_arg 是否匹配参数名 param_name（跳过 % 前缀）", "_translate_struct_ctor_kw": "结构体构造函数（关键字参数版本）", "_translate_call_with_kwargs": "带关键字参数的函数调用，按参数顺序构建调用链表\n\n 使用 llvmlite 访问器函数绕过 stub 类型限制：\n found_func.Params/RetTy/Name 和 cur_param.Name/Ty/Next 在 stub\n 类型中字段不足，直接访问会被 TransPyC 静默跳过。\n ", "_translate_t_type_cast": "类型转换：根据参数类型和目标类型选择转换方式", "_extract_type_args_from_slice": "从 Subscript.slice 提取类型实参名列表\n\n list[int] -> [\"int\"]\n dict[str, int] -> [\"str\", \"int\"]\n list[AST | t.CPtr] -> [\"AST|t.CPtr\"]\n ", "translate_call": "翻译函数调用，返回 call 结果 Value", "translate_closure_call": "通过闭包指针间接调用", "translate_printf_call": "翻译 printf 调用（特殊处理格式字符串）", "FuncEntry": "函数表条目", "init_func_table": "分配并清零函数表数组", "find_func_in_table": "在函数表中按名称查找，返回 Function 指针或 None", "add_func_to_table": "添加函数到函数表，返回 0 成功", "_init_global_func_table": "初始化全局函数表（用 stdlib.malloc 分配，生命周期=整个程序）", "_register_global_func": "注册函数到全局表（已存在则跳过），返回 0 成功\n\n 直接操作全局表，不调用 add_func_to_table，避免递归。\n ", "find_func_global": "在全局函数表中按名称查找，返回 Function 指针或 None", "ExprCallHandle": "函数调用处理器：继承 Mixin 获得 Trans 回指针", "ExprCallHandle.HandleCall": "翻译函数调用，从 self.Trans 获取共享状态", "ExprCallHandle.HandlePrintfCall": "翻译 printf 调用，从 self.Trans 获取共享状态"} \ No newline at end of file diff --git a/TransPyV/temp/b29b80b6106e4be7.pyi b/TransPyV/temp/b29b80b6106e4be7.pyi new file mode 100644 index 0000000..065139b --- /dev/null +++ b/TransPyV/temp/b29b80b6106e4be7.pyi @@ -0,0 +1,173 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesExprCall.py +Module: lib.core.Handles.HandlesExprCall +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import stdlib +import sys +import viperlib +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesClassDef as HandlesClassDef + +_g_sha1_arr: t.CExtern | bytes +_g_mod_arr: t.CExtern | bytes +_g_sha1_count: t.CExtern | int + +def set_module_sha1_map(sha1_arr: bytes, mod_arr: bytes, count: int) -> int: pass + +def _lookup_module_sha1(mod_name: str) -> str: pass + +def _lookup_module_sha1_suffix(mod_suffix: str) -> str: pass + + +_g_cexport_sha1_arr: t.CExtern | bytes +_g_cexport_func_arr: t.CExtern | bytes +_g_cexport_count: t.CExtern | int +_CEXPORT_MAX: t.CDefine = 256 + +def register_cexport_func(sha1: str, func_name: str) -> int: pass + +def is_cexport_func(sha1: str, func_name: str) -> int: pass + +def _lookup_mod_sha1(pool: memhub.MemBuddy | t.CPtr, from_imports: str, class_name: str, allow_star_fallback: int) -> str: pass + +def find_func_in_module(mod: llvmlite.LLVMModule | t.CPtr, name: str) -> llvmlite.Function | t.CPtr: pass + +def _infer_external_func_ret_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _get_external_func_param_ty(pool: memhub.MemBuddy | t.CPtr, func_name: str, param_idx: int) -> llvmlite.LLVMType | t.CPtr: pass + +def _emit_llvm_memcpy_intrinsic(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, dst_val: llvmlite.Value | t.CPtr, src_val: llvmlite.Value | t.CPtr, num_val: llvmlite.Value | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_deref(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_derefas(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_addr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_c_load(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +ASM_MAX_OPERANDS: t.CDefine = 16 +ASM_TEXT_BUF_SIZE: t.CDefine = 2048 +ASM_CONST_BUF_SIZE: t.CDefine = 512 +ASM_ARGS_BUF_SIZE: t.CDefine = 512 +ASM_LINE_BUF_SIZE: t.CDefine = 4096 + +@t.NoVTable +class AsmOperand: + Value: llvmlite.Value | t.CPtr + Constraint: str + IsOutput: int + +def _asm_append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _asm_escape_for_ir(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _resolve_register(attr_name: str) -> str: pass + +def _resolve_asm_descr(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr) -> str: pass + +def _asm_get_operand(base: AsmOperand | t.CPtr, idx: int) -> AsmOperand | t.CPtr: pass + +def _asm_is_c_call(node: ast.Call | t.CPtr) -> str: pass + +def _asm_add_operand(operands: AsmOperand | t.CPtr, count: int, val: llvmlite.Value | t.CPtr, constraint: str, is_output: int) -> int: pass + +def translate_c_asm(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + + +LLVMIR_MAX_OPS: t.CDefine = 16 +LLVMIR_TEMPLATE_BUF: t.CDefine = 512 + +def _llvmir_get_op(buf: AsmOperand | t.CPtr, i: int) -> AsmOperand | t.CPtr: pass + +def _llvmir_find_op_marker(s: t.CChar | t.CPtr) -> t.CPtr: pass + +def _llvmir_parse_op_num(marker: t.CChar | t.CPtr) -> int: pass + +def _llvmir_match_icmp_pred(pred: str) -> int: pass + +def _llvmir_match_fcmp_pred(pred: str) -> int: pass + +def _llvmir_extract_two_ops(template: t.CChar | t.CPtr, out_nums: int | t.CPtr) -> int: pass + +def _llvmir_resolve_op(op_num: int, all_kinds: int | t.CPtr, all_indices: int | t.CPtr, input_ops: AsmOperand | t.CPtr, output_targets: AsmOperand | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _llvmir_store_outputs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, result: llvmlite.Value | t.CPtr, output_targets: AsmOperand | t.CPtr, output_count: int) -> llvmlite.Value | t.CPtr: pass + +def translate_c_llvmir(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _apply_struct_defaults(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, tmp: llvmlite.Value | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + +def _translate_struct_ctor(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _do_virtual_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, args_head: llvmlite.Value | t.CPtr, args_count: int, found_func: llvmlite.Function | t.CPtr, ret_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _call_method_on_ptr(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, self_ptr: llvmlite.Value | t.CPtr, extra_args: t.CSizeT | t.CPtr, extra_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _infer_method_ret_ty(pool: memhub.MemBuddy | t.CPtr, method_name: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _translate_method_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, class_name: str, method_name: str, obj_ptr: llvmlite.Value | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _kwarg_name_matches(kw_arg: str, param_name: t.CChar | t.CPtr) -> int: pass + +def _translate_struct_ctor_kw(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_call_with_kwargs(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, func_name: str, found_func: llvmlite.Function | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT, keywords: list[ast.AST | t.CPtr] | t.CPtr, kw_count: t.CSizeT, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _translate_t_type_cast(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, val: llvmlite.Value | t.CPtr, target_ty: llvmlite.LLVMType | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def _extract_type_args_from_slice(pool: memhub.MemBuddy | t.CPtr, slice_node: ast.AST | t.CPtr) -> list[str] | t.CPtr: pass + +def translate_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, node: ast.AST | t.CPtr, funcs_ptr: t.CPtr, func_count: int, trans: HT.Translator | t.CPtr) -> llvmlite.Value | t.CPtr: pass + +def translate_closure_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, closure_alloca: llvmlite.Value | t.CPtr, cargs: list[ast.AST | t.CPtr] | t.CPtr, can: t.CSizeT) -> llvmlite.Value | t.CPtr: pass + +def translate_printf_call(pool: memhub.MemBuddy | t.CPtr, builder: llvmlite.IRBuilder | t.CPtr, mod: llvmlite.LLVMModule | t.CPtr, cl: ast.Call | t.CPtr, trans: HT.Translator | t.CPtr) -> t.CInt: pass + + +@t.NoVTable +class FuncEntry: + Name: str + Func: llvmlite.Function | t.CPtr + +def init_func_table(pool: memhub.MemBuddy | t.CPtr, max_count: int) -> FuncEntry | t.CPtr: pass + +def find_func_in_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str) -> llvmlite.Function | t.CPtr: pass + +def add_func_to_table(funcs: FuncEntry | t.CPtr, func_count: int, name: str, func: llvmlite.Function | t.CPtr, max_count: int) -> int: pass + + +GLOBAL_FUNC_MAX: t.CDefine = 8192 +_global_funcs: t.CExtern | FuncEntry | t.CPtr +_global_func_count: t.CExtern | int + +def _init_global_func_table() -> int: pass + +def _register_global_func(name: str, func: llvmlite.Function | t.CPtr) -> int: pass + +def find_func_global(name: str) -> llvmlite.Function | t.CPtr: pass + + +@t.NoVTable +class ExprCallHandle(HandlesBase.Mixin): + def __init__(self: ExprCallHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def HandleCall(self: ExprCallHandle, node: ast.AST | t.CPtr) -> llvmlite.Value | t.CPtr: pass + def HandlePrintfCall(self: ExprCallHandle, cl: ast.Call | t.CPtr) -> t.CInt: pass + +def NewExprCallHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ExprCallHandle | t.CPtr: pass diff --git a/TransPyV/temp/b547ac4f380bddb6.pyi b/TransPyV/temp/b547ac4f380bddb6.pyi new file mode 100644 index 0000000..e960341 --- /dev/null +++ b/TransPyV/temp/b547ac4f380bddb6.pyi @@ -0,0 +1,166 @@ +""" +Auto-generated Python stub file from ast.tokens.py +Module: ast.tokens +""" + + +import t, c +from stdint import * +import memhub +import string + +class TokenType(t.CEnum): + EndMarker: t.State + NewLine: t.State + Indent: t.State + Dedent: t.State + Name: t.State + Number: t.State + String: t.State + FStringStart: t.State + FStringMiddle: t.State + FStringEnd: t.State + Op: t.State + Nl: t.State + Comment: t.State + Encoding: t.State +class Keyword(t.CEnum): + NotKeyword: t.State + False_: t.State + None_: t.State + True_: t.State + And: t.State + As: t.State + Assert: t.State + Async: t.State + Await: t.State + Break: t.State + Class: t.State + Continue: t.State + Def: t.State + Del: t.State + Elif: t.State + Else: t.State + Except: t.State + Finally: t.State + For: t.State + From: t.State + Global: t.State + If: t.State + Import: t.State + In: t.State + Is: t.State + Lambda: t.State + Nonlocal: t.State + Not: t.State + Or: t.State + Pass: t.State + Raise: t.State + Return: t.State + Try: t.State + While: t.State + With: t.State + Yield: t.State + Match: t.State + Case: t.State + Type: t.State +class TokOp(t.CEnum): + LPar: t.State + RPar: t.State + Lsqb: t.State + Rsqb: t.State + LBrace: t.State + RBrace: t.State + Comma: t.State + Colon: t.State + Dot: t.State + Semi: t.State + At: t.State + Equal: t.State + RArrow: t.State + PlusEq: t.State + MinusEq: t.State + StarEq: t.State + SlashEq: t.State + DSlashEq: t.State + PercentEq: t.State + AtEq: t.State + AmpEq: t.State + VBarEq: t.State + CaretEq: t.State + GtGtEq: t.State + LtLtEq: t.State + StarEqEq: t.State + DSlash: t.State + StarStar: t.State + LtLt: t.State + GtGt: t.State + LessEq: t.State + GreaterEq: t.State + EqEq: t.State + ExclaimEq: t.State + Less: t.State + Greater: t.State + Plus: t.State + Minus: t.State + Star: t.State + Slash: t.State + Percent: t.State + Amp: t.State + VBar: t.State + Caret: t.State + Tilde: t.State + ColonEq: t.State + Ellipsis: t.State + Bang: t.State +class Token: + type: t.CInt + op_subtype: t.CInt + kw_subtype: t.CInt + str_val: str + int_val: t.CInt64T + float_val: t.CDouble + is_float: t.CInt + is_complex: t.CInt + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + next: Token | t.CPtr + def __new__(self: Token, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass +class KwEntry: + name: str + kw_id: t.CInt + next: KwEntry | t.CPtr + def __new__(self: KwEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_kw_head: t.CExtern | KwEntry | t.CPtr +_kw_pool: t.CExtern | memhub.MemManager | t.CPtr + +def _kw_intern(pool: memhub.MemManager | t.CPtr, name: str, kw_id: t.CInt) -> t.CInt: pass + +def _kw_lookup(name: str) -> t.CInt: pass + +def _init_keywords(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + + +class OpEntry: + name: str + op_id: t.CInt + length: t.CInt + next: OpEntry | t.CPtr + def __new__(self: OpEntry, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +_op_head: t.CExtern | OpEntry | t.CPtr + +def _op_intern(pool: memhub.MemManager | t.CPtr, name: str, op_id: t.CInt) -> t.CInt: pass + +def _init_operators(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _init_tables(pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_token(pool: memhub.MemManager | t.CPtr, ttype: t.CInt, lineno: t.CInt, col_offset: t.CInt) -> Token | t.CPtr: pass + +def token_set_str(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, src: str, start: t.CSizeT, length: t.CSizeT) -> t.CInt: pass + +def token_set_str_literal(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, text: str) -> t.CInt: pass diff --git a/TransPyV/temp/b5a965302cded36b.pyi b/TransPyV/temp/b5a965302cded36b.pyi new file mode 100644 index 0000000..bfdb955 --- /dev/null +++ b/TransPyV/temp/b5a965302cded36b.pyi @@ -0,0 +1,44 @@ +""" +Auto-generated Python stub file from sys.py +Module: sys +""" + + +import t, c +from stdint import * +import platmacro +import memhub +import string +import stdio +import w32.win32base +import w32.win32process +import posix + +_mbuddy: t.CExtern | memhub.MemBuddy | t.CPtr + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +def exit(status: INT) -> VOID | t.CExtern | t.CExport: pass + + +maxsize: t.CDefine = 9223372036854775807 + +def platform() -> str: pass + +def get_pid() -> int: pass + + +_argc: t.CExtern | INT +_argv: t.CExtern | str | t.CPtr + +def _init_argv() -> t.CInt: pass + +def _init_argv_win32() -> t.CInt: pass + +def _init_argv_posix() -> t.CInt: pass + +def argc() -> int: pass + +def argv(index: int) -> str: pass diff --git a/TransPyV/temp/b5d05dbd7fee36ef.doc.json b/TransPyV/temp/b5d05dbd7fee36ef.doc.json new file mode 100644 index 0000000..86441ed --- /dev/null +++ b/TransPyV/temp/b5d05dbd7fee36ef.doc.json @@ -0,0 +1 @@ +{"_load_includes_sha1_set": "读取 _sha1_map.txt，将 includes/ 开头的 SHA1 写入 sha1_set\n sha1_set 大小为 MAX_INCLUDES_SHA1 * 17（每个 SHA1 16字符+null）\n 返回找到的 includes SHA1 数量，-1 表示错误", "_is_in_sha1_set": "检查 sha1 是否在集合中", "WriteIncludesSha1Map": "将 includes 扫描结果写入 _sha1_map.txt（供 StubMerger 使用）\n\n 若 filter_set 不为 None 且 filter_count > 0，只写入 filter_set 中的条目。\n ", "_IsFuncDeclaredOrDefined": "检查 out_buf 中是否已有函数的 declare 或 define 行", "_CommentOutDeclareInBuf": "在 out_buf 中搜索 declare 行（包含 func_sig），将其注释掉", "_CommentOutGlobalInBuf": "在 out_buf 中搜索全局变量定义行（以 global_name 开头），将行首改为 ';'", "_CommentOutOpaqueTypeInBuf": "在 out_buf 中搜索 type_name_prefix 对应的 opaque 声明行，将行首改为 ';'\n\n type_name_prefix 格式: '%\"sha1.ClassName\"'（不含 '=' 和后续内容）\n 返回 1=已注释, 0=未找到\n ", "_HasFullTypeDefinition": "检查 out_buf 中是否已有 type_name_prefix 的完整定义", "_LoadAndAppendStub": "加载依赖 stub 并追加（跳过 header，去重）", "_LoadAndAppendTextDeclares": "从依赖 text.ll 提取 declare（define 转 declare），追加到 out_buf", "_PathToModuleName": "将文件路径转换为模块名\n\n 特殊处理 __init__.py：includes/ast/__init__.py → ast（而非 ast.__init__）\n ", "_BuildIncludesSha1Map": "从 _sha1_map.txt 构建 sha1→module_name 映射，返回条目数", "_FindSha1ByModName": "从模块名查找 SHA1（精确匹配）", "_AddSha1ToSet": "将 SHA1 加入集合，返回新 count", "_BuildReachableSha1Set": "构建可达 SHA1 集合（依赖图按需翻译）", "BuildCombinedIR": "组合本地 stub + 依赖 stubs + 本地 text → out_buf（按需加载）"} \ No newline at end of file diff --git a/TransPyV/temp/b5d05dbd7fee36ef.pyi b/TransPyV/temp/b5d05dbd7fee36ef.pyi new file mode 100644 index 0000000..4466a75 --- /dev/null +++ b/TransPyV/temp/b5d05dbd7fee36ef.pyi @@ -0,0 +1,60 @@ +""" +Auto-generated Python stub file from lib.core.StubMerger.py +Module: lib.core.StubMerger +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import llvmlite +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import viperlib +import sys +import ast +import lib.core.IncludesScanner as IncludesScanner +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.Handles.HandlesTranslator as HandlesTranslator + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_INCLUDES_SHA1: t.CDefine = 256 +STUB_READ_BUF_SIZE: t.CDefine = 262144 +MAX_INCLUDES: t.CDefine = 256 + +def _load_includes_sha1_set(pool: memhub.MemBuddy | t.CPtr, temp_dir: str, sha1_set: str) -> int: pass + +def _is_in_sha1_set(sha1: str, sha1_set: str, set_count: int) -> int: pass + +def WriteIncludesSha1Map(mb: memhub.MemBuddy | t.CPtr, temp_dir: str, scan_result: IncludesScanner.ScanResult | t.CPtr, filter_set: t.CChar | t.CPtr, filter_count: int) -> int: pass + +def _IsFuncDeclaredOrDefined(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutDeclareInBuf(out_buf: bytes, func_sig: str) -> int: pass + +def _CommentOutGlobalInBuf(out_buf: bytes, global_name: str) -> int: pass + +def _CommentOutOpaqueTypeInBuf(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _HasFullTypeDefinition(out_buf: bytes, type_name_prefix: str) -> int: pass + +def _LoadAndAppendStub(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _LoadAndAppendTextDeclares(temp_dir: str, td_len: t.CSizeT, dep_sha1: str, dep_buf: bytes, out_buf: bytes, out_size: t.CSizeT, out_pos: t.CSizeT) -> t.CSizeT: pass + +def _PathToModuleName(path: str) -> str: pass + +def _BuildIncludesSha1Map(temp_dir: str, td_len: t.CSizeT, sha1_list: t.CChar | t.CPtr, mod_list: t.CChar | t.CPtr) -> int: pass + +def _FindSha1ByModName(sha1_arr: t.CChar | t.CPtr, mod_arr: t.CChar | t.CPtr, count: int, mod_name: str) -> str: pass + +def _AddSha1ToSet(set_buf: t.CChar | t.CPtr, count: int, sha1: str) -> int: pass + +def _BuildReachableSha1Set(mb: memhub.MemBuddy | t.CPtr, source_dir: str, temp_dir: str, reachable_set: t.CChar | t.CPtr) -> int: pass + +def BuildCombinedIR(temp_dir: str, local_sha1: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass diff --git a/TransPyV/temp/b8c66c8ff44eb874.pyi b/TransPyV/temp/b8c66c8ff44eb874.pyi new file mode 100644 index 0000000..85bdb8f --- /dev/null +++ b/TransPyV/temp/b8c66c8ff44eb874.pyi @@ -0,0 +1,46 @@ +""" +Auto-generated Python stub file from hashtable.py +Module: hashtable +""" + + +import t, c +import string +import memhub +from stdint import * + +_HT_EMPTY: t.CDefine = 0 +_HT_TOMBSTONE: t.CDefine = 1 +_HT_INIT_CAP: t.CDefine = 16 +_HT_SLOT_SIZE: t.CDefine = 24 +_FNV_OFFSET: t.CExtern | t.CUInt64T +_FNV_PRIME: t.CExtern | t.CUInt64T + +def _fnv1a_hash(key: str) -> t.CUInt64T: pass + + +class HashTable: + __slots__: t.CVoid | t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 16 + __tombstones__: t.CSizeT = 0 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + def __new__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: pass + def __init__(self: HashTable, mb: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + def _slot_hash(self: HashTable, idx: t.CSizeT) -> t.CUInt64T: pass + def _slot_key(self: HashTable, idx: t.CSizeT) -> str: pass + def _slot_value(self: HashTable, idx: t.CSizeT) -> t.CPtr: pass + def _set_slot(self: HashTable, idx: t.CSizeT, h: t.CUInt64T, key: str, val: t.CPtr) -> t.CInt: pass + def _find_slot(self: HashTable, key: str, h: t.CUInt64T) -> t.CSizeT: pass + def _resize(self: HashTable, new_cap: t.CSizeT) -> t.CInt: pass + def __setitem__(self: HashTable, key: str, val: t.CNeedPtr) -> t.CInt: pass + def __getitem__(self: HashTable, key: str) -> t.CPtr: pass + def __contains__(self: HashTable, key: str) -> t.CInt: pass + def __delitem__(self: HashTable, key: str) -> t.CInt: pass + def __len__(self: HashTable) -> t.CSizeT: pass + def __iter__(self: HashTable) -> HashTable | t.CPtr: pass + def __next__(self: HashTable) -> str: pass + def get(self: HashTable, key: str, default: t.CPtr) -> t.CPtr: pass + def set_int(self: HashTable, key: str, val: int) -> t.CInt: pass + def set_str(self: HashTable, key: str, val: str) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/bbdf3bbd4c3bc28c.pyi b/TransPyV/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/TransPyV/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/TransPyV/temp/bd82e3ca81546c76.doc.json b/TransPyV/temp/bd82e3ca81546c76.doc.json new file mode 100644 index 0000000..0858afd --- /dev/null +++ b/TransPyV/temp/bd82e3ca81546c76.doc.json @@ -0,0 +1 @@ +{"FileEntry": "扫描到的文件条目", "ScanResult": "扫描结果集合", "create_scan_result": "创建扫描结果容器，预分配 MAX_FILES 个槽位", "add_file_entry": "添加文件条目到扫描结果，返回 0 成功", "compute_file_sha1": "读取文件内容并计算 SHA1 前16字符（CRLF→LF 后计算，与 Projectrans.py 一致）", "scan_directory_recursive": "递归扫描目录，收集 .py 文件", "scan_includes": "扫描 includes 目录，返回所有 .py 文件的扫描结果", "find_entry_by_module": "按模块名查找文件条目"} \ No newline at end of file diff --git a/TransPyV/temp/bd82e3ca81546c76.pyi b/TransPyV/temp/bd82e3ca81546c76.pyi new file mode 100644 index 0000000..ba0eee6 --- /dev/null +++ b/TransPyV/temp/bd82e3ca81546c76.pyi @@ -0,0 +1,45 @@ +""" +Auto-generated Python stub file from lib.core.IncludesScanner.py +Module: lib.core.IncludesScanner +""" + + +import t, c +from stdint import * +import memhub +import string +import stdio +import stdlib +import w32.win32file as win32file +import w32.win32base as win32base +import w32.fileio as fileio +import hashlib +import viperlib + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +MAX_PATH_LEN: t.CDefine = 512 +MAX_FILES: t.CDefine = 256 + +@t.NoVTable +class FileEntry: + Path: str + Sha1: str + RelPath: str + ModuleName: str +@t.NoVTable +class ScanResult: + Entries: FileEntry | t.CPtr + Count: t.CInt + Capacity: t.CInt + +def create_scan_result(pool: memhub.MemBuddy | t.CPtr) -> ScanResult | t.CPtr: pass + +def add_file_entry(result: ScanResult | t.CPtr, pool: memhub.MemBuddy | t.CPtr, abs_path: str, rel_path: str, sha1: str) -> int: pass + +def compute_file_sha1(pool: memhub.MemBuddy | t.CPtr, file_path: str) -> str: pass + +def scan_directory_recursive(pool: memhub.MemBuddy | t.CPtr, root_dir: str, rel_prefix: str, result: ScanResult | t.CPtr) -> int: pass + +def scan_includes(pool: memhub.MemBuddy | t.CPtr, includes_dir: str) -> ScanResult | t.CPtr: pass + +def find_entry_by_module(result: ScanResult | t.CPtr, module_name: str) -> FileEntry | t.CPtr: pass diff --git a/TransPyV/temp/c3b259b4059f8668.pyi b/TransPyV/temp/c3b259b4059f8668.pyi new file mode 100644 index 0000000..67ff4ce --- /dev/null +++ b/TransPyV/temp/c3b259b4059f8668.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from viperlib.py +Module: viperlib +""" + + +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: pass + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: pass + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: pass + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: pass + +def _check_special_float(fv: t.CDouble) -> t.CInt: pass + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: pass diff --git a/TransPyV/temp/c4f2d483ec0f4db5.doc.json b/TransPyV/temp/c4f2d483ec0f4db5.doc.json new file mode 100644 index 0000000..a8e6f20 --- /dev/null +++ b/TransPyV/temp/c4f2d483ec0f4db5.doc.json @@ -0,0 +1 @@ +{"_is_cenum_base": "检查 base 节点是否为 t.CEnum，返回 1=是 / 0=否", "_is_enum_class": "检查 ClassDef 是否继承自 t.CEnum，返回 1=是 / 0=否", "_is_cunion_base": "检查 base 节点是否为 t.CUnion，返回 1=是 / 0=否", "_is_union_class": "检查 ClassDef 是否继承自 t.CUnion，返回 1=是 / 0=否", "_has_decorator": "检查 decorator_list 中是否有指定装饰器名", "_is_marker_base": "检查基类名是否为标记基类，返回 1=是 / 0=否", "_get_base_name": "从 base AST 节点提取基类名，返回 None=失败", "_get_parent_class": "获取 ClassDef 的真实父类名，返回 None=无真实父类", "_has_any_func_cvtable": "检查类中是否有任何函数标记了 @t.CVTable，返回 1=有 / 0=无", "_should_method_be_virtual": "判断方法是否应该进入虚表，返回 1=包含 / 0=排除", "_detect_vtable_status": "检测类是否应该启用虚表", "_resolve_enum_member_type": "解析枚举成员的类型注解为 LLVM 整数类型", "_is_state_annotation": "检查注解节点是否为 t.State，返回 1=是 / 0=否", "_extract_int_value": "从 AST 节点提取整数值，失败返回 0", "_translate_enum_def": "翻译枚举类定义，返回 0", "_get_type_size": "计算 LLVM 类型的字节大小", "_translate_union_def": "翻译联合体定义，返回 0", "_is_generic_class": "检查 ClassDef 是否有类型参数（泛型类），返回 1=是 / 0=否", "_register_generic_template": "注册泛型类模板（类名 + ClassDef 节点）", "_find_generic_template": "查找泛型类模板，返回 ClassDef 节点或 None", "_mangle_generic_class_name": "生成特化类名: list + [int] -> list[int]\n\n 分配新缓冲区拼接，避免原地 strcat 导致缓冲区溢出\n ", "_clone_annotation": "拷贝类型注解表达式，替换类型参数 T 为具体类型名\n\n tp_names: 类型参数名列表 ['T']\n type_args: 类型实参名列表 ['int']\n ", "_clone_arguments": "拷贝 Arguments 节点，替换 arg annotation 中的 T", "_clone_classdef_for_spec": "拷贝 ClassDef 节点用于泛型特化\n\n - 替换类名为 spec_name\n - 清除 type_params（特化后不是泛型类）\n - 拷贝 children 中的 AnnAssign 和 FunctionDef，替换类型注解中的 T\n - 方法体直接引用原始节点（通过 type_map 上下文处理 T）\n ", "_find_cached_spec": "查找已缓存的特化名，未找到返回 None", "_cache_spec": "缓存特化结果", "_specialize_generic_class": "特化泛型类，返回特化类名\n\n 1. 查找泛型模板\n 2. 生成特化名\n 3. 检查缓存\n 4. 拷贝 ClassDef 节点（替换类型注解中的 T）\n 5. 设置 type_map 上下文（用于方法体中的 T 替换）\n 6. 调用 translate_class_def 翻译特化类\n 7. 返回特化名\n ", "translate_class_def": "翻译 class 定义，返回 0", "_vtable_strcat": "将 src 追加到 dst 末尾", "_vtable_name_needs_quote": "检查 LLVM 标识符是否需要引号（以数字开头或含 '.'）", "_format_func_type_str": "从 Function 对象生成函数类型字符串", "_generate_vtable": "生成 VTable 全局变量", "_translate_oop_methods": "扫描 class body 中的方法并翻译，生成 __before_init__", "_translate_method": "翻译单个方法，返回 0", "_generate_before_init": "生成 __before_init__ 函数，返回 0"} \ No newline at end of file diff --git a/TransPyV/temp/c4f2d483ec0f4db5.pyi b/TransPyV/temp/c4f2d483ec0f4db5.pyi new file mode 100644 index 0000000..62b2e80 --- /dev/null +++ b/TransPyV/temp/c4f2d483ec0f4db5.pyi @@ -0,0 +1,104 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesClassDef.py +Module: lib.core.Handles.HandlesClassDef +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import stdio +import viperlib +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesEnum as HandlesEnum + +ENUM_MEMBER_MAX_LOCAL: t.CDefine = 64 + +def _is_cenum_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_enum_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _is_cunion_base(base_node: ast.AST | t.CPtr) -> int: pass + +def _is_union_class(cd: ast.ClassDef | t.CPtr) -> int: pass + + +_MARKER_BASES: t.CDefine = 8 + +def _has_decorator(decorator_list: list[ast.AST | t.CPtr] | t.CPtr, deco_name: str) -> int: pass + +def _is_marker_base(base_name: str) -> int: pass + +def _get_base_name(base_node: ast.AST | t.CPtr) -> str: pass + +def _get_parent_class(cd: ast.ClassDef | t.CPtr) -> str: pass + +def _has_any_func_cvtable(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _should_method_be_virtual(cd: ast.ClassDef | t.CPtr, fd: ast.FunctionDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _detect_vtable_status(cd: ast.ClassDef | t.CPtr, trans: HT.Translator | t.CPtr) -> int: pass + +def _resolve_enum_member_type(pool: memhub.MemBuddy | t.CPtr, ann_node: ast.AST | t.CPtr, imported_modules: str, from_imports: str) -> llvmlite.LLVMType | t.CPtr: pass + +def _is_state_annotation(node: ast.AST | t.CPtr) -> int: pass + +def _extract_int_value(node: ast.AST | t.CPtr) -> t.CInt64T: pass + +def _translate_enum_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + +def _get_type_size(ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def _translate_union_def(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr) -> int: pass + + +_generic_class_names: t.CExtern | list[str] | t.CPtr +_generic_class_nodes: t.CExtern | list[ast.ClassDef | t.CPtr] | t.CPtr + +def _is_generic_class(cd: ast.ClassDef | t.CPtr) -> int: pass + +def _register_generic_template(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr) -> t.CInt: pass + +def _find_generic_template(class_name: str) -> ast.ClassDef | t.CPtr: pass + +def _mangle_generic_class_name(pool: memhub.MemBuddy | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def _clone_annotation(pool: memhub.MemBuddy | t.CPtr, node: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_arguments(pool: memhub.MemBuddy | t.CPtr, args: ast.AST | t.CPtr, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.AST | t.CPtr: pass + +def _clone_classdef_for_spec(pool: memhub.MemBuddy | t.CPtr, cd: ast.ClassDef | t.CPtr, spec_name: str, tp_names: list[str] | t.CPtr, type_args: list[str] | t.CPtr) -> ast.ClassDef | t.CPtr: pass + + +_spec_keys: t.CExtern | list[str] | t.CPtr +_spec_names: t.CExtern | list[str] | t.CPtr + +def _find_cached_spec(spec_key: str) -> str: pass + +def _cache_spec(pool: memhub.MemBuddy | t.CPtr, spec_key: str, spec_name: str) -> t.CInt: pass + +def _specialize_generic_class(trans: HT.Translator | t.CPtr, class_name: str, type_args: list[str] | t.CPtr) -> str: pass + +def translate_class_def(trans: HT.Translator | t.CPtr, node: ast.AST | t.CPtr) -> int: pass + + +VTBL_METHOD_MAX: t.CDefine = 32 + +def _vtable_strcat(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _vtable_name_needs_quote(name: str) -> int: pass + +def _format_func_type_str(pool: memhub.MemBuddy | t.CPtr, buf: t.CChar | t.CPtr, buf_size: t.CSizeT, func: llvmlite.Function | t.CPtr) -> int: pass + +def _generate_vtable(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, class_name: str) -> int: pass + +def _translate_oop_methods(trans: HT.Translator | t.CPtr, cd: ast.ClassDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _translate_method(trans: HT.Translator | t.CPtr, fd: ast.FunctionDef | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass + +def _generate_before_init(trans: HT.Translator | t.CPtr, struct_ty: llvmlite.LLVMType | t.CPtr, class_name: str) -> int: pass diff --git a/TransPyV/temp/c9d54a4158f7f5a8.pyi b/TransPyV/temp/c9d54a4158f7f5a8.pyi new file mode 100644 index 0000000..3e2d738 --- /dev/null +++ b/TransPyV/temp/c9d54a4158f7f5a8.pyi @@ -0,0 +1,34 @@ +""" +Auto-generated Python stub file from hashlib.__sha256.py +Module: hashlib.__sha256 +""" + + +import t, c + +SHA256_BLOCK_LEN: t.CDefine = 64 +SHA256_DIGEST_LEN: t.CDefine = 32 +sha256_K: t.CExtern | t.CArray[t.CUInt32T, 64] + +def s0(x: t.CUInt32T) -> t.CUInt32T: pass + +def s1(x: t.CUInt32T) -> t.CUInt32T: pass + +def S0(x: t.CUInt32T) -> t.CUInt32T: pass + +def S1(x: t.CUInt32T) -> t.CUInt32T: pass + +def Ch(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + +def Maj(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: pass + + +@t.Object +class sha256: + state: t.CArray[t.CUInt32T, 8] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA256_BLOCK_LEN] + def __init__(self: sha256) -> t.CInt: pass + def transform(self: sha256, block: t.CUInt8T | t.CPtr) -> t.CInt: pass + def update(self: sha256, s: str) -> t.CInt: pass + def final(self: sha256, out: t.CArray[t.CUInt8T, SHA256_DIGEST_LEN]) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/c9f4be41ca1cc2b4.pyi b/TransPyV/temp/c9f4be41ca1cc2b4.pyi new file mode 100644 index 0000000..fdbd7ec --- /dev/null +++ b/TransPyV/temp/c9f4be41ca1cc2b4.pyi @@ -0,0 +1,22 @@ +""" +Auto-generated Python stub file from viperio.py +Module: viperio +""" + + +import t, c +from stdint import * + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + def __init__(self: Buf, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT, owned: bool) -> t.CInt: pass + def clear(self: Buf) -> t.CInt: pass + def write(self: Buf, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: pass + def cstr(self: Buf) -> t.CChar | t.CPtr: pass + def reset(self: Buf) -> t.CInt: pass + def __enter__(self: Buf) -> 'Buf' | t.CPtr: pass + def __exit__(self: Buf) -> t.CInt: pass + def free(self: Buf) -> t.CInt: pass \ No newline at end of file diff --git a/TransPyV/temp/d1599ecbac5b6586.doc.json b/TransPyV/temp/d1599ecbac5b6586.doc.json new file mode 100644 index 0000000..a6b0616 --- /dev/null +++ b/TransPyV/temp/d1599ecbac5b6586.doc.json @@ -0,0 +1 @@ +{"TranslateFileGetTrans": "翻译文件，返回 Translator 对象（调用者负责 dump_ir），None 失败\n\n current_package: 当前文件所属包名（用于解析相对导入），None 表示顶级模块\n ", "ensure_dir": "递归创建目录（类似 mkdir -p），目录已存在视为成功\n\n Args:\n path: 目录路径（支持 / 或 \\ 分隔符）\n\n Returns:\n 0 成功（包括目录已存在），非 0 失败\n ", "write_ir_to_file": "将 IR 缓冲区写入 .ll 文件\n\n Args:\n ir_buf: IR 文本缓冲区\n ir_len: IR 文本长度\n output_dir: 输出目录（temp 或 output）\n module_name: 模块名（如 \"main\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_ll_to_obj": "调用 llc 将 .ll 编译为 .obj\n\n Args:\n ir_path: .ll 文件路径\n output_dir: 输出目录\n module_name: 模块名（用于生成 .obj 文件名）\n cc_cmd: 编译器命令（如 \"llc\"）\n cc_flags: 编译器参数（如 \"-filetype=obj -relocation-model=pic\"）\n\n Returns:\n 0 成功，非 0 失败\n ", "collect_obj_files": "扫描 includes_binary_dir 目录，收集所有 .obj 文件路径到 out_buf\n\n Args:\n includes_binary_dir: includes.binary 目录路径\n out_buf: 输出缓冲区（用于存放空格分隔的 .obj 文件路径）\n out_size: 输出缓冲区大小\n\n Returns:\n 写入的字节数（不含 null 终止符），0 表示无文件或错误\n ", "link_obj_to_exe": "调用 clang++ 链接 .obj 为 .exe\n\n Args:\n output_dir: 输出目录（包含 .obj 文件）\n module_name: 模块名\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件名（如 \"test.exe\"）\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "compile_module_to_obj": "编译 IR 到 .obj（不链接）\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名（SHA1）\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n\n Returns:\n 0 成功，非 0 失败\n ", "link_objs_to_exe": "链接多个 .obj 文件为 .exe\n\n Args:\n obj_paths: 空格分隔的 .obj 文件完整路径字符串\n obj_paths_len: obj_paths 长度\n linker_cmd: 链接器命令（如 \"clang++\"）\n linker_flags: 链接器参数\n linker_output: 输出文件完整路径\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n 0 成功，非 0 失败\n ", "run_pipeline": "执行完整编译管线\n\n Args:\n ir_buf: LLVM IR 文本缓冲区\n ir_len: IR 文本长度\n temp_dir: 临时目录\n output_dir: 输出目录\n module_name: 模块名\n cc_cmd: 编译器命令\n cc_flags: 编译器参数\n linker_cmd: 链接器命令\n linker_flags: 链接器参数\n linker_output: 输出文件名\n includes_binary_dir: includes.binary 目录路径（链接时附加预编译 .obj）\n\n Returns:\n BuildResult 对象\n "} \ No newline at end of file diff --git a/TransPyV/temp/d1599ecbac5b6586.pyi b/TransPyV/temp/d1599ecbac5b6586.pyi new file mode 100644 index 0000000..c155594 --- /dev/null +++ b/TransPyV/temp/d1599ecbac5b6586.pyi @@ -0,0 +1,51 @@ +""" +Auto-generated Python stub file from lib.core.BuildPipeline.py +Module: lib.core.BuildPipeline +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import subprocess +import viperlib +import ast +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesImports as HandlesImports + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def TranslateFileGetTrans(mb: memhub.MemBuddy | t.CPtr, file_path: str, sha1_val: str, current_package: str) -> HandlesTranslator.Translator | t.CPtr: pass + + +class BuildResult: + Success: t.CInt + OutputPath: str + ErrorMsg: str + def __new__(self: BuildResult) -> t.CPtr: pass + def __init__(self: BuildResult) -> t.CInt: pass + +def ensure_dir(path: str) -> int: pass + +def write_ir_to_file(ir_buf: bytes, ir_len: t.CSizeT, output_dir: str, module_name: str) -> int: pass + +def compile_ll_to_obj(ir_path: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def collect_obj_files(includes_binary_dir: str, out_buf: bytes, out_size: t.CSizeT) -> t.CSizeT: pass + +def link_obj_to_exe(output_dir: str, module_name: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def compile_module_to_obj(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str) -> int: pass + +def link_objs_to_exe(obj_paths: str, obj_paths_len: t.CSizeT, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> int: pass + +def run_pipeline(ir_buf: bytes, ir_len: t.CSizeT, temp_dir: str, output_dir: str, module_name: str, cc_cmd: str, cc_flags: str, linker_cmd: str, linker_flags: str, linker_output: str, includes_binary_dir: str) -> BuildResult | t.CPtr: pass diff --git a/TransPyV/temp/d4117f5e00a6c3b5.pyi b/TransPyV/temp/d4117f5e00a6c3b5.pyi new file mode 100644 index 0000000..ee44bdb --- /dev/null +++ b/TransPyV/temp/d4117f5e00a6c3b5.pyi @@ -0,0 +1,62 @@ +""" +Auto-generated Python stub file from llvmlite.__verify.py +Module: llvmlite.__verify +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__function import Function, BasicBlock, Line, Param +from .__module import LLVMModule + +VERIFY_OK: t.CDefine = 0 +VERIFY_ERR_NO_TERMINATOR: t.CDefine = 1 +VERIFY_ERR_DUPLICATE_DEF: t.CDefine = 2 +VERIFY_ERR_UNDEFINED_USE: t.CDefine = 3 +VERIFY_ERR_PHI_NOT_FIRST: t.CDefine = 4 + +class NameEntry(GSListNode[NameEntry]): + Name: t.CChar | t.CPtr +class VerifyResult: + ErrorCount: t.CInt + WarningCount: t.CInt + ErrorCode: t.CInt + ErrorBuf: t.CChar | t.CPtr + ErrorBufSize: t.CSizeT + +def _new_verify_result(pool: memhub.MemBuddy | t.CPtr, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: pass + +def _report_error(result: VerifyResult | t.CPtr, code: t.CInt, msg: t.CChar | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def _new_name_table(pool: memhub.MemBuddy | t.CPtr) -> GSList[NameEntry] | t.CPtr: pass + + +NENTRY_SIZE: t.CDefine = 16 + +def _name_table_add(table: GSList[NameEntry] | t.CPtr, pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_table_contains(table: GSList[NameEntry] | t.CPtr, name: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_char_name(ch: t.CChar) -> t.CInt: pass + +def _extract_def_name(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> t.CChar | t.CPtr: pass + +def _is_terminator(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _is_phi(text: t.CChar | t.CPtr) -> t.CInt: pass + +def _check_uses_in_line(table: GSList[NameEntry] | t.CPtr, text: t.CChar | t.CPtr, def_name: t.CChar | t.CPtr) -> t.CInt: pass + +def _name_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, table: GSList[NameEntry] | t.CPtr) -> t.CInt: pass + +def _str_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, target: t.CChar | t.CPtr) -> t.CInt: pass + +def verify_function(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass + +def verify_module(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: pass diff --git a/TransPyV/temp/d9322f0db660edd8.doc.json b/TransPyV/temp/d9322f0db660edd8.doc.json new file mode 100644 index 0000000..4dba15e --- /dev/null +++ b/TransPyV/temp/d9322f0db660edd8.doc.json @@ -0,0 +1 @@ +{"ReturnHandle": "Return 语句处理器：继承 Mixin 获得 Trans 回指针", "ReturnHandle.Handle": "翻译 Return 语句"} \ No newline at end of file diff --git a/TransPyV/temp/d9322f0db660edd8.pyi b/TransPyV/temp/d9322f0db660edd8.pyi new file mode 100644 index 0000000..abcbbff --- /dev/null +++ b/TransPyV/temp/d9322f0db660edd8.pyi @@ -0,0 +1,23 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesReturn.py +Module: lib.core.Handles.HandlesReturn +""" + + +import t, c +from stdint import * +import ast +import llvmlite +import memhub +import string +import lib.core.Handles.HandlesBase as HandlesBase +import lib.core.Handles.HandlesTranslator as HT +import lib.core.Handles.HandlesExpr as HandlesExpr +import lib.core.Handles.HandlesVar as HandlesVar + +@t.NoVTable +class ReturnHandle(HandlesBase.Mixin): + def __init__(self: ReturnHandle, trans: HT.Translator | t.CPtr) -> t.CInt: pass + def Handle(self: ReturnHandle, node: ast.AST | t.CPtr) -> int: pass + +def NewReturnHandle(pool: memhub.MemBuddy | t.CPtr, trans: HT.Translator | t.CPtr) -> ReturnHandle | t.CPtr: pass diff --git a/TransPyV/temp/db7c4a8f09d02a17.doc.json b/TransPyV/temp/db7c4a8f09d02a17.doc.json new file mode 100644 index 0000000..56e8c7e --- /dev/null +++ b/TransPyV/temp/db7c4a8f09d02a17.doc.json @@ -0,0 +1 @@ +{"RunPhase1": "Phase1: 扫描 includes 目录，按需翻译并生成 stub"} \ No newline at end of file diff --git a/TransPyV/temp/db7c4a8f09d02a17.pyi b/TransPyV/temp/db7c4a8f09d02a17.pyi new file mode 100644 index 0000000..c2761c1 --- /dev/null +++ b/TransPyV/temp/db7c4a8f09d02a17.pyi @@ -0,0 +1,32 @@ +""" +Auto-generated Python stub file from lib.core.Phase1.py +Module: lib.core.Phase1 +""" + + +import t, c +from stdint import * +import stdio +import string +import stdlib +import memhub +import viperlib +import w32.fileio as fileio +import w32.win32file +import w32.win32base +import ast +import llvmlite +import lib.core.VLogger as VLogger +import lib.core.Handles.HandlesTranslator as HandlesTranslator +import lib.core.Handles.HandlesStruct as HandlesStruct +import lib.core.Handles.HandlesType as HandlesType +import lib.core.Handles.HandlesExprCall as HandlesExprCall +import lib.core.Handles.HandlesImports as HandlesImports +import lib.core.IncludesScanner as IncludesScanner +import lib.core.StubMerger as StubMerger +import lib.Projectrans.Config as Config + +_mbuddy: t.CExtern | memhub.MemManager | t.CPtr +SRC_BUF_SIZE: t.CDefine = 1048576 + +def RunPhase1(mb: memhub.MemBuddy | t.CPtr, includes_dir: str, temp_dir: str, log: VLogger.Logger | t.CPtr) -> int: pass diff --git a/TransPyV/temp/e0b84369e50c937c.pyi b/TransPyV/temp/e0b84369e50c937c.pyi new file mode 100644 index 0000000..c10ea89 --- /dev/null +++ b/TransPyV/temp/e0b84369e50c937c.pyi @@ -0,0 +1,93 @@ +""" +Auto-generated Python stub file from llvmlite.__function.py +Module: llvmlite.__function +""" + + +import t, c +from stdint import * +import string +import viperlib +import memhub +import stdio +from linkedlist import GSListNode, GSList +from .__types import LLVMType, ParamNode, TypePrint + +class Line(GSListNode[Line]): + Buf: t.CChar | t.CPtr + Len: t.CSizeT +class BasicBlock(GSListNode[BasicBlock]): + Name: t.CChar | t.CPtr + Lines: GSList[Line] | t.CPtr + IsTerminated: t.CInt +class Param(GSListNode[Param]): + Ty: LLVMType | t.CPtr + Name: t.CChar | t.CPtr + Attrs: t.CChar | t.CPtr +class Function(GSListNode[Function]): + Name: t.CChar | t.CPtr + RetTy: LLVMType | t.CPtr + Params: GSList[Param] | t.CPtr + Blocks: GSList[BasicBlock] | t.CPtr + IsDeclared: t.CInt + IsVarArg: t.CInt + Attrs: t.CChar | t.CPtr + +LINE_SIZE: t.CDefine = 24 + +def new_line(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> Line | t.CPtr: pass + +def _new_gslist(pool: memhub.MemBuddy | t.CPtr, list_size: t.CSizeT) -> t.CPtr: pass + + +BLOCK_SIZE: t.CDefine = 32 + +def new_basic_block(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: pass + + +PARAM_SIZE: t.CDefine = 32 + +def new_param(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: pass + + +FUNC_SIZE: t.CDefine = 56 + +def new_function(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: pass + +def function_add_param(func: Function | t.CPtr, param: Param | t.CPtr) -> t.CInt: pass + +def function_add_block(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def function_move_block_to_end(func: Function | t.CPtr, block: BasicBlock | t.CPtr) -> t.CInt: pass + +def block_append_line(block: BasicBlock | t.CPtr, line: Line | t.CPtr) -> t.CInt: pass + +def block_append_text(pool: memhub.MemBuddy | t.CPtr, block: BasicBlock | t.CPtr, text: t.CChar | t.CPtr) -> t.CInt: pass + +def function_set_attrs(func: Function | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def param_set_attrs(param: Param | t.CPtr, attrs: t.CChar | t.CPtr) -> t.CInt: pass + +def _ll_name_needs_quote(name: str) -> t.CInt: pass + +def FunctionPrint(buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr, pool: memhub.MemBuddy | t.CPtr) -> t.CInt: pass + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr) -> t.CInt: pass + +def function_get_name(func: Function | t.CPtr) -> t.CChar | t.CPtr: pass + +def function_get_ret_ty(func: Function | t.CPtr) -> LLVMType | t.CPtr: pass + +def function_get_params(func: Function | t.CPtr) -> GSList[Param] | t.CPtr: pass + +def function_get_param_head(func: Function | t.CPtr) -> Param | t.CPtr: pass + +def function_get_next(func: Function | t.CPtr) -> Function | t.CPtr: pass + +def function_is_declared(func: Function | t.CPtr) -> t.CInt: pass + +def param_get_name(param: Param | t.CPtr) -> t.CChar | t.CPtr: pass + +def param_get_ty(param: Param | t.CPtr) -> LLVMType | t.CPtr: pass + +def param_get_next(param: Param | t.CPtr) -> Param | t.CPtr: pass diff --git a/TransPyV/temp/e5be27ae4fe7ce04.doc.json b/TransPyV/temp/e5be27ae4fe7ce04.doc.json new file mode 100644 index 0000000..7307589 --- /dev/null +++ b/TransPyV/temp/e5be27ae4fe7ce04.doc.json @@ -0,0 +1 @@ +{"init_enum_table": "初始化枚举注册表，返回 1 成功", "_get_enum_entry": "获取第 i 个枚举条目", "_get_enum_member": "获取枚举中第 i 个成员条目", "register_enum": "注册枚举类型，返回 EnumEntry 指针", "add_enum_member": "向枚举添加成员，返回成员索引（-1 失败）", "find_enum": "按枚举类名查找，返回 EnumEntry 或 None", "lookup_enum_member": "按枚举类名和成员名查找，返回 EnumMember 或 None", "is_enum_class": "检查类名是否为已注册枚举，返回 1=是 / 0=否", "get_enum_base_type": "按枚举类名获取基准类型"} \ No newline at end of file diff --git a/TransPyV/temp/e5be27ae4fe7ce04.pyi b/TransPyV/temp/e5be27ae4fe7ce04.pyi new file mode 100644 index 0000000..ee312c4 --- /dev/null +++ b/TransPyV/temp/e5be27ae4fe7ce04.pyi @@ -0,0 +1,48 @@ +""" +Auto-generated Python stub file from lib.core.Handles.HandlesEnum.py +Module: lib.core.Handles.HandlesEnum +""" + + +import t, c +from stdint import * +import memhub +import string +import llvmlite +import stdio + +ENUM_MAX: t.CDefine = 64 +ENUM_MEMBER_MAX: t.CDefine = 64 + +@t.NoVTable +class EnumMember: + Name: t.CChar | t.CPtr + Value: t.CInt64T + Ty: llvmlite.LLVMType | t.CPtr +@t.NoVTable +class EnumEntry: + Name: t.CChar | t.CPtr + BaseTy: llvmlite.LLVMType | t.CPtr + MemberCount: int + Members: EnumMember | t.CPtr + +_enum_table: t.CExtern | EnumEntry | t.CPtr +_enum_count: t.CExtern | int + +def init_enum_table(pool: memhub.MemBuddy | t.CPtr) -> int: pass + +def _get_enum_entry(i: int) -> EnumEntry | t.CPtr: pass + +def _get_enum_member(enum_entry: EnumEntry | t.CPtr, i: int) -> EnumMember | t.CPtr: pass + +def register_enum(pool: memhub.MemBuddy | t.CPtr, name: str, base_ty: llvmlite.LLVMType | t.CPtr) -> EnumEntry | t.CPtr: pass + +def add_enum_member(pool: memhub.MemBuddy | t.CPtr, enum_entry: EnumEntry | t.CPtr, member_name: str, member_val: t.CInt64T, member_ty: llvmlite.LLVMType | t.CPtr) -> int: pass + +def find_enum(name: str) -> EnumEntry | t.CPtr: pass + +def lookup_enum_member(enum_name: str, member_name: str) -> EnumMember | t.CPtr: pass + +def is_enum_class(name: str) -> int: pass + +def get_enum_base_type(enum_name: str) -> llvmlite.LLVMType | t.CPtr: pass diff --git a/TransPyV/temp/ee084e9fc6ee413a.pyi b/TransPyV/temp/ee084e9fc6ee413a.pyi new file mode 100644 index 0000000..45551ef --- /dev/null +++ b/TransPyV/temp/ee084e9fc6ee413a.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from memhub.py +Module: memhub +""" + + +import t, c +from stdint import * +import string +import atom +import viperio + +MEMHUB_ALIGN: t.CDefine = 8 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: pass + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: pass + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: pass + + +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr + size: t.CSizeT + def __init__(self: MemManager, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemManager, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemManager, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemManager) -> t.CInt: pass + def calloc(self: MemManager, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def realloc(self: MemManager, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def __enter__(self: MemManager) -> 'MemManager' | t.CPtr: pass + def __exit__(self: MemManager) -> t.CInt: pass + def alloc_buf(self: MemManager, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: pass +class MemPool(MemManager): + offset: t.CSizeT + high_water: t.CSizeT + def __init__(self: MemPool, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemPool, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemPool, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemPool) -> t.CInt: pass +class MemSlab(MemManager): + block_size: t.CSizeT + block_count: t.CSizeT + used_count: t.CSizeT + free_list: t.CVoid | t.CPtr + alloc_map: t.CUInt8T | t.CPtr + alloc_map_size: t.CSizeT + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemSlab, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT) -> t.CInt: pass + def alloc(self: MemSlab, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemSlab, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemSlab) -> t.CInt: pass +class MemBuddy(MemManager): + max_order: t.CInt + free_lists: t.CUInt64T | t.CPtr + lock_val: t.CVolatile | t.CInt + usable: t.CVoid | t.CPtr + usable_size: t.CSizeT + def __init__(self: MemBuddy, base: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CInt: pass + def _fl_push(self: MemBuddy, order: t.CInt, block: t.CVoid | t.CPtr) -> t.CInt: pass + def _fl_pop(self: MemBuddy, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _fl_find_and_remove(self: MemBuddy, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: pass + def _buddy_of(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: pass + def _order_for_size(self: MemBuddy, size: t.CSizeT) -> t.CInt: pass + def _split_to_order(self: MemBuddy, to_order: t.CInt) -> t.CVoid | t.CPtr: pass + def _coalesce(self: MemBuddy, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CInt: pass + def _is_valid_ptr(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def _lock(self: MemBuddy) -> t.CInt: pass + def _unlock(self: MemBuddy) -> t.CInt: pass + def alloc(self: MemBuddy, size: t.CSizeT) -> t.CVoid | t.CPtr: pass + def free(self: MemBuddy, ptr: t.CVoid | t.CPtr) -> t.CInt: pass + def reset(self: MemBuddy) -> t.CInt: pass + def realloc(self: MemBuddy, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: pass \ No newline at end of file diff --git a/TransPyV/temp/ee52c08239684346.pyi b/TransPyV/temp/ee52c08239684346.pyi new file mode 100644 index 0000000..ee09339 --- /dev/null +++ b/TransPyV/temp/ee52c08239684346.pyi @@ -0,0 +1,215 @@ +""" +Auto-generated Python stub file from ast.parser.py +Module: ast.parser +""" + + +import t, c +from stdint import * +import memhub +import string +from .tokens import Token, TokenType, Keyword, TokOp +from .base import AST, ASTCtx, ASTKind, OpKind, ASTFlag, _set_pos, _binop_from_op, _augop_from_op, _cmpop_from_op, _copy_str, _inherit_pos, _init_ast, CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, FLAG_IS_ASYNC +from .exprs import BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, ListComp, SetComp, DictComp, GeneratorExp, Await, Yield, YieldFrom, FormattedValue, JoinedStr, Constant, NamedExpr, Attribute, Subscript, Starred, Name, List, Tuple, Slice, Call, Compare, OpNode +from .stmts import Module, Expression, Interactive, FunctionType, FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, For, While, If, With, Raise, Try, Assert, Global, Nonlocal, Pass, Break, Continue, Expr, Import, ImportFrom, Match +from .astaux import ExceptHandler, Arguments, Arg, Keyword as KwNode, Alias, WithItem, Comprehension +from .match import MatchCase, MatchValue, MatchSingleton, MatchSequence, MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr +from .lexer import Lexer, _lexer_init, tokenize, new_lexer + +class Parser: + tokens: Token | t.CPtr + cur: Token | t.CPtr + pool: memhub.MemManager | t.CPtr + error_count: t.CInt + pending_decorators: list[AST | t.CPtr] | t.CPtr + def __new__(self: Parser, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def new_parser(pool: memhub.MemManager | t.CPtr) -> Parser | t.CPtr: pass + +def _parser_init(ps: Parser | t.CPtr, tokens: Token | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CInt: pass + +def _advance(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_type(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_kw(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_str(ps: Parser | t.CPtr) -> str: pass + +def _cur_lineno(ps: Parser | t.CPtr) -> t.CInt: pass + +def _cur_col(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nl(ps: Parser | t.CPtr) -> t.CInt: pass + +def _skip_nls_and_newlines(ps: Parser | t.CPtr) -> t.CInt: pass + +def _expect_type(ps: Parser | t.CPtr, ttype: t.CInt) -> t.CInt: pass + +def _expect_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _match_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _accept_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: pass + +def _new_const_int(pool: memhub.MemManager | t.CPtr, val: t.CInt64T, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_float(pool: memhub.MemManager | t.CPtr, val: t.CDouble, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_str(pool: memhub.MemManager | t.CPtr, val: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_bool(pool: memhub.MemManager | t.CPtr, val: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _new_const_none(pool: memhub.MemManager | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _set_ctx(node: AST | t.CPtr, ctx: t.CInt) -> t.CInt: pass + +def _fix_store_ctx(node: AST | t.CPtr) -> t.CInt: pass + +def _parse_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_namedexpr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_lambda(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_or_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_and_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_not_test(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_comparison(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _match_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _advance_comp_op(ps: Parser | t.CPtr) -> t.CInt: pass + +def _parse_bitor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitxor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_bitand(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_shift(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_arith(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_term(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_factor(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_power(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_call(ps: Parser | t.CPtr, func: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_subscript(ps: Parser | t.CPtr, value: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_slice(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_atom(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_paren(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_brace(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sub_expr(ps: Parser | t.CPtr, src: str) -> AST | t.CPtr: pass + +def _parse_fstring(ps: Parser | t.CPtr, content: str, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comprehension(ps: Parser | t.CPtr, elt: AST | t.CPtr, vtype: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_comp_for(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_target_list(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_target(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_yield(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_varargslist(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_arg_def(ps: Parser | t.CPtr, is_lambda: t.CInt) -> AST | t.CPtr: pass + +def _parse_suite_into_children(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_statement_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_simple_stmt_into(ps: Parser | t.CPtr, parent: AST | t.CPtr) -> t.CInt: pass + +def _parse_small_stmt(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _maybe_tuple(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_testlist(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_expr_stmt(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: pass + +def _parse_return(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_raise(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_global(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_nonlocal(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_from_import(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_alias_list(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_del(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_assert(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_compound(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_if_tail(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_while(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_for(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_try(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_except_handler(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_with(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_with_item(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_funcdef(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: pass + +def _parse_classdef(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_match(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _make_name_ast(pool: memhub.MemManager | t.CPtr, node: Name | t.CPtr) -> AST | t.CPtr: pass + +def _parse_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_closed_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_sequence_pattern(ps: Parser | t.CPtr, close_op: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_mapping_pattern(ps: Parser | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_class_pattern(ps: Parser | t.CPtr, cls: AST | t.CPtr, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: pass + +def _parse_match_case(ps: Parser | t.CPtr) -> AST | t.CPtr: pass + +def _parse_decorators(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: pass + +def _parse_module_body(ps: Parser | t.CPtr, module_node: AST | t.CPtr) -> t.CInt: pass + +def parse_tokens(pool: memhub.MemManager | t.CPtr, tokens: Token | t.CPtr) -> AST | t.CPtr: pass diff --git a/TransPyV/temp/f2ecbf8ced20575c.pyi b/TransPyV/temp/f2ecbf8ced20575c.pyi new file mode 100644 index 0000000..fdcae1d --- /dev/null +++ b/TransPyV/temp/f2ecbf8ced20575c.pyi @@ -0,0 +1,30 @@ +""" +Auto-generated Python stub file from _list.py +Module: _list +""" + + +import t, c +import memhub +import string +from stdint import * + +class list[T]: + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + def __new__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __init__(self: list, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT) -> t.CInt: pass + def __len__(self: list) -> t.CSizeT: pass + def append(self: list, item: T) -> t.CInt: pass + def get(self: list, index: t.CSizeT) -> T: pass + def __getitem__(self: list, index: t.CSizeT) -> T: pass + def set(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def __setitem__(self: list, index: t.CSizeT, value: T) -> t.CInt: pass + def pop(self: list) -> T: pass + def clear(self: list) -> t.CInt: pass + def __iter__(self: list) -> list[T] | t.CPtr: pass + def __next__(self: list) -> T: pass \ No newline at end of file diff --git a/TransPyV/temp/f5522571bcce7bcb.pyi b/TransPyV/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/TransPyV/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/TransPyV/temp/f6b51804a0ba8ff0.pyi b/TransPyV/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/TransPyV/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/c.py b/c.py new file mode 100644 index 0000000..d382954 --- /dev/null +++ b/c.py @@ -0,0 +1 @@ +from lib.includes.c import * \ No newline at end of file diff --git a/find.py b/find.py new file mode 100644 index 0000000..1af572b --- /dev/null +++ b/find.py @@ -0,0 +1,163 @@ +""" +REPL 模式文件内容搜索工具 +支持正则或纯文本搜索，默认搜索当前目录 +""" +import os +import re +import sys + +# 硬编码排除目录 +EXCLUDE_DIRS = { + '.git', '.vs', 'node_modules', '__pycache__', '.idea', + 'miniprogram_npm', '_backup_pre_redesign', +} + +# 排除的文件扩展名 +EXCLUDE_EXTS = { + '.png', '.jpg', '.jpeg', '.gif', '.svg', '.ico', '.bmp', + '.mp3', '.mp4', '.wav', '.avi', + '.zip', '.rar', '.7z', '.tar', '.gz', + '.exe', '.dll', '.so', '.dylib', + '.pyc', '.pyo', '.class', '.o', '.obj', + '.db', '.sqlite', '.vsidx', +} + +# 默认搜索根目录 +DEFAULT_ROOT = os.path.dirname(os.path.abspath(__file__)) + + +def should_skip(dirpath): + """判断目录是否应跳过""" + parts = dirpath.replace('\\', '/').split('/') + return any(p in EXCLUDE_DIRS for p in parts) + + +def search(pattern, root=DEFAULT_ROOT, use_regex=True, max_results=50): + """搜索文件内容""" + if use_regex: + try: + compiled = re.compile(pattern, re.IGNORECASE) + except re.error as e: + print(f' 正则语法错误: {e}') + return + + count = 0 + for dirpath, dirnames, filenames in os.walk(root): + # 就地修改 dirnames 以跳过排除目录 + dirnames[:] = [d for d in dirnames if d not in EXCLUDE_DIRS] + + for fname in filenames: + ext = os.path.splitext(fname)[1].lower() + if ext in EXCLUDE_EXTS: + continue + + fpath = os.path.join(dirpath, fname) + relpath = os.path.relpath(fpath, root) + try: + with open(fpath, 'r', encoding='utf-8', errors='ignore') as f: + for lineno, line in enumerate(f, 1): + line_stripped = line.rstrip('\n\r') + if use_regex: + match = compiled.search(line_stripped) + if match: + highlight = line_stripped[:match.start()] + \ + f'>>> {match.group()} <<<' + \ + line_stripped[match.end():] + print(f' {relpath}:{lineno} {highlight}') + count += 1 + else: + if pattern.lower() in line_stripped.lower(): + print(f' {relpath}:{lineno} {line_stripped}') + count += 1 + + if count >= max_results: + print(f'\n ... 已达上限 {max_results} 条，停止') + return + except (PermissionError, OSError): + continue + + if count == 0: + print(' 未找到匹配内容') + else: + print(f'\n 共 {count} 条匹配') + + +def main(): + print('=== 文件内容搜索工具 ===') + print(f'搜索根目录: {DEFAULT_ROOT}') + print(f'排除目录: {EXCLUDE_DIRS}') + print() + print('命令:') + print(' /regex 正则搜索') + print(' /text 纯文本搜索') + print(' /max 设置结果上限(默认50)') + print(' /dir 切换搜索目录') + print(' /help 显示帮助') + print(' /quit 退出') + print() + print('直接输入内容默认使用【纯文本】搜索，需要正则请加 /regex 前缀') + print() + + root = DEFAULT_ROOT + use_regex = False # 修改1：默认纯文本 + max_results = 50 + + while True: + try: + raw = input('find> ').strip() + except (EOFError, KeyboardInterrupt): + print('\n再见') + break + + if not raw: + continue + + if raw == '/quit': + print('再见') + break + + if raw == '/help': + print(' /regex 正则搜索') + print(' /text 纯文本搜索') + print(' /max 设置结果上限') + print(' /dir 切换搜索目录') + print(' /help 显示帮助') + print(' /quit 退出') + print(' 直接输入内容默认纯文本搜索') + continue + + if raw.startswith('/regex '): + pattern = raw[7:].strip() + if pattern: + search(pattern, root, use_regex=True, max_results=max_results) + continue + + if raw.startswith('/text '): + pattern = raw[6:].strip() + if pattern: + search(pattern, root, use_regex=False, max_results=max_results) + continue + + if raw.startswith('/max '): + try: + max_results = int(raw[5:].strip()) + print(f' 结果上限设为 {max_results}') + except ValueError: + print(' 请输入数字') + continue + + if raw.startswith('/dir '): + new_dir = raw[5:].strip() + if os.path.isdir(new_dir): + root = os.path.abspath(new_dir) + print(f' 搜索目录切换为: {root}') + else: + print(f' 目录不存在: {new_dir}') + continue + + # 修改2：直接输入走纯文本搜索 + search(raw, root, use_regex=False, max_results=max_results) + + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/includes/__builtin.py.bak b/includes/__builtin.py.bak new file mode 100644 index 0000000..7226c3f --- /dev/null +++ b/includes/__builtin.py.bak @@ -0,0 +1,425 @@ +# GCC/Clang 内建函数声明 +# 所有函数均为仅声明（t.State 等价于 CExtern | CExport），由编译器/运行时提供实现 +import t + + +# ============================================================ +# 随机数生成 (RDRAND / RDSEED) +# ============================================================ + +def __builtin_ia32_rdrand16_step(p: t.CUInt16T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_rdrand32_step(p: t.CUInt32T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_rdrand64_step(p: t.CUInt64T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_rdseed16_step(p: t.CUInt16T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_rdseed32_step(p: t.CUInt32T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_rdseed64_step(p: t.CUInt64T | t.CPtr) -> t.CInt | t.State: pass + + +# ============================================================ +# 时间戳计数器 (RDTSC / RDTSCP) +# ============================================================ + +def __builtin_ia32_rdtsc() -> t.CUInt64T | t.State: pass +def __builtin_ia32_rdtscp(p: t.CUInt32T | t.CPtr) -> t.CUInt64T | t.State: pass +def __builtin_ia32_rdpmc(counter: t.CUInt32T) -> t.CUInt64T | t.State: pass + + +# ============================================================ +# CPUID +# ============================================================ + +def __builtin_ia32_cpuid(level: t.CUInt32T, a: t.CUInt32T | t.CPtr, b: t.CUInt32T | t.CPtr, c: t.CUInt32T | t.CPtr, d: t.CUInt32T | t.CPtr) -> t.CUInt32T | t.State: pass +def __builtin_ia32_cpuid_count(level: t.CUInt32T, count: t.CUInt32T, a: t.CUInt32T | t.CPtr, b: t.CUInt32T | t.CPtr, c: t.CUInt32T | t.CPtr, d: t.CUInt32T | t.CPtr) -> t.CUInt32T | t.State: pass + + +# ============================================================ +# 内存屏障 (Fence) +# ============================================================ + +def __builtin_ia32_lfence() -> t.State: pass +def __builtin_ia32_mfence() -> t.State: pass +def __builtin_ia32_sfence() -> t.State: pass +def __builtin_ia32_pause() -> t.State: pass + + +# ============================================================ +# 位操作 (POPCNT / BSF / BSR / TZCNT / LZCNT / BEXTR / PDEP / PEXT / ANDN) +# ============================================================ + +def __builtin_ia32_popcntsi(x: t.CUInt32T) -> t.CInt | t.State: pass +def __builtin_ia32_popcntdi(x: t.CUInt64T) -> t.CInt | t.State: pass +def __builtin_ia32_popcntqi(x: t.CUInt64T) -> t.CInt | t.State: pass + +def __builtin_ia32_bsfsi(src: t.CUInt32T, dst: t.CUInt32T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_bsrsi(src: t.CUInt32T, dst: t.CUInt32T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_bsfqi(src: t.CUInt64T, dst: t.CUInt64T | t.CPtr) -> t.CInt | t.State: pass +def __builtin_ia32_bsrqi(src: t.CUInt64T, dst: t.CUInt64T | t.CPtr) -> t.CInt | t.State: pass + +def __builtin_ia32_tzcntsi(x: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_tzcntdi(x: t.CUInt64T) -> t.CUInt64T | t.State: pass +def __builtin_ia32_lzcntsi(x: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_lzcntdi(x: t.CUInt64T) -> t.CUInt64T | t.State: pass + +def __builtin_ia32_bextrsi(src: t.CUInt32T, start: t.CUInt32T, len: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_bextrdi(src: t.CUInt64T, start: t.CUInt32T, len: t.CUInt32T) -> t.CUInt64T | t.State: pass + +def __builtin_ia32_pdep_si(src: t.CUInt32T, mask: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_pext_si(src: t.CUInt32T, mask: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_pdep_di(src: t.CUInt64T, mask: t.CUInt64T) -> t.CUInt64T | t.State: pass +def __builtin_ia32_pext_di(src: t.CUInt64T, mask: t.CUInt64T) -> t.CUInt64T | t.State: pass + +def __builtin_ia32_andn_si(a: t.CUInt32T, b: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_andn_di(a: t.CUInt64T, b: t.CUInt64T) -> t.CUInt64T | t.State: pass + + +# ============================================================ +# 字节序交换 (BSWAP) +# ============================================================ + +def __builtin_bswap16(x: t.CUInt16T) -> t.CUInt16T | t.State: pass +def __builtin_bswap32(x: t.CUInt32T) -> t.CUInt32T | t.State: pass +def __builtin_bswap64(x: t.CUInt64T) -> t.CUInt64T | t.State: pass + + +# ============================================================ +# 缓存操作 (Cache) +# ============================================================ + +def __builtin_ia32_clflush(p: t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_clflushopt(p: t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_clwb(p: t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_clzero() -> t.State: pass + + +# ============================================================ +# 预取 (Prefetch) +# ============================================================ + +def __builtin_prefetch(addr: t.CConst | t.CPtr, rw: t.CInt = 0, locality: t.CInt = 3) -> t.State: pass +def __builtin_ia32_prefetch0(p: t.CChar | t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_prefetch1(p: t.CChar | t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_prefetch2(p: t.CChar | t.CConst | t.CPtr) -> t.State: pass +def __builtin_ia32_prefetchnta(p: t.CChar | t.CConst | t.CPtr) -> t.State: pass + + +# ============================================================ +# 非临时存储 (Non-Temporal Store) +# ============================================================ + +def __builtin_ia32_movnti(p: t.CUInt32T | t.CPtr, val: t.CUInt32T) -> t.State: pass +def __builtin_ia32_movnti64(p: t.CUInt64T | t.CPtr, val: t.CUInt64T) -> t.State: pass +def __builtin_ia32_movntdq(p: t.CPtr, val: t.CPtr) -> t.State: pass +def __builtin_ia32_movntps(p: t.CFloat | t.CPtr, val: t.CFloat) -> t.State: pass +def __builtin_ia32_movntpd(p: t.CDouble | t.CPtr, val: t.CDouble) -> t.State: pass + + +# ============================================================ +# I/O 端口操作 (Port I/O) +# ============================================================ + +def __builtin_ia32_inb(port: t.CUInt16T) -> t.CUInt8T | t.State: pass +def __builtin_ia32_inw(port: t.CUInt16T) -> t.CUInt16T | t.State: pass +def __builtin_ia32_inl(port: t.CUInt16T) -> t.CUInt32T | t.State: pass +def __builtin_ia32_outb(val: t.CUInt8T, port: t.CUInt16T) -> t.State: pass +def __builtin_ia32_outw(val: t.CUInt16T, port: t.CUInt16T) -> t.State: pass +def __builtin_ia32_outl(val: t.CUInt32T, port: t.CUInt16T) -> t.State: pass + + +# ============================================================ +# 模型特定寄存器 (MSR) +# ============================================================ + +def __builtin_ia32_rdmsr(msr: t.CUInt32T, val: t.CUInt64T | t.CPtr) -> t.State: pass +def __builtin_ia32_wrmsr(msr: t.CUInt32T, val: t.CUInt64T) -> t.State: pass + + +# ============================================================ +# 系统指令 (System) +# ============================================================ + +def __builtin_ia32_hlt() -> t.State: pass +def __builtin_ia32_invd() -> t.State: pass +def __builtin_ia32_wbinvd() -> t.State: pass +def __builtin_ia32_invlpg(p: t.CPtr) -> t.State: pass +def __builtin_ia32_clts() -> t.State: pass +def __builtin_ia32_cli() -> t.State: pass +def __builtin_ia32_sti() -> t.State: pass +def __builtin_ia32_int3() -> t.State: pass +def __builtin_ia32_int(n: t.CUInt8T) -> t.State: pass +def __builtin_ia32_ud2() -> t.State: pass + + +# ============================================================ +# 描述符表 (GDT / IDT / LDT / TR) +# ============================================================ + +def __builtin_ia32_lgdt(p: t.CPtr) -> t.State: pass +def __builtin_ia32_sgdt(p: t.CPtr) -> t.State: pass +def __builtin_ia32_lidt(p: t.CPtr) -> t.State: pass +def __builtin_ia32_sidt(p: t.CPtr) -> t.State: pass +def __builtin_ia32_lldt(sel: t.CUInt32T) -> t.State: pass +def __builtin_ia32_sldt(p: t.CPtr) -> t.State: pass +def __builtin_ia32_ltr(sel: t.CUInt32T) -> t.State: pass +def __builtin_ia32_str(p: t.CPtr) -> t.State: pass + + +# ============================================================ +# 控制寄存器 (CR / MSW) +# ============================================================ + +def __builtin_ia32_lmsw(val: t.CUInt32T) -> t.State: pass +def __builtin_ia32_smsw(p: t.CPtr) -> t.State: pass + + +# ============================================================ +# 监控等待 (MONITOR / MWAIT) +# ============================================================ + +def __builtin_ia32_monitor(p: t.CPtr, extensions: t.CUInt32T, hints: t.CUInt32T) -> t.State: pass +def __builtin_ia32_mwait(extensions: t.CUInt32T, hints: t.CUInt32T) -> t.State: pass + + +# ============================================================ +# XSAVE 状态操作 +# ============================================================ + +def __builtin_ia32_xsave(p: t.CPtr, mask: t.CUInt64T) -> t.State: pass +def __builtin_ia32_xrstor(p: t.CConst | t.CPtr, mask: t.CUInt64T) -> t.State: pass +def __builtin_ia32_xsaveopt(p: t.CPtr, mask: t.CUInt64T) -> t.State: pass +def __builtin_ia32_xsavec(p: t.CPtr, mask: t.CUInt64T) -> t.State: pass +def __builtin_ia32_xrstors(p: t.CConst | t.CPtr, mask: t.CUInt64T) -> t.State: pass +def __builtin_ia32_xsaves(p: t.CPtr, mask: t.CUInt64T) -> t.State: pass + + +# ============================================================ +# 通用内建函数 (Generic Builtins) +# ============================================================ + +def __builtin_expect(exp: t.CLong, c: t.CLong) -> t.CLong | t.State: pass +def __builtin_unreachable() -> t.State: pass +def __builtin_trap() -> t.State: pass +def __builtin_abort() -> t.State: pass +def __builtin_assume_aligned(addr: t.CConst | t.CPtr, align: t.CSizeT, *args) -> t.CPtr | t.State: pass +def __builtin_constant_p(expr: t.CVoid) -> t.CInt | t.State: pass +def __builtin_types_compatible_p(t1: t.CVoid, t2: t.CVoid) -> t.CInt | t.State: pass +def __builtin_choose_expr(const_exp: t.CUInt32T, exp1: t.CVoid, exp2: t.CVoid) -> t.CPtr | t.State: pass +def __builtin_offsetof(typ: t.CVoid, member: t.CVoid) -> t.CSizeT | t.State: pass +def __builtin_function_start(func: t.CPtr) -> t.CPtr | t.State: pass + + +# ============================================================ +# 可变参数 (VaList) +# ============================================================ + +def __builtin_va_start(va: t.CPtr, *args) -> t.State: pass +def __builtin_va_end(va: t.CPtr) -> t.State: pass +def __builtin_va_arg(va: t.CPtr, typ: t.CVoid) -> t.State: pass +def __builtin_va_copy(dst: t.CPtr, src: t.CConst | t.CPtr) -> t.State: pass + + +# ============================================================ +# 栈与调用信息 (Stack / Frame) +# ============================================================ + +def __builtin_return_address(level: t.CUInt32T) -> t.CPtr | t.State: pass +def __builtin_frame_address(level: t.CUInt32T) -> t.CPtr | t.State: pass +def __builtin_alloca(size: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin_alloca_with_align(size: t.CSizeT, alignment: t.CSizeT) -> t.CPtr | t.State: pass + + +# ============================================================ +# 内存操作 (Memory) +# ============================================================ + +def __builtin_memcpy(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin_memset(s: t.CPtr, c: t.CInt, n: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin_memmove(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin_memcmp(s1: t.CConst | t.CPtr, s2: t.CConst | t.CPtr, n: t.CSizeT) -> t.CInt | t.State: pass +def __builtin_mempcpy(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin_memchr(s: t.CConst | t.CPtr, c: t.CInt, n: t.CSizeT) -> t.CPtr | t.State: pass + + +# ============================================================ +# 字符串操作 (String) +# ============================================================ + +def __builtin_strcpy(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strncpy(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, n: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strcat(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strncat(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, n: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strlen(s: t.CChar | t.CConst | t.CPtr) -> t.CSizeT | t.State: pass +def __builtin_strcmp(s1: t.CChar | t.CConst | t.CPtr, s2: t.CChar | t.CConst | t.CPtr) -> t.CInt | t.State: pass +def __builtin_strncmp(s1: t.CChar | t.CConst | t.CPtr, s2: t.CChar | t.CConst | t.CPtr, n: t.CSizeT) -> t.CInt | t.State: pass +def __builtin_strchr(s: t.CChar | t.CConst | t.CPtr, c: t.CInt) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strrchr(s: t.CChar | t.CConst | t.CPtr, c: t.CInt) -> t.CChar | t.CPtr | t.State: pass +def __builtin_strstr(haystack: t.CChar | t.CConst | t.CPtr, needle: t.CChar | t.CConst | t.CPtr) -> t.CChar | t.CPtr | t.State: pass + + +# ============================================================ +# 数学内建函数 (Math) +# ============================================================ + +def __builtin_sqrt(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_sqrtf(x: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_sqrtl(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_pow(x: t.CDouble, y: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_powf(x: t.CFloat, y: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_sin(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_sinf(x: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_cos(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_cosf(x: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_tan(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_tanf(x: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_atan(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_exp(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_log(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_log2(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_log10(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_fabs(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_fabsf(x: t.CFloat) -> t.CFloat | t.State: pass +def __builtin_abs(x: t.CInt) -> t.CInt | t.State: pass +def __builtin_labs(x: t.CLong) -> t.CLong | t.State: pass +def __builtin_llabs(x: t.CInt64T) -> t.CInt64T | t.State: pass +def __builtin_fma(x: t.CDouble, y: t.CDouble, z: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_floor(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_ceil(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_trunc(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_round(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_nearbyint(x: t.CDouble) -> t.CDouble | t.State: pass +def __builtin_rint(x: t.CDouble) -> t.CDouble | t.State: pass + + +# ============================================================ +# 位计数内建函数 (Bit Counting) +# ============================================================ + +def __builtin_ffs(x: t.CUInt32T) -> t.CInt | t.State: pass +def __builtin_clz(x: t.CUInt32T) -> t.CInt | t.State: pass +def __builtin_ctz(x: t.CUInt32T) -> t.CInt | t.State: pass +def __builtin_popcount(x: t.CUInt32T) -> t.CInt | t.State: pass +def __builtin_parity(x: t.CUInt32T) -> t.CInt | t.State: pass + +def __builtin_ffsl(x: t.CULong) -> t.CInt | t.State: pass +def __builtin_clzl(x: t.CULong) -> t.CInt | t.State: pass +def __builtin_ctzl(x: t.CULong) -> t.CInt | t.State: pass +def __builtin_popcountl(x: t.CULong) -> t.CInt | t.State: pass +def __builtin_parityl(x: t.CULong) -> t.CInt | t.State: pass + +def __builtin_ffsll(x: t.CUInt64T) -> t.CInt | t.State: pass +def __builtin_clzll(x: t.CUInt64T) -> t.CInt | t.State: pass +def __builtin_ctzll(x: t.CUInt64T) -> t.CInt | t.State: pass +def __builtin_popcountll(x: t.CUInt64T) -> t.CInt | t.State: pass +def __builtin_parityll(x: t.CUInt64T) -> t.CInt | t.State: pass + + +# ============================================================ +# 对象大小检查 (Object size check) +# ============================================================ + +def __builtin_object_size(ptr: t.CConst | t.CPtr, typ: t.CInt) -> t.CSizeT | t.State: pass +def __builtin_dynamic_object_size(ptr: t.CConst | t.CPtr, typ: t.CInt) -> t.CSizeT | t.State: pass + +def __builtin___memcpy_chk(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT, objsize: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin___memmove_chk(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT, objsize: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin___memset_chk(s: t.CPtr, c: t.CInt, n: t.CSizeT, objsize: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin___mempcpy_chk(dst: t.CPtr, src: t.CConst | t.CPtr, n: t.CSizeT, objsize: t.CSizeT) -> t.CPtr | t.State: pass +def __builtin___strcpy_chk(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, objsize: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin___strncpy_chk(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, n: t.CSizeT, objsize: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin___strcat_chk(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, objsize: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin___strncat_chk(dst: t.CChar | t.CPtr, src: t.CChar | t.CConst | t.CPtr, n: t.CSizeT, objsize: t.CSizeT) -> t.CChar | t.CPtr | t.State: pass +def __builtin___snprintf_chk(s: t.CChar | t.CPtr, maxlen: t.CSizeT, flag: t.CInt, slen: t.CSizeT, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.State: pass +def __builtin___sprintf_chk(s: t.CChar | t.CPtr, flag: t.CInt, slen: t.CSizeT, fmt: t.CChar | t.CConst | t.CPtr, *args) -> t.CInt | t.State: pass + + +# ============================================================ +# 浮点分类 (FP Classification) +# ============================================================ + +def __builtin_isnan(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isinf(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isinf_sign(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isfinite(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isnormal(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_signbit(x: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isgreater(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isgreaterequal(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isless(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass +def __builtin_islessequal(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass +def __builtin_islessgreater(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass +def __builtin_isunordered(x: t.CDouble, y: t.CDouble) -> t.CInt | t.State: pass + + +# ============================================================ +# 特殊值 (Special Values) +# ============================================================ + +def __builtin_nan(s: t.CChar | t.CConst | t.CPtr) -> t.CDouble | t.State: pass +def __builtin_nanf(s: t.CChar | t.CConst | t.CPtr) -> t.CFloat | t.State: pass +def __builtin_nanl(s: t.CChar | t.CConst | t.CPtr) -> t.CDouble | t.State: pass +def __builtin_nans(s: t.CChar | t.CConst | t.CPtr) -> t.CDouble | t.State: pass +def __builtin_nansf(s: t.CChar | t.CConst | t.CPtr) -> t.CFloat | t.State: pass +def __builtin_inf() -> t.CDouble | t.State: pass +def __builtin_inff() -> t.CFloat | t.State: pass +def __builtin_infl() -> t.CDouble | t.State: pass +def __builtin_huge_val() -> t.CDouble | t.State: pass +def __builtin_huge_valf() -> t.CFloat | t.State: pass +def __builtin_huge_vall() -> t.CDouble | t.State: pass +def __builtin_inf_sign() -> t.CInt | t.State: pass + + +# ============================================================ +# 类型转换与溢出检查 (Overflow) +# ============================================================ + +def __builtin_add_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_sub_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_mul_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_sadd_overflow(a: t.CInt64T, b: t.CInt64T, res: t.CInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_ssub_overflow(a: t.CInt64T, b: t.CInt64T, res: t.CInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_smul_overflow(a: t.CInt64T, b: t.CInt64T, res: t.CInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_uadd_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_usub_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass +def __builtin_umul_overflow(a: t.CUInt64T, b: t.CUInt64T, res: t.CUInt64T | t.CPtr) -> t.CBool | t.State: pass + + +# ============================================================ +# MMX / SSE 状态 +# ============================================================ + +def __builtin_ia32_emms() -> t.State: pass +def __builtin_ia32_femms() -> t.State: pass + + +# ============================================================ +# 排队指令 (ENQCMD) +# ============================================================ + +def __builtin_ia32_enqcmd(dst: t.CPtr, src: t.CConst | t.CPtr) -> t.CBool | t.State: pass +def __builtin_ia32_enqcmds(dst: t.CPtr, src: t.CConst | t.CPtr) -> t.CBool | t.State: pass + + +# ============================================================ +# 用户级等待 (UMWAIT / UMONITOR / TPAUSE) +# ============================================================ + +def __builtin_ia32_umonitor(p: t.CPtr) -> t.State: pass +def __builtin_ia32_umwait(extensions: t.CUInt32T, counter: t.CUInt64T) -> t.CUInt8T | t.State: pass +def __builtin_ia32_tpause(extensions: t.CUInt32T, counter: t.CUInt64T) -> t.CUInt8T | t.State: pass + + +# ============================================================ +# 访问控制 (CLAC / STAC) +# ============================================================ + +def __builtin_ia32_clac() -> t.State: pass +def __builtin_ia32_stac() -> t.State: pass + + +# ============================================================ +# TLB 无效化 (INVLPG 变体) +# ============================================================ + +def __builtin_ia32_invlpgb(aux: t.CUInt64T, pcid: t.CUInt64T) -> t.State: pass +def __builtin_ia32_tlbsync() -> t.State: pass diff --git a/includes/_dict.py b/includes/_dict.py new file mode 100644 index 0000000..4f3f067 --- /dev/null +++ b/includes/_dict.py @@ -0,0 +1,271 @@ +import t, c +import memhub +import string +import json +from stdint import * + +# Variant 类型标签 (与 _variant.py 保持一致) +VARIANT_NONE: t.CDefine = 0 # 无类型/空值 +VARIANT_INT: t.CDefine = 1 # int (i64) +VARIANT_FLOAT: t.CDefine = 2 # float (f32) +VARIANT_DOUBLE: t.CDefine = 3 # double (f64) +VARIANT_STR: t.CDefine = 4 # str (i8*) +VARIANT_DICT: t.CDefine = 5 # dict* +VARIANT_LIST: t.CDefine = 6 # list* +VARIANT_BOOL: t.CDefine = 7 # bool (i1/i32) +VARIANT_PTR: t.CDefine = 8 # 自定义指针类型 + +# Variant 结构体大小: vtype(i32, 4 bytes) + padding(4 bytes) + ptr(i8*, 8 bytes) = 16 bytes +_VARIANT_SIZE: t.CDefine = 16 + + +class dict: + """堆上动态字典容器，基于 mbuddy 分配器。 + 键类型固定为 str (i8*)，值使用 Variant 存储 (类型标签 + 无类型指针)。 + 支持嵌套 dict/list 和 JSON 序列化。 + + 用法: + mb: memhub.MemBuddy = memhub.MemBuddy(arena, arena_size) + d: dict = dict(mb) + d.set_int("x", 42) + p: t.CPtr = d["x"] + v: int = c.Deref(p) + + # 嵌套 dict + inner: dict = dict(mb) + inner.set_int("a", 1) + d.set_dict("sub", inner) + sub: dict = c.Deref(d["sub"]) + av: int = c.Deref(sub["a"]) + """ + __keys__: t.CPtr + __values__: t.CPtr + __count__: t.CSizeT = 0 + __capacity__: t.CSizeT = 8 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + + def __new__(self, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: + # 堆分配 dict 对象 + return mb.alloc(dict.__sizeof__()) + + def __init__(self, mb: memhub.MemBuddy | t.CPtr): + self.__mbuddy__ = mb + self.__keys__ = mb.alloc(self.__capacity__ * 8) + self.__values__ = mb.alloc(self.__capacity__ * _VARIANT_SIZE) + + def __len__(self) -> t.CSizeT: + return self.__count__ + + def _find(self, key: str) -> t.CSizeT: + """查找 key 的索引，找不到返回 __count__""" + for i in range(self.__count__): + key_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(self.__keys__) + i * 8) + if key_ptr[0] == key: return i + return self.__count__ + + def __getitem__(self, key: str) -> t.CPtr: + """返回值的无类型指针。用户用 c.Deref 解引用获取实际值。 + 例: p: t.CPtr = d["x"]; v: int = c.Deref(p)""" + idx: t.CSizeT = self._find(key) + if idx >= self.__count__: return None + # Variant.ptr 在 Variant 结构体偏移 8 处 + val_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(self.__values__) + idx * _VARIANT_SIZE + 8) + return val_ptr[0] + + def __setitem__(self, key: str, val: t.CNeedPtr): + """存储值为 VARIANT_PTR。val 自动取地址 (t.CNeedPtr)。 + 注意: val 指向的值生命周期由调用者管理。 + 如需持久存储，请使用 set_int / set_str 等类型化方法。""" + self._set_variant_raw(key, VARIANT_PTR, val) + + def _set_variant_raw(self, key: str, vtype: t.CInt, ptr: t.CPtr): + """内部方法: 存储 Variant (vtype + ptr)。不自动取地址。""" + idx: t.CSizeT = self._find(key) + if idx < self.__count__: + # 更新已有键 + vtype_ptr: t.CInt | t.CPtr = t.CPtr(t.CUInt64T(self.__values__) + idx * _VARIANT_SIZE) + vtype_ptr[0] = vtype + val_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(self.__values__) + idx * _VARIANT_SIZE + 8) + val_ptr[0] = ptr + return + if self.__count__ >= self.__capacity__: + new_cap: t.CSizeT = self.__capacity__ * 2 + new_keys: t.CPtr = self.__mbuddy__.alloc(new_cap * 8) + new_values: t.CPtr = self.__mbuddy__.alloc(new_cap * _VARIANT_SIZE) + if new_keys is None or new_values is None: return + string.memcpy(new_keys, self.__keys__, self.__count__ * 8) + string.memcpy(new_values, self.__values__, self.__count__ * _VARIANT_SIZE) + self.__keys__ = new_keys + self.__values__ = new_values + self.__capacity__ = new_cap + key_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(self.__keys__) + self.__count__ * 8) + key_ptr[0] = key + vtype_ptr: t.CInt | t.CPtr = t.CPtr(t.CUInt64T(self.__values__) + self.__count__ * _VARIANT_SIZE) + vtype_ptr[0] = vtype + val_ptr: str | t.CPtr = t.CPtr(t.CUInt64T(self.__values__) + self.__count__ * _VARIANT_SIZE + 8) + val_ptr[0] = ptr + self.__count__ += 1 + + # ============================================================ + # 类型化 setter — 在 __mbuddy__ 中分配持久存储 + # ============================================================ + + def set_int(self, key: str, val: int): + """存储 int 值 (VARIANT_INT)。值复制到 __mbuddy__ 持久存储。""" + storage: t.CPtr = self.__mbuddy__.alloc(8) + int_ptr: int | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + int_ptr[0] = val + self._set_variant_raw(key, VARIANT_INT, storage) + + def set_str(self, key: str, val: str): + """存储 str 值 (VARIANT_STR)。直接存储字符串指针。""" + self._set_variant_raw(key, VARIANT_STR, val) + + def set_float(self, key: str, val: float): + """存储 float 值 (VARIANT_FLOAT)。值复制到 __mbuddy__ 持久存储。""" + storage: t.CPtr = self.__mbuddy__.alloc(8) + float_ptr: float | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + float_ptr[0] = val + self._set_variant_raw(key, VARIANT_FLOAT, storage) + + def set_double(self, key: str, val: t.CDouble): + """存储 double 值 (VARIANT_DOUBLE)。值复制到 __mbuddy__ 持久存储。""" + storage: t.CPtr = self.__mbuddy__.alloc(8) + dbl_ptr: t.CDouble | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + dbl_ptr[0] = val + self._set_variant_raw(key, VARIANT_DOUBLE, storage) + + def set_dict(self, key: str, val: dict | t.CPtr): + """存储嵌套 dict 指针 (VARIANT_DICT)。""" + self._set_variant_raw(key, VARIANT_DICT, val) + + def set_list(self, key: str, val: t.CPtr): + """存储嵌套 list 指针 (VARIANT_LIST)。""" + self._set_variant_raw(key, VARIANT_LIST, val) + + def set_bool(self, key: str, val: t.CInt): + """存储 bool 值 (VARIANT_BOOL)。值复制到 __mbuddy__ 持久存储。""" + storage: t.CPtr = self.__mbuddy__.alloc(8) + bool_ptr: t.CInt | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + bool_ptr[0] = val + self._set_variant_raw(key, VARIANT_BOOL, storage) + + # ============================================================ + # 查询方法 + # ============================================================ + + def get_vtype(self, key: str) -> t.CInt: + """返回 key 对应值的 Variant 类型标签。""" + idx: t.CSizeT = self._find(key) + if idx >= self.__count__: + return VARIANT_NONE + vtype_ptr: t.CInt | t.CPtr = t.CVoid(t.CUInt64T(self.__values__) + idx * _VARIANT_SIZE, t.CPtr) + return vtype_ptr[0] + + def get(self, key: str, default: t.CPtr) -> t.CPtr: + """返回值指针，不存在则返回 default。""" + idx: t.CSizeT = self._find(key) + if idx >= self.__count__: + return default + val_ptr: str | t.CPtr = t.CVoid(t.CUInt64T(self.__values__) + idx * _VARIANT_SIZE + 8, t.CPtr) + return val_ptr[0] + + def contains(self, key: str) -> t.CInt: + """检查 key 是否存在。""" + idx: t.CSizeT = self._find(key) + if idx < self.__count__: + return 1 + return 0 + + def __iter__(self) -> dict | t.CPtr: + self.__iter_index__ = 0 + return self + + def __next__(self) -> str: + if self.__iter_index__ >= self.__count__: + raise StopIteration + idx: t.CSizeT = self.__iter_index__ + self.__iter_index__ = idx + 1 + key_ptr: str | t.CPtr = t.CVoid(t.CUInt64T(self.__keys__) + idx * 8, t.CPtr) + return key_ptr[0] + + # ============================================================ + # JSON 处理：利用 includes/json 库实现序列化/反序列化 + # ============================================================ + + def to_json(self, pool: memhub.MemManager | t.CPtr) -> json.JsonValue | t.CPtr: + """将 dict 转换为 JsonValue 对象 (JSON_OBJECT)。 + 根据 Variant 类型标签自动选择合适的 JSON 值类型。""" + obj: json.JsonValue | t.CPtr = json.object(pool) + i: t.CSizeT = 0 + while i < self.__count__: + key_ptr: str | t.CPtr = t.CVoid(t.CUInt64T(self.__keys__) + i * 8, t.CPtr) + vtype_ptr: t.CInt | t.CPtr = t.CVoid(t.CUInt64T(self.__values__) + i * _VARIANT_SIZE, t.CPtr) + val_ptr: str | t.CPtr = t.CVoid(t.CUInt64T(self.__values__) + i * _VARIANT_SIZE + 8, t.CPtr) + vt: t.CInt = vtype_ptr[0] + vp: t.CPtr = val_ptr[0] + jv: json.JsonValue | t.CPtr = json.null(pool) + if jv != None: + if vt == VARIANT_INT: + jv.vtype = 2 # JSON_INT + ip: int | t.CPtr = t.CVoid(t.CUInt64T(vp), t.CPtr) + jv.int_val = ip[0] + elif vt == VARIANT_STR: + jv.vtype = 4 # JSON_STRING + jv.str_val = vp + elif vt == VARIANT_BOOL: + jv.vtype = 1 # JSON_BOOL + bp: t.CInt | t.CPtr = t.CVoid(t.CUInt64T(vp), t.CPtr) + jv.bool_val = bp[0] + elif vt == VARIANT_FLOAT: + jv.vtype = 3 # JSON_FLOAT + fp: float | t.CPtr = t.CVoid(t.CUInt64T(vp), t.CPtr) + jv.float_val = fp[0] + elif vt == VARIANT_PTR: + # 默认: 将 PTR 作为整数处理 + jv.vtype = 2 # JSON_INT + ip: int | t.CPtr = t.CVoid(t.CUInt64T(vp), t.CPtr) + jv.int_val = ip[0] + json.object_set(pool, obj, key_ptr[0], jv) + i += 1 + return obj + + def from_json(self, pool: memhub.MemManager | t.CPtr, root: json.JsonValue | t.CPtr): + """从 JsonValue 对象填充 dict。 + 根据 JSON 值类型自动提取并存储为对应 Variant 类型。""" + if root == None: + return + if root.vtype != 6: # JSON_OBJECT + return + cur: json.JsonValue | t.CPtr = root.child + while cur != None: + if cur.key != None: + if cur.vtype == 2: # JSON_INT + storage: t.CPtr = self.__mbuddy__.alloc(8) + ip: int | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + ip[0] = cur.int_val + self._set_variant_raw(cur.key, VARIANT_INT, storage) + elif cur.vtype == 4: # JSON_STRING + self._set_variant_raw(cur.key, VARIANT_STR, cur.str_val) + elif cur.vtype == 1: # JSON_BOOL + storage: t.CPtr = self.__mbuddy__.alloc(8) + bp: t.CInt | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + bp[0] = cur.bool_val + self._set_variant_raw(cur.key, VARIANT_BOOL, storage) + elif cur.vtype == 3: # JSON_FLOAT + storage: t.CPtr = self.__mbuddy__.alloc(8) + fp: float | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + fp[0] = cur.float_val + self._set_variant_raw(cur.key, VARIANT_FLOAT, storage) + cur = cur.next + + def dumps(self, pool: memhub.MemManager | t.CPtr) -> str: + """将 dict 序列化为 JSON 字符串。""" + obj: json.JsonValue | t.CPtr = self.to_json(pool) + return json.write(pool, obj, False) + + def loads(self, pool: memhub.MemManager | t.CPtr, json_str: str): + """从 JSON 字符串解析并填充 dict。""" + root: json.JsonValue | t.CPtr = json.parse(pool, json_str) + self.from_json(pool, root) diff --git a/includes/_fakeduck.py b/includes/_fakeduck.py new file mode 100644 index 0000000..602e441 --- /dev/null +++ b/includes/_fakeduck.py @@ -0,0 +1,319 @@ +import t +import memhub +import string + + +class _Type: + def type_id(self) -> int: + return 99 + + +class _str: + # 编译期元数据：声明此类希望从 with 上下文获取 __mbuddy__ 字段（非必要，找不到不报错） + __requires__: list[str] = ['__mbuddy__'] + __data__: str + __mbuddy__: memhub.MemBuddy | t.CPtr + + def upper(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + for i, ch in enumerate(src): + if ch >= 97: + if ch <= 122: + buf[i] = ch - 32 + else: + buf[i] = ch + else: + buf[i] = ch + buf[length] = 0 + return str(buf) + + def lower(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + for i, ch in enumerate(src): + if ch >= 65: + if ch <= 90: + buf[i] = ch + 32 + else: + buf[i] = ch + else: + buf[i] = ch + buf[length] = 0 + return str(buf) + + def capitalize(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + first: bool = True + for i, ch in enumerate(src): + if first: + first = False + if ch >= 97: + if ch <= 122: + buf[i] = ch - 32 + else: + buf[i] = ch + else: + buf[i] = ch + else: + if ch >= 65: + if ch <= 90: + buf[i] = ch + 32 + else: + buf[i] = ch + else: + buf[i] = ch + buf[length] = 0 + return str(buf) + + def swapcase(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + for i, ch in enumerate(src): + if ch >= 97: + if ch <= 122: + buf[i] = ch - 32 + else: + buf[i] = ch + else: + if ch >= 65: + if ch <= 90: + buf[i] = ch + 32 + else: + buf[i] = ch + else: + buf[i] = ch + buf[length] = 0 + return str(buf) + + def title(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + prev_alpha: bool = False + for i, ch in enumerate(src): + cur_alpha: bool = False + if ch >= 65: + if ch <= 90: + cur_alpha = True + else: + if ch >= 97: + if ch <= 122: + cur_alpha = True + if cur_alpha: + if prev_alpha: + if ch >= 65: + if ch <= 90: + buf[i] = ch + 32 + else: + buf[i] = ch + else: + buf[i] = ch + else: + if ch >= 97: + if ch <= 122: + buf[i] = ch - 32 + else: + buf[i] = ch + else: + buf[i] = ch + else: + buf[i] = ch + prev_alpha = cur_alpha + buf[length] = 0 + return str(buf) + + def casefold(self) -> str: + src: str = self.__data__ + length: t.CSizeT = len(src) + buf: bytes = self.__mbuddy__.alloc(length + 1) + if buf == None: return self.__data__ + for i, ch in enumerate(src): + if ch >= 65: + if ch <= 90: + buf[i] = ch + 32 + else: + buf[i] = ch + else: + buf[i] = ch + buf[length] = 0 + return str(buf) + + def isupper(self) -> bool: + src: str = self.__data__ + has_alpha: bool = False + for ch in src: + if ch >= 65: + if ch <= 90: + has_alpha = True + else: + if ch >= 97: + if ch <= 122: + return False + return has_alpha + + def islower(self) -> bool: + src: str = self.__data__ + has_alpha: bool = False + for ch in src: + if ch >= 97: + if ch <= 122: + has_alpha = True + if ch >= 65: + if ch <= 90: + return False + return has_alpha + + def isalpha(self) -> bool: + src: str = self.__data__ + has_char: bool = False + for ch in src: + has_char = True + cur_alpha: bool = False + if ch >= 65: + if ch <= 90: + cur_alpha = True + else: + if ch >= 97: + if ch <= 122: + cur_alpha = True + if not cur_alpha: + return False + return has_char + + def isdigit(self) -> bool: + src: str = self.__data__ + has_char: bool = False + for ch in src: + has_char = True + if ch >= 48: + if ch > 57: + return False + else: + return False + return has_char + + def isalnum(self) -> bool: + src: str = self.__data__ + has_char: bool = False + for ch in src: + has_char = True + ok: bool = False + if ch >= 48: + if ch <= 57: + ok = True + if not ok: + if ch >= 65: + if ch <= 90: + ok = True + else: + if ch >= 97: + if ch <= 122: + ok = True + if not ok: + return False + return has_char + + def isspace(self) -> bool: + src: str = self.__data__ + has_char: bool = False + for ch in src: + has_char = True + is_space: bool = False + if ch == 32 or ch == 9 or ch == 10 or ch == 13 or ch == 11 or ch == 12: + is_space = True + if not is_space: + return False + return has_char + + def isascii(self) -> bool: + src: str = self.__data__ + has_char: bool = False + for ch in src: + has_char = True + if ch > 127: return False + return has_char + + def istitle(self) -> bool: + src: str = self.__data__ + has_char: bool = False + prev_alpha: bool = False + for ch in src: + has_char = True + cur_alpha: bool = False + if ch >= 65: + if ch <= 90: + cur_alpha = True + else: + if ch >= 97: + if ch <= 122: + cur_alpha = True + if cur_alpha: + if not prev_alpha: + if ch >= 97: + if ch <= 122: + return False + else: + if ch >= 65: + if ch <= 90: + return False + prev_alpha = cur_alpha + return has_char + + def split(self, delimiter: str) -> list[str]: + result = list[str](self.__mbuddy__) + src: str = self.__data__ + length: t.CSizeT = len(src) + delim_len: t.CSizeT = string.strlen(delimiter) + start: t.CSizeT = 0 + i: t.CSizeT = 0 + cur: str = src + sub_len: t.CSizeT = 0 + sub: bytes = None + src_off: bytes = None + term: bytes = None + if delim_len == 0: + sub_len = length + sub = self.__mbuddy__.alloc(sub_len + 1) + if sub != None: + src_off = bytes(t.CUInt64T(src) + start) + string.memcpy(sub, src_off, sub_len) + term = sub + term[sub_len] = 0 + result.append(str(sub)) + return result + while i + delim_len <= length: + if string.strncmp(cur, delimiter, delim_len) == 0: + sub_len = i - start + sub = self.__mbuddy__.alloc(sub_len + 1) + if sub != None: + src_off = bytes(t.CUInt64T(src) + start) + string.memcpy(sub, src_off, sub_len) + term = sub + term[sub_len] = 0 + result.append(str(sub)) + start = i + delim_len + cur += delim_len + i += delim_len + else: + cur += 1 + i += 1 + sub_len = length - start + sub = self.__mbuddy__.alloc(sub_len + 1) + if sub != None: + src_off = bytes(t.CUInt64T(src) + start) + string.memcpy(sub, src_off, sub_len) + term = sub + term[sub_len] = 0 + result.append(str(sub)) + return result diff --git a/includes/_list.py b/includes/_list.py new file mode 100644 index 0000000..8176346 --- /dev/null +++ b/includes/_list.py @@ -0,0 +1,90 @@ +import t, c +import memhub +import string +from stdint import * + + +class list[T]: + """堆上动态列表容器，基于 memhub.MemManager 多态分配器。 + + 用法: + pool: memhub.MemManager = memhub.MemBuddy(arena, arena_size) + nums: list[int] = list[int](pool) + nums.append(42) + x: int = nums.get(0) + + 对于指针类型（如 AST | t.CPtr），传 elem_size=8 避免对象切片: + lst: list[AST | t.CPtr] = list[AST | t.CPtr](pool, 8) + """ + __data__: t.CVoid | t.CPtr + __count__: t.CSizeT + __capacity__: t.CSizeT + __pool__: memhub.MemManager | t.CPtr + __elem_size__: t.CSizeT + __iter_index__: t.CSizeT + + def __new__(self, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT = 0): + # 堆分配 list[T] 对象 (6 字段 × 8 = 48 字节，无 vtable) + return pool.alloc(48) + + def __init__(self, pool: memhub.MemManager | t.CPtr, elem_size: t.CSizeT = 0): + self.__pool__ = pool + self.__count__ = 0 + self.__capacity__ = 8 + if elem_size == 0: + self.__elem_size__ = T.__sizeof__() + else: + self.__elem_size__ = elem_size + self.__data__ = pool.alloc(self.__capacity__ * self.__elem_size__) + self.__iter_index__ = 0 + + def __len__(self) -> t.CSizeT: + return self.__count__ + + def append(self, item: T): + if self.__count__ >= self.__capacity__: + new_cap: t.CSizeT = self.__capacity__ * 2 + new_data: t.CPtr = self.__pool__.alloc(new_cap * self.__elem_size__) + if new_data == None: return + string.memcpy(new_data, self.__data__, self.__count__ * self.__elem_size__) + self.__data__ = new_data + self.__capacity__ = new_cap + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + self.__count__ * self.__elem_size__) + elem_ptr[0] = item + self.__count__ += 1 + + def get(self, index: t.CSizeT) -> T: + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__) + return elem_ptr[0] + + def __getitem__(self, index: t.CSizeT) -> T: + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__) + return elem_ptr[0] + + def set(self, index: t.CSizeT, value: T): + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__) + elem_ptr[0] = value + + def __setitem__(self, index: t.CSizeT, value: T): + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + index * self.__elem_size__) + elem_ptr[0] = value + + def pop(self) -> T: + self.__count__ -= 1 + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + self.__count__ * self.__elem_size__) + return elem_ptr[0] + + def clear(self): + self.__count__ = 0 + + def __iter__(self) -> list[T] | t.CPtr: + self.__iter_index__ = 0 + return self + + def __next__(self) -> T: + if self.__iter_index__ >= self.__count__: + raise StopIteration + idx: t.CSizeT = self.__iter_index__ + self.__iter_index__ = idx + 1 + elem_ptr: T | t.CPtr = t.CPtr(t.CUInt64T(self.__data__) + idx * self.__elem_size__) + return elem_ptr[0] diff --git a/includes/_variant.py b/includes/_variant.py new file mode 100644 index 0000000..6cc06b3 --- /dev/null +++ b/includes/_variant.py @@ -0,0 +1,64 @@ +import t, c +from stdint import * + +# Variant 类型标签 (固定枚举 + 自定义 PTR) +VARIANT_NONE: t.CDefine = 0 # 无类型/空值 +VARIANT_INT: t.CDefine = 1 # int (i64) +VARIANT_FLOAT: t.CDefine = 2 # float (f32) +VARIANT_DOUBLE: t.CDefine = 3 # double (f64) +VARIANT_STR: t.CDefine = 4 # str (i8*) +VARIANT_DICT: t.CDefine = 5 # dict* +VARIANT_LIST: t.CDefine = 6 # list* +VARIANT_BOOL: t.CDefine = 7 # bool (i1/i32) +VARIANT_PTR: t.CDefine = 8 # 自定义指针类型 + + +class Variant: + """通用变体类型，存储类型标签 + 无类型指针。 + 用于 dict/list 的万能对象存储，支持嵌套和 JSON 序列化。 + + 用法: + v: Variant = Variant(VARIANT_INT, t.CNeedPtr(42)) + v2: Variant = Variant(VARIANT_STR, t.CNeedPtr("hello")) + + # 获取值 + ptr: t.CPtr = v.ptr + i: int = c.Deref(ptr) + """ + vtype: t.CInt + ptr: t.CPtr + + def __init__(self, vtype: t.CInt, val: t.CNeedPtr): + """构造 Variant。 + val 使用 t.CNeedPtr 自动取地址（非指针值自动 alloca+store）。 + 注意: val 的生命周期由调用者管理，Variant 仅存储指针。 + """ + self.vtype = vtype + self.ptr = val + + def is_none(self) -> t.CInt: + return self.vtype == VARIANT_NONE + + def is_int(self) -> t.CInt: + return self.vtype == VARIANT_INT + + def is_float(self) -> t.CInt: + return self.vtype == VARIANT_FLOAT + + def is_double(self) -> t.CInt: + return self.vtype == VARIANT_DOUBLE + + def is_str(self) -> t.CInt: + return self.vtype == VARIANT_STR + + def is_dict(self) -> t.CInt: + return self.vtype == VARIANT_DICT + + def is_list(self) -> t.CInt: + return self.vtype == VARIANT_LIST + + def is_bool(self) -> t.CInt: + return self.vtype == VARIANT_BOOL + + def is_ptr(self) -> t.CInt: + return self.vtype == VARIANT_PTR diff --git a/includes/_withcontent.py b/includes/_withcontent.py new file mode 100644 index 0000000..42147cb --- /dev/null +++ b/includes/_withcontent.py @@ -0,0 +1,67 @@ +import t, c +from stdint import * +from w32.win32process import TlsAlloc, TlsGetValue, TlsSetValue +import stdlib +import stdio +import string + + +# 最大提供者栈深度（编译器会检查是否够用，不够再往上提） +_MAX_PROVIDER_DEPTH: t.CDefine = 32 + + +@t.Object +class _ProviderStack: + """提供者栈，存储字段名和指针对。""" + names: t.CArray[VOIDPTR, 32] + ptrs: t.CArray[VOIDPTR, 32] + top: UINT64 + + +# TLS 索引（全局，所有线程共享同一个索引） +_tls_index: ULONG = 0 + + +@t.TLS +def _init_tls(): + """初始化 TLS 索引（一次性执行，后续调用跳过）。""" + global _tls_index + _tls_index = TlsAlloc() + + +def _get_stack() -> _ProviderStack | t.CPtr: + """获取当前线程的提供者栈，不存在则分配。""" + _init_tls() + stack: _ProviderStack | t.CPtr = TlsGetValue(_tls_index) + if stack == None: + stack = stdlib.malloc(_ProviderStack.__sizeof__()) + string.memset(stack, 0, _ProviderStack.__sizeof__()) + TlsSetValue(_tls_index, stack) + return stack + + +def _push_provider(field_name: str, ptr: t.CPtr): + """压入提供者 (field_name, ptr) 到栈中。""" + stack: _ProviderStack | t.CPtr = _get_stack() + if stack.top >= _MAX_PROVIDER_DEPTH: + stdio.printf("[with] ERROR: provider stack overflow (depth=%lu, max=%d), field '%s' not injected\n", stack.top, _MAX_PROVIDER_DEPTH, field_name) + return + stack.names[stack.top] = field_name + stack.ptrs[stack.top] = ptr + stack.top = stack.top + 1 + + +def _pop_provider(): + """弹出栈顶提供者。""" + stack: _ProviderStack | t.CPtr = _get_stack() + if stack.top > 0: + stack.top = stack.top - 1 + + +def _find_provider(field_name: str) -> t.CPtr: + """从栈顶向下查找匹配 field_name 的提供者，返回对应的指针。找不到返回 NULL。""" + stack: _ProviderStack | t.CPtr = _get_stack() + for i in range(stack.top - 1, -1, -1): + if stack.names[i] == field_name: + return stack.ptrs[i] + return None diff --git a/includes/argparse.py b/includes/argparse.py new file mode 100644 index 0000000..67ffc05 --- /dev/null +++ b/includes/argparse.py @@ -0,0 +1,458 @@ +import t, c +from stdint import * +import string +import memhub +import stdio +import hashtable + + +# ============================================================ +# argparse.py — 命令行参数解析库（Python argparse 风格） +# +# 用法: +# argparse._mbuddy = mb +# parser = argparse.ArgumentParser("myprog", "A test", pool=mb) +# parser.add_argument("--input", "-i", arg_type=argparse.STRING, +# required=True, help="input file") +# parser.add_argument("--count", "-c", arg_type=argparse.INT, +# default=0, help="repeat count") +# parser.add_argument("--verbose", "-v", action=argparse.STORE_TRUE, +# help="verbose mode") +# parser.add_argument("output", arg_type=argparse.STRING, +# help="output path") +# args = parser.parse_args(argc, argv) +# if args is not None: +# inp: str = args.get_str("input") +# cnt: INT = args.get_int("count") +# verb: bool = args.get_bool("verbose") +# out: str = args.get_str("output") +# argparse.release(args) +# ============================================================ + +# 类型常量 +STRING: t.CDefine = 0 +INT: t.CDefine = 1 +BOOL: t.CDefine = 2 + +# 动作常量 +STORE: t.CDefine = 0 +STORE_TRUE: t.CDefine = 1 +STORE_FALSE: t.CDefine = 2 +COUNT: t.CDefine = 3 + +# 限制 +MAX_ARGS: t.CDefine = 32 + +# 模块级 mbuddy 指针（由用户在 main 中赋值） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================ +# Argument — 参数定义结构体 +# ============================================================ + +class Argument: + name: str # "--input" 或 "output"（位置参数） + short_name: str # "-i" 或 None + help_text: str # 帮助文本 + arg_type: INT # STRING, INT, BOOL + action: INT # STORE, STORE_TRUE, STORE_FALSE, COUNT + default_int: INT # INT 类型默认值 + default_str: str # STRING 类型默认值 + required: bool # 是否必须 + is_positional: bool # 是否位置参数 + dest: str # 查找键名（"--input" → "input"） + + +# ============================================================ +# ParsedArgs — 解析结果 +# ============================================================ + +class ParsedArgs: + _ht: hashtable.HashTable | t.CPtr + __mbuddy__: memhub.MemBuddy | t.CPtr + + def __new__(self, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: + return mb.alloc(16) + + def __init__(self, mb: memhub.MemBuddy | t.CPtr): + self.__mbuddy__ = mb + self._ht = hashtable.HashTable(mb) + + def get_str(self, name: str) -> str: + """获取字符串值。键不存在返回 None。""" + if self._ht.__contains__(name) == 0: return None + return self._ht[name] + + def get_int(self, name: str) -> INT: + """获取整数值。键不存在返回 0。""" + if self._ht.__contains__(name) == 0: return 0 + ptr: t.CPtr = self._ht[name] + return c.Deref(t.CInt(t.CUInt64T(ptr), t.CPtr)) + + def get_bool(self, name: str) -> INT: + """获取布尔值（返回 1 或 0）。键不存在返回 0。""" + if self._ht.__contains__(name) == 0: return 0 + ptr: t.CPtr = self._ht[name] + return 1 if c.Deref(t.CInt(t.CUInt64T(ptr), t.CPtr)) != 0 else 0 + + def has(self, name: str) -> bool: + """检查键是否存在。""" + return self._ht.__contains__(name) != 0 + + def _set_str(self, name: str, val: str): + self._ht.set_str(name, val) + + def _set_int(self, name: str, val: INT): + self._ht.set_int(name, val) + + def _set_bool(self, name: str, val: bool): + if val: + self._ht.set_int(name, 1) + else: + self._ht.set_int(name, 0) + + +# ============================================================ +# ArgumentParser — 命令行参数解析器 +# ============================================================ + +class ArgumentParser: + _prog: str + _description: str + _args: Argument | t.CPtr + _arg_count: INT + __mbuddy__: memhub.MemBuddy | t.CPtr + + def __new__(self, prog: str = "", description: str = "", + pool: memhub.MemBuddy | t.CPtr = None) -> t.CPtr: + mb: memhub.MemBuddy | t.CPtr = pool + if mb is None: mb = _mbuddy + return mb.alloc(40) + + def __init__(self, prog: str = "", description: str = "", + pool: memhub.MemBuddy | t.CPtr = None): + mb: memhub.MemBuddy | t.CPtr = pool + if mb is None: mb = _mbuddy + self.__mbuddy__ = mb + self._prog = prog + self._description = description + self._arg_count = 0 + self._args = mb.alloc(MAX_ARGS * Argument.__sizeof__()) + + def add_argument(self, name: str = "", short: str = None, + arg_type: INT = 0, default: INT = 0, + default_str: str = None, required: bool = False, + action: INT = 0, help: str = None): + """注册一个命令行参数。 + + Args: + name: 参数名（"--input" 或 "output"） + short: 短选项（"-i"），可选 + arg_type: 类型（STRING/INT/BOOL） + default: INT 默认值 + default_str: STRING 默认值 + required: 是否必须 + action: 动作（STORE/STORE_TRUE/STORE_FALSE/COUNT） + help: 帮助文本 + """ + if self._arg_count >= MAX_ARGS: return + idx: INT = self._arg_count + arg: Argument | t.CPtr = t.CPtr(t.CUInt64T(self._args) + idx * Argument.__sizeof__()) + arg.name = name + arg.short_name = short + arg.help_text = help + arg.arg_type = arg_type + arg.action = action + arg.default_int = default + arg.default_str = default_str + arg.required = required + # 判断是否位置参数 + if name is not None and name[0] != '-': + arg.is_positional = True + arg.dest = name + else: + arg.is_positional = False + if name is not None and name[0] == '-' and name[1] == '-': + arg.dest = name + 2 + else: + arg.dest = name + 1 + self._arg_count = idx + 1 + + def _get_arg(self, idx: INT) -> Argument | t.CPtr: + return t.CPtr(t.CUInt64T(self._args) + idx * Argument.__sizeof__()) + + def _find_long(self, token: str) -> INT: + """精确匹配长选项名""" + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + if string.strcmp(arg.name, token) == 0: + return i + i += 1 + return -1 + + def _find_long_prefix(self, token: str) -> INT: + """匹配 --name=value 中的 name 前缀""" + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + name_len: INT = string.strlen(arg.name) + if string.strncmp(token, arg.name, name_len) == 0: + if token[name_len] == '=': + return i + i += 1 + return -1 + + def _find_short(self, token: str) -> INT: + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + if arg.short_name is not None: + if string.strcmp(arg.short_name, token) == 0: + return i + i += 1 + return -1 + + def _find_positional(self, pos_idx: INT) -> INT: + count: INT = 0 + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + if arg.is_positional: + if count == pos_idx: + return i + count += 1 + i += 1 + return -1 + + def _store_value(self, args: ParsedArgs | t.CPtr, + arg: Argument | t.CPtr, value: str): + if arg.arg_type == INT: + args._set_int(arg.dest, string.atoi(value)) + elif arg.arg_type == STRING: + args._set_str(arg.dest, value) + elif arg.arg_type == BOOL: + v: INT = string.atoi(value) + if v != 0: + args._set_bool(arg.dest, True) + else: + args._set_bool(arg.dest, False) + + def _apply_action(self, args: ParsedArgs | t.CPtr, arg: Argument | t.CPtr): + if arg.action == STORE_TRUE: + args._set_bool(arg.dest, True) + elif arg.action == STORE_FALSE: + args._set_bool(arg.dest, False) + elif arg.action == COUNT: + cur: INT = args.get_int(arg.dest) + args._set_int(arg.dest, cur + 1) + + def parse_args(self, argc: INT, argv: str | t.CPtr) -> ParsedArgs | t.CPtr: + """解析命令行参数。 + + Args: + argc: 参数个数 + argv: 参数数组（char**，argv[0] 为程序名） + Returns: + ParsedArgs 指针，失败返回 None + """ + if self.__mbuddy__ is None: return None + + args: ParsedArgs | t.CPtr = ParsedArgs(self.__mbuddy__) + if args is None: return None + + # 初始化默认值 + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + if arg.action == STORE: + if arg.arg_type == INT: + args._set_int(arg.dest, arg.default_int) + elif arg.arg_type == STRING: + args._set_str(arg.dest, arg.default_str) + elif arg.action == STORE_TRUE: + args._set_bool(arg.dest, False) + elif arg.action == STORE_FALSE: + args._set_bool(arg.dest, True) + elif arg.action == COUNT: + args._set_int(arg.dest, 0) + i += 1 + + # 解析 argv + idx: INT = 1 + pos_idx: INT = 0 + token: str = None + found: INT = 0 + arg: Argument | t.CPtr = None + value: str = None + entry: str | t.CPtr = None + name_len: INT = 0 + + while idx < argc: + entry = t.CPtr(t.CUInt64T(argv) + idx * 8) + token = entry[0] + if token is None: break + + # "--" 结束符：后续全部为位置参数 + if token[0] == '-' and token[1] == '-' and token[2] == 0: + idx += 1 + while idx < argc: + entry = t.CPtr(t.CUInt64T(argv) + idx * 8) + token = entry[0] + found = self._find_positional(pos_idx) + if found < 0: + stdio.printf("error: unexpected argument '%s'\n", token) + release(args) + return None + arg = self._get_arg(found) + self._store_value(args, arg, token) + pos_idx += 1 + idx += 1 + break + + # 长选项 --name 或 --name=value + if token[0] == '-' and token[1] == '-': + # 检查是否含 = + has_eq: bool = False + p: str = token + while p[0] != 0: + if p[0] == '=': + has_eq = True + break + p += 1 + + if has_eq: + # --name=value + found = self._find_long_prefix(token) + if found < 0: + stdio.printf( + "error: unrecognized argument '%s'\n", token) + release(args) + return None + arg = self._get_arg(found) + # 提取 = 后的值 + arg_name_len: INT = string.strlen(arg.name) + value = token + arg_name_len + 1 + self._store_value(args, arg, value) + idx += 1 + else: + # --name（值在下一个 argv 或标志） + found = self._find_long(token) + if found < 0: + stdio.printf( + "error: unrecognized argument '%s'\n", token) + release(args) + return None + arg = self._get_arg(found) + if arg.action == STORE: + if idx + 1 >= argc: + stdio.printf( + "error: %s requires a value\n", arg.name) + release(args) + return None + entry = t.CPtr(t.CUInt64T(argv) + (idx + 1) * 8) + value = entry[0] + self._store_value(args, arg, value) + idx += 2 + else: + self._apply_action(args, arg) + idx += 1 + + # 短选项 -x + elif token[0] == '-': + found = self._find_short(token) + if found < 0: + stdio.printf( + "error: unrecognized argument '%s'\n", token) + release(args) + return None + arg = self._get_arg(found) + if arg.action == STORE: + if idx + 1 >= argc: + stdio.printf( + "error: %s requires a value\n", arg.name) + release(args) + return None + entry = t.CPtr(t.CUInt64T(argv) + (idx + 1) * 8) + value = entry[0] + self._store_value(args, arg, value) + idx += 2 + else: + self._apply_action(args, arg) + idx += 1 + + # 位置参数 + else: + found = self._find_positional(pos_idx) + if found < 0: + stdio.printf( + "error: unexpected argument '%s'\n", token) + release(args) + return None + arg = self._get_arg(found) + self._store_value(args, arg, token) + pos_idx += 1 + idx += 1 + + # 检查必须参数 + i = 0 + while i < self._arg_count: + arg = self._get_arg(i) + if arg.required: + if not args.has(arg.dest): + stdio.printf( + "error: missing required argument '%s'\n", + arg.name) + release(args) + return None + i += 1 + + return args + + def print_help(self): + """打印帮助信息""" + stdio.printf("usage: %s [options]\n", self._prog) + if self._description is not None and self._description[0] != 0: + stdio.printf("\n%s\n", self._description) + stdio.printf("\npositional arguments:\n") + i: INT = 0 + while i < self._arg_count: + arg: Argument | t.CPtr = self._get_arg(i) + if arg.is_positional: + ht: str = arg.help_text + if ht is None: ht = "" + stdio.printf(" %-20s %s\n", arg.name, ht) + i += 1 + stdio.printf("\noptional arguments:\n") + i = 0 + while i < self._arg_count: + arg = self._get_arg(i) + if not arg.is_positional: + ht: str = arg.help_text + if ht is None: ht = "" + if arg.short_name is not None: + stdio.printf(" %s, %-18s %s\n", + arg.short_name, arg.name, ht) + else: + stdio.printf(" %-22s %s\n", arg.name, ht) + i += 1 + + def print_usage(self): + """打印用法简述""" + stdio.printf("usage: %s [options]\n", self._prog) + + +# ============================================================ +# release — 释放 ParsedArgs 及其内部资源 +# ============================================================ + +def release(args: ParsedArgs | t.CPtr): + """释放 ParsedArgs 对象及其内部 HashTable。""" + if args is None: return + if args.__mbuddy__ is None: return + if args._ht is not None: + if args._ht.__slots__ is not None: + args.__mbuddy__.free(args._ht.__slots__) + args.__mbuddy__.free(t.CVoid(t.CUInt64T(args._ht), t.CPtr)) + args.__mbuddy__.free(t.CVoid(t.CUInt64T(args), t.CPtr)) diff --git a/includes/asm.py b/includes/asm.py new file mode 100644 index 0000000..ef64063 --- /dev/null +++ b/includes/asm.py @@ -0,0 +1,153 @@ +import t, c + + +def cli(): + c.Asm("cli", [t.ASM_DESCR.CLOBBER_MEMORY]) + +def sti(): + c.Asm("sti", [t.ASM_DESCR.CLOBBER_MEMORY]) + +def hlt(): + c.Asm("hlt", [t.ASM_DESCR.CLOBBER_MEMORY]) + +def nop(): + c.Asm("nop", [t.ASM_DESCR.CLOBBER_MEMORY]) + +def pause(): + c.Asm("pause", op=[t.ASM_DESCR.CLOBBER_MEMORY]) + +def inb(port: t.CUInt16T) -> t.CUInt8T: + value: t.CUInt32T + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + in al, dx + movzx eax, al""", + out = [c.AsmOut(value, t.ASM_DESCR.OUTPUT_REG)], + op =[t.ASM_DESCR.CLOBBER_RDX]) + return t.CUInt8T(value) + +def outb(port: t.CUInt16T, value: t.CUInt8T): + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + mov eax, {c.AsmInp(t.CUInt32T(value), t.ASM_DESCR.REG_ANY)} + out dx, al""", + op = [t.ASM_DESCR.CLOBBER_RDX, t.ASM_DESCR.CLOBBER_RAX]) + + +def inw(port: t.CUInt16T) -> t.CUInt16T: + value: t.CUInt32T + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + in ax, dx + movzx eax, ax""", + out = [c.AsmOut(value, t.ASM_DESCR.OUTPUT_REG)], + op =[t.ASM_DESCR.CLOBBER_RDX]) + return t.CUInt16T(value) + +def outw(port: t.CUInt16T, value: t.CUInt16T): + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + mov eax, {c.AsmInp(t.CUInt32T(value), t.ASM_DESCR.REG_ANY)} + out dx, ax""", + op = [t.ASM_DESCR.CLOBBER_RDX, t.ASM_DESCR.CLOBBER_RAX]) + +def inl(port: t.CUInt16T) -> t.CUInt32T: + value: t.CUInt32T + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + in eax, dx""", + out = [c.AsmOut(value, t.ASM_DESCR.OUTPUT_REG)], + op =[t.ASM_DESCR.CLOBBER_RDX]) + return value + +def outl(port: t.CUInt16T, value: t.CUInt32T): + c.Asm(f"""mov edx, {c.AsmInp(t.CUInt32T(port), t.ASM_DESCR.REG_ANY)} + mov eax, {c.AsmInp(value, t.ASM_DESCR.REG_ANY)} + out dx, eax""", + op = [t.ASM_DESCR.CLOBBER_RDX, t.ASM_DESCR.CLOBBER_RAX]) + +def io_wait(): + c.Asm(f"mov dx, 0x80; out dx, al", + op = [t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_DX]) + +def pushall() -> t.CInline | t.CVoid: + c.Asm(f""" + push rax + push rbx + push rcx + push rdx + push rsi + push rdi + push rbp + push r8 + push r9 + push r10 + push r11 + push r12 + push r13 + push r14 + push r15""", + [t.ASM_DESCR.CLOBBER_MEMORY] + ) + +def popall() -> t.CInline | t.CVoid: + c.Asm(f""" + pop r15 + pop r14 + pop r13 + pop r12 + pop r11 + pop r10 + pop r9 + pop r8 + pop rbp + pop rdi + pop rsi + pop rdx + pop rcx + pop rbx + pop rax""", + [t.ASM_DESCR.CLOBBER_MEMORY] + ) + +def cpu_switch_context(old_stack: t.CUInt64T | t.CPtr, new_stack: t.CUInt64T): + c.Asm(f""" + push rax + push rbx + push rcx + push rdx + push rsi + push rdi + push rbp + push r8 + push r9 + push r10 + push r11 + push r12 + push r13 + push r14 + push r15 + mov [{c.AsmInp(old_stack, t.ASM_DESCR.REG_ANY)}], rsp + mov rsp, {c.AsmInp(new_stack, t.ASM_DESCR.REG_ANY)} + pop r15 + pop r14 + pop r13 + pop r12 + pop r11 + pop r10 + pop r9 + pop r8 + pop rbp + pop rdi + pop rsi + pop rdx + pop rcx + pop rbx + pop rax""", + op=[t.ASM_DESCR.CLOBBER_MEMORY]) + +def BSSClean(): + c.Asm( + "lea rdi, [rip + __bss_start]\n" + "lea rcx, [rip + __bss_end]\n" + "sub rcx, rdi\n" + "xor eax, eax\n" + "cld\n" + "rep stosb", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RDI, t.ASM_DESCR.CLOBBER_RCX, t.ASM_DESCR.CLOBBER_RAX] + ) diff --git a/includes/ast/__init__.py b/includes/ast/__init__.py new file mode 100644 index 0000000..8a26c77 --- /dev/null +++ b/includes/ast/__init__.py @@ -0,0 +1,105 @@ +import t, c +from stdint import * +import memhub +import string + +# 导入子模块 +from .tokens import ( + Token, TokenType, Keyword, TokOp, _init_tables, +) +# 公共部分：基类、枚举、常量、辅助函数、dump +from .base import ( + AST, ASTKind, ASTCtx, OpKind, ASTFlag, + # 常量 + CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, + FLAG_IS_ASYNC, FLAG_SIMPLE, FLAG_HAS_STAR, + # 辅助函数 + _init_ast, _copy_str, _set_pos, _inherit_pos, _set_parent_list, + _binop_from_op, _augop_from_op, _cmpop_from_op, + _emit, _emit_str, _emit_int, _dump_list, + dump, +) +# 语句节点 +from .stmts import ( + # 模块层 + Module, Expression, Interactive, FunctionType, + # 语句 + FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, + For, While, If, With, Raise, Try, Assert, Global, Nonlocal, + Pass, Break, Continue, Expr, Import, ImportFrom, Match, +) +# 表达式节点 +from .exprs import ( + # 表达式 + BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, + ListComp, SetComp, DictComp, GeneratorExp, + Await, Yield, YieldFrom, FormattedValue, JoinedStr, + Constant, NamedExpr, Attribute, Subscript, Starred, Name, + List, Tuple, Slice, + Call, Compare, OpNode, +) +# 辅助节点 +from .astaux import ( + ExceptHandler, Arguments, Arg, Keyword, Alias, WithItem, + Comprehension, +) +# 模式匹配 +from .match import ( + MatchCase, MatchValue, MatchSingleton, MatchSequence, + MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr, +) +from .visitor import ASTVisitor +from .lexer import Lexer, _lexer_init, tokenize, new_lexer +from .parser import Parser, _parser_init, parse_tokens, new_parser +from .parser import _parse_test + +# ============================================================ +# 全局内存池指针（遵循项目规则：声明 _mbuddy） +# 用户在 main 中初始化：ast._mbuddy = memhub.MemBuddy(arena, size) +# ============================================================ +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================ +# 公共 API: parse +# ============================================================ + +def parse(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: + """解析 Python 源代码，返回 AST Module 节点。 + + 参数: + src: NUL 结尾的源代码字符串 + pool: memhub.MemManager 实例，用于分配 AST 节点 + + 返回: + AST Module 节点，失败返回 None + """ + if src is None or pool is None: + return None + _init_tables(pool) + lx: Lexer | t.CPtr = new_lexer(pool) + if lx is None: + return None + _lexer_init(lx, src, pool) + tokens: Token | t.CPtr = tokenize(lx) + if tokens is None: + return None + tree: AST | t.CPtr = parse_tokens(pool, tokens) + return tree + + +def parse_expression(src: str, pool: memhub.MemManager | t.CPtr) -> AST | t.CPtr: + """解析单个表达式，返回 Expression 节点""" + if src is None or pool is None: + return None + _init_tables(pool) + lx: Lexer | t.CPtr = new_lexer(pool) + if lx is None: return None + _lexer_init(lx, src, pool) + tokens: Token | t.CPtr = tokenize(lx) + if tokens is None: return None + ps: Parser | t.CPtr = new_parser(pool) + if ps is None: return None + _parser_init(ps, tokens, pool) + body: AST | t.CPtr = _parse_test(ps) + return Expression(pool, body) diff --git a/includes/ast/astaux.py b/includes/ast/astaux.py new file mode 100644 index 0000000..3a4b083 --- /dev/null +++ b/includes/ast/astaux.py @@ -0,0 +1,337 @@ +import t, c +from stdint import * +import memhub +import string +from .base import ( + AST, ASTKind, + _init_ast, _copy_str, _set_parent_list, + _emit, _emit_str, _emit_int, _dump_list, +) + + +# ============================================================ +# ExceptHandler 节点 +# ============================================================ + +class ExceptHandler(AST): + """ExceptHandler(type: AST, name: str)""" + type: AST | t.CPtr + name: str + + def __new__(self, pool: memhub.MemManager | t.CPtr, + type: AST | t.CPtr, name: str): + ptr: ExceptHandler | t.CPtr = pool.alloc(ExceptHandler.__sizeof__()) + if ptr: + string.memset(ptr, 0, ExceptHandler.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + type: AST | t.CPtr, name: str): + _init_ast(self, pool, 0, 0) + self.type = type + self.name = _copy_str(pool, name) + + def kind(self) -> t.CInt: + return ASTKind.ExceptHandler + + def type_name(self) -> str: + return "ExceptHandler" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "ExceptHandler(type=") + if self.type is not None: + pos = self.type.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "', body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Arguments 节点 +# ============================================================ + +class Arguments(AST): + """Arguments(args: list[AST | CPtr], vararg: AST, kwarg: AST, + defaults: list[AST | CPtr], kw_defaults: list[AST | CPtr])""" + args: list[AST | t.CPtr] | t.CPtr + vararg: AST | t.CPtr + kwarg: AST | t.CPtr + defaults: list[AST | t.CPtr] | t.CPtr + kw_defaults: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + args: list[AST | t.CPtr] | t.CPtr, + vararg: AST | t.CPtr, kwarg: AST | t.CPtr, + defaults: list[AST | t.CPtr] | t.CPtr, + kw_defaults: list[AST | t.CPtr] | t.CPtr): + ptr: Arguments | t.CPtr = pool.alloc(Arguments.__sizeof__()) + if ptr: + string.memset(ptr, 0, Arguments.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + args: list[AST | t.CPtr] | t.CPtr, + vararg: AST | t.CPtr, kwarg: AST | t.CPtr, + defaults: list[AST | t.CPtr] | t.CPtr, + kw_defaults: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.args = args + self.vararg = vararg + self.kwarg = kwarg + self.defaults = defaults + self.kw_defaults = kw_defaults + _set_parent_list(args, self) + _set_parent_list(defaults, self) + _set_parent_list(kw_defaults, self) + + def kind(self) -> t.CInt: + return ASTKind.Arguments + + def type_name(self) -> str: + return "Arguments" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Arguments(args=") + pos = _dump_list(self.args, buf, size, pos) + pos = _emit(buf, size, pos, ", vararg=") + if self.vararg is not None: + pos = self.vararg.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", kwarg=") + if self.kwarg is not None: + pos = self.kwarg.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", defaults=") + pos = _dump_list(self.defaults, buf, size, pos) + pos = _emit(buf, size, pos, ", kw_defaults=") + pos = _dump_list(self.kw_defaults, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Arg 节点 +# ============================================================ + +class Arg(AST): + """Arg(arg: str, annotation: AST)""" + arg: str + annotation: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + arg: str, annotation: AST | t.CPtr): + ptr: Arg | t.CPtr = pool.alloc(Arg.__sizeof__()) + if ptr: + string.memset(ptr, 0, Arg.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + arg: str, annotation: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.arg = _copy_str(pool, arg) + self.annotation = annotation + + def kind(self) -> t.CInt: + return ASTKind.Arg + + def type_name(self) -> str: + return "Arg" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Arg(arg='") + pos = _emit_str(buf, size, pos, self.arg) + pos = _emit(buf, size, pos, "', annotation=") + if self.annotation is not None: + pos = self.annotation.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Keyword 节点 +# ============================================================ + +class Keyword(AST): + """Keyword(arg: str, value: AST)""" + arg: str + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + arg: str, value: AST | t.CPtr): + ptr: Keyword | t.CPtr = pool.alloc(Keyword.__sizeof__()) + if ptr: + string.memset(ptr, 0, Keyword.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + arg: str, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.arg = _copy_str(pool, arg) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.Keyword + + def type_name(self) -> str: + return "Keyword" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Keyword(arg='") + pos = _emit_str(buf, size, pos, self.arg) + pos = _emit(buf, size, pos, "', value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Alias 节点 +# ============================================================ + +class Alias(AST): + """Alias(name: str, asname: str)""" + name: str + asname: str + + def __new__(self, pool: memhub.MemManager | t.CPtr, + name: str, asname: str): + ptr: Alias | t.CPtr = pool.alloc(Alias.__sizeof__()) + if ptr: + string.memset(ptr, 0, Alias.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + name: str, asname: str): + _init_ast(self, pool, 0, 0) + self.name = _copy_str(pool, name) + self.asname = _copy_str(pool, asname) + + def kind(self) -> t.CInt: + return ASTKind.Alias + + def type_name(self) -> str: + return "Alias" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Alias(name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "', asname='") + pos = _emit_str(buf, size, pos, self.asname) + pos = _emit(buf, size, pos, "')") + return pos + + +# ============================================================ +# WithItem 节点 +# ============================================================ + +class WithItem(AST): + """WithItem(context_expr: AST, optional_vars: AST)""" + context_expr: AST | t.CPtr + optional_vars: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr): + ptr: WithItem | t.CPtr = pool.alloc(WithItem.__sizeof__()) + if ptr: + string.memset(ptr, 0, WithItem.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + context_expr: AST | t.CPtr, optional_vars: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.context_expr = context_expr + self.optional_vars = optional_vars + + def kind(self) -> t.CInt: + return ASTKind.WithItem + + def type_name(self) -> str: + return "WithItem" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "WithItem(context_expr=") + if self.context_expr is not None: + pos = self.context_expr.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", optional_vars=") + if self.optional_vars is not None: + pos = self.optional_vars.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Comprehension 节点 +# ============================================================ + +class Comprehension(AST): + """Comprehension(target: AST, iter: AST, ifs: list[AST | CPtr], is_async: CInt)""" + target: AST | t.CPtr + iter: AST | t.CPtr + ifs: list[AST | t.CPtr] | t.CPtr + is_async: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, iter: AST | t.CPtr, + ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt): + ptr: Comprehension | t.CPtr = pool.alloc(Comprehension.__sizeof__()) + if ptr: + string.memset(ptr, 0, Comprehension.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, iter: AST | t.CPtr, + ifs: list[AST | t.CPtr] | t.CPtr, is_async: t.CInt): + _init_ast(self, pool, 0, 0) + self.target = target + self.iter = iter + self.ifs = ifs + self.is_async = is_async + _set_parent_list(ifs, self) + + def kind(self) -> t.CInt: + return ASTKind.Comprehension + + def type_name(self) -> str: + return "Comprehension" + + def dump(self, buf: bytes, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Comprehension(target=") + if self.target is not None: + pos = self.target.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", iter=") + if self.iter is not None: + pos = self.iter.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", ifs=") + pos = _dump_list(self.ifs, buf, size, pos) + pos = _emit(buf, size, pos, ", is_async=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.is_async)) + pos = _emit(buf, size, pos, ")") + return pos diff --git a/includes/ast/base.py b/includes/ast/base.py new file mode 100644 index 0000000..3b2820c --- /dev/null +++ b/includes/ast/base.py @@ -0,0 +1,474 @@ +import t, c +from stdint import * +import memhub +import string +import viperlib +from .tokens import TokOp + + +# ============================================================ +# ASTKind - 节点类型枚举（kind() 虚函数返回值） +# ============================================================ + +class ASTKind(t.CEnum): + # 核心 15 节点 + Module: t.State + FunctionDef: t.State + ClassDef: t.State + Assign: t.State + If: t.State + For: t.State + While: t.State + Return: t.State + Expr: t.State + Name: t.State + Constant: t.State + BinOp: t.State + UnaryOp: t.State + Call: t.State + Compare: t.State + # 模块层 + Expression: t.State + Interactive: t.State + FunctionType: t.State + # 语句 + Delete: t.State + AugAssign: t.State + AnnAssign: t.State + With: t.State + Raise: t.State + Try: t.State + Assert: t.State + Global: t.State + Nonlocal: t.State + Pass: t.State + Break: t.State + Continue: t.State + Import: t.State + ImportFrom: t.State + Match: t.State + # 表达式 + BoolOp: t.State + Lambda: t.State + IfExp: t.State + Dict: t.State + Set: t.State + ListComp: t.State + SetComp: t.State + DictComp: t.State + GeneratorExp: t.State + Await: t.State + Yield: t.State + YieldFrom: t.State + FormattedValue: t.State + JoinedStr: t.State + Attribute: t.State + Subscript: t.State + Starred: t.State + List: t.State + Tuple: t.State + Slice: t.State + NamedExpr: t.State + # 辅助节点 + ExceptHandler: t.State + Arguments: t.State + Arg: t.State + Keyword: t.State + Alias: t.State + WithItem: t.State + Comprehension: t.State + OpNode: t.State + # 模式匹配 + MatchCase: t.State + MatchValue: t.State + MatchSingleton: t.State + MatchSequence: t.State + MatchMapping: t.State + MatchClass: t.State + MatchStar: t.State + MatchAs: t.State + MatchOr: t.State + + +# ============================================================ +# ASTCtx - 名称上下文（Load/Store/Del） +# ============================================================ + +class ASTCtx(t.CEnum): + Load: t.State + Store: t.State + Del: t.State + + +# ============================================================ +# OpKind - 运算符枚举（BinOp/UnaryOp/Compare 共用） +# ============================================================ + +class OpKind(t.CEnum): + # 二元算术/位运算 + Add: t.State + Sub: t.State + Mult: t.State + MatMult: t.State + Div: t.State + Mod: t.State + Pow: t.State + LShift: t.State + RShift: t.State + BitOr: t.State + BitXor: t.State + BitAnd: t.State + FloorDiv: t.State + # 布尔运算 + And: t.State + Or: t.State + # 比较运算 + Eq: t.State + Ne: t.State + Lt: t.State + Le: t.State + Gt: t.State + Ge: t.State + Is: t.State + IsNot: t.State + In: t.State + NotIn: t.State + # 一元运算 + Not: t.State + UAdd: t.State + USub: t.State + Invert: t.State + # 空 + NoneOp: t.State + + +# Constant 子类型标识 +CONST_INT: t.CDefine = 1 +CONST_FLOAT: t.CDefine = 2 +CONST_STR: t.CDefine = 3 +CONST_BOOL: t.CDefine = 4 +CONST_NONE: t.CDefine = 5 + +# 标志位 +FLAG_IS_ASYNC: t.CDefine = 1 +FLAG_SIMPLE: t.CDefine = 2 +FLAG_HAS_STAR: t.CDefine = 4 + + +# ============================================================ +# ASTFlag - 节点标志位枚举 +# ============================================================ + +class ASTFlag(t.CEnum): + IsAsync: t.State + Simple: t.State + HasStar: t.State + + +# ============================================================ +# 辅助函数 +# ============================================================ + +def _copy_str(pool: memhub.MemManager | t.CPtr, src: str) -> str: + """从 pool 分配并复制 C 字符串""" + if src is None: return None + slen: t.CSizeT = string.strlen(src) + buf: str = pool.alloc(slen + 1) + if buf is None: return None + string.memcpy(buf, src, slen + 1) + return buf + + +def _set_pos(node: AST | t.CPtr, lineno: t.CInt, col_offset: t.CInt, + end_lineno: t.CInt, end_col_offset: t.CInt): + if node is None: return + node.lineno = lineno + node.col_offset = col_offset + node.end_lineno = end_lineno + node.end_col_offset = end_col_offset + + +def _inherit_pos(node: AST | t.CPtr, ref: AST | t.CPtr): + if node is None or ref is None: return + node.lineno = ref.lineno + node.col_offset = ref.col_offset + node.end_lineno = ref.end_lineno + node.end_col_offset = ref.end_col_offset + + +def _init_ast(node: AST | t.CPtr, pool: memhub.MemManager | t.CPtr, + lineno: t.CInt, col: t.CInt): + """初始化 AST 基类字段""" + node.parent = None + node.pool = pool + node.lineno = lineno + node.col_offset = col + node.end_lineno = lineno + node.end_col_offset = col + node.children = None + + +def _set_parent_list(lst: list[AST | t.CPtr] | t.CPtr, parent: AST | t.CPtr): + """为 list 中的每个子节点设置 parent 指针""" + if lst is None: return + n: t.CSizeT = lst.__len__() + i: t.CSizeT = 0 + while i < n: + child: AST | t.CPtr = lst.get(i) + if child is not None: + child.parent = parent + i += 1 + + +def _append_child(lst: list[AST | t.CPtr] | t.CPtr, node: AST | t.CPtr, + pool: memhub.MemManager | t.CPtr): + """向 list 追加子节点（模块级函数，避免与 AST.append 方法名冲突）。 + + 在 AST.append 方法中调用此函数，绕过编译器把 list.append 误解析为 + AST.append 虚函数调用的类型推断问题。""" + if lst is None or node is None: + return + lst.append(node) + + +# ============================================================ +# 运算符映射：TokOp -> OpKind +# ============================================================ + +def _binop_from_op(op_id: t.CInt) -> t.CInt: + if op_id == TokOp.Plus: return OpKind.Add + if op_id == TokOp.Minus: return OpKind.Sub + if op_id == TokOp.Star: return OpKind.Mult + if op_id == TokOp.At: return OpKind.MatMult + if op_id == TokOp.Slash: return OpKind.Div + if op_id == TokOp.Percent: return OpKind.Mod + if op_id == TokOp.StarStar: return OpKind.Pow + if op_id == TokOp.LtLt: return OpKind.LShift + if op_id == TokOp.GtGt: return OpKind.RShift + if op_id == TokOp.VBar: return OpKind.BitOr + if op_id == TokOp.Caret: return OpKind.BitXor + if op_id == TokOp.Amp: return OpKind.BitAnd + if op_id == TokOp.DSlash: return OpKind.FloorDiv + return OpKind.NoneOp + + +def _augop_from_op(op_id: t.CInt) -> t.CInt: + if op_id == TokOp.PlusEq: return OpKind.Add + if op_id == TokOp.MinusEq: return OpKind.Sub + if op_id == TokOp.StarEq: return OpKind.Mult + if op_id == TokOp.AtEq: return OpKind.MatMult + if op_id == TokOp.SlashEq: return OpKind.Div + if op_id == TokOp.PercentEq: return OpKind.Mod + if op_id == TokOp.StarEqEq: return OpKind.Pow + if op_id == TokOp.LtLtEq: return OpKind.LShift + if op_id == TokOp.GtGtEq: return OpKind.RShift + if op_id == TokOp.VBarEq: return OpKind.BitOr + if op_id == TokOp.CaretEq: return OpKind.BitXor + if op_id == TokOp.AmpEq: return OpKind.BitAnd + if op_id == TokOp.DSlashEq: return OpKind.FloorDiv + return OpKind.NoneOp + + +def _cmpop_from_op(op_id: t.CInt) -> t.CInt: + if op_id == TokOp.EqEq: return OpKind.Eq + if op_id == TokOp.ExclaimEq: return OpKind.Ne + if op_id == TokOp.Less: return OpKind.Lt + if op_id == TokOp.LessEq: return OpKind.Le + if op_id == TokOp.Greater: return OpKind.Gt + if op_id == TokOp.GreaterEq: return OpKind.Ge + return OpKind.NoneOp + + +# ============================================================ +# dump 辅助函数 +# ============================================================ + +def _emit(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, + text: str) -> t.CSizeT: + """追加字面量到 buf+pos，返回新 pos""" + if buf is None or pos >= size: + return pos + cur: t.CChar | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(buf) + pos) + rem: t.CSizeT = size - pos + n: t.CInt = viperlib.snprintf(cur, rem, "%s", text) + if n < 0: + return pos + return pos + t.CSizeT(n) + + +def _emit_str(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, + text: str) -> t.CSizeT: + """追加字符串值到 buf+pos（用 %s 格式）""" + if buf is None or pos >= size: + return pos + cur: t.CChar | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(buf) + pos) + rem: t.CSizeT = size - pos + n: t.CInt = viperlib.snprintf(cur, rem, "%s", text) + if n < 0: + return pos + return pos + t.CSizeT(n) + + +def _emit_int(buf: t.CChar | t.CPtr, size: t.CSizeT, pos: t.CSizeT, + val: t.CInt64T) -> t.CSizeT: + """追加整数值到 buf+pos""" + if buf is None or pos >= size: + return pos + cur: t.CChar | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(buf) + pos) + rem: t.CSizeT = size - pos + n: t.CInt = viperlib.snprintf(cur, rem, "%lld", val) + if n < 0: + return pos + return pos + t.CSizeT(n) + + +def _op_name(op: t.CInt) -> str: + """返回 OpKind 的字符串名称""" + if op == OpKind.Add: return "Add" + if op == OpKind.Sub: return "Sub" + if op == OpKind.Mult: return "Mult" + if op == OpKind.MatMult: return "MatMult" + if op == OpKind.Div: return "Div" + if op == OpKind.Mod: return "Mod" + if op == OpKind.Pow: return "Pow" + if op == OpKind.LShift: return "LShift" + if op == OpKind.RShift: return "RShift" + if op == OpKind.BitOr: return "BitOr" + if op == OpKind.BitXor: return "BitXor" + if op == OpKind.BitAnd: return "BitAnd" + if op == OpKind.FloorDiv: return "FloorDiv" + if op == OpKind.And: return "And" + if op == OpKind.Or: return "Or" + if op == OpKind.Eq: return "Eq" + if op == OpKind.Ne: return "Ne" + if op == OpKind.Lt: return "Lt" + if op == OpKind.Le: return "Le" + if op == OpKind.Gt: return "Gt" + if op == OpKind.Ge: return "Ge" + if op == OpKind.Is: return "Is" + if op == OpKind.IsNot: return "IsNot" + if op == OpKind.In: return "In" + if op == OpKind.NotIn: return "NotIn" + if op == OpKind.Not: return "Not" + if op == OpKind.UAdd: return "UAdd" + if op == OpKind.USub: return "USub" + if op == OpKind.Invert: return "Invert" + return "NoneOp" + + +def _dump_list(lst: list[AST | t.CPtr] | t.CPtr, buf: t.CChar | t.CPtr, + size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: + """遍历 list[AST | CPtr] 容器，多态 dump 每个元素""" + if lst is None: + pos = _emit(buf, size, pos, "[]") + return pos + pos = _emit(buf, size, pos, "[") + n: t.CSizeT = lst.__len__() + i: t.CSizeT = 0 + while i < n: + if i > 0: + pos = _emit(buf, size, pos, ", ") + child: AST | t.CPtr = lst.get(i) + if child is not None: + pos = child.dump(buf, size, pos) + i += 1 + pos = _emit(buf, size, pos, "]") + return pos + + +def _dump_op_list(lst: t.CPtr, buf: t.CChar | t.CPtr, + size: t.CSizeT, pos: t.CSizeT) -> t.CSizeT: + """遍历 list[CInt] (OpKind 值)，输出 op 名称列表""" + if lst is None: + pos = _emit(buf, size, pos, "[]") + return pos + ops_list: list[t.CInt] | t.CPtr = (list[t.CInt] | t.CPtr)(lst) + pos = _emit(buf, size, pos, "[") + n: t.CSizeT = ops_list.__len__() + i: t.CSizeT = 0 + while i < n: + if i > 0: + pos = _emit(buf, size, pos, ", ") + op: t.CInt = ops_list.get(i) + pos = _emit(buf, size, pos, _op_name(op)) + i += 1 + pos = _emit(buf, size, pos, "]") + return pos + + +# ============================================================ +# AST - 多态基类 +# +# @t.CVTable 启用 vtable，支持 kind()/type_name()/dump()/accept() 虚函数 +# 子节点用 list[AST | CPtr] 容器存储（O(1) 随机访问，比链表高效） +# parent 字段维护父指针（用于向上遍历） +# ============================================================ + +@t.CVTable +class AST: + """AST 节点基类。所有具体节点类继承此类。 + + 字段: + parent: 父节点指针（向上遍历） + pool: 分配器（用于子节点/字符串分配） + lineno/col_offset: 起始位置 + end_lineno/end_col_offset: 结束位置 + children: 子节点列表（body 语句块，通过 append 添加） + """ + parent: AST | t.CPtr + pool: memhub.MemManager | t.CPtr + lineno: t.CInt + col_offset: t.CInt + end_lineno: t.CInt + end_col_offset: t.CInt + children: list[AST | t.CPtr] | t.CPtr + + def kind(self) -> t.CInt: + """返回节点类型（ASTKind 值），子类覆盖""" + return 0 + + def type_name(self) -> str: + """返回节点类型名，子类覆盖""" + return "AST" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + """多态 dump：子类覆盖以自定义格式。返回写入后的 pos""" + return pos + + def accept(self, visitor: t.CPtr): + """访问者模式入口（待实现）：ASTVisitor 在 __visitor.py 定义后， + 子类覆盖此方法分派到 visitor.visit_X(self)。当前为空实现。""" + pass + + def append(self, node: AST | t.CPtr): + """将 node 追加为子节点（添加到 children 列表），设置 parent 指针。 + + 用于 parser 构建语句块：module.append(stmt), if_node.append(stmt) 等。 + 首次调用时懒初始化 children 列表。 + + 注意：list[AST|CPtr] 当前按值复制存储（编译器限制），所以必须先设置 + node.parent 再 append，这样副本中的 parent 才是正确的。""" + if self is None or node is None: + return + if self.children is None: + self.children = list[AST | t.CPtr](self.pool, 8) + node.parent = self + _append_child(self.children, node, self.pool) + + +# ============================================================ +# 模块级 dump 函数（供 ast.dump(node, buf, size) 调用） +# ============================================================ + +def dump(node: AST | t.CPtr, buf: t.CChar | t.CPtr, size: t.CSizeT): + """将 AST 节点序列化到 buf（容量 size）""" + if node is None or buf is None or size == 0: + return + buf[0] = '\0' + final_pos: t.CSizeT = node.dump(buf, size, 0) + # NUL 终止，防止 printf 越界 + if final_pos < size: + buf[final_pos] = '\0' + else: + buf[size - 1] = '\0' diff --git a/includes/ast/exprs.py b/includes/ast/exprs.py new file mode 100644 index 0000000..92b7b87 --- /dev/null +++ b/includes/ast/exprs.py @@ -0,0 +1,1234 @@ +import t, c +from stdint import * +import memhub +import string +from .base import ( + AST, ASTKind, ASTCtx, OpKind, + _init_ast, _copy_str, _set_parent_list, + _emit, _emit_str, _emit_int, _dump_list, _dump_op_list, _op_name, + CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE +) + + +# ============================================================ +# Name 节点 +# ============================================================ + +class Name(AST): + """Name(id: str, ctx: ASTCtx)""" + id: str + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + id: str, ctx: t.CInt): + ptr: Name | t.CPtr = pool.alloc(Name.__sizeof__()) + if ptr: + string.memset(ptr, 0, Name.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + id: str, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.id = _copy_str(pool, id) + self.ctx = ctx + + def kind(self) -> t.CInt: + return ASTKind.Name + + def type_name(self) -> str: + return "Name" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Name(id='") + pos = _emit_str(buf, size, pos, self.id) + pos = _emit(buf, size, pos, "', ctx=") + if self.ctx == ASTCtx.Load: + pos = _emit(buf, size, pos, "Load") + elif self.ctx == ASTCtx.Store: + pos = _emit(buf, size, pos, "Store") + else: + pos = _emit(buf, size, pos, "Del") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Constant 节点（用 const_kind 标识子类型） +# ============================================================ + +class Constant(AST): + """Constant(int_val/float_val/str_val, const_kind)""" + int_val: t.CInt64T + float_val: t.CDouble + str_val: str + const_kind: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + const_kind: t.CInt, int_val: t.CInt64T, + float_val: t.CDouble, str_val: str, + lineno: t.CInt, col: t.CInt): + ptr: Constant | t.CPtr = pool.alloc(Constant.__sizeof__()) + if ptr: + string.memset(ptr, 0, Constant.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + const_kind: t.CInt, int_val: t.CInt64T, + float_val: t.CDouble, str_val: str, + lineno: t.CInt, col: t.CInt): + _init_ast(self, pool, lineno, col) + self.const_kind = const_kind + self.int_val = int_val + self.float_val = float_val + if str_val is not None: + self.str_val = _copy_str(pool, str_val) + else: + self.str_val = None + + def kind(self) -> t.CInt: + return ASTKind.Constant + + def type_name(self) -> str: + return "Constant" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Constant(value=") + if self.const_kind == CONST_INT: + pos = _emit_int(buf, size, pos, self.int_val) + elif self.const_kind == CONST_FLOAT: + pos = _emit(buf, size, pos, "") + elif self.const_kind == CONST_STR: + pos = _emit(buf, size, pos, "'") + if self.str_val is not None: + pos = _emit_str(buf, size, pos, self.str_val) + else: + pos = _emit(buf, size, pos, "(null)") + pos = _emit(buf, size, pos, "'") + elif self.const_kind == CONST_BOOL: + if self.int_val != 0: + pos = _emit(buf, size, pos, "True") + else: + pos = _emit(buf, size, pos, "False") + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# BinOp 节点 +# ============================================================ + +class BinOp(AST): + """BinOp(left: AST, op: OpKind, right: AST)""" + left: AST | t.CPtr + op: t.CInt + right: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr): + ptr: BinOp | t.CPtr = pool.alloc(BinOp.__sizeof__()) + if ptr: + string.memset(ptr, 0, BinOp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + left: AST | t.CPtr, op: t.CInt, right: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.left = left + self.op = op + self.right = right + + def kind(self) -> t.CInt: + return ASTKind.BinOp + + def type_name(self) -> str: + return "BinOp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "BinOp(left=") + if self.left is not None: + pos = self.left.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", op=") + pos = _emit(buf, size, pos, _op_name(self.op)) + pos = _emit(buf, size, pos, ", right=") + if self.right is not None: + pos = self.right.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# UnaryOp 节点 +# ============================================================ + +class UnaryOp(AST): + """UnaryOp(op: OpKind, operand: AST)""" + op: t.CInt + operand: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + op: t.CInt, operand: AST | t.CPtr): + ptr: UnaryOp | t.CPtr = pool.alloc(UnaryOp.__sizeof__()) + if ptr: + string.memset(ptr, 0, UnaryOp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + op: t.CInt, operand: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.op = op + self.operand = operand + + def kind(self) -> t.CInt: + return ASTKind.UnaryOp + + def type_name(self) -> str: + return "UnaryOp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "UnaryOp(op=") + pos = _emit(buf, size, pos, _op_name(self.op)) + pos = _emit(buf, size, pos, ", operand=") + if self.operand is not None: + pos = self.operand.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Compare 节点 +# ============================================================ + +class Compare(AST): + """Compare(left: AST, ops: list[CInt], comparators: list[AST | CPtr])""" + left: AST | t.CPtr + ops: t.CPtr + comparators: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + left: AST | t.CPtr, + ops: t.CPtr, + comparators: list[AST | t.CPtr] | t.CPtr): + ptr: Compare | t.CPtr = pool.alloc(Compare.__sizeof__()) + if ptr: + string.memset(ptr, 0, Compare.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + left: AST | t.CPtr, + ops: t.CPtr, + comparators: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.left = left + self.ops = ops + self.comparators = comparators + _set_parent_list(comparators, self) + + def kind(self) -> t.CInt: + return ASTKind.Compare + + def type_name(self) -> str: + return "Compare" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Compare(left=") + if self.left is not None: + pos = self.left.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", ops=") + pos = _dump_op_list(self.ops, buf, size, pos) + pos = _emit(buf, size, pos, ", comparators=") + pos = _dump_list(self.comparators, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Call 节点 +# ============================================================ + +class Call(AST): + """Call(func: AST, args: list[AST | CPtr], keywords: list[AST | CPtr])""" + func: AST | t.CPtr + args: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + func: AST | t.CPtr, + args: list[AST | t.CPtr] | t.CPtr, + keywords: list[AST | t.CPtr] | t.CPtr): + ptr: Call | t.CPtr = pool.alloc(Call.__sizeof__()) + if ptr: + string.memset(ptr, 0, Call.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + func: AST | t.CPtr, + args: list[AST | t.CPtr] | t.CPtr, + keywords: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.func = func + self.args = args + self.keywords = keywords + _set_parent_list(args, self) + _set_parent_list(keywords, self) + + def kind(self) -> t.CInt: + return ASTKind.Call + + def type_name(self) -> str: + return "Call" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Call(func=") + if self.func is not None: + pos = self.func.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", args=") + pos = _dump_list(self.args, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# BoolOp 节点 +# ============================================================ + +class BoolOp(AST): + """BoolOp(op: CInt, values: list[AST | CPtr])""" + op: t.CInt + values: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + op: t.CInt, + values: list[AST | t.CPtr] | t.CPtr): + ptr: BoolOp | t.CPtr = pool.alloc(BoolOp.__sizeof__()) + if ptr: + string.memset(ptr, 0, BoolOp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + op: t.CInt, + values: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.op = op + self.values = values + _set_parent_list(values, self) + + def kind(self) -> t.CInt: + return ASTKind.BoolOp + + def type_name(self) -> str: + return "BoolOp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "BoolOp(op=") + pos = _emit(buf, size, pos, _op_name(self.op)) + pos = _emit(buf, size, pos, ", values=") + pos = _dump_list(self.values, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Lambda 节点 +# ============================================================ + +class Lambda(AST): + """Lambda(args: AST, body: AST)""" + args: AST | t.CPtr + body: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + args: AST | t.CPtr, body: AST | t.CPtr): + ptr: Lambda | t.CPtr = pool.alloc(Lambda.__sizeof__()) + if ptr: + string.memset(ptr, 0, Lambda.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + args: AST | t.CPtr, body: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.args = args + self.body = body + + def kind(self) -> t.CInt: + return ASTKind.Lambda + + def type_name(self) -> str: + return "Lambda" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Lambda(args=") + if self.args is not None: + pos = self.args.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + if self.body is not None: + pos = self.body.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# IfExp 节点 +# ============================================================ + +class IfExp(AST): + """IfExp(test: AST, body: AST, orelse: AST)""" + test: AST | t.CPtr + body: AST | t.CPtr + orelse: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, body: AST | t.CPtr, + orelse: AST | t.CPtr): + ptr: IfExp | t.CPtr = pool.alloc(IfExp.__sizeof__()) + if ptr: + string.memset(ptr, 0, IfExp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, body: AST | t.CPtr, + orelse: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.test = test + self.body = body + self.orelse = orelse + + def kind(self) -> t.CInt: + return ASTKind.IfExp + + def type_name(self) -> str: + return "IfExp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "IfExp(test=") + if self.test is not None: + pos = self.test.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + if self.body is not None: + pos = self.body.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", orelse=") + if self.orelse is not None: + pos = self.orelse.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Dict 节点 +# ============================================================ + +class Dict(AST): + """Dict(keys: list[AST | CPtr], values: list[AST | CPtr])""" + keys: list[AST | t.CPtr] | t.CPtr + values: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + keys: list[AST | t.CPtr] | t.CPtr, + values: list[AST | t.CPtr] | t.CPtr): + ptr: Dict | t.CPtr = pool.alloc(Dict.__sizeof__()) + if ptr: + string.memset(ptr, 0, Dict.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + keys: list[AST | t.CPtr] | t.CPtr, + values: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.keys = keys + self.values = values + _set_parent_list(keys, self) + _set_parent_list(values, self) + + def kind(self) -> t.CInt: + return ASTKind.Dict + + def type_name(self) -> str: + return "Dict" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Dict(keys=") + pos = _dump_list(self.keys, buf, size, pos) + pos = _emit(buf, size, pos, ", values=") + pos = _dump_list(self.values, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Set 节点 +# ============================================================ + +class Set(AST): + """Set(elts: list[AST | CPtr])""" + elts: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr): + ptr: Set | t.CPtr = pool.alloc(Set.__sizeof__()) + if ptr: + string.memset(ptr, 0, Set.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.elts = elts + _set_parent_list(elts, self) + + def kind(self) -> t.CInt: + return ASTKind.Set + + def type_name(self) -> str: + return "Set" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Set(elts=") + pos = _dump_list(self.elts, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# ListComp 节点 +# ============================================================ + +class ListComp(AST): + """ListComp(elt: AST, generators: list[AST | CPtr])""" + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + ptr: ListComp | t.CPtr = pool.alloc(ListComp.__sizeof__()) + if ptr: + string.memset(ptr, 0, ListComp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.elt = elt + self.generators = generators + _set_parent_list(generators, self) + + def kind(self) -> t.CInt: + return ASTKind.ListComp + + def type_name(self) -> str: + return "ListComp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "ListComp(elt=") + if self.elt is not None: + pos = self.elt.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", generators=") + pos = _dump_list(self.generators, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# SetComp 节点 +# ============================================================ + +class SetComp(AST): + """SetComp(elt: AST, generators: list[AST | CPtr])""" + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + ptr: SetComp | t.CPtr = pool.alloc(SetComp.__sizeof__()) + if ptr: + string.memset(ptr, 0, SetComp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.elt = elt + self.generators = generators + _set_parent_list(generators, self) + + def kind(self) -> t.CInt: + return ASTKind.SetComp + + def type_name(self) -> str: + return "SetComp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "SetComp(elt=") + if self.elt is not None: + pos = self.elt.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", generators=") + pos = _dump_list(self.generators, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# DictComp 节点 +# ============================================================ + +class DictComp(AST): + """DictComp(key: AST, value: AST, generators: list[AST | CPtr])""" + key: AST | t.CPtr + value: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + key: AST | t.CPtr, value: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + ptr: DictComp | t.CPtr = pool.alloc(DictComp.__sizeof__()) + if ptr: + string.memset(ptr, 0, DictComp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + key: AST | t.CPtr, value: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.key = key + self.value = value + self.generators = generators + _set_parent_list(generators, self) + + def kind(self) -> t.CInt: + return ASTKind.DictComp + + def type_name(self) -> str: + return "DictComp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "DictComp(key=") + if self.key is not None: + pos = self.key.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", generators=") + pos = _dump_list(self.generators, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# GeneratorExp 节点 +# ============================================================ + +class GeneratorExp(AST): + """GeneratorExp(elt: AST, generators: list[AST | CPtr])""" + elt: AST | t.CPtr + generators: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + ptr: GeneratorExp | t.CPtr = pool.alloc(GeneratorExp.__sizeof__()) + if ptr: + string.memset(ptr, 0, GeneratorExp.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elt: AST | t.CPtr, + generators: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.elt = elt + self.generators = generators + _set_parent_list(generators, self) + + def kind(self) -> t.CInt: + return ASTKind.GeneratorExp + + def type_name(self) -> str: + return "GeneratorExp" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "GeneratorExp(elt=") + if self.elt is not None: + pos = self.elt.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", generators=") + pos = _dump_list(self.generators, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Await 节点 +# ============================================================ + +class Await(AST): + """Await(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: Await | t.CPtr = pool.alloc(Await.__sizeof__()) + if ptr: + string.memset(ptr, 0, Await.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.Await + + def type_name(self) -> str: + return "Await" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Await(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Yield 节点 +# ============================================================ + +class Yield(AST): + """Yield(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: Yield | t.CPtr = pool.alloc(Yield.__sizeof__()) + if ptr: + string.memset(ptr, 0, Yield.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.Yield + + def type_name(self) -> str: + return "Yield" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Yield(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# YieldFrom 节点 +# ============================================================ + +class YieldFrom(AST): + """YieldFrom(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: YieldFrom | t.CPtr = pool.alloc(YieldFrom.__sizeof__()) + if ptr: + string.memset(ptr, 0, YieldFrom.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.YieldFrom + + def type_name(self) -> str: + return "YieldFrom" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "YieldFrom(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# FormattedValue 节点 +# ============================================================ + +class FormattedValue(AST): + """FormattedValue(value: AST, conversion: CInt, format_spec: AST)""" + value: AST | t.CPtr + conversion: t.CInt + format_spec: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, conversion: t.CInt, + format_spec: AST | t.CPtr): + ptr: FormattedValue | t.CPtr = pool.alloc(FormattedValue.__sizeof__()) + if ptr: + string.memset(ptr, 0, FormattedValue.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, conversion: t.CInt, + format_spec: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + self.conversion = conversion + self.format_spec = format_spec + + def kind(self) -> t.CInt: + return ASTKind.FormattedValue + + def type_name(self) -> str: + return "FormattedValue" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "FormattedValue(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", conversion=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.conversion)) + pos = _emit(buf, size, pos, ", format_spec=") + if self.format_spec is not None: + pos = self.format_spec.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# JoinedStr 节点 +# ============================================================ + +class JoinedStr(AST): + """JoinedStr(values: list[AST | CPtr])""" + values: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + values: list[AST | t.CPtr] | t.CPtr): + ptr: JoinedStr | t.CPtr = pool.alloc(JoinedStr.__sizeof__()) + if ptr: + string.memset(ptr, 0, JoinedStr.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + values: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.values = values + _set_parent_list(values, self) + + def kind(self) -> t.CInt: + return ASTKind.JoinedStr + + def type_name(self) -> str: + return "JoinedStr" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "JoinedStr(values=") + pos = _dump_list(self.values, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Attribute 节点 +# ============================================================ + +class Attribute(AST): + """Attribute(value: AST, attr: str, ctx: CInt)""" + value: AST | t.CPtr + attr: str + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, attr: str, ctx: t.CInt): + ptr: Attribute | t.CPtr = pool.alloc(Attribute.__sizeof__()) + if ptr: + string.memset(ptr, 0, Attribute.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, attr: str, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.value = value + self.attr = _copy_str(pool, attr) + self.ctx = ctx + + def kind(self) -> t.CInt: + return ASTKind.Attribute + + def type_name(self) -> str: + return "Attribute" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Attribute(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", attr='") + pos = _emit_str(buf, size, pos, self.attr) + pos = _emit(buf, size, pos, "', ctx=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.ctx)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Subscript 节点 +# ============================================================ + +class Subscript(AST): + """Subscript(value: AST, slice: AST, ctx: CInt)""" + value: AST | t.CPtr + slice: AST | t.CPtr + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt): + ptr: Subscript | t.CPtr = pool.alloc(Subscript.__sizeof__()) + if ptr: + string.memset(ptr, 0, Subscript.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, slice: AST | t.CPtr, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.value = value + self.slice = slice + self.ctx = ctx + + def kind(self) -> t.CInt: + return ASTKind.Subscript + + def type_name(self) -> str: + return "Subscript" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Subscript(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", slice=") + if self.slice is not None: + pos = self.slice.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", ctx=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.ctx)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Starred 节点 +# ============================================================ + +class Starred(AST): + """Starred(value: AST, ctx: CInt)""" + value: AST | t.CPtr + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, ctx: t.CInt): + ptr: Starred | t.CPtr = pool.alloc(Starred.__sizeof__()) + if ptr: + string.memset(ptr, 0, Starred.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + value: AST | t.CPtr, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.value = value + self.ctx = ctx + + def kind(self) -> t.CInt: + return ASTKind.Starred + + def type_name(self) -> str: + return "Starred" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Starred(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", ctx=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.ctx)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# List 节点 +# ============================================================ + +class List(AST): + """List(elts: list[AST | CPtr], ctx: CInt)""" + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt): + ptr: List | t.CPtr = pool.alloc(List.__sizeof__()) + if ptr: + string.memset(ptr, 0, List.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.elts = elts + self.ctx = ctx + _set_parent_list(elts, self) + + def kind(self) -> t.CInt: + return ASTKind.List + + def type_name(self) -> str: + return "List" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "List(elts=") + pos = _dump_list(self.elts, buf, size, pos) + pos = _emit(buf, size, pos, ", ctx=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.ctx)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Tuple 节点 +# ============================================================ + +class Tuple(AST): + """Tuple(elts: list[AST | CPtr], ctx: CInt)""" + elts: list[AST | t.CPtr] | t.CPtr + ctx: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt): + ptr: Tuple | t.CPtr = pool.alloc(Tuple.__sizeof__()) + if ptr: + string.memset(ptr, 0, Tuple.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + elts: list[AST | t.CPtr] | t.CPtr, ctx: t.CInt): + _init_ast(self, pool, 0, 0) + self.elts = elts + self.ctx = ctx + _set_parent_list(elts, self) + + def kind(self) -> t.CInt: + return ASTKind.Tuple + + def type_name(self) -> str: + return "Tuple" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Tuple(elts=") + pos = _dump_list(self.elts, buf, size, pos) + pos = _emit(buf, size, pos, ", ctx=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.ctx)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Slice 节点 +# ============================================================ + +class Slice(AST): + """Slice(lower: AST, upper: AST, step: AST)""" + lower: AST | t.CPtr + upper: AST | t.CPtr + step: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + lower: AST | t.CPtr, upper: AST | t.CPtr, + step: AST | t.CPtr): + ptr: Slice | t.CPtr = pool.alloc(Slice.__sizeof__()) + if ptr: + string.memset(ptr, 0, Slice.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + lower: AST | t.CPtr, upper: AST | t.CPtr, + step: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.lower = lower + self.upper = upper + self.step = step + + def kind(self) -> t.CInt: + return ASTKind.Slice + + def type_name(self) -> str: + return "Slice" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Slice(lower=") + if self.lower is not None: + pos = self.lower.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", upper=") + if self.upper is not None: + pos = self.upper.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", step=") + if self.step is not None: + pos = self.step.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# NamedExpr 节点 +# ============================================================ + +class NamedExpr(AST): + """NamedExpr(target: AST, value: AST)""" + target: AST | t.CPtr + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, value: AST | t.CPtr): + ptr: NamedExpr | t.CPtr = pool.alloc(NamedExpr.__sizeof__()) + if ptr: + string.memset(ptr, 0, NamedExpr.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.target = target + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.NamedExpr + + def type_name(self) -> str: + return "NamedExpr" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "NamedExpr(target=") + if self.target is not None: + pos = self.target.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# OpNode 节点 +# ============================================================ + +class OpNode(AST): + """OpNode(op: CInt)""" + op: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, op: t.CInt): + ptr: OpNode | t.CPtr = pool.alloc(OpNode.__sizeof__()) + if ptr: + string.memset(ptr, 0, OpNode.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, op: t.CInt): + _init_ast(self, pool, 0, 0) + self.op = op + + def kind(self) -> t.CInt: + return ASTKind.OpNode + + def type_name(self) -> str: + return "OpNode" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "OpNode(op=") + pos = _emit(buf, size, pos, _op_name(self.op)) + pos = _emit(buf, size, pos, ")") + return pos diff --git a/includes/ast/lexer.py b/includes/ast/lexer.py new file mode 100644 index 0000000..ae01c93 --- /dev/null +++ b/includes/ast/lexer.py @@ -0,0 +1,743 @@ +import t, c +from stdint import * +import memhub +import string +from .tokens import ( + Token, TokenType, TokOp, _init_tables, _kw_lookup, _op_head, + OpEntry, new_token, token_set_str, token_set_str_literal, +) + + +# 缩进栈最大深度 +LEXER_MAX_INDENT: t.CDefine = 256 + + +# ============================================================ +# Lexer 状态结构体 +# ============================================================ +class Lexer: + src: str # 源代码（NUL 结尾） + pos: t.CSizeT # 当前位置 + len: t.CSizeT # 源代码长度 + lineno: t.CInt # 当前行号（从 1 开始） + col: t.CInt # 当前列号（从 0 开始） + pool: memhub.MemManager | t.CPtr # 内存池 + indent_stack: t.CInt | t.CPtr # 缩进栈（int 数组，从 pool 分配） + indent_top: t.CInt # 栈顶索引（-1 表示空栈） + paren_depth: t.CInt # 括号深度（>0 时不产生 TokenType.NewLine） + tokens_head: Token | t.CPtr # token 链表头 + tokens_tail: Token | t.CPtr # token 链表尾 + at_line_start: t.CInt # 1=行首（需处理缩进） + error_count: t.CInt # 错误计数 + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Lexer | t.CPtr = pool.alloc(Lexer.__sizeof__()) + if ptr: + string.memset(ptr, 0, Lexer.__sizeof__()) + return ptr + + +def new_lexer(pool: memhub.MemManager | t.CPtr) -> Lexer | t.CPtr: + """从 pool 分配并返回 Lexer（跨模块安全包装）""" + return Lexer(pool) + + +def _lexer_init(lx: Lexer | t.CPtr, src: str, + pool: memhub.MemManager | t.CPtr): + """初始化 Lexer""" + if lx == None: return + lx.src = src + lx.pos = 0 + lx.len = string.strlen(src) if src != None else 0 + lx.lineno = 1 + lx.col = 0 + lx.pool = pool + # 分配缩进栈 + lx.indent_stack = pool.alloc(LEXER_MAX_INDENT * 8) + if lx.indent_stack != None: + string.memset(lx.indent_stack, 0, LEXER_MAX_INDENT * 8) + lx.indent_top = -1 + # 栈底压入 0（基础缩进） + if lx.indent_stack != None: + lx.indent_stack[0] = 0 + lx.indent_top = 0 + lx.paren_depth = 0 + lx.tokens_head = None + lx.tokens_tail = None + lx.at_line_start = 1 + lx.error_count = 0 + _init_tables(pool) + + +# ============================================================ +# 辅助函数 +# ============================================================ + +def _peek(lx: Lexer | t.CPtr, offset: t.CSizeT) -> t.CChar: + """查看 pos+offset 处字符，越界返回 '\0'""" + p: t.CSizeT = lx.pos + offset + if p >= lx.len: + return '\0' + return lx.src[p] + + +def _peek_cur(lx: Lexer | t.CPtr) -> t.CChar: + if lx.pos >= lx.len: + return '\0' + return lx.src[lx.pos] + + +def _advance(lx: Lexer | t.CPtr) -> t.CChar: + """前进一个字符，返回该字符""" + if lx.pos >= lx.len: + return '\0' + ch: t.CChar = lx.src[lx.pos] + lx.pos += 1 + if ch == '\n': + lx.lineno += 1 + lx.col = 0 + else: + lx.col += 1 + return ch + + +def _is_alpha(ch: t.CChar) -> t.CInt: + if ch >= 'a' and ch <= 'z': return 1 + if ch >= 'A' and ch <= 'Z': return 1 + if ch == '_': return 1 + return 0 + + +def _is_digit(ch: t.CChar) -> t.CInt: + if ch >= '0' and ch <= '9': return 1 + return 0 + + +def _is_alnum(ch: t.CChar) -> t.CInt: + if _is_alpha(ch): return 1 + if _is_digit(ch): return 1 + return 0 + + +def _is_hex(ch: t.CChar) -> t.CInt: + if ch >= '0' and ch <= '9': return 1 + if ch >= 'a' and ch <= 'f': return 1 + if ch >= 'A' and ch <= 'F': return 1 + return 0 + + +def _is_space(ch: t.CChar) -> t.CInt: + if ch == ' ': return 1 + if ch == '\t': return 1 + if ch == '\f': return 1 + if ch == '\v': return 1 + if ch == '\r': return 1 + return 0 + + +def _emit(lx: Lexer | t.CPtr, tok: Token | t.CPtr): + """将 token 追加到链表""" + if tok == None: return + if lx.tokens_head == None: + lx.tokens_head = tok + lx.tokens_tail = tok + else: + lx.tokens_tail.next = tok + lx.tokens_tail = tok + + +def _make_op_token(lx: Lexer | t.CPtr, op_id: t.CInt, length: t.CInt, + lineno: t.CInt, col: t.CInt) -> Token | t.CPtr: + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Op, lineno, col) + if tok == None: return None + tok.op_subtype = op_id + tok.end_col_offset = col + length + return tok + + +# ============================================================ +# 缩进处理 +# ============================================================ + +def _handle_indent(lx: Lexer | t.CPtr): + """处理行首缩进，生成 TokenType.Indent/TokenType.Dedent。 + 跳过空行和注释行（不产生缩进变化）。""" + # 先跳过空行和注释行 + while lx.pos < lx.len: + # 记住行首位置，用于非空行回退 + line_start_pos: t.CSizeT = lx.pos + line_start_col: t.CInt = lx.col + # 跳过空白（不跨行） + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch == ' ' or ch == '\t' or ch == '\f' or ch == '\v': + lx.pos += 1 + lx.col += 1 + elif ch == '\r': + # CRLF 行尾符的 \r，跳过（让 \n 触发空行处理） + lx.pos += 1 + else: + break + if lx.pos >= lx.len: + break + ch = lx.src[lx.pos] + if ch == '\n': + # 空行，跳过 + lx.pos += 1 + lx.lineno += 1 + lx.col = 0 + continue + if ch == '\r': + # CRLF 空行（\r 后跟 \n），跳过 \r + lx.pos += 1 + if lx.pos < lx.len and lx.src[lx.pos] == '\n': + lx.pos += 1 + lx.lineno += 1 + lx.col = 0 + continue + if ch == '#': + # 注释行，跳过到行尾 + while lx.pos < lx.len and lx.src[lx.pos] != '\n': + lx.pos += 1 + continue + # 非空行：回退到行首，让后续缩进计算能读到前导空格 + lx.pos = line_start_pos + lx.col = line_start_col + break + + if lx.pos >= lx.len: + # 文件末尾，不再处理缩进 + lx.at_line_start = 0 + return + + # 计算当前行缩进 + indent: t.CInt = 0 + save_pos: t.CSizeT = lx.pos + save_col: t.CInt = lx.col + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch == ' ': + indent += 1 + lx.pos += 1 + lx.col += 1 + elif ch == '\t': + indent += 8 - (indent % 8) + lx.pos += 1 + lx.col += 1 + else: + break + + # 如果行尾（空行），不处理 + if lx.pos < lx.len: + ch = lx.src[lx.pos] + if ch == '\n' or ch == '\r' or ch == '#': + # 空行或注释行，回退位置让主循环处理 + lx.pos = save_pos + lx.col = save_col + lx.at_line_start = 0 + return + + # 与缩进栈比较 + top_indent: t.CInt = lx.indent_stack[lx.indent_top] + if indent > top_indent: + # TokenType.Indent + if lx.indent_top + 1 >= LEXER_MAX_INDENT: + lx.error_count += 1 + return + lx.indent_top += 1 + lx.indent_stack[lx.indent_top] = indent + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Indent, lx.lineno, 0) + _emit(lx, tok) + else: + while indent < lx.indent_stack[lx.indent_top] and lx.indent_top > 0: + lx.indent_top -= 1 + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Dedent, lx.lineno, 0) + _emit(lx, tok) + lx.at_line_start = 0 + + +# ============================================================ +# 标识符/关键字 +# ============================================================ + +def _read_name(lx: Lexer | t.CPtr): + start: t.CSizeT = lx.pos + start_lineno: t.CInt = lx.lineno + start_col: t.CInt = lx.col + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if _is_alnum(ch): + lx.pos += 1 + lx.col += 1 + else: + break + length: t.CSizeT = lx.pos - start + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Name, start_lineno, start_col) + if tok == None: return + token_set_str(lx.pool, tok, lx.src, start, length) + tok.end_lineno = lx.lineno + tok.end_col_offset = start_col + t.CInt(length) + # 查找关键字 + kw_id: t.CInt = _kw_lookup(tok.str_val) + if kw_id: + tok.kw_subtype = kw_id + _emit(lx, tok) + + +# ============================================================ +# 数字 +# ============================================================ + +def _read_number(lx: Lexer | t.CPtr): + start: t.CSizeT = lx.pos + start_lineno: t.CInt = lx.lineno + start_col: t.CInt = lx.col + is_float: t.CInt = 0 + is_complex: t.CInt = 0 + + ch0: t.CChar = lx.src[lx.pos] + # 处理 0x/0o/0b 前缀 + if ch0 == '0' and lx.pos + 1 < lx.len: + ch1: t.CChar = lx.src[lx.pos + 1] + if ch1 == 'x' or ch1 == 'X': + lx.pos += 2 + lx.col += 2 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if _is_hex(ch) or ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + _emit_number(lx, start, start_lineno, start_col, 0, 0, 0) + return + if ch1 == 'o' or ch1 == 'O': + lx.pos += 2 + lx.col += 2 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch >= '0' and ch <= '7': + lx.pos += 1 + lx.col += 1 + elif ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + _emit_number(lx, start, start_lineno, start_col, 0, 0, 0) + return + if ch1 == 'b' or ch1 == 'B': + lx.pos += 2 + lx.col += 2 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch == '0' or ch == '1' or ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + _emit_number(lx, start, start_lineno, start_col, 0, 0, 0) + return + + # 十进制整数/浮点 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if _is_digit(ch) or ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + + # 小数部分 + if lx.pos < lx.len and lx.src[lx.pos] == '.': + is_float = 1 + lx.pos += 1 + lx.col += 1 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if _is_digit(ch) or ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + + # 指数部分 + if lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch == 'e' or ch == 'E': + is_float = 1 + lx.pos += 1 + lx.col += 1 + if lx.pos < lx.len: + ch = lx.src[lx.pos] + if ch == '+' or ch == '-': + lx.pos += 1 + lx.col += 1 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if _is_digit(ch) or ch == '_': + lx.pos += 1 + lx.col += 1 + else: + break + + # 虚数后缀 + if lx.pos < lx.len and lx.src[lx.pos] == 'j': + is_complex = 1 + is_float = 1 + lx.pos += 1 + lx.col += 1 + + _emit_number(lx, start, start_lineno, start_col, is_float, is_complex, 0) + + +def _emit_number(lx: Lexer | t.CPtr, start: t.CSizeT, + start_lineno: t.CInt, start_col: t.CInt, + is_float: t.CInt, is_complex: t.CInt, is_hex: t.CInt): + length: t.CSizeT = lx.pos - start + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Number, start_lineno, start_col) + if tok == None: return + token_set_str(lx.pool, tok, lx.src, start, length) + tok.is_float = is_float + tok.is_complex = is_complex + tok.end_lineno = lx.lineno + tok.end_col_offset = start_col + t.CInt(length) + # 解析数值 + if is_float: + tok.float_val = string.atof(tok.str_val) + else: + tok.int_val = string.atoll(tok.str_val) + _emit(lx, tok) + + +# ============================================================ +# 字符串 +# ============================================================ + +def _read_string(lx: Lexer | t.CPtr): + """读取字符串字面量。 + 支持前缀: r, b, u, f, rb, br, rf, fr（大小写不敏感） + 支持引号: ', ", ''', """ + start_lineno: t.CInt = lx.lineno + start_col: t.CInt = lx.col + prefix_start: t.CSizeT = lx.pos + + # 读取前缀（r/b/u/f） + is_raw: t.CInt = 0 + is_bytes: t.CInt = 0 + is_fstring: t.CInt = 0 + while lx.pos < lx.len: + ch: t.CChar = lx.src[lx.pos] + if ch == 'r' or ch == 'R': + is_raw = 1 + lx.pos += 1 + lx.col += 1 + elif ch == 'b' or ch == 'B': + is_bytes = 1 + lx.pos += 1 + lx.col += 1 + elif ch == 'u' or ch == 'U': + lx.pos += 1 + lx.col += 1 + elif ch == 'f' or ch == 'F': + is_fstring = 1 + lx.pos += 1 + lx.col += 1 + else: + break + + if lx.pos >= lx.len: + lx.error_count += 1 + return + + # 确定引号 + quote: t.CChar = lx.src[lx.pos] + if quote != '\'' and quote != '"': + lx.error_count += 1 + return + + # 检查三引号 + is_triple: t.CInt = 0 + if lx.pos + 2 < lx.len: + if lx.src[lx.pos] == quote and lx.src[lx.pos + 1] == quote and lx.src[lx.pos + 2] == quote: + is_triple = 1 + + content_start: t.CSizeT = lx.pos + if is_triple: + lx.pos += 3 + lx.col += 3 + else: + lx.pos += 1 + lx.col += 1 + + # 读取字符串内容 + content_buf_start: t.CSizeT = lx.pos + while lx.pos < lx.len: + ch = lx.src[lx.pos] + if is_triple: + if ch == quote: + if lx.pos + 2 < lx.len: + if lx.src[lx.pos + 1] == quote and lx.src[lx.pos + 2] == quote: + break + elif lx.pos + 2 == lx.len: + break + if ch == '\\': + lx.pos += 1 + lx.col += 1 + if lx.pos < lx.len: + if lx.src[lx.pos] == '\n': + lx.lineno += 1 + lx.col = 0 + lx.pos += 1 + lx.col += 1 + continue + if ch == '\n': + lx.lineno += 1 + lx.col = 0 + lx.pos += 1 + lx.col += 1 + else: + if ch == quote: + break + if ch == '\n': + # 单引号字符串不能跨行（除非续行） + lx.error_count += 1 + return + if ch == '\\' and not is_raw: + lx.pos += 1 + lx.col += 1 + if lx.pos < lx.len: + lx.pos += 1 + lx.col += 1 + continue + lx.pos += 1 + lx.col += 1 + + content_end: t.CSizeT = lx.pos + # 跳过结束引号 + if is_triple: + if lx.pos + 2 < lx.len: + lx.pos += 3 + lx.col += 3 + else: + lx.pos = lx.len + else: + if lx.pos < lx.len: + lx.pos += 1 + lx.col += 1 + + # 创建 token + tok: Token | t.CPtr = new_token(lx.pool, TokenType.String, start_lineno, start_col) + if tok == None: return + content_len: t.CSizeT = content_end - content_buf_start + + # 处理转义序列（非 raw 字符串） + if is_raw == 0 and content_len > 0: + # 分配转换缓冲区（最大情况：每个字符都是普通字符，无需扩展） + conv_buf: str = lx.pool.alloc(content_len + 1) + if conv_buf == None: + token_set_str(lx.pool, tok, lx.src, content_buf_start, content_len) + else: + wpos: t.CSizeT = 0 + rpos: t.CSizeT = 0 + while rpos < content_len: + ch: t.CChar = lx.src[content_buf_start + rpos] + if ch == '\\' and rpos + 1 < content_len: + rpos += 1 + next_ch: t.CChar = lx.src[content_buf_start + rpos] + if next_ch == 'n': + conv_buf[wpos] = '\n' + elif next_ch == 't': + conv_buf[wpos] = '\t' + elif next_ch == 'r': + conv_buf[wpos] = '\r' + elif next_ch == '\\': + conv_buf[wpos] = '\\' + elif next_ch == '\'': + conv_buf[wpos] = '\'' + elif next_ch == '"': + conv_buf[wpos] = '"' + elif next_ch == '0': + conv_buf[wpos] = '\0' + else: + # 未知转义，保留原样（包括 \xNN 等以后再处理） + conv_buf[wpos] = '\\' + wpos += 1 + conv_buf[wpos] = next_ch + wpos += 1 + rpos += 1 + else: + conv_buf[wpos] = ch + wpos += 1 + rpos += 1 + conv_buf[wpos] = '\0' + token_set_str_literal(lx.pool, tok, conv_buf) + else: + token_set_str(lx.pool, tok, lx.src, content_buf_start, content_len) + tok.end_lineno = lx.lineno + tok.end_col_offset = lx.col + # flags 字段复用：bit0=raw, bit1=bytes, bit2=fstring, bit3=triple + if is_raw: tok.kw_subtype = tok.kw_subtype | 1 + if is_bytes: tok.kw_subtype = tok.kw_subtype | 2 + if is_fstring: tok.kw_subtype = tok.kw_subtype | 4 + if is_triple: tok.kw_subtype = tok.kw_subtype | 8 + _emit(lx, tok) + + +# ============================================================ +# 运算符 +# ============================================================ + +def _read_operator(lx: Lexer | t.CPtr): + """贪婪匹配运算符""" + start_lineno: t.CInt = lx.lineno + start_col: t.CInt = lx.col + remaining: t.CSizeT = lx.len - lx.pos + + # 遍历运算符表，找最长匹配 + best_entry: OpEntry | t.CPtr = None + best_len: t.CInt = 0 + cur: OpEntry | t.CPtr = _op_head + while cur != None: + if cur.length <= t.CInt(remaining): + # 比较 cur.name 与 src[pos:pos+cur.length] + match: t.CInt = 1 + i: t.CSizeT = 0 + while i < t.CSizeT(cur.length): + if lx.src[lx.pos + i] != cur.name[i]: + match = 0 + break + i += 1 + if match: + if cur.length > best_len: + best_len = cur.length + best_entry = cur + cur = cur.next + + if best_entry == None: + # 未知字符，报错并跳过 + lx.error_count += 1 + lx.pos += 1 + lx.col += 1 + return + + # 更新括号深度 + if best_entry.op_id == TokOp.LPar or best_entry.op_id == TokOp.Lsqb or best_entry.op_id == TokOp.LBrace: + lx.paren_depth += 1 + elif best_entry.op_id == TokOp.RPar or best_entry.op_id == TokOp.Rsqb or best_entry.op_id == TokOp.RBrace: + if lx.paren_depth > 0: + lx.paren_depth -= 1 + + tok: Token | t.CPtr = _make_op_token(lx, best_entry.op_id, best_entry.length, + start_lineno, start_col) + if tok == None: return + token_set_str(lx.pool, tok, lx.src, lx.pos, t.CSizeT(best_len)) + lx.pos += t.CSizeT(best_len) + lx.col += best_len + _emit(lx, tok) + + +# ============================================================ +# 主词法分析入口 +# ============================================================ + +def tokenize(lx: Lexer | t.CPtr) -> Token | t.CPtr: + """执行词法分析，返回 token 链表头""" + if lx == None: return None + while lx.pos < lx.len: + # 行首缩进处理 + if lx.at_line_start and lx.paren_depth == 0: + _handle_indent(lx) + if lx.pos >= lx.len: + break + + ch: t.CChar = _peek_cur(lx) + if ch == '\0': + break + + # 换行 + if ch == '\n': + if lx.paren_depth > 0: + # 括号内，产生 TokenType.Nl + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Nl, lx.lineno, lx.col) + _emit(lx, tok) + else: + # 逻辑换行 + tok: Token | t.CPtr = new_token(lx.pool, TokenType.NewLine, lx.lineno, lx.col) + _emit(lx, tok) + lx.at_line_start = 1 + lx.pos += 1 + lx.lineno += 1 + lx.col = 0 + continue + + # 反斜线续行 + if ch == '\\' and lx.pos + 1 < lx.len and lx.src[lx.pos + 1] == '\n': + lx.pos += 2 + lx.lineno += 1 + lx.col = 0 + continue + + # 空白 + if _is_space(ch): + lx.pos += 1 + lx.col += 1 + continue + + # 注释 + if ch == '#': + while lx.pos < lx.len and lx.src[lx.pos] != '\n': + lx.pos += 1 + continue + + # 字符串前缀（如 r"..."）或字符串 + if ch == '\'' or ch == '"': + _read_string(lx) + continue + # 字符串前缀字母后跟引号（必须在 _is_alpha 之前检测，否则 f/r/b/u 会被当作 TokenType.Name） + if (ch == 'r' or ch == 'R' or ch == 'b' or ch == 'B' or + ch == 'u' or ch == 'U' or ch == 'f' or ch == 'F'): + if lx.pos + 1 < lx.len: + ch1: t.CChar = lx.src[lx.pos + 1] + if ch1 == '\'' or ch1 == '"': + _read_string(lx) + continue + # 双前缀如 rb, br, rf, fr + if lx.pos + 2 < lx.len: + ch2: t.CChar = lx.src[lx.pos + 2] + if ((ch1 == 'r' or ch1 == 'R' or ch1 == 'b' or ch1 == 'B' or + ch1 == 'f' or ch1 == 'F' or ch1 == 'u' or ch1 == 'U') and + (ch2 == '\'' or ch2 == '"')): + _read_string(lx) + continue + + # 标识符/关键字 + if _is_alpha(ch): + _read_name(lx) + continue + + # 数字 + if _is_digit(ch): + _read_number(lx) + continue + + # 运算符 + _read_operator(lx) + + # 文件结尾：产生 TokenType.NewLine（如果有内容） + if lx.tokens_tail != None: + if lx.tokens_tail.type != TokenType.NewLine and lx.tokens_tail.type != TokenType.Nl: + tok: Token | t.CPtr = new_token(lx.pool, TokenType.NewLine, lx.lineno, lx.col) + _emit(lx, tok) + + # 产生剩余 TokenType.Dedent + while lx.indent_top > 0: + lx.indent_top -= 1 + tok: Token | t.CPtr = new_token(lx.pool, TokenType.Dedent, lx.lineno, 0) + _emit(lx, tok) + + # TokenType.EndMarker + tok: Token | t.CPtr = new_token(lx.pool, TokenType.EndMarker, lx.lineno, 0) + _emit(lx, tok) + + return lx.tokens_head diff --git a/includes/ast/match.py b/includes/ast/match.py new file mode 100644 index 0000000..e3df402 --- /dev/null +++ b/includes/ast/match.py @@ -0,0 +1,363 @@ +import t, c +from stdint import * +import memhub +import string +from .base import ( + AST, ASTKind, + _init_ast, _copy_str, _set_parent_list, + _emit, _emit_str, _emit_int, _dump_list, +) + + +# ============================================================ +# MatchCase 节点 +# ============================================================ + +class MatchCase(AST): + """MatchCase(pattern: AST, guard: AST)""" + pattern: AST | t.CPtr + guard: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + pattern: AST | t.CPtr, guard: AST | t.CPtr): + ptr: MatchCase | t.CPtr = pool.alloc(MatchCase.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchCase.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + pattern: AST | t.CPtr, guard: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.pattern = pattern + self.guard = guard + + def kind(self) -> t.CInt: + return ASTKind.MatchCase + + def type_name(self) -> str: + return "MatchCase" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchCase(pattern=") + if self.pattern is not None: + pos = self.pattern.dump(buf, size, pos) + pos = _emit(buf, size, pos, ", guard=") + if self.guard is not None: + pos = self.guard.dump(buf, size, pos) + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# MatchValue 节点 +# ============================================================ + +class MatchValue(AST): + """MatchValue(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: MatchValue | t.CPtr = pool.alloc(MatchValue.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchValue.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.MatchValue + + def type_name(self) -> str: + return "MatchValue" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchValue(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# MatchSingleton 节点 +# ============================================================ + +class MatchSingleton(AST): + """MatchSingleton(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: MatchSingleton | t.CPtr = pool.alloc(MatchSingleton.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchSingleton.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.MatchSingleton + + def type_name(self) -> str: + return "MatchSingleton" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchSingleton(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# MatchSequence 节点 +# ============================================================ + +class MatchSequence(AST): + """MatchSequence(patterns: list[AST | t.CPtr])""" + patterns: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr): + ptr: MatchSequence | t.CPtr = pool.alloc(MatchSequence.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchSequence.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.patterns = patterns + _set_parent_list(patterns, self) + + def kind(self) -> t.CInt: + return ASTKind.MatchSequence + + def type_name(self) -> str: + return "MatchSequence" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchSequence(patterns=") + pos = _dump_list(self.patterns, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# MatchMapping 节点 +# ============================================================ + +class MatchMapping(AST): + """MatchMapping(keys: list[AST | t.CPtr], patterns: list[AST | t.CPtr], rest: str)""" + keys: list[AST | t.CPtr] | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + rest: str + + def __new__(self, pool: memhub.MemManager | t.CPtr, + keys: list[AST | t.CPtr] | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr, + rest: str): + ptr: MatchMapping | t.CPtr = pool.alloc(MatchMapping.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchMapping.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + keys: list[AST | t.CPtr] | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr, + rest: str): + _init_ast(self, pool, 0, 0) + self.keys = keys + self.patterns = patterns + self.rest = _copy_str(pool, rest) + _set_parent_list(keys, self) + _set_parent_list(patterns, self) + + def kind(self) -> t.CInt: + return ASTKind.MatchMapping + + def type_name(self) -> str: + return "MatchMapping" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchMapping(keys=") + pos = _dump_list(self.keys, buf, size, pos) + pos = _emit(buf, size, pos, ", patterns=") + pos = _dump_list(self.patterns, buf, size, pos) + pos = _emit(buf, size, pos, ", rest='") + pos = _emit_str(buf, size, pos, self.rest) + pos = _emit(buf, size, pos, "')") + return pos + + +# ============================================================ +# MatchClass 节点 +# ============================================================ + +class MatchClass(AST): + """MatchClass(cls: AST, patterns: list[AST | t.CPtr], + kwd_attrs: list[AST | t.CPtr], kwd_patterns: list[AST | t.CPtr])""" + cls: AST | t.CPtr + patterns: list[AST | t.CPtr] | t.CPtr + kwd_attrs: list[AST | t.CPtr] | t.CPtr + kwd_patterns: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + cls: AST | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr, + kwd_attrs: list[AST | t.CPtr] | t.CPtr, + kwd_patterns: list[AST | t.CPtr] | t.CPtr): + ptr: MatchClass | t.CPtr = pool.alloc(MatchClass.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchClass.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + cls: AST | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr, + kwd_attrs: list[AST | t.CPtr] | t.CPtr, + kwd_patterns: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.cls = cls + self.patterns = patterns + self.kwd_attrs = kwd_attrs + self.kwd_patterns = kwd_patterns + _set_parent_list(patterns, self) + _set_parent_list(kwd_attrs, self) + _set_parent_list(kwd_patterns, self) + + def kind(self) -> t.CInt: + return ASTKind.MatchClass + + def type_name(self) -> str: + return "MatchClass" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchClass(cls=") + if self.cls is not None: + pos = self.cls.dump(buf, size, pos) + pos = _emit(buf, size, pos, ", patterns=") + pos = _dump_list(self.patterns, buf, size, pos) + pos = _emit(buf, size, pos, ", kwd_attrs=") + pos = _dump_list(self.kwd_attrs, buf, size, pos) + pos = _emit(buf, size, pos, ", kwd_patterns=") + pos = _dump_list(self.kwd_patterns, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# MatchStar 节点 +# ============================================================ + +class MatchStar(AST): + """MatchStar(name: str)""" + name: str + + def __new__(self, pool: memhub.MemManager | t.CPtr, name: str): + ptr: MatchStar | t.CPtr = pool.alloc(MatchStar.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchStar.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, name: str): + _init_ast(self, pool, 0, 0) + self.name = _copy_str(pool, name) + + def kind(self) -> t.CInt: + return ASTKind.MatchStar + + def type_name(self) -> str: + return "MatchStar" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchStar(name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "')") + return pos + + +# ============================================================ +# MatchAs 节点 +# ============================================================ + +class MatchAs(AST): + """MatchAs(pattern: AST, name: str)""" + pattern: AST | t.CPtr + name: str + + def __new__(self, pool: memhub.MemManager | t.CPtr, + pattern: AST | t.CPtr, name: str): + ptr: MatchAs | t.CPtr = pool.alloc(MatchAs.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchAs.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + pattern: AST | t.CPtr, name: str): + _init_ast(self, pool, 0, 0) + self.pattern = pattern + self.name = _copy_str(pool, name) + + def kind(self) -> t.CInt: + return ASTKind.MatchAs + + def type_name(self) -> str: + return "MatchAs" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchAs(pattern=") + if self.pattern is not None: + pos = self.pattern.dump(buf, size, pos) + pos = _emit(buf, size, pos, ", name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "')") + return pos + + +# ============================================================ +# MatchOr 节点 +# ============================================================ + +class MatchOr(AST): + """MatchOr(patterns: list[AST | t.CPtr])""" + patterns: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr): + ptr: MatchOr | t.CPtr = pool.alloc(MatchOr.__sizeof__()) + if ptr: + string.memset(ptr, 0, MatchOr.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + patterns: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.patterns = patterns + _set_parent_list(patterns, self) + + def kind(self) -> t.CInt: + return ASTKind.MatchOr + + def type_name(self) -> str: + return "MatchOr" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "MatchOr(patterns=") + pos = _dump_list(self.patterns, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos diff --git a/includes/ast/parser.py b/includes/ast/parser.py new file mode 100644 index 0000000..f1f8d7e --- /dev/null +++ b/includes/ast/parser.py @@ -0,0 +1,2345 @@ +import t, c +from stdint import * +import memhub +import string +from .tokens import (Token, TokenType, Keyword, TokOp) +from .base import ( + AST, ASTCtx, ASTKind, OpKind, ASTFlag, + _set_pos, _binop_from_op, _augop_from_op, _cmpop_from_op, _copy_str, _inherit_pos, + _init_ast, + CONST_INT, CONST_FLOAT, CONST_STR, CONST_BOOL, CONST_NONE, + FLAG_IS_ASYNC, +) +from .exprs import ( + BoolOp, BinOp, UnaryOp, Lambda, IfExp, Dict, Set, + ListComp, SetComp, DictComp, GeneratorExp, + Await, Yield, YieldFrom, FormattedValue, JoinedStr, + Constant, NamedExpr, Attribute, Subscript, Starred, Name, + List, Tuple, Slice, Call, Compare, OpNode, +) +from .stmts import ( + Module, Expression, Interactive, FunctionType, + FunctionDef, ClassDef, Return, Delete, Assign, AugAssign, AnnAssign, + For, While, If, With, Raise, Try, Assert, Global, Nonlocal, + Pass, Break, Continue, Expr, Import, ImportFrom, Match, +) +from .astaux import ( + ExceptHandler, Arguments, Arg, Keyword as KwNode, Alias, WithItem, Comprehension, +) +from .match import ( + MatchCase, MatchValue, MatchSingleton, MatchSequence, + MatchMapping, MatchClass, MatchStar, MatchAs, MatchOr, +) +from .lexer import Lexer, _lexer_init, tokenize, new_lexer + + +# ============================================================ +# Parser 状态 +# ============================================================ +class Parser: + tokens: Token | t.CPtr # token 链表头 + cur: Token | t.CPtr # 当前 token + pool: memhub.MemManager | t.CPtr # 内存池 + error_count: t.CInt # 错误计数 + pending_decorators: list[AST | t.CPtr] | t.CPtr # 待处理的装饰器 list + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Parser | t.CPtr = pool.alloc(Parser.__sizeof__()) + if ptr: + string.memset(ptr, 0, Parser.__sizeof__()) + return ptr + + +def new_parser(pool: memhub.MemManager | t.CPtr) -> Parser | t.CPtr: + """从 pool 分配并返回 Parser（跨模块安全包装）""" + return Parser(pool) + + +def _parser_init(ps: Parser | t.CPtr, tokens: Token | t.CPtr, + pool: memhub.MemManager | t.CPtr): + if ps is None: return + ps.tokens = tokens + ps.cur = tokens + ps.pool = pool + ps.error_count = 0 + ps.pending_decorators = None + + +# ============================================================ +# Token 导航辅助 +# ============================================================ + +def _advance(ps: Parser | t.CPtr): + """前进到下一个 token""" + if ps.cur is None: return + if ps.cur.next is not None: + ps.cur = ps.cur.next + + +def _cur_type(ps: Parser | t.CPtr) -> t.CInt: + if ps.cur is None: return TokenType.EndMarker + return ps.cur.type + + +def _cur_op(ps: Parser | t.CPtr) -> t.CInt: + if ps.cur is None: return 0 + return ps.cur.op_subtype + + +def _cur_kw(ps: Parser | t.CPtr) -> t.CInt: + if ps.cur is None: return 0 + return ps.cur.kw_subtype + + +def _cur_str(ps: Parser | t.CPtr) -> str: + if ps.cur is None: return None + return ps.cur.str_val + + +def _cur_lineno(ps: Parser | t.CPtr) -> t.CInt: + if ps.cur is None: return 0 + return ps.cur.lineno + + +def _cur_col(ps: Parser | t.CPtr) -> t.CInt: + if ps.cur is None: return 0 + return ps.cur.col_offset + + +def _skip_nl(ps: Parser | t.CPtr): + """跳过 TokenType.Nl token""" + while ps.cur is not None and ps.cur.type == TokenType.Nl: + ps.cur = ps.cur.next + + +def _skip_nls_and_newlines(ps: Parser | t.CPtr): + """跳过 TokenType.Nl 和 TokenType.NewLine（用于续行）""" + while ps.cur is not None: + if ps.cur.type == TokenType.Nl or ps.cur.type == TokenType.NewLine: + ps.cur = ps.cur.next + else: + break + + +def _expect_type(ps: Parser | t.CPtr, ttype: t.CInt) -> t.CInt: + """期望当前 token 是指定类型，是则前进并返回 1，否则返回 0""" + if _cur_type(ps) == ttype: + _advance(ps) + return 1 + ps.error_count += 1 + return 0 + + +def _expect_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: + """期望当前是某个运算符""" + if _cur_type(ps) == TokenType.Op and _cur_op(ps) == op_id: + _advance(ps) + return 1 + ps.error_count += 1 + return 0 + + +def _match_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: + """检查但不消耗""" + if _cur_type(ps) == TokenType.Op and _cur_op(ps) == op_id: + return 1 + return 0 + + +def _accept_op(ps: Parser | t.CPtr, op_id: t.CInt) -> t.CInt: + """如果匹配则消耗并返回 1""" + if _match_op(ps, op_id): + _advance(ps) + return 1 + return 0 + + +# ============================================================ +# 常量构造辅助（新 API：Constant 直接接受 lineno/col） +# ============================================================ + +def _new_const_int(pool: memhub.MemManager | t.CPtr, val: t.CInt64T, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + return Constant(pool, CONST_INT, val, 0.0, None, lineno, col) + + +def _new_const_float(pool: memhub.MemManager | t.CPtr, val: t.CDouble, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + return Constant(pool, CONST_FLOAT, 0, val, None, lineno, col) + + +def _new_const_str(pool: memhub.MemManager | t.CPtr, val: str, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + return Constant(pool, CONST_STR, 0, 0.0, val, lineno, col) + + +def _new_const_bool(pool: memhub.MemManager | t.CPtr, val: t.CInt, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + return Constant(pool, CONST_BOOL, val, 0.0, None, lineno, col) + + +def _new_const_none(pool: memhub.MemManager | t.CPtr, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + return Constant(pool, CONST_NONE, 0, 0.0, None, lineno, col) + + +# ============================================================ +# ctx 辅助（递归修正赋值目标的 ctx） +# ============================================================ + +def _set_ctx(node: AST | t.CPtr, ctx: t.CInt): + """设置节点的 ctx 字段（按 kind 转型后赋值）""" + if node is None: return + k: t.CInt = node.kind() + if k == ASTKind.Name: + nm: Name | t.CPtr = (Name | t.CPtr)(node) + nm.ctx = ctx + elif k == ASTKind.Attribute: + at: Attribute | t.CPtr = (Attribute | t.CPtr)(node) + at.ctx = ctx + elif k == ASTKind.Subscript: + sb: Subscript | t.CPtr = (Subscript | t.CPtr)(node) + sb.ctx = ctx + elif k == ASTKind.Tuple: + tp: Tuple | t.CPtr = (Tuple | t.CPtr)(node) + tp.ctx = ctx + elif k == ASTKind.List: + ls: List | t.CPtr = (List | t.CPtr)(node) + ls.ctx = ctx + elif k == ASTKind.Starred: + st: Starred | t.CPtr = (Starred | t.CPtr)(node) + st.ctx = ctx + + +def _fix_store_ctx(node: AST | t.CPtr): + """递归修正赋值目标的 ctx 为 Store""" + if node is None: return + _set_ctx(node, ASTCtx.Store) + k: t.CInt = node.kind() + if k == ASTKind.Tuple or k == ASTKind.List: + tp_elts: list[AST | t.CPtr] | t.CPtr = None + if k == ASTKind.Tuple: + tp: Tuple | t.CPtr = (Tuple | t.CPtr)(node) + tp_elts = tp.elts + else: + ls: List | t.CPtr = (List | t.CPtr)(node) + tp_elts = ls.elts + if tp_elts is not None: + n: t.CSizeT = tp_elts.__len__() + i: t.CSizeT = 0 + while i < n: + cur: AST | t.CPtr = tp_elts.get(i) + _fix_store_ctx(cur) + i += 1 + elif k == ASTKind.Starred: + st2: Starred | t.CPtr = (Starred | t.CPtr)(node) + _fix_store_ctx(st2.value) + + +# ============================================================ +# 表达式解析 +# ============================================================ + +def _parse_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: + """expr: yield_expr | namedexpr_test""" + if _cur_kw(ps) == Keyword.Yield: + return _parse_yield(ps) + return _parse_namedexpr(ps) + + +def _parse_namedexpr(ps: Parser | t.CPtr) -> AST | t.CPtr: + """namedexpr_test: test [':=' test]""" + left: AST | t.CPtr = _parse_test(ps) + if left is None: return None + if _match_op(ps, TokOp.ColonEq): + _advance(ps) + right: AST | t.CPtr = _parse_test(ps) + node: AST | t.CPtr = NamedExpr(ps.pool, left, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + return node + return left + + +def _parse_test(ps: Parser | t.CPtr) -> AST | t.CPtr: + """test: or_test ['if' or_test 'else' test] | lambdef""" + if _cur_kw(ps) == Keyword.Lambda: + return _parse_lambda(ps) + left: AST | t.CPtr = _parse_or_test(ps) + if left is None: return None + if _cur_kw(ps) == Keyword.If: + _advance(ps) + test: AST | t.CPtr = _parse_or_test(ps) + if _cur_kw(ps) == Keyword.Else: + _advance(ps) + orelse: AST | t.CPtr = _parse_test(ps) + node: AST | t.CPtr = IfExp(ps.pool, test, left, orelse) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + return node + ps.error_count += 1 + return left + return left + + +def _parse_lambda(ps: Parser | t.CPtr) -> AST | t.CPtr: + """lambdef: 'lambda' [varargslist] ':' test""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # lambda + args: AST | t.CPtr = None + if not _match_op(ps, TokOp.Colon): + args = _parse_varargslist(ps, 1) + if not _expect_op(ps, TokOp.Colon): + return None + body: AST | t.CPtr = _parse_test(ps) + node: AST | t.CPtr = Lambda(ps.pool, args, body) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_or_test(ps: Parser | t.CPtr) -> AST | t.CPtr: + """or_test: and_test ('or' and_test)*""" + left: AST | t.CPtr = _parse_and_test(ps) + if left is None: return None + if _cur_kw(ps) == Keyword.Or: + values: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + values.append(left) + while _cur_kw(ps) == Keyword.Or: + _advance(ps) + right: AST | t.CPtr = _parse_and_test(ps) + if right is None: break + values.append(right) + node: AST | t.CPtr = BoolOp(ps.pool, OpKind.Or, values) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + return node + return left + + +def _parse_and_test(ps: Parser | t.CPtr) -> AST | t.CPtr: + """and_test: not_test ('and' not_test)*""" + left: AST | t.CPtr = _parse_not_test(ps) + if left is None: return None + if _cur_kw(ps) == Keyword.And: + values: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + values.append(left) + while _cur_kw(ps) == Keyword.And: + _advance(ps) + right: AST | t.CPtr = _parse_not_test(ps) + if right is None: break + values.append(right) + node: AST | t.CPtr = BoolOp(ps.pool, OpKind.And, values) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + return node + return left + + +def _parse_not_test(ps: Parser | t.CPtr) -> AST | t.CPtr: + """not_test: 'not' not_test | comparison""" + if _cur_kw(ps) == Keyword.Not: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + operand: AST | t.CPtr = _parse_not_test(ps) + node: AST | t.CPtr = UnaryOp(ps.pool, OpKind.Not, operand) + _set_pos(node, lineno, col, 0, 0) + return node + return _parse_comparison(ps) + + +def _parse_comparison(ps: Parser | t.CPtr) -> AST | t.CPtr: + """comparison: bitor_expr (comp_op bitor_expr)* + + 单比较设计：a < b → Compare(left=a, op=Lt, right=b) + 链式比较 a < b < c → BoolOp(And, [Compare(a,<,b), Compare(b,<,c)]) + """ + left: AST | t.CPtr = _parse_bitor(ps) + if left is None: return None + + op_id: t.CInt = _match_comp_op(ps) + if op_id == OpKind.NoneOp: + return left + _advance_comp_op(ps) + right: AST | t.CPtr = _parse_bitor(ps) + # 单比较：构造 ops 和 comparators list + ops: list[t.CInt] | t.CPtr = list[t.CInt](ps.pool, 8) + ops.append(op_id) + comparators: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + comparators.append(right) + first_cmp: AST | t.CPtr = Compare(ps.pool, left, ops, comparators) + _set_pos(first_cmp, left.lineno, left.col_offset, 0, 0) + + # 检查链式比较 + next_op: t.CInt = _match_comp_op(ps) + if next_op == OpKind.NoneOp: + return first_cmp + + # 链式：a < b < c → BoolOp(And, [Compare(a,<,b), Compare(b,<,c), ...]) + values: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + values.append(first_cmp) + cur_left: AST | t.CPtr = right + while next_op != OpKind.NoneOp: + _advance_comp_op(ps) + cur_right: AST | t.CPtr = _parse_bitor(ps) + ops2: list[t.CInt] | t.CPtr = list[t.CInt](ps.pool, 8) + ops2.append(next_op) + comparators2: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + comparators2.append(cur_right) + cmp: AST | t.CPtr = Compare(ps.pool, cur_left, ops2, comparators2) + _set_pos(cmp, cur_left.lineno, cur_left.col_offset, 0, 0) + values.append(cmp) + cur_left = cur_right + next_op = _match_comp_op(ps) + node: AST | t.CPtr = BoolOp(ps.pool, OpKind.And, values) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + return node + + +def _match_comp_op(ps: Parser | t.CPtr) -> t.CInt: + """检查当前是否是比较运算符，返回 OP_* 或 OpKind.NoneOp""" + if _cur_type(ps) == TokenType.Op: + op_id: t.CInt = _cur_op(ps) + if op_id == TokOp.EqEq: return OpKind.Eq + if op_id == TokOp.ExclaimEq: return OpKind.Ne + if op_id == TokOp.Less: return OpKind.Lt + if op_id == TokOp.LessEq: return OpKind.Le + if op_id == TokOp.Greater: return OpKind.Gt + if op_id == TokOp.GreaterEq: return OpKind.Ge + if _cur_kw(ps) == Keyword.Is: + if ps.cur.next is not None and ps.cur.next.type == TokenType.Name and ps.cur.next.kw_subtype == Keyword.Not: + return OpKind.IsNot + return OpKind.Is + if _cur_kw(ps) == Keyword.Not: + if ps.cur.next is not None and ps.cur.next.kw_subtype == Keyword.In: + return OpKind.NotIn + if _cur_kw(ps) == Keyword.In: + return OpKind.In + return OpKind.NoneOp + + +def _advance_comp_op(ps: Parser | t.CPtr): + """消耗比较运算符 token""" + if _cur_type(ps) == TokenType.Op: + _advance(ps) + return + if _cur_kw(ps) == Keyword.Is: + _advance(ps) + if _cur_kw(ps) == Keyword.Not: + _advance(ps) + return + if _cur_kw(ps) == Keyword.Not: + _advance(ps) + if _cur_kw(ps) == Keyword.In: + _advance(ps) + return + if _cur_kw(ps) == Keyword.In: + _advance(ps) + + +def _parse_bitor(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_bitxor(ps) + if left is None: return None + while _match_op(ps, TokOp.VBar): + _advance(ps) + right: AST | t.CPtr = _parse_bitxor(ps) + node: AST | t.CPtr = BinOp(ps.pool, left, OpKind.BitOr, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_bitxor(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_bitand(ps) + if left is None: return None + while _match_op(ps, TokOp.Caret): + _advance(ps) + right: AST | t.CPtr = _parse_bitand(ps) + node: AST | t.CPtr = BinOp(ps.pool, left, OpKind.BitXor, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_bitand(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_shift(ps) + if left is None: return None + while _match_op(ps, TokOp.Amp): + _advance(ps) + right: AST | t.CPtr = _parse_shift(ps) + node: AST | t.CPtr = BinOp(ps.pool, left, OpKind.BitAnd, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_shift(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_arith(ps) + if left is None: return None + while _match_op(ps, TokOp.LtLt) or _match_op(ps, TokOp.GtGt): + op_token: t.CInt = _cur_op(ps) + _advance(ps) + right: AST | t.CPtr = _parse_arith(ps) + ast_op: t.CInt = OpKind.LShift + if op_token == TokOp.GtGt: + ast_op = OpKind.RShift + node: AST | t.CPtr = BinOp(ps.pool, left, ast_op, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_arith(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_term(ps) + if left is None: return None + while _match_op(ps, TokOp.Plus) or _match_op(ps, TokOp.Minus): + op_token: t.CInt = _cur_op(ps) + _advance(ps) + right: AST | t.CPtr = _parse_term(ps) + ast_op: t.CInt = OpKind.Add + if op_token == TokOp.Minus: + ast_op = OpKind.Sub + node: AST | t.CPtr = BinOp(ps.pool, left, ast_op, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_term(ps: Parser | t.CPtr) -> AST | t.CPtr: + left: AST | t.CPtr = _parse_factor(ps) + if left is None: return None + while True: + op_token: t.CInt = 0 + if _match_op(ps, TokOp.Star): op_token = TokOp.Star + elif _match_op(ps, TokOp.Slash): op_token = TokOp.Slash + elif _match_op(ps, TokOp.DSlash): op_token = TokOp.DSlash + elif _match_op(ps, TokOp.Percent): op_token = TokOp.Percent + elif _match_op(ps, TokOp.At): op_token = TokOp.At + if op_token == 0: break + _advance(ps) + right: AST | t.CPtr = _parse_factor(ps) + ast_op: t.CInt = _binop_from_op(op_token) + node: AST | t.CPtr = BinOp(ps.pool, left, ast_op, right) + _set_pos(node, left.lineno, left.col_offset, 0, 0) + left = node + return left + + +def _parse_factor(ps: Parser | t.CPtr) -> AST | t.CPtr: + """factor: '+' factor | '-' factor | '~' factor | power""" + if _match_op(ps, TokOp.Plus) or _match_op(ps, TokOp.Minus) or _match_op(ps, TokOp.Tilde): + op_token: t.CInt = _cur_op(ps) + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + operand: AST | t.CPtr = _parse_factor(ps) + ast_op: t.CInt = OpKind.UAdd + if op_token == TokOp.Minus: + ast_op = OpKind.USub + elif op_token == TokOp.Tilde: + ast_op = OpKind.Invert + node: AST | t.CPtr = UnaryOp(ps.pool, ast_op, operand) + _set_pos(node, lineno, col, 0, 0) + return node + return _parse_power(ps) + + +def _parse_power(ps: Parser | t.CPtr) -> AST | t.CPtr: + """power: await_expr ['**' factor] | atom_expr ['**' factor]""" + base: AST | t.CPtr = _parse_atom_expr(ps) + if base is None: return None + if _match_op(ps, TokOp.StarStar): + _advance(ps) + exp: AST | t.CPtr = _parse_factor(ps) + node: AST | t.CPtr = BinOp(ps.pool, base, OpKind.Pow, exp) + _set_pos(node, base.lineno, base.col_offset, 0, 0) + return node + return base + + +def _parse_atom_expr(ps: Parser | t.CPtr) -> AST | t.CPtr: + """atom_expr: ['await'] atom trailer*""" + if _cur_kw(ps) == Keyword.Await: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + inner: AST | t.CPtr = _parse_atom_expr(ps) + node: AST | t.CPtr = Await(ps.pool, inner) + _set_pos(node, lineno, col, 0, 0) + return node + base: AST | t.CPtr = _parse_atom(ps) + if base is None: return None + # trailer 循环 + while True: + if _match_op(ps, TokOp.Dot): + _advance(ps) + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + break + attr_name: str = _cur_str(ps) + attr_lineno: t.CInt = _cur_lineno(ps) + attr_col: t.CInt = _cur_col(ps) + _advance(ps) + ctx: t.CInt = ASTCtx.Load + if _match_op(ps, TokOp.Equal): + ctx = ASTCtx.Store + elif _match_op(ps, TokOp.PlusEq) or _match_op(ps, TokOp.MinusEq): + ctx = ASTCtx.Store + base = Attribute(ps.pool, base, attr_name, ctx) + _set_pos(base, attr_lineno, attr_col, 0, 0) + elif _match_op(ps, TokOp.LPar): + base = _parse_call(ps, base) + elif _match_op(ps, TokOp.Lsqb): + base = _parse_subscript(ps, base) + else: + break + return base + + +def _parse_call(ps: Parser | t.CPtr, func: AST | t.CPtr) -> AST | t.CPtr: + """解析函数调用 trailer。args 和 keywords 均为 list。""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # ( + _skip_nls_and_newlines(ps) # 跳过 ( 后的换行 + args: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + keywords: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + + if not _match_op(ps, TokOp.RPar): + while True: + # **expr + if _match_op(ps, TokOp.StarStar): + _advance(ps) + _skip_nls_and_newlines(ps) + val: AST | t.CPtr = _parse_expr(ps) + kw: AST | t.CPtr = KwNode(ps.pool, None, val) + _set_pos(kw, _cur_lineno(ps), _cur_col(ps), 0, 0) + keywords.append(kw) + # *expr + elif _match_op(ps, TokOp.Star): + _advance(ps) + _skip_nls_and_newlines(ps) + val = _parse_expr(ps) + star: AST | t.CPtr = Starred(ps.pool, val, ASTCtx.Load) + _set_pos(star, val.lineno, val.col_offset, 0, 0) + args.append(star) + # keyword = expr + elif _cur_type(ps) == TokenType.Name: + if ps.cur.next is not None and ps.cur.next.type == TokenType.Op and ps.cur.next.op_subtype == TokOp.Equal: + kw_name: str = _cur_str(ps) + kw_lineno: t.CInt = _cur_lineno(ps) + kw_col: t.CInt = _cur_col(ps) + _advance(ps) # name + _advance(ps) # = + _skip_nls_and_newlines(ps) + val = _parse_expr(ps) + kw = KwNode(ps.pool, kw_name, val) + _set_pos(kw, kw_lineno, kw_col, 0, 0) + keywords.append(kw) + else: + val = _parse_expr(ps) + args.append(val) + else: + val = _parse_expr(ps) + args.append(val) + _skip_nls_and_newlines(ps) # 跳过参数后的换行 + if not _accept_op(ps, TokOp.Comma): + break + _skip_nls_and_newlines(ps) # 跳过逗号后的换行 + _skip_nls_and_newlines(ps) # 跳过 ) 前的换行 + _expect_op(ps, TokOp.RPar) + node: AST | t.CPtr = Call(ps.pool, func, args, keywords) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_subscript(ps: Parser | t.CPtr, value: AST | t.CPtr) -> AST | t.CPtr: + """解析下标 trailer""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # [ + _skip_nl(ps) + slice_node: AST | t.CPtr = _parse_slice(ps) + _expect_op(ps, TokOp.Rsqb) + ctx: t.CInt = ASTCtx.Load + if _match_op(ps, TokOp.Equal): + ctx = ASTCtx.Store + node: AST | t.CPtr = Subscript(ps.pool, value, slice_node, ctx) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_slice(ps: Parser | t.CPtr) -> AST | t.CPtr: + """解析 slice：[expr] ':' [expr] [':' [expr]] | expr | expr, expr, ...""" + lower: AST | t.CPtr = None + if not _match_op(ps, TokOp.Colon): + lower = _parse_expr(ps) + + # 检查是否有逗号（元组语法: a[x, y, ...]） + if _match_op(ps, TokOp.Comma): + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + if lower is not None: + elts.append(lower) + while _accept_op(ps, TokOp.Comma): + if _match_op(ps, TokOp.Rsqb): + break # 尾随逗号 + elem: AST | t.CPtr = _parse_expr(ps) + if elem is not None: + elts.append(elem) + lineno: t.CInt = 0 + if lower is not None: + lineno = lower.lineno + node: AST | t.CPtr = Tuple(ps.pool, elts, ASTCtx.Load) + _set_pos(node, lineno, 0, 0, 0) + return node + + if _match_op(ps, TokOp.Colon): + _advance(ps) + upper: AST | t.CPtr = None + if not _match_op(ps, TokOp.Colon) and not _match_op(ps, TokOp.Rsqb): + upper = _parse_expr(ps) + step: AST | t.CPtr = None + if _accept_op(ps, TokOp.Colon): + if not _match_op(ps, TokOp.Rsqb): + step = _parse_expr(ps) + lineno: t.CInt = 0 + if lower is not None: + lineno = lower.lineno + elif upper is not None: + lineno = upper.lineno + else: + lineno = _cur_lineno(ps) + node: AST | t.CPtr = Slice(ps.pool, lower, upper, step) + _set_pos(node, lineno, 0, 0, 0) + return node + return lower + + +def _parse_atom(ps: Parser | t.CPtr) -> AST | t.CPtr: + """atom: 最小表达式单元""" + ttype: t.CInt = _cur_type(ps) + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + + if ttype == TokenType.Name: + kw: t.CInt = _cur_kw(ps) + if kw == Keyword.True_: + _advance(ps) + return _new_const_bool(ps.pool, 1, lineno, col) + if kw == Keyword.False_: + _advance(ps) + return _new_const_bool(ps.pool, 0, lineno, col) + if kw == Keyword.None_: + _advance(ps) + return _new_const_none(ps.pool, lineno, col) + name_str: str = _cur_str(ps) + _advance(ps) + node: AST | t.CPtr = Name(ps.pool, name_str, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + + if ttype == TokenType.Number: + tok: Token | t.CPtr = ps.cur + _advance(ps) + if tok.is_float: + return _new_const_float(ps.pool, tok.float_val, lineno, col) + return _new_const_int(ps.pool, tok.int_val, lineno, col) + + if ttype == TokenType.String: + sval: str = _cur_str(ps) + is_fstr: t.CInt = ps.cur.kw_subtype & 4 + _advance(ps) + if is_fstr: + return _parse_fstring(ps, sval, lineno, col) + return _new_const_str(ps.pool, sval, lineno, col) + + if _match_op(ps, TokOp.Ellipsis): + _advance(ps) + return _new_const_none(ps.pool, lineno, col) + + if _match_op(ps, TokOp.LPar): + return _parse_paren(ps) + if _match_op(ps, TokOp.Lsqb): + return _parse_list(ps) + if _match_op(ps, TokOp.LBrace): + return _parse_brace(ps) + + ps.error_count += 1 + return None + + +def _parse_paren(ps: Parser | t.CPtr) -> AST | t.CPtr: + """括号表达式: () | (expr) | (expr,) | (expr, ...) | (yield)""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # ( + if _match_op(ps, TokOp.RPar): + _advance(ps) + # 空元组 + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = Tuple(ps.pool, empty, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + # yield + if _cur_kw(ps) == Keyword.Yield: + y: AST | t.CPtr = _parse_yield(ps) + _expect_op(ps, TokOp.RPar) + return y + first: AST | t.CPtr = _parse_expr(ps) + if _match_op(ps, TokOp.Comma): + # 元组 + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + while _accept_op(ps, TokOp.Comma): + if _match_op(ps, TokOp.RPar): + break + e: AST | t.CPtr = _parse_expr(ps) + elts.append(e) + _expect_op(ps, TokOp.RPar) + node = Tuple(ps.pool, elts, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + _expect_op(ps, TokOp.RPar) + # 检查推导式 + if _cur_kw(ps) == Keyword.For: + return _parse_comprehension(ps, first, ASTKind.GeneratorExp, lineno, col) + return first + + +def _parse_list(ps: Parser | t.CPtr) -> AST | t.CPtr: + """列表: [] | [expr_list] | [expr_list comp_for]""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # [ + if _match_op(ps, TokOp.Rsqb): + _advance(ps) + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = List(ps.pool, empty, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + first: AST | t.CPtr = _parse_expr(ps) + if _cur_kw(ps) == Keyword.For: + node = _parse_comprehension(ps, first, ASTKind.ListComp, lineno, col) + _expect_op(ps, TokOp.Rsqb) + return node + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + while _accept_op(ps, TokOp.Comma): + _skip_nl(ps) + if _match_op(ps, TokOp.Rsqb): + break + e: AST | t.CPtr = _parse_expr(ps) + elts.append(e) + _expect_op(ps, TokOp.Rsqb) + node = List(ps.pool, elts, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_brace(ps: Parser | t.CPtr) -> AST | t.CPtr: + """花括号: {} | {set_items} | {dict_items} | {comp_for}""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # { + _skip_nl(ps) + if _match_op(ps, TokOp.RBrace): + _advance(ps) + empty_k: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + empty_v: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = Dict(ps.pool, empty_k, empty_v) + _set_pos(node, lineno, col, 0, 0) + return node + first: AST | t.CPtr = _parse_expr(ps) + # dict: key: value + if _match_op(ps, TokOp.Colon): + _advance(ps) + val: AST | t.CPtr = _parse_expr(ps) + if _cur_kw(ps) == Keyword.For: + # dict comprehension + gens: list[AST | t.CPtr] | t.CPtr = _parse_comp_for(ps) + node = DictComp(ps.pool, first, val, gens) + _set_pos(node, lineno, col, 0, 0) + _expect_op(ps, TokOp.RBrace) + return node + keys: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + values: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + keys.append(first) + values.append(val) + while _accept_op(ps, TokOp.Comma): + _skip_nl(ps) + if _match_op(ps, TokOp.RBrace): + break + k: AST | t.CPtr = _parse_expr(ps) + _expect_op(ps, TokOp.Colon) + v: AST | t.CPtr = _parse_expr(ps) + keys.append(k) + values.append(v) + _expect_op(ps, TokOp.RBrace) + node = Dict(ps.pool, keys, values) + _set_pos(node, lineno, col, 0, 0) + return node + # set + if _cur_kw(ps) == Keyword.For: + node = _parse_comprehension(ps, first, ASTKind.SetComp, lineno, col) + _expect_op(ps, TokOp.RBrace) + return node + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + while _accept_op(ps, TokOp.Comma): + if _match_op(ps, TokOp.RBrace): + break + e: AST | t.CPtr = _parse_expr(ps) + elts.append(e) + _expect_op(ps, TokOp.RBrace) + node = Set(ps.pool, elts) + _set_pos(node, lineno, col, 0, 0) + return node + + +# ============================================================ +# f-string 解析 +# ============================================================ + +def _parse_sub_expr(ps: Parser | t.CPtr, src: str) -> AST | t.CPtr: + """创建子 lexer + parser 解析表达式（用于 f-string 内的 {expr}）""" + pool: memhub.MemManager | t.CPtr = ps.pool + lx: Lexer | t.CPtr = new_lexer(pool) + if lx is None: return None + _lexer_init(lx, src, pool) + tokens: Token | t.CPtr = tokenize(lx) + if tokens is None: return None + sub_ps: Parser | t.CPtr = new_parser(pool) + if sub_ps is None: return None + _parser_init(sub_ps, tokens, pool) + return _parse_test(sub_ps) + + +def _parse_fstring(ps: Parser | t.CPtr, content: str, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + """解析 f-string 内容，构建 JoinedStr 或 FormattedValue。""" + pool: memhub.MemManager | t.CPtr = ps.pool + values: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](pool, 8) + i: t.CSizeT = 0 + content_len: t.CSizeT = string.strlen(content) + # 字面量文本缓冲区 + lit_buf: str = pool.alloc(content_len + 1) + if lit_buf is None: return None + lit_pos: t.CSizeT = 0 + + while i < content_len: + ch: t.CChar = content[i] + if ch == '{': + # {{ → 字面量 { + if i + 1 < content_len and content[i + 1] == '{': + lit_buf[lit_pos] = '{' + lit_pos += 1 + i += 2 + continue + # 刷新字面量缓冲区 + if lit_pos > 0: + lit_buf[lit_pos] = '\0' + lit_node: AST | t.CPtr = _new_const_str(pool, lit_buf, lineno, col) + values.append(lit_node) + lit_buf = pool.alloc(content_len + 1) + if lit_buf is None: break + lit_pos = 0 + # 查找匹配的 }，同时跟踪嵌套 {} 和字符串字面量 + j: t.CSizeT = i + 1 + depth: t.CInt = 1 + conv_pos: t.CSizeT = 0 + fmt_pos: t.CSizeT = 0 + while j < content_len and depth > 0: + c: t.CChar = content[j] + if c == '{': + depth += 1 + elif c == '}': + depth -= 1 + if depth == 0: + break + elif depth == 1: + if c == '!' and conv_pos == 0 and fmt_pos == 0: + if j + 1 < content_len: + nc: t.CChar = content[j + 1] + if nc == 's' or nc == 'r' or nc == 'a': + if j + 2 < content_len: + after: t.CChar = content[j + 2] + if after == ':' or after == '}': + conv_pos = j + elif j + 2 == content_len: + conv_pos = j + elif c == ':' and fmt_pos == 0: + fmt_pos = j + j += 1 + if depth != 0: + ps.error_count += 1 + break + # 提取表达式文本 + expr_end: t.CSizeT = j + if conv_pos > 0: + expr_end = conv_pos + elif fmt_pos > 0: + expr_end = fmt_pos + expr_len: t.CSizeT = expr_end - i - 1 + if expr_len == 0: + ps.error_count += 1 + i = j + 1 + continue + expr_buf: str = pool.alloc(expr_len + 1) + if expr_buf is None: break + string.memcpy(expr_buf, content + i + 1, expr_len) + expr_buf[expr_len] = '\0' + # 解析表达式 + expr_node: AST | t.CPtr = _parse_sub_expr(ps, expr_buf) + # 转换符 + conversion: t.CInt = -1 + if conv_pos > 0: + conv_char: t.CChar = content[conv_pos + 1] + if conv_char == 's': + conversion = 115 + elif conv_char == 'r': + conversion = 114 + elif conv_char == 'a': + conversion = 97 + # 格式说明符 + format_spec: AST | t.CPtr = None + if fmt_pos > 0: + spec_len: t.CSizeT = j - fmt_pos - 1 + if spec_len > 0: + spec_buf: str = pool.alloc(spec_len + 1) + if spec_buf is not None: + string.memcpy(spec_buf, content + fmt_pos + 1, spec_len) + spec_buf[spec_len] = '\0' + format_spec = _new_const_str(pool, spec_buf, lineno, col) + fv: AST | t.CPtr = FormattedValue(pool, expr_node, conversion, format_spec) + _set_pos(fv, lineno, col, 0, 0) + values.append(fv) + i = j + 1 + elif ch == '}': + if i + 1 < content_len and content[i + 1] == '}': + lit_buf[lit_pos] = '}' + lit_pos += 1 + i += 2 + else: + ps.error_count += 1 + i += 1 + else: + lit_buf[lit_pos] = ch + lit_pos += 1 + i += 1 + + # 刷新剩余字面量 + if lit_pos > 0: + lit_buf[lit_pos] = '\0' + lit_node2: AST | t.CPtr = _new_const_str(pool, lit_buf, lineno, col) + values.append(lit_node2) + + # 如果没有内容，返回空字符串 + if values.__len__() == 0: + return _new_const_str(pool, "", lineno, col) + # 如果只有一个 FormattedValue 且无字面量，返回 FormattedValue + if values.__len__() == 1: + first: AST | t.CPtr = values.get(0) + if first.kind() == ASTKind.FormattedValue: + return first + node: AST | t.CPtr = JoinedStr(pool, values) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_comprehension(ps: Parser | t.CPtr, elt: AST | t.CPtr, + vtype: t.CInt, lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + """通用推导式：elt comp_for""" + gens: list[AST | t.CPtr] | t.CPtr = _parse_comp_for(ps) + node: AST | t.CPtr = None + if vtype == ASTKind.ListComp: + node = ListComp(ps.pool, elt, gens) + elif vtype == ASTKind.SetComp: + node = SetComp(ps.pool, elt, gens) + else: + node = GeneratorExp(ps.pool, elt, gens) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_comp_for(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: + """comp_for: ['async'] 'for' exprlist 'in' or_test [comp_if] (comp_for)* + 返回 generators list。""" + gens: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while True: + is_async: t.CInt = 0 + if _cur_kw(ps) == Keyword.Async: + is_async = 1 + _advance(ps) + if _cur_kw(ps) != Keyword.For: + break + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + target: AST | t.CPtr = _parse_target_list(ps) + if _cur_kw(ps) != Keyword.In: + ps.error_count += 1 + break + _advance(ps) + iter_: AST | t.CPtr = _parse_or_test(ps) + # 收集 ifs 到 list + ifs: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while _cur_kw(ps) == Keyword.If: + _advance(ps) + cond: AST | t.CPtr = _parse_or_test(ps) + ifs.append(cond) + comp: AST | t.CPtr = Comprehension(ps.pool, target, iter_, ifs, is_async) + _set_pos(comp, lineno, col, 0, 0) + gens.append(comp) + return gens + + +def _parse_target_list(ps: Parser | t.CPtr) -> AST | t.CPtr: + """解析赋值目标列表（用逗号分隔）""" + first: AST | t.CPtr = _parse_target(ps) + if _match_op(ps, TokOp.Comma): + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + while _accept_op(ps, TokOp.Comma): + if _cur_kw(ps) == Keyword.In: + break + e: AST | t.CPtr = _parse_target(ps) + elts.append(e) + node: AST | t.CPtr = Tuple(ps.pool, elts, ASTCtx.Store) + _set_pos(node, first.lineno, first.col_offset, 0, 0) + return node + return first + + +def _parse_target(ps: Parser | t.CPtr) -> AST | t.CPtr: + """解析单个赋值目标""" + if _match_op(ps, TokOp.Star): + _advance(ps) + inner: AST | t.CPtr = _parse_atom_expr(ps) + node: AST | t.CPtr = Starred(ps.pool, inner, ASTCtx.Store) + _set_pos(node, inner.lineno, inner.col_offset, 0, 0) + return node + if _match_op(ps, TokOp.LPar): + _advance(ps) + _skip_nl(ps) + t_: AST | t.CPtr = _parse_target_list(ps) + _skip_nl(ps) + _expect_op(ps, TokOp.RPar) + return t_ + if _match_op(ps, TokOp.Lsqb): + _advance(ps) + _skip_nl(ps) + first: AST | t.CPtr = _parse_target(ps) + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + while _accept_op(ps, TokOp.Comma): + _skip_nl(ps) + if _match_op(ps, TokOp.Rsqb): + break + e: AST | t.CPtr = _parse_target(ps) + elts.append(e) + _expect_op(ps, TokOp.Rsqb) + node = List(ps.pool, elts, ASTCtx.Store) + _set_pos(node, first.lineno, first.col_offset, 0, 0) + return node + # 普通 atom_expr，但 ctx=Store + node = _parse_atom_expr(ps) + if node is not None: + _set_ctx(node, ASTCtx.Store) + return node + + +def _parse_yield(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # yield + if _cur_kw(ps) == Keyword.From: + _advance(ps) + val: AST | t.CPtr = _parse_expr(ps) + node: AST | t.CPtr = YieldFrom(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + val2: AST | t.CPtr = None + if not _match_op(ps, TokOp.RPar) and _cur_type(ps) != TokenType.NewLine and _cur_type(ps) != TokenType.EndMarker: + if _cur_type(ps) == TokenType.Op and (_cur_op(ps) == TokOp.Comma or _cur_op(ps) == TokOp.Rsqb): + pass + else: + val2 = _parse_test(ps) + node = Yield(ps.pool, val2) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_varargslist(ps: Parser | t.CPtr, is_lambda: t.CInt = 0) -> AST | t.CPtr: + """函数参数列表解析（lambda 和 def 共用） + 新 API：args/defaults/kw_defaults 为 list，vararg/kwarg 为字段。""" + args: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + vararg: AST | t.CPtr = None + kwarg: AST | t.CPtr = None + defaults: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + kw_defaults: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + + while True: + _skip_nl(ps) + if _match_op(ps, TokOp.RPar): + break + # *args 或 * (kwonly 分隔) + if _match_op(ps, TokOp.Star): + _advance(ps) + _skip_nl(ps) + if _cur_type(ps) == TokenType.Name: + vararg = _parse_arg_def(ps, is_lambda) + if not _accept_op(ps, TokOp.Comma): + break + _skip_nl(ps) + continue + # **kwargs + if _match_op(ps, TokOp.StarStar): + _advance(ps) + _skip_nl(ps) + kwarg = _parse_arg_def(ps, is_lambda) + if not _accept_op(ps, TokOp.Comma): + break + _skip_nl(ps) + continue + # / 分隔符（posonly 结束）— 新 API 中 posonly/regular 不区分 + if _match_op(ps, TokOp.Slash): + _advance(ps) + if not _accept_op(ps, TokOp.Comma): + break + _skip_nl(ps) + continue + # 普通参数 + a: AST | t.CPtr = _parse_arg_def(ps, is_lambda) + if a is None: break + args.append(a) + if _match_op(ps, TokOp.Equal): + _advance(ps) + d: AST | t.CPtr = _parse_test(ps) + defaults.append(d) + if not _accept_op(ps, TokOp.Comma): + break + _skip_nl(ps) + return Arguments(ps.pool, args, vararg, kwarg, defaults, kw_defaults) + + +def _parse_arg_def(ps: Parser | t.CPtr, is_lambda: t.CInt = 0) -> AST | t.CPtr: + """arg: TokenType.Name [':' annotation]""" + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + return None + name_str: str = _cur_str(ps) + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + annotation: AST | t.CPtr = None + if is_lambda == 0: + if _accept_op(ps, TokOp.Colon): + annotation = _parse_test(ps) + result: AST | t.CPtr = Arg(ps.pool, name_str, annotation) + _set_pos(result, lineno, col, 0, 0) + return result + + +# ============================================================ +# 语句解析（设计：直接 append 到 parent，不返回 list） +# ============================================================ + +def _parse_suite_into_children(ps: Parser | t.CPtr, parent: AST | t.CPtr): + """解析语句块，直接 append 为 parent 的子节点。 + suite: TokenType.NewLine TokenType.Indent stmt+ TokenType.Dedent | simple_stmt""" + _skip_nl(ps) + if _cur_type(ps) == TokenType.NewLine: + _advance(ps) + _skip_nl(ps) + if _cur_type(ps) != TokenType.Indent: + ps.error_count += 1 + return + _advance(ps) + sidx: t.CInt = 0 + while _cur_type(ps) != TokenType.Dedent and _cur_type(ps) != TokenType.EndMarker: + sidx += 1 + # 装饰器（类体内方法也可以有装饰器） + if _match_op(ps, TokOp.At): + ps.pending_decorators = _parse_decorators(ps) + before_cur: Token | t.CPtr = ps.cur + _parse_statement_into(ps, parent) + # 防御：若解析未前进 token，强制前进以避免死循环 + if ps.cur is before_cur: + ps.error_count += 1 + _advance(ps) + _skip_nl(ps) + _expect_type(ps, TokenType.Dedent) + return + # 单行 suite + _parse_simple_stmt_into(ps, parent) + + +def _parse_statement_into(ps: Parser | t.CPtr, parent: AST | t.CPtr): + """解析一条语句（复合或简单），append 到 parent。""" + ttype: t.CInt = _cur_type(ps) + kw: t.CInt = _cur_kw(ps) + if ttype == TokenType.Name: + if kw == Keyword.If or kw == Keyword.While or kw == Keyword.For or kw == Keyword.Try: + stmt: AST | t.CPtr = _parse_compound(ps) + if stmt is not None: + parent.append(stmt) + return + if kw == Keyword.With or kw == Keyword.Def or kw == Keyword.Class: + stmt = _parse_compound(ps) + if stmt is not None: + parent.append(stmt) + return + if kw == Keyword.Match or kw == Keyword.Async: + stmt = _parse_compound(ps) + if stmt is not None: + parent.append(stmt) + return + _parse_simple_stmt_into(ps, parent) + + +def _parse_simple_stmt_into(ps: Parser | t.CPtr, parent: AST | t.CPtr): + """simple_stmt: small_stmt (';' small_stmt)* [';'] TokenType.NewLine""" + first: AST | t.CPtr = _parse_small_stmt(ps) + if first is not None: + parent.append(first) + while _accept_op(ps, TokOp.Semi): + if _cur_type(ps) == TokenType.NewLine or _cur_type(ps) == TokenType.EndMarker: + break + s: AST | t.CPtr = _parse_small_stmt(ps) + if s is None: break + parent.append(s) + _skip_nl(ps) + if _cur_type(ps) == TokenType.NewLine: + _advance(ps) + + +def _parse_small_stmt(ps: Parser | t.CPtr) -> AST | t.CPtr: + """small_stmt: 各种简单语句""" + kw: t.CInt = _cur_kw(ps) + if kw == Keyword.Pass: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + node: AST | t.CPtr = Pass(ps.pool) + _set_pos(node, lineno, col, 0, 0) + return node + if kw == Keyword.Break: + lineno = _cur_lineno(ps) + col = _cur_col(ps) + _advance(ps) + node = Break(ps.pool) + _set_pos(node, lineno, col, 0, 0) + return node + if kw == Keyword.Continue: + lineno = _cur_lineno(ps) + col = _cur_col(ps) + _advance(ps) + node = Continue(ps.pool) + _set_pos(node, lineno, col, 0, 0) + return node + if kw == Keyword.Return: + return _parse_return(ps) + if kw == Keyword.Raise: + return _parse_raise(ps) + if kw == Keyword.Global: + return _parse_global(ps) + if kw == Keyword.Nonlocal: + return _parse_nonlocal(ps) + if kw == Keyword.Import: + return _parse_import(ps) + if kw == Keyword.From: + return _parse_from_import(ps) + if kw == Keyword.Del: + return _parse_del(ps) + if kw == Keyword.Assert: + return _parse_assert(ps) + if kw == Keyword.Yield: + return _parse_expr_stmt(ps, _parse_yield(ps)) + # 表达式语句 / 赋值 / augassign / annassign + return _parse_expr_stmt(ps, None) + + +def _maybe_tuple(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: + """如果 first 后面有逗号，收集逗号分隔的表达式为 Tuple。""" + if first is None: return None + if not _match_op(ps, TokOp.Comma): + return first + lineno: t.CInt = first.lineno + col: t.CInt = first.col_offset + elts: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elts.append(first) + _accept_op(ps, TokOp.Comma) + while True: + tt: t.CInt = _cur_type(ps) + if tt == TokenType.NewLine or tt == TokenType.EndMarker: + break + if tt == TokenType.Op: + op: t.CInt = _cur_op(ps) + if op == TokOp.Equal or op == TokOp.Semi: + break + e: AST | t.CPtr = _parse_expr(ps) + if e is None: break + elts.append(e) + if not _accept_op(ps, TokOp.Comma): + break + node: AST | t.CPtr = Tuple(ps.pool, elts, ASTCtx.Load) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_testlist(ps: Parser | t.CPtr) -> AST | t.CPtr: + """解析逗号分隔的表达式列表，多于1个则形成 Tuple。""" + first: AST | t.CPtr = _parse_expr(ps) + return _maybe_tuple(ps, first) + + +def _parse_expr_stmt(ps: Parser | t.CPtr, first: AST | t.CPtr) -> AST | t.CPtr: + """表达式语句，处理赋值/augassign/annassign""" + if first is None: + first = _parse_testlist(ps) + else: + first = _maybe_tuple(ps, first) + if first is None: + return None + lineno: t.CInt = first.lineno + col: t.CInt = first.col_offset + # 赋值 + if _match_op(ps, TokOp.Equal): + targets: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + targets.append(first) + _fix_store_ctx(first) + value: AST | t.CPtr = None + accept_result: t.CInt = _accept_op(ps, TokOp.Equal) + while accept_result: + v: AST | t.CPtr = _parse_testlist(ps) + if v is None: break + if value is not None: + targets.append(value) + _fix_store_ctx(value) + value = v + accept_result = _accept_op(ps, TokOp.Equal) + node: AST | t.CPtr = Assign(ps.pool, targets, value) + _set_pos(node, lineno, col, 0, 0) + return node + # augassign + if _cur_type(ps) == TokenType.Op: + op_id: t.CInt = _cur_op(ps) + aug_op: t.CInt = _augop_from_op(op_id) + if aug_op != OpKind.NoneOp: + _advance(ps) + value2: AST | t.CPtr = _parse_testlist(ps) + _fix_store_ctx(first) + node2: AST | t.CPtr = AugAssign(ps.pool, first, aug_op, value2) + _set_pos(node2, lineno, col, 0, 0) + return node2 + # annassign: target ':' annotation [= value] + if _match_op(ps, TokOp.Colon): + _advance(ps) + annotation: AST | t.CPtr = _parse_test(ps) + value3: AST | t.CPtr = None + if _accept_op(ps, TokOp.Equal): + value3 = _parse_testlist(ps) + if ps.pool is None: + return None + _fix_store_ctx(first) + if ps.pool is None: + return None + node3: AST | t.CPtr = AnnAssign(ps.pool, first, annotation, value3, 1) + _set_pos(node3, lineno, col, 0, 0) + return node3 + # 表达式语句 + node4: AST | t.CPtr = Expr(ps.pool, first) + _set_pos(node4, lineno, col, 0, 0) + return node4 + + +def _parse_return(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + val: AST | t.CPtr = None + if _cur_type(ps) != TokenType.NewLine and _cur_type(ps) != TokenType.EndMarker and _cur_type(ps) != TokenType.Op: + val = _parse_testlist(ps) + elif _cur_type(ps) == TokenType.Op and _cur_op(ps) != TokOp.Semi: + val = _parse_testlist(ps) + node: AST | t.CPtr = Return(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_raise(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + exc: AST | t.CPtr = None + cause: AST | t.CPtr = None + if _cur_type(ps) != TokenType.NewLine and _cur_type(ps) != TokenType.EndMarker: + exc = _parse_test(ps) + if _cur_kw(ps) == Keyword.From: + _advance(ps) + cause = _parse_test(ps) + node: AST | t.CPtr = Raise(ps.pool, exc, cause) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_global(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + names: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + if _cur_type(ps) == TokenType.Name: + n: AST | t.CPtr = Name(ps.pool, _cur_str(ps), ASTCtx.Store) + _set_pos(n, _cur_lineno(ps), _cur_col(ps), 0, 0) + names.append(n) + _advance(ps) + while _accept_op(ps, TokOp.Comma): + if _cur_type(ps) != TokenType.Name: break + n = Name(ps.pool, _cur_str(ps), ASTCtx.Store) + _set_pos(n, _cur_lineno(ps), _cur_col(ps), 0, 0) + names.append(n) + _advance(ps) + node: AST | t.CPtr = Global(ps.pool, names) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_nonlocal(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + names: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + if _cur_type(ps) == TokenType.Name: + n: AST | t.CPtr = Name(ps.pool, _cur_str(ps), ASTCtx.Store) + _set_pos(n, _cur_lineno(ps), _cur_col(ps), 0, 0) + names.append(n) + _advance(ps) + while _accept_op(ps, TokOp.Comma): + if _cur_type(ps) != TokenType.Name: break + n = Name(ps.pool, _cur_str(ps), ASTCtx.Store) + _set_pos(n, _cur_lineno(ps), _cur_col(ps), 0, 0) + names.append(n) + _advance(ps) + node: AST | t.CPtr = Nonlocal(ps.pool, names) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_import(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + names: list[AST | t.CPtr] | t.CPtr = _parse_alias_list(ps) + node: AST | t.CPtr = Import(ps.pool, names) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_from_import(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + # 解析 module（可能带点号） + level: t.CInt = 0 + while _match_op(ps, TokOp.Dot): + level += 1 + _advance(ps) + while _match_op(ps, TokOp.Ellipsis): + level += 3 + _advance(ps) + module_name: str = None + if _cur_type(ps) == TokenType.Name: + module_name = _cur_str(ps) + _advance(ps) + while _match_op(ps, TokOp.Dot): + _advance(ps) + if _cur_type(ps) == TokenType.Name: + next_name: str = _cur_str(ps) + mlen: t.CInt = string.strlen(module_name) + nlen: t.CInt = string.strlen(next_name) + new_mod: str = ps.pool.alloc(mlen + 1 + nlen + 1) + if new_mod is not None: + string.strcpy(new_mod, module_name) + new_mod[mlen] = '.' + string.strcpy(new_mod + mlen + 1, next_name) + module_name = new_mod + _advance(ps) + if _cur_kw(ps) != Keyword.Import: + ps.error_count += 1 + return None + _advance(ps) + # import * + if _match_op(ps, TokOp.Star): + _advance(ps) + star_alias: AST | t.CPtr = Alias(ps.pool, "*", None) + _set_pos(star_alias, lineno, col, 0, 0) + star_names: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + star_names.append(star_alias) + node: AST | t.CPtr = ImportFrom(ps.pool, module_name, star_names, level) + _set_pos(node, lineno, col, 0, 0) + return node + # ( names ) or names + if _match_op(ps, TokOp.LPar): + _advance(ps) + _skip_nl(ps) + names: list[AST | t.CPtr] | t.CPtr = _parse_alias_list(ps) + _skip_nl(ps) + _expect_op(ps, TokOp.RPar) + else: + names = _parse_alias_list(ps) + node: AST | t.CPtr = ImportFrom(ps.pool, module_name, names, level) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_alias_list(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: + """返回 alias list。""" + names: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while True: + if _cur_type(ps) != TokenType.Name: + break + name_str: str = _cur_str(ps) + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + # dotted name + while _match_op(ps, TokOp.Dot): + _advance(ps) + if _cur_type(ps) == TokenType.Name: + next_n: str = _cur_str(ps) + nmlen: t.CInt = string.strlen(name_str) + nxtlen: t.CInt = string.strlen(next_n) + new_name: str = ps.pool.alloc(nmlen + 1 + nxtlen + 1) + if new_name is not None: + string.strcpy(new_name, name_str) + new_name[nmlen] = '.' + string.strcpy(new_name + nmlen + 1, next_n) + name_str = new_name + _advance(ps) + asname: str = None + if _cur_kw(ps) == Keyword.As: + _advance(ps) + if _cur_type(ps) == TokenType.Name: + asname = _cur_str(ps) + _advance(ps) + a: AST | t.CPtr = Alias(ps.pool, name_str, asname) + _set_pos(a, lineno, col, 0, 0) + names.append(a) + if not _accept_op(ps, TokOp.Comma): + break + return names + + +def _parse_del(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + targets: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while True: + t_: AST | t.CPtr = _parse_atom_expr(ps) + if t_ is None: break + _set_ctx(t_, ASTCtx.Del) + targets.append(t_) + if not _accept_op(ps, TokOp.Comma): + break + node: AST | t.CPtr = Delete(ps.pool, targets) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_assert(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + test: AST | t.CPtr = _parse_test(ps) + msg: AST | t.CPtr = None + if _accept_op(ps, TokOp.Comma): + msg = _parse_test(ps) + node: AST | t.CPtr = Assert(ps.pool, test, msg) + _set_pos(node, lineno, col, 0, 0) + return node + + +# ============================================================ +# 复合语句 +# ============================================================ + +def _parse_compound(ps: Parser | t.CPtr) -> AST | t.CPtr: + kw: t.CInt = _cur_kw(ps) + if kw == Keyword.If: + return _parse_if(ps) + if kw == Keyword.While: + return _parse_while(ps) + if kw == Keyword.For: + return _parse_for(ps, 0) + if kw == Keyword.Try: + return _parse_try(ps) + if kw == Keyword.With: + return _parse_with(ps, 0) + if kw == Keyword.Def: + return _parse_funcdef(ps, 0) + if kw == Keyword.Class: + return _parse_classdef(ps) + if kw == Keyword.Async: + _advance(ps) + if _cur_kw(ps) == Keyword.Def: + return _parse_funcdef(ps, 1) + if _cur_kw(ps) == Keyword.For: + return _parse_for(ps, 1) + if _cur_kw(ps) == Keyword.With: + return _parse_with(ps, 1) + ps.error_count += 1 + return None + if kw == Keyword.Match: + return _parse_match(ps) + ps.error_count += 1 + return None + + +def _parse_if(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # if + test: AST | t.CPtr = _parse_test(ps) + _expect_op(ps, TokOp.Colon) + # 先用空 orelse 构造，后面填充 + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = If(ps.pool, test, empty) + _parse_suite_into_children(ps, node) + orelse: list[AST | t.CPtr] | t.CPtr = _parse_if_tail(ps) + node.orelse = orelse + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_if_tail(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: + """Parse elif/else chain and return orelse list. + + elif: orelse = list containing single If node (nested) + else: orelse = statement list + """ + if _cur_kw(ps) == Keyword.Elif: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) + elif_test: AST | t.CPtr = _parse_test(ps) + _expect_op(ps, TokOp.Colon) + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + elif_node: AST | t.CPtr = If(ps.pool, elif_test, empty) + _parse_suite_into_children(ps, elif_node) + elif_orelse: list[AST | t.CPtr] | t.CPtr = _parse_if_tail(ps) + elif_node.orelse = elif_orelse + _set_pos(elif_node, lineno, col, 0, 0) + orelse: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + orelse.append(elif_node) + return orelse + if _cur_kw(ps) == Keyword.Else: + _advance(ps) + _expect_op(ps, TokOp.Colon) + scratch: Module | t.CPtr = Module(ps.pool) + _parse_suite_into_children(ps, scratch) + return scratch.children + return list[AST | t.CPtr](ps.pool, 8) # 空 list + + +# ============================================================ +# 复合语句解析（续） +# ============================================================ + +def _parse_while(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # while + test: AST | t.CPtr = _parse_test(ps) + _expect_op(ps, TokOp.Colon) + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = While(ps.pool, test, empty) + _parse_suite_into_children(ps, node) + if _cur_kw(ps) == Keyword.Else: + _advance(ps) + _expect_op(ps, TokOp.Colon) + scratch: Module | t.CPtr = Module(ps.pool) + _parse_suite_into_children(ps, scratch) + node.orelse = scratch.children + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_for(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # for + target: AST | t.CPtr = _parse_target_list(ps) + if _cur_kw(ps) != Keyword.In: + ps.error_count += 1 + return None + _advance(ps) + iter_: AST | t.CPtr = _parse_or_test(ps) + _expect_op(ps, TokOp.Colon) + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = For(ps.pool, target, iter_, empty, is_async) + _fix_store_ctx(target) + _parse_suite_into_children(ps, node) + if _cur_kw(ps) == Keyword.Else: + _advance(ps) + _expect_op(ps, TokOp.Colon) + scratch: Module | t.CPtr = Module(ps.pool) + _parse_suite_into_children(ps, scratch) + node.orelse = scratch.children + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_try(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # try + _expect_op(ps, TokOp.Colon) + empty: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = Try(ps.pool, empty, empty, empty) + _parse_suite_into_children(ps, node) + # handlers (except) + handlers: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while _cur_kw(ps) == Keyword.Except: + h: AST | t.CPtr = _parse_except_handler(ps) + if h is not None: + handlers.append(h) + node.handlers = handlers + # else + if _cur_kw(ps) == Keyword.Else: + _advance(ps) + _expect_op(ps, TokOp.Colon) + scratch: Module | t.CPtr = Module(ps.pool) + _parse_suite_into_children(ps, scratch) + node.orelse = scratch.children + # finally + if _cur_kw(ps) == Keyword.Finally: + _advance(ps) + _expect_op(ps, TokOp.Colon) + scratch2: Module | t.CPtr = Module(ps.pool) + _parse_suite_into_children(ps, scratch2) + node.finalbody = scratch2.children + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_except_handler(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # except + type_: AST | t.CPtr = None + name: str = None + if _cur_op(ps) != TokOp.Colon: + type_ = _parse_test(ps) + if _cur_kw(ps) == Keyword.As: + _advance(ps) + if _cur_type(ps) == TokenType.Name: + name = _cur_str(ps) + _advance(ps) + _expect_op(ps, TokOp.Colon) + node: AST | t.CPtr = ExceptHandler(ps.pool, type_, name) + _parse_suite_into_children(ps, node) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_with(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # with + items: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + # items + while True: + item: AST | t.CPtr = _parse_with_item(ps) + if item is None: break + items.append(item) + if not _accept_op(ps, TokOp.Comma): + break + # 检查是否可以续行（with (...): 形式） + if _cur_type(ps) == TokenType.NewLine or _cur_type(ps) == TokenType.Nl: + break + _expect_op(ps, TokOp.Colon) + node: AST | t.CPtr = With(ps.pool, items, is_async) + _parse_suite_into_children(ps, node) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_with_item(ps: Parser | t.CPtr) -> AST | t.CPtr: + """with_item: test ['as' expr]""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + context_expr: AST | t.CPtr = _parse_or_test(ps) + if context_expr is None: return None + optional_vars: AST | t.CPtr = None + if _cur_kw(ps) == Keyword.As: + _advance(ps) + optional_vars = _parse_target(ps) + _fix_store_ctx(optional_vars) + node: AST | t.CPtr = WithItem(ps.pool, context_expr, optional_vars) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_funcdef(ps: Parser | t.CPtr, is_async: t.CInt) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # def + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + return None + name: str = _cur_str(ps) + _advance(ps) + args: AST | t.CPtr = None + if _accept_op(ps, TokOp.LPar): + args = _parse_varargslist(ps, 0) + _expect_op(ps, TokOp.RPar) + returns: AST | t.CPtr = None + if _accept_op(ps, TokOp.RArrow): + returns = _parse_test(ps) + _expect_op(ps, TokOp.Colon) + dec_list: list[AST | t.CPtr] | t.CPtr = ps.pending_decorators + ps.pending_decorators = None + node: AST | t.CPtr = FunctionDef(ps.pool, name, args, dec_list, returns, is_async) + _parse_suite_into_children(ps, node) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_classdef(ps: Parser | t.CPtr) -> AST | t.CPtr: + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # class + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + return None + name: str = _cur_str(ps) + _advance(ps) + # 解析类型参数 [T] 或 [T, U]（PEP 695 泛型类语法） + type_params: list[str] | t.CPtr = None + if _accept_op(ps, TokOp.Lsqb): + type_params = list[str](ps.pool, 8) + while not _match_op(ps, TokOp.Rsqb): + if _cur_type(ps) == TokenType.EndMarker: break + if _cur_type(ps) == TokenType.Name: + tp_name: str = _cur_str(ps) + _advance(ps) + # 跳过可选的 bound/constraint: [T: bound] 或 [T: (A, B)] + if _match_op(ps, TokOp.Colon): + _advance(ps) + _parse_test(ps) + type_params.append(tp_name) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, TokOp.Rsqb) + bases: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + keywords: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + # bases / keywords + if _accept_op(ps, TokOp.LPar): + while not _match_op(ps, TokOp.RPar): + if _cur_type(ps) == TokenType.EndMarker: break + # base 或 keyword=value + if _cur_type(ps) == TokenType.Name: + # 解析为表达式，若后跟 = 则作为 keyword + e: AST | t.CPtr = _parse_or_test(ps) + if _match_op(ps, TokOp.Equal): + _advance(ps) + v: AST | t.CPtr = _parse_test(ps) + kw_node: AST | t.CPtr = KwNode(ps.pool, e, v) + keywords.append(kw_node) + else: + bases.append(e) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, TokOp.RPar) + _expect_op(ps, TokOp.Colon) + dec_list: list[AST | t.CPtr] | t.CPtr = ps.pending_decorators + ps.pending_decorators = None + node: AST | t.CPtr = ClassDef(ps.pool, name, bases, keywords, dec_list, type_params) + _parse_suite_into_children(ps, node) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_match(ps: Parser | t.CPtr) -> AST | t.CPtr: + """match 语句：match subject ':' NewLine Indent case+ Dedent""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # match + subject: AST | t.CPtr = _parse_test(ps) + _expect_op(ps, TokOp.Colon) + cases: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + node: AST | t.CPtr = Match(ps.pool, subject, cases) + # 解析 body：与 _parse_suite_into_children 相同的 NewLine/Indent/Dedent 模式， + # 但解析 case 子句到 cases 列表而非普通语句到 children + _skip_nl(ps) + if _cur_type(ps) == TokenType.NewLine: + _advance(ps) + _skip_nl(ps) + if _cur_type(ps) == TokenType.Indent: + _advance(ps) + while _cur_type(ps) != TokenType.Dedent and _cur_type(ps) != TokenType.EndMarker: + if _cur_kw(ps) == Keyword.Case: + case_node: AST | t.CPtr = _parse_match_case(ps) + if case_node is not None: + cases.append(case_node) + else: + _advance(ps) # 跳过非 case token，避免死循环 + _expect_type(ps, TokenType.Dedent) + _set_pos(node, lineno, col, 0, 0) + return node + + +# ============================================================ +# Match pattern 解析 +# ============================================================ + +def _make_name_ast(pool: memhub.MemManager | t.CPtr, node: Name | t.CPtr) -> AST | t.CPtr: + """将 Name* 转换为 AST*（解决 TransPyC 类型推断问题）""" + return node + + +def _parse_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: + """pattern: or_pattern ['as' NAME] + or_pattern: closed_pattern ('|' closed_pattern)*""" + pat: AST | t.CPtr = _parse_closed_pattern(ps) + if pat is None: return None + # OR pattern + if _match_op(ps, TokOp.VBar): + patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + patterns.append(pat) + while _accept_op(ps, TokOp.VBar): + p: AST | t.CPtr = _parse_closed_pattern(ps) + if p is not None: + patterns.append(p) + or_node: AST | t.CPtr = MatchOr(ps.pool, patterns) + _set_pos(or_node, pat.lineno, pat.col_offset, 0, 0) + pat = or_node + # AS pattern + if _cur_kw(ps) == Keyword.As: + _advance(ps) + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + return pat + name_str: str = _cur_str(ps) + _advance(ps) + as_node: AST | t.CPtr = MatchAs(ps.pool, pat, name_str) + _set_pos(as_node, pat.lineno, pat.col_offset, 0, 0) + return as_node + return pat + + +def _parse_closed_pattern(ps: Parser | t.CPtr) -> AST | t.CPtr: + """closed_pattern: literal | capture | wildcard | value | group | sequence | mapping | class""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + ttype: t.CInt = _cur_type(ps) + val: AST | t.CPtr + node: AST | t.CPtr + tok: Token | t.CPtr + sval: str + + # 数字字面量 pattern + if ttype == TokenType.Number: + tok = ps.cur + _advance(ps) + if tok.is_float: + val = _new_const_float(ps.pool, tok.float_val, lineno, col) + else: + val = _new_const_int(ps.pool, tok.int_val, lineno, col) + node = MatchValue(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + + # 字符串字面量 pattern + if ttype == TokenType.String: + sval = _cur_str(ps) + _advance(ps) + val = _new_const_str(ps.pool, sval, lineno, col) + node = MatchValue(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + + # 负数字面量 pattern: -1, -1.5 + if _match_op(ps, TokOp.Minus): + _advance(ps) + if _cur_type(ps) == TokenType.Number: + tok = ps.cur + _advance(ps) + if tok.is_float: + val = _new_const_float(ps.pool, -tok.float_val, lineno, col) + else: + val = _new_const_int(ps.pool, -tok.int_val, lineno, col) + node = MatchValue(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + ps.error_count += 1 + return None + + # Name: None/True/False/_ /capture/value/class + if ttype == TokenType.Name: + kw: t.CInt = _cur_kw(ps) + # None/True/False → MatchSingleton + if kw == Keyword.None_ or kw == Keyword.True_ or kw == Keyword.False_: + _advance(ps) + if kw == Keyword.None_: + val = _new_const_none(ps.pool, lineno, col) + elif kw == Keyword.True_: + val = _new_const_bool(ps.pool, 1, lineno, col) + else: + val = _new_const_bool(ps.pool, 0, lineno, col) + node = MatchSingleton(ps.pool, val) + _set_pos(node, lineno, col, 0, 0) + return node + name_str: str = _cur_str(ps) + # 通配符 _ + if string.strcmp(name_str, "_") == 0: + _advance(ps) + node = MatchAs(ps.pool, None, None) + _set_pos(node, lineno, col, 0, 0) + return node + _advance(ps) + # 检查 dotted name（value pattern 或 class pattern） + if _match_op(ps, TokOp.Dot): + name_node: Name | t.CPtr = Name(ps.pool, name_str, ASTCtx.Load) + _set_pos(name_node, lineno, col, 0, 0) + base: AST | t.CPtr = _make_name_ast(ps.pool, name_node) + while _match_op(ps, TokOp.Dot): + _advance(ps) + if _cur_type(ps) != TokenType.Name: + ps.error_count += 1 + break + attr_name: str = _cur_str(ps) + attr_lineno: t.CInt = _cur_lineno(ps) + attr_col: t.CInt = _cur_col(ps) + _advance(ps) + base = Attribute(ps.pool, base, attr_name, ASTCtx.Load) + _set_pos(base, attr_lineno, attr_col, 0, 0) + # class pattern: dotted.name(...) + if _match_op(ps, TokOp.LPar): + return _parse_class_pattern(ps, base, lineno, col) + # value pattern: dotted.name + node = MatchValue(ps.pool, base) + _set_pos(node, lineno, col, 0, 0) + return node + # class pattern: Name(...) + if _match_op(ps, TokOp.LPar): + cls_node: Name | t.CPtr = Name(ps.pool, name_str, ASTCtx.Load) + _set_pos(cls_node, lineno, col, 0, 0) + cls: AST | t.CPtr = _make_name_ast(ps.pool, cls_node) + return _parse_class_pattern(ps, cls, lineno, col) + # capture pattern: NAME + node = MatchAs(ps.pool, None, name_str) + _set_pos(node, lineno, col, 0, 0) + return node + + # group pattern / sequence pattern (tuple): (...) + if _match_op(ps, TokOp.LPar): + _advance(ps) + inner: AST | t.CPtr = _parse_pattern(ps) + if _accept_op(ps, TokOp.Comma): + # sequence pattern (tuple) + patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + if inner is not None: + patterns.append(inner) + while not _match_op(ps, TokOp.RPar) and _cur_type(ps) != TokenType.EndMarker: + p: AST | t.CPtr = _parse_pattern(ps) + if p is not None: + patterns.append(p) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, TokOp.RPar) + node = MatchSequence(ps.pool, patterns) + _set_pos(node, lineno, col, 0, 0) + return node + _expect_op(ps, TokOp.RPar) + return inner # group pattern + + # sequence pattern (list): [...] + if _match_op(ps, TokOp.Lsqb): + return _parse_sequence_pattern(ps, TokOp.Rsqb, lineno, col) + + # mapping pattern: {...} + if _match_op(ps, TokOp.LBrace): + return _parse_mapping_pattern(ps, lineno, col) + + ps.error_count += 1 + return None + + +def _parse_sequence_pattern(ps: Parser | t.CPtr, close_op: t.CInt, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + """sequence_pattern: '[' [pattern_list] ']'""" + patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + _advance(ps) # [ + while not _match_op(ps, close_op) and _cur_type(ps) != TokenType.EndMarker: + # *name (star pattern) + if _match_op(ps, TokOp.Star): + _advance(ps) + if _cur_type(ps) == TokenType.Name: + star_name: str = _cur_str(ps) + _advance(ps) + star_node: AST | t.CPtr + if string.strcmp(star_name, "_") == 0: + star_node = MatchStar(ps.pool, None) + else: + star_node = MatchStar(ps.pool, star_name) + _set_pos(star_node, lineno, col, 0, 0) + patterns.append(star_node) + else: + p: AST | t.CPtr = _parse_pattern(ps) + if p is not None: + patterns.append(p) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, close_op) + node: AST | t.CPtr = MatchSequence(ps.pool, patterns) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_mapping_pattern(ps: Parser | t.CPtr, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + """mapping_pattern: '{' [items] '}' + items: key ':' pattern (',' key ':' pattern)* [',' '**' NAME] | '**' NAME""" + keys: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + rest: str = None + node: AST | t.CPtr + key: AST | t.CPtr + val_pat: AST | t.CPtr + _advance(ps) # { + if _match_op(ps, TokOp.RBrace): + _advance(ps) + node = MatchMapping(ps.pool, keys, patterns, rest) + _set_pos(node, lineno, col, 0, 0) + return node + while _cur_type(ps) != TokenType.EndMarker: + if _match_op(ps, TokOp.StarStar): + _advance(ps) + if _cur_type(ps) == TokenType.Name: + rest = _cur_str(ps) + _advance(ps) + break + key = _parse_expr(ps) + _expect_op(ps, TokOp.Colon) + val_pat = _parse_pattern(ps) + if key is not None: + keys.append(key) + if val_pat is not None: + patterns.append(val_pat) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, TokOp.RBrace) + node = MatchMapping(ps.pool, keys, patterns, rest) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_class_pattern(ps: Parser | t.CPtr, cls: AST | t.CPtr, + lineno: t.CInt, col: t.CInt) -> AST | t.CPtr: + """class_pattern: cls '(' [pattern_list] ')' + pattern_list: positional_pattern (',' positional_pattern)* + [',' keyword_pattern]* | keyword_pattern (',' keyword_pattern)* + keyword_pattern: NAME '=' pattern""" + patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + kwd_attrs: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + kwd_patterns: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + p: AST | t.CPtr + _advance(ps) # ( + if not _match_op(ps, TokOp.RPar): + while _cur_type(ps) != TokenType.EndMarker: + # keyword pattern: NAME '=' pattern + if _cur_type(ps) == TokenType.Name and ps.cur is not None: + nxt: Token | t.CPtr = ps.cur.next + if nxt is not None and nxt.type == TokenType.Op and nxt.op_subtype == TokOp.Equal: + attr_name: str = _cur_str(ps) + _advance(ps) # NAME + _advance(ps) # = + kwd_pat: AST | t.CPtr = _parse_pattern(ps) + attr_n: Name | t.CPtr = Name(ps.pool, attr_name, ASTCtx.Load) + _set_pos(attr_n, lineno, col, 0, 0) + attr_node: AST | t.CPtr = _make_name_ast(ps.pool, attr_n) + kwd_attrs.append(attr_node) + if kwd_pat is not None: + kwd_patterns.append(kwd_pat) + else: + p = _parse_pattern(ps) + if p is not None: + patterns.append(p) + else: + p = _parse_pattern(ps) + if p is not None: + patterns.append(p) + if not _accept_op(ps, TokOp.Comma): + break + _expect_op(ps, TokOp.RPar) + node: AST | t.CPtr = MatchClass(ps.pool, cls, patterns, kwd_attrs, kwd_patterns) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_match_case(ps: Parser | t.CPtr) -> AST | t.CPtr: + """case pattern [if guard] ':' suite""" + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # case + guard: AST | t.CPtr = None + pattern: AST | t.CPtr = _parse_pattern(ps) + if _cur_kw(ps) == Keyword.If: + _advance(ps) + guard = _parse_or_test(ps) + _expect_op(ps, TokOp.Colon) + node: AST | t.CPtr = MatchCase(ps.pool, pattern, guard) + _parse_suite_into_children(ps, node) + _set_pos(node, lineno, col, 0, 0) + return node + + +def _parse_decorators(ps: Parser | t.CPtr) -> list[AST | t.CPtr] | t.CPtr: + """解析装饰器链，返回 list。""" + decs: list[AST | t.CPtr] | t.CPtr = list[AST | t.CPtr](ps.pool, 8) + while _match_op(ps, TokOp.At): + lineno: t.CInt = _cur_lineno(ps) + col: t.CInt = _cur_col(ps) + _advance(ps) # @ + expr: AST | t.CPtr = _parse_namedexpr(ps) + _skip_nl(ps) + if _cur_type(ps) == TokenType.NewLine: + _advance(ps) + decs.append(expr) + return decs + + +# ============================================================ +# 模块入口 +# ============================================================ + +def _parse_module_body(ps: Parser | t.CPtr, module_node: AST | t.CPtr): + """解析模块体，append 到 module_node。""" + _skip_nl(ps) + stmt_idx: t.CInt = 0 + while _cur_type(ps) != TokenType.EndMarker: + stmt_idx += 1 + # 装饰器 + if _match_op(ps, TokOp.At): + ps.pending_decorators = _parse_decorators(ps) + # 记录解析前的 token 指针，用于检测未前进的死循环 + before_cur: Token | t.CPtr = ps.cur + _parse_statement_into(ps, module_node) + # 防御：若解析未前进 token（如 _parse_atom 遇到无法识别的 token 返回 None）， + # 强制前进以避免主循环死循环；累计 error_count 供调用方诊断 + if ps.cur is before_cur: + ps.error_count += 1 + _advance(ps) + _skip_nl(ps) + + +def parse_tokens(pool: memhub.MemManager | t.CPtr, + tokens: Token | t.CPtr) -> AST | t.CPtr: + """从 token 链表解析 AST Module。""" + if pool is None or tokens is None: + return None + ps: Parser | t.CPtr = new_parser(pool) + if ps is None: return None + _parser_init(ps, tokens, pool) + module_node: AST | t.CPtr = Module(pool) + _parse_module_body(ps, module_node) + return module_node diff --git a/includes/ast/stmts.py b/includes/ast/stmts.py new file mode 100644 index 0000000..471001e --- /dev/null +++ b/includes/ast/stmts.py @@ -0,0 +1,1150 @@ +import t, c +from stdint import * +import memhub +import string +from .base import ( + AST, ASTKind, ASTCtx, OpKind, + _init_ast, _copy_str, _set_parent_list, + _emit, _emit_str, _emit_int, _dump_list, + FLAG_IS_ASYNC, +) + + +# ============================================================ +# Module 节点 +# ============================================================ + +@t.CVTable +class Module(AST): + """Module(children: list[AST | t.CPtr])""" + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Module | t.CPtr = pool.alloc(Module.__sizeof__()) + if ptr: + string.memset(ptr, 0, Module.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr): + _init_ast(self, pool, 0, 0) + + def kind(self) -> t.CInt: + return ASTKind.Module + + def type_name(self) -> str: + return "Module" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Module(body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Assign 节点 +# ============================================================ + +class Assign(AST): + """Assign(targets: list[AST | t.CPtr], value: AST)""" + targets: list[AST | t.CPtr] | t.CPtr + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr): + ptr: Assign | t.CPtr = pool.alloc(Assign.__sizeof__()) + if ptr: + string.memset(ptr, 0, Assign.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + targets: list[AST | t.CPtr] | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.targets = targets + self.value = value + _set_parent_list(targets, self) + + def kind(self) -> t.CInt: + return ASTKind.Assign + + def type_name(self) -> str: + return "Assign" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Assign(targets=") + pos = _dump_list(self.targets, buf, size, pos) + pos = _emit(buf, size, pos, ", value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# If 节点 +# ============================================================ + +class If(AST): + """If(test: AST, orelse: list[AST | t.CPtr])""" + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr): + ptr: If | t.CPtr = pool.alloc(If.__sizeof__()) + if ptr: + string.memset(ptr, 0, If.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.test = test + self.orelse = orelse + _set_parent_list(orelse, self) + + def kind(self) -> t.CInt: + return ASTKind.If + + def type_name(self) -> str: + return "If" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "If(test=") + if self.test is not None: + pos = self.test.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ", orelse=") + pos = _dump_list(self.orelse, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# For 节点 +# ============================================================ + +class For(AST): + """For(target: AST, iter: AST, orelse: list[AST | t.CPtr], is_async: CInt)""" + target: AST | t.CPtr + iter: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, iter: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr, + is_async: t.CInt): + ptr: For | t.CPtr = pool.alloc(For.__sizeof__()) + if ptr: + string.memset(ptr, 0, For.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, iter: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr, + is_async: t.CInt): + _init_ast(self, pool, 0, 0) + self.target = target + self.iter = iter + self.orelse = orelse + if is_async: + self.flags = FLAG_IS_ASYNC + else: + self.flags = 0 + _set_parent_list(orelse, self) + + def kind(self) -> t.CInt: + return ASTKind.For + + def type_name(self) -> str: + return "For" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "For(target=") + if self.target is not None: + pos = self.target.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", iter=") + if self.iter is not None: + pos = self.iter.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# While 节点 +# ============================================================ + +class While(AST): + """While(test: AST, orelse: list[AST | t.CPtr])""" + test: AST | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr): + ptr: While | t.CPtr = pool.alloc(While.__sizeof__()) + if ptr: + string.memset(ptr, 0, While.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.test = test + self.orelse = orelse + _set_parent_list(orelse, self) + + def kind(self) -> t.CInt: + return ASTKind.While + + def type_name(self) -> str: + return "While" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "While(test=") + if self.test is not None: + pos = self.test.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Return 节点 +# ============================================================ + +class Return(AST): + """Return(value: AST | None)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: Return | t.CPtr = pool.alloc(Return.__sizeof__()) + if ptr: + string.memset(ptr, 0, Return.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.Return + + def type_name(self) -> str: + return "Return" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Return(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Expr 节点（表达式语句） +# ============================================================ + +class Expr(AST): + """Expr(value: AST)""" + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + ptr: Expr | t.CPtr = pool.alloc(Expr.__sizeof__()) + if ptr: + string.memset(ptr, 0, Expr.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.Expr + + def type_name(self) -> str: + return "Expr" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Expr(value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# FunctionDef 节点 +# ============================================================ + +class FunctionDef(AST): + """FunctionDef(name: str, args: AST, + decorator_list: list[AST | t.CPtr], returns: AST, flags: CInt)""" + name: str + args: AST | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + flags: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + name: str, args: AST | t.CPtr, + decorator_list: list[AST | t.CPtr] | t.CPtr, + returns: AST | t.CPtr, is_async: t.CInt): + ptr: FunctionDef | t.CPtr = pool.alloc(FunctionDef.__sizeof__()) + if ptr: + string.memset(ptr, 0, FunctionDef.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + name: str, args: AST | t.CPtr, + decorator_list: list[AST | t.CPtr] | t.CPtr, + returns: AST | t.CPtr, is_async: t.CInt): + _init_ast(self, pool, 0, 0) + self.name = _copy_str(pool, name) + self.args = args + self.decorator_list = decorator_list + self.returns = returns + if is_async: + self.flags = FLAG_IS_ASYNC + else: + self.flags = 0 + _set_parent_list(decorator_list, self) + + def kind(self) -> t.CInt: + return ASTKind.FunctionDef + + def type_name(self) -> str: + return "FunctionDef" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "FunctionDef(name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "', args=") + if self.args is not None: + pos = self.args.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ", decorator_list=") + pos = _dump_list(self.decorator_list, buf, size, pos) + pos = _emit(buf, size, pos, ", returns=") + if self.returns is not None: + pos = self.returns.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Expression 节点 +# ============================================================ + +class Expression(AST): + """Expression(body: AST)""" + body: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + body: AST | t.CPtr): + ptr: Expression | t.CPtr = pool.alloc(Expression.__sizeof__()) + if ptr: + string.memset(ptr, 0, Expression.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + body: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.body = body + + def kind(self) -> t.CInt: + return ASTKind.Expression + + def type_name(self) -> str: + return "Expression" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Expression(body=") + if self.body is not None: + pos = self.body.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Interactive 节点 +# ============================================================ + +class Interactive(AST): + """Interactive(body: AST)""" + body: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + body: AST | t.CPtr): + ptr: Interactive | t.CPtr = pool.alloc(Interactive.__sizeof__()) + if ptr: + string.memset(ptr, 0, Interactive.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + body: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.body = body + + def kind(self) -> t.CInt: + return ASTKind.Interactive + + def type_name(self) -> str: + return "Interactive" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Interactive(body=") + if self.body is not None: + pos = self.body.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# FunctionType 节点 +# ============================================================ + +class FunctionType(AST): + """FunctionType(argtypes: list[AST | t.CPtr], returns: AST)""" + argtypes: list[AST | t.CPtr] | t.CPtr + returns: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + argtypes: list[AST | t.CPtr] | t.CPtr, + returns: AST | t.CPtr): + ptr: FunctionType | t.CPtr = pool.alloc(FunctionType.__sizeof__()) + if ptr: + string.memset(ptr, 0, FunctionType.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + argtypes: list[AST | t.CPtr] | t.CPtr, + returns: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.argtypes = argtypes + self.returns = returns + + def kind(self) -> t.CInt: + return ASTKind.FunctionType + + def type_name(self) -> str: + return "FunctionType" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "FunctionType(argtypes=") + pos = _dump_list(self.argtypes, buf, size, pos) + pos = _emit(buf, size, pos, ", returns=") + if self.returns is not None: + pos = self.returns.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# ClassDef 节点 +# ============================================================ + +class ClassDef(AST): + """ClassDef(name: str, bases: list[AST | t.CPtr], keywords: list[AST | t.CPtr], + decorator_list: list[AST | t.CPtr], type_params: list[str] | t.CPtr) + + type_params: 泛型类型参数名列表（PEP 695 class list[T]: 中的 [T]）， + None 表示非泛型类。""" + name: str + bases: list[AST | t.CPtr] | t.CPtr + keywords: list[AST | t.CPtr] | t.CPtr + decorator_list: list[AST | t.CPtr] | t.CPtr + type_params: list[str] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + name: str, + bases: list[AST | t.CPtr] | t.CPtr, + keywords: list[AST | t.CPtr] | t.CPtr, + decorator_list: list[AST | t.CPtr] | t.CPtr, + type_params: list[str] | t.CPtr = None): + ptr: ClassDef | t.CPtr = pool.alloc(ClassDef.__sizeof__()) + if ptr: + string.memset(ptr, 0, ClassDef.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + name: str, + bases: list[AST | t.CPtr] | t.CPtr, + keywords: list[AST | t.CPtr] | t.CPtr, + decorator_list: list[AST | t.CPtr] | t.CPtr, + type_params: list[str] | t.CPtr = None): + _init_ast(self, pool, 0, 0) + self.name = _copy_str(pool, name) + self.bases = bases + self.keywords = keywords + self.decorator_list = decorator_list + self.type_params = type_params + _set_parent_list(decorator_list, self) + + def kind(self) -> t.CInt: + return ASTKind.ClassDef + + def type_name(self) -> str: + return "ClassDef" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "ClassDef(name='") + pos = _emit_str(buf, size, pos, self.name) + pos = _emit(buf, size, pos, "', bases=") + pos = _dump_list(self.bases, buf, size, pos) + pos = _emit(buf, size, pos, ", keywords=") + pos = _dump_list(self.keywords, buf, size, pos) + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ", decorator_list=") + pos = _dump_list(self.decorator_list, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Delete 节点 +# ============================================================ + +class Delete(AST): + """Delete(targets: list[AST | t.CPtr])""" + targets: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + targets: list[AST | t.CPtr] | t.CPtr): + ptr: Delete | t.CPtr = pool.alloc(Delete.__sizeof__()) + if ptr: + string.memset(ptr, 0, Delete.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + targets: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.targets = targets + _set_parent_list(targets, self) + + def kind(self) -> t.CInt: + return ASTKind.Delete + + def type_name(self) -> str: + return "Delete" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Delete(targets=") + pos = _dump_list(self.targets, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# AugAssign 节点 +# ============================================================ + +class AugAssign(AST): + """AugAssign(target: AST, op: CInt, value: AST)""" + target: AST | t.CPtr + op: t.CInt + value: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr): + ptr: AugAssign | t.CPtr = pool.alloc(AugAssign.__sizeof__()) + if ptr: + string.memset(ptr, 0, AugAssign.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, op: t.CInt, value: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.target = target + self.op = op + self.value = value + + def kind(self) -> t.CInt: + return ASTKind.AugAssign + + def type_name(self) -> str: + return "AugAssign" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "AugAssign(target=") + if self.target is not None: + pos = self.target.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", op=") + pos = _emit(buf, size, pos, _op_name(self.op)) + pos = _emit(buf, size, pos, ", value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# AnnAssign 节点 +# ============================================================ + +class AnnAssign(AST): + """AnnAssign(target: AST, annotation: AST, value: AST, simple: CInt)""" + target: AST | t.CPtr + annotation: AST | t.CPtr + value: AST | t.CPtr + simple: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, annotation: AST | t.CPtr, + value: AST | t.CPtr, simple: t.CInt): + ptr: AnnAssign | t.CPtr = pool.alloc(AnnAssign.__sizeof__()) + if ptr: + string.memset(ptr, 0, AnnAssign.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + target: AST | t.CPtr, annotation: AST | t.CPtr, + value: AST | t.CPtr, simple: t.CInt): + _init_ast(self, pool, 0, 0) + self.target = target + self.annotation = annotation + self.value = value + self.simple = simple + + def kind(self) -> t.CInt: + return ASTKind.AnnAssign + + def type_name(self) -> str: + return "AnnAssign" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "AnnAssign(target=") + if self.target is not None: + pos = self.target.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", annotation=") + if self.annotation is not None: + pos = self.annotation.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", value=") + if self.value is not None: + pos = self.value.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", simple=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.simple)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# With 节点 +# ============================================================ + +class With(AST): + """With(items: list[AST | t.CPtr], is_async: CInt)""" + items: list[AST | t.CPtr] | t.CPtr + flags: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + items: list[AST | t.CPtr] | t.CPtr, + is_async: t.CInt): + ptr: With | t.CPtr = pool.alloc(With.__sizeof__()) + if ptr: + string.memset(ptr, 0, With.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + items: list[AST | t.CPtr] | t.CPtr, + is_async: t.CInt): + _init_ast(self, pool, 0, 0) + self.items = items + if is_async: + self.flags = FLAG_IS_ASYNC + else: + self.flags = 0 + _set_parent_list(items, self) + + def kind(self) -> t.CInt: + return ASTKind.With + + def type_name(self) -> str: + return "With" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "With(items=") + pos = _dump_list(self.items, buf, size, pos) + pos = _emit(buf, size, pos, ", body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Raise 节点 +# ============================================================ + +class Raise(AST): + """Raise(exc: AST, cause: AST)""" + exc: AST | t.CPtr + cause: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + exc: AST | t.CPtr, cause: AST | t.CPtr): + ptr: Raise | t.CPtr = pool.alloc(Raise.__sizeof__()) + if ptr: + string.memset(ptr, 0, Raise.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + exc: AST | t.CPtr, cause: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.exc = exc + self.cause = cause + + def kind(self) -> t.CInt: + return ASTKind.Raise + + def type_name(self) -> str: + return "Raise" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Raise(exc=") + if self.exc is not None: + pos = self.exc.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", cause=") + if self.cause is not None: + pos = self.cause.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Try 节点 +# ============================================================ + +class Try(AST): + """Try(handlers: list[AST | t.CPtr], + orelse: list[AST | t.CPtr], finalbody: list[AST | t.CPtr])""" + handlers: list[AST | t.CPtr] | t.CPtr + orelse: list[AST | t.CPtr] | t.CPtr + finalbody: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + handlers: list[AST | t.CPtr] | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr, + finalbody: list[AST | t.CPtr] | t.CPtr): + ptr: Try | t.CPtr = pool.alloc(Try.__sizeof__()) + if ptr: + string.memset(ptr, 0, Try.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + handlers: list[AST | t.CPtr] | t.CPtr, + orelse: list[AST | t.CPtr] | t.CPtr, + finalbody: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.handlers = handlers + self.orelse = orelse + self.finalbody = finalbody + _set_parent_list(handlers, self) + _set_parent_list(orelse, self) + _set_parent_list(finalbody, self) + + def kind(self) -> t.CInt: + return ASTKind.Try + + def type_name(self) -> str: + return "Try" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Try(body=") + pos = _dump_list(self.children, buf, size, pos) + pos = _emit(buf, size, pos, ", handlers=") + pos = _dump_list(self.handlers, buf, size, pos) + pos = _emit(buf, size, pos, ", orelse=") + pos = _dump_list(self.orelse, buf, size, pos) + pos = _emit(buf, size, pos, ", finalbody=") + pos = _dump_list(self.finalbody, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Assert 节点 +# ============================================================ + +class Assert(AST): + """Assert(test: AST, msg: AST)""" + test: AST | t.CPtr + msg: AST | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, msg: AST | t.CPtr): + ptr: Assert | t.CPtr = pool.alloc(Assert.__sizeof__()) + if ptr: + string.memset(ptr, 0, Assert.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + test: AST | t.CPtr, msg: AST | t.CPtr): + _init_ast(self, pool, 0, 0) + self.test = test + self.msg = msg + + def kind(self) -> t.CInt: + return ASTKind.Assert + + def type_name(self) -> str: + return "Assert" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Assert(test=") + if self.test is not None: + pos = self.test.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", msg=") + if self.msg is not None: + pos = self.msg.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Global 节点 +# ============================================================ + +class Global(AST): + """Global(names: list[AST | t.CPtr])""" + names: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + ptr: Global | t.CPtr = pool.alloc(Global.__sizeof__()) + if ptr: + string.memset(ptr, 0, Global.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.names = names + _set_parent_list(names, self) + + def kind(self) -> t.CInt: + return ASTKind.Global + + def type_name(self) -> str: + return "Global" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Global(names=") + pos = _dump_list(self.names, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Nonlocal 节点 +# ============================================================ + +class Nonlocal(AST): + """Nonlocal(names: list[AST | t.CPtr])""" + names: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + ptr: Nonlocal | t.CPtr = pool.alloc(Nonlocal.__sizeof__()) + if ptr: + string.memset(ptr, 0, Nonlocal.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.names = names + _set_parent_list(names, self) + + def kind(self) -> t.CInt: + return ASTKind.Nonlocal + + def type_name(self) -> str: + return "Nonlocal" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Nonlocal(names=") + pos = _dump_list(self.names, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Pass 节点 +# ============================================================ + +class Pass(AST): + """Pass()""" + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Pass | t.CPtr = pool.alloc(Pass.__sizeof__()) + if ptr: + string.memset(ptr, 0, Pass.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr): + _init_ast(self, pool, 0, 0) + + def kind(self) -> t.CInt: + return ASTKind.Pass + + def type_name(self) -> str: + return "Pass" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Pass()") + return pos + + +# ============================================================ +# Break 节点 +# ============================================================ + +class Break(AST): + """Break()""" + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Break | t.CPtr = pool.alloc(Break.__sizeof__()) + if ptr: + string.memset(ptr, 0, Break.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr): + _init_ast(self, pool, 0, 0) + + def kind(self) -> t.CInt: + return ASTKind.Break + + def type_name(self) -> str: + return "Break" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Break()") + return pos + + +# ============================================================ +# Continue 节点 +# ============================================================ + +class Continue(AST): + """Continue()""" + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: Continue | t.CPtr = pool.alloc(Continue.__sizeof__()) + if ptr: + string.memset(ptr, 0, Continue.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr): + _init_ast(self, pool, 0, 0) + + def kind(self) -> t.CInt: + return ASTKind.Continue + + def type_name(self) -> str: + return "Continue" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Continue()") + return pos + + +# ============================================================ +# Import 节点 +# ============================================================ + +class Import(AST): + """Import(names: list[AST | t.CPtr])""" + names: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + ptr: Import | t.CPtr = pool.alloc(Import.__sizeof__()) + if ptr: + string.memset(ptr, 0, Import.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + names: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.names = names + _set_parent_list(names, self) + + def kind(self) -> t.CInt: + return ASTKind.Import + + def type_name(self) -> str: + return "Import" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Import(names=") + pos = _dump_list(self.names, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# ImportFrom 节点 +# ============================================================ + +class ImportFrom(AST): + """ImportFrom(module: str, names: list[AST | t.CPtr], level: CInt)""" + module: str + names: list[AST | t.CPtr] | t.CPtr + level: t.CInt + + def __new__(self, pool: memhub.MemManager | t.CPtr, + module: str, + names: list[AST | t.CPtr] | t.CPtr, + level: t.CInt): + ptr: ImportFrom | t.CPtr = pool.alloc(ImportFrom.__sizeof__()) + if ptr: + string.memset(ptr, 0, ImportFrom.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + module: str, + names: list[AST | t.CPtr] | t.CPtr, + level: t.CInt): + _init_ast(self, pool, 0, 0) + self.module = _copy_str(pool, module) + self.names = names + self.level = level + _set_parent_list(names, self) + + def kind(self) -> t.CInt: + return ASTKind.ImportFrom + + def type_name(self) -> str: + return "ImportFrom" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "ImportFrom(module='") + pos = _emit_str(buf, size, pos, self.module) + pos = _emit(buf, size, pos, "', names=") + pos = _dump_list(self.names, buf, size, pos) + pos = _emit(buf, size, pos, ", level=") + pos = _emit_int(buf, size, pos, t.CInt64T(self.level)) + pos = _emit(buf, size, pos, ")") + return pos + + +# ============================================================ +# Match 节点 +# ============================================================ + +class Match(AST): + """Match(subject: AST, cases: list[AST | t.CPtr])""" + subject: AST | t.CPtr + cases: list[AST | t.CPtr] | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, + subject: AST | t.CPtr, + cases: list[AST | t.CPtr] | t.CPtr): + ptr: Match | t.CPtr = pool.alloc(Match.__sizeof__()) + if ptr: + string.memset(ptr, 0, Match.__sizeof__()) + return ptr + + def __init__(self, pool: memhub.MemManager | t.CPtr, + subject: AST | t.CPtr, + cases: list[AST | t.CPtr] | t.CPtr): + _init_ast(self, pool, 0, 0) + self.subject = subject + self.cases = cases + _set_parent_list(cases, self) + + def kind(self) -> t.CInt: + return ASTKind.Match + + def type_name(self) -> str: + return "Match" + + def dump(self, buf: t.CChar | t.CPtr, size: t.CSizeT, + pos: t.CSizeT) -> t.CSizeT: + pos = _emit(buf, size, pos, "Match(subject=") + if self.subject is not None: + pos = self.subject.dump(buf, size, pos) + else: + pos = _emit(buf, size, pos, "None") + pos = _emit(buf, size, pos, ", cases=") + pos = _dump_list(self.cases, buf, size, pos) + pos = _emit(buf, size, pos, ")") + return pos diff --git a/includes/ast/tokens.py b/includes/ast/tokens.py new file mode 100644 index 0000000..d3a4ea0 --- /dev/null +++ b/includes/ast/tokens.py @@ -0,0 +1,391 @@ +import t, c +from stdint import * +import memhub +import string + + +# ============================================================ +# Token 类型枚举（参考 CPython Lib/token.py） +# 用 t.CEnum 组织为类型安全枚举；旧名 XXX 保留为兼容别名 +# ============================================================ + +class TokenType(t.CEnum): + EndMarker: t.State + NewLine: t.State + Indent: t.State + Dedent: t.State + Name: t.State + Number: t.State + String: t.State + FStringStart: t.State + FStringMiddle: t.State + FStringEnd: t.State + Op: t.State + Nl: t.State + Comment: t.State + Encoding: t.State + + +# ============================================================ +# 关键字枚举（参考 CPython Lib/keyword.py） +# 用 t.CEnum 组织为类型安全枚举；旧名 KW_XXX 保留为兼容别名 +# NotKeyword(0) 表示非关键字（哨兵，避免 False_=0 与 kw_subtype=0 冲突） +# ============================================================ + +class Keyword(t.CEnum): + NotKeyword: t.State + False_: t.State + None_: t.State + True_: t.State + And: t.State + As: t.State + Assert: t.State + Async: t.State + Await: t.State + Break: t.State + Class: t.State + Continue: t.State + Def: t.State + Del: t.State + Elif: t.State + Else: t.State + Except: t.State + Finally: t.State + For: t.State + From: t.State + Global: t.State + If: t.State + Import: t.State + In: t.State + Is: t.State + Lambda: t.State + Nonlocal: t.State + Not: t.State + Or: t.State + Pass: t.State + Raise: t.State + Return: t.State + Try: t.State + While: t.State + With: t.State + Yield: t.State + Match: t.State + Case: t.State + Type: t.State + + +# ============================================================ +# 运算符/分隔符枚举 +# 用 t.CEnum 组织为类型安全枚举；旧名 OP_XXX 保留为兼容别名 +# 存储在 token.op_subtype 字段 +# ============================================================ + +class TokOp(t.CEnum): + LPar: t.State # ( + RPar: t.State # ) + Lsqb: t.State # [ + Rsqb: t.State # ] + LBrace: t.State # { + RBrace: t.State # } + Comma: t.State # , + Colon: t.State # : + Dot: t.State # . + Semi: t.State # ; + At: t.State # @ + Equal: t.State # = + RArrow: t.State # -> + PlusEq: t.State # += + MinusEq: t.State # -= + StarEq: t.State # *= + SlashEq: t.State # /= + DSlashEq: t.State # //= + PercentEq: t.State # %= + AtEq: t.State # @= + AmpEq: t.State # &= + VBarEq: t.State # |= + CaretEq: t.State # ^= + GtGtEq: t.State # >>= + LtLtEq: t.State # <<= + StarEqEq: t.State # **= + DSlash: t.State # // + StarStar: t.State # ** + LtLt: t.State # << + GtGt: t.State # >> + LessEq: t.State # <= + GreaterEq: t.State # >= + EqEq: t.State # == + ExclaimEq: t.State # != + Less: t.State # < + Greater: t.State # > + Plus: t.State # + + Minus: t.State # - + Star: t.State # * + Slash: t.State # / + Percent: t.State # % + Amp: t.State # & + VBar: t.State # | + Caret: t.State # ^ + Tilde: t.State # ~ + ColonEq: t.State # := + Ellipsis: t.State # ... + Bang: t.State # ! (非标准，用于解析容错) + + +# ============================================================ +# Token 结构体 +# ============================================================ +class Token: + type: t.CInt # Token 类型（TokenType.EndMarker/TokenType.NewLine/TokenType.Name/...） + op_subtype: t.CInt # 运算符子类型（OP_*），仅 type==TokenType.Op 时有效 + kw_subtype: t.CInt # 关键字子类型（KW_*），仅 type==TokenType.Name 且是关键字时有效 + str_val: str # token 文本（NUL 结尾，从 pool 分配） + int_val: t.CInt64T # 整数值（TokenType.Number 用） + float_val: t.CDouble # 浮点值（TokenType.Number 用） + is_float: t.CInt # 1=浮点数, 0=整数 + is_complex: t.CInt # 1=复数, 0=实数 + lineno: t.CInt # 起始行号 + col_offset: t.CInt # 起始列偏移 + end_lineno: t.CInt # 结束行号 + end_col_offset: t.CInt # 结束列偏移 + next: Token | t.CPtr # 链表下一个 token + + def __new__(self, pool: memhub.MemManager | t.CPtr): + """从 mpool 分配 Token 结构体内存并零初始化""" + ptr: Token | t.CPtr = pool.alloc(Token.__sizeof__()) + if ptr: + string.memset(ptr, 0, Token.__sizeof__()) + return ptr + + +# ============================================================ +# 关键字查找表 +# 由于 TransPyC 无 dict 字面量，用链表实现 +# ============================================================ +class KwEntry: + name: str + kw_id: t.CInt + next: KwEntry | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: KwEntry | t.CPtr = pool.alloc(KwEntry.__sizeof__()) + if ptr: + string.memset(ptr, 0, KwEntry.__sizeof__()) + return ptr + + +# 全局关键字链表头（懒初始化） +_kw_head: KwEntry | t.CPtr = None +_kw_pool: memhub.MemManager | t.CPtr = None + + +def _kw_intern(pool: memhub.MemManager | t.CPtr, name: str, kw_id: t.CInt): + """将关键字名注册到链表""" + global _kw_head + entry: KwEntry | t.CPtr = KwEntry(pool) + if entry == None: return + slen: t.CSizeT = string.strlen(name) + buf: str = pool.alloc(slen + 1) + if buf == None: return + string.memcpy(buf, name, slen + 1) + entry.name = buf + entry.kw_id = kw_id + entry.next = _kw_head + _kw_head = entry + + +def _kw_lookup(name: str) -> t.CInt: + """查找关键字，返回 kw_id（0 表示非关键字）""" + cur: KwEntry | t.CPtr = _kw_head + while cur != None: + if string.strcmp(cur.name, name) == 0: + return cur.kw_id + cur = cur.next + return 0 + + +def _init_keywords(pool: memhub.MemManager | t.CPtr): + """初始化关键字表（仅一次）""" + global _kw_head, _kw_pool + if _kw_head != None: return + _kw_pool = pool + _kw_intern(pool, "False", Keyword.False_) + _kw_intern(pool, "None", Keyword.None_) + _kw_intern(pool, "True", Keyword.True_) + _kw_intern(pool, "and", Keyword.And) + _kw_intern(pool, "as", Keyword.As) + _kw_intern(pool, "assert", Keyword.Assert) + _kw_intern(pool, "async", Keyword.Async) + _kw_intern(pool, "await", Keyword.Await) + _kw_intern(pool, "break", Keyword.Break) + _kw_intern(pool, "class", Keyword.Class) + _kw_intern(pool, "continue", Keyword.Continue) + _kw_intern(pool, "def", Keyword.Def) + _kw_intern(pool, "del", Keyword.Del) + _kw_intern(pool, "elif", Keyword.Elif) + _kw_intern(pool, "else", Keyword.Else) + _kw_intern(pool, "except", Keyword.Except) + _kw_intern(pool, "finally", Keyword.Finally) + _kw_intern(pool, "for", Keyword.For) + _kw_intern(pool, "from", Keyword.From) + _kw_intern(pool, "global", Keyword.Global) + _kw_intern(pool, "if", Keyword.If) + _kw_intern(pool, "import", Keyword.Import) + _kw_intern(pool, "in", Keyword.In) + _kw_intern(pool, "is", Keyword.Is) + _kw_intern(pool, "lambda", Keyword.Lambda) + _kw_intern(pool, "nonlocal", Keyword.Nonlocal) + _kw_intern(pool, "not", Keyword.Not) + _kw_intern(pool, "or", Keyword.Or) + _kw_intern(pool, "pass", Keyword.Pass) + _kw_intern(pool, "raise", Keyword.Raise) + _kw_intern(pool, "return", Keyword.Return) + _kw_intern(pool, "try", Keyword.Try) + _kw_intern(pool, "while", Keyword.While) + _kw_intern(pool, "with", Keyword.With) + _kw_intern(pool, "yield", Keyword.Yield) + _kw_intern(pool, "match", Keyword.Match) + _kw_intern(pool, "case", Keyword.Case) + _kw_intern(pool, "type", Keyword.Type) + + +# ============================================================ +# 运算符查找表 +# 由于 TransPyC 无 dict，用数组+线性查找 +# 运算符按长度降序排列，以便贪婪匹配 +# ============================================================ +class OpEntry: + name: str # 运算符文本（如 "==", "+="） + op_id: t.CInt + length: t.CInt + next: OpEntry | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr): + ptr: OpEntry | t.CPtr = pool.alloc(OpEntry.__sizeof__()) + if ptr: + string.memset(ptr, 0, OpEntry.__sizeof__()) + return ptr + + +_op_head: OpEntry | t.CPtr = None + + +def _op_intern(pool: memhub.MemManager | t.CPtr, name: str, op_id: t.CInt): + global _op_head + slen: t.CSizeT = string.strlen(name) + entry: OpEntry | t.CPtr = OpEntry(pool) + if entry == None: return + buf: str = pool.alloc(slen + 1) + if buf == None: return + string.memcpy(buf, name, slen + 1) + entry.name = buf + entry.op_id = op_id + entry.length = t.CInt(slen) + entry.next = _op_head + _op_head = entry + + +def _init_operators(pool: memhub.MemManager | t.CPtr): + """初始化运算符表（按长度降序注册，便于贪婪匹配）""" + global _op_head + if _op_head != None: return + # 3 字符运算符 + _op_intern(pool, "**=", TokOp.StarEqEq) + _op_intern(pool, "//=", TokOp.DSlashEq) + _op_intern(pool, ">>=", TokOp.GtGtEq) + _op_intern(pool, "<<=", TokOp.LtLtEq) + _op_intern(pool, "...", TokOp.Ellipsis) + # 2 字符运算符 + _op_intern(pool, "->", TokOp.RArrow) + _op_intern(pool, "+=", TokOp.PlusEq) + _op_intern(pool, "-=", TokOp.MinusEq) + _op_intern(pool, "*=", TokOp.StarEq) + _op_intern(pool, "/=", TokOp.SlashEq) + _op_intern(pool, "%=", TokOp.PercentEq) + _op_intern(pool, "@=", TokOp.AtEq) + _op_intern(pool, "&=", TokOp.AmpEq) + _op_intern(pool, "|=", TokOp.VBarEq) + _op_intern(pool, "^=", TokOp.CaretEq) + _op_intern(pool, "//", TokOp.DSlash) + _op_intern(pool, "**", TokOp.StarStar) + _op_intern(pool, "<<", TokOp.LtLt) + _op_intern(pool, ">>", TokOp.GtGt) + _op_intern(pool, "<=", TokOp.LessEq) + _op_intern(pool, ">=", TokOp.GreaterEq) + _op_intern(pool, "==", TokOp.EqEq) + _op_intern(pool, "!=", TokOp.ExclaimEq) + _op_intern(pool, ":=", TokOp.ColonEq) + # 1 字符运算符 + _op_intern(pool, "(", TokOp.LPar) + _op_intern(pool, ")", TokOp.RPar) + _op_intern(pool, "[", TokOp.Lsqb) + _op_intern(pool, "]", TokOp.Rsqb) + _op_intern(pool, "{", TokOp.LBrace) + _op_intern(pool, "}", TokOp.RBrace) + _op_intern(pool, ",", TokOp.Comma) + _op_intern(pool, ":", TokOp.Colon) + _op_intern(pool, ".", TokOp.Dot) + _op_intern(pool, ";", TokOp.Semi) + _op_intern(pool, "@", TokOp.At) + _op_intern(pool, "=", TokOp.Equal) + _op_intern(pool, "<", TokOp.Less) + _op_intern(pool, ">", TokOp.Greater) + _op_intern(pool, "+", TokOp.Plus) + _op_intern(pool, "-", TokOp.Minus) + _op_intern(pool, "*", TokOp.Star) + _op_intern(pool, "/", TokOp.Slash) + _op_intern(pool, "%", TokOp.Percent) + _op_intern(pool, "&", TokOp.Amp) + _op_intern(pool, "|", TokOp.VBar) + _op_intern(pool, "^", TokOp.Caret) + _op_intern(pool, "~", TokOp.Tilde) + + +def _init_tables(pool: memhub.MemManager | t.CPtr): + """初始化所有查找表（幂等）""" + _init_keywords(pool) + _init_operators(pool) + + +# ============================================================ +# Token 创建辅助函数 +# ============================================================ + +def new_token(pool: memhub.MemManager | t.CPtr, ttype: t.CInt, lineno: t.CInt, + col_offset: t.CInt) -> Token | t.CPtr: + """创建一个空 Token，调用方填充字段""" + tok: Token | t.CPtr = Token(pool) + if tok == None: return None + tok.type = ttype + tok.lineno = lineno + tok.col_offset = col_offset + tok.end_lineno = lineno + tok.end_col_offset = col_offset + return tok + + +def token_set_str(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, + src: str, start: t.CSizeT, length: t.CSizeT): + """从 src[start:start+length] 复制文本到 token.str_val + + 注意：length == 0 时也分配 1 字节存 NUL，让空字符串与 NULL 指针区分开。 + """ + if tok == None: return + buf: str = pool.alloc(length + 1) + if buf == None: return + i: t.CSizeT = 0 + while i < length: + buf[i] = src[start + i] + i += 1 + buf[length] = '\0' + tok.str_val = buf + + +def token_set_str_literal(pool: memhub.MemManager | t.CPtr, tok: Token | t.CPtr, + text: str): + """直接复制一个 C 字符串到 token.str_val""" + if tok == None or text == None: return + slen: t.CSizeT = string.strlen(text) + buf: str = pool.alloc(slen + 1) + if buf == None: return + string.memcpy(buf, text, slen + 1) + tok.str_val = buf diff --git a/includes/atom.py b/includes/atom.py new file mode 100644 index 0000000..c01664f --- /dev/null +++ b/includes/atom.py @@ -0,0 +1,42 @@ +import t, c + + +ATOMIC_RELAXED: t.CDefine = 0 +ATOMIC_CONSUME: t.CDefine = 1 +ATOMIC_ACQUIRE: t.CDefine = 2 +ATOMIC_RELEASE: t.CDefine = 3 +ATOMIC_ACQ_REL: t.CDefine = 4 +ATOMIC_SEQ_CST: t.CDefine = 5 + + +def __atomic_test_and_set(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CBool: + result: t.CUInt8T = 1 + c.Asm(f"""mov al, 1 + xchg byte ptr [{c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)}], al + mov {c.AsmOut(result, t.ASM_DESCR.OUTPUT_REG)}, al""", + out = [c.AsmOut(result, t.ASM_DESCR.OUTPUT_REG)], + inp = [c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)], + op = [t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX]) + return t.CBool(result) + + +def __atomic_clear(ptr: t.CUInt64T | t.CPtr, order: t.CInt) -> t.CVoid: + c.Asm(f"""mov byte ptr [{c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)}], 0""", + inp = [c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)], + op = [t.ASM_DESCR.CLOBBER_MEMORY]) + + +def __atomic_thread_fence(order: t.CInt) -> t.CVoid: + c.Asm("mfence", op=[t.ASM_DESCR.CLOBBER_MEMORY]) + + +def __atomic_signal_fence(order: t.CInt) -> t.CVoid: + pass + + +def __atomic_always_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: + return t.CBool(1) + + +def __atomic_is_lock_free(size: t.CSizeT, ptr: t.CVoid | t.CPtr) -> t.CBool: + return t.CBool(1) diff --git a/includes/base64.py b/includes/base64.py new file mode 100644 index 0000000..bf18d0b --- /dev/null +++ b/includes/base64.py @@ -0,0 +1,78 @@ +import t, c + + +# ============================================== +# Base64 编解码 纯裸实现 +# ============================================== +# Base64 编码对照表 +b64_tab: t.CArray[t.CChar, None] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" + +# Base64 反向解码表，映射字符到索引值 +b64_dec_tab: t.CArray[t.CInt8T, 80] = [ + 62, -1, -1, -1, 63, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, + -1, -1, -1, -1, -1, -1, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, + 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, + -1, -1, -1, -1, -1, -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, + 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51 +] + +# * @brief Base64 编码 +# * @param in: 输入二进制数据 +# * @param in_len: 输入长度 +# * @param out: 输出Base64字符串缓冲区 +# * @retval 编码后字符串长度 +def b64encode(s: str, out: str) -> t.CSizeT: + _s = t.CUInt8T(s, t.CPtr) + length: t.CSizeT = len(s) + i: t.CSizeT = 0 + j: t.CSizeT = 0 + val: t.CUInt32T = 0 + # 每3字节一组转4字节Base64 + for i in range(0, length, 3): + # 拼接24bit分组 + val = ((t.CUInt32T(_s[i + 0]) << 16 ) | + (t.CUInt32T(_s[i + 1]) << 8 if (i + 1 < length) else 0) | + (t.CUInt32T(_s[i + 2]) << 0 if (i + 2 < length) else 0)) + # 拆分4个6bit 查表 + out[j + 0] = b64_tab[(val >> 18) & 0x3F] + out[j + 1] = b64_tab[(val >> 12) & 0x3F] + out[j + 2] = b64_tab[(val >> 6) & 0x3F] if (i + 1 < length) else '=' + out[j + 3] = b64_tab[(val >> 0) & 0x3F] if (i + 2 < length) else '=' + j += 4 + out[j] = '\0' # 字符串结尾 + return j + + +# * @brief Base64 解码 +# * @param in: Base64字符串 +# * @param out: 输出二进制缓冲区 +# * @retval 解码后二进制长度 +def b64decode(s: str, out: t.CUInt8T | t.CPtr) -> t.CSizeT: + length: t.CSizeT = len(s) + i: t.CSizeT = 0 + j: t.CSizeT = 0 + val: t.CUInt32T = 0 + c: t.CInt + # 每4个Base64字符还原3字节 + for i in range(0, length, 4): + if s[i] == '=': break + val = 0 + # 逐个解析6bit + c = s[i] - 43 + val |= (b64_dec_tab[c] & 0x3F) << 18 + c = s[i+1] - 43 + val |= (b64_dec_tab[c] & 0x3F) << 12 + c = s[i+2] - 43 + if s[i+2] != '=': val |= (b64_dec_tab[c] & 0x3F) << 6 + c = s[i+3] - 43 + if s[i+3] != '=': val |= (b64_dec_tab[c] & 0x3F) + # 还原3字节 + out[j] = (val >> 16) & 0xFF + j += 1 + if s[i+2] != '=': + out[j] = (val >> 8) & 0xFF + j += 1 + if s[i+3] != '=': + out[j] = val & 0xFF + j += 1 + return j diff --git a/includes/binascii.py b/includes/binascii.py new file mode 100644 index 0000000..4c9b7b5 --- /dev/null +++ b/includes/binascii.py @@ -0,0 +1,348 @@ +import t, c + + +HEX_CHARS: t.CArray[t.CChar, None] = "0123456789abcdef" +HEX_VALS: t.CArray[t.CInt, 256] + +B64_CHARS: t.CArray[t.CChar, None] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" +B64_VALS: t.CArray[t.CInt, 256] + +HQX_CHARS: t.CArray[t.CChar, 64] = [ + ' ', '!', chr(34), '#', '$', '%', '&', "'", '(', ')', '*', '+', ',', '-', '.', '/', + '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', + '@', + 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', + 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'X', 'Y', 'Z', + '[', chr(92), '`', + 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm' +] +HQX_VALS: t.CArray[t.CInt, 256] + +UU_CHARS: t.CArray[t.CChar, None] = " !\"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_" + +CRC32_TABLE: t.CArray[t.CUnsignedInt, 256] = [ + 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3, + 0x0EDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91, + 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7, + 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5, + 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B, + 0x35B5A8FA, 0x42B2986C, 0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59, + 0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F, + 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D, + 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433, + 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01, + 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457, + 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65, + 0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB, + 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9, + 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F, + 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD, + 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683, + 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1, + 0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7, + 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5, + 0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B, + 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79, + 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F, + 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D, + 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713, + 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21, + 0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777, + 0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45, + 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB, + 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5, 0x47B2CF7F, 0x30B5FFE9, + 0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF, + 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D +] + +CRC_HQX_TABLE: t.CArray[t.CUnsignedShort, 256] = [ + 0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50A5, 0x60C6, 0x70E7, + 0x8108, 0x9129, 0xA14A, 0xB16B, 0xC18C, 0xD1AD, 0xE1CE, 0xF1EF, + 0x1231, 0x0210, 0x3273, 0x2252, 0x52B5, 0x4294, 0x72F7, 0x62D6, + 0x9339, 0x8318, 0xB37B, 0xA35A, 0xD3BD, 0xC39C, 0xF3FF, 0xE3DE, + 0x2462, 0x3443, 0x0420, 0x1401, 0x64E6, 0x74C7, 0x44A4, 0x5485, + 0xA56A, 0xB54B, 0x8528, 0x9509, 0xE5EE, 0xF5CF, 0xC5AC, 0xD58D, + 0x3653, 0x2672, 0x1611, 0x0630, 0x76D7, 0x66F6, 0x5695, 0x46B4, + 0xB75B, 0xA77A, 0x9719, 0x8738, 0xF7DF, 0xE7FE, 0xD79D, 0xC7BC, + 0x48C4, 0x58E5, 0x6886, 0x78A7, 0x0840, 0x1861, 0x2802, 0x3823, + 0xC9CC, 0xD9ED, 0xE98E, 0xF9AF, 0x8948, 0x9969, 0xA90A, 0xB92B, + 0x5AF5, 0x4AD4, 0x7AB7, 0x6A96, 0x1A71, 0x0A50, 0x3A33, 0x2A12, + 0xDBFD, 0xCBDC, 0xFBBF, 0xEB9E, 0x9B79, 0x8B58, 0xBB3B, 0xAB1A, + 0x6CA6, 0x7C87, 0x4CE4, 0x5CC5, 0x2C22, 0x3C03, 0x0C60, 0x1C41, + 0xEDAE, 0xFD8F, 0xCDEC, 0xDDCD, 0xAD2A, 0xBD0B, 0x8D68, 0x9D49, + 0x7E97, 0x6EB6, 0x5ED5, 0x4EF4, 0x3E13, 0x2E32, 0x1E51, 0x0E70, + 0xFF9F, 0xEFBE, 0xDFDD, 0xCFFC, 0xBF1B, 0xAF3A, 0x9F59, 0x8F78, + 0x9188, 0x81A9, 0xB1CA, 0xA1EB, 0xD10C, 0xC12D, 0xF14E, 0xE16F, + 0x1080, 0x00A1, 0x30C2, 0x20E3, 0x5004, 0x4025, 0x7046, 0x6067, + 0x83B9, 0x9398, 0xA3FB, 0xB3DA, 0xC33D, 0xD31C, 0xE37F, 0xF35E, + 0x02B1, 0x1290, 0x22F3, 0x32D2, 0x4235, 0x5214, 0x6277, 0x7256, + 0xB5EA, 0xA5CB, 0x95A8, 0x8589, 0xF56E, 0xE54F, 0xD52C, 0xC50D, + 0x34E2, 0x24C3, 0x14A0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405, + 0xA7DB, 0xB7FA, 0x8799, 0x97B8, 0xE75F, 0xF77E, 0xC71D, 0xD73C, + 0x26D3, 0x36F2, 0x0691, 0x16B0, 0x6657, 0x7676, 0x4615, 0x5634, + 0xD94C, 0xC96D, 0xF90E, 0xE92F, 0x99C8, 0x89E9, 0xB98A, 0xA9AB, + 0x5844, 0x4865, 0x7806, 0x6827, 0x18C0, 0x08E1, 0x3882, 0x28A3, + 0xCB7D, 0xDB5C, 0xEB3F, 0xFB1E, 0x8BF9, 0x9BD8, 0xABBB, 0xBB9A, + 0x4A75, 0x5A54, 0x6A37, 0x7A16, 0x0AF1, 0x1AD0, 0x2AB3, 0x3A92, + 0xFD2E, 0xED0F, 0xDD6C, 0xCD4D, 0xBDAA, 0xAD8B, 0x9DE8, 0x8DC9, + 0x7C26, 0x6C07, 0x5C64, 0x4C45, 0x3CA2, 0x2C83, 0x1CE0, 0x0CC1, + 0xEF1F, 0xFF3E, 0xCF5D, 0xDF7C, 0xAF9B, 0xBFBA, 0x8FD9, 0x9FF8, + 0x6E17, 0x7E36, 0x4E55, 0x5E74, 0x2E93, 0x3EB2, 0x0ED1, 0x1EF0 +] + +def init_binascii(): + i: t.CInt + for i in range(256): + HEX_VALS[i] = -1 + B64_VALS[i] = -1 + HQX_VALS[i] = -1 + for i in range(16): + HEX_VALS[t.CInt(HEX_CHARS[i])] = i + HEX_VALS[t.CInt(HEX_CHARS[i] - 32)] = i + for i in range(64): + B64_VALS[t.CInt(B64_CHARS[i])] = i + for i in range(64): + HQX_CHARS[i] = t.CChar(32 + i) + for i in range(64): + HQX_VALS[t.CUnsignedChar(HQX_CHARS[i])] = i + +def crc32(data: str, length: t.CInt, crc: t.CUnsignedInt) -> t.CUnsignedInt: + i: t.CInt + s: t.CUnsignedInt = crc ^ 0xFFFFFFFF + for i in range(length): + s = CRC32_TABLE[(s ^ t.CUnsignedChar(data[i])) & 0xFF] ^ (s >> 8) + return s ^ 0xFFFFFFFF + +def crc_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort) -> t.CUnsignedShort: + i: t.CInt + s: t.CUnsignedShort = crc + for i in range(length): + s = CRC_HQX_TABLE[t.CUnsignedChar(s >> 8)] ^ (s << 1 | (data[i] & 0xFF)) + return s + +def hexlify(data: str, length: t.CInt) -> str: + if not data or length < 0: return None + out: str = str(malloc(length * 2 + 1)) + if not out: return None + oi: t.CInt = 0 + for i in range(length): + s: t.CUnsignedChar = t.CUnsignedChar(data[i]) + out[oi] = HEX_CHARS[(s >> 4) & 0x0F] + oi += 1 + out[oi] = HEX_CHARS[(s >> 0) & 0x0F] + oi += 1 + out[oi] = '\0' + return out + +def b2a_hex(data: str, length: t.CInt) -> str: + return hexlify(data, length) + +def a2b_hex(hexstr: str, length: t.CInt) -> str: + if not hexstr or length < 0 or length % 2 != 0: return None + out_len: t.CInt = length // 2 + out: str = str(malloc(out_len + 1)) + if not out: return None + oi: t.CInt = 0 + for i in range(0, length, 2): + hi: t.CInt = HEX_VALS[t.CUnsignedChar(hexstr[i])] + lo: t.CInt = HEX_VALS[t.CUnsignedChar(hexstr[i + 1])] + if hi < 0 or lo < 0: + free(out) + return None + out[oi] = t.CChar((hi << 4) | lo) + oi += 1 + out[oi] = '\0' + return out + +def unhexlify(data: str, length: t.CInt): + return a2b_hex(data, length) + +def b2a_base64(data: str, length: t.CInt) -> str: + if not data or length < 0: return None + olen: t.CInt = (length // 3) * 4 + if length % 3 != 0: + olen += 4 + olen += 1 + out: str = str(malloc(olen + 1)) + if not out: return None + oi: t.CInt = 0 + for i in range(0, length, 3): + b0: t.CUnsignedChar = t.CUnsignedChar(data[i]) + b1: t.CUnsignedChar = 0 + b2: t.CUnsignedChar = 0 + if i + 1 < length: + b1 = t.CUnsignedChar(data[i + 1]) + if i + 2 < length: + b2 = t.CUnsignedChar(data[i + 2]) + out[oi] = B64_CHARS[(b0 >> 2) & 0x3F] + oi += 1 + out[oi] = B64_CHARS[((b0 & 0x03) << 4) | ((b1 >> 4) & 0x0F)] + oi += 1 + if i + 1 < length: + out[oi] = B64_CHARS[((b1 & 0x0F) << 2) | ((b2 >> 6) & 0x03)] + else: + out[oi] = '=' + oi += 1 + if i + 2 < length: + out[oi] = B64_CHARS[b2 & 0x3F] + else: + out[oi] = '=' + oi += 1 + out[oi] = '\0' + return out + +def a2b_base64(data: str, length: t.CInt) -> str: + if not data or length < 4: return None + olen: t.CInt = (length // 4) * 3 + if length >= 4 and data[length - 1] == '=': + olen -= 1 + if length >= 4 and data[length - 2] == '=': + olen -= 1 + out: str = str(malloc(olen + 1)) + if not out: return None + oi: t.CInt = 0 + for i in range(0, length, 4): + v0: t.CInt = 0 + v1: t.CInt = 0 + v2: t.CInt = 0 + v3: t.CInt = 0 + j: t.CInt + for j in range(4): + c: t.CChar = data[i + j] + if c == '=': + pass + else: + k: t.CInt + for k in range(64): + if B64_CHARS[k] == c: + if j == 0: v0 = k + elif j == 1: v1 = k + elif j == 2: v2 = k + elif j == 3: v3 = k + break + out[oi] = t.CChar((v0 << 2) | (v1 >> 4)) + oi += 1 + if data[i + 2] != '=': + out[oi] = t.CChar(((v1 & 0x0F) << 4) | (v2 >> 2)) + oi += 1 + if data[i + 3] != '=': + out[oi] = t.CChar(((v2 & 0x03) << 6) | v3) + oi += 1 + out[oi] = '\0' + return out + +def b2a_uu(data: str, length: t.CInt) -> str: + if not data or length < 1 or length > 45: return None + out: str = str(malloc(length * 2 + 2)) + if not out: return None + oi: t.CInt = 0 + out[oi] = UU_CHARS[length] + oi += 1 + for i in range(0, length, 3): + b0: t.CUnsignedChar = t.CUnsignedChar(data[i]) + b1: t.CUnsignedChar = 0 + b2: t.CUnsignedChar = 0 + if i + 1 < length: + b1 = t.CUnsignedChar(data[i + 1]) + if i + 2 < length: + b2 = t.CUnsignedChar(data[i + 2]) + m: t.CUnsignedInt = (t.CUnsignedInt(b0) << 16) | (t.CUnsignedInt(b1) << 8) | t.CUnsignedInt(b2) + out[oi] = UU_CHARS[(m >> 18) & 0x3F] + oi += 1 + out[oi] = UU_CHARS[(m >> 12) & 0x3F] + oi += 1 + out[oi] = UU_CHARS[(m >> 6) & 0x3F] + oi += 1 + out[oi] = UU_CHARS[m & 0x3F] + oi += 1 + out[oi] = '\n' + oi += 1 + out[oi] = '\0' + return out + +def a2b_uu(data: str) -> str: + if not data or data[0] == ' ' or data[0] == '`': return None + length: t.CInt = t.CInt(data[0]) - 32 + if length < 0 or length > 45: return None + out: str = str(malloc(length + 1)) + if not out: return None + oi: t.CInt = 0 + i: t.CInt = 1 + while oi < length: + v0: t.CInt = data[i] - 32 + v1: t.CInt = data[i + 1] - 32 + v2: t.CInt = data[i + 2] - 32 + v3: t.CInt = data[i + 3] - 32 + if v0 < 0 or v0 > 63: v0 = 0 + if v1 < 0 or v1 > 63: v1 = 0 + if v2 < 0 or v2 > 63: v2 = 0 + if v3 < 0 or v3 > 63: v3 = 0 + m: t.CUnsignedInt = (t.CUnsignedInt(v0) << 18) | (t.CUnsignedInt(v1) << 12) | (t.CUnsignedInt(v2) << 6) | t.CUnsignedInt(v3) + if oi < length: + out[oi] = t.CChar((m >> 16) & 0xFF) + oi += 1 + if oi < length: + out[oi] = t.CChar((m >> 8) & 0xFF) + oi += 1 + if oi < length: + out[oi] = t.CChar(m & 0xFF) + oi += 1 + i += 4 + out[oi] = '\0' + return out + +def b2a_hqx(data: str, length: t.CInt, crc: t.CUnsignedShort = 0) -> tuple[str, t.CUnsignedShort]: + if not data or length != 3: + return None, crc + out: str = str(malloc(5)) + if not out: + return None, crc + # 3字节 → 4字节 HQX + m: t.CUnsignedInt = (t.CUnsignedChar(data[0]) << 16) | (t.CUnsignedChar(data[1]) << 8) | t.CUnsignedChar(data[2]) + out[0] = HQX_CHARS[(m >> 18) & 0x3F] + out[1] = HQX_CHARS[(m >> 12) & 0x3F] + out[2] = HQX_CHARS[(m >> 6) & 0x3F] + out[3] = HQX_CHARS[m & 0x3F] + out[4] = '\0' + # CRC 初始化 + 正确计算 + s: t.CUnsignedShort = crc + for i in range(3): + byte: t.CUnsignedChar = t.CUnsignedChar(data[i]) + s = (s << 8) ^ CRC_HQX_TABLE[((s >> 8) ^ byte) & 0xFF] + return out, s + + +def a2b_hqx(data: str, crc: t.CUnsignedShort = 0) -> tuple[str, t.CUnsignedShort]: + if not data: + return None, crc + # 检查字符是否有效 + for i in range(4): + cchar: t.CUnsignedChar = t.CUnsignedChar(data[i]) + if HQX_VALS[cchar] < 0: + return None, crc + # 解码 + m: t.CUnsignedInt = 0 + for i in range(4): + m = (m << 6) | t.CUnsignedInt(HQX_VALS[t.CUnsignedChar(data[i])]) + out: str = str(malloc(4)) + if not out: + return None, crc + out[0] = t.CChar((m >> 16) & 0xFF) + out[1] = t.CChar((m >> 8) & 0xFF) + out[2] = t.CChar(m & 0xFF) + out[3] = '\0' + # CRC 初始化 + 正确计算 + s: t.CUnsignedShort = crc + for i in range(3): + byte: t.CUnsignedChar = t.CUnsignedChar(out[i]) + s = (s << 8) ^ CRC_HQX_TABLE[((s >> 8) ^ byte) & 0xFF] + return out, s + +def adler32(data: str, length: t.CInt, adler: t.CUnsignedInt) -> t.CUnsignedInt: + a: t.CUnsignedInt = adler & 0xFFFF + b: t.CUnsignedInt = (adler >> 16) & 0xFFFF + i: t.CInt + for i in range(length): + a = (a + t.CUnsignedChar(data[i])) % 65521 + b = (b + a) % 65521 + return (b << 16) | a diff --git a/includes/builtins.py b/includes/builtins.py new file mode 100644 index 0000000..13fe189 --- /dev/null +++ b/includes/builtins.py @@ -0,0 +1,11 @@ +import t + + +def print(*args, sep: str = " ", end: str = "\n") -> t.State: + ... +def input(prompt: str = "") -> str | t.State: + ... + +class type: + def __init__(self): ... + def __str__(self): ... \ No newline at end of file diff --git a/includes/condition.py b/includes/condition.py new file mode 100644 index 0000000..f999e4e --- /dev/null +++ b/includes/condition.py @@ -0,0 +1,53 @@ +import t, c +from stdint import * +from w32.win32base import * +from w32.win32sync import ( + CRITICAL_SECTION, + InitializeCriticalSection, EnterCriticalSection, LeaveCriticalSection, DeleteCriticalSection, + InitializeConditionVariable, SleepConditionVariableCS, + WakeConditionVariable, WakeAllConditionVariable +) + + +@t.Object +class Condition: + """基于 Windows 条件变量的条件变量，内部包含 CRITICAL_SECTION。 + + 用法: + cond: Condition = Condition() + with cond: + while not ready: + cond.wait() + cond.notify() + """ + cond_var: VOIDPTR + cs: CRITICAL_SECTION + + def __init__(self): + InitializeCriticalSection(c.Addr(self.cs)) + InitializeConditionVariable(c.Addr(self.cond_var)) + + def wait(self, timeout_ms: ULONG = INFINITE) -> BOOL: + return SleepConditionVariableCS(c.Addr(self.cond_var), c.Addr(self.cs), timeout_ms) + + def notify(self): + WakeConditionVariable(c.Addr(self.cond_var)) + + def notify_all(self): + WakeAllConditionVariable(c.Addr(self.cond_var)) + + def acquire(self): + EnterCriticalSection(c.Addr(self.cs)) + + def release(self): + LeaveCriticalSection(c.Addr(self.cs)) + + def __enter__(self) -> 'Condition' | t.CPtr: + self.acquire() + return self + + def __exit__(self): + self.release() + + def delete(self): + DeleteCriticalSection(c.Addr(self.cs)) diff --git a/includes/ctraits.pyi b/includes/ctraits.pyi new file mode 100644 index 0000000..9649dc0 --- /dev/null +++ b/includes/ctraits.pyi @@ -0,0 +1,20 @@ +"""ctraits — 编译时类型特征库 + +提供编译时类型判断函数，不纳入运行时编译。 +所有函数仅在编译时求值，可与 c.CIf 等编译时条件配合使用。 +""" +import t + + +def isptr(x: t.CType) -> bool: + """编译时判断 x 的类型是否为指针类型。 + + 返回 True 如果 x 是指针（ptr_count > 0），否则返回 False。 + 可与 c.CIf 配合使用实现条件编译。 + + 示例: + if c.CIf(ctraits.isptr(my_var)): + # 仅当 my_var 是指针类型时编译此分支 + ... + """ + pass diff --git a/includes/event.py b/includes/event.py new file mode 100644 index 0000000..449c06b --- /dev/null +++ b/includes/event.py @@ -0,0 +1,36 @@ +import t, c +from stdint import * +from w32.win32base import * +from w32.win32sync import CreateEventA, SetEvent, ResetEvent, WaitForSingleObject + + +@t.Object +class Event: + """基于 Windows Event 的事件对象，支持手动/自动重置。 + + 用法: + evt: Event = Event(FALSE, FALSE) + evt.set() + evt.wait() + """ + handle: HANDLE + manual_reset: BOOL + + def __init__(self, manual_reset: BOOL = FALSE, initial_state: BOOL = FALSE): + self.manual_reset = manual_reset + self.handle = CreateEventA(NULL, manual_reset, initial_state, NULL) + + def set(self): + SetEvent(self.handle) + + def clear(self): + ResetEvent(self.handle) + + def is_set(self) -> BOOL: + return WaitForSingleObject(self.handle, 0) == WAIT_OBJECT_0 + + def wait(self, timeout_ms: ULONG = INFINITE) -> BOOL: + return WaitForSingleObject(self.handle, timeout_ms) == WAIT_OBJECT_0 + + def close(self): + CloseHandle(self.handle) diff --git a/includes/hashlib/__init__.py b/includes/hashlib/__init__.py new file mode 100644 index 0000000..3ee2214 --- /dev/null +++ b/includes/hashlib/__init__.py @@ -0,0 +1,4 @@ +from .__md5 import md5, MD5_DIGEST_LEN +from .__sha1 import sha1, SHA1_DIGEST_LEN +from .__sha256 import sha256, SHA256_DIGEST_LEN +from .__sha512 import sha512, SHA512_DIGEST_LEN diff --git a/includes/hashlib/__md5.py b/includes/hashlib/__md5.py new file mode 100644 index 0000000..ad0f100 --- /dev/null +++ b/includes/hashlib/__md5.py @@ -0,0 +1,131 @@ +import string +import t, c + + +# ============================================== +# MD5 哈希算法 +# ============================================== +MD5_BLOCK_LEN: t.CDefine = 64 # 分组块大小 +MD5_DIGEST_LEN: t.CDefine = 16 # 摘要输出长度 + +# MD5 四轮非线性函数 +def md5_F(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return (x & y) | (~x & z) +def md5_G(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return (x & z) | (y & ~z) +def md5_H(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return x ^ y ^ z +def md5_I(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return y ^ (x | ~z) +# 循环左移 +def md5_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: return (x << n) | (x >> (32 - n)) + +# MD5 常量表、移位表 +md5_T: t.CArray[t.CUInt32T, 64] = [ + 0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee, 0xf57c0faf, 0x4787c62a, 0xa8304613, 0xfd469501, + 0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be, 0x6b901122, 0xfd987193, 0xa679438e, 0x49b40821, + 0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa, 0xd62f105d, 0x02441453, 0xd8a1e681, 0xe7d3fbc8, + 0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed, 0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a, + 0xfffa3942, 0x8771f681, 0x6d9d6122, 0xfde5380c, 0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70, + 0x289b7ec6, 0xeaa127fa, 0xd4ef3085, 0x04881d05, 0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665, + 0xf4292244, 0x432aff97, 0xab9423a7, 0xfc93a039, 0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1, + 0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1, 0xf7537e82, 0xbd3af235, 0x2ad7d2bb, 0xeb86d391 +] + +md5_S: t.CArray[t.CInt, 64] = [ + 7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22, + 5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20, + 4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23, + 6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21 +] + +# MD5 上下文结构体 +@t.Object +class md5: + state: t.CArray[t.CUInt32T, 4] # 4个32位状态寄存器 + count: t.CUInt64T # 已处理比特数 + buf: t.CArray[t.CUInt8T, MD5_BLOCK_LEN] # 分组缓冲区 + # 初始化MD5上下文 + def __init__(self): + string.memset(self.buf, 0, MD5_BLOCK_LEN) + self.state[0] = 0x67452301 + self.state[1] = 0xefcdab89 + self.state[2] = 0x98badcfe + self.state[3] = 0x10325476 + self.count = 0 + # MD5 单分组压缩变换 + def transform(self, block: t.CUInt8T | t.CPtr): + a: t.CUInt32T = self.state[0] + b: t.CUInt32T = self.state[1] + c: t.CUInt32T = self.state[2] + d: t.CUInt32T = self.state[3] + w: t.CArray[t.CUInt32T, 16] + # 字节序转32位字 + for i in range(16): + w[i] = ((t.CUInt32T(block[i * 4 + 0]) << 0) | + (t.CUInt32T(block[i * 4 + 1]) << 8) | + (t.CUInt32T(block[i * 4 + 2]) << 16) | + (t.CUInt32T(block[i * 4 + 3]) << 24)) + # 4轮64步迭代 + for i in range(16): + tmp: t.CUInt32T = md5_F(b, c, d) + a + md5_T[i] + w[i] + a = d; d = c; c = b + b += md5_rotl(tmp, md5_S[i]) + # 在 C 中的 for i 0: + left: t.CSizeT = MD5_BLOCK_LEN - idx + if length >= left: + string.memcpy(self.buf + idx, data, left) + self.transform(self.buf) + data += left + length -= left + idx = 0 + while length >= MD5_BLOCK_LEN: + self.transform(data) + data += MD5_BLOCK_LEN + length -= MD5_BLOCK_LEN + if length > 0: + string.memcpy(self.buf + idx, data, length) + # 结束计算，输出16字节摘要 + def final(self, out: t.CArray[t.CUInt8T, MD5_DIGEST_LEN]): + idx: t.CSizeT = self.count % MD5_BLOCK_LEN + # 填充0x80结尾 + self.buf[idx] = 0x80 + idx += 1 + # 空间不够则先压缩一次 + if idx > MD5_BLOCK_LEN - 8: + string.memset(self.buf + idx, 0, MD5_BLOCK_LEN - idx) + self.transform(self.buf) + idx = 0 + # 补0到末尾8字节 + string.memset(self.buf + idx, 0, MD5_BLOCK_LEN - 8 - idx) + # 写入总比特数 + bits: t.CUInt64T = self.count << 3 + for i in range(8): + self.buf[MD5_BLOCK_LEN - 8 + i] = (bits >> (i * 8)) & 0xFF + self.transform(self.buf) + # 状态转小端字节输出 + for i in range(4): + out[i * 4 + 0] = (self.state[i] >> 0) & 0xFF + out[i * 4 + 1] = (self.state[i] >> 8) & 0xFF + out[i * 4 + 2] = (self.state[i] >> 16) & 0xFF + out[i * 4 + 3] = (self.state[i] >> 24) & 0xFF diff --git a/includes/hashlib/__sha1.py b/includes/hashlib/__sha1.py new file mode 100644 index 0000000..413cfc2 --- /dev/null +++ b/includes/hashlib/__sha1.py @@ -0,0 +1,120 @@ +import string +import t, c + + + +# ============================================== +# SHA1 哈希算法 +# ============================================== +SHA1_BLOCK_LEN: t.CDefine = 64 +SHA1_DIGEST_LEN: t.CDefine = 20 + +def sha1_rotl(x: t.CUInt32T, n: t.CInt) -> t.CUInt32T: return (x << n) | (x >> (32 - n)) +# 三轮逻辑函数 +def sha1_f1(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: return (b & c) | (~b & d) +def sha1_f2(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: return b ^ c ^ d +def sha1_f3(b: t.CUInt32T, c: t.CUInt32T, d: t.CUInt32T) -> t.CUInt32T: return (b & c) | (b & d) | (c & d) + +# SHA1 上下文 +@t.Object +class sha1: + state: t.CArray[t.CUInt32T, 5] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA1_BLOCK_LEN] + def __init__(self): + memset(self.buf, 0, SHA1_BLOCK_LEN) + self.state[0] = 0x67452301 + self.state[1] = 0xefcdab89 + self.state[2] = 0x98badcfe + self.state[3] = 0x10325476 + self.state[4] = 0xc3d2e1f0 + self.count = 0 + # 单分组压缩 + def transform(self, block: t.CUInt8T | t.CPtr): + w: t.CArray[t.CUInt32T, 80] + data: t.CUInt8T | t.CPtr = block + # 前16个字 + for i in range(16): + w[i] = (((t.CUInt32T(data[i * 4 + 0])) << 24) | + ((t.CUInt32T(data[i * 4 + 1])) << 16) | + ((t.CUInt32T(data[i * 4 + 2])) << 8) | + ((t.CUInt32T(data[i * 4 + 3])) << 0)) + # 扩展到80个字 + for i in range(16, 80): + w[i] = sha1_rotl(w[i-3] ^ w[i-8] ^ w[i-14] ^ w[i-16],1) + a: t.CUInt32T = self.state[0] + b: t.CUInt32T = self.state[1] + c: t.CUInt32T = self.state[2] + d: t.CUInt32T = self.state[3] + e: t.CUInt32T = self.state[4] + k: t.CUInt32T + tmp: t.CUInt32T + # 4轮迭代 + for i in range(0, 20): + k = 0x5a827999 + tmp = sha1_rotl(a,5) + sha1_f1(b, c, d) + e + k + w[i] + e = d; d = c + c = sha1_rotl(b, 30) + b = a; a = tmp + for i in range(20, 40): + k = 0x6ed9eba1 + tmp = sha1_rotl(a,5) + sha1_f2(b, c, d) + e + k + w[i] + e = d; d = c + c = sha1_rotl(b, 30) + b = a; a = tmp + for i in range(40, 60): + k = 0x8f1bbcdc + tmp = sha1_rotl(a,5) + sha1_f3(b, c, d) + e + k + w[i] + e = d; d = c + c = sha1_rotl(b, 30) + b = a; a = tmp + for i in range(60, 80): + k = 0xca62c1d6 + tmp = sha1_rotl(a,5) + sha1_f2(b, c, d) + e + k + w[i] + e = d; d = c + c = sha1_rotl(b, 30) + b = a; a = tmp + self.state[0] += a + self.state[1] += b + self.state[2] += c + self.state[3] += d + self.state[4] += e + def update(self, s: str): + data: t.CUInt8T | t.CPtr = t.CUInt8T(s, t.CPtr) + length: t.CSizeT = len(s) + idx: t.CSizeT = self.count % SHA1_BLOCK_LEN + self.count += length + if idx > 0: + left: t.CSizeT = SHA1_BLOCK_LEN - idx + if length >= left: + memcpy(self.buf + idx, data, left) + self.transform(self.buf) + data += left + length -= left + idx = 0 + while length >= SHA1_BLOCK_LEN: + self.transform(data) + data += SHA1_BLOCK_LEN + length -= SHA1_BLOCK_LEN + if length > 0: + memcpy(self.buf + idx, data, length) + def final(self, out: t.CArray[t.CUInt8T, SHA1_DIGEST_LEN]): + idx: t.CSizeT = self.count % SHA1_BLOCK_LEN + self.buf[idx] = 0x80 + idx += 1 + if idx > SHA1_BLOCK_LEN - 8: + memset(self.buf + idx, 0, SHA1_BLOCK_LEN - idx) + self.transform(self.buf) + idx = 0 + memset(self.buf + idx,0,SHA1_BLOCK_LEN-8-idx) + bits: t.CUInt64T = self.count << 3 + # SHA1 大端写入长度 + for i in range(8): + self.buf[SHA1_BLOCK_LEN - 8 + i] = (bits >> (56 - i * 8)) & 0xFF + self.transform(self.buf) + # 大端输出摘要 + for i in range(5): + out[i * 4 + 0] = (self.state[i] >> 24) & 0xFF + out[i * 4 + 1] = (self.state[i] >> 16) & 0xFF + out[i * 4 + 2] = (self.state[i] >> 8) & 0xFF + out[i * 4 + 3] = (self.state[i] >> 0) & 0xFF diff --git a/includes/hashlib/__sha256.py b/includes/hashlib/__sha256.py new file mode 100644 index 0000000..718a3c2 --- /dev/null +++ b/includes/hashlib/__sha256.py @@ -0,0 +1,123 @@ +import t, c + + +# ============================================== +# SHA256 哈希算法 +# ============================================== +SHA256_BLOCK_LEN: t.CDefine = 64 +SHA256_DIGEST_LEN: t.CDefine = 32 + +# SHA256 常量K +sha256_K: t.CArray[t.CUInt32T, 64] = [ + 0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5, + 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174, + 0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da, + 0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967, + 0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85, + 0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070, + 0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3, + 0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2 +] + +# SHA256 辅助函数 +def s0(x: t.CUInt32T) -> t.CUInt32T: return ((x >> 7) | (x << 25)) ^ ((x >> 18) | (x << 14)) ^ ((x >> 3)) +def s1(x: t.CUInt32T) -> t.CUInt32T: return ((x >> 17) | (x << 15)) ^ ((x >> 19) | (x << 13)) ^ ((x >> 10)) +def S0(x: t.CUInt32T) -> t.CUInt32T: return ((x >> 2) | (x << 30)) ^ ((x >> 13) | (x << 19)) ^ ((x >> 22) | (x << 10)) +def S1(x: t.CUInt32T) -> t.CUInt32T: return ((x >> 6) | (x << 26)) ^ ((x >> 11) | (x << 21)) ^ ((x >> 25) | (x << 7)) +def Ch(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return (x & y) ^ (~x & z) +def Maj(x: t.CUInt32T, y: t.CUInt32T, z: t.CUInt32T) -> t.CUInt32T: return (x & y) ^ (x & z) ^ (y & z) + +# SHA256 上下文 +@t.Object +class sha256: + state: t.CArray[t.CUInt32T, 8] + count: t.CUInt64T + buf: t.CArray[t.CUInt8T, SHA256_BLOCK_LEN] + def __init__(self): + memset(self, 0, sha256.__sizeof__()) + self.state[0] = 0x6a09e667 + self.state[1] = 0xbb67ae85 + self.state[2] = 0x3c6ef372 + self.state[3] = 0xa54ff53a + self.state[4] = 0x510e527f + self.state[5] = 0x9b05688c + self.state[6] = 0x1f83d9ab + self.state[7] = 0x5be0cd19 + self.count = 0 + # memset(self.buf, 0, SHA256_BLOCK_LEN) + # 单分组压缩 + def transform(self, block: t.CUInt8T | t.CPtr): + w: t.CArray[t.CUInt32T, 64] + for i in range(16): + w[i] = ((t.CUInt32T(block[i * 4 + 0]) << 24) | + (t.CUInt32T(block[i * 4 + 1]) << 16) | + (t.CUInt32T(block[i * 4 + 2]) << 8 ) | + (t.CUInt32T(block[i * 4 + 3]) << 0 )) + # 扩展64字 + for i in range(16, 64): + w[i] = w[i-16] + s0(w[i-15]) + w[i-7] + s1(w[i-2]) + a: t.CUInt32T = self.state[0] + b: t.CUInt32T = self.state[1] + c: t.CUInt32T = self.state[2] + d: t.CUInt32T = self.state[3] + e: t.CUInt32T = self.state[4] + f: t.CUInt32T = self.state[5] + g: t.CUInt32T = self.state[6] + h: t.CUInt32T = self.state[7] + t1: t.CUInt32T + t2: t.CUInt32T + # 64步迭代 + for i in range(64): + t1 = h + S1(e) + Ch(e,f,g) + sha256_K[i] + w[i] + t2 = S0(a) + Maj(a,b,c) + h = g; g = f; f = e + e = d + t1 + d = c; c = b; b = a + a = t1 + t2 + # 累加状态 + self.state[0] += a + self.state[1] += b + self.state[2] += c + self.state[3] += d + self.state[4] += e + self.state[5] += f + self.state[6] += g + self.state[7] += h + def update(self, s: str): + data: t.CUInt8T | t.CPtr = t.CUInt8T(s, t.CPtr) + length: t.CSizeT = len(s) + idx: t.CSizeT = self.count % SHA256_BLOCK_LEN + self.count += length + if idx > 0: + left: t.CSizeT = SHA256_BLOCK_LEN - idx + if length >= left: + memcpy(self.buf + idx, data, left) + self.transform(self.buf) + data += left + length -= left + idx = 0 + while length >= SHA256_BLOCK_LEN: + self.transform(data) + data += SHA256_BLOCK_LEN + length -= SHA256_BLOCK_LEN + if length > 0: + memcpy(self.buf + idx, data, length) + def final(self, out: t.CArray[t.CUInt8T, SHA256_DIGEST_LEN]): + idx: t.CSizeT = self.count % SHA256_BLOCK_LEN + self.buf[idx] = 0x80 + idx += 1 + if idx > SHA256_BLOCK_LEN - 8: + memset(self.buf + idx, 0, SHA256_BLOCK_LEN - idx) + self.transform(self.buf) + idx = 0 + memset(self.buf + idx, 0, SHA256_BLOCK_LEN - 8 - idx) + bits: t.CUInt64T = self.count << 3 + for i in range(8): + self.buf[SHA256_BLOCK_LEN-8+i] = (bits >> (56 - i*8)) & 0xFF + self.transform(self.buf) + # 大端输出32字节摘要 + for i in range(8): + out[i * 4 + 0] = (self.state[i] >> 24) & 0xFF + out[i * 4 + 1] = (self.state[i] >> 16) & 0xFF + out[i * 4 + 2] = (self.state[i] >> 8) & 0xFF + out[i * 4 + 3] = (self.state[i] >> 0) & 0xFF diff --git a/includes/hashlib/__sha512.py b/includes/hashlib/__sha512.py new file mode 100644 index 0000000..3db4c86 --- /dev/null +++ b/includes/hashlib/__sha512.py @@ -0,0 +1,148 @@ +import t, c + + +# ============================================== +# SHA512 哈希算法 +# ============================================== +SHA512_BLOCK_LEN: t.CDefine = 128 +SHA512_DIGEST_LEN: t.CDefine = 64 + +def sha512_ror(x: t.CUInt64T, n: int) -> t.CUInt64T: return (x >> n) | (x << (64 - n)) +def sha512_shr(x: t.CUInt64T, n: int) -> t.CUInt64T: return x >> n +def sha512_Ch (x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: return (x & y) ^ (~x & z) +def sha512_Maj(x: t.CUInt64T, y: t.CUInt64T, z: t.CUInt64T) -> t.CUInt64T: return (x & y) ^ (x & z) ^ (y & z) +def sha512_BigSigma0(x: t.CUInt64T) -> t.CUInt64T: return sha512_ror(x, 28) ^ sha512_ror(x, 34) ^ sha512_ror(x, 39) +def sha512_BigSigma1(x: t.CUInt64T) -> t.CUInt64T: return sha512_ror(x, 14) ^ sha512_ror(x, 18) ^ sha512_ror(x, 41) +def sha512_Sigma0(x: t.CUInt64T) -> t.CUInt64T: return sha512_ror(x, 1) ^ sha512_ror(x, 8) ^ sha512_shr(x, 7) +def sha512_Sigma1(x: t.CUInt64T) -> t.CUInt64T: return sha512_ror(x, 19) ^ sha512_ror(x, 61) ^ sha512_shr(x, 6) + +sha512_K: t.CArray[t.CUInt64T, 80] = [ + 0x428a2f98d728ae22, 0x7137449123ef65cd, 0xb5c0fbcfec4d3b2f, 0xe9b5dba58189dbbc, + 0x3956c25bf348b538, 0x59f111f1b605d019, 0x923f82a4af194f9b, 0xab1c5ed5da6d8118, + 0xd807aa98a3030242, 0x12835b0145706fbe, 0x243185be4ee4b28c, 0x550c7dc3d5ffb4e2, + 0x72be5d74f27b896f, 0x80deb1fe3b1696b1, 0x9bdc06a725c71235, 0xc19bf174cf692694, + 0xe49b69c19ef14ad2, 0xefbe4786384f25e3, 0x0fc19dc68b8cd5b5, 0x240ca1cc77ac9c65, + 0x2de92c6f592b0275, 0x4a7484aa6ea6e483, 0x5cb0a9dcbd41fbd4, 0x76f988da831153b5, + 0x983e5152ee66dfab, 0xa831c66d2db43210, 0xb00327c898fb213f, 0xbf597fc7beef0ee4, + 0xc6e00bf33da88fc2, 0xd5a79147930aa725, 0x06ca6351e003826f, 0x142929670a0e6e70, + 0x27b70a8546d22ffc, 0x2e1b21385c26c926, 0x4d2c6dfc5ac42aed, 0x53380d139d95b3df, + 0x650a73548baf63de, 0x766a0abb3c77b2a8, 0x81c2c92e47edaee6, 0x92722c851482353b, + 0xa2bfe8a14cf10364, 0xa81a664bbc423001, 0xc24b8b70d0f89791, 0xc76c51a30654be30, + 0xd192e819d6ef5218, 0xd69906245565a910, 0xf40e35855771202a, 0x106aa07032bbd1b8, + 0x19a4c116b8d2d0c8, 0x1e376c085141ab53, 0x2748774cdf8eeb99, 0x34b0bcb5e19b48a8, + 0x391c0cb3c5c95a63, 0x4ed8aa4ae3418acb, 0x5b9cca4f7763e373, 0x682e6ff3d6b2b8a3, + 0x748f82ee5defb2fc, 0x78a5636f43172f60, 0x84c87814a1f0ab72, 0x8cc702081a6439ec, + 0x90befffa23631e28, 0xa4506cebde82bde9, 0xbef9a3f7b2c67915, 0xc67178f2e372532b, + 0xca273eceea26619c, 0xd186b8c721c0c207, 0xeada7dd6cde0eb1e, 0xf57d4f7fee6ed178, + 0x06f067aa72176fba, 0x0a637dc5a2c898a6, 0x113f9804bef90dae, 0x1b710b35131c471b, + 0x28db77f523047d84, 0x32caab7b40c72493, 0x3c9ebe0a15c9bebc, 0x431d67c49c100d4c, + 0x4cc5d4becb3e42b6, 0x597f299cfc657e2a, 0x5fcb6fab3ad6faec, 0x6c44198c4a475817 +] + +@t.Object +class sha512: + state: t.CArray[t.CUInt64T, 8] + count: t.CArray[t.CUInt64T, 2] + buf: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN] + def __init__(self): + memset(self, 0, sha512.__sizeof__()) + self.state[0] = 0x6a09e667f3bcc908 + self.state[1] = 0xbb67ae8584caa73b + self.state[2] = 0x3c6ef372fe94f82b + self.state[3] = 0xa54ff53a5f1d36f1 + self.state[4] = 0x510e527fade682d1 + self.state[5] = 0x9b05688c2b3e6c1f + self.state[6] = 0x1f83d9abfb41bd6b + self.state[7] = 0x5be0cd19137e2179 + def transform(self, block: t.CArray[t.CUInt8T, SHA512_BLOCK_LEN]): + w: t.CArray[t.CUInt64T, 80] + a: t.CUInt64T + b: t.CUInt64T + c: t.CUInt64T + d: t.CUInt64T + e: t.CUInt64T + f: t.CUInt64T + g: t.CUInt64T + h: t.CUInt64T + t1: t.CUInt64T + t2: t.CUInt64T + for i in range(16): + w[i] = t.CUInt64T(block[i * 8 + 0]) << 56 + w[i] |= t.CUInt64T(block[i * 8 + 1]) << 48 + w[i] |= t.CUInt64T(block[i * 8 + 2]) << 40 + w[i] |= t.CUInt64T(block[i * 8 + 3]) << 32 + w[i] |= t.CUInt64T(block[i * 8 + 4]) << 24 + w[i] |= t.CUInt64T(block[i * 8 + 5]) << 16 + w[i] |= t.CUInt64T(block[i * 8 + 6]) << 8 + w[i] |= t.CUInt64T(block[i * 8 + 7]) << 0 + for i in range(16, 80): + w[i] = sha512_Sigma1(w[i-2]) + w[i-7] + sha512_Sigma0(w[i-15]) + w[i-16] + a = self.state[0] + b = self.state[1] + c = self.state[2] + d = self.state[3] + e = self.state[4] + f = self.state[5] + g = self.state[6] + h = self.state[7] + for i in range(80): + t1 = h + sha512_BigSigma1(e) + sha512_Ch(e, f, g) + sha512_K[i] + w[i] + t2 = sha512_BigSigma0(a) + sha512_Maj(a, b, c) + h = g; g = f; f = e + e = d + t1 + d = c; c = b; b = a + a = t1 + t2 + self.state[0] += a + self.state[1] += b + self.state[2] += c + self.state[3] += d + self.state[4] += e + self.state[5] += f + self.state[6] += g + self.state[7] += h + def update(self, s: str): + data: t.CUInt8T | t.CPtr = t.CUInt8T(s, t.CPtr) + length: t.CSizeT = len(s) + idx: t.CUInt32T = t.CUInt32T(self.count[0] % SHA512_BLOCK_LEN) + self.count[0] += length + if self.count[0] < length: + self.count[1] += 1 + if idx > 0: + left: t.CSizeT = SHA512_BLOCK_LEN - idx + if length >= left: + memcpy(self.buf + idx, data, left) + self.transform(self.buf) + data += left + length -= left + idx = 0 + while length >= SHA512_BLOCK_LEN: + self.transform(data) + data += SHA512_BLOCK_LEN + length -= SHA512_BLOCK_LEN + if length > 0: + memcpy(self.buf + idx, data, length) + def final(self, out: t.CArray[t.CUInt8T, SHA512_DIGEST_LEN]): + bits_lo: t.CUInt64T = self.count[0] << 3 + bits_hi: t.CUInt64T = (self.count[1] << 3) | (self.count[0] >> 61) + idx: t.CUInt32T = t.CUInt32T(self.count[0] % SHA512_BLOCK_LEN) + self.buf[idx] = 0x80 + idx += 1 + if idx > SHA512_BLOCK_LEN - 16: + memset(self.buf + idx, 0, SHA512_BLOCK_LEN - idx) + self.transform(self.buf) + idx = 0 + memset(self.buf + idx, 0, SHA512_BLOCK_LEN - 16 - idx) + for i in range(8): + self.buf[SHA512_BLOCK_LEN - 16 + i] = t.CUInt8T(bits_hi >> (56 - i * 8)) + for i in range(8): + self.buf[SHA512_BLOCK_LEN - 8 + i] = t.CUInt8T(bits_lo >> (56 - i * 8)) + self.transform(self.buf) + for i in range(8): + out[i * 8 + 0] = t.CUInt8T(self.state[i] >> 56) + out[i * 8 + 1] = t.CUInt8T(self.state[i] >> 48) + out[i * 8 + 2] = t.CUInt8T(self.state[i] >> 40) + out[i * 8 + 3] = t.CUInt8T(self.state[i] >> 32) + out[i * 8 + 4] = t.CUInt8T(self.state[i] >> 24) + out[i * 8 + 5] = t.CUInt8T(self.state[i] >> 16) + out[i * 8 + 6] = t.CUInt8T(self.state[i] >> 8) + out[i * 8 + 7] = t.CUInt8T(self.state[i] >> 0) diff --git a/includes/hashtable.py b/includes/hashtable.py new file mode 100644 index 0000000..831436e --- /dev/null +++ b/includes/hashtable.py @@ -0,0 +1,240 @@ +# hashtable - 开放寻址哈希表库 +# 使用线性探测处理冲突，支持自动扩容和墓碑删除 +# 键类型固定为 str (i8*)，值类型为 void* (通用指针) +import t, c +import string +import memhub +from stdint import * + +# 空槽标记 (hash == 0) +_HT_EMPTY: t.CDefine = 0 +# 墓碑标记 (hash == 1) +_HT_TOMBSTONE: t.CDefine = 1 +# 默认初始容量 (必须是 2 的幂) +_HT_INIT_CAP: t.CDefine = 16 +# 槽大小: hash(8) + key(8) + value(8) = 24 字节 +_HT_SLOT_SIZE: t.CDefine = 24 +# FNV-1a 64-bit 常量 +_FNV_OFFSET: t.CUInt64T = 0xcbf29ce484222325 +_FNV_PRIME: t.CUInt64T = 0x100000001b3 + + +def _fnv1a_hash(key: str) -> t.CUInt64T: + """FNV-1a 64-bit 哈希函数""" + h: t.CUInt64T = _FNV_OFFSET + k: str = key + while k[0] != 0: + h = h ^ t.CUInt64T(k[0]) + h = h * _FNV_PRIME + k += 1 + # 保证 h >= 2 (避开空槽 0 和墓碑 1) + if h < 2: + h = 2 + return h + + +class HashTable: + """开放寻址哈希表，使用线性探测处理冲突。 + + 键类型固定为 str (i8*)，值类型为 void* (通用指针)。 + 使用 mbuddy 分配器管理内存。负载因子超过 0.75 时自动扩容。 + + 用法: + mb: memhub.MemBuddy = memhub.MemBuddy(arena, arena_size) + ht: HashTable = HashTable(mb) + ht.set_int("x", 42) + p: t.CPtr = ht["x"] + v: int = c.Deref(p) + """ + __slots__: t.CVoid | t.CPtr # 槽位数组基址 + __count__: t.CSizeT = 0 # 已用条目数 + __capacity__: t.CSizeT = 16 # 槽位总数 (2 的幂) + __tombstones__: t.CSizeT = 0 # 墓碑数 + __mbuddy__: memhub.MemBuddy | t.CPtr + __iter_index__: t.CSizeT = 0 + + def __new__(self, mb: memhub.MemBuddy | t.CPtr) -> t.CPtr: + # 堆分配 HashTable 对象 (6 字段 x 8 字节 = 48 字节) + return mb.alloc(48) + + def __init__(self, mb: memhub.MemBuddy | t.CPtr): + self.__mbuddy__ = mb + self.__capacity__ = _HT_INIT_CAP + self.__count__ = 0 + self.__tombstones__ = 0 + self.__slots__ = mb.alloc(self.__capacity__ * _HT_SLOT_SIZE) + # 初始化所有槽为空 (hash = 0) + string.memset(self.__slots__, 0, self.__capacity__ * _HT_SLOT_SIZE) + + def _slot_hash(self, idx: t.CSizeT) -> t.CUInt64T: + """读取槽位的 hash 值""" + ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE) + return ptr[0] + + def _slot_key(self, idx: t.CSizeT) -> str: + """读取槽位的 key 指针""" + ptr: str | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE + 8) + return ptr[0] + + def _slot_value(self, idx: t.CSizeT) -> t.CPtr: + """读取槽位的 value 指针""" + ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE + 16) + return ptr[0] + + def _set_slot(self, idx: t.CSizeT, h: t.CUInt64T, key: str, val: t.CPtr): + """设置槽位的三个字段""" + h_ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE) + h_ptr[0] = h + k_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE + 8) + k_ptr[0] = key + v_ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE + 16) + v_ptr[0] = val + + def _find_slot(self, key: str, h: t.CUInt64T) -> t.CSizeT: + """查找 key 应该插入或已存在的槽位索引。 + + 返回: 如果找到 key，返回其索引；否则返回第一个可插入的空槽/墓碑索引。 + """ + mask: t.CSizeT = self.__capacity__ - 1 + idx: t.CSizeT = t.CSizeT(h) & mask + first_tombstone: t.CSizeT = self.__capacity__ + i: t.CSizeT = 0 + while i < self.__capacity__: + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY: + if first_tombstone < self.__capacity__: + return first_tombstone + return idx + if sh == _HT_TOMBSTONE: + if first_tombstone >= self.__capacity__: + first_tombstone = idx + else: + if sh == h: + sk: str = self._slot_key(idx) + if sk == key: + return idx + idx = (idx + 1) & mask + i += 1 + if first_tombstone < self.__capacity__: + return first_tombstone + return self.__capacity__ + + def _resize(self, new_cap: t.CSizeT): + """扩容并重新哈希所有条目""" + old_slots: t.CVoid | t.CPtr = self.__slots__ + old_cap: t.CSizeT = self.__capacity__ + new_slots: t.CVoid | t.CPtr = self.__mbuddy__.alloc(new_cap * _HT_SLOT_SIZE) + if new_slots == None: + return + string.memset(new_slots, 0, new_cap * _HT_SLOT_SIZE) + self.__slots__ = new_slots + self.__capacity__ = new_cap + self.__count__ = 0 + self.__tombstones__ = 0 + i: t.CSizeT = 0 + while i < old_cap: + old_h_ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(old_slots) + i * _HT_SLOT_SIZE) + h: t.CUInt64T = old_h_ptr[0] + if h != _HT_EMPTY and h != _HT_TOMBSTONE: + old_k_ptr: str | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(old_slots) + i * _HT_SLOT_SIZE + 8) + old_v_ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(old_slots) + i * _HT_SLOT_SIZE + 16) + k: str = old_k_ptr[0] + v: t.CPtr = old_v_ptr[0] + idx: t.CSizeT = self._find_slot(k, h) + self._set_slot(idx, h, k, v) + self.__count__ += 1 + i += 1 + + def __setitem__(self, key: str, val: t.CNeedPtr): + """插入或更新键值对。val 自动取地址 (t.CNeedPtr: 已是指针则不取)。""" + h: t.CUInt64T = _fnv1a_hash(key) + idx: t.CSizeT = self._find_slot(key, h) + if idx >= self.__capacity__: + return + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY or sh == _HT_TOMBSTONE: + if sh == _HT_TOMBSTONE: + self.__tombstones__ -= 1 + self._set_slot(idx, h, key, val) + self.__count__ += 1 + # 负载因子 > 0.75 时扩容 + load: t.CSizeT = self.__count__ + self.__tombstones__ + if load * 4 >= self.__capacity__ * 3: + self._resize(self.__capacity__ * 2) + else: + self._set_slot(idx, h, key, val) + + def __getitem__(self, key: str) -> t.CPtr: + """查找 key，返回值指针。不存在返回 None。""" + h: t.CUInt64T = _fnv1a_hash(key) + idx: t.CSizeT = self._find_slot(key, h) + if idx >= self.__capacity__: + return None + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY or sh == _HT_TOMBSTONE: + return None + return self._slot_value(idx) + + def __contains__(self, key: str) -> t.CInt: + """检查 key 是否存在。""" + h: t.CUInt64T = _fnv1a_hash(key) + idx: t.CSizeT = self._find_slot(key, h) + if idx >= self.__capacity__: + return 0 + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY or sh == _HT_TOMBSTONE: + return 0 + return 1 + + def __delitem__(self, key: str) -> t.CInt: + """删除 key。成功返回 1，key 不存在返回 0。""" + h: t.CUInt64T = _fnv1a_hash(key) + idx: t.CSizeT = self._find_slot(key, h) + if idx >= self.__capacity__: + return 0 + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY or sh == _HT_TOMBSTONE: + return 0 + h_ptr: t.CUInt64T | t.CPtr = (t.CVoid | t.CPtr)(t.CUInt64T(self.__slots__) + idx * _HT_SLOT_SIZE) + h_ptr[0] = _HT_TOMBSTONE + self.__count__ -= 1 + self.__tombstones__ += 1 + return 1 + + def __len__(self) -> t.CSizeT: + return self.__count__ + + def __iter__(self) -> HashTable | t.CPtr: + self.__iter_index__ = 0 + return self + + def __next__(self) -> str: + while self.__iter_index__ < self.__capacity__: + idx: t.CSizeT = self.__iter_index__ + self.__iter_index__ = idx + 1 + sh: t.CUInt64T = self._slot_hash(idx) + if sh != _HT_EMPTY and sh != _HT_TOMBSTONE: + return self._slot_key(idx) + raise StopIteration + + def get(self, key: str, default: t.CPtr) -> t.CPtr: + """返回值指针，不存在则返回 default。""" + h: t.CUInt64T = _fnv1a_hash(key) + idx: t.CSizeT = self._find_slot(key, h) + if idx >= self.__capacity__: + return default + sh: t.CUInt64T = self._slot_hash(idx) + if sh == _HT_EMPTY or sh == _HT_TOMBSTONE: + return default + return self._slot_value(idx) + + def set_int(self, key: str, val: int): + """存储 int 值。值复制到 mbuddy 持久存储。""" + storage: t.CVoid | t.CPtr = self.__mbuddy__.alloc(8) + int_ptr: int | t.CPtr = t.CVoid(t.CUInt64T(storage), t.CPtr) + int_ptr[0] = val + self[key] = storage + + def set_str(self, key: str, val: str): + """存储 str 值。直接存储字符串指针。""" + self[key] = val diff --git a/includes/json/__init__.py b/includes/json/__init__.py new file mode 100644 index 0000000..cc87fd6 --- /dev/null +++ b/includes/json/__init__.py @@ -0,0 +1,237 @@ +import t, c +from stdint import * +import memhub +import string +import viperio + +# ============================================================ +# JSON 值类型枚举 +# ============================================================ +JSON_NULL: t.CDefine = 0 +JSON_BOOL: t.CDefine = 1 +JSON_INT: t.CDefine = 2 +JSON_FLOAT: t.CDefine = 3 +JSON_STRING: t.CDefine = 4 +JSON_ARRAY: t.CDefine = 5 +JSON_OBJECT: t.CDefine = 6 + + +# ============================================================ +# JsonValue - JSON 值节点 +# 使用 mpool 分配：JsonValue(pool) 从池中分配内存 +# ============================================================ +class JsonValue: + vtype: t.CInt + bool_val: t.CInt + int_val: t.CInt64T + float_val: t.CDouble + str_val: t.CChar | t.CPtr + next: 'JsonValue' | t.CPtr + key: t.CChar | t.CPtr + child: 'JsonValue' | t.CPtr + child_count: t.CSizeT + + def __new__(self, pool: memhub.MemManager | t.CPtr): + """从 mpool 分配 JsonValue 结构体内存并零初始化""" + ptr: JsonValue | t.CPtr = pool.alloc(JsonValue.__sizeof__()) + if ptr: + string.memset(ptr, 0, JsonValue.__sizeof__()) + return ptr + + def type(self) -> t.CInt: + return self.vtype + + def is_null(self) -> bool: + return self.vtype == JSON_NULL + + def is_bool(self) -> bool: + return self.vtype == JSON_BOOL + + def is_int(self) -> bool: + return self.vtype == JSON_INT + + def is_float(self) -> bool: + return self.vtype == JSON_FLOAT + + def is_string(self) -> bool: + return self.vtype == JSON_STRING + + def is_array(self) -> bool: + return self.vtype == JSON_ARRAY + + def is_object(self) -> bool: + return self.vtype == JSON_OBJECT + + def as_bool(self) -> bool: + return self.bool_val != 0 + + def as_int(self) -> t.CInt64T: + return self.int_val + + def as_float(self) -> t.CDouble: + return self.float_val + + def as_string(self) -> t.CChar | t.CPtr: + return self.str_val + + def __len__(self) -> t.CSizeT: + return self.child_count + + def __getitem__(self, key: t.CChar | t.CPtr) -> 'JsonValue' | t.CPtr: + """Python风格取值：v["key"] 取对象字段""" + if self.vtype != JSON_OBJECT: + return None + if key == None: + return None + cur: JsonValue | t.CPtr = self.child + while cur != None: + if cur.key != None: + if cur.key == key: + return cur + cur = cur.next + return None + + def __setitem__(self, key: t.CChar | t.CPtr, val: 'JsonValue' | t.CPtr): + """Python风格赋值：v["key"] = val""" + if self.vtype != JSON_OBJECT or key == None or val == None: + return + cur: JsonValue | t.CPtr = self.child + while cur != None: + if cur.key != None: + if cur.key == key: + cur.vtype = val.vtype + cur.bool_val = val.bool_val + cur.int_val = val.int_val + cur.float_val = val.float_val + cur.str_val = val.str_val + cur.child = val.child + cur.child_count = val.child_count + return + cur = cur.next + # key 不存在，追加新子节点 + klen: t.CSizeT = string.strlen(key) + kbuf: t.CChar | t.CPtr = self._pool_alloc(klen + 1) + if kbuf != None: + string.memcpy(kbuf, key, klen + 1) + val.key = kbuf + self._append_child(val) + + def get_item(self, index: t.CSizeT) -> 'JsonValue' | t.CPtr: + """按索引取数组元素：v.get_item(0)""" + if self.vtype != JSON_ARRAY and self.vtype != JSON_OBJECT: + return None + cur: JsonValue | t.CPtr = self.child + i: t.CSizeT = 0 + while cur != None: + if i == index: + return cur + cur = cur.next + i += 1 + return None + + def _pool_alloc(self, size: t.CSizeT) -> t.CChar | t.CPtr: + """内部方法：从当前对象的 mpool 分配内存（预留，暂返回 None）""" + return None + + def _append_child(self, child: 'JsonValue' | t.CPtr): + if self.child == None: + self.child = child + else: + cur: JsonValue | t.CPtr = self.child + while cur.next != None: + cur = cur.next + cur.next = child + self.child_count += 1 + + +# ============================================================ +# 构造函数：创建各类 JsonValue（从 mpool 分配） +# ============================================================ + +def null(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_NULL + return v + +def bool_val(pool: memhub.MemManager | t.CPtr, val: bool) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_BOOL + v.bool_val = 1 if val else 0 + return v + +def int_val(pool: memhub.MemManager | t.CPtr, val: t.CInt64T) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_INT + v.int_val = val + return v + +def float_val(pool: memhub.MemManager | t.CPtr, val: t.CDouble) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_FLOAT + v.float_val = val + return v + +def string_val(pool: memhub.MemManager | t.CPtr, val: t.CChar | t.CPtr) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_STRING + if val != None: + slen: t.CSizeT = string.strlen(val) + buf: t.CChar | t.CPtr = pool.alloc(slen + 1) + if buf != None: + string.memcpy(buf, val, slen + 1) + v.str_val = buf + return v + +def array(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_ARRAY + return v + +def object(pool: memhub.MemManager | t.CPtr) -> JsonValue | t.CPtr: + v: JsonValue | t.CPtr = JsonValue(pool) + if v == None: return None + v.vtype = JSON_OBJECT + return v + +# 向数组追加元素 +def array_append(pool: memhub.MemManager | t.CPtr, arr: JsonValue | t.CPtr, item: JsonValue | t.CPtr): + if arr == None or item == None: return + if arr.vtype != JSON_ARRAY: return + arr._append_child(item) + +# 向对象设置键值对 +def object_set(pool: memhub.MemManager | t.CPtr, obj: JsonValue | t.CPtr, key: t.CChar | t.CPtr, val: JsonValue | t.CPtr): + if obj == None or key == None or val == None: return + if obj.vtype != JSON_OBJECT: return + cur: JsonValue | t.CPtr = obj.child + while cur != None: + if cur.key != None: + if cur.key == key: + cur.vtype = val.vtype + cur.bool_val = val.bool_val + cur.int_val = val.int_val + cur.float_val = val.float_val + cur.str_val = val.str_val + cur.child = val.child + cur.child_count = val.child_count + return + cur = cur.next + klen: t.CSizeT = string.strlen(key) + kbuf: t.CChar | t.CPtr = pool.alloc(klen + 1) + if kbuf != None: + string.memcpy(kbuf, key, klen + 1) + val.key = kbuf + obj._append_child(val) + + +# ============================================================ +# 导入子模块 +# ============================================================ +from .__parser import parse +from .__writer import write diff --git a/includes/json/__parser.py b/includes/json/__parser.py new file mode 100644 index 0000000..2c9a300 --- /dev/null +++ b/includes/json/__parser.py @@ -0,0 +1,291 @@ +import t, c +from stdint import * +import memhub +import string +from . import ( + JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, + JSON_STRING, JSON_ARRAY, JSON_OBJECT, + null, bool_val, int_val, float_val, string_val, array, object, +) + + +class ParseState: + src: t.CChar | t.CPtr + pos: t.CSizeT + len: t.CSizeT + pool: memhub.MemManager | t.CPtr + + def __init__(self, src: t.CChar | t.CPtr, pool: memhub.MemManager | t.CPtr): + self.src = src + self.pos = 0 + self.len = string.strlen(src) if src != None else 0 + self.pool = pool + + +def _peek(s: ParseState | t.CPtr) -> t.CChar: + if s.pos >= s.len: + return '\0' + return s.src[s.pos] + +def _advance(s: ParseState | t.CPtr) -> t.CChar: + if s.pos >= s.len: + return '\0' + ch: t.CChar = s.src[s.pos] + s.pos += 1 + return ch + +def _skip_whitespace(s: ParseState | t.CPtr): + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch == ' ' or ch == '\t' or ch == '\n' or ch == '\r': + s.pos += 1 + else: + break + +def _expect(s: ParseState | t.CPtr, ch: t.CChar) -> bool: + _skip_whitespace(s) + if s.pos >= s.len: + return False + if s.src[s.pos] != ch: + return False + s.pos += 1 + return True + + +def _parse_string(s: ParseState | t.CPtr) -> t.CChar | t.CPtr: + if _peek(s) != '"': + return None + _advance(s) + + start: t.CSizeT = s.pos + slen: t.CSizeT = 0 + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch == '"': + break + if ch == '\\': + s.pos += 1 + slen += 1 + slen += 1 + s.pos += 1 + + buf: t.CChar | t.CPtr = s.pool.alloc(slen + 1) + if buf == None: + return None + + s.pos = start + di: t.CSizeT = 0 + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch == '"': + s.pos += 1 + break + if ch == '\\': + s.pos += 1 + if s.pos < s.len: + esc: t.CChar = s.src[s.pos] + if esc == 'n': + buf[di] = '\n' + elif esc == 't': + buf[di] = '\t' + elif esc == 'r': + buf[di] = '\r' + elif esc == '\\': + buf[di] = '\\' + elif esc == '"': + buf[di] = '"' + elif esc == '/': + buf[di] = '/' + else: + buf[di] = esc + di += 1 + s.pos += 1 + continue + buf[di] = ch + di += 1 + s.pos += 1 + + buf[di] = '\0' + return buf + + +def _parse_number(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: + _skip_whitespace(s) + is_float: bool = False + start: t.CSizeT = s.pos + + if s.pos < s.len and s.src[s.pos] == '-': + s.pos += 1 + + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch >= '0' and ch <= '9': + s.pos += 1 + else: + break + + if s.pos < s.len and s.src[s.pos] == '.': + is_float = True + s.pos += 1 + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch >= '0' and ch <= '9': + s.pos += 1 + else: + break + + if s.pos < s.len and (s.src[s.pos] == 'e' or s.src[s.pos] == 'E'): + is_float = True + s.pos += 1 + if s.pos < s.len and (s.src[s.pos] == '+' or s.src[s.pos] == '-'): + s.pos += 1 + while s.pos < s.len: + ch: t.CChar = s.src[s.pos] + if ch >= '0' and ch <= '9': + s.pos += 1 + else: + break + + num_len: t.CSizeT = s.pos - start + buf: t.CChar | t.CPtr = s.pool.alloc(num_len + 1) + if buf == None: + return null(s.pool) + + i: t.CSizeT = 0 + while i < num_len: + buf[i] = s.src[start + i] + i += 1 + buf[num_len] = '\0' + + if is_float: + val: t.CDouble = string.atof(buf) + return float_val(s.pool, val) + else: + ival: t.CInt64T = string.atoll(buf) + return int_val(s.pool, ival) + + +def _parse_value(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: + _skip_whitespace(s) + if s.pos >= s.len: + return null(s.pool) + + ch: t.CChar = _peek(s) + + if ch == '"': + sv: t.CChar | t.CPtr = _parse_string(s) + return string_val(s.pool, sv) + + if ch == '{': + return _parse_object(s) + + if ch == '[': + return _parse_array(s) + + if ch == '-' or (ch >= '0' and ch <= '9'): + return _parse_number(s) + + if ch == 't': + if s.pos + 3 < s.len: + if s.src[s.pos + 1] == 'r' and s.src[s.pos + 2] == 'u' and s.src[s.pos + 3] == 'e': + s.pos += 4 + return bool_val(s.pool, True) + return null(s.pool) + + if ch == 'f': + if s.pos + 4 < s.len: + if s.src[s.pos + 1] == 'a' and s.src[s.pos + 2] == 'l' and s.src[s.pos + 3] == 's' and s.src[s.pos + 4] == 'e': + s.pos += 5 + return bool_val(s.pool, False) + return null(s.pool) + + if ch == 'n': + if s.pos + 3 < s.len: + if s.src[s.pos + 1] == 'u' and s.src[s.pos + 2] == 'l' and s.src[s.pos + 3] == 'l': + s.pos += 4 + return null(s.pool) + return null(s.pool) + + return null(s.pool) + + +def _parse_array(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: + if not _expect(s, '['): + return null(s.pool) + + arr: JsonValue | t.CPtr = array(s.pool) + + _skip_whitespace(s) + if s.pos < s.len and _peek(s) == ']': + _advance(s) + return arr + + item: JsonValue | t.CPtr = _parse_value(s) + arr._append_child(item) + + while True: + _skip_whitespace(s) + if s.pos >= s.len: + break + if _peek(s) != ',': + break + _advance(s) + item = _parse_value(s) + arr._append_child(item) + + _expect(s, ']') + return arr + + +def _parse_object(s: ParseState | t.CPtr) -> JsonValue | t.CPtr: + if not _expect(s, '{'): + return null(s.pool) + + obj: JsonValue | t.CPtr = object(s.pool) + + _skip_whitespace(s) + if s.pos < s.len and _peek(s) == '}': + _advance(s) + return obj + + _skip_whitespace(s) + key: t.CChar | t.CPtr = _parse_string(s) + _expect(s, ':') + val: JsonValue | t.CPtr = _parse_value(s) + if key != None and val != None: + klen: t.CSizeT = string.strlen(key) + kbuf: t.CChar | t.CPtr = s.pool.alloc(klen + 1) + if kbuf != None: + string.memcpy(kbuf, key, klen + 1) + val.key = kbuf + obj._append_child(val) + + while True: + _skip_whitespace(s) + if s.pos >= s.len: + break + if _peek(s) != ',': + break + _advance(s) + _skip_whitespace(s) + key = _parse_string(s) + _expect(s, ':') + val = _parse_value(s) + if key != None and val != None: + klen2: t.CSizeT = string.strlen(key) + kbuf2: t.CChar | t.CPtr = s.pool.alloc(klen2 + 1) + if kbuf2 != None: + string.memcpy(kbuf2, key, klen2 + 1) + val.key = kbuf2 + obj._append_child(val) + + _expect(s, '}') + return obj + + +def parse(pool: memhub.MemManager | t.CPtr, src: t.CChar | t.CPtr) -> JsonValue | t.CPtr: + """解析 JSON 字符串，返回 JsonValue 树""" + if src == None: + return null(pool) + s: ParseState | t.CPtr = ParseState(src, pool) + return _parse_value(s) diff --git a/includes/json/__writer.py b/includes/json/__writer.py new file mode 100644 index 0000000..4b6435b --- /dev/null +++ b/includes/json/__writer.py @@ -0,0 +1,119 @@ +import t, c +from stdint import * +import memhub +import string +from stdio import snprintf +from . import ( + JsonValue, JSON_NULL, JSON_BOOL, JSON_INT, JSON_FLOAT, + JSON_STRING, JSON_ARRAY, JSON_OBJECT, +) + + +def _write_char(buf: t.CChar | t.CPtr, pos: t.CSizeT, ch: t.CChar) -> t.CSizeT: + buf[pos] = ch + return pos + 1 + +def _write_str(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: + if s == None: return pos + slen: t.CSizeT = string.strlen(s) + string.memcpy(t.CChar(t.CUInt64T(buf) + pos, t.CPtr), s, slen) + return pos + slen + +def _write_string_escaped(buf: t.CChar | t.CPtr, pos: t.CSizeT, s: t.CChar | t.CPtr) -> t.CSizeT: + if s == None: + pos = _write_str(buf, pos, "null") + return pos + pos = _write_char(buf, pos, '"') + i: t.CSizeT = 0 + slen: t.CSizeT = string.strlen(s) + while i < slen: + ch: t.CChar = s[i] + if ch == '"': + pos = _write_str(buf, pos, "\\\"") + elif ch == '\\': + pos = _write_str(buf, pos, "\\\\") + elif ch == '\n': + pos = _write_str(buf, pos, "\\n") + elif ch == '\r': + pos = _write_str(buf, pos, "\\r") + elif ch == '\t': + pos = _write_str(buf, pos, "\\t") + else: + pos = _write_char(buf, pos, ch) + i += 1 + pos = _write_char(buf, pos, '"') + return pos + + +def _write_value(buf: t.CChar | t.CPtr, pos: t.CSizeT, val: JsonValue | t.CPtr, pool: memhub.MemManager | t.CPtr) -> t.CSizeT: + if val == None: + pos = _write_str(buf, pos, "null") + return pos + + if val.vtype == JSON_NULL: + pos = _write_str(buf, pos, "null") + + elif val.vtype == JSON_BOOL: + if val.bool_val != 0: + pos = _write_str(buf, pos, "true") + else: + pos = _write_str(buf, pos, "false") + + elif val.vtype == JSON_INT: + tmp: t.CChar | t.CPtr = pool.alloc(32) + if tmp != None: + snprintf(tmp, 32, "%lld", val.int_val) + pos = _write_str(buf, pos, tmp) + + elif val.vtype == JSON_FLOAT: + tmp2: t.CChar | t.CPtr = pool.alloc(32) + if tmp2 != None: + snprintf(tmp2, 32, "%.6g", val.float_val) + pos = _write_str(buf, pos, tmp2) + + elif val.vtype == JSON_STRING: + pos = _write_string_escaped(buf, pos, val.str_val) + + elif val.vtype == JSON_ARRAY: + if val.child == None: + pos = _write_str(buf, pos, "[]") + return pos + pos = _write_char(buf, pos, '[') + cur: JsonValue | t.CPtr = val.child + first: bool = True + while cur != None: + if not first: + pos = _write_char(buf, pos, ',') + pos = _write_value(buf, pos, cur, pool) + first = False + cur = cur.next + pos = _write_char(buf, pos, ']') + + elif val.vtype == JSON_OBJECT: + if val.child == None: + pos = _write_str(buf, pos, "{}") + return pos + pos = _write_char(buf, pos, '{') + cur2: JsonValue | t.CPtr = val.child + first2: bool = True + while cur2 != None: + if not first2: + pos = _write_char(buf, pos, ',') + pos = _write_string_escaped(buf, pos, cur2.key) + pos = _write_char(buf, pos, ':') + pos = _write_value(buf, pos, cur2, pool) + first2 = False + cur2 = cur2.next + pos = _write_char(buf, pos, '}') + + return pos + + +def write(pool: memhub.MemManager | t.CPtr, val: JsonValue | t.CPtr, pretty: bool) -> t.CChar | t.CPtr: + """将 JsonValue 序列化为 JSON 字符串""" + buf: t.CChar | t.CPtr = pool.alloc(4096) + if buf == None: + return None + end_pos: t.CSizeT = _write_value(buf, 0, val, pool) + buf[end_pos] = '\0' + return buf diff --git a/includes/linkedlist.py b/includes/linkedlist.py new file mode 100644 index 0000000..5f6a990 --- /dev/null +++ b/includes/linkedlist.py @@ -0,0 +1,450 @@ +import t, c +from stdint import * + + +# ============================================================ +# LinkedNode: 非多态继承基类（@t.NoVTable） +# +# 子类继承此类即"启用"标准链表/树形结构能力： +# - 双向兄弟链表（next/prev）→ O(1) 删除 +# - 父子树结构（child/last_child/parent）→ O(1) 追加子节点 +# - 子节点计数（child_count） +# +# 字段展平嵌入子类，无 vtable 开销（对应 C++ 非多态继承）。 +# OOP 方法通过 _generate_inherited_method_wrappers 自动生成子类包装 +# （self bitcast 为父类指针后调用），子类可直接调用继承的方法。 +# +# 用法： +# class MyNode(LinkedNode): +# value: t.CInt +# root: MyNode = MyNode() +# child: MyNode = MyNode() +# root.append(child) # OOP 调用，self = root +# child.detach() # 从兄弟链摘除（分离自己） +# ============================================================ +@t.NoVTable +class LinkedNode: + next: "LinkedNode" | t.CPtr # 下一个兄弟节点 + prev: "LinkedNode" | t.CPtr # 上一个兄弟节点（双向链表，O(1) 删除） + child: "LinkedNode" | t.CPtr # 第一个子节点 + last_child: "LinkedNode" | t.CPtr # 最后一个子节点（O(1) 追加） + parent: "LinkedNode" | t.CPtr # 父节点 + child_count: t.CSizeT # 子节点数 + + def append(self, node: "LinkedNode" | t.CPtr): + """将 node 追加为 self 的最后一个子节点（O(1)）。 + + 若 node 已在某链表中，调用方应先 node.detach() 再 append。 + """ + if self is None or node is None: return + node.parent = self + node.next = None + node.prev = self.last_child + if self.child is None: + self.child = node + else: + self.last_child.next = node + self.last_child = node + self.child_count += 1 + + def prepend(self, node: "LinkedNode" | t.CPtr): + """将 node 插入为 self 的第一个子节点（O(1)）""" + if self is None or node is None: return + node.parent = self + node.prev = None + node.next = self.child + if self.child is not None: + self.child.prev = node + else: + self.last_child = node + self.child = node + self.child_count += 1 + + def insert_before(self, node: "LinkedNode" | t.CPtr, + new_node: "LinkedNode" | t.CPtr): + """在 node 之前插入 new_node（node 必须是 self 的子节点，O(1)）""" + if self is None or node is None or new_node is None: return + new_node.parent = self + new_node.prev = node.prev + new_node.next = node + if node.prev is not None: + node.prev.next = new_node + else: + self.child = new_node + node.prev = new_node + self.child_count += 1 + + def insert_after(self, node: "LinkedNode" | t.CPtr, + new_node: "LinkedNode" | t.CPtr): + """在 node 之后插入 new_node（node 必须是 self 的子节点，O(1)）""" + if self is None or node is None or new_node is None: return + new_node.parent = self + new_node.prev = node + new_node.next = node.next + if node.next is not None: + node.next.prev = new_node + else: + self.last_child = new_node + node.next = new_node + self.child_count += 1 + + def remove_child(self, node: "LinkedNode" | t.CPtr): + """从 self 的子链表中摘除 node（O(1)）。node 必须是 self 的直接子节点。""" + if self is None or node is None: return + # 修复 prev.next + if node.prev is not None: + node.prev.next = node.next + else: + self.child = node.next + # 修复 next.prev + if node.next is not None: + node.next.prev = node.prev + else: + self.last_child = node.prev + # 清除 node 的兄弟/父链接 + node.next = None + node.prev = None + node.parent = None + if self.child_count > 0: + self.child_count -= 1 + + def detach(self): + """将 self 从其所在兄弟链表中摘除（O(1)）。 + + 调用后 self 的 next/prev/parent 被清空，child 链不受影响。 + """ + if self is None: return + # 修复 prev.next + if self.prev is not None: + self.prev.next = self.next + elif self.parent is not None: + self.parent.child = self.next + # 修复 next.prev + if self.next is not None: + self.next.prev = self.prev + elif self.parent is not None: + self.parent.last_child = self.prev + # 递减父节点计数 + if self.parent is not None: + if self.parent.child_count > 0: + self.parent.child_count -= 1 + # 清除 self 链接 + self.next = None + self.prev = None + self.parent = None + + def unlink(self): + """断开 self 的所有连接（兄弟 + 父子），不递归处理子节点。 + + 调用后 self 成为孤立节点，但其 child 链仍指向原子节点 + （子节点的 parent 仍指向 self）。若需完全隔离，调用方应 + 先逐个 self.remove_child 子节点。 + """ + if self is None: return + # 先从兄弟链摘除 + self.detach() + # 再断开父子链接 + self.child = None + self.last_child = None + self.child_count = 0 + + def has_children(self) -> t.CInt: + """返回 1 若有子节点，否则 0""" + if self is None: return 0 + if self.child is not None: return 1 + return 0 + + def child_at(self, index: t.CSizeT) -> "LinkedNode" | t.CPtr: + """返回第 index 个子节点（O(n)，从 0 开始），越界返回 None""" + if self is None: return None + cur: "LinkedNode" | t.CPtr = self.child + i: t.CSizeT = 0 + while cur is not None: + if i == index: return cur + i += 1 + cur = cur.next + return None + + def count_children(self) -> t.CSizeT: + """遍历计算子节点数（O(n)），用于校验 child_count""" + if self is None: return 0 + n: t.CSizeT = 0 + cur: "LinkedNode" | t.CPtr = self.child + while cur is not None: + n += 1 + cur = cur.next + return n + + def first_sibling(self) -> "LinkedNode" | t.CPtr: + """返回兄弟链表的头节点（沿 prev 回溯）""" + if self is None: return None + cur: "LinkedNode" | t.CPtr = self + while cur.prev is not None: + cur = cur.prev + return cur + + def last_sibling(self) -> "LinkedNode" | t.CPtr: + """返回兄弟链表的尾节点（沿 next 前进）""" + if self is None: return None + cur: "LinkedNode" | t.CPtr = self + while cur.next is not None: + cur = cur.next + return cur + + +# ============================================================ +# SListNode: 单向链表节点（@t.NoVTable） +# +# 轻量级单向链表，仅 Next 指针，无 prev/parent/child。 +# 适用于队列/栈/简单链场景（如 llvmlite 的 Param/BasicBlock/Function/Line/Value）。 +# O(1) 追加需调用方维护 tail 指针（append_after）； +# O(n) 追加仅需 head（append）。 +# ============================================================ +@t.NoVTable +class SListNode: + Next: "SListNode" | t.CPtr # 下一个节点 + + def append(self, node: "SListNode" | t.CPtr) -> "SListNode" | t.CPtr: + """将 node 追加到链表末尾（O(n)），返回 self（head）。 + + 若 self 为 None，应直接使用 node 作为 head（调用方需自行处理）。 + """ + if self is None: return node + if node is None: return self + node.Next = None + cur: "SListNode" | t.CPtr = self + while cur.Next is not None: + cur = cur.Next + cur.Next = node + return self + + def append_after(self, node: "SListNode" | t.CPtr) -> "SListNode" | t.CPtr: + """将 node 追加到 self 之后（O(1)），返回新的 tail（即 node）。 + + 调用方需自行维护 head 指针。若 self 为 None，返回 node 作为首节点。 + """ + if node is None: return self + node.Next = None + if self is not None: + self.Next = node + return node + + def count(self) -> t.CSizeT: + """遍历计算链表长度（O(n)）""" + if self is None: return 0 + n: t.CSizeT = 0 + cur: "SListNode" | t.CPtr = self + while cur is not None: + n += 1 + cur = cur.Next + return n + + def at(self, index: t.CSizeT) -> "SListNode" | t.CPtr: + """返回第 index 个节点（O(n)，从 0 开始），越界返回 None""" + if self is None: return None + cur: "SListNode" | t.CPtr = self + i: t.CSizeT = 0 + while cur is not None: + if i == index: return cur + i += 1 + cur = cur.Next + return None + + def remove(self, node: "SListNode" | t.CPtr) -> "SListNode" | t.CPtr: + """从链表中移除 node（O(n)），返回新的 head（self）。 + + node 的 Next 被清空。若 node 是 head，返回 head.Next。 + """ + if self is None or node is None: return self + if self is node: + new_head: "SListNode" | t.CPtr = self.Next + self.Next = None + return new_head + cur: "SListNode" | t.CPtr = self + while cur.Next is not None: + if cur.Next is node: + cur.Next = node.Next + node.Next = None + break + cur = cur.Next + return self + + +# ============================================================ +# GSListNode[T]: 泛型单向链表节点（@t.NoVTable + PEP 695 泛型） +# +# 相比 SListNode（非泛型，Next: SListNode|CPtr），GSListNode[T] 的 Next +# 字段类型为 T|CPtr，特化后直接是具体节点类型，无需向下转型。 +# +# 递归泛型用法（子类继承自身特化版本）： +# class GNode(GSListNode[GNode]): +# value: t.CInt +# # 此时 GNode.Next: GNode|CPtr（强类型，无转型） +# +# 技术验证点： +# 1. 递归泛型 class GNode(GSListNode[GNode]) +# 2. @t.NoVTable + 泛型组合 +# 3. 泛型类作字段类型 list: GSList[GNode] +# 4. 泛型方法继承（GSList[T].append 在特化后可用） +# ============================================================ +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr + + +# ============================================================ +# GSList[T]: 泛型单向链表容器（@t.NoVTable + PEP 695 泛型） +# +# 持有 Head/Tail/Count，提供 O(1) append 和 O(n) at。 +# 节点类型 T 必须继承 GSListNode[T]（以获得 Next 字段）。 +# +# 用法： +# list: GSList[GNode] | t.CPtr = GSList[GNode]() +# list.append(node) # 方法调用，非全局函数 +# ============================================================ +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + + def __init__(self): + self.Head = None + self.Tail = None + self.Count = 0 + + def append(self, node: T): + """将 node 追加到链表末尾（O(1)）""" + if node is None: return + node.Next = None + if self.Head is None: + self.Head = node + else: + self.Tail.Next = node + self.Tail = node + self.Count += 1 + + def count(self) -> t.CSizeT: + """返回节点数（O(1)）""" + return self.Count + + def at(self, index: t.CSizeT) -> T | t.CPtr: + """返回第 index 个节点（O(n)，从 0 开始），越界返回 None""" + if self.Head is None: return None + cur: T | t.CPtr = self.Head + i: t.CSizeT = 0 + while cur is not None: + if i == index: return cur + i += 1 + cur = cur.Next + return None + + +# ============================================================ +# GTreeNode[T]: 泛型树节点（@t.NoVTable + PEP 695 泛型） +# +# 强类型版本的 LinkedNode，所有指针字段类型为 T|t.CPtr。 +# 提供： +# - 双向兄弟链表（Next/Prev）→ O(1) 删除 +# - 父子树结构（Child/LastChild/Parent）→ O(1) 追加子节点 +# - 子节点计数（Count） +# +# 递归泛型用法（子类继承自身特化版本）： +# @t.CVTable +# class AST(GTreeNode[AST]): +# def kind(self) -> t.CInt: +# return 0 +# # 此时 AST.Next/Prev/Child/LastChild/Parent 均为 AST|t.CPtr（强类型） +# +# 字段展平嵌入子类，无 vtable 开销（对应 C++ 非多态继承）。 +# 子类若为 @t.CVTable，则获得自己的 vtable（offset 0）， +# GTreeNode 字段展平在 vtable 之后。 +# +# OOP 方法通过 _generate_inherited_method_wrappers 自动生成子类包装 +# （self bitcast 为父类指针后调用），子类可直接调用继承的方法。 +# ============================================================ +@t.NoVTable +class GTreeNode[T]: + Next: T | t.CPtr # 下一个兄弟节点 + Prev: T | t.CPtr # 上一个兄弟节点（双向链表，O(1) 删除） + Child: T | t.CPtr # 第一个子节点 + LastChild: T | t.CPtr # 最后一个子节点（O(1) 追加） + Parent: T | t.CPtr # 父节点 + Count: t.CSizeT # 子节点数 + + def append(self, node: T | t.CPtr): + """将 node 追加为 self 的最后一个子节点（O(1)）。 + + 若 node 已在某链表中，调用方应先 node.detach() 再 append。 + """ + if self is None or node is None: return + node.Parent = self + node.Next = None + node.Prev = self.LastChild + if self.Child is None: + self.Child = node + else: + self.LastChild.Next = node + self.LastChild = node + self.Count += 1 + + def prepend(self, node: T | t.CPtr): + """将 node 插入为 self 的第一个子节点（O(1)）""" + if self is None or node is None: return + node.Parent = self + node.Prev = None + node.Next = self.Child + if self.Child is not None: + self.Child.Prev = node + else: + self.LastChild = node + self.Child = node + self.Count += 1 + + def detach(self): + """将 self 从其所在兄弟链表中摘除（O(1)）。 + + 调用后 self 的 Next/Prev/Parent 被清空，Child 链不受影响。 + """ + if self is None: return + if self.Prev is not None: + self.Prev.Next = self.Next + elif self.Parent is not None: + self.Parent.Child = self.Next + if self.Next is not None: + self.Next.Prev = self.Prev + elif self.Parent is not None: + self.Parent.LastChild = self.Prev + if self.Parent is not None: + if self.Parent.Count > 0: + self.Parent.Count -= 1 + self.Next = None + self.Prev = None + self.Parent = None + + def has_children(self) -> t.CInt: + """返回 1 若有子节点，否则 0""" + if self is None: return 0 + if self.Child is not None: return 1 + return 0 + + def child_at(self, index: t.CSizeT) -> T | t.CPtr: + """返回第 index 个子节点（O(n)，从 0 开始），越界返回 None""" + if self is None: return None + cur: T | t.CPtr = self.Child + i: t.CSizeT = 0 + while cur is not None: + if i == index: return cur + i += 1 + cur = cur.Next + return None + + def count_children(self) -> t.CSizeT: + """遍历计算子节点数（O(n)），用于校验 Count""" + if self is None: return 0 + n: t.CSizeT = 0 + cur: T | t.CPtr = self.Child + while cur is not None: + n += 1 + cur = cur.Next + return n diff --git a/includes/llvmlite/README.md b/includes/llvmlite/README.md new file mode 100644 index 0000000..6c0a380 --- /dev/null +++ b/includes/llvmlite/README.md @@ -0,0 +1,301 @@ +# llvmlite 设计方案 + +用 TransPyC 自身实现的轻量级 LLVM IR 文本生成库，替代 Python `llvmlite` 依赖，为自举（见 `wiki/13-bootstrapping.md` 路径 B）铺路。 + +## 1. 目标与定位 + +- **输入**：类型/值/指令的语义对象 +- **输出**：合法的 LLVM IR 文本（`.ll`），可直接喂给 `llc` +- **不做什么**：不解析 IR、不做优化、不绑定 LLVM C API（纯文本生成） +- **自举约束**：本库必须能用当前 TransPyC 编译器完整编译，不得依赖任何 Python 运行时 + +## 2. 核心设计决策 + +### 2.1 类型系统用 `t.REnum`（tagged union） + +LLVM 类型变体少、无通用字段、字段都是指针/小整数、操作主要通过 `match` 分派 —— 完全契合 REnum 的 tag+union 模型。 + +### 2.2 指令 opcode 用 `t.CEnum` + +指令类别（Binary/Unary/Cast/Memory/Terminator/Compare）与具体 opcode（Add/Sub/Load/Store/Br/Ret/...）用 `t.CEnum` 组织，类型安全且零开销。 + +### 2.3 IR 文本生成用 `viperlib.snprintf` + +所有 IR 文本通过 `viperlib.snprintf(buf, size, fmt, *args)` 格式化写入缓冲区，避免手写逐字符拼接。`snprintf` 已支持宽度/精度/符号/`%s`/`%d`/`%x` 等，足够覆盖 IR 语法。 + +### 2.4 对象分配用普通类 + `mpool` + +像 `mbuddy` 一样用普通类（无 `__new__`），由调用方注入 `mpool.MPool` 做批量分配。库内部不持有全局 `mbuddy`，遵循项目约定（`_mbuddy` 指针由 `os.py` 或调用方设置）。 + +--- + +## 3. 为什么 AST 不用 REnum，而 llvmlite 用？ + +用户质疑：REnum 是 tag+union（Rust enum），配合 `match` 模式匹配，按理很适合 AST。下面解释两者的适配性差异。 + +### 3.1 REnum 的硬限制（来自 `HandlesClassDef._EmitREnumLlvm`） + +1. **无通用字段共享**：每个 variant 独立声明字段，通用字段（如 `lineno`/`child`/`next`）必须在每 variant 重复声明。 +2. **共享最大变体尺寸**：布局为 `{ i32 __tag, }`，小变体浪费内存。 +3. **payload 仅位置绑定**：`match` 通过位置绑定 payload，不能直接 `node.child` 属性访问 payload 字段。 +4. **无方法**：variant 不能挂方法。 + +### 3.2 AST 不适合 REnum 的四个原因 + +| 维度 | AST 需求 | REnum 限制 | 冲突 | +|------|----------|------------|------| +| 变体数量 | 70+ 节点类型 | 共享最大变体尺寸 | 内存膨胀（每节点都占最大变体空间） | +| 通用字段 | lineno/child/next/parent 等 ~8 个字段全节点共享 | 无通用字段共享 | 要么每 variant 重复声明（冗长），要么放弃统一访问 | +| 访问模式 | 频繁直接 `node.child`/`node.next`/`node.lineno` | payload 仅位置绑定，需 match 分派 | 遍历代码爆炸（每次访问都要 match） | +| 字段复用 | `int_val`/`str_val`/`op` 等字段按 vtype 复用语义 | 每 variant 独立 payload | 无法复用 | + +AST 采用"胖节点"设计（所有节点共用一个 `AST` struct，用 `vtype` 区分）：一次 `malloc`、统一链表遍历（`child`/`next`）、字段复用。这与 REnum"每变体独立 payload"的哲学根本冲突。 + +### 3.3 llvmlite 的 LLVMType 适合 REnum 的原因 + +| 维度 | LLVMType 需求 | REnum 特性 | 契合度 | +|------|---------------|------------|--------| +| 变体数量 | ~10 个（Int/Ptr/Func/Array/Struct/Void/Float/Label/...） | 共享最大变体尺寸 | 浪费可接受（最大变体 ~24 字节） | +| 通用字段 | 无（每变体字段不同） | 无通用字段共享 | 完美匹配 | +| 访问模式 | 类型操作天然通过 match 分派（"是 Int 取 bits，是 Ptr 取 pointee"） | payload 位置绑定 + match | 天然契合 | +| 字段尺寸 | 都是指针(8B)或小整数(4B) | 共享最大变体 | 浪费小 | + +**结论**：REnum 是"少变体、无共性、match 分派"场景的最佳工具 —— llvmlite 类型系统正好如此，AST 正好相反。 + +--- + +## 4. 文件结构 + +``` +includes/llvmlite/ +├── __init__.py # 公共导出 + 便捷工厂函数 +├── __types.py # LLVMType (REnum) + 类型构造/打印 +├── __values.py # Value/Constant/SSA 值表示 +├── __module.py # Module 容器（函数列表 + 目标三元组 + 输出 .ll） +├── __function.py # Function + BasicBlock + 参数管理 +├── __builder.py # IRBuilder（指令发射 + SSA 命名 + 块跳转） +└── README.md # 本文件 +``` + +### 职责划分 + +- `__types.py`：`LLVMType` REnum 定义 + `TypePrint(buf, ty)` 将类型序列化为 IR 文本（如 `i32`/`i32*`/`{i32, i8*}`） +- `__values.py`：`Value` 表示一个 SSA 值（`%0`/`%result`/常量），含类型指针 + 名字 + 是否常量 +- `__module.py`：`Module` 持有函数链表 + 目标三元组 + 数据布局，`ModulePrint` 输出完整 `.ll` +- `__function.py`：`Function` 持有基本块链表 + 参数 + 返回类型；`BasicBlock` 持有指令文本缓冲 +- `__builder.py`：`IRBuilder` 游标式 API，`build_add`/`build_load`/`build_br`/... 发射指令到当前块 + +--- + +## 5. 类型系统设计（REnum） + +```python +import t +from stdint import * + +class LLVMType(t.REnum): + # 变体 Int：整数类型 + IntBits: t.CInt # 位宽 1/8/16/32/64/128 + + # 变体 Ptr：指针类型 + PtrPointee: LLVMType | t.CPtr # 指向的类型（自引用用指针） + + # 变体 Func：函数类型 + FuncRet: LLVMType | t.CPtr # 返回类型 + FuncParams: LLVMType | t.CPtr # 参数类型链表头 + FuncPCount: t.CInt # 参数数量 + + # 变体 Array：数组类型 + ArrayElem: LLVMType | t.CPtr # 元素类型 + ArrayCount: t.CInt # 元素数量 + + # 变体 Struct：结构体类型 + StructHead: LLVMType | t.CPtr # 字段类型链表头 + StructCount: t.CInt # 字段数量 + + # 变体 Float：浮点类型 + FloatBits: t.CInt # 16/32/64/128 + + # 变体 Void / Label / Metadata：无 payload + # （REnum 允许空变体，仅靠 __tag 区分） +``` + +### 类型工厂（`__init__.py` 导出） + +```python +def Int1() -> LLVMType | t.CPtr: ... # i1 +def Int8() -> LLVMType | t.CPtr: ... # i8 +def Int32() -> LLVMType | t.CPtr: ... # i32 +def Int64() -> LLVMType | t.CPtr: ... # i64 +def Ptr(pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: ... +def Func(ret: LLVMType | t.CPtr, params: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: ... +def Void() -> LLVMType | t.CPtr: ... +``` + +### 类型打印（match 分派） + +```python +def TypePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr): + match ty: + case LLVMType.Int: # 位置绑定 IntBits + viperlib.snprintf(buf, size, "i%d", ty.IntBits) + case LLVMType.Ptr: # 位置绑定 PtrPointee + TypePrint(inner, size2, ty.PtrPointee) + # 拼接 "*" + case LLVMType.Func: + # 打印 ret (params) + case LLVMType.Void: + string.strcpy(buf, "void") + ... +``` + +--- + +## 6. Value / SSA 设计 + +```python +class Value: + Ty: LLVMType | t.CPtr # 值的类型 + Name: t.CChar | t.CPtr # SSA 名（"%0"/"%result"）或常量文本 + IsConst: t.CInt # 1=常量字面量，0=SSA 临时值 + Next: Value | t.CPtr # 链表（函数内的值池） +``` + +`IRBuilder` 维护一个递增的 `%N` 计数器，每发射一条产生结果的指令就分配一个新名。 + +--- + +## 7. IRBuilder 设计 + +游标式 API，跟踪当前 `BasicBlock`，发射指令文本到块的缓冲区。 + +```python +class IRBuilder: + Func: Function | t.CPtr # 所属函数 + CurBlock: BasicBlock | t.CPtr # 当前插入点 + Counter: t.CInt # SSA 名计数器 + + def build_add(self, lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: + # 1. 分配 SSA 名 %N + # 2. snprintf(buf, size, " %%%d = add %s %s, %s\n", N, ty, lhs, rhs) + # 3. 追加到 CurBlock 缓冲 + # 4. 返回新 Value + + def build_load(self, ty: LLVMType | t.CPtr, ptr: Value | t.CPtr) -> Value | t.CPtr: ... + def build_store(self, val: Value | t.CPtr, ptr: Value | t.CPtr): ... + def build_br(self, target: BasicBlock | t.CPtr): ... + def build_cond_br(self, cond: Value | t.CPtr, then: BasicBlock | t.CPtr, else_: BasicBlock | t.CPtr): ... + def build_ret(self, val: Value | t.CPtr): ... + def build_alloca(self, ty: LLVMType | t.CPtr) -> Value | t.CPtr: ... + def build_gep(self, ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: ... + def build_call(self, callee: t.CChar | t.CPtr, args: Value | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Value | t.CPtr: ... +``` + +### 指令 opcode（CEnum） + +```python +class IROp(t.CEnum): + # Terminator + Ret: t.CEnum = 1 + Br: t.CEnum = 2 + CondBr: t.CEnum = 3 + Switch: t.CEnum = 4 + # Binary + Add: t.CEnum = 10 + Sub: t.CEnum = 11 + Mul: t.CEnum = 12 + SDiv: t.CEnum = 13 + # Memory + Load: t.CEnum = 20 + Store: t.CEnum = 21 + Alloca: t.CEnum = 22 + Gep: t.CEnum = 23 + # Cast + BitCast: t.CEnum = 30 + SExt: t.CEnum = 31 + Trunc: t.CEnum = 32 + # Compare + ICmp: t.CEnum = 40 + # Call + Call: t.CEnum = 50 +``` + +--- + +## 8. IR 文本生成（snprintf 用法） + +每条指令发射用 `viperlib.snprintf` 格式化到块缓冲区。示例： + +```python +import viperlib + +def _emit_add(block: BasicBlock | t.CPtr, name: t.CChar | t.CPtr, + ty_str: t.CChar | t.CPtr, lhs: t.CChar | t.CPtr, rhs: t.CChar | t.CPtr): + line: t.CChar | t.CPtr = mpool.alloc(128) # 行缓冲 + viperlib.snprintf(line, 128, " %s = add %s %s, %s\n", name, ty_str, lhs, rhs) + _block_append(block, line) +``` + +### Module 输出 + +```python +def ModulePrint(mod: Module | t.CPtr, out_path: t.CChar | t.CPtr): + # 1. 写头部：target triple / datalayout + # 2. 遍历函数，写 declare/define + # 3. 每函数：签名 + 基本块 + 指令 + # 用 snprintf 拼接，写到文件（viperio 或 w32/win32file） +``` + +--- + +## 9. 内存管理 + +- 库内所有对象从 `mpool.MPool` 分配（像 `ast` 库一样） +- `IRBuilder`/`Module`/`Function` 不持有 `mbuddy`，由调用方传入 `mpool` +- 文本缓冲区按行分配（每指令一行，128 字节起步），避免频繁 realloc + +--- + +## 10. 基础构建范围（第一阶段） + +最小可用集，能生成一个完整的小程序 IR： + +1. **类型**：`IntType`(1/8/16/32/64) + `PointerType` + `FunctionType` + `VoidType` + `StructType` +2. **值**：`Value` + 常量整数字面量 +3. **Module/Function/BasicBlock**：容器与链表 +4. **IRBuilder** 指令子集： + - `alloca` / `load` / `store` + - `add` / `sub` / `mul` / `sdiv` + - `icmp` (eq/ne/sgt/sge/slt/sle) + - `br` / `cond_br` / `ret` + - `call`（外部函数） + - `bitcast` / `sext` / `trunc` + - `getelementptr` +5. **ModulePrint**：输出完整 `.ll` 文件 + +### 后续扩展路线 + +- **第二阶段**：浮点指令、`phi` 节点、`switch`、字符串常量全局、属性（`nounwind`/`nocapture`） +- **第三阶段**：内联汇编、`atomic` 指令、`va_arg`、精确的 SSA 验证（支配边界、phi 完整性） +- **第四阶段**：对接 TransPyC 的 `LLVMCG`，替换 Python `llvmlite`，完成路径 B 自举 + +--- + +## 11. 测试计划 + +独立测试项目（`Test/LLvmLiteTest/`）： + +1. **类型打印测试**：构造各 `LLVMType`，`TypePrint` 输出，对比预期字符串 +2. **小函数生成**：用 `IRBuilder` 生成 `i32 add(i32 a, i32 b) { ret i32 %r }`，`ModulePrint` 写 `.ll`，用 `llc` 编译验证语法合法 +3. **match 分派测试**：构造各变体 `LLVMType`，`match` 正确绑定 payload +4. **snprintf 格式测试**：验证 `%d`/`%s`/`%x` 在 IR 文本中的正确性 + +--- + +## 12. 与现有架构的对接点 + +- **`lib/core/Codegen/LLVMCG.py`**：当前用 Python `llvmlite` 生成 IR，自举时改用本库 +- **`includes/viperlib.py`**：提供 `snprintf` / `sprintf` +- **`includes/mpool.py`**：提供 `MPool` 内存池 +- **`includes/string.py`**：提供 `strcpy`/`strlen`/`memcpy` +- **`includes/ast/`**：参考其"胖节点 + mpool + CEnum 常量"的组织方式（但类型系统用 REnum 而非胖节点） diff --git a/includes/llvmlite/__builder.py b/includes/llvmlite/__builder.py new file mode 100644 index 0000000..e5d580f --- /dev/null +++ b/includes/llvmlite/__builder.py @@ -0,0 +1,1220 @@ +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__types import LLVMType, TypePrint +from .__values import Value, new_value, SSAValue +from .__function import Function, BasicBlock, block_append_text, function_move_block_to_end, _ll_name_needs_quote + +# ============================================================ +# IRBuilder: 游标式 IR 指令发射器 +# +# 跟踪当前 BasicBlock，将指令文本追加到块的指令链表。 +# 维护递增的 SSA 名计数器（%0, %1, %2, ...）。 +# +# 用法: +# builder = new_builder(pool, func) +# builder.PositionAtEnd(entry_block) +# a = builder.build_alloca(i32_ty) +# b = builder.build_alloca(i32_ty) +# builder.build_store(ConstInt(42), a) +# r = builder.build_load(i32_ty, a) +# builder.build_ret(r) +# ============================================================ +class IRBuilder: + Pool: memhub.MemBuddy | t.CPtr # 内存池 + Func: Function | t.CPtr # 所属函数 + CurBlock: BasicBlock | t.CPtr # 当前插入点 + Counter: t.CInt # SSA 名计数器 + + +# ============================================================ +# 工厂函数 +# ============================================================ +# IRBuilder 硬编码大小: Pool(8) + Func(8) + CurBlock(8) + Counter(4) = 32 字节 +BUILDER_SIZE: t.CDefine = 32 + +def new_builder(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr) -> IRBuilder | t.CPtr: + """创建 IRBuilder，关联到函数 func""" + ptr: IRBuilder | t.CPtr = pool.alloc(BUILDER_SIZE) + if ptr is None: return None + string.memset(ptr, 0, BUILDER_SIZE) + ptr.Pool = pool + ptr.Func = func + ptr.CurBlock = None + ptr.Counter = 0 + return ptr + + +# ============================================================ +# 块定位 +# ============================================================ +def position_at_end(builder: IRBuilder | t.CPtr, block: BasicBlock | t.CPtr): + """将插入点设为 block 的末尾，并把 block 移到函数块链表末尾。 + + 移动 block 是为了保证 BB 文本输出顺序与控制流（position）顺序一致， + 避免 SSA 名在文本中非单调递增导致 llc 编译失败。 + """ + if builder is None: return + builder.CurBlock = block + if builder.Func is not None: + function_move_block_to_end(builder.Func, block) + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _alloc_ssa_name(builder: IRBuilder | t.CPtr) -> t.CChar | t.CPtr: + """分配一个 SSA 名字符串 "%N" 并递增计数器""" + pool: memhub.MemBuddy | t.CPtr = builder.Pool + buf: t.CChar | t.CPtr = pool.alloc(16) + if buf is None: return None + viperlib.snprintf(buf, 16, "%%%d", builder.Counter) + builder.Counter += 1 + return buf + + +def _type_str(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: + """获取类型的 IR 文本（从 pool 分配 512 字节缓冲区） + + 注意：复杂结构体类型（如 13 方法的虚表 {i8*,i8*,...}）可能超过 64 字节， + 必须使用足够大的缓冲区，否则 '}' 会被截断导致 IR 语法错误。 + """ + pool: memhub.MemBuddy | t.CPtr = builder.Pool + buf: t.CChar | t.CPtr = pool.alloc(512) + if buf is None: return None + buf[0] = '\0' + TypePrint(buf, 512, ty, pool) + return buf + + +def _emit(builder: IRBuilder | t.CPtr, text: t.CChar | t.CPtr): + """将指令文本追加到当前块""" + if builder is None or builder.CurBlock is None: return + block_append_text(builder.Pool, builder.CurBlock, text) + + +def _mark_terminated(builder: IRBuilder | t.CPtr): + """标记当前块已终止""" + if builder is None or builder.CurBlock is None: return + builder.CurBlock.IsTerminated = 1 + + +def builder_cur_block_is_terminated(builder: IRBuilder | t.CPtr) -> t.CInt: + """检查当前块的 IsTerminated 标志 + + 供其他模块绕过 stub 类型限制使用： + builder.CurBlock.IsTerminated 在 stub 类型中 BasicBlock 为 opaque， + 直接访问会被 TransPyC 静默跳过（不生成 GEP，不报错）， + 导致返回寄存器残留值，br 回边指令被错误跳过。 + """ + if builder is None or builder.CurBlock is None: + return 1 + return builder.CurBlock.IsTerminated + + +# ============================================================ +# 内存指令 +# ============================================================ +def build_alloca(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = alloca + + 返回类型为 ty* 的 SSA 值 + """ + if builder is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, ty) + line: t.CChar | t.CPtr = pool.alloc(128) + if line is None: return None + viperlib.snprintf(line, 128, "%s = alloca %s", name, ty_s) + _emit(builder, line) + + # 结果类型是 ty* + result_ty: LLVMType | t.CPtr = LLVMType.Ptr(ty) + result_ty_ptr: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if result_ty_ptr is not None: + string.memset(result_ty_ptr, 0, LLVMType.__sizeof__()) + c.DerefAs(result_ty_ptr, result_ty) + return SSAValue(pool, result_ty_ptr, name) + + +def build_load(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, + ptr: Value | t.CPtr) -> Value | t.CPtr: + """%N = load , """ + if builder is None or ptr is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, ty) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = load %s, %s %s", name, ty_s, ptr_ty_s, ptr.Name) + _emit(builder, line) + return SSAValue(pool, ty, name) + + +def build_store(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, ptr: Value | t.CPtr): + """store , """ + if builder is None or val is None or ptr is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + val_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return + viperlib.snprintf(line, 256, "store %s %s, %s %s", val_ty_s, val.Name, ptr_ty_s, ptr.Name) + _emit(builder, line) + + +# ============================================================ +# 二元运算指令 +# ============================================================ +def build_add(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = add , """ + return _build_binop(builder, "add", lhs, rhs) + + +def build_sub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = sub , """ + return _build_binop(builder, "sub", lhs, rhs) + + +def build_mul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = mul , """ + return _build_binop(builder, "mul", lhs, rhs) + + +def build_sdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = sdiv , """ + return _build_binop(builder, "sdiv", lhs, rhs) + + +def build_udiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = udiv , """ + return _build_binop(builder, "udiv", lhs, rhs) + + +def build_srem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = srem , """ + return _build_binop(builder, "srem", lhs, rhs) + + +def build_urem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = urem , """ + return _build_binop(builder, "urem", lhs, rhs) + + +def build_and(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = and , """ + return _build_binop(builder, "and", lhs, rhs) + + +def build_or(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = or , """ + return _build_binop(builder, "or", lhs, rhs) + + +def build_xor(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = xor , """ + return _build_binop(builder, "xor", lhs, rhs) + + +def build_shl(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = shl , """ + return _build_binop(builder, "shl", lhs, rhs) + + +def build_lshr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = lshr , """ + return _build_binop(builder, "lshr", lhs, rhs) + + +def build_ashr(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = ashr , """ + return _build_binop(builder, "ashr", lhs, rhs) + + +# ============================================================ +# 浮点二元运算指令 +# ============================================================ +def build_fadd(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = fadd , """ + return _build_binop(builder, "fadd", lhs, rhs) + + +def build_fsub(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = fsub , """ + return _build_binop(builder, "fsub", lhs, rhs) + + +def build_fmul(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = fmul , """ + return _build_binop(builder, "fmul", lhs, rhs) + + +def build_fdiv(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = fdiv , """ + return _build_binop(builder, "fdiv", lhs, rhs) + + +def build_frem(builder: IRBuilder | t.CPtr, lhs: Value | t.CPtr, + rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = frem , """ + return _build_binop(builder, "frem", lhs, rhs) + + +def build_fneg(builder: IRBuilder | t.CPtr, val: Value | t.CPtr) -> Value | t.CPtr: + """%N = fneg """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + line: t.CChar | t.CPtr = pool.alloc(128) + if line is None: return None + viperlib.snprintf(line, 128, "%s = fneg %s %s", name, ty_s, val.Name) + _emit(builder, line) + return SSAValue(pool, val.Ty, name) + + +def _build_binop(builder: IRBuilder | t.CPtr, op: t.CChar | t.CPtr, + lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: + """二元运算通用实现""" + if builder is None or lhs is None or rhs is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, lhs.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = %s %s %s, %s", name, op, ty_s, lhs.Name, rhs.Name) + _emit(builder, line) + return SSAValue(pool, lhs.Ty, name) + + +# ============================================================ +# 比较指令 +# ============================================================ +# ICMP 谓词常量 +ICMP_EQ: t.CDefine = 0 +ICMP_NE: t.CDefine = 1 +ICMP_SGT: t.CDefine = 2 +ICMP_SGE: t.CDefine = 3 +ICMP_SLT: t.CDefine = 4 +ICMP_SLE: t.CDefine = 5 +ICMP_UGT: t.CDefine = 6 +ICMP_UGE: t.CDefine = 7 +ICMP_ULT: t.CDefine = 8 +ICMP_ULE: t.CDefine = 9 + +# FCMP 谓词常量 +FCMP_FALSE: t.CDefine = 0 +FCMP_OEQ: t.CDefine = 1 +FCMP_OGT: t.CDefine = 2 +FCMP_OGE: t.CDefine = 3 +FCMP_OLT: t.CDefine = 4 +FCMP_OLE: t.CDefine = 5 +FCMP_ONE: t.CDefine = 6 +FCMP_ORD: t.CDefine = 7 +FCMP_UNO: t.CDefine = 8 +FCMP_UEQ: t.CDefine = 9 +FCMP_UGT: t.CDefine = 10 +FCMP_UGE: t.CDefine = 11 +FCMP_ULT: t.CDefine = 12 +FCMP_ULE: t.CDefine = 13 +FCMP_UNE: t.CDefine = 14 +FCMP_TRUE: t.CDefine = 15 + + +def _icmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: + if pred == ICMP_EQ: return "eq" + if pred == ICMP_NE: return "ne" + if pred == ICMP_SGT: return "sgt" + if pred == ICMP_SGE: return "sge" + if pred == ICMP_SLT: return "slt" + if pred == ICMP_SLE: return "sle" + if pred == ICMP_UGT: return "ugt" + if pred == ICMP_UGE: return "uge" + if pred == ICMP_ULT: return "ult" + if pred == ICMP_ULE: return "ule" + return "eq" + + +def _fcmp_pred_str(pred: t.CInt) -> t.CChar | t.CPtr: + if pred == FCMP_FALSE: return "false" + if pred == FCMP_OEQ: return "oeq" + if pred == FCMP_OGT: return "ogt" + if pred == FCMP_OGE: return "oge" + if pred == FCMP_OLT: return "olt" + if pred == FCMP_OLE: return "ole" + if pred == FCMP_ONE: return "one" + if pred == FCMP_ORD: return "ord" + if pred == FCMP_UNO: return "uno" + if pred == FCMP_UEQ: return "ueq" + if pred == FCMP_UGT: return "ugt" + if pred == FCMP_UGE: return "uge" + if pred == FCMP_ULT: return "ult" + if pred == FCMP_ULE: return "ule" + if pred == FCMP_UNE: return "une" + if pred == FCMP_TRUE: return "true" + return "oeq" + + +def build_icmp(builder: IRBuilder | t.CPtr, pred: t.CInt, + lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = icmp , + + 返回类型为 i1 的 SSA 值 + """ + if builder is None or lhs is None or rhs is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, lhs.Ty) + pred_s: t.CChar | t.CPtr = _icmp_pred_str(pred) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = icmp %s %s %s, %s", name, pred_s, ty_s, lhs.Name, rhs.Name) + _emit(builder, line) + + # 结果类型是 i1 + i1_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if i1_ty is not None: + string.memset(i1_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(i1_ty, LLVMType.Int(1)) + return SSAValue(pool, i1_ty, name) + + +def build_fcmp(builder: IRBuilder | t.CPtr, pred: t.CInt, + lhs: Value | t.CPtr, rhs: Value | t.CPtr) -> Value | t.CPtr: + """%N = fcmp , + + 返回类型为 i1 的 SSA 值 + """ + if builder is None or lhs is None or rhs is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, lhs.Ty) + pred_s: t.CChar | t.CPtr = _fcmp_pred_str(pred) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = fcmp %s %s %s, %s", name, pred_s, ty_s, lhs.Name, rhs.Name) + _emit(builder, line) + + # 结果类型是 i1 + i1_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if i1_ty is not None: + string.memset(i1_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(i1_ty, LLVMType.Int(1)) + return SSAValue(pool, i1_ty, name) + + +# ============================================================ +# 终止指令 +# ============================================================ +def build_br(builder: IRBuilder | t.CPtr, target: BasicBlock | t.CPtr): + """br label %""" + if builder is None or target is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + line: t.CChar | t.CPtr = pool.alloc(64) + if line is None: return + viperlib.snprintf(line, 64, "br label %%%s", target.Name) + _emit(builder, line) + _mark_terminated(builder) + + +def build_cond_br(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, + then_blk: BasicBlock | t.CPtr, else_blk: BasicBlock | t.CPtr): + """br i1 , label %, label %""" + if builder is None or cond is None or then_blk is None or else_blk is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + line: t.CChar | t.CPtr = pool.alloc(128) + if line is None: return + viperlib.snprintf(line, 128, "br i1 %s, label %%%s, label %%%s", + cond.Name, then_blk.Name, else_blk.Name) + _emit(builder, line) + _mark_terminated(builder) + + +def build_ret(builder: IRBuilder | t.CPtr, val: Value | t.CPtr): + """ret """ + if builder is None or val is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + line: t.CChar | t.CPtr = pool.alloc(128) + if line is None: return + viperlib.snprintf(line, 128, "ret %s %s", ty_s, val.Name) + _emit(builder, line) + _mark_terminated(builder) + + +def build_ret_void(builder: IRBuilder | t.CPtr): + """ret void""" + if builder is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + line: t.CChar | t.CPtr = pool.alloc(16) + if line is None: return + string.strcpy(line, "ret void") + _emit(builder, line) + _mark_terminated(builder) + + +# ============================================================ +# 调用指令 +# ============================================================ +def build_call(builder: IRBuilder | t.CPtr, callee: t.CChar | t.CPtr, + args: Value | t.CPtr, arg_count: t.CInt, + ret_ty: LLVMType | t.CPtr, + is_variadic: t.CInt = 0) -> Value | t.CPtr: + """%N = call @() + + args 是 Value 链表头，arg_count 是参数数量。 + 若 ret_ty 是 void，不分配 SSA 名（使用 call 而非 %N = call）。 + is_variadic=1 时发射函数类型签名（如 i32 (i8*, ...) @printf）。 + """ + if builder is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + + # 判断返回类型是否为 void + is_void: t.CInt = 0 + match ret_ty: + case LLVMType.Void(): + is_void = 1 + case _: + is_void = 0 + + name: t.CChar | t.CPtr = None + if is_void == 0: + name = _alloc_ssa_name(builder) + + ret_ty_s: t.CChar | t.CPtr = _type_str(builder, ret_ty) + + # 构建参数列表文本 + 变参函数类型签名 + args_buf: t.CChar | t.CPtr = pool.alloc(256) + if args_buf is None: return None + args_buf[0] = '\0' + func_ty_buf: t.CChar | t.CPtr = pool.alloc(256) + if func_ty_buf is None: return None + func_ty_buf[0] = '\0' + cur: Value | t.CPtr = args + first: t.CInt = 1 + while cur is not None: + if first == 0: + _append_cstr(args_buf, 256, ", ") + _append_cstr(func_ty_buf, 256, ", ") + arg_ty_s: t.CChar | t.CPtr = _type_str(builder, cur.Ty) + _append_cstr(args_buf, 256, arg_ty_s) + _append_cstr(args_buf, 256, " ") + if cur.Name is not None: + _append_cstr(args_buf, 256, cur.Name) + _append_cstr(func_ty_buf, 256, arg_ty_s) + cur = cur.Next + first = 0 + if is_variadic == 1: + if first == 0: + _append_cstr(func_ty_buf, 256, ", ") + _append_cstr(func_ty_buf, 256, "...") + + # 构造带引号的 callee 名字（SHA1 前缀名含 '.' 或以数字开头需要加引号） + callee_buf: t.CChar | t.CPtr = pool.alloc(128) + if callee_buf is None: return None + callee_buf[0] = '\0' + if _ll_name_needs_quote(callee) != 0: + _append_cstr(callee_buf, 128, "\"") + _append_cstr(callee_buf, 128, callee) + _append_cstr(callee_buf, 128, "\"") + else: + _append_cstr(callee_buf, 128, callee) + + line: t.CChar | t.CPtr = pool.alloc(512) + if line is None: return None + if is_variadic == 1: + if is_void == 1: + viperlib.snprintf(line, 512, "call %s (%s) @%s(%s)", ret_ty_s, func_ty_buf, callee_buf, args_buf) + else: + viperlib.snprintf(line, 512, "%s = call %s (%s) @%s(%s)", name, ret_ty_s, func_ty_buf, callee_buf, args_buf) + else: + if is_void == 1: + viperlib.snprintf(line, 512, "call %s @%s(%s)", ret_ty_s, callee_buf, args_buf) + else: + viperlib.snprintf(line, 512, "%s = call %s @%s(%s)", name, ret_ty_s, callee_buf, args_buf) + _emit(builder, line) + + if is_void == 1: + return None + return SSAValue(pool, ret_ty, name) + + +# ============================================================ +# 间接调用指令: call %callee() +# +# callee_val 是函数指针 SSA 值（必须已是函数指针类型，如 i32(i8*)*） +# ============================================================ +def build_call_indirect(builder: IRBuilder | t.CPtr, callee_val: Value | t.CPtr, + args: Value | t.CPtr, arg_count: t.CInt, + ret_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = call %callee() + + 间接调用：callee_val 是函数指针 SSA 值。 + """ + if builder is None or callee_val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + + is_void: t.CInt = 0 + match ret_ty: + case LLVMType.Void(): + is_void = 1 + case _: + is_void = 0 + + name: t.CChar | t.CPtr = None + if is_void == 0: + name = _alloc_ssa_name(builder) + + ret_ty_s: t.CChar | t.CPtr = _type_str(builder, ret_ty) + + # 构建参数列表文本 + args_buf: t.CChar | t.CPtr = pool.alloc(256) + if args_buf is None: return None + args_buf[0] = '\0' + cur: Value | t.CPtr = args + first: t.CInt = 1 + while cur is not None: + if first == 0: + _append_cstr(args_buf, 256, ", ") + arg_ty_s: t.CChar | t.CPtr = _type_str(builder, cur.Ty) + _append_cstr(args_buf, 256, arg_ty_s) + _append_cstr(args_buf, 256, " ") + if cur.Name is not None: + _append_cstr(args_buf, 256, cur.Name) + cur = cur.Next + first = 0 + + line: t.CChar | t.CPtr = pool.alloc(512) + if line is None: return None + if is_void == 1: + viperlib.snprintf(line, 512, "call %s %s(%s)", ret_ty_s, callee_val.Name, args_buf) + else: + viperlib.snprintf(line, 512, "%s = call %s %s(%s)", name, ret_ty_s, callee_val.Name, args_buf) + _emit(builder, line) + + if is_void == 1: + return None + return SSAValue(pool, ret_ty, name) + + +# ============================================================ +# 类型转换指令 +# ============================================================ +def build_bitcast(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = bitcast to """ + if builder is None or val is None: + return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + if val.Name is None: + return None + viperlib.snprintf(line, 256, "%s = bitcast %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + ret: Value | t.CPtr = SSAValue(pool, to_ty, name) + return ret + + +def build_sext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = sext to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = sext %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_trunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = trunc to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = trunc %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_zext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = zext to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = zext %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_ptrtoint(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = ptrtoint to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = ptrtoint %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_inttoptr(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = inttoptr to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = inttoptr %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +# ============================================================ +# 浮点转换指令 +# ============================================================ +def build_fpext(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = fpext to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = fpext %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_fptrunc(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = fptrunc to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = fptrunc %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_fp2si(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = fptosi to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = fptosi %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_si2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = sitofp to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = sitofp %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_fp2ui(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = fptoui to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = fptoui %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +def build_ui2fp(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + to_ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """%N = uitofp to """ + if builder is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + from_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + to_ty_s: t.CChar | t.CPtr = _type_str(builder, to_ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = uitofp %s %s to %s", name, from_ty_s, val.Name, to_ty_s) + _emit(builder, line) + return SSAValue(pool, to_ty, name) + + +# ============================================================ +# GEP 指令 +# ============================================================ +def build_gep(builder: IRBuilder | t.CPtr, elem_ty: LLVMType | t.CPtr, + ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: + """%N = getelementptr , , """ + if builder is None or ptr is None or idx is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + elem_ty_s: t.CChar | t.CPtr = _type_str(builder, elem_ty) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + idx_ty_s: t.CChar | t.CPtr = _type_str(builder, idx.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = getelementptr %s, %s %s, %s %s", + name, elem_ty_s, ptr_ty_s, ptr.Name, idx_ty_s, idx.Name) + _emit(builder, line) + + # GEP 结果类型是 elem_ty* + result_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if result_ty is not None: + string.memset(result_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(result_ty, LLVMType.Ptr(elem_ty)) + return SSAValue(pool, result_ty, name) + + +# ============================================================ +# build_gep_array — 数组双索引 GEP: getelementptr [N x ty], [N x ty]* %ptr, i32 0, i32 %idx +# ============================================================ +def build_gep_array(builder: IRBuilder | t.CPtr, array_ty: LLVMType | t.CPtr, + elem_ty: LLVMType | t.CPtr, + ptr: Value | t.CPtr, idx: Value | t.CPtr) -> Value | t.CPtr: + """%N = getelementptr [N x elem_ty], [N x elem_ty]* %ptr, i32 0, i32 %idx + + 结果类型是 elem_ty* + """ + if builder is None or ptr is None or idx is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + array_ty_s: t.CChar | t.CPtr = _type_str(builder, array_ty) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + idx_ty_s: t.CChar | t.CPtr = _type_str(builder, idx.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = getelementptr %s, %s %s, i32 0, %s %s", + name, array_ty_s, ptr_ty_s, ptr.Name, idx_ty_s, idx.Name) + _emit(builder, line) + + # GEP 结果类型是 elem_ty* + result_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if result_ty is not None: + string.memset(result_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(result_ty, LLVMType.Ptr(elem_ty)) + return SSAValue(pool, result_ty, name) + + +# ============================================================ +# build_gep_struct — 结构体字段 GEP: getelementptr struct_ty, struct_ty* %ptr, i32 0, i32 %field_idx +# ============================================================ +def build_gep_struct(builder: IRBuilder | t.CPtr, struct_ty: LLVMType | t.CPtr, + field_ty: LLVMType | t.CPtr, + ptr: Value | t.CPtr, field_idx: t.CInt) -> Value | t.CPtr: + """%N = getelementptr struct_ty, struct_ty* %ptr, i32 0, i32 %field_idx + + 结果类型是 field_ty* + """ + if builder is None or ptr is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + struct_ty_s: t.CChar | t.CPtr = _type_str(builder, struct_ty) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + line: t.CChar | t.CPtr = pool.alloc(2048) + if line is None: return None + viperlib.snprintf(line, 2048, "%s = getelementptr %s, %s %s, i32 0, i32 %d", + name, struct_ty_s, ptr_ty_s, ptr.Name, field_idx) + _emit(builder, line) + + # GEP 结果类型是 field_ty* + result_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if result_ty is not None: + string.memset(result_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(result_ty, LLVMType.Ptr(field_ty)) + return SSAValue(pool, result_ty, name) + + +# ============================================================ +# PhiIncoming: phi 节点的入边 (value, block) 对 +# 继承 GSListNode[PhiIncoming] 获取强类型 Next +# ============================================================ +class PhiIncoming(GSListNode[PhiIncoming]): + Val: Value | t.CPtr + Block: BasicBlock | t.CPtr + + +# PhiIncoming 硬编码大小: Next(8) + Val(8) + Block(8) = 24 字节 +PHIINC_SIZE: t.CDefine = 24 + +def new_phi_incoming(pool: memhub.MemBuddy | t.CPtr, + val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> PhiIncoming | t.CPtr: + ptr: PhiIncoming | t.CPtr = pool.alloc(PHIINC_SIZE) + if ptr is None: return None + string.memset(ptr, 0, PHIINC_SIZE) + ptr.Val = val + ptr.Block = block + return ptr + + +# ============================================================ +# Phi 指令 +# ============================================================ +def build_phi(builder: IRBuilder | t.CPtr, ty: LLVMType | t.CPtr, + incoming_head: PhiIncoming | t.CPtr, count: t.CInt) -> Value | t.CPtr: + """%N = phi [val1, %blk1], [val2, %blk2], ... + + incoming_head 是 PhiIncoming 链表头，count 是入边数量。 + """ + if builder is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, ty) + + # 构建入边列表文本: [val, %blk], [val, %blk], ... + parts: t.CChar | t.CPtr = pool.alloc(512) + if parts is None: return None + parts[0] = '\0' + cur: PhiIncoming | t.CPtr = incoming_head + first: t.CInt = 1 + while cur is not None: + if first == 0: + _append_cstr(parts, 512, ", ") + _append_cstr(parts, 512, "[") + # 使用中间变量避免链式结构体成员访问被编译器丢弃 + inc_val: Value | t.CPtr = cur.Val + if inc_val is not None: + inc_val_name: t.CChar | t.CPtr = inc_val.Name + if inc_val_name is not None: + _append_cstr(parts, 512, inc_val_name) + _append_cstr(parts, 512, ", %") + inc_blk: BasicBlock | t.CPtr = cur.Block + if inc_blk is not None: + inc_blk_name: t.CChar | t.CPtr = inc_blk.Name + if inc_blk_name is not None: + _append_cstr(parts, 512, inc_blk_name) + _append_cstr(parts, 512, "]") + cur = cur.Next + first = 0 + + line: t.CChar | t.CPtr = pool.alloc(640) + if line is None: return None + viperlib.snprintf(line, 640, "%s = phi %s %s", name, ty_s, parts) + _emit(builder, line) + return SSAValue(pool, ty, name) + + +# ============================================================ +# SwitchCase: switch 指令的 case 分支 +# ============================================================ +class SwitchCase(GSListNode[SwitchCase]): + CaseVal: Value | t.CPtr + Block: BasicBlock | t.CPtr + + +# SwitchCase 硬编码大小: Next(8) + CaseVal(8) + Block(8) = 24 字节 +SWCASE_SIZE: t.CDefine = 24 + +def new_switch_case(pool: memhub.MemBuddy | t.CPtr, + case_val: Value | t.CPtr, block: BasicBlock | t.CPtr) -> SwitchCase | t.CPtr: + ptr: SwitchCase | t.CPtr = pool.alloc(SWCASE_SIZE) + if ptr is None: return None + string.memset(ptr, 0, SWCASE_SIZE) + ptr.CaseVal = case_val + ptr.Block = block + return ptr + + +# ============================================================ +# Switch 指令 +# ============================================================ +def build_switch(builder: IRBuilder | t.CPtr, val: Value | t.CPtr, + default_blk: BasicBlock | t.CPtr, + cases_head: SwitchCase | t.CPtr, case_count: t.CInt): + """switch , label % [, label % ...]""" + if builder is None or val is None or default_blk is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + + # 构建主体: switch , label %default + line: t.CChar | t.CPtr = pool.alloc(512) + if line is None: return + viperlib.snprintf(line, 512, "switch %s %s, label %%%s", + ty_s, val.Name, default_blk.Name) + _append_cstr(line, 512, " [") + _append_cstr(line, 512, "\n") + + # 遍历 case 列表 + cur: SwitchCase | t.CPtr = cases_head + case_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + while cur is not None: + case_line: t.CChar | t.CPtr = pool.alloc(128) + if case_line is not None: + # 使用中间变量避免链式结构体成员访问被编译器丢弃 + cv: Value | t.CPtr = cur.CaseVal + cb: BasicBlock | t.CPtr = cur.Block + if cv is not None and cb is not None: + cv_name: t.CChar | t.CPtr = cv.Name + cb_name: t.CChar | t.CPtr = cb.Name + if cv_name is not None and cb_name is not None: + viperlib.snprintf(case_line, 128, " %s %s, label %%%s\n", + case_ty_s, cv_name, cb_name) + _append_cstr(line, 512, case_line) + cur = cur.Next + + _append_cstr(line, 512, "]") + _emit(builder, line) + _mark_terminated(builder) + + +# ============================================================ +# Select 指令 +# ============================================================ +def build_select(builder: IRBuilder | t.CPtr, cond: Value | t.CPtr, + true_val: Value | t.CPtr, false_val: Value | t.CPtr) -> Value | t.CPtr: + """%N = select i1 , , """ + if builder is None or cond is None or true_val is None or false_val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ty_s: t.CChar | t.CPtr = _type_str(builder, true_val.Ty) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = select i1 %s, %s %s, %s %s", + name, cond.Name, ty_s, true_val.Name, ty_s, false_val.Name) + _emit(builder, line) + return SSAValue(pool, true_val.Ty, name) + + +# ============================================================ +# Unreachable 指令 +# ============================================================ +def build_unreachable(builder: IRBuilder | t.CPtr): + """unreachable""" + if builder is None: return + pool: memhub.MemBuddy | t.CPtr = builder.Pool + line: t.CChar | t.CPtr = pool.alloc(16) + if line is None: return + string.strcpy(line, "unreachable") + _emit(builder, line) + _mark_terminated(builder) + + +# ============================================================ +# 原子指令 (Phase 3) +# ============================================================ +# AtomicRMW 操作常量 +ATOMIC_XCHG: t.CDefine = 0 +ATOMIC_ADD: t.CDefine = 1 +ATOMIC_SUB: t.CDefine = 2 +ATOMIC_AND: t.CDefine = 3 +ATOMIC_NAND: t.CDefine = 4 +ATOMIC_OR: t.CDefine = 5 +ATOMIC_XOR: t.CDefine = 6 +ATOMIC_MAX: t.CDefine = 7 +ATOMIC_MIN: t.CDefine = 8 +ATOMIC_UMAX: t.CDefine = 9 +ATOMIC_UMIN: t.CDefine = 10 + +# 内存序常量 +ATOMIC_ORDER_NOTATOMIC: t.CDefine = 0 +ATOMIC_ORDER_UNORDERED: t.CDefine = 1 +ATOMIC_ORDER_MONOTONIC: t.CDefine = 2 +ATOMIC_ORDER_ACQUIRE: t.CDefine = 3 +ATOMIC_ORDER_RELEASE: t.CDefine = 4 +ATOMIC_ORDER_ACQ_REL: t.CDefine = 5 +ATOMIC_ORDER_SEQ_CST: t.CDefine = 6 + + +def _atomic_op_str(op: t.CInt) -> t.CChar | t.CPtr: + if op == ATOMIC_XCHG: return "xchg" + if op == ATOMIC_ADD: return "add" + if op == ATOMIC_SUB: return "sub" + if op == ATOMIC_AND: return "and" + if op == ATOMIC_NAND: return "nand" + if op == ATOMIC_OR: return "or" + if op == ATOMIC_XOR: return "xor" + if op == ATOMIC_MAX: return "max" + if op == ATOMIC_MIN: return "min" + if op == ATOMIC_UMAX: return "umax" + if op == ATOMIC_UMIN: return "umin" + return "xchg" + + +def _atomic_order_str(ord: t.CInt) -> t.CChar | t.CPtr: + if ord == ATOMIC_ORDER_UNORDERED: return "unordered" + if ord == ATOMIC_ORDER_MONOTONIC: return "monotonic" + if ord == ATOMIC_ORDER_ACQUIRE: return "acquire" + if ord == ATOMIC_ORDER_RELEASE: return "release" + if ord == ATOMIC_ORDER_ACQ_REL: return "acq_rel" + if ord == ATOMIC_ORDER_SEQ_CST: return "seq_cst" + return "monotonic" + + +def build_atomicrmw(builder: IRBuilder | t.CPtr, op: t.CInt, + ptr: Value | t.CPtr, val: Value | t.CPtr, + order: t.CInt) -> Value | t.CPtr: + """%N = atomicrmw , """ + if builder is None or ptr is None or val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + val_ty_s: t.CChar | t.CPtr = _type_str(builder, val.Ty) + op_s: t.CChar | t.CPtr = _atomic_op_str(op) + ord_s: t.CChar | t.CPtr = _atomic_order_str(order) + line: t.CChar | t.CPtr = pool.alloc(256) + if line is None: return None + viperlib.snprintf(line, 256, "%s = atomicrmw %s %s %s, %s %s %s", + name, op_s, ptr_ty_s, ptr.Name, val_ty_s, val.Name, ord_s) + _emit(builder, line) + return SSAValue(pool, val.Ty, name) + + +def build_cmpxchg(builder: IRBuilder | t.CPtr, ptr: Value | t.CPtr, + cmp_val: Value | t.CPtr, new_val: Value | t.CPtr, + succ_order: t.CInt, fail_order: t.CInt) -> Value | t.CPtr: + """%N = cmpxchg , , + + 返回 {, i1} 类型的 SSA 值（简化为返回 i64 表示的值对） + """ + if builder is None or ptr is None or cmp_val is None or new_val is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + name: t.CChar | t.CPtr = _alloc_ssa_name(builder) + ptr_ty_s: t.CChar | t.CPtr = _type_str(builder, ptr.Ty) + cmp_ty_s: t.CChar | t.CPtr = _type_str(builder, cmp_val.Ty) + new_ty_s: t.CChar | t.CPtr = _type_str(builder, new_val.Ty) + succ_s: t.CChar | t.CPtr = _atomic_order_str(succ_order) + fail_s: t.CChar | t.CPtr = _atomic_order_str(fail_order) + line: t.CChar | t.CPtr = pool.alloc(384) + if line is None: return None + viperlib.snprintf(line, 384, "%s = cmpxchg %s %s, %s %s, %s %s %s %s", + name, ptr_ty_s, ptr.Name, cmp_ty_s, cmp_val.Name, + new_ty_s, new_val.Name, succ_s, fail_s) + _emit(builder, line) + # 返回原始值的类型 + return SSAValue(pool, new_val.Ty, name) + + +# ============================================================ +# 内联汇编 (Phase 3) +# ============================================================ +def build_inline_asm(builder: IRBuilder | t.CPtr, ret_ty: LLVMType | t.CPtr, + asm_text: t.CChar | t.CPtr, constraints: t.CChar | t.CPtr, + has_side_effects: t.CInt) -> Value | t.CPtr: + """%N = call asm [sideeffect] "" ""() + + 简化的内联汇编调用，无参数版本。 + """ + if builder is None: return None + pool: memhub.MemBuddy | t.CPtr = builder.Pool + + # 判断返回类型是否为 void + is_void: t.CInt = 0 + match ret_ty: + case LLVMType.Void(): + is_void = 1 + case _: + is_void = 0 + + name: t.CChar | t.CPtr = None + if is_void == 0: + name = _alloc_ssa_name(builder) + + ret_ty_s: t.CChar | t.CPtr = _type_str(builder, ret_ty) + line: t.CChar | t.CPtr = pool.alloc(512) + if line is None: return None + + if has_side_effects != 0: + if is_void == 1: + viperlib.snprintf(line, 512, "call %s asm sideeffect \"%s\", \"%s\"()", + ret_ty_s, asm_text, constraints) + else: + viperlib.snprintf(line, 512, "%s = call %s asm sideeffect \"%s\", \"%s\"()", + name, ret_ty_s, asm_text, constraints) + else: + if is_void == 1: + viperlib.snprintf(line, 512, "call %s asm \"%s\", \"%s\"()", + ret_ty_s, asm_text, constraints) + else: + viperlib.snprintf(line, 512, "%s = call %s asm \"%s\", \"%s\"()", + name, ret_ty_s, asm_text, constraints) + _emit(builder, line) + + if is_void == 1: + return None + return SSAValue(pool, ret_ty, name) + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 C 字符串 src 追加到 dst 末尾""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' diff --git a/includes/llvmlite/__function.py b/includes/llvmlite/__function.py new file mode 100644 index 0000000..5db70fb --- /dev/null +++ b/includes/llvmlite/__function.py @@ -0,0 +1,456 @@ +import t, c +from stdint import * +import string +import viperlib +import memhub +import stdio +from linkedlist import GSListNode, GSList +from .__types import LLVMType, ParamNode, TypePrint + + +# ============================================================ +# Line: 指令文本行 +# +# 每条 LLVM IR 指令序列化为一行文本，通过链表组织。 +# 继承 GSListNode[Line] 获取强类型 Next: Line|CPtr +# ============================================================ +class Line(GSListNode[Line]): + Buf: t.CChar | t.CPtr # 行文本（NUL 结尾） + Len: t.CSizeT # 文本长度（不含 NUL） + + +# ============================================================ +# BasicBlock: 基本块 +# +# 持有名称 + 指令行链表 + 终止指令标志。 +# 继承 GSListNode[BasicBlock] 获取强类型 Next: BasicBlock|CPtr +# Lines 为 GSList[Line] 指针，提供 O(1) 追加。 +# ============================================================ +class BasicBlock(GSListNode[BasicBlock]): + Name: t.CChar | t.CPtr # 块名（如 "entry"/"if.then"） + Lines: GSList[Line] | t.CPtr # 指令行链表容器（Head/Tail/Count） + IsTerminated: t.CInt # 1=已有终止指令（br/ret） + + +# ============================================================ +# Param: 函数参数 +# +# 继承 GSListNode[Param] 获取强类型 Next: Param|CPtr +# Attrs 存储参数属性文本（如 "noundef"/"nocapture"/"readonly"），None=无属性 +# ============================================================ +class Param(GSListNode[Param]): + Ty: LLVMType | t.CPtr # 参数类型 + Name: t.CChar | t.CPtr # 参数名（如 "%0"/"%a"） + Attrs: t.CChar | t.CPtr # 参数属性文本（None=无属性） + + +# ============================================================ +# Function: 函数 +# +# 持有名称 + 返回类型 + 参数链表 + 基本块链表。 +# 继承 GSListNode[Function] 获取强类型 Next: Function|CPtr +# Params/Blocks 为 GSList 指针，提供 O(1) 追加。 +# Attrs 存储函数级属性文本（如 "nounwind"/"noinline"），None=无属性 +# ============================================================ +class Function(GSListNode[Function]): + Name: t.CChar | t.CPtr # 函数名 + RetTy: LLVMType | t.CPtr # 返回类型 + Params: GSList[Param] | t.CPtr # 参数链表容器 + Blocks: GSList[BasicBlock] | t.CPtr # 基本块链表容器 + IsDeclared: t.CInt # 1=仅声明(declare), 0=定义(define) + IsVarArg: t.CInt # 1=变参函数(...) + Attrs: t.CChar | t.CPtr # 函数属性文本（None=无属性） + + +# ============================================================ +# 工厂函数 +# ============================================================ +# Line 硬编码大小: Next(8) + Buf(8) + Len(8) = 24 字节 +LINE_SIZE: t.CDefine = 24 + +def new_line(pool: memhub.MemBuddy | t.CPtr, text: t.CChar | t.CPtr) -> Line | t.CPtr: + """创建一个指令行，复制 text 到新分配的缓冲区""" + ptr: Line | t.CPtr = pool.alloc(LINE_SIZE) + if ptr is None: return None + string.memset(ptr, 0, LINE_SIZE) + if text is not None: + slen: t.CSizeT = string.strlen(text) + buf: t.CChar | t.CPtr = pool.alloc(slen + 1) + if buf is not None: + string.strcpy(buf, text) + ptr.Buf = buf + ptr.Len = slen + return ptr + + +def _new_gslist(pool: memhub.MemBuddy | t.CPtr, list_size: t.CSizeT) -> t.CPtr: + """从 pool 分配并零初始化一个 GSList 容器（Head/Tail/Count 全零）""" + gsl: t.CPtr = pool.alloc(list_size) + if gsl is None: return None + string.memset(gsl, 0, list_size) + return gsl + + +# BasicBlock 硬编码大小: Next(8) + Name(8) + Lines(8) + IsTerminated(4) = 32 字节 +BLOCK_SIZE: t.CDefine = 32 + +def new_basic_block(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: + """创建一个基本块""" + ptr: BasicBlock | t.CPtr = pool.alloc(BLOCK_SIZE) + if ptr is None: return None + string.memset(ptr, 0, BLOCK_SIZE) + ptr.Name = name + ptr.Lines = _new_gslist(pool, 24) # GSList: Head+Tail+Count + ptr.IsTerminated = 0 + return ptr + + +# Param 硬编码大小: Next(8) + Ty(8) + Name(8) + Attrs(8) = 32 字节 +PARAM_SIZE: t.CDefine = 32 + +def new_param(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, + name: t.CChar | t.CPtr) -> Param | t.CPtr: + """创建一个函数参数""" + ptr: Param | t.CPtr = pool.alloc(PARAM_SIZE) + if ptr is None: return None + string.memset(ptr, 0, PARAM_SIZE) + ptr.Ty = ty + ptr.Name = name + ptr.Attrs = None + return ptr + + +# Function 硬编码大小: Next(8) + Name(8) + RetTy(8) + Params(8) + Blocks(8) + IsDeclared(4) + IsVarArg(4) + Attrs(8) = 56 字节 +FUNC_SIZE: t.CDefine = 56 + +def new_function(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr, + ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: + """创建一个函数""" + ptr: Function | t.CPtr = pool.alloc(FUNC_SIZE) + if ptr is None: return None + string.memset(ptr, 0, FUNC_SIZE) + ptr.Name = name + ptr.RetTy = ret_ty + ptr.Params = _new_gslist(pool, 24) # GSList: Head+Tail+Count + ptr.Blocks = _new_gslist(pool, 24) + ptr.IsDeclared = 0 + ptr.Attrs = None + return ptr + + +# ============================================================ +# 链表操作（委托 GSList.append） +# ============================================================ +def function_add_param(func: Function | t.CPtr, param: Param | t.CPtr): + """将参数追加到函数参数链表""" + if func is None or param is None: return + if func.Params is None: return + func.Params.append(param) + + +def function_add_block(func: Function | t.CPtr, block: BasicBlock | t.CPtr): + """将基本块追加到函数块链表""" + if func is None or block is None: return + if func.Blocks is None: return + func.Blocks.append(block) + + +def function_move_block_to_end(func: Function | t.CPtr, block: BasicBlock | t.CPtr): + """将 block 移动到函数块链表末尾（保证 BB 文本顺序与控制流顺序一致）。 + + 用于解决 SSA 名非单调问题：BB 按创建顺序排列，但指令按控制流顺序生成， + 导致 SSA 编号在文本输出时非单调递增，llc 报错。 + 在 position_at_end 时调用此函数，使 BB 顺序与 position 顺序一致。 + """ + if func is None or block is None: return + blocks: GSList | t.CPtr = func.Blocks + if blocks is None: return + + # 已经是 Tail，无需移动 + if blocks.Tail is block: + return + + # 从链表中移除 block + if blocks.Head is block: + # block 是 Head，直接后移 Head + blocks.Head = block.Next + else: + # 遍历找前驱 + prev: BasicBlock | t.CPtr = blocks.Head + while prev is not None and prev.Next is not block: + prev = prev.Next + if prev is None: + # block 不在链表中，直接 append + blocks.append(block) + return + prev.Next = block.Next + + # block.Next 由移除逻辑处理，重置为 None + block.Next = None + + # append 到末尾（不调用 blocks.append 以避免 Count 重复递增） + if blocks.Tail is not None: + blocks.Tail.Next = block + blocks.Tail = block + + +def block_append_line(block: BasicBlock | t.CPtr, line: Line | t.CPtr): + """将指令行追加到基本块""" + if block is None or line is None: return + if block.Lines is None: return + block.Lines.append(line) + + +def block_append_text(pool: memhub.MemBuddy | t.CPtr, block: BasicBlock | t.CPtr, + text: t.CChar | t.CPtr): + """创建指令行并追加到基本块""" + if block is None or text is None: return + line: Line | t.CPtr = new_line(pool, text) + if line is not None: + block_append_line(block, line) + + +# ============================================================ +# 属性设置 +# ============================================================ +def function_set_attrs(func: Function | t.CPtr, attrs: t.CChar | t.CPtr): + """设置函数级属性文本（如 "nounwind"/"noinline"）""" + if func is None: return + func.Attrs = attrs + + +def param_set_attrs(param: Param | t.CPtr, attrs: t.CChar | t.CPtr): + """设置参数属性文本（如 "noundef"/"nocapture"/"readonly"）""" + if param is None: return + param.Attrs = attrs + + +# ============================================================ +# _ll_name_needs_quote - 检查 LLVM 标识符是否需要加引号 +# +# LLVM IR 规范：标识符以数字开头或含特殊字符（如 '.'）时必须 +# 用双引号包围。SHA1 命名空间的函数名格式为 "sha1hex.funcname" +# （含 '.'），以 hex 字符开头（可能以数字开头），因此需要加引号。 +# 普通函数名（如 printf/malloc/main）不含特殊字符，不加引号。 +# ============================================================ +def _ll_name_needs_quote(name: str) -> t.CInt: + """检查 LLVM 标识符是否需要加引号（以数字开头或含 '.'） + + 例外: LLVM 内联函数名（如 llvm.memcpy/llvm.memset）以 'llvm.' 开头， + 虽含 '.'，但不能加引号——加引号后 LLVM 解析器不识别为内联函数， + 导致 declare/call 签名不匹配（Intrinsic called with incompatible signature）。 + """ + if name is None: + return 0 + c0: t.CChar = name[0] + if c0 >= '0' and c0 <= '9': + return 1 + name_len: t.CSizeT = string.strlen(name) + # 豁免 llvm.* 前缀的内联函数名 + if name_len >= 5: + if name[0] == 'l' and name[1] == 'l' and name[2] == 'v' and name[3] == 'm' and name[4] == '.': + return 0 + i: t.CSizeT = 0 + while i < name_len: + if name[i] == '.': + return 1 + i += 1 + return 0 + + +# ============================================================ +# FunctionPrint: 将函数序列化为 IR 文本 +# +# 格式: +# define @() { +# : +# +# +# : +# +# } +# ============================================================ +def FunctionPrint(buf: t.CChar | t.CPtr, size: t.CSizeT, + func: Function | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """将函数 func 序列化为 IR 文本写入 buf""" + if func is None or buf is None or size == 0: return + + ty_buf: t.CChar | t.CPtr = pool.alloc(128) + if ty_buf is None: return + ty_buf[0] = '\0' + + # define @() + if func.IsDeclared == 1: + _append_cstr(buf, size, "declare ") + else: + _append_cstr(buf, size, "define ") + ty_buf[0] = '\0' + TypePrint(ty_buf, 128, func.RetTy, pool) + _append_cstr(buf, size, ty_buf) + _append_cstr(buf, size, " @") + if func.Name is not None: + if _ll_name_needs_quote(func.Name) != 0: + _append_cstr(buf, size, "\"") + _append_cstr(buf, size, func.Name) + _append_cstr(buf, size, "\"") + else: + _append_cstr(buf, size, func.Name) + _append_cstr(buf, size, "(") + + # 参数列表（从 GSList[Param].Head 遍历） + cur: Param | t.CPtr = None + if func.Params is not None: + cur = func.Params.Head + first: t.CInt = 1 + while cur is not None: + if first == 0: + _append_cstr(buf, size, ", ") + # 参数属性（如 "noundef"/"nocapture"）在类型之前 + if cur.Attrs is not None: + _append_cstr(buf, size, cur.Attrs) + _append_cstr(buf, size, " ") + ty_buf[0] = '\0' + TypePrint(ty_buf, 128, cur.Ty, pool) + _append_cstr(buf, size, ty_buf) + # declare 声明不打印参数名 + if func.IsDeclared == 0 and cur.Name is not None: + _append_cstr(buf, size, " ") + # LLVM IR 要求参数名以 % 开头 + if cur.Name[0] != '%': + _append_cstr(buf, size, "%") + _append_cstr(buf, size, cur.Name) + cur = cur.Next + first = 0 + # 变参函数追加 ... + if func.IsVarArg == 1: + if func.Params is not None and func.Params.Head is not None: + _append_cstr(buf, size, ", ...") + else: + _append_cstr(buf, size, "...") + _append_cstr(buf, size, ")") + + # 函数级属性（如 "nounwind"/"noinline"）在 ) 之后 + if func.Attrs is not None: + _append_cstr(buf, size, " ") + _append_cstr(buf, size, func.Attrs) + + if func.IsDeclared == 1: + _append_cstr(buf, size, "\n") + return + + # 函数体 + _append_cstr(buf, size, " {\n") + + blk: BasicBlock | t.CPtr = None + if func.Blocks is not None: + blk = func.Blocks.Head + while blk is not None: + # 块标签 + if blk.Name is not None: + _append_cstr(buf, size, blk.Name) + _append_cstr(buf, size, ":\n") + # 指令行（从 GSList[Line].Head 遍历） + line: Line | t.CPtr = None + if blk.Lines is not None: + line = blk.Lines.Head + while line is not None: + if line.Buf is not None: + _append_cstr(buf, size, " ") + _append_cstr(buf, size, line.Buf) + _append_cstr(buf, size, "\n") + line = line.Next + blk = blk.Next + + _append_cstr(buf, size, "}\n") + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 C 字符串 src 追加到 dst 末尾""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' + + +# ============================================================ +# 字段访问器（供其他模块绕过 stub 类型限制使用） +# +# 跨模块编译时，使用方模块只有 Function/Param 的 stub 类型 +# （字段不足且字段类型可能错误），导致 TransPyC 静默跳过 +# 字段访问（不生成 GEP，不报错），字段值永远为 null。 +# +# 这些访问器在拥有完整类型定义的本模块内执行，能正确访问 +# 所有字段。其他模块通过函数调用获取字段值，绕过 stub 限制。 +# ============================================================ +def function_get_name(func: Function | t.CPtr) -> t.CChar | t.CPtr: + """获取函数名""" + if func is None: + return None + return func.Name + + +def function_get_ret_ty(func: Function | t.CPtr) -> LLVMType | t.CPtr: + """获取函数返回类型""" + if func is None: + return None + return func.RetTy + + +def function_get_params(func: Function | t.CPtr) -> GSList[Param] | t.CPtr: + """获取函数参数链表容器""" + if func is None: + return None + return func.Params + + +def function_get_param_head(func: Function | t.CPtr) -> Param | t.CPtr: + """获取函数参数链表头节点（一步到位，避免调用方访问 GSList.Head）""" + if func is None: + return None + if func.Params is None: + return None + return func.Params.Head + + +def function_get_next(func: Function | t.CPtr) -> Function | t.CPtr: + """获取链表下一个函数""" + if func is None: + return None + return func.Next + + +def function_is_declared(func: Function | t.CPtr) -> t.CInt: + """获取函数是否为声明（1=declare 仅声明, 0=define 定义）""" + if func is None: + return 0 + return func.IsDeclared + + +def param_get_name(param: Param | t.CPtr) -> t.CChar | t.CPtr: + """获取参数名""" + if param is None: + return None + return param.Name + + +def param_get_ty(param: Param | t.CPtr) -> LLVMType | t.CPtr: + """获取参数类型""" + if param is None: + return None + return param.Ty + + +def param_get_next(param: Param | t.CPtr) -> Param | t.CPtr: + """获取链表下一个参数""" + if param is None: + return None + return param.Next diff --git a/includes/llvmlite/__init__.py b/includes/llvmlite/__init__.py new file mode 100644 index 0000000..8c36883 --- /dev/null +++ b/includes/llvmlite/__init__.py @@ -0,0 +1,403 @@ +import t, c +from stdint import * +import memhub +import string +import viperlib + +# 导入子模块 +from .__types import ( + LLVMType, ParamNode, LLVMTypeCore, + Int1, Int8, Int16, Int32, Int64, Ptr, Func, Void, Array, Struct, + Half, Float, Double, FP128, Label, + new_param_node, param_list_append, + paramnode_get_ty, paramnode_get_next, paramnode_set_ty, paramnode_set_next, + TypePrint, PrintStructDefinition, get_struct_name, +) +from .__values import ( + Value, new_value, ConstInt, ConstNull, ConstZero, ConstFloat, SSAValue, ValuePrint, + value_get_ty, value_get_next, + value_set_ty, value_set_next, value_set_name, value_set_isconst, +) +from .__function import ( + Line, BasicBlock, Param, Function, + new_line, new_basic_block, new_param, new_function, + function_add_param, function_add_block, + block_append_line, block_append_text, + function_set_attrs, param_set_attrs, + function_get_name, function_get_ret_ty, function_get_params, + function_get_param_head, function_get_next, function_is_declared, + param_get_name, param_get_ty, param_get_next, + FunctionPrint, +) +from .__module import ( + LLVMModule, GlobalVariable, NamedTypeNode, new_module, new_global_variable, + module_add_function, module_add_global, module_add_named_type, + module_set_target, module_set_datalayout, + global_set_constant, global_set_linkage, global_set_unnamed_addr, global_set_section, + LLVMModulePrint, OUTPUT_FULL, OUTPUT_STUB, OUTPUT_TEXT, + module_ensure_opaque_for_type, +) +from .__builder import ( + IRBuilder, new_builder, position_at_end, + build_alloca, build_load, build_store, + build_add, build_sub, build_mul, build_sdiv, build_udiv, build_srem, build_urem, + build_and, build_or, build_xor, build_shl, build_lshr, build_ashr, + build_fadd, build_fsub, build_fmul, build_fdiv, build_frem, build_fneg, + build_icmp, build_fcmp, + build_br, build_cond_br, build_ret, build_ret_void, + build_call, build_call_indirect, build_bitcast, build_sext, build_trunc, build_zext, + build_ptrtoint, build_inttoptr, + build_fpext, build_fptrunc, build_fp2si, build_si2fp, build_fp2ui, build_ui2fp, + build_gep, build_gep_array, build_gep_struct, + build_phi, build_switch, build_select, build_unreachable, + build_atomicrmw, build_cmpxchg, build_inline_asm, + PhiIncoming, new_phi_incoming, SwitchCase, new_switch_case, + ICMP_EQ, ICMP_NE, ICMP_SGT, ICMP_SGE, ICMP_SLT, ICMP_SLE, + ICMP_UGT, ICMP_UGE, ICMP_ULT, ICMP_ULE, + FCMP_FALSE, FCMP_OEQ, FCMP_OGT, FCMP_OGE, FCMP_OLT, FCMP_OLE, + FCMP_ONE, FCMP_ORD, FCMP_UNO, FCMP_UEQ, FCMP_UGT, FCMP_UGE, + FCMP_ULT, FCMP_ULE, FCMP_UNE, FCMP_TRUE, + ATOMIC_XCHG, ATOMIC_ADD, ATOMIC_SUB, ATOMIC_AND, ATOMIC_NAND, + ATOMIC_OR, ATOMIC_XOR, ATOMIC_MAX, ATOMIC_MIN, ATOMIC_UMAX, ATOMIC_UMIN, + ATOMIC_ORDER_NOTATOMIC, ATOMIC_ORDER_UNORDERED, ATOMIC_ORDER_MONOTONIC, + ATOMIC_ORDER_ACQUIRE, ATOMIC_ORDER_RELEASE, ATOMIC_ORDER_ACQ_REL, + ATOMIC_ORDER_SEQ_CST, +) +from .__verify import ( + VerifyResult, NameEntry, + verify_function, verify_module, + VERIFY_OK, VERIFY_ERR_NO_TERMINATOR, VERIFY_ERR_DUPLICATE_DEF, + VERIFY_ERR_UNDEFINED_USE, VERIFY_ERR_PHI_NOT_FIRST, + _new_verify_result, +) + + +# ============================================================ +# 便捷工厂: 字符串复制 +# ============================================================ +def _str_dup(pool: memhub.MemBuddy | t.CPtr, src: t.CChar | t.CPtr) -> t.CChar | t.CPtr: + """从 pool 分配并复制一个 C 字符串""" + if src is None: return None + slen: t.CSizeT = string.strlen(src) + buf: t.CChar | t.CPtr = pool.alloc(slen + 1) + if buf is None: return None + string.strcpy(buf, src) + return buf + + +# ============================================================ +# 便捷工厂: 创建带名称的基本块 +# ============================================================ +def create_block(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, + name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: + """创建基本块并添加到函数""" + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + blk: BasicBlock | t.CPtr = new_basic_block(pool, name_dup) + if blk is not None: + function_add_block(func, blk) + return blk + + +# ============================================================ +# 便捷工厂: 创建整数常量值 +# ============================================================ +def const_int(pool: memhub.MemBuddy | t.CPtr, bits: t.CInt, + val: t.CInt64T) -> Value | t.CPtr: + """创建整数常量值（自动创建类型 + 名字文本） + + 参数: + bits: 位宽 (1/8/16/32/64) + val: 整数值 + """ + # 创建类型 + ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if ty is None: return None + string.memset(ty, 0, LLVMType.__sizeof__()) + c.DerefAs(ty, LLVMType.Int(bits)) + + # 创建名字文本 + name: t.CChar | t.CPtr = pool.alloc(32) + if name is None: return None + viperlib.snprintf(name, 32, "%lld", val) + + return ConstInt(pool, ty, val, name) + + +def const_int32(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: + """便捷: 创建 i32 整数常量""" + return const_int(pool, 32, val) + + +def const_int64(pool: memhub.MemBuddy | t.CPtr, val: t.CInt64T) -> Value | t.CPtr: + """便捷: 创建 i64 整数常量""" + return const_int(pool, 64, val) + + +# ============================================================ +# 便捷工厂: 创建函数并添加到模块 +# ============================================================ +def create_function(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, + name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: + """创建函数并添加到模块""" + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + func: Function | t.CPtr = new_function(pool, name_dup, ret_ty) + if func is not None: + module_add_function(mod, func) + return func + + +def add_param(pool: memhub.MemBuddy | t.CPtr, func: Function | t.CPtr, + ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Param | t.CPtr: + """创建参数并添加到函数""" + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + param: Param | t.CPtr = new_param(pool, ty, name_dup) + if param is not None: + function_add_param(func, param) + return param + + +# ============================================================ +# 便捷工厂: 创建函数声明（declare） +# ============================================================ +def create_declare(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, + name: t.CChar | t.CPtr, ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: + """创建函数声明（declare，无函数体）并添加到模块 + + 与 create_function 不同，声明不生成函数体，仅生成: + declare @() + """ + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + func: Function | t.CPtr = new_function(pool, name_dup, ret_ty) + if func is not None: + func.IsDeclared = 1 + module_add_function(mod, func) + return func + + +# ============================================================ +# 便捷工厂: 创建字符串常量全局变量 +# +# 生成 LLVM IR: +# @.str = private unnamed_addr constant [N x i8] c"contents\00" +# ============================================================ +def create_global_string(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, + name: t.CChar | t.CPtr, text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: + """创建字符串常量全局变量并添加到模块 + + 参数: + name: 全局变量名（如 ".str"） + text: 字符串内容（C 字符串，不含末尾 NUL，NUL 由本函数自动添加） + + 返回: GlobalVariable 指针，其类型为 [len+1 x i8] + """ + if text is None: return None + slen: t.CSizeT = string.strlen(text) + count: t.CInt = slen + 1 # 含末尾 NUL + + # 创建 i8 类型 + i8_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if i8_ty is None: return None + string.memset(i8_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(i8_ty, LLVMType.Int(8)) + + # 创建 [count x i8] 数组类型 + arr_ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if arr_ty is None: return None + string.memset(arr_ty, 0, LLVMType.__sizeof__()) + c.DerefAs(arr_ty, LLVMType.Array(i8_ty, count)) + + # 创建全局变量 + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + gv: GlobalVariable | t.CPtr = new_global_variable(pool, name_dup, arr_ty) + if gv is None: return None + + # 设置属性: private + unnamed_addr + constant + global_set_linkage(gv, "private") + global_set_unnamed_addr(gv, 1) + global_set_constant(gv, 1) + + # 构造初始化值文本: c"contents\00" + # LLVM IR 字符串字面量格式: c"" 其中非打印字符用 \xx 转义 + # 这里简单处理: 直接复制原始字节，末尾加 \00 + init_buf: t.CChar | t.CPtr = pool.alloc(slen + 8) # c"..." + \00 + " + if init_buf is None: return None + pos: t.CSizeT = 0 + init_buf[pos] = 'c' + pos += 1 + init_buf[pos] = '"' + pos += 1 + i: t.CSizeT = 0 + while i < slen: + init_buf[pos] = text[i] + pos += 1 + i += 1 + # 末尾 NUL 字节 + init_buf[pos] = '\\' + pos += 1 + init_buf[pos] = '0' + pos += 1 + init_buf[pos] = '0' + pos += 1 + init_buf[pos] = '"' + pos += 1 + init_buf[pos] = '\0' + gv.Initializer = init_buf + + module_add_global(mod, gv) + return gv + + +# ============================================================ +# 便捷工厂: 创建带初始化值的整数全局变量 +# ============================================================ +def create_global_int(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, + name: t.CChar | t.CPtr, bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: + """创建整数全局变量并添加到模块 + + 生成 LLVM IR: + @ = global i + """ + # 创建整数类型 + ty: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if ty is None: return None + string.memset(ty, 0, LLVMType.__sizeof__()) + c.DerefAs(ty, LLVMType.Int(bits)) + + name_dup: t.CChar | t.CPtr = _str_dup(pool, name) + gv: GlobalVariable | t.CPtr = new_global_variable(pool, name_dup, ty) + if gv is None: return None + + # 构造初始化值文本: "42" (类型由 _print_global 单独打印) + init_buf: t.CChar | t.CPtr = pool.alloc(48) + if init_buf is None: return None + viperlib.snprintf(init_buf, 48, "%lld", val) + gv.Initializer = init_buf + + module_add_global(mod, gv) + return gv + + +# ============================================================ +# LLVMModuleCore: 模块/函数/值创建核心，封装 pool 避免重复传参 +# +# 用法: +# mcore: LLVMModuleCore | t.CPtr = LLVMModuleCore(pool) +# mod: LLVMModule | t.CPtr = mcore.NewModule("test") +# func: Function | t.CPtr = mcore.CreateFunction(mod, "add", ty_i32) +# val: Value | t.CPtr = mcore.ConstInt32(42) +# ============================================================ +@t.NoVTable +class LLVMModuleCore: + """模块/函数/值创建核心，封装 pool 避免每次创建都传 pool""" + Pool: memhub.MemBuddy | t.CPtr + + def __init__(self, pool: memhub.MemBuddy | t.CPtr): + self.Pool = pool + + # ---- 模块操作 ---- + def NewModule(self, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: + return new_module(self.Pool, name) + + def CreateFunction(self, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, + ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: + return create_function(self.Pool, mod, name, ret_ty) + + def CreateDeclare(self, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, + ret_ty: LLVMType | t.CPtr) -> Function | t.CPtr: + """创建函数声明（declare，无函数体）""" + return create_declare(self.Pool, mod, name, ret_ty) + + def AddParam(self, func: Function | t.CPtr, ty: LLVMType | t.CPtr, + name: t.CChar | t.CPtr) -> Param | t.CPtr: + return add_param(self.Pool, func, ty, name) + + def FunctionSetAttrs(self, func: Function | t.CPtr, attrs: t.CChar | t.CPtr): + """设置函数级属性文本""" + function_set_attrs(func, attrs) + + def ParamSetAttrs(self, param: Param | t.CPtr, attrs: t.CChar | t.CPtr): + """设置参数属性文本""" + param_set_attrs(param, attrs) + + def CreateBlock(self, func: Function | t.CPtr, + name: t.CChar | t.CPtr) -> BasicBlock | t.CPtr: + return create_block(self.Pool, func, name) + + def NewBuilder(self, func: Function | t.CPtr) -> IRBuilder | t.CPtr: + return new_builder(self.Pool, func) + + # ---- 全局变量 ---- + def NewGlobal(self, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, + ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: + """创建全局变量并添加到模块""" + name_dup: t.CChar | t.CPtr = _str_dup(self.Pool, name) + gv: GlobalVariable | t.CPtr = new_global_variable(self.Pool, name_dup, ty) + if gv is not None: + module_add_global(mod, gv) + return gv + + def CreateGlobalString(self, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, + text: t.CChar | t.CPtr) -> GlobalVariable | t.CPtr: + """创建字符串常量全局变量""" + return create_global_string(self.Pool, mod, name, text) + + def CreateGlobalInt(self, mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr, + bits: t.CInt, val: t.CInt64T) -> GlobalVariable | t.CPtr: + """创建带初始化值的整数全局变量""" + return create_global_int(self.Pool, mod, name, bits, val) + + def GlobalSetConstant(self, gv: GlobalVariable | t.CPtr, is_const: t.CInt): + global_set_constant(gv, is_const) + + def GlobalSetLinkage(self, gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr): + global_set_linkage(gv, linkage) + + def GlobalSetUnnamedAddr(self, gv: GlobalVariable | t.CPtr, level: t.CInt): + global_set_unnamed_addr(gv, level) + + def GlobalSetSection(self, gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr): + global_set_section(gv, section) + + # ---- 值创建 ---- + def ConstInt(self, bits: t.CInt, val: t.CInt64T) -> Value | t.CPtr: + return const_int(self.Pool, bits, val) + + def ConstInt32(self, val: t.CInt64T) -> Value | t.CPtr: + return const_int32(self.Pool, val) + + def ConstInt64(self, val: t.CInt64T) -> Value | t.CPtr: + return const_int64(self.Pool, val) + + def ConstFloat(self, ty: LLVMType | t.CPtr, val: t.CDouble) -> Value | t.CPtr: + return ConstFloat(self.Pool, ty, val) + + def SSAValue(self, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: + return SSAValue(self.Pool, ty, name) + + def ConstNull(self, ty: LLVMType | t.CPtr, name: t.CChar | t.CPtr) -> Value | t.CPtr: + return ConstNull(self.Pool, ty, name) + + # ---- 打印 ---- + def PrintValue(self, buf: t.CChar | t.CPtr, size: t.CSizeT, val: Value | t.CPtr): + ValuePrint(buf, size, val, self.Pool) + + def PrintFunction(self, buf: t.CChar | t.CPtr, size: t.CSizeT, func: Function | t.CPtr): + FunctionPrint(buf, size, func, self.Pool) + + def PrintModule(self, buf: t.CChar | t.CPtr, size: t.CSizeT, mod: LLVMModule | t.CPtr): + LLVMModulePrint(buf, size, mod, self.Pool, OUTPUT_FULL) + + # ---- SSA 验证 ---- + def NewVerifyResult(self, buf_size: t.CSizeT) -> VerifyResult | t.CPtr: + """创建验证结果对象（buf_size 为错误消息缓冲区大小）""" + return _new_verify_result(self.Pool, buf_size) + + def VerifyFunction(self, func: Function | t.CPtr, + result: VerifyResult | t.CPtr) -> t.CInt: + """验证单个函数的 SSA 正确性，返回错误数（0=通过）""" + return verify_function(self.Pool, func, result) + + def VerifyModule(self, mod: LLVMModule | t.CPtr, + result: VerifyResult | t.CPtr) -> t.CInt: + """验证模块中所有函数的 SSA 正确性，返回错误数（0=通过）""" + return verify_module(self.Pool, mod, result) diff --git a/includes/llvmlite/__module.py b/includes/llvmlite/__module.py new file mode 100644 index 0000000..cc6157a --- /dev/null +++ b/includes/llvmlite/__module.py @@ -0,0 +1,504 @@ +import t, c +from stdint import * +import string +import stdio +import viperlib +import memhub +from linkedlist import GSList, GSListNode +from .__types import LLVMType, TypePrint, Array, PrintStructDefinition, get_struct_name, Struct +from .__function import Function, FunctionPrint, _ll_name_needs_quote, Param + + +# ============================================================ +# GlobalVariable: LLVM IR 全局变量 +# +# 持有名称 + 类型 + 初始化值 + 常量标志 + 链接类型。 +# 继承 GSListNode[GlobalVariable] 获取强类型 Next。 +# ============================================================ +@t.NoVTable +class GlobalVariable(GSListNode[GlobalVariable]): + Name: t.CChar | t.CPtr # 全局变量名（如 ".str" / "my_global"） + Ty: LLVMType | t.CPtr # 变量类型 + Initializer: t.CVoid | t.CPtr # 初始化值 Value*（None = 无初始化） + IsConstant: t.CInt # 1=constant, 0=variable + Linkage: t.CChar | t.CPtr # 链接类型（"private"/"internal"/"external"/...） + Section: t.CChar | t.CPtr # 段名（None = 默认） + UnnamedAddr: t.CInt # 1=unnamed_addr, 2=local_unnamed_addr, 0=none + + +# ============================================================ +# NamedTypeNode: 命名结构体类型链表节点 +# +# 持有 LLVMType* 指针（必须是命名 Struct 类型）。 +# 继承 GSListNode[NamedTypeNode] 获取强类型 Next。 +# ============================================================ +@t.NoVTable +class NamedTypeNode(GSListNode[NamedTypeNode]): + Ty: LLVMType | t.CPtr # 命名结构体类型指针 + + +# ============================================================ +# LLVMModule: LLVM IR 模块容器 +# +# 使用嵌入式 Head/Tail/Count 字段代替 GSList 指针， +# 避免泛型 GSList 的 __sizeof__ 返回 0 和 vtable 偏移不一致问题。 +# LLVMModulePrint 输出完整 .ll 文件内容。 +# 注意: 类名必须为 LLVMModule 而非 Module，避免与 ast.Module 命名冲突 +# ============================================================ +@t.NoVTable +class LLVMModule: + Name: t.CChar | t.CPtr # 模块名 + FuncHead: Function | t.CPtr # 函数链表头 + FuncTail: Function | t.CPtr # 函数链表尾 + FuncCount: t.CSizeT # 函数计数 + GlobalHead: GlobalVariable | t.CPtr # 全局变量链表头 + GlobalTail: GlobalVariable | t.CPtr # 全局变量链表尾 + GlobalCount: t.CSizeT # 全局变量计数 + TargetTriple: t.CChar | t.CPtr # 目标三元组（如 "x86_64-pc-linux-gnu"） + DataLayout: t.CChar | t.CPtr # 数据布局（如 "e-m:e-p270:32:32-...") + NamedTypeHead: NamedTypeNode | t.CPtr # 命名结构体类型链表头 + NamedTypeTail: NamedTypeNode | t.CPtr # 命名结构体类型链表尾 + NamedTypeCount: t.CSizeT # 命名结构体类型计数 + # 总计: 12 字段 * 8 = 96 字节（无 vtable） + + +# ============================================================ +# 工厂函数 +# ============================================================ +def new_module(pool: memhub.MemBuddy | t.CPtr, name: t.CChar | t.CPtr) -> LLVMModule | t.CPtr: + """创建一个模块""" + # 硬编码 96 字节：12 字段 * 8 字节（无 vtable，@t.NoVTable） + ptr: LLVMModule | t.CPtr = pool.alloc(96) + if ptr is None: return None + string.memset(ptr, 0, 96) + ptr.Name = name + ptr.FuncHead = None + ptr.FuncTail = None + ptr.FuncCount = 0 + ptr.GlobalHead = None + ptr.GlobalTail = None + ptr.GlobalCount = 0 + ptr.TargetTriple = None + ptr.DataLayout = None + ptr.NamedTypeHead = None + ptr.NamedTypeTail = None + ptr.NamedTypeCount = 0 + return ptr + + +# GlobalVariable 硬编码大小: Next(8) + Name(8) + Ty(8) + Initializer(8) + IsConstant(4) + Linkage(8) + Section(8) + UnnamedAddr(4) = 64 字节 +GV_SIZE: t.CDefine = 64 + +# NamedTypeNode 硬编码大小: Next(8) + Ty(8) = 16 字节 +NAMED_TYPE_NODE_SIZE: t.CDefine = 16 + +def new_global_variable(pool: memhub.MemBuddy | t.CPtr, + name: t.CChar | t.CPtr, ty: LLVMType | t.CPtr) -> GlobalVariable | t.CPtr: + """创建一个全局变量""" + ptr: GlobalVariable | t.CPtr = pool.alloc(GV_SIZE) + if ptr is None: return None + string.memset(ptr, 0, GV_SIZE) + ptr.Name = name + ptr.Ty = ty + ptr.Initializer = None + ptr.IsConstant = 0 + ptr.Linkage = "external" + ptr.Section = None + ptr.UnnamedAddr = 0 + return ptr + + +# ============================================================ +# 链表操作（嵌入式 Head/Tail/Count，O(1) 追加） +# ============================================================ +def module_add_function(mod: LLVMModule | t.CPtr, func: Function | t.CPtr): + """将函数追加到模块函数链表（O(1)）""" + if mod is None or func is None: return + func.Next = None + if mod.FuncHead is None: + mod.FuncHead = func + else: + mod.FuncTail.Next = func + mod.FuncTail = func + mod.FuncCount += 1 + + +def module_add_global(mod: LLVMModule | t.CPtr, gv: GlobalVariable | t.CPtr): + """将全局变量追加到模块全局变量链表（O(1)）""" + if mod is None or gv is None: return + gv.Next = None + if mod.GlobalHead is None: + mod.GlobalHead = gv + else: + mod.GlobalTail.Next = gv + mod.GlobalTail = gv + mod.GlobalCount += 1 + + +def module_set_target(mod: LLVMModule | t.CPtr, triple: t.CChar | t.CPtr): + """设置目标三元组""" + if mod is None: return + mod.TargetTriple = triple + + +def module_set_datalayout(mod: LLVMModule | t.CPtr, layout: t.CChar | t.CPtr): + """设置数据布局""" + if mod is None: return + mod.DataLayout = layout + + +# ============================================================ +# 命名结构体类型管理 +# ============================================================ +def module_add_named_type(mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, + ty: LLVMType | t.CPtr): + """将命名结构体类型添加到模块（用于输出 %"name" = type { ... } 定义行） + + 非命名结构体（Name 为 None 或非 Struct 类型）会被忽略。 + 按 Name 去重，相同名称的类型只保留第一个。 + """ + if mod is None or pool is None or ty is None: + return + name: t.CChar | t.CPtr = get_struct_name(ty) + if name is None: + return + # 去重检查 + cur: NamedTypeNode | t.CPtr = mod.NamedTypeHead + while cur is not None: + if cur.Ty is not None: + existing_name: t.CChar | t.CPtr = get_struct_name(cur.Ty) + if existing_name is not None: + if string.strcmp(existing_name, name) == 0: + return + cur = cur.Next + # 创建新节点并追加到链表尾部 + node: NamedTypeNode | t.CPtr = pool.alloc(NAMED_TYPE_NODE_SIZE) + if node is None: + return + string.memset(node, 0, NAMED_TYPE_NODE_SIZE) + node.Ty = ty + node.Next = None + if mod.NamedTypeHead is None: + mod.NamedTypeHead = node + else: + mod.NamedTypeTail.Next = node + mod.NamedTypeTail = node + mod.NamedTypeCount += 1 + + +# ============================================================ +# module_ensure_opaque_for_type - 为类型中的跨模块结构体引用添加 opaque 声明 +# +# 递归扫描 LLVMType 树，当遇到 Ptr(Struct(name)) 且 name 包含 '.'（SHA1 限定名）时， +# 如果该名称未在模块中定义，则添加 opaque 声明（%"name" = type opaque）。 +# 这解决了跨模块结构体引用导致的 llc "use of undefined type" 错误。 +# ============================================================ +def _is_type_defined_in_module(mod: LLVMModule | t.CPtr, name: t.CChar | t.CPtr) -> int: + """检查模块中是否已定义指定名称的结构体类型，返回 1=已定义 / 0=未定义""" + if mod is None or name is None: + return 0 + cur: NamedTypeNode | t.CPtr = mod.NamedTypeHead + while cur is not None: + if cur.Ty is not None: + existing_name: t.CChar | t.CPtr = get_struct_name(cur.Ty) + if existing_name is not None: + if string.strcmp(existing_name, name) == 0: + return 1 + cur = cur.Next + return 0 + + +def _scan_type_for_cross_module_refs(pool: memhub.MemBuddy | t.CPtr, + mod: LLVMModule | t.CPtr, + ty: LLVMType | t.CPtr): + """递归扫描类型树，为跨模块结构体引用添加 opaque 声明""" + if ty is None or mod is None or pool is None: + return + match ty: + case LLVMType.Ptr(pointee): + if pointee is not None: + match pointee: + case LLVMType.Struct(sfields, sfcount, sname): + if sname is not None: + # 检查是否为跨模块引用（名称包含 '.'） + if string.strchr(sname, 46) is not None: + # 检查是否已在模块中定义 + if _is_type_defined_in_module(mod, sname) == 0: + # 创建 opaque 类型并添加到模块 + opaque_ty: LLVMType | t.CPtr = Struct(pool, None, 0, sname) + if opaque_ty is not None: + module_add_named_type(mod, pool, opaque_ty) + case _: + _scan_type_for_cross_module_refs(pool, mod, pointee) + case LLVMType.Array(elem_ty, acount): + if elem_ty is not None: + _scan_type_for_cross_module_refs(pool, mod, elem_ty) + case _: + pass + + +def module_ensure_opaque_for_type(mod: LLVMModule | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + ty: LLVMType | t.CPtr): + """为类型中的跨模块结构体引用添加 opaque 声明 + + 当类型引用了 Ptr(Struct(name)) 且 name 包含 '.'（SHA1 限定名）时， + 如果该名称未在模块中定义，则添加 opaque 声明。 + """ + _scan_type_for_cross_module_refs(pool, mod, ty) + + +# ============================================================ +# 全局变量属性设置 +# ============================================================ +def global_set_constant(gv: GlobalVariable | t.CPtr, is_const: t.CInt): + """设置全局变量是否为常量""" + if gv is None: return + gv.IsConstant = is_const + + +def global_set_linkage(gv: GlobalVariable | t.CPtr, linkage: t.CChar | t.CPtr): + """设置链接类型 (private/internal/external/... )""" + if gv is None: return + gv.Linkage = linkage + + +def global_set_unnamed_addr(gv: GlobalVariable | t.CPtr, level: t.CInt): + """设置 unnamed_addr 属性 (0=none, 1=unnamed_addr, 2=local_unnamed_addr)""" + if gv is None: return + gv.UnnamedAddr = level + + +def global_set_section(gv: GlobalVariable | t.CPtr, section: t.CChar | t.CPtr): + """设置段名""" + if gv is None: return + gv.Section = section + + +# ============================================================ +# 输出模式常量 +# ============================================================ +OUTPUT_FULL: t.CDefine = 0 # 完整 IR（stub + text） +OUTPUT_STUB: t.CDefine = 1 # 仅声明（header + 类型定义 + declare + external global） +OUTPUT_TEXT: t.CDefine = 2 # 仅代码（define + global 带初值） + + +# ============================================================ +# LLVMModulePrint: 将模块序列化为 .ll 文本 +# +# mode: OUTPUT_FULL=完整, OUTPUT_STUB=仅声明, OUTPUT_TEXT=仅代码 +# ============================================================ +def LLVMModulePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, + mod: LLVMModule | t.CPtr, pool: memhub.MemBuddy | t.CPtr, + mode: t.CInt): + """将模块 mod 序列化为 .ll 文本写入 buf""" + if mod is None or buf is None or size == 0: return + + # 头部注释（stub 和 full 模式输出，text 模式不输出） + if mode == OUTPUT_FULL or mode == OUTPUT_STUB: + _append_cstr(buf, size, "; ModuleID = '") + if mod.Name is not None: + _append_cstr(buf, size, mod.Name) + _append_cstr(buf, size, "'\n") + + # 目标三元组 + if mod.TargetTriple is not None: + _append_cstr(buf, size, "target triple = \"") + _append_cstr(buf, size, mod.TargetTriple) + _append_cstr(buf, size, "\"\n") + + # 数据布局 + if mod.DataLayout is not None: + _append_cstr(buf, size, "target datalayout = \"") + _append_cstr(buf, size, mod.DataLayout) + _append_cstr(buf, size, "\"\n") + + # 空行 + _append_cstr(buf, size, "\n") + + # 命名结构体类型定义（stub 和 full 模式输出，text 模式不输出） + if mode == OUTPUT_FULL or mode == OUTPUT_STUB: + # 扫描所有函数签名，为跨模块结构体引用添加 opaque 定义 + # 确保 stub.ll 包含被 declare/define 引用的跨模块类型定义 + # 修复：alloc_buf 返回 viperio.Buf，但 Buf 类型未在 NamedTypeHead 中注册 + func_scan: Function | t.CPtr = mod.FuncHead + while func_scan is not None: + if func_scan.RetTy is not None: + module_ensure_opaque_for_type(mod, pool, func_scan.RetTy) + if func_scan.Params is not None: + param_scan: Param | t.CPtr = func_scan.Params.Head + while param_scan is not None: + if param_scan.Ty is not None: + module_ensure_opaque_for_type(mod, pool, param_scan.Ty) + param_scan = param_scan.Next + func_scan = func_scan.Next + nt: NamedTypeNode | t.CPtr = mod.NamedTypeHead + while nt is not None: + if nt.Ty is not None: + ty_buf: t.CChar | t.CPtr = pool.alloc(256) + if ty_buf is not None: + ty_buf[0] = '\0' + PrintStructDefinition(ty_buf, 256, nt.Ty, pool) + _append_cstr(buf, size, ty_buf) + _append_cstr(buf, size, "\n") + nt = nt.Next + + if mod.NamedTypeHead is not None: + _append_cstr(buf, size, "\n") + + # 全局变量 + gv: GlobalVariable | t.CPtr = mod.GlobalHead + if mode == OUTPUT_STUB: + # stub 模式：输出 external global（无初值） + while gv is not None: + _print_global_stub(buf, size, gv, pool) + _append_cstr(buf, size, "\n") + gv = gv.Next + else: + # full 和 text 模式：输出完整全局变量 + # text 模式跳过无初值的全局变量（纯声明） + gv_idx: t.CInt = 0 + while gv is not None: + if mode == OUTPUT_TEXT and gv.Initializer is None: + gv = gv.Next + gv_idx += 1 + continue + _print_global(buf, size, gv, pool) + _append_cstr(buf, size, "\n") + gv = gv.Next + gv_idx += 1 + + if mod.GlobalHead is not None and (mode == OUTPUT_FULL or mode == OUTPUT_STUB): + _append_cstr(buf, size, "\n") + elif mod.GlobalHead is not None and mode == OUTPUT_TEXT: + # text 模式下如果输出了全局变量，也加空行 + had_global: t.CInt = 0 + gv2: GlobalVariable | t.CPtr = mod.GlobalHead + while gv2 is not None: + if gv2.Initializer is not None: + had_global = 1 + break + gv2 = gv2.Next + if had_global != 0: + _append_cstr(buf, size, "\n") + + # 遍历函数 + func: Function | t.CPtr = mod.FuncHead + func_idx: t.CInt = 0 + while func is not None: + if mode == OUTPUT_STUB: + # stub 模式：仅输出 declare（IsDeclared==1） + if func.IsDeclared == 1: + FunctionPrint(buf, size, func, pool) + _append_cstr(buf, size, "\n") + elif mode == OUTPUT_TEXT: + # text 模式：仅输出 define（IsDeclared==0） + if func.IsDeclared == 0: + FunctionPrint(buf, size, func, pool) + _append_cstr(buf, size, "\n") + else: + # full 模式：输出所有函数 + FunctionPrint(buf, size, func, pool) + _append_cstr(buf, size, "\n") + func = func.Next + func_idx += 1 + + +# ============================================================ +# 全局变量打印 +# ============================================================ +def _print_global(buf: t.CChar | t.CPtr, size: t.CSizeT, + gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """打印单个全局变量: @ = [linkage] [unnamed_addr] [constant] """ + if gv is None: return + + _append_cstr(buf, size, "@") + if gv.Name is not None: + if _ll_name_needs_quote(gv.Name) != 0: + _append_cstr(buf, size, "\"") + _append_cstr(buf, size, gv.Name) + _append_cstr(buf, size, "\"") + else: + _append_cstr(buf, size, gv.Name) + _append_cstr(buf, size, " = ") + + # 链接类型（有初始值时不输出 external，因为 external + 初始值是非法 IR） + if gv.Linkage is not None: + if gv.Initializer is not None and string.strcmp(gv.Linkage, "external") == 0: + pass + else: + _append_cstr(buf, size, gv.Linkage) + _append_cstr(buf, size, " ") + + # unnamed_addr + if gv.UnnamedAddr == 1: + _append_cstr(buf, size, "unnamed_addr ") + elif gv.UnnamedAddr == 2: + _append_cstr(buf, size, "local_unnamed_addr ") + + # constant / global + if gv.IsConstant == 1: + _append_cstr(buf, size, "constant ") + else: + _append_cstr(buf, size, "global ") + + # 类型 + ty_buf: t.CChar | t.CPtr = pool.alloc(128) + if ty_buf is not None: + ty_buf[0] = '\0' + TypePrint(ty_buf, 128, gv.Ty, pool) + _append_cstr(buf, size, ty_buf) + + # 初始化值 + if gv.Initializer is not None: + _append_cstr(buf, size, " ") + _append_cstr(buf, size, gv.Initializer) + + # 段名 + if gv.Section is not None: + _append_cstr(buf, size, ", section \"") + _append_cstr(buf, size, gv.Section) + _append_cstr(buf, size, "\"") + + +# ============================================================ +# _print_global_stub - 打印全局变量的声明形式（external global，无初值） +# ============================================================ +def _print_global_stub(buf: t.CChar | t.CPtr, size: t.CSizeT, + gv: GlobalVariable | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """打印全局变量声明: @ = external global """ + if gv is None: return + + _append_cstr(buf, size, "@") + if gv.Name is not None: + if _ll_name_needs_quote(gv.Name) != 0: + _append_cstr(buf, size, "\"") + _append_cstr(buf, size, gv.Name) + _append_cstr(buf, size, "\"") + else: + _append_cstr(buf, size, gv.Name) + _append_cstr(buf, size, " = external global ") + + # 类型 + ty_buf: t.CChar | t.CPtr = pool.alloc(128) + if ty_buf is not None: + ty_buf[0] = '\0' + TypePrint(ty_buf, 128, gv.Ty, pool) + _append_cstr(buf, size, ty_buf) + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 C 字符串 src 追加到 dst 末尾""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' \ No newline at end of file diff --git a/includes/llvmlite/__types.py b/includes/llvmlite/__types.py new file mode 100644 index 0000000..3ac24d7 --- /dev/null +++ b/includes/llvmlite/__types.py @@ -0,0 +1,524 @@ +import t, c +from stdint import * +import viperlib +import string +import memhub +from linkedlist import GSListNode + + +# ============================================================ +# ParamNode: 函数参数类型链表节点 +# 继承 GSListNode[ParamNode] 获取强类型 Next: ParamNode|CPtr +# Ty 用 t.CVoid | t.CPtr 前向引用，使用时转型为 (LLVMType | t.CPtr) +# ============================================================ +class ParamNode(GSListNode[ParamNode]): + Ty: t.CVoid | t.CPtr # LLVMType*（前向引用，使用时转型） + + +# ============================================================ +# LLVMType: LLVM IR 类型系统（t.REnum tagged union） +# +# 变体少、无通用字段、操作通过 match 分派 —— 契合 REnum 模型。 +# 布局: { i32 __tag, } +# ============================================================ +class LLVMType(t.REnum): + # 整数类型: i1/i8/i16/i32/i64 + class Int: + Bits: t.CInt + + # 指针类型: pointee* + class Ptr: + Pointee: LLVMType | t.CPtr + + # 函数类型: ret (params) + class Func: + Ret: LLVMType | t.CPtr + Params: ParamNode | t.CPtr + PCount: t.CInt + + # 数组类型: [N x elem_ty] + class Array: + ElemTy: LLVMType | t.CPtr + Count: t.CInt + + # 结构体类型: {ty1, ty2, ...} 或 %"name" = type {ty1, ty2, ...} + # Fields 复用 ParamNode 链表（ParamNode.Ty 即字段类型） + # Name 非 None 时为命名结构体，TypePrint 输出 %"name" 而非 {...} + class Struct: + Fields: ParamNode | t.CPtr + FCount: t.CInt + Name: t.CChar | t.CPtr + + # 浮点类型: half/float/double/fp128 + class Float: + Bits: t.CInt # 16=half, 32=float, 64=double, 128=fp128 + + # 无值类型 + class Void: + pass + + # 标签类型 + class Label: + pass + + # 元数据类型 + class Metadata: + pass + + +# ============================================================ +# 内部分配辅助 +# ============================================================ +def _new_type(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """从 mpool 分配一个零初始化的 LLVMType""" + ptr: LLVMType | t.CPtr = pool.alloc(LLVMType.__sizeof__()) + if ptr: + string.memset(ptr, 0, LLVMType.__sizeof__()) + return ptr + + +# ============================================================ +# 类型工厂函数 +# ============================================================ +def Int1(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Int(1)) + return ptr + + +def Int8(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Int(8)) + return ptr + + +def Int16(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Int(16)) + return ptr + + +def Int32(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Int(32)) + return ptr + + +def Int64(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Int(64)) + return ptr + + +def Ptr(pool: memhub.MemBuddy | t.CPtr, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Ptr(pointee)) + return ptr + + +def Func(pool: memhub.MemBuddy | t.CPtr, ret: LLVMType | t.CPtr, + params: ParamNode | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Func(ret, params, count)) + return ptr + + +def Void(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Void()) + return ptr + + +def Array(pool: memhub.MemBuddy | t.CPtr, elem_ty: LLVMType | t.CPtr, + count: t.CInt) -> LLVMType | t.CPtr: + """创建数组类型 [count x elem_ty]""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Array(elem_ty, count)) + return ptr + + +def Struct(pool: memhub.MemBuddy | t.CPtr, fields: ParamNode | t.CPtr, + fcount: t.CInt, name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: + """创建结构体类型 {ty1, ty2, ...} + + fields 为 ParamNode 链表头，fcount 为字段数量。 + name 非 None 时为命名结构体（如 "sha1.ClassName"），TypePrint 输出 %"name"。 + name 为 None 时为匿名结构体，TypePrint 输出 {ty1, ty2, ...}。 + """ + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Struct(fields, fcount, name)) + return ptr + + +def Half(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """创建半精度浮点类型 half (16-bit)""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Float(16)) + return ptr + + +def Float(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """创建单精度浮点类型 float (32-bit)""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Float(32)) + return ptr + + +def Double(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """创建双精度浮点类型 double (64-bit)""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Float(64)) + return ptr + + +def FP128(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """创建四精度浮点类型 fp128 (128-bit)""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Float(128)) + return ptr + + +def Label(pool: memhub.MemBuddy | t.CPtr) -> LLVMType | t.CPtr: + """创建标签类型 label""" + ptr: LLVMType | t.CPtr = _new_type(pool) + if ptr is None: return None + c.DerefAs(ptr, LLVMType.Label()) + return ptr + + +# ============================================================ +# ParamNode 辅助 +# ============================================================ +# ParamNode 硬编码大小: Next(8) + Ty(8) = 16 字节 +PNODE_SIZE: t.CDefine = 16 + +def new_param_node(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: + """创建一个参数节点，Ty 设为给定类型指针""" + ptr: ParamNode | t.CPtr = pool.alloc(PNODE_SIZE) + if ptr is None: return None + string.memset(ptr, 0, PNODE_SIZE) + ptr.Ty = ty + ptr.Next = None + return ptr + + +def param_list_append(head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, + node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: + """将 node 追加到参数链表尾部，返回 head 和更新后的 tail""" + if node is None: return head + node.Next = None + if head is None: + return node + if tail is not None: + tail.Next = node + return node + + +# ============================================================ +# ParamNode 字段访问器（供其他模块绕过 stub 类型限制使用） +# +# 跨模块编译时，使用方模块只有 ParamNode 的 stub 类型， +# 导致 TransPyC 静默跳过字段访问。这些访问器在拥有完整类型定义的 +# 本模块内执行，能正确访问所有字段。 +# ============================================================ +def paramnode_get_ty(node: ParamNode | t.CPtr) -> LLVMType | t.CPtr: + """获取参数节点的类型""" + if node is None: + return None + return (LLVMType | t.CPtr)(node.Ty) + + +def paramnode_get_next(node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: + """获取链表下一个参数节点""" + if node is None: + return None + return node.Next + + +def paramnode_set_ty(node: ParamNode | t.CPtr, ty: LLVMType | t.CPtr): + """设置参数节点的类型""" + if node is None: + return + node.Ty = ty + + +def paramnode_set_next(node: ParamNode | t.CPtr, nxt: ParamNode | t.CPtr): + """设置链表下一个参数节点""" + if node is None: + return + node.Next = nxt + + +# ============================================================ +# LLVMTypeCore: 类型创建核心，封装 pool 避免重复传参 +# +# 用法: +# core: LLVMTypeCore | t.CPtr = LLVMTypeCore(pool) +# ty_i32: LLVMType | t.CPtr = core.Int32() +# ty_ptr: LLVMType | t.CPtr = core.Ptr(ty_i32) +# ty_void: LLVMType | t.CPtr = core.Void() +# ============================================================ +@t.NoVTable +class LLVMTypeCore: + """类型创建核心，封装 pool 避免每次创建类型都传 pool""" + Pool: memhub.MemBuddy | t.CPtr + + def __init__(self, pool: memhub.MemBuddy | t.CPtr): + self.Pool = pool + + def Int1(self) -> LLVMType | t.CPtr: + return Int1(self.Pool) + + def Int8(self) -> LLVMType | t.CPtr: + return Int8(self.Pool) + + def Int16(self) -> LLVMType | t.CPtr: + return Int16(self.Pool) + + def Int32(self) -> LLVMType | t.CPtr: + return Int32(self.Pool) + + def Int64(self) -> LLVMType | t.CPtr: + return Int64(self.Pool) + + def Ptr(self, pointee: LLVMType | t.CPtr) -> LLVMType | t.CPtr: + return Ptr(self.Pool, pointee) + + def Func(self, ret: LLVMType | t.CPtr, params: ParamNode | t.CPtr, + count: t.CInt) -> LLVMType | t.CPtr: + return Func(self.Pool, ret, params, count) + + def Array(self, elem_ty: LLVMType | t.CPtr, count: t.CInt) -> LLVMType | t.CPtr: + return Array(self.Pool, elem_ty, count) + + def Struct(self, fields: ParamNode | t.CPtr, fcount: t.CInt, + name: t.CChar | t.CPtr) -> LLVMType | t.CPtr: + return Struct(self.Pool, fields, fcount, name) + + def Void(self) -> LLVMType | t.CPtr: + return Void(self.Pool) + + def Half(self) -> LLVMType | t.CPtr: + return Half(self.Pool) + + def Float(self) -> LLVMType | t.CPtr: + return Float(self.Pool) + + def Double(self) -> LLVMType | t.CPtr: + return Double(self.Pool) + + def FP128(self) -> LLVMType | t.CPtr: + return FP128(self.Pool) + + def Label(self) -> LLVMType | t.CPtr: + return Label(self.Pool) + + def NewParamNode(self, ty: LLVMType | t.CPtr) -> ParamNode | t.CPtr: + return new_param_node(self.Pool, ty) + + # ---- 链表辅助 ---- + def ParamListAppend(self, head: ParamNode | t.CPtr, tail: ParamNode | t.CPtr, + node: ParamNode | t.CPtr) -> ParamNode | t.CPtr: + """将 node 追加到参数/字段链表，返回新的 tail""" + return param_list_append(head, tail, node) + + # ---- 打印 ---- + def PrintType(self, buf: t.CChar | t.CPtr, size: t.CSizeT, ty: LLVMType | t.CPtr): + TypePrint(buf, size, ty, self.Pool) + + +# ============================================================ +# TypePrint: 将 LLVMType 序列化为 IR 文本 +# +# 示例: +# Int(32) -> "i32" +# Int(8) -> "i8" +# Ptr(Int(32)) -> "i32*" +# Void -> "void" +# Array(Int(32),4) -> "[4 x i32]" +# Struct(...) -> "{i32, i8*}" +# Func(...) -> "i32 (i32, i8*)" +# +# pool 用于复杂类型（Array/Struct/Func）递归打印子类型时分配临时缓冲区。 +# ============================================================ +def TypePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, + ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """将类型 ty 序列化为 IR 文本写入 buf（容量 size） + + pool 用于递归打印子类型时分配临时缓冲区。 + """ + if ty is None or buf is None or size == 0: + return + buf[0] = '\0' + match ty: + case LLVMType.Int(bits): + viperlib.snprintf(buf, size, "i%d", bits) + case LLVMType.Ptr(pointee): + # 先打印 pointee，再追加 '*' + TypePrint(buf, size, pointee, pool) + pos: t.CSizeT = string.strlen(buf) + if pos + 1 < size: + buf[pos] = '*' + buf[pos + 1] = '\0' + case LLVMType.Array(elem_ty, count): + # [N x elem_ty] + viperlib.snprintf(buf, size, "[%d x ", count) + tmp: t.CChar | t.CPtr = pool.alloc(128) + if tmp is not None: + TypePrint(tmp, 128, elem_ty, pool) + _strappend(buf, size, tmp) + _strappend(buf, size, "]") + case LLVMType.Struct(fields, fcount, name): + if name is not None: + # 命名结构体: %"name" + buf[0] = '%' + buf[1] = '"' + buf[2] = '\0' + _strappend(buf, size, name) + _strappend(buf, size, "\"") + else: + # 匿名结构体: {ty1, ty2, ...} + _print_struct_body(buf, size, fields, pool) + case LLVMType.Func(ret, params, count): + # ret (ty1, ty2, ...) + TypePrint(buf, size, ret, pool) + _strappend(buf, size, " (") + cur2: ParamNode | t.CPtr = params + first2: t.CInt = 1 + tmp3: t.CChar | t.CPtr = pool.alloc(128) + while cur2 is not None: + if first2 == 0: + _strappend(buf, size, ", ") + if tmp3 is not None: + TypePrint(tmp3, 128, cur2.Ty, pool) + _strappend(buf, size, tmp3) + cur2 = cur2.Next + first2 = 0 + _strappend(buf, size, ")") + case LLVMType.Void(): + string.strcpy(buf, "void") + case LLVMType.Label(): + string.strcpy(buf, "label") + case LLVMType.Metadata(): + string.strcpy(buf, "metadata") + case LLVMType.Float(bits): + if bits == 16: + string.strcpy(buf, "half") + elif bits == 32: + string.strcpy(buf, "float") + elif bits == 64: + string.strcpy(buf, "double") + elif bits == 128: + string.strcpy(buf, "fp128") + else: + viperlib.snprintf(buf, size, "float%x", bits) + case _: + string.strcpy(buf, "void") + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _strappend(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 src 追加到 dst 末尾（不超过 dst_size）""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' + + +# ============================================================ +# _print_struct_body - 打印结构体字段部分 {ty1, ty2, ...} +# +# 供 TypePrint（匿名结构体）和 PrintStructDefinition（命名结构体定义行）复用 +# ============================================================ +def _print_struct_body(buf: t.CChar | t.CPtr, size: t.CSizeT, + fields: ParamNode | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """打印 {ty1, ty2, ...} 到 buf（追加模式，不覆盖已有内容）""" + if buf is None or size == 0: + return + _strappend(buf, size, "{") + cur: ParamNode | t.CPtr = fields + first: t.CInt = 1 + tmp: t.CChar | t.CPtr = pool.alloc(128) + while cur is not None: + if first == 0: + _strappend(buf, size, ", ") + if tmp is not None: + TypePrint(tmp, 128, cur.Ty, pool) + _strappend(buf, size, tmp) + cur = cur.Next + first = 0 + _strappend(buf, size, "}") + + +# ============================================================ +# PrintStructDefinition - 打印命名结构体类型定义行 +# +# 输出格式: %"name" = type { ty1, ty2, ... } +# 供 LLVMModulePrint 在全局变量/函数之前输出类型定义 +# ============================================================ +def PrintStructDefinition(buf: t.CChar | t.CPtr, size: t.CSizeT, + ty: LLVMType | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """打印命名结构体类型定义行 %"name" = type { ty1, ty2, ... } + + 当 fcount==0 且 fields is None 时，输出 opaque 声明: %"name" = type opaque + """ + if ty is None or buf is None or size == 0: + return + match ty: + case LLVMType.Struct(fields, fcount, name): + if name is None: + return + buf[0] = '%' + buf[1] = '"' + buf[2] = '\0' + _strappend(buf, size, name) + # opaque 类型: fcount==0 且 fields is None（跨模块前向声明） + if fcount == 0 and fields is None: + _strappend(buf, size, "\" = type opaque") + else: + _strappend(buf, size, "\" = type ") + _print_struct_body(buf, size, fields, pool) + case _: + return + + +# ============================================================ +# get_struct_name - 获取命名结构体的名称 +# +# 返回 Name 字段（如 "sha1.ClassName"），非 Struct 类型或匿名 Struct 返回 None +# 供 module_add_named_type 检查是否为命名结构体及去重 +# ============================================================ +def get_struct_name(ty: LLVMType | t.CPtr) -> t.CChar | t.CPtr: + """获取命名结构体的名称，非命名结构体返回 None""" + if ty is None: + return None + match ty: + case LLVMType.Struct(fields, fcount, name): + return name + case _: + return None + + + diff --git a/includes/llvmlite/__values.py b/includes/llvmlite/__values.py new file mode 100644 index 0000000..4af3140 --- /dev/null +++ b/includes/llvmlite/__values.py @@ -0,0 +1,212 @@ +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode +from .__types import LLVMType, TypePrint + + +# ============================================================ +# Value: SSA 值表示 +# +# 表示一个 LLVM IR 值：SSA 临时值（%0/%result）或常量字面量（42/0）。 +# 每个值有类型指针、名字、是否常量标志。 +# 继承 GSListNode[Value] 获取强类型 Next: Value|CPtr（值池链表） +# ============================================================ +class Value(GSListNode[Value]): + Ty: LLVMType | t.CPtr # 值的类型 + Name: t.CChar | t.CPtr # SSA 名（"%0"/"%result"）或常量文本（"42"/"1.5e+00"） + IsConst: t.CInt # 1=常量字面量, 0=SSA 临时值 + IntVal: t.CInt64T # 整数常量值（当 IsConst=1 时使用） + FloatVal: t.CDouble # 浮点常量值（当 IsConst=1 且 Ty 为 Float 时使用） + + +# ============================================================ +# 工厂函数 +# ============================================================ +# Value 硬编码大小: Next(8) + Ty(8) + Name(8) + IsConst(4) + IntVal(8) + FloatVal(8) = 48 字节 +VALUE_SIZE: t.CDefine = 48 + +def new_value(pool: memhub.MemBuddy | t.CPtr) -> Value | t.CPtr: + """从 mpool 分配一个零初始化的 Value""" + ptr: Value | t.CPtr = pool.alloc(VALUE_SIZE) + if ptr: + string.memset(ptr, 0, VALUE_SIZE) + return ptr + + +def ConstInt(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, + val: t.CInt64T, name: t.CChar | t.CPtr) -> Value | t.CPtr: + """创建整数常量值 + + 参数: + ty: 整数类型（Int1/Int8/Int32/Int64） + val: 整数值 + name: 常量文本（如 "42"），由调用方提供 + """ + v: Value | t.CPtr = new_value(pool) + if v is None: return None + v.Ty = ty + v.Name = name + v.IsConst = 1 + v.IntVal = val + return v + + +def ConstNull(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, + name: t.CChar | t.CPtr) -> Value | t.CPtr: + """创建 null 指针常量""" + v: Value | t.CPtr = new_value(pool) + if v is None: return None + v.Ty = ty + v.Name = name + v.IsConst = 1 + v.IntVal = 0 + return v + + +def ConstZero(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr) -> Value | t.CPtr: + """创建 zeroinitializer 常量（零初始化聚合类型：结构体/数组）""" + v: Value | t.CPtr = new_value(pool) + if v is None: return None + v.Ty = ty + v.Name = "zeroinitializer" + v.IsConst = 1 + v.IntVal = 0 + return v + + +def ConstFloat(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, + val: t.CDouble) -> Value | t.CPtr: + """创建浮点常量值 + + 参数: + ty: 浮点类型（Half/Float/Double/FP128） + val: 浮点数值 + 名字自动格式化为 LLVM IR 科学计数法文本（如 "1.500000e+00"）。 + """ + v: Value | t.CPtr = new_value(pool) + if v is None: return None + v.Ty = ty + v.IsConst = 1 + v.FloatVal = val + v.IntVal = 0 + # 格式化名字: LLVM IR 使用 "%e" 科学计数法（6 位小数） + name_buf: t.CChar | t.CPtr = pool.alloc(32) + if name_buf is not None: + viperlib.snprintf(name_buf, 32, "%e", val) + v.Name = name_buf + return v + + +def SSAValue(pool: memhub.MemBuddy | t.CPtr, ty: LLVMType | t.CPtr, + name: t.CChar | t.CPtr) -> Value | t.CPtr: + """创建 SSA 临时值 + + 参数: + ty: 值类型 + name: SSA 名（如 "%0"/"%result"），由调用方提供 + """ + v: Value | t.CPtr = new_value(pool) + if v is None: return None + v.Ty = ty + v.Name = name + v.IsConst = 0 + v.IntVal = 0 + return v + + +# ============================================================ +# ValuePrint: 将值序列化为 IR 文本 +# +# 示例: +# ConstInt(i32, 42, "42") -> "i32 42" +# SSAValue(i32, "%0") -> "i32 %0" +# ============================================================ +def ValuePrint(buf: t.CChar | t.CPtr, size: t.CSizeT, + val: Value | t.CPtr, pool: memhub.MemBuddy | t.CPtr): + """将值 val 序列化为 IR 文本写入 buf + + 格式: + pool 用于复杂类型（Array/Struct/Func）递归打印时分配临时缓冲区。 + """ + if val is None or buf is None or size == 0: + return + # 先写类型 + TypePrint(buf, size, val.Ty, pool) + # 追加空格 + pos: t.CSizeT = string.strlen(buf) + if pos + 1 < size: + buf[pos] = ' ' + buf[pos + 1] = '\0' + # 追加名字 + if val.Name is not None: + pos = string.strlen(buf) + _append_str(buf, size, val.Name) + + +# ============================================================ +# Value 字段访问器（供其他模块绕过 stub 类型限制使用） +# +# 跨模块编译时，使用方模块只有 Value 的 stub 类型（字段不足或偏移错误）， +# 导致 TransPyC 静默跳过字段访问（不生成 GEP，不报错）。 +# 这些访问器在拥有完整类型定义的本模块内执行，能正确访问所有字段。 +# ============================================================ +def value_get_ty(val: Value | t.CPtr) -> LLVMType | t.CPtr: + """获取值的类型""" + if val is None: + return None + return val.Ty + + +def value_get_next(val: Value | t.CPtr) -> Value | t.CPtr: + """获取链表下一个值""" + if val is None: + return None + return val.Next + + +def value_set_ty(val: Value | t.CPtr, ty: LLVMType | t.CPtr): + """设置值的类型""" + if val is None: + return + val.Ty = ty + + +def value_set_next(val: Value | t.CPtr, nxt: Value | t.CPtr): + """设置链表下一个值""" + if val is None: + return + val.Next = nxt + + +def value_set_name(val: Value | t.CPtr, name: t.CChar | t.CPtr): + """设置值的名字""" + if val is None: + return + val.Name = name + + +def value_set_isconst(val: Value | t.CPtr, is_const: t.CInt): + """设置是否为常量""" + if val is None: + return + val.IsConst = is_const + + +# ============================================================ +# 内部辅助 +# ============================================================ +def _append_str(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 src 追加到 dst 末尾""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' diff --git a/includes/llvmlite/__verify.py b/includes/llvmlite/__verify.py new file mode 100644 index 0000000..2bd6d9c --- /dev/null +++ b/includes/llvmlite/__verify.py @@ -0,0 +1,396 @@ +import t, c +from stdint import * +import string +import viperlib +import memhub +from linkedlist import GSListNode, GSList +from .__function import Function, BasicBlock, Line, Param +from .__module import LLVMModule + + +# ============================================================ +# SSA 验证器 +# +# 检查项: +# 1. 块终止: 每个基本块必须以终止指令结尾 (ret/br/switch/unreachable) +# 2. SSA 唯一性: 每个 %name 只能被定义一次 +# 3. 未定义引用: 所有 %name 使用必须有对应定义 +# 4. Phi 位置: phi 指令只能出现在块首（非 phi 指令之前） +# ============================================================ + +# 错误码 +VERIFY_OK: t.CDefine = 0 +VERIFY_ERR_NO_TERMINATOR: t.CDefine = 1 +VERIFY_ERR_DUPLICATE_DEF: t.CDefine = 2 +VERIFY_ERR_UNDEFINED_USE: t.CDefine = 3 +VERIFY_ERR_PHI_NOT_FIRST: t.CDefine = 4 + + +# ============================================================ +# NameEntry: 已定义 SSA 名条目（链表节点） +# ============================================================ +class NameEntry(GSListNode[NameEntry]): + Name: t.CChar | t.CPtr # SSA 名（如 "%0"/"%result"） + + +# ============================================================ +# VerifyResult: 验证结果 +# ============================================================ +class VerifyResult: + ErrorCount: t.CInt # 错误总数 + WarningCount: t.CInt # 警告总数 + ErrorCode: t.CInt # 第一个错误码 + ErrorBuf: t.CChar | t.CPtr # 错误消息缓冲区 + ErrorBufSize: t.CSizeT # 缓冲区大小 + + +# ============================================================ +# 辅助函数 +# ============================================================ +def _new_verify_result(pool: memhub.MemBuddy | t.CPtr, + buf_size: t.CSizeT) -> VerifyResult | t.CPtr: + """创建验证结果对象""" + ptr: VerifyResult | t.CPtr = pool.alloc(VerifyResult.__sizeof__()) + if ptr is None: return None + string.memset(ptr, 0, VerifyResult.__sizeof__()) + ptr.ErrorCount = 0 + ptr.WarningCount = 0 + ptr.ErrorCode = VERIFY_OK + buf: t.CChar | t.CPtr = pool.alloc(buf_size) + if buf is not None: + buf[0] = '\0' + ptr.ErrorBuf = buf + ptr.ErrorBufSize = buf_size + return ptr + + +def _report_error(result: VerifyResult | t.CPtr, code: t.CInt, msg: t.CChar | t.CPtr): + """记录一个错误""" + if result is None: return + result.ErrorCount += 1 + if result.ErrorCode == VERIFY_OK: + result.ErrorCode = code + if result.ErrorBuf is not None and msg is not None: + _append_cstr(result.ErrorBuf, result.ErrorBufSize, msg) + _append_cstr(result.ErrorBuf, result.ErrorBufSize, "\n") + + +def _append_cstr(dst: t.CChar | t.CPtr, dst_size: t.CSizeT, src: t.CChar | t.CPtr): + """将 src 追加到 dst 末尾""" + if dst is None or src is None: return + dlen: t.CSizeT = string.strlen(dst) + slen: t.CSizeT = string.strlen(src) + remain: t.CSizeT = dst_size - dlen + if remain <= 0: return + i: t.CSizeT = 0 + while i < slen and i + 1 < remain: + dst[dlen + i] = src[i] + i += 1 + dst[dlen + i] = '\0' + + +# ============================================================ +# SSA 名表操作 +# ============================================================ +def _new_name_table(pool: memhub.MemBuddy | t.CPtr) -> GSList[NameEntry] | t.CPtr: + """创建空名表""" + gsl: GSList[NameEntry] | t.CPtr = pool.alloc(24) # GSList: Head+Tail+Count + if gsl is None: return None + string.memset(gsl, 0, 24) + return gsl + + +# NameEntry 硬编码大小: Next(8) + Name(8) = 16 字节 +NENTRY_SIZE: t.CDefine = 16 + +def _name_table_add(table: GSList[NameEntry] | t.CPtr, + pool: memhub.MemBuddy | t.CPtr, + name: t.CChar | t.CPtr) -> t.CInt: + """添加名到表。返回 1=重复, 0=成功添加""" + if table is None or name is None: return 0 + # 检查重复 + cur: NameEntry | t.CPtr = table.Head + while cur is not None: + if string.strcmp(cur.Name, name) == 0: + return 1 + cur = cur.Next + # 添加新条目 + entry: NameEntry | t.CPtr = pool.alloc(NENTRY_SIZE) + if entry is None: return 0 + string.memset(entry, 0, NENTRY_SIZE) + entry.Name = name + table.append(entry) + return 0 + + +def _name_table_contains(table: GSList[NameEntry] | t.CPtr, + name: t.CChar | t.CPtr) -> t.CInt: + """检查名是否在表中。返回 1=存在, 0=不存在""" + if table is None or name is None: return 0 + cur: NameEntry | t.CPtr = table.Head + while cur is not None: + if string.strcmp(cur.Name, name) == 0: + return 1 + cur = cur.Next + return 0 + + +# ============================================================ +# 文本分析辅助 +# ============================================================ +def _is_char_name(ch: t.CChar) -> t.CInt: + """判断字符是否为 SSA 名字符 [a-zA-Z0-9._]""" + if ch >= 'a' and ch <= 'z': return 1 + if ch >= 'A' and ch <= 'Z': return 1 + if ch >= '0' and ch <= '9': return 1 + if ch == '.' or ch == '_': return 1 + return 0 + + +def _extract_def_name(pool: memhub.MemBuddy | t.CPtr, + text: t.CChar | t.CPtr) -> t.CChar | t.CPtr: + """从定义行提取被定义的 SSA 名。 + + 定义行格式: "% = " + 返回分配的 NUL 结尾名串（如 "%0"），或 None 若非定义行。 + """ + if text is None: return None + if text[0] != '%': return None + # 扫描 " = " 模式 + i: t.CInt = 1 + while text[i] != '\0': + if text[i] == ' ' and text[i + 1] == '=' and text[i + 2] == ' ': + # 找到 " = " 在位置 i + name_len: t.CInt = i + name: t.CChar | t.CPtr = pool.alloc(name_len + 1) + if name is None: return None + j: t.CInt = 0 + while j < name_len: + name[j] = text[j] + j += 1 + name[name_len] = '\0' + return name + i += 1 + return None + + +def _is_terminator(text: t.CChar | t.CPtr) -> t.CInt: + """判断文本是否为终止指令""" + if text is None: return 0 + if string.strncmp(text, "ret", 3) == 0: return 1 + if string.strncmp(text, "br ", 3) == 0: return 1 + if string.strncmp(text, "switch ", 7) == 0: return 1 + if string.strncmp(text, "unreachable", 11) == 0: return 1 + if string.strncmp(text, "indirectbr ", 11) == 0: return 1 + if string.strncmp(text, "resume ", 7) == 0: return 1 + return 0 + + +def _is_phi(text: t.CChar | t.CPtr) -> t.CInt: + """判断文本是否为 phi 指令""" + if text is None: return 0 + if string.strstr(text, " = phi ") is not None: return 1 + return 0 + + +def _check_uses_in_line(table: GSList[NameEntry] | t.CPtr, + text: t.CChar | t.CPtr, + def_name: t.CChar | t.CPtr) -> t.CInt: + """检查行中所有 SSA 名使用是否已定义。 + + 跳过 def_name（定义位置的名字）。 + 返回未定义使用数量。 + """ + if text is None: return 0 + i: t.CInt = 0 + undef_count: t.CInt = 0 + while text[i] != '\0': + if text[i] == '%': + # 提取名 + name_start: t.CInt = i + i += 1 + while text[i] != '\0' and _is_char_name(text[i]) == 1: + i += 1 + name_len: t.CInt = i - name_start + if name_len > 1: + # 构建临时名串进行比较 + # 由于无法在栈上分配，用 strstr 进行简化检查 + # 这里采用逐字符比较方式 + if _name_matches_at(text, name_start, name_len, table) == 0: + # 检查是否是 def_name（定义本身的位置） + if def_name is not None: + if _str_matches_at(text, name_start, name_len, def_name) == 0: + undef_count += 1 + else: + undef_count += 1 + else: + i += 1 + return undef_count + + +def _name_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, + table: GSList[NameEntry] | t.CPtr) -> t.CInt: + """检查 text[start:start+length] 是否匹配名表中的某个条目。 + 返回 1=匹配, 0=不匹配 + """ + if table is None: return 0 + cur: NameEntry | t.CPtr = table.Head + while cur is not None: + if cur.Name is not None: + if _str_matches_at(text, start, length, cur.Name) == 1: + return 1 + cur = cur.Next + return 0 + + +def _str_matches_at(text: t.CChar | t.CPtr, start: t.CInt, length: t.CInt, + target: t.CChar | t.CPtr) -> t.CInt: + """检查 text[start:start+length] 是否等于 target 字符串。 + 返回 1=匹配, 0=不匹配 + """ + if text is None or target is None: return 0 + # 检查 target 长度 + target_len: t.CInt = string.strlen(target) + if target_len != length: return 0 + i: t.CInt = 0 + while i < length: + if text[start + i] != target[i]: return 0 + i += 1 + return 1 + + +# ============================================================ +# 主验证函数 +# ============================================================ +def verify_function(pool: memhub.MemBuddy | t.CPtr, + func: Function | t.CPtr, + result: VerifyResult | t.CPtr) -> t.CInt: + """验证函数的 SSA 正确性。 + + 参数: + pool: 内存池 + func: 要验证的函数 + result: 验证结果（调用前需用 _new_verify_result 创建） + + 返回: 错误数量（0 = 验证通过） + """ + if func is None or result is None: return 1 + + # 创建名表 + name_table: GSList[NameEntry] | t.CPtr = _new_name_table(pool) + if name_table is None: return 1 + + # 1. 将函数参数加入名表（参数是已定义的） + param: Param | t.CPtr = None + if func.Params is not None: + param = func.Params.Head + while param is not None: + if param.Name is not None: + _name_table_add(name_table, pool, param.Name) + param = param.Next + + # 1b. 将基本块名加入名表（块标签如 %then 是 br/phi 的合法引用目标） + blk_iter: BasicBlock | t.CPtr = None + if func.Blocks is not None: + blk_iter = func.Blocks.Head + while blk_iter is not None: + if blk_iter.Name is not None: + blk_name_len: t.CSizeT = string.strlen(blk_iter.Name) + blk_ref: t.CChar | t.CPtr = pool.alloc(blk_name_len + 2) + if blk_ref is not None: + # 构造 "%" 字符串 + blk_ref[0] = '%' + k: t.CSizeT = 0 + while k < blk_name_len: + blk_ref[k + 1] = blk_iter.Name[k] + k += 1 + blk_ref[blk_name_len + 1] = '\0' + _name_table_add(name_table, pool, blk_ref) + blk_iter = blk_iter.Next + + # 2. 遍历所有块 + blk: BasicBlock | t.CPtr = None + if func.Blocks is not None: + blk = func.Blocks.Head + while blk is not None: + # 检查终止指令 + if blk.IsTerminated == 0: + # 进一步检查最后一行是否为终止指令 + has_term: t.CInt = 0 + line_check: Line | t.CPtr = None + if blk.Lines is not None: + line_check = blk.Lines.Head + while line_check is not None: + if line_check.Buf is not None: + if _is_terminator(line_check.Buf) == 1: + has_term = 1 + line_check = line_check.Next + if has_term == 0: + blk_name: t.CChar | t.CPtr = "" + if blk.Name is not None: + blk_name = blk.Name + err_buf: t.CChar | t.CPtr = pool.alloc(128) + if err_buf is not None: + viperlib.snprintf(err_buf, 128, "ERROR: block '%s' missing terminator", blk_name) + _report_error(result, VERIFY_ERR_NO_TERMINATOR, err_buf) + + # 遍历块内指令行 + seen_non_phi: t.CInt = 0 + line: Line | t.CPtr = None + if blk.Lines is not None: + line = blk.Lines.Head + while line is not None: + if line.Buf is not None: + text: t.CChar | t.CPtr = line.Buf + + # 检查 phi 位置 + if _is_phi(text) == 1: + if seen_non_phi == 1: + phi_err: t.CChar | t.CPtr = pool.alloc(128) + if phi_err is not None: + viperlib.snprintf(phi_err, 128, "ERROR: phi not at block start") + _report_error(result, VERIFY_ERR_PHI_NOT_FIRST, phi_err) + else: + # 非空行且非 phi 标记为已见非 phi + if string.strlen(text) > 0: + seen_non_phi = 1 + + # 提取定义名 + def_name: t.CChar | t.CPtr = _extract_def_name(pool, text) + if def_name is not None: + # 检查重复定义 + dup: t.CInt = _name_table_add(name_table, pool, def_name) + if dup == 1: + dup_err: t.CChar | t.CPtr = pool.alloc(128) + if dup_err is not None: + viperlib.snprintf(dup_err, 128, "ERROR: duplicate definition '%s'", def_name) + _report_error(result, VERIFY_ERR_DUPLICATE_DEF, dup_err) + + # 检查未定义使用 + undef: t.CInt = _check_uses_in_line(name_table, text, def_name) + if undef > 0: + use_err: t.CChar | t.CPtr = pool.alloc(128) + if use_err is not None: + viperlib.snprintf(use_err, 128, "ERROR: %d undefined use(s) in: %s", undef, text) + _report_error(result, VERIFY_ERR_UNDEFINED_USE, use_err) + + line = line.Next + + blk = blk.Next + + return result.ErrorCount + + +def verify_module(pool: memhub.MemBuddy | t.CPtr, mod: LLVMModule | t.CPtr, result: VerifyResult | t.CPtr) -> t.CInt: + """验证模块中所有函数的 SSA 正确性。 + + 返回: 错误总数 + """ + if mod is None or result is None: return 1 + func: Function | t.CPtr = mod.FuncHead + while func is not None: + if func.IsDeclared == 0: + # 只验证有函数体的定义，跳过声明 + verify_function(pool, func, result) + func = func.Next + return result.ErrorCount diff --git a/includes/lock.py b/includes/lock.py new file mode 100644 index 0000000..b3ce100 --- /dev/null +++ b/includes/lock.py @@ -0,0 +1,39 @@ +import t, c +from stdint import * +from w32.win32base import * +from w32.win32sync import CRITICAL_SECTION, InitializeCriticalSection, EnterCriticalSection, LeaveCriticalSection, DeleteCriticalSection, TryEnterCriticalSection + + +@t.Object +class Lock: + """基于 CRITICAL_SECTION 的互斥锁，支持递归加锁。 + + 用法: + lock: Lock = Lock() + with lock: + # 临界区 + pass + """ + cs: CRITICAL_SECTION + + def __init__(self): + InitializeCriticalSection(c.Addr(self.cs)) + + def acquire(self): + EnterCriticalSection(c.Addr(self.cs)) + + def release(self): + LeaveCriticalSection(c.Addr(self.cs)) + + def try_acquire(self) -> BOOL: + return TryEnterCriticalSection(c.Addr(self.cs)) + + def __enter__(self) -> 'Lock' | t.CPtr: + self.acquire() + return self + + def __exit__(self): + self.release() + + def delete(self): + DeleteCriticalSection(c.Addr(self.cs)) diff --git a/includes/memhub.py b/includes/memhub.py new file mode 100644 index 0000000..7b1c44a --- /dev/null +++ b/includes/memhub.py @@ -0,0 +1,479 @@ +import t, c +from stdint import * +import string +import atom +import viperio + +# ============================================================ +# memhub - 统一内存管理库 +# MemManager (多态基类, vtable) +# ├─ MemPool (arena bump 分配器, 不支持单个 free) +# ├─ MemSlab (定长块池, 支持 alloc/free 位图跟踪) +# └─ MemBuddy (伙伴系统, 支持 alloc/free/realloc 合并) +# 所有子类通过 vtable 覆盖 alloc/free/reset, 可用 MemManager 指针多态调用 +# ============================================================ + +MEMHUB_ALIGN: t.CDefine = 8 + +# MemSlab 位图常量 +MEMSLAB_MIN_BLOCK: t.CDefine = 16 +MEMSLAB_BITMAP_BYTES: t.CDefine = 256 + +# MemBuddy 伙伴系统常量 +MEMBUDDY_MIN_BLOCK: t.CDefine = 32 +MEMBUDDY_MAX_ORDERS: t.CDefine = 32 +MEMBUDDY_HEADER_SIZE: t.CDefine = 8 + + +def _align_up(val: t.CSizeT, align: t.CSizeT) -> t.CSizeT: + if align == 0: return val + return (val + align - 1) & ~(align - 1) + + +def _largest_pow2_le(val: t.CSizeT) -> t.CSizeT: + if val == 0: return 0 + p: t.CSizeT = 1 + while p * 2 <= val: + p = p * 2 + return p + + +def _block_size_at_order(order: t.CInt) -> t.CSizeT: + bs: t.CSizeT = MEMBUDDY_MIN_BLOCK + i: t.CInt + for i in range(order): + bs = bs << 1 + return bs + + +# ============================================================ +# MemManager - 多态基类 +# @t.CVTable 显式标记, 确保跨模块导入时也被识别为 vtable 类 +# (HandlesImports 只通过装饰器检测 IsCVTable, 不像同模块自动推断) +# ============================================================ +@t.CVTable +class MemManager: + __provides__: list[str] = ['__memmgr__'] + base: t.CVoid | t.CPtr # 内存区起始地址 + size: t.CSizeT # 内存区总大小 + + def __init__(self, base: t.CVoid | t.CPtr, size: t.CSizeT): + self.base = base + self.size = size + + # === 虚函数 (子类覆盖) === + + def alloc(self, size: t.CSizeT) -> t.CVoid | t.CPtr: + # 默认: 不支持分配 + return None + + def free(self, ptr: t.CVoid | t.CPtr) -> t.CInt: + # 默认: 无操作 + return 0 + + def reset(self) -> t.CInt: + # 默认: 无操作 + return 0 + + # === 具体方法 (调用虚函数, 自动多态分派) === + + def calloc(self, count: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr: + total: t.CSizeT = count * size + ptr: t.CVoid | t.CPtr = self.alloc(total) + if ptr is not None: + string.memset(ptr, 0, total) + return ptr + + def realloc(self, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: + if ptr is None: + return self.alloc(new_size) + if new_size == 0: + self.free(ptr) + return None + new_ptr: t.CVoid | t.CPtr = self.alloc(new_size) + if new_ptr is None: + return ptr + copy_size: t.CSizeT = old_size + if new_size < old_size: + copy_size = new_size + string.memcpy(new_ptr, ptr, copy_size) + self.free(ptr) + return new_ptr + + def __enter__(self) -> 'MemManager' | t.CPtr: + return self + + def __exit__(self): + self.reset() + + def alloc_buf(self, capacity: t.CSizeT) -> viperio.Buf | t.CPtr: + buf: viperio.Buf | t.CPtr = self.alloc(viperio.Buf.__sizeof__()) + if buf is None: + return None + data_ptr: t.CVoid | t.CPtr = self.alloc(capacity) + if data_ptr is None: + return None + buf.__before_init__() + buf.__init__(t.CChar(t.CUInt64T(data_ptr), t.CPtr), capacity) + return buf + + +# ============================================================ +# MemPool - arena bump 分配器 +# 不支持单个 free, reset 一次性回收全部 +# ============================================================ +class MemPool(MemManager): + offset: t.CSizeT # bump 游标 + high_water: t.CSizeT # 峰值水位 + + def __init__(self, base: t.CVoid | t.CPtr, size: t.CSizeT): + self.base = base + self.size = size + self.offset = 0 + self.high_water = 0 + + def alloc(self, size: t.CSizeT) -> t.CVoid | t.CPtr: + if size == 0: return None + aligned: t.CSizeT = _align_up(size, MEMHUB_ALIGN) + if self.offset + aligned > self.size: return None + ptr: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(self.base) + self.offset, t.CPtr) + self.offset += aligned + if self.offset > self.high_water: + self.high_water = self.offset + return ptr + + def free(self, ptr: t.CVoid | t.CPtr) -> t.CInt: + # bump 分配器不支持单个 free + return 1 + + def reset(self) -> t.CInt: + self.offset = 0 + self.high_water = 0 + return 1 + + +# ============================================================ +# MemSlab - 定长块池 +# arena 开头存位图, 后续区域按 block_size 切分, 空闲链管理 +# ============================================================ +class MemSlab(MemManager): + block_size: t.CSizeT # 每块大小 (对齐后) + block_count: t.CSizeT # 总块数 + used_count: t.CSizeT # 已用块数 + free_list: t.CVoid | t.CPtr # 空闲链头 + alloc_map: t.CUInt8T | t.CPtr # 分配位图 (位于 arena 开头) + alloc_map_size: t.CSizeT # 位图字节数 + usable: t.CVoid | t.CPtr # 可用块区起始 (跳过位图) + usable_size: t.CSizeT # 可用块区大小 + + def __init__(self, base: t.CVoid | t.CPtr, size: t.CSizeT, block_size: t.CSizeT): + self.base = base + self.size = size + self.block_size = 0 + self.block_count = 0 + self.used_count = 0 + self.free_list = None + self.alloc_map = None + self.alloc_map_size = 0 + self.usable = None + self.usable_size = 0 + + if base is None: return + + bs: t.CSizeT = _align_up(block_size, MEMHUB_ALIGN) + if bs < MEMSLAB_MIN_BLOCK: bs = MEMSLAB_MIN_BLOCK + + map_bytes: t.CSizeT = _align_up(MEMSLAB_BITMAP_BYTES, MEMHUB_ALIGN) + if size < map_bytes + bs: return + + self.alloc_map = base + self.alloc_map_size = map_bytes + self.usable = t.CVoid(t.CUInt64T(base) + map_bytes, t.CPtr) + self.usable_size = size - map_bytes + self.block_size = bs + + # 清零位图 + idx: t.CSizeT = 0 + while idx < map_bytes: + self.alloc_map[idx] = 0 + idx += 1 + + self.block_count = self.usable_size / bs + if self.block_count == 0: return + + # 构建空闲链: 每块首 8 字节存下一块指针 + self.free_list = None + i: t.CSizeT = 0 + while i < self.block_count: + block: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(self.usable) + i * bs, t.CPtr) + c.DerefAs(block, self.free_list) + self.free_list = block + i += 1 + + def alloc(self, size: t.CSizeT) -> t.CVoid | t.CPtr: + # slab 模式: 仅当请求 <= block_size 时分配一块 + if self.free_list is None: return None + if size > self.block_size: return None + block: t.CVoid | t.CPtr = self.free_list + self.free_list = t.CVoid(c.Deref(t.CUInt64T(block, t.CPtr)), t.CPtr) + self.used_count += 1 + idx: t.CSizeT = (t.CUInt64T(block) - t.CUInt64T(self.usable)) / self.block_size + self.alloc_map[idx / 8] = self.alloc_map[idx / 8] | t.CUInt8T(1 << (idx % 8)) + return block + + def free(self, ptr: t.CVoid | t.CPtr) -> t.CInt: + if ptr is None: return 0 + p: t.CUInt64T = t.CUInt64T(ptr) + if p < t.CUInt64T(self.usable): return 0 + if p >= t.CUInt64T(self.usable) + self.block_count * self.block_size: return 0 + if (p - t.CUInt64T(self.usable)) % self.block_size != 0: return 0 + idx: t.CSizeT = (p - t.CUInt64T(self.usable)) / self.block_size + byte_idx: t.CSizeT = idx / 8 + bit_idx: t.CInt = t.CInt(idx % 8) + if (self.alloc_map[byte_idx] >> bit_idx) & 1 == 0: return 0 + self.alloc_map[byte_idx] = self.alloc_map[byte_idx] & t.CUInt8T(~(1 << bit_idx)) + c.DerefAs(ptr, self.free_list) + self.free_list = ptr + self.used_count -= 1 + return 1 + + def reset(self) -> t.CInt: + self.used_count = 0 + self.free_list = None + i: t.CSizeT = 0 + while i < self.block_count: + block: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(self.usable) + i * self.block_size, t.CPtr) + c.DerefAs(block, self.free_list) + self.free_list = block + if i / 8 < self.alloc_map_size: + self.alloc_map[i / 8] = 0 + i += 1 + return 1 + + +# ============================================================ +# MemBuddy - 伙伴系统分配器 +# arena 开头存空闲链头数组, 后续区域按 2 的幂管理 +# 线程安全 (自旋锁), 支持合并 +# ============================================================ +class MemBuddy(MemManager): + max_order: t.CInt # 最大阶数 + free_lists: t.CUInt64T | t.CPtr # 空闲链头数组 (位于 arena 开头) + lock_val: t.CVolatile | t.CInt # 自旋锁 + usable: t.CVoid | t.CPtr # 可用区起始 (跳过 free_lists) + usable_size: t.CSizeT # 可用区大小 (2 的幂) + + def __init__(self, base: t.CVoid | t.CPtr, size: t.CSizeT): + self.base = base + self.size = size + self.max_order = 0 + self.free_lists = None + self.lock_val = 0 + self.usable = None + self.usable_size = 0 + + self.lock_val = 0 + fl_bytes: t.CSizeT = (MEMBUDDY_MAX_ORDERS + 1) * 8 + self.free_lists = base + + # 初始化所有空闲链头为 0 + i: t.CInt + for i in range(MEMBUDDY_MAX_ORDERS + 1): + self.free_lists[i] = 0 + + if size <= fl_bytes: + self.usable = None + self.usable_size = 0 + self.max_order = 0 + return + + remaining: t.CSizeT = size - fl_bytes + usable: t.CSizeT = _largest_pow2_le(remaining) + + if usable < MEMBUDDY_MIN_BLOCK: + self.usable = None + self.usable_size = 0 + self.max_order = 0 + return + + self.usable = t.CVoid(t.CUInt64T(base) + fl_bytes, t.CPtr) + self.usable_size = usable + + # 计算 max_order: log2(usable / MIN_BLOCK) + self.max_order = 0 + bs: t.CSizeT = MEMBUDDY_MIN_BLOCK + while bs < usable: + bs = bs << 1 + self.max_order += 1 + + # 整个可用区作为 max_order 阶空闲块 + self._fl_push(self.max_order, self.usable) + + # === 空闲链操作 === + + def _fl_push(self, order: t.CInt, block: t.CVoid | t.CPtr): + old_head: t.CUInt64T = self.free_lists[order] + c.DerefAs(block, t.CVoid(old_head, t.CPtr)) + self.free_lists[order] = t.CUInt64T(block) + + def _fl_pop(self, order: t.CInt) -> t.CVoid | t.CPtr: + head_val: t.CUInt64T = self.free_lists[order] + if head_val == 0: + return None + block: t.CVoid | t.CPtr = t.CVoid(head_val, t.CPtr) + next_ptr: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(block, t.CPtr)), t.CPtr) + self.free_lists[order] = t.CUInt64T(next_ptr) + return block + + def _fl_find_and_remove(self, order: t.CInt, target: t.CVoid | t.CPtr) -> t.CInt: + head_val: t.CUInt64T = self.free_lists[order] + if head_val == 0: + return 0 + head: t.CVoid | t.CPtr = t.CVoid(head_val, t.CPtr) + if t.CUInt64T(head) == t.CUInt64T(target): + next_ptr: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(head, t.CPtr)), t.CPtr) + self.free_lists[order] = t.CUInt64T(next_ptr) + return 1 + prev: t.CVoid | t.CPtr = head + cur: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(head, t.CPtr)), t.CPtr) + while cur is not None: + if t.CUInt64T(cur) == t.CUInt64T(target): + next_ptr: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(cur, t.CPtr)), t.CPtr) + c.DerefAs(prev, next_ptr) + return 1 + prev = cur + cur = t.CVoid(c.Deref(t.CUInt64T(cur, t.CPtr)), t.CPtr) + return 0 + + # === 伙伴系统核心 === + + def _buddy_of(self, block: t.CVoid | t.CPtr, order: t.CInt) -> t.CVoid | t.CPtr: + offset: t.CSizeT = t.CUInt64T(block) - t.CUInt64T(self.usable) + bs: t.CSizeT = _block_size_at_order(order) + buddy_offset: t.CSizeT = offset ^ bs + return t.CVoid(t.CUInt64T(self.usable) + buddy_offset, t.CPtr) + + def _order_for_size(self, size: t.CSizeT) -> t.CInt: + order: t.CInt = 0 + bs: t.CSizeT = MEMBUDDY_MIN_BLOCK + while bs < size: + bs = bs << 1 + order += 1 + return order + + def _split_to_order(self, to_order: t.CInt) -> t.CVoid | t.CPtr: + found_order: t.CInt = to_order + while found_order <= self.max_order: + if self.free_lists[found_order] != 0: + break + found_order += 1 + if found_order > self.max_order: + return None + block: t.CVoid | t.CPtr = self._fl_pop(found_order) + while found_order > to_order: + found_order -= 1 + bs: t.CSizeT = _block_size_at_order(found_order) + buddy: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(block) + bs, t.CPtr) + self._fl_push(found_order, buddy) + return block + + def _coalesce(self, block: t.CVoid | t.CPtr, order: t.CInt): + while order < self.max_order: + buddy: t.CVoid | t.CPtr = self._buddy_of(block, order) + found: t.CInt = self._fl_find_and_remove(order, buddy) + if found == 0: + break + if t.CUInt64T(buddy) < t.CUInt64T(block): + block = buddy + order += 1 + self._fl_push(order, block) + + def _is_valid_ptr(self, ptr: t.CVoid | t.CPtr) -> t.CInt: + if ptr is None: return 0 + if self.usable is None: return 0 + block: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(ptr) - MEMBUDDY_HEADER_SIZE, t.CPtr) + if t.CUInt64T(block) < t.CUInt64T(self.usable): return 0 + if t.CUInt64T(block) >= t.CUInt64T(self.usable) + self.usable_size: return 0 + offset: t.CSizeT = t.CUInt64T(block) - t.CUInt64T(self.usable) + if offset % MEMBUDDY_MIN_BLOCK != 0: return 0 + stored: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(block, t.CPtr)), t.CPtr) + stored_val: t.CUInt64T = t.CUInt64T(stored) + if (stored_val & 1) == 0: return 0 + order: t.CInt = t.CInt(stored_val >> 1) + if order < 0: return 0 + if order > self.max_order: return 0 + return 1 + + # === 锁 === + + def _lock(self): + while atom.__atomic_test_and_set(c.Addr(self.lock_val), atom.ATOMIC_ACQUIRE): + pass + + def _unlock(self): + atom.__atomic_clear(c.Addr(self.lock_val), atom.ATOMIC_RELEASE) + + # === 虚函数覆盖 === + + def alloc(self, size: t.CSizeT) -> t.CVoid | t.CPtr: + self._lock() + result: t.CVoid | t.CPtr = None + if self.usable is not None: + if size != 0: + needed: t.CSizeT = size + MEMBUDDY_HEADER_SIZE + order: t.CInt = self._order_for_size(needed) + if order <= self.max_order: + block: t.CVoid | t.CPtr = self._split_to_order(order) + if block is not None: + c.DerefAs(block, t.CVoid(t.CUInt64T((order << 1) | 1), t.CPtr)) + result = t.CVoid(t.CUInt64T(block) + MEMBUDDY_HEADER_SIZE, t.CPtr) + self._unlock() + return result + + def free(self, ptr: t.CVoid | t.CPtr) -> t.CInt: + self._lock() + if ptr is not None: + if self._is_valid_ptr(ptr) != 0: + block: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(ptr) - MEMBUDDY_HEADER_SIZE, t.CPtr) + stored: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(block, t.CPtr)), t.CPtr) + stored_val: t.CUInt64T = t.CUInt64T(stored) + order: t.CInt = t.CInt(stored_val >> 1) + c.DerefAs(block, t.CVoid(0, t.CPtr)) + self._coalesce(block, order) + self._unlock() + return 1 + + def reset(self) -> t.CInt: + if self.usable is None: + return 0 + i: t.CInt + for i in range(MEMBUDDY_MAX_ORDERS + 1): + self.free_lists[i] = 0 + self._fl_push(self.max_order, self.usable) + return 1 + + # realloc 覆盖: 伙伴系统从块头读取旧阶数, 无需 old_size + def realloc(self, ptr: t.CVoid | t.CPtr, old_size: t.CSizeT, new_size: t.CSizeT) -> t.CVoid | t.CPtr: + if ptr is None: + return self.alloc(new_size) + if new_size == 0: + self.free(ptr) + return None + if self._is_valid_ptr(ptr) == 0: + return None + block: t.CVoid | t.CPtr = t.CVoid(t.CUInt64T(ptr) - MEMBUDDY_HEADER_SIZE, t.CPtr) + stored: t.CVoid | t.CPtr = t.CVoid(c.Deref(t.CUInt64T(block, t.CPtr)), t.CPtr) + stored_val: t.CUInt64T = t.CUInt64T(stored) + old_order: t.CInt = t.CInt(stored_val >> 1) + needed: t.CSizeT = new_size + MEMBUDDY_HEADER_SIZE + new_order: t.CInt = self._order_for_size(needed) + if new_order <= old_order: + return ptr + new_ptr: t.CVoid | t.CPtr = self.alloc(new_size) + if new_ptr is None: + return ptr + old_block_size: t.CSizeT = _block_size_at_order(old_order) + old_data_size: t.CSizeT = old_block_size - MEMBUDDY_HEADER_SIZE + string.memcpy(new_ptr, ptr, old_data_size) + self.free(ptr) + return new_ptr diff --git a/includes/numpy/__init__.py b/includes/numpy/__init__.py new file mode 100644 index 0000000..b06f2e6 --- /dev/null +++ b/includes/numpy/__init__.py @@ -0,0 +1,1163 @@ +from stdint import * +import stdlib +import string +import vipermath +import stdio +import t, c +import memhub + + +# ============================================================ +# Constants +# ============================================================ +MAX_NDIM: t.CDefine = 4 + +float64: t.CTypedef = t.CDouble +float32: t.CTypedef = t.CFloat +int64: t.CTypedef = t.CLong +int32: t.CTypedef = t.CInt +uint8: t.CTypedef = t.CUnsignedChar + +pi: t.CDefine = 3.14159265358979323846 +e: t.CDefine = 2.71828182845904523536 + +# ============================================================ +# ndarray - generic n-dimensional array (PEP 695) +# +# Element type T is a numeric type (t.CDouble / t.CFloat / t.CInt / ...). +# T is erased to i8* at the IR level but each ndarray[T] instantiation +# generates specialized code (data access uses T's load/store width). +# +# Default usage: ndarray[t.CDouble] for floating point. +# ============================================================ +@t.Object +class ndarray[T]: + data: T | t.CPtr + shape: t.CArray[t.CSizeT, MAX_NDIM] + strides: t.CArray[t.CSizeT, MAX_NDIM] + ndim: t.CInt + size: t.CSizeT + owns_data: t.CInt + pool: memhub.MemManager | t.CPtr + + def __new__(self, pool: memhub.MemManager | t.CPtr, n: t.CSizeT): + self.pool = pool + self.data = pool.alloc(n * T.__sizeof__()) + if self.data: + memset(self.data, 0, n * T.__sizeof__()) + self.ndim = 1 + self.size = n + self.shape[0] = n + self.strides[0] = 1 + self.owns_data = 1 + + def __add__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + result.data[i] = self.data[i] + other.data[i] + i += 1 + return result + + def __sub__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + result.data[i] = self.data[i] - other.data[i] + i += 1 + return result + + def __mul__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + result.data[i] = self.data[i] * other.data[i] + i += 1 + return result + + def __truediv__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + if other.data[i] != T(0): + result.data[i] = self.data[i] / other.data[i] + else: + result.data[i] = T(0) + i += 1 + return result + + def __floordiv__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + if other.data[i] != T(0): + result.data[i] = vipermath.floor(self.data[i] / other.data[i]) + else: + result.data[i] = T(0) + i += 1 + return result + + def __mod__(self, other: ndarray[T] | t.CPtr) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + if other.data[i] != T(0): + result.data[i] = vipermath.fmod(self.data[i], other.data[i]) + else: + result.data[i] = T(0) + i += 1 + return result + + def __neg__(self) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + i: t.CSizeT = 0 + while i < self.size: + result.data[i] = -self.data[i] + i += 1 + return result + + def __len__(self) -> t.CInt: + if self.ndim > 0: return t.CInt(self.shape[0]) + return 0 + + def at2d(self, row: t.CSizeT, col: t.CSizeT) -> T: + return self.data[row * self.shape[1] + col] + + def set2d(self, row: t.CSizeT, col: t.CSizeT, val: T): + self.data[row * self.shape[1] + col] = val + + def delete(self): + if self.owns_data and self.data: + self.pool.free(self.data) + self.data = None + self.pool.free(self) + + def fill(self, val: T): + i: t.CSizeT = 0 + while i < self.size: + self.data[i] = val + i += 1 + + def copy(self) -> ndarray[T] | t.CPtr: + result: ndarray[T] | t.CPtr = _empty_like(self) + if not result: return None + memcpy(result.data, self.data, self.size * T.__sizeof__()) + return result + + def reshape(self, new_shape: INTPTR, new_ndim: t.CInt) -> ndarray[T] | t.CPtr: + new_size: t.CSizeT = 1 + i: t.CInt = 0 + for i in range(new_ndim): + new_size *= t.CSizeT(new_shape[i]) + if new_size != self.size: return None + result: ndarray[T] | t.CPtr = _alloc_ndarray(self.pool) + if not result: return None + result.data = self.data + result.pool = self.pool + result.ndim = new_ndim + result.size = self.size + result.owns_data = 0 + for i in range(new_ndim): + result.shape[i] = t.CSizeT(new_shape[i]) + _compute_strides(result) + return result + + def sum(self) -> T: + s: T = T(0) + i: t.CSizeT = 0 + while i < self.size: + s += self.data[i] + i += 1 + return s + + def mean(self) -> T: + if self.size == 0: return T(0) + return self.sum() / T(self.size) + + def min(self) -> T: + if self.size == 0: return T(0) + m: T = self.data[0] + i: t.CSizeT = 1 + while i < self.size: + if self.data[i] < m: m = self.data[i] + i += 1 + return m + + def max(self) -> T: + if self.size == 0: return T(0) + m: T = self.data[0] + i: t.CSizeT = 1 + while i < self.size: + if self.data[i] > m: m = self.data[i] + i += 1 + return m + + def argmax(self) -> t.CInt: + if self.size == 0: return -1 + m: T = self.data[0] + idx: t.CInt = 0 + i: t.CSizeT = 1 + while i < self.size: + if self.data[i] > m: + m = self.data[i] + idx = t.CInt(i) + i += 1 + return idx + + def argmin(self) -> t.CInt: + if self.size == 0: return -1 + m: T = self.data[0] + idx: t.CInt = 0 + i: t.CSizeT = 1 + while i < self.size: + if self.data[i] < m: + m = self.data[i] + idx = t.CInt(i) + i += 1 + return idx + + def dot(self, other: ndarray[T] | t.CPtr) -> T: + s: T = T(0) + n: t.CSizeT = self.size if (self.size < other.size) else other.size + i: t.CSizeT = 0 + while i < n: + s += self.data[i] * other.data[i] + i += 1 + return s + + def T(self) -> ndarray[T] | t.CPtr: + if self.ndim != 2: return None + rows: t.CSizeT = self.shape[0] + cols: t.CSizeT = self.shape[1] + result: ndarray[T] | t.CPtr = empty2d(self.pool, rows, cols) + if not result: return None + # Swap shape for transpose + result.shape[0] = cols + result.shape[1] = rows + _compute_strides(result) + i: t.CSizeT = 0 + while i < rows: + j: t.CSizeT = 0 + while j < cols: + result.data[j * rows + i] = self.data[i * cols + j] + j += 1 + i += 1 + return result + + def print_arr(self): + if self.ndim == 1: + printf("[") + i: t.CSizeT = 0 + while i < self.size: + if i > 0: printf(", ") + printf("%.4f", self.data[i]) + i += 1 + printf("]\n") + elif self.ndim == 2: + rows: t.CSizeT = self.shape[0] + cols: t.CSizeT = self.shape[1] + printf("[") + i: t.CSizeT = 0 + while i < rows: + if i > 0: printf(" ") + printf("[") + j: t.CSizeT = 0 + while j < cols: + if j > 0: printf(", ") + printf("%.4f", self.data[i * cols + j]) + j += 1 + if i < rows - 1: + printf("]\n") + else: + printf("]]\n") + i += 1 + + +# ============================================================ +# Convenience aliases: typed ndarray instantiations +# (must be after class ndarray[T] definition to avoid forward reference) +# ============================================================ +Float64Array: t.CTypedef = ndarray[t.CDouble] +Float32Array: t.CTypedef = ndarray[t.CFloat] +Int64Array: t.CTypedef = ndarray[t.CLong] +Int32Array: t.CTypedef = ndarray[t.CInt] +Uint8Array: t.CTypedef = ndarray[t.CUnsignedChar] + + +# ============================================================ +# Internal helpers (specialized for float64 by default) +# Use _alloc_ndarray_t[T] / _empty_like_t[T] for other types +# ============================================================ +def _alloc_ndarray(pool: memhub.MemManager | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = pool.alloc(ndarray[t.CDouble].__sizeof__()) + if a: + memset(a, 0, ndarray[t.CDouble].__sizeof__()) + return a + +def _compute_strides(a: ndarray[t.CDouble] | t.CPtr): + if a.ndim <= 0: return + a.strides[a.ndim - 1] = 1 + i: t.CInt = a.ndim - 2 + while i >= 0: + a.strides[i] = a.strides[i + 1] * a.shape[i + 1] + i -= 1 + +def _empty_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _alloc_ndarray(a.pool) + if not result: return None + result.data = a.pool.alloc(a.size * t.CDouble.__sizeof__()) + if not result.data: + a.pool.free(result) + return None + result.pool = a.pool + result.ndim = a.ndim + result.size = a.size + result.owns_data = 1 + i: t.CInt = 0 + for i in range(a.ndim): + result.shape[i] = a.shape[i] + _compute_strides(result) + return result + + +# ============================================================ +# Array creation functions (float64 default) +# ============================================================ +def array(pool: memhub.MemManager | t.CPtr, data: t.CDouble | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = _alloc_ndarray(pool) + if not a: return None + a.data = pool.alloc(n * t.CDouble.__sizeof__()) + if not a.data: + pool.free(a) + return None + memcpy(a.data, data, n * t.CDouble.__sizeof__()) + a.pool = pool + a.ndim = 1 + a.size = n + a.shape[0] = n + a.strides[0] = 1 + a.owns_data = 1 + return a + +def zeros(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = _alloc_ndarray(pool) + if not a: return None + a.data = pool.alloc(n * t.CDouble.__sizeof__()) + if not a.data: + pool.free(a) + return None + memset(a.data, 0, n * t.CDouble.__sizeof__()) + a.pool = pool + a.ndim = 1 + a.size = n + a.shape[0] = n + a.strides[0] = 1 + a.owns_data = 1 + return a + +def ones(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = zeros(pool, n) + if not a: return None + a.fill(t.CDouble(1.0)) + return a + +def full(pool: memhub.MemManager | t.CPtr, n: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = zeros(pool, n) + if not a: return None + a.fill(val) + return a + +def arange(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, step: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + if step == t.CDouble(0.0): return None + n: t.CSizeT = 0 + if step > t.CDouble(0.0): + v: t.CDouble = start + while v < stop: + n += 1 + v += step + else: + v: t.CDouble = start + while v > stop: + n += 1 + v += step + if n == 0: return None + a: ndarray[t.CDouble] | t.CPtr = zeros(pool, n) + if not a: return None + i: t.CSizeT = 0 + v = start + while i < n: + a.data[i] = v + v += step + i += 1 + return a + +def linspace(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble, num: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + if num == 0: return None + a: ndarray[t.CDouble] | t.CPtr = zeros(pool, num) + if not a: return None + if num == 1: + a.data[0] = start + return a + step: t.CDouble = (stop - start) / t.CDouble(num - 1) + i: t.CSizeT = 0 + while i < num: + a.data[i] = start + t.CDouble(i) * step + i += 1 + return a + +def empty2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = _alloc_ndarray(pool) + if not a: return None + total: t.CSizeT = rows * cols + a.data = pool.alloc(total * t.CDouble.__sizeof__()) + if not a.data: + pool.free(a) + return None + memset(a.data, 0, total * t.CDouble.__sizeof__()) + a.pool = pool + a.ndim = 2 + a.size = total + a.shape[0] = rows + a.shape[1] = cols + a.owns_data = 1 + _compute_strides(a) + return a + +def zeros2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + return empty2d(pool, rows, cols) + +def ones2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = empty2d(pool, rows, cols) + if not a: return None + a.fill(t.CDouble(1.0)) + return a + +def eye(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = empty2d(pool, n, n) + if not a: return None + i: t.CSizeT = 0 + while i < n: + a.data[i * n + i] = t.CDouble(1.0) + i += 1 + return a + +def diag(pool: memhub.MemManager | t.CPtr, vals: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + n: t.CSizeT = vals.size + a: ndarray[t.CDouble] | t.CPtr = empty2d(pool, n, n) + if not a: return None + i: t.CSizeT = 0 + while i < n: + a.data[i * n + i] = vals.data[i] + i += 1 + return a + + +# ============================================================ +# Mathematical functions (element-wise, float64 default) +# ============================================================ +def np_abs(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + v: t.CDouble = a.data[i] + result.data[i] = v if v >= t.CDouble(0.0) else -v + i += 1 + return result + +def np_sqrt(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.sqrt(a.data[i]) + i += 1 + return result + +def np_exp(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.exp(a.data[i]) + i += 1 + return result + +def np_log(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.log(a.data[i]) + i += 1 + return result + +def np_sin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.sin(a.data[i]) + i += 1 + return result + +def np_cos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.cos(a.data[i]) + i += 1 + return result + +def np_tan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.tan(a.data[i]) + i += 1 + return result + +def np_pow(a: ndarray[t.CDouble] | t.CPtr, p: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.pow(a.data[i], p) + i += 1 + return result + + +# ============================================================ +# Scalar operations (float64 default) +# ============================================================ +def add_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[i] + s + i += 1 + return result + +def mul_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[i] * s + i += 1 + return result + +def sub_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[i] - s + i += 1 + return result + +def div_scalar(a: ndarray[t.CDouble] | t.CPtr, s: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + if s != t.CDouble(0.0): + result.data[i] = a.data[i] / s + else: + result.data[i] = t.CDouble(0.0) + i += 1 + return result + + +# ============================================================ +# Matrix operations (float64 default) +# ============================================================ +def matmul(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + # 2D matrix multiply: a(m,k) x b(k,n) = c(m,n) + if a.ndim != 2 or b.ndim != 2: return None + m: t.CSizeT = a.shape[0] + k: t.CSizeT = a.shape[1] + k2: t.CSizeT = b.shape[0] + n: t.CSizeT = b.shape[1] + if k != k2: return None + c: ndarray[t.CDouble] | t.CPtr = empty2d(a.pool, m, n) + if not c: return None + i: t.CSizeT = 0 + while i < m: + j: t.CSizeT = 0 + while j < n: + s: t.CDouble = t.CDouble(0.0) + p: t.CSizeT = 0 + while p < k: + s += a.data[i * k + p] * b.data[p * n + j] + p += 1 + c.data[i * n + j] = s + j += 1 + i += 1 + return c + +def dot_product(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return a.dot(b) + + +# ============================================================ +# Reduction operations (float64 default) +# ============================================================ +def np_sum(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return a.sum() + +def np_mean(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return a.mean() + +def np_min(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return a.min() + +def np_max(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return a.max() + +def np_argmax(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: + return a.argmax() + +def np_argmin(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: + return a.argmin() + + +# ============================================================ +# Utility (float64 default) +# ============================================================ +def var(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + if a.size == 0: return t.CDouble(0.0) + m: t.CDouble = a.mean() + s: t.CDouble = t.CDouble(0.0) + i: t.CSizeT = 0 + while i < a.size: + d: t.CDouble = a.data[i] - m + s += d * d + i += 1 + return s / t.CDouble(a.size) + +def std(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return vipermath.sqrt(var(a)) + +def norm(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + return vipermath.sqrt(a.dot(a)) + +def clip(a: ndarray[t.CDouble] | t.CPtr, lo: t.CDouble, hi: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + v: t.CDouble = a.data[i] + if v < lo: v = lo + if v > hi: v = hi + result.data[i] = v + i += 1 + return result + +def concatenate(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + if a.ndim != 1 or b.ndim != 1: return None + total: t.CSizeT = a.size + b.size + result: ndarray[t.CDouble] | t.CPtr = zeros(a.pool, total) + if not result: return None + memcpy(result.data, a.data, a.size * t.CDouble.__sizeof__()) + memcpy(result.data + a.size, b.data, b.size * t.CDouble.__sizeof__()) + return result + +def sort_arr(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = a.copy() + if not result: return None + # Simple insertion sort + i: t.CSizeT = 1 + while i < result.size: + key: t.CDouble = result.data[i] + j: t.CSizeT = i - 1 + while j < result.size and result.data[j] > key: + result.data[j + 1] = result.data[j] + if j == 0: break + j -= 1 + if j > 0 or result.data[0] <= key: + result.data[j + 1] = key + else: + result.data[1] = result.data[0] + result.data[0] = key + i += 1 + return result + +def reverse(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[a.size - 1 - i] + i += 1 + return result + + +# ============================================================ +# Additional math functions (element-wise, float64 default) +# ============================================================ +def np_log10(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.log10(a.data[i]) + i += 1 + return result + +def np_log2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.log2(a.data[i]) + i += 1 + return result + +def np_floor(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.floor(a.data[i]) + i += 1 + return result + +def np_ceil(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.ceil(a.data[i]) + i += 1 + return result + +def np_round(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.round(a.data[i]) + i += 1 + return result + +def np_sign(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + if a.data[i] > t.CDouble(0.0): + result.data[i] = t.CDouble(1.0) + elif a.data[i] < t.CDouble(0.0): + result.data[i] = t.CDouble(-1.0) + else: + result.data[i] = t.CDouble(0.0) + i += 1 + return result + +def np_tanh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.tanh(a.data[i]) + i += 1 + return result + +def np_sinh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.sinh(a.data[i]) + i += 1 + return result + +def np_cosh(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.cosh(a.data[i]) + i += 1 + return result + +def np_arcsin(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.asin(a.data[i]) + i += 1 + return result + +def np_arccos(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.acos(a.data[i]) + i += 1 + return result + +def np_arctan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.atan(a.data[i]) + i += 1 + return result + +def np_arctan2(y: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(y) + if not result: return None + i: t.CSizeT = 0 + while i < y.size: + result.data[i] = vipermath.atan2(y.data[i], x.data[i]) + i += 1 + return result + +def np_degrees(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.degrees(a.data[i]) + i += 1 + return result + +def np_radians(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = vipermath.radians(a.data[i]) + i += 1 + return result + +def np_isnan(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(vipermath.isnan(a.data[i])) + i += 1 + return result + +def np_isinf(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(vipermath.isinf(a.data[i])) + i += 1 + return result + +def np_maximum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[i] if a.data[i] > b.data[i] else b.data[i] + i += 1 + return result + +def np_minimum(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = a.data[i] if a.data[i] < b.data[i] else b.data[i] + i += 1 + return result + + +# ============================================================ +# Additional array operations (float64 default) +# ============================================================ +def cumsum(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + s: t.CDouble = t.CDouble(0.0) + i: t.CSizeT = 0 + while i < a.size: + s += a.data[i] + result.data[i] = s + i += 1 + return result + +def diff(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + if a.size < 2: return None + result: ndarray[t.CDouble] | t.CPtr = zeros(a.pool, a.size - 1) + if not result: return None + i: t.CSizeT = 0 + while i < a.size - 1: + result.data[i] = a.data[i + 1] - a.data[i] + i += 1 + return result + +def flatten(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + result.ndim = 1 + result.shape[0] = a.size + result.strides[0] = 1 + memcpy(result.data, a.data, a.size * t.CDouble.__sizeof__()) + return result + +def trace(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + if a.ndim != 2: return t.CDouble(0.0) + n: t.CSizeT = a.shape[0] if a.shape[0] < a.shape[1] else a.shape[1] + s: t.CDouble = t.CDouble(0.0) + i: t.CSizeT = 0 + while i < n: + s += a.data[i * a.shape[1] + i] + i += 1 + return s + +def outer(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = empty2d(a.pool, a.size, b.size) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + j: t.CSizeT = 0 + while j < b.size: + result.data[i * b.size + j] = a.data[i] * b.data[j] + j += 1 + i += 1 + return result + +def np_where(condition: ndarray[t.CDouble] | t.CPtr, x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(x) + if not result: return None + i: t.CSizeT = 0 + while i < x.size: + if condition.data[i] != t.CDouble(0.0): + result.data[i] = x.data[i] + else: + result.data[i] = y.data[i] + i += 1 + return result + +def np_count_nonzero(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: + count: t.CInt = 0 + i: t.CSizeT = 0 + while i < a.size: + if a.data[i] != t.CDouble(0.0): + count += 1 + i += 1 + return count + +def np_all(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: + i: t.CSizeT = 0 + while i < a.size: + if a.data[i] == t.CDouble(0.0): + return 0 + i += 1 + return 1 + +def np_any(a: ndarray[t.CDouble] | t.CPtr) -> t.CInt: + i: t.CSizeT = 0 + while i < a.size: + if a.data[i] != t.CDouble(0.0): + return 1 + i += 1 + return 0 + +def np_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] == b.data[i] else t.CDouble(0.0) + i += 1 + return result + +def np_not_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] != b.data[i] else t.CDouble(0.0) + i += 1 + return result + +def np_less(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] < b.data[i] else t.CDouble(0.0) + i += 1 + return result + +def np_greater(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] > b.data[i] else t.CDouble(0.0) + i += 1 + return result + +def np_less_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] <= b.data[i] else t.CDouble(0.0) + i += 1 + return result + +def np_greater_equal(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + i: t.CSizeT = 0 + while i < a.size: + result.data[i] = t.CDouble(1.0) if a.data[i] >= b.data[i] else t.CDouble(0.0) + i += 1 + return result + + +# ============================================================ +# Additional creation functions (float64 default) +# ============================================================ +def empty(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> ndarray[t.CDouble] | t.CPtr: + return zeros(pool, n) + +def full2d(pool: memhub.MemManager | t.CPtr, rows: t.CSizeT, cols: t.CSizeT, val: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + a: ndarray[t.CDouble] | t.CPtr = empty2d(pool, rows, cols) + if not a: return None + a.fill(val) + return a + +def zeros_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _alloc_ndarray(a.pool) + if not result: return None + result.data = a.pool.alloc(a.size * t.CDouble.__sizeof__()) + if not result.data: + a.pool.free(result) + return None + memset(result.data, 0, a.size * t.CDouble.__sizeof__()) + result.pool = a.pool + result.ndim = a.ndim + result.size = a.size + result.owns_data = 1 + i: t.CInt = 0 + for i in range(a.ndim): + result.shape[i] = a.shape[i] + _compute_strides(result) + return result + +def ones_like(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + result: ndarray[t.CDouble] | t.CPtr = _empty_like(a) + if not result: return None + result.fill(t.CDouble(1.0)) + return result + +def arange1(pool: memhub.MemManager | t.CPtr, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + return arange(pool, t.CDouble(0.0), stop, t.CDouble(1.0)) + +def linspace2(pool: memhub.MemManager | t.CPtr, start: t.CDouble, stop: t.CDouble) -> ndarray[t.CDouble] | t.CPtr: + return linspace(pool, start, stop, 50) + +def meshgrid(x: ndarray[t.CDouble] | t.CPtr, y: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + # Returns flattened grid pairs - simplified version + # Returns X grid as 2D array (each row is x) + result: ndarray[t.CDouble] | t.CPtr = empty2d(x.pool, y.size, x.size) + if not result: return None + i: t.CSizeT = 0 + while i < y.size: + j: t.CSizeT = 0 + while j < x.size: + result.data[i * x.size + j] = x.data[j] + j += 1 + i += 1 + return result + + +# ============================================================ +# Linear algebra helpers (float64 default) +# ============================================================ +def det2x2(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + # Determinant of 2x2 matrix + if a.ndim != 2: return t.CDouble(0.0) + if a.shape[0] != 2 or a.shape[1] != 2: return t.CDouble(0.0) + return a.data[0] * a.data[3] - a.data[1] * a.data[2] + +def inv2x2(a: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + # Inverse of 2x2 matrix + d: t.CDouble = det2x2(a) + if d == t.CDouble(0.0): return None + result: ndarray[t.CDouble] | t.CPtr = empty2d(a.pool, 2, 2) + if not result: return None + result.data[0] = a.data[3] / d + result.data[1] = -a.data[1] / d + result.data[2] = -a.data[2] / d + result.data[3] = a.data[0] / d + return result + +def cross3(a: ndarray[t.CDouble] | t.CPtr, b: ndarray[t.CDouble] | t.CPtr) -> ndarray[t.CDouble] | t.CPtr: + # Cross product of 3D vectors + result: ndarray[t.CDouble] | t.CPtr = zeros(a.pool, 3) + if not result: return None + result.data[0] = a.data[1] * b.data[2] - a.data[2] * b.data[1] + result.data[1] = a.data[2] * b.data[0] - a.data[0] * b.data[2] + result.data[2] = a.data[0] * b.data[1] - a.data[1] * b.data[0] + return result + +def np_interp(x: t.CDouble, xp: ndarray[t.CDouble] | t.CPtr, fp: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + # 1D linear interpolation + if xp.size == 0: return t.CDouble(0.0) + if x <= xp.data[0]: return fp.data[0] + if x >= xp.data[xp.size - 1]: return fp.data[xp.size - 1] + i: t.CSizeT = 0 + while i < xp.size - 1: + if x >= xp.data[i] and x <= xp.data[i + 1]: + t_val: t.CDouble = (x - xp.data[i]) / (xp.data[i + 1] - xp.data[i]) + return fp.data[i] + t_val * (fp.data[i + 1] - fp.data[i]) + i += 1 + return fp.data[xp.size - 1] + +def prod(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + p: t.CDouble = t.CDouble(1.0) + i: t.CSizeT = 0 + while i < a.size: + p *= a.data[i] + i += 1 + return p + +def median(a: ndarray[t.CDouble] | t.CPtr) -> t.CDouble: + if a.size == 0: return t.CDouble(0.0) + s: ndarray[t.CDouble] | t.CPtr = sort_arr(a) + if not s: return t.CDouble(0.0) + mid: t.CSizeT = a.size / 2 + if a.size % 2 == 1: + result: t.CDouble = s.data[mid] + else: + result = (s.data[mid - 1] + s.data[mid]) / t.CDouble(2.0) + s.delete() + return result + +def percentile(a: ndarray[t.CDouble] | t.CPtr, q: t.CDouble) -> t.CDouble: + if a.size == 0: return t.CDouble(0.0) + s: ndarray[t.CDouble] | t.CPtr = sort_arr(a) + if not s: return t.CDouble(0.0) + idx: t.CDouble = q / t.CDouble(100.0) * t.CDouble(a.size - 1) + lo: t.CSizeT = t.CSizeT(vipermath.floor(idx)) + hi: t.CSizeT = t.CSizeT(vipermath.ceil(idx)) + frac: t.CDouble = idx - vipermath.floor(idx) + result: t.CDouble = s.data[lo] + frac * (s.data[hi] - s.data[lo]) + s.delete() + return result diff --git a/includes/os/__init__.py b/includes/os/__init__.py new file mode 100644 index 0000000..a73046c --- /dev/null +++ b/includes/os/__init__.py @@ -0,0 +1,133 @@ +import t, c +from stdint import * +import platmacro +import memhub +import os.path +import os._win32 +import os._posix + + +# os 包入口 +# 根据 platmacro.IS_WINDOWS 分派到 _win32 或 _posix 后端 +# os.path 为跨平台纯字符串操作，直接可用 + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# 文件/目录属性查询 +# ============================================================================= + +def exists(path: str) -> bool: + """检查文件/目录是否存在""" + if platmacro.IS_WINDOWS: + return os._win32.exists(path) + return os._posix.exists(path) + + +def isdir(path: str) -> bool: + """检查路径是否为目录""" + if platmacro.IS_WINDOWS: + return os._win32.isdir(path) + return os._posix.isdir(path) + + +def isfile(path: str) -> bool: + """检查路径是否为普通文件""" + if platmacro.IS_WINDOWS: + return os._win32.isfile(path) + return os._posix.isfile(path) + + +def getsize(path: str) -> t.CInt64T: + """返回文件大小（字节）。失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.getsize(path) + return os._posix.getsize(path) + + +# ============================================================================= +# 工作目录 +# ============================================================================= + +def getcwd() -> str: + """返回当前工作目录（通过 _mbuddy 分配内存）""" + if platmacro.IS_WINDOWS: + return os._win32.getcwd() + return os._posix.getcwd_impl() + + +def chdir(path: str) -> int: + """切换当前工作目录。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.chdir(path) + return os._posix.chdir_impl(path) + + +# ============================================================================= +# 文件/目录操作 +# ============================================================================= + +def mkdir(path: str) -> int: + """创建目录。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.mkdir(path) + return os._posix.mkdir_impl(path) + + +def remove(path: str) -> int: + """删除文件。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.remove(path) + return os._posix.remove(path) + + +def rmdir(path: str) -> int: + """删除空目录。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.rmdir(path) + return os._posix.rmdir_impl(path) + + +def rename(old_path: str, new_path: str) -> int: + """重命名/移动文件或目录。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return os._win32.rename(old_path, new_path) + return os._posix.rename_impl(old_path, new_path) + + +def chmod(path: str, mode: int) -> int: + """修改文件权限。成功返回 0，失败返回 -1。 + Windows 上只处理只读位（S_IWUSR）；POSIX 上使用完整权限位。""" + if platmacro.IS_WINDOWS: + return os._win32.chmod(path, mode) + return os._posix.chmod_impl(path, mode) + + +# ============================================================================= +# 目录列举 +# ============================================================================= + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: + """列举目录下的文件/子目录名。 + out_names 为调用者提供的字符串指针数组（char**），每个元素指向一个文件名。 + 文件名通过 _mbuddy 分配内存。 + max_count 为 out_names 数组最大容量。 + 返回实际列举的条目数。失败返回 0。""" + if platmacro.IS_WINDOWS: + return os._win32.listdir(path, out_names, max_count) + return os._posix.listdir(path, out_names, max_count) + + +# ============================================================================= +# 命令执行 +# ============================================================================= + +def system(cmd: str) -> int: + """执行命令行命令（通过 C 运行时 system()）。 + 命令输出直接到终端（继承父进程 stdout/stderr）。 + 返回命令的退出码（0 表示成功）。""" + if platmacro.IS_WINDOWS: + return os._win32.system(cmd) + return os._posix.system_impl(cmd) diff --git a/includes/os/_posix.py b/includes/os/_posix.py new file mode 100644 index 0000000..f5d9114 --- /dev/null +++ b/includes/os/_posix.py @@ -0,0 +1,190 @@ +import t, c +from stdint import * +import posix +import memhub +import stdlib + + +# POSIX 路径最大长度 +MAX_PATH_LEN: t.CDefine = 4096 + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# 文件/目录属性查询 +# ============================================================================= + +def _stat(path: str) -> posix.Stat | t.CPtr: + """调用 stat 并返回 Stat 结构体指针。失败返回 None。 + 注意：Stat 结构体通过 _mbuddy 分配，调用者负责释放。""" + st: posix.Stat = posix.Stat() + result: INT = posix.stat(path, t.CPtr(c.Addr(st))) + if result != 0: + return None + return t.CPtr(c.Addr(st)) + + +def exists(path: str) -> bool: + """检查文件/目录是否存在""" + result: INT = posix.access(path, posix.F_OK) + if result == 0: + return True + return False + + +def isdir(path: str) -> bool: + """检查路径是否为目录""" + st: posix.Stat = posix.Stat() + result: INT = posix.stat(path, t.CPtr(c.Addr(st))) + if result != 0: + return False + if posix.S_ISDIR(st.st_mode): + return True + return False + + +def isfile(path: str) -> bool: + """检查路径是否为普通文件""" + st: posix.Stat = posix.Stat() + result: INT = posix.stat(path, t.CPtr(c.Addr(st))) + if result != 0: + return False + if posix.S_ISREG(st.st_mode): + return True + return False + + +def getsize(path: str) -> t.CInt64T: + """返回文件大小（字节）。失败返回 -1。""" + st: posix.Stat = posix.Stat() + result: INT = posix.stat(path, t.CPtr(c.Addr(st))) + if result != 0: + return -1 + return st.st_size + + +# ============================================================================= +# 工作目录 +# ============================================================================= + +def getcwd_impl() -> str: + """返回当前工作目录（通过 _mbuddy 分配内存）""" + buf: bytes = _mbuddy.alloc(MAX_PATH_LEN) + if buf == None: return None + result: CHARPTR = posix.getcwd(buf, MAX_PATH_LEN) + if result == None: + return None + return str(buf) + + +def chdir_impl(path: str) -> int: + """切换当前工作目录。成功返回 0，失败返回 -1。""" + result: INT = posix.chdir(path) + if result != 0: + return -1 + return 0 + + +# ============================================================================= +# 文件/目录操作 +# ============================================================================= + +def mkdir_impl(path: str) -> int: + """创建目录。成功返回 0，失败返回 -1。 + 默认权限 0o755（rwxr-xr-x）。""" + result: INT = posix.mkdir(path, 0o755) + if result != 0: + return -1 + return 0 + + +def remove(path: str) -> int: + """删除文件。成功返回 0，失败返回 -1。""" + result: INT = posix.unlink(path) + if result != 0: + return -1 + return 0 + + +def rmdir_impl(path: str) -> int: + """删除空目录。成功返回 0，失败返回 -1。""" + result: INT = posix.rmdir(path) + if result != 0: + return -1 + return 0 + + +def rename_impl(old_path: str, new_path: str) -> int: + """重命名/移动文件或目录。成功返回 0，失败返回 -1。""" + result: INT = posix.rename(old_path, new_path) + if result != 0: + return -1 + return 0 + + +def chmod_impl(path: str, mode: int) -> int: + """修改文件权限。成功返回 0，失败返回 -1。 + mode 为 POSIX 权限位（如 0o755）。""" + result: INT = posix.chmod(path, posix.mode_t(mode)) + if result != 0: + return -1 + return 0 + + +# ============================================================================= +# 目录列举（opendir / readdir / closedir） +# ============================================================================= + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: + """列举目录下的文件/子目录名。 + out_names 为调用者提供的字符串指针数组（char**），每个元素指向一个文件名。 + 文件名通过 _mbuddy 分配内存。 + max_count 为 out_names 数组最大容量。 + 返回实际列举的条目数。失败返回 0。""" + dirp: posix.DIR | t.CPtr = posix.opendir(path) + if dirp == None: + return 0 + + count: ULONG = 0 + while count < max_count: + entry: posix.Dirent | t.CPtr = posix.readdir(dirp) + if entry == None: + break + + # 读取 d_name 字段 + name: str = entry.d_name + if name[0] == '.': + if name[1] == 0: + # "." + continue + if name[1] == '.' and name[2] == 0: + # ".." + continue + + # 复制文件名到新分配的内存 + name_len: t.CSizeT = len(name) + name_buf: bytes = _mbuddy.alloc(name_len + 1) + if name_buf == None: break + i: t.CSizeT = 0 + for ch in name: + name_buf[i] = ch + i += 1 + name_buf[name_len] = 0 + out_names[count] = str(name_buf) + count += 1 + + posix.closedir(dirp) + return count + + +# ============================================================================= +# 命令执行 +# ============================================================================= + +def system_impl(cmd: str) -> int: + """执行命令行命令（通过 C 运行时 system()）。 + 命令输出直接到子进程的终端（继承父进程 stdout/stderr）。 + 返回命令的退出码（0 表示成功）。""" + return stdlib.system(cmd) diff --git a/includes/os/_win32.py b/includes/os/_win32.py new file mode 100644 index 0000000..6292e53 --- /dev/null +++ b/includes/os/_win32.py @@ -0,0 +1,234 @@ +import t, c +from stdint import * +import w32.win32base +import w32.win32file +import memhub +import stdlib + + +# Windows 路径最大长度 +MAX_PATH_LEN: t.CDefine = 260 + +# INVALID_FILE_ATTRIBUTES +INVALID_FILE_ATTRIBUTES: t.CDefine = 0xFFFFFFFF + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# 文件/目录属性查询 +# ============================================================================= + +def _get_attrs(path: str) -> ULONG: + """获取文件属性。失败返回 INVALID_FILE_ATTRIBUTES。""" + return w32.win32file.GetFileAttributesA(path) + + +def exists(path: str) -> bool: + """检查文件/目录是否存在""" + attrs: ULONG = _get_attrs(path) + if attrs == INVALID_FILE_ATTRIBUTES: + return False + return True + + +def isdir(path: str) -> bool: + """检查路径是否为目录""" + attrs: ULONG = _get_attrs(path) + if attrs == INVALID_FILE_ATTRIBUTES: + return False + if (attrs & w32.win32file.FILE_ATTRIBUTE_DIRECTORY) != 0: + return True + return False + + +def isfile(path: str) -> bool: + """检查路径是否为普通文件""" + attrs: ULONG = _get_attrs(path) + if attrs == INVALID_FILE_ATTRIBUTES: + return False + if (attrs & w32.win32file.FILE_ATTRIBUTE_DIRECTORY) != 0: + return False + return True + + +def getsize(path: str) -> t.CInt64T: + """返回文件大小（字节）。失败返回 -1。""" + handle: w32.win32base.HANDLE = w32.win32file.CreateFileA( + path, w32.win32file.GENERIC_READ, w32.win32file.FILE_SHARE_READ, + t.CPtr(0), w32.win32file.OPEN_EXISTING, w32.win32file.FILE_ATTRIBUTE_NORMAL, t.CPtr(0) + ) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + return -1 + li: w32.win32base.LARGE_INTEGER = w32.win32base.LARGE_INTEGER() + result: BOOL = w32.win32file.GetFileSizeEx(handle, t.CPtr(c.Addr(li))) + w32.win32base.CloseHandle(handle) + if result == 0: + return -1 + return li.QuadPart + + +# ============================================================================= +# 工作目录 +# ============================================================================= + +def getcwd() -> str: + """返回当前工作目录（通过 _mbuddy 分配内存）""" + buf: bytes = _mbuddy.alloc(MAX_PATH_LEN + 1) + if buf == None: return None + length: ULONG = w32.win32file.GetCurrentDirectoryA(MAX_PATH_LEN + 1, buf) + if length == 0: + return None + if length > MAX_PATH_LEN: + # 缓冲区不够，重新分配 + buf2: bytes = _mbuddy.alloc(length + 1) + if buf2 == None: return None + w32.win32file.GetCurrentDirectoryA(length + 1, buf2) + buf2[length] = 0 + return str(buf2) + buf[length] = 0 + return str(buf) + + +def chdir(path: str) -> int: + """切换当前工作目录。成功返回 0，失败返回 -1。""" + result: BOOL = w32.win32file.SetCurrentDirectoryA(path) + if result == 0: + return -1 + return 0 + + +# ============================================================================= +# 文件/目录操作 +# ============================================================================= + +def mkdir(path: str) -> int: + """创建目录。成功返回 0，失败返回 -1。""" + result: BOOL = w32.win32file.CreateDirectoryA(path, t.CPtr(0)) + if result == 0: + return -1 + return 0 + + +def remove(path: str) -> int: + """删除文件。成功返回 0，失败返回 -1。""" + result: BOOL = w32.win32file.DeleteFileA(path) + if result == 0: + return -1 + return 0 + + +def rmdir(path: str) -> int: + """删除空目录。成功返回 0，失败返回 -1。""" + result: BOOL = w32.win32file.RemoveDirectoryA(path) + if result == 0: + return -1 + return 0 + + +def rename(old_path: str, new_path: str) -> int: + """重命名/移动文件或目录。成功返回 0，失败返回 -1。""" + result: BOOL = w32.win32file.MoveFileExA( + old_path, new_path, + w32.win32file.MOVEFILE_REPLACE_EXISTING | w32.win32file.MOVEFILE_COPY_ALLOWED + ) + if result == 0: + return -1 + return 0 + + +def chmod(path: str, mode: int) -> int: + """修改文件属性（Windows 简化版：只处理只读位）。 + mode 的 S_IWUSR 位（0o200）为 0 时设为只读，为 1 时设为可写。 + 成功返回 0，失败返回 -1。""" + if (mode & 0o200) != 0: + # 可写：清除只读属性 + result: BOOL = w32.win32file.SetFileAttributesA( + path, w32.win32file.FILE_ATTRIBUTE_NORMAL + ) + else: + # 只读：设置只读属性 + result: BOOL = w32.win32file.SetFileAttributesA( + path, w32.win32file.FILE_ATTRIBUTE_READONLY + ) + if result == 0: + return -1 + return 0 + + +# ============================================================================= +# 目录列举（FindFirstFile / FindNextFile） +# ============================================================================= + +def listdir(path: str, out_names: str, max_count: ULONG) -> ULONG: + """列举目录下的文件/子目录名。 + out_names 为调用者提供的字符串指针数组（char**），每个元素指向一个文件名。 + 文件名通过 _mbuddy 分配内存。 + max_count 为 out_names 数组最大容量。 + 返回实际列举的条目数。失败返回 0。""" + # 构造搜索模式：path\* + path_len: t.CSizeT = len(path) + pattern: bytes = _mbuddy.alloc(path_len + 3) + if pattern == None: return 0 + i: t.CSizeT = 0 + for ch in path: + pattern[i] = ch + i += 1 + # 添加 \* 后缀 + if path[path_len - 1] != '\\' and path[path_len - 1] != '/': + pattern[i] = '\\' + i += 1 + pattern[i] = '*' + i += 1 + pattern[i] = 0 + + find_data: w32.win32file.WIN32_FIND_DATAA = w32.win32file.WIN32_FIND_DATAA() + handle: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(pattern, t.CPtr(c.Addr(find_data))) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + return 0 + + count: ULONG = 0 + while count < max_count: + # 跳过 . 和 .. + name: str = find_data.cFileName + if name[0] == '.': + if name[1] == 0: + # "." + if w32.win32file.FindNextFileA(handle, t.CPtr(c.Addr(find_data))) == 0: + break + continue + if name[1] == '.' and name[2] == 0: + # ".." + if w32.win32file.FindNextFileA(handle, t.CPtr(c.Addr(find_data))) == 0: + break + continue + + # 复制文件名到新分配的内存 + name_len: t.CSizeT = len(name) + name_buf: bytes = _mbuddy.alloc(name_len + 1) + if name_buf == None: break + j: t.CSizeT = 0 + for ch in name: + name_buf[j] = ch + j += 1 + name_buf[name_len] = 0 + out_names[count] = str(name_buf) + count += 1 + + if w32.win32file.FindNextFileA(handle, t.CPtr(c.Addr(find_data))) == 0: + break + + w32.win32file.FindClose(handle) + return count + + +# ============================================================================= +# 命令执行 +# ============================================================================= + +def system(cmd: str) -> int: + """执行命令行命令（通过 C 运行时 system()）。 + 命令输出直接到子进程的终端（继承父进程 stdout/stderr）。 + 返回命令的退出码（0 表示成功）。""" + return stdlib.system(cmd) diff --git a/includes/os/path.py b/includes/os/path.py new file mode 100644 index 0000000..af38b91 --- /dev/null +++ b/includes/os/path.py @@ -0,0 +1,316 @@ +import t, c +from stdint import * +import string +import memhub +import platmacro + + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# 路径分隔符 +# ============================================================================= + +def sep() -> t.CChar: + """返回当前平台的路径分隔符（Windows: \\，POSIX: /）""" + if platmacro.IS_WINDOWS: + return '\\' + return '/' + + +def _is_sep(ch: t.CChar) -> bool: + """判断字符是否为路径分隔符（/ 或 \\，跨平台识别）""" + return ch == '/' or ch == '\\' + + +# ============================================================================= +# 内部辅助函数 +# ============================================================================= + +def _dup_str(s: str) -> str: + """复制字符串（通过全局 MBuddy 分配内存）""" + if s == None: return None + length: t.CSizeT = len(s) + buf: bytes = _mbuddy.alloc(length + 1) + if buf == None: return None + i: t.CSizeT = 0 + for ch in s: + buf[i] = ch + i += 1 + buf[length] = 0 + return str(buf) + + +def _substr(s: str, start: t.CSizeT, end: t.CSizeT) -> str: + """提取子串 s[start:end]（通过全局 MBuddy 分配内存）""" + if s == None: return None + if start >= end: return _dup_str("") + result_len: t.CSizeT = end - start + buf: bytes = _mbuddy.alloc(result_len + 1) + if buf == None: return None + i: t.CSizeT = 0 + k: t.CSizeT = start + while k < end: + buf[i] = s[k] + i += 1 + k += 1 + buf[i] = 0 + return str(buf) + + +# ============================================================================= +# 路径判断 +# ============================================================================= + +def isabs(path: str) -> bool: + """判断路径是否为绝对路径。 + POSIX: 以 / 开头。 + Windows: 以 \\ 开头（UNC）或 X:\\ / X:/ 开头（盘符）。""" + if path == None: return False + if path[0] == 0: return False + if path[0] == '/' or path[0] == '\\': return True + if platmacro.IS_WINDOWS: + # 检查 X:\ 或 X:/ 模式 + c0: t.CChar = path[0] + if (c0 >= 'A' and c0 <= 'Z') or (c0 >= 'a' and c0 <= 'z'): + if path[1] == ':': + if path[2] == '\\' or path[2] == '/': + return True + return False + + +# ============================================================================= +# 路径拼接与分割 +# ============================================================================= + +def join(path1: str, path2: str) -> str: + """拼接两个路径。如果 path2 是绝对路径，返回 path2 的副本。""" + if path1 == None or path1[0] == 0: + return _dup_str(path2) + if path2 == None or path2[0] == 0: + return _dup_str(path1) + if isabs(path2): + return _dup_str(path2) + + len1: t.CSizeT = len(path1) + len2: t.CSizeT = len(path2) + + # 检查 path1 是否以分隔符结尾 + need_sep: bool = True + if _is_sep(path1[len1 - 1]): + need_sep = False + + total: t.CSizeT = len1 + len2 + 1 + if need_sep: + total += 1 + + buf: bytes = _mbuddy.alloc(total) + if buf == None: return None + + # 复制 path1 + i: t.CSizeT = 0 + for ch in path1: + buf[i] = ch + i += 1 + + # 添加分隔符 + if need_sep: + buf[i] = sep() + i += 1 + + # 复制 path2 + for ch in path2: + buf[i] = ch + i += 1 + + buf[i] = 0 + return str(buf) + + +def basename(path: str) -> str: + """返回路径的文件名部分（最后一个分隔符之后的内容）""" + if path == None: return None + length: t.CSizeT = len(path) + if length == 0: return _dup_str("") + + # 从后向前找分隔符 + i: t.CSizeT = length + while i > 0: + i -= 1 + if _is_sep(path[i]): + return _substr(path, i + 1, length) + + # 没有分隔符，返回整个路径的副本 + return _dup_str(path) + + +def dirname(path: str) -> str: + """返回路径的目录部分（最后一个分隔符之前的内容）""" + if path == None: return None + length: t.CSizeT = len(path) + if length == 0: return _dup_str("") + + # 从后向前找分隔符 + i: t.CSizeT = length + while i > 0: + i -= 1 + if _is_sep(path[i]): + if i == 0: + # 路径以分隔符开头，返回根目录 + return _dup_str("/") + return _substr(path, 0, i) + + # 没有分隔符，返回空字符串 + return _dup_str("") + + +# ============================================================================= +# 扩展名处理 +# ============================================================================= + +def splitext_name(path: str) -> str: + """返回路径去除扩展名后的主名部分。 + 扩展名从最后一个 '.' 开始（该 '.' 不能是路径最后一个组件的第一个字符）。""" + if path == None: return None + length: t.CSizeT = len(path) + if length == 0: return _dup_str("") + + # 从后向前找 '.'，遇到分隔符停止 + i: t.CSizeT = length + while i > 0: + i -= 1 + if path[i] == '.': + return _substr(path, 0, i) + if _is_sep(path[i]): + break + + # 没有扩展名，返回整个路径的副本 + return _dup_str(path) + + +def splitext_ext(path: str) -> str: + """返回路径的扩展名部分（包含 '.'）。无扩展名时返回空字符串。""" + if path == None: return None + length: t.CSizeT = len(path) + if length == 0: return _dup_str("") + + # 从后向前找 '.'，遇到分隔符停止 + i: t.CSizeT = length + while i > 0: + i -= 1 + if path[i] == '.': + return _substr(path, i, length) + if _is_sep(path[i]): + break + + # 没有扩展名，返回空字符串 + return _dup_str("") + + +# ============================================================================= +# 路径规范化 +# ============================================================================= + +def normpath(path: str) -> str: + """规范化路径：统一分隔符、消除多余的 . 和 .. 组件、合并连续分隔符。 + 注意：不解析符号链接，不调用系统 API。""" + if path == None: return None + length: t.CSizeT = len(path) + if length == 0: return _dup_str(".") + + # 分配输出缓冲区（最坏情况：长度不变 + 前缀 /） + buf: bytes = _mbuddy.alloc(length + 2) + if buf == None: return None + + out_pos: t.CSizeT = 0 + i: t.CSizeT = 0 + s: str = sep() + + # 处理绝对路径前缀 + is_absolute: bool = False + if _is_sep(path[0]): + is_absolute = True + buf[out_pos] = s + out_pos += 1 + i = 1 + + # Windows 盘符前缀（X:\ 或 X:/） + drive_prefix_len: t.CSizeT = 0 + if platmacro.IS_WINDOWS: + if length >= 3: + c0: t.CChar = path[0] + if (c0 >= 'A' and c0 <= 'Z') or (c0 >= 'a' and c0 <= 'z'): + if path[1] == ':': + if _is_sep(path[2]): + buf[out_pos] = path[0] + out_pos += 1 + buf[out_pos] = ':' + out_pos += 1 + buf[out_pos] = s + out_pos += 1 + drive_prefix_len = 3 + i = 3 + is_absolute = True + + # 逐组件处理 + while i < length: + # 跳过连续分隔符 + while i < length and _is_sep(path[i]): + i += 1 + if i >= length: + break + + # 提取组件 + comp_start: t.CSizeT = i + while i < length and not _is_sep(path[i]): + i += 1 + comp_len: t.CSizeT = i - comp_start + + # 处理 "." 组件：跳过 + if comp_len == 1 and path[comp_start] == '.': + continue + + # 处理 ".." 组件：回退一级 + if comp_len == 2 and path[comp_start] == '.' and path[comp_start + 1] == '.': + if out_pos > drive_prefix_len: + # 回退到上一个分隔符前 + out_pos -= 1 + while out_pos > drive_prefix_len and not _is_sep(buf[out_pos - 1]): + out_pos -= 1 + # 如果回退到了根目录前缀，不删除前缀 + if out_pos == drive_prefix_len and is_absolute: + out_pos = drive_prefix_len + if drive_prefix_len == 0: + # POSIX 根：保留 / + buf[out_pos] = s + out_pos += 1 + continue + + # 普通组件：复制到输出 + if out_pos > drive_prefix_len and not (drive_prefix_len == 0 and out_pos == 1 and is_absolute): + buf[out_pos] = s + out_pos += 1 + elif out_pos == 0 and not is_absolute: + pass # 第一个组件，不需要前导分隔符 + elif is_absolute and out_pos == drive_prefix_len: + # 已经有前导分隔符，不重复添加 + pass + else: + buf[out_pos] = s + out_pos += 1 + + j: t.CSizeT = comp_start + while j < comp_start + comp_len: + buf[out_pos] = path[j] + out_pos += 1 + j += 1 + + # 处理空结果 + if out_pos == 0: + buf[out_pos] = '.' + out_pos += 1 + + buf[out_pos] = 0 + return str(buf) diff --git a/includes/platmacro.py b/includes/platmacro.py new file mode 100644 index 0000000..5560617 --- /dev/null +++ b/includes/platmacro.py @@ -0,0 +1,57 @@ +import t, c + + +IS_WINDOWS: t.CDefine = 1 +IS_WIN64: t.CDefine = 1 +IS_X86_64: t.CDefine = 1 +IS_LP64: t.CDefine = 1 +IS_LITTLE_ENDIAN: t.CDefine = 1 + +IS_LINUX: t.CDefine = 0 +IS_MACOS: t.CDefine = 0 +IS_ARM64: t.CDefine = 0 +IS_BIG_ENDIAN: t.CDefine = 0 +IS_ILP32: t.CDefine = 0 + +PTR_SIZE: t.CDefine = t.CSizeT().Size +INT_SIZE: t.CDefine = 4 +LONG_SIZE: t.CDefine = t.CLong().Size +WORD_SIZE: t.CDefine = t.CSizeT().Size + +CACHE_LINE_SIZE: t.CDefine = 64 +PAGE_SIZE: t.CDefine = 4096 +STACK_ALIGN: t.CDefine = 16 +MAX_PATH_LEN: t.CDefine = 260 + +PLATFORM_NAME: t.CDefine = 1 +ARCH_NAME: t.CDefine = 1 + +WINVER: t.CDefine = 0x0A00 +_WIN32_WINNT: t.CDefine = 0x0A00 +NTDDI_VERSION: t.CDefine = 0x0A000006 + +HAS_SSE2: t.CDefine = 1 +HAS_AVX: t.CDefine = 1 +HAS_AVX2: t.CDefine = 1 +HAS_FMA: t.CDefine = 1 +HAS_NEON: t.CDefine = 0 + +ALIGNOF_INT: t.CDefine = 4 +ALIGNOF_LONG: t.CDefine = t.CLong().Size // 8 +ALIGNOF_PTR: t.CDefine = t.CSizeT().Size // 8 +ALIGNOF_DOUBLE: t.CDefine = 8 +ALIGNOF_LONG_DOUBLE: t.CDefine = 16 + +SIZEOF_SHORT: t.CDefine = 2 +SIZEOF_INT: t.CDefine = 4 +SIZEOF_LONG: t.CDefine = t.CLong().Size // 8 +SIZEOF_LONG_LONG: t.CDefine = 8 +SIZEOF_FLOAT: t.CDefine = 4 +SIZEOF_DOUBLE: t.CDefine = 8 +SIZEOF_LONG_DOUBLE: t.CDefine = 16 +SIZEOF_POINTER: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_SIZE_T: t.CDefine = t.CSizeT().Size // 8 +SIZEOF_PTRDIFF_T: t.CDefine = t.CPtrDiffT().Size // 8 + +BYTE_ORDER_LE: t.CDefine = 1 +BYTE_ORDER_BE: t.CDefine = 0 diff --git a/includes/plist.py b/includes/plist.py new file mode 100644 index 0000000..dcb2fa2 --- /dev/null +++ b/includes/plist.py @@ -0,0 +1,5 @@ +import t, c + + +#class PList: +# def diff --git a/includes/posix.py b/includes/posix.py new file mode 100644 index 0000000..40ba987 --- /dev/null +++ b/includes/posix.py @@ -0,0 +1,265 @@ +import t, c +from stdint import * + + +# ============================================================================= +# POSIX 类型别名 +# 注意：ssize_t 已在 stdint.py 中定义为 t.CPtrDiffT，此处不再重复 +# ============================================================================= +DIR: t.CTypedef = t.CVoid | t.CPtr # 不透明目录流句柄 +mode_t: t.CTypedef = t.CUInt32T # 文件模式/权限位 +pid_t: t.CTypedef = t.CInt32T # 进程 ID +off_t: t.CTypedef = t.CInt64T # 文件偏移量（Linux x86_64） +dev_t: t.CTypedef = t.CUInt64T # 设备号 +ino_t: t.CTypedef = t.CUInt64T # inode 号 +nlink_t: t.CTypedef = t.CUInt64T # 硬链接数 +uid_t: t.CTypedef = t.CUInt32T # 用户 ID +gid_t: t.CTypedef = t.CUInt32T # 组 ID +blksize_t: t.CTypedef = t.CInt64T # 文件系统 I/O 块大小 +blkcnt_t: t.CTypedef = t.CInt64T # 已分配块数 +time_t: t.CTypedef = t.CInt64T # 日历时间秒数 + + +# ============================================================================= +# open() 标志位 +# ============================================================================= +O_RDONLY: t.CDefine = 0 +O_WRONLY: t.CDefine = 1 +O_RDWR: t.CDefine = 2 +O_CREAT: t.CDefine = 0o100 +O_EXCL: t.CDefine = 0o200 +O_TRUNC: t.CDefine = 0o1000 +O_APPEND: t.CDefine = 0o2000 +O_NONBLOCK: t.CDefine = 0o4000 +O_CLOEXEC: t.CDefine = 0o2000000 + + +# ============================================================================= +# access() 测试标志 +# ============================================================================= +F_OK: t.CDefine = 0 # 文件存在 +R_OK: t.CDefine = 4 # 可读 +W_OK: t.CDefine = 2 # 可写 +X_OK: t.CDefine = 1 # 可执行 + + +# ============================================================================= +# stat 结构体的 st_mode 掩码与类型位 +# ============================================================================= +S_IFMT: t.CDefine = 0o170000 # 类型掩码 +S_IFSOCK: t.CDefine = 0o140000 # 套接字 +S_IFLNK: t.CDefine = 0o120000 # 符号链接 +S_IFREG: t.CDefine = 0o100000 # 普通文件 +S_IFBLK: t.CDefine = 0o060000 # 块设备 +S_IFDIR: t.CDefine = 0o040000 # 目录 +S_IFCHR: t.CDefine = 0o020000 # 字符设备 +S_IFIFO: t.CDefine = 0o010000 # FIFO/管道 + +# 权限位 +S_ISUID: t.CDefine = 0o4000 # set-user-ID +S_ISGID: t.CDefine = 0o2000 # set-group-ID +S_ISVTX: t.CDefine = 0o1000 # sticky 位 +S_IRWXU: t.CDefine = 0o700 # 用户读写执行 +S_IRUSR: t.CDefine = 0o400 # 用户读 +S_IWUSR: t.CDefine = 0o200 # 用户写 +S_IXUSR: t.CDefine = 0o100 # 用户执行 +S_IRWXG: t.CDefine = 0o070 # 组读写执行 +S_IRGRP: t.CDefine = 0o040 # 组读 +S_IWGRP: t.CDefine = 0o020 # 组写 +S_IXGRP: t.CDefine = 0o010 # 组执行 +S_IRWXO: t.CDefine = 0o007 # 其他读写执行 +S_IROTH: t.CDefine = 0o004 # 其他读 +S_IWOTH: t.CDefine = 0o002 # 其他写 +S_IXOTH: t.CDefine = 0o001 # 其他执行 + + +# ============================================================================= +# lseek() whence 值 +# ============================================================================= +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 + + +# ============================================================================= +# 路径/名称长度限制 +# ============================================================================= +PATH_MAX: t.CDefine = 4096 +NAME_MAX: t.CDefine = 255 + + +# ============================================================================= +# sysconf() 参数 +# ============================================================================= +_SC_NPROCESSORS_ONLN: t.CDefine = 84 +_SC_PAGESIZE: t.CDefine = 30 +_SC_CLK_TCK: t.CDefine = 2 + + +# ============================================================================= +# dirent 结构体的 d_type 值 +# ============================================================================= +DT_UNKNOWN: t.CDefine = 0 +DT_FIFO: t.CDefine = 1 +DT_CHR: t.CDefine = 2 +DT_DIR: t.CDefine = 4 +DT_BLK: t.CDefine = 6 +DT_REG: t.CDefine = 8 +DT_LNK: t.CDefine = 10 +DT_SOCK: t.CDefine = 12 +DT_WHT: t.CDefine = 14 + + +# ============================================================================= +# waitpid() 选项 +# ============================================================================= +WNOHANG: t.CDefine = 1 +WUNTRACED: t.CDefine = 2 +WCONTINUED: t.CDefine = 8 + + +# ============================================================================= +# 结构体：Linux x86_64 布局 +# ============================================================================= + +# struct stat（144 字节，Linux x86_64 glibc 布局） +# 注意：st_atim/st_mtim/st_ctim 在 C 中是 struct timespec { tv_sec; tv_nsec; } +# 此处拆分为 _sec/_nsec 两个 LONGLONG 字段以保持二进制兼容 +class Stat(): + st_dev: dev_t # @ 0 + st_ino: ino_t # @ 8 + st_nlink: nlink_t # @ 16 + st_mode: mode_t # @ 24 + st_uid: uid_t # @ 28 + st_gid: gid_t # @ 32 + __pad0: INT # @ 36 + st_rdev: dev_t # @ 40 + st_size: off_t # @ 48 + st_blksize: blksize_t # @ 56 + st_blocks: blkcnt_t # @ 64 + st_atim_sec: time_t # @ 72 + st_atim_nsec: LONGLONG # @ 80 + st_mtim_sec: time_t # @ 88 + st_mtim_nsec: LONGLONG # @ 96 + st_ctim_sec: time_t # @ 104 + st_ctim_nsec: LONGLONG # @ 112 + __unused0: LONGLONG # @ 120 + __unused1: LONGLONG # @ 128 + __unused2: LONGLONG # @ 136 + + +# struct dirent（Linux x86_64，d_name 为 char[256]） +# 注意：d_name 实际是定长 char[256]，TransPyC 用 CHARPTR 占位（8 字节）。 +# 用户通常通过 readdir 返回的只读指针访问，不直接分配 Dirent 结构体。 +class Dirent(): + d_ino: ino_t # @ 0 + d_off: off_t # @ 8 + d_reclen: USHORT # @ 16 + d_type: BYTE # @ 18 + d_name: CHARPTR # @ 19（实际为 char[256]，此处占位） + + +# ============================================================================= +# 文件操作函数（10 个） +# ============================================================================= +def open(path: t.CConst | str, oflag: INT, *args) -> INT | t.State: pass +def close(fd: INT) -> INT | t.State: pass +def read(fd: INT, buf: bytes, count: t.CSizeT) -> ssize_t | t.State: pass +def write(fd: INT, buf: t.CConst | bytes, count: t.CSizeT) -> ssize_t | t.State: pass +def lseek(fd: INT, offset: off_t, whence: INT) -> off_t | t.State: pass +def unlink(path: t.CConst | str) -> INT | t.State: pass +def access(path: t.CConst | str, amode: INT) -> INT | t.State: pass +def realpath(path: t.CConst | str, resolved: CHARPTR) -> CHARPTR | t.State: pass +def rename(oldpath: t.CConst | str, newpath: t.CConst | str) -> INT | t.State: pass +def chmod(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass + + +# ============================================================================= +# stat 系列（2 个） +# ============================================================================= +def stat(path: t.CConst | str, buf: Stat | t.CPtr) -> INT | t.State: pass +def fstat(fd: INT, buf: Stat | t.CPtr) -> INT | t.State: pass + + +# ============================================================================= +# 目录操作（6 个） +# ============================================================================= +def opendir(name: t.CConst | str) -> DIR | t.CPtr | t.State: pass +def readdir(dirp: DIR | t.CPtr) -> Dirent | t.CPtr | t.State: pass +def closedir(dirp: DIR | t.CPtr) -> INT | t.State: pass +def mkdir(path: t.CConst | str, mode: mode_t) -> INT | t.State: pass +def rmdir(path: t.CConst | str) -> INT | t.State: pass +def getcwd(buf: CHARPTR, size: t.CSizeT) -> CHARPTR | t.State: pass +def chdir(path: t.CConst | str) -> INT | t.State: pass + + +# ============================================================================= +# 进程/管道（8 个） +# ============================================================================= +def fork() -> pid_t | t.State: pass +def execvp(file: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass +def execv(path: t.CConst | str, argv: CHARPTR | t.CPtr) -> INT | t.State: pass +def waitpid(pid: pid_t, status: INT | t.CPtr, options: INT) -> pid_t | t.State: pass +def pipe(fds: INT | t.CPtr) -> INT | t.State: pass +def dup2(oldfd: INT, newfd: INT) -> INT | t.State: pass +def _exit(status: INT) -> t.CVoid | t.State: pass +def getpid() -> pid_t | t.State: pass + + +# ============================================================================= +# 环境/系统（4 个） +# ============================================================================= +def getenv(name: t.CConst | str) -> CHARPTR | t.State: pass +def setenv(name: t.CConst | str, value: t.CConst | str, overwrite: INT) -> INT | t.State: pass +def sysconf(name: INT) -> LONG | t.State: pass +def isatty(fd: INT) -> INT | t.State: pass + + +# ============================================================================= +# 内联宏（C 宏，用 TransPyC 普通函数实现，不标记 t.CExtern） +# ============================================================================= + +# st_mode 类型判定 +def S_ISREG(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFREG + + +def S_ISDIR(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFDIR + + +def S_ISLNK(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFLNK + + +def S_ISCHR(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFCHR + + +def S_ISBLK(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFBLK + + +def S_ISFIFO(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFIFO + + +def S_ISSOCK(mode: mode_t) -> bool: + return (mode & S_IFMT) == S_IFSOCK + + +# waitpid 状态解析 +def WIFEXITED(status: INT) -> bool: + return (status & 0x7F) == 0 + + +def WEXITSTATUS(status: INT) -> INT: + return (status >> 8) & 0xFF + + +def WIFSIGNALED(status: INT) -> bool: + return ((status & 0x7F) != 0) and ((status & 0x7F) != 0x7F) + + +def WTERMSIG(status: INT) -> INT: + return status & 0x7F diff --git a/includes/requests.py b/includes/requests.py new file mode 100644 index 0000000..1b8f8aa --- /dev/null +++ b/includes/requests.py @@ -0,0 +1,361 @@ +import t, c +from stdint import * +import string +import socket +import memhub +import stdio + + +# ============================================================ +# requests.py — HTTP 客户端库（Python 原版风格） +# +# 用法: +# requests._mbuddy = my_mbuddy +# resp = requests.get("http://example.com/path") +# if resp is not None: +# print(resp.status_code) +# print(resp.text) +# requests.release(resp) +# ============================================================ + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值） +_mbuddy: memhub.MemBuddy | t.CPtr + +# Socket 初始化标记 +_socket_ready: bool = False + +HTTP_PORT: t.CDefine = 80 +HTTP_BUFSIZ: t.CDefine = 4096 + + +# ============================================================ +# Response — HTTP 响应对象 +# ============================================================ + +class Response: + """HTTP 响应对象，贴近 Python requests.Response 用法。 + + 从 _mbuddy 分配，使用完毕后调用 release(resp) 释放。 + """ + status_code: INT # HTTP 状态码（如 200、404） + text: str # 响应体文本（指向 _raw_buf 内部） + body_len: INT # 响应体长度 + _raw_buf: str # 原始缓冲区指针 + + +# ============================================================ +# 内部辅助函数 +# ============================================================ + +def _EnsureSocketInit() -> INT: + """确保 Socket 库已初始化（幂等）。""" + global _socket_ready + if _socket_ready: return 0 + result: INT = socket.SocketInit() + if result == 0: + _socket_ready = True + return result + + +def _AllocResponse() -> Response | t.CPtr: + """从 _mbuddy 分配 Response 结构体并就地初始化。""" + if _mbuddy is None: return None + resp: Response | t.CPtr = _mbuddy.alloc(Response.__sizeof__()) + if resp is None: return None + resp.status_code = 0 + resp.text = None + resp.body_len = 0 + resp._raw_buf = None + return resp + + +def _ParseUrl(url: str, host_buf: str, host_size: INT, path_buf: str, path_size: INT, port: INT | t.CPtr) -> INT: + """解析 URL，提取 host、port、path。""" + if url is None: return -1 + if host_buf is None or path_buf is None or port is None: return -1 + + p: str = url + + # 跳过 "http://" + if string.strncmp(p, "http://", 7) == 0: + p += 7 + + # 解析 host + host_start: str = p + host_len: INT = 0 + while p[0] != '\0' and p[0] != ':' and p[0] != '/': + host_len += 1 + p += 1 + + if host_len == 0 or host_len >= host_size: return -1 + string.strncpy(host_buf, host_start, host_len) + host_buf[host_len] = 0 + + # 解析 port + if p[0] == ':': + p += 1 + port_val: INT = 0 + while p[0] >= '0' and p[0] <= '9': + port_val = port_val * 10 + INT(p[0] - '0') + p += 1 + c.DerefAs(port, port_val) + else: + c.DerefAs(port, HTTP_PORT) + + # 解析 path + if p[0] == '/': + path_start: str = p + path_len: INT = 0 + while p[0] != '\0': + path_len += 1 + p += 1 + if path_len >= path_size: return -1 + string.strncpy(path_buf, path_start, path_len) + path_buf[path_len] = 0 + else: + if path_size < 2: return -1 + path_buf[0] = '/' + path_buf[1] = 0 + + return 0 + + +def _ParseStatusLine(resp_text: str, out_code: INT | t.CPtr) -> INT: + """解析 HTTP 状态行，提取状态码。返回状态行结束偏移。""" + if resp_text is None or out_code is None: return -1 + + p: str = resp_text + + # 跳过 "HTTP/x.x " + while p[0] != '\0' and p[0] != ' ': + p += 1 + if p[0] == ' ': p += 1 + + # 解析状态码 + code: INT = 0 + while p[0] >= '0' and p[0] <= '9': + code = code * 10 + INT(p[0] - '0') + p += 1 + + c.DerefAs(out_code, code) + + # 找到 \n + while p[0] != '\0' and p[0] != '\n': + p += 1 + if p[0] == '\n': p += 1 + + return INT(t.CUInt64T(p) - t.CUInt64T(resp_text)) + + +def _FindBodyStart(resp_text: str, resp_len: INT) -> INT: + """找到 HTTP 响应体起始位置（\\r\\n\\r\\n 之后）。""" + if resp_text is None: return -1 + + end_of_headers: str = string.strstr(resp_text, "\r\n\r\n") + if end_of_headers is None: + end_of_headers = string.strstr(resp_text, "\n\n") + if end_of_headers is None: + return -1 + return INT(t.CUInt64T(end_of_headers) - t.CUInt64T(resp_text)) + 2 + + return INT(t.CUInt64T(end_of_headers) - t.CUInt64T(resp_text)) + 4 + + +# ============================================================ +# 公共 API +# ============================================================ + +def get(url: str) -> Response | t.CPtr: + """HTTP GET 请求。 + + Args: + url: 完整 URL (如 "http://example.com/path") + Returns: + Response 对象指针（从 _mbuddy 分配），失败返回 None。 + 使用完毕后调用 release(resp) 释放。 + """ + if _mbuddy is None: return None + + resp: Response | t.CPtr = _AllocResponse() + if resp is None: return None + + if _EnsureSocketInit() != 0: + release(resp) + return None + + # 解析 URL + host_buf: t.CArray[t.CChar, 256] + path_buf: t.CArray[t.CChar, 1024] + port: INT = 0 + + if _ParseUrl(url, host_buf, 256, path_buf, 1024, c.Addr(port)) != 0: + release(resp) + return None + + # 创建并连接 Socket + sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM) + if sock.fd == socket.INVALID_SOCKET: + release(resp) + return None + + if sock.connect(host_buf, port) != 0: + sock.close() + release(resp) + return None + + # 构建 HTTP 请求 + req_buf: str = _mbuddy.alloc(HTTP_BUFSIZ) + if req_buf is None: + sock.close() + release(resp) + return None + + stdio.snprintf(req_buf, HTTP_BUFSIZ, + "GET %s HTTP/1.1\r\nHost: %s\r\nConnection: close\r\n\r\n", + path_buf, host_buf) + + req_len: INT = INT(string.strlen(req_buf)) + sent: INT = sock.send(req_buf, req_len) + _mbuddy.free(req_buf) + + if sent <= 0: + sock.close() + release(resp) + return None + + # 接收响应 + resp_buf: str = _mbuddy.alloc(HTTP_BUFSIZ) + if resp_buf is None: + sock.close() + release(resp) + return None + + total: INT = 0 + while total < HTTP_BUFSIZ - 1: + n: INT = sock.recv(resp_buf + total, HTTP_BUFSIZ - 1 - total) + if n <= 0: break + total += n + + resp_buf[total] = 0 + sock.close() + + resp._raw_buf = resp_buf + + # 解析状态码 + _ParseStatusLine(resp_buf, c.Addr(resp.status_code)) + + # 找到 body + body_offset: INT = _FindBodyStart(resp_buf, total) + if body_offset < 0 or body_offset >= total: + resp.text = None + resp.body_len = 0 + else: + resp.text = resp_buf + body_offset + resp.body_len = total - body_offset + + return resp + + +def post(url: str, data: str = None, content_type: str = "application/x-www-form-urlencoded") -> Response | t.CPtr: + """HTTP POST 请求。 + + Args: + url: 完整 URL + data: 请求体数据 + content_type: Content-Type 头（默认 "application/x-www-form-urlencoded"） + Returns: + Response 对象指针（从 _mbuddy 分配），失败返回 None。 + """ + if _mbuddy is None: return None + + resp: Response | t.CPtr = _AllocResponse() + if resp is None: return None + + if _EnsureSocketInit() != 0: + release(resp) + return None + + # 解析 URL + host_buf: t.CArray[t.CChar, 256] + path_buf: t.CArray[t.CChar, 1024] + port: INT = 0 + + if _ParseUrl(url, host_buf, 256, path_buf, 1024, c.Addr(port)) != 0: + release(resp) + return None + + body_len: INT = INT(string.strlen(data)) if data is not None else 0 + ct: str = content_type if content_type is not None else "application/x-www-form-urlencoded" + + # 创建并连接 Socket + sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM) + if sock.fd == socket.INVALID_SOCKET: + release(resp) + return None + + if sock.connect(host_buf, port) != 0: + sock.close() + release(resp) + return None + + # 构建 HTTP 请求 + req_buf: str = _mbuddy.alloc(HTTP_BUFSIZ) + if req_buf is None: + sock.close() + release(resp) + return None + + stdio.snprintf(req_buf, HTTP_BUFSIZ, + "POST %s HTTP/1.1\r\nHost: %s\r\nContent-Type: %s\r\nContent-Length: %d\r\nConnection: close\r\n\r\n", + path_buf, host_buf, ct, body_len) + + hdr_len: INT = INT(string.strlen(req_buf)) + + # 发送头部 + body + sent: INT = sock.send(req_buf, hdr_len) + if sent > 0 and body_len > 0 and data is not None: + sock.send(data, body_len) + + _mbuddy.free(req_buf) + + # 接收响应 + resp_buf: str = _mbuddy.alloc(HTTP_BUFSIZ) + if resp_buf is None: + sock.close() + release(resp) + return None + + total: INT = 0 + while total < HTTP_BUFSIZ - 1: + n: INT = sock.recv(resp_buf + total, HTTP_BUFSIZ - 1 - total) + if n <= 0: break + total += n + + resp_buf[total] = 0 + sock.close() + + resp._raw_buf = resp_buf + + _ParseStatusLine(resp_buf, c.Addr(resp.status_code)) + body_offset: INT = _FindBodyStart(resp_buf, total) + if body_offset < 0 or body_offset >= total: + resp.text = None + resp.body_len = 0 + else: + resp.text = resp_buf + body_offset + resp.body_len = total - body_offset + + return resp + + +def release(resp: Response | t.CPtr): + """释放 Response 对象及其内部缓冲区。 + + Args: + resp: 要释放的 Response 对象 + """ + if resp is None: return + if _mbuddy is None: return + if resp._raw_buf is not None: + _mbuddy.free(resp._raw_buf) + _mbuddy.free(t.CVoid(t.CUInt64T(resp), t.CPtr)) \ No newline at end of file diff --git a/includes/rwlock.py b/includes/rwlock.py new file mode 100644 index 0000000..0166bfb --- /dev/null +++ b/includes/rwlock.py @@ -0,0 +1,46 @@ +import t, c +from stdint import * +from w32.win32base import * +from w32.win32sync import ( + InitializeSRWLock, + AcquireSRWLockExclusive, ReleaseSRWLockExclusive, + AcquireSRWLockShared, ReleaseSRWLockShared +) + + +@t.Object +class RWLock: + """基于 SRWLock 的读写锁，支持多读单写，不支持递归。 + + 用法: + rwlock: RWLock = RWLock() + rwlock.acquire_read() + # ... 读取共享数据 + rwlock.release_read() + rwlock.acquire_write() + # ... 修改共享数据 + rwlock.release_write() + """ + lock: VOIDPTR + + def __init__(self): + InitializeSRWLock(c.Addr(self.lock)) + + def acquire_read(self): + AcquireSRWLockShared(c.Addr(self.lock)) + + def release_read(self): + ReleaseSRWLockShared(c.Addr(self.lock)) + + def acquire_write(self): + AcquireSRWLockExclusive(c.Addr(self.lock)) + + def release_write(self): + ReleaseSRWLockExclusive(c.Addr(self.lock)) + + def __enter__(self) -> 'RWLock' | t.CPtr: + self.acquire_write() + return self + + def __exit__(self): + self.release_write() diff --git a/includes/shutil.py b/includes/shutil.py new file mode 100644 index 0000000..4bc7a48 --- /dev/null +++ b/includes/shutil.py @@ -0,0 +1,283 @@ +import t, c +from stdint import * +import platmacro +import memhub +import string + +# Windows 后端依赖 +import w32.win32base +import w32.win32file + +# POSIX 后端依赖 +import posix + + +# 文件拷贝缓冲区大小 +COPY_BUF_SIZE: t.CDefine = 65536 + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# copyfile：拷贝文件内容（不保留权限/元数据） +# ============================================================================= + +def copyfile(src: str, dst: str) -> int: + """拷贝文件内容。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return _copyfile_win32(src, dst) + return _copyfile_posix(src, dst) + + +def _copyfile_win32(src: str, dst: str) -> int: + """Windows：使用 CopyFileA 内核实现（高效）""" + result: BOOL = w32.win32file.CopyFileA(src, dst, 0) + if result == 0: + return -1 + return 0 + + +def _copyfile_posix(src: str, dst: str) -> int: + """POSIX：手动 read/write 循环拷贝""" + fd_in: INT = posix.open(src, posix.O_RDONLY, 0) + if fd_in < 0: + return -1 + fd_out: INT = posix.open(dst, posix.O_WRONLY | posix.O_CREAT | posix.O_TRUNC, 0o644) + if fd_out < 0: + posix.close(fd_in) + return -1 + buf: bytes = _mbuddy.alloc(COPY_BUF_SIZE) + if buf == None: + posix.close(fd_in) + posix.close(fd_out) + return -1 + n: ssize_t = 0 + result: int = 0 + n = posix.read(fd_in, buf, COPY_BUF_SIZE) + while n > 0: + written: ssize_t = posix.write(fd_out, buf, t.CSizeT(n)) + if written < 0: + result = -1 + break + n = posix.read(fd_in, buf, COPY_BUF_SIZE) + if n < 0: + result = -1 + _mbuddy.free(buf) + posix.close(fd_in) + posix.close(fd_out) + return result + + +# ============================================================================= +# copy：拷贝文件并保留权限 +# ============================================================================= + +def copy(src: str, dst: str) -> int: + """拷贝文件并保留权限（POSIX 上调用 stat+chmod）。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return _copy_win32(src, dst) + return _copy_posix(src, dst) + + +def _copy_win32(src: str, dst: str) -> int: + """Windows：拷贝文件（CopyFileA，Windows 权限模型简单，直接委托 copyfile）""" + return _copyfile_win32(src, dst) + + +def _copy_posix(src: str, dst: str) -> int: + """POSIX：拷贝文件并保留权限（stat+chmod）""" + result: int = _copyfile_posix(src, dst) + if result != 0: + return -1 + st: posix.Stat = posix.Stat() + if posix.stat(src, t.CPtr(c.Addr(st))) == 0: + posix.chmod(dst, st.st_mode) + return 0 + + +# ============================================================================= +# rmtree：递归删除目录树 +# ============================================================================= + +def rmtree(path: str) -> int: + """递归删除目录树。成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return _rmtree_win32(path) + return _rmtree_posix(path) + + +def _rmtree_win32(path: str) -> int: + """Windows：FindFirstFileA/FindNextFileA 递归删除""" + path_len: t.CSizeT = len(path) + # 构造搜索模式 path\* + pattern: bytes = _mbuddy.alloc(path_len + 3) + if pattern == None: + return -1 + i: t.CSizeT = 0 + for ch in path: + pattern[i] = ch + i += 1 + if path_len > 0: + if path[path_len - 1] != '\\' and path[path_len - 1] != '/': + pattern[i] = '\\' + i += 1 + pattern[i] = '*' + i += 1 + pattern[i] = 0 + + find_data: w32.win32file.WIN32_FIND_DATAA = w32.win32file.WIN32_FIND_DATAA() + handle: w32.win32base.HANDLE = w32.win32file.FindFirstFileA(pattern, t.CPtr(c.Addr(find_data))) + _mbuddy.free(pattern) + if handle == w32.win32base.INVALID_HANDLE_VALUE: + return -1 + + result: int = 0 + has_more: int = 1 + while has_more == 1: + name: str = find_data.cFileName + # 跳过 . 和 .. + skip: bool = False + if name[0] == '.': + if name[1] == 0: + skip = True + elif name[1] == '.' and name[2] == 0: + skip = True + if not skip: + # 构造完整路径 path\name + name_len: t.CSizeT = len(name) + full_path: bytes = _mbuddy.alloc(path_len + name_len + 2) + if full_path == None: + result = -1 + else: + j: t.CSizeT = 0 + for ch in path: + full_path[j] = ch + j += 1 + if path_len > 0: + if path[path_len - 1] != '\\' and path[path_len - 1] != '/': + full_path[j] = '\\' + j += 1 + for ch in name: + full_path[j] = ch + j += 1 + full_path[j] = 0 + # 判断是文件还是目录 + if (find_data.dwFileAttributes & w32.win32file.FILE_ATTRIBUTE_DIRECTORY) != 0: + # 递归删除子目录 + sub_result: int = _rmtree_win32(str(full_path)) + if sub_result != 0: + result = -1 + else: + # 删除文件 + if w32.win32file.DeleteFileA(str(full_path)) == 0: + result = -1 + _mbuddy.free(full_path) + # 继续搜索下一个条目 + if w32.win32file.FindNextFileA(handle, t.CPtr(c.Addr(find_data))) == 0: + has_more = 0 + + w32.win32file.FindClose(handle) + # 最后删除空目录本身 + if result == 0: + if w32.win32file.RemoveDirectoryA(path) == 0: + result = -1 + return result + + +def _rmtree_posix(path: str) -> int: + """POSIX：opendir/readdir 递归删除""" + dirp: posix.DIR = posix.opendir(path) + if dirp == None: + return -1 + + path_len: t.CSizeT = len(path) + result: int = 0 + entry: posix.Dirent = posix.readdir(dirp) + while entry != None: + name: str = entry.d_name + # 跳过 . 和 .. + skip: bool = False + if name[0] == '.': + if name[1] == 0: + skip = True + elif name[1] == '.' and name[2] == 0: + skip = True + if not skip: + # 构造完整路径 path/name + name_len: t.CSizeT = len(name) + full_path: bytes = _mbuddy.alloc(path_len + name_len + 2) + if full_path == None: + result = -1 + else: + j: t.CSizeT = 0 + for ch in path: + full_path[j] = ch + j += 1 + if path_len > 0: + if path[path_len - 1] != '/': + full_path[j] = '/' + j += 1 + for ch in name: + full_path[j] = ch + j += 1 + full_path[j] = 0 + # 判断条目类型 + if entry.d_type == posix.DT_DIR: + # 递归删除子目录 + sub_result: int = _rmtree_posix(str(full_path)) + if sub_result != 0: + result = -1 + else: + # 删除文件/符号链接等 + if posix.unlink(str(full_path)) != 0: + result = -1 + _mbuddy.free(full_path) + entry = posix.readdir(dirp) + + posix.closedir(dirp) + # 最后删除空目录本身 + if result == 0: + if posix.rmdir(path) != 0: + result = -1 + return result + + +# ============================================================================= +# move：移动文件/目录 +# ============================================================================= + +def move(src: str, dst: str) -> int: + """移动/重命名文件或目录。同分区时直接 rename，跨分区时 copy+delete。 + 成功返回 0，失败返回 -1。""" + if platmacro.IS_WINDOWS: + return _move_win32(src, dst) + return _move_posix(src, dst) + + +def _move_win32(src: str, dst: str) -> int: + """Windows：MoveFileExA（同分区 rename，跨分区 copy+delete）""" + rename_result: BOOL = w32.win32file.MoveFileExA( + src, dst, + w32.win32file.MOVEFILE_REPLACE_EXISTING | w32.win32file.MOVEFILE_COPY_ALLOWED + ) + if rename_result != 0: + return 0 + # rename 失败，尝试 copy + remove + if copyfile(src, dst) != 0: + return -1 + if w32.win32file.DeleteFileA(src) == 0: + return -1 + return 0 + + +def _move_posix(src: str, dst: str) -> int: + """POSIX：rename（同分区），跨分区 copy+unlink""" + if posix.rename(src, dst) == 0: + return 0 + # rename 失败（可能跨分区），尝试 copy + unlink + if copyfile(src, dst) != 0: + return -1 + if posix.unlink(src) != 0: + return -1 + return 0 diff --git a/includes/socket.py b/includes/socket.py new file mode 100644 index 0000000..11d2b82 --- /dev/null +++ b/includes/socket.py @@ -0,0 +1,451 @@ +import t, c +from stdint import * +import string +import platmacro +import stdio +import w32.winsock2 + + +# ============================================================ +# socket.py — 跨平台 Socket 封装 +# Python 风格 OOP API + 兼容旧版过程式函数 +# +# 用法示例: +# sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM) +# sock.connect("example.com", 80) +# sock.send(data, length) +# n = sock.recv(buf, bufsize) +# sock.close() +# +# 仅 3 处平台差异（通过 platmacro.IS_WINDOWS 分支）: +# 1. WSAStartup() / WSACleanup() +# 2. closesocket() vs close() +# 3. WSAGetLastError() vs errno +# ============================================================ + + +# ============================================================ +# 常量 +# ============================================================ +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 10 + +SOCK_STREAM: t.CDefine = 1 +SOCK_DGRAM: t.CDefine = 2 +SOCK_RAW: t.CDefine = 3 + +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 + +SOL_SOCKET: t.CDefine = 1 +SO_RCVTIMEO: t.CDefine = 20 +SO_SNDTIMEO: t.CDefine = 21 +SO_REUSEADDR: t.CDefine = 2 + +INADDR_ANY: t.CDefine = 0 + +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = -1 + +SOCKET_DEFAULT_TIMEOUT: t.CDefine = 0 + + +# ============================================================ +# 结构体 +# ============================================================ + +class SocketAddr: + """sockaddr_in — IPv4 地址（16 字节，POSIX 布局）""" + family: u16 + port: u16 + addr: u32 + zero: u64 + + +class HostEnt: + """struct hostent — DNS 解析结果""" + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: INT + h_length: INT + h_addr_list: CHARPTR + + +class timeval: + """struct timeval — 超时设置""" + tv_sec: INT + tv_usec: INT + + +# ============================================================ +# 外部 C 函数 — 使用 u64 句柄，跨平台兼容 +# (Windows SOCKET=u64, POSIX fd=int → u64 零扩展兼容) +# ============================================================ + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.CExtern | t.CExport: pass +def connect(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass +def send(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass +def recv(fd: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.CExtern | t.CExport: pass +def bind(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT) -> INT | t.CExtern | t.CExport: pass +def listen(fd: u64, backlog: INT) -> INT | t.CExtern | t.CExport: pass +def accept(fd: u64, addr: SocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.CExtern | t.CExport: pass +def setsockopt(fd: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.CExtern | t.CExport: pass +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> HostEnt | t.CPtr | t.CExtern | t.CExport: pass +def ntohs(netshort: u16) -> u16 | t.CExtern | t.CExport: pass +def htons(hostshort: u16) -> u16 | t.CExtern | t.CExport: pass +def close(fd: u64) -> INT | t.CExtern | t.CExport: pass + + +# ============================================================ +# 内部辅助函数 +# ============================================================ + +def _CloseFd(fd: INT) -> INT: + """平台差异：Windows 用 closesocket，POSIX 用 close。""" + if platmacro.IS_WINDOWS: + return w32.winsock2.closesocket(u64(fd)) + return close(fd) + + +def _MakeAddr(host: str, port: INT) -> SocketAddr: + """构建 SocketAddr。先尝试点分十进制 IP，失败则 DNS 解析。""" + addr: SocketAddr + string.memset(c.Addr(addr), 0, 16) + addr.family = u16(AF_INET) + addr.port = htons(u16(port)) + + ip: u32 = 0 + if _ParseIPv4(host, c.Addr(ip)): + addr.addr = ip + return addr + + # DNS 解析 — gethostbyname 返回 struct hostent* + # h_addr_list 是 char** → h_addr_list[0] 是 char* → *(uint32_t*)h_addr_list[0] 是 IP + if platmacro.IS_WINDOWS: + he_raw: w32.winsock2.WSAHostEnt | t.CPtr = w32.winsock2.gethostbyname(host) + if he_raw is not None: + # he_raw.h_addr_list → he_raw->h_addr_list (char**) + # c.Deref(t.CUInt64T(he_raw.h_addr_list, t.CPtr)) → *(uint64_t*)(he_raw->h_addr_list) → h_addr_list[0] + ip_ptr: u64 = u64(c.Deref(t.CUInt64T(he_raw.h_addr_list, t.CPtr))) + if ip_ptr != 0: + addr.addr = u32(c.Deref(t.CUInt32T(ip_ptr, t.CPtr))) + else: + he: HostEnt | t.CPtr = gethostbyname(host) + if he is not None: + ip_ptr: u64 = u64(c.Deref(t.CUInt64T(he.h_addr_list, t.CPtr))) + if ip_ptr != 0: + addr.addr = u32(c.Deref(t.CUInt32T(ip_ptr, t.CPtr))) + + return addr + + +def _ParseIPv4(ip_str: str, out_addr: u32 | t.CPtr) -> bool: + """解析点分十进制 IPv4 地址。""" + if ip_str is None or out_addr is None: return False + + parts: t.CArray[INT, 4] + part_count: INT = 0 + current: INT = 0 + started: bool = False + + for ch in ip_str: + if ch >= '0' and ch <= '9': + if not started: started = True + current = current * 10 + INT(ch - '0') + if current > 255: return False + elif ch == '.': + if not started: return False + if part_count >= 4: return False + parts[part_count] = current + part_count += 1 + current = 0 + started = False + else: + break + + if started and part_count < 4: + parts[part_count] = current + part_count += 1 + + if part_count != 4: return False + + result: u32 = 0 + result = result | (u32(parts[0]) << 24) + result = result | (u32(parts[1]) << 16) + result = result | (u32(parts[2]) << 8) + result = result | u32(parts[3]) + c.DerefAs(out_addr, result) + return True + + +# ============================================================ +# Socket 类 — Python 风格 OOP API +# ============================================================ + +class Socket: + """跨平台 Socket 封装，贴近 Python socket.socket 用法。 + + 用法: + sock = Socket(AF_INET, SOCK_STREAM) + sock.connect("example.com", 80) + sock.send(data, length) + n = sock.recv(buf, bufsize) + sock.close() + """ + fd: INT + family: INT + type: INT + protocol: INT + + def __init__(self, family: INT, type: INT, protocol: INT = 0): + """创建 Socket。 + + Args: + family: 地址族 (AF_INET) + type: 套接字类型 (SOCK_STREAM) + protocol: 协议 (默认 0，自动匹配) + """ + self.family = family + self.type = type + self.protocol = protocol + + if platmacro.IS_WINDOWS: + self.fd = INT(w32.winsock2.socket(family, type, protocol)) + else: + self.fd = socket(family, type, protocol) + + def connect(self, host: str, port: INT) -> INT: + """连接到远程主机。 + + Args: + host: 主机名或 IP + port: 端口号 + Returns: + 0 成功，非 0 失败 + """ + if self.fd == INVALID_SOCKET: return -1 + addr: SocketAddr = _MakeAddr(host, port) + + result: INT = 0 + if platmacro.IS_WINDOWS: + result = w32.winsock2.connect(u64(self.fd), c.Addr(addr), 16) + if result != 0: + err: INT = w32.winsock2.WSAGetLastError() + else: + result = connect(self.fd, c.Addr(addr), 16) + return result + + def send(self, data: t.CVoid | t.CPtr, length: INT) -> INT: + """发送数据。 + + Returns: + 已发送字节数，SOCKET_ERROR 失败 + """ + if self.fd == INVALID_SOCKET: return SOCKET_ERROR + if data is None or length <= 0: return 0 + + if platmacro.IS_WINDOWS: + return w32.winsock2.send(u64(self.fd), data, length, 0) + return send(self.fd, data, length, 0) + + def recv(self, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: + """接收数据。 + + Returns: + 已接收字节数，SOCKET_ERROR 失败，0 表示连接关闭 + """ + if self.fd == INVALID_SOCKET: return SOCKET_ERROR + if buf is None or bufsize <= 0: return 0 + + if platmacro.IS_WINDOWS: + return w32.winsock2.recv(u64(self.fd), buf, bufsize, 0) + return recv(self.fd, buf, bufsize, 0) + + def sendall(self, data: t.CVoid | t.CPtr, length: INT) -> INT: + """发送全部数据（循环发送直到完成或出错）。 + + Returns: + 0 成功，SOCKET_ERROR 失败 + """ + if self.fd == INVALID_SOCKET: return SOCKET_ERROR + if data is None or length <= 0: return 0 + + total: INT = 0 + while total < length: + n: INT = self.send(t.CVoid(t.CUInt64T(data) + total, t.CPtr), length - total) + if n <= 0: return SOCKET_ERROR + total += n + return 0 + + def close(self) -> INT: + """关闭 Socket。""" + if self.fd == INVALID_SOCKET: return 0 + result: INT = _CloseFd(self.fd) + self.fd = INVALID_SOCKET + return result + + def bind(self, host: str, port: INT) -> INT: + """绑定地址和端口（服务器端）。 + + Returns: + 0 成功，非 0 失败 + """ + if self.fd == INVALID_SOCKET: return -1 + addr: SocketAddr = _MakeAddr(host, port) + + if platmacro.IS_WINDOWS: + return w32.winsock2.bind(u64(self.fd), c.Addr(addr), 16) + return bind(self.fd, c.Addr(addr), 16) + + def listen(self, backlog: INT) -> INT: + """开始监听连接（服务器端）。 + + Returns: + 0 成功，非 0 失败 + """ + if self.fd == INVALID_SOCKET: return -1 + + if platmacro.IS_WINDOWS: + return w32.winsock2.listen(u64(self.fd), backlog) + return listen(self.fd, backlog) + + def accept(self, out_client: 'Socket' | t.CPtr) -> INT: + """接受一个客户端连接。 + + Args: + out_client: 输出参数，接收客户端 Socket + Returns: + 客户端 fd（>= 0 成功），SOCKET_ERROR 失败 + """ + if self.fd == INVALID_SOCKET: return SOCKET_ERROR + if out_client is None: return SOCKET_ERROR + + addr: SocketAddr + string.memset(c.Addr(addr), 0, 16) + addrlen: INT = 16 + + if platmacro.IS_WINDOWS: + client_fd: u64 = w32.winsock2.accept(u64(self.fd), c.Addr(addr), c.Addr(addrlen)) + if client_fd == w32.winsock2.INVALID_SOCKET: + return SOCKET_ERROR + out_client.fd = INT(client_fd) + out_client.family = self.family + out_client.type = self.type + out_client.protocol = self.protocol + return INT(client_fd) + + client_fd: INT = accept(self.fd, c.Addr(addr), c.Addr(addrlen)) + if client_fd == SOCKET_ERROR: + return SOCKET_ERROR + out_client.fd = client_fd + out_client.family = self.family + out_client.type = self.type + out_client.protocol = self.protocol + return client_fd + + def settimeout(self, timeout_ms: INT) -> INT: + """设置接收和发送超时（毫秒）。0 表示无超时。 + + Returns: + 0 成功，非 0 失败 + """ + if self.fd == INVALID_SOCKET: return -1 + + tv: timeval + tv.tv_sec = timeout_ms / 1000 + tv.tv_usec = (timeout_ms % 1000) * 1000 + optlen: INT = 8 + + if platmacro.IS_WINDOWS: + r1: INT = w32.winsock2.setsockopt(u64(self.fd), SOL_SOCKET, SO_RCVTIMEO, c.Addr(tv), optlen) + r2: INT = w32.winsock2.setsockopt(u64(self.fd), SOL_SOCKET, SO_SNDTIMEO, c.Addr(tv), optlen) + else: + r1 = setsockopt(self.fd, SOL_SOCKET, SO_RCVTIMEO, c.Addr(tv), optlen) + r2 = setsockopt(self.fd, SOL_SOCKET, SO_SNDTIMEO, c.Addr(tv), optlen) + + if r1 != 0: return r1 + return r2 + + def fileno(self) -> INT: + """返回底层文件描述符 / SOCKET 句柄。""" + return self.fd + + +# ============================================================ +# 平台初始化 +# ============================================================ + +def SocketInit() -> INT: + """初始化 Socket 库。Windows 下调用 WSAStartup，POSIX 无操作。""" + if platmacro.IS_WINDOWS: + wsa_data: t.CArray[BYTE, 400] + string.memset(c.Addr(wsa_data), 0, 400) + return w32.winsock2.WSAStartup(w32.winsock2.WINSOCK_VERSION, c.Addr(wsa_data)) + return 0 + + +def SocketCleanup() -> INT: + """清理 Socket 库。""" + if platmacro.IS_WINDOWS: + return w32.winsock2.WSACleanup() + return 0 + + +# ============================================================ +# 兼容旧版过程式 API（委托到 Socket 方法） +# ============================================================ + +def SocketCreate(family: INT, type: INT, protocol: INT, out_sock: Socket | t.CPtr) -> INT: + """创建 Socket 句柄（兼容旧版 API）。建议使用 Socket(...) 构造。""" + if out_sock is None: return -1 + out_sock.family = family + out_sock.type = type + out_sock.protocol = protocol + + if platmacro.IS_WINDOWS: + fd: u64 = w32.winsock2.socket(family, type, protocol) + out_sock.fd = INT(fd) + if fd == w32.winsock2.INVALID_SOCKET: return -1 + else: + fd: INT = socket(family, type, protocol) + out_sock.fd = fd + if fd == INVALID_SOCKET: return -1 + return 0 + + +def SocketConnect(sock: Socket | t.CPtr, host: str, port: INT) -> INT: + return sock.connect(host, port) + + +def SocketSend(sock: Socket | t.CPtr, data: t.CVoid | t.CPtr, length: INT) -> INT: + return sock.send(data, length) + + +def SocketRecv(sock: Socket | t.CPtr, buf: t.CVoid | t.CPtr, bufsize: INT) -> INT: + return sock.recv(buf, bufsize) + + +def SocketClose(sock: Socket | t.CPtr) -> INT: + return sock.close() + + +def SocketBind(sock: Socket | t.CPtr, host: str, port: INT) -> INT: + return sock.bind(host, port) + + +def SocketListen(sock: Socket | t.CPtr, backlog: INT) -> INT: + return sock.listen(backlog) + + +def SocketAccept(sock: Socket | t.CPtr, out_client: Socket | t.CPtr) -> INT: + return sock.accept(out_client) + + +def SocketSetTimeout(sock: Socket | t.CPtr, timeout_ms: INT) -> INT: + return sock.settimeout(timeout_ms) + + +def SocketIsValid(sock: Socket | t.CPtr) -> bool: + """检查 Socket 是否有效。""" + if sock is None: return False + return sock.fd != INVALID_SOCKET \ No newline at end of file diff --git a/includes/spinlock.py b/includes/spinlock.py new file mode 100644 index 0000000..4208a1b --- /dev/null +++ b/includes/spinlock.py @@ -0,0 +1,18 @@ +import t, c +import atom + + +@t.Object +class _spinlock: + locked: t.CVolatile | t.CInt + name: t.CArray[t.CChar, 32] + + def __init__(self): + self.locked = 0 + + def lock(self): + while atom.__atomic_test_and_set(c.Addr(self.locked), atom.ATOMIC_ACQUIRE): + pass + + def unlock(self): + atom.__atomic_clear(c.Addr(self.locked), atom.ATOMIC_RELEASE) diff --git a/includes/stdarg.py b/includes/stdarg.py new file mode 100644 index 0000000..1d0192f --- /dev/null +++ b/includes/stdarg.py @@ -0,0 +1,9 @@ +import t + +def va_start(args: t.CPtr, last_arg: t.CPtr) -> t.State: pass +def va_arg(args: t.CPtr, type: t.CPtr) -> t.CPtr | t.State: pass +def va_end(args: t.CPtr) -> t.State | t.State: pass + +va_list: t.CTypedef = t.CUnsignedChar | t.CPtr + +def arg(type: t.CType) -> t.State: pass \ No newline at end of file diff --git a/includes/stdint.py b/includes/stdint.py new file mode 100644 index 0000000..727b759 --- /dev/null +++ b/includes/stdint.py @@ -0,0 +1,133 @@ +import t + + +# ============================================================================= +# 大写别名（Win32/TransPyC 风格） +# 注意：INTPTR/UINTPTR 在此模块中是 int*/unsigned int* 指针类型（历史命名）。 +# 它们与 C 标准的 intptr_t/uintptr_t 整数类型不同，后者见下方小写 _t 区域。 +# ============================================================================= +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr + +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid + +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr + +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong + +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD + +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble + +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T + + +# ============================================================================= +# 定长整数大写别名 +# ============================================================================= +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T + +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T + +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr + +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr + +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr + +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T + +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T + +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT + + +# ============================================================================= +# C 标准库 类型名（小写 _t 后缀） +# 这些是 C 语言标准类型名，从 _CNAME_ALIAS 挪到此处作为 typedef 提供。 +# 用户 `from stdint import *` 后即可直接使用 int8_t / size_t 等。 +# ============================================================================= +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T + +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T + +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT + +# intptr_t/uintptr_t 是足以容纳指针的整数类型（不是指针！） +# 与上方的 INTPTR/UINTPTR（int 指针）语义不同。 +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT + +# 宽字符 / 字符类型 +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T + +# 浮点特化类型 +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T + +_Bool: t.CTypedef = t.CBool diff --git a/includes/stdio.py b/includes/stdio.py new file mode 100644 index 0000000..66030fe --- /dev/null +++ b/includes/stdio.py @@ -0,0 +1,16 @@ +import t, c + + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass +def puts(s: t.CConst | str) -> t.CInt | t.State: pass +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass +def fflush(stream: bytes) -> t.CInt | t.State: pass + +# 标准流指针（全局变量，仅声明供 printf 等函数使用） +stdin: bytes = 0 +stdout: bytes = 0 +stderr: bytes = 0 diff --git a/includes/stdlib.py b/includes/stdlib.py new file mode 100644 index 0000000..deb3b84 --- /dev/null +++ b/includes/stdlib.py @@ -0,0 +1,10 @@ +from stdint import * +import t + + + +def malloc(size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass +def calloc(nmemb: t.CSizeT, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass +def realloc(p: t.CVoid | t.CPtr, size: t.CSizeT) -> t.CVoid | t.CPtr | t.State: pass +def free(p: t.CVoid | t.CPtr) -> t.State: pass +def system(cmd: t.CConst | t.CChar | t.CPtr) -> INT | t.State: pass \ No newline at end of file diff --git a/includes/string.py b/includes/string.py new file mode 100644 index 0000000..678191b --- /dev/null +++ b/includes/string.py @@ -0,0 +1,354 @@ +from stdint import * +import t, c + + +# 字符串复制函数 +def strcpy(dest: str, src: str) -> str: + if dest == None or src == None: return None + original_dest: str = dest + for s in src: + dest[0] = s + dest += 1 + dest[0] = 0 + return original_dest + + +# 字符串拼接函数: 将 src 追加到 dest 末尾 +def strcat(dest: str, src: str) -> str: + if dest == None or src == None: return dest + original_dest: str = dest + # 找到 dest 末尾 + while dest[0] != '\0': + dest += 1 + # 复制 src + for s in src: + dest[0] = s + dest += 1 + dest[0] = 0 + return original_dest + + +# 有限长度字符串复制函数 +def strncpy(dest: str, src: str, n: t.CSizeT) -> str: + if dest == None or src == None or n == 0: return dest + original_dest: str = dest + i: t.CSizeT = 0 + for s in src: + if i >= n: break + dest[0] = s + dest += 1 + i += 1 + # 填充剩余空间为 '\0' + while i < n: + dest[0] = 0 + dest += 1 + i += 1 + return original_dest + + +# 字符串长度函数 +def strlen(src: str) -> t.CSizeT | t.CExport: + length: t.CSizeT = 0 + for _ in src: + length += 1 + return length + +# 字符串比较函数 +def strcmp(str1: str, str2: str) -> t.CInt: + while str1[0] != '\0' and str2[0] != '\0': + if str1[0] != str2[0]: + return str1[0] - str2[0] + str1 += 1 + str2 += 1 + return str1[0] - str2[0] + +def samestr(str1: str, str2: str) -> bool: + return str1 == str2 + + +# 有限长度字符串比较函数 +def strncmp(str1: str, str2: str, n: t.CSizeT) -> t.CInt: + i: t.CSizeT = 0 + while str1[0] != '\0' and str2[0] != '\0' and i < n: + if str1[0] != str2[0]: + return str1[0] - str2[0] + str1 += 1 + str2 += 1 + i += 1 + if i < n: + return str1[0] - str2[0] + return 0 + + +# 内存比较函数 +def memcmp(ptr1: t.CVoid | t.CPtr, ptr2: t.CVoid | t.CPtr, n: t.CSizeT) -> t.CInt: + p1: BYTEPTR = BYTEPTR(ptr1) + p2: BYTEPTR = BYTEPTR(ptr2) + i: t.CSizeT = 0 + while i < n: + if p1[i] != p2[i]: + return t.CInt(p1[i]) - t.CInt(p2[i]) + i += 1 + return 0 + + +# 字符串查找函数 +def strchr(s: str, cr: t.CInt) -> str: + while s[0] != '\0': + if s[0] == t.CChar(cr): + return str(s) + s += 1 + if cr == '\0': + return str(s) + return None + +# 字符串反向查找函数（找最后一次出现的字符） +def strrchr(s: str, cr: t.CInt) -> str: + last: str = None + while s[0] != '\0': + if s[0] == t.CChar(cr): + last = s + s += 1 + if cr == '\0': return s + return last + + +# 子串查找函数（在 s 中查找 needle 第一次出现的位置，返回指向该位置的指针，未找到返回 None） +def strstr(s: str, needle: str) -> str: + if needle[0] == 0: + return s + needle_len: t.CSizeT = strlen(needle) + if needle_len == 0: + return s + i: t.CSizeT = 0 + s_len: t.CSizeT = strlen(s) + while i + needle_len <= s_len: + if strncmp(s + i, needle, needle_len) == 0: + return s + i + i += 1 + return None + +# 跳过字符 +def strspn(s: str, skip: str = " ") -> int: + si: int = 0 + while s[si] != 0: + found: int = 0 + sj: int = 0 + while skip[sj] != 0: + if s[si] == skip[sj]: + found = 1 + break + sj += 1 + if found == 0: + break + si += 1 + return si + +# 内存设置函数 +def memset(ptr: t.CVoid | t.CPtr, value: t.CInt, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: + # 原始实现: + # p: BYTEPTR = BYTEPTR(ptr) + # i: t.CSizeT + # for i in range(num): + # p[i] = t.CUnsignedChar(value) + if num == 0: return ptr + c.Asm(f"""mov rdi, {c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)} + mov eax, {c.AsmInp(value, t.ASM_DESCR.REG_ANY)} + mov rcx, {c.AsmInp(num, t.ASM_DESCR.REG_ANY)} + rep stosb""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX, + t.ASM_DESCR.CLOBBER_RCX, t.ASM_DESCR.CLOBBER_RDI]) + return ptr + + +def memset32(ptr: t.CVoid | t.CPtr, value: t.CUInt32T, count: t.CSizeT) -> t.CVoid | t.CPtr: + # 原始实现: + # p: UINT32PTR = UINT32PTR(ptr) + # i: t.CSizeT + # for i in range(count): + # p[i] = value + if count == 0: return ptr + c.Asm(f"""mov rdi, {c.AsmInp(ptr, t.ASM_DESCR.REG_ANY)} + mov eax, {c.AsmInp(value, t.ASM_DESCR.REG_ANY)} + mov rcx, {c.AsmInp(count, t.ASM_DESCR.REG_ANY)} + rep stosd""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX, + t.ASM_DESCR.CLOBBER_RCX, t.ASM_DESCR.CLOBBER_RDI]) + return ptr + + +# 内存复制函数 +def memcpy(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr | t.CExport: + # 原始实现: + # d: BYTEPTR = BYTEPTR(dest) + # s: BYTEPTR = BYTEPTR(src) + # i: t.CSizeT + # for i in range(num): + # d[i] = s[i] + if num == 0: return dest + c.Asm(f"""mov rdi, {c.AsmInp(dest, t.ASM_DESCR.REG_ANY)} + mov rsi, {c.AsmInp(src, t.ASM_DESCR.REG_ANY)} + mov rcx, {c.AsmInp(num, t.ASM_DESCR.REG_ANY)} + rep movsb""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RCX, + t.ASM_DESCR.CLOBBER_RDI, t.ASM_DESCR.CLOBBER_RSI]) + return dest + + +# 内存移动函数 +def memmove(dest: t.CVoid | t.CPtr, src: t.CVoid | t.CPtr, num: t.CSizeT) -> t.CVoid | t.CPtr: + # 原始实现: + # d: BYTEPTR = BYTEPTR(dest) + # s: BYTEPTR = BYTEPTR(src) + # i: t.CSizeT + # if d < s: + # for i in range(num): + # d[i] = s[i] + # elif d > s: + # for i in range(num, 0, -1): + # d[i - 1] = s[i - 1] + if num == 0: return dest + c.Asm(f"""mov rdi, {c.AsmInp(dest, t.ASM_DESCR.REG_ANY)} + mov rsi, {c.AsmInp(src, t.ASM_DESCR.REG_ANY)} + mov rcx, {c.AsmInp(num, t.ASM_DESCR.REG_ANY)} + cmp rdi, rsi + jb 2f + std + add rdi, rcx + dec rdi + add rsi, rcx + dec rsi + rep movsb + cld + jmp 3f + 2: + rep movsb + 3:""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RCX, + t.ASM_DESCR.CLOBBER_RDI, t.ASM_DESCR.CLOBBER_RSI]) + return dest + + +def atoi(src: str) -> t.CInt: + num: t.CInt = 0 + sign: t.CInt = 1 + s: str = src + if s == None: return 0 + # 跳过空白 + while s[0] == ' ' or s[0] == '\t' or s[0] == '\n' or s[0] == '\r': + s += 1 + # 符号 + if s[0] == '-': + sign = -1 + s += 1 + elif s[0] == '+': + s += 1 + for ch in s: + if ch < '0' or ch > '9': break + digit: t.CInt = ch - '0' + num = num * 10 + digit + return num * sign + + +def atoll(src: str) -> t.CInt64T: + num: t.CInt64T = 0 + sign: t.CInt64T = 1 + s: str = src + if s == None: return 0 + while s[0] == ' ' or s[0] == '\t' or s[0] == '\n' or s[0] == '\r': + s += 1 + if s[0] == '-': + sign = -1 + s += 1 + elif s[0] == '+': + s += 1 + for ch in s: + if ch < '0' or ch > '9': break + d: t.CInt = ch - '0' + digit: t.CInt64T = t.CInt64T(d) + num = num * 10 + digit + return num * sign + + +def atof(src: str) -> t.CDouble: + s: str = src + if s == None: return 0.0 + # 跳过空白 + while s[0] == ' ' or s[0] == '\t' or s[0] == '\n' or s[0] == '\r': + s += 1 + # 符号 + sign: t.CDouble = 1.0 + if s[0] == '-': + sign = -1.0 + s += 1 + elif s[0] == '+': + s += 1 + # 整数部分 + int_part: t.CDouble = 0.0 + for ch in s: + if ch < '0' or ch > '9': break + d1: t.CInt = ch - '0' + int_part = int_part * 10.0 + t.CDouble(d1) + s += 1 + # 小数部分 + frac_part: t.CDouble = 0.0 + frac_div: t.CDouble = 1.0 + if s[0] == '.': + s += 1 + for ch in s: + if ch < '0' or ch > '9': break + d2: t.CInt = ch - '0' + frac_div = frac_div * 10.0 + frac_part = frac_part * 10.0 + t.CDouble(d2) + s += 1 + # 指数部分 + exp: t.CInt = 0 + exp_sign: t.CInt = 1 + if s[0] == 'e' or s[0] == 'E': + s += 1 + if s[0] == '-': + exp_sign = -1 + s += 1 + elif s[0] == '+': + s += 1 + for ch in s: + if ch < '0' or ch > '9': break + d3: t.CInt = ch - '0' + exp = exp * 10 + d3 + s += 1 + exp = exp * exp_sign + result: t.CDouble = sign * (int_part + frac_part / frac_div) + # 应用指数 + if exp > 0: + while exp > 0: + result = result * 10.0 + exp -= 1 + elif exp < 0: + while exp < 0: + result = result / 10.0 + exp += 1 + return result + +def split(s: str, delim: str, result: t.CArray[str]) -> int: + count: int = 0 + start: str = s + if not s or not delim or not result: + return 0 + d: t.CChar = delim[0] # 单分隔符 + + while start[0]: + # 跳过分隔符 + while start[0] == d: start += 1 + if not start[0]: break + + # 记录token起始 + result[count] = start + count += 1 + + # 找到下一个分隔符，截断 + while start[0] and start[0] != d: start += 1 + if start[0]: + start[0] = '\0' + start += 1 + result[count] = None # 结尾标记 + return count \ No newline at end of file diff --git a/includes/subprocess.py b/includes/subprocess.py new file mode 100644 index 0000000..37e699c --- /dev/null +++ b/includes/subprocess.py @@ -0,0 +1,196 @@ +import t, c +from stdint import * +import platmacro +import memhub +import string + +# Windows 后端依赖 +import w32.win32base +import w32.win32file +import w32.win32process +import w32.win32sync + +# POSIX 后端依赖 +import posix + + +# 子进程输出缓冲区大小 +OUTPUT_BUF_SIZE: t.CDefine = 65536 + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# CompletedProcess 结果类 +# ============================================================================= + +class CompletedProcess: + """subprocess.run 的返回结果""" + args: str # 原始命令行 + returncode: int # 进程退出码（0 表示成功） + stdout: str # 标准输出（capture_output=True 时） + stderr: str # 标准错误（capture_output=True 时） + + def __new__(self, args: str, returncode: int, stdout: str, stderr: str) -> t.CPtr: + # 堆分配（避免返回栈地址导致 use-after-return） + return t.CPtr(_mbuddy.alloc(CompletedProcess.__sizeof__())) + + def __init__(self, args: str, returncode: int, stdout: str, stderr: str): + self.args = args + self.returncode = returncode + self.stdout = stdout + self.stderr = stderr + + +# ============================================================================= +# 入口函数 +# ============================================================================= + +def run(args: str, capture_output: bool = True, text: bool = True) -> CompletedProcess | t.CPtr: + """同步执行子进程并等待完成。 + args 为完整命令行字符串。 + capture_output=True 时捕获 stdout/stderr。 + text=True 时输出为字符串（当前实现总是返回字符串）。 + 返回 CompletedProcess 对象。""" + if platmacro.IS_WINDOWS: + return _run_win32(args, capture_output, text) + return _run_posix(args, capture_output, text) + + +# ============================================================================= +# Windows 后端实现 +# ============================================================================= + +def _run_win32(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: + """Windows 平台子进程执行（CreateProcessA + 管道）""" + # 复制命令行（CreateProcessA 需要可修改的 lpCommandLine） + args_len: t.CSizeT = len(args) + cmd_line: bytes = _mbuddy.alloc(args_len + 1) + if cmd_line == None: + return CompletedProcess(args, -1, None, None) + i: t.CSizeT = 0 + for ch in args: + cmd_line[i] = ch + i += 1 + cmd_line[args_len] = 0 + + # 安全属性（允许句柄继承） + sa: w32.win32base.SECURITY_ATTRIBUTES = w32.win32base.SECURITY_ATTRIBUTES() + sa.nLength = w32.win32base.SECURITY_ATTRIBUTES.__sizeof__() + sa.lpSecurityDescriptor = t.CPtr(0) + sa.bInheritHandle = 1 # TRUE + + # 创建 stdout 管道 + stdout_read: w32.win32base.HANDLE = t.CPtr(0) + stdout_write: w32.win32base.HANDLE = t.CPtr(0) + if capture_output: + pipe_result: BOOL = w32.win32file.CreatePipe( + t.CPtr(c.Addr(stdout_read)), + t.CPtr(c.Addr(stdout_write)), + t.CPtr(c.Addr(sa)), + 0 + ) + if pipe_result == 0: + return CompletedProcess(args, -1, None, None) + # 防止读端被子进程继承 + w32.win32file.SetHandleInformation(stdout_read, w32.win32file.HANDLE_FLAG_INHERIT, 0) + + # 设置启动信息 + si: w32.win32process.STARTUPINFOA = w32.win32process.STARTUPINFOA() + si.cb = w32.win32process.STARTUPINFOA.__sizeof__() + if capture_output: + si.dwFlags = w32.win32process.STARTF_USESTDHANDLES + si.hStdOutput = stdout_write + si.hStdError = stdout_write # stderr 合并到 stdout + + pi: w32.win32process.PROCESS_INFORMATION = w32.win32process.PROCESS_INFORMATION() + + # 创建进程 + # capture_output=True 时用 CREATE_NO_WINDOW（输出到管道，不需要控制台） + # capture_output=False 时用 0（继承父进程控制台，让 stdout 直接显示） + create_flags: ULONG = w32.win32process.CREATE_NO_WINDOW if capture_output else 0 + create_result: BOOL = w32.win32process.CreateProcessA( + t.CPtr(0), # lpApplicationName (NULL，从命令行解析) + cmd_line, # lpCommandLine (可修改) + t.CPtr(0), # lpProcessAttributes + t.CPtr(0), # lpThreadAttributes + 1, # bInheritHandles = TRUE + create_flags, # dwCreationFlags + t.CPtr(0), # lpEnvironment (继承父进程) + t.CPtr(0), # lpCurrentDirectory (继承父进程) + t.CPtr(c.Addr(si)), + t.CPtr(c.Addr(pi)) + ) + + if create_result == 0: + # 创建失败 + if capture_output: + w32.win32base.CloseHandle(stdout_read) + w32.win32base.CloseHandle(stdout_write) + return CompletedProcess(args, -1, None, None) + + # 关闭父进程的写端（子进程已有写端的继承副本） + if capture_output: + w32.win32base.CloseHandle(stdout_write) + + # 读取 stdout + 等待进程结束 + # 注意：必须边读边等，否则子进程输出超过管道缓冲区时会死锁 + # （子进程阻塞在写 stdout，父进程阻塞在 WaitForSingleObject） + stdout_str: str = None + if capture_output: + buf_size: ULONG = OUTPUT_BUF_SIZE + buf: bytes = _mbuddy.alloc(buf_size) + if buf != None: + total: ULONG = 0 + while 1: + bytes_read: ULONG = 0 + read_result: BOOL = w32.win32file.ReadFile( + stdout_read, + buf + total, + buf_size - 1 - total, + t.CPtr(c.Addr(bytes_read)), + t.CPtr(0) + ) + if read_result == 0 or bytes_read == 0: + break + total += bytes_read + # 缓冲区将满，扩容（翻倍） + if total >= buf_size - 1 - 4096: + new_size: ULONG = buf_size * 2 + new_buf: bytes = _mbuddy.alloc(new_size) + if new_buf != None: + string.memcpy(new_buf, buf, total) + buf = new_buf + buf_size = new_size + else: + # 扩容失败，停止读取 + break + buf[total] = 0 + stdout_str = str(buf) + + # 等待进程结束（如果还没退出的话） + w32.win32sync.WaitForSingleObject(pi.hProcess, w32.win32base.INFINITE) + + # 获取退出码 + exit_code: ULONG = 0 + w32.win32process.GetExitCodeProcess(pi.hProcess, t.CPtr(c.Addr(exit_code))) + + # 关闭句柄 + w32.win32base.CloseHandle(pi.hProcess) + w32.win32base.CloseHandle(pi.hThread) + if capture_output: + w32.win32base.CloseHandle(stdout_read) + + return CompletedProcess(args, exit_code, stdout_str, None) + + +# ============================================================================= +# POSIX 后端实现 +# ============================================================================= + +def _run_posix(args: str, capture_output: bool, text: bool) -> CompletedProcess | t.CPtr: + """POSIX 平台子进程执行（fork + execvp + pipe + waitpid） + 注意：当前为桩函数。POSIX 实现需要 pipe/fork/waitpid， + 这些函数在 Windows MinGW 上不可用，待 POSIX 平台测试时再完善。""" + return CompletedProcess(args, -1, None, None) diff --git a/includes/sys.py b/includes/sys.py new file mode 100644 index 0000000..1027df0 --- /dev/null +++ b/includes/sys.py @@ -0,0 +1,200 @@ +import t, c +from stdint import * +import platmacro +import memhub +import string +import stdio + +# Windows 后端依赖 +import w32.win32base +import w32.win32process + +# POSIX 后端依赖 +import posix + +# 模块级全局 MBuddy 指针（由用户在 main 中赋值初始化） +_mbuddy: memhub.MemBuddy | t.CPtr + + +# ============================================================================= +# C 标准库 IO 声明 +# ============================================================================= + +def fgets(buf: t.CChar | t.CPtr, size: t.CInt, stream: t.CVoid | t.CPtr) -> t.CChar | t.CPtr | t.CExtern | t.CExport: pass + +def fflush(stream: t.CVoid | t.CPtr) -> t.CInt | t.CExtern | t.CExport: pass + +# C 标准库 exit（刷新缓冲区后终止进程，跨平台） +def exit(status: INT) -> VOID | t.CExtern | t.CExport: pass + + +# ============================================================================= +# 平台信息 +# ============================================================================= + +# 最大整数值（64 位平台 ssize_t 上限） +maxsize: t.CDefine = 9223372036854775807 + + +def platform() -> str: + """返回平台标识字符串（'win32' 或 'linux'）""" + if platmacro.IS_WINDOWS: + return "win32" + return "linux" + + +# ============================================================================= +# 进程信息 +# ============================================================================= + +def get_pid() -> int: + """返回当前进程 ID""" + if platmacro.IS_WINDOWS: + return w32.win32process.GetCurrentProcessId() + return posix.getpid() + + +# ============================================================================= +# 命令行参数（argv） +# ============================================================================= + +# 全局 argv 状态（首次访问时初始化） +_argc: INT = 0 +# _argv 是指针数组（char**），必须用 str | t.CPtr 使索引按 sizeof(char*) 步长 +# str 本身是 char*，| t.CPtr 再加一层指针 = char** +# bytes | t.CPtr 也是 char**，可以同样用 +_argv: str | t.CPtr + + +def _init_argv(): + """初始化全局 argv（首次访问时调用一次）""" + global _argc, _argv + if _argc != 0: + return + if platmacro.IS_WINDOWS: + _init_argv_win32() + else: + _init_argv_posix() + + +def _init_argv_win32(): + """Windows：GetCommandLineA + 手动解析（处理空格/双引号/反斜杠）""" + global _argc, _argv + cmdline: CHARPTR = w32.win32base.GetCommandLineA() + if cmdline == None: + return + cmdlen: t.CSizeT = string.strlen(cmdline) + if cmdlen == 0: + return + + # 分配参数字符串缓冲区（最坏情况：整个命令行 + argc 个 \0） + buf: bytes = _mbuddy.alloc(cmdlen + 1) + if buf == None: + return + + # 分配指针数组（最坏情况：每个字符都是一个参数） + max_args: t.CSizeT = cmdlen + 1 + argv_arr: str | t.CPtr = (str | t.CPtr)(_mbuddy.alloc(max_args * platmacro.SIZEOF_POINTER)) + if argv_arr == None: + return + + argc: INT = 0 + i: t.CSizeT = 0 + j: t.CSizeT = 0 + in_quote: bool = False + in_arg: bool = False + + while i < cmdlen: + ch: t.CInt = cmdline[i] + if ch == 34: # 双引号 + in_quote = not in_quote + if not in_arg: + argv_arr[argc] = str(buf + j) + argc += 1 + in_arg = True + elif (ch == 32 or ch == 9) and not in_quote: + # 空格或 Tab（引号外）：参数分隔符 + if in_arg: + buf[j] = 0 + j += 1 + in_arg = False + else: + if not in_arg: + argv_arr[argc] = str(buf + j) + argc += 1 + in_arg = True + buf[j] = ch + j += 1 + i += 1 + + if in_arg: + buf[j] = 0 + + _argc = argc + _argv = argv_arr + + +def _init_argv_posix(): + """POSIX：读取 /proc/self/cmdline（参数以 \\0 分隔，末尾双 \\0）""" + global _argc, _argv + fd: INT = posix.open("/proc/self/cmdline", posix.O_RDONLY, 0) + if fd < 0: + return + + # 读取 cmdline（4KB 足够大多数情况） + buf: bytes = _mbuddy.alloc(4096) + if buf == None: + posix.close(fd) + return + + total: ssize_t = posix.read(fd, buf, 4096) + posix.close(fd) + if total <= 0: + return + + # 计算参数个数（\0 分隔） + argc: INT = 0 + i: t.CSizeT = 0 + total_size: t.CSizeT = t.CSizeT(total) + while i < total_size: + if buf[i] == 0: + argc += 1 + i += 1 + + if argc == 0: + return + + # 分配指针数组 + argv_arr: str | t.CPtr = (str | t.CPtr)(_mbuddy.alloc(argc * platmacro.SIZEOF_POINTER)) + if argv_arr == None: + return + + # 填充指针（每个参数的起始地址） + argv_arr[0] = str(buf) + idx: INT = 1 + i = 0 + while i < total_size and idx < argc: + if buf[i] == 0: + argv_arr[idx] = str(buf + i + 1) + idx += 1 + i += 1 + + _argc = argc + _argv = argv_arr + + +def argc() -> int: + """返回命令行参数个数。首次调用时自动初始化。""" + if _argc == 0: + _init_argv() + return _argc + + +def argv(index: int) -> str: + """返回第 index 个命令行参数（index 从 0 开始）。 + 首次调用时自动初始化。索引越界返回 None。""" + if _argc == 0: + _init_argv() + if index < 0 or index >= _argc: + return None + return _argv[index] diff --git a/includes/testcheck.py b/includes/testcheck.py new file mode 100644 index 0000000..629cc83 --- /dev/null +++ b/includes/testcheck.py @@ -0,0 +1,71 @@ +# testcheck - 回归测试格式化输出库 +# 提供 PASS/FAIL 计数、分节标题、汇总报告 +import t, c +import stdio +from w32.win32console import SetConsoleOutputCP, SetConsoleCP + +CP_UTF8: t.CDefine = 65001 + +_pass_count: t.CInt = 0 +_fail_count: t.CInt = 0 +_total_pass: t.CInt = 0 +_total_fail: t.CInt = 0 + + +def begin(name: str): + """开始测试套件，打印头部并重置当前套件计数器（累计计数不变）""" + global _pass_count, _fail_count + _pass_count = 0 + _fail_count = 0 + SetConsoleOutputCP(CP_UTF8) + SetConsoleCP(CP_UTF8) + stdio.printf("=== %s ===\n\n", name) + + +def section(name: str): + """打印分节标题""" + stdio.printf("--- %s ---\n", name) + + +def ok(msg: str): + """打印 PASS 并计数""" + global _pass_count + _pass_count += 1 + stdio.printf("PASS: %s\n", msg) + + +def fail(msg: str): + """打印 FAIL 并计数""" + global _fail_count + _fail_count += 1 + stdio.printf("FAIL: %s\n", msg) + + +def check(cond: t.CInt, ok_msg: str, fail_msg: str): + """条件检查，自动选择 PASS/FAIL""" + if cond: + ok(ok_msg) + else: + fail(fail_msg) + + +def info(msg: str): + """打印信息消息（不计入 PASS/FAIL）""" + stdio.printf("%s\n", msg) + + +def end() -> t.CInt: + """打印当前套件汇总，累加到总计，返回当前套件失败数""" + global _total_pass, _total_fail + _total_pass += _pass_count + _total_fail += _fail_count + stdio.printf("\n--- Summary ---\n") + stdio.printf("PASS: %d, FAIL: %d\n", _pass_count, _fail_count) + return _fail_count + + +def summary() -> t.CInt: + """打印所有套件的总计汇总，返回总失败数""" + stdio.printf("\n=== Total Summary ===\n") + stdio.printf("Total PASS: %d, Total FAIL: %d\n", _total_pass, _total_fail) + return _total_fail diff --git a/includes/this.py b/includes/this.py new file mode 100644 index 0000000..a401c2e --- /dev/null +++ b/includes/this.py @@ -0,0 +1,39 @@ +import stdio, t, c + + +def this(): + s: str = """Ivcre Cevapvcyrf, ol GrezvArkhf + +Ornhgl orngf htyvarff. +Rkcyvpvg orngf vzcyvpvg. +Fvzcyr orngf pbzcyrk. +Pbapvfr orngf ireobfr. +Syng orngf qrrc arfgvat. +Fcnpvbhf orngf pyhggrerq. +Ernqnovyvgl vf pehpvny. +Fcrpvny pnfrf qba’g whfgvsl oernxvat gur ehyrf. +Ohg cenpgvpnyvgl orngf cher vqrnyf. +Reebef fubhyq arire or fvyragyl vtaberq. +Hayrff rkcyvpvgyl fgngrq gb or vtaberq. +Jura snprq jvgu nzovthbhf frznagvpf, erfvfg znxvat nffhzcgvbaf. +Hayrff pyrneyl nyybjrq. +Gurer fubhyq or bar — cersrenoyl bayl bar — boivbhf jnl gb qb vg. +Rira vs vg’f uneq gb tenfc ng svefg, hayrff lbh qrrcyl haqrefgnaq gur haqreylvat flfgrz. +Fgnegvat vzzrqvngryl orngf vaqrsvavgr qrynl. +Ohg unfgl npgvba vf bsgra jbefr guna pnershy jnvgvat. +Vs na vzcyrzragngvba vf bofpher naq uneq gb qrfpevor, vg’f n onq fbyhgvba. +Vs na vzcyrzragngvba vf pyrne naq rnfl gb haqrefgnaq, punaprf ner vg’f n tbbq fbyhgvba. +Anzrfcnprf ner vaperqvoyl inyhnoyr — hfr gurz trarebhfyl!""" + + # ASCII + m: t.CArray[t.CChar, 256] = [0] + l: t.CArray[int, 2] = [65, 97] + + for k in l: + for i in range(26): + m[k + i] = (k + ((i + 13) % 26)) # 隐式类型转换 + + for ch in s: + _: t.CUInt8T = ch + stdio.printf("%c", m[_] if m[_] else ch) + stdio.printf("\n") diff --git a/includes/vector.py b/includes/vector.py new file mode 100644 index 0000000..62dd751 --- /dev/null +++ b/includes/vector.py @@ -0,0 +1,66 @@ +import t, c +from stdint import * +import memhub + + +class Vector[T](): + data: t.CVoid | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + elem_size: t.CSizeT + pool: memhub.MemManager | t.CPtr + + def __init__(self, pool: memhub.MemManager | t.CPtr, capacity: t.CSizeT, _hint: T): + self.elem_size = T.__sizeof__() + self.capacity = capacity + self.length = 0 + self.pool = pool + self.data = pool.alloc(capacity * self.elem_size) + + def push(self, value: T): + if self.data == None: return + if self.length >= self.capacity: + self._grow() + if self.length >= self.capacity: return + offset: t.CUInt64T = t.CUInt64T(self.data) + self.length * self.elem_size + dst: t.CVoid | t.CPtr = t.CVoid(offset, t.CPtr) + c.DerefAs(dst, value) + self.length += 1 + + def _grow(self): + if self.pool == None: return + new_capacity: t.CSizeT = self.capacity * 2 + if new_capacity == 0: new_capacity = 4 + old_size: t.CSizeT = self.capacity * self.elem_size + new_size: t.CSizeT = new_capacity * self.elem_size + new_data: t.CVoid | t.CPtr = self.pool.realloc(self.data, old_size, new_size) + if new_data == None: return + self.data = new_data + self.capacity = new_capacity + + def get(self, index: t.CSizeT) -> T: + if self.data == None: return T(0) + offset: t.CUInt64T = t.CUInt64T(self.data) + index * self.elem_size + typed_ptr = T(offset, t.CPtr) + return c.Deref(typed_ptr) + + def set(self, index: t.CSizeT, value: T): + if index >= self.length: return + if self.data == None: return + offset: t.CUInt64T = t.CUInt64T(self.data) + index * self.elem_size + dst: t.CVoid | t.CPtr = t.CVoid(offset, t.CPtr) + c.DerefAs(dst, value) + + def len(self) -> t.CSizeT: + return self.length + + def clear(self): + self.length = 0 + + def free(self): + if self.data != None: + if self.pool != None: + self.pool.free(self.data) + self.data = None + self.length = 0 + self.capacity = 0 diff --git a/includes/viperio.py b/includes/viperio.py new file mode 100644 index 0000000..9485919 --- /dev/null +++ b/includes/viperio.py @@ -0,0 +1,57 @@ +import t, c +from stdint import * + + +class Buf: + data: t.CChar | t.CPtr + length: t.CSizeT + capacity: t.CSizeT + owned: bool + + def __init__(self, data: t.CChar | t.CPtr, capacity: t.CSizeT, length: t.CSizeT = 0, owned: bool = False): + self.data = data + self.capacity = capacity + self.length = length + self.owned = owned + + def clear(self): + self.length = 0 + if self.data != None and self.capacity > 0: + self.data[0] = 0 + + def write(self, src: t.CChar | t.CPtr, count: t.CSizeT) -> t.CSizeT: + if self.data == None: return 0 + if src == None: return 0 + if count == 0: return 0 + avail: t.CSizeT = self.capacity - self.length + if avail <= 0: return 0 + n: t.CSizeT = count if count < avail else avail + i: t.CSizeT = 0 + while i < n: + self.data[self.length + i] = src[i] + i += 1 + self.length += n + self.data[self.length] = 0 + return n + + def cstr(self) -> t.CChar | t.CPtr: + return self.data + + def reset(self): + self.length = 0 + if self.data != None and self.capacity > 0: + self.data[0] = 0 + + def __enter__(self) -> 'Buf' | t.CPtr: + return self + + def __exit__(self): + self.reset() + + def free(self): + if self.owned and self.data != None: + self.data[0] = 0 + self.owned = False + self.data = None + self.length = 0 + self.capacity = 0 diff --git a/includes/viperlib.py b/includes/viperlib.py new file mode 100644 index 0000000..68ead99 --- /dev/null +++ b/includes/viperlib.py @@ -0,0 +1,1364 @@ +import t, c +import viperio +from stdarg import arg, va_arg + +TEMP_BUF_SIZE: t.CDefine = 68 + +def get_digit_count(num: t.CUnsignedInt, base: t.CInt) -> t.CStatic | t.CInt: + count: t.CInt = 0 + if num == 0: return 1 + while num > 0: + count += 1 + num //= base + return count + +def sprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CVoid | t.CExport: + if buf is None or rule is None: return + vsnprintf(buf, 0x7FFFFFFF, rule, args) + +def vsprintf(buf: t.CChar | t.CPtr, rule: str, *args) -> t.CInt: + if buf is None or rule is None: return 0 + return vsnprintf(buf, 0x7FFFFFFF, rule, args) + +def snprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, *args) -> t.CInt | t.CExport: + if buf is None or fmt is None or size == 0: return 0 + return vsnprintf(buf, size, fmt, args) + +# 检测浮点特殊值（IEEE 754） +# 返回: 0=normal, 1=+inf, 2=-inf, 3=nan +def _check_special_float(fv: t.CDouble) -> t.CInt: + fbits: t.CUInt64T = 0 + c.Load(c.Addr(fbits), c.Addr(fv)) + if ((fbits >> 52) & 0x7FF) != 0x7FF: + return 0 + is_neg: t.CInt = 1 if (fbits >> 63) != 0 else 0 + if (fbits & 0x000FFFFFFFFFFFFF) == 0: + if is_neg: + return 2 + return 1 + return 3 + +def vsnprintf(buf: t.CChar | t.CPtr, size: t.CSizeT, fmt: str, ap: t.CChar | t.CPtr) -> t.CInt | t.CExport: + if buf is None or fmt is None or size == 0: return 0 + ptr: t.CChar | t.CPtr = buf + write_count: t.CInt = 0 + temp: t.CArray[t.CChar, 68] + temp[0] = '\t' + while c.Deref(fmt) != 0 and write_count < size - 1: + if c.Deref(fmt) != 37: + ptr[0] = c.Deref(fmt) + write_count += 1 + ptr += 1 + fmt += 1 + else: + fmt += 1 + if c.Deref(fmt) == 0: break + width: t.CInt = 0 + pad_zero: t.CInt = 0 + left_align: t.CInt = 0 + show_sign: t.CInt = 0 + space_sign: t.CInt = 0 + alt_form: t.CInt = 0 + precision: t.CInt = -1 + is_long: t.CInt = 0 + is_long_long: t.CInt = 0 + is_size: t.CInt = 0 + is_short: t.CInt = 0 + is_char: t.CInt = 0 + while True: + if c.Deref(fmt) == 45: + left_align = 1 + fmt += 1 + elif c.Deref(fmt) == 48: + pad_zero = 1 + fmt += 1 + elif c.Deref(fmt) == 43: + show_sign = 1 + fmt += 1 + elif c.Deref(fmt) == 32: + space_sign = 1 + fmt += 1 + elif c.Deref(fmt) == 35: + alt_form = 1 + fmt += 1 + else: break + if show_sign: space_sign = 0 + while c.Deref(fmt) >= 48 and c.Deref(fmt) <= 57: + width = width * 10 + (c.Deref(fmt) - 48) + fmt += 1 + if c.Deref(fmt) == 46: + fmt += 1 + precision = 0 + while c.Deref(fmt) >= 48 and c.Deref(fmt) <= 57: + precision = precision * 10 + (c.Deref(fmt) - 48) + fmt += 1 + if left_align: pad_zero = 0 + if c.Deref(fmt) == 108: + is_long = 1 + fmt += 1 + if c.Deref(fmt) == 108: + is_long_long = 1 + is_long = 0 + fmt += 1 + elif c.Deref(fmt) == 104: + is_short = 1 + fmt += 1 + if c.Deref(fmt) == 104: + is_char = 1 + is_short = 0 + fmt += 1 + elif c.Deref(fmt) == 122: + is_size = 1 + fmt += 1 + elif c.Deref(fmt) == 106: + is_long_long = 1 + fmt += 1 + elif c.Deref(fmt) == 116: + is_long = 1 + fmt += 1 + if c.Deref(fmt) == 99: + if write_count < size - 1: + ptr_i: t.CInt = va_arg(ap) + ptr[0] = t.CChar(ptr_i) + write_count += 1 + ptr += 1 + else: + ptr_i: t.CInt = va_arg(ap) + elif c.Deref(fmt) == 115: + s: str = va_arg(ap, t.CPtr) + if s is None: s = "(None)" + slen: t.CInt = 0 + sc: str = s + while c.Deref(sc) != 0: + slen += 1 + sc += 1 + if not left_align and width > slen: + pad_count: t.CInt = width - slen + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + while c.Deref(s) != 0 and write_count < size - 1: + ptr[0] = c.Deref(s) + write_count += 1 + ptr += 1 + s += 1 + if left_align and width > slen: + pad_count = width - slen + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 100 or c.Deref(fmt) == 105: + if is_long_long or is_size or is_long: + num_l: t.CInt64T = va_arg(ap, t.CInt64T) + else: + num_l: t.CInt64T = t.CInt64T(va_arg(ap, t.CInt)) + if is_char: + num_l = t.CInt64T(t.CChar(num_l)) + elif is_short: + num_l = t.CInt64T(t.CShort(num_l)) + sign: t.CInt = 0 + if num_l < 0: + sign = 1 + if num_l == (-9223372036854775807 - 1): + u_num: t.CUInt64T = t.CUInt64T(0x8000000000000000) + else: + u_num: t.CUInt64T = t.CUInt64T(-num_l) + else: + u_num: t.CUInt64T = t.CUInt64T(num_l) + if show_sign: sign = 2 + elif space_sign: sign = 3 + i: t.CInt = 0 + while True: + if i >= TEMP_BUF_SIZE - 1: break + temp[i] = t.CChar(u_num % 10 + 48) + i += 1 + u_num //= 10 + if u_num <= 0: break + if sign == 1 and i < TEMP_BUF_SIZE - 1: + temp[i] = 45 + i += 1 + elif sign == 2 and i < TEMP_BUF_SIZE - 1: + temp[i] = 43 + i += 1 + elif sign == 3 and i < TEMP_BUF_SIZE - 1: + temp[i] = 32 + i += 1 + if not left_align and not pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 117: + if is_long or is_long_long or is_size: + u_num_l: t.CUInt64T = va_arg(ap, t.CUInt64T) + else: + u_num_l: t.CUInt64T = t.CUInt64T(t.CUnsignedInt(va_arg(ap))) + i = 0 + while True: + if i >= TEMP_BUF_SIZE - 1: break + temp[i] = t.CChar(u_num_l % 10 + 48) + i += 1 + u_num_l //= 10 + if u_num_l <= 0: break + if not left_align and not pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 120: + if is_long or is_long_long or is_size: + u_num_l: t.CUInt64T = va_arg(ap, t.CUInt64T) + else: + u_num_l: t.CUInt64T = t.CUInt64T(t.CUnsignedInt(va_arg(ap))) + is_nonzero: t.CInt = 0 + if u_num_l != 0: + is_nonzero = 1 + i = 0 + while True: + if i >= TEMP_BUF_SIZE - 1: break + digit: t.CInt = t.CInt(u_num_l % 16) + if digit < 10: + temp[i] = t.CChar(digit + 48) + else: + temp[i] = t.CChar(digit - 10 + 97) + u_num_l //= 16 + i += 1 + if u_num_l <= 0: break + digit_count: t.CInt = i + prefix_len: t.CInt = 0 + if alt_form and is_nonzero: + prefix_len = 2 + effective_len: t.CInt = digit_count + prefix_len + if not left_align and not pad_zero and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if prefix_len > 0: + if write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = 120 + write_count += 1 + ptr += 1 + if pad_zero and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 88: + if is_long or is_long_long or is_size: + u_num_l: t.CUInt64T = va_arg(ap) + else: + u_num_l: t.CUInt64T = t.CUInt64T(t.CUnsignedInt(va_arg(ap))) + is_nonzero = 0 + if u_num_l != 0: + is_nonzero = 1 + i = 0 + while True: + if i >= TEMP_BUF_SIZE - 1: break + digit = t.CInt(u_num_l % 16) + if digit < 10: + temp[i] = t.CChar(digit + 48) + else: + temp[i] = t.CChar(digit - 10 + 65) + u_num_l //= 16 + i += 1 + if u_num_l <= 0: break + digit_count = i + prefix_len = 0 + if alt_form and is_nonzero: + prefix_len = 2 + effective_len = digit_count + prefix_len + if not left_align and not pad_zero and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if prefix_len > 0: + if write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = 88 + write_count += 1 + ptr += 1 + if pad_zero and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > effective_len: + pad_count = width - effective_len + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 111: + if is_long or is_long_long or is_size: + u_num_l: t.CUInt64T = va_arg(ap, t.CUInt64T) + else: + u_num_l: t.CUInt64T = t.CUInt64T(t.CUnsignedInt(va_arg(ap))) + if is_char: + u_num_l = t.CUInt64T(t.CChar(u_num_l)) + elif is_short: + u_num_l = t.CUInt64T(t.CShort(u_num_l)) + i = 0 + if u_num_l == 0 and not alt_form: + temp[0] = 48 + i = 1 + else: + while True: + if i >= TEMP_BUF_SIZE - 1: break + temp[i] = t.CChar(u_num_l % 8 + 48) + i += 1 + u_num_l //= 8 + if u_num_l <= 0: break + if alt_form and i < TEMP_BUF_SIZE - 1: + temp[i] = 48 + i += 1 + if not left_align and not pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 98: + if is_long or is_long_long or is_size: + u_num_l: t.CUInt64T = va_arg(ap, t.CUInt64T) + else: + u_num_l: t.CUInt64T = t.CUInt64T(t.CUnsignedInt(va_arg(ap))) + if is_char: + u_num_l = t.CUInt64T(t.CChar(u_num_l)) + elif is_short: + u_num_l = t.CUInt64T(t.CShort(u_num_l)) + i = 0 + if u_num_l == 0: + temp[0] = 48 + i = 1 + else: + while True: + if i >= TEMP_BUF_SIZE - 1: break + temp[i] = t.CChar(u_num_l % 2 + 48) + i += 1 + u_num_l //= 2 + if u_num_l <= 0: break + if alt_form and i < TEMP_BUF_SIZE - 1: + temp[i] = 98 + i += 1 + if not left_align and not pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 112: + ptr_val: t.CUInt64T = t.CUInt64T(va_arg(ap, t.CPtr)) + if ptr_val == 0: + temp[0] = 40 + temp[1] = 110 + temp[2] = 105 + temp[3] = 108 + temp[4] = 41 + i = 5 + else: + u_num_l = ptr_val + i = 0 + while True: + if i >= TEMP_BUF_SIZE - 1: break + digit = t.CInt(u_num_l % 16) + if digit < 10: + temp[i] = t.CChar(digit + 48) + else: + temp[i] = t.CChar(digit - 10 + 97) + u_num_l //= 16 + i += 1 + if u_num_l <= 0: break + if i < TEMP_BUF_SIZE - 1: + temp[i] = 120 + i += 1 + if i < TEMP_BUF_SIZE - 1: + temp[i] = 48 + i += 1 + if not left_align and not pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while i > 0 and write_count < size - 1: + i -= 1 + ptr[0] = temp[i] + write_count += 1 + ptr += 1 + if left_align and width > i: + pad_count = width - i + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 102 or c.Deref(fmt) == 70: + f_upper: t.CInt = 1 if c.Deref(fmt) == 70 else 0 + if precision < 0: precision = 6 + fv: t.CDouble = va_arg(ap, t.CDouble) + f_sp: t.CInt = _check_special_float(fv) + if f_sp != 0: + f_sgn_len: t.CInt = 0 + if f_sp == 2: + f_sgn_len = 1 + elif f_sp == 1 and show_sign: + f_sgn_len = 1 + elif f_sp == 1 and space_sign: + f_sgn_len = 1 + f_total_len: t.CInt = f_sgn_len + 3 + if not left_align and width > f_total_len: + f_pad: t.CInt = width - f_total_len + while f_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + f_pad -= 1 + if f_sp == 2: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + elif f_sp == 1 and show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif f_sp == 1 and space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + fi_c: t.CInt = 0 + while fi_c < 3 and write_count < size - 1: + fc: t.CInt = 0 + if f_sp == 3: + # nan + if fi_c == 0: + fc = 110 + elif fi_c == 1: + fc = 97 + else: + fc = 110 + else: + # inf (f_sp==1 +inf 或 f_sp==2 -inf) + if fi_c == 0: + fc = 105 + elif fi_c == 1: + fc = 110 + else: + fc = 102 + if f_upper and fc >= 97 and fc <= 122: + fc -= 32 + ptr[0] = t.CChar(fc) + write_count += 1 + ptr += 1 + fi_c += 1 + if left_align and width > f_total_len: + f_pad = width - f_total_len + while f_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + f_pad -= 1 + fmt += 1 + continue + if fv < 0.0: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + fv = 0.0 - fv + elif show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + di: t.CInt = 0 + work: t.CDouble = fv + if work < 1.0: + temp[0] = 48 + di = 1 + else: + scale: t.CDouble = 1.0 + while work >= scale * 10.0: + scale *= 10.0 + while scale >= 1.0 and di < TEMP_BUF_SIZE - 1: + d: t.CInt = t.CInt(work / scale) + if d > 9: d = 9 + temp[di] = t.CChar(d + 48) + di += 1 + work = work - t.CDouble(d) * scale + scale /= 10.0 + frac_val: t.CDouble = work + p: t.CInt = 0 + while p < precision: + frac_val *= 10.0 + p += 1 + frac_part: t.CUInt64T = t.CUInt64T(frac_val + 0.5) + f_total: t.CInt = di + (precision + 1 if precision > 0 else 0) + if not left_align and not pad_zero and width > f_total: + pad_count = width - f_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > f_total: + pad_count = width - f_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + while di > 0 and write_count < size - 1: + di -= 1 + ptr[0] = temp[di] + write_count += 1 + ptr += 1 + if precision > 0: + if write_count < size - 1: + ptr[0] = 46 + write_count += 1 + ptr += 1 + fi: t.CInt = 0 + while fi < precision and fi < TEMP_BUF_SIZE - 1: + temp[fi] = t.CChar(frac_part % 10 + 48) + fi += 1 + frac_part //= 10 + while fi < precision: + temp[fi] = 48 + fi += 1 + while fi > 0 and write_count < size - 1: + fi -= 1 + ptr[0] = temp[fi] + write_count += 1 + ptr += 1 + elif alt_form: + if write_count < size - 1: + ptr[0] = 46 + write_count += 1 + ptr += 1 + if left_align and width > f_total: + pad_count = width - f_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 101 or c.Deref(fmt) == 69: + if precision < 0: precision = 6 + fv: t.CDouble = va_arg(ap, t.CDouble) + e_upper: t.CInt = 1 if c.Deref(fmt) == 69 else 0 + e_sp: t.CInt = _check_special_float(fv) + if e_sp != 0: + e_sgn_len: t.CInt = 0 + if e_sp == 2: + e_sgn_len = 1 + elif e_sp == 1 and show_sign: + e_sgn_len = 1 + elif e_sp == 1 and space_sign: + e_sgn_len = 1 + e_total_len: t.CInt = e_sgn_len + 3 + if not left_align and width > e_total_len: + e_pad: t.CInt = width - e_total_len + while e_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + e_pad -= 1 + if e_sp == 2: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + elif e_sp == 1 and show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif e_sp == 1 and space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + ei_c: t.CInt = 0 + while ei_c < 3 and write_count < size - 1: + ec: t.CInt = 0 + if e_sp == 3: + # nan + if ei_c == 0: + ec = 110 + elif ei_c == 1: + ec = 97 + else: + ec = 110 + else: + # inf (e_sp==1 +inf 或 e_sp==2 -inf) + if ei_c == 0: + ec = 105 + elif ei_c == 1: + ec = 110 + else: + ec = 102 + if e_upper and ec >= 97 and ec <= 122: + ec -= 32 + ptr[0] = t.CChar(ec) + write_count += 1 + ptr += 1 + ei_c += 1 + if left_align and width > e_total_len: + e_pad = width - e_total_len + while e_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + e_pad -= 1 + fmt += 1 + continue + if fv < 0.0: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + fv = 0.0 - fv + elif show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + exp10: t.CInt = 0 + if fv != 0.0: + while fv >= 10.0: + fv /= 10.0 + exp10 += 1 + while fv < 1.0: + fv *= 10.0 + exp10 -= 1 + int_digit: t.CInt = t.CInt(fv) + if int_digit > 9: int_digit = 9 + if int_digit < 0: int_digit = 0 + frac_v: t.CDouble = fv - t.CDouble(int_digit) + p: t.CInt = 0 + while p < precision: + frac_v *= 10.0 + p += 1 + frac_p: t.CUInt64T = t.CUInt64T(frac_v + 0.5) + carry: t.CUInt64T = 1 + p = 0 + while p < precision: + carry *= 10 + p += 1 + if frac_p >= carry and carry > 0: + frac_p = 0 + int_digit += 1 + if int_digit > 9: + int_digit = 1 + exp10 += 1 + e_total: t.CInt = 1 + (precision + 1 if precision > 0 else 0) + 4 + if not left_align and not pad_zero and width > e_total: + pad_count = width - e_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > e_total: + pad_count = width - e_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + if write_count < size - 1: + ptr[0] = t.CChar(int_digit + 48) + write_count += 1 + ptr += 1 + if precision > 0: + if write_count < size - 1: + ptr[0] = 46 + write_count += 1 + ptr += 1 + fi: t.CInt = 0 + while fi < precision and fi < TEMP_BUF_SIZE - 1: + temp[fi] = t.CChar(frac_p % 10 + 48) + fi += 1 + frac_p //= 10 + while fi < precision: + temp[fi] = 48 + fi += 1 + while fi > 0 and write_count < size - 1: + fi -= 1 + ptr[0] = temp[fi] + write_count += 1 + ptr += 1 + elif alt_form: + if write_count < size - 1: + ptr[0] = 46 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = 101 if e_upper == 0 else 69 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = 45 if exp10 < 0 else 43 + write_count += 1 + ptr += 1 + exp_abs: t.CUInt64T = t.CUInt64T(exp10) if exp10 >= 0 else t.CUInt64T(0 - exp10) + ei: t.CInt = 0 + if exp_abs == 0: + temp[0] = 48 + temp[1] = 48 + ei = 2 + else: + while exp_abs > 0 and ei < TEMP_BUF_SIZE - 1: + temp[ei] = t.CChar(exp_abs % 10 + 48) + ei += 1 + exp_abs //= 10 + while ei < 2: + temp[ei] = 48 + ei += 1 + while ei > 0 and write_count < size - 1: + ei -= 1 + ptr[0] = temp[ei] + write_count += 1 + ptr += 1 + if left_align and width > e_total: + pad_count = width - e_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 103 or c.Deref(fmt) == 71: + if precision < 0: precision = 6 + if precision == 0: precision = 1 + fv: t.CDouble = va_arg(ap, t.CDouble) + g_upper: t.CInt = 1 if c.Deref(fmt) == 71 else 0 + g_sp: t.CInt = _check_special_float(fv) + if g_sp != 0: + g_sgn_len: t.CInt = 0 + if g_sp == 2: + g_sgn_len = 1 + elif g_sp == 1 and show_sign: + g_sgn_len = 1 + elif g_sp == 1 and space_sign: + g_sgn_len = 1 + g_total_len: t.CInt = g_sgn_len + 3 + if not left_align and width > g_total_len: + g_pad: t.CInt = width - g_total_len + while g_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + g_pad -= 1 + if g_sp == 2: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + elif g_sp == 1 and show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif g_sp == 1 and space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + gi_c: t.CInt = 0 + while gi_c < 3 and write_count < size - 1: + gc: t.CInt = 0 + if g_sp == 3: + # nan + if gi_c == 0: + gc = 110 + elif gi_c == 1: + gc = 97 + else: + gc = 110 + else: + # inf (g_sp==1 +inf 或 g_sp==2 -inf) + if gi_c == 0: + gc = 105 + elif gi_c == 1: + gc = 110 + else: + gc = 102 + if g_upper and gc >= 97 and gc <= 122: + gc -= 32 + ptr[0] = t.CChar(gc) + write_count += 1 + ptr += 1 + gi_c += 1 + if left_align and width > g_total_len: + g_pad = width - g_total_len + while g_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + g_pad -= 1 + fmt += 1 + continue + if fv < 0.0: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + fv = 0.0 - fv + elif show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + fv_orig: t.CDouble = fv + exp10: t.CInt = 0 + if fv != 0.0: + while fv >= 10.0: + fv /= 10.0 + exp10 += 1 + while fv < 1.0: + fv *= 10.0 + exp10 -= 1 + g_use_e: t.CInt = 0 + if exp10 < -4 or exp10 >= precision: + g_use_e = 1 + g_len: t.CInt = 0 + if g_use_e: + int_digit: t.CInt = t.CInt(fv) + if int_digit > 9: int_digit = 9 + if int_digit < 0: int_digit = 0 + frac_v: t.CDouble = fv - t.CDouble(int_digit) + prec_e: t.CInt = precision - 1 + p: t.CInt = 0 + while p < prec_e: + frac_v *= 10.0 + p += 1 + frac_p: t.CUInt64T = t.CUInt64T(frac_v + 0.5) + carry: t.CUInt64T = 1 + p = 0 + while p < prec_e: + carry *= 10 + p += 1 + if frac_p >= carry and carry > 0: + frac_p = 0 + int_digit += 1 + if int_digit > 9: + int_digit = 1 + exp10 += 1 + temp[0] = t.CChar(int_digit + 48) + g_len = 1 + if prec_e > 0: + temp[1] = 46 + g_len = 2 + fi: t.CInt = 0 + while fi < prec_e and g_len < TEMP_BUF_SIZE - 1: + temp[g_len] = t.CChar(frac_p % 10 + 48) + g_len += 1 + frac_p //= 10 + lo: t.CInt = 2 + hi: t.CInt = g_len - 1 + while lo < hi: + tc: t.CChar = temp[lo] + temp[lo] = temp[hi] + temp[hi] = tc + lo += 1 + hi -= 1 + else: + prec_f: t.CInt = precision - 1 - exp10 + if prec_f < 0: prec_f = 0 + work: t.CDouble = fv_orig + if work < 1.0: + temp[0] = 48 + g_len = 1 + else: + scale: t.CDouble = 1.0 + while work >= scale * 10.0: + scale *= 10.0 + while scale >= 1.0 and g_len < TEMP_BUF_SIZE - 1: + d: t.CInt = t.CInt(work / scale) + if d > 9: d = 9 + temp[g_len] = t.CChar(d + 48) + g_len += 1 + work = work - t.CDouble(d) * scale + scale /= 10.0 + if prec_f > 0 and g_len < TEMP_BUF_SIZE - 1: + temp[g_len] = 46 + g_len += 1 + frac_val: t.CDouble = work + p = 0 + while p < prec_f: + frac_val *= 10.0 + p += 1 + frac_part: t.CUInt64T = t.CUInt64T(frac_val + 0.5) + fi = 0 + while fi < prec_f and g_len < TEMP_BUF_SIZE - 1: + temp[g_len] = t.CChar(frac_part % 10 + 48) + g_len += 1 + frac_part //= 10 + lo = g_len - prec_f + hi = g_len - 1 + while lo < hi: + tc = temp[lo] + temp[lo] = temp[hi] + temp[hi] = tc + lo += 1 + hi -= 1 + if not alt_form and g_len > 0: + while g_len > 0: + if temp[g_len - 1] == 48: + g_len -= 1 + else: + break + if g_len > 0 and temp[g_len - 1] == 46: + g_len -= 1 + g_total: t.CInt = g_len + if g_use_e: g_total = g_len + 4 + if not left_align and not pad_zero and width > g_total: + pad_count = width - g_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + if pad_zero and width > g_total: + pad_count = width - g_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + pad_count -= 1 + gi: t.CInt = 0 + while gi < g_len and write_count < size - 1: + ptr[0] = temp[gi] + write_count += 1 + ptr += 1 + gi += 1 + if g_use_e: + if write_count < size - 1: + ptr[0] = 101 if g_upper == 0 else 69 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = 45 if exp10 < 0 else 43 + write_count += 1 + ptr += 1 + exp_abs: t.CUInt64T = t.CUInt64T(exp10) if exp10 >= 0 else t.CUInt64T(0 - exp10) + ei: t.CInt = 0 + if exp_abs == 0: + temp[0] = 48 + temp[1] = 48 + ei = 2 + else: + while exp_abs > 0 and ei < TEMP_BUF_SIZE - 1: + temp[ei] = t.CChar(exp_abs % 10 + 48) + ei += 1 + exp_abs //= 10 + while ei < 2: + temp[ei] = 48 + ei += 1 + while ei > 0 and write_count < size - 1: + ei -= 1 + ptr[0] = temp[ei] + write_count += 1 + ptr += 1 + if left_align and width > g_total: + pad_count = width - g_total + while pad_count > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + pad_count -= 1 + elif c.Deref(fmt) == 97 or c.Deref(fmt) == 65: + a_upper: t.CInt = 1 if c.Deref(fmt) == 65 else 0 + fv: t.CDouble = va_arg(ap, t.CDouble) + a_sp: t.CInt = _check_special_float(fv) + if a_sp != 0: + a_sgn_len: t.CInt = 0 + if a_sp == 2: + a_sgn_len = 1 + elif a_sp == 1 and show_sign: + a_sgn_len = 1 + elif a_sp == 1 and space_sign: + a_sgn_len = 1 + a_total_len: t.CInt = a_sgn_len + 3 + if not left_align and width > a_total_len: + a_pad: t.CInt = width - a_total_len + while a_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + a_pad -= 1 + if a_sp == 2: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + elif a_sp == 1 and show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif a_sp == 1 and space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + ai_c: t.CInt = 0 + while ai_c < 3 and write_count < size - 1: + ac: t.CInt = 0 + if a_sp == 3: + # nan + if ai_c == 0: + ac = 110 + elif ai_c == 1: + ac = 97 + else: + ac = 110 + else: + # inf (a_sp==1 +inf 或 a_sp==2 -inf) + if ai_c == 0: + ac = 105 + elif ai_c == 1: + ac = 110 + else: + ac = 102 + if a_upper and ac >= 97 and ac <= 122: + ac -= 32 + ptr[0] = t.CChar(ac) + write_count += 1 + ptr += 1 + ai_c += 1 + if left_align and width > a_total_len: + a_pad = width - a_total_len + while a_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + a_pad -= 1 + fmt += 1 + continue + fbits: t.CUInt64T = 0 + c.Load(c.Addr(fbits), c.Addr(fv)) + a_neg: t.CInt = 1 if (fbits >> 63) != 0 else 0 + a_exp: t.CInt = t.CInt((fbits >> 52) & 0x7FF) + a_frac: t.CUInt64T = fbits & 0x000FFFFFFFFFFFFF + a_leading: t.CInt = 0 + a_binexp: t.CInt = 0 + if a_exp == 0: + a_leading = 0 + if a_frac == 0: + a_binexp = 0 + else: + a_binexp = -1022 + else: + a_leading = 1 + a_binexp = a_exp - 1023 + a_prec: t.CInt = precision + if a_prec < 0: + a_prec = 13 + if a_prec > 50: + a_prec = 50 + a_hi: t.CInt = 0 + while a_hi < 13 and a_hi < a_prec: + a_digit: t.CInt = t.CInt((a_frac >> (48 - 4 * a_hi)) & 0xF) + if a_digit < 10: + temp[a_hi] = t.CChar(a_digit + 48) + else: + if a_upper: + temp[a_hi] = t.CChar(a_digit - 10 + 65) + else: + temp[a_hi] = t.CChar(a_digit - 10 + 97) + a_hi += 1 + while a_hi < a_prec: + temp[a_hi] = 48 + a_hi += 1 + a_hd_count: t.CInt = a_prec + if precision < 0: + while a_hd_count > 0 and temp[a_hd_count - 1] == 48: + a_hd_count -= 1 + a_has_dot: t.CInt = 0 + if a_hd_count > 0: + a_has_dot = 1 + elif alt_form: + a_has_dot = 1 + a_eb: t.CInt = a_binexp + a_esign: t.CInt = 0 + if a_binexp < 0: + a_esign = 1 + a_eb = -a_binexp + a_ei: t.CInt = 0 + if a_eb == 0: + temp[60] = 48 + a_ei = 1 + else: + while a_eb > 0 and a_ei < 7: + temp[60 + a_ei] = t.CChar(a_eb % 10 + 48) + a_ei += 1 + a_eb //= 10 + a_ecount: t.CInt = a_ei + a_sign_len: t.CInt = 0 + if a_neg: + a_sign_len = 1 + elif show_sign: + a_sign_len = 1 + elif space_sign: + a_sign_len = 1 + a_content_len: t.CInt = a_sign_len + 2 + 1 + a_has_dot + a_hd_count + 1 + 1 + a_ecount + if not left_align and not pad_zero and width > a_content_len: + a_pad = width - a_content_len + while a_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + a_pad -= 1 + if a_neg: + if write_count < size - 1: + ptr[0] = 45 + write_count += 1 + ptr += 1 + elif show_sign: + if write_count < size - 1: + ptr[0] = 43 + write_count += 1 + ptr += 1 + elif space_sign: + if write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + if pad_zero and width > a_content_len: + a_pad = width - a_content_len + while a_pad > 0 and write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + a_pad -= 1 + if write_count < size - 1: + ptr[0] = 48 + write_count += 1 + ptr += 1 + if write_count < size - 1: + if a_upper: + ptr[0] = 88 + else: + ptr[0] = 120 + write_count += 1 + ptr += 1 + if write_count < size - 1: + ptr[0] = t.CChar(a_leading + 48) + write_count += 1 + ptr += 1 + if a_has_dot: + if write_count < size - 1: + ptr[0] = 46 + write_count += 1 + ptr += 1 + a_hj: t.CInt = 0 + while a_hj < a_hd_count and write_count < size - 1: + ptr[0] = temp[a_hj] + write_count += 1 + ptr += 1 + a_hj += 1 + if write_count < size - 1: + if a_upper: + ptr[0] = 80 + else: + ptr[0] = 112 + write_count += 1 + ptr += 1 + if write_count < size - 1: + if a_esign: + ptr[0] = 45 + else: + ptr[0] = 43 + write_count += 1 + ptr += 1 + a_ek: t.CInt = a_ecount + while a_ek > 0 and write_count < size - 1: + a_ek -= 1 + ptr[0] = temp[60 + a_ek] + write_count += 1 + ptr += 1 + if left_align and width > a_content_len: + a_pad = width - a_content_len + while a_pad > 0 and write_count < size - 1: + ptr[0] = 32 + write_count += 1 + ptr += 1 + a_pad -= 1 + elif c.Deref(fmt) == 110: + if is_long_long or is_long: + np: t.CInt64T | t.CPtr = va_arg(ap, t.CInt64T | t.CPtr) + if np != None: + np[0] = t.CInt64T(write_count) + elif is_short: + sp: t.CShort | t.CPtr = va_arg(ap, t.CShort | t.CPtr) + if sp != None: + sp[0] = t.CShort(write_count) + elif is_char: + cp: t.CChar | t.CPtr = va_arg(ap, t.CChar | t.CPtr) + if cp != None: + cp[0] = t.CChar(write_count) + else: + ip: t.CInt | t.CPtr = va_arg(ap, t.CInt | t.CPtr) + if ip != None: + ip[0] = write_count + elif c.Deref(fmt) == 37: + if write_count < size - 1: + ptr[0] = 37 + write_count += 1 + ptr += 1 + else: + if write_count < size - 1: + ptr[0] = c.Deref(fmt) + write_count += 1 + ptr += 1 + fmt += 1 + ptr[0] = 0 + return write_count diff --git a/includes/vipermath.py b/includes/vipermath.py new file mode 100644 index 0000000..bf38b32 --- /dev/null +++ b/includes/vipermath.py @@ -0,0 +1,542 @@ +import t, c + + +U_M_PI: t.CDefine = 3.14159265358979323846 +U_M_E: t.CDefine = 2.71828182845904523536 +U_M_PI_2: t.CDefine = 1.57079632679489661923 +U_M_PI_4: t.CDefine = 0.78539816339744830962 +U_M_1_PI: t.CDefine = 0.31830988618379067154 +U_M_2_PI: t.CDefine = 0.63661977236758134308 + +U_M_LN2: t.CDefine = 0.69314718055994530942 +U_M_LN10: t.CDefine = 2.30258509299404568402 + +U_M_2_SQRT_PI: t.CDefine = 1.77245385090551602730 + +def radians(degrees: t.CDouble) -> t.CDouble: + return degrees * (U_M_PI / t.CDouble(180.0)) + +def degrees(radians: t.CDouble) -> t.CDouble: + return radians * (t.CDouble(180.0) / U_M_PI) + +def sin(x: t.CDouble) -> t.CDouble: + while x > U_M_PI: + x -= t.CDouble(2.0) * U_M_PI + while x < -U_M_PI: + x += t.CDouble(2.0) * U_M_PI + x2: t.CDouble = x * x + result: t.CDouble = x + term: t.CDouble = x + term = term * x2 / t.CDouble(6.0) + result = result - term + term = term * x2 / t.CDouble(20.0) + result = result + term + term = term * x2 / t.CDouble(42.0) + result = result - term + term = term * x2 / t.CDouble(72.0) + result = result + term + term = term * x2 / t.CDouble(110.0) + result = result - term + term = term * x2 / t.CDouble(156.0) + result = result + term + term = term * x2 / t.CDouble(210.0) + result = result - term + term = term * x2 / t.CDouble(272.0) + result = result + term + return result + + +def cos(x: t.CDouble) -> t.CDouble: + while x > U_M_PI: + x -= t.CDouble(2.0) * U_M_PI + while x < -U_M_PI: + x += t.CDouble(2.0) * U_M_PI + x2: t.CDouble = x * x + result: t.CDouble = t.CDouble(1.0) + term: t.CDouble = t.CDouble(1.0) + term = term * x2 / t.CDouble(2.0) + result = result - term + term = term * x2 / t.CDouble(12.0) + result = result + term + term = term * x2 / t.CDouble(30.0) + result = result - term + term = term * x2 / t.CDouble(56.0) + result = result + term + term = term * x2 / t.CDouble(90.0) + result = result - term + term = term * x2 / t.CDouble(132.0) + result = result + term + term = term * x2 / t.CDouble(182.0) + result = result - term + term = term * x2 / t.CDouble(240.0) + result = result + term + return result + + +def tan(x: t.CDouble) -> t.CDouble: + return sin(x) / cos(x) + + +def asin(x: t.CDouble) -> t.CDouble: + if x > t.CDouble(1.0): + x = t.CDouble(1.0) + if x < t.CDouble(-1.0): + x = t.CDouble(-1.0) + x2: t.CDouble = x * x + result: t.CDouble = x + result += x * x2 / t.CDouble(6.0) + result += t.CDouble(3.0) * x * x2 * x2 / t.CDouble(40.0) + result += t.CDouble(5.0) * x * x2 * x2 * x2 / t.CDouble(112.0) + result += t.CDouble(35.0) * x * x2 * x2 * x2 * x2 / t.CDouble(1152.0) + result += t.CDouble(63.0) * x * x2 * x2 * x2 * x2 * x2 / t.CDouble(2816.0) + return result + + +def acos(x: t.CDouble) -> t.CDouble: + return U_M_PI / t.CDouble(2.0) - asin(x) + + +def _atan_core(x: t.CDouble) -> t.CDouble: + x2: t.CDouble = x * x + result: t.CDouble = x + term: t.CDouble = x + term = term * x2 / t.CDouble(3.0) + result = result - term + term = term * x2 / t.CDouble(5.0) + result = result + term + term = term * x2 / t.CDouble(7.0) + result = result - term + term = term * x2 / t.CDouble(9.0) + result = result + term + term = term * x2 / t.CDouble(11.0) + result = result - term + term = term * x2 / t.CDouble(13.0) + result = result + term + term = term * x2 / t.CDouble(15.0) + result = result - term + term = term * x2 / t.CDouble(17.0) + result = result + term + term = term * x2 / t.CDouble(19.0) + result = result - term + term = term * x2 / t.CDouble(21.0) + result = result + term + return result + + +def atan(x: t.CDouble) -> t.CDouble: + ax: t.CDouble = x + if ax < t.CDouble(0.0): + ax = -ax + if ax > t.CDouble(1.0): + return t.CDouble(0.0) - _atan_core(t.CDouble(1.0) / ax) + U_M_PI_2 if x > t.CDouble(0.0) else _atan_core(t.CDouble(1.0) / ax) - U_M_PI_2 + return _atan_core(x) if x >= t.CDouble(0.0) else t.CDouble(0.0) - _atan_core(ax) + + +def atan2(y: t.CDouble, x: t.CDouble) -> t.CDouble: + if x > t.CDouble(0.0): + return atan(y / x) + elif x < t.CDouble(0.0): + if y >= t.CDouble(0.0): + return atan(y / x) + U_M_PI + else: + return atan(y / x) - U_M_PI + else: + if y > t.CDouble(0.0): + return U_M_PI_2 + elif y < t.CDouble(0.0): + return t.CDouble(0.0) - U_M_PI_2 + else: + return t.CDouble(0.0) + + +def sinh(x: t.CDouble) -> t.CDouble: + ex: t.CDouble = exp(x) + e_negx: t.CDouble = exp(-x) + return (ex - e_negx) / t.CDouble(2.0) + + +def cosh(x: t.CDouble) -> t.CDouble: + ex: t.CDouble = exp(x) + e_negx: t.CDouble = exp(-x) + return (ex + e_negx) / t.CDouble(2.0) + + +def tanh(x: t.CDouble) -> t.CDouble: + ex: t.CDouble = exp(x) + e_negx: t.CDouble = exp(-x) + return (ex - e_negx) / (ex + e_negx) + + +def exp(x: t.CDouble) -> t.CDouble: + if x > t.CDouble(709.0): + return t.CDouble(1.7976931348623157e308) + if x < t.CDouble(-709.0): + return t.CDouble(0.0) + result: t.CDouble = t.CDouble(1.0) + term: t.CDouble = t.CDouble(1.0) + term = term * x / t.CDouble(1.0) + result = result + term + term = term * x / t.CDouble(2.0) + result = result + term + term = term * x / t.CDouble(3.0) + result = result + term + term = term * x / t.CDouble(4.0) + result = result + term + term = term * x / t.CDouble(5.0) + result = result + term + term = term * x / t.CDouble(6.0) + result = result + term + term = term * x / t.CDouble(7.0) + result = result + term + term = term * x / t.CDouble(8.0) + result = result + term + term = term * x / t.CDouble(9.0) + result = result + term + term = term * x / t.CDouble(10.0) + result = result + term + term = term * x / t.CDouble(11.0) + result = result + term + term = term * x / t.CDouble(12.0) + result = result + term + term = term * x / t.CDouble(13.0) + result = result + term + term = term * x / t.CDouble(14.0) + result = result + term + term = term * x / t.CDouble(15.0) + result = result + term + term = term * x / t.CDouble(16.0) + result = result + term + term = term * x / t.CDouble(17.0) + result = result + term + term = term * x / t.CDouble(18.0) + result = result + term + return result + + +def log(x: t.CDouble) -> t.CDouble: + if x <= t.CDouble(0.0): + return t.CDouble(-1e308) + exponent: t.CInt = 0 + while x >= t.CDouble(2.0): + x /= t.CDouble(2.0) + exponent += 1 + while x < t.CDouble(1.0): + x *= t.CDouble(2.0) + exponent -= 1 + y: t.CDouble = (x - t.CDouble(1.0)) / (x + t.CDouble(1.0)) + y2: t.CDouble = y * y + result: t.CDouble = t.CDouble(2.0) * y + term: t.CDouble = t.CDouble(2.0) * y + term = term * y2 + result = result + term / t.CDouble(3.0) + term = term * y2 + result = result + term / t.CDouble(5.0) + term = term * y2 + result = result + term / t.CDouble(7.0) + term = term * y2 + result = result + term / t.CDouble(9.0) + term = term * y2 + result = result + term / t.CDouble(11.0) + term = term * y2 + result = result + term / t.CDouble(13.0) + term = term * y2 + result = result + term / t.CDouble(15.0) + term = term * y2 + result = result + term / t.CDouble(17.0) + term = term * y2 + result = result + term / t.CDouble(19.0) + term = term * y2 + result = result + term / t.CDouble(21.0) + result += t.CDouble(exponent) * U_M_LN2 + return result + + +def sqrt(x: t.CDouble) -> t.CDouble: + if x < t.CDouble(0.0): + return t.CDouble(0.0) + if x == t.CDouble(0.0): + return t.CDouble(0.0) + guess: t.CDouble = x / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + guess = (guess + x / guess) / t.CDouble(2.0) + return guess + + +def abs(x: t.CInt) -> t.CInt: + return x if x >= 0 else -x + +def fabs(x: t.CDouble) -> t.CDouble: + return x if x >= t.CDouble(0.0) else -x + +def labs(x: t.CLong) -> t.CLong: + return x if x >= 0 else -x + +def factorial(n: t.CInt) -> t.CDouble: + if n < 0: return t.CDouble(0.0) + if n == 0 or n == 1: return t.CDouble(1.0) + result: t.CDouble = t.CDouble(1.0) + for i in range(2, n + 1): + result *= i + return result + + +def combination(n: t.CInt, k: t.CInt) -> t.CDouble: + if k < 0 or k > n: return t.CDouble(0.0) + if k == 0 or k == n: return t.CDouble(1.0) + k = k if k < n - k else n - k + result: t.CDouble = t.CDouble(1.0) + for i in range(1, k + 1): + result = result * (n - k + i) / i + return result + + +def permutation(n: t.CInt, k: t.CInt) -> t.CDouble: + if k < 0 or k > n: return t.CDouble(0.0) + result: t.CDouble = t.CDouble(1.0) + for i in range(0, k): + result *= (n - i) + return result + +def pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: + if x == t.CDouble(0.0): + if y > t.CDouble(0.0): + return t.CDouble(0.0) + return t.CDouble(1.0) + if x < t.CDouble(0.0): + iy: t.CDouble = y * log(-x) + return exp(iy) + return exp(y * log(x)) + + +def powf(x: t.CDouble, y: t.CDouble) -> t.CDouble: + return pow(x, y) + + +def cbrt(x: t.CDouble) -> t.CDouble: + return pow(x, t.CDouble(1.0) / t.CDouble(3.0)) + +def hypot(x: t.CDouble, y: t.CDouble) -> t.CDouble: + return sqrt(x * x + y * y) + + +def floor(x: t.CDouble) -> t.CDouble: + if x >= t.CDouble(0.0): + return t.CDouble(t.CInt(x)) + i: t.CInt = t.CInt(x) + if x < t.CDouble(i): + return t.CDouble(i - 1) + else: + return t.CDouble(i) + + +def ceil(x: t.CDouble) -> t.CDouble: + if x < t.CDouble(0.0): + return t.CDouble(t.CInt(x)) + i: t.CInt = t.CInt(x) + if x > t.CDouble(i): + return t.CDouble(i + 1) + else: + return t.CDouble(i) + + +def round(x: t.CDouble) -> t.CDouble: + return floor(x + t.CDouble(0.5)) + +def trunc(x: t.CDouble) -> t.CDouble: + if x >= t.CDouble(0.0): + return floor(x) + else: + return ceil(x) + + +def fmod(x: t.CDouble, y: t.CDouble) -> t.CDouble: + if y == t.CDouble(0.0): + return t.CDouble(0.0) + result: t.CDouble = x - y * floor(x / y) + if result < t.CDouble(0.0): + result += y + return result + + +def fmodf(x: float, y: float) -> float: + if y == float(0.0): + return float(0.0) + result: float = x - y * floorf(x / y) + if result < float(0.0): + result += y + return result + + +def modf(x: t.CDouble, iptr: t.CDouble | t.CPtr) -> t.CDouble: + int_part: t.CDouble = floor(x) + c.Set(c.Deref(iptr), int_part) + return x - int_part + + +def log10(x: t.CDouble) -> t.CDouble: + return log(x) / U_M_LN10 + + +def log2(x: t.CDouble) -> t.CDouble: + return log(x) / U_M_LN2 + + +def exp2(x: t.CDouble) -> t.CDouble: + return pow(t.CDouble(2.0), x) + + +def expm1(x: t.CDouble) -> t.CDouble: + return exp(x) - t.CDouble(1.0) + + +def log1p(x: t.CDouble) -> t.CDouble: + return log(t.CDouble(1.0) + x) + + +def asinh(x: t.CDouble) -> t.CDouble: + return log(x + sqrt(x * x + t.CDouble(1.0))) + +def acosh(x: t.CDouble) -> t.CDouble: + return log(x + sqrt(x * x - t.CDouble(1.0))) + +def atanh(x: t.CDouble) -> t.CDouble: + return (log(t.CDouble(1.0) + x) - log(t.CDouble(1.0) - x)) / t.CDouble(2.0) + + +def gamma(x: t.CDouble) -> t.CDouble: + if x <= t.CDouble(0.0): return t.CDouble(0.0) + if x == t.CDouble(1.0) or x == t.CDouble(2.0): return t.CDouble(1.0) + ln_gamma: t.CDouble = t.CDouble(0.5) * log(t.CDouble(2.0) * U_M_PI / x) + x * log(x / U_M_E) + return exp(ln_gamma) + + +def erf(x: t.CDouble) -> t.CDouble: + sign: t.CDouble = t.CDouble(1.0) if x >= t.CDouble(0.0) else t.CDouble(-1.0) + abs_x: t.CDouble = fabs(x) + t0: t.CDouble = t.CDouble(4.0) * abs_x * abs_x / U_M_PI + exp_term: t.CDouble = exp(-t0) + return sign * sqrt(t.CDouble(1.0) - exp_term) + + +def erfc(x: t.CDouble) -> t.CDouble: + return t.CDouble(1.0) - erf(x) + +def sqrtf(x: t.CFloat) -> t.CFloat: + if x < t.CFloat(0.0): + return t.CFloat(0.0) + if x == t.CFloat(0.0): + return t.CFloat(0.0) + guess: t.CFloat = x / t.CFloat(2.0) + for i in range(10): + guess = (guess + x / guess) / t.CFloat(2.0) + return guess + +def sinf(x: t.CFloat) -> t.CFloat: + while x > t.CFloat(3.14159265): + x -= t.CFloat(6.28318530) + while x < t.CFloat(-3.14159265): + x += t.CFloat(6.28318530) + x2: t.CFloat = x * x + result: t.CFloat = x + term: t.CFloat = x + term = term * x2 / t.CFloat(6.0) + result = result - term + term = term * x2 / t.CFloat(20.0) + result = result + term + term = term * x2 / t.CFloat(42.0) + result = result - term + term = term * x2 / t.CFloat(72.0) + result = result + term + term = term * x2 / t.CFloat(110.0) + result = result - term + term = term * x2 / t.CFloat(156.0) + result = result + term + term = term * x2 / t.CFloat(210.0) + result = result - term + term = term * x2 / t.CFloat(272.0) + result = result + term + return result + +def cosf(x: t.CFloat) -> t.CFloat: + while x > t.CFloat(3.14159265): + x -= t.CFloat(6.28318530) + while x < t.CFloat(-3.14159265): + x += t.CFloat(6.28318530) + x2: t.CFloat = x * x + result: t.CFloat = t.CFloat(1.0) + term: t.CFloat = t.CFloat(1.0) + term = term * x2 / t.CFloat(2.0) + result = result - term + term = term * x2 / t.CFloat(12.0) + result = result + term + term = term * x2 / t.CFloat(30.0) + result = result - term + term = term * x2 / t.CFloat(56.0) + result = result + term + term = term * x2 / t.CFloat(90.0) + result = result - term + term = term * x2 / t.CFloat(132.0) + result = result + term + term = term * x2 / t.CFloat(182.0) + result = result - term + term = term * x2 / t.CFloat(240.0) + result = result + term + return result + +def tanf(x: t.CFloat) -> t.CFloat: + c: t.CFloat = cosf(x) + if c == t.CFloat(0.0): + return t.CFloat(0.0) + return sinf(x) / c + +def fabsf(x: t.CFloat) -> t.CFloat: + return x if x >= t.CFloat(0.0) else -x + +def floorf(x: t.CFloat) -> t.CFloat: + if x >= t.CFloat(0.0): + return t.CFloat(t.CInt(x)) + i: t.CInt = t.CInt(x) + if x < t.CFloat(i): + return t.CFloat(i - 1) + else: + return t.CFloat(i) + +def ceilf(x: t.CFloat) -> t.CFloat: + if x < t.CFloat(0.0): + return t.CFloat(t.CInt(x)) + i: t.CInt = t.CInt(x) + if x > t.CFloat(i): + return t.CFloat(i + 1) + else: + return t.CFloat(i) + + +class _U(t.CUnion): + d: t.CDouble + u: t.CUInt64T + +def isnan(x: t.CDouble) -> t.CStatic | t.CInt: + u: t.CUInt64T = t.CUInt64T(x) + exp_part: t.CUInt64T = (u >> 52) & 0x7FF + mantissa_part: t.CUInt64T = u & 0xFFFFFFFFFFFFF + if exp_part == 0x7FF: + if mantissa_part != 0: + return 1 + return 0 + +def isinf(x: t.CDouble) -> t.CStatic | t.CInt: + u: t.CUInt64T = t.CUInt64T(x) + exp_part: t.CUInt64T = (u >> 52) & 0x7FF + mantissa_part: t.CUInt64T = u & 0xFFFFFFFFFFFFF + if exp_part == 0x7FF: + if mantissa_part == 0: + return 1 + return 0 diff --git a/includes/vipersimd.py b/includes/vipersimd.py new file mode 100644 index 0000000..1e3b109 --- /dev/null +++ b/includes/vipersimd.py @@ -0,0 +1,460 @@ +import t, c + + +# ============================================================ +# AVX2 双精度向量运算 (4 x double, 256-bit YMM) +# ============================================================ + +def simd_add4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] + b[0:4]""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vaddpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_sub4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] - b[0:4]""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vsubpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_mul4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] * b[0:4]""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vmulpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_div4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] / b[0:4]""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vdivpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_sqrt4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = sqrt(a[0:4])""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vsqrtpd ymm0, ymm0 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0']) + + +def simd_neg4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = -a[0:4] (flip sign bit via XOR)""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vpcmpeqd ymm1, ymm1, ymm1 +vpsllq ymm1, ymm1, 63 +vxorpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_abs4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = |a[0:4]| (clear sign bit via ANDN)""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vpcmpeqd ymm1, ymm1, ymm1 +vpsllq ymm1, ymm1, 63 +vandnpd ymm0, ymm1, ymm0 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_max4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = max(a[0:4], b[0:4])""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vmaxpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_min4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = min(a[0:4], b[0:4])""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vminpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_mul_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] * scalar (broadcast)""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vbroadcastsd ymm1, [{c.AsmInp(scalar, t.ASM_DESCR.REG_ANY)}] +vmulpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_add_scalar4d(a: t.CPtr, scalar: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4] + scalar (broadcast)""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vbroadcastsd ymm1, [{c.AsmInp(scalar, t.ASM_DESCR.REG_ANY)}] +vaddpd ymm0, ymm0, ymm1 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_hsum4d(a: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0] = a[0]+a[1]+a[2]+a[3] (horizontal sum)""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vextractf128 xmm1, ymm0, 1 +vaddpd xmm0, xmm0, xmm1 +vhaddpd xmm0, xmm0, xmm0 +vmovsd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_dot4d(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0] = dot(a[0:4], b[0:4])""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vmulpd ymm0, ymm0, ymm1 +vextractf128 xmm1, ymm0, 1 +vaddpd xmm0, xmm0, xmm1 +vhaddpd xmm0, xmm0, xmm0 +vmovsd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1']) + + +def simd_fma4d(a: t.CPtr, b: t.CPtr, c_val: t.CPtr, out: t.CPtr) -> t.CVoid: + """AVX2: out[0:4] = a[0:4]*b[0:4] + c[0:4] (fused multiply-add) + Uses vfmadd213pd: ymm0 = ymm1 * ymm0 + ymm2 + Load order: b->ymm0, a->ymm1, c->ymm2""" + c.Asm(f"""vmovupd ymm0, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm1, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +vmovupd ymm2, [{c.AsmInp(c_val, t.ASM_DESCR.REG_ANY)}] +vfmadd213pd ymm0, ymm1, ymm2 +vmovupd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], ymm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'ymm2']) + + +# ============================================================ +# SSE2 单精度向量运算 (4 x float, 128-bit XMM) +# ============================================================ + +def simd_add4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """SSE2: out[0:4] = a[0:4] + b[0:4] (float)""" + c.Asm(f"""movups xmm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +movups xmm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +addps xmm0, xmm1 +movups [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'xmm0', 'xmm1']) + + +def simd_sub4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """SSE2: out[0:4] = a[0:4] - b[0:4] (float)""" + c.Asm(f"""movups xmm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +movups xmm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +subps xmm0, xmm1 +movups [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'xmm0', 'xmm1']) + + +def simd_mul4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """SSE2: out[0:4] = a[0:4] * b[0:4] (float)""" + c.Asm(f"""movups xmm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +movups xmm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +mulps xmm0, xmm1 +movups [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'xmm0', 'xmm1']) + + +def simd_div4f(a: t.CPtr, b: t.CPtr, out: t.CPtr) -> t.CVoid: + """SSE2: out[0:4] = a[0:4] / b[0:4] (float)""" + c.Asm(f"""movups xmm0, [{c.AsmInp(a, t.ASM_DESCR.REG_ANY)}] +movups xmm1, [{c.AsmInp(b, t.ASM_DESCR.REG_ANY)}] +divps xmm0, xmm1 +movups [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm0""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'xmm0', 'xmm1']) + + +# ============================================================ +# 批量数组运算 (AVX2 循环, 处理 N 个 double) +# ============================================================ + +def simd_add_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = a[i:i+4] + b[i:i+4], i=0,4,8,...""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r9, {c.AsmInp(b, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +test rcx, rcx +jz .Ladd_end +.Ladd_loop: +vmovupd ymm0, [r8] +vmovupd ymm1, [r9] +vaddpd ymm0, ymm0, ymm1 +vmovupd [r10], ymm0 +add r8, 32 +add r9, 32 +add r10, 32 +dec rcx +jnz .Ladd_loop +.Ladd_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'rcx', 'r8', 'r9', 'r10']) + + +def simd_sub_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = a[i:i+4] - b[i:i+4]""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r9, {c.AsmInp(b, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +test rcx, rcx +jz .Lsub_end +.Lsub_loop: +vmovupd ymm0, [r8] +vmovupd ymm1, [r9] +vsubpd ymm0, ymm0, ymm1 +vmovupd [r10], ymm0 +add r8, 32 +add r9, 32 +add r10, 32 +dec rcx +jnz .Lsub_loop +.Lsub_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'rcx', 'r8', 'r9', 'r10']) + + +def simd_mul_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = a[i:i+4] * b[i:i+4]""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r9, {c.AsmInp(b, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +test rcx, rcx +jz .Lmul_end +.Lmul_loop: +vmovupd ymm0, [r8] +vmovupd ymm1, [r9] +vmulpd ymm0, ymm0, ymm1 +vmovupd [r10], ymm0 +add r8, 32 +add r9, 32 +add r10, 32 +dec rcx +jnz .Lmul_loop +.Lmul_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'rcx', 'r8', 'r9', 'r10']) + + +def simd_div_array(a: t.CPtr, b: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = a[i:i+4] / b[i:i+4]""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r9, {c.AsmInp(b, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +test rcx, rcx +jz .Ldiv_end +.Ldiv_loop: +vmovupd ymm0, [r8] +vmovupd ymm1, [r9] +vdivpd ymm0, ymm0, ymm1 +vmovupd [r10], ymm0 +add r8, 32 +add r9, 32 +add r10, 32 +dec rcx +jnz .Ldiv_loop +.Ldiv_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'rcx', 'r8', 'r9', 'r10']) + + +def simd_sqrt_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = sqrt(a[i:i+4])""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +test rcx, rcx +jz .Lsqrt_end +.Lsqrt_loop: +vmovupd ymm0, [r8] +vsqrtpd ymm0, ymm0 +vmovupd [r10], ymm0 +add r8, 32 +add r10, 32 +dec rcx +jnz .Lsqrt_loop +.Lsqrt_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'rcx', 'r8', 'r10']) + + +def simd_abs_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = |a[i:i+4]|""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +vpcmpeqd ymm2, ymm2, ymm2 +vpsllq ymm2, ymm2, 63 +test rcx, rcx +jz .Labs_end +.Labs_loop: +vmovupd ymm0, [r8] +vandnpd ymm0, ymm2, ymm0 +vmovupd [r10], ymm0 +add r8, 32 +add r10, 32 +dec rcx +jnz .Labs_loop +.Labs_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm2', 'rcx', 'r8', 'r10']) + + +def simd_neg_array(a: t.CPtr, out: t.CPtr, n: t.CSizeT) -> t.CVoid: + """AVX2 batch: out[i:i+4] = -a[i:i+4] (flip sign bit via XOR)""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r10, {c.AsmInp(out, t.ASM_DESCR.REG_ANY)} +vpcmpeqd ymm2, ymm2, ymm2 +vpsllq ymm2, ymm2, 63 +test rcx, rcx +jz .Lneg_end +.Lneg_loop: +vmovupd ymm0, [r8] +vxorpd ymm0, ymm0, ymm2 +vmovupd [r10], ymm0 +add r8, 32 +add r10, 32 +dec rcx +jnz .Lneg_loop +.Lneg_end: +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm2', 'rcx', 'r8', 'r10']) + + +def simd_dot_array(a: t.CPtr, b: t.CPtr, n: t.CSizeT, out: t.CPtr) -> t.CVoid: + """AVX2 batch: out[0] = dot(a[0:n], b[0:n]) using FMA accumulation""" + c.Asm(f"""mov rcx, {c.AsmInp(n, t.ASM_DESCR.REG_ANY)} +shr rcx, 2 +mov r8, {c.AsmInp(a, t.ASM_DESCR.REG_ANY)} +mov r9, {c.AsmInp(b, t.ASM_DESCR.REG_ANY)} +vxorpd ymm2, ymm2, ymm2 +test rcx, rcx +jz .Ldot_end +.Ldot_loop: +vmovupd ymm0, [r8] +vmovupd ymm1, [r9] +vfmadd231pd ymm2, ymm0, ymm1 +add r8, 32 +add r9, 32 +dec rcx +jnz .Ldot_loop +.Ldot_end: +vextractf128 xmm1, ymm2, 1 +vaddpd xmm2, xmm2, xmm1 +vhaddpd xmm2, xmm2, xmm2 +vmovsd [{c.AsmInp(out, t.ASM_DESCR.REG_ANY)}], xmm2 +vzeroupper""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, 'ymm0', 'ymm1', 'ymm2', 'rcx', 'r8', 'r9']) + + +# ============================================================ +# LLVMIR 精确标量运算 (LLVM 内置函数) +# ============================================================ + +def simd_sqrt(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: sqrt(x)""" + return c.LLVMIR(f"call double @llvm.sqrt.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_fabs(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: |x|""" + return c.LLVMIR(f"call double @llvm.fabs.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_floor(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: floor(x)""" + return c.LLVMIR(f"call double @llvm.floor.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_ceil(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: ceil(x)""" + return c.LLVMIR(f"call double @llvm.ceil.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_round(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: round(x)""" + return c.LLVMIR(f"call double @llvm.round.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_trunc(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: trunc(x)""" + return c.LLVMIR(f"call double @llvm.trunc.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_fma(a: t.CDouble, b: t.CDouble, c_val: t.CDouble) -> t.CDouble: + """LLVM intrinsic: a*b + c (single rounding)""" + return c.LLVMIR(f"call double @llvm.fma.f64(double {c.LInp(a)}, double {c.LInp(b)}, double {c.LInp(c_val)})", t.CDouble) + + +def simd_copysign(mag: t.CDouble, sign: t.CDouble) -> t.CDouble: + """LLVM intrinsic: copysign(mag, sign)""" + return c.LLVMIR(f"call double @llvm.copysign.f64(double {c.LInp(mag)}, double {c.LInp(sign)})", t.CDouble) + + +def simd_neg(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: -x (via copysign, correct for negative/-0.0/NaN inputs) + + copysign(x, -1.0) is wrong for x < 0 (returns -|x| instead of +|x|). + copysign(x, -x) = |x| * sign(-x) = -x is correct for all IEEE 754 values. + """ + return c.LLVMIR(f"call double @llvm.copysign.f64(double {c.LInp(x)}, double {c.LInp(t.CDouble(0.0) - x)})", t.CDouble) + + +def simd_minnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: + """LLVM intrinsic: min(a, b) (NaN-aware)""" + return c.LLVMIR(f"call double @llvm.minnum.f64(double {c.LInp(a)}, double {c.LInp(b)})", t.CDouble) + + +def simd_maxnum(a: t.CDouble, b: t.CDouble) -> t.CDouble: + """LLVM intrinsic: max(a, b) (NaN-aware)""" + return c.LLVMIR(f"call double @llvm.maxnum.f64(double {c.LInp(a)}, double {c.LInp(b)})", t.CDouble) + + +def simd_exp(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: e^x""" + return c.LLVMIR(f"call double @llvm.exp.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_log(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: ln(x)""" + return c.LLVMIR(f"call double @llvm.log.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_sin(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: sin(x)""" + return c.LLVMIR(f"call double @llvm.sin.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_cos(x: t.CDouble) -> t.CDouble: + """LLVM intrinsic: cos(x)""" + return c.LLVMIR(f"call double @llvm.cos.f64(double {c.LInp(x)})", t.CDouble) + + +def simd_pow(x: t.CDouble, y: t.CDouble) -> t.CDouble: + """LLVM intrinsic: x^y""" + return c.LLVMIR(f"call double @llvm.pow.f64(double {c.LInp(x)}, double {c.LInp(y)})", t.CDouble) diff --git a/includes/viperstring.py b/includes/viperstring.py new file mode 100644 index 0000000..212d749 --- /dev/null +++ b/includes/viperstring.py @@ -0,0 +1 @@ +from .string import * diff --git a/includes/vrandom.py b/includes/vrandom.py new file mode 100644 index 0000000..55356b9 --- /dev/null +++ b/includes/vrandom.py @@ -0,0 +1,379 @@ +# vrandom - 硬件随机数生成库 +# 基于 x86 RDRAND / RDSEED 指令，使用内嵌汇编实现 +import t, c +import vipermath + + +# 重试次数限制（RDRAND 通常不需要重试，RDSEED 可能需要） +RDRAND_RETRY_LIMIT: t.CDefine = 10 +RDSEED_RETRY_LIMIT: t.CDefine = 100 + + +# ============================================================ +# RDRAND - 硬件随机数（伪随机数发生器输出） +# ============================================================ + +def rdrand16() -> t.CUInt16T: + val: t.CUInt32T = 0 + c.Asm(f"rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return t.CUInt16T(val) + + +def rdrand32() -> t.CUInt32T: + val: t.CUInt32T = 0 + c.Asm(f"rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return val + + +def rdrand64() -> t.CUInt64T: + val: t.CUInt64T = 0 + c.Asm(f"rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return val + + +# ============================================================ +# RDSEED - 硬件随机种子（熵源直接输出） +# ============================================================ + +def rdseed16() -> t.CUInt16T: + val: t.CUInt32T = 0 + c.Asm(f"rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return t.CUInt16T(val) + + +def rdseed32() -> t.CUInt32T: + val: t.CUInt32T = 0 + c.Asm(f"rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return val + + +def rdseed64() -> t.CUInt64T: + val: t.CUInt64T = 0 + c.Asm(f"rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)}", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC]) + return val + + +# ============================================================ +# Step 版本（带成功标志，匹配 GCC builtin 接口） +# 返回 1 表示成功，0 表示失败；随机值写入 *p +# ============================================================ + +def rdrand16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: + val: t.CUInt32T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = t.CUInt16T(val) + return t.CInt(ok) + + +def rdrand32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: + val: t.CUInt32T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = val + return t.CInt(ok) + + +def rdrand64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: + val: t.CUInt64T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdrand {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = val + return t.CInt(ok) + + +def rdseed16_step(p: t.CUInt16T | t.CPtr) -> t.CInt: + val: t.CUInt32T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = t.CUInt16T(val) + return t.CInt(ok) + + +def rdseed32_step(p: t.CUInt32T | t.CPtr) -> t.CInt: + val: t.CUInt32T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = val + return t.CInt(ok) + + +def rdseed64_step(p: t.CUInt64T | t.CPtr) -> t.CInt: + val: t.CUInt64T = 0 + ok: t.CUInt32T = 0 + c.Asm(f"""rdseed {c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG)} + setc al + movzx {c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)}, al""", + out=[c.AsmOut(val, t.ASM_DESCR.OUTPUT_REG), c.AsmOut(ok, t.ASM_DESCR.OUTPUT_REG)], + op=[t.ASM_DESCR.CLOBBER_CC, t.ASM_DESCR.CLOBBER_AL]) + p[0] = val + return t.CInt(ok) + + +# ============================================================ +# 带重试的便捷接口（自动重试直到成功或达到上限） +# ============================================================ + +def random_u16() -> t.CUInt16T: + r: t.CUInt16T = 0 + i: t.CInt = 0 + while i < RDRAND_RETRY_LIMIT: + if rdrand16_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +def random_u32() -> t.CUInt32T: + r: t.CUInt32T = 0 + i: t.CInt = 0 + while i < RDRAND_RETRY_LIMIT: + if rdrand32_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +def random_u64() -> t.CUInt64T: + r: t.CUInt64T = 0 + i: t.CInt = 0 + while i < RDRAND_RETRY_LIMIT: + if rdrand64_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +def seed_u16() -> t.CUInt16T: + r: t.CUInt16T = 0 + i: t.CInt = 0 + while i < RDSEED_RETRY_LIMIT: + if rdseed16_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +def seed_u32() -> t.CUInt32T: + r: t.CUInt32T = 0 + i: t.CInt = 0 + while i < RDSEED_RETRY_LIMIT: + if rdseed32_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +def seed_u64() -> t.CUInt64T: + r: t.CUInt64T = 0 + i: t.CInt = 0 + while i < RDSEED_RETRY_LIMIT: + if rdseed64_step(c.Addr(r)) == 1: + return r + i += 1 + return 0 + + +# ============================================================ +# Python random 标准库函数实现 +# 默认类型：t.CUInt64T (u64), t.CDouble (f64), t.CInt (i32) +# ============================================================ + +# ---------- 一、随机整数 ---------- + +def randint(a: t.CInt, b: t.CInt) -> t.CInt: + return t.CInt(a + t.CInt(random_u64() % t.CUInt64T(b - a + 1))) + + +def randrange(start: t.CInt, stop: t.CInt, step: t.CInt = 1) -> t.CInt: + width: t.CInt = 0 + if step > 0: + width = (stop - start + step - 1) // step + else: + width = (start - stop - step - 1) // (-step) + return t.CInt(start + t.CInt(random_u64() % t.CUInt64T(width)) * step) + + +def getrandbits(k: t.CInt) -> t.CUInt64T: + r: t.CUInt64T = rdrand64() + if k >= 64: + return r + mask: t.CUInt64T = (t.CUInt64T(1) << k) - t.CUInt64T(1) + return r & mask + + +# ---------- 二、随机浮点数 ---------- + +def random() -> t.CDouble: + return t.CDouble(rdrand64() >> 11) * (t.CDouble(1.0) / t.CDouble(9007199254740992.0)) + + +def uniform(a: t.CDouble, b: t.CDouble) -> t.CDouble: + return a + (b - a) * random() + + +def triangular(low: t.CDouble, high: t.CDouble, mode: t.CDouble) -> t.CDouble: + u: t.CDouble = random() + c: t.CDouble = (mode - low) / (high - low) + if u > c: + u = t.CDouble(1.0) - u + c = t.CDouble(1.0) - c + tmp: t.CDouble = low + low = high + high = tmp + return low + (high - low) * vipermath.sqrt(u * c) + + +def betavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: + y: t.CDouble = gammavariate(alpha, t.CDouble(1.0)) + if y != t.CDouble(0.0): + return y / (y + gammavariate(beta, t.CDouble(1.0))) + return t.CDouble(0.0) + + +def expovariate(lambd: t.CDouble) -> t.CDouble: + u: t.CDouble = random() + if u < t.CDouble(1e-300): + u = t.CDouble(1e-300) + return -vipermath.log(t.CDouble(1.0) - u) / lambd + + +def gammavariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: + u: t.CDouble = t.CDouble(0.0) + d: t.CDouble = t.CDouble(0.0) + c: t.CDouble = t.CDouble(0.0) + x: t.CDouble = t.CDouble(0.0) + v: t.CDouble = t.CDouble(0.0) + if alpha <= t.CDouble(0.0) or beta <= t.CDouble(0.0): + return t.CDouble(0.0) + if alpha < t.CDouble(1.0): + u = random() + if u < t.CDouble(1e-300): + u = t.CDouble(1e-300) + return gammavariate(alpha + t.CDouble(1.0), beta) * vipermath.pow(u, t.CDouble(1.0) / alpha) + d = alpha - t.CDouble(1.0) / t.CDouble(3.0) + c = t.CDouble(1.0) / vipermath.sqrt(t.CDouble(9.0) * d) + while 1: + x = gauss(t.CDouble(0.0), t.CDouble(1.0)) + v = t.CDouble(1.0) + c * x + if v > t.CDouble(0.0): + v = v * v * v + u = random() + if u < t.CDouble(1.0) - t.CDouble(0.0331) * x * x * x * x: + return d * v * beta + if vipermath.log(u) < t.CDouble(0.5) * x * x + d * (t.CDouble(1.0) - v + vipermath.log(v)): + return d * v * beta + + +def gauss(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: + u1: t.CDouble = random() + u2: t.CDouble = random() + if u1 < t.CDouble(1e-300): + u1 = t.CDouble(1e-300) + z0: t.CDouble = vipermath.sqrt(t.CDouble(-2.0) * vipermath.log(u1)) * vipermath.cos(t.CDouble(2.0) * vipermath.U_M_PI * u2) + return mu + sigma * z0 + + +def normalvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: + NV_MAGICCONST: t.CDouble = t.CDouble(1.7155277699214135) + u1: t.CDouble = t.CDouble(0.0) + u2: t.CDouble = t.CDouble(0.0) + z: t.CDouble = t.CDouble(0.0) + zz: t.CDouble = t.CDouble(0.0) + while 1: + u1 = random() + u2 = random() + if u2 < t.CDouble(1e-300): + continue + z = NV_MAGICCONST * (u1 - t.CDouble(0.5)) / u2 + zz = z * z / t.CDouble(4.0) + if zz <= t.CDouble(0.0) - vipermath.log(u2): + break + return mu + z * sigma + + +def lognormvariate(mu: t.CDouble, sigma: t.CDouble) -> t.CDouble: + return vipermath.exp(gauss(mu, sigma)) + + +def vonmisesvariate(mu: t.CDouble, kappa: t.CDouble) -> t.CDouble: + if kappa <= t.CDouble(1e-6): + return vipermath.U_M_PI * t.CDouble(2.0) * random() + s: t.CDouble = t.CDouble(0.5) / kappa + r: t.CDouble = s + vipermath.sqrt(t.CDouble(1.0) + s * s) + u1: t.CDouble = t.CDouble(0.0) + u2: t.CDouble = t.CDouble(0.0) + u3: t.CDouble = t.CDouble(0.0) + z: t.CDouble = t.CDouble(0.0) + d: t.CDouble = t.CDouble(0.0) + q: t.CDouble = t.CDouble(0.0) + f: t.CDouble = t.CDouble(0.0) + theta: t.CDouble = t.CDouble(0.0) + two_pi: t.CDouble = vipermath.U_M_PI * t.CDouble(2.0) + while 1: + u1 = random() + z = vipermath.cos(vipermath.U_M_PI * u1) + d = t.CDouble(1.0) / (r + z) + u2 = random() + if u2 < t.CDouble(1.0) - d * d: + break + if u2 < (t.CDouble(1.0) - z) * vipermath.exp(kappa * (z - t.CDouble(1.0)) - d * kappa * r): + break + q = t.CDouble(1.0) / r + f = (q + z) / (t.CDouble(1.0) + q * z) + u3 = random() + if u3 > t.CDouble(0.5): + theta = mu + vipermath.acos(f) + else: + theta = mu - vipermath.acos(f) + while theta < t.CDouble(0.0): + theta += two_pi + while theta >= two_pi: + theta -= two_pi + return theta + + +def paretovariate(alpha: t.CDouble) -> t.CDouble: + u: t.CDouble = random() + if u < t.CDouble(1e-300): + u = t.CDouble(1e-300) + return t.CDouble(1.0) / vipermath.pow(t.CDouble(1.0) - u, t.CDouble(1.0) / alpha) + + +def weibullvariate(alpha: t.CDouble, beta: t.CDouble) -> t.CDouble: + u: t.CDouble = random() + if u < t.CDouble(1e-300): + u = t.CDouble(1e-300) + return alpha * vipermath.pow(t.CDouble(-1.0) * vipermath.log(t.CDouble(1.0) - u), t.CDouble(1.0) / beta) diff --git a/includes/vthreading.py b/includes/vthreading.py new file mode 100644 index 0000000..a8162fb --- /dev/null +++ b/includes/vthreading.py @@ -0,0 +1,104 @@ +import t, c +from stdint import * +from w32.win32base import * +from w32.win32process import CreateThread, GetExitCodeThread, GetCurrentThreadId, TlsAlloc, TlsFree, TlsGetValue, TlsSetValue +from w32.win32sync import WaitForSingleObject + +# 全部导入锁类，用户也可按需导入 +from lock import Lock +from event import Event +from condition import Condition +from rwlock import RWLock + + +@t.Object +class Thread: + """线程类，封装 Windows CreateThread。 + + 线程回调签名: def thread_func(arg: t.CPtr) -> t.CPtr + 注意: Windows API 限制返回值为 DWORD(32位)，64位指针会被截断。 + 如需返回 64 位结果，请通过 arg 参数结构体传递。 + + 用法: + def worker(arg: t.CPtr) -> t.CPtr: + # 线程逻辑 + return NULL + + t: Thread = Thread(worker, arg_ptr) + t.start() + t.join() + """ + handle: HANDLE + thread_id: ULONG + target: VOIDPTR + arg: VOIDPTR + started: BOOL + + def __init__(self, target: VOIDPTR, arg: VOIDPTR = NULL): + self.handle = NULL + self.thread_id = 0 + self.target = target + self.arg = arg + self.started = FALSE + + def start(self): + if self.started: + return + self.handle = CreateThread(NULL, 0, self.target, self.arg, 0, c.Addr(self.thread_id)) + self.started = TRUE + + def join(self, timeout_ms: ULONG = INFINITE) -> BOOL: + if not self.started: + return FALSE + result: ULONG = WaitForSingleObject(self.handle, timeout_ms) + return result == WAIT_OBJECT_0 + + def is_alive(self) -> BOOL: + if not self.started: + return FALSE + result: ULONG = WaitForSingleObject(self.handle, 0) + return result == WAIT_TIMEOUT + + def get_exit_code(self) -> ULONG: + code: ULONG = 0 + GetExitCodeThread(self.handle, c.Addr(code)) + return code + + def close(self): + if self.handle != NULL: + CloseHandle(self.handle) + self.handle = NULL + + +@t.Object +class local: + """TLS (Thread Local Storage) 封装。 + + 用法: + tls: local = local() + tls.set(value_ptr) + val: t.CPtr = tls.get() + """ + index: ULONG + + def __init__(self): + self.index = TlsAlloc() + + def get(self) -> VOIDPTR: + return TlsGetValue(self.index) + + def set(self, value: VOIDPTR) -> BOOL: + return TlsSetValue(self.index, value) + + def free(self) -> BOOL: + return TlsFree(self.index) + + +def get_current_thread_id() -> ULONG: + """获取当前线程 ID""" + return GetCurrentThreadId() + + +def sleep_ms(ms: ULONG): + """让当前线程休眠指定毫秒数""" + Sleep(ms) diff --git a/includes/w32/fileio.py b/includes/w32/fileio.py new file mode 100644 index 0000000..2e7dd28 --- /dev/null +++ b/includes/w32/fileio.py @@ -0,0 +1,394 @@ +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 + + +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 + +INVALID_SET_FILE_POINTER: t.CDefine = -1 + +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + + def __init__(self, filename: str, mode: ULONG, share: ULONG = SHARE_READ): + self.closed = True + self.can_read = False + self.can_write = False + self.is_append = False + self._share_mode = share + + access: ULONG = 0 + disposition: ULONG = w32.win32file.OPEN_EXISTING + + if mode == MODE.R: + access = w32.win32file.GENERIC_READ + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + elif mode == MODE.W: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_ALWAYS + self.can_write = True + elif mode == MODE.A: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_ALWAYS + self.can_write = True + self.is_append = True + elif mode == MODE.RP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + self.can_write = True + elif mode == MODE.WP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_ALWAYS + self.can_read = True + self.can_write = True + elif mode == MODE.AP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_ALWAYS + self.can_read = True + self.can_write = True + self.is_append = True + elif mode == MODE.X: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_NEW + self.can_write = True + elif mode == MODE.XP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_NEW + self.can_read = True + self.can_write = True + else: + access = w32.win32file.GENERIC_READ + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + + self.handle = w32.win32file.CreateFileA( + filename, access, self._share_mode, + None, disposition, w32.win32file.FILE_ATTRIBUTE_NORMAL, None + ) + + if self.handle == w32.win32base.INVALID_HANDLE_VALUE: + self.closed = True + return + + self.closed = False + + if self.is_append: + self.seek(0, SEEK_END) + + def __enter__(self) -> 'File' | t.CPtr: + return self + + def __exit__(self): + self.close() + + def read(self, buf: bytes, count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + bytes_read: ULONG = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf, count, t.CPtr(c.Addr(bytes_read)), None) + if result == 0: return FRESULT.ERR_IO + return LONG(bytes_read) + + def write(self, buf: bytes, count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_write: return FRESULT.ERR_PERM + bytes_written: ULONG = 0 + result: BOOL = w32.win32file.WriteFile(self.handle, buf, count, t.CPtr(c.Addr(bytes_written)), None) + if result == 0: return FRESULT.ERR_IO + return LONG(bytes_written) + + def write_str(self, s: str) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_write: return FRESULT.ERR_PERM + length: ULONG = 0 + p: bytes = s + while c.Deref(p) != 0: + length += 1 + p += 1 + return self.write(s, length) + + def seek(self, offset: LONG, origin: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + result: LONG = w32.win32file.SetFilePointer(self.handle, offset, None, origin) + if result == -1: return FRESULT.ERR_IO + return result + + def tell(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + return self.seek(0, SEEK_CUR) + + def close(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + w32.win32base.CloseHandle(self.handle) + self.closed = True + return FRESULT.OK + + def flush(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + w32.win32file.FlushFileBuffers(self.handle) + return FRESULT.OK + + def size(self) -> LONGLONG: + if self.closed: return -1 + li: w32.win32base.LARGE_INTEGER = w32.win32base.LARGE_INTEGER() + result: BOOL = w32.win32file.GetFileSizeEx(self.handle, t.CPtr(c.Addr(li))) + if result == 0: return -1 + return li.QuadPart + + def readline(self, buf: bytes, max_count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + if max_count < 2: return FRESULT.ERR + total: ULONG = 0 + bytes_read: ULONG = 0 + ch: t.CInt = 0 + while total < max_count - 1: + bytes_read = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf + total, 1, t.CPtr(c.Addr(bytes_read)), None) + if result == 0 or bytes_read == 0: break + ch = c.Deref(buf + total) + total += 1 + if ch == 10: break + if total < max_count: + buf[total] = 0 + return LONG(total) + + def read_all(self, buf: bytes, max_count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + if max_count < 2: return FRESULT.ERR + # stdio.printf("[DBG read_all] before size()\n") + file_size: LONGLONG = self.size() + # stdio.printf("[DBG read_all] after size(), file_size=%lld\n", file_size) + if file_size < 0: return FRESULT.ERR_IO + to_read: ULONG = 0 + if file_size < max_count - 1: + to_read = ULONG(file_size) + else: + to_read = max_count - 1 + # stdio.printf("[DBG read_all] to_read=%u, before seek\n", to_read) + self.seek(0, SEEK_SET) + # stdio.printf("[DBG read_all] after seek, before ReadFile\n") + bytes_read: ULONG = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf, to_read, t.CPtr(c.Addr(bytes_read)), None) + # stdio.printf("[DBG read_all] after ReadFile, result=%d, bytes_read=%u\n", result, bytes_read) + if result == 0: return FRESULT.ERR_IO + if bytes_read < max_count: + buf[bytes_read] = 0 + return LONG(bytes_read) + + +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + + def __init__(self, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG = SHARE_READ): + self.closed = True + self.can_read = False + self.can_write = False + self.is_append = False + self._share_mode = share + + access: ULONG = 0 + disposition: ULONG = w32.win32file.OPEN_EXISTING + + if mode == MODE.R: + access = w32.win32file.GENERIC_READ + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + elif mode == MODE.W: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_ALWAYS + self.can_write = True + elif mode == MODE.A: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_ALWAYS + self.can_write = True + self.is_append = True + elif mode == MODE.RP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + self.can_write = True + elif mode == MODE.WP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_ALWAYS + self.can_read = True + self.can_write = True + elif mode == MODE.AP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.OPEN_ALWAYS + self.can_read = True + self.can_write = True + self.is_append = True + elif mode == MODE.X: + access = w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_NEW + self.can_write = True + elif mode == MODE.XP: + access = w32.win32file.GENERIC_READ | w32.win32file.GENERIC_WRITE + disposition = w32.win32file.CREATE_NEW + self.can_read = True + self.can_write = True + else: + access = w32.win32file.GENERIC_READ + disposition = w32.win32file.OPEN_EXISTING + self.can_read = True + + self.handle = w32.win32file.CreateFileW( + filename, access, self._share_mode, + None, disposition, w32.win32file.FILE_ATTRIBUTE_NORMAL, None + ) + + if self.handle == w32.win32base.INVALID_HANDLE_VALUE: + self.closed = True + return + + self.closed = False + + if self.is_append: + self.seek(0, SEEK_END) + + def __enter__(self) -> 'FileW' | t.CPtr: + return self + + def __exit__(self): + self.close() + + def read(self, buf: bytes, count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + bytes_read: ULONG = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf, count, t.CPtr(c.Addr(bytes_read)), None) + if result == 0: return FRESULT.ERR_IO + return LONG(bytes_read) + + def write(self, buf: bytes, count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_write: return FRESULT.ERR_PERM + bytes_written: ULONG = 0 + result: BOOL = w32.win32file.WriteFile(self.handle, buf, count, t.CPtr(c.Addr(bytes_written)), None) + if result == 0: return FRESULT.ERR_IO + return LONG(bytes_written) + + def write_str(self, s: str) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_write: return FRESULT.ERR_PERM + length: ULONG = 0 + p: bytes = s + while c.Deref(p) != 0: + length += 1 + p += 1 + return self.write(s, length) + + def seek(self, offset: LONG, origin: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + result: LONG = w32.win32file.SetFilePointer(self.handle, offset, None, origin) + if result == -1: return FRESULT.ERR_IO + return result + + def tell(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + return self.seek(0, SEEK_CUR) + + def close(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + w32.win32base.CloseHandle(self.handle) + self.closed = True + return FRESULT.OK + + def flush(self) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + w32.win32file.FlushFileBuffers(self.handle) + return FRESULT.OK + + def size(self) -> LONGLONG: + if self.closed: return -1 + li: w32.win32base.LARGE_INTEGER = w32.win32base.LARGE_INTEGER() + result: BOOL = w32.win32file.GetFileSizeEx(self.handle, t.CPtr(c.Addr(li))) + if result == 0: return -1 + return li.QuadPart + + def readline(self, buf: bytes, max_count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + if max_count < 2: return FRESULT.ERR + total: ULONG = 0 + bytes_read: ULONG = 0 + ch: t.CInt = 0 + while total < max_count - 1: + bytes_read = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf + total, 1, t.CPtr(c.Addr(bytes_read)), None) + if result == 0 or bytes_read == 0: break + ch = c.Deref(buf + total) + total += 1 + if ch == 10: break + if total < max_count: + buf[total] = 0 + return LONG(total) + + def read_all(self, buf: bytes, max_count: ULONG) -> LONG: + if self.closed: return FRESULT.ERR_CLOSED + if not self.can_read: return FRESULT.ERR_PERM + if max_count < 2: return FRESULT.ERR + file_size: LONGLONG = self.size() + if file_size < 0: return FRESULT.ERR_IO + to_read: ULONG = 0 + if file_size < max_count - 1: + to_read = ULONG(file_size) + else: + to_read = max_count - 1 + self.seek(0, SEEK_SET) + bytes_read: ULONG = 0 + result: BOOL = w32.win32file.ReadFile(self.handle, buf, to_read, t.CPtr(c.Addr(bytes_read)), None) + if result == 0: return FRESULT.ERR_IO + if bytes_read < max_count: + buf[bytes_read] = 0 + return LONG(bytes_read) + + +def open(filename: str, mode: ULONG, share: ULONG = SHARE_READ) -> File | t.CPtr: + return File(filename, mode, share) + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG = SHARE_READ) -> FileW | t.CPtr: + return FileW(filename, mode, share) diff --git a/includes/w32/win32base.py b/includes/w32/win32base.py new file mode 100644 index 0000000..883fbe8 --- /dev/null +++ b/includes/w32/win32base.py @@ -0,0 +1,80 @@ +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr + +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES(): + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL + +class OVERLAPPED(): + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE + +class FILETIME(): + dwLowDateTime: DWORD + dwHighDateTime: DWORD + +class SYSTEMTIME(): + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD + +class GUID(): + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR + +class LARGE_INTEGER(): + QuadPart: LONGLONG + +class ULARGE_INTEGER(): + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass +def SetLastError(dwErrCode: ULONG) -> t.State: pass +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass +def Sleep(dwMilliseconds: ULONG) -> t.State: pass +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass +def GetTickCount() -> ULONG | t.State: pass +def GetTickCount64() -> ULONGLONG | t.State: pass +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/includes/w32/win32console.py b/includes/w32/win32console.py new file mode 100644 index 0000000..a90c0c0 --- /dev/null +++ b/includes/w32/win32console.py @@ -0,0 +1,111 @@ +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 + +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 + +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 + +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD(): + X: SHORT + Y: SHORT + +class SMALL_RECT(): + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT + +class CONSOLE_SCREEN_BUFFER_INFO(): + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD + +class CONSOLE_CURSOR_INFO(): + dwSize: ULONG + bVisible: BOOL + +class CHAR_INFO(): + UnicodeChar: WCHAR + Attributes: WORD + +class KEY_EVENT_RECORD(): + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG + +class MOUSE_EVENT_RECORD(): + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG + +class WINDOW_BUFFER_SIZE_RECORD(): + dwSize: COORD + +class INPUT_RECORD(): + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass +def GetConsoleCP() -> UINT | t.State: pass +def GetConsoleOutputCP() -> UINT | t.State: pass +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass +def AllocConsole() -> BOOL | t.State: pass +def FreeConsole() -> BOOL | t.State: pass +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/includes/w32/win32file.py b/includes/w32/win32file.py new file mode 100644 index 0000000..aedfe35 --- /dev/null +++ b/includes/w32/win32file.py @@ -0,0 +1,143 @@ +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 + +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 + +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 + +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 + +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 + +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 + +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) + +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION(): + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG + +class WIN32_FIND_DATAA(): + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] + +class WIN32_FIND_DATAW(): + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 diff --git a/includes/w32/win32memory.py b/includes/w32/win32memory.py new file mode 100644 index 0000000..824f39a --- /dev/null +++ b/includes/w32/win32memory.py @@ -0,0 +1,64 @@ +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 + +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 + +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION(): + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass +def GetProcessHeap() -> HANDLE | t.State: pass +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/includes/w32/win32process.py b/includes/w32/win32process.py new file mode 100644 index 0000000..c46cd01 --- /dev/null +++ b/includes/w32/win32process.py @@ -0,0 +1,112 @@ +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF + +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF + +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 + +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 + +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 + +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 + +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA(): + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE + +class STARTUPINFOW(): + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE + +class PROCESS_INFORMATION(): + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass +def GetCurrentProcess() -> HANDLE | t.State: pass +def GetCurrentProcessId() -> ULONG | t.State: pass +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass +def GetCurrentThread() -> HANDLE | t.State: pass +def GetCurrentThreadId() -> ULONG | t.State: pass +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass +def TlsAlloc() -> ULONG | t.State: pass +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/includes/w32/win32sync.py b/includes/w32/win32sync.py new file mode 100644 index 0000000..9f13e6c --- /dev/null +++ b/includes/w32/win32sync.py @@ -0,0 +1,65 @@ +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION(): + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +# 条件变量 (Condition Variables) - Vista+ +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +# 一次性初始化 (One-Time Initialization) - Vista+ +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/includes/w32/winsock2.py b/includes/w32/winsock2.py new file mode 100644 index 0000000..04450c5 --- /dev/null +++ b/includes/w32/winsock2.py @@ -0,0 +1,104 @@ +import t, c +from stdint import * +from w32.win32base import * + + +# ============================================================ +# w32/winsock2.py — WinSock2 API 声明 (ws2_32.dll) +# ============================================================ + +# 版本常量 +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 + +# 地址族 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 + +# 套接字类型 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 + +# 协议 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 + +# 选项级别 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 + +# 特殊值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 + +# 发送/接收标志 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 + +# 关闭方式 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + + +# ============================================================ +# 结构体 +# ============================================================ +# struct sockaddr_in — 16 字节 +class WSASocketAddr: + family: u16 # sin_family + port: u16 # sin_port + addr: u32 # sin_addr.s_addr + zero: u64 # sin_zero + + +# struct WSADATA — 400 字节 +class WSAData: + wVersion: WORD # @ 0 + wHighVersion: WORD # @ 2 + szDescription: BYTE # @ 4 (257 bytes + padding) + szSystemStatus: BYTE # @ 261 (129 bytes + padding) + iMaxSockets: u16 # @ 390 + iMaxUdpDg: u16 # @ 392 + lpVendorInfo: CHARPTR # @ 394 (实际上指针是 8 字节对齐) + + +# struct hostent — 自然对齐: h_name(8) + h_aliases(8) + h_addrtype(2) + h_length(2) + auto-pad(4) + h_addr_list(8) = 32 +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR + + +# struct timeval +class WinTimeVal: + tv_sec: LONG # 秒 + tv_usec: LONG # 微秒 + + +# ============================================================ +# WinSock2 API 函数 +# ============================================================ + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass +def WSACleanup() -> INT | t.State: pass +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass +def closesocket(s: u64) -> INT | t.State: pass +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass +def listen(s: u64, backlog: INT) -> INT | t.State: pass +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass +def shutdown(s: u64, how: INT) -> INT | t.State: pass +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass +def ntohs(netshort: u16) -> u16 | t.State: pass +def htons(hostshort: u16) -> u16 | t.State: pass +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass \ No newline at end of file diff --git a/includes/zlib/__init__.py b/includes/zlib/__init__.py new file mode 100644 index 0000000..829b69e --- /dev/null +++ b/includes/zlib/__init__.py @@ -0,0 +1 @@ +from .pyzlib import * diff --git a/includes/zlib/pyzlib.py b/includes/zlib/pyzlib.py new file mode 100644 index 0000000..538b5a7 --- /dev/null +++ b/includes/zlib/pyzlib.py @@ -0,0 +1,695 @@ +from stdint import * +import zlib.zdeflate as zdeflate +import zlib.zinflate as zinflate +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import stdlib +import string +import memhub +import t, c + + +# ============================================================ +# Constants - matching Python zlib module names and values +# ============================================================ + +# Compression levels +Z_NO_COMPRESSION: t.CDefine = 0 +Z_BEST_SPEED: t.CDefine = 1 +Z_BEST_COMPRESSION: t.CDefine = 9 +Z_DEFAULT_COMPRESSION: t.CDefine = (-1) + +# Compression methods +DEFLATED: t.CDefine = 8 + +# Flush modes +Z_NO_FLUSH: t.CDefine = 0 +Z_PARTIAL_FLUSH: t.CDefine = 1 +Z_SYNC_FLUSH: t.CDefine = 2 +Z_FULL_FLUSH: t.CDefine = 3 +Z_FINISH: t.CDefine = 4 +Z_BLOCK: t.CDefine = 5 +Z_TREES: t.CDefine = 6 + +# Strategies +Z_DEFAULT_STRATEGY: t.CDefine = 0 +Z_FILTERED: t.CDefine = 1 +Z_HUFFMAN_ONLY: t.CDefine = 2 +Z_RLE: t.CDefine = 3 +Z_FIXED: t.CDefine = 4 + +# Return codes +Z_OK: t.CDefine = 0 +Z_STREAM_END: t.CDefine = 1 +Z_NEED_DICT: t.CDefine = 2 +Z_ERRNO: t.CDefine = (-1) +Z_STREAM_ERROR: t.CDefine = (-2) +Z_DATA_ERROR: t.CDefine = (-3) +Z_MEM_ERROR: t.CDefine = (-4) +Z_BUF_ERROR: t.CDefine = (-5) +Z_VERSION_ERROR: t.CDefine = (-6) + +# Window / Buffer constants +MAX_WBITS: t.CDefine = 15 +DEF_BUF_SIZE: t.CDefine = 16384 +DEF_MEM_LEVEL: t.CDefine = 8 + +# Version +ZLIB_VERSION: t.CDefine = "1.3.2" + +pyzlib_error_msg: t.CArray[t.CChar, 512] = "" +pyzlib_error_code_val: t.CInt = 0 + +# ============================================================ +# Structures +# ============================================================ +@t.Object +class Compress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + is_finished: t.CInt + level: t.CInt + method: t.CInt + wbits: t.CInt + memLevel: t.CInt + strategy: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + last_pos: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + header_written: t.CInt + + def compress(self, + data: BYTEPTR, data_len: t.CSizeT, + out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + if self.is_finished: + set_error(Z_STREAM_ERROR, "compress object already finished") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + if data_len == 0: + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p, 0, 1) + return p + while self.input_buf_len + data_len > self.input_buf_cap: + self.input_buf_cap *= 2 + if self.pool == None: + new_buf: BYTEPTR = BYTEPTR(realloc(self.input_buf, self.input_buf_cap)) + else: + new_buf: BYTEPTR = BYTEPTR(self.pool.alloc(self.input_buf_cap)) + if new_buf: + memcpy(new_buf, self.input_buf, self.input_buf_len) + self.pool.free(self.input_buf) + if not new_buf: + set_error(Z_MEM_ERROR, "out of memory") + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p2: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p2, 0, 1) + return p2 + self.input_buf = new_buf + memcpy(self.input_buf + self.input_buf_len, data, data_len) + self.input_buf_len += data_len + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p3: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p3, 0, 1) + return p3 + + def flush(self, mode: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + if mode == Z_FINISH: + if not self.header_written: + self.header_written = 1 + input_data: BYTEPTR = self.input_buf + input_len: t.CSizeT = self.input_buf_len + if input_len == 0: + s.writer.write_bits(1, 1) + s.writer.write_bits(1, 2) + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + lit_tree.encode_symbol(256, c.Addr(s.writer)) + else: + s.adler = zchecksum.zchecksum_adler32(input_data, input_len, 1) + s.compress_block(input_data, input_len, 1) + s.is_finished = 1 + if s.wbits >= 0: + s.writer.align() + adler: t.CUInt32T = s.adler + s.writer.buf[s.writer.byte_pos + 0] = (adler >> 24) & 0xFF + s.writer.buf[s.writer.byte_pos + 1] = (adler >> 16) & 0xFF + s.writer.buf[s.writer.byte_pos + 2] = (adler >> 8) & 0xFF + s.writer.buf[s.writer.byte_pos + 3] = (adler >> 0) & 0xFF + s.writer.byte_pos += 4 + self.is_finished = 1 + total: t.CSizeT = s.writer.total() + if self.pool == None: + result: BYTEPTR = BYTEPTR(malloc(total)) + else: + result: BYTEPTR = BYTEPTR(self.pool.alloc(total)) + if result: memcpy(result, s.writer.buf, total) + c.Set(c.Deref(out_len), total) + if self.pool == None: + free(self.input_buf) + else: + self.pool.free(self.input_buf) + self.input_buf = None + self.input_buf_len = 0 + self.input_buf_cap = 0 + return result + elif mode == Z_SYNC_FLUSH or mode == Z_FULL_FLUSH: + input_data: BYTEPTR = self.input_buf + input_len: t.CSizeT = self.input_buf_len + if input_len > 0: + s.adler = zchecksum.zchecksum_adler32(input_data, input_len, s.adler) + s.compress_block(input_data, input_len, 0) + self.input_buf_len = 0 + s.writer.write_bits(0, 1) + s.writer.write_bits(0, 2) + s.writer.align() + prev_pos: t.CSizeT = self.last_pos + total: t.CSizeT = s.writer.total() + delta: t.CSizeT = total - prev_pos + self.last_pos = s.writer.byte_pos + if delta == 0: + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p4: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p4, 0, 1) + return p4 + + if self.pool == None: + result: BYTEPTR = BYTEPTR(malloc(delta)) + else: + result: BYTEPTR = BYTEPTR(self.pool.alloc(delta)) + if result: memcpy(result, s.writer.buf + prev_pos, delta) + c.Set(c.Deref(out_len), delta) + return result + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p5: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p5, 0, 1) + return p5 + + def copy(self) -> Compress | t.CPtr: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "compress object not initialized") + return None + if self.pool == None: + copy_obj: Compress | t.CPtr = calloc(1, Compress.__sizeof__()) + else: + copy_obj: Compress | t.CPtr = self.pool.alloc(Compress.__sizeof__()) + memset(copy_obj, 0, Compress.__sizeof__()) + if not copy_obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + copy_obj.pool = self.pool + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + copy_obj.stream = s.copy() + if not copy_obj.stream: + if self.pool == None: + free(copy_obj) + else: + self.pool.free(copy_obj) + set_error(Z_MEM_ERROR, "out of memory") + return None + copy_obj.is_initialized = 1 + copy_obj.is_finished = self.is_finished + copy_obj.level = self.level + copy_obj.method = self.method + copy_obj.wbits = self.wbits + copy_obj.memLevel = self.memLevel + copy_obj.strategy = self.strategy + copy_obj.last_pos = self.last_pos + copy_obj.header_written = self.header_written + if self.input_buf and self.input_buf_len > 0: + if self.pool == None: + copy_obj.input_buf = BYTEPTR(calloc(1, self.input_buf_cap)) + else: + copy_obj.input_buf = BYTEPTR(self.pool.alloc(self.input_buf_cap)) + memset(copy_obj.input_buf, 0, self.input_buf_cap) + if copy_obj.input_buf: + memcpy(copy_obj.input_buf, self.input_buf, self.input_buf_len) + copy_obj.input_buf_cap = self.input_buf_cap + copy_obj.input_buf_len = self.input_buf_len + else: + if self.pool == None: + copy_obj.input_buf = BYTEPTR(calloc(1, 256)) + else: + copy_obj.input_buf = BYTEPTR(self.pool.alloc(256)) + memset(copy_obj.input_buf, 0, 256) + copy_obj.input_buf_cap = 256 + copy_obj.input_buf_len = 0 + if self.zdict and self.zdict_len > 0: + copy_obj.zdict = clone_bytes(self.pool, self.zdict, self.zdict_len) + copy_obj.zdict_len = self.zdict_len + return copy_obj + + def delete(self): + if self.is_initialized and self.stream: + s: zdeflate.zdeflate_stream | t.CPtr = self.stream # 隐式类型转换 + s.destroy() + if self.pool == None: + free(self.zdict) + free(self.input_buf) + free(self) + else: + self.pool.free(self.zdict) + self.pool.free(self.input_buf) + self.pool.free(self) + + def __del__(self): + self.delete() + + +@t.Object +class Decompress: + pool: memhub.MemManager | t.CPtr + stream: VOIDPTR + is_initialized: t.CInt + eof: t.CInt + wbits: t.CInt + zdict: BYTEPTR + zdict_len: t.CSizeT + _unused_data: BYTEPTR + _unused_data_len: t.CSizeT + _unused_data_cap: t.CSizeT + _unconsumed_tail: BYTEPTR + _unconsumed_tail_len: t.CSizeT + _unconsumed_tail_cap: t.CSizeT + input_buf: BYTEPTR + input_buf_len: t.CSizeT + input_buf_cap: t.CSizeT + + def decompress(self, data: BYTEPTR, data_len: t.CSizeT, + max_length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + if self.eof: + if data and data_len > 0: + append_bytes(self.pool, c.Addr(self._unused_data), c.Addr(self._unused_data_len), + c.Addr(self._unused_data_cap), data, data_len) + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p6: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p6, 0, 1) + return p6 + self._unconsumed_tail_len = 0 + if data and data_len > 0: + append_bytes(self.pool, c.Addr(self.input_buf), c.Addr(self.input_buf_len), + c.Addr(self.input_buf_cap), data, data_len) + if self.input_buf_len == 0: + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p7: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p7, 0, 1) + return p7 + s: zinflate.zinflate_stream | t.CPtr = zinflate.zinflate_create(self.pool, self.wbits) + if not s: + set_error(Z_MEM_ERROR, "out of memory") + return None + if self.zdict and self.zdict_len > 0: + s.set_dictionary(self.zdict, self.zdict_len) + out: t.CUInt8T | t.CPtr = None + err: t.CInt = s.decompress(self.input_buf, self.input_buf_len, + max_length, c.Addr(out), out_len) + if err != 0: + s.destroy() + set_error(Z_DATA_ERROR, "decompression failed") + return None + if s.is_finished: + self.eof = 1 + if s.input_pos < s.input_len: + append_bytes(self.pool, c.Addr(self._unused_data), c.Addr(self._unused_data_len), + c.Addr(self._unused_data_cap), + s.input_data + s.input_pos, + s.input_len - s.input_pos) + self.input_buf_len = 0 + else: + consumed: t.CSizeT = s.input_pos + if consumed < self.input_buf_len: + remaining: t.CSizeT = self.input_buf_len - consumed + memmove(self.input_buf, self.input_buf + consumed, remaining) + self.input_buf_len = remaining + else: + self.input_buf_len = 0 + s.destroy() + return out + + def flush(self, length: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + c.Set(c.Deref(out_len), 0) + if self.pool == None: + return BYTEPTR(calloc(1, 1)) + else: + p8: BYTEPTR = BYTEPTR(self.pool.alloc(1)) + memset(p8, 0, 1) + return p8 + + def copy(self) -> Decompress | t.CPtr: + if not self.is_initialized: + set_error(Z_STREAM_ERROR, "decompress object not initialized") + return None + if self.pool == None: + copy_obj: Decompress | t.CPtr = calloc(1, Decompress.__sizeof__()) + else: + copy_obj: Decompress | t.CPtr = self.pool.alloc(Decompress.__sizeof__()) + memset(copy_obj, 0, Decompress.__sizeof__()) + if not copy_obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + copy_obj.pool = self.pool + copy_obj.is_initialized = 1 + copy_obj.eof = self.eof + copy_obj.wbits = self.wbits + if self.zdict and self.zdict_len > 0: + copy_obj.zdict = clone_bytes(self.pool, self.zdict, self.zdict_len) + copy_obj.zdict_len = self.zdict_len + if self._unused_data and self._unused_data_len > 0: + copy_obj._unused_data = clone_bytes(self.pool, self._unused_data, self._unused_data_len) + copy_obj._unused_data_len = self._unused_data_len + copy_obj._unused_data_cap = self._unused_data_len + if self._unconsumed_tail and self._unconsumed_tail_len > 0: + copy_obj._unconsumed_tail = clone_bytes(self.pool, self._unconsumed_tail, self._unconsumed_tail_len) + copy_obj._unconsumed_tail_len = self._unconsumed_tail_len + copy_obj._unconsumed_tail_cap = self._unconsumed_tail_len + if self.input_buf and self.input_buf_len > 0: + copy_obj.input_buf = clone_bytes(self.pool, self.input_buf, self.input_buf_len) + copy_obj.input_buf_len = self.input_buf_len + copy_obj.input_buf_cap = self.input_buf_cap + else: + if self.pool == None: + copy_obj.input_buf = BYTEPTR(calloc(1, 256)) + else: + copy_obj.input_buf = BYTEPTR(self.pool.alloc(256)) + memset(copy_obj.input_buf, 0, 256) + copy_obj.input_buf_cap = 256 + copy_obj.input_buf_len = 0 + return copy_obj + + def delete(self): + if self.pool == None: + free(self.zdict) + free(self._unused_data) + free(self._unconsumed_tail) + free(self.input_buf) + free(self) + else: + self.pool.free(self.zdict) + self.pool.free(self._unused_data) + self.pool.free(self._unconsumed_tail) + self.pool.free(self.input_buf) + self.pool.free(self) + + def unused_data(self, length: t.CSizeT | t.CPtr) -> BYTEPTR: + #if not self: + # if length: + # c.Set(c.Deref(length), 0) + # return None + if length: + c.Set(c.Deref(length), self._unused_data_len) + return self._unused_data + + def unconsumed_tail(self, length: t.CSizeT | t.CPtr) -> BYTEPTR: + #if not self: + # if length: + # c.Set(c.Deref(length), 0) + # return None + if length: + c.Set(c.Deref(length), self._unconsumed_tail_len) + return self._unconsumed_tail + +# def eof(self) -> t.CInt: +# # if not obj: return 0 +# return self.eof + + + +def set_error(code: int, msg: str): + global pyzlib_error_code_val, pyzlib_error_msg + pyzlib_error_code_val = code + if msg: + strncpy(pyzlib_error_msg, msg, pyzlib_error_msg.__sizeof__() - 1) + pyzlib_error_msg[pyzlib_error_msg.__sizeof__() - 1] = '\0' + else: + pyzlib_error_msg[0] = '\0' + +def clone_bytes(pool: memhub.MemManager | t.CPtr, src: BYTEPTR, length: t.CSizeT) -> BYTEPTR: + if not src or length == 0: return None + if pool == None: + dst: BYTEPTR = BYTEPTR(malloc(length)) + else: + dst: BYTEPTR = BYTEPTR(pool.alloc(length)) + if dst: memcpy(dst, src, length) + return dst + +def append_bytes(pool: memhub.MemManager | t.CPtr, buf: BYTE | t.CPtr[t.CPtr], length: t.CSizeT | t.CPtr, cap: t.CSizeT | t.CPtr, + data: BYTEPTR, data_len: t.CSizeT) -> t.CInt: + if not data or data_len == 0: return 0 + needed: t.CSizeT = c.Deref(length) + data_len + if needed > c.Deref(cap): + new_cap: t.CSizeT = c.Deref(cap) * 2 + if new_cap < needed: new_cap = needed + if pool == None: + new_buf: BYTEPTR = BYTEPTR(realloc(c.Deref(buf), new_cap)) + else: + new_buf: BYTEPTR = BYTEPTR(pool.alloc(new_cap)) + if new_buf: + memcpy(new_buf, c.Deref(buf), c.Deref(length)) + pool.free(c.Deref(buf)) + if not new_buf: return -1 + c.Set(c.Deref(buf), new_buf) + c.Set(c.Deref(cap), new_cap) + memcpy(c.Deref(buf) + c.Deref(length), data, data_len) + c.Set(c.Deref(length), c.Deref(length) + data_len) + return 0 + + +def shrink_to_fit(pool: memhub.MemManager | t.CPtr, buf: BYTEPTR, length: t.CSizeT) -> BYTEPTR: + if length == 0: + if pool == None: + free(buf) + return BYTEPTR(calloc(1, 1)) + else: + pool.free(buf) + p: BYTEPTR = BYTEPTR(pool.alloc(1)) + memset(p, 0, 1) + return p + if pool == None: + result: BYTEPTR = BYTEPTR(realloc(buf, length)) + return result if result else buf + else: + result: BYTEPTR = BYTEPTR(pool.alloc(length)) + if result: + memcpy(result, buf, length) + pool.free(buf) + return result + return buf + + +def runtime_version() -> str: + return ZLIB_VERSION + +def get_error() -> str: + return pyzlib_error_msg + +def get_error_code() -> t.CInt: + return pyzlib_error_code_val + +def clear_error(): + global pyzlib_error_msg, pyzlib_error_code_val + pyzlib_error_msg[0] = '\0' + pyzlib_error_code_val = 0 + +def compress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, + level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not data and data_len > 0: + set_error(Z_STREAM_ERROR, "data is None but data_len > 0") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: + actual_wbits = -MAX_WBITS + raw_result: UINT8PTR = zdeflate.zdeflate_one_shot(pool, data, data_len, level, actual_wbits, out_len) + if not raw_result: + set_error(Z_DATA_ERROR, "compression failed") + return None + if wbits > MAX_WBITS: + gzip_len: t.CSizeT = 10 + c.Deref(out_len) + 8 + if pool == None: + gzip_buf: BYTEPTR = BYTEPTR(malloc(gzip_len)) + else: + gzip_buf: BYTEPTR = BYTEPTR(pool.alloc(gzip_len)) + if not gzip_buf: + if pool == None: + free(raw_result) + else: + pool.free(raw_result) + set_error(Z_MEM_ERROR, "out of memory") + return None + pos: t.CSizeT = 0 + gzip_buf[pos + 0] = 0x1F + gzip_buf[pos + 1] = 0x8B + gzip_buf[pos + 2] = 0x08 + gzip_buf[pos + 3] = 0x00 + gzip_buf[pos + 4] = 0x00 + gzip_buf[pos + 5] = 0x00 + gzip_buf[pos + 6] = 0x00 + gzip_buf[pos + 7] = 0x00 + gzip_buf[pos + 8] = 0x00 + gzip_buf[pos + 9] = 0xFF + pos += 10 + + memcpy(gzip_buf + pos, raw_result, c.Deref(out_len)) + pos += c.Deref(out_len) + if pool == None: + free(raw_result) + else: + pool.free(raw_result) + + crc: t.CUInt32T = zchecksum.zchecksum_crc32(data, data_len, 0) + gzip_buf[pos + 0] = (crc) & 0xFF + gzip_buf[pos + 1] = (crc >> 8) & 0xFF + gzip_buf[pos + 2] = (crc >> 16) & 0xFF + gzip_buf[pos + 3] = (crc >> 24) & 0xFF + gzip_buf[pos + 4] = (data_len) & 0xFF + gzip_buf[pos + 5] = (data_len >> 8) & 0xFF + gzip_buf[pos + 6] = (data_len >> 16) & 0xFF + gzip_buf[pos + 7] = (data_len >> 24) & 0xFF + pos += 8 + c.Set(c.Deref(out_len), pos) + return shrink_to_fit(pool, gzip_buf, pos) + return shrink_to_fit(pool, raw_result, c.Deref(out_len)) + + +def decompress(pool: memhub.MemManager | t.CPtr, data: BYTEPTR, data_len: t.CSizeT, + wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> BYTEPTR: + if not data and data_len > 0: + set_error(Z_STREAM_ERROR, "data is None but data_len > 0") + return None + if not out_len: + set_error(Z_STREAM_ERROR, "out_len is None") + return None + result: UINT8PTR = zinflate.zinflate_one_shot(pool, data, data_len, wbits, bufsize, out_len) + if not result: + set_error(Z_DATA_ERROR, "decompression failed") + return None + return shrink_to_fit(pool, result, c.Deref(out_len)) + +def compressobj(pool: memhub.MemManager | t.CPtr, level: t.CInt, method: t.CInt, wbits: t.CInt, + memLevel: t.CInt, strategy: t.CInt, + zdict: BYTEPTR, zdict_len: t.CSizeT) -> Compress | t.CPtr: + if pool == None: + obj: Compress | t.CPtr = calloc(1, Compress.__sizeof__()) + else: + obj: Compress | t.CPtr = pool.alloc(Compress.__sizeof__()) + memset(obj, 0, Compress.__sizeof__()) + if not obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + obj.pool = pool + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: actual_wbits = wbits - 16 + s: zdeflate.zdeflate_stream | t.CPtr = zdeflate.zdeflate_create(pool, level, actual_wbits, memLevel, strategy) + if not s: + if pool == None: + free(obj) + else: + pool.free(obj) + set_error(Z_MEM_ERROR, "out of memory") + return None + obj.stream = s + obj.is_initialized = 1 + obj.level = level + obj.method = method + obj.wbits = wbits + obj.memLevel = memLevel + obj.strategy = strategy + if pool == None: + obj.input_buf = BYTEPTR(calloc(1, 256)) + else: + obj.input_buf = BYTEPTR(pool.alloc(256)) + memset(obj.input_buf, 0, 256) + obj.input_buf_cap = 256 + obj.input_buf_len = 0 + obj.header_written = 0 + if zdict and zdict_len > 0: + s.set_dictionary(zdict, zdict_len) + obj.zdict = clone_bytes(pool, zdict, zdict_len) + obj.zdict_len = zdict_len + return obj + + +def decompressobj(pool: memhub.MemManager | t.CPtr, wbits: t.CInt, zdict: BYTEPTR, zdict_len: t.CSizeT) -> Decompress | t.CPtr: + if pool == None: + obj: Decompress | t.CPtr = calloc(1, Decompress.__sizeof__()) + else: + obj: Decompress | t.CPtr = pool.alloc(Decompress.__sizeof__()) + memset(obj, 0, Decompress.__sizeof__()) + if not obj: + set_error(Z_MEM_ERROR, "out of memory") + return None + obj.pool = pool + actual_wbits: t.CInt = wbits + if wbits > MAX_WBITS: actual_wbits = wbits - 16 + obj.stream = None + obj.is_initialized = 1 + obj.wbits = actual_wbits + if zdict and zdict_len > 0: + obj.zdict = clone_bytes(pool, zdict, zdict_len) + obj.zdict_len = zdict_len + if pool == None: + obj.input_buf = BYTEPTR(calloc(1, 256)) + else: + obj.input_buf = BYTEPTR(pool.alloc(256)) + memset(obj.input_buf, 0, 256) + obj.input_buf_cap = 256 + obj.input_buf_len = 0 + return obj + +def zlib_adler32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: + return zchecksum.zchecksum_adler32(data, length, UINT32(value)) + +def zlib_crc32(data: BYTEPTR, length: t.CSizeT, value: t.CUnsignedLong) -> t.CUnsignedLong: + return zchecksum.zchecksum_crc32(data, length, UINT32(value)) diff --git a/includes/zlib/zchecksum.py b/includes/zlib/zchecksum.py new file mode 100644 index 0000000..d46246f --- /dev/null +++ b/includes/zlib/zchecksum.py @@ -0,0 +1,89 @@ +from stdint import * +import t, c + + +def zchecksum_adler32(data: UINT8PTR, length: t.CSizeT, init: UINT32) -> t.CUInt32T: + a: t.CUInt32T = (init >> 0) & 0xFFFF + b: t.CUInt32T = (init >> 16) & 0xFFFF + if not data or length == 0: + return init + i: t.CSizeT + for i in range(length): + a = (a + data[i]) % 65521 + b = (b + a) % 65521 + return (b << 16) | a + +crc32_table: t.CArray[t.CUInt32T, 256] = [ + 0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, + 0x076DC419, 0x706AF48F, 0xE963A535, 0x9E6495A3, + 0x0EDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988, + 0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91, + 0x1DB71064, 0x6AB020F2, 0xF3B97148, 0x84BE41DE, + 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7, + 0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, + 0x14015C4F, 0x63066CD9, 0xFA0F3D63, 0x8D080DF5, + 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172, + 0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B, + 0x35B5A8FA, 0x42B2986C, 0xDBBBBBD6, 0xACBCCB40, + 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59, + 0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, + 0x21B4F4B5, 0x56B3C423, 0xCFBA9599, 0xB8BDA50F, + 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924, + 0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D, + 0x76DC4190, 0x01DB7106, 0x98D220BC, 0xEFD5102A, + 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433, + 0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, + 0x7F6A0DBB, 0x086D3D2D, 0x91646C97, 0xE6635C01, + 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E, + 0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457, + 0x65B0D9C6, 0x12B7E950, 0x8BBEB8EA, 0xFCB9887C, + 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65, + 0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, + 0x4ADFA541, 0x3DD895D7, 0xA4D1C46D, 0xD3D6F4FB, + 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0, + 0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7D49, + 0x5005713C, 0x270241AA, 0xBE0B1010, 0xC90C2086, + 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F, + 0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, + 0x59B33D17, 0x2EB40D81, 0xB7BD5C3B, 0xC0BA6CAD, + 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A, + 0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683, + 0xE3630B12, 0x94643B84, 0x0D6D6A3E, 0x7A6A5AA8, + 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1, + 0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, + 0xF762575D, 0x806567CB, 0x196C3671, 0x6E6B06E7, + 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC, + 0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5, + 0xD6D6A3E8, 0xA1D1937E, 0x38D8C2C4, 0x4FDFF252, + 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B, + 0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, + 0xDF60EFC3, 0xA867DF55, 0x316E8EEF, 0x4669BE79, + 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236, + 0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F, + 0xC5BA3BBE, 0xB2BD0B28, 0x2BB45A92, 0x5CB36A04, + 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D, + 0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, + 0x9C0906A9, 0xEB0E363F, 0x72076785, 0x05005713, + 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38, + 0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21, + 0x86D3D2D4, 0xF1D4E242, 0x68DDB3F8, 0x1FDA836E, + 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777, + 0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, + 0x8F659EFF, 0xF862AE69, 0x616BFFD3, 0x166CCF45, + 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2, + 0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB, + 0xAED16A4A, 0xD9D65ADC, 0x40DF0B66, 0x37D83BF0, + 0xA9BCAE53, 0xDede9EC5, 0x47D7897F, 0x30D0B8E9, + 0xBDDA8B1C, 0xCADD8B8A, 0x53D3903A, 0x24D4C2AC, + 0xBAD03605, 0xCDD70693, 0x54DE5729, 0x23D967BF, + 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94, + 0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D +] + +def zchecksum_crc32(data: UINT8PTR, length: t.CSizeT, init: t.CUInt32T) -> t.CUInt32T: + crc: t.CUInt32T = init ^ 0xFFFFFFFF + if not data or length == 0: return init + i: t.CSizeT + for i in range(length): + crc = crc32_table[(crc ^ data[i]) & 0xFF] ^ (crc >> 8) + return crc ^ 0xFFFFFFFF diff --git a/includes/zlib/zdef.py b/includes/zlib/zdef.py new file mode 100644 index 0000000..88d5006 --- /dev/null +++ b/includes/zlib/zdef.py @@ -0,0 +1,288 @@ +from stdint import * +import stddef +import string +import stdlib +import memhub +import t, c + + +# ============================================================ +# DEFLATE / zlib constants +# ============================================================ +ZDEFLATE_WINDOW_BITS: t.CDefine = 15 +ZDEFLATE_WINDOW_SIZE: t.CDefine = (1 << ZDEFLATE_WINDOW_BITS) +ZDEFLATE_WINDOW_MASK: t.CDefine = (ZDEFLATE_WINDOW_SIZE - 1) + +ZDEFLATE_MIN_MATCH: t.CDefine = 3 +ZDEFLATE_MAX_MATCH: t.CDefine = 258 + +ZDEFLATE_HASH_BITS: t.CDefine = 15 +ZDEFLATE_HASH_SIZE: t.CDefine = (1 << ZDEFLATE_HASH_BITS) +ZDEFLATE_HASH_MASK: t.CDefine = (ZDEFLATE_HASH_SIZE - 1) + +ZDEFLATE_MAX_CODES: t.CDefine = 288 +ZDEFLATE_MAX_DIST_CODES: t.CDefine = 32 +ZDEFLATE_MAX_BITS: t.CDefine = 15 +ZDEFLATE_MAX_CODELEN_CODES: t.CDefine = 19 + +ZDEFLATE_LIT_COUNT: t.CDefine = 286 +ZDEFLATE_DIST_COUNT: t.CDefine = 30 + +ZDEFLATE_LEN_SYMBOLS_BASE: t.CDefine = 257 +ZDEFLATE_END_OF_BLOCK: t.CDefine = 256 + +# ============================================================ +# Length / Distance extra bits tables (RFC 1951) +# ============================================================ +zdeflate_len_extra_bits: t.CArray[t.CInt, 29] = [ + 0, 0, 0, 0, 0, 0, 0, 0, + 1, 1, 1, 1, + 2, 2, 2, 2, + 3, 3, 3, 3, + 4, 4, 4, 4, + 5, 5, 5, 5, + 0 +] + +zdeflate_len_base: t.CArray[t.CInt, 29] = [ + 3, 4, 5, 6, 7, 8, 9, 10, + 11, 13, 15, 17, + 19, 23, 27, 31, + 35, 43, 51, 59, + 67, 83, 99, 115, + 131, 163, 195, 227, + 258 +] + +zdeflate_dist_extra_bits: t.CArray[t.CInt, 30] = [ + 0, 0, 0, 0, + 1, 1, + 2, 2, + 3, 3, + 4, 4, + 5, 5, + 6, 6, + 7, 7, + 8, 8, + 9, 9, + 10, 10, + 11, 11, + 12, 12, + 13, 13 +] + +zdeflate_dist_base: t.CArray[t.CInt, 30] = [ + 1, 2, 3, 4, + 5, 7, + 9, 13, + 17, 25, + 33, 49, + 65, 97, + 129, 193, + 257, 385, + 513, 769, + 1025, 1537, + 2049, 3073, + 4097, 6145, + 8193, 12289, + 16385, 24577 +] + +zdeflate_codelen_order: t.CArray[int, 19] = [ + 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15 +] + +# ============================================================ +# Bit writer - writes bits LSB first into a byte buffer +# ============================================================ +@t.Object +class zbit_writer: + pool: memhub.MemManager | t.CPtr + buf: BYTE | t.CPtr + cap: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + + def __init__(self, pool: memhub.MemManager | t.CPtr): + self.pool = pool + self.cap = 4096 + if pool == None: + self.buf = BYTEPTR(malloc(self.cap)) + else: + self.buf = BYTEPTR(pool.alloc(self.cap)) + memset(self.buf, 0, self.cap) + self.byte_pos = 0 + self.bit_pos = 0 + + def ensure(self, need: t.CSizeT): + while self.byte_pos + need + 4 >= self.cap: + new_cap: t.CSizeT = self.cap * 2 + if self.pool == None: + new_buf: BYTE | t.CPtr = BYTEPTR(realloc(self.buf, new_cap)) + else: + new_buf: BYTE | t.CPtr = BYTEPTR(self.pool.alloc(new_cap)) + if new_buf: + memcpy(new_buf, self.buf, self.byte_pos + (self.bit_pos + 7) / 8) + self.pool.free(self.buf) + if not new_buf: return + memset(new_buf + self.cap, 0, new_cap - self.cap) + self.buf = new_buf + self.cap = new_cap + + def write_bits(self, value: UINT, nbits: t.CInt): + self.ensure((nbits + 7) / 8 + 1) + for i in range(nbits): + if value & (UINT(1) << i): + self.buf[self.byte_pos] |= (UINT(1) << self.bit_pos) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + + def write_bits_rev(self, code: UINT, nbits: t.CInt): + self.ensure((nbits + 7) / 8 + 1) + for i in range(nbits - 1, -1, -1): + if code & (UINT(1) << i): + self.buf[self.byte_pos] |= (UINT(1) << self.bit_pos) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + + def stored_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + self.write_bits(1 if final else 0, 1) + self.write_bits(0, 2) + self.align() + + pos: t.CSizeT = 0 + while pos < length: + block_length: t.CSizeT = length - pos + if block_length > 65535: + block_length = 65535 + n: t.CUInt16T = t.CUInt16T(block_length) + ncomp: t.CUInt16T = ~n + self.write_bits(n, 16) + self.write_bits(ncomp, 16) + + self.ensure(block_length) + i: t.CSizeT = 0 + for i in range(block_length): + self.buf[self.byte_pos] = data[pos + i] + self.byte_pos += 1 + pos += block_length + if pos < length: + self.write_bits(0, 1) + self.write_bits(0, 2) + self.align() + + def write_zlib_header(self, wbits: t.CInt, level: t.CInt): + cinfo: t.CInt = wbits - 8 + if cinfo < 1: cinfo = 1 + if cinfo > 7: cinfo = 7 + cmf: t.CUnsignedChar = t.CUnsignedChar((cinfo << 4) | 8) + + flevel: t.CInt = 0 + if level >= 5: flevel = 3 + elif level >= 3: flevel = 1 + + flg: t.CUnsignedChar = t.CUnsignedChar(flevel << 6) + flg |= t.CUnsignedChar(31 - (cmf * 256 + flg) % 31) + + self.buf[self.byte_pos + 0] = cmf + self.buf[self.byte_pos + 1] = flg + self.byte_pos += 2 + + + def write_gzip_header(self): + self.buf[self.byte_pos + 0] = 0x1F + self.buf[self.byte_pos + 1] = 0x8B + self.buf[self.byte_pos + 2] = 0x08 + self.buf[self.byte_pos + 3] = 0x00 + self.buf[self.byte_pos + 4] = 0x00 + self.buf[self.byte_pos + 5] = 0x00 + self.buf[self.byte_pos + 6] = 0x00 + self.buf[self.byte_pos + 7] = 0x00 + self.buf[self.byte_pos + 8] = 0x00 + self.buf[self.byte_pos + 9] = 0xFF + self.byte_pos += 10 + + def align(self): + if self.bit_pos > 0: + self.bit_pos = 0 + self.byte_pos += 1 + + def total(self) -> t.CSizeT: + return self.byte_pos + (1 if (self.bit_pos > 0) else 0) + + def free(self): + if self.pool == None: + free(self.buf) + else: + self.pool.free(self.buf) + self.buf = None + self.cap = 0 + self.byte_pos = 0 + self.bit_pos = 0 + + def __del__(self): + self.free() + +# ============================================================ +# Bit reader - reads bits LSB first from a byte buffer +# ============================================================ +@t.Object +class zbit_reader: + buf: BYTE | t.CPtr + length: t.CSizeT + byte_pos: t.CSizeT + bit_pos: t.CInt + + def init(self, buf: BYTE | t.CPtr, length: t.CSizeT): + self.buf = buf + self.length = length + self.byte_pos = 0 + self.bit_pos = 0 + + def read_bits(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits): + if self.byte_pos >= self.length: return -1 + if self.buf[self.byte_pos] & (UINT(1) << self.bit_pos): + val |= (UINT(1) << i) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + def read_bits_rev(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits - 1, 0, -1): + if self.byte_pos >= self.length: return -1 + if self.buf[self.byte_pos] & (UINT(1) << self.bit_pos): + val |= (UINT(1) << i) + self.bit_pos += 1 + if self.bit_pos == 8: + self.bit_pos = 0 + self.byte_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + def align(self): + if self.bit_pos > 0: + self.bit_pos = 0 + self.byte_pos += 1 + +# ============================================================ +# Memory allocation helper for bare metal +# Can be replaced with custom allocator +# ============================================================ +def zdef_alloc(pool: memhub.MemManager | t.CPtr, n: t.CSizeT) -> VOIDPTR: + if pool == None: return malloc(n) + return pool.alloc(n) +def zdef_free(pool: memhub.MemManager | t.CPtr, p: VOIDPTR): + if pool == None: + free(p) + return + pool.free(p) diff --git a/includes/zlib/zdeflate.py b/includes/zlib/zdeflate.py new file mode 100644 index 0000000..dfbb6eb --- /dev/null +++ b/includes/zlib/zdeflate.py @@ -0,0 +1,434 @@ +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zhuff as zhuff +import zlib.zdef as zdef +import string +import memhub +import t, c + + +@t.Object +class zdeflate_stream: + pool: memhub.MemManager | t.CPtr + writer: zdef.zbit_writer + window: BYTE | t.CPtr + window_pos: t.CInt + window_size: t.CInt + hash_head: t.CInt | t.CPtr + hash_prev: t.CInt | t.CPtr + level: t.CInt + strategy: t.CInt + wbits: t.CInt + is_finished: t.CInt + adler: t.CUInt32T + + def find_match(self, data: BYTE | t.CPtr, data_len: t.CSizeT, + pos: t.CSizeT, best_dist: t.CInt | t.CPtr) -> t.CInt: + if pos + 2 >= data_len: return 0 + h: t.CUnsignedInt = zdeflate_hash(data + pos) + chain: t.CInt = self.hash_head[h] + best_len: t.CInt = zdef.ZDEFLATE_MIN_MATCH - 1 + c.Set(c.Deref(best_dist), 0) + max_chain: t.CInt = 128 if (self.level >= 5) else (32 if (self.level >= 2) else 4) + limit: t.CInt = (1 << self.wbits) if (self.wbits > 0) else zdef.ZDEFLATE_WINDOW_SIZE + if limit > zdef.ZDEFLATE_WINDOW_SIZE: + limit = zdef.ZDEFLATE_WINDOW_SIZE + attempts: t.CInt = 0 + while chain >= 0 and attempts < max_chain: + dist: t.CInt = t.CInt(pos) - chain + if dist <= 0 or dist > limit: break + match_len: t.CInt = 0 + max_len: t.CInt = t.CInt(data_len - pos) + if max_len > zdef.ZDEFLATE_MAX_MATCH: + max_len = zdef.ZDEFLATE_MAX_MATCH + while match_len < max_len and data[pos + match_len] == data[chain + match_len]: + match_len += 1 + if match_len > best_len: + best_len = match_len + c.Set(c.Deref(best_dist), dist) + if best_len >= zdef.ZDEFLATE_MAX_MATCH: break + chain = self.hash_prev[chain & zdef.ZDEFLATE_WINDOW_MASK] + if chain <= t.CInt(pos) - limit: break + attempts += 1 + return best_len if (best_len >= zdef.ZDEFLATE_MIN_MATCH) else 0 + + def update_hash(self, data: BYTE | t.CPtr, pos: t.CSizeT): + if pos + 2 < pos: return + h: t.CUnsignedInt = zdeflate_hash(data + pos) + self.hash_prev[pos & zdef.ZDEFLATE_WINDOW_MASK] = self.hash_head[h] + self.hash_head[h] = t.CInt(pos) + + def write_fixed_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + dist_tree.build_fixed_dist_tree() + self.writer.write_bits(1 if final else 0, 1) + self.writer.write_bits(1, 2) + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if self.level > 0 and pos + 2 < length: + match_len = self.find_match(data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_tree.encode_symbol(len_sym, c.Addr(self.writer)) + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_sym - 257] + if extra > 0: + self.writer.write_bits(match_len - zdef.zdeflate_len_base[len_sym - 257], extra) + dist_sym: t.CInt = zdeflate_dist_to_symbol(best_dist) + dist_tree.encode_symbol(dist_sym, c.Addr(self.writer)) + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + self.writer.write_bits(best_dist - zdef.zdeflate_dist_base[dist_sym], extra) + for i in range(match_len): + self.update_hash(data, pos + i) + pos += match_len + else: + lit_tree.encode_symbol(data[pos], c.Addr(self.writer)) + self.update_hash(data, pos) + pos += 1 + lit_tree.encode_symbol(256, c.Addr(self.writer)) + + def encode_block_data(self, data: UINT8PTR, length: t.CSizeT, + lit_tree: zhuff.zhuff_tree | t.CPtr, dist_tree: zhuff.zhuff_tree | t.CPtr): + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if self.level > 0 and pos + 2 < length: + match_len = self.find_match(data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_tree.encode_symbol(len_sym, c.Addr(self.writer)) + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_sym - 257] + if extra > 0: + self.writer.write_bits(match_len - zdef.zdeflate_len_base[len_sym - 257], extra) + dist_sym: t.CInt = zdeflate_dist_to_symbol(best_dist) + dist_tree.encode_symbol(dist_sym, c.Addr(self.writer)) + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + self.writer.write_bits(best_dist - zdef.zdeflate_dist_base[dist_sym], extra) + for j in range(match_len): + self.update_hash(data, pos + j) + pos += match_len + else: + lit_tree.encode_symbol(data[pos], c.Addr(self.writer)) + self.update_hash(data, pos) + pos += 1 + lit_tree.encode_symbol(256, c.Addr(self.writer)) + + + def write_dynamic_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + lit_freqs: t.CArray[t.CInt, 288] + dist_freqs: t.CArray[t.CInt, 32] + zdeflate_count_freqs(lit_freqs, dist_freqs, self, data, length) + hlit: t.CInt = 286 + while hlit > 257 and lit_freqs[hlit - 1] == 0: hlit -= 1 + hdist: t.CInt = 30 + while hdist > 1 and dist_freqs[hdist - 1] == 0: hdist -= 1 + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_codes(lit_freqs, hlit, zdef.ZDEFLATE_MAX_BITS) + dist_tree.build_codes(dist_freqs, hdist, zdef.ZDEFLATE_MAX_BITS) + all_lengths: t.CArray[t.CInt, 288 + 32] + for i in range(hlit): all_lengths[i] = lit_tree.codes[i].bits + for i in range(hdist): all_lengths[hlit + i] = dist_tree.codes[i].bits + total_lengths: t.CInt = hlit + hdist + cl_freqs: t.CArray[t.CInt, 19] + zdeflate_count_cl_freqs(all_lengths, total_lengths, cl_freqs) + cl_tree = zhuff.zhuff_tree() + cl_tree.build_codes(cl_freqs, 19, 7) + hclen: int = 19 + while hclen > 4 and cl_tree.codes[zdef.zdeflate_codelen_order[hclen - 1]].bits == 0: hclen -= 1 + self.writer.write_bits(1 if final else 0, 1) + self.writer.write_bits(2, 2) + self.writer.write_bits(hlit - 257, 5) + self.writer.write_bits(hdist - 1, 5) + self.writer.write_bits(hclen - 4, 4) + for j in range(hclen): + self.writer.write_bits(cl_tree.codes[zdef.zdeflate_codelen_order[j]].bits, 3) + zdeflate_write_cl_encoded(all_lengths, total_lengths, c.Addr(self.writer), c.Addr(cl_tree)) + self.encode_block_data(data, length, c.Addr(lit_tree), c.Addr(dist_tree)) + + def compress_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + if self.level == 0: + self.write_stored_block(data, length, final) + elif self.strategy == 4 or length < 128: + self.write_fixed_block(data, length, final) + else: + self.write_dynamic_block(data, length, final) + + def write_stored_block(self, data: UINT8PTR, length: t.CSizeT, final: t.CInt): + self.writer.stored_block(data, length, final) + + def compress(self, data: UINT8PTR, length: t.CSizeT, + out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if self.is_finished: return -2 + if not data or length == 0: + c.Set(c.Deref(out), None) + c.Set(c.Deref(out_len), 0) + return 0 + self.adler = zchecksum.zchecksum_adler32(data, length, self.adler) + memset(self.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(self.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + self.compress_block(data, length, 0) + c.Set(c.Deref(out_len), self.writer.total()) + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(self.pool, c.Deref(out_len)))) + if not c.Deref(out): return -4 + memcpy(c.Deref(out), self.writer.buf, c.Deref(out_len)) + return 0 + + def flush(self, mode: t.CInt, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if mode == 4: + if not self.is_finished: + if self.level == 0: + self.writer.write_bits(1, 1) + self.writer.write_bits(0, 2) + self.writer.align() + else: + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + self.writer.write_bits(1, 1) + self.writer.write_bits(1, 2) + lit_tree.encode_symbol(256, c.Addr(self.writer)) + self.is_finished = 1 + if self.wbits >= 0: + self.writer.align() + adler: t.CUInt32T = self.adler + self.writer.buf[self.writer.byte_pos + 0] = (adler >> 24) & 0xFF + self.writer.buf[self.writer.byte_pos + 1] = (adler >> 16) & 0xFF + self.writer.buf[self.writer.byte_pos + 2] = (adler >> 8) & 0xFF + self.writer.buf[self.writer.byte_pos + 3] = (adler >> 0) & 0xFF + self.writer.byte_pos += 4 + elif mode == 2 or mode == 3: + self.writer.write_bits(0, 1) + self.writer.write_bits(0, 2) + self.writer.align() + c.Set(c.Deref(out_len), self.writer.total()) + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(self.pool, c.Deref(out_len)))) + if not c.Deref(out): return -4 + memcpy(c.Deref(out), self.writer.buf, c.Deref(out_len)) + return 0 + + def set_dictionary(self, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: + if not dict: return -2 + use: t.CSizeT = length + if use > zdef.ZDEFLATE_WINDOW_SIZE: + dict += length - zdef.ZDEFLATE_WINDOW_SIZE + use = zdef.ZDEFLATE_WINDOW_SIZE + memcpy(self.window, dict, use) + self.window_size = t.CInt(use) + self.window_pos = t.CInt(use) + self.adler = zchecksum.zchecksum_adler32(dict, length, self.adler) + return 0 + + def copy(self) -> zdeflate_stream | t.CPtr: + z: zdeflate_stream | t.CPtr = zdef.zdef_alloc(self.pool, zdeflate_stream.__sizeof__()) + if not z: return None + memcpy(z, self, zdeflate_stream.__sizeof__()) + z.writer.buf = BYTEPTR(zdef.zdef_alloc(self.pool, self.writer.cap)) + if not z.writer.buf: + zdef.zdef_free(self.pool, z) + return None + memcpy(z.writer.buf, self.writer.buf, self.writer.cap) + z.window = BYTEPTR(zdef.zdef_alloc(self.pool, zdef.ZDEFLATE_WINDOW_SIZE)) + z.hash_head = INTPTR(zdef.zdef_alloc(self.pool, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__())) + z.hash_prev = INTPTR(zdef.zdef_alloc(self.pool, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__())) + if not z.window or not z.hash_head or not z.hash_prev: + if z.writer.buf: zdef.zdef_free(self.pool, z.writer.buf) + zdef.zdef_free(self.pool, z) + return None + memcpy(z.window, self.window, zdef.ZDEFLATE_WINDOW_SIZE) + memcpy(z.hash_head, self.hash_head, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memcpy(z.hash_prev, self.hash_prev, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + return z + + def destroy(self): + if self.writer.buf: zdef.zdef_free(self.pool, self.writer.buf) + if self.window: zdef.zdef_free(self.pool, self.window) + if self.hash_head: zdef.zdef_free(self.pool, self.hash_head) + if self.hash_prev: zdef.zdef_free(self.pool, self.hash_prev) + zdef.zdef_free(self.pool, self) + + + + +def zdeflate_count_freqs(lit_freqs: INTPTR, dist_freqs: INTPTR, + s: zdeflate_stream | t.CPtr, data: UINT8PTR, length: t.CSizeT): + memset(lit_freqs, 0, 288 * int.__sizeof__()) + memset(dist_freqs, 0, 32 * int.__sizeof__()) + pos: t.CSizeT = 0 + while pos < length: + best_dist: t.CInt = 0 + match_len: t.CInt = 0 + if s.level > 0 and pos + 2 < length: + match_len = zdeflate_stream.find_match(s, data, length, pos, c.Addr(best_dist)) + if match_len >= zdef.ZDEFLATE_MIN_MATCH: + len_sym: t.CInt = zdeflate_len_to_symbol(match_len) + lit_freqs[len_sym] += 1 + dist_freqs[zdeflate_dist_to_symbol(best_dist)] += 1 + for i in range(match_len): + zdeflate_stream.update_hash(s, data, pos + i) + pos += match_len + else: + lit_freqs[data[pos]] += 1 + zdeflate_stream.update_hash(s, data, pos) + pos += 1 + lit_freqs[256] = 1 + + + +def zdeflate_hash(p: BYTE | t.CPtr) -> t.CUnsignedInt: + return (t.CUnsignedInt(p[0]) ^ (t.CUnsignedInt(p[1]) << 5) ^ (t.CUnsignedInt(p[2]) << 10)) & zdef.ZDEFLATE_HASH_MASK + + +def zdeflate_len_to_symbol(length: t.CInt) -> t.CInt: + for i in range(29): + if length <= zdef.zdeflate_len_base[i] + (1 << zdef.zdeflate_len_extra_bits[i]) - 1: + return 257 + i + return 285 + +def zdeflate_dist_to_symbol(dist: t.CInt) -> t.CInt: + for i in range(30): + if dist <= zdef.zdeflate_dist_base[i] + (1 << zdef.zdeflate_dist_extra_bits[i]) - 1: + return i + return 29 + +def zdeflate_count_cl_freqs(all_lengths: INTPTR, total: t.CInt, cl_freqs: INTPTR): + memset(cl_freqs, 0, 19 * int.__sizeof__()) + i: t.CInt = 0 + while i < total: + if all_lengths[i] == 0: + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == 0: run += 1 + while run > 0: + if run >= 11: + count: t.CInt = run + if count > 138: count = 138 + cl_freqs[18] += 1 + run -= count + elif run >= 3: + count: t.CInt = run + if count > 10: count = 10 + cl_freqs[17] += 1 + run -= count + else: + cl_freqs[0] += 1 + run -= 1 + while i < total and all_lengths[i] == 0: i += 1 + else: + cl_freqs[all_lengths[i]] += 1 + val: t.CInt = all_lengths[i] + i += 1 + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == val: run += 1 + while run >= 3: + count: t.CInt = run + if count > 6: count = 6 + cl_freqs[16] += 1 + run -= count + i += run + +def zdeflate_write_cl_encoded(all_lengths: INTPTR, total: t.CInt, + w: zdef.zbit_writer | t.CPtr, cl_tree: zhuff.zhuff_tree | t.CPtr): + i: t.CInt = 0 + while i < total: + if all_lengths[i] == 0: + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == 0: run += 1 + while run > 0: + if run >= 11: + count: t.CInt = run + if count > 138: count = 138 + cl_tree.encode_symbol(18, w) + w.write_bits(count - 11, 7) + run -= count + elif run >= 3: + count: t.CInt = run + if count > 10: count = 10 + cl_tree.encode_symbol(17, w) + w.write_bits(count - 3, 3) + run -= count + else: + cl_tree.encode_symbol(0, w) + run -= 1 + while i < total and all_lengths[i] == 0: i += 1 + else: + cl_tree.encode_symbol(all_lengths[i], w) + val: t.CInt = all_lengths[i] + i += 1 + run: t.CInt = 0 + while i + run < total and all_lengths[i + run] == val: run += 1 + while run >= 3: + count: t.CInt = run + if count > 6: count = 6 + cl_tree.encode_symbol(16, w) + w.write_bits(count - 3, 2) + run -= count + i += run + + + +def zdeflate_create(pool: memhub.MemManager | t.CPtr, level: t.CInt, wbits: t.CInt, mem_level: t.CInt, strategy: t.CInt) -> zdeflate_stream | t.CPtr: + n: t.CSizeT = zdeflate_stream.__sizeof__() + s: zdeflate_stream | t.CPtr = zdef.zdef_alloc(pool, n) + if not s: return None + memset(s, 0, zdeflate_stream.__sizeof__()) + s.pool = pool + s.writer = zdef.zbit_writer(pool) + s.window = BYTEPTR(zdef.zdef_alloc(pool, zdef.ZDEFLATE_WINDOW_SIZE)) + s.hash_head = INTPTR(zdef.zdef_alloc(pool, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__())) + s.hash_prev = INTPTR(zdef.zdef_alloc(pool, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__())) + if not s.window or not s.hash_head or not s.hash_prev: + s.destroy() + return None + memset(s.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(s.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + s.level = level + s.wbits = wbits + s.strategy = strategy + s.is_finished = 0 + s.adler = 1 + if wbits > 0: + s.writer.write_zlib_header(wbits, level) + elif wbits == 0: + s.writer.write_zlib_header(15, level) + return s + + +def zdeflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, + level: t.CInt, wbits: t.CInt, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: + s: zdeflate_stream | t.CPtr = zdeflate_create(pool, level, wbits, 8, 0) + if not s: return None + + memset(s.hash_head, -1, zdef.ZDEFLATE_HASH_SIZE * int.__sizeof__()) + memset(s.hash_prev, -1, zdef.ZDEFLATE_WINDOW_SIZE * int.__sizeof__()) + + s.adler = zchecksum.zchecksum_adler32(data, length, 1) + + if length == 0: + s.writer.write_bits(1, 1) + s.writer.write_bits(1, 2) + lit_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + lit_tree.encode_symbol(256, c.Addr(s.writer)) + else: + s.compress_block(data, length, 1) + + s.is_finished = 1 + + if wbits >= 0: + s.writer.align() + adler: t.CUInt32T = s.adler + s.writer.buf[s.writer.byte_pos + 0] = (adler >> 24) & 0xFF + s.writer.buf[s.writer.byte_pos + 1] = (adler >> 16) & 0xFF + s.writer.buf[s.writer.byte_pos + 2] = (adler >> 8) & 0xFF + s.writer.buf[s.writer.byte_pos + 3] = (adler >> 0) & 0xFF + s.writer.byte_pos += 4 + c.Set(c.Deref(out_len), s.writer.total()) + result: UINT8PTR = UINT8PTR(zdef.zdef_alloc(pool, c.Deref(out_len))) + if result: memcpy(result, s.writer.buf, c.Deref(out_len)) + s.destroy() + return result diff --git a/includes/zlib/zhuff.py b/includes/zlib/zhuff.py new file mode 100644 index 0000000..2b6648c --- /dev/null +++ b/includes/zlib/zhuff.py @@ -0,0 +1,356 @@ +#include +from stdint import * +import zlib.zdef as zdef +import memhub +import t, c + + +ZHUFF_MAX_CODES: t.CDefine = 288 +ZHUFF_MAX_BITS: t.CDefine = 15 + +class zhuff_code: + code: UINT + bits: t.CInt + +@t.Object +class zhuff_tree: + pool: memhub.MemManager | t.CPtr + codes: t.CArray[zhuff_code, ZHUFF_MAX_CODES] + count: t.CInt + max_bits: t.CInt + + def __init__(self): + self.pool = None + + def build_codes(self, freqs: INTPTR, count: t.CInt, max_bits: t.CInt): + lengths: t.CArray[t.CInt, ZHUFF_MAX_CODES] + bl_count: t.CArray[t.CInt, ZHUFF_MAX_BITS + 1] + next_code: t.CArray[UINT, ZHUFF_MAX_BITS + 1] + self.count = count + self.max_bits = max_bits + self.build_code_lengths(lengths, freqs, count, max_bits) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, max_bits + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + for i in range(count): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def build_fixed_lit_tree(self): + lengths: t.CArray[t.CInt, 288] + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + self.get_fixed_lit_lengths(lengths) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(288): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, 9 + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + self.count = 288 + self.max_bits = 9 + for i in range(288): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def build_fixed_dist_tree(self): + lengths: t.CArray[t.CInt, 32] + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + self.get_fixed_dist_lengths(lengths) + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(32): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + code: t.CUnsignedInt = 0 + next_code[0] = 0 + for bits in range(1, 5 + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + self.count = 32 + self.max_bits = 5 + for i in range(32): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + def encode_symbol(self, symbol: int, writer: zdef.zbit_writer | t.CPtr): + if symbol < 0 or symbol >= self.count: + return + if self.codes[symbol].bits == 0: + return + writer.write_bits_rev(self.codes[symbol].code, self.codes[symbol].bits) + + + def build_code_lengths(self, lengths: t.CInt | t.CPtr, freqs: t.CInt | t.CPtr, count: t.CInt, max_bits: t.CInt): + bl_count: t.CArray[t.CInt, ZHUFF_MAX_BITS + 2] + sort_count: t.CInt = 0 + memset(bl_count, 0, bl_count.__sizeof__()) + memset(lengths, 0, int.__sizeof__() * count) + for i in range(count): + if freqs[i] > 0: + sort_count += 1 + if sort_count == 0: return + if sort_count == 1: + for i in range(count): + if freqs[i] > 0: + lengths[i] = 1 + return + items: t.CInt | t.CPtr = INTPTR(zdef.zdef_alloc(self.pool, sort_count * int.__sizeof__())) + item_freqs: t.CInt | t.CPtr = INTPTR(zdef.zdef_alloc(self.pool, sort_count * int.__sizeof__())) + idx: t.CInt = 0 + for i in range(count): + if freqs[i] > 0: + items[idx] = i + item_freqs[idx] = freqs[i] + idx += 1 + for i in range(sort_count): + key_item: t.CInt = items[i] + key_freq: t.CInt = item_freqs[i] + j: t.CInt = i - 1 + while j >= 0 and item_freqs[j] > key_freq: + items[j + 1] = items[j] + item_freqs[j + 1] = item_freqs[j] + j -= 1 + items[j + 1] = key_item + item_freqs[j + 1] = key_freq + parent: INTPTR = INTPTR(zdef.zdef_alloc(self.pool, (sort_count * 2) * int.__sizeof__())) + for i in range(sort_count * 2): + parent[i] = -1 + heap: INTPTR = INTPTR(zdef.zdef_alloc(self.pool, (sort_count + 1) * int.__sizeof__())) + heap_size: t.CInt = 0 + for i in range(sort_count): + heap[heap_size] = i + heap_size += 1 + pos: t.CInt = heap_size - 1 + while pos > 0: + par: t.CInt = (pos - 1) / 2 + if item_freqs[heap[par]] > item_freqs[heap[pos]]: + tmp: t.CInt = heap[par] + heap[par] = heap[pos] + heap[pos] = tmp + pos = par + else: break + combined_freq: INTPTR = INTPTR(zdef.zdef_alloc(self.pool, (sort_count * 2) * int.__sizeof__())) + for i in range(sort_count): + combined_freq[i] = item_freqs[i] + internal: t.CInt = sort_count + while heap_size > 1: + a: t.CInt = heap[0] + heap[0] = heap[heap_size - 1] + heap_size -= 1 + pos: t.CInt = 0 + while True: + left: t.CInt = 2 * pos + 1 + right: t.CInt = 2 * pos + 2 + smallest: t.CInt = pos + if left < heap_size and combined_freq[heap[left]] < combined_freq[heap[smallest]]: + smallest = left + if right < heap_size and combined_freq[heap[right]] < combined_freq[heap[smallest]]: + smallest = right + if smallest != pos: + tmp: t.CInt = heap[pos] + heap[pos] = heap[smallest] + heap[smallest] = tmp + pos = smallest + else: break + b: t.CInt = heap[0] + heap[0] = heap[heap_size - 1] + heap_size -= 1 + pos = 0 + while True: + left: t.CInt = 2 * pos + 1 + right: t.CInt = 2 * pos + 2 + smallest: t.CInt = pos + if left < heap_size and combined_freq[heap[left]] < combined_freq[heap[smallest]]: + smallest = left + if right < heap_size and combined_freq[heap[right]] < combined_freq[heap[smallest]]: + smallest = right + if smallest != pos: + tmp: t.CInt = heap[pos] + heap[pos] = heap[smallest] + heap[smallest] = tmp + pos = smallest + else: break + if internal >= sort_count * 2 - 1: break + combined_freq[internal] = combined_freq[a] + combined_freq[b] + parent[a] = internal + parent[b] = internal + heap[heap_size] = internal + heap_size += 1 + pos = heap_size - 1 + while pos > 0: + par: t.CInt = (pos - 1) / 2 + if combined_freq[heap[par]] > combined_freq[heap[pos]]: + tmp: t.CInt = heap[par] + heap[par] = heap[pos] + heap[pos] = tmp + pos = par + else: break + internal += 1 + for i in range(sort_count): + node: t.CInt = i + length: t.CInt = 0 + while parent[node] >= 0: + length += 1 + node = parent[node] + lengths[items[i]] = length + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + overflow: t.CInt = 0 + for i in range(count): + if lengths[i] > max_bits: + overflow += 1 + bl_count[lengths[i]] -= 1 + lengths[i] = max_bits + bl_count[max_bits] += 1 + while overflow > 0: + bits: t.CInt = max_bits - 1 + while bits > 0 and bl_count[bits] == 0: + bits -= 1 + if bits == 0: break + bl_count[bits] -= 1 + bl_count[bits + 1] += 2 + bl_count[max_bits] -= 1 + overflow -= 2 + sorted_items: INTPTR = INTPTR(zdef.zdef_alloc(self.pool, sort_count * int.__sizeof__())) + si: t.CInt = 0 + for i in range(count): + if freqs[i] > 0: + sorted_items[si] = i + si += 1 + for i in range(si - 1): + for j in range(i + 1, si): + if lengths[sorted_items[i]] < lengths[sorted_items[j]]: + tmp: t.CInt = sorted_items[i] + sorted_items[i] = sorted_items[j] + sorted_items[j] = tmp + sidx: t.CInt = 0 + for bits in range(max_bits, 1, -1): + n: t.CInt = bl_count[bits] + while n > 0 and sidx < si: + lengths[sorted_items[sidx]] = bits + sidx += 1 + n -= 1 + zdef.zdef_free(self.pool, sorted_items) + zdef.zdef_free(self.pool, heap) + zdef.zdef_free(self.pool, parent) + zdef.zdef_free(self.pool, combined_freq) + zdef.zdef_free(self.pool, item_freqs) + zdef.zdef_free(self.pool, items) + + def get_fixed_lit_lengths(self, lengths: INTPTR): + for i in range( 0, 143 + 1): lengths[i] = 8 + for i in range(143 + 1, 255 + 1): lengths[i] = 9 + for i in range(255 + 1, 279 + 1): lengths[i] = 7 + for i in range(279 + 1, 287 + 1): lengths[i] = 8 + + def get_fixed_dist_lengths(self, lengths: INTPTR): + for i in range(32): + lengths[i] = 5 + + def build_tree_from_lengths(self, lengths: INTPTR, count: t.CInt, max_bits: t.CInt): + bl_count: t.CArray[t.CInt, 16] + next_code: t.CArray[t.CUnsignedInt, 16] + + memset(bl_count, 0, bl_count.__sizeof__()) + for i in range(count): + if lengths[i] > 0: + bl_count[lengths[i]] += 1 + + code: UINT = 0 + next_code[0] = 0 + for bits in range(1, max_bits + 1): + code = (code + bl_count[bits - 1]) << 1 + next_code[bits] = code + + self.count = count + self.max_bits = max_bits + for i in range(count): + self.codes[i].bits = lengths[i] + if lengths[i] > 0: + self.codes[i].code = next_code[lengths[i]] + next_code[lengths[i]] += 1 + else: + self.codes[i].code = 0 + + +class zhuff_decode_node: + children: t.CArray[t.CInt, 2] + symbol: t.CInt + + +@t.Object +class zhuff_decode_tree: + pool: memhub.MemManager | t.CPtr + nodes: t.CArray[zhuff_decode_node, 2 * ZHUFF_MAX_CODES] + node_count: t.CInt + root: t.CInt + + def __init__(self): + self.pool = None + + def build_decode_tree(self, ht: zhuff_tree | t.CPtr): + self.node_count = 1 + self.root = 0 + self.nodes[0].children[0] = -1 + self.nodes[0].children[1] = -1 + self.nodes[0].symbol = -1 + for i in range(ht.count): + if ht.codes[i].bits == 0: continue + node: int = 0 + for bit in range(ht.codes[i].bits - 1, -1, -1): + dir: t.CInt = (ht.codes[i].code >> bit) & 1 + if bit > 0: + # Internal node: traverse or create + if self.nodes[node].children[dir] == -1: + new_node: int = self.node_count + self.node_count += 1 + self.nodes[new_node].children[0] = -1 + self.nodes[new_node].children[1] = -1 + self.nodes[new_node].symbol = -1 + self.nodes[node].children[dir] = new_node + node = self.nodes[node].children[dir] + else: + # Leaf: bit 0 - set symbol on the child + if self.nodes[node].children[dir] == -1: + leaf: int = self.node_count + self.node_count += 1 + self.nodes[leaf].children[0] = -1 + self.nodes[leaf].children[1] = -1 + self.nodes[leaf].symbol = i + self.nodes[node].children[dir] = leaf + else: + self.nodes[self.nodes[node].children[dir]].symbol = i + + def decode_symbol(self, reader: zdef.zbit_reader | t.CPtr) -> t.CInt: + node: int = self.root + while self.nodes[node].symbol == -1: + bit: t.CUnsignedInt + if reader.read_bits(1, c.Addr(bit)) != 0: return -1 + dir: t.CInt = t.CInt(bit) + if self.nodes[node].children[dir] == -1: return -1 + node = self.nodes[node].children[dir] + return self.nodes[node].symbol diff --git a/includes/zlib/zinflate.py b/includes/zlib/zinflate.py new file mode 100644 index 0000000..8adf6a7 --- /dev/null +++ b/includes/zlib/zinflate.py @@ -0,0 +1,456 @@ +from stdint import * +import zlib.zchecksum as zchecksum +import zlib.zdef as zdef +import zlib.zhuff as zhuff +import string +import memhub +import t, c + + +@t.Object +class zinflate_stream: + pool: memhub.MemManager | t.CPtr + reader: zdef.zbit_reader + + window: BYTEPTR + window_pos: t.CInt + window_size: t.CInt + + wbits: t.CInt + is_finished: t.CInt + is_initialized: t.CInt + + header_parsed: t.CInt + is_gzip: t.CInt + + adler: t.CUInt32T + crc: t.CUInt32T + + output: BYTEPTR + output_len: t.CSizeT + output_cap: t.CSizeT + + input_data: t.CConst | BYTEPTR + input_len: t.CSizeT + input_pos: t.CSizeT + + max_length: t.CSizeT + + def __init__(self): + pass + + def output_byte(self, byte: BYTE): + if self.max_length > 0 and self.output_len >= self.max_length: return + + if self.output_len >= self.output_cap: + new_cap: t.CSizeT = self.output_cap * 2 + if new_cap < 256: new_cap = 256 + new_buf: BYTEPTR = BYTEPTR(zdef.zdef_alloc(self.pool, new_cap)) + if not new_buf: return + memcpy(new_buf, self.output, self.output_len) + zdef.zdef_free(self.pool, self.output) + self.output = new_buf + self.output_cap = new_cap + self.output[self.output_len] = byte + self.output_len += 1 + self.window[self.window_pos & (zdef.ZDEFLATE_WINDOW_SIZE - 1)] = byte + self.window_pos += 1 + + + def parse_header(self) -> t.CInt: + if self.header_parsed: return 0 + if self.wbits < 0: + self.header_parsed = 1 + self.is_gzip = 0 + return 0 + if self.input_pos + 2 > self.input_len: return -3 + b0: BYTE = self.input_data[self.input_pos] + b1: BYTE = self.input_data[self.input_pos + 1] + if b0 == 0x1F and b1 == 0x8B: + self.is_gzip = 1 + if self.input_pos + 10 > self.input_len: return -3 + self.input_pos += 10 + flg: BYTE = self.input_data[self.input_pos - 6] + if flg & 0x04: + if self.input_pos + 2 > self.input_len: return -3 + xlen: UINT = self.input_data[self.input_pos] | (self.input_data[self.input_pos + 1] << 8) + self.input_pos += 2 + xlen + if flg & 0x08: + while self.input_pos < self.input_len and self.input_data[self.input_pos] != 0: self.input_pos += 1 + self.input_pos += 1 + if flg & 0x10: + while self.input_pos < self.input_len and self.input_data[self.input_pos] != 0: self.input_pos += 1 + self.input_pos += 1 + if flg & 0x02: + self.input_pos += 2 + if self.input_pos > self.input_len: return -3 + else: + self.is_gzip = 0 + if (b0 * 256 + b1) % 31 != 0: return -3 + cm: t.CInt = b0 & 0x0F + if cm != 8: return -3 + self.input_pos += 2 + self.header_parsed = 1 + return 0 + + + def read_bits_from_input(self, nbits: t.CInt, out: UINTPTR) -> t.CInt: + val: UINT = 0 + for i in range(nbits): + if self.input_pos >= self.input_len: return -3 + if self.input_data[self.input_pos] & (UINT(1) << (self.reader.bit_pos)): + val |= (UINT(1) << i) + self.reader.bit_pos += 1 + if self.reader.bit_pos == 8: + self.reader.bit_pos = 0 + self.input_pos += 1 + c.Set(c.Deref(out), val) + return 0 + + + def read_huffman(self, dt: zhuff.zhuff_decode_tree | t.CPtr) -> t.CInt: + node: int = dt.root + while dt.nodes[node].symbol == -1: + bit: UINT + if self.read_bits_from_input(1, c.Addr(bit)) != 0: return -1 + dir: int = t.CInt(bit) + if dt.nodes[node].children[dir] == -1: return -3 + node = dt.nodes[node].children[dir] + return dt.nodes[node].symbol + + def inflate_block_stored(self) -> t.CInt: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 4 > self.input_len: return -3 + length: UINT = self.input_data[self.input_pos] | (self.input_data[self.input_pos + 1] << 8) + nlen: UINT = self.input_data[self.input_pos + 2] | (self.input_data[self.input_pos + 3] << 8) + self.input_pos += 4 + if (length ^ nlen) != 0xFFFF: return -3 + if self.input_pos + length > self.input_len: return -3 + i: t.CUnsignedInt + for i in range(length): + self.output_byte(self.input_data[self.input_pos]) + self.input_pos += 1 + return 0 + + + def inflate_block_fixed(self) -> t.CInt: + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + lit_tree.build_fixed_lit_tree() + dist_tree.build_fixed_dist_tree() + + lit_dt: zhuff.zhuff_decode_tree | t.CPtr = zdef.zdef_alloc(self.pool, zhuff.zhuff_decode_tree.__sizeof__()) + dist_dt: zhuff.zhuff_decode_tree | t.CPtr = zdef.zdef_alloc(self.pool, zhuff.zhuff_decode_tree.__sizeof__()) + if not lit_dt or not dist_dt: + if lit_dt: zdef.zdef_free(self.pool, lit_dt) + if dist_dt: zdef.zdef_free(self.pool, dist_dt) + return -4 + memset(lit_dt, 0, zhuff.zhuff_decode_tree.__sizeof__()) + memset(dist_dt, 0, zhuff.zhuff_decode_tree.__sizeof__()) + lit_dt.build_decode_tree(c.Addr(lit_tree)) + dist_dt.build_decode_tree(c.Addr(dist_tree)) + + result: t.CInt = 0 + while True: + if self.max_length > 0 and self.output_len >= self.max_length: break + + sym: t.CInt = self.read_huffman(lit_dt) + if sym < 0: + result = -3 + break + if sym == 256: break + if sym < 256: + self.output_byte(BYTE(sym)) + else: + len_idx: t.CInt = sym - 257 + if len_idx < 0 or len_idx >= 29: + result = -3 + break + length: t.CInt = zdef.zdeflate_len_base[len_idx] + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_idx] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: + result = -3 + break + length += extra_val + + dist_sym: t.CInt = self.read_huffman(dist_dt) + if dist_sym < 0 or dist_sym >= 30: + result = -3 + break + dist: t.CInt = zdef.zdeflate_dist_base[dist_sym] + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: + result = -3 + break + dist += extra_val + + for i in range(length): + if self.max_length > 0 and self.output_len >= self.max_length: break + src_pos: t.CInt = (self.window_pos - dist) & (zdef.ZDEFLATE_WINDOW_SIZE - 1) + byte: BYTE = self.window[src_pos] + self.output_byte(byte) + + zdef.zdef_free(self.pool, lit_dt) + zdef.zdef_free(self.pool, dist_dt) + return result + + + def inflate_block_dynamic(self) -> t.CInt: + hlit_val: UINT + hdist_val: UINT + hclen_val: UINT + if self.read_bits_from_input(5, c.Addr(hlit_val)) != 0: return -3 + if self.read_bits_from_input(5, c.Addr(hdist_val)) != 0: return -3 + if self.read_bits_from_input(4, c.Addr(hclen_val)) != 0: return -3 + + hlit: t.CInt = t.CInt(hlit_val) + 257 + hdist: t.CInt = t.CInt(hdist_val) + 1 + hclen: t.CInt = t.CInt(hclen_val) + 4 + + cl_lengths: t.CArray[t.CInt, 19] + memset(cl_lengths, 0, cl_lengths.__sizeof__()) + for i in range(hclen): + len_val: UINT + if self.read_bits_from_input(3, c.Addr(len_val)) != 0: return -3 + cl_lengths[zdef.zdeflate_codelen_order[i]] = t.CInt(len_val) + + cl_tree = zhuff.zhuff_tree() + cl_tree.build_tree_from_lengths(cl_lengths, 19, 7) + + cl_dt = zhuff.zhuff_decode_tree() + cl_dt.build_decode_tree(c.Addr(cl_tree)) + + total: t.CInt = hlit + hdist + all_lengths: t.CInt | t.CPtr = zdef.zdef_alloc(self.pool, total * int.__sizeof__()) + if not all_lengths: return -4 + + idx: t.CInt = 0 + while idx < total: + sym: t.CInt = self.read_huffman(c.Addr(cl_dt)) + if sym < 0 or sym > 18: + zdef.zdef_free(self.pool, all_lengths) + return -3 + if sym < 16: + all_lengths[idx] = sym + idx += 1 + elif sym == 16: + rep: UINT + if self.read_bits_from_input(2, c.Addr(rep)) != 0: + zdef.zdef_free(self.pool, all_lengths) + return -3 + rep += 3 + if idx == 0 or idx + rep > total: + zdef.zdef_free(self.pool, all_lengths) + return -3 + prev: t.CInt = all_lengths[idx - 1] + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = prev + idx += 1 + elif sym == 17: + rep: t.CUnsignedInt + if self.read_bits_from_input(3, c.Addr(rep)) != 0: + zdef.zdef_free(self.pool, all_lengths) + return -3 + rep += 3 + if idx + rep > total: + zdef.zdef_free(self.pool, all_lengths) + return -3 + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = 0 + idx += 1 + else: + rep: t.CUnsignedInt + if self.read_bits_from_input(7, c.Addr(rep)) != 0: + zdef.zdef_free(self.pool, all_lengths) + return -3 + rep += 11 + if idx + rep > total: + zdef.zdef_free(self.pool, all_lengths) + return -3 + i: t.CUnsignedInt + for i in range(rep): + all_lengths[idx] = 0 + idx += 1 + + lit_tree = zhuff.zhuff_tree() + dist_tree = zhuff.zhuff_tree() + + lit_lengths: t.CArray[t.CInt, 288] + memset(lit_lengths, 0, lit_lengths.__sizeof__()) + for i in range(hlit): + lit_lengths[i] = all_lengths[i] + lit_tree.build_tree_from_lengths(lit_lengths, hlit, zdef.ZDEFLATE_MAX_BITS) + + dist_lengths: t.CArray[t.CInt, 32] + memset(dist_lengths, 0, dist_lengths.__sizeof__()) + for i in range(hdist): + dist_lengths[i] = all_lengths[hlit + i] + dist_tree.build_tree_from_lengths(dist_lengths, hdist, zdef.ZDEFLATE_MAX_BITS) + + zdef.zdef_free(self.pool, all_lengths) + + lit_dt = zhuff.zhuff_decode_tree() + dist_dt = zhuff.zhuff_decode_tree() + lit_dt.build_decode_tree(c.Addr(lit_tree)) + dist_dt.build_decode_tree(c.Addr(dist_tree)) + + while True: + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + sym: t.CInt = self.read_huffman(c.Addr(lit_dt)) + if sym < 0: return -3 + if sym == 256: return 0 + if sym < 256: + self.output_byte(BYTE(sym)) + else: + len_idx: t.CInt = sym - 257 + if len_idx < 0 or len_idx >= 29: return -3 + length: t.CInt = zdef.zdeflate_len_base[len_idx] + extra: t.CInt = zdef.zdeflate_len_extra_bits[len_idx] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: return -3 + length += extra_val + dist_sym: t.CInt = self.read_huffman(c.Addr(dist_dt)) + if dist_sym < 0 or dist_sym >= 30: return -3 + dist: t.CInt = zdef.zdeflate_dist_base[dist_sym] + extra = zdef.zdeflate_dist_extra_bits[dist_sym] + if extra > 0: + extra_val: UINT + if self.read_bits_from_input(extra, c.Addr(extra_val)) != 0: return -3 + dist += extra_val + for i in range(length): + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + src_pos: t.CInt = (self.window_pos - dist) & (zdef.ZDEFLATE_WINDOW_SIZE - 1) + byte: BYTE = self.window[src_pos] + self.output_byte(byte) + + def inflate_stream(self) -> t.CInt: + err: t.CInt = self.parse_header() + if err != 0: return err + while not self.is_finished: + bfinal_val: UINT + btype_val: UINT + if self.read_bits_from_input(1, c.Addr(bfinal_val)) != 0: return -3 + if self.read_bits_from_input(2, c.Addr(btype_val)) != 0: return -3 + bfinal: t.CInt = t.CInt(bfinal_val) + btype: t.CInt = t.CInt(btype_val) + if btype == 0: + err = self.inflate_block_stored() + elif btype == 1: + err = self.inflate_block_fixed() + elif btype == 2: + err = self.inflate_block_dynamic() + else: + return -3 + if err != 0: return err + if bfinal: self.is_finished = 1 + if self.max_length > 0 and self.output_len >= self.max_length: return 0 + if self.is_gzip: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 8 <= self.input_len: + self.crc = ((t.CUInt32T(self.input_data[self.input_pos + 0]) << 0) | + (t.CUInt32T(self.input_data[self.input_pos + 1]) << 8) | + (t.CUInt32T(self.input_data[self.input_pos + 2]) << 16) | + (t.CUInt32T(self.input_data[self.input_pos + 3]) << 24)) + self.input_pos += 8 + elif self.wbits >= 0: + if self.reader.bit_pos != 0: + self.reader.bit_pos = 0 + self.input_pos += 1 + if self.input_pos + 4 <= self.input_len: + self.adler = ((t.CUInt32T(self.input_data[self.input_pos + 0]) << 24) | + (t.CUInt32T(self.input_data[self.input_pos + 1]) << 16) | + (t.CUInt32T(self.input_data[self.input_pos + 2]) << 8) | + (t.CUInt32T(self.input_data[self.input_pos + 3]) << 0)) + self.input_pos += 4 + return 0 + + def decompress(self, data: UINT8PTR, length: t.CSizeT, + max_length: t.CSizeT, out: t.CUInt8T | t.CPtr[t.CPtr], out_len: t.CSizeT | t.CPtr) -> t.CInt: + if not self.is_initialized: return -2 + self.input_data = data + self.input_len = length + self.input_pos = 0 + self.reader.bit_pos = 0 + self.output_len = 0 + self.max_length = max_length + err: t.CInt = self.inflate_stream() + if err != 0: return err + c.Set(c.Deref(out_len), self.output_len) + if self.output_len == 0: + c.Set(c.Deref(out), UINT8PTR(zdef.zdef_alloc(self.pool, 1))) + else: + dst: UINT8PTR = UINT8PTR(zdef.zdef_alloc(self.pool, self.output_len)) + c.Set(c.Deref(out), dst) + if dst: + memcpy(dst, self.output, self.output_len) + return 0 + + def set_dictionary(self, dict: UINT8PTR, length: t.CSizeT) -> t.CInt: + if not dict: return -2 + use: t.CSizeT = length + if use > zdef.ZDEFLATE_WINDOW_SIZE: + dict += length - zdef.ZDEFLATE_WINDOW_SIZE + use = zdef.ZDEFLATE_WINDOW_SIZE + memcpy(self.window, dict, use) + self.window_pos = t.CInt(use) + return 0 + + def copy(self) -> zinflate_stream | t.CPtr: + z: zinflate_stream | t.CPtr = zdef.zdef_alloc(self.pool, zinflate_stream.__sizeof__()) + if not z: return None + memcpy(z, self, zinflate_stream.__sizeof__()) + z.window = BYTEPTR(zdef.zdef_alloc(self.pool, zdef.ZDEFLATE_WINDOW_SIZE)) + z.output = BYTEPTR(zdef.zdef_alloc(self.pool, self.output_cap)) + if not z.window or not z.output: + zinflate_stream.destroy(z) + return None + memcpy(z.window, self.window, zdef.ZDEFLATE_WINDOW_SIZE) + memcpy(z.output, self.output, self.output_cap) + return z + + def destroy(self): + if not self: return + if self.window: zdef.zdef_free(self.pool, self.window) + if self.output: zdef.zdef_free(self.pool, self.output) + zdef.zdef_free(self.pool, self) + + +def zinflate_one_shot(pool: memhub.MemManager | t.CPtr, data: UINT8PTR, length: t.CSizeT, + wbits: t.CInt, bufsize: t.CSizeT, out_len: t.CSizeT | t.CPtr) -> t.CUInt8T | t.CPtr: + s: zinflate_stream | t.CPtr = zinflate_create(pool, wbits) + if not s: return None + out: t.CUInt8T | t.CPtr = None + err: t.CInt = s.decompress(data, length, 0, c.Addr(out), out_len) + s.destroy() + if err != 0: return None + return out + + +def zinflate_create(pool: memhub.MemManager | t.CPtr, wbits: t.CInt) -> zinflate_stream | t.CPtr: + s: zinflate_stream | t.CPtr = zdef.zdef_alloc(pool, zinflate_stream.__sizeof__()) + if not s: return None + memset(s, 0, zinflate_stream.__sizeof__()) + s.pool = pool + s.wbits = wbits + s.is_initialized = 1 + s.adler = 1 + s.window = BYTEPTR(zdef.zdef_alloc(pool, zdef.ZDEFLATE_WINDOW_SIZE)) + s.output = BYTEPTR(zdef.zdef_alloc(pool, 256)) + s.output_cap = 256 + s.output_len = 0 + s.max_length = 0 + if not s.window or not s.output: + s.destroy() + return None + return s diff --git a/lib/Projectrans/Config.py b/lib/Projectrans/Config.py new file mode 100644 index 0000000..26dbaab --- /dev/null +++ b/lib/Projectrans/Config.py @@ -0,0 +1,280 @@ +from __future__ import annotations + +import os +import sys +import json +import glob as glob_module +import argparse +import shutil +import subprocess +from typing import Any + +from lib.core.VLogger import get_logger as _vlog +from lib.Projectrans.Phase1Generator import Phase1Generator +from lib.Projectrans.Phase2Translator import Phase2Translator + + +def save_sha1_map(temp_dir: str, sha1_map: dict[str, str]) -> None: + """将 SHA1 映射表保存到文件""" + map_path: str = os.path.join(temp_dir, '_sha1_map.txt') + with open(map_path, 'w', encoding='utf-8') as f: + for sha1, rel in sorted(sha1_map.items()): + f.write(f"{sha1}:{rel}\n") + + +def Load_sha1_map(temp_dir: str) -> dict[str, str]: + """从文件加载 SHA1 映射表""" + map_path: str = os.path.join(temp_dir, '_sha1_map.txt') + result: dict[str, str] = {} + if os.path.exists(map_path): + with open(map_path, 'r', encoding='utf-8') as f: + for line in f: + line: str = line.strip() + if ':' in line: + sha1: str + rel: str + sha1, rel = line.split(':', 1) + result[sha1] = rel + return result + + +def Load_project_config(project_file: str) -> dict[str, Any]: + """从 project.json 加载配置""" + with open(project_file, 'r', encoding='utf-8') as f: + return json.load(f) + + +def resolve_paths(config: dict[str, Any], project_file: str) -> dict[str, Any]: + """将配置中的相对路径转换为绝对路径""" + project_dir: str = os.path.dirname(os.path.abspath(project_file)) + result: dict[str, Any] = dict(config) + + def resolve(p: str | None) -> str | None: + if p is None: + return None + if os.path.isabs(p): + return p + return os.path.normpath(os.path.join(project_dir, p)) + + if 'source_dir' in result: + result['source_dir'] = resolve(result.get('source_dir')) + elif 'sources' in result: + resolved_sources: list[str] = [] + for src in result['sources']: + if '*' in src or '?' in src: + for matched in glob_module.glob(os.path.join(project_dir, src), recursive=True): + if os.path.isfile(matched): + resolved_sources.append(matched) + else: + resolved_sources.append(resolve(src)) + result['sources'] = [s for s in resolved_sources if os.path.isfile(s)] + + result['temp_dir'] = resolve(config.get('temp_dir')) + result['output_dir'] = resolve(config.get('output_dir')) + + if 'includes' in result: + result['includes'] = [resolve(inc) for inc in result['includes']] + + result['_project_dir'] = project_dir + return result + + +def main() -> None: + parser: argparse.ArgumentParser = argparse.ArgumentParser( + description='TransPyC 工程翻译器 - 基于 project.json 的两阶段翻译', + formatter_class=argparse.RawDescriptionHelpFormatter, + epilog=''' +示例: + # 从 project.json 读取配置并执行（默认查找当前目录） + python ProjectTrans.py + + # 指定 project.json 文件 + python ProjectTrans.py --project ./TestProject/project.json + + # 使用命令行参数覆盖 project.json 配置 + python ProjectTrans.py --src ./src --temp ./decl --output ./out --phase all + + # 仅运行阶段一 + python ProjectTrans.py --phase 1 + + # 仅运行阶段二 + python ProjectTrans.py --phase 2 + + # 清理临时目录 + python ProjectTrans.py --clean + + # 编译并立即执行 + python Projectrans.py --project ./TestProject/project.json --clean --run + + # 删除 includes.binary 预编译缓存并重新编译所有 includes + python Projectrans.py --project ./TestProject/project.json --rebuild-includes +''' + ) + parser.add_argument('--project', help='project.json 路径（默认查找当前目录）') + parser.add_argument('--src', help='源文件目录（覆盖 project.json）') + parser.add_argument('--temp', help='声明接口临时目录（覆盖 project.json）') + parser.add_argument('--output', help='输出目录（覆盖 project.json）') + parser.add_argument('--phase', choices=['1', '2', 'all'], help='阶段: 1=生成声明, 2=翻译+编译, all=全部') + parser.add_argument('--cc', help='LLVM 编译器命令（覆盖 project.json）') + parser.add_argument('--clean', action='store_true', help='清理 output 和 temp 目录') + parser.add_argument('--run', action='store_true', help='编译成功后立即执行生成的可执行文件') + parser.add_argument('--rebuild-includes', action='store_true', help='删除 includes.binary 预编译缓存并重新编译所有 includes') + parser.add_argument('--clear-cache', action='store_true', help='清除 .transpyc_cache 全局缓存（.pyi/.stub.ll/.doc.json）') + + args: argparse.Namespace = parser.parse_args() + + project_file: str | None = args.project + if not project_file: + candidates: list[str] = ['project.json', './project.json'] + for c in candidates: + if os.path.exists(c): + project_file = c + break + + if project_file and os.path.exists(project_file): + _vlog().info(f"读取配置文件: {project_file}") + config: dict[str, Any] = Load_project_config(project_file) + project_dir: str = os.path.dirname(os.path.abspath(project_file)) + _vlog().info(f"项目目录: {project_dir}") + else: + _vlog().error("未找到 project.json，请使用 --project 指定") + sys.exit(1) + + resolved: dict[str, Any] = resolve_paths(config, project_file) + src: str | list[str] | None = args.src or resolved.get('source_dir') or resolved.get('sources') + temp_dir: str | None = args.temp or resolved.get('temp_dir') + output_dir: str | None = args.output or resolved.get('output_dir') + phase: str = args.phase or 'all' + cc_cmd: str = args.cc or resolved.get('compiler', {}).get('cmd', 'llc') + cc_flags: list[str] = resolved.get('compiler', {}).get('flags', ['-filetype=obj']) + slice_level: int = resolved.get('options', {}).get('slice_level', 3) + project_include_dirs: list[str] = resolved.get('includes', []) + target_triple: str | None = resolved.get('target', {}).get('triple') + target_datalayout: str | None = resolved.get('target', {}).get('datalayout') + + if isinstance(src, list): + if len(src) == 1 and os.path.isdir(src[0]): + src = src[0] + else: + src = src[0] if len(src) == 1 else os.path.dirname(os.path.commonpath(src)) if all(os.path.exists(p) for p in src) else src[0] + + _vlog().info(f"源目录: {src}") + _vlog().info(f"临时目录: {temp_dir}") + _vlog().info(f"输出目录: {output_dir}") + _vlog().info(f"阶段: {phase}") + _vlog().info(f"编译器: {cc_cmd} {' '.join(cc_flags)}") + if target_triple: + _vlog().info(f"目标三元组: {target_triple}") + if target_datalayout: + _vlog().info(f"数据布局: {target_datalayout}") + + if args.clean: + if output_dir and os.path.exists(output_dir): + for f in os.listdir(output_dir): + fpath: str = os.path.join(output_dir, f) + try: + if os.path.isfile(fpath) or os.path.islink(fpath): + os.remove(fpath) + elif os.path.isdir(fpath): + shutil.rmtree(fpath, ignore_errors=True) + except: + pass + _vlog().info(f"output: 已清空 {output_dir}") + if temp_dir and os.path.exists(temp_dir): + for f in os.listdir(temp_dir): + fpath: str = os.path.join(temp_dir, f) + try: + if os.path.isfile(fpath) or os.path.islink(fpath): + os.remove(fpath) + elif os.path.isdir(fpath): + shutil.rmtree(fpath, ignore_errors=True) + except: + pass + _vlog().info(f"temp: 已清空 {temp_dir}") + _vlog().info("清理完成") + if (not phase or phase == 'all') and not args.project: + return + + if args.clear_cache: + # 清除 .transpyc_cache 全局缓存 + cache_dir: str = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), '.transpyc_cache') + if os.path.isdir(cache_dir): + shutil.rmtree(cache_dir, ignore_errors=True) + os.makedirs(cache_dir, exist_ok=True) + _vlog().info(f"已清除全局缓存: {cache_dir}") + else: + _vlog().info(f"全局缓存目录不存在: {cache_dir}") + + if phase in ('1', 'all'): + if isinstance(src, list): + _vlog().error("阶段一需要指定源目录，不支持文件列表") + sys.exit(1) + all_include_dirs: list[str] = list(project_include_dirs) if project_include_dirs else [] + for d in Phase2Translator.INCLUDE_DIRS: + if os.path.isdir(d): + d = os.path.abspath(d) + if d not in [os.path.abspath(x) for x in all_include_dirs]: + all_include_dirs.append(d) + gen: Phase1Generator = Phase1Generator(src, temp_dir, include_dirs=all_include_dirs, target_triple=target_triple, target_datalayout=target_datalayout) + gen.run() + save_sha1_map(temp_dir, gen.sha1_map) + + if phase in ('2', 'all'): + if not output_dir: + _vlog().error("阶段二需要指定 --output 目录") + sys.exit(1) + if not os.path.exists(temp_dir): + _vlog().error(f"声明目录不存在，请先运行阶段一: {temp_dir}") + sys.exit(1) + if isinstance(src, list): + _vlog().error("阶段二需要指定源目录，不支持文件列表") + sys.exit(1) + linker_cmd: str | None = resolved.get('linker', {}).get('cmd') + linker_flags: list[str] = resolved.get('linker', {}).get('flags', []) + linker_output: str | None = resolved.get('linker', {}).get('output') + target_triple = resolved.get('target', {}).get('triple') + target_datalayout = resolved.get('target', {}).get('datalayout') + force_recompile_includes: bool = bool(resolved.get('options', {}).get('force_recompile_includes', False)) + if args.rebuild_includes: + force_recompile_includes = True + all_inc_dirs: list[str] = list(project_include_dirs) + list(Phase2Translator.INCLUDE_DIRS) + seen_dirs: set[str] = set() + for inc_dir in all_inc_dirs: + abs_inc: str = os.path.abspath(inc_dir) + if abs_inc in seen_dirs: + continue + seen_dirs.add(abs_inc) + if not os.path.isdir(abs_inc): + continue + inc_parent: str = os.path.dirname(abs_inc) + inc_base: str = os.path.basename(abs_inc) + precompiled_dir: str = os.path.join(inc_parent, inc_base + '.binary') + if os.path.isdir(precompiled_dir): + cleared: int = 0 + for fname in os.listdir(precompiled_dir): + fpath: str = os.path.join(precompiled_dir, fname) + try: + if os.path.isfile(fpath) or os.path.islink(fpath): + os.remove(fpath) + cleared += 1 + except: + pass + _vlog().info(f"includes.binary: 已清空 {precompiled_dir} ({cleared} 个文件)") + trans: Phase2Translator = Phase2Translator(src, temp_dir, output_dir, compile_cmd=cc_cmd, compile_flags=cc_flags, linker_cmd=linker_cmd, linker_flags=linker_flags, linker_output=linker_output, include_dirs=project_include_dirs, target_triple=target_triple, target_datalayout=target_datalayout, force_recompile_includes=force_recompile_includes) + trans.sha1_map = Load_sha1_map(temp_dir) + trans.slice_level = slice_level + trans.run() + + if args.run and linker_output and not linker_output.endswith('.bin'): + exe_path: str = os.path.join(output_dir, linker_output) + if os.path.isfile(exe_path): + _vlog().info(f"\n[运行] 执行: {exe_path}") + try: + result: subprocess.CompletedProcess = subprocess.run([exe_path], cwd=output_dir) + _vlog().info(f"[运行] 退出码: {result.returncode}") + except Exception as e: + _vlog().error(f"[运行] 执行失败: {e}") + sys.exit(1) + else: + _vlog().error(f"[运行] 找不到可执行文件: {exe_path}") + sys.exit(1) diff --git a/lib/Projectrans/DeclarationGenerator.py b/lib/Projectrans/DeclarationGenerator.py new file mode 100644 index 0000000..fb7f1da --- /dev/null +++ b/lib/Projectrans/DeclarationGenerator.py @@ -0,0 +1,1016 @@ +from __future__ import annotations + +import ast +import re +from typing import List + +import llvmlite.ir as ir + +from lib.includes import t +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import IsListAnnotation, AnnotationContainsName + + +class DeclarationGenerator: + """从 .pyi AST 生成纯 LLVM IR 声明（不使用字符串操作）""" + + # 匹配 LLVM 数组类型：[N x elem_type] + _ARRAY_TYPE_RE: re.Pattern = re.compile(r'^\[(\d+)\s+x\s+(.+)\]$') + + @staticmethod + def _decay_array_to_ptr(type_str: str) -> str: + """C 语言中数组参数退化为指针：[N x elem_type] → elem_type*""" + m = DeclarationGenerator._ARRAY_TYPE_RE.match(type_str) + if m: + return f'{m.group(2)}*' + return type_str + + def __init__(self, struct_names: set[str] | None = None, enum_names: set[str] | None = None, module_sha1: str | None = None, target_triple: str | None = None, target_datalayout: str | None = None, struct_sha1_map: dict[str, str] | None = None, exception_names: set[str] | None = None, typedef_map: dict[str, ast.AST] | None = None, class_def_map: dict[str, ast.ClassDef] | None = None) -> None: + self.ir: ir.Module = ir + self.module: ir.Module | None = None + self.builder: ir.IRBuilder | None = None + self._DefineConstants: dict[str, int | str] = {} + self.struct_names: set[str] = struct_names or set() + self.enum_names: set[str] = enum_names or set() + self.module_sha1: str | None = module_sha1 + self.struct_sha1_map: dict[str, str] = struct_sha1_map or {} + self.exception_names: set[str] = exception_names or set() + self.target_triple: str = target_triple or "x86_64-none-elf" + self.target_datalayout: str = target_datalayout or "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + self.typedef_map: dict[str, ast.AST] = typedef_map or {} + self._pyi_tree: ast.Module | None = None + self._t_alias_map: dict[str, type] = {} + # 跨模块类定义映射：{class_name: ClassDef_node}，供 _get_inherited_members + # 查找其他模块定义的父类（如 exprs.py 的 Name(AST) 查找 base.py 的 AST） + self._cross_module_class_defs: dict[str, ast.ClassDef] = class_def_map or {} + t.configure_platform(self.target_triple) + + def generate(self, pyi_content: str, src_path: str) -> str: + tree: ast.Module = ast.parse(pyi_content) + + self._DefineConstants: dict[str, int | str] = {} + self._pyi_tree = tree + global_typedef_map: dict[str, ast.AST] = self.typedef_map + self.typedef_map: dict[str, ast.AST] = {} + if global_typedef_map: + self.typedef_map.update(global_typedef_map) + + # 构建 t/c 类型别名映射（from t import CInt as myint） + self._t_alias_map: dict[str, type] = {} + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.ImportFrom): + if node.module in ('t', 'c'): + for alias in node.names: + name: str = alias.name + asname: str = alias.asname or name + if node.module == 't': + t_cls: type | None = getattr(t, name, None) + if isinstance(t_cls, type) and issubclass(t_cls, t.CType): + self._t_alias_map[asname] = t_cls + + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.AnnAssign) and isinstance(node.target, ast.Name): + var_name: str = node.target.id + if node.value and isinstance(node.value, ast.Constant): + self._DefineConstants[var_name] = node.value.value + elif node.value and isinstance(node.value, ast.Name): + self._DefineConstants[var_name] = node.value.id + is_typedef: bool = False + if isinstance(node.annotation, ast.Attribute) and hasattr(node.annotation, 'attr') and node.annotation.attr == 'CTypedef': + is_typedef = True + elif isinstance(node.annotation, ast.Name) and node.annotation.id == 'CTypedef': + is_typedef = True + elif isinstance(node.annotation, ast.BinOp) and isinstance(node.annotation.left, ast.Attribute) and hasattr(node.annotation.left, 'attr') and node.annotation.left.attr == 'CTypedef': + is_typedef = True + if is_typedef: + if node.value: + self.typedef_map[var_name] = node.value + elif isinstance(node.annotation, ast.BinOp) and node.annotation.right: + self.typedef_map[var_name] = node.annotation.right + + lines: list[str] = [] + lines.append('; ModuleID = "transpyc_decl"') + lines.append(f'target triple = "{self.target_triple}"') + lines.append(f'target datalayout = "{self.target_datalayout}"') + lines.append('') + + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.FunctionDef): + if hasattr(node, 'type_params') and node.type_params: + continue + decl: str | None = self._generate_func_decl(node) + if decl: + lines.append(decl) + elif isinstance(node, ast.AnnAssign): + decl: str | None = self._generate_global_decl(node) + if decl: + lines.append(decl) + elif isinstance(node, ast.Assign): + decl: str | None = self._generate_global_assign_decl(node) + if decl: + lines.append(decl) + elif isinstance(node, ast.ClassDef): + if hasattr(node, 'type_params') and node.type_params: + continue + decls: list[str] = self._generate_class_decl(node) + lines.extend(decls) + + # 后处理：扫描引用但未定义的结构体类型，自动添加 opaque 声明 + # 这解决了泛型类（如 list[T]）在 stub 中被跳过导致的 "undefined type" 错误 + opaque_decls: list[str] = self._collect_undefined_type_decls(lines) + if opaque_decls: + # 在 header 之后插入 opaque 声明（ModuleID, triple, datalayout, 空行 = 4 行） + insert_idx: int = 4 + for i, od in enumerate(opaque_decls): + lines.insert(insert_idx + i, od) + + return '\n'.join(lines) + + def _collect_undefined_type_decls(self, lines: list[str]) -> list[str]: + """扫描所有引用的 %"sha1.TypeName" 类型，返回未定义类型的 opaque 声明列表""" + # 收集已定义的类型名 + defined_types: set[str] = set() + type_def_re: re.Pattern = re.compile(r'^(%".+?")\s*=\s*type\s') + for line in lines: + m = type_def_re.match(line.strip()) + if m: + defined_types.add(m.group(1)) + + # 扫描所有引用的类型名 + referenced_types: set[str] = set() + type_ref_re: re.Pattern = re.compile(r'%"[a-f0-9]+\.[^"]+"') + for line in lines: + for m in type_ref_re.finditer(line): + ref: str = m.group(0) + if ref not in defined_types: + referenced_types.add(ref) + + # 生成 opaque 声明（排序确保确定性） + result: list[str] = [] + for type_name in sorted(referenced_types): + result.append(f'{type_name} = type opaque') + return result + + def _generate_func_decl(self, node: ast.FunctionDef) -> str | None: + """生成函数声明""" + func_name: str = node.name + is_export: bool = self._is_export_func(node) + if self.module_sha1 and not is_export: + func_name = f"{self.module_sha1}.{func_name}" + CReturnTypes: list[ast.AST] = [] + if node.decorator_list: + for decorator in node.decorator_list: + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Attribute): + if decorator.func.attr == 'CReturn': + for arg in decorator.args: + CReturnTypes.append(arg) + if node.returns: + if isinstance(node.returns, ast.Subscript) and isinstance(node.returns.value, ast.Name) and node.returns.value.id == 'tuple': + slice_node: ast.AST = node.returns.slice + if isinstance(slice_node, ast.Tuple): + for elt in slice_node.elts: + CReturnTypes.append(elt) + else: + CReturnTypes.append(slice_node) + ret_type: str + if CReturnTypes: + elem_types: list[str] = [self._get_type_str(rt, embedded=True) for rt in CReturnTypes] + ret_type = '{ ' + ', '.join(elem_types) + ' }' + else: + ret_type = self._get_type_str(node.returns) + if not ret_type or ret_type == 'void': + ret_type = 'void' + elif ret_type == 'i8*': + if node.returns is None: + ret_type = 'void' + + params: list[str] = [] + for arg in node.args.args: + arg_type: str + if arg.annotation: + arg_type = self._get_type_str(arg.annotation) + else: + arg_type = 'i8*' + # C 语言中数组参数退化为指针：[N x elem_type] → elem_type* + arg_type = self._decay_array_to_ptr(arg_type) + if arg_type and arg_type != 'void': + params.append(arg_type) + + param_str: str = ', '.join(params) if params else '' + if node.args.vararg: + param_str = param_str + ', ...' if param_str else '...' + if func_name[0].isdigit(): + return f'declare {ret_type} @"{func_name}"({param_str})' + return f'declare {ret_type} @{func_name}({param_str})' + + def _is_export_func(self, node: ast.FunctionDef) -> bool: + """检查函数是否标记为 CExport""" + if not node.returns: + return False + return self._check_annotation_for_export(node.returns) + + def _check_annotation_for_export(self, annotation: ast.AST | None) -> bool: + """递归检查类型注解中是否包含 CExport 或 t.State""" + return bool(annotation) and (AnnotationContainsName(annotation, 'CExport') or AnnotationContainsName(annotation, 'State')) + + def _generate_global_decl(self, node: ast.AnnAssign) -> str | None: + """生成全局变量声明""" + if not isinstance(node.target, ast.Name): + return None + var_name: str = node.target.id + if isinstance(node.annotation, ast.Attribute) and hasattr(node.annotation, 'attr') and node.annotation.attr == 'CDefine': + return None + if isinstance(node.annotation, ast.Attribute) and hasattr(node.annotation, 'attr') and node.annotation.attr == 'CTypedef': + return None + if isinstance(node.annotation, ast.Name) and node.annotation.id == 'CTypedef': + return None + if isinstance(node.annotation, ast.BinOp) and isinstance(node.annotation.left, ast.Attribute) and hasattr(node.annotation.left, 'attr') and node.annotation.left.attr == 'CTypedef': + return None + if isinstance(node.annotation, ast.List): + for elt in node.annotation.elts: + if isinstance(elt, ast.Attribute) and hasattr(elt, 'attr') and elt.attr in ('CTypedef', 'Callable'): + return None + if isinstance(elt, ast.Subscript) and isinstance(elt.value, ast.Attribute) and isinstance(elt.value.value, ast.Name) and elt.value.value.id == 't' and elt.value.attr == 'Callable': + return None + VarType: str = self._get_type_str(node.annotation) + if VarType.startswith('[0 x ') and node.value and isinstance(node.value, ast.List): + elem_type: str = VarType[5:-1] + actual_count: int = len(node.value.elts) + if actual_count > 0: + VarType = f'[{actual_count} x {elem_type}]' + return f'@{var_name} = external global {VarType}' + + def _generate_global_assign_decl(self, node: ast.Assign) -> str | None: + """从无类型标注赋值推断并生成全局变量声明""" + if not node.targets or not isinstance(node.targets[0], ast.Name): + return None + var_name: str = node.targets[0].id + VarType: str | None = self._infer_type(node.value) + if VarType: + return f'@{var_name} = external global {VarType}' + return None + + def _resolve_base_kind(self, base_name: str) -> str | None: + """解析基类名称的类型种类（支持别名） + + Returns: + 'enum' | 'union' | 'renum' | 'struct' | None + """ + # 1) 查 t 别名映射（from t import CEnum as en） + if base_name in self._t_alias_map: + t_cls: type = self._t_alias_map[base_name] + if issubclass(t_cls, (t.CEnum, t.REnum)): + return 'renum' if issubclass(t_cls, t.REnum) else 'enum' + if issubclass(t_cls, t.CUnion): + return 'union' + if issubclass(t_cls, t.CStruct): + return 'struct' + + # 2) 直接查 t 模块属性 + t_cls: type | None = getattr(t, base_name, None) + if isinstance(t_cls, type) and issubclass(t_cls, t.CType): + if issubclass(t_cls, (t.CEnum, t.REnum)): + return 'renum' if issubclass(t_cls, t.REnum) else 'enum' + if issubclass(t_cls, t.CUnion): + return 'union' + if issubclass(t_cls, t.CStruct): + return 'struct' + + return None + + def _is_marker_base(self, base_name: str) -> bool: + """检查是否为标记基类（Object/CVTable/Exception/CEnum/CUnion/CStruct/REnum 及其别名）""" + if base_name in ('Object', 'CVTable', 'Exception', 'CEnum', 'Enum', 'CStruct', 'CUnion', 'REnum'): + return True + if self._resolve_base_kind(base_name) is not None: + return True + return False + + def _generate_class_decl(self, node: ast.ClassDef) -> List[str]: + """生成结构体/类声明""" + decls: list[str] = [] + class_name: str = node.name + is_enum: bool = False + is_renum: bool = False + is_union: bool = False + is_exception: bool = False + if node.bases: + for base in node.bases: + base_name: str | None = None + if isinstance(base, ast.Attribute) and hasattr(base, 'attr'): + base_name = base.attr + elif isinstance(base, ast.Name) and hasattr(base, 'id'): + base_name = base.id + if base_name: + if base_name in self.exception_names or base_name == 'Exception': + is_exception = True + break + kind: str | None = self._resolve_base_kind(base_name) + if kind == 'enum': + is_enum = True + break + elif kind == 'union': + is_union = True + elif kind == 'renum': + is_renum = True + if is_exception: + return decls + if is_enum: + enum_values: dict[str, int] = {} + next_value: int = 0 + for item in node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + var_name: str = item.target.id + if item.value and isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + enum_values[var_name] = item.value.value + else: + enum_values[var_name] = next_value + next_value += 1 + elif isinstance(item, ast.Assign) and len(item.targets) == 1 and isinstance(item.targets[0], ast.Name): + var_name: str = item.targets[0].id + if item.value and isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + enum_values[var_name] = item.value.value + next_value = item.value.value + 1 + else: + enum_values[var_name] = next_value + next_value += 1 + for var_name, value in enum_values.items(): + decls.append(f'@__config_{class_name}_{var_name} = external global i32') + return decls + if is_renum: + # REnum 布局: { i32 __tag, } + # 对齐 _EmitREnumLlvm (HandlesClassDef.py) 的 Phase 2 逻辑 + # Bug 修复：按每个位置的 max 字段尺寸构建布局，而非选总尺寸最大的变体。 + # 不同变体在同一位置可能有不同尺寸的字段（如 Concrete 位置3是 i32， + # 但 Var 位置3是 MetaType*），用最大变体的布局会导致指针截断。 + variant_fields_list: list[list[str]] = [] + for item in node.body: + if isinstance(item, ast.ClassDef): + variant_fields: list[str] = [] + for nested_item in item.body: + if isinstance(nested_item, ast.AnnAssign) and isinstance(nested_item.target, ast.Name): + ft: str = self._get_type_str(nested_item.annotation, embedded=True) + if ft: + variant_fields.append(ft) + variant_fields_list.append(variant_fields) + elif isinstance(item, ast.Assign): + # 显式值变体（无 payload） + variant_fields_list.append([]) + # 计算每个位置的最大尺寸 + max_field_count: int = 0 + for vf in variant_fields_list: + if len(vf) > max_field_count: + max_field_count = len(vf) + renum_member_types: list[str] = ['i32'] # __tag + for pos in range(max_field_count): + pos_max_size: int = 0 + for vf in variant_fields_list: + if pos < len(vf): + sz: int = self._llvm_type_size_str(vf[pos]) + if sz > pos_max_size: + pos_max_size = sz + # <=4 用 i32，>4 用 i64（8字节，可容纳指针） + if pos_max_size <= 4: + renum_member_types.append('i32') + else: + renum_member_types.append('i64') + if len(renum_member_types) < 2: + # 无类变体或所有变体无字段时，使用 i32 作为占位（对齐 _EmitREnumLlvm 默认行为） + renum_member_types = ['i32', 'i32'] + struct_type_name: str = f'%"{self.module_sha1}.{class_name}"' if self.module_sha1 else f'%struct.{class_name}' + struct_decl: str = f'{struct_type_name} = type {{ {", ".join(renum_member_types)} }}' + decls.append(struct_decl) + return decls + member_types: list[str] = [] + has_bitfield: bool = False + total_bits: int = 0 + has_methods: bool = any(isinstance(item, ast.FunctionDef) for item in node.body) + is_cvtable: bool = False + is_cpython_object: bool = False + is_novtable: bool = False + if hasattr(node, 'decorator_list') and node.decorator_list: + for decorator in node.decorator_list: + if isinstance(decorator, ast.Attribute): + if getattr(decorator.value, 'id', None) == 't': + if decorator.attr == 'CVTable': + is_cvtable = True + elif decorator.attr == 'Object': + is_cpython_object = True + elif decorator.attr == 'NoVTable': + is_novtable = True + elif isinstance(decorator, ast.Name): + if decorator.id == 'CVTable': + is_cvtable = True + elif decorator.id == 'Object': + is_cpython_object = True + elif decorator.id == 'NoVTable': + is_novtable = True + if has_methods and not is_cpython_object: + is_cpython_object = True + has_parent_class: bool = False + parent_is_novtable: bool = False + for base in node.bases: + base_name: str | None = None + if isinstance(base, ast.Attribute): + base_name = base.attr + elif isinstance(base, ast.Name): + base_name = base.id + elif isinstance(base, ast.Subscript): + if isinstance(base.value, ast.Attribute): + base_name = base.value.attr + elif isinstance(base.value, ast.Name): + base_name = base.value.id + if base_name and not self._is_marker_base(base_name): + has_parent_class = True + # 检查父类是否为 @t.NoVTable（非多态继承） + _p_node: ast.ClassDef | None = None + for n in ast.iter_child_nodes(self._pyi_tree): + if isinstance(n, ast.ClassDef) and n.name == base_name: + _p_node = n + break + if _p_node is None: + _p_node = self._cross_module_class_defs.get(base_name) + if _p_node is not None and hasattr(_p_node, 'decorator_list') and _p_node.decorator_list: + for dec in _p_node.decorator_list: + if isinstance(dec, ast.Attribute) and getattr(dec.value, 'id', None) == 't' and dec.attr == 'NoVTable': + parent_is_novtable = True + elif isinstance(dec, ast.Name) and dec.id == 'NoVTable': + parent_is_novtable = True + break + # 仅当非 NoVTable 且父类非 NoVTable 时才因继承启用 CVTable + # （@t.NoVTable 标记的类及其子类不使用 vtable，字段展平嵌入） + if has_parent_class and not is_cvtable and not is_novtable and not parent_is_novtable: + is_cvtable = True + base_has_vtable: bool = False + for base in node.bases: + base_name: str | None = None + if isinstance(base, ast.Attribute): + base_name = base.attr + elif isinstance(base, ast.Name): + base_name = base.id + elif isinstance(base, ast.Subscript): + if isinstance(base.value, ast.Attribute): + base_name = base.value.attr + elif isinstance(base.value, ast.Name): + base_name = base.value.id + if base_name and not self._is_marker_base(base_name): + # 先在当前 .pyi 中查找父类，找不到则查跨模块类定义映射 + _base_node: ast.ClassDef | None = None + for n in ast.iter_child_nodes(self._pyi_tree): + if isinstance(n, ast.ClassDef) and n.name == base_name: + _base_node = n + break + if _base_node is None: + _base_node = self._cross_module_class_defs.get(base_name) + if _base_node is not None and hasattr(_base_node, 'decorator_list') and _base_node.decorator_list: + for dec in _base_node.decorator_list: + if isinstance(dec, ast.Attribute) and getattr(dec.value, 'id', None) == 't' and dec.attr == 'CVTable': + base_has_vtable = True + elif isinstance(dec, ast.Name) and dec.id == 'CVTable': + base_has_vtable = True + if base_has_vtable: + break + # NoVTable 类不添加 vtable 指针（即使有方法和父类） + if has_methods and is_cvtable and not base_has_vtable and not is_novtable: + member_types.append('i8*') + seen_member_names: set[str] = set() + for base in node.bases: + base_name: str | None = None + if isinstance(base, ast.Attribute): + base_name = base.attr + elif isinstance(base, ast.Name): + base_name = base.id + elif isinstance(base, ast.Subscript): + if isinstance(base.value, ast.Attribute): + base_name = base.value.attr + elif isinstance(base.value, ast.Name): + base_name = base.value.id + if base_name and not self._is_marker_base(base_name): + base_member_types: list[str] + base_seen: set[str] + base_member_types, base_seen = self._get_inherited_members(base_name) + for mt in base_member_types: + member_types.append(mt) + seen_member_names.update(base_seen) + for item in node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + seen_member_names.add(item.target.id) + init_members: list[ast.AnnAssign] = [] + for item in node.body: + if isinstance(item, ast.FunctionDef) and item.name == '__init__': + for stmt in item.body: + if (isinstance(stmt, ast.AnnAssign) + and isinstance(stmt.target, ast.Attribute) + and isinstance(stmt.target.value, ast.Name) + and stmt.target.value.id == 'self'): + init_members.append(stmt) + untyped_self_members: list[ast.Assign] = [] + for m_name in [m.attr for m in init_members if isinstance(m.target, ast.Attribute)]: + seen_member_names.add(m_name) + for item in node.body: + if isinstance(item, ast.FunctionDef): + if hasattr(item, 'type_params') and item.type_params: + continue + for stmt in item.body: + if (isinstance(stmt, ast.Assign) + and len(stmt.targets) == 1 + and isinstance(stmt.targets[0], ast.Attribute) + and isinstance(stmt.targets[0].value, ast.Name) + and stmt.targets[0].value.id == 'self'): + attr_name: str = stmt.targets[0].attr + if attr_name not in seen_member_names: + seen_member_names.add(attr_name) + untyped_self_members.append(stmt) + for item in list(node.body) + init_members + untyped_self_members: + if isinstance(item, ast.AnnAssign): + # 跳过编译期元数据字段（__provides__/__requires__/__require_must__） + if isinstance(item.target, ast.Name) and item.target.id in ('__provides__', '__requires__', '__require_must__'): + continue + bit_width: int | None = self._get_bitfield_width(item.annotation) + if bit_width is not None: + has_bitfield = True + total_bits += bit_width + else: + VarType: str = self._get_type_str(item.annotation, embedded=True) + if IsListAnnotation(item.annotation): + slice_node: ast.AST = item.annotation.slice + is_dynamic: bool = False + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + count_node: ast.AST = slice_node.elts[1] + if isinstance(count_node, ast.Constant) and count_node.value is None: + is_dynamic = True + elif not (isinstance(count_node, ast.Constant) and isinstance(count_node.value, int) and count_node.value > 0): + if not isinstance(count_node, (ast.Name, ast.BinOp)): + is_dynamic = True + elif not isinstance(slice_node, ast.Tuple): + is_dynamic = True + if is_dynamic: + elem_type: str = self._get_type_str(slice_node if not isinstance(slice_node, ast.Tuple) else slice_node.elts[0], embedded=True) + init_len: int = 0 + if isinstance(item.value, ast.List): + init_len = len(item.value.elts) + elif isinstance(item.value, ast.Constant) and isinstance(item.value.value, str): + init_len = len(item.value.value) + 1 + VarType = f'[{init_len} x {elem_type}]' + else: + # 固定大小数组: t.CArray[elem_type, count] → [count x elem_type] + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + elem_type: str = self._get_type_str(slice_node.elts[0], embedded=True) + count_val: int = self._get_const_int(slice_node.elts[1]) + if count_val > 0: + VarType = f'[{count_val} x {elem_type}]' + elif isinstance(slice_node, (ast.Attribute, ast.Name, ast.Subscript)): + # 单参数 t.CArray[elem_type] → 指针模式 + elem_type: str = self._get_type_str(slice_node, embedded=True) + VarType = f'{elem_type}*' + member_types.append(VarType) + elif isinstance(item, ast.Assign): + if (len(item.targets) == 1 + and isinstance(item.targets[0], ast.Attribute) + and isinstance(item.targets[0].value, ast.Name) + and item.targets[0].value.id == 'self'): + attr_name: str = item.targets[0].attr + if attr_name in seen_member_names: + continue + InferredType: str = 'i32' + if item.value and isinstance(item.value, ast.Constant): + if isinstance(item.value.value, float): + InferredType = 'float' + elif isinstance(item.value.value, bool): + InferredType = 'i8' + elif isinstance(item.value.value, str): + InferredType = 'i8*' + member_types.append(InferredType) + else: + member_types.append('i32') + if has_bitfield: + if total_bits <= 8: + member_types.insert(0, 'i8') + elif total_bits <= 16: + member_types.insert(0, 'i16') + elif total_bits <= 32: + member_types.insert(0, 'i32') + else: + member_types.insert(0, 'i64') + struct_type_name: str = f'%"{self.module_sha1}.{class_name}"' if self.module_sha1 else f'%struct.{class_name}' + struct_decl: str = f'{struct_type_name} = type {{ {", ".join(member_types)} }}' + decls.append(struct_decl) + if is_cpython_object or is_cvtable: + new_func_name: str = f'{class_name}.__before_init__' + if self.module_sha1: + new_func_name = f"{self.module_sha1}.{new_func_name}" + new_func_decl: str + if new_func_name[0].isdigit(): + new_func_decl = f'declare void @"{new_func_name}"({struct_type_name}*)' + else: + new_func_decl = f'declare void @{new_func_name}({struct_type_name}*)' + decls.append(new_func_decl) + for item in node.body: + if isinstance(item, ast.FunctionDef): + if hasattr(item, 'type_params') and item.type_params: + continue + method_name: str = f'{class_name}.{item.name}' + if self.module_sha1: + method_name = f"{self.module_sha1}.{method_name}" + ret_type: str = self._get_type_str(item.returns) if item.returns else 'void' + if not ret_type: + ret_type = 'void' + params: list[str] = [] + for arg_idx, arg in enumerate(item.args.args): + arg_type: str + if arg.annotation: + arg_type = self._get_type_str(arg.annotation) + elif arg_idx == 0 and arg.arg == 'self': + arg_type = f'{struct_type_name}*' + else: + arg_type = 'i8*' + # C 语言中数组参数退化为指针：[N x elem_type] → elem_type* + arg_type = self._decay_array_to_ptr(arg_type) + params.append(arg_type) + param_str: str = ', '.join(params) if params else '' + if method_name[0].isdigit(): + decls.append(f'declare {ret_type} @"{method_name}"({param_str})') + else: + decls.append(f'declare {ret_type} @{method_name}({param_str})') + return decls + + def _get_inherited_members(self, base_name: str) -> tuple[list[str], set[str]]: + member_types: list[str] = [] + seen_names: set[str] = set() + if not hasattr(self, '_pyi_tree') or self._pyi_tree is None: + return member_types, seen_names + base_node: ast.ClassDef | None = None + for node in ast.iter_child_nodes(self._pyi_tree): + if isinstance(node, ast.ClassDef) and node.name == base_name: + base_node = node + break + if base_node is None: + # 跨模块父类查找：当前 .pyi 中没有父类定义时，从跨模块类定义映射中查找 + # （如 exprs.py 的 Name(AST)，AST 定义在 base.py） + base_node = self._cross_module_class_defs.get(base_name) + if base_node is None: + return member_types, seen_names + base_decls: list[str] = self._generate_class_decl(base_node) + for decl in base_decls: + if '= type {' in decl: + type_body: str = decl.split('= type {')[1].rstrip('}').strip() + if type_body: + for t in type_body.split(','): + t = t.strip() + if t: + member_types.append(t) + break + for item in base_node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + seen_names.add(item.target.id) + elif isinstance(item, ast.Assign) and len(item.targets) == 1: + if isinstance(item.targets[0], ast.Attribute) and isinstance(item.targets[0].value, ast.Name) and item.targets[0].value.id == 'self': + seen_names.add(item.targets[0].attr) + for item in base_node.body: + if isinstance(item, ast.FunctionDef) and item.name == '__init__': + for stmt in item.body: + if isinstance(stmt, ast.AnnAssign) and isinstance(stmt.target, ast.Attribute) and isinstance(stmt.target.value, ast.Name) and stmt.target.value.id == 'self': + seen_names.add(stmt.target.attr) + elif isinstance(stmt, ast.Assign) and len(stmt.targets) == 1 and isinstance(stmt.targets[0], ast.Attribute) and isinstance(stmt.targets[0].value, ast.Name) and stmt.targets[0].value.id == 'self': + seen_names.add(stmt.targets[0].attr) + return member_types, seen_names + + def _llvm_type_size_str(self, type_str: str) -> int: + """估算 LLVM 类型字符串的大小（字节），用于 REnum 变体大小比较 + + 对齐 _EmitREnumLlvm 中的大小计算逻辑： + - i1=1, i8=1, i16=2, i32=4, i64=8, float=4, double=8, ptr=8 + - [N x T] = N * size(T) + - 其他结构体类型按指针大小 8 估算（保守值） + """ + s: str = type_str.strip() + if not s: + return 0 + # 指针类型 + if s.endswith('*'): + return 8 + # 数组类型 [N x T] + m = self._ARR_TYPE_RE.match(s) if hasattr(self, '_ARR_TYPE_RE') else None + if m: + try: + n: int = int(m.group(1)) + return n * self._llvm_type_size_str(m.group(2)) + except ValueError: + return 8 + # 基本整数/浮点类型 + size_map: dict[str, int] = { + 'i1': 1, 'i8': 1, 'i16': 2, 'i32': 4, 'i64': 8, + 'float': 4, 'double': 8, 'half': 2, 'fp128': 16, + 'void': 0, + } + if s in size_map: + return size_map[s] + # %{...} 结构体类型 - 保守估算为指针大小 + if s.startswith('%') or s.startswith('{'): + return 8 + # 其他未知类型保守估算 + return 8 + + def _get_type_str(self, annotation: ast.AST | None, embedded: bool = False) -> str: + if annotation is None: + return 'i8*' + + non_struct_types: set[str] = {'list', 'dict', 'tuple', 'set', 'array', 'CArray'} + + if isinstance(annotation, ast.Name): + if annotation.id == 'None': + return 'void' + if annotation.id in ('str', 'bytes'): + return 'i8*' + llvm_type: str | None = CTypeRegistry.NameToLLVM(annotation.id) + if llvm_type is not None: + return llvm_type + resolved: tuple[type, int] | None = CTypeRegistry.ResolveName(annotation.id) + if resolved is not None: + ctype_cls: type + ptr_level: int + ctype_cls, ptr_level = resolved + base: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if ptr_level > 0: + if base == 'void': + return 'i8*' + if '*' in base: + return base + return f'{base}*' + return base + if annotation.id in self.struct_names: + sha1: str | None = self.struct_sha1_map.get(annotation.id, self.module_sha1) + sname: str = f'%"{sha1}.{annotation.id}"' if sha1 else f'%struct.{annotation.id}' + if embedded: + return sname + else: + return f'{sname}*' + if annotation.id in non_struct_types: + return 'i32' + if annotation.id in self.enum_names: + return 'i32' + if annotation.id in self.typedef_map: + resolved_str: str = self._get_type_str(self.typedef_map[annotation.id], embedded=embedded) + return resolved_str + # 检查 t 类型别名（from t import CInt as myint） + if annotation.id in self._t_alias_map: + t_cls: type = self._t_alias_map[annotation.id] + llvm: str | None = CTypeRegistry.CTypeToLLVM(t_cls) + if llvm: + return llvm + sha1: str | None = self.struct_sha1_map.get(annotation.id, self.module_sha1) + sname: str = f'%"{sha1}.{annotation.id}"' if sha1 else f'%struct.{annotation.id}' + return f'{sname}*' + elif isinstance(annotation, ast.Attribute): + attr_name: str = annotation.attr if hasattr(annotation, 'attr') else '' + module_name: str = '' + if hasattr(annotation, 'value'): + if isinstance(annotation.value, ast.Name): + module_name = annotation.value.id + elif isinstance(annotation.value, ast.Attribute) and hasattr(annotation.value, 'attr'): + module_name = annotation.value.attr + if (module_name == 't' and attr_name == 'State') or attr_name == 'State': + return '' + if (module_name == 't' and attr_name == 'Callable') or attr_name == 'Callable': + return 'i8*' + llvm_type: str | None = CTypeRegistry.NameToLLVM(attr_name) + if llvm_type is not None: + return llvm_type + resolved: tuple[type, int] | None = CTypeRegistry.ResolveName(attr_name) + if resolved is not None: + ctype_cls: type + ptr_level: int + ctype_cls, ptr_level = resolved + base: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if ptr_level > 0: + if base == 'void': + return 'i8*' + if '*' in base: + return base + return f'{base}*' + return base + if attr_name in self.typedef_map: + return self._get_type_str(self.typedef_map[attr_name], embedded=embedded) + if attr_name in self.enum_names: + return 'i32' + if attr_name in self.struct_names: + sha1: str | None = self.struct_sha1_map.get(attr_name, self.module_sha1) + sname: str = f'%"{sha1}.{attr_name}"' if sha1 else f'%struct.{attr_name}' + if embedded: + return sname + else: + return f'{sname}*' + if attr_name and attr_name[0].isupper() and attr_name not in non_struct_types: + sha1: str | None = self.struct_sha1_map.get(attr_name, self.module_sha1) + sname: str = f'%"{sha1}.{attr_name}"' if sha1 else f'%struct.{attr_name}' + return f'{sname}*' + if attr_name and attr_name not in non_struct_types: + sha1: str | None = self.struct_sha1_map.get(attr_name, self.module_sha1) + sname: str = f'%"{sha1}.{attr_name}"' if sha1 else f'%struct.{attr_name}' + return f'{sname}*' + return 'i32' + elif isinstance(annotation, ast.BinOp): + left_type: str = self._get_type_str(annotation.left, embedded=embedded) + right_type: str = self._get_type_str(annotation.right, embedded=embedded) + if left_type in ('', 'void'): + return right_type if right_type not in ('', 'void') else (left_type or right_type) + if right_type in ('', 'void'): + return left_type + # t.CPtr (i8*) 与 left 合并：对齐 Phase 2 _MergeAllComponentTypes 的 + # "first CPtr absorption" 语义：struct | t.CPtr → struct*（吸收，struct 变量默认是指针）。 + # 但对于本身已是指针的类型（如 bytes=str=i8*），bytes | t.CPtr → i8**（不吸收）。 + # 例外：泛型特化类型（含 '['）回退为 i8*，避免引用未定义的特化结构体。 + if right_type == 'i8*': + if '[' in left_type: + return 'i8*' + if left_type in ('', 'void'): + return 'i8*' + if left_type.startswith('%"') and left_type.endswith('*'): + return left_type + if '*' in left_type: + return f'{left_type}*' + return f'{left_type}*' + if '*' in left_type: + return left_type + if '*' in right_type: + return right_type + return left_type + elif isinstance(annotation, ast.Subscript): + base: str = self._get_type_str(annotation.value) + if base == 'i8*' and isinstance(annotation.slice, ast.Constant): + return f'[{self._get_const_int(annotation.slice)} x i8]' + if isinstance(annotation.slice, ast.Constant) and isinstance(annotation.slice.value, int): + return f'[{annotation.slice.value} x {base}]' + # 处理 t.CArray[elem_type, count] 注解 → [count x elem_type] + # 支持 t.CArray (ast.Attribute) 和 CArray (ast.Name) 两种写法 + ValueIsCArray: bool = ( + (isinstance(annotation.value, ast.Attribute) and annotation.value.attr == 'CArray') + or (isinstance(annotation.value, ast.Name) and annotation.value.id == 'CArray') + ) + if ValueIsCArray: + SliceNode: ast.AST = annotation.slice + if isinstance(SliceNode, ast.Tuple) and len(SliceNode.elts) == 2: + ElemType: str = self._get_type_str(SliceNode.elts[0], embedded=True) + CountVal: int = self._get_const_int(SliceNode.elts[1]) + if CountVal > 0: + return f'[{CountVal} x {ElemType}]' + # count 为 None 或 0 → 零长度数组（外部符号） + return f'[0 x {ElemType}]' + elif isinstance(SliceNode, (ast.Attribute, ast.Name, ast.Subscript)): + # 单参数 t.CArray[elem_type] → 指针模式 + ElemType: str = self._get_type_str(SliceNode, embedded=True) + return f'{ElemType}*' + if isinstance(annotation.value, ast.Name) and annotation.value.id == 'tuple': + slice_node: ast.AST = annotation.slice + elem_types: list[str] + if isinstance(slice_node, ast.Tuple): + elem_types = [self._get_type_str(e, embedded=True) for e in slice_node.elts] + else: + elem_types = [self._get_type_str(slice_node, embedded=True)] + return '{ ' + ', '.join(elem_types) + ' }' + # 处理泛型类类型注解（如 list[str]） + if isinstance(annotation.value, ast.Name): + gc_name: str = annotation.value.id + if gc_name in non_struct_types and gc_name != 'tuple': + gc_slice: ast.AST = annotation.slice + gc_args: list[str] = [] + if isinstance(gc_slice, ast.Name): + gc_args.append(gc_slice.id) + elif isinstance(gc_slice, ast.Tuple): + gc_valid: bool = True + for gc_elt in gc_slice.elts: + if isinstance(gc_elt, ast.Name): + gc_args.append(gc_elt.id) + else: + gc_valid = False + break + if not gc_valid: + gc_args = [] + if gc_args: + gc_mangled: list[str] = [] + for gc_ta in gc_args: + gc_llvm: str | None = CTypeRegistry.NameToLLVM(gc_ta) + if gc_llvm: + if gc_llvm in ('float', 'double', 'half', 'fp128'): + gc_cls: type | None = CTypeRegistry.GetClassByName(gc_ta) + gc_sz: int = gc_cls().Size if gc_cls else 0 + gc_mangled.append('f' + str(gc_sz)) + else: + gc_mangled.append(gc_llvm) + else: + gc_mangled.append(gc_ta) + gc_spec: str = gc_name + '[' + ']['.join(gc_mangled) + ']' + gc_sha1: str | None = self.module_sha1 + gc_sname: str = f'%"{gc_sha1}.{gc_spec}"' if gc_sha1 else f'%struct.{gc_spec}' + if embedded: + return gc_sname + return f'{gc_sname}*' + return f'{base}' + elif isinstance(annotation, ast.Constant): + if annotation.value is None: + return 'void' + if isinstance(annotation.value, int): + return 'i32' + if isinstance(annotation.value, str): + type_name: str = annotation.value + if type_name in self.enum_names: + return 'i32' + if type_name in self.struct_names: + sha1: str | None = self.struct_sha1_map.get(type_name, self.module_sha1) + sname: str = f'%"{sha1}.{type_name}"' if sha1 else f'%struct.{type_name}' + if embedded: + return sname + else: + return f'{sname}*' + llvm_type: str | None = CTypeRegistry.NameToLLVM(type_name) + if llvm_type is not None: + return llvm_type + resolved: tuple[type, int] | None = CTypeRegistry.ResolveName(type_name) + if resolved is not None: + ctype_cls: type + ptr_level: int + ctype_cls, ptr_level = resolved + base: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if ptr_level > 0: + if base == 'void': + return 'i8*' + if '*' in base: + return base + return f'{base}*' + return base + sha1: str | None = self.struct_sha1_map.get(type_name, self.module_sha1) + sname: str = f'%"{sha1}.{type_name}"' if sha1 else f'%struct.{type_name}' + return f'{sname}*' + return 'i8*' + elif isinstance(annotation, ast.Call): + if isinstance(annotation.func, ast.Name) and annotation.func.id == 'callable': + return 'i8*' + if isinstance(annotation.func, ast.Attribute) and annotation.func.attr == 'Callable': + return 'i8*' + return 'i32' + + def _infer_type(self, value: ast.AST) -> str: + """从值推断类型""" + if isinstance(value, ast.Constant): + if isinstance(value.value, int): + return 'i32' + elif isinstance(value.value, float): + return 'double' + elif isinstance(value.value, str): + return 'i8*' + elif isinstance(value.value, bool): + return 'i8' + elif isinstance(value, ast.List): + return 'i8*' + elif isinstance(value, ast.Dict): + return 'i8*' + elif isinstance(value, ast.Name): + return 'i32' + elif isinstance(value, ast.BinOp): + return self._infer_type(value.left) + elif isinstance(value, ast.Call): + return 'i8*' + return 'i32' + + def _get_bitfield_width(self, annotation: ast.AST) -> int | None: + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + right_width: int | None = self._get_bitfield_width(annotation.right) + if right_width is not None: + return right_width + return self._get_bitfield_width(annotation.left) + if isinstance(annotation, ast.Call): + if isinstance(annotation.func, ast.Attribute) and annotation.func.attr == 'Bit': + if annotation.args: + arg: ast.AST = annotation.args[0] + if isinstance(arg, ast.Constant) and isinstance(arg.value, int): + return arg.value + if isinstance(annotation, ast.Subscript): + if isinstance(annotation.value, ast.Attribute) and annotation.value.attr == 'Bit': + if isinstance(annotation.slice, ast.Constant) and isinstance(annotation.slice.value, int): + return annotation.slice.value + return None + + def _get_const_int(self, node: ast.AST) -> int: + """获取常量整数值，支持符号常量和简单表达式""" + if isinstance(node, ast.Constant) and isinstance(node.value, int): + return node.value + if isinstance(node, ast.Name): + if node.id in self._DefineConstants: + val: int | str = self._DefineConstants[node.id] + if isinstance(val, int): + return val + if isinstance(node, ast.BinOp): + left_val: int = self._get_const_int(node.left) + right_val: int = self._get_const_int(node.right) + if left_val and right_val: + if isinstance(node.op, ast.Add): + return left_val + right_val + if isinstance(node.op, ast.Sub): + return left_val - right_val + if isinstance(node.op, ast.Mult): + return left_val * right_val + if isinstance(node.op, ast.Div): + return left_val // right_val + if isinstance(node.op, ast.FloorDiv): + return left_val // right_val + return 0 diff --git a/lib/Projectrans/Phase1Generator.py b/lib/Projectrans/Phase1Generator.py new file mode 100644 index 0000000..c30ed83 --- /dev/null +++ b/lib/Projectrans/Phase1Generator.py @@ -0,0 +1,595 @@ +from __future__ import annotations + +import os +import ast +import json +import shutil +import traceback + +from lib.Projectrans.Utils import compute_sha1, get_file_dependencies, find_reachable_files_from_entries, parse_python_file +from lib.Projectrans.DeclarationGenerator import DeclarationGenerator +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _ConfigMode +from StubGen import PythonToStubConverter + +# 全局缓存目录（不被 --clean 清除），用于缓存库文件的 .pyi / .stub.ll / .doc.json +_COMPILER_ROOT: str = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +_GLOBAL_CACHE_DIR: str = os.path.join(_COMPILER_ROOT, '.transpyc_cache') + + +def _TryGlobalCache(Sha1: str, Ext: str, TempPath: str) -> bool: + """检查全局缓存中是否存在指定文件，若存在则复制到 temp_dir 并返回 True。""" + GlobalPath: str = os.path.join(_GLOBAL_CACHE_DIR, f"{Sha1}.{Ext}") + if os.path.isfile(GlobalPath): + try: + os.makedirs(os.path.dirname(TempPath), exist_ok=True) + shutil.copy2(GlobalPath, TempPath) + return True + except Exception: + pass + return False + + +def _SaveToGlobalCache(Sha1: str, Ext: str, TempPath: str) -> None: + """将文件保存到全局缓存供后续使用。""" + GlobalPath: str = os.path.join(_GLOBAL_CACHE_DIR, f"{Sha1}.{Ext}") + try: + os.makedirs(_GLOBAL_CACHE_DIR, exist_ok=True) + shutil.copy2(TempPath, GlobalPath) + except Exception: + pass + + +class Phase1Generator: + """阶段一：从源文件生成声明接口""" + + def __init__(self, src_root: str, temp_dir: str, include_dirs: list[str] | None = None, entry_files: list[str] | None = None, target_triple: str | None = None, target_datalayout: str | None = None) -> None: + self.src_root: str = os.path.abspath(src_root) + self.temp_dir: str = os.path.abspath(temp_dir) + self.include_dirs: list[str] = include_dirs or [] + self.sha1_map: dict[str, str] = {} + self.include_py_map: dict[str, str] = {} + self.entry_files: list[str] | None = entry_files + self.target_triple: str | None = target_triple + self.target_datalayout: str | None = target_datalayout + os.makedirs(self.temp_dir, exist_ok=True) + # 增量编译 manifest：{abs_path: {sha1, mtime, size}} + # 未变更文件（mtime+size 匹配）跳过 open+read+sha1，仅需 stat + self._manifest: dict[str, dict] = {} + self._typedef_map_cache: dict[str, ast.AST] | None = None + self._manifest_path: str = os.path.join(self.temp_dir, '_phase1_manifest.json') + self._load_manifest() + + def _load_manifest(self) -> None: + """加载上次运行的 manifest（mtime+size → sha1 缓存）""" + try: + if os.path.isfile(self._manifest_path): + with open(self._manifest_path, 'r', encoding='utf-8') as f: + self._manifest = json.load(f) + except Exception: + self._manifest = {} + + def _save_manifest(self) -> None: + """保存 manifest 供下次增量编译使用""" + try: + with open(self._manifest_path, 'w', encoding='utf-8') as f: + json.dump(self._manifest, f) + except Exception: + pass + + def _get_sha1(self, src_path: str) -> str: + """用 mtime+size 缓存加速 sha1 计算。 + + 未变更文件（mtime+size 匹配 manifest）直接返回缓存的 sha1， + 跳过 open+read+compute_sha1，仅需一次 stat 调用。 + """ + abs_path: str = os.path.abspath(src_path) + try: + st: os.stat_result = os.stat(abs_path) + except OSError: + return '' + mtime: float = st.st_mtime + size: int = st.st_size + entry: dict = self._manifest.get(abs_path, {}) + if entry.get('mtime') == mtime and entry.get('size') == size: + sha1: str = entry.get('sha1', '') + if sha1: + return sha1 + # mtime/size 不匹配，需读取内容重新计算 + try: + with open(abs_path, 'r', encoding='utf-8') as f: + content: str = f.read() + except Exception: + return '' + sha1 = compute_sha1(content) + self._manifest[abs_path] = {'sha1': sha1, 'mtime': mtime, 'size': size} + return sha1 + + def _get_needed_include_files(self, reachable_source_files: set[str]) -> list[tuple[str, str, str]]: + """从可达源文件收集所有被引用（含传递依赖）的 include 文件""" + include_file_map: dict[str, tuple[str, str, str]] = {} + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + for root, dirs, files in os.walk(includes_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith('.py') or fname.endswith('.pyi'): + src_path: str = os.path.join(root, fname) + rel_from_inc: str = os.path.relpath(src_path, includes_dir) + ModulePath: str = rel_from_inc.replace(os.sep, '.').replace('/', '.') + module_name: str = os.path.splitext(ModulePath)[0] + info: tuple[str, str, str] = (src_path, rel_from_inc, includes_dir) + include_file_map[module_name] = info + top_pkg: str = module_name.split('.')[0] + # __init__.py 是包入口，应优先作为 top_pkg 的代表 + # （参考 find_reachable_files_from_entries 的优先级逻辑）， + # 否则若 os/path.py 先被遍历，'os' 会错误指向 path.py + # 而非 os/__init__.py，导致包入口 SHA1 未注册到 sha1_map + if module_name == f"{top_pkg}.__init__": + include_file_map[top_pkg] = info + elif top_pkg not in include_file_map: + include_file_map[top_pkg] = info + + imported_from_src: set[str] = set() + for src_path in reachable_source_files: + deps: set[str] = get_file_dependencies(src_path, self.src_root) + imported_from_src.update(deps) + + # 检测 dict/list 容器使用，自动添加 _dict/_list/json 依赖 + for src_path in reachable_source_files: + try: + # 复用 parse_python_file 的 AST 缓存（get_file_dependencies 已解析过同一文件） + content, tree = parse_python_file(src_path) + if tree is None: + continue + for node in ast.walk(tree): + if isinstance(node, ast.Subscript) and isinstance(node.value, ast.Name): + if node.value.id == 'dict': + imported_from_src.add('_dict') + imported_from_src.add('json') + break + elif node.value.id == 'list': + imported_from_src.add('_list') + break + if isinstance(node, ast.Call) and isinstance(node.func, ast.Name): + if node.func.id == 'dict': + imported_from_src.add('_dict') + imported_from_src.add('json') + break + elif node.func.id == 'list': + imported_from_src.add('_list') + break + # 检测类型注解 d: dict / l: list (非泛型形式) + if isinstance(node, ast.AnnAssign) and isinstance(node.annotation, ast.Name): + if node.annotation.id == 'dict': + imported_from_src.add('_dict') + imported_from_src.add('json') + break + elif node.annotation.id == 'list': + imported_from_src.add('_list') + break + except Exception: + pass + + needed: set[str] = set() + queue: list[str] = list(imported_from_src) + while queue: + mod_name: str = queue.pop(0) + if mod_name in needed: + continue + if mod_name in include_file_map: + needed.add(mod_name) + src_path: str = include_file_map[mod_name][0] + includes_dir: str = include_file_map[mod_name][2] + deps: set[str] = get_file_dependencies(src_path, includes_dir) + for dep in deps: + if dep in include_file_map and dep not in needed: + queue.append(dep) + + needed_infos: list[tuple[str, str, str]] = [] + for mod_name in sorted(needed): + if mod_name in include_file_map: + info: tuple[str, str, str] = include_file_map[mod_name] + if info not in needed_infos: + needed_infos.append(info) + + return needed_infos + + def run(self) -> None: + """扫描源目录，只处理入口文件可达的 .py 文件（导入遍历）""" + py_files: list[str] + if self.entry_files: + py_files = list(self.entry_files) + else: + main_py: str = os.path.join(self.src_root, 'main.py') + if os.path.exists(main_py): + py_files = [main_py] + else: + py_files = [] + for root, dirs, files in os.walk(self.src_root): + dirs[:] = [d for d in dirs if d not in ('__pycache__',)] + for file in files: + if file.endswith('.py'): + py_files.append(os.path.join(root, file)) + + if not py_files: + _vlog().warning("未找到入口文件或源文件") + return + + reachable: set[str] + if self.entry_files: + reachable = find_reachable_files_from_entries(self.src_root, py_files) + else: + reachable = find_reachable_files_from_entries(self.src_root, py_files) + + _vlog().info(f"找到 {len(reachable)} 个可达源文件（从入口遍历）") + + for i, src_path in enumerate(sorted(reachable), 1): + rel: str = os.path.relpath(src_path, self.src_root) + _vlog().info(f"[{i}/{len(reachable)}] 生成签名: {rel}") + try: + self._process_file_pyi(src_path, rel) + except Exception as e: + _vlog().error(f"生成签名失败: {e}") + traceback.print_exc() + + needed_includes: list[tuple[str, str, str]] = self._get_needed_include_files(reachable) + self._process_include_py_files_pyi(needed_includes) + + struct_names: set[str] + enum_names: set[str] + struct_sha1_map: dict[str, str] + exception_names: set[str] + struct_names, enum_names, struct_sha1_map, exception_names, class_def_map = self._build_struct_registry() + _vlog().info(f"结构体注册表: {len(struct_names)} 个类型, {len(enum_names)} 个枚举") + for name in sorted(struct_names): + _vlog().debug(f" {name}") + + # 预先收集 typedef 映射，避免在 _process_file_stub 中对每个源文件重复扫描所有 .pyi + self._typedef_map_cache: dict[str, ast.AST] = self._collect_typedef_map() + + for i, src_path in enumerate(sorted(reachable), 1): + rel: str = os.path.relpath(src_path, self.src_root) + try: + self._process_file_stub(src_path, rel, struct_names, enum_names=enum_names, struct_sha1_map=struct_sha1_map, exception_names=exception_names, class_def_map=class_def_map) + except Exception as e: + _vlog().error(f"生成声明失败: {e}") + traceback.print_exc() + + self._process_include_py_files_stub(struct_names, needed_includes, enum_names=enum_names, struct_sha1_map=struct_sha1_map, exception_names=exception_names, class_def_map=class_def_map) + + # 保存 manifest 供下次增量编译使用 + self._save_manifest() + + _vlog().success(f"声明接口生成到: {self.temp_dir}") + _vlog().info(f"SHA1 映射表 ({len(self.sha1_map)} 个文件):") + for sha1, rel in sorted(self.sha1_map.items()): + _vlog().debug(f" {sha1} -> {rel}") + + def _collect_include_py_files(self) -> list[tuple[str, str, str]]: + include_py_files: list[tuple[str, str, str]] = [] + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + for root, dirs, files in os.walk(includes_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith('.py') or fname.endswith('.pyi'): + src_path: str = os.path.join(root, fname) + rel_from_inc: str = os.path.relpath(src_path, includes_dir) + include_py_files.append((src_path, rel_from_inc, includes_dir)) + return include_py_files + + def _process_include_py_files_pyi(self, needed_includes: list[tuple[str, str, str]] | None = None) -> None: + include_py_files: list[tuple[str, str, str]] + if needed_includes is not None: + include_py_files = needed_includes + else: + include_py_files = self._collect_include_py_files() + if not include_py_files: + return + _vlog().info(f"处理 {len(include_py_files)} 个被引用的 Python 库文件") + for src_path, rel_from_inc, includes_dir in include_py_files: + ModulePath: str = rel_from_inc.replace(os.sep, '.').replace('/', '.') + module_name: str = os.path.splitext(ModulePath)[0] + try: + # 增量优化：用 mtime+size 获取 sha1，未变更文件跳过 open+read + sha1: str = self._get_sha1(src_path) + if not sha1: + continue + self.sha1_map[sha1] = f"includes/{rel_from_inc}" + top_module: str = rel_from_inc.split(os.sep)[0].split('/')[0] + if top_module not in self.include_py_map: + self.include_py_map[top_module] = sha1 + self.include_py_map[module_name] = sha1 + + sig_path: str = os.path.join(self.temp_dir, f"{sha1}.pyi") + if os.path.isfile(sig_path): + _vlog().info(f" 缓存命中: {rel_from_inc} -> {sha1}.pyi") + continue + + # 全局缓存检查（不被 --clean 清除） + if _TryGlobalCache(sha1, 'pyi', sig_path): + _vlog().info(f" 缓存命中(全局): {rel_from_inc} -> {sha1}.pyi") + continue + + # 缓存未命中，需读取 content 生成 .pyi + with open(src_path, 'r', encoding='utf-8') as f: + content: str = f.read() + sig_content: str = PythonToStubConverter.convert(content, module_name) + with open(sig_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(sig_content) + _SaveToGlobalCache(sha1, 'pyi', sig_path) + _vlog().info(f" 生成签名: {rel_from_inc} -> {sha1}.pyi") + except Exception as e: + _vlog().error(f"处理 include 文件失败 {rel_from_inc}: {e}") + + def _collect_typedef_map(self) -> dict[str, ast.AST]: + typedef_map: dict[str, ast.AST] = {} + if not os.path.isdir(self.temp_dir): + return typedef_map + for fname in os.listdir(self.temp_dir): + if not fname.endswith('.pyi'): + continue + pyi_path: str = os.path.join(self.temp_dir, fname) + try: + with open(pyi_path, 'r', encoding='utf-8') as f: + content: str = f.read() + tree: ast.Module = ast.parse(content) + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.AnnAssign) and isinstance(node.target, ast.Name): + var_name: str = node.target.id + is_typedef: bool = False + if isinstance(node.annotation, ast.Attribute) and hasattr(node.annotation, 'attr') and node.annotation.attr == 'CTypedef': + is_typedef = True + elif isinstance(node.annotation, ast.Name) and node.annotation.id == 'CTypedef': + is_typedef = True + elif isinstance(node.annotation, ast.BinOp) and isinstance(node.annotation.left, ast.Attribute) and hasattr(node.annotation.left, 'attr') and node.annotation.left.attr == 'CTypedef': + is_typedef = True + if is_typedef and node.value and var_name not in typedef_map: + typedef_map[var_name] = node.value + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"收集 typedef 映射失败: {_e}", "Exception") + return typedef_map + + def _process_include_py_files_stub(self, struct_names: set[str], needed_includes: list[tuple[str, str, str]] | None = None, enum_names: set[str] | None = None, struct_sha1_map: dict[str, str] | None = None, exception_names: set[str] | None = None, class_def_map: dict[str, ast.ClassDef] | None = None) -> None: + include_py_files: list[tuple[str, str, str]] + if needed_includes is not None: + include_py_files = needed_includes + else: + include_py_files = self._collect_include_py_files() + if not include_py_files: + return + typedef_map: dict[str, ast.AST] = self._collect_typedef_map() + for src_path, rel_from_inc, includes_dir in include_py_files: + ModulePath: str = rel_from_inc.replace(os.sep, '.').replace('/', '.') + module_name: str = os.path.splitext(ModulePath)[0] + try: + # 增量优化：用 mtime+size 获取 sha1，未变更文件跳过 open+read + sha1: str = self._get_sha1(src_path) + if not sha1: + continue + + stub_path: str = os.path.join(self.temp_dir, f"{sha1}.stub.ll") + if os.path.isfile(stub_path): + _vlog().info(f" 缓存命中: {rel_from_inc} -> {sha1}.stub.ll") + continue + + # 全局缓存检查（不被 --clean 清除） + if _TryGlobalCache(sha1, 'stub.ll', stub_path): + _vlog().info(f" 缓存命中(全局): {rel_from_inc} -> {sha1}.stub.ll") + continue + + sig_path: str = os.path.join(self.temp_dir, f"{sha1}.pyi") + if not os.path.isfile(sig_path): + _TryGlobalCache(sha1, 'pyi', sig_path) + with open(sig_path, 'r', encoding='utf-8') as f: + sig_content: str = f.read() + + self._generate_decl_ll(sig_content, stub_path, src_path, struct_names, enum_names=enum_names, module_sha1=sha1, struct_sha1_map=struct_sha1_map, exception_names=exception_names, typedef_map=typedef_map, class_def_map=class_def_map) + _SaveToGlobalCache(sha1, 'stub.ll', stub_path) + _vlog().info(f" 生成声明: {rel_from_inc} -> {sha1}.stub.ll") + except Exception as e: + _vlog().error(f"处理 include 文件失败 {rel_from_inc}: {e}") + + def _extract_docstrings(self, content: str) -> dict[str, str]: + """从源码 AST 提取所有函数/结构体/方法的首条裸字符串字面量 docstring。 + + - 顶层 FunctionDef: key = 函数名 + - 顶层 ClassDef: key = 类名；同时遍历方法，key = "ClassName.method" + - docstring 必须是 body[0] 为 ast.Expr + ast.Constant(str) + - 空字符串不计入 + """ + result: dict[str, str] = {} + try: + tree = ast.parse(content) + except SyntaxError: + return result + for node in tree.body: + if isinstance(node, ast.FunctionDef): + doc = self._get_first_docstring(node.body) + if doc is not None and doc != '': + result[node.name] = doc + elif isinstance(node, ast.ClassDef): + doc = self._get_first_docstring(node.body) + if doc is not None and doc != '': + result[node.name] = doc + for sub in node.body: + if isinstance(sub, ast.FunctionDef): + m_doc = self._get_first_docstring(sub.body) + if m_doc is not None and m_doc != '': + result[f"{node.name}.{sub.name}"] = m_doc + return result + + def _get_first_docstring(self, body: list[ast.stmt]) -> str | None: + """返回 body 首条裸字符串字面量，否则 None""" + if body and isinstance(body[0], ast.Expr) and isinstance(body[0].value, ast.Constant) and isinstance(body[0].value.value, str): + return body[0].value.value + return None + + def _process_file_pyi(self, src_path: str, rel_path: str) -> None: + # 增量优化：先用 mtime+size 获取 sha1，未变更文件跳过 open+read + sha1: str = self._get_sha1(src_path) + if not sha1: + return + self.sha1_map[sha1] = rel_path + + sig_path: str = os.path.join(self.temp_dir, f"{sha1}.pyi") + if os.path.isfile(sig_path): + _vlog().info(f" -> {sha1}.pyi (缓存)") + # 缓存命中时补全缺失的 .doc.json + doc_path: str = os.path.join(self.temp_dir, f"{sha1}.doc.json") + if not os.path.isfile(doc_path): + if not _TryGlobalCache(sha1, 'doc.json', doc_path): + # doc.json 缺失时需读 content 提取 docstring + with open(src_path, 'r', encoding='utf-8') as f: + content: str = f.read() + docs: dict[str, str] = self._extract_docstrings(content) + try: + with open(doc_path, 'w', encoding='utf-8', newline='\n') as f: + json.dump(docs, f, ensure_ascii=False) + _SaveToGlobalCache(sha1, 'doc.json', doc_path) + except Exception as e: + _vlog().warning(f"补写 {sha1}.doc.json 失败: {e}") + return + + # 全局缓存检查 + if _TryGlobalCache(sha1, 'pyi', sig_path): + _vlog().info(f" -> {sha1}.pyi (全局缓存)") + doc_path: str = os.path.join(self.temp_dir, f"{sha1}.doc.json") + if not os.path.isfile(doc_path): + _TryGlobalCache(sha1, 'doc.json', doc_path) + return + + # 缓存未命中，需读取 content 生成 .pyi + with open(src_path, 'r', encoding='utf-8') as f: + content = f.read() + + module_name: str = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + sig_content: str = PythonToStubConverter.convert(content, module_name) + with open(sig_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(sig_content) + _SaveToGlobalCache(sha1, 'pyi', sig_path) + _vlog().info(f" -> {sha1}.pyi (签名)") + + # 提取 docstring 写入 {sha1}.doc.json，供 Phase2 跨模块 __doc__ 加载 + docs: dict[str, str] = self._extract_docstrings(content) + doc_path: str = os.path.join(self.temp_dir, f"{sha1}.doc.json") + try: + with open(doc_path, 'w', encoding='utf-8', newline='\n') as f: + json.dump(docs, f, ensure_ascii=False) + _SaveToGlobalCache(sha1, 'doc.json', doc_path) + except Exception as e: + _vlog().warning(f"写入 {sha1}.doc.json 失败: {e}") + + def _process_file_stub(self, src_path: str, rel_path: str, struct_names: set[str], enum_names: set[str] | None = None, struct_sha1_map: dict[str, str] | None = None, exception_names: set[str] | None = None, class_def_map: dict[str, ast.ClassDef] | None = None) -> None: + # 增量优化：用 mtime+size 获取 sha1，未变更文件跳过 open+read + sha1: str = self._get_sha1(src_path) + if not sha1: + return + stub_path: str = os.path.join(self.temp_dir, f"{sha1}.stub.ll") + if os.path.isfile(stub_path): + _vlog().info(f" -> {sha1}.stub.ll (缓存)") + return + + sig_path: str = os.path.join(self.temp_dir, f"{sha1}.pyi") + with open(sig_path, 'r', encoding='utf-8') as f: + sig_content: str = f.read() + + typedef_map: dict[str, ast.AST] = self._typedef_map_cache or self._collect_typedef_map() + self._generate_decl_ll(sig_content, stub_path, src_path, struct_names, enum_names=enum_names, module_sha1=sha1, struct_sha1_map=struct_sha1_map, exception_names=exception_names, typedef_map=typedef_map, class_def_map=class_def_map) + _vlog().info(f" -> {sha1}.stub.ll (声明)") + + def _build_struct_registry(self) -> tuple[set[str], set[str], dict[str, str], set[str], dict[str, ast.ClassDef]]: + struct_names: set[str] = set() + enum_names: set[str] = set() + exception_names: set[str] = set() + struct_sha1_map: dict[str, str] = {} + class_def_map: dict[str, ast.ClassDef] = {} + valid_sha1_keys: set[str] = set(self.sha1_map.keys()) + for fname in os.listdir(self.temp_dir): + if not fname.endswith('.pyi'): + continue + sha1_key: str = fname.replace('.pyi', '') + if sha1_key not in valid_sha1_keys: + continue + pyi_path: str = os.path.join(self.temp_dir, fname) + try: + with open(pyi_path, 'r', encoding='utf-8') as f: + content: str = f.read() + tree: ast.Module = ast.parse(content) + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.ClassDef): + is_enum: bool = False + is_renum: bool = False + is_exception: bool = False + if node.bases: + for base in node.bases: + if isinstance(base, ast.Attribute) and hasattr(base, 'attr'): + if base.attr == 'REnum': + is_renum = True + break + elif base.attr in ('CEnum', 'Enum'): + is_enum = True + break + elif base.attr == 'Exception' or base.attr in exception_names: + is_exception = True + break + elif isinstance(base, ast.Name) and hasattr(base, 'id'): + if base.id == 'REnum': + is_renum = True + break + elif base.id in ('CEnum', 'Enum'): + is_enum = True + break + elif base.id == 'Exception' or base.id in exception_names: + is_exception = True + break + # 也检查装饰器形式 @t.CEnum / @t.REnum + if not is_enum and not is_renum and not is_exception and hasattr(node, 'decorator_list') and node.decorator_list: + for deco in node.decorator_list: + deco_attr: str | None = None + if isinstance(deco, ast.Attribute) and hasattr(deco, 'attr'): + deco_attr = deco.attr + elif isinstance(deco, ast.Name) and hasattr(deco, 'id'): + deco_attr = deco.id + if deco_attr == 'REnum': + is_renum = True + break + elif deco_attr in ('CEnum', 'Enum'): + is_enum = True + break + # REnum 是结构体类型（带 __tag + payload），需加入 struct_names 以便 + # DeclarationGenerator._get_type_str 返回 %Name* 而非 i32 + if is_renum: + struct_names.add(node.name) + struct_sha1_map[node.name] = sha1_key + elif is_enum: + enum_names.add(node.name) + elif is_exception: + exception_names.add(node.name) + else: + struct_names.add(node.name) + struct_sha1_map[node.name] = sha1_key + # 收集所有类（含枚举/异常）的 ClassDef 节点，供 DeclarationGenerator + # 跨模块字段展平使用（如 exprs.py 的 Name(AST) 需查找 base.py 的 AST） + class_def_map[node.name] = node + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"构建结构体/枚举注册表失败: {_e}", "Exception") + return struct_names, enum_names, struct_sha1_map, exception_names, class_def_map + + def _generate_decl_ll(self, pyi_content: str, ll_path: str, src_path: str, struct_names: set[str] | None = None, enum_names: set[str] | None = None, module_sha1: str | None = None, struct_sha1_map: dict[str, str] | None = None, exception_names: set[str] | None = None, typedef_map: dict[str, ast.AST] | None = None, class_def_map: dict[str, ast.ClassDef] | None = None) -> None: + try: + decl_gen: DeclarationGenerator = DeclarationGenerator(struct_names=struct_names, enum_names=enum_names, module_sha1=module_sha1, target_triple=self.target_triple, target_datalayout=self.target_datalayout, struct_sha1_map=struct_sha1_map, exception_names=exception_names, typedef_map=typedef_map, class_def_map=class_def_map) + decl_ll: str = decl_gen.generate(pyi_content, src_path) + with open(ll_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(decl_ll) + except Exception as e: + _vlog().warning(f".ll 声明生成失败: {e}") + with open(ll_path, 'w', encoding='utf-8') as f: + f.write(f"; declaration for {os.path.basename(src_path)}\n; error: {e}\n") diff --git a/lib/Projectrans/Phase2Translator.py b/lib/Projectrans/Phase2Translator.py new file mode 100644 index 0000000..7fa9b17 --- /dev/null +++ b/lib/Projectrans/Phase2Translator.py @@ -0,0 +1,3055 @@ +from __future__ import annotations + +import sys +import os +import re +import shutil +import subprocess +import ast +import traceback +import json +import pickle +import time +from concurrent.futures import ProcessPoolExecutor, ThreadPoolExecutor, as_completed, wait, FIRST_COMPLETED +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.includes import t +from lib.core.SymbolUtils import AnnotationContainsName +from lib.Projectrans.Utils import compute_sha1, _check_annotation_for_export, find_reachable_files_from_entries, topological_sort_files, parse_python_file +from lib.Projectrans.Phase1Generator import _TryGlobalCache, _SaveToGlobalCache, _GLOBAL_CACHE_DIR +import TransPyC +from StubGen import PythonToStubConverter + + +def _ExtractCDefineConstantsFromPyi(pyi_path: str, all_dc: dict[str, object]) -> None: + """从 .pyi 文件中提取 CDefine 常量到 all_dc。文件不存在时静默跳过。""" + if not os.path.exists(pyi_path): + return + try: + # 复用 parse_python_file 的 AST 缓存（.pyi 文件被多处解析） + _, tree = parse_python_file(pyi_path) + if tree is None: + return + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.AnnAssign) and isinstance(node.target, ast.Name): + if AnnotationContainsName(node.annotation, 'CDefine') and node.value: + val = None + if isinstance(node.value, ast.Constant): + val = node.value.value + elif isinstance(node.value, ast.Call) and isinstance(node.value.func, ast.Name) and node.value.func.id == 'float' and node.value.args and isinstance(node.value.args[0], ast.Constant): + val = node.value.args[0].value + if val is not None: + all_dc[f"{node.target.id}"] = val + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + + +def _parallel_translate_worker(src_path: str, out_path: str, src_root: str, temp_dir: str, output_dir: str, + include_dirs: list[str], triple: str | None, datalayout: str | None, + sha1_map: dict[str, str], sig_files: dict[str, str], + stub_files: dict[str, str], include_py_map: dict[str, str], slice_level: int, + shared_sym_pickle_path: str, struct_sha1_map: dict[str, str], + prebuilt_ModuleSha1Map: dict[str, str] | None = None) -> tuple[str, str, str]: + try: + trans = Phase2Translator(src_root, temp_dir, output_dir, + compile_cmd='llc', include_dirs=include_dirs, + target_triple=triple, target_datalayout=datalayout) + trans.sha1_map = sha1_map + trans.sig_files = sig_files + trans.stub_files = stub_files + trans.include_py_map = include_py_map + trans.slice_level = slice_level + trans.struct_sha1_map = struct_sha1_map + + if shared_sym_pickle_path and os.path.exists(shared_sym_pickle_path): + # 使用 pickle.load(f) 流式加载，避免 pickle.loads(f.read()) 一次性 + # 将整个 pickle 文件读入内存（磁盘副本 + 反序列化对象同时占用内存）导致 MemoryError。 + # 同时提高递归限制，防止深层嵌套对象反序列化失败。 + sys.setrecursionlimit(max(sys.getrecursionlimit(), 100000)) + with open(shared_sym_pickle_path, 'rb') as f: + shared_data = pickle.load(f) + trans._shared_symbol_table = shared_data['symbol_table'] + trans._shared_source_module_sig_files = shared_data['source_module_sig_files'] + trans._shared_all_dc = shared_data['all_dc'] + trans._shared_export_extern_funcs = shared_data['export_extern_funcs'] + trans.inline_func_symbols = shared_data['inline_func_symbols'] + trans.function_default_args = shared_data['function_default_args'] + trans.function_param_names = shared_data.get('function_param_names', {}) + trans._shared_generic_class_templates = shared_data.get('generic_class_templates', {}) + trans._shared_doc_meta = shared_data.get('shared_doc_meta', {}) + trans._shared_class_vtable = shared_data.get('class_vtable', set()) + trans._shared_cross_module_vtable = shared_data.get('cross_module_vtable', set()) + trans._shared_class_methods = shared_data.get('class_methods', {}) + trans._shared_cross_module_novtable = shared_data.get('cross_module_novtable', set()) + else: + trans._precollect_inline_symbols() + trans._build_shared_symbol_data() + + trans._translate_file(src_path, out_path, prebuilt_ModuleSha1Map=prebuilt_ModuleSha1Map) + return ('ok', src_path, '') + except Exception as e: + return ('error', src_path, f'{e}\n{traceback.format_exc()}') + + +def _count_braces_outside_strings(text: str) -> int: + """统计花括号差值，跳过 c"..." 字符串字面量内部的括号""" + count = 0 + i = 0 + while i < len(text): + if text[i:i+2] == 'c"' and (i == 0 or text[i-1] != '\\'): + # 跳过 c"..." 字面量 + i += 2 + while i < len(text): + if text[i] == '"' and (i == 0 or text[i-1] != '\\'): + i += 1 + break + i += 1 + continue + if text[i] == '{': + count += 1 + elif text[i] == '}': + count -= 1 + i += 1 + return count + + +class Phase2Translator: + """阶段二：使用声明接口翻译源文件""" + + INCLUDE_LIB_EXTENSIONS: tuple[str, ...] = ('.dll', '.ll', '.so', '.o', '.a', '.lib') + INCLUDE_SRC_EXTENSIONS: tuple[str, ...] = ('.c', '.cpp', '.cc', '.cxx', '.m', '.mm') + INCLUDE_DIRS: list[str] = [ + os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), 'includes'), + ] + + def __init__(self, src_root: str, temp_dir: str, output_dir: str, compile_cmd: str = 'llc', compile_flags: list[str] | None = None, linker_cmd: str | None = None, linker_flags: list[str] | None = None, linker_output: str | None = None, include_dirs: list[str] | None = None, entry_files: list[str] | None = None, target_triple: str | None = None, target_datalayout: str | None = None, force_recompile_includes: bool = False) -> None: + self.src_root = os.path.abspath(src_root) + self.temp_dir = os.path.abspath(temp_dir) + self.output_dir = os.path.abspath(output_dir) + self.compile_cmd = compile_cmd + self.compile_flags = compile_flags or ['-filetype=obj'] + self.triple = target_triple + self.datalayout = target_datalayout + if not self.triple: + for i, flag in enumerate(self.compile_flags): + if flag.startswith('-mtriple='): + self.triple = flag[len('-mtriple='):] + elif flag == '-mtriple' and i + 1 < len(self.compile_flags): + self.triple = self.compile_flags[i + 1] + self.linker_cmd = linker_cmd + self.linker_flags = linker_flags or [] + self.linker_output = linker_output + self.slice_level = 3 + self.sha1_map: dict[str, str] = {} + self.sig_files: dict[str, str] = {} + self.stub_files: dict[str, str] = {} + self.include_py_map: dict[str, str] = {} + self.used_includes: set[str] = set() + self._last_ErrorStack: list | None = None + self.function_default_args: dict[str, list] = {} + self.function_param_names: dict[str, list] = {} + self.inline_func_symbols: dict[str, CTypeInfo] = {} + self.extra_link_files: list[str] = [] + self.extra_compile_files: list[str] = [] + self.extra_py_files: list[str] = [] + self._include_sha1s: set[str] = set() + # 持有 ProcessPoolExecutor 引用，避免局部变量 GC 时 finalizer 阻塞 ~99s + # （join_executor_internals 在 GC 时同步等待 worker 退出队列） + self._executor_pool: ProcessPoolExecutor | None = None + self.entry_files: list[str] | None = entry_files + self.force_recompile_includes: bool = force_recompile_includes + self._shared_symbol_table: object | None = None + self._shared_source_module_sig_files: dict[str, str] = {} + self._shared_all_dc: dict[str, dict] = {} + self._shared_export_extern_funcs: dict[str, str] = {} + self.include_dirs: list[str] = [] + self.struct_sha1_map: dict[str, str] = {} + self._shared_generic_class_templates: dict[str, dict] = {} + self._shared_doc_meta: dict[str, dict[str, str]] = {} + # 共享 vtable 状态跨模块（修复 pool.alloc(48) 虚表分派失败导致段错误） + self._shared_class_vtable: set[str] = set() + self._shared_cross_module_vtable: set[str] = set() + self._shared_class_methods: dict[str, list[str]] = {} + self._shared_cross_module_novtable: set[str] = set() + default_dirs = [d for d in self.INCLUDE_DIRS if os.path.isdir(d)] + if include_dirs: + seen = set() + self.include_dirs = [] + for d in list(include_dirs) + default_dirs: + d = os.path.abspath(d) + if d not in seen and os.path.isdir(d): + self.include_dirs.append(d) + seen.add(d) + else: + self.include_dirs = default_dirs + self._LoadSha1Map() + + def _LoadSha1Map(self) -> None: + """从 temp_dir 加载 SHA1 映射、签名文件和 stub 声明文件列表。 + temp_dir 中缺失的文件会从全局缓存补充。 + """ + if not os.path.exists(self.temp_dir): + _vlog().error(f"声明目录不存在: {self.temp_dir}") + return + + for fname in os.listdir(self.temp_dir): + fpath = os.path.join(self.temp_dir, fname) + if fname.startswith('_'): + continue + if fname.endswith('.pyi'): + sha1 = fname[:-4] + self.sig_files[sha1] = fpath + elif fname.endswith('.stub.ll'): + sha1 = fname[:-8] + self.stub_files[sha1] = fpath + + # 从全局缓存补充 temp_dir 中缺失的 .pyi / .stub.ll + if os.path.isdir(_GLOBAL_CACHE_DIR): + for fname in os.listdir(_GLOBAL_CACHE_DIR): + if fname.startswith('_'): + continue + if fname.endswith('.pyi'): + sha1 = fname[:-4] + if sha1 not in self.sig_files: + self.sig_files[sha1] = os.path.join(_GLOBAL_CACHE_DIR, fname) + elif fname.endswith('.stub.ll'): + sha1 = fname[:-8] + if sha1 not in self.stub_files: + self.stub_files[sha1] = os.path.join(_GLOBAL_CACHE_DIR, fname) + + sha1_file_path = os.path.join(self.temp_dir, '_sha1_map.txt') + if os.path.exists(sha1_file_path): + with open(sha1_file_path, 'r', encoding='utf-8') as f: + for line in f: + line = line.strip() + if ':' in line: + sha1, rel = line.split(':', 1) + self.sha1_map[sha1] = rel + if rel.startswith('includes/'): + module_name = os.path.splitext(os.path.basename(rel))[0] + self.include_py_map[module_name] = sha1 + else: + for sha1, stub_path in self.stub_files.items(): + self.sha1_map[sha1] = sha1 + + def _build_struct_sha1_map(self) -> dict[str, str]: + struct_sha1_map = {} + valid_sha1_keys = set(self.sha1_map.keys()) + # 收集 temp_dir 和全局缓存中的 .pyi 文件 + pyi_dirs: list[str] = [self.temp_dir] + if os.path.isdir(_GLOBAL_CACHE_DIR): + pyi_dirs.append(_GLOBAL_CACHE_DIR) + for scan_dir in pyi_dirs: + if not os.path.isdir(scan_dir): + continue + for fname in os.listdir(scan_dir): + if not fname.endswith('.pyi'): + continue + sha1_key = fname.replace('.pyi', '') + if sha1_key not in valid_sha1_keys: + continue + if sha1_key in struct_sha1_map: + continue + pyi_path = os.path.join(scan_dir, fname) + # 复用 parse_python_file 的 AST 缓存（.pyi 文件可能被多处解析） + _, tree = parse_python_file(pyi_path) + if tree is None: + continue + for node in ast.walk(tree): + if isinstance(node, ast.ClassDef): + struct_sha1_map[node.name] = sha1_key + return struct_sha1_map + + def run(self) -> None: + """扫描源目录，翻译并生成 .ll 含代码文件（导入遍历）""" + import sys as _sys + if self.entry_files: + py_files = list(self.entry_files) + else: + main_py = os.path.join(self.src_root, 'main.py') + if os.path.exists(main_py): + py_files = [main_py] + else: + py_files = [] + for root, dirs, files in os.walk(self.src_root): + dirs[:] = [d for d in dirs if d not in ('__pycache__',)] + for file in files: + if file.endswith('.py'): + py_files.append(os.path.join(root, file)) + + if not py_files: + _vlog().info("未找到入口文件或源文件") + return + + reachable = find_reachable_files_from_entries(self.src_root, py_files) + py_files = list(reachable) + + py_files = topological_sort_files(py_files, self.src_root) + + _vlog().info(f"找到 {len(py_files)} 个可达源文件（已按依赖排序）") + _vlog().info("处理顺序:") + for i, f in enumerate(py_files[:10], 1): + rel = os.path.relpath(f, self.src_root) + _vlog().info(f" {i}. {rel}") + os.makedirs(self.output_dir, exist_ok=True) + + valid_sha1s = set(self.sha1_map.keys()) + old_stub_count = len(self.stub_files) + old_sig_count = len(self.sig_files) + self.stub_files = {k: v for k, v in self.stub_files.items() if k in valid_sha1s} + self.sig_files = {k: v for k, v in self.sig_files.items() if k in valid_sha1s} + if old_stub_count != len(self.stub_files) or old_sig_count != len(self.sig_files): + _vlog().info(f"过滤旧文件: stub {old_stub_count}->{len(self.stub_files)}, sig {old_sig_count}->{len(self.sig_files)}") + + active_sha1s = set() + self._precollect_inline_symbols() + self._build_shared_symbol_data() + self._precollect_vtable_state() + self.struct_sha1_map = self._build_struct_sha1_map() + + need_translate = [] + for i, src_path in enumerate(py_files, 1): + rel = os.path.relpath(src_path, self.src_root) + # 复用 parse_python_file 的 AST 缓存（get_file_dependencies 已 parse 过） + content, tree = parse_python_file(src_path) + sha1 = compute_sha1(content) + active_sha1s.add(sha1) + out_path = os.path.join(self.output_dir, f"{sha1}.ll") + + current_ModuleSha1Map = self._build_current_ModuleSha1Map(sha1) + + if tree is not None: + try: + for node in ast.walk(tree): + if isinstance(node, ast.Import): + for alias in node.names: + self.used_includes.add(alias.name.split('.')[0]) + elif isinstance(node, ast.ImportFrom): + if node.module: + self.used_includes.add(node.module.split('.')[0]) + # 检测 list/dict 容器使用，自动添加 _list/_dict/json 到 used_includes + # 以便 _scan_include_libraries 能发现并编译这些库文件 + if isinstance(node, ast.Subscript) and isinstance(node.value, ast.Name): + if node.value.id == 'dict': + self.used_includes.add('_dict') + self.used_includes.add('json') + elif node.value.id == 'list': + self.used_includes.add('_list') + if isinstance(node, ast.Call) and isinstance(node.func, ast.Name): + if node.func.id == 'dict': + self.used_includes.add('_dict') + self.used_includes.add('json') + elif node.func.id == 'list': + self.used_includes.add('_list') + # 检测类型注解 d: dict / l: list (非泛型形式) + if isinstance(node, ast.AnnAssign) and isinstance(node.annotation, ast.Name): + if node.annotation.id == 'dict': + self.used_includes.add('_dict') + self.used_includes.add('json') + elif node.annotation.id == 'list': + self.used_includes.add('_list') + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"收集源文件导入的 include 模块失败: {_e}", "Exception") + + if os.path.isfile(out_path): + deps_changed = self._check_deps_changed(sha1, current_ModuleSha1Map) + if not deps_changed: + _vlog().info(f"[{i}/{len(py_files)}] 跳过(缓存): {rel} -> {sha1}.ll") + continue + else: + if self._recombine_ll(sha1, current_ModuleSha1Map): + _vlog().info(f"[{i}/{len(py_files)}] 重组(依赖变化): {rel} -> {sha1}.ll") + continue + _vlog().info(f"[{i}/{len(py_files)}] 重译(依赖变化): {rel} -> {sha1}.ll") + + else: + stub_cache = os.path.join(self.output_dir, f"{sha1}.stub.ll") + text_cache = os.path.join(self.output_dir, f"{sha1}.text.ll") + if os.path.isfile(stub_cache) and os.path.isfile(text_cache): + if self._recombine_ll(sha1, current_ModuleSha1Map): + _vlog().info(f"[{i}/{len(py_files)}] 重组(缓存恢复): {rel} -> {sha1}.ll") + continue + + need_translate.append((i, src_path, out_path, sha1, current_ModuleSha1Map)) + + if need_translate: + # 提前从 include 文件源码中提取函数参数名，供主项目编译时 keyword 参数匹配使用 + self._extract_include_param_names_from_dirs() + # 提前从 App 源文件中提取函数默认参数，供跨模块调用参数检查使用 + # （避免并行编译时，被调用函数的文件尚未编译完，导致默认参数信息缺失） + self._extract_app_default_args(need_translate) + + _vlog().info(f"\n需要翻译 {len(need_translate)} 个文件") + t_start = time.time() + + n_workers = min(8, len(need_translate)) + + if n_workers > 1 and len(need_translate) > 1: + _vlog().info(f" 并行翻译 (workers={n_workers})") + + shared_pickle_path = os.path.join(self.temp_dir, '_shared_sym.json') + try: + shared_data = { + 'symbol_table': self._shared_symbol_table, + 'source_module_sig_files': self._shared_source_module_sig_files, + 'all_dc': self._shared_all_dc, + 'export_extern_funcs': self._shared_export_extern_funcs, + 'inline_func_symbols': self.inline_func_symbols, + 'function_default_args': self.function_default_args, + 'function_param_names': self.function_param_names, + 'generic_class_templates': self._shared_generic_class_templates, + 'shared_doc_meta': self._shared_doc_meta, + 'class_vtable': self._shared_class_vtable, + 'cross_module_vtable': self._shared_cross_module_vtable, + 'class_methods': self._shared_class_methods, + 'cross_module_novtable': self._shared_cross_module_novtable, + } + with open(shared_pickle_path, 'wb') as f: + f.write(pickle.dumps(shared_data)) + _vlog().info(f" 共享符号表已序列化 ({os.path.getsize(shared_pickle_path)//1024}KB)") + except Exception as e: + _vlog().warning(f"符号表序列化失败({e})，worker将自行构建") + shared_pickle_path = '' + + errors = [] + # 复用 self._executor_pool：避免局部变量 GC 时 finalizer 阻塞 ~99s + # （join_executor_internals 在 GC 时同步等待 worker 退出队列） + if self._executor_pool is None: + self._executor_pool = ProcessPoolExecutor(max_workers=n_workers) + executor = self._executor_pool + try: + futures = {} + for i, src_path, out_path, sha1, msm in need_translate: + rel = os.path.relpath(src_path, self.src_root) + future = executor.submit( + _parallel_translate_worker, + src_path, out_path, + self.src_root, self.temp_dir, self.output_dir, + self.include_dirs, self.triple, self.datalayout, + self.sha1_map, self.sig_files, self.stub_files, + self.include_py_map, self.slice_level, + shared_pickle_path, self.struct_sha1_map, + prebuilt_ModuleSha1Map=msm + ) + futures[future] = (i, rel) + + done_count = 0 + for future in as_completed(futures): + i, rel = futures[future] + done_count += 1 + try: + status, _, err = future.result() + if status == 'ok': + _vlog().info(f" [{done_count}/{len(need_translate)}] 完成: {rel}") + else: + _vlog().error(f" {rel}: {err}") + errors.append((rel, err)) + except Exception as e: + _vlog().error(f" {rel}: {e}") + errors.append((rel, str(e))) + except Exception: + # 异常时立即清理，正常流程不 shutdown（留给 includes 阶段复用） + if self._executor_pool is not None: + self._executor_pool.shutdown(wait=False, cancel_futures=True) + self._executor_pool = None + raise + + if errors: + _vlog().error(f"\n{len(errors)} 个文件翻译失败，继续编译 includes") + else: + for idx, (i, src_path, out_path, sha1, msm) in enumerate(need_translate): + rel = os.path.relpath(src_path, self.src_root) + t0 = time.time() + try: + self._translate_file(src_path, out_path, prebuilt_ModuleSha1Map=msm) + t1 = time.time() + _vlog().info(f" [{idx+1}/{len(need_translate)}] {rel} ({t1-t0:.1f}s)") + except Exception as e: + _vlog().error(f" {rel}: {e}") + traceback.print_exc() + t_end = time.time() + _vlog().info(f" 翻译总耗时: {t_end-t_start:.1f}s") + + _vlog().info(f"\n[阶段二完成] .ll 文件生成到: {self.output_dir}") + self._scan_include_libraries() + self._compile_include_py_files(active_sha1s) + self._compile_ll_files(active_sha1s) + self._compile_include_sources() + if self.linker_cmd: + self._link_obj_files(active_sha1s) + # 清理 ProcessPoolExecutor：所有翻译已完成，shutdown(wait=False) 让 worker + # 后台退出。避免 GC finalizer 在程序末尾阻塞 ~99s（join_executor_internals） + if self._executor_pool is not None: + self._executor_pool.shutdown(wait=False, cancel_futures=True) + self._executor_pool = None + + def _build_current_ModuleSha1Map(self, self_sha1: str | None) -> dict[str, str]: + """构建当前模块的依赖 SHA1 映射（与 _translate_file 中逻辑一致）""" + ModuleSha1Map = {} + for sha1_key, sig_path in self.sig_files.items(): + if sha1_key == self_sha1: + continue + rel_path = self.sha1_map.get(sha1_key, sha1_key) + if rel_path.startswith('includes/'): + inc_mod_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + inc_mod_name = inc_mod_name[len('includes/'):] + inc_short_name = inc_mod_name.split('.')[-1] if '.' in inc_mod_name else inc_mod_name + actual_sha1 = sha1_key + for inc_dir in self.include_dirs: + if os.path.isdir(inc_dir): + py_path = os.path.join(inc_dir, rel_path[len('includes/'):].replace(os.sep, '/')) + py_path = os.path.splitext(py_path)[0] + '.py' + if os.path.isfile(py_path): + with open(py_path, 'r', encoding='utf-8') as f: + py_sha1 = compute_sha1(f.read()) + actual_sha1 = py_sha1 + break + ModuleSha1Map[inc_mod_name] = actual_sha1 + ModuleSha1Map[inc_short_name] = actual_sha1 + # Also add parent package name (e.g., 'hashlib' for 'hashlib.__init__') + if inc_short_name == '__init__' and '.' in inc_mod_name: + parent_pkg = inc_mod_name.rsplit('.', 1)[0] + ModuleSha1Map[parent_pkg] = actual_sha1 + continue + module_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + ModuleSha1Map[module_name] = sha1_key + short_name = module_name.split('.')[-1] if '.' in module_name else module_name + ModuleSha1Map[short_name] = sha1_key + # 包入口文件：注册父包名（如 'os.__init__' -> 'os'），使 import os 能解析到包级符号 + if short_name == '__init__' and '.' in module_name: + parent_pkg = module_name.rsplit('.', 1)[0] + ModuleSha1Map[parent_pkg] = sha1_key + return ModuleSha1Map + + @staticmethod + def _extract_llvm_type_name(line: str) -> str | None: + """从 LLVM IR 类型定义行提取类型名（字符串操作，不使用正则） + + 支持 list[str]、dict[str, int] 等含特殊字符的类型名。 + 例如: + %"18e6990aa81cc5b6.list[str]" = type { ... } → "18e6990aa81cc5b6.list[str]" + %struct.foo = type { ... } → "struct.foo" + """ + stripped = line.strip() + if not stripped.startswith('%'): + return None + # 处理带引号的情况: %"name" = type ... + if stripped.startswith('%"'): + end_quote = stripped.find('"', 2) + if end_quote > 0: + return stripped[2:end_quote] + return None + # 处理不带引号的情况: %name = type ... + eq_pos = stripped.find('=') + if eq_pos > 1: + return stripped[1:eq_pos].strip() + return None + + @staticmethod + def _find_all_type_refs(text: str) -> set[str]: + """查找文本中所有 %"..."/%name 类型引用（字符串操作，不使用正则） + + 支持 list[str]、dict[str, int] 等含特殊字符的类型名。 + """ + refs: set[str] = set() + i: int = 0 + n: int = len(text) + while i < n: + if text[i] == '%': + if i + 1 < n and text[i + 1] == '"': + # 带引号: %"name" + end: int = text.find('"', i + 2) + if end > 0: + name: str = text[i + 2:end] + if name: + refs.add(name) + i = end + 1 + continue + else: + # 不带引号: %name (仅匹配字母、数字、下划线、点) + j: int = i + 1 + while j < n and (text[j].isalnum() or text[j] in '._'): + j += 1 + if j > i + 1: + refs.add(text[i + 1:j]) + i = j + continue + i += 1 + return refs + + @staticmethod + def _split_ll(ll_content: str) -> tuple[str, str]: + """将 .ll 内容拆分为 stub（声明）和 text（代码）两部分 + + stub: target, 注释, %type = type, declare, @xxx = external global + text: define ... { ... }, @xxx = global/constant (带初始化器) + """ + stub_lines = [] + text_lines = [] + in_define = False + brace_depth = 0 + module_sha1 = None + defined_names = set() + + for line in ll_content.splitlines(True): + stripped = line.strip() + + if stripped.startswith('define '): + dm = re.match(r'define\s+[^@]*@"?([a-f0-9]+)\.', stripped) + if dm and module_sha1 is None: + module_sha1 = dm.group(1) + dm2 = re.match(r'define\s+[^@]*@"?([^"\s(]+)"?', stripped) + if dm2: + defined_names.add(dm2.group(1)) + + in_global_def = False + global_var_name = None + global_type_parts = [] + global_brace_depth = 0 + + for line in ll_content.splitlines(True): + stripped = line.strip() + + if in_global_def: + text_lines.append(line) + global_type_parts.append(stripped) + global_brace_depth += _count_braces_outside_strings(stripped) + if global_brace_depth <= 0: + full_type = ' '.join(global_type_parts).strip() + depth = 0 + type_end = len(full_type) + for ci, cc in enumerate(full_type): + if cc in ('{', '['): + depth += 1 + elif cc in ('}', ']'): + depth -= 1 + elif depth == 0 and cc == ' ': + type_end = ci + break + var_type = full_type[:type_end].strip() + if var_type: + stub_lines.append(f'{global_var_name} = external global {var_type}\n') + in_global_def = False + global_var_name = None + global_type_parts = [] + global_brace_depth = 0 + continue + + if in_define: + text_lines.append(line) + brace_depth += _count_braces_outside_strings(stripped) + if brace_depth <= 0: + in_define = False + continue + + if stripped.startswith('define '): + text_lines.append(line) + in_define = True + brace_depth = _count_braces_outside_strings(stripped) + decl_line = re.sub(r'^define\s+', 'declare ', stripped) + decl_line = re.sub(r'\b(linkonce_odr|weak_odr|linkonce|weak|common|appending|internal|private)\s+', '', decl_line) + decl_line = re.sub(r'\balwaysinline\s+', '', decl_line) + paren_pos = decl_line.rfind(')') + if paren_pos >= 0: + decl_line = decl_line[:paren_pos + 1] + stub_lines.append(decl_line + '\n') + continue + + if stripped == '{' and text_lines and not in_define: + prev_stripped = text_lines[-1].strip() if text_lines else '' + if prev_stripped.startswith('define ') or prev_stripped.endswith('noredzone') or prev_stripped.endswith(')'): + text_lines.append(line) + brace_depth = 1 + in_define = True + if prev_stripped.startswith('define '): + decl_line = re.sub(r'^define\s+', 'declare ', prev_stripped) + decl_line = re.sub(r'\b(linkonce_odr|weak_odr|linkonce|weak|common|appending|internal|private)\s+', '', decl_line) + decl_line = re.sub(r'\balwaysinline\s+', '', decl_line) + paren_pos = decl_line.rfind(')') + if paren_pos >= 0: + decl_line = decl_line[:paren_pos + 1] + stub_lines.append(decl_line + '\n') + continue + + if stripped.startswith('@') and ' global ' in stripped and 'external global' not in stripped: + text_lines.append(line) + if ' internal ' not in stripped and ' private ' not in stripped: + name_match = re.match(r'(@\S+)', stripped) + if name_match: + var_name = name_match.group(1) + global_pos = stripped.find(' global ') + after_global = stripped[global_pos + 8:] + depth = 0 + type_end = len(after_global) + for ci, cc in enumerate(after_global): + if cc in ('{', '['): + depth += 1 + elif cc in ('}', ']'): + depth -= 1 + elif depth == 0 and cc == ' ': + type_end = ci + break + var_type = after_global[:type_end].strip() + brace_depth_gv = 0 + for cc in var_type: + if cc == '{': + brace_depth_gv += 1 + elif cc == '}': + brace_depth_gv -= 1 + if brace_depth_gv > 0: + in_global_def = True + global_var_name = var_name + global_type_parts = [var_type] + global_brace_depth = brace_depth_gv + elif var_type: + stub_lines.append(f'{var_name} = external global {var_type}\n') + continue + if stripped.startswith('@') and ' constant ' in stripped: + text_lines.append(line) + continue + + if stripped.startswith('declare'): + dm = re.match(r'declare\s+[^@]*@"?([^"\s(]+)"?', stripped) + if dm: + fname = dm.group(1) + if fname in defined_names: + stub_lines.append(line) + continue + dot_pos = fname.find('.') + if dot_pos > 0: + fname_sha1 = fname[:dot_pos] + if module_sha1 and fname_sha1 != module_sha1 and len(fname_sha1) >= 16: + continue + if re.match(r'^[a-f0-9]{16}\.', fname): + continue + stub_lines.append(line) + continue + + stub_lines.append(line) + + return ''.join(stub_lines), ''.join(text_lines) + + def _save_deps(self, sha1: str, ModuleSha1Map: dict[str, str]) -> None: + """保存依赖指纹到 .deps.json""" + deps_path = os.path.join(self.output_dir, f"{sha1}.deps.json") + deps = {} + for mod_name, dep_sha1 in ModuleSha1Map.items(): + deps[mod_name] = dep_sha1 + with open(deps_path, 'w', encoding='utf-8') as f: + json.dump(deps, f) + + def _LoadDeps(self, sha1: str) -> dict[str, str] | None: + """加载依赖指纹""" + deps_path = os.path.join(self.output_dir, f"{sha1}.deps.json") + if os.path.isfile(deps_path): + try: + with open(deps_path, 'r', encoding='utf-8') as f: + return json.load(f) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"加载依赖指纹 JSON 失败: {_e}", "Exception") + return None + + def _check_deps_changed(self, sha1: str, current_ModuleSha1Map: dict[str, str]) -> bool: + """检查依赖指纹是否变化""" + saved_deps = self._LoadDeps(sha1) + if saved_deps is None: + return True + for mod_name, dep_sha1 in current_ModuleSha1Map.items(): + if saved_deps.get(mod_name) != dep_sha1: + return True + for mod_name, dep_sha1 in saved_deps.items(): + if current_ModuleSha1Map.get(mod_name) != dep_sha1: + return True + return False + + def _recombine_ll(self, sha1: str, current_ModuleSha1Map: dict[str, str]) -> bool: + """依赖变化时重新组合 .ll：更新 .stub.ll 中的 SHA1 前缀，拼接 .stub.ll + .text.ll""" + saved_deps = self._LoadDeps(sha1) + if saved_deps is None: + return False + + stub_path = os.path.join(self.output_dir, f"{sha1}.stub.ll") + text_path = os.path.join(self.output_dir, f"{sha1}.text.ll") + ll_path = os.path.join(self.output_dir, f"{sha1}.ll") + + if not os.path.isfile(stub_path) or not os.path.isfile(text_path): + return False + + with open(stub_path, 'r', encoding='utf-8') as f: + stub_content = f.read() + with open(text_path, 'r', encoding='utf-8') as f: + text_content = f.read() + + sha1_replacements = {} + for mod_name, old_sha1 in saved_deps.items(): + new_sha1 = current_ModuleSha1Map.get(mod_name) + if new_sha1 and old_sha1 != new_sha1: + sha1_replacements[old_sha1] = new_sha1 + + if sha1_replacements: + for old_sha1, new_sha1 in sha1_replacements.items(): + stub_content = stub_content.replace(old_sha1, new_sha1) + text_content = text_content.replace(old_sha1, new_sha1) + + combined = stub_content + if not combined.endswith('\n'): + combined += '\n' + combined += text_content + + with open(ll_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(combined) + with open(stub_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(stub_content) + with open(text_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(text_content) + + self._save_deps(sha1, current_ModuleSha1Map) + + obj_path = os.path.join(self.output_dir, f"{sha1}.obj") + if os.path.isfile(obj_path): + os.remove(obj_path) + + return True + + def _inject_auto_imports(self, code: str, src_path: str) -> str: + """检测源码中对 list/dict 容器的使用，自动注入 import _list / import _dict。 + 当使用 dict 时，同时注入 import json (用于 dict 的 JSON 处理功能)。 + + 检测模式: + - list[T] / dict[V] 作为类型注解 (ast.Subscript, value=Name(id='list'/'dict')) + - list[T](...) / dict[V](...) 作为构造函数调用 (ast.Call, func=Subscript) + - list(...) / dict(...) 作为构造函数调用 (ast.Call, func=Name) + """ + try: + tree = ast.parse(code) + except SyntaxError: + return code + + # 检查是否已经导入了 _list / _dict / json + has_list_import: bool = False + has_dict_import: bool = False + has_json_import: bool = False + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.Import): + for alias in node.names: + if alias.name == '_list': + has_list_import = True + elif alias.name == '_dict': + has_dict_import = True + elif alias.name == 'json': + has_json_import = True + elif isinstance(node, ast.ImportFrom): + if node.module == '_list': + has_list_import = True + elif node.module == '_dict': + has_dict_import = True + elif node.module == 'json': + has_json_import = True + if has_list_import and has_dict_import and has_json_import: + break + + # 检查是否使用了 list / dict 容器 + uses_list: bool = False + uses_dict: bool = False + for node in ast.walk(tree): + if isinstance(node, ast.Subscript): + if isinstance(node.value, ast.Name): + if node.value.id == 'list': + uses_list = True + elif node.value.id == 'dict': + uses_dict = True + if isinstance(node, ast.Call) and isinstance(node.func, ast.Name): + if node.func.id == 'list': + uses_list = True + elif node.func.id == 'dict': + uses_dict = True + # 检测类型注解 d: dict / l: list (非泛型形式) + if isinstance(node, ast.AnnAssign) and isinstance(node.annotation, ast.Name): + if node.annotation.id == 'dict': + uses_dict = True + elif node.annotation.id == 'list': + uses_list = True + + # 注入 import 到源码开头 + prefix: str = '' + if uses_list and not has_list_import: + prefix += 'import _list\n' + if uses_dict and not has_dict_import: + prefix += 'import _dict\n' + if uses_dict and not has_json_import: + prefix += 'import json\n' + if prefix: + return prefix + code + return code + + def _translate_file(self, src_path: str, out_path: str, prebuilt_ModuleSha1Map: dict[str, str] | None = None) -> None: + """翻译单个源文件，嵌入相关 .stub.ll 声明""" + with open(src_path, 'r', encoding='utf-8') as f: + code = f.read() + + # SHA1 必须在自动导入注入之前计算，以保持与 Phase1 生成的文件名/函数名前缀一致 + sha1 = compute_sha1(code) + + # 自动导入: 检测 list 容器使用，自动注入 import _list + code = self._inject_auto_imports(code, src_path) + + if prebuilt_ModuleSha1Map is not None: + ModuleSha1Map = prebuilt_ModuleSha1Map + else: + ModuleSha1Map = self._build_current_ModuleSha1Map(sha1) + + trans = TransPyC.TransPyC(code=code, triple=self.triple, datalayout=self.datalayout) + trans.translator.CurrentFile = src_path + trans.SliceLevel = self.slice_level + trans.translator.SliceLevel = self.slice_level + trans.translator.SliceCount = 0 + trans.translator.SliceInfos = [] + trans.translator.LlvmGen = None + trans.translator._module_sha1 = sha1 + trans.translator._ModuleSha1Map = ModuleSha1Map + trans.translator._struct_sha1_map = self.struct_sha1_map + trans.translator._global_function_default_args = self.function_default_args + trans.translator._global_function_param_names = self.function_param_names + trans.translator._temp_dir = self.temp_dir + trans.translator._shared_doc_meta = self._shared_doc_meta + # 传递共享 vtable 状态 + trans.translator._shared_class_vtable = self._shared_class_vtable + trans.translator._shared_cross_module_vtable = self._shared_cross_module_vtable + trans.translator._shared_class_methods = self._shared_class_methods + trans.translator._shared_cross_module_novtable = self._shared_cross_module_novtable + + if self._shared_symbol_table is not None: + trans.translator.SymbolTable = self._shared_symbol_table + # 确保共享符号表的 translator 指向当前 translator，以便 import 别名解析正确 + if hasattr(self._shared_symbol_table, 'translator'): + self._shared_symbol_table.translator = trans.translator + trans.translator._source_module_sig_files = self._shared_source_module_sig_files + trans.translator._all_define_constants = self._shared_all_dc + trans.translator._export_extern_funcs = self._shared_export_extern_funcs + if self._shared_generic_class_templates: + if not hasattr(trans.translator.ClassHandler, '_generic_class_templates'): + trans.translator.ClassHandler._generic_class_templates = {} + trans.translator.ClassHandler._generic_class_templates.update(self._shared_generic_class_templates) + else: + export_extern_funcs = {} + for sha1_key, sig_path in self.sig_files.items(): + if sha1_key == sha1: + continue + try: + with open(sig_path, 'r', encoding='utf-8') as f: + sig_content = f.read() + sig_tree = ast.parse(sig_content) + for node in ast.iter_child_nodes(sig_tree): + if isinstance(node, ast.FunctionDef): + if _check_annotation_for_export(node.returns): + export_extern_funcs[node.name] = sha1_key + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"收集导出外部函数声明失败: {_e}", "Exception") + if export_extern_funcs: + trans.translator._export_extern_funcs = export_extern_funcs + + for includes_dir in self.include_dirs: + if os.path.isdir(includes_dir): + for pyi_file in os.listdir(includes_dir): + if pyi_file.endswith('.pyi'): + pyi_path = os.path.join(includes_dir, pyi_file) + module_name = os.path.splitext(pyi_file)[0] + trans.translator.SymbolTable.LoadModuleSymbols(pyi_path, module_name, lineno=0) + for py_file in os.listdir(includes_dir): + if py_file.endswith('.py') and (not py_file.startswith('_') or py_file == '_list.py' or py_file == '_dict.py'): + module_name = os.path.splitext(py_file)[0] + sha1_key = self.include_py_map.get(module_name) + if sha1_key and sha1_key in self.sig_files: + sig_path = self.sig_files[sha1_key] + trans.translator.SymbolTable.LoadModuleSymbols(sig_path, module_name, lineno=0) + # 收集需要重导出的包，等所有模块符号加载完后再处理 + _pending_reexports = [] + for sha1_key, sig_path in self.sig_files.items(): + rel_path = self.sha1_map.get(sha1_key, sha1_key) + if rel_path.startswith('includes/'): + mod_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + mod_name = mod_name[len('includes/'):] + trans.translator.SymbolTable.LoadModuleSymbols(sig_path, mod_name, lineno=0) + if mod_name.endswith('.__init__'): + package_name = mod_name[:-len('.__init__')] + _pending_reexports.append((sig_path, package_name)) + + # 所有 includes 模块符号已加载，现在处理包重导出 + for sig_path, package_name in _pending_reexports: + self._register_package_reexports(trans, sig_path, package_name) + + for sha1_key, sig_path in self.sig_files.items(): + if sha1_key == sha1: + continue + rel_path = self.sha1_map.get(sha1_key, sha1_key) + if rel_path.startswith('includes/'): + continue + module_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + trans.translator.SymbolTable.LoadModuleSymbols(sig_path, module_name, lineno=0) + trans.translator._source_module_sig_files[module_name] = sig_path + short_name = module_name.split('.')[-1] if '.' in module_name else module_name + if short_name != module_name: + trans.translator._source_module_sig_files[short_name] = sig_path + + all_dc = {} + for sha1_key, stub_path in self.stub_files.items(): + if sha1_key == sha1: + continue + if os.path.exists(stub_path): + try: + with open(stub_path, 'r', encoding='utf-8') as f: + stub_content = f.read() + for line in stub_content.splitlines(): + line = line.strip() + if line.startswith('@') and '= global' in line and 'i32' in line and not line.startswith('@llvm'): + var_name = line.split('=')[0].strip().lstrip('@') + val_part = line.split('i32')[-1].strip().rstrip(']').lstrip('[').lstrip('i32') + try: + val = int(val_part) + all_dc[var_name] = val + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + for sha1_key, pyi_path in self.sig_files.items(): + if sha1_key == sha1: + continue + _ExtractCDefineConstantsFromPyi(pyi_path, all_dc) + if all_dc: + trans.translator._all_define_constants = all_dc + + for sym_name, sym_info in self.inline_func_symbols.items(): + if sym_name not in trans.translator.SymbolTable: + trans.translator.SymbolTable[sym_name] = sym_info + else: + existing = trans.translator.SymbolTable[sym_name] + if not getattr(existing, 'InlineBody', None) and getattr(sym_info, 'InlineBody', None): + existing.IsInline = sym_info.IsInline + existing.InlineBody = sym_info.InlineBody + existing.InlineParams = sym_info.InlineParams + + try: + result = trans.Convert( + OutputFilename=out_path, + SourceFilename=src_path, + target='llvm' + ) + except Exception as e: + error_stack = getattr(trans.translator, '_ErrorStack', []) + if error_stack: + self._last_ErrorStack = error_stack + chain_lines = [] + for entry in reversed(error_stack): + if len(entry) >= 3: + exc_msg, line_info = entry[1], entry[2] + chain_lines.append(" %s\n %s" % (line_info, exc_msg)) + if chain_lines: + enriched = str(e) + "\n" + "\n".join(chain_lines) + raise type(e)(enriched) from e + node_info = "" + if hasattr(trans.translator, 'LlvmGen') and trans.translator.LlvmGen: + node_info = trans.translator.LlvmGen._get_node_info() + if node_info: + enriched = "%s\n %s" % (str(e), node_info.strip()) + raise type(e)(enriched) from e + raise + + if result and isinstance(result, str): + for func_name, func_def in trans.translator.FunctionDefCache.items(): + if hasattr(func_def, 'args'): + mangled = trans.translator.LlvmGen._mangle_func_name(func_name) if trans.translator.LlvmGen else func_name + # 提取参数名 + if hasattr(func_def.args, 'args'): + param_names = [arg.arg for arg in func_def.args.args] + self.function_param_names[mangled] = param_names + self.function_param_names[func_name] = param_names + # 提取默认值 + if hasattr(func_def.args, 'defaults') and func_def.args.defaults: + defaults = [] + for d in func_def.args.defaults: + if isinstance(d, ast.Constant): + defaults.append(d.value) + else: + defaults.append(None) + self.function_default_args[mangled] = defaults + self.function_default_args[func_name] = defaults + for sym_name, sym_info in trans.translator.SymbolTable.items(): + if isinstance(sym_info, CTypeInfo) and getattr(sym_info, 'IsInline', False) and getattr(sym_info, 'InlineBody', None): + self.inline_func_symbols[sym_name] = sym_info + + out_dir = os.path.dirname(out_path) + if out_dir: + os.makedirs(out_dir, exist_ok=True) + # 修复 LLVM 22+: external global 不能有初值，改为 linkonce + result = re.sub(r'^(@[\w.]+)\s*=\s*external\s+global\s+(\w[\w*]*)\s+(-?\d+|null|zeroinitializer|true|false)$', r'\1 = linkonce global \2 \3', result, flags=re.MULTILINE) + # 后处理：添加 llvmlite 不支持的 LLVM 属性（如 willreturn, mustprogress） + pending_str_attrs = getattr(trans.translator.LlvmGen, '_pending_str_attrs', None) + if pending_str_attrs: + for func_name, attrs in pending_str_attrs.items(): + attrs_str = ' '.join(attrs) + # 匹配 define ... @func_name(...) ... { 并在 { 之前添加属性 + pattern = rf'(define\s+[^@]*@{re.escape(func_name)}\s*\([^)]*\))([^{{]*)(\{{)' + result, n = re.subn(pattern, rf'\1\2 {attrs_str}\3', result, count=1) + if n == 0: + _vlog().warning(f"后处理添加属性失败: 未找到函数 {func_name} 的定义") + with open(out_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(result) + + stub_content, text_content = self._split_ll(result) + base = os.path.splitext(out_path)[0] + with open(base + '.stub.ll', 'w', encoding='utf-8', newline='\n') as f: + f.write(stub_content) + with open(base + '.text.ll', 'w', encoding='utf-8', newline='\n') as f: + f.write(text_content) + self._save_deps(sha1, ModuleSha1Map) + + _vlog().info(f" -> {os.path.basename(out_path)}") + else: + stub_path = self.stub_files.get(sha1) + if stub_path and os.path.exists(stub_path): + out_dir = os.path.dirname(out_path) + if out_dir: + os.makedirs(out_dir, exist_ok=True) + with open(stub_path, 'r', encoding='utf-8') as f: + stub_content = f.read() + with open(out_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(stub_content) + _vlog().info(f" -> {os.path.basename(out_path)} (仅声明)") + + def _register_package_reexports(self, trans, init_pyi_path: str, package_name: str) -> None: + """解析 __init__.pyi 中的 from .xxx import yyy 或 from __xxx import yyy，在包级别注册导出符号""" + try: + # 复用 parse_python_file 的 AST 缓存 + _, tree = parse_python_file(init_pyi_path) + if tree is None: + return + for node in ast.walk(tree): + if isinstance(node, ast.ImportFrom) and node.module: + sub_module = node.module.lstrip('.') + for alias in node.names: + symbol_name = alias.name + exported_name = alias.asname if alias.asname else symbol_name + source_keys = [ + f"{package_name}.{sub_module}.{symbol_name}", + f"{package_name}.__{sub_module}.{symbol_name}" if not sub_module.startswith('__') else None, + symbol_name, + ] + target_key = f"{package_name}.{exported_name}" + if target_key not in trans.translator.SymbolTable: + for source_key in source_keys: + if source_key and source_key in trans.translator.SymbolTable: + trans.translator.SymbolTable[target_key] = trans.translator.SymbolTable[source_key] + break + if exported_name not in trans.translator.SymbolTable: + for source_key in source_keys: + if source_key and source_key in trans.translator.SymbolTable: + trans.translator.SymbolTable[exported_name] = trans.translator.SymbolTable[source_key] + break + except Exception as e: + _vlog().warning(f" [警告] 解析包重导出失败 {init_pyi_path}: {e}") + + def _compile_ll_files(self, active_sha1s: set[str]) -> None: + """将 output_dir 中属于 active_sha1s 的 .ll 文件编译为 .obj（跳过已编译过的 include 文件）""" + # 第一次 walk：清理 stale 文件（.obj/.deps.json/纯 .ll，排除 .stub.ll/.text.ll） + stale_files = [] + for root, dirs, files in os.walk(self.output_dir): + for file in files: + if file.endswith('.obj'): + sha1 = file[:-4] + if sha1 not in active_sha1s: + stale_files.append(os.path.join(root, file)) + elif file.endswith('.deps.json'): + sha1 = file[:-10] + if sha1 not in active_sha1s: + stale_files.append(os.path.join(root, file)) + elif file.endswith('.ll') and not file.endswith('.stub.ll') and not file.endswith('.text.ll'): + sha1 = file[:-3] + if sha1 not in active_sha1s: + stale_files.append(os.path.join(root, file)) + if stale_files: + for fpath in stale_files: + try: + os.remove(fpath) + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + _vlog().info(f"[清理] 删除 {len(stale_files)} 个过时文件") + + # 第二次 walk：收集需要编译的 .ll 文件（排除 .stub.ll/.text.ll） + ll_files = [] + for root, dirs, files in os.walk(self.output_dir): + for file in files: + if file.endswith('.ll') and not file.endswith('.stub.ll') and not file.endswith('.text.ll'): + sha1 = file[:-3] + if sha1 in active_sha1s: + fpath = os.path.join(root, file) + obj_path = os.path.join(root, file[:-3] + '.obj') + if not os.path.isfile(obj_path): + ll_files.append(fpath) + if not ll_files: + _vlog().info("[编译] 无 .ll 文件需要重新编译") + else: + _vlog().info(f"\n[编译] 找到 {len(ll_files)} 个 .ll 文件需要编译") + + # 第三次 walk：收集 all_ll_files（用于 alwaysinline 检测，排除 include sha1s） + all_ll_files = [] + for root, dirs, files in os.walk(self.output_dir): + for file in files: + if file.endswith('.ll') and not file.endswith('.stub.ll') and not file.endswith('.text.ll'): + sha1 = file[:-3] + if sha1 in active_sha1s and sha1 not in self._include_sha1s: + all_ll_files.append(os.path.join(root, file)) + + has_inline = False + for ll_path in all_ll_files: + try: + with open(ll_path, 'r', encoding='utf-8') as f: + content = f.read() + if 'alwaysinline' in content: + has_inline = True + break + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + + if has_inline: + llvm_link = shutil.which('llvm-link') or shutil.which('llvm-link.exe') + opt_cmd = shutil.which('opt') or shutil.which('opt.exe') + if llvm_link and opt_cmd: + _vlog().info(" [内联] 检测到 alwaysinline 函数，执行跨模块内联优化...") + for ll_path in all_ll_files: + try: + with open(ll_path, 'r', encoding='utf-8') as f: + content = f.read() + type_defs = [] + type_def_indices = set() + for i, line in enumerate(content.splitlines()): + stripped = line.strip() + if stripped.startswith('%') and '= type' in stripped: + type_defs.append(line) + type_def_indices.add(i) + if type_def_indices: + lines = content.splitlines(True) + content_no_types = ''.join(line for i, line in enumerate(lines) if i not in type_def_indices) + header_end = 0 + for i, line in enumerate(content_no_types.splitlines()): + stripped = line.strip() + if stripped.startswith(';') or stripped.startswith('target ') or stripped.startswith('source_filename') or not stripped: + header_end = i + 1 + continue + break + lines2 = content_no_types.splitlines(True) + fixed = ''.join(lines2[:header_end]) + '\n'.join(type_defs) + '\n' + ''.join(lines2[header_end:]) + with open(ll_path, 'w', encoding='utf-8') as f: + f.write(fixed) + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + merged_ll = os.path.join(self.output_dir, '_merged.ll') + optimized_ll = os.path.join(self.output_dir, '_merged_opt.ll') + try: + link_cmd = [llvm_link] + all_ll_files + ['-o', merged_ll] + result = subprocess.run(link_cmd, capture_output=True, text=True) + if result.returncode != 0: + _vlog().warning(f" llvm-link 失败: {result.stderr.strip()}，回退到单独编译") + has_inline = False + else: + opt_result_cmd = [opt_cmd, '-passes=always-inline', merged_ll, '-S', '-o', optimized_ll] + result = subprocess.run(opt_result_cmd, capture_output=True, text=True) + if result.returncode != 0: + _vlog().warning(f" opt 失败: {result.stderr.strip()}，回退到单独编译") + has_inline = False + else: + _vlog().info(" [内联] 跨模块内联优化完成") + with open(optimized_ll, 'r', encoding='utf-8') as f: + opt_content = f.read() + opt_content = re.sub(r'^(define|declare)\s+(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external)\s+', r'\1 \2 dso_local ', opt_content, flags=re.MULTILINE) + opt_content = re.sub(r'^(define|declare)\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external|dso_local|@)', r'\1 dso_local ', opt_content, flags=re.MULTILINE) + # LLVM 22+: external 不允许在带初值的全局定义上显式写出，改为 linkonce 保留多模块合并语义 + opt_content = re.sub(r'(@[\w.]+)\s*=\s*external\s+global\s+(\w[\w*]*)\s+(-?\d+|null|zeroinitializer|true|false)$', r'\1 = linkonce global \2 \3', opt_content, flags=re.MULTILINE) + opt_content = re.sub(r'(@[\w.]+)\s*=\s*(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr)\s+(global|constant)', r'\1 = \2 dso_local \3', opt_content) + opt_content = re.sub(r'(@[\w.]+)\s*=\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|dso_local)(global|constant)', r'\1 = dso_local \2', opt_content) + with open(optimized_ll, 'w', encoding='utf-8') as f: + f.write(opt_content) + merged_obj = os.path.join(self.output_dir, '_merged.obj') + try: + cmd = [self.compile_cmd] + self.compile_flags + ['-o', merged_obj, optimized_ll] + result = subprocess.run(cmd, capture_output=True, text=True, + cwd=self.output_dir) + if result.returncode == 0: + _vlog().success(f" 合并模块编译完成") + for ll_path in ll_files: + obj_path = ll_path[:-3] + '.obj' + sha1_obj = os.path.join(self.output_dir, os.path.basename(ll_path)[:-3] + '.obj') + if os.path.isfile(sha1_obj): + os.remove(sha1_obj) + return + else: + _vlog().warning(f" 合并模块编译失败: {result.stderr.strip()}，回退到单独编译") + has_inline = False + except Exception as e: + _vlog().warning(f" 合并模块编译异常: {e}，回退到单独编译") + has_inline = False + except FileNotFoundError: + _vlog().warning(f" 找不到 llvm-link 或 opt，回退到单独编译") + has_inline = False + else: + _vlog().warning(" 未找到 llvm-link/opt 工具，回退到单独编译（alwaysinline 可能不生效）") + + def _compile_one(ll_path: str) -> tuple[str, str, str]: + rel = os.path.relpath(ll_path, self.output_dir) + obj_path = ll_path[:-3] + '.obj' + if os.path.isfile(obj_path): + return (rel, 'cached', '') + try: + with open(ll_path, 'r', encoding='utf-8') as f: + ll_content = f.read() + sha1 = os.path.basename(ll_path)[:-3] + deps = self._LoadDeps(sha1) + stub_prefix = '' + if deps: + # 去重：同一个全局变量如果同时有 external 声明和实际定义， + # 移除 external 声明，避免 "redefinition of global" 错误 + # 注意：此去重必须在收集现有符号之前执行，否则行号偏移会导致 + # lines_to_remove 误删其他行（如全局变量定义） + ll_lines_for_dedup = ll_content.splitlines() + external_globals: dict[str, int] = {} + non_external_globals: dict[str, int] = {} + for i, line in enumerate(ll_lines_for_dedup): + stripped = line.strip() + if stripped.startswith('@') and (' global ' in stripped or ' constant ' in stripped): + gm = re.match(r'(@\S+)', stripped) + if gm: + gname = gm.group(1) + is_external = re.match(r'@\S+\s*=\s*external\s+', stripped) is not None + if is_external: + external_globals[gname] = i + else: + non_external_globals[gname] = i + lines_to_remove_global_dedup: set[int] = set() + for gname in external_globals: + if gname in non_external_globals: + lines_to_remove_global_dedup.add(external_globals[gname]) + if lines_to_remove_global_dedup: + ll_content = '\n'.join( + line for i, line in enumerate(ll_lines_for_dedup) + if i not in lines_to_remove_global_dedup + ) + '\n' + existing_symbols: dict[str, str] = {} + opaque_line_indices: dict[str, int] = {} + existing_declares: set[str] = set() + existing_declare_lines: dict[str, set[int]] = {} + for i, line in enumerate(ll_content.splitlines()): + stripped = line.strip() + if stripped.startswith('%') and '= type' in stripped: + name = Phase2Translator._extract_llvm_type_name(stripped) + if name: + if '= type opaque' in stripped: + existing_symbols[name] = 'opaque' + opaque_line_indices[name] = i + else: + existing_symbols[name] = 'full' + elif stripped.startswith('declare'): + dm = re.match(r'declare\s+[^@]*@"?([^"\s(]+)"?', stripped) + if dm: + fname = dm.group(1) + existing_declares.add(fname) + existing_declare_lines.setdefault(fname, set()).add(i) + seen_stub_sha1: set[str] = set() + replaced_opaque_names: set[str] = set() + replaced_declare_names: set[str] = set() + all_stub_lines: list[str | None] = [] + stub_type_map: dict[str, tuple[bool, int]] = {} + stub_declare_names: set[str] = set() + stub_declare_lines: dict[str, str] = {} + existing_defines: set[str] = set() + existing_global_defs: set[str] = set() + stub_global_defs: set[str] = set() + for line in ll_content.splitlines(): + stripped = line.strip() + if stripped.startswith('define'): + dm = re.match(r'define\s+[^@]*@"?([^"\s(]+)"?', stripped) + if dm: + existing_defines.add(dm.group(1)) + elif stripped.startswith('@') and (' global ' in stripped or ' constant ' in stripped): + gm = re.match(r'(@\S+)', stripped) + if gm: + existing_global_defs.add(gm.group(1)) + for dep_name, dep_sha1 in deps.items(): + if dep_sha1 in seen_stub_sha1: + continue + seen_stub_sha1.add(dep_sha1) + stub_path = os.path.join(self.output_dir, f"{dep_sha1}.stub.ll") + if not os.path.isfile(stub_path): + stub_path = os.path.join(self.temp_dir, f"{dep_sha1}.stub.ll") + if os.path.isfile(stub_path): + with open(stub_path, 'r', encoding='utf-8') as sf: + stub_content = sf.read() + for line in stub_content.splitlines(): + stripped = line.strip() + if stripped.startswith(';') or stripped.startswith('target ') or stripped.startswith('source_filename') or not stripped: + continue + if stripped.startswith('declare'): + dm = re.match(r'declare\s+[^@]*@"?([^"\s(]+)"?', stripped) + if dm: + fname = dm.group(1) + if 'zdef_alloc' in fname: + _vlog().debug(f"[DEBUG] stub declare for {fname}: existing_declares={fname in existing_declares}, stub_declare_names={fname in stub_declare_names}, has_struct={'%' in stripped}, existing={existing_declare_lines.get(fname)}") + if fname in existing_defines: + continue + if fname in existing_declares or fname in stub_declare_names: + existing_decl_line = stub_declare_lines.get(fname) + if existing_decl_line is not None: + has_struct = '%' in stripped + existing_has_struct = '%' in existing_decl_line if existing_decl_line else False + if has_struct and not existing_has_struct: + for i, l in enumerate(all_stub_lines): + if l is not None and l.strip().startswith('declare'): + em = re.match(r'declare\s+[^@]*@"?([^"\s(]+)"?', l.strip()) + if em and em.group(1) == fname: + all_stub_lines[i] = line + break + continue + elif fname in existing_declares: + has_struct = '%' in stripped + if has_struct: + replaced_declare_names.add(fname) + stub_declare_names.add(fname) + stub_declare_lines[fname] = line + all_stub_lines.append(line) + continue + continue + stub_declare_names.add(fname) + stub_declare_lines[fname] = line + all_stub_lines.append(line) + continue + if stripped.startswith('%') and '= type' in stripped: + name = Phase2Translator._extract_llvm_type_name(stripped) + if name: + is_opaque = '= type opaque' in stripped + if name in existing_symbols: + if existing_symbols[name] == 'opaque' and not is_opaque: + replaced_opaque_names.add(name) + existing_symbols[name] = 'full' + idx = len(all_stub_lines) + all_stub_lines.append(line) + stub_type_map[name] = (is_opaque, idx) + continue + if name in stub_type_map: + prev_opaque, prev_idx = stub_type_map[name] + if prev_opaque and not is_opaque: + all_stub_lines[prev_idx] = None + idx = len(all_stub_lines) + all_stub_lines.append(line) + stub_type_map[name] = (is_opaque, idx) + continue + idx = len(all_stub_lines) + all_stub_lines.append(line) + stub_type_map[name] = (is_opaque, idx) + else: + all_stub_lines.append(line) + elif stripped.startswith('@') and (' global ' in stripped or ' constant ' in stripped): + gm = re.match(r'(@\S+)', stripped) + if gm and (gm.group(1) in existing_global_defs or gm.group(1) in stub_global_defs): + continue + if gm: + stub_global_defs.add(gm.group(1)) + all_stub_lines.append(line) + else: + all_stub_lines.append(line) + filtered = [l for l in all_stub_lines if l is not None] + if filtered: + stub_prefix = '\n'.join(filtered) + '\n' + lines_to_remove = set() + if replaced_opaque_names: + for name in replaced_opaque_names: + if name in opaque_line_indices: + lines_to_remove.add(opaque_line_indices[name]) + if replaced_declare_names: + _vlog().debug(f"[DEBUG] replaced_declare_names: {replaced_declare_names}") + for fname in replaced_declare_names: + if fname in existing_declare_lines: + lines_to_remove.update(existing_declare_lines[fname]) + _vlog().debug(f"[DEBUG] removing lines {existing_declare_lines[fname]} for {fname}") + else: + _vlog().debug(f"[DEBUG] fname {fname} not in existing_declare_lines: {list(existing_declare_lines.keys())}") + if lines_to_remove: + ll_lines = ll_content.splitlines(True) + ll_content = ''.join(line for i, line in enumerate(ll_lines) if i not in lines_to_remove) + type_defs_in_ll = [] + type_def_line_indices = set() + for i, line in enumerate(ll_content.splitlines()): + stripped = line.strip() + if stripped.startswith('%') and '= type' in stripped: + type_defs_in_ll.append(line) + type_def_line_indices.add(i) + if type_def_line_indices: + ll_lines = ll_content.splitlines(True) + ll_content = ''.join(line for i, line in enumerate(ll_lines) if i not in type_def_line_indices) + if type_defs_in_ll: + stub_prefix += '\n'.join(type_defs_in_ll) + '\n' + if stub_prefix: + header_end = 0 + for i, line in enumerate(ll_content.splitlines()): + stripped = line.strip() + if stripped.startswith(';') or stripped.startswith('target ') or stripped.startswith('source_filename') or not stripped: + header_end = i + 1 + continue + break + ll_lines = ll_content.splitlines(True) + ll_content = ''.join(ll_lines[:header_end]) + stub_prefix + ''.join(ll_lines[header_end:]) + ll_content = re.sub(r'^(define|declare)\s+(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external)\s+', r'\1 \2 dso_local ', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(define|declare)\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external|dso_local|@)', r'\1 dso_local ', ll_content, flags=re.MULTILINE) + # LLVM 22+: external 不允许在带初值的全局定义上显式写出，改为 linkonce 保留多模块合并语义 + ll_content = re.sub(r'^(@[\w.]+)\s*=\s*external\s+global\s+(\w[\w*]*)\s+(-?\d+|null|zeroinitializer|true|false)$', r'\1 = linkonce global \2 \3', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(@[\w.]+)\s*=\s*(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr)\s+(global|constant)', r'\1 = \2 dso_local \3', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(@[\w.]+)\s*=\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|dso_local)(global|constant)', r'\1 = dso_local \2', ll_content, flags=re.MULTILINE) + # LLVM 22+: 移除同一模块中已有 define 的 declare 语句（LLVM 22 将 declare+define 视为重复定义） + defined_funcs: set[str] = set() + for m in re.finditer(r'^define\s+(?:dso_local\s+)?[^@]*@\"?([^"\s\(]+)\"?\s*\(', ll_content, re.MULTILINE): + defined_funcs.add(m.group(1)) + if defined_funcs: + lines: list[str] = ll_content.splitlines(True) + kept_lines: list[str] = [] + for line in lines: + m = re.match(r'^declare\s+(?:dso_local\s+)?[^@]*@\"?([^"\s\(]+)\"?\s*\(', line) + if m and m.group(1) in defined_funcs: + continue + kept_lines.append(line) + ll_content = ''.join(kept_lines) + # 移除重复的 declare 语句（同一函数多次 declare 但没有 define） + seen_declares: set[str] = set() + dedup_lines: list[str] = ll_content.splitlines(True) + dedup_kept: list[str] = [] + for line in dedup_lines: + dm = re.match(r'^declare\s+(?:dso_local\s+)?[^@]*@\"?([^"\s\(]+)\"?\s*\(', line) + if dm: + fname = dm.group(1) + if fname in seen_declares: + continue + seen_declares.add(fname) + dedup_kept.append(line) + ll_content = ''.join(dedup_kept) + final_type_defs = [] + final_type_def_indices = set() + seen_type_names_main = set() + for i, line in enumerate(ll_content.splitlines()): + stripped = line.strip() + if stripped.startswith('%') and '= type' in stripped: + type_name_main = stripped.split('=')[0].strip() + final_type_def_indices.add(i) + if type_name_main in seen_type_names_main: + continue + seen_type_names_main.add(type_name_main) + final_type_defs.append(line) + if final_type_def_indices: + final_lines = ll_content.splitlines(True) + ll_content_no_types = ''.join(line for i, line in enumerate(final_lines) if i not in final_type_def_indices) + header_end = 0 + for i, line in enumerate(ll_content_no_types.splitlines()): + stripped = line.strip() + if stripped.startswith(';') or stripped.startswith('target ') or stripped.startswith('source_filename') or not stripped: + header_end = i + 1 + continue + break + final_lines2 = ll_content_no_types.splitlines(True) + ll_content = ''.join(final_lines2[:header_end]) + '\n'.join(final_type_defs) + '\n' + ''.join(final_lines2[header_end:]) + dso_ll_path = ll_path[:-3] + '_dso.ll' + with open(dso_ll_path, 'w', encoding='utf-8') as f: + f.write(ll_content) + cmd = [self.compile_cmd] + self.compile_flags + ['-o', obj_path, dso_ll_path] + result = subprocess.run(cmd, capture_output=True, text=True, + cwd=os.path.dirname(ll_path) or '.') + if result.returncode == 0: + return (rel, 'ok', '') + else: + return (rel, 'error', result.stderr.strip()) + except FileNotFoundError: + return (rel, 'error', f'找不到编译器: {self.compile_cmd}') + except Exception as e: + return (rel, 'error', str(e)) + + need_compile = [] + for ll_path in ll_files: + obj_path = ll_path[:-3] + '.obj' + if not os.path.isfile(obj_path): + need_compile.append(ll_path) + + if need_compile: + n_workers = min(os.cpu_count() or 4, len(need_compile), 8) + _vlog().info(f" 并行编译 {len(need_compile)} 个文件 (workers={n_workers})") + errors = [] + with ThreadPoolExecutor(max_workers=n_workers) as executor: + futures = {executor.submit(_compile_one, ll_path): ll_path for ll_path in need_compile} + for future in as_completed(futures): + rel, status, err = future.result() + if status == 'ok': + _vlog().success(f" {rel}") + elif status == 'error': + _vlog().error(f" {rel}: {err}") + errors.append((rel, err)) + elif status == 'cached': + pass + if errors: + _vlog().error(f"\n{len(errors)} 个文件编译失败") + sys.exit(1) + + def _precollect_vtable_state(self) -> None: + """预扫描 includes .py 文件，收集 @t.CVTable 类的虚表信息。 + + 并行编译（ProcessPoolExecutor）创建独立内存的 worker 进程， + vtable 状态必须预收集并通过 pickle 共享。 + 复用 parse_python_file 的 AST 缓存。 + """ + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + for root, dirs, files in os.walk(includes_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if not fname.endswith('.py'): + continue + py_path: str = os.path.join(root, fname) + code, tree = parse_python_file(py_path) + # 字符串预过滤：不含 CVTable 的文件跳过 AST 遍历 + if tree is None or 'CVTable' not in code: + continue + try: + for node in ast.iter_child_nodes(tree): + if not isinstance(node, ast.ClassDef): + continue + has_cvtable: bool = False + for dec in node.decorator_list: + if isinstance(dec, ast.Attribute): + if getattr(dec.value, 'id', None) == 't' and dec.attr == 'CVTable': + has_cvtable = True + elif isinstance(dec, ast.Name): + if dec.id == 'CVTable': + has_cvtable = True + if not has_cvtable: + continue + class_name: str = node.name + has_methods: bool = False + if class_name not in self._shared_class_methods: + self._shared_class_methods[class_name] = [] + for item in node.body: + if isinstance(item, ast.FunctionDef): + has_methods = True + method_name: str = item.name + # 构造函数不放入 vtable（避免子类与基类 vtable 大小不一致破坏多态分派） + if method_name in ('__new__', '__init__', '__before_init__'): + continue + if method_name == '__init__': + func_full_name: str = f"{class_name}.__init__" + elif method_name == '__call__': + func_full_name = f"{class_name}.__call__" + else: + func_full_name = f"{class_name}.{method_name}" + if func_full_name not in self._shared_class_methods[class_name]: + self._shared_class_methods[class_name].append(func_full_name) + if has_methods: + self._shared_class_vtable.add(class_name) + self._shared_cross_module_vtable.add(class_name) + except Exception: + pass + + def _build_include_py_index(self, includes_dir: str) -> set[str]: + """预扫描 includes 目录，构建所有 .py 文件路径集合 + + 用 set 查询替代 os.path.isfile，避免 3490 次磁盘 stat（83s→<0.1s）。 + 同时规范化路径为小写（Windows 不区分大小写）以加速查找。 + """ + py_index: set[str] = set() + for root, dirs, files in os.walk(includes_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith('.py'): + py_index.add(os.path.normcase(os.path.join(root, fname))) + return py_index + + def _scan_include_libraries(self) -> None: + """扫描 includes 目录，检测被使用的模块对应的库文件和源文件""" + if not self.used_includes: + return + + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + + # 预构建 .py 文件索引，避免递归发现时反复 os.path.isfile + py_index = self._build_include_py_index(includes_dir) + + for module_name in self.used_includes: + module_dir = os.path.join(includes_dir, module_name) + if os.path.isdir(module_dir): + self._scan_dir_for_libs(module_dir, module_name) + for root, dirs, files in os.walk(module_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith('.py'): + py_path = os.path.join(root, fname) + if py_path not in self.extra_py_files: + self.extra_py_files.append(py_path) + _vlog().info(f" [includes] 发现 Python 包文件: {os.path.relpath(py_path, includes_dir)}") + + for ext in self.INCLUDE_LIB_EXTENSIONS: + lib_path = os.path.join(includes_dir, module_name + ext) + if os.path.isfile(lib_path): + self.extra_link_files.append(lib_path) + _vlog().info(f" [includes] 发现库文件: {os.path.relpath(lib_path, includes_dir)}") + + for ext in self.INCLUDE_SRC_EXTENSIONS: + src_path = os.path.join(includes_dir, module_name + ext) + if os.path.isfile(src_path): + self.extra_compile_files.append(src_path) + _vlog().info(f" [includes] 发现源文件: {os.path.relpath(src_path, includes_dir)}") + + py_path = os.path.join(includes_dir, module_name + '.py') + if os.path.normcase(py_path) in py_index: + self.extra_py_files.append(py_path) + _vlog().info(f" [includes] 发现 Python 源文件: {os.path.relpath(py_path, includes_dir)}") + + discovered = set() + for ef in list(self.extra_py_files): + self._discover_transitive_deps(ef, includes_dir, discovered, py_index) + + if self.extra_link_files: + _vlog().info(f"\n[includes] 需要链接的库文件: {len(self.extra_link_files)} 个") + if self.extra_compile_files: + _vlog().info(f"[includes] 需要编译的源文件: {len(self.extra_compile_files)} 个") + + def _scan_dir_for_libs(self, directory: str, module_name: str) -> None: + """递归扫描目录中的库文件和源文件""" + for root, dirs, files in os.walk(directory): + for fname in files: + fpath = os.path.join(root, fname) + _, ext = os.path.splitext(fname) + ext = ext.lower() + if ext in self.INCLUDE_LIB_EXTENSIONS: + self.extra_link_files.append(fpath) + _vlog().info(f" [includes] 发现库文件: {os.path.relpath(fpath, directory)}") + elif ext in self.INCLUDE_SRC_EXTENSIONS: + self.extra_compile_files.append(fpath) + _vlog().info(f" [includes] 发现源文件: {os.path.relpath(fpath, directory)}") + + def _discover_transitive_deps(self, py_path: str, includes_dir: str, discovered: set[str], py_index: set[str] | None = None) -> None: + """递归发现 Python 文件的间接依赖 + + 复用 parse_python_file 的 AST 缓存，避免重复 open+read+ast.parse。 + py_index 为预构建的 .py 文件路径集合，避免 os.path.isfile 磁盘 stat。 + """ + if py_path in discovered: + return + discovered.add(py_path) + _, tree = parse_python_file(py_path) + if tree is None: + return + try: + for node in ast.walk(tree): + if isinstance(node, ast.Import): + for alias in node.names: + self._try_add_include_dep(alias.name, includes_dir, discovered, py_index) + elif isinstance(node, ast.ImportFrom): + if node.module and node.level == 0: + self._try_add_include_dep(node.module, includes_dir, discovered, py_index) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"发现 include 传递依赖失败: {_e}", "Exception") + + def _try_add_include_dep(self, module_name: str, includes_dir: str, discovered: set, py_index: set[str] | None = None): + """尝试将模块添加为 include 依赖（支持子包，如 w32.win32base） + + py_index 非空时用 set 查询替代 os.path.isfile（O(1) vs 24ms/次）。 + """ + # 候选路径：子包路径（w32/win32base.py）、顶层模块（w32.py）、子包 __init__.py + candidates: list[str] = [] + # 子包路径：w32.win32base → w32/win32base.py + candidates.append(os.path.join(includes_dir, module_name.replace('.', os.sep) + '.py')) + # 子包 __init__.py：w32.win32base → w32/win32base/__init__.py + candidates.append(os.path.join(includes_dir, module_name.replace('.', os.sep), '__init__.py')) + # 顶层模块：取第一个组件 → w32.py + top_module = module_name.split('.')[0] + if top_module != module_name: + candidates.append(os.path.join(includes_dir, top_module + '.py')) + + for py_path in candidates: + exists: bool + if py_index is not None: + exists = os.path.normcase(py_path) in py_index + else: + exists = os.path.isfile(py_path) + if exists and py_path not in self.extra_py_files: + self.extra_py_files.append(py_path) + _vlog().info(f" [includes] 发现间接依赖: {os.path.relpath(py_path, includes_dir)}") + self._discover_transitive_deps(py_path, includes_dir, discovered, py_index) + return # 找到第一个匹配即可 + + def _is_decl_only_file(self, src_path: str) -> bool: + """复用 parse_python_file 的 AST 缓存""" + _, tree = parse_python_file(src_path) + if tree is None: + return False + try: + for node in ast.walk(tree): + if isinstance(node, ast.ClassDef): + return False + if isinstance(node, ast.FunctionDef): + body = node.body + if len(body) == 1: + stmt = body[0] + if isinstance(stmt, ast.Expr) and isinstance(stmt.value, ast.Constant) and stmt.value.value is ...: + continue + if isinstance(stmt, ast.Pass): + continue + return False + return True + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"判断是否为声明文件失败: {_e}", "Exception") + return False + + def _sort_include_files_by_deps(self, file_list: list[str]) -> tuple[list[str], dict[str, set[str]]]: + """按依赖关系对 include 文件进行拓扑排序，确保被依赖的文件先编译。 + + 返回 (排序后的文件列表, 依赖字典)。 + 依赖字典: {fpath: set(依赖的fpath)}，仅包含 file_list 中的依赖。 + """ + # 构建文件名到路径的映射 + name_to_path = {} + for fpath in file_list: + basename = os.path.splitext(os.path.basename(fpath))[0] + name_to_path[basename] = fpath + + # 也支持包内模块: vpsdk/window -> path + for fpath in file_list: + # 尝试从 includes 目录推断模块全名 + for includes_dir in self.include_dirs: + try: + rel = os.path.relpath(fpath, includes_dir) + mod_name = os.path.splitext(rel)[0].replace(os.sep, '.').replace('/', '.') + name_to_path[mod_name] = fpath + except ValueError: + pass + + # 分析每个文件的 import 依赖（复用 parse_python_file 的 AST 缓存） + deps = {} # path -> set of paths it depends on + for fpath in file_list: + deps[fpath] = set() + try: + _, tree = parse_python_file(fpath) + if tree is None: + continue + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.Import): + for alias in node.names: + # 尝试完整路径和各级前缀，确保同包内子模块依赖能被正确解析 + # 例如 import zlib.zhuff -> 应先匹配 zlib.zhuff，再回退到 zlib + parts = alias.name.split('.') + for i in range(len(parts), 0, -1): + mod = '.'.join(parts[:i]) + if mod in name_to_path and name_to_path[mod] != fpath: + deps[fpath].add(name_to_path[mod]) + break + elif isinstance(node, ast.ImportFrom): + if node.module and node.level == 0: + # 绝对导入：同样尝试完整路径和各级前缀 + parts = node.module.split('.') + for i in range(len(parts), 0, -1): + mod = '.'.join(parts[:i]) + if mod in name_to_path and name_to_path[mod] != fpath: + deps[fpath].add(name_to_path[mod]) + break + elif node.level > 0: + # 相对导入：from .xxx import yyy 或 from ..xxx import yyy + # 根据当前文件位置推断目标模块的完整路径 + current_dir = os.path.dirname(fpath) + # node.level == 1 表示当前包，node.level == 2 表示上一级包 + package_dir = current_dir + for _ in range(node.level - 1): + package_dir = os.path.dirname(package_dir) + # 推断当前包名（从 includes 目录的相对路径） + package_name: str | None = None + for includes_dir in self.include_dirs: + try: + rel = os.path.relpath(package_dir, includes_dir) + if rel and rel != '.': + package_name = rel.replace(os.sep, '.').replace('/', '.') + break + except ValueError: + pass + # 构建目标模块名并查找 + target_mod: str | None = None + if node.module: + target_mod = f"{package_name}.{node.module}" if package_name else node.module + else: + # from . import yyy：yyy 是子模块，在 names 中 + target_mod = package_name + if target_mod: + parts = target_mod.split('.') + for i in range(len(parts), 0, -1): + mod = '.'.join(parts[:i]) + if mod in name_to_path and name_to_path[mod] != fpath: + deps[fpath].add(name_to_path[mod]) + break + # from . import yyy：yyy 可能是子模块 + if node.level > 0 and not node.module: + for alias in node.names: + sub_mod = f"{package_name}.{alias.name}" if package_name else alias.name + if sub_mod in name_to_path and name_to_path[sub_mod] != fpath: + deps[fpath].add(name_to_path[sub_mod]) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"分析 include 文件导入依赖失败: {_e}", "Exception") + + # 拓扑排序 (Kahn's algorithm) + # in_degree 只计入 file_list 中的依赖——decl-only 文件（如 stdint.py） + # 不在 file_list 中，不应计入入度，否则依赖它们的文件入度永远不为 0， + # 被误判为循环依赖通过 fallback 添加，导致编译顺序错误。 + file_set = set(file_list) + in_degree = {fpath: sum(1 for dep in deps[fpath] if dep in file_set) for fpath in file_list} + # 反向邻接表：如果 A 依赖 B，则 B -> A + reverse_adj = {fpath: [] for fpath in file_list} + for fpath in file_list: + for dep in deps[fpath]: + if dep in reverse_adj: + reverse_adj[dep].append(fpath) + + result = [] + queue = [fpath for fpath in file_list if in_degree[fpath] == 0] + # 对队列排序以保持确定性 + queue.sort(key=lambda x: x) + + while queue: + current = queue.pop(0) + result.append(current) + for neighbor in reverse_adj[current]: + in_degree[neighbor] -= 1 + if in_degree[neighbor] == 0: + queue.append(neighbor) + queue.sort(key=lambda x: x) + + # 如果有循环依赖，把剩余文件也加入 + for fpath in file_list: + if fpath not in result: + result.append(fpath) + + # 仅保留 file_list 内的依赖（用于波次并行翻译） + file_set_result = set(file_list) + deps_in_set = {fpath: (deps[fpath] & file_set_result) for fpath in file_list} + return result, deps_in_set + + def _get_precompiled_obj_path(self, src_path: str, sha1: str) -> str | None: + """计算 include 文件在 includes.binary/ 中的预编译 .obj 路径。 + + 预编译目录位于 include 目录同级，命名为 {basename}.binary/。 + 文件命名为 {sha1}.{triple_safe}.{rel_model}.obj，区分目标平台和重定位模型。 + """ + AbsSrcPath: str = os.path.abspath(src_path) + RelModel: str = 'static' + for _flag in self.compile_flags: + if _flag.startswith('-relocation-model='): + RelModel = _flag[len('-relocation-model='):] + break + for inc_dir in self.include_dirs: + abs_inc: str = os.path.abspath(inc_dir) + try: + rel: str = os.path.relpath(AbsSrcPath, abs_inc) + if rel.startswith('..'): + continue + inc_parent: str = os.path.dirname(abs_inc) + inc_base: str = os.path.basename(abs_inc) + precompiled_dir: str = os.path.join(inc_parent, inc_base + '.binary') + triple_safe: str = (self.triple or 'default').replace(os.sep, '_').replace('/', '_').replace('\\', '_').replace(':', '_') + return os.path.join(precompiled_dir, f"{sha1}.{triple_safe}.{RelModel}.obj") + except ValueError: + continue + return None + + def _compile_include_py_files(self, active_sha1s: set[str]) -> None: + """通过 TransPyC 翻译器编译 includes 目录中发现的 .py 文件 + + 优化流程：预注册符号 → 并行翻译 → 串行后处理 → 并行编译 → 注册+缓存 + """ + if not self.extra_py_files: + return + + # 按依赖关系排序 include 文件，确保被依赖的文件先编译 + sorted_files, file_deps = self._sort_include_files_by_deps(self.extra_py_files) + _vlog().info(f"\n[includes 编译] 找到 {len(sorted_files)} 个 Python 源文件") + + # 构建 include 文件的模块 SHA1 映射（预计算所有文件，构建完整映射） + include_ModuleSha1Map = self._build_current_ModuleSha1Map(None) + need_translate = [] # [(src_path, sha1, module_name, ll_path, obj_path, precompiled_obj)] + for src_path in sorted_files: + module_name = os.path.splitext(os.path.basename(src_path))[0] + # 复用 parse_python_file 缓存的 content，避免重复 open+read + py_content, _ = parse_python_file(src_path) + sha1 = compute_sha1(py_content) + self.include_py_map[module_name] = sha1 + active_sha1s.add(sha1) + self._include_sha1s.add(sha1) + + for includes_dir in self.include_dirs: + try: + rel = os.path.relpath(src_path, includes_dir) + mod_full = os.path.splitext(rel)[0].replace(os.sep, '.').replace('/', '.') + include_ModuleSha1Map[mod_full] = sha1 + include_ModuleSha1Map[module_name] = sha1 + # 包入口文件：注册父包名（如 'os.__init__' -> 'os'） + if module_name == '__init__' and '.' in mod_full: + parent_pkg = mod_full.rsplit('.', 1)[0] + include_ModuleSha1Map[parent_pkg] = sha1 + except ValueError: + pass + include_ModuleSha1Map[module_name] = sha1 + ll_path = os.path.join(self.output_dir, f"{sha1}.ll") + obj_path = os.path.join(self.output_dir, f"{sha1}.obj") + precompiled_obj = self._get_precompiled_obj_path(src_path, sha1) + + if self._is_decl_only_file(src_path): + _vlog().info(f" 跳过(声明文件): {module_name}.py") + # 声明文件（如 stdint.py）虽不编译为 .obj，但其 typedef/类型别名 + # 必须注册到共享符号表，否则依赖它的 include 文件编译时 + # 类型解析失败（如 UINT32PTR 被当作命名结构体而非 i32*） + self._register_compiled_include(src_path, sha1, None, include_ModuleSha1Map) + self._collect_inline_symbols(src_path) + continue + + # 优先检查 includes.binary 预编译缓存（跨项目共享） + if not self.force_recompile_includes and precompiled_obj and os.path.isfile(precompiled_obj): + _vlog().info(f" 跳过(预编译): {module_name}.py -> {os.path.basename(precompiled_obj)}") + # 预编译跳过的文件仍需注册 .pyi/.stub.ll 声明和符号， + # 否则依赖它的 include 文件（如 vpui.py import vpsdk.window） + # 在 _EmitModuleDeclarationsLlvm 中找不到声明，导致 Undefined symbol + self._register_compiled_include(src_path, sha1, None, include_ModuleSha1Map) + self._collect_inline_symbols(src_path) + self.extra_link_files.append(precompiled_obj) + continue + + if not self.force_recompile_includes and os.path.isfile(obj_path): + _vlog().info(f" 跳过(缓存): {module_name}.py -> {sha1}.obj") + self._register_compiled_include(src_path, sha1, None, include_ModuleSha1Map) + self._collect_inline_symbols(src_path) + self.extra_link_files.append(obj_path) + continue + + need_translate.append((src_path, sha1, module_name, ll_path, obj_path, precompiled_obj)) + + if not need_translate: + return + + # 预注册所有待翻译文件的符号（从 Phase1 缓存加载 .pyi / .stub.ll） + # 这样并行翻译时所有跨模块符号都已可用 + for src_path, sha1, module_name, ll_path, obj_path, precompiled_obj in need_translate: + self._register_compiled_include(src_path, sha1, None, include_ModuleSha1Map) + + # 预收集 stub_decls（所有文件共用，避免重复扫描 .stub.ll） + stub_decls = [] + for other_sha1, other_stub in self.stub_files.items(): + if os.path.exists(other_stub): + try: + with open(other_stub, 'r', encoding='utf-8') as sf: + for sline in sf: + sl = sline.strip() + if sl.startswith('%') and '= type' in sl and 'opaque' not in sl: + struct_name = Phase2Translator._extract_llvm_type_name(sl) + if struct_name: + stub_decls.append(sl) + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + all_type_defs = {} + for sdecl in stub_decls: + tname = Phase2Translator._extract_llvm_type_name(sdecl) + if tname: + all_type_defs[tname] = sdecl + + # ========== 阶段1：并行翻译 ========== + _vlog().info(f" 翻译 {len(need_translate)} 个文件") + if len(need_translate) > 1: + # 序列化共享符号表供 worker 使用 + shared_pickle_path = os.path.join(self.output_dir, '_shared_sym_includes.pickle') + try: + shared_data = { + 'symbol_table': self._shared_symbol_table, + 'source_module_sig_files': self._shared_source_module_sig_files, + 'all_dc': self._shared_all_dc, + 'export_extern_funcs': self._shared_export_extern_funcs, + 'inline_func_symbols': self.inline_func_symbols, + 'function_default_args': self.function_default_args, + 'function_param_names': self.function_param_names, + 'generic_class_templates': self._shared_generic_class_templates, + 'shared_doc_meta': self._shared_doc_meta, + 'class_vtable': self._shared_class_vtable, + 'cross_module_vtable': self._shared_cross_module_vtable, + 'class_methods': self._shared_class_methods, + 'cross_module_novtable': self._shared_cross_module_novtable, + } + with open(shared_pickle_path, 'wb') as f: + f.write(pickle.dumps(shared_data)) + except Exception as e: + _vlog().warning(f"符号表序列化失败({e})，include将串行翻译") + shared_pickle_path = '' + + if shared_pickle_path and os.path.exists(shared_pickle_path): + n_workers = min(8, len(need_translate)) + errors = [] + # 波次并行翻译：按依赖关系分批提交，确保被依赖的文件先翻译完成。 + # 根因：_TryLoadStructFromStub 扫描 output 目录加载跨模块 struct 定义， + # 若依赖文件尚未完成翻译（output stub 未写入），只能取到 Phase1 temp + # 中的不完整 stub（缺少跨模块继承字段），导致运行时崩溃。 + item_by_path = {item[0]: item for item in need_translate} + translate_paths = set(item_by_path.keys()) + # 每个文件的依赖（仅限本轮待翻译集合内） + translate_deps = { + p: (file_deps.get(p, set()) & translate_paths) + for p in translate_paths + } + # 复用 self._executor_pool（与 App 源文件阶段共享），避免 GC 阻塞 + if self._executor_pool is None: + self._executor_pool = ProcessPoolExecutor(max_workers=n_workers) + executor = self._executor_pool + try: + completed_paths: set[str] = set() + remaining = list(need_translate) # 保持拓扑序 + in_progress: dict = {} # future -> item + done_count = 0 + + while remaining or in_progress: + # 提交所有依赖已完成的文件 + ready = [ + item for item in remaining + if translate_deps[item[0]] <= completed_paths + ] + for item in ready: + remaining.remove(item) + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + future = executor.submit( + _parallel_translate_worker, + src_path, ll_path, + self.src_root, self.temp_dir, self.output_dir, + self.include_dirs, self.triple, self.datalayout, + self.sha1_map, self.sig_files, self.stub_files, + self.include_py_map, self.slice_level, + shared_pickle_path, self.struct_sha1_map, + include_ModuleSha1Map + ) + in_progress[future] = item + + if not in_progress: + # 无可提交且无进行中：循环依赖 fallback + for item in remaining: + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + future = executor.submit( + _parallel_translate_worker, + src_path, ll_path, + self.src_root, self.temp_dir, self.output_dir, + self.include_dirs, self.triple, self.datalayout, + self.sha1_map, self.sig_files, self.stub_files, + self.include_py_map, self.slice_level, + shared_pickle_path, self.struct_sha1_map, + include_ModuleSha1Map + ) + in_progress[future] = item + remaining.clear() + + # 等待至少一个完成 + done_set, _ = wait(in_progress, return_when=FIRST_COMPLETED) + for future in done_set: + item = in_progress.pop(future) + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + completed_paths.add(src_path) + done_count += 1 + try: + status, _, err = future.result() + if status == 'ok': + _vlog().info(f" [{done_count}/{len(need_translate)}] 翻译完成: {module_name}.py") + else: + _vlog().error(f" {module_name}.py: {err}") + errors.append((module_name, err)) + except Exception as e: + _vlog().error(f" {module_name}.py: {e}") + errors.append((module_name, str(e))) + except Exception: + if self._executor_pool is not None: + self._executor_pool.shutdown(wait=False, cancel_futures=True) + self._executor_pool = None + raise + + if errors: + _vlog().error(f"\n{len(errors)} 个 include 文件翻译失败") + sys.exit(1) + else: + # 串行回退 + for item in need_translate: + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + _vlog().info(f" 翻译: {module_name}.py -> {sha1}.ll") + self._translate_file(src_path, ll_path, prebuilt_ModuleSha1Map=include_ModuleSha1Map) + else: + item = need_translate[0] + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + _vlog().info(f" 翻译: {module_name}.py -> {sha1}.ll") + self._translate_file(src_path, ll_path, prebuilt_ModuleSha1Map=include_ModuleSha1Map) + + # ========== 阶段2：串行后处理（dso_local + stub_decls 注入）========== + compile_tasks = [] # [(dso_ll_path, obj_path, src_path, sha1, module_name, ll_path, precompiled_obj, ll_content)] + for item in need_translate: + src_path, sha1, module_name, ll_path, obj_path, precompiled_obj = item + if not os.path.isfile(ll_path): + _vlog().error(f" 翻译未生成 .ll 文件: {module_name}.py") + continue + + with open(ll_path, 'r', encoding='utf-8') as f: + ll_content = f.read() + + # dso_local 正则标注（使用 ^ + MULTILINE 避免匹配字符串常量内容中的 define/declare） + ll_content = re.sub(r'^(define|declare)\s+(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external)\s+', r'\1 \2 dso_local ', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(define|declare)\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|external|dso_local)', r'\1 dso_local ', ll_content, flags=re.MULTILINE) + # LLVM 22+: external 不允许在带初值的全局定义上显式写出，改为 linkonce 保留多模块合并语义 + ll_content = re.sub(r'^(@[\w.]+)\s*=\s*external\s+global\s+(\w[\w*]*)\s+(-?\d+|null|zeroinitializer|true|false)$', r'\1 = linkonce global \2 \3', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(@[\w.]+)\s*=\s*(private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr)\s+(global|constant)', r'\1 = \2 dso_local \3', ll_content, flags=re.MULTILINE) + ll_content = re.sub(r'^(@[\w.]+)\s*=\s+(?!private|internal|available_externally|linkonce|weak|common|appending|extern_weak|linkonce_odr|weak_odr|dso_local)(global|constant)', r'\1 = dso_local \2', ll_content, flags=re.MULTILINE) + + # stub_decls 注入（复用预收集的 stub_decls） + if stub_decls: + for sdecl in stub_decls: + sname = Phase2Translator._extract_llvm_type_name(sdecl) + if sname: + short = sname.split('.')[-1] if '.' in sname else sname + ll_content = re.sub( + r'%"?' + re.escape(sname) + r'"?\s*=\s*type\s*opaque', + sdecl, ll_content) + if short != sname: + ll_content = re.sub( + r'%"?' + re.escape(short) + r'"?\s*=\s*type\s*opaque', + sdecl, ll_content) + + max_iterations = 10 + for _ in range(max_iterations): + defined_types = set() + for line in ll_content.splitlines(): + sl = line.strip() + if sl.startswith('%') and '= type' in sl: + tname = Phase2Translator._extract_llvm_type_name(sl) + if tname: + defined_types.add(tname) + # 同时添加短名，防止 missing 循环因全名/短名不匹配而重复添加 + if '.' in tname: + parts = tname.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + defined_types.add(parts[1]) + + referenced_types = Phase2Translator._find_all_type_refs(ll_content) + missing = referenced_types - defined_types + new_defs = [] + for tname in sorted(missing): + if tname in all_type_defs: + new_defs.append(all_type_defs[tname]) + else: + short_key = tname.split('.')[-1] if '.' in tname else tname + for full_name, defn in all_type_defs.items(): + if full_name.split('.')[-1] == short_key: + if short_key not in defined_types: + new_defs.append(defn) + break + else: + # 未在 stub 中找到定义（如泛型类 list[T]）， + # 声明为 opaque 以满足 llc 的类型前向声明要求 + new_defs.append(f'%"{tname}" = type opaque') + + if not new_defs: + break + + inject_block = '\n'.join(new_defs) + '\n' + lines = ll_content.splitlines(True) + insert_pos = 0 + for i, line in enumerate(lines): + stripped = line.strip() + if stripped and not stripped.startswith(';') and not stripped.startswith('target ') and not stripped.startswith('source_filename') and not stripped.startswith('declare ') and not stripped.startswith('@') and not stripped.startswith('define ') and not stripped.startswith('attributes ') and not stripped.startswith('!') and not stripped.startswith('module '): + insert_pos = i + break + if insert_pos == 0: + insert_pos = len(lines) + ll_content = ''.join(lines[:insert_pos]) + inject_block + ''.join(lines[insert_pos:]) + + # 将所有类型定义移到 declares 之前 + # （llc 要求 by-value struct 参数的类型在 declare 前定义，否则报 invalid type for function argument） + lines = ll_content.splitlines(True) + type_def_lines = [] + other_lines = [] + seen_type_names_inc = set() + for line in lines: + stripped = line.strip() + if stripped.startswith('%') and '= type' in stripped: + type_name_inc = stripped.split('=')[0].strip() + if type_name_inc in seen_type_names_inc: + continue + seen_type_names_inc.add(type_name_inc) + type_def_lines.append(line) + else: + other_lines.append(line) + if type_def_lines: + insert_pos = 0 + for i, line in enumerate(other_lines): + stripped = line.strip() + if stripped and not stripped.startswith(';') and not stripped.startswith('target ') and not stripped.startswith('source_filename') and not stripped.startswith('!'): + insert_pos = i + break + if insert_pos == 0: + insert_pos = len(other_lines) + ll_content = ''.join(other_lines[:insert_pos]) + ''.join(type_def_lines) + ''.join(other_lines[insert_pos:]) + + dso_ll_path = ll_path[:-3] + '_dso.ll' + with open(dso_ll_path, 'w', encoding='utf-8') as f: + f.write(ll_content) + compile_tasks.append((dso_ll_path, obj_path, src_path, sha1, module_name, ll_path, precompiled_obj, ll_content)) + + if not compile_tasks: + return + + # ========== 阶段3：并行编译（llc）========== + _vlog().info(f" 编译 {len(compile_tasks)} 个 .ll 文件") + compile_results = {} # task -> result + + def _compile_one_include(task): + dso_ll_path, obj_path = task[0], task[1] + cmd = [self.compile_cmd] + self.compile_flags + ['-o', obj_path, dso_ll_path] + result = subprocess.run(cmd, capture_output=True, text=True, cwd=self.output_dir) + return (task, result) + + if len(compile_tasks) > 1: + n_workers = min(8, len(compile_tasks)) + with ThreadPoolExecutor(max_workers=n_workers) as executor: + futures = {executor.submit(_compile_one_include, task): task for task in compile_tasks} + done_count = 0 + for future in as_completed(futures): + done_count += 1 + try: + task, result = future.result() + module_name = task[4] + if result.returncode == 0: + _vlog().success(f" [{done_count}/{len(compile_tasks)}] 编译成功: {module_name}.py") + else: + _vlog().warning(f" [{done_count}/{len(compile_tasks)}] 编译失败: {module_name}.py: {result.stderr.strip()}") + compile_results[task] = result + except Exception as e: + task = futures[future] + module_name = task[4] + _vlog().error(f" [{done_count}/{len(compile_tasks)}] 编译异常: {module_name}.py: {e}") + compile_results[task] = None + else: + task = compile_tasks[0] + task_result, result = _compile_one_include(task) + module_name = task[4] + if result.returncode == 0: + _vlog().success(f" 编译成功: {module_name}.py") + else: + _vlog().warning(f" 编译失败: {module_name}.py: {result.stderr.strip()}") + compile_results[task] = result + + # ========== 阶段4：注册 + 缓存 ========== + has_error = False + for task in compile_tasks: + dso_ll_path, obj_path, src_path, sha1, module_name, ll_path, precompiled_obj, ll_content = task + result = compile_results.get(task) + if result and result.returncode == 0: + self.extra_link_files.append(obj_path) + # 复制到 includes.binary 预编译目录 + if precompiled_obj: + precompiled_dir = os.path.dirname(precompiled_obj) + try: + if not os.path.exists(precompiled_dir): + os.makedirs(precompiled_dir, exist_ok=True) + shutil.copy2(obj_path, precompiled_obj) + except Exception as _e: + if _config_mode == "strict": + _vlog().warning(f"预编译缓存写入失败: {_e}") + # 注册符号 + self._register_compiled_include(src_path, sha1, ll_content, include_ModuleSha1Map) + # 清理临时文件（仅成功时清理，失败时保留以便调试） + # DEBUG: 保留 .ll 文件以便静态分析 + # for ext in ['.ll', '_dso.ll']: + # p = ll_path[:-3] + ext + # if os.path.isfile(p): + # os.remove(p) + else: + has_error = True + _vlog().warning(f" 保留失败文件以便调试: {ll_path}") + + if has_error: + _vlog().error(f"\nincludes 文件编译失败") + sys.exit(1) + + # 从 include 文件源码中提取函数参数名，供主项目编译时 keyword 参数匹配使用 + self._extract_include_param_names(sorted_files) + + def _extract_include_param_names_from_dirs(self) -> None: + """从 include 目录的 Python 源码中提取函数参数名，存入 function_param_names + + 优化：单次递归遍历 AST，维护当前类上下文，O(N) 复杂度。 + 复用 parse_python_file 的 AST 缓存。 + """ + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + for filename in os.listdir(includes_dir): + if not filename.endswith('.py'): + continue + src_path = os.path.join(includes_dir, filename) + try: + source, tree = parse_python_file(src_path) + if tree is None: + continue + sha1 = compute_sha1(source) + self._collect_function_params(tree, sha1) + except Exception as e: + _vlog().warning(f" 提取参数名失败 {src_path}: {e}") + + def _collect_function_params(self, tree: ast.AST, sha1: str | None) -> None: + """单次递归遍历 AST 收集函数参数名（含类方法），O(N) 复杂度 + + 遍历时维护 current_class 上下文：遇到 ClassDef 进入类作用域， + 遇到 FunctionDef 记录参数名并标注所属类，函数内部的嵌套定义 + 不再属于外层类。 + """ + def visit(node: ast.AST, current_class: str | None) -> None: + for child in ast.iter_child_nodes(node): + if isinstance(child, ast.ClassDef): + # 进入类作用域，递归处理类体 + visit(child, child.name) + elif isinstance(child, (ast.FunctionDef, ast.AsyncFunctionDef)): + func_name = child.name + param_names = [arg.arg for arg in child.args.args] + self.function_param_names[func_name] = param_names + if sha1: + self.function_param_names[f'{sha1}.{func_name}'] = param_names + if current_class: + self.function_param_names[f'{current_class}.{func_name}'] = param_names + if sha1: + self.function_param_names[f'{sha1}.{current_class}.{func_name}'] = param_names + # 函数内部的嵌套定义不属于外层类 + visit(child, None) + else: + visit(child, current_class) + visit(tree, None) + + def _extract_app_default_args(self, need_translate: list) -> None: + """从 App 源文件中提取函数默认参数，存入 function_default_args + + 在 Phase 2 并行编译开始前调用，确保跨模块调用参数检查能正确识别默认参数。 + need_translate 中每个元素是 (i, src_path, out_path, sha1, current_ModuleSha1Map) + 复用 parse_python_file 的 AST 缓存（run() 循环已 parse 过）。 + """ + for entry in need_translate: + src_path = entry[1] + sha1 = entry[3] + _, tree = parse_python_file(src_path) + if tree is None: + continue + try: + for node in ast.walk(tree): + if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)): + func_name = node.name + defaults = getattr(node.args, 'defaults', []) + if defaults: + default_vals = [] + for d in defaults: + if isinstance(d, ast.Constant): + default_vals.append(d.value) + else: + default_vals.append(None) + # 存储多种键格式，确保跨模块调用检查能找到 + self.function_default_args[func_name] = default_vals + self.function_default_args[f'{sha1}.{func_name}'] = default_vals + # 也更新参数名（预扫描时补充） + param_names = [arg.arg for arg in node.args.args] + if func_name not in self.function_param_names: + self.function_param_names[func_name] = param_names + if f'{sha1}.{func_name}' not in self.function_param_names: + self.function_param_names[f'{sha1}.{func_name}'] = param_names + except Exception as e: + _vlog().warning(f" 提取默认参数失败 {src_path}: {e}") + + def _extract_include_param_names(self, sorted_files: list[str]) -> None: + """从 include 文件的 Python 源码中提取函数参数名，存入 function_param_names + + 优化：复用 _collect_function_params 的 O(N) 单次遍历 + parse_python_file 缓存。 + """ + for src_path in sorted_files: + if self._is_decl_only_file(src_path): + continue + try: + _, tree = parse_python_file(src_path) + if tree is None: + continue + module_name = os.path.splitext(os.path.basename(src_path))[0] + sha1 = self.include_py_map.get(module_name) + self._collect_function_params(tree, sha1) + except Exception as e: + _vlog().warning(f" 提取参数名失败 {src_path}: {e}") + + def _register_compiled_include(self, src_path: str, sha1: str, ll_content: str, include_ModuleSha1Map: dict[str, str]) -> None: + """编译 include 文件成功后，生成签名和 stub 文件并注册到符号表中""" + module_name = os.path.splitext(os.path.basename(src_path))[0] + + # 确定完整模块名（如 vpsdk.window） + mod_full = module_name + for includes_dir in self.include_dirs: + try: + rel = os.path.relpath(src_path, includes_dir) + mod_full = os.path.splitext(rel)[0].replace(os.sep, '.').replace('/', '.') + break + except ValueError: + pass + + # 1. 生成 .pyi 签名文件（如果还没有的话） + sig_path = os.path.join(self.temp_dir, f"{sha1}.pyi") + if not os.path.isfile(sig_path): + # 全局缓存检查（不被 --clean 清除） + _TryGlobalCache(sha1, 'pyi', sig_path) + if not os.path.isfile(sig_path): + try: + with open(src_path, 'r', encoding='utf-8') as f: + py_content = f.read() + sig_content = PythonToStubConverter.convert(py_content, mod_full) + os.makedirs(self.temp_dir, exist_ok=True) + with open(sig_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(sig_content) + _SaveToGlobalCache(sha1, 'pyi', sig_path) + except Exception as e: + if _config_mode == "strict": + raise + _vlog().warning(f"签名生成失败不阻塞编译: {e}", "Exception") + + # 注册到 sig_files 和 _source_module_sig_files + if os.path.isfile(sig_path): + self.sig_files[sha1] = sig_path + self.sha1_map[sha1] = f"includes/{mod_full.replace('.', os.sep)}.py" + if self._shared_source_module_sig_files is not None: + self._shared_source_module_sig_files[mod_full] = sig_path + self._shared_source_module_sig_files[module_name] = sig_path + + # 将签名信息加载到共享符号表中 + if self._shared_symbol_table is not None: + try: + self._shared_symbol_table.LoadModuleSymbols(sig_path, mod_full, lineno=0) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"加载模块签名到共享符号表失败: {_e}", "Exception") + + # 2. 生成 .stub.ll 文件 + stub_path = os.path.join(self.output_dir, f"{sha1}.stub.ll") + if not os.path.isfile(stub_path): + # 全局缓存检查（不被 --clean 清除） + _TryGlobalCache(sha1, 'stub.ll', stub_path) + if not os.path.isfile(stub_path) and ll_content: + try: + stub_content, _ = self._split_ll(ll_content) + with open(stub_path, 'w', encoding='utf-8', newline='\n') as f: + f.write(stub_content) + _SaveToGlobalCache(sha1, 'stub.ll', stub_path) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"生成 .stub.ll 文件失败: {_e}", "Exception") + + # 注册到 stub_files + if os.path.isfile(stub_path): + self.stub_files[sha1] = stub_path + elif os.path.isfile(sig_path): + # 如果 stub 文件不存在，也尝试从 temp 目录查找 + temp_stub = os.path.join(self.temp_dir, f"{sha1}.stub.ll") + if os.path.isfile(temp_stub): + self.stub_files[sha1] = temp_stub + else: + # 最后尝试从全局缓存直接读取 + _TryGlobalCache(sha1, 'stub.ll', temp_stub) + if os.path.isfile(temp_stub): + self.stub_files[sha1] = temp_stub + + def _precollect_inline_symbols(self) -> None: + """在主翻译循环之前，扫描includes目录中used_includes相关的模块收集内联函数AST""" + if not self.used_includes: + return + for includes_dir in self.include_dirs: + if not os.path.isdir(includes_dir): + continue + for module_name in self.used_includes: + module_dir = os.path.join(includes_dir, module_name) + if os.path.isdir(module_dir): + for root, dirs, files in os.walk(module_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith('.py'): + src_path = os.path.join(root, fname) + self._collect_inline_symbols(src_path) + py_path = os.path.join(includes_dir, module_name + '.py') + if os.path.isfile(py_path): + self._collect_inline_symbols(py_path) + for ef in self.extra_py_files: + self._collect_inline_symbols(ef) + + def _collect_inline_symbols(self, src_path: str): + """解析源文件，收集内联函数的AST信息到inline_func_symbols + + 复用 parse_python_file 的 AST 缓存，避免重复 open+read+ast.parse。 + """ + try: + _, tree = parse_python_file(src_path) + if tree is None: + return + module_name = os.path.splitext(os.path.basename(src_path))[0] + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.FunctionDef) and node.returns: + is_inline = False + if isinstance(node.returns, ast.BinOp) and isinstance(node.returns.op, ast.BitOr): + for part in [node.returns.left, node.returns.right]: + if isinstance(part, ast.Attribute) and part.attr == 'CInline': + is_inline = True + break + if isinstance(part, ast.Name) and part.id == 'CInline': + is_inline = True + break + elif isinstance(node.returns, ast.Attribute) and node.returns.attr == 'CInline': + is_inline = True + elif isinstance(node.returns, ast.Name) and node.returns.id == 'CInline': + is_inline = True + if is_inline: + func_name = node.name + sym_key = f"{module_name}.{func_name}" + info = CTypeInfo() + info.Name = func_name + info.IsFunction = True + info.IsInline = True + info.InlineBody = node.body + info.InlineParams = [arg.arg for arg in node.args.args] + info.Storage = t.CInline() + self.inline_func_symbols[func_name] = info + self.inline_func_symbols[sym_key] = info + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"预收集内联函数符号失败: {_e}", "Exception") + + def _build_shared_symbol_data(self) -> None: + """一次性构建所有文件共享的符号表数据，避免每个文件重复加载""" + t0 = time.time() + + _vlog().info("[缓存] 构建共享符号表数据...") + + base_trans = TransPyC.TransPyC(code="pass", triple=self.triple, datalayout=self.datalayout) + base_trans.translator.LlvmGen = None + base_trans.translator._ModuleSha1Map = {} + base_trans.translator._global_function_default_args = self.function_default_args + base_trans.translator._global_function_param_names = self.function_param_names + + for includes_dir in self.include_dirs: + if os.path.isdir(includes_dir): + for pyi_file in os.listdir(includes_dir): + if pyi_file.endswith('.pyi'): + pyi_path = os.path.join(includes_dir, pyi_file) + module_name = os.path.splitext(pyi_file)[0] + base_trans.translator.SymbolTable.LoadModuleSymbols(pyi_path, module_name, lineno=0) + for py_file in os.listdir(includes_dir): + if py_file.endswith('.py') and (not py_file.startswith('_') or py_file == '_list.py' or py_file == '_dict.py'): + module_name = os.path.splitext(py_file)[0] + sha1_key = self.include_py_map.get(module_name) + if sha1_key and sha1_key in self.sig_files: + sig_path = self.sig_files[sha1_key] + base_trans.translator.SymbolTable.LoadModuleSymbols(sig_path, module_name, lineno=0) + # 收集需要重导出的包，等所有模块符号加载完后再处理 + _pending_reexports = [] + for sha1_key, sig_path in self.sig_files.items(): + rel_path = self.sha1_map.get(sha1_key, sha1_key) + if rel_path.startswith('includes/'): + mod_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + mod_name = mod_name[len('includes/'):] + base_trans.translator.SymbolTable.LoadModuleSymbols(sig_path, mod_name, lineno=0) + if mod_name.endswith('.__init__'): + package_name = mod_name[:-len('.__init__')] + _pending_reexports.append((sig_path, package_name)) + + # 所有 includes 模块符号已加载，现在处理包重导出 + for sig_path, package_name in _pending_reexports: + self._register_package_reexports(base_trans, sig_path, package_name) + + for sha1_key, sig_path in self.sig_files.items(): + rel_path = self.sha1_map.get(sha1_key, sha1_key) + if rel_path.startswith('includes/'): + continue + module_name = os.path.splitext(rel_path)[0].replace(os.sep, '.').replace('/', '.') + base_trans.translator.SymbolTable.LoadModuleSymbols(sig_path, module_name, lineno=0) + base_trans.translator._source_module_sig_files[module_name] = sig_path + short_name = module_name.split('.')[-1] if '.' in module_name else module_name + if short_name != module_name: + base_trans.translator._source_module_sig_files[short_name] = sig_path + + all_dc = {} + for sha1_key, stub_path in self.stub_files.items(): + if os.path.exists(stub_path): + try: + with open(stub_path, 'r', encoding='utf-8') as f: + stub_content = f.read() + for line in stub_content.splitlines(): + line = line.strip() + if line.startswith('@') and '= global' in line and 'i32' in line and not line.startswith('@llvm'): + var_name = line.split('=')[0].strip().lstrip('@') + val_part = line.split('i32')[-1].strip().rstrip(']').lstrip('[').lstrip('i32') + try: + val = int(val_part) + all_dc[var_name] = val + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + except Exception as e: + _vlog().warning(f"Phase2Translator: 忽略异常 {e}", exc_info=e) + + for sha1_key, pyi_path in self.sig_files.items(): + _ExtractCDefineConstantsFromPyi(pyi_path, all_dc) + + export_extern_funcs = {} + for sha1_key, sig_path in self.sig_files.items(): + try: + # 复用 parse_python_file 的 AST 缓存（sig 文件被多处解析） + _, sig_tree = parse_python_file(sig_path) + if sig_tree is None: + continue + for node in ast.iter_child_nodes(sig_tree): + if isinstance(node, ast.FunctionDef): + if _check_annotation_for_export(node.returns): + export_extern_funcs[node.name] = sha1_key + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"收集共享导出外部函数失败: {_e}", "Exception") + + for sym_name, sym_info in self.inline_func_symbols.items(): + if sym_name not in base_trans.translator.SymbolTable: + base_trans.translator.SymbolTable[sym_name] = sym_info + else: + existing = base_trans.translator.SymbolTable[sym_name] + if not getattr(existing, 'InlineBody', None) and getattr(sym_info, 'InlineBody', None): + existing.IsInline = sym_info.IsInline + existing.InlineBody = sym_info.InlineBody + existing.InlineParams = sym_info.InlineParams + + self._shared_symbol_table = base_trans.translator.SymbolTable + self._shared_source_module_sig_files = dict(base_trans.translator._source_module_sig_files) + self._shared_all_dc = all_dc + self._shared_export_extern_funcs = export_extern_funcs + + generic_class_templates = {} + for includes_dir in self.include_dirs: + if os.path.isdir(includes_dir): + abs_includes = os.path.join(self.src_root, includes_dir) if not os.path.isabs(includes_dir) else includes_dir + if not os.path.isdir(abs_includes): + abs_includes = includes_dir + for py_file in os.listdir(abs_includes): + if py_file.endswith('.py') and (not py_file.startswith('_') or py_file == '_list.py' or py_file == '_dict.py'): + py_path = os.path.join(abs_includes, py_file) + try: + # 复用 parse_python_file 的 AST 缓存 + _, py_tree = parse_python_file(py_path) + if py_tree is None: + continue + for node in py_tree.body: + if isinstance(node, ast.ClassDef) and hasattr(node, 'type_params') and node.type_params: + type_params = [tp.name for tp in node.type_params] + generic_class_templates[node.name] = { + 'node': node, + 'type_params': type_params, + } + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"收集泛型类模板失败: {_e}", "Exception") + self._shared_generic_class_templates = generic_class_templates + + # 加载所有 {sha1}.doc.json 到 _shared_doc_meta，供跨模块 __doc__ 查询 + shared_doc_meta: dict[str, dict[str, str]] = {} + if os.path.isdir(self.temp_dir): + for fname in os.listdir(self.temp_dir): + if fname.endswith('.doc.json'): + sha1_key = fname[:-len('.doc.json')] + try: + with open(os.path.join(self.temp_dir, fname), 'r', encoding='utf-8') as f: + data = json.load(f) + if isinstance(data, dict): + shared_doc_meta[sha1_key] = {str(k): str(v) for k, v in data.items()} + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"加载 {fname} 失败: {_e}", "Exception") + # 也从全局缓存加载 .doc.json（temp_dir 被 --clean 清空时补充） + if os.path.isdir(_GLOBAL_CACHE_DIR): + for fname in os.listdir(_GLOBAL_CACHE_DIR): + if fname.endswith('.doc.json'): + sha1_key = fname[:-len('.doc.json')] + if sha1_key in shared_doc_meta: + continue + try: + with open(os.path.join(_GLOBAL_CACHE_DIR, fname), 'r', encoding='utf-8') as f: + data = json.load(f) + if isinstance(data, dict): + shared_doc_meta[sha1_key] = {str(k): str(v) for k, v in data.items()} + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"加载全局缓存 {fname} 失败: {_e}", "Exception") + self._shared_doc_meta = shared_doc_meta + + t1 = time.time() + _vlog().info(f"[缓存] 共享符号表构建完成 ({t1-t0:.2f}s, {len(self._shared_symbol_table)} 符号, {len(generic_class_templates)} 泛型模板, {len(shared_doc_meta)} doc 元数据)") + + def _compile_include_sources(self): + """扫描项目目录中的汇编存根，编译 includes 目录中发现的 C/C++ 源文件""" + extra_sources = list(self.extra_compile_files) + scan_dirs = [] + if self.src_root and os.path.isdir(self.src_root): + scan_dirs.append(self.src_root) + project_dir = os.path.dirname(self.output_dir) + if project_dir and os.path.isdir(project_dir) and project_dir not in scan_dirs: + scan_dirs.append(project_dir) + for scan_dir in scan_dirs: + for root, dirs, files in os.walk(scan_dir): + dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__'] + for fname in files: + if fname.endswith(('.S', '.s', '.c', '.cpp')): + if fname.startswith('test_'): + continue + fpath = os.path.join(root, fname) + if fpath not in extra_sources: + extra_sources.append(fpath) + if not extra_sources: + return + + cc_map = { + '.c': 'gcc', + '.cpp': 'g++', '.cc': 'g++', '.cxx': 'g++', + '.m': 'clang', '.mm': 'clang++', + '.s': 'gcc', '.S': 'gcc', + } + + _vlog().info(f"\n[includes 编译] 找到 {len(extra_sources)} 个源文件") + + for src_path in extra_sources: + _, ext = os.path.splitext(src_path) + ext = ext.lower() + obj_path = os.path.join(self.output_dir, os.path.splitext(os.path.basename(src_path))[0] + '.obj') + if ext in ('.s', '.S') and any('oformat' in f for f in self.linker_flags): + cc = 'clang' + compile_src = src_path + extra_flags = ['-c', '-target', 'x86_64-none-elf', '-o', obj_path, compile_src] + else: + cc = cc_map.get(ext, 'gcc') + extra_flags = ['-c', '-o', obj_path, src_path] + + _vlog().info(f" 编译: {os.path.basename(src_path)} -> {os.path.basename(obj_path)}") + + try: + cmd = [cc] + extra_flags + result = subprocess.run(cmd, capture_output=True, text=True) + if result.returncode == 0: + _vlog().success(" 编译成功") + self.extra_link_files.append(obj_path) + else: + _vlog().error(f" 编译失败: {result.stderr.strip()}") + _vlog().error(f"\nC/C++ 源文件编译失败，立即终止编译。") + sys.exit(1) + except FileNotFoundError: + _vlog().error(f" 找不到编译器: {cc}") + _vlog().error(f"\n找不到编译器，立即终止编译。") + sys.exit(1) + except Exception as e: + _vlog().error(f" {e}") + _vlog().error(f"\n编译异常，立即终止编译。") + sys.exit(1) + + def _strip_debug_sections(self, exe_path: str) -> None: + """剥离 .exe 中的调试段（跳过裸机二进制）""" + if not os.path.exists(exe_path): + return + ext = os.path.splitext(exe_path)[1].lower() + if ext not in ('.exe', '.dll', '.obj', '.o', '.elf'): + _vlog().info(f" [剥离] 跳过（{ext} 格式无需剥离）") + return + for tool in ['llvm-strip', 'strip']: + strip_cmd = shutil.which(tool) + if strip_cmd: + try: + result = subprocess.run( + [strip_cmd, '--strip-debug', exe_path], + capture_output=True, text=True + ) + if result.returncode == 0: + _vlog().info(f" [剥离] 调试段已移除 ({tool})") + else: + _vlog().warning(f" [剥离] 警告: {result.stderr.strip()}") + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"剥离可执行文件调试段失败: {_e}", "Exception") + break + + def _link_obj_files(self, active_sha1s: set[str]) -> None: + """将 active_sha1s 对应的 .obj 文件和 includes 库文件链接为可执行文件""" + obj_files = [] + merged_obj = os.path.join(self.output_dir, '_merged.obj') + if os.path.isfile(merged_obj): + obj_files.append(merged_obj) + else: + for root, dirs, files in os.walk(self.output_dir): + for file in files: + if file.endswith('.obj'): + sha1 = file[:-4] + if sha1 in active_sha1s and sha1 not in self._include_sha1s: + obj_files.append(os.path.join(root, file)) + + if not obj_files and not self.extra_link_files: + _vlog().info("[链接] 无文件可链接") + return + + _vlog().info(f"\n[链接] 找到 {len(obj_files)} 个 .obj 文件") + + all_link_files = obj_files + self.extra_link_files + seen = set() + all_link_files = [f for f in all_link_files if not (f in seen or seen.add(f))] + if self.extra_link_files: + _vlog().info(f"[链接] 额外库文件: {len(self.extra_link_files)} 个") + for lib in self.extra_link_files: + _vlog().info(f" - {os.path.basename(lib)}") + + if not self.linker_cmd: + _vlog().info("[链接] 未配置链接器") + return + + exe_path = os.path.join(self.output_dir, self.linker_output) if self.linker_output else os.path.join(self.output_dir, 'a.exe') + is_binary = self.linker_output and self.linker_output.endswith('.bin') + + # 检查是否直接输出二进制（使用 --oformat binary / -oformat binary 标志） + is_direct_binary = is_binary and any('oformat' in f for f in self.linker_flags) + + if is_binary and not is_direct_binary: + link_output = exe_path[:-4] + '_pe.exe' + else: + link_output = exe_path + + # 自动复制或生成 linker.ld 到输出目录 + linker_ld_src = os.path.join(os.path.dirname(self.output_dir), 'linker.ld') + linker_ld_dst = os.path.join(self.output_dir, 'linker.ld') + if os.path.isfile(linker_ld_src) and not os.path.isfile(linker_ld_dst): + shutil.copy2(linker_ld_src, linker_ld_dst) + _vlog().info(f" [链接] 使用链接脚本: linker.ld") + elif not os.path.isfile(linker_ld_dst) and any('-T' in f or '-T' in f.replace(',', ' ') for f in self.linker_flags): + default_ld = """ENTRY(_start) + +SECTIONS { + . = 0x100000; + + .text : { + *(.text) + *(.text.*) + } + + .rodata : { + *(.rodata) + *(.rodata.*) + } + + .data : { + *(.data) + *(.data.*) + } + + .bss : { + *(.bss) + *(.bss.*) + } + + /DISCARD/ : { + *(.comment) + *(.note) + *(.eh_frame) + *(.eh_frame_hdr) + *(.reloc) + *(.rela) + *(.debug*) + } +} +""" + with open(linker_ld_dst, 'w', encoding='utf-8', newline='\n') as f: + f.write(default_ld) + _vlog().info(f" [链接] 自动创建默认链接脚本: linker.ld") + + try: + # 将 linker_flags 拆分为非库标志和库标志（-l/-L），库标志放在目标文件之后 + # 这样 Unix 风格链接器才能正确解析库中的符号 + # -Wl, 选项是链接器全局选项，放在目标文件之前 + non_lib_flags = [] + lib_flags = [] + i = 0 + while i < len(self.linker_flags): + flag = self.linker_flags[i] + if flag in ('-l', '-L') and i + 1 < len(self.linker_flags): + # -l xxx / -L xxx 分开的形式 + lib_flags.extend([flag, self.linker_flags[i + 1]]) + i += 2 + elif flag.startswith('-l') or flag.startswith('-L'): + # -lxxx / -Lxxx 合并的形式 + lib_flags.append(flag) + i += 1 + else: + # 其他所有标志（包括 -Wl, 选项）放在目标文件之前 + non_lib_flags.append(flag) + i += 1 + cmd = [self.linker_cmd] + non_lib_flags + ['-o', link_output] + all_link_files + lib_flags + _vlog().info(f" 执行: {' '.join(cmd)}") + result = subprocess.run( + cmd, + capture_output=True, + text=True, + cwd=self.output_dir or '.' + ) + if result.returncode == 0: + if is_binary and not is_direct_binary: + objcopy = shutil.which('llvm-objcopy') or shutil.which('objcopy') + if objcopy: + conv = subprocess.run( + [objcopy, '-O', 'binary', link_output, exe_path], + capture_output=True, text=True + ) + if conv.returncode == 0: + os.remove(link_output) + _vlog().success(f" 生成裸机二进制: {exe_path}") + file_size = os.path.getsize(exe_path) + _vlog().info(f" [大小] {file_size} 字节") + else: + _vlog().error(f" 二进制转换失败: {conv.stderr.strip()}") + sys.exit(1) + else: + _vlog().error(" 找不到 llvm-objcopy，无法生成裸机二进制") + sys.exit(1) + else: + _vlog().success(f" 生成: {exe_path}") + file_size = os.path.getsize(exe_path) + _vlog().info(f" [大小] {file_size} 字节") + if not is_binary: + self._strip_debug_sections(exe_path) + else: + _vlog().error(f" 链接失败: {result.stderr.strip()}") + _vlog().error(f"\n链接失败，立即终止编译。") + sys.exit(1) + except FileNotFoundError: + _vlog().error(f" 找不到链接器: {self.linker_cmd}") + _vlog().error(f"\n找不到链接器，立即终止编译。") + sys.exit(1) + except Exception as e: + _vlog().error(f" {e}") + _vlog().error(f"\n链接异常，立即终止编译。") + sys.exit(1) diff --git a/lib/Projectrans/Utils.py b/lib/Projectrans/Utils.py new file mode 100644 index 0000000..187eaba --- /dev/null +++ b/lib/Projectrans/Utils.py @@ -0,0 +1,231 @@ +from __future__ import annotations + +import hashlib +import os +import ast + +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _ConfigMode +from lib.core.SymbolUtils import AnnotationContainsName + + +def compute_sha1(content: str) -> str: + return hashlib.sha1(content.encode('utf-8')).hexdigest()[:16] + + +# get_file_dependencies 缓存：键 (filepath, src_root)，值 set[str] +# 单次编译中同一文件被 find_reachable_files 和 topological_sort 重复解析， +# 缓存避免重复 open+read+ast.parse+ast.walk（~350 次调用→~175 次） +_FILE_DEPS_CACHE: dict[tuple[str, str], set[str]] = {} + +# Python 源文件 AST 缓存：键 filepath，值 (mtime, content, tree) +# 多处解析同一 .py 文件（get_file_dependencies、used_includes 收集、 +# _collect_inline_symbols 等）共享 parse 结果，避免重复 ast.parse（~18s→~9s） +_FILE_AST_CACHE: dict[str, tuple[float, str, ast.Module]] = {} + + +def _check_annotation_for_export(annotation: ast.AST | None) -> bool: + """递归检查类型注解中是否包含 CExport/CExtern/State（t.State = t.CExtern | t.CExport）""" + if not annotation: + return False + return (AnnotationContainsName(annotation, 'CExport') or + AnnotationContainsName(annotation, 'CExtern') or + AnnotationContainsName(annotation, 'State')) + + +def sha1_file(filepath: str) -> str: + with open(filepath, 'r', encoding='utf-8') as f: + content: str = f.read() + return compute_sha1(content) + + +def parse_python_file(filepath: str) -> tuple[str, ast.Module] | tuple[str, None]: + """读取并解析 Python 文件，带 mtime 校验的 AST 缓存 + + 多处调用共享同一文件的 parse 结果，避免重复 open+read+ast.parse。 + 返回 (content, tree)；文件读取或解析失败返回 (content, None)。 + """ + try: + st: os.stat_result = os.stat(filepath) + except OSError: + return ('', None) + mtime: float = st.st_mtime + cached: tuple[float, str, ast.Module] | None = _FILE_AST_CACHE.get(filepath) + if cached is not None and cached[0] == mtime: + return (cached[1], cached[2]) + try: + with open(filepath, 'r', encoding='utf-8') as f: + content: str = f.read() + except Exception: + return ('', None) + try: + tree: ast.Module = ast.parse(content) + except SyntaxError: + return (content, None) + _FILE_AST_CACHE[filepath] = (mtime, content, tree) + return (content, tree) + + +def get_file_dependencies(filepath: str, src_root: str) -> set[str]: + """解析 Python 文件的导入依赖（带缓存） + + 单次编译中 find_reachable_files_from_entries 和 topological_sort_files + 会对同一批文件重复调用，缓存避免重复 ast.parse（~8.6s→~4.3s）。 + """ + cache_key: tuple[str, str] = (filepath, src_root) + cached: set[str] | None = _FILE_DEPS_CACHE.get(cache_key) + if cached is not None: + return cached + dependencies: set[str] = set() + content, tree = parse_python_file(filepath) + if tree is not None: + try: + for node in ast.walk(tree): + if isinstance(node, ast.Import): + for alias in node.names: + module: str = alias.name + dependencies.add(module) + elif isinstance(node, ast.ImportFrom): + if node.module: + module: str = node.module + if node.level > 0: + # from .module import name → 依赖 pkg.module（模块本身） + pkg_dir: str = os.path.dirname(filepath) + pkg_rel: str = os.path.relpath(pkg_dir, src_root).replace(os.sep, '.').replace('/', '.') + if pkg_rel == '.': + pkg_rel = '' + dep_module: str + if pkg_rel: + dep_module = f'{pkg_rel}.{module}' + else: + dep_module = module + dependencies.add(dep_module) + else: + if not module.startswith('_'): + dependencies.add(module) + elif node.level > 0 and node.names: + # from . import name → 依赖 pkg.name（子模块） + for alias in node.names: + pkg_dir: str = os.path.dirname(filepath) + pkg_rel: str = os.path.relpath(pkg_dir, src_root).replace(os.sep, '.').replace('/', '.') + if pkg_rel == '.': + pkg_rel = '' + dep_module: str = f'{pkg_rel}.{alias.name}' if pkg_rel else alias.name + dependencies.add(dep_module) + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"解析文件导入依赖失败: {_e}", "Exception") + _FILE_DEPS_CACHE[cache_key] = dependencies + return dependencies + + +def clear_file_deps_cache() -> None: + """清除 get_file_dependencies 和 AST 缓存（跨编译或文件变更时调用）""" + _FILE_DEPS_CACHE.clear() + _FILE_AST_CACHE.clear() + + +def find_reachable_files_from_entries(src_root: str, entry_files: list[str]) -> set[str]: + """从入口文件出发，找出所有可达的 .py 文件（通过导入关系）""" + all_files: dict[str, str] = {} # module_name -> filepath + for root, dirs, files in os.walk(src_root): + dirs[:] = [d for d in dirs if d not in ('__pycache__',)] + for file in files: + if file.endswith('.py'): + filepath: str = os.path.join(root, file) + rel: str = os.path.relpath(filepath, src_root) + module: str = os.path.splitext(rel)[0].replace(os.sep, '.').replace('/', '.') + if filepath.endswith('__init__.py'): + all_files[module] = filepath + parent_module: str = module.rsplit('.', 1)[0] if '.' in module else module + if parent_module not in all_files or not all_files[parent_module].endswith('__init__.py'): + all_files[parent_module] = filepath + elif module not in all_files: + all_files[module] = filepath + top_module: str = module.split('.')[0] + if top_module not in all_files: + all_files[top_module] = filepath + + reachable: set[str] = set() + queue: list[str] = list(entry_files) + + # BFS trace: record each file processed and unresolved deps + _bfs_trace: list[str] = [] + _unresolved_deps: list[tuple[str, str]] = [] # (current_file, dep) + + while queue: + current: str = queue.pop(0) + if current in reachable: + continue + reachable.add(current) + deps: set[str] = get_file_dependencies(current, src_root) + _bfs_trace.append(f" {os.path.basename(current)} deps={len(deps)}") + for dep in deps: + if dep in all_files and all_files[dep] not in reachable: + queue.append(all_files[dep]) + else: + _matched = False + for key, val in all_files.items(): + if key.endswith('.' + dep) or key == dep: + if val not in reachable: + queue.append(val) + _matched = True + break + if not _matched: + _unresolved_deps.append((os.path.basename(current), dep)) + + return reachable + + +def topological_sort_files(py_files: list[str], src_root: str) -> list[str]: + """对 Python 文件进行拓扑排序，确保依赖模块先被处理""" + # 构建模块名到文件路径的映射 + module_to_file: dict[str, str] = {} + for filepath in py_files: + rel: str = os.path.relpath(filepath, src_root) + module: str = os.path.splitext(rel)[0].replace(os.sep, '.').replace('/', '.') + module_to_file[module] = filepath + # 也添加顶层模块名 + top_module: str = module.split('.')[0] + if top_module not in module_to_file: + module_to_file[top_module] = filepath + + # 构建依赖图 + graph: dict[str, set[str]] = {f: set() for f in py_files} + for filepath in py_files: + deps: set[str] = get_file_dependencies(filepath, src_root) + for dep in deps: + if dep in module_to_file: + dep_file: str = module_to_file[dep] + if dep_file != filepath: # 避免自依赖 + graph[filepath].add(dep_file) + + # 拓扑排序 (Kahn's algorithm) + in_degree: dict[str, int] = {f: 0 for f in py_files} + for f, deps in graph.items(): + for dep in deps: + if dep in in_degree: + in_degree[f] += 1 + + queue: list[str] = [f for f, degree in in_degree.items() if degree == 0] + sorted_files: list[str] = [] + + while queue: + # 按字母顺序处理同级别的文件，确保确定性 + queue.sort() + current: str = queue.pop(0) + sorted_files.append(current) + + for f, deps in graph.items(): + if current in deps: + in_degree[f] -= 1 + if in_degree[f] == 0: + queue.append(f) + + # 如果有环，添加剩余文件 + if len(sorted_files) < len(py_files): + remaining: list[str] = [f for f in py_files if f not in sorted_files] + sorted_files.extend(remaining) + + return sorted_files diff --git a/lib/Projectrans/__init__.py b/lib/Projectrans/__init__.py new file mode 100644 index 0000000..501e098 --- /dev/null +++ b/lib/Projectrans/__init__.py @@ -0,0 +1,31 @@ +"""Projectrans 包 - 工程级 TransPyC 翻译器""" +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） + +from lib.core.VLogger import Logger, LogLevel, set_logger + +# 创建全局日志实例 +logger = Logger( + name="TransPyC", + log_file="build.log", + level=LogLevel.INFO, + use_colors=True +) +set_logger(logger) + +from lib.Projectrans.Utils import ( + compute_sha1, _check_annotation_for_export, sha1_file, + get_file_dependencies, find_reachable_files_from_entries, + topological_sort_files +) +from lib.Projectrans.Phase1Generator import Phase1Generator +from lib.Projectrans.DeclarationGenerator import DeclarationGenerator +from lib.Projectrans.Phase2Translator import Phase2Translator, _parallel_translate_worker +from lib.Projectrans.Config import save_sha1_map, Load_sha1_map, Load_project_config, resolve_paths, main + +__all__ = [ + 'compute_sha1', '_check_annotation_for_export', 'sha1_file', + 'get_file_dependencies', 'find_reachable_files_from_entries', + 'topological_sort_files', 'Phase1Generator', 'DeclarationGenerator', + 'Phase2Translator', '_parallel_translate_worker', + 'save_sha1_map', 'Load_sha1_map', 'Load_project_config', 'resolve_paths', 'main' +] diff --git a/lib/Shell/Config.py b/lib/Shell/Config.py new file mode 100644 index 0000000..5bfcfc7 --- /dev/null +++ b/lib/Shell/Config.py @@ -0,0 +1,8 @@ +from __future__ import annotations + + +class Config: + """配置类""" + def __init__(self) -> None: + self.debug: bool = False + self.mode: str = 'relaxed' # 'relaxed' 或 'strict' diff --git a/lib/Shell/Serializer.py b/lib/Shell/Serializer.py new file mode 100644 index 0000000..f87409e --- /dev/null +++ b/lib/Shell/Serializer.py @@ -0,0 +1,61 @@ +from __future__ import annotations + +import json +import struct +from typing import Any + + +def SerializeSymbolTable(SymbolTable: dict[str, Any]) -> bytes: + """将符号表序列化为二进制格式 + + 格式: + - 4字节: JSON数据长度（小端序） + - N字节: JSON数据（UTF-8编码） + + Args: + SymbolTable: 符号表字典 + + Returns: + 二进制字节数据 + """ + JsonData: bytes = json.dumps(SymbolTable, ensure_ascii=False).encode('utf-8') + length: int = len(JsonData) + # 使用4字节小端序整数表示长度 + header: bytes = struct.pack(' dict[str, Any]: + """从二进制数据反序列化符号表 + + Args: + data: 二进制字节数据 + + Returns: + 符号表字典 + """ + if len(data) < 4: + raise ValueError("Invalid symbin file: data too short") + # 解析4字节长度头（小端序） + length: int = struct.unpack(' None: + """初始化SymbolFile对象 + + Args: + file: 文件路径 + string: 代码字符串 + type: 文件类型，支持"c"或"py" + encoding: 文件编码 + """ + self.FilePath: str | None = file + self.CodeString: str | None = string + self.FileType: str | None = type + self.encoding: str = encoding + self.symbols: dict[str, Any] = {} diff --git a/lib/Shell/TransPyC.py b/lib/Shell/TransPyC.py new file mode 100644 index 0000000..2aa92e7 --- /dev/null +++ b/lib/Shell/TransPyC.py @@ -0,0 +1,645 @@ +from __future__ import annotations + +import sys +import os +import ast +import shlex +import datetime +import tempfile +import subprocess +from typing import Any +from lib.includes import c +from lib.includes import t +from lib.constants.config import ( + DEFAULT_INPUT_FILE, DEFAULT_OUTPUT_FILE, + DEFAULT_COMPILE_COMMAND, + DEFAULT_COMPILE_FLAGS, + HELP_MESSAGE, DEFAULT_METADATA +) +from lib.utils.helpers import ( + DetectFileType, ExecuteCommand, + AppendFileContent +) +from lib.core.VLogger import info, warning, error, success, debug +from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.Shell.Serializer import SerializeSymbolTable, SymbolFile +from lib.Shell.Config import Config + + +class TransPyC: + """TransPyC 主类""" + + def __init__(self, code: str | None = None, debug: bool = False, triple: str | None = None, datalayout: str | None = None) -> None: + """初始化TransPyC对象 + + Args: + code: 代码字符串 + debug: 调试模式 + triple: LLVM target triple (e.g. "x86_64-pc-linux-gnu") + datalayout: LLVM data layout string + """ + self.Args: dict[str, Any] = {} + self.HeaderFiles: list[str] = [] + self.HelperFiles: list[str] = [] + self.AnnotationFiles: list[str] = [] + self.translator: Translator = Translator() + self.code: str | None = code + self.translator.Content = code + self.config: Config = Config() + self.translator.config = self.config + self.config.debug = debug + self.SymbolFiles: list[SymbolFile] = [] + self.SliceLevel: int | str = 3 + self.SliceCount: int = 0 + self.SliceInfos: list[Any] = [] + self.metadata: dict[str, Any] = DEFAULT_METADATA.copy() + self.triple: str | None = triple + self.datalayout: str | None = datalayout + self._source_module_sig_files: dict[str, Any] = {} + + @staticmethod + def PreProcessSymbol(SymbolFile: SymbolFile, debug: bool = False) -> bytes | tuple[bytes, str]: + """预处理符号文件，提取符号表并返回二进制数据 + + Args: + SymbolFile: SymbolFile对象，包含文件路径和类型信息 + debug: 是否启用调试模式，如果为True，返回 (bytes, DebugInfo) + 对于Python文件，DebugInfo包含p2c日志 + 对于C文件，DebugInfo包含基本处理信息 + + Returns: + 如果debug为False: bytes - 序列化后的符号表二进制数据 + 如果debug为True: (bytes, str) - (二进制数据, 调试日志) + """ + translator: Translator = Translator() + DebugLogs: list[str] | str = [] + + try: + # 读取文件内容 + if SymbolFile.FilePath: + with open(SymbolFile.FilePath, 'r', encoding=SymbolFile.encoding) as f: + content: str = f.read() + elif SymbolFile.CodeString: + content = SymbolFile.CodeString + else: + raise ValueError("SymbolFile must have either FilePath or CodeString") + + # 根据文件类型处理 + FileType: str | None = SymbolFile.FileType + if not FileType and SymbolFile.FilePath: + # 从文件路径推断类型 + if SymbolFile.FilePath.endswith('.py'): + FileType = 'py' + elif SymbolFile.FilePath.endswith('.c') or SymbolFile.FilePath.endswith('.h'): + FileType = 'c' + + if FileType == 'py': + # Python文件处理 + translator.OriginalLines = content.split('\n') + translator.Content = content + + # 设置调试文件（如果需要） + if debug: + DebugFile: str = SymbolFile.FilePath.replace('.py', '.p2c') if SymbolFile.FilePath else 'debug.p2c' + translator.SetDebugFile(DebugFile) + # 清空调试文件 + with open(DebugFile, 'w', encoding=SymbolFile.encoding) as f: + f.write('') + + # 解析AST并提取符号 + tree: ast.Module = ast.parse(content) + + if debug: + with open(DebugFile, 'a', encoding=SymbolFile.encoding) as f: + f.write('=== AST Tree (Compact) ===\n') + f.write(ast.dump(tree)) + f.write('\n\n') + + # 遍历AST提取符号 + for node in ast.iter_child_nodes(tree): + if isinstance(node, ast.ClassDef): + translator.SymbolTable[node.name] = {'type': 'struct', 'members': {}} + for item in node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + VarName: str = item.target.id + TypeInfo: CTypeInfo | None = translator.TypeMergeHandler.GetCTypeInfo(item.annotation) if item.annotation else None + VarType: str = TypeInfo.ToString() if TypeInfo else 'unknown' + IsPtr: bool = TypeInfo.IsPtr if TypeInfo else False + translator.SymbolTable[node.name]['members'][VarName] = { + 'type': VarType, + 'IsPtr': IsPtr + } + elif isinstance(node, ast.FunctionDef): + translator.SymbolTable[node.name] = {'type': 'function'} + elif isinstance(node, ast.AnnAssign): + if isinstance(node.target, ast.Name): + VarName: str = node.target.id + VarType: str = 'unknown' + IsPtr: bool = False + if node.annotation: + try: + TypeInfo: CTypeInfo | None = translator.TypeMergeHandler.GetCTypeInfo(node.annotation) + VarType = TypeInfo.ToString() if TypeInfo else 'unknown' + IsPtr = TypeInfo.IsPtr if TypeInfo else False + except: + pass + translator.SymbolTable[VarName] = { + 'type': 'variable', + 'declared_type': VarType, + 'IsPtr': IsPtr + } + + if debug: + with open(DebugFile, 'a', encoding=SymbolFile.encoding) as f: + f.write('=== Symbol Table ===\n') + f.write(str(translator.SymbolTable)) + f.write('\n\n') + # 读取调试日志 + with open(DebugFile, 'r', encoding=SymbolFile.encoding) as f: + DebugLogs = f.read() + + elif FileType == 'c': + # C文件处理 - 使用现有的ParseHelperFiles逻辑 + translator.ParseHelperFiles([SymbolFile.FilePath], SymbolFile.encoding) + + if debug: + DebugLogs = f"=== C File Symbol Extraction ===\n" + DebugLogs += f"File: {SymbolFile.FilePath}\n" + DebugLogs += f"=== Symbol Table ===\n" + DebugLogs += str(translator.SymbolTable) + DebugLogs += "\n\n" + else: + raise ValueError(f"Unsupported file type: {FileType}") + + # 序列化符号表 + BinaryData: bytes = SerializeSymbolTable(translator.SymbolTable) + + if debug: + return BinaryData, DebugLogs + return BinaryData + + except Exception as e: + error_msg: str = f"Error processing symbol file: {e}" + if debug: + return b'', error_msg + raise RuntimeError(error_msg) + + def ParseArgs(self) -> None: + """解析命令行参数""" + I: int = 1 + while I < len(sys.argv): + if sys.argv[I] == '-f': + if I + 1 < len(sys.argv): + self.Args['Input'] = sys.argv[I + 1] + I += 2 + else: + error('-f requires an argument') + sys.exit(1) + elif sys.argv[I] == '-o': + if I + 1 < len(sys.argv): + self.Args['Output'] = sys.argv[I + 1] + I += 2 + else: + error('-o requires an argument') + sys.exit(1) + elif sys.argv[I] == '-wh': + if I + 1 < len(sys.argv): + self.HeaderFiles = sys.argv[I + 1:] + break + else: + error('-wh requires arguments') + sys.exit(1) + elif sys.argv[I] == '-debug': + if I + 1 < len(sys.argv): + self.Args['Debug'] = sys.argv[I + 1] + I += 2 + else: + error('-debug requires an argument') + sys.exit(1) + elif sys.argv[I] == '-cc': + # 编译命令 + if I + 1 < len(sys.argv): + self.Args['CompileCommand'] = sys.argv[I + 1] + I += 2 + else: + error('-cc requires an argument') + sys.exit(1) + elif sys.argv[I] == '-cflags': + # 编译标志 + if I + 1 < len(sys.argv): + self.Args['CompileFlags'] = sys.argv[I + 1] + I += 2 + else: + error('-cflags requires an argument') + sys.exit(1) + elif sys.argv[I] == '-run': + # 是否运行生成的程序 + self.Args['Run'] = True + I += 1 + elif sys.argv[I] == '-args': + # 运行时参数 + if I + 1 < len(sys.argv): + self.Args['RunArgs'] = sys.argv[I + 1] + I += 2 + else: + error('-args requires an argument') + sys.exit(1) + elif sys.argv[I] == '-level' or sys.argv[I] == '-l': + if I + 1 < len(sys.argv): + LevelArg: str = sys.argv[I + 1] + if LevelArg.isdigit(): + self.Args['SliceLevel'] = int(LevelArg) + elif LevelArg == 's': + self.Args['SliceLevel'] = 's' + else: + self.Args['SliceLevel'] = 3 + I += 1 + else: + self.Args['SliceLevel'] = 3 + elif sys.argv[I] == '-h': + # 辅助文件，用于解析符号信息 + if I + 1 < len(sys.argv): + # 收集所有后续参数作为辅助文件，直到遇到下一个以-开头的参数 + while I + 1 < len(sys.argv) and not sys.argv[I + 1].startswith('-'): + self.HelperFiles.append(sys.argv[I + 1]) + I += 1 + I += 1 + else: + error('-h requires arguments') + sys.exit(1) + elif sys.argv[I] == '-presym': + # 预处理符号文件，生成.symbin文件 + # 格式: -presym -o [-debug ] + if I + 1 < len(sys.argv): + InputFile: str = sys.argv[I + 1] + I += 2 + + # 解析可选参数 + OutputFile: str | None = None + DebugFile: str | None = None + while I < len(sys.argv) and sys.argv[I].startswith('-'): + if sys.argv[I] == '-o': + if I + 1 < len(sys.argv): + OutputFile = sys.argv[I + 1] + I += 2 + else: + error('-o requires an argument') + sys.exit(1) + elif sys.argv[I] == '-debug': + if I + 1 < len(sys.argv): + DebugFile = sys.argv[I + 1] + I += 2 + else: + error('-debug requires an argument') + sys.exit(1) + else: + break + + # 如果没有指定输出文件，使用默认名称 + if not OutputFile: + OutputFile = InputFile.replace('.py', '.symbin').replace('.c', '.symbin') + + # 执行预处理 + self.Args['PreSym'] = { + 'input': InputFile, + 'output': OutputFile, + 'debug': DebugFile + } + else: + error('-presym requires an argument') + sys.exit(1) + elif sys.argv[I] == '-e' or sys.argv[I] == '--encoding': + if I + 1 < len(sys.argv): + self.Args['Encoding'] = sys.argv[I + 1] + I += 2 + else: + error('-e/--encoding requires an argument') + sys.exit(1) + elif sys.argv[I] == '--triple': + if I + 1 < len(sys.argv): + self.triple = sys.argv[I + 1] + I += 2 + else: + error('--triple requires an argument') + sys.exit(1) + elif sys.argv[I] == '--datalayout': + if I + 1 < len(sys.argv): + self.datalayout = sys.argv[I + 1] + I += 2 + else: + error('--datalayout requires an argument') + sys.exit(1) + else: + error(f'Unknown argument {sys.argv[I]}') + sys.exit(1) + + # 检查是否有预处理符号文件的请求，如果有则跳过输入输出检查 + if 'PreSym' not in self.Args and ('Input' not in self.Args or 'Output' not in self.Args): + error('缺少参数') + info(HELP_MESSAGE) + sys.exit(1) + + def CompileAndRun(self, OutputFile: str) -> None: + """编译并运行生成的C代码""" + # 获取编译命令 + CompileCmd: str = self.Args.get('CompileCommand', DEFAULT_COMPILE_COMMAND) + CompileFlags: str = self.Args.get('CompileFlags', DEFAULT_COMPILE_FLAGS) + + # 构建编译命令（使用列表形式，避免 shell 注入） + FullCompileCmd: list[str] = shlex.split(CompileCmd) + shlex.split(CompileFlags) + [OutputFile, '-o', f'{OutputFile}.exe'] + info(f'编译命令: {" ".join(FullCompileCmd)}') + + # 执行编译命令 + try: + CompileResult: subprocess.CompletedProcess[str] | None = ExecuteCommand(FullCompileCmd) + if CompileResult: + success('编译成功!') + + # 检查是否需要运行 + if self.Args.get('Run', False): + RunArgs: str = self.Args.get('RunArgs', '') + RunCmd: list[str] = [f'{OutputFile}.exe'] + (shlex.split(RunArgs) if RunArgs else []) + info(f'运行: {" ".join(RunCmd)}') + + # 执行运行命令 + RunResult: subprocess.CompletedProcess[str] | None = ExecuteCommand(RunCmd) + if RunResult: + info('执行输出:') + info(RunResult.stdout) + if RunResult.stderr: + warning('执行错误:') + warning(RunResult.stderr) + except Exception as e: + error(f'编译失败: {e}') + + def WriteDebugInfo(self, content: str) -> None: + """写入调试信息到指定文件""" + # 获取调试文件路径 + if 'Debug' in self.Args: + DebugFile: str = self.Args['Debug'] + else: + DebugFile: str = self.Args.get('Output', '').replace('.c', '.p2c') + + if DebugFile: + AppendFileContent(DebugFile, content) + + def PythonToC(self, InputFile: str, OutputFile: str) -> None: + """Python到C的转换""" + # 获取编码参数 + encoding: str = self.Args.get('Encoding', 'utf-8') + + # 设置调试文件路径（使用 .p2c 扩展名） + if 'Debug' in self.Args: + DebugFile: str = self.Args['Debug'] + else: + # 默认使用 .p2c 文件 + DebugFile: str = OutputFile.replace('.c', '.p2c') + + # 设置翻译器的调试文件 + self.translator.SetDebugFile(DebugFile) + + # 先清空调试文件 + with open(DebugFile, 'w', encoding=encoding) as f: + f.write('') + + # 解析辅助文件，提取符号信息 + if self.HelperFiles: + self.translator.ParseHelperFiles(self.HelperFiles, encoding) + + # 加载注解文件，提取类型定义 + if self.AnnotationFiles: + self.translator.LoadAnnotationFiles(self.AnnotationFiles) + + # 获取编码参数 + encoding = self.Args.get('Encoding', 'utf-8') + with open(InputFile, 'r', encoding=encoding) as F: + Content: str = F.read() + + # 保存原始代码行 + self.translator.OriginalLines = Content.split('\n') + self.translator.Content = Content + + # 解析Python代码为AST + Tree: ast.Module = ast.parse(Content) + + # 解析文件以提取类型信息（用于 EmbeddedAssignments） + self.translator.ParsePythonFile(InputFile, encoding) + + # 写入AST树信息（压缩格式） + with open(DebugFile, 'a', encoding=encoding) as f: + f.write('=== AST Tree (Compact) ===\n') + f.write(ast.dump(Tree)) + f.write('\n\n') + + Target: str = 'llvm' + CCode: str = self.translator.GenerateCCode(Tree, target=Target) + + timestamp: str = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') + filename: str = os.path.basename(OutputFile) + + encoding = self.Args.get('Encoding', 'utf-8') + with open(OutputFile, 'w', encoding=encoding) as F: + F.write(CCode) + F.write('\n') + + def CToPython(self, InputFile: str, OutputFile: str, HeaderFiles: list[str]) -> None: + """C到Python的转换（暂时禁用）""" + warning("C到Python的转换功能暂时禁用，请使用Python到C的转换功能。") + + def AddSymbol(self, SymbolFiles: SymbolFile | list[SymbolFile]) -> None: + """添加符号文件 + + Args: + SymbolFiles: 符号文件或符号文件列表 + """ + if isinstance(SymbolFiles, list): + self.SymbolFiles.extend(SymbolFiles) + else: + self.SymbolFiles.append(SymbolFiles) + + # 收集文件路径和编码 + HelperFiles: list[str] = [] + encodings: list[str] = [] + for SymbolFile in self.SymbolFiles: + FilePath: str | None = SymbolFile.FilePath + encoding: str = SymbolFile.encoding + if FilePath: + HelperFiles.append(FilePath) + encodings.append(encoding) + + # 解析辅助文件 + if HelperFiles: + # 暂时使用第一个文件的编码 + encoding = encodings[0] if encodings else 'utf-8' + self.translator.ParseHelperFiles(HelperFiles, encoding) + + def Convert(self, OutputFilename: str | None = None, SourceFilename: str | None = None, target: str = 'llvm') -> str | tuple[str, dict[str, Any]] | None: + """转换代码 + + Args: + OutputFilename: 输出文件名（用于模板渲染） + SourceFilename: 源文件路径（用于设置CurrentFile，影响导入模块的查找） + target: 目标代码类型，仅支持 'llvm' + + Returns: + 如果debug是字符串，则返回生成的代码 + 如果debug是True，则返回生成的代码和调试信息 + """ + if not self.code: + error('未提供代码') + return None + + from lib.core.Handles.HandlesImports import ImportHandle + ImportHandle._struct_Load_cache.clear() + + # 如果提供了 SourceFilename，先设置 CurrentFile + # 这样导入模块时会相对于源文件所在目录查找 + if SourceFilename: + self.translator.CurrentFile = SourceFilename + + # 加载注解文件，提取类型定义 + if self.AnnotationFiles: + self.translator.LoadAnnotationFiles(self.AnnotationFiles) + + # 保存原始代码行 + self.translator.OriginalLines = self.code.split('\n') + + # 解析Python代码为AST + Tree: ast.Module = ast.parse(self.code) + + # 从源代码直接预扫描 import 语句，注册别名 + # 注意：不能依赖 Tree 中的 import 节点，因为 ParsePythonFile 可能修改了 AST + if not getattr(self.translator, '_ImportedModules', None): + self.translator._ImportedModules = set() + _prescan_tree: ast.Module = ast.parse(self.code) + for _node in ast.iter_child_nodes(_prescan_tree): + if isinstance(_node, ast.Import): + for _alias in _node.names: + _name: str = _alias.name + if _name in ('c', 't'): + continue + self.translator._ImportedModules.add(_name) + if _alias.asname: + self.translator.SymbolTable.import_aliases[_alias.asname] = _name + elif isinstance(_node, ast.ImportFrom): + _module_name: str | None = _node.module + if _module_name and _module_name not in ('c', 't'): + self.translator._ImportedModules.add(_module_name) + for _alias in _node.names: + if _alias.asname: + self.translator.SymbolTable.import_aliases[_alias.asname] = f"{_module_name}.{_alias.name}" + + # 解析代码以提取类型信息（用于 EmbeddedAssignments 和 TypedefAssignments） + # 创建临时文件路径用于解析 + with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False, encoding='utf-8') as f: + f.write(self.code) + TempFile: str = f.name + try: + # UpdateCurrentFile=False 表示不修改 CurrentFile + if SourceFilename: + self.translator.ParsePythonFile(TempFile, UpdateCurrentFile=False) + else: + self.translator.ParsePythonFile(TempFile) + finally: + os.unlink(TempFile) + + timestamp: str = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') + + if self.triple: + self.translator.triple = self.triple + if self.datalayout: + self.translator.datalayout = self.datalayout + + # 生成代码 + code: str = self.translator.GenerateCCode(Tree, target='llvm') + filename: str = os.path.basename(OutputFilename) if OutputFilename else 'generated.ll' + + # 处理调试信息 + if isinstance(self.config.debug, str): + # 如果debug是字符串，写入调试信息到文件 + with open(self.config.debug, 'w', encoding='utf-8') as f: + f.write('=== Symbol Table ===\n') + f.write(str(self.translator.SymbolTable) + '\n\n') + f.write('=== AST Tree ===\n') + f.write(ast.dump(Tree, indent=2) + '\n\n') + f.write('=== Export Table ===\n') + f.write(self.translator.Exportable.to_json() + '\n\n') + f.write('=== Generated Code ===\n') + f.write(code + '\n') + return code + elif self.config.debug: + # 如果debug是True，返回生成的代码和调试信息 + DebugInfo: dict[str, Any] = { + 'SymbolTable': self.translator.SymbolTable, + 'ast_tree': ast.dump(Tree, indent=2), + 'Exportable': self.translator.Exportable.to_dict(), + 'generated_code': code + } + return code, DebugInfo + else: + # 如果debug是False，只返回生成的代码 + return code + + def Run(self) -> None: + """主运行函数""" + # 检查是否有命令行参数 + if len(sys.argv) > 1: + # 有命令行参数，使用 ParseArgs 方法解析 + self.ParseArgs() + + # 检查是否有预处理符号文件的请求 + if 'PreSym' in self.Args: + PresymConfig: dict[str, Any] = self.Args['PreSym'] + InputFile: str = PresymConfig['input'] + OutputFile: str = PresymConfig['output'] + DebugFile: str | None = PresymConfig['debug'] + + # 获取编码参数 + encoding: str = self.Args.get('Encoding', 'utf-8') + + # 推断文件类型 + FileType: str | None = 'py' if InputFile.endswith('.py') else 'c' if InputFile.endswith('.c') else None + + # 创建 SymbolFile 对象，传入编码参数 + SymbolFile: SymbolFile = SymbolFile(file=InputFile, type=FileType, encoding=encoding) + + # 调用 PreProcessSymbol + if DebugFile: + BinaryData: bytes + DebugInfo: str + BinaryData, DebugInfo = TransPyC.PreProcessSymbol(SymbolFile, debug=True) + # 写入调试文件 + with open(DebugFile, 'w', encoding=encoding) as f: + f.write(DebugInfo) + info(f"调试信息写入: {DebugFile}") + else: + BinaryData: bytes = TransPyC.PreProcessSymbol(SymbolFile, debug=False) + + # 写入二进制符号文件 + with open(OutputFile, 'wb') as f: + f.write(BinaryData) + + success(f"符号文件生成: {OutputFile}") + return + + InputFile: str = self.Args['Input'] + OutputFile: str = self.Args['Output'] + else: + # 没有命令行参数，使用默认的测试文件名称 + InputFile: str = DEFAULT_INPUT_FILE + OutputFile: str = DEFAULT_OUTPUT_FILE + info('使用默认测试文件: test.py -> test.c') + + FileType: str = DetectFileType(InputFile) + + if FileType == '.py': + self.PythonToC(InputFile, OutputFile) + # 检查是否需要编译和运行 + if 'CompileCommand' in self.Args or self.Args.get('Run', False): + self.CompileAndRun(OutputFile) + elif FileType == '.c': + self.CToPython(InputFile, OutputFile, self.HeaderFiles) + else: + error(f'不支持的文件类型: {FileType}') + sys.exit(1) diff --git a/lib/Shell/__init__.py b/lib/Shell/__init__.py new file mode 100644 index 0000000..ffec9e5 --- /dev/null +++ b/lib/Shell/__init__.py @@ -0,0 +1,6 @@ +"""Shell 包 - TransPyC 命令行入口""" +from lib.Shell.Serializer import SerializeSymbolTable, deSerializeSymbolTable, SymbolFile +from lib.Shell.Config import Config +from lib.Shell.TransPyC import TransPyC + +__all__ = ['SerializeSymbolTable', 'deSerializeSymbolTable', 'SymbolFile', 'Config', 'TransPyC'] diff --git a/lib/StubGen/Config.py b/lib/StubGen/Config.py new file mode 100644 index 0000000..4e3b135 --- /dev/null +++ b/lib/StubGen/Config.py @@ -0,0 +1,43 @@ +"""StubGen 配置模块""" +from __future__ import annotations +import json +from dataclasses import dataclass, field +from typing import List, Dict, Optional + + +@dataclass +class StubGenConfig: + """存根生成器配置""" + InputDir: Optional[str] = None + OutputDir: str = "./stubs" + InputFiles: List[str] = field(default_factory=list) + IncludePatterns: List[str] = field(default_factory=lambda: ["*.h", "*.c", "*.py"]) + ExcludePatterns: List[str] = field(default_factory=list) + TypeMappings: Dict[str, str] = field(default_factory=dict) + PreserveStructure: bool = True # 保持目录结构 + GenerateGuards: bool = True # 生成宏守卫 + verbose: bool = False + DryRun: bool = False + + @classmethod + def from_dict(cls, data: dict) -> StubGenConfig: + """从字典创建配置""" + return cls( + InputDir=data.get('InputDir'), + OutputDir=data.get('OutputDir', './stubs'), + InputFiles=data.get('InputFiles', []), + IncludePatterns=data.get('IncludePatterns', ['*.h', '*.c', '*.py']), + ExcludePatterns=data.get('ExcludePatterns', []), + TypeMappings=data.get('TypeMappings', {}), + PreserveStructure=data.get('PreserveStructure', True), + GenerateGuards=data.get('GenerateGuards', True), + verbose=data.get('verbose', False), + DryRun=data.get('DryRun', False), + ) + + @classmethod + def from_file(cls, FilePath: str) -> StubGenConfig: + """从 JSON 文件加载配置""" + with open(FilePath, 'r', encoding='utf-8') as f: + data: dict = json.load(f) + return cls.from_dict(data) \ No newline at end of file diff --git a/lib/StubGen/Converter.py b/lib/StubGen/Converter.py new file mode 100644 index 0000000..d7ee5f7 --- /dev/null +++ b/lib/StubGen/Converter.py @@ -0,0 +1,983 @@ +"""StubGen 转换器模块 - 将 Python 文件转换为存根格式""" +from __future__ import annotations +import ast +import re +import fnmatch +from typing import List, Dict +from lib.core.SymbolUtils import IsListAnnotation + + +class PythonToStubConverter: + """将 Python 文件转换为存根格式""" + + # __include 别名映射表 {alias: ModuleName} + IncludeAliasMap: Dict[str, str] = {} + + # 变量排除列表 - 匹配这些模式的变量不会被添加到存根文件 + VarExcludeList: List[str] = [] + + # 宏排除列表 - 匹配这些模式的宏不会被添加到存根文件 + MacroExcludeList: List[str] = [] + + @classmethod + def SetIncludeAliasMap(cls, alias_map: Dict[str, str]) -> None: + """设置 __include 别名映射表""" + cls.IncludeAliasMap = alias_map + + @classmethod + def SetVarExcludeList(cls, exclude_list: List[str]) -> None: + """设置变量排除列表""" + cls.VarExcludeList = exclude_list + + @classmethod + def SetMacroExcludeList(cls, exclude_list: List[str]) -> None: + """设置宏排除列表""" + cls.MacroExcludeList = exclude_list + + @classmethod + def _ShouldExcludeVar(cls, VarName: str) -> bool: + """检查变量是否应该被排除""" + for pattern in cls.VarExcludeList: + if fnmatch.fnmatch(VarName, pattern): + return True + return False + + @classmethod + def _ShouldExcludeMacro(cls, MacroName: str) -> bool: + """检查宏是否应该被排除""" + for pattern in cls.MacroExcludeList: + if fnmatch.fnmatch(MacroName, pattern): + return True + return False + + @staticmethod + def convert(PyContent: str, ModuleName: str) -> str: + """将 Python 代码转换为存根格式""" + lines: list[str] = PyContent.split('\n') + StubLines: list[str] = [] + + # 添加文件头 + StubLines.append('"""') + StubLines.append(f'Auto-generated Python stub file from {ModuleName}.py') + StubLines.append(f'Module: {ModuleName}') + StubLines.append('"""') + StubLines.append('') + + # 解析 Python 代码，检查是否已经导入了 t 和 c + HasImportT: bool = False + HasImportC: bool = False + try: + tree: ast.Module = ast.parse(PyContent) + for node in tree.body: + if isinstance(node, ast.Import): + for alias in node.names: + if alias.name == 't': + HasImportT = True + elif alias.name == 'c': + HasImportC = True + except: + pass + + # 添加默认导入（如果源代码中没有） + if not HasImportT: + StubLines.append('import t') + if not HasImportC: + StubLines.append('import c') + if not HasImportT or not HasImportC: + StubLines.append('') + + # 添加 c.CPragma("once") + # StubLines.append('c.CPragma("once")') + StubLines.append('') + + # 生成文件级别宏守卫名称 + #FileGuardName = f'__{ModuleName.upper()}_DEFINE__' + + # 添加文件开头宏守卫 + #StubLines.append(f'c.CIfndef({FileGuardName})') + #StubLines.append(f'{FileGuardName}: t.CDefine') + #StubLines.append('') + + # 解析 Python 代码，提取类型定义 + try: + tree = ast.parse(PyContent) + + # 第一遍扫描：收集 Postdefinition 变量 + PostdefVars: dict[str, list[tuple[str, str, ast.AnnAssign]]] = {} # {ClassName: [(VarName, TypeStr, node), ...]} + for node in tree.body: + if isinstance(node, ast.AnnAssign) and isinstance(node.target, ast.Name): + TypeStr: str = PythonToStubConverter._GetTypeString(node.annotation) + # 检查是否是 Postdefinition 类型 + if 't.Postdefinition' in TypeStr: + # 提取引用的类名 + ClassName: str | None = PythonToStubConverter._ExtractPostdefClass(node.annotation) + if ClassName: + if ClassName not in PostdefVars: + PostdefVars[ClassName] = [] + PostdefVars[ClassName].append((node.target.id, TypeStr, node)) + + # 按原始顺序处理节点，保持代码的原始顺序 + PrevType: str | None = None # 记录前一个节点的类型 + + for node in tree.body: + CurrentType: str | None = None + + if isinstance(node, ast.ClassDef): + CurrentType = 'class' + # 如果前一个不是类，添加空行分隔 + if PrevType is not None and PrevType != 'class': + StubLines.append('') + # 检查是否有对应的 Postdefinition 变量 + ClassPostdefs: list[tuple[str, str, ast.AnnAssign]] = PostdefVars.get(node.name, []) + ClassLines: list[str] = PythonToStubConverter._ConvertClass( + node, SourceLines=lines, PostdefVars=ClassPostdefs + ) + StubLines.extend(ClassLines) + elif isinstance(node, ast.FunctionDef): + CurrentType = 'function' + # 如果前一个是类或者是第一个函数，添加空行分隔 + if PrevType == 'class' or (PrevType is not None and PrevType != 'function'): + StubLines.append('') + FuncLines: list[str] = PythonToStubConverter._ConvertFunction(node, SourceLines=lines) + StubLines.extend(FuncLines) + StubLines.append('') # 函数后空行 + elif isinstance(node, ast.AnnAssign): + CurrentType = 'variable' + # 跳过 Postdefinition 变量（已经在类中处理） + TypeStr = PythonToStubConverter._GetTypeString(node.annotation) + if 't.Postdefinition' in TypeStr: + continue + # 检查是否带有 t.CStatic 标志，如果有则不排除 + HasStatic: bool = 't.CStatic' in TypeStr + # 检查变量是否应该被排除 + ShouldExclude: bool = False + if isinstance(node.target, ast.Name) and not HasStatic: + VarName: str = node.target.id + if PythonToStubConverter._ShouldExcludeVar(VarName): + ShouldExclude = True + if ShouldExclude: + continue + # 如果前一个不是变量，添加空行分隔 + if PrevType is not None and PrevType != 'variable': + StubLines.append('') + VarLines: list[str] = PythonToStubConverter._ConvertVariable(node, SourceLines=lines) + # 移除多余的空行 + VarLines = [line for line in VarLines if line.strip()] + StubLines.extend(VarLines) + elif isinstance(node, ast.Import): + # 处理导入语句 + CurrentType = 'import' + ImportStr: str = PythonToStubConverter._GetImportString(node, SourceLines=lines) + if ImportStr: + if PrevType is not None and PrevType != 'import': + StubLines.append('') + StubLines.append(ImportStr) + elif isinstance(node, ast.Assign): + CurrentType = 'variable' + # 处理模块级别的全局变量（无类型标注） + if PrevType is not None and PrevType != 'variable': + StubLines.append('') + AssignLines: list[str] = PythonToStubConverter._ConvertGlobalAssign(node) + StubLines.extend(AssignLines) + elif isinstance(node, ast.ImportFrom): + # 处理 from ... import ... 语句 + CurrentType = 'import' + ImportStr = PythonToStubConverter._GetImportFromString(node, SourceLines=lines) + if ImportStr: + if PrevType is not None and PrevType != 'import': + StubLines.append('') + StubLines.append(ImportStr) + elif isinstance(node, ast.Expr): + # 处理模块级别的宏调用，如 c.CIf(...), c.CEndif() + ExprStr: str = PythonToStubConverter._GetExprString(node.value) + if ExprStr: + CurrentType = 'macro' + StubLines.append(ExprStr) + StubLines.append('') # 宏后空行 + elif isinstance(node, ast.If): + # 处理模块级别的 if 宏条件，如 if c.CIf(FF_MULTI_PARTITION): + IfStr: str = PythonToStubConverter._GetModuleIfMacroString(node) + if IfStr: + CurrentType = 'macro' + StubLines.append(IfStr) + StubLines.append('') # 宏后空行 + + if CurrentType: + PrevType = CurrentType + + # 添加文件结尾宏守卫 + #StubLines.append('') + #StubLines.append('c.CEndif()') + + except SyntaxError as e: + # 如果解析失败，添加注释说明 + StubLines.append(f'# Warning: Failed to parse Python code: {e}') + StubLines.append('# Original content:') + StubLines.append('"""') + StubLines.append(PyContent[:1000]) # 只显示前1000字符 + StubLines.append('"""') + + return '\n'.join(StubLines) + + @staticmethod + def _ConvertClass(node: ast.ClassDef, SourceLines: list[str] | None = None, PostdefVars: list[tuple[str, str, ast.AnnAssign]] | None = None, indent: int = 0) -> list[str]: + """转换类定义 + + Args: + node: 类定义节点 + SourceLines: 源代码行列表 + PostdefVars: Postdefinition 变量列表 [(VarName, TypeStr, node), ...] + indent: 缩进级别（用于嵌套类） + """ + lines: list[str] = [] + PostdefVars = PostdefVars or [] + IndentStr: str = ' ' * indent + + # 处理类装饰器 + for decorator in node.decorator_list: + DecoratorStr: str = PythonToStubConverter._GetDecoratorString(decorator) + if DecoratorStr: + lines.append(f'{IndentStr}@{DecoratorStr}') + + # 检查是否是结构体、联合体或枚举 + BaseNames: list[str] = [base.attr if isinstance(base, ast.Attribute) else base.id + for base in node.bases + if isinstance(base, (ast.Name, ast.Attribute))] + + # 保留原始基类 + if node.bases: + BaseStrs: list[str] = [] + for base in node.bases: + if isinstance(base, ast.Name): + BaseStrs.append(base.id) + elif isinstance(base, ast.Attribute): + BaseStrs.append(f'{base.value.id}.{base.attr}') + elif isinstance(base, ast.Subscript): + # 处理泛型类型，如 memory_block_t[MAX_ORDER + 1] + BaseStrs.append(PythonToStubConverter._GetTypeString(base)) + ClassTypeParamStr: str = '' + if hasattr(node, 'type_params') and node.type_params: + param_names: list[str] = [tp.name for tp in node.type_params] + ClassTypeParamStr = f'[{", ".join(param_names)}]' + if BaseStrs: + lines.append(f'{IndentStr}class {node.name}{ClassTypeParamStr}({", ".join(BaseStrs)}):') + else: + lines.append(f'{IndentStr}class {node.name}{ClassTypeParamStr}:') + else: + ClassTypeParamStr = '' + if hasattr(node, 'type_params') and node.type_params: + param_names = [tp.name for tp in node.type_params] + ClassTypeParamStr = f'[{", ".join(param_names)}]' + lines.append(f'{IndentStr}class {node.name}{ClassTypeParamStr}:') + + # 处理类成员 + HasMembers: bool = False + seen_member_names: set[str] = set() + for item in node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + seen_member_names.add(item.target.id) + init_members: list[ast.AnnAssign] = [] + for item in node.body: + if isinstance(item, ast.FunctionDef) and item.name == '__init__': + for stmt in item.body: + if (isinstance(stmt, ast.AnnAssign) + and isinstance(stmt.target, ast.Attribute) + and isinstance(stmt.target.value, ast.Name) + and stmt.target.value.id == 'self'): + init_members.append(stmt) + untyped_self_members: list[ast.Assign] = [] + for m_name in [m.target.attr for m in init_members if isinstance(m.target, ast.Attribute)]: + seen_member_names.add(m_name) + for item in node.body: + if isinstance(item, ast.FunctionDef): + for stmt in item.body: + if (isinstance(stmt, ast.Assign) + and len(stmt.targets) == 1 + and isinstance(stmt.targets[0], ast.Attribute) + and isinstance(stmt.targets[0].value, ast.Name) + and stmt.targets[0].value.id == 'self'): + attr_name: str = stmt.targets[0].attr + if attr_name not in seen_member_names: + seen_member_names.add(attr_name) + untyped_self_members.append(stmt) + for item in list(node.body) + init_members + untyped_self_members: + if isinstance(item, ast.Assign): + if (len(item.targets) == 1 + and isinstance(item.targets[0], ast.Attribute) + and isinstance(item.targets[0].value, ast.Name) + and item.targets[0].value.id == 'self'): + HasMembers = True + MemberIndent: str = IndentStr + ' ' + attr_name = item.targets[0].attr + if attr_name in seen_member_names: + continue + InferredType: str = 'int' + if item.value and isinstance(item.value, ast.Constant): + if isinstance(item.value.value, float): + InferredType = 'float' + elif isinstance(item.value.value, bool): + InferredType = 'bool' + elif isinstance(item.value.value, str): + InferredType = 'str' + lines.append(f'{MemberIndent}{attr_name}: {InferredType}') + else: + HasMembers = True + MemberIndent = IndentStr + ' ' + if SourceLines and hasattr(item, 'lineno'): + StartLine: int = item.lineno - 1 + EndLine: int = item.end_lineno if hasattr(item, 'end_lineno') and item.end_lineno else StartLine + 1 + for i in range(StartLine, min(EndLine, len(SourceLines))): + lines.append(MemberIndent + SourceLines[i].lstrip()) + elif isinstance(item, ast.AnnAssign): + HasMembers = True + TypeStr: str = PythonToStubConverter._GetTypeString(item.annotation) + MemberIndent = IndentStr + ' ' + # __provides__/__requires__/__require_must__ 元数据：保留值以便跨模块加载 + if isinstance(item.target, ast.Name) and item.target.id in ('__provides__', '__requires__', '__require_must__'): + if item.value is not None: + try: + value_str: str = ast.unparse(item.value) + except Exception: + value_str = '' + lines.append(f'{MemberIndent}{item.target.id}: {TypeStr} = {value_str}') + else: + lines.append(f'{MemberIndent}{item.target.id}: {TypeStr}') + continue + # 检查是否是宏变量（类型包含 t.CDefine） + if 't.CDefine' in TypeStr: + # 检查宏变量是否应该被排除 + if isinstance(item.target, ast.Name): + MacroName: str = item.target.id + if PythonToStubConverter._ShouldExcludeMacro(MacroName): + continue # 跳过该宏 + # 宏变量保持原样（包括赋值） + if SourceLines and hasattr(item, 'lineno'): + StartLine = item.lineno - 1 + EndLine = item.end_lineno if hasattr(item, 'end_lineno') and item.end_lineno else StartLine + 1 + for i in range(StartLine, min(EndLine, len(SourceLines))): + lines.append(MemberIndent + SourceLines[i].lstrip()) + else: + lines.append(f'{MemberIndent}{item.target.id}: {TypeStr}') + else: + if isinstance(item.target, ast.Attribute): + MemberName: str = item.target.attr + else: + MemberName = item.target.id + FinalTypeStr: str = TypeStr + if IsListAnnotation(item.annotation): + slice_node: ast.AST = item.annotation.slice + has_size: bool = False + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + count_node: ast.AST = slice_node.elts[1] + if isinstance(count_node, ast.Constant) and isinstance(count_node.value, int) and count_node.value > 0: + has_size = True + else: + try: + count_expr: str = ast.unparse(count_node) + except Exception: + count_expr = '' + if count_expr and count_expr != 'None': + has_size = True + elem_type_str: str = PythonToStubConverter._GetTypeString(slice_node.elts[0]) + FinalTypeStr = f't.CArray[{elem_type_str}, {count_expr}]' + if not has_size: + elem_type_str = PythonToStubConverter._GetTypeString(slice_node if not isinstance(slice_node, ast.Tuple) else slice_node.elts[0]) + init_len: int = 0 + if item.value is not None: + if isinstance(item.value, ast.List): + init_len = len(item.value.elts) + elif isinstance(item.value, ast.Constant) and isinstance(item.value.value, str): + init_len = len(item.value.value) + 1 + if init_len > 0: + FinalTypeStr = f't.CArray[{elem_type_str}, {init_len}]' + else: + FinalTypeStr = f't.CArray[{elem_type_str}, None]' + if item.value is not None and isinstance(item.value, ast.Constant): + try: + value_repr: str = ast.unparse(item.value) + lines.append(f'{MemberIndent}{MemberName}: {FinalTypeStr} = {value_repr}') + except Exception: + lines.append(f'{MemberIndent}{MemberName}: {FinalTypeStr}') + else: + lines.append(f'{MemberIndent}{MemberName}: {FinalTypeStr}') + elif isinstance(item, ast.FunctionDef): + HasMembers = True + FuncStub: list[str] = PythonToStubConverter._ConvertFunction(item, indent=indent+1, SourceLines=SourceLines, class_name=node.name) + lines.extend(FuncStub) + elif isinstance(item, ast.Expr): + # 处理宏调用，如 c.CIf(...), c.CEndif(), c.CElif(), c.CElse() + ExprStr: str = PythonToStubConverter._GetExprString(item.value) + if ExprStr: + lines.append(f'{IndentStr} {ExprStr}') + elif isinstance(item, ast.If): + # 处理 if c.CIf(...): 形式的宏条件 + IfStr: str = PythonToStubConverter._get_if_macro_string(item, indent=indent+1) + if IfStr: + lines.append(IfStr) + elif isinstance(item, ast.ClassDef): + # 处理嵌套类 + HasMembers = True + NestedClassLines: list[str] = PythonToStubConverter._ConvertClass(item, SourceLines=SourceLines, PostdefVars=[], indent=indent+1) + # 移除嵌套类的宏守卫（因为已经在父类的宏守卫内） + FilteredLines: list[str] = [] + for line in NestedClassLines: + # 跳过宏守卫相关行 + if line.strip().startswith('c.CIfndef(') or line.strip().startswith('c.CEndif()'): + continue + if ': t.CDefine' in line and '__' in line: + continue + FilteredLines.append(line) + lines.extend(FilteredLines) + + if not HasMembers: + lines.append(f'{IndentStr} pass') + + # 添加 Postdefinition 变量 + if PostdefVars: + lines.append('') + for VarName, TypeStr, _ in PostdefVars: + lines.append(f'{VarName}: {TypeStr}') + + return lines + + @staticmethod + def _ConvertFunction(node: ast.FunctionDef, indent: int = 0, SourceLines: list[str] | None = None, class_name: str | None = None) -> list[str]: + """转换函数定义""" + lines: list[str] = [] + IndentStr: str = ' ' * indent + + # 检查是否是宏函数（返回类型包含 t.CDefine） + is_macro_func: bool = False + if node.returns: + ReturnType: str = PythonToStubConverter._GetTypeString(node.returns) + if 't.CDefine' in ReturnType: + is_macro_func = True + + # 如果是宏函数，检查是否应该被排除 + if is_macro_func: + if PythonToStubConverter._ShouldExcludeMacro(node.name): + return lines # 返回空列表，排除该宏 + + # 保持原样（包括代码块） + if SourceLines: + StartLine: int = node.lineno - 1 + EndLine: int = node.end_lineno if hasattr(node, 'end_lineno') and node.end_lineno else StartLine + 1 + for i in range(StartLine, min(EndLine, len(SourceLines))): + lines.append(SourceLines[i]) + if lines and lines[-1].strip() and not lines[-1].rstrip().endswith(':'): + pass + else: + lines.append(f'{IndentStr} pass') + return lines + + # 处理函数装饰器 + for decorator in node.decorator_list: + DecoratorStr: str = PythonToStubConverter._GetDecoratorString(decorator) + if DecoratorStr: + lines.append(f'{IndentStr}@{DecoratorStr}') + + # 构建参数列表 + params: list[str] = [] + for arg_idx, arg in enumerate(node.args.args): + ArgName: str = arg.arg + if arg.annotation: + TypeStr: str = PythonToStubConverter._GetTypeString(arg.annotation) + params.append(f'{ArgName}: {TypeStr}') + elif class_name and arg_idx == 0 and ArgName == 'self': + params.append(f'self: {class_name}') + else: + params.append(ArgName) + + # 处理 *args + if node.args.vararg: + ArgName = node.args.vararg.arg + if node.args.vararg.annotation: + TypeStr = PythonToStubConverter._GetTypeString(node.args.vararg.annotation) + params.append(f'*{ArgName}: {TypeStr}') + else: + params.append(f'*{ArgName}') + + # 处理 **kwargs + if node.args.kwarg: + ArgName = node.args.kwarg.arg + if node.args.kwarg.annotation: + TypeStr = PythonToStubConverter._GetTypeString(node.args.kwarg.annotation) + params.append(f'**{ArgName}: {TypeStr}') + else: + params.append(f'**{ArgName}') + + ParamStr: str = ', '.join(params) + + # 返回类型 - 保持原始类型（不添加t.State） + if node.returns: + ReturnType = PythonToStubConverter._GetTypeString(node.returns) + else: + ReturnType = 't.CInt' + + TypeParamStr: str = '' + if hasattr(node, 'type_params') and node.type_params: + param_names: list[str] = [tp.name for tp in node.type_params] + TypeParamStr = f'[{", ".join(param_names)}]' + + lines.append(f'{IndentStr}def {node.name}{TypeParamStr}({ParamStr}) -> {ReturnType}: pass') + return lines + + @staticmethod + def _ConvertVariable(node: ast.AnnAssign, SourceLines: list[str] | None = None) -> list[str]: + """转换变量定义""" + lines: list[str] = [] + if isinstance(node.target, ast.Name): + VarName: str = node.target.id + TypeStr: str = PythonToStubConverter._GetTypeString(node.annotation) + # 检查是否是宏变量（类型包含 t.CDefine） + if 't.CDefine' in TypeStr: + # 检查宏变量是否应该被排除 + if PythonToStubConverter._ShouldExcludeMacro(VarName): + return lines # 返回空列表，排除该宏 + # 宏变量保持原样（包括赋值） + if SourceLines and hasattr(node, 'lineno'): + StartLine: int = node.lineno - 1 + EndLine: int = node.end_lineno if hasattr(node, 'end_lineno') and node.end_lineno else StartLine + 1 + for i in range(StartLine, min(EndLine, len(SourceLines))): + lines.append(SourceLines[i]) + else: + lines.append(f'{VarName}: {TypeStr}') + return lines + # 非宏变量：不加 t.State + # 含有 typedef 的变量不加 extern，普通变量需要加 extern + has_typedef: bool = 't.CTypedef' in TypeStr or 'CTypedef' in TypeStr + if has_typedef and node.value is not None: + ValueTypeStr: str = PythonToStubConverter._GetTypeString(node.value) + lines.append(f'{VarName}: {TypeStr} = {ValueTypeStr}') + else: + if not has_typedef and 't.CExtern' not in TypeStr and 't.CStatic' not in TypeStr: + TypeStr = f't.CExtern | {TypeStr}' + lines.append(f'{VarName}: {TypeStr}') + return lines + + @staticmethod + def _ConvertGlobalAssign(node: ast.Assign) -> list[str]: + """转换模块级别的全局变量（无类型标注）""" + lines: list[str] = [] + if not node.targets or not isinstance(node.targets[0], ast.Name): + return lines + + VarName: str = node.targets[0].id + + # 跳过私有变量（以下划线开头） + if VarName.startswith('_'): + return lines + + # 从值推断类型 + inferred_type: str | None = PythonToStubConverter._InferTypeFromValue(node.value) + if inferred_type: + lines.append(f'{VarName}: {inferred_type}') + else: + lines.append(f'{VarName}: t.CInt') + + return lines + + @staticmethod + def _InferTypeFromValue(value: ast.AST) -> str | None: + """从值推断 C 类型""" + if isinstance(value, ast.Constant): + if isinstance(value.value, int): + return 't.CInt' + elif isinstance(value.value, float): + return 't.CDouble' + elif isinstance(value.value, str): + return 't.CCharPtr' + elif isinstance(value.value, bool): + return 't.CInt' + elif isinstance(value, ast.List): + return 't.CVoidPtr' + elif isinstance(value, ast.Dict): + return 't.CVoidPtr' + elif isinstance(value, ast.Name): + return 't.CInt' + elif isinstance(value, ast.BinOp): + left_type: str | None = PythonToStubConverter._InferTypeFromValue(value.left) + if left_type: + return left_type + return 't.CInt' + elif isinstance(value, ast.Call): + # 函数调用返回值，假设为指针类型 + if isinstance(value.func, ast.Name): + func_name: str = value.func.id + if func_name == 'malloc': + return 't.CVoidPtr' + elif func_name in ('ctypes.cast', 'cast'): + return 't.CVoidPtr' + return 't.CVoidPtr' + return None + + @staticmethod + def _GetTypeString(annotation: ast.AST) -> str: + """获取类型字符串""" + if isinstance(annotation, ast.Name): + return annotation.id + elif isinstance(annotation, ast.Attribute): + if isinstance(annotation.value, ast.Name): + return f'{annotation.value.id}.{annotation.attr}' + else: + # 处理嵌套属性访问，如 t.attr.packed + ParentStr: str = PythonToStubConverter._GetTypeString(annotation.value) + return f'{ParentStr}.{annotation.attr}' + elif isinstance(annotation, ast.Subscript): + base: str = PythonToStubConverter._GetTypeString(annotation.value) + if isinstance(annotation.slice, ast.Tuple): + slice_strs: list[str] = [PythonToStubConverter._GetTypeString(e) for e in annotation.slice.elts] + SliceStr: str = ', '.join(slice_strs) + else: + SliceStr = PythonToStubConverter._GetTypeString(annotation.slice) + return f'{base}[{SliceStr}]' + elif isinstance(annotation, ast.BinOp): + # 处理 BinOp，如 t.CExtern | fd_table.__set_default__(...) + left: ast.AST = annotation.left + right: ast.AST = annotation.right + + # 检查右侧是否是 __set_default__ 调用 + if isinstance(right, ast.Call) and isinstance(right.func, ast.Attribute) and right.func.attr in ('__set_default__', '__set_default'): + # 去除 __set_default__，保留左侧，右侧替换为函数名部分 + LeftStr: str = PythonToStubConverter._GetTypeString(left) + RightStr: str = PythonToStubConverter._GetTypeString(right.func.value) + if isinstance(annotation.op, ast.BitOr): + return f'{LeftStr} | {RightStr}' + + LeftStr = PythonToStubConverter._GetTypeString(left) + RightStr = PythonToStubConverter._GetTypeString(right) + if isinstance(annotation.op, ast.BitOr): + return f'{LeftStr} | {RightStr}' + elif isinstance(annotation.op, ast.Add): + return f'{LeftStr} + {RightStr}' + elif isinstance(annotation.op, ast.Sub): + return f'{LeftStr} - {RightStr}' + elif isinstance(annotation.op, ast.Mult): + return f'{LeftStr} * {RightStr}' + elif isinstance(annotation.op, ast.Div): + return f'{LeftStr} / {RightStr}' + elif isinstance(annotation.op, ast.Mod): + return f'{LeftStr} % {RightStr}' + elif isinstance(annotation, ast.UnaryOp): + if isinstance(annotation.op, ast.Not): + operand: str = PythonToStubConverter._GetTypeString(annotation.operand) + return f'not {operand}' + elif isinstance(annotation.op, ast.Invert): + operand = PythonToStubConverter._GetTypeString(annotation.operand) + return f'~{operand}' + elif isinstance(annotation.op, ast.USub): + operand = PythonToStubConverter._GetTypeString(annotation.operand) + return f'-{operand}' + elif isinstance(annotation, ast.Compare): + # 处理比较操作，如 FF_MAX_SS != FF_MIN_SS + left: str = PythonToStubConverter._GetTypeString(annotation.left) + comparators: list[str] = [PythonToStubConverter._GetTypeString(c) for c in annotation.comparators] + ops: list[str] = [] + for op in annotation.ops: + if isinstance(op, ast.Eq): + ops.append('==') + elif isinstance(op, ast.NotEq): + ops.append('!=') + elif isinstance(op, ast.Lt): + ops.append('<') + elif isinstance(op, ast.LtE): + ops.append('<=') + elif isinstance(op, ast.Gt): + ops.append('>') + elif isinstance(op, ast.GtE): + ops.append('>=') + else: + ops.append('?') + result: str = left + for i, op in enumerate(ops): + result += f' {op} {comparators[i]}' + return result + elif isinstance(annotation, ast.Constant): + return repr(annotation.value) + elif isinstance(annotation, ast.Index): + return PythonToStubConverter._GetTypeString(annotation.value) + elif isinstance(annotation, ast.List): + # 处理列表，如 [MAX_FILE_DESCRIPTORS] + elements: list[str] = [PythonToStubConverter._GetTypeString(elem) for elem in annotation.elts] + return f'[{", ".join(elements)}]' + elif isinstance(annotation, ast.Call): + # 处理函数调用形式的类型注解，如 t.Postdefinition(xxa) 或 callable(t.CVoid, app_id = t.CInt) + # 如果是 __set_default__ 调用，只返回函数名部分 + if isinstance(annotation.func, ast.Attribute) and annotation.func.attr in ('__set_default__', '__set_default'): + return PythonToStubConverter._GetTypeString(annotation.func.value) + + FuncStr: str = PythonToStubConverter._GetTypeString(annotation.func) + ArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in annotation.args]) + # 处理关键字参数 + keywords_str: str = ', '.join([f'{kw.arg} = {PythonToStubConverter._GetTypeString(kw.value)}' for kw in annotation.keywords]) + # 合并位置参数和关键字参数 + if ArgsStr and keywords_str: + return f'{FuncStr}({ArgsStr}, {keywords_str})' + elif keywords_str: + return f'{FuncStr}({keywords_str})' + else: + return f'{FuncStr}({ArgsStr})' + return 't.CVoid' + + @staticmethod + def _ExtractPostdefClass(annotation: ast.AST) -> str | None: + """从 Postdefinition 类型注解中提取类名 + + 例如：t.Postdefinition(__buddy_system) -> '__buddy_system' + t.Postdefinition(xxx) | t.CTypedef -> 'xxx' + """ + # 处理 BinOp (如: t.Postdefinition(xxx) | t.CTypedef) + if isinstance(annotation, ast.BinOp): + # 递归检查左操作数 + left_result: str | None = PythonToStubConverter._ExtractPostdefClass(annotation.left) + if left_result: + return left_result + # 递归检查右操作数 + right_result: str | None = PythonToStubConverter._ExtractPostdefClass(annotation.right) + if right_result: + return right_result + return None + + # 处理函数调用 (如: t.Postdefinition(__buddy_system)) + if isinstance(annotation, ast.Call): + FuncStr: str = PythonToStubConverter._GetTypeString(annotation.func) + if FuncStr == 't.Postdefinition' and annotation.args: + # 提取第一个参数 + first_arg: ast.AST = annotation.args[0] + if isinstance(first_arg, ast.Name): + return first_arg.id + elif isinstance(first_arg, ast.Attribute): + return f'{first_arg.value.id}.{first_arg.attr}' + + return None + + @staticmethod + def _GetImportString(node: ast.Import, SourceLines: list[str] | None = None) -> str: + """获取导入语句字符串""" + parts: list[str] = [] + IncludeAliasMap: dict[str, str] = PythonToStubConverter.IncludeAliasMap + + for alias in node.names: + ModuleName: str = alias.name + # 检查是否是 __include.xxx 格式的导入 + if ModuleName.startswith('__include.'): + # 提取别名（xxx 部分） + SubModule: str = ModuleName[len('__include.'):] + if SubModule in IncludeAliasMap: + # 替换为实际模块路径 + RealModule: str = IncludeAliasMap[SubModule] + if alias.asname: + parts.append(f'{RealModule} as {alias.asname}') + else: + parts.append(f'{RealModule} as {SubModule}') + else: + # 不在映射表中，保持原样 + if alias.asname: + parts.append(f'{ModuleName} as {alias.asname}') + else: + parts.append(ModuleName) + else: + if alias.asname: + parts.append(f'{alias.name} as {alias.asname}') + else: + parts.append(alias.name) + + result: str = f'import {", ".join(parts)}' + + # 提取并保留注释 + if SourceLines and hasattr(node, 'lineno'): + LineIdx: int = node.lineno - 1 + if 0 <= LineIdx < len(SourceLines): + line: str = SourceLines[LineIdx] + CommentMatch: re.Match | None = re.search(r'#.*$', line) + if CommentMatch: + result += ' ' + CommentMatch.group(0) + + return result + + @staticmethod + def _GetImportFromString(node: ast.ImportFrom, SourceLines: list[str] | None = None) -> str: + """获取 from ... import ... 语句字符串""" + module: str = node.module or '' + parts: list[str] = [] + IncludeAliasMap: dict[str, str] = PythonToStubConverter.IncludeAliasMap + + # 处理相对导入 (from . import xxx 或 from ..package import xxx) + if node.level > 0: + if module: + if node.level == 1: + full_module: str = f'.{module}' + else: + full_module = '.' * node.level + module + else: + full_module = '.' * node.level + for alias in node.names: + if alias.asname: + parts.append(f'{alias.name} as {alias.asname}') + else: + parts.append(alias.name) + return f'from {full_module} import {", ".join(parts)}' + + # 提取注释（在生成结果前获取） + comment: str = '' + if SourceLines and hasattr(node, 'lineno'): + LineIdx = node.lineno - 1 + if 0 <= LineIdx < len(SourceLines): + line = SourceLines[LineIdx] + CommentMatch: re.Match | None = re.search(r'#.*$', line) + if CommentMatch: + comment = ' ' + CommentMatch.group(0) + + # 检查是否是 from __include import xxx 格式 + if module == '__include': + for alias in node.names: + name: str = alias.name + if name in IncludeAliasMap: + # 替换为实际模块路径 + RealModule: str = IncludeAliasMap[name] + if alias.asname: + parts.append(f'import {RealModule} as {alias.asname}{comment}') + else: + parts.append(f'import {RealModule} as {name}{comment}') + else: + # 不在映射表中，保持原样 + if alias.asname: + parts.append(f'from {module} import {name} as {alias.asname}{comment}') + else: + parts.append(f'from {module} import {name}{comment}') + return '\n'.join(parts) + + # 普通 from ... import ... 语句 + for alias in node.names: + if alias.asname: + parts.append(f'{alias.name} as {alias.asname}') + else: + parts.append(alias.name) + return f'from {module} import {", ".join(parts)}{comment}' + + @staticmethod + def _GetDecoratorString(decorator: ast.AST) -> str: + """获取装饰器字符串""" + if isinstance(decorator, ast.Name): + return decorator.id + elif isinstance(decorator, ast.Attribute): + if isinstance(decorator.value, ast.Name): + return f'{decorator.value.id}.{decorator.attr}' + return decorator.attr + elif isinstance(decorator, ast.Call): + # 处理带参数的装饰器，如 @c.Attribute(t.attr.packed) + FuncStr: str = PythonToStubConverter._GetDecoratorString(decorator.func) + ArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in decorator.args]) + return f'{FuncStr}({ArgsStr})' + return '' + + @staticmethod + def _GetExprString(expr: ast.AST) -> str: + """获取表达式字符串（用于宏调用）""" + if isinstance(expr, ast.Call): + # 处理函数调用，如 c.CIf(...), c.CEndif(), c.CUndef(...), c.CPragma(...) + FuncStr: str = PythonToStubConverter._GetDecoratorString(expr.func) + if FuncStr and ('CIf' in FuncStr or 'CEndif' in FuncStr or 'CElif' in FuncStr or 'CElse' in FuncStr or 'CUndef' in FuncStr or 'CPragma' in FuncStr): + ArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in expr.args]) + return f'{FuncStr}({ArgsStr})' if ArgsStr else f'{FuncStr}()' + return '' + + @staticmethod + def _get_if_macro_string(node: ast.If, indent: int = 1) -> str: + """获取类内部的 if 宏字符串（如 if c.CIf(...):）""" + IndentStr: str = ' ' * indent + InnerIndent: str = ' ' * (indent + 1) + # 检查条件是否是宏调用 + if isinstance(node.test, ast.Call): + FuncStr: str = PythonToStubConverter._GetDecoratorString(node.test.func) + if FuncStr and 'CIf' in FuncStr: + ArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in node.test.args]) + result: list[str] = [f'{IndentStr}if {FuncStr}({ArgsStr}):' if ArgsStr else f'{IndentStr}if {FuncStr}():'] + # 处理 if 体内的语句 + for item in node.body: + if isinstance(item, ast.AnnAssign): + TypeStr: str = PythonToStubConverter._GetTypeString(item.annotation) + result.append(f'{InnerIndent}{item.target.id}: {TypeStr}') + elif isinstance(item, ast.Expr): + ExprStr: str = PythonToStubConverter._GetExprString(item.value) + if ExprStr: + result.append(f'{InnerIndent}{ExprStr}') + return '\n'.join(result) + return '' + + @staticmethod + def _GetModuleIfMacroString(node: ast.If) -> str: + """获取模块级别的 if 宏字符串（如 if c.CIf(FF_MULTI_PARTITION):）""" + return PythonToStubConverter._ProcessIfMacro(node, indent=0) + + @staticmethod + def _ProcessIfMacro(node: ast.If, indent: int = 0) -> str: + """处理 if 宏节点，支持 elif 和 else""" + IndentStr: str = ' ' * indent + + # 检查条件是否是宏调用 + if isinstance(node.test, ast.Call): + FuncStr: str = PythonToStubConverter._GetDecoratorString(node.test.func) + if FuncStr and 'CIf' in FuncStr: + ArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in node.test.args]) + result: list[str] = [f'{IndentStr}if {FuncStr}({ArgsStr}):' if ArgsStr else f'{IndentStr}if {FuncStr}():'] + # 处理 if 体内的语句 + for item in node.body: + result.extend(PythonToStubConverter._ProcessMacroBodyItem(item, indent + 1)) + + # 处理 elif 和 else + current: ast.If = node + while current.orelse: + if len(current.orelse) == 1 and isinstance(current.orelse[0], ast.If): + # elif 情况 + elif_node: ast.If = current.orelse[0] + if isinstance(elif_node.test, ast.Call): + ElifFuncStr: str = PythonToStubConverter._GetDecoratorString(elif_node.test.func) + if ElifFuncStr and 'CIf' in ElifFuncStr: + ElifArgsStr: str = ', '.join([PythonToStubConverter._GetTypeString(arg) for arg in elif_node.test.args]) + result.append(f'{IndentStr}elif {ElifFuncStr}({ElifArgsStr}):' if ElifArgsStr else f'{IndentStr}elif {ElifFuncStr}():') + for item in elif_node.body: + result.extend(PythonToStubConverter._ProcessMacroBodyItem(item, indent + 1)) + current = elif_node + continue + break + else: + # else 情况 + result.append(f'{IndentStr}else:') + for item in current.orelse: + result.extend(PythonToStubConverter._ProcessMacroBodyItem(item, indent + 1)) + break + + return '\n'.join(result) + return '' + + @staticmethod + def _ProcessMacroBodyItem(item: ast.AST, indent: int = 0) -> list[str]: + """处理宏体内的单个语句""" + IndentStr: str = ' ' * indent + result: list[str] = [] + + if isinstance(item, ast.ClassDef): + class_stub: list[str] = PythonToStubConverter._ConvertClass(item) + for line in class_stub: + if line.strip(): + result.append(f'{IndentStr}{line}') + else: + result.append(line) + elif isinstance(item, ast.FunctionDef): + FuncStub: list[str] = PythonToStubConverter._ConvertFunction(item, indent=indent) + result.extend(FuncStub) + elif isinstance(item, ast.AnnAssign): + TypeStr: str = PythonToStubConverter._GetTypeString(item.annotation) + result.append(f'{IndentStr}{item.target.id}: {TypeStr}') + elif isinstance(item, ast.Expr): + ExprStr: str = PythonToStubConverter._GetExprString(item.value) + if ExprStr: + result.append(f'{IndentStr}{ExprStr}') + elif isinstance(item, ast.If): + # 处理嵌套的 if 宏 + NestedIf: str = PythonToStubConverter._ProcessIfMacro(item, indent=indent) + if NestedIf: + result.append(NestedIf) + + return result \ No newline at end of file diff --git a/lib/StubGen/Generator.py b/lib/StubGen/Generator.py new file mode 100644 index 0000000..526071f --- /dev/null +++ b/lib/StubGen/Generator.py @@ -0,0 +1,313 @@ +"""StubGen 生成器模块 - 存根生成器主类和命令行入口""" +from __future__ import annotations +import sys +import os +import re +import argparse +import logging +import traceback +from pathlib import Path +from typing import List + +from lib.core.VLogger import get_logger as _vlog +from lib.core.stub_generator import CHeaderParser, PythonStubGenerator, CTypeMapper +from lib.StubGen.Config import StubGenConfig +from lib.StubGen.Converter import PythonToStubConverter + + +class StubGen: + """存根生成器主类""" + + def __init__(self, config: StubGenConfig) -> None: + self.config: StubGenConfig = config + self.logger: logging.Logger = self._SetupLogger() + self.GeneratedFiles: List[str] = [] + self.FailedFiles: List[str] = [] + + # 应用自定义类型映射 + if config.TypeMappings: + CTypeMapper.BASIC_TYPE_MAP.update(config.TypeMappings) + + def _SetupLogger(self) -> logging.Logger: + """设置日志""" + logger: logging.Logger = logging.getLogger('StubGen') + logger.setLevel(logging.DEBUG if self.config.verbose else logging.INFO) + + if not logger.handlers: + handler: logging.StreamHandler = logging.StreamHandler(sys.stdout) + formatter: logging.Formatter = logging.Formatter( + '%(asctime)s - %(name)s - %(levelname)s - %(message)s', + datefmt='%H:%M:%S' + ) + handler.setFormatter(formatter) + logger.addHandler(handler) + + return logger + + def FindInputFiles(self) -> list[tuple[str, str]]: + """查找输入文件，返回 (文件路径, 相对路径) 列表""" + files: list[tuple[str, str]] = [] + + # 添加显式指定的文件 + for FilePath in self.config.InputFiles: + if os.path.exists(FilePath): + RelPath: str = os.path.relpath(FilePath, self.config.InputDir or '.') + files.append((FilePath, RelPath)) + else: + self.logger.warning(f"Input file not found: {FilePath}") + + # 从输入目录查找 + if self.config.InputDir and os.path.exists(self.config.InputDir): + InputPath: Path = Path(self.config.InputDir) + for pattern in self.config.IncludePatterns: + for FilePath in InputPath.rglob(pattern): + # 检查是否在排除列表中 + excluded: bool = False + RelPath: str = os.path.relpath(str(FilePath), self.config.InputDir) + + for exclude_pattern in self.config.ExcludePatterns: + if FilePath.match(exclude_pattern) or RelPath == exclude_pattern: + excluded = True + break + + if not excluded: + files.append((str(FilePath), RelPath)) + + # 去重并保持顺序 + seen: set[str] = set() + UniqueFiles: list[tuple[str, str]] = [] + for FilePath, RelPath in files: + key: str = FilePath + if key not in seen: + seen.add(key) + UniqueFiles.append((FilePath, RelPath)) + + return UniqueFiles + + def _GenerateGuardName(self, FilePath: str) -> str: + """生成宏守卫名称""" + # 获取文件名（不含扩展名） + BaseName: str = os.path.splitext(os.path.basename(FilePath))[0] + + # 转换为大写，替换特殊字符为下划线 + guard: str = re.sub(r'[^a-zA-Z0-9_]', '_', BaseName).upper() + guard = re.sub(r'_+', '_', guard) # 合并多个下划线 + guard = guard.strip('_') + + return f'{guard}_DEFINE_H' + + def _GetOutputPath(self, RelPath: str) -> str: + """获取输出文件路径""" + # 更改扩展名为 .pyi (Python stub file) + BaseName: str = os.path.splitext(RelPath)[0] + OutputRelPath: str = BaseName + '.pyi' + + # 构建完整输出路径 + if self.config.OutputDir: + OutputPath: str = os.path.join(self.config.OutputDir, OutputRelPath) + else: + OutputPath = OutputRelPath + + return OutputPath + + def GenerateStub(self, InputFile: str, RelPath: str) -> bool: + """生成单个存根文件""" + try: + self.logger.info(f"Processing: {InputFile}") + + # 确定输出文件路径 + OutputFile: str = self._GetOutputPath(RelPath) + + # 确保输出目录存在 + OutputDir: str = os.path.dirname(OutputFile) + if OutputDir and not os.path.exists(OutputDir): + if not self.config.DryRun: + os.makedirs(OutputDir, exist_ok=True) + self.logger.debug(f"Created directory: {OutputDir}") + + if self.config.DryRun: + self.logger.info(f"[DRY RUN] Would generate: {OutputFile}") + return True + + # 根据文件类型选择处理方式 + ext: str = os.path.splitext(InputFile)[1].lower() + + if ext in ['.h', '.c']: + # C 文件：使用 CHeaderParser + content: str = self._GenerateFromC(InputFile, RelPath) + elif ext == '.py': + # Python 文件：使用 PythonToStubConverter + content = self._GenerateFromPy(InputFile, RelPath) + else: + self.logger.warning(f"Unsupported file type: {ext}") + return False + + # 写入文件 + with open(OutputFile, 'w', encoding='utf-8') as f: + f.write(content) + + self.GeneratedFiles.append(OutputFile) + self.logger.info(f"Generated: {OutputFile}") + return True + + except Exception as e: + self.logger.error(f"Failed to process {InputFile}: {e}") + traceback.print_exc() + self.FailedFiles.append(InputFile) + return False + + def _GenerateFromC(self, InputFile: str, RelPath: str) -> str: + """从 C 文件生成存根""" + parser: CHeaderParser = CHeaderParser() + parser.parse_file(InputFile) + + generator: PythonStubGenerator = PythonStubGenerator(parser) + ModuleName: str = os.path.splitext(os.path.basename(InputFile))[0] + content: str = generator.generate(ModuleName) + + # 添加宏守卫（如果需要） + if self.config.GenerateGuards: + guard_name: str = self._GenerateGuardName(InputFile) + guard_comment: str = f"\n# Guard: {guard_name}\n" + content = content + guard_comment + + return content + + def _GenerateFromPy(self, InputFile: str, RelPath: str) -> str: + """从 Python 文件生成存根""" + with open(InputFile, 'r', encoding='utf-8') as f: + PyContent: str = f.read() + + ModuleName: str = os.path.splitext(os.path.basename(InputFile))[0] + content: str = PythonToStubConverter.convert(PyContent, ModuleName) + + # 添加宏守卫（如果需要） + if self.config.GenerateGuards: + guard_name: str = self._GenerateGuardName(InputFile) + guard_comment: str = f"\n# Guard: {guard_name}\n" + content = content + guard_comment + + return content + + def run(self) -> bool: + """运行生成器""" + self.logger.info("=" * 60) + self.logger.info("StubGen - C/H/Py to Python Stub Generator") + self.logger.info("=" * 60) + + # 查找输入文件 + InputFiles: list[tuple[str, str]] = self.FindInputFiles() + + if not InputFiles: + self.logger.warning("No input files found!") + return False + + self.logger.info(f"Found {len(InputFiles)} input file(s)") + for FilePath, RelPath in InputFiles: + self.logger.debug(f" - {RelPath}") + + # 处理每个文件 + SuccessCount: int = 0 + for FilePath, RelPath in InputFiles: + if self.GenerateStub(FilePath, RelPath): + SuccessCount += 1 + + # 输出统计信息 + self.logger.info("=" * 60) + self.logger.info(f"Summary: {SuccessCount}/{len(InputFiles)} files generated successfully") + + if self.FailedFiles: + self.logger.warning(f"Failed files ({len(self.FailedFiles)}):") + for f in self.FailedFiles: + self.logger.warning(f" - {f}") + + return SuccessCount == len(InputFiles) + + +def main() -> None: + parser: argparse.ArgumentParser = argparse.ArgumentParser( + description='StubGen - Generate Python stub files from C/H/Py sources', + formatter_class=argparse.RawDescriptionHelpFormatter, + epilog=''' +Examples: + # 使用配置文件 + %(prog)s -c config.json + + # 处理单个文件 + %(prog)s -i input.h -o output.py + + # 批量处理目录（保持目录结构） + %(prog)s -d ./kernel -o ./kernel/include + + # 干运行（不实际生成文件） + %(prog)s -d ./kernel --dry-run + + # 不生成宏守卫 + %(prog)s -d ./kernel --no-guards + ''' + ) + + # 输入选项 + InputGroup: argparse._MutuallyExclusiveGroup = parser.add_mutually_exclusive_group(required=True) + InputGroup.add_argument('-c', '--config', help='Configuration file (JSON)') + InputGroup.add_argument('-i', '--input', help='Input file') + InputGroup.add_argument('-d', '--directory', help='Input directory') + + # 输出选项 + parser.add_argument('-o', '--output', help='Output directory') + + # 其他选项 + parser.add_argument('--include', action='append', + help='Include patterns (default: *.h, *.c, *.py)') + parser.add_argument('--exclude', action='append', + help='Exclude patterns') + parser.add_argument('--no-guards', action='store_true', + help='Do not generate guard macros') + parser.add_argument('--no-structure', action='store_true', + help='Do not preserve directory structure') + parser.add_argument('-v', '--verbose', action='store_true', + help='Verbose output') + parser.add_argument('--dry-run', action='store_true', + help='Dry run (do not create files)') + + args: argparse.Namespace = parser.parse_args() + + # 加载配置 + if args.config: + if not os.path.exists(args.config): + _vlog().error(f"配置文件未找到: {args.config}") + sys.exit(1) + config: StubGenConfig = StubGenConfig.from_file(args.config) + else: + config = StubGenConfig() + config.IncludePatterns = args.include or ['*.h', '*.c', '*.py'] + config.ExcludePatterns = args.exclude or [] + config.GenerateGuards = not args.no_guards + config.PreserveStructure = not args.no_structure + config.verbose = args.verbose + config.DryRun = args.dry_run + + if args.input: + config.InputFiles = [args.input] + if args.output: + config.OutputDir = os.path.dirname(args.output) or '.' + elif args.directory: + config.InputDir = args.directory + if args.output: + config.OutputDir = args.output + + # 创建生成器并运行 + generator: StubGen = StubGen(config) + + # 如果指定了单个输出文件，直接处理 + if args.input and args.output and not os.path.isdir(args.output): + RelPath: str = os.path.relpath(args.input, config.InputDir or '.') + success: bool = generator.GenerateStub(args.input, RelPath) + else: + success = generator.run() + + sys.exit(0 if success else 1) + + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/lib/StubGen/__init__.py b/lib/StubGen/__init__.py new file mode 100644 index 0000000..00a45bd --- /dev/null +++ b/lib/StubGen/__init__.py @@ -0,0 +1,6 @@ +"""StubGen 包 - 存根文件生成器""" +from lib.StubGen.Config import StubGenConfig +from lib.StubGen.Converter import PythonToStubConverter +from lib.StubGen.Generator import StubGen, main + +__all__ = ['StubGenConfig', 'PythonToStubConverter', 'StubGen', 'main'] diff --git a/lib/__init__.py b/lib/__init__.py new file mode 100644 index 0000000..ea4ac0f --- /dev/null +++ b/lib/__init__.py @@ -0,0 +1 @@ +"""TransPyC 编译器核心库""" diff --git a/lib/_bootstrap.py b/lib/_bootstrap.py new file mode 100644 index 0000000..8089950 --- /dev/null +++ b/lib/_bootstrap.py @@ -0,0 +1,37 @@ +"""集中化 sys.path 引导模块 + +在首次导入时一次性设置好编译器运行所需的 sys.path： +- 项目根目录：让 ``import TransPyC``、``from StubGen import``、动态 importlib 能解析 +- lib/includes：让用户注解模块中的 ``import t`` / ``import c`` 能解析 + +入口点（TransPyC.py / Projectrans.py）应在最开头 ``import lib._bootstrap``。 +lib 内部模块也可 ``import lib._bootstrap`` 作为安全网，重复导入是无副作用的 no-op。 +""" +from __future__ import annotations + +import os +import sys + +_BOOTSTRAPPED: bool = False + + +def _bootstrap() -> None: + global _BOOTSTRAPPED + if _BOOTSTRAPPED: + return + _BOOTSTRAPPED = True + + # lib/_bootstrap.py 所在目录 + _LibDir: str = os.path.dirname(os.path.abspath(__file__)) + # 项目根目录 = lib 的父目录 + _ProjectRoot: str = os.path.dirname(_LibDir) + # lib/includes 目录（编译器内部类型系统 t.py / c.py） + _IncludesDir: str = os.path.join(_LibDir, 'includes') + + if _ProjectRoot not in sys.path: + sys.path.insert(0, _ProjectRoot) + if _IncludesDir not in sys.path: + sys.path.append(_IncludesDir) + + +_bootstrap() diff --git a/lib/constants/__init__.py b/lib/constants/__init__.py new file mode 100644 index 0000000..013292a --- /dev/null +++ b/lib/constants/__init__.py @@ -0,0 +1 @@ +"""全局常量定义""" diff --git a/lib/constants/config.py b/lib/constants/config.py new file mode 100644 index 0000000..250a029 --- /dev/null +++ b/lib/constants/config.py @@ -0,0 +1,43 @@ +# 常量定义 +from __future__ import annotations + +DEFAULT_INPUT_FILE: str = 'test.py' +DEFAULT_OUTPUT_FILE: str = 'test.ll' + +SLICE_THRESHOLD: int = 50 +SLICE_LEVEL_DEFAULT: int = 3 +SLICE_LEVELS: dict[int | str, str] = { + 0: 'no_optimize', + 1: 'conservative', + 2: 'moderate', + 3: 'aggressive', + 's': 'size', +} + +DEFAULT_COMPILE_COMMAND: str = 'gcc' +DEFAULT_COMPILE_FLAGS: str = '' + +DEFAULT_METADATA: dict[str, str] = { + 'PROJECT_NAME': 'TransPyC Project', + 'AUTHOR': 'TransPyC Team', + 'VERSION': '1.0.0', + 'DESCRIPTION': 'Auto-generated LLVM IR file', + 'COPYRIGHT_YEAR': '2026', + 'COPYRIGHT_HOLDER': 'Galactic Vorpal Software Development Studio (e.g. GVSDS Inc.)' +} + +ERROR_MESSAGES: dict[str, str] = { + 'MISSING_ARGS': 'Missing required arguments -f and/or -o', + 'UNKNOWN_ARG': 'Unknown argument', + 'UNSUPPORTED_TYPE': 'Unsupported file type', + 'FAILED_PARSE': 'Failed to parse file', + 'COMPILE_FAILED': 'Compilation or execution failed' +} + +HELP_MESSAGE: str = ''' +Usage: python TransPyC.py -f InputFile -o OutputFile [-wh header_files] [-debug DebugFile] + [-cc compile_command] [-cflags CompileFlags] [-run] [-args RunArgs] [-h HelperFiles] + -h: Specify helper files (C or Python) to help identify structs, functions, variables, and ptrs +''' + +mode: str = "strict" diff --git a/lib/core/ConstEvaluator.py b/lib/core/ConstEvaluator.py new file mode 100644 index 0000000..9897e2b --- /dev/null +++ b/lib/core/ConstEvaluator.py @@ -0,0 +1,515 @@ +"""统一常量表达式求值器 + +整合了原先分散在 5 处的常量求值逻辑： + - SymbolTable._eval_const_expr + - CTypeInfo.TryEvalConstExpr + - ConstEvalMixin._try_eval_const_expr + - HandlesAssign._eval_const_expr + - HandlesIf._eval_const_expr + +ConstEvaluator 提供三个级别的求值： + 1. eval_basic(node) — 纯 AST 常量 + 算术（无外部依赖） + 2. eval_with_symtab(node) — + SymbolTable 中的 define 查找 + 3. eval_full(node, ctx) — + Gen._define_constants / _all_define_constants / 平台宏 +""" +from __future__ import annotations + +import ast +from typing import Any, Callable, Optional, TYPE_CHECKING + +import llvmlite.ir as ir + +from lib.core.Handles.HandlesBase import CTypeInfo + +if TYPE_CHECKING: + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator + + +class ConstEvaluator: + """统一常量表达式求值器""" + + # ================================================================== + # Level 2: + SymbolTable define 查找 + # ================================================================== + + @staticmethod + def eval_with_symtab(node: ast.AST, symtab: Any) -> Optional[Any]: + """求值常量表达式，支持从 SymbolTable 查找 define 值。 + + 支持: Constant, BinOp, UnaryOp, Name(define), Attribute(define) + """ + if isinstance(node, ast.Constant): + return node.value + if isinstance(node, ast.Name): + if hasattr(symtab, '__contains__') and node.id in symtab: + info: CTypeInfo = symtab[node.id] + if info.IsDefine and isinstance(info.DefineValue, int): + return info.DefineValue + return None + if isinstance(node, ast.Attribute): + # 先尝试 t.CSizeT().Size 模式（类型构造调用 + .Size 属性） + size_val = ConstEvaluator._eval_ctype_size_attr(node) + if size_val is not None: + return size_val + return ConstEvaluator._eval_attribute_define(node, symtab) + return ConstEvaluator._eval_arith(node, lambda n: ConstEvaluator.eval_with_symtab(n, symtab)) + + # ================================================================== + # Level 3: + Gen._define_constants / _all_define_constants / 平台宏 + # ================================================================== + + @staticmethod + def eval_full(node: ast.AST, ctx: 'EvalContext') -> Optional[Any]: + """完整求值，支持 define_constants、平台宏、BoolOp、Compare。 + + ctx: EvalContext 实例，封装了 Gen、SymbolTable 等上下文。 + """ + # BoolOp (HandlesIf 逻辑) + if isinstance(node, ast.BoolOp): + if isinstance(node.op, ast.And): + for val in node.values: + result: Any = ConstEvaluator.eval_full(val, ctx) + if result is None: + return None + if not result: + return 0 + return 1 + elif isinstance(node.op, ast.Or): + for val in node.values: + result: Any = ConstEvaluator.eval_full(val, ctx) + if result is None: + return None + if result: + return 1 + return 0 + return None + + # Compare (HandlesIf 逻辑) + if isinstance(node, ast.Compare): + left: Any = ConstEvaluator.eval_full(node.left, ctx) + if left is None: + return None + op: ast.cmpop + comparator: ast.AST + for op, comparator in zip(node.ops, node.comparators): + right: Any = ConstEvaluator.eval_full(comparator, ctx) + if right is None: + return None + result: bool | None = ConstEvaluator._eval_compare_op(op, left, right) + if result is None: + return None + if not result: + return 0 + left = right + return 1 + + # Constant — 直接返回原始值，bool→0/1 转换由 HandlesIf 自行处理 + if isinstance(node, ast.Constant): + return node.value + + # Name — 查 define_constants / SymbolTable / 平台宏 + if isinstance(node, ast.Name): + return ctx.lookup_name(node.id) + + # Attribute — 查 define_constants / SymbolTable / 平台宏 + if isinstance(node, ast.Attribute): + # 先尝试 t.CSizeT().Size 模式（类型构造调用 + .Size 属性） + size_val = ConstEvaluator._eval_ctype_size_attr(node) + if size_val is not None: + return size_val + return ctx.lookup_attribute(node) + + # UnaryOp + if isinstance(node, ast.UnaryOp): + operand: Any = ConstEvaluator.eval_full(node.operand, ctx) + if operand is None: + return None + if isinstance(node.op, ast.Not): + return 1 if not operand else 0 + if isinstance(node.op, ast.USub): + return -operand + if isinstance(node.op, ast.UAdd): + return +operand + if isinstance(node.op, ast.Invert): + return ~operand + return None + + # Call — 编译时函数调用（如 ctraits.isptr(x)） + if isinstance(node, ast.Call): + return ConstEvaluator._eval_compile_time_call(node, ctx) + + # BinOp + return ConstEvaluator._eval_arith(node, lambda n: ConstEvaluator.eval_full(n, ctx)) + + # ================================================================== + # 内部辅助 + # ================================================================== + + @staticmethod + def _eval_compile_time_call(node: ast.Call, ctx: 'EvalContext') -> Optional[Any]: + """处理编译时函数调用，如 ctraits.isptr(x)""" + func: ast.expr = node.func + func_name: str | None = None + module_name: str | None = None + + if isinstance(func, ast.Attribute): + if isinstance(func.value, ast.Name): + module_name = func.value.id + func_name = func.attr + elif isinstance(func, ast.Name): + func_name = func.id + + # ctraits.isptr(x) — 带模块前缀 + # isptr(x) — 通过 from ctraits import isptr 导入 + is_isptr: bool = False + if module_name == 'ctraits' and func_name == 'isptr': + is_isptr = True + elif func_name == 'isptr' and not module_name: + # 检查是否通过 from ctraits import isptr 导入 + t_c_imported = getattr(ctx.translator, '_t_c_imported_names', {}) if ctx.translator else {} + if func_name in t_c_imported: + src_module, _ = t_c_imported[func_name] + if src_module == 'ctraits': + is_isptr = True + # 也检查符号表中是否有 ctraits.isptr + if not is_isptr and ctx.symtab: + sym = ctx.symtab.lookup('ctraits.isptr') if hasattr(ctx.symtab, 'lookup') else None + if sym: + is_isptr = True + + if is_isptr and node.args and ctx.translator: + try: + arg_node: ast.expr = node.args[0] + # 局部变量：优先通过 LLVM 类型判断（GetCTypeInfo 无法解析局部变量名） + if ctx.Gen and isinstance(arg_node, ast.Name): + var_name: str = arg_node.id + variables: dict = getattr(ctx.Gen, 'variables', {}) + if var_name in variables: + var_val: Any = variables[var_name] + if hasattr(var_val, 'type') and isinstance(var_val.type, ir.PointerType): + pointee: ir.Type = var_val.type.pointee + if isinstance(pointee, ir.PointerType): + return 1 + return 0 + # 类型名 / typedef：通过 TypeMergeHandler 获取 CTypeInfo + type_info = ctx.translator.TypeMergeHandler.GetCTypeInfo(arg_node) + if type_info and (type_info.IsPtr or type_info.PtrCount > 0): + return 1 + if type_info and not (type_info.IsPtr or type_info.PtrCount > 0): + return 0 + return None + except Exception: + return None + return None + + @staticmethod + def _eval_arith(node: ast.AST, recurse: Callable[[ast.AST], Any]) -> Optional[Any]: + """通用算术求值：BinOp + UnaryOp""" + if isinstance(node, ast.BinOp): + left: Any = recurse(node.left) + right: Any = recurse(node.right) + if left is None or right is None: + return None + return ConstEvaluator._apply_binop(node.op, left, right) + if isinstance(node, ast.UnaryOp): + operand: Any = recurse(node.operand) + if operand is None: + return None + if isinstance(node.op, ast.USub): + return -operand + if isinstance(node.op, ast.UAdd): + return +operand + if isinstance(node.op, ast.Invert): + return ~operand + return None + return None + + @staticmethod + def _apply_binop(op: ast.operator, left: Any, right: Any) -> Optional[Any]: + """应用二元运算符""" + try: + if isinstance(op, ast.Add): + return left + right + elif isinstance(op, ast.Sub): + return left - right + elif isinstance(op, ast.Mult): + return left * right + elif isinstance(op, ast.Div): + return left // right if isinstance(left, int) and isinstance(right, int) else left / right + elif isinstance(op, ast.FloorDiv): + if isinstance(right, (int, float)) and right == 0: + return None + return left // right + elif isinstance(op, ast.Mod): + if isinstance(right, (int, float)) and right == 0: + return None + return left % right + elif isinstance(op, ast.Pow): + return left ** right + elif isinstance(op, ast.LShift): + return left << right + elif isinstance(op, ast.RShift): + return left >> right + elif isinstance(op, ast.BitOr): + return left | right + elif isinstance(op, ast.BitXor): + return left ^ right + elif isinstance(op, ast.BitAnd): + return left & right + except Exception: + return None + return None + + @staticmethod + def _eval_compare_op(op: ast.cmpop, left: Any, right: Any) -> Optional[bool]: + """应用比较运算符""" + if isinstance(op, ast.Eq): + return left == right + elif isinstance(op, ast.NotEq): + return left != right + elif isinstance(op, ast.Lt): + return left < right + elif isinstance(op, ast.LtE): + return left <= right + elif isinstance(op, ast.Gt): + return left > right + elif isinstance(op, ast.GtE): + return left >= right + return None + + @staticmethod + def _eval_ctype_size_attr(node: ast.Attribute) -> Optional[int]: + """求值 t.CSizeT().Size 等类型构造调用 + .Size 属性模式。 + + 匹配 AST 形态: Attribute(attr='Size', value=Call(func=Attribute(value=Name(id='t'), attr='CSizeT'))) + 或直接 Call(func=Name(id='CSizeT'))（from t import CSizeT 后的使用方式）。 + 返回类型大小（位数），如 CSizeT().Size 返回 64（64 位平台）。 + """ + if node.attr != 'Size': + return None + if not isinstance(node.value, ast.Call): + return None + call_node: ast.Call = node.value + if call_node.args or call_node.keywords: + return None + func: ast.expr = call_node.func + type_name: str | None = None + if isinstance(func, ast.Attribute) and isinstance(func.value, ast.Name) and func.value.id == 't': + type_name = func.attr + elif isinstance(func, ast.Name): + type_name = func.id + if not type_name: + return None + # 延迟导入避免循环依赖 + try: + from lib.includes.t import CTypeRegistry + except Exception: + return None + cls = CTypeRegistry.GetClassByName(type_name) + if cls is None: + return None + try: + instance = cls() + size_val = getattr(instance, 'Size', None) + if isinstance(size_val, int): + return size_val + except Exception: + pass + return None + + @staticmethod + def _eval_attribute_define(node: ast.Attribute, symtab: Any) -> Optional[Any]: + """从 SymbolTable 查找 Attribute 形式的 define 值""" + parts: list[str] = [] + current: ast.AST = node + while isinstance(current, ast.Attribute): + parts.append(current.attr) + current = current.value + if isinstance(current, ast.Name): + parts.append(current.id) + parts.reverse() + attr_name: str = parts[-1] + possible_keys: list[str] = [attr_name, '.'.join(parts)] + for key in possible_keys: + if hasattr(symtab, '__contains__') and key in symtab: + info: CTypeInfo = symtab[key] + if info.IsDefine and isinstance(info.DefineValue, int): + return info.DefineValue + # 模糊匹配 + if len(parts) >= 2: + mod_key: str + mod_info: CTypeInfo + for mod_key, mod_info in (symtab.items() if hasattr(symtab, 'items') else []): + if mod_info.IsDefine and isinstance(mod_info.DefineValue, int): + if mod_key.endswith('.' + attr_name) or mod_key == attr_name: + return mod_info.DefineValue + return None + + @staticmethod + def _intify(val: Any) -> Any: + """float → int 转换（仅当 float 值恰好是整数时）""" + if isinstance(val, float) and val == int(val): + return int(val) + return val + + +class EvalContext: + """常量求值上下文，封装 Gen / SymbolTable / 平台宏查找逻辑。 + + 由调用方构造并传入 ConstEvaluator.eval_full()。 + """ + + def __init__(self, Gen: LlvmCodeGenerator | None = None, symtab: Any = None, platform_macros: dict | None = None, translator: Any = None): + self.Gen: LlvmCodeGenerator | None = Gen + self.symtab: Any = symtab + self._platform_macros: dict | None = platform_macros + self.translator: Any = translator + + def lookup_name(self, name: str) -> Optional[Any]: + """查找 Name 节点的值：define_constants → SymbolTable → 平台宏""" + # 1. Gen._define_constants + if self.Gen and name in self.Gen._define_constants: + val: Any = self.Gen._define_constants[name] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + + # 2. Gen._DefineConstants (大写版本) + if self.Gen and hasattr(self.Gen, '_DefineConstants') and name in self.Gen._DefineConstants: + val: Any = self.Gen._DefineConstants[name] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + + # 3. SymbolTable + if self.symtab and name in self.symtab: + info: CTypeInfo = self.symtab[name] + val: Any = info.value + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + if info.IsDefine and isinstance(info.DefineValue, (int, float)): + return ConstEvaluator._intify(info.DefineValue) + + # 4. 平台宏 + macros: dict = self._get_platform_macros() + if name in macros: + return macros[name] + + return None + + def lookup_attribute(self, node: ast.Attribute) -> Optional[Any]: + """查找 Attribute 节点的值""" + module_name: str | None = node.value.id if isinstance(node.value, ast.Name) else None + attr_name: str = node.attr + + if module_name and attr_name: + combined_key: str = f"{module_name}.{attr_name}" + + # 1. Gen._define_constants + if self.Gen and combined_key in self.Gen._define_constants: + val: Any = self.Gen._define_constants[combined_key] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + + # 2. SymbolTable (精确匹配 + 后缀匹配) + if self.symtab: + key: str + info: CTypeInfo + for key, info in self.symtab.items(): + if key.endswith(f".{combined_key}") or key == combined_key: + if info.IsDefine and isinstance(info.DefineValue, (int, float)): + return ConstEvaluator._intify(info.DefineValue) + + # 3. _all_define_constants + all_dc: dict = self._get_all_define_constants() + if combined_key in all_dc: + val: Any = all_dc[combined_key] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + if attr_name in all_dc: + val: Any = all_dc[attr_name] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + dc_key: str + dc_val: Any + for dc_key, dc_val in all_dc.items(): + if dc_key == attr_name or dc_key.endswith(f".{attr_name}"): + if isinstance(dc_val, (int, float)): + return ConstEvaluator._intify(dc_val) + + # 4. SymbolTable 短名查找 + if self.symtab and attr_name in self.symtab: + info: CTypeInfo = self.symtab[attr_name] + if info.IsDefine and isinstance(info.DefineValue, (int, float)): + return ConstEvaluator._intify(info.DefineValue) + + # 5. Gen._define_constants 短名 + if self.Gen and attr_name in self.Gen._define_constants: + val: Any = self.Gen._define_constants[attr_name] + if isinstance(val, (int, float)): + return ConstEvaluator._intify(val) + + # 6. 平台宏 + macros: dict = self._get_platform_macros() + full_key: str | None = self._get_attr_full_name(node) + if full_key: + if full_key in macros: + return macros[full_key] + short_name: str = full_key.split('.')[-1] if '.' in full_key else full_key + if short_name in macros: + return macros[short_name] + + return None + + def _get_all_define_constants(self) -> dict: + """获取 _all_define_constants""" + if self.Gen and hasattr(self.Gen, '_all_define_constants'): + return self.Gen._all_define_constants + return {} + + def _get_platform_macros(self) -> dict: + """获取平台宏""" + if self._platform_macros is not None: + return self._platform_macros + if not self.Gen: + return {} + macros: dict[str, int] = {} + pi: dict = self.Gen._platform_info + ptr_size: int = self.Gen.ptr_size + if pi.get('is_windows'): + macros['_WIN32'] = 1 + if not pi.get('is_32bit'): + macros['_WIN64'] = 1 + macros['WIN64'] = 1 + macros['WIN32'] = 1 + if pi.get('is_linux'): + macros['__linux__'] = 1 + if pi.get('is_macos'): + macros['__APPLE__'] = 1 + macros['__MACH__'] = 1 + if pi.get('is_x86_64'): + macros['__x86_64__'] = 1 + macros['__x86_64'] = 1 + macros['_M_X64'] = 1 + elif pi.get('is_arm64'): + macros['__aarch64__'] = 1 + macros['_M_ARM64'] = 1 + if not pi.get('is_32bit') and pi.get('is_linux'): + macros['__LP64__'] = 1 + elif pi.get('is_32bit'): + macros['_ILP32'] = 1 + macros['SIZEOF_VOID_P'] = ptr_size + macros['__SIZEOF_POINTER__'] = ptr_size + macros['NDEBUG'] = 0 + return macros + + @staticmethod + def _get_attr_full_name(node: ast.Attribute) -> Optional[str]: + """从 Attribute 节点提取完整名称 (e.g. 'module.name')""" + parts: list[str] = [] + current: ast.AST = node + while isinstance(current, ast.Attribute): + parts.append(current.attr) + current = current.value + if isinstance(current, ast.Name): + parts.append(current.id) + parts.reverse() + return '.'.join(parts) if parts else None diff --git a/lib/core/DecoratorPass.py b/lib/core/DecoratorPass.py new file mode 100644 index 0000000..475f3f3 --- /dev/null +++ b/lib/core/DecoratorPass.py @@ -0,0 +1,547 @@ +""" +DecoratorPass - Viper 语言自定义装饰器的 IR 层包装转换 (v4) + +v4 增强版特性： +1. 栈帧局部上下文 (ctx)：替代全局变量，解决并发/递归状态错乱 +2. 流程劫持：handler 返回 i32 控制原函数调用次数 + - 返回 0：跳过原函数调用（@cache, @mock） + - 返回 1：正常调用一次（默认） + - 返回 N > 1：循环调用 N 次（@repeat） +3. alwaysinline 属性：消除 wrapper 调用开销 +4. 参数打包与修改：pre-phase 后从 args_struct 读回参数，装饰器可修改入参 +5. 返回值修改：ret_ptr 可写，装饰器可在后置阶段修改返回值 +6. 递归装饰保护：最外层 wrapper 检查全局标志位，递归调用时跳过所有装饰逻辑 + +装饰器调用约定 (v4)： + i32 decor_name(i8* ctx, i8* func_name, i32 phase, + i8* args_ptr, i8* ret_ptr, i8* decor_args_ptr) + + 参数说明： + ctx : i8* - 栈帧局部上下文缓冲区（32字节），每次调用独立分配 + 前置阶段可写入状态，后置阶段可读取，生命周期绑定本次调用 + 多线程/递归完全隔离，替代全局变量 + func_name : i8* - 原始函数名字符串 + phase : i32 - 执行阶段（0=前置, 1=后置） + args_ptr : i8* - 指向函数参数打包结构体（可读写，无参数时为 null） + 前置阶段可写入修改参数值，wrapper 会从结构体读回修改后的参数 + ret_ptr : i8* - 指向返回值（前置阶段为 null，void 函数始终为 null） + 后置阶段可读写，装饰器可修改返回值 + decor_args_ptr : i8* - 指向装饰器参数打包结构体（无参数装饰器为 null） + + 返回值（仅前置阶段有效，后置阶段忽略）： + 0 : 跳过原函数调用（@cache, @mock, @skip） + 1 : 调用原函数一次（默认行为，@log, @timing, @trace） + N>1: 循环调用原函数 N 次（@repeat, @benchmark） + + 编译器生成的 wrapper（以 add(i32, i32) -> i32, @log 为例）： + define i32 @__decor_wrap_add(i32 %a, i32 %b) alwaysinline { + entry: + ; 递归保护：检查标志位 + %rec_val = Load i8, i8* @__decor_rec___decor_wrap_add + %is_rec = icmp ne i8 %rec_val, 0 + br i1 %is_rec, label %recursive_call, label %decorated_entry + + recursive_call: + ; 递归调用：跳过所有装饰逻辑，直接调用原函数 + %rec_result = call i32 @add(i32 %a, i32 %b) + ret i32 %rec_result + + decorated_entry: + ; 设置递归保护标志 + store i8 1, i8* @__decor_rec___decor_wrap_add + + ; 分配栈帧局部上下文 + %ctx = alloca [32 x i8] + %ctx_ptr = bitcast [32 x i8]* %ctx to i8* + + ; 打包函数参数到结构体（装饰器可通过 args_ptr 修改） + %args = alloca {i32, i32} + store i32 %a, i32* getelementptr({i32, i32}* %args, i32 0, i32 0) + store i32 %b, i32* getelementptr({i32, i32}* %args, i32 0, i32 1) + %args_ptr = bitcast {i32, i32}* %args to i8* + + ; 分配返回值存储（零初始化） + %ret = alloca i32 + store i32 0, i32* %ret + + ; 前置阶段：handler 返回调用次数 + %n = call i32 @log(i8* %ctx_ptr, i8* "add", i32 0, + i8* %args_ptr, i8* null, i8* null) + br label %loop.header + + loop.header: + %i = phi i32 [0, %decorated_entry], [%i.next, %loop.body] + %cond = icmp slt i32 %i, %n + br i1 %cond, label %loop.body, label %post + + loop.body: + ; 从结构体读回参数（装饰器可能已修改） + %a.Loaded = Load i32, i32* getelementptr({i32, i32}* %args, i32 0, i32 0) + %b.Loaded = Load i32, i32* getelementptr({i32, i32}* %args, i32 0, i32 1) + %result = call i32 @add(i32 %a.Loaded, i32 %b.Loaded) + store i32 %result, i32* %ret + %i.next = add i32 %i, 1 + br label %loop.header + + post: + %ret_ptr = bitcast i32* %ret to i8* + call i32 @log(i8* %ctx_ptr, i8* "add", i32 1, + i8* %args_ptr, i8* %ret_ptr, i8* null) + + ; 清除递归保护标志 + store i8 0, i8* @__decor_rec___decor_wrap_add + + %final = Load i32, i32* %ret + ret i32 %final + } + +链式装饰器（从下到上嵌套）： + @log + @timing + def f(x) -> int: + 等价于 log(timing(f)) + 执行顺序：log_pre → timing_pre → f → timing_post → log_post + 生成：__decor_wrap_f (log) → __decor_wrap_f_timing (timing) → f + 递归保护仅在最外层 wrapper 生效，递归调用直接跳到原始 f +""" +from __future__ import annotations +import llvmlite.ir as ir +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator + + +# 栈帧局部上下文缓冲区大小（字节） +_CTX_SIZE: int = 32 + +# 装饰器处理函数的 LLVM 签名：i32 (i8*, i8*, i32, i8*, i8*, i8*) +_DECOR_HANDLER_PARAM_TYPES: ir.FunctionType | None = None # 延迟初始化 + + +def _get_decor_handler_func_type() -> ir.FunctionType: + """获取装饰器处理函数的 LLVM FunctionType: i32 (i8*, i8*, i32, i8*, i8*, i8*)""" + global _DECOR_HANDLER_PARAM_TYPES + if _DECOR_HANDLER_PARAM_TYPES is None: + i8_ptr: ir.PointerType = ir.IntType(8).as_pointer() + _DECOR_HANDLER_PARAM_TYPES = ir.FunctionType( + ir.IntType(32), + [i8_ptr, i8_ptr, ir.IntType(32), i8_ptr, i8_ptr, i8_ptr] + ) + return _DECOR_HANDLER_PARAM_TYPES + + +def _find_or_declare_handler(module: ir.Module, handler_name: str) -> ir.Function: + """在 module 中查找或声明装饰器处理函数""" + f: ir.Function + for f in module.functions: + if f.name == handler_name: + return f + func_type: ir.FunctionType = _get_decor_handler_func_type() + handler: ir.Function = ir.Function(module, func_type, name=handler_name) + return handler + + +def _make_string_constant(module: ir.Module, text: str, name_prefix: str = "decor.str") -> ir.Constant: + """在 module 中创建一个全局字符串常量，返回 i8* 指针""" + counter: int = getattr(_make_string_constant, '_counter', 0) + 1 + _make_string_constant._counter = counter + const_name: str = f"{name_prefix}.{counter}" + + encoded: bytes = text.encode('utf-8') + b'\x00' + const_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(encoded)) + const_val: ir.Constant = ir.Constant(const_type, bytearray(encoded)) + + global_var: ir.GlobalVariable = ir.GlobalVariable(module, const_type, name=const_name) + global_var.global_constant = True + global_var.linkage = 'private' + global_var.initializer = const_val + + return global_var.bitcast(ir.IntType(8).as_pointer()) + + +def _make_decor_args_constant(module: ir.Module, decorator_info: dict) -> ir.Constant: + """ + 为带参数的装饰器生成全局常量结构体，返回 i8* 指针。 + 无参数装饰器返回 null。 + + 支持的参数类型： + - int → i32 / i64 + - float → f64 + - bool → i1 + - str → i8*（全局字符串常量） + """ + deco_args: list = decorator_info.get('args', []) + deco_kwargs: dict = decorator_info.get('kwargs', {}) + + if not deco_args and not deco_kwargs: + return ir.Constant(ir.IntType(8).as_pointer(), None) + + field_types: list = [] + field_values: list = [] + + arg: int | str | float | bool + for arg in deco_args: + llvm_type: ir.Type | None + llvm_val: ir.Constant | None + llvm_type, llvm_val = _python_value_to_llvm(module, arg) + if llvm_type is None: + return ir.Constant(ir.IntType(8).as_pointer(), None) + field_types.append(llvm_type) + field_values.append(llvm_val) + + kw: str + for kw in sorted(deco_kwargs.keys()): + llvm_type: ir.Type | None + llvm_val: ir.Constant | None + llvm_type, llvm_val = _python_value_to_llvm(module, deco_kwargs[kw]) + if llvm_type is None: + return ir.Constant(ir.IntType(8).as_pointer(), None) + field_types.append(llvm_type) + field_values.append(llvm_val) + + struct_type: ir.LiteralStructType = ir.LiteralStructType(field_types) + struct_const: ir.Constant = ir.Constant(struct_type, field_values) + + counter: int = getattr(_make_decor_args_constant, '_counter', 0) + 1 + _make_decor_args_constant._counter = counter + global_name: str = f"__decor_args.{counter}" + + global_var: ir.GlobalVariable = ir.GlobalVariable(module, struct_type, name=global_name) + global_var.global_constant = True + global_var.linkage = 'private' + global_var.initializer = struct_const + + return global_var.bitcast(ir.IntType(8).as_pointer()) + + +def _python_value_to_llvm(module: ir.Module, value: int | str | float | bool) -> tuple[ir.Type | None, ir.Constant | None]: + """将 Python 值转换为 (LLVM类型, LLVM常量) 元组。""" + if isinstance(value, bool): + return ir.IntType(1), ir.Constant(ir.IntType(1), int(value)) + elif isinstance(value, int): + if -2**31 <= value < 2**31: + return ir.IntType(32), ir.Constant(ir.IntType(32), value) + else: + return ir.IntType(64), ir.Constant(ir.IntType(64), value) + elif isinstance(value, float): + return ir.DoubleType(), ir.Constant(ir.DoubleType(), value) + elif isinstance(value, str): + str_const = _make_string_constant(module, value, name_prefix="decor.arg") + return ir.IntType(8).as_pointer(), str_const + else: + return None, None + + +def _ZeroConstant(llvm_type: ir.Type) -> ir.Constant | None: + """为给定 LLVM 类型创建零值常量""" + if isinstance(llvm_type, ir.IntType): + return ir.Constant(llvm_type, 0) + elif isinstance(llvm_type, (ir.FloatType, ir.DoubleType)): + return ir.Constant(llvm_type, 0.0) + elif isinstance(llvm_type, ir.PointerType): + return ir.Constant(llvm_type, None) + elif isinstance(llvm_type, ir.ArrayType): + elem_zero: ir.Constant | None = _ZeroConstant(llvm_type.element) + if elem_zero is None: + return None + return ir.Constant(llvm_type, [elem_zero] * llvm_type.count) + else: + return None + + +def _generate_single_wrapper(module: ir.Module, original_func: ir.Function, wrapper_name: str, func_name_str: str, + decorator_info: dict, is_export: bool = False, + is_outermost: bool = False, true_original_func: ir.Function | None = None) -> ir.Function: + """ + 为单个装饰器生成一层 wrapper 函数。 + + v4 Wrapper 结构（最外层含递归保护）： + entry → 递归保护检查（仅最外层） + recursive_call → 递归时直接调用原函数，跳过所有装饰（仅最外层） + decorated_entry → 设置递归标志，分配 ctx/args/ret，调用 handler 前置 + loop.header → phi 计数器，比较 i < n_calls + loop.body → 从 args_struct 读回参数（支持修改），调用原函数 + post → 调用 handler 后置，清除递归标志，返回结果 + """ + decor_name: str = decorator_info['name'] + handler: ir.Function = _find_or_declare_handler(module, decor_name) + func_name_const: ir.Constant = _make_string_constant(module, func_name_str, name_prefix="decor.fn") + + func_type: ir.FunctionType = original_func.ftype + return_type: ir.Type = func_type.return_type + param_types: list[ir.Type] = [p.type for p in original_func.args] + is_void: bool = isinstance(return_type, ir.VoidType) + + wrapper_type: ir.FunctionType = ir.FunctionType(return_type, param_types) + wrapper: ir.Function = ir.Function(module, wrapper_type, name=wrapper_name) + + # 添加 alwaysinline 属性，优化器完全内联消除调用开销 + wrapper.attributes.add('alwaysinline') + + i32: ir.IntType = ir.IntType(32) + i8: ir.IntType = ir.IntType(8) + i8_ptr: ir.PointerType = ir.IntType(8).as_pointer() + phase_pre: ir.Constant = ir.Constant(i32, 0) + phase_post: ir.Constant = ir.Constant(i32, 1) + null_ptr: ir.Constant = ir.Constant(i8_ptr, None) + zero_i32: ir.Constant = ir.Constant(i32, 0) + one_i32: ir.Constant = ir.Constant(i32, 1) + one_i8: ir.Constant = ir.Constant(i8, 1) + zero_i8: ir.Constant = ir.Constant(i8, 0) + + # 递归保护：仅最外层 wrapper 生成 + rec_flag: ir.GlobalVariable | None = None + if is_outermost and true_original_func is not None: + rec_flag_name: str = f"__decor_rec_{wrapper_name}" + # 检查是否已存在 + rec_flag = None + g: ir.GlobalValue + for g in module.global_values: + if g.name == rec_flag_name: + rec_flag = g + break + if rec_flag is None: + rec_flag = ir.GlobalVariable(module, i8, name=rec_flag_name) + rec_flag.global_constant = False + rec_flag.linkage = 'internal' + rec_flag.initializer = zero_i8 + + # 创建基本块 + entry_block: ir.Block = wrapper.append_basic_block("entry") + + recursive_call_block: ir.Block | None + decorated_entry_block: ir.Block | None + if rec_flag is not None: + recursive_call_block = wrapper.append_basic_block("recursive_call") + decorated_entry_block = wrapper.append_basic_block("decorated_entry") + else: + recursive_call_block = None + decorated_entry_block = None + + loop_header_block: ir.Block = wrapper.append_basic_block("loop.header") + loop_body_block: ir.Block = wrapper.append_basic_block("loop.body") + post_block: ir.Block = wrapper.append_basic_block("post") + + # ==== Entry block ==== + builder: ir.IRBuilder = ir.IRBuilder(entry_block) + + loop_predecessor: ir.Block + if rec_flag is not None: + # 递归保护：检查标志位 + rec_val: ir.LoadInstr = builder.load(rec_flag, name="rec_val") + is_recursive: ir.ICMPInstr = builder.icmp_signed('!=', rec_val, zero_i8) + builder.cbranch(is_recursive, recursive_call_block, decorated_entry_block) + + # ==== Recursive call block ==== + builder = ir.IRBuilder(recursive_call_block) + # 递归调用：直接调用真正的原函数，跳过所有装饰逻辑 + rec_call_args: list[ir.Argument] = [arg for arg in wrapper.args] + rec_ret_val: ir.CallInstr = builder.call(true_original_func, rec_call_args) + if is_void: + builder.ret_void() + else: + builder.ret(rec_ret_val) + + # ==== Decorated entry block ==== + builder = ir.IRBuilder(decorated_entry_block) + # 设置递归保护标志 + builder.store(one_i8, rec_flag) + + # 记录 loop.header 的前置块为 decorated_entry_block + loop_predecessor = decorated_entry_block + else: + loop_predecessor = entry_block + + # 1. 分配栈帧局部上下文（每次调用独立，线程安全，递归安全） + ctx_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), _CTX_SIZE) + ctx_alloca: ir.AllocaInstr = builder.alloca(ctx_type, name="ctx") + ctx_ptr: ir.BitCastInstr = builder.bitcast(ctx_alloca, i8_ptr) + + # 2. 打包函数参数到结构体（装饰器可通过 args_ptr 读写修改） + args_alloca: ir.AllocaInstr | None = None + args_struct_type: ir.LiteralStructType | None = None + if param_types: + args_struct_type = ir.LiteralStructType(param_types) + args_alloca = builder.alloca(args_struct_type, name="args") + zero: ir.Constant = ir.Constant(i32, 0) + i: int + arg: ir.Argument + for i, arg in enumerate(wrapper.args): + field_idx: ir.Constant = ir.Constant(i32, i) + gep: ir.GEPInstr = builder.gep(args_alloca, [zero, field_idx], inbounds=True) + builder.store(arg, gep) + args_ptr: ir.BitCastInstr | ir.Constant = builder.bitcast(args_alloca, i8_ptr) + else: + args_ptr = null_ptr + + # 3. 分配返回值存储（零初始化，确保跳过原函数时返回安全默认值） + ret_alloca: ir.AllocaInstr | None + if not is_void: + ret_alloca = builder.alloca(return_type, name="ret") + zero_val: ir.Constant | None = _ZeroConstant(return_type) + if zero_val is not None: + builder.store(zero_val, ret_alloca) + else: + ret_alloca = None + + # 4. 获取装饰器参数指针 + decor_args_ptr: ir.Constant = _make_decor_args_constant(module, decorator_info) + + # 5. 前置阶段：handler 返回调用次数 + n_calls: ir.CallInstr = builder.call(handler, [ctx_ptr, func_name_const, phase_pre, + args_ptr, null_ptr, decor_args_ptr]) + + # 6. 跳转到循环头 + builder.branch(loop_header_block) + + # ==== Loop header block ==== + builder = ir.IRBuilder(loop_header_block) + i_phi: ir.PhiInstr = builder.phi(i32, "i") + i_phi.add_incoming(zero_i32, loop_predecessor) + cond: ir.ICMPInstr = builder.icmp_signed('<', i_phi, n_calls) + builder.cbranch(cond, loop_body_block, post_block) + + # ==== Loop body block ==== + builder = ir.IRBuilder(loop_body_block) + + # v4: 从 args_struct 读回参数（装饰器可能在 pre-phase 中修改了参数） + call_args: list[ir.LoadInstr | ir.Argument] = [] + if args_alloca is not None and args_struct_type is not None: + zero = ir.Constant(i32, 0) + for i in range(len(param_types)): + field_idx: ir.Constant = ir.Constant(i32, i) + gep: ir.GEPInstr = builder.gep(args_alloca, [zero, field_idx], inbounds=True) + Loaded_arg: ir.LoadInstr = builder.load(gep, name=f"arg.{i}") + call_args.append(Loaded_arg) + + ret_val: ir.CallInstr = builder.call(original_func, call_args) + if ret_alloca is not None: + builder.store(ret_val, ret_alloca) + i_next: ir.AddInstr = builder.add(i_phi, one_i32) + i_phi.add_incoming(i_next, loop_body_block) + builder.branch(loop_header_block) + + # ==== Post block ==== + builder = ir.IRBuilder(post_block) + + # 后置阶段 + ret_ptr: ir.BitCastInstr | ir.Constant + if ret_alloca is not None: + ret_ptr = builder.bitcast(ret_alloca, i8_ptr) + else: + ret_ptr = null_ptr + builder.call(handler, [ctx_ptr, func_name_const, phase_post, + args_ptr, ret_ptr, decor_args_ptr]) + + # 递归保护：清除标志（必须在 post-phase 之后、return 之前） + if rec_flag is not None: + builder.store(zero_i8, rec_flag) + + # 返回 + if is_void: + builder.ret_void() + else: + final_ret: ir.LoadInstr = builder.load(ret_alloca, name="final_ret") + builder.ret(final_ret) + + return wrapper + + +def _redirect_calls(module: ir.Module, old_func: ir.Function, new_func: ir.Function, skip_func_names: set[str] | None = None) -> None: + """ + 替换 module 中所有对 old_func 的调用指令为对 new_func 的调用。 + 跳过 skip_func_names 中的函数（wrapper 函数内部不应被重定向）。 + """ + if skip_func_names is None: + skip_func_names = set() + + func: ir.Function + for func in module.functions: + if func.name in skip_func_names: + continue + if func.name == new_func.name: + continue + block: ir.Block + for block in func.blocks: + instr: ir.Instruction + for instr in list(block.instructions): + if not isinstance(instr, ir.CallInstr): + continue + callee: ir.Function = instr.callee + if isinstance(callee, ir.Function) and callee is old_func: + instr.callee = new_func + + +def run(Gen: LlvmCodeGenerator) -> None: + """ + 执行 DecoratorPass。 + + 处理流程: + 1. 遍历 _decorated_funcs 中所有带装饰标记的函数 + 2. 对每个函数，按装饰器列表从底到顶生成嵌套 wrapper + 3. 将原函数改为 internal 链接 + 4. 替换所有调用点 + """ + if not Gen._decorated_funcs: + return + + module: ir.Module = Gen.module + decorated: dict = Gen._decorated_funcs.copy() + + mangled_name: str + info: dict + for mangled_name, info in decorated.items(): + decorators: list = info['decorators'] + func_name: str = info['func_name'] + is_export: bool = info['is_export'] + + # 查找原始函数 + original_func: ir.Function | None = None + f: ir.Function + for f in module.functions: + if f.name == mangled_name: + original_func = f + break + if original_func is None: + continue + + # 原函数改为 internal 链接（仅 wrapper 调用） + original_func.linkage = 'internal' + + # 按装饰器列表从底到顶生成嵌套 wrapper + # Python 装饰器顺序：@a @b def f() => a(b(f)) + # 执行顺序：先应用最靠近函数的 @b，再应用 @a + + current_func: ir.Function = original_func + all_wrapper_names: set[str] = set() + + i: int + for i in range(len(decorators) - 1, -1, -1): + deco: dict = decorators[i] + is_outermost: bool = (i == 0) + deco_name: str = deco['name'] + + wrapper_name: str + if is_outermost: + wrapper_name = f"__decor_wrap_{mangled_name}" + else: + wrapper_name = f"__decor_wrap_{mangled_name}_{deco_name}" + + all_wrapper_names.add(wrapper_name) + + wrapper: ir.Function = _generate_single_wrapper( + module, current_func, wrapper_name, + func_name, deco, is_export=is_export, + is_outermost=is_outermost, + true_original_func=original_func if is_outermost else None + ) + current_func = wrapper + + # 更新 functions 映射，使原函数名指向最外层 wrapper + Gen.functions[mangled_name] = current_func + if func_name in Gen.functions: + Gen.functions[func_name] = current_func + + # 替换所有非 wrapper 函数中对原函数的调用 + _redirect_calls(module, original_func, current_func, all_wrapper_names) diff --git a/lib/core/DiagnosticCollector.py b/lib/core/DiagnosticCollector.py new file mode 100644 index 0000000..33b7357 --- /dev/null +++ b/lib/core/DiagnosticCollector.py @@ -0,0 +1,48 @@ +"""诊断系统 — 收集编译过程中的警告和错误 + +替代原先静默回退的模式： + - return 'i32' → 记录 Warning + 回退到 'i32' + - return [] → 记录 Error + 回退到 [] + +使用方式: + diag = DiagnosticCollector() + diag.warn(file, lineno, "无法解析类型，回退到 i32") + diag.error(file, lineno, f"加载模块失败: {path}") + diag.report() # 打印所有诊断信息 +""" +from __future__ import annotations + +import os +from dataclasses import dataclass, field +from enum import Enum +from typing import List + + +class DiagnosticLevel(Enum): + WARNING = "WARNING" + ERROR = "ERROR" + + +@dataclass +class Diagnostic: + level: DiagnosticLevel + file: str + lineno: int + message: str + + def __str__(self) -> str: + short_file: str = os.path.basename(self.file) if self.file else "" + return f"[{self.level.value}] {short_file}:{self.lineno} — {self.message}" + + +class DiagnosticCollector: + """诊断信息收集器""" + + def __init__(self) -> None: + self._diagnostics: list[Diagnostic] = [] + + def warn(self, file: str, lineno: int, message: str) -> None: + self._diagnostics.append(Diagnostic(DiagnosticLevel.WARNING, file, lineno, message)) + + def error(self, file: str, lineno: int, message: str) -> None: + self._diagnostics.append(Diagnostic(DiagnosticLevel.ERROR, file, lineno, message)) diff --git a/lib/core/Exportable.py b/lib/core/Exportable.py new file mode 100644 index 0000000..ca6f9d7 --- /dev/null +++ b/lib/core/Exportable.py @@ -0,0 +1,161 @@ +#!/usr/bin/env python3 +""" +导出表模块 - 用于收集和存储文件的公开符号信息 +""" +from __future__ import annotations +from dataclasses import dataclass, field +from typing import Dict, List, Optional, Any + + +@dataclass +class EnumMember: + """枚举成员信息""" + name: str + value: int + lineno: int + + +@dataclass +class EnumInfo: + """枚举类型信息""" + name: str + members: List[EnumMember] = field(default_factory=list) + lineno: int = 0 + is_public: bool = True + + +@dataclass +class FunctionParam: + """函数参数信息""" + name: str + type_name: str + is_pointer: bool = False + + +@dataclass +class FunctionInfo: + """函数信息""" + name: str + return_type: str + params: List[FunctionParam] = field(default_factory=list) + lineno: int = 0 + is_public: bool = True + + +@dataclass +class StructMember: + """结构体成员信息""" + name: str + type_name: str + is_pointer: bool = False + array_size: Optional[int] = None + + +@dataclass +class StructInfo: + """结构体类型信息""" + name: str + members: List[StructMember] = field(default_factory=list) + lineno: int = 0 + is_public: bool = True + + +@dataclass +class TypedefInfo: + """类型别名信息""" + name: str + original_type: str + lineno: int = 0 + is_public: bool = True + + +@dataclass +class Exportable: + """导出表 - 存储文件的所有公开符号""" + filename: str + enums: List[EnumInfo] = field(default_factory=list) + functions: List[FunctionInfo] = field(default_factory=list) + structs: List[StructInfo] = field(default_factory=list) + typedefs: List[TypedefInfo] = field(default_factory=list) + + def add_enum(self, name: str, lineno: int, is_public: bool = True) -> EnumInfo: + """添加枚举类型""" + enum_info: EnumInfo = EnumInfo(name=name, lineno=lineno, is_public=is_public) + self.enums.append(enum_info) + return enum_info + + def add_function(self, name: str, return_type: str, lineno: int, is_public: bool = True) -> FunctionInfo: + """添加函数""" + func_info: FunctionInfo = FunctionInfo(name=name, return_type=return_type, lineno=lineno, is_public=is_public) + self.functions.append(func_info) + return func_info + + def add_struct(self, name: str, lineno: int, is_public: bool = True) -> StructInfo: + """添加结构体""" + struct_info: StructInfo = StructInfo(name=name, lineno=lineno, is_public=is_public) + self.structs.append(struct_info) + return struct_info + + def add_typedef(self, name: str, original_type: str, lineno: int, is_public: bool = True) -> TypedefInfo: + """添加类型别名""" + typedef_info: TypedefInfo = TypedefInfo(name=name, original_type=original_type, lineno=lineno, is_public=is_public) + self.typedefs.append(typedef_info) + return typedef_info + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + 'filename': self.filename, + 'enums': [ + { + 'name': enum.name, + 'lineno': enum.lineno, + 'is_public': enum.is_public, + 'members': [ + {'name': m.name, 'value': m.value, 'lineno': m.lineno} + for m in enum.members + ] + } + for enum in self.enums + ], + 'functions': [ + { + 'name': func.name, + 'return_type': func.return_type, + 'lineno': func.lineno, + 'is_public': func.is_public, + 'params': [ + {'name': p.name, 'type_name': p.type_name, 'is_pointer': p.is_pointer} + for p in func.params + ] + } + for func in self.functions + ], + 'structs': [ + { + 'name': struct.name, + 'lineno': struct.lineno, + 'is_public': struct.is_public, + 'members': [ + { + 'name': m.name, + 'type_name': m.type_name, + 'is_pointer': m.is_pointer, + 'array_size': m.array_size + } + for m in struct.members + ] + } + for struct in self.structs + ], + 'typedefs': [ + { + 'name': typedef.name, + 'original_type': typedef.original_type, + 'lineno': typedef.lineno, + 'is_public': typedef.is_public + } + for typedef in self.typedefs + ] + } + diff --git a/lib/core/Handles/HandlesAnnAssign.py b/lib/core/Handles/HandlesAnnAssign.py new file mode 100644 index 0000000..388b093 --- /dev/null +++ b/lib/core/Handles/HandlesAnnAssign.py @@ -0,0 +1,704 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +import ast +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.core.SymbolUtils import IsListAnnotation, ParseListAnnotation, FindStructNameInAnnotation, AnnotationContainsName + + +class AnnAssignHandle(BaseHandle): + def _StoreOrStrCopy(self, VarName: str, VarPtr: ir.Value, Value: ir.Value) -> None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + DestPtr: ir.Value = Gen._load(VarPtr, name=VarName) + if isinstance(DestPtr.type, ir.PointerType) and isinstance(DestPtr.type.pointee, ir.IntType) and DestPtr.type.pointee.width == 8: + if isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, ir.IntType) and Value.type.pointee.width == 8: + if 'llvm.memcpy' not in Gen.functions: + memcpy_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64), ir.IntType(32), ir.IntType(1)]) + Gen.functions['llvm.memcpy'] = ir.Function(Gen.module, memcpy_type, name='llvm.memcpy') + if isinstance(DestPtr.type.pointee, ir.IntType) and not isinstance(DestPtr.type.pointee, ir.IntType(8)): + DestPtr = Gen.builder.bitcast(DestPtr, ir.PointerType(ir.IntType(8)), name=f"strcpy_dst_cast_{VarName}") + if isinstance(Value.type.pointee, ir.IntType) and not isinstance(Value.type.pointee, ir.IntType(8)): + Value = Gen.builder.bitcast(Value, ir.PointerType(ir.IntType(8)), name=f"strcpy_src_cast_{VarName}") + size: ir.Constant = ir.Constant(ir.IntType(64), 8) + align: ir.Constant = ir.Constant(ir.IntType(32), 1) + isvolatile: ir.Constant = ir.Constant(ir.IntType(1), 0) + Gen.builder.call(Gen.functions['llvm.memcpy'], [DestPtr, Value, size, align, isvolatile], name=f"strcpy_call_{VarName}") + return + try: + CastedVal: ir.Value = Gen.builder.bitcast(Value, VarPtr.type.pointee, name=f"cast_{VarName}") + Gen._store(CastedVal, VarPtr) + except Exception: # 回退：bitcast 失败时 alloca 新变量 + NewVar: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=VarName) + Gen._store(Value, NewVar) + Gen.variables[VarName] = NewVar + + def _HandleAttributeStoreLlvm(self, Target: ast.Attribute, Value: ir.Value) -> None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + AttrName: str = Target.attr + if isinstance(Target.value, ast.Name) and Target.value.id == 'self': + ClassName: str | None = self._CurrentCpythonObjectClass + if not ClassName: + self_var: ir.Value | None = Gen.variables.get('self') + if self_var: + var_type: ir.Type = self_var.type + if isinstance(var_type, ir.PointerType) and isinstance(var_type.pointee, ir.PointerType): + var_type = var_type.pointee + if isinstance(var_type, ir.PointerType): + pointee: ir.Type = var_type.pointee + for cn, st in Gen.structs.items(): + if st == pointee: + ClassName = cn + break + if ClassName and ClassName in Gen.structs: + SelfVar: ir.Value | None = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: ir.Value = Gen._load(SelfVar, name="self") + offset: int = Gen._get_member_offset(AttrName, ClassName) + MemberPtr: ir.Value = Gen.builder.gep(SelfPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + elif isinstance(Target.value, ast.Name): + VarName: str = Target.value.id + + # FakeDuck: 检查是否有影子结构体 a__meta__ (str 变量的 per-instance 字段存储) + # 直接通过 a.__field__ = value 访问影子结构体字段 (语法糖) + _fd_meta_name: str = f"{VarName}__meta__" + if _fd_meta_name in Gen.variables and '_str' in Gen.structs: + _fd_offset = Gen._get_member_offset(AttrName, '_str') + if _fd_offset is not None: + _fd_meta_ptr: ir.Value = Gen.variables[_fd_meta_name] + _fd_field_ptr: ir.Value = Gen.builder.gep(_fd_meta_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _fd_offset)], name=f"{VarName}_{AttrName}_ptr") + self._StoreWithCoerce(Value, _fd_field_ptr) + return + + ClassName: str | None = Gen.var_struct_class.get(VarName) + if not ClassName: + for CN in Gen.structs: + if VarName == CN.lower() or VarName.startswith(CN): + ClassName = CN + break + if not ClassName and VarName in Gen.variables: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType): + pointee: ir.Type = VarPtr.type.pointee + if isinstance(pointee, ir.PointerType): + pointee = pointee.pointee + for CN, ST in Gen.structs.items(): + if pointee == ST: + ClassName = CN + break + IsUnion: bool = False + if ClassName: + ClassTypeInfo: Any = self.Trans.SymbolTable.lookup(ClassName) + if ClassTypeInfo: + if ClassTypeInfo.IsUnion: + IsUnion = True + if ClassName and ClassName in Gen.structs: + ObjVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + if ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.IntType) and ObjVal.type.pointee.width == 8: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{ClassName}") + if isinstance(ObjVal.type, ir.PointerType) and ObjVal.type.pointee == Gen.structs[ClassName]: + if IsUnion: + NestedStructName: str = f"{ClassName}_{AttrName}" + if NestedStructName in Gen.structs: + NestedStructType: ir.Type = Gen.structs[NestedStructName] + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(NestedStructType), name=f"cast_{NestedStructName}") + offset: int = Gen._get_member_offset(AttrName, NestedStructName) + if offset is not None and offset < len(NestedStructType.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + return + offset: int = Gen._get_member_offset(AttrName, ClassName) + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + elif isinstance(Target.value, ast.Attribute): + ObjVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + if ObjVal and isinstance(ObjVal.type, ir.PointerType): + pointee: ir.Type = ObjVal.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + offset: int = Gen._get_member_offset(AttrName, CN) + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + break + + def _StoreWithCoerce(self, Value: ir.Value, Ptr: ir.Value) -> None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(Ptr.type, ir.PointerType): + TargetType: ir.Type = Ptr.type.pointee + if Value.type != TargetType: + if isinstance(TargetType, ir.IntType) and isinstance(Value.type, ir.IntType): + if TargetType.width < Value.type.width: + Value = Gen.builder.trunc(Value, TargetType, name="trunc") + else: + Value = Gen.builder.zext(Value, TargetType, name="zext") + elif isinstance(TargetType, ir.PointerType) and isinstance(Value.type, ir.PointerType): + Value = Gen.builder.bitcast(Value, TargetType, name="ptr_bitcast") + elif isinstance(TargetType, ir.PointerType) and isinstance(Value.type, ir.IntType): + Value = Gen.builder.inttoptr(Value, TargetType, name="int_to_ptr") + elif isinstance(TargetType, ir.IntType) and isinstance(Value.type, ir.PointerType): + Value = Gen.builder.ptrtoint(Value, TargetType, name="ptr_to_int") + Gen._store(Value, Ptr) + + def _HandleGenericListLiteralAssign(self, VarName: str, Node: ast.AnnAssign, Gen: LlvmCodeGenerator, list_sub: ast.Subscript) -> None: + """处理 list[T] = [...] → 构造 list[T] 对象 + append 序列。 + + 将 list 字面量翻译为： + 1. 特化 list[T] 类 + 2. alloca + __before_init__ + __new__(pool) + __before_init__ + __init__(pool) + 3. 对每个元素调用 append(elem) + 4. 将堆指针存入变量 + list_sub 是从注解中提取的 list[...] Subscript 节点（支持 list[T] 和 list[T] | t.CPtr） + """ + # 1. 解析元素类型 + slice_node: ast.expr = list_sub.slice + result: tuple[str, str] | None = self.Trans.ExprCallHandle._resolve_generic_slice_type(slice_node) + if result is None: + return + type_arg: str = result[0] + type_name: str = result[1] + + # 2. 特化 list[T] 类 + spec_name: str | None = self.Trans.ClassHandler._specialize_generic_class( + 'list', [type_arg], Gen, type_names=[type_name]) + if spec_name is None: + return + + # 3. 获取结构体类型 + StructType: ir.Type | None = Gen.structs.get(spec_name) + if StructType is None: + return + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + + # 4. 查找 pool 变量（pool / _mbuddy / mbuddy / mb） + pool_val: ir.Value | None = None + for pool_name in ('pool', '_mbuddy', 'mbuddy', 'mb'): + if pool_name in Gen.variables: + pool_var: ir.Value = Gen.variables[pool_name] + if isinstance(pool_var.type, ir.PointerType): + pointee: ir.Type = pool_var.type.pointee + if isinstance(pointee, ir.PointerType): + # alloca of pointer → load + pool_val = Gen._load(pool_var, name=f"pool_load_{VarName}") + else: + pool_val = pool_var + else: + pool_val = pool_var + break + if pool_val is None: + return + + # 5. alloca 临时结构体 + alloca_ptr: ir.AllocaInstr = Gen._allocaEntry(StructType, name=f"{VarName}_alloca") + + # 6. 调用 __before_init__(alloca_ptr) + before_init_name: str = f'{spec_name}.__before_init__' + before_init_func: ir.Function | None = None + if Gen._has_function(before_init_name): + before_init_func = Gen._get_function(before_init_name) + Gen.builder.call(before_init_func, [alloca_ptr], name=f"before_init_{VarName}") + + # 7. 调用 __new__(alloca_ptr, pool_val) → heap_ptr + new_method_name: str = f'{spec_name}.__new__' + heap_ptr: ir.Value = alloca_ptr + has_new: bool = Gen._find_function(new_method_name) is not None + if has_new: + new_func: ir.Function | None = Gen._find_function(new_method_name) + if new_func: + new_args: list[ir.Value] = [alloca_ptr, pool_val] + self.Trans.ExprCallHandle._fill_default_args(new_args, new_func, new_method_name) + new_args = Gen._apply_auto_addr(new_method_name, new_args) + adjusted_new: list[ir.Value] = Gen._adjust_args(new_args, new_func) + heap_ptr = Gen.builder.call(new_func, adjusted_new, name=f"new_{VarName}") + # 转换返回指针类型 + target_ptr_type: ir.PointerType = ir.PointerType(StructType) + if isinstance(heap_ptr.type, ir.PointerType) and heap_ptr.type != target_ptr_type: + heap_ptr = Gen.builder.bitcast(heap_ptr, target_ptr_type, name=f"new_{VarName}_cast") + elif isinstance(heap_ptr.type, ir.IntType) and target_ptr_type: + if heap_ptr.type.width < 64: + i64_val: ir.Value = Gen.builder.zext(heap_ptr, ir.IntType(64), name=f"new_{VarName}_i64") + heap_ptr = Gen.builder.inttoptr(i64_val, target_ptr_type, name=f"new_{VarName}_ptr") + else: + heap_ptr = Gen.builder.inttoptr(heap_ptr, target_ptr_type, name=f"new_{VarName}_ptr") + # 对堆指针重新调用 __before_init__ + if before_init_func: + Gen.builder.call(before_init_func, [heap_ptr], name=f"before_init_{VarName}_heap") + + # 8. 调用 __init__(heap_ptr, pool_val, elem_size=0) + init_name: str = f'{spec_name}.__init__' + if Gen._has_function(init_name): + init_func: ir.Function | None = Gen._get_function(init_name) + if init_func: + elem_size_val: ir.Constant = ir.Constant(ir.IntType(64), 0) + init_args: list[ir.Value] = [heap_ptr, pool_val, elem_size_val] + self.Trans.ExprCallHandle._fill_default_args(init_args, init_func, init_name) + init_args = Gen._apply_auto_addr(init_name, init_args) + adjusted_init: list[ir.Value] = Gen._adjust_args(init_args, init_func) + Gen.builder.call(init_func, adjusted_init, name=f"init_{VarName}") + + # 9. 对每个元素调用 append(heap_ptr, elem_val) + append_name: str = f'{spec_name}.append' + append_func: ir.Function | None = Gen._find_function(append_name) + if append_func and Node.value and isinstance(Node.value, ast.List): + elts: list[ast.expr] = Node.value.elts + for i, elem_node in enumerate(elts): + elem_val: ir.Value | None = self.HandleExprLlvm(elem_node) + if elem_val: + append_args: list[ir.Value] = [heap_ptr, elem_val] + self.Trans.ExprCallHandle._fill_default_args(append_args, append_func, append_name) + append_args = Gen._apply_auto_addr(append_name, append_args) + adjusted_append: list[ir.Value] = Gen._adjust_args(append_args, append_func) + Gen.builder.call(append_func, adjusted_append, name=f"append_{VarName}_{i}") + + # 10. 创建变量并存储堆指针 + # 判断变量类型：list[T] | t.CPtr → i8*；list[T] → Ptr(StructType) + is_cptr_union: bool = isinstance(Node.annotation, ast.BinOp) and isinstance(Node.annotation.op, ast.BitOr) + if is_cptr_union: + # list[T] | t.CPtr → 变量类型为 i8* + i8_ptr_type: ir.PointerType = ir.PointerType(ir.IntType(8)) + var: ir.AllocaInstr = Gen._allocaEntry(i8_ptr_type, name=VarName) + # bitcast heap_ptr (Ptr(StructType)) → i8* + if isinstance(heap_ptr.type, ir.PointerType) and heap_ptr.type != i8_ptr_type: + stored_val: ir.Value = Gen.builder.bitcast(heap_ptr, i8_ptr_type, name=f"store_cast_{VarName}") + else: + stored_val = heap_ptr + Gen._store(stored_val, var) + else: + # list[T] → 变量类型为 Ptr(StructType) + var = Gen._allocaEntry(ir.PointerType(StructType), name=VarName) + Gen._store(heap_ptr, var) + Gen.variables[VarName] = var + Gen.var_struct_class[VarName] = spec_name + Gen.global_struct_class[VarName] = spec_name + Gen._record_var_signedness(VarName, False) + + def _HandleAnnAssignLlvm(self, Node: ast.AnnAssign) -> None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + self.Trans._current_assign_node = Node + try: + self._HandleAnnAssignLlvmInner(Node) + finally: + self.Trans._current_assign_node = None + + def _HandleAnnAssignLlvmInner(self, Node: ast.AnnAssign) -> None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(Node.target, ast.Name): + VarName: str = Node.target.id + TypeInfo: CTypeInfo | None = self.Trans.TypeMergeHandler.GetCTypeInfo(Node.annotation) + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + IsPtr: bool = TypeInfo.IsPtr + + if TypeInfo.IsDefine: + if Node.value and isinstance(Node.value, ast.Constant): + if not hasattr(Gen, '_define_constants'): + Gen._define_constants = {} + Gen._define_constants[VarName] = Node.value.value + info: CTypeInfo = CTypeInfo() + info.IsDefine = True + info.DefineValue = Node.value.value + self.Trans.SymbolTable.insert(VarName, info) + Gen._record_var_signedness(VarName, TypeInfo.IsUInt) + elif Node.value: + # 非 Constant 表达式：尝试编译期求值（如 t.CSizeT().Size // 8） + from lib.core.ConstEvaluator import ConstEvaluator, EvalContext + ctx = EvalContext(Gen=Gen, symtab=self.Trans.SymbolTable, translator=self.Trans) + eval_val = ConstEvaluator.eval_full(Node.value, ctx) + if eval_val is not None and isinstance(eval_val, (int, float)): + if not hasattr(Gen, '_define_constants'): + Gen._define_constants = {} + Gen._define_constants[VarName] = eval_val + info: CTypeInfo = CTypeInfo() + info.IsDefine = True + info.DefineValue = eval_val + self.Trans.SymbolTable.insert(VarName, info) + Gen._record_var_signedness(VarName, TypeInfo.IsUInt) + return + if TypeInfo.IsStr or (isinstance(Node.annotation, ast.Name) and Node.annotation.id == 'str'): + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CChar() + TypeInfo.PtrCount = 1 + IsPtr = True + # FakeDuck: 为 str 变量创建影子结构体 a__meta__ (栈上, per-instance) + # a 保持为 char* 用于 printf; a__meta__ 存储 __data__ 和 __mbuddy__ 字段 + # 仅对 str 类型创建，bytes 和 t.CChar|t.CPtr 不创建 + if '_str' in Gen.structs and isinstance(Node.annotation, ast.Name) and Node.annotation.id == 'str': + _meta_name: str = f"{VarName}__meta__" + _meta_var: ir.AllocaInstr = Gen._allocaEntry(Gen.structs['_str'], name=_meta_name) + Gen.variables[_meta_name] = _meta_var + # 标记指针到 str/bytes 的变量 (BinOp 注解如 str | t.CPtr) + # elem_ptr[0] 应加载/存储 8 字节 (指针), 而非 1 字节 (字符) + # 仅当注解包含 str 或 bytes 时设置（t.CChar | t.CPtr 不应设置，它是纯字符指针） + if isinstance(Node.annotation, ast.BinOp) and isinstance(Node.annotation.op, ast.BitOr): + if AnnotationContainsName(Node.annotation, 'str') or AnnotationContainsName(Node.annotation, 'bytes'): + Gen.var_ptr_element[VarName] = True + # 同时记录到全局字典，使全局变量在跨函数访问时仍能用 8 字节指针步长 + Gen.global_var_ptr_element[VarName] = True + parse_result = ParseListAnnotation(Node.annotation) + if parse_result: + elem_type_node: ast.expr = parse_result.elem_type_node + ElemType, ElemTypeInfo = self.ResolveListElementType( + elem_type_node, Gen, + handle_str=True, void_fallback_width=32, + ) + if parse_result.is_pointer: + # 单参数 list[type] → 指针 + # 但如果有列表字面量初始化值 [a, b, c]，则按数组 list[type, N] 处理 + if not (Node.value and isinstance(Node.value, ast.List)): + VarType: ir.Type = ir.PointerType(ElemType) + var: ir.AllocaInstr = Gen._allocaEntry(VarType, name=VarName) + Gen.variables[VarName] = var + Gen._record_var_signedness(VarName, False) + if self.Trans.VarScopes: + self.Trans.VarScopes[-1][VarName] = ElemTypeInfo + return + # 有列表字面量值，继续走数组路径（count_node=None 时从列表长度推断） + count_node: ast.expr | None = parse_result.count_node + ArrayCount: int = self.ParseArrayCount(count_node, Gen, Node.value, mode='local') + VarType = ir.ArrayType(ElemType, ArrayCount) + var = Gen._allocaEntry(VarType, name=VarName) + Gen.variables[VarName] = var + Gen._record_var_signedness(VarName, False) + if Node.value and isinstance(Node.value, ast.List): + InitConstants: list[ir.Constant] | None = self.Trans._BuildArrayInitConstants(Node.value, ElemType, elem_type_node, ArrayCount, Gen, byte_order=ElemTypeInfo.ByteOrder if ElemTypeInfo else '') + if InitConstants: + try: + InitVal: ir.Constant = ir.Constant(VarType, InitConstants) + Gen.builder.store(InitVal, var) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + elif Node.value and isinstance(Node.value, ast.Constant) and isinstance(Node.value.value, str): + str_val: str = Node.value.value + '\x00' + str_bytes: bytearray = bytearray(str_val, 'utf-8') + if isinstance(VarType, ir.ArrayType) and len(str_bytes) < VarType.count: + str_bytes.extend(b'\x00' * (VarType.count - len(str_bytes))) + try: + InitVal: ir.Constant = ir.Constant(VarType, str_bytes[:VarType.count] if isinstance(VarType, ir.ArrayType) else str_bytes) + Gen.builder.store(InitVal, var) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + if self.Trans.VarScopes: + self.Trans.VarScopes[-1][VarName] = ElemTypeInfo + # 记录数组元素字节序 + if ElemTypeInfo and ElemTypeInfo.ByteOrder == 'big' and isinstance(ElemType, ir.IntType) and ElemType.width in (16, 32, 64): + Gen.local_var_byteorders[VarName] = 'big' + return + # list[T] 泛型类注解 + list 字面量值 → 构造 list[T] 对象 + append 序列 + # （list[str] = ["a", "b"] 不应走 C 数组路径，而应构造 list[str] 对象） + # 支持两种注解形式：list[T] 和 list[T] | t.CPtr + list_sub: ast.Subscript | None = None + if (isinstance(Node.annotation, ast.Subscript) + and isinstance(Node.annotation.value, ast.Name) + and Node.annotation.value.id == 'list'): + list_sub = Node.annotation + elif (isinstance(Node.annotation, ast.BinOp) + and isinstance(Node.annotation.op, ast.BitOr)): + for side in (Node.annotation.left, Node.annotation.right): + if (isinstance(side, ast.Subscript) + and isinstance(side.value, ast.Name) + and side.value.id == 'list'): + list_sub = side + break + if (list_sub is not None + and not isinstance(list_sub.slice, ast.Tuple) + and Node.value is not None + and isinstance(Node.value, ast.List) + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and 'list' in self.Trans.ClassHandler._generic_class_templates): + self._HandleGenericListLiteralAssign(VarName, Node, Gen, list_sub) + return + if TypeInfo.IsVoid and isinstance(Node.annotation, ast.BinOp): + found: str | None = FindStructNameInAnnotation(Node.annotation, Gen.structs) + if found: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CStruct(name=found) + TypeInfo.PtrCount = 1 + IsPtr = True + elif isinstance(Node.annotation.left, ast.Attribute): + AttrTypeInfo: CTypeInfo | None = self.Trans.TypeMergeHandler.GetCTypeInfo(Node.annotation.left) + if AttrTypeInfo and AttrTypeInfo.BaseType: + TypeInfo = AttrTypeInfo + TypeInfo.PtrCount = max(TypeInfo.PtrCount, 1) + IsPtr = True + # 对于局部变量声明 s: TestStruct（不带 | t.CPtr），应为栈上结构体而非指针 + # GetCTypeInfo 默认设置 PtrCount=1，但纯结构体名注解应为值类型 + if (isinstance(Node.annotation, ast.Name) + and Node.annotation.id in Gen.structs + and TypeInfo.IsStruct + and Node.annotation.id not in Gen.class_vtable): + TypeInfo.PtrCount = 0 + IsPtr = False + VarType: ir.Type = Gen._ctype_to_llvm(TypeInfo) + if not IsPtr and TypeInfo.IsStruct and TypeInfo.Name: + stripped: str = TypeInfo.Name + if stripped in Gen.class_vtable and stripped in Gen.structs: + VarType = ir.PointerType(Gen.structs[stripped]) + IsPtr = True + if isinstance(VarType, ir.IdentifiedStructType): + struct_name: str = getattr(VarType, 'name', '') + if struct_name and Gen.class_vtable: + for cls_name in Gen.class_vtable: + if struct_name.endswith(f'.{cls_name}') or struct_name == cls_name: + VarType = ir.PointerType(VarType) + IsPtr = True + break + if isinstance(VarType, ir.VoidType): + VarType = ir.IntType(32) + IsUnsigned: bool = TypeInfo.IsUInt + # 修复 3: BinOp 注解（如 ast.Arguments | t.CPtr）的 var_struct_class 设置 + # TypeInfo.Name 由 _HandleSubscript 设置（特化名如 list[sha1.ASTptr]）或 + # GetCTypeInfo 设置（普通结构体名），经 _MergeCTypeInfoInto 传播后保留。 + # 此处确保 BinOp 注解的变量也能被 _get_var_class 识别，避免方法调用分派 + # fallthrough 到 char pointer fallback 错误匹配其他类（如 AST.__len__）。 + # 使用 not in 条件，不覆盖后续 L337-339/L396-398 对 ast.Name 注解的精确设置。 + if TypeInfo.IsStruct and TypeInfo.Name and TypeInfo.Name in Gen.structs: + if VarName not in Gen.var_struct_class: + Gen.var_struct_class[VarName] = TypeInfo.Name + Gen.global_struct_class[VarName] = TypeInfo.Name + if VarName in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[VarName] + if isinstance(OldVal.type, ir.PointerType): + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.variables[VarName] = var + del Gen._reg_values[VarName] + if Node.value: + InitValue: ir.Value | None = self.HandleExprLlvm(Node.value, VarType=VarType) + if InitValue: + self._StoreOrStrCopy(VarName, var, InitValue) + Gen._record_var_signedness(VarName, IsUnsigned) + return + del Gen._reg_values[VarName] + InitValue: ir.Value | None = None + if Node.value: + InitValue = self.HandleExprLlvm(Node.value, VarType=VarType) + if VarName in Gen._reg_values: + del Gen._reg_values[VarName] + if VarName in Gen.variables and Gen.variables[VarName] is not None: + if InitValue and isinstance(InitValue.type, ir.PointerType): + pointee: ir.Type = InitValue.type.pointee + if isinstance(pointee, ir.IdentifiedStructType): + struct_name: str = getattr(pointee, 'name', '') + is_vtable_class: bool = False + for cls_name in Gen.class_vtable: + if struct_name.endswith(f'.{cls_name}') or struct_name == cls_name: + is_vtable_class = True + break + if is_vtable_class: + var: ir.AllocaInstr = Gen._allocaEntry(InitValue.type, name=VarName) + Gen._store(InitValue, var) + Gen.variables[VarName] = var + if isinstance(Node.annotation, ast.Name) and Node.annotation.id in Gen.structs: + Gen.var_struct_class[VarName] = Node.annotation.id + Gen.global_struct_class[VarName] = Node.annotation.id + Gen._record_var_signedness(VarName, IsUnsigned) + return + if InitValue: + try: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType): + TargetType: ir.Type = VarPtr.type.pointee + if InitValue.type != TargetType: + if isinstance(InitValue.type, ir.IntType) and isinstance(TargetType, ir.IntType): + if TargetType.width > InitValue.type.width: + InitValue = Gen.builder.zext(InitValue, TargetType, name=f"zext_{VarName}") + elif TargetType.width < InitValue.type.width: + InitValue = Gen.builder.trunc(InitValue, TargetType, name=f"trunc_{VarName}") + elif isinstance(InitValue.type, ir.PointerType) and isinstance(TargetType, ir.PointerType): + InitValue = Gen.builder.bitcast(InitValue, TargetType, name=f"cast_{VarName}") + Gen._store(InitValue, VarPtr) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + Gen._record_var_signedness(VarName, IsUnsigned) + return + if VarName in Gen.global_vars: + if VarName in Gen.module.globals: + GVar: ir.GlobalVariable = Gen.module.globals[VarName] + if InitValue: + Gen._store(InitValue, GVar) + Gen.variables[VarName] = GVar + Gen._record_var_signedness(VarName, IsUnsigned) + return + var: ir.AllocaInstr = Gen._allocaEntry(VarType, name=VarName) + if InitValue and isinstance(InitValue.type, ir.PointerType): + pointee: ir.Type = InitValue.type.pointee + # 当 InitValue 是 i8* (t.CPtr) 且注解是 Gen.structs 中的结构体名时， + # bitcast 为对应的结构体指针，以便正确存储和设置 var_struct_class + # 例如: sub: dict = sub_p (sub_p 是 t.CPtr) + if (isinstance(pointee, ir.IntType) and pointee.width == 8 + and isinstance(Node.annotation, ast.Name) + and Node.annotation.id in Gen.structs): + TargetStructType: ir.Type = Gen.structs[Node.annotation.id] + if isinstance(TargetStructType, ir.PointerType): + TargetStructType = TargetStructType.pointee + InitValue = Gen.builder.bitcast(InitValue, ir.PointerType(TargetStructType), name=f"cast_{VarName}_to_{Node.annotation.id}") + pointee = InitValue.type.pointee + if isinstance(pointee, ir.IdentifiedStructType): + struct_name: str = getattr(pointee, 'name', '') + is_vtable_class: bool = False + for cls_name in Gen.class_vtable: + if struct_name.endswith(f'.{cls_name}') or struct_name == cls_name: + is_vtable_class = True + break + if is_vtable_class and not isinstance(VarType, ir.PointerType): + var = Gen._allocaEntry(InitValue.type, name=VarName) + VarType = InitValue.type + IsPtr = True + Gen._store(InitValue, var) + Gen.variables[VarName] = var + if isinstance(Node.annotation, ast.Name) and Node.annotation.id in Gen.structs: + Gen.var_struct_class[VarName] = Node.annotation.id + Gen.global_struct_class[VarName] = Node.annotation.id + Gen._record_var_signedness(VarName, IsUnsigned) + return + for CN, ST in Gen.structs.items(): + if pointee is ST or pointee == ST: + if CN != Gen.var_struct_class.get(VarName): + Gen.var_struct_class[VarName] = CN + Gen.global_struct_class[VarName] = CN + if isinstance(VarType, ir.PointerType) and VarType.pointee is not ST and VarType.pointee != ST: + var = Gen._allocaEntry(ir.PointerType(pointee), name=VarName) + VarType = ir.PointerType(pointee) + break + if InitValue: + try: + if InitValue.type != VarType: + if isinstance(InitValue.type, ir.IntType) and isinstance(VarType, ir.IntType): + if VarType.width > InitValue.type.width: + InitValue = Gen.builder.zext(InitValue, VarType, name=f"zext_{VarName}") + elif VarType.width < InitValue.type.width: + InitValue = Gen.builder.trunc(InitValue, VarType, name=f"trunc_{VarName}") + elif isinstance(InitValue.type, ir.PointerType) and isinstance(VarType, ir.PointerType): + InitValue = Gen.builder.bitcast(InitValue, VarType, name=f"cast_{VarName}") + elif isinstance(VarType, ir.PointerType) and isinstance(InitValue.type, ir.IntType): + InitValue = Gen.builder.inttoptr(InitValue, VarType, name=f"int2ptr_{VarName}") + elif isinstance(VarType, ir.IntType) and isinstance(InitValue.type, ir.PointerType): + InitValue = Gen.builder.ptrtoint(InitValue, VarType, name=f"ptr2int_{VarName}") + # 大端局部变量：存储时 bswap + InitValue = Gen._apply_bswap_if_big(InitValue, TypeInfo.ByteOrder, f"bswap_store_{VarName}") + Gen._store(InitValue, var) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + Gen.variables[VarName] = var + Gen._record_var_signedness(VarName, IsUnsigned) + # FakeDuck: 初始化影子结构体 a__meta__ 的字段 + _meta_name_check: str = f"{VarName}__meta__" + if _meta_name_check in Gen.variables and '_str' in Gen.structs: + _meta_var_init: ir.Value = Gen.variables[_meta_name_check] + _str_st: ir.Type = Gen.structs['_str'] + # __data__ 字段 = 字符串指针 + _data_off = Gen._get_member_offset('__data__', '_str') + if _data_off is not None and InitValue: + _data_ptr = Gen.builder.gep(_meta_var_init, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _data_off)], name=f"{VarName}_meta_data") + Gen._store(InitValue, _data_ptr) + # __mbuddy__ 字段 = NULL + _mpool_off = Gen._get_member_offset('__mbuddy__', '_str') + if _mpool_off is not None: + _mpool_ptr = Gen.builder.gep(_meta_var_init, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _mpool_off)], name=f"{VarName}_meta_mbuddy") + Gen._store(ir.Constant(ir.PointerType(ir.IntType(8)), None), _mpool_ptr) + # FakeDuck: 从 with 上下文自动注入 requires 字段（如 __mbuddy__） + self.Trans.ExprCallHandle._InjectRequiresFields(_meta_var_init, '_str', Gen) + # 记录局部变量字节序 + if TypeInfo.ByteOrder == 'big': + if isinstance(VarType, ir.IntType) and VarType.width in (16, 32, 64): + Gen.local_var_byteorders[VarName] = 'big' + elif isinstance(VarType, ir.ArrayType) and isinstance(VarType.element, ir.IntType) and VarType.element.width in (16, 32, 64): + Gen.local_var_byteorders[VarName] = 'big' + + if isinstance(Node.annotation, ast.BinOp) and isinstance(Node.annotation.op, ast.BitOr): + found_struct: str | None = FindStructNameInAnnotation(Node.annotation, Gen.structs) + if found_struct: + Gen.var_struct_class[VarName] = found_struct + Gen.global_struct_class[VarName] = found_struct + + if InitValue and isinstance(InitValue.type, ir.PointerType): + pointee: ir.Type = InitValue.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + Gen._var_to_heap_ptr[VarName] = InitValue + if CN != Gen.var_struct_class.get(VarName): + Gen.var_struct_class[VarName] = CN + Gen.global_struct_class[VarName] = CN + break + + if self.Trans.VarScopes: + if TypeInfo: + self.Trans.VarScopes[-1][VarName] = TypeInfo + else: + DefaultInfo: CTypeInfo = CTypeInfo() + DefaultInfo.BaseType = t.CInt + self.Trans.VarScopes[-1][VarName] = DefaultInfo + elif isinstance(Node.target, ast.Attribute): + if isinstance(Node.target.value, ast.Name) and Node.target.value.id == 'self': + if Node.value: + AttrVarType: Any = None + try: + if IsListAnnotation(Node.annotation): + slice_node: ast.AST = Node.annotation.slice + if isinstance(slice_node, ast.Name): + AttrVarType = f't.CArray[{slice_node.id}]' + elif isinstance(slice_node, ast.Tuple): + elts: str = ', '.join(getattr(e, 'id', str(e)) for e in slice_node.elts) + AttrVarType = f't.CArray[{elts}]' + elif isinstance(slice_node, ast.Attribute): + # 还原 t.xxx 形式（如 t.CInt、t.CArray） + if isinstance(slice_node.value, ast.Name): + AttrVarType = f't.CArray[{slice_node.value.id}.{slice_node.attr}]' + else: + AttrVarType = f't.CArray[{ast.unparse(slice_node)}]' + if AttrVarType is None: + AttrTypeInfo: CTypeInfo | None = CTypeInfo.FromNode(Node.annotation, self.Trans.SymbolTable) + if AttrTypeInfo: + AttrVarType = Gen._ctype_to_llvm(AttrTypeInfo) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + Value: ir.Value | None = self.HandleExprLlvm(Node.value, VarType=AttrVarType) + if Value: + self._HandleAttributeStoreLlvm(Node.target, Value) + AttrName: str = Node.target.attr + len_name: str = f'{AttrName}__len' + ClassName: str | None = self._CurrentCpythonObjectClass + if not ClassName: + self_var: ir.Value | None = Gen.variables.get('self') + if self_var: + var_type: ir.Type = self_var.type + if isinstance(var_type, ir.PointerType) and isinstance(var_type.pointee, ir.PointerType): + var_type = var_type.pointee + if isinstance(var_type, ir.PointerType): + pointee: ir.Type = var_type.pointee + for cn, st in Gen.structs.items(): + if st == pointee: + ClassName = cn + break + if ClassName and ClassName in Gen.class_members: + for m_name, m_type in Gen.class_members[ClassName]: + if m_name == len_name: + list_len: ir.Constant | None = None + if isinstance(Node.value, ast.List): + list_len = ir.Constant(ir.IntType(64), len(Node.value.elts)) + if list_len is not None: + SelfVar: ir.Value | None = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: ir.Value = Gen._load(SelfVar, name="self") + offset: int = Gen._get_member_offset(len_name, ClassName) + LenPtr: ir.Value = Gen.builder.gep(SelfPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=len_name) + Gen._store(list_len, LenPtr) + break \ No newline at end of file diff --git a/lib/core/Handles/HandlesAssert.py b/lib/core/Handles/HandlesAssert.py new file mode 100644 index 0000000..93b0392 --- /dev/null +++ b/lib/core/Handles/HandlesAssert.py @@ -0,0 +1,60 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle, EXCEPTION_CODE_MAP +import ast +import llvmlite.ir as ir + +class AssertHandle(BaseHandle): + def _HandleAssertLlvm(self, Node: ast.Assert) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + cond_val: ir.Value | None = self.HandleExprLlvm(Node.test) + if not cond_val: + return + if not isinstance(cond_val.type, ir.IntType) or cond_val.type.width != 1: + zero: ir.Constant = ir.Constant(cond_val.type, 0) + cond_val = Gen.builder.icmp_signed('!=', cond_val, zero, name="assert_cond") + AssertFailBB: ir.Block = Gen.func.append_basic_block(name="assert.fail") + AssertOkBB: ir.Block = Gen.func.append_basic_block(name="assert.ok") + Gen.builder.cbranch(cond_val, AssertOkBB, AssertFailBB) + Gen.builder.position_at_start(AssertFailBB) + exc_code: int = 9 + exc_msg: ir.Value | None = None + if Node.msg: + if isinstance(Node.msg, ast.Constant) and isinstance(Node.msg.value, str): + exc_msg = self.HandleExprLlvm(Node.msg) + elif isinstance(Node.msg, ast.Call) and isinstance(Node.msg.func, ast.Name): + exc_code = EXCEPTION_CODE_MAP.get(Node.msg.func.id, 99) + if Node.msg.args: + first_arg: ast.expr = Node.msg.args[0] + if isinstance(first_arg, ast.Constant) and isinstance(first_arg.value, str): + exc_msg = self.HandleExprLlvm(first_arg) + elif isinstance(Node.msg, ast.Name): + exc_code = EXCEPTION_CODE_MAP.get(Node.msg.id, 99) + else: + exc_msg = self.HandleExprLlvm(Node.msg) + if Gen.eh_except_block_stack: + ExceptBB: ir.Block = Gen.eh_except_block_stack[-1][0] + EndBB: ir.Block | None = Gen.eh_except_block_stack[-1][1] + exception_code: ir.AllocaInstr = Gen.eh_except_block_stack[-1][2] + eh_message: ir.AllocaInstr = Gen.eh_except_block_stack[-1][3] + Gen._store(ir.Constant(ir.IntType(32), exc_code), exception_code) + if exc_msg: + Gen._store(exc_msg, eh_message) + else: + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + Gen._store(null_ptr, eh_message) + Gen.builder.branch(ExceptBB) + else: + printf_func: ir.Function = Gen.get_or_declare_c_func('printf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + if exc_msg: + fmt: ir.Value = Gen.emit_constant("AssertionError: %s\n", 'string') + Gen.builder.call(printf_func, [fmt, exc_msg], name="assert_fail_print") + else: + fmt: ir.Value = Gen.emit_constant("AssertionError\n", 'string') + Gen.builder.call(printf_func, [fmt], name="assert_fail_print") + exit_func: ir.Function = Gen.get_or_declare_c_func('exit', ir.FunctionType(ir.VoidType(), [ir.IntType(32)])) + Gen.builder.call(exit_func, [ir.Constant(ir.IntType(32), 1)], name="call_exit") + Gen.builder.unreachable() + Gen.builder.position_at_start(AssertOkBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesAssign.py b/lib/core/Handles/HandlesAssign.py new file mode 100644 index 0000000..3ad0f32 --- /dev/null +++ b/lib/core/Handles/HandlesAssign.py @@ -0,0 +1,1673 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator +import ast +import sys +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo, FuncMeta +from lib.core.VLogger import get_logger as _vlog +from lib.core.ConstEvaluator import ConstEvaluator, EvalContext +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import IsTModule, ParseListAnnotation, AnnotationContainsName + +class AssignHandle(BaseHandle): + def __init__(self, translator: "Translator") -> None: + super().__init__(translator) + self._CurrentCpythonObjectClass: str | None = None + + @staticmethod + def _contains_identified_struct(var_type: ir.Type) -> bool: + if isinstance(var_type, ir.IdentifiedStructType): + return True + if isinstance(var_type, ir.ArrayType): + return AssignHandle._contains_identified_struct(var_type.element) + if isinstance(var_type, ir.PointerType): + return AssignHandle._contains_identified_struct(var_type.pointee) + if isinstance(var_type, ir.BaseStructType): + if var_type.elements: + return any(AssignHandle._contains_identified_struct(e) for e in var_type.elements) + return False + + def HandleExprLlvm(self, Node: ast.expr, VarType: ir.Type | None = None) -> ir.Value | None: + return self.Trans.ExprHandler.HandleExprLlvm(Node, VarType) + + def _HandleAssignLlvm(self, Node: ast.Assign) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + Gen._set_node_info(Node, "Assign") + if len(Node.targets) == 1: + Target: ast.expr = Node.targets[0] + if isinstance(Target, ast.Name): + VarName: str = Target.id + if VarName in Gen.var_const_flags: + src_info: str = Gen._get_node_info() + _vlog().error(f"不能对 const 变量 '{VarName}' 赋值{src_info}") + sys.exit(1) + if isinstance(Node.value, ast.Attribute) and isinstance(Node.value.value, ast.Name) and IsTModule(Node.value.value.id, self.Trans.SymbolTable): + AttrName: str = Node.value.attr + Gen.var_type_info[VarName] = {'type': 't_type', 'name': AttrName} + Gen.var_type_assignments.setdefault(VarName, []).append({'type': 't_type', 'name': AttrName}) + c_type_name: str = '' + resolved: tuple[type, int] | None = CTypeRegistry.ResolveName(AttrName) + if resolved is not None: + ctype_cls: type = resolved[0] + ptr_level: int = resolved[1] + c_type_name = CTypeRegistry.CTypeToName(ctype_cls) + if ptr_level > 0 and c_type_name: + c_type_name = c_type_name + ' *' + if not c_type_name: + if AttrName == 'CPtr': + c_type_name = 'void *' + is_unsigned: bool = Gen._is_type_unsigned(c_type_name) if c_type_name else False + fmt_str: str = "%u\n" if is_unsigned else "%d\n" + fmt_ptr: ir.Value = Gen._create_string_global(fmt_str) + fmt_var_name: str = f"__{VarName}_fmt" + if fmt_var_name not in Gen.variables: + fmt_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name=fmt_var_name) + Gen.variables[fmt_var_name] = fmt_alloca + Gen.builder.store(fmt_ptr, Gen.variables[fmt_var_name]) + kind_var_name: str = f"__{VarName}_kind" + if kind_var_name not in Gen.variables: + kind_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=kind_var_name) + Gen.variables[kind_var_name] = kind_alloca + Gen.builder.store(ir.Constant(ir.IntType(32), 0), Gen.variables[kind_var_name]) + if VarName not in Gen.variables: + dummy_var: ir.AllocaInstr = Gen._alloca(ir.IntType(8), name=VarName) + Gen.variables[VarName] = dummy_var + return + if isinstance(Node.value, ast.Name) and Node.value.id in Gen.structs: + ClassName: str = Node.value.id + Gen.var_type_info[VarName] = {'type': 'class_cast', 'name': ClassName} + Gen.var_type_assignments.setdefault(VarName, []).append({'type': 'class_cast', 'name': ClassName}) + fmt_str: str = "%p\n" + fmt_ptr: ir.Value = Gen._create_string_global(fmt_str) + fmt_var_name: str = f"__{VarName}_fmt" + if fmt_var_name not in Gen.variables: + fmt_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name=fmt_var_name) + Gen.variables[fmt_var_name] = fmt_alloca + Gen.builder.store(fmt_ptr, Gen.variables[fmt_var_name]) + kind_var_name: str = f"__{VarName}_kind" + if kind_var_name not in Gen.variables: + kind_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=kind_var_name) + Gen.variables[kind_var_name] = kind_alloca + Gen.builder.store(ir.Constant(ir.IntType(32), 1), Gen.variables[kind_var_name]) + if VarName not in Gen.variables: + dummy_var: ir.AllocaInstr = Gen._alloca(ir.IntType(8), name=VarName) + Gen.variables[VarName] = dummy_var + return + if isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name) and Node.value.func.id == 'va_list': + va_list_ptr: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(8).as_pointer(), name="va_list") + Gen.variables[VarName] = va_list_ptr + return + IsStructCtor: bool = (isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name) + and Node.value.func.id in Gen.structs) + if IsStructCtor: + ClassName: str = Node.value.func.id + StructType: ir.Type = Gen.structs[ClassName] + + IsUnion: bool = False + TypeInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(ClassName) + if TypeInfo and TypeInfo.IsUnion: + IsUnion = True + + if IsUnion: + Value: ir.Value | None = self.HandleExprLlvm(Node.value) + if Value: + var: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=VarName) + Gen._store(Value, var) + Gen.variables[VarName] = var + return + + ExistingPtr: ir.Value | None = None + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type.pointee, ir.PointerType): + ExistingPtr = VarPtr + elif VarName in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[VarName] + if isinstance(OldVal.type, ir.PointerType): + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.variables[VarName] = var + del Gen._reg_values[VarName] + ExistingPtr = var + elif VarName in Gen._direct_values: + OldVal: ir.Value = Gen._direct_values.pop(VarName) + if isinstance(OldVal.type, ir.PointerType): + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.variables[VarName] = var + ExistingPtr = var + if ExistingPtr: + pointee: ir.Type = ExistingPtr.type.pointee + # 仅当 alloca 直接存储结构体值（pointee 不是指针）时才 in-place 初始化。 + # 当 pointee 是 PointerType 时，alloca 存储的是堆对象指针： + # - 若 AnnAssign 分支未执行，alloca 内容是未初始化的栈垃圾 + # （可能恰好等于 pool 地址），_InitStructAtPtr 会向垃圾地址写字段 → 内存损坏 + # - 即使指针有效，node = Class(...) 的语义也是创建新对象并更新指针， + # 而非在旧对象位置 in-place 重初始化 + # 此时必须 fall through 到通用路径 HandleExprLlvm → _HandleClassNewLlvm + # 以正确调用 __new__/__before_init__/__init__ 构造新对象。 + if not isinstance(pointee, ir.PointerType) and pointee == StructType: + self._InitStructAtPtr(Node.value, ClassName, ExistingPtr) + return + if isinstance(Node.value, ast.Attribute): + enum_type_name: str | None = self._CheckEnumMemberAssignment(Node.value) + if enum_type_name: + Gen.var_type_info[VarName] = {'type': 'enum', 'name': enum_type_name} + var_type_for_list: ir.Type | None = None + if isinstance(Node.value, ast.List) and VarName in Gen.var_type_info: + type_info: dict[str, str] = Gen.var_type_info[VarName] + if type_info.get('type') == 'CArray': + var_type_for_list = type_info.get('full_type') + Value: ir.Value | None = self.HandleExprLlvm(Node.value, VarType=var_type_for_list) + if not Value: + return + if isinstance(Value.type, ir.VoidType): + return + Gen._UnregisterTempPtr(Value) + # 大端局部变量：赋值时 bswap + Value = Gen._apply_bswap_if_big(Value, Gen.local_var_byteorders.get(VarName, ""), f"bswap_assign_{VarName}") + if isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + found_class: str | None = None + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(Value.type.pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + found_class = CN + Gen.var_struct_class[VarName] = CN + Gen._var_to_heap_ptr[VarName] = Value + if found_class is None and isinstance(Value.type.pointee, ir.IdentifiedStructType): + pointee_name: str = Value.type.pointee.name + pointee_short: str = Gen._extract_short_name(pointee_name) + best_elem_count: int = 0 + for CN, ST in Gen.structs.items(): + if isinstance(ST, ir.IdentifiedStructType): + st_short: str = Gen._extract_short_name(ST.name) + if st_short == pointee_short: + elem_count: int = len(ST.elements) if ST.elements else 0 + if elem_count > best_elem_count: + found_class = CN + best_elem_count = elem_count + if found_class: + Gen.var_struct_class[VarName] = found_class + Gen._var_to_heap_ptr[VarName] = Value + # 泛型类特化触发：返回类型是泛型类特化（如 list[str]）但结构体尚未创建， + # 或虽然找到了结构体但它是 opaque（来自 stub 声明，尚未发射方法体） + _fd_struct_opaque: bool = False + if found_class and found_class in Gen.structs: + _fd_st: ir.IdentifiedStructType = Gen.structs[found_class] + if isinstance(_fd_st, ir.IdentifiedStructType) and (_fd_st.elements is None or len(_fd_st.elements) == 0): + _fd_struct_opaque = True + if (found_class is None or _fd_struct_opaque) and isinstance(Value.type.pointee, ir.IdentifiedStructType): + pn: str = Value.type.pointee.name + short_pn: str = pn + if '.' in pn: + pn_parts: list[str] = pn.split('.', 1) + if len(pn_parts[0]) >= 8 and all(c in '0123456789abcdef' for c in pn_parts[0]): + short_pn = pn_parts[1] + if '[' in short_pn and short_pn.endswith(']'): + gc_base: str = short_pn.split('[')[0] + gc_args_raw: str = short_pn[len(gc_base):] + gc_type_args: list[str] = [] + for gc_arg_part in gc_args_raw.split(']['): + gc_arg_part = gc_arg_part.strip('[]') + if gc_arg_part: + gc_type_args.append(gc_arg_part) + if (gc_type_args + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and gc_base in self.Trans.ClassHandler._generic_class_templates): + gc_type_names: list[str] = [] + for gc_ta in gc_type_args: + gc_tn: str = gc_ta + if gc_ta == 'int': + gc_tn = 'CInt' + elif gc_ta == 'double': + gc_tn = 'CDouble' + elif gc_ta == 'float': + gc_tn = 'CFloat' + elif gc_ta == 'char': + gc_tn = 'CChar' + elif gc_ta == 'bool': + gc_tn = 'CBool' + gc_type_names.append(gc_tn) + gc_spec: str | None = self.Trans.ClassHandler._specialize_generic_class( + gc_base, gc_type_args, Gen, type_names=gc_type_names) + if gc_spec and gc_spec in Gen.structs: + found_class = gc_spec + Gen.var_struct_class[VarName] = gc_spec + Gen._var_to_heap_ptr[VarName] = Value + else: + if VarName in ('mctx', 's1ctx', 's2ctx', 's5ctx'): + pass + if isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + if VarName in Gen.variables and Gen.variables[VarName] is not None: + Gen._store(Value, Gen.variables[VarName]) + else: + var: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=VarName) + Gen._store(Value, var) + Gen.variables[VarName] = var + if isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name): + call_func_name: str = Node.value.func.id + if call_func_name in Gen.var_type_info: + type_info: dict[str, str] = Gen.var_type_info[call_func_name] + if type_info['type'] in ('t_type', 'class_cast'): + src_fmt_var: str = f"__{call_func_name}_fmt" + dst_fmt_var: str = f"__{VarName}_fmt" + if src_fmt_var in Gen.variables: + if dst_fmt_var not in Gen.variables: + dst_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name=dst_fmt_var) + Gen.variables[dst_fmt_var] = dst_alloca + src_fmt_ptr: ir.Value = Gen._load(Gen.variables[src_fmt_var], name=f"Load_{call_func_name}_fmt_for_{VarName}") + Gen.builder.store(src_fmt_ptr, Gen.variables[dst_fmt_var]) + src_kind_var: str = f"__{call_func_name}_kind" + dst_kind_var: str = f"__{VarName}_kind" + if src_kind_var in Gen.variables: + if dst_kind_var not in Gen.variables: + dst_kind_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=dst_kind_var) + Gen.variables[dst_kind_var] = dst_kind_alloca + src_kind_val: ir.Value = Gen._load(Gen.variables[src_kind_var], name=f"Load_{call_func_name}_kind_for_{VarName}") + Gen.builder.store(src_kind_val, Gen.variables[dst_kind_var]) + Gen.var_type_info[VarName] = {'type': 't_type_runtime', 'source': call_func_name} + return + if VarName in Gen.global_vars: + if VarName in Gen.module.globals: + GVar: ir.GlobalVariable = Gen.module.globals[VarName] + Gen._store(Value, GVar) + Gen.variables[VarName] = GVar + if isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(Value.type.pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + Gen.var_struct_class[VarName] = CN + elif VarName in Gen.variables and Gen.variables[VarName] is not None: + Gen._store(Value, Gen.variables[VarName]) + return + if VarName in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[VarName] + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.variables[VarName] = var + del Gen._reg_values[VarName] + if VarName in Gen._direct_values: + OldVal: ir.Value = Gen._direct_values.pop(VarName) + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.variables[VarName] = var + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: ir.Value = Gen.variables[VarName] + TargetType: ir.Type | None = VarPtr.type.pointee if isinstance(VarPtr.type, ir.PointerType) else None + types_differ: bool + try: + types_differ = TargetType and Value.type != TargetType + except AssertionError: + types_differ = False + if isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, ir.PointerType): + LoadedValue: ir.Value = Gen._load(Value, name=f"Load_{VarName}") + Loaded_differ: bool + try: + Loaded_differ = TargetType and LoadedValue.type != TargetType + except AssertionError: + Loaded_differ = False + if Loaded_differ: + try: + Coerced: ir.Value | None = Gen._coerce_value(LoadedValue, TargetType) + if Coerced is not None: + LoadedValue = Coerced + except AssertionError: + pass + Gen._store(LoadedValue, VarPtr) + elif types_differ: + try: + Coerced: ir.Value | None = Gen._coerce_value(Value, TargetType) + if Coerced is not None: + Gen._store(Coerced, VarPtr) + else: + Gen._store(Value, VarPtr) + except AssertionError: + Gen._store(Value, VarPtr) + else: + Gen._store(Value, VarPtr) + if VarName not in Gen.variables: + # 检查是否为模块级全局变量 + # main 函数代表模块级作用域，直接修改全局变量 + # 其他函数中，未声明 global 且未先读取的赋值创建局部变量（Python 作用域语义） + # 先读取过的变量已通过 HandleExprLlvm 加入 Gen.variables，会走 L289 分支存储到全局 + CurrentFuncName: str = getattr(Gen.func, 'name', '') if Gen.func else '' + IsMainScope: bool = CurrentFuncName == 'main' or CurrentFuncName.endswith('.main') + if VarName in Gen.module.globals and (VarName in Gen.global_vars or IsMainScope): + GVar: ir.GlobalVariable = Gen.module.globals[VarName] + Gen._store(Value, GVar) + Gen.variables[VarName] = GVar + elif isinstance(Value.type, ir.PointerType) and isinstance(Value.type.pointee, ir.PointerType): + LoadedValue: ir.Value = Gen._load(Value, name=f"Load_{VarName}") + var: ir.AllocaInstr = Gen._allocaEntry(LoadedValue.type, name=VarName) + Gen._store(LoadedValue, var) + Gen.variables[VarName] = var + else: + var: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=VarName) + Gen._store(Value, var) + Gen.variables[VarName] = var + is_u: bool = Gen._check_node_unsigned(Node.value) + Gen._record_var_signedness(VarName, 'unsigned int' if is_u else 'int') + if isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name): + call_func_name: str = Node.value.func.id + if call_func_name in Gen.var_type_info: + type_info: dict[str, str] = Gen.var_type_info[call_func_name] + if type_info['type'] in ('t_type', 'class_cast'): + src_fmt_var: str = f"__{call_func_name}_fmt" + dst_fmt_var: str = f"__{VarName}_fmt" + if src_fmt_var in Gen.variables: + if dst_fmt_var not in Gen.variables: + dst_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name=dst_fmt_var) + Gen.variables[dst_fmt_var] = dst_alloca + src_fmt_ptr: ir.Value = Gen._load(Gen.variables[src_fmt_var], name=f"Load_{call_func_name}_fmt_for_{VarName}") + Gen.builder.store(src_fmt_ptr, Gen.variables[dst_fmt_var]) + src_kind_var: str = f"__{call_func_name}_kind" + dst_kind_var: str = f"__{VarName}_kind" + if src_kind_var in Gen.variables: + if dst_kind_var not in Gen.variables: + dst_kind_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=dst_kind_var) + Gen.variables[dst_kind_var] = dst_kind_alloca + src_kind_val: ir.Value = Gen._load(Gen.variables[src_kind_var], name=f"Load_{call_func_name}_kind_for_{VarName}") + Gen.builder.store(src_kind_val, Gen.variables[dst_kind_var]) + Gen.var_type_info[VarName] = {'type': 't_type_runtime', 'source': call_func_name} + else: + Gen.var_type_info[VarName] = {'type': 't_type_runtime', 'source': call_func_name} + elif isinstance(Target, ast.Attribute): + TargetType: ir.Type | None = None + AttrName: str = Target.attr + VarName: str | None = None + if isinstance(Target.value, ast.Name): + VarName = Target.value.id + if VarName: + ClassName: str | None = Gen.var_struct_class.get(VarName) + if not ClassName and Gen.global_struct_class: + ClassName = Gen.global_struct_class.get(VarName) + if ClassName: + Gen.var_struct_class[VarName] = ClassName + if not ClassName: + for CN in Gen.structs: + if VarName == CN.lower() or VarName.startswith(CN): + ClassName = CN + break + if ClassName and ClassName in Gen.class_members: + for m_name, m_type in Gen.class_members[ClassName]: + if m_name == AttrName: + TargetType = m_type + break + Value: ir.Value | None = self.HandleExprLlvm(Node.value, VarType=TargetType) + if Value: + self._HandleAttributeStoreLlvm(Target, Value) + elif isinstance(Target, ast.Subscript): + TargetType: ir.Type | None = None + SubVarName: str | None = None + if isinstance(Target.value, ast.Name): + SubVarName = Target.value.id + if SubVarName in Gen.var_struct_class: + ClassName: str = Gen.var_struct_class[SubVarName] + if ClassName in Gen.class_members: + for m_name, m_type in Gen.class_members[ClassName]: + if m_name == SubVarName: + if isinstance(m_type, ir.PointerType): + pointee: ir.Type = m_type.pointee + if isinstance(pointee, ir.ArrayType): + TargetType = pointee.element + elif isinstance(m_type, ir.ArrayType): + TargetType = m_type.element + break + if not TargetType and isinstance(Target.value, ast.Name): + SubVarName = Target.value.id + var_val: ir.Value | None = Gen.variables.get(SubVarName) + if var_val is None and SubVarName in Gen._direct_values: + var_val = Gen._direct_values[SubVarName] + if var_val is not None: + var_type: ir.Type = var_val.type + if isinstance(var_type, ir.PointerType): + pointee: ir.Type = var_type.pointee + if isinstance(pointee, ir.ArrayType): + TargetType = pointee.element + elif isinstance(var_type, ir.ArrayType): + TargetType = var_type.element + Value: ir.Value | None = self.HandleExprLlvm(Node.value, VarType=TargetType) + if Value: + self._HandleSubscriptStoreLlvm(Target, Value) + elif isinstance(Target, ast.Tuple): + if isinstance(Node.value, ast.Call): + FuncName: str | None = None + ModulePath: str | None = None + if isinstance(Node.value.func, ast.Name): + FuncName = Node.value.func.id + elif isinstance(Node.value.func, ast.Attribute): + FuncName = Node.value.func.attr + ModulePath = self.Trans.ExprCallHandle._get_ModulePath(Node.value.func.value) + CReturnTypes: list[ast.expr] = [] + FuncDef: ast.FunctionDef | None = self.Trans.FunctionDefCache.get(FuncName) if FuncName else None + if FuncDef: + if FuncDef.returns and isinstance(FuncDef.returns, ast.Subscript): + if isinstance(FuncDef.returns.value, ast.Name) and FuncDef.returns.value.id == 'tuple': + slice_node: ast.expr = FuncDef.returns.slice + if isinstance(slice_node, ast.Tuple): + CReturnTypes = slice_node.elts + else: + CReturnTypes = [slice_node] + if FuncDef.decorator_list: + for decorator in FuncDef.decorator_list: + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Attribute): + if decorator.func.attr == 'CReturn': + for arg in decorator.args: + CReturnTypes.append(arg) + if not CReturnTypes and FuncName: + func: ir.Function | None = Gen.functions.get(FuncName) + if func: + func_ft: ir.FunctionType | None = getattr(func, 'ftype', None) + if func_ft: + ret_type: ir.Type = func_ft.return_type + if isinstance(ret_type, ir.LiteralStructType): + CReturnTypes = [None] * len(ret_type.elements) + if not CReturnTypes: + sym_key: str = FuncName + if not self.Trans.SymbolTable.has(sym_key): + if ModulePath: + sym_key = f"{ModulePath}.{FuncName}" + sym_info: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(sym_key) + if sym_info and sym_info.IsFunction: + ret_type_info: CTypeInfo | None = sym_info.FuncPtrReturn + param_type_infos: list[tuple[str, CTypeInfo]] = [pt for _, pt in (sym_info.FuncPtrParams or [])] + if ret_type_info: + if isinstance(ret_type_info, CTypeInfo) and ret_type_info.BaseType: + ret_type: ir.Type = ret_type_info.ToLLVM(Gen) + else: + ret_type = Gen._type_str_to_llvm(str(ret_type_info) if ret_type_info else 'i32', False) + if isinstance(ret_type, ir.LiteralStructType): + CReturnTypes = [None] * len(ret_type.elements) + if FuncName not in Gen.functions: + if isinstance(ret_type, ir.VoidType): + ret_type = ir.IntType(32) + llvm_param_types: list[ir.Type] = [] + for pt in param_type_infos: + if isinstance(pt, CTypeInfo) and pt.BaseType: + lp: ir.Type = pt.ToLLVM(Gen) + else: + lp = Gen._type_str_to_llvm(str(pt) if pt else 'i32', '*' in str(pt) if isinstance(pt, str) else False) + if isinstance(lp, ir.VoidType): + lp = ir.IntType(8).as_pointer() + llvm_param_types.append(lp) + func_type: ir.FunctionType = ir.FunctionType(ret_type, llvm_param_types) + MangledName: str = Gen._mangle_func_name(FuncName) + func_decl: ir.Function = ir.Function(Gen.module, func_type, name=MangledName) + Gen.functions[MangledName] = func_decl + Gen.functions[FuncName] = func_decl + if not CReturnTypes and FuncName: + func: ir.Function | None = Gen.functions.get(FuncName) + if func: + func_ft: ir.FunctionType | None = getattr(func, 'ftype', None) + if func_ft: + ret_type: ir.Type = func_ft.return_type + if isinstance(ret_type, ir.LiteralStructType): + CReturnTypes = [None] * len(ret_type.elements) + if CReturnTypes and len(Target.elts) == len(CReturnTypes): + CallArgs: list[ir.Value | None] = [] + for arg in Node.value.args: + ArgVal: ir.Value | None = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + for kw in Node.value.keywords: + pass + func: ir.Function | None = Gen.functions.get(FuncName) + func_ft: ir.FunctionType | None = getattr(func, 'ftype', None) if func else None + func_params: list[ir.Type] = list(getattr(func_ft, 'args', []) or []) if func_ft else [] + while len(CallArgs) < len(func_params): + FuncDef: ast.FunctionDef | None = self.Trans.FunctionDefCache.get(FuncName) + if FuncDef and len(CallArgs) < len(FuncDef.args.args): + arg_def: list[ast.expr] = FuncDef.args.defaults + arg_idx: int = len(CallArgs) + total_args: int = len(FuncDef.args.args) + default_start: int = total_args - len(arg_def) + if arg_idx >= default_start: + def_idx: int = arg_idx - default_start + if def_idx < len(arg_def): + DefVal: ir.Value | None = self.HandleExprLlvm(arg_def[def_idx]) + if DefVal: + expected_type: ir.Type = func_params[arg_idx] + if DefVal.type != expected_type: + if isinstance(expected_type, ir.IntType) and isinstance(DefVal.type, ir.IntType): + if DefVal.type.width < expected_type.width: + DefVal = Gen.builder.zext(DefVal, expected_type, name=f"zext_default_{arg_idx}") + elif DefVal.type.width > expected_type.width: + DefVal = Gen.builder.trunc(DefVal, expected_type, name=f"trunc_default_{arg_idx}") + CallArgs.append(DefVal) + continue + if len(CallArgs) < len(func_params): + CallArgs.append(ir.Constant(func_params[len(CallArgs)], 0)) + else: + break + adjusted: list[ir.Value] = Gen._adjust_args(CallArgs, func) + call_result: ir.Value = Gen.builder.call(func, adjusted, name=f"call_{FuncName}") + func_ft: ir.FunctionType | None = getattr(func, 'ftype', None) + ret_type: ir.Type | None = getattr(func_ft, 'return_type', None) if func_ft else None + for j, elt in enumerate(Target.elts): + if isinstance(elt, ast.Name): + VarName: str = elt.id + field_val: ir.Value = Gen.builder.extract_value(call_result, j, name=f"extract_{FuncName}_{j}") + VarType: ir.Type + if CReturnTypes[j] is not None: + ReturnTypeInfo: CTypeInfo | None = CTypeInfo.FromNode(CReturnTypes[j], self.Trans.SymbolTable) + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + VarType = Gen._ctype_to_llvm(ReturnTypeInfo) + elif ret_type and isinstance(ret_type, ir.LiteralStructType) and j < len(ret_type.elements): + VarType = ret_type.elements[j] + else: + VarType = field_val.type + if isinstance(VarType, ir.VoidType): + VarType = ir.IntType(32) + var: ir.AllocaInstr = Gen._alloca(VarType, name=VarName) + Gen.variables[VarName] = var + if CReturnTypes[j] is not None: + Gen._record_var_signedness(VarName, ReturnTypeStr) + elif ret_type and isinstance(ret_type, ir.LiteralStructType) and j < len(ret_type.elements): + elem_type: ir.Type = ret_type.elements[j] + if isinstance(elem_type, ir.IntType): + if elem_type.width == 16: + Gen._record_var_signedness(VarName, 'unsigned short') + elif elem_type.width == 32: + Gen._record_var_signedness(VarName, 'unsigned int') + elif elem_type.width == 64: + Gen._record_var_signedness(VarName, 'unsigned long long') + Gen._store(field_val, var) + return + + def _InitStructAtPtr(self, call_node: ast.Call, ClassName: str, VarPtr: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if ClassName not in Gen.structs: + return + StructType: ir.Type = Gen.structs[ClassName] + StructPtrType: ir.PointerType = ir.PointerType(StructType) + RawPtr: ir.Value = Gen._load(VarPtr, name=getattr(VarPtr, 'name', 'ptr')) + StructPtr: ir.Value = Gen.builder.bitcast(RawPtr, StructPtrType, name=ClassName) + members: list[tuple[str, ir.Type]] = Gen.class_members.get(ClassName, []) + defaults: dict[str, ir.Value] = Gen.class_member_defaults.get(ClassName, {}) + member_values: dict[str, ir.Value] = {} + for member_name, member_type in members: + if member_name in defaults: + member_values[member_name] = defaults[member_name] + if call_node.args: + for i, arg in enumerate(call_node.args): + if i < len(members): + member_name: str = members[i][0] + val: ir.Value | None = self.HandleExprLlvm(arg) + if val: + member_values[member_name] = val + if call_node.keywords: + for kw in call_node.keywords: + val: ir.Value | None = self.HandleExprLlvm(kw.value) + if val: + member_values[kw.arg] = val + has_vtable: bool = ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes + base: int = 1 if has_vtable else 0 + for i, (member_name, member_type) in enumerate(members): + if isinstance(member_type, ir.VoidType): + continue + if member_name in member_values: + ElemPtr: ir.Value = Gen.builder.gep(StructPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), base + i)], name=f"{ClassName}_{member_name}") + try: + Gen._store(member_values[member_name], ElemPtr) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + InitFuncName: str = f'{ClassName}.__init__' + if Gen._has_function(InitFuncName): + Args: list[ir.Value | None] = [self.HandleExprLlvm(arg) for arg in call_node.args] + Args = [a for a in Args if a] + Gen.builder.call(Gen._get_function(InitFuncName), [StructPtr] + Args, name=f"call_{InitFuncName}") + + def _HandleAttributeStoreLlvm(self, Target: ast.Attribute, Value: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + AttrName: str = Target.attr + if isinstance(Target.value, ast.Name) and Target.value.id == 'self': + ClassName: str | None = self.Trans._CurrentCpythonObjectClass + if ClassName and ClassName in Gen.structs: + PropKey: str = f'{ClassName}.{AttrName}' + PropInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_SETTER in PropInfo.MetaList: + SelfVar: ir.Value | None = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: ir.Value = Gen._load(SelfVar, name="self") + SetterFunc: ir.Function | None = Gen._get_function(PropKey + '$set') + if SetterFunc and SelfPtr: + Gen.builder.call(SetterFunc, [SelfPtr, Value], name=f"prop_set_{PropKey}") + return + SelfVar: ir.Value | None = Gen._get_var_ptr('self') + if not SelfVar: + return + if SelfVar: + SelfPtr: ir.Value = Gen._load(SelfVar, name="self") + is_vtable_method: bool = False + method_idx: int = 0 + if ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes: + if ClassName in Gen.class_methods: + for mi, mn in enumerate(Gen.class_methods[ClassName]): + method_short: str = mn.split('.')[-1] if '.' in mn else mn + if method_short == AttrName or mn == AttrName: + is_vtable_method = True + method_idx = mi + break + if is_vtable_method: + VtableSlotPtr: ir.Value = Gen.builder.gep(SelfPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"vtable_slot_{ClassName}") + VtablePtr: ir.Value = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}") + methods: list[str] = Gen.class_methods[ClassName] + VtableArrayType: ir.ArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + ClassVtable: ir.GlobalVariable | None = Gen.Vtables.get(ClassName) + if ClassVtable: + ClassVtableAddr: ir.Value = Gen.builder.bitcast(ClassVtable, ir.PointerType(ir.IntType(8)), name=f"class_vtable_addr_{ClassName}") + IsSameVtable: ir.Value = Gen.builder.icmp_unsigned('==', VtablePtr, ClassVtableAddr, name=f"is_same_vtable_{ClassName}") + CopyBlock: ir.Block = Gen.func.append_basic_block(name=f"vtable_copy_self_{AttrName}") + SkipBlock: ir.Block = Gen.func.append_basic_block(name=f"vtable_skip_self_{AttrName}") + Gen.builder.cbranch(IsSameVtable, CopyBlock, SkipBlock) + Gen.builder.position_at_end(CopyBlock) + Gen._vtable_copy_counter += 1 + CopyName: str = f"{Gen._mangle_name(ClassName)}_vtable_copy_self_{Gen._vtable_copy_counter}" + VtableCopyGlobal: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VtableArrayType, name=CopyName) + VtableCopyGlobal.linkage = 'internal' + VtableCopyGlobal.initializer = ir.Constant(VtableArrayType, [ir.Constant(ir.PointerType(ir.IntType(8)), None)] * len(methods)) + VtableCopyTyped: ir.Value = Gen.builder.bitcast(VtableCopyGlobal, ir.PointerType(ir.IntType(8)), name=f"vtable_copy_typed") + SrcTyped: ir.Value = Gen.builder.bitcast(ClassVtable, ir.PointerType(ir.IntType(8)), name=f"src_vtable_typed") + VtableSize: int = len(methods) * Gen.ptr_size + MemcpyFunc: ir.Function | None = Gen._find_function('llvm.memcpy.p0i8.p0i8.i64') + if not MemcpyFunc: + MemcpyType: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64), ir.IntType(1)]) + MemcpyFunc = ir.Function(Gen.module, MemcpyType, name='llvm.memcpy.p0i8.p0i8.i64') + Gen.functions['llvm.memcpy.p0i8.p0i8.i64'] = MemcpyFunc + Gen.builder.call(MemcpyFunc, [VtableCopyTyped, SrcTyped, ir.Constant(ir.IntType(64), VtableSize), ir.Constant(ir.IntType(1), 0)]) + VtableCopyGlobalAddr: ir.Value = Gen.builder.bitcast(VtableCopyGlobal, ir.PointerType(ir.IntType(8)), name=f"vtable_copy_addr") + Gen._store(VtableCopyGlobalAddr, VtableSlotPtr) + Gen.builder.branch(SkipBlock) + Gen.builder.position_at_end(SkipBlock) + VtablePtr = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}_after_copy") + VtableTyped: ir.Value = Gen.builder.bitcast(VtablePtr, ir.PointerType(VtableArrayType), name=f"vtable_typed_{ClassName}") + MethodPtrAddr: ir.Value = Gen.builder.gep(VtableTyped, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), method_idx)], name=f"vmethod_slot_{AttrName}") + FuncPtrI8: ir.Value = Gen.builder.bitcast(Value, ir.PointerType(ir.IntType(8)), name=f"func_ptr_i8_{AttrName}") + Gen._store(FuncPtrI8, MethodPtrAddr) + return + offset: int = Gen._get_member_offset(AttrName, ClassName) + struct_type: ir.Type = Gen.structs[ClassName] + max_offset: int = len(struct_type.elements) if isinstance(struct_type, ir.IdentifiedStructType) else 0 + if offset >= max_offset: + VarName: str = f"self.{AttrName}" + if VarName not in Gen.variables: + NewVar: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=AttrName) + Gen.variables[VarName] = NewVar + Gen._store(Value, Gen.variables[VarName]) + return + MemberPtr: ir.Value = Gen.builder.gep(SelfPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + elif isinstance(Target.value, ast.Name): + VarName: str = Target.value.id + + # FakeDuck: 检查是否有影子结构体 a__meta__ (str 变量的 per-instance 字段存储) + # 直接通过 a.__field__ = value 访问影子结构体字段 (语法糖) + _fd_meta_name: str = f"{VarName}__meta__" + if _fd_meta_name in Gen.variables and '_str' in Gen.structs: + _fd_offset = Gen._get_member_offset(AttrName, '_str') + if _fd_offset is not None: + _fd_meta_ptr: ir.Value = Gen.variables[_fd_meta_name] + _fd_field_ptr: ir.Value = Gen.builder.gep(_fd_meta_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _fd_offset)], name=f"{VarName}_{AttrName}_ptr") + self._StoreWithCoerce(Value, _fd_field_ptr) + return + + ClassName: str | None = Gen.var_struct_class.get(VarName) + + if not ClassName and Gen.global_struct_class: + ClassName = Gen.global_struct_class.get(VarName) + if ClassName: + Gen.var_struct_class[VarName] = ClassName + + if not ClassName: + for CN in Gen.structs: + if VarName == CN.lower() or VarName.startswith(CN): + ClassName = CN + break + + if not ClassName and VarName in Gen.module.globals: + GVar: ir.GlobalVariable = Gen.module.globals[VarName] + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(GVar.type.pointee) + if found: + ClassName = found[0] + Gen.var_struct_class[VarName] = ClassName + if Gen.global_struct_class: + Gen.global_struct_class[VarName] = ClassName + + if not ClassName and VarName in Gen.variables: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType): + pointee: ir.Type = VarPtr.type.pointee + if isinstance(pointee, ir.PointerType): + pointee = pointee.pointee + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + ClassName = CN + elif isinstance(pointee, ir.IdentifiedStructType): + # sha1 冲突 fallback：find_struct_by_pointee 因 sha1 前缀不匹配失败时， + # 用短名（去掉 sha1 前缀）匹配，优先选字段最多的完整版本 + pointee_short: str = Gen._extract_short_name(pointee.name) + best_elem_count: int = 0 + for CN, ST in Gen.structs.items(): + if isinstance(ST, ir.IdentifiedStructType): + st_short: str = Gen._extract_short_name(ST.name) + if st_short == pointee_short: + elem_count: int = len(ST.elements) if ST.elements else 0 + if elem_count > best_elem_count: + ClassName = CN + best_elem_count = elem_count + + IsUnion: bool = False + if ClassName: + TypeInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(ClassName) + if TypeInfo and TypeInfo.IsUnion: + IsUnion = True + + if ClassName and ClassName in Gen.structs: + PropKey: str = f'{ClassName}.{AttrName}' + PropInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_SETTER in PropInfo.MetaList: + ObjVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + SetterFunc: ir.Function | None = Gen._get_function(PropKey + '$set') + if SetterFunc and ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{ClassName}") + Gen.builder.call(SetterFunc, [ObjVal, Value], name=f"prop_set_{PropKey}") + return + ObjVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + if ObjVal: + if BaseHandle._is_char_pointer(ObjVal): + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{ClassName}") + # 跨模块 opaque 类型：ObjVal.type.pointee 与 Gen.structs[ClassName] 可能是 + # 不同的 IdentifiedStructType 对象（虽然 name 相同），== 返回 False。 + # 使用 name 匹配作为 fallback，与链式赋值路径（第 926-927 行）一致。 + # sha1 冲突时还需短名匹配（如 6bc91356c652f8ca.Function vs 8ad0c0a2934ca5b4.Function） + _ObjPointee = ObjVal.type.pointee if isinstance(ObjVal.type, ir.PointerType) else None + _TargetStruct = Gen.structs.get(ClassName) + _MatchesStruct = (_ObjPointee is not None and _TargetStruct is not None and + (_ObjPointee == _TargetStruct or + (isinstance(_ObjPointee, ir.IdentifiedStructType) and isinstance(_TargetStruct, ir.IdentifiedStructType) and + (_ObjPointee.name == _TargetStruct.name or + Gen._extract_short_name(_ObjPointee.name) == Gen._extract_short_name(_TargetStruct.name))))) + if isinstance(ObjVal.type, ir.PointerType) and _MatchesStruct: + if IsUnion: + NestedStructName: str = f"{ClassName}_{AttrName}" + if NestedStructName in Gen.structs: + NestedStructType: ir.Type = Gen.structs[NestedStructName] + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(NestedStructType), name=f"cast_{NestedStructName}") + offset: int | None = Gen._get_member_offset(AttrName, NestedStructName) + if offset is not None and offset < len(NestedStructType.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + return + bitfield_offsets: dict[str, int] = Gen.class_member_bitoffsets.get(ClassName, {}) + if AttrName in bitfield_offsets: + self._store_bitfield_member(ObjVal, ClassName, AttrName, Value) + return + bitfields: dict[str, int] = Gen.class_member_bitfields.get(ClassName, {}) + if AttrName in bitfields and bitfields[AttrName] > 0: + if ClassName not in Gen.class_member_bitoffsets: + Gen.class_member_bitoffsets[ClassName] = {} + if AttrName not in Gen.class_member_bitoffsets[ClassName]: + bo: int = 0 + for bf_name, bf_width in bitfields.items(): + if bf_name == AttrName: + break + bo += bf_width + Gen.class_member_bitoffsets[ClassName][AttrName] = bo + self._store_bitfield_member(ObjVal, ClassName, AttrName, Value) + return + is_vtable_method: bool = False + method_idx: int = 0 + if ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes: + if ClassName in Gen.class_methods: + for mi, mn in enumerate(Gen.class_methods[ClassName]): + method_short: str = mn.split('.')[-1] if '.' in mn else mn + if method_short == AttrName or mn == AttrName: + is_vtable_method = True + method_idx = mi + break + if is_vtable_method: + VtableSlotPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"vtable_slot_{ClassName}") + VtablePtr: ir.Value = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}") + methods: list[str] = Gen.class_methods[ClassName] + VtableArrayType: ir.ArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + ClassVtable: ir.GlobalVariable | None = Gen.Vtables.get(ClassName) + if ClassVtable: + ClassVtableAddr: ir.Value = Gen.builder.bitcast(ClassVtable, ir.PointerType(ir.IntType(8)), name=f"class_vtable_addr_{ClassName}") + IsSameVtable: ir.Value = Gen.builder.icmp_unsigned('==', VtablePtr, ClassVtableAddr, name=f"is_same_vtable_{ClassName}") + CopyBlock: ir.Block = Gen.func.append_basic_block(name=f"vtable_copy_{AttrName}") + SkipBlock: ir.Block = Gen.func.append_basic_block(name=f"vtable_skip_{AttrName}") + Gen.builder.cbranch(IsSameVtable, CopyBlock, SkipBlock) + Gen.builder.position_at_end(CopyBlock) + Gen._vtable_copy_counter += 1 + CopyName: str = f"{Gen._mangle_name(ClassName)}_vtable_copy_{Gen._vtable_copy_counter}" + VtableCopyGlobal: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VtableArrayType, name=CopyName) + VtableCopyGlobal.linkage = 'internal' + VtableCopyGlobal.initializer = ir.Constant(VtableArrayType, [ir.Constant(ir.PointerType(ir.IntType(8)), None)] * len(methods)) + VtableCopyTyped: ir.Value = Gen.builder.bitcast(VtableCopyGlobal, ir.PointerType(ir.IntType(8)), name=f"vtable_copy_typed") + SrcTyped: ir.Value = Gen.builder.bitcast(ClassVtable, ir.PointerType(ir.IntType(8)), name=f"src_vtable_typed") + VtableSize: int = len(methods) * Gen.ptr_size + MemcpyFunc: ir.Function | None = Gen._find_function('llvm.memcpy.p0i8.p0i8.i64') + if not MemcpyFunc: + MemcpyType: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64), ir.IntType(1)]) + MemcpyFunc = ir.Function(Gen.module, MemcpyType, name='llvm.memcpy.p0i8.p0i8.i64') + Gen.functions['llvm.memcpy.p0i8.p0i8.i64'] = MemcpyFunc + Gen.builder.call(MemcpyFunc, [VtableCopyTyped, SrcTyped, ir.Constant(ir.IntType(64), VtableSize), ir.Constant(ir.IntType(1), 0)]) + VtableCopyGlobalAddr: ir.Value = Gen.builder.bitcast(VtableCopyGlobal, ir.PointerType(ir.IntType(8)), name=f"vtable_copy_addr") + Gen._store(VtableCopyGlobalAddr, VtableSlotPtr) + Gen.builder.branch(SkipBlock) + Gen.builder.position_at_end(SkipBlock) + VtablePtr = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}_after_copy") + VtableTyped: ir.Value = Gen.builder.bitcast(VtablePtr, ir.PointerType(VtableArrayType), name=f"vtable_typed_{ClassName}") + MethodPtrAddr: ir.Value = Gen.builder.gep(VtableTyped, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), method_idx)], name=f"vmethod_slot_{AttrName}") + FuncPtrI8: ir.Value = Gen.builder.bitcast(Value, ir.PointerType(ir.IntType(8)), name=f"func_ptr_i8_{AttrName}") + Gen._store(FuncPtrI8, MethodPtrAddr) + return + offset: int | None = Gen._get_member_offset(AttrName, ClassName) + StructType: ir.Type | None = Gen.structs.get(ClassName) + if isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(ClassName, Gen) + StructType = Gen.structs.get(ClassName) + if isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0): + return + # 重新计算 offset：_TryLoadStructFromStub 刚加载了 struct body， + # 之前 offset 可能因 struct opaque / class_members 不完整而返回 None + offset = Gen._get_member_offset(AttrName, ClassName) + # sha1 冲突修复：Gen.structs[ClassName] 可能是字段不全的版本（如 2 字段 Function + # 来自 MetaType.py scope，而正确版本是 8 字段 Function 来自 __function.py）。 + # _get_member_offset 通过 class_sha1_map 已返回正确的 offset（如 5）， + # 但 StructType 可能只有 2 个 elements，导致 offset < len(elements) 失败。 + # 在所有 structs 中查找同短名且字段最多的版本。 + if offset is not None and isinstance(StructType, ir.IdentifiedStructType) and StructType.elements is not None and offset >= len(StructType.elements): + _short: str = Gen._extract_short_name(StructType.name) + _best: int = len(StructType.elements) + for _cn, _st in Gen.structs.items(): + if isinstance(_st, ir.IdentifiedStructType) and _st.elements is not None: + if Gen._extract_short_name(_st.name) == _short and len(_st.elements) > _best: + StructType = _st + _best = len(_st.elements) + # bitcast：当 ObjVal 的 pointee 是 opaque 或与 StructType 不是同一个类型对象 + # （sha1 前缀不同但短名相同）时，需要 bitcast 到 StructType 才能正确 GEP + if StructType is not None and StructType.elements is not None and isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and (ObjVal.type.pointee.elements is None or ObjVal.type.pointee != StructType): + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(StructType), name=f"cast_{ClassName}") + if offset is not None and StructType is not None and StructType.elements is not None and offset < len(StructType.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + else: + return + + byte_order: str = Gen.class_member_byteorders.get(ClassName, {}).get(AttrName, "") + + if byte_order == 'big': + if StructType is None or StructType.elements is None or len(StructType.elements) == 0: + return + member_type: ir.Type = StructType.elements[offset] + if isinstance(member_type, ir.IntType): + Value = Gen._apply_bswap_if_big(Value, byte_order, f"bswap_store_{AttrName}") + + self._StoreWithCoerce(Value, MemberPtr) + if not ClassName: + imported_modules: list[str] | None = getattr(self.Trans, '_ImportedModules', None) + resolved_mod: str = self.Trans.SymbolTable.resolve_alias(VarName) + if imported_modules and (VarName in imported_modules or resolved_mod in imported_modules): + PossibleKeys: list[str] = [f"{VarName}.{AttrName}", AttrName] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + key: str = f"{mod_name}.{AttrName}" + if key not in PossibleKeys: + PossibleKeys.append(key) + for key in PossibleKeys: + if key in Gen.module.globals: + GVar: ir.GlobalVariable = Gen.module.globals[key] + if isinstance(GVar, ir.Function): + return + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return + self._StoreWithCoerce(Value, GVar) + return + ModuleSha1Map: dict[str, str] = Gen.ModuleSha1Map + sha1_candidates: list[str] = [VarName, resolved_mod] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + if mod_name not in sha1_candidates: + sha1_candidates.append(mod_name) + for cand in sha1_candidates: + sha1: str | None = ModuleSha1Map.get(cand) + if sha1: + prefixed: str = f"{sha1}.{AttrName}" + if prefixed in Gen.module.globals: + GVar2: ir.GlobalVariable = Gen.module.globals[prefixed] + if isinstance(GVar2, ir.Function): + return + if isinstance(GVar2.type, ir.PointerType) and isinstance(GVar2.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return + self._StoreWithCoerce(Value, GVar2) + return + elif isinstance(Target.value, ast.Attribute): + ObjVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + if ObjVal and isinstance(ObjVal.type, ir.PointerType): + pointee: ir.Type = ObjVal.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + if isinstance(pointee, ir.IdentifiedStructType) and pointee.elements is None: + for CN, ST in Gen.structs.items(): + if isinstance(ST, ir.IdentifiedStructType) and ST.name == pointee.name and ST.elements is not None: + pointee = ST + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{CN}") + break + found_match: bool = False + for CN, ST in Gen.structs.items(): + if pointee == ST or (isinstance(pointee, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType) and pointee.name == ST.name): + found_match = True + if ST.elements is None: + self.Trans.ImportHandler._TryLoadStructFromStub(CN, Gen) + ST = Gen.structs.get(CN, ST) + if ST.elements is not None and isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and ObjVal.type.pointee.elements is None: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{CN}") + bitfield_offsets: dict[str, int] = Gen.class_member_bitoffsets.get(CN, {}) + if AttrName in bitfield_offsets: + self._store_bitfield_member(ObjVal, CN, AttrName, Value) + return + offset: int = Gen._get_member_offset(AttrName, CN) + if offset is not None and ST.elements is not None and offset < len(ST.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + break + if not found_match: + if isinstance(pointee, ir.IdentifiedStructType): + matched_cn: str | None = None + for CN in Gen.class_members: + members: list[tuple[str, ir.Type]] = Gen.class_members[CN] + for mname, mtype in members: + if mname == AttrName: + matched_cn = CN + break + if matched_cn: + break + if matched_cn and matched_cn in Gen.structs: + ST: ir.Type = Gen.structs[matched_cn] + if isinstance(ST, ir.IdentifiedStructType) and (ST.elements is None or len(ST.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(matched_cn, Gen) + ST = Gen.structs.get(matched_cn, ST) + if ST.elements is not None and isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and ObjVal.type.pointee.elements is None: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{matched_cn}") + offset: int = Gen._get_member_offset(AttrName, matched_cn) + if offset is not None and ST.elements is not None and offset < len(ST.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + return + elif isinstance(Target.value, ast.Call): + # Check if this is c.Deref(...) — we need the pointer, not the Loaded value + ObjVal: ir.Value | None = None + is_cderef: bool = (isinstance(Target.value.func, ast.Attribute) and + isinstance(Target.value.func.value, ast.Name) and + Target.value.func.value.id == 'c' and + Target.value.func.attr == 'Deref' and + Target.value.args) + if is_cderef: + # Get the pointer from c.Deref argument, don't Load the value + deref_arg: ast.expr = Target.value.args[0] + inner_val: ir.Value | None = self.HandleExprLlvm(deref_arg) + if inner_val and isinstance(inner_val.type, ir.PointerType): + if isinstance(inner_val.type.pointee, ir.PointerType): + ObjVal = Gen._load(inner_val, name="deref_load_ptr") + else: + ObjVal = inner_val + if ObjVal is None: + ObjVal = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + if ObjVal and isinstance(ObjVal.type, ir.PointerType): + pointee: ir.Type = ObjVal.type.pointee + if isinstance(pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name="load_ptr_for_attr") + pointee = ObjVal.type.pointee + for CN, ST in Gen.structs.items(): + if pointee == ST or (isinstance(pointee, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType) and pointee.name == ST.name): + if isinstance(ST, ir.IdentifiedStructType) and (ST.elements is None or len(ST.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(CN, Gen) + ST = Gen.structs.get(CN, ST) + if ST.elements is not None and isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and ObjVal.type.pointee.elements is None: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{CN}") + bitfield_offsets: dict[str, int] = Gen.class_member_bitoffsets.get(CN, {}) + if AttrName in bitfield_offsets: + self._store_bitfield_member(ObjVal, CN, AttrName, Value) + return + offset: int = Gen._get_member_offset(AttrName, CN) + if offset is not None and ST.elements is not None and offset < len(ST.elements): + MemberPtr: ir.Value = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + break + elif isinstance(Target.value, ast.Subscript): + SubPtr: ir.Value | None = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(Target.value) + if SubPtr and isinstance(SubPtr.type, ir.PointerType): + pointee: ir.Type = SubPtr.type.pointee + + if isinstance(Target.value.value, ast.Attribute): + AttrNameSub: str = Target.value.value.attr + if AttrNameSub == 'items' and 'Item' in Gen.structs: + ObjValCast: ir.Value = Gen.builder.bitcast(SubPtr, ir.PointerType(Gen.structs['Item']), name=f"cast_Item") + offset: int = Gen._get_member_offset(AttrName, 'Item') + MemberPtr: ir.Value = Gen.builder.gep(ObjValCast, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + return + + if BaseHandle._is_char_pointer(SubPtr): + for CN, ST in Gen.structs.items(): + ObjValCast: ir.Value = Gen.builder.bitcast(SubPtr, ir.PointerType(ST), name=f"cast_{CN}") + offset: int = Gen._get_member_offset(AttrName, CN) + MemberPtr: ir.Value = Gen.builder.gep(ObjValCast, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + break + elif isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + if isinstance(ST, ir.IdentifiedStructType) and (ST.elements is None or len(ST.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(CN, Gen) + ST = Gen.structs.get(CN, ST) + if ST.elements is not None and isinstance(SubPtr.type, ir.PointerType) and isinstance(SubPtr.type.pointee, ir.IdentifiedStructType) and SubPtr.type.pointee.elements is None: + SubPtr = Gen.builder.bitcast(SubPtr, ir.PointerType(ST), name=f"cast_{CN}") + offset: int = Gen._get_member_offset(AttrName, CN) + if offset is not None and ST.elements is not None and offset < len(ST.elements): + MemberPtr: ir.Value = Gen.builder.gep(SubPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + elif isinstance(pointee, ir.PointerType): + LoadedPtr: ir.Value = Gen._load(SubPtr, name=f"Load_{AttrName}_ptr") + if isinstance(LoadedPtr.type, ir.PointerType) and isinstance(LoadedPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(LoadedPtr.type.pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + offset: int = Gen._get_member_offset(AttrName, CN) + MemberPtr: ir.Value = Gen.builder.gep(LoadedPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + self._StoreWithCoerce(Value, MemberPtr) + + def _StoreArrElemWithCoerceBswap(self, Value: ir.Value, Ptr: ir.Value, arr_byte_order: str) -> None: + """对数组元素存储应用 bswap 后再 coerce 存储""" + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + Value = Gen._apply_bswap_if_big(Value, arr_byte_order, "bswap_arr_store") + self._StoreWithCoerce(Value, Ptr) + + def _HandleSubscriptStoreLlvm(self, Target: ast.Subscript, Value: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + # 查询数组元素字节序 + arr_byte_order: str = '' + if isinstance(Target.value, ast.Name) and Target.value.id in Gen.local_var_byteorders: + arr_byte_order = Gen.local_var_byteorders[Target.value.id] + if isinstance(Target.value, ast.Name): + SubVarName: str = Target.value.id + if SubVarName in Gen.var_const_flags: + src_info: str = Gen._get_node_info() + self.Trans.LogError(f"不能对 const 变量 '{SubVarName}' 的元素赋值{src_info}") + ClassName: str | None = self.Trans.ExprHandler._get_var_class(Target.value, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__setitem__'): + obj_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.value) + idx_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Target.slice) + if obj_val and idx_val: + # 处理 obj_val 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + if isinstance(obj_val.type, ir.IntType) and obj_val.type.width == 8: + obj_val = Gen.builder.inttoptr(obj_val, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if BaseHandle._is_char_pointer(obj_val): + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + setitem_args: list[ir.Value] = [obj_val, idx_val, Value] + setitem_func_name: str = f'{ClassName}.__setitem__' + setitem_args = Gen._apply_auto_addr(setitem_func_name, setitem_args) + setitem_func: ir.Function = Gen._get_function(setitem_func_name) + setitem_args = Gen._adjust_args(setitem_args, setitem_func) + Gen.builder.call(setitem_func, setitem_args, name=f"call_{setitem_func_name}") + return + ValueVal: ir.Value | None = self.HandleExprLlvm(Target.value) + IndexVal: ir.Value | None = self.HandleExprLlvm(Target.slice) + if not ValueVal or not IndexVal: + return + # 当变量本身是二级指针（如 elem_ptr: T | t.CPtr = AST**）， + # 且存储的值也是指针时，elem_ptr[i] = item 应直接存储指针（8 字节）， + # 而非解引用后复制结构体（如 48 字节的 AST），避免对象切片 + if isinstance(Target.value, ast.Name): + VarName: str = Target.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarAlloca: ir.Value = Gen.variables[VarName] + if isinstance(VarAlloca.type, ir.PointerType) and isinstance(VarAlloca.type.pointee, ir.PointerType): + if (isinstance(ValueVal.type, ir.PointerType) and isinstance(Value.type, ir.PointerType) + and isinstance(Value.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType))): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index_store") + # ValueVal 是 AST*（加载自 elem_ptr 的基指针）， + # 但 GEP 默认将其视为 AST 结构体数组。需要先 bitcast 为 + # AST**（指向 AST* 的指针），让 GEP 正确索引 AST* 指针数组， + # 而非 AST 结构体数组。然后用 builder.store 直接存储指针， + # 避免 _store 的类型强制转换将指针解引用为结构体。 + PtrToStruct: ir.Type = Value.type.pointee + BaseAsPP: ir.Value = Gen.builder.bitcast(ValueVal, + ir.PointerType(ir.PointerType(PtrToStruct)), name="cast_base_pp") + ElemPtr: ir.Value = Gen.builder.gep(BaseAsPP, [IndexVal], name="subscript") + Gen.builder.store(Value, ElemPtr) + return + if BaseHandle._is_char_pointer(IndexVal): + Loaded_byte: ir.Value = Gen._load(IndexVal, name="Load_char_idx_store") + IndexVal = Gen.builder.zext(Loaded_byte, ir.IntType(32), name="char_to_int_idx_store") + if isinstance(ValueVal.type, ir.IntType): + var_ptr: ir.Value | None = self._get_int_ptr(Target.value) + if var_ptr: + i8_ptr: ir.Value = Gen.builder.bitcast(var_ptr, ir.PointerType(ir.IntType(8)), name="int_to_i8ptr_store") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_int_index_store") + ptr: ir.Value = Gen.builder.gep(i8_ptr, [IndexVal], name="int_subscript_ptr_store") + self._StoreWithCoerce(Value, ptr) + return + if isinstance(ValueVal.type, ir.PointerType) and isinstance(ValueVal.type.pointee, ir.PointerType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index_store") + ElemPtr: ir.Value = Gen.builder.gep(ValueVal, [IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + return + if isinstance(ValueVal.type, ir.PointerType): + pointee: ir.Type = ValueVal.type.pointee + if isinstance(pointee, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr: ir.Value = Gen.builder.gep(ValueVal, [zero, IndexVal], name="subscript") + elif (isinstance(pointee, ir.IntType) and pointee.width == 8 + and isinstance(Target.value, ast.Name) + and (Gen.var_ptr_element.get(Target.value.id, False) + or Gen.global_var_ptr_element.get(Target.value.id, False))): + # 指针到 str/bytes 的变量 (BinOp 注解如 str | t.CPtr) + # elem_ptr 是 i8* 但实际指向 i8* (字符串指针), elem_ptr[0] = item 应存储 8 字节 + CastedPtrPtr: ir.Value = Gen.builder.bitcast(ValueVal, ir.PointerType(ir.PointerType(ir.IntType(8))), name="cast_pp_str_store") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_pp_str_index_store") + ElemPtr = Gen.builder.gep(CastedPtrPtr, [IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + return + else: + ElemPtr: ir.Value = Gen.builder.gep(ValueVal, [IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + elif isinstance(ValueVal.type, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + if isinstance(Target.value, ast.Name): + VarName: str = Target.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr: ir.Value = Gen.builder.gep(VarPtr, [zero, IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + return + if isinstance(Target.value, ast.Subscript): + InnerPtr: ir.Value | None = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(Target.value) + if InnerPtr: + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr: ir.Value = Gen.builder.gep(InnerPtr, [zero, IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + return + if isinstance(Target.value, ast.Attribute): + AttrPtr: ir.Value | None = self.Trans.ExprAttrHandle._get_attr_ptr(Target.value) + if AttrPtr and isinstance(AttrPtr.type, ir.PointerType) and isinstance(AttrPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr: ir.Value = Gen.builder.gep(AttrPtr, [zero, IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + return + arr_alloc: ir.AllocaInstr = Gen._allocaEntry(ValueVal.type, name="arr_subscript_tmp") + Gen._store(ValueVal, arr_alloc) + ElemPtr: ir.Value = Gen.builder.gep(arr_alloc, [zero, IndexVal], name="subscript") + self._StoreArrElemWithCoerceBswap(Value, ElemPtr, arr_byte_order) + ModifiedArr: ir.Value = Gen._load(arr_alloc, name="modified_arr") + if isinstance(Target.value, ast.Subscript): + OuterPtr: ir.Value | None = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(Target.value) + if OuterPtr: + Gen._store(ModifiedArr, OuterPtr) + elif isinstance(Target.value, ast.Attribute): + OuterPtr: ir.Value | None = self.Trans.ExprAttrHandle._get_attr_ptr(Target.value) + if OuterPtr: + Gen._store(ModifiedArr, OuterPtr) + + def _StoreWithCoerce(self, Value: ir.Value, Ptr: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if isinstance(Ptr.type, ir.PointerType): + TargetType: ir.Type = Ptr.type.pointee + if Value.type != TargetType: + if isinstance(TargetType, ir.ArrayType) and isinstance(Value.type, ir.PointerType): + ElemType: ir.Type = TargetType.element + ElemSize: int = Gen._get_struct_size(ElemType) + for i in range(TargetType.count): + DstElem: ir.Value = Gen.builder.gep(Ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), i)], name=f"arr_dst_{i}") + SrcPtr: ir.Value + if ElemSize == 1: + SrcPtr = Gen.builder.gep(Value, [ir.Constant(ir.IntType(32), i)], name=f"arr_src_{i}") + else: + PtrInt: ir.Value = Gen.builder.ptrtoint(Value, ir.IntType(64)) + Offset: ir.Constant = ir.Constant(ir.IntType(64), i * ElemSize) + NewPtrInt: ir.Value = Gen.builder.add(PtrInt, Offset) + SrcPtr = Gen.builder.inttoptr(NewPtrInt, ir.PointerType(ElemType), name=f"arr_src_{i}") + SrcVal: ir.Value = Gen._load(SrcPtr, name=f"arr_elem_{i}") + if SrcVal.type != ElemType: + try: + SrcVal = Gen.builder.bitcast(SrcVal, ElemType, name=f"arr_cast_{i}") + except Exception: # 回退：bitcast 失败时跳过该元素 + continue + Gen._store(SrcVal, DstElem) + return + if isinstance(TargetType, ir.PointerType) and isinstance(Value, ir.Constant) and Value.constant is None: + Value = ir.Constant(TargetType, None) + elif isinstance(TargetType, ir.PointerType) and isinstance(Value.type, ir.PointerType): + try: + Value = Gen.builder.bitcast(Value, TargetType, name="null_ptr_bitcast") + except Exception: # 回退：bitcast 失败时设为 null 常量 + Value = ir.Constant(TargetType, None) + else: + Coerced: ir.Value | None = Gen._coerce_value(Value, TargetType) + if Coerced is not None: + Value = Coerced + Gen._store(Value, Ptr) + + + def _get_int_ptr(self, target: ast.expr) -> ir.Value | None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if isinstance(target, ast.Name): + VarName: str = target.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: ir.Value = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType): + pointee: ir.Type = VarPtr.type.pointee + if isinstance(pointee, ir.ArrayType): + return Gen.builder.gep(VarPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"arr_start_{VarName}") + return None + + def _EmitGlobalAnnAssignLlvm(self, Node: ast.AnnAssign, Gen: "Translator.LlvmGen") -> None: + if isinstance(Node.target, ast.Name): + VarName: str = Node.target.id + Info: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(VarName) + if Info: + if Info.IsTypedef: + return + parse_result = ParseListAnnotation(Node.annotation) + if parse_result: + elem_type_node: ast.expr = parse_result.elem_type_node + ElemType, ElemTypeInfo = self.ResolveListElementType( + elem_type_node, Gen, + check_func_ptr=True, void_fallback_width=32, fallback_to_cint=True, + ) + if parse_result.is_pointer: + # 单参数 list[type] → 指针 + VarType: ir.Type = ir.PointerType(ElemType) + if VarName not in Gen.variables and VarName not in Gen.module.globals: + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VarType, name=VarName) + GlobalVar.initializer = ir.Constant(VarType, None) + Gen.variables[VarName] = GlobalVar + if isinstance(elem_type_node, ast.Name) and elem_type_node.id in Gen.structs: + Gen.global_struct_class[VarName] = elem_type_node.id + return + count_node: ast.expr | None = parse_result.count_node + ArrayCount: int = self.ParseArrayCount(count_node, Gen, Node.value, mode='global') + VarType = ir.ArrayType(ElemType, ArrayCount) + if VarName not in Gen.variables and VarName not in Gen.module.globals: + GlobalVar = ir.GlobalVariable(Gen.module, VarType, name=VarName) + if isinstance(ElemType, ir.IdentifiedStructType): + GlobalVar.linkage = 'common' + elif Node.value and isinstance(Node.value, ast.Constant) and isinstance(Node.value.value, str): + str_val: str = Node.value.value + '\x00' + str_bytes: bytearray = bytearray(str_val, 'utf-8') + if len(str_bytes) < ArrayCount: + str_bytes.extend(b'\x00' * (ArrayCount - len(str_bytes))) + try: + GlobalVar.initializer = ir.Constant(VarType, str_bytes[:ArrayCount]) + except Exception: # 回退：初始化常量失败时用零值 + GlobalVar.initializer = Gen._zero_value_for_type(VarType) + else: + InitConstants: list[ir.Constant] | None = self.Trans._BuildArrayInitConstants(Node.value, ElemType, elem_type_node, ArrayCount, Gen, byte_order=ElemTypeInfo.ByteOrder if ElemTypeInfo else '') + if InitConstants: + GlobalVar.initializer = ir.Constant(VarType, InitConstants) + elif ArrayCount > 256: + GlobalVar.linkage = 'common' + else: + if self._contains_identified_struct(VarType): + GlobalVar.linkage = 'common' + else: + GlobalVar.initializer = ir.Constant(VarType, [ir.Constant(ElemType, ir.Undefined)] * ArrayCount) + Gen.variables[VarName] = GlobalVar + if isinstance(elem_type_node, ast.Name) and elem_type_node.id in Gen.structs: + Gen.global_struct_class[VarName] = elem_type_node.id + return + if isinstance(Node.annotation, ast.Name) and Node.annotation.id in Gen.structs: + StructName: str = Node.annotation.id + VarType: ir.Type = Gen.structs[StructName] + if VarName not in Gen.variables and VarName not in Gen.module.globals: + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VarType, name=VarName) + if Node.value and isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name) and Node.value.func.id == StructName: + const: ir.Constant | None = self.Trans.ClassHandler._BuildStructConstant(Node.value, StructName, Gen) + if const: + GlobalVar.initializer = const + else: + GlobalVar.initializer = ir.Constant(VarType, None) + else: + GlobalVar.initializer = ir.Constant(VarType, None) + Gen.variables[VarName] = GlobalVar + Gen.global_struct_class[VarName] = StructName + return + TypeInfo: CTypeInfo | None = CTypeInfo.FromNode(Node.annotation, self.Trans.SymbolTable) + IsCDefine: bool = False + if TypeInfo and TypeInfo.BaseType: + if isinstance(TypeInfo.BaseType, t._CTypedef): + return + if isinstance(TypeInfo.BaseType, t.CDefine): + IsCDefine = True + if not IsCDefine and self._IsTypedefAnnotation(Node.annotation): + IsCDefine = True + TTypeInfo: CTypeInfo | None = self.Trans.TypeMergeHandler.GetCTypeInfo(Node.annotation) + if TTypeInfo: + TTypeInfo.IsTypedef = True + TTypeInfo.Name = VarName + self.Trans.SymbolTable.insert(VarName, TTypeInfo) + if IsCDefine: + if Node.value: + define_constants: dict[str, int | float] = vars(Gen).setdefault('_define_constants', {}) + try: + const_value: int | float | None = self._eval_const_expr(Node.value, Gen) + if const_value is not None: + define_constants[VarName] = const_value + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + return + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + VarType: ir.Type = Gen._ctype_to_llvm(TypeInfo) + if isinstance(VarType, ir.VoidType): + VarType = ir.IntType(32) + if VarName in Gen.variables or VarName in Gen.module.globals: + # 时序修复：第一遍创建全局时类未处理（SymbolTable 无枚举成员）， + # initializer 回退为 0。第二遍类已处理，尝试用解析出的枚举值更新 initializer。 + if VarName in Gen.module.globals and Node.value: + existing: Any = Gen.module.globals[VarName] + if isinstance(existing, ir.GlobalVariable) and isinstance(VarType, ir.IntType): + new_init: ir.Constant | None = self._eval_global_const(Node.value, VarType, Gen) + if new_init is not None: + existing.initializer = new_init + return + if isinstance(VarType, (ir.IdentifiedStructType, ir.LiteralStructType)): + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VarType, name=VarName) + GlobalVar.initializer = ir.Constant(VarType, None) + Gen.variables[VarName] = GlobalVar + if TypeInfo and TypeInfo.IsStruct and TypeInfo.Name: + Gen.global_struct_class[VarName] = TypeInfo.Name + return + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VarType, name=VarName) + # 全局变量指针元素标志：bytes|t.CPtr 或 str|t.CPtr 注解的变量， + # 使 var[index] 使用 8 字节指针步长（而非 1 字节字符步长） + if isinstance(Node.annotation, ast.BinOp) and isinstance(Node.annotation.op, ast.BitOr): + if AnnotationContainsName(Node.annotation, 'str') or AnnotationContainsName(Node.annotation, 'bytes'): + Gen.global_var_ptr_element[VarName] = True + if Node.value: + if isinstance(Node.value, ast.Constant) and isinstance(Node.value.value, str) and isinstance(VarType, ir.PointerType): + str_val: str = Node.value.value + '\x00' + str_bytes: bytes = str_val.encode('utf-8') + arr_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(str_bytes)) + str_gv_name: str = VarName + '_str' + str_gv: ir.GlobalVariable + if str_gv_name not in Gen.module.globals: + str_gv = ir.GlobalVariable(Gen.module, arr_type, name=str_gv_name) + str_gv.initializer = ir.Constant(arr_type, bytearray(str_bytes)) + str_gv.linkage = 'internal' + else: + str_gv = Gen.module.globals[str_gv_name] + ptr_type: ir.PointerType = ir.PointerType(ir.IntType(8)) + ptr_gv: ir.GlobalVariable + if VarName not in Gen.module.globals: + ptr_gv = ir.GlobalVariable(Gen.module, ptr_type, name=VarName) + ptr_gv.initializer = ir.Constant(ptr_type, str_gv) + ptr_gv.linkage = 'internal' + else: + ptr_gv = Gen.module.globals[VarName] + Gen.variables[VarName] = ptr_gv + elif isinstance(VarType, ir.PointerType) and isinstance(VarType.pointee, ir.FunctionType) and isinstance(Node.value, ast.Name): + # 函数指针赋值：f3: t.Callable[[int], int] = f + RefName: str = Node.value.id + SymInfo: Any = self.Trans.SymbolTable.lookup(RefName) + if SymInfo and (SymInfo.IsFunction or SymInfo.IsFuncPtr): + MangledName: str = Gen._mangle_func_name(RefName) + if MangledName in Gen.module.globals: + func: Any = Gen.module.globals[MangledName] + if isinstance(func, ir.Function): + GlobalVar.initializer = func + else: + if isinstance(VarType, ir.ArrayType) and isinstance(Node.value, (ast.List, ast.Set)): + InitConstants: list[ir.Constant] | None = self.Trans._BuildMultiDimArrayInitConstants(Node.value, VarType, Gen) + if InitConstants: + GlobalVar.initializer = ir.Constant(VarType, InitConstants) + else: + if self._contains_identified_struct(VarType): + GlobalVar.linkage = 'common' + else: + zv: ir.Constant | None = Gen._zero_value_for_type(VarType) + if zv: + GlobalVar.initializer = zv + else: + GlobalVar.linkage = 'common' + else: + init_val: ir.Constant | None = self._eval_global_const(Node.value, VarType, Gen) + if init_val: + GlobalVar.initializer = init_val + elif isinstance(VarType, ir.BaseStructType): + GlobalVar.linkage = 'common' + else: + GlobalVar.initializer = Gen._zero_value_for_type(VarType) + else: + if isinstance(VarType, ir.BaseStructType) or self._contains_identified_struct(VarType): + GlobalVar.linkage = 'common' + else: + GlobalVar.initializer = Gen._zero_value_for_type(VarType) + Gen.variables[VarName] = GlobalVar + + def _eval_global_const(self, value_node: ast.expr, var_type: ir.Type, Gen: "Translator.LlvmGen") -> ir.Constant | None: + if isinstance(value_node, ast.Constant): + if isinstance(value_node.value, bool): + if isinstance(var_type, ir.IntType): + return ir.Constant(var_type, int(value_node.value)) + elif isinstance(value_node.value, int): + if isinstance(var_type, ir.IntType): + return ir.Constant(var_type, value_node.value) + elif isinstance(var_type, (ir.FloatType, ir.DoubleType)): + return ir.Constant(var_type, float(value_node.value)) + elif isinstance(value_node.value, float): + if isinstance(var_type, (ir.FloatType, ir.DoubleType)): + return ir.Constant(var_type, value_node.value) + elif isinstance(var_type, ir.IntType): + return ir.Constant(var_type, int(value_node.value)) + elif isinstance(value_node.value, str): + return None + # Handle type constructor calls like t.CDouble(1e-9), t.CFloat(3.14), t.CInt(42) + if isinstance(value_node, ast.Call): + inner_val: ir.Constant | None = self._eval_type_ctor_const(value_node, var_type, Gen) + if inner_val is not None: + return inner_val + if isinstance(value_node, ast.UnaryOp) and isinstance(value_node.op, ast.USub): + inner: ir.Constant | None = self._eval_global_const(value_node.operand, var_type, Gen) + if inner: + return ir.Constant(var_type, -inner.constant) + # Handle enum member references like ASTVType.Module (ast.Attribute) + if isinstance(value_node, ast.Attribute): + attr_name: str = value_node.attr + # 优先用带前缀的 key 查找（避免短名被同名函数/变量覆盖） + if isinstance(value_node.value, ast.Name): + FullKey2: str = f"{value_node.value.id}.{attr_name}" + FullInfo: Any = self.Trans.SymbolTable.lookup(FullKey2) + if FullInfo and FullInfo.IsEnumMember and isinstance(FullInfo.value, int): + if isinstance(var_type, ir.IntType): + return ir.Constant(var_type, FullInfo.value) + SymInfo: Any = self.Trans.SymbolTable.lookup(attr_name) + if SymInfo: + if SymInfo.IsEnumMember and isinstance(SymInfo.value, int): + if isinstance(var_type, ir.IntType): + return ir.Constant(var_type, SymInfo.value) + return None + + def _eval_type_ctor_const(self, call_node: ast.Call, var_type: ir.Type, Gen: "Translator.LlvmGen") -> ir.Constant | None: + """Try to evaluate t.CDouble(val), t.CFloat(val), t.CInt(val) etc. as a compile-time constant.""" + if not isinstance(call_node.func, ast.Attribute): + return None + if not isinstance(call_node.func.value, ast.Name): + return None + if call_node.func.value.id != 't': + return None + if not call_node.args or len(call_node.args) != 1: + return None + arg: ast.expr = call_node.args[0] + # Recursively evaluate the argument as a constant + arg_val: ir.Constant | None = self._eval_global_const(arg, var_type, Gen) + if arg_val is not None: + return arg_val + return None + + def _store_bitfield_member(self, struct_ptr: ir.Value, ClassName: str, field_name: str, Value: ir.Value) -> bool: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + bitfield_offsets: dict[str, int] = Gen.class_member_bitoffsets.get(ClassName, {}) + bitfield_widths: dict[str, int] = Gen.class_member_bitfields.get(ClassName, {}) + if field_name not in bitfield_offsets: + return False + bit_offset: int = bitfield_offsets[field_name] + bit_width: int = bitfield_widths.get(field_name, 0) + if bit_width == 0: + return False + struct_type: ir.Type | None = Gen.structs.get(ClassName) + if struct_type is None or struct_type.elements is None or len(struct_type.elements) == 0: + return False + storage_type: ir.Type = struct_type.elements[0] + member_ptr: ir.Value = Gen.builder.gep(struct_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"{field_name}_storage_ptr") + old_val: ir.Value = Gen._load(member_ptr, name=f"{field_name}_old_storage") + mask: int = (1 << bit_width) - 1 + shifted_mask: int = mask << bit_offset + inverted_mask: ir.Constant = ir.Constant(storage_type, ~shifted_mask & ((1 << storage_type.width) - 1)) + cleared: ir.Value = Gen.builder.and_(old_val, inverted_mask, name=f"{field_name}_cleared") + if isinstance(Value.type, ir.IntType) and Value.type.width < storage_type.width: + if getattr(Value.type, 'is_signed', False): + Value = Gen.builder.sext(Value, storage_type, name=f"{field_name}_sext_store") + else: + Value = Gen.builder.zext(Value, storage_type, name=f"{field_name}_zext_store") + elif Value.type.width != storage_type.width: + Value = Gen.builder.trunc(Value, storage_type, name=f"{field_name}_trunc_store") if Value.type.width > storage_type.width else Gen.builder.zext(Value, storage_type, name=f"{field_name}_zext_store") + shifted_value: ir.Value = Gen.builder.shl(Value, ir.Constant(storage_type, bit_offset), name=f"{field_name}_shift_store") + masked_value: ir.Value = Gen.builder.and_(shifted_value, ir.Constant(storage_type, shifted_mask), name=f"{field_name}_mask_store") + new_val: ir.Value = Gen.builder.or_(cleared, masked_value, name=f"{field_name}_new_storage") + Gen._store(new_val, member_ptr) + return True + + def _CheckEnumMemberAssignment(self, node: ast.expr) -> str | None: + def _get_attr_path(n: ast.expr) -> str | None: + if isinstance(n, ast.Name): + return n.id + elif isinstance(n, ast.Attribute): + return f"{_get_attr_path(n.value)}.{n.attr}" + return None + attr_path: str | None = _get_attr_path(node) + if not attr_path: + return None + parts: list[str] = attr_path.split('.') + if len(parts) >= 2: + enum_class_name: str = parts[-2] + member_name: str = parts[-1] + qualified_name: str = f"{enum_class_name}.{member_name}" + info: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(qualified_name) + if info: + if info.IsEnumMember and info.EnumName: + return info.EnumName + under_name: str = f"{enum_class_name}_{member_name}" + if self.Trans.SymbolTable.has(under_name): + info: "SymbolTable.SymbolInfo" = self.Trans.SymbolTable[under_name] + if info.IsEnumMember and info.EnumName: + return info.EnumName + for key in self.Trans.SymbolTable: + if key == member_name: + info: "SymbolTable.SymbolInfo" = self.Trans.SymbolTable[key] + if info.IsEnumMember and info.EnumName and info.EnumName == enum_class_name: + return enum_class_name + return None + + def _IsTypedefAnnotation(self, annotation: ast.expr) -> bool: + if isinstance(annotation, ast.Attribute): + if annotation.attr in ('CTypedef', 'CDefine'): + return True + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + return self._IsTypedefAnnotation(annotation.left) or self._IsTypedefAnnotation(annotation.right) + if isinstance(annotation, ast.Subscript): + return self._IsTypedefAnnotation(annotation.value) + return False + + def _EmitGlobalAssignLlvm(self, Node: ast.Assign, Gen: "Translator.LlvmGen") -> None: + if len(Node.targets) == 1 and isinstance(Node.targets[0], ast.Name): + VarName: str = Node.targets[0].id + if VarName in Gen.variables or VarName in Gen.module.globals: + return + if isinstance(Node.value, ast.Call) and isinstance(Node.value.func, ast.Name): + FuncName: str = Node.value.func.id + if FuncName in Gen.structs: + IsUnion: bool = False + TypeInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(FuncName) + if TypeInfo: + if TypeInfo.IsUnion: + IsUnion = True + if IsUnion: + StructType: ir.Type = Gen.structs[FuncName] + StructPtrType: ir.PointerType = ir.PointerType(StructType) + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, StructPtrType, name=VarName) + GlobalVar.initializer = ir.Constant(StructPtrType, None) + Gen.variables[VarName] = GlobalVar + Gen.global_struct_class[VarName] = FuncName + return + StructName: str = FuncName + VarType: ir.Type = Gen.structs[StructName] + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, VarType, name=VarName) + const: ir.Constant | None = self.Trans.ClassHandler._BuildStructConstant(Node.value, StructName, Gen) + if const: + GlobalVar.initializer = const + else: + GlobalVar.linkage = 'common' + Gen.variables[VarName] = GlobalVar + Gen.global_struct_class[VarName] = StructName + return + GlobalVar: ir.GlobalVariable = ir.GlobalVariable(Gen.module, ir.IntType(32), name=VarName) + init_val: ir.Constant | None = self._eval_global_const(Node.value, ir.IntType(32), Gen) + GlobalVar.initializer = init_val if init_val else ir.Constant(ir.IntType(32), 0) + Gen.variables[VarName] = GlobalVar + + def _eval_const_expr(self, node: ast.expr, Gen: "Translator.LlvmGen") -> int | float | None: + """计算常量表达式的值（委托到 ConstEvaluator）""" + ctx: EvalContext = EvalContext(Gen=Gen, symtab=self.Trans.SymbolTable, translator=self.Trans) + return ConstEvaluator.eval_full(node, ctx) + + def _eval_global_count(self, node: ast.expr, Gen: "Translator.LlvmGen") -> int | None: + val: int | float | None = self._eval_const_expr(node, Gen) + if val is not None: + if isinstance(val, float): + return int(val) + if isinstance(val, int): + return val + return None \ No newline at end of file diff --git a/lib/core/Handles/HandlesAugAssign.py b/lib/core/Handles/HandlesAugAssign.py new file mode 100644 index 0000000..ea78caf --- /dev/null +++ b/lib/core/Handles/HandlesAugAssign.py @@ -0,0 +1,224 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +import ast +import sys +import llvmlite.ir as ir +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.VLogger import get_logger as _vlog +from lib.core.Handles.HandlesBase import BaseHandle + + +class AugAssignHandle(BaseHandle): + _AUGOP_MAP: dict[type, str] = { + ast.Add: '__iadd__', ast.Sub: '__isub__', ast.Mult: '__imul__', + ast.Div: '__itruediv__', ast.FloorDiv: '__ifloordiv__', ast.Mod: '__imod__', ast.Pow: '__ipow__', + } + _AUGOP_FALLBACK: dict[type, str] = { + ast.Add: '__add__', ast.Sub: '__sub__', ast.Mult: '__mul__', + ast.Div: '__truediv__', ast.FloorDiv: '__floordiv__', ast.Mod: '__mod__', ast.Pow: '__pow__', + } + + def _is_aug_unsigned(self, Node: ast.AugAssign, OldVal: ir.Value | None, Value: ir.Value) -> bool: + """判断增量赋值操作是否应使用无符号指令""" + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if isinstance(Node.target, ast.Name): + if Gen._check_node_unsigned(Node.target): + return True + elif isinstance(Node.target, ast.Attribute): + if Gen._check_node_unsigned(Node.target): + return True + elif isinstance(Node.target, ast.Subscript): + if Gen._check_node_unsigned(Node.target): + return True + if Node.value is not None and Gen._check_node_unsigned(Node.value): + return True + return False + + def _HandleAugAssignLlvm(self, Node: ast.AugAssign) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if isinstance(Node.target, ast.Name): + VarName: str = Node.target.id + if (VarName not in Gen.variables + and VarName not in Gen.global_vars + and VarName not in Gen._reg_values + and VarName not in Gen._direct_values): + src_info: str = Gen._get_node_info() + _vlog().error(f"变量 '{VarName}' 未声明，不能用于增量赋值（需用 global 声明或先声明局部变量）{src_info}") + sys.exit(1) + if VarName in Gen.var_const_flags: + src_info: str = Gen._get_node_info() + _vlog().error(f"不能对 const 变量 '{VarName}' 增量赋值{src_info}") + sys.exit(1) + ClassName: str | None = Gen.var_struct_class.get(VarName) + iop_name: str | None = self._AUGOP_MAP.get(type(Node.op)) + if ClassName and iop_name: + iFuncName: str = f'{ClassName}.{iop_name}' + FuncName: str = f'{ClassName}.{self._AUGOP_FALLBACK.get(type(Node.op), "")}' + if iFuncName in Gen.functions or FuncName in Gen.functions: + OldVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(ast.Name(id=VarName, ctx=ast.Load())) + Value: ir.Value | None = self.HandleExprLlvm(Node.value) + if OldVal and Value: + result: ir.Value | None + if iFuncName in Gen.functions: + result = self.Trans.ExprHandler._try_operator_overLoad(ClassName, iop_name, OldVal, Gen, Value) + else: + result = self.Trans.ExprHandler._try_operator_overLoad(ClassName, self._AUGOP_FALLBACK[type(Node.op)], OldVal, Gen, Value) + if result is not None: + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: ir.Value = Gen.variables[VarName] + if VarPtr.type.pointee == result.type: + Gen._store(result, VarPtr) + elif isinstance(VarPtr.type.pointee, ir.PointerType) and isinstance(result.type, ir.PointerType): + CastedVal: ir.Value = Gen.builder.bitcast(result, VarPtr.type.pointee, name=f"cast_{VarName}") + Gen._store(CastedVal, VarPtr) + else: + NewVar: ir.AllocaInstr = Gen._allocaEntry(result.type, name=VarName) + Gen._store(result, NewVar) + Gen.variables[VarName] = NewVar + else: + Gen._reg_values[VarName] = result + Gen.variables[VarName] = None + return + Value: ir.Value | None = self.HandleExprLlvm(Node.value) + if not Value: + return + is_unsigned: bool = self._is_aug_unsigned(Node, None, Value) + if isinstance(Node.target, ast.Name): + VarName: str = Node.target.id + Op: str = self.Trans.ExprHandler.GetOpSymbol(Node.op) + if VarName in Gen.global_vars and VarName in Gen.module.globals: + GVar: ir.GlobalVariable = Gen.module.globals[VarName] + Gen.variables[VarName] = GVar + OldVal: ir.Value = Gen._load(GVar, name=VarName) + NewVal: ir.Value = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + Gen._store(NewVal, GVar) + return + if VarName in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[VarName] + saved_block: ir.Block = Gen.builder.block + entry_block: ir.Block = Gen.func.entry_basic_block + Gen.builder.position_at_start(entry_block) + var: ir.AllocaInstr = Gen.builder.alloca(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.builder.position_at_end(saved_block) + Gen.variables[VarName] = var + del Gen._reg_values[VarName] + if VarName in Gen._direct_values: + OldVal: ir.Value = Gen._direct_values.pop(VarName) + saved_block: ir.Block = Gen.builder.block + entry_block: ir.Block = Gen.func.entry_basic_block + Gen.builder.position_at_start(entry_block) + var: ir.AllocaInstr = Gen.builder.alloca(OldVal.type, name=VarName) + Gen._store(OldVal, var) + Gen.builder.position_at_end(saved_block) + Gen.variables[VarName] = var + if Gen.builder.block.is_terminated: + return + if VarName in Gen.variables and Gen.variables[VarName] is not None: + OldVal: ir.Value = Gen._load(Gen.variables[VarName], name=VarName) + store_ptr: ir.Value = Gen.variables[VarName] + if isinstance(OldVal.type, ir.PointerType) and isinstance(Value.type, ir.IntType): + if Value.type.width != 64: + Value = Gen.builder.zext(Value, ir.IntType(64), name="zext_ptr_offset") + NewVal: ir.Value + if Op == '-' or Op == '+': + if Op == '-': + NegValue: ir.Value = Gen.builder.neg(Value, name="neg_ptr_offset") + NewVal = Gen.builder.gep(OldVal, [NegValue], name="ptr_sub") + else: + NewVal = Gen.builder.gep(OldVal, [Value], name="ptr_add") + else: + ptr_val: ir.Value = OldVal + OldVal = Gen._load(OldVal, name="deref_ptr_for_op") + orig_type: ir.Type = OldVal.type + if isinstance(OldVal.type, ir.IntType) and isinstance(Value.type, ir.IntType): + if OldVal.type.width < Value.type.width: + OldVal = Gen.builder.zext(OldVal, Value.type, name="zext_left") + elif OldVal.type.width > Value.type.width: + Value = Gen.builder.zext(Value, OldVal.type, name="zext_right") + NewVal = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + if isinstance(NewVal.type, ir.IntType) and isinstance(orig_type, ir.IntType): + if NewVal.type.width > orig_type.width: + NewVal = Gen.builder.trunc(NewVal, orig_type, name="trunc_result") + store_ptr = ptr_val + else: + if isinstance(OldVal.type, ir.IntType) and isinstance(Value.type, ir.IntType): + if OldVal.type.width < Value.type.width: + OldVal = Gen.builder.zext(OldVal, Value.type, name="zext_left") + elif OldVal.type.width > Value.type.width: + Value = Gen.builder.zext(Value, OldVal.type, name="zext_right") + elif isinstance(OldVal.type, ir.PointerType) and isinstance(Value.type, ir.IntType): + if Op in ('+', '-'): + if Value.type.width < 64: + Value = Gen.builder.zext(Value, ir.IntType(64), name="zext_ptr_offset_aug") + NewVal: ir.Value + if Op == '-': + NegValue: ir.Value = Gen.builder.neg(Value, name="neg_ptr_offset_aug") + NewVal = Gen.builder.gep(OldVal, [NegValue], name="ptr_sub_aug") + else: + NewVal = Gen.builder.gep(OldVal, [Value], name="ptr_add_aug") + Gen._store(NewVal, store_ptr) + return + NewVal: ir.Value = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + if isinstance(store_ptr.type, ir.PointerType): + TargetType: ir.Type = store_ptr.type.pointee + if NewVal.type != TargetType: + Coerced: ir.Value | None = Gen._coerce_value(NewVal, TargetType) + if Coerced is not None: + NewVal = Coerced + Gen._store(NewVal, store_ptr) + elif VarName == 'pos': + var: ir.AllocaInstr = Gen._allocaEntry(Value.type, name=VarName) + Gen.variables[VarName] = var + OldVal: ir.Value = Gen._load(var, name=VarName) + NewVal: ir.Value = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + Gen._store(NewVal, var) + elif isinstance(Node.target, ast.Attribute): + AttrName: str = Node.target.attr + OldVal: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Node.target) + if OldVal: + def _deref_if_ptr(val: ir.Value) -> ir.Value: + if isinstance(val.type, ir.PointerType): + pointee: ir.Type = val.type.pointee + if isinstance(pointee, (ir.IntType, ir.FloatType, ir.DoubleType)): + return Gen._load(val, name="deref_aug") + return val + OldVal = _deref_if_ptr(OldVal) + Op: str = self.Trans.ExprHandler.GetOpSymbol(Node.op) + if isinstance(OldVal.type, ir.IntType) and isinstance(Value.type, ir.IntType): + if OldVal.type.width < Value.type.width: + OldVal = Gen.builder.zext(OldVal, Value.type, name="zext_left_attr") + elif OldVal.type.width > Value.type.width: + Value = Gen.builder.zext(Value, OldVal.type, name="zext_right_attr") + NewVal: ir.Value = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + self.Trans.AssignHandler._HandleAttributeStoreLlvm(Node.target, NewVal) + elif isinstance(Node.target, ast.Subscript): + SubTarget: ast.Subscript = Node.target + ElemPtr: ir.Value | None = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(SubTarget) + if ElemPtr: + if isinstance(ElemPtr.type, ir.PointerType): + OldVal: ir.Value = Gen._load(ElemPtr, name="old_subscript_val") + Op: str = self.Trans.ExprHandler.GetOpSymbol(Node.op) + if isinstance(OldVal.type, ir.IntType) and isinstance(Value.type, ir.IntType): + if OldVal.type.width < Value.type.width: + OldVal = Gen.builder.zext(OldVal, Value.type, name="zext_left_sub") + elif OldVal.type.width > Value.type.width: + Value = Gen.builder.zext(Value, OldVal.type, name="zext_right_sub") + elif isinstance(OldVal.type, ir.PointerType) and isinstance(Value.type, ir.IntType): + if Op in ('+', '-'): + if Value.type.width < 64: + Value = Gen.builder.zext(Value, ir.IntType(64), name="zext_ptr_offset_sub") + NewVal: ir.Value + if Op == '-': + NegValue: ir.Value = Gen.builder.neg(Value, name="neg_ptr_offset_sub") + NewVal = Gen.builder.gep(OldVal, [NegValue], name="ptr_sub_sub") + else: + NewVal = Gen.builder.gep(OldVal, [Value], name="ptr_add_sub") + Gen._store(NewVal, ElemPtr) + return + NewVal: ir.Value = Gen.emit_binary_op(Op, OldVal, Value, is_unsigned=is_unsigned) + if isinstance(NewVal.type, ir.IntType) and isinstance(ElemPtr.type.pointee, ir.IntType): + if NewVal.type.width > ElemPtr.type.pointee.width: + NewVal = Gen.builder.trunc(NewVal, ElemPtr.type.pointee, name="trunc_sub_result") + Gen._store(NewVal, ElemPtr) \ No newline at end of file diff --git a/lib/core/Handles/HandlesBase.py b/lib/core/Handles/HandlesBase.py new file mode 100644 index 0000000..35225bb --- /dev/null +++ b/lib/core/Handles/HandlesBase.py @@ -0,0 +1,832 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, List, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator + +import ast +from enum import IntFlag, auto + +from llvmlite import ir as _ir + +from lib.constants import config as _config +from lib.includes import t +from lib.includes.t import CTypeRegistry +from lib.core.TypeSpec import TypeSpec, SymbolMeta, FIELD_ROUTES +from lib.core.VLogger import get_logger as _vlog + + +EXCEPTION_CODE_MAP: dict[str, int] = { + 'ValueError': 1, 'TypeError': 2, 'RuntimeError': 3, + 'ZeroDivisionError': 4, 'IndexError': 5, 'KeyError': 6, + 'IOError': 7, 'OSError': 8, 'AssertionError': 9, 'Exception': 99, +} + + +class FuncMeta(IntFlag): + NONE = 0 + STATIC_METHOD = auto() + PROPERTY_GETTER = auto() + PROPERTY_SETTER = auto() + PROPERTY_DELETER = auto() + CLASS_METHOD = auto() + ABSTRACT = auto() + + +class _Delegated: + """CTypeInfo 旧字段名委托描述符 — 静态委托到 _ts/_sm 的字段 + + 替代旧的 FIELD_ROUTES + __setattr__/__getattr__ 动态路由： + 描述符在类创建时绑定，访问时直接走 __get__/__set__， + 无 dict 查找和字符串比较的运行时开销。 + """ + __slots__ = ('_target_attr', '_field_name') + + def __init__(self, target_attr: str, field_name: str) -> None: + self._target_attr: str = target_attr # '_ts' or '_sm' + self._field_name: str = field_name + + def __get__(self, obj: Any, objtype: type | None = None) -> Any: + if obj is None: + return self + target: Any = object.__getattribute__(obj, self._target_attr) + return getattr(target, self._field_name) + + def __set__(self, obj: Any, value: Any) -> None: + target: Any = object.__getattribute__(obj, self._target_attr) + setattr(target, self._field_name, value) + + +class CTypeInfo: + """ + C 类型信息对象 - 统一符号表条目和类型计算 + + 设计原则: + - BaseType 是 CType 实例（来自 t 模块），永远不使用硬编码字符串 + - 描述符分为两种: + - VarConst/VarVolatile: 变量本身（指针）的限定符，如 int * const + - DataConst/DataVolatile: 指向数据的限定符，如 const int * + + 属性（类型计算）: + - BaseType: t.CType - 基础类型实例 (如 t.CInt(), t.CVoid()) + - PtrCount: int - 指针层数 + - VarConst: bool - 变量本身（指针）是否 const，如 int * const + - VarVolatile: bool - 变量本身（指针）是否 volatile + - DataConst: bool - 指向的数据是否 const，如 const int * + - DataVolatile: bool - 指向的数据是否 volatile + - ArrayDims: List[str] - 数组维度 + - Storage: t.CType - 存储类 (static, extern 等) + - IsFuncPtr: bool - 是否函数指针 + - FuncPtrParams: List[Tuple[str, CTypeInfo]] - 函数指针参数列表，每个元素是 (参数名, 参数类型) + - FuncPtrReturn: CTypeInfo - 函数指针返回类型 + - IsBitField: bool - 是否位域 + - BitWidth: int - 位域宽度 + - IsTypedef/IsStruct/IsEnum/IsUnion: bool - 类型种类 + + 属性（符号表元数据）: + - Name: str - 名称（struct/enum/union/typedef 名） + - Members: Dict[str, CTypeInfo] - 成员字典（用于 struct/union） + - OriginalType: str - typedef 的原始类型 + - DeclaredType: str - 声明类型 + - CreturnTypes: list - C 返回类型列表 + - Lineno: int - 定义行号 + - IsAnonymous: bool - 是否匿名类型 + - IsCpythonObject: bool - 是否 CPython 对象 + - IsEnumMember: bool - 是否枚举成员 + """ + + def __init__(self) -> None: + object.__setattr__(self, '_ts', TypeSpec()) + sm: SymbolMeta = SymbolMeta() + sm.meta_list = FuncMeta.NONE + object.__setattr__(self, '_sm', sm) + + # --- 新公共 API：供新代码直接访问 TypeSpec / SymbolMeta / SymbolKind --- + @property + def ts(self) -> TypeSpec: + """TypeSpec — 纯类型描述（只读）""" + return self._ts + + @property + def sm(self) -> SymbolMeta: + """SymbolMeta — 符号表元数据（只读）""" + return self._sm + + @property + def kind(self) -> 'SymbolKind': + """SymbolKind — 符号类型种类枚举""" + return self._sm.kind + + # 注：旧字段名（PtrCount/IsStruct/Name 等）的委托通过 _Delegated 描述符实现， + # 在模块末尾从 FIELD_ROUTES 静态绑定，无需 __setattr__/__getattr__ 动态路由 + + + + @property + def BaseType(self) -> t.CType | tuple | None: + return self._ts._base_type + + @BaseType.setter + def BaseType(self, value: t.CType | type | tuple | List | None) -> None: + """设置 BaseType，接受以下类型： + + - None: 清空 + - t.CType 实例: 单个类型，自动设置 IsStruct/IsEnum/IsUnion 标志 + - type (CType 子类): 如 t.CStruct，自动实例化 + - tuple/list of t.CType: 多个组合类型 + """ + ts: TypeSpec = self._ts + sm: SymbolMeta = self._sm + + if value is None: + ts._base_type = None + sm.is_struct = False + sm.is_enum = False + sm.is_union = False + sm.is_renum = False + return + + if isinstance(value, (tuple, list)): + ts._base_type = tuple(value) + for v in value: + if isinstance(v, t.CStruct): + sm.is_struct = True + elif isinstance(v, t.CEnum): + sm.is_enum = True + elif isinstance(v, t.CUnion): + sm.is_union = True + elif isinstance(v, t.REnum): + sm.is_renum = True + sm.is_enum = True + return + + if isinstance(value, type) and issubclass(value, t.CType): + ts._base_type = value() + return + + if isinstance(value, t.CType): + ts._base_type = value + if isinstance(value, t.CStruct): + sm.is_struct = True + sm.is_enum = False + sm.is_union = False + sm.is_renum = False + elif isinstance(value, t.REnum): + sm.is_renum = True + sm.is_enum = True + sm.is_struct = False + sm.is_union = False + elif isinstance(value, t.CEnum): + sm.is_enum = True + sm.is_struct = False + sm.is_union = False + sm.is_renum = False + elif isinstance(value, t.CUnion): + sm.is_union = True + sm.is_struct = False + sm.is_enum = False + sm.is_renum = False + return + + raise TypeError(f"BaseType must be t.CType instance, type subclass, or tuple of t.CType, got {type(value)}") + + @property + def TypeCls(self) -> type | None: + """获取 CType 类（从 BaseType 推断）""" + bt: t.CType | tuple | None = self._ts._base_type + if bt: + return type(bt) + sm: SymbolMeta = self._sm + if sm.is_struct: + return t.CStruct + if sm.is_enum: + return t.CEnum + if sm.is_union: + return t.CUnion + if sm.is_typedef: + return t._CTypedef + return None + + @property + def IsPtr(self) -> bool: + return self._ts.ptr_count > 0 + + @property + def IsVoid(self) -> bool: + return isinstance(self._ts._base_type, t.CVoid) + + @property + def IsStr(self) -> bool: + return isinstance(self._ts._base_type, t.CChar) and self._ts.ptr_count > 0 + + @property + def IsInt(self) -> bool: + bt: t.CType | tuple | None = self._ts._base_type + return isinstance(bt, t.CType) and getattr(bt, 'IsSigned', None) is not None and bt.IsSigned is True and not isinstance(bt, t.CVoid) + + @property + def IsUInt(self) -> bool: + bt: t.CType | tuple | None = self._ts._base_type + return isinstance(bt, t.CType) and getattr(bt, 'IsSigned', None) is False + + @property + def IsFloat(self) -> bool: + bt: t.CType | tuple | None = self._ts._base_type + return isinstance(bt, t.CType) and getattr(bt, 'IsSigned', None) is None and getattr(bt, 'Size', 0) in (32, 64, 128) and not isinstance(bt, t.CVoid) + + def ToString(self) -> str: + ts: TypeSpec = self._ts + sm: SymbolMeta = self._sm + bt: t.CType | tuple | None = ts._base_type + + if ts.is_func_ptr: + return 'Callable' + parts: list[str] = [] + + if ts.storage and not isinstance(ts.storage, t.CExport): + parts.append(type(ts.storage).__name__) + + if ts.data_const or ts.data_volatile: + qualifiers: list[str] = [] + if ts.data_const: + qualifiers.append("const") + if ts.data_volatile: + qualifiers.append("volatile") + parts.append("_".join(qualifiers)) + + if sm.is_typedef and sm.name: + return sm.name + elif sm.is_renum and sm.name: + base_name: str = sm.name + elif bt: + if isinstance(bt, str): + base_name = bt + elif isinstance(bt, type) and issubclass(bt, t.CType): + base_name = bt.__name__ + elif isinstance(bt, t.CType): + # 对命名的 CType（CStruct/CEnum/CUnion/REnum），优先使用其 name 属性 + # 否则回退到类名（如 'CInt'、'CChar'） + _named: str | None = getattr(bt, 'name', None) + base_name = _named if _named else type(bt).__name__ + else: + base_name = str(bt) + else: + base_name = "void" + + if ts.is_array_ptr: + ptr_str: str = "" + if ts.ptr_count > 0: + ptr_str = "*" * ts.ptr_count + if ts.array_ptr: + inner: str = ts.array_ptr.ToString() + if ptr_str: + parts.append(f"{base_name} {ptr_str}({inner})") + else: + parts.append(f"{base_name} ({inner})") + else: + if ptr_str: + parts.append(f"{base_name} {ptr_str}()") + else: + parts.append(f"{base_name} ()") + elif ts.array_ptr: + inner = ts.array_ptr.ToString() + if ts.ptr_count > 0: + parts.append(f"{base_name} {'*' * ts.ptr_count}({inner})") + else: + parts.append(f"{base_name} ({inner})") + else: + parts.append(base_name) + if ts.ptr_count > 0: + parts.append("*" * ts.ptr_count) + if ts.var_const: + parts.append("const") + if ts.var_volatile: + parts.append("volatile") + + for dim in ts.array_dims: + if dim: + parts.append(f"[{dim}]") + else: + parts.append("[]") + + return " ".join(parts) if parts else "void" + + @staticmethod + def CreateFromTypeName(TypeName: str) -> "CTypeInfo": + """从类型名字符串创建 CTypeInfo（兼容旧 API）""" + info: CTypeInfo = CTypeInfo() + if TypeName.startswith('struct '): + info.BaseType = t.CStruct(name=TypeName[7:].strip()) if TypeName[7:].strip() else t.CStruct() + elif TypeName.startswith('enum '): + info.BaseType = t.CEnum(name=TypeName[5:].strip()) if TypeName[5:].strip() else t.CEnum() + elif TypeName.startswith('union '): + info.BaseType = t.CUnion(name=TypeName[6:].strip()) if TypeName[6:].strip() else t.CUnion() + else: + info.BaseType = t.CStruct(name=TypeName) + return info + + # LLVM primitive type name → (CTypeClass, PtrCount) mapping + # Used when generic type inference produces LLVM type names like 'i64', 'double', etc. + _LLVM_PRIMITIVE_MAP: dict[str, tuple[type, int]] | None = None + + @classmethod + def _get_llvm_primitive_map(cls) -> dict[str, tuple[type, int]]: + if cls._LLVM_PRIMITIVE_MAP is not None: + return cls._LLVM_PRIMITIVE_MAP + cls._LLVM_PRIMITIVE_MAP = { + 'void': (t.CVoid, 0), + 'i1': (t.CInt, 0), + 'i8': (t.CChar, 0), + 'i16': (t.CShort, 0), + 'i32': (t.CInt, 0), + 'i64': (t.CLong, 0), + 'i128': (t.CInt, 0), + 'float': (t.CFloat, 0), + 'double': (t.CDouble, 0), + 'fp128': (t.CFloat128T, 0), + } + return cls._LLVM_PRIMITIVE_MAP + + @staticmethod + def FromTypeName(TypeName: str) -> "CTypeInfo": + """从简单类型名构造 CTypeInfo（不经过 FromStr 字符串解析）""" + from lib.core.TypeAnnotationResolver import TypeAnnotationResolver + return TypeAnnotationResolver.from_type_name(TypeName) + + # FromStr 已移除 - 类型解析不再经过中间字符串格式 + # typedef 展开直接走 CTypeInfo 对象，不经过字符串解析 + + @staticmethod + def TryEvalConstExpr(node: ast.AST, SymbolTable: Any) -> Any: + from lib.core.TypeAnnotationResolver import TypeAnnotationResolver + return TypeAnnotationResolver.try_eval_const_expr(node, SymbolTable) + + @classmethod + def FromNode(cls, Node: ast.AST, SymbolTable: Any) -> "CTypeInfo": + """从 AST 节点解析 CTypeInfo（类方法） + + Args: + Node: AST 节点（如 ast.Name, ast.Attribute 等） + SymbolTable: 符号表字典 + """ + from lib.core.TypeAnnotationResolver import TypeAnnotationResolver + return TypeAnnotationResolver.from_node(Node, SymbolTable) + + + def Copy(self) -> "CTypeInfo": + NewInfo: CTypeInfo = CTypeInfo() + NewInfo._ts = self._ts.copy() + NewInfo._sm = self._sm.copy() + # CTypeInfo 引用字段需要深拷贝 + if isinstance(NewInfo._ts.func_ptr_return, CTypeInfo): + NewInfo._ts.func_ptr_return = NewInfo._ts.func_ptr_return.Copy() + if isinstance(NewInfo._ts.original_type, CTypeInfo): + NewInfo._ts.original_type = NewInfo._ts.original_type.Copy() + if NewInfo._ts.array_ptr: + NewInfo._ts.array_ptr = NewInfo._ts.array_ptr.Copy() + return NewInfo + + def get(self, key: str, default: Any = None) -> Any: + """获取额外属性""" + return self._sm._extra.get(key, default) + + def set(self, key: str, value: Any) -> None: + """设置额外属性""" + self._sm._extra[key] = value + + @staticmethod + def VoidTypeInfo() -> "CTypeInfo": + info: CTypeInfo = CTypeInfo() + info.BaseType = t.CVoid() + return info + + def ToLLVM(self, Gen: LlvmCodeGenerator) -> _ir.Type: + if Gen is None: + return _ir.IntType(32) + return Gen._ctype_to_llvm(self) + + def __str__(self) -> str: + return self.ToString() + + def __repr__(self) -> str: + ts: TypeSpec = self._ts + sm: SymbolMeta = self._sm + return f"CTypeInfo(BaseType={ts._base_type}, PtrCount={ts.ptr_count}, IsTypedef={sm.is_typedef}, OriginalType={ts.original_type!r}, VarConst={ts.var_const}, DataConst={ts.data_const})" + + +# 从 FIELD_ROUTES 静态绑定 _Delegated 描述符到 CTypeInfo 类 +# 这是一次性配置（类创建期执行），替代运行时 __setattr__/__getattr__ 动态路由 +for _old_name, (_target, _field) in FIELD_ROUTES.items(): + setattr(CTypeInfo, _old_name, _Delegated('_' + _target, _field)) +del _old_name, _target, _field + + +class CTypeHelper: + + @staticmethod + def GetTModuleCType(TypeName: str) -> type | None: + result: type | None = CTypeRegistry.GetClassByName(TypeName) + if result is not None: + return result + TypeClass: type | None = getattr(t, TypeName, None) + if TypeClass and isinstance(TypeClass, type) and issubclass(TypeClass, t.CType): + return TypeClass + FallbackClass: type | None = getattr(t, f'_{TypeName}', None) + if FallbackClass and isinstance(FallbackClass, type) and issubclass(FallbackClass, t.CType): + return FallbackClass + return None + + @staticmethod + def GetCName(type_or_name: str | type) -> str: + if isinstance(type_or_name, str): + cls: type | None = CTypeRegistry.GetClassByName(type_or_name) + if cls is None: + cls = CTypeHelper.GetTModuleCType(type_or_name) + if cls: + return cls.__name__ + return type_or_name + elif isinstance(type_or_name, type) and issubclass(type_or_name, t.CType): + return type_or_name.__name__ + return '' + + +class BuiltinTypeMap: + """ + 内置类型映射表 - 字符串到 (CType类, PtrCount) 的映射 + + 用法：BuiltinTypeMap.Get('int') -> (t.CInt, 0) + BuiltinTypeMap.Get('BYTEPTR') -> (t.CUnsignedChar, 1) + """ + + _map: dict[str, tuple[type, int]] | None = None + + @classmethod + def _build_map(cls) -> dict[str, tuple[type, int]]: + if cls._map is not None: + return cls._map + CTypeRegistry._build() + cls._map = {} + # 仅使用 Python 类名 → (CType 类, ptr_level) 映射。 + # C 类型名（如 'int'、'int32_t'）不再通过 _cname_to_class 提供。 + # C 头文件 stub 生成器使用独立的硬编码表（CTypeMapper._CNAME_TO_PY）。 + for pyname, ctype_cls in CTypeRegistry._name_to_class.items(): + pos: frozenset = getattr(ctype_cls, 'position', frozenset()) + if t.CType.POINTER in pos and t.CType.BASE not in pos: + cls._map[pyname] = (ctype_cls, 1) + else: + cls._map[pyname] = (ctype_cls, 0) + # 保留一些常用的字符串速记（这些不是 CName，而是硬编码的便捷别名）。 + # 包含 Python 内置类型名（int/float/bool）和 C 类型名速记（char/short/long 等）。 + _SPECIAL: dict[str, tuple[type, int]] = { + # Python 内置类型名 + 'str': (t.CChar, 1), 'bytes': (t.CChar, 1), + 'int': (t.CInt, 0), 'float': (t.CFloat, 0), 'bool': (t.CBool, 0), + # C 类型名速记 + 'char': (t.CChar, 0), 'short': (t.CShort, 0), + 'long': (t.CLong, 0), 'long long': (t.CLongLong, 0), + 'double': (t.CDouble, 0), + 'unsigned': (t.CUnsigned, 0), + 'signed': (t.CInt, 0), 'signed int': (t.CInt, 0), + 'signed char': (t.CSignedChar, 0), + 'void': (t.CVoid, 0), 'Void': (t.CVoid, 0), + } + for k, v in _SPECIAL.items(): + if k not in cls._map: + cls._map[k] = v + _FLOAT_ALIASES: dict[str, tuple[type, int]] = { + 'FLOAT8': (t.CFloat8T, 0), 'FLOAT16': (t.CFloat16T, 0), + 'FLOAT32': (t.CFloat32T, 0), 'FLOAT64': (t.CFloat64T, 0), + 'FLOAT128': (t.CFloat128T, 0), + } + for k, v in _FLOAT_ALIASES.items(): + if k not in cls._map: + cls._map[k] = v + return cls._map + + @classmethod + def Get(cls, type_name: str) -> tuple[type, int] | None: + """获取类型类和指针层级""" + return cls._build_map().get(type_name) + + +class BaseHandle: + def __init__(self, translator: "Translator") -> None: + self.Trans: Translator = translator + self._CurrentCpythonObjectClass: str | None = None + + @staticmethod + def _is_char_pointer(val: _ir.Value) -> bool: + """检查值是否为 char* (i8*) 指针""" + if not isinstance(val.type, _ir.PointerType): + return False + pointee: _ir.Type = val.type.pointee + return isinstance(pointee, _ir.IntType) and pointee.width == 8 + + def HandleExprLlvm(self, Node: ast.AST, VarType: _ir.Type | str | None = None) -> _ir.Value | None: + return self.Trans.ExprHandler.HandleExprLlvm(Node, VarType) + + def HandleBodyLlvm(self, Body: ast.AST | list[ast.stmt]) -> Any: + return self.Trans.BodyHandler.HandleBodyLlvm(Body) + + def _get_int_ptr(self, Node: ast.AST) -> _ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(Node, ast.Name): + if Node.id in Gen.variables: + return Gen.variables[Node.id] + elif isinstance(Node, ast.Attribute): + obj_val: Any = self.HandleExprLlvm(Node.value) + if not obj_val: + return None + if isinstance(obj_val.type, _ir.PointerType): + pointee: _ir.Type = obj_val.type.pointee + if isinstance(pointee, (_ir.IdentifiedStructType, _ir.LiteralStructType)): + StructName: str | None = None + if isinstance(pointee, _ir.IdentifiedStructType): + StructName = pointee.name + if not StructName: + for ClassName, struct_type in Gen.structs.items(): + if struct_type == pointee: + StructName = ClassName + break + if StructName and StructName in Gen.structs: + offset: int = self.Trans.ExprHandler._get_llvm_member_offset(Node.attr, StructName, Gen) + member_ptr: Any = Gen.builder.gep(obj_val, [_ir.Constant(_ir.IntType(32), 0), _ir.Constant(_ir.IntType(32), offset)], name=f"{Node.attr}_ptr") + return member_ptr + return None + + def _IsTModuleType(self, TypeName: str) -> bool: + if TypeName in ('t', 'State', 'Bit', 'Anonymous', 'Postdefinition'): + return True + TypeClass: type | None = getattr(t, TypeName, None) + if TypeClass and isinstance(TypeClass, type): + if issubclass(TypeClass, t.CType): + return True + FallbackClass: type | None = getattr(t, f'_{TypeName}', None) + if FallbackClass and isinstance(FallbackClass, type): + if issubclass(FallbackClass, t.CType): + return True + return False + + def ResolveListElementType( + self, + elem_type_node: ast.expr, + Gen: Any, + *, + check_func_ptr: bool = False, + handle_str: bool = False, + void_fallback_width: int = 32, + fallback_to_cint: bool = False, + ) -> tuple[Any, CTypeInfo | None]: + """解析 list 元素类型，返回 (LLVM类型, CTypeInfo|None)。 + + 统一处理结构体、str、函数指针、VoidType 回退等变体。 + + Args: + check_func_ptr: 是否检查函数指针（全局变量路径需要） + handle_str: 是否处理 str → i8*（局部变量路径需要） + void_fallback_width: VoidType 回退宽度 (32 或 8) + fallback_to_cint: 无类型信息时回退到 CInt + """ + elem_type_info: CTypeInfo | None = CTypeInfo.FromNode(elem_type_node, self.Trans.SymbolTable) + + # 结构体类型 + if isinstance(elem_type_node, ast.Name) and elem_type_node.id in Gen.structs: + if check_func_ptr: + sym_entry: CTypeInfo | None = self.Trans.SymbolTable.lookup(elem_type_node.id) + if sym_entry and isinstance(sym_entry, CTypeInfo) and sym_entry.IsFuncPtr: + return _ir.IntType(8).as_pointer(), elem_type_info + return Gen.structs[elem_type_node.id], elem_type_info + + # str → i8* + if handle_str and isinstance(elem_type_node, ast.Name) and elem_type_node.id == 'str': + return _ir.PointerType(_ir.IntType(8)), elem_type_info + + # 函数指针检查 + if check_func_ptr and elem_type_info and elem_type_info.IsFuncPtr: + return _ir.IntType(8).as_pointer(), elem_type_info + + # 回退到 CInt + if elem_type_info is None and fallback_to_cint: + elem_type_info = CTypeInfo() + elem_type_info.BaseType = t.CInt() + + elem_type: Any = Gen._ctype_to_llvm(elem_type_info) + if isinstance(elem_type, _ir.VoidType): + elem_type = _ir.IntType(void_fallback_width) + + return elem_type, elem_type_info + + def ParseArrayCount( + self, + count_node: ast.expr | None, + Gen: Any, + value_node: ast.expr | None = None, + *, + mode: str = 'global', + ) -> int: + """解析数组计数，返回计数（默认 1）。 + + 统一处理 Constant/Name/Attribute/BinOp/None 等形式。 + None 或 Constant(None) 时从 value_node 推断（List 长度或 str 长度+1）。 + + Args: + count_node: 计数 AST 节点 + value_node: 赋值右侧值节点（用于 count_node 为 None 时推断） + mode: 'global' 使用 _eval_global_count + SymbolTable + _define_constants； + 'local' 使用 TryEvalConstExpr + """ + # None 或 Constant(None) → 从 value_node 推断 + if count_node is None or (isinstance(count_node, ast.Constant) and count_node.value is None): + if value_node and isinstance(value_node, ast.List): + return len(value_node.elts) + if value_node and isinstance(value_node, ast.Constant) and isinstance(value_node.value, str): + return len(value_node.value) + 1 + return 1 + + # Constant(int) + if isinstance(count_node, ast.Constant) and isinstance(count_node.value, int): + return count_node.value + + if mode == 'global': + # Name → SymbolTable + _define_constants + if isinstance(count_node, ast.Name): + ArrayCount: int = 1 + if self.Trans.SymbolTable.has(count_node.id): + sym_info: Any = self.Trans.SymbolTable[count_node.id] + if isinstance(sym_info.value, int): + ArrayCount = sym_info.value + if ArrayCount == 1 and count_node.id in Gen._define_constants: + def_val: Any = Gen._define_constants[count_node.id] + if isinstance(def_val, int): + ArrayCount = def_val + return ArrayCount + # Attribute / BinOp → _eval_global_count + if isinstance(count_node, (ast.Attribute, ast.BinOp)): + eval_fn: Any = getattr(self, '_eval_global_count', None) + if eval_fn is not None: + ev: int | None = eval_fn(count_node, Gen) + if ev is not None: + return ev + return 1 + else: + # local mode → TryEvalConstExpr + eval_count: Any = CTypeInfo.TryEvalConstExpr(count_node, self.Trans.SymbolTable) + if eval_count is not None and isinstance(eval_count, int) and eval_count > 0: + return eval_count + return 1 + + return 1 + + def ResolveAnnotationType(self, annotation: ast.expr) -> CTypeInfo | None: + """解析注解类型，BinOp(BitOr) 优先使用 MergeTypes，其他使用 FromNode。 + + 统一处理函数返回类型和参数类型的 BinOp 联合类型解析。 + """ + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + merger: Any = getattr(self.Trans, 'TypeMergeHandler', None) + if merger: + info: CTypeInfo | None = merger.MergeTypes(annotation) + if info is not None and isinstance(info, CTypeInfo): + return info + return CTypeInfo.FromNode(annotation, self.Trans.SymbolTable) + + def CollectAnnAssignMember( + self, item: ast.AST, Gen: Any + ) -> tuple[str, _ir.Type, CTypeInfo | None] | None: + """从 ast.AnnAssign 节点解析成员信息。 + + 返回 (VarName, MemberType, TypeInfo)；TypeInfo 为 None 表示解析回退到 i32。 + 非 AnnAssign/Name 目标时返回 None。 + """ + if not isinstance(item, ast.AnnAssign) or not isinstance(item.target, ast.Name): + return None + VarName: str = item.target.id + try: + TypeInfo: CTypeInfo | None = self.ResolveAnnotationType(item.annotation) + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + MemberType: _ir.Type = Gen._ctype_to_llvm(TypeInfo) + if isinstance(MemberType, _ir.VoidType): + MemberType = _ir.PointerType(_ir.IntType(8)) + return VarName, MemberType, TypeInfo + except Exception as e: + _vlog().warning(f"HandlesBase: 忽略异常 {e}", exc_info=e) + return VarName, _ir.IntType(32), None + + def LookupFunctionSymbol(self, name: str, *, module_path: str | None = None) -> CTypeInfo | None: + """查找函数符号，先尝试 module_path.name，再回退到 name。 + + 统一处理 HandlesExprCall / HandlesAssign 中重复的二级查找模式。 + 返回 CTypeInfo（不限定 IsFunction），由调用方检查标志位。 + """ + st = self.Trans.SymbolTable + if module_path: + sym: CTypeInfo | None = st.lookup(f'{module_path}.{name}') + if sym: + return sym + return st.lookup(name) + + def BuildLLVMFuncTypeFromSig( + self, + sym_info: CTypeInfo, + Gen: Any, + *, + mangled_name: str | None = None, + use_infer_fallback: bool = False, + ) -> tuple[Any, list[Any], bool] | None: + """从 CTypeInfo 函数签名构建 LLVM 函数类型。 + + 返回 (ret_type, llvm_param_types, is_variadic)，非函数返回 None。 + 统一处理 FuncPtrReturn/FuncPtrParams → ToLLVM 转换、VoidType 回退。 + + Args: + mangled_name: 用于 _infer_return_type + stub 查找回退 + use_infer_fallback: 为 True 时启用 _infer_return_type + stub 查找回退链 + """ + if not sym_info.IsFunction: + return None + is_variadic: bool = sym_info.IsVariadic + ret_type_info = sym_info.FuncPtrReturn + if isinstance(ret_type_info, CTypeInfo) and ret_type_info.BaseType: + ret_type: Any = ret_type_info.ToLLVM(Gen) + elif use_infer_fallback and mangled_name: + inferred: Any = Gen._infer_return_type(mangled_name) + if isinstance(inferred, _ir.IntType) and inferred.width == 32: + try: + stub_ft: Any = self.Trans.ImportHandler._LookupStubFuncType(mangled_name, Gen) + if stub_ft and hasattr(stub_ft, 'return_type'): + inferred = stub_ft.return_type + except Exception: + pass + ret_type = inferred + else: + ret_type = Gen._CType2LLVM('i32', False) + param_type_infos = [pt for _, pt in (sym_info.FuncPtrParams or [])] + if isinstance(ret_type, _ir.VoidType): + ret_type = _ir.IntType(32) + llvm_param_types: list[Any] = [] + for pt in param_type_infos: + if isinstance(pt, CTypeInfo) and pt.BaseType: + lp: Any = pt.ToLLVM(Gen) + else: + lp = Gen._CType2LLVM(pt if isinstance(pt, str) else 'i32', '*' in pt if isinstance(pt, str) else False) + if isinstance(lp, _ir.VoidType): + lp = _ir.IntType(8).as_pointer() + llvm_param_types.append(lp) + return ret_type, llvm_param_types, is_variadic + + def GetOrCreateFuncDecl( + self, + name: str, + ret_type: Any, + param_types: list[Any], + Gen: Any, + *, + is_variadic: bool = False, + replace_name: str | None = None, + ) -> Any: + """创建或替换 LLVM 函数声明。 + + 若 name 已存在且类型不匹配，通过 _replace_function_decl 替换。 + replace_name 传递给 _replace_function_decl（通常为未 mangle 的 func_name）。 + """ + if name not in Gen.functions: + func_type = _ir.FunctionType(ret_type, param_types, var_arg=is_variadic) + func_decl = _ir.Function(Gen.module, func_type, name=name) + Gen.functions[name] = func_decl + else: + existing = Gen.functions[name] + existing_type = getattr(existing, 'ftype', None) + new_type = _ir.FunctionType(ret_type, param_types, var_arg=is_variadic) + if existing_type and existing_type != new_type: + replaced = self.Trans.FunctionHandler._replace_function_decl(Gen, replace_name or name, new_type) + if replaced and replaced != existing: + Gen.functions[name] = replaced + return Gen.functions[name] + + def GetOrCreateStubFuncDecl( + self, + decl_name: str, + stub_func_type: Any, + Gen: Any, + *, + alias: str | None = None, + check_existing: bool = True, + ) -> Any: + """创建 stub 函数声明，可选注册别名。 + + check_existing=False 时跳过存在性检查（用于确定不存在的场景）。 + """ + if check_existing and decl_name in Gen.functions: + func_decl = Gen.functions[decl_name] + else: + func_decl = _ir.Function(Gen.module, stub_func_type, name=decl_name) + Gen.functions[decl_name] = func_decl + if alias and alias != decl_name: + if not check_existing or alias not in Gen.functions: + Gen.functions[alias] = func_decl + return func_decl + + +StrictMode: bool = _config.mode == 'strict' \ No newline at end of file diff --git a/lib/core/Handles/HandlesBody.py b/lib/core/Handles/HandlesBody.py new file mode 100644 index 0000000..86b54d8 --- /dev/null +++ b/lib/core/Handles/HandlesBody.py @@ -0,0 +1,77 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class BodyHandle(BaseHandle): + def __init__(self, translator: "Translator") -> None: + super().__init__(translator) + + def HandleBodyLlvm(self, Body: list[ast.stmt]) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + for Node in Body: + if Gen.builder and Gen.builder.block.is_terminated: + break + Gen._set_node_info(Node) + try: + if isinstance(Node, ast.Expr): + self.HandleExprLlvm(Node.value) + elif isinstance(Node, ast.Return): + self.Trans.ReturnHandler._HandleReturnLlvm(Node) + elif isinstance(Node, ast.Assign): + self.Trans.AssignHandler._HandleAssignLlvm(Node) + elif isinstance(Node, ast.AugAssign): + self.Trans.AugAssignHandler._HandleAugAssignLlvm(Node) + elif isinstance(Node, ast.AnnAssign): + self.Trans.AnnAssignHandler._HandleAnnAssignLlvm(Node) + elif isinstance(Node, ast.If): + self.Trans.IfHandler._HandleIfLlvm(Node) + elif isinstance(Node, ast.For): + self.Trans.ForHandler._HandleForLlvm(Node) + elif isinstance(Node, ast.While): + self.Trans.WhileHandler._HandleWhileLlvm(Node) + elif isinstance(Node, ast.Break): + if Gen.loop_break_targets: + Gen.builder.branch(Gen.loop_break_targets[-1]) + elif isinstance(Node, ast.Continue): + if Gen.loop_continue_targets: + Gen.builder.branch(Gen.loop_continue_targets[-1]) + elif isinstance(Node, ast.With): + self.Trans.WithHandler._HandleWithLlvm(Node) + elif isinstance(Node, ast.ClassDef): + pass + elif getattr(ast, 'Match', None) and isinstance(Node, ast.Match): + self.Trans.MatchHandler._HandleMatchLlvm(Node) + elif isinstance(Node, ast.Try): + self.Trans.TryHandler._HandleTryLlvm(Node) + elif isinstance(Node, ast.Raise): + self.Trans.RaiseHandler._HandleRaiseLlvm(Node) + elif isinstance(Node, ast.Assert): + self.Trans.AssertHandler._HandleAssertLlvm(Node) + elif isinstance(Node, ast.Global): + self.Trans.ForHandler._HandleGlobalLlvm(Node) + elif isinstance(Node, ast.FunctionDef): + self.Trans.ForHandler._HandleNestedFunctionLlvm(Node) + elif isinstance(Node, ast.Nonlocal): + self.Trans.ForHandler._HandleNonlocalLlvm(Node) + elif isinstance(Node, ast.Delete): + self.Trans.DeleteHandler._HandleDeleteLlvm(Node) + except Exception as e: + lineno: int = getattr(Node, 'lineno', 0) + source_line: str | None = Gen._get_source_line(lineno) + src_file: str = Gen._current_source_file or '' + src_path: str = src_file if src_file else 'unknown' + line_info: str = "源代码 (%s, line %d)" % (src_path, lineno) + if source_line: + line_info += ":\n %d | %s" % (lineno, source_line) + self.Trans._ErrorStack.append((type(e).__name__, str(e), line_info)) + raise + if Gen.builder and not Gen.builder.block.is_terminated: + Gen._EmitTempFrees() + + def HandleExprLlvm(self, Node: ast.expr, VarType: ir.Type | None = None) -> ir.Value | None: + return self.Trans.ExprHandler.HandleExprLlvm(Node, VarType) \ No newline at end of file diff --git a/lib/core/Handles/HandlesClassDef.py b/lib/core/Handles/HandlesClassDef.py new file mode 100644 index 0000000..5d58f21 --- /dev/null +++ b/lib/core/Handles/HandlesClassDef.py @@ -0,0 +1,1259 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +import ast +import copy +import llvmlite.ir as ir +from lib.core.Exportable import EnumMember as ExportEnumMember +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo, FuncMeta +from lib.core.SymbolUtils import IsTModule, FindStructNameInAnnotation +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.includes.t import CTypeRegistry + + +class ClassHandle(BaseHandle): + def _is_exception_class(self, Node: ast.ClassDef) -> bool: + if not Node.bases: + return False + for base in Node.bases: + if hasattr(base, 'id'): + if base.id == 'Exception': + return True + if base.id in self.Trans.exception_registry: + return True + elif hasattr(base, 'attr'): + if base.attr == 'Exception': + return True + if base.attr in self.Trans.exception_registry: + return True + return False + + def _get_exception_parent(self, Node: ast.ClassDef) -> str | None: + if not Node.bases: + return None + for base in Node.bases: + if hasattr(base, 'id'): + if base.id != 'Exception' and base.id in self.Trans.exception_registry: + return base.id + elif hasattr(base, 'attr'): + if base.attr != 'Exception' and base.attr in self.Trans.exception_registry: + return base.attr + return None + + def _RegisterExceptionClass(self, Node: ast.ClassDef) -> None: + ClassName: str = Node.name + if ClassName in self.Trans.exception_registry: + return + code: int = self.Trans._next_exception_code + self.Trans._next_exception_code += 1 + self.Trans.exception_registry[ClassName] = code + parent: str | None = self._get_exception_parent(Node) + if parent: + self.Trans.exception_parents[ClassName] = parent + ExcTypeInfo: CTypeInfo = CTypeInfo() + ExcTypeInfo.Name = ClassName + ExcTypeInfo.IsExceptionClass = True + ExcTypeInfo.value = code + self.Trans.SymbolTable.insert(ClassName, ExcTypeInfo) + + def _is_generic_class(self, Node: ast.ClassDef) -> bool: + if hasattr(Node, 'type_params') and Node.type_params: + return True + return False + + def _mangle_generic_class_name(self, class_name: str, type_args: list[str]) -> str: + mangled_args: list[str] = [] + for ta in type_args: + llvm_str: str | None = CTypeRegistry.NameToLLVM(ta) + if llvm_str: + if llvm_str in ('float', 'double', 'half', 'fp128'): + mangled: str = 'f' + str(CTypeRegistry.GetClassByName(ta)().Size) + else: + mangled = llvm_str + else: + mangled = ta + mangled_args.append(mangled) + return class_name + '[' + ']['.join(mangled_args) + ']' + + def _specialize_generic_class(self, ClassName: str, type_args: list[str], Gen: LlvmGeneratorMixin, type_names: list[str] | None = None, declare_only: bool = False) -> str | None: + if not hasattr(self, '_generic_class_templates') or ClassName not in self._generic_class_templates: + return None + # 防御：共享符号表构建阶段 Gen 可能为 None，此时无法发射特化 IR + if Gen is None: + return None + template: dict = self._generic_class_templates[ClassName] + Node: ast.ClassDef = template['node'] + type_param_names: list[str] = template['type_params'] + if len(type_args) != len(type_param_names): + return None + spec_key: str = ClassName + '<' + ','.join(type_args) + '>' + # 缓存必须在 Gen 上（每个模块独立），不能在 ClassHandler 上跨模块共享， + # 否则临时 Gen（如共享符号表构建阶段）的特化会污染正式翻译阶段的缓存 + if not hasattr(Gen, '_generic_class_specializations'): + Gen._generic_class_specializations = {} + if spec_key in Gen._generic_class_specializations: + cached: object = Gen._generic_class_specializations[spec_key] + # 缓存格式: (spec_name, was_declare_only) + if isinstance(cached, tuple): + cached_name: str = cached[0] + cached_declare_only: bool = cached[1] + # 完整特化请求但缓存只有 declare_only 版本 → 需要补发方法体 + if not declare_only and cached_declare_only: + spec_name = cached_name + # 标记为正在完整特化，防止递归 + Gen._generic_class_specializations[spec_key] = (spec_name, False) + else: + return cached_name + else: + # 旧格式（纯字符串）：直接返回 + return cached + else: + spec_name: str = self._mangle_generic_class_name(ClassName, type_args) + # 预注册到缓存，防止特化过程中方法体引用自身类型导致无限递归 + Gen._generic_class_specializations[spec_key] = (spec_name, declare_only) + type_map: dict[str, str] = {} + for i, tp_name in enumerate(type_param_names): + if type_names and i < len(type_names): + type_map[tp_name] = type_names[i] + else: + type_map[tp_name] = type_args[i] + # 内建类型名集合（必须在 cptr_params 检测之前定义） + _BUILTIN_TYPE_NAMES: set = { + 'CInt', 'CUInt', 'CInt8T', 'CUInt8T', 'CInt16T', 'CUInt16T', + 'CInt32T', 'CUInt32T', 'CInt64T', 'CUInt64T', 'CChar', 'CShort', + 'CLong', 'CDouble', 'CFloat', 'CFloat32T', 'CFloat64T', 'CBool', + 'CVoid', 'CSizeT', 'CSSizeT', 'CPtr', 'CArray', 'CEnum', 'CUnion', + 'CStruct', 'CDefine', + 'int', 'double', 'float', 'i8', 'i16', 'i32', 'i64', 'void', + 'half', 'fp128', 'char', 'bool', + } + # 检测哪些类型参数在原始类型实参中带有 | t.CPtr + # 当 type_arg 以 'ptr' 结尾（_resolve_type_arg_str 将 t.CPtr 规范化为 'ptr'） + # 且对应的 type_name 不是内建类型名时，说明调用方写了 list[AST | t.CPtr] + cptr_params: set[str] = set() + for i, tp_name in enumerate(type_param_names): + type_arg: str = type_args[i] + if type_arg.endswith('ptr'): + tn: str = type_names[i] if type_names and i < len(type_names) else type_args[i] + if tn not in _BUILTIN_TYPE_NAMES: + cptr_params.add(tp_name) + # Bug 2 修复：识别特化为 class 类型的类型参数 + # 这类参数在方法签名中是裸 T（无 | t.CPtr），特化后为裸 ClassName（值类型） + # 需要加 | t.CPtr 使其成为指针类型（class 实例应按指针传递） + class_type_names: set[str] = set() + for i, tp_name in enumerate(type_param_names): + tn: str = type_names[i] if type_names and i < len(type_names) else type_args[i] + if tn not in _BUILTIN_TYPE_NAMES: + class_type_names.add(tn) + # 仅对 t 模块内建类型名注册 _t_c_imported_names（如 CInt -> ('t','CInt')） + # 类名（如 AST）不应注册为 t.AST，否则会污染名称解析 + if type_names: + if not hasattr(self.Trans, '_t_c_imported_names'): + self.Trans._t_c_imported_names = {} + for tn in type_names: + if tn in _BUILTIN_TYPE_NAMES and tn not in self.Trans._t_c_imported_names: + self.Trans._t_c_imported_names[tn] = ('t', tn) + SpecNode: ast.ClassDef = copy.deepcopy(Node) + SpecNode.type_params = [] + SpecNode.name = spec_name + for item in SpecNode.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + if item.annotation: + item.annotation = self.Trans.FunctionHandler._replace_type_in_annotation(item.annotation, type_map, cptr_params) + elif isinstance(item, ast.FunctionDef): + for arg in item.args.args: + if arg.annotation: + arg.annotation = self.Trans.FunctionHandler._replace_type_in_annotation(arg.annotation, type_map, cptr_params) + # class 类型参数加指针语义：T -> GNode -> GNode | t.CPtr + # 去除 SHA1 前缀后检查（如 fc06023a9bcfc93a.AST -> AST） + if isinstance(arg.annotation, ast.Name): + raw_name: str = arg.annotation.id + if '.' in raw_name: + raw_name = raw_name.split('.')[-1] + if raw_name in class_type_names: + arg.annotation = ast.BinOp( + left=ast.Name(id=arg.annotation.id, ctx=ast.Load()), + op=ast.BitOr(), + right=ast.Attribute(value=ast.Name(id='t', ctx=ast.Load()), attr='CPtr', ctx=ast.Load()) + ) + if item.returns: + item.returns = self.Trans.FunctionHandler._replace_type_in_annotation(item.returns, type_map, cptr_params) + # 返回类型也加指针语义 + if isinstance(item.returns, ast.Name): + raw_name: str = item.returns.id + if '.' in raw_name: + raw_name = raw_name.split('.')[-1] + if raw_name in class_type_names: + item.returns = ast.BinOp( + left=ast.Name(id=item.returns.id, ctx=ast.Load()), + op=ast.BitOr(), + right=ast.Attribute(value=ast.Name(id='t', ctx=ast.Load()), attr='CPtr', ctx=ast.Load()) + ) + self.Trans.FunctionHandler._apply_type_map_to_body(item.body, type_map, cptr_params) + saved_builder: ir.IRBuilder = Gen.builder + saved_func: ir.Function = Gen.func + saved_variables: dict = dict(Gen.variables) if Gen.variables else {} + saved_direct_values: dict = dict(Gen._direct_values) if Gen._direct_values else {} + saved_var_type_info: dict = dict(Gen.var_type_info) if Gen.var_type_info else {} + saved_var_signedness: dict = dict(Gen.var_signedness) if Gen.var_signedness else {} + saved_global_vars: set = set(Gen.global_vars) if Gen.global_vars else set() + saved_var_scopes: list = [dict(s) for s in self.Trans.VarScopes] if self.Trans.VarScopes else [] + saved_block: ir.Block | None = None + if Gen.builder and Gen.builder.block and not Gen.builder.block.is_terminated: + saved_block = Gen.builder.block + try: + self._EmitClassLlvm(SpecNode, Gen, declare_only=declare_only) + finally: + # 确保 Gen 状态恢复，即使 _EmitClassLlvm 抛出异常 + Gen.builder = saved_builder + if saved_block is not None and Gen.builder is not None: + Gen.builder.position_at_end(saved_block) + Gen.func = saved_func + Gen.variables = saved_variables + Gen._direct_values = saved_direct_values + Gen.var_type_info = saved_var_type_info + Gen.var_signedness = saved_var_signedness + Gen.global_vars = saved_global_vars + self.Trans.VarScopes = saved_var_scopes + Gen._generic_class_specializations[spec_key] = (spec_name, declare_only) + if not hasattr(self.Trans, '_generic_class_specializations'): + self.Trans._generic_class_specializations = {} + self.Trans._generic_class_specializations[spec_key] = spec_name + if hasattr(self.Trans, '_module_sha1') and self.Trans._module_sha1: + Gen._struct_sha1_map[spec_name] = self.Trans._module_sha1 + return spec_name + + def _resolve_base_kind(self, base_name: str, Gen: LlvmGeneratorMixin) -> str | None: + """解析基类名称的类型种类 + + 查询顺序： + 1. _t_type_symbols（from t import CEnum as en） + 2. t 模块属性（t.CEnum / t.CUnion / t.CStruct / t.REnum） + 3. 返回 None 表示不是已知标记基类 + + Returns: + 'enum' | 'union' | 'renum' | 'struct' | None + """ + # 1) 查 _t_type_symbols（别名导入） + symtab: object = self.Trans.SymbolTable + cls: type | None = symtab._t_type_symbols.get(base_name) + if cls is not None: + if issubclass(cls, (t.CEnum, t.REnum)): + return 'renum' if issubclass(cls, t.REnum) else 'enum' + if issubclass(cls, t.CUnion): + return 'union' + if issubclass(cls, t.CStruct): + return 'struct' + + # 2) 查 t 模块属性（直接引用 t.CEnum 等） + t_cls: type | None = getattr(t, base_name, None) + if isinstance(t_cls, type) and issubclass(t_cls, t.CType): + if issubclass(t_cls, (t.CEnum, t.REnum)): + return 'renum' if issubclass(t_cls, t.REnum) else 'enum' + if issubclass(t_cls, t.CUnion): + return 'union' + if issubclass(t_cls, t.CStruct): + return 'struct' + + return None + + def _resolve_decorator_kind(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin) -> str | None: + """从装饰器列表中解析结构体类型种类 + + 支持 @t.CStruct, @t.CUnion, @t.CEnum, @t.REnum + 也支持 from t import CStruct 等别名形式 + + Returns: + 'enum' | 'union' | 'renum' | 'struct' | None + """ + if not hasattr(Node, 'decorator_list') or not Node.decorator_list: + return None + symtab: object = self.Trans.SymbolTable + for decorator in Node.decorator_list: + attr_name: str | None = None + # @t.CStruct 形式 + if isinstance(decorator, ast.Attribute) and isinstance(decorator.value, ast.Name): + if IsTModule(decorator.value.id, symtab): + attr_name = decorator.attr + # @CStruct 形式 (from t import CStruct) + elif isinstance(decorator, ast.Name): + cls: type | None = getattr(t, decorator.id, None) + if cls is None and hasattr(symtab, '_t_type_symbols'): + cls = symtab._t_type_symbols.get(decorator.id) + if cls is not None and isinstance(cls, type) and issubclass(cls, t.CType): + attr_name = cls.__name__ + if attr_name: + # 查 _t_type_symbols 和 t 模块属性以确定种类 + resolved_cls: type | None = None + if hasattr(symtab, '_t_type_symbols'): + resolved_cls = symtab._t_type_symbols.get(attr_name) + if resolved_cls is None: + resolved_cls = getattr(t, attr_name, None) + if resolved_cls is not None and isinstance(resolved_cls, type): + if issubclass(resolved_cls, t.REnum): + return 'renum' + if issubclass(resolved_cls, t.CEnum): + return 'enum' + if issubclass(resolved_cls, t.CUnion): + return 'union' + if issubclass(resolved_cls, t.CStruct): + return 'struct' + return None + + def _resolve_class_flags(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin) -> tuple[bool, bool, str | None, bool]: + """解析类的 Object/CVTable 标志和父类信息 + + Returns: + tuple: (IsCpythonObject, IsCVTable, ParentClassName, IsNoVTable) + """ + ClassName: str = Node.name + IsCpythonObject: bool = False + IsCVTable: bool = False + IsNoVTable: bool = False + if hasattr(Node, 'decorator_list') and Node.decorator_list: + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Attribute): + if hasattr(decorator.value, 'id') and IsTModule(decorator.value.id, self.Trans.SymbolTable): + if decorator.attr == 'Object': + IsCpythonObject = True + elif decorator.attr == 'CVTable': + IsCVTable = True + elif decorator.attr == 'NoVTable': + IsNoVTable = True + elif isinstance(decorator, ast.Name): + if decorator.id == 'Object': + IsCpythonObject = True + elif decorator.id == 'CVTable': + IsCVTable = True + elif decorator.id == 'NoVTable': + IsNoVTable = True + elif isinstance(decorator, ast.Call): + # 检测 @c.Attribute(t.attr.packed) + if isinstance(decorator.func, ast.Attribute): + if getattr(decorator.func.value, 'id', None) == 'c' and decorator.func.attr == 'Attribute': + for arg in decorator.args: + if isinstance(arg, ast.Attribute): + if isinstance(arg.value, ast.Attribute): + if IsTModule(getattr(arg.value.value, 'id', ''), self.Trans.SymbolTable) and arg.value.attr == 'attr' and arg.attr == 'packed': + Gen.class_packed.add(ClassName) + class_graph: dict = Gen._class_graph + if class_graph.get(ClassName, {}).get('has_methods', False) and not IsNoVTable: + if not IsCpythonObject: + IsCpythonObject = True + # 仅当该类作为其他类的基类时，才启用 CVTable + # 检查 _class_graph 中是否有任何类将此类作为基类 + is_base_of_any: bool = False + for other_cls, info in class_graph.items(): + if ClassName in info.get('bases', []): + is_base_of_any = True + break + if is_base_of_any: + IsCVTable = True + HasParentClass: bool = False + ParentClassName: str | None = None + ParentIsNoVTable: bool = False + if Node.bases: + for base in Node.bases: + base_name: str | None = None + if hasattr(base, 'id'): + base_name = base.id + elif hasattr(base, 'attr'): + base_name = base.attr + if base_name: + kind: str | None = self._resolve_base_kind(base_name, Gen) + if kind is not None or base_name in t._MARKER_BASES: + continue # 标记基类，跳过 + if base_name in class_graph or base_name in Gen.class_members or base_name in Gen.structs: + HasParentClass = True + ParentClassName = base_name + # 检测父类是否有 @t.NoVTable 标记（非多态继承） + if class_graph.get(base_name, {}).get('has_novtable', False): + ParentIsNoVTable = True + # 跨模块父类：检查 _cross_module_novtable 集合 + if base_name in Gen._cross_module_novtable: + ParentIsNoVTable = True + break + if HasParentClass and not IsCVTable and not IsNoVTable and not ParentIsNoVTable: + IsCVTable = True + if IsCVTable: + Gen.class_vtable.add(ClassName) + # 沿继承链向上传播，将所有祖先类加入 vtable + visited: set[str] = {ClassName} + ancestor: str | None = ParentClassName + while ancestor and ancestor not in visited: + visited.add(ancestor) + Gen.class_vtable.add(ancestor) + # 查询祖先类的父类 + ancestor_info: dict | None = class_graph.get(ancestor) + if ancestor_info: + ancestor_bases: list = ancestor_info.get('bases', []) + next_ancestor: str | None = None + for ab in ancestor_bases: + if ab not in t._MARKER_BASES and ab not in visited: + next_ancestor = ab + break + ancestor = next_ancestor + else: + # 类图中没有，尝试从 class_parent 查询 + ancestor = Gen.class_parent.get(ancestor) + return IsCpythonObject, IsCVTable, ParentClassName, IsNoVTable + + def _EmitClassLlvm(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin, declare_only: bool = False) -> None: + ClassName: str = Node.name + if self._is_generic_class(Node): + if not hasattr(self, '_generic_class_templates'): + self._generic_class_templates = {} + type_params: list[str] = [tp.name for tp in Node.type_params] + self._generic_class_templates[ClassName] = { + 'node': Node, + 'type_params': type_params, + } + return + if self._is_exception_class(Node): + self._RegisterExceptionClass(Node) + return + IsCenum: bool = False + IsCunion: bool = False + IsRenum: bool = False + # 先检查装饰器形式（@t.CStruct / @t.CEnum / @t.CUnion / @t.REnum） + deco_kind: str | None = self._resolve_decorator_kind(Node, Gen) + if deco_kind == 'enum': + IsCenum = True + elif deco_kind == 'union': + IsCunion = True + elif deco_kind == 'renum': + IsRenum = True + # 再检查基类继承形式（class X(t.CStruct):），与装饰器合并 + if not IsCenum and not IsCunion and not IsRenum and Node.bases: + for base in Node.bases: + base_name: str | None = None + if hasattr(base, 'attr'): + base_name = base.attr + elif hasattr(base, 'id'): + base_name = base.id + if base_name: + kind: str | None = self._resolve_base_kind(base_name, Gen) + if kind == 'enum': + IsCenum = True + break + elif kind == 'union': + IsCunion = True + elif kind == 'renum': + IsRenum = True + if IsCenum: + self._RegisterEnumMembers(Node) + return + if IsCunion: + self._EmitUnionLlvm(Node, Gen) + return + if IsRenum: + self._EmitREnumLlvm(Node, Gen) + return + IsCpythonObject: bool + IsCVTable: bool + ParentClassName: str | None + IsNoVTable: bool + IsCpythonObject, IsCVTable, ParentClassName, IsNoVTable = self._resolve_class_flags(Node, Gen) + # 跟踪正在发射的类，防止递归特化时 _generate_structs 过早为空成员类设置 body + Gen._classes_in_progress.add(ClassName) + # 标记正在收集字段，阻止继承处理（_ResolveGenericSlice → _specialize_generic_class） + # 和 CollectAnnAssignMember 触发的嵌套 _generate_structs 用不完整 class_members 调用 set_body + Gen._collecting_members_set.add(ClassName) + if ClassName not in Gen.class_methods: + Gen.class_methods[ClassName] = [] + else: + # 防御性清除：移除导入阶段可能添加的构造函数（__new__/__init__/__before_init__）。 + # 构造函数不参与虚分派，放入 vtable 会导致子类与基类 vtable 大小不一致， + # 索引偏移破坏多态分派（如 Module 有 __new__/__init__ 使 vtable=7，而 AST 无它们 vtable=5， + # 调用方按 AST 布局取 append@idx4 实际取到 Module.dump@idx4）。 + Gen.class_methods[ClassName] = [ + m for m in Gen.class_methods[ClassName] + if m.split('.')[-1] not in ('__new__', '__init__', '__before_init__') + ] + # 当前文件定义的类是权威定义，覆盖之前导入阶段可能添加的外部声明。 + # 关键：如果本地类是 @t.NoVTable 但同名导入类有 vtable（名称冲突）， + # 必须清除 vtable 状态，否则 _generate_structs 会错误添加 vtable 指针。 + if IsNoVTable: + Gen.class_vtable.discard(ClassName) + Gen._cross_module_vtable_classes.discard(ClassName) + Gen._cross_module_novtable.add(ClassName) + # 清除可能从导入 stub 加载的旧 struct body（含错误 vtable 指针）， + # 让 _generate_structs 用本地 class_members 重新生成 + _old_st = Gen.structs.get(ClassName) + if isinstance(_old_st, ir.IdentifiedStructType) and _old_st.elements is not None: + _old_st._elements = None + _ih = getattr(Gen, '_import_handler_ref', None) + if _ih is not None: + _ck = (id(Gen), ClassName) + if _ck in _ih._struct_Load_cache: + del _ih._struct_Load_cache[_ck] + Gen.class_members[ClassName] = [] + Gen.class_member_defaults[ClassName] = {} + Gen.class_member_signeds[ClassName] = {} + Gen.class_member_bitfields[ClassName] = {} + Gen.class_member_byteorders[ClassName] = {} + Gen.class_member_bitoffsets[ClassName] = {} + ParentClass: str | None = None + if Node.bases: + class_graph: dict = Gen._class_graph + for base in Node.bases: + if hasattr(base, 'id'): + base_name = base.id + elif hasattr(base, 'attr'): + base_name = base.attr + elif isinstance(base, ast.Subscript): + # 泛型特化基类：GSListNode[GNode] + # 特化泛型类，用特化名作为父类名 + spec_name: str | None = self.Trans.ExprCallHandle._ResolveGenericSlice(base) + if spec_name: + base_name = spec_name + else: + continue + else: + continue + if base_name == ClassName: + continue + if base_name in ('Object', 'CVTable', 'NoVTable', 'CStruct', 'CUnion', 'CEnum', 'REnum', 'Exception', 'Enum'): + continue + # 总是记录父类名，即使父类 class_members 未加载。 + # _get_member_offset 会沿 class_parent 链回退查找继承字段。 + ParentClass = base_name + Gen.class_parent[ClassName] = ParentClass + break + if ParentClass and (ParentClass in class_graph or ParentClass in Gen.class_members): + parent_members: list = list(Gen.class_members[ParentClass]) + parent_defaults: dict = dict(Gen.class_member_defaults.get(ParentClass, {})) + existing_names: set = {m[0] for m in Gen.class_members[ClassName]} + inherited: list[tuple] = [] + for pm_name, pm_type in parent_members: + if pm_name not in existing_names: + inherited.append((pm_name, pm_type)) + if pm_name in parent_defaults: + Gen.class_member_defaults[ClassName][pm_name] = parent_defaults[pm_name] + Gen.class_members[ClassName] = inherited + Gen.class_members[ClassName] + if ParentClass in Gen.class_methods: + parent_methods: list = list(Gen.class_methods[ParentClass]) + for pm in parent_methods: + method_name: str = pm.split('.')[-1] if '.' in pm else pm + child_method: str = f"{ClassName}.{method_name}" + if child_method not in Gen.class_methods[ClassName]: + Gen.class_methods[ClassName].append(child_method) + Gen.register_method(ClassName, child_method) + # 提取首个裸字符串字面量作为结构体 __doc__ docstring + if (Node.body and isinstance(Node.body[0], ast.Expr) + and isinstance(Node.body[0].value, ast.Constant) + and isinstance(Node.body[0].value.value, str)): + Gen._create_doc_global(ClassName, Node.body[0].value.value) + else: + Gen._create_doc_global(ClassName, None) + # 解析 __provides__/__requires__/__require_must__ 编译期元数据 + for item in Node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + field_name: str = item.target.id + if field_name in ('__provides__', '__requires__', '__require_must__') and item.value: + values: list[str] = [] + if isinstance(item.value, ast.List): + for elt in item.value.elts: + if isinstance(elt, ast.Constant) and isinstance(elt.value, str): + values.append(elt.value) + if field_name == '__provides__': + Gen.class_provides[ClassName] = values + elif field_name == '__requires__': + Gen.class_requires[ClassName] = values + elif field_name == '__require_must__': + Gen.class_require_must[ClassName] = values + for item in Node.body: + # 跳过编译期元数据字段（__provides__/__requires__/__require_must__） + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + if item.target.id in ('__provides__', '__requires__', '__require_must__'): + continue + parsed = self.CollectAnnAssignMember(item, Gen) + if parsed: + VarName, MemberType, TypeInfo = parsed + Gen.class_members[ClassName].append((VarName, MemberType)) + if TypeInfo: + if isinstance(MemberType, ir.PointerType) and isinstance(MemberType.pointee, ir.IntType) and MemberType.pointee.width == 8: + if isinstance(item.annotation, ast.BinOp) and isinstance(item.annotation.op, ast.BitOr): + element_class: str | None = FindStructNameInAnnotation(item.annotation, Gen.structs) + # Fallback: FindStructNameInAnnotation 不处理 Subscript slice 为 BinOp 的情况 + # (如 list[AST | t.CPtr] | t.CPtr)，使用 TypeInfo.Name 作为回退。 + # TypeInfo.Name 由 _HandleSubscript 设置为特化名（如 'list[sha1.ASTptr]'）， + # 经 _MergeCTypeInfoInto 传播后保留。 + if not element_class and TypeInfo.IsStruct and TypeInfo.Name and TypeInfo.Name in Gen.structs: + element_class = TypeInfo.Name + if element_class: + if ClassName not in Gen.class_member_element_class: + Gen.class_member_element_class[ClassName] = {} + Gen.class_member_element_class[ClassName][VarName] = element_class + if hasattr(TypeInfo.BaseType, 'IsSigned'): + Gen.class_member_signeds[ClassName][VarName] = TypeInfo.BaseType.IsSigned + else: + Gen.class_member_signeds[ClassName][VarName] = None + if TypeInfo.IsBitField: + Gen.class_member_bitfields[ClassName][VarName] = TypeInfo.BitWidth + else: + Gen.class_member_bitfields[ClassName][VarName] = 0 + if TypeInfo.ByteOrder: + Gen.class_member_byteorders[ClassName][VarName] = TypeInfo.ByteOrder + else: + Gen.class_member_byteorders[ClassName][VarName] = "" + if item.value: + const: ir.Constant | None = self._BuildScalarConstant(item.value, MemberType, Gen) + if const: + Gen.class_member_defaults[ClassName][VarName] = const + else: + Gen.class_member_signeds[ClassName][VarName] = None + Gen.class_member_bitfields[ClassName][VarName] = 0 + # 字段收集完成，清除标志，允许后续 _generate_structs 正常设置 body + Gen._collecting_members_set.discard(ClassName) + existing_members: set[str] = {m[0] for m in Gen.class_members[ClassName]} + for item in Node.body: + if isinstance(item, ast.FunctionDef): + self._current_method_args = {} + for arg in item.args.args: + if arg.arg != 'self' and arg.annotation: + self._current_method_args[arg.arg] = arg.annotation + self._scan_method_body_for_self_members(item.body, ClassName, existing_members, Gen) + self._current_method_args = {} + for item in Node.body: + if isinstance(item, ast.FunctionDef): + MethodName: str = item.name + FullMethodName: str = f"{ClassName}.{MethodName}" + # property setter/deleter 使用不同的函数名后缀 + item_meta: FuncMeta = self.Trans.FunctionHandler._ExtractFuncMeta(item.decorator_list) + if FuncMeta.PROPERTY_SETTER in item_meta: + FullMethodName = FullMethodName + '$set' + elif FuncMeta.PROPERTY_DELETER in item_meta: + FullMethodName = FullMethodName + '$del' + # 构造函数（__new__/__init__）不放入 vtable，避免子类与基类 + # vtable 大小不一致导致虚分派索引偏移。__before_init__ 由 TPC + # 自动生成，不会出现在 FunctionDef 中，此处一并防御。 + if MethodName not in ('__new__', '__init__', '__before_init__'): + if FullMethodName not in Gen.class_methods[ClassName]: + Gen.class_methods[ClassName].append(FullMethodName) + Gen.register_method(ClassName, FullMethodName) + if IsCVTable: + Gen.class_vtable.add(ClassName) + Gen._generate_structs() + Gen._create_Vtable_globals() + # _create_Vtable_globals 会跳过正在发射中的类（方法列表可能不完整）， + # 当前类的方法已全部注册（L559-571），此处显式创建其 vtable + if IsCVTable: + Gen._create_vtable_for_class(ClassName) + if ClassName in Gen.structs: + self._EmitNewFunctionLlvm(ClassName, Gen, IsCVTable=IsCVTable, IsCpythonObject=IsCpythonObject) + for item in Node.body: + if isinstance(item, ast.FunctionDef): + self.Trans.FunctionHandler._EmitFunctionForwardDeclLlvm(item, Gen, ClassName=ClassName) + if not declare_only: + for item in Node.body: + if isinstance(item, ast.FunctionDef): + self.Trans.FunctionHandler._EmitFunctionLlvm(item, Gen, ClassName=ClassName) + if IsCVTable and ClassName in Gen.Vtables: + self._fill_vtable_with_methods(ClassName, Gen) + # Generate wrapper functions for inherited methods that are not overridden + if ParentClass: + self._generate_inherited_method_wrappers(ClassName, Gen) + Gen._classes_in_progress.discard(ClassName) + + def _generate_inherited_method_wrappers(self, ClassName: str, Gen: LlvmGeneratorMixin) -> None: + """为子类继承但未覆写的方法生成包装函数，使跨模块调用能正确链接。 + + 例如 Label 继承 Widget.place，生成 Label.place 函数， + 内部将 self 从 Label* bitcast 为 Widget*，然后调用 Widget.place。 + """ + methods: list = Gen.class_methods.get(ClassName, []) + if not methods: + return + if ClassName not in Gen.structs: + return + + for method_name in methods: + method_short: str = method_name.split('.')[-1] if '.' in method_name else method_name + child_full: str = f"{ClassName}.{method_short}" + + # 如果子类已经有该方法的实现，跳过 + if Gen._find_function(child_full): + continue + + # 沿继承链查找父类实现 + parent_func: ir.Function | None = None + parent_class: str | None = None + p: str | None = Gen.class_parent.get(ClassName) + while p: + parent_full: str = f"{p}.{method_short}" + parent_func = Gen._find_function(parent_full) + if parent_func: + parent_class = p + break + p = Gen.class_parent.get(p) + + if not parent_func or not parent_class: + continue + + # 父类和子类都需要有 struct 定义 + if parent_class not in Gen.structs: + continue + + # 构造子类包装函数签名：与父类函数相同，但 self 参数类型为子类指针 + parent_ftype: ir.FunctionType = parent_func.function_type + parent_ret_type: ir.Type = parent_ftype.return_type + parent_param_types: list[ir.Type] = list(parent_ftype.args) + is_vararg: bool = parent_ftype.var_arg + + # 判断父类方法是否是静态方法： + # 如果第一个参数类型是父类指针，说明是实例方法；否则是静态方法 + parent_struct_ptr_type: ir.PointerType | None = ir.PointerType(Gen.structs[parent_class]) if parent_class in Gen.structs else None + is_static: bool = True + if parent_param_types and parent_struct_ptr_type: + first_param: ir.Type = parent_param_types[0] + # 检查第一个参数是否是父类指针（或可以 bitcast 为父类指针的指针类型） + if isinstance(first_param, ir.PointerType): + if isinstance(first_param.pointee, ir.IdentifiedStructType): + if first_param.pointee.name == Gen.structs[parent_class].name: + is_static = False + elif first_param == ir.IntType(8): + # i8* 可能是前向声明时的 self 类型，视为实例方法 + is_static = False + + if is_static: + # 静态方法：签名与父类完全一致，不需要 self 参数 + child_param_types: list[ir.Type] = list(parent_param_types) + child_struct_ptr: ir.PointerType | None = None + else: + # 实例方法：替换第一个参数（self）的类型为子类指针 + child_struct_ptr = ir.PointerType(Gen.structs[ClassName]) + if parent_param_types: + child_param_types = [child_struct_ptr] + parent_param_types[1:] + else: + child_param_types = [child_struct_ptr] + + child_func_type: ir.FunctionType = ir.FunctionType(parent_ret_type, child_param_types, var_arg=is_vararg) + child_mangled: str = Gen._mangle_func_name(child_full) + child_func: ir.Function = Gen._get_or_declare_function(child_mangled, child_func_type) + Gen.functions[child_full] = child_func + + # 生成函数体 + entry_block: ir.Block = child_func.append_basic_block(name="entry") + saved_builder: ir.IRBuilder = Gen.builder + saved_func: ir.Function = Gen.func + Gen.builder = ir.IRBuilder(entry_block) + Gen.func = child_func + + # 准备参数 + call_args: list[ir.Value] = [] + if is_static: + # 静态方法：参数直接传递，不做 bitcast + for arg in child_func.args: + call_args.append(arg) + else: + # 实例方法：bitcast self，其余参数直接传递 + for i, arg in enumerate(child_func.args): + if i == 0 and parent_param_types: + actual_self_type: ir.Type = parent_param_types[0] + casted_self: ir.Value = Gen.builder.bitcast(arg, actual_self_type, name="self_cast") + call_args.append(casted_self) + else: + call_args.append(arg) + + result: ir.Value = Gen.builder.call(parent_func, call_args, name="inherited_call") + + if isinstance(parent_ret_type, ir.VoidType): + Gen.builder.ret_void() + else: + Gen.builder.ret(result) + + Gen.builder = saved_builder + Gen.func = saved_func + + def _fill_vtable_with_methods(self, ClassName: str, Gen: LlvmGeneratorMixin) -> None: + methods: list = Gen.class_methods.get(ClassName, []) + if not methods or ClassName not in Gen.Vtables: + return + Vtable: ir.GlobalVariable = Gen.Vtables[ClassName] + VtableType: ir.Type = Vtable.type.pointee + null_i8ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + initializers: list[ir.Constant] = [] + for mi, method_name in enumerate(methods): + method_short: str = method_name.split('.')[-1] if '.' in method_name else method_name + func: ir.Function | None = Gen._find_function(f"{ClassName}.{method_short}") + if not func: + func = Gen._find_function(f"{ClassName}.{method_short}__") + if not func: + p: str | None = Gen.class_parent.get(ClassName) + while p: + func = Gen._find_function(f"{p}.{method_short}") + if not func: + func = Gen._find_function(f"{p}.{method_short}__") + if func: + break + p = Gen.class_parent.get(p) + if func: + initializers.append(func.bitcast(ir.PointerType(ir.IntType(8)))) + else: + initializers.append(null_i8ptr) + if initializers: + Vtable.initializer = ir.Constant(VtableType, initializers) + + def _scan_method_body_for_self_members(self, body: list[ast.AST], ClassName: str, existing_members: set[str], Gen: LlvmGeneratorMixin) -> None: + for stmt in body: + self._scan_node_for_self_assign(stmt, ClassName, existing_members, Gen) + + def _scan_node_for_self_assign(self, node: ast.AST, ClassName: str, existing_members: set[str], Gen: LlvmGeneratorMixin) -> None: + if isinstance(node, ast.AnnAssign): + if (isinstance(node.target, ast.Attribute) + and isinstance(node.target.value, ast.Name) + and node.target.value.id == 'self'): + VarName: str = node.target.attr + if VarName not in existing_members: + try: + TypeInfo: CTypeInfo | None = CTypeInfo.FromNode(node.annotation, self.Trans.SymbolTable) + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + MemberType: ir.Type = Gen._ctype_to_llvm(TypeInfo) + if isinstance(MemberType, ir.VoidType): + MemberType = ir.IntType(32) + except Exception as e: # 回退：成员类型解析失败时使用默认 i32 + _vlog().warning(f"HandlesClassDef: 忽略异常 {e}", exc_info=e) + MemberType = ir.IntType(32) + Gen.class_members[ClassName].append((VarName, MemberType)) + Gen.class_member_signeds[ClassName][VarName] = None + Gen.class_member_bitfields[ClassName][VarName] = 0 + Gen.class_member_byteorders[ClassName][VarName] = "" + Gen.class_member_bitoffsets[ClassName][VarName] = 0 + existing_members.add(VarName) + len_name = f'{VarName}__len' + if isinstance(MemberType, ir.PointerType) and node.value and isinstance(node.value, ast.List): + if len_name not in existing_members: + Gen.class_members[ClassName].append((len_name, ir.IntType(64))) + Gen.class_member_signeds[ClassName][len_name] = None + Gen.class_member_bitfields[ClassName][len_name] = 0 + Gen.class_member_byteorders[ClassName][len_name] = "" + Gen.class_member_bitoffsets[ClassName][len_name] = 0 + Gen.class_member_defaults[ClassName][len_name] = ir.Constant(ir.IntType(64), len(node.value.elts)) + existing_members.add(len_name) + elif isinstance(node, ast.Assign): + for target in node.targets: + if (isinstance(target, ast.Attribute) + and isinstance(target.value, ast.Name) + and target.value.id == 'self'): + VarName = target.attr + if VarName not in existing_members: + MemberType = ir.IntType(32) + if node.value: + try: + if isinstance(node.value, ast.Constant): + if isinstance(node.value.value, int): + MemberType = ir.IntType(32) + elif isinstance(node.value.value, float): + MemberType = ir.DoubleType() + elif isinstance(node.value.value, str): + MemberType = ir.PointerType(ir.IntType(8)) + elif isinstance(node.value, ast.Name): + if hasattr(self, '_current_method_args'): + arg_info = self._current_method_args.get(node.value.id) + if arg_info: + TypeInfo = CTypeInfo.FromNode(arg_info, self.Trans.SymbolTable) + if TypeInfo: + InferredType = Gen._ctype_to_llvm(TypeInfo) + if not isinstance(InferredType, ir.VoidType): + MemberType = InferredType + elif isinstance(node.value, ast.Call): + if isinstance(node.value.func, ast.Name): + if node.value.func.id in ('float', 'double'): + MemberType = ir.DoubleType() + elif node.value.func.id in ('int', 'long', 'short', 'char'): + MemberType = ir.IntType(32) + elif isinstance(node.value.func, ast.Attribute): + if node.value.func.attr == 'len': + MemberType = ir.IntType(32) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + Gen.class_members[ClassName].append((VarName, MemberType)) + Gen.class_member_signeds[ClassName][VarName] = None + Gen.class_member_bitfields[ClassName][VarName] = 0 + Gen.class_member_byteorders[ClassName][VarName] = "" + Gen.class_member_bitoffsets[ClassName][VarName] = 0 + existing_members.add(VarName) + for child in ast.iter_child_nodes(node): + if isinstance(child, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef)): + continue + self._scan_node_for_self_assign(child, ClassName, existing_members, Gen) + + def _EmitUnionLlvm(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin) -> None: + ClassName: str = Node.name + union_member_types: list[ir.Type] = [] + for item in Node.body: + if isinstance(item, ast.ClassDef): + nested_class_name: str = item.name + nested_member_types: list[ir.Type] = [] + for nested_item in item.body: + parsed = self.CollectAnnAssignMember(nested_item, Gen) + if parsed: + nested_member_types.append(parsed[1]) + if nested_member_types: + nested_struct_type: ir.IdentifiedStructType = ir.IdentifiedStructType(Gen.module, f"{ClassName}_{nested_class_name}") + nested_struct_type.set_body(*nested_member_types) + Gen.structs[f"{ClassName}_{nested_class_name}"] = nested_struct_type + union_member_types.append(nested_struct_type) + nested_class_full_name: str = f"{ClassName}_{nested_class_name}" + if nested_class_full_name not in Gen.class_members: + Gen.class_members[nested_class_full_name] = [] + if nested_class_full_name not in Gen.class_member_signeds: + Gen.class_member_signeds[nested_class_full_name] = {} + for nested_item in item.body: + parsed = self.CollectAnnAssignMember(nested_item, Gen) + if parsed: + VarName, MemberType, TypeInfo = parsed + Gen.class_members[nested_class_full_name].append((VarName, MemberType)) + if TypeInfo and hasattr(TypeInfo.BaseType, 'IsSigned'): + Gen.class_member_signeds[nested_class_full_name][VarName] = TypeInfo.BaseType.IsSigned + else: + Gen.class_member_signeds[nested_class_full_name][VarName] = None + if union_member_types: + max_size: int = 0 + for member_type in union_member_types: + try: + size: int = member_type.get_abi_size(ir.DataLayout(Gen.module.data_layout)) + except Exception as e: # 回退：获取类型大小失败时使用默认值 8 + _vlog().warning(f"HandlesClassDef: 忽略异常 {e}", exc_info=e) + size = 8 + max_size = max(max_size, size) + union_type: ir.IdentifiedStructType = ir.IdentifiedStructType(Gen.module, ClassName) + union_type.set_body(ir.ArrayType(ir.IntType(8), max_size)) + Gen.structs[ClassName] = union_type + UnionNode: CTypeInfo = CTypeInfo() + UnionNode.Name = ClassName + UnionNode.IsUnion = True + self.Trans.SymbolTable.insert(ClassName, UnionNode) + + def _EmitREnumLlvm(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin) -> None: + ClassName: str = Node.name + variant_info: list[tuple] = [] + variant_names: list[str] = [] + + # 预注册 REnum opaque 结构体到 Gen.structs 和 SymbolTable， + # 使自引用成员（如 LLVMType | t.CPtr）在收集成员时能正确解析为 %LLVMType*。 + # 若不预注册，GetCTypeInfo 找不到 LLVMType → 回退到 IsEnum → i32 → | t.CPtr → i32*（错误）。 + PreRenumType: ir.IdentifiedStructType = ir.IdentifiedStructType(Gen.module, ClassName) + Gen.structs[ClassName] = PreRenumType + PreTypeInfo: CTypeInfo = CTypeInfo() + PreTypeInfo.Name = ClassName + PreTypeInfo.BaseType = t.REnum(ClassName) + PreTypeInfo.IsRenum = True + PreTypeInfo.IsEnum = True + # REnum 在内存布局上是 struct（{ i32 __tag, }）， + # 必须设置 IsStruct=True 才能让 HandlesTypeMerge 的 CPtr 吸收逻辑生效： + # `REnumType | t.CPtr` 应得到 REnumType*（PtrCount=1）而非 REnumType**（PtrCount=2）。 + # 否则 include-imported 函数返回类型会出现双指针，与声明单指针不匹配 → 运行时崩溃。 + PreTypeInfo.IsStruct = True + PreTypeInfo.RenumVariants = [] + self.Trans.SymbolTable.insert(ClassName, PreTypeInfo) + + # 第一遍：收集显式赋值的 tag 值，确定起始索引 + explicit_values: dict[str, int] = {} + for item in Node.body: + if isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + if item.value and isinstance(item.value, ast.Constant): + explicit_values[target.id] = item.value.value + # 类变体从 max(explicit_values, default=-1) + 1 开始 + next_class_tag: int = max(explicit_values.values()) + 1 if explicit_values else 0 + variant_index: int = 0 + + for item in Node.body: + if isinstance(item, ast.ClassDef): + VariantName: str = item.name + variant_names.append(VariantName) + member_types: list[ir.Type] = [] + member_names: list[str] = [] + member_annotations: list[ast.AST | None] = [] + for nested_item in item.body: + parsed = self.CollectAnnAssignMember(nested_item, Gen) + if parsed: + VarName, MemberType, TypeInfo = parsed + member_types.append(MemberType) + member_names.append(VarName) + member_annotations.append(nested_item.annotation if TypeInfo else None) + # 类变体使用专用计数器，避免与显式值冲突 + tag: int = next_class_tag + next_class_tag += 1 + variant_info.append((VariantName, member_types, member_names, member_annotations, item.lineno, tag)) + elif isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + VarName = target.id + variant_names.append(VarName) + value: int | None = None + if item.value and isinstance(item.value, ast.Constant): + value = item.value.value + else: + value = variant_index + variant_index += 1 + variant_info.append((VarName, [], [], [], item.lineno, value)) + # REnum 布局是所有变体的联合体。每个字段位置必须能容纳任何变体在该位置的字段。 + # Bug 修复：原来只选总尺寸最大的变体，但不同变体在同一位置可能有不同尺寸的字段。 + # 例如 MetaType: Concrete 位置3是 i32(4字节)，但 Var 位置3是 MetaType*(8字节)。 + # 用 Concrete 的布局会导致 Var.Instance 指针被截断为4字节 → 运行时崩溃。 + # 修复：按每个位置的 max 字段尺寸构建布局。 + def _ir_type_size(t: ir.Type) -> int: + if isinstance(t, ir.IntType): + return max(4, t.width // 8) + elif isinstance(t, ir.PointerType): + return 8 + elif isinstance(t, ir.FloatType): + return 4 + elif isinstance(t, ir.DoubleType): + return 8 + elif isinstance(t, ir.ArrayType): + if isinstance(t.element, ir.IntType): + return t.element.width // 8 * t.count + return 8 * t.count + else: + return 8 + + all_variant_types: list[list[ir.Type]] = [] + max_field_count: int = 0 + for _, member_types, _, _, _, _ in variant_info: + if member_types: + full_types: list[ir.Type] = [ir.IntType(32)] + member_types + all_variant_types.append(full_types) + if len(full_types) > max_field_count: + max_field_count = len(full_types) + max_variant_struct: list[ir.Type] = [] + for pos in range(max_field_count): + pos_max_size: int = 0 + for vt in all_variant_types: + if pos < len(vt): + sz: int = _ir_type_size(vt[pos]) + if sz > pos_max_size: + pos_max_size = sz + # 按最大尺寸选择 LLVM 类型：<=4 用 i32，>4 用 i64（8字节，可容纳指针） + if pos_max_size <= 4: + max_variant_struct.append(ir.IntType(32)) + else: + max_variant_struct.append(ir.IntType(64)) + if not max_variant_struct: + max_variant_struct = [ir.IntType(32), ir.IntType(32)] + # 复用预注册的 opaque 结构体，设置 body 使其完整 + PreRenumType.set_body(*max_variant_struct) + for VariantName, member_types, member_names, member_annotations, lineno, tag_value in variant_info: + NestedStructName: str = f"{ClassName}_{VariantName}" + if member_types: + padded_member_types: list[ir.Type] = list(max_variant_struct) + nested_struct_type: ir.IdentifiedStructType = ir.IdentifiedStructType(Gen.module, NestedStructName) + nested_struct_type.set_body(*padded_member_types) + Gen.structs[NestedStructName] = nested_struct_type + if NestedStructName not in Gen.class_members: + Gen.class_members[NestedStructName] = [] + if NestedStructName not in Gen.class_member_signeds: + Gen.class_member_signeds[NestedStructName] = {} + Gen.class_members[NestedStructName].append(('__tag', ir.IntType(32))) + Gen.class_member_signeds[NestedStructName]['__tag'] = None + for i, (mname, mtype) in enumerate(zip(member_names, member_types)): + Gen.class_members[NestedStructName].append((mname, mtype)) + try: + ti: CTypeInfo | None = CTypeInfo.FromNode(member_annotations[i], self.Trans.SymbolTable) + if ti and hasattr(ti.BaseType, 'IsSigned'): + Gen.class_member_signeds[NestedStructName][mname] = ti.BaseType.IsSigned + else: + Gen.class_member_signeds[NestedStructName][mname] = None + except Exception as e: # 回退：签名信息解析失败时设为 None + _vlog().warning(f"HandlesClassDef: 忽略异常 {e}", exc_info=e) + Gen.class_member_signeds[NestedStructName][mname] = None + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = VariantName + MemberNode.BaseType = t.CEnum(ClassName) + MemberNode.value = tag_value + MemberNode.EnumName = ClassName + MemberNode.Lineno = lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VariantName, MemberNode) + # 更新预注册的 REnum 类型信息（填充变体列表） + PreTypeInfo.RenumVariants = variant_names + + def _RegisterEnumMembers(self, Node: ast.ClassDef) -> None: + ClassName: str = Node.name + EnumTypeInfo: CTypeInfo = CTypeInfo() + EnumTypeInfo.Name = ClassName + EnumTypeInfo.BaseType = t.CEnum(ClassName) + EnumTypeInfo.IsEnum = True + self.Trans.SymbolTable.insert(ClassName, EnumTypeInfo) + enum_export = self.Trans.Exportable.add_enum( + name=ClassName, + lineno=Node.lineno, + is_public=True + ) + next_enum_value: int = 0 + for item in Node.body: + if isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + VarName: str = target.id + value: int | str | None = None + if item.value: + if isinstance(item.value, ast.Constant): + value = item.value.value + next_enum_value = value + 1 + elif isinstance(item.value, ast.UnaryOp) and isinstance(item.value.op, ast.USub): + if isinstance(item.value.operand, ast.Constant): + value = -item.value.operand.value + next_enum_value = value + 1 + elif isinstance(item.value, ast.Name): + value = item.value.id + ref_info: CTypeInfo | None = self.Trans.SymbolTable.lookup(value) + if ref_info: + if ref_info.value is not None and isinstance(ref_info.value, int): + next_enum_value = ref_info.value + 1 + else: + value = next_enum_value + next_enum_value += 1 + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = VarName + MemberNode.BaseType = t.CEnum(ClassName) + MemberNode.value = value + MemberNode.EnumName = ClassName + MemberNode.Lineno = item.lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VarName, MemberNode) + enum_export.members.append(ExportEnumMember( + name=VarName, + value=value, + lineno=item.lineno + )) + elif isinstance(item, ast.AnnAssign): + if isinstance(item.target, ast.Name): + VarName = item.target.id + value = None + if item.value: + if isinstance(item.value, ast.Constant): + value = item.value.value + next_enum_value = value + 1 + elif isinstance(item.value, ast.UnaryOp) and isinstance(item.value.op, ast.USub): + if isinstance(item.value.operand, ast.Constant): + value = -item.value.operand.value + next_enum_value = value + 1 + elif isinstance(item.value, ast.Name): + value = item.value.id + ref_info: CTypeInfo | None = self.Trans.SymbolTable.lookup(value) + if ref_info: + if ref_info.value is not None and isinstance(ref_info.value, int): + next_enum_value = ref_info.value + 1 + else: + value = next_enum_value + next_enum_value += 1 + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = VarName + MemberNode.BaseType = t.CEnum(ClassName) + MemberNode.value = value + MemberNode.EnumName = ClassName + MemberNode.Lineno = item.lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VarName, MemberNode) + enum_export.members.append(ExportEnumMember( + name=VarName, + value=value, + lineno=item.lineno + )) + + def _EmitNewFunctionLlvm(self, ClassName: str, Gen: LlvmGeneratorMixin, IsCVTable: bool = False, IsCpythonObject: bool = False) -> None: + NewFuncName: str = f'{ClassName}.__before_init__' + existing_func: ir.Function | None = Gen.functions.get(NewFuncName) + if existing_func and len(existing_func.blocks) > 0: + return + MangledName: str = Gen._mangle_name(NewFuncName) + StructType: ir.Type = Gen.structs[ClassName] + if isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(ClassName, Gen) + StructType = Gen.structs.get(ClassName, StructType) + StructPtrType: ir.PointerType = ir.PointerType(StructType) + FuncType: ir.FunctionType = ir.FunctionType(ir.VoidType(), [StructPtrType]) + if existing_func: + func: ir.Function = existing_func + else: + func = Gen._get_or_declare_function(MangledName, FuncType) + Gen.functions[NewFuncName] = func + EntryBlock: ir.Block = func.append_basic_block(name="entry") + saved_builder: ir.IRBuilder = Gen.builder + saved_func: ir.Function = Gen.func + Gen.builder = ir.IRBuilder(EntryBlock) + Gen.func = func + StructPtr: ir.Argument = func.args[0] + base: int = 1 if IsCVTable else 0 + if IsCVTable and ClassName in Gen.Vtables: + VtablePtr: ir.Value = Gen.builder.gep(StructPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="vtable_slot") + VtableAddr: ir.Value = Gen.builder.bitcast(Gen.Vtables[ClassName], ir.PointerType(ir.IntType(8)), name="vtable_addr") + Gen.builder.store(VtableAddr, VtablePtr) + members: list[tuple] = Gen.class_members.get(ClassName, []) + defaults: dict = Gen.class_member_defaults.get(ClassName, {}) + struct_elems: tuple = StructType.elements if hasattr(StructType, 'elements') and StructType.elements is not None else () + for i, (member_name, member_type) in enumerate(members): + elem_idx: int = base + i + if len(struct_elems) > 0 and elem_idx >= len(struct_elems): + break + ElemPtr: ir.Value = Gen.builder.gep(StructPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), elem_idx)], name=f"{ClassName}_{member_name}") + if member_name in defaults: + try: + val: ir.Constant = defaults[member_name] + if not isinstance(val, ir.Constant): + val = ir.Constant(ElemPtr.type.pointee, val) + Gen.builder.store(val, ElemPtr) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + else: + try: + ZeroVal: ir.Constant = ir.Constant(ElemPtr.type.pointee, None) + Gen.builder.store(ZeroVal, ElemPtr) + except Exception: + pass + Gen.builder.ret_void() + Gen.builder = saved_builder + Gen.func = saved_func + + def _BuildScalarConstant(self, node: ast.AST, llvm_type: ir.Type, Gen: LlvmGeneratorMixin | None = None) -> ir.Constant | None: + """构建标量常量，委托给统一的 _BuildScalarConstantUnified 方法。""" + if Gen is None: + Gen = self.Trans.LlvmGen + return self.Trans._BuildScalarConstantUnified(node, llvm_type, Gen) + + def _BuildStructConstant(self, call_node: ast.Call, StructName: str, Gen: LlvmGeneratorMixin) -> ir.Constant | None: + struct_type: ir.Type = Gen.structs[StructName] + members: list[tuple] = Gen.class_members.get(StructName, []) + defaults: dict = Gen.class_member_defaults.get(StructName, {}) + member_values: list[ir.Constant] = [] + for member_name, member_type in members: + if member_name in defaults: + val: ir.Constant = defaults[member_name] + if not isinstance(val, ir.Constant): + try: + val = ir.Constant(member_type, val) + except Exception: # 回退：常量转换失败时使用零值 + val = ir.Constant(member_type, 0) + member_values.append(val) + else: + try: + if isinstance(member_type, (ir.IdentifiedStructType, ir.LiteralStructType, ir.PointerType, ir.ArrayType)): + member_values.append(ir.Constant(member_type, None)) + else: + member_values.append(ir.Constant(member_type, ir.Undefined)) + except Exception: # 回退：常量创建失败时使用零值 + member_values.append(ir.Constant(member_type, 0)) + if call_node.args: + for i, arg in enumerate(call_node.args): + if i < len(members): + _, member_type = members[i] + const: ir.Constant | None = self._BuildScalarConstant(arg, member_type, Gen) + if const is not None: + member_values[i] = const + try: + return ir.Constant(struct_type, member_values) + except Exception: # 回退：结构体常量创建失败 + return None diff --git a/lib/core/Handles/HandlesDelete.py b/lib/core/Handles/HandlesDelete.py new file mode 100644 index 0000000..f4871d9 --- /dev/null +++ b/lib/core/Handles/HandlesDelete.py @@ -0,0 +1,102 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle, FuncMeta +import ast +import llvmlite.ir as ir + + +class DeleteHandle(BaseHandle): + def _HandleDeleteLlvm(self, Node: ast.Delete) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + for target in Node.targets: + if isinstance(target, ast.Subscript): + ClassName: str | None = self.Trans.ExprHandler._get_var_class(target.value, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__delitem__'): + obj_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(target.value) + idx_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(target.slice) + if obj_val and idx_val: + # 处理 obj_val 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + if isinstance(obj_val.type, ir.IntType) and obj_val.type.width == 8: + obj_val = Gen.builder.inttoptr(obj_val, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + Gen.builder.call(Gen._get_function(f'{ClassName}.__delitem__'), [obj_val, idx_val], name=f"call_{ClassName}.__delitem__") + elif isinstance(target, ast.Name): + ClassName: str | None = Gen.var_struct_class.get(target.id) + if ClassName and Gen._has_function(f'{ClassName}.__delete__'): + obj_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(target) + if obj_val: + # 处理 obj_val 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + if isinstance(obj_val.type, ir.IntType) and obj_val.type.width == 8: + obj_val = Gen.builder.inttoptr(obj_val, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + Gen.builder.call(Gen._get_function(f'{ClassName}.__delete__'), [obj_val], name=f"call_{ClassName}.__delete__") + elif isinstance(target, ast.Attribute): + # 检查 property deleter + AttrName: str = target.attr + ClassName: str | None = None + obj_val_for_prop: ir.Value | None = None + + if isinstance(target.value, ast.Name) and target.value.id == 'self': + # del self.attr — 检查当前类的 property deleter + ClassName = self.Trans._CurrentCpythonObjectClass + if ClassName: + PropKey: str = f'{ClassName}.{AttrName}' + PropInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_DELETER in PropInfo.MetaList: + SelfVar: ir.Value | None = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: ir.Value = Gen._load(SelfVar, name="self") + DeleterFunc: ir.Function | None = Gen._get_function(PropKey + '$del') + if DeleterFunc and SelfPtr: + Gen.builder.call(DeleterFunc, [SelfPtr], name=f"prop_del_{PropKey}") + continue + else: + # del obj.attr — 检查 obj 类的 property deleter + ClassName = self.Trans.ExprHandler._get_var_class(target.value, Gen) + if ClassName: + PropKey: str = f'{ClassName}.{AttrName}' + PropInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_DELETER in PropInfo.MetaList: + obj_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(target.value) + if obj_val: + DeleterFunc: ir.Function | None = Gen._get_function(PropKey + '$del') + if DeleterFunc: + Gen.builder.call(DeleterFunc, [obj_val], name=f"prop_del_{PropKey}") + continue + + # 原有的 __del__ 处理逻辑 + obj_val: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(target) + if obj_val: + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.PointerType): + obj_val = Gen._load(obj_val, name="Load_del_ptr") + ClassName = None + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + for CN, ST in Gen.structs.items(): + if obj_val.type.pointee == ST: + ClassName = CN + break + elif isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + if isinstance(target.value, ast.Name) and target.value.id == 'self': + AttrName = target.attr + SelfClassName: str | None = self.Trans._CurrentCpythonObjectClass + if SelfClassName and SelfClassName in Gen.class_members: + for member_name, member_type in Gen.class_members[SelfClassName]: + if member_name == AttrName: + for CN in Gen.structs: + if CN == AttrName or Gen._has_function(f'{CN}.__del__'): + ClassName = CN + break + break + if not ClassName: + for CN, ST in Gen.structs.items(): + if Gen._has_function(f'{CN}.__del__'): + ClassName = CN + break + if ClassName and Gen._has_function(f'{ClassName}.__del__'): + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + Gen.builder.call(Gen._get_function(f'{ClassName}.__del__'), [obj_val], name=f"call_{ClassName}.__del__") \ No newline at end of file diff --git a/lib/core/Handles/HandlesExpr.py b/lib/core/Handles/HandlesExpr.py new file mode 100644 index 0000000..1128d5e --- /dev/null +++ b/lib/core/Handles/HandlesExpr.py @@ -0,0 +1,446 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +import ast +import re +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle +from lib.constants.config import mode as _config_mode +from lib.includes.t import CTypeRegistry as _CR + + +class ExprHandle(BaseHandle): + + # list 元素类型 → (字节大小, LLVM 元素类型) 映射表 + # 用于 _HandleListLlvm 中 alloc_size / byte_offset / elem_ptr 类型统一查表， + # 消除原代码三处重复的 if-elif 链。 + _ELEM_TYPE_INFO: dict[str, tuple[int, ir.Type]] = { + 'i8': (1, ir.IntType(8)), + 'i16': (2, ir.IntType(16)), + 'i32': (4, ir.IntType(32)), + 'float': (4, ir.FloatType()), + 'double': (8, ir.DoubleType()), + } + + def __init__(self, translator: "Translator") -> None: + super().__init__(translator) + + def GetOpSymbol(self, Op: ast.cmpop | ast.operator | ast.unaryoperator) -> str | None: + return self.Trans.ExprUtils.GetOpSymbol(Op) + + def GetUnaryOpSymbol(self, Op: ast.unaryoperator) -> str | None: + return self.Trans.ExprUtils.GetUnaryOpSymbol(Op) + + def GetComparatorSymbol(self, Op: ast.cmpop) -> str | None: + return self.Trans.ExprUtils.GetComparatorSymbol(Op) + + def _get_var_class(self, node: ast.AST, Gen: LlvmCodeGenerator) -> str | None: + return self.Trans.ExprUtils._get_var_class(node, Gen) + + def _try_operator_overLoad(self, ClassName: str, op_name: str, obj_val: ir.Value, Gen: LlvmCodeGenerator, other_val: ir.Value | None = None) -> ir.Value | None: + return self.Trans.ExprUtils._try_operator_overLoad(ClassName, op_name, obj_val, Gen, other_val) + + def _get_llvm_member_offset(self, field_name: str, ClassName: str, Gen: LlvmCodeGenerator) -> int | None: + return self.Trans.ExprAttrHandle._get_llvm_member_offset(field_name, ClassName, Gen) + + def HandleExprLlvm(self, Node: ast.AST, VarType: ir.Type | str | None = None) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if not Gen or not Gen.builder: + return None + if isinstance(Node, ast.Constant): + return self._HandleConstantLlvm(Node, VarType) + elif isinstance(Node, ast.Name): + return self._HandleNameLlvm(Node, VarType) + elif isinstance(Node, ast.BinOp): + return self.Trans.ExprOpsHandle._HandleBinOpLlvm(Node, VarType) + elif isinstance(Node, ast.BoolOp): + return self.Trans.ExprOpsHandle._HandleBoolOpLlvm(Node) + elif isinstance(Node, ast.UnaryOp): + return self.Trans.ExprOpsHandle._HandleUnaryOpLlvm(Node) + elif isinstance(Node, ast.Call): + return self.Trans.ExprCallHandle._HandleCallLlvm(Node) + elif isinstance(Node, ast.Compare): + return self.Trans.ExprOpsHandle._HandleCompareLlvm(Node) + elif isinstance(Node, ast.Attribute): + return self.Trans.ExprAttrHandle._HandleAttributeLlvm(Node) + elif isinstance(Node, ast.Subscript): + return self.Trans.ExprAttrHandle._HandleSubscriptLlvm(Node) + elif isinstance(Node, ast.IfExp): + return self.Trans.ExprLambdaHandle._HandleIfExpLlvm(Node) + elif isinstance(Node, ast.NamedExpr): + return self.Trans.ExprLambdaHandle._HandleNamedExprLlvm(Node) + elif isinstance(Node, ast.JoinedStr): + return self.Trans.ExprFormatHandle._HandleJoinedStrLlvm(Node) + elif isinstance(Node, ast.Lambda): + return self.Trans.ExprLambdaHandle._HandleLambdaLlvm(Node) + elif isinstance(Node, ast.List): + return self._HandleListLlvm(Node, VarType) + return ir.Constant(ir.IntType(32), 0) + + def _HandleListLlvm(self, Node: ast.List, VarType: ir.Type | str | None = None) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + elements: list[ast.expr] = Node.elts + if not elements: + return ir.Constant(ir.IntType(8).as_pointer(), None) + + array_len: int = len(elements) + elem_type: str = 'i8' + + if VarType and isinstance(VarType, str) and 'list[' in VarType: + match: re.Match | None = re.search(r'list\[([^,\]]+)(?:,\s*(\d+))?\]', VarType) + if match: + type_str: str = match.group(1).strip() + if match.group(2): + array_len = int(match.group(2)) + if 'CChar' in type_str or 'char' in type_str.lower(): + elem_type = 'i8' + elif 'CFloat' in type_str or type_str.lower() == 'float': + elem_type = 'float' + elif 'CDouble' in type_str or type_str.lower() == 'double': + elem_type = 'double' + elif 'CInt' in type_str or ('int' in type_str.lower() and 'unsigned' not in type_str.lower()): + elem_type = 'i32' + elif 'CUnsignedInt' in type_str or 'unsigned' in type_str.lower(): + elem_type = 'i32' + elif 'CUnsignedChar' in type_str or 'unsigned char' in type_str.lower(): + elem_type = 'i8' + elif 'CShort' in type_str or 'short' in type_str.lower(): + elem_type = 'i16' + elif 'CUnsignedShort' in type_str: + elem_type = 'i16' + elif elements and all(isinstance(e, (ast.Constant,)) and isinstance(getattr(e, 'value', None), float) for e in elements): + elem_type = 'float' + + alloc_size: int = array_len + elem_info: tuple[int, ir.Type] = self._ELEM_TYPE_INFO.get(elem_type, (1, ir.IntType(8))) + elem_size: int = elem_info[0] + elem_llvm_type: ir.Type = elem_info[1] + alloc_size *= elem_size + + malloc_fn: Any = None + malloc_arg_type: ir.IntType = ir.IntType(64) + for fn in Gen.module.global_values: + if fn.name == 'malloc': + malloc_fn = fn + func_ftype: Any = getattr(fn, 'ftype', None) + if func_ftype and getattr(func_ftype, 'args', None): + if func_ftype.args: + malloc_arg_type = fn.ftype.args[0] + break + + if not malloc_fn: + try: + malloc_fn_type: ir.FunctionType = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.IntType(64)]) + malloc_fn = ir.Function(Gen.module, malloc_fn_type, name='malloc') + Gen.functions['malloc'] = malloc_fn + except Exception: # 回退：malloc 函数创建失败时返回 null + return ir.Constant(ir.IntType(8).as_pointer(), None) + + alloc_ptr: Any = Gen.builder.call(malloc_fn, [ir.Constant(malloc_arg_type, alloc_size)]) + cast_ptr: Any = Gen.builder.bitcast(alloc_ptr, ir.PointerType(ir.IntType(8))) + + for i, elem in enumerate(elements): + if i >= array_len: + break + + elem_val: Any = self.HandleExprLlvm(elem, VarType=elem_type) + if elem_val is None: + continue + + byte_offset: int = i + if elem_type == 'i32': + byte_offset = i * 4 + elif elem_type == 'i16': + byte_offset = i * 2 + elif elem_type == 'float': + byte_offset = i * 4 + elif elem_type == 'double': + byte_offset = i * 8 + + ptr_as_int: Any = Gen.builder.ptrtoint(cast_ptr, ir.IntType(64)) + new_ptr_val: Any = Gen.builder.add(ptr_as_int, ir.Constant(ir.IntType(64), byte_offset)) + if elem_type == 'i32': + elem_ptr: Any = Gen.builder.inttoptr(new_ptr_val, ir.PointerType(ir.IntType(32))) + elif elem_type == 'i16': + elem_ptr = Gen.builder.inttoptr(new_ptr_val, ir.PointerType(ir.IntType(16))) + elif elem_type == 'float': + elem_ptr = Gen.builder.inttoptr(new_ptr_val, ir.PointerType(ir.FloatType())) + elif elem_type == 'double': + elem_ptr = Gen.builder.inttoptr(new_ptr_val, ir.PointerType(ir.DoubleType())) + else: + elem_ptr = Gen.builder.inttoptr(new_ptr_val, ir.PointerType(ir.IntType(8))) + + if elem_type == 'i8': + if isinstance(elem_val.type, ir.IntType) and elem_val.type.width > 8: + elem_val = Gen.builder.trunc(elem_val, ir.IntType(8), name=f"trunc_elem_{i}") + elif isinstance(elem_val.type, ir.PointerType): + elem_val = Gen.builder.ptrtoint(elem_val, ir.IntType(64), name=f"ptrtoint_elem_{i}") + elem_val = Gen.builder.trunc(elem_val, ir.IntType(8), name=f"trunc_elem_{i}") + elif elem_type == 'i32': + if isinstance(elem_val.type, ir.IntType): + if elem_val.type.width < 32: + elem_val = Gen.builder.zext(elem_val, ir.IntType(32), name=f"zext_elem_{i}") + elif elem_val.type.width > 32: + elem_val = Gen.builder.trunc(elem_val, ir.IntType(32), name=f"trunc_elem_{i}") + elif isinstance(elem_val.type, ir.PointerType): + elem_val = Gen.builder.ptrtoint(elem_val, ir.IntType(32), name=f"ptrtoint_elem_{i}") + elif elem_type == 'i16': + if isinstance(elem_val.type, ir.IntType): + if elem_val.type.width < 16: + elem_val = Gen.builder.zext(elem_val, ir.IntType(16), name=f"zext_elem_{i}") + elif elem_val.type.width > 16: + elem_val = Gen.builder.trunc(elem_val, ir.IntType(16), name=f"trunc_elem_{i}") + elif elem_type == 'float': + if isinstance(elem_val.type, ir.DoubleType): + elem_val = Gen.builder.fptrunc(elem_val, ir.FloatType(), name=f"fptrunc_elem_{i}") + elif isinstance(elem_val.type, ir.IntType): + elem_val = Gen.builder.sitofp(elem_val, ir.FloatType(), name=f"sitofp_elem_{i}") + elif elem_type == 'double': + if isinstance(elem_val.type, ir.FloatType): + elem_val = Gen.builder.fpext(elem_val, ir.DoubleType(), name=f"fpext_elem_{i}") + elif isinstance(elem_val.type, ir.IntType): + elem_val = Gen.builder.sitofp(elem_val, ir.DoubleType(), name=f"sitofp_elem_{i}") + + Gen.builder.store(elem_val, elem_ptr) + + if elem_type in ('float', 'double'): + return Gen.builder.bitcast(alloc_ptr, ir.PointerType(elem_llvm_type), name=f"list_{elem_type}_ptr") + return cast_ptr + + def _HandleConstantLlvm(self, Node: ast.Constant, VarType: ir.Type | str | None = None) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + Value: Any = Node.value + if Value is None: + if VarType is not None and isinstance(VarType, ir.PointerType): + return ir.Constant(VarType, None) + return ir.Constant(ir.IntType(8).as_pointer(), None) + elif isinstance(Value, bool): + return ir.Constant(ir.IntType(32), 1 if Value else 0) + elif isinstance(Value, int): + if VarType is not None: + if isinstance(VarType, ir.IntType): + return ir.Constant(VarType, Value & ((1 << VarType.width) - 1)) + if isinstance(VarType, ir.PointerType): + return ir.Constant(ir.IntType(64), Value) + if isinstance(VarType, (ir.FloatType, ir.DoubleType)): + return ir.Constant(VarType, float(Value)) + if isinstance(VarType, str) and '*' in VarType: + return ir.Constant(ir.IntType(64), Value) + if isinstance(VarType, str): + resolved: Any = _CR.ResolveName(VarType) or _CR.LLVMToCType(VarType) + if resolved: + ctype_cls: type = resolved[0] if isinstance(resolved, tuple) else resolved + inst: Any = ctype_cls() + size: int = getattr(inst, 'Size', 32) + is_unsigned: bool = getattr(inst, 'IsSigned', True) is False + mask: int = (1 << size) - 1 + return ir.Constant(ir.IntType(size), Value & mask) + llvm_match: re.Match | None = re.match(r'^i(\d+)$', VarType) + if llvm_match: + width: int = int(llvm_match.group(1)) + return ir.Constant(ir.IntType(width), Value & ((1 << width) - 1)) + if Value < 0 or Value > 0xFFFFFFFF: + return ir.Constant(ir.IntType(64), Value) + return ir.Constant(ir.IntType(32), Value & 0xFFFFFFFF) + elif isinstance(Value, float): + if VarType and isinstance(VarType, ir.FloatType): + return ir.Constant(VarType, Value) + elif VarType and isinstance(VarType, ir.DoubleType): + return ir.Constant(VarType, Value) + return ir.Constant(ir.DoubleType(), Value) + elif isinstance(Value, str): + is_wide: bool = getattr(Node, 'kind', None) == 'u' + # 引号类型推断：单引号单字符 → i8，双引号/多字符/三重 → i8*，空 → 0(None) + quote_char: str | None = None + is_triple: bool = False + if not is_wide: + col_offset: int = getattr(Node, 'col_offset', 0) + node_lineno: int = getattr(Node, 'lineno', 0) + # AST 的 lineno 可能不准确（常量节点可能指向下一行）， + # 因此在 lineno 和 lineno-1 两行中查找引号 + for try_lineno in (node_lineno, node_lineno - 1): + source_line: str | None = Gen._get_source_line(try_lineno) + if source_line and col_offset < len(source_line): + ch: str = source_line[col_offset] + if ch in ("'", '"'): + quote_char = ch + if col_offset + 2 < len(source_line) and source_line[col_offset:col_offset + 3] in ("'''", '"""'): + is_triple = True + break + # 空字符串 → 创建指向 \0 的全局变量 (而非 null, 避免 strcmp 解引用空指针崩溃) + # 走与非空字符串相同的路径, 由下方 else 分支处理 + # 单引号单字符（非三重）→ i8 + is_char_literal: bool = (not is_triple) and (quote_char == "'") and (len(Value) == 1) and not is_wide + if is_char_literal: + if VarType is not None and isinstance(VarType, ir.PointerType): + pass # 类型不匹配：目标是 ptr 但值是 char，走 string 路径 + elif isinstance(VarType, ir.IntType): + return ir.Constant(VarType, ord(Value[0])) + elif isinstance(VarType, str): + resolved = _CR.ResolveName(VarType) or _CR.LLVMToCType(VarType) + if resolved: + ctype_cls = resolved[0] if isinstance(resolved, tuple) else resolved + inst = ctype_cls() + size = getattr(inst, 'Size', 32) + return ir.Constant(ir.IntType(size), ord(Value[0])) + llvm_match = re.match(r'^i(\d+)$', VarType) + if llvm_match: + return ir.Constant(ir.IntType(int(llvm_match.group(1))), ord(Value[0])) + elif VarType is not None: + return ir.Constant(ir.IntType(8), ord(Value[0])) + else: + # 无类型提示：单引号单字符默认为 i8 + return ir.Constant(ir.IntType(8), ord(Value[0])) + if is_wide: + str_value: str = Value + '\x00' + wide_chars: list[int] = [ord(c) for c in str_value] + target_count: int = len(wide_chars) + str_type: ir.ArrayType = ir.ArrayType(ir.IntType(16), target_count) + gv_name: str = f"str_const_{Gen.string_const_counter}" + gv: ir.GlobalVariable = ir.GlobalVariable(Gen.module, str_type, name=gv_name) + gv.initializer = ir.Constant(str_type, wide_chars) + gv.linkage = 'internal' + Gen.string_const_counter += 1 + ptr: Any = Gen.builder.gep(gv, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"{gv_name}_cast") + return ptr + else: + str_value = Value + '\x00' + str_bytes: bytes = str_value.encode('utf-8') + target_count = len(str_bytes) + if isinstance(VarType, ir.ArrayType) and isinstance(VarType.element, ir.IntType) and VarType.element.width == 8: + target_count = VarType.count + if len(str_bytes) < target_count: + str_bytes = str_bytes + b'\x00' * (target_count - len(str_bytes)) + else: + str_bytes = str_bytes[:target_count] + str_type = ir.ArrayType(ir.IntType(8), target_count) + gv_name = f"str_const_{Gen.string_const_counter}" + gv = ir.GlobalVariable(Gen.module, str_type, name=gv_name) + gv.initializer = ir.Constant(str_type, bytearray(str_bytes)) + gv.linkage = 'internal' + Gen.string_const_counter += 1 + ptr = Gen.builder.gep(gv, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"{gv_name}_cast") + return ptr + return ir.Constant(ir.IntType(32), 0) + + def _HandleNameLlvm(self, Node: ast.Name, VarType: ir.Type | str | None = None) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + VarName: str = Node.id + define_constants: dict[str, Any] = getattr(Gen, '_define_constants', {}) + if VarName in define_constants: + val: Any = define_constants[VarName] + if isinstance(val, int): + # Use VarType hint if available to avoid unnecessary type mismatch + if VarType is not None and isinstance(VarType, ir.IntType): + return ir.Constant(VarType, val & ((1 << VarType.width) - 1)) + if VarName in Gen.var_signedness and Gen.var_signedness[VarName]: + if val < -(1 << 31) or val > 0xFFFFFFFF: + return ir.Constant(ir.IntType(64), val & 0xFFFFFFFFFFFFFFFF) + return ir.Constant(ir.IntType(32), val & 0xFFFFFFFF) + if val < -(1 << 31) or val > 0xFFFFFFFF: + return ir.Constant(ir.IntType(64), val) + return ir.Constant(ir.IntType(32), val & 0xFFFFFFFF) + elif isinstance(val, float): + return ir.Constant(ir.DoubleType(), val) + elif isinstance(val, str): + return Gen._create_string_global(val) + # 如果 _define_constants 中没有，尝试从符号表查找 + if VarName not in getattr(Gen, '_define_constants', {}): + try: + sym_info: Any = self.translator.SymbolTable.lookup(VarName) + if sym_info and sym_info.IsDefine and sym_info.DefineValue is not None: + val = sym_info.DefineValue + if isinstance(val, int): + # Use VarType hint if available + if VarType is not None and isinstance(VarType, ir.IntType): + return ir.Constant(VarType, val & ((1 << VarType.width) - 1)) + if val < -(1 << 31) or val > 0xFFFFFFFF: + return ir.Constant(ir.IntType(64), val) + return ir.Constant(ir.IntType(32), val & 0xFFFFFFFF) + elif isinstance(val, float): + return ir.Constant(ir.DoubleType(), val) + elif isinstance(val, str): + return Gen._create_string_global(val) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + # 优先检查 variables（alloca），因为变量可能已被 += 等操作迁移到 variables + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: Any = Gen.variables[VarName] + if isinstance(VarPtr, ir.GlobalVariable) and isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + if VarName in Gen.global_struct_class: + Gen.var_struct_class[VarName] = Gen.global_struct_class[VarName] + return VarPtr + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + if VarName in Gen.global_struct_class: + Gen.var_struct_class[VarName] = Gen.global_struct_class[VarName] + return VarPtr + loaded: ir.Value = Gen._load(VarPtr, name=VarName) + # 大端局部变量：读取时 bswap 还原 + loaded = Gen._apply_bswap_if_big(loaded, getattr(Gen, 'local_var_byteorders', {}).get(VarName, ""), f"bswap_load_{VarName}") + return loaded + if VarName in Gen._reg_values: + return Gen._reg_values[VarName] + if VarName in Gen._direct_values: + return Gen._direct_values[VarName] + if VarName in Gen.global_vars and VarName in Gen.module.globals: + GVar: Any = Gen.module.globals[VarName] + Gen.variables[VarName] = GVar + if isinstance(GVar, ir.Function): + return GVar + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar + return Gen._load(GVar, name=VarName) + if VarName in Gen.module.globals: + GVar = Gen.module.globals[VarName] + Gen.variables[VarName] = GVar + if isinstance(GVar, ir.Function): + return GVar + if VarName in Gen.global_struct_class: + Gen.var_struct_class[VarName] = Gen.global_struct_class[VarName] + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar + return Gen._load(GVar, name=VarName) + if VarName == 'self': + return Gen.GetVarPtr('self') + if Gen._has_function(VarName): + func: Any = Gen._get_function(VarName) + if func: + return Gen.builder.bitcast(func, ir.IntType(8).as_pointer(), name=f"funcptr_{VarName}") + if VarName in Gen.class_methods: + return ir.Constant(ir.IntType(8).as_pointer(), None) + if VarName == 'True' or VarName == 'False': + return ir.Constant(ir.IntType(32), 1 if VarName == 'True' else 0) + SymInfo: Any = self.Trans.SymbolTable.lookup(VarName) + if SymInfo: + if SymInfo.IsEnumMember and isinstance(SymInfo.value, int): + return ir.Constant(ir.IntType(32), SymInfo.value) + if SymInfo.IsFunction or SymInfo.IsFuncPtr: + MangledName: str = Gen._mangle_func_name(VarName) + if MangledName in Gen.module.globals: + g: Any = Gen.module.globals[MangledName] + if isinstance(g, ir.Function): + Gen.functions[VarName] = g + return Gen.builder.bitcast(g, ir.IntType(8).as_pointer(), name=f"funcptr_{VarName}") + for sha1 in Gen.ModuleSha1Map.values(): + prefixed: str = f"{sha1}.{VarName}" + if prefixed in Gen.module.globals: + g = Gen.module.globals[prefixed] + if isinstance(g, ir.Function): + Gen.functions[VarName] = g + return Gen.builder.bitcast(g, ir.IntType(8).as_pointer(), name=f"funcptr_{VarName}") + t_c_imported: dict[str, tuple[str, str]] = getattr(self.Trans, '_t_c_imported_names', {}) + if VarName in t_c_imported: + src_module: str + src_name: str + src_module, src_name = t_c_imported[VarName] + virtual_attr: ast.Attribute = ast.Attribute( + value=ast.Name(id=src_module, ctx=ast.Load()), + attr=src_name, + ctx=ast.Load() + ) + ast.copy_location(virtual_attr, Node) + return self.Trans.ExprAttrHandle._HandleAttributeLlvm(virtual_attr) + return ir.Constant(ir.IntType(32), 0) \ No newline at end of file diff --git a/lib/core/Handles/HandlesExprAsm.py b/lib/core/Handles/HandlesExprAsm.py new file mode 100644 index 0000000..22cff72 --- /dev/null +++ b/lib/core/Handles/HandlesExprAsm.py @@ -0,0 +1,580 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +import ast +import re +import llvmlite.ir as ir +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import BaseHandle +from lib.core.SymbolUtils import IsTModule +from lib.includes import t + +# 猴子补丁 llvmlite.ir.InlineAsm 以支持 asm_dialect 参数 +# 添加保护避免重复补丁 +_InlineAsmPatched = getattr(ir.InlineAsm, '_transpyc_patched', False) + +if not _InlineAsmPatched: + _OrigInlineAsmInit = ir.InlineAsm.__init__ + _OrigInlineAsmDescr = ir.InlineAsm.descr + + def _patched_inline_asm_init(self, ftype, asm, constraint, side_effect=False, asm_dialect=None): + _OrigInlineAsmInit(self, ftype, asm, constraint, side_effect) + self.asm_dialect = asm_dialect + + def _patched_inline_asm_descr(self, buf): + sideeffect = 'sideeffect' if self.side_effect else '' + dialect_str = getattr(self, 'asm_dialect', None) + if dialect_str == 'intel': + dialect = 'inteldialect' + else: + dialect = '' + fmt = 'asm {sideeffect} {dialect} "{asm}", "{constraint}"' + buf.append(fmt.format(sideeffect=sideeffect, dialect=dialect, asm=self.asm, + constraint=self.constraint)) + + ir.InlineAsm.__init__ = _patched_inline_asm_init + ir.InlineAsm.descr = _patched_inline_asm_descr + ir.InlineAsm._transpyc_patched = True + + +class ExprAsmHandle(BaseHandle): + @staticmethod + def _format_to_dialect(asm_format: str | None) -> str | None: + """将用户指定的 format 参数转换为 LLVM asm_dialect 值""" + fmt: str = asm_format.lower() if asm_format else 'intel' + if fmt == 'att': + return 'att' + elif fmt == 'arm': + return None # ARM 没有专门的 dialect 标记 + else: + return 'intel' # 默认 Intel + + def _HandleAsmLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + + op_arg: Any = None + input_arg: Any = None + output_arg: Any = None + asm_format: str = 'Intel' # 汇编格式: Intel / AT&T / ARM + for kw in Node.keywords: + if kw.arg == 'op': + op_arg = kw.value + elif kw.arg in ('input', 'inp', 'inputs'): + input_arg = kw.value + elif kw.arg in ('output', 'out', 'outputs'): + output_arg = kw.value + elif kw.arg == 'format': + if isinstance(kw.value, ast.Constant) and isinstance(kw.value.value, str): + asm_format = kw.value.value + + if len(Node.args) >= 1: + pos_op: Any = None + if len(Node.args) >= 2 and op_arg is None: + pos_op = Node.args[1] + return self._HandleAsmWithOpLlvm(Node.args[0], op_arg or pos_op, input_arg, output_arg, asm_format) + + if len(Node.args) < 2: + return ir.Constant(ir.IntType(32), 1) + + args: list[ast.AST] = Node.args + + if len(args) == 4: + output_type_arg: ast.AST = args[0] + asm_template_arg: ast.AST = args[1] + constraint_arg: ast.AST = args[2] + clobber_arg: ast.AST = args[3] + + output_type: ir.Type | None = self._infer_expr_llvm_type(output_type_arg) + if not output_type: + output_type = ir.IntType(32) + + asm_template: str = '' + if isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str): + asm_template = self._prepare_intel_asm(self._mangle_asm_calls(asm_template_arg.value)) + + constraint: str = '' + if isinstance(constraint_arg, ast.Constant) and isinstance(constraint_arg.value, str): + constraint = constraint_arg.value + + clobber: str = '' + if isinstance(clobber_arg, ast.Constant) and isinstance(clobber_arg.value, str): + clobber = clobber_arg.value + + full_constraint: str = constraint + func_type: ir.FunctionType = ir.FunctionType(output_type, []) + asm_dialect: str | None = self._format_to_dialect(asm_format) + inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, constraint, side_effect=True, asm_dialect=asm_dialect) + result: Any = Gen.builder.call(inline_asm, [], name="asm_result") + return result + + elif len(args) >= 2: + output_type_arg: ast.AST = args[0] + asm_template_arg: ast.AST = args[1] + + output_type: ir.Type | None = self._infer_expr_llvm_type(output_type_arg) + if not output_type: + output_type = ir.IntType(32) + + asm_template: str = '' + if isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str): + asm_template = self._prepare_intel_asm(self._mangle_asm_calls(asm_template_arg.value)) + + output_constraints: list[str] = [] + output_values: list[Any] = [] + if len(args) > 2 and isinstance(args[2], (ast.List, ast.Tuple)): + for item in args[2].elts: + if isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2: + constraint_node: ast.AST = item.elts[0] + value_node: ast.AST = item.elts[1] + if isinstance(constraint_node, ast.Constant) and isinstance(constraint_node.value, str): + output_constraints.append(constraint_node.value) + value: Any = self.HandleExprLlvm(value_node) + if value: + output_values.append(value) + + input_constraints: list[str] = [] + input_values: list[Any] = [] + if len(args) > 3 and isinstance(args[3], (ast.List, ast.Tuple)): + for item in args[3].elts: + if isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2: + constraint_node: ast.AST = item.elts[0] + value_node: ast.AST = item.elts[1] + if isinstance(constraint_node, ast.Constant) and isinstance(constraint_node.value, str): + input_constraints.append(constraint_node.value) + value: Any = self.HandleExprLlvm(value_node) + if value: + input_values.append(value) + + clobbers: list[str] = [] + if len(args) > 4 and isinstance(args[4], (ast.List, ast.Tuple)): + for item in args[4].elts: + if isinstance(item, ast.Constant) and isinstance(item.value, str): + clobbers.append(item.value) + + constraint_parts: list[str] = [] + for oc in output_constraints: + constraint_parts.append(oc) + + input_start_idx: int = len(output_constraints) + for ic in input_constraints: + constraint_parts.append(ic) + + full_constraint: str = ",".join(constraint_parts) + + if clobbers: + if full_constraint: + full_constraint += "," + for c in clobbers: + full_constraint += f"~{{{c}}}" + + operands: list[Any] = output_values + input_values + param_types: list[ir.Type] = [v.type for v in operands] + func_type: ir.FunctionType = ir.FunctionType(output_type, param_types) + asm_dialect: str | None = self._format_to_dialect(asm_format) + inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=False, asm_dialect=asm_dialect) + result: Any = Gen.builder.call(inline_asm, operands, name="asm_result") + return result + + return None + + def _prepare_intel_asm(self, template: str) -> str: + template = re.sub(r'%(\d+)', r'$\1', template) + template = re.sub(r'%([a-zA-Z][a-zA-Z0-9]*)', r'\1', template) + return template + + def _normalize_asm_template(self, template: str) -> str: + lines = template.split('\n') + if len(lines) <= 1: + return template + first_line = lines[0] + rest_lines = lines[1:] + stripped = [] + for line in rest_lines: + s = line.lstrip() + if s: + stripped.append(s) + else: + stripped.append('') + return '\n'.join([first_line] + stripped) + + def _process_joined_str_template(self, node: ast.AST, out_input_ops: list[Any], out_input_constraints: list[str], + out_output_ops: list[Any], out_output_constraints: list[str]) -> str: + if not isinstance(node, ast.JoinedStr): + if isinstance(node, ast.Constant) and isinstance(node.value, str): + return node.value + return '' + + raw_parts: list[tuple[str, str | int]] = [] + for value in node.values: + if isinstance(value, ast.Constant) and isinstance(value.value, str): + raw_parts.append(('text', value.value)) + elif isinstance(value, ast.FormattedValue): + expr: ast.AST = value.value + if isinstance(expr, ast.Call) and isinstance(expr.func, ast.Attribute): + if isinstance(expr.func.value, ast.Name) and expr.func.value.id == 'c': + if expr.func.attr == 'AsmInp': + if expr.args: + v: Any = self.HandleExprLlvm(expr.args[0]) + if v: + c: str = 'r' + if len(expr.args) >= 2: + c = self._parse_asm_descr(expr.args[1]) + if not c: + c = 'r' + out_input_ops.append(v) + out_input_constraints.append(c) + raw_parts.append(('input', len(out_input_ops) - 1)) + elif expr.func.attr == 'AsmOut': + if expr.args: + target_ptr: Any = self._get_var_ptr_for_asm(expr.args[0]) + if target_ptr: + v = target_ptr + else: + v = self.HandleExprLlvm(expr.args[0]) + if v: + c: str = '=r' + if len(expr.args) >= 2: + c = self._parse_asm_descr(expr.args[1]) + if not c: + c = '=r' + if not c.startswith('='): + c = '=' + c + out_output_ops.append(v) + out_output_constraints.append(c) + raw_parts.append(('output', len(out_output_ops) - 1)) + else: + fallback: Any = self.HandleExprLlvm(expr) + if fallback and isinstance(fallback.type, ir.IntType): + out_input_ops.append(fallback) + out_input_constraints.append('r') + raw_parts.append(('input', len(out_input_ops) - 1)) + else: + raw_parts.append(('text', '')) + + num_outputs: int = len(out_output_ops) + parts: list[str] = [] + for kind, data in raw_parts: + if kind == 'text': + parts.append(data) + elif kind == 'output': + parts.append(f'%{data}') + elif kind == 'input': + parts.append(f'%{num_outputs + data}') + return ''.join(parts) + + _X86_REGISTERS = frozenset({ + 'rax', 'rbx', 'rcx', 'rdx', 'rsi', 'rdi', 'rbp', 'rsp', + 'r8', 'r9', 'r10', 'r11', 'r12', 'r13', 'r14', 'r15', + 'eax', 'ebx', 'ecx', 'edx', 'esi', 'edi', 'ebp', 'esp', + 'ax', 'bx', 'cx', 'dx', 'si', 'di', 'bp', 'sp', + 'al', 'bl', 'cl', 'dl', 'ah', 'bh', 'ch', 'dh', + 'sil', 'dil', 'bpl', 'spl', + 'r8d', 'r9d', 'r10d', 'r11d', 'r12d', 'r13d', 'r14d', 'r15d', + 'r8w', 'r9w', 'r10w', 'r11w', 'r12w', 'r13w', 'r14w', 'r15w', + 'r8b', 'r9b', 'r10b', 'r11b', 'r12b', 'r13b', 'r14b', 'r15b', + }) + + def _mangle_asm_calls(self, template: str) -> str: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + sha1: str = getattr(Gen, 'module_sha1', '') + if not sha1: + return template + + def replace_call(m: re.Match[str]) -> str: + prefix: str = m.group(1) + func_name: str = m.group(2) + if func_name.lower() in self._X86_REGISTERS: + return m.group(0) + mangled: str = Gen._mangle_func_name(func_name) + if '.' in mangled: + return f'{prefix}\\22{mangled}\\22' + return f"{prefix}{mangled}" + + return re.sub(r'(call\s+)(\w+)', replace_call, template) + + def _HandleAsmWithOpLlvm(self, asm_template_arg: ast.AST, op_arg: ast.AST | None = None, input_arg: ast.AST | None = None, output_arg: ast.AST | None = None, asm_format: str = 'Intel') -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + + input_ops: list[Any] = [] + input_constraints: list[str] = [] + output_ops: list[Any] = [] + output_constraints: list[str] = [] + + if output_arg and isinstance(output_arg, (ast.List, ast.Tuple)): + for item in output_arg.elts: + if isinstance(item, ast.Call) and isinstance(item.func, ast.Attribute): + if item.func.value.id == 'c' and item.func.attr == 'AsmOut': + if len(item.args) >= 1: + target_ptr: Any = self._get_var_ptr_for_asm(item.args[0]) + if target_ptr: + output_ops.append(target_ptr) + else: + value: Any = self.HandleExprLlvm(item.args[0]) + if value: + output_ops.append(value) + constraint: str = '=r' + if len(item.args) >= 2: + base_constraint: str = self._parse_asm_descr(item.args[1]) + if base_constraint: + if base_constraint.startswith('='): + constraint = base_constraint + else: + constraint = '=' + base_constraint + output_constraints.append(constraint) + + raw_template: str + if isinstance(asm_template_arg, ast.JoinedStr): + raw_template = self._normalize_asm_template( + self._process_joined_str_template( + asm_template_arg, input_ops, input_constraints, output_ops, output_constraints + ) + ) + elif isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str): + raw_template = self._normalize_asm_template(asm_template_arg.value) + else: + raw_template = '' + + raw_template = self._mangle_asm_calls(raw_template) + asm_dialect: str | None = self._format_to_dialect(asm_format) + asm_template: str + if asm_format == 'AT&T': + asm_template = raw_template # AT&T 不需要转换 + elif asm_format == 'ARM': + asm_template = raw_template # ARM 不需要转换 + else: + asm_template = self._prepare_intel_asm(raw_template) # Intel → LLVM IR 格式 + + if input_arg and isinstance(input_arg, (ast.List, ast.Tuple)): + for item in input_arg.elts: + if isinstance(item, ast.Call) and isinstance(item.func, ast.Attribute): + if item.func.value.id == 'c' and item.func.attr == 'AsmInp': + if len(item.args) >= 1: + value: Any = self.HandleExprLlvm(item.args[0]) + if value: + input_ops.append(value) + constraint: str = 'r' + if len(item.args) >= 2: + constraint = self._parse_asm_descr(item.args[1]) + if not constraint: + constraint = 'r' + input_constraints.append(constraint) + elif isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2: + value_node: ast.AST = item.elts[0] + descr_node: ast.AST = item.elts[1] + value: Any = self.HandleExprLlvm(value_node) + if value: + input_ops.append(value) + constraint: str = self._parse_asm_descr(descr_node) + if not constraint: + constraint = 'r' + input_constraints.append(constraint) + + clobbers: list[str] = [] + if op_arg and isinstance(op_arg, (ast.List, ast.Tuple)): + for item in op_arg.elts: + clobber: str = self._parse_asm_descr(item) + if clobber: + clobbers.append(clobber) + + constraint_parts: list[str] = [] + for oc in output_constraints: + c: str = oc if oc.startswith('=') else ('=' + oc) + constraint_parts.append(c) + for ic in input_constraints: + if ic in ('d', 'edx', 'rdx'): + constraint_parts.append('{dx}') + else: + constraint_parts.append(ic) + + full_constraint: str = ",".join(constraint_parts) + + constraint_to_reg: dict[str, str] = { + 'a': 'rax', 'b': 'rbx', 'c': 'rcx', 'd': 'rdx', + 'S': 'rsi', 'D': 'rdi', + 'A': 'rax', 'U': 'r8', + } + used_regs: set[str] = set() + for ic in input_constraints: + if ic in constraint_to_reg: + used_regs.add(constraint_to_reg[ic]) + for oc in output_constraints: + base_oc: str = oc.lstrip('=').lstrip('+') + if base_oc in constraint_to_reg: + used_regs.add(constraint_to_reg[base_oc]) + + if clobbers: + has_memory: bool = 'memory' in clobbers + if has_memory and output_ops: + clobbers = [c for c in clobbers if c != 'memory'] + clobber_final: list[str] + if asm_format == 'ARM': + # ARM 不需要 x86 的 e→r 寄存器名转换 + clobber_final = [c for c in clobbers if c not in used_regs] + else: + clobber_64bit: list[str] = [] + for c in clobbers: + if c in ['eax', 'ebx', 'ecx', 'edx', 'esi', 'edi', 'ebp', 'esp']: + clobber_64bit.append(c.replace('e', 'r')) + else: + clobber_64bit.append(c) + clobber_final = [c for c in clobber_64bit if c not in used_regs] + if clobber_final: + if full_constraint: + full_constraint += "," + full_constraint += ",".join([f"~{{{c}}}" for c in clobber_final]) + + if output_ops: + single_out: bool = len(output_ops) == 1 + if single_out: + op: Any = output_ops[0] + ret_type: ir.Type + if isinstance(op, ir.AllocaInstr): + ret_type = op.type.pointee + elif isinstance(op, ir.GlobalVariable): + ret_type = op.type.pointee + elif isinstance(op, (ir.GEPInstr, ir.CastInstr)): + ret_type = op.type.pointee if isinstance(op.type, ir.PointerType) else op.type + else: + ret_type = op.type + else: + def _get_ret_type(op: Any) -> ir.Type: + if isinstance(op, (ir.AllocaInstr, ir.GlobalVariable)): + return op.type.pointee + return op.type + ret_type = ir.LiteralStructType([_get_ret_type(op) for op in output_ops]) + if isinstance(ret_type, ir.PointerType) and isinstance(ret_type.pointee, ir.IdentifiedStructType): + ret_type = ir.IntType(64) + operands: list[Any] = input_ops + param_types: list[ir.Type] = [op.type for op in operands] + func_type: ir.FunctionType = ir.FunctionType(ret_type, param_types) + inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=len(output_ops) == 0, asm_dialect=asm_dialect) + result: Any = Gen.builder.call(inline_asm, operands, name="asm_result") + if single_out: + target: Any = output_ops[0] + if isinstance(target, ir.AllocaInstr): + if target.type.pointee != result.type: + i64t: ir.IntType = ir.IntType(64) + iv: Any = Gen.builder.ptrtoint(result, i64t) + ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t)) + Gen.builder.store(iv, ptr) + else: + Gen.builder.store(result, target) + elif isinstance(target, (ir.GlobalVariable, ir.GEPInstr, ir.CastInstr)): + target_pointee: ir.Type = target.type.pointee + if target_pointee != result.type: + if isinstance(target_pointee, ir.PointerType) and isinstance(result, ir.Instruction) and result.type == ir.IntType(64): + ptr_val: Any = Gen.builder.inttoptr(result, target_pointee) + Gen.builder.store(ptr_val, target) + else: + i64t: ir.IntType = ir.IntType(64) + iv: Any = Gen.builder.ptrtoint(result, i64t) + ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t)) + Gen.builder.store(iv, ptr) + else: + Gen.builder.store(result, target) + else: + dst: Any = Gen._allocaEntry(result.type) + Gen.builder.store(result, dst) + else: + for i, out_op in enumerate(output_ops): + val: Any = Gen.builder.extract_value(result, i, name=f"asm_out_{i}") + target: Any = out_op if isinstance(out_op, ir.AllocaInstr) else Gen._allocaEntry(val.type) + if isinstance(target, ir.AllocaInstr) and target.type.pointee != val.type: + i64t: ir.IntType = ir.IntType(64) + iv: Any = Gen.builder.ptrtoint(val, i64t) + ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t)) + Gen.builder.store(iv, ptr) + else: + Gen.builder.store(val, target) + else: + operands: list[Any] = input_ops + param_types: list[ir.Type] = [op.type for op in operands] + func_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), param_types) + inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=True, asm_dialect=asm_dialect) + Gen.builder.call(inline_asm, operands, name="asm_call") + return ir.Constant(ir.IntType(32), 1) + + def _get_var_ptr_for_asm(self, node: ast.AST) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(node, ast.Name): + var_name: str = node.id + if var_name in Gen.variables and Gen.variables[var_name] is not None: + ptr: Any = Gen.variables[var_name] + if isinstance(ptr, ir.AllocaInstr): + return ptr + if isinstance(ptr, ir.GlobalVariable): + return ptr + elif isinstance(node, ast.Attribute): + obj_ptr: Any = self._get_var_ptr_for_asm(node.value) + if obj_ptr is not None: + obj_type: ir.Type = obj_ptr.type.pointee + if isinstance(obj_type, ir.PointerType) and isinstance(obj_type.pointee, ir.IdentifiedStructType): + Loaded: Any = Gen.builder.load(obj_ptr, name="asm_attr_Load") + struct_type: ir.IdentifiedStructType = obj_type.pointee + class_name: str | None = Gen._resolve_class_for_var(node.value.id) if isinstance(node.value, ast.Name) else None + if class_name: + offset: int = Gen._get_member_offset(node.attr, class_name) + gep: Any = Gen.builder.gep(Loaded, [ir.Constant(ir.IntType(32), 0), + ir.Constant(ir.IntType(32), offset)], + inbounds=True) + return gep + for idx, elem_type in enumerate(struct_type.elements): + gep: Any = Gen.builder.gep(Loaded, [ir.Constant(ir.IntType(32), 0), + ir.Constant(ir.IntType(32), idx)], + inbounds=True) + return gep + elif isinstance(obj_type, ir.IdentifiedStructType): + class_name: str | None = Gen._resolve_class_for_var(node.value.id) if isinstance(node.value, ast.Name) else None + if class_name: + offset: int = Gen._get_member_offset(node.attr, class_name) + gep: Any = Gen.builder.gep(obj_ptr, [ir.Constant(ir.IntType(32), 0), + ir.Constant(ir.IntType(32), offset)], + inbounds=True) + return gep + for idx, elem_type in enumerate(obj_type.elements): + gep: Any = Gen.builder.gep(obj_ptr, [ir.Constant(ir.IntType(32), 0), + ir.Constant(ir.IntType(32), idx)], + inbounds=True) + return gep + return None + + def _parse_asm_descr(self, expr: ast.AST) -> str: + if isinstance(expr, ast.BinOp): + left_val: str = self._parse_asm_descr(expr.left) + right_val: str = self._parse_asm_descr(expr.right) + return left_val + right_val + + elif isinstance(expr, ast.Attribute): + if isinstance(expr.value, ast.Attribute): + if (isinstance(expr.value.value, ast.Name) and + IsTModule(expr.value.value.id, self.Trans.SymbolTable) and + expr.value.attr == 'ASM_DESCR'): + AttrName: str = expr.attr + return getattr(t.ASM_DESCR, AttrName, "") + elif isinstance(expr.value, ast.Name) and IsTModule(expr.value.id, self.Trans.SymbolTable): + AttrName: str = expr.attr + return getattr(t.ASM_DESCR, AttrName, "") + + elif isinstance(expr, ast.Constant): + return expr.value + + return "" + + def _infer_expr_llvm_type(self, expr: ast.AST) -> ir.Type | None: + if isinstance(expr, ast.Name): + type_name: str = expr.id + if hasattr(t, type_name): + ctype: Any = getattr(t, type_name) + if isinstance(ctype, type) and issubclass(ctype, t.CType): + size: Any = getattr(ctype, 'Size', None) + if size is not None: + return ir.IntType(size) + if type_name == 'CFloat': + return ir.FloatType() + elif type_name == 'CDouble': + return ir.DoubleType() + elif isinstance(expr, ast.Attribute): + if isinstance(expr.value, ast.Name) and IsTModule(expr.value.id, self.Trans.SymbolTable): + return self._infer_expr_llvm_type(ast.Name(id=expr.attr, ctx=ast.Load())) + return None diff --git a/lib/core/Handles/HandlesExprAttr.py b/lib/core/Handles/HandlesExprAttr.py new file mode 100644 index 0000000..90ab2c4 --- /dev/null +++ b/lib/core/Handles/HandlesExprAttr.py @@ -0,0 +1,1379 @@ +from __future__ import annotations +from typing import Any, TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import BaseHandle, FuncMeta +import ast +import llvmlite.ir as ir + + +class ExprAttrHandle(BaseHandle): + + # ========== 辅助方法 ========== + + def _apply_bswap_on_Load(self, val: ir.Value, ClassName: str, AttrName: str) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + byte_order: str = getattr(Gen, 'class_member_byteorders', {}).get(ClassName, {}).get(AttrName, "") + return Gen._apply_bswap_if_big(val, byte_order, f"bswap_load_{AttrName}") + + def _load_arr_elem_with_bswap(self, Gen: LlvmCodeGenerator, ElemPtr: ir.Value, arr_byte_order: str) -> ir.Value: + """加载数组元素并按字节序应用 bswap""" + val: ir.Value = Gen._load(ElemPtr, name="arr_elem_load") + return Gen._apply_bswap_if_big(val, arr_byte_order, "bswap_arr_load") + + def _gep_and_Load_member(self, Gen: LlvmCodeGenerator, ObjVal: ir.Value, offset: int | None, AttrName: str, ClassName: str, ST: ir.Type | None = None) -> ir.Value | None: + """从结构体指针通过 GEP 取成员，处理 array/struct pointer/class_members 特殊情况""" + if offset is None: + return None + if isinstance(ObjVal.type, ir.PointerType): + pointee: Any = ObjVal.type.pointee + if isinstance(pointee, ir.IdentifiedStructType) and pointee.elements is not None and offset >= len(pointee.elements): + # sha1 冲突修复：pointee 字段不足时，尝试 bitcast 到 ST（更完整的版本） + if ST is not None and isinstance(ST, ir.IdentifiedStructType) and ST.elements is not None and offset < len(ST.elements): + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_gep_{ClassName}") + else: + return None + MemberPtr: Any = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=AttrName) + if isinstance(MemberPtr.type, ir.PointerType) and isinstance(MemberPtr.type.pointee, ir.ArrayType): + return MemberPtr + if isinstance(MemberPtr.type, ir.PointerType) and isinstance(MemberPtr.type.pointee, ir.IdentifiedStructType): + return MemberPtr + # 检查 pointer-to-pointer 且实际元素是 ArrayType + if isinstance(MemberPtr.type, ir.PointerType) and isinstance(MemberPtr.type.pointee, ir.PointerType): + actual_elem_type: Any = None + if ST is not None and isinstance(ST, ir.IdentifiedStructType) and ST.elements is not None and offset < len(ST.elements): + actual_elem_type = ST.elements[offset] + if isinstance(actual_elem_type, ir.ArrayType): + return Gen.builder.bitcast(MemberPtr, ir.PointerType(actual_elem_type), name=f"cast_arr_{AttrName}") + # 检查 class_members 中的 ArrayType + if ClassName in Gen.class_members: + for member_name, member_type in Gen.class_members[ClassName]: + if member_name == AttrName and isinstance(member_type, ir.ArrayType): + return Gen.builder.bitcast(MemberPtr, ir.PointerType(member_type), name=f"cast_arr_{AttrName}") + val: Any = Gen._load(MemberPtr, name=AttrName) + return self._apply_bswap_on_Load(val, ClassName, AttrName) + + def _try_Load_struct_from_stub(self, Gen: LlvmCodeGenerator, ClassName: str) -> tuple[ir.Type | None, bool]: + """尝试从 stub 加载结构体定义，返回 (struct_type, ok)""" + if ClassName not in Gen.structs: + return None, False + st: Any = Gen.structs[ClassName] + if isinstance(st, ir.IdentifiedStructType) and (st.elements is None or len(st.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(ClassName, Gen) + st = Gen.structs.get(ClassName, st) + if isinstance(st, ir.IdentifiedStructType) and (st.elements is None or len(st.elements) == 0): + return st, False + return st, True + + def _find_specialized_struct_with_body(self, Gen: LlvmCodeGenerator, opaque_struct: Any) -> tuple[ir.IdentifiedStructType, str] | None: + """查找 opaque 非特化泛型结构体的特化版本（如 GSList -> GSList[Param]）。 + 所有特化版本布局相同，可用于 GEP 偏移计算。""" + if not isinstance(opaque_struct, ir.IdentifiedStructType): + return None + base_name: str = opaque_struct.name or "" + short_name: str = base_name.split('.')[-1] if '.' in base_name else base_name + if not short_name or '[' in short_name: + return None + for CN2, ST2 in Gen.structs.items(): + if not isinstance(ST2, ir.IdentifiedStructType): + continue + if ST2 is opaque_struct: + continue + if ST2.elements is None: + continue + st2_short: str = (ST2.name or "").split('.')[-1] if '.' in (ST2.name or "") else (ST2.name or "") + if st2_short.startswith(short_name + '['): + return (ST2, CN2) + return None + + def _try_access_opaque_generic_member(self, Gen: LlvmCodeGenerator, ObjVal: ir.Value, opaque_struct: Any, Node: ast.Attribute, orig_CN: str) -> ir.Value | None: + """通过特化版本访问 opaque 非特化泛型结构体成员。""" + spec_result: Any = self._find_specialized_struct_with_body(Gen, opaque_struct) + if spec_result is None: + return None + spec_st: ir.IdentifiedStructType + spec_CN: str + spec_st, spec_CN = spec_result + CastedObj: Any = Gen.builder.bitcast(ObjVal, ir.PointerType(spec_st), name=f"cast_spec_{orig_CN}") + for lookup_CN in (spec_CN, orig_CN): + bitfield_offsets: Any = Gen.class_member_bitoffsets.get(lookup_CN, {}) + if Node.attr in bitfield_offsets: + return self._load_bitfield_member(CastedObj, lookup_CN, Node.attr) + offset: int | None = Gen._get_member_offset(Node.attr, lookup_CN) + if offset is not None: + return self._gep_and_Load_member(Gen, CastedObj, offset, Node.attr, lookup_CN, spec_st) + return None + + def _make_define_constant(self, Gen: LlvmCodeGenerator, val: int | float | str) -> ir.Value | None: + """将 CDefine 值转为 LLVM 常量""" + if isinstance(val, int): + if abs(val) > 2147483647: + return ir.Constant(ir.IntType(64), val) + return ir.Constant(ir.IntType(32), val) + elif isinstance(val, float): + return ir.Constant(ir.DoubleType(), val) + elif isinstance(val, str): + return Gen._create_string_global(val) + return None + + def _lookup_cdefine(self, Gen: LlvmCodeGenerator, VarName: str, AttrName: str) -> ir.Value | None: + """在符号表和模块中查找 CDefine 常量值""" + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + if not imported_modules: + return None + PossibleKeys: list[str] = [f"{VarName}.{AttrName}", AttrName] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + key: str = f"{mod_name}.{AttrName}" + if key not in PossibleKeys: + PossibleKeys.append(key) + for lookup_key in PossibleKeys: + SymInfo: Any = self.Trans.SymbolTable.lookup(lookup_key) + if SymInfo and SymInfo.IsDefine and SymInfo.DefineValue is not None: + return self._make_define_constant(Gen, SymInfo.DefineValue) + # 也检查 _define_constants + define_constants: Any = getattr(Gen, '_define_constants', {}) + for key in (f"{VarName}.{AttrName}", AttrName): + if key in define_constants: + return self._make_define_constant(Gen, define_constants[key]) + return None + + def _lookup_module_global(self, Gen: LlvmCodeGenerator, VarName: str, AttrName: str) -> ir.Value | None: + """在模块全局变量中查找 import 的属性""" + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + if not imported_modules: + return None + resolved_mod: str = self.Trans.SymbolTable.resolve_alias(VarName) + if VarName not in imported_modules and resolved_mod not in imported_modules: + return None + PossibleKeys: list[str] = [f"{VarName}.{AttrName}", AttrName] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + key: str = f"{mod_name}.{AttrName}" + if key not in PossibleKeys: + PossibleKeys.append(key) + for key in PossibleKeys: + if key in Gen.module.globals: + GVar: Any = Gen.module.globals[key] + if isinstance(GVar, ir.Function): + return GVar + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar + return Gen._load(GVar, name=AttrName) + # SHA1 前缀查找 + ModuleSha1Map: Any = getattr(Gen, 'ModuleSha1Map', {}) + sha1_candidates: list[str] = [VarName, resolved_mod] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + if mod_name not in sha1_candidates: + sha1_candidates.append(mod_name) + for cand in sha1_candidates: + sha1: str | None = ModuleSha1Map.get(cand) + if sha1: + prefixed: str = f"{sha1}.{AttrName}" + if prefixed in Gen.module.globals: + GVar2: Any = Gen.module.globals[prefixed] + if isinstance(GVar2, ir.Function): + return GVar2 + if isinstance(GVar2.type, ir.PointerType) and isinstance(GVar2.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar2 + return Gen._load(GVar2, name=AttrName) + if AttrName in Gen.module.globals: + gv: Any = Gen.module.globals[AttrName] + if AttrName in Gen.variables and Gen.variables[AttrName] is None: + str_gv_name: str = AttrName + '_str' + if str_gv_name in Gen.module.globals: + str_gv: Any = Gen.module.globals[str_gv_name] + return Gen.builder.gep(str_gv, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"{AttrName}_str_ptr") + return Gen.builder.load(gv, name=AttrName) + return None + + def _resolve_var_classname(self, Gen: LlvmCodeGenerator, VarName: str) -> str | None: + """根据变量名解析其结构体类名""" + ClassName: str | None = Gen.var_struct_class.get(VarName) + if not ClassName and getattr(Gen, 'global_struct_class', None): + ClassName = Gen.global_struct_class.get(VarName) + if ClassName: + Gen.var_struct_class[VarName] = ClassName + if not ClassName and VarName in Gen.module.globals: + GVar: Any = Gen.module.globals[VarName] + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found: Any = Gen.find_struct_by_pointee(GVar.type.pointee) + if found: + ClassName = found[0] + Gen.var_struct_class[VarName] = ClassName + if getattr(Gen, 'global_struct_class', None): + Gen.global_struct_class[VarName] = ClassName + if not ClassName and VarName in Gen.variables: + VarPtr: Any = Gen.variables[VarName] + if VarPtr is not None and isinstance(VarPtr.type, ir.PointerType): + pointee: Any = VarPtr.type.pointee + if isinstance(pointee, ir.PointerType): + pointee = pointee.pointee + found = Gen.find_struct_by_pointee(pointee) + if found: + ClassName = found[0] + return ClassName + + def _check_struct_match(self, Gen: LlvmCodeGenerator, ObjVal: ir.Value, ClassName: str) -> bool: + """检查 ObjVal 是否指向 ClassName 对应的结构体""" + if ClassName not in Gen.structs: + return False + if isinstance(ObjVal.type, ir.PointerType): + ST: Any = Gen.structs[ClassName] + if ObjVal.type.pointee == ST: + return True + if isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType): + # 精确名匹配（含 sha1 前缀）或短名匹配（去掉 sha1 前缀） + if ObjVal.type.pointee.name == ST.name: + return True + if Gen._extract_short_name(ObjVal.type.pointee.name) == Gen._extract_short_name(ST.name): + return True + return False + + # ========== _HandleAttributeLlvm 子方法 ========== + + def _handle_attr_self(self, Node: ast.Attribute, VarName: str) -> ir.Value | None: + """处理 self.xxx 属性访问""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + ClassName: str | None = self.Trans._CurrentCpythonObjectClass + if not ClassName: + return None + # 检查 property getter + PropKey: str = f'{ClassName}.{Node.attr}' + PropInfo: Any = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_GETTER in PropInfo.MetaList: + SelfVar: Any = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: Any = Gen._load(SelfVar, name="self") + GetterFunc: Any = Gen._get_function(PropKey) + if GetterFunc and SelfPtr: + return Gen.builder.call(GetterFunc, [SelfPtr], name=f"prop_{PropKey}") + offset: int | None = Gen._get_member_offset(Node.attr, ClassName) + SelfVar = Gen._get_var_ptr('self') + if not SelfVar: + return None + SelfPtr = Gen._load(SelfVar, name="self") + if not isinstance(SelfPtr.type, ir.PointerType): + return None + pointee: Any = SelfPtr.type.pointee + if isinstance(pointee, ir.IdentifiedStructType) and pointee.elements is None: + # 修复 Bug 4/5：NoVTable fix 清除 struct elements 后，_handle_attr_self + # 必须主动从 stub 文件重新加载 struct 定义，否则 self.Trans 等属性访问 + # 会因 pointee.elements 为 None 返回 None，导致 AnnAssign/Store 不生成 IR。 + # 与 _handle_attr_nested 的处理保持一致。 + # 策略：先用 ClassName 直接尝试加载；若失败，遍历 Gen.structs 按名字匹配 + # 找到对应的 dict key（可能是 sha1 前缀形式），再用该 key 加载。 + loaded: bool = False + for TryCN in (ClassName,): + st, ok = self._try_Load_struct_from_stub(Gen, TryCN) + if ok and isinstance(st, ir.IdentifiedStructType) and st.elements is not None: + pointee = st + loaded = True + break + if not loaded: + for CN2, ST2 in Gen.structs.items(): + if isinstance(ST2, ir.IdentifiedStructType) and ST2.name == pointee.name: + st2, ok2 = self._try_Load_struct_from_stub(Gen, CN2) + if ok2 and isinstance(st2, ir.IdentifiedStructType) and st2.elements is not None: + pointee = st2 + loaded = True + break + if not loaded: + # 最后回退：在 Gen.structs 中查找同名非 opaque 版本 + for CN2, ST2 in Gen.structs.items(): + if isinstance(ST2, ir.IdentifiedStructType) and ST2.name == pointee.name and ST2.elements is not None: + pointee = ST2 + break + # 当 pointee 元素数不足以容纳 offset 时（常见于跨模块继承：子类 struct + # 在父类 class_members 加载前生成，只有部分继承字段），从 Gen.structs + # 中查找同名且元素数更多的完整定义。 + if (isinstance(pointee, ir.IdentifiedStructType) + and offset is not None + and pointee.elements is not None + and offset >= len(pointee.elements)): + _cur_count: int = len(pointee.elements) + for CN2, ST2 in Gen.structs.items(): + if (isinstance(ST2, ir.IdentifiedStructType) + and ST2.name == pointee.name + and ST2.elements is not None + and len(ST2.elements) > _cur_count + and offset < len(ST2.elements)): + pointee = ST2 + break + if isinstance(pointee, ir.IdentifiedStructType) and pointee.elements is not None and offset is not None and offset < len(pointee.elements): + if isinstance(pointee.elements[offset], ir.VoidType): + return None + return self._gep_and_Load_member(Gen, SelfPtr, offset, Node.attr, ClassName, pointee) + DynamicVarName: str = f"self.{Node.attr}" + if DynamicVarName in Gen.variables: + return Gen._load(Gen.variables[DynamicVarName], name=Node.attr) + return None + + def _handle_attr_enum(self, Node: ast.Attribute, VarName: str) -> ir.Value | None: + """处理枚举成员访问 (VarName 是枚举类型名)""" + SymInfo: Any = self.Trans.SymbolTable.lookup(VarName) + if not SymInfo or not SymInfo.IsEnum: + return None + # 优先用带前缀的 key 查找（避免短名被同名函数/变量覆盖） + FullKey: str = f"{VarName}.{Node.attr}" + MemberInfo: Any = self.Trans.SymbolTable.lookup(FullKey) + if not MemberInfo or not MemberInfo.IsEnumMember or MemberInfo.EnumName != VarName: + # 回退到短名查找 + MemberInfo = self.Trans.SymbolTable.lookup(Node.attr) + if not MemberInfo or not MemberInfo.IsEnumMember or MemberInfo.EnumName != VarName: + # 最后遍历符号表 + for key, info in self.Trans.SymbolTable.items(): + if info.IsEnumMember and info.EnumName == VarName and key == Node.attr: + if isinstance(info.value, int): + return ir.Constant(ir.IntType(32), info.value) + return None + if isinstance(MemberInfo.value, int): + return ir.Constant(ir.IntType(32), MemberInfo.value) + return None + + def _handle_attr_name(self, Node: ast.Attribute, VarName: str) -> ir.Value | None: + """处理 x.attr 形式 (x 是普通 Name)""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + AttrName: str = Node.attr + + # __doc__ 魔法字符串：函数/结构体的 docstring + if AttrName == '__doc__': + doc_ptr: ir.Value | None = Gen._get_doc_ptr(VarName) + if doc_ptr is not None: + return doc_ptr + + # 先尝试枚举成员 + result: Any = self._handle_attr_enum(Node, VarName) + if result is not None: + return result + + # 尝试 CDefine 常量 + result = self._lookup_cdefine(Gen, VarName, AttrName) + if result is not None: + return result + + # 尝试 import 模块全局变量 + result = self._lookup_module_global(Gen, VarName, AttrName) + if result is not None: + return result + + # FakeDuck: 检查是否有影子结构体 a__meta__ (str 变量的 per-instance 字段存储) + # 当 a 是 str 变量时, a 本身是 char*, 但 a__meta__ 存储了 __data__/__mbuddy__ 等字段 + # 直接通过 a.__field__ 访问影子结构体字段 (语法糖) + _fd_meta_name: str = f"{VarName}__meta__" + if _fd_meta_name in Gen.variables and '_str' in Gen.structs: + _fd_offset = Gen._get_member_offset(AttrName, '_str') + if _fd_offset is not None: + _fd_meta_ptr: ir.Value = Gen.variables[_fd_meta_name] + _fd_field_ptr: ir.Value = Gen.builder.gep(_fd_meta_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _fd_offset)], name=f"{VarName}_{AttrName}_ptr") + return Gen._load(_fd_field_ptr, name=f"{VarName}_{AttrName}_val") + + # 解析结构体类名：若 VarName 是结构体变量，优先走结构体字段访问路径 + # 避免结构体字段名（如 stdout）与全局变量名冲突时被误解析为全局变量 + ClassName: str | None = self._resolve_var_classname(Gen, VarName) + + # 尝试 attr 直接作为全局变量（仅当 VarName 不是结构体变量时） + if not ClassName and AttrName in Gen.module.globals: + gv: Any = Gen.module.globals[AttrName] + if AttrName in Gen.variables and Gen.variables[AttrName] is None: + str_gv_name: str = AttrName + '_str' + if str_gv_name in Gen.module.globals: + str_gv: Any = Gen.module.globals[str_gv_name] + return Gen.builder.gep(str_gv, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"{AttrName}_str_ptr") + return Gen.builder.load(gv, name=AttrName) + + if not ClassName: + return None + + # 检查 property getter + PropKey: str = f'{ClassName}.{AttrName}' + PropInfo: Any = self.Trans.SymbolTable.lookup(PropKey) + if PropInfo and PropInfo.MetaList and FuncMeta.PROPERTY_GETTER in PropInfo.MetaList: + ObjVal: Any = self.HandleExprLlvm(Node.value) + GetterFunc: Any = Gen._get_function(PropKey) + if GetterFunc and ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + return Gen.builder.call(GetterFunc, [ObjVal], name=f"prop_{PropKey}") + + # 获取类型信息 + TypeInfo: Any = self.Trans.SymbolTable.lookup(ClassName) + IsUnion: bool = TypeInfo.IsUnion if TypeInfo else False + IsCenum: bool = TypeInfo.IsEnum if TypeInfo else False + IsRenum: bool = TypeInfo.IsRenum if TypeInfo else False + + # REnum 处理 + if IsRenum: + if AttrName == 'tag': + ObjVal = self.HandleExprLlvm(Node.value) + if ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + tag_ptr: Any = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="tag_ptr") + return Gen._load(tag_ptr, name="tag_val") + NestedStructName: str = f"{ClassName}_{AttrName}" + if NestedStructName in Gen.structs: + NestedStructType: Any = Gen.structs[NestedStructName] + ObjVal = self.HandleExprLlvm(Node.value) + if ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + return Gen.builder.bitcast(ObjVal, ir.PointerType(NestedStructType), name=f"cast_{NestedStructName}") + + # CEnum 处理 + if IsCenum: + for qname in (f"{ClassName}.{AttrName}", f"{ClassName}_{AttrName}"): + info: Any = self.Trans.SymbolTable.lookup(qname) + if info and info.IsEnumMember and isinstance(info.value, int): + return ir.Constant(ir.IntType(32), info.value) + for key, info in self.Trans.SymbolTable.items(): + if key == AttrName: + if info.IsEnumMember and info.EnumName == ClassName: + if isinstance(info.value, int): + return ir.Constant(ir.IntType(32), info.value) + break + + # CUnion 处理 + if IsUnion: + NestedStructName = f"{ClassName}_{AttrName}" + if NestedStructName in Gen.structs: + NestedStructType = Gen.structs[NestedStructName] + ObjVal = self.HandleExprLlvm(Node.value) + if ObjVal: + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + return Gen.builder.bitcast(ObjVal, ir.PointerType(NestedStructType), name=f"cast_{NestedStructName}") + + # 普通结构体成员访问 + ObjVal = self.HandleExprLlvm(Node.value) + if not ObjVal: + return None + # char* → struct* cast + if self._is_char_pointer(ObjVal): + if ClassName in Gen.structs: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + # 二级指针解引用 + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + + # 检查结构体匹配 + struct_match: bool = self._check_struct_match(Gen, ObjVal, ClassName) + if not struct_match and ClassName and VarName: + if VarName in Gen.var_struct_class: + CN: str = Gen.var_struct_class[VarName] + if CN != ClassName and CN in Gen.structs: + ClassName = CN + struct_match = self._check_struct_match(Gen, ObjVal, ClassName) + # 最终回退：var_struct_class 可能被前一个函数污染（如 cur 在不同函数中 + # 分别是 Line/Param/BasicBlock），通过 ObjVal 的实际 pointee 类型恢复正确类名 + if not struct_match and isinstance(ObjVal.type, ir.PointerType): + found: Any = Gen.find_struct_by_pointee(ObjVal.type.pointee) + if found: + ClassName = found[0] + struct_match = True + if VarName: + Gen.var_struct_class[VarName] = ClassName + elif isinstance(ObjVal.type.pointee, ir.IdentifiedStructType): + # sha1 冲突 fallback：find_struct_by_pointee 因 sha1 前缀不匹配失败时， + # 用短名匹配，优先选字段最多的完整版本 + pointee_short: str = Gen._extract_short_name(ObjVal.type.pointee.name) + best_elem_count: int = 0 + for CN, ST in Gen.structs.items(): + if isinstance(ST, ir.IdentifiedStructType): + st_short: str = Gen._extract_short_name(ST.name) + if st_short == pointee_short: + elem_count: int = len(ST.elements) if ST.elements else 0 + if elem_count > best_elem_count: + ClassName = CN + best_elem_count = elem_count + struct_match = True + if struct_match and VarName: + Gen.var_struct_class[VarName] = ClassName + if not struct_match: + return None + + st: Any + ok: bool + st, ok = self._try_Load_struct_from_stub(Gen, ClassName) + if not ok: + return None + + # bitfield 处理 + bitfield_offsets: Any = Gen.class_member_bitoffsets.get(ClassName, {}) + bitfields: Any = Gen.class_member_bitfields.get(ClassName, {}) + if AttrName in bitfield_offsets and bitfields.get(AttrName, 0) > 0: + return self._load_bitfield_member(ObjVal, ClassName, AttrName) + if AttrName in bitfields and bitfields[AttrName] > 0: + if ClassName not in Gen.class_member_bitoffsets: + Gen.class_member_bitoffsets[ClassName] = {} + if AttrName not in Gen.class_member_bitoffsets[ClassName]: + bo: int = 0 + for bf_name, bf_width in bitfields.items(): + if bf_name == AttrName: + break + bo += bf_width + Gen.class_member_bitoffsets[ClassName][AttrName] = bo + return self._load_bitfield_member(ObjVal, ClassName, AttrName) + + offset = Gen._get_member_offset(AttrName, ClassName) + # 检查 offset 是否超出实际结构体 + if isinstance(ObjVal.type, ir.PointerType): + actual_pointee: Any = ObjVal.type.pointee + if isinstance(actual_pointee, ir.IdentifiedStructType): + actual_elems: Any = actual_pointee.elements + if actual_elems is not None and offset is not None and offset >= len(actual_elems): + self.Trans.ImportHandler._TryLoadStructFromStub(ClassName, Gen) + actual_elems = actual_pointee.elements + if actual_elems is not None and offset >= len(actual_elems): + # sha1 冲突修复：ObjVal 的 pointee 可能是字段不全的版本（如 2 字段 Function）， + # 而 st（来自 _try_Load_struct_from_stub）可能是字段完整的版本。 + # bitcast ObjVal 到 st 的类型，使 GEP 能正确访问完整字段。 + if st is not None and isinstance(st, ir.IdentifiedStructType) and st.elements is not None and offset < len(st.elements): + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(st), name=f"cast_load_{ClassName}") + else: + return None + result = self._gep_and_Load_member(Gen, ObjVal, offset, AttrName, ClassName, st) + return result + + def _handle_attr_nested(self, Node: ast.Attribute) -> ir.Value | None: + """处理 a.b.c 嵌套属性访问""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(Node.value.value, ast.Name): + ParentVarName: str = Node.value.value.id + ParentAttrName: str = Node.value.attr + # __doc__ 魔法字符串：ClassName.method.__doc__ 或 module.ClassName.__doc__ + if Node.attr == '__doc__': + full_symbol: str = f"{ParentVarName}.{ParentAttrName}" + doc_ptr: ir.Value | None = Gen._get_doc_ptr(full_symbol) + if doc_ptr is not None: + return doc_ptr + # 回退：module.ClassName.__doc__ → 仅用 ClassName 查找 + doc_ptr = Gen._get_doc_ptr(ParentAttrName) + if doc_ptr is not None: + return doc_ptr + enum_result: Any = self._try_resolve_cross_module_enum(ParentVarName, ParentAttrName, Node.attr) + if enum_result is not None: + return enum_result + # 尝试跨模块子模块全局变量解析: a.b.c 其中 a.b 是已知子模块, c 是该模块的全局变量 + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + if imported_modules: + full_ModulePath: str = f"{ParentVarName}.{ParentAttrName}" + if full_ModulePath in imported_modules: + result: Any = self._lookup_module_global(Gen, full_ModulePath, Node.attr) + if result is not None: + return result + elif isinstance(Node.value.value, ast.Attribute) and isinstance(Node.value.value.value, ast.Name): + # 三层嵌套：module.ClassName.method.__doc__ + if Node.attr == '__doc__': + ModuleName: str = Node.value.value.value.id + ClassName2: str = Node.value.value.attr + MethodName: str = Node.value.attr + FullSymbol3: str = f"{ModuleName}.{ClassName2}.{MethodName}" + doc_ptr3: ir.Value | None = Gen._get_doc_ptr(FullSymbol3) + if doc_ptr3 is not None: + return doc_ptr3 + # 回退：ClassName.method.__doc__ + doc_ptr3 = Gen._get_doc_ptr(f"{ClassName2}.{MethodName}") + if doc_ptr3 is not None: + return doc_ptr3 + # 三层嵌套：module.submodule.EnumClass.MEMBER + ModuleAlias3: str = f"{Node.value.value.value.id}.{Node.value.value.attr}" + EnumClassName3: str = Node.value.attr + enum_result3: Any = self._try_resolve_cross_module_enum(ModuleAlias3, EnumClassName3, Node.attr) + if enum_result3 is not None: + return enum_result3 + cdefine_result: Any = self._try_resolve_nested_cdefine(Node) + if cdefine_result is not None: + return cdefine_result + ObjVal: Any = self.HandleExprLlvm(Node.value) + if not ObjVal or not isinstance(ObjVal.type, ir.PointerType): + return None + pointee: Any = ObjVal.type.pointee + + # char* → struct* cast + 成员访问 + if isinstance(pointee, ir.IntType) and pointee.width == 8: + AttrClassName: str | None = self._get_attr_class(Node.value, Gen) + if AttrClassName and AttrClassName in Gen.structs: + st: Any + ok: bool + st, ok = self._try_Load_struct_from_stub(Gen, AttrClassName) + if not isinstance(st, ir.IdentifiedStructType) or st.elements is None: + return None + CastedObj: Any = Gen.builder.bitcast(ObjVal, ir.PointerType(st), name=f"cast_{AttrClassName}") + offset: int | None = Gen._get_member_offset(Node.attr, AttrClassName) + if offset is None: + return None + if offset < len(st.elements) and isinstance(st.elements[offset], ir.VoidType): + return None + return self._gep_and_Load_member(Gen, CastedObj, offset, Node.attr, AttrClassName, st) + + # 已知结构体类型 → 直接成员访问 + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST or (isinstance(pointee, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType) and pointee.name == ST.name): + st, ok = self._try_Load_struct_from_stub(Gen, CN) + if not isinstance(st, ir.IdentifiedStructType) or st.elements is None: + spec_val: Any = self._try_access_opaque_generic_member(Gen, ObjVal, st if isinstance(st, ir.IdentifiedStructType) else pointee, Node, CN) + if spec_val is not None: + return spec_val + break + bitfield_offsets: Any = Gen.class_member_bitoffsets.get(CN, {}) + if Node.attr in bitfield_offsets: + return self._load_bitfield_member(ObjVal, CN, Node.attr) + offset = Gen._get_member_offset(Node.attr, CN) + if offset is None: + break + return self._gep_and_Load_member(Gen, ObjVal, offset, Node.attr, CN, st) + return None + + def _handle_attr_call_result(self, Node: ast.Attribute) -> ir.Value | None: + """处理 func().attr 属性访问""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + # Check if this is c.Deref(...) — we need the pointer, not the Loaded value + ObjVal: Any = None + is_cderef: bool = (isinstance(Node.value, ast.Call) and + isinstance(Node.value.func, ast.Attribute) and + isinstance(Node.value.func.value, ast.Name) and + Node.value.func.value.id == 'c' and + Node.value.func.attr == 'Deref' and + Node.value.args) + if is_cderef: + deref_arg: ast.expr = Node.value.args[0] + inner_val: Any = self.HandleExprLlvm(deref_arg) + if inner_val and isinstance(inner_val.type, ir.PointerType): + if isinstance(inner_val.type.pointee, ir.PointerType): + ObjVal = Gen._load(inner_val, name="deref_load_ptr") + else: + ObjVal = inner_val + if ObjVal is None: + ObjVal = self.HandleExprLlvm(Node.value) + if not ObjVal or not isinstance(ObjVal.type, ir.PointerType): + return None + pointee: Any = ObjVal.type.pointee + if isinstance(pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name="load_ptr_for_attr") + pointee = ObjVal.type.pointee + found: Any = Gen.find_struct_by_pointee(pointee) + if not found: + return None + CN: str + ST: Any + CN, ST = found + st: Any + ok: bool + st, ok = self._try_Load_struct_from_stub(Gen, CN) + if not isinstance(st, ir.IdentifiedStructType) or st.elements is None: + spec_val: Any = self._try_access_opaque_generic_member(Gen, ObjVal, st if isinstance(st, ir.IdentifiedStructType) else pointee, Node, CN) + if spec_val is not None: + return spec_val + return None + bitfield_offsets: Any = Gen.class_member_bitoffsets.get(CN, {}) + if Node.attr in bitfield_offsets: + return self._load_bitfield_member(ObjVal, CN, Node.attr) + offset: int | None = Gen._get_member_offset(Node.attr, CN) + if offset is None: + return None + return self._gep_and_Load_member(Gen, ObjVal, offset, Node.attr, CN, st) + + def _handle_attr_subscript_result(self, Node: ast.Attribute) -> ir.Value | None: + """处理 arr[i].attr 属性访问""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + ObjVal: Any = self.HandleExprLlvm(Node.value) + if not ObjVal or not isinstance(ObjVal.type, ir.PointerType): + return None + pointee: Any = ObjVal.type.pointee + if not isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + return None + found: Any = Gen.find_struct_by_pointee(pointee) + if not found: + return None + CN: str + ST: Any + CN, ST = found + st: Any + ok: bool + st, ok = self._try_Load_struct_from_stub(Gen, CN) + if not isinstance(st, ir.IdentifiedStructType) or st.elements is None: + spec_val: Any = self._try_access_opaque_generic_member(Gen, ObjVal, st if isinstance(st, ir.IdentifiedStructType) else pointee, Node, CN) + if spec_val is not None: + return spec_val + return None + offset: int | None = Gen._get_member_offset(Node.attr, CN) + if offset is None: + return None + return self._gep_and_Load_member(Gen, ObjVal, offset, Node.attr, CN, st) + + def _handle_attr_fallback(self, Node: ast.Attribute) -> ir.Value | None: + """兜底：对任意值尝试指针解引用后查找结构体成员""" + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + ObjVal: Any = self.HandleExprLlvm(Node.value) + if not ObjVal or not isinstance(ObjVal.type, ir.PointerType): + return None + pointee: Any = ObjVal.type.pointee + if isinstance(pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name="load_ptr_for_attr_fallback") + pointee = ObjVal.type.pointee + if not isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + return None + found: Any = Gen.find_struct_by_pointee(pointee) + if not found: + return None + CN: str + ST: Any + CN, ST = found + st: Any + ok: bool + st, ok = self._try_Load_struct_from_stub(Gen, CN) + if not isinstance(st, ir.IdentifiedStructType) or st.elements is None: + spec_val: Any = self._try_access_opaque_generic_member(Gen, ObjVal, st if isinstance(st, ir.IdentifiedStructType) else pointee, Node, CN) + if spec_val is not None: + return spec_val + return None + offset: int | None = Gen._get_member_offset(Node.attr, CN) + if offset is None: + return None + return self._gep_and_Load_member(Gen, ObjVal, offset, Node.attr, CN, st) + + # ========== 主入口 ========== + + def _HandleAttributeLlvm(self, Node: ast.Attribute) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + + if isinstance(Node.value, ast.Name): + VarName: str = Node.value.id + # 处理 t/c import 的别名重定向 + t_c_imported: Any = getattr(self.Trans, '_t_c_imported_names', {}) + if VarName != 'self' and VarName in t_c_imported: + src_module: str + src_name: str + src_module, src_name = t_c_imported[VarName] + virtual_attr: ast.Attribute = ast.Attribute( + value=ast.Name(id=src_module, ctx=ast.Load()), + attr=src_name, ctx=ast.Load() + ) + ast.copy_location(virtual_attr, Node) + return self._HandleAttributeLlvm(ast.Attribute( + value=virtual_attr, attr=Node.attr, ctx=ast.Load() + )) + if VarName == 'self': + return self._handle_attr_self(Node, VarName) + return self._handle_attr_name(Node, VarName) + + elif isinstance(Node.value, ast.Attribute): + return self._handle_attr_nested(Node) + + elif isinstance(Node.value, ast.Call): + return self._handle_attr_call_result(Node) + + elif isinstance(Node.value, ast.Subscript): + return self._handle_attr_subscript_result(Node) + + # 兜底 + return self._handle_attr_fallback(Node) + + # ========== HandleSubscript 及其辅助 ========== + + def _HandleSubscriptLlvm(self, Node: ast.Subscript) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + # 查询数组元素字节序 + arr_byte_order: str = '' + if isinstance(Node.value, ast.Name) and Node.value.id in Gen.local_var_byteorders: + arr_byte_order = Gen.local_var_byteorders[Node.value.id] + ClassName: str | None = self.Trans.ExprHandler._get_var_class(Node.value, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__getitem__'): + obj_val: Any = self.HandleExprLlvm(Node.value) + idx_val: Any = self.HandleExprLlvm(Node.slice) + if obj_val and idx_val: + # 处理 obj_val 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + # 需要先 inttoptr 转换为 i8*，再 bitcast 到目标 struct 指针 + if isinstance(obj_val.type, ir.IntType) and obj_val.type.width == 8: + obj_val = Gen.builder.inttoptr(obj_val, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + if isinstance(idx_val.type, ir.IntType) and idx_val.type.width < 64: + idx_val = Gen.builder.zext(idx_val, ir.IntType(64), name="zext_idx") + _getitem_func: Any = Gen._get_function(f'{ClassName}.__getitem__') + # 处理泛型特化类型不匹配：obj_val 可能是通用类型，而函数期望特化类型 + if _getitem_func.ftype.args and _getitem_func.ftype.args[0] != obj_val.type: + _expected_type = _getitem_func.ftype.args[0] + if isinstance(_expected_type, ir.PointerType) and isinstance(obj_val.type, ir.PointerType): + obj_val = Gen.builder.bitcast(obj_val, _expected_type, name=f"cast_{ClassName}_getitem") + result: Any = Gen.builder.call(_getitem_func, [obj_val, idx_val], name=f"call_{ClassName}.__getitem__") + return result + ValueVal: Any = self.HandleExprLlvm(Node.value) + if not ValueVal: + return None + IndexVal: Any = self.HandleExprLlvm(Node.slice) + if not IndexVal: + return None + if isinstance(IndexVal.type, ir.PointerType) and isinstance(IndexVal.type.pointee, ir.IntType) and IndexVal.type.pointee.width == 8: + Loaded_byte: Any = Gen._load(IndexVal, name="Load_char_idx") + IndexVal = Gen.builder.zext(Loaded_byte, ir.IntType(32), name="char_to_int_idx") + if isinstance(ValueVal.type, ir.IntType): + var_ptr: Any = self._get_int_ptr(Node.value) + if var_ptr: + i8_ptr: Any = Gen.builder.bitcast(var_ptr, ir.PointerType(ir.IntType(8)), name="int_to_i8ptr") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_int_index") + ptr: Any = Gen.builder.gep(i8_ptr, [IndexVal], name="int_subscript_ptr") + return Gen._load(ptr, name="int_subscript_val") + return None + # 当变量本身是二级指针（如 elem_ptr: T | t.CPtr = AST**）， + # 但 HandleExprLlvm 已自动 load 返回 AST*（一级指针）， + # 需要检查 Gen.variables 中的原始 alloca 类型， + # 若是二级指针且存储的是结构体指针，应 bitcast ValueVal 为 AST** 后 GEP 再 load， + # 避免 GEP AST 结构体数组返回错误的 AST*（指向存储位置而非实际对象） + if (isinstance(Node.value, ast.Name) + and isinstance(ValueVal.type, ir.PointerType) + and isinstance(ValueVal.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType))): + VarName: str = Node.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarAlloca: ir.Value = Gen.variables[VarName] + if (isinstance(VarAlloca.type, ir.PointerType) + and isinstance(VarAlloca.type.pointee, ir.PointerType)): + # 原始 alloca 是二级指针，ValueVal 是已 load 的一级结构体指针 + # 该指针指向数据缓冲区，缓冲区存储的是 AST* 指针数组， + # 需 bitcast ValueVal 为 AST** 后 GEP 指针数组再 load 出 AST* + PtrToStruct: ir.Type = ValueVal.type.pointee + BaseAsPP: ir.Value = Gen.builder.bitcast(ValueVal, + ir.PointerType(ir.PointerType(PtrToStruct)), name="cast_base_pp_load") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index_load") + ElemPtrLoad: ir.Value = Gen.builder.gep(BaseAsPP, [IndexVal], name="subscript_pp_load") + return Gen._load(ElemPtrLoad, name="ptr_elem_load") + if isinstance(ValueVal.type, ir.PointerType) and isinstance(ValueVal.type.pointee, ir.PointerType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index") + ElemPtr: Any = Gen.builder.gep(ValueVal, [IndexVal], name="subscript") + if isinstance(ValueVal.type.pointee.pointee, ir.IntType) and ValueVal.type.pointee.pointee.width == 8: + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + if isinstance(ValueVal.type, ir.PointerType): + pointee: Any = ValueVal.type.pointee + if isinstance(pointee, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(ValueVal, [zero, IndexVal], name="subscript") + elem_type: Any = pointee.element + if isinstance(elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + else: + if isinstance(pointee, ir.IntType) and pointee.width == 8: + elem_class: str | None = self._infer_element_struct_class(Node.value, Gen) + if elem_class and elem_class in Gen.structs: + CastedPtr: Any = Gen.builder.bitcast(ValueVal, ir.PointerType(Gen.structs[elem_class]), name=f"cast_{elem_class}_arr") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_arr_index") + ElemPtr = Gen.builder.gep(CastedPtr, [IndexVal], name="subscript") + if isinstance(pointee, ir.IntType): + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + return ElemPtr + # 指针到 str/bytes 的变量 (BinOp 注解如 str | t.CPtr) + # elem_ptr 是 i8* 但实际指向 i8* (字符串指针), elem_ptr[0] 应加载 8 字节 + if isinstance(Node.value, ast.Name): + _ptr_elem_flag: bool = (getattr(Gen, 'var_ptr_element', {}).get(Node.value.id, False) + or getattr(Gen, 'global_var_ptr_element', {}).get(Node.value.id, False)) + if _ptr_elem_flag: + CastedPtrPtr: Any = Gen.builder.bitcast(ValueVal, ir.PointerType(ir.PointerType(ir.IntType(8))), name="cast_pp_str") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_pp_str_index") + ElemPtr = Gen.builder.gep(CastedPtrPtr, [IndexVal], name="subscript") + return Gen._load(ElemPtr, name="str_ptr_load") + arr_type: Any = self._infer_array_member_type(Node.value, Gen) + if arr_type and isinstance(arr_type, ir.ArrayType): + arr_ptr: Any = Gen.builder.bitcast(ValueVal, ir.PointerType(arr_type), name=f"cast_arr_subscript") + zero = ir.Constant(ir.IntType(32), 0) + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(arr_ptr, [zero, IndexVal], name="subscript") + arr_elem_type: Any = arr_type.element + if isinstance(arr_elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + ElemPtr = Gen.builder.gep(ValueVal, [IndexVal], name="subscript") + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + elif isinstance(ValueVal.type, ir.ArrayType): + zero = ir.Constant(ir.IntType(32), 0) + if isinstance(Node.value, ast.Name): + VarName: str = Node.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: Any = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(VarPtr, [zero, IndexVal], name="subscript") + var_arr_elem_type: Any = VarPtr.type.pointee.element + if isinstance(var_arr_elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + if isinstance(Node.value, ast.Attribute): + AttrPtr: Any = self._get_attr_ptr(Node.value) + if AttrPtr and isinstance(AttrPtr.type, ir.PointerType) and isinstance(AttrPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(AttrPtr, [zero, IndexVal], name="subscript") + attr_arr_elem_type: Any = AttrPtr.type.pointee.element + if isinstance(attr_arr_elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + if isinstance(Node.value, ast.Subscript): + InnerPtr: Any = self.HandleSubscriptPtrLlvm(Node.value) + if InnerPtr and isinstance(InnerPtr.type, ir.PointerType): + inner_pointee: Any = InnerPtr.type.pointee + if isinstance(inner_pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(InnerPtr, [zero, IndexVal], name="subscript") + inner_elem_type: Any = inner_pointee.element + if isinstance(inner_elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + arr_alloc: Any = Gen._allocaEntry(ValueVal.type, name="arr_subscript_tmp") + Gen._store(ValueVal, arr_alloc) + ElemPtr = Gen.builder.gep(arr_alloc, [zero, IndexVal], name="subscript") + alloc_arr_elem_type: Any = ValueVal.type.element + if isinstance(alloc_arr_elem_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + return ElemPtr + return self._load_arr_elem_with_bswap(Gen, ElemPtr, arr_byte_order) + return None + + def HandleSubscriptPtrLlvm(self, Node: ast.Subscript) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + ValueVal: Any = self.HandleExprLlvm(Node.value) + if not ValueVal: + return None + IndexVal: Any = self.HandleExprLlvm(Node.slice) + if not IndexVal: + return None + if isinstance(IndexVal.type, ir.PointerType) and isinstance(IndexVal.type.pointee, ir.IntType) and IndexVal.type.pointee.width == 8: + Loaded_byte: Any = Gen._load(IndexVal, name="Load_char_idx_ptr") + IndexVal = Gen.builder.zext(Loaded_byte, ir.IntType(32), name="char_to_int_idx_ptr") + if isinstance(ValueVal.type, ir.PointerType): + pointee: Any = ValueVal.type.pointee + if isinstance(pointee, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr: Any = Gen.builder.gep(ValueVal, [zero, IndexVal], name="subscript_ptr") + return ElemPtr + elif isinstance(pointee, ir.PointerType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index_ptr") + ElemPtr = Gen.builder.gep(ValueVal, [IndexVal], name="subscript_ptr") + return ElemPtr + else: + # 指针到 str/bytes 的变量 (BinOp 注解如 str | t.CPtr, bytes | t.CPtr) + # elem_ptr 是 i8* 但实际指向 i8* (指针数组), 应按 8 字节步长 GEP + if isinstance(Node.value, ast.Name): + _ptr_elem_flag: bool = (getattr(Gen, 'var_ptr_element', {}).get(Node.value.id, False) + or getattr(Gen, 'global_var_ptr_element', {}).get(Node.value.id, False)) + if _ptr_elem_flag: + CastedPtrPtr: Any = Gen.builder.bitcast(ValueVal, ir.PointerType(ir.PointerType(ir.IntType(8))), name="cast_pp_str_ptr") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_pp_str_index_ptr") + ElemPtr = Gen.builder.gep(CastedPtrPtr, [IndexVal], name="subscript_ptr") + return ElemPtr + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index") + ElemPtr = Gen.builder.gep(ValueVal, [IndexVal], name="subscript_ptr") + return ElemPtr + elif isinstance(ValueVal.type, ir.IntType): + var_ptr: Any = self._get_int_ptr(Node.value) + if var_ptr: + i8_ptr: Any = Gen.builder.bitcast(var_ptr, ir.PointerType(ir.IntType(8)), name="int_to_i8ptr") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_int_index") + ElemPtr = Gen.builder.gep(i8_ptr, [IndexVal], name="int_subscript_ptr") + return ElemPtr + elif isinstance(ValueVal.type, ir.ArrayType): + zero = ir.Constant(ir.IntType(32), 0) + if isinstance(Node.value, ast.Name): + VarName: str = Node.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: Any = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(VarPtr, [zero, IndexVal], name="subscript_ptr") + return ElemPtr + if isinstance(Node.value, ast.Subscript): + InnerPtr: Any = self.HandleSubscriptPtrLlvm(Node.value) + if InnerPtr: + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(InnerPtr, [zero, IndexVal], name="subscript_ptr") + return ElemPtr + if isinstance(Node.value, ast.Attribute): + AttrPtr: Any = self._get_attr_ptr(Node.value) + if AttrPtr and isinstance(AttrPtr.type, ir.PointerType) and isinstance(AttrPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(AttrPtr, [zero, IndexVal], name="subscript_ptr") + return ElemPtr + arr_alloc: Any = Gen._allocaEntry(ValueVal.type, name="arr_subscript_tmp") + Gen._store(ValueVal, arr_alloc) + ElemPtr = Gen.builder.gep(arr_alloc, [zero, IndexVal], name="subscript_ptr") + return ElemPtr + return None + + # ========== _get_attr_ptr / _get_int_ptr / _get_attr_class ========== + + def _get_attr_ptr(self, Node: ast.Attribute) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if not isinstance(Node, ast.Attribute): + return None + if isinstance(Node.value, ast.Name): + VarName: str = Node.value.id + if VarName == 'self': + ClassName: str | None = self.Trans._CurrentCpythonObjectClass + if ClassName: + offset: int | None = Gen._get_member_offset(Node.attr, ClassName) + SelfVar: Any = Gen._get_var_ptr('self') + if SelfVar: + SelfPtr: Any = Gen._load(SelfVar, name="self") + if isinstance(SelfPtr.type, ir.PointerType): + MemberPtr: Any = Gen.builder.gep(SelfPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=Node.attr) + return MemberPtr + ClassName = Gen.var_struct_class.get(VarName) + if not ClassName and getattr(Gen, 'global_struct_class', None): + ClassName = Gen.global_struct_class.get(VarName) + if not ClassName: + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + resolved_mod: str = self.Trans.SymbolTable.resolve_alias(VarName) + if imported_modules and (VarName in imported_modules or resolved_mod in imported_modules): + AttrName: str = Node.attr + ModuleSha1Map: Any = getattr(Gen, 'ModuleSha1Map', {}) + candidates: list[str] = [VarName, resolved_mod] + for mod_name in imported_modules: + if mod_name.endswith(f".{VarName}") or mod_name == VarName: + if mod_name not in candidates: + candidates.append(mod_name) + for cand in candidates: + sha1: str | None = ModuleSha1Map.get(cand) + if sha1: + prefixed: str = f"{sha1}.{AttrName}" + if prefixed in Gen.module.globals: + return Gen.module.globals[prefixed] + if AttrName in Gen.module.globals: + gv: Any = Gen.module.globals[AttrName] + if isinstance(gv, ir.GlobalVariable): + return gv + if ClassName and ClassName in Gen.structs: + VarPtr: Any = Gen.variables.get(VarName) + if VarPtr: + ObjVal: Any = VarPtr + if isinstance(ObjVal.type, ir.PointerType) and isinstance(ObjVal.type.pointee, ir.PointerType): + ObjVal = Gen._load(ObjVal, name=f"Load_{VarName}") + if isinstance(ObjVal.type, ir.PointerType) and ObjVal.type.pointee == Gen.structs[ClassName]: + ST: Any = Gen.structs[ClassName] + if isinstance(ST, ir.IdentifiedStructType) and (ST.elements is None or len(ST.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(ClassName, Gen) + ST = Gen.structs.get(ClassName, ST) + if ST.elements is not None and isinstance(ObjVal.type.pointee, ir.IdentifiedStructType) and ObjVal.type.pointee.elements is None: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{ClassName}") + offset = Gen._get_member_offset(Node.attr, ClassName) + if offset is not None and ST.elements is not None: + MemberPtr = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=Node.attr) + return MemberPtr + elif isinstance(Node.value, ast.Attribute): + InnerPtr: Any = self._get_attr_ptr(Node.value) + if InnerPtr and isinstance(InnerPtr.type, ir.PointerType): + pointee: Any = InnerPtr.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found: Any = Gen.find_struct_by_pointee(pointee) + if found: + CN: str + ST: Any + CN, ST = found + if isinstance(ST, ir.IdentifiedStructType) and (ST.elements is None or len(ST.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(CN, Gen) + ST = Gen.structs.get(CN, ST) + if ST.elements is not None and isinstance(InnerPtr.type, ir.PointerType) and isinstance(InnerPtr.type.pointee, ir.IdentifiedStructType) and InnerPtr.type.pointee.elements is None: + InnerPtr = Gen.builder.bitcast(InnerPtr, ir.PointerType(ST), name=f"cast_{CN}") + offset = Gen._get_member_offset(Node.attr, CN) + if offset is not None and ST.elements is not None: + MemberPtr = Gen.builder.gep(InnerPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=Node.attr) + return MemberPtr + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + AttrClassName: str | None = self._get_attr_class(Node.value, Gen) + if AttrClassName and AttrClassName in Gen.structs: + CastedObj: Any = Gen.builder.bitcast(InnerPtr, ir.PointerType(Gen.structs[AttrClassName]), name=f"cast_{AttrClassName}") + offset = Gen._get_member_offset(Node.attr, AttrClassName) + if offset is not None: + MemberPtr = Gen.builder.gep(CastedObj, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=Node.attr) + return MemberPtr + elif isinstance(Node.value, ast.Subscript): + SubPtr: Any = self.HandleSubscriptPtrLlvm(Node.value) + if SubPtr and isinstance(SubPtr.type, ir.PointerType): + pointee = SubPtr.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found = Gen.find_struct_by_pointee(pointee) + if found: + CN, ST = found + offset = Gen._get_member_offset(Node.attr, CN) + if offset is not None: + MemberPtr = Gen.builder.gep(SubPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=Node.attr) + return MemberPtr + return None + + def _HandleSliceLlvm(self, base_val: ir.Value, slice_node: ast.Slice | ast.Index, base_ast: ast.AST) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + lower_val: Any = None + upper_val: Any = None + step_val: Any = None + if isinstance(slice_node, ast.Slice): + if slice_node.lower: + lower_val = self.HandleExprLlvm(slice_node.lower) + if slice_node.upper: + upper_val = self.HandleExprLlvm(slice_node.upper) + if slice_node.step: + step_val = self.HandleExprLlvm(slice_node.step) + if isinstance(slice_node, ast.Index): + return self.HandleExprLlvm(slice_node.value) + return None + + def _infer_element_struct_class(self, node: ast.AST, Gen: LlvmCodeGenerator) -> str | None: + if isinstance(node, ast.Attribute): + attr_name: str = node.attr + if isinstance(node.value, ast.Name) and node.value.id == 'self': + current_class: str | None = self.Trans._CurrentCpythonObjectClass + if current_class: + if current_class in Gen.class_member_element_class: + elem_class: str | None = Gen.class_member_element_class[current_class].get(attr_name) + if elem_class and elem_class in Gen.structs: + return elem_class + if current_class in Gen.class_members: + for member_name, member_type in Gen.class_members[current_class]: + if member_name == attr_name: + if isinstance(member_type, ir.IdentifiedStructType): + found: Any = Gen.find_struct_by_pointee(member_type) + if found: + return found[0] + break + class_name: str | None = Gen.var_struct_class.get(attr_name) + if class_name: + return class_name + elif isinstance(node, ast.Name): + var_name: str = node.id + class_name = Gen.var_struct_class.get(var_name) + if class_name: + return class_name + return None + + def _get_int_ptr(self, target: ast.AST) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(target, ast.Name): + VarName: str = target.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr: Any = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType): + pointee: Any = VarPtr.type.pointee + if isinstance(pointee, ir.ArrayType): + return Gen.builder.gep(VarPtr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"arr_start_{VarName}") + return VarPtr + elif isinstance(target, ast.Attribute): + if isinstance(target.value, ast.Name): + obj_name: str = target.value.id + if obj_name in Gen.variables and Gen.variables[obj_name] is not None: + obj_ptr: Any = Gen.variables[obj_name] + if isinstance(obj_ptr.type, ir.PointerType): + pointee = obj_ptr.type.pointee + if isinstance(pointee, ir.PointerType): + obj_ptr = Gen._load(obj_ptr, name=f"Load_{obj_name}") + pointee = obj_ptr.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + StructName: str | None = None + if isinstance(pointee, ir.IdentifiedStructType): + StructName = pointee.name + if not StructName: + found: Any = Gen.find_struct_by_pointee(pointee) + if found: + StructName = found[0] + if StructName and StructName in Gen.structs: + offset: int | None = Gen._get_member_offset(target.attr, StructName) + if offset is not None: + return Gen.builder.gep(obj_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=f"{target.attr}_ptr") + obj_val: Any = self.HandleExprLlvm(target.value) + if obj_val and isinstance(obj_val.type, ir.PointerType): + if isinstance(obj_val.type.pointee, ir.PointerType): + obj_val = Gen._load(obj_val, name="Load_attr_ptr") + pointee = obj_val.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + StructName = None + if isinstance(pointee, ir.IdentifiedStructType): + StructName = pointee.name + if not StructName: + found = Gen.find_struct_by_pointee(pointee) + if found: + StructName = found[0] + if StructName and StructName in Gen.structs: + offset = Gen._get_member_offset(target.attr, StructName) + if offset is not None: + return Gen.builder.gep(obj_val, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=f"{target.attr}_ptr") + return None + + def _resolve_struct_class_for_attr(self, Node: ast.AST, Gen: LlvmCodeGenerator) -> str | None: + if isinstance(Node, ast.Name): + VarName: str = Node.id + if VarName in Gen.var_struct_class: + return Gen.var_struct_class[VarName] + elif isinstance(Node, ast.Attribute): + ClassName: str | None = self._resolve_struct_class_for_attr(Node.value, Gen) + if ClassName and ClassName in Gen.class_members: + for member_name, _ in Gen.class_members[ClassName]: + if member_name == Node.attr: + return ClassName + return None + + def _infer_array_member_type(self, Node: ast.AST, Gen: LlvmCodeGenerator) -> ir.ArrayType | None: + if isinstance(Node, ast.Attribute): + parent_class: str | None = self._get_attr_class(Node, Gen) + if parent_class and parent_class in Gen.class_members: + for m_name, m_type in Gen.class_members[parent_class]: + if m_name == Node.attr and isinstance(m_type, ir.ArrayType): + return m_type + if parent_class and parent_class in Gen.structs: + st: Any = Gen.structs[parent_class] + if isinstance(st, ir.IdentifiedStructType) and st.elements is not None: + offset: int | None = Gen._get_member_offset(Node.attr, parent_class) + if offset is not None and offset < len(st.elements): + elem: Any = st.elements[offset] + if isinstance(elem, ir.ArrayType): + return elem + return None + + def _build_attr_path(self, node: ast.AST) -> list[str]: + parts: list[str] = [] + current: ast.AST = node + while isinstance(current, ast.Attribute): + parts.append(current.attr) + current = current.value + if isinstance(current, ast.Name): + parts.append(current.id) + parts.reverse() + return parts + + def _try_resolve_nested_cdefine(self, Node: ast.Attribute) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + if not imported_modules: + return None + parts: list[str] = self._build_attr_path(Node) + if len(parts) < 2: + return None + attr_name: str = parts[-1] + module_parts: list[str] = parts[:-1] + possible_keys: list[str] = [attr_name] + full_path: str = '.'.join(parts) + possible_keys.append(full_path) + if module_parts: + ModulePath: str = '.'.join(module_parts) + resolved_first: str = self.Trans.SymbolTable.resolve_alias(module_parts[0]) + if resolved_first != module_parts[0]: + resolved_parts: list[str] = [resolved_first] + module_parts[1:] + resolved_path: str = '.'.join(resolved_parts) + possible_keys.append(f"{resolved_path}.{attr_name}") + for mod_name in imported_modules: + if mod_name.endswith('.' + ModulePath) or mod_name == ModulePath: + key: str = f"{mod_name}.{attr_name}" + if key not in possible_keys: + possible_keys.append(key) + if module_parts and (mod_name.endswith('.' + module_parts[-1]) or mod_name == module_parts[-1]): + key = f"{mod_name}.{attr_name}" + if key not in possible_keys: + possible_keys.append(key) + for lookup_key in possible_keys: + SymInfo: Any = self.Trans.SymbolTable.lookup(lookup_key) + if SymInfo and SymInfo.IsDefine and SymInfo.DefineValue is not None: + val: Any = SymInfo.DefineValue + if isinstance(val, int): + if val > 0x7FFFFFFF or val < -0x80000000: + return ir.Constant(ir.IntType(64), val & 0xFFFFFFFFFFFFFFFF) + return ir.Constant(ir.IntType(32), val) + elif isinstance(val, float): + return ir.Constant(ir.FloatType(), val) + return None + + def _try_resolve_cross_module_enum(self, module_alias: str, enum_class_name: str, member_name: str) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + imported_modules: Any = getattr(self.Trans, '_ImportedModules', None) + if not imported_modules: + return None + resolved_module: str = self.Trans.SymbolTable.resolve_alias(module_alias) + if resolved_module not in imported_modules and module_alias not in imported_modules: + return None + enum_keys: list[str] = [enum_class_name, f"{resolved_module}.{enum_class_name}"] + for enum_key in enum_keys: + SymInfo: Any = self.Trans.SymbolTable.lookup(enum_key) + if SymInfo: + if SymInfo.IsEnum: + for qname in (f"{enum_key}.{member_name}", f"{enum_key}_{member_name}"): + info: Any = self.Trans.SymbolTable.lookup(qname) + if info and info.IsEnumMember and isinstance(info.value, int): + return ir.Constant(ir.IntType(32), info.value) + for key, info in self.Trans.SymbolTable.items(): + if info.IsEnumMember and info.EnumName == enum_key and key == member_name: + if isinstance(info.value, int): + return ir.Constant(ir.IntType(32), info.value) + for key, info in self.Trans.SymbolTable.items(): + if info.IsEnumMember and key == member_name: + if isinstance(info.value, int): + for enum_key in enum_keys: + if info.EnumName == enum_key: + return ir.Constant(ir.IntType(32), info.value) + return None + + def _get_attr_class(self, Node: ast.AST, Gen: LlvmCodeGenerator) -> str | None: + if isinstance(Node, ast.Attribute): + # 注: 不再硬编码 ZLIB_VERSION / C_OK，它们应从符号表查找 + if isinstance(Node.value, ast.Name) and Node.value.id == 'self': + ClassName: str | None = self.Trans._CurrentCpythonObjectClass + if ClassName and ClassName in Gen.class_member_element_class: + return Gen.class_member_element_class[ClassName].get(Node.attr) + if isinstance(Node.value, ast.Name): + VarName: str = Node.value.id + if VarName in Gen.var_struct_class: + ParentClass: str = Gen.var_struct_class[VarName] + if ParentClass in Gen.class_member_element_class: + return Gen.class_member_element_class[ParentClass].get(Node.attr) + if isinstance(Node.value, ast.Attribute): + ParentClass = self._get_attr_class(Node.value, Gen) + if ParentClass and ParentClass in Gen.class_member_element_class: + return Gen.class_member_element_class[ParentClass].get(Node.attr) + return None + + def _get_llvm_member_offset(self, field_name: str, ClassName: str, Gen: LlvmCodeGenerator) -> int | None: + return Gen._get_member_offset(field_name, ClassName) \ No newline at end of file diff --git a/lib/core/Handles/HandlesExprBuiltin.py b/lib/core/Handles/HandlesExprBuiltin.py new file mode 100644 index 0000000..9125f36 --- /dev/null +++ b/lib/core/Handles/HandlesExprBuiltin.py @@ -0,0 +1,988 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +import ast +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import IsTModule, ExtractTypeNameFromBinOp + + +class ExprBuiltinHandle(BaseHandle): + _C_LIB_FUNCS: dict[str, object] = { + 'strlen': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'strlength': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'memset': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(32), ir.IntType(64)]), + 'memcpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'memmove': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'memcmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'strncmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'strncpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcat': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'puts': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8))]), + 'atoi': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8))]), + 'atol': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'abs': lambda: (ir.IntType(32), [ir.IntType(32)]), + 'labs': lambda: (ir.IntType(64), [ir.IntType(64)]), + } + + def _HandleBuiltinCallLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not isinstance(Node.func, ast.Name): + return None + FuncName: str = Node.func.id + + if FuncName == 'print': + return self._HandlePrintLlvm(Node) + elif FuncName == 'len': + if Node.args: + return self._HandleLenLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'sizeof': + if Node.args: + return self._HandleSizeofLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'abs': + if Node.args: + return self._HandleAbsLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'arg': + return self._HandleArgLlvm(Node.args) + elif FuncName == 'bool': + if Node.args: + Val = self.HandleExprLlvm(Node.args[0]) + if Val: + if isinstance(Val.type, ir.IntType): + if Val.type.width == 1: + return Val + return Gen.builder.icmp_signed('!=', Val, Gen._ZeroConst(Val.type), name="bool_result") + elif isinstance(Val.type, ir.PointerType): + return Gen.builder.icmp_signed('!=', Val, Gen._ZeroConst(Val.type), name="bool_result") + return Val + return ir.Constant(ir.IntType(1), 0) + elif FuncName == 'int': + if Node.args: + Val = self.HandleExprLlvm(Node.args[0]) + if Val: + if isinstance(Val.type, ir.IntType): + if Val.type.width < 32: + return Gen.builder.zext(Val, ir.IntType(32), name="cast_int") + elif Val.type.width > 32: + return Gen.builder.trunc(Val, ir.IntType(32), name="trunc_int") + return Val + elif isinstance(Val.type, ir.PointerType): + if isinstance(Val.type.pointee, ir.IntType) and Val.type.pointee.width == 8: + result = self._HandleAtoiLlvm(Val) + if result: + return result + elif isinstance(Val.type, (ir.FloatType, ir.DoubleType)): + return Gen.builder.fptosi(Val, ir.IntType(32), name="float_to_int") + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'float' or FuncName == 'double': + if Node.args: + Val = self.HandleExprLlvm(Node.args[0]) + if Val: + target_type = ir.DoubleType() if FuncName == 'double' else ir.FloatType() + if isinstance(Val.type, (ir.FloatType, ir.DoubleType)): + if Val.type != target_type: + if isinstance(target_type, ir.DoubleType): + return Gen.builder.fpext(Val, target_type, name="fpext_double") + return Gen.builder.fptrunc(Val, target_type, name="fptrunc_float") + return Val + elif isinstance(Val.type, ir.IntType): + if Val.type.width == 1: + return Gen.builder.uitofp(Val, target_type, name="uitofp") + return Gen.builder.sitofp(Val, target_type, name="sitofp") + return ir.Constant(ir.DoubleType() if FuncName == 'double' else ir.FloatType(), 0.0) + elif FuncName == 'min': + if len(Node.args) >= 2: + a = self.HandleExprLlvm(Node.args[0]) + b = self.HandleExprLlvm(Node.args[1]) + if a and b: + is_float = isinstance(a.type, (ir.FloatType, ir.DoubleType)) or isinstance(b.type, (ir.FloatType, ir.DoubleType)) + if is_float: + fa = Gen._to_float(a, ir.FloatType()) if isinstance(a.type, (ir.FloatType, ir.DoubleType)) else Gen.builder.sitofp(a, ir.FloatType(), name="min_int2float") + fb = Gen._to_float(b, ir.FloatType()) if isinstance(b.type, (ir.FloatType, ir.DoubleType)) else Gen.builder.sitofp(b, ir.FloatType(), name="min_int2float") + cmp = Gen.builder.fcmp_ordered('olt', fa, fb, name="min_cmp") + return Gen.builder.select(cmp, fa, fb, name="min_val") + else: + cmp = Gen.builder.icmp_signed('<', a, b, name="min_cmp") + return Gen.builder.select(cmp, a, b, name="min_val") + elif len(Node.args) == 1: + return self.HandleExprLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'max': + if len(Node.args) >= 2: + a = self.HandleExprLlvm(Node.args[0]) + b = self.HandleExprLlvm(Node.args[1]) + if a and b: + is_float = isinstance(a.type, (ir.FloatType, ir.DoubleType)) or isinstance(b.type, (ir.FloatType, ir.DoubleType)) + if is_float: + fa = Gen._to_float(a, ir.FloatType()) if isinstance(a.type, (ir.FloatType, ir.DoubleType)) else Gen.builder.sitofp(a, ir.FloatType(), name="max_int2float") + fb = Gen._to_float(b, ir.FloatType()) if isinstance(b.type, (ir.FloatType, ir.DoubleType)) else Gen.builder.sitofp(b, ir.FloatType(), name="max_int2float") + cmp = Gen.builder.fcmp_ordered('ogt', fa, fb, name="max_cmp") + return Gen.builder.select(cmp, fa, fb, name="max_val") + else: + cmp = Gen.builder.icmp_signed('>', a, b, name="max_cmp") + return Gen.builder.select(cmp, a, b, name="max_val") + elif len(Node.args) == 1: + return self.HandleExprLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'str': + if Node.args: + arg_node = Node.args[0] + arg_val = self.HandleExprLlvm(arg_node) + if arg_val: + if isinstance(arg_val.type, ir.PointerType): + ClassName = self.Trans.ExprHandler._get_var_class(arg_node, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__str__'): + if isinstance(arg_val.type, ir.PointerType) and isinstance(arg_val.type.pointee, ir.IntType) and arg_val.type.pointee.width == 8: + if ClassName in Gen.structs: + arg_val = Gen.builder.bitcast(arg_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}_str") + str_val = Gen.builder.call(Gen._get_function(f'{ClassName}.__str__'), [arg_val], name=f"call_{ClassName}.__str__") + return str_val + return arg_val + return self._EmitStringFromValue(arg_val) + return ir.Constant(ir.IntType(8).as_pointer(), None) + elif FuncName == 'bytes': + # bytes() — 类型转换为 i8* 指针 (等同于 t.CPtr / str) + # bytes(int_addr) → inttoptr; bytes(ptr) → bitcast + if Node.args: + arg_val = self.HandleExprLlvm(Node.args[0]) + if arg_val: + if isinstance(arg_val.type, ir.PointerType): + if isinstance(arg_val.type.pointee, ir.IntType) and arg_val.type.pointee.width == 8: + return arg_val + return Gen.builder.bitcast(arg_val, ir.IntType(8).as_pointer(), name="bytes_cast") + if isinstance(arg_val.type, ir.IntType): + return Gen.builder.inttoptr(arg_val, ir.IntType(8).as_pointer(), name="bytes_int2ptr") + return ir.Constant(ir.IntType(8).as_pointer(), None) + elif FuncName == 'input': + return self._HandleInputLlvm(Node) + elif FuncName == 'ord': + if Node.args: + return self._HandleOrdLlvm(Node.args[0]) + return ir.Constant(ir.IntType(32), 0) + elif FuncName == 'chr': + if Node.args: + return self._HandleChrLlvm(Node.args[0]) + return ir.Constant(ir.IntType(8), 0) + elif FuncName == 'malloc': + return self._HandleMallocLlvm(Node.args) + elif FuncName == 'realloc': + return self._HandleReallocLlvm(Node.args) + elif FuncName == 'free': + return self._HandleFreeLlvm(Node.args) + elif FuncName == 'memcpy': + return self._HandleMemcpyLlvm(Node.args) + elif FuncName == 'memset': + return self._HandleMemsetLlvm(Node.args) + elif FuncName == 'va_start': + return self._HandleVaStartLlvm(Node.args) + elif FuncName == 'va_end': + return self._HandleVaEndLlvm(Node.args) + elif FuncName == 'va_arg': + return self._HandleVaArgLlvm(Node.args) + elif FuncName == 'dir': + return self._HandleDirLlvm(Node) + elif FuncName == 'type': + return self._HandleTypeLlvm(Node) + return None + + def _HandlePrintLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + printf_func: ir.Function = Gen.get_or_declare_c_func('printf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + args: list[ast.expr] = Node.args + end_arg: ast.expr | None = None + sep_arg: ast.expr | None = None + if Node.keywords: + for kw in Node.keywords: + if kw.arg == 'end': + end_arg = kw.value + elif kw.arg == 'sep': + sep_arg = kw.value + sep_str: str = ' ' + if sep_arg and isinstance(sep_arg, ast.Constant) and isinstance(sep_arg.value, str): + sep_str = sep_arg.value + end_str: str = '\n' + if end_arg: + if isinstance(end_arg, ast.Constant) and isinstance(end_arg.value, str): + end_str = end_arg.value + elif isinstance(end_arg, ast.Constant) and end_arg.value is None: + end_str = '' + if not args: + if end_str: + nl_fmt = Gen.emit_constant(end_str, 'string') + Gen.builder.call(printf_func, [nl_fmt], name="print_nl") + return ir.Constant(ir.IntType(32), 0) + has_fstring = any(isinstance(arg, ast.JoinedStr) for arg in args) + if has_fstring: + return self._HandlePrintWithFString(Node, printf_func, args, sep_str, end_str) + fmt_parts = [] + fmt_args = [] + for i, arg in enumerate(args): + if i > 0: + fmt_parts.append(sep_str) + val = self.HandleExprLlvm(arg) + if not val: + fmt_parts.append('(null)') + continue + if isinstance(val, ir.Constant) and isinstance(val.type, ir.PointerType) and val.constant is None: + fmt_parts.append('nil') + continue + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if isinstance(pointee, ir.IntType) and pointee.width == 8: + fmt_parts.append('%s') + fmt_args.append(val) + elif isinstance(pointee, ir.ArrayType) and isinstance(pointee.element, ir.IntType) and pointee.element.width == 8: + elem_ptr = Gen.builder.gep(val, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="print_arr_to_ptr") + fmt_parts.append('%s') + fmt_args.append(elem_ptr) + elif isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + ClassName = self.Trans.ExprHandler._get_var_class(arg, Gen) + if not ClassName and isinstance(arg, ast.Name): + ClassName = Gen.var_struct_class.get(arg.id) + if ClassName and Gen._has_function(f'{ClassName}.__str__'): + if isinstance(val.type, ir.PointerType) and isinstance(val.type.pointee, ir.IntType) and val.type.pointee.width == 8: + if ClassName in Gen.structs: + val = Gen.builder.bitcast(val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + str_val = Gen.builder.call(Gen._get_function(f'{ClassName}.__str__'), [val], name=f"call_{ClassName}.__str__") + fmt_parts.append('%s') + fmt_args.append(str_val) + else: + int_val = Gen.builder.ptrtoint(val, ir.IntType(64), name="print_ptr_to_int") + fmt_parts.append('0x%llx') + fmt_args.append(int_val) + else: + int_val = Gen.builder.ptrtoint(val, ir.IntType(64), name="print_ptr_to_int") + fmt_parts.append('%lld') + fmt_args.append(int_val) + elif isinstance(val.type, ir.IntType): + is_unsigned = self._is_val_unsigned(arg, val) + if val.type.width == 1: + fmt_parts.append('%d') + val = Gen.builder.zext(val, ir.IntType(32), name="zext_bool_print") + elif val.type.width == 8: + if self._is_char_type(arg): + char_var = Gen._allocaEntry(ir.IntType(8), name="print_char_buf") + Gen.builder.store(val, char_var) + fmt_parts.append('%c') + fmt_args.append(char_var) + continue + fmt_parts.append('%d') + if is_unsigned: + val = Gen.builder.zext(val, ir.IntType(32), name="zext_i8_print") + else: + val = Gen.builder.sext(val, ir.IntType(32), name="sext_i8_print") + elif val.type.width == 16: + fmt_parts.append('%u' if is_unsigned else '%d') + if is_unsigned: + val = Gen.builder.zext(val, ir.IntType(32), name="zext_i16_print") + else: + val = Gen.builder.sext(val, ir.IntType(32), name="sext_i16_print") + elif val.type.width == 32: + fmt_parts.append('%u' if is_unsigned else '%d') + elif val.type.width == 64: + fmt_parts.append('%llu' if is_unsigned else '%lld') + else: + fmt_parts.append('%d') + val = Gen.builder.zext(val, ir.IntType(32), name="zext_int_print") + fmt_args.append(val) + elif isinstance(val.type, ir.FloatType): + fmt_parts.append('%f') + fmt_args.append(Gen.builder.fpext(val, ir.DoubleType(), name="fpext_printf_float")) + elif isinstance(val.type, ir.DoubleType): + fmt_parts.append('%f') + fmt_args.append(val) + else: + fmt_parts.append('(unknown)') + fmt_parts.append(end_str) + fmt_str = ''.join(fmt_parts) + fmt_ptr = Gen._create_string_global(fmt_str) + call_args = [fmt_ptr] + fmt_args + return Gen.builder.call(printf_func, call_args, name="print_call") + + def _HandlePrintWithFString(self, Node: ast.Call, printf_func: ir.Function, args: list[ast.expr], sep_str: str, end_str: str) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + for i, arg in enumerate(args): + if i > 0: + sep_const = Gen.emit_constant(sep_str, 'string') + Gen.builder.call(printf_func, [sep_const], name="print_sep") + if isinstance(arg, ast.JoinedStr): + str_val = self.HandleExprLlvm(arg) + if str_val: + Gen._emit_printf([str_val], unsigned_flags=[False], sep=None, end=None) + else: + ClassName = self.Trans.ExprHandler._get_var_class(arg, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__str__'): + obj_val = self.HandleExprLlvm(arg) + if obj_val: + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + if ClassName in Gen.structs: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + str_val = Gen.builder.call(Gen._get_function(f'{ClassName}.__str__'), [obj_val], name=f"call_{ClassName}.__str__") + Gen._RegisterTempPtr(str_val) + Gen._emit_printf([str_val], unsigned_flags=[False], sep=None, end=None) + else: + val = self.HandleExprLlvm(arg) + if val: + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if not (isinstance(pointee, ir.IntType) and pointee.width == 8): + try: + val = Gen._load(val, name="print_Load") + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + is_u = Gen._check_node_unsigned(arg) + if not is_u and id(val) in Gen._unsigned_results: + is_u = True + Gen._emit_printf([val], unsigned_flags=[is_u], sep=None, end=None) + if end_str: + end_const = Gen.emit_constant(end_str, 'string') + Gen.builder.call(printf_func, [end_const], name="print_end") + return ir.Constant(ir.IntType(32), 0) + + def _is_char_type(self, arg_node: ast.AST) -> bool: + if isinstance(arg_node, ast.Call): + if isinstance(arg_node.func, ast.Name): + return arg_node.func.id in ('CChar', 'CUInt8T', 'CInt8T') + if isinstance(arg_node.func, ast.Attribute): + return arg_node.func.attr in ('CChar', 'CUInt8T', 'CInt8T') + return False + + def _is_val_unsigned(self, arg_node: ast.AST, val: ir.Value) -> bool: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if isinstance(arg_node, ast.Name): + var_name: str = arg_node.id + signedness: object = Gen.var_signedness.get(var_name) + if signedness: + if isinstance(signedness, bool): + return signedness + return 'unsigned' in signedness + if isinstance(val.type, ir.IntType): + return False + return False + + def _try_declare_c_lib_func(self, func_name: str, Gen: LlvmGeneratorMixin) -> ir.Function | None: + if func_name not in self._C_LIB_FUNCS: + return None + ret_type: ir.Type + param_types: list[ir.Type] + ret_type, param_types = self._C_LIB_FUNCS[func_name]() + func_type: ir.FunctionType = ir.FunctionType(ret_type, param_types) + func: ir.Function = ir.Function(Gen.module, func_type, name=func_name) + Gen.functions[func_name] = func + return func + + def _HandleLenLlvm(self, arg_node: ast.AST) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if isinstance(arg_node, ast.Attribute) and isinstance(arg_node.value, ast.Name) and arg_node.value.id == 'self': + attr_name = arg_node.attr + len_member = f'{attr_name}__len' + current_class = self.Trans._CurrentCpythonObjectClass + if current_class and current_class in Gen.class_members: + for m_name, m_type in Gen.class_members[current_class]: + if m_name == len_member: + self_val = Gen._get_var_ptr('self') + if self_val: + self_ptr = Gen._load(self_val, name="self_for_len") + offset = Gen._get_member_offset(len_member, current_class) + len_ptr = Gen.builder.gep(self_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=f"self_{len_member}_ptr") + return Gen._load(len_ptr, name=f"self_{len_member}") + val = self.HandleExprLlvm(arg_node) + if not val: + return None + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if isinstance(pointee, ir.PointerType): + val = Gen._load(val, name="len_deref") + pointee = val.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + result = self.Trans.ExprUtils._try_operator_overLoad(CN, '__len__', val, Gen) + if result is not None: + return result + break + if isinstance(pointee, ir.ArrayType): + return ir.Constant(ir.IntType(64), pointee.count) + if isinstance(pointee, (ir.FloatType, ir.DoubleType, ir.IntType)) and pointee.width != 8: + return ir.Constant(ir.IntType(64), 0) + if val.type != ir.PointerType(ir.IntType(8)): + val = Gen.builder.bitcast(val, ir.PointerType(ir.IntType(8)), name="str_ptr_cast") + correct_strlen_type = ir.FunctionType(ir.IntType(64), [ir.PointerType(ir.IntType(8))]) + strlen_func = None + if 'strlen' in Gen.functions: + existing = Gen.functions['strlen'] + if existing.ftype == correct_strlen_type: + strlen_func = existing + if not strlen_func: + try: + strlen_func = ir.Function(Gen.module, correct_strlen_type, name='strlen') + Gen.functions['strlen'] = strlen_func + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + if strlen_func: + result = Gen.builder.call(strlen_func, [val], name="strlen_call") + return result + elif isinstance(val.type, ir.ArrayType): + return ir.Constant(ir.IntType(64), val.type.count) + return ir.Constant(ir.IntType(64), 0) + + def _resolve_nested_opaque_structs(self, struct_type: ir.Type, Gen: LlvmGeneratorMixin, visited: set | None = None) -> None: + """递归解析嵌套的 opaque struct，从 stub 文件中加载其定义""" + if visited is None: + visited = set() + if isinstance(struct_type, ir.PointerType): + self._resolve_nested_opaque_structs(struct_type.pointee, Gen, visited) + return + if not isinstance(struct_type, ir.IdentifiedStructType): + return + if struct_type in visited: + return + visited.add(struct_type) + if struct_type.elements is None or len(struct_type.elements) == 0: + # 尝试从 stub 加载 opaque struct + try: + st_name = struct_type.name + except Exception: + st_name = None + if st_name: + # 提取短名称（去掉模块前缀） + if '.' in st_name: + short_name = st_name.split('.', 1)[1] + else: + short_name = st_name + self.Trans.ImportHandler._TryLoadStructFromStub(short_name, Gen) + return + # 递归解析嵌套的 struct 成员 + for elem in struct_type.elements: + self._resolve_nested_opaque_structs(elem, Gen, visited) + + def _HandleSizeofLlvm(self, Node: ast.AST) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + type_name: str | None + if isinstance(Node, ast.Name): + type_name = Node.id + elif isinstance(Node, ast.Attribute): + type_name = Node.attr + else: + return ir.Constant(ir.IntType(64), 0) + + if type_name not in Gen.structs: + # str/bytes 是指针类型 (char*/void*)，sizeof = 指针大小 + if type_name in ('str', 'bytes'): + return ir.Constant(ir.IntType(64), 8) + ctype_cls = CTypeRegistry.GetClassByName(type_name) + if ctype_cls is not None: + try: + ctype_inst = ctype_cls() + size_bits = getattr(ctype_inst, 'Size', 0) or 0 + size_bytes = size_bits // 8 + return ir.Constant(ir.IntType(64), size_bytes) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + resolved = CTypeRegistry.ResolveName(type_name) + if resolved: + ctype_cls, ptr_level = resolved + try: + ctype_inst = ctype_cls() + size_bits = getattr(ctype_inst, 'Size', 0) or 0 + size_bytes = size_bits // 8 + if ptr_level > 0: + size_bytes = 8 + return ir.Constant(ir.IntType(64), size_bytes) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + return ir.Constant(ir.IntType(64), 0) + + struct_type = Gen.structs[type_name] + if isinstance(struct_type, ir.PointerType): + struct_type = struct_type.pointee + if isinstance(struct_type, ir.IdentifiedStructType) and (struct_type.elements is None or len(struct_type.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(type_name, Gen) + struct_type = Gen.structs.get(type_name, struct_type) + # 递归解析所有嵌套的 opaque struct，确保 sizeof 计算正确 + self._resolve_nested_opaque_structs(struct_type, Gen) + size = Gen._get_struct_size(struct_type) + if size > 0: + return ir.Constant(ir.IntType(64), size) + return ir.Constant(ir.IntType(64), 0) + + def _HandleAbsLlvm(self, arg_node: ast.AST) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + val: ir.Value | None = self.HandleExprLlvm(arg_node) + if not val: + return None + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + result = self.Trans.ExprUtils._try_operator_overLoad(CN, '__abs__', val, Gen) + if result is not None: + return result + break + if isinstance(val.type, ir.IntType): + neg_val = Gen.builder.neg(val, name="abs_neg") + is_neg = Gen.builder.icmp_signed('<', val, ir.Constant(val.type, 0), name="is_neg") + return Gen.builder.select(is_neg, neg_val, val, name="abs_result") + elif isinstance(val.type, (ir.FloatType, ir.DoubleType)): + zero = ir.Constant(val.type, 0.0) + neg_val = Gen.builder.fneg(val, name="fabs_neg") + is_neg = Gen.builder.fcmp_ordered('<', val, zero, name="f_is_neg") + return Gen.builder.select(is_neg, neg_val, val, name="fabs_result") + return None + + def _HandleDirLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _HandleTypeLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + type_str: str + if not Node.args: + type_str = "" + else: + arg = Node.args[0] + enum_type_name = None + + def _get_attr_path(node: ast.AST) -> str | None: + if isinstance(node, ast.Name): + return node.id + elif isinstance(node, ast.Attribute): + return f"{_get_attr_path(node.value)}.{node.attr}" + return None + + if isinstance(arg, ast.Name): + arg_name = arg.id + Gen = self.Trans.LlvmGen + if arg_name in Gen.var_type_info: + type_info = Gen.var_type_info[arg_name] + if isinstance(type_info, dict) and type_info.get('type') == 'enum': + enum_type_name = type_info.get('name') + if not enum_type_name and self.Trans.SymbolTable.has(arg_name): + TypeInfo = self.Trans.SymbolTable[arg_name] + if TypeInfo.IsEnum: + enum_type_name = arg_name + elif isinstance(arg, ast.Attribute): + LastPart = None + enum_class_name = None + if isinstance(arg.value, ast.Name): + LastPart = arg.attr + if self.Trans.SymbolTable.has(LastPart): + AttrInfo = self.Trans.SymbolTable[LastPart] + if AttrInfo.IsEnumMember and AttrInfo.EnumName: + enum_type_name = AttrInfo.EnumName + elif isinstance(arg.value, ast.Attribute): + attr_path = _get_attr_path(arg) + if attr_path: + parts = attr_path.split('.') + if len(parts) >= 2: + enum_class_name = parts[-2] + LastPart = parts[-1] + qualified_name_dot = f"{enum_class_name}.{LastPart}" + qualified_name_under = f"{enum_class_name}_{LastPart}" + enum_member_found = None + for qname in (qualified_name_dot, qualified_name_under): + if self.Trans.SymbolTable.has(qname): + info = self.Trans.SymbolTable[qname] + if info.IsEnumMember and info.EnumName == enum_class_name: + enum_member_found = info + break + if not enum_member_found: + for key in self.Trans.SymbolTable: + if key == LastPart: + info = self.Trans.SymbolTable[key] + if info.IsEnumMember and info.EnumName == enum_class_name: + enum_member_found = info + break + if enum_member_found: + enum_type_name = enum_class_name + if enum_type_name: + type_str = f"" + else: + llvm_type = self.Trans.ExprUtils._infer_expr_llvm_type_full(arg) + if isinstance(llvm_type, ir.PointerType): + pointee = llvm_type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + llvm_type = pointee + type_info = self.Trans.ExprUtils._llvm_type_to_detailed_string(llvm_type) + type_str = f"" + return Gen.emit_constant(type_str, 'string') + + def _HandleInputLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if Node.args and isinstance(Node.args[0], ast.Constant) and isinstance(Node.args[0].value, str): + prompt = Node.args[0].value + prompt_ptr = Gen.emit_constant(prompt, 'string') + printf_func = Gen.get_or_declare_c_func('printf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + Gen.builder.call(printf_func, [prompt_ptr]) + scanf_func = Gen.get_or_declare_c_func('scanf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + input_buffer = Gen._alloca(ir.ArrayType(ir.IntType(8), 256), name="input_buffer") + result = Gen.builder.call(scanf_func, [Gen.emit_constant("%255s", 'string'), input_buffer]) + return Gen.builder.gep(input_buffer, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="input_result") + + def _HandleAtoiLlvm(self, str_ptr: ir.Value) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + atoi_func: ir.Function = Gen.get_or_declare_c_func('atoi', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))])) + return Gen.builder.call(atoi_func, [str_ptr], name="atoi_result") + + def _HandleOrdLlvm(self, expr_node: ast.AST) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + val: ir.Value | None = self.HandleExprLlvm(expr_node) + if val and isinstance(val.type, ir.PointerType) and isinstance(val.type.pointee, ir.IntType) and val.type.pointee.width == 8: + return Gen._load(val, name="ord_result") + return ir.Constant(ir.IntType(32), 0) + + def _HandleChrLlvm(self, expr_node: ast.AST) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + val: ir.Value | None = self.HandleExprLlvm(expr_node) + if val: + char_ptr = Gen._alloca(ir.IntType(8), name="chr_char") + Gen.builder.store(Gen.builder.trunc(val, ir.IntType(8), name="trunc_chr"), char_ptr) + return Gen.builder.gep(char_ptr, [ir.Constant(ir.IntType(32), 0)], name="chr_result") + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _EmitStringFromValue(self, val: ir.Value) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if isinstance(val.type, ir.IntType): + if val.type.width == 8: + buf = Gen._alloca(ir.IntType(8), name="str_buf", size=ir.Constant(ir.IntType(32), 2)) + null_ptr = Gen.builder.gep(buf, [ir.Constant(ir.IntType(32), 1)], name="null_byte") + Gen.builder.store(ir.Constant(ir.IntType(8), 0), null_ptr) + Gen.builder.store(val, buf) + return Gen.builder.gep(buf, [ir.Constant(ir.IntType(32), 0)], name="str_result") + buf = Gen._alloca(ir.IntType(8), name="str_buf", size=ir.Constant(ir.IntType(32), 32)) + snprintf_func = Gen.get_or_declare_c_func('snprintf', ir.FunctionType(ir.IntType(32), [ + ir.PointerType(ir.IntType(8)), + ir.IntType(64), + ir.PointerType(ir.IntType(8)) + ], var_arg=True)) + fmt_ptr = Gen.emit_constant("%d", 'string') + Gen.builder.call(snprintf_func, [buf, ir.Constant(ir.IntType(64), 32), fmt_ptr, val]) + return Gen.builder.gep(buf, [ir.Constant(ir.IntType(32), 0)], name="str_result") + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _HandleMallocLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + malloc_type: ir.FunctionType = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.IntType(64)]) + malloc_func: ir.Function = Gen._get_or_declare_func('malloc', malloc_type) + param_type: ir.Type = malloc_func.type.pointee.args[0] + size_val: ir.Value | None = None + if args: + size_val = self.HandleExprLlvm(args[0]) + if size_val and isinstance(size_val.type, ir.IntType): + if isinstance(param_type, ir.IntType) and size_val.type.width != param_type.width: + if size_val.type.width < param_type.width: + size_val = Gen.builder.zext(size_val, param_type, name="zext_malloc_size") + else: + size_val = Gen.builder.trunc(size_val, param_type, name="trunc_malloc_size") + if not size_val: + size_val = ir.Constant(param_type, 1) + result: ir.Value = Gen.builder.call(malloc_func, [size_val], name="malloc_result") + return Gen.builder.bitcast(result, ir.IntType(8).as_pointer(), name="malloc_cast") + + def _HandleReallocLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + realloc_type: ir.FunctionType = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(64)]) + realloc_func: ir.Function = Gen._get_or_declare_func('realloc', realloc_type) + size_param_type: ir.Type = realloc_func.type.pointee.args[1] + ptr_val: ir.Value | None = None + size_val: ir.Value | None = None + if len(args) >= 2: + ptr_val = self.HandleExprLlvm(args[0]) + size_val = self.HandleExprLlvm(args[1]) + elif len(args) == 1: + size_val = self.HandleExprLlvm(args[0]) + if not ptr_val: + ptr_val = ir.Constant(ir.PointerType(ir.IntType(8)), None) + if isinstance(ptr_val.type, ir.PointerType) and not (isinstance(ptr_val.type.pointee, ir.IntType) and ptr_val.type.pointee.width == 8): + ptr_val = Gen.builder.bitcast(ptr_val, ir.PointerType(ir.IntType(8)), name="realloc_cast_ptr") + if size_val and isinstance(size_val.type, ir.IntType): + if isinstance(size_param_type, ir.IntType) and size_val.type.width != size_param_type.width: + if size_val.type.width < size_param_type.width: + size_val = Gen.builder.zext(size_val, size_param_type, name="zext_realloc_size") + else: + size_val = Gen.builder.trunc(size_val, size_param_type, name="trunc_realloc_size") + if not size_val: + size_val = ir.Constant(size_param_type, 0) + result: ir.Value = Gen.builder.call(realloc_func, [ptr_val, size_val], name="realloc_result") + return Gen.builder.bitcast(result, ir.IntType(8).as_pointer(), name="realloc_cast") + + def _HandleFreeLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + free_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))]) + free_func: ir.Function = Gen._get_or_declare_func('free', free_type) + if args: + var_name: str | None = None + if isinstance(args[0], ast.Name): + var_name = args[0].id + ptr_val: ir.Value | None = self.HandleExprLlvm(args[0]) + if ptr_val is not None: + if var_name and hasattr(Gen, '_var_to_heap_ptr') and var_name in Gen._var_to_heap_ptr: + registered_val = Gen._var_to_heap_ptr[var_name] + Gen._unregister_local_heap_ptr(registered_val) + del Gen._var_to_heap_ptr[var_name] + if isinstance(ptr_val.type, ir.PointerType): + if isinstance(ptr_val.type.pointee, ir.IntType) and ptr_val.type.pointee.width == 8: + pass + else: + ptr_val = Gen.builder.bitcast(ptr_val, ir.PointerType(ir.IntType(8)), name="free_cast_ptr") + elif isinstance(ptr_val.type, ir.IntType): + ptr_val = Gen.builder.inttoptr(ptr_val, ir.PointerType(ir.IntType(8)), name="free_inttoptr") + Gen.builder.call(free_func, [ptr_val], name="free_call") + return ir.Constant(ir.IntType(32), 0) + + def _HandleMemcpyLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if 'llvm.memcpy' not in Gen.functions: + memcpy_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64), ir.IntType(1)]) + Gen.functions['llvm.memcpy'] = ir.Function(Gen.module, memcpy_type, name='llvm.memcpy') + if len(args) >= 3: + dst: ir.Value | None = self.HandleExprLlvm(args[0]) + src: ir.Value | None = self.HandleExprLlvm(args[1]) + size: ir.Value | None = self.HandleExprLlvm(args[2]) + if dst and src and size: + i8ptr: ir.PointerType = ir.PointerType(ir.IntType(8)) + if isinstance(dst.type, ir.PointerType) and dst.type != i8ptr: + dst = Gen.builder.bitcast(dst, i8ptr, name="memcpy_dst_cast") + elif isinstance(dst.type, ir.IntType): + if dst.type.width < 64: + dst = Gen.builder.zext(dst, ir.IntType(64), name="zext_dst_ptr") + dst = Gen.builder.inttoptr(dst, i8ptr, name="dst_int2ptr") + if isinstance(src.type, ir.PointerType) and src.type != i8ptr: + src = Gen.builder.bitcast(src, i8ptr, name="memcpy_src_cast") + elif isinstance(src.type, ir.IntType): + if src.type.width < 64: + src = Gen.builder.zext(src, ir.IntType(64), name="zext_src_ptr") + src = Gen.builder.inttoptr(src, i8ptr, name="src_int2ptr") + if isinstance(size.type, ir.IntType) and size.type.width < 64: + size = Gen.builder.zext(size, ir.IntType(64), name="zext_memcpy_size") + isvolatile: ir.Constant = ir.Constant(ir.IntType(1), 0) + Gen.builder.call(Gen.functions['llvm.memcpy'], [dst, src, size, isvolatile], name="memcpy_call") + return ir.Constant(ir.IntType(32), 0) + + def _HandleMemsetLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + memset_func: ir.Function = Gen.get_or_declare_c_func('memset', ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(32), ir.IntType(64)])) + if len(args) >= 3: + dst: ir.Value | None = self.HandleExprLlvm(args[0]) + val: ir.Value | None = self.HandleExprLlvm(args[1]) + size: ir.Value | None = self.HandleExprLlvm(args[2]) + if dst and val is not None and size: + if isinstance(dst.type, ir.PointerType) and not (isinstance(dst.type.pointee, ir.IntType) and dst.type.pointee.width == 8): + dst = Gen.builder.bitcast(dst, ir.PointerType(ir.IntType(8)), name="memset_dst_cast") + elif isinstance(dst.type, ir.IntType): + if dst.type.width < 64: + dst = Gen.builder.zext(dst, ir.IntType(64), name="zext_memset_dst") + elif dst.type.width > 64: + dst = Gen.builder.trunc(dst, ir.IntType(64), name="trunc_memset_dst") + dst = Gen.builder.inttoptr(dst, ir.PointerType(ir.IntType(8)), name="memset_dst_int2ptr") + if isinstance(val.type, ir.IntType) and val.type.width != 32: + if val.type.width < 32: + val = Gen.builder.zext(val, ir.IntType(32), name="zext_memset_val") + else: + val = Gen.builder.trunc(val, ir.IntType(32), name="trunc_memset_val") + if isinstance(size.type, ir.IntType) and size.type.width < 64: + size = Gen.builder.zext(size, ir.IntType(64), name="zext_memset_size") + elif isinstance(size.type, ir.IntType) and size.type.width > 64: + size = Gen.builder.trunc(size, ir.IntType(64), name="trunc_memset_size") + Gen.builder.call(memset_func, [dst, val, size], name="memset_call") + return ir.Constant(ir.IntType(32), 0) + + def _HandleVaStartLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + return ir.Constant(ir.IntType(32), 0) + + def _HandleVaEndLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + return ir.Constant(ir.IntType(32), 0) + + def _HandleArgLlvm(self, args: list[ast.expr], CallNode: ast.Call | None = None) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + va_list_ptr: ir.Value | None = None + variadic_info: dict | None = getattr(Gen, '_va_arg_info', None) + if not variadic_info: + variadic_info = getattr(Gen, '_variadic_info', None) + if variadic_info: + va_list_ptr = variadic_info.get('va_list_ptr') + if not va_list_ptr: + vararg_name: str = variadic_info.get('vararg_name', 'args') if variadic_info else 'args' + if vararg_name in Gen._direct_values: + va_list_ptr = Gen._direct_values[vararg_name] + elif vararg_name in Gen.variables: + va_list_ptr = Gen.variables[vararg_name] + if not va_list_ptr: + return ir.Constant(ir.IntType(32), 0) + target_type: ir.Type = ir.IntType(32) + if args: + type_arg: ast.expr = args[0] + if isinstance(type_arg, ast.Name): + type_name: str = type_arg.id + if type_name in ('str', 'bytes'): + target_type = ir.PointerType(ir.IntType(8)) + elif type_name == 'CPtr': + target_type = ir.PointerType(ir.IntType(8)) + elif type_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[type_name]) + else: + ctype_cls: type | None = CTypeRegistry.GetClassByName(type_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + else: + resolved: tuple | None = CTypeRegistry.ResolveName(type_name) + if resolved is not None: + ctype_cls, ptr_level = resolved + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str: + base: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if base is not None: + for _ in range(ptr_level): + if isinstance(base, ir.VoidType): + base = ir.IntType(8).as_pointer() + else: + base = ir.PointerType(base) + target_type = base + elif isinstance(type_arg, ast.Attribute): + if isinstance(type_arg.value, ast.Name) and IsTModule(type_arg.value.id, self.Trans.SymbolTable): + attr_name: str = type_arg.attr + ctype_cls: type | None = CTypeRegistry.GetClassByName(attr_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + else: + assign_node: ast.AST | None = getattr(self.Trans, '_current_assign_node', None) + if assign_node: + ann: ast.AST | None = getattr(assign_node, 'annotation', None) + if ann: + ann_name: str | None = None + if isinstance(ann, ast.Name): + ann_name = ann.id + elif isinstance(ann, ast.Attribute): + ann_name = ann.attr + elif isinstance(ann, ast.BinOp) and isinstance(ann.op, ast.BitOr): + ann_name = ExtractTypeNameFromBinOp(ann) + if ann_name: + ctype_cls: type | None = CTypeRegistry.GetClassByName(ann_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + elif ann_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[ann_name]) + if not getattr(Gen, '_va_arg_counter', None): + Gen._va_arg_counter = 0 + Gen._va_arg_counter += 1 + result: ir.Value | None = Gen.emit_va_arg(va_list_ptr, target_type) + if result is not None and getattr(result, 'name', None): + result.name = f"va_arg_result_{Gen._va_arg_counter}" + return result + + def _resolve_vararg_type(self, type_arg: ast.expr, Gen: LlvmGeneratorMixin) -> ir.Type: + """将类型表达式解析为 LLVM 类型（供 arg/va_arg 共用）。""" + target_type: ir.Type = ir.IntType(32) + if isinstance(type_arg, ast.Name): + type_name: str = type_arg.id + if type_name in ('str', 'bytes'): + target_type = ir.PointerType(ir.IntType(8)) + elif type_name == 'CPtr': + target_type = ir.PointerType(ir.IntType(8)) + elif type_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[type_name]) + else: + ctype_cls: type | None = CTypeRegistry.GetClassByName(type_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + else: + resolved_tuple: tuple | None = CTypeRegistry.ResolveName(type_name) + if resolved_tuple is not None: + ctype_cls, ptr_level = resolved_tuple + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str: + base: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if base is not None: + for _ in range(ptr_level): + if isinstance(base, ir.VoidType): + base = ir.IntType(8).as_pointer() + else: + base = ir.PointerType(base) + target_type = base + elif isinstance(type_arg, ast.Attribute): + if isinstance(type_arg.value, ast.Name) and IsTModule(type_arg.value.id, self.Trans.SymbolTable): + attr_name: str = type_arg.attr + ctype_cls: type | None = CTypeRegistry.GetClassByName(attr_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + return target_type + + def _infer_vararg_type_from_context(self, Gen: LlvmGeneratorMixin) -> ir.Type: + """从当前赋值上下文推断 va_arg/arg 的目标类型。""" + target_type: ir.Type = ir.IntType(32) + assign_node: ast.AST | None = getattr(self.Trans, '_current_assign_node', None) + if assign_node: + ann: ast.AST | None = getattr(assign_node, 'annotation', None) + if ann: + ann_name: str | None = None + if isinstance(ann, ast.Name): + ann_name = ann.id + elif isinstance(ann, ast.Attribute): + ann_name = ann.attr + elif isinstance(ann, ast.BinOp) and isinstance(ann.op, ast.BitOr): + ann_name = ExtractTypeNameFromBinOp(ann) + if ann_name: + ctype_cls: type | None = CTypeRegistry.GetClassByName(ann_name) + if ctype_cls is not None: + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + elif ann_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[ann_name]) + return target_type + + def _HandleVaArgLlvm(self, args: list[ast.expr]) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not args: + return ir.Constant(ir.IntType(32), 0) + # 求值 va_list 指针（args[0]） + va_list_ptr: ir.Value = self.HandleExprLlvm(args[0]) + if va_list_ptr is None: + return ir.Constant(ir.IntType(32), 0) + # 解析目标类型 + if len(args) >= 2: + target_type: ir.Type = self._resolve_vararg_type(args[1], Gen) + else: + target_type: ir.Type = self._infer_vararg_type_from_context(Gen) + if not getattr(Gen, '_va_arg_counter', None): + Gen._va_arg_counter = 0 + Gen._va_arg_counter += 1 + result: ir.Value | None = Gen.emit_va_arg(va_list_ptr, target_type) + if result is not None and getattr(result, 'name', None): + result.name = f"va_arg_explicit_{Gen._va_arg_counter}" + return result if result is not None else ir.Constant(ir.IntType(32), 0) diff --git a/lib/core/Handles/HandlesExprCall.py b/lib/core/Handles/HandlesExprCall.py new file mode 100644 index 0000000..e0fdd3c --- /dev/null +++ b/lib/core/Handles/HandlesExprCall.py @@ -0,0 +1,3779 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +import ast +import os +import re +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo, FuncMeta +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import IsTModule, ExtractTypeNameFromBinOp + + +class ExprCallHandle(BaseHandle): + def _append_eh_msg_out_arg(self, CallArgs: list[ir.Value], func: ir.Function | None, Gen: LlvmGeneratorMixin) -> None: + if not func: + return + func_args: list[ir.Type] = func.function_type.args + n_user_args: int = len(CallArgs) + has_eh_msg: bool = False + has_eh_code: bool = False + if len(func_args) > n_user_args: + last_type: ir.Type = func_args[-1] + if (isinstance(last_type, ir.PointerType) + and isinstance(last_type.pointee, ir.IntType) + and last_type.pointee.width == 32): + has_eh_code = True + if len(func_args) > n_user_args + (1 if has_eh_code else 0): + check_idx: int = len(func_args) - 1 - (1 if has_eh_code else 0) + check_type: ir.Type = func_args[check_idx] + if (isinstance(check_type, ir.PointerType) + and isinstance(check_type.pointee, ir.PointerType) + and isinstance(check_type.pointee.pointee, ir.IntType) + and check_type.pointee.pointee.width == 8): + has_eh_msg = True + if not has_eh_msg: + return + eh_msg_ptr: ir.Value | None = None + eh_code_ptr: ir.Value | None = None + if Gen.eh_except_block_stack: + _, _, exception_code, eh_message = Gen.eh_except_block_stack[-1] + if eh_message is not None: + eh_msg_ptr = eh_message + if exception_code is not None: + eh_code_ptr = exception_code + if eh_msg_ptr is None and Gen.func and len(Gen.func.args) > 0: + for arg in Gen.func.args: + if arg.name == '__eh_msg_out__' and eh_msg_ptr is None: + eh_msg_ptr = arg + elif arg.name == '__eh_code_out__' and eh_code_ptr is None: + eh_code_ptr = arg + if eh_msg_ptr is None: + eh_msg_ptr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name="eh_msg_out_null") + if eh_code_ptr is None: + eh_code_ptr = Gen._allocaEntry(ir.IntType(32), name="eh_code_out_null") + Gen._store(ir.Constant(ir.PointerType(ir.IntType(8)), None), eh_msg_ptr) + Gen._store(ir.Constant(ir.IntType(32), 0), eh_code_ptr) + CallArgs.append(eh_msg_ptr) + if has_eh_code: + CallArgs.append(eh_code_ptr) + + def _check_eh_return(self, result: ir.Value, func_name: str, Gen: LlvmGeneratorMixin, called_func: ir.Function | None = None) -> ir.Value: + has_eh_msg_out: bool = False + has_eh_code_out: bool = False + if called_func and hasattr(called_func, 'args') and len(called_func.args) > 0: + for arg in called_func.args: + if arg.name == '__eh_msg_out__': + has_eh_msg_out = True + elif arg.name == '__eh_code_out__': + has_eh_code_out = True + if not has_eh_msg_out: + return result + eh_msg_ptr: ir.Value | None = None + eh_code_ptr: ir.Value | None = None + if Gen.eh_except_block_stack: + _, _, exception_code, eh_message = Gen.eh_except_block_stack[-1] + if eh_message is not None: + eh_msg_ptr = eh_message + if exception_code is not None: + eh_code_ptr = exception_code + if eh_msg_ptr is None and Gen.func and len(Gen.func.args) > 0: + for arg in Gen.func.args: + if arg.name == '__eh_msg_out__' and eh_msg_ptr is None: + eh_msg_ptr = arg + elif arg.name == '__eh_code_out__' and eh_code_ptr is None: + eh_code_ptr = arg + if eh_msg_ptr is not None: + eh_msg_val: ir.Value = Gen._load(eh_msg_ptr, name=f"Load_eh_msg_{func_name}") + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + is_error: ir.Value = Gen.builder.icmp_signed('!=', eh_msg_val, null_ptr, name=f"check_eh_msg_{func_name}") + OkBB: ir.Block = Gen.func.append_basic_block(name=f"ok_after_{func_name}") + ErrorBB: ir.Block = Gen.func.append_basic_block(name=f"error_after_{func_name}") + Gen.builder.cbranch(is_error, ErrorBB, OkBB) + Gen.builder.position_at_start(ErrorBB) + ret_type: ir.Type | None = getattr(Gen.func, 'ftype', None) and Gen.func.ftype.return_type if getattr(Gen, 'func', None) else None + error_ret_val: ir.Constant | None = None + if isinstance(ret_type, ir.IdentifiedStructType): + if ret_type.elements: + error_ret_val = ir.Constant(ret_type, [ + ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) + else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) + else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + error_ret_val = ir.Constant(ret_type, None) + elif isinstance(ret_type, ir.PointerType): + error_ret_val = ir.Constant(ret_type, None) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + error_ret_val = ir.Constant(ret_type, 0.0) + elif isinstance(ret_type, ir.IntType): + error_ret_val = ir.Constant(ret_type, 1) + elif isinstance(ret_type, ir.VoidType): + pass + else: + error_ret_val = ir.Constant(ir.IntType(32), 1) + if Gen.eh_except_block_stack: + ExceptBB, _, exception_code, _ = Gen.eh_except_block_stack[-1] + if ExceptBB is not None: + if has_eh_code_out and eh_code_ptr is not None: + eh_code_val = Gen._load(eh_code_ptr, name=f"Load_eh_code_{func_name}") + Gen._store(eh_code_val, exception_code) + else: + Gen._store(ir.Constant(ir.IntType(32), 1), exception_code) + Gen.builder.branch(ExceptBB) + else: + if isinstance(ret_type, ir.VoidType): + Gen.builder.ret_void() + else: + Gen.builder.ret(error_ret_val) + else: + if isinstance(ret_type, ir.VoidType): + Gen.builder.ret_void() + else: + Gen.builder.ret(error_ret_val) + Gen.builder.position_at_start(OkBB) + + def _fill_default_args(self, CallArgs: list[ir.Value], func: ir.Function | None, func_name: str | None = None) -> None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + global_defaults: dict = self.Trans._global_function_default_args + func_defaults: list | None = None + if func_name: + func_defaults = global_defaults.get(func_name) + if not func_defaults and func: + func_defaults = global_defaults.get(func.name) + if not func_defaults and func_name: + for gk, gv in global_defaults.items(): + if func_name in gk or gk.endswith(func_name) or gk.endswith('.' + func_name): + func_defaults = gv + break + if func and len(CallArgs) < len(func.args): + total_params: int = len(func.args) + if func_defaults: + num_defaults: int = len(func_defaults) + default_start: int = total_params - num_defaults + else: + num_defaults = 0 + default_start = total_params + while len(CallArgs) < total_params: + arg_idx: int = len(CallArgs) + arg = func.args[arg_idx] + param_type: ir.Type = arg.type if hasattr(arg, 'type') else arg + if func_defaults and arg_idx >= default_start: + def_idx: int = arg_idx - default_start + if def_idx < num_defaults and func_defaults[def_idx] is not None: + default_val = func_defaults[def_idx] + if isinstance(param_type, ir.IntType): + max_val: int = (1 << param_type.width) - 1 + if isinstance(default_val, bool): + default_val = 1 if default_val else 0 + if isinstance(default_val, int) and default_val > max_val: + default_val = default_val & max_val + if isinstance(default_val, int) and default_val < 0: + default_val = default_val & max_val + CallArgs.append(ir.Constant(param_type, default_val)) + elif isinstance(param_type, ir.PointerType): + CallArgs.append(ir.Constant(param_type, None)) + else: + try: + _zv = Gen._zero_value_for_type(param_type) + CallArgs.append(_zv if _zv is not None else ir.Constant(ir.IntType(32), 0)) + except (AssertionError, TypeError): + CallArgs.append(ir.Constant(ir.IntType(32), 0)) + else: + if isinstance(param_type, ir.PointerType): + CallArgs.append(ir.Constant(param_type, None)) + elif isinstance(param_type, ir.IntType): + CallArgs.append(ir.Constant(param_type, 0)) + else: + try: + _zv = Gen._zero_value_for_type(param_type) + CallArgs.append(_zv if _zv is not None else ir.Constant(ir.IntType(32), 0)) + except (AssertionError, TypeError): + CallArgs.append(ir.Constant(ir.IntType(32), 0)) + else: + if isinstance(param_type, ir.PointerType): + CallArgs.append(ir.Constant(param_type, None)) + elif isinstance(param_type, ir.IntType): + CallArgs.append(ir.Constant(param_type, 0)) + else: + try: + _zv = Gen._zero_value_for_type(param_type) + CallArgs.append(_zv if _zv is not None else ir.Constant(ir.IntType(32), 0)) + except (AssertionError, TypeError): + CallArgs.append(ir.Constant(ir.IntType(32), 0)) + + def _ApplyKeywordArgs(self, Node: ast.Call, CallArgs: list[ir.Value], func: ir.Function | None, func_name: str | None, self_offset: int = 1) -> None: + """在 _fill_default_args 之后调用：用 keyword args 按 param_names 覆盖 CallArgs 对应位置。 + self_offset=1 表示 CallArgs[0] 是 self，参数从 CallArgs[1] 开始（__init__/__new__/__call__/method）。 + self_offset=0 表示无 self，参数从 CallArgs[0] 开始（普通函数）。 + """ + if not Node.keywords: + return + global_param_names: dict = self.Trans._global_function_param_names + param_names: list[str] | None = None + if func_name: + param_names = global_param_names.get(func_name) + if not param_names and func is not None: + param_names = global_param_names.get(func.name) + if not param_names and func is not None and hasattr(func, 'args'): + llvm_names: list[str] = [a.name for a in func.args] if func.args else [] + if llvm_names and not all(n.isdigit() for n in llvm_names): + param_names = llvm_names + if not param_names: + # 无法获取参数名，回退：按顺序追加（保持旧行为，避免静默丢弃） + for kw in Node.keywords: + if kw.arg is None: + continue + ArgVal = self.HandleExprLlvm(kw.value) + if ArgVal: + CallArgs.append(ArgVal) + return + for kw in Node.keywords: + if kw.arg is None: + continue + ArgVal = self.HandleExprLlvm(kw.value) + if not ArgVal: + continue + if kw.arg in param_names: + idx: int = param_names.index(kw.arg) + target_idx: int = idx - self_offset + if 0 <= target_idx < len(CallArgs) - self_offset: + CallArgs[self_offset + target_idx] = ArgVal + + def _HandleCallLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + FuncAttr: str | None = None + ModulePath: str | None = None + if isinstance(Node.func, ast.Name): + FuncName = Node.func.id + if hasattr(self.Trans.FunctionHandler, '_generic_templates') and FuncName in self.Trans.FunctionHandler._generic_templates: + return self._HandleGenericCallLlvm(Node, FuncName) + result = self.Trans.ExprBuiltinHandle._HandleBuiltinCallLlvm(Node) + if result is not None: + return result + if Gen._has_function(FuncName): + return self._HandleClosureCallLlvm(Node, FuncName) + SymInfo = self.Trans.SymbolTable.lookup(FuncName) + if SymInfo: + # 名字冲突处理：类名可能与 CEnum 成员名冲突（如 ASTKind.Name 枚举成员与 Name 类同名） + # 若该名字同时是类（在 Gen.structs 中或存在 __before_init__），优先作为类构造函数处理 + IsAlsoClass: bool = (FuncName in Gen.structs) or Gen._has_function(f'{FuncName}.__before_init__') + if SymInfo.IsEnum and not (IsAlsoClass and getattr(SymInfo, 'IsEnumMember', False)): + if Node.args: + arg_val = self.HandleExprLlvm(Node.args[0]) + if arg_val: + if isinstance(arg_val.type, ir.IntType) and arg_val.type.width != 32: + if arg_val.type.width < 32: + arg_val = Gen.builder.zext(arg_val, ir.IntType(32), name="zext_enum_arg") + else: + arg_val = Gen.builder.trunc(arg_val, ir.IntType(32), name="trunc_enum_arg") + return arg_val + return ir.Constant(ir.IntType(32), 0) + if SymInfo.IsTypedef: + return self._HandleTypedefCastLlvm(Node, FuncName) + NewFuncName = f'{FuncName}.__before_init__' + InitFuncName = f'{FuncName}.__init__' + if Gen._has_function(NewFuncName): + return self._HandleClassNewLlvm(Node, FuncName) + if Gen._has_function(InitFuncName) and FuncName in Gen.structs: + return self._HandleClassNewLlvm(Node, FuncName) + if FuncName in Gen.structs: + return self._HandleClassNewLlvm(Node, FuncName) + if hasattr(self.Trans.ClassHandler, '_generic_class_templates') and FuncName in self.Trans.ClassHandler._generic_class_templates: + return self._HandleClassNewLlvm(Node, FuncName) + if self.Trans.FunctionDefCache.get(FuncName): + return self._HandleClosureCallLlvm(Node, FuncName) + if FuncName in self._C_LIB_FUNCS: + return self._HandleClosureCallLlvm(Node, FuncName) + ClassName = Gen.var_struct_class.get(FuncName) + if ClassName and Gen._has_function(f'{ClassName}.__call__'): + return self._HandleInstanceCallLlvm(Node, FuncName, ClassName) + t_c_imported = getattr(self.Trans, '_t_c_imported_names', {}) + if FuncName in t_c_imported: + src_module, src_name = t_c_imported[FuncName] + virtual_node = self._make_virtual_attr_call(Node, src_module, src_name) + return self._HandleCallLlvm(virtual_node) + # 检查是否是全局变量中的函数指针（t.Callable 类型在模块级定义） + if FuncName not in Gen.variables and FuncName in Gen.global_vars: + Gen._Load_var(FuncName) + if FuncName not in Gen.variables and FuncName in Gen.module.globals: + GVar = Gen.module.globals[FuncName] + if not isinstance(GVar, ir.Function): + StoredType = GVar.type.pointee if isinstance(GVar.type, ir.PointerType) else GVar.type + if isinstance(StoredType, ir.PointerType) and isinstance(StoredType.pointee, ir.FunctionType): + Gen.variables[FuncName] = GVar + # 检查是否是闭包变量（lambda 赋值给变量后调用） + if FuncName in Gen.variables and Gen.variables[FuncName] is not None: + VarPtr = Gen.variables[FuncName] + # 检查是否是函数指针变量（t.Callable 类型），直接加载并调用 + if isinstance(VarPtr, ir.GlobalVariable) and isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, ir.PointerType) and isinstance(VarPtr.type.pointee.pointee, ir.FunctionType): + fn_ptr = Gen._load(VarPtr, name=f"Load_fn_{FuncName}") + args = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + args.append(ArgVal) + return Gen.builder.call(fn_ptr, args, name=f"call_{FuncName}") + closure_ptr = Gen._load(VarPtr, name=f"Load_closure_{FuncName}") + if isinstance(closure_ptr.type, ir.PointerType) and isinstance(closure_ptr.type.pointee, ir.FunctionType): + fn_ptr = closure_ptr + args = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + args.append(ArgVal) + return Gen.builder.call(fn_ptr, args, name=f"call_{FuncName}") + if isinstance(closure_ptr.type, ir.PointerType) and isinstance(closure_ptr.type.pointee, ir.LiteralStructType): + st = closure_ptr.type.pointee + if len(st.elements) == 2 and isinstance(st.elements[1], ir.PointerType) and isinstance(st.elements[1].pointee, ir.FunctionType): + fn_ptr_type = st.elements[1] + fn_ptr = Gen.builder.gep(closure_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 1)], name=f"fn_ptr_{FuncName}") + fn_val = Gen._load(fn_ptr, name=f"fn_{FuncName}") + args = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + args.append(ArgVal) + env_ptr = Gen.builder.gep(closure_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"env_ptr_{FuncName}") + env_val = Gen._load(env_ptr, name=f"env_{FuncName}") + call_args = [env_val] + args + expected_fn_type = fn_ptr_type.pointee + if len(call_args) < len(expected_fn_type.args): + while len(call_args) < len(expected_fn_type.args): + call_args.append(ir.Constant(expected_fn_type.args[len(call_args)], 0)) + # 如果函数期望的 env 类型与闭包中存储的 i8* 不同，做 bitcast + if len(expected_fn_type.args) > 0 and call_args[0].type != expected_fn_type.args[0]: + call_args[0] = Gen.builder.bitcast(call_args[0], expected_fn_type.args[0], name=f"cast_env_{FuncName}") + return Gen.builder.call(fn_val, call_args, name=f"call_{FuncName}") + raise Exception(f"Undefined function: '{FuncName}'") + if isinstance(Node.func, ast.Subscript): + # Handle list[int](mb) syntax — generic class constructor with explicit type args + sub: ast.Subscript = Node.func + if isinstance(sub.value, ast.Name): + ClassName: str = sub.value.id + if hasattr(self.Trans.ClassHandler, '_generic_class_templates') and ClassName in self.Trans.ClassHandler._generic_class_templates: + return self._HandleGenericClassNewLlvm(Node, ClassName, explicit_slice=sub.slice) + if isinstance(Node.func, ast.Attribute): + if Node.func.attr in ('update', 'final', 'transform'): + pass + if Node.func.attr == '__sizeof__': + if isinstance(Node.func.value, ast.Subscript): + spec_name = self._ResolveGenericSlice(Node.func.value) + if spec_name: + return self._HandleSizeofLlvm(ast.Name(id=spec_name, ctx=ast.Load())) + # 对于 BinOp/BitOr 联合类型（如 AST | t.CPtr）， + # sizeof 应返回指针大小 8，而非结构体大小 + # 必须在 HandleExprLlvm 之前检查，避免对类型名求值产生副作用 + if isinstance(Node.func.value, ast.BinOp) and isinstance(Node.func.value.op, ast.BitOr): + if self._AnnotationContainsCPtr(Node.func.value): + return ir.Constant(ir.IntType(64), 8) + # 对于 ast.Attribute（如 ast.ClassDef.__sizeof__() / t.CInt.__sizeof__()）， + # 直接调用 _HandleSizeofLlvm，不走 HandleExprLlvm + # 因为 ast.Attribute 在 sizeof 上下文中一定是类型名，不是变量 + if isinstance(Node.func.value, ast.Attribute): + type_name = Node.func.value.attr + SizeofNode = ast.Name(id=type_name, ctx=ast.Load()) + _r = self._HandleSizeofLlvm(SizeofNode) + return _r + # 对于 ast.Name，需要区分变量和类型名： + # - 如果是已知变量（Gen.variables/_reg_values/_direct_values/self）， + # 走 HandleExprLlvm 路径（如 myvar.__sizeof__()） + # - 否则视为类型名，直接调用 _HandleSizeofLlvm + # 这避免了 HandleExprLlvm 对类型名求值产生副作用 + # （如修改 builder 状态、污染 variables 缓存等）， + # 这些副作用会导致泛型特化后 then 块为空 + if isinstance(Node.func.value, ast.Name): + nm: str = Node.func.value.id + is_variable: bool = ( + nm == 'self' + or nm in Gen.variables + or nm in getattr(Gen, '_reg_values', {}) + or nm in getattr(Gen, '_direct_values', {}) + ) + if not is_variable: + SizeofNode = ast.Name(id=nm, ctx=ast.Load()) + _r = self._HandleSizeofLlvm(SizeofNode) + return _r + # 变量/实例的 sizeof: 通过 HandleExprLlvm 获取值 + ObjVal = self.HandleExprLlvm(Node.func.value) + if ObjVal and isinstance(ObjVal.type, ir.PointerType): + pointee = ObjVal.type.pointee + if isinstance(pointee, ir.ArrayType): + # Calculate element size in bytes + elem = pointee.element + if isinstance(elem, ir.IntType): + elem_size = elem.width // 8 + elif isinstance(elem, ir.PointerType): + elem_size = 8 # 64-bit pointers + elif isinstance(elem, (ir.LiteralStructType, ir.IdentifiedStructType)): + # Approximate: sum of field sizes + elem_size = 0 + for field in getattr(elem, 'elements', []): + if isinstance(field, ir.IntType): + elem_size += field.width // 8 + elif isinstance(field, ir.PointerType): + elem_size += 8 + elif isinstance(field, ir.ArrayType): + if isinstance(field.element, ir.IntType): + elem_size += field.count * (field.element.width // 8) + else: + elem_size += field.count * 8 + else: + elem_size += 8 + elif isinstance(elem, ir.ArrayType): + if isinstance(elem.element, ir.IntType): + elem_size = elem.count * (elem.element.width // 8) + else: + elem_size = elem.count * 8 + elif isinstance(elem, ir.FloatType): + elem_size = 4 + elif isinstance(elem, ir.DoubleType): + elem_size = 8 + else: + elem_size = 8 # fallback + return ir.Constant(ir.IntType(64), pointee.count * elem_size) + elif isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + # 结构体实例: 直接计算大小 + self._resolve_opaque_struct_members(pointee, Gen) + size = Gen._get_struct_size(pointee) + if size > 0: + return ir.Constant(ir.IntType(64), size) + # 回退: 尝试作为类型名处理 + if isinstance(Node.func.value, ast.Name): + type_name = Node.func.value.id + SizeofNode = ast.Name(id=type_name, ctx=ast.Load()) + return self._HandleSizeofLlvm(SizeofNode) + ModulePath = self._get_ModulePath(Node.func.value) + if ModulePath: + first_part = ModulePath.split('.')[0] + resolved_first = self.Trans.SymbolTable.resolve_alias(first_part) + if resolved_first != first_part: + ModulePath = resolved_first + ModulePath[len(first_part):] + # 统一获取模块 SHA1，用于跨模块类型引用（确保用源模块的 SHA1 而非当前模块） + # 放在 ModulePath 和 alias 解析之后，供所有 _HandleClassNewLlvm/_ensure_struct_declared 调用点使用 + _ctor_module_sha1: str | None = None + if ModulePath and ModulePath not in {'t', 'c'}: + _ctor_sha1_map: dict[str, str] = getattr(Gen, 'ModuleSha1Map', {}) + _ctor_module_sha1 = _ctor_sha1_map.get(ModulePath) + if not _ctor_module_sha1 and '.' in ModulePath: + _ctor_module_sha1 = _ctor_sha1_map.get(ModulePath.split('.')[-1]) + if ModulePath: + is_instance_var = isinstance(Node.func.value, ast.Name) and ( + Node.func.value.id in Gen.var_struct_class + or Node.func.value.id in Gen.variables + or Node.func.value.id in Gen.global_vars + ) and Node.func.value.id not in Gen.ModuleSha1Map + is_class_name = isinstance(Node.func.value, ast.Name) and (Node.func.value.id in Gen.structs or self.Trans.SymbolTable.is_struct(Node.func.value.id)) + # 链式方法调用：如 self.__mbuddy__.alloc() 或 trans.ImportsH.HandleImport(child) + # Node.func.value 是 ast.Attribute，基础是实例变量（如 self/trans） + _is_chained_instance: bool = False + if isinstance(Node.func.value, ast.Attribute): + _base_node: ast.AST = Node.func.value + while isinstance(_base_node, ast.Attribute): + _base_node = _base_node.value + if isinstance(_base_node, ast.Name): + _base_name: str = _base_node.id + # 检查基名是否为已知变量（self、局部变量、全局变量、已知 struct 变量） + # 且不在 ModuleSha1Map 中（避免把模块名误判为变量） + if _base_name == 'self' or ( + (_base_name in Gen.var_struct_class + or _base_name in Gen.variables + or _base_name in Gen.global_vars) + and _base_name not in Gen.ModuleSha1Map + ): + _is_chained_instance = True + if (is_instance_var or _is_chained_instance) and not is_class_name: + return self._HandleMethodCallLlvm(Node) + FuncAttr = Node.func.attr + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + if hasattr(self.Trans.ClassHandler, '_generic_class_templates') and FuncAttr in self.Trans.ClassHandler._generic_class_templates: + return self._HandleGenericClassNewLlvm(Node, FuncAttr) + # 跨模块 REnum 变体构造：ClassName.Variant(args) + # 直接检查 Node.func.value 是否是 REnum 类名（不依赖变体短名注册） + if isinstance(Node.func.value, ast.Name): + RenumClassName: str = Node.func.value.id + if self.Trans.SymbolTable.has(RenumClassName): + RenumInfo = self.Trans.SymbolTable[RenumClassName] + if RenumInfo.IsRenum and hasattr(RenumInfo, 'RenumVariants') and RenumInfo.RenumVariants: + if FuncAttr in RenumInfo.RenumVariants: + TagValue: int = RenumInfo.RenumVariants.index(FuncAttr) + result = self._HandleREnumConstructLlvm(Node, RenumClassName, FuncAttr, TagValue) + if result is not None: + return result + if self.Trans.SymbolTable.has(FuncAttr): + SymInfo = self.Trans.SymbolTable[FuncAttr] + if SymInfo.IsStruct: + if FuncAttr not in Gen.structs: + self._ensure_struct_declared(FuncAttr, module_sha1=_ctor_module_sha1) + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + if SymInfo.IsEnumMember and SymInfo.EnumName: + # When called as module.func(args), check if module.func is a + # function/variable first. If so, skip REnum construction and + # let the function call path handle it. This fixes cases where + # a factory function shares a name with an REnum variant + # (e.g., llvmlite.Ptr vs LLVMType.Ptr). + _SkipRenum: bool = False + if ModulePath: + _FullKey = f"{ModulePath}.{FuncAttr}" + _FullSym = self.Trans.SymbolTable.lookup(_FullKey) + if _FullSym and (_FullSym.IsFunction or _FullSym.IsVariable): + _SkipRenum = True + if not _SkipRenum: + EnumName = SymInfo.EnumName + if self.Trans.SymbolTable.has(EnumName): + EnumInfo = self.Trans.SymbolTable[EnumName] + if EnumInfo.IsRenum: + result = self._HandleREnumConstructLlvm(Node, EnumName, FuncAttr, SymInfo.value) + if result is not None: + return result + FullAttrKey = f"{ModulePath}.{FuncAttr}" + if self.Trans.SymbolTable.has(FullAttrKey): + SymInfo = self.Trans.SymbolTable[FullAttrKey] + if SymInfo.IsStruct: + if FuncAttr not in Gen.structs: + self._ensure_struct_declared(FuncAttr, module_sha1=_ctor_module_sha1) + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + # 跨模块 REnum 变体构造：FullName 查找（如 LLVMType.Int） + if SymInfo.IsEnumMember and SymInfo.EnumName: + EnumName = SymInfo.EnumName + if self.Trans.SymbolTable.has(EnumName): + EnumInfo = self.Trans.SymbolTable[EnumName] + if EnumInfo.IsRenum: + result = self._HandleREnumConstructLlvm(Node, EnumName, FuncAttr, SymInfo.value) + if result is not None: + return result + if is_class_name: + ClassName = Node.func.value.id + result = self._HandleStaticMethodCallLlvm(Node, ClassName, FuncAttr) + if result is not None: + return result + FullAttrKey = f"{ModulePath}.{FuncAttr}" + if self.Trans.SymbolTable.has(FullAttrKey): + SymInfo = self.Trans.SymbolTable[FullAttrKey] + if SymInfo.IsTypedef: + return ir.Constant(ir.IntType(32), 0) + if SymInfo.IsStruct: + if FuncAttr not in Gen.structs: + self._ensure_struct_declared(FuncAttr, module_sha1=_ctor_module_sha1) + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + if ModulePath == 'c': + if FuncAttr == 'Asm': + return self.Trans.ExprAsmHandle._HandleAsmLlvm(Node) + elif FuncAttr == 'Addr': + return self.Trans.CSpecialCallHandle._HandleCAddrLlvm(Node) + elif FuncAttr == 'Set': + return self.Trans.CSpecialCallHandle._HandleCSetLlvm(Node) + elif FuncAttr == 'Load': + return self.Trans.CSpecialCallHandle._HandleCLoadLlvm(Node) + elif FuncAttr == 'Deref': + return self.Trans.CSpecialCallHandle._HandleCDerefLlvm(Node) + elif FuncAttr == 'DerefAs': + return self._HandleCDerefAsLlvm(Node) + elif FuncAttr == 'PtrToInt': + return self.Trans.CSpecialCallHandle._HandleCPtrToIntLlvm(Node) + elif FuncAttr in ('CIf', 'CElif'): + return self.Trans.CSpecialCallHandle._HandleCIfLlvm(Node) + elif FuncAttr == 'CError': + msg = "compile-time error" + if Node.args: + arg = Node.args[0] + if isinstance(arg, ast.Constant) and isinstance(arg.value, str): + msg = arg.value + lineno = getattr(Node, 'lineno', 0) + raise Exception(f"#error: {msg} (line {lineno})") + elif FuncAttr in ('AsmInp', 'AsmOut'): + return ir.Constant(ir.IntType(32), 0) + elif FuncAttr == 'LLVMIR': + return self._HandleLLVMIRLlvm(Node) + elif FuncAttr in ('LInp', 'LOut'): + return ir.Constant(ir.IntType(32), 0) + if ModulePath == 't': + MethodResult_t = self._HandleMethodCallLlvm(Node) + if MethodResult_t is not None: + return MethodResult_t + raise Exception(f"Undefined method: 't.{FuncAttr}'") + if isinstance(Node.func.value, ast.Attribute): + inner_attr = Node.func.value.attr + if inner_attr in Gen.structs or self.Trans.SymbolTable.is_struct(inner_attr): + static_result = self._HandleStaticMethodCallLlvm(Node, inner_attr, FuncAttr) + if static_result is not None: + return static_result + MethodResult_attr = self._HandleMethodCallLlvm(Node) + is_zero_attr = isinstance(MethodResult_attr, ir.Constant) and isinstance(MethodResult_attr.type, ir.IntType) and MethodResult_attr.constant == 0 + if MethodResult_attr is not None and not is_zero_attr: + return MethodResult_attr + result = self._HandleExternalCallLlvm(Node, FuncAttr, ModulePath) + if result is not None: + return result + MethodResult = self._HandleMethodCallLlvm(Node) + IsZeroResult = isinstance(MethodResult, ir.Constant) and isinstance(MethodResult.type, ir.IntType) and MethodResult.constant == 0 + if IsZeroResult and FuncAttr is not None: + mangled_name = Gen._mangle_func_name(FuncAttr, ModulePath if ModulePath and ModulePath != 't' else None) + if Gen._has_function(mangled_name): + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + return Gen.builder.call(Gen._get_function(mangled_name), CallArgs, name=f"call_{FuncAttr}") + if MethodResult is None or IsZeroResult: + imported = getattr(self.Trans, '_ImportedModules', set()) + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + IsKnown = (ModulePath in imported or ModulePath in self.Trans.SymbolTable.import_aliases or + ModulePath in ModuleSha1Map) + if not IsKnown and ModulePath and '.' in ModulePath: + LastPart = ModulePath.split('.')[-1] + IsKnown = (LastPart in imported or LastPart in self.Trans.SymbolTable.import_aliases or + LastPart in ModuleSha1Map) + # 最后尝试: 用 FuncAttr 直接查找 Gen.functions（_find_function 会遍历 SHA1 前缀） + if FuncAttr and ModulePath and ModulePath not in {'t', 'c'}: + # _ctor_module_sha1 已在函数顶部统一获取，此处无需重复 + # 先检查 FuncAttr 是否是函数/变量，如果是则跳过 struct 声明 + _attr_sym = self.Trans.SymbolTable.lookup(FuncAttr) + _is_func_or_var = _attr_sym and (_attr_sym.IsFunction or _attr_sym.IsVariable) + if not _is_func_or_var and ModulePath: + _full_key = f"{ModulePath}.{FuncAttr}" + _full_sym = self.Trans.SymbolTable.lookup(_full_key) + if _full_sym and (_full_sym.IsFunction or _full_sym.IsVariable): + _is_func_or_var = True + # 优先检查 FuncAttr 是否为 struct/类构造器 + if not _is_func_or_var and FuncAttr not in Gen.structs: + self._ensure_struct_declared(FuncAttr, module_sha1=_ctor_module_sha1) + if not _is_func_or_var and FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + # 检查 FullAttrKey 是否为 struct（如 hashlib.md5） + if ModulePath: + FullAttrKey = f"{ModulePath}.{FuncAttr}" + if self.Trans.SymbolTable.has(FullAttrKey): + FullSymInfo = self.Trans.SymbolTable[FullAttrKey] + if FullSymInfo.IsStruct or FullSymInfo.IsCpythonObject: + if FuncAttr not in Gen.structs: + self._ensure_struct_declared(FuncAttr, module_sha1=_ctor_module_sha1) + if FuncAttr in Gen.structs: + result = self._HandleClassNewLlvm(Node, FuncAttr, module_sha1=_ctor_module_sha1) + if result is not None: + return result + fallback_func = Gen._find_function(FuncAttr) + if fallback_func is not None: + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + self._append_eh_msg_out_arg(CallArgs, fallback_func, Gen) + result = Gen.builder.call(fallback_func, CallArgs, name=f"call_{FuncAttr}") + return self._check_eh_return(result, FuncAttr, Gen, called_func=fallback_func) + # 尝试从 stub 按需加载函数声明 + if ModulePath in ModuleSha1Map: + sha1 = ModuleSha1Map[ModulePath] + stub_mangled = f"{sha1}.{FuncAttr}" + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(stub_mangled, Gen) + except Exception: + stub_func_type = None + if not stub_func_type: + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(FuncAttr, Gen) + except Exception: + stub_func_type = None + if stub_func_type: + decl_name = stub_mangled if stub_mangled in getattr(self.Trans.ImportHandler, '_stub_func_cache', {}) else FuncAttr + func_decl = self.GetOrCreateStubFuncDecl(decl_name, stub_func_type, Gen, alias=FuncAttr) + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + self._append_eh_msg_out_arg(CallArgs, func_decl, Gen) + result = Gen.builder.call(func_decl, CallArgs, name=f"call_{FuncAttr}") + return self._check_eh_return(result, FuncAttr, Gen, called_func=func_decl) + if ModulePath and IsKnown: + raise Exception(f"Undefined symbol: '{ModulePath}.{FuncAttr}'") + if ModulePath and ModulePath not in {'t', 'c'} and not IsKnown: + raise Exception(f"Unknown module: '{ModulePath}' (undefined symbol: '{ModulePath}.{FuncAttr}')") + if not ModulePath and FuncAttr: + raise Exception(f"Undefined method: '{FuncAttr}'") + return MethodResult if MethodResult is not None else ir.Constant(ir.IntType(32), 0) + if isinstance(Node.func, ast.BinOp) and isinstance(Node.func.op, ast.BitOr): + return self._HandleTypeUnionCastLlvm(Node) + raise Exception(f"Unsupported call expression") + + def _ensure_struct_declared(self, class_name: str, module_sha1: str | None = None) -> None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + # 如果名称是函数或变量，不应创建 struct + SymInfo = self.Trans.SymbolTable.lookup(class_name) + if SymInfo and (SymInfo.IsFunction or SymInfo.IsVariable): + return + # 跨模块类型引用：如果提供了 module_sha1，优先检查带 SHA1 前缀的完整名 + full_key: str | None = f"{module_sha1}.{class_name}" if module_sha1 else None + if full_key and full_key in Gen.structs: + existing = Gen.structs[full_key] + if isinstance(existing, ir.IdentifiedStructType) and (existing.elements is None or len(existing.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(class_name, Gen, source_sha1=module_sha1) + return + if class_name in Gen.structs: + existing = Gen.structs[class_name] + if isinstance(existing, ir.IdentifiedStructType) and (existing.elements is None or len(existing.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(class_name, Gen, source_sha1=module_sha1) + existing = Gen.structs.get(class_name, existing) + if isinstance(existing, ir.IdentifiedStructType) and existing.elements is not None and len(existing.elements) > 0: + return + else: + # 短名已存在但 SHA1 可能不匹配：如果 module_sha1 与现有 struct 的 SHA1 不一致， + # 需要用正确的 SHA1 创建独立 struct + if module_sha1: + _existing_sha1: str = Gen._extract_struct_sha1(existing) + if _existing_sha1 and _existing_sha1 != module_sha1: + # SHA1 不匹配，用正确的 SHA1 创建 + correct_st = Gen._get_or_create_struct(class_name, source_sha1=module_sha1) + self.Trans.ImportHandler._TryLoadStructFromStub(class_name, Gen, source_sha1=module_sha1) + return + SymInfo = self.Trans.SymbolTable.lookup(class_name) + if SymInfo and SymInfo.Members: + member_types = [] + if isinstance(SymInfo.Members, dict): + for member_name, member_info in SymInfo.Members.items(): + if isinstance(member_info, CTypeInfo): + mt = member_info.ToLLVM(Gen) + elif isinstance(member_info, str): + mt = Gen._CType2LLVM(member_info, '*' in member_info) + else: + mt = Gen._CType2LLVM(str(member_info) if member_info else 'int', False) + if isinstance(mt, ir.VoidType): + mt = ir.IntType(8) + member_types.append(mt) + else: + for item in SymInfo.Members: + if isinstance(item, (list, tuple)) and len(item) == 2: + member_name, member_type_str = item + mt = Gen._CType2LLVM(member_type_str, '*' in member_type_str) + if isinstance(mt, ir.VoidType): + mt = ir.IntType(8) + member_types.append(mt) + if member_types: + st = Gen._get_or_create_struct(class_name, source_sha1=module_sha1) + if isinstance(st, ir.IdentifiedStructType) and (st.elements is None or len(st.elements) == 0): + st.set_body(*member_types) + else: + Gen._get_or_create_struct(class_name, source_sha1=module_sha1) + else: + if class_name not in Gen.structs and not (full_key and full_key in Gen.structs): + self.Trans.ImportHandler._TryLoadStructFromStub(class_name, Gen, source_sha1=module_sha1) + Gen._get_or_create_struct(class_name, source_sha1=module_sha1) + + def _HandleStaticMethodCallLlvm(self, Node: ast.Call, ClassName: str, MethodName: str) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + class_sha1 = struct_sha1_map.get(ClassName) + if class_sha1: + MangledName = f'{class_sha1}.{ClassName}.{MethodName}' + else: + MangledName = f'{ClassName}.{MethodName}' + func = Gen._find_function(MangledName) + if not func: + func = Gen._find_function(f'{MangledName}__') + if not func: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(MangledName, Gen) + if stub_func_type: + func = self.GetOrCreateStubFuncDecl(MangledName, stub_func_type, Gen) + if not func and class_sha1: + plain_name = f'{ClassName}.{MethodName}' + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(plain_name, Gen) + if stub_func_type: + func = self.GetOrCreateStubFuncDecl(MangledName, stub_func_type, Gen) + if not func: + if ClassName not in Gen.structs: + self._ensure_struct_declared(ClassName) + SymInfo = self.LookupFunctionSymbol(MethodName, module_path=ClassName) + if SymInfo and SymInfo.IsFunction: + sig = self.BuildLLVMFuncTypeFromSig(SymInfo, Gen) + if sig is not None: + ret_type, llvm_param_types, _ = sig + is_static = FuncMeta.STATIC_METHOD in SymInfo.MetaList + if ClassName in Gen.structs and not is_static: + llvm_param_types.insert(0, ir.PointerType(Gen.structs[ClassName])) + func_type = ir.FunctionType(ret_type, llvm_param_types) + func = ir.Function(Gen.module, func_type, name=MangledName) + Gen.functions[MangledName] = func + if not func: + p = Gen.class_parent.get(ClassName) + while p: + parent_mangled = Gen._mangle_name(f'{p}.{MethodName}') + parent_func = Gen.functions.get(parent_mangled) + if parent_func: + func = parent_func + break + parent_func = Gen._find_function(f'{p}.{MethodName}') + if parent_func: + func = parent_func + break + p = Gen.class_parent.get(p) + if not func: + return None + CallArgs = [] + SymInfo = self.LookupFunctionSymbol(MethodName, module_path=ClassName) + is_static_call = SymInfo is not None and FuncMeta.STATIC_METHOD in SymInfo.MetaList + is_classmethod_call = SymInfo is not None and FuncMeta.CLASS_METHOD in SymInfo.MetaList + # @classmethod: 在参数列表前插入 cls（栈上分配的类实例指针） + if is_classmethod_call and ClassName in Gen.structs: + ClsPtr = Gen._alloca(Gen.structs[ClassName], name=f"cls_{ClassName}") + CallArgs.append(ClsPtr) + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + if (ClassName not in Gen.class_vtable + and ClassName not in Gen._cross_module_vtable_classes + and class_sha1): + self._try_register_vtable_ondemand(ClassName, class_sha1, Gen) + if (ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes) and ClassName in Gen.class_methods and CallArgs and not is_static_call: + methods = Gen.class_methods[ClassName] + if class_sha1: + FullMethodName = f'{class_sha1}.{ClassName}.{MethodName}' + else: + FullMethodName = f'{ClassName}.{MethodName}' + method_idx = -1 + for mi, mn in enumerate(methods): + if mn == FullMethodName or mn == MethodName or mn.endswith(f'.{MethodName}'): + method_idx = mi + break + if method_idx >= 0: + ObjVal = CallArgs[0] + if self._is_char_pointer(ObjVal): + if ClassName in Gen.structs: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}_static") + CallArgs[0] = ObjVal + VtableSlotPtr = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"vtable_slot_{ClassName}_static") + VtablePtr = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}_static") + VtableArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + VtableTyped = Gen.builder.bitcast(VtablePtr, ir.PointerType(VtableArrayType), name=f"vtable_typed_{ClassName}_static") + MethodPtrAddr = Gen.builder.gep(VtableTyped, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), method_idx)], name=f"vmethod_ptr_addr_{MethodName}_static") + MethodPtrI8 = Gen._load(MethodPtrAddr, name=f"vmethod_ptr_{MethodName}_static") + FuncPtrType = ir.PointerType(func.function_type) + MethodPtrTyped = Gen.builder.bitcast(MethodPtrI8, FuncPtrType, name=f"vmethod_typed_{MethodName}_static") + self._fill_default_args(CallArgs, func, FullMethodName) + self._append_eh_msg_out_arg(CallArgs, func, Gen) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(MethodPtrTyped, adjusted, name=f"vcall_{FullMethodName}") + self._check_eh_return(result, FullMethodName, Gen, called_func=func) + return result + self._append_eh_msg_out_arg(CallArgs, func, Gen) + CallArgs = Gen._apply_auto_addr(MangledName, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{MangledName}") + self._check_eh_return(result, MangledName, Gen, called_func=func) + return result + + def _HandleInstanceCallLlvm(self, Node: ast.Call, VarName: str, ClassName: str) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + ObjVal = self.HandleExprLlvm(ast.Name(id=VarName, ctx=ast.Load())) + if not ObjVal: + raise Exception(f"Cannot evaluate object '{VarName}' for __call__") + if self._is_char_pointer(ObjVal): + if ClassName in Gen.structs: + ObjVal = Gen.builder.bitcast(ObjVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + CallArgs = [ObjVal] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + CallFuncName = f'{ClassName}.__call__' + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + class_sha1 = struct_sha1_map.get(ClassName) + if class_sha1: + CallFuncName = f'{class_sha1}.{ClassName}.__call__' + if not Gen._has_function(CallFuncName): + raise Exception(f"Class '{ClassName}' has no __call__ method") + func = Gen._get_function(CallFuncName) + self._fill_default_args(CallArgs, func, CallFuncName) + self._ApplyKeywordArgs(Node, CallArgs, func, CallFuncName, self_offset=1) + self._append_eh_msg_out_arg(CallArgs, func, Gen) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{CallFuncName}") + self._check_eh_return(result, CallFuncName, Gen, called_func=func) + return result + + def _make_virtual_attr_call(self, Node: ast.Call, module_name: str, attr_name: str) -> ast.Call: + virtual_func = ast.Attribute( + value=ast.Name(id=module_name, ctx=ast.Load()), + attr=attr_name, + ctx=ast.Load() + ) + new_node = ast.Call( + func=virtual_func, + args=Node.args, + keywords=Node.keywords + ) + ast.copy_location(new_node, Node) + return new_node + + def _get_ModulePath(self, node: ast.AST) -> str | None: + parts = [] + while isinstance(node, ast.Attribute): + parts.append(node.attr) + node = node.value + if isinstance(node, ast.Name): + parts.append(node.id) + parts.reverse() + return '.'.join(parts) if parts else None + + def _HandleLLVMIRLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not Node.args: + return ir.Constant(ir.IntType(32), 0) + + all_ops = [] + input_ops = [] + output_targets = [] + operand_seq = 0 + + ret_type = ir.IntType(32) + if len(Node.args) >= 2: + ret_type_arg = Node.args[1] + inferred = self.Trans.ExprAsmHandle._infer_expr_llvm_type(ret_type_arg) + if inferred: + ret_type = inferred + + ir_template = "" + first_arg = Node.args[0] + + if isinstance(first_arg, ast.JoinedStr): + parts = [] + for value in first_arg.values: + if isinstance(value, ast.Constant) and isinstance(value.value, str): + parts.append(value.value) + elif isinstance(value, ast.FormattedValue): + expr = value.value + if isinstance(expr, ast.Call) and isinstance(expr.func, ast.Attribute): + if isinstance(expr.func.value, ast.Name) and expr.func.value.id == 'c': + if expr.func.attr == 'LInp': + if expr.args: + val = self.HandleExprLlvm(expr.args[0]) + if val: + input_ops.append(val) + all_ops.append(('inp', len(input_ops) - 1)) + parts.append(f'%__OP{operand_seq}__') + operand_seq += 1 + elif expr.func.attr == 'LOut': + if expr.args: + target_ptr = self.Trans.ExprAsmHandle._get_var_ptr_for_asm(expr.args[0]) + if target_ptr: + val = target_ptr + else: + val = self.HandleExprLlvm(expr.args[0]) + if val: + output_targets.append(val) + all_ops.append(('out', len(output_targets) - 1)) + parts.append(f'%__OP{operand_seq}__') + operand_seq += 1 + else: + fallback = self.HandleExprLlvm(expr) + if fallback: + input_ops.append(fallback) + all_ops.append(('inp', len(input_ops) - 1)) + parts.append(f'%__OP{operand_seq}__') + operand_seq += 1 + else: + fallback = self.HandleExprLlvm(expr) + if fallback: + input_ops.append(fallback) + all_ops.append(('inp', len(input_ops) - 1)) + parts.append(f'%__OP{operand_seq}__') + operand_seq += 1 + ir_template = ''.join(parts) + elif isinstance(first_arg, ast.Constant) and isinstance(first_arg.value, str): + ir_template = first_arg.value + + return self._EmitLLVMIRInstruction(Gen, ir_template, input_ops, output_targets, all_ops, ret_type) + + def _EmitLLVMIRInstruction(self, Gen: LlvmGeneratorMixin, ir_template: str, input_ops: list[ir.Value], output_targets: list[ir.Value], all_ops: list[tuple], ret_type: ir.Type) -> ir.Value: + template = ir_template.strip() + + def resolve_op(idx): + if idx < len(all_ops): + kind, local_idx = all_ops[idx] + if kind == 'inp' and local_idx < len(input_ops): + return input_ops[local_idx] + elif kind == 'out' and local_idx < len(output_targets): + return output_targets[local_idx] + return None + + assign_match = re.match(r'%__OP(\d+)__\s*=\s*(.+)', template) + if assign_match: + template = assign_match.group(2).strip() + + binop_map = { + 'add': 'add', 'fadd': 'fadd', 'sub': 'sub', 'fsub': 'fsub', + 'mul': 'mul', 'fmul': 'fmul', 'udiv': 'udiv', 'sdiv': 'sdiv', + 'fdiv': 'fdiv', 'urem': 'urem', 'srem': 'srem', 'frem': 'frem', + 'shl': 'shl', 'lshr': 'lshr', 'ashr': 'ashr', + 'and': 'and_', 'or': 'or_', 'xor': 'xor_', + } + + icmp_match = re.match(r'icmp\s+(\w+)\s+(\w+)\s+%__OP(\d+)__,\s*%__OP(\d+)__', template) + if icmp_match: + pred = icmp_match.group(1) + op1 = resolve_op(int(icmp_match.group(3))) + op2 = resolve_op(int(icmp_match.group(4))) + if op1 and op2: + result = Gen.builder.icmp_signed(pred, op1, op2, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + fcmp_match = re.match(r'fcmp\s+(\w+(?:\s+fast)?)\s+(\w+)\s+%__OP(\d+)__,\s*%__OP(\d+)__', template) + if fcmp_match: + pred = fcmp_match.group(1) + op1 = resolve_op(int(fcmp_match.group(3))) + op2 = resolve_op(int(fcmp_match.group(4))) + if op1 and op2: + result = Gen.builder.fcmp_ordered(pred, op1, op2, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + for op_name, method_name in binop_map.items(): + pattern = re.compile( + rf'{op_name}\s+(\w+)\s+%__OP(\d+)__,\s*%__OP(\d+)__', + re.IGNORECASE if op_name in ('and', 'or', 'xor') else 0 + ) + m = pattern.match(template) + if m: + op1 = resolve_op(int(m.group(2))) + op2 = resolve_op(int(m.group(3))) + if op1 and op2: + method = getattr(Gen.builder, method_name, None) + if method: + result = method(op1, op2, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + select_match = re.match(r'select\s+i1\s+%__OP(\d+)__,\s*(\w+)\s+%__OP(\d+)__,\s*(\w+)\s+%__OP(\d+)__', template) + if select_match: + cond = resolve_op(int(select_match.group(1))) + true_val = resolve_op(int(select_match.group(3))) + false_val = resolve_op(int(select_match.group(5))) + if cond and true_val and false_val: + result = Gen.builder.select(cond, true_val, false_val, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + cast_map = { + 'zext': 'zext', 'sext': 'sext', 'trunc': 'trunc', + 'bitcast': 'bitcast', 'inttoptr': 'inttoptr', 'ptrtoint': 'ptrtoint', + 'uitofp': 'uitofp', 'sitofp': 'sitofp', 'fptoui': 'fptoui', 'fptosi': 'fptosi', + 'fpext': 'fpext', 'fptrunc': 'fptrunc', + } + for op_name, method_name in cast_map.items(): + pattern = re.compile(rf'{op_name}\s+(\w+)\s+%__OP(\d+)__\s+to\s+(\w+)') + m = pattern.match(template) + if m: + src_val = resolve_op(int(m.group(2))) + dest_type_str = m.group(3) + dest_type = self._ParseLLVMType(Gen, dest_type_str) + if src_val and dest_type: + method = getattr(Gen.builder, method_name, None) + if method: + result = method(src_val, dest_type, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + Load_match = re.match(r'Load\s+(\w+),\s*(\w+)\s+%__OP(\d+)__', template) + if Load_match: + ptr_val = resolve_op(int(Load_match.group(3))) + if ptr_val: + result = Gen.builder.load(ptr_val, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + store_match = re.match(r'store\s+(\w+)\s+%__OP(\d+)__,\s*(\w+)\s+%__OP(\d+)__', template) + if store_match: + val = resolve_op(int(store_match.group(2))) + ptr = resolve_op(int(store_match.group(4))) + if val and ptr: + Gen.builder.store(val, ptr) + return ir.Constant(ir.IntType(32), 0) + + gep_match = re.match(r'getelementptr\s+(?:inbounds\s+)?(\w+),\s*(\w+)\s+%__OP(\d+)__,\s*(.+)', template) + if gep_match: + ptr_val = resolve_op(int(gep_match.group(3))) + indices_str = gep_match.group(4) + if ptr_val: + indices = [] + for idx_part in re.split(r',\s*', indices_str): + idx_part = idx_part.strip() + idx_m = re.match(r'%__OP(\d+)__', idx_part) + if idx_m: + idx_op = resolve_op(int(idx_m.group(1))) + if idx_op: + indices.append(idx_op) + else: + try: + indices.append(ir.Constant(ir.IntType(32), int(idx_part))) + except ValueError: + break + if indices: + result = Gen.builder.gep(ptr_val, indices, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + neg_match = re.match(r'sub\s+(\w+)\s+(\d+),\s*%__OP(\d+)__', template) + if neg_match: + op = resolve_op(int(neg_match.group(3))) + if op: + zero = ir.Constant(op.type, int(neg_match.group(2))) + result = Gen.builder.sub(zero, op, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + not_match = re.match(r'xor\s+(\w+)\s+(-1|\d+),\s*%__OP(\d+)__', template) + if not_match: + op = resolve_op(int(not_match.group(3))) + if op: + mask = ir.Constant(op.type, int(not_match.group(2))) + result = Gen.builder.xor_(op, mask, name="llvmir_result") + return self._StoreLLVMIROutputs(Gen, result, output_targets) + + call_match = re.match(r'call\s+(\w+)\s+@([\w.]+)\((.+)\)', template) + if call_match: + call_ret_str = call_match.group(1) + func_name = call_match.group(2) + args_str = call_match.group(3) + call_args = [] + for arg_part in re.split(r',\s*', args_str): + arg_part = arg_part.strip() + typed_op = re.match(r'(\w+\*?)\s+%__OP(\d+)__', arg_part) + if typed_op: + expected_type_str = typed_op.group(1) + op_idx = int(typed_op.group(2)) + val = resolve_op(op_idx) + if val: + val = self._ConvertLLVMIRType(Gen, val, expected_type_str) + if val: + call_args.append(val) + else: + const_m = re.match(r'(\w+)\s+(-?\d+)', arg_part) + if const_m: + const_type = self._ParseLLVMType(Gen, const_m.group(1)) + if const_type: + call_args.append(ir.Constant(const_type, int(const_m.group(2)))) + if call_args: + call_ret_type = self._ParseLLVMType(Gen, call_ret_str) or ir.VoidType() + func_type = ir.FunctionType(call_ret_type, [a.type for a in call_args]) + if func_name.startswith('llvm.'): + existing = Gen.module.globals.get(func_name) + if existing: + func = existing + else: + try: + func = Gen.module.declare_intrinsic(func_name, [a.type for a in call_args]) + except Exception: + func = ir.Function(Gen.module, func_type, name=func_name) + else: + existing = Gen.module.globals.get(func_name) + if existing: + func = existing + else: + func = ir.Function(Gen.module, func_type, name=func_name) + result = Gen.builder.call(func, call_args, name="llvmir_result") + if output_targets: + return self._StoreLLVMIROutputs(Gen, result, output_targets) + if isinstance(call_ret_type, ir.VoidType): + return ir.Constant(ir.IntType(32), 0) + return result + + if output_targets: + for i, target in enumerate(output_targets): + if isinstance(target, ir.AllocaInstr): + Gen.builder.store(ir.Constant(target.type.pointee, 0), target) + return ir.Constant(ir.IntType(32), 0) + + return ir.Constant(ret_type, 0) + + def _StoreLLVMIROutputs(self, Gen: LlvmGeneratorMixin, result: ir.Value, output_targets: list[ir.Value]) -> ir.Value: + for target in output_targets: + if isinstance(target, ir.AllocaInstr): + if target.type.pointee == result.type: + Gen.builder.store(result, target) + else: + i64t = ir.IntType(64) + if isinstance(result.type, ir.IntType) and result.type.width == 64 and isinstance(target.type.pointee, ir.PointerType): + ptr_val = Gen.builder.inttoptr(result, target.type.pointee) + Gen.builder.store(ptr_val, target) + elif isinstance(result.type, ir.PointerType) and isinstance(target.type.pointee, ir.IntType) and target.type.pointee.width == 64: + int_val = Gen.builder.ptrtoint(result, i64t) + Gen.builder.store(int_val, target) + else: + try: + cast_val = Gen.builder.bitcast(result, target.type.pointee) + Gen.builder.store(cast_val, target) + except Exception: + Gen.builder.store(result, target) + elif isinstance(target, (ir.GlobalVariable, ir.GEPInstr)): + if isinstance(target.type, ir.PointerType): + if target.type.pointee == result.type: + Gen.builder.store(result, target) + else: + try: + cast_val = Gen.builder.bitcast(result, target.type.pointee) + Gen.builder.store(cast_val, target) + except Exception: + Gen.builder.store(result, target) + return result + + def _ParseLLVMType(self, Gen: LlvmGeneratorMixin, type_str: str) -> ir.Type | None: + type_map = { + 'i1': ir.IntType(1), 'i8': ir.IntType(8), 'i16': ir.IntType(16), + 'i32': ir.IntType(32), 'i64': ir.IntType(64), 'i128': ir.IntType(128), + 'float': ir.FloatType(), 'double': ir.DoubleType(), + 'void': ir.VoidType(), + } + if type_str in type_map: + return type_map[type_str] + if type_str.endswith('*'): + pointee = self._ParseLLVMType(Gen, type_str[:-1]) + if pointee: + return ir.PointerType(pointee) + return None + + def _ConvertLLVMIRType(self, Gen: LlvmGeneratorMixin, val: ir.Value, expected_type_str: str) -> ir.Value | None: + expected_type = self._ParseLLVMType(Gen, expected_type_str.rstrip('*')) + if not expected_type: + return val + IsPtr = expected_type_str.endswith('*') + if IsPtr: + expected_ptr_type = ir.PointerType(expected_type) + if isinstance(val.type, ir.PointerType) and val.type != expected_ptr_type: + try: + return Gen.builder.bitcast(val, expected_ptr_type) + except Exception: + return val + elif isinstance(val.type, ir.IntType) and val.type.width == 64: + try: + return Gen.builder.inttoptr(val, expected_ptr_type) + except Exception: + return val + return val + if val.type == expected_type: + return val + if isinstance(val.type, ir.IntType) and isinstance(expected_type, ir.IntType): + if expected_type.width < val.type.width: + return Gen.builder.trunc(val, expected_type) + else: + return Gen.builder.zext(val, expected_type) + if isinstance(val.type, ir.PointerType) and isinstance(expected_type, ir.IntType): + if expected_type.width == 64: + return Gen.builder.ptrtoint(val, expected_type) + if isinstance(val.type, ir.IntType) and isinstance(expected_type, ir.PointerType): + if val.type.width == 64: + return Gen.builder.inttoptr(val, expected_type) + return val + + def _HandleExternalCallLlvm(self, Node: ast.Call, func_name: str, ModulePath: str | None) -> ir.Value | None: + Gen = self.Trans.LlvmGen + + # 解析 import 别名: 如 window -> vpsdk.window + if ModulePath and ModulePath not in ('c', 't'): + ModulePath = self.Trans.SymbolTable.resolve_alias(ModulePath) + + is_user_module = (ModulePath and ModulePath not in ('c', 't') and + (ModulePath in Gen.ModuleSha1Map or + ModulePath in getattr(self.Trans, '_ImportedModules', set()) or + ModulePath in self.Trans.SymbolTable.import_aliases)) + if ModulePath and ModulePath not in ('c', 't') and not is_user_module: + is_user_module = (ModulePath in Gen.ModuleSha1Map or + ModulePath.split('.')[-1] in Gen.ModuleSha1Map or + ModulePath.split('.')[-1] in getattr(self.Trans, '_ImportedModules', set())) + if is_user_module and ModulePath not in Gen.ModuleSha1Map and ModulePath not in getattr(self.Trans, '_ImportedModules', set()): + ModulePath = ModulePath.split('.')[-1] + + if ModulePath and ModulePath not in ('c', 't') and not is_user_module: + return None + + if is_user_module: + mangled_name = Gen._mangle_func_name(func_name, ModulePath) + sym_info = self.LookupFunctionSymbol(func_name, module_path=ModulePath) + is_inline = sym_info and (sym_info.IsInline or isinstance(sym_info.Storage, t.CInline)) + if is_inline and sym_info.InlineBody: + self._HandleInlineExpandLlvm(Node, sym_info) + return ir.Constant(ir.IntType(32), 1) + sym_info_exact = self.Trans.SymbolTable.lookup(f'{ModulePath}.{func_name}') + if sym_info_exact: + exact_params = sym_info_exact.FuncPtrParams + exact_param_names = [pn for pn, _ in exact_params] + if exact_param_names: + provided = len(Node.args) + kw_provided = set() + for kw in Node.keywords: + if kw.arg and kw.arg in exact_param_names: + kw_provided.add(exact_param_names.index(kw.arg)) + # 获取默认参数信息，计算必需参数数量（与 _fill_default_args 一致的查找逻辑） + global_defaults: dict = self.Trans._global_function_default_args + func_defaults: list | None = None + if mangled_name in global_defaults: + func_defaults = global_defaults[mangled_name] + elif func_name in global_defaults: + func_defaults = global_defaults[func_name] + else: + for gk, gv in global_defaults.items(): + if func_name in gk or gk.endswith(func_name) or gk.endswith('.' + func_name): + func_defaults = gv + break + total_params: int = len(exact_param_names) + num_defaults: int = len(func_defaults) if func_defaults else 0 + min_args: int = total_params - num_defaults + # 只检查必需参数（有默认值的参数允许缺失） + for i in range(min_args): + if i >= provided and i not in kw_provided: + raise Exception(f"调用 {ModulePath}.{func_name}() 缺少必需参数 '{exact_param_names[i]}'，该参数没有默认值") + elif ModulePath in Gen.ModuleSha1Map and len(Node.args) == 0: + func = Gen.functions.get(mangled_name) + if func and len(func.function_type.args) > 0: + raise Exception(f"调用 {ModulePath}.{func_name}() 缺少必需参数，函数需要 {len(func.function_type.args)} 个参数但未传入任何参数") + if Gen._has_function(mangled_name): + return self._HandleClosureCallLlvm(Node, mangled_name) + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(mangled_name, Gen) + except Exception: + stub_func_type = None + if not stub_func_type and mangled_name != func_name: + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(func_name, Gen) + except Exception: + stub_func_type = None + if stub_func_type: + if is_user_module: + decl_name = mangled_name + else: + decl_name = mangled_name if mangled_name in self.Trans.ImportHandler._stub_func_cache else func_name + self.GetOrCreateStubFuncDecl(decl_name, stub_func_type, Gen, alias=func_name) + return self._HandleClosureCallLlvm(Node, decl_name) + sym_info = self.LookupFunctionSymbol(func_name, module_path=ModulePath) + if sym_info: + sig = self.BuildLLVMFuncTypeFromSig(sym_info, Gen, mangled_name=mangled_name, use_infer_fallback=True) + if sig is not None: + ret_type, llvm_param_types, sym_is_variadic = sig + self.GetOrCreateFuncDecl(mangled_name, ret_type, llvm_param_types, Gen, + is_variadic=sym_is_variadic, replace_name=func_name) + return self._HandleClosureCallLlvm(Node, mangled_name) + # 回退: 直接用 func_name 查找（_find_function 会遍历 SHA1 前缀） + fallback_func = Gen._find_function(func_name) + if fallback_func is not None: + return self._HandleClosureCallLlvm(Node, func_name) + return None + + if ModulePath == 'c': + if func_name == 'Asm': + return self.Trans.ExprAsmHandle._HandleAsmLlvm(Node) + elif func_name == 'Addr': + return self.Trans.CSpecialCallHandle._HandleCAddrLlvm(Node) + elif func_name == 'Set': + return self.Trans.CSpecialCallHandle._HandleCSetLlvm(Node) + elif func_name == 'Load': + return self.Trans.CSpecialCallHandle._HandleCLoadLlvm(Node) + elif func_name == 'Deref': + return self.Trans.CSpecialCallHandle._HandleCDerefLlvm(Node) + elif func_name == 'DerefAs': + return self._HandleCDerefAsLlvm(Node) + elif func_name == 'PtrToInt': + return self.Trans.CSpecialCallHandle._HandleCPtrToIntLlvm(Node) + elif func_name in ('CIf', 'CElif'): + return self.Trans.CSpecialCallHandle._HandleCIfLlvm(Node) + elif func_name == 'CError': + msg = "compile-time error" + if Node.args: + arg = Node.args[0] + if isinstance(arg, ast.Constant) and isinstance(arg.value, str): + msg = arg.value + lineno = getattr(Node, 'lineno', 0) + raise Exception(f"#error: {msg} (line {lineno})") + elif func_name in ('AsmInp', 'AsmOut'): + return ir.Constant(ir.IntType(32), 0) + elif func_name == 'LLVMIR': + return self._HandleLLVMIRLlvm(Node) + elif func_name in ('LInp', 'LOut'): + return ir.Constant(ir.IntType(32), 0) + + win32_apis = { + 'SetConsoleOutputCP': (ir.IntType(32), [ir.IntType(32)]), + 'SetConsoleCP': (ir.IntType(32), [ir.IntType(32)]), + 'GetConsoleOutputCP': (ir.IntType(32), []), + 'GetConsoleCP': (ir.IntType(32), []), + } + if ModulePath and 'win32console' in ModulePath and func_name in win32_apis: + ret_type, param_types = win32_apis[func_name] + if func_name not in Gen.functions: + func_type = ir.FunctionType(ret_type, param_types) + func_decl = ir.Function(Gen.module, func_type, name=func_name) + func_decl.linkage = 'external' + Gen.functions[func_name] = func_decl + call_args = [] + if Node.args: + arg_val = self.HandleExprLlvm(Node.args[0]) + if arg_val: + target_type = param_types[0] if param_types else ir.IntType(32) + if isinstance(arg_val.type, ir.IntType) and arg_val.type.width < target_type.width: + arg_val = Gen.builder.zext(arg_val, target_type, name="zext_win32_arg") + elif isinstance(arg_val.type, ir.IntType) and arg_val.type.width > target_type.width: + arg_val = Gen.builder.trunc(arg_val, target_type, name="trunc_win32_arg") + call_args.append(arg_val) + return Gen.builder.call(Gen.functions[func_name], call_args, name=f"call_{func_name}") + + if Gen._has_function(func_name): + return self._HandleClosureCallLlvm(Node, func_name) + + if func_name in self.Trans.FunctionDefCache: + return self._HandleClosureCallLlvm(Node, func_name) + + if func_name in self._C_LIB_FUNCS: + return self._HandleClosureCallLlvm(Node, func_name) + + mangled_name = Gen._mangle_func_name(func_name, ModulePath) + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(mangled_name, Gen) + except Exception: + stub_func_type = None + if not stub_func_type and mangled_name != func_name: + try: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(func_name, Gen) + except Exception: + stub_func_type = None + if stub_func_type: + decl_name = mangled_name if mangled_name in self.Trans.ImportHandler._stub_func_cache else func_name + self.GetOrCreateStubFuncDecl(decl_name, stub_func_type, Gen, alias=func_name, check_existing=False) + return self._HandleClosureCallLlvm(Node, decl_name) + + sym_info = self.LookupFunctionSymbol(func_name, module_path=ModulePath or None) + if sym_info: + sig = self.BuildLLVMFuncTypeFromSig(sym_info, Gen) + if sig is not None: + ret_type, llvm_param_types, sym_is_variadic = sig + mangled_name = Gen._mangle_func_name(func_name, ModulePath) + self.GetOrCreateFuncDecl(mangled_name, ret_type, llvm_param_types, Gen, + is_variadic=sym_is_variadic, replace_name=func_name) + return self._HandleClosureCallLlvm(Node, mangled_name) + + return None + + def _HandleTypeUnionCastLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen = self.Trans.LlvmGen + if not Node.args: + return None + type_info = None + try: + type_info = getattr(self.Trans, 'TypeMergeHandler', None) and self.Trans.TypeMergeHandler.MergeTypes(Node.func) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + if not isinstance(type_info, CTypeInfo): + return None + target_type_str = type_info.ToString() + IsPtr = type_info.IsPtr + if IsPtr and target_type_str and '*' not in target_type_str: + target_type_str += ' *' + if not target_type_str: + return None + if type_info.IsPtr and not IsPtr: + IsPtr = True + if '*' not in target_type_str: + target_type_str += ' *' + if isinstance(type_info.Storage, t.CExport) or isinstance(type_info.Storage, t.CExtern): + pass + if type_info.IsState: + Gen._export_funcs.add('_type_union_cast') + return self._HandleTypeCastLlvm(Node.args[0], target_type_str) + + def _HandleTypeCastLlvm(self, expr_node: ast.AST, target_type_str: str) -> ir.Value | None: + Gen = self.Trans.LlvmGen + val = self.HandleExprLlvm(expr_node) + if val is None: + return Gen.emit_constant(0, target_type_str) + target_type = Gen._CType2LLVM(target_type_str) + if not target_type: + return val + if val.type == target_type: + if Gen._is_type_unsigned(target_type_str): + Gen._unsigned_results[id(val)] = True + return val + src_is_float = isinstance(val.type, (ir.FloatType, ir.DoubleType)) + dst_is_float = isinstance(target_type, (ir.FloatType, ir.DoubleType)) + src_is_int = isinstance(val.type, ir.IntType) + dst_is_int = isinstance(target_type, ir.IntType) + src_IsPtr = isinstance(val.type, ir.PointerType) + dst_IsPtr = isinstance(target_type, ir.PointerType) + if src_is_int and dst_is_float: + is_unsigned = Gen._check_node_unsigned(expr_node) + if is_unsigned: + return Gen.builder.uitofp(val, target_type, name="uint2float") + return Gen.builder.sitofp(val, target_type, name="int2float") + if src_is_float and dst_is_int: + return Gen.builder.fptosi(val, target_type, name="float2int") + if src_is_int and dst_is_int: + is_dst_unsigned = Gen._is_type_unsigned(target_type_str) + if val.type.width < target_type.width: + if val.type.width == 1: + # i1 是布尔值（来自 icmp/fcmp/__contains__），无符号语义，必须 zext + result = Gen.builder.zext(val, target_type, name="zext_bool_cast") + else: + is_unsigned = Gen._check_node_unsigned(expr_node) + if is_unsigned or is_dst_unsigned: + result = Gen.builder.zext(val, target_type, name="zext_cast") + else: + result = Gen.builder.sext(val, target_type, name="sext_cast") + if is_dst_unsigned: + Gen._unsigned_results[id(result)] = True + return result + if val.type.width > target_type.width: + result = Gen.builder.trunc(val, target_type, name="trunc_cast") + if is_dst_unsigned: + Gen._unsigned_results[id(result)] = True + return result + if is_dst_unsigned: + Gen._unsigned_results[id(val)] = True + return val + if src_is_float and dst_is_float: + if isinstance(val.type, ir.FloatType) and isinstance(target_type, ir.DoubleType): + return Gen.builder.fpext(val, target_type, name="fpext_cast") + if isinstance(val.type, ir.DoubleType) and isinstance(target_type, ir.FloatType): + return Gen.builder.fptrunc(val, target_type, name="fptrunc_cast") + return val + if src_IsPtr and dst_is_int: + if target_type.width < 64: + ptr_to_i64 = Gen.builder.ptrtoint(val, ir.IntType(64), name="ptr2i64") + return Gen.builder.trunc(ptr_to_i64, target_type, name="ptr2int_cast") + return Gen.builder.ptrtoint(val, target_type, name="ptr2int_cast") + if src_is_int and dst_IsPtr: + return Gen.builder.inttoptr(val, target_type, name="int2ptr_cast") + if src_IsPtr and dst_IsPtr: + if isinstance(val.type.pointee, ir.PointerType) and isinstance(target_type.pointee, ir.IntType): + Loaded = Gen._load(val, name="load_ptr") + return Loaded + return Gen.builder.bitcast(val, target_type, name="ptrcast") + try: + return Gen.builder.bitcast(val, target_type, name="cast") + except Exception: # 回退：bitcast 失败时返回原值 + return val + + def _HandleREnumConstructLlvm(self, Node: ast.Call, RenumName: str, VariantName: str, TagValue: int) -> ir.Value | None: + Gen = self.Trans.LlvmGen + if RenumName not in Gen.structs: + self.Trans.ImportHandler._TryLoadStructFromStub(RenumName, Gen) + if RenumName not in Gen.structs: + return None + RenumType = Gen.structs[RenumName] + result = Gen._allocaEntry(RenumType, name=f"{RenumName}_{VariantName}") + tag_ptr = Gen.builder.gep(result, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="renum_tag_ptr") + Gen._store(ir.Constant(ir.IntType(32), TagValue), tag_ptr) + NestedStructName = f"{RenumName}_{VariantName}" + if NestedStructName in Gen.structs and Node.args: + NestedStructType = Gen.structs[NestedStructName] + NestedStructPtrType = ir.PointerType(NestedStructType) + variant_ptr = Gen.builder.bitcast(result, NestedStructPtrType, name=f"cast_{NestedStructName}") + members = Gen.class_members.get(NestedStructName, []) + payLoad_members = [(n, t) for n, t in members if n != '__tag'] + for i, arg in enumerate(Node.args): + if i < len(payLoad_members): + member_name, member_type = payLoad_members[i] + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + Coerced = Gen._coerce_value(ArgVal, member_type) + if Coerced is not None: + ArgVal = Coerced + elem_ptr = Gen.builder.gep(variant_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), i + 1)], name=f"{VariantName}_{member_name}") + # REnum 嵌套结构体共享 max_variant_struct 布局，成员类型可能与 + # 结构体字段类型不一致（如 Bits:i32 与 Ret:LLVMType* 重叠）。 + # 需 bitcast elem_ptr 到成员类型，避免 _store 按 struct 字段类型做错误转换。 + if isinstance(elem_ptr.type, ir.PointerType) and elem_ptr.type.pointee != member_type: + elem_ptr = Gen.builder.bitcast(elem_ptr, ir.PointerType(member_type), name=f"{VariantName}_{member_name}_cast") + Gen._store(ArgVal, elem_ptr) + return result + + def _HandleTypedefCastLlvm(self, Node: ast.Call, TypedefName: str) -> ir.Value: + Gen = self.Trans.LlvmGen + SymInfo = self.Trans.SymbolTable.lookup(TypedefName) + if not SymInfo or not SymInfo.IsTypedef: + return ir.Constant(ir.IntType(32), 0) + # 确保 Name 已设置，以便 _resolve_typedef_ctype 能正确解析 + if not SymInfo.Name: + SymInfo.Name = TypedefName + TargetType = Gen._ctype_to_llvm(SymInfo) + if not Node.args: + if isinstance(TargetType, ir.IntType): + return ir.Constant(TargetType, 0) + elif isinstance(TargetType, ir.PointerType): + return ir.Constant(TargetType, None) + return ir.Constant(ir.IntType(32), 0) + ArgVal = self.HandleExprLlvm(Node.args[0]) + if not ArgVal: + if isinstance(TargetType, ir.IntType): + return ir.Constant(TargetType, 0) + return ir.Constant(ir.IntType(32), 0) + if ArgVal.type == TargetType: + return ArgVal + if isinstance(TargetType, ir.IntType) and isinstance(ArgVal.type, ir.IntType): + if ArgVal.type.width < TargetType.width: + return Gen.builder.zext(ArgVal, TargetType, name=f"zext_{TypedefName}_cast") + elif ArgVal.type.width > TargetType.width: + return Gen.builder.trunc(ArgVal, TargetType, name=f"trunc_{TypedefName}_cast") + return ArgVal + if isinstance(TargetType, ir.PointerType): + if isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.bitcast(ArgVal, TargetType, name=f"bitcast_{TypedefName}_cast") + if isinstance(ArgVal.type, ir.IntType): + return Gen.builder.inttoptr(ArgVal, TargetType, name=f"inttoptr_{TypedefName}_cast") + # 如果参数是结构体值，需要 alloca 并返回指针 + if isinstance(ArgVal.type, ir.BaseStructType): + alloca = Gen.builder.alloca(ArgVal.type, name=f"struct_alloca_{TypedefName}") + Gen.builder.store(ArgVal, alloca, align=8) + return Gen.builder.bitcast(alloca, TargetType, name=f"bitcast_struct_{TypedefName}") + if isinstance(TargetType, ir.IntType) and isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.ptrtoint(ArgVal, TargetType, name=f"ptrtoint_{TypedefName}_cast") + if isinstance(TargetType, (ir.FloatType, ir.DoubleType)) and isinstance(ArgVal.type, ir.IntType): + return Gen.builder.sitofp(ArgVal, TargetType, name=f"sitofp_{TypedefName}_cast") + if isinstance(TargetType, ir.IntType) and isinstance(ArgVal.type, (ir.FloatType, ir.DoubleType)): + return Gen.builder.fptosi(ArgVal, TargetType, name=f"fptosi_{TypedefName}_cast") + return ArgVal + + def _HandleInlineExpandLlvm(self, Node: ast.Call, sym_info: CTypeInfo) -> None: + Gen = self.Trans.LlvmGen + inline_body = sym_info.InlineBody + inline_params = sym_info.InlineParams or [] + call_args = [] + for arg in Node.args: + call_args.append(self.Trans.ExprHandler.HandleExprLlvm(arg)) + saved_vars = {} + for i, param_name in enumerate(inline_params): + if i < len(call_args): + saved_vars[param_name] = call_args[i] + old_var_scopes = list(self.Trans.VarScopes) + for param_name, val in saved_vars.items(): + if val is not None: + alloca = Gen._allocaEntry(val.type, name=f"inline_arg_{param_name}") + Gen._store(val, alloca) + self.Trans.VarScopes.append((param_name, alloca)) + self.Trans.BodyHandler.HandleBodyLlvm(inline_body) + for param_name, _ in saved_vars.items(): + if self.Trans.VarScopes and self.Trans.VarScopes[-1][0] == param_name: + self.Trans.VarScopes.pop() + self.Trans.VarScopes = old_var_scopes + return None + + def _HandleClosureCallLlvm(self, Node: ast.Call, FuncName: str) -> ir.Value: + Gen = self.Trans.LlvmGen + func = Gen.functions.get(FuncName) + if not func: + if FuncName in self.Trans.FunctionDefCache: + FuncDef = self.Trans.FunctionDefCache[FuncName] + self.Trans.FunctionHandler._EmitFunctionForwardDeclLlvm(FuncDef, Gen) + func = Gen.functions.get(FuncName) + if not func: + func = self._try_declare_c_lib_func(FuncName, Gen) + if not func: + raise Exception(f"Undefined function: '{FuncName}'") + + FuncDef = self.Trans.FunctionDefCache.get(FuncName) + param_names = [] + FuncDef_args = getattr(FuncDef, 'args', None) if FuncDef else None + if FuncDef_args: + param_names = [arg.arg for arg in FuncDef.args.args] + # 回退到全局参数名表（支持跨模块 include 函数） + if not param_names: + global_param_names = getattr(self.Trans, '_global_function_param_names', {}) + param_names = global_param_names.get(FuncName) or [] + + num_params = len(func.function_type.args) + nonlocal_param_count = len(Gen.nonlocal_params.get(FuncName, [])) + eh_msg_out_count = 0 + if num_params > 0 and func.args: + for arg in func.args: + if arg.name in ('__eh_msg_out__', '__eh_code_out__'): + eh_msg_out_count += 1 + regular_param_count = num_params - nonlocal_param_count - eh_msg_out_count + is_variadic = False + func_ft = getattr(func, 'function_type', None) + is_variadic = getattr(func_ft, 'var_arg', False) if func_ft else False + + if not is_variadic and len(Node.args) > regular_param_count: + raise Exception(f"调用 {FuncName}() 传入了 {len(Node.args)} 个参数，但函数只需要 {regular_param_count} 个参数") + + max_args = len(Node.args) if is_variadic else regular_param_count + resolved = [None] * max_args + + for i, arg in enumerate(Node.args): + if i < max_args: + arg_var_type = None + if i < num_params: + arg_var_type = func.function_type.args[i] + ArgVal = self.HandleExprLlvm(arg, VarType=arg_var_type) + if ArgVal: + resolved[i] = ArgVal + + for kw in Node.keywords: + if kw.arg is None: + if isinstance(kw.value, ast.Dict): + pass + elif isinstance(kw.value, ast.Call): + pass + continue + ArgVal = self.HandleExprLlvm(kw.value) + if not ArgVal: + continue + kw_name = kw.arg + if kw_name in param_names: + idx = param_names.index(kw_name) + if idx < regular_param_count: + resolved[idx] = ArgVal + + FuncDef_args_defaults = getattr(getattr(FuncDef, 'args', None), 'defaults', []) if FuncDef else [] + if FuncDef_args_defaults: + defaults = FuncDef.args.defaults + num_defaults = len(defaults) + if num_defaults > 0: + min_args = regular_param_count - num_defaults + for i in range(regular_param_count): + if resolved[i] is None and i >= min_args: + default_idx = i - min_args + if default_idx < len(defaults): + default_node = defaults[default_idx] + default_val = self.HandleExprLlvm(default_node) + if default_val: + param_type = func.function_type.args[i] + if default_val.type != param_type: + if isinstance(param_type, ir.IntType) and isinstance(default_val.type, ir.IntType): + if default_val.type.width < param_type.width: + default_val = Gen.builder.zext(default_val, param_type, name=f"zext_default_{i}") + elif default_val.type.width > param_type.width: + default_val = Gen.builder.trunc(default_val, param_type, name=f"trunc_default_{i}") + elif isinstance(param_type, ir.PointerType) and isinstance(default_val.type, ir.IntType): + default_val = Gen.builder.inttoptr(default_val, param_type, name=f"inttoptr_default_{i}") + resolved[i] = default_val + + if FuncDef and param_names: + min_args = regular_param_count - num_defaults if FuncDef_args_defaults else regular_param_count + provided = len(Node.args) + kw_provided = set() + for kw in Node.keywords: + if kw.arg and kw.arg in param_names: + kw_provided.add(param_names.index(kw.arg)) + for i in range(min(min_args, len(param_names))): + if i >= provided and i not in kw_provided: + raise Exception(f"调用 {FuncName}() 缺少必需参数 '{param_names[i]}'，该参数没有默认值") + + CallArgs = [] + for i in range(max_args): + if resolved[i] is not None: + CallArgs.append(resolved[i]) + elif i < regular_param_count: + arg_type = func.function_type.args[i] + if isinstance(arg_type, ir.PointerType): + CallArgs.append(ir.Constant(arg_type, None)) + elif isinstance(arg_type, (ir.IdentifiedStructType, ir.LiteralStructType)): + zero_val = Gen._zero_value_for_type(arg_type) + if zero_val is not None: + CallArgs.append(zero_val) + else: + CallArgs.append(ir.Constant(ir.PointerType(arg_type), None)) + elif isinstance(arg_type, ir.ArrayType): + zero_val = Gen._zero_value_for_type(arg_type) + if zero_val is not None: + CallArgs.append(zero_val) + else: + CallArgs.append(ir.Constant(arg_type, 0)) + else: + CallArgs.append(ir.Constant(arg_type, 0)) + + if FuncName in Gen.nonlocal_params: + for var_name, ptr_type in Gen.nonlocal_params[FuncName]: + if var_name in Gen.variables and Gen.variables[var_name] is not None: + CallArgs.append(Gen.variables[var_name]) + elif var_name in Gen._reg_values: + OldVal = Gen._reg_values[var_name] + var = Gen._allocaEntry(OldVal.type, name=var_name) + Gen._store(OldVal, var) + Gen.variables[var_name] = var + del Gen._reg_values[var_name] + CallArgs.append(var) + + self._append_eh_msg_out_arg(CallArgs, func, Gen) + + if is_variadic: + # 对于可变参数函数，需要手动处理参数类型转换 + adjusted = [] + for i, arg in enumerate(CallArgs): + if i < num_params: + # 固定参数：尝试类型转换 + param_type = func.function_type.args[i] + if arg.type != param_type: + try: + if isinstance(arg.type, ir.IntType) and isinstance(param_type, ir.PointerType): + adjusted.append(self.builder.inttoptr(arg, param_type)) + elif isinstance(arg.type, ir.PointerType) and isinstance(param_type, ir.IntType): + adjusted.append(Gen.builder.ptrtoint(arg, param_type)) + elif isinstance(arg.type, ir.IntType) and arg.type.width < param_type.width: + adjusted.append(Gen.builder.zext(arg, param_type)) + elif isinstance(arg.type, ir.PointerType) and isinstance(param_type, ir.PointerType): + adjusted.append(Gen.builder.bitcast(arg, param_type, name=f"cast_arg_{i}")) + else: + adjusted.append(arg) + except Exception: # 回退：参数类型调整失败时使用原值 + if isinstance(arg.type, ir.PointerType) and isinstance(param_type, ir.PointerType): + try: + adjusted.append(Gen.builder.bitcast(arg, param_type, name=f"cast_arg_{i}")) + except Exception: + adjusted.append(arg) + else: + adjusted.append(arg) + else: + adjusted.append(arg) + else: + # 可变参数：应用默认参数提升（float -> double, i32 -> i64） + if isinstance(arg.type, ir.FloatType): + arg = Gen.builder.fpext(arg, ir.DoubleType(), name=f"fpext_vararg_{FuncName}") + elif isinstance(arg.type, ir.IntType) and arg.type.width < 64: + # i1 是布尔值（来自 icmp/fcmp/__contains__），无符号语义，必须 zext + if arg.type.width == 1: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_bool_vararg_{FuncName}") + else: + # 有符号整数用 sext，无符号整数用 zext + is_signed = True + arg_node = Node.args[i] if i < len(Node.args) else None + if arg_node is not None: + is_u = Gen._check_node_unsigned(arg_node) + if is_u: + is_signed = False + if is_signed: + arg = Gen.builder.sext(arg, ir.IntType(64), name=f"sext_vararg_{FuncName}") + else: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_vararg_{FuncName}") + adjusted.append(arg) + result = Gen.builder.call(func, adjusted, name=f"call_{FuncName}") + else: + CallArgs = Gen._apply_auto_addr(FuncName, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + # 兜底：确保所有指针类型参数与函数签名匹配（如 struct* -> i8*） + for i, (arg, param) in enumerate(zip(adjusted, func.args)): + if arg.type != param.type: + if isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.PointerType): + adjusted[i] = Gen.builder.bitcast(arg, param.type, name=f"cast_arg_{i}") + result = Gen.builder.call(func, adjusted, name=f"call_{FuncName}") + self._check_eh_return(result, FuncName, Gen, called_func=func) + return result + + def _try_register_vtable_ondemand(self, ClassName: str, class_sha1: str, Gen: "LlvmGeneratorMixin") -> None: + """按需 vtable 注册：从 .pyi stub 检查 @t.CVTable 并注册到 vtable 集合。 + + 当 vcall 决策点发现 ClassName 不在 vtable 集合中时调用。 + 读取 temp/{sha1}.pyi 文件，查找 ClassName 的 ClassDef， + 检查 @t.CVTable 装饰器，注册 vtable 和方法列表。 + """ + try: + import_handler = self.Trans.ImportHandler + project_root = import_handler._find_project_root() + pyi_path = os.path.join(project_root, 'temp', f'{class_sha1}.pyi') + if not os.path.isfile(pyi_path): + return + with open(pyi_path, 'r', encoding='utf-8') as f: + pyi_code = f.read() + pyi_tree = ast.parse(pyi_code) + # 查找目标 ClassDef + target_node: ast.ClassDef | None = None + for node in ast.iter_child_nodes(pyi_tree): + if isinstance(node, ast.ClassDef) and node.name == ClassName: + target_node = node + break + if target_node is None: + return + # 检查 @t.CVTable 装饰器 + is_cvtable: bool = False + for decorator in target_node.decorator_list: + if isinstance(decorator, ast.Attribute) and getattr(decorator.value, 'id', None) == 't' and decorator.attr == 'CVTable': + is_cvtable = True + break + if isinstance(decorator, ast.Call): + inner = decorator.func + if isinstance(inner, ast.Attribute) and getattr(inner.value, 'id', None) == 't' and inner.attr == 'CVTable': + is_cvtable = True + break + if isinstance(inner, ast.Name) and inner.id == 'CVTable': + is_cvtable = True + break + if isinstance(decorator, ast.Name) and decorator.id == 'CVTable': + is_cvtable = True + break + if not is_cvtable: + return + # 注册到 vtable 集合（共享集合） + Gen._cross_module_vtable_classes.add(ClassName) + Gen.class_vtable.add(ClassName) + # 预注册方法列表（与 prescan 逻辑一致） + if ClassName not in Gen.class_methods: + Gen.class_methods[ClassName] = [] + for item in target_node.body: + if isinstance(item, ast.FunctionDef): + method_name = item.name + # 构造函数不放入 vtable（避免子类与基类 vtable 大小不一致破坏多态分派） + if method_name in ('__new__', '__init__', '__before_init__'): + continue + if method_name == '__init__': + full_name = f"{ClassName}.__init__" + elif method_name == '__call__': + full_name = f"{ClassName}.__call__" + else: + full_name = f"{ClassName}.{method_name}" + if full_name not in Gen.class_methods[ClassName]: + Gen.class_methods[ClassName].append(full_name) + except Exception: + pass + + def _HandleMethodCallLlvm(self, Node: ast.Call) -> ir.Value: + Gen = self.Trans.LlvmGen + MethodName = Node.func.attr + # 检查是否为 @staticmethod 或 @classmethod，如果是则走特殊路径 + if isinstance(Node.func.value, ast.Name): + VarName = Node.func.value.id + ClassName = Gen.var_struct_class.get(VarName) + if ClassName: + SymKey = f'{ClassName}.{MethodName}' + SymInfo = self.Trans.SymbolTable.lookup(SymKey) + if SymInfo and SymInfo.MetaList: + if FuncMeta.STATIC_METHOD in SymInfo.MetaList: + return self._HandleStaticMethodCallLlvm(Node, ClassName, MethodName) + if FuncMeta.CLASS_METHOD in SymInfo.MetaList: + return self._HandleStaticMethodCallLlvm(Node, ClassName, MethodName) + if isinstance(Node.func.value, ast.Name) and IsTModule(Node.func.value.id, self.Trans.SymbolTable): + result = self.Trans.TSpecialCallHandle._HandleTSpecialCallLlvm(Node) + if result is not None: + return result + ctype_cls = CTypeRegistry.GetClassByName(MethodName) + if ctype_cls is not None: + try: + ctype_inst = ctype_cls() + except Exception: + ctype_inst = None + target_bits = getattr(ctype_inst, 'Size', None) if ctype_inst else None + if target_bits is None: + target_bits = 0 + is_float = getattr(ctype_inst, 'IsSigned', False) is None and target_bits > 0 if ctype_inst else False + IsPtr_cast = len(Node.args) >= 2 and ( + (isinstance(Node.args[1], ast.Attribute) and Node.args[1].attr == 'CPtr' and isinstance(Node.args[1].value, ast.Name) and IsTModule(Node.args[1].value.id, self.Trans.SymbolTable)) or + (isinstance(Node.args[1], ast.Name) and Node.args[1].id == 'CPtr') + ) + if Node.args: + ArgNode = Node.args[0] + if isinstance(ArgNode, ast.Subscript): + SubscriptPtr = self._HandleSubscriptPtrLlvm(ArgNode) + if SubscriptPtr: + if MethodName == 'CPtr' or IsPtr_cast: + return SubscriptPtr + if isinstance(SubscriptPtr.type, ir.PointerType): + val = Gen._load(SubscriptPtr, name="Load_subscript_val") + else: + val = SubscriptPtr + if target_bits > 0 and isinstance(val.type, ir.IntType) and not is_float: + if val.type.width < target_bits: + val = Gen.builder.zext(val, ir.IntType(target_bits), name="zext_cast") + elif val.type.width > target_bits: + val = Gen.builder.trunc(val, ir.IntType(target_bits), name="trunc_cast") + return val + val = self.HandleExprLlvm(ArgNode) + if IsPtr_cast: + if target_bits > 0 and not is_float: + target_ptr_type = ir.PointerType(ir.IntType(target_bits)) + elif is_float and target_bits == 32: + target_ptr_type = ir.PointerType(ir.FloatType()) + elif is_float and target_bits == 64: + target_ptr_type = ir.PointerType(ir.DoubleType()) + else: + target_ptr_type = ir.PointerType(ir.IntType(8)) + if isinstance(val.type, ir.PointerType): + if val.type != target_ptr_type: + val = Gen.builder.bitcast(val, target_ptr_type, name="ptr_cast") + elif isinstance(val.type, ir.IntType): + if val.type.width < 64: + val = Gen.builder.zext(val, ir.IntType(64), name="zext_ptr_cast") + val = Gen.builder.inttoptr(val, target_ptr_type, name="int_to_ptr_cast") + return val + if val and target_bits > 0 and not is_float: + if isinstance(val.type, ir.IntType): + if val.type.width < target_bits: + val = Gen.builder.zext(val, ir.IntType(target_bits), name="zext_cast") + elif val.type.width > target_bits: + val = Gen.builder.trunc(val, ir.IntType(target_bits), name="trunc_cast") + elif isinstance(val.type, ir.PointerType): + val = Gen.builder.ptrtoint(val, ir.IntType(target_bits), name="ptrtoint_cast") + return val + if is_float: + return ir.Constant(ir.FloatType() if target_bits == 32 else ir.DoubleType(), 0.0) + return ir.Constant(ir.IntType(target_bits) if target_bits > 0 else ir.IntType(32), 0) + if self.Trans.SymbolTable.has(MethodName): + SymInfo = self.Trans.SymbolTable[MethodName] + if SymInfo.IsTypedef: + return self._HandleTypedefCastLlvm(Node, MethodName) + if isinstance(Node.func.value, ast.Name) and Node.func.value.id == 'c': + if MethodName == 'Asm': + return self.Trans.ExprAsmHandle._HandleAsmLlvm(Node) + if MethodName == 'Addr': + return self.Trans.CSpecialCallHandle._HandleCAddrLlvm(Node) + elif MethodName == 'Set': + return self.Trans.CSpecialCallHandle._HandleCSetLlvm(Node) + elif MethodName == 'Load': + return self.Trans.CSpecialCallHandle._HandleCLoadLlvm(Node) + elif MethodName == 'Deref': + return self.Trans.CSpecialCallHandle._HandleCDerefLlvm(Node) + elif MethodName == 'DerefAs': + return self._HandleCDerefAsLlvm(Node) + elif MethodName == 'PtrToInt': + return self.Trans.CSpecialCallHandle._HandleCPtrToIntLlvm(Node) + elif MethodName in ('CIf', 'CElif'): + if Node.args: + ArgVal = self.HandleExprLlvm(Node.args[0]) + if ArgVal: + if isinstance(ArgVal, ir.Constant) and isinstance(ArgVal.type, ir.IntType): + return ir.Constant(ir.IntType(1), 1 if ArgVal.constant != 0 else 0) + if isinstance(ArgVal.type, ir.IntType): + return Gen.builder.icmp_signed('!=', ArgVal, Gen._ZeroConst(ArgVal.type), name="cif_cond") + if isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.icmp_signed('!=', ArgVal, Gen._ZeroConst(ArgVal.type), name="cif_cond") + return ir.Constant(ir.IntType(1), 0) + elif MethodName in ('CIfdef', 'CIfndef'): + return ir.Constant(ir.IntType(1), 1) + elif MethodName == 'CError': + msg = "compile-time error" + if Node.args: + arg = Node.args[0] + if isinstance(arg, ast.Constant) and isinstance(arg.value, str): + msg = arg.value + else: + try: + val = self.HandleExprLlvm(arg) + if isinstance(val, str): + msg = val + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理函数调用失败: {_e}", "Exception") + lineno = getattr(Node, 'lineno', 0) + raise Exception(f"#error: {msg} (line {lineno})") + elif MethodName in ('CElse', 'CEndif', 'CPragma', 'CUndef'): + return ir.Constant(ir.IntType(1), 1) + elif MethodName in ('LLVMIR', 'LInp', 'LOut'): + return None + ObjVal = self.HandleExprLlvm(Node.func.value) + if not ObjVal: + return None + if isinstance(ObjVal.type, (ir.IdentifiedStructType, ir.LiteralStructType)): + if isinstance(Node.func.value, ast.Name): + var_name = Node.func.value.id + if var_name in Gen.variables: + var_ptr = Gen.variables[var_name] + if isinstance(var_ptr.type, ir.PointerType) and isinstance(var_ptr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + ObjVal = var_ptr + else: + tmp = Gen._allocaEntry(ObjVal.type, name="chain_method_tmp") + Gen._store(ObjVal, tmp) + ObjVal = tmp + else: + tmp = Gen._allocaEntry(ObjVal.type, name="chain_method_tmp") + Gen._store(ObjVal, tmp) + ObjVal = tmp + else: + tmp = Gen._allocaEntry(ObjVal.type, name="chain_method_tmp") + Gen._store(ObjVal, tmp) + ObjVal = tmp + if isinstance(Node.func.value, ast.Name): + ModuleName = Node.func.value.id + imported = getattr(self.Trans, '_ImportedModules', set()) + actual_module = self.Trans.SymbolTable.resolve_alias(ModuleName) + if ModuleName in imported or actual_module in imported: + mangled_name = Gen._mangle_func_name(MethodName, actual_module) + if Gen._has_function(mangled_name): + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + func = Gen._get_function(mangled_name) + self._fill_default_args(CallArgs, func, MethodName) + func_ft = getattr(func, 'function_type', None) + is_variadic = getattr(func_ft, 'var_arg', False) if func_ft else False + num_fixed_params = len(func_ft.args) if func_ft else 0 + if is_variadic: + adjusted = [] + for i, arg in enumerate(CallArgs): + if i < num_fixed_params: + param_type = func_ft.args[i] + if arg.type != param_type: + try: + if isinstance(arg.type, ir.IntType) and isinstance(param_type, ir.PointerType): + adjusted.append(self.builder.inttoptr(arg, param_type)) + elif isinstance(arg.type, ir.PointerType) and isinstance(param_type, ir.IntType): + adjusted.append(Gen.builder.ptrtoint(arg, param_type)) + elif isinstance(arg.type, ir.IntType) and arg.type.width < param_type.width: + adjusted.append(Gen.builder.zext(arg, param_type)) + else: + adjusted.append(arg) + except Exception: # 回退：参数类型调整失败时使用原值 + adjusted.append(arg) + else: + adjusted.append(arg) + else: + if isinstance(arg.type, ir.FloatType): + arg = Gen.builder.fpext(arg, ir.DoubleType(), name=f"fpext_vararg_{MethodName}") + elif isinstance(arg.type, ir.IntType) and arg.type.width < 64: + if arg.type.width == 1: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_bool_vararg_{MethodName}") + else: + is_signed = True + arg_node = Node.args[i] if i < len(Node.args) else None + if arg_node is not None: + is_u = Gen._check_node_unsigned(arg_node) + if is_u: + is_signed = False + if is_signed: + arg = Gen.builder.sext(arg, ir.IntType(64), name=f"sext_vararg_{MethodName}") + else: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_vararg_{MethodName}") + adjusted.append(arg) + result = Gen.builder.call(func, adjusted, name=f"call_{MethodName}") + else: + CallArgs = Gen._apply_auto_addr(MethodName, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{MethodName}") + return result + if Gen._has_function(MethodName): + _mf = Gen._get_function(MethodName) + if '.' not in _mf.name: + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + func = _mf + self._fill_default_args(CallArgs, func, MethodName) + func_ft = getattr(func, 'function_type', None) + is_variadic = getattr(func_ft, 'var_arg', False) if func_ft else False + num_fixed_params = len(func_ft.args) if func_ft else 0 + if is_variadic: + adjusted = [] + for i, arg in enumerate(CallArgs): + if i < num_fixed_params: + param_type = func_ft.args[i] + if arg.type != param_type: + try: + if isinstance(arg.type, ir.IntType) and isinstance(param_type, ir.PointerType): + adjusted.append(self.builder.inttoptr(arg, param_type)) + elif isinstance(arg.type, ir.PointerType) and isinstance(param_type, ir.IntType): + adjusted.append(Gen.builder.ptrtoint(arg, param_type)) + elif isinstance(arg.type, ir.IntType) and arg.type.width < param_type.width: + adjusted.append(Gen.builder.zext(arg, param_type)) + else: + adjusted.append(arg) + except Exception: # 回退：参数类型调整失败时使用原值 + adjusted.append(arg) + else: + adjusted.append(arg) + else: + if isinstance(arg.type, ir.FloatType): + arg = Gen.builder.fpext(arg, ir.DoubleType(), name=f"fpext_vararg_{MethodName}") + elif isinstance(arg.type, ir.IntType) and arg.type.width < 64: + if arg.type.width == 1: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_bool_vararg_{MethodName}") + else: + is_signed = True + arg_node = Node.args[i] if i < len(Node.args) else None + if arg_node is not None: + is_u = Gen._check_node_unsigned(arg_node) + if is_u: + is_signed = False + if is_signed: + arg = Gen.builder.sext(arg, ir.IntType(64), name=f"sext_vararg_{MethodName}") + else: + arg = Gen.builder.zext(arg, ir.IntType(64), name=f"zext_vararg_{MethodName}") + adjusted.append(arg) + result = Gen.builder.call(func, adjusted, name=f"call_{MethodName}") + else: + CallArgs = Gen._apply_auto_addr(MethodName, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{MethodName}") + return result + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(mangled_name, Gen) + if not stub_func_type: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(MethodName, Gen) + if not stub_func_type and mangled_name != MethodName: + for sha1 in getattr(Gen, 'ModuleSha1Map', {}).values(): + alt_name = f"{sha1}.{MethodName}" + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(alt_name, Gen) + if stub_func_type: + mangled_name = alt_name + break + if stub_func_type: + func = self.GetOrCreateStubFuncDecl(mangled_name, stub_func_type, Gen) + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + self._fill_default_args(CallArgs, func, mangled_name) + self._append_eh_msg_out_arg(CallArgs, func, Gen) + CallArgs = Gen._apply_auto_addr(mangled_name, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{MethodName}") + self._check_eh_return(result, mangled_name, Gen, called_func=func) + return result + return ir.Constant(ir.IntType(32), 0) + CallArgs = [] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + CallArgs.append(ArgVal) + ClassName = None + if isinstance(Node.func.value, ast.Name): + ClassName = Gen.var_struct_class.get(Node.func.value.id) + if not ClassName and isinstance(Node.func.value, ast.Attribute): + # 链式属性访问（如 func.Params.append）：通过 _get_var_class 推导泛型特化类名 + ClassName = self.Trans.ExprHandler._get_var_class(Node.func.value, Gen) + if not ClassName: + for CN, ST in Gen.structs.items(): + if isinstance(ObjVal.type, ir.PointerType) and ObjVal.type.pointee == ST: + ClassName = CN + break + if not ClassName and self._is_char_pointer(ObjVal): + MethodName = Node.func.attr + # FakeDuck: 检查是否有影子结构体 a__meta__ (per-instance 字段存储) + meta_ptr: ir.Value | None = None + if isinstance(Node.func.value, ast.Name): + meta_name = f"{Node.func.value.id}__meta__" + if meta_name in Gen.variables: + meta_ptr = Gen.variables[meta_name] + for CN, ST in Gen.structs.items(): + method_key = f'{CN}.{MethodName}' + if not Gen._has_function(method_key): + continue + # FakeDuck: 不跳过空结构体 (扩展类可能没有数据成员只有方法) + try: + # 如果有影子结构体且该类有字段, 使用影子结构体作为 self (而非 bitcast char*) + if meta_ptr is not None and hasattr(ST, 'elements') and len(ST.elements) > 0: + ObjVal = meta_ptr + elif CN == '_str' and hasattr(ST, 'elements') and len(ST.elements) > 0: + # FakeDuck: 字符串字面量/匿名 str 表达式调用 _str 方法时， + # 创建临时影子结构体 (栈上), 设置 __data__ 和 __mbuddy__ 字段。 + # bitcast i8* → _str* 会导致读取垃圾指针 → ACCESS_VIOLATION + tmp_meta: ir.AllocaInstr = Gen._allocaEntry(ST, name="tmp_str_meta") + _data_off = Gen._get_member_offset('__data__', '_str') + if _data_off is not None: + _data_ptr = Gen.builder.gep(tmp_meta, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _data_off)], name="tmp_str_data") + Gen._store(ObjVal, _data_ptr) + _mpool_off = Gen._get_member_offset('__mbuddy__', '_str') + if _mpool_off is not None: + _mpool_ptr = Gen.builder.gep(tmp_meta, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), _mpool_off)], name="tmp_str_mbuddy") + Gen._store(ir.Constant(ir.PointerType(ir.IntType(8)), None), _mpool_ptr) + # 从 with 上下文自动注入 requires 字段 (如 __mbuddy__) + self._InjectRequiresFields(tmp_meta, '_str', Gen) + ObjVal = tmp_meta + else: + casted = Gen.builder.bitcast(ObjVal, ir.PointerType(ST), name=f"cast_{CN}") + ObjVal = casted + ClassName = CN + break + except Exception: # 回退：bitcast 失败时尝试下一个类 + continue + if not ClassName and isinstance(Node.func.value, ast.Call): + if isinstance(Node.func.value.func, ast.Attribute): + InnerAttrName = Node.func.value.func.attr + for CN in Gen.structs: + if CN.endswith(f'__{InnerAttrName}') or f'__{InnerAttrName}' in CN: + continue + method_name = f'{CN}__{InnerAttrName}' + if Gen._has_function(method_name): + func = Gen._get_function(method_name) + ret_type = func.fnty.return_type + if isinstance(ret_type, ir.PointerType): + for CN2, ST in Gen.structs.items(): + if ret_type.pointee == ST: + ClassName = CN2 + break + elif isinstance(ret_type, ir.IdentifiedStructType): + for CN2, ST in Gen.structs.items(): + if ret_type == ST: + ClassName = CN2 + break + if ClassName: + break + if not ClassName: + return ir.Constant(ir.IntType(32), 0) + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + class_sha1 = struct_sha1_map.get(ClassName) + if class_sha1: + FullMethodName = f'{class_sha1}.{ClassName}.{MethodName}' + else: + FullMethodName = f'{ClassName}.{MethodName}' + + # 提前查找函数以获取参数信息 + func_for_kw = Gen._find_function(FullMethodName) + if not func_for_kw: + func_for_kw = Gen._find_function(f'{FullMethodName}__') + if not func_for_kw and class_sha1: + func_for_kw = Gen._find_function(f'{ClassName}.{MethodName}') + + # 处理关键字参数: 先用 _fill_default_args 填充默认值，再用 keyword args 覆盖 + if Node.keywords and func_for_kw: + # 优先从全局参数名表获取 Python 参数名（支持跨模块 include 函数） + global_param_names = getattr(self.Trans, '_global_function_param_names', {}) + param_names = global_param_names.get(FullMethodName) + if not param_names: + param_names = global_param_names.get(f'{ClassName}.{MethodName}') + if not param_names: + # 回退到 LLVM 函数参数名（仅当参数名不是自动生成的数字名时有效） + llvm_names = [arg.name for arg in func_for_kw.args] if func_for_kw.args else [] + # 检测是否为 LLVM 自动生成的数字名（如 "0", "1", "2"） + if llvm_names and not all(n.isdigit() for n in llvm_names): + param_names = llvm_names + if param_names: + # 构造临时 call_args (含 self) 调用 _fill_default_args 填充默认值 + temp_call_args = [ObjVal] + list(CallArgs) + self._fill_default_args(temp_call_args, func_for_kw, FullMethodName) + # 提取回非 self 参数 (现在已填充默认值) + CallArgs.clear() + CallArgs.extend(temp_call_args[1:]) + # 用 keyword args 覆盖对应位置 + for kw in Node.keywords: + if kw.arg is None: + continue + ArgVal = self.HandleExprLlvm(kw.value) + if not ArgVal: + continue + kw_name = kw.arg + if kw_name in param_names: + idx = param_names.index(kw_name) + # idx 0 是 self, 所以 CallArgs 中的偏移是 idx-1 + target_idx = idx - 1 + if 0 <= target_idx < len(CallArgs): + CallArgs[target_idx] = ArgVal + + # 按需 vtable 注册：当 ClassName 不在 vtable 集合中但有 class_sha1 时， + # 从 .pyi stub 检查 @t.CVTable 装饰器并注册。 + # 修复场景：list[str].__new__ 中 pool.alloc(48) 编译时， + # MemManager 可能因导入时序问题未注册到当前 Gen 的 vtable 集合， + # 导致生成 call_（直接调用）而非 vcall_（虚分派），运行时段错误。 + if (ClassName not in Gen.class_vtable + and ClassName not in Gen._cross_module_vtable_classes + and class_sha1): + self._try_register_vtable_ondemand(ClassName, class_sha1, Gen) + + if (ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes) and ClassName in Gen.class_methods: + methods = Gen.class_methods[ClassName] + method_idx = -1 + for mi, mn in enumerate(methods): + if mn == FullMethodName or mn == MethodName or mn.endswith(f'.{MethodName}'): + method_idx = mi + break + if method_idx >= 0: + func = Gen._find_function(FullMethodName) + if not func: + func = Gen._find_function(f'{FullMethodName}__') + if not func: + p = Gen.class_parent.get(ClassName) + while p: + parent_full = f'{p}.{MethodName}' + func = Gen._find_function(parent_full) + if not func: + func = Gen._find_function(f'{parent_full}__') + if func: + break + p = Gen.class_parent.get(p) + if func: + VtableSlotPtr = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name=f"vtable_slot_{ClassName}") + VtablePtr = Gen._load(VtableSlotPtr, name=f"vtable_ptr_{ClassName}") + VtableArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + VtableTyped = Gen.builder.bitcast(VtablePtr, ir.PointerType(VtableArrayType), name=f"vtable_typed_{ClassName}") + MethodPtrAddr = Gen.builder.gep(VtableTyped, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), method_idx)], name=f"vmethod_ptr_addr_{MethodName}") + MethodPtrI8 = Gen._load(MethodPtrAddr, name=f"vmethod_ptr_{MethodName}") + FuncPtrType = ir.PointerType(func.function_type) + MethodPtrTyped = Gen.builder.bitcast(MethodPtrI8, FuncPtrType, name=f"vmethod_typed_{MethodName}") + call_args = [ObjVal] + CallArgs + self._fill_default_args(call_args, func, FullMethodName) + self._append_eh_msg_out_arg(call_args, func, Gen) + adjusted = Gen._adjust_args(call_args, func) + result = Gen.builder.call(MethodPtrTyped, adjusted, name=f"vcall_{FullMethodName}") + self._check_eh_return(result, FullMethodName, Gen, called_func=func) + return result + func = Gen._find_function(FullMethodName) + if not func: + func = Gen._find_function(f'{FullMethodName}__') + if not func and class_sha1: + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(FullMethodName, Gen) + if stub_func_type: + func = self.GetOrCreateStubFuncDecl(FullMethodName, stub_func_type, Gen) + if not func and class_sha1: + plain_name = f'{ClassName}.{MethodName}' + stub_func_type = self.Trans.ImportHandler._LookupStubFuncType(plain_name, Gen) + if stub_func_type: + func = self.GetOrCreateStubFuncDecl(FullMethodName, stub_func_type, Gen) + # 最终回退：尝试无 sha1 前缀的函数名（与 func_for_kw 的双路径回退一致） + if not func: + plain_name = f'{ClassName}.{MethodName}' + func = Gen._find_function(plain_name) + if func: + call_args = [ObjVal] + CallArgs + self._fill_default_args(call_args, func, FullMethodName) + self._append_eh_msg_out_arg(call_args, func, Gen) + call_args = Gen._apply_auto_addr(FullMethodName, call_args) + adjusted = Gen._adjust_args(call_args, func) + result = Gen.builder.call(func, adjusted, name=f"call_{FullMethodName}") + self._check_eh_return(result, FullMethodName, Gen, called_func=func) + return result + i8ptr = ir.PointerType(ir.IntType(8)) + member_idx = -1 + if ClassName in Gen.class_members: + for mi, (mn, mt) in enumerate(Gen.class_members[ClassName]): + if mn == MethodName: + member_idx = mi + break + if member_idx >= 0 and isinstance(ObjVal.type, ir.PointerType): + has_vtable = ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes + base = 1 if has_vtable else 0 + member_ptr = Gen.builder.gep(ObjVal, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), base + member_idx)], name=f"funcptr_{MethodName}_ptr") + raw_ptr = Gen._load(member_ptr, name=f"funcptr_{MethodName}_raw") + func_ptr_type = ir.PointerType(ir.FunctionType(ir.IntType(32), [i8ptr] + [a.type for a in CallArgs])) + typed_ptr = Gen.builder.bitcast(raw_ptr, func_ptr_type, name=f"funcptr_{MethodName}_typed") + obj_as_i8 = Gen.builder.bitcast(ObjVal, i8ptr, name=f"funcptr_self_{ClassName}") + call_args = [obj_as_i8] + CallArgs + result = Gen.builder.call(typed_ptr, call_args, name=f"call_funcptr_{ClassName}.{MethodName}") + return result + if class_sha1 and ClassName in Gen.structs: + struct_type = Gen.structs[ClassName] + self_ptr_type = ir.PointerType(struct_type) + param_types = [self_ptr_type] + [a.type for a in CallArgs] + func_type = ir.FunctionType(ir.IntType(32), param_types) + func = ir.Function(Gen.module, func_type, name=FullMethodName) + Gen.functions[FullMethodName] = func + if isinstance(ObjVal.type, ir.PointerType) and ObjVal.type.pointee == struct_type: + call_self = ObjVal + else: + call_self = Gen.builder.bitcast(ObjVal, self_ptr_type, name=f"method_self_{ClassName}") + adjusted = Gen._adjust_args([call_self] + CallArgs, func) + result = Gen.builder.call(func, adjusted, name=f"call_{FullMethodName}") + return result + raise Exception(f"Undefined method '{MethodName}' in class '{ClassName}' (no function declaration found for '{FullMethodName}')") + + _C_LIB_FUNCS = { + 'malloc': lambda: (ir.PointerType(ir.IntType(8)), [ir.IntType(64)]), + 'calloc': lambda: (ir.PointerType(ir.IntType(8)), [ir.IntType(64), ir.IntType(64)]), + 'realloc': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'free': lambda: (ir.VoidType(), [ir.PointerType(ir.IntType(8))]), + 'strlen': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'strlength': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'memset': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(32), ir.IntType(64)]), + 'memcpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'memmove': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'memcmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'strncmp': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'strncpy': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64)]), + 'strcat': lambda: (ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8))]), + 'puts': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8))]), + 'atoi': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8))]), + 'atol': lambda: (ir.IntType(64), [ir.PointerType(ir.IntType(8))]), + 'abs': lambda: (ir.IntType(32), [ir.IntType(32)]), + 'labs': lambda: (ir.IntType(64), [ir.IntType(64)]), + 'system': lambda: (ir.IntType(32), [ir.PointerType(ir.IntType(8))]), + } + + def _try_declare_c_lib_func(self, func_name: str, Gen: LlvmGeneratorMixin) -> ir.Function | None: + if func_name not in self._C_LIB_FUNCS: + return None + ret_type, param_types = self._C_LIB_FUNCS[func_name]() + func_type = ir.FunctionType(ret_type, param_types) + func = ir.Function(Gen.module, func_type, name=func_name) + Gen.functions[func_name] = func + return func + + def _HandleLenLlvm(self, arg_node: ast.AST) -> ir.Value | None: + Gen = self.Trans.LlvmGen + if isinstance(arg_node, ast.Attribute) and isinstance(arg_node.value, ast.Name) and arg_node.value.id == 'self': + attr_name = arg_node.attr + len_member = f'{attr_name}__len' + current_class = self.Trans._CurrentCpythonObjectClass + if current_class and current_class in Gen.class_members: + for m_name, m_type in Gen.class_members[current_class]: + if m_name == len_member: + self_val = Gen._get_var_ptr('self') + if self_val: + self_ptr = Gen._load(self_val, name="self_for_len") + offset = Gen._get_member_offset(len_member, current_class) + len_ptr = Gen.builder.gep(self_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), offset)], name=f"self_{len_member}_ptr") + return Gen._load(len_ptr, name=f"self_{len_member}") + val = self.HandleExprLlvm(arg_node) + if not val: + return None + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if isinstance(pointee, ir.PointerType): + val = Gen._load(val, name="len_deref") + pointee = val.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + result = self.Trans.ExprUtils._try_operator_overLoad(CN, '__len__', val, Gen) + if result is not None: + return result + break + if isinstance(pointee, ir.ArrayType): + return ir.Constant(ir.IntType(64), pointee.count) + if isinstance(pointee, (ir.FloatType, ir.DoubleType, ir.IntType)) and pointee.width != 8: + return ir.Constant(ir.IntType(64), 0) + if val.type != ir.PointerType(ir.IntType(8)): + val = Gen.builder.bitcast(val, ir.PointerType(ir.IntType(8)), name="str_ptr_cast") + correct_strlen_type = ir.FunctionType(ir.IntType(64), [ir.PointerType(ir.IntType(8))]) + strlen_func = None + if 'strlen' in Gen.functions: + existing = Gen.functions['strlen'] + if existing.ftype == correct_strlen_type: + strlen_func = existing + if not strlen_func: + try: + strlen_func = ir.Function(Gen.module, correct_strlen_type, name='strlen') + Gen.functions['strlen'] = strlen_func + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + if strlen_func: + result = Gen.builder.call(strlen_func, [val], name="strlen_call") + return result + elif isinstance(val.type, ir.ArrayType): + return ir.Constant(ir.IntType(64), val.type.count) + return ir.Constant(ir.IntType(64), 0) + + def _resolve_opaque_struct_members(self, struct_type: ir.Type, Gen: LlvmGeneratorMixin, _visited: set | None = None) -> None: + """Recursively resolve opaque struct members by loading from stub files.""" + if _visited is None: + _visited = set() + if not isinstance(struct_type, ir.IdentifiedStructType): + return + # 防止循环引用导致无限递归 + try: + st_id = id(struct_type) + except Exception: + st_id = struct_type + if st_id in _visited: + return + _visited.add(st_id) + if struct_type.elements is None or len(struct_type.elements) == 0: + # Find the class name for this opaque struct and try to resolve + for cn, st in Gen.structs.items(): + try: + if st == struct_type or (hasattr(st, 'name') and hasattr(struct_type, 'name') and st.name == struct_type.name): + self.Trans.ImportHandler._TryLoadStructFromStub(cn, Gen) + break + except (AssertionError, RuntimeError): + pass + # Re-get the struct in case it was resolved + if struct_type.elements: + for elem in struct_type.elements: + if isinstance(elem, ir.IdentifiedStructType): + self._resolve_opaque_struct_members(elem, Gen, _visited) + elif isinstance(elem, ir.PointerType): + pointee = elem.pointee + if isinstance(pointee, ir.IdentifiedStructType): + self._resolve_opaque_struct_members(pointee, Gen, _visited) + + def _resolve_type_arg_str(self, node: ast.AST) -> str: + """将类型参数 AST 节点解析为规范化的类型字符串（用于泛型特化的 mangling）。 + + 走和普通类型签名同样的规范化路径（不单独创建规则）： + - ast.Name: 若为已注册结构体，加 SHA1 前缀（如 'AST' → '9ada227b160c07a1.AST'）， + 与 DeclarationGenerator._get_type_str 对 ast.Name 的处理一致 + - ast.Attribute: t.CPtr → 'ptr'（指针标记，等价于普通签名中 t.CPtr → i8* 的指针语义； + 用 'ptr' 而非 '*' 是因为 _get_or_create_struct 的 clean_name 会移除 '*'， + 导致 Phase2 直接编译与 .stub.ll 加载路径的 mangled 名不一致）； + 其他 t 模块类型按 CTypeRegistry.NameToLLVM 解析 + - ast.BinOp: 递归处理左右操作数并直接拼接（如 AST | t.CPtr → 'sha1.ASTptr'）， + 避免空格/管道符导致 _get_or_create_struct 的 clean_name 清洗不一致 + """ + if isinstance(node, ast.Name): + tn: str = node.id + t_c_imported = getattr(self.Trans, '_t_c_imported_names', {}) + if tn in t_c_imported: + src_module, src_name = t_c_imported[tn] + return f'{src_module}.{src_name}' + # 若为已注册结构体，加 SHA1 前缀（和普通签名路径一致） + Gen = self.Trans.LlvmGen + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) if Gen is not None else {} + if tn in struct_sha1_map: + return f'{struct_sha1_map[tn]}.{tn}' + return tn + if isinstance(node, ast.Attribute): + if isinstance(node.value, ast.Name): + full_name: str = f'{node.value.id}.{node.attr}' + # t.CPtr → 'ptr'（指针标记，和普通签名中 t.CPtr 被解析为指针类型 i8* 语义一致） + if full_name == 't.CPtr': + return 'ptr' + # 其他 t 模块类型按 CTypeRegistry 解析（和普通签名路径一致） + llvm_str: str | None = CTypeRegistry.NameToLLVM(node.attr) + if llvm_str is not None: + return llvm_str + return full_name + return ast.unparse(node) + if isinstance(node, ast.BinOp): + left_str: str = self._resolve_type_arg_str(node.left) + right_str: str = self._resolve_type_arg_str(node.right) + # 直接拼接（如 'sha1.AST' + 'ptr' = 'sha1.ASTptr'），避免空格导致清洗不一致 + return f'{left_str}{right_str}' + return ast.unparse(node) + + def _get_raw_type_name(self, node: ast.AST) -> str: + """获取类型节点的原始名称（不带模块前缀），用于泛型 T 替换。 + + ast.Name → sn.id (如 'AST', 'CInt') + ast.Attribute → sn.attr (如 t.CInt → 'CInt') + ast.BinOp → 左侧的原始名称 (如 AST | CPtr → 'AST') + """ + if isinstance(node, ast.Name): + return node.id + if isinstance(node, ast.Attribute): + return node.attr + if isinstance(node, ast.BinOp): + return self._get_raw_type_name(node.left) + return ast.unparse(node) + + def _resolve_generic_slice_type(self, sn: ast.AST) -> tuple[str, str] | None: + """解析泛型 slice 中的单个类型参数节点，返回 (type_arg, type_name) 或 None。 + + type_arg: 用于 mangling，走和普通签名同样的规范化路径 + （如 'AST | t.CPtr' → '9ada227b160c07a1.ASTptr'，AST 加 SHA1 前缀，t.CPtr 处理为 ptr） + type_name: 用于替换泛型类型参数 T（如 'AST' 或 'CInt'，_specialize_generic_class 会自动加 | t.CPtr 给类名） + """ + if isinstance(sn, ast.Name): + tn: str = sn.id + ts: str = 'int' + if tn == 'int': + ts, tn = 'int', 'CInt' + elif tn == 'double': + ts, tn = 'double', 'CDouble' + elif tn == 'float': + ts, tn = 'float', 'CFloat' + elif tn == 'char': + ts, tn = 'i8', 'CChar' + elif tn == 'bool': + ts, tn = 'i8', 'CBool' + elif tn in ('CInt', 'CUInt', 'CInt32T', 'CUInt32T', 'CInt64T', 'CUInt64T', 'CInt16T', 'CUInt16T', 'CInt8T', 'CUInt8T', 'CChar', 'CShort', 'CLong'): + ts = 'int' + elif tn in ('CDouble', 'CFloat', 'CFloat64T', 'CFloat32T'): + ts = 'double' + else: + ts = tn + return (ts, tn) + # BinOp (如 AST | CPtr) 或 Attribute (如 t.CPtr / t.CInt) + type_str: str = self._resolve_type_arg_str(sn) + # type_name 取原始名称（不带模块前缀），用于 T 替换 + raw_name: str = self._get_raw_type_name(sn) + return (type_str, raw_name) + + def _ResolveGenericSlice(self, Sub: ast.Subscript) -> str | None: + """从泛型 Subscript (如 Stack[int]) 解析特化名，确保特化类型已生成。""" + if not isinstance(Sub.value, ast.Name): + return None + ClassName: str = Sub.value.id + if not (hasattr(self.Trans.ClassHandler, '_generic_class_templates') and ClassName in self.Trans.ClassHandler._generic_class_templates): + return None + Gen = self.Trans.LlvmGen + # 共享符号表构建阶段（_build_shared_symbol_data / ParsePythonFile）LlvmGen 为 None， + # 此时不应发射特化 IR；调用方（GetCTypeInfo）已用 mangled 名设置符号表信息 + if Gen is None: + return None + template: dict = self.Trans.ClassHandler._generic_class_templates[ClassName] + type_param_names: list[str] = template['type_params'] + type_args: list[str] = [] + type_names: list[str] = [] + slice_nodes: list[ast.AST] = [] + if isinstance(Sub.slice, ast.Tuple): + slice_nodes = list(Sub.slice.elts) + else: + slice_nodes = [Sub.slice] + for sn in slice_nodes: + result = self._resolve_generic_slice_type(sn) + if result is not None: + type_args.append(result[0]) + type_names.append(result[1]) + while len(type_args) < len(type_param_names): + type_args.append('int') + type_names.append('CInt') + return self.Trans.ClassHandler._specialize_generic_class(ClassName, type_args, Gen, type_names=type_names) + + @staticmethod + def _AnnotationContainsCPtr(Node: ast.AST) -> bool: + """检查类型注解 AST 是否包含 t.CPtr（即是否为指针类型）""" + if isinstance(Node, ast.Attribute): + if Node.attr == 'CPtr': + return True + if isinstance(Node, ast.BinOp) and isinstance(Node.op, ast.BitOr): + return (ExprCallHandle._AnnotationContainsCPtr(Node.left) + or ExprCallHandle._AnnotationContainsCPtr(Node.right)) + return False + + @staticmethod + def _is_sizeof_type_name(name: str, Gen: "LlvmGeneratorMixin") -> bool: + """检查名称是否是已知的类型名（用于 __sizeof__ 上下文）。 + 检查顺序: Gen.structs → SHA1 前缀 → 内建类型 → CTypeRegistry + """ + if not name: + return False + # 1. 直接在 Gen.structs 中查找 + if name in Gen.structs: + return True + # 2. 带 SHA1 前缀查找（如 ClassDef → 657e182b27c2a022.ClassDef） + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + if name in struct_sha1_map: + return True + # 遍历所有 SHA1 前缀 + for sha1 in struct_sha1_map.values(): + prefixed = f"{sha1}.{name}" + if prefixed in Gen.structs: + return True + # 3. 内建类型名 + if name in ('str', 'bytes', 'int', 'double', 'float', 'char', 'bool', + 'void', 'i8', 'i16', 'i32', 'i64', 'half', 'fp128'): + return True + # 4. CTypeRegistry 中的类型 + if CTypeRegistry.GetClassByName(name) is not None: + return True + if CTypeRegistry.ResolveName(name): + return True + return False + + def _HandleSizeofLlvm(self, Node: ast.AST) -> ir.Constant: + Gen = self.Trans.LlvmGen + if isinstance(Node, ast.Name): + type_name = Node.id + elif isinstance(Node, ast.Attribute): + type_name = Node.attr + else: + return ir.Constant(ir.IntType(64), 0) + + if type_name not in Gen.structs: + # SHA1 前缀查找：如 ClassDef → 657e182b27c2a022.ClassDef + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + resolved_name: str | None = None + if type_name in struct_sha1_map: + resolved_name = f"{struct_sha1_map[type_name]}.{type_name}" + else: + for sha1 in struct_sha1_map.values(): + prefixed = f"{sha1}.{type_name}" + if prefixed in Gen.structs: + resolved_name = prefixed + break + if resolved_name is not None and resolved_name in Gen.structs: + type_name = resolved_name + if type_name not in Gen.structs: + # str/bytes 是指针类型 (char*/void*)，sizeof = 指针大小 + if type_name in ('str', 'bytes'): + return ir.Constant(ir.IntType(64), 8) + ctype_cls = CTypeRegistry.GetClassByName(type_name) + if ctype_cls is not None: + try: + ctype_inst = ctype_cls() + size_bits = getattr(ctype_inst, 'Size', 0) or 0 + size_bytes = size_bits // 8 + return ir.Constant(ir.IntType(64), size_bytes) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + resolved = CTypeRegistry.ResolveName(type_name) + if resolved: + ctype_cls, ptr_level = resolved + try: + ctype_inst = ctype_cls() + size_bits = getattr(ctype_inst, 'Size', 0) or 0 + size_bytes = size_bits // 8 + if ptr_level > 0: + size_bytes = 8 + return ir.Constant(ir.IntType(64), size_bytes) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型失败: {_e}", "Exception") + # 所有查找策略均失败（不在 Gen.structs、SHA1 前缀查找失败、CTypeRegistry 也找不到）： + # 报错终止，拒绝静默返回 0（会导致 malloc(0)/alloc(0) 堆溢出） + raise ValueError( + f"[FATAL][__sizeof__] 类型 '{type_name}' 无法解析" + f"（不在 Gen.structs、SHA1 前缀查找失败、CTypeRegistry 也找不到），" + f"拒绝静默返回 0" + ) + + struct_type = Gen.structs[type_name] + if isinstance(struct_type, ir.PointerType): + struct_type = struct_type.pointee + # 保存 builder 状态：_TryLoadStructFromStub / _resolve_opaque_struct_members + # 可能触发重入编译（编译其他模块的函数），这会改变 Gen.func/Gen.builder， + # 导致当前函数的指令被写到错误函数的 block 中（then 块为空的根因） + _saved_func = Gen.func + _saved_builder = Gen.builder + _saved_block = Gen.builder.block if (Gen.builder and Gen.builder.block) else None + _saved_variables = dict(Gen.variables) if Gen.variables else {} + _saved_reg_values = dict(getattr(Gen, '_reg_values', {})) + _saved_direct_values = dict(getattr(Gen, '_direct_values', {})) + _saved_var_type_info = dict(getattr(Gen, 'var_type_info', {})) + _saved_var_signedness = dict(getattr(Gen, 'var_signedness', {})) + _saved_global_vars = set(getattr(Gen, 'global_vars', set())) + _saved_var_scopes = [dict(s) for s in getattr(self.Trans, 'VarScopes', [])] + try: + if isinstance(struct_type, ir.IdentifiedStructType) and (struct_type.elements is None or len(struct_type.elements) == 0): + self.Trans.ImportHandler._TryLoadStructFromStub(type_name, Gen) + struct_type = Gen.structs.get(type_name, struct_type) + # Recursively resolve opaque struct members + self._resolve_opaque_struct_members(struct_type, Gen) + finally: + Gen.func = _saved_func + Gen.builder = _saved_builder + if _saved_block is not None and Gen.builder is not None: + Gen.builder.position_at_end(_saved_block) + Gen.variables = _saved_variables + Gen._reg_values = _saved_reg_values + Gen._direct_values = _saved_direct_values + Gen.var_type_info = _saved_var_type_info + Gen.var_signedness = _saved_var_signedness + Gen.global_vars = _saved_global_vars + self.Trans.VarScopes = _saved_var_scopes + size = Gen._get_struct_size(struct_type) + if size > 0: + return ir.Constant(ir.IntType(64), size) + return ir.Constant(ir.IntType(64), 0) + + def _HandleAbsLlvm(self, arg_node: ast.AST) -> ir.Value | None: + Gen = self.Trans.LlvmGen + val = self.HandleExprLlvm(arg_node) + if not val: + return None + if isinstance(val.type, ir.PointerType): + pointee = val.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + result = self.Trans.ExprUtils._try_operator_overLoad(CN, '__abs__', val, Gen) + if result is not None: + return result + break + if isinstance(val.type, ir.IntType): + neg_val = Gen.builder.neg(val, name="abs_neg") + is_neg = Gen.builder.icmp_signed('<', val, ir.Constant(val.type, 0), name="is_neg") + return Gen.builder.select(is_neg, neg_val, val, name="abs_result") + elif isinstance(val.type, (ir.FloatType, ir.DoubleType)): + zero = ir.Constant(val.type, 0.0) + neg_val = Gen.builder.fneg(val, name="fabs_neg") + is_neg = Gen.builder.fcmp_ordered('<', val, zero, name="f_is_neg") + return Gen.builder.select(is_neg, neg_val, val, name="fabs_result") + return None + + def _HandleDirLlvm(self, Node: ast.Call) -> ir.Constant: + Gen = self.Trans.LlvmGen + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _HandleTypeLlvm(self, Node: ast.Call) -> ir.Value: + Gen = self.Trans.LlvmGen + if not Node.args: + type_str = "" + else: + arg = Node.args[0] + enum_type_name = None + + def _get_attr_path(node): + if isinstance(node, ast.Name): + return node.id + elif isinstance(node, ast.Attribute): + return f"{_get_attr_path(node.value)}.{node.attr}" + return None + + if isinstance(arg, ast.Name): + arg_name = arg.id + Gen = self.Trans.LlvmGen + if arg_name in Gen.var_type_info: + type_info = Gen.var_type_info[arg_name] + if isinstance(type_info, dict) and type_info.get('type') == 'enum': + enum_type_name = type_info.get('name') + if not enum_type_name: + TypeInfo = self.Trans.SymbolTable.lookup(arg_name) + if TypeInfo and TypeInfo.IsEnum: + enum_type_name = arg_name + elif isinstance(arg, ast.Attribute): + LastPart = None + enum_class_name = None + if isinstance(arg.value, ast.Name): + LastPart = arg.attr + AttrInfo = self.Trans.SymbolTable.lookup(LastPart) + if AttrInfo and AttrInfo.IsEnumMember and AttrInfo.EnumName: + enum_type_name = AttrInfo.EnumName + elif isinstance(arg.value, ast.Attribute): + attr_path = _get_attr_path(arg) + if attr_path: + parts = attr_path.split('.') + if len(parts) >= 2: + enum_class_name = parts[-2] + LastPart = parts[-1] + qualified_name_dot = f"{enum_class_name}.{LastPart}" + qualified_name_under = f"{enum_class_name}_{LastPart}" + enum_member_found = None + for qname in (qualified_name_dot, qualified_name_under): + info = self.Trans.SymbolTable.lookup(qname) + if info and info.IsEnumMember and info.EnumName == enum_class_name: + enum_member_found = info + break + if not enum_member_found: + for key in self.Trans.SymbolTable: + if key == LastPart: + info = self.Trans.SymbolTable[key] + if info.IsEnumMember and info.EnumName == enum_class_name: + enum_member_found = info + break + if enum_member_found: + enum_type_name = enum_class_name + if enum_type_name: + type_str = f"" + else: + llvm_type = self.Trans.ExprUtils._infer_expr_llvm_type_full(arg) + if isinstance(llvm_type, ir.PointerType): + pointee = llvm_type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + llvm_type = pointee + type_info = self.Trans.ExprUtils._llvm_type_to_detailed_string(llvm_type) + type_str = f"" + return Gen.emit_constant(type_str, 'string') + + def _HandleInputLlvm(self, Node: ast.Call) -> ir.Value: + Gen = self.Trans.LlvmGen + if Node.args and isinstance(Node.args[0], ast.Constant) and isinstance(Node.args[0].value, str): + prompt = Node.args[0].value + prompt_ptr = Gen.emit_constant(prompt, 'string') + printf_func = Gen.get_or_declare_c_func('printf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + Gen.builder.call(printf_func, [prompt_ptr]) + scanf_func = Gen.get_or_declare_c_func('scanf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + input_buffer = Gen._alloca(ir.ArrayType(ir.IntType(8), 256), name="input_buffer") + result = Gen.builder.call(scanf_func, [Gen.emit_constant("%255s", 'string'), input_buffer]) + return Gen.builder.gep(input_buffer, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="input_result") + + def _HandleAtoiLlvm(self, str_ptr: ir.Value) -> ir.Value: + Gen = self.Trans.LlvmGen + atoi_func = Gen.get_or_declare_c_func('atoi', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))])) + return Gen.builder.call(atoi_func, [str_ptr], name="atoi_result") + + def _HandleOrdLlvm(self, expr_node: ast.AST) -> ir.Value: + Gen = self.Trans.LlvmGen + val = self.HandleExprLlvm(expr_node) + if val and self._is_char_pointer(val): + return Gen._load(val, name="ord_result") + return ir.Constant(ir.IntType(32), 0) + + def _HandleChrLlvm(self, expr_node: ast.AST) -> ir.Value: + Gen = self.Trans.LlvmGen + val = self.HandleExprLlvm(expr_node) + if val: + char_ptr = Gen._alloca(ir.IntType(8), name="chr_char") + Gen.builder.store(Gen.builder.trunc(val, ir.IntType(8), name="trunc_chr"), char_ptr) + return Gen.builder.gep(char_ptr, [ir.Constant(ir.IntType(32), 0)], name="chr_result") + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _EmitStringFromValue(self, val: ir.Value) -> ir.Value: + Gen = self.Trans.LlvmGen + if isinstance(val.type, ir.IntType): + buf = Gen._alloca(ir.IntType(8), name="str_buf", size=ir.Constant(ir.IntType(32), 32)) + snprintf_func = Gen.get_or_declare_c_func('snprintf', ir.FunctionType(ir.IntType(32), [ + ir.PointerType(ir.IntType(8)), + ir.IntType(32), + ir.PointerType(ir.IntType(8)) + ])) + fmt_ptr = Gen.emit_constant("%d", 'string') + Gen.builder.call(snprintf_func, [buf, ir.Constant(ir.IntType(32), 32), fmt_ptr, val]) + return Gen.builder.gep(buf, [ir.Constant(ir.IntType(32), 0)], name="str_result") + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _HandleMallocLlvm(self, args: list[ast.AST]) -> ir.Value: + Gen = self.Trans.LlvmGen + malloc_type = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.IntType(64)]) + malloc_func = Gen._get_or_declare_func('malloc', malloc_type) + param_type = malloc_func.type.pointee.args[0] + size_val = None + if args: + size_val = self.HandleExprLlvm(args[0]) + if size_val and isinstance(size_val.type, ir.IntType): + if isinstance(param_type, ir.IntType) and size_val.type.width != param_type.width: + if size_val.type.width < param_type.width: + size_val = Gen.builder.zext(size_val, param_type, name="zext_malloc_size") + else: + size_val = Gen.builder.trunc(size_val, param_type, name="trunc_malloc_size") + if not size_val: + size_val = ir.Constant(param_type, 1) + result = Gen.builder.call(malloc_func, [size_val], name="malloc_result") + return Gen.builder.bitcast(result, ir.IntType(8).as_pointer(), name="malloc_cast") + + def _HandleReallocLlvm(self, args: list[ast.AST]) -> ir.Value: + Gen = self.Trans.LlvmGen + realloc_type = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(64)]) + realloc_func = Gen._get_or_declare_func('realloc', realloc_type) + size_param_type = realloc_func.type.pointee.args[1] + ptr_val = None + size_val = None + if len(args) >= 2: + ptr_val = self.HandleExprLlvm(args[0]) + size_val = self.HandleExprLlvm(args[1]) + elif len(args) == 1: + size_val = self.HandleExprLlvm(args[0]) + if not ptr_val: + ptr_val = ir.Constant(ir.PointerType(ir.IntType(8)), None) + if isinstance(ptr_val.type, ir.PointerType) and not self._is_char_pointer(ptr_val): + ptr_val = Gen.builder.bitcast(ptr_val, ir.PointerType(ir.IntType(8)), name="realloc_cast_ptr") + if size_val and isinstance(size_val.type, ir.IntType): + if isinstance(size_param_type, ir.IntType) and size_val.type.width != size_param_type.width: + if size_val.type.width < size_param_type.width: + size_val = Gen.builder.zext(size_val, size_param_type, name="zext_realloc_size") + else: + size_val = Gen.builder.trunc(size_val, size_param_type, name="trunc_realloc_size") + if not size_val: + size_val = ir.Constant(size_param_type, 0) + result = Gen.builder.call(realloc_func, [ptr_val, size_val], name="realloc_result") + return Gen.builder.bitcast(result, ir.IntType(8).as_pointer(), name="realloc_cast") + + def _HandleFreeLlvm(self, args: list[ast.AST]) -> ir.Constant: + Gen = self.Trans.LlvmGen + free_type = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))]) + free_func = Gen._get_or_declare_func('free', free_type) + if args: + var_name = None + if isinstance(args[0], ast.Name): + var_name = args[0].id + ptr_val = self.HandleExprLlvm(args[0]) + if ptr_val: + if var_name and hasattr(Gen, '_var_to_heap_ptr') and var_name in Gen._var_to_heap_ptr: + registered_val = Gen._var_to_heap_ptr[var_name] + Gen._unregister_local_heap_ptr(registered_val) + del Gen._var_to_heap_ptr[var_name] + if isinstance(ptr_val.type, ir.PointerType) and not self._is_char_pointer(ptr_val): + ptr_val = Gen.builder.bitcast(ptr_val, ir.PointerType(ir.IntType(8)), name="free_cast_ptr") + Gen.builder.call(free_func, [ptr_val], name="free_call") + return ir.Constant(ir.IntType(32), 0) + + def _HandleMemcpyLlvm(self, args: list[ast.AST]) -> None: + Gen = self.Trans.LlvmGen + if 'llvm.memcpy' not in Gen.functions: + memcpy_type = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8)), ir.PointerType(ir.IntType(8)), ir.IntType(64), ir.IntType(1)]) + Gen.functions['llvm.memcpy'] = ir.Function(Gen.module, memcpy_type, name='llvm.memcpy') + if len(args) >= 3: + dst = self.HandleExprLlvm(args[0]) + src = self.HandleExprLlvm(args[1]) + size = self.HandleExprLlvm(args[2]) + if dst and src and size: + i8ptr = ir.PointerType(ir.IntType(8)) + if isinstance(dst.type, ir.PointerType) and dst.type != i8ptr: + dst = Gen.builder.bitcast(dst, i8ptr, name="memcpy_dst_cast") + elif isinstance(dst.type, ir.IntType): + if dst.type.width < 64: + dst = Gen.builder.zext(dst, ir.IntType(64), name="zext_dst_ptr") + dst = Gen.builder.inttoptr(dst, i8ptr, name="dst_int2ptr") + if isinstance(src.type, ir.PointerType) and src.type != i8ptr: + src = Gen.builder.bitcast(src, i8ptr, name="memcpy_src_cast") + elif isinstance(src.type, ir.IntType): + if src.type.width < 64: + src = Gen.builder.zext(src, ir.IntType(64), name="zext_src_ptr") + src = Gen.builder.inttoptr(src, i8ptr, name="src_int2ptr") + if isinstance(size.type, ir.IntType) and size.type.width < 64: + size = Gen.builder.zext(size, ir.IntType(64), name="zext_memcpy_size") + isvolatile = ir.Constant(ir.IntType(1), 0) + Gen.builder.call(Gen.functions['llvm.memcpy'], [dst, src, size, isvolatile], name="memcpy_call") + return None + + def _HandleMemsetLlvm(self, args: list[ast.AST]) -> None: + Gen = self.Trans.LlvmGen + memset_func = Gen.get_or_declare_c_func('memset', ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.PointerType(ir.IntType(8)), ir.IntType(32), ir.IntType(64)])) + if len(args) >= 3: + dst = self.HandleExprLlvm(args[0]) + val = self.HandleExprLlvm(args[1]) + size = self.HandleExprLlvm(args[2]) + if dst and val is not None and size: + if isinstance(dst.type, ir.PointerType) and not self._is_char_pointer(dst): + dst = Gen.builder.bitcast(dst, ir.PointerType(ir.IntType(8)), name="memset_dst_cast") + if isinstance(val.type, ir.IntType) and val.type.width != 32: + if val.type.width < 32: + val = Gen.builder.zext(val, ir.IntType(32), name="zext_memset_val") + else: + val = Gen.builder.trunc(val, ir.IntType(32), name="trunc_memset_val") + if isinstance(size.type, ir.IntType) and size.type.width < 64: + size = Gen.builder.zext(size, ir.IntType(64), name="zext_memset_size") + elif isinstance(size.type, ir.IntType) and size.type.width > 64: + size = Gen.builder.trunc(size, ir.IntType(64), name="trunc_memset_size") + Gen.builder.call(memset_func, [dst, val, size], name="memset_call") + return None + + def _HandleVaStartLlvm(self, args: list[ast.AST]) -> None: + Gen = self.Trans.LlvmGen + return None + + def _HandleVaEndLlvm(self, args: list[ast.AST]) -> None: + Gen = self.Trans.LlvmGen + return None + + def _HandleArgLlvm(self, args: list[ast.AST], CallNode: ast.Call | None = None) -> ir.Value | None: + Gen = self.Trans.LlvmGen + va_list_ptr = None + variadic_info = getattr(Gen, '_va_arg_info', None) + if not variadic_info: + variadic_info = getattr(Gen, '_variadic_info', None) + if variadic_info: + va_list_ptr = variadic_info.get('va_list_ptr') + if not va_list_ptr: + vararg_name = variadic_info.get('vararg_name', 'args') if variadic_info else 'args' + if vararg_name in Gen._direct_values: + va_list_ptr = Gen._direct_values[vararg_name] + elif vararg_name in Gen.variables: + va_list_ptr = Gen.variables[vararg_name] + if not va_list_ptr: + return ir.Constant(ir.IntType(32), 0) + target_type = ir.IntType(32) + if args: + type_arg = args[0] + if isinstance(type_arg, ast.Name): + type_name = type_arg.id + if type_name in ('str', 'bytes'): + target_type = ir.PointerType(ir.IntType(8)) + elif type_name == 'CPtr': + target_type = ir.PointerType(ir.IntType(8)) + elif type_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[type_name]) + else: + ctype_cls = CTypeRegistry.GetClassByName(type_name) + if ctype_cls is not None: + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + else: + resolved = CTypeRegistry.ResolveName(type_name) + if resolved is not None: + ctype_cls, ptr_level = resolved + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str: + base = Gen._type_str_to_llvm(llvm_str) + if base is not None: + for _ in range(ptr_level): + if isinstance(base, ir.VoidType): + base = ir.IntType(8).as_pointer() + else: + base = ir.PointerType(base) + target_type = base + elif isinstance(type_arg, ast.Attribute): + if isinstance(type_arg.value, ast.Name) and IsTModule(type_arg.value.id, self.Trans.SymbolTable): + attr_name = type_arg.attr + ctype_cls = CTypeRegistry.GetClassByName(attr_name) + if ctype_cls is not None: + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + else: + assign_node = getattr(self.Trans, '_current_assign_node', None) + if assign_node: + ann = getattr(assign_node, 'annotation', None) + if ann: + ann_name = None + if isinstance(ann, ast.Name): + ann_name = ann.id + elif isinstance(ann, ast.Attribute): + ann_name = ann.attr + elif isinstance(ann, ast.BinOp) and isinstance(ann.op, ast.BitOr): + ann_name = ExtractTypeNameFromBinOp(ann) + if ann_name: + ctype_cls = CTypeRegistry.GetClassByName(ann_name) + if ctype_cls is not None: + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + resolved = Gen._type_str_to_llvm(llvm_str) + if resolved: + target_type = resolved + elif ann_name in Gen.structs: + target_type = ir.PointerType(Gen.structs[ann_name]) + va_arg_counter = getattr(Gen, '_va_arg_counter', 0) + Gen._va_arg_counter = va_arg_counter + 1 + result = Gen.emit_va_arg(va_list_ptr, target_type) + if result is not None and getattr(result, 'name', ''): + result.name = f"va_arg_result_{Gen._va_arg_counter}" + return result + + def _HandleVaArgLlvm(self, args: list[ast.AST]) -> ir.Constant: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + return ir.Constant(ir.IntType(32), 0) + + def _HandleCDerefAsLlvm(self, Node: ast.Call) -> ir.Constant: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if len(Node.args) < 2: + return ir.Constant(ir.IntType(32), 0) + ptr_val: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + val_val: ir.Value | None = self.HandleExprLlvm(Node.args[1]) + if ptr_val is None or val_val is None: + return ir.Constant(ir.IntType(32), 0) + if isinstance(ptr_val.type, ir.PointerType): + store_val: ir.Value = val_val + # 两端均为指向同一结构体类型的指针时执行结构体拷贝（*ptr = *val）， + # 而非存储指针值。这对 REnum 变体构造（c.DerefAs(ptr, LLVMType.Int(32))) 至关重要。 + if (isinstance(val_val.type, ir.PointerType) and + isinstance(ptr_val.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)) and + ptr_val.type.pointee == val_val.type.pointee): + loaded: ir.Value = Gen._load(val_val, name="deref_as_struct_load") + Gen._store(loaded, ptr_val) + return ir.Constant(ir.IntType(32), 0) + if isinstance(val_val.type, ir.PointerType): + void_pp: ir.PointerType = ir.PointerType(ir.PointerType(ir.IntType(8))) + if ptr_val.type != void_pp: + ptr_val = Gen.builder.bitcast(ptr_val, void_pp, name="deref_as_pp") + if val_val.type.pointee != ir.IntType(8): + store_val = Gen.builder.bitcast(val_val, ir.PointerType(ir.IntType(8)), name="deref_as_vp") + elif isinstance(val_val.type, ir.IntType) and isinstance(ptr_val.type.pointee, ir.IntType): + if val_val.type.width != ptr_val.type.pointee.width: + if ptr_val.type.pointee.width == 8 and val_val.type.width > 8: + target_ptr_type = ir.PointerType(val_val.type) + ptr_val = Gen.builder.bitcast(ptr_val, target_ptr_type, name="deref_as_cast") + elif val_val.type.width < ptr_val.type.pointee.width: + store_val = Gen.builder.zext(val_val, ptr_val.type.pointee, name="deref_as_zext") + else: + store_val = Gen.builder.trunc(val_val, ptr_val.type.pointee, name="deref_as_trunc") + elif isinstance(val_val.type, ir.IntType) and isinstance(ptr_val.type.pointee, ir.PointerType): + if val_val.type.width == 64: + store_val = Gen.builder.inttoptr(val_val, ptr_val.type.pointee, name="deref_as_inttoptr") + else: + ext = Gen.builder.zext(val_val, ir.IntType(64), name="deref_as_zext") + store_val = Gen.builder.inttoptr(ext, ptr_val.type.pointee, name="deref_as_inttoptr") + elif val_val.type != ptr_val.type.pointee: + target_ptr_type = ir.PointerType(val_val.type) + ptr_val = Gen.builder.bitcast(ptr_val, target_ptr_type, name="deref_as_cast") + Gen.builder.store(store_val, ptr_val) + return ir.Constant(ir.IntType(32), 0) + + def _InjectRequiresFields(self, result: ir.Value, ClassName: str, Gen: LlvmGeneratorMixin) -> None: + """在 __new__ 后、__init__ 前自动注入 __requires__/__require_must__ 字段。 + + - 编译期：对 __require_must__ 做纯静态可达性检查（遍历 _with_provides_stack） + - 运行时：对每个 requires 字段调用 _find_provider，非 NULL 则 bitcast 写入 + """ + requires_all: list[str] = [] + requires_all.extend(Gen.class_requires.get(ClassName, [])) + requires_all.extend(Gen.class_require_must.get(ClassName, [])) + if not requires_all: + return + # 编译期 __require_must__ 静态检查 + must_fields: list[str] = Gen.class_require_must.get(ClassName, []) + if must_fields: + # 跳过类自身方法内的检查（类的实现内部不需要 with 上下文） + current_class = getattr(Gen, '_current_class_name', None) + if current_class != ClassName: + available: set[str] = set() + for provides_list in Gen._with_provides_stack: + available.update(provides_list) + for field in must_fields: + if field not in available: + raise SyntaxError( + f"[TransPyC] Class '{ClassName}' __require_must__ field '{field}' " + f"not provided by any enclosing 'with' context" + ) + # 运行时注入 + find_func = Gen.functions.get('_find_provider') + if not find_func: + return + members = Gen.class_members.get(ClassName, []) + has_vtable = ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes + base = 1 if has_vtable else 0 + field_map: dict[str, tuple[int, ir.Type]] = {} + for i, (mname, mtype) in enumerate(members): + if mname in requires_all: + field_map[mname] = (base + i, mtype) + for field_name in requires_all: + if field_name not in field_map: + continue + field_idx, field_type = field_map[field_name] + field_str = Gen._create_string_global(field_name) + ptr_val = Gen.builder.call(find_func, [field_str], name=f"find_{field_name}") + null_ptr = ir.Constant(ptr_val.type, None) + is_null = Gen.builder.icmp_unsigned('==', ptr_val, null_ptr, name=f"is_null_{field_name}") + skip_bb = Gen.func.append_basic_block(f"skip_inject_{field_name}") + inject_bb = Gen.func.append_basic_block(f"inject_{field_name}") + Gen.builder.cbranch(is_null, skip_bb, inject_bb) + Gen.builder.position_at_end(inject_bb) + if isinstance(field_type, ir.PointerType): + casted = Gen.builder.bitcast(ptr_val, field_type, name=f"cast_{field_name}") + elif isinstance(field_type, ir.IntType): + casted = Gen.builder.ptrtoint(ptr_val, field_type, name=f"cast_{field_name}") + else: + casted = Gen.builder.bitcast(ptr_val, ir.PointerType(ir.IntType(8)), name=f"cast_{field_name}") + elem_ptr = Gen.builder.gep(result, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), field_idx)], name=f"field_{field_name}_ptr") + Gen._store(casted, elem_ptr) + Gen.builder.branch(skip_bb) + Gen.builder.position_at_end(skip_bb) + + def _HandleClassNewLlvm(self, Node: ast.Call, ClassName: str, module_sha1: str | None = None) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + # 跨模块类型引用：优先用 {module_sha1}.{ClassName} 查找 struct + _struct_full_key: str | None = f"{module_sha1}.{ClassName}" if module_sha1 else None + + def _get_struct_type(name: str) -> ir.Type | None: + """优先用 SHA1 前缀完整名查找，回退到短名""" + if _struct_full_key and _struct_full_key in Gen.structs: + return Gen.structs[_struct_full_key] + return Gen.structs.get(name) + + if hasattr(self.Trans.ClassHandler, '_generic_class_templates') and ClassName in self.Trans.ClassHandler._generic_class_templates: + return self._HandleGenericClassNewLlvm(Node, ClassName) + NewFuncName = f'{ClassName}.__before_init__' + InitFuncName = f'{ClassName}.__init__' + NewMethodFuncName = f'{ClassName}.__new__' + has_new_method = Gen._find_function(NewMethodFuncName) is not None + if Gen._has_function(NewFuncName): + NewFunc = Gen._get_function(NewFuncName) + StructType = _get_struct_type(ClassName) + if StructType: + result = Gen._allocaEntry(StructType, name=f"{ClassName}_alloca") + else: + return ir.Constant(ir.IntType(32), 0) + Gen.builder.call(NewFunc, [result], name=f"before_init_{ClassName}") + if has_new_method: + NewMethodFunc = Gen._find_function(NewMethodFuncName) + if NewMethodFunc: + NewMethodArgs = [result] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + NewMethodArgs.append(ArgVal) + self._fill_default_args(NewMethodArgs, NewMethodFunc, NewMethodFuncName) + self._ApplyKeywordArgs(Node, NewMethodArgs, NewMethodFunc, NewMethodFuncName, self_offset=1) + NewMethodArgs = Gen._apply_auto_addr(NewMethodFuncName, NewMethodArgs) + adjusted_new = Gen._adjust_args(NewMethodArgs, NewMethodFunc) + heap_ptr = Gen.builder.call(NewMethodFunc, adjusted_new, name=f"new_{ClassName}") + if heap_ptr: + target_ptr_type = ir.PointerType(StructType) if StructType else None + if isinstance(heap_ptr.type, ir.PointerType): + # Bitcast the returned pointer to the struct pointer type if needed + if target_ptr_type and heap_ptr.type != target_ptr_type: + result = Gen.builder.bitcast(heap_ptr, target_ptr_type, name=f"new_{ClassName}_cast") + else: + result = heap_ptr + elif isinstance(heap_ptr.type, ir.IntType) and target_ptr_type: + # __new__ returned an integer (pointer as int), convert back to pointer + if heap_ptr.type.width < 64: + i64_val = Gen.builder.zext(heap_ptr, ir.IntType(64), name=f"new_{ClassName}_i64") + result = Gen.builder.inttoptr(i64_val, target_ptr_type, name=f"new_{ClassName}_ptr") + else: + result = Gen.builder.inttoptr(heap_ptr, target_ptr_type, name=f"new_{ClassName}_ptr") + # __new__ 后：对堆指针重新调用 __before_init__ 初始化默认值（alloca 的默认值已随 alloca 丢弃） + if has_new_method: + Gen.builder.call(NewFunc, [result], name=f"before_init_{ClassName}_heap") + # __new__ 后、__init__ 前：自动注入 requires 字段 + self._InjectRequiresFields(result, ClassName, Gen) + if Gen._has_function(InitFuncName): + InitFunc = Gen._get_function(InitFuncName) + InitArgs = [result] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + InitArgs.append(ArgVal) + self._fill_default_args(InitArgs, InitFunc, InitFuncName) + self._ApplyKeywordArgs(Node, InitArgs, InitFunc, InitFuncName, self_offset=1) + self._append_eh_msg_out_arg(InitArgs, InitFunc, Gen) + InitArgs = Gen._apply_auto_addr(InitFuncName, InitArgs) + adjusted_init = Gen._adjust_args(InitArgs, InitFunc) + Gen.builder.call(InitFunc, adjusted_init, name=f"init_{ClassName}") + return result + if _struct_full_key and _struct_full_key in Gen.structs: + StructType = Gen.structs[_struct_full_key] + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + is_opaque = isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0) + if is_opaque: + self._ensure_struct_declared(ClassName, module_sha1=module_sha1) + StructType = Gen.structs.get(_struct_full_key, Gen.structs.get(ClassName, StructType)) + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + is_opaque = isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0) + if is_opaque: + return None + elif ClassName in Gen.structs: + StructType = Gen.structs[ClassName] + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + is_opaque = isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0) + if is_opaque: + self._ensure_struct_declared(ClassName, module_sha1=module_sha1) + StructType = Gen.structs.get(ClassName, StructType) + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + is_opaque = isinstance(StructType, ir.IdentifiedStructType) and (StructType.elements is None or len(StructType.elements) == 0) + if is_opaque: + return None + result = Gen._allocaEntry(StructType, name=f"{ClassName}_alloca") + # Call __new__ method if it exists (may replace self with heap pointer) + if has_new_method: + NewMethodFunc = Gen._find_function(NewMethodFuncName) + if NewMethodFunc: + NewMethodArgs = [result] + for arg in Node.args: + ArgVal = self.HandleExprLlvm(arg) + if ArgVal: + NewMethodArgs.append(ArgVal) + self._fill_default_args(NewMethodArgs, NewMethodFunc, NewMethodFuncName) + self._ApplyKeywordArgs(Node, NewMethodArgs, NewMethodFunc, NewMethodFuncName, self_offset=1) + NewMethodArgs = Gen._apply_auto_addr(NewMethodFuncName, NewMethodArgs) + adjusted_new = Gen._adjust_args(NewMethodArgs, NewMethodFunc) + heap_ptr = Gen.builder.call(NewMethodFunc, adjusted_new, name=f"new_{ClassName}") + if heap_ptr: + target_ptr_type = ir.PointerType(StructType) if StructType else None + if isinstance(heap_ptr.type, ir.PointerType): + # Bitcast the returned pointer to the struct pointer type if needed + if target_ptr_type and heap_ptr.type != target_ptr_type: + result = Gen.builder.bitcast(heap_ptr, target_ptr_type, name=f"new_{ClassName}_cast") + else: + result = heap_ptr + elif isinstance(heap_ptr.type, ir.IntType) and target_ptr_type: + # __new__ returned an integer (pointer as int), convert back to pointer + if heap_ptr.type.width < 64: + i64_val = Gen.builder.zext(heap_ptr, ir.IntType(64), name=f"new_{ClassName}_i64") + result = Gen.builder.inttoptr(i64_val, target_ptr_type, name=f"new_{ClassName}_ptr") + else: + result = Gen.builder.inttoptr(heap_ptr, target_ptr_type, name=f"new_{ClassName}_ptr") + # __new__ 后：对堆指针调用 __before_init__ 初始化默认值 + if has_new_method and Gen._has_function(NewFuncName): + NewFunc = Gen._get_function(NewFuncName) + Gen.builder.call(NewFunc, [result], name=f"before_init_{ClassName}_heap") + # __new__ 后、__init__ 前：自动注入 requires 字段 + self._InjectRequiresFields(result, ClassName, Gen) + InitFunc = Gen.functions.get(InitFuncName) + if not InitFunc: + g = Gen.module.globals.get(InitFuncName) + if g and isinstance(g, ir.Function): + Gen.functions[InitFuncName] = g + InitFunc = g + if InitFunc: + InitArgs = [result] + init_param_types = InitFunc.function_type.args if hasattr(InitFunc, 'function_type') else [] + for i, arg in enumerate(Node.args): + arg_var_type = init_param_types[i + 1] if i + 1 < len(init_param_types) else None + ArgVal = self.HandleExprLlvm(arg, VarType=arg_var_type) + if ArgVal: + InitArgs.append(ArgVal) + self._fill_default_args(InitArgs, InitFunc, InitFuncName) + self._ApplyKeywordArgs(Node, InitArgs, InitFunc, InitFuncName, self_offset=1) + self._append_eh_msg_out_arg(InitArgs, InitFunc, Gen) + InitArgs = Gen._apply_auto_addr(InitFuncName, InitArgs) + adjusted_init = Gen._adjust_args(InitArgs, InitFunc) + Gen.builder.call(InitFunc, adjusted_init, name=f"init_{ClassName}") + elif Node.args or Node.keywords: + self._InitStructDefaults(result, ClassName, StructType) + self._InitStructFromArgs(result, Node, ClassName, StructType) + else: + self._InitStructDefaults(result, ClassName, StructType) + return result + return ir.Constant(ir.IntType(32), 0) + + def _ZeroInitStruct(self, struct_ptr: ir.Value, StructType: ir.Type) -> None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if isinstance(StructType, ir.IdentifiedStructType) and StructType.elements: + for i, elem_type in enumerate(StructType.elements): + elem_ptr = Gen.builder.gep(struct_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), i)], name=f"zero_elem_{i}") + zero_val = self._GetZeroValue(elem_type) + if zero_val is not None: + Gen._store(zero_val, elem_ptr) + elif isinstance(StructType, ir.IntType): + Gen._store(ir.Constant(StructType, 0), struct_ptr) + elif isinstance(StructType, ir.ArrayType) and isinstance(StructType.pointee, ir.IntType): + zero_arr = ir.Constant(StructType, None) + Gen._store(zero_arr, struct_ptr) + + def _GetZeroValue(self, typ: ir.Type) -> ir.Constant | None: + if isinstance(typ, ir.IntType): + return ir.Constant(typ, 0) + elif isinstance(typ, ir.PointerType): + return ir.Constant(typ, None) + elif isinstance(typ, (ir.FloatType, ir.DoubleType)): + return ir.Constant(typ, 0.0) + elif isinstance(typ, ir.ArrayType): + return ir.Constant(typ, None) + return None + + def _InitStructDefaults(self, struct_ptr: ir.Value, ClassName: str, StructType: ir.Type) -> None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + defaults: dict = Gen.class_member_defaults.get(ClassName, {}) + members: list[tuple] = Gen.class_members.get(ClassName, []) + if not defaults or not members: + return + if not isinstance(StructType, ir.IdentifiedStructType): + return + has_vtable = ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes + base = 1 if has_vtable else 0 + for i, (member_name, member_type) in enumerate(members): + if member_name in defaults: + default_val = defaults[member_name] + if isinstance(default_val, ir.Constant): + elem_ptr = Gen.builder.gep(struct_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), base + i)], name=f"default_{ClassName}_{member_name}") + Gen._store(default_val, elem_ptr) + + def _InitStructFromArgs(self, struct_ptr: ir.Value, Node: ast.Call, ClassName: str, StructType: ir.Type) -> None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not isinstance(StructType, ir.IdentifiedStructType): + return + members = Gen.class_members.get(ClassName, []) + has_vtable = ClassName in Gen.class_vtable or ClassName in Gen._cross_module_vtable_classes + base = 1 if has_vtable else 0 + for i, arg in enumerate(Node.args): + if i >= len(members): + break + val = self.HandleExprLlvm(arg) + if not val: + continue + member_name, member_type = members[i] + elem_idx = base + i + if StructType.elements is None or elem_idx >= len(StructType.elements): + break + elem_type = StructType.elements[elem_idx] + if isinstance(elem_type, ir.IntType) and isinstance(val.type, ir.IntType): + if val.type.width < elem_type.width: + val = Gen.builder.zext(val, elem_type, name=f"zext_init_{ClassName}_{i}") + elif val.type.width > elem_type.width: + val = Gen.builder.trunc(val, elem_type, name=f"trunc_init_{ClassName}_{i}") + elif isinstance(elem_type, ir.PointerType) and isinstance(val.type, ir.PointerType): + if val.type.pointee != elem_type.pointee: + val = Gen.builder.bitcast(val, elem_type, name=f"bitcast_init_{ClassName}_{i}") + elem_ptr = Gen.builder.gep(struct_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), elem_idx)], name=f"init_elem_{ClassName}_{member_name}") + Gen._store(val, elem_ptr) + if Node.keywords: + member_map = {name: (base + i) for i, (name, _) in enumerate(members)} + for kw in Node.keywords: + if kw.arg in member_map: + val = self.HandleExprLlvm(kw.value) + if not val: + continue + elem_idx = member_map[kw.arg] + if StructType.elements is None or elem_idx >= len(StructType.elements): + continue + elem_type = StructType.elements[elem_idx] + if isinstance(elem_type, ir.IntType) and isinstance(val.type, ir.IntType): + if val.type.width < elem_type.width: + val = Gen.builder.zext(val, elem_type, name=f"zext_init_{ClassName}_{kw.arg}") + elif val.type.width > elem_type.width: + val = Gen.builder.trunc(val, elem_type, name=f"trunc_init_{ClassName}_{kw.arg}") + elif isinstance(elem_type, ir.PointerType) and isinstance(val.type, ir.PointerType): + if val.type.pointee != elem_type.pointee: + val = Gen.builder.bitcast(val, elem_type, name=f"bitcast_init_{ClassName}_{kw.arg}") + elem_ptr = Gen.builder.gep(struct_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), elem_idx)], name=f"init_elem_{ClassName}_{kw.arg}") + Gen._store(val, elem_ptr) + + def _HandleSubscriptPtrLlvm(self, Node: ast.Subscript) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + ClassName: str | None = self.Trans.ExprHandler._get_var_class(Node.value, Gen) + if ClassName and ClassName in Gen.structs and Gen._has_function(f'{ClassName}.__getitem__'): + obj_val = self.HandleExprLlvm(Node.value) + idx_val = self.HandleExprLlvm(Node.slice) + if obj_val and idx_val: + # 处理 obj_val 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + if isinstance(obj_val.type, ir.IntType) and obj_val.type.width == 8: + obj_val = Gen.builder.inttoptr(obj_val, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if self._is_char_pointer(obj_val): + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + if isinstance(idx_val.type, ir.IntType) and idx_val.type.width < 64: + idx_val = Gen.builder.zext(idx_val, ir.IntType(64), name="zext_idx") + result = Gen.builder.call(Gen._get_function(f'{ClassName}.__getitem__'), [obj_val, idx_val], name=f"call_{ClassName}.__getitem__") + return result + ValueVal = self.HandleExprLlvm(Node.value) + if not ValueVal: + return None + IndexVal = self.HandleExprLlvm(Node.slice) + if not IndexVal: + return None + if isinstance(ValueVal.type, ir.IntType): + var_ptr = self._get_int_ptr(Node.value) + if var_ptr: + i8_ptr = Gen.builder.bitcast(var_ptr, ir.PointerType(ir.IntType(8)), name="int_to_i8ptr") + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_int_index") + ptr = Gen.builder.gep(i8_ptr, [IndexVal], name="int_subscript_ptr") + return ptr + return None + if isinstance(ValueVal.type, ir.PointerType) and isinstance(ValueVal.type.pointee, ir.PointerType): + LoadedValueVal = Gen._load(ValueVal, name="Load_subscript") + if isinstance(LoadedValueVal.type, ir.PointerType): + ElemPtr = Gen.builder.gep(LoadedValueVal, [IndexVal], name="subscript") + return ElemPtr + if isinstance(ValueVal.type, ir.PointerType): + pointee = ValueVal.type.pointee + if isinstance(pointee, ir.ArrayType): + zero = ir.Constant(ir.IntType(32), 0) + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(ValueVal, [zero, IndexVal], name="subscript") + return Gen._load(ElemPtr, name="arr_subscript_val") + else: + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width < 64: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(64), name="zext_ptr_index") + ElemPtr = Gen.builder.gep(ValueVal, [IndexVal], name="subscript") + if isinstance(pointee, ir.IntType): + return Gen._load(ElemPtr, name="subscript_val") + return ElemPtr + elif isinstance(ValueVal.type, ir.ArrayType): + zero = ir.Constant(ir.IntType(32), 0) + if isinstance(Node.value, ast.Name): + VarName = Node.value.id + if VarName in Gen.variables and Gen.variables[VarName] is not None: + VarPtr = Gen.variables[VarName] + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, ir.ArrayType): + if isinstance(IndexVal.type, ir.IntType) and IndexVal.type.width != 32: + if IndexVal.type.width < 32: + IndexVal = Gen.builder.zext(IndexVal, ir.IntType(32), name="zext_arr_index") + else: + IndexVal = Gen.builder.trunc(IndexVal, ir.IntType(32), name="trunc_arr_index") + ElemPtr = Gen.builder.gep(VarPtr, [zero, IndexVal], name="subscript") + return ElemPtr + arr_alloc = Gen._allocaEntry(ValueVal.type, name="arr_subscript_tmp") + Gen._store(ValueVal, arr_alloc) + ElemPtr = Gen.builder.gep(arr_alloc, [zero, IndexVal], name="subscript") + return ElemPtr + return None + + def _HandleGenericCallLlvm(self, Node: ast.Call, FuncName: str) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + template: dict = self.Trans.FunctionHandler._generic_templates[FuncName] + type_param_names: list[str] = template['type_params'] + type_args: list[str] = [] + CallArgs: list[ir.Value] = [] + for i, arg_node in enumerate(Node.args): + arg_val = self.HandleExprLlvm(arg_node) + if arg_val: + CallArgs.append(arg_val) + type_str = self.Trans.FunctionHandler._infer_type_arg_from_value(arg_val, Gen) + type_args.append(type_str) + else: + type_args.append('int') + while len(type_args) < len(type_param_names): + type_args.append('int') + type_args = type_args[:len(type_param_names)] + spec_name = self.Trans.FunctionHandler._specialize_generic_function(FuncName, type_args, Gen) + if spec_name is None: + return None + mangled_spec = Gen._mangle_func_name(spec_name) + func = None + if mangled_spec in Gen.functions: + func = Gen.functions[mangled_spec] + elif spec_name in Gen.functions: + func = Gen.functions[spec_name] + else: + for fname, fobj in Gen.functions.items(): + if spec_name in fname or fname.endswith(spec_name): + func = fobj + break + if func is None: + return None + if not CallArgs: + CallArgs = [ir.Constant(ir.IntType(32), 0) for _ in func.args] + CallArgs = Gen._apply_auto_addr(spec_name, CallArgs) + adjusted = Gen._adjust_args(CallArgs, func) + return Gen.builder.call(func, adjusted, name=f"call_{spec_name}") + + def _HandleGenericClassNewLlvm(self, Node: ast.Call, ClassName: str, explicit_slice: ast.AST | None = None) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + template: dict = self.Trans.ClassHandler._generic_class_templates[ClassName] + type_param_names: list[str] = template['type_params'] + type_args: list[str] = [] + type_names: list[str] = [] + # 如果有显式类型参数 (如 list[int](mb))，直接从 slice 提取 + if explicit_slice is not None: + slice_nodes: list[ast.AST] = [] + if isinstance(explicit_slice, ast.Tuple): + slice_nodes = list(explicit_slice.elts) + else: + slice_nodes = [explicit_slice] + for sn in slice_nodes: + result = self._resolve_generic_slice_type(sn) + if result is not None: + type_args.append(result[0]) + type_names.append(result[1]) + # 补齐缺少的类型参数 + while len(type_args) < len(type_param_names): + type_args.append('int') + type_names.append('CInt') + spec_name = self.Trans.ClassHandler._specialize_generic_class(ClassName, type_args, Gen, type_names=type_names) + if spec_name is None: + return ir.Constant(ir.IntType(32), 0) + return self._HandleClassNewLlvm(Node, spec_name) + all_inferred: list[tuple[str, str]] = [] + for i, arg_node in enumerate(Node.args): + type_str = 'int' + type_name = 'CInt' + if isinstance(arg_node, ast.Constant): + if isinstance(arg_node.value, float): + type_str = 'double' + type_name = 'CDouble' + elif isinstance(arg_node.value, int): + type_str = 'int' + type_name = 'CInt' + elif isinstance(arg_node.value, bool): + type_str = 'i8' + type_name = 'CBool' + elif isinstance(arg_node, ast.Call): + if isinstance(arg_node.func, ast.Name): + func_id = arg_node.func.id + if func_id == 'float': + type_str = 'double' + type_name = 'CDouble' + elif func_id == 'int': + type_str = 'int' + type_name = 'CInt' + else: + float_types = {'CDouble', 'CFloat', 'CFloat64T', 'CFloat32T'} + int_types = {'CInt', 'CUInt', 'CInt32T', 'CUInt32T', 'CInt64T', 'CUInt64T', 'CInt16T', 'CUInt16T', 'CInt8T', 'CUInt8T', 'CChar', 'CShort', 'CLong'} + if func_id in float_types: + type_str = 'double' + type_name = func_id + elif func_id in int_types: + type_str = 'int' + type_name = func_id + elif isinstance(arg_node.func, ast.Attribute): + attr_name = arg_node.func.attr + float_types = {'CDouble', 'CFloat', 'CFloat64T', 'CFloat32T'} + int_types = {'CInt', 'CUInt', 'CInt32T', 'CUInt32T', 'CInt64T', 'CUInt64T', 'CInt16T', 'CUInt16T', 'CInt8T', 'CUInt8T', 'CChar', 'CShort', 'CLong'} + if attr_name in float_types: + type_str = 'double' + type_name = attr_name + elif attr_name in int_types: + type_str = 'int' + type_name = attr_name + elif isinstance(arg_node, ast.Name): + var_info = self.Trans.SymbolTable.lookup(arg_node.id) + if var_info and isinstance(var_info, dict): + var_type = var_info.get('type', '') + if var_type in ('double', 'float'): + type_str = 'double' + type_name = 'CDouble' + all_inferred.append((type_str, type_name)) + hint_types = [(ts, tn) for ts, tn in all_inferred if tn != 'CInt' or ts != 'int'] + for i in range(len(type_param_names)): + if i < len(hint_types): + type_args.append(hint_types[i][0]) + type_names.append(hint_types[i][1]) + elif i < len(all_inferred): + type_args.append(all_inferred[i][0]) + type_names.append(all_inferred[i][1]) + else: + type_args.append('int') + type_names.append('CInt') + spec_name = self.Trans.ClassHandler._specialize_generic_class(ClassName, type_args, Gen, type_names=type_names) + if spec_name is None: + return ir.Constant(ir.IntType(32), 0) + return self._HandleClassNewLlvm(Node, spec_name) + diff --git a/lib/core/Handles/HandlesExprFormat.py b/lib/core/Handles/HandlesExprFormat.py new file mode 100644 index 0000000..02693cf --- /dev/null +++ b/lib/core/Handles/HandlesExprFormat.py @@ -0,0 +1,271 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import re +import llvmlite.ir as ir + + +class ExprFormatHandle(BaseHandle): + def _HandleJoinedStrLlvm(self, Node: ast.JoinedStr) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if not Gen or not Gen.builder: + return ir.Constant(ir.IntType(8).as_pointer(), None) + format_str: str = "" + cast_args: list[Any] = [] + for part in Node.values: + if isinstance(part, ast.Constant) and isinstance(part.value, str): + format_str += part.value.replace('\\', '\\\\').replace('%', '%%').replace('\n', '\\n').replace('\t', '\\t').replace('\0', '\\0') + elif isinstance(part, ast.FormattedValue): + fmt_class: str | None = self.Trans.ExprHandler._get_var_class(part.value, Gen) + if fmt_class and Gen._has_function(f'{fmt_class}.__str__'): + obj_val: Any = self.HandleExprLlvm(part.value) + if obj_val: + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + if fmt_class in Gen.structs: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[fmt_class]), name=f"cast_{fmt_class}") + val: Any = Gen.builder.call(Gen._get_function(f'{fmt_class}.__str__'), [obj_val], name=f"call_{fmt_class}.__str__") + Gen._RegisterTempPtr(val) + format_str += "%s" + cast_args.append(val) + else: + format_str += "%s" + cast_args.append(ir.Constant(ir.PointerType(ir.IntType(8)), None)) + continue + val = self.HandleExprLlvm(part.value) + if not val: + format_str += "%d" + cast_args.append(ir.Constant(ir.IntType(32), 0)) + continue + # 检查 format_spec（Python f-string 格式说明符） + if part.format_spec is not None: + c_fmt: str | None = self._parse_format_spec(part.format_spec, val) + if c_fmt is not None: + cast_val: Any = self._cast_value_for_format(val, c_fmt, Gen, part.value) + format_str += c_fmt + cast_args.append(cast_val) + continue + if isinstance(val.type, ir.PointerType): + pointee: ir.Type = val.type.pointee + if isinstance(pointee, ir.IntType) and pointee.width == 8: + format_str += "%s" + cast_args.append(val) + elif isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + int_val: Any = Gen.builder.ptrtoint(val, ir.IntType(64), name="fstr_ptr2int") + trunc_val: Any = Gen.builder.trunc(int_val, ir.IntType(32), name="fstr_trunc") + format_str += "%d" + cast_args.append(trunc_val) + else: + Loaded: Any = Gen._load(val, name="fstr_deref") + if isinstance(Loaded.type, ir.IntType): + if Loaded.type.width == 8: + format_str += "%c" + ext: Any = Gen.builder.zext(Loaded, ir.IntType(32), name="fstr_char2int") + cast_args.append(ext) + else: + format_str += "%d" + cast_args.append(Loaded) + elif isinstance(Loaded.type, (ir.FloatType, ir.DoubleType)): + format_str += "%f" + cast_args.append(Loaded) + elif isinstance(Loaded.type, ir.PointerType): + format_str += "%s" + cast_args.append(Loaded) + else: + format_str += "%d" + cast_args.append(Loaded) + elif isinstance(val.type, ir.IntType): + if val.type.width == 8: + format_str += "%c" + ext = Gen.builder.zext(val, ir.IntType(32), name="fstr_char2int") + cast_args.append(ext) + elif val.type.width == 1: + zext: Any = Gen.builder.zext(val, ir.IntType(32), name="fstr_bool2int") + format_str += "%d" + cast_args.append(zext) + else: + is_u: bool = Gen._check_node_unsigned(part.value) + format_str += "%u" if is_u else "%d" + cast_args.append(val) + elif isinstance(val.type, (ir.FloatType, ir.DoubleType)): + format_str += "%f" + cast_args.append(val) + else: + format_str += "%d" + cast_args.append(val) + if not format_str: + return Gen.emit_constant("", 'string') + format_ptr: Any = Gen._create_string_global(format_str) + # 统一使用 snprintf 生成字符串，返回 i8* + # __str__ 方法内使用堆分配（返回值需持久化），其他场景使用栈分配（自动释放） + in_str_method: bool = bool(getattr(Gen, 'func', None) and getattr(Gen.func, 'name', '').endswith('.__str__')) + snprintf_type: ir.FunctionType = ir.FunctionType(ir.IntType(32), [ + ir.PointerType(ir.IntType(8)), + ir.IntType(64), + ir.PointerType(ir.IntType(8)) + ], var_arg=True) + snprintf_func: Any = Gen.get_or_declare_c_func('snprintf', snprintf_type) + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + # 第一遍：获取所需长度（NULL 缓冲，size=0） + len_val: ir.Value = Gen.builder.call(snprintf_func, [null_ptr, ir.Constant(ir.IntType(64), 0), format_ptr] + cast_args, name="fstr_len") + # 加 1 用于 null 终止符 + size_val: ir.Value = Gen.builder.add(len_val, ir.Constant(ir.IntType(32), 1), name="fstr_size") + size_ext: ir.Value = Gen.builder.zext(size_val, ir.IntType(64), name="fstr_size_ext") + if in_str_method: + # 堆分配（__str__ 返回值需跨函数生命周期） + malloc_type: ir.FunctionType = ir.FunctionType(ir.PointerType(ir.IntType(8)), [ir.IntType(64)]) + malloc_func: Any = Gen._get_or_declare_func('malloc', malloc_type) + buf_ptr: ir.Value = Gen.builder.call(malloc_func, [size_ext], name="fstr_heap_buf") + Gen.builder.call(snprintf_func, [buf_ptr, size_ext, format_ptr] + cast_args, name="fstr_snprintf") + Gen._RegisterTempPtr(buf_ptr) + return buf_ptr + # 栈分配（局部使用，自动释放） + buf_ptr: ir.Value = Gen._alloca(ir.IntType(8), name="fstr_stack_buf", size=size_val) + Gen.builder.call(snprintf_func, [buf_ptr, size_ext, format_ptr] + cast_args, name="fstr_snprintf") + return buf_ptr + + def _parse_format_spec(self, format_spec_node: ast.AST, val: ir.Value) -> str | None: + """将 Python f-string format_spec 转换为 C printf 格式说明符。 + + 支持: d/x/X/o/c/f/s 及 [[fill]align][sign][#][0][width][.precision][type] 语法。 + 返回 None 表示无法解析，调用方应回退到类型推断。 + """ + if isinstance(format_spec_node, ast.JoinedStr): + parts: list[str] = [] + for v in format_spec_node.values: + if isinstance(v, ast.Constant) and isinstance(v.value, str): + parts.append(v.value) + else: + return None + spec_str: str = ''.join(parts) + elif isinstance(format_spec_node, ast.Constant) and isinstance(format_spec_node.value, str): + spec_str = format_spec_node.value + else: + return None + + if not spec_str: + return None + + m: Any = re.match(r'^(.?[<>=^])?([+\- ])?(#)?(0)?(\d+)?(,)?(\.\d+)?([bcdeEfFgGnosxX%])?$', spec_str) + if not m: + return None + + fill_align, sign, alt, zero, width, comma, precision, fmt_type = m.groups() + + c_flags: str = '' + if sign == '+': + c_flags += '+' + elif sign == ' ': + c_flags += ' ' + if alt: + c_flags += '#' + + if fill_align: + align_char: str = fill_align[-1] if len(fill_align) > 1 else fill_align[0] + if align_char in ('<', '^'): + c_flags += '-' + if zero: + c_flags += '0' + + c_width: str = width or '' + c_precision: str = precision or '' + + length_mod: str = '' + if isinstance(val.type, ir.IntType): + if val.type.width == 64: + length_mod = 'll' + elif val.type.width == 32: + length_mod = 'l' + elif val.type.width == 16: + length_mod = 'h' + elif val.type.width == 8: + length_mod = 'hh' + + c_specifier: str = '' + if fmt_type in ('d', 'n'): + c_specifier = f'{length_mod}d' + elif fmt_type == 'x': + c_specifier = f'{length_mod}x' + elif fmt_type == 'X': + c_specifier = f'{length_mod}X' + elif fmt_type == 'o': + c_specifier = f'{length_mod}o' + elif fmt_type == 'b': + return None + elif fmt_type == 'c': + c_specifier = 'c' + elif fmt_type in ('f', 'F'): + c_specifier = 'f' + elif fmt_type in ('e', 'E', 'g', 'G'): + c_specifier = fmt_type + elif fmt_type == 's': + c_specifier = 's' + elif fmt_type == '%': + c_specifier = 'f%%' + else: + if isinstance(val.type, ir.IntType): + if val.type.width == 8: + c_specifier = 'c' + elif val.type.width == 1: + c_specifier = 'd' + else: + c_specifier = f'{length_mod}d' + elif isinstance(val.type, (ir.FloatType, ir.DoubleType)): + c_specifier = 'f' + elif isinstance(val.type, ir.PointerType): + c_specifier = 's' + else: + c_specifier = 'd' + + return f'%{c_flags}{c_width}{c_precision}{c_specifier}' + + def _cast_value_for_format(self, val: ir.Value, c_fmt: str, Gen: Any, arg_node: ast.AST) -> ir.Value: + """根据 C printf 格式说明符转换值类型。""" + fmt_type_char: str = '' + for ch in reversed(c_fmt): + if ch.isalpha() and ch not in ('l', 'h'): + fmt_type_char = ch + break + elif ch == '%': + break + + if fmt_type_char == 's': + if isinstance(val.type, ir.PointerType): + pointee: ir.Type = val.type.pointee + if isinstance(pointee, ir.IntType) and pointee.width == 8: + return val + if isinstance(pointee, ir.ArrayType) and isinstance(pointee.element, ir.IntType) and pointee.element.width == 8: + return Gen.builder.gep(val, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)], name="fstr_arr2ptr") + return val + elif fmt_type_char == 'c': + if isinstance(val.type, ir.IntType) and val.type.width < 32: + return Gen.builder.zext(val, ir.IntType(32), name="fstr_zext_c") + return val + elif fmt_type_char in ('d', 'x', 'X', 'o', 'i'): + if isinstance(val.type, ir.IntType): + if val.type.width == 8: + is_u: bool = Gen._check_node_unsigned(arg_node) + if is_u: + return Gen.builder.zext(val, ir.IntType(32), name="fstr_zext_i8") + return Gen.builder.sext(val, ir.IntType(32), name="fstr_sext_i8") + elif val.type.width == 16: + is_u = Gen._check_node_unsigned(arg_node) + if is_u: + return Gen.builder.zext(val, ir.IntType(32), name="fstr_zext_i16") + return Gen.builder.sext(val, ir.IntType(32), name="fstr_sext_i16") + elif val.type.width == 1: + return Gen.builder.zext(val, ir.IntType(32), name="fstr_zext_bool") + return val + elif isinstance(val.type, (ir.FloatType, ir.DoubleType)): + return Gen.builder.fptosi(val, ir.IntType(32), name="fstr_f2i") + return val + elif fmt_type_char in ('f', 'e', 'E', 'g', 'G'): + if isinstance(val.type, ir.FloatType): + return Gen.builder.fpext(val, ir.DoubleType(), name="fstr_fpext") + elif isinstance(val.type, ir.IntType): + return Gen.builder.sitofp(val, ir.DoubleType(), name="fstr_i2f") + return val + return val \ No newline at end of file diff --git a/lib/core/Handles/HandlesExprLambda.py b/lib/core/Handles/HandlesExprLambda.py new file mode 100644 index 0000000..7003785 --- /dev/null +++ b/lib/core/Handles/HandlesExprLambda.py @@ -0,0 +1,249 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class ExprLambdaHandle(BaseHandle): + def _HandleLambdaLlvm(self, Node: ast.Lambda) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if not getattr(Gen, '_lambda_counter', None): + Gen._lambda_counter = 0 + lambda_id: int = Gen._lambda_counter + Gen._lambda_counter += 1 + fn_name: str = f"lambda_fn_{lambda_id}" + captured_vars: list[tuple[str, Any]] = self._get_lambda_captured_vars(Node) + env_types: list[ir.Type] = [] + env_var_names: list[str] = [] + for var_name, var_val in captured_vars: + env_types.append(var_val.type) + env_var_names.append(var_name) + env_struct_type: ir.LiteralStructType = ir.LiteralStructType(env_types) if env_types else ir.LiteralStructType([]) + ret_type: ir.Type = self.Trans.ExprUtils._infer_expr_llvm_type_full(Node.body) + param_types: list[ir.Type] = [ir.PointerType(env_struct_type)] + if Node.args: + for arg in Node.args.args: + param_types.append(ir.IntType(32)) + fn_type: ir.FunctionType = ir.FunctionType(ret_type, param_types) + fn: ir.Function = ir.Function(Gen.module, fn_type, name=fn_name) + Gen.functions[fn_name] = fn + entry_block: ir.Block = fn.append_basic_block(name=f"{fn_name}_entry") + prev_builder: Any = Gen.builder + prev_func: Any = Gen.func + prev_vars: dict[str, Any] = dict(Gen.variables) + Gen.builder = ir.IRBuilder(entry_block) + Gen.func = fn + Gen.variables = {} + for i, arg in enumerate(fn.args): + arg.name = f"arg_{i}" if i > 0 else "env" + if env_types: + for idx, var_name in enumerate(env_var_names): + env_ptr: Any = fn.args[0] + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + member_ptr: Any = Gen.builder.gep(env_ptr, [zero, ir.Constant(ir.IntType(32), idx)], name=f"env_{var_name}") + Loaded: Any = Gen._load(member_ptr, name=f"Load_env_{var_name}") + temp_alloca: Any = Gen._alloca(Loaded.type, name=f"temp_{var_name}") + Gen._store(Loaded, temp_alloca) + Gen.variables[var_name] = temp_alloca + if Node.args: + for i, arg in enumerate(Node.args.args): + Gen.variables[arg.arg] = Gen._alloca(ir.IntType(32), name=arg.arg) + Gen._store(fn.args[i + 1], Gen.variables[arg.arg]) + body_val: Any = self.HandleExprLlvm(Node.body) + if body_val: + if isinstance(body_val.type, ir.PointerType) and not isinstance(ret_type, ir.PointerType): + body_val = Gen._load(body_val, name="Load_lambda_ret") + if body_val.type != ret_type: + if isinstance(body_val.type, ir.IntType) and isinstance(ret_type, ir.IntType): + if body_val.type.width < ret_type.width: + body_val = Gen.builder.zext(body_val, ret_type, name="zext_lambda_ret") + elif body_val.type.width > ret_type.width: + body_val = Gen.builder.trunc(body_val, ret_type, name="trunc_lambda_ret") + elif isinstance(body_val.type, ir.PointerType) and isinstance(ret_type, ir.PointerType): + body_val = Gen.builder.bitcast(body_val, ret_type, name="bitcast_lambda_ret") + Gen.builder.ret(body_val) + else: + if isinstance(ret_type, ir.PointerType): + Gen.builder.ret(ir.Constant(ret_type, None)) + elif isinstance(ret_type, ir.IntType): + Gen.builder.ret(ir.Constant(ret_type, 0)) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ret_type, 0.0)) + elif isinstance(ret_type, ir.BaseStructType): + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ + ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) + else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) + else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + Gen.builder.ret(zero_val) + else: + Gen.builder.ret(ir.Constant(ret_type, 0)) + Gen.builder = prev_builder + Gen.func = prev_func + Gen.variables = prev_vars + closure_struct_type: ir.LiteralStructType = ir.LiteralStructType([ + ir.PointerType(ir.IntType(8)), + ir.PointerType(fn_type) + ]) + closure_ptr: Any = Gen._alloca(closure_struct_type, name=f"closure_{lambda_id}") + env_ptr: Any = Gen._alloca(env_struct_type, name=f"env_{lambda_id}") + if env_types: + for idx, (var_name, var_val) in enumerate(captured_vars): + zero = ir.Constant(ir.IntType(32), 0) + member_ptr = Gen.builder.gep(env_ptr, [zero, ir.Constant(ir.IntType(32), idx)], name=f"store_env_{var_name}") + Gen._store(var_val, member_ptr) + zero = ir.Constant(ir.IntType(32), 0) + env_field_ptr: Any = Gen.builder.gep(closure_ptr, [zero, zero], name="closure_env_ptr") + env_i8_ptr: Any = Gen.builder.bitcast(env_ptr, ir.PointerType(ir.IntType(8)), name="env_i8_ptr") + Gen._store(env_i8_ptr, env_field_ptr) + fn_field_ptr: Any = Gen.builder.gep(closure_ptr, [zero, ir.Constant(ir.IntType(32), 1)], name="closure_fn_ptr") + Gen._store(fn, fn_field_ptr) + return closure_ptr + + def _get_lambda_captured_vars(self, Node: ast.Lambda) -> list[tuple[str, ir.Value]]: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + captured: list[tuple[str, Any]] = [] + free_vars: list[str] = self._get_lambda_free_vars(Node) + for var_name in free_vars: + if var_name in Gen.variables: + var_ptr: Any = Gen.variables[var_name] + if isinstance(var_ptr, ir.AllocaInstr) or isinstance(var_ptr, ir.GlobalVariable): + Loaded: Any = Gen._load(var_ptr, name=f"capture_{var_name}") + captured.append((var_name, Loaded)) + return captured + + def _get_lambda_free_vars(self, Node: ast.Lambda) -> list[str]: + bound_vars: set[str] = set() + free_vars: set[str] = set() + # 收集 lambda body 中的 bound 和 free 变量 + self.Trans.ExprUtils._collect_names(Node.body, bound_vars, free_vars) + # lambda 参数也是 bound 的 + if Node.args: + for arg in Node.args.args: + bound_vars.add(arg.arg) + # free 变量 = 在 body 中引用(Load)但不是参数也不是 body 内赋值的变量 + result: list[str] = list(free_vars - bound_vars) + return result + + def _HandleIfExpLlvm(self, Node: ast.IfExp) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + TestVal: Any = self.HandleExprLlvm(Node.test) + if not TestVal: + return None + if not isinstance(TestVal.type, ir.IntType) or TestVal.type.width != 1: + TestVal = Gen.builder.icmp_signed('!=', TestVal, Gen._ZeroConst(TestVal.type), name="ifexp_cond") + ThenBB: ir.Block = Gen.func.append_basic_block(name="ifexp.then") + ElseBB: ir.Block = Gen.func.append_basic_block(name="ifexp.else") + MergeBB: ir.Block = Gen.func.append_basic_block(name="ifexp.end") + Gen.builder.cbranch(TestVal, ThenBB, ElseBB) + # 先在 ThenBB 中生成 BodyVal + Gen.builder.position_at_start(ThenBB) + BodyVal: Any = self.HandleExprLlvm(Node.body) + if not BodyVal: + BodyVal = ir.Constant(ir.IntType(32), 0) + ThenEndBB: ir.Block = Gen.builder.block + # 在 ElseBB 中生成 OrelseVal + Gen.builder.position_at_start(ElseBB) + OrelseVal: Any = self.HandleExprLlvm(Node.orelse) + if not OrelseVal: + OrelseVal = ir.Constant(ir.IntType(32), 0) + ElseEndBB: ir.Block = Gen.builder.block + # 确定 phi 节点的目标类型 + result_type: ir.Type = BodyVal.type + if BodyVal.type != OrelseVal.type: + if isinstance(BodyVal.type, ir.IntType) and isinstance(OrelseVal.type, ir.IntType): + result_type = BodyVal.type if BodyVal.type.width >= OrelseVal.type.width else OrelseVal.type + elif isinstance(BodyVal.type, ir.PointerType) and isinstance(OrelseVal.type, ir.PointerType): + # 指针类型，使用 BodyVal 的类型 + result_type = BodyVal.type + elif isinstance(BodyVal.type, ir.IntType) and isinstance(OrelseVal.type, ir.PointerType): + # BodyVal 是整数，OrelseVal 是指针（字符串字面量） + # 如果指针指向 i8，将指针转换为 i8（加载字符） + if isinstance(OrelseVal.type.pointee, ir.IntType) and OrelseVal.type.pointee.width == 8: + result_type = ir.IntType(8) + else: + result_type = OrelseVal.type + elif isinstance(BodyVal.type, ir.PointerType) and isinstance(OrelseVal.type, ir.IntType): + # BodyVal 是指针（字符串字面量），OrelseVal 是整数 + # 如果指针指向 i8，将指针转换为 i8（加载字符） + if isinstance(BodyVal.type.pointee, ir.IntType) and BodyVal.type.pointee.width == 8: + result_type = ir.IntType(8) + else: + result_type = BodyVal.type + else: + result_type = OrelseVal.type + # 在 ThenBB 末尾添加类型转换（如果需要）和分支 + if not ThenEndBB.is_terminated: + Gen.builder.position_at_end(ThenEndBB) + if BodyVal.type != result_type: + if isinstance(BodyVal.type, ir.IntType) and isinstance(result_type, ir.IntType): + if BodyVal.type.width < result_type.width: + BodyVal = Gen.builder.zext(BodyVal, result_type, name="ifexp_body_zext") + else: + BodyVal = Gen.builder.trunc(BodyVal, result_type, name="ifexp_body_trunc") + elif isinstance(BodyVal.type, ir.IntType) and isinstance(result_type, ir.PointerType): + # 如果 result_type 是 i8*，说明 Else 分支是字符串字面量 + # 不应该将整数转换为指针，而应该保持整数类型 + # 这里将 result_type 改为 i8 + result_type = ir.IntType(8) + # 重新处理 Else 分支的类型转换 + elif isinstance(BodyVal.type, ir.PointerType) and isinstance(result_type, ir.IntType): + # 指针转整数 + BodyVal = Gen.builder.ptrtoint(BodyVal, result_type, name="ifexp_body_ptr2int") + Gen.builder.branch(MergeBB) + # 在 ElseBB 末尾添加类型转换（如果需要）和分支 + if not ElseEndBB.is_terminated: + Gen.builder.position_at_end(ElseEndBB) + if OrelseVal.type != result_type: + if isinstance(OrelseVal.type, ir.IntType) and isinstance(result_type, ir.IntType): + if OrelseVal.type.width < result_type.width: + OrelseVal = Gen.builder.zext(OrelseVal, result_type, name="ifexp_orelse_zext") + else: + OrelseVal = Gen.builder.trunc(OrelseVal, result_type, name="ifexp_orelse_trunc") + elif isinstance(OrelseVal.type, ir.IntType) and isinstance(result_type, ir.PointerType): + # 如果 result_type 是 i8*，说明 BodyVal 分支是字符串字面量 + # 不应该将整数转换为指针，而应该保持整数类型 + # 这里将 result_type 改为 i8 + result_type = ir.IntType(8) + # 重新处理 BodyVal 分支的类型转换（已经在上面处理过了） + elif isinstance(OrelseVal.type, ir.PointerType) and isinstance(result_type, ir.IntType): + # 如果指针指向 i8，加载字符值 + if isinstance(OrelseVal.type.pointee, ir.IntType) and OrelseVal.type.pointee.width == 8: + OrelseVal = Gen._load(OrelseVal, name="ifexp_orelse_Load_char") + else: + OrelseVal = Gen.builder.ptrtoint(OrelseVal, result_type, name="ifexp_orelse_ptr2int") + Gen.builder.branch(MergeBB) + Gen.builder.position_at_start(MergeBB) + result_phi: Any = Gen.builder.phi(result_type, name="ifexp.result") + result_phi.add_incoming(BodyVal, ThenEndBB) + result_phi.add_incoming(OrelseVal, ElseEndBB) + return result_phi + + def _HandleNamedExprLlvm(self, Node: ast.NamedExpr) -> ir.Value: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + ValueVal: Any = self.HandleExprLlvm(Node.value) + if not ValueVal: + return None + TargetName: str = Node.target.id + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + if TargetName in Gen.variables and Gen.variables[TargetName] is not None: + VarPtr: Any = Gen.variables[TargetName] + if VarPtr.type.pointee == ValueVal.type: + Gen._store(ValueVal, VarPtr) + else: + NewVar: Any = Gen._alloca(ValueVal.type, name=TargetName) + Gen._store(ValueVal, NewVar) + Gen.variables[TargetName] = NewVar + else: + NewVar = Gen._alloca(ValueVal.type, name=TargetName) + Gen._store(ValueVal, NewVar) + Gen.variables[TargetName] = NewVar + Gen._reg_values[TargetName] = ValueVal + return ValueVal \ No newline at end of file diff --git a/lib/core/Handles/HandlesExprOps.py b/lib/core/Handles/HandlesExprOps.py new file mode 100644 index 0000000..c3756da --- /dev/null +++ b/lib/core/Handles/HandlesExprOps.py @@ -0,0 +1,453 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class ExprOpsHandle(BaseHandle): + def _HandleBinOpLlvm(self, Node: ast.BinOp, VarType: ir.Type | str | None = None) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + Gen._set_node_info(Node, "BinOp") + LeftVal: Any = self.HandleExprLlvm(Node.left, VarType) + if not LeftVal: + return None + RightVal: Any = self.HandleExprLlvm(Node.right, VarType) + if not RightVal: + return None + Op: str | None = self.Trans.ExprHandler.GetOpSymbol(Node.op) + if not Op: + return None + OP_OVERLOAD_MAP: dict[str, str] = { + '+': '__add__', '-': '__sub__', '*': '__mul__', + '/': '__truediv__', '//': '__floordiv__', '%': '__mod__', + '**': '__pow__', + } + if Op in OP_OVERLOAD_MAP: + LeftClassName: str | None = None + if isinstance(LeftVal.type, ir.PointerType): + pointee: ir.Type = LeftVal.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + LeftClassName = CN + break + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + LeftClassName = self.Trans.ExprUtils._get_var_class(Node.left, Gen) + if LeftClassName and LeftClassName in Gen.structs: + TargetStructPtr: ir.PointerType = ir.PointerType(Gen.structs[LeftClassName]) + LeftVal = Gen.builder.bitcast(LeftVal, TargetStructPtr, name=f"bitcast_to_{LeftClassName}") + if LeftClassName is None and isinstance(Node.left, ast.Name): + LeftClassName = Gen.var_struct_class.get(Node.left.id) + if LeftClassName and LeftClassName in Gen.structs: + if isinstance(LeftVal.type, ir.PointerType) and isinstance(LeftVal.type.pointee, ir.IntType) and LeftVal.type.pointee.width == 8: + TargetStructPtr = ir.PointerType(Gen.structs[LeftClassName]) + LeftVal = Gen.builder.bitcast(LeftVal, TargetStructPtr, name=f"bitcast_to_{LeftClassName}") + if LeftClassName: + op_name: str = OP_OVERLOAD_MAP[Op] + result: Any = self.Trans.ExprUtils._try_operator_overLoad(LeftClassName, op_name, LeftVal, Gen, RightVal) + if result is not None: + return result + if Op in ('+', 'Add') and isinstance(LeftVal.type, ir.PointerType) and isinstance(RightVal.type, ir.IntType): + if isinstance(LeftVal.type.pointee, ir.IntType) and LeftVal.type.pointee.width == 8: + if isinstance(Node.left, ast.Constant) and isinstance(Node.left.value, str) and len(Node.left.value) == 1: + CharVal: Any = Gen._load(LeftVal, name="Load_char_const") + CharAsInt: Any = Gen.builder.zext(CharVal, RightVal.type, name="char_to_int") + result = Gen.builder.add(CharAsInt, RightVal, name="char_add") + TruncResult: Any = Gen.builder.trunc(result, ir.IntType(8), name="add_to_char") + return TruncResult + is_unsigned: bool = Gen._check_node_unsigned(Node.left) or Gen._check_node_unsigned(Node.right) + if isinstance(LeftVal.type, ir.IntType) and isinstance(RightVal.type, ir.IntType): + if LeftVal.type.width < RightVal.type.width: + need_zext: bool = is_unsigned or Gen._check_node_unsigned(Node.left) or LeftVal.type.width == 8 or LeftVal.type.width == 1 + if not need_zext and isinstance(RightVal, ir.Constant): + signed_max: int = (1 << (LeftVal.type.width - 1)) - 1 + try: + if RightVal.constant > signed_max: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + LeftVal = Gen.builder.zext(LeftVal, RightVal.type, name="zext_left") + else: + LeftVal = Gen.builder.sext(LeftVal, RightVal.type, name="sext_left") + elif LeftVal.type.width > RightVal.type.width: + need_zext = is_unsigned or Gen._check_node_unsigned(Node.right) or RightVal.type.width == 8 or RightVal.type.width == 1 + if not need_zext and isinstance(LeftVal, ir.Constant): + signed_max = (1 << (RightVal.type.width - 1)) - 1 + try: + if LeftVal.constant > signed_max: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + RightVal = Gen.builder.zext(RightVal, LeftVal.type, name="zext_right") + else: + RightVal = Gen.builder.sext(RightVal, LeftVal.type, name="sext_right") + elif isinstance(LeftVal.type, ir.PointerType) and isinstance(RightVal.type, ir.IntType): + if Op not in ('+', '-', 'Add', 'Sub') and isinstance(LeftVal.type.pointee, ir.IntType): + LeftVal = Gen._load(LeftVal, name="deref_ptr_binop") + if LeftVal.type.width < RightVal.type.width: + LeftVal = Gen.builder.zext(LeftVal, RightVal.type, name="zext_deref_left") + elif LeftVal.type.width > RightVal.type.width: + RightVal = Gen.builder.zext(RightVal, LeftVal.type, name="zext_right_deref") + elif isinstance(RightVal.type, ir.PointerType) and isinstance(LeftVal.type, ir.IntType): + if Op not in ('+', '-', 'Add', 'Sub') and isinstance(RightVal.type.pointee, ir.IntType): + RightVal = Gen._load(RightVal, name="deref_ptr_binop_r") + if RightVal.type.width < LeftVal.type.width: + RightVal = Gen.builder.zext(RightVal, LeftVal.type, name="zext_deref_right") + elif RightVal.type.width > LeftVal.type.width: + LeftVal = Gen.builder.zext(LeftVal, RightVal.type, name="zext_left_deref") + if Op in ('>>', 'RShift') and isinstance(Node.left, ast.Name): + vname: str = Node.left.id + if vname in Gen.var_signedness and Gen.var_signedness[vname]: + is_unsigned = True + result = Gen.emit_binary_op(Op, LeftVal, RightVal, is_unsigned=is_unsigned) + return result + + def _HandleBoolOpLlvm(self, Node: ast.BoolOp) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + + def _to_bool(val: Any, name: str) -> Any: + if isinstance(val.type, (ir.FloatType, ir.DoubleType)): + return Gen.builder.fcmp_ordered('!=', val, ir.Constant(val.type, 0.0), name=name) + if not isinstance(val.type, ir.IntType) or val.type.width != 1: + return Gen.builder.icmp_signed('!=', val, Gen._ZeroConst(val.type), name=name) + return val + + values: list[ast.expr] = Node.values + if len(values) < 2: + return self.HandleExprLlvm(values[0]) if values else None + if isinstance(Node.op, ast.And): + phi_incomings: list[tuple[ir.Constant, ir.Block]] = [] + end_bb: ir.Block = Gen.func.append_basic_block(name="and.end") + for idx in range(len(values) - 1): + val: Any = self.HandleExprLlvm(values[idx]) + if not val: + if not end_bb.is_terminated: + saved_block: ir.Block = Gen.builder.block + Gen.builder.position_at_start(end_bb) + Gen.builder.unreachable() + Gen.builder.position_at_end(saved_block) + return None + if not isinstance(val.type, ir.IntType) or val.type.width != 1: + val = _to_bool(val, f"andcond{idx}") + cur_bb: ir.Block = Gen.builder.block + next_bb: ir.Block = Gen.func.append_basic_block(name=f"and.rhs{idx}") + Gen.builder.cbranch(val, next_bb, end_bb) + phi_incomings.append((ir.Constant(ir.IntType(1), 0), cur_bb)) + Gen.builder.position_at_start(next_bb) + last_val: Any = self.HandleExprLlvm(values[-1]) + if not last_val: + last_val = ir.Constant(ir.IntType(1), 0) + if not isinstance(last_val.type, ir.IntType) or last_val.type.width != 1: + last_val = _to_bool(last_val, "andcond_last") + last_bb: ir.Block = Gen.builder.block + Gen.builder.branch(end_bb) + Gen.builder.position_at_start(end_bb) + result_phi: Any = Gen.builder.phi(ir.IntType(1), name="and.result") + for const_val, bb in phi_incomings: + result_phi.add_incoming(const_val, bb) + result_phi.add_incoming(last_val, last_bb) + return result_phi + elif isinstance(Node.op, ast.Or): + phi_incomings = [] + end_bb = Gen.func.append_basic_block(name="or.end") + for idx in range(len(values) - 1): + val = self.HandleExprLlvm(values[idx]) + if not val: + if not end_bb.is_terminated: + saved_block = Gen.builder.block + Gen.builder.position_at_start(end_bb) + Gen.builder.unreachable() + Gen.builder.position_at_end(saved_block) + return None + if not isinstance(val.type, ir.IntType) or val.type.width != 1: + val = _to_bool(val, f"orcond{idx}") + cur_bb = Gen.builder.block + next_bb = Gen.func.append_basic_block(name=f"or.rhs{idx}") + Gen.builder.cbranch(val, end_bb, next_bb) + phi_incomings.append((ir.Constant(ir.IntType(1), 1), cur_bb)) + Gen.builder.position_at_start(next_bb) + last_val = self.HandleExprLlvm(values[-1]) + if not last_val: + last_val = ir.Constant(ir.IntType(1), 1) + if not isinstance(last_val.type, ir.IntType) or last_val.type.width != 1: + last_val = _to_bool(last_val, "orcond_last") + last_bb = Gen.builder.block + Gen.builder.branch(end_bb) + Gen.builder.position_at_start(end_bb) + result_phi = Gen.builder.phi(ir.IntType(1), name="or.result") + for const_val, bb in phi_incomings: + result_phi.add_incoming(const_val, bb) + result_phi.add_incoming(last_val, last_bb) + return result_phi + return None + + def _HandleUnaryOpLlvm(self, Node: ast.UnaryOp) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + OperandVal: Any = self.HandleExprLlvm(Node.operand) + if not OperandVal: + return None + OpSymbol: str | None = self.Trans.ExprHandler.GetUnaryOpSymbol(Node.op) + if not OpSymbol: + return None + UNARY_OVERLOAD_MAP: dict[str, str] = { + '-': '__neg__', + '+': '__pos__', + '~': '__invert__', + } + if OpSymbol in UNARY_OVERLOAD_MAP: + ClassName: str | None = None + if isinstance(OperandVal.type, ir.PointerType): + pointee: ir.Type = OperandVal.type.pointee + if isinstance(pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + for CN, ST in Gen.structs.items(): + if pointee == ST: + ClassName = CN + break + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + ClassName = self.Trans.ExprUtils._get_var_class(Node.operand, Gen) + if ClassName and ClassName in Gen.structs: + TargetStructPtr: ir.PointerType = ir.PointerType(Gen.structs[ClassName]) + OperandVal = Gen.builder.bitcast(OperandVal, TargetStructPtr, name=f"bitcast_to_{ClassName}") + if ClassName is None and isinstance(Node.operand, ast.Name): + ClassName = Gen.var_struct_class.get(Node.operand.id) + if ClassName and ClassName in Gen.structs: + if isinstance(OperandVal.type, ir.PointerType) and isinstance(OperandVal.type.pointee, ir.IntType) and OperandVal.type.pointee.width == 8: + TargetStructPtr = ir.PointerType(Gen.structs[ClassName]) + OperandVal = Gen.builder.bitcast(OperandVal, TargetStructPtr, name=f"bitcast_to_{ClassName}") + if ClassName: + op_name: str = UNARY_OVERLOAD_MAP[OpSymbol] + result: Any = self.Trans.ExprUtils._try_operator_overLoad(ClassName, op_name, OperandVal, Gen) + if result is not None: + return result + if OpSymbol == 'not': + if isinstance(OperandVal.type, ir.IntType) and OperandVal.type.width == 1: + return Gen.builder.not_(OperandVal, name="not_result") + if isinstance(OperandVal.type, ir.PointerType): + null_val: ir.Constant = ir.Constant(OperandVal.type, None) + return Gen.builder.icmp_signed('==', OperandVal, null_val, name="not_result") + zero: ir.Constant = ir.Constant(OperandVal.type, 0) + return Gen.builder.icmp_signed('==', OperandVal, zero, name="not_result") + elif OpSymbol == '~': + if isinstance(OperandVal.type, ir.IntType): + mask: int = (1 << OperandVal.type.width) - 1 + return Gen.builder.xor(OperandVal, ir.Constant(OperandVal.type, mask), name="bnot_result") + return Gen.builder.xor(OperandVal, ir.Constant(OperandVal.type, -1), name="bnot_result") + elif OpSymbol == '+': + return OperandVal + elif OpSymbol == '-': + if isinstance(OperandVal.type, ir.IntType): + return Gen.builder.neg(OperandVal, name="neg_result") + elif isinstance(OperandVal.type, (ir.FloatType, ir.DoubleType)): + return Gen.builder.fneg(OperandVal, name="fneg_result") + return None + + def _HandleCompareLlvm(self, Node: ast.Compare) -> ir.Value | None: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + LeftVal: Any = self.HandleExprLlvm(Node.left) + if not LeftVal: + return None + is_unsigned: bool = Gen._check_node_unsigned(Node.left) or (Gen._check_node_unsigned(Node.comparators[0]) if Node.comparators else False) + if len(Node.ops) == 1: + Op: ast.cmpop = Node.ops[0] + # in/not in: 调用 __contains__ + if isinstance(Op, (ast.In, ast.NotIn)): + RightVal: Any = self.HandleExprLlvm(Node.comparators[0]) + if RightVal: + ClassName: str | None = self.Trans.ExprHandler._get_var_class(Node.comparators[0], Gen) + if ClassName and Gen._has_function(f'{ClassName}.__contains__'): + # 处理 RightVal 为 i8（整数）的情况：联合类型 | t.CPtr 降级后可能被 load 为 i8 + if isinstance(RightVal.type, ir.IntType) and RightVal.type.width == 8: + RightVal = Gen.builder.inttoptr(RightVal, ir.PointerType(ir.IntType(8)), name=f"i8_to_ptr_{ClassName}") + if isinstance(RightVal.type, ir.PointerType) and isinstance(RightVal.type.pointee, ir.IntType) and RightVal.type.pointee.width == 8: + RightVal = Gen.builder.bitcast(RightVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + contains_func: ir.Function = Gen._get_function(f'{ClassName}.__contains__') + result: ir.Value = Gen.builder.call(contains_func, [RightVal, LeftVal], name=f"call_{ClassName}.__contains__") + # 归一化为 i1 + if isinstance(result.type, ir.IntType) and result.type.width > 1: + zero: ir.Constant = ir.Constant(result.type, 0) + result = Gen.builder.icmp_signed('!=', result, zero, name="contains_bool") + if isinstance(Op, ast.NotIn): + result = Gen.builder.not_(result, name="not_in_result") + return result + return ir.Constant(ir.IntType(1), 0) + ComparatorSymbol: str | None = self.Trans.ExprHandler.GetComparatorSymbol(Op) + if not ComparatorSymbol: + return None + RightVal: Any = self.HandleExprLlvm(Node.comparators[0]) + if not RightVal: + return None + if isinstance(LeftVal, ir.Constant) and isinstance(RightVal, ir.Constant): + if isinstance(LeftVal.type, ir.IntType) and isinstance(RightVal.type, ir.IntType): + lv: int = LeftVal.constant + rv: int = RightVal.constant + lw: int = LeftVal.type.width + rw: int = RightVal.type.width + if lw < rw: + lv = lv if lv >= 0 else lv + (1 << lw) + elif rw < lw: + rv = rv if rv >= 0 else rv + (1 << rw) + if is_unsigned: + mask_l: int = (1 << max(lw, rw)) - 1 + lv = lv & mask_l + rv = rv & mask_l + cmp_map: dict[str, bool] = { + '==': lv == rv, '!=': lv != rv, + '<': lv < rv, '>': lv > rv, + '<=': lv <= rv, '>=': lv >= rv, + } + result: bool = cmp_map.get(ComparatorSymbol, False) + return ir.Constant(ir.IntType(1), 1 if result else 0) + if isinstance(LeftVal.type, ir.IntType) and isinstance(RightVal.type, ir.IntType): + if LeftVal.type.width < RightVal.type.width: + # 如果宽类型常量值超出窄类型有符号范围，则必须用zext + # i8 类型默认使用 zext（系统编程中 i8 几乎总是无符号字节） + need_zext: bool = is_unsigned or Gen._check_node_unsigned(Node.left) or LeftVal.type.width == 8 or LeftVal.type.width == 1 + if not need_zext and isinstance(RightVal, ir.Constant): + signed_max: int = (1 << (LeftVal.type.width - 1)) - 1 + try: + if RightVal.constant > signed_max or RightVal.constant < 0: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + LeftVal = Gen.builder.zext(LeftVal, RightVal.type, name="zext_left_cmp") + else: + LeftVal = Gen.builder.sext(LeftVal, RightVal.type, name="sext_left_cmp") + elif LeftVal.type.width > RightVal.type.width: + need_zext = is_unsigned or Gen._check_node_unsigned(Node.comparators[0]) or RightVal.type.width == 8 or RightVal.type.width == 1 + if not need_zext and isinstance(LeftVal, ir.Constant): + signed_max = (1 << (RightVal.type.width - 1)) - 1 + try: + if LeftVal.constant > signed_max or LeftVal.constant < 0: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + RightVal = Gen.builder.zext(RightVal, LeftVal.type, name="zext_right_cmp") + else: + RightVal = Gen.builder.sext(RightVal, LeftVal.type, name="sext_right_cmp") + elif isinstance(LeftVal.type, ir.PointerType) and isinstance(RightVal.type, ir.IntType): + if isinstance(LeftVal.type.pointee, ir.IntType) and LeftVal.type.pointee.width == 8 and isinstance(RightVal.type, ir.IntType) and RightVal.type.width == 8: + LeftVal = Gen._load(LeftVal, name="Load_char_ptr_cmp") + else: + if isinstance(RightVal, ir.Constant) and RightVal.constant == 0: + RightVal = ir.Constant(LeftVal.type, None) + else: + # 不将整数转为指针（避免 strcmp 解引用无效地址），改为将指针转为整数走正常 == + LeftVal = Gen.builder.ptrtoint(LeftVal, RightVal.type, name="ptr2int_cmp") + elif isinstance(RightVal.type, ir.PointerType) and isinstance(LeftVal.type, ir.IntType): + if isinstance(RightVal.type.pointee, ir.IntType) and RightVal.type.pointee.width == 8 and isinstance(LeftVal.type, ir.IntType) and LeftVal.type.width == 8: + RightVal = Gen._load(RightVal, name="Load_char_ptr_cmp") + else: + if isinstance(LeftVal, ir.Constant) and LeftVal.constant == 0: + LeftVal = ir.Constant(RightVal.type, None) + else: + # 不将整数转为指针（避免 strcmp 解引用无效地址），改为将指针转为整数走正常 == + RightVal = Gen.builder.ptrtoint(RightVal, LeftVal.type, name="ptr2int_cmp") + # is / is not: 地址比较（指针）或值比较（int） + if ComparatorSymbol in ('is', 'is not'): + if isinstance(LeftVal.type, ir.PointerType) and isinstance(RightVal.type, ir.PointerType): + addr_cmp = Gen.builder.icmp_unsigned('==', LeftVal, RightVal, name="is_cmp") + if ComparatorSymbol == 'is not': + addr_cmp = Gen.builder.not_(addr_cmp, name="is_not_result") + return addr_cmp + if isinstance(LeftVal.type, ir.IntType) and isinstance(RightVal.type, ir.IntType): + val_cmp = Gen.builder.icmp_signed('==', LeftVal, RightVal, name="is_cmp") + if ComparatorSymbol == 'is not': + val_cmp = Gen.builder.not_(val_cmp, name="is_not_result") + return val_cmp + # str/bytes == str/bytes: 值比较（调用 strcmp），None 比较用地址 + if ComparatorSymbol in ('==', '!='): + if (isinstance(LeftVal.type, ir.PointerType) and isinstance(LeftVal.type.pointee, ir.IntType) and LeftVal.type.pointee.width == 8 + and isinstance(RightVal.type, ir.PointerType) and isinstance(RightVal.type.pointee, ir.IntType) and RightVal.type.pointee.width == 8): + left_is_null: bool = isinstance(LeftVal, ir.Constant) and LeftVal.constant is None + right_is_null: bool = isinstance(RightVal, ir.Constant) and RightVal.constant is None + if not (left_is_null or right_is_null): + strcmp_func = Gen.get_or_declare_c_func('strcmp', ir.FunctionType(ir.IntType(32), [ir.IntType(8).as_pointer(), ir.IntType(8).as_pointer()])) + cmp_result = Gen.builder.call(strcmp_func, [LeftVal, RightVal], name="str_eq_cmp") + zero = ir.Constant(ir.IntType(32), 0) + eq_result = Gen.builder.icmp_signed('==', cmp_result, zero, name="str_eq") + if ComparatorSymbol == '!=': + eq_result = Gen.builder.not_(eq_result, name="str_neq") + return eq_result + if isinstance(LeftVal.type, ir.PointerType) and isinstance(RightVal.type, ir.PointerType): + result = Gen.builder.icmp_unsigned(ComparatorSymbol, LeftVal, RightVal, name="cmpresult") + elif isinstance(LeftVal.type, (ir.FloatType, ir.DoubleType)) or isinstance(RightVal.type, (ir.FloatType, ir.DoubleType)): + if isinstance(LeftVal.type, (ir.FloatType, ir.DoubleType)) and isinstance(RightVal.type, ir.IntType): + RightVal = Gen.builder.sitofp(RightVal, LeftVal.type, name="int_to_float_cmp") + elif isinstance(RightVal.type, (ir.FloatType, ir.DoubleType)) and isinstance(LeftVal.type, ir.IntType): + LeftVal = Gen.builder.sitofp(LeftVal, RightVal.type, name="int_to_float_cmp") + result = Gen.builder.fcmp_unordered(ComparatorSymbol, LeftVal, RightVal, name="cmpresult") + elif is_unsigned: + result = Gen.builder.icmp_unsigned(ComparatorSymbol, LeftVal, RightVal, name="cmpresult") + else: + result = Gen.builder.icmp_signed(ComparatorSymbol, LeftVal, RightVal, name="cmpresult") + return result + EndBB: ir.Block = Gen.func.append_basic_block(name="cmp.end") + phi_incomings: list[tuple[ir.Value, ir.Block]] = [] + for i, (Op, Comparator) in enumerate(zip(Node.ops, Node.comparators)): + ComparatorSymbol = self.Trans.ExprHandler.GetComparatorSymbol(Op) + if not ComparatorSymbol: + continue + RightVal = self.HandleExprLlvm(Comparator) + if not RightVal: + continue + if isinstance(LeftVal.type, ir.IntType) and isinstance(RightVal.type, ir.IntType): + if LeftVal.type.width < RightVal.type.width: + need_zext = is_unsigned or Gen._check_node_unsigned(Node.left) or LeftVal.type.width == 8 or LeftVal.type.width == 1 + if not need_zext and isinstance(RightVal, ir.Constant): + signed_max = (1 << (LeftVal.type.width - 1)) - 1 + try: + if RightVal.constant > signed_max or RightVal.constant < 0: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + LeftVal = Gen.builder.zext(LeftVal, RightVal.type, name=f"zext_left_{i}") + else: + LeftVal = Gen.builder.sext(LeftVal, RightVal.type, name=f"sext_left_{i}") + elif LeftVal.type.width > RightVal.type.width: + need_zext = is_unsigned or Gen._check_node_unsigned(Comparator) or RightVal.type.width == 8 or RightVal.type.width == 1 + if not need_zext and isinstance(LeftVal, ir.Constant): + signed_max = (1 << (RightVal.type.width - 1)) - 1 + try: + if LeftVal.constant > signed_max or LeftVal.constant < 0: + need_zext = True + except (TypeError, ValueError): + pass + if need_zext: + RightVal = Gen.builder.zext(RightVal, LeftVal.type, name=f"zext_right_{i}") + else: + RightVal = Gen.builder.sext(RightVal, LeftVal.type, name=f"sext_right_{i}") + if is_unsigned: + cmp_result: Any = Gen.builder.icmp_unsigned(ComparatorSymbol, LeftVal, RightVal, name=f"cmp_{i}") + elif isinstance(LeftVal.type, (ir.FloatType, ir.DoubleType)) or isinstance(RightVal.type, (ir.FloatType, ir.DoubleType)): + if isinstance(LeftVal.type, (ir.FloatType, ir.DoubleType)) and isinstance(RightVal.type, ir.IntType): + RightVal = Gen.builder.sitofp(RightVal, LeftVal.type, name=f"int_to_float_cmp_{i}") + elif isinstance(RightVal.type, (ir.FloatType, ir.DoubleType)) and isinstance(LeftVal.type, ir.IntType): + LeftVal = Gen.builder.sitofp(LeftVal, RightVal.type, name=f"int_to_float_cmp_{i}") + cmp_result = Gen.builder.fcmp_unordered(ComparatorSymbol, LeftVal, RightVal, name=f"cmp_{i}") + else: + cmp_result = Gen.builder.icmp_signed(ComparatorSymbol, LeftVal, RightVal, name=f"cmp_{i}") + if i < len(Node.ops) - 1: + NextBB: ir.Block = Gen.func.append_basic_block(name=f"cmp.next.{i}") + cur_bb: ir.Block = Gen.builder.block + Gen.builder.cbranch(cmp_result, NextBB, EndBB) + phi_incomings.append((ir.Constant(ir.IntType(1), 0), cur_bb)) + Gen.builder.position_at_start(NextBB) + LeftVal = RightVal + else: + cur_bb = Gen.builder.block + Gen.builder.branch(EndBB) + phi_incomings.append((cmp_result, cur_bb)) + Gen.builder.position_at_start(EndBB) + final_result: Any = Gen.builder.phi(ir.IntType(1), name="final_cmp") + for val, bb in phi_incomings: + final_result.add_incoming(val, bb) + return final_result \ No newline at end of file diff --git a/lib/core/Handles/HandlesExprUtils.py b/lib/core/Handles/HandlesExprUtils.py new file mode 100644 index 0000000..29c555f --- /dev/null +++ b/lib/core/Handles/HandlesExprUtils.py @@ -0,0 +1,266 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +import ast +import llvmlite.ir as ir +from lib.core.SymbolUtils import IsTModule + + +class ExprUtils: + def __init__(self, translator: "Translator") -> None: + self.Trans: Translator = translator + + def GetOpSymbol(self, Op: ast.cmpop | ast.operator | ast.unaryoperator) -> str | None: + OpMap: dict[type, str] = { + ast.Add: '+', ast.Sub: '-', ast.Mult: '*', ast.Div: '/', + ast.FloorDiv: '//', ast.Mod: '%', ast.Pow: '**', ast.BitAnd: '&', + ast.BitOr: '|', ast.BitXor: '^', ast.LShift: '<<', ast.RShift: '>>', + ast.MatMult: '@', + } + return OpMap.get(type(Op), None) + + def GetUnaryOpSymbol(self, Op: ast.unaryoperator) -> str | None: + OpMap: dict[type, str] = {ast.Invert: '~', ast.Not: 'not', ast.UAdd: '+', ast.USub: '-'} + return OpMap.get(type(Op), None) + + def GetComparatorSymbol(self, Op: ast.cmpop) -> str | None: + OpMap: dict[type, str] = { + ast.Eq: '==', ast.NotEq: '!=', ast.Lt: '<', ast.LtE: '<=', + ast.Gt: '>', ast.GtE: '>=', ast.Is: 'is', ast.IsNot: 'is not', + ast.In: 'in', ast.NotIn: 'not in', + } + return OpMap.get(type(Op), None) + + def _get_var_class(self, node: ast.AST, Gen: LlvmCodeGenerator) -> str | None: + if isinstance(node, ast.Name): + VarName: str = node.id + return Gen.var_struct_class.get(VarName) + if isinstance(node, ast.Attribute): + AttrName: str = node.attr + # 从父对象的类成员类型推导（如 self._ht → ParsedArgs._ht → HashTable） + # 必须先查 BaseClass，避免字段名和变量名冲突导致错误分派 + # （如 old_args.args 匹配到函数参数 args 的 var_struct_class='AST'） + BaseClass: str | None = self._get_var_class(node.value, Gen) + if not BaseClass and isinstance(node.value, ast.Name) and node.value.id == 'self': + BaseClass = getattr(self.Trans, '_CurrentCpythonObjectClass', None) + if BaseClass and BaseClass in Gen.class_members: + for m_name, m_type in Gen.class_members[BaseClass]: + if m_name == AttrName: + if isinstance(m_type, ir.PointerType) and isinstance(m_type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + found = Gen.find_struct_by_pointee(m_type.pointee) + if found: + return found[0] + # 联合类型（如 GSList[Param] | t.CPtr）降级为 i8* 时， + # 从 class_member_element_class 查找泛型特化类名 + element_class_map = getattr(Gen, 'class_member_element_class', {}) + element_class = element_class_map.get(BaseClass, {}).get(AttrName) + if element_class: + return element_class + break + # Fallback: 直接查找 var_struct_class（放在 BaseClass 查找之后， + # 避免字段名和变量名冲突导致错误分派） + cls: str | None = Gen.var_struct_class.get(AttrName) + if cls: + return cls + return None + return None + + def _try_operator_overLoad(self, ClassName: str, op_name: str, obj_val: ir.Value, Gen: LlvmCodeGenerator, other_val: ir.Value | None = None) -> ir.Value | None: + FullMethodName: str = f'{ClassName}.{op_name}' + if not Gen._has_function(FullMethodName): + FullMethodName = f'{ClassName}.{op_name}__' + if Gen._has_function(FullMethodName): + func: Any = Gen._get_function(FullMethodName) + call_args: list[Any] = [obj_val] + if other_val is not None: + call_args.append(other_val) + if len(func.ftype.args) == len(call_args): + for i, (expected_type, actual_val) in enumerate(zip(func.ftype.args, call_args)): + if isinstance(expected_type, ir.PointerType) and isinstance(actual_val.type, ir.PointerType): + if expected_type != actual_val.type: + if isinstance(expected_type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + call_args[i] = Gen.builder.bitcast(actual_val, expected_type, name=f"bitcast_arg_{i}") + result: Any = Gen.builder.call(func, call_args, name=f"call_{FullMethodName}") + return result + return None + + def _llvm_type_to_detailed_string(self, llvm_type: ir.Type, var_name: str = "") -> dict[str, Any]: + info: dict[str, Any] = { + 'name': "unknown", + 'size': 0, + 'signed': False, + 'ptr': False + } + if isinstance(llvm_type, ir.IntType): + info['size'] = llvm_type.width // 8 + info['signed'] = True + if llvm_type.width == 8: + info['name'] = "char" + elif llvm_type.width == 16: + info['name'] = "short" + elif llvm_type.width == 32: + info['name'] = "int" + elif llvm_type.width == 64: + info['name'] = "long long" + else: + info['name'] = f"int{llvm_type.width}" + elif isinstance(llvm_type, ir.DoubleType): + info['size'] = 8 + info['signed'] = True + info['name'] = "double" + elif isinstance(llvm_type, ir.FloatType): + info['size'] = 4 + info['signed'] = True + info['name'] = "float" + elif isinstance(llvm_type, ir.PointerType): + info['size'] = 8 + info['signed'] = False + info['ptr'] = True + pointee_info: dict[str, Any] = self._llvm_type_to_detailed_string(llvm_type.pointee) + info['name'] = f"{pointee_info['name']}*" + elif isinstance(llvm_type, (ir.LiteralStructType, ir.IdentifiedStructType)): + size: int = 0 + for elem in llvm_type.elements: + elem_info: dict[str, Any] = self._llvm_type_to_detailed_string(elem) + size += elem_info['size'] + info['size'] = size + info['signed'] = False + if isinstance(llvm_type, ir.IdentifiedStructType): + info['name'] = llvm_type.name + else: + info['name'] = "struct" + elif isinstance(llvm_type, ir.ArrayType): + elem_info = self._llvm_type_to_detailed_string(llvm_type.element) + info['size'] = elem_info['size'] * llvm_type.count + info['signed'] = elem_info['signed'] + info['name'] = f"{elem_info['name']}[]" + elif isinstance(llvm_type, ir.VoidType): + info['size'] = 0 + info['signed'] = False + info['name'] = "void" + else: + info['size'] = 0 + info['signed'] = False + info['name'] = str(llvm_type) + return info + + def _infer_expr_llvm_type_full(self, node: ast.AST) -> ir.Type: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + if isinstance(node, ast.Constant): + if isinstance(node.value, bool): + return ir.IntType(32) + elif isinstance(node.value, int): + return ir.IntType(32) + elif isinstance(node.value, float): + return ir.DoubleType() + elif isinstance(node.value, str): + if getattr(node, 'kind', None) == 'u': + return ir.PointerType(ir.IntType(16)) + return ir.PointerType(ir.IntType(8)) + return ir.IntType(32) + elif isinstance(node, ast.Name): + if node.id in Gen.variables: + var_ptr: Any = Gen.variables[node.id] + if isinstance(var_ptr.type, ir.PointerType): + return var_ptr.type.pointee + return var_ptr.type + if node.id in Gen._reg_values: + return Gen._reg_values[node.id].type + return ir.IntType(32) + elif isinstance(node, ast.Attribute): + attr_name: str = node.attr + AttrInfo: Any = self.Trans.SymbolTable.lookup(attr_name) + if AttrInfo and getattr(AttrInfo, 'IsEnumMember', None): + return ir.IntType(32) + return ir.IntType(32) + elif isinstance(node, ast.BinOp): + left_type: ir.Type = self._infer_expr_llvm_type_full(node.left) + right_type: ir.Type = self._infer_expr_llvm_type_full(node.right) + if isinstance(left_type, (ir.FloatType, ir.DoubleType)) or isinstance(right_type, (ir.FloatType, ir.DoubleType)): + return ir.DoubleType() + if isinstance(left_type, ir.PointerType) or isinstance(right_type, ir.PointerType): + return ir.PointerType(ir.IntType(8)) + return ir.IntType(32) + elif isinstance(node, ast.UnaryOp): + return self._infer_expr_llvm_type_full(node.operand) + elif isinstance(node, ast.Subscript): + val_type: ir.Type = self._infer_expr_llvm_type_full(node.value) + if isinstance(val_type, ir.PointerType): + if isinstance(val_type.pointee, ir.IntType) and val_type.pointee.width == 8: + return ir.IntType(8) + if isinstance(val_type.pointee, ir.ArrayType): + return ir.PointerType(val_type.pointee.element) + return val_type.pointee + if isinstance(val_type, ir.ArrayType): + return val_type.element + if isinstance(val_type, ir.IntType): + return ir.IntType(8) + return ir.IntType(32) + elif isinstance(node, ast.Call): + if isinstance(node.func, ast.Name): + func_name: str = node.func.id + if Gen._has_function(func_name): + fn: Any = Gen._get_function(func_name) + return fn.function_type.return_type + if func_name == 'len': + return ir.IntType(32) + if func_name == 'sizeof': + return ir.IntType(32) + if func_name == 'ord': + return ir.IntType(32) + if func_name == 'chr': + return ir.IntType(8) + if func_name == 'int': + return ir.IntType(32) + if func_name == 'float' or func_name == 'double': + return ir.DoubleType() + elif isinstance(node.func, ast.Attribute): + if isinstance(node.func.value, ast.Name) and node.func.value.id == 'c': + if node.func.attr == 'Deref': + return ir.IntType(64) + if isinstance(node.func.value, ast.Name) and IsTModule(node.func.value.id, self.Trans.SymbolTable): + if node.func.attr == 'CChar': + return ir.PointerType(ir.IntType(8)) + return ir.IntType(32) + elif isinstance(node, ast.Compare): + return ir.IntType(1) + elif isinstance(node, ast.BoolOp): + return ir.IntType(1) + elif isinstance(node, ast.IfExp): + return self._infer_expr_llvm_type_full(node.body) + elif isinstance(node, ast.Attribute): + return ir.PointerType(ir.IntType(8)) + return ir.IntType(32) + + def _collect_names(self, node: ast.AST, bound: set[str], free: set[str] | None = None) -> None: + """收集 AST 节点中的绑定变量和自由变量 + + Args: + node: AST 节点 + bound: 集合，收集绑定变量（ast.Store 上下文） + free: 集合，收集自由变量（ast.Load 上下文），可为 None + """ + if free is None: + free = set() + if isinstance(node, ast.Name): + if isinstance(node.ctx, ast.Store): + bound.add(node.id) + elif isinstance(node.ctx, ast.Load): + free.add(node.id) + elif isinstance(node, (ast.Lambda, ast.FunctionDef, ast.AsyncFunctionDef)): + # 这些节点有自己的作用域，不递归进入 + # 但需要收集参数名作为 bound + if hasattr(node, 'args') and node.args: + for arg in node.args.args: + bound.add(arg.arg) + elif getattr(node, '_fields', None): + for field in node._fields: + value: Any = getattr(node, field, None) + if isinstance(value, list): + for item in value: + if isinstance(item, (ast.stmt, ast.expr)): + self._collect_names(item, bound, free) + elif isinstance(value, (ast.stmt, ast.expr)): + self._collect_names(value, bound, free) \ No newline at end of file diff --git a/lib/core/Handles/HandlesFor.py b/lib/core/Handles/HandlesFor.py new file mode 100644 index 0000000..9806d68 --- /dev/null +++ b/lib/core/Handles/HandlesFor.py @@ -0,0 +1,1046 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class ForHandle(BaseHandle): + def _HandleForLlvm(self, Node: ast.For) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + Gen._set_node_info(Node, "For") + # zip() 内置：for a, b in zip(iter1, iter2): + if (isinstance(Node.iter, ast.Call) + and isinstance(Node.iter.func, ast.Name) + and Node.iter.func.id == 'zip' + and isinstance(Node.target, ast.Tuple)): + self._HandleForZipLlvm(Node) + return + # enumerate() 内置：for idx, val in enumerate(iterable): + if (isinstance(Node.iter, ast.Call) + and isinstance(Node.iter.func, ast.Name) + and Node.iter.func.id == 'enumerate' + and isinstance(Node.target, ast.Tuple) + and Node.iter.args): + self._HandleForEnumerateLlvm(Node) + return + if not isinstance(Node.target, ast.Name): + return + TargetName: str = Node.target.id + StartVal: ir.Value | None = None + EndVal: ir.Value | None = None + StepVal: ir.Value | None = None + IsRange: bool = False + CondOp: str = '<' + if isinstance(Node.iter, ast.Call): + CallFunc: ast.expr = Node.iter.func + if isinstance(CallFunc, ast.Name) and CallFunc.id == 'range': + IsRange = True + RangeArgs: list[ast.expr] = Node.iter.args + RangeEndArg: ast.expr | None = RangeArgs[1] if len(RangeArgs) >= 2 else RangeArgs[0] if len(RangeArgs) >= 1 else None + OptimizedEnd: bool = False + if len(RangeArgs) >= 2 and isinstance(RangeEndArg, ast.BinOp): + if isinstance(RangeEndArg.op, ast.Add): + if (isinstance(RangeEndArg.right, ast.Constant) and RangeEndArg.right.value == 1): + EndVal = self.HandleExprLlvm(RangeEndArg.left) + CondOp = '<=' + OptimizedEnd = True + elif (isinstance(RangeEndArg.left, ast.Constant) and RangeEndArg.left.value == 1): + EndVal = self.HandleExprLlvm(RangeEndArg.right) + CondOp = '<=' + OptimizedEnd = True + elif isinstance(RangeEndArg.op, ast.Sub): + if isinstance(RangeEndArg.right, ast.Constant) and RangeEndArg.right.value == 1: + EndVal = self.HandleExprLlvm(RangeEndArg.left) + CondOp = '<' + OptimizedEnd = True + if not OptimizedEnd: + if len(RangeArgs) == 1: + EndVal = self.HandleExprLlvm(RangeArgs[0]) + else: + EndVal = self.HandleExprLlvm(RangeEndArg) + if len(RangeArgs) == 1: + StartVal = ir.Constant(ir.IntType(32), 0) + elif len(RangeArgs) >= 2: + StartVal = self.HandleExprLlvm(RangeArgs[0]) + if len(RangeArgs) >= 3: + StepVal = self.HandleExprLlvm(RangeArgs[2]) + # Check if step is a negative constant — need to flip condition + step_arg: ast.expr = RangeArgs[2] + if isinstance(step_arg, ast.UnaryOp) and isinstance(step_arg.op, ast.USub): + if isinstance(step_arg.operand, ast.Constant) and isinstance(step_arg.operand.value, (int, float)): + if step_arg.operand.value > 0: + CondOp = '>' + elif isinstance(step_arg, ast.Constant) and isinstance(step_arg.value, (int, float)): + if step_arg.value < 0: + CondOp = '>' + if not EndVal: + return + if not isinstance(EndVal.type, ir.IntType): + if isinstance(EndVal.type, ir.PointerType) and isinstance(EndVal.type.pointee, ir.IntType): + EndVal = Gen._load(EndVal, name="Load_end") + else: + EndVal = Gen.builder.ptrtoint(EndVal, ir.IntType(64), name="end") + if StartVal and not isinstance(StartVal.type, ir.IntType): + if isinstance(StartVal.type, ir.PointerType) and isinstance(StartVal.type.pointee, ir.IntType): + StartVal = Gen._load(StartVal, name="Load_start") + else: + StartVal = Gen.builder.ptrtoint(StartVal, ir.IntType(64), name="start") + else: + StartVal = ir.Constant(ir.IntType(64), 0) + if not StartVal: + StartVal = ir.Constant(EndVal.type, 0) + if StepVal is None: + StepVal = ir.Constant(EndVal.type, 1) + if not IsRange: + # 检测 c.Addr(x) 模式：按原始变量类型迭代（for i64 in i64* 语义） + if (isinstance(Node.iter, ast.Call) + and isinstance(Node.iter.func, ast.Attribute) + and isinstance(Node.iter.func.value, ast.Name) + and Node.iter.func.value.id == 'c' + and Node.iter.func.attr == 'Addr' + and Node.iter.args): + AddrArg: ast.expr = Node.iter.args[0] + if isinstance(AddrArg, ast.Name) and AddrArg.id in Gen.variables: + _var_ptr: ir.Value | None = Gen.variables[AddrArg.id] + if _var_ptr is not None and isinstance(_var_ptr.type, ir.PointerType): + _pointee: ir.Type = _var_ptr.type.pointee + # 标量整数变量 + if isinstance(_pointee, ir.IntType): + if _pointee.width == 8: + self._HandleForStrLlvm(Node, _var_ptr) + return + self._HandleForPtrLlvm(Node, _var_ptr, _pointee) + return + # 数组变量：取元素类型 + if isinstance(_pointee, ir.ArrayType) and isinstance(_pointee.element, ir.IntType): + _elem_type: ir.IntType = _pointee.element + if _elem_type.width == 8: + self._HandleForStrLlvm(Node, _var_ptr) + return + _typed_ptr: ir.Value = Gen.builder.bitcast(_var_ptr, ir.PointerType(_elem_type), name="addr_arr_elem_cast") + self._HandleForPtrLlvm(Node, _typed_ptr, _elem_type) + return + IterClassName: str | None = self.Trans.ExprHandler._get_var_class(Node.iter, Gen) + if not IterClassName and isinstance(Node.iter, ast.Call) and isinstance(Node.iter.func, ast.Name): + IterClassName = Node.iter.func.id + if IterClassName not in Gen.structs: + IterClassName = None + if IterClassName and Gen._has_function(f'{IterClassName}.__iter__') and Gen._has_function(f'{IterClassName}.__next__'): + self._HandleForIterLlvm(Node, IterClassName) + return + IterVal: ir.Value | None = self.HandleExprLlvm(Node.iter) + if not IterVal: + return + # 回退：通过 LLVM 类型查找结构体类名，检测 __iter__/__next__ + if not IterClassName and isinstance(IterVal.type, ir.PointerType): + _pointee: ir.Type = IterVal.type.pointee + if isinstance(_pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + _found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(_pointee) + if _found: + _cn: str = _found[0] + if not (Gen._has_function(f'{_cn}.__iter__') and Gen._has_function(f'{_cn}.__next__')): + # temp stub 可能不包含泛型特化方法，尝试从 output stub 按需加载 + self.Trans.ImportHandler._TryLoadFuncDeclsFromOutputStub(_cn, Gen) + if Gen._has_function(f'{_cn}.__iter__') and Gen._has_function(f'{_cn}.__next__'): + Gen._UnregisterTempPtr(IterVal) + self._HandleForIterLlvm(Node, _cn) + return + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.IntType) and IterVal.type.pointee.width == 8: + self._HandleForStrLlvm(Node, IterVal) + return + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.ArrayType): + self._HandleForListLlvm(Node, IterVal) + return + StartVal = ir.Constant(ir.IntType(32), 0) + EndVal = IterVal + if not isinstance(EndVal.type, ir.IntType): + EndVal = Gen.builder.ptrtoint(EndVal, ir.IntType(64), name="end") + StartVal = ir.Constant(ir.IntType(64), 0) + StepVal = ir.Constant(EndVal.type, 1) + CondOp = '<' + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + LoopVar: ir.AllocaInstr + if TargetName in Gen.variables and Gen.variables[TargetName] is not None: + LoopVar = Gen.variables[TargetName] + if isinstance(LoopVar.type, ir.PointerType) and isinstance(LoopVar.type.pointee, ir.IntType) and LoopVar.type.pointee == EndVal.type and isinstance(EndVal.type, ir.IntType) and StartVal.type == LoopVar.type.pointee and EndVal.type == StartVal.type: + Gen._store(StartVal, LoopVar) + else: + NewVar: ir.AllocaInstr = Gen._allocaEntry(EndVal.type, name=TargetName) + if StartVal.type != EndVal.type: + if isinstance(EndVal.type, ir.IntType) and isinstance(StartVal.type, ir.IntType): + if StartVal.type.width < EndVal.type.width: + StartVal = Gen.builder.zext(StartVal, EndVal.type, name="zext_loop_start") + elif StartVal.type.width > EndVal.type.width: + StartVal = Gen.builder.trunc(StartVal, EndVal.type, name="trunc_loop_start") + Gen._store(StartVal, NewVar) + Gen.variables[TargetName] = NewVar + LoopVar = NewVar + else: + LoopVar = Gen._allocaEntry(EndVal.type, name=TargetName) + Gen.variables[TargetName] = LoopVar + Gen._store(StartVal, LoopVar) + if StepVal and isinstance(StepVal.type, ir.IntType) and isinstance(EndVal.type, ir.IntType): + if StepVal.type.width < EndVal.type.width: + StepVal = Gen.builder.zext(StepVal, EndVal.type, name="zext_loop_step") + elif StepVal.type.width > EndVal.type.width: + StepVal = Gen.builder.trunc(StepVal, EndVal.type, name="trunc_loop_step") + Gen._record_var_signedness(TargetName, 'int') + CondBB: ir.Block = Gen.func.append_basic_block(name="for.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="for.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="for.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="for.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="for.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + LoopVal: ir.Value = Gen._load(LoopVar, name=TargetName) + is_unsigned: bool = Gen._is_var_unsigned(TargetName) + Cond: ir.Value + if is_unsigned: + Cond = Gen.builder.icmp_unsigned(CondOp, LoopVal, EndVal, name="forcond") + else: + Cond = Gen.builder.icmp_signed(CondOp, LoopVal, EndVal, name="forcond") + Gen.builder.cbranch(Cond, BodyBB, ElseBB if ElseBB else AfterBB) + Gen.builder.position_at_start(BodyBB) + CachedLoopVal: ir.Value = Gen._load(LoopVar, name=TargetName) + Gen._reg_values[TargetName] = CachedLoopVal + self.HandleBodyLlvm(Node.body) + StepLoopVal: ir.Value = CachedLoopVal + if TargetName in Gen._reg_values and Gen._reg_values[TargetName] is CachedLoopVal: + del Gen._reg_values[TargetName] + else: + StepLoopVal = Gen._load(LoopVar, name=TargetName) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + Gen.builder.position_at_start(StepBB) + StepLoopVal2: ir.Value = Gen._load(LoopVar, name=TargetName) + NextVal: ir.Value = Gen.builder.add(StepLoopVal2, StepVal, name="next") + Gen._store(NextVal, LoopVar) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _HandleGlobalLlvm(self, Node: ast.Global) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + self._RegisterGlobalNames(Node.names, Gen, clear_reg_values=True) + + def _RegisterGlobalNames(self, names: list[str], Gen: "Translator.LlvmGen", clear_reg_values: bool = False) -> None: + for name in names: + Gen.global_vars.add(name) + if name in Gen.module.globals: + Gen.variables[name] = Gen.module.globals[name] + if clear_reg_values and name in Gen._reg_values: + del Gen._reg_values[name] + + def _HandleNonlocalLlvm(self, Node: ast.Nonlocal) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + for name in Node.names: + if name in Gen._reg_values and name not in Gen.variables: + OldVal: ir.Value = Gen._reg_values[name] + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=name) + Gen._store(OldVal, var) + Gen.variables[name] = var + del Gen._reg_values[name] + + def _collect_implicit_nonlocal(self, Node: ast.FunctionDef, Gen: "Translator.LlvmGen") -> list[str]: + param_names: set[str] = set() + for arg in Node.args.args: + param_names.add(arg.arg) + local_names: set[str] = set() + for child in ast.walk(Node): + if isinstance(child, ast.Assign): + for target in child.targets: + if isinstance(target, ast.Name): + local_names.add(target.id) + elif isinstance(target, ast.Tuple): + for elt in target.elts: + if isinstance(elt, ast.Name): + local_names.add(elt.id) + elif isinstance(child, ast.AugAssign): + if isinstance(child.target, ast.Name): + local_names.add(child.target.id) + elif isinstance(child, ast.AnnAssign): + if isinstance(child.target, ast.Name): + local_names.add(child.target.id) + elif isinstance(child, ast.For): + if isinstance(child.target, ast.Name): + local_names.add(child.target.id) + read_names: set[str] = set() + for child in ast.walk(Node): + if isinstance(child, ast.Name) and isinstance(child.ctx, ast.Load): + name: str = child.id + if name not in param_names and name not in local_names: + read_names.add(name) + implicit: list[str] = [] + for name in read_names: + if name in Gen.variables and Gen.variables[name] is not None: + implicit.append(name) + elif name in Gen._reg_values: + implicit.append(name) + for called_name in read_names: + if called_name in Gen.nonlocal_params: + for nv, _ in Gen.nonlocal_params[called_name]: + if nv not in param_names and nv not in local_names: + if nv not in read_names: + if nv in Gen.variables and Gen.variables[nv] is not None: + implicit.append(nv) + elif nv in Gen._reg_values: + implicit.append(nv) + return implicit + + def _collect_all_nonlocal_names(self, Node: ast.FunctionDef) -> set[str]: + """Recursively collect all nonlocal variable names from a function and its nested functions.""" + names: set[str] = set() + for stmt in Node.body: + if isinstance(stmt, ast.Nonlocal): + for name in stmt.names: + names.add(name) + elif isinstance(stmt, ast.FunctionDef): + # Recursively collect from nested functions + sub_names: set[str] = self._collect_all_nonlocal_names(stmt) + names.update(sub_names) + return names + + def _HandleNestedFunctionLlvm(self, Node: ast.FunctionDef) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + nonlocal_var_names: list[str] = [] + for stmt in Node.body: + if isinstance(stmt, ast.Nonlocal): + for name in stmt.names: + nonlocal_var_names.append(name) + # Pre-scan ALL nested functions recursively for their nonlocal needs (passthrough) + for stmt in Node.body: + if isinstance(stmt, ast.FunctionDef): + sub_nonlocal_names: set[str] = self._collect_all_nonlocal_names(stmt) + for name in sub_nonlocal_names: + if name not in nonlocal_var_names: + nonlocal_var_names.append(name) + implicit_names: list[str] = self._collect_implicit_nonlocal(Node, Gen) + for name in implicit_names: + if name not in nonlocal_var_names: + nonlocal_var_names.append(name) + extra_params: list[tuple[str, ir.Type]] = [] + for var_name in nonlocal_var_names: + if var_name in Gen.variables and Gen.variables[var_name] is not None: + var: ir.Value = Gen.variables[var_name] + extra_params.append((var_name, var.type)) + elif var_name in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[var_name] + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=var_name) + Gen._store(OldVal, var) + Gen.variables[var_name] = var + del Gen._reg_values[var_name] + extra_params.append((var_name, var.type)) + else: + # Search outer scope stack for the variable + found_var: ir.Value | None = None + for scope in reversed(Gen._variable_scope_stack): + if var_name in scope and scope[var_name] is not None: + found_var = scope[var_name] + break + if found_var is not None: + # Add to current scope so it can be passed to nested functions + Gen.variables[var_name] = found_var + extra_params.append((var_name, found_var.type)) + else: + var = Gen._allocaEntry(ir.IntType(32), name=var_name) + Gen.variables[var_name] = var + extra_params.append((var_name, var.type)) + if extra_params: + Gen.nonlocal_params[Node.name] = [(name, ptr_type) for name, ptr_type in extra_params] + # Push current variables to scope stack before compiling nested function + Gen._variable_scope_stack.append(Gen.variables.copy()) + saved_builder: ir.IRBuilder = Gen.builder + saved_func: ir.Function = Gen.func + saved_variables: dict[str, ir.Value] = Gen.variables.copy() + saved_reg_values: dict[str, ir.Value] = Gen._reg_values.copy() + saved_global_vars: set[str] = Gen.global_vars.copy() + saved_var_signedness: dict[str, str] = Gen.var_signedness.copy() + saved_var_type_info: dict[str, dict[str, str]] = Gen.var_type_info.copy() + saved_var_type_assignments: dict[str, list[dict[str, str]]] = Gen.var_type_assignments.copy() + saved_CurrentCReturnTypes: object | None = getattr(self.Trans, 'CurrentCReturnTypes', None) + saved_CurrentCpythonObjectClass: str | None = getattr(self.Trans, '_CurrentCpythonObjectClass', None) + saved_VarScopes: int = len(self.Trans.VarScopes) + self.Trans.FunctionHandler._EmitFunctionLlvm(Node, Gen, extra_params=extra_params if extra_params else None) + Gen.builder = saved_builder + Gen.func = saved_func + Gen.variables = saved_variables + Gen._reg_values = saved_reg_values + Gen.global_vars = saved_global_vars + Gen.var_signedness = saved_var_signedness + Gen.var_type_info = saved_var_type_info + Gen.var_type_assignments = saved_var_type_assignments + self.Trans.CurrentCReturnTypes = saved_CurrentCReturnTypes + self.Trans._CurrentCpythonObjectClass = saved_CurrentCpythonObjectClass + while len(self.Trans.VarScopes) > saved_VarScopes: + self.Trans.VarScopes.pop() + # Pop scope stack + if Gen._variable_scope_stack: + Gen._variable_scope_stack.pop() + + def _HandleForListLlvm(self, Node: ast.For, ArrPtr: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Name): + return + TargetName: str = Node.target.id + ElemType: ir.Type = ArrPtr.type.pointee.element + ArrayCount: int = ArrPtr.type.pointee.count + IdxVal: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=f"arr_idx") + Gen.builder.store(ir.Constant(ir.IntType(32), 0), IdxVal) + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + LoopVar: ir.AllocaInstr = Gen._allocaEntry(ElemType, name=TargetName) + Gen.variables[TargetName] = LoopVar + Gen._store(ir.Constant(ElemType, None), LoopVar) + CondBB: ir.Block = Gen.func.append_basic_block(name="arrfor.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="arrfor.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="arrfor.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="arrfor.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="arrfor.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + CurIdx: ir.Value = Gen._load(IdxVal, name="cur_idx") + IsDone: ir.Value = Gen.builder.icmp_signed('<', CurIdx, ir.Constant(ir.IntType(32), ArrayCount), name="arr_in_bounds") + Gen.builder.cbranch(IsDone, BodyBB, ElseBB if ElseBB else AfterBB) + Gen.builder.position_at_start(BodyBB) + ElemPtr: ir.Value = Gen.builder.gep(ArrPtr, [ir.Constant(ir.IntType(32), 0), CurIdx], name="arr_elem_ptr") + ElemVal: ir.Value = Gen._load(ElemPtr, name="arr_elem_val") + Gen._store(ElemVal, LoopVar) + Gen._record_var_signedness(TargetName, 'ptr') + Gen._reg_values[TargetName] = ElemVal + self.HandleBodyLlvm(Node.body) + if TargetName in Gen._reg_values and Gen._reg_values[TargetName] is ElemVal: + del Gen._reg_values[TargetName] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + Gen.builder.position_at_start(StepBB) + CurIdx2: ir.Value = Gen._load(IdxVal, name="cur_idx2") + NextIdx: ir.Value = Gen.builder.add(CurIdx2, ir.Constant(ir.IntType(32), 1), name="next_idx") + Gen._store(NextIdx, IdxVal) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _HandleForStrLlvm(self, Node: ast.For, StrVal: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Name): + return + TargetName: str = Node.target.id + Gen._UnregisterTempPtr(StrVal) + PtrVal: ir.AllocaInstr = Gen.builder.alloca(ir.IntType(8).as_pointer(), name=f"str_ptr_copy") + Gen.builder.store(StrVal, PtrVal) + CharType: ir.IntType = ir.IntType(8) + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + LoopVar: ir.AllocaInstr = Gen._allocaEntry(CharType, name=TargetName) + Gen.variables[TargetName] = LoopVar + Gen._store(ir.Constant(CharType, 0), LoopVar) + CondBB: ir.Block = Gen.func.append_basic_block(name="strfor.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="strfor.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="strfor.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="strfor.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="strfor.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + CurrentPtr: ir.Value = Gen._load(PtrVal, name="current_ptr") + RawChar: ir.Value = Gen._load(CurrentPtr, name="raw_char") + NullCond: ir.Value = Gen.builder.icmp_signed('==', RawChar, ir.Constant(CharType, 0), name="is_null") + Gen.builder.cbranch(NullCond, ElseBB if ElseBB else AfterBB, BodyBB) + Gen.builder.position_at_start(BodyBB) + Gen._store(RawChar, LoopVar) + Gen._record_var_signedness(TargetName, 'char') + Gen._reg_values[TargetName] = RawChar + self.HandleBodyLlvm(Node.body) + if TargetName in Gen._reg_values and Gen._reg_values[TargetName] is RawChar: + del Gen._reg_values[TargetName] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + Gen.builder.position_at_start(StepBB) + CurrentPtr2: ir.Value = Gen._load(PtrVal, name="current_ptr2") + NextPtr: ir.Value = Gen.builder.gep(CurrentPtr2, [ir.Constant(ir.IntType(32), 1)], name="next_ptr") + Gen._store(NextPtr, PtrVal) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _HandleForPtrLlvm(self, Node: ast.For, PtrVal: ir.Value, ElemType: ir.IntType) -> None: + """通用指针迭代：按 ElemType 大小逐元素迭代，终止条件为元素值 == 0 + + 适用于 for i in c.Addr(u) where u: i32/i64 等非 i8 整数类型。 + """ + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Name): + return + TargetName: str = Node.target.id + # 确保 PtrVal 是 ElemType* + TypedPtr: ir.Value = PtrVal + if isinstance(PtrVal.type, ir.PointerType) and PtrVal.type.pointee is not ElemType: + TypedPtr = Gen.builder.bitcast(PtrVal, ir.PointerType(ElemType), name="ptr_elem_cast") + Gen._UnregisterTempPtr(TypedPtr) + # 分配指针副本 + PtrCopy: ir.AllocaInstr = Gen.builder.alloca(ir.PointerType(ElemType), name="ptr_copy") + Gen.builder.store(TypedPtr, PtrCopy) + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + LoopVar: ir.AllocaInstr = Gen._allocaEntry(ElemType, name=TargetName) + Gen.variables[TargetName] = LoopVar + Gen._store(ir.Constant(ElemType, 0), LoopVar) + CondBB: ir.Block = Gen.func.append_basic_block(name="ptrfor.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="ptrfor.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="ptrfor.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="ptrfor.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="ptrfor.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + CurrentPtr: ir.Value = Gen._load(PtrCopy, name="current_ptr") + RawElem: ir.Value = Gen._load(CurrentPtr, name="raw_elem") + NullCond: ir.Value = Gen.builder.icmp_signed('==', RawElem, ir.Constant(ElemType, 0), name="is_null") + Gen.builder.cbranch(NullCond, ElseBB if ElseBB else AfterBB, BodyBB) + Gen.builder.position_at_start(BodyBB) + Gen._store(RawElem, LoopVar) + Gen._record_var_signedness(TargetName, 'int') + Gen._reg_values[TargetName] = RawElem + self.HandleBodyLlvm(Node.body) + if TargetName in Gen._reg_values and Gen._reg_values[TargetName] is RawElem: + del Gen._reg_values[TargetName] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + Gen.builder.position_at_start(StepBB) + CurrentPtr2: ir.Value = Gen._load(PtrCopy, name="current_ptr2") + NextPtr: ir.Value = Gen.builder.gep(CurrentPtr2, [ir.Constant(ir.IntType(32), 1)], name="next_ptr") + Gen._store(NextPtr, PtrCopy) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _HandleForIterLlvm(self, Node: ast.For, ClassName: str) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Name): + return + TargetName: str = Node.target.id + IterVal: ir.Value | None = self.HandleExprLlvm(Node.iter) + if not IterVal: + return + Gen._UnregisterTempPtr(IterVal) + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.IntType) and IterVal.type.pointee.width == 8: + IterVal = Gen.builder.bitcast(IterVal, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + IterCall: ir.Function = Gen._get_function(f'{ClassName}.__iter__') + IterResult: ir.Value = Gen.builder.call(IterCall, [IterVal], name=f"call_{ClassName}.__iter__") + Gen._UnregisterTempPtr(IterResult) + if isinstance(IterResult.type, ir.PointerType) and isinstance(IterResult.type.pointee, ir.IntType) and IterResult.type.pointee.width == 8: + IterResult = Gen.builder.bitcast(IterResult, ir.PointerType(Gen.structs[ClassName]), name=f"cast_iter_{ClassName}") + NextCall: ir.Function = Gen._get_function(f'{ClassName}.__next__') + StopFlagPtr: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(1), name="stop_iter_flag") + Gen.builder.store(ir.Constant(ir.IntType(1), 0), StopFlagPtr) + NextReturnType: ir.Type = NextCall.function_type.return_type + if TargetName in Gen._reg_values: + del Gen._reg_values[TargetName] + LoopVar: ir.AllocaInstr = Gen._allocaEntry(NextReturnType, name=TargetName) + Gen.variables[TargetName] = LoopVar + CondBB: ir.Block = Gen.func.append_basic_block(name="iter.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="iter.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="iter.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="iter.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="iter.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + Gen.builder.store(ir.Constant(ir.IntType(1), 0), StopFlagPtr) + NextCallArgs: list[ir.Value] = [IterResult, StopFlagPtr] + if NextCall and len(NextCall.function_type.args) > len(NextCallArgs): + last_param_type: ir.Type = NextCall.function_type.args[-1] + if (isinstance(last_param_type, ir.PointerType) + and isinstance(last_param_type.pointee, ir.PointerType) + and isinstance(last_param_type.pointee.pointee, ir.IntType) + and last_param_type.pointee.pointee.width == 8): + null_msg: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name="eh_msg_out_null") + NextCallArgs.append(null_msg) + NextVal: ir.Value = Gen.builder.call(NextCall, NextCallArgs, name=f"call_{ClassName}.__next__") + StopFlag: ir.Value = Gen.builder.load(StopFlagPtr, name="stop_flag") + Cond: ir.Value = Gen.builder.icmp_signed('==', StopFlag, ir.Constant(ir.IntType(1), 0), name="iter_cond") + Gen.builder.cbranch(Cond, BodyBB, ElseBB if ElseBB else AfterBB) + Gen.builder.position_at_start(BodyBB) + Gen._store(NextVal, LoopVar) + Gen._reg_values[TargetName] = NextVal + self.HandleBodyLlvm(Node.body) + if TargetName in Gen._reg_values and Gen._reg_values[TargetName] is NextVal: + del Gen._reg_values[TargetName] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + Gen.builder.position_at_start(StepBB) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + # ===== zip() 内置支持 ===== + + def _HandleForZipLlvm(self, Node: ast.For) -> None: + """for a, b in zip(iter1, iter2): — 多迭代器并行迭代 + + 兼容 range、__iter__/__next__、字符串、数组、指针等所有迭代类型。 + 任一迭代器停止则退出循环。 + """ + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Tuple): + return + Targets: list[ast.Name] = [t for t in Node.target.elts if isinstance(t, ast.Name)] + ZipArgs: list[ast.expr] = Node.iter.args + if len(Targets) != len(ZipArgs) or len(Targets) < 2: + return + + # 为每个可迭代对象设置迭代状态 + IterInfos: list[dict] = [] + for i, arg in enumerate(ZipArgs): + target_name: str = Targets[i].id + info: dict | None = self._zip_setup_iter(arg, target_name, Gen) + if info is None: + return + IterInfos.append(info) + + CondBB: ir.Block = Gen.func.append_basic_block(name="zip.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="zip.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="zip.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="zip.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="zip.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + + # Cond 块：为每个迭代器获取下一个值和停止条件 + Gen.builder.position_at_start(CondBB) + AnyStopped: ir.Value | None = None + for info in IterInfos: + val: ir.Value + stopped: ir.Value + val, stopped = self._zip_get_next(info, Gen) + info['last_value'] = val + if AnyStopped is None: + AnyStopped = stopped + else: + AnyStopped = Gen.builder.or_(AnyStopped, stopped, name="zip_any_stop") + Gen.builder.cbranch(AnyStopped, ElseBB if ElseBB else AfterBB, BodyBB) + + # Body 块：赋值并执行循环体 + Gen.builder.position_at_start(BodyBB) + for info in IterInfos: + Gen._store(info['last_value'], info['loop_var']) + Gen._reg_values[info['target_name']] = info['last_value'] + self.HandleBodyLlvm(Node.body) + for info in IterInfos: + tn: str = info['target_name'] + if tn in Gen._reg_values and Gen._reg_values[tn] is info['last_value']: + del Gen._reg_values[tn] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + + # Step 块：跳回 Cond（推进已在 Cond 中完成） + Gen.builder.position_at_start(StepBB) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _zip_setup_iter(self, arg: ast.expr, target_name: str, Gen: "Translator.LlvmGen") -> dict | None: + """为 zip 参数设置迭代状态，返回包含迭代信息的 dict。""" + if target_name in Gen._reg_values: + del Gen._reg_values[target_name] + + # Case 1: range(n) / range(start, end) / range(start, end, step) + if isinstance(arg, ast.Call) and isinstance(arg.func, ast.Name) and arg.func.id == 'range': + return self._zip_setup_range(arg, target_name, Gen) + + # Case 2: c.Addr(x) — 指针/数组迭代 + if (isinstance(arg, ast.Call) + and isinstance(arg.func, ast.Attribute) + and isinstance(arg.func.value, ast.Name) + and arg.func.value.id == 'c' + and arg.func.attr == 'Addr' + and arg.args): + AddrArg: ast.expr = arg.args[0] + if isinstance(AddrArg, ast.Name) and AddrArg.id in Gen.variables: + _var_ptr: ir.Value | None = Gen.variables[AddrArg.id] + if _var_ptr is not None and isinstance(_var_ptr.type, ir.PointerType): + _pointee: ir.Type = _var_ptr.type.pointee + if isinstance(_pointee, ir.IntType): + if _pointee.width == 8: + return self._zip_setup_str(_var_ptr, target_name, Gen) + return self._zip_setup_ptr(_var_ptr, target_name, _pointee, Gen) + if isinstance(_pointee, ir.ArrayType) and isinstance(_pointee.element, ir.IntType): + _elem_type: ir.IntType = _pointee.element + if _elem_type.width == 8: + return self._zip_setup_str(_var_ptr, target_name, Gen) + _typed_ptr: ir.Value = Gen.builder.bitcast(_var_ptr, ir.PointerType(_elem_type), name="zip_addr_cast") + return self._zip_setup_ptr(_typed_ptr, target_name, _elem_type, Gen) + + # Case 3: __iter__/__next__ 类迭代 + IterClassName: str | None = self.Trans.ExprHandler._get_var_class(arg, Gen) + if not IterClassName and isinstance(arg, ast.Call) and isinstance(arg.func, ast.Name): + IterClassName = arg.func.id + if IterClassName not in Gen.structs: + IterClassName = None + if IterClassName and Gen._has_function(f'{IterClassName}.__iter__') and Gen._has_function(f'{IterClassName}.__next__'): + return self._zip_setup_iter_class(arg, target_name, IterClassName, Gen) + + # Case 4-6: 求值表达式后按类型分发 + IterVal: ir.Value | None = self.HandleExprLlvm(arg) + if not IterVal: + return None + # 回退：通过 LLVM 类型查找结构体类名，检测 __iter__/__next__ + if not IterClassName and isinstance(IterVal.type, ir.PointerType): + _pointee: ir.Type = IterVal.type.pointee + if isinstance(_pointee, (ir.IdentifiedStructType, ir.LiteralStructType)): + _found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(_pointee) + if _found: + _cn: str = _found[0] + if not (Gen._has_function(f'{_cn}.__iter__') and Gen._has_function(f'{_cn}.__next__')): + self.Trans.ImportHandler._TryLoadFuncDeclsFromOutputStub(_cn, Gen) + if Gen._has_function(f'{_cn}.__iter__') and Gen._has_function(f'{_cn}.__next__'): + Gen._UnregisterTempPtr(IterVal) + return self._zip_setup_iter_class_from_val(IterVal, target_name, _cn, Gen) + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.IntType) and IterVal.type.pointee.width == 8: + return self._zip_setup_str(IterVal, target_name, Gen) + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.ArrayType): + return self._zip_setup_array(IterVal, target_name, Gen) + # Case 6: 整数计数器 + return self._zip_setup_int(IterVal, target_name, Gen) + + def _zip_setup_range(self, arg: ast.Call, target_name: str, Gen: "Translator.LlvmGen") -> dict | None: + RangeArgs: list[ast.expr] = arg.args + if len(RangeArgs) == 1: + StartVal: ir.Value = ir.Constant(ir.IntType(32), 0) + EndVal: ir.Value = self.HandleExprLlvm(RangeArgs[0]) + StepVal: ir.Value = ir.Constant(ir.IntType(32), 1) + elif len(RangeArgs) == 2: + StartVal = self.HandleExprLlvm(RangeArgs[0]) + EndVal = self.HandleExprLlvm(RangeArgs[1]) + StepVal = ir.Constant(ir.IntType(32), 1) + elif len(RangeArgs) >= 3: + StartVal = self.HandleExprLlvm(RangeArgs[0]) + EndVal = self.HandleExprLlvm(RangeArgs[1]) + StepVal = self.HandleExprLlvm(RangeArgs[2]) + else: + return None + if not EndVal: + return None + if not isinstance(EndVal.type, ir.IntType): + if isinstance(EndVal.type, ir.PointerType) and isinstance(EndVal.type.pointee, ir.IntType): + EndVal = Gen._load(EndVal, name="zip_end") + else: + EndVal = Gen.builder.ptrtoint(EndVal, ir.IntType(64), name="zip_end") + if not isinstance(StartVal.type, ir.IntType): + StartVal = ir.Constant(EndVal.type, 0) + if isinstance(StartVal.type, ir.IntType) and isinstance(EndVal.type, ir.IntType): + if StartVal.type.width < EndVal.type.width: + StartVal = Gen.builder.zext(StartVal, EndVal.type, name="zip_start_zext") + elif StartVal.type.width > EndVal.type.width: + EndVal = Gen.builder.zext(EndVal, StartVal.type, name="zip_end_zext") + if isinstance(StepVal.type, ir.IntType) and isinstance(EndVal.type, ir.IntType): + if StepVal.type.width < EndVal.type.width: + StepVal = Gen.builder.zext(StepVal, EndVal.type, name="zip_step_zext") + IdxVar: ir.AllocaInstr = Gen._allocaEntry(EndVal.type, name=f"zip_{target_name}_idx") + Gen._store(StartVal, IdxVar) + LoopVar: ir.AllocaInstr = Gen._allocaEntry(EndVal.type, name=target_name) + Gen.variables[target_name] = LoopVar + Gen._store(StartVal, LoopVar) + Gen._record_var_signedness(target_name, 'int') + return { + 'iter_type': 'range', + 'target_name': target_name, + 'value_type': EndVal.type, + 'idx_var': IdxVar, + 'end_val': EndVal, + 'step_val': StepVal, + 'loop_var': LoopVar, + } + + def _zip_setup_iter_class(self, arg: ast.expr, target_name: str, ClassName: str, Gen: "Translator.LlvmGen") -> dict | None: + IterVal: ir.Value | None = self.HandleExprLlvm(arg) + if not IterVal: + return None + return self._zip_setup_iter_class_from_val(IterVal, target_name, ClassName, Gen) + + def _zip_setup_iter_class_from_val(self, IterVal: ir.Value, target_name: str, ClassName: str, Gen: "Translator.LlvmGen") -> dict | None: + Gen._UnregisterTempPtr(IterVal) + if isinstance(IterVal.type, ir.PointerType) and isinstance(IterVal.type.pointee, ir.IntType) and IterVal.type.pointee.width == 8: + IterVal = Gen.builder.bitcast(IterVal, ir.PointerType(Gen.structs[ClassName]), name=f"zip_cast_{ClassName}") + IterCall: ir.Function = Gen._get_function(f'{ClassName}.__iter__') + IterResult: ir.Value = Gen.builder.call(IterCall, [IterVal], name=f"zip_call_{ClassName}.__iter__") + Gen._UnregisterTempPtr(IterResult) + if isinstance(IterResult.type, ir.PointerType) and isinstance(IterResult.type.pointee, ir.IntType) and IterResult.type.pointee.width == 8: + IterResult = Gen.builder.bitcast(IterResult, ir.PointerType(Gen.structs[ClassName]), name=f"zip_cast_iter_{ClassName}") + NextCall: ir.Function = Gen._get_function(f'{ClassName}.__next__') + StopFlagPtr: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(1), name=f"zip_stop_{target_name}") + Gen.builder.store(ir.Constant(ir.IntType(1), 0), StopFlagPtr) + NextReturnType: ir.Type = NextCall.function_type.return_type + LoopVar: ir.AllocaInstr = Gen._allocaEntry(NextReturnType, name=target_name) + Gen.variables[target_name] = LoopVar + return { + 'iter_type': 'iter', + 'target_name': target_name, + 'value_type': NextReturnType, + 'iter_obj': IterResult, + 'next_call': NextCall, + 'stop_flag': StopFlagPtr, + 'loop_var': LoopVar, + } + + def _zip_setup_str(self, StrVal: ir.Value, target_name: str, Gen: "Translator.LlvmGen") -> dict: + Gen._UnregisterTempPtr(StrVal) + PtrVal: ir.AllocaInstr = Gen.builder.alloca(ir.IntType(8).as_pointer(), name=f"zip_{target_name}_ptr") + Gen.builder.store(StrVal, PtrVal) + CharType: ir.IntType = ir.IntType(8) + LoopVar: ir.AllocaInstr = Gen._allocaEntry(CharType, name=target_name) + Gen.variables[target_name] = LoopVar + Gen._store(ir.Constant(CharType, 0), LoopVar) + Gen._record_var_signedness(target_name, 'char') + return { + 'iter_type': 'str', + 'target_name': target_name, + 'value_type': CharType, + 'ptr_var': PtrVal, + 'loop_var': LoopVar, + } + + def _zip_setup_array(self, ArrPtr: ir.Value, target_name: str, Gen: "Translator.LlvmGen") -> dict: + ElemType: ir.Type = ArrPtr.type.pointee.element + ArrayCount: int = ArrPtr.type.pointee.count + IdxVar: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name=f"zip_{target_name}_idx") + Gen.builder.store(ir.Constant(ir.IntType(32), 0), IdxVar) + LoopVar: ir.AllocaInstr = Gen._allocaEntry(ElemType, name=target_name) + Gen.variables[target_name] = LoopVar + Gen._store(ir.Constant(ElemType, None), LoopVar) + Gen._record_var_signedness(target_name, 'ptr') + return { + 'iter_type': 'array', + 'target_name': target_name, + 'value_type': ElemType, + 'arr_ptr': ArrPtr, + 'idx_var': IdxVar, + 'count_val': ir.Constant(ir.IntType(32), ArrayCount), + 'loop_var': LoopVar, + } + + def _zip_setup_ptr(self, PtrVal: ir.Value, target_name: str, ElemType: ir.IntType, Gen: "Translator.LlvmGen") -> dict: + TypedPtr: ir.Value = PtrVal + if isinstance(PtrVal.type, ir.PointerType) and PtrVal.type.pointee is not ElemType: + TypedPtr = Gen.builder.bitcast(PtrVal, ir.PointerType(ElemType), name="zip_ptr_cast") + Gen._UnregisterTempPtr(TypedPtr) + PtrCopy: ir.AllocaInstr = Gen.builder.alloca(ir.PointerType(ElemType), name=f"zip_{target_name}_ptr") + Gen.builder.store(TypedPtr, PtrCopy) + LoopVar: ir.AllocaInstr = Gen._allocaEntry(ElemType, name=target_name) + Gen.variables[target_name] = LoopVar + Gen._store(ir.Constant(ElemType, 0), LoopVar) + Gen._record_var_signedness(target_name, 'int') + return { + 'iter_type': 'ptr', + 'target_name': target_name, + 'value_type': ElemType, + 'ptr_var': PtrCopy, + 'loop_var': LoopVar, + } + + def _zip_setup_int(self, IterVal: ir.Value, target_name: str, Gen: "Translator.LlvmGen") -> dict: + EndVal: ir.Value = IterVal + if not isinstance(EndVal.type, ir.IntType): + EndVal = Gen.builder.ptrtoint(EndVal, ir.IntType(64), name="zip_int_end") + StartVal: ir.Value = ir.Constant(EndVal.type, 0) + StepVal: ir.Value = ir.Constant(EndVal.type, 1) + IdxVar: ir.AllocaInstr = Gen._allocaEntry(EndVal.type, name=f"zip_{target_name}_idx") + Gen._store(StartVal, IdxVar) + LoopVar: ir.AllocaInstr = Gen._allocaEntry(EndVal.type, name=target_name) + Gen.variables[target_name] = LoopVar + Gen._store(StartVal, LoopVar) + Gen._record_var_signedness(target_name, 'int') + return { + 'iter_type': 'range', + 'target_name': target_name, + 'value_type': EndVal.type, + 'idx_var': IdxVar, + 'end_val': EndVal, + 'step_val': StepVal, + 'loop_var': LoopVar, + } + + def _zip_get_next(self, info: dict, Gen: "Translator.LlvmGen") -> tuple[ir.Value, ir.Value]: + """生成获取下一个值和停止条件的 IR。返回 (value, stopped_cond)。""" + iter_type: str = info['iter_type'] + + if iter_type == 'range': + idx: ir.Value = Gen._load(info['idx_var'], name="zip_range_idx") + stopped: ir.Value = Gen.builder.icmp_signed('>=', idx, info['end_val'], name="zip_range_stop") + next_idx: ir.Value = Gen.builder.add(idx, info['step_val'], name="zip_range_next") + Gen._store(next_idx, info['idx_var']) + return idx, stopped + + if iter_type == 'iter': + Gen.builder.store(ir.Constant(ir.IntType(1), 0), info['stop_flag']) + args: list[ir.Value] = [info['iter_obj'], info['stop_flag']] + NextCall: ir.Function = info['next_call'] + if NextCall and len(NextCall.function_type.args) > len(args): + last_param_type: ir.Type = NextCall.function_type.args[-1] + if (isinstance(last_param_type, ir.PointerType) + and isinstance(last_param_type.pointee, ir.PointerType) + and isinstance(last_param_type.pointee.pointee, ir.IntType) + and last_param_type.pointee.pointee.width == 8): + null_msg: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name="zip_eh_msg") + args.append(null_msg) + val: ir.Value = Gen.builder.call(NextCall, args, name="zip_iter_next") + stopped = Gen.builder.load(info['stop_flag'], name="zip_iter_stop") + return val, stopped + + if iter_type == 'str': + ptr: ir.Value = Gen._load(info['ptr_var'], name="zip_str_ptr") + val = Gen._load(ptr, name="zip_str_val") + stopped = Gen.builder.icmp_signed('==', val, ir.Constant(ir.IntType(8), 0), name="zip_str_stop") + next_ptr: ir.Value = Gen.builder.gep(ptr, [ir.Constant(ir.IntType(32), 1)], name="zip_str_next") + Gen._store(next_ptr, info['ptr_var']) + return val, stopped + + if iter_type == 'ptr': + ptr = Gen._load(info['ptr_var'], name="zip_ptr_cur") + val = Gen._load(ptr, name="zip_ptr_val") + elem_type: ir.IntType = info['value_type'] + stopped = Gen.builder.icmp_signed('==', val, ir.Constant(elem_type, 0), name="zip_ptr_stop") + next_ptr = Gen.builder.gep(ptr, [ir.Constant(ir.IntType(32), 1)], name="zip_ptr_next") + Gen._store(next_ptr, info['ptr_var']) + return val, stopped + + if iter_type == 'array': + idx = Gen._load(info['idx_var'], name="zip_arr_idx") + stopped = Gen.builder.icmp_signed('>=', idx, info['count_val'], name="zip_arr_stop") + elem_ptr: ir.Value = Gen.builder.gep(info['arr_ptr'], [ir.Constant(ir.IntType(32), 0), idx], name="zip_arr_elem") + val = Gen._load(elem_ptr, name="zip_arr_val") + next_idx = Gen.builder.add(idx, ir.Constant(ir.IntType(32), 1), name="zip_arr_next") + Gen._store(next_idx, info['idx_var']) + return val, stopped + + # 回退：立即停止 + return ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(1), 1) + + # ===== enumerate() 内置支持 ===== + + def _HandleForEnumerateLlvm(self, Node: ast.For) -> None: + """for idx, val in enumerate(iterable): — 带索引迭代 + + 等价于 zip(range(0, ∞), iterable)，但无需 len()。 + idx 从 0 开始，每次迭代 +1。 + """ + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not isinstance(Node.target, ast.Tuple): + return + Targets: list[ast.Name] = [t for t in Node.target.elts if isinstance(t, ast.Name)] + if len(Targets) != 2: + return + IdxTargetName: str = Targets[0].id + ValTargetName: str = Targets[1].id + IterArg: ast.expr = Node.iter.args[0] + + # 设置内部可迭代对象（复用 zip 基础设施） + if ValTargetName in Gen._reg_values: + del Gen._reg_values[ValTargetName] + if IdxTargetName in Gen._reg_values: + del Gen._reg_values[IdxTargetName] + IterInfo: dict | None = self._zip_setup_iter(IterArg, ValTargetName, Gen) + if IterInfo is None: + return + + # 索引计数器（i32，从 0 开始） + IdxType: ir.IntType = ir.IntType(32) + IdxVar: ir.AllocaInstr = Gen._allocaEntry(IdxType, name=f"enum_{IdxTargetName}") + Gen._store(ir.Constant(IdxType, 0), IdxVar) + IdxLoopVar: ir.AllocaInstr = Gen._allocaEntry(IdxType, name=IdxTargetName) + Gen.variables[IdxTargetName] = IdxLoopVar + Gen._store(ir.Constant(IdxType, 0), IdxLoopVar) + Gen._record_var_signedness(IdxTargetName, 'int') + + CondBB: ir.Block = Gen.func.append_basic_block(name="enum.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="enum.body") + StepBB: ir.Block = Gen.func.append_basic_block(name="enum.step") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="enum.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="enum.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(StepBB) + Gen.builder.branch(CondBB) + + # Cond 块：获取下一个值和停止条件 + Gen.builder.position_at_start(CondBB) + CurIdx: ir.Value = Gen._load(IdxVar, name="enum_idx") + Val, Stopped = self._zip_get_next(IterInfo, Gen) + IterInfo['last_value'] = Val + Gen.builder.cbranch(Stopped, ElseBB if ElseBB else AfterBB, BodyBB) + + # Body 块：赋值并执行循环体 + Gen.builder.position_at_start(BodyBB) + Gen._store(CurIdx, IdxLoopVar) + Gen._store(Val, IterInfo['loop_var']) + Gen._reg_values[IdxTargetName] = CurIdx + Gen._reg_values[ValTargetName] = Val + self.HandleBodyLlvm(Node.body) + if IdxTargetName in Gen._reg_values and Gen._reg_values[IdxTargetName] is CurIdx: + del Gen._reg_values[IdxTargetName] + if ValTargetName in Gen._reg_values and Gen._reg_values[ValTargetName] is Val: + del Gen._reg_values[ValTargetName] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(StepBB) + + # Step 块：索引 +1，跳回 Cond + Gen.builder.position_at_start(StepBB) + CurIdx2: ir.Value = Gen._load(IdxVar, name="enum_idx2") + NextIdx: ir.Value = Gen.builder.add(CurIdx2, ir.Constant(IdxType, 1), name="enum_next") + Gen._store(NextIdx, IdxVar) + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesFunctions.py b/lib/core/Handles/HandlesFunctions.py new file mode 100644 index 0000000..b64f19c --- /dev/null +++ b/lib/core/Handles/HandlesFunctions.py @@ -0,0 +1,1852 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +import ast +import copy +import traceback +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo, FuncMeta +from lib.core.SymbolUtils import IsTModule, FindStructNameInAnnotation +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.includes.t import CTypeRegistry + + +# t 模块存储类标记装饰器名称集合 +_T_STORAGE_DECORATORS = frozenset({'CExport', 'CExtern', 'CStatic', 'State', 'CInline', 'TLS'}) + + +class FunctionHandle(BaseHandle): + @staticmethod + def _ExtractFuncMeta(decorator_list: list[ast.AST]) -> FuncMeta: + meta: FuncMeta = FuncMeta.NONE + if not decorator_list: + return meta + for d in decorator_list: + if isinstance(d, ast.Name): + if d.id == 'staticmethod': + meta |= FuncMeta.STATIC_METHOD + elif d.id == 'property': + meta |= FuncMeta.PROPERTY_GETTER + elif d.id == 'classmethod': + meta |= FuncMeta.CLASS_METHOD + elif isinstance(d, ast.Attribute): + if isinstance(d.value, ast.Name): + # 支持 @property.setter 和 @propname.setter 两种形式 + # @propname.setter 中 propname 是之前用 @property 定义的同名属性 + if d.value.id == 'property' or d.attr in ('setter', 'getter', 'deleter'): + if d.attr == 'setter': + meta |= FuncMeta.PROPERTY_SETTER + elif d.attr == 'getter': + meta |= FuncMeta.PROPERTY_GETTER + elif d.attr == 'deleter': + meta |= FuncMeta.PROPERTY_DELETER + return meta + + def _extract_decorator_storage_flags(self, Node: ast.FunctionDef) -> dict[str, bool]: + """从装饰器列表中提取 t 模块存储类标记 + + 支持 @t.CExport, @t.CExtern, @t.CStatic, @t.State, @t.CInline + 也支持 from t import CExport 等别名形式 + + Returns: + dict with keys: 'is_export', 'is_extern', 'is_static', 'is_state', 'is_inline' + """ + flags: dict[str, bool] = { + 'is_export': False, 'is_extern': False, + 'is_static': False, 'is_state': False, 'is_inline': False, + } + if not Node.decorator_list: + return flags + symtab: object = self.Trans.SymbolTable + for decorator in Node.decorator_list: + attr_name: str | None = None + # @t.CExport 形式 + if isinstance(decorator, ast.Attribute) and isinstance(decorator.value, ast.Name): + if IsTModule(decorator.value.id, symtab): + attr_name = decorator.attr + # @CExport 形式 (from t import CExport) + elif isinstance(decorator, ast.Name): + cls: type | None = getattr(t, decorator.id, None) + if cls is None and hasattr(symtab, '_t_type_symbols'): + cls = symtab._t_type_symbols.get(decorator.id) + if cls is not None and isinstance(cls, type) and issubclass(cls, t.CType): + attr_name = cls.__name__ + if attr_name and attr_name in _T_STORAGE_DECORATORS: + if attr_name == 'CExport': + flags['is_export'] = True + elif attr_name == 'CExtern': + flags['is_extern'] = True + elif attr_name == 'CStatic': + flags['is_static'] = True + elif attr_name == 'State': + flags['is_state'] = True + elif attr_name == 'CInline': + flags['is_inline'] = True + return flags + + @staticmethod + def _is_t_storage_decorator(decorator: ast.AST, symtab: object) -> bool: + """检查装饰器是否是 t 模块存储类标记（用于排除 behavior_decorators）""" + attr_name: str | None = None + if isinstance(decorator, ast.Attribute) and isinstance(decorator.value, ast.Name): + if IsTModule(decorator.value.id, symtab): + attr_name = decorator.attr + elif isinstance(decorator, ast.Name): + cls: type | None = getattr(t, decorator.id, None) + if cls is None and hasattr(symtab, '_t_type_symbols'): + cls = symtab._t_type_symbols.get(decorator.id) + if cls is not None and isinstance(cls, type) and issubclass(cls, t.CType): + attr_name = cls.__name__ + elif decorator.id in _T_STORAGE_DECORATORS and hasattr(t, decorator.id): + # TLS 是普通函数标记，不是 CType 子类 + attr_name = decorator.id + return attr_name in _T_STORAGE_DECORATORS if attr_name else False + + # ------------------------------------------------------------------ + # 函数签名公共辅助：以下三个方法供 _GetFunctionSignatureLlvm 与 + # _EmitFunctionLlvm 共享，避免 ~80 行重复代码。 + # ------------------------------------------------------------------ + + def _compute_func_name_meta(self, Node: ast.FunctionDef, ClassName: str | None) -> tuple[str, str, FuncMeta]: + """计算函数全名（含 setter/deleter 后缀）和 func_meta + + Returns: + (FuncName, RawFuncName, func_meta) + """ + RawFuncName: str = Node.name + FuncName: str = f"{ClassName}.{RawFuncName}" if ClassName else RawFuncName + func_meta: FuncMeta = self._ExtractFuncMeta(Node.decorator_list) + # property setter/deleter 使用不同的函数名后缀，避免与 getter 冲突 + if FuncMeta.PROPERTY_SETTER in func_meta: + FuncName = FuncName + '$set' + elif FuncMeta.PROPERTY_DELETER in func_meta: + FuncName = FuncName + '$del' + return FuncName, RawFuncName, func_meta + + def _resolve_method_class(self, Node: ast.FunctionDef, ClassName: str | None, func_meta: FuncMeta, FuncName: str, Gen: LlvmCodeGenerator) -> tuple[bool, bool, str | None]: + """判定函数是否是实例方法 / 类方法，并解析所属类名 + + Returns: + (IsMethod, IsClassMethod, ResolvedClassName) + """ + RawFuncName: str = Node.name + IsMethod: bool = False + IsClassMethod: bool = False + ResolvedClassName: str | None = ClassName + if not ResolvedClassName: + for potential_class in Gen.class_methods: + if FuncName.startswith(f"{potential_class}."): + IsMethod = True + ResolvedClassName = potential_class + break + else: + IsMethod = FuncMeta.STATIC_METHOD not in func_meta and FuncMeta.CLASS_METHOD not in func_meta + IsClassMethod = FuncMeta.CLASS_METHOD in func_meta + if RawFuncName == '__init__': + IsMethod = False + return IsMethod, IsClassMethod, ResolvedClassName + + def _apply_struct_self_cls_types(self, ParamTypes: list[ir.Type], ParamNames: list[str], IsMethod: bool, IsClassMethod: bool, ResolvedClassName: str | None, Gen: LlvmCodeGenerator) -> bool: + """根据 IsMethod / IsClassMethod 调整 self/cls 参数的 LLVM 类型 + + - 若是实例方法且无 self 参数，自动插入一个 struct 指针类型； + 此时调用方需要同步插入对应的 ParamTypeStrs（本方法不维护）。 + - 若是 @classmethod，将 cls 参数覆盖为 struct 指针类型。 + + Returns: + True 表示为实例方法自动插入了 self 参数；False 表示未插入。 + """ + if not ResolvedClassName or ResolvedClassName not in Gen.structs: + return False + # 治本修复：同名类 sha1 冲突检测。 + # 当多个模块定义同名类（如 stmts.py:Module 和 __module.py:Module）时， + # Gen.structs[short_name] 可能指向错误模块的 struct（set_body 已锁定错误布局）。 + # 通过 class_sha1_map 获取当前类的源模块 sha1，查找以 sha1 为前缀的完整 struct。 + TargetStruct: ir.IdentifiedStructType = Gen.structs[ResolvedClassName] + ClassSha1Map: dict[str, str] = getattr(Gen, 'class_sha1_map', {}) + ExpectedSha1: str | None = ClassSha1Map.get(ResolvedClassName) + if ExpectedSha1 and isinstance(TargetStruct, ir.IdentifiedStructType): + ExistingSha1: str = Gen._extract_struct_sha1(TargetStruct) + if ExistingSha1 and ExistingSha1 != ExpectedSha1: + # sha1 冲突：查找或创建正确的 struct + FullKey: str = f"{ExpectedSha1}.{ResolvedClassName}" + if FullKey in Gen.structs: + TargetStruct = Gen.structs[FullKey] + else: + # 创建新 struct 并从 stub 加载 body + TargetStruct = Gen._get_or_create_struct(ResolvedClassName, source_sha1=ExpectedSha1) + if hasattr(self.Trans, 'ImportHandler'): + self.Trans.ImportHandler._TryLoadStructFromStub(ResolvedClassName, Gen, source_sha1=ExpectedSha1) + if FullKey in Gen.structs: + TargetStruct = Gen.structs[FullKey] + # 安全回退：如果新 struct 仍为 opaque（body 未加载），回退到原始短名 struct， + # 避免方法编译生成无效 GEP（访问 opaque struct 元素）导致运行时崩溃。 + if isinstance(TargetStruct, ir.IdentifiedStructType): + try: + _is_opaque: bool = TargetStruct.is_opaque + except (AssertionError, AttributeError): + _is_opaque = False + if _is_opaque: + TargetStruct = Gen.structs[ResolvedClassName] + if IsMethod: + StructPtrType: ir.PointerType = ir.PointerType(TargetStruct) + SelfIdx: int = next((i for i, n in enumerate(ParamNames) if n == "self"), -1) + if SelfIdx >= 0: + ParamTypes[SelfIdx] = StructPtrType + return False + ParamTypes.insert(0, StructPtrType) + ParamNames.insert(0, "self") + return True + # 非方法但属于类（如 staticmethod / 顶层函数位于类下） + SelfIdx = next((i for i, n in enumerate(ParamNames) if n == "self"), -1) + if SelfIdx >= 0: + StructPtrType = ir.PointerType(TargetStruct) + ParamTypes[SelfIdx] = StructPtrType + if IsClassMethod: + ClsIdx: int = next((i for i, n in enumerate(ParamNames) if n == "cls"), -1) + if ClsIdx >= 0: + StructPtrType = ir.PointerType(TargetStruct) + ParamTypes[ClsIdx] = StructPtrType + return False + + def _body_contains_raise(self, body: list[ast.AST]) -> bool: + for node in ast.walk(ast.Module(body=body, type_ignores=[])): + if isinstance(node, ast.Raise): + if node.exc: + if isinstance(node.exc, ast.Name) and node.exc.id == 'StopIteration': + continue + if isinstance(node.exc, ast.Call) and isinstance(node.exc.func, ast.Name) and node.exc.func.id == 'StopIteration': + continue + return True + if isinstance(node, ast.Call): + called_name: str | None = None + if isinstance(node.func, ast.Name): + called_name = node.func.id + elif isinstance(node.func, ast.Attribute): + called_name = node.func.attr + if called_name: + Gen: LlvmCodeGenerator = self.Trans.LlvmGen + for prefix in ['', f'{called_name}.', f'__']: + for suffix in ['', f'.{called_name}']: + fname: str = f'{prefix}{called_name}{suffix}' + if fname in Gen.functions: + func: Any = Gen.functions[fname] + if len(func.args) > 0: + for arg in func.args: + if hasattr(arg, 'name') and arg.name in ('__eh_msg_out__', '__eh_code_out__'): + return True + break + return False + + _LLVM_ATTR_NAMES = frozenset({ + 'nobuiltin', 'nounwind', 'noredzone', 'willreturn', 'mustprogress', + 'optnone', 'noinline', 'alwaysinline', 'readnone', 'readonly', + 'writeonly', 'inaccessiblememonly', 'inaccessiblemem_or_argmemonly', + }) + + def _ExtractLLVMAttrs(self, node: ast.AST) -> list[str]: + attrs: list[str] = [] + if isinstance(node, ast.BinOp): + attrs.extend(self._ExtractLLVMAttrs(node.left)) + attrs.extend(self._ExtractLLVMAttrs(node.right)) + elif isinstance(node, ast.Attribute): + attr_name: str | None = self._GetAttrLLVMName(node) + if attr_name: + attrs.append(attr_name) + return attrs + + def _GetAttrLLVMName(self, node: ast.AST) -> str | None: + if not isinstance(node, ast.Attribute): + return None + if node.attr not in self._LLVM_ATTR_NAMES: + return None + value: ast.AST = node.value + if not isinstance(value, ast.Attribute) or value.attr != 'llvm': + return None + inner: ast.AST = value.value + if not isinstance(inner, ast.Attribute) or inner.attr != 'attr': + return None + if not isinstance(inner.value, ast.Name) or inner.value.id != 't': + return None + return node.attr + + def _infer_return_type_from_body(self, Node: ast.AST, Gen: LlvmCodeGenerator | None = None, ClassName: str | None = None) -> tuple[str, bool] | None: + local_var_types: dict[str, tuple[str, bool]] = {} + for child in ast.walk(Node): + if isinstance(child, ast.Assign) and len(child.targets) == 1: + target: ast.AST = child.targets[0] + if isinstance(target, ast.Name) and isinstance(child.value, ast.Subscript): + sub: ast.Subscript = child.value + if (isinstance(sub.value, ast.Attribute) + and isinstance(sub.value.value, ast.Name) + and sub.value.value.id == 'self' + and ClassName and ClassName in Gen.class_members): + member_name: str = sub.value.attr + for m_name, m_type in Gen.class_members[ClassName]: + if m_name == member_name: + if isinstance(m_type, ir.ArrayType): + elem_type: ir.Type = m_type.element + if isinstance(elem_type, ir.FloatType): + local_var_types[target.id] = ('float', False) + elif isinstance(elem_type, ir.DoubleType): + local_var_types[target.id] = ('double', False) + elif isinstance(elem_type, ir.IntType): + if elem_type.width == 8: + local_var_types[target.id] = ('char', False) + else: + local_var_types[target.id] = ('int', False) + elif isinstance(elem_type, ir.PointerType): + if isinstance(elem_type.pointee, ir.IntType) and elem_type.pointee.width == 8: + local_var_types[target.id] = ('char', True) + else: + local_var_types[target.id] = ('void', True) + elif isinstance(elem_type, ir.IdentifiedStructType): + struct_name: str = elem_type.name + if struct_name: + for sn, st in Gen.structs.items(): + if st.name == struct_name or sn == struct_name: + local_var_types[target.id] = (sn, False) + break + else: + local_var_types[target.id] = (elem_type.name, False) + elif isinstance(m_type, ir.PointerType): + pointee: ir.Type = m_type.pointee + if isinstance(pointee, ir.FloatType): + local_var_types[target.id] = ('float', False) + elif isinstance(pointee, ir.DoubleType): + local_var_types[target.id] = ('double', False) + elif isinstance(pointee, ir.IntType): + if pointee.width == 8: + local_var_types[target.id] = ('char', False) + else: + local_var_types[target.id] = ('int', False) + elif isinstance(pointee, ir.PointerType): + local_var_types[target.id] = ('void', True) + elif isinstance(pointee, ir.IdentifiedStructType): + for sn, st in Gen.structs.items(): + if st == pointee or st.name == pointee.name: + local_var_types[target.id] = (sn, True) + break + break + for child in ast.walk(Node): + if isinstance(child, ast.Return) and child.value: + val: ast.AST = child.value + if isinstance(val, ast.Name) and val.id == 'self' and ClassName: + return (ClassName, True) + if isinstance(val, ast.Call): + if isinstance(val.func, ast.Name): + callee: str = val.func.id + if callee == 'malloc': + return ('char', True) + if callee == 'chr': + return ('char', False) + if Gen and callee in Gen.functions: + func: Any = Gen.functions[callee] + if hasattr(func, 'ftype'): + ret_type: ir.Type = func.ftype.return_type + if isinstance(ret_type, ir.PointerType): + if isinstance(ret_type.pointee, ir.IntType) and ret_type.pointee.width == 8: + return ('char', True) + return ('void', True) + elif isinstance(ret_type, ir.IntType): + if ret_type.width == 8: + return ('char', False) + return ('int', False) + elif isinstance(ret_type, ir.VoidType): + return ('void', False) + elif isinstance(val.func, ast.Attribute): + pass + callee_name: str | None = None + if isinstance(val.func, ast.Name): + callee_name = val.func.id + elif isinstance(val.func, ast.Attribute): + callee_name = getattr(val.func, 'attr', None) + if callee_name: + sym: Any = self.Trans.SymbolTable.lookup(callee_name) + if sym and isinstance(sym, dict): + ret_type_str: str = sym.get('return_type', '') + if ret_type_str and ret_type_str != 'int': + IsPtr: bool = sym.get('IsPtr', False) + return (ret_type_str, IsPtr) + elif isinstance(val, ast.Constant): + if val.value is None: + return ('char', True) + elif isinstance(val.value, str) and len(val.value) <= 1: + return ('char', True) + elif isinstance(val, ast.Name): + var_name: str = val.id + if var_name in local_var_types: + return local_var_types[var_name] + var_info: Any = self.Trans.SymbolTable.lookup(var_name) + if var_info and isinstance(var_info, dict): + var_type: str = var_info.get('type', '') + if var_type and var_type != 'int': + IsPtr: bool = var_info.get('IsPtr', False) + return (var_type, IsPtr) + elif isinstance(val, ast.Subscript): + if (isinstance(val.value, ast.Attribute) + and isinstance(val.value.value, ast.Name) + and val.value.value.id == 'self' + and ClassName): + member_name: str = val.value.attr + if ClassName in Gen.class_members: + for m_name, m_type in Gen.class_members[ClassName]: + if m_name == member_name: + if isinstance(m_type, ir.ArrayType): + elem_type: ir.Type = m_type.element + if isinstance(elem_type, ir.FloatType): + return ('float', False) + elif isinstance(elem_type, ir.DoubleType): + return ('double', False) + elif isinstance(elem_type, ir.IntType): + if elem_type.width == 8: + return ('char', False) + return ('int', False) + elif isinstance(elem_type, ir.PointerType): + if isinstance(elem_type.pointee, ir.IntType) and elem_type.pointee.width == 8: + return ('char', True) + return ('void', True) + elif isinstance(elem_type, ir.IdentifiedStructType): + struct_name: str = elem_type.name + if struct_name: + for sn, st in Gen.structs.items(): + if st.name == struct_name or sn == struct_name: + return (sn, False) + return (elem_type.name, False) + elif isinstance(m_type, ir.PointerType): + pointee: ir.Type = m_type.pointee + if isinstance(pointee, ir.FloatType): + return ('float', False) + elif isinstance(pointee, ir.DoubleType): + return ('double', False) + elif isinstance(pointee, ir.IntType): + if pointee.width == 8: + return ('char', False) + return ('int', False) + elif isinstance(pointee, ir.PointerType): + return ('void', True) + elif isinstance(pointee, ir.IdentifiedStructType): + for sn, st in Gen.structs.items(): + if st == pointee or st.name == pointee.name: + return (sn, True) + return (pointee.name, True) + break + if isinstance(val.value, ast.Name): + var_name: str = val.value.id + var_info: Any = self.Trans.SymbolTable.lookup(var_name) + if var_info and isinstance(var_info, dict): + var_type: str = var_info.get('type', '') + if var_type.startswith('list[') or var_type.startswith('List[') or var_type.startswith('t.CArray['): + inner: str = var_type[var_type.index('[') + 1:var_type.rindex(']')] + parts: list[str] = inner.split(',') + elem_type_str: str = parts[0].strip() + if elem_type_str and elem_type_str != 'int': + IsPtr: bool = var_info.get('IsPtr', False) + return (elem_type_str, IsPtr) + return None + + def _GetFunctionSignatureLlvm(self, Node: ast.FunctionDef, Gen: LlvmCodeGenerator, ClassName: str | None = None, extra_params: list[tuple[str, Any]] | None = None) -> tuple[str, ir.FunctionType, Any, list[Any], list[ast.AST], bool, str | None, bool]: + FuncName, RawFuncName, func_meta = self._compute_func_name_meta(Node, ClassName) + CReturnTypes: list[ast.AST] = [] + if Node.decorator_list: + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Attribute): + if decorator.func.attr == 'CReturn': + for arg in decorator.args: + CReturnTypes.append(arg) + if Node.returns: + llvm_attrs: list[str] = self._ExtractLLVMAttrs(Node.returns) + if llvm_attrs: + if not hasattr(self, '_pending_llvm_attrs'): + self._pending_llvm_attrs = {} + self._pending_llvm_attrs[FuncName] = llvm_attrs + if isinstance(Node.returns, ast.Subscript) and isinstance(Node.returns.value, ast.Name) and Node.returns.value.id == 'tuple': + slice_node: ast.AST = Node.returns.slice + if isinstance(slice_node, ast.Tuple): + for elt in slice_node.elts: + CReturnTypes.append(elt) + else: + CReturnTypes.append(slice_node) + IsPtr: bool = False + ReturnTypeInfo: Any = None + if Node.returns: + if CReturnTypes: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + else: + try: + if isinstance(Node.returns, ast.BinOp) and isinstance(Node.returns.op, ast.BitOr): + ReturnTypeInfo = getattr(self.Trans, 'TypeMergeHandler', None) and self.Trans.TypeMergeHandler.MergeTypes(Node.returns) + if not isinstance(ReturnTypeInfo, CTypeInfo): + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + if (ReturnTypeInfo.IsVoid and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState) or ReturnTypeInfo.PtrCount >= 2: + def _FindNonVoidCTypeInfo(node: ast.AST) -> Any: + if isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + results: list[Any] = [] + for child in [node.left, node.right]: + r: Any = _FindNonVoidCTypeInfo(child) + if r and not r.IsVoid: + results.append(r) + for r in results: + if not r.IsPtr and not r.IsStruct: + return r + return results[0] if results else None + try: + SideInfo: Any = CTypeInfo.FromNode(node, self.Trans.SymbolTable) + if SideInfo: + return SideInfo + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + return None + found: Any = _FindNonVoidCTypeInfo(Node.returns) + if found and not found.IsVoid: + ReturnTypeInfo = found + if ReturnTypeInfo.IsVoid and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + else: + is_str_type: bool = False + if isinstance(Node.returns, ast.Name) and Node.returns.id in ('str', 'bytes'): + is_str_type = True + ReturnTypeInfo = CTypeInfo.FromNode(Node.returns, self.Trans.SymbolTable) + if ReturnTypeInfo and ReturnTypeInfo.IsDefine: + inferred: Any = self._infer_return_type_from_body(Node, Gen, ClassName) + if inferred: + ReturnTypeInfo = CTypeInfo.FromTypeName(inferred[0]) + ReturnTypeInfo.PtrCount = 1 if inferred[1] else 0 + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + if not hasattr(self, '_cdefine_funcs'): + self._cdefine_funcs = set() + self._cdefine_funcs.add(FuncName) + elif ReturnTypeInfo and ReturnTypeInfo.IsState: + if ReturnTypeInfo.PtrCount > 0 or (ReturnTypeInfo.BaseType and not isinstance(ReturnTypeInfo.BaseType, t.CVoid)): + pass + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + if is_str_type or ReturnTypeInfo.IsStr: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + # 检查泛型类返回类型 (如 list[str]) + # 注意：_from_node_subscript 对未识别的泛型下标（如 list[str]）返回空 CTypeInfo（BaseType=None）， + # 此时 IsVoid 为 False（因为 IsVoid 仅检查 t.CVoid 实例），需额外检测 BaseType is None + if (isinstance(Node.returns, ast.Subscript) + and isinstance(Node.returns.value, ast.Name) + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and Node.returns.value.id in self.Trans.ClassHandler._generic_class_templates + and (ReturnTypeInfo.IsVoid or ReturnTypeInfo.BaseType is None) and ReturnTypeInfo.PtrCount == 0): + ret_class: str = Node.returns.value.id + ret_slice: ast.AST = Node.returns.slice + ret_type_args: list[str] = [] + ret_type_names: list[str] = [] + if isinstance(ret_slice, ast.Tuple): + for elt in ret_slice.elts: + if isinstance(elt, ast.Name): + tn2: str = elt.id + ts2: str = tn2 + if tn2 == 'int': + ts2, tn2 = 'int', 'CInt' + ret_type_args.append(ts2) + ret_type_names.append(tn2) + elif isinstance(ret_slice, ast.Name): + tn2: str = ret_slice.id + ts2: str = tn2 + if tn2 == 'int': + ts2, tn2 = 'int', 'CInt' + ret_type_args.append(ts2) + ret_type_names.append(tn2) + # 返回类型解析只需 struct 定义，不需方法体；使用 declare_only=True + # 避免过早触发完整特化（此时 MemManager.alloc 可能尚未注册， + # 导致 __new__ 的 pool.alloc(48) 解析失败，IR 块未终止）。 + # 完整特化由后续类实例化（L3688/L3753）触发 declare_only=False。 + spec_name2: str | None = self.Trans.ClassHandler._specialize_generic_class( + ret_class, ret_type_args, Gen, type_names=ret_type_names, declare_only=True) + if spec_name2 and spec_name2 in Gen.structs: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CStruct(name=spec_name2) + ReturnTypeInfo.PtrCount = 1 + if (ReturnTypeInfo.IsVoid or ReturnTypeInfo.BaseType is None) and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + except Exception: # 回退：设置默认返回类型为 CInt + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + inferred: Any = self._infer_return_type_from_body(Node, Gen, ClassName) + if inferred: + ReturnTypeInfo = CTypeInfo.FromTypeName(inferred[0]) + ReturnTypeInfo.PtrCount = 1 if inferred[1] else 0 + IsPtr = ReturnTypeInfo.IsPtr if ReturnTypeInfo else False + ReturnType: ir.Type = Gen._ctype_to_llvm(ReturnTypeInfo) + if isinstance(ReturnType, ir.PointerType) and isinstance(ReturnType.pointee, ir.PointerType): + if isinstance(ReturnType.pointee.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + ReturnType = ir.PointerType(ReturnType.pointee.pointee) + if isinstance(ReturnType, ir.IntType) and ReturnType.width == 8: + if RawFuncName == '__str__': + ReturnType = ir.PointerType(ir.IntType(8)) + if isinstance(ReturnType, ir.VoidType) and FuncName == 'main': + ReturnType = ir.IntType(32) + IsMethod, IsClassMethod, ResolvedClassName = self._resolve_method_class(Node, ClassName, func_meta, FuncName, Gen) + # __new__ methods should return a pointer to the struct (for heap allocation replacement) + # 修复：前向声明也必须使用正确的返回类型，否则 _replace_function_decl 可能失败， + # 导致 vtable 引用错误的函数类型（i32 而非 ClassName*） + if RawFuncName == '__new__' and ResolvedClassName: + StructType = Gen.structs.get(ResolvedClassName) + if StructType: + if isinstance(StructType, ir.PointerType): + ReturnType = StructType + else: + ReturnType = ir.PointerType(StructType) + ParamTypes: list[ir.Type] = [] + ParamNames: list[str] = [] + ParamTypeStrs: list[Any] = [] + auto_addr_modes: list[str | None] = [] + CReturnLlvmTypes: list[ir.Type] = [] + if CReturnTypes: + for i, ReturnTypeNode in enumerate(CReturnTypes): + ReturnTypeInfo = CTypeInfo.FromNode(ReturnTypeNode, self.Trans.SymbolTable) + if ReturnTypeInfo and ReturnTypeInfo.IsStr: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + RetType: ir.Type = Gen._ctype_to_llvm(ReturnTypeInfo) + CReturnLlvmTypes.append(RetType) + ReturnType = ir.LiteralStructType(CReturnLlvmTypes) + for Arg in Node.args.args: + ParamIsUnsigned: bool = False + if Arg.annotation: + try: + ParamTypeInfo: Any + ParamTypeInfo = self.ResolveAnnotationType(Arg.annotation) + if ParamTypeInfo is None: + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CInt() + IsPtr = ParamTypeInfo.IsPtr + if ParamTypeInfo and getattr(ParamTypeInfo, 'IsCpythonObject', False) and not IsPtr: + IsPtr = True + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + if ParamTypeInfo.IsStr or (isinstance(Arg.annotation, ast.Name) and Arg.annotation.id in ('str', 'bytes')): + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CChar() + ParamTypeInfo.PtrCount = 1 + IsPtr = True + # C 语言中数组参数退化为指针：list[type, N] → type* + if ParamTypeInfo.ArrayDims: + ParamTypeInfo.ArrayDims = [] + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + ParamType: ir.Type = Gen._ctype_to_llvm(ParamTypeInfo) + ParamTypeStr: Any = ParamTypeInfo + except Exception as e: + ParamType = ir.IntType(32) + ParamTypeStr = CTypeInfo() + ParamTypeStr.BaseType = t.CInt() + else: + if ClassName and Arg.arg != 'self': + op_method_names: list[str] = ['__add__', '__sub__', '__mul__', '__truediv__', '__floordiv__', '__mod__', '__pow__', + '__radd__', '__rsub__', '__rmul__', '__rtruediv__', '__rfloordiv__', '__rmod__', '__rpow__', + '__iadd__', '__isub__', '__imul__', '__itruediv__', '__ifloordiv__', '__imod__', '__ipow__', + '__eq__', '__ne__', '__lt__', '__le__', '__gt__', '__ge__', + '__neg__', '__pos__', '__abs__', '__call__'] + if RawFuncName in op_method_names and ClassName in Gen.structs: + ParamType = ir.PointerType(Gen.structs[ClassName]) + else: + ParamType = ir.IntType(32) + else: + ParamType = ir.IntType(32) + ParamTypeStr = CTypeInfo() + ParamTypeStr.BaseType = t.CInt() + ParamTypes.append(ParamType) + ParamNames.append(Arg.arg) + ParamTypeStrs.append(ParamTypeStr) + # 检测 t.CNeedPtr / t.CAutoPtr 参数注解 + auto_mode: str | None = None + if ParamTypeStr is not None and hasattr(ParamTypeStr, 'BaseType') and ParamTypeStr.BaseType is not None: + if isinstance(ParamTypeStr.BaseType, t.CAutoPtr): + auto_mode = 'always' + elif isinstance(ParamTypeStr.BaseType, t.CNeedPtr): + auto_mode = 'need' + auto_addr_modes.append(auto_mode) + if self._apply_struct_self_cls_types(ParamTypes, ParamNames, IsMethod, IsClassMethod, ResolvedClassName, Gen): + ParamTypeStrs.insert(0, f"struct {ResolvedClassName}") + IsVariadic: bool = Node.args.vararg is not None + if extra_params: + for var_name, ptr_type in extra_params: + ParamTypes.append(ptr_type) + ParamNames.append(f'__nonlocal_{var_name}__') + if IsMethod and RawFuncName == '__next__': + ParamTypes.append(ir.PointerType(ir.IntType(1))) + ParamNames.append('__stop_iter_flag__') + HasRaise: bool = self._body_contains_raise(Node.body) + if HasRaise and RawFuncName != 'main': + ParamTypes.append(ir.PointerType(ir.PointerType(ir.IntType(8)))) + ParamNames.append('__eh_msg_out__') + ParamTypes.append(ir.PointerType(ir.IntType(32))) + ParamNames.append('__eh_code_out__') + FuncType: ir.FunctionType = ir.FunctionType(ReturnType, ParamTypes, var_arg=IsVariadic) + # 存储 auto_addr_modes（为额外参数补 None） + total_params: int = len(ParamTypes) + while len(auto_addr_modes) < total_params: + auto_addr_modes.append(None) + Gen._auto_addr_params[FuncName] = auto_addr_modes + return FuncName, FuncType, ReturnTypeInfo, ParamTypeStrs, CReturnTypes, IsMethod, ResolvedClassName, IsVariadic + + def _is_generic_function(self, Node: ast.FunctionDef) -> bool: + if hasattr(Node, 'type_params') and Node.type_params: + return True + return False + + def _mangle_generic_name(self, func_name: str, type_args: list[str], has_cexport: bool = False) -> str: + mangled_args: list[str] = [] + for ta in type_args: + llvm_str: str | None = CTypeRegistry.NameToLLVM(ta) + if llvm_str: + if llvm_str in ('float', 'double', 'half', 'fp128'): + mangled: str = 'f' + str(CTypeRegistry.GetClassByName(ta)().Size) + else: + mangled = llvm_str + else: + mangled = ta + mangled_args.append(mangled) + if has_cexport: + return func_name + '_T_' + '_'.join(mangled_args) + return func_name + '[' + ']['.join(mangled_args) + ']' + + def _apply_type_map_to_node(self, Node: ast.FunctionDef, type_map: dict[str, str], cptr_params: set[str] | None = None) -> None: + for arg in Node.args.args: + if arg.annotation: + arg.annotation = self._replace_type_in_annotation(arg.annotation, type_map, cptr_params) + if Node.returns: + Node.returns = self._replace_type_in_annotation(Node.returns, type_map, cptr_params) + self._apply_type_map_to_body(Node.body, type_map, cptr_params) + + def _replace_type_in_annotation(self, node: ast.AST, type_map: dict[str, str], cptr_params: set[str] | None = None) -> ast.AST: + if isinstance(node, ast.Name): + if node.id in type_map: + replacement: str = type_map[node.id] + return self._make_replacement_node(replacement) + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + if isinstance(node.left, ast.Name) and node.left.id in type_map: + replacement: str = type_map[node.left.id] + node.left = self._make_replacement_node(replacement) + return node + node.left = self._replace_type_in_annotation(node.left, type_map, cptr_params) + node.right = self._replace_type_in_annotation(node.right, type_map, cptr_params) + elif isinstance(node, ast.Attribute): + if hasattr(node, 'attr') and node.attr in type_map: + replacement: str = type_map[node.attr] + return self._make_replacement_node(replacement) + elif isinstance(node, ast.Subscript): + # 嵌套泛型（如 list[T]）：递归替换 value 和 slice 中的类型参数 + node.value = self._replace_type_in_annotation(node.value, type_map, cptr_params) + node.slice = self._replace_type_in_annotation(node.slice, type_map, cptr_params) + return node + + def _make_replacement_node(self, replacement: str) -> ast.AST: + """将 type_map 中的替换字符串转换为 AST 节点。 + mangled 名称如 'sha1.ASTptr' 需要还原为 AST | t.CPtr 结构， + 否则类型解析会丢失 PtrCount 信息。 + 含点号的名称（如 'sha1.AST'）需解析为 ast.Attribute 链。 + """ + if replacement.endswith('ptr'): + # 提取基础名称: 'sha1.ASTptr' -> 'sha1.AST' + base: str = replacement[:-3] + return ast.BinOp( + left=self._make_name_node(base), + op=ast.BitOr(), + right=ast.Attribute(value=ast.Name(id='t', ctx=ast.Load()), attr='CPtr', ctx=ast.Load()) + ) + return self._make_name_node(replacement) + + @staticmethod + def _make_name_node(name: str) -> ast.AST: + """将可能含点号的名称转换为 ast.Name 或 ast.Attribute 链。 + 'AST' -> ast.Name(id='AST') + 'sha1.AST' -> ast.Attribute(value=ast.Name(id='sha1'), attr='AST') + """ + if '.' not in name: + return ast.Name(id=name, ctx=ast.Load()) + parts: list[str] = name.split('.') + node: ast.AST = ast.Name(id=parts[0], ctx=ast.Load()) + for part in parts[1:]: + node = ast.Attribute(value=node, attr=part, ctx=ast.Load()) + return node + + def _apply_type_map_to_body(self, body: list[ast.AST], type_map: dict[str, str], cptr_params: set[str] | None = None) -> None: + for stmt in body: + self._apply_type_map_to_stmt(stmt, type_map, cptr_params) + + def _apply_type_map_to_stmt(self, node: ast.AST, type_map: dict[str, str], cptr_params: set[str] | None = None) -> None: + if isinstance(node, ast.Return) and node.value: + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.Assign): + for t in node.targets: + self._apply_type_map_to_expr(t, type_map, cptr_params) + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.AnnAssign): + if node.annotation: + node.annotation = self._replace_type_in_annotation(node.annotation, type_map, cptr_params) + if node.value: + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.AugAssign): + self._apply_type_map_to_expr(node.target, type_map, cptr_params) + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.Expr): + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.If): + self._apply_type_map_to_expr(node.test, type_map, cptr_params) + for s in node.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for s in node.orelse: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + elif isinstance(node, ast.For): + self._apply_type_map_to_expr(node.iter, type_map, cptr_params) + for s in node.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for s in node.orelse: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + elif isinstance(node, ast.While): + self._apply_type_map_to_expr(node.test, type_map, cptr_params) + for s in node.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for s in node.orelse: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + elif isinstance(node, ast.Try): + for s in node.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for handler in node.handlers: + for s in handler.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for s in node.orelse: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + for s in node.finalbody: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + elif isinstance(node, ast.With): + for item in node.items: + self._apply_type_map_to_expr(item.context_expr, type_map, cptr_params) + if item.optional_vars: + self._apply_type_map_to_expr(item.optional_vars, type_map, cptr_params) + for s in node.body: + self._apply_type_map_to_stmt(s, type_map, cptr_params) + elif isinstance(node, ast.Raise): + if node.exc: + self._apply_type_map_to_expr(node.exc, type_map, cptr_params) + if node.cause: + self._apply_type_map_to_expr(node.cause, type_map, cptr_params) + elif isinstance(node, ast.Assert): + self._apply_type_map_to_expr(node.test, type_map, cptr_params) + if node.msg: + self._apply_type_map_to_expr(node.msg, type_map, cptr_params) + + def _apply_type_map_to_expr(self, node: ast.AST | None, type_map: dict[str, str], cptr_params: set[str] | None = None) -> None: + if node is None: + return + if isinstance(node, ast.Call): + if isinstance(node.func, ast.Name) and node.func.id in type_map: + replacement: str = type_map[node.func.id] + node.func = ast.Name(id=replacement, ctx=ast.Load()) + elif isinstance(node.func, ast.Attribute): + if isinstance(node.func.value, ast.Name) and node.func.value.id in type_map: + replacement: str = type_map[node.func.value.id] + # 指针类型参数（如 list[AST | t.CPtr] 中的 T）在表达式上下文中 + # 需替换为 BinOp(AST | t.CPtr)，使 __sizeof__ 等处理器能正确识别 + # 指针语义并直接返回 8，避免触发 struct 加载和重入编译 + if cptr_params and node.func.value.id in cptr_params: + node.func.value = ast.BinOp( + left=self._make_name_node(replacement), + op=ast.BitOr(), + right=ast.Attribute(value=ast.Name(id='t', ctx=ast.Load()), attr='CPtr', ctx=ast.Load()) + ) + else: + node.func.value = ast.Name(id=replacement, ctx=ast.Load()) + else: + self._apply_type_map_to_expr(node.func.value, type_map, cptr_params) + for arg in node.args: + self._apply_type_map_to_expr(arg, type_map, cptr_params) + elif isinstance(node, ast.Attribute): + if isinstance(node.value, ast.Name) and node.value.id in type_map: + replacement: str = type_map[node.value.id] + if cptr_params and node.value.id in cptr_params: + node.value = ast.BinOp( + left=self._make_name_node(replacement), + op=ast.BitOr(), + right=ast.Attribute(value=ast.Name(id='t', ctx=ast.Load()), attr='CPtr', ctx=ast.Load()) + ) + else: + node.value = ast.Name(id=replacement, ctx=ast.Load()) + else: + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + elif isinstance(node, ast.BinOp): + self._apply_type_map_to_expr(node.left, type_map, cptr_params) + self._apply_type_map_to_expr(node.right, type_map, cptr_params) + elif isinstance(node, ast.UnaryOp): + self._apply_type_map_to_expr(node.operand, type_map, cptr_params) + elif isinstance(node, ast.Compare): + self._apply_type_map_to_expr(node.left, type_map, cptr_params) + for comp in node.comparators: + self._apply_type_map_to_expr(comp, type_map, cptr_params) + elif isinstance(node, ast.BoolOp): + for val in node.values: + self._apply_type_map_to_expr(val, type_map, cptr_params) + elif isinstance(node, ast.IfExp): + self._apply_type_map_to_expr(node.test, type_map, cptr_params) + self._apply_type_map_to_expr(node.body, type_map, cptr_params) + self._apply_type_map_to_expr(node.orelse, type_map, cptr_params) + elif isinstance(node, ast.Subscript): + self._apply_type_map_to_expr(node.value, type_map, cptr_params) + self._apply_type_map_to_expr(node.slice, type_map, cptr_params) + elif isinstance(node, ast.Name): + if node.id in type_map: + node.id = type_map[node.id] + + def _infer_type_arg_from_value(self, val: ir.Value, Gen: LlvmCodeGenerator) -> str: + if val is None: + return 'int' + if isinstance(val.type, ir.IntType): + if val.type.width == 64: + return 'i64' + elif val.type.width == 16: + return 'i16' + elif val.type.width == 8: + return 'i8' + return 'int' + elif isinstance(val.type, ir.DoubleType): + return 'double' + elif isinstance(val.type, ir.FloatType): + return 'float' + elif isinstance(val.type, ir.PointerType): + return 'void*' + return 'int' + + def _specialize_generic_function(self, FuncName: str, type_args: list[str], Gen: LlvmCodeGenerator) -> str | None: + if not hasattr(self, '_generic_templates') or FuncName not in self._generic_templates: + return None + template: dict[str, Any] = self._generic_templates[FuncName] + Node: ast.FunctionDef = template['node'] + type_param_names: list[str] = template['type_params'] + ClassName: str | None = template['class_name'] + if len(type_args) != len(type_param_names): + return None + spec_key: str = FuncName + '<' + ','.join(type_args) + '>' + if not hasattr(self, '_generic_specializations'): + self._generic_specializations = {} + if spec_key in self._generic_specializations: + return self._generic_specializations[spec_key] + has_cexport: bool = False + if Node.returns: + try: + RetTypeInfo: Any = None + if isinstance(Node.returns, ast.BinOp) and isinstance(Node.returns.op, ast.BitOr): + RetTypeInfo = self.Trans.TypeMergeHandler.MergeTypes(Node.returns) + else: + RetTypeInfo = CTypeInfo.FromNode(Node.returns, self.Trans.SymbolTable) + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CExport): + has_cexport = True + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + spec_name: str = self._mangle_generic_name(FuncName, type_args, has_cexport) + type_map: dict[str, str] = {} + for i, tp_name in enumerate(type_param_names): + type_map[tp_name] = type_args[i] + SpecNode: ast.FunctionDef = copy.deepcopy(Node) + SpecNode.type_params = [] + SpecNode.name = spec_name + self._apply_type_map_to_node(SpecNode, type_map) + if has_cexport: + if not hasattr(Gen, '_export_funcs'): + Gen._export_funcs = set() + Gen._export_funcs.add(spec_name) + saved_builder: Any = Gen.builder + saved_variables: dict[str, Any] = dict(Gen.variables) if Gen.variables else {} + saved_direct_values: dict[str, Any] = dict(Gen._direct_values) if Gen._direct_values else {} + saved_var_type_info: dict[str, Any] = dict(Gen.var_type_info) if Gen.var_type_info else {} + saved_var_signedness: dict[str, Any] = dict(Gen.var_signedness) if Gen.var_signedness else {} + saved_global_vars: set[str] = set(Gen.global_vars) if Gen.global_vars else set() + saved_var_scopes: list[dict[str, Any]] = [dict(s) for s in self.Trans.VarScopes] if self.Trans.VarScopes else [] + saved_func: Any = Gen.func + saved_current_func_name: str | None = getattr(Gen, '_current_func_name', None) + saved_block: Any = None + if Gen.builder and Gen.builder.block and not Gen.builder.block.is_terminated: + saved_block = Gen.builder.block + self._EmitFunctionForwardDeclLlvm(SpecNode, Gen, ClassName=ClassName) + self._EmitFunctionLlvm(SpecNode, Gen, ClassName=ClassName) + Gen.builder = saved_builder + if saved_block is not None and Gen.builder is not None: + Gen.builder.position_at_end(saved_block) + Gen.variables = saved_variables + Gen._direct_values = saved_direct_values + Gen.var_type_info = saved_var_type_info + Gen.var_signedness = saved_var_signedness + Gen.global_vars = saved_global_vars + Gen.func = saved_func + if saved_current_func_name is not None: + Gen._current_func_name = saved_current_func_name + self.Trans.VarScopes = saved_var_scopes + self._generic_specializations[spec_key] = spec_name + if not hasattr(self.Trans, '_generic_specializations'): + self.Trans._generic_specializations = {} + self.Trans._generic_specializations[spec_key] = spec_name + return spec_name + + def _EmitFunctionForwardDeclLlvm(self, Node: ast.FunctionDef, Gen: LlvmCodeGenerator, ClassName: str | None = None) -> None: + if self._is_generic_function(Node): + if not hasattr(self, '_generic_templates'): + self._generic_templates = {} + RawFuncName: str = Node.name + if ClassName: + FuncName: str = f"{ClassName}.{RawFuncName}" + else: + FuncName = RawFuncName + type_params: list[str] = [tp.name for tp in Node.type_params] + self._generic_templates[FuncName] = { + 'node': Node, + 'type_params': type_params, + 'class_name': ClassName, + } + return + result: Any = self._GetFunctionSignatureLlvm(Node, Gen, ClassName) + if result is None: + return + FuncName: str + FuncType: ir.FunctionType + ReturnTypeInfo: Any + ParamTypeStrs: list[Any] + CReturnTypes: list[ast.AST] + IsMethod: bool + ResolvedClassName: str | None + IsVariadic: bool + FuncName, FuncType, ReturnTypeInfo, ParamTypeStrs, CReturnTypes, IsMethod, ResolvedClassName, IsVariadic = result + IsUserMarkedExport: bool = False + if Node.returns: + try: + RetTypeInfo: Any = None + RetTypeInfo = self.ResolveAnnotationType(Node.returns) + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CExport) and not RetTypeInfo.IsState: + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CExtern): + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.IsState: + Gen._export_funcs.add(FuncName) # t.State 标记的函数必须保持原始名称，以便链接器解析 + IsUserMarkedExport = True + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + # 装饰器形式的存储类标记（@t.CExport / @t.CExtern / @t.State 等） + deco_flags: dict[str, bool] = self._extract_decorator_storage_flags(Node) + if deco_flags['is_export'] or deco_flags['is_extern'] or deco_flags['is_state']: + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + # 注册当前模块定义的函数名，使 _mangle_func_name 跳过 _export_extern_funcs 检查 + # （_export_extern_funcs 是跨模块共享集合，包含所有 CExtern 函数名；当前模块定义的 + # 同名用户函数应使用 SHA1 前缀，而非被误判为外部 C 函数返回裸名） + Gen._current_module_func_names.add(FuncName) + # 移除导入污染：用户函数未显式标记导出，但名称在 _export_funcs 中（来自导入的同名 C 库函数） + if not IsUserMarkedExport and FuncName in Gen._export_funcs: + Gen._export_funcs.discard(FuncName) + MangledName: str = Gen._mangle_func_name(FuncName) + need_new: bool = MangledName not in Gen.functions + if not need_new: + existing: Any = Gen.functions[MangledName] + if getattr(existing, 'name', MangledName) != MangledName: + need_new = True + if need_new: + try: + func: ir.Function = ir.Function(Gen.module, FuncType, name=MangledName) + func.attributes.add('noredzone') + if hasattr(self, '_cdefine_funcs') and FuncName in self._cdefine_funcs: + func.linkage = 'linkonce_odr' + func.attributes.add('alwaysinline') + if hasattr(self, '_pending_llvm_attrs') and FuncName in self._pending_llvm_attrs: + for attr_name in self._pending_llvm_attrs[FuncName]: + try: + func.attributes.add(attr_name) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"添加函数属性失败: {_e}", "Exception") + Gen.functions[MangledName] = func + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = func + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + else: + existing = Gen.functions[MangledName] + existing_type: ir.FunctionType | None = getattr(existing, 'ftype', None) + if existing_type and existing_type != FuncType: + Gen.functions[MangledName] = self._replace_function_decl(Gen, FuncName, FuncType) + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = Gen.functions[MangledName] + elif not existing_type: + try: + Gen.functions[MangledName] = self._replace_function_decl(Gen, FuncName, FuncType) + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = Gen.functions[MangledName] + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + self.Trans.FunctionDefCache[FuncName] = Node + + def _replace_function_decl(self, Gen: LlvmCodeGenerator, FuncName: str, NewFuncType: ir.FunctionType) -> ir.Function: + MangledName: str = Gen._mangle_func_name(FuncName) + old_func: Any = Gen.functions.get(MangledName) + if old_func is None: + try: + return ir.Function(Gen.module, NewFuncType, name=MangledName) + except Exception: # 回退：函数已存在时返回旧函数 + return old_func + try: + old_name: str = old_func.name + if old_name in getattr(Gen.module, 'globals', {}): + del Gen.module.globals[old_name] + module_scope: Any = getattr(Gen.module, 'scope', None) + if getattr(module_scope, '_useset', None): + module_scope._useset.discard(old_name) + if getattr(Gen.module, '_guessed_names', None): + Gen.module._guessed_names.discard(FuncName) + new_func: ir.Function = ir.Function(Gen.module, NewFuncType, name=MangledName) + Gen.functions[MangledName] = new_func + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = new_func + return new_func + except Exception as ex: + traceback.print_exc() + Gen.functions[MangledName] = old_func + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = old_func + return old_func + + def _EmitFunctionLlvm(self, Node: ast.FunctionDef, Gen: LlvmCodeGenerator, ClassName: str | None = None, extra_params: list[tuple[str, ir.Type]] | None = None) -> ir.Function | None: + if self._is_generic_function(Node): + return + FuncName, RawFuncName, func_meta = self._compute_func_name_meta(Node, ClassName) + # 保存父级 _CurrentCpythonObjectClass，防止嵌套函数发射污染父方法类上下文 + # （例如 HandleImport body 内触发 list[ast.ASTptr] 方法发射，结束后会清空 _CurrentCpythonObjectClass） + saved_cpython_class: str | None = self.Trans._CurrentCpythonObjectClass + CReturnTypes = [] + IsPtr = False + fn_attrs = {} + # 自定义行为装饰器列表，用于生成 wrapper 函数 + # 任何 @name 或 @name(args) 只要不是内置装饰器，都视为自定义行为装饰器 + behavior_decorators = [] + # 内置装饰器名称，这些由编译器特殊处理，不作为行为装饰器 + _BUILTIN_DECORATORS = {'staticmethod', 'property', 'classmethod'} + _symtab: object = self.Trans.SymbolTable + if Node.decorator_list: + for decorator in Node.decorator_list: + # 跳过 t 模块存储类标记装饰器（@t.CExport / @CExport 等） + if self._is_t_storage_decorator(decorator, _symtab): + continue + # 识别自定义行为装饰器：@name（ast.Name 形式） + if isinstance(decorator, ast.Name): + if decorator.id not in _BUILTIN_DECORATORS: + behavior_decorators.append({'name': decorator.id, 'args': [], 'kwargs': {}}) + continue + # 识别带参数的自定义行为装饰器：@name(arg1, key=val)（ast.Call + ast.Name 形式） + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Name): + deco_name = decorator.func.id + if deco_name not in _BUILTIN_DECORATORS: + deco_info = {'name': deco_name, 'args': [], 'kwargs': {}} + for arg in decorator.args: + if isinstance(arg, ast.Constant): + deco_info['args'].append(arg.value) + for kw in decorator.keywords: + if isinstance(kw.value, ast.Constant): + deco_info['kwargs'][kw.arg] = kw.value.value + behavior_decorators.append(deco_info) + continue + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Attribute): + if decorator.func.attr == 'CReturn': + for arg in decorator.args: + CReturnTypes.append(arg) + elif decorator.func.attr == 'Attribute' and isinstance(decorator.func.value, ast.Name) and decorator.func.value.id == 'c': + for arg in decorator.args: + if isinstance(arg, ast.Call) and isinstance(arg.func, ast.Attribute): + if isinstance(arg.func.value, ast.Attribute) and isinstance(arg.func.value.value, ast.Name) and IsTModule(arg.func.value.value.id, self.Trans.SymbolTable) and arg.func.value.attr == 'attr': + attr_name = arg.func.attr + str_args = [a.value for a in arg.args if isinstance(a, ast.Constant) and isinstance(a.value, str)] + int_args = [a.value for a in arg.args if isinstance(a, ast.Constant) and isinstance(a.value, int)] + if attr_name == 'section' and str_args: + fn_attrs['section'] = str_args[0] + elif attr_name == 'aligned' and int_args: + fn_attrs['aligned'] = int_args[0] + elif attr_name == 'visibility' and str_args: + fn_attrs['visibility'] = str_args[0] + elif attr_name in ('noreturn', 'always_inline', 'noinline', 'cold', 'hot', 'constructor', 'destructor', 'pure', 'const', 'malloc', 'returns_nonnull', 'used', 'naked', 'no_instrument_function', 'warn_unused_result', 'weak', 'fallthrough'): + if not arg.args: + fn_attrs[attr_name] = True + elif isinstance(arg.func.value, ast.Attribute) and arg.func.value.attr == 'llvm' and isinstance(arg.func.value.value, ast.Attribute) and arg.func.value.value.attr == 'attr' and isinstance(arg.func.value.value.value, ast.Name) and IsTModule(arg.func.value.value.value.id, self.Trans.SymbolTable): + llvm_attr_name = arg.func.attr + if llvm_attr_name in self._LLVM_ATTR_NAMES: + fn_attrs[f'llvm.{llvm_attr_name}'] = True + elif isinstance(arg, ast.Attribute): + if isinstance(arg.value, ast.Attribute) and isinstance(arg.value.value, ast.Name) and IsTModule(arg.value.value.id, self.Trans.SymbolTable) and arg.value.attr == 'attr': + attr_name = arg.attr + if attr_name in ('noreturn', 'always_inline', 'noinline', 'cold', 'hot', 'constructor', 'destructor', 'pure', 'const', 'malloc', 'returns_nonnull', 'used', 'naked', 'no_instrument_function', 'warn_unused_result', 'weak', 'fallthrough', 'packed'): + fn_attrs[attr_name] = True + elif isinstance(arg.value, ast.Attribute) and arg.value.attr == 'llvm' and isinstance(arg.value.value, ast.Attribute) and arg.value.value.attr == 'attr' and isinstance(arg.value.value.value, ast.Name) and IsTModule(arg.value.value.value.id, self.Trans.SymbolTable): + llvm_attr_name = arg.attr + if llvm_attr_name in self._LLVM_ATTR_NAMES: + fn_attrs[f'llvm.{llvm_attr_name}'] = True + # 检查 @t.TLS 装饰器（一次性初始化标记） + is_tls_func: bool = False + if Node.decorator_list: + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Attribute) and isinstance(decorator.value, ast.Name): + if IsTModule(decorator.value.id, _symtab) and decorator.attr == 'TLS': + is_tls_func = True + break + elif isinstance(decorator, ast.Name) and decorator.id == 'TLS' and hasattr(t, 'TLS'): + is_tls_func = True + break + if Node.returns: + if isinstance(Node.returns, ast.Subscript) and isinstance(Node.returns.value, ast.Name) and Node.returns.value.id == 'tuple': + slice_node = Node.returns.slice + if isinstance(slice_node, ast.Tuple): + for elt in slice_node.elts: + CReturnTypes.append(elt) + else: + CReturnTypes.append(slice_node) + + # 先强制检查 str 类型注解 + is_str_return = False + ReturnTypeInfo = None + if Node.returns: + if isinstance(Node.returns, ast.Name) and Node.returns.id in ('str', 'bytes'): + is_str_return = True + else: + try: + rt_info = CTypeInfo.FromNode(Node.returns, self.Trans.SymbolTable) + if rt_info and rt_info.IsStr: + is_str_return = True + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + + if is_str_return: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + elif Node.returns: + if CReturnTypes: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + else: + try: + if isinstance(Node.returns, ast.BinOp) and isinstance(Node.returns.op, ast.BitOr): + ReturnTypeInfo = getattr(self.Trans, 'TypeMergeHandler', None) and self.Trans.TypeMergeHandler.MergeTypes(Node.returns) + if not isinstance(ReturnTypeInfo, CTypeInfo): + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + if (ReturnTypeInfo.IsVoid and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState) or ReturnTypeInfo.PtrCount >= 2: + def _FindNonVoidCTypeInfo(node): + if isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + results = [] + for child in [node.left, node.right]: + r = _FindNonVoidCTypeInfo(child) + if r and not r.IsVoid: + results.append(r) + for r in results: + if not r.IsPtr and not r.IsStruct: + return r + return results[0] if results else None + try: + SideInfo = CTypeInfo.FromNode(node, self.Trans.SymbolTable) + if SideInfo: + return SideInfo + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + return None + found = _FindNonVoidCTypeInfo(Node.returns) + if found and not found.IsVoid: + ReturnTypeInfo = found + if ReturnTypeInfo.IsVoid and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + else: + is_str_type = False + if isinstance(Node.returns, ast.Name) and Node.returns.id in ('str', 'bytes'): + is_str_type = True + ReturnTypeInfo = CTypeInfo.FromNode(Node.returns, self.Trans.SymbolTable) + if ReturnTypeInfo and ReturnTypeInfo.IsDefine: + inferred = self._infer_return_type_from_body(Node, Gen, ClassName) + if inferred: + ReturnTypeInfo = CTypeInfo.FromTypeName(inferred[0]) + ReturnTypeInfo.PtrCount = 1 if inferred[1] else 0 + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + if not hasattr(self, '_cdefine_funcs'): + self._cdefine_funcs = set() + self._cdefine_funcs.add(FuncName) + elif ReturnTypeInfo and ReturnTypeInfo.IsState: + pass # t.State is a modifier, preserve the actual return type + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CVoid() + if is_str_type or ReturnTypeInfo.IsStr: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + # 检查泛型类返回类型 (如 list[str]) + # 注意：_from_node_subscript 对未识别的泛型下标（如 list[str]）返回空 CTypeInfo（BaseType=None）， + # 此时 IsVoid 为 False（因为 IsVoid 仅检查 t.CVoid 实例），需额外检测 BaseType is None + if (isinstance(Node.returns, ast.Subscript) + and isinstance(Node.returns.value, ast.Name) + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and Node.returns.value.id in self.Trans.ClassHandler._generic_class_templates + and (ReturnTypeInfo.IsVoid or ReturnTypeInfo.BaseType is None) and ReturnTypeInfo.PtrCount == 0): + ret_class: str = Node.returns.value.id + ret_slice: ast.AST = Node.returns.slice + ret_type_args: list[str] = [] + ret_type_names: list[str] = [] + if isinstance(ret_slice, ast.Tuple): + for elt in ret_slice.elts: + if isinstance(elt, ast.Name): + tn2: str = elt.id + ts2: str = tn2 + if tn2 == 'int': + ts2, tn2 = 'int', 'CInt' + ret_type_args.append(ts2) + ret_type_names.append(tn2) + elif isinstance(ret_slice, ast.Name): + tn2: str = ret_slice.id + ts2: str = tn2 + if tn2 == 'int': + ts2, tn2 = 'int', 'CInt' + ret_type_args.append(ts2) + ret_type_names.append(tn2) + # 返回类型解析只需 struct 定义，不需方法体；使用 declare_only=True + # 避免过早触发完整特化（此时 MemManager.alloc 可能尚未注册， + # 导致 __new__ 的 pool.alloc(48) 解析失败，IR 块未终止）。 + spec_name2: str | None = self.Trans.ClassHandler._specialize_generic_class( + ret_class, ret_type_args, Gen, type_names=ret_type_names, declare_only=True) + if spec_name2 and spec_name2 in Gen.structs: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CStruct(name=spec_name2) + ReturnTypeInfo.PtrCount = 1 + if (ReturnTypeInfo.IsVoid or ReturnTypeInfo.BaseType is None) and ReturnTypeInfo.PtrCount == 0 and not ReturnTypeInfo.IsState: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + except Exception: # 回退：设置默认返回类型为 CInt + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + inferred = self._infer_return_type_from_body(Node, Gen, ClassName) + if inferred: + ReturnTypeInfo = CTypeInfo.FromTypeName(inferred[0]) + ReturnTypeInfo.PtrCount = 1 if inferred[1] else 0 + + ReturnType = Gen._ctype_to_llvm(ReturnTypeInfo) + + if isinstance(ReturnType, ir.VoidType) and FuncName == 'main': + ReturnType = ir.IntType(32) + IsMethod, IsClassMethod, ResolvedClassName = self._resolve_method_class(Node, ClassName, func_meta, FuncName, Gen) + # __new__ methods should return a pointer to the struct (for heap allocation replacement) + if RawFuncName == '__new__' and ResolvedClassName: + StructType = Gen.structs.get(ResolvedClassName) + if StructType: + if isinstance(StructType, ir.PointerType): + ReturnType = StructType + else: + ReturnType = ir.PointerType(StructType) + ParamTypes = [] + ParamNames = [] + ParamTypeStrs = [] + auto_addr_modes_emit: list[str | None] = [] + self.Trans.VarScopes.append({}) + for Arg in Node.args.args: + ParamIsUnsigned = False + if Arg.annotation: + try: + ParamTypeInfo = self.ResolveAnnotationType(Arg.annotation) + if ParamTypeInfo is None: + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CInt() + IsPtr = ParamTypeInfo.IsPtr + if ParamTypeInfo and getattr(ParamTypeInfo, 'IsCpythonObject', False) and not IsPtr: + IsPtr = True + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + if ParamTypeInfo.IsStr or (isinstance(Arg.annotation, ast.Name) and Arg.annotation.id in ('str', 'bytes')): + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CChar() + ParamTypeInfo.PtrCount = 1 + IsPtr = True + # C 语言中数组参数退化为指针：list[type, N] → type* + if ParamTypeInfo.ArrayDims: + ParamTypeInfo.ArrayDims = [] + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + ParamType = Gen._ctype_to_llvm(ParamTypeInfo) + if ParamTypeInfo and ParamTypeInfo.IsFuncPtr: + ParamType = ir.IntType(8).as_pointer() + ParamIsUnsigned = ParamTypeInfo.IsUInt + except Exception: # 回退：参数类型解析失败时使用默认 i32 + ParamType = ir.IntType(32) + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CInt() + else: + ParamType = ir.IntType(32) + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CInt() + ParamTypes.append(ParamType) + ParamNames.append(Arg.arg) + ParamTypeStrs.append(ParamTypeInfo) + # 检测 t.CNeedPtr / t.CAutoPtr 参数注解 + auto_mode_emit: str | None = None + if ParamTypeInfo is not None and hasattr(ParamTypeInfo, 'BaseType') and ParamTypeInfo.BaseType is not None: + if isinstance(ParamTypeInfo.BaseType, t.CAutoPtr): + auto_mode_emit = 'always' + elif isinstance(ParamTypeInfo.BaseType, t.CNeedPtr): + auto_mode_emit = 'need' + auto_addr_modes_emit.append(auto_mode_emit) + Gen._record_var_signedness(Arg.arg, ParamIsUnsigned) + self.Trans.VarScopes[-1][Arg.arg] = ParamTypeInfo + if Arg.annotation: + try: + PInfo = self.ResolveAnnotationType(Arg.annotation) + if PInfo and (PInfo.DataConst or PInfo.VarConst): + Gen.var_const_flags[Arg.arg] = True + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + if CReturnTypes: + CReturnLlvmTypes = [] + for i, ReturnTypeNode in enumerate(CReturnTypes): + ReturnTypeInfo = CTypeInfo.FromNode(ReturnTypeNode, self.Trans.SymbolTable) + if ReturnTypeInfo and ReturnTypeInfo.IsStr: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + RetType = Gen._ctype_to_llvm(ReturnTypeInfo) + CReturnLlvmTypes.append(RetType) + ReturnType = ir.LiteralStructType(CReturnLlvmTypes) + self.Trans.CurrentCReturnTypes = CReturnTypes + else: + self.Trans.CurrentCReturnTypes = None + self._apply_struct_self_cls_types(ParamTypes, ParamNames, IsMethod, IsClassMethod, ResolvedClassName, Gen) + IsVariadic = Node.args.vararg is not None + if extra_params: + for var_name, ptr_type in extra_params: + ParamTypes.append(ptr_type) + ParamNames.append(f'__nonlocal_{var_name}__') + HasStopIterFlag = False + if IsMethod and RawFuncName == '__next__': + ParamTypes.append(ir.PointerType(ir.IntType(1))) + ParamNames.append('__stop_iter_flag__') + HasStopIterFlag = True + HasRaise = self._body_contains_raise(Node.body) + if HasRaise and RawFuncName != 'main': + ParamTypes.append(ir.PointerType(ir.PointerType(ir.IntType(8)))) + ParamNames.append('__eh_msg_out__') + ParamTypes.append(ir.PointerType(ir.IntType(32))) + ParamNames.append('__eh_code_out__') + FuncType = ir.FunctionType(ReturnType, ParamTypes, var_arg=IsVariadic) + # 存储 auto_addr_modes（为额外参数补 None） + total_params_emit: int = len(ParamTypes) + while len(auto_addr_modes_emit) < total_params_emit: + auto_addr_modes_emit.append(None) + Gen._auto_addr_params[FuncName] = auto_addr_modes_emit + self.Trans.FunctionDefCache[FuncName] = Node + IsInlineFunc = False + IsExternFunc = False + IsStateFunc = False + IsUserMarkedExport: bool = False + if Node.returns: + try: + RetTypeInfo = None + RetTypeInfo = self.ResolveAnnotationType(Node.returns) + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CInline): + IsInlineFunc = True + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CExport) and not RetTypeInfo.IsState: + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.Storage and isinstance(RetTypeInfo.Storage, t.CExtern): + IsExternFunc = True + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + if isinstance(RetTypeInfo, CTypeInfo) and RetTypeInfo.IsState: + IsStateFunc = True + Gen._export_funcs.add(FuncName) # 外部 C 函数声明必须保持原始名称 + IsUserMarkedExport = True + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + # 装饰器形式的存储类标记（@t.CExport / @t.CExtern / @t.CStatic / @t.State / @t.CInline） + deco_flags: dict[str, bool] = self._extract_decorator_storage_flags(Node) + if deco_flags['is_inline']: + IsInlineFunc = True + if deco_flags['is_extern']: + IsExternFunc = True + if deco_flags['is_state']: + IsStateFunc = True + if deco_flags['is_export'] or deco_flags['is_extern'] or deco_flags['is_state']: + Gen._export_funcs.add(FuncName) + IsUserMarkedExport = True + # 注册当前模块定义的函数名，使 _mangle_func_name 跳过 _export_extern_funcs 检查 + Gen._current_module_func_names.add(FuncName) + # 移除导入污染：用户函数未显式标记导出，但名称在 _export_funcs 中（来自导入的同名 C 库函数） + if not IsUserMarkedExport and FuncName in Gen._export_funcs: + Gen._export_funcs.discard(FuncName) + MangledName = Gen._mangle_func_name(FuncName) + if MangledName in Gen.functions: + func = Gen.functions[MangledName] + if getattr(func, 'name', MangledName) != MangledName: + func = ir.Function(Gen.module, FuncType, name=MangledName) + Gen.functions[MangledName] = func + else: + existing_type = getattr(func, 'ftype', None) + if existing_type and existing_type != FuncType: + func = self._replace_function_decl(Gen, FuncName, FuncType) + else: + func = ir.Function(Gen.module, FuncType, name=MangledName) + Gen.functions[MangledName] = func + if MangledName == FuncName or FuncName not in Gen._export_extern_funcs: + Gen.functions[FuncName] = func + func.attributes.add('noredzone') + if IsInlineFunc: + func.linkage = 'linkonce_odr' + func.attributes.add('alwaysinline') + if hasattr(self, '_cdefine_funcs') and FuncName in self._cdefine_funcs: + func.linkage = 'linkonce_odr' + func.attributes.add('alwaysinline') + if fn_attrs: + _LLVM_FN_ATTR_MAP = { + 'noreturn': 'noreturn', + 'always_inline': 'alwaysinline', + 'noinline': 'noinline', + 'cold': 'cold', + 'hot': 'hot', + 'constructor': 'constructor', + 'destructor': 'destructor', + 'malloc': 'malloc', + 'returns_nonnull': 'returns_nonnull', + 'used': 'used', + 'naked': 'naked', + 'no_instrument_function': 'no_instrument_function', + 'warn_unused_result': 'warn_unused_result', + 'fallthrough': 'fallthrough', + 'pure': 'readonly', + 'const': 'readnone', + } + if 'section' in fn_attrs: + func.section = fn_attrs['section'] + if 'visibility' in fn_attrs: + func.linkage = 'internal' + func.attributes.add('visibility') + if fn_attrs.get('weak'): + func.linkage = 'weak' + for attr_name, llvm_name in _LLVM_FN_ATTR_MAP.items(): + if fn_attrs.get(attr_name): + try: + func.attributes.add(llvm_name) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"添加函数属性失败: {_e}", "Exception") + for key in fn_attrs: + if key.startswith('llvm.'): + llvm_attr_name = key[5:] + try: + func.attributes.add(llvm_attr_name) + except Exception as _e: + # llvmlite 不支持某些较新的 LLVM 属性（如 willreturn, mustprogress） + # 记录下来，后续通过 IR 后处理添加 + if not hasattr(Gen, '_pending_str_attrs'): + Gen._pending_str_attrs = {} + Gen._pending_str_attrs.setdefault(MangledName, []).append(llvm_attr_name) + if hasattr(self, '_pending_llvm_attrs') and FuncName in self._pending_llvm_attrs: + for attr_name in self._pending_llvm_attrs.pop(FuncName): + try: + func.attributes.add(attr_name) + except Exception as _e: + # llvmlite 不支持的属性，记录下来后续通过 IR 后处理添加 + if not hasattr(Gen, '_pending_str_attrs'): + Gen._pending_str_attrs = {} + Gen._pending_str_attrs.setdefault(MangledName, []).append(attr_name) + # 记录自定义行为装饰器信息，供 DecoratorPass 使用 + if behavior_decorators: + Gen._decorated_funcs[MangledName] = { + 'decorators': behavior_decorators, + 'func_name': FuncName, + 'func_type': FuncType, + 'return_type': FuncType.return_type, + 'param_types': [p.type for p in func.args], + 'param_names': ParamNames, + 'is_export': FuncName in Gen._export_funcs, + } + if IsExternFunc or IsStateFunc: + return + EntryBlock = func.append_basic_block(name="entry") + Gen.builder = ir.IRBuilder(EntryBlock) + Gen.func = func + Gen.variables = {} + Gen._reg_values = {} + Gen.global_vars = set() + if 'aligned' in fn_attrs: + Gen._emit_stack_align(fn_attrs['aligned']) + # 保存当前的 var_type_info，以便函数结束时恢复 + saved_var_type_info = Gen.var_type_info.copy() + saved_var_type_assignments = Gen.var_type_assignments.copy() + # 函数内部定义的变量会在赋值时添加到 var_type_info 中 + # 这样函数内部定义的元类型变量（如 a = t.CInt32T）就能被正确处理 + for stmt in Node.body: + if isinstance(stmt, ast.Global): + self.Trans.ForHandler._RegisterGlobalNames(stmt.names, Gen) + for param, param_name in zip(func.args, ParamNames): + param.name = param_name + if param_name == '__stop_iter_flag__': + Gen._stop_iter_flag_param = param + Gen.builder.store(ir.Constant(ir.IntType(1), 0), param) + elif param_name.startswith('__nonlocal_') and param_name.endswith('__'): + var_name = param_name[len('__nonlocal_'):-2] + Gen.variables[var_name] = param + else: + if isinstance(param.type, ir.PointerType) and isinstance(param.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + var = Gen.builder.alloca(param.type, name=param_name) + Gen._store(param, var) + Gen.variables[param_name] = var + for CN, ST in Gen.structs.items(): + if param.type.pointee == ST or (isinstance(param.type.pointee, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType) and param.type.pointee.name == ST.name): + Gen.var_struct_class[param_name] = CN + break + elif isinstance(param.type, (ir.LiteralStructType, ir.IdentifiedStructType)): + var = Gen.builder.alloca(param.type, name=param_name) + Gen._store(param, var) + Gen.variables[param_name] = var + for CN, ST in Gen.structs.items(): + if param.type == ST or (isinstance(param.type, ir.IdentifiedStructType) and isinstance(ST, ir.IdentifiedStructType) and param.type.name == ST.name): + Gen.var_struct_class[param_name] = CN + break + else: + # 所有参数都存入 alloca，确保循环体中对参数的修改能正确反映 + var = Gen._allocaEntry(param.type, name=param_name) + Gen._store(param, var) + Gen.variables[param_name] = var + # 为联合类型注解（如 Function | t.CPtr）的参数设置 var_struct_class， + # 确保编译器能正确生成 GEP 访问结构体成员（如 func.Blocks、func.Params） + for i, arg in enumerate(Node.args.args): + if arg.annotation and isinstance(arg.annotation, ast.BinOp) and isinstance(arg.annotation.op, ast.BitOr): + param_name: str = ParamNames[i] if i < len(ParamNames) else arg.arg + if param_name in Gen.variables: + found_struct: str | None = FindStructNameInAnnotation(arg.annotation, Gen.structs) + if found_struct: + Gen.var_struct_class[param_name] = found_struct + if ResolvedClassName and (IsMethod or RawFuncName == '__init__'): + self.Trans._CurrentCpythonObjectClass = ResolvedClassName + Gen._variadic_info = None + if IsVariadic and Node.args.vararg: + vararg_name = getattr(Node.args.vararg, 'arg', 'args') + last_param_name = ParamNames[-1] if ParamNames else None + last_param_ptr = None + for param, pn in zip(func.args, ParamNames): + if pn == last_param_name: + last_param_ptr = param + break + triple = getattr(Gen, 'module_triple', '') or getattr(Gen.module, 'triple', '') + is_windows = 'windows' in triple if triple else False + ptr_size = getattr(Gen, 'ptr_size', 8) + # va_list 大小取决于 ABI：Windows x64 = 1个指针，Linux x64 = __va_list_tag[1]（24字节） + # 32位平台：Windows = 4字节指针，Linux = 12字节 + if ptr_size == 4: + va_list_size = 4 if is_windows else 12 + va_list_align = 4 + else: + va_list_size = 8 if is_windows else 24 + va_list_align = 8 if is_windows else 16 + va_list_type = ir.ArrayType(ir.IntType(8), va_list_size) + va_list_alloc = Gen._allocaEntry(va_list_type, name=f"{vararg_name}_va_list", align=va_list_align) + va_list_ptr = Gen.builder.bitcast(va_list_alloc, ir.IntType(8).as_pointer(), name=f"{vararg_name}_va_list_i8ptr") + Gen._variadic_info = { + 'vararg_name': vararg_name, + 'va_list_ptr': va_list_ptr, + 'last_param_ptr': last_param_ptr, + 'va_start_called': False, + } + if last_param_ptr: + last_param_alloc = Gen._allocaEntry(last_param_ptr.type, name="last_param_copy") + Gen.builder.store(last_param_ptr, last_param_alloc) + Gen.emit_va_start(va_list_ptr) + Gen._variadic_info['va_start_called'] = True + # 用 _direct_values 注册，Name 解析时直接返回 va_list 指针（i8*）， + # 不走 variables 的 load 路径（否则会把 i8* 当 char* 加载一个字节） + Gen._direct_values[vararg_name] = va_list_ptr + # @t.TLS 一次性初始化检查：首次调用执行函数体，后续调用直接返回 + if is_tls_func: + flag_name: str = f"__tls_init_done_{FuncName}" + flag_global = None + for gv in Gen.module.global_values: + if gv.name == flag_name: + flag_global = gv + break + if flag_global is None: + flag_global = ir.GlobalVariable(Gen.module, ir.IntType(1), name=flag_name) + flag_global.initializer = ir.Constant(ir.IntType(1), 0) + flag_global.linkage = 'internal' + flag_val = Gen.builder.load(flag_global, name="tls_flag_check") + skip_bb = func.append_basic_block("tls_skip") + cont_bb = func.append_basic_block("tls_continue") + Gen.builder.cbranch(flag_val, skip_bb, cont_bb) + Gen.builder.position_at_end(skip_bb) + ActualReturnType = func.function_type.return_type + if isinstance(ActualReturnType, ir.VoidType): + Gen.builder.ret_void() + elif isinstance(ActualReturnType, ir.IntType): + Gen.builder.ret(ir.Constant(ActualReturnType, 0)) + elif isinstance(ActualReturnType, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ActualReturnType, 0.0)) + elif isinstance(ActualReturnType, ir.PointerType): + Gen.builder.ret(ir.Constant(ActualReturnType, None)) + else: + Gen.builder.ret(ir.Constant(ActualReturnType, None)) + Gen.builder.position_at_end(cont_bb) + Gen.builder.store(ir.Constant(ir.IntType(1), 1), flag_global) + _saved_class_name = getattr(Gen, '_current_class_name', None) + Gen._current_class_name = ClassName + # 提取首个裸字符串字面量作为 __doc__ docstring，并跳过该节点避免生成死代码 + func_body: list[ast.stmt] = Node.body + doc_symbol: str = f"{ClassName}.{Node.name}" if ClassName else Node.name + if (func_body and isinstance(func_body[0], ast.Expr) + and isinstance(func_body[0].value, ast.Constant) + and isinstance(func_body[0].value.value, str)): + Gen._create_doc_global(doc_symbol, func_body[0].value.value) + func_body = func_body[1:] + else: + Gen._create_doc_global(doc_symbol, None) + try: + self.Trans.BodyHandler.HandleBodyLlvm(func_body) + except Exception as _body_exc: + # 方法体编译失败：确保 block 终止（防止 expected instruction opcode 错误）， + # 然后重新抛出异常让编译立即终止（不吞掉错误） + _fn_name: str = Node.name + _cls_name: str = ClassName if ClassName else '' + import sys as _sys + import traceback as _tb + print(f"[FATAL] 方法体编译失败 ({_cls_name}.{_fn_name}): {type(_body_exc).__name__}: {_body_exc}", file=_sys.stderr) + _tb.print_exc(file=_sys.stderr) + if Gen.builder is not None: + try: + if not Gen.builder.block.is_terminated: + _ret_ty = func.function_type.return_type + if isinstance(_ret_ty, ir.VoidType): + Gen.builder.ret_void() + elif isinstance(_ret_ty, ir.PointerType): + Gen.builder.ret(ir.Constant(_ret_ty, None)) + elif isinstance(_ret_ty, ir.IntType): + Gen.builder.ret(ir.Constant(_ret_ty, 0)) + else: + Gen.builder.ret(ir.Constant(_ret_ty, None)) + except Exception: + try: + if Gen.builder and not Gen.builder.block.is_terminated: + Gen.builder.ret_void() + except Exception: + pass + raise + if not Gen.builder.block.is_terminated: + if 'naked' in fn_attrs: + Gen.builder.unreachable() + self.Trans._CurrentCpythonObjectClass = saved_cpython_class + return + Gen._emit_local_heap_frees() + if Gen._variadic_info and Gen._variadic_info.get('va_start_called'): + va_list_ptr = Gen._variadic_info.get('va_list_ptr') + if va_list_ptr: + Gen.emit_va_end(va_list_ptr) + ActualReturnType = func.function_type.return_type + if isinstance(ActualReturnType, ir.VoidType): + Gen.builder.ret_void() + elif isinstance(ActualReturnType, ir.IntType): + Gen.builder.ret(ir.Constant(ActualReturnType, 0)) + elif isinstance(ActualReturnType, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ActualReturnType, 0.0)) + elif isinstance(ActualReturnType, ir.PointerType): + Gen.builder.ret(ir.Constant(ActualReturnType, None)) + elif isinstance(ActualReturnType, ir.IdentifiedStructType): + if ActualReturnType.elements: + zero_val = ir.Constant(ActualReturnType, [ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) else ir.Constant(et, ir.Undefined) for et in ActualReturnType.elements]) + else: + zero_val = ir.Constant(ActualReturnType, None) + Gen.builder.ret(zero_val) + elif isinstance(ActualReturnType, ir.ArrayType): + Gen.builder.ret(ir.Constant(ActualReturnType, None)) + else: + Gen.builder.ret_void() + Gen.builder = None + Gen.func = None + Gen._current_class_name = _saved_class_name + Gen.variables = {} + Gen.var_signedness = {} + Gen._reg_values = {} + Gen._direct_values = {} + Gen.var_struct_class = {} + Gen.var_ptr_element = {} + Gen.var_const_flags = {} + # 恢复保存的 var_type_info + Gen.var_type_info = saved_var_type_info + Gen.var_type_assignments = saved_var_type_assignments + Gen._stop_iter_flag_param = None + Gen._local_heap_ptrs = [] + Gen._var_to_heap_ptr = {} + Gen._variadic_info = None + self.Trans._CurrentCpythonObjectClass = saved_cpython_class + self.Trans.CurrentCReturnTypes = None + if self.Trans.VarScopes: + self.Trans.VarScopes.pop() + self.Trans.FunctionReturnTypes[FuncName] = ReturnTypeInfo + if not self.Trans.SymbolTable.has(FuncName): + FuncInfo = CTypeInfo() + FuncInfo.Name = FuncName + FuncInfo.IsFunction = True + FuncInfo.MetaList = func_meta + if IsInlineFunc: + FuncInfo.IsInline = True + FuncInfo.InlineBody = Node.body + FuncInfo.InlineParams = [arg.arg for arg in Node.args.args] + self.Trans.SymbolTable.insert(FuncName, FuncInfo) + else: + existing = self.Trans.SymbolTable[FuncName] + if existing.MetaList == FuncMeta.NONE and func_meta != FuncMeta.NONE: + existing.MetaList = func_meta + if IsInlineFunc: + existing.IsInline = True + existing.InlineBody = Node.body + existing.InlineParams = [arg.arg for arg in Node.args.args] + # property setter/deleter: 在原始 PropKey（不带后缀）下注册 MetaList + if FuncMeta.PROPERTY_SETTER in func_meta or FuncMeta.PROPERTY_DELETER in func_meta: + BasePropKey = f"{ClassName}.{RawFuncName}" if ClassName else RawFuncName + base_existing = self.Trans.SymbolTable.lookup(BasePropKey) + if base_existing: + if func_meta != FuncMeta.NONE: + base_existing.MetaList = base_existing.MetaList | func_meta + else: + PropInfo = CTypeInfo() + PropInfo.Name = BasePropKey + PropInfo.IsFunction = True + PropInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(BasePropKey, PropInfo) + return func diff --git a/lib/core/Handles/HandlesIf.py b/lib/core/Handles/HandlesIf.py new file mode 100644 index 0000000..2fd361a --- /dev/null +++ b/lib/core/Handles/HandlesIf.py @@ -0,0 +1,204 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +import ast +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle +from lib.core.ConstEvaluator import ConstEvaluator, EvalContext + + +class IfHandle(BaseHandle): + def _get_attr_full_name(self, node: ast.expr) -> str | None: + if isinstance(node, ast.Attribute): + parts: list[str] = [] + cur: ast.expr = node + while isinstance(cur, ast.Attribute): + parts.append(cur.attr) + cur = cur.value + if isinstance(cur, ast.Name): + parts.append(cur.id) + parts.reverse() + return '.'.join(parts) + return None + + def _is_cif_call(self, node: ast.expr) -> bool: + if isinstance(node, ast.Call): + if isinstance(node.func, ast.Attribute): + if isinstance(node.func.value, ast.Name) and node.func.value.id == 'c': + return node.func.attr in ('CIf', 'CIfdef', 'CIfndef') + if isinstance(node.func, ast.Name): + return node.func.id in ('CIf', 'CIfdef', 'CIfndef') + return False + + def _resolve_macro_name(self, arg: ast.expr) -> str | None: + if isinstance(arg, ast.Name): + return arg.id + if isinstance(arg, ast.Constant) and isinstance(arg.value, str): + return arg.value + if isinstance(arg, ast.Attribute): + full_name: str | None = self._get_attr_full_name(arg) + if full_name: + return full_name + return None + + def _evaluate_cif_condition(self, node: ast.Call) -> int | None: + if not isinstance(node, ast.Call): + return None + args: list[ast.expr] = node.args + if not args: + return 0 + kind: str | None = None + if isinstance(node.func, ast.Attribute): + kind = node.func.attr + elif isinstance(node.func, ast.Name): + kind = node.func.id + else: + return None + if kind == 'CIfdef': + name: str | None = self._resolve_macro_name(args[0]) + if name is None: + return 0 + return 1 if self._is_macro_defined(name) else 0 + elif kind == 'CIfndef': + name: str | None = self._resolve_macro_name(args[0]) + if name is None: + return 1 + return 0 if self._is_macro_defined(name) else 1 + elif kind == 'CIf': + return self._eval_const_expr(args[0]) + return None + + def _is_macro_defined(self, name: str) -> bool: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if hasattr(Gen, '_define_constants') and name in Gen._define_constants: + return True + if self.Trans.SymbolTable.is_define(name): + return True + platform_macros: dict[str, int] = self._get_platform_macros() + return name in platform_macros + + def _get_macro_value(self, name: str) -> int | float | None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if hasattr(Gen, '_define_constants') and name in Gen._define_constants: + val: int | float | None = Gen._define_constants[name] + if isinstance(val, (int, float)): + return val + info: "SymbolTable.SymbolInfo" = self.Trans.SymbolTable.lookup(name) + if info: + if info.IsDefine: + val: int | float | None = info.DefineValue + if isinstance(val, (int, float)): + return val + platform_macros: dict[str, int] = self._get_platform_macros() + if name in platform_macros: + return platform_macros[name] + return None + + def _get_platform_macros(self) -> dict[str, int]: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + macros: dict[str, int] = {} + pi: dict[str, object] = Gen._platform_info + ptr_size: int = getattr(Gen, 'ptr_size', 8) + # 根据目标平台设置宏（从三元组推导） + if pi.get('is_windows'): + macros['_WIN32'] = 1 + if not pi.get('is_32bit'): + macros['_WIN64'] = 1 + macros['WIN64'] = 1 + macros['WIN32'] = 1 + if pi.get('is_linux'): + macros['__linux__'] = 1 + if pi.get('is_macos'): + macros['__APPLE__'] = 1 + macros['__MACH__'] = 1 + if pi.get('is_x86_64'): + macros['__x86_64__'] = 1 + macros['__x86_64'] = 1 + macros['_M_X64'] = 1 + elif pi.get('is_arm64'): + macros['__aarch64__'] = 1 + macros['_M_ARM64'] = 1 + if not pi.get('is_32bit') and pi.get('is_linux'): + macros['__LP64__'] = 1 + elif pi.get('is_32bit'): + macros['_ILP32'] = 1 + macros['SIZEOF_VOID_P'] = ptr_size + macros['__SIZEOF_POINTER__'] = ptr_size + macros['NDEBUG'] = 0 + return macros + + def _eval_const_expr(self, node: ast.expr) -> int | None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + ctx: EvalContext = EvalContext(Gen=Gen, symtab=self.Trans.SymbolTable, translator=self.Trans) + result: int | float | bool | None = ConstEvaluator.eval_full(node, ctx) + if result is None: + # 保留 CIf 嵌套调用处理（HandlesIf 特有逻辑） + if isinstance(node, ast.Call) and self._is_cif_call(node): + return self._evaluate_cif_condition(node) + return None + # HandlesIf 需要 bool/float → int 转换（#if 条件编译语义） + if isinstance(result, bool): + return 1 if result else 0 + if isinstance(result, float): + return 1 if result != 0.0 else 0 + return result + + def _HandleIfLlvm(self, Node: ast.If) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if self._is_cif_call(Node.test): + compile_time_val: int | None = self._evaluate_cif_condition(Node.test) + if compile_time_val is not None: + if compile_time_val: + self.Trans.BodyHandler.HandleBodyLlvm(Node.body) + elif Node.orelse: + if isinstance(Node.orelse, list) and len(Node.orelse) == 1 and isinstance(Node.orelse[0], ast.If): + self._HandleIfLlvm(Node.orelse[0]) + else: + self.Trans.BodyHandler.HandleBodyLlvm(Node.orelse) + return + Cond: ir.Value | None = self.Trans.ExprHandler.HandleExprLlvm(Node.test) + if not Cond: + return + is_const: bool = isinstance(Cond, ir.Constant) and isinstance(Cond.type, ir.IntType) and Cond.type.width == 1 + if is_const: + if Cond.constant: + self.Trans.BodyHandler.HandleBodyLlvm(Node.body) + elif Node.orelse: + if isinstance(Node.orelse, list) and len(Node.orelse) == 1 and isinstance(Node.orelse[0], ast.If): + self._HandleIfLlvm(Node.orelse[0]) + else: + self.Trans.BodyHandler.HandleBodyLlvm(Node.orelse) + return + if not isinstance(Cond.type, ir.IntType) or Cond.type.width != 1: + ClassName: str | None = self.Trans.ExprHandler._get_var_class(Node.test, Gen) + if ClassName and Gen._has_function(f'{ClassName}.__bool__'): + obj_val: ir.Value = Cond + if isinstance(obj_val.type, ir.PointerType) and isinstance(obj_val.type.pointee, ir.IntType) and obj_val.type.pointee.width == 8: + obj_val = Gen.builder.bitcast(obj_val, ir.PointerType(Gen.structs[ClassName]), name=f"cast_{ClassName}") + Cond = Gen.builder.call(Gen._get_function(f'{ClassName}.__bool__'), [obj_val], name=f"call_{ClassName}.__bool__") + if isinstance(Cond.type, ir.IntType) and Cond.type.width != 1: + Cond = Gen.builder.icmp_signed('!=', Cond, Gen._ZeroConst(Cond.type), name="bool_result") + else: + if isinstance(Cond.type, (ir.FloatType, ir.DoubleType)): + Cond = Gen.builder.fcmp_ordered('!=', Cond, ir.Constant(Cond.type, 0.0), name="ifcond") + else: + Cond = Gen.builder.icmp_signed('!=', Cond, Gen._ZeroConst(Cond.type), name="ifcond") + ThenBB: ir.Block = Gen.func.append_basic_block(name="then") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="else") if Node.orelse else None + MergeBB: ir.Block = Gen.func.append_basic_block(name="endif") + if not Gen.builder.block.is_terminated: + Gen.builder.cbranch(Cond, ThenBB, ElseBB if ElseBB else MergeBB) + Gen.builder.position_at_start(ThenBB) + self.Trans.BodyHandler.HandleBodyLlvm(Node.body) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(MergeBB) + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.Trans.BodyHandler.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(MergeBB) + if not MergeBB.is_terminated: + Gen.builder.position_at_start(MergeBB) + else: + Gen.builder.position_at_end(MergeBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesImports.py b/lib/core/Handles/HandlesImports.py new file mode 100644 index 0000000..55b17fd --- /dev/null +++ b/lib/core/Handles/HandlesImports.py @@ -0,0 +1,2589 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +import ast +import inspect +import os +import re +import hashlib +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo, FuncMeta +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode +from lib.includes import t +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import ParseListAnnotation, ExtractTypeNameFromBinOp, AnnotationContainsName + + +class ImportHandle(BaseHandle): + _struct_Load_cache = {} + _pyi_cache = {} + _pyi_cache_dir = None + _project_root_cache = None + # 类级 stub 索引：class_name -> list of (full_name, struct_body, file_sha1, from_output) + # 一次扫描所有 .stub.ll 文件构建索引，避免 _TryLoadStructFromStub 每次调用都全量扫描 + # （原 609 次调用 × ~100 stub 文件 × ~50 行 = 5.5M 正则匹配 → 一次扫描 ~5000 匹配） + _stub_index: dict[str, list[tuple[str, str, str, bool]]] = {} + _stub_index_signature: tuple[int, int] | None = None # (output_stub_count, temp_stub_count) + + def __init__(self, translator: "Translator") -> None: + super().__init__(translator) + # 实例级去重集合：每个 translator 独立跟踪已加载的 .pyi 和已发射的模块声明。 + # 若作为类变量共享，main.py 编译时加载的 __nodes.py 会阻止 __parser.py 编译时 + # 重新加载，导致函数未注册到 __parser.py 的 Gen.functions（NamedExpr bug 根因）。 + self._loaded_pyi_paths: set[str] = set() + self._emitted_module_decls: set[tuple] = set() + + def _reset_file_cache(self) -> None: + self._struct_Load_cache.clear() + self._loaded_pyi_paths.clear() + self._emitted_module_decls.clear() + # stub 索引在文件间保留（同一 worker 进程内），但标记需检查是否过期 + # 不清除 _stub_index，通过 _stub_index_signature 在下次使用时校验新鲜度 + + def _find_project_root(self) -> str: + if self._project_root_cache is not None: + return self._project_root_cache + # 优先使用 translator 上显式设置的 _temp_dir（由 Phase2Translator._translate_file 传入）， + # 这样可以正确处理 include 文件——其源文件目录不在项目根目录下， + # 从源文件目录向上搜索 temp/output 会失败并回退到错误的路径。 + temp_dir_attr = getattr(self.Trans, '_temp_dir', None) + if temp_dir_attr and os.path.isdir(temp_dir_attr): + project_root = os.path.dirname(os.path.abspath(temp_dir_attr)) + if os.path.isdir(os.path.join(project_root, 'temp')) and os.path.isdir(os.path.join(project_root, 'output')): + self._project_root_cache = project_root + return project_root + current_file = getattr(self.Trans, 'CurrentFile', '') or '' + if current_file: + current_dir = os.path.dirname(os.path.abspath(current_file)) + search_dir = current_dir + for _ in range(10): + if os.path.isdir(os.path.join(search_dir, 'temp')) and os.path.isdir(os.path.join(search_dir, 'output')): + self._project_root_cache = search_dir + return search_dir + parent = os.path.dirname(search_dir) + if parent == search_dir: + break + search_dir = parent + # 回退到当前工作目录（编译运行时的项目根目录） + cwd = os.getcwd() + if os.path.isdir(os.path.join(cwd, 'temp')) and os.path.isdir(os.path.join(cwd, 'output')): + self._project_root_cache = cwd + return cwd + if current_file: + self._project_root_cache = current_dir + return current_dir + self._project_root_cache = cwd + return cwd + + def _EmitImportDeclarationsLlvm(self, Node: ast.Import, Gen: LlvmGeneratorMixin) -> None: + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + for alias in Node.names: + name = alias.name + if name in ('c', 't'): + if alias.asname: + self.Trans.SymbolTable.import_aliases[alias.asname] = name + AliasInfo = CTypeInfo() + AliasInfo.IsModuleAlias = True + AliasInfo.ResolvedModule = name + self.Trans.SymbolTable.insert(alias.asname, AliasInfo) + continue + self.Trans._ImportedModules.add(name) + if alias.asname: + self.Trans.SymbolTable.import_aliases[alias.asname] = name + # 同时在符号表中注册模块别名，以便 CTypeInfo.FromNode 能解析 + AliasInfo = CTypeInfo() + AliasInfo.IsModuleAlias = True + AliasInfo.ResolvedModule = name + self.Trans.SymbolTable.insert(alias.asname, AliasInfo) + current_module = self._get_current_module_name() + self._EmitModuleDeclarationsLlvm(name, Gen, register_module_name=current_module) + + def _RegisterFromImportAliases(self, Node: ast.ImportFrom, Gen: LlvmGeneratorMixin, module: str) -> None: + """Register 'from module import y as z' aliases after module declarations are Loaded""" + if not Node.names: + return + for alias in Node.names: + name = alias.name + asname = alias.asname + # For 'from X import Y' (no asname), register CDefine constants + # into _define_constants so they can be found by _HandleNameLlvm + if not asname or asname == name: + # Check if this name is a CDefine constant in SymbolTable + sym_info = self.Trans.SymbolTable.lookup(name) + if sym_info and sym_info.IsDefine and sym_info.DefineValue is not None: + define_constants = vars(Gen).setdefault('_define_constants', {}) + if name not in define_constants: + define_constants[name] = sym_info.DefineValue + # Also check with module prefix + for prefix_key in [f"{module}.{name}", name]: + sym_info2 = self.Trans.SymbolTable.lookup(prefix_key) + if sym_info2 and sym_info2.IsDefine and sym_info2.DefineValue is not None: + define_constants = vars(Gen).setdefault('_define_constants', {}) + if name not in define_constants: + define_constants[name] = sym_info2.DefineValue + break + continue + if not hasattr(self.Trans, '_t_c_imported_names'): + self.Trans._t_c_imported_names = {} + self.Trans._t_c_imported_names[asname] = (module, name) + if name in Gen.functions: + Gen.functions[asname] = Gen.functions[name] + if name in Gen.structs: + Gen.structs[asname] = Gen.structs[name] + info = self.Trans.SymbolTable.lookup(name) + if info: + self.Trans.SymbolTable.insert(asname, info) + if name in Gen.variables: + Gen.variables[asname] = Gen.variables[name] + if name in Gen.global_vars: + Gen.global_vars[asname] = Gen.global_vars[name] + for func_name in list(Gen.functions.keys()): + if func_name.startswith(f'{name}.'): + method_suffix = func_name[len(name):] + Gen.functions[f'{asname}{method_suffix}'] = Gen.functions[func_name] + for key in list(self.Trans.SymbolTable.keys()): + if key.startswith(f'{name}.'): + suffix = key[len(name):] + self.Trans.SymbolTable.insert(f'{asname}{suffix}', self.Trans.SymbolTable[key]) + for meta_dict in (Gen.class_members, Gen.class_member_defaults, + Gen.class_member_signeds, Gen.class_member_bitfields, + Gen.class_member_byteorders, Gen.class_member_bitoffsets, + Gen.class_methods): + if name in meta_dict: + meta_dict[asname] = meta_dict[name] + if name in Gen.class_vtable: + Gen.class_vtable.add(asname) + if name in Gen.class_packed: + Gen.class_packed.add(asname) + struct_sha1_map = getattr(Gen, '_struct_sha1_map', {}) + if name in struct_sha1_map: + struct_sha1_map[asname] = struct_sha1_map[name] + + def _EmitImportFromDeclarationsLlvm(self, Node: ast.ImportFrom, Gen: LlvmGeneratorMixin) -> None: + module = Node.module if Node.module else '' + if module in ('c', 't'): + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(module) + if not hasattr(self.Trans, '_t_c_imported_names'): + self.Trans._t_c_imported_names = {} + # 注册 t/c 类型别名到 _t_type_symbols（CType 子类） + t_type_syms = self.Trans.SymbolTable._t_type_symbols + for alias in Node.names: + name = alias.name + asname = alias.asname or name + self.Trans._t_c_imported_names[asname] = (module, name) + # 仅注册 CType 子类（CEnum/CUnion/CStruct/REnum 等） + if module == 't': + cls = getattr(t, name, None) + if isinstance(cls, type) and issubclass(cls, t.CType): + t_type_syms[asname] = cls + return + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + current_module = self._get_current_module_name() + if Node.level and Node.level > 0: + current_file = getattr(self.Trans, 'CurrentFile', '') or '' + current_dir = os.path.dirname(os.path.abspath(current_file)) + parent_dir = current_dir + for _ in range(Node.level - 1): + parent_dir = os.path.dirname(parent_dir) + if module: + SearchExtensions = ['.pyi', '.py'] + sub_path_base = os.path.join(parent_dir, module) + for ext in SearchExtensions: + candidate = sub_path_base + ext + if os.path.isfile(candidate): + self.Trans._ImportedModules.add(module) + self._LoadModuleDeclarationsFromFile(candidate, Gen, module_name=module, register_module_name=current_module) + self._RegisterFromImportAliases(Node, Gen, module) + return + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + # 尝试完整模块名和短模块名 + full_mod_name = f"{current_module}.{module}" if current_module else module + target_sha1 = ModuleSha1Map.get(full_mod_name) or ModuleSha1Map.get(module) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + self.Trans._ImportedModules.add(module) + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=module, register_module_name=current_module) + self._RegisterFromImportAliases(Node, Gen, module) + return + else: + mod_dir = parent_dir + + SearchExtensions = ['.pyi', '.py'] + pkg_name = '' + if not module: + # When 'from . import name' is used, Load the package's __init__ + # to make package-level names (functions, classes, constants) available + pkg_name = self._LoadPackageInitForRelativeImport(mod_dir, Gen, current_module) + + # Also try to Load each alias as a submodule + for alias in Node.names: + found_sub = False + for ext in SearchExtensions: + sub_path = os.path.join(mod_dir, alias.name + ext) + if os.path.isfile(sub_path): + self.Trans._ImportedModules.add(alias.name) + self._LoadModuleDeclarationsFromFile(sub_path, Gen, module_name=alias.name, register_module_name=current_module) + if alias.asname: + self.Trans.SymbolTable.import_aliases[alias.asname] = alias.name + found_sub = True + break + if not found_sub: + # SHA1 map fallback for submodule + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + sub_ModulePath = f"{pkg_name}.{alias.name}" if pkg_name else alias.name + target_sha1 = ModuleSha1Map.get(sub_ModulePath) or ModuleSha1Map.get(alias.name) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + self.Trans._ImportedModules.add(sub_ModulePath) + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=alias.name, register_module_name=current_module) + if alias.asname: + self.Trans.SymbolTable.import_aliases[alias.asname] = alias.name + self._RegisterFromImportAliases(Node, Gen, module or pkg_name) + return + self.Trans._ImportedModules.add(module) + self._EmitModuleDeclarationsLlvm(module, Gen, register_module_name=current_module) + self._RegisterFromImportAliases(Node, Gen, module) + + def _RegisterCrossModuleDocstrings(self, pyi_path: str, Gen: LlvmGeneratorMixin, module_prefix: str | None) -> None: + """从 {sha1}.doc.json 加载跨模块 docstring 注册到 Gen._doc_strings。 + + - 长名 {module_prefix}.{symbol} 总是注册 + - 短名 {symbol} 只在不存在时注册（避免覆盖本模块的） + - slice_level != 0（优化开启）时注册为 None + - 跨模块 docstring 在 _get_doc_ptr 访问时创建本地常量副本 + """ + if not module_prefix: + return + basename: str = os.path.basename(pyi_path) + # 若传入 .py 文件，则从 _source_module_sig_files 查找对应的 {sha1}.pyi + if basename.endswith('.py'): + source_sig_files = getattr(self.Trans, '_source_module_sig_files', None) + if source_sig_files: + sig_path = source_sig_files.get(module_prefix) + if sig_path and os.path.isfile(sig_path): + pyi_path = sig_path + basename = os.path.basename(pyi_path) + else: + return + else: + return + if not basename.endswith('.pyi'): + return + sha1: str = basename[:-len('.pyi')] + shared_doc_meta = getattr(self.Trans, '_shared_doc_meta', None) + if not shared_doc_meta: + return + docs: dict[str, str] | None = shared_doc_meta.get(sha1) + if not docs: + return + trans = getattr(Gen, '_Trans', None) + slice_level: int = getattr(trans, 'SliceLevel', 3) + optimized: bool = slice_level != 0 + for symbol, doc in docs.items(): + long_key: str = f"{module_prefix}.{symbol}" + if optimized: + Gen._doc_strings[long_key] = None + if symbol not in Gen._doc_strings: + Gen._doc_strings[symbol] = None + else: + Gen._doc_strings[long_key] = doc + if symbol not in Gen._doc_strings: + Gen._doc_strings[symbol] = doc + + def _LoadModuleDeclarationsFromFile(self, pyi_path: str, Gen: LlvmGeneratorMixin, module_name: str | None = None, register_module_name: str | None = None, actual_module_name: str | None = None, reexport_package: str | None = None) -> None: + if not pyi_path or not os.path.isfile(pyi_path): + return + # 去重：同一 .pyi 文件只加载一次（菱形依赖下避免重复 read+parse） + abs_pyi = os.path.abspath(pyi_path) + if abs_pyi in self._loaded_pyi_paths: + return + self._loaded_pyi_paths.add(abs_pyi) + # 注册跨模块 __doc__ docstring 到 Gen._doc_strings + self._RegisterCrossModuleDocstrings(pyi_path, Gen, register_module_name or module_name) + try: + with open(pyi_path, 'r', encoding='utf-8') as f: + ModuleCode = f.read() + ModuleTree = ast.parse(ModuleCode) + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + return + + # 计算源模块名（从 .pyi 文件路径推导）。 + # 治本修复：module_name 参数可能来自 alias.name（如 `from stmts import Module` + # 中的 'Module' 是类名而非模块名），直接传给 _EmitExternalClassDeclLlvm 会导致 + # ModuleSha1Map 查找失败 → _self_struct_sha1=None → 使用错误的同名 struct。 + # 使用文件名推导的模块名（如 'stmts'）作为 source_module_name 才是正确的。 + # 对于 sha1 命名的 .pyi 文件（如 e64295c03b564a53.pyi），文件名是 sha1 而非 + # 模块名，需回退到 actual_module_name 或 module_name。 + file_module_name = os.path.splitext(os.path.basename(pyi_path))[0] + if len(file_module_name) == 16 and all(_c in '0123456789abcdef' for _c in file_module_name): + file_module_name = actual_module_name or module_name or file_module_name + + for node in ModuleTree.body: + try: + if isinstance(node, ast.FunctionDef): + if node.name in ('va_start', 'va_arg', 'va_end'): + continue + effective_register = register_module_name or module_name + reexport_names = [] + if module_name and module_name != file_module_name and module_name != effective_register: + reexport_names.append(module_name) + # When Loading a submodule for 'from .xxx import yyy' inside a package's __init__.py, + # re-export functions under the package name so 'pkg.func()' resolves correctly + if reexport_package and reexport_package != file_module_name and reexport_package not in reexport_names: + reexport_names.append(reexport_package) + self._EmitExternalFuncDeclLlvm(node, Gen, source_module_name=file_module_name, register_module_name=effective_register, reexport_module_names=reexport_names if reexport_names else None) + elif isinstance(node, ast.AnnAssign): + self._EmitExternalGlobalDeclLlvm(node, Gen, module_name=module_name, ModulePath=pyi_path) + elif isinstance(node, ast.Assign): + self._EmitExternalGlobalDeclLlvm(node, Gen, module_name=module_name, ModulePath=pyi_path) + elif isinstance(node, ast.ClassDef): + if hasattr(node, 'type_params') and node.type_params: + if not hasattr(self.Trans.ClassHandler, '_generic_class_templates'): + self.Trans.ClassHandler._generic_class_templates = {} + type_params = [tp.name for tp in node.type_params] + self.Trans.ClassHandler._generic_class_templates[node.name] = { + 'node': node, + 'type_params': type_params, + } + # 治本修复：使用 file_module_name（源模块名）而非 module_name（可能是别名/类名）。 + # 这样 _EmitExternalClassDeclLlvm 内的 ModuleSha1Map 查找才能命中， + # _EmitExternalFuncDeclLlvm 才能拿到正确的 source_module_name → 正确的 struct sha1。 + # 注意：actual_module_name 保持原值（可能为 None），不回退到 module_name（可能是别名） + self._EmitExternalClassDeclLlvm(node, Gen, module_name=file_module_name, actual_module_name=actual_module_name) + elif isinstance(node, ast.ImportFrom): + if node.level > 0: + current_dir = os.path.dirname(os.path.abspath(pyi_path)) + parent_dir = current_dir + for _ in range(node.level - 1): + parent_dir = os.path.dirname(parent_dir) + if node.module: + SearchExtensions = ['.pyi', '.py'] + sub_path_base = os.path.join(parent_dir, node.module) + found_sub = False + for ext in SearchExtensions: + candidate = sub_path_base + ext + if os.path.isfile(candidate): + actual_module = f"{register_module_name}.{node.module}" if register_module_name else node.module + self._RegisterSubModuleSha1(candidate, actual_module, node.module, Gen) + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(actual_module) + for alias in node.names: + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(candidate, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + else: + self._LoadModuleDeclarationsFromFile(candidate, Gen, module_name=alias.name, register_module_name=register_module_name, actual_module_name=actual_module, reexport_package=reexport_package or module_name) + found_sub = True + break + # SHA1 map 回退 + if not found_sub: + actual_module = f"{register_module_name}.{node.module}" if register_module_name else node.module + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + target_sha1 = ModuleSha1Map.get(actual_module) or ModuleSha1Map.get(node.module) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(actual_module) + for alias in node.names: + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + else: + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=alias.name, register_module_name=register_module_name, actual_module_name=actual_module, reexport_package=reexport_package or module_name) + else: + mod_dir = parent_dir + SearchExtensions = ['.pyi', '.py'] + # Load __init__.py to make package-level names available + self._LoadPackageInitForRelativeImport(mod_dir, Gen, register_module_name) + for alias in node.names: + found_alias = False + for ext in SearchExtensions: + sub_path = os.path.join(mod_dir, alias.name + ext) + if os.path.isfile(sub_path): + sub_ModulePath = f"{register_module_name}.{alias.name}" if register_module_name else alias.name + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(sub_ModulePath) + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sub_path, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + else: + self._LoadModuleDeclarationsFromFile(sub_path, Gen, module_name=alias.name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + found_alias = True + break + # SHA1 map 回退 + if not found_alias: + sub_ModulePath = f"{register_module_name}.{alias.name}" if register_module_name else alias.name + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + target_sha1 = ModuleSha1Map.get(sub_ModulePath) or ModuleSha1Map.get(alias.name) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(sub_ModulePath) + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + else: + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=alias.name, register_module_name=register_module_name, reexport_package=reexport_package or module_name) + except Exception as _e: + _vlog().warning( + f"HandlesImports[_LoadModuleDeclarationsFromFile]: 忽略节点 " + f"{type(node).__name__} {getattr(node, 'name', '?')} 异常: {_e}", + exc_info=_e + ) + continue + + def _get_current_module_name(self) -> str | None: + current_file = getattr(self.Trans, 'CurrentFile', '') or '' + if current_file: + return os.path.splitext(os.path.basename(current_file))[0] + return None + + def _RegisterSubModuleSha1(self, candidate_path: str, actual_module: str, short_module: str, Gen: LlvmGeneratorMixin) -> None: + """Register a submodule's SHA1 in ModuleSha1Map so cross-module class references work""" + try: + with open(candidate_path, 'r', encoding='utf-8') as f: + sub_content = f.read() + sub_sha1 = hashlib.sha1(sub_content.encode('utf-8')).hexdigest()[:16] + if hasattr(Gen, 'ModuleSha1Map'): + Gen.ModuleSha1Map[actual_module] = sub_sha1 + if short_module and short_module not in Gen.ModuleSha1Map: + Gen.ModuleSha1Map[short_module] = sub_sha1 + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + + def _LoadPackageInitForRelativeImport(self, mod_dir: str, Gen: LlvmGeneratorMixin, register_module_name: str | None) -> str: + """Load __init__.py from the package directory for 'from . import name' resolution. + This makes package-level names (functions, classes, constants) available. + Returns the package name.""" + pkg_name = os.path.basename(mod_dir) + init_Loaded = False + + # 跳过当前正在编译的文件，避免重复加载导致 class_members 重复 + current_sha1 = getattr(Gen, 'module_sha1', None) + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + pkg_sha1 = ModuleSha1Map.get(pkg_name) + if pkg_sha1 and pkg_sha1 == current_sha1: + return pkg_name + + # Try SHA1 map first (correct mangled names from stub) + if pkg_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{pkg_sha1}.pyi") + if os.path.isfile(sha1_pyi): + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=pkg_name, register_module_name=register_module_name) + init_Loaded = True + + # Fallback: try Loading __init__.py directly from the package directory + if not init_Loaded: + for ext in ['.pyi', '.py']: + init_path = os.path.join(mod_dir, '__init__' + ext) + if os.path.isfile(init_path): + self._LoadModuleDeclarationsFromFile(init_path, Gen, module_name=pkg_name, register_module_name=register_module_name) + init_Loaded = True + break + + return pkg_name + + def _EmitModuleDeclarationsLlvm(self, module_name: str, Gen: LlvmGeneratorMixin, register_module_name: str | None = None) -> None: + # 去重：同一模块声明只发射一次（避免多个 import 同一模块时重复加载） + decl_key = (module_name, register_module_name) + if decl_key in self._emitted_module_decls: + return + self._emitted_module_decls.add(decl_key) + if module_name in ('pyzlib',): + source_sig_files = getattr(self.Trans, '_source_module_sig_files', None) + ModulePath = module_name.replace('.', os.sep) + '.pyi' + PackageInitPath = module_name.replace('.', os.sep) + os.sep + '__init__.pyi' + ProjectRoot = os.path.dirname(os.path.abspath(getattr(self.Trans, 'CurrentFile', '') or '')) or os.getcwd() + + HandlesFile = os.path.abspath(inspect.getfile(self.__class__)) + TransPyCRoot = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(HandlesFile)))) + + FullModulePath = None + SearchDirs = [os.path.join(TransPyCRoot, 'includes'), os.path.join(TransPyCRoot, 'lib', 'includes')] + list(self.Trans.LibraryPaths) + + for LibPath in SearchDirs: + SearchPath = os.path.join(ProjectRoot, LibPath) if not os.path.isabs(LibPath) else LibPath + CandidatePath = os.path.join(SearchPath, ModulePath) + if os.path.isfile(CandidatePath): + FullModulePath = CandidatePath + break + CandidatePath = os.path.join(SearchPath, PackageInitPath) + if os.path.isfile(CandidatePath): + FullModulePath = CandidatePath + break + ModuleParts = ModulePath.split(os.sep) + if len(ModuleParts) > 1: + LibPathBasename = os.path.basename(LibPath.rstrip('/\\')) + if LibPathBasename == ModuleParts[0]: + RemainingPath = os.sep.join(ModuleParts[1:]) + CandidatePath = os.path.join(SearchPath, RemainingPath) + if os.path.isfile(CandidatePath): + FullModulePath = CandidatePath + break + if not FullModulePath: + ModulePathPy = module_name.replace('.', os.sep) + '.py' + PackageInitPathPy = module_name.replace('.', os.sep) + os.sep + '__init__.py' + SearchDirsForPy = [os.path.join(TransPyCRoot, 'includes'), os.path.join(TransPyCRoot, 'lib', 'includes')] + list(self.Trans.LibraryPaths) + for LibPath in SearchDirsForPy: + SearchPath = os.path.join(ProjectRoot, LibPath) if not os.path.isabs(LibPath) else LibPath + CandidatePath = os.path.join(SearchPath, ModulePathPy) + if os.path.isfile(CandidatePath): + FullModulePath = CandidatePath + break + CandidatePath = os.path.join(SearchPath, PackageInitPathPy) + if os.path.isfile(CandidatePath): + FullModulePath = CandidatePath + break + if not FullModulePath or not os.path.isfile(FullModulePath): + source_sig_files = getattr(self.Trans, '_source_module_sig_files', None) + if source_sig_files: + sig_path = source_sig_files.get(module_name) + if sig_path and os.path.isfile(sig_path): + FullModulePath = sig_path + else: + for key, val in source_sig_files.items(): + if key.endswith('.' + module_name) or key == module_name: + FullModulePath = val + break + if not FullModulePath or not os.path.isfile(FullModulePath): + if module_name in ('pyzlib', 'zdeflate', 'zinflate', 'zchecksum', 'zdef', 'zhuff'): + pass + if not FullModulePath: + CurrentFile = getattr(self.Trans, 'CurrentFile', None) or '' + if CurrentFile: + CurrentFileDir = os.path.dirname(os.path.abspath(CurrentFile)) + SearchFromDirs = [CurrentFileDir, os.getcwd()] + else: + SearchFromDirs = [os.getcwd()] + SearchDirsForSubmodule = SearchFromDirs + [os.path.join(TransPyCRoot, 'includes'), os.path.join(TransPyCRoot, 'lib', 'includes')] + list(self.Trans.LibraryPaths) + for SearchPath in SearchDirsForSubmodule: + if os.path.isdir(SearchPath): + for root, dirs, files in os.walk(SearchPath): + if '__pycache__' in root: + continue + if module_name + '.py' in files: + FullModulePath = os.path.join(root, module_name + '.py') + break + if module_name + '.pyi' in files: + FullModulePath = os.path.join(root, module_name + '.pyi') + break + if FullModulePath: + break + # 注册跨模块 __doc__ docstring（module_name 是被导入模块名） + if FullModulePath: + self._RegisterCrossModuleDocstrings(FullModulePath, Gen, module_name) + self._LoadDeclarationsFromStubLlvm(module_name, Gen) + if module_name == 'pyzlib': + pass + try: + with open(FullModulePath, 'r', encoding='utf-8') as f: + ModuleCode = f.read() + ModuleTree = ast.parse(ModuleCode) + if module_name == 'pyzlib': + func_names = [n.name for n in ModuleTree.body if isinstance(n, ast.FunctionDef)] + has_functions_or_classes = False + for node in ModuleTree.body: + if module_name == 'pyzlib': + node_type = type(node).__name__ + name = getattr(node, 'name', getattr(node, 'attr', '?')) + try: + if isinstance(node, ast.FunctionDef): + if node.name in ('va_start', 'va_arg', 'va_end'): + continue + effective_register = register_module_name or module_name + self._EmitExternalFuncDeclLlvm(node, Gen, source_module_name=module_name, register_module_name=effective_register) + has_functions_or_classes = True + elif isinstance(node, ast.AnnAssign): + self._EmitExternalGlobalDeclLlvm(node, Gen, module_name=module_name, ModulePath=FullModulePath) + has_functions_or_classes = True + elif isinstance(node, ast.Assign): + self._EmitExternalGlobalDeclLlvm(node, Gen, module_name=module_name, ModulePath=FullModulePath) + has_functions_or_classes = True + elif isinstance(node, ast.ClassDef): + if hasattr(node, 'type_params') and node.type_params: + if not hasattr(self.Trans.ClassHandler, '_generic_class_templates'): + self.Trans.ClassHandler._generic_class_templates = {} + type_params = [tp.name for tp in node.type_params] + self.Trans.ClassHandler._generic_class_templates[node.name] = { + 'node': node, + 'type_params': type_params, + } + self._EmitExternalClassDeclLlvm(node, Gen, module_name=module_name) + has_functions_or_classes = True + elif isinstance(node, ast.ImportFrom): + if node.level > 0: + current_dir = os.path.dirname(os.path.abspath(FullModulePath)) + parent_dir = current_dir + for _ in range(node.level - 1): + parent_dir = os.path.dirname(parent_dir) + if node.module: + SearchExtensions = ['.pyi', '.py'] + sub_path_base = os.path.join(parent_dir, node.module) + found_sub = False + for ext in SearchExtensions: + candidate = sub_path_base + ext + if os.path.isfile(candidate): + actual_module = f"{module_name}.{node.module}" if module_name else node.module + self._RegisterSubModuleSha1(candidate, actual_module, node.module, Gen) + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(actual_module) + for alias in node.names: + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(candidate, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=module_name) + else: + self._LoadModuleDeclarationsFromFile(candidate, Gen, module_name=alias.name, register_module_name=register_module_name, actual_module_name=actual_module, reexport_package=module_name) + has_functions_or_classes = True + found_sub = True + break + # SHA1 map 回退：当文件查找失败时，从 SHA1 映射中查找子模块 stub + if not found_sub: + actual_module = f"{module_name}.{node.module}" if module_name else node.module + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + # 尝试完整模块名和短模块名 + target_sha1 = ModuleSha1Map.get(actual_module) or ModuleSha1Map.get(node.module) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(actual_module) + for alias in node.names: + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=module_name) + else: + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=alias.name, register_module_name=register_module_name, actual_module_name=actual_module, reexport_package=module_name) + has_functions_or_classes = True + else: + mod_dir = parent_dir + SearchExtensions = ['.pyi', '.py'] + # Load __init__.py to make package-level names available + self._LoadPackageInitForRelativeImport(mod_dir, Gen, register_module_name or module_name) + for alias in node.names: + found_alias = False + for ext in SearchExtensions: + sub_path = os.path.join(mod_dir, alias.name + ext) + if os.path.isfile(sub_path): + sub_ModulePath = f"{module_name}.{alias.name}" if module_name else alias.name + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(sub_ModulePath) + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sub_path, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=module_name) + else: + self._LoadModuleDeclarationsFromFile(sub_path, Gen, module_name=alias.name, register_module_name=register_module_name, reexport_package=module_name) + has_functions_or_classes = True + found_alias = True + break + # SHA1 map 回退 + if not found_alias: + sub_ModulePath = f"{module_name}.{alias.name}" if module_name else alias.name + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + target_sha1 = ModuleSha1Map.get(sub_ModulePath) or ModuleSha1Map.get(alias.name) + if target_sha1: + temp_dir = getattr(Gen, '_temp_dir', None) + if temp_dir: + sha1_pyi = os.path.join(temp_dir, f"{target_sha1}.pyi") + if os.path.isfile(sha1_pyi): + if not getattr(self.Trans, '_ImportedModules', None): + self.Trans._ImportedModules = set() + self.Trans._ImportedModules.add(sub_ModulePath) + if alias.name == '*': + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=module_name, register_module_name=register_module_name, reexport_package=module_name) + else: + self._LoadModuleDeclarationsFromFile(sha1_pyi, Gen, module_name=alias.name, register_module_name=register_module_name, reexport_package=module_name) + has_functions_or_classes = True + except Exception as e: + if module_name == 'pyzlib': + node_type = type(node).__name__ + name = getattr(node, 'name', getattr(node, 'attr', '?')) + _vlog().warning(f"HandlesImports: 忽略异常 {e}", exc_info=e) + continue + if not has_functions_or_classes and module_name: + self._LoadDeclarationsFromStubLlvm(module_name, Gen) + elif 'includes' in FullModulePath and module_name: + self._LoadDeclarationsFromStubLlvm(module_name, Gen) + except Exception as e: + if module_name: + self._LoadDeclarationsFromStubLlvm(module_name, Gen) + + def _EmitExternalGlobalDeclLlvm(self, Node: ast.AnnAssign | ast.Assign, Gen: LlvmGeneratorMixin, module_name: str | None = None, ModulePath: str | None = None) -> None: + """Emit external global variable declaration from imported module""" + if isinstance(Node, ast.AnnAssign) and isinstance(Node.target, ast.Name): + VarName = Node.target.id + # 不再跳过 _ 开头的全局变量，跨模块引用需要声明它们 + var_type = ir.IntType(32) + IsPtr = False + is_string_val = False + TypeInfo = None + if VarName in Gen.module.globals: + existing = Gen.module.globals[VarName] + if isinstance(existing, ir.GlobalVariable): + return + if Node.annotation: + parse_result = ParseListAnnotation(Node.annotation) + if parse_result and not parse_result.is_pointer: + # list[type, count] 数组声明 + elem_type_node = parse_result.elem_type_node + count_node = parse_result.count_node + ElemTypeInfo = CTypeInfo.FromNode(elem_type_node, self.Trans.SymbolTable) + if ElemTypeInfo: + elem_type = Gen._ctype_to_llvm(ElemTypeInfo) + if isinstance(elem_type, ir.VoidType): + elem_type = ir.IntType(8) + if isinstance(count_node, ast.Constant) and isinstance(count_node.value, int): + var_type = ir.ArrayType(elem_type, count_node.value) + elif isinstance(count_node, ast.Constant) and count_node.value is None: + var_type = ir.ArrayType(elem_type, 0) + else: + var_type = Gen._ctype_to_llvm(ElemTypeInfo) + IsPtr = False + else: + TypeInfo = CTypeInfo.FromNode(Node.annotation, self.Trans.SymbolTable) + if TypeInfo: + var_type = Gen._ctype_to_llvm(TypeInfo) + IsPtr = TypeInfo.IsPtr + if isinstance(var_type, ir.IdentifiedStructType) and (var_type.elements is None or len(var_type.elements) == 0): + var_type = ir.IntType(32) + if Node.value and ModulePath and IsPtr and isinstance(Node.value, ast.Constant) and isinstance(Node.value.value, str): + str_val = Node.value.value + '\x00' + str_bytes = str_val.encode('utf-8') + arr_type = ir.ArrayType(ir.IntType(8), len(str_bytes)) + str_gv_name = VarName + '_str' + if str_gv_name not in Gen.module.globals: + str_gv = ir.GlobalVariable(Gen.module, arr_type, name=str_gv_name) + str_gv.initializer = ir.Constant(arr_type, bytearray(str_bytes)) + str_gv.linkage = 'internal' + ptr_type = ir.PointerType(ir.IntType(8)) + if VarName not in Gen.module.globals: + ptr_gv = ir.GlobalVariable(Gen.module, ptr_type, name=VarName) + str_gv = Gen.module.globals[str_gv_name] + ptr_gv.initializer = ir.Constant.gep(str_gv, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), 0)]) + ptr_gv.linkage = 'internal' + Gen.variables[VarName] = None + return + + # 检查是否是 CDefine 常量或 CTypedef 类型别名 + IsCDefine = False + IsCTypedef = False + if Node.annotation: + if isinstance(Node.annotation, ast.Attribute): + if getattr(Node.annotation.value, 'id', None) == 't' and Node.annotation.attr == 'CDefine': + IsCDefine = True + elif getattr(Node.annotation.value, 'id', None) == 't' and Node.annotation.attr == 'CTypedef': + IsCTypedef = True + elif isinstance(Node.annotation, ast.Name): + if Node.annotation.id == 'CDefine': + IsCDefine = True + elif Node.annotation.id == 'CTypedef': + IsCTypedef = True + + if IsCTypedef: + # 检查符号是否已存在且已正确解析（OriginalType 已设置），避免覆盖 + existing_sym = self.Trans.SymbolTable.lookup(VarName) if hasattr(self.Trans.SymbolTable, 'lookup') else None + if existing_sym and isinstance(existing_sym, CTypeInfo) and existing_sym.IsTypedef and existing_sym.OriginalType: + return + if Node.value: + ValueTypeInfo = None + if isinstance(Node.value, ast.BinOp) and isinstance(Node.value.op, ast.BitOr): + ValueTypeInfo = self.Trans.TypeMergeHandler.MergeTypes(Node.value) + else: + ValueTypeInfo = self.Trans.TypeMergeHandler.GetCTypeInfo(Node.value) + if isinstance(ValueTypeInfo, CTypeInfo) and (ValueTypeInfo.BaseType or ValueTypeInfo.PtrCount > 0): + # GetCTypeInfo 对 t.CInt 等 Attribute 返回的 BaseType 可能是 CType 基类实例 + # （而非具体子类如 CInt），导致 _resolve_typedef_int_width 无法解析位宽。 + # 此时从 Node.value 提取类型名，设置 OriginalType 为字符串， + # 让 _resolve_typedef_int_width 通过 CTypeRegistry.GetClassByName 解析。 + if ValueTypeInfo.BaseType and type(ValueTypeInfo.BaseType) is t.CType: + if isinstance(Node.value, ast.Attribute) and getattr(Node.value.value, 'id', None) == 't': + type_name: str = Node.value.attr + resolved_cls = CTypeRegistry.GetClassByName(type_name) + if resolved_cls is not None: + ValueTypeInfo.OriginalType = type_name + ValueTypeInfo.IsTypedef = True + ValueTypeInfo.Name = VarName + # 设置 OriginalType，确保 _resolve_name 能解析（MergeTypes 不为 BinOp 结果设置 OriginalType） + if not ValueTypeInfo.OriginalType: + OriginalCopy: CTypeInfo = ValueTypeInfo.Copy() + OriginalCopy.IsTypedef = False + OriginalCopy.Name = '' + ValueTypeInfo.OriginalType = OriginalCopy + self.Trans.SymbolTable.insert(VarName, ValueTypeInfo) + if module_name: + FullName = f"{module_name}.{VarName}" + self.Trans.SymbolTable.insert(FullName, ValueTypeInfo) + return + TTypeInfo = self.Trans.TypeMergeHandler.GetCTypeInfo(Node.annotation) + if TTypeInfo: + TTypeInfo.IsTypedef = True + TTypeInfo.Name = VarName + self.Trans.SymbolTable.insert(VarName, TTypeInfo) + if module_name: + FullName = f"{module_name}.{VarName}" + self.Trans.SymbolTable.insert(FullName, TTypeInfo) + return + + if IsCDefine: + # CDefine 宏不生成 LLVM 全局变量，只注册到 SymbolTable + # 引用时会直接替换为立即数 + DefineValue = None + if Node.value: + DefineValue = self._ExtractConstValue(Node.value) + # 注册不带前缀的键名（例如 Z_NO_COMPRESSION） + # 但要先检查：如果符号已经存在且是 typedef，不覆盖它 + existing = self.Trans.SymbolTable.lookup(VarName) + if not (existing and existing.IsTypedef and existing.OriginalType): + self._RegisterCDefineSymbol(VarName, DefineValue, Node.lineno, ModulePath or 'unknown') + # 如果提供了模块名，也注册带模块前缀的键名 + if module_name: + FullName = f"{module_name}.{VarName}" + existing_full = self.Trans.SymbolTable.lookup(FullName) + if not (existing_full and existing_full.IsTypedef and existing_full.OriginalType): + self._RegisterCDefineSymbol(FullName, DefineValue, Node.lineno, ModulePath or 'unknown') + return # 不再继续生成 LLVM 全局变量 + + # t.State 标记的全局变量：只声明不定义（external linkage），避免 DSO/non-DSO 重定义 + if TypeInfo and TypeInfo.IsState: + if VarName not in Gen.module.globals: + gv = ir.GlobalVariable(Gen.module, var_type, name=VarName) + gv.linkage = 'external' + Gen._export_funcs.add(VarName) + return + + gv = ir.GlobalVariable(Gen.module, var_type, name=VarName) + # 外部全局变量声明：只声明不定义（external linkage），避免多重定义 + gv.linkage = 'external' + elif isinstance(node, ast.Assign) and Node.targets and isinstance(Node.targets[0], ast.Name): + VarName = Node.targets[0].id + if VarName.startswith('_'): + return + if VarName in Gen.module.globals: + return + gv = ir.GlobalVariable(Gen.module, ir.IntType(32), name=VarName) + gv.linkage = 'external' + + def _ExtractValue(self, value_node: ast.AST, target_type: ir.Type, var_name: str | None = None) -> ir.Constant | None: + """Extract constant value from AST node for global variable initializer""" + # 处理负号 + if isinstance(value_node, ast.UnaryOp) and isinstance(value_node.op, ast.USub): + if isinstance(value_node.operand, ast.Constant) and isinstance(value_node.operand.value, int): + neg_val = -value_node.operand.value + if isinstance(target_type, ir.IntType): + return ir.Constant(target_type, neg_val) + return ir.Constant(ir.IntType(32), neg_val) + # 处理正号 + elif isinstance(value_node, ast.UnaryOp) and isinstance(value_node.op, ast.UAdd): + return self._ExtractValue(value_node.operand, target_type) + + if isinstance(value_node, ast.Constant): + if isinstance(value_node.value, bool): + if isinstance(target_type, ir.IntType): + return ir.Constant(target_type, 1 if value_node.value else 0) + return ir.Constant(ir.IntType(32), 1 if value_node.value else 0) + elif isinstance(value_node.value, int): + if isinstance(target_type, ir.IntType): + return ir.Constant(target_type, value_node.value) + return ir.Constant(ir.IntType(32), value_node.value) + elif isinstance(value_node.value, str): + str_val = value_node.value + '\x00' + str_bytes = str_val.encode('utf-8') + arr_type = ir.ArrayType(ir.IntType(8), len(str_bytes)) + Gen = self.Trans.LlvmGen + gv_name = var_name + '_str' if var_name else f"str_global_{id(value_node)}" + if gv_name in Gen.module.globals: + gv = Gen.module.globals[gv_name] + else: + gv = ir.GlobalVariable(Gen.module, arr_type, name=gv_name) + gv.initializer = ir.Constant(arr_type, bytearray(str_bytes)) + gv.linkage = 'internal' + ptr_type = ir.PointerType(ir.IntType(8)) + return ir.Constant(ptr_type, gv.reference) + elif isinstance(value_node.value, bool): + return ir.Constant(ir.IntType(32), 1 if value_node.value else 0) + elif value_node.value is None: + if isinstance(target_type, ir.BaseStructType): + return ir.Constant(target_type, None) + if isinstance(target_type, ir.PointerType): + return ir.Constant(target_type, None) + return ir.Constant(target_type, 0) + elif isinstance(value_node, ast.Name): + if value_node.id == 'None': + if isinstance(target_type, ir.BaseStructType): + return ir.Constant(target_type, None) + if isinstance(target_type, ir.PointerType): + return ir.Constant(target_type, None) + return ir.Constant(target_type, 0) + return None + + def _ExtractConstValue(self, value_node: ast.AST) -> object: + """Extract constant value from AST node for SymbolTable registration + 支持常量、一元运算符、二元运算符和常量引用 + """ + if isinstance(value_node, ast.Constant): + return value_node.value + elif isinstance(value_node, ast.BinOp): + left = self._ExtractConstValue(value_node.left) + right = self._ExtractConstValue(value_node.right) + if left is None or right is None: + return None + if isinstance(value_node.op, ast.Add): + return left + right + elif isinstance(value_node.op, ast.Sub): + return left - right + elif isinstance(value_node.op, ast.Mult): + return left * right + elif isinstance(value_node.op, ast.Div): + return left // right if isinstance(left, int) and isinstance(right, int) else left / right + elif isinstance(value_node.op, ast.FloorDiv): + return left // right + elif isinstance(value_node.op, ast.Mod): + return left % right + elif isinstance(value_node.op, ast.Pow): + return left ** right + elif isinstance(value_node.op, ast.LShift): + return left << right + elif isinstance(value_node.op, ast.RShift): + return left >> right + elif isinstance(value_node.op, ast.BitOr): + return left | right + elif isinstance(value_node.op, ast.BitXor): + return left ^ right + elif isinstance(value_node.op, ast.BitAnd): + return left & right + elif isinstance(value_node, ast.UnaryOp): + operand = self._ExtractConstValue(value_node.operand) + if operand is None: + return None + if isinstance(value_node.op, ast.USub): + return -operand + elif isinstance(value_node.op, ast.UAdd): + return +operand + elif isinstance(value_node.op, ast.Invert): + return ~operand + elif isinstance(value_node, ast.Name): + if value_node.id in getattr(self.Trans, 'SymbolTable', {}): + info = self.Trans.SymbolTable[value_node.id] + if info.IsDefine and info.DefineValue is not None: + return info.DefineValue + elif isinstance(value_node, ast.Call): + if isinstance(value_node.func, ast.Attribute): + if getattr(value_node.func.value, 'id', None) == 't': + if value_node.args: + return self._ExtractConstValue(value_node.args[0]) + elif isinstance(value_node.func, ast.Name): + if value_node.args: + return self._ExtractConstValue(value_node.args[0]) + return None + + def _ExtractTypedefOriginalType(self, value_node: ast.AST) -> str | None: + if isinstance(value_node, ast.Attribute): + if getattr(value_node.value, 'id', None) == 't': + llvm_str = CTypeRegistry.NameToLLVM(value_node.attr) + if llvm_str: + return llvm_str + elif isinstance(value_node, ast.Name): + llvm_str = CTypeRegistry.NameToLLVM(value_node.id) + if llvm_str: + return llvm_str + elif isinstance(value_node, ast.BinOp) and isinstance(value_node.op, ast.BitOr): + left = self._ExtractTypedefOriginalType(value_node.left) + right = self._ExtractTypedefOriginalType(value_node.right) + parts = [] + if left: + parts.append(left) + if right: + parts.append(right) + if parts: + return '|'.join(parts) + elif isinstance(value_node, ast.Call): + if isinstance(value_node.func, ast.Attribute): + if getattr(value_node.func.value, 'id', None) == 't': + llvm_str = CTypeRegistry.NameToLLVM(value_node.func.attr) + if llvm_str: + return llvm_str + elif isinstance(value_node.func, ast.Name): + llvm_str = CTypeRegistry.NameToLLVM(value_node.func.id) + if llvm_str: + return llvm_str + return None + + def _RegisterCDefineSymbol(self, FullName: str, DefineValue: object, lineno: int, FilePath: str) -> None: + """Register CDefine constant to SymbolTable""" + # 防御：新值为 None 但已存在非 None 值时保留旧值。 + # 翻译期间声明发射调用 _ExtractConstValue（不支持 ast.Attribute 形式如 t.CSizeT().Size）， + # 会返回 None 并覆盖共享符号表中已由 ConstEvaluator 正确求值的值（如 8）。 + # 共享符号表在 _build_shared_symbol_data 阶段已正确求值，此处不应覆盖。 + if DefineValue is None: + existing: Any = self.Trans.SymbolTable.lookup(FullName) + if existing and existing.IsDefine and existing.DefineValue is not None: + return + info = CTypeInfo() + info.IsDefine = True + info.DefineValue = DefineValue + info.Lineno = lineno + info.file = FilePath + # 直接添加到 SymbolTable 字典 + self.Trans.SymbolTable.insert(FullName, info) + + def _check_annotation_for_state(self, annotation: ast.AST) -> bool: + return (AnnotationContainsName(annotation, 'CExport') or + AnnotationContainsName(annotation, 'CExtern') or + AnnotationContainsName(annotation, 'State')) + + def _check_decorators_for_state(self, Node: ast.FunctionDef) -> bool: + """检查装饰器列表中是否包含 @t.CExport / @t.CExtern / @t.State（t.CExport 既能作返回值注解也能作装饰器）""" + if not Node.decorator_list: + return False + for decorator in Node.decorator_list: + if (AnnotationContainsName(decorator, 'CExport') or + AnnotationContainsName(decorator, 'CExtern') or + AnnotationContainsName(decorator, 'State')): + return True + return False + + def _EmitExternalFuncDeclLlvm(self, Node: ast.FunctionDef, Gen: LlvmGeneratorMixin, is_class_method: bool = False, source_module_name: str | None = None, register_module_name: str | None = None, reexport_module_names: list[str] | None = None) -> None: + FuncName = Node.name + if (Node.returns and self._check_annotation_for_state(Node.returns)) or self._check_decorators_for_state(Node): + Gen._export_funcs.add(FuncName) # t.State 标记的外部 C 函数必须保持原始名称，以便链接器解析 + if source_module_name and source_module_name not in ('c', 't'): + _src_sha1 = Gen.ModuleSha1Map.get(source_module_name) + if not _src_sha1: + _init_name = f"{source_module_name}.__init__" + if _init_name in Gen.ModuleSha1Map: + _src_sha1 = Gen.ModuleSha1Map[_init_name] + if _src_sha1: + Gen._export_extern_funcs[FuncName] = _src_sha1 + if FuncName in Gen.functions: + return + CReturnTypes = [] + if Node.decorator_list: + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Attribute): + if decorator.func.attr == 'CReturn': + for arg in decorator.args: + CReturnTypes.append(arg) + IsPtr = False + ReturnTypeInfo = None + if Node.returns: + try: + ReturnTypeInfo = self.ResolveAnnotationType(Node.returns) + if ReturnTypeInfo and ReturnTypeInfo.IsStr: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CChar() + ReturnTypeInfo.PtrCount = 1 + if ReturnTypeInfo is None: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + IsPtr = ReturnTypeInfo.IsPtr + except Exception as e: # 回退：返回类型解析失败时使用默认 CInt + _vlog().warning(f"HandlesImports: 忽略异常 {e}", exc_info=e) + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + else: + ReturnTypeInfo = CTypeInfo() + ReturnTypeInfo.BaseType = t.CInt() + ReturnType = Gen._ctype_to_llvm(ReturnTypeInfo) + if (isinstance(ReturnType, ir.VoidType) or (isinstance(ReturnType, ir.PointerType) and isinstance(ReturnType.pointee, ir.VoidType))) and Node.returns: + RetAnnName = None + if isinstance(Node.returns, ast.Name): + RetAnnName = Node.returns.id + elif isinstance(Node.returns, ast.Attribute): + RetAnnName = Node.returns.attr + elif isinstance(Node.returns, ast.BinOp) and isinstance(Node.returns.op, ast.BitOr): + left = Node.returns.left + if isinstance(left, ast.Name): + RetAnnName = left.id + elif isinstance(left, ast.Attribute): + RetAnnName = left.attr + if RetAnnName and RetAnnName in Gen.structs: + ReturnType = ir.PointerType(Gen.structs[RetAnnName]) + elif IsPtr or RetAnnName: + ReturnType = ir.PointerType(ir.IntType(8)) + # __new__ 方法应返回类指针类型（堆分配替换），与 _GetFunctionSignatureLlvm/_EmitFunctionLlvm 保持一致 + # .pyi 生成器对无返回类型注解的 __new__ 默认使用 t.CInt，这里修正为 ClassName* + if '.__new__' in FuncName: + _np = FuncName.split('.') + if len(_np[0]) == 16 and all(_c in '0123456789abcdef' for _c in _np[0]): + _new_class_name = _np[1] + else: + _new_class_name = _np[0] + # 治本修复：优先用 source sha1 查找正确的 struct，避免同名类冲突时返回错误类型 + _new_struct = None + _new_src_sha1: str | None = None + if source_module_name and hasattr(Gen, 'ModuleSha1Map'): + _new_src_sha1 = Gen.ModuleSha1Map.get(source_module_name) + if not _new_src_sha1: + _init_name = f"{source_module_name}.__init__" + if _init_name in Gen.ModuleSha1Map: + _new_src_sha1 = Gen.ModuleSha1Map[_init_name] + if _new_src_sha1 and _new_class_name: + _new_struct = Gen.structs.get(f"{_new_src_sha1}.{_new_class_name}") + if _new_struct is None: + _new_struct = Gen.structs.get(_new_class_name) + if _new_struct is None: + _full_name = FuncName.split('.__new__')[0] + _new_struct = Gen.structs.get(_full_name) + if _new_struct: + if isinstance(_new_struct, ir.PointerType): + ReturnType = _new_struct + else: + ReturnType = ir.PointerType(_new_struct) + ParamTypes = [] + is_method = is_class_method or '.__' in FuncName + class_name_for_method = FuncName.split('.')[0] if '.' in FuncName else None + class_is_cpython = class_name_for_method and self.Trans.SymbolTable.is_struct(class_name_for_method) + # 治本修复：通过 source_module_name 获取源模块 sha1，查找正确的 struct。 + # 当存在同名类冲突（如 stmts.py:Module 和 __module.py:Module）时， + # Gen.structs[short_name] 可能指向错误模块的 struct（set_body 已锁定错误布局）。 + # 优先查找以 sha1 为前缀的完整 key，避免使用错误的 struct 作为 self 参数类型。 + _self_struct_sha1: str | None = None + if source_module_name and hasattr(Gen, 'ModuleSha1Map'): + _self_struct_sha1 = Gen.ModuleSha1Map.get(source_module_name) + if not _self_struct_sha1: + _init_name = f"{source_module_name}.__init__" + if _init_name in Gen.ModuleSha1Map: + _self_struct_sha1 = Gen.ModuleSha1Map[_init_name] + for i, Arg in enumerate(Node.args.args): + if i == 0 and is_method: + # self parameter of a method should always be a pointer to the struct + _self_struct = None + if _self_struct_sha1 and class_name_for_method: + _full_key = f"{_self_struct_sha1}.{class_name_for_method}" + _self_struct = Gen.structs.get(_full_key) + if _self_struct is None: + _self_struct = Gen.structs.get(class_name_for_method or FuncName.split('.')[0], ir.IntType(8)) + ParamType = ir.PointerType(_self_struct) + elif Arg.annotation: + try: + ParamTypeInfo = self.ResolveAnnotationType(Arg.annotation) + if ParamTypeInfo is None: + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CInt() + ParamIsPtr = ParamTypeInfo.IsPtr + if i == 0 and is_method and Arg.arg == 'self' and class_is_cpython: + ParamIsPtr = True + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + if ParamTypeInfo.IsStr: + ParamTypeInfo = CTypeInfo() + ParamTypeInfo.BaseType = t.CChar() + ParamTypeInfo.PtrCount = 1 + ParamIsPtr = True + # C 语言中数组参数退化为指针：list[type, N] → type* + if ParamTypeInfo.ArrayDims: + ParamTypeInfo.ArrayDims = [] + ParamTypeInfo.PtrCount = max(ParamTypeInfo.PtrCount, 1) + ParamType = Gen._ctype_to_llvm(ParamTypeInfo) + except Exception as e: # 回退：参数类型解析失败时使用默认 i32 + _vlog().warning(f"HandlesImports: 忽略异常 {e}", exc_info=e) + ParamType = ir.IntType(32) + if isinstance(ParamType, ir.VoidType) or (isinstance(ParamType, ir.PointerType) and isinstance(ParamType.pointee, ir.VoidType)): + AnnName = None + if isinstance(Arg.annotation, ast.Name): + AnnName = Arg.annotation.id + elif isinstance(Arg.annotation, ast.Attribute): + AnnName = Arg.annotation.attr + elif isinstance(Arg.annotation, ast.BinOp) and isinstance(Arg.annotation.op, ast.BitOr): + AnnName = ExtractTypeNameFromBinOp(Arg.annotation) + if AnnName and AnnName in Gen.structs: + ParamType = ir.PointerType(Gen.structs[AnnName]) + else: + ParamType = ir.PointerType(ir.IntType(8)) + else: + ParamType = ir.IntType(32) + ParamTypes.append(ParamType) + if CReturnTypes: + for _ in CReturnTypes: + ParamTypes.append(ir.PointerType(ir.IntType(8))) + IsVariadic = Node.args.vararg is not None + FuncType = ir.FunctionType(ReturnType, ParamTypes, var_arg=IsVariadic) + _is_export = False + if isinstance(ReturnTypeInfo, CTypeInfo) and ReturnTypeInfo.IsState: + _is_export = True + if isinstance(ReturnTypeInfo, CTypeInfo) and ReturnTypeInfo.Storage and isinstance(ReturnTypeInfo.Storage, t.CExport) and not ReturnTypeInfo.IsState: + Gen._export_funcs.add(FuncName) + _is_export = True + if isinstance(ReturnTypeInfo, CTypeInfo) and ReturnTypeInfo.Storage and isinstance(ReturnTypeInfo.Storage, t.CExtern): + Gen._export_funcs.add(FuncName) + _is_export = True + if not _is_export: + if Node.returns and self._check_annotation_for_state(Node.returns): + _is_export = True + elif self._check_decorators_for_state(Node): + _is_export = True + MangledName = Gen._mangle_func_name(FuncName, module_name=source_module_name) + func = Gen._get_or_declare_function(MangledName, FuncType) + Gen.functions[MangledName] = func + Gen.functions[FuncName] = func + func_meta = FuncMeta.NONE + if Node.decorator_list: + for d in Node.decorator_list: + if isinstance(d, ast.Name): + if d.id == 'staticmethod': + func_meta |= FuncMeta.STATIC_METHOD + elif d.id == 'property': + func_meta |= FuncMeta.PROPERTY_GETTER + elif d.id == 'classmethod': + func_meta |= FuncMeta.CLASS_METHOD + elif isinstance(d, ast.Attribute): + if isinstance(d.value, ast.Name): + # 支持 @property.setter 和 @propname.setter 两种形式 + if d.value.id == 'property' or d.attr in ('setter', 'getter', 'deleter'): + if d.attr == 'setter': + func_meta |= FuncMeta.PROPERTY_SETTER + elif d.attr == 'getter': + func_meta |= FuncMeta.PROPERTY_GETTER + elif d.attr == 'deleter': + func_meta |= FuncMeta.PROPERTY_DELETER + if not self.Trans.SymbolTable.has(FuncName): + FuncInfo = CTypeInfo() + FuncInfo.Name = FuncName + FuncInfo.IsFunction = True + FuncInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(FuncName, FuncInfo) + else: + existing = self.Trans.SymbolTable[FuncName] + if existing.MetaList == FuncMeta.NONE and func_meta != FuncMeta.NONE: + existing.MetaList = func_meta + # 同时注册带模块前缀的符号，以便 module.func_name 查找能命中 + if source_module_name and source_module_name not in ('c', 't'): + FullSymKey = f"{source_module_name}.{FuncName}" + if not self.Trans.SymbolTable.has(FullSymKey): + FullFuncInfo = CTypeInfo() + FullFuncInfo.Name = FullSymKey + FullFuncInfo.IsFunction = True + FullFuncInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(FullSymKey, FullFuncInfo) + # property setter/deleter: 在原始 PropKey（不带后缀）下注册 MetaList + if FuncMeta.PROPERTY_SETTER in func_meta or FuncMeta.PROPERTY_DELETER in func_meta: + BasePropKey = FuncName.replace('$set', '').replace('$del', '') + base_existing = self.Trans.SymbolTable.lookup(BasePropKey) + if base_existing: + if func_meta != FuncMeta.NONE: + base_existing.MetaList = base_existing.MetaList | func_meta + else: + PropInfo = CTypeInfo() + PropInfo.Name = BasePropKey + PropInfo.IsFunction = True + PropInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(BasePropKey, PropInfo) + if register_module_name and register_module_name != source_module_name: + ReexportName = Gen._mangle_func_name(FuncName, module_name=register_module_name) + Gen.functions[ReexportName] = func + if reexport_module_names: + for reexport_mod in reexport_module_names: + if reexport_mod and reexport_mod != source_module_name: + ReexportName2 = Gen._mangle_func_name(FuncName, module_name=reexport_mod) + if ReexportName2 not in Gen.functions: + Gen.functions[ReexportName2] = func + + def _EmitExternalClassDeclLlvm(self, Node: ast.ClassDef, Gen: LlvmGeneratorMixin, module_name: str | None = None, actual_module_name: str | None = None, source_sha1: str | None = None) -> None: + ClassName = Node.name + if hasattr(Node, 'type_params') and Node.type_params: + return + IsCenum = False + IsRenum = False + IsCpythonObject = False + IsCVTable = False + IsNoVTable = False + if Node.bases: + for base in Node.bases: + if getattr(base, 'attr', None): + if base.attr == 'REnum': + IsRenum = True + break + elif base.attr == 'CEnum' or base.attr == 'Enum': + IsCenum = True + break + elif getattr(base, 'id', None): + if base.id == 'REnum': + IsRenum = True + break + elif base.id == 'CEnum' or base.id == 'Enum': + IsCenum = True + break + if not IsCenum and not IsRenum and getattr(Node, 'decorator_list', None): + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Attribute): + if getattr(decorator.value, 'id', None) == 't': + if decorator.attr == 'Object': + IsCpythonObject = True + elif decorator.attr == 'CVTable': + IsCVTable = True + elif decorator.attr == 'NoVTable': + IsNoVTable = True + elif decorator.attr == 'REnum': + IsRenum = True + elif decorator.attr == 'CEnum' or decorator.attr == 'Enum': + IsCenum = True + elif isinstance(decorator, ast.Call): + if isinstance(decorator.func, ast.Attribute): + if getattr(decorator.func.value, 'id', None) == 't': + if decorator.func.attr == 'Object': + IsCpythonObject = True + elif decorator.func.attr == 'CVTable': + IsCVTable = True + elif decorator.func.attr == 'NoVTable': + IsNoVTable = True + elif isinstance(decorator.func, ast.Name): + if decorator.func.id == 'Object': + IsCpythonObject = True + elif decorator.func.id == 'CVTable': + IsCVTable = True + elif decorator.func.id == 'NoVTable': + IsNoVTable = True + if ClassName.startswith('__') and not IsCpythonObject and not IsCenum and not IsRenum: + return + IsPacked = False + if getattr(Node, 'decorator_list', None): + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Call): + if isinstance(decorator.func, ast.Attribute): + if getattr(decorator.func.value, 'id', None) == 'c' and decorator.func.attr == 'Attribute': + for arg in decorator.args: + if isinstance(arg, ast.Attribute): + if isinstance(arg.value, ast.Attribute): + if getattr(arg.value.value, 'id', None) == 't' and arg.value.attr == 'attr' and arg.attr == 'packed': + IsPacked = True + if IsPacked: + Gen.class_packed.add(ClassName) + if IsCpythonObject: + CInfo = CTypeInfo() + CInfo.Name = ClassName + CInfo.IsCpythonObject = True + CInfo.IsStruct = True + if module_name: + FullName = f"{module_name}.{ClassName}" + self.Trans.SymbolTable.insert(FullName, CInfo) + self.Trans.SymbolTable.insert(ClassName, CInfo) + if IsCenum: + EnumTypeNode = CTypeInfo() + EnumTypeNode.Name = ClassName + EnumTypeNode.BaseType = t.CEnum(ClassName) + EnumTypeNode.IsEnum = True + self.Trans.SymbolTable.insert(ClassName, EnumTypeNode) + next_enum_value: int = 0 + for item in Node.body: + VarName = None + ItemValue = None + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + VarName = item.target.id + ItemValue = item.value + elif isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + VarName = target.id + break + ItemValue = item.value + if VarName is None: + continue + # 提取枚举值：优先用显式赋值，否则自动编号 + if ItemValue is not None and isinstance(ItemValue, ast.Constant) and isinstance(ItemValue.value, int): + value = ItemValue.value + next_enum_value = ItemValue.value + 1 + else: + value = next_enum_value + next_enum_value += 1 + MemberNode = CTypeInfo() + MemberNode.Name = VarName + MemberNode.BaseType = t.CEnum(ClassName) + MemberNode.value = value + MemberNode.EnumName = ClassName + MemberNode.Lineno = item.lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VarName, MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}.{VarName}", MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}_{VarName}", MemberNode) + return + if IsRenum: + # REnum 跨模块导入：注册 REnum 类型 + 变体成员到 SymbolTable + # 对齐同模块 _EmitREnumLlvm 的注册逻辑 + RenumTypeNode = CTypeInfo() + RenumTypeNode.Name = ClassName + RenumTypeNode.IsRenum = True + RenumTypeNode.IsEnum = True # 兼容 IsEnum 检查 + # REnum 在内存布局上是 struct，必须设置 IsStruct=True 才能让 + # HandlesTypeMerge 的 CPtr 吸收逻辑生效（REnumType | t.CPtr = REnumType*） + RenumTypeNode.IsStruct = True + if module_name: + FullName = f"{module_name}.{ClassName}" + self.Trans.SymbolTable.insert(FullName, RenumTypeNode) + self.Trans.SymbolTable.insert(ClassName, RenumTypeNode) + # 从 stub.ll 加载 REnum 结构体布局到 Gen.structs + self._TryLoadStructFromStub(ClassName, Gen) + # 记录 class_sha1_map（跨模块名修饰） + source_sha1 = None + if actual_module_name and hasattr(Gen, 'ModuleSha1Map') and actual_module_name in Gen.ModuleSha1Map: + source_sha1 = Gen.ModuleSha1Map[actual_module_name] + elif module_name and hasattr(Gen, 'ModuleSha1Map') and module_name in Gen.ModuleSha1Map: + source_sha1 = Gen.ModuleSha1Map[module_name] + if source_sha1: + if not hasattr(Gen, 'class_sha1_map'): + Gen.class_sha1_map = {} + Gen.class_sha1_map[ClassName] = source_sha1 + # 遍历嵌套 ClassDef 注册变体成员 + # REnum body 是嵌套 ClassDef（变体定义），不是 AnnAssign/Assign + next_tag: int = 0 + variant_names: list[str] = [] + for item in Node.body: + if isinstance(item, ast.ClassDef): + VariantName = item.name + MemberNode = CTypeInfo() + MemberNode.Name = VariantName + MemberNode.value = next_tag + MemberNode.EnumName = ClassName + MemberNode.Lineno = item.lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VariantName, MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}.{VariantName}", MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}_{VariantName}", MemberNode) + variant_names.append(VariantName) + next_tag += 1 + # 加载嵌套变体结构体（如 LLVMType_Int）到 Gen.structs， + # 并注册 class_members 以支持 _HandleREnumConstructLlvm 的 payload 存储和 + # _HandleRenumMatchLlvm 的 binding 提取。 + NestedStructName: str = f"{ClassName}_{VariantName}" + self._TryLoadStructFromStub(NestedStructName, Gen) + if NestedStructName not in Gen.class_members: + Gen.class_members[NestedStructName] = [] + if NestedStructName not in Gen.class_member_signeds: + Gen.class_member_signeds[NestedStructName] = {} + if Gen.class_members[NestedStructName]: + # 已有成员（_TryLoadStructFromStub 可能已加载），跳过 + pass + else: + Gen.class_members[NestedStructName].append(('__tag', ir.IntType(32))) + Gen.class_member_signeds[NestedStructName]['__tag'] = None + for nested_item in item.body: + parsed = self.CollectAnnAssignMember(nested_item, Gen) + if parsed: + MName, MType, MTypeInfo = parsed + Gen.class_members[NestedStructName].append((MName, MType)) + Gen.class_member_signeds[NestedStructName][MName] = None + # 当 stub 中没有嵌套变体结构体定义时（_TryLoadStructFromStub 失败）， + # 使用主 REnum 结构体的 body (max_variant_struct) 创建嵌套变体结构体， + # 确保跨模块布局一致。否则 _generate_structs() Step 4 会从 class_members + # 设置最小布局 body，导致 GEP 偏移量与定义模块不匹配（offset 4 vs offset 8）。 + NestedStructExisting = Gen.structs.get(NestedStructName) + NestedHasBody = (isinstance(NestedStructExisting, ir.IdentifiedStructType) + and NestedStructExisting.elements is not None + and len(NestedStructExisting.elements) > 0) + if not NestedHasBody: + MainRenumStruct = Gen.structs.get(ClassName) + if (isinstance(MainRenumStruct, ir.IdentifiedStructType) + and MainRenumStruct.elements is not None + and len(MainRenumStruct.elements) > 0): + NestedStruct = Gen._get_or_create_struct(NestedStructName, source_sha1=source_sha1) + if isinstance(NestedStruct, ir.IdentifiedStructType) and NestedStruct.is_opaque: + NestedStruct.set_body(*MainRenumStruct.elements) + elif isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + # 显式值变体（无 payload） + VariantName = item.target.id + value = next_tag + if item.value and isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + value = item.value.value + MemberNode = CTypeInfo() + MemberNode.Name = VariantName + MemberNode.value = value + MemberNode.EnumName = ClassName + MemberNode.Lineno = item.lineno + MemberNode.IsEnumMember = True + self.Trans.SymbolTable.insert(VariantName, MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}.{VariantName}", MemberNode) + self.Trans.SymbolTable.insert(f"{ClassName}_{VariantName}", MemberNode) + variant_names.append(VariantName) + next_tag = value + 1 + RenumTypeNode.RenumVariants = variant_names + return + # 先初始化 class_methods/class_members 等容器，确保 _TryLoadStructFromStub + # 和 _get_or_create_struct 中的 IsEnum 守卫（clean_name not in class_methods） + # 能正确识别 AST 节点类名（避免 ASTKind 枚举名冲突时把节点类当枚举返回 i32） + if ClassName not in Gen.class_members: + Gen.class_members[ClassName] = [] + if ClassName not in Gen.class_member_defaults: + Gen.class_member_defaults[ClassName] = {} + if ClassName not in Gen.class_member_signeds: + Gen.class_member_signeds[ClassName] = {} + if ClassName not in Gen.class_member_bitfields: + Gen.class_member_bitfields[ClassName] = {} + if ClassName not in Gen.class_member_byteorders: + Gen.class_member_byteorders[ClassName] = {} + if ClassName not in Gen.class_member_bitoffsets: + Gen.class_member_bitoffsets[ClassName] = {} + if ClassName not in Gen.class_methods: + Gen.class_methods[ClassName] = [] + # --- 预扫描预注册：修复编译时序问题 --- + # _TryLoadStructFromStub → _parse_llvm_type → _specialize_generic_class + # 可能触发 list[str].__new__ 编译（其中调用 pool.alloc(48)）， + # 此时 MemManager 尚未注册到虚表，导致直接调用返回 NULL → 段错误。 + # 修复：在 _TryLoadStructFromStub 之前预扫描方法列表并提前注册虚表。 + _has_methods_prescan = False + for _item in Node.body: + if isinstance(_item, ast.FunctionDef): + _has_methods_prescan = True + _MethodName = _item.name + # 构造函数（__new__/__init__/__before_init__）不放入 vtable， + # 避免子类与基类 vtable 大小不一致导致虚分派索引偏移 + # （如 Module 有 __new__/__init__ 使 vtable=7，AST 无它们 vtable=5， + # 调用方按 AST 布局取 append@idx4 实际取到 Module.dump@idx4）。 + if _MethodName in ('__new__', '__init__', '__before_init__'): + continue + _FuncFullName = f"{ClassName}.__init__" if _MethodName == "__init__" else f"{ClassName}.__call__" if _MethodName == "__call__" else f"{ClassName}.{_MethodName}" + if _FuncFullName not in Gen.class_methods[ClassName]: + Gen.class_methods[ClassName].append(_FuncFullName) + if _has_methods_prescan and IsCVTable: + Gen._cross_module_vtable_classes.add(ClassName) + Gen.class_vtable.add(ClassName) + # 先计算 source_sha1，确保首次 _TryLoadStructFromStub 调用就能使用正确的 sha1 + # 优先使用传入的 source_sha1（来自 _TryLoadClassMembersFromPyi 的直接调用， + # 当前模块未直接导入定义模块时 module_name 查找会失败，需依赖传入的 source_sha1） + if not source_sha1: + if actual_module_name and hasattr(Gen, 'ModuleSha1Map') and actual_module_name in Gen.ModuleSha1Map: + source_sha1 = Gen.ModuleSha1Map[actual_module_name] + elif module_name and hasattr(Gen, 'ModuleSha1Map') and module_name in Gen.ModuleSha1Map: + source_sha1 = Gen.ModuleSha1Map[module_name] + # 首次加载就传入 source_sha1，避免从其他模块的 output stub 中加载到错误的结构体定义 + # （如 HandlesExprCall.py 的 output stub 中 43f27dda2e5b5923.Value 只有 5 字段， + # 而正确的 aaf8ba449d1e64c1.Value 有 6 字段） + self._TryLoadStructFromStub(ClassName, Gen, source_sha1=source_sha1) + # Track which module defines this class (for cross-module name mangling) + if source_sha1: + if not hasattr(Gen, 'class_sha1_map'): + Gen.class_sha1_map = {} + Gen.class_sha1_map[ClassName] = source_sha1 + Gen._get_or_create_struct(ClassName, source_sha1=source_sha1, packed=IsPacked) + # 如果 _get_or_create_struct 因 sha1 冲突创建了新的 opaque struct， + # 需用 source_sha1 重新加载正确的 stub body。 + if source_sha1: + self._TryLoadStructFromStub(ClassName, Gen, source_sha1=source_sha1) + has_methods = False + # 解析 __provides__/__requires__/__require_must__ 编译期元数据（跨模块加载） + for item in Node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + field_name: str = item.target.id + if field_name in ('__provides__', '__requires__', '__require_must__') and item.value: + values: list[str] = [] + if isinstance(item.value, ast.List): + for elt in item.value.elts: + if isinstance(elt, ast.Constant) and isinstance(elt.value, str): + values.append(elt.value) + if field_name == '__provides__': + Gen.class_provides[ClassName] = values + elif field_name == '__requires__': + Gen.class_requires[ClassName] = values + elif field_name == '__require_must__': + Gen.class_require_must[ClassName] = values + _existing_member_names = {m[0] for m in Gen.class_members[ClassName]} + for item in Node.body: + # 跳过编译期元数据字段（__provides__/__requires__/__require_must__） + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + if item.target.id in ('__provides__', '__requires__', '__require_must__'): + continue + parsed = self.CollectAnnAssignMember(item, Gen) + if parsed: + VarName, MemberType, TypeInfo = parsed + if TypeInfo is None: + if VarName not in _existing_member_names: + Gen.class_members[ClassName].append((VarName, MemberType)) + _existing_member_names.add(VarName) + Gen.class_member_signeds[ClassName][VarName] = None + Gen.class_member_bitfields[ClassName][VarName] = 0 + else: + if TypeInfo.IsBitField: + Gen.class_member_bitfields[ClassName][VarName] = TypeInfo.BitWidth + else: + if VarName not in _existing_member_names: + Gen.class_members[ClassName].append((VarName, MemberType)) + _existing_member_names.add(VarName) + Gen.class_member_bitfields[ClassName][VarName] = 0 + if TypeInfo.ByteOrder: + Gen.class_member_byteorders[ClassName][VarName] = TypeInfo.ByteOrder + else: + Gen.class_member_byteorders[ClassName][VarName] = "" + if item.value: + const = self._BuildScalarConstant(item.value, MemberType, Gen) + if const: + Gen.class_member_defaults[ClassName][VarName] = const + elif isinstance(item, ast.FunctionDef): + has_methods = True + MethodName = item.name + FuncFullName = f"{ClassName}.__init__" if MethodName == "__init__" else f"{ClassName}.__call__" if MethodName == "__call__" else f"{ClassName}.{MethodName}" + # 构造函数（__new__/__init__/__before_init__）不放入 vtable， + # 避免子类与基类 vtable 大小不一致导致虚分派索引偏移。 + # register_method 内部也会过滤，此处仅跳过 class_methods.append。 + if MethodName not in ('__new__', '__init__', '__before_init__'): + if FuncFullName not in Gen.class_methods[ClassName]: + Gen.class_methods[ClassName].append(FuncFullName) + Gen.register_method(ClassName, FuncFullName) + is_item_static = any(isinstance(d, ast.Name) and d.id == 'staticmethod' for d in item.decorator_list) + is_item_property = any(isinstance(d, ast.Name) and d.id == 'property' for d in item.decorator_list) + is_item_classmethod = any(isinstance(d, ast.Name) and d.id == 'classmethod' for d in item.decorator_list) + is_item_prop_setter = any(isinstance(d, ast.Attribute) and d.attr == 'setter' and isinstance(d.value, ast.Name) and d.value.id == 'property' for d in item.decorator_list) + is_item_prop_getter = any(isinstance(d, ast.Attribute) and d.attr == 'getter' and isinstance(d.value, ast.Name) and d.value.id == 'property' for d in item.decorator_list) + is_item_prop_deleter = any(isinstance(d, ast.Attribute) and d.attr == 'deleter' and isinstance(d.value, ast.Name) and d.value.id == 'property' for d in item.decorator_list) + # setter/deleter 使用不同的函数名后缀 + DeclFuncName = FuncFullName + if is_item_prop_setter: + DeclFuncName = FuncFullName + '$set' + elif is_item_prop_deleter: + DeclFuncName = FuncFullName + '$del' + if DeclFuncName not in Gen.functions: + FuncDeclNode = ast.FunctionDef( + name=DeclFuncName, + args=item.args, + body=item.body, + decorator_list=item.decorator_list, + returns=item.returns + ) + try: + self._EmitExternalFuncDeclLlvm(FuncDeclNode, Gen, is_class_method=not is_item_static and not is_item_classmethod, source_module_name=module_name) + except Exception as e: + raise + func_meta = FuncMeta.NONE + if is_item_static: + func_meta |= FuncMeta.STATIC_METHOD + if is_item_property: + func_meta |= FuncMeta.PROPERTY_GETTER + if is_item_classmethod: + func_meta |= FuncMeta.CLASS_METHOD + if is_item_prop_setter: + func_meta |= FuncMeta.PROPERTY_SETTER + if is_item_prop_getter: + func_meta |= FuncMeta.PROPERTY_GETTER + if is_item_prop_deleter: + func_meta |= FuncMeta.PROPERTY_DELETER + # setter/deleter 的 SymKey 使用带后缀的函数名 + SymKey = DeclFuncName + if not self.Trans.SymbolTable.has(SymKey): + FuncInfo = CTypeInfo() + FuncInfo.Name = SymKey + FuncInfo.IsFunction = True + FuncInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(SymKey, FuncInfo) + else: + existing = self.Trans.SymbolTable[SymKey] + if existing.MetaList == FuncMeta.NONE and func_meta != FuncMeta.NONE: + existing.MetaList = func_meta + # setter/deleter: 同时在原始 PropKey（不带后缀）下注册 MetaList + if is_item_prop_setter or is_item_prop_deleter: + base_existing = self.Trans.SymbolTable.lookup(FuncFullName) + if base_existing: + if func_meta != FuncMeta.NONE: + base_existing.MetaList = base_existing.MetaList | func_meta + else: + PropInfo = CTypeInfo() + PropInfo.Name = FuncFullName + PropInfo.IsFunction = True + PropInfo.MetaList = func_meta + self.Trans.SymbolTable.insert(FuncFullName, PropInfo) + # --- 跨模块继承字段展平 --- + # _EmitExternalClassDeclLlvm 只添加了类自身定义的字段（来自 .pyi）， + # 没有展平父类（如 GSListNode[T]）的继承字段，导致 class_members 缺少 + # 继承字段，_get_member_offset 返回错误偏移。 + # 修复：先尝试从已加载的父类获取真实字段名；若父类未加载， + # 则用 struct body 元素数推断继承字段数并补充占位条目 __inherited_N。 + _ParentClass: str | None = None + if Node.bases: + for _base in Node.bases: + _base_name: str | None = None + if hasattr(_base, 'id'): + _base_name = _base.id + elif hasattr(_base, 'attr'): + _base_name = _base.attr + elif isinstance(_base, ast.Subscript): + _spec_name: str | None = self.Trans.ExprCallHandle._ResolveGenericSlice(_base) + if _spec_name: + _base_name = _spec_name + else: + # _ResolveGenericSlice 失败（泛型基类未在当前模块注册为模板， + # 常见于跨模块导入：如 HandlesExprCall.py 导入 Value(GSListNode[Value])， + # 但 GSListNode 定义在 includes/linkedlist.py，尚未编译）。 + # 直接从 AST 提取泛型基类名（如 GSListNode[Value] -> GSListNode）， + # 后续继承字段展平逻辑会从 .pyi 加载该基类的字段。 + if isinstance(_base.value, ast.Name): + _base_name = _base.value.id + elif isinstance(_base.value, ast.Attribute): + _base_name = _base.value.attr + else: + continue + else: + continue + if _base_name == ClassName: + continue + if _base_name in ('CVTable', 'Object', 'NoVTable', 'CStruct', 'CUnion', 'CEnum', 'REnum', 'Exception', 'Enum'): + continue + # 总是记录父类名，即使父类 class_members 未加载。 + # _get_member_offset 会沿 class_parent 链回退查找继承字段。 + _ParentClass = _base_name + if not hasattr(Gen, 'class_parent'): + Gen.class_parent = {} + Gen.class_parent[ClassName] = _ParentClass + break + # 治本修复：在继承展开前，尝试从 stub/.pyi 加载父类的 class_members。 + # 确保子类 class_members 能正确展平父类真实字段名（而非 __inherited_N 占位符）， + # 避免 struct body 从不完整 class_members 生成导致 GEP 越界。 + # 同时处理父类 class_members 全为占位符的情况（清除后重新加载）。 + if _ParentClass: + _parent_members_check: list = Gen.class_members.get(_ParentClass, []) + _parent_all_placeholders: bool = bool(_parent_members_check) and all( + n.startswith('__inherited_') for n, _ in _parent_members_check + ) + if not _parent_members_check or _parent_all_placeholders: + # 清除占位条目以便 _TryLoadClassMembersFromPyi 的 guard 通过 + _old_parent_members: list = _parent_members_check + if _parent_all_placeholders: + Gen.class_members[_ParentClass] = [] + _sha1_map: dict[str, str] = getattr(Gen, 'ModuleSha1Map', {}) + _parent_loaded: bool = False + for _mod_name, _mod_sha1 in _sha1_map.items(): + _stub_name: str = f"{_mod_sha1}.stub.ll" + try: + self._TryLoadClassMembersFromPyi(_ParentClass, _stub_name, Gen) + except Exception: + pass + _parent_members_check = Gen.class_members.get(_ParentClass, []) + if _parent_members_check and not all(n.startswith('__inherited_') for n, _ in _parent_members_check): + _parent_loaded = True + break + if not _parent_loaded and _old_parent_members: + # 加载失败，回滚到旧占位条目（有总比没有好） + Gen.class_members[_ParentClass] = _old_parent_members + if _ParentClass and _ParentClass in Gen.class_members: + _parent_members: list = list(Gen.class_members[_ParentClass]) + _parent_defaults: dict = dict(Gen.class_member_defaults.get(_ParentClass, {})) + _existing_names: set = {m[0] for m in Gen.class_members[ClassName]} + _inherited: list[tuple] = [] + for _pm_name, _pm_type in _parent_members: + if _pm_name not in _existing_names: + _inherited.append((_pm_name, _pm_type)) + if _pm_name in _parent_defaults: + Gen.class_member_defaults[ClassName][_pm_name] = _parent_defaults[_pm_name] + Gen.class_members[ClassName] = _inherited + Gen.class_members[ClassName] + else: + # 父类未加载到 class_members（常见于泛型特化基类如 GSListNode[Value]， + # 其特化不在 .pyi 中）。用 struct body 元素数推断继承字段数。 + _st = Gen.structs.get(ClassName) + if isinstance(_st, ir.IdentifiedStructType) and _st.elements is not None: + _num_elements = len(_st.elements) + # 如果类有 vtable ptr（struct 首元素），需减去 1 + # 否则 vtable ptr 会被误当成 __inherited_0 占位条目， + # 导致子类继承时多出一个字段（如 Name(AST) 从 9 字段变成 10） + _has_vtable_ptr = IsCVTable or (has_methods and not IsNoVTable) + if _has_vtable_ptr and _num_elements > 0: + _num_elements -= 1 + _num_members = len(Gen.class_members[ClassName]) + if _num_elements > _num_members: + _num_inherited = _num_elements - _num_members + _inherited_ph: list[tuple] = [(f'__inherited_{i}', _st.elements[i]) for i in range(_num_inherited)] + Gen.class_members[ClassName] = _inherited_ph + Gen.class_members[ClassName] + if has_methods: + if IsCVTable: + Gen._cross_module_vtable_classes.add(ClassName) + Gen.class_vtable.add(ClassName) + if IsNoVTable: + Gen._cross_module_novtable.add(ClassName) + # __before_init__ 对所有跨模块结构体声明（定义侧无条件生成，导入侧也应无条件声明） + NewFuncName = f'{ClassName}.__before_init__' + if not Gen._has_function(NewFuncName): + # 复用前面已查找的 source_sha1（优先 actual_module_name 再 module_name）， + # 通过 class_sha1_map 获取，避免此处只查 module_name 导致子模块（如 w32.win32file）查找失败 + source_sha1 = getattr(Gen, 'class_sha1_map', {}).get(ClassName) + MangledName = Gen._mangle_name(NewFuncName) if not source_sha1 else f"{source_sha1}.{NewFuncName}" + StructType = Gen.structs.get(ClassName) + if StructType is not None: + StructPtrType = ir.PointerType(StructType) + NewFuncType = ir.FunctionType(ir.VoidType(), [StructPtrType]) + NewFunc = ir.Function(Gen.module, NewFuncType, name=MangledName) + Gen.functions[NewFuncName] = NewFunc + bitfields = Gen.class_member_bitfields.get(ClassName, {}) + if any(v > 0 for v in bitfields.values()): + current_bit_offset = 0 + for name, _ in Gen.class_members.get(ClassName, []): + bw = bitfields.get(name, 0) + Gen.class_member_bitoffsets[ClassName][name] = current_bit_offset + current_bit_offset += bw + + def _build_stub_index(self) -> None: + """扫描所有 .stub.ll 文件构建 class_name → 条目列表索引 + + 一次扫描替代 _TryLoadStructFromStub 的 609 次全量扫描。 + 通过文件数签名检测新增 stub，支持增量编译。 + """ + ProjectRoot = self._find_project_root() + temp_dir = os.path.join(ProjectRoot, 'temp') + output_dir = os.path.join(ProjectRoot, 'output') + if not os.path.isdir(temp_dir) and not os.path.isdir(output_dir): + return + + # 文件数签名：检测 stub 文件新增（增量编译时 output 逐步填充） + output_count = sum(1 for f in os.listdir(output_dir) if f.endswith('.stub.ll')) if os.path.isdir(output_dir) else 0 + temp_count = sum(1 for f in os.listdir(temp_dir) if f.endswith('.stub.ll')) if os.path.isdir(temp_dir) else 0 + signature = (output_count, temp_count) + if self._stub_index_signature == signature and self._stub_index: + return # 索引未过期 + + # 重建索引 + self._stub_index.clear() + for scan_dir, is_output in [(output_dir, True), (temp_dir, False)]: + if not os.path.isdir(scan_dir): + continue + for filename in os.listdir(scan_dir): + if not filename.endswith('.stub.ll'): + continue + stub_path = os.path.join(scan_dir, filename) + # 预检查文件存在性：os.listdir 与 open 之间文件可能被 --clean 或 + # 并发写入删除（独立 profile 脚本曾触发 FileNotFoundError） + if not os.path.isfile(stub_path): + continue + file_sha1 = filename.replace('.stub.ll', '') + try: + with open(stub_path, 'r', encoding='utf-8') as f: + content = f.read() + for line in content.splitlines(): + stripped = line.strip() + match = re.match(r'%"?((?:[a-f0-9]+\.)?([\w\[\]]+))"?\s*=\s*type\s*(.*)', stripped) + if match: + full_name = match.group(1) + short_name = match.group(2) + struct_body = match.group(3).strip() + # 索引键：short_name 和 full_name 都映射到同一条目 + self._stub_index.setdefault(short_name, []).append((full_name, struct_body, file_sha1, is_output)) + if full_name != short_name: + self._stub_index.setdefault(full_name, []).append((full_name, struct_body, file_sha1, is_output)) + except FileNotFoundError: + # 竞态：isfile 检查后文件被删除，跳过即可（不应触发 strict raise） + continue + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"构建 stub 索引失败: {_e}", "Exception") + self._stub_index_signature = signature + + def _TryLoadStructFromStub(self, class_name: str, Gen: LlvmGeneratorMixin, source_sha1: str | None = None) -> None: + # 缓存键包含 Gen 实例标识，确保每个模块的 Gen 都能独立加载 struct 定义， + # 同时仍能防止同一 Gen 内的递归重入。原先使用纯 class_name 作为键会导致 + # 跨模块 struct 类型对象无法分别 set_body（每个模块有独立的 self.structs）。 + # 治本修复：当 source_sha1 指定时（同名类冲突情况），使用 sha1 前缀全名作为 + # 缓存键，确保不同模块的同名类（如 stmts.py:Module 和 __module.py:Module） + # 不会互相阻止 stub 加载。 + cache_key = (id(Gen), f"{source_sha1}.{class_name}" if source_sha1 else class_name) + if cache_key in self._struct_Load_cache: + return + # 先设置缓存防止递归重入，但如果加载失败则清除缓存允许后续重试 + self._struct_Load_cache[cache_key] = True + + # 使用 stub 索引替代全量扫描（609 次 × 100 文件 → 1 次扫描 + O(1) 查询） + self._build_stub_index() + entries = self._stub_index.get(class_name, []) + + if not entries: + del self._struct_Load_cache[cache_key] + return + + # 选择最佳条目： + # 优先级 0: 当 source_sha1 指定时，优先选择 type_sha1 匹配的条目（治本修复： + # 同名类冲突时确保选择正确模块的 struct 定义） + # 优先级 1: 类型定义模块自身的 stub（file_sha1 == full_name 中的 sha1 前缀） + # 定义模块的 stub 包含完整的字段展平（含跨模块继承字段） + # 优先级 2: output（Phase2 生成）优于 temp（Phase1 生成） + # 优先级 3: 非 opaque（具体定义）优于 opaque + def _entry_priority(e: tuple[str, str, str, bool]) -> tuple[int, int, int, int]: + full_name, struct_body, file_sha1, from_output = e + type_sha1 = file_sha1 + if '.' in full_name: + parts = full_name.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + type_sha1 = parts[0] + sha1_match = 0 if (source_sha1 and type_sha1 == source_sha1) else 1 + is_defining = 0 if type_sha1 == file_sha1 else 1 + is_temp = 0 if from_output else 1 + is_opaque = 0 if (struct_body and struct_body != 'opaque') else 1 + return (sha1_match, is_defining, is_temp, is_opaque) + + best_entry = min(entries, key=_entry_priority) + full_name, struct_body, file_sha1, from_output = best_entry + + # Bug 1 修复：从 full_name 提取 sha1 前缀（类型定义模块的 sha1）， + # 而非使用 file_sha1（stub 文件所在模块的 sha1，可能只是导入方而非定义方）。 + # 例如 output/35ffd4b9fbd082d1.stub.ll (main.py) 中包含 + # %"5cde8c6813a2b6f3.Function" = type {...}， + # 此时 file_sha1='35ffd4b9fbd082d1' (main.py) 但类型实际定义在 + # __function.py (sha1='5cde8c6813a2b6f3')，必须用后者作为 source_sha1。 + actual_sha1 = file_sha1 + if '.' in full_name: + parts = full_name.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + actual_sha1 = parts[0] + + is_packed = struct_body.startswith('<{') and struct_body.endswith('}>') + if struct_body and struct_body != 'opaque': + elem_types = [] + if is_packed: + inner = struct_body[2:-1].strip() + else: + inner = struct_body.strip('{}').strip() + if inner: + for elem in inner.split(','): + elem = elem.strip() + if elem: + et = self._parse_llvm_type(elem, Gen, source_sha1=actual_sha1) + elem_types.append(et if et else ir.IntType(32)) + try: + st = Gen._get_or_create_struct(class_name, source_sha1=actual_sha1, packed=is_packed) + if isinstance(st, ir.IdentifiedStructType) and st.is_opaque: + st.set_body(*elem_types) + if is_packed: + Gen.class_packed.add(class_name) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + stub_filename = actual_sha1 + '.stub.ll' + self._TryLoadClassMembersFromPyi(class_name, stub_filename, Gen) + + def _TryLoadClassMembersFromPyi(self, class_name: str, stub_filename: str, Gen: LlvmGeneratorMixin) -> None: + ProjectRoot = self._find_project_root() + temp_dir = os.path.join(ProjectRoot, 'temp') + if class_name in Gen.class_members and len(Gen.class_members[class_name]) > 0: + return + pyi_basename = stub_filename.replace('.stub.ll', '.pyi') + pyi_path = os.path.join(temp_dir, pyi_basename) + + if self._pyi_cache_dir != temp_dir: + self._pyi_cache.clear() + self._pyi_cache_dir = temp_dir + + if pyi_path not in self._pyi_cache: + if not os.path.isfile(pyi_path): + self._pyi_cache[pyi_path] = None + return + try: + with open(pyi_path, 'r', encoding='utf-8') as f: + self._pyi_cache[pyi_path] = ast.parse(f.read()) + except Exception as e: + _vlog().warning(f"HandlesImports: 忽略异常 {e}", exc_info=e) + self._pyi_cache[pyi_path] = None + return + + pyi_tree = self._pyi_cache[pyi_path] + if pyi_tree is None: + return + source_sha1 = stub_filename.replace('.stub.ll', '') + module_name = None + if hasattr(Gen, 'ModuleSha1Map'): + for mod_name, mod_sha1 in Gen.ModuleSha1Map.items(): + if mod_sha1 == source_sha1 and '.' not in mod_name: + module_name = mod_name + break + if module_name is None: + for mod_name, mod_sha1 in Gen.ModuleSha1Map.items(): + if mod_sha1 == source_sha1: + module_name = mod_name + break + if module_name is None and hasattr(Gen, 'module_sha1') and Gen.module_sha1 == source_sha1: + module_name = '__self__' + # module_name 反查失败时（当前模块未直接导入定义模块）， + # 仍需注册类成员和 vtable 信息。传 source_sha1 让 _EmitExternalClassDeclLlvm + # 正确设置 class_sha1_map 和注册 vtable。 + found = False + for node in pyi_tree.body: + if isinstance(node, ast.ClassDef) and node.name == class_name: + self._EmitExternalClassDeclLlvm(node, Gen, module_name=module_name, source_sha1=source_sha1) + found = True + break + if not found: + return + if class_name in Gen.class_members and len(Gen.class_members[class_name]) > 0: + return + return + + def _BuildScalarConstant(self, value_node: ast.AST, target_type: ir.Type, Gen: LlvmGeneratorMixin) -> ir.Constant | None: + """Build a scalar constant value from AST node, delegated to unified method.""" + return self.Trans._BuildScalarConstantUnified(value_node, target_type, Gen) + + def _LookupStubFuncType(self, func_name: str, Gen: LlvmGeneratorMixin) -> ir.FunctionType | None: + """从 stub.ll 文件中查找指定函数的 LLVM 类型""" + + if not getattr(self, '_stub_func_cache', None): + self._stub_func_cache = {} + + ProjectRoot = self._find_project_root() + temp_dir = os.path.join(ProjectRoot, 'temp') + + if not os.path.isdir(temp_dir): + return None + + if func_name not in self._stub_func_cache: + for filename in os.listdir(temp_dir): + if not filename.endswith('.stub.ll'): + continue + + stub_path = os.path.join(temp_dir, filename) + source_sha1 = filename.replace('.stub.ll', '') + try: + with open(stub_path, 'r', encoding='utf-8') as f: + content = f.read() + + for line in content.splitlines(): + stripped = line.strip() + if not stripped.startswith('declare '): + continue + + name_match = stripped.split('@', 1) + if len(name_match) < 2: + continue + + stub_func_name = name_match[1].split('(', 1)[0].strip().strip('"') + + match = re.match(r'declare\s+(.+?)\s+@("?[\w.]+"?)\s*\((.*)\)$', stripped) + if match: + stub_ret = match.group(1).strip() + stub_params = match.group(3).strip() + self._stub_func_cache[stub_func_name] = (stub_ret, stub_params, '...' in stub_params, source_sha1) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + + if func_name not in self._stub_func_cache: + return None + + ret_str, params_str, is_variadic, source_sha1 = self._stub_func_cache[func_name] + ret_type = self._parse_llvm_type(ret_str, Gen, source_sha1=source_sha1, create_structs=False) + if ret_type is None: + ret_type = ir.IntType(32) + param_types = [] + if params_str and params_str != 'void': + depth = 0 + current = '' + for ch in params_str: + if ch == ',' and depth == 0: + p = self._strip_llvm_param_name(current.strip()) + if p and p != '...': + pt = self._parse_llvm_type(p, Gen, source_sha1=source_sha1, create_structs=False) + if pt is not None: + param_types.append(pt) + else: + param_types.append(ir.PointerType(ir.IntType(8))) + current = '' + else: + if ch == '{': + depth += 1 + elif ch == '}': + depth -= 1 + current += ch + p = self._strip_llvm_param_name(current.strip()) + if p and p != '...': + pt = self._parse_llvm_type(p, Gen, source_sha1=source_sha1, create_structs=False) + if pt is not None: + param_types.append(pt) + else: + param_types.append(ir.PointerType(ir.IntType(8))) + return ir.FunctionType(ret_type, param_types, var_arg=is_variadic) + + def _strip_llvm_param_name(self, param_str: str) -> str: + """剥离 LLVM IR 参数字符串中的参数名，仅保留类型部分""" + param_str = param_str.strip() + result = re.sub(r'\s+%[\w.]+\s*$', '', param_str) + return result.strip() + + def _parse_simple_llvm_type(self, type_str: str) -> ir.Type | None: + """解析简单的 LLVM 类型字符串（不创建结构体，避免副作用）""" + + type_str = type_str.strip() + + type_map = { + 'void': ir.VoidType(), + 'i1': ir.IntType(1), + 'i8': ir.IntType(8), + 'i16': ir.IntType(16), + 'i32': ir.IntType(32), + 'i64': ir.IntType(64), + 'float': ir.FloatType(), + 'double': ir.DoubleType(), + } + + if type_str in type_map: + return type_map[type_str] + + if type_str.endswith('*'): + pointee = type_str[:-1].strip() + pointee_type = self._parse_simple_llvm_type(pointee) + if pointee_type: + return ir.PointerType(pointee_type) + return ir.PointerType(ir.IntType(8)) + + if type_str.startswith('[') and ']' in type_str: + arr_match = re.match(r'\[(\d+)\s*x\s+(.+)\]', type_str) + if arr_match: + size = int(arr_match.group(1)) + elem_type = self._parse_simple_llvm_type(arr_match.group(2).strip()) + if elem_type: + return ir.ArrayType(elem_type, size) + + if type_str.startswith('%'): + return ir.PointerType(ir.IntType(8)) + + return None + + def _LoadDeclarationsFromStubLlvm(self, module_name: str, Gen: LlvmGeneratorMixin) -> None: + """从 stub.ll 文件加载函数声明和结构体定义到 Gen""" + ProjectRoot = self._find_project_root() + temp_dir = os.path.join(ProjectRoot, 'temp') + + if not os.path.isdir(temp_dir): + return + + all_stub_files = [f for f in os.listdir(temp_dir) if f.endswith('.stub.ll')] + + source_sig_files = getattr(self.Trans, '_source_module_sig_files', {}) + target_sha1 = None + if module_name: + for key, val in source_sig_files.items(): + if key == module_name or key.endswith('.' + module_name): + target_sha1 = os.path.basename(val).replace('.pyi', '') + break + if not target_sha1: + ModuleSha1Map = getattr(Gen, 'ModuleSha1Map', {}) + target_sha1 = ModuleSha1Map.get(module_name) + + if target_sha1: + target_stub = f"{target_sha1}.stub.ll" + if target_stub in all_stub_files: + stub_files = [target_stub] + else: + stub_files = [] + else: + stub_files = [] + + needed_sha1s = set() + for filename in stub_files: + stub_path = os.path.join(temp_dir, filename) + try: + with open(stub_path, 'r', encoding='utf-8') as f: + content = f.read() + for line in content.splitlines(): + stripped = line.strip() + if stripped.startswith('%') and '=' in stripped and 'type' in stripped: + for m in re.finditer(r'%\"?([a-f0-9]{16})\.', stripped): + needed_sha1s.add(m.group(1)) + elif stripped.startswith('declare '): + for m in re.finditer(r'%\"?([a-f0-9]{16})\.', stripped): + needed_sha1s.add(m.group(1)) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + + preLoad_files = [] + for sha1 in needed_sha1s: + dep_stub = f"{sha1}.stub.ll" + if dep_stub in all_stub_files and dep_stub not in stub_files: + preLoad_files.append(dep_stub) + + Load_order = preLoad_files + stub_files + + for filename in Load_order: + stub_path = os.path.join(temp_dir, filename) + source_sha1 = filename.replace('.stub.ll', '') + is_preLoad = filename in preLoad_files + try: + with open(stub_path, 'r', encoding='utf-8') as f: + content = f.read() + + for line in content.splitlines(): + stripped = line.strip() + + if stripped.startswith('%') and '=' in stripped and 'type' in stripped: + struct_def_match = re.match(r'%"?((?:[a-f0-9]+\.)?([\w\[\]]+))"?\s*=\s*type\s*(.*)', stripped) + if struct_def_match: + raw_name = struct_def_match.group(1) + # 治本修复：从 raw_name 提取类型定义的实际 sha1 前缀， + # 而非使用 stub 文件的 sha1。同一 stub 文件可能包含多个模块的 + # struct 定义（前向引用），必须用类型名中的 sha1 作为 source_sha1， + # 否则同名类冲突检测会失效（如 __module.py stub 中的 + # e64295c03b564a53.Module 前向引用会被错误地用 6e3f70566907c6cf 作为 sha1）。 + actual_type_sha1 = source_sha1 + if '.' in raw_name: + _rn_parts = raw_name.split('.', 1) + if len(_rn_parts[0]) == 16 and all(_c in '0123456789abcdef' for _c in _rn_parts[0]): + actual_type_sha1 = _rn_parts[0] + clean_name = _rn_parts[1] + else: + clean_name = raw_name + struct_body = struct_def_match.group(3).strip() + if struct_body and struct_body != 'opaque': + elem_types = [] + is_packed = struct_body.startswith('<{') and struct_body.endswith('}>') + if is_packed: + inner = struct_body[2:-1].strip() + else: + inner = struct_body.strip('{}') + if inner: + for elem in inner.split(','): + elem = elem.strip() + if elem: + elem_type = self._parse_llvm_type(elem, Gen, source_sha1=actual_type_sha1) + if elem_type is not None: + elem_types.append(elem_type) + else: + elem_types.append(ir.IntType(32)) + st = Gen._get_or_create_struct(clean_name, source_sha1=actual_type_sha1, packed=is_packed) + if isinstance(st, ir.IdentifiedStructType) and st.elements is None: + try: + st.set_body(*elem_types) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + continue + + if not stripped.startswith('declare '): + continue + + if is_preLoad: + continue + + func_name_match = stripped.split('@', 1) + if len(func_name_match) < 2: + continue + + func_name = func_name_match[1].split('(', 1)[0].strip().strip('"') + + if not module_name: + continue + + # bare 名（无 16 位 hex SHA1 前缀）表示 CExport，调用点需用纯名 + _Prefix: str = func_name.split('.', 1)[0] + if not (len(_Prefix) == 16 and all(_c in '0123456789abcdef' for _c in _Prefix)): + Gen._export_funcs.add(func_name) + + try: + func_type = self._parse_llvm_declare(stripped, Gen) + if func_type is not None: + # __new__ 方法应返回类指针类型（堆分配替换），与 _GetFunctionSignatureLlvm/_EmitFunctionLlvm 保持一致 + # .stub.ll 由 .pyi 生成，.pyi 生成器对无返回类型注解的 __new__ 默认使用 t.CInt，这里修正为 ClassName* + if '.__new__' in func_name: + _np = func_name.split('.') + if len(_np[0]) == 16 and all(_c in '0123456789abcdef' for _c in _np[0]): + _new_class = _np[1] + else: + _new_class = _np[0] + _new_struct = Gen.structs.get(_new_class) + if _new_struct is None: + _full_name = func_name.split('.__new__')[0] + _new_struct = Gen.structs.get(_full_name) + if _new_struct: + _ret_type = _new_struct if isinstance(_new_struct, ir.PointerType) else ir.PointerType(_new_struct) + func_type = ir.FunctionType(_ret_type, func_type.args, var_arg=func_type.var_arg) + if func_name in Gen.functions: + existing_func = Gen.functions[func_name] + existing_ftype = getattr(existing_func, 'ftype', None) or existing_func.type.pointee + if str(existing_ftype) != str(func_type): + # 类型不一致：清理 module.globals + module.scope._useset 后重建，避免 DuplicatedNameError + old_name = existing_func.name + if old_name in Gen.module.globals: + del Gen.module.globals[old_name] + module_scope = getattr(Gen.module, 'scope', None) + if getattr(module_scope, '_useset', None): + module_scope._useset.discard(old_name) + func = ir.Function(Gen.module, func_type, name=func_name) + Gen.functions[func_name] = func + if '.' in func_name: + short_name = func_name.split('.', 1)[1] + Gen.functions[short_name] = func + else: + func = Gen._get_or_declare_function(func_name, func_type) + Gen.functions[func_name] = func + if '.' in func_name: + short_name = func_name.split('.', 1)[1] + Gen.functions[short_name] = func + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理导入失败: {_e}", "Exception") + + def _TryLoadFuncDeclsFromOutputStub(self, struct_name: str, Gen: LlvmGeneratorMixin) -> bool: + """按需从 output 目录的 stub.ll 加载泛型特化方法声明。 + + 当 temp stub 不包含泛型特化（如 list[str].__iter__）时， + 从 output stub（由 _split_ll 从完整 IR 生成）中按需加载。 + + struct_name: 结构体名，如 '9163064cf3eb88f4.list[str]' 或 'list[str]' + 返回 True 如果找到了并加载了新函数声明。 + """ + ProjectRoot = self._find_project_root() + output_dir = os.path.join(ProjectRoot, 'output') + if not os.path.isdir(output_dir): + return False + + # 提取短名称和 SHA1 前缀 + sha1_prefix: str | None = None + if '.' in struct_name: + parts: list[str] = struct_name.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + sha1_prefix = parts[0] + short_name: str = parts[1] + else: + short_name = struct_name + else: + short_name = struct_name + + # 确定要搜索的 stub 文件列表 + stub_files_to_search: list[str] = [] + if sha1_prefix: + target_stub = f'{sha1_prefix}.stub.ll' + target_path = os.path.join(output_dir, target_stub) + if os.path.isfile(target_path): + stub_files_to_search.append(target_stub) + else: + # 没有SHA1前缀，搜索所有 output stub + for fname in os.listdir(output_dir): + if fname.endswith('.stub.ll'): + stub_files_to_search.append(fname) + + loaded_any: bool = False + for fname in stub_files_to_search: + stub_path = os.path.join(output_dir, fname) + try: + with open(stub_path, 'r', encoding='utf-8') as f: + content = f.read() + for line in content.splitlines(): + stripped = line.strip() + if not stripped.startswith('declare '): + continue + # 检查是否是目标结构体的方法 + if f'.{short_name}.' not in stripped: + continue + try: + func_type = self._parse_llvm_declare(stripped, Gen) + if func_type is None: + continue + # 提取函数名：最后一个 @ 到第一个 ( 之间的内容 + paren_pos: int = stripped.index('(') + at_pos: int = stripped.rindex('@', 0, paren_pos) + func_name: str = stripped[at_pos + 1:paren_pos].strip().strip('"') + if func_name not in Gen.functions: + func = ir.Function(Gen.module, func_type, name=func_name) + Gen.functions[func_name] = func + if '.' in func_name: + short_fn = func_name.split('.', 1)[1] + Gen.functions[short_fn] = func + loaded_any = True + except Exception: + pass + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"按需加载 output stub 失败: {_e}", "Exception") + return loaded_any + + def _parse_llvm_declare(self, declare_line: str, Gen: LlvmGeneratorMixin) -> ir.FunctionType | None: + """解析 LLVM declare 语句""" + match = re.match(r'declare\s+(.+?)\s+@("?[^()]+"?)\s*\((.*)\)$', declare_line) + if not match: + return None + + ret_type_str = match.group(1).strip() + params_str = match.group(3).strip() + + ret_type = self._parse_llvm_type(ret_type_str, Gen) + if ret_type is None: + return None + + param_types = [] + if params_str and params_str != 'void': + depth = 0 + current = '' + for ch in params_str: + if ch == ',' and depth == 0: + p = self._strip_llvm_param_name(current.strip()) + if p and p != '...': + pt = self._parse_llvm_type(p, Gen) + if pt is not None: + param_types.append(pt) + current = '' + else: + if ch == '{': + depth += 1 + elif ch == '}': + depth -= 1 + current += ch + p = self._strip_llvm_param_name(current.strip()) + if p and p != '...': + pt = self._parse_llvm_type(p, Gen) + if pt is not None: + param_types.append(pt) + + is_variadic = '...' in params_str + return ir.FunctionType(ret_type, param_types, var_arg=is_variadic) + + def _parse_llvm_type(self, type_str: str, Gen: LlvmGeneratorMixin, source_sha1: str | None = None, create_structs: bool = True) -> ir.Type | None: + """解析 LLVM 类型字符串""" + + type_str = type_str.strip() + + type_map = { + 'void': ir.VoidType(), + 'i1': ir.IntType(1), + 'i8': ir.IntType(8), + 'i16': ir.IntType(16), + 'i32': ir.IntType(32), + 'i64': ir.IntType(64), + 'float': ir.FloatType(), + 'double': ir.DoubleType(), + } + + if type_str in type_map: + return type_map[type_str] + + if type_str.startswith('{') and type_str.endswith('}'): + inner = type_str[1:-1].strip() + field_types = [] + depth = 0 + current = '' + for ch in inner: + if ch == ',' and depth == 0: + field_types.append(current.strip()) + current = '' + else: + if ch == '{': + depth += 1 + elif ch == '}': + depth -= 1 + current += ch + if current.strip(): + field_types.append(current.strip()) + llvm_fields = [] + for ft in field_types: + parsed = self._parse_llvm_type(ft, Gen, source_sha1=source_sha1, create_structs=create_structs) + if parsed is not None: + llvm_fields.append(parsed) + else: + return None + if llvm_fields: + return ir.LiteralStructType(llvm_fields) + return None + + if type_str.endswith('*'): + pointee = type_str[:-1].strip() + pointee_type = self._parse_llvm_type(pointee, Gen, source_sha1=source_sha1, create_structs=create_structs) + if pointee_type is not None: + return ir.PointerType(pointee_type) + return ir.PointerType(ir.IntType(8)) + + if type_str.startswith('[') and ']' in type_str: + arr_match = re.match(r'\[(\d+)\s*x\s+(.+)\]', type_str) + if arr_match: + size = int(arr_match.group(1)) + elem_type = self._parse_llvm_type(arr_match.group(2).strip(), Gen, source_sha1=source_sha1, create_structs=create_structs) + if elem_type is not None: + return ir.ArrayType(elem_type, size) + + typedef_names = {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum'} + struct_match = re.match(r'%"?((?:[a-f0-9]+\.)?[\w.\[\]]+)"?', type_str) + if struct_match: + raw_name = struct_match.group(1) + actual_source_sha1 = source_sha1 + if '.' in raw_name: + parts = raw_name.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + actual_source_sha1 = parts[0] + struct_name = parts[1] + else: + struct_name = raw_name + if struct_name not in typedef_names: + # 泛型类特化（如 list[str]）：触发特化创建结构体和方法声明 + # 使用 declare_only=True 避免在引用模块中生成方法体（定义由定义模块提供） + if '[' in struct_name and actual_source_sha1: + if struct_name in Gen.structs: + return Gen.structs[struct_name] + gc_base: str = struct_name.split('[')[0] + gc_args_raw: str = struct_name[len(gc_base):] + gc_type_args: list[str] = [] + for gc_arg_part in gc_args_raw.split(']['): + gc_arg_part = gc_arg_part.strip('[]') + if gc_arg_part: + gc_type_args.append(gc_arg_part) + if (gc_type_args + and hasattr(self.Trans, 'ClassHandler') + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and gc_base in self.Trans.ClassHandler._generic_class_templates): + gc_type_names: list[str] = [] + for gc_ta in gc_type_args: + gc_tn: str = gc_ta + if gc_ta == 'int': + gc_tn = 'CInt' + elif gc_ta == 'double': + gc_tn = 'CDouble' + elif gc_ta == 'float': + gc_tn = 'CFloat' + elif gc_ta == 'char': + gc_tn = 'CChar' + elif gc_ta == 'bool': + gc_tn = 'CBool' + gc_type_names.append(gc_tn) + saved_module_sha1: str | None = getattr(Gen, 'module_sha1', None) + saved_trans_sha1: str | None = getattr(self.Trans, '_module_sha1', None) + Gen.module_sha1 = actual_source_sha1 + self.Trans._module_sha1 = actual_source_sha1 + try: + gc_spec: str | None = self.Trans.ClassHandler._specialize_generic_class( + gc_base, gc_type_args, Gen, type_names=gc_type_names, + declare_only=True) + except Exception: + gc_spec = None + Gen.module_sha1 = saved_module_sha1 + self.Trans._module_sha1 = saved_trans_sha1 + if gc_spec and gc_spec in Gen.structs: + return Gen.structs[gc_spec] + if create_structs: + return Gen._get_or_create_struct(struct_name, source_sha1=actual_source_sha1) + else: + existing = Gen.structs.get(struct_name) + if existing is not None: + return existing + return None + return ir.IntType(32) + + return None + + def _strip_param_name(self, param_str: str) -> str: + """从 LLVM 参数字符串中剥离参数名，只保留类型部分 + 例如: '%"sha1.struct_name"* %param_name' -> '%"sha1.struct_name"*' + 'i8* %path' -> 'i8*' + 'i32' -> 'i32' + """ + param_str = param_str.strip() + stripped = re.sub(r'\s+%[\w.]+\s*$', '', param_str) + return stripped.strip() diff --git a/lib/core/Handles/HandlesMatch.py b/lib/core/Handles/HandlesMatch.py new file mode 100644 index 0000000..4b8e78a --- /dev/null +++ b/lib/core/Handles/HandlesMatch.py @@ -0,0 +1,382 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class MatchHandle(BaseHandle): + def _HandleMatchLlvm(self, Node: ast.Match) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + SubjectVal: ir.Value | None = self.HandleExprLlvm(Node.subject) + if not SubjectVal: + return + + IsRenumMatch: bool = False + RenumName: str | None = None + SubjectPtr: ir.Value | None = None + if isinstance(Node.subject, ast.Name): + VarName: str = Node.subject.id + TypeInfo: "SymbolTable.SymbolInfo | None" = self.Trans.SymbolTable.lookup(VarName) + if TypeInfo and TypeInfo.IsRenum: + IsRenumMatch = True + RenumName = TypeInfo.Name + SubjectPtr = Gen._Load_var(VarName) + + if not IsRenumMatch: + for case in Node.cases: + if isinstance(case.pattern, ast.MatchClass): + cls_node: ast.expr = case.pattern.cls + VariantName: str | None = None + QualifiedName: str | None = None + if isinstance(cls_node, ast.Name): + VariantName = cls_node.id + elif isinstance(cls_node, ast.Attribute): + VariantName = cls_node.attr + # Extract enum name for qualified lookup to avoid collision + # with same-name factory functions (e.g. def Ptr vs LLVMType.Ptr) + if isinstance(cls_node.value, ast.Attribute): + QualifiedName = f"{cls_node.value.attr}.{VariantName}" + elif isinstance(cls_node.value, ast.Name): + QualifiedName = f"{cls_node.value.id}.{VariantName}" + if VariantName: + # Try qualified name first (e.g., "LLVMType.Ptr") to avoid + # collision with same-name functions (e.g., def Ptr(...)) + SymInfo: "SymbolTable.SymbolInfo" = None + if QualifiedName: + SymInfo = self.Trans.SymbolTable.lookup(QualifiedName) + if not (SymInfo and SymInfo.IsEnumMember): + SymInfo = self.Trans.SymbolTable.lookup(VariantName) + if SymInfo and SymInfo.IsEnumMember and SymInfo.EnumName: + EnumName: str = SymInfo.EnumName + EnumInfo: "SymbolTable.SymbolInfo" = self.Trans.SymbolTable.lookup(EnumName) + if EnumInfo and EnumInfo.IsRenum: + IsRenumMatch = True + RenumName = EnumName + if SubjectPtr is None: + SubjectPtr = self.HandleExprLlvm(Node.subject) + break + + if IsRenumMatch and SubjectPtr: + self._HandleRenumMatchLlvm(Node, RenumName, SubjectPtr) + return + + if not isinstance(SubjectVal.type, ir.IntType): + try: + SubjectVal = Gen.builder.ptrtoint(SubjectVal, ir.IntType(64), name="match_subj") + SubjectVal = Gen.builder.trunc(SubjectVal, ir.IntType(32), name="match_subj_i32") + except Exception: # 回退：ptrtoint 失败时直接返回 + return + SwitchIntType: ir.IntType = SubjectVal.type + DefaultBB: ir.Block = Gen.func.append_basic_block(name="match.default") + AfterBB: ir.Block = Gen.func.append_basic_block(name="match.end") + CaseBBs: list[ir.Block] = [] + CaseValues: list[ir.Value] = [] + HasDefault: bool = False + HasNoBreak: list[bool] = [] + for i, case in enumerate(Node.cases): + pattern: ast.pattern = case.pattern + if isinstance(pattern, ast.MatchValue): + Val: ir.Value | None = self.HandleExprLlvm(pattern.value) + if Val: + CaseVal: ir.Value + if isinstance(Val.type, ir.IntType): + if Val.type != SwitchIntType: + if Val.type.width > SwitchIntType.width: + CaseVal = ir.Constant(SwitchIntType, Val.constant & ((1 << SwitchIntType.width) - 1)) + else: + CaseVal = ir.Constant(SwitchIntType, Val.constant) + else: + CaseVal = Val + else: + try: + CaseVal = Gen.builder.ptrtoint(Val, SwitchIntType, name=f"case_val_{i}") + except Exception: # 回退：ptrtoint 失败时设默认值 0 + CaseVal = ir.Constant(SwitchIntType, 0) + CaseValues.append(CaseVal) + CaseBBs.append(Gen.func.append_basic_block(name=f"match.case_{i}")) + elif isinstance(pattern, ast.MatchOr): + for j, SubPattern in enumerate(pattern.patterns): + if isinstance(SubPattern, ast.MatchValue): + Val: ir.Value | None = self.HandleExprLlvm(SubPattern.value) + if Val: + CaseVal: ir.Value + if isinstance(Val.type, ir.IntType): + if Val.type != SwitchIntType: + if Val.type.width > SwitchIntType.width: + CaseVal = ir.Constant(SwitchIntType, Val.constant & ((1 << SwitchIntType.width) - 1)) + else: + CaseVal = ir.Constant(SwitchIntType, Val.constant) + else: + CaseVal = Val + else: + try: + CaseVal = Gen.builder.ptrtoint(Val, SwitchIntType, name=f"case_val_{i}_{j}") + except Exception: # 回退：ptrtoint 失败时设默认值 0 + CaseVal = ir.Constant(SwitchIntType, 0) + CaseValues.append(CaseVal) + CaseBBs.append(Gen.func.append_basic_block(name=f"match.case_{i}_{j}")) + elif isinstance(pattern, ast.MatchSingleton): + if pattern.value is None: + CaseValues.append(ir.Constant(SwitchIntType, 0)) + CaseBBs.append(Gen.func.append_basic_block(name=f"match.case_{i}")) + elif isinstance(pattern, ast.MatchAs): + if pattern.pattern is None: + HasDefault = True + CaseBBs.append(DefaultBB) + elif isinstance(pattern, ast.MatchSequence): + HasDefault = True + CaseBBs.append(DefaultBB) + def _HasNoBreak(stmts: list[ast.stmt]) -> bool: + for stmt in stmts: + if isinstance(stmt, ast.Expr) and isinstance(stmt.value, ast.Call): + if isinstance(stmt.value.func, ast.Attribute): + if (isinstance(stmt.value.func.value, ast.Name) and + stmt.value.func.value.id == 'c' and + stmt.value.func.attr == 'NoBreak'): + return True + if getattr(stmt, 'body', None) and isinstance(stmt.body, list): + if _HasNoBreak(stmt.body): + return True + if getattr(stmt, 'orelse', None): + if isinstance(stmt.orelse, list) and _HasNoBreak(stmt.orelse): + return True + return False + HasNoBreak.append(_HasNoBreak(case.body) if case.body else False) + if not HasDefault: + CaseBBs.append(DefaultBB) + SwitchCases: list[tuple[ir.Value, ir.Block]] = [(val, bb) for val, bb in zip(CaseValues, CaseBBs) if bb != DefaultBB] + switch_instr: ir.SwitchInstr = Gen.builder.switch(SubjectVal, DefaultBB) + for val, bb in SwitchCases: + switch_instr.add_case(val, bb) + CaseIdx: int = 0 + for i, case in enumerate(Node.cases): + pattern: ast.pattern = case.pattern + if isinstance(pattern, ast.MatchOr): + NumSubCases: int = len(pattern.patterns) + for j in range(NumSubCases): + if CaseIdx < len(CaseBBs) and CaseBBs[CaseIdx] != DefaultBB: + Gen.builder.position_at_start(CaseBBs[CaseIdx]) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, (ast.MatchValue, ast.MatchSingleton)): + if CaseIdx < len(CaseBBs) and CaseBBs[CaseIdx] != DefaultBB: + Gen.builder.position_at_start(CaseBBs[CaseIdx]) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, ast.MatchAs): + if pattern.pattern is None: + Gen.builder.position_at_start(DefaultBB) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, ast.MatchSequence): + Gen.builder.position_at_start(DefaultBB) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + if not HasDefault: + Gen.builder.position_at_start(DefaultBB) + Gen.builder.branch(AfterBB) + elif not any(isinstance(c.pattern, ast.MatchAs) and c.pattern.pattern is None for c in Node.cases) and not any(isinstance(c.pattern, ast.MatchSequence) for c in Node.cases): + Gen.builder.position_at_start(DefaultBB) + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + + def _HandleRenumMatchLlvm(self, Node: ast.Match, RenumName: str, SubjectPtr: ir.Value) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if isinstance(SubjectPtr.type, ir.PointerType) and isinstance(SubjectPtr.type.pointee, ir.PointerType): + SubjectPtr = Gen._load(SubjectPtr, name="Load_match_subj") + # Bug fix: HandleExprLlvm 可能对 REnum 字段执行了 load，返回值而非指针。 + # 情况1: SubjectPtr 不是指针类型（如 IntType）→ alloca REnum 结构体 + store + # 情况2: SubjectPtr 是指针但 pointee 不是结构体（如 i32* 指向 __tag）→ bitcast + NeedAlloca: bool = not isinstance(SubjectPtr.type, ir.PointerType) + NeedBitcast: bool = (isinstance(SubjectPtr.type, ir.PointerType) and + not isinstance(SubjectPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType))) + if NeedAlloca or NeedBitcast: + RenumStructType: Any = Gen.structs.get(RenumName) + if RenumStructType is None: + # 跨模块编译时 REnum 结构体可能尚未在 Gen.structs 中注册 + # （如 import llvmlite 后 LLVMType 仅在符号表，Gen.structs 无条目）。 + # 使用 _get_or_create_struct 按需创建（可能为 opaque）。 + RenumStructType = Gen._get_or_create_struct(RenumName) + if NeedAlloca: + AllocaPtr: ir.Value = Gen._allocaEntry(RenumStructType, name="match_subj_alloca") + CastedPtr: ir.Value = Gen.builder.bitcast(AllocaPtr, ir.PointerType(SubjectPtr.type), name="match_subj_cast") + Gen._store(SubjectPtr, CastedPtr) + SubjectPtr = AllocaPtr + else: + SubjectPtr = Gen.builder.bitcast(SubjectPtr, ir.PointerType(RenumStructType), name="match_subj_recast") + # REnum 布局为 { i32 __tag, }，tag 在偏移 0。 + # 使用 bitcast 到 i32* 替代 gep [0,0]，兼容 opaque 结构体 + # （跨模块编译时 REnum 结构体可能未 set_body，gep 会失败）。 + tag_ptr: ir.Value = Gen.builder.bitcast(SubjectPtr, ir.PointerType(ir.IntType(32)), name="match_tag_ptr") + TagVal: ir.Value = Gen._load(tag_ptr, name="match_tag_val") + DefaultBB: ir.Block = Gen.func.append_basic_block(name="match.default") + AfterBB: ir.Block = Gen.func.append_basic_block(name="match.end") + CaseBBs: list[ir.Block] = [] + CaseValues: list[ir.Value] = [] + CaseBindings: list[list[tuple[str, str, ir.Type, int]]] = [] + HasDefault: bool = False + HasNoBreak: list[bool] = [] + for i, case in enumerate(Node.cases): + pattern: ast.pattern = case.pattern + bindings: list[tuple[str, str, ir.Type, int]] = [] + if isinstance(pattern, ast.MatchClass): + cls_node: ast.expr = pattern.cls + VariantName: str | None = None + QualifiedName: str | None = None + if isinstance(cls_node, ast.Name): + VariantName = cls_node.id + elif isinstance(cls_node, ast.Attribute): + VariantName = cls_node.attr + if isinstance(cls_node.value, ast.Attribute): + QualifiedName = f"{cls_node.value.attr}.{VariantName}" + elif isinstance(cls_node.value, ast.Name): + QualifiedName = f"{cls_node.value.id}.{VariantName}" + if VariantName: + TagValue: int | None = None + SymInfo: "SymbolTable.SymbolInfo" = None + if QualifiedName: + SymInfo = self.Trans.SymbolTable.lookup(QualifiedName) + if not (SymInfo and SymInfo.IsEnumMember): + SymInfo = self.Trans.SymbolTable.lookup(VariantName) + if SymInfo and SymInfo.IsEnumMember: + TagValue = SymInfo.value + if TagValue is not None: + CaseValues.append(ir.Constant(ir.IntType(32), TagValue)) + CaseBB: ir.Block = Gen.func.append_basic_block(name=f"match.case_{VariantName}") + CaseBBs.append(CaseBB) + NestedStructName: str = f"{RenumName}_{VariantName}" + if NestedStructName in Gen.structs: + members: list[tuple[str, ir.Type]] = Gen.class_members.get(NestedStructName, []) + payLoad_members: list[tuple[str, ir.Type]] = [(n, t) for n, t in members if n != '__tag'] + for j, sub_pat in enumerate(pattern.patterns): + if isinstance(sub_pat, ast.MatchAs) and sub_pat.name and j < len(payLoad_members): + bindings.append((sub_pat.name, payLoad_members[j][0], payLoad_members[j][1], j)) + CaseBindings.append(bindings) + else: + CaseValues.append(ir.Constant(ir.IntType(32), 0)) + CaseBBs.append(Gen.func.append_basic_block(name=f"match.case_{i}")) + CaseBindings.append([]) + elif isinstance(pattern, ast.MatchValue): + Val: ir.Value | None = self.HandleExprLlvm(pattern.value) + if Val: + CaseVal: ir.Value + if isinstance(Val.type, ir.IntType): + CaseVal = Val + else: + try: + CaseVal = Gen.builder.ptrtoint(Val, ir.IntType(32), name=f"case_val_{i}") + except Exception: # 回退：ptrtoint 失败时设默认值 0 + CaseVal = ir.Constant(ir.IntType(32), 0) + CaseValues.append(CaseVal) + CaseBBs.append(Gen.func.append_basic_block(name=f"match.case_{i}")) + CaseBindings.append([]) + elif isinstance(pattern, ast.MatchAs): + if pattern.pattern is None: + HasDefault = True + CaseBBs.append(DefaultBB) + CaseBindings.append([]) + elif isinstance(pattern, ast.MatchSequence): + HasDefault = True + CaseBBs.append(DefaultBB) + CaseBindings.append([]) + def _HasNoBreak(stmts: list[ast.stmt]) -> bool: + for stmt in stmts: + if isinstance(stmt, ast.Expr) and isinstance(stmt.value, ast.Call): + if isinstance(stmt.value.func, ast.Attribute): + if (isinstance(stmt.value.func.value, ast.Name) and + stmt.value.func.value.id == 'c' and + stmt.value.func.attr == 'NoBreak'): + return True + if getattr(stmt, 'body', None) and isinstance(stmt.body, list): + if _HasNoBreak(stmt.body): + return True + if getattr(stmt, 'orelse', None): + if isinstance(stmt.orelse, list) and _HasNoBreak(stmt.orelse): + return True + return False + HasNoBreak.append(_HasNoBreak(case.body) if case.body else False) + if not HasDefault: + CaseBBs.append(DefaultBB) + SwitchCases: list[tuple[ir.Value, ir.Block]] = [(val, bb) for val, bb in zip(CaseValues, CaseBBs) if bb != DefaultBB] + switch_instr: ir.SwitchInstr = Gen.builder.switch(TagVal, DefaultBB) + for val, bb in SwitchCases: + switch_instr.add_case(val, bb) + CaseIdx: int = 0 + for i, case in enumerate(Node.cases): + pattern: ast.pattern = case.pattern + if isinstance(pattern, ast.MatchClass): + if CaseIdx < len(CaseBBs) and CaseBBs[CaseIdx] != DefaultBB: + Gen.builder.position_at_start(CaseBBs[CaseIdx]) + bindings: list[tuple[str, str, ir.Type, int]] = CaseBindings[CaseIdx] if CaseIdx < len(CaseBindings) else [] + cls_node: ast.expr = pattern.cls + VariantName: str | None = None + if isinstance(cls_node, ast.Name): + VariantName = cls_node.id + elif isinstance(cls_node, ast.Attribute): + VariantName = cls_node.attr + if VariantName: + NestedStructName: str = f"{RenumName}_{VariantName}" + if NestedStructName in Gen.structs: + NestedStructType: ir.Type = Gen.structs[NestedStructName] + NestedStructPtrType: ir.PointerType = ir.PointerType(NestedStructType) + variant_ptr: ir.Value = Gen.builder.bitcast(SubjectPtr, NestedStructPtrType, name=f"match_cast_{VariantName}") + for bind_name, member_name, member_type, member_idx in bindings: + elem_ptr: ir.Value = Gen.builder.gep(variant_ptr, [ir.Constant(ir.IntType(32), 0), ir.Constant(ir.IntType(32), member_idx + 1)], name=f"match_{bind_name}") + # REnum 嵌套结构体共享 max_variant_struct 布局，成员类型可能与 + # 结构体字段类型不一致。bitcast 到成员类型以确保后续 load 得到正确类型。 + if isinstance(elem_ptr.type, ir.PointerType) and elem_ptr.type.pointee != member_type: + elem_ptr = Gen.builder.bitcast(elem_ptr, ir.PointerType(member_type), name=f"match_{bind_name}_cast") + Gen.variables[bind_name] = elem_ptr + self.HandleBodyLlvm(case.body) + for bind_name, _, _, _ in bindings: + if bind_name in Gen.variables: + del Gen.variables[bind_name] + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, ast.MatchValue): + if CaseIdx < len(CaseBBs) and CaseBBs[CaseIdx] != DefaultBB: + Gen.builder.position_at_start(CaseBBs[CaseIdx]) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, ast.MatchAs): + if pattern.pattern is None: + Gen.builder.position_at_start(DefaultBB) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + elif isinstance(pattern, ast.MatchSequence): + Gen.builder.position_at_start(DefaultBB) + self.HandleBodyLlvm(case.body) + if not Gen.builder.block.is_terminated: + if not HasNoBreak[i]: + Gen.builder.branch(AfterBB) + CaseIdx += 1 + if not HasDefault: + Gen.builder.position_at_start(DefaultBB) + Gen.builder.branch(AfterBB) + elif not any(isinstance(c.pattern, ast.MatchAs) and c.pattern.pattern is None for c in Node.cases) and not any(isinstance(c.pattern, ast.MatchSequence) for c in Node.cases): + Gen.builder.position_at_start(DefaultBB) + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesRaise.py b/lib/core/Handles/HandlesRaise.py new file mode 100644 index 0000000..9802e92 --- /dev/null +++ b/lib/core/Handles/HandlesRaise.py @@ -0,0 +1,155 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +import ast +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import BaseHandle, EXCEPTION_CODE_MAP +from lib.constants.config import mode as _config_mode + + +class RaiseHandle(BaseHandle): + def _get_exception_code(self, ExcName: str) -> int: + if ExcName in EXCEPTION_CODE_MAP: + return EXCEPTION_CODE_MAP[ExcName] + if ExcName in self.Trans.exception_registry: + return self.Trans.exception_registry[ExcName] + return 1 + + def _HandleRaiseLlvm(self, Node: ast.Raise) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + exc_val: ir.Constant = ir.Constant(ir.IntType(32), 1) + exc_msg: ir.Value | None = None + IsStopIteration: bool = False + if Node.exc: + if isinstance(Node.exc, ast.Constant) and isinstance(Node.exc.value, int): + exc_val = ir.Constant(ir.IntType(32), Node.exc.value) + elif isinstance(Node.exc, ast.Name): + ExcName: str = Node.exc.id + if ExcName == 'StopIteration': + IsStopIteration = True + exc_val = ir.Constant(ir.IntType(32), self._get_exception_code(ExcName)) + elif isinstance(Node.exc, ast.Call) and isinstance(Node.exc.func, ast.Name): + ExcName: str = Node.exc.func.id + if ExcName == 'StopIteration': + IsStopIteration = True + exc_val = ir.Constant(ir.IntType(32), self._get_exception_code(ExcName)) + if Node.exc.args: + first_arg: ast.expr = Node.exc.args[0] + if isinstance(first_arg, ast.Constant) and isinstance(first_arg.value, str): + exc_msg = self.HandleExprLlvm(first_arg) + elif isinstance(first_arg, (ast.Name, ast.Call, ast.Attribute)): + arg_val: ir.Value | None = self.HandleExprLlvm(first_arg) + if arg_val: + if isinstance(arg_val.type, ir.PointerType): + if self._is_char_pointer(arg_val): + exc_msg = arg_val + else: + try: + exc_msg = Gen.builder.bitcast(arg_val, ir.PointerType(ir.IntType(8)), name="exc_msg_cast") + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + else: + try: + exc_msg = Gen.builder.inttoptr(arg_val, ir.PointerType(ir.IntType(8)), name="exc_msg_ptr") + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + else: + val: ir.Value | None = self.HandleExprLlvm(Node.exc) + if val: + if isinstance(val.type, ir.IntType): + if val.type.width != 32: + try: + val = Gen.builder.trunc(val, ir.IntType(32), name="exc_trunc") + except Exception: # 回退：trunc 失败时设默认值 1 + val = ir.Constant(ir.IntType(32), 1) + exc_val = val + else: + try: + val = Gen.builder.ptrtoint(val, ir.IntType(32), name="exc_ptr2int") + exc_val = val + except Exception as _e: + if _config_mode == "strict": + self.Trans.LogWarning(f"异常被忽略: {_e}") + if IsStopIteration: + if Gen._stop_iter_flag_param is not None: + Gen.builder.store(ir.Constant(ir.IntType(1), 1), Gen._stop_iter_flag_param) + if Gen.func and Gen.func.type.pointee.return_type != ir.VoidType(): + ret_type: ir.Type = Gen.func.type.pointee.return_type + if isinstance(ret_type, ir.PointerType): + Gen.builder.ret(ir.Constant(ret_type, None)) + elif isinstance(ret_type, ir.IntType): + Gen.builder.ret(ir.Constant(ret_type, 0)) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ret_type, 0.0)) + elif isinstance(ret_type, ir.BaseStructType): + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ + ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) + else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) + else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + Gen.builder.ret(zero_val) + else: + Gen.builder.ret(ir.Constant(ret_type, 0)) + else: + Gen.builder.ret_void() + return + if Gen.eh_except_block_stack: + ExceptBB: ir.Block = Gen.eh_except_block_stack[-1][0] + EndBB: ir.Block | None = Gen.eh_except_block_stack[-1][1] + exception_code: ir.AllocaInstr = Gen.eh_except_block_stack[-1][2] + eh_message: ir.AllocaInstr = Gen.eh_except_block_stack[-1][3] + Gen._store(exc_val, exception_code) + if exc_msg: + Gen._store(exc_msg, eh_message) + else: + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + Gen._store(null_ptr, eh_message) + if Gen.func and ExceptBB and ExceptBB.function is Gen.func: + Gen.builder.branch(ExceptBB) + else: + eh_msg_arg: ir.Argument | None = None + eh_code_arg: ir.Argument | None = None + if Gen.func: + for arg in Gen.func.args: + if arg.name == '__eh_msg_out__': + eh_msg_arg = arg + elif arg.name == '__eh_code_out__': + eh_code_arg = arg + if eh_msg_arg is not None: + if exc_msg: + Gen.builder.store(exc_msg, eh_msg_arg) + else: + null_ptr = ir.Constant(ir.PointerType(ir.IntType(8)), None) + Gen.builder.store(null_ptr, eh_msg_arg) + if eh_code_arg is not None: + Gen.builder.store(exc_val, eh_code_arg) + if Gen.func and Gen.func.type.pointee.return_type == ir.IntType(32): + Gen.builder.ret(ir.Constant(ir.IntType(32), 1)) + else: + Gen.builder.ret_void() + else: + eh_msg_arg: ir.Argument | None = None + eh_code_arg: ir.Argument | None = None + if Gen.func: + for arg in Gen.func.args: + if arg.name == '__eh_msg_out__': + eh_msg_arg = arg + elif arg.name == '__eh_code_out__': + eh_code_arg = arg + if eh_msg_arg is not None: + if exc_msg: + Gen.builder.store(exc_msg, eh_msg_arg) + else: + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + Gen.builder.store(null_ptr, eh_msg_arg) + if eh_code_arg is not None: + Gen.builder.store(exc_val, eh_code_arg) + if Gen.func and Gen.func.type.pointee.return_type == ir.IntType(32): + Gen.builder.ret(ir.Constant(ir.IntType(32), 1)) + else: + Gen.builder.ret_void() \ No newline at end of file diff --git a/lib/core/Handles/HandlesReturn.py b/lib/core/Handles/HandlesReturn.py new file mode 100644 index 0000000..be02744 --- /dev/null +++ b/lib/core/Handles/HandlesReturn.py @@ -0,0 +1,165 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle, CTypeInfo +import ast +import llvmlite.ir as ir + + +class ReturnHandle(BaseHandle): + def _find_active_finally(self, Gen: "Translator.LlvmGen") -> tuple[ir.Block, ir.AllocaInstr, ir.AllocaInstr | None] | None: + for item in reversed(Gen.eh_finally_stack): + if item is not None and item[0] is not None: + return item + return None + + def _HandleReturnLlvm(self, Node: ast.Return) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + active_finally: tuple[ir.Block, ir.AllocaInstr, ir.AllocaInstr | None] | None = self._find_active_finally(Gen) + if getattr(self.Trans, 'CurrentCReturnTypes', None) and self.Trans.CurrentCReturnTypes and Node.value: + return_values: list[ast.expr] = [] + if isinstance(Node.value, ast.Tuple): + return_values = Node.value.elts + else: + return_values = [Node.value] + if len(return_values) == len(self.Trans.CurrentCReturnTypes): + llvm_vals: list[ir.Value] = [] + expected_types: list[ir.Type] = [] + for i, val in enumerate(return_values): + ReturnTypeInfo: CTypeInfo | None = CTypeInfo.FromNode(self.Trans.CurrentCReturnTypes[i], self.Trans.SymbolTable) + IsRetPtr: bool = ReturnTypeInfo.IsPtr if ReturnTypeInfo else False + if (isinstance(ReturnTypeInfo, CTypeInfo) and ReturnTypeInfo.IsStr) or (isinstance(self.Trans.CurrentCReturnTypes[i], ast.Name) and self.Trans.CurrentCReturnTypes[i].id == 'str'): + IsRetPtr = True + ExpectedType: ir.Type = ReturnTypeInfo.ToLLVM(Gen) if ReturnTypeInfo and ReturnTypeInfo.BaseType else Gen._CType2LLVM('int', IsRetPtr) + expected_types.append(ExpectedType) + Val: ir.Value | None = self.HandleExprLlvm(val) + if Val is None: + if isinstance(ExpectedType, ir.PointerType): + Val = ir.Constant(ExpectedType, None) + elif isinstance(ExpectedType, ir.IntType): + Val = ir.Constant(ExpectedType, 0) + else: + Val = ir.Constant(ExpectedType, ir.Undefined) + else: + if Val.type != ExpectedType: + if isinstance(ExpectedType, ir.IntType) and isinstance(Val.type, ir.IntType): + if Val.type.width < ExpectedType.width: + Val = Gen.builder.zext(Val, ExpectedType, name=f"zext_ret_{i}") + elif Val.type.width > ExpectedType.width: + Val = Gen.builder.trunc(Val, ExpectedType, name=f"trunc_ret_{i}") + elif isinstance(ExpectedType, ir.PointerType) and isinstance(Val.type, ir.PointerType): + Val = Gen.builder.bitcast(Val, ExpectedType, name=f"cast_ret_{i}") + llvm_vals.append(Val) + if llvm_vals: + struct_type: ir.LiteralStructType = ir.LiteralStructType(expected_types) + result: ir.Value = ir.Constant(struct_type, ir.Undefined) + for i, val in enumerate(llvm_vals): + result = Gen.builder.insert_value(result, val, i, name=f"ret_field_{i}") + if active_finally: + FinallyBB: ir.Block = active_finally[0] + pending_ret_flag: ir.AllocaInstr = active_finally[1] + pending_ret_val: ir.AllocaInstr | None = active_finally[2] + if pending_ret_val: + Gen._store(result, pending_ret_val) + Gen._store(ir.Constant(ir.IntType(32), 1), pending_ret_flag) + Gen.builder.branch(FinallyBB) + else: + Gen.emit_return(result) + return + if Node.value: + Val: ir.Value | None = self.HandleExprLlvm(Node.value) + if Val: + ret_type: ir.Type | None = getattr(Gen.func, 'ftype', None) and Gen.func.ftype.return_type if getattr(Gen, 'func', None) else None + if isinstance(ret_type, ir.IdentifiedStructType) and isinstance(Val, ir.Constant) and isinstance(Val.type, ir.PointerType) and Val.constant is None: + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ + ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) + else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) + else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + if active_finally: + FinallyBB: ir.Block = active_finally[0] + pending_ret_flag: ir.AllocaInstr = active_finally[1] + pending_ret_val: ir.AllocaInstr | None = active_finally[2] + if pending_ret_val: + Gen._store(zero_val, pending_ret_val) + Gen._store(ir.Constant(ir.IntType(32), 1), pending_ret_flag) + Gen.builder.branch(FinallyBB) + else: + Gen.emit_return(zero_val) + return + if getattr(self.Trans, 'CurrentCReturnTypes', None) and self.Trans.CurrentCReturnTypes: + struct_type: ir.Type = Gen.func.ftype.return_type + if isinstance(struct_type, ir.LiteralStructType): + result: ir.Value = ir.Constant(struct_type, ir.Undefined) + if isinstance(Val.type, ir.PointerType) and len(struct_type.elements) > 0 and isinstance(struct_type.elements[0], ir.PointerType): + result = Gen.builder.insert_value(result, Val, 0, name="ret_field_0") + for j in range(1, len(struct_type.elements)): + elem_type: ir.Type = struct_type.elements[j] + zero_val: ir.Constant = ir.Constant(elem_type, 0 if isinstance(elem_type, ir.IntType) else None) + result = Gen.builder.insert_value(result, zero_val, j, name=f"ret_field_{j}") + else: + for j, elem_type in enumerate(struct_type.elements): + if j == 0: + if Val.type == elem_type: + result = Gen.builder.insert_value(result, Val, j, name=f"ret_field_{j}") + elif isinstance(elem_type, ir.IntType) and isinstance(Val.type, ir.IntType): + if Val.type.width < elem_type.width: + Val = Gen.builder.zext(Val, elem_type, name=f"zext_ret_{j}") + elif Val.type.width > elem_type.width: + Val = Gen.builder.trunc(Val, elem_type, name=f"trunc_ret_{j}") + result = Gen.builder.insert_value(result, Val, j, name=f"ret_field_{j}") + else: + zero_val = ir.Constant(elem_type, 0 if isinstance(elem_type, ir.IntType) else None) + result = Gen.builder.insert_value(result, zero_val, j, name=f"ret_field_{j}") + else: + zero_val = ir.Constant(elem_type, 0 if isinstance(elem_type, ir.IntType) else None) + result = Gen.builder.insert_value(result, zero_val, j, name=f"ret_field_{j}") + if active_finally: + FinallyBB: ir.Block = active_finally[0] + pending_ret_flag: ir.AllocaInstr = active_finally[1] + pending_ret_val: ir.AllocaInstr | None = active_finally[2] + if pending_ret_val: + Gen._store(result, pending_ret_val) + Gen._store(ir.Constant(ir.IntType(32), 1), pending_ret_flag) + Gen.builder.branch(FinallyBB) + else: + Gen.emit_return(result) + return + if active_finally: + FinallyBB: ir.Block = active_finally[0] + pending_ret_flag: ir.AllocaInstr = active_finally[1] + pending_ret_val: ir.AllocaInstr | None = active_finally[2] + if pending_ret_val: + Gen._store(Val, pending_ret_val) + Gen._store(ir.Constant(ir.IntType(32), 1), pending_ret_flag) + Gen.builder.branch(FinallyBB) + else: + Gen.emit_return(Val) + else: + if active_finally: + FinallyBB: ir.Block = active_finally[0] + pending_ret_flag: ir.AllocaInstr = active_finally[1] + pending_ret_val: ir.AllocaInstr | None = active_finally[2] + Gen._store(ir.Constant(ir.IntType(32), 1), pending_ret_flag) + Gen.builder.branch(FinallyBB) + else: + ret_type: ir.Type | None = getattr(Gen.func, 'ftype', None) and Gen.func.ftype.return_type if getattr(Gen, 'func', None) else None + if isinstance(ret_type, ir.IntType): + Gen.builder.ret(ir.Constant(ret_type, 0)) + elif isinstance(ret_type, ir.PointerType): + Gen.builder.ret(ir.Constant(ret_type, None)) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ret_type, 0.0)) + elif isinstance(ret_type, ir.BaseStructType): + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + Gen.builder.ret(zero_val) + elif isinstance(ret_type, ir.ArrayType): + Gen.builder.ret(ir.Constant(ret_type, None)) + else: + Gen.emit_return() \ No newline at end of file diff --git a/lib/core/Handles/HandlesSpecialCall.py b/lib/core/Handles/HandlesSpecialCall.py new file mode 100644 index 0000000..61ea1de --- /dev/null +++ b/lib/core/Handles/HandlesSpecialCall.py @@ -0,0 +1,467 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +import ast +import llvmlite.ir as ir +if TYPE_CHECKING: + from lib.core.translator import Translator + from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +from lib.core.Handles.HandlesBase import BaseHandle, BuiltinTypeMap +from lib.includes.t import CTypeRegistry +from lib.core.SymbolUtils import IsTModule +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode + + +class CSpecialCallHandle(BaseHandle): + def _HandleCIfLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if Node.args: + ArgVal: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + if ArgVal: + if isinstance(ArgVal, ir.Constant) and isinstance(ArgVal.type, ir.IntType): + return ir.Constant(ir.IntType(1), 1 if ArgVal.constant != 0 else 0) + if isinstance(ArgVal.type, ir.IntType): + return Gen.builder.icmp_signed('!=', ArgVal, Gen._ZeroConst(ArgVal.type), name="cif_cond") + if isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.icmp_signed('!=', ArgVal, Gen._ZeroConst(ArgVal.type), name="cif_cond") + return ir.Constant(ir.IntType(1), 0) + def _HandleCAddrLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not Node.args: + return ir.Constant(ir.IntType(8).as_pointer(), None) + Arg = Node.args[0] + if isinstance(Arg, ast.Call): + ClassName = None + if isinstance(Arg.func, ast.Name): + ClassName = Arg.func.id + elif isinstance(Arg.func, ast.Attribute): + ClassName = Arg.func.attr + if ClassName and ClassName in Gen.structs: + StructType = Gen.structs[ClassName] + if isinstance(StructType, ir.PointerType): + StructType = StructType.pointee + ObjPtr = Gen._allocaEntry(StructType, name=f"{ClassName}_obj") + zero_val = Gen._zero_value_for_type(StructType) + if zero_val is not None: + Gen.builder.store(zero_val, ObjPtr) + ConstructorName = f"{ClassName}.__init__" + if ConstructorName in Gen.functions: + func = Gen.functions[ConstructorName] + InitArgs = [ObjPtr] + for carg in Arg.args: + ArgVal = self.HandleExprLlvm(carg) + if ArgVal: + InitArgs.append(ArgVal) + for kw in Arg.keywords: + ArgVal = self.HandleExprLlvm(kw.value) + if ArgVal: + InitArgs.append(ArgVal) + if hasattr(self.Trans.ExprCallHandle, '_append_eh_msg_out_arg'): + self.Trans.ExprCallHandle._append_eh_msg_out_arg(InitArgs, func, Gen) + adjusted = Gen._adjust_args(InitArgs, func) + Gen.builder.call(func, adjusted, name=f"{ClassName}_construct") + return Gen.builder.bitcast(ObjPtr, ir.IntType(8).as_pointer(), name=f"{ClassName}_as_i8ptr") + if isinstance(Arg, ast.Subscript): + SubPtr = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(Arg) + if SubPtr and isinstance(SubPtr.type, ir.PointerType): + return Gen.builder.bitcast(SubPtr, ir.IntType(8).as_pointer(), name="addr_subscript_as_i8ptr") + if isinstance(Arg, ast.Attribute): + AttrPtr = self.Trans.ExprAttrHandle._get_attr_ptr(Arg) + if AttrPtr and isinstance(AttrPtr.type, ir.PointerType): + return Gen.builder.bitcast(AttrPtr, ir.IntType(8).as_pointer(), name="addr_attr_as_i8ptr") + if isinstance(Arg, ast.Name): + VarName = Arg.id + if VarName in Gen.functions: + func = Gen.functions[VarName] + return Gen.builder.bitcast(func, ir.IntType(8).as_pointer(), name="addr_func_as_i8ptr") + if VarName in Gen.variables and Gen.variables[VarName] is not None: + var_ptr = Gen.variables[VarName] + if isinstance(var_ptr.type, ir.PointerType): + # For pointer-to-pointer allocas (var_ptr is T**): + # - If T is a named struct (@t.Object class), the variable semantically + # IS the struct, so c.Addr should return the struct pointer value. + # - If T is a primitive pointer type (e.g., i8*), the variable semantically + # holds a pointer, so c.Addr should return &var (alloca address). + if isinstance(var_ptr.type.pointee, ir.PointerType): + pointee = var_ptr.type.pointee.pointee + if isinstance(pointee, ir.IdentifiedStructType): + # @t.Object class variable — return Loaded struct pointer + ArgVal = self.HandleExprLlvm(Arg) + if ArgVal and isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.bitcast(ArgVal, ir.IntType(8).as_pointer(), name="addr_as_i8ptr") + # Pointer-type variable — return alloca address (&var) + return Gen.builder.bitcast(var_ptr, ir.IntType(8).as_pointer(), name="addr_as_i8ptr") + return Gen.builder.bitcast(var_ptr, ir.IntType(8).as_pointer(), name="addr_as_i8ptr") + ArgVal = self.HandleExprLlvm(Arg) + if ArgVal: + if isinstance(ArgVal.type, ir.IntType): + alloca = Gen._alloca(ArgVal.type, name="addr_tmp") + Gen._store(ArgVal, alloca) + return Gen.builder.bitcast(alloca, ir.IntType(8).as_pointer(), name="addr_as_i8ptr") + elif isinstance(ArgVal.type, ir.PointerType): + return Gen.builder.bitcast(ArgVal, ir.IntType(8).as_pointer(), name="addr_bitcast") + return ir.Constant(ir.IntType(8).as_pointer(), None) + + def _HandleCSetLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if len(Node.args) >= 2: + target_arg: ast.expr = Node.args[0] + value_arg: ast.expr = Node.args[1] + target_ptr: ir.Value | None = None + if isinstance(target_arg, ast.Call) and isinstance(target_arg.func, ast.Attribute): + if isinstance(target_arg.func.value, ast.Name) and target_arg.func.value.id == 'c': + if target_arg.func.attr == 'Deref' and target_arg.args: + deref_arg = target_arg.args[0] + if isinstance(deref_arg, ast.Name) and deref_arg.id in Gen.variables and Gen.variables[deref_arg.id] is not None: + target_ptr = Gen._load(Gen.variables[deref_arg.id], name="deref_load_ptr") + else: + inner_value = self.HandleExprLlvm(deref_arg) + if inner_value: + if isinstance(inner_value.type, ir.PointerType): + if isinstance(inner_value.type.pointee, ir.PointerType): + inner_value = Gen._load(inner_value, name="cast_deref") + target_ptr = inner_value + elif target_arg.func.attr == 'Addr' and target_arg.args: + target_ptr = self.HandleExprLlvm(target_arg) + elif isinstance(target_arg.func.value, ast.Name) and IsTModule(target_arg.func.value.id, self.Trans.SymbolTable): + type_attr = target_arg.func.attr + resolved_cname = self.Trans.TSpecialCallHandle._ResolveTAttrToCName(type_attr) + if target_arg.args and resolved_cname is not None: + inner_val = self.HandleExprLlvm(target_arg.args[0]) + if inner_val and isinstance(inner_val.type, ir.PointerType): + target_llvm_type = Gen._basic_type_to_llvm(resolved_cname) + if target_llvm_type: + target_ptr = Gen.builder.bitcast(inner_val, ir.PointerType(target_llvm_type), name="tcast_ptr") + if not target_ptr: + if isinstance(target_arg, ast.Subscript): + target_ptr = self.Trans.ExprAttrHandle.HandleSubscriptPtrLlvm(target_arg) + else: + target_ptr = self.HandleExprLlvm(target_arg) + Val = self.HandleExprLlvm(value_arg) + if target_ptr and Val: + if isinstance(target_ptr.type, ir.PointerType): + ValToStore = Val + pointee = target_ptr.type.pointee + if isinstance(pointee, ir.PointerType) and isinstance(pointee.pointee, ir.IntType) and pointee.pointee.width == 8: + if isinstance(Val.type, ir.PointerType) and isinstance(Val.type.pointee, ir.IntType) and Val.type.pointee.width == 8: + ValToStore = Val + elif isinstance(Val.type, ir.IntType): + ValToStore = Gen.builder.inttoptr(Val, pointee, name="int_to_ptr_set") + elif isinstance(pointee, ir.IntType): + if isinstance(Val.type, ir.ArrayType) and isinstance(Val.type.element, ir.IntType) and Val.type.element.width == 8: + zero = ir.Constant(ir.IntType(32), 0) + value_ptr = Gen.builder.gep(Val, [zero, zero], name="value_ptr") + ValToStore = Gen._load(value_ptr, name="Load_value") + elif isinstance(Val.type, ir.PointerType) and isinstance(Val.type.pointee, ir.IntType) and Val.type.pointee.width == 8: + ValToStore = Gen._load(Val, name="Load_char_value") + if isinstance(ValToStore.type, ir.IntType) and isinstance(pointee, ir.IntType): + if ValToStore.type.width < pointee.width: + ValToStore = Gen.builder.zext(ValToStore, pointee, name="zext_set") + elif ValToStore.type.width > pointee.width: + ValToStore = Gen.builder.trunc(ValToStore, pointee, name="trunc_set") + elif Val.type != pointee: + if isinstance(pointee, ir.PointerType) and isinstance(Val.type, ir.PointerType): + ValToStore = Gen.builder.bitcast(Val, pointee, name="bitcast_set") + Gen._store(ValToStore, target_ptr) + return ir.Constant(ir.IntType(32), 1) + + def _HandleCLoadLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if len(Node.args) < 2: + return ir.Constant(ir.IntType(32), 0) + # c.Load(a, b) 语义: *a = *b（a 是目标，b 是源） + # wiki 文档: c.Load(ptr, value) -> *ptr = *value; + dst_val: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + src_val: ir.Value | None = self.HandleExprLlvm(Node.args[1]) + if not src_val or not dst_val: + return ir.Constant(ir.IntType(32), 0) + if not isinstance(src_val.type, ir.PointerType) or not isinstance(dst_val.type, ir.PointerType): + return ir.Constant(ir.IntType(32), 0) + pointee = dst_val.type.pointee + # c.Addr 统一返回 i8* 通用指针，丢失原始类型信息。 + # 若 dst 是 c.Addr(var)，从 var 的 alloca 类型推断真实 pointee（如 i64）， + # 否则 c.Load 只会按 i8 加载 1 字节（IEEE 754 double→u64 位重解释 bug）。 + if (isinstance(pointee, ir.IntType) and pointee.width == 8 + and isinstance(Node.args[0], ast.Call) + and isinstance(Node.args[0].func, ast.Attribute) + and Node.args[0].func.attr == 'Addr' + and Node.args[0].args + and isinstance(Node.args[0].args[0], ast.Name)): + var_name = Node.args[0].args[0].id + if var_name in Gen.variables and Gen.variables[var_name] is not None: + var_ptr = Gen.variables[var_name] + if isinstance(var_ptr.type, ir.PointerType): + pointee = var_ptr.type.pointee + # 按 pointee 类型 bitcast src 加载，bitcast dst 存储 + if src_val.type.pointee != pointee: + src_cast = Gen.builder.bitcast(src_val, ir.PointerType(pointee), name="cLoad_src_cast") + else: + src_cast = src_val + Loaded = Gen._load(src_cast, name="cLoad_loaded") + if Loaded is None: + return ir.Constant(ir.IntType(32), 0) + if dst_val.type.pointee != pointee: + dst_cast = Gen.builder.bitcast(dst_val, ir.PointerType(pointee), name="cLoad_dst_cast") + else: + dst_cast = dst_val + Gen._store(Loaded, dst_cast) + return ir.Constant(ir.IntType(32), 1) + + def _HandleCDerefLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not Node.args: + return ir.Constant(ir.IntType(64), 0) + # 检测嵌套 c.Deref(c.Deref(...))：设置标志让内层 Deref 加载为指针 + IsNestedDeref: bool = ( + isinstance(Node.args[0], ast.Call) + and isinstance(Node.args[0].func, ast.Attribute) + and Node.args[0].func.attr == 'Deref' + ) + if IsNestedDeref: + Gen._nested_deref_load_ptr = True + try: + arg_val: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + finally: + if IsNestedDeref: + Gen._nested_deref_load_ptr = False + if arg_val is None: + return ir.Constant(ir.IntType(32), 0) + if isinstance(arg_val.type, ir.PointerType): + pointee: ir.Type = arg_val.type.pointee + # 嵌套 Deref 的内层：加载为指针 (i8*)，不推断目标类型 + # 因为结果会被外层 Deref 再次解引用 + NestedLoadPtr: bool = getattr(Gen, '_nested_deref_load_ptr', False) + if NestedLoadPtr: + PtrType: ir.Type = ir.PointerType(ir.IntType(8)) + casted: ir.Value = Gen.builder.bitcast(arg_val, ir.PointerType(PtrType), name="deref_cast") + Loaded = Gen._load(casted, name="deref") + if Loaded is not None: + return Loaded + # void* (i8*) 需要根据赋值目标类型推断加载类型 + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + target_type: ir.Type | None = self._infer_deref_target_type(Gen, Node) + if target_type is not None and not isinstance(target_type, ir.IntType): + # bitcast i8* -> target_type* then load + casted: ir.Value = Gen.builder.bitcast(arg_val, ir.PointerType(target_type), name="deref_cast") + Loaded = Gen._load(casted, name="deref") + if Loaded is not None: + return Loaded + elif target_type is not None and isinstance(target_type, ir.IntType): + # 目标是整数类型 (如 i32/i64): bitcast 并加载 + if target_type.width != 8: + casted: ir.Value = Gen.builder.bitcast(arg_val, ir.PointerType(target_type), name="deref_cast") + Loaded = Gen._load(casted, name="deref") + if Loaded is not None: + return Loaded + Loaded = Gen._load(arg_val, name="deref") + if Loaded is not None: + return Loaded + return arg_val + + def _infer_deref_target_type(self, Gen: LlvmGeneratorMixin, DerefNode: ast.Call | None = None) -> ir.Type | None: + """从当前赋值上下文推断 c.Deref 的目标加载类型。 + 当用户写 v: int = c.Deref(ptr) 时，从赋值注解推断目标类型。 + 注意：仅当 c.Deref 是赋值的直接 RHS 时才推断；若 c.Deref 嵌套在 + ifexp/compare/binop 中（如 `a = 1 if c.Deref(p) == 65 else 0`）， + 不应推断，否则会把 i8* 错误 bitcast 为 i32* 加载 4 字节。""" + assign_node: ast.AST | None = getattr(self.Trans, '_current_assign_node', None) + if not assign_node: + return None + # 仅当 c.Deref 是赋值的直接 RHS 时才推断类型 + if DerefNode is not None and getattr(assign_node, 'value', None) is not DerefNode: + return None + ann: ast.AST | None = getattr(assign_node, 'annotation', None) + if not ann: + return None + ann_name: str | None = None + if isinstance(ann, ast.Name): + ann_name = ann.id + elif isinstance(ann, ast.Attribute): + ann_name = ann.attr + elif isinstance(ann, ast.BinOp) and isinstance(ann.op, ast.BitOr): + # 处理 int | t.CPtr 形式的联合类型注解 + for side in (ann.left, ann.right): + if isinstance(side, ast.Name): + ann_name = side.id + break + elif isinstance(side, ast.Attribute): + ann_name = side.attr + break + if not ann_name: + return None + # 使用 BuiltinTypeMap 统一查询（同时支持 Python 类名和字符串速记） + entry: tuple[type, int] | None = BuiltinTypeMap.Get(ann_name) + if entry is not None: + ctype_cls: type = entry[0] + llvm_str: str | None = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str: + resolved: ir.Type | None = Gen._type_str_to_llvm(llvm_str) + if resolved: + return resolved + # 检查是否是已注册的结构体类型 + if ann_name in Gen.structs: + return Gen.structs[ann_name] + return None + + def _HandleCPtrToIntLlvm(self, Node: ast.Call) -> ir.Value: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not Node.args: + return ir.Constant(ir.IntType(64), 0) + arg_val: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + if arg_val is None: + return ir.Constant(ir.IntType(64), 0) + if isinstance(arg_val.type, ir.PointerType): + return Gen.builder.ptrtoint(arg_val, ir.IntType(64), name="ptrtoint_result") + if isinstance(arg_val.type, ir.IntType): + if arg_val.type.width < 64: + return Gen.builder.zext(arg_val, ir.IntType(64), name="zext_ptrtoint") + elif arg_val.type.width > 64: + return Gen.builder.trunc(arg_val, ir.IntType(64), name="trunc_ptrtoint") + return arg_val + return ir.Constant(ir.IntType(64), 0) + + +class TSpecialCallHandle(BaseHandle): + + @staticmethod + def _ResolveTAttrToCName(attr: str) -> str | None: + ctype_cls: type | None = CTypeRegistry.GetClassByName(attr) + if ctype_cls is not None: + try: + inst = ctype_cls() + cname = type(inst).__name__ + if cname: + return cname + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"处理函数调用失败: {_e}", "Exception") + return None + + def _HandleTSpecialCallLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if not Node.args: + return None + attr: str = Node.func.attr + + if attr == 'CType': + return self._HandleCTypeLlvm(Node) + + target_type: str | None = self._ResolveTAttrToCName(attr) + if target_type is not None: + IsPtr_cast = False + if len(Node.args) >= 2: + second_arg = Node.args[1] + if isinstance(second_arg, ast.Attribute) and isinstance(second_arg.value, ast.Name) and IsTModule(second_arg.value.id, self.Trans.SymbolTable) and second_arg.attr == 'CPtr': + IsPtr_cast = True + elif isinstance(second_arg, ast.Name) and second_arg.id == 'CPtr': + IsPtr_cast = True + if IsPtr_cast: + return self._HandleTPtrCastLlvm(Node, attr) + if target_type == 'void': + target_type = 'void *' + return self.Trans.ExprCallHandle._HandleTypeCastLlvm(Node.args[0], target_type) + return None + + def _HandleCTypeLlvm(self, Node: ast.Call) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + arg_val: ir.Value | None = self.HandleExprLlvm(Node.args[0]) + if not arg_val: + return None + target_type_str: str = 'unsigned long long' + if len(Node.args) >= 2: + type_parts: list[str] = [] + ptr_count: int = 0 + for i in range(1, len(Node.args)): + type_arg = Node.args[i] + if isinstance(type_arg, ast.Attribute) and isinstance(type_arg.value, ast.Name) and IsTModule(type_arg.value.id, self.Trans.SymbolTable): + type_attr = type_arg.attr + _PREFIX_MAP: dict[str, str] = { + 'CUnsigned': 'unsigned', 'CSigned': 'signed', + 'CConst': 'const', 'CVolatile': 'volatile', + } + if type_attr in _PREFIX_MAP: + type_parts.append(_PREFIX_MAP[type_attr]) + elif type_attr in ('CPtr', 'CArrayPtr'): + ptr_count += 1 + else: + resolved = self._ResolveTAttrToCName(type_attr) + if resolved is not None: + type_parts.append(resolved) + if type_parts: + target_type_str = ' '.join(type_parts) + if ptr_count > 0: + target_type_str += ' ' + '*' * ptr_count + if isinstance(arg_val.type, ir.PointerType): + int_val = Gen.builder.ptrtoint(arg_val, ir.IntType(64), name="ptr_to_int") + return int_val + return self.Trans.ExprCallHandle._HandleTypeCastLlvm(Node.args[0], target_type_str) + + def _HandleTPtrCastLlvm(self, Node: ast.Call, attr: str) -> ir.Value | None: + Gen: LlvmGeneratorMixin = self.Trans.LlvmGen + if attr in ('CChar', 'CUnsignedChar'): + first_arg = Node.args[0] + if isinstance(first_arg, ast.Call) and isinstance(first_arg.func, ast.Attribute) and isinstance(first_arg.func.value, ast.Name) and IsTModule(first_arg.func.value.id, self.Trans.SymbolTable) and first_arg.func.attr == 'CInt' and first_arg.args: + inner_val = self.HandleExprLlvm(first_arg.args[0]) + if isinstance(inner_val.type, ir.PointerType): + return Gen.builder.bitcast(inner_val, ir.PointerType(ir.IntType(8)), name=f"c{attr.lower()}_ptr_from_cint") + expr_val = self.HandleExprLlvm(Node.args[0]) + if isinstance(expr_val.type, ir.IntType) and expr_val.type.width == 8: + var = Gen._alloca(ir.ArrayType(ir.IntType(8), 2), name=f"c{attr.lower()}_ptr") + zero = ir.Constant(ir.IntType(32), 0) + char_ptr = Gen.builder.gep(var, [zero, zero], name="char_ptr") + Gen._store(expr_val, char_ptr) + null_char = ir.Constant(ir.IntType(8), 0) + null_ptr = Gen.builder.gep(var, [zero, ir.Constant(ir.IntType(32), 1)], name="null_ptr") + Gen._store(null_char, null_ptr) + return char_ptr + elif isinstance(expr_val.type, ir.PointerType): + if isinstance(expr_val.type.pointee, ir.PointerType): + Loaded_ptr = Gen._load(expr_val, name="load_ptr") + if isinstance(Loaded_ptr.type, ir.PointerType) and isinstance(Loaded_ptr.type.pointee, ir.IntType) and Loaded_ptr.type.pointee.width == 8: + return Loaded_ptr + if isinstance(expr_val.type.pointee, ir.IntType) and expr_val.type.pointee.width == 8: + return expr_val + return Gen.builder.bitcast(expr_val, ir.PointerType(ir.IntType(8)), name="cast_to_char_ptr") + elif isinstance(expr_val.type, ir.IntType): + if expr_val.type.width == 32: + expr_val = Gen.builder.zext(expr_val, ir.IntType(64), name="zext_to_64") + return Gen.builder.inttoptr(expr_val, ir.PointerType(ir.IntType(8)), name=f"inttoptr_c{attr.lower()}") + return Gen.emit_constant(0, 'int') + elif attr == 'CVoid': + return self.Trans.ExprCallHandle._HandleTypeCastLlvm(Node.args[0], 'void *') + elif attr == 'CInt': + expr_val = self.HandleExprLlvm(Node.args[0]) + if expr_val: + if isinstance(expr_val.type, ir.IntType): + if expr_val.type.width < 64: + expr_val = Gen.builder.zext(expr_val, ir.IntType(64), name="zext_to_64") + return Gen.builder.inttoptr(expr_val, ir.PointerType(ir.IntType(32)), name="cint_ptr") + return Gen.builder.bitcast(expr_val, ir.PointerType(ir.IntType(32)), name="cint_ptr") + else: + ctype_cls = CTypeRegistry.GetClassByName(attr) + if ctype_cls is not None: + llvm_str = CTypeRegistry.CTypeToLLVM(ctype_cls) + target_type = Gen._type_str_to_llvm(llvm_str) + if target_type: + # 基础类型 (int/float) → 加一层指针 → T* + if isinstance(target_type, (ir.IntType, ir.FloatType, ir.DoubleType)): + target_ptr_type = ir.PointerType(target_type) + expr_val = self.HandleExprLlvm(Node.args[0]) + if expr_val: + if isinstance(expr_val.type, ir.IntType): + if expr_val.type.width < 64: + expr_val = Gen.builder.zext(expr_val, ir.IntType(64), name="zext_to_64") + return Gen.builder.inttoptr(expr_val, target_ptr_type, name=f"{attr.lower()}_ptr") + return Gen.builder.bitcast(expr_val, target_ptr_type, name=f"{attr.lower()}_ptr") + # 指针类型 (如 CPtr → i8*) → 再加一层指针 → i8** + elif isinstance(target_type, ir.PointerType): + target_ptr_type = ir.PointerType(target_type) + expr_val = self.HandleExprLlvm(Node.args[0]) + if expr_val: + if isinstance(expr_val.type, ir.IntType): + if expr_val.type.width < 64: + expr_val = Gen.builder.zext(expr_val, ir.IntType(64), name="zext_to_64") + return Gen.builder.inttoptr(expr_val, target_ptr_type, name=f"{attr.lower()}_ptr") + return Gen.builder.bitcast(expr_val, target_ptr_type, name=f"{attr.lower()}_ptr") + return None diff --git a/lib/core/Handles/HandlesTry.py b/lib/core/Handles/HandlesTry.py new file mode 100644 index 0000000..1fd04e7 --- /dev/null +++ b/lib/core/Handles/HandlesTry.py @@ -0,0 +1,214 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle, EXCEPTION_CODE_MAP +import ast +import llvmlite.ir as ir + + +class TryHandle(BaseHandle): + def _get_exception_type_code(self, type_node: ast.expr) -> int: + if isinstance(type_node, ast.Name): + name: str = type_node.id + if name in EXCEPTION_CODE_MAP: + return EXCEPTION_CODE_MAP[name] + if name in self.Trans.exception_registry: + return self.Trans.exception_registry[name] + return 1 + if isinstance(type_node, ast.Attribute): + name: str = type_node.attr + if name in EXCEPTION_CODE_MAP: + return EXCEPTION_CODE_MAP[name] + if name in self.Trans.exception_registry: + return self.Trans.exception_registry[name] + return 1 + if isinstance(type_node, ast.Tuple): + for elt in type_node.elts: + code: int = self._get_exception_type_code(elt) + if code != 1: + return code + return 1 + return 1 + + def _get_exception_name(self, type_node: ast.expr) -> str | None: + if isinstance(type_node, ast.Name): + return type_node.id + if isinstance(type_node, ast.Attribute): + return type_node.attr + return None + + def _get_all_match_codes(self, type_node: ast.expr) -> list[int]: + name: str | None = self._get_exception_name(type_node) + if name is None: + return [self._get_exception_type_code(type_node)] + if name not in self.Trans.exception_registry: + return [self._get_exception_type_code(type_node)] + codes: list[int] = [self.Trans.exception_registry[name]] + self._collect_descendant_codes(name, codes) + return codes + + def _collect_descendant_codes(self, parent_name: str, codes: list[int]) -> None: + for child_name, parent in self.Trans.exception_parents.items(): + if parent == parent_name: + child_code: int | None = self.Trans.exception_registry.get(child_name) + if child_code is not None and child_code not in codes: + codes.append(child_code) + self._collect_descendant_codes(child_name, codes) + + def _HandleTryLlvm(self, Node: ast.Try) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if not Gen.func: + return + exception_code: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name="eh_code") + eh_message: ir.AllocaInstr = Gen._allocaEntry(ir.PointerType(ir.IntType(8)), name="eh_message") + Gen._store(ir.Constant(ir.IntType(32), 0), exception_code) + Gen._store(ir.Constant(ir.PointerType(ir.IntType(8)), None), eh_message) + HasFinally: bool = bool(Node.finalbody) + if HasFinally: + Gen.eh_finally_stack.append((None, None, None)) + else: + Gen.eh_finally_stack.append(None) + DispatchBB: ir.Block = Gen.func.append_basic_block(name="try.dispatch") + AfterBB: ir.Block = Gen.func.append_basic_block(name="try.after") + except_blocks: list[tuple[ir.Block, list[int], ast.ExceptHandler]] = [] + for i, handler in enumerate(Node.handlers): + ExcTypeCodes: list[int] = [1] + if handler.type: + ExcTypeCodes = self._get_all_match_codes(handler.type) + ExceptBB: ir.Block = Gen.func.append_basic_block(name=f"try.except_{i}") + Gen.eh_except_block_stack.append((DispatchBB, None, exception_code, eh_message)) + except_blocks.append((ExceptBB, ExcTypeCodes, handler)) + TryBB: ir.Block = Gen.func.append_basic_block(name="try.body") + ElseBB: ir.Block | None = None + if Node.orelse: + ElseBB = Gen.func.append_basic_block(name="try.else") + Gen.builder.branch(TryBB) + Gen.builder.position_at_start(TryBB) + self.HandleBodyLlvm(Node.body) + if not Gen.builder.block.is_terminated: + Gen._store(ir.Constant(ir.IntType(32), 0), exception_code) + if ElseBB: + Gen.builder.branch(ElseBB) + else: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(DispatchBB) + eh_code_val: ir.Value = Gen._load(exception_code, name="Load_eh_code") + next_check_bb: ir.Block | None = None + handler_var_map: dict[int, tuple[str, ir.AllocaInstr, ir.Value | None]] = {} + UnhandledBB: ir.Block = Gen.func.append_basic_block(name="try.unhandled") + for i, (ExceptBB, ExcTypeCodes, handler) in enumerate(except_blocks): + MatchBB: ir.Block = Gen.func.append_basic_block(name=f"try.match_{i}") + if i > 0 and next_check_bb is not None: + Gen.builder.position_at_start(next_check_bb) + eh_code_val = Gen._load(exception_code, name=f"Load_eh_code_{i}") + if handler.type is None: + Gen.builder.branch(MatchBB) + else: + match_cond: ir.Value | None = None + for code in ExcTypeCodes: + is_code_match: ir.Value = Gen.builder.icmp_signed('==', eh_code_val, ir.Constant(ir.IntType(32), code), name=f"eh_match_{i}_c{code}") + if match_cond is None: + match_cond = is_code_match + else: + match_cond = Gen.builder.or_(match_cond, is_code_match, name=f"eh_match_any_{i}") + if i < len(except_blocks) - 1: + next_check_bb = Gen.func.append_basic_block(name=f"try.check_{i}") + else: + next_check_bb = UnhandledBB + Gen.builder.cbranch(match_cond, MatchBB, next_check_bb) + Gen.builder.position_at_start(MatchBB) + if handler.name: + var_alloca: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(8).as_pointer(), name=handler.name) + eh_msg_val: ir.Value = Gen._load(eh_message, name=f"Load_eh_msg_{handler.name}") + Gen._store(eh_msg_val, var_alloca) + handler_var_map[i] = (handler.name, var_alloca, Gen.variables.get(handler.name)) + Gen.builder.branch(ExceptBB) + Gen.builder.position_at_start(UnhandledBB) + eh_msg_out_arg: ir.Argument | None = None + eh_code_out_arg: ir.Argument | None = None + if Gen.func and len(Gen.func.args) > 0: + for arg in Gen.func.args: + if arg.name == '__eh_msg_out__': + eh_msg_out_arg = arg + elif arg.name == '__eh_code_out__': + eh_code_out_arg = arg + if eh_msg_out_arg is not None and eh_code_out_arg is not None: + eh_msg_val: ir.Value = Gen._load(eh_message, name="Load_eh_msg_propagate") + eh_code_val_prop: ir.Value = Gen._load(exception_code, name="Load_eh_code_propagate") + Gen._store(eh_msg_val, eh_msg_out_arg) + Gen._store(eh_code_val_prop, eh_code_out_arg) + ret_type: ir.Type | None = getattr(Gen.func, 'ftype', None) and Gen.func.ftype.return_type if getattr(Gen, 'func', None) else None + if isinstance(ret_type, ir.IdentifiedStructType): + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ + ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) + else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) + else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + Gen.builder.ret(zero_val) + elif isinstance(ret_type, ir.PointerType): + Gen.builder.ret(ir.Constant(ret_type, None)) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + Gen.builder.ret(ir.Constant(ret_type, 0.0)) + elif isinstance(ret_type, ir.IntType): + Gen.builder.ret(ir.Constant(ret_type, 1)) + elif isinstance(ret_type, ir.VoidType): + Gen.builder.ret_void() + else: + Gen.builder.ret(ir.Constant(ir.IntType(32), 1)) + else: + Gen._store(ir.Constant(ir.IntType(32), 0), exception_code) + Gen.builder.branch(AfterBB) + for idx, (ExceptBB, ExcTypeCodes, handler) in enumerate(except_blocks): + Gen.builder.position_at_start(ExceptBB) + old_var: ir.Value | None = None + if idx in handler_var_map: + hname: str = handler_var_map[idx][0] + var_alloca: ir.AllocaInstr = handler_var_map[idx][1] + old_var = handler_var_map[idx][2] + Gen.variables[hname] = var_alloca + self.HandleBodyLlvm(handler.body) + if idx in handler_var_map: + hname = handler_var_map[idx][0] + old_var = handler_var_map[idx][2] + if old_var is not None: + Gen.variables[hname] = old_var + elif hname in Gen.variables: + del Gen.variables[hname] + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.eh_except_block_stack.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) + if HasFinally: + Gen.eh_finally_stack.pop() + if Node.finalbody: + FinallyBB: ir.Block = Gen.func.append_basic_block(name="try.finally") + pending_ret_flag: ir.AllocaInstr = Gen._allocaEntry(ir.IntType(32), name="pending_ret_flag") + pending_ret_val: ir.AllocaInstr | None = Gen._allocaEntry(ir.IntType(32), name="pending_ret_val") if Gen.func.type.pointee.return_type == ir.IntType(32) else None + Gen.eh_finally_stack.append((FinallyBB, pending_ret_flag, pending_ret_val)) + Gen.builder.branch(FinallyBB) + Gen.builder.position_at_start(FinallyBB) + Gen._store(ir.Constant(ir.IntType(32), 0), pending_ret_flag) + self.HandleBodyLlvm(Node.finalbody) + if Gen.eh_finally_stack: + _, pending_ret_flag, pending_ret_val = Gen.eh_finally_stack[-1] + Gen.eh_finally_stack.pop() + ret_flag_val: ir.Value = Gen._load(pending_ret_flag, name="Load_ret_flag") + ret_flag_is_set: ir.Value = Gen.builder.icmp_signed('!=', ret_flag_val, ir.Constant(ir.IntType(32), 0), name="check_ret_flag") + RetBB: ir.Block = Gen.func.append_basic_block(name="finally.ret") + ContinueBB: ir.Block = Gen.func.append_basic_block(name="try.continue") + Gen.builder.cbranch(ret_flag_is_set, RetBB, ContinueBB) + Gen.builder.position_at_start(RetBB) + if pending_ret_val: + ret_val: ir.Value = Gen._load(pending_ret_val, name="Load_ret_val") + Gen.builder.ret(ret_val) + else: + Gen.builder.ret_void() + Gen.builder.position_at_start(ContinueBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesTypeMerge.py b/lib/core/Handles/HandlesTypeMerge.py new file mode 100644 index 0000000..a3c0d93 --- /dev/null +++ b/lib/core/Handles/HandlesTypeMerge.py @@ -0,0 +1,975 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from typing import List +from lib.core.Handles.HandlesBase import BaseHandle, CTypeHelper, CTypeInfo +from lib.core.SymbolUtils import IsTModule, IsListAnnotation +from lib.includes import t +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） +import ast + + +class HandlesTypeMerge(BaseHandle): + """处理类型合并逻辑 - 所有类型分析的核心模块""" + + def _MakeVoidCTypeInfo(self) -> CTypeInfo: + Info = CTypeInfo() + Info.BaseType = t.CVoid + return Info + + @staticmethod + def _MakeCTypeInfoFromName(TypeName: str) -> CTypeInfo: + """直接从类型名构造 CTypeInfo，不经过 FromStr/字符串解析""" + return CTypeInfo.FromTypeName(TypeName) + + def GetCTypeInfo(self, Node: ast.AST) -> CTypeInfo: + """获取 CTypeInfo 对象""" + Result = self._GetCTypeInfoInternal(Node) + if isinstance(Result, CTypeInfo): + return Result + elif isinstance(Result, str): + return self._MakeCTypeInfoFromName(Result) + else: + return self._MakeVoidCTypeInfo() + + def _GetCTypeInfoInternal(self, Node: ast.AST) -> CTypeInfo | str | None: + """鍐呴儴鏂规硶锛岃繑鍥?CTypeInfo""" + LineNum = getattr(Node, 'lineno', None) + + self.Trans.DebugPrint(f"[GetCTypeInfo] Node type: {type(Node).__name__}, dump: {ast.dump(Node)[:60]}...") + + if isinstance(Node, ast.Constant): + TypeName = Node.value + return self._MakeCTypeInfoFromName(TypeName) + + if isinstance(Node, ast.Name): + TypeName = Node.id + t_c_imported = getattr(self.Trans, '_t_c_imported_names', {}) + if TypeName in t_c_imported: + src_module, src_name = t_c_imported[TypeName] + virtual_attr = ast.Attribute( + value=ast.Name(id=src_module, ctx=ast.Load()), + attr=src_name, + ctx=ast.Load() + ) + ast.copy_location(virtual_attr, Node) + return self.GetCTypeInfo(virtual_attr) + Entry = self.Trans.SymbolTable.lookup(TypeName) + if Entry and Entry.IsTypedef: + if isinstance(Entry, CTypeInfo) and Entry.BaseType: + if not isinstance(Entry.BaseType, (t._CTypedef,)) or Entry.PtrCount > 0: + Result = Entry.Copy() + Result.IsTypedef = True + Result.Name = TypeName + return Result + if Entry.OriginalType: + if isinstance(Entry.OriginalType, CTypeInfo) and Entry.OriginalType.IsFuncPtr: + Result = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = Entry.OriginalType.FuncPtrReturn or CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = list(Entry.OriginalType.FuncPtrParams) if Entry.OriginalType.FuncPtrParams else [] + Result.IsTypedef = True + Result.Name = TypeName + return Result + elif isinstance(Entry.OriginalType, CTypeInfo): + Resolved = Entry.OriginalType.Copy() + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + elif isinstance(Entry.OriginalType, str): + # 字符串 OriginalType（如 'void *'）需解析为 CTypeInfo， + # 否则返回的 Entry BaseType=None 会被下游误判为 i32 + Resolved = CTypeInfo.FromTypeName(Entry.OriginalType) + if Resolved and Resolved.BaseType: + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + return Entry + # REnum 优先于 IsEnum 处理（REnum 同时有 IsEnum=True 和 IsStruct=True） + # 必须设置 Name 和 PtrCount=1，否则 BinOp `LLVMType | t.CPtr` 合并后丢失 REnum 语义 + if isinstance(Entry, CTypeInfo) and Entry.IsRenum: + Result = Entry.Copy() + Result.Name = TypeName + Result.PtrCount = 1 # REnum 变量默认为指针（和 struct 一致） + return Result + # 类名与枚举成员名冲突时（如 Constant 既是 ASTKind.Constant 枚举成员又是结构体类名）， + # 优先检查 Gen.structs：类型注解应解析为结构体而非枚举值 + _Gen = getattr(self.Trans, 'LlvmGen', None) + if _Gen and TypeName in _Gen.structs: + Result = CTypeInfo() + Result.BaseType = t.CStruct(name=TypeName) + Result.IsStruct = True + Result.Name = TypeName + Result.PtrCount = 1 + return Result + if isinstance(Entry, CTypeInfo) and Entry.IsEnum: + return Entry.Copy() + if isinstance(Entry, dict) and Entry.get('IsEnum'): + Result = CTypeInfo() + Result.BaseType = t.CEnum(TypeName) + Result.IsEnum = True + return Result + # 检测结构体类型 (CTypeInfo with IsStruct=True, 来自 _InsertClassSymbol) + if isinstance(Entry, CTypeInfo) and Entry.IsStruct: + Result = Entry.Copy() + if Result.BaseType is None: + Result.BaseType = t.CStruct(name=TypeName) + Result.Name = TypeName + Result.PtrCount = 1 # struct 变量默认为指针 + return Result + return self._MakeCTypeInfoFromName(TypeName) + + elif isinstance(Node, ast.Attribute): + ModuleParts = [] + Current = Node + while isinstance(Current, ast.Attribute): + ModuleParts.insert(0, Current.attr) + Current = Current.value + + if isinstance(Current, ast.Name): + ModuleParts.insert(0, Current.id) + + if len(ModuleParts) >= 2: + TypeName = ModuleParts[-1] + ModulePath = '.'.join(ModuleParts[:-1]) + elif len(ModuleParts) == 1: + TypeName = ModuleParts[0] + ModulePath = None + else: + return self._MakeVoidCTypeInfo() + + # 解析 import 别名: 如 import fat32_types as types -> types -> fat32_types + if ModulePath: + import_aliases = self.Trans.SymbolTable.import_aliases + if ModulePath in import_aliases: + ModulePath = import_aliases[ModulePath] + else: + # 也尝试在符号表中查找模块别名 + entry = self.Trans.SymbolTable.lookup(ModulePath) + if isinstance(entry, CTypeInfo) and entry.IsModuleAlias: + resolved = entry.ResolvedModule + if resolved: + ModulePath = resolved + + self.Trans.DebugPrint(f"[GetCTypeInfo-Attribute] 查询: '{TypeName}' ModulePath: {ModulePath}") + + if ModulePath == 't' and TypeName == 'State': + Info = CTypeInfo() + Info.BaseType = t.CVoid + Info.IsState = True + return Info + + if ModulePath == 't' or (ModulePath and ModulePath.startswith('t.')): + if self._IsTModuleType(TypeName): + TypeClass = CTypeHelper.GetTModuleCType(TypeName) or getattr(t, TypeName, None) + if TypeClass == t.CArrayPtr: + Info = CTypeInfo() + Info.IsArrayPtr = True + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CPtr: + Info = CTypeInfo() + Info.PtrCount = 1 + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CConst: + Info = CTypeInfo() + Info.DataConst = True + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CVolatile: + Info = CTypeInfo() + Info.DataVolatile = True + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CInline: + Info = CTypeInfo() + Info.Storage = t.CInline() + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CStatic: + Info = CTypeInfo() + Info.Storage = t.CStatic() + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CExtern: + Info = CTypeInfo() + Info.Storage = t.CExtern() + Info.BaseType = t.CVoid + return Info + elif TypeClass == t.CExport: + Info = CTypeInfo() + Info.Storage = t.CExport() + Info.BaseType = t.CVoid + return Info + elif TypeClass in (t.BigEndian, t.LittleEndian): + Info = CTypeInfo() + Info.BaseType = t.CInt() + Info.ByteOrder = 'big' if TypeClass == t.BigEndian else 'little' + return Info + elif TypeClass: + Info = CTypeInfo() + Info.BaseType = TypeClass + return Info + elif CTypeHelper.GetCName(TypeName): + return self._MakeCTypeInfoFromName(CTypeHelper.GetCName(TypeName)) + + FullName = f"{ModulePath}.{TypeName}" if ModulePath else TypeName + + # 优先检查 Gen.structs：类名与枚举成员名冲突时（如 Import 既是 ASTKind.Import + # 枚举成员又是 ast.Import 结构体类名），类型注解应解析为结构体而非枚举值 + # （与 ast.Name 路径 line 100-107 的逻辑一致） + _Gen = getattr(self.Trans, 'LlvmGen', None) + if _Gen and TypeName in _Gen.structs: + Result = CTypeInfo() + Result.BaseType = t.CStruct(name=TypeName) + Result.IsStruct = True + Result.Name = TypeName + Result.PtrCount = 1 + return Result + + Entry = self.Trans.SymbolTable.lookup(TypeName) + if Entry and Entry.IsTypedef: + if isinstance(Entry, CTypeInfo) and Entry.BaseType: + if not isinstance(Entry.BaseType, (t._CTypedef,)) or Entry.PtrCount > 0: + Result = Entry.Copy() + Result.IsTypedef = True + Result.Name = TypeName + return Result + if Entry.OriginalType: + if isinstance(Entry.OriginalType, CTypeInfo) and Entry.OriginalType.IsFuncPtr: + Result = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = Entry.OriginalType.FuncPtrReturn or CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = list(Entry.OriginalType.FuncPtrParams) if Entry.OriginalType.FuncPtrParams else [] + Result.IsTypedef = True + Result.Name = TypeName + return Result + elif isinstance(Entry.OriginalType, CTypeInfo): + Resolved = Entry.OriginalType.Copy() + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + elif isinstance(Entry.OriginalType, str): + # 字符串 OriginalType（如 'void *'）需解析为 CTypeInfo， + # 否则返回的 Result BaseType=_CTypedef 会被下游误判为 i32 + Resolved = CTypeInfo.FromTypeName(Entry.OriginalType) + if Resolved and Resolved.BaseType: + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + Result = CTypeInfo() + Result.BaseType = t._CTypedef(TypeName) + Result.IsTypedef = True + Result.Name = TypeName + return Result + if Entry: + # If BaseType is None but this is a class/struct, set BaseType to CStruct + Result = Entry.Copy() + if Result.BaseType is None: + Result.BaseType = t.CStruct(name=TypeName) + Result.IsStruct = True + # struct 变量默认为指针（与 ast.Name 路径一致，见 line 121） + # 同时确保 Name 字段被设置，供 ToString() 使用 + if Result.IsStruct: + if Result.PtrCount == 0: + Result.PtrCount = 1 + if not Result.Name: + Result.Name = TypeName + return Result + return CTypeInfo() + + elif isinstance(Node, ast.Call): + if isinstance(Node.func, ast.Name) and Node.func.id == 'callable': + ReturnTypeInfo = self._GetCTypeInfoInternal(Node.args[0]) if Node.args else None + if isinstance(ReturnTypeInfo, CTypeInfo): + ReturnType = ReturnTypeInfo + else: + ReturnType = CTypeInfo() + ReturnType.BaseType = t.CVoid() + + ParamTypes = [] + ParamNames = [] + for kw in Node.keywords: + ParamTypeInfo = self._GetCTypeInfoInternal(kw.value) + if not isinstance(ParamTypeInfo, CTypeInfo): + ParamTypeInfo = self._MakeCTypeInfoFromName('int') + ParamTypes.append(ParamTypeInfo) + ParamNames.append(kw.arg) + + for arg in Node.args[1:]: + if isinstance(arg, ast.Tuple): + for elt in arg.elts: + ParamTypeInfo = self._GetCTypeInfoInternal(elt) + if not isinstance(ParamTypeInfo, CTypeInfo): + ParamTypeInfo = self._MakeCTypeInfoFromName('int') + ParamTypes.append(ParamTypeInfo) + ParamNames.append('') + else: + ParamTypeInfo = self._GetCTypeInfoInternal(arg) + if not isinstance(ParamTypeInfo, CTypeInfo): + ParamTypeInfo = self._MakeCTypeInfoFromName('int') + ParamTypes.append(ParamTypeInfo) + ParamNames.append('') + + if not ParamTypes: + ParamTypes.append(self._MakeCTypeInfoFromName('void')) + ParamNames.append('') + + Info = CTypeInfo() + Info.IsFuncPtr = True + Info.FuncPtrReturn = ReturnType + Info.FuncPtrParams = list(zip(ParamNames, ParamTypes)) + return Info + + if isinstance(Node.func, ast.Attribute) and isinstance(Node.func.value, ast.Name): + ModuleName = Node.func.value.id + TypeName = Node.func.attr + + if ModuleName == 't' and CTypeHelper.GetTModuleCType(TypeName) == t._CTypedef and Node.args: + TypeArg = Node.args[0] + if isinstance(TypeArg, ast.Name): + return TypeArg.id + elif isinstance(TypeArg, ast.Attribute): + AttrModule = TypeArg.value.id if isinstance(TypeArg.value, ast.Name) else None + AttrName = TypeArg.attr + if AttrModule == 't' and getattr(t, AttrName, None): + TypeObj = getattr(t, AttrName) + if isinstance(TypeObj, type) and issubclass(TypeObj, t.CType): + return TypeObj.__name__ + return AttrName + elif isinstance(TypeArg, ast.Constant): + return str(TypeArg.value) + + if ModuleName in self.Trans._UserTypeModules: + module = self.Trans._UserTypeModules[ModuleName] + if getattr(module, TypeName, None): + TypeObj = getattr(module, TypeName) + if isinstance(TypeObj, type): + if issubclass(TypeObj, t.CType): + return TypeObj.__name__ + + if ModuleName == 't': + TypeObj = CTypeHelper.GetTModuleCType(TypeName) + if TypeObj is None: + TypeObj = getattr(t, TypeName, None) + if isinstance(TypeObj, type) and issubclass(TypeObj, t.CType): + Result = CTypeInfo() + Result.BaseType = TypeObj + return Result + + if ModuleName == 't' and TypeName == 'State': + Result = CTypeInfo() + Result.IsState = True + return Result + + if ModuleName == 't' and TypeName == 'Bit': + if Node.args and isinstance(Node.args[0], ast.Constant): + BitWidth = Node.args[0].value + Result = CTypeInfo() + Result.IsBitField = True + Result.BitWidth = BitWidth + Result.BaseType = t.CInt + return Result + + Result = CTypeInfo() + Result.BaseType = t.CVoid + return Result + + elif isinstance(Node, ast.Subscript): + if (isinstance(Node.value, ast.Attribute) + and isinstance(Node.value.value, ast.Name) + and IsTModule(Node.value.value.id, self.Trans.SymbolTable) + and Node.value.attr == 'Bit'): + if isinstance(Node.slice, ast.Constant) and isinstance(Node.slice.value, int): + Result = CTypeInfo() + Result.IsBitField = True + Result.BitWidth = Node.slice.value + Result.BaseType = t.CInt + return Result + return self._HandleSubscript(Node) + + elif isinstance(Node, ast.BinOp): + return self.MergeTypes(Node) + + elif isinstance(Node, ast.List): + if Node.elts: + BaseTypes = [] + for elt in Node.elts: + EltType = self._GetCTypeInfoInternal(elt) + if isinstance(EltType, CTypeInfo) and EltType.BaseType: + BaseTypes.append(EltType.BaseType) + if BaseTypes: + Result = CTypeInfo() + Result.BaseType = tuple(BaseTypes) + return Result + return self._MakeVoidCTypeInfo() + + return self._MakeVoidCTypeInfo() + + def _HandleSubscript(self, Node: ast.Subscript) -> CTypeInfo: + """澶勭悊 Subscript 鑺傜偣锛屼娇鐢?CTypeInfo 鏍戝舰缁撴瀯鍒嗘瀽 AST + + 例如? + t.CVoid[5] -> void *[5] + t.CPtr[t.CPtr[t.CPtr]][5][6] -> void ***[5][6] + t.CArrayPtr[t.CArrayPtr] -> void *(*)[] + t.CPtr[t.CArrayPtr[t.CArrayPtr]] -> void *(*(*))[] + """ + def CollectAll(node: ast.AST, ParentIsArrayPtr: bool = False) -> CTypeInfo: + """鏀堕泦绫诲瀷淇℃伅锛岃繑鍥?CTypeInfo""" + if isinstance(node, ast.Subscript): + value_node = node.value + slice_node = node.slice + + if isinstance(value_node, ast.Attribute): + VtInfo = self.GetCTypeInfo(value_node) + + if isinstance(value_node, ast.Attribute) and value_node.attr == 'Callable' and isinstance(value_node.value, ast.Name) and IsTModule(value_node.value.id, self.Trans.SymbolTable): + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + params_list = slice_node.elts[0] + return_node = slice_node.elts[1] + ParamTypes = [] + ParamNames = [] + if isinstance(params_list, ast.List): + for elt in params_list.elts: + ParamTypeInfo = self._GetCTypeInfoInternal(elt) + if isinstance(ParamTypeInfo, CTypeInfo): + ParamTypes.append(ParamTypeInfo) + ParamNames.append('') + else: + ParamTypes.append(self._MakeCTypeInfoFromName('int')) + ParamNames.append('') + if not ParamTypes: + ParamTypes.append(self._MakeCTypeInfoFromName('void')) + ParamNames.append('') + ReturnTypeInfo = self._GetCTypeInfoInternal(return_node) + if isinstance(ReturnTypeInfo, CTypeInfo): + ReturnType = ReturnTypeInfo + else: + ReturnType = CTypeInfo() + ReturnType.BaseType = t.CVoid() + Info = CTypeInfo() + Info.IsFuncPtr = True + Info.FuncPtrReturn = ReturnType + Info.FuncPtrParams = list(zip(ParamNames, ParamTypes)) + return Info + + if VtInfo and VtInfo.IsArrayPtr: + ResultInfo = CTypeInfo() + ResultInfo.PtrCount = VtInfo.PtrCount + ResultInfo.IsArrayPtr = True + if isinstance(slice_node, ast.Attribute): + SliceVtInfo = self.GetCTypeInfo(slice_node) + if SliceVtInfo and SliceVtInfo.IsArrayPtr: + ResultInfo.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr.IsArrayPtr = True + ResultInfo.ArrayPtr.PtrCount = SliceVtInfo.PtrCount + if SliceVtInfo.ArrayPtr: + ResultInfo.ArrayPtr.ArrayPtr = SliceVtInfo.ArrayPtr + elif SliceVtInfo and SliceVtInfo.PtrCount > 0: + ResultInfo.ArrayPtr = SliceVtInfo + else: + ResultInfo.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr.IsArrayPtr = True + elif isinstance(slice_node, ast.Subscript): + ResultInfo.ArrayPtr = CollectAll(slice_node) + elif isinstance(slice_node, ast.Name): + ResultInfo.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr.BaseType = CTypeInfo.CreateFromTypeName(slice_node.id).BaseType + elif isinstance(slice_node, ast.BinOp): + ResultInfo.ArrayPtr = CTypeInfo() + try: + ResultInfo.ArrayPtr.ArrayDims.append(ast.unparse(slice_node)) + except Exception: # 回退：unparse 失败时用 'N' + ResultInfo.ArrayPtr.ArrayDims.append('N') + elif isinstance(slice_node, ast.Constant): + ResultInfo.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr.ArrayDims.append(str(slice_node.value)) + else: + ResultInfo.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr.IsArrayPtr = True + elif VtInfo and VtInfo.PtrCount == 1 and not VtInfo.ArrayPtr: + ResultInfo = CTypeInfo() + ResultInfo.BaseType = t.CVoid + ResultInfo.PtrCount = 1 + if isinstance(slice_node, ast.Subscript): + inner_info = CollectAll(slice_node) + if inner_info and inner_info.PtrCount > 0: + ResultInfo.PtrCount += inner_info.PtrCount + if inner_info.BaseType and not isinstance(inner_info.BaseType, t.CVoid): + ResultInfo.BaseType = inner_info.BaseType + elif inner_info and inner_info.ArrayPtr: + ResultInfo.ArrayPtr = inner_info + elif inner_info and inner_info.BaseType and not isinstance(inner_info.BaseType, t.CVoid): + ResultInfo.BaseType = inner_info.BaseType + elif isinstance(slice_node, ast.Attribute): + SliceVtInfo = self.GetCTypeInfo(slice_node) + if SliceVtInfo and SliceVtInfo.PtrCount == 1 and not SliceVtInfo.ArrayPtr: + ResultInfo.PtrCount = 2 + if SliceVtInfo.BaseType and not isinstance(SliceVtInfo.BaseType, t.CVoid): + ResultInfo.BaseType = SliceVtInfo.BaseType + elif SliceVtInfo and SliceVtInfo.IsArrayPtr: + inner = CTypeInfo() + inner.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr = inner + elif SliceVtInfo and SliceVtInfo.ArrayPtr: + ResultInfo.ArrayPtr = SliceVtInfo + elif SliceVtInfo and SliceVtInfo.BaseType and not isinstance(SliceVtInfo.BaseType, t.CVoid): + ResultInfo.BaseType = SliceVtInfo.BaseType + elif isinstance(slice_node, ast.Name): + SliceType = getattr(t, slice_node.id, None) + if SliceType == t.CPtr: + ResultInfo.PtrCount += 1 + elif SliceType == t.CArrayPtr: + inner = CTypeInfo() + inner.ArrayPtr = CTypeInfo() + ResultInfo.ArrayPtr = inner + else: + ResultInfo.BaseType = CTypeInfo.CreateFromTypeName(slice_node.id).BaseType + elif isinstance(slice_node, ast.BinOp): + merged = self.MergeTypes(slice_node) + if merged: + ResultInfo.PtrCount += merged.PtrCount + if merged.BaseType and not isinstance(merged.BaseType, t.CVoid): + ResultInfo.BaseType = merged.BaseType + if merged.ArrayPtr: + ResultInfo.ArrayPtr = merged.ArrayPtr + elif isinstance(slice_node, ast.Constant): + ResultInfo.ArrayDims.append(str(slice_node.value)) + elif VtInfo: + ResultInfo = CTypeInfo() + ResultInfo.BaseType = VtInfo.BaseType + if isinstance(slice_node, ast.Constant): + ResultInfo.ArrayDims.append(str(slice_node.value)) + elif isinstance(slice_node, ast.Name): + ResultInfo.ArrayDims.append(slice_node.id) + elif isinstance(slice_node, ast.BinOp): + try: + ResultInfo.ArrayDims.append(ast.unparse(slice_node)) + except Exception: # 回退：unparse 失败时用 'N' + ResultInfo.ArrayDims.append('N') + elif isinstance(slice_node, ast.Subscript): + ResultInfo.ArrayPtr = CollectAll(slice_node) + else: + ResultInfo = CTypeInfo() + if isinstance(slice_node, ast.Constant): + ResultInfo.ArrayDims.append(str(slice_node.value)) + elif isinstance(slice_node, ast.Name): + SliceType = getattr(t, slice_node.id, None) + if SliceType == t.CPtr: + ResultInfo.PtrCount += 1 + elif SliceType == t.CArrayPtr: + ResultInfo.ArrayPtr = CTypeInfo() + else: + ResultInfo.ArrayDims.append(slice_node.id) + elif isinstance(value_node, ast.Subscript): + ResultInfo = CollectAll(value_node) + if isinstance(slice_node, ast.Attribute): + SliceVtInfo = self.GetCTypeInfo(slice_node) + if SliceVtInfo and SliceVtInfo.PtrCount >= 1 and not SliceVtInfo.ArrayPtr: + ResultInfo.PtrCount += SliceVtInfo.PtrCount + if SliceVtInfo.BaseType and not isinstance(SliceVtInfo.BaseType, t.CVoid): + ResultInfo.BaseType = SliceVtInfo.BaseType + elif SliceVtInfo and SliceVtInfo.IsArrayPtr: + NewInfo = CTypeInfo() + NewInfo.ArrayPtr = ResultInfo + ResultInfo = NewInfo + elif SliceVtInfo and SliceVtInfo.BaseType and not isinstance(SliceVtInfo.BaseType, t.CVoid): + ResultInfo.BaseType = SliceVtInfo.BaseType + elif isinstance(slice_node, ast.Subscript): + InnerNode = CollectAll(slice_node) + if ResultInfo.ArrayPtr is None: + ResultInfo.ArrayPtr = InnerNode + else: + current = ResultInfo + while current.ArrayPtr is not None: + current = current.ArrayPtr + current.ArrayPtr = InnerNode + elif isinstance(slice_node, ast.Constant): + ResultInfo.ArrayDims.append(str(slice_node.value)) + elif isinstance(slice_node, ast.Name): + SliceType = getattr(t, slice_node.id, None) + if SliceType == t.CPtr: + ResultInfo.PtrCount += 1 + elif SliceType == t.CArrayPtr: + ResultInfo.ArrayPtr = CTypeInfo() + else: + ResultInfo.ArrayDims.append(slice_node.id) + elif isinstance(slice_node, ast.BinOp): + try: + ResultInfo.ArrayDims.append(ast.unparse(slice_node)) + except Exception: # 回退：unparse 失败时用 'N' + ResultInfo.ArrayDims.append('N') + elif isinstance(value_node, ast.Name): + ResultInfo = CTypeInfo() + if IsListAnnotation(node): + if not isinstance(slice_node, ast.Tuple): + ElementType = self._GetCTypeInfoInternal(slice_node) + if ElementType and isinstance(ElementType, CTypeInfo): + ResultInfo.BaseType = ElementType.BaseType if ElementType.BaseType else t.CVoid() + ResultInfo.PtrCount = ElementType.PtrCount + 1 + ResultInfo.IsFuncPtr = ElementType.IsFuncPtr + ResultInfo.FuncPtrReturn = ElementType.FuncPtrReturn + ResultInfo.FuncPtrParams = ElementType.FuncPtrParams + else: + ResultInfo.BaseType = t.CVoid() + ResultInfo.PtrCount = 1 + elif isinstance(slice_node, ast.Tuple) and len(slice_node.elts) >= 1: + ElementType = self._GetCTypeInfoInternal(slice_node.elts[0]) + if ElementType and isinstance(ElementType, CTypeInfo): + ResultInfo.BaseType = ElementType.BaseType if ElementType.BaseType else t.CVoid() + ResultInfo.PtrCount = ElementType.PtrCount + ResultInfo.IsFuncPtr = ElementType.IsFuncPtr + ResultInfo.FuncPtrReturn = ElementType.FuncPtrReturn + ResultInfo.FuncPtrParams = ElementType.FuncPtrParams + for elt in slice_node.elts[1:]: + if isinstance(elt, ast.Constant): + ResultInfo.ArrayDims.append(str(elt.value)) + elif isinstance(elt, ast.Name): + ResultInfo.ArrayDims.append(elt.id) + else: + try: + ResultInfo.ArrayDims.append(ast.unparse(elt)) + except Exception: # 回退：unparse 失败时用 'N' + ResultInfo.ArrayDims.append('N') + else: + ResultInfo.BaseType = CTypeInfo.CreateFromTypeName(value_node.id).BaseType + else: + # 检查是否为泛型类特化（如 list[str]） + _is_generic_template: bool = ( + hasattr(self.Trans, 'ClassHandler') + and hasattr(self.Trans.ClassHandler, '_generic_class_templates') + and value_node.id in self.Trans.ClassHandler._generic_class_templates + ) + if _is_generic_template: + # 走和 _ResolveGenericSlice 同样的规范化路径（_resolve_generic_slice_type）， + # 确保字段类型名与特化名完全一致： + # - ast.Name: 用原始名（如 'Param' → 'Param'，不加 SHA1 前缀）， + # 与 _resolve_generic_slice_type 的 ast.Name 分支一致 + # - BinOp/Attribute: 走 _resolve_type_arg_str（加 SHA1 前缀 / 'ptr' 标记） + # 避免字段类型用 'GSList[sha1.Param]' 而特化名用 'GSList[Param]' 导致 + # find_struct_by_pointee 匹配到 opaque 副本，方法分派失败 + type_arg_str: str + if isinstance(slice_node, ast.Tuple): + arg_parts: list[str] = [] + for elt in slice_node.elts: + slice_res = self.Trans.ExprCallHandle._resolve_generic_slice_type(elt) + arg_parts.append(slice_res[0] if slice_res else self.Trans.ExprCallHandle._resolve_type_arg_str(elt)) + # 必须用 '][' 分隔以匹配 _mangle_generic_class_name 的格式 + # （class_name + '[' + ']['.join(mangled_args) + ']'） + type_arg_str = ']['.join(arg_parts) + else: + slice_res = self.Trans.ExprCallHandle._resolve_generic_slice_type(slice_node) + type_arg_str = slice_res[0] if slice_res else self.Trans.ExprCallHandle._resolve_type_arg_str(slice_node) + full_spec_name: str = f'{value_node.id}[{type_arg_str}]' + ResultInfo.BaseType = t.CStruct(name=full_spec_name) + ResultInfo.Name = full_spec_name + ResultInfo.IsStruct = True + ResultInfo.PtrCount = 1 + # 触发泛型特化发射：确保特化的方法被发射 + struct sha1 注册 + # 与局部变量 AnnAssign 路径一致；否则函数参数注解路径 + # 只构建 mangled 名，方法调用分派时 class_sha1=None 导致 + # "Undefined method" 错误（如 lst.__len__() 在 list[T] | t.CPtr 参数上） + # _specialize_generic_class 内部有 spec_key 去重，重复调用安全 + if hasattr(self.Trans, 'ExprCallHandle') and self.Trans.ExprCallHandle is not None: + self.Trans.ExprCallHandle._ResolveGenericSlice(node) + else: + ResultInfo.BaseType = CTypeInfo.CreateFromTypeName(value_node.id).BaseType + if isinstance(slice_node, ast.Constant): + ResultInfo.ArrayDims.append(str(slice_node.value)) + elif isinstance(slice_node, ast.Name): + SliceType = getattr(t, slice_node.id, None) + if SliceType == t.CPtr: + ResultInfo.PtrCount += 1 + elif SliceType == t.CArrayPtr: + ResultInfo.ArrayPtr = CTypeInfo() + else: + ResultInfo.ArrayDims.append(slice_node.id) + elif isinstance(slice_node, ast.BinOp): + try: + ResultInfo.ArrayDims.append(ast.unparse(slice_node)) + except Exception: # 回退：unparse 失败时用 'N' + ResultInfo.ArrayDims.append('N') + elif isinstance(slice_node, ast.Subscript): + InnerNode = CollectAll(slice_node) + ResultInfo.ArrayDims.extend(InnerNode.ArrayDims) + else: + ResultInfo = CTypeInfo() + + return ResultInfo + elif isinstance(node, ast.Attribute): + VtInfo = self.GetCTypeInfo(node) + result = CTypeInfo() + if VtInfo and VtInfo.IsArrayPtr: + result.ArrayPtr = CTypeInfo() + return result + elif VtInfo and VtInfo.PtrCount == 1 and not VtInfo.ArrayPtr: + result.BaseType = t.CVoid + result.PtrCount = 1 + return result + elif VtInfo and VtInfo.BaseType: + result.BaseType = VtInfo.BaseType + return result + else: + return CTypeInfo() + elif isinstance(node, ast.Name): + result = CTypeInfo() + result.BaseType = CTypeInfo.CreateFromTypeName(node.id).BaseType + return result + else: + return CTypeInfo() + + return CollectAll(Node) + + def _CollectAllOperandTypes(self, Node: ast.AST) -> List: + """递归收集所有操作数的类型（用于一次性合并）""" + Types = [] + + if isinstance(Node, ast.BinOp) and isinstance(Node.op, ast.BitOr): + Types.extend(self._CollectAllOperandTypes(Node.left)) + Types.extend(self._CollectAllOperandTypes(Node.right)) + else: + Types.append(Node) + + return Types + + def _GetTypeFromNode(self, Node: ast.AST) -> CTypeInfo: + """浠?AST 鑺傜偣鑾峰彇绫诲瀷""" + if isinstance(Node, ast.Call): + if isinstance(Node.func, ast.Name) and Node.func.id == 'callable': + return self.GetCTypeInfo(Node) + elif isinstance(Node.func, ast.Attribute) and CTypeHelper.GetTModuleCType(Node.func.attr) == t._CTypedef: + return self.GetCTypeInfo(Node) + elif isinstance(Node.func, ast.Attribute) and Node.func.attr == 'Bit': + if Node.args and isinstance(Node.args[0], ast.Constant): + Result = CTypeInfo() + Result.BaseType = t.CInt + Result.IsBitField = True + Result.BitWidth = Node.args[0].value + return Result + return self._MakeVoidCTypeInfo() + elif isinstance(Node, ast.Subscript): + if (isinstance(Node.value, ast.Attribute) + and isinstance(Node.value.value, ast.Name) + and IsTModule(Node.value.value.id, self.Trans.SymbolTable) + and Node.value.attr == 'Bit'): + if isinstance(Node.slice, ast.Constant) and isinstance(Node.slice.value, int): + Result = CTypeInfo() + Result.BaseType = t.CInt + Result.IsBitField = True + Result.BitWidth = Node.slice.value + return Result + return self.GetCTypeInfo(Node) + else: + return self.GetCTypeInfo(Node) + + def MergeTypes(self, Node: ast.BinOp): + """处理类型组合，如 t.CInt | t.CPtr ?t.CStatic | t.CInt + + 这是 | 运算符的类型合并处理函数 + 收集所有操作数，一次性合? + """ + if not isinstance(Node, ast.BinOp) or not isinstance(Node.op, ast.BitOr): + return None + + AllOperands = self._CollectAllOperandTypes(Node) + + AllTypes = [] + for Operand in AllOperands: + TypeItem = self._GetTypeFromNode(Operand) + AllTypes.append(TypeItem) + + if isinstance(Node.left, ast.Attribute): + LeftParts = [] + current = Node.left + while isinstance(current, ast.Attribute): + attr_type = getattr(t, current.attr, None) + if attr_type and isinstance(attr_type, type) and issubclass(attr_type, t.CType): + LeftParts.insert(0, attr_type) + current = current.value + if isinstance(current, ast.Name): + name_type = getattr(t, current.id, None) + if name_type and isinstance(name_type, type) and issubclass(name_type, t.CType): + LeftParts.insert(0, name_type) + if LeftParts and LeftParts[-1] == t._CTypedef: + for type_item in AllTypes: + if isinstance(type_item, CTypeInfo) and type_item.IsFuncPtr: + type_item.IsTypedef = True + type_item.OriginalType = CTypeInfo() + type_item.OriginalType.IsFuncPtr = True + type_item.OriginalType.FuncPtrReturn = type_item.FuncPtrReturn or CTypeInfo.VoidTypeInfo() + type_item.OriginalType.FuncPtrParams = list(type_item.FuncPtrParams) if type_item.FuncPtrParams else [] + return type_item + Result = CTypeInfo() + if isinstance(AllTypes[-1], CTypeInfo): + Result.BaseType = AllTypes[-1].BaseType + else: + Result.BaseType = t.CInt + Result.IsTypedef = True + return Result + + return self._MergeAllComponentTypes(*AllTypes) + + def _MergeAllComponentTypes(self, *Types: CTypeInfo) -> CTypeInfo: + """合并多个类型 - 直接使用 CTypeInfo 类""" + if not Types: + return CTypeInfo() + + Result = CTypeInfo() + # 统计 t.CPtr（CVoid + PtrCount=1）的个数，用于正确累加指针层数 + # 因为 (AST | t.CPtr) | t.CPtr 展开后会包含多个 t.CPtr，每个都需要累加 + cptr_count: int = 0 + + for TypeItem in Types: + if TypeItem is None: + continue + if TypeItem.IsState: + Result.IsState = True + if not Result.Storage: + Result.Storage = t.CExport() + continue + if TypeItem.IsBitField: + Result.BaseType = t.CInt + Result.IsBitField = True + Result.BitWidth = TypeItem.BitWidth + return Result + if TypeItem.IsFuncPtr: + Result.BaseType = t.CVoid + Result.IsFuncPtr = True + Result.FuncPtrParams = TypeItem.FuncPtrParams + Result.FuncPtrReturn = TypeItem.FuncPtrReturn + return Result + if TypeItem.IsTypedef: + if TypeItem.BaseType and (not isinstance(TypeItem.BaseType, (t._CTypedef,)) or TypeItem.PtrCount > 0): + if Result.BaseType is None or isinstance(Result.BaseType, (t.CVoid, t._CTypedef)): + Result.BaseType = TypeItem.BaseType + Result.Name = TypeItem.Name + Result.IsTypedef = True + Result.PtrCount = max(Result.PtrCount, TypeItem.PtrCount) + if TypeItem.OriginalType: + Result.OriginalType = TypeItem.OriginalType + else: + Result.IsTypedef = True + Result.Name = TypeItem.Name + if TypeItem.OriginalType: + Result.OriginalType = TypeItem.OriginalType + continue + # 统计 t.CPtr（CVoid + PtrCount=1） + if TypeItem.PtrCount > 0 and TypeItem.BaseType and (TypeItem.BaseType is t.CVoid or isinstance(TypeItem.BaseType, t.CVoid)): + cptr_count += 1 + # BaseType 保持不变，只计数，后续统一累加 + if TypeItem.IsRenum and TypeItem.Name: + Result.IsRenum = True + Result.IsEnum = True + Result.IsStruct = True + if not Result.Name: + Result.Name = TypeItem.Name + continue + self._MergeCTypeInfoInto(Result, TypeItem) + + # 对所有 t.CPtr 统一累加指针层数 + # 仅对 struct 类型吸收第一个 CPtr（struct 变量默认是指针，AST | t.CPtr = AST*）。 + # 对于本身已是指针的类型（如 bytes=CChar*），bytes | t.CPtr = i8**（不吸收）。 + base_ptr_count: int = Result.PtrCount + Result.PtrCount += cptr_count + if Result.IsStruct and base_ptr_count > 0 and cptr_count > 0: + Result.PtrCount -= 1 + + if Result.BaseType is None: + Result.BaseType = t.CVoid + + return Result + + def _MergeCTypeInfoInto(self, Target: CTypeInfo, Source: CTypeInfo): + if Source.PtrCount > 0 and Source.BaseType and (Source.BaseType is t.CVoid or isinstance(Source.BaseType, t.CVoid)): + # t.CPtr 的处理已提升到 _MergeAllComponentTypes 中统一计数累加， + # 此处保留仅作为兜底（从其他路径直接调用 _MergeCTypeInfoInto 时） + if Target.PtrCount < Source.PtrCount: + Target.PtrCount = Source.PtrCount + elif Source.PtrCount > 0 and Source.BaseType and Source.BaseType is not t.CVoid and not isinstance(Source.BaseType, t.CVoid): + if Target.BaseType is None or Target.BaseType is t.CVoid or isinstance(Target.BaseType, t.CVoid): + Target.BaseType = Source.BaseType + # 当 Target 已有 PtrCount（来自之前的 t.CPtr），累加而非替换 + Target.PtrCount = Source.PtrCount + Target.PtrCount + elif Target.PtrCount == 0: + Target.PtrCount = Source.PtrCount + elif Source.PtrCount > Target.PtrCount: + Target.PtrCount = Source.PtrCount + elif Source.PtrCount > Target.PtrCount: + Target.PtrCount = Source.PtrCount + if Source.BaseType and Source.BaseType is not t.CVoid and not isinstance(Source.BaseType, (t.CVoid, t._CTypedef)): + if Target.BaseType is None: + Target.BaseType = Source.BaseType + elif (Target.BaseType is t.CVoid or isinstance(Target.BaseType, t.CVoid)) and Target.PtrCount == 0: + Target.BaseType = Source.BaseType + elif (Source.BaseType is t.CVoid or isinstance(Source.BaseType, t.CVoid)) and Target.BaseType is None: + Target.BaseType = Source.BaseType + # Struct 名称保留：BinOp `list[AST | t.CPtr] | t.CPtr` 合并时， + # _HandleSubscript 已在 Source.Name 设置了特化名（如 'list[5dab8cb390496d22.ASTptr]'）， + # 合并后 Target.Name 必须保留此名称，否则 class_member_element_class + # 无法获取正确的特化类名 → 方法分派 fallback 到错误的类 + if Source.IsStruct and Source.Name and not Target.Name: + Target.Name = Source.Name + # REnum 标志和名称保留（BinOp `LLVMType | t.CPtr` 合并时不能丢失 REnum 语义） + # 同时传播 IsStruct=True，让 CPtr 吸收逻辑生效（REnum 在内存布局上是 struct） + if Source.IsRenum and Source.Name: + Target.IsRenum = True + Target.IsEnum = True + Target.IsStruct = True + if not Target.Name: + Target.Name = Source.Name + if Source.IsTypedef and Source.Name: + if Source.BaseType and (not isinstance(Source.BaseType, (t._CTypedef,)) or Source.PtrCount > 0): + if Target.BaseType is None or isinstance(Target.BaseType, (t.CVoid, t._CTypedef)): + Target.BaseType = Source.BaseType + Target.IsTypedef = True + Target.Name = Source.Name + if Source.OriginalType: + Target.OriginalType = Source.OriginalType + elif Target.BaseType is None or isinstance(Target.BaseType, (t.CVoid, t._CTypedef)): + if Source.OriginalType: + if isinstance(Source.OriginalType, CTypeInfo) and Source.OriginalType.BaseType: + Target.BaseType = Source.OriginalType.BaseType + Target.PtrCount = max(Target.PtrCount, Source.OriginalType.PtrCount) + Target.IsTypedef = True + Target.Name = Source.Name + Target.OriginalType = Source.OriginalType + elif isinstance(Source.OriginalType, CTypeInfo): + Target.IsTypedef = True + Target.Name = Source.Name + Target.OriginalType = Source.OriginalType + else: + Target.BaseType = t.CStruct(name=Source.Name) + Target.IsStruct = True + else: + Target.BaseType = t.CStruct(name=Source.Name) + Target.IsStruct = True + if Source.ArrayPtr: + if Target.ArrayPtr is None: + Target.ArrayPtr = Source.ArrayPtr + else: + if Source.PtrCount > Target.PtrCount: + Target.PtrCount = Source.PtrCount + Target.ArrayPtr = Source.ArrayPtr + if Source.PtrCount > 0: + Target.ArrayDims = [] + elif Source.ArrayDims: + Target.ArrayDims.extend(Source.ArrayDims) + if Source.VarConst: + Target.VarConst = True + if Source.VarVolatile: + Target.VarVolatile = True + if Source.DataConst: + Target.DataConst = True + if Source.DataVolatile: + Target.DataVolatile = True + if Source.IsFuncPtr: + Target.IsFuncPtr = True + Target.FuncPtrParams = Source.FuncPtrParams + Target.FuncPtrReturn = Source.FuncPtrReturn + if Source.Storage and not Target.Storage: + Target.Storage = Source.Storage + if Source.ByteOrder and not Target.ByteOrder: + Target.ByteOrder = Source.ByteOrder diff --git a/lib/core/Handles/HandlesWhile.py b/lib/core/Handles/HandlesWhile.py new file mode 100644 index 0000000..b7b4eb4 --- /dev/null +++ b/lib/core/Handles/HandlesWhile.py @@ -0,0 +1,54 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle +import llvmlite.ir as ir + + +class WhileHandle(BaseHandle): + def _HandleWhileLlvm(self, Node: ast.While) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + if Gen._direct_values: + for var_name in list(Gen._direct_values.keys()): + OldVal: ir.Value = Gen._direct_values.pop(var_name) + if var_name not in Gen.variables or Gen.variables.get(var_name) is None: + saved_block: ir.Block = Gen.builder.block + entry_block: ir.Block = Gen.func.entry_basic_block + Gen.builder.position_at_start(entry_block) + var: ir.AllocaInstr = Gen.builder.alloca(OldVal.type, name=var_name) + Gen._store(OldVal, var) + Gen.builder.position_at_end(saved_block) + Gen.variables[var_name] = var + CondBB: ir.Block = Gen.func.append_basic_block(name="while.cond") + BodyBB: ir.Block = Gen.func.append_basic_block(name="while.body") + ElseBB: ir.Block | None = Gen.func.append_basic_block(name="while.else") if Node.orelse else None + AfterBB: ir.Block = Gen.func.append_basic_block(name="while.end") + Gen.loop_break_targets.append(AfterBB) + Gen.loop_continue_targets.append(CondBB) + Gen.builder.branch(CondBB) + Gen.builder.position_at_start(CondBB) + Cond: ir.Value | None = self.HandleExprLlvm(Node.test) + if Cond: + if not isinstance(Cond.type, ir.IntType) or Cond.type.width != 1: + if isinstance(Cond.type, (ir.FloatType, ir.DoubleType)): + Cond = Gen.builder.fcmp_ordered('!=', Cond, ir.Constant(Cond.type, 0.0), name="whilecond") + else: + Cond = Gen.builder.icmp_signed('!=', Cond, Gen._ZeroConst(Cond.type), name="whilecond") + if not Gen.builder.block.is_terminated: + Gen.builder.cbranch(Cond, BodyBB, ElseBB if ElseBB else AfterBB) + else: + if not Gen.builder.block.is_terminated: + Gen.builder.branch(ElseBB if ElseBB else AfterBB) + Gen.builder.position_at_start(BodyBB) + self.HandleBodyLlvm(Node.body) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(CondBB) + Gen.loop_break_targets.pop() + Gen.loop_continue_targets.pop() + if ElseBB: + Gen.builder.position_at_start(ElseBB) + self.HandleBodyLlvm(Node.orelse) + if not Gen.builder.block.is_terminated: + Gen.builder.branch(AfterBB) + Gen.builder.position_at_start(AfterBB) \ No newline at end of file diff --git a/lib/core/Handles/HandlesWith.py b/lib/core/Handles/HandlesWith.py new file mode 100644 index 0000000..91bfea4 --- /dev/null +++ b/lib/core/Handles/HandlesWith.py @@ -0,0 +1,172 @@ +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.translator import Translator +from lib.core.Handles.HandlesBase import BaseHandle +import ast +import llvmlite.ir as ir + + +class WithHandle(BaseHandle): + def _HandleWithLlvm(self, Node: ast.With) -> None: + Gen: "Translator.LlvmGen" = self.Trans.LlvmGen + for item in Node.items: + context_expr: ast.expr = item.context_expr + asname: str | None = None + target: ast.expr | None = getattr(item, 'optional_vars', getattr(item, 'target', getattr(item, 'asname', None))) + if target and isinstance(target, ast.Name): + asname = target.id + ClassName: str | None = None + if isinstance(context_expr, ast.Call): + if isinstance(context_expr.func, ast.Name): + ClassName = context_expr.func.id + elif isinstance(context_expr.func, ast.Attribute): + ClassName = context_expr.func.attr + if ClassName and ClassName not in Gen.structs: + SymInfo: "SymbolTable.SymbolInfo" = self.Trans.SymbolTable.lookup(ClassName) + if SymInfo: + if SymInfo.IsStruct or SymInfo.IsRenum: + pass + else: + ClassName = None + else: + ClassName = None + elif isinstance(context_expr, ast.Name): + VarName: str = context_expr.id + if VarName in Gen.var_struct_class: + ClassName = Gen.var_struct_class[VarName] + ctx_val: ir.Value | None = self.HandleExprLlvm(context_expr) + if not ctx_val: + continue + original_ctx_val: ir.Value = ctx_val + if ClassName and isinstance(ctx_val.type, ir.PointerType): + pointee: ir.Type = ctx_val.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + ClassName = CN + StructPtrType: ir.PointerType | None = ir.PointerType(Gen.structs[ClassName]) if ClassName and ClassName in Gen.structs else None + if StructPtrType and isinstance(ctx_val.type, ir.PointerType) and ctx_val.type.pointee != Gen.structs.get(ClassName): + ctx_val = Gen.builder.bitcast(ctx_val, StructPtrType, name="cast_with_ctx") + ctx_alloca: ir.AllocaInstr = Gen._allocaEntry(ctx_val.type, name="__with_ctx") + Gen._store(ctx_val, ctx_alloca) + Gen._UnregisterTempPtr(ctx_val) + if ClassName: + Gen.var_struct_class['__with_ctx'] = ClassName + Gen._var_to_heap_ptr['__with_ctx'] = ctx_val + original_asname_heap_ptr: ir.Value | None = Gen._var_to_heap_ptr.get(asname) if asname else None + ctx_is_var_ref: bool = isinstance(context_expr, ast.Name) + enter_result: ir.Value | None = None + if ClassName: + EnterFuncName: str = f'{ClassName}.__enter__' + if EnterFuncName in Gen.functions: + ctx_Loaded: ir.Value = Gen._load(ctx_alloca, name="__with_ctx") + enter_call: ir.CallInstr = Gen.builder.call(Gen.functions[EnterFuncName], [ctx_Loaded], name="call_enter") + enter_result = enter_call + if isinstance(enter_call.type, ir.PointerType): + pointee: ir.Type = enter_call.type.pointee + if isinstance(pointee, ir.PointerType) and isinstance(pointee.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + enter_result = Gen._load(enter_call, name="enter_deref") + elif self._is_char_pointer(enter_call) and StructPtrType: + enter_result = Gen.builder.bitcast(enter_call, StructPtrType, name="cast_enter") + if asname and enter_result: + if isinstance(enter_result.type, ir.PointerType) and isinstance(enter_result.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + found: tuple[str, ir.Type] | None = Gen.find_struct_by_pointee(enter_result.type.pointee) + if found: + CN: str = found[0] + ST: ir.Type = found[1] + Gen.var_struct_class[asname] = CN + Gen._var_to_heap_ptr[asname] = enter_result + Gen._register_local_heap_ptr(enter_result, VarName=asname) + if asname in Gen._reg_values: + OldVal: ir.Value = Gen._reg_values[asname] + var: ir.AllocaInstr = Gen._allocaEntry(OldVal.type, name=asname) + Gen._store(OldVal, var) + Gen.variables[asname] = var + del Gen._reg_values[asname] + if asname in Gen.variables and Gen.variables[asname] is not None: + VarPtr: ir.Value = Gen.variables[asname] + if VarPtr.type.pointee == enter_result.type: + Gen._store(enter_result, VarPtr) + elif isinstance(VarPtr.type.pointee, ir.PointerType) and isinstance(enter_result.type, ir.PointerType): + CastedVal: ir.Value = Gen.builder.bitcast(enter_result, VarPtr.type.pointee, name=f"cast_{asname}") + Gen._store(CastedVal, VarPtr) + else: + NewVar: ir.AllocaInstr = Gen._alloca(enter_result.type, name=asname) + Gen._store(enter_result, NewVar) + Gen.variables[asname] = NewVar + else: + Gen._reg_values[asname] = enter_result + Gen.variables[asname] = None + # 如果类有 __provides__，自动生成 push 代码 + if ClassName and ClassName in Gen.class_provides: + # 编译期检查: with provider 栈深度是否超过 _MAX_PROVIDER_DEPTH (32) + # _with_provides_stack 只反映当前嵌套路径，顺序执行的 with 会在退出时弹栈 + current_depth: int = sum(len(pl) for pl in Gen._with_provides_stack) + new_fields: int = len(Gen.class_provides[ClassName]) + if current_depth + new_fields > 32: + raise SyntaxError( + f"[TransPyC] with provider stack overflow at line {Node.lineno}: " + f"current depth {current_depth} + new {new_fields} fields > max 32. " + f"Class '{ClassName}' provides too many fields or with nesting too deep." + ) + # 编译期压栈：记录当前 with 上下文提供的字段名，供 __require_must__ 静态检查 + Gen._with_provides_stack.append(Gen.class_provides[ClassName]) + ctx_Loaded: ir.Value = Gen._load(ctx_alloca, name="__with_ctx") + ctx_i8: ir.Value = ctx_Loaded + if isinstance(ctx_Loaded.type, ir.PointerType) and ctx_Loaded.type != ir.PointerType(ir.IntType(8)): + ctx_i8 = Gen.builder.bitcast(ctx_Loaded, ir.PointerType(ir.IntType(8)), name="ctx_push_i8") + for field_name in Gen.class_provides[ClassName]: + field_str: ir.Value = Gen._create_string_global(field_name) + push_func = Gen.functions.get('_push_provider') + if push_func: + Gen.builder.call(push_func, [field_str, ctx_i8], name="call_push_provider") + self.Trans.VarScopes.append({}) + if asname: + self.Trans.VarScopes[-1][asname] = True + self.HandleBodyLlvm(Node.body) + if self.Trans.VarScopes: + self.Trans.VarScopes.pop() + # 如果类有 __provides__，自动生成 pop 代码 + if ClassName and ClassName in Gen.class_provides: + for _ in Gen.class_provides[ClassName]: + pop_func = Gen.functions.get('_pop_provider') + if pop_func: + Gen.builder.call(pop_func, [], name="call_pop_provider") + # 编译期弹栈 + if Gen._with_provides_stack: + Gen._with_provides_stack.pop() + if ClassName: + ExitFuncName: str = f'{ClassName}.__exit__' + if ExitFuncName in Gen.functions: + ctx_Loaded: ir.Value = Gen._load(ctx_alloca, name="__with_ctx") + Gen.builder.call(Gen.functions[ExitFuncName], [ctx_Loaded], name="call_exit") + Gen._unregister_local_heap_ptr(original_ctx_val) + if ctx_val is not original_ctx_val: + Gen._unregister_local_heap_ptr(ctx_val) + if ctx_is_var_ref and original_asname_heap_ptr and original_asname_heap_ptr is not original_ctx_val and original_asname_heap_ptr is not ctx_val: + Gen._unregister_local_heap_ptr(original_asname_heap_ptr) + if '__with_ctx' in Gen._var_to_heap_ptr: + del Gen._var_to_heap_ptr['__with_ctx'] + if asname and enter_result is not None and isinstance(ctx_val.type, ir.PointerType) and isinstance(enter_result.type, ir.PointerType): + ctx_int: ir.Value = Gen.builder.ptrtoint(ctx_val, ir.IntType(64), name="ctx_int") + enter_int: ir.Value = Gen.builder.ptrtoint(enter_result, ir.IntType(64), name="enter_int") + same_ptr: ir.Value = Gen.builder.icmp_unsigned('==', ctx_int, enter_int, name="same_ptr_check") + with_free_bb: ir.Block = Gen.func.append_basic_block("with_free_ctx") + with_skip_bb: ir.Block = Gen.func.append_basic_block("with_skip_free") + Gen.builder.cbranch(same_ptr, with_skip_bb, with_free_bb) + Gen.builder.position_at_start(with_free_bb) + raw: ir.Value = Gen.builder.bitcast(ctx_val, ir.PointerType(ir.IntType(8)), name="with_free_cast") + free_func: ir.Function = Gen.get_or_declare_c_func('free', ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))])) + Gen.builder.call(free_func, [raw]) + Gen.builder.branch(with_skip_bb) + Gen.builder.position_at_start(with_skip_bb) + Gen._unregister_local_heap_ptr(enter_result) + Gen._unregister_local_heap_ptr(original_ctx_val) + elif isinstance(ctx_val.type, ir.PointerType): + raw: ir.Value = Gen.builder.bitcast(ctx_val, ir.PointerType(ir.IntType(8)), name="with_free_cast") + free_func: ir.Function = Gen.get_or_declare_c_func('free', ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))])) + Gen.builder.call(free_func, [raw]) + Gen._unregister_local_heap_ptr(original_ctx_val) \ No newline at end of file diff --git a/lib/core/Handles/__init__.py b/lib/core/Handles/__init__.py new file mode 100644 index 0000000..b7c469e --- /dev/null +++ b/lib/core/Handles/__init__.py @@ -0,0 +1,32 @@ +""" +Handles 包 - 使用 Mixin 模式组织 Handle 函数 +""" + +from .HandlesIf import IfHandle +from .HandlesFor import ForHandle +from .HandlesWhile import WhileHandle +from .HandlesAnnAssign import AnnAssignHandle +from .HandlesTypeMerge import HandlesTypeMerge +from .HandlesExpr import ExprHandle +from .HandlesExprUtils import ExprUtils +from .HandlesExprOps import ExprOpsHandle +from .HandlesExprAttr import ExprAttrHandle +from .HandlesExprBuiltin import ExprBuiltinHandle +from .HandlesExprAsm import ExprAsmHandle +from .HandlesExprFormat import ExprFormatHandle +from .HandlesExprLambda import ExprLambdaHandle +from .HandlesExprCall import ExprCallHandle +from .HandlesSpecialCall import CSpecialCallHandle, TSpecialCallHandle +from .HandlesBody import BodyHandle +from .HandlesReturn import ReturnHandle +from .HandlesAssign import AssignHandle +from .HandlesAugAssign import AugAssignHandle +from .HandlesDelete import DeleteHandle +from .HandlesWith import WithHandle +from .HandlesTry import TryHandle +from .HandlesAssert import AssertHandle +from .HandlesRaise import RaiseHandle +from .HandlesMatch import MatchHandle +from .HandlesClassDef import ClassHandle +from .HandlesFunctions import FunctionHandle +from .HandlesImports import ImportHandle diff --git a/lib/core/LLVMCG/BaseGen.py b/lib/core/LLVMCG/BaseGen.py new file mode 100644 index 0000000..8689680 --- /dev/null +++ b/lib/core/LLVMCG/BaseGen.py @@ -0,0 +1,657 @@ +from __future__ import annotations +from typing import Any +import ast +import os +import llvmlite.ir as ir +from lib.includes import t as _t +from lib.includes.t import CTypeRegistry +from lib.core.Handles.HandlesBase import CTypeInfo as _CTypeInfo, BuiltinTypeMap + + +class VaArgInstruction(ir.Instruction): + def descr(self, buf: list[str]) -> None: + ptr: ir.Value = self.operands[0] + buf.append("va_arg {0} {1}, {2}\n".format( + ptr.type, ptr.get_reference(), self.type)) + + +class BaseGenMixin: + + @staticmethod + def _parse_triple(triple: str | None) -> dict[str, bool | int]: + """解析目标三元组，返回平台信息字典""" + info: dict[str, bool | int] = { + 'is_windows': False, 'is_linux': False, 'is_macos': False, + 'is_x86_64': False, 'is_arm64': False, 'is_32bit': False, + 'ptr_size': 8, 'long_size': 8, 'wchar_t_size': 32, + } + if not triple: + return info + triple_lower: str = triple.lower() + # 检测操作系统 + if 'windows' in triple_lower or 'win32' in triple_lower or 'mingw' in triple_lower or 'msvc' in triple_lower or 'cygwin' in triple_lower: + info['is_windows'] = True + # Windows LLP64: long = 32 位, wchar_t = 16 位 + info['long_size'] = 32 + info['wchar_t_size'] = 16 + elif 'linux' in triple_lower: + info['is_linux'] = True + elif 'darwin' in triple_lower or 'macos' in triple_lower or 'apple' in triple_lower: + info['is_macos'] = True + # 检测架构 + if 'x86_64' in triple_lower or 'amd64' in triple_lower or 'x64' in triple_lower: + info['is_x86_64'] = True + elif 'aarch64' in triple_lower or 'arm64' in triple_lower: + info['is_arm64'] = True + elif 'i386' in triple_lower or 'i686' in triple_lower or 'arm' in triple_lower: + info['is_32bit'] = True + info['ptr_size'] = 4 + info['long_size'] = 32 + if info['is_windows']: + info['wchar_t_size'] = 16 + else: + info['wchar_t_size'] = 32 + return info + + def __init__(self, triple: str | None = None, datalayout: str | None = None) -> None: + self.module: ir.Module = ir.Module(name="transpyc_module") + if triple: + self.module.triple = triple + else: + self.module.triple = "x86_64-none-elf" + if datalayout: + self.module.data_layout = datalayout + else: + self.module.data_layout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + + # 解析目标平台信息 + self._platform_info: dict[str, bool | int] = self._parse_triple(self.module.triple) + self.ptr_size: int = self._platform_info['ptr_size'] + # 根据目标平台配置 t 模块中的平台相关类型大小 + _t.configure_platform(self.module.triple) + self.builder: ir.IRBuilder | None = None + self.func: ir.Function | None = None + self.functions: dict[str, ir.Function] = {} + self.variables: dict[str, ir.Value | None] = {} + self._direct_values: dict[str, ir.Value] = {} + self._reg_values: dict[str, ir.Value] = {} + self.var_signedness: dict[str, bool] = {} + self._unsigned_results: set[int] = {} + self.string_const_counter: int = 0 + self.Vtables: dict[str, ir.GlobalVariable] = {} + self.class_methods: dict[str, list[str]] = {} + self.class_members: dict[str, list[tuple[str, ir.Type]]] = {} + self.class_member_defaults: dict[str, dict[str, Any]] = {} + self.class_member_signeds: dict[str, dict[str, bool]] = {} + self.class_member_bitfields: dict[str, dict[str, int]] = {} + self.class_member_byteorders: dict[str, dict[str, str]] = {} + self.local_var_byteorders: dict[str, str] = {} + self.class_member_element_class: dict[str, dict[str, str]] = {} + self.class_member_bitoffsets: dict[str, dict[str, int]] = {} + self.structs: dict[str, ir.IdentifiedStructType] = {} + self.typedef_int_widths: dict[str, int] = {} # typedef名称 -> 底层整数位宽 (如 'UINT' -> 32, 'BYTE' -> 8) + self._cross_module_vtable_classes: set[str] = set() + self._cross_module_novtable: set[str] = set() + self.class_vtable: set[str] = set() + self._class_graph: dict[str, Any] = {} + self.class_parent: dict[str, str] = {} + # Store member annotations for deferred type resolution (cross-module class references) + self.class_member_annotations: dict[str, dict[str, Any]] = {} + self.loop_break_targets: list[ir.Block] = [] + self.loop_continue_targets: list[ir.Block] = [] + self.eh_jmp_buf_stack: list[ir.Value] = [] + self.eh_except_block_stack: list[ir.Block] = [] + self.eh_finally_stack: list[Any] = [] + self.global_vars: set[str] = set() + self.nonlocal_params: dict[str, Any] = {} + self._variable_scope_stack: list[dict[str, Any]] = [] # Stack of outer scope variables for nested function nonlocal lookup + self.var_struct_class: dict[str, str] = {} + self.var_ptr_element: dict[str, bool] = {} + # 全局变量的指针元素标志（持久，不受函数重置影响） + # 用于全局变量 _argv: bytes | t.CPtr 等，使 _argv[index] 使用 8 字节指针步长 + self.global_var_ptr_element: dict[str, bool] = {} + self.class_provides: dict[str, list[str]] = {} + self.class_requires: dict[str, list[str]] = {} + self.class_require_must: dict[str, list[str]] = {} + # 编译期 with 上下文栈：记录当前嵌套 with 链中所有可用的 provides 字段名 + # 用于 __require_must__ 的纯编译期静态可达性检查 + self._with_provides_stack: list[list[str]] = [] + # 函数参数自动取地址模式: func_name -> [None, None, 'need', 'always', ...] + # 'need': t.CNeedPtr — 仅对非指针值取地址 + # 'always': t.CAutoPtr — 无条件取地址 + self._auto_addr_params: dict[str, list[str | None]] = {} + self.global_struct_class: dict[str, str] = {} + self.var_type_info: dict[str, Any] = {} + self.var_const_flags: dict[str, bool] = {} + self._current_source_file: str | None = None + self._current_source_lines: list[str] = [] + self._ns_cache: dict[str, str] = {} + self._type_cache: dict[tuple[str, bool], ir.Type] = {} + self._typedef_cache: dict[str, Any] = {} + self._func_sig_cache: dict[str, Any] = {} + self.var_type_assignments: dict[str, Any] = {} + self._temp_struct_ptrs: list[ir.Value] = [] + self._stop_iter_flag_param: ir.Value | None = None + self._local_heap_ptrs: list[tuple[ir.Value, str, str | None]] = [] + self._var_to_heap_ptr: dict[str, ir.Value] = {} + self.known_return_types: dict[str, ir.Type] = {} + self._current_lineno: int = 0 + self._current_node_info: str = "" + self.module_sha1: str | None = None + self.ModuleSha1Map: dict[str, str] = {} + self._temp_dir: str | None = None + self._export_funcs: set[str] = set() + self._export_extern_funcs: dict[str, str] = {} + self._current_module_func_names: set[str] = set() + # __doc__ 魔法字符串注册表：symbol_name -> docstring 文本（None 表示空/优化剥离） + self._doc_strings: dict[str, str | None] = {} + self.class_packed: set[str] = set() + self._define_constants: dict[str, int] = {} + # 以下属性原先在外部懒注入，此处显式初始化以避免 hasattr/getattr 动态反射 + self._struct_sha1_map: dict[str, str] = {} + self.SymbolTable: Any = None + self._Trans: Any = None + self.class_sha1_map: dict[str, str] = {} + # 以下属性原先懒注入，此处显式初始化以消除 hasattr/getattr 动态反射 + self._import_handler_ref: Any = None + self._classes_in_progress: set[str] = set() + # 字段收集期间添加当前类名，阻止 CollectAnnAssignMember 触发的嵌套 + # _generate_structs 用不完整 class_members 调用 set_body（set_body 只能调用一次） + # 用集合而非单值，因为嵌套特化类 _EmitClassLlvm 会清除单值，导致外层类被错误设置 body + self._collecting_members_set: set[str] = set() + self._last_var_name: str | None = None + self._variadic_info: dict | None = None + self._vtable_copy_counter: int = 0 + self._decorated_funcs: dict = {} + self._current_tree: Any = None + self._DefineConstants: dict = {} + self._all_define_constants: dict = {} + pi: dict[str, bool | int] = self._platform_info + # 根据目标平台设置宏（声明式，从三元组推导） + if pi['is_windows']: + self._define_constants['_WIN32'] = 1 + if not pi['is_32bit']: + self._define_constants['_WIN64'] = 1 + self._define_constants['WIN64'] = 1 + self._define_constants['WIN32'] = 1 + if pi['is_linux']: + self._define_constants['__linux__'] = 1 + if pi['is_macos']: + self._define_constants['__APPLE__'] = 1 + self._define_constants['__MACH__'] = 1 + if pi['is_x86_64']: + self._define_constants['__x86_64__'] = 1 + self._define_constants['__x86_64'] = 1 + self._define_constants['_M_X64'] = 1 + elif pi['is_arm64']: + self._define_constants['__aarch64__'] = 1 + self._define_constants['_M_ARM64'] = 1 + if not pi['is_32bit'] and pi['is_linux']: + self._define_constants['__LP64__'] = 1 + elif pi['is_32bit']: + self._define_constants['_ILP32'] = 1 + self._define_constants['SIZEOF_VOID_P'] = pi['ptr_size'] + self._define_constants['__SIZEOF_POINTER__'] = pi['ptr_size'] + + def find_struct_by_pointee(self, pointee: ir.Type) -> tuple[str, ir.IdentifiedStructType] | None: + """根据 pointee 类型查找匹配的结构体，返回 (class_name, struct_type) 或 None""" + for class_name, struct_type in self.structs.items(): + if pointee == struct_type: + return (class_name, struct_type) + return None + + @staticmethod + def _extract_short_name(name: str) -> str: + """从结构体类型名中提取短名（去掉 sha1 前缀）。 + 如 "6bc91356c652f8ca.Function" → "Function" + 如 "Function" → "Function"（无 sha1 前缀时原样返回） + """ + if '.' in name: + prefix: str = name.split('.', 1)[0] + if len(prefix) == 16 and all(c in '0123456789abcdef' for c in prefix): + return name.split('.', 1)[1] + return name + + @staticmethod + def _extract_struct_sha1(st: ir.IdentifiedStructType) -> str: + """从 IdentifiedStructType 的 name 中提取 sha1 前缀（16 位十六进制）。 + 返回空字符串表示无 sha1 前缀。用于同名类冲突检测。 + """ + if not isinstance(st, ir.IdentifiedStructType): + return '' + try: + name: str = st.name + except (AssertionError, AttributeError): + return '' + if '.' in name: + prefix: str = name.split('.', 1)[0] + if len(prefix) == 16 and all(c in '0123456789abcdef' for c in prefix): + return prefix + return '' + + def _get_or_create_struct(self, name: str, source_sha1: str | None = None, packed: bool = False) -> ir.IdentifiedStructType: + clean_name: str = name.replace(' *', '').replace('*', '').replace(' ', '_').replace('const', 'const_').replace('volatile', 'volatile_') + # 如果名称包含模块前缀（如 "a9fa0f6200c09e65.zbit_writer"）， + # 先尝试用短名称（如 "zbit_writer"）查找已注册的 struct， + # 避免为同一 struct 创建多个不同的类型对象导致跨模块引用类型不一致 + if '.' in clean_name and clean_name not in self.structs: + parts: list[str] = clean_name.split('.', 1) + if len(parts[0]) == 16 and all(c in '0123456789abcdef' for c in parts[0]): + short_name: str = parts[1] + if short_name in self.structs: + existing_short: ir.IdentifiedStructType = self.structs[short_name] + # 治本修复：同名类 sha1 冲突检测。 + # 当请求的 sha1 前缀与现有 struct 的 sha1 前缀不一致时 + # （如 stmts.py:Module 与 __module.py:Module 同名冲突）， + # 不能复用已有 struct（set_body 已锁定为错误布局）， + # 需创建独立 struct 并以 sha1 前缀全名存储。 + _existing_sha1: str = self._extract_struct_sha1(existing_short) + if _existing_sha1 and _existing_sha1 != parts[0]: + if clean_name in self.structs: + return self.structs[clean_name] + st_new: ir.IdentifiedStructType = ir.IdentifiedStructType(self.module, name=clean_name, packed=packed) + self.structs[clean_name] = st_new + return st_new + return existing_short + # 优先解析 typedef 整数类型（如 INT/UINT/BYTE），避免将其注册为 opaque struct。 + # 必须在 `if clean_name in self.structs` 缓存检查之前执行， + # 否则首次调用（SymbolTable 未就绪时）会将 typedef 名缓存为 opaque struct， + # 后续调用即使 SymbolTable 就绪也只会返回缓存的错误类型。 + # 此修复同时解决两个问题： + # 1. alloca unsized type：变量声明 `argc: INT = 6` 不再生成 `alloca %"...INT"` + # 2. 下标访问：FindStructNameInAnnotation 不再匹配 typedef 名（因不在 Gen.structs） + td_width: int | None = self._resolve_typedef_int_width(clean_name) + if td_width is not None: + if td_width > 0: + self.typedef_int_widths[clean_name] = td_width + return ir.IntType(td_width) + # td_width == 0: 特殊 typedef 名称（str/bytes/list 等），返回 i8 + self.typedef_int_widths[clean_name] = 0 + return ir.IntType(8) + if clean_name in self.structs: + existing: ir.IdentifiedStructType = self.structs[clean_name] + # 治本修复：同名类 sha1 冲突检测（短名调用路径）。 + # 当 source_sha1 与现有 struct 的 sha1 前缀不一致时（跨模块同名类冲突）， + # 不能复用已有 struct（set_body 已锁定为错误布局）， + # 需创建独立 struct 并以 sha1 前缀全名存储，不覆盖短名条目。 + if source_sha1: + _existing_sha1: str = self._extract_struct_sha1(existing) + if _existing_sha1 and _existing_sha1 != source_sha1: + full_key: str = f"{source_sha1}.{clean_name}" + if full_key in self.structs: + st_full: ir.IdentifiedStructType = self.structs[full_key] + if packed and isinstance(st_full, ir.IdentifiedStructType) and not st_full.packed: + st_full.packed = True + return st_full + ns_name: str = f"{source_sha1}.{clean_name}" + st_new: ir.IdentifiedStructType = ir.IdentifiedStructType(self.module, name=ns_name, packed=packed) + self.structs[full_key] = st_new + return st_new + if packed and isinstance(existing, ir.IdentifiedStructType) and not existing.packed: + existing.packed = True + return existing + if clean_name in {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum', 'Void', 'void'}: + return ir.IntType(8) + if self.SymbolTable: + Entry: _CTypeInfo | None = self.SymbolTable.lookup(clean_name) + if Entry and Entry.IsTypedef: + resolved_str: str = self._resolve_typedef(clean_name) + if resolved_str != clean_name: + return self._type_str_to_llvm(resolved_str) + if Entry.BaseType: + if not isinstance(Entry.BaseType, (_t._CTypedef,)) or Entry.PtrCount > 0: + resolved: str = self._resolve_ctype_to_str(Entry) + if resolved: + return self._type_str_to_llvm(resolved) + if Entry and Entry.IsRenum: + if clean_name in self.structs: + return self.structs[clean_name] + # REnum 结构体尚未创建时不能 fall through 到 IsEnum（会返回 i32）， + # 需要继续到下方正常结构体创建路径，创建 opaque 结构体。 + # 这对自引用 REnum（如 LLVMType 包含 LLVMType*）至关重要： + # _parse_llvm_type 先通过此路径创建 opaque 结构体，_TryLoadStructFromStub 再 set_body。 + elif Entry and Entry.IsEnum and clean_name not in self.class_methods: + return ir.IntType(32) + if Entry and Entry.IsExceptionClass: + return ir.IntType(32) + if Entry and Entry.IsFunction: + return ir.PointerType(ir.IntType(8)) + if Entry and Entry.IsVariable: + return ir.PointerType(ir.IntType(8)) + if Entry and Entry.BaseType: + if isinstance(Entry.BaseType, type) and issubclass(Entry.BaseType, _t.CEnum): + return ir.IntType(32) + if Entry.BaseType is _t.CEnum: + return ir.IntType(32) + ns_name: str + if source_sha1: + ns_name = f"{source_sha1}.{clean_name}" + elif clean_name in self._struct_sha1_map: + ns_name = f"{self._struct_sha1_map[clean_name]}.{clean_name}" + else: + ns_name = self._mangle_name(clean_name) + st: ir.IdentifiedStructType = ir.IdentifiedStructType(self.module, name=ns_name, packed=packed) + self.structs[clean_name] = st + # 尝试解析 typedef 的底层整数位宽 + width: int | None = self._resolve_typedef_int_width(clean_name) + if width is not None: + self.typedef_int_widths[clean_name] = width + return st + + def _resolve_typedef_int_width(self, name: str) -> int | None: + """解析 typedef 名称的底层整数位宽 + + 通过 CTypeRegistry 和 BuiltinTypeMap 动态解析， + 替代硬编码的 TYPEDEF_NAMES 映射表。 + 返回 int 位宽或 None（无法解析）。 + """ + # 1. 先查 BuiltinTypeMap（覆盖 BYTEPTR 等指针别名） + bm: Any = BuiltinTypeMap.Get(name) + if bm: + ctype_cls: Any + ptr_count: int + ctype_cls, ptr_count = bm + if ptr_count > 0: + return None # 指针类型，不是整数 typedef + llvm_str: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str and llvm_str.startswith('i') and llvm_str[1:].isdigit(): + return int(llvm_str[1:]) + + # 2. 再查 SymbolTable（用户定义的 typedef） + if self.SymbolTable: + Entry: _CTypeInfo | None = self.SymbolTable.lookup(name) + if Entry and Entry.IsTypedef: + if Entry.BaseType: + if not isinstance(Entry.BaseType, (_t._CTypedef,)) and Entry.PtrCount == 0: + llvm_str: str = CTypeRegistry.CTypeToLLVM(type(Entry.BaseType) if isinstance(Entry.BaseType, _t.CType) else Entry.BaseType) + if llvm_str and llvm_str.startswith('i') and llvm_str[1:].isdigit(): + return int(llvm_str[1:]) + if Entry.OriginalType: + if isinstance(Entry.OriginalType, _CTypeInfo) and Entry.OriginalType.BaseType: + if not isinstance(Entry.OriginalType.BaseType, (_t._CTypedef,)) and Entry.OriginalType.PtrCount == 0: + llvm_str: str = CTypeRegistry.CTypeToLLVM(type(Entry.OriginalType.BaseType) if isinstance(Entry.OriginalType.BaseType, _t.CType) else Entry.OriginalType.BaseType) + if llvm_str and llvm_str.startswith('i') and llvm_str[1:].isdigit(): + return int(llvm_str[1:]) + elif isinstance(Entry.OriginalType, str): + # OriginalType 是类型名字符串（如 'CInt', 'CUnsignedInt'）， + # 通过 CTypeRegistry 解析为整数位宽 + cls: Any = CTypeRegistry.GetClassByName(Entry.OriginalType) + if cls: + llvm_str: str = CTypeRegistry.CTypeToLLVM(cls) + if llvm_str and llvm_str.startswith('i') and llvm_str[1:].isdigit(): + return int(llvm_str[1:]) + + # 3. 尝试 CTypeRegistry 直接按名称查找 + ctype_cls: Any = CTypeRegistry.GetClassByName(name) + if ctype_cls: + llvm_str: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str and llvm_str.startswith('i') and llvm_str[1:].isdigit(): + return int(llvm_str[1:]) + + # 4. 特殊名称（非整数类型，但需要标记为 typedef 以跳过 opaque 定义） + _SPECIAL_TYPEDEFS: set[str] = { + 'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', + 'CStruct', 'enum', 'REnum', 'renum', + 'list', 'dict', 'tuple', 'set', 'str', 'bytes', 'array', 'type', 'CArray', + } + if name in _SPECIAL_TYPEDEFS: + return 0 # 标记为 typedef，但位宽为 0（非整数） + + return None + + def _set_node_info(self, node: ast.AST, extra_info: str = "") -> None: + if hasattr(node, 'lineno'): + self._current_lineno = node.lineno + else: + self._current_lineno = 0 + if hasattr(node, 'col_offset'): + self._current_node_info = f"line {self._current_lineno}, col {node.col_offset}" + else: + self._current_node_info = f"line {self._current_lineno}" + if extra_info: + self._current_node_info += f" ({extra_info})" + + def _set_source_info(self, filepath: str) -> None: + self._current_source_file = filepath + try: + with open(filepath, 'r', encoding='utf-8') as f: + self._current_source_lines = f.readlines() + except Exception: # 文件读取失败时回退为空行列表 + self._current_source_lines = [] + + def _skip_mangle(self, name: str) -> bool: + skip_prefixes: tuple[str, ...] = ('llvm.', 'str_const_', 'printf', 'memset_pattern') + if any(name.startswith(p) for p in skip_prefixes): + return True + if '.' in name: + parts: list[str] = name.split('.', 1) + if len(parts[0]) >= 8 and all(c in '0123456789abcdef' for c in parts[0]): + return True + return False + + def _mangle_name(self, name: str) -> str: + if self._skip_mangle(name): + return name + if name in self._export_funcs: + return name + if self.module_sha1: + return f"{self.module_sha1}.{name}" + return name + + def _find_function(self, name: str) -> ir.Function | None: + if name in self.functions: + return self.functions[name] + g: Any = self.module.globals.get(name) + if g and isinstance(g, ir.Function): + self.functions[name] = g + return g + mangled: str = self._mangle_func_name(name) + if mangled != name: + if mangled in self.functions: + self.functions[name] = self.functions[mangled] + return self.functions[mangled] + g = self.module.globals.get(mangled) + if g and isinstance(g, ir.Function): + self.functions[mangled] = g + self.functions[name] = g + return g + for sha1 in self.ModuleSha1Map.values(): + prefixed: str = f"{sha1}.{name}" + if prefixed in self.functions: + self.functions[name] = self.functions[prefixed] + return self.functions[prefixed] + g = self.module.globals.get(prefixed) + if g and isinstance(g, ir.Function): + self.functions[prefixed] = g + self.functions[name] = g + return g + return None + + def _has_function(self, name: str) -> bool: + return self._find_function(name) is not None + + def _get_or_declare_function(self, mangled_name: str, func_type: ir.FunctionType) -> ir.Function: + g = self.module.globals.get(mangled_name) + if g and isinstance(g, ir.Function): + return g + return ir.Function(self.module, func_type, name=mangled_name) + + def _get_function(self, name: str) -> ir.Function | None: + func: ir.Function | None = self._find_function(name) + return func + + def _mangle_func_name(self, name: str, module_name: str | None = None) -> str: + if name in self._export_funcs and not module_name: + return name + if name in self._export_extern_funcs: + target_sha1: str = self._export_extern_funcs[name] + if module_name: + # 显式带 module_name 调用（如 stdlib.system）：即使当前模块定义了同名函数， + # 也应按模块匹配 extern，返回裸名。snprintf 之所以工作而 system 不工作， + # 就是因为 os._win32 定义了自己的 system 包装，导致旧逻辑跳过了 extern 匹配。 + if module_name == target_sha1: + return name + if module_name in self.ModuleSha1Map and self.ModuleSha1Map[module_name] == target_sha1: + return name + init_name: str = f"{module_name}.__init__" + if init_name in self.ModuleSha1Map and self.ModuleSha1Map[init_name] == target_sha1: + return name + # 兜底：module_name 不在 ModuleSha1Map 中或 sha1 不匹配， + # 但函数在 _export_extern_funcs 中（C extern 不应混淆）， + # 仅当当前模块未定义同名函数时才返回裸名。 + if name not in self._current_module_func_names: + return name + elif name not in self._current_module_func_names: + # 无 module_name 调用（如 from stdlib import system; system()）： + # 仅当当前模块未定义同名函数时才返回 extern 裸名， + # 否则让本地定义走正常 mangle 路径（本地 def 遮蔽 import 语义）。 + return name + # main 函数作为程序入口点，永远不混淆 + if name == 'main': + return name + # Try class_sha1_map first for class methods (e.g., 'md5.__init__' -> class 'md5') + class_sha1_map: dict[str, str] = self.class_sha1_map + if class_sha1_map: + if '.' in name: + base_class: str = name.split('.')[0] + if base_class in class_sha1_map: + sha1: str = class_sha1_map[base_class] + return f"{sha1}.{name}" + elif name in class_sha1_map: + sha1: str = class_sha1_map[name] + return f"{sha1}.{name}" + # If module_name is already a SHA1 hash (16 hex chars), use it directly + if module_name and len(module_name) == 16 and all(c in '0123456789abcdef' for c in module_name): + return f"{module_name}.{name}" + if module_name and module_name in self.ModuleSha1Map: + sha1: str = self.ModuleSha1Map[module_name] + return f"{sha1}.{name}" + # Try parent package name (e.g., 'hashlib' -> 'hashlib.__init__') + if module_name: + init_name: str = f"{module_name}.__init__" + if init_name in self.ModuleSha1Map: + sha1: str = self.ModuleSha1Map[init_name] + return f"{sha1}.{name}" + if self._skip_mangle(name): + return name + if self.module_sha1: + return f"{self.module_sha1}.{name}" + return name + + def _get_source_line(self, lineno: int) -> str | None: + if 0 < lineno <= len(self._current_source_lines): + return self._current_source_lines[lineno - 1].rstrip('\r\n') + return None + + def _get_node_info(self) -> str: + info: str = "" + if self._current_node_info: + info = f" [{self._current_node_info}]" + if self._current_source_file: + info += f" in {self._current_source_file}" + lineno: int = self._current_lineno + if lineno > 0: + info += f", line {lineno}:" + source_line: str | None = self._get_source_line(lineno) + if source_line: + info += f"\n → {source_line}" + return info + + # ========== 统一工具方法 ========== + + def _create_string_global(self, s: str, name: str | None = None) -> ir.Constant: + """创建字符串全局变量并返回 i8* 指针。 + + 统一字符串全局变量创建模式：编码为 utf-8 + '\\x00' 终止符， + 创建 internal linkage 的 GlobalVariable，返回 i8* 指针常量。 + """ + str_val: str = s + '\x00' + str_bytes: bytes = str_val.encode('utf-8') + arr_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(str_bytes)) + gv_name: str = name or f"str_const_{self.string_const_counter}" + self.string_const_counter += 1 + gv: ir.GlobalVariable = ir.GlobalVariable(self.module, arr_type, name=gv_name) + gv.initializer = ir.Constant(arr_type, bytearray(str_bytes)) + gv.linkage = 'internal' + return ir.Constant(ir.PointerType(ir.IntType(8)), gv.get_reference()) + + def _create_doc_global(self, symbol_name: str, docstring: str | None) -> None: + """为符号创建 __doc__ 全局常量。 + + - docstring 为 None 或空字符串时，不创建常量（访问时内联空指针） + - slice_level != 0（优化开启）时，内容视为空 + - 非空时创建 internal linkage、global_constant=True 的只读字符串常量 + - 常量名格式：{module_sha1}.{symbol_name}.__doc__ + """ + slice_level: int = getattr(self._Trans, 'SliceLevel', 3) + if slice_level != 0: + docstring = None + if docstring is None or docstring == '': + self._doc_strings[symbol_name] = None + return + self._doc_strings[symbol_name] = docstring + mangled: str = self._mangle_name(f"{symbol_name}.__doc__") + self._create_string_global(docstring, name=mangled) + gv: ir.GlobalVariable | None = self.module.globals.get(mangled) + if gv is not None: + gv.global_constant = True + + def _get_doc_ptr(self, symbol_name: str) -> ir.Value | None: + """获取符号 __doc__ 的 i8* 指针。 + + - 若符号未注册 __doc__，返回 None（调用方回退到其他解析） + - 若 __doc__ 为空（无 docstring 或优化剥离），返回 i8* null 内联空指针 + - 若 __doc__ 非空，返回指向字符串全局常量的 i8* 指针 + - 跨模块 docstring（在 _doc_strings 中有文本但 module.globals 无对应常量）首次访问时创建本地 internal linkage 只读副本 + """ + if symbol_name not in self._doc_strings: + return None + docstring: str | None = self._doc_strings[symbol_name] + if docstring is None or docstring == '': + return ir.Constant(ir.PointerType(ir.IntType(8)), None) + mangled: str = self._mangle_name(f"{symbol_name}.__doc__") + gv: ir.GlobalVariable | None = self.module.globals.get(mangled) + if gv is None: + # 跨模块 docstring：创建本地 internal linkage 只读字符串常量副本 + self._create_string_global(docstring, name=mangled) + gv = self.module.globals.get(mangled) + if gv is None: + return ir.Constant(ir.PointerType(ir.IntType(8)), None) + gv.global_constant = True + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + return self.builder.gep(gv, [zero, zero], name=f"{symbol_name}_doc_ptr") + + def _apply_bswap_if_big(self, val: ir.Value, byte_order: str, name: str) -> ir.Value: + """大端字节序时对整数应用 bswap，否则原值返回。 + + 仅对 16/32/64 位整数生效，统一 bswap 逻辑。 + """ + if byte_order == 'big' and isinstance(val.type, ir.IntType) and val.type.width in (16, 32, 64): + return self.builder.bswap(val, name=name) + return val + + def _zero_value_for_type(self, llvm_type: ir.Type) -> ir.Constant | None: + """为指定 LLVM 类型生成零值常量。 + + 合并原 HandlesAssign._zero_value 与 StructGen._zero_value_for_type 的逻辑。 + 返回 ir.Constant 或 None（无法生成零值的类型，如 opaque struct）。 + """ + if isinstance(llvm_type, ir.IntType): + return ir.Constant(llvm_type, 0) + elif isinstance(llvm_type, (ir.FloatType, ir.DoubleType)): + return ir.Constant(llvm_type, 0.0) + elif isinstance(llvm_type, ir.PointerType): + return ir.Constant(llvm_type, None) + elif isinstance(llvm_type, ir.ArrayType): + elem_zv: ir.Constant | None = self._zero_value_for_type(llvm_type.element) + if elem_zv is None: + return None + return ir.Constant(llvm_type, [elem_zv] * llvm_type.count) + elif isinstance(llvm_type, ir.BaseStructType): + if llvm_type.elements is not None and len(llvm_type.elements) > 0: + elem_zvs: list[ir.Constant | None] = [self._zero_value_for_type(m) for m in llvm_type.elements] + if any(zv is None for zv in elem_zvs): + return None + return ir.Constant(llvm_type, elem_zvs) + return None + return None \ No newline at end of file diff --git a/lib/core/LLVMCG/ExprGen.py b/lib/core/LLVMCG/ExprGen.py new file mode 100644 index 0000000..e8eb6a1 --- /dev/null +++ b/lib/core/LLVMCG/ExprGen.py @@ -0,0 +1,368 @@ +from __future__ import annotations +from typing import Any +import logging +import llvmlite.ir as ir +from lib.constants.config import mode as _config_mode + + +class ExprGenMixin: + def emit_constant(self, value: Any, type_name: str = 'int') -> ir.Value: + if type_name == 'int': + return ir.Constant(ir.IntType(32), int(value)) + elif type_name == 'uint64_t' or type_name == 'unsigned long long': + return ir.Constant(ir.IntType(64), int(value)) + elif type_name == 'uint32_t' or type_name == 'unsigned int': + return ir.Constant(ir.IntType(32), int(value)) + elif type_name == 'uint16_t' or type_name == 'unsigned short': + return ir.Constant(ir.IntType(16), int(value)) + elif type_name == 'uint8_t' or type_name == 'unsigned char': + return ir.Constant(ir.IntType(8), int(value)) + elif type_name == 'int64_t' or type_name == 'long long': + return ir.Constant(ir.IntType(64), int(value)) + elif type_name == 'int32_t' or type_name == 'int': + return ir.Constant(ir.IntType(32), int(value)) + elif type_name == 'int16_t' or type_name == 'short': + return ir.Constant(ir.IntType(16), int(value)) + elif type_name == 'int8_t' or type_name == 'char': + return ir.Constant(ir.IntType(8), int(value)) + elif type_name == 'float' or type_name == 'float32_t' or type_name == 'FLOAT32': + return ir.Constant(ir.FloatType(), float(value)) + elif type_name == 'double' or type_name == 'float64_t' or type_name == 'FLOAT64': + return ir.Constant(ir.DoubleType(), float(value)) + elif type_name == 'float16_t' or type_name == 'FLOAT16': + return ir.Constant(ir.IntType(16), int(value)) + elif type_name == 'float8_t' or type_name == 'FLOAT8': + return ir.Constant(ir.IntType(8), int(value)) + elif type_name == 'float128_t' or type_name == 'FLOAT128': + return ir.Constant(ir.IntType(128), int(value)) + elif type_name == 'string': + encoded: bytes = value.encode('utf-8') + b'\x00' + str_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(encoded)) + str_const: ir.Constant = ir.Constant(str_type, bytearray(encoded)) + global_var: ir.GlobalVariable = ir.GlobalVariable(self.module, str_type, name=f"str_const_{self.string_const_counter}") + self.string_const_counter += 1 + global_var.initializer = str_const + global_var.linkage = 'internal' + return self.builder.bitcast(global_var, ir.PointerType(ir.IntType(8)), name="str") + return ir.Constant(ir.IntType(32), int(value)) + + def emit_binary_op(self, op: str, left: ir.Value, right: ir.Value, is_unsigned: bool = False) -> ir.Value: + if isinstance(left.type, ir.PointerType) and isinstance(right.type, ir.IntType): + # 数组指针需要先 decay 为元素指针，否则 GEP 会以整个数组为步长 + # 例如 [64 x i8]* + 1 会前进 64 字节而非 1 字节 + if isinstance(left.type.pointee, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + left = self.builder.gep(left, [zero, zero], name="array_decay") + right = self.builder.zext(right, ir.IntType(64), name="int2ptrint") + if op == '+' or op == 'Add': + return self.builder.gep(left, [right], name="ptr_add") + elif op == '-' or op == 'Sub': + return self.builder.gep(left, [self.builder.neg(right, name="neg")], name="ptr_sub") + if isinstance(left.type, ir.IntType) and isinstance(right.type, ir.PointerType): + if isinstance(right.type.pointee, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + right = self.builder.gep(right, [zero, zero], name="array_decay_rev") + left = self.builder.zext(left, ir.IntType(64), name="int2ptrint2") + if op == '+' or op == 'Add': + return self.builder.gep(right, [left], name="ptr_add_rev") + is_float: bool = isinstance(left.type, (ir.FloatType, ir.DoubleType)) or \ + isinstance(right.type, (ir.FloatType, ir.DoubleType)) + if is_float: + ftype: ir.Type + if isinstance(left.type, (ir.FloatType, ir.DoubleType)): + ftype = left.type + elif isinstance(right.type, (ir.FloatType, ir.DoubleType)): + ftype = right.type + else: + ftype = ir.DoubleType() + if not isinstance(left.type, (ir.FloatType, ir.DoubleType)): + left = self._to_float(left, ftype) + elif left.type != ftype: + if isinstance(left.type, ir.FloatType) and isinstance(ftype, ir.DoubleType): + left = self.builder.fpext(left, ftype, name="fpext_l") + else: + left = self.builder.fptrunc(left, ftype, name="fptrunc_l") + if not isinstance(right.type, (ir.FloatType, ir.DoubleType)): + right = self._to_float(right, ftype) + elif right.type != ftype: + if isinstance(right.type, ir.FloatType) and isinstance(ftype, ir.DoubleType): + right = self.builder.fpext(right, ftype, name="fpext_r") + else: + right = self.builder.fptrunc(right, ftype, name="fptrunc_r") + float_ops: dict[str, Any] = { + '+': self.builder.fadd, 'Add': self.builder.fadd, + '-': self.builder.fsub, 'Sub': self.builder.fsub, + '*': self.builder.fmul, 'Mult': self.builder.fmul, + '/': self.builder.fdiv, 'Div': self.builder.fdiv, + '%': self.builder.frem, 'Mod': self.builder.frem, + } + float_cmp_ops: dict[str, str] = { + '==': '==', 'Eq': '==', '!=': '!=', 'NotEq': '!=', + '<': '<', 'Lt': '<', '<=': '<=', 'LtE': '<=', + '>': '>', 'Gt': '>', '>=': '>=', 'GtE': '>=', + } + if op in float_ops: + return float_ops[op](left, right, name=op) + if op in float_cmp_ops: + return self.builder.fcmp_ordered(float_cmp_ops[op], left, right, name=op) + ops: dict[str, Any] + if is_unsigned: + ops = { + '+': self.builder.add, 'Add': self.builder.add, + '-': self.builder.sub, 'Sub': self.builder.sub, + '*': self.builder.mul, 'Mult': self.builder.mul, + '/': self.builder.udiv, 'Div': self.builder.udiv, + '%': self.builder.urem, 'Mod': self.builder.urem, + '//': self.builder.udiv, 'FloorDiv': self.builder.udiv, + } + else: + ops = { + '+': self.builder.add, 'Add': self.builder.add, + '-': self.builder.sub, 'Sub': self.builder.sub, + '*': self.builder.mul, 'Mult': self.builder.mul, + '/': self.builder.sdiv, 'Div': self.builder.sdiv, + '%': self.builder.srem, 'Mod': self.builder.srem, + '//': self.builder.sdiv, 'FloorDiv': self.builder.sdiv, + } + cmp_ops: dict[str, str] = { + '==': '==', 'Eq': '==', '!=': '!=', 'NotEq': '!=', + '<': '<', 'Lt': '<', '<=': '<=', 'LtE': '<=', + '>': '>', 'Gt': '>', '>=': '>=', 'GtE': '>=', + } + logic_ops: dict[str, Any] = { + '&&': self.builder.and_, 'And': self.builder.and_, + '||': self.builder.or_, 'Or': self.builder.or_, + } + if op in ('>>', 'RShift') and not is_unsigned: + if isinstance(left.type, ir.IntType): + if left.type.width == 8: + is_unsigned = True + elif left.type.width == 16: + is_unsigned = True + elif left.type.width == 32: + if self._last_var_name: + is_unsigned = self._is_var_unsigned(self._last_var_name) + elif left.type.width == 64: + if self._last_var_name: + is_unsigned = self._is_var_unsigned(self._last_var_name) + shift_ops: dict[str, Any] = { + '>>': self.builder.lshr if is_unsigned else self.builder.ashr, + 'RShift': self.builder.lshr if is_unsigned else self.builder.ashr, + '<<': self.builder.shl, 'LShift': self.builder.shl, + } + bit_ops: dict[str, Any] = { + '&': self.builder.and_, + '|': self.builder.or_, + '^': self.builder.xor, + } + if op == '**' or op == 'Pow': + if isinstance(left, ir.Constant) and isinstance(right, ir.Constant): + try: + lv: Any = left.constant + rv: Any = right.constant + if isinstance(lv, int) and isinstance(rv, int) and rv >= 0: + result: int = lv ** rv + return ir.Constant(left.type, result) + except Exception as _e: + if _config_mode == "strict": + logging.warning(f"异常被忽略: {_e}") + pass + ftype: ir.Type + if isinstance(left.type, (ir.FloatType, ir.DoubleType)): + ftype = left.type + elif isinstance(right.type, (ir.FloatType, ir.DoubleType)): + ftype = right.type + else: + ftype = ir.DoubleType() + lf: ir.Value = self._to_float(left, ftype) + rf: ir.Value = self._to_float(right, ftype) + powf: ir.Function | None = self.module.globals.get('llvm.pow.f64') + if not powf: + fnty: ir.FunctionType = ir.FunctionType(ir.DoubleType(), [ir.DoubleType(), ir.DoubleType()]) + powf = ir.Function(self.module, fnty, name='llvm.pow.f64') + if lf.type != ir.DoubleType(): + lf = self.builder.fpext(lf, ir.DoubleType(), name="ext_f64") + if rf.type != ir.DoubleType(): + rf = self.builder.fpext(rf, ir.DoubleType(), name="ext_f64") + result: ir.Value = self.builder.call(powf, [lf, rf], name="pow") + both_int: bool = isinstance(left.type, ir.IntType) and isinstance(right.type, ir.IntType) + right_is_pos_int_const: bool = isinstance(right, ir.Constant) and isinstance(right.constant, int) and right.constant >= 0 + if both_int and right_is_pos_int_const: + return self.builder.fptosi(result, left.type, name="pow2int") + if isinstance(left.type, ir.FloatType): + return self.builder.fptrunc(result, ir.FloatType(), name="pow2f32") + return result + if op in ops: + try: + return ops[op](left, right, name=op) + except Exception as e: + raise ValueError(f"{e}{self._get_node_info()}") + if op in shift_ops: + try: + return shift_ops[op](left, right, name=op) + except Exception as e: + raise ValueError(f"{e}{self._get_node_info()}") + if op in bit_ops: + try: + return bit_ops[op](left, right, name=op) + except Exception as e: + raise ValueError(f"{e}{self._get_node_info()}") + if op in cmp_ops: + try: + if is_unsigned: + return self.builder.icmp_unsigned(cmp_ops[op], left, right, name=op) + return self.builder.icmp_signed(cmp_ops[op], left, right, name=op) + except Exception as e: + raise ValueError(f"{e}{self._get_node_info()}") + if op in logic_ops: + try: + return logic_ops[op](left, right, name=op) + except Exception as e: + raise ValueError(f"{e}{self._get_node_info()}") + return left + + def _to_float(self, val: ir.Value, ftype: ir.Type | None = None) -> ir.Value: + if ftype is None: + ftype = ir.DoubleType() + if isinstance(val.type, (ir.FloatType, ir.DoubleType)): + if val.type == ftype: + return val + if isinstance(val.type, ir.FloatType) and isinstance(ftype, ir.DoubleType): + return self.builder.fpext(val, ftype, name="fpext") + return self.builder.fptrunc(val, ftype, name="fptrunc") + if isinstance(val.type, ir.IntType): + if val.type.width == 1: + val = self.builder.zext(val, ir.IntType(32), name="bool2i32") + return self.builder.sitofp(val, ftype, name="int2float") + return val + + def emit_return(self, value: ir.Value | None = None) -> None: + if not self.builder or self.builder.block.is_terminated: + return + if value is not None: + self._unregister_local_heap_ptr(value) + self._emit_local_heap_frees() + if self._variadic_info and self._variadic_info.get('va_start_called'): + va_list_ptr: ir.Value = self._variadic_info['va_list_ptr'] + self.emit_va_end(va_list_ptr) + if value is None: + if self.func and hasattr(self.func, 'ftype'): + ret_type: ir.Type = self.func.ftype.return_type + if isinstance(ret_type, ir.IntType): + self.builder.ret(ir.Constant(ret_type, 0)) + elif isinstance(ret_type, ir.PointerType): + self.builder.ret(ir.Constant(ret_type, None)) + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)): + self.builder.ret(ir.Constant(ret_type, 0.0)) + elif isinstance(ret_type, ir.BaseStructType): + if ret_type.elements: + zero_val: ir.Constant = ir.Constant(ret_type, [ir.Constant(et, None) if isinstance(et, (ir.PointerType, ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)) else ir.Constant(et, 0) if isinstance(et, (ir.IntType, ir.FloatType, ir.DoubleType)) else ir.Constant(et, ir.Undefined) for et in ret_type.elements]) + else: + zero_val = ir.Constant(ret_type, None) + self.builder.ret(zero_val) + elif isinstance(ret_type, ir.ArrayType): + self.builder.ret(ir.Constant(ret_type, None)) + else: + self.builder.ret_void() + else: + self.builder.ret_void() + else: + if self.func and hasattr(self.func, 'ftype'): + ret_type: ir.Type = self.func.ftype.return_type + if ret_type != value.type: + if isinstance(ret_type, ir.PointerType) and isinstance(value.type, ir.PointerType): + value = self.builder.bitcast(value, ret_type, name="ret_cast") + elif isinstance(value.type, ir.PointerType) and not isinstance(ret_type, ir.PointerType): + if isinstance(value.type.pointee, ir.IntType) and isinstance(ret_type, ir.IntType) and value.type.pointee.width == ret_type.width: + value = self._load(value, name="ret_Load") + elif isinstance(ret_type, ir.IntType): + value = self.builder.ptrtoint(value, ret_type, name="ret_ptrtoint") + elif isinstance(value.type.pointee, ret_type.__class__) and not isinstance(value.type.pointee, ir.IntType): + value = self._load(value, name="ret_Load") + elif isinstance(value.type.pointee, ir.PointerType): + Loaded: ir.Value = self._load(value, name="ret_deref") + if Loaded.type == ret_type: + value = Loaded + elif isinstance(ret_type, ir.IntType): + value = self.builder.ptrtoint(Loaded, ret_type, name="ret_ptrtoint") + elif isinstance(ret_type, ir.IntType) and isinstance(value.type, ir.IntType): + if value.type.width < ret_type.width: + if value.type.width == 1: + # i1 是布尔值（来自 icmp/fcmp/__contains__），无符号语义，必须 zext + value = self.builder.zext(value, ret_type, name="ret_zext_bool") + else: + is_unsigned: bool = id(value) in self._unsigned_results + if is_unsigned: + value = self.builder.zext(value, ret_type, name="ret_zext") + else: + value = self.builder.sext(value, ret_type, name="ret_sext") + elif value.type.width > ret_type.width: + value = self.builder.trunc(value, ret_type, name="ret_trunc") + elif isinstance(ret_type, ir.PointerType) and isinstance(value.type, ir.IntType): + value = self.builder.inttoptr(value, ret_type, name="ret_inttoptr") + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)) and isinstance(value.type, ir.IntType): + value = self.builder.sitofp(value, ret_type, name="ret_int2float") + elif isinstance(ret_type, ir.IntType) and isinstance(value.type, (ir.FloatType, ir.DoubleType)): + value = self.builder.fptosi(value, ret_type, name="ret_float2int") + elif isinstance(ret_type, (ir.FloatType, ir.DoubleType)) and isinstance(value.type, (ir.FloatType, ir.DoubleType)): + if ret_type != value.type: + if isinstance(value.type, ir.FloatType) and isinstance(ret_type, ir.DoubleType): + value = self.builder.fpext(value, ret_type, name="ret_fpext") + elif isinstance(value.type, ir.DoubleType) and isinstance(ret_type, ir.FloatType): + value = self.builder.fptrunc(value, ret_type, name="ret_fptrunc") + self.builder.ret(value) + + def _emit_printf(self, args: list[ir.Value], unsigned_flags: list[bool] | None = None, sep: str = " ", end: str = "\n") -> ir.Value | None: + if not args: + return None + if unsigned_flags is None: + unsigned_flags = [] + format_str: str = "" + cast_args: list[ir.Value] = [] + for i, arg in enumerate(args): + is_u: bool = i < len(unsigned_flags) and unsigned_flags[i] + if isinstance(arg.type, ir.IntType): + if arg.type.width == 1: + # i1 是布尔值（来自 icmp/fcmp/__contains__），zext 到 i32 后用 %d + format_str += "%d" + ext: ir.Value = self.builder.zext(arg, ir.IntType(32), name="bool2int") + cast_args.append(ext) + elif arg.type.width == 8: + format_str += "%c" + ext: ir.Value = self.builder.zext(arg, ir.IntType(32), name="char2int") + cast_args.append(ext) + elif arg.type.width == 64: + format_str += "%llu" if is_u else "%lld" + cast_args.append(arg) + else: + format_str += "%u" if is_u else "%d" + cast_args.append(arg) + elif isinstance(arg.type, (ir.FloatType, ir.DoubleType)): + format_str += "%f" + cast_args.append(arg) + elif isinstance(arg.type, ir.PointerType): + if isinstance(arg.type.pointee, ir.IntType) and arg.type.pointee.width == 8: + format_str += "%s" + cast_args.append(arg) + else: + format_str += "%d" + int_val: ir.Value = self.builder.ptrtoint(arg, ir.IntType(64), name="ptr2int") + trunc_val: ir.Value = self.builder.trunc(int_val, ir.IntType(32), name="trunc") + cast_args.append(trunc_val) + else: + format_str += "%d" + cast_args.append(arg) + if sep is not None and i < len(args) - 1: + format_str += sep + if end is not None: + format_str += end + string_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(format_str.encode('utf-8')) + 1) + string_const: ir.Constant = ir.Constant(string_type, bytearray(format_str + '\x00', 'utf-8')) + global_var: ir.GlobalVariable = ir.GlobalVariable(self.module, string_type, name=f"str_const_{self.string_const_counter}") + self.string_const_counter += 1 + global_var.initializer = string_const + global_var.linkage = 'internal' + format_ptr: ir.Value = self.builder.bitcast(global_var, ir.PointerType(ir.IntType(8))) + call_args: list[ir.Value] = [format_ptr] + cast_args + printf_func: ir.Function | None = self.get_or_declare_c_func('printf', ir.FunctionType(ir.IntType(32), [ir.PointerType(ir.IntType(8))], var_arg=True)) + return self.builder.call(printf_func, call_args, name="call_printf") \ No newline at end of file diff --git a/lib/core/LLVMCG/FuncGen.py b/lib/core/LLVMCG/FuncGen.py new file mode 100644 index 0000000..dc743c6 --- /dev/null +++ b/lib/core/LLVMCG/FuncGen.py @@ -0,0 +1,524 @@ +from __future__ import annotations +from typing import Any +import ast +import os +import re +import llvmlite.ir as ir +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.VLogger import get_logger as _vlog +from lib.constants.config import mode as _config_mode + + +class FuncGenMixin: + + def setup_from_symbol_table(self, SymbolTable: Any) -> None: + self.SymbolTable: Any = SymbolTable + for name, info in SymbolTable.items(): + if isinstance(info, CTypeInfo): + # CTypeInfo 对象：struct/union 由 StructGen 处理，此处只处理 dict 格式 + continue + if not isinstance(info, dict): + if hasattr(info, 'get'): + info = dict(info) if hasattr(info, '__iter__') else {} + else: + continue + TypeKind: str = info.get('type', '') + if TypeKind == 'struct': + ClassName: str = name + self.class_methods[ClassName] = [] + self.class_members[ClassName] = [] + members: dict[str, Any] = info.get('members', {}) + if members: + for MemberName, MemberInfo in members.items(): + if isinstance(MemberInfo, dict): + MemberType: ir.Type = self._type_str_to_llvm(MemberInfo.get('type', 'int'), MemberInfo.get('IsPtr', False)) + self.class_members[ClassName].append((MemberName, MemberType)) + elif isinstance(MemberInfo, CTypeInfo): + MemberType: ir.Type = self._ctype_to_llvm(MemberInfo) + if MemberType is None: + MemberType = self._type_str_to_llvm(MemberInfo.ToString(), MemberInfo.IsPtr) + self.class_members[ClassName].append((MemberName, MemberType)) + self._generate_structs() + self._create_Vtable_globals() + + def register_method(self, ClassName: str, MethodName: str) -> None: + if ClassName not in self.class_methods: + self.class_methods[ClassName] = [] + if ClassName not in self.class_members: + self.class_members[ClassName] = [] + self._generate_structs() + self._create_Vtable_globals() + # 构造函数（__new__/__init__/__before_init__）不放入 vtable， + # 避免子类与基类 vtable 大小不一致导致虚分派索引偏移 + _short_name: str = MethodName.split('.')[-1] if '.' in MethodName else MethodName + if _short_name in ('__new__', '__init__', '__before_init__'): + return + if MethodName not in self.class_methods[ClassName]: + self.class_methods[ClassName].append(MethodName) + + def _apply_auto_addr(self, func_name: str, args: list[ir.Value]) -> list[ir.Value]: + """对函数参数应用自动取地址逻辑（t.CNeedPtr / t.CAutoPtr）。 + + 在调用前对标记为 auto-addr 的参数自动取地址： + - 'need' (t.CNeedPtr): 仅对非指针值取地址（alloca + store + get pointer） + - 'always' (t.CAutoPtr): 无条件取地址 + """ + modes: list[str | None] | None = self._auto_addr_params.get(func_name) + if not modes: + return args + new_args: list[ir.Value] = list(args) + for i, mode in enumerate(modes): + if i >= len(new_args) or mode is None: + continue + arg: ir.Value = new_args[i] + is_ptr: bool = isinstance(arg.type, ir.PointerType) + if mode == 'always' or (mode == 'need' and not is_ptr): + # alloca + store + get pointer + slot: ir.Value = self.builder.alloca(arg.type) + self.builder.store(arg, slot) + new_args[i] = slot + return new_args + + def _adjust_args(self, args: list[ir.Value], func: ir.Function) -> list[ir.Value]: + adjusted: list[ir.Value] = [] + for i, (arg, param) in enumerate(zip(args, func.args)): + if arg.type != param.type: + if isinstance(arg.type, ir.PointerType) and isinstance(param.type, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + if arg.type.pointee == param.type or (hasattr(arg.type.pointee, 'name') and hasattr(param.type, 'name') and arg.type.pointee.name == param.type.name): + # C 语言中数组参数退化为指针，不做 load 而是退化为元素指针 + if isinstance(param.type, ir.ArrayType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + elem_ptr: ir.Value = self.builder.gep(arg, [zero, zero], name=f"array_decay_arg_{i}") + adjusted.append(elem_ptr) + else: + adjusted.append(self._load(arg, name=f"Load_arg_{i}")) + continue + try: + if isinstance(arg.type, ir.IntType) and isinstance(param.type, ir.PointerType): + if isinstance(param.type.pointee, ir.IntType) and param.type.pointee.width == 8 and arg.type.width == 8: + # 当参数是 i8（字符值）而函数期望 i8*（字符串指针）时 + # 为字符值分配内存，创建 null-terminated 字符串 + tmp: ir.AllocaInstr = self._alloca(ir.ArrayType(ir.IntType(8), 2), name=f"char2str_arg_{i}") + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + char_ptr: ir.Value = self.builder.gep(tmp, [zero, zero], name=f"char2str_ptr_{i}") + self._store(arg, char_ptr) + null_ptr: ir.Value = self.builder.gep(tmp, [zero, ir.Constant(ir.IntType(32), 1)], name=f"char2str_null_{i}") + self._store(ir.Constant(ir.IntType(8), 0), null_ptr) + adjusted.append(char_ptr) + else: + if isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.IntType): + # 当参数是指针而函数期望整数时，使用 ptrtoint + adjusted.append(self.builder.ptrtoint(arg, param.type, name=f"ptrtoint_arg_{i}")) + else: + if arg.type.width < 32: + arg = self.builder.zext(arg, ir.IntType(32), name=f"zext_arg_{i}") + adjusted.append(self.builder.inttoptr(arg, param.type, name=f"inttoptr_arg_{i}")) + elif isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.IntType): + # 当参数是指针而函数期望整数时，使用 ptrtoint + adjusted.append(self.builder.ptrtoint(arg, param.type, name=f"ptrtoint_arg_{i}")) + elif isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.PointerType): + if isinstance(arg.type.pointee, ir.PointerType) and isinstance(param.type.pointee, ir.IntType): + Loaded: ir.Value = self._load(arg, name=f"Load_arg_{i}") + adjusted.append(Loaded) + elif isinstance(arg.type.pointee, ir.ArrayType) and isinstance(param.type.pointee, (ir.IntType, ir.FloatType, ir.DoubleType)): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + elem_ptr: ir.Value = self.builder.gep(arg, [zero, zero], name=f"array_decay_arg_{i}") + if elem_ptr.type != param.type: + adjusted.append(self.builder.bitcast(elem_ptr, param.type, name=f"cast_arg_{i}")) + else: + adjusted.append(elem_ptr) + elif isinstance(arg.type.pointee, ir.ArrayType) and isinstance(param.type.pointee, ir.PointerType): + zero: ir.Constant = ir.Constant(ir.IntType(32), 0) + elem_ptr: ir.Value = self.builder.gep(arg, [zero, zero], name=f"array_decay_arg_{i}") + if elem_ptr.type != param.type: + adjusted.append(self.builder.bitcast(elem_ptr, param.type, name=f"cast_arg_{i}")) + else: + adjusted.append(elem_ptr) + else: + adjusted.append(self.builder.bitcast(arg, param.type, name=f"cast_arg_{i}")) + else: + if isinstance(arg.type, ir.PointerType) and isinstance(param.type, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + if arg.type.pointee == param.type or (hasattr(arg.type.pointee, 'name') and hasattr(param.type, 'name') and arg.type.pointee.name == param.type.name): + adjusted.append(self._load(arg, name=f"Load_arg_{i}")) + elif isinstance(arg.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + tmp: ir.AllocaInstr = self._alloca(param.type, name=f"temp_arg_{i}") + Loaded: ir.Value = self.builder.bitcast(arg, ir.PointerType(param.type), name=f"cast_arg_{i}") + Loaded_val: ir.Value = self._load(Loaded, name=f"Load_arg_{i}") + self._store(Loaded_val, tmp) + adjusted.append(self._load(tmp, name=f"reLoad_arg_{i}")) + else: + tmp: ir.AllocaInstr = self._alloca(param.type, name=f"temp_arg_{i}") + cast_ptr: ir.Value = self.builder.bitcast(arg, ir.PointerType(param.type), name=f"cast_arg_{i}") + Loaded_val: ir.Value = self._load(cast_ptr, name=f"Load_arg_{i}") + self._store(Loaded_val, tmp) + adjusted.append(self._load(tmp, name=f"reLoad_arg_{i}")) + elif isinstance(arg.type, ir.PointerType) and arg.type.pointee == param.type: + adjusted.append(self._load(arg, name=f"Load_arg_{i}")) + elif isinstance(param.type, ir.PointerType) and param.type.pointee == arg.type: + tmp: ir.AllocaInstr = self._alloca(arg.type, name=f"temp_arg_{i}") + self._store(arg, tmp) + adjusted.append(tmp) + elif isinstance(arg.type, ir.IntType) and isinstance(param.type, ir.IntType): + if arg.type.width < param.type.width: + # 整数扩展：使用 sext（符号扩展）以正确处理负数 + adjusted.append(self.builder.sext(arg, param.type, name=f"sext_arg_{i}")) + else: + adjusted.append(self.builder.trunc(arg, param.type, name=f"trunc_arg_{i}")) + elif isinstance(arg.type, (ir.FloatType, ir.DoubleType)) and isinstance(param.type, (ir.FloatType, ir.DoubleType)): + if isinstance(arg.type, ir.DoubleType) and isinstance(param.type, ir.FloatType): + adjusted.append(self.builder.fptrunc(arg, param.type, name=f"fptrunc_arg_{i}")) + elif isinstance(arg.type, ir.FloatType) and isinstance(param.type, ir.DoubleType): + adjusted.append(self.builder.fpext(arg, param.type, name=f"fpext_arg_{i}")) + else: + adjusted.append(arg) + elif isinstance(arg.type, (ir.FloatType, ir.DoubleType)) and isinstance(param.type, ir.IntType): + adjusted.append(self.builder.fptosi(arg, param.type, name=f"fptosi_arg_{i}")) + elif isinstance(arg.type, ir.IntType) and isinstance(param.type, (ir.FloatType, ir.DoubleType)): + adjusted.append(self.builder.sitofp(arg, param.type, name=f"sitofp_arg_{i}")) + elif isinstance(arg.type, ir.IntType) and isinstance(param.type, ir.PointerType): + # 整数转指针：inttoptr + if arg.type.width < 64: + i64_val: ir.Value = self.builder.sext(arg, ir.IntType(64), name=f"sext_arg_{i}") + adjusted.append(self.builder.inttoptr(i64_val, param.type, name=f"inttoptr_arg_{i}")) + else: + adjusted.append(self.builder.inttoptr(arg, param.type, name=f"inttoptr_arg_{i}")) + elif isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.IntType): + # 指针转整数：ptrtoint + i64_val: ir.Value = self.builder.ptrtoint(arg, ir.IntType(64), name=f"ptrtoint_arg_{i}") + if i64_val.type.width > param.type.width: + adjusted.append(self.builder.trunc(i64_val, param.type, name=f"trunc_arg_{i}")) + elif i64_val.type.width < param.type.width: + adjusted.append(self.builder.sext(i64_val, param.type, name=f"sext_arg_{i}")) + else: + adjusted.append(i64_val) + else: + adjusted.append(self.builder.bitcast(arg, param.type, name=f"cast_arg_{i}")) + except Exception: # 参数类型调整失败时使用简化回退逻辑 + if isinstance(arg.type, ir.PointerType) and isinstance(param.type, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + if arg.type.pointee == param.type or (hasattr(arg.type.pointee, 'name') and hasattr(param.type, 'name') and arg.type.pointee.name == param.type.name): + adjusted.append(self._load(arg, name=f"Load_arg_{i}")) + else: + adjusted.append(self.builder.bitcast(arg, param.type, name=f"cast_arg_{i}")) + elif isinstance(arg.type, ir.PointerType) and arg.type.pointee == param.type: + adjusted.append(self._load(arg, name=f"Load_arg_{i}")) + elif isinstance(param.type, ir.PointerType) and param.type.pointee == arg.type: + tmp: ir.AllocaInstr = self._alloca(arg.type, name=f"temp_arg_{i}") + self._store(arg, tmp) + adjusted.append(tmp) + elif isinstance(arg.type, ir.PointerType) and isinstance(param.type, ir.PointerType): + # 任意指针到指针的类型转换（如 struct* -> i8*） + adjusted.append(self.builder.bitcast(arg, param.type, name=f"cast_arg_{i}")) + else: + adjusted.append(arg) + else: + adjusted.append(arg) + while len(adjusted) < len(func.args): + missing_idx: int = len(adjusted) + param: ir.Argument = func.args[missing_idx] + param_type: ir.Type = param.type if hasattr(param, 'type') else param + if isinstance(param_type, ir.PointerType): + adjusted.append(ir.Constant(param_type, None)) + elif isinstance(param_type, ir.IntType): + adjusted.append(ir.Constant(param_type, 0)) + elif isinstance(param_type, (ir.FloatType, ir.DoubleType)): + adjusted.append(ir.Constant(param_type, 0.0)) + else: + adjusted.append(ir.Constant(ir.IntType(32), 0)) + if func.type.pointee.var_arg: + for i in range(len(func.args), len(args)): + arg: ir.Value = args[i] + adjusted.append(arg) + return adjusted + + def _get_member_offset(self, field_name: str, ClassName: str | None = None) -> int | None: + # 治本修复：同名类 sha1 冲突检测。 + # 当 class_sha1_map[ClassName] 的 sha1 与 Gen.structs[ClassName] 的 sha1 不一致时， + # class_members[ClassName] 可能存储了错误模块的同名类成员。 + # 此处检测冲突，加载正确模块的 class_members 并以 sha1 前缀全名缓存。 + _ResolvedClass: str = ClassName if ClassName else '' + if ClassName: + _ClassSha1Map: dict[str, str] = getattr(self, 'class_sha1_map', {}) + _ExpectedSha1: str | None = _ClassSha1Map.get(ClassName) + if _ExpectedSha1: + _ExistingStruct = self.structs.get(ClassName) + if isinstance(_ExistingStruct, ir.IdentifiedStructType): + _ExistingSha1: str = self._extract_struct_sha1(_ExistingStruct) + if _ExistingSha1 and _ExistingSha1 != _ExpectedSha1: + # sha1 冲突：使用全名查找 class_members + _FullKey: str = f"{_ExpectedSha1}.{ClassName}" + if _FullKey in self.class_members and self.class_members[_FullKey]: + _ResolvedClass = _FullKey + else: + # 从正确 stub 加载 class_members + if self._Trans and hasattr(self._Trans, 'ImportHandler'): + _StubName: str = f"{_ExpectedSha1}.stub.ll" + # 临时保存旧值，清除短名条目以绕过 guard，加载后恢复 + _OldMembers = self.class_members.get(ClassName) + self.class_members[ClassName] = [] + try: + self._Trans.ImportHandler._TryLoadClassMembersFromPyi(ClassName, _StubName, self) + except Exception: + pass + _NewMembers = self.class_members.get(ClassName) + if _NewMembers: + self.class_members[_FullKey] = _NewMembers + _ResolvedClass = _FullKey + # 恢复旧值（避免影响其他代码对短名的依赖） + if _OldMembers: + self.class_members[ClassName] = _OldMembers + else: + self.class_members.pop(ClassName, None) + has_vtable: bool = ClassName and (ClassName in self.class_vtable or ClassName in self._cross_module_vtable_classes) + base: int = 1 if has_vtable else 0 + if ClassName and ClassName in self.structs: + struct_type: ir.IdentifiedStructType = self.structs[ClassName] + if isinstance(struct_type, ir.IdentifiedStructType) and struct_type.elements is not None and len(struct_type.elements) > 0: + if ClassName in self.class_members: + expected_fields: int = len(self.class_members[ClassName]) + actual_elements: int = len(struct_type.elements) + if base > 0 and actual_elements == expected_fields: + base = 0 + elif base == 0 and actual_elements == expected_fields + 1: + # 跨模块类可能未被检测为 CVTable（如 Name 继承 AST@t.CVTable）， + # 但实际 struct 首元素是 vtable ptr → 需 base=1。 + # 但若类有继承（class_parent 有值），多出的 1 可能是继承字段而非 vtable， + # 需检查 struct 首元素类型：i8*/i8** 为 vtable，其他为继承字段。 + _first_elem = struct_type.elements[0] if actual_elements > 0 else None + _has_parent = self.class_parent.get(ClassName) is not None + _is_vtable_ptr = (isinstance(_first_elem, ir.PointerType) and + isinstance(_first_elem.pointee, ir.IntType) and _first_elem.pointee.width == 8) + if not _has_parent or _is_vtable_ptr: + base = 1 + if _ResolvedClass and _ResolvedClass in self.class_members: + for i, (name, _) in enumerate(self.class_members[_ResolvedClass]): + if name == field_name: + return base + i + # 继承链回退：当字段不在当前类的 class_members 中时（常见于跨模块导入 + # 父类未加载，字段被填充为 __inherited_N 占位名），沿 class_parent 链向上查找。 + # 子类 struct 布局为 [vtable?] + [基类字段...] + [子类字段...]， + # 基类字段在子类 struct 中的起始偏移即为 base（子类）， + # 因此祖先 class_members 中的索引 i 对应子类 struct 中的偏移 base + i。 + if ClassName and field_name: + _parent: str | None = self.class_parent.get(ClassName) + _visited: set = {ClassName} if ClassName else set() + while _parent and _parent not in _visited: + _visited.add(_parent) + # 跳过 __inherited_N 占位条目，只在真实字段名中查找 + if _parent in self.class_members: + _members: list = self.class_members[_parent] + _has_placeholder: bool = any(n.startswith('__inherited_') for n, _ in _members) + if not _has_placeholder: + for i, (name, _) in enumerate(_members): + if name == field_name: + return base + i + # 父类 class_members 未加载或含占位条目，尝试从 stub 加载真实字段 + if self._Trans and hasattr(self._Trans, 'ImportHandler'): + sha1_map: dict[str, str] = getattr(self, 'ModuleSha1Map', {}) + for mod_name, mod_sha1 in sha1_map.items(): + stub_name: str = f"{mod_sha1}.stub.ll" + self._Trans.ImportHandler._TryLoadClassMembersFromPyi(_parent, stub_name, self) + if _parent in self.class_members and not any(n.startswith('__inherited_') for n, _ in self.class_members[_parent]): + for i, (name, _) in enumerate(self.class_members[_parent]): + if name == field_name: + return base + i + break + _parent = self.class_parent.get(_parent) + short_name: str | None = ClassName.split('.')[-1] if ClassName and '.' in ClassName else None + if short_name and short_name in self.class_members: + for i, (name, _) in enumerate(self.class_members[short_name]): + if name == field_name: + return base + i + if ClassName and field_name: + if ClassName not in self.class_members or field_name not in [n for n, _ in self.class_members.get(ClassName, [])]: + if self._Trans and hasattr(self._Trans, 'ImportHandler'): + sha1_map: dict[str, str] = getattr(self, 'ModuleSha1Map', {}) + for mod_name, mod_sha1 in sha1_map.items(): + stub_name: str = f"{mod_sha1}.stub.ll" + self._Trans.ImportHandler._TryLoadClassMembersFromPyi(ClassName, stub_name, self) + if ClassName in self.class_members and len(self.class_members[ClassName]) > 0: + for i, (name, _) in enumerate(self.class_members[ClassName]): + if name == field_name: + return base + i + break + if short_name: + for mod_name, mod_sha1 in sha1_map.items(): + stub_name: str = f"{mod_sha1}.stub.ll" + self._Trans.ImportHandler._TryLoadClassMembersFromPyi(short_name, stub_name, self) + if short_name in self.class_members and len(self.class_members[short_name]) > 0: + for i, (name, _) in enumerate(self.class_members[short_name]): + if name == field_name: + return base + i + break + if ClassName and field_name: + for cn_key in self.class_members: + if cn_key == ClassName or cn_key.endswith(f'.{ClassName}') or (short_name and (cn_key == short_name or cn_key.endswith(f'.{short_name}'))): + for i, (name, _) in enumerate(self.class_members[cn_key]): + if name == field_name: + return base + i + # REnum 变体搜索: ClassName 是 REnum（如 MetaType），成员注册在变体结构 + # （MetaType_Var, MetaType_Concrete, MetaType_App）中。变体与 REnum 主结构 + # 共享 max_variant_struct 布局，偏移一致（__tag 在偏移 0，字段从偏移 1 开始） + if ClassName and field_name: + variant_prefix: str = f"{ClassName}_" + for cn_key in self.class_members: + if cn_key.startswith(variant_prefix): + for i, (name, _) in enumerate(self.class_members[cn_key]): + if name == field_name: + return i + if short_name: + variant_prefix = f"{short_name}_" + for cn_key in self.class_members: + if cn_key.startswith(variant_prefix): + for i, (name, _) in enumerate(self.class_members[cn_key]): + if name == field_name: + return i + if ClassName and ClassName in self.structs: + struct_type: ir.IdentifiedStructType = self.structs[ClassName] + if isinstance(struct_type, ir.IdentifiedStructType) and struct_type.elements is not None and len(struct_type.elements) > 0: + for cn_key, members in self.class_members.items(): + if cn_key == short_name or (short_name and cn_key.endswith(f'.{short_name}')): + for i, (name, _) in enumerate(members): + if name == field_name and i < len(struct_type.elements): + return i + break + else: + if short_name: + for cn_key, members in self.class_members.items(): + sn: str = cn_key.split('.')[-1] if '.' in cn_key else cn_key + if sn == short_name: + for i, (name, _) in enumerate(members): + if name == field_name and i < len(struct_type.elements): + return i + break + if ClassName and field_name and ClassName not in self.class_members: + if self._Trans and hasattr(self._Trans, 'ImportHandler'): + for temp_dir_candidate in [getattr(self, '_temp_dir', None)]: + if temp_dir_candidate and os.path.isdir(temp_dir_candidate): + for pyi_file in os.listdir(temp_dir_candidate): + if pyi_file.endswith('.pyi'): + stub_name: str = pyi_file.replace('.pyi', '.stub.ll') + short_cn: str = short_name if short_name else ClassName + self._Trans.ImportHandler._TryLoadClassMembersFromPyi(short_cn, stub_name, self) + if short_cn in self.class_members and len(self.class_members[short_cn]) > 0: + for i, (name, _) in enumerate(self.class_members[short_cn]): + if name == field_name: + return base + i + break + if ClassName and ClassName in self.structs and (ClassName not in self.class_members or len(self.class_members.get(ClassName, [])) == 0): + struct_type: ir.IdentifiedStructType = self.structs[ClassName] + if isinstance(struct_type, ir.IdentifiedStructType) and struct_type.elements is not None: + if self._Trans and hasattr(self._Trans, 'ClassHandler'): + class_handler: Any = self._Trans.ClassHandler + if self._current_tree: + for node in ast.iter_child_nodes(self._current_tree): + if isinstance(node, ast.ClassDef) and (node.name == ClassName or node.name == short_name): + if node.name not in self.class_members: + self.class_members[node.name] = [] + if len(self.class_members[node.name]) == 0: + class_handler._PreRegisterClassMembers(node, self) + if ClassName in self.class_members: + for i, (name, _) in enumerate(self.class_members[ClassName]): + if name == field_name: + return base + i + if short_name and short_name in self.class_members: + for i, (name, _) in enumerate(self.class_members[short_name]): + if name == field_name: + return base + i + break + # 最后回退：当继承链回退失败但 struct body 已加载时， + # 从 struct elements 数和 class_members 数推断继承字段偏移。 + # 继承字段在 struct 开头（base 之后），此 fallback 仅适用于单继承字段的情况 + # （如 GSListNode[T] 的 Next 字段）。子类 struct 布局： + # [vtable?] + [基类字段...] + [子类字段...] + # 当 class_members 只含子类自身字段、基类 class_members 未加载时， + # 继承字段数 = struct_elements - class_members_count， + # 若该值 == 1，则所查字段（如 Next）的偏移 = base + 0。 + if ClassName and field_name: + _st_final = self.structs.get(ClassName) + if isinstance(_st_final, ir.IdentifiedStructType) and _st_final.elements is not None: + _num_elements_final: int = len(_st_final.elements) + _members_final: list = self.class_members.get(ClassName, []) + _num_members_final: int = len(_members_final) + if _num_elements_final > _num_members_final: + _num_inherited_final: int = _num_elements_final - _num_members_final + if _num_inherited_final == 1: + # 单继承字段，偏移 = base（vtable 之后第一个位置） + return base + 0 + return None + + def _resolve_class_for_var(self, var_name: str) -> str | None: + for ClassName in self.class_methods: + if var_name == ClassName or var_name.startswith(f"__with_{ClassName}_"): + return ClassName + if var_name in self.variables: + var: ir.Value = self.variables[var_name] + if isinstance(var.type, ir.PointerType): + for ClassName, struct_type in self.structs.items(): + if var.type.pointee == struct_type: + return ClassName + if isinstance(var.type.pointee, ir.PointerType) and var.type.pointee.pointee == struct_type: + return ClassName + return None + + def _infer_return_type(self, FuncName: str) -> ir.Type: + if FuncName in self.functions: + return self.functions[FuncName].type.pointee.return_type + # 也尝试混淆后的名称 + mangled: str = self._mangle_func_name(FuncName) + if mangled != FuncName and mangled in self.functions: + return self.functions[mangled].type.pointee.return_type + if FuncName in self.known_return_types: + return self.known_return_types[FuncName] + if mangled != FuncName and mangled in self.known_return_types: + return self.known_return_types[mangled] + for ClassName, methods in self.class_methods.items(): + for method in methods: + if FuncName == method: + return ir.VoidType() + if self._import_handler_ref: + try: + # 先用原始名称查找，再用混淆名称查找 + stub_func_type: Any = self._import_handler_ref._LookupStubFuncType(FuncName, self) + if stub_func_type is None and mangled != FuncName: + stub_func_type = self._import_handler_ref._LookupStubFuncType(mangled, self) + if stub_func_type and hasattr(stub_func_type, 'return_type'): + return stub_func_type.return_type + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"查找桩函数返回类型失败: {_e}", "Exception") + return ir.IntType(32) + + def _get_or_declare_func(self, name: str, func_type: ir.FunctionType) -> ir.Function: + for fname, fobj in self.functions.items(): + if fname == name: + return fobj + if re.match(r'^[0-9a-f]{16,}\.' + re.escape(name) + r'$', fname): + return fobj + func_decl: ir.Function = ir.Function(self.module, func_type, name=name) + self.functions[name] = func_decl + return func_decl + + def get_or_declare_c_func(self, name: str, fallback_type: ir.FunctionType | None = None) -> ir.Function | None: + if name in self.functions: + return self.functions[name] + for fname, fobj in self.functions.items(): + if re.match(r'^[0-9a-f]{16,}\.' + re.escape(name) + r'$', fname): + return fobj + stub_func_type: ir.FunctionType | None = None + if self._import_handler_ref: + try: + stub_func_type = self._import_handler_ref._LookupStubFuncType(name, self) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"查找桩函数类型失败: {_e}", "Exception") + if stub_func_type: + func_decl: ir.Function = ir.Function(self.module, stub_func_type, name=name) + self.functions[name] = func_decl + return func_decl + if fallback_type is not None: + func_decl: ir.Function = ir.Function(self.module, fallback_type, name=name) + self.functions[name] = func_decl + return func_decl + return None \ No newline at end of file diff --git a/lib/core/LLVMCG/MemoryOps.py b/lib/core/LLVMCG/MemoryOps.py new file mode 100644 index 0000000..07003e6 --- /dev/null +++ b/lib/core/LLVMCG/MemoryOps.py @@ -0,0 +1,286 @@ +from __future__ import annotations +import llvmlite.ir as ir + + +class MemoryOpsMixin: + + def _register_local_heap_ptr(self, val: ir.Value, ClassName: str | None = None, VarName: str | None = None) -> None: + if isinstance(val.type, ir.PointerType): + pointee: ir.Type = val.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + self._local_heap_ptrs.append((val, 'struct', ClassName)) + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + self._local_heap_ptrs.append((val, 'i8', ClassName)) + if VarName: + self._var_to_heap_ptr[VarName] = val + + def _unregister_local_heap_ptr(self, val: ir.Value) -> None: + self._local_heap_ptrs = [(v, t, cn) for v, t, cn in self._local_heap_ptrs if v is not val] + + def _emit_local_heap_frees(self) -> None: + if not self._local_heap_ptrs or not self.builder: + return + deleted_objects: set[ir.Value] = set() + for ptr, ptr_type, ClassName in self._local_heap_ptrs: + if ptr_type == 'i8': + Loaded_ptr: ir.Value = self.builder.load(ptr, name="load_ptr") + null_ptr: ir.Constant = ir.Constant(ir.PointerType(ir.IntType(8)), None) + is_null: ir.Value = self.builder.icmp_unsigned('==', Loaded_ptr, null_ptr, name="is_null") + not_null_block: ir.Block = self.builder.append_basic_block(name="not_null") + next_block: ir.Block = self.builder.append_basic_block(name="next") + self.builder.cbranch(is_null, next_block, not_null_block) + self.builder.position_at_end(not_null_block) + if ClassName and self._has_function(f'{ClassName}.__del__'): + cast_ptr: ir.Value + if isinstance(ptr.type, ir.PointerType) and isinstance(ptr.type.pointee, ir.IntType) and ptr.type.pointee.width == 8: + cast_ptr = self.builder.bitcast(ptr, ir.PointerType(self.structs[ClassName]), name=f"cast_{ClassName}") + else: + cast_ptr = ptr + if cast_ptr not in deleted_objects: + self.builder.call(self._get_function(f'{ClassName}.__del__'), [cast_ptr], name=f"call_{ClassName}.__del__") + deleted_objects.add(cast_ptr) + raw: ir.Value + if ptr_type == 'struct': + raw = self.builder.bitcast(ptr, ir.PointerType(ir.IntType(8)), name="local_free_cast") + elif ptr_type == 'i8': + raw = ptr + else: + continue + free_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))]) + free_func: ir.Function = self._get_or_declare_func('free', free_type) + self.builder.call(free_func, [raw]) + if ptr_type == 'i8': + self.builder.branch(next_block) + self.builder.position_at_end(next_block) + self._local_heap_ptrs = [] + + def _RegisterTempPtr(self, val: ir.Value) -> None: + if isinstance(val.type, ir.PointerType): + pointee: ir.Type = val.type.pointee + if isinstance(pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + for CN, ST in self.structs.items(): + if pointee == ST: + self._temp_struct_ptrs.append(val) + return + elif isinstance(pointee, ir.IntType) and pointee.width == 8: + self._temp_struct_ptrs.append(val) + + def _UnregisterTempPtr(self, val: ir.Value) -> None: + self._temp_struct_ptrs = [t for t in self._temp_struct_ptrs if t is not val] + + def _EmitTempFrees(self) -> None: + if not self._temp_struct_ptrs or not self.builder: + return + for ptr in self._temp_struct_ptrs: + raw: ir.Value + if isinstance(ptr.type, ir.PointerType) and isinstance(ptr.type.pointee, (ir.LiteralStructType, ir.IdentifiedStructType)): + raw = self.builder.bitcast(ptr, ir.PointerType(ir.IntType(8)), name="free_cast") + elif isinstance(ptr.type, ir.PointerType) and isinstance(ptr.type.pointee, ir.IntType) and ptr.type.pointee.width == 8: + raw = ptr + else: + continue + free_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.PointerType(ir.IntType(8))]) + free_func: ir.Function = self._get_or_declare_func('free', free_type) + self.builder.call(free_func, [raw]) + self._temp_struct_ptrs = [] + + def _ZeroConst(self, typ: ir.Type) -> ir.Constant: + if isinstance(typ, ir.PointerType): + return ir.Constant(typ, None) + if isinstance(typ, (ir.LiteralStructType, ir.IdentifiedStructType)): + return ir.Constant(typ, None) + if isinstance(typ, ir.ArrayType): + return ir.Constant(typ, None) + return ir.Constant(typ, 0) + + def _alloca(self, typ: ir.Type, name: str = '', align: int | None = None, size: ir.Value | None = None) -> ir.AllocaInstr: + if align is None: + align = self._get_align(typ) + instr: ir.AllocaInstr + if size is not None: + instr = self.builder.alloca(typ, size=size, name=name) + else: + instr = self.builder.alloca(typ, name=name) + if align: + instr.align = align + return instr + + def _allocaEntry(self, typ: ir.Type, name: str = '', align: int | None = None) -> ir.AllocaInstr: + if align is None: + align = self._get_align(typ) + saved_block: ir.Block = self.builder.block + entry_block: ir.Block = self.func.entry_basic_block + instr: ir.AllocaInstr + if entry_block.instructions: + first_instr: ir.Instruction = entry_block.instructions[0] + self.builder.position_before(first_instr) + instr = self.builder.alloca(typ, name=name) + if align: + instr.align = align + self.builder.position_at_end(saved_block) + else: + self.builder.position_at_start(entry_block) + instr = self.builder.alloca(typ, name=name) + if align: + instr.align = align + self.builder.position_at_end(saved_block) + return instr + + def _load(self, ptr: ir.Value, name: str = '', align: int | None = None) -> ir.LoadInstr: + if align is None: + if isinstance(ptr.type, ir.PointerType): + align = self._get_align(ptr.type.pointee) + else: + align = 0 + if isinstance(ptr.type, ir.PointerType): + pointee_type: ir.Type = ptr.type.pointee + if isinstance(pointee_type, ir.IdentifiedStructType) and pointee_type.name in self.typedef_int_widths: + width: int = self.typedef_int_widths[pointee_type.name] + if width > 0: + actual_type: ir.IntType = ir.IntType(width) + bitcast_ptr: ir.Value = self.builder.bitcast(ptr, ir.PointerType(actual_type), name="typedef_bitcast_Load") + return self.builder.load(bitcast_ptr, name=name, align=align) + return self.builder.load(ptr, name=name, align=align) + + def _Load_var(self, name: str) -> ir.Value | None: + if name in self._direct_values: + return self._direct_values[name] + if name in self.variables and self.variables[name] is not None: + VarPtr: ir.Value = self.variables[name] + if isinstance(VarPtr, ir.GlobalVariable) and isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return VarPtr + if isinstance(VarPtr.type, ir.PointerType) and isinstance(VarPtr.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return VarPtr + return self._load(VarPtr, name=name) + if name in self.global_vars and name in self.module.globals: + GVar: ir.GlobalVariable = self.module.globals[name] + self.variables[name] = GVar + if isinstance(GVar, ir.Function): + return GVar + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar + return self._load(GVar, name=name) + if name in self.module.globals: + GVar = self.module.globals[name] + self.variables[name] = GVar + if isinstance(GVar, ir.Function): + return GVar + if isinstance(GVar.type, ir.PointerType) and isinstance(GVar.type.pointee, (ir.IdentifiedStructType, ir.LiteralStructType, ir.ArrayType)): + return GVar + return self._load(GVar, name=name) + return None + + def _coerce_value(self, val: ir.Value, target_type: ir.Type) -> ir.Value | None: + if val.type == target_type: + return val + if isinstance(target_type, ir.IdentifiedStructType) and target_type.name in self.typedef_int_widths: + if isinstance(val.type, ir.IntType): + return val + if isinstance(target_type, ir.IntType) and isinstance(val.type, ir.IntType): + if target_type.width < val.type.width: + return self.builder.trunc(val, target_type, name="trunc_store") + elif val.type.width == 1: + # i1 是布尔值（来自 icmp/fcmp/__contains__ 归一化），无符号语义， + # 必须用 zext 而非 sext，否则 i1 的 1 会被符号扩展为 -1 + return self.builder.zext(val, target_type, name="zext_bool_store") + else: + return self.builder.sext(val, target_type, name="sext_store") + if isinstance(target_type, ir.PointerType) and isinstance(val.type, ir.PointerType): + if isinstance(val, ir.Constant) and val.constant is None: + return ir.Constant(target_type, None) + return self.builder.bitcast(val, target_type, name="ptr_bitcast_store") + if isinstance(target_type, ir.PointerType) and isinstance(val.type, ir.IntType): + if val.type.width < 64: + val = self.builder.zext(val, ir.IntType(64), name="zext_ptr_store") + return self.builder.inttoptr(val, target_type, name="int_to_ptr_store") + if isinstance(target_type, ir.IntType) and isinstance(val.type, ir.PointerType): + if isinstance(val.type.pointee, ir.IntType) and val.type.pointee.width == target_type.width: + return self.builder.load(val, name="load_ptr_as_int_store") + if target_type.width <= 8 and isinstance(val.type.pointee, ir.IntType): + Loaded: ir.Value = self.builder.load(val, name="load_ptr_byte_store") + if Loaded.type != target_type: + if isinstance(Loaded.type, ir.IntType) and isinstance(target_type, ir.IntType): + if target_type.width < Loaded.type.width: + return self.builder.trunc(Loaded, target_type, name="trunc_ptr_byte_store") + return self.builder.zext(Loaded, target_type, name="zext_ptr_byte_store") + return Loaded + return self.builder.ptrtoint(val, target_type, name="ptr_to_int_store") + if isinstance(target_type, ir.FloatType) and isinstance(val.type, ir.DoubleType): + return self.builder.fptrunc(val, target_type, name="fptrunc_store") + if isinstance(target_type, ir.DoubleType) and isinstance(val.type, ir.FloatType): + return self.builder.fpext(val, target_type, name="fpext_store") + if isinstance(target_type, ir.IntType) and isinstance(val.type, (ir.FloatType, ir.DoubleType)): + return self.builder.fptosi(val, target_type, name="fptosi_store") + if isinstance(target_type, (ir.FloatType, ir.DoubleType)) and isinstance(val.type, ir.IntType): + return self.builder.sitofp(val, target_type, name="sitofp_store") + if isinstance(target_type, ir.ArrayType) and isinstance(val.type, ir.ArrayType): + if target_type.count == val.type.count and target_type.element == val.type.element: + return val + if (isinstance(target_type.element, ir.IntType) and target_type.element.width == 8 + and isinstance(val.type.element, ir.IntType) and val.type.element.width == 8): + if target_type.count > val.type.count: + padded: list[ir.Constant] = list(val.constant) if hasattr(val, 'constant') else [] + if padded and len(padded) == val.type.count: + padded.extend([ir.Constant(ir.IntType(8), 0)] * (target_type.count - val.type.count)) + return ir.Constant(target_type, padded) + elif target_type.count < val.type.count: + trimmed: list[ir.Constant] = list(val.constant) if hasattr(val, 'constant') else [] + if trimmed and len(trimmed) == val.type.count: + return ir.Constant(target_type, trimmed[:target_type.count]) + if isinstance(val.type, ir.PointerType) and not isinstance(target_type, ir.PointerType): + if val.type.pointee == target_type: + return self.builder.load(val, name="Load_struct_for_store") + if isinstance(val.type.pointee, ir.IdentifiedStructType) and isinstance(target_type, ir.IdentifiedStructType): + if val.type.pointee.name == target_type.name: + Loaded: ir.Value = self.builder.load(val, name="Load_struct_for_store") + if Loaded.type == target_type: + return Loaded + return None + + def _store(self, val: ir.Value, ptr: ir.Value, align: int | None = None) -> ir.StoreInstr: + if align is None: + if isinstance(ptr.type, ir.PointerType): + align = self._get_align(ptr.type.pointee) + else: + align = 0 + if isinstance(ptr.type, ir.PointerType): + target_type: ir.Type = ptr.type.pointee + if isinstance(target_type, ir.IdentifiedStructType) and target_type.name in self.typedef_int_widths: + if isinstance(val.type, ir.IntType): + width: int = self.typedef_int_widths[target_type.name] + if width > 0: + actual_type: ir.IntType = ir.IntType(width) + bitcast_ptr: ir.Value = self.builder.bitcast(ptr, ir.PointerType(actual_type), name="typedef_bitcast") + if val.type.width != actual_type.width: + if val.type.width > actual_type.width: + val = self.builder.trunc(val, actual_type, name="trunc_typedef") + else: + val = self.builder.zext(val, actual_type, name="zext_typedef") + return self.builder.store(val, bitcast_ptr, align=align) + if val.type != target_type: + if isinstance(val, ir.Constant) and val.constant is None: + if isinstance(target_type, ir.PointerType): + val = ir.Constant(target_type, None) + elif isinstance(target_type, ir.IdentifiedStructType): + val = ir.Constant(ir.PointerType(target_type), None) + ptr = self.builder.bitcast(ptr, ir.PointerType(ir.PointerType(target_type)), name="null_struct_ptr_cast") + else: + coerced: ir.Value | None = self._coerce_value(val, target_type) + if coerced is not None: + val = coerced + else: + src_info: str = self._get_node_info() + raise TypeError(f"cannot store {val.type} to {ptr.type}: mismatching types{src_info}") + return self.builder.store(val, ptr, align=align) + + def _get_var_ptr(self, name: str) -> ir.Value | None: + if name in self._reg_values: + val: ir.Value = self._reg_values[name] + var: ir.AllocaInstr = self._alloca(val.type, name=name) + self._store(val, var) + self.variables[name] = var + del self._reg_values[name] + return var + if name in self.variables and self.variables[name] is not None: + return self.variables[name] + return None \ No newline at end of file diff --git a/lib/core/LLVMCG/StructGen.py b/lib/core/LLVMCG/StructGen.py new file mode 100644 index 0000000..3ab012a --- /dev/null +++ b/lib/core/LLVMCG/StructGen.py @@ -0,0 +1,587 @@ +from __future__ import annotations +from typing import Any +import re +import ast +import llvmlite.ir as ir +from llvmlite.ir.values import _Undefined +from lib.includes import t as _t + + +class StructGenMixin: + + def _get_align(self, llvm_type: ir.Type) -> int: + if isinstance(llvm_type, ir.IntType): + return max(1, llvm_type.width // 8) + if isinstance(llvm_type, ir.FloatType): + return 4 + if isinstance(llvm_type, ir.DoubleType): + return 8 + if isinstance(llvm_type, ir.PointerType): + return self.ptr_size + if isinstance(llvm_type, ir.ArrayType): + return self._get_align(llvm_type.element) + if isinstance(llvm_type, ir.LiteralStructType): + return max((self._get_align(f) for f in llvm_type.elements), default=1) + if isinstance(llvm_type, ir.IdentifiedStructType): + if llvm_type.elements is None or len(llvm_type.elements) == 0: + return self.ptr_size # opaque struct, assume pointer alignment + return max((self._get_align(f) for f in llvm_type.elements), default=1) + if isinstance(llvm_type, ir.VoidType): + return 0 + return 4 + + def _resolve_opaque_struct(self, llvm_type: ir.Type) -> ir.Type: + """尝试从 Gen.structs 中查找已解析的同名 struct，解决跨模块 struct 类型对象不一致的问题""" + if not isinstance(llvm_type, ir.IdentifiedStructType): + return llvm_type + if llvm_type.elements is not None and len(llvm_type.elements) > 0: + return llvm_type + # opaque struct，尝试通过名称在 Gen.structs 中查找已解析的版本 + st_name: str + try: + st_name = llvm_type.name + except Exception: + return llvm_type + if not st_name: + return llvm_type + # 提取短名称 + short_name: str = st_name + if '.' in st_name: + short_name = st_name.split('.', 1)[1] + # 尝试短名称查找 + if short_name in self.structs: + resolved: ir.IdentifiedStructType = self.structs[short_name] + if isinstance(resolved, ir.IdentifiedStructType) and resolved.elements is not None and len(resolved.elements) > 0: + return resolved + # 尝试全名查找 + if st_name in self.structs: + resolved: ir.IdentifiedStructType = self.structs[st_name] + if isinstance(resolved, ir.IdentifiedStructType) and resolved.elements is not None and len(resolved.elements) > 0: + return resolved + # 尝试通过导入处理器从 stub 文件加载 struct 定义 + import_handler: Any = self._import_handler_ref + if import_handler is not None: + try: + import_handler._TryLoadStructFromStub(short_name, self) + except Exception: + pass + # 加载后再次查找 + if short_name in self.structs: + resolved = self.structs[short_name] + if isinstance(resolved, ir.IdentifiedStructType) and resolved.elements is not None and len(resolved.elements) > 0: + return resolved + if st_name in self.structs: + resolved = self.structs[st_name] + if isinstance(resolved, ir.IdentifiedStructType) and resolved.elements is not None and len(resolved.elements) > 0: + return resolved + return llvm_type + + def _get_struct_size(self, llvm_type: ir.Type) -> int: + if isinstance(llvm_type, ir.IntType): + return max(1, llvm_type.width // 8) + if isinstance(llvm_type, ir.FloatType): + return 4 + if isinstance(llvm_type, ir.DoubleType): + return 8 + if isinstance(llvm_type, ir.PointerType): + return self.ptr_size + if isinstance(llvm_type, ir.ArrayType): + return self._get_struct_size(llvm_type.element) * llvm_type.count + if isinstance(llvm_type, ir.LiteralStructType): + total: int = 0 + max_align: int = 1 + for f in llvm_type.fields: + fa: int = self._get_align(f) + fs: int = self._get_struct_size(f) + max_align = max(max_align, fa) + if fa > 1: + total = (total + fa - 1) // fa * fa + total += fs + if max_align > 1: + total = (total + max_align - 1) // max_align * max_align + return total + if isinstance(llvm_type, ir.IdentifiedStructType): + # 尝试解析 opaque struct 为已注册的版本 + llvm_type = self._resolve_opaque_struct(llvm_type) + if llvm_type.elements is None or len(llvm_type.elements) == 0: + return self.ptr_size # opaque struct, assume pointer size + total: int = 0 + max_align: int = 1 + for f in llvm_type.elements: + fa: int = self._get_align(f) + fs: int = self._get_struct_size(f) + max_align = max(max_align, fa) + if fa > 1: + total = (total + fa - 1) // fa * fa + total += fs + if max_align > 1: + total = (total + max_align - 1) // max_align * max_align + return total + return 4 + + @staticmethod + def _strip_unnecessary_quotes(ir_str: str) -> str: + def _unquote(m: re.Match) -> str: + prefix: str = m.group(1) + name: str = m.group(2) + if '.' in name: + return m.group(0) + if re.match(r'^[a-zA-Z_$][a-zA-Z0-9_$]*$', name): + return prefix + name + return m.group(0) + return re.sub(r'([@%])"([^"]+)"', _unquote, ir_str) + + def finalize(self) -> str: + self._set_Vtable_initializers() + self._fix_global_var_init() + ir_text: str + try: + ir_text = str(self.module) + except TypeError as e: + for func in self.module.functions: + for block in func.blocks: + for instr in block.instructions: + try: + str(instr) + except TypeError as te: + pass + raise + ir_text = self._strip_unnecessary_quotes(ir_text) + struct_defs: list[str] = [] + struct_names: set[str] = set() + for ClassName, struct_type in self.structs.items(): + if isinstance(struct_type, ir.IdentifiedStructType): + sname: str = struct_type.name + sname_ir: str + if sname[0].isdigit() or '.' in sname or not sname.replace('_', '').replace('.', '').isalnum(): + sname_ir = f'%"{sname}"' + else: + sname_ir = f'%{sname}' + if struct_type.elements is None or len(struct_type.elements) == 0: + if ClassName not in self.typedef_int_widths: + struct_names.add(sname) + struct_defs.append(f'{sname_ir} = type opaque') + continue + elems: str = ', '.join(str(e) for e in struct_type.elements) + if struct_type.packed: + struct_defs.append(f'{sname_ir} = type <{{ {elems} }}>') + else: + struct_defs.append(f'{sname_ir} = type {{ {elems} }}') + struct_names.add(sname) + if struct_defs: + lines: list[str] = ir_text.split('\n') + filtered: list[str] = [] + for line in lines: + stripped: str = line.strip() + skip: bool = False + for name in struct_names: + patterns: list[str] + if name[0].isdigit() or '.' in name or not name.replace('_', '').replace('.', '').isalnum(): + patterns = [f'%"{name}" = type', f'%"{name}"=type', f'%{name} = type', f'%{name}=type'] + else: + patterns = [f'%{name} = type', f'%{name}= type', f'%{name}=type'] + for p in patterns: + if stripped.startswith(p): + skip = True + break + if skip: + break + if not skip: + filtered.append(line) + ir_text = '\n'.join(filtered) + header_end: int = ir_text.find('\ndefine') + if header_end == -1: + header_end = ir_text.find('\ndeclare') + if header_end == -1: + header_end = ir_text.find('\n@') + if header_end == -1: + header_end = len(ir_text) + insert_pos: int = ir_text.rfind('\n', 0, header_end) + if insert_pos == -1: + insert_pos = header_end + struct_block: str = '\n'.join(struct_defs) + '\n' + ir_text = ir_text[:insert_pos + 1] + struct_block + ir_text[insert_pos + 1:] + for name in struct_names: + if name[0].isdigit() or '.' in name: + ir_text = re.sub( + r'%' + re.escape(name) + r'(?=[^a-zA-Z0-9_$\.]|$)', + r'%"' + name + r'"', + ir_text + ) + + return ir_text + + def _generate_structs(self) -> None: + all_classes: set[str] = set(self.class_methods.keys()) | set(self.class_members.keys()) + + preserved_structs: dict[str, ir.IdentifiedStructType] = {} + for name, st in self.structs.items(): + if name not in all_classes: + preserved_structs[name] = st + elif isinstance(st, ir.IdentifiedStructType): + preserved_structs[name] = st + + self.structs.clear() + + for name, st in preserved_structs.items(): + self.structs[name] = st + + # Step 1: 先为所有类创建空的 struct，确保后面可以正确引用 + all_classes = set(self.class_methods.keys()) | set(self.class_members.keys()) + for ClassName in all_classes: + is_packed: bool = ClassName in self.class_packed + if self.SymbolTable: + Entry = self.SymbolTable.lookup(ClassName) + if Entry and Entry.IsExceptionClass: + continue + # 注意：类名可能与 CEnum 成员名冲突（如 ASTKind.Module 枚举成员与 Module 类同名） + # 如果 ClassName 在 class_methods 中，表明是 _EmitClassLlvm 显式处理的常规类， + # 不应被 IsEnum/IsTypedef 检查跳过 + if Entry and Entry.IsEnum and ClassName not in self.class_methods: + if not Entry.IsRenum: + continue + if Entry and Entry.IsTypedef and ClassName not in self.class_methods: + if Entry.BaseType and not isinstance(Entry.BaseType, (type(None),)): + if not isinstance(Entry.BaseType, (_t._CTypedef,)): + continue + if ClassName in self.structs: + if is_packed: + self.structs[ClassName].packed = True + continue + mangled_name: str = self._mangle_name(ClassName) + struct_type: ir.IdentifiedStructType = ir.IdentifiedStructType(self.module, mangled_name, packed=is_packed) + self.structs[ClassName] = struct_type + + # Step 2: 解析跨模块的成员类型 + for ClassName, annotations in self.class_member_annotations.items(): + if ClassName not in self.class_members: + continue + for i, (member_name, member_type) in enumerate(self.class_members[ClassName]): + if member_name in annotations: + annotation: Any = annotations[member_name] + ResolvedType: ir.Type | None = None + + if isinstance(annotation, ast.Attribute): + ClassTypeName: str = annotation.attr + if ClassTypeName in self.structs: + ResolvedType = self.structs[ClassTypeName] + elif isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + left: Any = annotation.left + if isinstance(left, ast.Attribute): + ClassTypeName: str = left.attr + if ClassTypeName in self.structs: + ResolvedType = self.structs[ClassTypeName] + + if ResolvedType: + self.class_members[ClassName][i] = (member_name, ResolvedType) + + struct_name_map: dict[str, ir.IdentifiedStructType] = {} + for struct_name, struct in self.structs.items(): + try: + struct_name_map[struct.name] = struct + except AssertionError: + struct_name_map[struct_name] = struct + orig: Any = getattr(struct, '_original_name', None) + if orig: + struct_name_map[orig] = struct + + # Step 3: 更新成员类型，确保使用正确的 mangled struct 引用 + for ClassName in all_classes: + if ClassName not in self.class_members: + continue + for i, (member_name, member_type) in enumerate(self.class_members[ClassName]): + if isinstance(member_type, ir.PointerType): + pointee: ir.Type = member_type.pointee + if isinstance(pointee, ir.IdentifiedStructType): + pname: str | None + try: + pname = pointee.name + except AssertionError: + pname = None + struct: ir.IdentifiedStructType | None = struct_name_map.get(pname) if pname else None + if struct is None: + orig: Any = getattr(pointee, '_original_name', None) + struct = struct_name_map.get(orig) if orig else None + if struct is not None: + self.class_members[ClassName][i] = (member_name, ir.PointerType(struct)) + else: + raw_name: str = pname or '' + if raw_name.startswith('struct.'): + raw_name = raw_name[7:] + new_struct: ir.IdentifiedStructType = self._get_or_create_struct(raw_name) + self.class_members[ClassName][i] = (member_name, ir.PointerType(new_struct)) + elif isinstance(member_type, ir.IdentifiedStructType): + mname: str | None + try: + mname = member_type.name + except AssertionError: + mname = None + struct: ir.IdentifiedStructType | None = struct_name_map.get(mname) if mname else None + if struct is None: + orig: Any = getattr(member_type, '_original_name', None) + struct = struct_name_map.get(orig) if orig else None + if struct is not None: + self.class_members[ClassName][i] = (member_name, struct) + else: + raw_name: str = mname or '' + if raw_name.startswith('struct.'): + raw_name = raw_name[7:] + new_struct: ir.IdentifiedStructType = self._get_or_create_struct(raw_name) + self.class_members[ClassName][i] = (member_name, new_struct) + + # Step 4: 创建最终的 structs 并设置正确的成员类型 + for ClassName in all_classes: + if self.SymbolTable: + Entry = self.SymbolTable.lookup(ClassName) + if Entry and Entry.IsExceptionClass: + continue + # 注意：类名可能与 CEnum 成员名冲突（如 ASTKind.Constant 枚举成员与 Constant 类同名） + # 如果 ClassName 在 class_methods 中，表明是 _EmitClassLlvm 显式处理的常规类， + # 不应被 IsEnum 检查跳过（与 Step 1 的守卫保持一致） + if Entry and Entry.IsEnum and ClassName not in self.class_methods: + if not Entry.IsRenum: + continue + existing_st: ir.IdentifiedStructType | None = self.structs.get(ClassName) + if isinstance(existing_st, ir.IdentifiedStructType) and existing_st.elements is not None and len(existing_st.elements) > 0: + # 治本修复：检查 struct body 是否不完整（元素数少于 class_members + vtable 预期）。 + # 常见于跨模块继承：子类 struct 在父类 class_members 加载前从 stub 生成， + # stub 中的定义可能不完整或类型不正确（如 GSList* 代替 AST*、i8* 代替 i32）。 + # 此处检测到不完整后清除 _elements 和 stub 缓存，让后续代码重新从 stub 加载。 + _has_vtable_precheck: bool = ClassName in self.class_vtable or ClassName in self._cross_module_vtable_classes + if not _has_vtable_precheck: + _p_pre: str | None = self.class_parent.get(ClassName) + while _p_pre: + if _p_pre in self.class_vtable or _p_pre in self._cross_module_vtable_classes: + _has_vtable_precheck = True + break + _p_pre = self.class_parent.get(_p_pre) + _expected_count_pre: int = len(self.class_members.get(ClassName, [])) + (1 if _has_vtable_precheck else 0) + # has_vtable 时，首字段必须是 i8*（vtable 指针）。 + # 若首字段非指针（如 i8 占位符），struct body 不完整，需清除重新加载。 + _vtable_first_ok: bool = True + if _has_vtable_precheck and len(existing_st.elements) > 0: + _vtable_first_ok = isinstance(existing_st.elements[0], ir.PointerType) + if len(existing_st.elements) >= _expected_count_pre and _vtable_first_ok: + if ClassName in self.class_packed and not existing_st.packed: + existing_st.packed = True + continue + # struct body 不完整，清除 _elements 和 stub 缓存以便重新加载 + existing_st._elements = None + _import_handler_pre: Any = self._import_handler_ref + if _import_handler_pre is not None: + _cache_key_pre: tuple = (id(self), ClassName) + if _cache_key_pre in _import_handler_pre._struct_Load_cache: + del _import_handler_pre._struct_Load_cache[_cache_key_pre] + has_vtable: bool = ClassName in self.class_vtable or ClassName in self._cross_module_vtable_classes + if not has_vtable: + p: str | None = self.class_parent.get(ClassName) + while p: + if p in self.class_vtable or p in self._cross_module_vtable_classes: + has_vtable = True + self.class_vtable.add(ClassName) + break + p = self.class_parent.get(p) + has_bitfield: bool = ClassName in self.class_member_bitfields and any( + self.class_member_bitfields[ClassName].get(name, 0) > 0 + for name, _ in self.class_members.get(ClassName, []) + ) + all_members: list[tuple[str, ir.Type]] = self.class_members.get(ClassName, []) + # 跳过正在发射中且成员尚未填充的类（递归特化时避免过早设置空 body） + if not all_members and not has_vtable and ClassName in self._classes_in_progress: + continue + # 字段收集期间触发的嵌套 _generate_structs：class_members 可能不完整， + # set_body 只能调用一次，跳过等字段收集完成后再设置 body + if ClassName in self._collecting_members_set: + continue + all_bitfield: bool = all( + self.class_member_bitfields.get(ClassName, {}).get(name, 0) > 0 + for name, _ in all_members + ) if all_members else False + + mangled_name: str = self._mangle_name(ClassName) + + member_types: list[ir.Type] + if has_bitfield and all_bitfield: + total_bits: int = sum(self.class_member_bitfields.get(ClassName, {}).get(name, 0) + for name, _ in all_members) + storage_type: ir.IntType + if total_bits <= 8: + storage_type = ir.IntType(8) + elif total_bits <= 16: + storage_type = ir.IntType(16) + elif total_bits <= 32: + storage_type = ir.IntType(32) + else: + storage_type = ir.IntType(64) + member_types = [storage_type] + bitfield_offsets: dict[str, int] = {} + current_bit_offset: int = 0 + for name, _ in all_members: + bit_width: int = self.class_member_bitfields.get(ClassName, {}).get(name, 0) + bitfield_offsets[name] = current_bit_offset + current_bit_offset += bit_width + self.class_member_bitoffsets[ClassName] = bitfield_offsets + else: + member_types = [] + if has_vtable: + member_types.append(ir.PointerType(ir.IntType(8))) + if ClassName in self.class_members: + for _, member_type in self.class_members[ClassName]: + if member_type is None: + member_type = ir.IntType(32) + member_types.append(member_type) + else: + if has_vtable: + # has_vtable 但 class_members 为空（跨模块导入类）： + # 保持 member_types 为空（[i8*] 占位会导致 stub 加载跳过， + # 因为 set_body 后 is_opaque=False）。让 struct 保持 opaque， + # 由 _TryLoadStructFromStub 设置完整 body。 + member_types = [] + else: + member_types = [ir.IntType(8)] + + # 获取之前创建的 struct 并设置成员 + struct_type: ir.IdentifiedStructType = self.structs[ClassName] + if ClassName in self.class_packed: + struct_type.packed = True + # 只在未设置 body 时设置，避免重复定义错误 + if struct_type.elements is None: + # 跨模块 struct 的 class_members 可能不完整（当前模块只解析了部分字段）， + # 优先从 stub.ll 加载完整定义。set_body 只能对 opaque struct 调用一次， + # 若用不完整 class_members 设置 body，后续无法修正，会导致 GEP 越界。 + # 但本地定义的类（_classes_in_progress）不从 stub 加载，因为 stub 可能 + # 来自同名导入类，其 vtable 状态与本地定义不同（@t.NoVTable 冲突） + import_handler: Any = self._import_handler_ref + if import_handler is not None and ClassName not in self._classes_in_progress: + try: + import_handler._TryLoadStructFromStub(ClassName, self) + except Exception: + pass + struct_type = self.structs.get(ClassName, struct_type) + if struct_type.elements is None: + # 防御性检查：member_types 为空时不调用 set_body。 + # set_body(*[]) 会设置 elements=() 使 is_opaque=False， + # 后续无法再次 set_body（llvmlite 限制），导致跨模块类型永久损坏。 + # 根因：register_method 在 class_members[ClassName]=[] 时立即调用 + # _generate_structs，此时 member_types 为空。保持 opaque 状态， + # 让后续 _TryLoadStructFromStub 能正确设置 body。 + if member_types: + struct_type.set_body(*member_types) + + def _create_Vtable_globals(self) -> None: + for ClassName, methods in self.class_methods.items(): + if ClassName in self.Vtables: + continue + if not methods: + continue + if ClassName not in self.class_vtable and ClassName not in self._cross_module_vtable_classes: + continue + # 跳过正在发射中的类：它们的方法列表可能还不完整（如继承方法已添加但自有方法未添加）， + # 此时创建 vtable 会导致大小不匹配。这些类的 vtable 会在其 _EmitClassLlvm 完成后 + # 由 _create_vtable_for_class 显式创建，或在模块级 _create_Vtable_globals 调用中创建。 + if ClassName in self._classes_in_progress: + continue + # 跨模块虚表过滤：_precollect_vtable_state 会将所有 includes 的有方法类同时加入 + # _shared_class_vtable 和 _shared_cross_module_vtable，导致 class_vtable 与 + # _cross_module_vtable_classes 内容相同，无法用以区分本模块类与跨模块类。 + # _generate_structs 会为所有 class_methods 的类创建空 struct type，导致 self.structs + # 也包含所有类。唯一可靠的过滤条件是 self.functions：_EmitClassLlvm 在本函数之前 + # 调用（LlvmGenerator.py L178-182），本模块定义的类方法已加入 self.functions； + # import 声明的跨模块类方法也在 self.functions 中。若 self.functions 中没有任何 + # 以 "{ClassName}." 开头的键，说明本模块既未定义也未导入该类的方法，虚表会保持 + # 全 null，虚表分派时调用 null 指针引发段错误。此时应跳过虚表创建，让链接器 + # 从定义该类的模块解析虚表符号。 + has_local_method: bool = False + prefix: str = f"{ClassName}." + for func_name in self.functions: + if func_name.startswith(prefix): + has_local_method = True + break + if not has_local_method: + continue + VtableType: ir.ArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + Vtable: ir.GlobalVariable = ir.GlobalVariable(self.module, VtableType, name=self._mangle_name(f"{ClassName}_Vtable")) + Vtable.initializer = ir.Constant(VtableType, [ir.Constant(ir.PointerType(ir.IntType(8)), None)] * len(methods)) + Vtable.linkage = 'internal' + self.Vtables[ClassName] = Vtable + + def _create_vtable_for_class(self, ClassName: str) -> None: + """为单个类创建 vtable（用于 _EmitClassLlvm 中当前类的方法已全部注册后）。""" + methods: list = self.class_methods.get(ClassName, []) + if not methods: + return + if ClassName not in self.class_vtable and ClassName not in self._cross_module_vtable_classes: + return + _vtable_name: str = self._mangle_name(f"{ClassName}_Vtable") + # 如果虚表已在导入阶段创建（方法列表不完整，缺少继承方法），删除旧虚表 + if ClassName in self.Vtables: + _old_vtable = self.Vtables[ClassName] + _old_name = _old_vtable.name + if _old_name in self.module.globals: + del self.module.globals[_old_name] + # llvmlite 的 NameScope._useset 追踪所有已使用的名字， + # del module.globals[name] 不会从 _useset 移除， + # 导致再次创建同名 GlobalVariable 时 scope.register() 报 DuplicatedNameError + self.module.scope._useset.discard(_old_name) + del self.Vtables[ClassName] + # 防御性：如果全局变量已存在但不在 Vtables 字典中，也删除 + if _vtable_name in self.module.globals: + del self.module.globals[_vtable_name] + self.module.scope._useset.discard(_vtable_name) + VtableType: ir.ArrayType = ir.ArrayType(ir.PointerType(ir.IntType(8)), len(methods)) + Vtable: ir.GlobalVariable = ir.GlobalVariable(self.module, VtableType, name=_vtable_name) + Vtable.initializer = ir.Constant(VtableType, [ir.Constant(ir.PointerType(ir.IntType(8)), None)] * len(methods)) + Vtable.linkage = 'internal' + self.Vtables[ClassName] = Vtable + + def _fix_global_var_init(self) -> None: + for gv_name, gv in list(self.module.globals.items()): + if not isinstance(gv, ir.GlobalVariable): + continue + if gv.linkage in ('external', 'extern_weak'): + continue + needs_fix: bool = False + if gv.initializer is None: + needs_fix = True + elif hasattr(gv.initializer, 'constant') and isinstance(gv.initializer.constant, _Undefined): + needs_fix = True + elif isinstance(gv.initializer, ir.Constant): + if isinstance(gv.initializer.constant, list): + if any(hasattr(c, 'constant') and isinstance(c.constant, _Undefined) for c in gv.initializer.constant): + needs_fix = True + if not needs_fix: + continue + gv.initializer = ir.Constant(gv.value_type, None) + if gv_name.startswith('str_const_') or gv_name.endswith('_Vtable') or gv_name.endswith('_str'): + gv.linkage = 'internal' + + def _set_Vtable_initializers(self) -> None: + for ClassName, methods in self.class_methods.items(): + if ClassName not in self.Vtables: + continue + if not methods: + continue + struct_type: ir.IdentifiedStructType | None = self.structs.get(ClassName) + if struct_type is None: + continue + Vtable: ir.GlobalVariable = self.Vtables[ClassName] + Vtable_entries: list[ir.Constant] = [] + for MethodName in methods: + short_name: str = MethodName.split('.')[-1] if '.' in MethodName else MethodName + full_MethodName: str = f"{ClassName}.{short_name}" + func: ir.Function | None = None + if full_MethodName in self.functions: + func = self.functions[full_MethodName] + if func is None: + parent_cls: str | None = self.class_parent.get(ClassName) + while parent_cls: + parent_full: str = f"{parent_cls}.{short_name}" + if parent_full in self.functions: + func = self.functions[parent_full] + break + parent_cls = self.class_parent.get(parent_cls) + if func is None: + Vtable_entries.append(ir.Constant(ir.PointerType(ir.IntType(8)), None)) + continue + Vtable_entries.append(func.bitcast(ir.PointerType(ir.IntType(8)))) + Vtable.initializer = ir.Constant(Vtable.type.pointee, Vtable_entries) \ No newline at end of file diff --git a/lib/core/LLVMCG/TypeConvert.py b/lib/core/LLVMCG/TypeConvert.py new file mode 100644 index 0000000..f32d5b9 --- /dev/null +++ b/lib/core/LLVMCG/TypeConvert.py @@ -0,0 +1,767 @@ +from __future__ import annotations +from typing import Any +import ast +import re +import logging +import llvmlite.ir as ir +from lib.includes import t as _t +from lib.includes.t import CTypeRegistry +from lib.core.Handles.HandlesBase import CTypeInfo as _CTypeInfo, BuiltinTypeMap +from lib.constants.config import mode as _config_mode +from lib.core.VLogger import get_logger as _vlog +from lib.core.SymbolUtils import IsTModule + + +class TypeConvertMixin: + + # 限定符前缀映射表：(前缀字符串, 去除长度) + # 这些限定符（const/volatile/static/extern/register/inline/export/CAuto 等） + # 仅影响 C 语义，对 LLVM IR 类型本身无影响，统一去除后递归解析内部类型。 + _QUALIFIER_PREFIXES: tuple[tuple[str, int], ...] = ( + ('const ', 6), ('volatile ', 9), ('const_', 6), ('volatile_', 9), + ('static ', 7), ('extern ', 7), ('register ', 9), ('inline ', 7), + ('export ', 7), + ('CStatic ', 8), ('CExtern ', 8), ('CRegister ', 10), + ('CInline ', 8), ('CAuto ', 6), + ) + + def _ctype_to_llvm(self, type_info: _CTypeInfo | str | None) -> ir.Type: + if type_info is None: + return ir.IntType(32) + + if isinstance(type_info, str): + return self._type_str_to_llvm(type_info) + + if not isinstance(type_info, _CTypeInfo): + return ir.IntType(32) + + if type_info.IsFuncPtr: + ret_type: ir.Type = self._ctype_to_llvm(type_info.FuncPtrReturn) if type_info.FuncPtrReturn else ir.VoidType() + param_types: list[ir.Type] = [] + if type_info.FuncPtrParams: + for _name, param_info in type_info.FuncPtrParams: + if isinstance(param_info, _CTypeInfo): + param_types.append(self._ctype_to_llvm(param_info)) + else: + param_types.append(ir.IntType(32)) + fn_type: ir.FunctionType = ir.FunctionType(ret_type, param_types) + return ir.PointerType(fn_type) + + if type_info.IsState and type_info.PtrCount == 0 and not type_info.IsTypedef: + if type_info.BaseType is None or (isinstance(type_info.BaseType, type) and issubclass(type_info.BaseType, _t.CVoid)) or isinstance(type_info.BaseType, _t.CVoid): + return ir.VoidType() + + if type_info.IsTypedef and type_info.Name: + resolved: ir.Type | None = self._resolve_typedef_ctype(type_info) + if resolved is not None: + return resolved + + # REnum 类型：通过 Name 查找或创建结构体（BaseType 可能为 None） + # 这对 BinOp `LLVMType | t.CPtr` 合并后的 CTypeInfo 至关重要 + if type_info.IsRenum and type_info.Name: + if type_info.Name in self.structs: + base: ir.Type = self.structs[type_info.Name] + else: + base = self._get_or_create_struct(type_info.Name) + result: ir.Type = base + for _ in range(type_info.PtrCount): + result = ir.PointerType(result) + return result + + BaseType: ir.Type = self._base_ctype_to_llvm(type_info.BaseType, type_info) + + result: ir.Type = BaseType + for _ in range(type_info.PtrCount): + if isinstance(result, ir.VoidType): + result = ir.IntType(8).as_pointer() + else: + result = ir.PointerType(result) + + for dim in reversed(type_info.ArrayDims): + try: + dim_val: int = int(dim) + if dim_val > 0: + result = ir.ArrayType(result, dim_val) + except (ValueError, TypeError): + pass + + return result + + def _resolve_typedef_ctype(self, type_info: _CTypeInfo) -> ir.Type | None: + name: str = type_info.Name + if not name or not self.SymbolTable: + return None + + entry: Any = self.SymbolTable.lookup(name) + if not entry or not isinstance(entry, _CTypeInfo): + return None + + if entry.BaseType and (not isinstance(entry.BaseType, _t._CTypedef) or entry.PtrCount > 0): + resolved: Any = entry.Copy() + resolved.IsTypedef = False + if type_info.PtrCount > resolved.PtrCount: + resolved.PtrCount = type_info.PtrCount + return self._ctype_to_llvm(resolved) + + if entry.OriginalType: + if isinstance(entry.OriginalType, _CTypeInfo): + resolved: Any = entry.OriginalType.Copy() + resolved.IsTypedef = False + if type_info.PtrCount > resolved.PtrCount: + resolved.PtrCount = type_info.PtrCount + return self._ctype_to_llvm(resolved) + elif isinstance(entry.OriginalType, str): + resolved: Any = _CTypeInfo.FromTypeName(entry.OriginalType) + if resolved and resolved.BaseType: + resolved.IsTypedef = False + if type_info.PtrCount > resolved.PtrCount: + resolved.PtrCount = type_info.PtrCount + return self._ctype_to_llvm(resolved) + + return None + + def _base_ctype_to_llvm(self, BaseType: Any, type_info: _CTypeInfo | None = None) -> ir.Type: + if BaseType is None: + return ir.VoidType() + + if isinstance(BaseType, (tuple, list)): + elem_types: list[ir.Type] = [self._base_ctype_to_llvm(bt, type_info) for bt in BaseType] + return ir.LiteralStructType(elem_types) if elem_types else ir.IntType(32) + + if isinstance(BaseType, type) and issubclass(BaseType, _t.CType): + BaseType = BaseType() + + if isinstance(BaseType, _t.CVoid): + return ir.VoidType() + + if isinstance(BaseType, _t.CStruct): + struct_name: str = getattr(BaseType, 'name', '') or (type_info.Name if type_info else '') + if struct_name and struct_name in self.structs: + return self.structs[struct_name] + if struct_name: + return self._get_or_create_struct(struct_name) + return ir.LiteralStructType([]) + + if isinstance(BaseType, (_t.CEnum, _t.REnum)): + enum_name: str = type_info.Name if type_info else '' + if enum_name and enum_name in self.structs: + return self.structs[enum_name] + return ir.IntType(32) + + if isinstance(BaseType, _t.CUnion): + union_name: str = type_info.Name if type_info else '' + if union_name and union_name in self.structs: + return self.structs[union_name] + if union_name: + return self._get_or_create_struct(union_name) + return ir.IntType(32) + + if isinstance(BaseType, _t._CTypedef): + typedef_name: str = getattr(BaseType, 'value', '') or (type_info.Name if type_info else '') + if typedef_name and self.SymbolTable: + entry: Any = self.SymbolTable.lookup(typedef_name) + if entry and isinstance(entry, _CTypeInfo) and entry.BaseType: + if not isinstance(entry.BaseType, _t._CTypedef) or entry.PtrCount > 0: + return self._base_ctype_to_llvm(entry.BaseType, entry) + return ir.IntType(32) + + if isinstance(BaseType, _t.CDefine): + # Use 64-bit type if the define value exceeds 32-bit range + if type_info and isinstance(type_info.DefineValue, int): + if type_info.DefineValue < 0 or type_info.DefineValue > 0xFFFFFFFF: + return ir.IntType(64) + return ir.IntType(32) + + if isinstance(BaseType, _t.CPass): + return ir.VoidType() + + llvm_str: str = CTypeRegistry.CTypeToLLVM(type(BaseType)) + return self._llvm_str_to_ir(llvm_str) + + def _get_source_sha1_from_type_str(self, type_str: str) -> str | None: + """从类型字符串中提取模块名，返回对应的源模块 SHA1。 + 用于跨模块类型引用（如 memhub.MemBuddy），确保用源模块的 SHA1 创建 struct。 + """ + if not type_str or '.' not in type_str: + return None + # 去除指针标记和空格 + clean: str = type_str.replace('*', '').strip() + if '.' not in clean: + return None + # 提取模块名（最后一个 '.' 之前的部分） + parts: list[str] = clean.rsplit('.', 1) + if len(parts) < 2: + return None + module_name: str = parts[0] + # 跳过 t/c 等内建模块 + if module_name in {'t', 'c'}: + return None + # 从 ModuleSha1Map 查找 SHA1（支持完整路径名和短名） + sha1_map: dict[str, str] = getattr(self, 'ModuleSha1Map', {}) + source_sha1: str | None = sha1_map.get(module_name) + if not source_sha1 and '.' in module_name: + source_sha1 = sha1_map.get(module_name.split('.')[-1]) + return source_sha1 + + def _resolve_typedef(self, type_name: str) -> str: + if type_name in {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum', 'CPtr'}: + return type_name + if type_name == 'Callable': + return 'Callable' + visited: set[str] = set() + current: str = type_name + while current not in visited: + visited.add(current) + if self.SymbolTable: + Entry: _CTypeInfo | None = self.SymbolTable.lookup(current) + if isinstance(Entry, dict) and Entry.get('type') == 'typedef': + OriginalType: str = Entry.get('OriginalType', '') + if OriginalType: + current = OriginalType + continue + elif Entry and Entry.IsTypedef: + if Entry.OriginalType: + if isinstance(Entry.OriginalType, _CTypeInfo): + if Entry.OriginalType.IsFuncPtr: + return 'Callable' + return Entry.OriginalType.ToString() if Entry.OriginalType.BaseType else current + elif isinstance(Entry.OriginalType, str): + current = Entry.OriginalType + continue + if Entry.BaseType: + if not isinstance(Entry.BaseType, (_t._CTypedef,)) or Entry.PtrCount > 0: + resolved: str = self._resolve_ctype_to_str(Entry) + if resolved and resolved != current: + current = resolved + continue + bm: Any = BuiltinTypeMap.Get(current) + if bm: + base_cls: Any = bm[0] + try: + instance: Any = base_cls() + cname: str = type(instance).__name__ + if cname: + current = cname + ' *' * bm[1] + continue + except Exception as _e: + if _config_mode == "strict": + logging.warning(f"异常被忽略: {_e}") + pass + bm: Any = BuiltinTypeMap.Get(current) + if bm and bm[1] > 0: + base_cls: Any = bm[0] + try: + instance: Any = base_cls() + cname: str = type(instance).__name__ + if cname: + current = cname + ' *' * bm[1] + continue + except Exception as _e: + if _config_mode == "strict": + logging.warning(f"异常被忽略: {_e}") + pass + break + return current + + def _resolve_ctype_to_str(self, type_info: _CTypeInfo) -> str: + parts: list[str] = [] + base: Any = type_info.BaseType + if base is None or base is _t.CVoid or isinstance(base, _t.CVoid): + parts.append('void') + elif isinstance(base, _t._CTypedef) or base is _t._CTypedef: + return '' + elif isinstance(base, _t.CStruct) or base is _t.CStruct: + parts.append(f'struct {getattr(base, "name", "")}') + elif isinstance(base, _t.CEnum) or base is _t.CEnum: + parts.append(f'enum {getattr(base, "name", "")}') + else: + cname: str = '' + if isinstance(base, type) and issubclass(base, _t.CType): + cname = base.__name__ + elif isinstance(base, _t.CType): + cname = type(base).__name__ + if cname: + parts.append(cname) + else: + return '' + if type_info.PtrCount > 0: + parts.append('*' * type_info.PtrCount) + return ' '.join(parts) + + def _type_str_to_llvm(self, type_str: str, IsPtr: bool = False) -> ir.Type: + type_str = type_str.strip() + cache_key: tuple[str, bool] = (type_str, IsPtr) + cached: ir.Type | None = self._type_cache.get(cache_key) + if cached is not None: + return cached + result: ir.Type = self._type_str_to_llvm_impl(type_str, IsPtr) + if result is not None: + if isinstance(result, (ir.IntType, ir.FloatType, ir.DoubleType, ir.VoidType)): + self._type_cache[cache_key] = result + elif isinstance(result, ir.PointerType) and isinstance(result.pointee, (ir.IntType, ir.FloatType, ir.DoubleType, ir.VoidType)): + self._type_cache[cache_key] = result + elif isinstance(result, ir.ArrayType) and not isinstance(result.element, (ir.IdentifiedStructType, ir.LiteralStructType)): + self._type_cache[cache_key] = result + return result + + _CType2LLVM = _type_str_to_llvm + + def _type_str_to_llvm_impl(self, type_str: str, IsPtr: bool = False) -> ir.Type: + array_dims: list[int] = [] + dim_match: list[str] = re.findall(r'\[(\d+)\]', type_str) + if dim_match: + array_dims = [int(d) for d in dim_match] + type_str = re.sub(r'\s*\[\d+\]', '', type_str).strip() + result: ir.Type = self._type_str_to_llvm_base(type_str, IsPtr) + for dim in reversed(array_dims): + result = ir.ArrayType(result, dim) + return result + + def _type_str_to_llvm_base(self, type_str: str, IsPtr: bool = False) -> ir.Type: + ptr_depth: int = 0 + while type_str.endswith('*'): + ptr_depth += 1 + type_str = type_str[:-1].strip() + if ptr_depth > 0: + IsPtr = True + if type_str in {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum'}: + result: ir.Type + if IsPtr: + result = ir.IntType(8).as_pointer() + for _ in range(ptr_depth - 1): + result = ir.PointerType(result) + return result + return ir.IntType(32) + if type_str in {'const', 'volatile'}: + return ir.IntType(8) + if type_str == 'Callable': + return ir.IntType(8).as_pointer() + for _q_prefix, _q_offset in self._QUALIFIER_PREFIXES: + if type_str.startswith(_q_prefix): + return self._type_str_to_llvm(type_str[_q_offset:].strip(), IsPtr) + resolved: str = self._resolve_typedef(type_str) + if resolved != type_str: + if '*' in resolved: + IsPtr = False + return self._type_str_to_llvm(resolved, IsPtr) + if type_str.endswith('*'): + BaseType: str = type_str[:-1].strip() + base_llvm: ir.Type | None = self._basic_type_to_llvm(BaseType) + if base_llvm is not None and isinstance(base_llvm, ir.VoidType): + result: ir.Type = ir.IntType(8).as_pointer() + for _ in range(ptr_depth): + result = ir.PointerType(result) + return result + if base_llvm is not None and not isinstance(base_llvm, ir.VoidType): + if isinstance(base_llvm, ir.PointerType): + return base_llvm + return ir.PointerType(base_llvm) + if BaseType in self.structs: + return ir.PointerType(self.structs[BaseType]) + if BaseType.startswith('struct '): + stripped: str = BaseType[7:].strip() + if stripped in self.structs: + return ir.PointerType(self.structs[stripped]) + if '.' in BaseType: + LastPart: str = BaseType.split('.')[-1] + if LastPart in self.structs: + return ir.PointerType(self.structs[LastPart]) + _src_sha1: str | None = self._get_source_sha1_from_type_str(BaseType) + placeholder: ir.IdentifiedStructType = self._get_or_create_struct(LastPart, source_sha1=_src_sha1) + return ir.PointerType(placeholder) + if base_llvm is not None and isinstance(base_llvm, ir.VoidType): + return ir.PointerType(ir.IntType(8)) + if type_str.startswith('{') and type_str.endswith('}'): + inner: str = type_str[1:-1].strip() + elem_strs: list[str] = [] + depth: int = 0 + current: str = '' + for ch in inner: + if ch == ',' and depth == 0: + elem_strs.append(current.strip()) + current = '' + else: + if ch == '{': + depth += 1 + elif ch == '}': + depth -= 1 + current += ch + if current.strip(): + elem_strs.append(current.strip()) + elem_types: list[ir.Type] = [] + for es in elem_strs: + et: ir.Type = self._type_str_to_llvm(es, '*' in es) + if isinstance(et, ir.VoidType): + et = ir.IntType(8).as_pointer() + elem_types.append(et) + if elem_types: + return ir.LiteralStructType(elem_types) + return ir.IntType(32) + if type_str in {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum'}: + return ir.IntType(32) + if type_str.startswith('%'): + stripped: str = type_str[1:] + if stripped in {'typedef', 'CType', 'CVolatile', 'CEnum', 'CUnion', 'CStruct', 'enum', 'REnum', 'renum'}: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + basic_stripped: ir.Type | None = self._basic_type_to_llvm(stripped) + if basic_stripped is not None: + if IsPtr and not isinstance(basic_stripped, ir.VoidType): + return ir.PointerType(basic_stripped) + return basic_stripped + if self.SymbolTable: + _Entry: _CTypeInfo | None = self.SymbolTable.lookup(type_str) + if _Entry and _Entry.IsTypedef: + resolved_str: str = self._resolve_typedef(type_str) + if resolved_str != type_str: + return self._type_str_to_llvm(resolved_str, IsPtr) + if _Entry.BaseType: + if not isinstance(_Entry.BaseType, (_t._CTypedef,)) or _Entry.PtrCount > 0: + resolved: str = self._resolve_ctype_to_str(_Entry) + if resolved: + return self._type_str_to_llvm(resolved, IsPtr) + if _Entry and _Entry.IsRenum: + if type_str in self.structs: + if IsPtr: + return ir.PointerType(self.structs[type_str]) + return self.structs[type_str] + # REnum 结构体尚未创建时不能 fall through 到 IsEnum（会返回 i32）， + # 需要通过 _get_or_create_struct 创建 opaque 结构体（自引用安全）。 + _renum_sha1: str | None = self._get_source_sha1_from_type_str(type_str) + opaque: ir.Type = self._get_or_create_struct(type_str, source_sha1=_renum_sha1) + if IsPtr: + return ir.PointerType(opaque) + return opaque + if _Entry and _Entry.IsStruct: + # 结构体类型（来自 _InsertClassSymbol）：必须返回结构体类型而非 i32， + # 否则 (ast.Import | t.CPtr) 联合转换会得到 i32* 导致字段访问 GEP 偏移错误 + if type_str in self.structs: + if IsPtr: + return ir.PointerType(self.structs[type_str]) + return self.structs[type_str] + # 结构体尚未创建时，通过 _get_or_create_struct 创建 opaque 结构体 + # （_TryLoadStructFromStub 会在后续阶段填充字段） + _struct_sha1: str | None = self._get_source_sha1_from_type_str(type_str) + opaque_struct: ir.IdentifiedStructType = self._get_or_create_struct(type_str, source_sha1=_struct_sha1) + if IsPtr: + return ir.PointerType(opaque_struct) + return opaque_struct + if _Entry and _Entry.IsEnum: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + if _Entry and _Entry.IsExceptionClass: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + if _Entry and _Entry.IsFunction: + return ir.PointerType(ir.IntType(8)) + if _Entry and _Entry.BaseType: + if (isinstance(_Entry.BaseType, type) and issubclass(_Entry.BaseType, _t.CEnum)) or _Entry.BaseType is _t.CEnum: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + for ClassName in self.structs: + if type_str == ClassName: + if IsPtr: + return ir.PointerType(self.structs[ClassName]) + return self.structs[ClassName] + if '.' in type_str and not type_str.startswith('%'): + LastPart: str = type_str.split('.')[-1] + basic_last: ir.Type | None = self._basic_type_to_llvm(LastPart) + if basic_last is not None: + if IsPtr and not isinstance(basic_last, ir.VoidType): + return ir.PointerType(basic_last) + return basic_last + if self.SymbolTable: + _Entry: _CTypeInfo | None = self.SymbolTable.lookup(LastPart) + if _Entry and _Entry.IsTypedef: + resolved_str: str = self._resolve_typedef(LastPart) + if resolved_str != LastPart: + return self._type_str_to_llvm(resolved_str, IsPtr) + if _Entry.BaseType: + if not isinstance(_Entry.BaseType, (_t._CTypedef,)) or _Entry.PtrCount > 0: + resolved: str = self._resolve_ctype_to_str(_Entry) + if resolved: + return self._type_str_to_llvm(resolved, IsPtr) + if _Entry and _Entry.IsRenum: + if LastPart in self.structs: + if IsPtr: + return ir.PointerType(self.structs[LastPart]) + return self.structs[LastPart] + # REnum 结构体尚未创建时不能 fall through 到 IsEnum（会返回 i32）。 + _lp_sha1: str | None = self._get_source_sha1_from_type_str(type_str) + opaque_lp: ir.Type = self._get_or_create_struct(LastPart, source_sha1=_lp_sha1) + if IsPtr: + return ir.PointerType(opaque_lp) + return opaque_lp + if _Entry and _Entry.IsEnum: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + if _Entry and _Entry.IsExceptionClass: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + if _Entry and _Entry.IsFunction: + return ir.PointerType(ir.IntType(8)) + if _Entry and _Entry.BaseType: + if (isinstance(_Entry.BaseType, type) and issubclass(_Entry.BaseType, _t.CEnum)) or _Entry.BaseType is _t.CEnum: + return ir.IntType(32) if not IsPtr else ir.PointerType(ir.IntType(32)) + if LastPart in self.structs: + if IsPtr: + return ir.PointerType(self.structs[LastPart]) + return self.structs[LastPart] + _lp_sha1_2: str | None = self._get_source_sha1_from_type_str(type_str) + placeholder: ir.IdentifiedStructType = self._get_or_create_struct(LastPart, source_sha1=_lp_sha1_2) + if IsPtr: + return ir.PointerType(placeholder) + return placeholder + if type_str.startswith('struct '): + stripped: str = type_str[7:].strip().replace(' *', '').replace('*', '') + if stripped in self.structs: + if IsPtr: + return ir.PointerType(self.structs[stripped]) + return self.structs[stripped] + basic_match: ir.Type | None = self._basic_type_to_llvm(stripped) + if basic_match is not None: + if IsPtr and not isinstance(basic_match, ir.VoidType): + return ir.PointerType(basic_match) + return basic_match + _stripped_sha1: str | None = self._get_source_sha1_from_type_str(stripped) + placeholder: ir.IdentifiedStructType = self._get_or_create_struct(stripped, source_sha1=_stripped_sha1) + if IsPtr: + return ir.PointerType(placeholder) + return placeholder + if type_str.startswith('%struct.'): + stripped: str = type_str[8:].strip().replace(' *', '').replace('*', '') + if stripped in self.structs: + if IsPtr: + return ir.PointerType(self.structs[stripped]) + return self.structs[stripped] + basic_match: ir.Type | None = self._basic_type_to_llvm(stripped) + if basic_match is not None: + if IsPtr and not isinstance(basic_match, ir.VoidType): + return ir.PointerType(basic_match) + return basic_match + _stripped_sha1_2: str | None = self._get_source_sha1_from_type_str(stripped) + placeholder: ir.IdentifiedStructType = self._get_or_create_struct(stripped, source_sha1=_stripped_sha1_2) + if IsPtr: + return ir.PointerType(placeholder) + return placeholder + if type_str.startswith('union '): + stripped: str = type_str[6:].strip() + basic_match: ir.Type | None = self._basic_type_to_llvm(stripped) + if basic_match is not None: + if IsPtr and not isinstance(basic_match, ir.VoidType): + return ir.PointerType(basic_match) + return basic_match + if type_str.endswith('*'): + pointee: ir.Type = self._type_str_to_llvm(type_str[:-1].strip()) + if isinstance(pointee, ir.VoidType): + return ir.PointerType(ir.IntType(8)) + return ir.PointerType(pointee) + result: ir.Type | None = self._basic_type_to_llvm(type_str) + if result is None: + # 未知类型：可能是跨模块结构体（如 'Import' 来自 ast 模块）。 + # 尝试通过 _get_or_create_struct 创建/查找结构体， + # 该方法内部会检查 typedef/enum 等并返回正确类型。 + # 必须在 fallback 到 i32 之前执行，否则 (ast.Import | t.CPtr) 联合转换 + # 会得到 i32* 导致字段访问 GEP 偏移错误（AccessViolation 崩溃）。 + _cand_sha1: str | None = self._get_source_sha1_from_type_str(type_str) + candidate: ir.Type = self._get_or_create_struct(type_str, source_sha1=_cand_sha1) + if isinstance(candidate, ir.IdentifiedStructType): + result = candidate + else: + result = ir.IntType(32) + if IsPtr: + if isinstance(result, ir.VoidType): + result = ir.PointerType(ir.IntType(8)) + elif isinstance(result, ir.PointerType): + pass + else: + result = ir.PointerType(result) + for _ in range(ptr_depth - 1): + result = ir.PointerType(result) + return result + + def _llvm_str_to_ir(self, llvm_str: str) -> ir.Type | None: + _LLVM_IR_TYPES: dict[str, ir.Type] = { + 'void': ir.VoidType(), 'i1': ir.IntType(1), 'i8': ir.IntType(8), + 'i16': ir.IntType(16), 'i32': ir.IntType(32), 'i64': ir.IntType(64), + 'i128': ir.IntType(128), 'float': ir.FloatType(), 'double': ir.DoubleType(), + 'half': ir.IntType(16), 'fp128': ir.IntType(128), + } + return _LLVM_IR_TYPES.get(llvm_str) + + def _basic_type_to_llvm(self, type_str: str) -> ir.Type | None: + _LLVM_IR_TYPES: dict[str, ir.Type] = { + 'void': ir.VoidType(), 'i1': ir.IntType(1), 'i8': ir.IntType(8), + 'i16': ir.IntType(16), 'i32': ir.IntType(32), 'i64': ir.IntType(64), + 'i128': ir.IntType(128), 'f32': ir.FloatType(), 'f64': ir.DoubleType(), + 'half': ir.IntType(16), 'fp128': ir.IntType(128), + } + if type_str in _LLVM_IR_TYPES: + return _LLVM_IR_TYPES[type_str] + resolved: Any = CTypeRegistry.ResolveName(type_str) + if resolved is not None: + ctype_cls: Any + ptr_level: int + ctype_cls, ptr_level = resolved + llvm_str: str = CTypeRegistry.CTypeToLLVM(ctype_cls) + if llvm_str: + # CTypeToLLVM 对 CPtr 等指针类型已返回含 * 的串（如 'i8*'）， + # 此时 ptr_level 已内含在 llvm_str 中，直接解析即可 + if '*' in llvm_str: + return self._type_str_to_llvm(llvm_str) + base: ir.Type | None = self._llvm_str_to_ir(llvm_str) + if base is not None: + for _ in range(ptr_level): + if isinstance(base, ir.VoidType): + base = ir.IntType(8).as_pointer() + else: + base = ir.PointerType(base) + return base + if type_str == '*': + return ir.IntType(8) + if type_str in ('const_', 'const'): + return ir.IntType(8) + return None + + def _is_type_unsigned(self, type_str: str) -> bool: + s: str = type_str.strip() + if s.endswith('*'): + return True + if s.startswith('unsigned'): + return True + resolved: Any = CTypeRegistry.ResolveName(s) + if resolved is not None: + ctype_cls: Any + _: int + ctype_cls, _ = resolved + try: + inst: Any = ctype_cls() + if getattr(inst, 'IsSigned', None) is False: + return True + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"判断类型符号性失败: {_e}", "Exception") + return False + + def _record_var_signedness(self, name: str, type_str_or_bool: str | bool) -> None: + if isinstance(type_str_or_bool, bool): + self.var_signedness[name] = type_str_or_bool + else: + self.var_signedness[name] = self._is_type_unsigned(type_str_or_bool) + + def _is_var_unsigned(self, name: str) -> bool: + return self.var_signedness.get(name, False) + + def _check_node_unsigned(self, node: ast.AST) -> bool: + if isinstance(node, ast.Name): + return self._is_var_unsigned(node.id) + if isinstance(node, ast.BinOp): + return self._check_node_unsigned(node.left) or self._check_node_unsigned(node.right) + if isinstance(node, ast.UnaryOp): + return self._check_node_unsigned(node.operand) + if isinstance(node, ast.Call): + return self._check_call_unsigned(node) + if isinstance(node, ast.Subscript): + return self._check_node_unsigned(node.value) + if isinstance(node, ast.Attribute): + # 检查是否为结构体字段，并查询其符号性 + field_signed: bool | None = self._check_attribute_signedness(node) + if field_signed is not None: + return not field_signed # IsSigned=False 表示无符号 → 返回 True + return self._check_node_unsigned(node.value) + return False + + def _check_call_unsigned(self, node: ast.Call) -> bool: + """推断函数调用返回值的无符号性。 + + 修复：原代码对所有 Call 节点返回 False，导致函数调用返回值 + 直接参与 >>/% 等运算时丢失无符号性，退化成 ashr/srem。 + + 规则（用户确认）： + - a = b 赋值时，若 a 无显式类型注解，则自动推导为 b 的类型（含符号性） + - 若 a 有显式有符号类型，b 无符号，则相当于将 b 强转为 a 的有符号类型 + - 此处仅推断 b（调用本身）的符号性，赋值时的强转由赋值逻辑处理 + """ + func_node: ast.AST = node.func + func_name: str = '' + + if isinstance(func_node, ast.Name): + # 直接调用: rdrand64() + func_name = func_node.id + elif isinstance(func_node, ast.Attribute): + # t.CUInt64T(x) 等类型转换调用 + if hasattr(func_node.value, 'id') and IsTModule(func_node.value.id, self.SymbolTable): + if self._is_type_unsigned(func_node.attr): + return True + # module.func() 形式的调用 + try: + func_name = f"{func_node.value.id}.{func_node.attr}" + except AttributeError: + return False + else: + return False + + if not func_name or not self.SymbolTable: + return False + + try: + sym_info: Any = self.SymbolTable.lookup(func_name) + except Exception: + return False + + if sym_info and isinstance(sym_info, _CTypeInfo): + if sym_info.IsFunction or sym_info.IsFuncPtr: + ret_info: Any = sym_info.FuncPtrReturn + if ret_info and isinstance(ret_info, _CTypeInfo): + # 指针类型视为无符号 + if ret_info.PtrCount > 0: + return True + # 检查返回类型的 IsUInt 属性（IsSigned is False） + if ret_info.IsUInt: + return True + elif sym_info.IsTypedef: + # 修复：typedef 强转调用（如 ULONGLONG(0x11)）需要解析底层类型的无符号性 + # IsUInt 对组合类型（CType | CType 产生的 list）会失败，需多策略回退 + if sym_info.IsUInt: + return True + # 策略1：直接检查 BaseType（可能是组合类型，如 ULONGLONG = CUnsignedLong | CLong） + bt: Any = sym_info.BaseType + if isinstance(bt, (tuple, list)): + for elem in bt: + if isinstance(elem, _t.CType) and getattr(elem, 'IsSigned', None) is False: + return True + elif isinstance(bt, _t.CType) and getattr(bt, 'IsSigned', None) is False: + return True + # 策略2：通过 OriginalType 解析 + if sym_info.OriginalType: + ot: Any = sym_info.OriginalType + if isinstance(ot, str): + if self._is_type_unsigned(ot): + return True + elif isinstance(ot, _CTypeInfo): + if ot.IsUInt: + return True + elif isinstance(ot, (tuple, list)): + for elem in ot: + if isinstance(elem, _t.CType) and getattr(elem, 'IsSigned', None) is False: + return True + # 策略3：通过 _resolve_typedef 解析为字符串后判断 + resolved_str: str = self._resolve_typedef(func_name) + if resolved_str and self._is_type_unsigned(resolved_str): + return True + return False + + def _check_attribute_signedness(self, node: ast.Attribute) -> bool | None: + """检查 Attribute 节点是否引用具有已知符号性的结构体字段。 + + Returns: + True: 字段是有符号类型 + False: 字段是无符号类型 + None: 无法确定（非结构体字段或符号性未知） + """ + if not isinstance(node.value, ast.Name): + return None + var_name: str = node.value.id + field_name: str = node.attr + class_name: str | None = self._resolve_class_for_var(var_name) + if not class_name: + return None + signed_dict: dict[str, bool] = self.class_member_signeds.get(class_name, {}) + if field_name in signed_dict: + return signed_dict[field_name] + return None \ No newline at end of file diff --git a/lib/core/LLVMCG/VaArg.py b/lib/core/LLVMCG/VaArg.py new file mode 100644 index 0000000..b349639 --- /dev/null +++ b/lib/core/LLVMCG/VaArg.py @@ -0,0 +1,94 @@ +from __future__ import annotations +import llvmlite.ir as ir +from lib.core.LLVMCG.BaseGen import VaArgInstruction + + +class VaArgMixin: + + def _get_va_start_intrinsic(self) -> ir.Function: + if 'llvm.va_start' in self.functions: + return self.functions['llvm.va_start'] + ftype: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.IntType(8).as_pointer()]) + func: ir.Function = ir.Function(self.module, ftype, name='llvm.va_start') + self.functions['llvm.va_start'] = func + return func + + def _get_va_end_intrinsic(self) -> ir.Function: + if 'llvm.va_end' in self.functions: + return self.functions['llvm.va_end'] + ftype: ir.FunctionType = ir.FunctionType(ir.VoidType(), [ir.IntType(8).as_pointer()]) + func: ir.Function = ir.Function(self.module, ftype, name='llvm.va_end') + self.functions['llvm.va_end'] = func + return func + + def emit_va_start(self, va_list_ptr: ir.Value) -> None: + if not self.builder: + return None + func: ir.Function = self._get_va_start_intrinsic() + i8ptr_type: ir.PointerType = ir.IntType(8).as_pointer() + if va_list_ptr.type == i8ptr_type: + self.builder.call(func, [va_list_ptr]) + else: + ptr: ir.Value = self.builder.bitcast(va_list_ptr, i8ptr_type, name="va_list_i8ptr") + self.builder.call(func, [ptr]) + return None + + def _emit_stack_align(self, alignment: int = 16) -> None: + if not self.builder: + return None + void: ir.VoidType = ir.VoidType() + asm_type: ir.FunctionType = ir.FunctionType(void, []) + mask: int = -alignment + asm_str: str = f"andq $${mask}, %rsp" + asm_func: ir.InlineAsm = ir.InlineAsm(asm_type, asm_str, "~{dirflag},~{fpsr},~{flags},~{rsp}", side_effect=True) + self.builder.call(asm_func, [], name="align_stack") + return None + + def emit_va_end(self, va_list_ptr: ir.Value) -> None: + if not self.builder: + return None + func: ir.Function = self._get_va_end_intrinsic() + i8ptr_type: ir.PointerType = ir.IntType(8).as_pointer() + if va_list_ptr.type == i8ptr_type: + self.builder.call(func, [va_list_ptr]) + else: + ptr: ir.Value = self.builder.bitcast(va_list_ptr, i8ptr_type, name="va_list_i8ptr") + self.builder.call(func, [ptr]) + return None + + def emit_va_arg(self, va_list_ptr: ir.Value, arg_type: ir.Type) -> ir.Value: + if not self.builder: + return self._ZeroConst(arg_type) + if not hasattr(self, '_va_arg_counter'): + self._va_arg_counter: int = 0 + self._va_arg_counter += 1 + is_x86_64: bool = self._platform_info['is_x86_64'] + if is_x86_64: + va_arg_type: ir.Type + if isinstance(arg_type, ir.PointerType): + va_arg_type = ir.IntType(64) + elif isinstance(arg_type, (ir.FloatType, ir.DoubleType)): + va_arg_type = ir.DoubleType() + elif isinstance(arg_type, ir.IntType) and arg_type.width < 64: + va_arg_type = ir.IntType(64) + else: + va_arg_type = arg_type + instr: ir.Instruction = VaArgInstruction(self.builder.block, va_arg_type, 'va_arg', (va_list_ptr,), name=f'va_arg_raw_{self._va_arg_counter}') + self.builder._insert(instr) + result: ir.Value + if va_arg_type != arg_type: + if isinstance(arg_type, ir.PointerType): + result = self.builder.inttoptr(instr, arg_type, name=f'va_arg_ptr_{self._va_arg_counter}') + elif isinstance(arg_type, ir.FloatType): + result = self.builder.fptrunc(instr, arg_type, name=f'va_arg_fptrunc_{self._va_arg_counter}') + else: + result = self.builder.trunc(instr, arg_type, name=f'va_arg_trunc_{self._va_arg_counter}') + else: + result = instr + return result + else: + if isinstance(arg_type, ir.IntType) and arg_type.width < 64: + arg_type = ir.IntType(64) + instr: ir.Instruction = VaArgInstruction(self.builder.block, arg_type, 'va_arg', (va_list_ptr,), name=f'va_arg_raw_{self._va_arg_counter}') + self.builder._insert(instr) + return instr \ No newline at end of file diff --git a/lib/core/LLVMCG/__init__.py b/lib/core/LLVMCG/__init__.py new file mode 100644 index 0000000..6b231e9 --- /dev/null +++ b/lib/core/LLVMCG/__init__.py @@ -0,0 +1,34 @@ +""" +LLVMCG 包 - LLVM 代码生成器模块 + +将 LlvmCodeGenerator.py 拆分为多个 Mixin 模块： +- BaseGen: 基础设施（__init__、平台解析、名称修饰、函数查找） +- TypeConvert: 类型转换（CType -> LLVM IR 类型） +- StructGen: 结构体生成（struct/vtable 定义与初始化） +- MemoryOps: 内存操作（alloca/Load/store/coerce） +- ExprGen: 表达式生成（常量、二元运算、返回、printf） +- FuncGen: 函数处理（参数调整、成员偏移、函数声明） +- VaArg: 变长参数处理（va_start/va_end/va_arg） +""" + +from lib.core.LLVMCG.BaseGen import ( + BaseGenMixin, + VaArgInstruction, +) +from lib.core.LLVMCG.TypeConvert import TypeConvertMixin +from lib.core.LLVMCG.StructGen import StructGenMixin +from lib.core.LLVMCG.MemoryOps import MemoryOpsMixin +from lib.core.LLVMCG.ExprGen import ExprGenMixin +from lib.core.LLVMCG.FuncGen import FuncGenMixin +from lib.core.LLVMCG.VaArg import VaArgMixin + +__all__ = [ + 'BaseGenMixin', + 'TypeConvertMixin', + 'StructGenMixin', + 'MemoryOpsMixin', + 'ExprGenMixin', + 'FuncGenMixin', + 'VaArgMixin', + 'VaArgInstruction', +] diff --git a/lib/core/LLVMTypeMapper.py b/lib/core/LLVMTypeMapper.py new file mode 100644 index 0000000..0e8cbe9 --- /dev/null +++ b/lib/core/LLVMTypeMapper.py @@ -0,0 +1,301 @@ +"""LLVMTypeMapper — AST 注解节点到 LLVM IR 类型字符串的映射器 + +从 SymbolTable 中提取的类型解析逻辑，职责单一： +将 Python AST 注解节点翻译为 LLVM IR 类型字符串（如 'i32', 'i8*', '{ i32, i8* }'）。 + +原 SymbolTable 上的 4 个方法： + - _GetLLVMTypeStr → LLVMTypeMapper.get_llvm_type_str + - _GetFuncRetTypeStr → LLVMTypeMapper.get_func_ret_type_str + - _GetFuncParamTypeStr → LLVMTypeMapper.get_func_param_type_str + - _ResolveTypedefValueType → LLVMTypeMapper.resolve_typedef_value_type +""" +from __future__ import annotations + +import ast +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + +from lib.core.Handles.HandlesBase import CTypeInfo, CTypeHelper, BuiltinTypeMap +from lib.core.SymbolUtils import IsTModule, IsListAnnotation +from lib.includes import t as _t +from lib.includes.t import CTypeRegistry + + +class LLVMTypeMapper: + """AST 注解节点 → LLVM IR 类型字符串 映射器""" + + def __init__(self, symtab: SymbolTable): + self._symtab = symtab + + # ================================================================== + # 公共 API + # ================================================================== + + def get_llvm_type_str(self, node: ast.AST | None) -> str: + """将 AST 注解节点转换为 LLVM IR 类型字符串""" + if node is None: + self._symtab.diagnostics.warn('', 0, "类型注解为 None，回退到 i32") + return 'i32' + if isinstance(node, ast.Name): + return self._resolve_name(node) + elif isinstance(node, ast.Attribute): + return self._resolve_attribute(node) + elif isinstance(node, ast.Subscript): + return self._resolve_subscript(node) + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + return self._resolve_bitor(node) + elif isinstance(node, ast.Constant): + if isinstance(node.value, bool): + return 'i8' + return 'i32' + elif isinstance(node, ast.Call): + return self._resolve_call(node) + self._symtab.diagnostics.warn('', 0, f"无法解析类型节点 {type(node).__name__}，回退到 i32") + return 'i32' + + def get_func_ret_type_str(self, returns_node: ast.AST | None) -> str: + """提取函数返回类型的 LLVM 字符串""" + if returns_node is None: + return 'i32' + if isinstance(returns_node, ast.BinOp) and isinstance(returns_node.op, ast.BitOr): + left_str: str = self.get_llvm_type_str(returns_node.left) + right_str: str = self.get_llvm_type_str(returns_node.right) + if right_str and right_str.endswith('*'): + return right_str + if left_str and left_str.endswith('*'): + return left_str + if left_str and left_str not in ('i32', 'void', 'i64'): + return left_str + if right_str and right_str not in ('i32', 'void', 'i64'): + return right_str + return left_str + result: str = self.get_llvm_type_str(returns_node) + return result if result else 'i32' + + def get_func_param_type_str(self, annotation_node: ast.AST | None) -> str: + """提取函数参数类型的 LLVM 字符串""" + if annotation_node is None: + return 'i8*' + if isinstance(annotation_node, ast.BinOp) and isinstance(annotation_node.op, ast.BitOr): + left_str: str = self.get_llvm_type_str(annotation_node.left) + right_str: str = self.get_llvm_type_str(annotation_node.right) + if right_str and right_str.endswith('*'): + return right_str + if left_str and left_str.endswith('*'): + return left_str + if right_str and right_str not in ('i32', 'void', 'i64'): + return right_str + if left_str and left_str not in ('i32', 'void', 'i64'): + return left_str + return right_str if right_str else left_str + result: str = self.get_llvm_type_str(annotation_node) + return result if result else 'i8*' + + def resolve_typedef_value_type(self, node: ast.AST | None) -> str: + """解析 typedef 值表达式的 C 类型字符串""" + if node is None: + return '' + if isinstance(node, ast.Name): + if node.id in self._symtab: + Entry: CTypeInfo = self._symtab[node.id] + if Entry.IsTypedef and Entry.OriginalType: + return Entry.OriginalType + return '' + if isinstance(node, ast.Attribute): + if isinstance(node.value, ast.Name) and IsTModule(node.value.id): + # 直接判断属性名是否为 CPtr（消除对 CName=='*' 的依赖） + if node.attr == 'CPtr': + return 'ptr' + CName: str | None = CTypeHelper.GetCName(node.attr) + if CName == 'CPtr': + return 'ptr' + if CName: + return CName + return '' + if isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + left_str: str = self.resolve_typedef_value_type(node.left) + right_str: str = self.resolve_typedef_value_type(node.right) + if right_str == 'ptr' or (right_str and right_str.endswith('*')): + base: str = left_str if left_str and left_str != 'ptr' else '' + return base + ' *' if base else 'void *' + if left_str == 'ptr' or (left_str and left_str.endswith('*')): + base: str = right_str if right_str and right_str != 'ptr' else '' + return base + ' *' if base else 'void *' + if right_str and right_str not in ('ptr', 'void', 'i32', 'i64'): + return right_str + if left_str and left_str not in ('ptr', 'void', 'i32', 'i64'): + return left_str + return left_str if left_str else right_str + return '' + + # ================================================================== + # 内部解析方法 + # ================================================================== + + def _ctype_to_llvm_str(self, info: CTypeInfo) -> str: + """从 CTypeInfo 的 BaseType+PtrCount 生成归一化的 LLVM 类型字符串。 + + void* 在 LLVM IR 中归一化为 i8*；struct/enum/union 等非基础类型返回空字符串。 + """ + if not info or not info.BaseType: + return '' + base_type = info.BaseType + base_llvm: str = '' + if isinstance(base_type, type) and issubclass(base_type, _t.CType): + base_llvm = CTypeRegistry.CTypeToLLVM(base_type) + elif isinstance(base_type, _t.CType): + base_llvm = CTypeRegistry.CTypeToLLVM(type(base_type)) + if not base_llvm: + return '' + if info.PtrCount > 0: + if base_llvm == 'void': + return 'i8*' + return base_llvm + '*' * info.PtrCount + return base_llvm + + def _ctype_str_to_llvm(self, type_str: str) -> str: + """将 C 类型名字符串（如 'CUnsignedInt *'）转换为 LLVM 类型字符串（如 'i32*'） + + 支持： + - C 类名：'CUnsignedInt' → 'i32'，'CChar' → 'i8' + - 指针：'CUnsignedInt *' → 'i32*'，'void *' → 'i8*' + - 已是 LLVM 类型：'i32' → 'i32'，'i8*' → 'i8*' + 无法识别返回空字符串。 + """ + s: str = type_str.strip() + ptr_count: int = 0 + while s.endswith('*'): + ptr_count += 1 + s = s[:-1].strip() + base_llvm: str = '' + if s in ('void', 'CVoid'): + base_llvm = 'void' + else: + base_llvm = CTypeRegistry.NameToLLVM(s) + if not base_llvm and s in ('i8', 'i16', 'i32', 'i64', 'float', 'double', 'half', 'fp128', 'void'): + base_llvm = s + if not base_llvm: + return '' + if ptr_count > 0: + if base_llvm == 'void': + return 'i8*' + return base_llvm + '*' * ptr_count + return base_llvm + + def _resolve_name(self, node: ast.Name) -> str: + """解析 ast.Name 节点""" + id_map: dict[str, str] = { + 'str': 'i8*', 'int': 'i32', 'bool': 'i8', + 'float': 'double', 'None': 'void', + 'UINT8PTR': 'i8*', 'INT8PTR': 'i8*', 'BYTEPTR': 'i8*', + 'UINT16PTR': 'i16*', 'INT16PTR': 'i16*', + 'UINT32PTR': 'i32*', 'INT32PTR': 'i32*', + 'UINT64PTR': 'i64*', 'INT64PTR': 'i64*', + } + if node.id in id_map: + return id_map[node.id] + # 通过 BuiltinTypeMap 解析 VOIDPTR、INTPTR 等指针别名 + builtin_entry: tuple[type, int] | None = BuiltinTypeMap.Get(node.id) + if builtin_entry: + TypeClass: type + base_ptr: int + TypeClass, base_ptr = builtin_entry + base_llvm: str = CTypeRegistry.CTypeToLLVM(TypeClass) + if base_ptr > 0: + if base_llvm == 'void': + return 'i8*' + return base_llvm + '*' * base_ptr + return base_llvm if base_llvm else 'i32' + # 查找 SymbolTable 中的 typedef + if node.id in self._symtab: + entry: CTypeInfo = self._symtab[node.id] + if entry and entry.IsTypedef: + # 优先用 OriginalType 解析 + if entry.OriginalType: + if isinstance(entry.OriginalType, CTypeInfo) and entry.OriginalType.IsFuncPtr: + return 'i8*' + elif isinstance(entry.OriginalType, CTypeInfo) and entry.OriginalType.BaseType: + # 用 CTypeRegistry 生成归一化 LLVM 字符串（void*→i8*），struct/enum 回退 ToString() + resolved: str = self._ctype_to_llvm_str(entry.OriginalType) + if resolved: + return resolved + return entry.OriginalType.ToString() + elif isinstance(entry.OriginalType, str): + # 字符串形式的 OriginalType（如 'CUnsignedInt *'）需转换为 LLVM 类型字符串 + llvm_str: str = self._ctype_str_to_llvm(entry.OriginalType) + if llvm_str: + return llvm_str + # OriginalType 为空或字符串无法转换时，回退到 BaseType + if entry.BaseType and (not isinstance(entry.BaseType, (_t._CTypedef,)) or entry.PtrCount > 0): + resolved = self._ctype_to_llvm_str(entry) + if resolved: + return resolved + self._symtab.diagnostics.warn('', 0, f"无法解析 Name 类型 '{node.id}'，回退到 i32") + return 'i32' + + def _resolve_attribute(self, node: ast.Attribute) -> str: + """解析 ast.Attribute 节点""" + attr_name: str = node.attr if hasattr(node, 'attr') else '' + llvm_str: str | None = CTypeRegistry.NameToLLVM(attr_name) + if llvm_str is not None: + return llvm_str + if attr_name in ('CState', 'CDefine', 'CTypedef', 'CExtern', 'CStatic', 'CConst', 'State'): + return '' + if attr_name in ('CCharPtr', 'CIntPtr', 'CVoidPtr', 'CArrayPtr'): + return 'i8*' + if attr_name == 'CVoidPtr': + return 'i8*' + if attr_name and attr_name[0].isupper() and attr_name not in CTypeRegistry._name_to_class: + return f'%struct.{attr_name}*' + if attr_name and attr_name not in CTypeRegistry._name_to_class: + return f'%struct.{attr_name}*' + self._symtab.diagnostics.warn('', 0, f"无法解析 Attribute 类型 '{attr_name}'，回退到 i32") + return 'i32' + + def _resolve_subscript(self, node: ast.Subscript) -> str: + """解析 ast.Subscript 节点""" + if isinstance(node.value, ast.Name) and node.value.id == 'tuple': + slice_node: ast.AST = node.slice + elem_types: list[str] = [] + if isinstance(slice_node, ast.Tuple): + elt: ast.AST + for elt in slice_node.elts: + elem_types.append(self.get_llvm_type_str(elt)) + else: + elem_types.append(self.get_llvm_type_str(slice_node)) + if elem_types: + return '{ ' + ', '.join(elem_types) + ' }' + if IsListAnnotation(node): + slice_node: ast.AST = node.slice + elem_type: str + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) >= 1: + elem_type = self.get_llvm_type_str(slice_node.elts[0]) + else: + elem_type = self.get_llvm_type_str(slice_node) + return elem_type + '*' + self._symtab.diagnostics.warn('', 0, "无法解析 Subscript 类型，回退到 i32") + return 'i32' + + def _resolve_bitor(self, node: ast.BinOp) -> str: + """解析 ast.BinOp(BitOr) 节点""" + left: str = self.get_llvm_type_str(node.left) + right: str = self.get_llvm_type_str(node.right) + if right and right.endswith('*'): + return right + if left and left.endswith('*'): + return left + if left and left not in ('i32', 'void', 'i64'): + return left + if right and right not in ('i32', 'void', 'i64'): + return right + return left if left else right + + def _resolve_call(self, node: ast.Call) -> str: + """解析 ast.Call 节点""" + if isinstance(node.func, ast.Attribute): + if hasattr(node.func.value, 'id') and IsTModule(node.func.value.id): + return self.get_llvm_type_str(ast.Attribute(value=ast.Name(id='t'), attr=node.func.attr)) + self._symtab.diagnostics.warn('', 0, "无法解析 Call 类型，回退到 i32") + return 'i32' diff --git a/lib/core/LlvmCodeGenerator.py b/lib/core/LlvmCodeGenerator.py new file mode 100644 index 0000000..b09db65 --- /dev/null +++ b/lib/core/LlvmCodeGenerator.py @@ -0,0 +1,45 @@ +# LLVM 代码生成器主类 +# 通过 Mixin 模式组合各功能模块，保持向后兼容的导入路径 +# +# 拆分后的模块位于 lib/core/LLVMCG/ 目录下： +# - BaseGen.py: __init__、平台解析、名称修饰、函数查找 +# - TypeConvert.py: 类型转换（CType -> LLVM IR） +# - StructGen.py: 结构体生成（struct/vtable） +# - MemoryOps.py: 内存操作（alloca/Load/store） +# - ExprGen.py: 表达式生成（常量、二元运算、返回） +# - FuncGen.py: 函数处理（参数调整、成员偏移） +# - VaArg.py: 变长参数处理 +from __future__ import annotations + +from lib.core.LLVMCG.BaseGen import BaseGenMixin +from lib.core.LLVMCG.TypeConvert import TypeConvertMixin +from lib.core.LLVMCG.StructGen import StructGenMixin +from lib.core.LLVMCG.MemoryOps import MemoryOpsMixin +from lib.core.LLVMCG.ExprGen import ExprGenMixin +from lib.core.LLVMCG.FuncGen import FuncGenMixin +from lib.core.LLVMCG.VaArg import VaArgMixin + +# from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin + + +class LlvmCodeGenerator( + BaseGenMixin, + TypeConvertMixin, + StructGenMixin, + MemoryOpsMixin, + ExprGenMixin, + FuncGenMixin, + VaArgMixin, +): + """LLVM IR 代码生成器 + + 通过 Mixin 模式组合各功能模块。 + 实际方法实现分布在 lib/core/LLVMCG/ 下的各模块中。 + """ + #builder: ir.IRBuilder = None + #_Trans: LlvmGeneratorMixin = None + pass + + +# 向后兼容：重新导出常量和辅助类 +__all__: list[str] = ['LlvmCodeGenerator', 'VaArgInstruction'] diff --git a/lib/core/SymbolData.py b/lib/core/SymbolData.py new file mode 100644 index 0000000..c422de0 --- /dev/null +++ b/lib/core/SymbolData.py @@ -0,0 +1,85 @@ +"""符号表处理的数据结构：四阶段管线中传递的中间数据""" +from __future__ import annotations +from dataclasses import dataclass, field +from typing import Dict, List, Optional, Tuple, TYPE_CHECKING +import ast + +if TYPE_CHECKING: + from lib.core.Handles.HandlesBase import CTypeInfo + + +@dataclass +class ClassSymbolData: + """从 AST 提取的类符号数据""" + name: str + type_kind: str # 'struct', 'union', 'enum', 'exception' + lineno: int + members: Dict[str, CTypeInfo] + is_cpython_object: bool = False + is_packed: bool = False + enum_members: List[Tuple[str, int, int]] = field(default_factory=list) # (name, value, lineno) + + +@dataclass +class TypedefSymbolData: + """Typedef 符号数据（含原始注解和解析结果）""" + name: str + lineno: int + # 原始注解数据（由 Extractor 设置，Resolver 使用） + annotation: ast.expr | None = None + value: ast.expr | int | str | float | None = None + has_cdefine: bool = False + has_postdef: bool = False + has_ctypedef: bool = False + # 解析结果（由 Resolver 设置，Inserter 使用） + original_type_kind: Optional[str] = None + original_class: type | None = None + members: Optional[Dict] = None + + +@dataclass +class FuncSymbolData: + """函数符号数据（含原始注解和解析结果）""" + name: str + lineno: int + # 原始数据（由 Extractor 设置） + returns_node: ast.expr | None = None + params: List = field(default_factory=list) # [(name, annotation_node), ...] + is_variadic: bool = False + is_inline: bool = False + # 解析结果（由 Resolver 设置） + ret_type: str | None = None + param_types: List = field(default_factory=list) # [str, ...] + # 解析结果（CTypeInfo，保留符号性，由 Resolver 设置） + ret_ctype: Optional[CTypeInfo] = None + + +@dataclass +class DefineSymbolData: + """Define 符号数据（含原始值和解析结果）""" + name: str + lineno: int + # 原始数据（由 Extractor 设置） + value_node: ast.expr | None = None + # 解析结果（由 Resolver 设置） + value: int | str | float | None = None + + +@dataclass +class AnonymousTypeData: + """匿名类型数据""" + name: str + is_union: bool + members: Dict[str, CTypeInfo] + lineno: int + + +@dataclass +class ModuleSymbols: + """从模块提取的完整符号数据，在四阶段管线中传递""" + file_path: str + classes: List[ClassSymbolData] = field(default_factory=list) + typedefs: List[TypedefSymbolData] = field(default_factory=list) + functions: List[FuncSymbolData] = field(default_factory=list) + defines: List[DefineSymbolData] = field(default_factory=list) + anonymous_types: Dict[str, AnonymousTypeData] = field(default_factory=dict) diff --git a/lib/core/SymbolExtractor.py b/lib/core/SymbolExtractor.py new file mode 100644 index 0000000..f3fba6a --- /dev/null +++ b/lib/core/SymbolExtractor.py @@ -0,0 +1,307 @@ +"""Pass 1: ASTSymbolExtractor — 从 AST 提取原始符号信息""" +from __future__ import annotations +import ast +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + +from lib.core.SymbolData import ( + ModuleSymbols, ClassSymbolData, TypedefSymbolData, + FuncSymbolData, DefineSymbolData, AnonymousTypeData +) +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.SymbolUtils import AnnotationContainsTType, CheckAnnotationHasCInline, IsTModule, AnnotationContainsName +from lib.includes import t + + +class ASTSymbolExtractor: + """从 Python 源文件的 AST 中提取所有符号信息""" + + def __init__(self, symbol_table: SymbolTable) -> None: + self._symtab: SymbolTable = symbol_table + + def extract(self, file_path: str) -> ModuleSymbols: + """提取文件中的所有符号信息""" + with open(file_path, 'r', encoding='utf-8') as f: + code: str = f.read() + tree: ast.Module = ast.parse(code) + + result: ModuleSymbols = ModuleSymbols(file_path=file_path) + + # 第一步：收集所有类的成员和匿名类型 + class_members: dict[str, dict[str, CTypeInfo]] = {} + for node in tree.body: + if isinstance(node, ast.ClassDef): + members: dict[str, CTypeInfo] = {} + self._collect_members(node, members, result.anonymous_types) + class_members[node.name] = members + + # 第二步：从每个顶层节点提取符号信息 + for node in tree.body: + if isinstance(node, ast.ClassDef): + members_info: dict[str, CTypeInfo] = class_members.get(node.name, {}) + result.classes.append(self._extract_class(node, members_info)) + elif isinstance(node, ast.AnnAssign) and isinstance(node.target, ast.Name): + self._extract_ann_assign(node, result) + elif isinstance(node, ast.FunctionDef): + result.functions.append(self._extract_function(node)) + + return result + + # ------------------------------------------------------------------ + # 成员收集（递归处理嵌套匿名类） + # ------------------------------------------------------------------ + + def _collect_members(self, class_node: ast.ClassDef, members: dict[str, CTypeInfo], anonymous_types: dict[str, AnonymousTypeData], prefix: str = '') -> None: + """递归收集类成员，包括嵌套匿名类型""" + for item in class_node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + member_name: str = item.target.id + full_name: str = f'{prefix}{member_name}' if prefix else member_name + member_type: CTypeInfo | None = CTypeInfo.FromNode(item.annotation, self._symtab) + if member_type is None: + member_type = CTypeInfo() + member_type.BaseType = t.CInt() + is_ptr: bool = member_type.IsPtr + if isinstance(item.annotation, ast.BinOp): + if AnnotationContainsName(item.annotation, 'CPtr'): + is_ptr = True + members[full_name] = member_type.Copy() + + elif isinstance(item, ast.ClassDef): + is_anonymous: bool = any( + any(AnnotationContainsTType(base, 'Anonymous') for base in item.bases) + ) if item.bases else False + + if is_anonymous: + is_union: bool = any( + any(AnnotationContainsTType(base, 'CUnion') for base in item.bases) + ) + # 也检查装饰器形式 @t.CUnion + if not is_union and hasattr(item, 'decorator_list') and item.decorator_list: + for deco in item.decorator_list: + if isinstance(deco, ast.Attribute) and hasattr(deco.value, 'id') and IsTModule(deco.value.id, self._symtab) and deco.attr == 'CUnion': + is_union = True + break + elif isinstance(deco, ast.Name): + cls2: type | None = getattr(t, deco.id, None) + if cls2 is None and hasattr(self._symtab, '_t_type_symbols'): + cls2 = self._symtab._t_type_symbols.get(deco.id) + if cls2 is not None and isinstance(cls2, type) and issubclass(cls2, t.CUnion): + is_union = True + break + + nested_name: str = f'{prefix}{item.name}' if prefix else item.name + members[nested_name] = CTypeInfo() + members[nested_name].BaseType = f'union {item.name}' if is_union else f'struct {item.name}' + members[nested_name].PtrCount = 0 + members[nested_name].ArrayDims = [] + + new_prefix: str = f'{prefix}{item.name}.' if prefix else f'{item.name}.' + self._collect_members(item, members, anonymous_types, new_prefix) + + anonymous_members: dict[str, CTypeInfo] = {} + for full_name, member_info in members.items(): + if full_name.startswith(f"{nested_name}."): + member_name: str = full_name[len(nested_name) + 1:] + anonymous_members[member_name] = member_info + + if item.name not in anonymous_types: + anonymous_types[item.name] = AnonymousTypeData( + name=item.name, + is_union=is_union, + members=anonymous_members, + lineno=item.lineno + ) + + # ------------------------------------------------------------------ + # 类符号提取 + # ------------------------------------------------------------------ + + def _extract_class(self, node: ast.ClassDef, members_info: dict[str, CTypeInfo]) -> ClassSymbolData: + """从 ClassDef 节点提取类符号数据""" + type_kind: str = 'struct' + is_cpython_object: bool = False + is_packed: bool = False + + # 检查装饰器 + if hasattr(node, 'decorator_list') and node.decorator_list: + for decorator in node.decorator_list: + if isinstance(decorator, ast.Attribute): + if (hasattr(decorator.value, 'id') and + IsTModule(decorator.value.id, self._symtab)): + if decorator.attr == 'Object': + is_cpython_object = True + elif decorator.attr == 'CUnion': + type_kind = 'union' + elif decorator.attr == 'CEnum': + type_kind = 'enum' + elif decorator.attr == 'REnum': + type_kind = 'renum' + elif decorator.attr == 'CStruct': + type_kind = 'struct' + elif isinstance(decorator, ast.Name): + # from t import CStruct / CUnion / CEnum / REnum + cls: type | None = getattr(t, decorator.id, None) + if cls is None and hasattr(self._symtab, '_t_type_symbols'): + cls = self._symtab._t_type_symbols.get(decorator.id) + if cls is not None and isinstance(cls, type) and issubclass(cls, t.CType): + if issubclass(cls, t.CUnion): + type_kind = 'union' + elif issubclass(cls, t.REnum): + type_kind = 'renum' + elif issubclass(cls, t.CEnum): + type_kind = 'enum' + elif issubclass(cls, t.CStruct): + type_kind = 'struct' + elif decorator.id == 'Object': + is_cpython_object = True + elif isinstance(decorator, ast.Call): + if isinstance(decorator.func, ast.Attribute): + if hasattr(decorator.func.value, 'id') and decorator.func.value.id == 'c' and decorator.func.attr == 'Attribute': + for arg in decorator.args: + if isinstance(arg, ast.Attribute): + if isinstance(arg.value, ast.Attribute): + if hasattr(arg.value.value, 'id') and IsTModule(arg.value.value.id, self._symtab) and arg.value.attr == 'attr' and arg.attr == 'packed': + is_packed = True + + # 检查基类 + for base in node.bases: + if AnnotationContainsTType(base, 'CUnion'): + type_kind = 'union' + break + elif AnnotationContainsTType(base, 'CEnum'): + type_kind = 'enum' + break + elif AnnotationContainsTType(base, 'REnum'): + type_kind = 'renum' + break + elif AnnotationContainsTType(base, 'CStruct'): + type_kind = 'struct' + break + elif AnnotationContainsTType(base, 'Object'): + is_cpython_object = True + type_kind = 'struct' + break + elif isinstance(base, ast.Name) and base.id == 'Exception': + type_kind = 'exception' + break + elif isinstance(base, ast.Name): + base_entry: CTypeInfo | None = self._symtab.get(base.id) + if base_entry and base_entry.IsExceptionClass: + type_kind = 'exception' + break + + # 提取枚举成员 + enum_members: list[tuple[str, int, int]] = [] + if type_kind == 'enum': + next_enum_value: int = 0 + for item in node.body: + if isinstance(item, ast.Assign): + if len(item.targets) == 1 and isinstance(item.targets[0], ast.Name): + member_name: str = item.targets[0].id + if isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + next_enum_value = item.value.value + 1 + else: + next_enum_value += 1 + enum_members.append((member_name, next_enum_value - 1, item.lineno)) + elif isinstance(item, ast.AnnAssign): + if isinstance(item.target, ast.Name): + member_name: str = item.target.id + if item.value: + if isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + next_enum_value = item.value.value + 1 + else: + next_enum_value += 1 + else: + next_enum_value += 1 + enum_members.append((member_name, next_enum_value - 1, item.lineno)) + + # REnum 变体提取（嵌套 ClassDef + Assign 显式值） + # 与 HandlesClassDef._EmitREnumLlvm 的 tag 分配逻辑对齐： + # - 先收集 Assign 显式值 + # - ClassDef 变体从 max(explicit)+1 开始顺序编号 + # - Assign 无显式值时用独立计数器 variant_index + if type_kind == 'renum': + explicit_values: dict[str, int] = {} + for item in node.body: + if isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + if item.value and isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + explicit_values[target.id] = item.value.value + next_class_tag: int = max(explicit_values.values()) + 1 if explicit_values else 0 + variant_index: int = 0 + for item in node.body: + if isinstance(item, ast.ClassDef): + variant_name_r: str = item.name + tag: int = next_class_tag + next_class_tag += 1 + enum_members.append((variant_name_r, tag, item.lineno)) + elif isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name): + var_name: str = target.id + if item.value and isinstance(item.value, ast.Constant) and isinstance(item.value.value, int): + value: int = item.value.value + else: + value = variant_index + variant_index += 1 + enum_members.append((var_name, value, item.lineno)) + + return ClassSymbolData( + name=node.name, + type_kind=type_kind, + lineno=node.lineno, + members=members_info, + is_cpython_object=is_cpython_object, + is_packed=is_packed, + enum_members=enum_members, + ) + + # ------------------------------------------------------------------ + # AnnAssign 提取（typedef / define） + # ------------------------------------------------------------------ + + def _extract_ann_assign(self, node: ast.AnnAssign, result: ModuleSymbols) -> None: + """从 AnnAssign 节点提取 typedef 或 define 数据""" + var_name: str = node.target.id + has_cdefine: bool = AnnotationContainsTType(node.annotation, 'CDefine') if node.annotation else False + has_postdef: bool = AnnotationContainsTType(node.annotation, 'Postdefinition') if node.annotation else False + has_ctypedef: bool = AnnotationContainsTType(node.annotation, 'CTypedef') if node.annotation else False + + if has_cdefine: + result.defines.append(DefineSymbolData( + name=var_name, + lineno=node.lineno, + value_node=node.value, + )) + else: + result.typedefs.append(TypedefSymbolData( + name=var_name, + lineno=node.lineno, + annotation=node.annotation, + value=node.value, + has_cdefine=has_cdefine, + has_postdef=has_postdef, + has_ctypedef=has_ctypedef, + )) + + # ------------------------------------------------------------------ + # 函数提取 + # ------------------------------------------------------------------ + + def _extract_function(self, node: ast.FunctionDef) -> FuncSymbolData: + """从 FunctionDef 节点提取函数符号数据""" + params: list[tuple[str, ast.AST | None]] = [(arg.arg, arg.annotation) for arg in node.args.args] + is_variadic: bool = node.args.vararg is not None + is_inline: bool = CheckAnnotationHasCInline(node.returns) + + return FuncSymbolData( + name=node.name, + lineno=node.lineno, + returns_node=node.returns, + params=params, + is_variadic=is_variadic, + is_inline=is_inline, + ) diff --git a/lib/core/SymbolInserter.py b/lib/core/SymbolInserter.py new file mode 100644 index 0000000..01dc3a1 --- /dev/null +++ b/lib/core/SymbolInserter.py @@ -0,0 +1,89 @@ +"""Pass 3: SymbolInserter — 将解析后的符号插入 SymbolTable（主命名空间）""" +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + +from lib.core.SymbolData import ModuleSymbols + + +class SymbolInserter: + """将解析后的符号插入 SymbolTable 的主命名空间（无前缀）""" + + def __init__(self, symbol_table: SymbolTable) -> None: + self._symtab: SymbolTable = symbol_table + self._namespace: str | None = None + + def set_namespace(self, namespace: str) -> None: + """设置当前模块的命名空间""" + self._namespace = namespace + self._symtab.register_namespace(namespace) + + def insert(self, module_symbols: ModuleSymbols) -> list[str]: + """插入所有符号到主命名空间，返回已加载的符号名列表""" + loaded: list[str] = [] + file_path: str = module_symbols.file_path + ns: dict[str, object] = self._symtab._namespaces.get(self._namespace, {}) if self._namespace else {} + + # 插入类符号 + for cls in module_symbols.classes: + self._symtab._InsertClassSymbol( + cls.name, cls.type_kind, cls.lineno, file_path, + cls.members, cls.is_cpython_object, cls.is_packed + ) + loaded.append(cls.name) + if self._namespace: + ns[cls.name] = self._symtab._symbols[cls.name] + + # 插入枚举成员（短名 + 类名.成员名）—— CEnum 和 REnum 共用 + if cls.type_kind in ('enum', 'renum'): + for member_name, member_value, member_lineno in cls.enum_members: + self._symtab._InsertEnumMemberSymbol(member_name, cls.name, member_lineno, file_path, value=member_value) + loaded.append(member_name) + class_member_name: str = f"{cls.name}.{member_name}" + self._symtab._InsertEnumMemberSymbol(class_member_name, cls.name, member_lineno, file_path, value=member_value) + loaded.append(class_member_name) + # REnum: 记录变体名列表到类符号（供 HandlesExprCall 跨模块变体构造识别） + if cls.type_kind == 'renum': + renum_info = self._symtab._symbols[cls.name] + renum_info.RenumVariants = [m[0] for m in cls.enum_members] + + # 插入 typedef 符号 + for td in module_symbols.typedefs: + self._symtab._InsertTypedefSymbol( + td.name, td.original_type_kind, td.original_class, + td.lineno, file_path, td.members + ) + loaded.append(td.name) + if self._namespace: + ns[td.name] = self._symtab._symbols[td.name] + + # 插入函数符号 + for func in module_symbols.functions: + # 优先使用 CTypeInfo（保留符号性），回退到 LLVM 类型字符串 + ret_type = func.ret_ctype if func.ret_ctype else func.ret_type + self._symtab._InsertFuncSymbol( + func.name, ret_type, func.param_types, + func.lineno, file_path, func.is_variadic, func.is_inline + ) + loaded.append(func.name) + if self._namespace: + ns[func.name] = self._symtab._symbols[func.name] + + # 插入 define 符号 + for define in module_symbols.defines: + self._symtab._InsertDefineSymbol( + define.name, define.value, define.lineno, file_path + ) + loaded.append(define.name) + if self._namespace: + ns[define.name] = self._symtab._symbols[define.name] + + # 插入匿名类型符号（短名） + for anon_name, anon_data in module_symbols.anonymous_types.items(): + self._symtab._InsertAnonymousSymbol( + anon_name, anon_data.is_union, anon_data.members, + anon_data.lineno, file_path + ) + + return loaded diff --git a/lib/core/SymbolNode.py b/lib/core/SymbolNode.py new file mode 100644 index 0000000..f1aa0ef --- /dev/null +++ b/lib/core/SymbolNode.py @@ -0,0 +1,224 @@ +from __future__ import annotations +from typing import Any, Dict, Optional + +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.includes import t + + +class SymbolNode: + """符号表节点，类似 AST 节点的树形结构 + + attributes 存储 CTypeInfo 对象，不再使用 TypeInfo + """ + + def __init__(self, name: str, NodeType: str, lineno: int = 0, file: str = '', parent: SymbolNode | None = None): + self.name: str = name + self.type: str = NodeType + self.lineno: int = lineno + self.file: str = file + self.parent: SymbolNode | None = parent + self.children: Dict[str, SymbolNode] = {} + self.attributes: CTypeInfo = CTypeInfo() + self.attributes.Name = name + self.attributes.Lineno = lineno + + def get(self, key: str, default: Any = None) -> Any: + """获取属性值""" + if hasattr(self.attributes, key): + return getattr(self.attributes, key) + return self.attributes._extra.get(key, default) + + def set(self, key: str, value: Any) -> None: + """设置属性值""" + if hasattr(self.attributes, key): + attr: Any = getattr(type(self.attributes), key, None) + if isinstance(attr, property) and attr.fset is None: + return + setattr(self.attributes, key, value) + else: + self.attributes._extra[key] = value + + def HasChild(self, name: str) -> bool: + """检查是否有子节点""" + return name in self.children + + def GetChild(self, name: str) -> Optional[SymbolNode]: + """获取子节点""" + return self.children.get(name) + + def AddChild(self, node: SymbolNode): + """添加子节点""" + node.parent = self + self.children[node.name] = node + + def RemoveChild(self, name: str) -> bool: + """删除子节点""" + if name in self.children: + del self.children[name] + return True + return False + + def find(self, path: str, separator: str = '.') -> Optional[SymbolNode]: + """根据路径查找节点，如 'test.ClassA'""" + parts: list[str] = path.split(separator) + current: SymbolNode = self + + for part in parts: + if current.HasChild(part): + current = current.GetChild(part) + else: + return None + + return current + + def ToDict(self) -> Dict[str, Any]: + """转换为字典格式""" + result: dict[str, Any] = { + 'type': self.type, + 'lineno': self.lineno, + 'file': self.file, + } + # 序列化 CTypeInfo 属性 + attrs: CTypeInfo = self.attributes + if attrs.BaseType: + result['BaseType'] = str(attrs.BaseType) + if attrs.PtrCount: + result['PtrCount'] = attrs.PtrCount + if attrs.IsTypedef: + result['IsTypedef'] = True + if attrs.IsStruct: + result['IsStruct'] = True + if attrs.IsEnum: + result['IsEnum'] = True + if attrs.IsUnion: + result['IsUnion'] = True + if attrs.IsFunction: + result['IsFunction'] = True + if attrs.IsVariable: + result['IsVariable'] = True + if attrs.IsDefine: + result['IsDefine'] = True + if attrs.OriginalType: + result['OriginalType'] = str(attrs.OriginalType) + if attrs._sm._extra: + result.update(attrs._sm._extra) + + if self.children: + result['children'] = {name: child.ToDict() for name, child in self.children.items()} + + return result + + @classmethod + def FromDict(cls, name: str, data: Dict[str, Any]) -> SymbolNode: + """从字典创建节点（兼容性）""" + node = cls(name=name, NodeType=data.get('type', 'unknown')) + node.lineno = data.get('lineno', 0) + node.file = data.get('file', '') + + for key, value in data.items(): + if key not in ('type', 'lineno', 'file', 'children'): + node.set(key, value) + + if 'children' in data: + for ChildName, ChildData in data['children'].items(): + node.AddChild(cls.FromDict(ChildName, ChildData)) + + return node + + @classmethod + def CreateMember(cls, name: str, TypeName: str | None = None, TypeInfo: 'CTypeInfo | None' = None, IsPtr: bool = False, dims: list[int] | None = None, lineno: int = 0, file: str = '') -> SymbolNode: + """创建成员节点（工厂函数）""" + node = cls(name=name, NodeType='member', lineno=lineno, file=file) + node.attributes = CTypeInfo() + node.attributes.Name = name + node.attributes.Lineno = lineno + if TypeInfo is not None: + node.attributes = TypeInfo + node.attributes.Name = name + node.attributes.Lineno = lineno + else: + if TypeName is not None: + node.attributes.BaseType = CTypeInfo.CreateFromTypeName(TypeName) + if IsPtr: + node.attributes.PtrCount = 1 + if dims: + node.attributes.ArrayDims = list(dims) + return node + + @classmethod + def CreateClass(cls, name: str, TypeKind: type, members: dict | None = None, lineno: int = 0, file: str = '', IsCpythonObject: bool = False, IsPacked: bool = False) -> SymbolNode: + """创建类节点（工厂函数）""" + node: SymbolNode = cls(name=name, NodeType=TypeKind, lineno=lineno, file=file) + node.attributes = CTypeInfo() + node.attributes.Name = name + node.attributes.Lineno = lineno + node.attributes.IsCpythonObject = IsCpythonObject + node.attributes.IsPacked = IsPacked + if members: + for MemberName, MemberInfo in members.items(): + if isinstance(MemberInfo, CTypeInfo): + node.attributes.Members[MemberName] = MemberInfo + elif isinstance(MemberInfo, dict): + CTypeMember: CTypeInfo = CTypeInfo() + type_str: str = MemberInfo.get('type', 'int') + IsPtr: bool = MemberInfo.get('IsPtr', False) + if isinstance(type_str, str): + try: + resolved: CTypeInfo | None = CTypeInfo.FromTypeName(type_str) + if resolved and resolved.BaseType: + CTypeMember = resolved + else: + CTypeMember.BaseType = t.CInt() + except Exception: + CTypeMember.BaseType = t.CInt() + else: + CTypeMember.BaseType = t.CInt() + if IsPtr and not CTypeMember.IsPtr: + CTypeMember.PtrCount = max(CTypeMember.PtrCount, 1) + node.attributes.Members[MemberName] = CTypeMember + if TypeKind == 'struct': + node.attributes.IsStruct = True + elif TypeKind == 'enum': + node.attributes.IsEnum = True + elif TypeKind == 'union': + node.attributes.IsUnion = True + elif TypeKind == 'typedef': + node.attributes.IsTypedef = True + elif TypeKind == 'exception': + node.attributes.IsExceptionClass = True + return node + + @classmethod + def CreateTypedef(cls, name: str, OriginalType: CTypeInfo | str | None, members: dict | None = None, lineno: int = 0, file: str = '') -> SymbolNode: + node: SymbolNode = cls(name=name, NodeType='typedef', lineno=lineno, file=file) + node.attributes = CTypeInfo() + node.attributes.Name = name + node.attributes.Lineno = lineno + node.attributes.IsTypedef = True + if isinstance(OriginalType, CTypeInfo): + node.attributes.OriginalType = OriginalType + elif isinstance(OriginalType, str): + node.attributes.OriginalType = CTypeInfo.FromTypeName(OriginalType) if OriginalType else None + else: + node.attributes.OriginalType = OriginalType + return node + + @classmethod + def CreateAnonymous(cls, name: str, IsUnion: bool, members: dict, lineno: int = 0, file: str = '') -> SymbolNode: + """创建匿名类型节点（工厂函数）""" + NodeType: type = t.CUnion if IsUnion else t.CStruct + node: SymbolNode = cls(name=name, NodeType=NodeType, lineno=lineno, file=file) + node.attributes = CTypeInfo() + node.attributes.Name = name + node.attributes.Lineno = lineno + node.attributes.IsAnonymous = True + if IsUnion: + node.attributes.IsUnion = True + else: + node.attributes.IsStruct = True + return node + + @classmethod + def CreateModule(cls, name: str, lineno: int = 0, file: str = '') -> SymbolNode: + """创建模块节点（工厂函数）""" + return cls(name=name, NodeType='module', lineno=lineno, file=file) diff --git a/lib/core/SymbolReexporter.py b/lib/core/SymbolReexporter.py new file mode 100644 index 0000000..ae86b80 --- /dev/null +++ b/lib/core/SymbolReexporter.py @@ -0,0 +1,99 @@ +"""Pass 4: PackageReexporter — 将符号重新导出到命名空间前缀下""" +from __future__ import annotations +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + +from lib.core.SymbolData import ModuleSymbols + + +class PackageReexporter: + """将已解析的符号重新导出到命名空间前缀下（如 memhub.MemPool）""" + + def __init__(self, symbol_table: SymbolTable) -> None: + self._symtab: SymbolTable = symbol_table + self._namespace: str | None = None + + def set_namespace(self, namespace: str) -> None: + """设置当前模块的命名空间""" + self._namespace = namespace + + def reexport(self, module_symbols: ModuleSymbols, prefixes: list[str], lineno: int = 0) -> list[str]: + """将符号重新导出到所有命名空间前缀下，返回新增的符号名列表""" + loaded: list[str] = [] + file_path: str = module_symbols.file_path + ns: dict[str, object] = self._symtab._namespaces.get(self._namespace, {}) if self._namespace else {} + + for prefix in prefixes: + # 插入类符号（前缀.类名） + for cls in module_symbols.classes: + full_name: str = f"{prefix}.{cls.name}" + self._symtab._InsertClassSymbol( + full_name, cls.type_kind, cls.lineno, file_path, + cls.members, cls.is_cpython_object, cls.is_packed + ) + loaded.append(full_name) + if self._namespace: + ns[full_name] = self._symtab._symbols[full_name] + + # 插入枚举成员（前缀.成员名 + 前缀.类名.成员名）—— CEnum 和 REnum 共用 + if cls.type_kind in ('enum', 'renum'): + for member_name, member_value, member_lineno in cls.enum_members: + full_member_name: str = f"{prefix}.{member_name}" + self._symtab._InsertEnumMemberSymbol(full_member_name, cls.name, member_lineno, file_path, value=member_value) + loaded.append(full_member_name) + # 同时注册 prefix.ClassName.member_name，以支持 module.Class.MEMBER 三层访问 + qualified_member_name: str = f"{prefix}.{cls.name}.{member_name}" + self._symtab._InsertEnumMemberSymbol(qualified_member_name, cls.name, member_lineno, file_path, value=member_value) + loaded.append(qualified_member_name) + # REnum: 记录变体名列表到前缀类符号 + if cls.type_kind == 'renum': + renum_info = self._symtab._symbols[full_name] + renum_info.RenumVariants = [m[0] for m in cls.enum_members] + + # 插入 typedef 符号（前缀.名称） + for td in module_symbols.typedefs: + full_name: str = f"{prefix}.{td.name}" + self._symtab._InsertTypedefSymbol( + full_name, td.original_type_kind, td.original_class, + td.lineno, file_path, td.members + ) + loaded.append(full_name) + if self._namespace: + ns[full_name] = self._symtab._symbols[full_name] + + # 插入函数符号（前缀.名称） + for func in module_symbols.functions: + full_name: str = f"{prefix}.{func.name}" + # 优先使用 CTypeInfo（保留符号性），回退到 LLVM 类型字符串 + ret_type = func.ret_ctype if func.ret_ctype else func.ret_type + self._symtab._InsertFuncSymbol( + full_name, ret_type, func.param_types, + func.lineno, file_path, func.is_variadic, func.is_inline + ) + loaded.append(full_name) + if self._namespace: + ns[full_name] = self._symtab._symbols[full_name] + + # 插入 define 符号（前缀.名称） + for define in module_symbols.defines: + full_name: str = f"{prefix}.{define.name}" + self._symtab._InsertDefineSymbol( + full_name, define.value, define.lineno, file_path + ) + loaded.append(full_name) + if self._namespace: + ns[full_name] = self._symtab._symbols[full_name] + + # 插入模块别名 + self._symtab._InsertModuleSymbol(prefix, lineno, file_path) + + # 插入匿名类型符号（前缀.名称） + for anon_name, anon_data in module_symbols.anonymous_types.items(): + full_type_name: str = f"{prefix}.{anon_name}" + self._symtab._InsertAnonymousSymbol( + full_type_name, anon_data.is_union, anon_data.members, + anon_data.lineno, file_path + ) + + return loaded diff --git a/lib/core/SymbolTable.py b/lib/core/SymbolTable.py new file mode 100644 index 0000000..9646b54 --- /dev/null +++ b/lib/core/SymbolTable.py @@ -0,0 +1,531 @@ +from __future__ import annotations +from typing import TYPE_CHECKING, Any, Iterator +if TYPE_CHECKING: + from lib.core.translator import Translator + +import ast +import traceback + +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.includes import t +from lib.core.VLogger import get_logger as _vlog +from lib.core.DiagnosticCollector import DiagnosticCollector +from lib.core.LLVMTypeMapper import LLVMTypeMapper +from lib.core.SymbolExtractor import ASTSymbolExtractor +from lib.core.SymbolTypeResolver import TypeResolver +from lib.core.SymbolInserter import SymbolInserter +from lib.core.SymbolReexporter import PackageReexporter +from lib.core.ConstEvaluator import ConstEvaluator +from lib.core.SymbolData import ModuleSymbols + +# 从 SymbolUtils 重导出，保持向后兼容 +from lib.core.SymbolUtils import CheckAnnotationHasCInline as _CheckAnnotationHasCInline + + +class SymbolTable: + def __init__(self, translator: Translator) -> None: + self.translator: Translator = translator + self._symbols: dict[str, CTypeInfo] = {} + self._namespaces: dict[str, dict[str, CTypeInfo]] = {} # 模块名 -> 符号字典 + self._t_type_symbols: dict[str, type] = {} # 别名 -> t.CType 子类 (from t import CEnum as en) + self.diagnostics: DiagnosticCollector = DiagnosticCollector() + self._type_mapper: LLVMTypeMapper = LLVMTypeMapper(self) + self.import_aliases: dict[str, str] = {} # 别名 -> 模块路径 + + def __getstate__(self) -> dict: + """pickle 序列化：排除不可序列化的字段（translator/_type_mapper/diagnostics）""" + state: dict = self.__dict__.copy() + state.pop('translator', None) + state.pop('_type_mapper', None) + state.pop('diagnostics', None) + return state + + def __setstate__(self, state: dict) -> None: + """pickle 反序列化：重建可序列化的字段，translator 由调用方后续设置""" + self.__dict__.update(state) + self.translator = None # 由调用方赋值 + self.diagnostics = DiagnosticCollector() + self._type_mapper = LLVMTypeMapper(self) + + def clear(self) -> None: + self._symbols.clear() + self._namespaces.clear() + self._t_type_symbols.clear() + self.import_aliases.clear() + + def get(self, name: str, default: CTypeInfo | None = None) -> CTypeInfo | None: + return self._symbols.get(name, default) + + def set(self, name: str, value: CTypeInfo | dict[str, Any]) -> None: + if isinstance(value, CTypeInfo): + self._symbols[name] = value + else: + # 兼容旧代码传入 dict 的情况 + info: CTypeInfo = self._CTypeInfoFromDict(name, value) + self._symbols[name] = info + + def update(self, symbols: dict[str, CTypeInfo | dict[str, Any]]) -> None: + for name, value in symbols.items(): + self.set(name, value) + + def keys(self) -> Iterator[str]: + return iter(self._symbols.keys()) + + def values(self) -> Iterator[CTypeInfo]: + return iter(self._symbols.values()) + + def items(self) -> Iterator[tuple[str, CTypeInfo]]: + return iter(self._symbols.items()) + + def pop(self, name: str, *args: Any) -> CTypeInfo: + return self._symbols.pop(name, *args) + + def __getitem__(self, name: str) -> CTypeInfo: + if name in self._symbols: + return self._symbols[name] + raise KeyError(name) + + def __setitem__(self, name: str, value: CTypeInfo | dict[str, Any]) -> None: + self.set(name, value) + + def __delitem__(self, name: str) -> None: + del self._symbols[name] + + def __contains__(self, name: str) -> bool: + return name in self._symbols + + def __len__(self) -> int: + return len(self._symbols) + + def __iter__(self) -> Iterator[str]: + return iter(self._symbols) + + # ================================================================== + # 服务接口 — 语义化的符号查询与操作方法 + # ================================================================== + + def lookup(self, name: str) -> CTypeInfo | None: + """查找符号，支持别名解析和 dotted name + + 解析顺序： + 1. 直接匹配 _symbols + 2. 如果包含 '.'，解析第一个段为别名后查找剩余路径 + 3. 查 _t_type_symbols（t/c 类型别名，返回模拟 CTypeInfo） + """ + if name in self._symbols: + return self._symbols[name] + + if '.' not in name: + return self._lookup_t_type(name) + + # dotted name: 尝试解析第一个段为别名 + parts: list[str] = name.split('.', 1) + first: str = parts[0] + rest: str = parts[1] + + # 1) 别名解析：import X as Y → Y.Symbol → X.Symbol + resolved: str | None = self.import_aliases.get(first) + if resolved: + alias_name: str = f"{resolved}.{rest}" + if alias_name in self._symbols: + return self._symbols[alias_name] + + # 2) 纯路径查找：module.ClassName + # （已在 line 100 检查过，此处为兜底，保留注释说明路径查找语义） + + return None + + def _lookup_t_type(self, name: str) -> CTypeInfo | None: + """在 _t_type_symbols 中查找 t/c 类型别名，返回模拟 CTypeInfo""" + cls: type | None = self._t_type_symbols.get(name) + if cls is None: + return None + info: CTypeInfo = CTypeInfo() + info.Name = name + info.BaseType = cls() + if issubclass(cls, (t.CEnum, t.REnum)): + info.IsEnum = True + elif issubclass(cls, t.CUnion): + info.IsUnion = True + elif issubclass(cls, t.CStruct): + info.IsStruct = True + return info + + def register_namespace(self, module_name: str) -> None: + """注册一个模块命名空间""" + if module_name not in self._namespaces: + self._namespaces[module_name] = {} + + def has(self, name: str) -> bool: + """检查符号是否存在（替代 `name in SymbolTable`）""" + return name in self._symbols + + def is_struct(self, name: str) -> bool: + """检查名称是否为结构体""" + entry: CTypeInfo | None = self._symbols.get(name) + return entry is not None and (entry.IsStruct or entry.IsCpythonObject) + + def is_define(self, name: str) -> bool: + """检查名称是否为常量定义""" + entry: CTypeInfo | None = self._symbols.get(name) + return entry is not None and entry.IsDefine + + def insert(self, name: str, info: CTypeInfo) -> None: + """插入符号（替代 `SymbolTable[name] = info`）""" + self._symbols[name] = info + + def resolve_alias(self, module_path: str) -> str: + """解析 import 别名，无别名返回原路径""" + return self.import_aliases.get(module_path, module_path) + + def ToDict(self) -> dict[str, Any]: + """序列化为字典格式，BaseType 使用可重建的 dict 格式 + + 返回的 dict 包含： + - 符号条目：{name: {attrs}}（与旧格式兼容） + - '__namespaces__': {namespace: [symbol_names]} 命名空间索引 + - '__import_aliases__': {alias: module_path} 导入别名 + """ + result: dict[str, Any] = {} + for name, info in self._symbols.items(): + entry: dict[str, Any] = {'name': name} + if info.BaseType: + entry['BaseType'] = self._SerializeBaseType(info.BaseType) + if info.PtrCount: + entry['PtrCount'] = info.PtrCount + if info.IsTypedef: + entry['IsTypedef'] = True + if info.IsStruct: + entry['IsStruct'] = True + if info.IsEnum: + entry['IsEnum'] = True + if info.IsUnion: + entry['IsUnion'] = True + if info.IsFunction: + entry['IsFunction'] = True + if info.IsVariable: + entry['IsVariable'] = True + if info.IsDefine: + entry['IsDefine'] = True + if info.OriginalType: + entry['OriginalType'] = str(info.OriginalType) + if info.Lineno: + entry['lineno'] = info.Lineno + if info.file: + entry['file'] = info.file + if info.Members: + entry['members'] = info.Members + extra: dict[str, Any] = info._sm._extra + if extra: + entry.update(extra) + result[name] = entry + + # 序列化命名空间信息 + if self._namespaces: + result['__namespaces__'] = { + ns: list(symbols.keys()) for ns, symbols in self._namespaces.items() + } + if self.import_aliases: + result['__import_aliases__'] = dict(self.import_aliases) + if self._t_type_symbols: + result['__t_type_symbols__'] = { + name: cls.__name__ for name, cls in self._t_type_symbols.items() + } + + return result + + @staticmethod + def _SerializeBaseType(base_type: t.CType | tuple | list | str | None) -> dict | str: + """将 BaseType 序列化为可重建的格式""" + if isinstance(base_type, t.CType): + return {'class': type(base_type).__name__} + if isinstance(base_type, (tuple, list)): + return {'classes': [type(b).__name__ for b in base_type if isinstance(b, t.CType)]} + return str(base_type) + + @staticmethod + def _DeserializeBaseType(data: Any) -> Any: + """从序列化格式恢复 BaseType""" + if isinstance(data, dict): + if 'class' in data: + cls: type | None = getattr(t, data['class'], None) + if cls and isinstance(cls, type) and issubclass(cls, t.CType): + return cls() + elif 'classes' in data: + types: list[t.CType] = [] + for cname in data['classes']: + cls: type | None = getattr(t, cname, None) + if cls and isinstance(cls, type) and issubclass(cls, t.CType): + types.append(cls()) + if types: + return tuple(types) + elif isinstance(data, str): + cls: type | None = getattr(t, data, None) + if cls and isinstance(cls, type) and issubclass(cls, t.CType): + return cls() + return None + + def FromDict(self, symbols: dict[str, Any]) -> None: + """从字典格式反序列化（兼容带/不带命名空间信息的两种格式）""" + self._symbols.clear() + self._namespaces.clear() + self.import_aliases.clear() + self._t_type_symbols.clear() + + _SPECIAL_KEYS: frozenset[str] = frozenset({'__namespaces__', '__import_aliases__', '__t_type_symbols__'}) + + for name, attrs in symbols.items(): + if name in _SPECIAL_KEYS: + continue + if isinstance(attrs, dict): + info: CTypeInfo = self._CTypeInfoFromDict(name, attrs) + self._symbols[name] = info + elif isinstance(attrs, CTypeInfo): + self._symbols[name] = attrs + + # 恢复命名空间信息 + ns_data: dict[str, list[str]] | None = symbols.get('__namespaces__') + if ns_data: + for ns, sym_names in ns_data.items(): + self._namespaces[ns] = {} + for sym_name in sym_names: + if sym_name in self._symbols: + self._namespaces[ns][sym_name] = self._symbols[sym_name] + aliases_data: dict[str, str] | None = symbols.get('__import_aliases__') + if aliases_data: + self.import_aliases.update(aliases_data) + + # 恢复 t 类型别名 + t_type_data: dict[str, str] | None = symbols.get('__t_type_symbols__') + if t_type_data: + for name, cls_name in t_type_data.items(): + cls: type | None = getattr(t, cls_name, None) + if cls and isinstance(cls, type) and issubclass(cls, t.CType): + self._t_type_symbols[name] = cls + + def _CTypeInfoFromDict(self, name: str, attrs: dict[str, Any]) -> CTypeInfo: + """从旧 dict 格式创建 CTypeInfo（兼容旧代码）""" + info: CTypeInfo = CTypeInfo() + info.Name = name + node_type: str = attrs.get('type', '') + + if node_type == 'struct' or attrs.get('IsStruct'): + info.IsStruct = True + elif node_type == 'enum' or attrs.get('IsEnum'): + info.IsEnum = True + elif node_type == 'union' or attrs.get('IsUnion'): + info.IsUnion = True + elif node_type == 'typedef' or attrs.get('IsTypedef'): + info.IsTypedef = True + elif node_type == 'function' or attrs.get('IsFunction'): + info.IsFunction = True + elif node_type == 'variable' or attrs.get('IsVariable'): + info.IsVariable = True + elif node_type == 'define' or attrs.get('IsDefine'): + info.IsDefine = True + elif node_type == 'enum_member' or attrs.get('IsEnumMember'): + info.IsEnumMember = True + + if 'PtrCount' in attrs: + info.PtrCount = attrs['PtrCount'] + if 'OriginalType' in attrs: + info.OriginalType = attrs['OriginalType'] + if 'lineno' in attrs: + info.Lineno = attrs['lineno'] + if 'file' in attrs: + info.file = attrs['file'] + if 'members' in attrs: + info.Members = attrs['members'] + if 'IsPtr' in attrs: + info.IsPtr = attrs['IsPtr'] + if 'dims' in attrs: + info.ArrayDims = attrs['dims'] + if 'IsCpythonObject' in attrs: + info.IsCpythonObject = attrs['IsCpythonObject'] + if 'IsAnonymous' in attrs: + info.IsAnonymous = attrs['IsAnonymous'] + if 'IsPacked' in attrs: + info.IsPacked = attrs['IsPacked'] + if 'EnumName' in attrs: + info.EnumName = attrs['EnumName'] + if 'DefineValue' in attrs: + info.DefineValue = attrs['DefineValue'] + if 'BaseType' in attrs: + info.BaseType = self._DeserializeBaseType(attrs['BaseType']) + if 'name' in attrs and attrs['name'] != name: + info.Name = attrs['name'] + + # 其他属性存入 _extra + skip_keys: frozenset[str] = frozenset({'type', 'name', 'PtrCount', 'OriginalType', 'lineno', 'file', + 'members', 'IsPtr', 'dims', 'IsCpythonObject', 'IsAnonymous', + 'IsPacked', 'EnumName', 'DefineValue', 'BaseType', + 'IsStruct', 'IsEnum', 'IsUnion', 'IsTypedef', + 'IsFunction', 'IsVariable', 'IsDefine', 'IsEnumMember'}) + for key, value in attrs.items(): + if key not in skip_keys: + info.set(key, value) + + return info + + # ================================================================== + # 模块符号加载：四阶段管线 + # ================================================================== + + def LoadModuleSymbols(self, FullModulePath: str, asname: str, lineno: int = 0) -> list[str]: + return self._LoadSymbolsFromFile(FullModulePath, asname, lineno) + + def _LoadSymbolsFromFile(self, FilePath: str, namespace_prefix: str | list[str], lineno: int = 0) -> list[str]: + try: + # 提取模块名作为命名空间 + module_name: str = namespace_prefix[0] if isinstance(namespace_prefix, list) else namespace_prefix + + # Pass 1: 从 AST 提取原始符号信息 + extractor: ASTSymbolExtractor = ASTSymbolExtractor(self) + module_symbols: ModuleSymbols = extractor.extract(FilePath) + + # Pass 2: 解析类型信息（typedef、函数签名、define 值） + resolver: TypeResolver = TypeResolver(self) + resolver.resolve(module_symbols) + + # Pass 3: 插入符号到主命名空间（无前缀） + inserter: SymbolInserter = SymbolInserter(self) + inserter.set_namespace(module_name) + loaded: list[str] = inserter.insert(module_symbols) + + # Pass 4: 重新导出到命名空间前缀下 + prefixes: list[str] = [namespace_prefix] if isinstance(namespace_prefix, str) else namespace_prefix + reexporter: PackageReexporter = PackageReexporter(self) + reexporter.set_namespace(module_name) + loaded.extend(reexporter.reexport(module_symbols, prefixes, lineno)) + + return loaded + + except Exception as e: + _vlog().error(traceback.format_exc()) + self.diagnostics.error(FilePath, lineno, f"加载模块失败: {e}") + return [] + + # ================================================================== + # 类型解析辅助方法（委托到 LLVMTypeMapper） + # ================================================================== + + @staticmethod + def _CheckAnnotationHasCInline(annotation_node: ast.AST | None) -> bool: + return _CheckAnnotationHasCInline(annotation_node) + + # ================================================================== + # 符号插入方法（供 SymbolInserter / PackageReexporter 调用） + # ================================================================== + + def _InsertClassSymbol(self, FullName: str, TypeKind: str, lineno: int, FilePath: str, members: dict[str, CTypeInfo], IsCpythonObject: bool, IsPacked: bool = False) -> None: + info: CTypeInfo = CTypeInfo() + info.Lineno = lineno + info.file = FilePath + info.Members = members if members else {} + + if TypeKind == 'struct': + info.IsStruct = True + elif TypeKind == 'union': + info.IsUnion = True + elif TypeKind == 'enum': + info.IsEnum = True + elif TypeKind == 'renum': + info.IsStruct = True + info.IsRenum = True + info.IsEnum = True + elif TypeKind == 'exception': + info.IsStruct = True + info.IsExceptionClass = True + + if IsCpythonObject: + info.IsCpythonObject = True + if IsPacked: + info.IsPacked = True + + self._symbols[FullName] = info + + def _InsertEnumMemberSymbol(self, FullName: str, EnumName: str, lineno: int, FilePath: str, value: int | None = None) -> None: + info: CTypeInfo = CTypeInfo() + info.IsEnumMember = True + info.EnumName = EnumName + info.Lineno = lineno + info.file = FilePath + if value is not None: + info.value = value + self._symbols[FullName] = info + + def _InsertTypedefSymbol(self, FullName: str, OriginalType_kind: str | None, OriginalClass: CTypeInfo | str | None, lineno: int, FilePath: str, members: dict[str, CTypeInfo] | None = None) -> None: + info: CTypeInfo = CTypeInfo() + info.IsTypedef = True + info.Name = FullName + info.Lineno = lineno + info.file = FilePath + if members: + info.Members = members + + if isinstance(OriginalClass, CTypeInfo): + info.OriginalType = OriginalClass + elif OriginalType_kind == 'typedef' and OriginalClass and isinstance(OriginalClass, str) and ('*' in OriginalClass): + OriginalType: str = OriginalClass + if OriginalType == 'CVoid *': + OriginalType = 'void *' + info.OriginalType = OriginalType + elif OriginalType_kind == 'typedef' and OriginalClass == 'void': + info.OriginalType = 'void *' + elif OriginalType_kind == 'typedef' and OriginalClass: + info.OriginalType = OriginalClass + elif OriginalType_kind and OriginalClass and isinstance(OriginalClass, str): + info.OriginalType = f'{OriginalType_kind} {OriginalClass}' + else: + info.OriginalType = OriginalClass + + self._symbols[FullName] = info + + def _InsertModuleSymbol(self, name: str, lineno: int, FilePath: str) -> None: + info: CTypeInfo = CTypeInfo() + info.IsModuleAlias = True + info.Lineno = lineno + info.file = FilePath + self._symbols[name] = info + + def _InsertFuncSymbol(self, FullName: str, RetType: str | CTypeInfo | None, ParamTypes: list[str], lineno: int, FilePath: str, IsVariadic: bool = False, IsInline: bool = False) -> None: + info: CTypeInfo = CTypeInfo() + info.IsFunction = True + if isinstance(RetType, str): + info.FuncPtrReturn = CTypeInfo.FromTypeName(RetType) if RetType else CTypeInfo.VoidTypeInfo() + elif isinstance(RetType, CTypeInfo): + info.FuncPtrReturn = RetType + else: + info.FuncPtrReturn = CTypeInfo.VoidTypeInfo() + info.FuncPtrParams = [(f'arg{i}', pt) for i, pt in enumerate(ParamTypes)] + info.IsVariadic = IsVariadic + info.IsInline = IsInline + if IsInline: + info.Storage = t.CInline() + info.Lineno = lineno + info.file = FilePath + self._symbols[FullName] = info + + def _InsertDefineSymbol(self, FullName: str, DefineValue: int | str | float | bool | None, lineno: int, FilePath: str) -> None: + info: CTypeInfo = CTypeInfo() + info.IsDefine = True + info.DefineValue = DefineValue + info.Lineno = lineno + info.file = FilePath + self._symbols[FullName] = info + + def _eval_const_expr(self, node: ast.AST) -> int | float | str | None: + """计算常量表达式的值（委托到 ConstEvaluator）""" + return ConstEvaluator.eval_with_symtab(node, self) + + def _InsertAnonymousSymbol(self, FullName: str, IsUnion: bool, members: dict[str, CTypeInfo], lineno: int, FilePath: str) -> None: + info: CTypeInfo = CTypeInfo() + if IsUnion: + info.IsUnion = True + else: + info.IsStruct = True + info.IsAnonymous = True + info.Members = members if members else {} + info.Lineno = lineno + info.file = FilePath + self._symbols[FullName] = info diff --git a/lib/core/SymbolTypeResolver.py b/lib/core/SymbolTypeResolver.py new file mode 100644 index 0000000..deb567c --- /dev/null +++ b/lib/core/SymbolTypeResolver.py @@ -0,0 +1,272 @@ +"""Pass 2: TypeResolver — 解析 typedef 类型、函数签名、define 值""" +from __future__ import annotations +import ast +from typing import TYPE_CHECKING +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + +from lib.core.SymbolData import ( + ModuleSymbols, TypedefSymbolData, FuncSymbolData, DefineSymbolData +) +from lib.core.Handles.HandlesBase import CTypeInfo, CTypeHelper +from lib.core.LLVMTypeMapper import LLVMTypeMapper +from lib.core.SymbolUtils import IsTModule +from lib.includes import t + + +class TypeResolver: + """解析从 AST 提取的原始符号中的类型信息""" + + def __init__(self, symbol_table: SymbolTable): + self._symtab = symbol_table + self._type_mapper = LLVMTypeMapper(symbol_table) + # 本模块内已解析的 typedef: name → original_class（供同模块 typedef 间引用解析） + self._resolved_typedefs: dict[str, str | None] = {} + + def resolve(self, module_symbols: ModuleSymbols) -> ModuleSymbols: + """解析所有类型信息，就地更新 ModuleSymbols""" + # 构建 class_members 查找表（typedef Postdefinition 需要） + class_members: dict[str, dict[str, CTypeInfo]] = {cls.name: cls.members for cls in module_symbols.classes} + + # 解析 typedef（按顺序记录，供后续 typedef 引用同模块已解析的 typedef） + td: TypedefSymbolData + for td in module_symbols.typedefs: + self._resolve_typedef(td, class_members) + self._resolved_typedefs[td.name] = td.original_class + + # 解析函数签名 + func: FuncSymbolData + for func in module_symbols.functions: + self._resolve_function(func) + + # 解析 define 值 + define: DefineSymbolData + for define in module_symbols.defines: + self._resolve_define(define) + + return module_symbols + + # ------------------------------------------------------------------ + # Typedef 解析 + # ------------------------------------------------------------------ + + def _resolve_typedef(self, td: TypedefSymbolData, class_members: dict[str, dict[str, CTypeInfo]]): + """解析单个 typedef 的类型信息""" + is_postdef_with_typedef: bool = td.has_postdef and td.has_ctypedef + + if is_postdef_with_typedef: + self._resolve_postdef_typedef(td, class_members) + elif td.has_ctypedef: + self._resolve_ctypedef_typedef(td) + else: + # 无特殊注解的 typedef + td.original_type_kind = None + td.original_class = None + + def _resolve_postdef_typedef(self, td: TypedefSymbolData, class_members: dict[str, dict[str, CTypeInfo]]): + """解析 Postdefinition + CTypedef 组合注解""" + original_class: str | None = self._find_postdef_arg(td.annotation) + if original_class and original_class in class_members: + members_info: dict[str, CTypeInfo] = class_members[original_class] + original_type_kind: str = 'struct' + if original_class in self._symtab: + orig_type_info: CTypeInfo = self._symtab[original_class] + if orig_type_info.type_cls in (t.CStruct, t.CEnum, t.CUnion): + if orig_type_info.type_cls is t.CStruct: + original_type_kind = 'struct' + elif orig_type_info.type_cls is t.CEnum: + original_type_kind = 'enum' + elif orig_type_info.type_cls is t.CUnion: + original_type_kind = 'union' + td.original_type_kind = original_type_kind + td.original_class = original_class + td.members = members_info + else: + td.original_type_kind = None + td.original_class = None + + def _resolve_ctypedef_typedef(self, td: TypedefSymbolData): + """解析 CTypedef 注解""" + base_type_name: str | None = self._find_base_type(td.annotation) + _TYPE_QUALIFIERS: frozenset[str] = frozenset({'CTypedef', 'CConst', 'CVolatile', 'CInline', 'CStatic', 'CExtern', 'CDefine', 'CPostdefinition'}) + + if base_type_name == 'Callable': + original_type: CTypeInfo = CTypeInfo() + original_type.IsFuncPtr = True + original_type.FuncPtrReturn = CTypeInfo.VoidTypeInfo() + original_type.FuncPtrParams = [] + td.original_type_kind = 'typedef' + td.original_class = original_type + elif base_type_name and base_type_name not in _TYPE_QUALIFIERS: + cname: str | None = CTypeHelper.GetCName(base_type_name) + if cname: + td.original_class = cname + else: + td.original_class = base_type_name + td.original_type_kind = 'typedef' + else: + # 尝试从值表达式解析 + self._resolve_typedef_from_value(td) + + def _resolve_typedef_from_value(self, td: TypedefSymbolData): + """从值表达式解析 typedef 类型 + + BinOp（如 UINT | t.CPtr）优先于 _find_base_type 处理， + 因为 _find_base_type 是静态方法无法解析 Name 引用（如 UINT）。 + """ + if not td.value: + td.original_type_kind = None + td.original_class = None + return + + # 优先处理 BinOp（_find_base_type 无法解析 Name 引用） + if isinstance(td.value, ast.BinOp) and isinstance(td.value.op, ast.BitOr): + left_type: str = self._resolve_type_expr(td.value.left) + right_type: str = self._resolve_type_expr(td.value.right) + is_ptr_type: bool = right_type == 'ptr' or left_type == 'ptr' + if is_ptr_type: + base_name: str = left_type if left_type != 'ptr' else (right_type if right_type != 'ptr' else '') + original_type: str = (base_name + ' *') if base_name else 'void *' + td.original_class = original_type + td.original_type_kind = 'typedef' + elif left_type: + td.original_class = left_type + td.original_type_kind = 'typedef' + elif right_type: + td.original_class = right_type + td.original_type_kind = 'typedef' + else: + td.original_type_kind = None + td.original_class = None + return + + # 非 BinOp：使用 _find_base_type + value_base_type: str | None = self._find_base_type(td.value) + _PTR_MODIFIERS: frozenset[str] = frozenset({'CPtr', 'CArrayPtr', 'CConst', 'CVolatile', 'CInline', 'CStatic', 'CExtern'}) + + if value_base_type and value_base_type not in _PTR_MODIFIERS: + cname: str | None = CTypeHelper.GetCName(value_base_type) + if cname: + td.original_class = cname + else: + td.original_class = value_base_type + td.original_type_kind = 'typedef' + return + + if isinstance(td.value, ast.Name): + ref_name: str = td.value.id + # 先查本模块已解析的 typedef + if ref_name in self._resolved_typedefs: + ref_class: str | None = self._resolved_typedefs[ref_name] + if ref_class: + td.original_class = ref_class + td.original_type_kind = 'typedef' + return + # 再查 symtab（跨模块） + if ref_name in self._symtab: + ref_info: CTypeInfo = self._symtab[ref_name] + if ref_info and ref_info.IsTypedef and ref_info.OriginalType: + td.original_class = ref_info.OriginalType + td.original_type_kind = 'typedef' + return + + td.original_type_kind = None + td.original_class = None + + def _resolve_type_expr(self, node: ast.AST) -> str: + """解析类型表达式为 C 类型名字符串，优先查本模块已解析的 typedef + + 解决同模块 typedef 间引用（如 UINTPTR = UINT | t.CPtr 中的 UINT） + 在 TypeResolver 阶段尚未插入 symtab 的问题。 + """ + if isinstance(node, ast.Name): + if node.id in self._resolved_typedefs: + return self._resolved_typedefs[node.id] or '' + return self._type_mapper.resolve_typedef_value_type(node) + + # ------------------------------------------------------------------ + # 函数签名解析 + # ------------------------------------------------------------------ + + def _resolve_function(self, func: FuncSymbolData): + """解析函数的返回类型和参数类型""" + func.ret_type = self._type_mapper.get_func_ret_type_str(func.returns_node) + # 同时解析为 CTypeInfo，保留符号性（LLVM 类型字符串 'i64' 不区分有无符号） + if func.returns_node is not None: + try: + func.ret_ctype = CTypeInfo.FromNode(func.returns_node, self._symtab) + except Exception: + func.ret_ctype = None + func.param_types = [] + _arg_name: str + annotation: ast.expr | None + for _arg_name, annotation in func.params: + if annotation: + func.param_types.append(self._type_mapper.get_func_param_type_str(annotation)) + else: + func.param_types.append('i8*') + + # ------------------------------------------------------------------ + # Define 值解析 + # ------------------------------------------------------------------ + + def _resolve_define(self, define: DefineSymbolData): + """解析 define 常量值""" + if define.value_node: + define.value = self._symtab._eval_const_expr(define.value_node) + + # ------------------------------------------------------------------ + # 静态工具方法（原 _LoadSymbolsFromFile 中的闭包） + # ------------------------------------------------------------------ + + @staticmethod + def _find_base_type(node: ast.AST | None) -> str | None: + """从注解 AST 节点中查找基础类型名称""" + if isinstance(node, ast.Attribute): + if hasattr(node.value, 'id') and IsTModule(node.value.id): + if node.attr == 'CPtr': + return 'ptr' + return node.attr + elif isinstance(node, ast.Subscript): + if isinstance(node.value, ast.Attribute) and isinstance(node.value.value, ast.Name) and IsTModule(node.value.value.id) and node.value.attr == 'Callable': + return 'Callable' + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + left_result: str | None = TypeResolver._find_base_type(node.left) + if left_result == 'ptr': + right_result: str | None = TypeResolver._find_base_type(node.right) + return (right_result if right_result and right_result != 'ptr' else 'void') + ' *' + if left_result == 'Callable': + return 'Callable' + right_result: str | None = TypeResolver._find_base_type(node.right) + if right_result == 'ptr': + return (left_result if left_result and left_result != 'ptr' else 'void') + ' *' + if right_result == 'Callable': + return 'Callable' + # 类型拼接语义：CLong | CLong → CLongLong, CUnsignedLong | CLong → CUnsignedLongLong + long_types: frozenset[str] = frozenset({'CLong', 'CUnsignedLong'}) + if left_result in long_types and right_result in long_types: + if left_result == 'CUnsignedLong' or right_result == 'CUnsignedLong': + return 'CUnsignedLongLong' + return 'CLongLong' + if left_result: + return left_result + if right_result: + return right_result + return None + + @staticmethod + def _find_postdef_arg(node: ast.AST | None) -> str | None: + """从注解 AST 节点中查找 Postdefinition 参数""" + if isinstance(node, ast.Call): + if isinstance(node.func, ast.Attribute): + if hasattr(node.func.value, 'id') and IsTModule(node.func.value.id) and node.func.attr == 'Postdefinition': + if node.args and isinstance(node.args[0], ast.Name): + return node.args[0].id + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + left_result: str | None = TypeResolver._find_postdef_arg(node.left) + if left_result: + return left_result + right_result: str | None = TypeResolver._find_postdef_arg(node.right) + if right_result: + return right_result + return None diff --git a/lib/core/SymbolUtils.py b/lib/core/SymbolUtils.py new file mode 100644 index 0000000..5022d04 --- /dev/null +++ b/lib/core/SymbolUtils.py @@ -0,0 +1,206 @@ +"""共享工具函数：符号表处理中使用的 AST 分析工具""" +from __future__ import annotations +import ast +from lib.includes import t + +_T_MODULE_NAMES: frozenset[str] = frozenset({'t', t.__name__}) + + +def IsTModule(module_id: str, symtable: object | None = None) -> bool: + """检查模块标识符是否指向 t 类型模块（支持别名如 import t as tt）""" + if module_id in _T_MODULE_NAMES: + return True + if symtable is not None and hasattr(symtable, 'import_aliases'): + resolved: str = symtable.import_aliases.get(module_id, module_id) + if resolved in _T_MODULE_NAMES: + return True + return False + + +def IsTModulePath(module_path: str) -> bool: + """检查模块路径是否指向 t 类型模块""" + if module_path in _T_MODULE_NAMES: + return True + if module_path: + t_name: str + for t_name in _T_MODULE_NAMES: + if module_path == t_name or module_path.startswith(t_name + '.'): + return True + return False + + +def AnnotationContainsTType(node: ast.AST, TypeName: str) -> bool: + """检测 AST 节点中是否包含指定名称的 t 模块类型""" + if isinstance(node, ast.Attribute): + if (hasattr(node.value, 'id') and IsTModule(node.value.id) and + node.attr == TypeName): + return True + elif isinstance(node, ast.BinOp): + return AnnotationContainsTType(node.left, TypeName) or AnnotationContainsTType(node.right, TypeName) + elif isinstance(node, ast.Call): + if AnnotationContainsTType(node.func, TypeName): + return True + for arg in node.args: + if AnnotationContainsTType(arg, TypeName): + return True + elif isinstance(node, ast.Subscript): + if AnnotationContainsTType(node.value, TypeName): + return True + if AnnotationContainsTType(node.slice, TypeName): + return True + return False + + +def IsListAnnotation(annotation: ast.expr) -> bool: + """检查注解是否为栈上固定数组类型（t.CArray[...] 或 list[type, count]）""" + if not isinstance(annotation, ast.Subscript): + return False + value = annotation.value + # t.CArray[...] + if (isinstance(value, ast.Attribute) + and isinstance(value.value, ast.Name) + and value.value.id == 't' + and value.attr == 'CArray'): + return True + # list[type, count] — 仅匹配双参数 Tuple 切片（固定数组）， + # 单参数 list[type] 不匹配（走泛型类 list[T] 路径） + if (isinstance(value, ast.Name) + and value.id == 'list' + and isinstance(annotation.slice, ast.Tuple) + and len(annotation.slice.elts) == 2): + return True + return False + + +def ExtractListFromBinOp(annotation: ast.expr) -> ast.Subscript | None: + """从 BinOp(BitOr) 注解中提取 list[...] 部分，如 list[i32] | t.CPtr""" + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + for side in (annotation.left, annotation.right): + if IsListAnnotation(side): + return side + return None + + +class ListAnnotationParseResult: + """list[...] 注解解析结果。 + + is_pointer=True 表示单参数 list[type] → 指针模式； + is_pointer=False 表示 Tuple(2) 模式 list[type, count]。 + """ + __slots__ = ('elem_type_node', 'count_node', 'is_pointer') + + def __init__(self, elem_type_node: ast.expr, count_node: ast.expr | None, is_pointer: bool) -> None: + self.elem_type_node: ast.expr = elem_type_node + self.count_node: ast.expr | None = count_node + self.is_pointer: bool = is_pointer + + +def ParseListAnnotation(annotation: ast.expr) -> ListAnnotationParseResult | None: + """解析 list[...] 注解，统一处理 list[type, count] / list[type] / BinOp(BitOr) 形式。 + + 自动处理 BinOp(BitOr) 包裹（如 list[i32, N] | t.CPtr）。 + 返回 None 表示不是 list 注解。 + """ + list_node: ast.expr = annotation + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + extracted: ast.Subscript | None = ExtractListFromBinOp(annotation) + if extracted is None: + return None + list_node = extracted + if not IsListAnnotation(list_node): + return None + slice_node: ast.AST = list_node.slice + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + return ListAnnotationParseResult(slice_node.elts[0], slice_node.elts[1], is_pointer=False) + # 单参数 list[type] → 指针模式 + return ListAnnotationParseResult(slice_node, None, is_pointer=True) + + +def CheckAnnotationHasCInline(annotation_node: ast.AST | None) -> bool: + """检测注解节点中是否包含 CInline""" + if annotation_node is None: + return False + if isinstance(annotation_node, ast.Attribute): + if hasattr(annotation_node.value, 'id') and IsTModule(annotation_node.value.id) and annotation_node.attr == 'CInline': + return True + if isinstance(annotation_node, ast.Name): + if annotation_node.id == 'CInline': + return True + if isinstance(annotation_node, ast.BinOp) and isinstance(annotation_node.op, ast.BitOr): + return CheckAnnotationHasCInline(annotation_node.left) or CheckAnnotationHasCInline(annotation_node.right) + return False + + +def ExtractTypeNameFromBinOp(annotation: ast.expr) -> str | None: + """从 BinOp(BitOr) 注解左侧提取类型名（如 TypeA | TypeB → TypeA）。 + + 返回 Name.id 或 Attribute.attr，非 BinOp 返回 None。 + """ + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + left: ast.expr = annotation.left + if isinstance(left, ast.Name): + return left.id + if isinstance(left, ast.Attribute): + return left.attr + return None + + +def FindStructNameInAnnotation(node: ast.AST, struct_names: dict | set) -> str | None: + """递归在注解中查找结构体名（支持 BinOp(BitOr) 递归）。 + + 检查 Name.id、Constant.str、Attribute.attr 是否在 struct_names 中。 + 支持泛型特化类型（如 GSList[Param]），构造特化名后检查是否在 struct_names 中。 + """ + if isinstance(node, ast.Name) and node.id in struct_names: + return node.id + if isinstance(node, ast.Constant) and isinstance(node.value, str) and node.value in struct_names: + return node.value + if isinstance(node, ast.Attribute) and node.attr in struct_names: + return node.attr + if isinstance(node, ast.Subscript): + base_name: str | None = None + if isinstance(node.value, ast.Name): + base_name = node.value.id + elif isinstance(node.value, ast.Attribute): + base_name = node.value.attr + if base_name: + slice_node: ast.AST = node.slice + if hasattr(ast, 'Index') and isinstance(slice_node, ast.Index): + slice_node = slice_node.value + type_args: list[str] = [] + if isinstance(slice_node, ast.Name): + type_args.append(slice_node.id) + elif isinstance(slice_node, ast.Attribute): + type_args.append(slice_node.attr) + elif isinstance(slice_node, ast.Tuple): + for elt in slice_node.elts: + if isinstance(elt, ast.Name): + type_args.append(elt.id) + elif isinstance(elt, ast.Attribute): + type_args.append(elt.attr) + if type_args: + spec_name: str = base_name + ''.join(f'[{arg}]' for arg in type_args) + if spec_name in struct_names: + return spec_name + if base_name in struct_names: + return base_name + if isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + result: str | None = FindStructNameInAnnotation(node.left, struct_names) + if result: + return result + return FindStructNameInAnnotation(node.right, struct_names) + return None + + +def AnnotationContainsName(annotation: ast.AST, name: str) -> bool: + """检查注解中是否包含指定名称（支持 BinOp(BitOr) 递归）。 + + 同时检查 Attribute.attr 和 Name.id，比 ast.dump 字符串匹配更精确。 + """ + if isinstance(annotation, ast.Attribute): + return annotation.attr == name + if isinstance(annotation, ast.Name): + return annotation.id == name + if isinstance(annotation, ast.BinOp) and isinstance(annotation.op, ast.BitOr): + return AnnotationContainsName(annotation.left, name) or AnnotationContainsName(annotation.right, name) + return False diff --git a/lib/core/Translator/AnnotationLoader.py b/lib/core/Translator/AnnotationLoader.py new file mode 100644 index 0000000..3e64d43 --- /dev/null +++ b/lib/core/Translator/AnnotationLoader.py @@ -0,0 +1,381 @@ +from __future__ import annotations + +import ast +import os +import importlib +import types +from typing import Any + +import lib._bootstrap # noqa: F401 设置 sys.path（项目根 + lib/includes） + +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.SymbolUtils import IsTModule, AnnotationContainsName +from lib.includes import t +from lib.includes.t import CType, CEnum +from lib.constants.config import mode as _config_mode +from lib.core.VLogger import get_logger as _vlog + + +class AnnotationLoaderMixin: + """注解模块加载 Mixin + + 提供注解模块/文件的加载与符号表注册能力。 + """ + + def _LoadAnnotationModule(self, ModuleName: str, library_name: str | None = None, parse_method: str = 'auto') -> int: + """加载注解模块并将 CType 子类注册为 typedef + + 支持两种模式： + 1. AST 模式：如果 ModuleName 是文件路径，解析文件提取 CType + 2. importlib 模式：如果 ModuleName 是模块名，导入模块提取 CType + + Args: + ModuleName: 模块名或文件路径 + library_name: 手动指定的库名称，用于注册到符号表。如果为 None，则从 ModuleName 推断 + parse_method: 解析方法，可选值： + - 'auto': 自动检测（默认） + - 'file': 强制使用 AST 模式（文件路径） + - 'module': 强制使用 importlib 模式（模块名） + """ + def register_to_symboltable(module: types.ModuleType, lib_name: str | None = None, source_file: str | None = None) -> int: + """将模块中的 CType 子类和 CEnum 类注册到符号表""" + count: int = 0 + for AttrName in dir(module): + attr: type = getattr(module, AttrName) + # 优先检查是否是 CEnum 子类（枚举） + if isinstance(attr, type) and issubclass(attr, CEnum) and attr is not CEnum: + # 注册枚举类型 + EnumNode: CTypeInfo = CTypeInfo() + EnumNode.Name = AttrName + EnumNode.BaseType = t.CEnum() + EnumNode.IsEnum = True + EnumNode.set('enum_class', attr) + EnumNode.set('source', 'annotation_module') + EnumNode.library_name = lib_name + EnumNode.file = source_file + self.SymbolTable.insert(AttrName, EnumNode) + if lib_name: + FullName: str = f'{lib_name}.{AttrName}' + full_EnumNode: CTypeInfo = CTypeInfo() + full_EnumNode.Name = FullName + full_EnumNode.IsEnum = True + full_EnumNode.set('enum_class', attr) + full_EnumNode.set('source', 'annotation_module') + full_EnumNode.library_name = lib_name + full_EnumNode.file = source_file + self.SymbolTable.insert(FullName, full_EnumNode) + # 枚举成员也需要注册 + for MemberName in dir(attr): + if not MemberName.startswith('_'): + member: int = getattr(attr, MemberName, None) + if isinstance(member, int): + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = MemberName + MemberNode.BaseType = t.CEnum(AttrName) + MemberNode.value = member + MemberNode.EnumName = AttrName + MemberNode.library_name = lib_name + MemberNode.IsEnumMember = True + self.SymbolTable.insert(MemberName, MemberNode) + self.SymbolTable.insert(f"{AttrName}.{MemberName}", MemberNode) + self.SymbolTable.insert(f"{AttrName}_{MemberName}", MemberNode) + if lib_name: + self.SymbolTable.insert(f"{lib_name}.{AttrName}.{MemberName}", MemberNode) + self.SymbolTable.insert(f"{lib_name}_{AttrName}_{MemberName}", MemberNode) + count += 1 + # 检查是否是 CType 子类（typedef 别名） + elif isinstance(attr, type) and issubclass(attr, CType) and attr is not CType: + # 继承 CType 的类是类型别名（typedef），不是结构体 + # 使用手动指定的库名称作为前缀 + if lib_name: + FullName: str = f'{lib_name}.{AttrName}' + else: + FullName = AttrName + # 同时注册带前缀和不带前缀的名称（都是 typedef 别名） + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = AttrName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = attr.__name__ + TypedefNode.set('source', 'annotation_module') + TypedefNode.library_name = lib_name + TypedefNode.file = source_file + self.SymbolTable.insert(AttrName, TypedefNode) + if lib_name and FullName != AttrName: + FullTypedef_node: CTypeInfo = CTypeInfo() + FullTypedef_node.Name = FullName + FullTypedef_node.IsTypedef = True + FullTypedef_node.OriginalType = attr.__name__ + FullTypedef_node.set('source', 'annotation_module') + FullTypedef_node.library_name = lib_name + FullTypedef_node.file = source_file + self.SymbolTable.insert(FullName, FullTypedef_node) + count += 1 + # 检查是否是下划线前缀的 CType 子类（如 _CTypedef -> CTypedef） + elif AttrName.startswith('_') and len(AttrName) > 1: + PublicName: str = AttrName[1:] + PublicAttr: type | None = getattr(module, PublicName, None) + if PublicAttr is not None and not isinstance(attr, type): + continue + if isinstance(attr, type) and issubclass(attr, CType) and attr is not CType: + if not self.SymbolTable.has(PublicName): + if lib_name: + FullName: str = f'{lib_name}.{PublicName}' + else: + FullName = PublicName + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = PublicName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = attr.__name__ + TypedefNode.set('source', 'annotation_module') + TypedefNode.library_name = lib_name + TypedefNode.file = source_file + self.SymbolTable.insert(PublicName, TypedefNode) + if lib_name and FullName != PublicName: + FullTypedef_node: CTypeInfo = CTypeInfo() + FullTypedef_node.Name = FullName + FullTypedef_node.IsTypedef = True + FullTypedef_node.OriginalType = attr.__name__ + FullTypedef_node.set('source', 'annotation_module') + FullTypedef_node.library_name = lib_name + FullTypedef_node.file = source_file + self.SymbolTable.insert(FullName, FullTypedef_node) + count += 1 + return count + + # 确定解析方法 + use_file_mode: bool = False + if parse_method == 'file': + use_file_mode = True + elif parse_method == 'module': + use_file_mode = False + else: # auto + use_file_mode = os.path.isfile(ModuleName) + + # 确定库名称 + lib_name: str | None = library_name + if lib_name is None: + if use_file_mode: + # 从文件路径推断库名称 + lib_name = os.path.splitext(os.path.basename(ModuleName))[0] + else: + # 模块名就是库名称 + lib_name = ModuleName + + # 根据解析方法加载模块 + if use_file_mode: + # AST 模式：使用 AST 解析文件，不执行代码 + try: + # 临时保存当前的 EmbeddedAssignments 和 TypedefAssignments + saved_embedded: dict = self.EmbeddedAssignments.copy() + saved_typedef: dict = self.TypedefAssignments.copy() + + # 清空 PD 收集（注解文件不应该收集 PD 语句） + self.EmbeddedAssignments = {} + self.TypedefAssignments = {} + + # 直接调用 ParsePythonFile 解析文件提取类型信息 + # 注意：这不会执行任何 Python 代码，只是解析 AST + # UpdateCurrentFile=False 表示不修改 CurrentFile + self.ParsePythonFile(ModuleName, UpdateCurrentFile=False) + + # 恢复主代码文件的 PD 收集 + self.EmbeddedAssignments = saved_embedded + self.TypedefAssignments = saved_typedef + + # 从文件名推断类型前缀（如 test_t -> test_t.xxx） + TypePrefix: str = lib_name + count: int = 0 + + # 检查符号表中新增的类型（注解文件解析后的所有 struct 类型） + # 需要检查这些类型是否继承自 CType，如果是则是 typedef 别名 + new_types: list[str] = [] + for TypeName, TypeInfo in self.SymbolTable.items(): + if TypeInfo.IsStruct: + new_types.append(TypeName) + + # 再次遍历 AST，检测继承自 CType 的类 + try: + with open(ModuleName, 'r', encoding='utf-8') as f: + ann_content: str = f.read() + ann_tree: ast.Module = ast.parse(ann_content) + for node in ann_tree.body: + if isinstance(node, ast.ClassDef): + ClassName: str = node.name + # 检查是否有基类 + for base in node.bases: + if isinstance(base, ast.Name) and base.id == 'CType': + # 这个类继承自 CType，是 typedef 别名 + # 更新符号表中的类型为 typedef + if self.SymbolTable.has(ClassName): + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = ClassName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = self.SymbolTable[ClassName].get('OriginalType', f'struct {ClassName}') + TypedefNode.set('source', 'annotation_module') + TypedefNode.set('is_ctype_subclass', True) + self.SymbolTable.insert(ClassName, TypedefNode) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析注解文件类型定义失败: {_e}", "Exception") + + # 为所有新增的类型添加库前缀和 source 标记 + for TypeName in new_types: + # 添加带前缀的类型名 + FullName: str = f'{TypePrefix}.{TypeName}' + TypeInfo: CTypeInfo | dict[str, Any] = self.SymbolTable[TypeName].copy() + TypeInfo['source'] = 'annotation_module' + TypeInfo['original_name'] = TypeName + self.SymbolTable.insert(FullName, TypeInfo) + count += 1 + + # 再次遍历 AST，查找注解文件中的 typedef 语句（AnnAssign with t.CTypedef） + # 注意：必须在类定义处理完之后再处理 typedef + typedef_annots: list[tuple[str, str]] = [] + # 查找 CEnum 成员变量（AnnAssign with t.CEnum | t.State 或 t.CEnum） + cEnumMembers: list[tuple[str, int | None]] = [] + try: + with open(ModuleName, 'r', encoding='utf-8') as f: + ann_content: str = f.read() + ann_tree: ast.Module = ast.parse(ann_content) + for node in ann_tree.body: + # 检测 xxx: t.CEnum 形式的枚举成员变量 + if isinstance(node, ast.AnnAssign): + if isinstance(node.target, ast.Name) and node.annotation: + TargetName: str = node.target.id + # 检查注解是否是 BinOp (Union) 或直接是 Attribute + IsCenum: bool = False + if isinstance(node.annotation, ast.BinOp): + # 检查是否包含 CEnum + if AnnotationContainsName(node.annotation, 'CEnum'): + IsCenum = True + elif isinstance(node.annotation, ast.Attribute): + # 直接是 t.CEnum + if node.annotation.attr == t.CEnum.__name__: + IsCenum = True + + if IsCenum: + # 检查值是否是 Constant (枚举成员值) + if node.value and isinstance(node.value, ast.Constant): + value: int | str | float | None = node.value.value + else: + value = None + cEnumMembers.append((TargetName, value)) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析 CEnum 成员变量失败: {_e}", "Exception") + + # 处理 CEnum 成员变量 + for TargetName, value in cEnumMembers: + # 注册枚举成员 + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = TargetName + MemberNode.BaseType = t.CEnum(TypePrefix) + MemberNode.value = value + MemberNode.EnumName = TypePrefix + MemberNode.library_name = TypePrefix + MemberNode.IsEnumMember = True + self.SymbolTable.insert(TargetName, MemberNode) + self.SymbolTable.insert(f"{TypePrefix}.{TargetName}", MemberNode) + self.SymbolTable.insert(f"{TypePrefix}_{TargetName}", MemberNode) + + try: + with open(ModuleName, 'r', encoding='utf-8') as f: + ann_content: str = f.read() + ann_tree: ast.Module = ast.parse(ann_content) + for node in ann_tree.body: + # 检测 xxx: t.CTypedef = xxx 形式的 typedef 语句 + if isinstance(node, ast.AnnAssign): + if isinstance(node.target, ast.Name) and node.annotation and node.value: + TargetName: str = node.target.id + # 检查注解是否是 t.CTypedef + if isinstance(node.annotation, ast.Attribute): + if isinstance(node.annotation.value, ast.Name): + if IsTModule(node.annotation.value.id) and node.annotation.attr == 'CTypedef': + # 检查值是否是 Name + if isinstance(node.value, ast.Name): + original_name: str = node.value.id + typedef_annots.append((TargetName, original_name)) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析 typedef 语句失败: {_e}", "Exception") + + # 处理 typedef 注解 + for TargetName, original_name in typedef_annots: + # 检查原始类型是否在符号表中（带前缀或不带前缀） + orig_with_prefix: str = f'{TypePrefix}.{original_name}' + if self.SymbolTable.has(orig_with_prefix): + # 添加带前缀的 typedef 别名 + FullTypedef_name: str = f'{TypePrefix}.{TargetName}' + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = FullTypedef_name + TypedefNode.IsTypedef = True + orig_entry: CTypeInfo | dict[str, Any] = self.SymbolTable.lookup(orig_with_prefix) + orig_type_str: str = f'struct {original_name}' + if isinstance(orig_entry, dict): + if orig_entry.get('type') == 'typedef': + orig_type_str = orig_entry.get('OriginalType', f'struct {original_name}') + elif orig_entry.get('type') == 'enum': + orig_type_str = f'enum {original_name}' + elif orig_entry.IsTypedef: + orig_type_str = orig_entry.OriginalType or f'struct {original_name}' + elif orig_entry.IsEnum: + orig_type_str = f'enum {original_name}' + TypedefNode.OriginalType = orig_type_str + TypedefNode.set('source', 'annotation_module') + TypedefNode.set('original_name', TargetName) + self.SymbolTable.insert(FullTypedef_name, TypedefNode) + count += 1 + + self.AnnotationModules.add(lib_name) + return count + except Exception as e: + # 恢复 PD 收集 + self.EmbeddedAssignments = saved_embedded + self.TypedefAssignments = saved_typedef + self.LogWarning(f"AST模式加载注解模块失败: {ModuleName}, 错误: {e}") + return 0 + else: + # importlib 模式：导入模块 + try: + module: types.ModuleType = importlib.import_module(ModuleName) + count: int = register_to_symboltable(module, lib_name, None) + self.AnnotationModules.add(lib_name) + return count + except ImportError: + self.LogWarning(f"importlib模式加载注解模块失败: {ModuleName}") + return 0 + except Exception as e: + self.LogWarning(f"加载注解模块失败: {ModuleName}, 错误: {e}") + return 0 + + def _LoadAnnotationFiles(self, FilePaths: list[str | tuple]) -> None: + """加载多个注解文件并注册到符号表 + + Args: + FilePaths: 文件路径列表，每个元素可以是： + - str: 文件路径或模块名 + - tuple: (path, options) 或 (path, library_name, parse_method) + """ + for item in FilePaths: + if isinstance(item, tuple): + if len(item) >= 3: + path: str + library_name: str + parse_method: str + path, library_name, parse_method = item[:3] + count: int = self._LoadAnnotationModule(path, library_name, parse_method) + elif len(item) == 2: + path: str + library_name: str + path, library_name = item + count: int = self._LoadAnnotationModule(path, library_name, 'auto') + else: + count: int = self._LoadAnnotationModule(item) + else: + count: int = self._LoadAnnotationModule(item) + + # 别名，保持向后兼容 + LoadAnnotationFiles = _LoadAnnotationFiles diff --git a/lib/core/Translator/BaseTranslator.py b/lib/core/Translator/BaseTranslator.py new file mode 100644 index 0000000..03f8537 --- /dev/null +++ b/lib/core/Translator/BaseTranslator.py @@ -0,0 +1,157 @@ +from __future__ import annotations + +import ast +from typing import Any, Callable, Dict, TYPE_CHECKING + +from lib.core.Exportable import Exportable + +if TYPE_CHECKING: + import llvmlite.ir as ir + +from lib.core.Handles import ( + IfHandle, ForHandle, WhileHandle, + AnnAssignHandle, + HandlesTypeMerge, ExprHandle, BodyHandle, + ReturnHandle, AssignHandle, AugAssignHandle, DeleteHandle, + WithHandle, TryHandle, AssertHandle, RaiseHandle, MatchHandle, + ExprUtils, ExprOpsHandle, ExprAttrHandle, ExprBuiltinHandle, + ExprAsmHandle, ExprFormatHandle, ExprLambdaHandle, ExprCallHandle, + CSpecialCallHandle, TSpecialCallHandle, + ClassHandle, + FunctionHandle, + ImportHandle +) +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.SymbolTable import SymbolTable +from lib.core.SymbolNode import SymbolNode +from lib.constants.config import mode as _ConfigMode +from lib.core.VLogger import get_logger as _vlog + + +def _StrictLog(logger: Callable[[str], None], msg: str) -> None: + """在 strict 模式下记录异常日志""" + if _ConfigMode == "strict": + logger(msg) + + +class BaseTranslatorMixin: + """代码转换器基础 Mixin + + 提供 __init__、日志、调试输出与表达式处理等基础能力。 + """ + + def __init__(self) -> None: + self.VarScopes: list[Dict[str, SymbolNode]] = [{}] # 跟踪变量作用域，第一个是全局作用域 + self.FunctionReturnTypes: Dict[str, CTypeInfo] = {} # 记录函数名和其返回类型 + self.CurrentCReturnTypes: list[CTypeInfo] | None = None # 当前函数的 CReturn 类型列表 + self._CurrentCpythonObjectClass: str | None = None # 当前正在处理的 CPython 对象类名 + self.FunctionDefCache: dict = {} # 函数定义缓存 + self.OriginalLines: list[str] = [] # 原始代码行 + self.Content: str = '' + self.DebugFile: str | None = None # 调试输出文件路径 + self.IsHeader: bool = False # 是否生成头文件（仅处理定义、宏、导入等，忽略函数体） + self.AnnotationModules: set[str] = set() # 注解模块集合，用于识别类型定义模块 + self.Warnings: list[str] = [] # 警告日志 + self._ErrorStack: list[tuple[str, str, str]] = [] # 错误栈，按顺序收集错误位置 + self.Errors: list[dict[str, str | None]] = [] # 错误日志 + self.SliceCount: int = 0 # 切片计数 + self.SliceInfos: list = [] # 切片信息列表 + self.SliceLevel: int = 3 # 切片优化级别 + self.GeneratedTypes: set[str] = set() # 在当前代码中生成的类型（struct/typedef） + self.EmbeddedAssignments: dict[str, list] = {} # {ClassName: [var1, var2, ...]} 嵌入的实例化变量 + self.TypedefAssignments: dict[str, list] = {} # {ClassName: [(var1, IsPtr), ...]} 嵌入的 typedef/指针变量 + self.ChainAssignmentArrays: list = [] # 链式赋值数组形式 [(ClassName, VarName, ArraySize_node), ...] + self.tree: ast.AST | None = None # 保存解析后的 AST 树 + self.LibraryPaths: list[str] = ['.'] # 库搜索路径列表，默认包含当前目录 + self.InClass: bool = False # 是否正在处理 class 定义 + self._UserTypeModules: dict = {} + self._source_module_sig_files: dict = {} + self._global_function_default_args: dict = {} + self._global_function_param_names: dict = {} + self.exception_registry: dict = {} + self.exception_parents: dict = {} + self._next_exception_code: int = 100 + # 以下属性原先在外部懒注入，此处显式初始化以避免 hasattr/getattr 动态反射 + self.config: Any = None + self._ImportedModules: set[str] | None = None + self._t_c_imported_names: dict = {} + + # 导出表 + self.Exportable = Exportable(filename='') + + self.SymbolTable = SymbolTable(self) + self.IfHandler = IfHandle(self) + self.ForHandler = ForHandle(self) + self.WhileHandler = WhileHandle(self) + self.AnnAssignHandler = AnnAssignHandle(self) + self.TypeMergeHandler = HandlesTypeMerge(self) + self.ExprHandler = ExprHandle(self) + self.ExprUtils = ExprUtils(self) + self.ExprOpsHandle = ExprOpsHandle(self) + self.ExprAttrHandle = ExprAttrHandle(self) + self.ExprBuiltinHandle = ExprBuiltinHandle(self) + self.ExprAsmHandle = ExprAsmHandle(self) + self.ExprFormatHandle = ExprFormatHandle(self) + self.ExprLambdaHandle = ExprLambdaHandle(self) + self.ExprCallHandle = ExprCallHandle(self) + self.CSpecialCallHandle = CSpecialCallHandle(self) + self.TSpecialCallHandle = TSpecialCallHandle(self) + self.ClassHandler = ClassHandle(self) + self.FunctionHandler = FunctionHandle(self) + self.ImportHandler = ImportHandle(self) + self.BodyHandler = BodyHandle(self) + self.ReturnHandler = ReturnHandle(self) + self.AssignHandler = AssignHandle(self) + self.AugAssignHandler = AugAssignHandle(self) + self.DeleteHandler = DeleteHandle(self) + self.WithHandler = WithHandle(self) + self.TryHandler = TryHandle(self) + self.AssertHandler = AssertHandle(self) + self.RaiseHandler = RaiseHandle(self) + self.MatchHandler = MatchHandle(self) + + def LogWarning(self, message: str, LineNum: int | None = None) -> None: + """记录警告 + + Args: + message: 警告消息 + LineNum: 行号（可选） + """ + warning = f"Warning: {message}" + if LineNum is not None: + warning += f" (line {LineNum})" + self.Warnings.append(warning) + + def LogError(self, message: str, LineNum: int | None = None) -> None: + """记录错误 + + Args: + message: 错误消息 + LineNum: 行号 + """ + _vlog().error(message) + entry: dict[str, str | int | None] = {'message': message, 'line': LineNum} + self.Errors.append(entry) + + # 检查 config 中是否有 debug 文件 + DebugFile: str | None = None + if hasattr(self, 'config') and self.config: + DebugFile = getattr(self.config, 'debug', None) + + if DebugFile: + with open(DebugFile, 'a', encoding='utf-8') as f: + LineInfo: str = f' line {LineNum}' if LineNum else '' + f.write(f'[ERROR]{LineInfo}: {message}\n') + + def SetDebugFile(self, FilePath: str) -> None: + """设置调试输出文件""" + self.DebugFile = FilePath + + def DebugPrint(self, *args: Any, **kwargs: Any) -> None: + """输出调试信息到文件""" + if self.DebugFile: + with open(self.DebugFile, 'a', encoding='utf-8') as f: + print(*args, file=f, **kwargs) + + def HandleExpr(self, Node: ast.AST, UseSingleQuote: bool = False, VarType: ir.Type | None = None) -> Any: + return self.ExprHandler.HandleExpr(Node, UseSingleQuote, VarType) diff --git a/lib/core/Translator/ConstEval.py b/lib/core/Translator/ConstEval.py new file mode 100644 index 0000000..8785365 --- /dev/null +++ b/lib/core/Translator/ConstEval.py @@ -0,0 +1,305 @@ +from __future__ import annotations + +import ast +from typing import TYPE_CHECKING +import llvmlite.ir as ir + +from lib.core.ConstEvaluator import ConstEvaluator + +if TYPE_CHECKING: + from lib.core.LlvmCodeGenerator import LlvmCodeGenerator + + +class ConstEvalMixin: + """常量表达式求值 Mixin + + 提供数组初始化常量构建与常量表达式求值能力。 + """ + + @staticmethod + def _bswap_constant_value(val: int, width: int) -> int: + """编译期字节序交换""" + if width == 16: + return ((val & 0xFF) << 8) | ((val >> 8) & 0xFF) + elif width == 32: + return ((val & 0xFF) << 24) | ((val & 0xFF00) << 8) | ((val >> 8) & 0xFF00) | ((val >> 24) & 0xFF) + elif width == 64: + return ((val & 0xFF) << 56) | ((val & 0xFF00) << 40) | ((val & 0xFF0000) << 24) | ((val & 0xFF000000) << 8) | ((val >> 8) & 0xFF000000) | ((val >> 24) & 0xFF0000) | ((val >> 40) & 0xFF00) | ((val >> 56) & 0xFF) + return val + + def _BuildArrayInitConstants(self, value_node: ast.AST, ElemType: ir.Type, elem_type_node: ast.AST, ArrayCount: int, Gen: LlvmCodeGenerator, byte_order: str = '') -> list[ir.Constant] | None: + _zv: ir.Constant | None = Gen._zero_value_for_type(ElemType) + _fallback: ir.Constant = _zv if _zv is not None else ir.Constant(ElemType, ir.Undefined) + if not isinstance(value_node, (ast.List, ast.Set)): + return None + if isinstance(ElemType, ir.ArrayType): + constants: list[ir.Constant] = [] + for elt in value_node.elts: + if isinstance(elt, (ast.List, ast.Set)): + inner_consts: list[ir.Constant] | None = self._BuildMultiDimArrayInitConstants(elt, ElemType, Gen) + if inner_consts: + try: + constants.append(ir.Constant(ElemType, inner_consts)) + except Exception: # fallback to _fallback on constant construction failure + constants.append(_fallback) + else: + constants.append(_fallback) + else: + constants.append(_fallback) + while len(constants) < ArrayCount: + constants.append(_fallback) + return constants[:ArrayCount] + StructName: str | None = None + if isinstance(elem_type_node, ast.Name) and elem_type_node.id in Gen.structs: + StructName = elem_type_node.id + constants = [] + for elt in value_node.elts: + if StructName and isinstance(elt, ast.Call) and isinstance(elt.func, ast.Name) and elt.func.id == StructName: + const: ir.Constant | None = self.ClassHandler._BuildStructConstant(elt, StructName, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + elif isinstance(elt, ast.Constant): + # 字符串常量：根据引号类型决定 i8 或 i8* + if isinstance(elt.value, str): + const: ir.Constant | None = self._BuildStringLiteralConstant(elt, ElemType, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + else: + const = self._BuildScalarConstantUnified(elt, ElemType, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + elif isinstance(elt, ast.UnaryOp): + const = self._BuildScalarConstantUnified(elt, ElemType, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + elif isinstance(elt, ast.Call) and isinstance(elt.func, ast.Name) and elt.func.id == 'chr': + if elt.args and isinstance(elt.args[0], ast.Constant) and isinstance(elt.args[0].value, int): + try: + constants.append(ir.Constant(ElemType, elt.args[0].value & 0xFF)) + except Exception: # fallback to _fallback on constant construction failure + constants.append(_fallback) + else: + constants.append(_fallback) + elif isinstance(elt, ast.Call) and isinstance(elt.func, ast.Name) and elt.func.id == 'ord': + if elt.args and isinstance(elt.args[0], ast.Constant) and isinstance(elt.args[0].value, str) and len(elt.args[0].value) == 1: + try: + constants.append(ir.Constant(ElemType, ord(elt.args[0].value))) + except Exception: # fallback to _fallback on constant construction failure + constants.append(_fallback) + else: + constants.append(_fallback) + else: + const_val: int | float | str | None = self._try_eval_const_call(elt, Gen) + if const_val is not None: + try: + constants.append(ir.Constant(ElemType, const_val)) + except Exception: # fallback to _fallback on constant construction failure + constants.append(_fallback) + else: + constants.append(_fallback) + while len(constants) < ArrayCount: + constants.append(_fallback) + if byte_order == 'big' and isinstance(ElemType, ir.IntType) and ElemType.width in (16, 32, 64): + swapped: list[ir.Constant] = [] + for c in constants: + if isinstance(c, ir.Constant) and isinstance(c.type, ir.IntType) and isinstance(c.constant, int): + swapped_val: int = self._bswap_constant_value(int(c.constant), ElemType.width) + swapped.append(ir.Constant(ElemType, swapped_val)) + else: + swapped.append(c) + constants = swapped + return constants[:ArrayCount] + + def _BuildStringLiteralConstant(self, node: ast.Constant, target_type: ir.Type, Gen: LlvmCodeGenerator) -> ir.Constant | None: + """根据引号类型决定字符串常量的类型： + + - 单引号单字符 → i8 (char 值) + - 双引号/单引号多字符/三重引号 → i8* (C 字符串指针) + - 空字符串 → 0 (None)，传入 ptr 目标自动隐式转换 + """ + if not isinstance(node.value, str): + return None + s: str = node.value + # 空字符串 → 0 (None) + if len(s) == 0: + try: + return ir.Constant(target_type, 0) + except Exception: + return None + # 获取源代码以判断引号类型 + quote_char: str | None = None + is_triple: bool = False + col_offset: int = getattr(node, 'col_offset', 0) + node_lineno: int = getattr(node, 'lineno', 0) + # AST 的 lineno 可能不准确（常量节点可能指向下一行）， + # 因此在 lineno 和 lineno-1 两行中查找引号 + for try_lineno in (node_lineno, node_lineno - 1): + source_line: str | None = Gen._get_source_line(try_lineno) + if source_line and col_offset < len(source_line): + ch: str = source_line[col_offset] + if ch in ("'", '"'): + quote_char = ch + if col_offset + 2 < len(source_line) and source_line[col_offset:col_offset + 3] in ("'''", '"""'): + is_triple = True + break + # 判断是否为 char (i8)：单引号、非三重、单字符 + is_char: bool = (not is_triple) and (quote_char == "'") and (len(s) == 1) + if is_char: + # 单引号单字符 → i8 + if isinstance(target_type, ir.IntType): + try: + return ir.Constant(target_type, ord(s)) + except Exception: + return None + # 目标类型不是整数 → 类型不匹配 + return None + else: + # 双引号/多字符/三重 → i8* (C 字符串) + if isinstance(target_type, ir.PointerType): + str_val: str = s + '\x00' + str_bytes: bytes = str_val.encode('utf-8') + arr_type: ir.ArrayType = ir.ArrayType(ir.IntType(8), len(str_bytes)) + gv_name: str = f"str_const_{id(node)}" + gv: ir.GlobalVariable = ir.GlobalVariable(Gen.module, arr_type, name=gv_name) + gv.initializer = ir.Constant(arr_type, bytearray(str_bytes)) + gv.linkage = 'internal' + return ir.Constant(target_type, gv.get_reference()) + # 目标类型不是指针 → 类型不匹配 + return None + + def _BuildScalarConstantUnified(self, node: ast.AST, target_type: ir.Type, Gen: LlvmCodeGenerator) -> ir.Constant | None: + """统一的标量常量构建方法。 + + 合并原 HandlesClassDef._BuildScalarConstant 与 HandlesImports._BuildScalarConstant 的特性： + - 支持 bool/int/float/str/ast.UnaryOp(USub)/ast.Name(True/False) + - 字符串处理委托给 _BuildStringLiteralConstant + - 显式拒绝 BaseStructType + - 全部 try/except 保护 + """ + # 显式拒绝结构体类型 + if isinstance(target_type, ir.BaseStructType): + return None + # 处理常量节点 + if isinstance(node, ast.Constant): + # bool 必须在 int 之前判断（bool 是 int 的子类） + if isinstance(node.value, bool): + try: + return ir.Constant(target_type, 1 if node.value else 0) + except Exception: + return None + elif isinstance(node.value, int): + try: + return ir.Constant(target_type, node.value) + except Exception: + return None + elif isinstance(node.value, float): + try: + return ir.Constant(target_type, node.value) + except Exception: + return None + elif isinstance(node.value, str): + # 字符串处理委托给 _BuildStringLiteralConstant + const: ir.Constant | None = self._BuildStringLiteralConstant(node, target_type, Gen) + if const is not None: + return const + # 回退：直接创建字符串全局变量（i8*） + try: + return Gen._create_string_global(node.value) + except Exception: + return None + # 处理一元负号 + elif isinstance(node, ast.UnaryOp) and isinstance(node.op, ast.USub): + if isinstance(node.operand, ast.Constant) and isinstance(node.operand.value, int): + try: + return ir.Constant(target_type, -node.operand.value) + except Exception: + return None + # 处理 True/False 名称节点 + elif isinstance(node, ast.Name): + if node.id == 'True': + try: + return ir.Constant(target_type, 1) + except Exception: + return None + elif node.id == 'False': + try: + if isinstance(target_type, ir.PointerType): + return ir.Constant(target_type, None) + return ir.Constant(target_type, 0) + except Exception: + return None + return None + + def _try_eval_const_call(self, node: ast.AST, Gen: LlvmCodeGenerator) -> int | None: + if isinstance(node, ast.Call): + func_name: str | None = None + if isinstance(node.func, ast.Name): + func_name = node.func.id + elif isinstance(node.func, ast.Attribute): + func_name = node.func.attr + if func_name: + arg_vals: list[int | float | str | None] = [] + for arg in node.args: + v: int | float | str | None = self._try_eval_const_expr(arg, Gen) + if v is None: + return None + arg_vals.append(v) + if func_name == 'COLOR_RGB' and len(arg_vals) == 3: + r: int | float | str | None + g: int | float | str | None + b: int | float | str | None + r, g, b = arg_vals + return (255 << 24) | (int(b) << 16) | (int(g) << 8) | int(r) + if func_name in Gen._DefineConstants: + return Gen._DefineConstants[func_name] + for key in Gen.functions: + if key.endswith(f'__{func_name}'): + return None + return None + return self._try_eval_const_expr(node, Gen) + + def _try_eval_const_expr(self, node: ast.AST, Gen: LlvmCodeGenerator) -> int | float | str | None: + return ConstEvaluator.eval_with_symtab(node, self.SymbolTable) + + def _BuildMultiDimArrayInitConstants(self, value_node: ast.AST, ArrayType: ir.ArrayType, Gen: LlvmCodeGenerator) -> list[ir.Constant] | None: + inner_type: ir.Type = ArrayType.element + count: int = ArrayType.count + elts: list[ast.AST] = list(value_node.elts) if hasattr(value_node, 'elts') else [] + _zv: ir.Constant | None = Gen._zero_value_for_type(inner_type) + _fallback: ir.Constant = _zv if _zv is not None else ir.Constant(inner_type, ir.Undefined) + constants: list[ir.Constant] = [] + for elt in elts: + if isinstance(inner_type, ir.ArrayType): + if isinstance(elt, (ast.List, ast.Set)): + inner_consts: list[ir.Constant] | None = self._BuildMultiDimArrayInitConstants(elt, inner_type, Gen) + if inner_consts: + constants.append(ir.Constant(inner_type, inner_consts)) + else: + constants.append(_fallback) + else: + constants.append(_fallback) + elif isinstance(elt, ast.Constant): + const: ir.Constant | None = self._BuildScalarConstantUnified(elt, inner_type, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + elif isinstance(elt, ast.UnaryOp): + const = self._BuildScalarConstantUnified(elt, inner_type, Gen) + if const: + constants.append(const) + else: + constants.append(_fallback) + else: + constants.append(_fallback) + while len(constants) < count: + constants.append(_fallback) + return constants[:count] diff --git a/lib/core/Translator/LlvmGenerator.py b/lib/core/Translator/LlvmGenerator.py new file mode 100644 index 0000000..4ae3d90 --- /dev/null +++ b/lib/core/Translator/LlvmGenerator.py @@ -0,0 +1,642 @@ +from __future__ import annotations + +import ast +import re +import llvmlite.ir as ir + +from lib.core.LlvmCodeGenerator import LlvmCodeGenerator +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.Handles.HandlesBase import CTypeInfo as _CTypeInfo, CTypeHelper +from lib.includes import t +from lib.constants.config import mode as _config_mode +from lib.core.VLogger import get_logger as _vlog +from lib.core.DecoratorPass import run as _run_decorator_pass +from lib.core.SymbolUtils import IsTModule, AnnotationContainsName + + +class LlvmGeneratorMixin: + """LLVM IR 生成 Mixin + + 提供 LLVM IR 直接生成、模块级 LLVM IR 处理与 C 代码生成入口能力。 + """ + + def GenerateLlvmDirect(self, Tree: ast.Module) -> str: + Gen: LlvmCodeGenerator = self.LlvmGen + + # === 新增：轻量 AST 遍历，收集类图信息，解决前向引用 === + Gen._class_graph: dict[str, dict[str, bool | list[str]]] = {} + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.ClassDef): + bases: list[str] = [] + for base in Node.bases: + if hasattr(base, 'id'): + bases.append(base.id) + elif hasattr(base, 'attr'): + bases.append(base.attr) + has_methods: bool = any(isinstance(item, ast.FunctionDef) for item in Node.body) + # 检测 @t.NoVTable 装饰器（非多态继承基类标记） + has_novtable: bool = False + if hasattr(Node, 'decorator_list') and Node.decorator_list: + for decorator in Node.decorator_list: + if isinstance(decorator, ast.Attribute): + if getattr(decorator.value, 'id', None) == 't' and decorator.attr == 'NoVTable': + has_novtable = True + elif isinstance(decorator, ast.Name): + if decorator.id == 'NoVTable': + has_novtable = True + Gen._class_graph[Node.name] = { + 'bases': bases, + 'has_methods': has_methods, + 'has_novtable': has_novtable, + } + # === 结束新增 === + + # 预扫描：为所有 -> str 注解的函数注册正确的返回类型 i8* + # 这样在类方法中调用这些函数时，前向声明会使用正确的类型 + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.FunctionDef): + if Node.returns and isinstance(Node.returns, ast.Name) and Node.returns.id == 'str': + Gen.known_return_types[Node.name] = ir.PointerType(ir.IntType(8)) + + # 首先收集所有 CDefine 常量，确保类定义中可以引用 + def _extract_call_const_val(node: ast.AST) -> int | str | float | None: + if isinstance(node, ast.Call): + if isinstance(node.func, ast.Attribute): + if getattr(node.func.value, 'id', None) == 't': + if node.args and isinstance(node.args[0], ast.Constant): + return node.args[0].value + elif isinstance(node.func, ast.Name): + if node.args and isinstance(node.args[0], ast.Constant): + return node.args[0].value + return None + + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.AnnAssign) and isinstance(Node.target, ast.Name) and Node.value: + TypeInfo: CTypeInfo | None = self.TypeMergeHandler.GetCTypeInfo(Node.annotation) + if TypeInfo and TypeInfo.BaseType and ((isinstance(TypeInfo.BaseType, type) and issubclass(TypeInfo.BaseType, t.CDefine)) or isinstance(TypeInfo.BaseType, t.CDefine)): + val: int | str | float | None = None + if isinstance(Node.value, ast.Constant): + val = Node.value.value + else: + val = _extract_call_const_val(Node.value) + if val is not None: + if not hasattr(Gen, '_define_constants'): + Gen._define_constants: dict[str, int | str | float | bool] = {} + Gen._define_constants[Node.target.id] = val + existing = self.SymbolTable.lookup(Node.target.id) + if not (existing and existing.IsTypedef and existing.OriginalType): + sym_info: _CTypeInfo = _CTypeInfo() + sym_info.IsDefine = True + sym_info.DefineValue = val + self.SymbolTable.insert(Node.target.id, sym_info) + elif isinstance(Node, ast.Assign): + for target in Node.targets: + if isinstance(target, ast.Name): + TypeInfo: CTypeInfo | None = None + if Node.type_comment: + try: + tc_ast: ast.Expression = ast.parse(Node.type_comment, mode='eval') + TypeInfo = self.TypeMergeHandler.GetCTypeInfo(tc_ast.body) + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析类型注释失败: {_e}", "Exception") + if TypeInfo and TypeInfo.BaseType and ((isinstance(TypeInfo.BaseType, type) and issubclass(TypeInfo.BaseType, t.CDefine)) or isinstance(TypeInfo.BaseType, t.CDefine)): + val: int | str | float | None = None + if isinstance(Node.value, ast.Constant): + val = Node.value.value + else: + val = _extract_call_const_val(Node.value) + if val is not None: + if not hasattr(Gen, '_define_constants'): + Gen._define_constants: dict[str, int | str | float | bool] = {} + Gen._define_constants[target.id] = val + sym_info: _CTypeInfo = _CTypeInfo() + sym_info.IsDefine = True + sym_info.DefineValue = val + self.SymbolTable.insert(target.id, sym_info) + + # 预扫描所有顶层函数定义，填充 FunctionDefCache（不创建 LLVM 声明） + # 这样类方法编译时遇到未声明的函数可以按需前向声明 + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.FunctionDef): + FuncName: str = Node.name + if FuncName not in self.FunctionDefCache: + self.FunctionDefCache[FuncName] = Node + + # 首先处理所有导入语句，确保模块在类型解析前被加载 + for Node in ast.iter_child_nodes(Tree): + Gen._set_node_info(Node) + try: + if isinstance(Node, ast.Import): + self.ImportHandler._EmitImportDeclarationsLlvm(Node, Gen) + elif isinstance(Node, ast.ImportFrom): + self.ImportHandler._EmitImportFromDeclarationsLlvm(Node, Gen) + except Exception as e: + lineno: int = getattr(Node, 'lineno', 0) + source_line: str | None = Gen._get_source_line(lineno) + src_file: str = getattr(Gen, '_current_source_file', '') + src_path: str = src_file if src_file else 'unknown' + line_info: str = "源代码 (%s, line %d)" % (src_path, lineno) + if source_line: + line_info += ":\n %d | %s" % (lineno, source_line) + self._ErrorStack.append((type(e).__name__, str(e), line_info)) + + # 前向声明所有顶层函数，确保全局变量初始化时能引用函数指针 + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.FunctionDef): + Gen._set_node_info(Node) + self.FunctionHandler._EmitFunctionForwardDeclLlvm(Node, Gen) + + # 预创建全局变量（带初始化器），确保类方法编译时能引用全局数组等 + Gen._current_tree: ast.Module = Tree + for Node in ast.iter_child_nodes(Tree): + Gen._set_node_info(Node) + if isinstance(Node, ast.AnnAssign): + self.AssignHandler._EmitGlobalAnnAssignLlvm(Node, Gen) + elif isinstance(Node, ast.Assign): + self.AssignHandler._EmitGlobalAssignLlvm(Node, Gen) + + # 首先处理所有类定义（包括 CUnion），确保类型定义在使用前完成 + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.ClassDef): + Gen._set_node_info(Node) + self.ClassHandler._EmitClassLlvm(Node, Gen) + + Gen._generate_structs() + Gen._create_Vtable_globals() + + for Node in ast.iter_child_nodes(Tree): + Gen._set_node_info(Node) + try: + if isinstance(Node, ast.Import): + continue + elif isinstance(Node, ast.ImportFrom): + continue + elif isinstance(Node, ast.ClassDef): + continue + elif isinstance(Node, ast.FunctionDef): + self.FunctionHandler._EmitFunctionLlvm(Node, Gen) + elif isinstance(Node, ast.AnnAssign): + self.AssignHandler._EmitGlobalAnnAssignLlvm(Node, Gen) + elif isinstance(Node, ast.Assign): + self.AssignHandler._EmitGlobalAssignLlvm(Node, Gen) + elif isinstance(Node, ast.If): + pass + elif isinstance(Node, ast.Expr): + self._HandleModuleLevelExprLlvm(Node, Gen) + except Exception as e: + lineno: int = getattr(Node, 'lineno', 0) + source_line: str | None = Gen._get_source_line(lineno) + src_file: str = getattr(Gen, '_current_source_file', '') + src_path: str = src_file if src_file else 'unknown' + line_info: str = "源代码 (%s, line %d)" % (src_path, lineno) + if source_line: + line_info += ":\n %d | %s" % (lineno, source_line) + self._ErrorStack.append((type(e).__name__, str(e), line_info)) + raise + Gen._set_Vtable_initializers() + # 执行 DecoratorPass：为带自定义装饰器的函数生成 wrapper + _run_decorator_pass(Gen) + ir_code: str = Gen.finalize() + return ir_code + + def _HandleModuleLevelExprLlvm(self, Node: ast.Expr, Gen: LlvmCodeGenerator) -> None: + if not isinstance(Node.value, ast.Call): + return + call: ast.Call = Node.value + if not isinstance(call.func, ast.Attribute): + return + if not isinstance(call.func.value, ast.Name) or call.func.value.id != 'c': + return + func_name: str = call.func.attr + if func_name == 'LLVMIR': + self._HandleModuleLevelLLVMIR(call, Gen) + + def _HandleModuleLevelLLVMIR(self, Node: ast.Call, Gen: LlvmCodeGenerator) -> None: + if not Node.args: + return + first_arg: ast.expr = Node.args[0] + if isinstance(first_arg, ast.Constant) and isinstance(first_arg.value, str): + ir_text: str = first_arg.value + elif isinstance(first_arg, ast.JoinedStr): + parts: list[str] = [] + for value in first_arg.values: + if isinstance(value, ast.Constant) and isinstance(value.value, str): + parts.append(value.value) + elif isinstance(value, ast.FormattedValue): + parts.append('') + ir_text = ''.join(parts) + else: + return + self._InsertModuleLevelLLVMIR(Gen, ir_text) + + def _InsertModuleLevelLLVMIR(self, Gen: LlvmCodeGenerator, ir_text: str) -> None: + for line in ir_text.strip().split('\n'): + line = line.strip() + if not line: + continue + module_flags_match: re.Match[str] | None = re.match(r'!llvm\.module\.flags\s*=\s*!\{(!\d+)\}', line) + if module_flags_match: + ref: str = module_flags_match.group(1) + if not hasattr(Gen, '_pending_module_flags'): + Gen._pending_module_flags: list = [] + Gen._pending_module_flags_refs: str = ref + continue + metadata_match: re.Match[str] | None = re.match(r'(!\d+)\s*=\s*!\{(.+)\}', line) + if metadata_match: + md_id: str = metadata_match.group(1) + md_body: str = metadata_match.group(2) + fields: list[ir.Constant | ir.MetaDataString | str] = [] + for field_str in re.split(r',\s*', md_body): + field_str = field_str.strip() + if not field_str: + continue + int_m: re.Match[str] | None = re.match(r'^i32\s+(\d+)$', field_str) + if int_m: + fields.append(ir.Constant(ir.IntType(32), int(int_m.group(1)))) + continue + str_m: re.Match[str] | None = re.match(r'^"([^"]*)"$', field_str) + if str_m: + fields.append(ir.MetaDataString(Gen.module, str_m.group(1))) + continue + mdstr_m: re.Match[str] | None = re.match(r'^!"([^"]*)"$', field_str) + if mdstr_m: + fields.append(ir.MetaDataString(Gen.module, mdstr_m.group(1))) + continue + int_m2: re.Match[str] | None = re.match(r'^(\d+)$', field_str) + if int_m2: + fields.append(ir.Constant(ir.IntType(32), int(int_m2.group(1)))) + continue + ref_m: re.Match[str] | None = re.match(r'^(!\d+)$', field_str) + if ref_m: + fields.append(ref_m.group(1)) + continue + if fields: + if not hasattr(Gen, '_pending_metadata'): + Gen._pending_metadata: list[tuple[str, list]] = [] + Gen._pending_metadata.append((md_id, fields)) + continue + self._FlushModuleMetadata(Gen) + + def _FlushModuleMetadata(self, Gen: LlvmCodeGenerator) -> None: + if not hasattr(Gen, '_pending_metadata') or not Gen._pending_metadata: + return + md_map: dict[str, ir.MetaDataString] = {} + for md_id, fields in Gen._pending_metadata: + resolved: list[ir.Constant | ir.MetaDataString] = [] + for f in fields: + if isinstance(f, str) and f.startswith('!'): + ref: ir.MetaDataString | None = md_map.get(f) + if ref: + resolved.append(ref) + else: + resolved.append(f) + md_value: ir.MetaDataString = Gen.module.add_metadata(resolved) + md_map[md_id] = md_value + if hasattr(Gen, '_pending_module_flags_refs'): + ref: str = Gen._pending_module_flags_refs + md_node: ir.MetaDataString | None = md_map.get(ref) + if md_node: + Gen.module.add_named_metadata('llvm.module.flags', md_node) + Gen._pending_metadata = [] + Gen._pending_module_flags_refs = None + + def GenerateCCode(self, Tree: ast.Module, target: str = 'llvm') -> str: + """生成代码 + + Args: + Tree: Python AST 树 + target: 目标代码类型，仅支持 'llvm' + + Returns: + 生成的代码字符串 + """ + self._generated_vars: set[str] = set() + self.VarScopes = [{}] # 全局作用域 + + # 预扫描 import 语句，注册别名，确保类型解析时能正确解析模块别名 + if not getattr(self, '_ImportedModules', None): + self._ImportedModules: set[str] = set() + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.Import): + for alias in Node.names: + name: str = alias.name + if name in ('c', 't'): + continue + self._ImportedModules.add(name) + if alias.asname: + self.SymbolTable.import_aliases[alias.asname] = name + elif isinstance(Node, ast.ImportFrom): + module_name: str | None = Node.module + if module_name and module_name not in ('c', 't'): + self._ImportedModules.add(module_name) + for alias in Node.names: + if alias.asname: + self.SymbolTable.import_aliases[alias.asname] = f"{module_name}.{alias.name}" + + for Node in ast.iter_child_nodes(Tree): + if isinstance(Node, ast.ClassDef): + ClassName: str = Node.name + + # 检查是否是特殊类型（枚举或联合体） + IsSpecialType: bool = False + if Node.bases: + for base in Node.bases: + if hasattr(base, 'attr'): + if base.attr in ['CEnum', 'Enum', 'CUnion', 'REnum']: + IsSpecialType = True + break + elif hasattr(base, 'id'): + if base.id in ['CEnum', 'Enum', 'CUnion', 'REnum']: + IsSpecialType = True + break + + # 如果是特殊类型，跳过这里的处理（由 HandleClassDef 处理） + if IsSpecialType: + continue + + members: dict[str, CTypeInfo] = {} + IsTypedef: bool = False + TypedefName: str | None = None + + if hasattr(Node, 'annotation') and Node.annotation: + try: + if AnnotationContainsName(Node.annotation, 'CTypedef'): + IsTypedef = True + if isinstance(Node.annotation, ast.Call): + if Node.annotation.args: + if isinstance(Node.annotation.args[0], ast.Constant): + TypedefName = Node.annotation.args[0].value + except Exception as _e: + if _config_mode == "strict": + raise + _vlog().warning(f"解析 typedef 注解失败: {_e}", "Exception") + + if not IsTypedef: + for item in Node.body: + if isinstance(item, ast.Assign): + for target in item.targets: + if isinstance(target, ast.Name) and target.id == '__annotations__': + if isinstance(item.value, ast.Dict): + for key, value in zip(item.value.keys, item.value.values): + if isinstance(key, ast.Constant) and key.value == '__type__': + if AnnotationContainsName(value, 'CTypedef'): + IsTypedef = True + + if IsTypedef: + TypedefKey: str = TypedefName if TypedefName else ClassName + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = TypedefKey + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = f'struct {ClassName}' + TypedefNode.set('IsComplete', True) + TypedefNode.file = '' + self.SymbolTable.insert(TypedefKey, TypedefNode) + + for item in Node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + VarName: str = item.target.id + try: + TypeInfo: CTypeInfo | None = self.TypeMergeHandler.GetCTypeInfo(item.annotation) + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + IsPtr: bool = TypeInfo.IsPtr + if not IsPtr: + if isinstance(item.annotation, ast.BinOp) and isinstance(item.annotation.op, ast.BitOr): + if AnnotationContainsName(item.annotation, 'CPtr'): + IsPtr = True + dims: list[int] = [] + if hasattr(item.annotation, 'slice') and isinstance(item.annotation.slice, ast.Constant): + if isinstance(item.annotation.value, ast.Name): + if IsTModule(item.annotation.value.id, self.SymbolTable): + if hasattr(item.annotation, 'slice'): + try: + dims.append(int(item.annotation.slice.value)) + except Exception as _e: + if _config_mode == "strict": + self.LogWarning(f"异常被忽略: {_e}") + MemberInfo: CTypeInfo = TypeInfo.Copy() + if dims: + MemberInfo.ArrayDims = [str(d) for d in dims] + members[VarName] = MemberInfo + except Exception as _e: + if _config_mode == "strict": + self.LogWarning(f"异常被忽略: {_e}") + + if not IsTypedef: + StructNode: CTypeInfo = CTypeInfo() + StructNode.Name = ClassName + StructNode.IsStruct = True + StructNode.Members = members or {} + StructNode.file = '' + existing: CTypeInfo | None = self.SymbolTable.lookup(ClassName) + if existing: + if hasattr(existing, 'IsCpythonObject') and existing.IsCpythonObject: + StructNode.IsCpythonObject = True + self.SymbolTable.insert(ClassName, StructNode) + elif isinstance(Node, ast.FunctionDef): + FuncName: str = Node.name + if not self.SymbolTable.has(FuncName): + FuncNode: CTypeInfo = CTypeInfo() + FuncNode.Name = FuncName + FuncNode.IsFunction = True + FuncNode.file = '' + self.SymbolTable.insert(FuncName, FuncNode) + elif isinstance(Node, ast.AnnAssign): + if isinstance(Node.target, ast.Name): + VarName: str = Node.target.id + + VarType: str = 'unknown' + IsPtr: bool = False + IsTypedef_assignment: bool = False + + if Node.annotation: + try: + if AnnotationContainsName(Node.annotation, 'CTypedef'): + IsTypedef_assignment = True + except Exception as _e: + if _config_mode == "strict": + self.LogWarning(f"异常被忽略: {_e}") + + if IsTypedef_assignment and Node.value: + if isinstance(Node.value, ast.Name): + OriginalType: str = Node.value.id + if OriginalType in self.GeneratedTypes: + OriginalInfo: CTypeInfo | None = self.SymbolTable.lookup(OriginalType) + if OriginalInfo: + if isinstance(OriginalInfo, dict): + actual_type: str = OriginalInfo.get('type', 'struct') + elif isinstance(OriginalInfo, CTypeInfo): + if OriginalInfo.IsEnum: + actual_type = 'enum' + elif OriginalInfo.IsTypedef: + actual_type = 'typedef' + elif OriginalInfo.IsStruct: + actual_type = 'struct' + else: + actual_type = 'struct' + else: + actual_type = 'struct' + if actual_type == 'enum': + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = f'enum {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + elif actual_type == 'typedef': + TypedefNode = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + if isinstance(OriginalInfo, dict): + TypedefNode.OriginalType = OriginalInfo.get('OriginalType', f'struct {OriginalType}') + elif isinstance(OriginalInfo, CTypeInfo) and OriginalInfo.OriginalType: + TypedefNode.OriginalType = OriginalInfo.OriginalType + else: + TypedefNode.OriginalType = f'struct {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + else: + TypedefNode = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = f'struct {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + self.GeneratedTypes.add(VarName) + continue + else: + OriginalInfo: CTypeInfo | None = self.SymbolTable.lookup(OriginalType) + if OriginalInfo: + if isinstance(OriginalInfo, dict): + actual_type: str = OriginalInfo.get('type', 'struct') + elif isinstance(OriginalInfo, CTypeInfo): + if OriginalInfo.IsEnum: + actual_type = 'enum' + elif OriginalInfo.IsTypedef: + actual_type = 'typedef' + elif OriginalInfo.IsStruct: + actual_type = 'struct' + else: + actual_type = 'struct' + else: + actual_type = 'struct' + if actual_type == 'typedef': + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + if isinstance(OriginalInfo, dict): + TypedefNode.OriginalType = OriginalInfo.get('OriginalType', f'struct {OriginalType}') + elif isinstance(OriginalInfo, CTypeInfo) and OriginalInfo.OriginalType: + TypedefNode.OriginalType = OriginalInfo.OriginalType + else: + TypedefNode.OriginalType = f'struct {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + self.GeneratedTypes.add(VarName) + if isinstance(OriginalInfo, dict): + OriginalInfo['skip_generation'] = True + continue + elif actual_type == 'struct': + TypedefNode = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = f'struct {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + self.GeneratedTypes.add(VarName) + if isinstance(OriginalInfo, dict): + OriginalInfo['skip_generation'] = True + continue + elif actual_type == 'enum': + TypedefNode = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = f'enum {OriginalType}' + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + self.GeneratedTypes.add(VarName) + if isinstance(OriginalInfo, dict): + OriginalInfo['skip_generation'] = True + continue + elif isinstance(Node.value, ast.Attribute): + CName: str | None = CTypeHelper.GetCName(Node.value.attr) + if CName: + TypedefNode: CTypeInfo = CTypeInfo() + TypedefNode.Name = VarName + TypedefNode.IsTypedef = True + TypedefNode.OriginalType = CName + TypedefNode.file = '' + self.SymbolTable.insert(VarName, TypedefNode) + self.GeneratedTypes.add(VarName) + continue + + if Node.annotation: + try: + TypeInfo: CTypeInfo | None = self.TypeMergeHandler.GetCTypeInfo(Node.annotation) + IsPtr = TypeInfo.IsPtr if TypeInfo else False + except Exception as _e: + if _config_mode == "strict": + self.LogWarning(f"异常被忽略: {_e}") + # 如果符号已经存在且是 typedef，不要覆盖它 + existing = self.SymbolTable.lookup(VarName) + if existing and existing.IsTypedef and existing.OriginalType: + # typedef 已正确插入，保持现有条目不覆盖 + pass + else: + var_node: CTypeInfo = CTypeInfo() + var_node.Name = VarName + var_node.IsVariable = True + var_node.PtrCount = 1 if IsPtr else 0 + var_node.file = '' + self.SymbolTable.insert(VarName, var_node) + + self.LlvmGen: LlvmCodeGenerator = LlvmCodeGenerator( + triple=getattr(self, 'triple', None), + datalayout=getattr(self, 'datalayout', None), + ) + self.LlvmGen._Trans = self + self.LlvmGen._import_handler_ref = self.ImportHandler + if hasattr(self, '_module_sha1'): + self.LlvmGen.module_sha1 = self._module_sha1 + if hasattr(self, '_ModuleSha1Map'): + self.LlvmGen.ModuleSha1Map = self._ModuleSha1Map + if hasattr(self, '_struct_sha1_map'): + self.LlvmGen._struct_sha1_map = self._struct_sha1_map + if hasattr(self, '_temp_dir'): + self.LlvmGen._temp_dir = self._temp_dir + if hasattr(self, '_export_extern_funcs'): + self.LlvmGen._export_extern_funcs = self._export_extern_funcs + self.LlvmGen.setup_from_symbol_table(self.SymbolTable) + if hasattr(self, 'CurrentFile') and self.CurrentFile: + self.LlvmGen._set_source_info(self.CurrentFile) + + # --- 修复：共享 vtable 状态跨模块 --- + # 根因：LlvmGen 每模块重新创建，class_vtable/class_methods 等是实例级属性， + # 不跨模块共享。导致模块 B 编译时 _specialize_generic_class 触发的 + # pool.alloc(48) 找不到 MemManager 的虚表状态，直接调用 stub 返回 NULL → 段错误。 + # 修复：将 vtable 状态存储在 Translator 对象上并赋值给每个新 Gen。 + if not hasattr(self, '_shared_class_vtable'): + self._shared_class_vtable: set[str] = set() + self._shared_cross_module_vtable: set[str] = set() + self._shared_class_methods: dict[str, list[str]] = {} + self._shared_cross_module_novtable: set[str] = set() + # 合并 setup_from_symbol_table 创建的实例级 class_methods 到共享 dict + # （仅添加新类，不覆盖已有方法列表） + for _cls_name, _methods in self.LlvmGen.class_methods.items(): + if _cls_name not in self._shared_class_methods: + self._shared_class_methods[_cls_name] = _methods + else: + for _m in _methods: + if _m not in self._shared_class_methods[_cls_name]: + self._shared_class_methods[_cls_name].append(_m) + # 替换为共享状态 + self.LlvmGen.class_vtable = self._shared_class_vtable + self.LlvmGen._cross_module_vtable_classes = self._shared_cross_module_vtable + self.LlvmGen.class_methods = self._shared_class_methods + self.LlvmGen._cross_module_novtable = self._shared_cross_module_novtable + # --- 修复结束 --- + + return self.GenerateLlvmDirect(Tree) diff --git a/lib/core/Translator/PythonParser.py b/lib/core/Translator/PythonParser.py new file mode 100644 index 0000000..9bcabcc --- /dev/null +++ b/lib/core/Translator/PythonParser.py @@ -0,0 +1,452 @@ +from __future__ import annotations + +import ast +import json +import struct + +from lib.utils.helpers import DetectFileType +from lib.core.Handles.HandlesBase import CTypeInfo +from lib.core.SymbolUtils import IsTModule, AnnotationContainsName +from lib.includes import t +from lib.constants.config import mode as _ConfigMode +from lib.core.VLogger import get_logger as _vlog + + +class PythonParserMixin: + """Python 文件解析 Mixin + + 提供辅助文件解析、symbin 文件加载与 Python 源文件解析能力。 + """ + + def ParseHelperFiles(self, HelperFiles: list[str], encoding: str = 'utf-8') -> None: + """解析辅助文件，提取符号信息 + + 支持 .py, .c, .h 和 .symbin 文件 + """ + for FilePath in HelperFiles: + ext: str = DetectFileType(FilePath) + if ext == '.py': + self.ParsePythonFile(FilePath, encoding, UpdateCurrentFile=False) + elif ext == '.c': + pass # .c 文件解析已移除，不再使用旧的正则解析路径 + elif FilePath.endswith('.symbin'): + self.LoadSymbinFile(FilePath) + + def LoadSymbinFile(self, FilePath: str) -> None: + """从.symbin文件加载符号表 + + Args: + FilePath: .symbin文件路径 + """ + try: + with open(FilePath, 'rb') as f: + data: bytes = f.read() + + # 导入序列化函数（避免循环导入） + # 解析二进制格式 + if len(data) < 4: + _vlog().warning(f"无效的 symbin 文件 (太短): {FilePath}") + return + + # 解析4字节长度头（小端序） + length: int = struct.unpack(' None: + """解析Python文件，提取类、函数、变量信息 + + Args: + FilePath: 要解析的文件路径 + encoding: 文件编码 + UpdateCurrentFile: 是否更新 CurrentFile（默认为 True，处理导入时设为 False） + """ + try: + with open(FilePath, 'r', encoding=encoding) as f: + content: str = f.read() + + # 保存当前处理的文件路径（仅在主文件处理时更新） + if UpdateCurrentFile: + self.CurrentFile = FilePath + + # 解析Python代码为AST + tree: ast.Module = ast.parse(content) + + # 保存 AST 树供后续使用 + self.tree = tree + + # 第一遍：检测特殊赋值语句，收集跟在 class 后面的 + # 遍历 tree.body，找到所有 ClassDef，然后收集紧随其后的赋值语句 + i: int = 0 + while i < len(tree.body): + node: ast.stmt = tree.body[i] + + # 检测 ClassDef + if isinstance(node, ast.ClassDef): + ClassName: str = node.name + # 使用类名_行号作为唯一 key + ClassKey: str = f"{ClassName}_{node.lineno}" + + # 检查后面的语句，收集特殊赋值 + j: int = i + 1 + while j < len(tree.body): + NextNode: ast.stmt = tree.body[j] + + if isinstance(NextNode, ast.Assign): + # 检测 xxx = xxx 或 xxx = yyy + if len(NextNode.targets) == 1 and isinstance(NextNode.targets[0], ast.Name): + TargetName: str = NextNode.targets[0].id + if isinstance(NextNode.value, ast.Name): + ValueName: str = NextNode.value.id + if ValueName == ClassName: + # 这是 class 的实例化 + if ClassKey not in self.EmbeddedAssignments: + self.EmbeddedAssignments[ClassKey] = [] + self.EmbeddedAssignments[ClassKey].append(TargetName) + j += 1 + continue + elif isinstance(NextNode, ast.AnnAssign): + # 检测 xxx: t.Postdefinition(yyy) 或 xxx: t.CTypedef = xxx 或 xxx: t.CPtr = xxx + if isinstance(NextNode.target, ast.Name): + TargetName: str = NextNode.target.id + # 先处理新格式：t.Postdefinition(...) + annotation: ast.expr | None = NextNode.annotation + PostdefNode: ast.Call | None = None + + # 递归搜索整个注解树，找到 t.Postdefinition(...) 调用 + def find_PostdefNode(node: ast.AST) -> ast.Call | None: + if isinstance(node, ast.Call): + if isinstance(node.func, ast.Attribute): + if hasattr(node.func.value, 'id') and IsTModule(node.func.value.id, self.SymbolTable) and node.func.attr == 'Postdefinition': + return node + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + LeftResult: ast.Call | None = find_PostdefNode(node.left) + if LeftResult: + return LeftResult + RightResult: ast.Call | None = find_PostdefNode(node.right) + if RightResult: + return RightResult + return None + + PostdefNode = find_PostdefNode(annotation) + # 检查是否找到 t.Postdefinition(...) 调用 + if PostdefNode: + if (isinstance(PostdefNode.func, ast.Attribute) and + hasattr(PostdefNode.func, 'value') and + hasattr(PostdefNode.func.value, 'id') and + IsTModule(PostdefNode.func.value.id, self.SymbolTable) and + PostdefNode.func.attr == 'Postdefinition' and + PostdefNode.args): + # 检查第一个参数是否是当前类名 + arg: ast.expr = PostdefNode.args[0] + if isinstance(arg, ast.Name) and arg.id == ClassName: + # 检查是否包含 CTypedef 或 CPtr + HasCtypedef: bool = AnnotationContainsName(NextNode.annotation, 'CTypedef') + IsPtr: bool = AnnotationContainsName(NextNode.annotation, 'CPtr') + + # 解析维度列表（第二个参数） + dimensions: list[ast.expr] = [] + if len(PostdefNode.args) > 1: + # 第二个参数应该是一个列表 + dim_arg: ast.expr = PostdefNode.args[1] + if isinstance(dim_arg, ast.List): + for DimExpr in dim_arg.elts: + dimensions.append(DimExpr) + + # 如果没有从 Postdefinition 参数中获取维度，尝试从类型注解中的 Subscript 获取 + # 例如：memory_block_t[MAX_ORDER + 1] | t.CPtr + if not dimensions: + def extract_subscript_dims(node: ast.AST) -> list[ast.expr]: + """递归提取 Subscript 节点的维度""" + dims: list[ast.expr] = [] + if isinstance(node, ast.Subscript): + # 提取当前维度 + if isinstance(node.slice, ast.Constant): + dims.append(ast.Constant(value=node.slice.value)) + elif isinstance(node.slice, ast.Name): + dims.append(ast.Name(id=node.slice.id, ctx=ast.Load())) + elif isinstance(node.slice, ast.BinOp): + dims.append(node.slice) + # 递归提取更外层的维度 + dims.extend(extract_subscript_dims(node.value)) + elif isinstance(node, ast.BinOp) and isinstance(node.op, ast.BitOr): + # 处理 Type | t.CPtr 的情况，检查左侧 + dims.extend(extract_subscript_dims(node.left)) + return dims + + # 从注解中提取维度 + subscript_dims: list[ast.expr] = extract_subscript_dims(annotation) + if subscript_dims: + # 反转维度顺序（从内到外） + dimensions = list(reversed(subscript_dims)) + + # 获取赋值值 + value: ast.expr | None = None + if hasattr(NextNode, 'value') and NextNode.value is not None: + value = NextNode.value + + # 添加到对应的 assignments + # 对于 t.Postdefinition(...)，只有包含 CTypedef 或 CPtr 时才视为 typedef 类型 + if HasCtypedef or IsPtr: + if ClassKey not in self.TypedefAssignments: + self.TypedefAssignments[ClassKey] = [] + # 如果有维度信息，使用字典存储 + if dimensions: + self.TypedefAssignments[ClassKey].append({ + 'name': TargetName, + 'IsPtr': IsPtr, + 'dimensions': dimensions, + 'value': value + }) + else: + self.TypedefAssignments[ClassKey].append((TargetName, IsPtr)) + else: + if ClassKey not in self.EmbeddedAssignments: + self.EmbeddedAssignments[ClassKey] = [] + # 如果有维度信息，使用字典存储 + if dimensions: + self.EmbeddedAssignments[ClassKey].append({ + 'name': TargetName, + 'dimensions': dimensions, + 'value': value + }) + else: + self.EmbeddedAssignments[ClassKey].append(TargetName) + + # 检查是否有 __set_default__ 初始化 + if AnnotationContainsName(NextNode.annotation, '__set_default__'): + self._handle_set_default_in_parse(ClassKey, TargetName, ClassName, NextNode) + j += 1 + continue + # 再处理旧格式：xxx: t.CTypedef = xxx 或 xxx: t.CPtr = xxx + # 注意：这种格式的 typedef 不应该被合并到结构体定义中， + # 而应该生成独立的 typedef 语句，所以不添加到 TypedefAssignments + elif NextNode.annotation and isinstance(NextNode.annotation, ast.Attribute): + if isinstance(NextNode.annotation.value, ast.Name): + annot_id: str = NextNode.annotation.value.id + annot_attr: str = NextNode.annotation.attr + if annot_id == 't' and NextNode.value: + if isinstance(NextNode.value, ast.Name): + ValueName: str = NextNode.value.id + if ValueName == ClassName: + # 检查是 CTypedef 还是 CPtr + if annot_attr == t.CPtr.__name__: + # t.CPtr 表示指针变量，加入 TypedefAssignments，设置 IsPtr=True + if ClassKey not in self.TypedefAssignments: + self.TypedefAssignments[ClassKey] = [] + self.TypedefAssignments[ClassKey].append((TargetName, True)) + j += 1 + continue + # 注意：t.CTypedef 不在这里处理，由 HandlesAnnAssign 生成独立的 typedef 语句 + # 如果不是特殊赋值，停止收集 + break + + i = j + continue + + i += 1 + + # 提取类定义（作为结构体） + for node in ast.walk(tree): + if isinstance(node, ast.ClassDef): + ClassName: str = node.name + # 先把类名加入 GeneratedTypes，这样 GetTypeName 可以正确识别 + self.GeneratedTypes.add(ClassName) + + # 检测是否是 CEnum + IsCenum: bool = False + for base in node.bases: + if isinstance(base, ast.Attribute): + if base.attr in ('CEnum', 'REnum'): + IsCenum = True + break + elif isinstance(base, ast.Name): + if base.id in ('CEnum', 'REnum'): + IsCenum = True + break + + # 检测是否有 @t.Object 装饰器或继承自 t.Object + IsCpythonObject: bool = False + if hasattr(node, 'decorator_list') and node.decorator_list: + for decorator in node.decorator_list: + if isinstance(decorator, ast.Attribute): + if (hasattr(decorator.value, 'id') and + IsTModule(decorator.value.id, self.SymbolTable) and + decorator.attr == 'Object'): + IsCpythonObject = True + break + elif isinstance(decorator, ast.Call): + if isinstance(decorator.func, ast.Attribute): + if (hasattr(decorator.func.value, 'id') and + IsTModule(decorator.func.value.id, self.SymbolTable) and + decorator.func.attr == 'Object'): + IsCpythonObject = True + break + elif isinstance(decorator.func, ast.Name): + if decorator.func.id == 'Object': + IsCpythonObject = True + break + elif isinstance(decorator, ast.Name): + if decorator.id == 'Object': + IsCpythonObject = True + break + if not IsCpythonObject and hasattr(node, 'bases') and node.bases: + for base in node.bases: + if isinstance(base, ast.Attribute): + if (hasattr(base.value, 'id') and + IsTModule(base.value.id, self.SymbolTable) and + base.attr == 'Object'): + IsCpythonObject = True + break + elif isinstance(base, ast.Name): + if base.id == 'Object': + IsCpythonObject = True + break + + if IsCenum: + # 注册为 enum 类型 + EnumNode: CTypeInfo = CTypeInfo() + EnumNode.Name = ClassName + EnumNode.IsEnum = True + EnumNode.set('file', '') + self.SymbolTable.insert(ClassName, EnumNode) + + # 注册 enum 成员（提取枚举值，与 SymbolExtractor 逻辑一致） + next_enum_value: int = 0 + for item in node.body: + MemberName: str | None = None + ItemValue: object | None = None + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + MemberName = item.target.id + ItemValue = item.value + elif isinstance(item, ast.Assign) and len(item.targets) == 1 and isinstance(item.targets[0], ast.Name): + MemberName = item.targets[0].id + ItemValue = item.value + if MemberName is None: + continue + MemberNode: CTypeInfo = CTypeInfo() + MemberNode.Name = MemberName + MemberNode.BaseType = t.CEnum(ClassName) + # 提取枚举值：优先用显式赋值，否则自动编号 + if ItemValue is not None and isinstance(ItemValue, ast.Constant) and isinstance(ItemValue.value, int): + MemberNode.value = ItemValue.value + next_enum_value = ItemValue.value + 1 + else: + MemberNode.value = next_enum_value + next_enum_value += 1 + MemberNode.EnumName = ClassName + MemberNode.library_name = None + MemberNode.IsEnumMember = True + self.SymbolTable.insert(MemberName, MemberNode) + self.SymbolTable.insert(f"{ClassName}.{MemberName}", MemberNode) + self.SymbolTable.insert(f"{ClassName}_{MemberName}", MemberNode) + + continue + + # 检测是否是 Anonymous + IsAnonymous: bool = False + + # 提取结构体成员信息 + members: dict[str, CTypeInfo] = {} + for item in node.body: + if isinstance(item, ast.AnnAssign) and isinstance(item.target, ast.Name): + VarName: str = item.target.id + try: + TypeInfo: CTypeInfo | None = self.TypeMergeHandler.GetCTypeInfo(item.annotation) + if TypeInfo is None: + TypeInfo = CTypeInfo() + TypeInfo.BaseType = t.CInt() + IsPtr: bool = TypeInfo.IsPtr + if not IsPtr: + if isinstance(item.annotation, ast.BinOp) and isinstance(item.annotation.op, ast.BitOr): + if AnnotationContainsName(item.annotation, 'CPtr'): + IsPtr = True + dims: list[int] = [] + if hasattr(item.annotation, 'slice') and isinstance(item.annotation.slice, ast.Constant): + if isinstance(item.annotation.value, ast.Name): + if IsTModule(item.annotation.value.id, self.SymbolTable): + if hasattr(item.annotation, 'slice'): + try: + dims.append(int(item.annotation.slice.value)) + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"解析数组维度失败: {_e}", "Exception") + MemberInfo: CTypeInfo = TypeInfo.Copy() + if dims: + MemberInfo.ArrayDims = [str(d) for d in dims] + members[VarName] = MemberInfo + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"解析结构体成员失败: {_e}", "Exception") + StructNode: CTypeInfo = CTypeInfo() + StructNode.Name = ClassName + StructNode.IsStruct = True + StructNode.Members = members or {} + StructNode.set('file', '') + StructNode.set('IsAnonymous', IsAnonymous) + StructNode.set('IsCpythonObject', IsCpythonObject) + self.SymbolTable.insert(ClassName, StructNode) + elif isinstance(node, ast.FunctionDef): + FuncName: str = node.name + if not self.SymbolTable.has(FuncName): + FuncNode: CTypeInfo = CTypeInfo() + FuncNode.Name = FuncName + FuncNode.IsFunction = True + FuncNode.set('file', '') + self.SymbolTable.insert(FuncName, FuncNode) + elif isinstance(node, ast.AnnAssign): + if isinstance(node.target, ast.Name): + VarName: str = node.target.id + # 检查是否是 t.CTypedef 格式，如果是则跳过（由 HandleClassDef 处理） + is_ctypedef: bool = False + if node.annotation: + if isinstance(node.annotation, ast.Attribute): + if isinstance(node.annotation.value, ast.Name): + if IsTModule(node.annotation.value.id, self.SymbolTable) and node.annotation.attr == 'CTypedef': + is_ctypedef = True + elif isinstance(node.annotation, ast.BinOp) and isinstance(node.annotation.op, ast.BitOr): + def _has_ctype(ann: ast.AST) -> bool: + if isinstance(ann, ast.Attribute): + return hasattr(ann.value, 'id') and IsTModule(ann.value.id) and ann.attr == 'CTypedef' + if isinstance(ann, ast.BinOp) and isinstance(ann.op, ast.BitOr): + return _has_ctype(ann.left) or _has_ctype(ann.right) + if isinstance(ann, ast.Subscript): + return _has_ctype(ann.value) + return False + if _has_ctype(node.annotation): + is_ctypedef = True + if not is_ctypedef: + # 如果符号已经存在且是 typedef，不要覆盖它 + existing = self.SymbolTable.lookup(VarName) + if existing and existing.IsTypedef and existing.OriginalType: + pass + else: + IsPtr: bool = False + if node.annotation: + try: + TypeInfo: CTypeInfo | None = self.TypeMergeHandler.GetCTypeInfo(node.annotation) + IsPtr = TypeInfo.IsPtr if TypeInfo else False + except Exception as _e: + if _ConfigMode == "strict": + raise + _vlog().warning(f"获取类型信息失败(ParsePythonFile): {_e}", "Exception") + var_node: CTypeInfo = CTypeInfo() + var_node.Name = VarName + var_node.IsVariable = True + var_node.set('IsPtr', IsPtr) + var_node.set('file', '') + self.SymbolTable.insert(VarName, var_node) + except Exception as e: + _vlog().warning(f'解析 Python 文件失败 {FilePath}: {e}') + + # ParseCFile 已移除 - .h/.c 文件的旧正则解析路径不再使用 + # 类型解析现在直接通过 CTypeInfo 对象进行，不经过中间字符串格式 diff --git a/lib/core/Translator/__init__.py b/lib/core/Translator/__init__.py new file mode 100644 index 0000000..84114dc --- /dev/null +++ b/lib/core/Translator/__init__.py @@ -0,0 +1,11 @@ +"""Translator 包 - 代码转换器模块""" +from lib.core.Translator.BaseTranslator import BaseTranslatorMixin, _StrictLog +from lib.core.Translator.AnnotationLoader import AnnotationLoaderMixin +from lib.core.Translator.PythonParser import PythonParserMixin +from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +from lib.core.Translator.ConstEval import ConstEvalMixin + +__all__ = [ + 'BaseTranslatorMixin', 'AnnotationLoaderMixin', 'PythonParserMixin', + 'LlvmGeneratorMixin', 'ConstEvalMixin', '_StrictLog' +] diff --git a/lib/core/TypeAnnotationResolver.py b/lib/core/TypeAnnotationResolver.py new file mode 100644 index 0000000..9ebcad1 --- /dev/null +++ b/lib/core/TypeAnnotationResolver.py @@ -0,0 +1,736 @@ +from __future__ import annotations +import ast +import warnings +from typing import TYPE_CHECKING + +from lib.core.Handles.HandlesBase import CTypeInfo, BuiltinTypeMap, CTypeHelper +from lib.core.ConstEvaluator import ConstEvaluator +from lib.core.SymbolUtils import IsTModule, IsTModulePath, IsListAnnotation +from lib.includes import t +from lib.constants.config import mode as _config_mode + +if TYPE_CHECKING: + from lib.core.SymbolTable import SymbolTable + + +class TypeAnnotationResolver: + """从 AST 节点解析类型注解，独立于 CTypeInfo 数据模型。""" + + @staticmethod + def from_type_name(TypeName: str) -> CTypeInfo: + """从简单类型名构造 CTypeInfo（不经过 FromStr 字符串解析）""" + # Handle pointer types like 'CUInt32T *', 'void *', etc. + ptr_count: int = 0 + base_name: str = TypeName + while base_name.endswith(' *') or base_name.endswith('*'): + ptr_count += 1 + base_name = base_name.rstrip(' *').rstrip() + + entry: tuple[type, int] | None = BuiltinTypeMap.Get(base_name) + if entry: + TypeClass: type + base_ptr: int + TypeClass, base_ptr = entry + info: CTypeInfo = CTypeInfo() + info.BaseType = TypeClass + info.PtrCount = base_ptr + ptr_count + return info + # Handle LLVM primitive type names (e.g., 'i64', 'double' from generic type inference) + llvm_map: dict = CTypeInfo._get_llvm_primitive_map() + llvm_entry: tuple[type, int] | None = llvm_map.get(base_name) + if llvm_entry: + TypeClass: type + base_ptr: int + TypeClass, base_ptr = llvm_entry + info: CTypeInfo = CTypeInfo() + info.BaseType = TypeClass + info.PtrCount = base_ptr + ptr_count + return info + # Fallback with pointer count + info: CTypeInfo = CTypeInfo.CreateFromTypeName(TypeName) + if ptr_count > 0: + info.PtrCount = ptr_count + return info + + @staticmethod + def try_eval_const_expr(node: ast.AST, SymbolTable: SymbolTable) -> int | float | str | None: + return ConstEvaluator.eval_with_symtab(node, SymbolTable) + + @staticmethod + def from_node(Node: ast.AST, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """从 AST 节点解析 CTypeInfo + + Args: + Node: AST 节点（如 ast.Name, ast.Attribute 等） + SymbolTable: 符号表字典 + """ + if isinstance(Node, ast.Constant): + return TypeAnnotationResolver._from_node_constant(Node, SymbolTable) + if isinstance(Node, ast.Name): + return TypeAnnotationResolver._from_node_name(Node, SymbolTable) + if isinstance(Node, ast.Call): + return TypeAnnotationResolver._from_node_call(Node, SymbolTable) + if isinstance(Node, ast.BinOp) and isinstance(Node.op, ast.BitOr): + return TypeAnnotationResolver._from_node_binop(Node, SymbolTable) + if isinstance(Node, ast.Subscript): + return TypeAnnotationResolver._from_node_subscript(Node, SymbolTable) + if isinstance(Node, ast.Attribute): + return TypeAnnotationResolver._from_node_attribute(Node, SymbolTable) + return CTypeInfo() + + @staticmethod + def _from_node_constant(Node: ast.AST, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.Constant 节点""" + TypeName: str = Node.value + return TypeAnnotationResolver.from_type_name(TypeName) + + @staticmethod + def _from_node_name(Node: ast.Name, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.Name 节点""" + TypeName: str = Node.id + TypeObj: type | None = CTypeHelper.GetTModuleCType(TypeName) + if TypeObj is None: + TypeObj = getattr(t, TypeName, None) + if isinstance(TypeObj, type) and issubclass(TypeObj, t.CType) and TypeObj is not t.CType: + if TypeObj == t.CPtr: + Result: CTypeInfo = CTypeInfo() + Result.PtrCount = 1 + Result.BaseType = t.CVoid() + return Result + if TypeObj == t.State: + Result: CTypeInfo = CTypeInfo() + Result.IsState = True + Result.BaseType = t.CVoid() + return Result + Inst: t.CType = TypeObj() + Result: CTypeInfo = CTypeInfo() + Result.Name = type(Inst).__name__ + # 存储类修饰符（CExtern, CExport, CInline, CThreadLocal 等）： + # position 含 STORAGE_CLASS 且不含 BASE → 纯修饰符，无实际类型 + if t.CType.STORAGE_CLASS in TypeObj.position and t.CType.BASE not in TypeObj.position: + Result.BaseType = t.CVoid() + Result.Storage = Inst + return Result + # 类型限定符（CConst, CVolatile 等） + if t.CType.TYPE_QUALIFIER in TypeObj.position and t.CType.BASE not in TypeObj.position: + Result.BaseType = t.CVoid() + if TypeObj == t.CConst: + Result.DataConst = True + elif TypeObj == t.CVolatile: + Result.DataVolatile = True + return Result + Result.BaseType = TypeObj + if hasattr(Inst, 'IsSigned'): + Result.IsSigned = Inst.IsSigned + return Result + # 检查 BuiltinTypeMap（覆盖 VOIDPTR, BYTEPTR 等指针别名） + builtin_entry: tuple[type, int] | None = BuiltinTypeMap.Get(TypeName) + if builtin_entry: + TypeClass: type + base_ptr: int + TypeClass, base_ptr = builtin_entry + Result: CTypeInfo = CTypeInfo() + Result.BaseType = TypeClass + Result.PtrCount = base_ptr + return Result + if TypeName in SymbolTable: + Entry: CTypeInfo | dict = SymbolTable[TypeName] + if Entry: + if isinstance(Entry, CTypeInfo): + if Entry.IsTypedef: + if Entry.BaseType and (not isinstance(Entry.BaseType, (t._CTypedef,)) or Entry.PtrCount > 0): + Result = Entry.Copy() + Result.IsTypedef = True + Result.Name = TypeName + return Result + if Entry.OriginalType: + if isinstance(Entry.OriginalType, CTypeInfo) and Entry.OriginalType.IsFuncPtr: + Result = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = Entry.OriginalType.FuncPtrReturn or CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = list(Entry.OriginalType.FuncPtrParams) if Entry.OriginalType.FuncPtrParams else [] + Result.IsTypedef = True + Result.Name = TypeName + return Result + elif isinstance(Entry.OriginalType, CTypeInfo): + Resolved = Entry.OriginalType.Copy() + elif isinstance(Entry.OriginalType, str) and Entry.OriginalType == 'Callable': + Result = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = [] + Result.IsTypedef = True + Result.Name = TypeName + return Result + elif isinstance(Entry.OriginalType, str): + Resolved = TypeAnnotationResolver.from_type_name(Entry.OriginalType) + else: + Resolved = Entry.OriginalType + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + TypeEntry = BuiltinTypeMap.Get(TypeName) + if TypeEntry: + Result = CTypeInfo() + Result.BaseType = TypeEntry[0]() + Result.PtrCount = TypeEntry[1] + Result.IsTypedef = True + Result.Name = TypeName + return Result + return Entry + if Entry.IsEnum: + Result = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsEnum = True + Result.Name = TypeName + return Result + if Entry.IsExceptionClass: + Result = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsExceptionClass = True + Result.Name = TypeName + return Result + if Entry.BaseType is None and (Entry.IsStruct or Entry.Name): + Entry.BaseType = t.CStruct(name=TypeName) + Entry.IsStruct = True + return Entry + elif isinstance(Entry, dict): + if Entry.get('type') == 'typedef': + OriginalType = Entry.get('OriginalType', '') + if OriginalType: + if OriginalType == 'Callable': + Result = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = [] + Result.IsTypedef = True + Result.Name = TypeName + return Result + Resolved = TypeAnnotationResolver.from_type_name(OriginalType) + Resolved.IsTypedef = True + Resolved.Name = TypeName + return Resolved + TypeEntry = BuiltinTypeMap.Get(TypeName) + if TypeEntry: + Result = CTypeInfo() + Result.BaseType = TypeEntry[0]() + Result.PtrCount = TypeEntry[1] + Result.IsTypedef = True + Result.Name = TypeName + return Result + Result = CTypeInfo() + Result.BaseType = t._CTypedef(TypeName) + Result.IsTypedef = True + Result.Name = TypeName + return Result + # 检查 t 类型别名（from t import CInt as myint） + if hasattr(SymbolTable, '_t_type_symbols'): + t_cls = SymbolTable._t_type_symbols.get(TypeName) + if t_cls is not None: + Result = CTypeInfo() + Result.BaseType = t_cls + return Result + # 特殊处理 str 类型，它应该是 char*（指针） + if TypeName in ('str', 'bytes'): + Result = CTypeInfo() + Result.BaseType = t.CChar() + Result.PtrCount = 1 + return Result + return TypeAnnotationResolver.from_type_name(TypeName) + + @staticmethod + def _from_node_call(Node: ast.Call, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.Call 节点""" + if isinstance(Node.func, ast.Name) and Node.func.id == 'callable': + Result: CTypeInfo = CTypeInfo() + Result.IsFuncPtr = True + Result.FuncPtrReturn = CTypeInfo.VoidTypeInfo() + Result.FuncPtrParams = [] + try: + if len(Node.args) > 0: + RetInfo: CTypeInfo = TypeAnnotationResolver.from_node(Node.args[0], SymbolTable) + if RetInfo: + Result.FuncPtrReturn = RetInfo + kw: ast.keyword + for kw in Node.keywords: + ParamInfo: CTypeInfo = TypeAnnotationResolver.from_node(kw.value, SymbolTable) + if ParamInfo: + Result.FuncPtrParams.append((kw.arg or '', ParamInfo)) + except Exception as _e: + if _config_mode == "strict": + warnings.warn(f"异常被忽略: {_e}") + return Result + if isinstance(Node.func, ast.Attribute): + if isinstance(Node.func.value, ast.Name) and IsTModule(Node.func.value.id, SymbolTable): + if Node.func.attr == 'Bit' and len(Node.args) > 0: + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsBitField = True + if isinstance(Node.args[0], ast.Constant) and isinstance(Node.args[0].value, int): + Result.BitWidth = Node.args[0].value + else: + Result.BitWidth = 1 + return Result + return CTypeInfo() + + @staticmethod + def _from_node_binop(Node: ast.BinOp, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.BinOp (BitOr) 节点""" + LeftInfo: CTypeInfo = TypeAnnotationResolver.from_node(Node.left, SymbolTable) + RightInfo: CTypeInfo = TypeAnnotationResolver.from_node(Node.right, SymbolTable) + + if LeftInfo.IsState or RightInfo.IsState: + Result: CTypeInfo = CTypeInfo() + Result.IsState = True + NonStateInfo: CTypeInfo = LeftInfo if not LeftInfo.IsState else RightInfo + StateInfo: CTypeInfo = LeftInfo if LeftInfo.IsState else RightInfo + if NonStateInfo.BaseType and (not isinstance(NonStateInfo.BaseType, (t._CTypedef,)) or NonStateInfo.PtrCount > 0): + Result.BaseType = NonStateInfo.BaseType + elif NonStateInfo.IsTypedef and NonStateInfo.Name: + if NonStateInfo.BaseType and (not isinstance(NonStateInfo.BaseType, (t._CTypedef,)) or NonStateInfo.PtrCount > 0): + Result.BaseType = NonStateInfo.BaseType + Result.PtrCount = NonStateInfo.PtrCount + Result.IsTypedef = True + Result.Name = NonStateInfo.Name + if NonStateInfo.OriginalType: + Result.OriginalType = NonStateInfo.OriginalType + else: + Result.BaseType = t.CStruct(name=NonStateInfo.Name) + Result.IsStruct = True + Result.IsTypedef = True + Result.Name = NonStateInfo.Name + if NonStateInfo.OriginalType: + Result.OriginalType = NonStateInfo.OriginalType + else: + Result.BaseType = t.CVoid() + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if not Result.Storage: + Result.Storage = t.CExport() + if LeftInfo.Storage and not isinstance(LeftInfo.Storage, t.CExport): + Result.Storage = LeftInfo.Storage + elif RightInfo.Storage and not isinstance(RightInfo.Storage, t.CExport): + Result.Storage = RightInfo.Storage + if LeftInfo.DataConst or RightInfo.DataConst: + Result.DataConst = True + if LeftInfo.VarConst or RightInfo.VarConst: + Result.VarConst = True + return Result + + if LeftInfo.IsPtr or RightInfo.IsPtr: + Result = CTypeInfo() + # 当双方都是指针时（如 (AST | t.CPtr) | t.CPtr），PtrCount 应相加 + # 而非取 max，否则 AST** 会被折叠为 AST*，导致对象切片问题 + if LeftInfo.IsPtr and RightInfo.IsPtr: + Result.PtrCount = LeftInfo.PtrCount + RightInfo.PtrCount + # BaseType 优先取非 void 侧（如 AST），而非 t.CPtr 的 void + if LeftInfo.BaseType and not isinstance(LeftInfo.BaseType, t.CVoid): + Result.BaseType = LeftInfo.BaseType + elif RightInfo.BaseType and not isinstance(RightInfo.BaseType, t.CVoid): + Result.BaseType = RightInfo.BaseType + else: + Result.BaseType = t.CUnsignedChar() + else: + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + NonPtrInfo = LeftInfo if not LeftInfo.IsPtr else RightInfo + PtrInfo = RightInfo if RightInfo.IsPtr else LeftInfo + if NonPtrInfo.BaseType and (not isinstance(NonPtrInfo.BaseType, (t._CTypedef,)) or NonPtrInfo.PtrCount > 0): + Result.BaseType = NonPtrInfo.BaseType + elif NonPtrInfo.IsTypedef and NonPtrInfo.Name: + if NonPtrInfo.BaseType and (not isinstance(NonPtrInfo.BaseType, (t._CTypedef,)) or NonPtrInfo.PtrCount > 0): + Result.BaseType = NonPtrInfo.BaseType + Result.IsTypedef = True + Result.Name = NonPtrInfo.Name + if NonPtrInfo.OriginalType: + Result.OriginalType = NonPtrInfo.OriginalType + else: + Result.BaseType = t.CStruct(name=NonPtrInfo.Name) + Result.IsStruct = True + elif PtrInfo.BaseType and (not isinstance(PtrInfo.BaseType, (t._CTypedef,)) or PtrInfo.PtrCount > 0): + Result.BaseType = PtrInfo.BaseType + else: + Result.BaseType = t.CUnsignedChar() + if LeftInfo.DataConst or RightInfo.DataConst: + Result.DataConst = True + if LeftInfo.VarConst or RightInfo.VarConst: + Result.VarConst = True + if LeftInfo.DataVolatile or RightInfo.DataVolatile: + Result.DataVolatile = True + if LeftInfo.VarVolatile or RightInfo.VarVolatile: + Result.VarVolatile = True + if LeftInfo.Storage: + Result.Storage = LeftInfo.Storage + elif RightInfo.Storage: + Result.Storage = RightInfo.Storage + return Result + + if LeftInfo.DataConst or RightInfo.DataConst: + Result = CTypeInfo() + BaseTypeSide = LeftInfo if not LeftInfo.DataConst else RightInfo + QualSide = LeftInfo if LeftInfo.DataConst else RightInfo + Result.BaseType = BaseTypeSide.BaseType if BaseTypeSide.BaseType and (not isinstance(BaseTypeSide.BaseType, t.CVoid) or BaseTypeSide.PtrCount > 0) else QualSide.BaseType + if not Result.BaseType: + Result.BaseType = t.CInt() + Result.DataConst = True + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if LeftInfo.VarConst or RightInfo.VarConst: + Result.VarConst = True + if LeftInfo.DataVolatile or RightInfo.DataVolatile: + Result.DataVolatile = True + if LeftInfo.Storage: + Result.Storage = LeftInfo.Storage + elif RightInfo.Storage: + Result.Storage = RightInfo.Storage + return Result + + if LeftInfo.DataVolatile or RightInfo.DataVolatile: + Result = CTypeInfo() + BaseTypeSide = LeftInfo if not LeftInfo.DataVolatile else RightInfo + QualSide = LeftInfo if LeftInfo.DataVolatile else RightInfo + Result.BaseType = BaseTypeSide.BaseType if BaseTypeSide.BaseType and (not isinstance(BaseTypeSide.BaseType, t.CVoid) or BaseTypeSide.PtrCount > 0) else QualSide.BaseType + if not Result.BaseType: + Result.BaseType = t.CInt() + Result.DataVolatile = True + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if LeftInfo.VarConst or RightInfo.VarConst: + Result.VarConst = True + if LeftInfo.Storage: + Result.Storage = LeftInfo.Storage + elif RightInfo.Storage: + Result.Storage = RightInfo.Storage + return Result + + if LeftInfo.Storage or RightInfo.Storage: + Result = CTypeInfo() + BaseTypeSide = LeftInfo if not LeftInfo.Storage else RightInfo + StorageSide = LeftInfo if LeftInfo.Storage else RightInfo + # CExtern | CExport 等价于 t.State：两边都有 Storage 时设置 IsState + if BaseTypeSide.IsState or StorageSide.IsState or (LeftInfo.Storage and RightInfo.Storage): + Result.IsState = True + Result.BaseType = t.CVoid() + Result.Storage = LeftInfo.Storage if LeftInfo.Storage else RightInfo.Storage + return Result + Result.BaseType = BaseTypeSide.BaseType if BaseTypeSide.BaseType and (not isinstance(BaseTypeSide.BaseType, t.CVoid) or BaseTypeSide.PtrCount > 0) else StorageSide.BaseType + if not Result.BaseType: + Result.BaseType = t.CInt() + Result.Storage = LeftInfo.Storage if LeftInfo.Storage else RightInfo.Storage + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if LeftInfo.DataConst or RightInfo.DataConst: + Result.DataConst = True + if LeftInfo.VarConst or RightInfo.VarConst: + Result.VarConst = True + return Result + + # 检查是否有位域类型 (t.Bit) + if LeftInfo.IsBitField: + Result = CTypeInfo() + Result.BaseType = LeftInfo.BaseType if LeftInfo.BaseType else t.CInt() + Result.IsBitField = True + Result.BitWidth = LeftInfo.BitWidth + return Result + if RightInfo.IsBitField: + Result = CTypeInfo() + Result.BaseType = RightInfo.BaseType if RightInfo.BaseType else t.CInt() + Result.IsBitField = True + Result.BitWidth = RightInfo.BitWidth + return Result + + # 处理字节序类型 (t.BigEndian, t.LittleEndian) + # 合并两侧：保留非字节序侧的实际类型（如 CUInt64T），取字节序侧的 ByteOrder + if LeftInfo.ByteOrder or RightInfo.ByteOrder: + Result = CTypeInfo() + ByteOrderSide: CTypeInfo = LeftInfo if LeftInfo.ByteOrder else RightInfo + OtherSide: CTypeInfo = RightInfo if LeftInfo.ByteOrder else LeftInfo + if OtherSide.BaseType and (not isinstance(OtherSide.BaseType, (t._CTypedef,)) or OtherSide.PtrCount > 0): + Result.BaseType = OtherSide.BaseType + elif ByteOrderSide.BaseType and (not isinstance(ByteOrderSide.BaseType, (t._CTypedef,)) or ByteOrderSide.PtrCount > 0): + Result.BaseType = ByteOrderSide.BaseType + else: + Result.BaseType = t.CInt() + Result.ByteOrder = LeftInfo.ByteOrder or RightInfo.ByteOrder + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + return Result + + if LeftInfo.IsFuncPtr or RightInfo.IsFuncPtr: + Result = CTypeInfo() + FuncPtrSide = LeftInfo if LeftInfo.IsFuncPtr else RightInfo + Result.IsFuncPtr = True + Result.FuncPtrParams = list(FuncPtrSide.FuncPtrParams) + Result.FuncPtrReturn = FuncPtrSide.FuncPtrReturn + if LeftInfo.Storage: + Result.Storage = LeftInfo.Storage + elif RightInfo.Storage: + Result.Storage = RightInfo.Storage + return Result + + if LeftInfo.IsTypedef and RightInfo.BaseType: + if LeftInfo.BaseType and (not isinstance(LeftInfo.BaseType, (t._CTypedef,)) or LeftInfo.PtrCount > 0): + Result = LeftInfo.Copy() + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if RightInfo.Storage: + Result.Storage = RightInfo.Storage + return Result + return RightInfo + if RightInfo.IsTypedef and LeftInfo.BaseType: + if RightInfo.BaseType and (not isinstance(RightInfo.BaseType, (t._CTypedef,)) or RightInfo.PtrCount > 0): + Result = RightInfo.Copy() + Result.PtrCount = max(LeftInfo.PtrCount, RightInfo.PtrCount) + if LeftInfo.Storage: + Result.Storage = LeftInfo.Storage + return Result + return LeftInfo + + if LeftInfo.BaseType: + return LeftInfo + elif RightInfo.BaseType: + return RightInfo + return CTypeInfo() + + @staticmethod + def _from_node_subscript(Node: ast.Subscript, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.Subscript 节点""" + base: ast.AST = Node.value + if isinstance(base, ast.Attribute): + if isinstance(base.value, ast.Name) and IsTModule(base.value.id, SymbolTable) and base.attr == 'Bit': + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsBitField = True + if isinstance(Node.slice, ast.Constant) and isinstance(Node.slice.value, int): + Result.BitWidth = Node.slice.value + else: + Result.BitWidth = 1 + return Result + if IsListAnnotation(Node): + slice_node: ast.AST = Node.slice + elts: list[ast.AST] = [] + if isinstance(slice_node, ast.Tuple): + elts = slice_node.elts + elif isinstance(slice_node, (ast.Attribute, ast.Name, ast.Subscript)): + elts = [slice_node] + if elts: + ElemInfo: CTypeInfo = TypeAnnotationResolver.from_node(elts[0], SymbolTable) + if ElemInfo and ElemInfo.BaseType: + Result: CTypeInfo = CTypeInfo() + Result.BaseType = ElemInfo.BaseType + Result.PtrCount = ElemInfo.PtrCount + Result.ArrayDims = list(ElemInfo.ArrayDims) + if len(elts) >= 2: + count_val: int | None = TypeAnnotationResolver.try_eval_const_expr(elts[1], SymbolTable) + if count_val is not None and isinstance(count_val, int) and count_val > 0: + Result.ArrayDims.insert(0, str(count_val)) + else: + Result.PtrCount += 1 + if ElemInfo.IsPtr and not ElemInfo.ArrayDims: + Result.PtrCount = max(Result.PtrCount, 1) + return Result + if isinstance(base, ast.Attribute): + parts: list[str] = [] + current: ast.AST = base + while isinstance(current, ast.Attribute): + parts.insert(0, current.attr) + current = current.value + if isinstance(current, ast.Name): + parts.insert(0, current.id) + if parts and IsTModule(parts[0], SymbolTable) and parts[-1] == t.CPtr.__name__: + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CVoid() + Result.PtrCount = 1 + slice_node: ast.AST = Node.slice + if isinstance(slice_node, ast.Subscript): + InnerInfo: CTypeInfo = TypeAnnotationResolver.from_node(slice_node, SymbolTable) + if InnerInfo: + if InnerInfo.PtrCount > 0: + Result.PtrCount += InnerInfo.PtrCount + if InnerInfo.BaseType and not isinstance(InnerInfo.BaseType, t.CVoid): + Result.BaseType = InnerInfo.BaseType + elif isinstance(slice_node, ast.Attribute): + SliceInfo: CTypeInfo = TypeAnnotationResolver.from_node(slice_node, SymbolTable) + if SliceInfo: + if SliceInfo.IsPtr: + Result.PtrCount += SliceInfo.PtrCount + elif SliceInfo.BaseType and not isinstance(SliceInfo.BaseType, t.CVoid): + Result.BaseType = SliceInfo.BaseType + elif isinstance(slice_node, ast.Name): + SliceType: type | None = getattr(t, slice_node.id, None) + if SliceType == t.CPtr: + Result.PtrCount += 1 + else: + SliceInfo = TypeAnnotationResolver.from_node(slice_node, SymbolTable) + if SliceInfo and SliceInfo.BaseType and not isinstance(SliceInfo.BaseType, t.CVoid): + Result.BaseType = SliceInfo.BaseType + return Result + if parts and IsTModule(parts[0], SymbolTable) and parts[-1] == 'Callable': + Result = CTypeInfo() + Result.IsFuncPtr = True + slice_node = Node.slice + if isinstance(slice_node, ast.Tuple) and len(slice_node.elts) == 2: + params_list = slice_node.elts[0] + return_node = slice_node.elts[1] + ParamTypes = [] + if isinstance(params_list, ast.List): + for elt in params_list.elts: + ParamTypeInfo = TypeAnnotationResolver.from_node(elt, SymbolTable) + if ParamTypeInfo: + ParamTypes.append(('', ParamTypeInfo)) + if not ParamTypes: + ParamTypes.append(('', TypeAnnotationResolver.from_type_name('void'))) + Result.FuncPtrParams = ParamTypes + ReturnTypeInfo = TypeAnnotationResolver.from_node(return_node, SymbolTable) + Result.FuncPtrReturn = ReturnTypeInfo if ReturnTypeInfo else CTypeInfo.VoidTypeInfo() + return Result + return CTypeInfo() + + @staticmethod + def _from_node_attribute(Node: ast.Attribute, SymbolTable: dict | SymbolTable) -> CTypeInfo: + """处理 ast.Attribute 节点""" + ModuleParts: list[str] = [] + Current: ast.AST = Node + while isinstance(Current, ast.Attribute): + ModuleParts.insert(0, Current.attr) + Current = Current.value + + if isinstance(Current, ast.Name): + ModuleParts.insert(0, Current.id) + + TypeName: str + ModulePath: str | None + if len(ModuleParts) >= 2: + TypeName = ModuleParts[-1] + ModulePath = '.'.join(ModuleParts[:-1]) + elif len(ModuleParts) == 1: + TypeName = ModuleParts[0] + ModulePath = None + else: + return CTypeInfo() + + # 解析 import 别名: 如 import vpsdk.window as window -> window -> vpsdk.window + if ModulePath: + resolved: bool = False + import_aliases: dict = SymbolTable.import_aliases + if ModulePath in import_aliases: + ModulePath = import_aliases[ModulePath] + resolved = True + if not resolved and ModulePath in SymbolTable: + entry: CTypeInfo = SymbolTable[ModulePath] + if isinstance(entry, CTypeInfo) and entry.IsModuleAlias: + resolved_name = entry.ResolvedModule + if resolved_name: + ModulePath = resolved_name + + if ModulePath and IsTModulePath(ModulePath): + TypeClass: type | None = CTypeHelper.GetTModuleCType(TypeName) + if TypeClass is None: + TypeClass = getattr(t, TypeName, None) + if TypeClass is not None and TypeClass.HasPosition(t.CType.POINTER): + Info: CTypeInfo = CTypeInfo() + Info.PtrCount = 1 + if TypeClass == t.CArrayPtr: + Info.IsArrayPtr = True + Info.BaseType = t.CVoid + return Info + if TypeClass is not None and TypeClass.IsStorageClass(): + Info: CTypeInfo = CTypeInfo() + Info.Storage = TypeClass() + if TypeClass == t.State: + Info.IsState = True + Info.BaseType = t.CVoid() + return Info + if TypeClass is not None and TypeClass.IsTypeQualifier(): + Info: CTypeInfo = CTypeInfo() + if TypeClass == t.CConst: + Info.DataConst = True + elif TypeClass == t.CVolatile: + Info.DataVolatile = True + return Info + if TypeClass is not None and TypeClass in (t.BigEndian, t.LittleEndian): + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CInt() + Result.ByteOrder = 'big' if TypeClass == t.BigEndian else 'little' + return Result + if (TypeClass is not None + and isinstance(TypeClass, type) + and issubclass(TypeClass, t.CType) + and TypeClass is not t.CType + and TypeClass.HasPosition(t.CType.BASE)): + if TypeClass == t.State: + Result: CTypeInfo = CTypeInfo() + Result.IsState = True + Result.BaseType = t.CVoid() + return Result + Inst: t.CType = TypeClass() + Result: CTypeInfo = CTypeInfo() + Result.BaseType = TypeClass + Result.Name = type(Inst).__name__ + if hasattr(Inst, 'IsSigned'): + Result.IsSigned = Inst.IsSigned + return Result + CNAME: str | None = CTypeHelper.GetCName(TypeName) + if CNAME: + return TypeAnnotationResolver.from_type_name(CNAME) + + # 处理字节序类型 + if TypeName == 'BigEndian': + Result = CTypeInfo() + Result.BaseType = t.CInt() + Result.ByteOrder = 'big' + return Result + if TypeName == 'LittleEndian': + Result = CTypeInfo() + Result.BaseType = t.CInt() + Result.ByteOrder = 'little' + return Result + + FullName: str = f"{ModulePath}.{TypeName}" if ModulePath else TypeName + + if TypeName in SymbolTable: + Entry: CTypeInfo = SymbolTable[TypeName] + if Entry and Entry.IsTypedef: + Resolved: CTypeInfo = Entry.Copy() + if Resolved.BaseType is None: + Resolved.BaseType = t._CTypedef(TypeName) + return Resolved + if Entry and Entry.IsEnum: + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsEnum = True + Result.Name = TypeName + return Result + if Entry and Entry.IsExceptionClass: + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t.CInt() + Result.IsExceptionClass = True + Result.Name = TypeName + return Result + if Entry: + Result: CTypeInfo = Entry.Copy() + if Result.BaseType is None and (Result.IsStruct or Result.IsCpythonObject or Result.Name): + Result.BaseType = t.CStruct(name=TypeName) + Result.IsStruct = True + return Result + return CTypeInfo() + + if FullName in SymbolTable: + Info: CTypeInfo = SymbolTable[FullName] + if Info and Info.IsTypedef: + OriginalType: str = Info.get('OriginalType', '') + if OriginalType and 'typedef' in OriginalType: + parts: list[str] = OriginalType.split() + if len(parts) >= 2: + BaseType_name: str = parts[1] + if not BaseType_name.startswith('C'): + BaseType_name = 'C' + BaseType_name + CNAME: str | None = CTypeHelper.GetCName(BaseType_name) + if CNAME: + return TypeAnnotationResolver.from_type_name(CNAME) + Result: CTypeInfo = CTypeInfo() + Result.BaseType = t._CTypedef(TypeName) + Result.IsTypedef = True + return Result + if Info: + Result: CTypeInfo = Info.Copy() + if Result.BaseType is None and (Result.IsStruct or Result.IsCpythonObject or Result.Name): + Result.BaseType = t.CStruct(name=TypeName) + Result.IsStruct = True + return Result + return CTypeInfo() + + return CTypeInfo() diff --git a/lib/core/TypeSpec.py b/lib/core/TypeSpec.py new file mode 100644 index 0000000..1b28bf5 --- /dev/null +++ b/lib/core/TypeSpec.py @@ -0,0 +1,332 @@ +"""TypeSpec, SymbolMeta, SymbolKind — CTypeInfo 的拆分类型 + +Phase 1 重构：将 CTypeInfo 的 40+ 平铺字段拆分为三个聚焦的类型： +- TypeSpec: 纯类型描述（BaseType + 指针 + 限定符 + 数组维度等） +- SymbolMeta: 符号表元数据（名称、行号、文件、类型种类、成员等） +- SymbolKind: 枚举类型种类（从 is_xxx bool 字段派生，供新代码使用） + +设计原则： +- CTypeInfo 内部组合 TypeSpec + SymbolMeta +- 旧 property 委托到新字段，外部代码零修改 +- 渐进式迁移：先加新字段，后删旧字段 +- 类型种类 bool 保持独立（因为 IsTypedef 和 IsStruct 可以同时为 True） +- SymbolKind 从 is_xxx bool 字段派生，供新代码使用 +""" +from __future__ import annotations + +import enum +from typing import TYPE_CHECKING, List, Dict, Tuple, Any, Optional + +if TYPE_CHECKING: + from lib.core.Handles.HandlesBase import CTypeInfo + +from lib.includes import t + + +class SymbolKind(enum.Enum): + """符号类型种类 — 从 is_xxx bool 字段派生的主类型 + + 注意：IsTypedef 与 IsStruct/IsEnum/IsUnion 正交（可以同时为 True）， + 因此 SymbolKind 只反映"主类型"，IsTypedef 需要单独查询。 + """ + UNKNOWN = 0 + VARIABLE = 1 + FUNCTION = 2 + STRUCT = 3 + UNION = 4 + ENUM = 5 + DEFINE = 6 + ENUM_MEMBER = 7 + MODULE = 8 + EXCEPTION = 9 + RENUM = 10 + FUNC_PTR = 11 + STATE = 12 + CPYTHON_OBJECT = 13 + + +# CTypeInfo 旧字段 → (目标, 新字段名) 的路由表 +# 目标: 'ts' = TypeSpec, 'sm' = SymbolMeta +FIELD_ROUTES = { + # --- TypeSpec 字段 --- + 'PtrCount': ('ts', 'ptr_count'), + 'VarConst': ('ts', 'var_const'), + 'VarVolatile': ('ts', 'var_volatile'), + 'DataConst': ('ts', 'data_const'), + 'DataVolatile': ('ts', 'data_volatile'), + 'ArrayDims': ('ts', 'array_dims'), + 'Storage': ('ts', 'storage'), + 'IsFuncPtr': ('ts', 'is_func_ptr'), + 'FuncPtrParams': ('ts', 'func_ptr_params'), + 'FuncParamNames': ('ts', 'func_param_names'), + 'FuncPtrReturn': ('ts', 'func_ptr_return'), + 'IsVariadic': ('ts', 'is_variadic'), + 'IsBitField': ('ts', 'is_bitfield'), + 'BitWidth': ('ts', 'bit_width'), + 'ByteOrder': ('ts', 'byte_order'), + 'IsArrayPtr': ('ts', 'is_array_ptr'), + 'ArrayPtr': ('ts', 'array_ptr'), + 'IsState': ('ts', 'is_state'), + 'OriginalType': ('ts', 'original_type'), + 'IsSigned': ('ts', 'is_signed'), + + # --- SymbolMeta 字段 --- + 'Name': ('sm', 'name'), + 'Lineno': ('sm', 'lineno'), + 'file': ('sm', 'file'), + 'ModuleName': ('sm', 'module_name'), + 'Members': ('sm', 'members'), + 'IsCpythonObject':('sm', 'is_cpython_object'), + 'IsPacked': ('sm', 'is_packed'), + 'IsAnonymous': ('sm', 'is_anonymous'), + 'IsInline': ('sm', 'is_inline'), + 'InlineBody': ('sm', 'inline_body'), + 'InlineParams': ('sm', 'inline_params'), + 'RenumVariants': ('sm', 'renum_variants'), + 'DeclaredType': ('sm', 'declared_type'), + 'CreturnTypes': ('sm', 'creturn_types'), + 'MetaList': ('sm', 'meta_list'), + 'IsModuleAlias': ('sm', 'is_module_alias'), + 'ResolvedModule': ('sm', 'resolved_module'), + 'DefineValue': ('sm', 'define_value'), + 'EnumName': ('sm', 'enum_name'), + 'value': ('sm', 'enum_value'), + + # --- 类型种类 bool（直接存储在 SymbolMeta 的 __slots__ 字段）--- + 'IsStruct': ('sm', 'is_struct'), + 'IsEnum': ('sm', 'is_enum'), + 'IsUnion': ('sm', 'is_union'), + 'IsTypedef': ('sm', 'is_typedef'), + 'IsVariable': ('sm', 'is_variable'), + 'IsFunction': ('sm', 'is_function'), + 'IsDefine': ('sm', 'is_define'), + 'IsEnumMember': ('sm', 'is_enum_member'), + 'IsRenum': ('sm', 'is_renum'), + 'IsExceptionClass': ('sm', 'is_exception_class'), + + # --- _extra（CTypeInfo 旧 _extra 字典，迁移到 SymbolMeta）--- + '_extra': ('sm', '_extra'), +} + + +class TypeSpec: + """纯类型描述 — 不包含符号表元数据 + + 职责：描述一个 C 类型的结构（基础类型、指针层数、限定符、数组维度等） + 不包含：名称、行号、文件路径、符号种类等元数据（这些归 SymbolMeta） + + 注意：当前阶段保持与 CTypeInfo 旧字段的兼容性， + PtrCount/VarConst/DataConst 等仍然使用扁平字段。 + 未来 Phase 2 会将指针改为嵌套 PointerLayer 表示。 + """ + + __slots__ = ( + '_base_type', 'ptr_count', + 'var_const', 'var_volatile', 'data_const', 'data_volatile', + 'array_dims', 'storage', + 'is_func_ptr', 'func_ptr_params', 'func_param_names', 'func_ptr_return', + 'is_variadic', + 'is_bitfield', 'bit_width', + 'byte_order', + 'is_array_ptr', 'array_ptr', + 'is_state', + 'original_type', + 'is_signed', + ) + + def __init__(self): + self._base_type: t.CType | tuple | None = None # t.CType 实例或 tuple + self.ptr_count: int = 0 + self.var_const: bool = False + self.var_volatile: bool = False + self.data_const: bool = False + self.data_volatile: bool = False + self.array_dims: List[str] = [] + self.storage: t.CType | None = None # t.CType (CStatic/CExtern/CExport 等) + self.is_func_ptr: bool = False + self.func_ptr_params: List[Tuple[str, CTypeInfo]] = [] # (参数名, CTypeInfo) + self.func_param_names: List[str] = [] + self.func_ptr_return: CTypeInfo | None = None # CTypeInfo + self.is_variadic: bool = False + self.is_bitfield: bool = False + self.bit_width: int = 0 + self.byte_order: str = "" # "big", "little", or "" + self.is_array_ptr: bool = False + self.array_ptr: CTypeInfo | None = None # CTypeInfo + self.is_state: bool = False + self.original_type: CTypeInfo | str | None = None # CTypeInfo | str | None (typedef 原始类型) + self.is_signed: Optional[bool] = None # None = 未设置, True/False = 有符号/无符号 + + @property + def base_type(self) -> t.CType | tuple | None: + return self._base_type + + @base_type.setter + def base_type(self, value: t.CType | tuple | None): + """设置 BaseType，与 CTypeInfo.BaseType setter 逻辑一致（不含 kind 副作用）""" + if value is None: + self._base_type = None + return + + if isinstance(value, (tuple, list)): + self._base_type = tuple(value) + return + + if isinstance(value, type) and issubclass(value, t.CType): + self._base_type = value() + return + + if isinstance(value, t.CType): + self._base_type = value + return + + raise TypeError( + f"base_type must be t.CType instance, type subclass, or tuple of t.CType, got {type(value)}" + ) + + def copy(self) -> TypeSpec: + """深拷贝""" + new: TypeSpec = TypeSpec() + new._base_type = self._base_type + new.ptr_count = self.ptr_count + new.var_const = self.var_const + new.var_volatile = self.var_volatile + new.data_const = self.data_const + new.data_volatile = self.data_volatile + new.array_dims = list(self.array_dims) + new.storage = self.storage + new.is_func_ptr = self.is_func_ptr + new.func_ptr_params = list(self.func_ptr_params) + new.func_param_names = list(self.func_param_names) + new.func_ptr_return = self.func_ptr_return + new.is_variadic = self.is_variadic + new.is_bitfield = self.is_bitfield + new.bit_width = self.bit_width + new.byte_order = self.byte_order + new.is_array_ptr = self.is_array_ptr + new.array_ptr = self.array_ptr + new.is_state = self.is_state + new.original_type = self.original_type + new.is_signed = self.is_signed + return new + + +class SymbolMeta: + """符号表条目的元数据 — 与类型描述无关 + + 职责：记录符号的名称、位置、种类、成员等元数据 + 不包含：类型结构（BaseType、指针、限定符等，这些归 TypeSpec） + + is_struct/is_enum/is_union/is_typedef/is_variable/is_function/is_define/ + is_enum_member/is_renum/is_exception_class: 独立 bool 字段，存储类型种类。 + 这些 flag 不是互斥的（IsTypedef 和 IsStruct 可以同时为 True）。 + SymbolKind 从这些 bool 字段派生，供新代码使用。 + """ + + __slots__ = ( + 'name', 'lineno', 'file', 'module_name', + 'members', + 'is_cpython_object', 'is_packed', + 'define_value', + 'enum_name', 'enum_value', + 'is_inline', 'inline_body', 'inline_params', + 'is_anonymous', + 'renum_variants', + 'declared_type', 'creturn_types', + 'meta_list', + 'is_module_alias', 'resolved_module', + 'is_struct', 'is_enum', 'is_union', 'is_typedef', + 'is_variable', 'is_function', 'is_define', + 'is_enum_member', 'is_renum', 'is_exception_class', + '_extra', + ) + + def __init__(self): + self.name: str = "" + self.lineno: int = 0 + self.file: str = "" + self.module_name: str = "" + self.members: Dict[str, CTypeInfo] = {} # 成员名 -> CTypeInfo + self.is_cpython_object: bool = False + self.is_packed: bool = False + self.define_value: int | str | float | bool | None = None + self.enum_name: str = "" # 枚举成员所属的枚举类名 + self.enum_value: int | None = None # 枚举成员的值 + self.is_inline: bool = False + self.inline_body: list | None = None + self.inline_params: list | None = None + self.is_anonymous: bool = False + self.renum_variants: list = [] + self.declared_type: str = "" + self.creturn_types: list = [] + self.meta_list: Any = None # FuncMeta + self.is_module_alias: bool = False + self.resolved_module: str | None = None + self.is_struct: bool = False + self.is_enum: bool = False + self.is_union: bool = False + self.is_typedef: bool = False + self.is_variable: bool = False + self.is_function: bool = False + self.is_define: bool = False + self.is_enum_member: bool = False + self.is_renum: bool = False + self.is_exception_class: bool = False + self._extra: dict = {} + + @property + def kind(self) -> SymbolKind: + """从 is_xxx bool 字段派生主类型种类""" + if self.is_struct or self.is_cpython_object: + return SymbolKind.STRUCT if not self.is_cpython_object else SymbolKind.CPYTHON_OBJECT + if self.is_enum: + return SymbolKind.RENUM if self.is_renum else SymbolKind.ENUM + if self.is_union: + return SymbolKind.UNION + if self.is_function: + return SymbolKind.FUNCTION + if self.is_variable: + return SymbolKind.VARIABLE + if self.is_define: + return SymbolKind.DEFINE + if self.is_enum_member: + return SymbolKind.ENUM_MEMBER + if self.is_exception_class: + return SymbolKind.EXCEPTION + return SymbolKind.UNKNOWN + + def copy(self) -> SymbolMeta: + """深拷贝""" + new: SymbolMeta = SymbolMeta() + new.name = self.name + new.lineno = self.lineno + new.file = self.file + new.module_name = self.module_name + new.members = dict(self.members) + new.is_cpython_object = self.is_cpython_object + new.is_packed = self.is_packed + new.define_value = self.define_value + new.enum_name = self.enum_name + new.enum_value = self.enum_value + new.is_inline = self.is_inline + new.inline_body = self.inline_body + new.inline_params = list(self.inline_params) if self.inline_params else None + new.is_anonymous = self.is_anonymous + new.renum_variants = list(self.renum_variants) + new.declared_type = self.declared_type + new.creturn_types = list(self.creturn_types) + new.meta_list = self.meta_list + new.is_module_alias = self.is_module_alias + new.resolved_module = self.resolved_module + new.is_struct = self.is_struct + new.is_enum = self.is_enum + new.is_union = self.is_union + new.is_typedef = self.is_typedef + new.is_variable = self.is_variable + new.is_function = self.is_function + new.is_define = self.is_define + new.is_enum_member = self.is_enum_member + new.is_renum = self.is_renum + new.is_exception_class = self.is_exception_class + new._extra = dict(self._extra) + return new diff --git a/lib/core/VLogger.py b/lib/core/VLogger.py new file mode 100644 index 0000000..043dc4c --- /dev/null +++ b/lib/core/VLogger.py @@ -0,0 +1,305 @@ +""" +TransPyC 彩色日志系统 +支持终端颜色输出和日志文件转储 +""" +from __future__ import annotations + +import sys +import os +from datetime import datetime +from typing import Any, Callable, Optional +from lib.constants.config import mode as _ConfigMode + +# ANSI 颜色代码 +class Colors: + RESET = '\033[0m' + BOLD = '\033[1m' + DIM = '\033[2m' + UNDERLINE = '\033[4m' + + # 前景色 + BLACK = '\033[30m' + RED = '\033[31m' + GREEN = '\033[32m' + YELLOW = '\033[33m' + BLUE = '\033[34m' + MAGENTA = '\033[35m' + CYAN = '\033[36m' + WHITE = '\033[37m' + + # 高亮前景色 + BRIGHT_RED = '\033[91m' + BRIGHT_GREEN = '\033[92m' + BRIGHT_YELLOW = '\033[93m' + BRIGHT_BLUE = '\033[94m' + BRIGHT_MAGENTA = '\033[95m' + BRIGHT_CYAN = '\033[96m' + + # 背景色 + BG_RED = '\033[41m' + BG_GREEN = '\033[42m' + BG_YELLOW = '\033[43m' + BG_BLUE = '\033[44m' + BG_MAGENTA = '\033[45m' + BG_CYAN = '\033[46m' + + # 渐变颜色（用于详细日志） + GRAY = '\033[90m' + +# 检查是否支持颜色 +def supports_color() -> bool: + if not hasattr(sys.stdout, 'isatty'): + return False + if not sys.stdout.isatty(): + return False + if os.environ.get('NO_COLOR'): + return False + if os.environ.get('FORCE_COLOR'): + return True + return True + +# 全局颜色支持标志 +USE_COLORS = supports_color() + +# 日志级别 +class LogLevel: + DEBUG = 0 + INFO = 1 + WARNING = 2 + ERROR = 3 + CRITICAL = 4 + +class Logger: + def __init__(self, name: str = "TransPyC", log_file: Optional[str] = None, + level: int = LogLevel.INFO, use_colors: bool = True): + self.name = name + self.level = level + self.use_colors = use_colors and USE_COLORS + self.log_file = log_file + self.file_handler = None + + if log_file: + os.makedirs(os.path.dirname(os.path.abspath(log_file)), exist_ok=True) + self.file_handler = open(log_file, 'w', encoding='utf-8') + + # 进度跟踪 + self.current_phase = "" + self.total_items = 0 + self.current_item = 0 + + def _colorize(self, text: str, color: str) -> str: + if self.use_colors: + return f"{color}{text}{Colors.RESET}" + return text + + def _format_time(self) -> str: + return datetime.now().strftime("%H:%M:%S") + + def _write(self, message: str, level: str = ""): + # 输出到控制台 + print(message, file=sys.stdout) + sys.stdout.flush() + + # 输出到文件 + if self.file_handler: + # 文件输出不带颜色 + clean_msg = message + for color in vars(Colors).values(): + if isinstance(color, str) and color.startswith('\033'): + clean_msg = clean_msg.replace(color, '') + timestamp = self._format_time() + self.file_handler.write(f"[{timestamp}] {level} {clean_msg}\n") + self.file_handler.flush() + + def set_phase(self, phase: str, total: int = 0) -> None: + self.current_phase = phase + self.total_items = total + self.current_item = 0 + self.progress(phase, 0, total) + + def progress(self, message: str = "", current: int = 0, total: int = 0) -> None: + if total > 0: + self.total_items = total + if current > 0: + self.current_item = current + + if self.total_items > 0: + percent: float = (self.current_item / self.total_items) * 100 + bar_length: int = 30 + filled: int = int(bar_length * self.current_item / self.total_items) + bar: str = '█' * filled + '░' * (bar_length - filled) + status: str = f"[{self._colorize(bar, Colors.BLUE)}] {self.current_item}/{self.total_items} ({percent:.1f}%)" + else: + status = "" + + msg: str = f"{self._colorize('▶', Colors.BRIGHT_CYAN)} {self.current_phase}" + if status: + msg += f" {status}" + if message: + msg += f" - {message}" + + # 使用 \r 实现进度更新 + sys.stdout.write('\r' + msg) + sys.stdout.flush() + + def debug(self, message: str, category: str = "") -> None: + if self.level > LogLevel.DEBUG: + return + + prefix: str = self._colorize("DEBUG", Colors.GRAY) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('├', Colors.GRAY)} {prefix}: {message}" + self._write(msg, "DEBUG") + + def info(self, message: str, category: str = "") -> None: + prefix: str = self._colorize("INFO", Colors.GREEN) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('├', Colors.GREEN)} {prefix}: {message}" + self._write(msg, "INFO") + + def warning(self, message: str, category: str = "", exc_info: Exception | None = None) -> None: + prefix: str = self._colorize("WARN", Colors.BRIGHT_YELLOW) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('├', Colors.YELLOW)} {prefix}: {message}" + self._write(msg, "WARNING") + + def error(self, message: str, category: str = "", exception: Optional[Exception] = None) -> None: + prefix: str = self._colorize("ERROR", Colors.BRIGHT_RED) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('├', Colors.RED)} {prefix}: {message}" + if exception: + msg += f"\n{self._colorize('│', Colors.RED)} Exception: {type(exception).__name__}: {exception}" + self._write(msg, "ERROR") + + def critical(self, message: str, category: str = "") -> None: + prefix: str = self._colorize("CRITICAL", Colors.BG_RED + Colors.WHITE + Colors.BOLD) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('╠', Colors.RED)} {prefix}: {message}" + self._write(msg, "CRITICAL") + + def success(self, message: str, category: str = "") -> None: + prefix: str = self._colorize("SUCCESS", Colors.BRIGHT_GREEN) + if category: + prefix += f"[{self._colorize(category, Colors.CYAN)}]" + + msg: str = f"{self._colorize('├', Colors.GREEN)} {prefix}: {message}" + self._write(msg, "SUCCESS") + + def compile_error(self, message: str, file: str = "", line: int = 0, code: str = "") -> None: + """编译错误格式化输出""" + header: str = self._colorize("编译错误", Colors.BG_RED + Colors.WHITE + Colors.BOLD) + + msg: str = f"\n{self._colorize('╔', Colors.RED)}{header}{self._colorize('╗', Colors.RED)}\n" + + if file and line > 0: + location: str = f"{self._colorize(file, Colors.CYAN)}:{self._colorize(str(line), Colors.YELLOW)}" + msg += f"{self._colorize('║', Colors.RED)} 位置：{location}\n" + + if code: + msg += f"{self._colorize('║', Colors.RED)} 代码：{self._colorize(code, Colors.WHITE)}\n" + + msg += f"{self._colorize('║', Colors.RED)} 错误：{self._colorize(message, Colors.BRIGHT_RED)}\n" + msg += f"{self._colorize('╚', Colors.RED)}{self._colorize('═' * 50, Colors.RED)}{self._colorize('╝', Colors.RED)}" + + self._write(msg, "ERROR") + + def compile_warning(self, message: str, file: str = "", line: int = 0, code: str = "") -> None: + """编译警告格式化输出""" + header: str = self._colorize("警告", Colors.BG_YELLOW + Colors.BLACK) + + msg: str = f"\n{self._colorize('╔', Colors.YELLOW)}{header}{self._colorize('╗', Colors.YELLOW)}\n" + + if file and line > 0: + location: str = f"{self._colorize(file, Colors.CYAN)}:{self._colorize(str(line), Colors.YELLOW)}" + msg += f"{self._colorize('║', Colors.YELLOW)} 位置：{location}\n" + + if code: + msg += f"{self._colorize('║', Colors.YELLOW)} 代码：{self._colorize(code, Colors.WHITE)}\n" + + msg += f"{self._colorize('║', Colors.YELLOW)} 警告：{self._colorize(message, Colors.BRIGHT_YELLOW)}\n" + msg += f"{self._colorize('╚', Colors.YELLOW)}{self._colorize('═' * 50, Colors.YELLOW)}{self._colorize('╝', Colors.YELLOW)}" + + self._write(msg, "WARNING") + + def summary(self, stats: dict[str, int | str]) -> None: + """输出编译摘要""" + header: str = self._colorize("编译摘要", Colors.BG_BLUE + Colors.WHITE + Colors.BOLD) + + msg: str = f"\n{self._colorize('╔', Colors.BLUE)}{header}{self._colorize('╗', Colors.BLUE)}\n" + + for key, value in stats.items(): + color: str = Colors.GREEN if value > 0 else Colors.WHITE + msg += f"{self._colorize('║', Colors.BLUE)} {key}: {self._colorize(str(value), color)}\n" + + msg += f"{self._colorize('╚', Colors.BLUE)}{self._colorize('═' * 50, Colors.BLUE)}{self._colorize('╝', Colors.BLUE)}" + + self._write(msg, "SUMMARY") + + def close(self) -> None: + if self.file_handler: + self.file_handler.close() + self.info(f"日志已保存到：{self.log_file}") + +# 全局日志实例 +_global_logger: Optional[Logger] = None + +def get_logger(name: str = "TransPyC", log_file: Optional[str] = None, + level: int = LogLevel.INFO, use_colors: bool = True) -> Logger: + global _global_logger + if _global_logger is None: + _global_logger = Logger(name, log_file, level, use_colors) + return _global_logger + +def set_logger(logger: Logger) -> None: + global _global_logger + _global_logger = logger + +# 便捷函数 +def debug(msg: str, category: str = "") -> None: + get_logger().debug(msg, category) + +def info(msg: str, category: str = "") -> None: + get_logger().info(msg, category) + +def warning(msg: str, category: str = "") -> None: + get_logger().warning(msg, category) + +def error(msg: str, category: str = "", exception: Exception | None = None) -> None: + get_logger().error(msg, category, exception) + +def success(msg: str, category: str = "") -> None: + get_logger().success(msg, category) + + +def safe_call(func: Callable[..., Any], *args: Any, context: str = "", **kwargs: Any) -> Any: + """安全调用函数，在异常时根据模式处理。 + + - strict 模式：重新抛出异常 + - relaxed 模式：打印彩色警告日志并返回 None + + Args: + func: 要调用的可调用对象 + *args, **kwargs: 传给 func 的参数 + context: 描述性上下文（用于日志消息） + + Returns: + func 的返回值，或 None（异常时） + """ + try: + return func(*args, **kwargs) + except Exception as e: + msg = f"{context}: {type(e).__name__}: {e}" if context else f"{type(e).__name__}: {e}" + if _ConfigMode == "strict": + raise + get_logger().warning(msg, "Exception") + return None diff --git a/lib/core/__init__.py b/lib/core/__init__.py new file mode 100644 index 0000000..ff06415 --- /dev/null +++ b/lib/core/__init__.py @@ -0,0 +1 @@ +"""编译器核心模块""" diff --git a/lib/core/stub_generator.py b/lib/core/stub_generator.py new file mode 100644 index 0000000..5b60618 --- /dev/null +++ b/lib/core/stub_generator.py @@ -0,0 +1,761 @@ +#!/usr/bin/env python3 +""" +C/H 文件到 Python 存根文件生成器 + +将 C 头文件 (.h) 或 C 源文件 (.c) 转换为 Python 存根文件 (.pyi) +生成的 Python 代码使用 TransPyC 语法，包含类型注解 +""" +from __future__ import annotations + +import os +import re +import sys +from typing import Any +from lib.core.VLogger import get_logger as _vlog + + +class CTypeMapper: + + # 硬编码 C 类型名 → Python 类型映射表 + # 这是 stub 生成器（C 头文件 → Python stub）专用的固定映射， + # 不再依赖运行时 CTypeRegistry._cname_to_class（该映射已删除）。 + _CNAME_TO_PY: dict[str, str] = { + # 基本类型 + 'char': 't.CChar', 'int': 't.CInt', 'short': 't.CShort', + 'long': 't.CLong', 'long long': 't.CLongLong', + 'float': 't.CFloat', 'double': 't.CDouble', 'void': 't.CVoid', + # unsigned / signed 变体 + 'unsigned': 't.CUnsigned', 'unsigned char': 't.CUnsignedChar', + 'unsigned int': 't.CUnsignedInt', 'unsigned short': 't.CUnsignedShort', + 'unsigned long': 't.CUnsignedLong', + 'unsigned long long': 't.CUnsignedLongLong', + 'signed char': 't.CSignedChar', + # bool / size_t + 'bool': 't.CBool', '_Bool': 't.CBool', 'size_t': 't.CSizeT', + 'ssize_t': 't.CLong', + # stdint 固定宽度类型 + 'int8_t': 't.CInt8T', 'int16_t': 't.CInt16T', + 'int32_t': 't.CInt32T', 'int64_t': 't.CInt64T', + 'uint8_t': 't.CUInt8T', 'uint16_t': 't.CUInt16T', + 'uint32_t': 't.CUInt32T', 'uint64_t': 't.CUInt64T', + 'intptr_t': 't.CIntPtrT', 'uintptr_t': 't.CUIntPtrT', + 'ptrdiff_t': 't.CPtrDiffT', + # 宽字符 / 字符类型 + 'wchar_t': 't.CWCharT', 'char8_t': 't.CChar8T', + 'char16_t': 't.CChar16T', 'char32_t': 't.CChar32T', + # 浮点特化类型 + 'float8_t': 't.CFloat8T', 'float16_t': 't.CFloat16T', + 'float32_t': 't.CFloat32T', 'float64_t': 't.CFloat64T', + 'float128_t': 't.CFloat128T', + # signed 速记 + 'signed': 't.CInt', 'signed int': 't.CInt', + 'signed short': 't.CShort', 'signed long': 't.CLong', + } + + @classmethod + def _ResolveCNameToPyType(cls, c_type: str) -> str: + """将 C 类型名解析为 Python 类型字符串（如 'int' → 't.CInt'） + + 使用硬编码映射表，不再依赖 CTypeRegistry._cname_to_class。 + 未识别的类型原样返回（用于 struct/union/enum/typedef 用户自定义名）。 + """ + py: str | None = cls._CNAME_TO_PY.get(c_type) + if py is not None: + return py + return c_type + + @classmethod + def MapType(cls, c_type: str, IsPtr: bool = False, IsArray: bool = False) -> str: + c_type = c_type.strip() + + IsConst: bool = 'const ' in c_type + c_type = c_type.replace('const ', '').replace('const', '').strip() + IsVolatile: bool = 'volatile ' in c_type + c_type = c_type.replace('volatile ', '').replace('volatile', '').strip() + IsStatic: bool = 'static ' in c_type + c_type = c_type.replace('static ', '').replace('static', '').strip() + IsExtern: bool = 'extern ' in c_type + c_type = c_type.replace('extern ', '').replace('extern', '').strip() + IsRegister: bool = 'register ' in c_type + c_type = c_type.replace('register ', '').replace('register', '').strip() + IsInline: bool = 'inline ' in c_type + c_type = c_type.replace('inline ', '').replace('inline', '').strip() + + PtrSuffix: str = '' + if IsPtr or c_type.endswith('*'): + PtrSuffix = ' | t.CPtr' + c_type = c_type.rstrip('*').strip() + + if c_type.startswith('struct '): + c_type = c_type[7:].strip() + elif c_type.startswith('union '): + c_type = c_type[6:].strip() + elif c_type.startswith('enum '): + c_type = c_type[5:].strip() + + PyType: str = cls._ResolveCNameToPyType(c_type) + + result: str = PyType + PtrSuffix + + if IsArray: + result += ' | t.CArrayPtr' + + return result + + +def _find_matching_brace(text: str, start: int) -> int: + depth: int = 0 + i: int = start + while i < len(text): + if text[i] == '{': + depth += 1 + elif text[i] == '}': + depth -= 1 + if depth == 0: + return i + elif text[i] == '"' or text[i] == "'": + quote: str = text[i] + i += 1 + while i < len(text) and text[i] != quote: + if text[i] == '\\': + i += 1 + i += 1 + elif text[i] == '/' and i + 1 < len(text) and text[i + 1] == '/': + while i < len(text) and text[i] != '\n': + i += 1 + continue + elif text[i] == '/' and i + 1 < len(text) and text[i + 1] == '*': + i += 2 + while i + 1 < len(text) and not (text[i] == '*' and text[i + 1] == '/'): + i += 1 + i += 1 + i += 1 + return -1 + + +def _strip_comments(content: str) -> str: + result: list[str] = [] + i: int = 0 + while i < len(content): + if content[i] == '/' and i + 1 < len(content) and content[i + 1] == '/': + while i < len(content) and content[i] != '\n': + i += 1 + elif content[i] == '/' and i + 1 < len(content) and content[i + 1] == '*': + i += 2 + while i + 1 < len(content) and not (content[i] == '*' and content[i + 1] == '/'): + i += 1 + i += 2 + elif content[i] == '"' or content[i] == "'": + quote: str = content[i] + result.append(content[i]) + i += 1 + while i < len(content) and content[i] != quote: + if content[i] == '\\' and i + 1 < len(content): + result.append(content[i]) + i += 1 + result.append(content[i]) + i += 1 + if i < len(content): + result.append(content[i]) + i += 1 + else: + result.append(content[i]) + i += 1 + return ''.join(result) + + +def _strip_preprocessor_guards(content: str) -> str: + lines: list[str] = content.split('\n') + filtered: list[str] = [] + for line in lines: + stripped: str = line.strip() + if stripped.startswith('#ifndef') and '_H' in stripped.upper(): + continue + if stripped.startswith('#define') and '_H' in stripped.upper(): + continue + if stripped == '#endif': + continue + filtered.append(line) + return '\n'.join(filtered) + + +class CHeaderParser: + """C 头文件解析器""" + + def __init__(self) -> None: + self.defines: list[dict[str, str | None]] = [] + self.structs: dict[str, dict[str, Any]] = {} + self.enums: dict[str, dict[str, Any]] = {} + self.typedefs: list[dict[str, str]] = [] + self.functions: dict[str, dict[str, Any]] = {} + self.variables: dict[str, dict[str, Any]] = {} + self.unions: dict[str, dict[str, Any]] = {} + self.func_ptr_typedefs: dict[str, dict[str, Any]] = {} + self._typedef_struct_map: dict[str, str] = {} + self._typedef_enum_map: dict[str, str] = {} + self._typedef_union_map: dict[str, str] = {} + + def ParseFile(self, FilePath: str) -> None: + with open(FilePath, 'r', encoding='utf-8') as f: + content: str = f.read() + self.ParseContent(content) + + def ParseContent(self, content: str) -> None: + content = _strip_comments(content) + content = _strip_preprocessor_guards(content) + self._ParseDefines(content) + self._ParseTypedefStructs(content) + self._ParseTypedefEnums(content) + self._ParseTypedefUnions(content) + self._ParseFuncPtrTypedefs(content) + self._ParseSimpleTypedefs(content) + self._ParseStandaloneStructs(content) + self._ParseStandaloneEnums(content) + self._ParseStandaloneUnions(content) + stripped: str = self._StripBodies(content) + self._ParseFunctions(stripped) + self._ParseVariables(stripped) + + def _StripBodies(self, content: str) -> str: + result: str = content + while True: + pos: int = result.find('{') + if pos < 0: + break + end: int = _find_matching_brace(result, pos) + if end < 0: + break + result = result[:pos] + result[end + 1:] + return result + + def _ParseDefines(self, content: str) -> None: + for match in re.finditer(r'#define\s+(\w+)(?:\s+(.+?))?\s*$', content, re.MULTILINE): + Name: str = match.group(1) + Value: str | None = match.group(2) + if Name.startswith('_') and Name.endswith('_'): + continue + if Name.startswith('__'): + continue + self.defines.append({ + 'name': Name, + 'value': Value.strip() if Value else None, + }) + + def _ParseTypedefStructs(self, content: str) -> None: + pattern: str = r'typedef\s+struct\s*(\w*)\s*\{' + for match in re.finditer(pattern, content): + brace_start: int = content.index('{', match.start()) + brace_end: int = _find_matching_brace(content, brace_start) + if brace_end < 0: + continue + body: str = content[brace_start + 1:brace_end] + rest: str = content[brace_end + 1:].lstrip() + AliasMatch: re.Match[str] | None = re.match(r'(\w+)\s*;', rest) + if not AliasMatch: + continue + AliasName: str = AliasMatch.group(1) + TagName: str | None = match.group(1) if match.group(1) else None + members: list[dict[str, Any]] = self._ParseStructMembers(body) + self.structs[AliasName] = { + 'name': AliasName, + 'tag': TagName, + 'members': members, + } + if TagName: + self._typedef_struct_map[TagName] = AliasName + + def _ParseTypedefEnums(self, content: str) -> None: + pattern: str = r'typedef\s+enum\s*(\w*)\s*\{' + for match in re.finditer(pattern, content): + brace_start: int = content.index('{', match.start()) + brace_end: int = _find_matching_brace(content, brace_start) + if brace_end < 0: + continue + body: str = content[brace_start + 1:brace_end] + rest: str = content[brace_end + 1:].lstrip() + AliasMatch: re.Match[str] | None = re.match(r'(\w+)\s*;', rest) + if not AliasMatch: + continue + AliasName: str = AliasMatch.group(1) + TagName: str | None = match.group(1) if match.group(1) else None + values: list[dict[str, str | None]] = self._ParseEnumValues(body) + self.enums[AliasName] = { + 'name': AliasName, + 'tag': TagName, + 'values': values, + } + if TagName: + self._typedef_enum_map[TagName] = AliasName + + def _ParseTypedefUnions(self, content: str) -> None: + pattern: str = r'typedef\s+union\s*(\w*)\s*\{' + for match in re.finditer(pattern, content): + brace_start: int = content.index('{', match.start()) + brace_end: int = _find_matching_brace(content, brace_start) + if brace_end < 0: + continue + body: str = content[brace_start + 1:brace_end] + rest: str = content[brace_end + 1:].lstrip() + AliasMatch: re.Match[str] | None = re.match(r'(\w+)\s*;', rest) + if not AliasMatch: + continue + AliasName: str = AliasMatch.group(1) + TagName: str | None = match.group(1) if match.group(1) else None + members: list[dict[str, Any]] = self._ParseStructMembers(body) + self.unions[AliasName] = { + 'name': AliasName, + 'tag': TagName, + 'members': members, + } + if TagName: + self._typedef_union_map[TagName] = AliasName + + def _ParseFuncPtrTypedefs(self, content: str) -> None: + pattern: str = r'typedef\s+([\w\s\*]+?)\s*\(\s*\*\s*(\w+)\s*\)\s*\(([^)]*)\)\s*;' + for match in re.finditer(pattern, content): + ReturnType: str = match.group(1).strip() + Name: str = match.group(2) + ParamsStr: str = match.group(3).strip() + params: list[dict[str, Any]] = self._ParseFuncParams(ParamsStr) + self.func_ptr_typedefs[Name] = { + 'name': Name, + 'return_type': ReturnType, + 'params': params, + } + + def _ParseSimpleTypedefs(self, content: str) -> None: + stripped: str = self._StripBodies(content) + pattern: str = r'typedef\s+([\w\s\*]+?)\s+(\w+)\s*;' + for match in re.finditer(pattern, stripped): + BaseType: str = match.group(1).strip() + AliasName: str = match.group(2) + if AliasName in self.structs: + continue + if AliasName in self.enums: + continue + if AliasName in self.unions: + continue + if AliasName in self.func_ptr_typedefs: + continue + if BaseType.startswith('typedef'): + continue + self.typedefs.append({'name': AliasName, 'base': BaseType}) + + def _ParseStandaloneStructs(self, content: str) -> None: + pattern: str = r'(? None: + pattern: str = r'(? None: + pattern: str = r'(? None: + pattern: str = r'(?:(?:static|extern|inline)\s+)*([\w][\w\s\*]*?)\b(\w+)\s*\(([^)]*)\)\s*;' + for match in re.finditer(pattern, content): + ReturnType: str = match.group(1).strip() + FuncName: str = match.group(2) + ParamsStr: str = match.group(3).strip() + if FuncName in self.structs or FuncName in self.enums or FuncName in self.unions: + continue + if FuncName in self.func_ptr_typedefs: + continue + if FuncName in self.defines: + continue + if ReturnType in ('struct', 'union', 'enum', 'typedef'): + continue + if re.match(r'^\d', ReturnType): + continue + if not re.match(r'^[A-Za-z_]', FuncName): + continue + params: list[dict[str, Any]] = self._ParseFuncParams(ParamsStr) + self.functions[FuncName] = { + 'name': FuncName, + 'return_type': ReturnType, + 'params': params, + } + + def _ParseVariables(self, content: str) -> None: + pattern: str = r'(?:(?:static|extern|volatile|const)\s+)*([\w][\w\s\*]*?)\s+(\w+)\s*(?:\[(\d*)\])?\s*;' + for match in re.finditer(pattern, content): + TypeStr: str = match.group(1).strip() + VarName: str = match.group(2) + ArraySize: str | None = match.group(3) + if VarName in self.functions or VarName in self.structs or VarName in self.enums: + continue + if VarName in self.unions or VarName in self.func_ptr_typedefs: + continue + if VarName in self.defines: + continue + if TypeStr.startswith('typedef'): + continue + if TypeStr in ('struct', 'union', 'enum', 'typedef'): + continue + if re.match(r'^(if|else|while|for|return|switch|case|break|continue|do|goto)$', VarName): + continue + if not re.match(r'^[A-Za-z_]', VarName): + continue + IsPtr: bool = '*' in TypeStr + IsArray: bool = ArraySize is not None + self.variables[VarName] = { + 'name': VarName, + 'type': TypeStr, + 'IsPtr': IsPtr, + 'is_array': IsArray, + 'array_size': int(ArraySize) if ArraySize and ArraySize.isdigit() else None, + } + + def _ParseStructMembers(self, body: str) -> list[dict[str, Any]]: + members: list[dict[str, Any]] = [] + for line in body.split(';'): + line = line.strip() + if not line: + continue + arr_match: re.Match[str] | None = re.match(r'([\w\s\*]+?)\s+(\w+)\s*\[(\d*)\]', line) + if arr_match: + MemberType: str = arr_match.group(1).strip() + MemberName: str = arr_match.group(2) + ArraySize: str = arr_match.group(3) + members.append({ + 'name': MemberName, + 'type': MemberType, + 'is_array': True, + 'array_size': int(ArraySize) if ArraySize and ArraySize.isdigit() else None, + }) + continue + ptr_match: re.Match[str] | None = re.match(r'([\w\s\*]+?)\s*\*\s*(\w+)', line) + if ptr_match: + MemberType: str = ptr_match.group(1).strip() + ' *' + MemberName: str = ptr_match.group(2) + members.append({ + 'name': MemberName, + 'type': MemberType, + 'IsPtr': True, + }) + continue + simple_match: re.Match[str] | None = re.match(r'([\w\s]+?)\s+(\w+)', line) + if simple_match: + MemberType: str = simple_match.group(1).strip() + MemberName: str = simple_match.group(2) + members.append({ + 'name': MemberName, + 'type': MemberType, + }) + return members + + def _ParseEnumValues(self, body: str) -> list[dict[str, str | None]]: + values: list[dict[str, str | None]] = [] + for item in body.split(','): + item = item.strip() + if not item: + continue + eq_pos: int = item.find('=') + if eq_pos >= 0: + Name: str = item[:eq_pos].strip() + Value: str | None = item[eq_pos + 1:].strip() + else: + Name = item.strip() + Value = None + if Name and re.match(r'^[A-Za-z_]\w*$', Name): + values.append({'name': Name, 'value': Value}) + return values + + def _ParseFuncParams(self, params_str: str) -> list[dict[str, Any]]: + params: list[dict[str, Any]] = [] + if not params_str or params_str.strip() == 'void' or params_str.strip() == '': + return params + for param in params_str.split(','): + param = param.strip() + if not param: + continue + func_ptr_match: re.Match[str] | None = re.match(r'([\w\s\*]+?)\s*\(\s*\*\s*(\w*)\s*\)\s*\([^)]*\)', param) + if func_ptr_match: + RetType: str = func_ptr_match.group(1).strip() + ParamName: str = func_ptr_match.group(2) or 'callback' + params.append({ + 'name': ParamName, + 'type': RetType + ' (*)', + 'is_func_ptr': True, + }) + continue + arr_match: re.Match[str] | None = re.match(r'([\w\s\*]+?)\s+(\w+)\s*\[\d*\]', param) + if arr_match: + ParamType: str = arr_match.group(1).strip() + ParamName: str = arr_match.group(2) + params.append({ + 'name': ParamName, + 'type': ParamType, + 'is_array': True, + }) + continue + tokens: list[str] = param.split() + if len(tokens) >= 2: + LastToken: str = tokens[-1] + PtrCount: int = 0 + while LastToken.startswith('*'): + PtrCount += 1 + LastToken = LastToken[1:] + TypeTokens: list[str] = tokens[:-1] + for _ in range(PtrCount): + TypeTokens.append('*') + ParamName: str = LastToken + ParamType: str = ' '.join(TypeTokens) + if not ParamName: + ParamName = 'arg' + ParamType = param + params.append({ + 'name': ParamName, + 'type': ParamType, + }) + elif len(tokens) == 1: + params.append({ + 'name': 'arg', + 'type': tokens[0], + }) + return params + + +class PythonStubGenerator: + """Python 存根文件生成器""" + + def __init__(self, parser: CHeaderParser) -> None: + self.parser: CHeaderParser = parser + self.OutputLines: list[str] = [] + + def generate(self, ModuleName: str | None = None) -> str: + self.OutputLines = [] + self._AddHeader(ModuleName) + self._GenerateImports() + self._GenerateDefines() + self._GenerateFuncPtrTypedefs() + self._GenerateSimpleTypedefs() + self._GenerateStructs() + self._GenerateUnions() + self._GenerateEnums() + self._GenerateVariables() + self._GenerateFunctions() + return '\n'.join(self.OutputLines) + + def _AddHeader(self, ModuleName: str | None = None) -> None: + self.OutputLines.append('"""') + self.OutputLines.append('Auto-generated Python stub file from C header') + if ModuleName: + self.OutputLines.append(f'Module: {ModuleName}') + self.OutputLines.append('"""') + self.OutputLines.append('') + + def _GenerateImports(self) -> None: + self.OutputLines.append('import t') + self.OutputLines.append('import c') + self.OutputLines.append('') + + def _GenerateDefines(self) -> None: + if not self.parser.defines: + return + for define in self.parser.defines: + Name: str = define['name'] + Value: str | None = define['value'] + if Value is not None: + self.OutputLines.append(f'{Name}: t.CDefine = {Value}') + else: + self.OutputLines.append(f'{Name}: t.CDefine') + self.OutputLines.append('') + + def _GenerateFuncPtrTypedefs(self) -> None: + if not self.parser.func_ptr_typedefs: + return + for Name, Info in self.parser.func_ptr_typedefs.items(): + RetType: str = CTypeMapper.MapType(Info['return_type']) + ParamTypes: list[str] = [] + for p in Info['params']: + ParamTypes.append(CTypeMapper.MapType(p['type'])) + ParamStr: str = ', '.join(ParamTypes) if ParamTypes else '' + self.OutputLines.append(f'{Name}: t.CTypedef = t.Callable[[{ParamStr}], {RetType}]') + self.OutputLines.append('') + + def _GenerateSimpleTypedefs(self) -> None: + if not self.parser.typedefs: + return + for td in self.parser.typedefs: + AliasName: str = td['name'] + BaseType: str = td['base'] + PyType: str = CTypeMapper.MapType(BaseType) + self.OutputLines.append(f'{AliasName}: t.CTypedef | {PyType}') + self.OutputLines.append('') + + def _GenerateStructs(self) -> None: + if not self.parser.structs: + return + for StructName, StructInfo in self.parser.structs.items(): + self.OutputLines.append(f'@c.Attribute(t.attr.packed)') + self.OutputLines.append(f'class {StructName}(t.CStruct):') + if StructInfo['members']: + for member in StructInfo['members']: + MemberName: str = member['name'] + IsPtr: bool = member.get('IsPtr', False) + IsArray: bool = member.get('is_array', False) + ArraySize: int | None = member.get('array_size') + PyType: str = CTypeMapper.MapType(member['type'], IsPtr=IsPtr) + if IsArray and ArraySize is not None: + self.OutputLines.append(f' {MemberName}: list[{PyType}, {ArraySize}]') + elif IsArray: + self.OutputLines.append(f' {MemberName}: {PyType} | t.CArrayPtr') + else: + self.OutputLines.append(f' {MemberName}: {PyType}') + else: + self.OutputLines.append(' pass') + self.OutputLines.append('') + + def _GenerateUnions(self) -> None: + if not self.parser.unions: + return + for UnionName, UnionInfo in self.parser.unions.items(): + self.OutputLines.append(f'@c.Attribute(t.attr.packed)') + self.OutputLines.append(f'class {UnionName}(t.CUnion):') + if UnionInfo['members']: + for member in UnionInfo['members']: + PyType: str = CTypeMapper.MapType(member['type'], IsPtr=member.get('IsPtr', False)) + self.OutputLines.append(f' {member["name"]}: {PyType}') + else: + self.OutputLines.append(' pass') + self.OutputLines.append('') + + def _GenerateEnums(self) -> None: + if not self.parser.enums: + return + for EnumName, EnumInfo in self.parser.enums.items(): + self.OutputLines.append(f'class __{EnumName.lower()}(t.CEnum):') + if EnumInfo['values']: + for value in EnumInfo['values']: + if value['value'] is not None: + self.OutputLines.append(f' {value["name"]}: t.CDefine = {value["value"]}') + else: + self.OutputLines.append(f' {value["name"]}: t.CDefine') + else: + self.OutputLines.append(' pass') + self.OutputLines.append(f'{EnumName}: t.CTypedef = __{EnumName.lower()}') + self.OutputLines.append('') + + def _GenerateVariables(self) -> None: + if not self.parser.variables: + return + for VarName, VarInfo in self.parser.variables.items(): + PyType: str = CTypeMapper.MapType(VarInfo['type'], IsPtr=VarInfo['IsPtr'], IsArray=VarInfo['is_array']) + self.OutputLines.append(f'{VarName}: {PyType}') + self.OutputLines.append('') + + def _GenerateFunctions(self) -> None: + if not self.parser.functions: + return + for FuncName, FuncInfo in self.parser.functions.items(): + params: list[str] = [] + for param in FuncInfo['params']: + if param.get('is_func_ptr'): + PyType: str = CTypeMapper.MapType(param['type']) + elif param.get('is_array'): + PyType: str = CTypeMapper.MapType(param['type'], IsPtr=True) + else: + PyType: str = CTypeMapper.MapType(param['type']) + params.append(f'{param["name"]}: {PyType}') + ParamStr: str = ', '.join(params) if params else '' + ReturnType: str = CTypeMapper.MapType(FuncInfo['return_type']) + self.OutputLines.append(f'def {FuncName}({ParamStr}) -> {ReturnType}: pass') + self.OutputLines.append('') + + +def GenerateStubFromC(InputFile: str, OutputFile: str | None = None) -> str: + parser: CHeaderParser = CHeaderParser() + parser.ParseFile(InputFile) + generator: PythonStubGenerator = PythonStubGenerator(parser) + ModuleName: str = os.path.splitext(os.path.basename(InputFile))[0] + content: str = generator.generate(ModuleName) + if OutputFile: + with open(OutputFile, 'w', encoding='utf-8') as f: + f.write(content) + _vlog().success(f"生成: {OutputFile}") + return content + + +if __name__ == '__main__': + if len(sys.argv) >= 3: + GenerateStubFromC(sys.argv[1], sys.argv[2]) + elif len(sys.argv) == 2: + _vlog().info(GenerateStubFromC(sys.argv[1])) + else: + test_code: str = ''' +struct memory_block { + uint64_t size; + uint8_t state; + uint8_t order; + struct memory_block *next; + struct memory_block *prev; +}; + +typedef struct memory_block memory_block_t; + +enum color_format { + COLOR_FORMAT_BGRA, + COLOR_FORMAT_RGBA, + COLOR_FORMAT_BGR, + COLOR_FORMAT_RGB +}; + +void *malloc(size_t size); +void free(void *ptr); +void *memcpy(void *dest, const void *src, size_t n); +''' + parser: CHeaderParser = CHeaderParser() + parser.ParseContent(test_code) + generator: PythonStubGenerator = PythonStubGenerator(parser) + _vlog().info(generator.generate("test_header")) diff --git a/lib/core/translator.py b/lib/core/translator.py new file mode 100644 index 0000000..219684d --- /dev/null +++ b/lib/core/translator.py @@ -0,0 +1,34 @@ +# 核心转换逻辑 +# 通过 Mixin 模式组合各功能模块 +# +# 拆分后的模块位于 lib/core/Translator/ 目录下： +# - BaseTranslator.py: __init__、日志、Handle 实例化 +# - AnnotationLoader.py: 注解模块加载 +# - PythonParser.py: Python 文件解析 +# - LlvmGenerator.py: LLVM IR 生成 +# - ConstEval.py: 常量表达式求值 +from __future__ import annotations + +from lib.core.Translator.BaseTranslator import BaseTranslatorMixin, _StrictLog +from lib.core.Translator.AnnotationLoader import AnnotationLoaderMixin +from lib.core.Translator.PythonParser import PythonParserMixin +from lib.core.Translator.LlvmGenerator import LlvmGeneratorMixin +from lib.core.Translator.ConstEval import ConstEvalMixin + + +class Translator( + BaseTranslatorMixin, + AnnotationLoaderMixin, + PythonParserMixin, + LlvmGeneratorMixin, + ConstEvalMixin, +): + """代码转换器 + + 通过 Mixin 模式组合各功能模块。 + 实际方法实现分布在 lib/core/Translator/ 下的各模块中。 + """ + pass + + +__all__: list[str] = ['Translator', '_StrictLog'] diff --git a/lib/includes/__init__.py b/lib/includes/__init__.py new file mode 100644 index 0000000..6129c5d --- /dev/null +++ b/lib/includes/__init__.py @@ -0,0 +1 @@ +"""类型系统与 C 语法定义 (t.py / c.py)""" diff --git a/lib/includes/c.py b/lib/includes/c.py new file mode 100644 index 0000000..5881be5 --- /dev/null +++ b/lib/includes/c.py @@ -0,0 +1,721 @@ +# C语法定义模块 +from __future__ import annotations + +import ast +import logging +from typing import Any, Optional + +from . import t +from lib.constants.config import mode as _config_mode + +def _to_str(node: Any) -> str: + if node is None: + return '' + if isinstance(node, str): + return node + if isinstance(node, list): + return ' && '.join([_to_str(n) for n in node]) + NodeType: str = type(node).__name__ + if NodeType == 'Constant': + return node.value + elif NodeType == 'ID': + return node.name + elif NodeType == 'BinaryOp': + left: str = _to_str(node.left) + right: str = _to_str(node.right) + return f'{left} {node.op} {right}' + elif NodeType == 'UnaryOp': + expr: str = _to_str(node.expr) + return f'{node.op}{expr}' + else: + return repr(node) + +class Asm: + @staticmethod + def _parse_asm_descr(expr: ast.AST) -> str: + """递归解析AST节点，确保获取t.ASM_DESCR的实际值，支持|组合 + 处理场景： + 1. t.ASM_DESCR.XXX → 取属性值 + 2. t.ASM_DESCR.XXX | t.ASM_DESCR.YYY → 拼接两个值 + 3. 常量字符串 → 直接返回 + """ + # 场景1：位或组合（XXX | YYY） + if isinstance(expr, ast.BinOp): + left_val: str = Asm._parse_asm_descr(expr.left) + right_val: str = Asm._parse_asm_descr(expr.right) + return left_val + right_val + + # 场景2：t.ASM_DESCR.XXX 多层属性访问 + elif isinstance(expr, ast.Attribute): + # 第一步：判断是否是 t.ASM_DESCR 的属性 + if isinstance(expr.value, ast.Attribute): + # 内层是 t.ASM_DESCR + if (isinstance(expr.value.value, ast.Name) and + expr.value.value.id == 't' and + expr.value.attr == 'ASM_DESCR'): + # 取 t.ASM_DESCR.XXX 的实际值 + AttrName: str = expr.attr + if hasattr(t.ASM_DESCR, AttrName): + return getattr(t.ASM_DESCR, AttrName, "") + + # 兼容 t.XXX 简化写法（如果有的话） + elif isinstance(expr.value, ast.Name) and expr.value.id == 't': + AttrName: str = expr.attr + if hasattr(t.ASM_DESCR, AttrName): + return getattr(t.ASM_DESCR, AttrName, "") + + # 场景3：直接传常量（如 "r"、"cc"） + elif isinstance(expr, ast.Constant): + return expr.value + + # 其他场景返回空 + return "" + + @staticmethod + def _extract_val_expr(translator: Any, expr_node: ast.AST) -> str: + """从AST表达式节点提取值表达式的字符串形式""" + val_ExprNode: Any = translator.HandleExpr(expr_node)[0] + if isinstance(val_ExprNode, str): + return val_ExprNode + try: + return str(val_ExprNode) + except Exception as _e: + if _config_mode == "strict": + logging.warning(f"异常被忽略: {_e}") + return '' + + @staticmethod + def _parse_asm_template(translator: Any, args: list[ast.AST]) -> tuple[str, list[tuple[str, str]], list[tuple[str, str]], int]: + """解析f-string或常量模板字符串，返回(asm_code, output_ops, input_ops, operand_seq)""" + output_ops: list[tuple[str, str]] = [] + input_ops: list[tuple[str, str]] = [] + operand_seq: int = 0 + asm_code: str = "" + first_arg: ast.AST = args[0] + + if isinstance(first_arg, ast.JoinedStr): + asm_parts: list[str] = [] + for part in first_arg.values: + if isinstance(part, ast.Constant): + asm_parts.append(part.value) + elif isinstance(part, ast.FormattedValue): + expr: ast.AST = part.value + if isinstance(expr, ast.Call) and isinstance(expr.func, ast.Attribute) and expr.func.value.id == 'c': + # 解析内联的AsmInp/AsmOut + call_node: ast.Call = expr + val_expr: str = Asm._extract_val_expr(translator, call_node.args[0]) if len(call_node.args)>=1 else "" + constraint: str = Asm._parse_asm_descr(call_node.args[1]) if len(call_node.args)>=2 else "" + + if call_node.func.attr == 'AsmOut': + output_ops.append((val_expr, constraint)) + asm_parts.append(f"%{operand_seq}") + operand_seq += 1 + elif call_node.func.attr == 'AsmInp': + input_ops.append((val_expr, constraint)) + asm_parts.append(f"%{operand_seq}") + operand_seq += 1 + else: + asm_parts.append(translator.HandleExpr(expr)[0]) + asm_code = ''.join(asm_parts) + elif isinstance(first_arg, ast.Constant): + asm_code = first_arg.value + + return asm_code, output_ops, input_ops, operand_seq + + @staticmethod + def _parse_asm_operands(translator: Any, args: list[ast.AST], keywords: list[ast.keyword], output_ops: list[tuple[str, str]], input_ops: list[tuple[str, str]], operand_seq: int) -> tuple[list[tuple[str, str]], list[tuple[str, str]], int]: + """解析位置参数和关键字参数中的输出/输入操作数""" + + def parse_operand(arg: ast.AST) -> Optional[tuple[str, str, str]]: + """解析 AsmInp/AsmOut 参数""" + if isinstance(arg, ast.Call) and isinstance(arg.func, ast.Attribute) and arg.func.value.id == 'c': + call_node: ast.Call = arg + val_expr: str = Asm._extract_val_expr(translator, call_node.args[0]) if len(call_node.args)>=1 else "" + constraint: str = Asm._parse_asm_descr(call_node.args[1]) if len(call_node.args)>=2 else "" + + if call_node.func.attr == 'AsmOut': + return ('out', val_expr, constraint) + elif call_node.func.attr == 'AsmInp': + return ('in', val_expr, constraint) + return None + + # 处理位置参数中的操作数 + for arg in args[1:]: + result: Optional[tuple[str, str, str]] = parse_operand(arg) + if result: + direction, val_expr, constraint = result + if direction == 'out': + output_ops.append((val_expr, constraint)) + operand_seq += 1 + elif direction == 'in': + input_ops.append((val_expr, constraint)) + operand_seq += 1 + + # 处理关键字参数中的操作数 + for kw in keywords: + if kw.arg == 'out': + # 处理 out 关键字参数 (输出操作数) + if isinstance(kw.value, ast.List): + for elt in kw.value.elts: + result = parse_operand(elt) + if result and result[0] == 'out': + output_ops.append((result[1], result[2])) + operand_seq += 1 + elif isinstance(kw.value, ast.Call): + result = parse_operand(kw.value) + if result and result[0] == 'out': + output_ops.append((result[1], result[2])) + operand_seq += 1 + elif kw.arg == 'inp' or kw.arg == 'inputs': + # 处理 inp/inputs 关键字参数 (输入操作数，追加到f-string中已有输入之后) + if isinstance(kw.value, ast.List): + for elt in kw.value.elts: + if isinstance(elt, ast.Tuple): + # 处理列表中的 (value, constraint) 元组格式 + if len(elt.elts) >= 2: + val_expr: str = Asm._extract_val_expr(translator, elt.elts[0]) if len(elt.elts) >= 1 else "" + constraint: str = Asm._parse_asm_descr(elt.elts[1]) + input_ops.append((val_expr, constraint)) + operand_seq += 1 + else: + result = parse_operand(elt) + if result and result[0] == 'in': + input_ops.append((result[1], result[2])) + operand_seq += 1 + elif isinstance(kw.value, ast.Call): + result = parse_operand(kw.value) + if result and result[0] == 'in': + input_ops.append((result[1], result[2])) + operand_seq += 1 + elif isinstance(kw.value, ast.Tuple): + # 处理 (value, constraint) 元组格式 + if len(kw.value.elts) >= 2: + val_expr: str = Asm._extract_val_expr(translator, kw.value.elts[0]) if len(kw.value.elts) >= 1 else "" + constraint: str = Asm._parse_asm_descr(kw.value.elts[1]) + input_ops.append((val_expr, constraint)) + operand_seq += 1 + + return output_ops, input_ops, operand_seq + + @staticmethod + def _parse_asm_clobbers(args: list[ast.AST], keywords: list[ast.keyword]) -> list[str]: + """解析位置参数和关键字参数中的破坏列表""" + clobbers: list[str] = [] + + # 处理位置参数中的破坏列表 + for arg in args[1:]: + if isinstance(arg, ast.List): + # 解析破坏列表（支持|组合） + for elt in arg.elts: + clobber_val: str = Asm._parse_asm_descr(elt) + if clobber_val: + clobbers.append(clobber_val) + + # 处理关键字参数中的破坏列表 + for kw in keywords: + if kw.arg == 'op': + # 处理 op 关键字参数 (clobber，破坏列表) + if isinstance(kw.value, ast.List): + for elt in kw.value.elts: + clobber_val: str = Asm._parse_asm_descr(elt) + if clobber_val: + clobbers.append(clobber_val) + elif kw.arg == 'clobber': + # 处理 clobber 关键字参数 + if isinstance(kw.value, ast.List): + for elt in kw.value.elts: + clobber_val: str = Asm._parse_asm_descr(elt) + if clobber_val: + clobbers.append(clobber_val) + + return clobbers + + @staticmethod + def _generate_asm_code(asm_code: str, output_ops: list[tuple[str, str]], input_ops: list[tuple[str, str]], clobbers: list[str], asm_format: str) -> list[str]: + """格式化汇编代码并生成最终的__asm__代码字符串""" + # 格式化汇编代码 + asm_lines: list[str] = [line.strip() for line in asm_code.split('\n') if line.strip()] + formatted_asm: str = '\n '.join([f'"{line}\\n"' for line in asm_lines]) + + # 拼接约束字符串 + # 输出约束：如果约束已包含 = 或 +，直接使用；否则添加 = + def format_constraint(constraint: str, is_output: bool = True) -> str: + if not constraint: + return "" + # 如果约束已包含修饰符，直接返回 + if '=' in constraint or '+' in constraint: + return constraint + # 否则添加适当的修饰符 + return ('=' if is_output else '') + constraint + + out_const: str = ', '.join([f'"{format_constraint(c, True)}"({v})' for v, c in output_ops]) if output_ops else "" + in_const: str = ', '.join([f'"{format_constraint(c, False)}"({v})' for v, c in input_ops]) if input_ops else "" + clobber_const: str = ', '.join([f'"{x}"' for x in clobbers]) if clobbers else "" + + # 构建约束部分：确保冒号格式正确 + constraint_parts: list[str] = [] + # 输出约束 + if output_ops: + constraint_parts.append(f': {out_const}') + elif input_ops or clobbers: + constraint_parts.append(':') + + # 输入约束 + if input_ops: + constraint_parts.append(f': {in_const}') + elif clobbers: + constraint_parts.append(':') + + # 破坏列表 + if clobbers: + constraint_parts.append(f': {clobber_const}') + + constraint_str: str = ' '.join(constraint_parts) + + # 生成最终代码 + if asm_format == 'ARM': + # ARM 使用 .arch 指令和不同的内联汇编语法 + final_asm: str = f'__asm__ __volatile__ (\n {formatted_asm} {constraint_str});' + elif asm_format == 'AT&T': + # AT&T 语法不需要额外标记，GCC 默认就是 AT&T + final_asm = f'__asm__ __volatile__ (\n {formatted_asm} {constraint_str});' + else: + # Intel 语法需要 .intel_syntax 指令 + final_asm = f'__asm__ __volatile__ (".intel_syntax noprefix\\n\\t"\n {formatted_asm} ".att_syntax prefix" {constraint_str});' + return [final_asm] + + @staticmethod + def HandleCall(translator: Any, args: list[ast.AST], keywords: list[ast.keyword]) -> list[str]: + if not args: + return ['__asm__ volatile ("nop");'] + + # 解析模板 + asm_code: str + output_ops: list[tuple[str, str]] + input_ops: list[tuple[str, str]] + operand_seq: int + asm_code, output_ops, input_ops, operand_seq = Asm._parse_asm_template(translator, args) + + # 解析操作数 + output_ops, input_ops, operand_seq = Asm._parse_asm_operands( + translator, args, keywords, output_ops, input_ops, operand_seq) + + # 解析破坏列表 + clobbers: list[str] = Asm._parse_asm_clobbers(args, keywords) + + # 解析汇编格式 + asm_format: str = 'Intel' + for kw in keywords: + if kw.arg == 'format': + if isinstance(kw.value, ast.Constant) and isinstance(kw.value.value, str): + asm_format = kw.value.value + + # 生成最终代码 + return Asm._generate_asm_code(asm_code, output_ops, input_ops, clobbers, asm_format) + +class AsmInp: + def __init__(self, value: Any, constraint: Any) -> None: + self.value: Any = value + self.constraint: Any = constraint + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[Any]: + return [] + +class AsmOut: + def __init__(self, value: Any, constraint: Any) -> None: + self.value: Any = value + self.constraint: Any = constraint + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[Any]: + return [] + +class NoBreak: + """ switch 分支无break """ + pass + +class Break: + """ switch 分支提前break """ + pass + +class Load: + """解引用a写入b，等价于 *b = *a，无拷贝副作用""" + def __init__(self, src: Any, dst: Any) -> None: + self.src: Any = src + self.dst: Any = dst + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + if len(args) >= 2: + src: Any = translator.HandleExpr(args[0])[0] + dst: Any = translator.HandleExpr(args[1])[0] + return [f'*({dst}) = *({src})'] + return [] + + +class Addr: + """取地址""" + def __init__(self, addr: Any) -> None: + self.addr: Any = addr + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.Addr() 调用""" + if args: + expr: Any = translator.HandleExpr(args[0])[0] + if isinstance(expr, str): + return [f'&{expr}'] + return [f'&{_to_str(expr)}'] + return ['0'] + + +class Deref: + """解引用""" + def __init__(self, ptr: Any) -> None: + self.ptr: Any = ptr + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.Deref() 调用""" + if args: + expr: Any = translator.HandleExpr(args[0])[0] + if hasattr(expr, 'op') and expr.op == '*': + return [expr] + return [f'*({_to_str(expr)})'] + return ['0'] + + +class DerefAs: + """解引用写入，等价于 *ptr = value""" + def __init__(self, ptr: Any, value: Any) -> None: + self.ptr: Any = ptr + self.value: Any = value + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.DerefAs() 调用""" + if len(args) >= 2: + ptr: Any = translator.HandleExpr(args[0])[0] + val: Any = translator.HandleExpr(args[1])[0] + return [f'*({_to_str(ptr)}) = {_to_str(val)}'] + return [] + + +class Set: + """设置值""" + def __init__(self, key: Any, value: Any) -> None: + self.key: Any = key + self.value: Any = value + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.Set() 调用""" + if len(args) >= 2: + target: Any = translator.HandleExpr(args[0])[0] + value: Any = translator.HandleExpr(args[1])[0] + return [f'{_to_str(target)} = {_to_str(value)}'] + return [] + + +class CReturn: + """多返回值装饰器 + + 用于实现函数多返回值，通过匿名结构体返回实现。 + 例如：@c.CReturn(t.CInt, t.CInt) 表示函数返回两个 int 值。 + 等价于 -> tuple[t.CInt, t.CInt] 返回类型注解。 + + 规则： + 1. CReturn 中有几个类型就说明要返回几个值 + 2. 函数返回类型为匿名结构体 { type1, type2, ... } + 3. return 语句使用 insert_value 构建结构体 + 4. 调用处使用 extract_value 提取各字段 + """ + def __init__(self, *ReturnTypes: Any) -> None: + self.ReturnTypes: tuple[Any, ...] = ReturnTypes + + def __call__(self, func: Any) -> Any: + """装饰器调用""" + # 将返回类型信息附加到函数上 + func._CreturnTypes = self.ReturnTypes + return func + + +class CDefine: + """#define 宏定义""" + def __init__(self, name: Any, value: Any) -> None: + self.name: Any = name + self.value: Any = value + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CDefine() 调用""" + if len(args) >= 1: + if isinstance(args[0], ast.Constant): + name: Any = args[0].value + else: + name = 'MACRO' + if len(args) >= 2: + value: Any = translator.HandleExpr(args[1])[0] + return ['#define ' + str(name) + ' ' + _to_str(value)] + return ['#define ' + str(name)] + return [] + + +class CIfndef: + """#ifndef 条件编译""" + def __init__(self, condition: Any) -> None: + self.condition: Any = condition + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CIfndef() 调用""" + if args: + condition: Any = translator.HandleExpr(args[0])[0] + return ['#ifndef ' + _to_str(condition)] + return ['#if 0'] + +class CIfdef: + """#ifdef 条件编译""" + def __init__(self, condition: Any) -> None: + self.condition: Any = condition + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CIfdef() 调用""" + if args: + condition: Any = translator.HandleExpr(args[0])[0] + return ['#ifdef ' + _to_str(condition)] + return ['#if 0'] + +class CError: + """#error 错误信息""" + def __init__(self, condition: Any) -> None: + self.condition: Any = condition + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CError() 调用""" + if args: + # 如果是字符串常量，保留引号 + if isinstance(args[0], ast.Constant) and isinstance(args[0].value, str): + return ['#error "' + args[0].value + '"'] + else: + condition: Any = translator.HandleExpr(args[0])[0] + return ['#error ' + _to_str(condition)] + return ['#error "Error: Condition is not met."'] + + +class TokenPast: + """## 连接符""" + def __init__(self, left: Any, right: Any) -> None: + self.left: Any = left + self.right: Any = right + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[Any]: + """处理 c.TokenPast() 调用""" + if len(args) >= 2: + # 处理左操作数 - 去掉引号 + if isinstance(args[0], ast.Constant): + left: str = str(args[0].value) + elif isinstance(args[0], ast.Name): + left = args[0].id + else: + left = _to_str(translator.HandleExpr(args[0])[0]) + + # 处理右操作数 - 去掉引号 + if isinstance(args[1], ast.Constant): + right: str = str(args[1].value) + elif isinstance(args[1], ast.Name): + right = args[1].id + else: + right = _to_str(translator.HandleExpr(args[1])[0]) + + # 返回字符串，但用特殊标记包装，让 _AstNodeToStr 能正确处理 + result: str = left + ' ## ' + right + # 使用一个特殊的类来包装结果，让 _AstNodeToStr 直接返回其值 + class TokenPastResult: + def __init__(self, value: str) -> None: + self.value: str = value + def __str__(self) -> str: + return self.value + return [TokenPastResult(result)] + return [''] + + +class CIf: + """#if 条件编译""" + def __init__(self, condition: Any) -> None: + self.condition: Any = condition + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CIf() 调用""" + if args: + condition: Any = translator.HandleExpr(args[0])[0] + return ['#if ' + _to_str(condition)] + return ['#if 0'] + + +class CElif: + """#elif 条件编译""" + def __init__(self, condition: Any) -> None: + self.condition: Any = condition + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CElif() 调用""" + if args: + condition: Any = translator.HandleExpr(args[0])[0] + return ['#elif ' + _to_str(condition)] + return ['#else'] + + +class CElse: + """#else 条件编译""" + def __init__(self) -> None: + pass + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CElse() 调用""" + return ['#else'] + + +class CEndif: + """#endif 条件编译""" + def __init__(self) -> None: + pass + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CEndif() 调用""" + return ['#endif'] + + +class CUndef: + """#undef 取消宏定义""" + def __init__(self, name: Any) -> None: + self.name: Any = name + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CUndef(name) 调用""" + if args: + arg: Any = args[0] + if isinstance(arg, str): + name: str = arg + elif isinstance(arg, ast.Name): + name = arg.id + elif isinstance(arg, ast.Constant): + name = str(arg.value) + else: + name = str(arg) + return [f'#undef {name}'] + return ['#undef'] + + +class LLVMIR: + """内联 LLVM IR + + 用法: c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + 或: c.LLVMIR(f"%{c.LOut(result)} = add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) + + c.LInp(expr) - 输入操作数，翻译后替换为对应的 LLVM 临时变量名 + c.LOut(expr) - 输出操作数，翻译后替换为对应的 LLVM 临时变量名 + """ + def __init__(self, ir_template: Any, ret_type: Any = None) -> None: + self.ir_template: Any = ir_template + self.ret_type: Any = ret_type + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + return ['/* LLVMIR */'] + + +class LInp: + """LLVM IR 输入操作数标记 + + 用法: c.LInp(expr) - 标记 expr 为输入操作数 + 在 c.LLVMIR 的 f-string 中使用，翻译后替换为 %N 形式的操作数引用 + """ + def __init__(self, value: Any) -> None: + self.value: Any = value + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[Any]: + return [] + + +class LOut: + """LLVM IR 输出操作数标记 + + 用法: c.LOut(expr) - 标记 expr 为输出操作数 + 在 c.LLVMIR 的 f-string 中使用，翻译后替换为 %N 形式的操作数引用 + """ + def __init__(self, value: Any) -> None: + self.value: Any = value + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[Any]: + return [] + + +class CPragma: + """#pragma 指令 + + 用于生成 C 语言的 #pragma 指令。 + 例如：c.CPragma("GCC diagnostic push") 生成 #pragma GCC diagnostic push + """ + def __init__(self, directive: Any) -> None: + self.directive: Any = directive + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 c.CPragma() 调用""" + if args: + arg: Any = args[0] + if isinstance(arg, ast.Constant): + directive: Any = arg.value + else: + expr: Any = translator.HandleExpr(arg)[0] + directive = _to_str(expr) + return [f'#pragma {directive}'] + return ['#pragma'] + + +# C 库类字典，用于 HandleCSpecialCall +Library_C = { + 'Asm': Asm, + 'Load': Load, + 'Addr': Addr, + 'Deref': Deref, + 'Set': Set, + 'AsmInp': AsmInp, + 'AsmOut': AsmOut, + 'CDefine': CDefine, + 'CIfndef': CIfndef, + 'CIfdef': CIfdef, + 'CIf': CIf, + 'CElif': CElif, + 'CElse': CElse, + 'CEndif': CEndif, + 'CUndef': CUndef, + 'CError': CError, + 'TokenPast': TokenPast, + 'CReturn': CReturn, + 'CPragma': CPragma, + 'LLVMIR': LLVMIR, + 'LInp': LInp, + 'LOut': LOut, +} + + +class Attribute: + """函数/变量属性装饰器""" + def __init__(self, *attrs: Any) -> None: + self.attrs: tuple[Any, ...] = attrs + + def __call__(self, func: Any) -> Any: + # 将属性附加到函数上 + func._c_attributes = self.attrs + return func diff --git a/lib/includes/t.py b/lib/includes/t.py new file mode 100644 index 0000000..2e1486d --- /dev/null +++ b/lib/includes/t.py @@ -0,0 +1,1163 @@ +# 类型定义模块 +from __future__ import annotations + +import sys +import types +from typing import Any, ClassVar, Optional, TypeVar, Generic, TypeAlias + +from lib.constants.config import mode as _ConfigMode +from lib.core.VLogger import get_logger as _vlog + + +# ============================================================================= +# 类型映射常量 +# ============================================================================= + +# 标记基类注册表（由 @_marker_base 装饰器自动注册，或手动添加） +_MARKER_BASES: set[str] = set() + +def _marker_base(cls: type) -> type: + """将类名注册为标记基类，供 HandlesClassDef 检测使用""" + _MARKER_BASES.add(cls.__name__) + return cls + +# 平台相关类型的默认 Size（可由 configure_platform 修改） +_LONG_SIZE = 64 +_UNSIGNED_LONG_SIZE = 64 +_WCHAR_T_SIZE = 32 +_SIZE_T_SIZE = 64 +_INTPTR_T_SIZE = 64 +_UINTPTR_T_SIZE = 64 +_PTRDIFF_T_SIZE = 64 + +def configure_platform(triple: Optional[str] = None) -> None: + """根据目标三元组配置平台相关类型的大小 + + Args: + triple: LLVM 目标三元组字符串，如 'x86_64-pc-windows-gnu' + """ + global _LONG_SIZE, _UNSIGNED_LONG_SIZE, _WCHAR_T_SIZE + global _SIZE_T_SIZE, _INTPTR_T_SIZE, _UINTPTR_T_SIZE, _PTRDIFF_T_SIZE + if not triple: + return + triple_lower: str = triple.lower() + is_windows: bool = any(k in triple_lower for k in ('windows', 'win32', 'mingw', 'msvc', 'cygwin')) + is_32bit: bool = any(k in triple_lower for k in ('i386', 'i686')) + if is_windows: + _LONG_SIZE = 32 # Windows LLP64: long = 32 位 + _UNSIGNED_LONG_SIZE = 32 + _WCHAR_T_SIZE = 16 # Windows: wchar_t = 16 位 + if is_32bit: + _SIZE_T_SIZE = 32 + _INTPTR_T_SIZE = 32 + _UINTPTR_T_SIZE = 32 + _PTRDIFF_T_SIZE = 32 + + +# ============================================================================= +# CType 基类 +# ============================================================================= + +class CType: + """C 类型基类""" + + PREFIX: ClassVar[str] = 'prefix' + BASE: ClassVar[str] = 'base' + POINTER: ClassVar[str] = 'ptr' + ARRAY: ClassVar[str] = 'array' + NAMED: ClassVar[str] = 'named' + STORAGE_CLASS: ClassVar[str] = 'storage_class' + TYPE_QUALIFIER: ClassVar[str] = 'type_qualifier' + + position: ClassVar[frozenset[str]] = frozenset({BASE}) # 类属性，支持联合类型如 BASE | SPECIAL + + def __init__(self, value: Any = None, *types: Any) -> None: + self.value: Any = value + self.types: tuple[Any, ...] = types + self.Name: str = '' + self.IsBasicType: bool = False + self.IsPointer: bool = False + self.IsSigned: Optional[bool] = None + self.Size: Optional[int] = None + + def GetPositions(self) -> frozenset[str]: + """获取 position 的 frozenset 形式""" + return self.position + + @classmethod + def HasPosition(cls, pos: str) -> bool: + """检查是否包含指定的位置""" + return pos in cls.position + + @classmethod + def IsNamed(cls) -> bool: + """检查是否是命名类型（struct/union/enum/typedef 等）""" + return cls.NAMED in cls.position + + @classmethod + def IsStorageClass(cls) -> bool: + """检查是否是存储类修饰符""" + return cls.STORAGE_CLASS in cls.position + + @classmethod + def IsTypeQualifier(cls) -> bool: + """检查是否是类型限定符""" + return cls.TYPE_QUALIFIER in cls.position + + @classmethod + def IsBasicType(cls) -> bool: + """检查是否是基本类型""" + return cls.BASE in cls.position + + @classmethod + def IsPrefix(cls) -> bool: + """检查是否是前缀类型""" + return cls.PREFIX in cls.position + + def rfind(self, s: str) -> None: + """注解""" + pass + + def get_position(self) -> frozenset[str]: + """获取类型在声明中的位置""" + return self.position + + def __merge__(self, types: Any) -> Any: + return types + + def __or__(self, other: Any) -> list[Any]: + return [self, other] + + def __set_default__(self, **kwargs: Any) -> CTypeDefault: + """设置默认值/初始值，用于 static/global 变量的初始化 + + Args: + **kwargs: 成员名称和值的键值对 + + Returns: + 包含默认值信息的 CTypeDefault 对象 + """ + return CTypeDefault(self, **kwargs) + + def GetFullType(self, VarName: str = '', ArraySizeStr: str = '') -> str: + """获取完整的类型声明字符串 + + Args: + VarName: 变量名 + ArraySizeStr: 数组大小字符串 + + Returns: + 完整的类型声明字符串 + """ + ptr_str: str = '*' if self.IsPointer else '' + if VarName: + return f'{self.Name}{ptr_str} {VarName}{ArraySizeStr}' + return f'{self.Name}{ptr_str}' + + def __repr__(self) -> str: + ptr_str: str = '*' if self.IsPointer else '' + return f'CType({self.Name}{ptr_str}, IsBasicType={self.IsBasicType})' + +class CChar(CType): + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 8 + +class CInt(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 32 + +class CShort(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 16 + +class CLong(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = _LONG_SIZE + +class CLongLong(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 64 + +class CFloat(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 32 + +class CDouble(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 64 + +class CFloat8T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 8 + +class CFloat16T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 16 + +class CFloat32T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 32 + +class CFloat64T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 64 + +class CFloat128T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 128 + +class CVoid(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: Optional[bool] = None + self.Size: int = 0 + +class CUnsigned(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 32 + +class CUnsignedChar(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 8 + +class CUnsignedInt(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 32 + +class CUnsignedShort(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 16 + +class CUnsignedLong(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = _UNSIGNED_LONG_SIZE + +class CUnsignedLongLong(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 64 + +class CSignedChar(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 8 + +@_marker_base +class CStruct(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE, CType.NAMED}) + def __init__(self, value: Any = None, name: Optional[str] = None) -> None: + self.name: Optional[str] = name + super().__init__(value) + +@_marker_base +class CUnion(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE, CType.NAMED}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +@_marker_base +class CEnum(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE, CType.NAMED}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +Enum = CEnum + +@_marker_base +class REnum(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE, CType.NAMED}) + tag: int = 0 # 无用，只是个标记 + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +''' +class Object(CType): + """Python 对象类型，用于支持类方法外联函数""" + position = frozenset({CType.BASE, CType.NAMED}) + def __init__(self, value=None): + super().__init__(value) + def __call__(self, cls): + return cls +''' +def Object() -> None: + pass + +def CVTable() -> None: + pass + +def NoVTable() -> None: + """标记类为非多态继承基类：被子类继承时字段直接展平嵌入，不生成 vtable，避免 vtable 开销。 + 区别于 @t.CVTable（启用多态 vtable）。对应 C++ 的非多态继承（struct A : B {} 无 virtual）。""" + pass + +def TLS() -> None: + """标记函数为 TLS 初始化函数，首次调用时执行，后续调用跳过。 + 编译器自动生成全局标志变量和一次性执行检查代码。 + """ + pass + +# Object 和 CVTable 是普通函数，Exception 是 Python 内建类型，Enum 是 CEnum 的别名 +# 这些名称无法通过 @_marker_base 装饰器自动注册，需手动添加到 _MARKER_BASES +# TLS 是函数装饰器标记，由编译器在函数处理阶段识别 +_MARKER_BASES.update({'Object', 'CVTable', 'Exception', 'Enum'}) + +CTypedef: TypeAlias = TypeAlias + +class _CTypedef(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.NAMED}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + + +class CAuto(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CRegister(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CStatic(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CExtern(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CConst(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.TYPE_QUALIFIER}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CInline(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CExport(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CVolatile(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.TYPE_QUALIFIER}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CSizeT(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = _SIZE_T_SIZE + +class CInt8T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 8 + +class CInt16T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 16 + +class CInt32T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 32 + +class CInt64T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = 64 + +class CUInt8T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 8 + +class CUInt16T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 16 + +class CUInt32T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 32 + +class CUInt64T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 64 + +class CIntPtrT(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = _INTPTR_T_SIZE + +class CUIntPtrT(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = _UINTPTR_T_SIZE + +class CPtrDiffT(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = _PTRDIFF_T_SIZE + +class CWCharT(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = True + self.Size: int = _WCHAR_T_SIZE + +class CChar8T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 8 + +class CChar16T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 16 + +class CChar32T(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 32 + +class CBool(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + self.IsSigned: bool = False + self.Size: int = 8 + +T = TypeVar("T") + +class CPtr(CType, Generic[T]): + position: ClassVar[frozenset[str]] = frozenset({CType.POINTER}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CNeedPtr(CPtr): + """函数签名注解：自动取地址（仅对非指针值）。 + 用作参数类型时，编译器在调用处自动将非指针值取地址为指针。 + 若参数已经是指针，则不做操作。""" + position: ClassVar[frozenset[str]] = frozenset({CType.POINTER}) + auto_addr_mode: ClassVar[str] = 'need' + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CAutoPtr(CPtr): + """函数签名注解：无条件取地址。 + 用作参数类型时，编译器在调用处无条件将值取地址为指针， + 即使值本身已经是指针也会再取一层地址。""" + position: ClassVar[frozenset[str]] = frozenset({CType.POINTER}) + auto_addr_mode: ClassVar[str] = 'always' + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CTypeDefault(CType): + """用于存储 C 类型默认值/初始化的包装类 + + 当使用 t.CStatic.__set_default__(open=mouse_open, ...) 时创建 + """ + def __init__(self, BaseType: Any, **kwargs: Any) -> None: + super().__init__() + self.BaseType: Any = BaseType + self.defaults: dict[str, Any] = kwargs + self.position: frozenset[str] = BaseType.position if hasattr(BaseType, 'position') else CType.BASE + + def __repr__(self) -> str: + return f'CTypeDefault({self.BaseType}, {self.defaults})' + +class CArrayPtr(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.POINTER}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CArray(CType): + """栈上固定数组类型注解。 + 用法: names: t.CArray[VOIDPTR, 32] # void*[32] + data: t.CArray[t.CInt, 64] # int[64] + 第一个参数为元素类型，第二个参数为数组大小（整数或常量名）。""" + position: ClassVar[frozenset[str]] = frozenset({CType.ARRAY}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + def __class_getitem__(cls, item: Any) -> type: + return cls + +class CDefine(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class CPass(CType): + position: ClassVar[frozenset[str]] = frozenset({CType.BASE}) + def __init__(self, value: Any = None) -> None: + super().__init__(value) + +class State(CType): + """状态标记，用于仅声明不定义，等价于 CExport | CExtern""" + position: ClassVar[frozenset[str]] = frozenset({CType.PREFIX, CType.STORAGE_CLASS}) + + @staticmethod + def HandleCall(translator: Any, args: Any, keywords: Any) -> list[str]: + """处理 t.State() 调用""" + return ['t.State'] + +class Postdefinition: + def __init__(self, c: Any) -> None: + pass + +class Bit: + def __init__(self, i: Any) -> None: + pass + +class BigEndian(CType): + """大端序（Big-Endian）字节序标记类 + + 用于标记结构体成员应使用大端序存储。 + 在小端序平台（如x86）上读取时，会自动进行字节交换。 + + 用法示例： + class MyStruct: + a: t.CInt | t.BigEndian # 大端序存储的整数 + """ + IsSigned: ClassVar[bool] = True + Size: ClassVar[int] = 32 + Align: ClassVar[int] = 32 + +class LittleEndian(CType): + """小端序（Little-Endian）字节序标记类 + + 用于标记结构体成员应使用小端序存储。 + 在小端序平台（如x86）上读取时，不需要进行字节交换。 + + 用法示例： + class MyStruct: + b: t.CInt | t.LittleEndian # 小端序存储的整数 + """ + IsSigned: ClassVar[bool] = True + Size: ClassVar[int] = 32 + Align: ClassVar[int] = 32 + +class ASM_DESCR: + """内嵌 ASM 破坏描述符类 + + 定义了所有可能的破坏描述符和约束字符，用于 GCC 内嵌汇编 + """ + # 操作数修饰符 + MODIFIER_OUTPUT = '=' # 输出操作数 + MODIFIER_READWRITE = '+' # 读写操作数 + MODIFIER_INPUT = '' # 输入操作数（默认） + MODIFIER_GLOBAL = '&' # 全局操作数 + + # 寄存器约束（32位命名） + REG_ANY = 'r' # 任何通用寄存器 + REG_EAX = 'a' # EAX/RAX 寄存器 + REG_EBX = 'b' # EBX/RBX 寄存器 + REG_ECX = 'c' # ECX/RCX 寄存器 + REG_EDX = 'd' # EDX/RDX 寄存器 + REG_ESI = 'S' # ESI/RSI 寄存器 + REG_EDI = 'D' # EDI/RDI 寄存器 + REG_STACK = 'q' # 任何通用寄存器（EAX, EBX, ECX, EDX） + REG_FLOAT = 'f' # 浮点寄存器 + REG_MMX = 'y' # MMX 寄存器 + REG_XMM = 'x' # XMM 寄存器 + + # 寄存器约束（64位命名，GCC约束字符与32位相同） + REG_RAX = 'a' # RAX 寄存器 + REG_RBX = 'b' # RBX 寄存器 + REG_RCX = 'c' # RCX 寄存器 + REG_RDX = 'd' # RDX 寄存器 + REG_RSI = 'S' # RSI 寄存器 + REG_RDI = 'D' # RDI 寄存器 + REG_R8 = 'r' # R8 寄存器（需通过r约束+显式指定） + REG_R9 = 'r' # R9 寄存器 + REG_R10 = 'r' # R10 寄存器 + REG_R11 = 'r' # R11 寄存器 + REG_R12 = 'r' # R12 寄存器 + REG_R13 = 'r' # R13 寄存器 + REG_R14 = 'r' # R14 寄存器 + REG_R15 = 'r' # R15 寄存器 + REG_RBP = 'r' # RBP 寄存器 + REG_RSP = 'r' # RSP 寄存器 + + # 内存约束 + MEMORY = 'm' # 内存操作数 + + # 立即数约束 + IMMEDIATE = 'i' # 立即数 + IMMEDIATE_CONST = 'n' # 常量立即数 + + # 综合约束 + ANY = 'g' # 任何通用寄存器、内存或立即数 + MEMORY_OR_REG = 'o' # 内存或寄存器 + + # 破坏描述符（clobber list） + CLOBBER_EAX = 'eax' # 破坏 EAX/RAX 寄存器 + CLOBBER_EBX = 'ebx' # 破坏 EBX/RBX 寄存器 + CLOBBER_ECX = 'ecx' # 破坏 ECX/RCX 寄存器 + CLOBBER_EDX = 'edx' # 破坏 EDX/RDX 寄存器 + CLOBBER_ESI = 'esi' # 破坏 ESI/RSI 寄存器 + CLOBBER_EDI = 'edi' # 破坏 EDI/RDI 寄存器 + CLOBBER_EBP = 'ebp' # 破坏 EBP/RBP 寄存器 + CLOBBER_ESP = 'esp' # 破坏 ESP/RSP 寄存器 + CLOBBER_AL = 'al' # 破坏 AL 寄存器（EAX/RAX 低8位） + CLOBBER_AH = 'ah' # 破坏 AH 寄存器（EAX/RAX 高8位） + CLOBBER_BL = 'bl' # 破坏 BL 寄存器（EBX/RBX 低8位） + CLOBBER_BH = 'bh' # 破坏 BH 寄存器（EBX/RBX 高8位） + CLOBBER_CL = 'cl' # 破坏 CL 寄存器（ECX/RCX 低8位） + CLOBBER_CH = 'ch' # 破坏 CH 寄存器（ECX/RCX 高8位） + CLOBBER_DL = 'dl' # 破坏 DL 寄存器（EDX/RDX 低8位） + CLOBBER_DH = 'dh' # 破坏 DH 寄存器（EDX/RDX 高8位） + CLOBBER_CC = 'cc' # 破坏条件代码寄存器（标志寄存器） + CLOBBER_MEMORY = 'memory' # 破坏内存（表示汇编代码修改了内存） + # 16 位寄存器破坏描述符 + CLOBBER_AX = 'ax' # 破坏 AX 寄存器 + CLOBBER_BX = 'bx' # 破坏 BX 寄存器 + CLOBBER_CX = 'cx' # 破坏 CX 寄存器 + CLOBBER_DX = 'dx' # 破坏 DX 寄存器 + CLOBBER_SI = 'si' # 破坏 SI 寄存器 + CLOBBER_DI = 'di' # 破坏 DI 寄存器 + CLOBBER_BP = 'bp' # 破坏 BP 寄存器 + + # 64位寄存器破坏描述符 + CLOBBER_RAX = 'rax' # 破坏 RAX 寄存器 + CLOBBER_RBX = 'rbx' # 破坏 RBX 寄存器 + CLOBBER_RCX = 'rcx' # 破坏 RCX 寄存器 + CLOBBER_RDX = 'rdx' # 破坏 RDX 寄存器 + CLOBBER_RSI = 'rsi' # 破坏 RSI 寄存器 + CLOBBER_RDI = 'rdi' # 破坏 RDI 寄存器 + CLOBBER_RBP = 'rbp' # 破坏 RBP 寄存器 + CLOBBER_RSP = 'rsp' # 破坏 RSP 寄存器 + CLOBBER_R8 = 'r8' # 破坏 R8 寄存器 + CLOBBER_R9 = 'r9' # 破坏 R9 寄存器 + CLOBBER_R10 = 'r10' # 破坏 R10 寄存器 + CLOBBER_R11 = 'r11' # 破坏 R11 寄存器 + CLOBBER_R12 = 'r12' # 破坏 R12 寄存器 + CLOBBER_R13 = 'r13' # 破坏 R13 寄存器 + CLOBBER_R14 = 'r14' # 破坏 R14 寄存器 + CLOBBER_R15 = 'r15' # 破坏 R15 寄存器 + + # 预定义的组合约束（32位命名） + OUTPUT_REG = '=r' # 输出操作数，使用任何通用寄存器 + OUTPUT_MEM = '=m' # 输出操作数，使用内存 + OUTPUT_EAX = '=a' # 输出操作数，使用 EAX/RAX 寄存器 + OUTPUT_EBX = '=b' # 输出操作数，使用 EBX/RBX 寄存器 + OUTPUT_ECX = '=c' # 输出操作数，使用 ECX/RCX 寄存器 + OUTPUT_EDX = '=d' # 输出操作数，使用 EDX/RDX 寄存器 + OUTPUT_ESI = '=S' # 输出操作数，使用 ESI/RSI 寄存器 + OUTPUT_EDI = '=D' # 输出操作数，使用 EDI/RDI 寄存器 + + # 预定义的组合约束（64位命名） + OUTPUT_RAX = '=a' # 输出操作数，使用 RAX 寄存器 + OUTPUT_RBX = '=b' # 输出操作数，使用 RBX 寄存器 + OUTPUT_RCX = '=c' # 输出操作数，使用 RCX 寄存器 + OUTPUT_RDX = '=d' # 输出操作数，使用 RDX 寄存器 + OUTPUT_RSI = '=S' # 输出操作数，使用 RSI 寄存器 + OUTPUT_RDI = '=D' # 输出操作数，使用 RDI 寄存器 + + INPUT_REG = 'r' # 输入操作数，使用任何通用寄存器 + INPUT_MEM = 'm' # 输入操作数，使用内存 + INPUT_EAX = 'a' # 输入操作数，使用 EAX/RAX 寄存器 + INPUT_EBX = 'b' # 输入操作数，使用 EBX/RBX 寄存器 + INPUT_ECX = 'c' # 输入操作数，使用 ECX/RCX 寄存器 + INPUT_EDX = 'd' # 输入操作数，使用 EDX/RDX 寄存器 + INPUT_ESI = 'S' # 输入操作数，使用 ESI/RSI 寄存器 + INPUT_EDI = 'D' # 输入操作数，使用 EDI/RDI 寄存器 + INPUT_IMM = 'i' # 输入操作数，使用立即数 + INPUT_ANY = 'g' # 输入操作数，使用任何通用寄存器、内存或立即数 + + # 输入约束（64位命名） + INPUT_RAX = 'a' # 输入操作数，使用 RAX 寄存器 + INPUT_RBX = 'b' # 输入操作数，使用 RBX 寄存器 + INPUT_RCX = 'c' # 输入操作数，使用 RCX 寄存器 + INPUT_RDX = 'd' # 输入操作数，使用 RDX 寄存器 + INPUT_RSI = 'S' # 输入操作数，使用 RSI 寄存器 + INPUT_RDI = 'D' # 输入操作数，使用 RDI 寄存器 + + # 所有单独约束字符列表 + ALL_CONSTRAINTS = [ + # 操作数修饰符 + MODIFIER_OUTPUT, MODIFIER_READWRITE, MODIFIER_INPUT, MODIFIER_GLOBAL, + + # 寄存器约束 + REG_ANY, REG_EAX, REG_EBX, REG_ECX, REG_EDX, REG_ESI, REG_EDI, + REG_STACK, REG_FLOAT, REG_MMX, REG_XMM, + + # 内存约束 + MEMORY, + + # 立即数约束 + IMMEDIATE, IMMEDIATE_CONST, + + # 综合约束 + ANY, MEMORY_OR_REG + ] + + # 所有破坏描述符列表 + ALL_CLOBBERS = [ + CLOBBER_EAX, CLOBBER_EBX, CLOBBER_ECX, CLOBBER_EDX, + CLOBBER_ESI, CLOBBER_EDI, CLOBBER_EBP, CLOBBER_ESP, + CLOBBER_CC, CLOBBER_MEMORY, + CLOBBER_RAX, CLOBBER_RBX, CLOBBER_RCX, CLOBBER_RDX, + CLOBBER_RSI, CLOBBER_RDI, CLOBBER_RBP, CLOBBER_RSP, + CLOBBER_R8, CLOBBER_R9, CLOBBER_R10, CLOBBER_R11, + CLOBBER_R12, CLOBBER_R13, CLOBBER_R14, CLOBBER_R15 + ] + + # 所有描述符列表 + ALL = ALL_CONSTRAINTS + ALL_CLOBBERS + + +# 定义 ASM_LIST，包含所有可能的汇编约束字符 +ASM_LIST: list[str] = ASM_DESCR.ALL + + +class attr: + """C 语言 __attribute__ 属性操作类型""" + + @staticmethod + def noreturn() -> str: + return "noreturn" + + @staticmethod + def format(printf: Any, arg1: Any, arg2: Any) -> str: + return f"format({printf}, {arg1}, {arg2})" + + @staticmethod + def section(name: str) -> str: + return f'section("{name}")' + + @staticmethod + def aligned(bytes: int) -> str: + return f"aligned({bytes})" + + @staticmethod + def packed() -> str: + return "packed" + + @staticmethod + def unused() -> str: + return "unused" + + @staticmethod + def used() -> str: + return "used" + + @staticmethod + def weak() -> str: + return "weak" + + @staticmethod + def alias(name: str) -> str: + return f'alias("{name}")' + + @staticmethod + def visibility(type: str) -> str: + return f'visibility("{type}")' + + @staticmethod + def constructor() -> str: + return "constructor" + + @staticmethod + def destructor() -> str: + return "destructor" + + @staticmethod + def always_inline() -> str: + return "always_inline" + + @staticmethod + def noinline() -> str: + return "noinline" + + @staticmethod + def pure() -> str: + return "pure" + + @staticmethod + def const() -> str: + return "const" + + @staticmethod + def malloc() -> str: + return "malloc" + + @staticmethod + def alloc_size(n: int) -> str: + return f"alloc_size({n})" + + @staticmethod + def warn_unused_result() -> str: + return "warn_unused_result" + + @staticmethod + def deprecated(msg: Optional[str] = None) -> str: + if msg: + return f'deprecated("{msg}")' + return "deprecated" + + @staticmethod + def fallthrough() -> str: + return "fallthrough" + + @staticmethod + def likely() -> str: + return "likely" + + @staticmethod + def unlikely() -> str: + return "unlikely" + + @staticmethod + def hot() -> str: + return "hot" + + @staticmethod + def cold() -> str: + return "cold" + + @staticmethod + def interrupt() -> str: + return "interrupt" + + @staticmethod + def naked() -> str: + return "naked" + + @staticmethod + def sentinel(n: int = 0) -> str: + return f"sentinel({n})" + + @staticmethod + def nonnull(*args: Any) -> str: + if args: + return f"nonnull({', '.join(map(str, args))})" + return "nonnull" + + @staticmethod + def returns_nonnull() -> str: + return "returns_nonnull" + + @staticmethod + def access(mode: Any, *args: Any) -> str: + return f"access({mode}, {', '.join(map(str, args))})" + + @staticmethod + def cleanup(func: str) -> str: + return f'cleanup({func})' + + @staticmethod + def transparent_union() -> str: + return "transparent_union" + + @staticmethod + def mode(mode_name: str) -> str: + return f"mode({mode_name})" + + @staticmethod + def vector_size(bytes: int) -> str: + return f"vector_size({bytes})" + + @staticmethod + def target(string: str) -> str: + return f'target("{string}")' + + @staticmethod + def optimize(level: str) -> str: + return f'optimize("{level}")' + + @staticmethod + def no_instrument_function() -> str: + return "no_instrument_function" + + @staticmethod + def no_sanitize(type: str) -> str: + return f"no_sanitize({type})" + + @staticmethod + def no_stack_protector() -> str: + return "no_stack_protector" + + @staticmethod + def stack_protector() -> str: + return "stack_protector" + + @staticmethod + def error(msg: str) -> str: + return f'error("{msg}")' + + @staticmethod + def warning(msg: str) -> str: + return f'warning("{msg}")' + + class llvm: + """LLVM 函数属性描述符 + + 用于在返回类型注解中指定 LLVM 函数属性。 + 例如: def foo() -> t.CInt | t.attr.llvm.nobuiltin | t.attr.llvm.nounwind: + """ + + # 以下空类仅作为类型注解占位符使用，由 HandlesFunctions._GetAttrLLVMName + # 从 AST 直接提取属性名（node.attr），不依赖 CName 属性。 + class nobuiltin: pass + class nounwind: pass + class noredzone: pass + class willreturn: pass + class mustprogress: pass + class optnone: pass + class noinline: pass + class alwaysinline: pass + class readnone: pass + class readonly: pass + class writeonly: pass + class inaccessiblememonly: pass + class inaccessiblemem_or_argmemonly: pass + + +class CTypeRegistry: + """基于 CType 元属性的类型注册表，替代手动映射表 + + 利用 CType 子类的 position/IsSigned/Size 元属性自动推导： + - LLVM IR 类型字符串 (如 'i32', 'double', 'i8*') + - C 类型名称 (如 'int', 'unsigned char', 'long long') + - Python CType 类引用 (如 t.CInt32T) + """ + + _name_to_class: ClassVar[Optional[dict[str, type]]] = None + + @classmethod + def _build(cls) -> None: + if cls._name_to_class is not None: + return + mod: Optional[types.ModuleType] = sys.modules.get(__name__) + if mod is None: + return + cls._name_to_class = {} + for name, obj in vars(mod).items(): + if not isinstance(obj, type) or not issubclass(obj, CType) or obj is CType: + continue + if obj in (CTypeDefault, BigEndian, LittleEndian): + continue + try: + inst: Any = obj() + except Exception: + continue + pos: frozenset[str] = getattr(obj, 'position', frozenset()) + if CType.PREFIX in pos and CType.BASE not in pos: + continue + if CType.POINTER in pos and CType.BASE not in pos: + cls._name_to_class[name] = obj + continue + # 注册基本类型（有 Size）和命名基本类型（CStruct/CUnion/CEnum/REnum）。 + # 不再依赖 CName 属性，依据 position/Size 推断。 + if getattr(inst, 'Size', None) is not None or CType.NAMED in pos: + cls._name_to_class[name] = obj + + @classmethod + def GetClassByName(cls, name: str) -> Optional[type]: + """根据 Python 类名获取 CType 类 (如 'CInt32T' -> t.CInt32T)""" + cls._build() + return cls._name_to_class.get(name) + + @staticmethod + def CTypeToLLVM(ctype_class: Optional[type]) -> str: + """从 CType 类的元属性推导 LLVM IR 类型字符串 + + 规则: + - position 含 POINTER 且不含 BASE → 'i8*' + - position 含 PREFIX 且不含 BASE → '' (修饰符,无LLVM类型) + - Size == 0 且 IsSigned is None → 'void' + - IsSigned is None 且 Size > 0 → 浮点: half/float/double/fp128 + - IsSigned is True/False 且 Size > 0 → 整数: i{Size} + """ + if ctype_class is None: + return 'i8*' + pos: frozenset[str] = getattr(ctype_class, 'position', frozenset()) + if CType.POINTER in pos and CType.BASE not in pos: + return 'i8*' + if CType.PREFIX in pos and CType.BASE not in pos: + return '' + try: + inst: Any = ctype_class() + except Exception: + return 'i8*' + size: Optional[int] = getattr(inst, 'Size', None) + is_signed: Optional[bool] = getattr(inst, 'IsSigned', None) + if size is None or size == 0: + if is_signed is None: + return 'void' + return 'i8*' + if is_signed is None: + float_map: dict[int, str] = {8: 'half', 16: 'half', 32: 'float', 64: 'double', 128: 'fp128'} + return float_map.get(size, f'f{size}') + return f'i{size}' + + _llvm_to_ctype_cache: ClassVar[Optional[dict[str, type]]] = None + + @classmethod + def LLVMToCType(cls, llvm_str: str) -> Optional[type]: + if cls._llvm_to_ctype_cache is None: + cls._build() + cls._llvm_to_ctype_cache = {} + for name, ctype_cls in cls._name_to_class.items(): + pos: frozenset[str] = getattr(ctype_cls, 'position', frozenset()) + if CType.PREFIX in pos and CType.BASE not in pos: + continue + llvm_key: str = cls.CTypeToLLVM(ctype_cls) + if llvm_key and llvm_key not in cls._llvm_to_ctype_cache: + cls._llvm_to_ctype_cache[llvm_key] = ctype_cls + return cls._llvm_to_ctype_cache.get(llvm_str) + + @classmethod + def NameToLLVM(cls, name: str) -> Optional[str]: + """根据 Python 类型名推导 LLVM IR 类型 (如 'CFloat64T' -> 'double') + + 修饰符类型 (CConst, CStatic 等) 返回空字符串。 + 指针类型 (CPtr) 返回 'i8*'。 + 未知类型返回 None。 + """ + cls._build() + mod: Optional[types.ModuleType] = sys.modules.get(__name__) + if mod and hasattr(mod, name): + obj: Any = getattr(mod, name) + if isinstance(obj, type) and issubclass(obj, CType): + return cls.CTypeToLLVM(obj) + return None + + @classmethod + def ResolveName(cls, name: str) -> Optional[tuple[type, int]]: + """解析类型名到 (CType类, 指针层级) + + 支持的格式: + - Python 类名: 'CInt32T' -> (t.CInt32T, 0) + - C 基本类型名/Python 内置名: 'int' -> (t.CInt, 0), 'bool' -> (t.CBool, 0) + + 注意：stdint.h 类型名（int8_t/uint8_t/size_t/intptr_t 等）和 + Win32 大写别名（INT8/UINT8/BYTE/DWORD/FLOAT8 等）由 stdint.py + 通过 t.CTypedef 提供，用户 `from stdint import *` 后即可使用。 + + 指针必须使用 t.CPtr[T] 显式表达（如 t.CPtr[t.CInt]）， + 本方法不再做 PTR 后缀的隐式类型推断，以避免违反类型自由原则。 + """ + cls._build() + direct: Optional[type] = cls._name_to_class.get(name) + if direct is not None: + pos: frozenset[str] = getattr(direct, 'position', frozenset()) + ptr_level: int = 1 if CType.POINTER in pos and CType.BASE not in pos else 0 + return (direct, ptr_level) + # C 基本类型名 / Python 内置类型名 → Python 类名硬编码表。 + # 仅保留语言内置名称；stdint.h 类型名由 stdint.py 的 typedef 提供。 + _CNAME_ALIAS: dict[str, str] = { + # Python 内置类型名 + 'bool': 'CBool', 'int': 'CInt', 'float': 'CFloat', + # C 基本类型名 + 'char': 'CChar', 'short': 'CShort', 'long': 'CLong', + 'long long': 'CLongLong', 'double': 'CDouble', 'void': 'CVoid', + 'unsigned': 'CUnsigned', 'signed': 'CInt', 'signed int': 'CInt', + 'signed char': 'CSignedChar', + 'unsigned char': 'CUnsignedChar', 'unsigned int': 'CUnsignedInt', + 'unsigned short': 'CUnsignedShort', 'unsigned long': 'CUnsignedLong', + 'unsigned long long': 'CUnsignedLongLong', + } + cname_alias: Optional[str] = _CNAME_ALIAS.get(name) + if cname_alias is not None: + alias_cls: Optional[type] = cls._name_to_class.get(cname_alias) + if alias_cls is not None: + return (alias_cls, 0) + return None \ No newline at end of file diff --git a/lib/utils/__init__.py b/lib/utils/__init__.py new file mode 100644 index 0000000..f68d323 --- /dev/null +++ b/lib/utils/__init__.py @@ -0,0 +1 @@ +"""通用工具函数""" diff --git a/lib/utils/helpers.py b/lib/utils/helpers.py new file mode 100644 index 0000000..225a19a --- /dev/null +++ b/lib/utils/helpers.py @@ -0,0 +1,47 @@ +# 工具函数 +from __future__ import annotations + +import os +import subprocess + +from lib.core.VLogger import get_logger as _vlog + + +def DetectFileType(FilePath: str) -> str: + """检测文件类型""" + _: str + ext: str + _, ext = os.path.splitext(FilePath) + return ext.lower() + + +def ExecuteCommand(command: list[str], shell: bool = False, CaptureOutput: bool = True, text: bool = True) -> subprocess.CompletedProcess[str] | None: + """执行命令 + command 应该是列表形式（如 ['gcc', 'file.c', '-o', 'file.exe']） + 避免字符串拼接以防止 shell 注入""" + try: + result: subprocess.CompletedProcess[str] = subprocess.run( + command, + shell=shell, + check=True, + capture_output=CaptureOutput, + text=text + ) + return result + except subprocess.CalledProcessError as e: + _vlog().error(f'编译失败: {e}') + if e.stderr: + _vlog().error(f'错误输出: {e.stderr}') + return None + + +def AppendFileContent(FilePath: str, content: str) -> bool: + """追加文件内容""" + try: + with open(FilePath, 'a', encoding='utf-8') as f: + f.write(content) + f.write('\n') + return True + except Exception as e: + _vlog().error(f'写入文件失败 {FilePath}: {e}') + return False diff --git a/rp.py b/rp.py new file mode 100644 index 0000000..72de661 --- /dev/null +++ b/rp.py @@ -0,0 +1,80 @@ +import os + +# 目录定义 +ROOT_DIR = r"./lib/" +EXTRA_FILES = [ + "./TransPyC.py", + "./StubGen.py", + "./Projectrans.py" +] + +def snake_to_upper_camel(s: str) -> str: + if not s: + return s + + if s.startswith('_'): + first = '_' + rest = s[1:] + else: + first = '' + rest = s + + parts = rest.split('_') + pascal = ''.join(part.capitalize() for part in parts) + return first + pascal + +while True: + old_str = input("请输入原蛇形字符串：").strip() + new_input = input("替换内容（留空自动转大驼峰）：").strip() + + if not old_str: + print("错误：原字符串不能为空\n") + continue + + total_mod = 0 + + # ====================== 1. 处理 lib 下所有 .py ====================== + for root, _, files in os.walk(ROOT_DIR): + for file in files: + if file.endswith(".py"): + file_path = os.path.join(root, file) + try: + with open(file_path, "r", encoding="utf-8") as f: + content = f.read() + + if new_input == "": + replace_target = snake_to_upper_camel(old_str) + new_content = content.replace(old_str, replace_target) + else: + new_content = content.replace(old_str, new_input) + + if new_content != content: + with open(file_path, "w", encoding="utf-8") as f: + f.write(new_content) + print(f"已修改：{file_path}") + total_mod += 1 + except Exception as err: + print(f"异常 {file_path}：{err}") + + # ====================== 2. 处理根目录下 3 个指定文件 ====================== + for file_path in EXTRA_FILES: + if os.path.exists(file_path): + try: + with open(file_path, "r", encoding="utf-8") as f: + content = f.read() + + if new_input == "": + replace_target = snake_to_upper_camel(old_str) + new_content = content.replace(old_str, replace_target) + else: + new_content = content.replace(old_str, new_input) + + if new_content != content: + with open(file_path, "w", encoding="utf-8") as f: + f.write(new_content) + print(f"已修改：{file_path}") + total_mod += 1 + except Exception as err: + print(f"异常 {file_path}：{err}") + + print(f"\n处理完成，共计修改 {total_mod} 个文件\n") \ No newline at end of file diff --git a/t.py b/t.py new file mode 100644 index 0000000..38e7cfc --- /dev/null +++ b/t.py @@ -0,0 +1 @@ +from lib.includes.t import * \ No newline at end of file diff --git a/thistest/App/benchmark_main.py b/thistest/App/benchmark_main.py new file mode 100644 index 0000000..c27abd9 --- /dev/null +++ b/thistest/App/benchmark_main.py @@ -0,0 +1,14 @@ +import w32.win32console as w32cmd +import this +import t, c + + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: + w32cmd.SetConsoleOutputCP(pagecode) + w32cmd.SetConsoleCP(pagecode) + print("你好，世界") + this.this() + return 0 \ No newline at end of file diff --git a/thistest/output/0035c95a18d4f8e8.deps.json b/thistest/output/0035c95a18d4f8e8.deps.json new file mode 100644 index 0000000..6b8ffc5 --- /dev/null +++ b/thistest/output/0035c95a18d4f8e8.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "this": "a7bc8c01684c0001", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "benchmark_main": "c1371d3dc6b676c2", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5"} \ No newline at end of file diff --git a/thistest/output/067c78e9f121dce3.deps.json b/thistest/output/067c78e9f121dce3.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/067c78e9f121dce3.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/output/06f53cc594b4ac6c.deps.json b/thistest/output/06f53cc594b4ac6c.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/06f53cc594b4ac6c.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/output/6446627d4f07a1b5.deps.json b/thistest/output/6446627d4f07a1b5.deps.json new file mode 100644 index 0000000..6b8ffc5 --- /dev/null +++ b/thistest/output/6446627d4f07a1b5.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "this": "a7bc8c01684c0001", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "benchmark_main": "c1371d3dc6b676c2", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5"} \ No newline at end of file diff --git a/thistest/output/72e2d5ccb7cedcf1.deps.json b/thistest/output/72e2d5ccb7cedcf1.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/72e2d5ccb7cedcf1.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/output/7e529fe7a078cfef.deps.json b/thistest/output/7e529fe7a078cfef.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/7e529fe7a078cfef.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/output/_shared_sym_includes.pickle b/thistest/output/_shared_sym_includes.pickle new file mode 100644 index 0000000..634f1e1 Binary files /dev/null and b/thistest/output/_shared_sym_includes.pickle differ diff --git a/thistest/output/a7bc8c01684c0001.deps.json b/thistest/output/a7bc8c01684c0001.deps.json new file mode 100644 index 0000000..6b8ffc5 --- /dev/null +++ b/thistest/output/a7bc8c01684c0001.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "this": "a7bc8c01684c0001", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "benchmark_main": "c1371d3dc6b676c2", "stdint": "f5522571bcce7bcb", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "0035c95a18d4f8e8", "fileio": "0035c95a18d4f8e8", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c", "w32.winsock2": "6446627d4f07a1b5", "winsock2": "6446627d4f07a1b5"} \ No newline at end of file diff --git a/thistest/output/bbdf3bbd4c3bc28c.deps.json b/thistest/output/bbdf3bbd4c3bc28c.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/bbdf3bbd4c3bc28c.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/output/c1371d3dc6b676c2.deps.json b/thistest/output/c1371d3dc6b676c2.deps.json new file mode 100644 index 0000000..d8aa459 --- /dev/null +++ b/thistest/output/c1371d3dc6b676c2.deps.json @@ -0,0 +1 @@ +{"stdio": "6f62fe05c5ea1ceb", "w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "this": "a7bc8c01684c0001", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb"} \ No newline at end of file diff --git a/thistest/output/f6b51804a0ba8ff0.deps.json b/thistest/output/f6b51804a0ba8ff0.deps.json new file mode 100644 index 0000000..215ffb4 --- /dev/null +++ b/thistest/output/f6b51804a0ba8ff0.deps.json @@ -0,0 +1 @@ +{"w32.win32base": "7e529fe7a078cfef", "win32base": "7e529fe7a078cfef", "w32.win32console": "bbdf3bbd4c3bc28c", "win32console": "bbdf3bbd4c3bc28c", "stdint": "f5522571bcce7bcb", "benchmark_main": "f71cf01f35e2325a", "w32.win32file": "f6b51804a0ba8ff0", "win32file": "f6b51804a0ba8ff0", "w32.fileio": "203a3999d5300163", "fileio": "203a3999d5300163", "w32.win32memory": "72e2d5ccb7cedcf1", "win32memory": "72e2d5ccb7cedcf1", "w32.win32process": "067c78e9f121dce3", "win32process": "067c78e9f121dce3", "w32.win32sync": "06f53cc594b4ac6c", "win32sync": "06f53cc594b4ac6c"} \ No newline at end of file diff --git a/thistest/project.json b/thistest/project.json new file mode 100644 index 0000000..96cd56f --- /dev/null +++ b/thistest/project.json @@ -0,0 +1,29 @@ +{ + "$schema": "https://raw.githubusercontent.com/TermiNexus/TransPyC/main/schemas/project-schema.json", + "name": "thistest", + "version": "1.0.0", + "source_dir": "./App", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-relocation-model=pic"] + }, + "linker": { + "cmd": "clang++", + "flags": ["-lmsvcrt", "-lucrt", "-lpthread", "-lmingwex", "-lkernel32", "-Wl,--allow-multiple-definition"], + "output": "app.exe" + }, + "includes": [ + "./includes" + ], + "target": { + "triple": "x86_64-pc-windows-gnu", + "datalayout": "e-m:w-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} diff --git a/thistest/temp/0035c95a18d4f8e8.pyi b/thistest/temp/0035c95a18d4f8e8.pyi new file mode 100644 index 0000000..3b3771f --- /dev/null +++ b/thistest/temp/0035c95a18d4f8e8.pyi @@ -0,0 +1,81 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import stdio +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/thistest/temp/067c78e9f121dce3.pyi b/thistest/temp/067c78e9f121dce3.pyi new file mode 100644 index 0000000..f4f767c --- /dev/null +++ b/thistest/temp/067c78e9f121dce3.pyi @@ -0,0 +1,136 @@ +""" +Auto-generated Python stub file from w32.win32process.py +Module: w32.win32process +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +PROCESS_TERMINATE: t.CDefine = 0x0001 +PROCESS_CREATE_THREAD: t.CDefine = 0x0002 +PROCESS_VM_OPERATION: t.CDefine = 0x0008 +PROCESS_VM_READ: t.CDefine = 0x0010 +PROCESS_VM_WRITE: t.CDefine = 0x0020 +PROCESS_QUERY_INFORMATION: t.CDefine = 0x0400 +PROCESS_ALL_ACCESS: t.CDefine = 0x001FFFFF +THREAD_TERMINATE: t.CDefine = 0x0001 +THREAD_SUSPEND_RESUME: t.CDefine = 0x0002 +THREAD_GET_CONTEXT: t.CDefine = 0x0008 +THREAD_SET_CONTEXT: t.CDefine = 0x0010 +THREAD_QUERY_INFORMATION: t.CDefine = 0x0040 +THREAD_ALL_ACCESS: t.CDefine = 0x001FFFFF +CREATE_SUSPENDED: t.CDefine = 0x00000004 +CREATE_NEW_CONSOLE: t.CDefine = 0x00000010 +CREATE_NEW_PROCESS_GROUP: t.CDefine = 0x00000200 +CREATE_NO_WINDOW: t.CDefine = 0x08000000 +DETACHED_PROCESS: t.CDefine = 0x00000008 +STARTF_USESHOWWINDOW: t.CDefine = 0x00000001 +STARTF_USESTDHANDLES: t.CDefine = 0x00000100 +SW_HIDE: t.CDefine = 0 +SW_SHOW: t.CDefine = 5 +SW_MINIMIZE: t.CDefine = 6 +NORMAL_PRIORITY_CLASS: t.CDefine = 0x00000020 +IDLE_PRIORITY_CLASS: t.CDefine = 0x00000040 +HIGH_PRIORITY_CLASS: t.CDefine = 0x00000080 +REALTIME_PRIORITY_CLASS: t.CDefine = 0x00000100 +BELOW_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00004000 +ABOVE_NORMAL_PRIORITY_CLASS: t.CDefine = 0x00008000 +STILL_ACTIVE: t.CDefine = 259 + +class STARTUPINFOA: + cb: ULONG + lpReserved: CHARPTR + lpDesktop: CHARPTR + lpTitle: CHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class STARTUPINFOW: + cb: ULONG + lpReserved: WCHARPTR + lpDesktop: WCHARPTR + lpTitle: WCHARPTR + dwX: ULONG + dwY: ULONG + dwXSize: ULONG + dwYSize: ULONG + dwXCountChars: ULONG + dwYCountChars: ULONG + dwFillAttribute: ULONG + dwFlags: ULONG + wShowWindow: WORD + cbReserved2: WORD + lpReserved2: VOIDPTR + hStdInput: HANDLE + hStdOutput: HANDLE + hStdError: HANDLE +class PROCESS_INFORMATION: + hProcess: HANDLE + hThread: HANDLE + dwProcessId: ULONG + dwThreadId: ULONG + +def CreateProcessA(lpApplicationName: LPCSTR, lpCommandLine: CHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCSTR, lpStartupInfo: STARTUPINFOA | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def CreateProcessW(lpApplicationName: LPCWSTR, lpCommandLine: WCHARPTR, lpProcessAttributes: SECURITY_ATTRIBUTES | t.CPtr, lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInheritHandles: BOOL, dwCreationFlags: ULONG, lpEnvironment: VOIDPTR, lpCurrentDirectory: LPCWSTR, lpStartupInfo: STARTUPINFOW | t.CPtr, lpProcessInformation: PROCESS_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def TerminateProcess(hProcess: HANDLE, uExitCode: UINT) -> BOOL | t.State: pass + +def GetExitCodeProcess(hProcess: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetExitCodeThread(hThread: HANDLE, lpExitCode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def OpenProcess(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwProcessId: ULONG) -> HANDLE | t.State: pass + +def GetCurrentProcess() -> HANDLE | t.State: pass + +def GetCurrentProcessId() -> ULONG | t.State: pass + +def CreateThread(lpThreadAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwStackSize: t.CSizeT, lpStartAddress: VOIDPTR, lpParameter: VOIDPTR, dwCreationFlags: ULONG, lpThreadId: ULONG | t.CPtr) -> HANDLE | t.State: pass + +def OpenThread(dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwThreadId: ULONG) -> HANDLE | t.State: pass + +def SuspendThread(hThread: HANDLE) -> ULONG | t.State: pass + +def ResumeThread(hThread: HANDLE) -> ULONG | t.State: pass + +def TerminateThread(hThread: HANDLE, dwExitCode: ULONG) -> BOOL | t.State: pass + +def GetCurrentThread() -> HANDLE | t.State: pass + +def GetCurrentThreadId() -> ULONG | t.State: pass + +def GetThreadId(hThread: HANDLE) -> ULONG | t.State: pass + +def GetProcessId(hProcess: HANDLE) -> ULONG | t.State: pass + +def SetThreadPriority(hThread: HANDLE, nPriority: INT) -> BOOL | t.State: pass + +def GetThreadPriority(hThread: HANDLE) -> INT | t.State: pass + +def ExitProcess(uExitCode: UINT) -> VOID | t.State: pass + +def ExitThread(dwExitCode: ULONG) -> VOID | t.State: pass + +def TlsAlloc() -> ULONG | t.State: pass + +def TlsFree(dwTlsIndex: ULONG) -> BOOL | t.State: pass + +def TlsGetValue(dwTlsIndex: ULONG) -> VOIDPTR | t.State: pass + +def TlsSetValue(dwTlsIndex: ULONG, lpTlsValue: VOIDPTR) -> BOOL | t.State: pass diff --git a/thistest/temp/06f53cc594b4ac6c.pyi b/thistest/temp/06f53cc594b4ac6c.pyi new file mode 100644 index 0000000..8b5a8cf --- /dev/null +++ b/thistest/temp/06f53cc594b4ac6c.pyi @@ -0,0 +1,109 @@ +""" +Auto-generated Python stub file from w32.win32sync.py +Module: w32.win32sync +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +class CRITICAL_SECTION: + DebugInfo: VOIDPTR + LockCount: LONG + RecursionCount: LONG + OwningThread: HANDLE + LockSemaphore: HANDLE + SpinCount: QWORD + +def InitializeCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def EnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def LeaveCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def DeleteCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> VOID | t.State: pass + +def TryEnterCriticalSection(lpCriticalSection: CRITICAL_SECTION | t.CPtr) -> BOOL | t.State: pass + +def SetCriticalSectionSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> ULONG | t.State: pass + +def InitializeCriticalSectionAndSpinCount(lpCriticalSection: CRITICAL_SECTION | t.CPtr, dwSpinCount: ULONG) -> BOOL | t.State: pass + +def CreateMutexA(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateMutexW(lpMutexAttributes: SECURITY_ATTRIBUTES | t.CPtr, bInitialOwner: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenMutexA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenMutexW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseMutex(hMutex: HANDLE) -> BOOL | t.State: pass + +def CreateEventA(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateEventW(lpEventAttributes: SECURITY_ATTRIBUTES | t.CPtr, bManualReset: BOOL, bInitialState: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenEventA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenEventW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def SetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def ResetEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def PulseEvent(hEvent: HANDLE) -> BOOL | t.State: pass + +def CreateSemaphoreA(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCSTR) -> HANDLE | t.State: pass + +def CreateSemaphoreW(lpSemaphoreAttributes: SECURITY_ATTRIBUTES | t.CPtr, lInitialCount: LONG, lMaximumCount: LONG, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreA(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCSTR) -> HANDLE | t.State: pass + +def OpenSemaphoreW(dwDesiredAccess: ULONG, bInheritHandle: BOOL, lpName: LPCWSTR) -> HANDLE | t.State: pass + +def ReleaseSemaphore(hSemaphore: HANDLE, lReleaseCount: LONG, lpPreviousCount: LONG | t.CPtr) -> BOOL | t.State: pass + +def WaitForSingleObject(hHandle: HANDLE, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForSingleObjectEx(hHandle: HANDLE, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def WaitForMultipleObjects(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG) -> ULONG | t.State: pass + +def WaitForMultipleObjectsEx(nCount: ULONG, lpHandles: VOIDPTR, bWaitAll: BOOL, dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def InitializeSRWLock(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def AcquireSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockExclusive(SRWLock: VOIDPTR) -> VOID | t.State: pass + +def ReleaseSRWLockShared(SRWLock: VOIDPTR) -> VOID | t.State: pass + + +CONDITION_VARIABLE_LOCKMODE_SHARED: t.CDefine = 0x1 + +def InitializeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def SleepConditionVariableCS(ConditionVariable: VOIDPTR, CriticalSection: CRITICAL_SECTION | t.CPtr, dwMilliseconds: ULONG) -> BOOL | t.State: pass + +def SleepConditionVariableSRW(ConditionVariable: VOIDPTR, SRWLock: VOIDPTR, dwMilliseconds: ULONG, Flags: ULONG) -> BOOL | t.State: pass + +def WakeConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + +def WakeAllConditionVariable(ConditionVariable: VOIDPTR) -> VOID | t.State: pass + + +INIT_ONCE_STATIC_INIT: t.CDefine = 0x00000001 +INIT_ONCE_CHECK_ONLY: t.CDefine = 0x00000002 +INIT_ONCE_ASYNC: t.CDefine = 0x00000004 +INIT_ONCE_INIT_FAILED: t.CDefine = 0x00000008 + +class INIT_ONCE: + Ptr: t.CPtr + +def InitOnceExecuteOnce(InitOnce: INIT_ONCE | t.CPtr, InitFn: VOIDPTR, Parameter: VOIDPTR, Context: VOIDPTR | t.CPtr) -> BOOL | t.State: pass diff --git a/thistest/temp/153805db5704f3e0.pyi b/thistest/temp/153805db5704f3e0.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/153805db5704f3e0.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/203a3999d5300163.pyi b/thistest/temp/203a3999d5300163.pyi new file mode 100644 index 0000000..a0c57dc --- /dev/null +++ b/thistest/temp/203a3999d5300163.pyi @@ -0,0 +1,80 @@ +""" +Auto-generated Python stub file from w32.fileio.py +Module: w32.fileio +""" + + +import t, c +from stdint import * +import w32.win32base +import w32.win32file + +class MODE(t.CEnum): + R = 0 + W = 1 + A = 2 + RP = 3 + WP = 4 + AP = 5 + X = 6 + XP = 7 +class FRESULT(t.CEnum): + OK = 0 + ERR = -1 + ERR_CLOSED = -2 + ERR_PERM = -3 + ERR_IO = -4 + ERR_NOTFOUND = -5 + +SEEK_SET: t.CDefine = 0 +SEEK_CUR: t.CDefine = 1 +SEEK_END: t.CDefine = 2 +INVALID_SET_FILE_POINTER: t.CDefine = -1 +SHARE_READ: t.CDefine = 0x00000001 +SHARE_WRITE: t.CDefine = 0x00000002 +SHARE_DELETE: t.CDefine = 0x00000004 + +class File: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: File, filename: str, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: File) -> 'File' | t.CPtr: pass + def __exit__(self: File) -> t.CInt: pass + def read(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write(self: File, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: File, s: str) -> LONG: pass + def seek(self: File, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: File) -> LONG: pass + def close(self: File) -> LONG: pass + def flush(self: File) -> LONG: pass + def size(self: File) -> LONGLONG: pass + def readline(self: File, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: File, buf: bytes, max_count: ULONG) -> LONG: pass +class FileW: + handle: w32.win32base.HANDLE + closed: bool + can_read: bool + can_write: bool + is_append: bool + _share_mode: ULONG + def __init__(self: FileW, filename: w32.win32base.LPCWSTR, mode: ULONG, share: ULONG) -> t.CInt: pass + def __enter__(self: FileW) -> 'FileW' | t.CPtr: pass + def __exit__(self: FileW) -> t.CInt: pass + def read(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write(self: FileW, buf: bytes, count: ULONG) -> LONG: pass + def write_str(self: FileW, s: str) -> LONG: pass + def seek(self: FileW, offset: LONG, origin: ULONG) -> LONG: pass + def tell(self: FileW) -> LONG: pass + def close(self: FileW) -> LONG: pass + def flush(self: FileW) -> LONG: pass + def size(self: FileW) -> LONGLONG: pass + def readline(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + def read_all(self: FileW, buf: bytes, max_count: ULONG) -> LONG: pass + +def open(filename: str, mode: ULONG, share: ULONG) -> File | t.CPtr: pass + +def openw(filename: LPCWSTR, mode: ULONG, share: ULONG) -> FileW | t.CPtr: pass diff --git a/thistest/temp/28d30831dd445f0f.pyi b/thistest/temp/28d30831dd445f0f.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/28d30831dd445f0f.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/2db2d238d218d3d8.pyi b/thistest/temp/2db2d238d218d3d8.pyi new file mode 100644 index 0000000..d78cb7d --- /dev/null +++ b/thistest/temp/2db2d238d218d3d8.pyi @@ -0,0 +1,9 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +def this() -> t.CInt: pass diff --git a/thistest/temp/2df45af226a48b4b.pyi b/thistest/temp/2df45af226a48b4b.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/2df45af226a48b4b.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/6446627d4f07a1b5.pyi b/thistest/temp/6446627d4f07a1b5.pyi new file mode 100644 index 0000000..d7bcff4 --- /dev/null +++ b/thistest/temp/6446627d4f07a1b5.pyi @@ -0,0 +1,86 @@ +""" +Auto-generated Python stub file from w32.winsock2.py +Module: w32.winsock2 +""" + + +import t, c +from stdint import * +from w32.win32base import * + +WINSOCK_VERSION: t.CDefine = 0x0202 # 2.2 +AF_INET: t.CDefine = 2 +AF_INET6: t.CDefine = 23 +SOCK_STREAM: t.CDefine = 1 # TCP +SOCK_DGRAM: t.CDefine = 2 # UDP +SOCK_RAW: t.CDefine = 3 # 原始套接字 +IPPROTO_TCP: t.CDefine = 6 +IPPROTO_UDP: t.CDefine = 17 +SOL_SOCKET: t.CDefine = 0xFFFF # WinSock2 值 +SO_RCVTIMEO: t.CDefine = 0x1006 # WinSock2 值 +SO_SNDTIMEO: t.CDefine = 0x1005 # WinSock2 值 +SO_REUSEADDR: t.CDefine = 0x0004 # WinSock2 值 +INADDR_ANY: t.CDefine = 0 +SOCKET_ERROR: t.CDefine = -1 +INVALID_SOCKET: t.CDefine = 0xFFFFFFFF # WinSock2: ~0 +MSG_NOSIGNAL: t.CDefine = 0 # Windows 不支持，设为 0 +SD_SEND: t.CDefine = 1 +SD_RECV: t.CDefine = 0 +SD_BOTH: t.CDefine = 2 + +class WSASocketAddr: + family: u16 + port: u16 + addr: u32 + zero: u64 +class WSAData: + wVersion: WORD + wHighVersion: WORD + szDescription: BYTE + szSystemStatus: BYTE + iMaxSockets: u16 + iMaxUdpDg: u16 + lpVendorInfo: CHARPTR +class WSAHostEnt: + h_name: CHARPTR + h_aliases: CHARPTR + h_addrtype: SHORT + h_length: SHORT + h_addr_list: CHARPTR +class WinTimeVal: + tv_sec: LONG + tv_usec: LONG + +def WSAStartup(wVersionRequested: WORD, lpWSAData: WSAData | t.CPtr) -> INT | t.State: pass + +def WSACleanup() -> INT | t.State: pass + +def WSAGetLastError() -> INT | t.State: pass + +def socket(family: INT, type: INT, protocol: INT) -> u64 | t.State: pass + +def closesocket(s: u64) -> INT | t.State: pass + +def connect(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def send(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def recv(s: u64, buf: t.CVoid | t.CPtr, len: INT, flags: INT) -> INT | t.State: pass + +def bind(s: u64, name: WSASocketAddr | t.CPtr, namelen: INT) -> INT | t.State: pass + +def listen(s: u64, backlog: INT) -> INT | t.State: pass + +def accept(s: u64, addr: WSASocketAddr | t.CPtr, addrlen: INT | t.CPtr) -> u64 | t.State: pass + +def setsockopt(s: u64, level: INT, optname: INT, optval: t.CVoid | t.CPtr, optlen: INT) -> INT | t.State: pass + +def shutdown(s: u64, how: INT) -> INT | t.State: pass + +def gethostbyname(name: t.CChar | t.CConst | t.CPtr) -> WSAHostEnt | t.CPtr | t.State: pass + +def ntohs(netshort: u16) -> u16 | t.State: pass + +def htons(hostshort: u16) -> u16 | t.State: pass + +def inet_addr(cp: t.CChar | t.CConst | t.CPtr) -> u32 | t.State: pass diff --git a/thistest/temp/6f62fe05c5ea1ceb.pyi b/thistest/temp/6f62fe05c5ea1ceb.pyi new file mode 100644 index 0000000..448ca53 --- /dev/null +++ b/thistest/temp/6f62fe05c5ea1ceb.pyi @@ -0,0 +1,28 @@ +""" +Auto-generated Python stub file from stdio.py +Module: stdio +""" + + +import t, c + +def printf(fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def fprintf(stream: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def sprintf(buf: bytes, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def snprintf(buf: bytes, size: t.CSizeT, fmt: t.CConst | str, *args) -> t.CInt | t.State: pass + +def puts(s: t.CConst | str) -> t.CInt | t.State: pass + +def fputs(s: t.CConst | str, stream: bytes) -> t.CInt | t.State: pass + +def fgets(buf: bytes, size: t.CInt, stream: bytes) -> bytes | t.State: pass + +def fflush(stream: bytes) -> t.CInt | t.State: pass + + +stdin: t.CExtern | bytes +stdout: t.CExtern | bytes +stderr: t.CExtern | bytes \ No newline at end of file diff --git a/thistest/temp/72e2d5ccb7cedcf1.pyi b/thistest/temp/72e2d5ccb7cedcf1.pyi new file mode 100644 index 0000000..5fceb08 --- /dev/null +++ b/thistest/temp/72e2d5ccb7cedcf1.pyi @@ -0,0 +1,91 @@ +""" +Auto-generated Python stub file from w32.win32memory.py +Module: w32.win32memory +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +MEM_COMMIT: t.CDefine = 0x00001000 +MEM_RESERVE: t.CDefine = 0x00002000 +MEM_DECOMMIT: t.CDefine = 0x00004000 +MEM_RELEASE: t.CDefine = 0x00008000 +MEM_FREE: t.CDefine = 0x00010000 +MEM_RESET: t.CDefine = 0x00080000 +MEM_TOP_DOWN: t.CDefine = 0x00100000 +MEM_WRITE_WATCH: t.CDefine = 0x00200000 +MEM_PHYSICAL: t.CDefine = 0x00400000 +MEM_LARGE_PAGES: t.CDefine = 0x20000000 +PAGE_NOACCESS: t.CDefine = 0x01 +PAGE_READONLY: t.CDefine = 0x02 +PAGE_READWRITE: t.CDefine = 0x04 +PAGE_WRITECOPY: t.CDefine = 0x08 +PAGE_EXECUTE: t.CDefine = 0x10 +PAGE_EXECUTE_READ: t.CDefine = 0x20 +PAGE_EXECUTE_READWRITE: t.CDefine = 0x40 +PAGE_EXECUTE_WRITECOPY: t.CDefine = 0x80 +PAGE_GUARD: t.CDefine = 0x100 +PAGE_NOCACHE: t.CDefine = 0x200 +PAGE_WRITECOMBINE: t.CDefine = 0x400 +HEAP_NO_SERIALIZE: t.CDefine = 0x00000001 +HEAP_GROWABLE: t.CDefine = 0x00000002 +HEAP_GENERATE_EXCEPTIONS: t.CDefine = 0x00000004 +HEAP_ZERO_MEMORY: t.CDefine = 0x00000008 +HEAP_REALLOC_IN_PLACE_ONLY: t.CDefine = 0x00000010 + +class MEMORY_BASIC_INFORMATION: + BaseAddress: VOIDPTR + AllocationBase: VOIDPTR + AllocationProtect: ULONG + RegionSize: t.CSizeT + State: ULONG + Protect: ULONG + Type: ULONG + +def VirtualAlloc(lpAddress: VOIDPTR, dwSize: t.CSizeT, flAllocationType: ULONG, flProtect: ULONG) -> VOIDPTR | t.State: pass + +def VirtualFree(lpAddress: VOIDPTR, dwSize: t.CSizeT, dwFreeType: ULONG) -> BOOL | t.State: pass + +def VirtualProtect(lpAddress: VOIDPTR, dwSize: t.CSizeT, flNewProtect: ULONG, lpflOldProtect: ULONG | t.CPtr) -> BOOL | t.State: pass + +def VirtualQuery(lpAddress: t.CConst | VOIDPTR, lpBuffer: MEMORY_BASIC_INFORMATION | t.CPtr, dwLength: t.CSizeT) -> t.CSizeT | t.State: pass + +def VirtualLock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def VirtualUnlock(lpAddress: VOIDPTR, dwSize: t.CSizeT) -> BOOL | t.State: pass + +def GetProcessHeap() -> HANDLE | t.State: pass + +def HeapCreate(flOptions: ULONG, dwInitialSize: t.CSizeT, dwMaximumSize: t.CSizeT) -> HANDLE | t.State: pass + +def HeapDestroy(hHeap: HANDLE) -> BOOL | t.State: pass + +def HeapAlloc(hHeap: HANDLE, dwFlags: ULONG, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapReAlloc(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def HeapFree(hHeap: HANDLE, dwFlags: ULONG, lpMem: VOIDPTR) -> BOOL | t.State: pass + +def HeapSize(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> t.CSizeT | t.State: pass + +def HeapValidate(hHeap: HANDLE, dwFlags: ULONG, lpMem: t.CConst | VOIDPTR) -> BOOL | t.State: pass + +def HeapCompact(hHeap: HANDLE, dwFlags: ULONG) -> t.CSizeT | t.State: pass + +def GlobalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def GlobalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalLock(hMem: VOIDPTR) -> VOIDPTR | t.State: pass + +def GlobalUnlock(hMem: VOIDPTR) -> BOOL | t.State: pass + +def GlobalSize(hMem: VOIDPTR) -> t.CSizeT | t.State: pass + +def LocalAlloc(uFlags: UINT, dwBytes: t.CSizeT) -> VOIDPTR | t.State: pass + +def LocalFree(hMem: VOIDPTR) -> VOIDPTR | t.State: pass diff --git a/thistest/temp/77dcdc6020d63dc5.pyi b/thistest/temp/77dcdc6020d63dc5.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/77dcdc6020d63dc5.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/7e529fe7a078cfef.pyi b/thistest/temp/7e529fe7a078cfef.pyi new file mode 100644 index 0000000..d7953a2 --- /dev/null +++ b/thistest/temp/7e529fe7a078cfef.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from w32.win32base.py +Module: w32.win32base +""" + +import c + + +import t +from stdint import * + +HANDLE: t.CTypedef = VOIDPTR +LPCSTR: t.CTypedef = t.CConst | t.CChar | t.CPtr +LPCWSTR: t.CTypedef = t.CConst | t.CUnsignedShort | t.CPtr +INVALID_HANDLE_VALUE: t.CDefine = t.CVoid(-1) +NULL: t.CDefine = 0 +TRUE: t.CDefine = 1 +FALSE: t.CDefine = 0 +INFINITE: t.CDefine = 0xFFFFFFFF +WAIT_FAILED: t.CDefine = 0xFFFFFFFF +WAIT_OBJECT_0: t.CDefine = 0 +WAIT_TIMEOUT: t.CDefine = 258 +WAIT_ABANDONED: t.CDefine = 0x80 +MAX_PATH: t.CDefine = 260 +ERROR_SUCCESS: t.CDefine = 0 +ERROR_FILE_NOT_FOUND: t.CDefine = 2 +ERROR_ACCESS_DENIED: t.CDefine = 5 +ERROR_INSUFFICIENT_BUFFER: t.CDefine = 122 + +class SECURITY_ATTRIBUTES: + nLength: ULONG + lpSecurityDescriptor: VOIDPTR + bInheritHandle: BOOL +class OVERLAPPED: + Internal: ULONGLONG + InternalHigh: ULONGLONG + Offset: ULONG + OffsetHigh: ULONG + hEvent: HANDLE +class FILETIME: + dwLowDateTime: DWORD + dwHighDateTime: DWORD +class SYSTEMTIME: + wYear: WORD + wMonth: WORD + wDayOfWeek: WORD + wDay: WORD + wHour: WORD + wMinute: WORD + wSecond: WORD + wMilliseconds: WORD +class GUID: + Data1: DWORD + Data2: WORD + Data3: WORD + Data4: BYTEPTR +class LARGE_INTEGER: + QuadPart: LONGLONG +class ULARGE_INTEGER: + QuadPart: ULONGLONG + +def GetLastError() -> ULONG | t.State: pass + +def SetLastError(dwErrCode: ULONG) -> t.State: pass + +def CloseHandle(hObject: HANDLE) -> BOOL | t.State: pass + +def GetProcAddress(hModule: HANDLE, lpProcName: LPCSTR) -> VOIDPTR | t.State: pass + +def GetModuleHandleA(lpModuleName: LPCSTR) -> HANDLE | t.State: pass + +def GetModuleHandleW(lpModuleName: LPCWSTR) -> HANDLE | t.State: pass + +def LoadLibraryA(lpLibFileName: LPCSTR) -> HANDLE | t.State: pass + +def LoadLibraryW(lpLibFileName: LPCWSTR) -> HANDLE | t.State: pass + +def FreeLibrary(hLibModule: HANDLE) -> BOOL | t.State: pass + +def GetSystemTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def GetLocalTime(lpSystemTime: SYSTEMTIME | t.CPtr) -> t.State: pass + +def FileTimeToSystemTime(lpFileTime: FILETIME | t.CPtr, lpSystemTime: SYSTEMTIME | t.CPtr) -> BOOL | t.State: pass + +def SystemTimeToFileTime(lpSystemTime: SYSTEMTIME | t.CPtr, lpFileTime: FILETIME | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceCounter(lpPerformanceCount: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def QueryPerformanceFrequency(lpFrequency: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def Sleep(dwMilliseconds: ULONG) -> t.State: pass + +def SleepEx(dwMilliseconds: ULONG, bAlertable: BOOL) -> ULONG | t.State: pass + +def GetTickCount() -> ULONG | t.State: pass + +def GetTickCount64() -> ULONGLONG | t.State: pass + +def GetCommandLineA() -> CHARPTR | t.State: pass diff --git a/thistest/temp/81d55b73067644b3.doc.json b/thistest/temp/81d55b73067644b3.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/thistest/temp/81d55b73067644b3.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/thistest/temp/81d55b73067644b3.pyi b/thistest/temp/81d55b73067644b3.pyi new file mode 100644 index 0000000..a51320e --- /dev/null +++ b/thistest/temp/81d55b73067644b3.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from benchmark_main.py +Module: benchmark_main +""" + + +import w32.win32console as w32cmd +import this +import t, c + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: pass diff --git a/thistest/temp/86df096a4b57c422.pyi b/thistest/temp/86df096a4b57c422.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/86df096a4b57c422.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/87f45d9188e9d92f.pyi b/thistest/temp/87f45d9188e9d92f.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/87f45d9188e9d92f.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/9664af47f66007d8.pyi b/thistest/temp/9664af47f66007d8.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/9664af47f66007d8.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/98028bc8b755b5f0.pyi b/thistest/temp/98028bc8b755b5f0.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/98028bc8b755b5f0.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/9db454840b9dbfe8.pyi b/thistest/temp/9db454840b9dbfe8.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/9db454840b9dbfe8.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/_phase1_manifest.json b/thistest/temp/_phase1_manifest.json new file mode 100644 index 0000000..0a89b99 --- /dev/null +++ b/thistest/temp/_phase1_manifest.json @@ -0,0 +1 @@ +{"D:\\Users\\TermiNexus\\Desktop\\TransPyC\\BenchmarkProject\\App\\benchmark_main.py": {"sha1": "f71cf01f35e2325a", "mtime": 1782991404.519475, "size": 233}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\thistest\\App\\benchmark_main.py": {"sha1": "c1371d3dc6b676c2", "mtime": 1784107847.4885626, "size": 263}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdint.py": {"sha1": "f5522571bcce7bcb", "mtime": 1782383975.8824987, "size": 4356}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32base.py": {"sha1": "7e529fe7a078cfef", "mtime": 1782488356.7736557, "size": 2662}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\w32\\win32console.py": {"sha1": "bbdf3bbd4c3bc28c", "mtime": 1781200703.5338137, "size": 5604}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "d:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\this.py": {"sha1": "a7bc8c01684c0001", "mtime": 1784109007.4091208, "size": 1345}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\thistest\\App\\benchmark_main.py": {"sha1": "c1371d3dc6b676c2", "mtime": 1784107847.4885626, "size": 263}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\stdio.py": {"sha1": "6f62fe05c5ea1ceb", "mtime": 1783239556.0959673, "size": 714}, "D:\\Users\\TermiNexus\\Desktop\\TransPyC\\includes\\this.py": {"sha1": "a7bc8c01684c0001", "mtime": 1784109007.4091208, "size": 1345}} \ No newline at end of file diff --git a/thistest/temp/_sha1_map.txt b/thistest/temp/_sha1_map.txt new file mode 100644 index 0000000..c553a6a --- /dev/null +++ b/thistest/temp/_sha1_map.txt @@ -0,0 +1,6 @@ +6f62fe05c5ea1ceb:includes/stdio.py +7e529fe7a078cfef:includes/w32\win32base.py +a7bc8c01684c0001:includes/this.py +bbdf3bbd4c3bc28c:includes/w32\win32console.py +c1371d3dc6b676c2:benchmark_main.py +f5522571bcce7bcb:includes/stdint.py diff --git a/thistest/temp/a642413df3ed705a.pyi b/thistest/temp/a642413df3ed705a.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/a642413df3ed705a.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/a7bc8c01684c0001.pyi b/thistest/temp/a7bc8c01684c0001.pyi new file mode 100644 index 0000000..d78cb7d --- /dev/null +++ b/thistest/temp/a7bc8c01684c0001.pyi @@ -0,0 +1,9 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +def this() -> t.CInt: pass diff --git a/thistest/temp/bbc118f202878cbe.pyi b/thistest/temp/bbc118f202878cbe.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/bbc118f202878cbe.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/bbdf3bbd4c3bc28c.pyi b/thistest/temp/bbdf3bbd4c3bc28c.pyi new file mode 100644 index 0000000..70dcdbc --- /dev/null +++ b/thistest/temp/bbdf3bbd4c3bc28c.pyi @@ -0,0 +1,138 @@ +""" +Auto-generated Python stub file from w32.win32console.py +Module: w32.win32console +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +ENABLE_PROCESSED_INPUT: t.CDefine = 0x0001 +ENABLE_LINE_INPUT: t.CDefine = 0x0002 +ENABLE_ECHO_INPUT: t.CDefine = 0x0004 +ENABLE_WINDOW_INPUT: t.CDefine = 0x0008 +ENABLE_MOUSE_INPUT: t.CDefine = 0x0010 +ENABLE_INSERT_MODE: t.CDefine = 0x0020 +ENABLE_QUICK_EDIT_MODE: t.CDefine = 0x0040 +ENABLE_EXTENDED_FLAGS: t.CDefine = 0x0080 +ENABLE_PROCESSED_OUTPUT: t.CDefine = 0x0001 +ENABLE_WRAP_AT_EOL_OUTPUT: t.CDefine = 0x0002 +ENABLE_VIRTUAL_TERMINAL_PROCESSING: t.CDefine = 0x0004 +DISABLE_NEWLINE_AUTO_RETURN: t.CDefine = 0x0008 +ENABLE_LVB_GRID_WORLDWIDE: t.CDefine = 0x0010 +FOREGROUND_BLUE: t.CDefine = 0x0001 +FOREGROUND_GREEN: t.CDefine = 0x0002 +FOREGROUND_RED: t.CDefine = 0x0004 +FOREGROUND_INTENSITY: t.CDefine = 0x0008 +BACKGROUND_BLUE: t.CDefine = 0x0010 +BACKGROUND_GREEN: t.CDefine = 0x0020 +BACKGROUND_RED: t.CDefine = 0x0040 +BACKGROUND_INTENSITY: t.CDefine = 0x0080 +KEY_EVENT: t.CDefine = 0x0001 +MOUSE_EVENT: t.CDefine = 0x0002 +WINDOW_BUFFER_SIZE_EVENT: t.CDefine = 0x0004 +MENU_EVENT: t.CDefine = 0x0008 +FOCUS_EVENT: t.CDefine = 0x0010 + +class COORD: + X: SHORT + Y: SHORT +class SMALL_RECT: + Left: SHORT + Top: SHORT + Right: SHORT + Bottom: SHORT +class CONSOLE_SCREEN_BUFFER_INFO: + dwSize: COORD + dwCursorPosition: COORD + wAttributes: WORD + srWindow: SMALL_RECT + dwMaximumWindowSize: COORD +class CONSOLE_CURSOR_INFO: + dwSize: ULONG + bVisible: BOOL +class CHAR_INFO: + UnicodeChar: WCHAR + Attributes: WORD +class KEY_EVENT_RECORD: + bKeyDown: BOOL + wRepeatCount: WORD + wVirtualKeyCode: WORD + wVirtualScanCode: WORD + uChar: WCHAR + dwControlKeyState: ULONG +class MOUSE_EVENT_RECORD: + dwMousePosition: COORD + dwButtonState: ULONG + dwControlKeyState: ULONG + dwEventFlags: ULONG +class WINDOW_BUFFER_SIZE_RECORD: + dwSize: COORD +class INPUT_RECORD: + EventType: WORD + Event: KEY_EVENT_RECORD + +def SetConsoleOutputCP(codepage: UINT) -> BOOL | t.State: pass + +def SetConsoleCP(codepage: UINT) -> BOOL | t.State: pass + +def GetConsoleCP() -> UINT | t.State: pass + +def GetConsoleOutputCP() -> UINT | t.State: pass + +def GetConsoleScreenBufferInfo(hConsoleOutput: HANDLE, lpConsoleScreenBufferInfo: CONSOLE_SCREEN_BUFFER_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleScreenBufferSize(hConsoleOutput: HANDLE, dwSize: COORD) -> BOOL | t.State: pass + +def SetConsoleCursorPosition(hConsoleOutput: HANDLE, dwCursorPosition: COORD) -> BOOL | t.State: pass + +def GetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleCursorInfo(hConsoleOutput: HANDLE, lpConsoleCursorInfo: CONSOLE_CURSOR_INFO | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleTextAttribute(hConsoleOutput: HANDLE, wAttributes: WORD) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterA(hConsoleOutput: HANDLE, cCharacter: t.CChar, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputCharacterW(hConsoleOutput: HANDLE, cCharacter: WCHAR, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfCharsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FillConsoleOutputAttribute(hConsoleOutput: HANDLE, wAttribute: WORD, nLength: ULONG, dwWriteCoord: COORD, lpNumberOfAttrsWritten: ULONG | t.CPtr) -> BOOL | t.State: pass + +def WriteConsoleA(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def WriteConsoleW(hConsoleOutput: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfCharsToWrite: ULONG, lpNumberOfCharsWritten: ULONG | t.CPtr, lpReserved: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleA(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def ReadConsoleW(hConsoleInput: HANDLE, lpBuffer: VOIDPTR, nNumberOfCharsToRead: ULONG, lpNumberOfCharsRead: ULONG | t.CPtr, pInputControl: VOIDPTR) -> BOOL | t.State: pass + +def GetConsoleMode(hConsoleHandle: HANDLE, lpMode: ULONG | t.CPtr) -> BOOL | t.State: pass + +def SetConsoleMode(hConsoleHandle: HANDLE, dwMode: ULONG) -> BOOL | t.State: pass + +def ReadConsoleInputA(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def ReadConsoleInputW(hConsoleInput: HANDLE, lpBuffer: INPUT_RECORD | t.CPtr, nLength: ULONG, lpNumberOfEventsRead: ULONG | t.CPtr) -> BOOL | t.State: pass + +def GetNumberOfConsoleInputEvents(hConsoleInput: HANDLE, lpNumberOfEvents: ULONG | t.CPtr) -> BOOL | t.State: pass + +def FlushConsoleInputBuffer(hConsoleInput: HANDLE) -> BOOL | t.State: pass + +def SetConsoleTitleA(lpConsoleTitle: LPCSTR) -> BOOL | t.State: pass + +def SetConsoleTitleW(lpConsoleTitle: LPCWSTR) -> BOOL | t.State: pass + +def GetConsoleTitleA(lpConsoleTitle: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetConsoleTitleW(lpConsoleTitle: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def AllocConsole() -> BOOL | t.State: pass + +def FreeConsole() -> BOOL | t.State: pass + +def SetConsoleWindowInfo(hConsoleOutput: HANDLE, bAbsolute: BOOL, lpConsoleWindow: SMALL_RECT | t.CPtr) -> BOOL | t.State: pass + +def ScrollConsoleScreenBufferA(hConsoleOutput: HANDLE, lpScrollRectangle: SMALL_RECT | t.CPtr, lpClipRectangle: SMALL_RECT | t.CPtr, dwDestinationOrigin: COORD, lpFill: CHAR_INFO | t.CPtr) -> BOOL | t.State: pass diff --git a/thistest/temp/bc52093344ff9400.pyi b/thistest/temp/bc52093344ff9400.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/bc52093344ff9400.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/c1371d3dc6b676c2.doc.json b/thistest/temp/c1371d3dc6b676c2.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/thistest/temp/c1371d3dc6b676c2.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/thistest/temp/c1371d3dc6b676c2.pyi b/thistest/temp/c1371d3dc6b676c2.pyi new file mode 100644 index 0000000..a51320e --- /dev/null +++ b/thistest/temp/c1371d3dc6b676c2.pyi @@ -0,0 +1,14 @@ +""" +Auto-generated Python stub file from benchmark_main.py +Module: benchmark_main +""" + + +import w32.win32console as w32cmd +import this +import t, c + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: pass diff --git a/thistest/temp/c38e082ec6d38c8e.pyi b/thistest/temp/c38e082ec6d38c8e.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/c38e082ec6d38c8e.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/e907dcfbc2e1b32b.pyi b/thistest/temp/e907dcfbc2e1b32b.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/e907dcfbc2e1b32b.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/f5522571bcce7bcb.pyi b/thistest/temp/f5522571bcce7bcb.pyi new file mode 100644 index 0000000..106a202 --- /dev/null +++ b/thistest/temp/f5522571bcce7bcb.pyi @@ -0,0 +1,100 @@ +""" +Auto-generated Python stub file from stdint.py +Module: stdint +""" + +import c + + +import t + +INT: t.CTypedef = t.CInt +INTPTR: t.CTypedef = t.CInt | t.CPtr +BOOL: t.CTypedef = t.CInt +UINT: t.CTypedef = t.CUnsignedInt +UINTPTR: t.CTypedef = UINT | t.CPtr +BYTE: t.CTypedef = t.CUnsignedChar +BYTEPTR: t.CTypedef = BYTE | t.CPtr +WORD: t.CTypedef = t.CUInt16T +DWORD: t.CTypedef = t.CUInt32T +QWORD: t.CTypedef = t.CUInt64T +TCHAR: t.CTypedef = t.CChar +CHARLIST: t.CTypedef = str | t.CPtr +VOID: t.CTypedef = t.CVoid +SHORT: t.CTypedef = t.CShort +SHORTPTR: t.CTypedef = t.CShort | t.CPtr +USHORT: t.CTypedef = t.CUnsignedShort +USHORTPTR: t.CTypedef = t.CUnsignedShort | t.CPtr +LONGLONG: t.CTypedef = t.CLongLong +ULONGLONG: t.CTypedef = t.CUnsignedLongLong +LONG: t.CTypedef = t.CLong +ULONG: t.CTypedef = t.CUnsignedLong +WCHAR: t.CTypedef = WORD +WCHARPTR: t.CTypedef = WORD | t.CPtr +CHARPTR: t.CTypedef = t.CChar | t.CPtr +FSIZE_t: t.CTypedef = DWORD +LBA_t: t.CTypedef = DWORD +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +FLOAT: t.CTypedef = t.CFloat +DOUBLE: t.CTypedef = t.CDouble +FLOAT8: t.CTypedef = t.CFloat8T +FLOAT16: t.CTypedef = t.CFloat16T +FLOAT32: t.CTypedef = t.CFloat32T +FLOAT64: t.CTypedef = t.CFloat64T +FLOAT128: t.CTypedef = t.CFloat128T +INT8: t.CTypedef = t.CInt8T +INT16: t.CTypedef = t.CInt16T +INT32: t.CTypedef = t.CInt32T +INT64: t.CTypedef = t.CInt64T +UINT8: t.CTypedef = t.CUInt8T +UINT16: t.CTypedef = t.CUInt16T +UINT32: t.CTypedef = t.CUInt32T +UINT64: t.CTypedef = t.CUInt64T +INT8PTR: t.CTypedef = t.CInt8T | t.CPtr +INT16PTR: t.CTypedef = t.CInt16T | t.CPtr +INT32PTR: t.CTypedef = t.CInt32T | t.CPtr +INT64PTR: t.CTypedef = t.CInt64T | t.CPtr +UINT8PTR: t.CTypedef = t.CUInt8T | t.CPtr +UINT16PTR: t.CTypedef = t.CUInt16T | t.CPtr +UINT32PTR: t.CTypedef = t.CUInt32T | t.CPtr +UINT64PTR: t.CTypedef = t.CUInt64T | t.CPtr +CHAR8: t.CTypedef = t.CChar8T +CHAR16: t.CTypedef = t.CChar16T +CHAR32: t.CTypedef = t.CChar32T +CHAR8PTR: t.CTypedef = t.CChar8T | t.CPtr +CHAR16PTR: t.CTypedef = t.CChar16T | t.CPtr +CHAR32PTR: t.CTypedef = t.CChar32T | t.CPtr +i8: t.CTypedef = t.CInt8T +i16: t.CTypedef = t.CInt16T +i32: t.CTypedef = t.CInt32T +i64: t.CTypedef = t.CInt64T +u8: t.CTypedef = t.CUInt8T +u16: t.CTypedef = t.CUInt16T +u32: t.CTypedef = t.CUInt32T +u64: t.CTypedef = t.CUInt64T +SIZE_T: t.CTypedef = t.CSizeT +SSIZE_T: t.CTypedef = t.CPtrDiffT +PTRDIFF_T: t.CTypedef = t.CPtrDiffT +int8_t: t.CTypedef = t.CInt8T +int16_t: t.CTypedef = t.CInt16T +int32_t: t.CTypedef = t.CInt32T +int64_t: t.CTypedef = t.CInt64T +uint8_t: t.CTypedef = t.CUInt8T +uint16_t: t.CTypedef = t.CUInt16T +uint32_t: t.CTypedef = t.CUInt32T +uint64_t: t.CTypedef = t.CUInt64T +size_t: t.CTypedef = t.CSizeT +ssize_t: t.CTypedef = t.CPtrDiffT +ptrdiff_t: t.CTypedef = t.CPtrDiffT +intptr_t: t.CTypedef = t.CIntPtrT +uintptr_t: t.CTypedef = t.CUIntPtrT +wchar_t: t.CTypedef = t.CWCharT +char8_t: t.CTypedef = t.CChar8T +char16_t: t.CTypedef = t.CChar16T +char32_t: t.CTypedef = t.CChar32T +float8_t: t.CTypedef = t.CFloat8T +float16_t: t.CTypedef = t.CFloat16T +float32_t: t.CTypedef = t.CFloat32T +float64_t: t.CTypedef = t.CFloat64T +float128_t: t.CTypedef = t.CFloat128T +_Bool: t.CTypedef = t.CBool \ No newline at end of file diff --git a/thistest/temp/f69ce03954b469e7.pyi b/thistest/temp/f69ce03954b469e7.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/f69ce03954b469e7.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/f6b51804a0ba8ff0.pyi b/thistest/temp/f6b51804a0ba8ff0.pyi new file mode 100644 index 0000000..0036d89 --- /dev/null +++ b/thistest/temp/f6b51804a0ba8ff0.pyi @@ -0,0 +1,192 @@ +""" +Auto-generated Python stub file from w32.win32file.py +Module: w32.win32file +""" + +import c + + +import t +from stdint import * +from w32.win32base import * + +GENERIC_READ: t.CDefine = 0x80000000 +GENERIC_WRITE: t.CDefine = 0x40000000 +GENERIC_EXECUTE: t.CDefine = 0x20000000 +GENERIC_ALL: t.CDefine = 0x10000000 +FILE_SHARE_READ: t.CDefine = 0x00000001 +FILE_SHARE_WRITE: t.CDefine = 0x00000002 +FILE_SHARE_DELETE: t.CDefine = 0x00000004 +CREATE_NEW: t.CDefine = 1 +CREATE_ALWAYS: t.CDefine = 2 +OPEN_EXISTING: t.CDefine = 3 +OPEN_ALWAYS: t.CDefine = 4 +TRUNCATE_EXISTING: t.CDefine = 5 +FILE_ATTRIBUTE_READONLY: t.CDefine = 0x00000001 +FILE_ATTRIBUTE_HIDDEN: t.CDefine = 0x00000002 +FILE_ATTRIBUTE_SYSTEM: t.CDefine = 0x00000004 +FILE_ATTRIBUTE_DIRECTORY: t.CDefine = 0x00000010 +FILE_ATTRIBUTE_ARCHIVE: t.CDefine = 0x00000020 +FILE_ATTRIBUTE_NORMAL: t.CDefine = 0x00000080 +FILE_ATTRIBUTE_TEMPORARY: t.CDefine = 0x00000100 +FILE_ATTRIBUTE_COMPRESSED: t.CDefine = 0x00000800 +FILE_ATTRIBUTE_OFFLINE: t.CDefine = 0x00001000 +FILE_ATTRIBUTE_ENCRYPTED: t.CDefine = 0x00004000 +FILE_FLAG_WRITE_THROUGH: t.CDefine = 0x80000000 +FILE_FLAG_OVERLAPPED: t.CDefine = 0x40000000 +FILE_FLAG_NO_BUFFERING: t.CDefine = 0x20000000 +FILE_FLAG_RANDOM_ACCESS: t.CDefine = 0x10000000 +FILE_FLAG_SEQUENTIAL_SCAN: t.CDefine = 0x08000000 +FILE_FLAG_DELETE_ON_CLOSE: t.CDefine = 0x04000000 +FILE_FLAG_BACKUP_SEMANTICS: t.CDefine = 0x02000000 +FILE_FLAG_POSIX_SEMANTICS: t.CDefine = 0x01000000 +FILE_BEGIN: t.CDefine = 0 +FILE_CURRENT: t.CDefine = 1 +FILE_END: t.CDefine = 2 +STD_INPUT_HANDLE: t.CDefine = t.CUnsignedLong(-10) +STD_OUTPUT_HANDLE: t.CDefine = t.CUnsignedLong(-11) +STD_ERROR_HANDLE: t.CDefine = t.CUnsignedLong(-12) +MOVEFILE_REPLACE_EXISTING: t.CDefine = 0x00000001 +MOVEFILE_COPY_ALLOWED: t.CDefine = 0x00000002 +MOVEFILE_WRITE_THROUGH: t.CDefine = 0x00000008 + +class BY_HANDLE_FILE_INFORMATION: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + dwVolumeSerialNumber: ULONG + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + nNumberOfLinks: ULONG + nFileIndexHigh: ULONG + nFileIndexLow: ULONG +class WIN32_FIND_DATAA: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CChar, 260] + cAlternateFileName: t.CArray[t.CChar, 14] +class WIN32_FIND_DATAW: + dwFileAttributes: ULONG + ftCreationTime: FILETIME + ftLastAccessTime: FILETIME + ftLastWriteTime: FILETIME + nFileSizeHigh: ULONG + nFileSizeLow: ULONG + dwReserved0: ULONG + dwReserved1: ULONG + cFileName: t.CArray[t.CUInt16T, 260] + cAlternateFileName: t.CArray[t.CUInt16T, 14] + +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def CreateFileW(lpFileName: LPCWSTR, dwDesiredAccess: ULONG, dwShareMode: ULONG, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, dwCreationDisposition: ULONG, dwFlagsAndAttributes: ULONG, hTemplateFile: HANDLE) -> HANDLE | t.State: pass + +def ReadFile(hFile: HANDLE, lpBuffer: VOIDPTR, nNumberOfBytesToRead: ULONG, lpNumberOfBytesRead: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def WriteFile(hFile: HANDLE, lpBuffer: t.CConst | VOIDPTR, nNumberOfBytesToWrite: ULONG, lpNumberOfBytesWritten: ULONG | t.CPtr, lpOverlapped: OVERLAPPED | t.CPtr) -> BOOL | t.State: pass + +def SetFilePointer(hFile: HANDLE, lDistanceToMove: LONG, lpDistanceToMoveHigh: LONG | t.CPtr, dwMoveMethod: ULONG) -> LONG | t.State: pass + +def SetFilePointerEx(hFile: HANDLE, liDistanceToMove: LARGE_INTEGER | t.CPtr, lpNewFilePointer: LARGE_INTEGER | t.CPtr, dwMoveMethod: ULONG) -> BOOL | t.State: pass + +def GetFileSize(hFile: HANDLE, lpFileSizeHigh: ULONG | t.CPtr) -> ULONG | t.State: pass + +def GetFileSizeEx(hFile: HANDLE, lpFileSize: LARGE_INTEGER | t.CPtr) -> BOOL | t.State: pass + +def SetEndOfFile(hFile: HANDLE) -> BOOL | t.State: pass + +def FlushFileBuffers(hFile: HANDLE) -> BOOL | t.State: pass + +def GetFileType(hFile: HANDLE) -> ULONG | t.State: pass + +def GetFileAttributesA(lpFileName: LPCSTR) -> ULONG | t.State: pass + +def GetFileAttributesW(lpFileName: LPCWSTR) -> ULONG | t.State: pass + +def SetFileAttributesA(lpFileName: LPCSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def SetFileAttributesW(lpFileName: LPCWSTR, dwFileAttributes: ULONG) -> BOOL | t.State: pass + +def DeleteFileA(lpFileName: LPCSTR) -> BOOL | t.State: pass + +def DeleteFileW(lpFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR) -> BOOL | t.State: pass + +def MoveFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR) -> BOOL | t.State: pass + +def MoveFileExA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def MoveFileExW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, dwFlags: ULONG) -> BOOL | t.State: pass + +def CopyFileA(lpExistingFileName: LPCSTR, lpNewFileName: LPCSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CopyFileW(lpExistingFileName: LPCWSTR, lpNewFileName: LPCWSTR, bFailIfExists: BOOL) -> BOOL | t.State: pass + +def CreateDirectoryA(lpPathName: LPCSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def CreateDirectoryW(lpPathName: LPCWSTR, lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr) -> BOOL | t.State: pass + +def RemoveDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def RemoveDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def FindFirstFileA(lpFileName: LPCSTR, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> HANDLE | t.State: pass + +def FindFirstFileW(lpFileName: LPCWSTR, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> HANDLE | t.State: pass + +def FindNextFileA(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAA | t.CPtr) -> BOOL | t.State: pass + +def FindNextFileW(hFindFile: HANDLE, lpFindFileData: WIN32_FIND_DATAW | t.CPtr) -> BOOL | t.State: pass + +def FindClose(hFindFile: HANDLE) -> BOOL | t.State: pass + +def GetStdHandle(nStdHandle: ULONG) -> HANDLE | t.State: pass + +def SetStdHandle(nStdHandle: ULONG, hHandle: HANDLE) -> BOOL | t.State: pass + +def GetFileInformationByHandle(hFile: HANDLE, lpFileInformation: BY_HANDLE_FILE_INFORMATION | t.CPtr) -> BOOL | t.State: pass + +def LockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToLockLow: ULONG, nNumberOfBytesToLockHigh: ULONG) -> BOOL | t.State: pass + +def UnlockFile(hFile: HANDLE, dwFileOffsetLow: ULONG, dwFileOffsetHigh: ULONG, nNumberOfBytesToUnlockLow: ULONG, nNumberOfBytesToUnlockHigh: ULONG) -> BOOL | t.State: pass + +def GetTempPathA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetTempPathW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def GetTempFileNameA(lpPathName: LPCSTR, lpPrefixString: LPCSTR, uUnique: UINT, lpTempFileName: CHARPTR) -> UINT | t.State: pass + +def GetTempFileNameW(lpPathName: LPCWSTR, lpPrefixString: LPCWSTR, uUnique: UINT, lpTempFileName: WCHARPTR) -> UINT | t.State: pass + +def GetCurrentDirectoryA(nBufferLength: ULONG, lpBuffer: CHARPTR) -> ULONG | t.State: pass + +def GetCurrentDirectoryW(nBufferLength: ULONG, lpBuffer: WCHARPTR) -> ULONG | t.State: pass + +def SetCurrentDirectoryA(lpPathName: LPCSTR) -> BOOL | t.State: pass + +def SetCurrentDirectoryW(lpPathName: LPCWSTR) -> BOOL | t.State: pass + +def GetModuleFileNameA(hModule: HANDLE, lpFilename: CHARPTR, nSize: ULONG) -> ULONG | t.State: pass + +def GetModuleFileNameW(hModule: HANDLE, lpFilename: WCHARPTR, nSize: ULONG) -> ULONG | t.State: pass + + +HANDLE_FLAG_INHERIT: t.CDefine = 0x00000001 +HANDLE_FLAG_PROTECT_FROM_CLOSE: t.CDefine = 0x00000002 + +def CreatePipe(hReadPipe: HANDLE | t.CPtr, hWritePipe: HANDLE | t.CPtr, lpPipeAttributes: SECURITY_ATTRIBUTES | t.CPtr, nSize: ULONG) -> BOOL | t.State: pass + +def SetHandleInformation(hObject: HANDLE, dwMask: ULONG, dwFlags: ULONG) -> BOOL | t.State: pass + +def DuplicateHandle(hSourceProcessHandle: HANDLE, hSourceHandle: HANDLE, hTargetProcessHandle: HANDLE, lpTargetHandle: HANDLE | t.CPtr, dwDesiredAccess: ULONG, bInheritHandle: BOOL, dwOptions: ULONG) -> BOOL | t.State: pass + + +DUPLICATE_SAME_ACCESS: t.CDefine = 0x00000002 \ No newline at end of file diff --git a/thistest/temp/f6d5ab5251ee8b21.pyi b/thistest/temp/f6d5ab5251ee8b21.pyi new file mode 100644 index 0000000..a00afbc --- /dev/null +++ b/thistest/temp/f6d5ab5251ee8b21.pyi @@ -0,0 +1,11 @@ +""" +Auto-generated Python stub file from this.py +Module: this +""" + + +import stdio, t, c + +s: t.CExtern | str + +def this() -> t.CInt: pass diff --git a/thistest/temp/f71cf01f35e2325a.doc.json b/thistest/temp/f71cf01f35e2325a.doc.json new file mode 100644 index 0000000..9e26dfe --- /dev/null +++ b/thistest/temp/f71cf01f35e2325a.doc.json @@ -0,0 +1 @@ +{} \ No newline at end of file diff --git a/thistest/temp/f71cf01f35e2325a.pyi b/thistest/temp/f71cf01f35e2325a.pyi new file mode 100644 index 0000000..d500571 --- /dev/null +++ b/thistest/temp/f71cf01f35e2325a.pyi @@ -0,0 +1,13 @@ +""" +Auto-generated Python stub file from benchmark_main.py +Module: benchmark_main +""" + + +import w32.win32console as w32cmd +import t, c + +pagecode: t.CDefine = 65001 + +@t.CExport +def main() -> int: pass diff --git a/tpc_profile.prof b/tpc_profile.prof new file mode 100644 index 0000000..0ebeab3 Binary files /dev/null and b/tpc_profile.prof differ diff --git a/tpc_profile2.prof b/tpc_profile2.prof new file mode 100644 index 0000000..fe4362c Binary files /dev/null and b/tpc_profile2.prof differ diff --git a/vdb.md b/vdb.md new file mode 100644 index 0000000..ce9df08 --- /dev/null +++ b/vdb.md @@ -0,0 +1,714 @@ +# TransPyC 调试器（vdb）设计文档 + +> **状态**：设计阶段，暂不实现 +> **创建时间**：2026-06-26 +> **目标**：为 TransPyC 编译的程序提供源码级调试能力，支持断点、变量查看、单步、调用栈、异常栈跟踪 + +--- + +## 1. 背景与动机 + +TransPyC 将 Python 子集编译为 LLVM IR 再编译为原生机器码。当前编译产物**完全不含调试信息**（无 DWARF/PDB），运行时崩溃只能看到错误码和消息，无法定位源码行、查看变量、回溯调用栈。 + +本设计文档规划一个 TransPyC 专用调试器 **vdb**，目标是： + +- 源码级断点（`break main.py:30`） +- 变量查看（`print x` / `info locals`） +- 单步执行（`step` / `next`） +- 调用栈回溯（`backtrace`） +- 异常栈跟踪（`raise` 时记录栈帧） +- 硬错误捕获（segfault / 溢出） + +--- + +## 2. 现状分析 + +### 2.1 已有可复用资产 + +| 资产 | 位置 | 复用方式 | +|------|------|---------| +| 变量映射三件套 `variables`/`_direct_values`/`_reg_values` | `lib/core/LLVMCG/BaseGen.py:79-81` | 序列化到 .tpd 供变量查看 | +| 变量类型信息 `var_type_info`/`var_signedness`/`var_struct_class` | `lib/core/LLVMCG/BaseGen.py:82-112,124` | 序列化为 VarBinding.TypeDesc | +| 源码行号跟踪 `_current_lineno`/`_set_node_info`/`_get_source_line` | `lib/core/LLVMCG/BaseGen.py:126-127,138,297-440` | 扩展为 SourceMap | +| 类元信息表 `class_members`/`class_member_defaults`/`class_parent`/`Vtables` | `lib/core/LLVMCG/BaseGen.py:85-100` | 序列化为 TypeMeta | +| 符号表序列化 `SymbolTable.ToDict` | `lib/core/SymbolTable.py:265-319` | vdb 直接加载，无需重建 | +| 异常机制 `__eh_msg_out__`/`__eh_code_out__` | `lib/core/Handles/HandlesRaise.py` / `HandlesTry.py` / `HandlesExprCall.py` | 扩展携带栈信息 | +| VLogger `compile_error(file, line, code)` 三元组 | `lib/core/VLogger.py:286` | 复用错误位置格式 | +| 本地堆指针记录 `_local_heap_ptrs`/`_var_to_heap_ptr` | `lib/core/LLVMCG/MemoryOps.py:7-15` | 堆对象追踪起点 | + +### 2.2 关键缺口 + +| 能力 | 现状 | 缺口说明 | +|------|------|---------| +| DWARF 调试信息 | ❌ 完全未生成 | `lib/` 下无任何 `DIBuilder`/`!dbg`/`DICompileUnit` 调用 | +| PDB 调试信息 | ❌ 完全未生成 | Windows 下 WinDbg/VS 无法源码级调试 | +| 源码行 → IR 指令映射 | ❌ 无数据结构 | 最核心缺口，需从零构建 SourceMap | +| 变量名 → 运行时内存 | ⚠️ 仅编译期 ephemeral | `Gen.variables` 随 Gen 实例销毁，需持久化 | +| 类型元信息运行时查询 | ❌ 无 RTTI/反射 | `class_members` 等均编译期状态，未发射到二进制 | +| 运行时调用栈 | ❌ 完全无 | 异常仅传 `(code, msg)`，无栈帧链 | +| 硬错误处理 | ❌ 无 | `__builtin_trap` 仅声明未实现，segfault/SIGFPE 无捕获 | +| 断点/单步/变量查看 | ❌ 完全无 | `lib/` 下搜索 `breakpoint`/`step`/`watch` 零命中 | + +--- + +## 3. 路线选择 + +### 3.1 方案 A：纯 DWARF + GDB/LLDB + +**优点**：生态成熟，无需自研前端 +**缺点**： +- llvmlite 的 DIBuilder 支持度需验证 +- Python 语义与 C 不完全对应（`self` 是指针但 Python 视为对象） +- SHA1 命名导致符号名难读 +- 泛型特化类型（`Stack[int]`）在 DWARF 中表达困难 +- 跨模块符号（`6d71a13705da41ad.main`）可读性差 + +### 3.2 方案 B：纯自研 vdb + +**优点**：完全匹配 Python 语义，可显示原始源码，理解泛型/SHA1/跨模块 +**缺点**：开发量大，需自研调试协议与前端 + +### 3.3 方案 C：混合（推荐） + +- 生成轻量 DWARF（仅行号 + 变量名）让 GDB 可用作后备 +- 同时生成 TransPyC 专用调试元数据 `.tpd` 文件 +- vdb 前端优先使用 `.tpd`，可降级到 GDB 后端 + +**本设计采用方案 C**，但优先实现 `.tpd` + vdb，DWARF 作为可选增强。 + +--- + +## 4. 核心数据结构 + +### 4.1 .tpd 文件格式（TransPyC Debug） + +每个 `.ll` 文件配套生成一个 `.tpd` 文件（JSON 格式），包含以下结构： + +```python +from dataclasses import dataclass, field +from typing import Any + +@dataclass +class IrLocation: + """单条 IR 指令的源码位置映射""" + FuncName: str # mangled 函数名（含 SHA1 头，如 '6d71a13705da41ad.main'） + BlockName: str # 基本块名（如 'entry'、'if.then'） + InstrIdx: int # 指令在块中的索引 + SourceFile: str # 源码相对路径 + LineNo: int # 源码行号 + ColNo: int # 列号 + +@dataclass +class VarBinding: + """变量绑定信息（作用域内）""" + VarName: str # Python 变量名（如 'x'、'self'） + LlvmName: str # alloca 名或 SSA（如 '%x'、'%".1"'） + TypeDesc: str # 类型描述（'i32' / 'Stack[int]*' / 'i8*'） + IsSigned: bool # 是否有符号整数 + StructClass: str | None # 所属结构体类名（如 'Stack[int]'） + StartLine: int # 作用域起始行 + EndLine: int # 作用域结束行 + Storage: str # 'stack' | 'register' | 'direct' | 'global' + StackOffset: int | None # 栈偏移（仅 'stack'） + +@dataclass +class TypeMeta: + """类型元数据""" + TypeName: str # 'Stack[int]' / 'IntPair' / 'memory_block' + Members: list[tuple[str, str, int]] # (成员名, 类型, 字节偏移) + Size: int # 总大小（字节） + Align: int # 对齐 + Parent: str | None # 父类名 + IsException: bool # 是否异常类 + ExceptionCode: int # 异常码（仅 IsException=True） + HasVTable: bool # 是否有虚表 + IsPacked: bool # 是否紧凑布局 + +@dataclass +class FuncSig: + """函数签名""" + FuncName: str # mangled 名 + RawName: str # 原始 Python 名（去 SHA1） + ReturnType: str # 返回类型描述 + ParamNames: list[str] + ParamTypes: list[str] + IsExport: bool # 是否导出（@t.CExport） + IsMethod: bool # 是否实例方法 + ClassName: str | None # 所属类名（方法） + SourceFile: str + StartLine: int + EndLine: int + HasException: bool # 是否可能抛异常（有 __eh_msg_out__） + +@dataclass +class TpdFile: + """完整的 .tpd 调试信息文件""" + SourceMap: list[IrLocation] + VarScopes: dict[str, list[VarBinding]] # FuncName -> bindings + TypeMeta: dict[str, TypeMeta] + FuncSignatures: dict[str, FuncSig] + ExceptionCodes: dict[int, str] # code -> exception name + SourceFiles: dict[str, list[str]] # file -> 源码行列表（供 vdb 显示上下文） + ModuleSha1: str | None + Version: str = "1.0" +``` + +### 4.2 序列化格式 + +`.tpd` 文件使用 JSON，字段名采用 PascalCase（与代码一致）。示例： + +```json +{ + "Version": "1.0", + "ModuleSha1": "6d71a13705da41ad", + "SourceFiles": { + "App/main.py": ["import t", "import stdio", ...] + }, + "SourceMap": [ + {"FuncName": "main", "BlockName": "entry", "InstrIdx": 0, "SourceFile": "App/main.py", "LineNo": 255, "ColNo": 0}, + ... + ], + "VarScopes": { + "6d71a13705da41ad.test_generic_stack": [ + {"VarName": "s", "LlvmName": "%s", "TypeDesc": "Stack[int]*", "IsSigned": false, "StructClass": "Stack[int]", "StartLine": 66, "EndLine": 73, "Storage": "stack", "StackOffset": 0} + ] + }, + "TypeMeta": { + "Stack[int]": { + "TypeName": "Stack[int]", + "Members": [["data", "[16 x i32]", 0], ["top", "i32", 64]], + "Size": 68, + "Align": 4, + "Parent": null, + "IsException": false, + "ExceptionCode": 0, + "HasVTable": false, + "IsPacked": false + } + }, + "ExceptionCodes": { + "1": "ValueError", + "2": "TypeError", + "99": "Exception", + "100": "MyCustomError" + } +} +``` + +--- + +## 5. 实现阶段 + +### 5.1 Phase 1：源码级断点（最小可用） + +**目标**：`break main.py:30` + `continue` + `run` + +#### 5.1.1 编译器改动 + +**位置**：`lib/core/LLVMCG/BaseGen.py` + +1. **扩展 `_set_node_info`**：记录当前指令的源码位置到 `SourceMap` + +```python +def _set_node_info(self, Node: ast.AST) -> None: + self._current_lineno = getattr(Node, 'lineno', 0) + self._current_col = getattr(Node, 'col_offset', 0) + self._current_node_info = f" [line {self._current_lineno}, col {self._current_col}]" + # 新增：记录到 SourceMap + if self.func and self.builder and self.builder.block: + loc: IrLocation = IrLocation( + FuncName=self.func.name, + BlockName=self.builder.block.name, + InstrIdx=len(self.builder.block.instructions), + SourceFile=self._current_source_file or '', + LineNo=self._current_lineno, + ColNo=self._current_col + ) + self._source_map.append(loc) +``` + +2. **插入断点桩**：在每行源码对应的首条 IR 指令前插入 + +```llvm +; 运行时断点桩声明 +declare void @__tpc_breakpoint(i32 %line, i8* %file) + +; 在 main.py:255 的首条指令前插入 +%src_file_ptr = getelementptr [12 x i8], [12 x i8]* @.src.main_py, i32 0, i32 0 +call void @__tpc_breakpoint(i32 255, i8* %src_file_ptr) +``` + +3. **条件性插入**：仅当 `--debug` 编译选项启用时插入，release 构建完全剥离 + +4. **生成 .tpd 文件**：函数编译结束时，序列化 `SourceMap` 到 `.tpd` + +#### 5.1.2 运行时库 + +新建 `includes/vdb_runtime.py`： + +```python +import t +import c + +# 全局断点表（行号 -> 是否启用） +_Breakpoints: t.CArray[t.CInt, 65536] | t.CPtr = [0] +# 当前调试器状态（0=运行, 1=暂停） +_DebugState: t.CInt = 0 +# 当前源码文件名（用于断点匹配） +_CurrentFile: t.CArray[t.CInt8T, 256] | t.CPtr = [0] + +@t.CExport +def __tpc_breakpoint(line: t.CInt, file: t.CPtr) -> None: + """断点桩入口，由编译器在每个源码行前插入调用""" + # 检查是否启用断点 + if _Breakpoints[line] == 0: + return + # 命中断点，陷入调试器循环 + _DebugState = 1 + _tpc_debug_loop(line, file) + +def _tpc_debug_loop(line: t.CInt, file: t.CPtr) -> None: + """调试器主循环，等待用户命令""" + # 通过 IPC 或 stdio 与 vdb 前端通信 + # 解析命令：continue / step / next / print / info / backtrace + while _DebugState == 1: + cmd: t.CArray[t.CInt8T, 256] | t.CPtr = [0] + stdio.scanf("%s", cmd) + # ... 命令分派 +``` + +#### 5.1.3 vdb 前端 + +新建 `lib/vdb/` 目录： + +``` +lib/vdb/ +├── __init__.py +├── TpdLoader.py # .tpd 文件加载器 +├── BreakpointManager.py # 断点管理 +├── DebugSession.py # 调试会话主循环 +├── CommandParser.py # 命令解析 +└── SourceView.py # 源码显示 +``` + +**核心命令**： +- `break :` / `b :` — 设置断点 +- `delete ` / `d ` — 删除断点 +- `continue` / `c` — 继续执行 +- `run` / `r` — 启动程序 +- `list` / `l` — 显示当前源码上下文 +- `info breakpoints` — 列出断点 + +### 5.2 Phase 2：变量查看 + +**目标**：`print x` / `info locals` / `info args` + +#### 5.2.1 编译器改动 + +**位置**：`lib/core/LLVMCG/FuncGen.py` 或 `lib/core/Handles/HandlesFunctions.py` + +函数编译结束时，序列化变量绑定： + +```python +def _serialize_var_scope(self, FuncName: str) -> list[VarBinding]: + bindings: list[VarBinding] = [] + for var_name, llvm_val in self.variables.items(): + if llvm_val is None: + continue + binding: VarBinding = VarBinding( + VarName=var_name, + LlvmName=llvm_val.name if hasattr(llvm_val, 'name') else '', + TypeDesc=self._describe_llvm_type(llvm_val.type), + IsSigned=self.var_signedness.get(var_name, True), + StructClass=self.var_struct_class.get(var_name), + StartLine=self._func_start_line, + EndLine=self._func_end_line, + Storage='stack', + StackOffset=self._get_alloca_offset(llvm_val) # 新增：记录栈偏移 + ) + bindings.append(binding) + # 同样处理 _direct_values 和全局变量 + return bindings +``` + +#### 5.2.2 运行时变量读取 + +调试器通过以下方式读取变量值： + +1. **栈变量**：通过 `StackOffset` + 当前栈帧基址读取 +2. **全局变量**：通过符号表查找全局符号地址 +3. **寄存器变量**：需调试器理解 SSA 值（较复杂，Phase 2 可跳过，仅支持 stack） + +#### 5.2.3 类型渲染 + +```python +def RenderValue(llvm_val: bytes, type_meta: TypeMeta, type_desc: str) -> str: + """根据类型渲染变量值""" + if type_desc == 'i32': + return str(int.from_bytes(llvm_val, 'little', signed=True)) + elif type_desc == 'i8*': + # 尝试作为 C 字符串读取 + return f'"{read_cstring(llvm_val)}"' + elif type_desc.endswith('*') and type_meta: + # 结构体指针：展开成员 + return RenderStruct(llvm_val, type_meta) + ... +``` + +### 5.3 Phase 3：单步执行 + +**目标**：`step` / `next` / `finish` + +基于 SourceMap 实现： + +- `step`（步入）：在下一行源码的首条 IR 指令前设临时断点，包括被调用函数内部 +- `next`（步过）：在当前函数的下一行设临时断点，跳过函数调用 +- `finish`（步出）：在当前函数的返回点设临时断点 + +**实现**： +1. 查 SourceMap 找到当前行的下一个源码行 +2. 设置临时断点（一次性） +3. `continue` +4. 命中后自动删除临时断点 + +### 5.4 Phase 4：调用栈跟踪 + +**目标**：`backtrace` / `bt` / `frame ` + +#### 5.4.1 编译器改动 + +每个函数入口/出口插入栈帧维护调用： + +```llvm +; 函数入口 +call void @__tpc_push_frame( + i8* getelementptr([N x i8], [N x i8]* @.func.name, i32 0, i32 0), ; 函数名 + i8* getelementptr([M x i8], [M x i8]* @.src.file, i32 0, i32 0), ; 源文件 + i32 255 ; 起始行 +) + +; 函数出口（每个 ret 前） +call void @__tpc_pop_frame() +``` + +#### 5.4.2 运行时栈结构 + +```python +import t + +@t.CStruct +class FrameInfo: + FuncName: t.CPtr # i8* 函数名 + SrcFile: t.CPtr # i8* 源文件 + LineNo: t.CInt # 当前行号（动态更新） + PrevFrame: 'FrameInfo' | t.CPtr # 上一帧（链表） + +# 全局栈顶 +_FrameTop: FrameInfo | t.CPtr = t.CPtr(0) + +@t.CExport +def __tpc_push_frame(func_name: t.CPtr, src_file: t.CPtr, line: t.CInt) -> None: + frame: FrameInfo | t.CPtr = c.cast(FrameInfo, mb.alloc(FrameInfo.__sizeof__())) + frame.FuncName = func_name + frame.SrcFile = src_file + frame.LineNo = line + frame.PrevFrame = _FrameTop + _FrameTop = frame + +@t.CExport +def __tpc_pop_frame() -> None: + if _FrameTop != t.CPtr(0): + old: FrameInfo | t.CPtr = _FrameTop + _FrameTop = _FrameTop.PrevFrame + mb.free(old) +``` + +#### 5.4.3 断点桩更新当前行 + +`__tpc_breakpoint` 命中时，更新 `_FrameTop.LineNo`： + +```python +@t.CExport +def __tpc_breakpoint(line: t.CInt, file: t.CPtr) -> None: + if _Breakpoints[line] == 0: + return + if _FrameTop != t.CPtr(0): + _FrameTop.LineNo = line + _DebugState = 1 + _tpc_debug_loop(line, file) +``` + +### 5.5 Phase 5：异常栈跟踪 + +**目标**：`raise ValueError("msg")` 时输出完整调用栈 + +#### 5.5.1 编译器改动 + +**位置**：`lib/core/Handles/HandlesRaise.py` + +在 `_HandleRaiseLlvm` 中，抛异常前捕获栈： + +```python +def _HandleRaiseLlvm(self, Node: ast.Raise) -> None: + # ... 原有错误码计算 ... + # 新增：捕获调用栈 + capture_call = self.Gen.builder.call( + self._get_or_declare_function('__tpc_capture_trace', ir.FunctionType(ir.VoidType(), [])), + [] + ) + # ... 原有异常传播逻辑 ... +``` + +#### 5.5.2 异常结构扩展 + +将 `__eh_msg_out__`（i8**）扩展为指向包含栈的结构： + +```python +import t + +@t.CStruct +class ExceptionInfo: + Code: t.CInt + Message: t.CArray[t.CInt8T, 256] | t.CPtr + TraceDepth: t.CInt + TraceFuncs: t.CArray[t.CPtr, 64] # 函数名指针数组 + TraceLines: t.CArray[t.CInt, 64] # 对应行号 + TraceFiles: t.CArray[t.CPtr, 64] # 对应源文件 +``` + +#### 5.5.3 未捕获异常处理 + +在 `main` 函数返回前检查异常，打印栈： + +```python +def main() -> t.CInt: + # ... 用户代码 ... + # 编译器自动插入：检查未捕获异常 + if _UnhandledException != t.CPtr(0): + _tpc_print_trace(_UnhandledException) + return 1 + return testcheck.end() +``` + +### 5.6 Phase 6：硬错误捕获 + +**目标**：segfault / 除零 / 栈溢出时进入调试器 + +#### 5.6.1 信号处理器注册 + +```python +import t +import c + +@t.CExport +def __tpc_signal_handler(sig: t.CInt) -> None: + """OS 信号处理入口""" + __tpc_capture_trace() + # 打印信号类型 + if sig == 11: # SIGSEGV + stdio.printf("Segmentation fault\n") + elif sig == 8: # SIGFPE + stdio.printf("Arithmetic exception\n") + # 陷入调试器 + _tpc_debug_loop(0, t.CPtr(0)) + +def __tpc_install_signal_handlers() -> None: + """在 main 入口前自动调用""" + c.signal(11, __tpc_signal_handler) # SIGSEGV + c.signal(8, __tpc_signal_handler) # SIGFPE +``` + +#### 5.6.2 编译器改动 + +在 `main` 函数入口自动插入 `__tpc_install_signal_handlers()` 调用（仅 `--debug` 模式）。 + +--- + +## 6. vdb 命令参考 + +### 6.1 启动与控制 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `run [args]` | `r` | 启动被调试程序 | +| `continue` | `c` | 继续执行至下一断点 | +| `step` | `s` | 单步进入函数 | +| `next` | `n` | 单步跳过函数 | +| `finish` | `f` | 执行至当前函数返回 | +| `quit` | `q` | 退出调试器 | + +### 6.2 断点管理 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `break :` | `b` | 设置源码行断点 | +| `break ` | `b` | 设置函数入口断点 | +| `tbreak :` | `tb` | 临时断点（命中后删除） | +| `delete ` | `d` | 删除断点 n | +| `info breakpoints` | `i b` | 列出所有断点 | + +### 6.3 变量查看 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `print ` | `p` | 打印表达式值 | +| `info locals` | `i lo` | 列出当前局部变量 | +| `info args` | `i ar` | 列出函数参数 | +| `display ` | - | 每次暂停自动显示 | +| `set =` | - |修改变量值 | + +### 6.4 栈操作 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `backtrace` | `bt` | 显示调用栈 | +| `frame ` | `fr` | 切换到第 n 帧 | +| `up` | - | 向上切换一帧 | +| `down` | `do` | 向下切换一帧 | + +### 6.5 源码浏览 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `list` | `l` | 显示当前行上下文 | +| `list :` | `l` | 显示指定位置 | +| `info source` | `i so` | 显示当前源文件信息 | + +### 6.6 类型与结构 + +| 命令 | 缩写 | 说明 | +|------|------|------| +| `ptype ` | `pt` | 显示类型定义 | +| `info types` | `i ty` | 列出所有类型 | +| `info exceptions` | `i ex` | 列出异常码表 | + +--- + +## 7. 实现优先级 + +| 优先级 | 阶段 | 价值 | 改动量 | +|--------|------|------|--------| +| P0 | Phase 1（源码断点）+ Phase 2（变量查看） | 80% 调试价值 | 小 | +| P1 | Phase 4（调用栈） | 崩溃定位关键 | 中 | +| P2 | Phase 5（异常栈） | Python 语义匹配 | 中 | +| P3 | Phase 3（单步） | 体验提升 | 中（依赖 Phase 1） | +| P4 | Phase 6（硬错误） | 兜底保护 | 中 | +| P5 | DWARF 生成 | GDB 互操作 | 大（可选） | + +--- + +## 8. 风险与考量 + +### 8.1 性能开销 + +- **断点桩**：仅 `--debug` 构建启用，release 完全剥离 +- **IR 膨胀**：每行插入断点桩约增加 10-20% IR 体积，可接受 +- **运行时开销**：断点桩检查为单次数组访问 + 分支预测，开销极低 +- **栈帧维护**：每函数调用增加 2 次 `push/pop_frame`，约 5-10% 调用开销 + +### 8.2 兼容性 + +- **SHA1 命名**：vdb 需理解 SHA1 前缀，通过 `ModuleSha1` 字段还原为 `模块名.函数名` 显示 +- **泛型特化**：`Stack[int]` 需作为完整类型名处理，TypeMeta 中保留特化名 +- **跨模块**：`.tpd` 文件按模块生成，vdb 启动时合并所有相关 `.tpd` +- **`main` 特化**：`main` 函数无 SHA1 头，vdb 需特殊处理 + +### 8.3 安全性 + +- **栈帧分配**：使用 `mb.alloc` 而非 `c.malloc`，确保与 TransPyC 内存管理一致 +- **信号处理器**：异步信号中避免调用非异步信号安全函数，仅设置标志位 + +### 8.4 限制 + +- **优化构建**：高优化级别（`-O2`+）下变量可能被优化掉，`.tpd` 中的 VarBinding 可能失效 +- **内联函数**：内联后栈帧信息丢失，需在 `.tpd` 中标记 `IsInline` +- **尾调用优化**：破坏调用栈链，需禁用或特殊处理 + +--- + +## 9. 文件结构规划 + +``` +lib/vdb/ # vdb 调试器前端 +├── __init__.py +├── TpdLoader.py # .tpd 文件加载与合并 +├── BreakpointManager.py # 断点管理 +├── DebugSession.py # 调试会话主循环 +├── CommandParser.py # 命令解析 +├── SourceView.py # 源码显示 +├── ValueRenderer.py # 变量值渲染 +├── StackWalker.py # 调用栈遍历 +└── IpcChannel.py # 与被调试进程的 IPC 通信 + +lib/core/LLVMCG/ +└── DebugInfoGen.py # 新增：.tpd 生成器（编译器侧） + +includes/ +└── vdb_runtime.py # 运行时断点/栈帧/信号处理库 +``` + +--- + +## 10. 与现有系统的集成点 + +### 10.1 编译器集成 + +- **`Phase2Translator.py`**：在生成 `.ll` 后，调用 `DebugInfoGen.GenerateTpd(module, gen, output_path)` +- **`BaseGen.py`**：`_set_node_info` 扩展记录 SourceMap；函数编译结束序列化 VarScope +- **`HandlesFunctions.py`**：函数入口/出口插入 `__tpc_push_frame`/`__tpc_pop_frame`（仅 debug 模式） +- **`HandlesRaise.py`**：`_HandleRaiseLlvm` 插入 `__tpc_capture_trace` + +### 10.2 project.json 配置 + +新增 `debug` 选项： + +```json +{ + "options": { + "debug": true, + "debug_info": "tpd" + } +} +``` + +- `debug: true` — 启用调试构建（插入断点桩、栈帧维护、信号处理） +- `debug_info: "tpd"` — 生成 .tpd 文件（默认） +- `debug_info: "dwarf"` — 生成 DWARF（未来支持） +- `debug_info: "both"` — 同时生成（未来支持） + +### 10.3 命令行集成 + +```bash +# 编译带调试信息 +transpyc build --debug + +# 直接启动调试器 +transpyc debug ./output/MyApp.exe +# 等价于 +vdb ./output/MyApp.exe +``` + +--- + +## 11. 开放问题 + +1. **llvmlite DIBuilder 支持度**：需验证 llvmlite 是否暴露 DIBuilder 接口，决定 DWARF 生成可行性 +2. **IPC 协议**：vdb 前端与被调试进程的通信方式（管道/Socket/共享内存）待定 +3. **远程调试**：是否支持远程调试（如 ViperOS 裸机目标）需进一步设计 +4. **表达式求值**：`print x.y.z` 等复杂表达式是否需要完整 Python 解释器，还是简化为路径访问 +5. **多线程**：TransPyC 当前是否支持多线程，调试器如何处理线程切换 + +--- + +## 12. 参考文档 + +- 现状分析：基于 2026-06-26 对 `lib/core/` 和 `lib/Projectrans/` 的完整搜索 +- 异常机制：`wiki/09-exceptions.md` +- 导入机制：`wiki/10-imports.md` +- 符号表：`lib/core/SymbolTable.py` +- 类型系统：`lib/core/TypeSpec.py` / `lib/core/Handles/HandlesBase.py` +- LLVM 生成：`lib/core/LLVMCG/BaseGen.py` / `FuncGen.py` / `MemoryOps.py` + +--- + +> **本文档为设计阶段产物，暂不实现。待社区需求或开发优先级提升后，按 Phase 1 → 2 → 4 → 5 → 3 → 6 顺序推进。** diff --git a/wiki/01-overview.md b/wiki/01-overview.md new file mode 100644 index 0000000..efec04b --- /dev/null +++ b/wiki/01-overview.md @@ -0,0 +1,357 @@ +# 01 - 语言概述与编译流程 + +## 语言定位 + +`Viper` 是一种 **系统级编程语言**，其语法基于 `Python`，但编译目标为 `LLVM IR`，最终生成原生机器码。`Viper` 不是 `Python` 的超集或子集，也不是"`C` 的语法糖"——它是一门拥有独立类型系统、独立编译模型、独立模块体系的语言。`Viper` 选择 `Python` 语法作为表达形式，是因为 `Python` 是人类公认的可读性最高的语言。 + +当前编译器 `TransPyC`（用 Python 实现）使用 Python 标准库 `ast` 解析源文件，将精力集中在语义翻译而非词法/语法分析上。但项目已在 [includes/ast](../includes/ast) 中实现**完全自研的 Python 解析器**（词法器 + 语法器 + AST 构造，约 4176 LOC，`AstTest` 综合测试通过），覆盖 CPython 几乎全部语法。这是向 `TransPyV`（用 Viper 重写编译器自身）完全自举的关键准备——自举后编译器将脱离 Python 运行时，解析、类型检查、IR 生成都由原生机器码完成。详见 [13-bootstrapping.md](13-bootstrapping.md)。 + +而实际上，下文中所有提到的 `C` 语言都是为了便于使用 `C` 的底层开发者理解。由于历史原因，在旧版本中我们使用 `C` 来作为中间语言，如同旧版本的 `C++` 一样，但现在 `target="c"` 的路径已经完全删除，`C` 路径永久不再受维护。 + +### 核心设计决策 + +1. **Python 语法，LLVM 语义**：源文件是完全合法的 `Python` 语法，但通过 `t` 模块的类型注解系统赋予完全不同的语义 —— `t.CInt` 不是 `Python` 的 `int`，而是 `LLVM` 的 `i32` +2. **类型注解即编译指令**：Viper 的类型注解不是可选的提示，而是编译器生成 `LLVM IR` 的决定性依据。`x: t.CInt` 生成 `i32`，`x: t.CInt | t.CPtr` 生成 `i32*`，但无注解时会自动推导，详见下文。 +3. **两阶段编译**：先从源文件提取声明接口（`.pyi` + `.stub.ll`），再使用声明接口翻译源文件为含代码的 `.ll`。这是 `Viper` 区别于简单"`Python` 转 `C`"工具的关键架构 +4. **`SHA1` 命名空间**：每个源文件按内容 `SHA1` 哈希命名，非导出函数和结构体自动加上 `SHA1` 前缀，从根本上消除跨模块的符号冲突 +5. **万物皆数据**：为便于底层开发，实际上所有的变量和类型一般并不适用于鸭子类型，但在语义层面我们会尽可能贴近鸭子类型。 + +### 与 `Python` 的关系 + +| 特性 | `Python` | `Viper` | +|------|--------|-------| +| 类型系统 | 动态类型 | 静态类型（通过 `t` 模块注解，编译时确定） | +| 内存管理 | `GC` 自动回收 | 手动管理，无 `GC`，无运行时 | +| 运行时 | 解释执行 + 字节码 | 编译为原生机器码（通过 `LLVM`） | +| 对象模型 | 万物皆对象 | `@t.Object` 启用 OOP，`@t.CVTable` 为多态（vtable），`@t.NoVTable` 为非多态继承（C++ 风格，零运行时开销），普通 `class` 仅为结构体布局 | +| 标准库 | `Python` 标准库 | `Viper` 标准库（`includes/`）+ `ViperOS SDK` | +| 空值 | `None` | `None`（编译为 `NULL`/零值） | +| 模块系统 | 运行时 `import` | 编译时解析，插入 `.stub.ll`，`SHA1` 命名空间隔离 | + +### 与 C 的关系 + +Viper 编译后的代码在底层等价于 `C` 编译后的机器码（都经过 LLVM），但 Viper 在语言层面提供了 C 所没有的能力： + +- **SHA1 命名空间**：自动消除符号冲突，无需手动管理 `static`/命名前缀 +- **类型组合语法**：`t.CConst | t.CInt | t.CPtr` 比 `const int*` 更具组合性 +- **声明式内联汇编**：`c.Asm(f"mov {c.AsmOut(x, t.ASM_DESCR.OUTPUT_REG)}, rdi")` 比裸 `__asm__` 更安全，也更具可读性。 +- **结构化预处理**：`c.CIfdef`/`c.CEndif()` 替代 `#ifdef`/`#endif`。 +- **面向对象**：`@t.Object` + `@t.CVTable` 提供 `vtable` 支持的多态 `OOP`；`@t.NoVTable` 提供 C++ 风格的非多态继承（零 vtable 开销）；PEP 695 泛型 + 递归泛型继承实现强类型自引用结构 +- **存根驱动的模块系统**：`.pyi` 存根文件实现跨模块类型解析，无需头文件 +- **更多内容**：额外更多的不依赖操作系统的函数和语法 + +## 两阶段编译流程 + +Viper 的编译由 `Projectrans.py` 驱动，分为两个阶段。这是 Viper 编译模型的核心，理解两阶段编译是理解 `t.CDefine`、`t.CExport`、`t.CInline` 等关键概念的前提。 + +``` +源文件 (.py) ────────────────────────────────────────────────────── + │ │ + │ ┌─────────────── 阶段一：声明提取 ───────────────┐ │ + │ │ │ │ + │ │ 1. 计算源文件 SHA1 │ │ + │ │ 2. 生成 .pyi（签名存根） │ │ + │ │ 3. 构建结构体注册表 │ │ + │ │ 4. 生成 .stub.ll（LLVM IR 声明） │ │ + │ │ │ │ + │ └────────────────────────────────────────────────┘ │ + │ │ + │ ┌─────────────── 阶段二：代码翻译 ───────────────┐ │ + │ │ │ │ + │ │ 1. 加载所有 .pyi 和 .stub.ll │ │ + │ │ 2. 构建共享符号表 │ │ + │ │ 3. 收集内联函数符号 │ │ + │ │ 4. 翻译源文件 → .ll（含代码） │ │ + │ │ 5. llc 编译 → .o（目标文件） │ │ + │ │ 6. ld.lld 链接 → 可执行文件 │ │ + │ │ │ │ + │ └────────────────────────────────────────────────┘ │ + │ │ + └──────────────────────────────────────────────────────────────┘ +``` + +### 阶段一：声明提取（Phase1Generator） + +阶段一的目标是从每个源文件提取**声明接口**，使其他模块在阶段二翻译时能够正确解析跨模块引用。 + +#### 步骤 1：可达文件发现与拓扑排序 + +从入口文件（`main.py` 或 `project.json` 指定）出发，通过 `import` 语句递归遍历，找出所有可达的 `.py` 源文件。然后对文件进行拓扑排序，确保被依赖的模块先被处理。 + +``` +main.py → import serial → import drivers.serial.uart.serial → ... +``` + +#### 步骤 2：生成 `.pyi` 签名存根 + +对每个源文件，计算其内容的 SHA1 哈希（16位），然后调用 `PythonToStubConverter` 生成签名存根文件 `.pyi`。 + +**SHA1 计算方式**： +```python +sha1 = hashlib.sha1(content.encode('utf-8')).hexdigest()[:16] +``` + +**存根文件的内容规则**： + +| 源文件元素 | 存根文件中的表示 | +|-----------|----------------| +| `t.CDefine` 常量 | 保留原样（含赋值）：`MAX_SIZE: t.CDefine = 1024` | +| `t.CDefine` 函数（返回 `t.CDefine` 的函数） | 保留完整函数体（宏展开需要） | +| 普通函数 | 仅保留签名，添加 `| c.State`：`def foo(x: t.CInt) -> t.CVoid | c.State: pass` | +| 类定义 | 保留成员类型注解和方法签名 | +| 全局变量 | 添加 `t.CExtern`：`x: t.CExtern | t.CInt` | +| `c.CIfdef`/`c.CEndif()` 等预处理指令 | 保留原样 | +| `import` 语句 | 保留原样 | + +**关键设计**：`t.CDefine` 常量和 `t.CDefine` 函数在存根中**保留完整定义**（包括赋值和函数体），因为它们是编译时常量/宏，其他模块引用时需要完整展开。普通函数只保留签名并标记 `c.State`（仅声明），因为函数体在阶段二生成。 + + +#### 步骤 3：构建结构体注册表 + +扫描所有已生成的 `.pyi` 文件，提取： +- **结构体名称集合**：所有非枚举、非异常的 `class` 定义 +- **枚举名称集合**：继承 `t.CEnum` 的类 +- **异常名称集合**：继承 `Exception` 的类 +- **结构体→SHA1 映射**：记录每个结构体定义在哪个模块中 + +这个注册表用于阶段一的声明生成和阶段二的类型解析，确保跨模块的结构体引用使用正确的 SHA1 命名空间。 + +#### 步骤 4：生成 `.stub.ll` LLVM IR 声明 + +对每个 `.pyi` 文件，由 `DeclarationGenerator` 生成对应的 LLVM IR 声明文件 `.stub.ll`。 + +**声明生成规则**： + +| 元素 | 生成的 LLVM IR | +|------|---------------| +| 普通函数 | `declare void @".foo"(i32)` — 非 CExport 函数加 SHA1 前缀 | +| CExport 函数 | `declare i32 @main()` — CExport 函数不加前缀，全局可见 | +| 结构体 | `%".ClassName" = type { i32, i32* }` — 类型名加 SHA1 前缀 | +| 全局变量 | `@varname = external global i32` | +| `t.CDefine` 常量 | **不生成声明**（编译时常量，直接内联展开） | +| `t.CTypedef` 别名 | **不生成声明**（类型别名，在类型解析时展开） | +| 枚举 | 为每个枚举成员生成 `@__config_EnumName_member = external global i32` | + +**增量编译**：如果 `.pyi` 或 `.stub.ll` 已存在，则跳过生成（缓存命中）。只有源文件内容变化导致 SHA1 变化时才重新生成。 + +### 阶段二：代码翻译（Phase2Translator） + +阶段二使用阶段一生成的声明接口，将源文件翻译为包含完整代码的 LLVM IR。 + +#### 步骤 1：加载声明接口 + +加载 `temp/` 目录中所有的 `.pyi` 和 `.stub.ll` 文件，构建： +- `sig_files`：SHA1 → `.pyi` 文件路径映射 +- `stub_files`：SHA1 → `.stub.ll` 文件路径映射 +- `sha1_map`：SHA1 → 源文件相对路径映射 + +#### 步骤 2：构建共享符号表 + +一次性构建所有文件共享的符号表数据，避免每个文件重复加载。将所有 `.pyi` 存根和 `.stub.ll` 声明中的类型信息注册到统一的 `SymbolTable` 中。 + +#### 步骤 3：收集内联函数符号 + +扫描 `includes/` 目录中被引用的模块，收集标记为 `t.CInline` 的函数。内联函数的完整 AST body 被保存，在翻译调用点时直接展开。值得注意的是，和 C 语言不同，`t.CInline` 不是优化建议，而是强制性的。 + +#### 步骤 4：翻译源文件 + +对每个源文件，使用 `TransPyC` 翻译器将 Python AST 翻译为 LLVM IR： +1. 将对应模块的 `.stub.ll` 声明嵌入到生成的 `.ll` 文件头部 +2. 所有被引用模块的 `.stub.ll` 声明也被嵌入 +3. 翻译 AST 节点为 LLVM IR 指令 +4. 输出 `.ll` 文件 + +#### 步骤 5：编译与链接 + +- `llc`：将每个 `.ll` 编译为 `.o` 目标文件 +- `ld.lld`：将所有 `.o` 链接为最终可执行文件 + +## SHA1 命名空间机制 + +SHA1 命名空间是 Viper 解决跨模块符号冲突的核心机制。 + +### 问题 + +在 C 语言中，多个模块可能定义同名函数或结构体，导致链接时符号冲突。C 的解决方案是 `static`（限制可见性）和命名前缀（手动避免冲突）。 + +### Viper 的解决方案 + +Viper 使用源文件内容的 SHA1 哈希作为命名空间前缀： + +``` +源文件 A.py（SHA1 = a1b2c3d4e5f6g7h8）中定义： + class Point: x: t.CInt; y: t.CInt + def draw(p: Point) -> t.CVoid: ... + +源文件 B.py（SHA1 = i9j0k1l2m3n4o5p6）中也定义： + class Point: x: t.CFloat; y: t.CFloat # 不同的结构体！ + def draw(p: Point) -> t.CVoid: ... + +编译后： + A.py 的结构体 → %"a1b2c3d4e5f6g7h8.Point" = type { i32, i32 } + A.py 的函数 → declare void @"a1b2c3d4e5f6g7h8.draw"(%"a1b2c3d4e5f6g7h8.Point"*) + + B.py 的结构体 → %"i9j0k1l2m3n4o5p6.Point" = type { float, float } + B.py 的函数 → declare void @"i9j0k1l2m3n4o5p6.draw"(%"i9j0k1l2m3n4o5p6.Point"*) +``` + +**没有符号冲突**——即使两个模块定义了同名类型和函数，它们的 LLVM IR 符号也是不同的。即便有两个内容完全相同的文件也不会冲突，他们会被识别为同一个文件，然后进行单次编译。 + +### SHA1 前缀的豁免：`t.CExport` + +标记为 `t.CExport` 的函数**不加 SHA1 前缀**，保持原始函数名。这是模块向外部暴露 API/ABI 的机制： + +```python +# main.py — 入口函数，必须全局可见 +def main() -> t.CInt | t.CExport: + return 0 + +# 编译后：define i32 @main() — 无 SHA1 前缀 +``` + +```python +# serial.py — 驱动接口，对外暴露 +def init() -> t.CVoid | t.CExport: + serial_puts("init\n") + +# 编译后：define void @init() — 无 SHA1 前缀，其他模块可直接调用 +``` + +### SHA1 与增量编译 + +SHA1 同时服务于增量编译： +- 源文件内容不变 → SHA1 不变 → `.pyi` 和 `.stub.ll` 缓存命中，跳过阶段一 +- 源文件内容变化 → SHA1 变化 → 重新生成声明接口 +- `temp/_sha1_map.txt` 记录 SHA1 → 源文件路径的映射，供阶段二加载 + +理论上，如果一个未经变动的模块的依赖路径上存在变动的模块，暂时的方法是将其简单做 SHA1 替换，暨将旧 SHA1 替换为新 SHA1，但不适用于签名改变的情况，不过签名改变这个未经变动的模块必然也需要改变，但是由于宏展开的存在，不能完全保证不会发生错误情况。且依赖路径的 SHA1 替换成功性未经完整测试，可能出现未定义行为。 + +## `t.CDefine` 深度解析 + +`t.CDefine` 是 Viper 中最特殊的类型——它不是数据类型，而是**编译时元指令**，控制编译器的代码生成行为。 + +### `t.CDefine` 常量 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +CPU_FEATURE_FPU: t.CDefine = (1 << 0) +``` + +**编译行为**： +1. **阶段一**：存根生成器保留 `t.CDefine` 常量的完整定义（包括赋值值），因为其他模块可能引用此常量 +2. **阶段一**：`.stub.ll` 声明生成器**跳过** `t.CDefine` 常量，不生成 `external global` 声明 +3. **阶段二**：翻译器遇到 `t.CDefine` 常量的引用时，直接内联其值，不生成任何加载指令 + +这意味着 `t.CDefine` 常量在 LLVM IR 层面**不存在**——它们在编译时被完全展开，等价于 C 的 `#define` 宏。 + +### `t.CDefine` 函数 + +当函数的返回类型注解包含 `t.CDefine` 时，该函数被编译器视为**宏函数**： + +```python +def MAKE_FLAG(bit) -> t.CDefine: + return (1 << bit) + +def GET_PAGE_ORDER(size) -> t.CDefine: + return size // PAGE_SIZE +``` + +**编译行为**： +1. **阶段一**：存根生成器保留 `t.CDefine` 函数的**完整函数体**（不像普通函数那样只保留签名） +2. **阶段二**：翻译器遇到 `t.CDefine` 函数的调用时，将调用内联展开为函数体的计算结果 + +`t.CDefine` 函数在 LLVM IR 中**不生成函数定义**——它们是纯粹的编译时宏。 + +### `t.CDefine` 与类型组合 + +`t.CDefine` 可以与具体类型组合使用，为常量指定底层类型： + +```python +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +DATA_SEG: t.CDefine | t.CUInt16T = 0x10 +``` + +这表示常量在类型检查时被视为 `t.CUInt16T`（`uint16_t`），但在代码生成时仍然内联展开。 + +## `t.CExport` 深度解析 + +`t.CExport` 控制函数的符号可见性，是 SHA1 命名空间的豁免机制。 + +### 语义 + +| 修饰 | LLVM IR 函数名 | 链接可见性 | 用途 | +|------|---------------|-----------|------| +| 无修饰 | `@.funcname` | 模块内部 | 模块私有函数 | +| `t.CExport` | `@funcname` | 全局可见 | 对外暴露的 API | +| `t.CExtern` | `@funcname`（仅声明） | 外部定义 | 引用外部函数 | + +### 使用场景 + +```python +# 入口函数 — 必须全局可见，链接器需要找到它 +def _start() -> t.CInt | t.CExport: + return kernel_main() + +# 驱动接口 — 其他模块/应用需要调用 +def init() -> t.CVoid | t.CExport: + uart_init() + +# 内部辅助函数 — 不需要全局可见，自动加 SHA1 前缀 +def _helper() -> t.CInt: + return 42 +``` + +### `t.CExport` 在存根中的表现 + +在 `.pyi` 存根文件中，`t.CExport` 函数与普通函数一样只保留签名（添加 `c.State`，表示单纯声明），但 `t.CExport` 标记被保留在返回类型注解中。阶段一的 `DeclarationGenerator` 检查 `t.CExport` 标记来决定是否添加 SHA1 前缀。 + +## `t.CInline` 深度解析 + +`t.CInline` 标记函数为内联函数，编译器在调用点直接展开函数体。 + +### 语义 + +```python +def fast_add(a: t.CInt, b: t.CInt) -> t.CInt | t.CInline: + return a + b +``` + +**编译行为**： +1. **阶段二预收集**：`_collect_inline_symbols` 扫描 `includes/` 目录中被引用的模块，收集所有 `t.CInline` 函数的 AST body +2. **翻译时展开**：遇到内联函数调用时，将函数体的 AST 直接嵌入调用点，替换参数为实际值 +3. **不生成独立函数**：内联函数不生成 LLVM IR 函数定义（除非也被非内联调用） + +### `t.CInline` 与 `t.CDefine` 函数的区别 + +| 特性 | `t.CInline` | `t.CDefine` 函数 | +|------|------------|-----------------| +| 展开时机 | 阶段二翻译时 | 阶段二翻译时 | +| 类型检查 | 完整的参数和返回类型检查 | 返回类型为宏，类型检查较弱 | +| 存根表示 | 签名 + `c.State` | 完整函数体 | +| LLVM IR | 可能生成函数定义（如果有非内联调用） | 不生成函数定义 | +| 适用场景 | 性能关键的短函数 | 编译时常量和宏计算 | + +## 目标平台 + +默认目标三元组：`x86_64-none-elf` + +数据布局：`e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128` + +## Hello World 示例 + +```python +import t, c + +def main() -> t.CInt | t.CExport: + print("Hello, ViperOS!\n") + return 0 +``` + +编译流程： + +1. **阶段一**：计算 `main.py` 的 SHA1，生成 `.pyi` 存根（`def main() -> t.CInt | t.CExport | c.State: pass`），生成 `.stub.ll` 声明（`declare i32 @main()`，因为是 CExport 不加前缀） +2. **阶段二**：翻译 `main.py` 为 `.ll`，嵌入 `print` 对应的 `puts`/`printf` 声明，生成 `define i32 @main()` 函数体 +3. **编译**：`llc` 将 `.ll` 编译为 `.o` +4. **链接**：`ld.lld` 链接为可执行文件 diff --git a/wiki/02-type-system.md b/wiki/02-type-system.md new file mode 100644 index 0000000..d4bcaef --- /dev/null +++ b/wiki/02-type-system.md @@ -0,0 +1,665 @@ +# 02 - 类型系统 + +Viper 的类型系统通过 `t` 模块提供，所有类型注解继承 `t.CType` 。大都以 `t.C{TypeName:PascalCase}`为类型，类型注解是 Viper 的核心机制，决定了变量的内存布局、大小和对齐方式。 + +## 基本类型 + +### 整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CChar` | `char` | 8 | 是 | +| `t.CShort` | `short` | 16 | 是 | +| `t.CInt` | `int` | 32 | 是 | +| `t.CLong` | `long` | 64 | 是 | +| `t.CUnsignedChar` | `unsigned char` | 8 | 否 | +| `t.CUnsignedShort` | `unsigned short` | 16 | 否 | +| `t.CUnsigned` / `t.CUnsignedInt` | `unsigned int` | 32 | 否 | +| `t.CUnsignedLong` | `unsigned long` | 64 | 否 | +| `t.CSignedChar` | `signed char` | 8 | 是 | + +值得注意的是，t.CUnsigned 一般不单独使用，而是配合其它有符号类型，替换表示无符号，比如 `t.CUnsigned | t.CInt`，由于为了便于使用一些常用类型，则将其组合为常用的 `t.CUnsignedInt` 等。 + +### 固定宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CInt8T` | `int8_t` | 8 | 是 | +| `t.CInt16T` | `int16_t` | 16 | 是 | +| `t.CInt32T` | `int32_t` | 32 | 是 | +| `t.CInt64T` | `int64_t` | 64 | 是 | +| `t.CUInt8T` | `uint8_t` | 8 | 否 | +| `t.CUInt16T` | `uint16_t` | 16 | 否 | +| `t.CUInt32T` | `uint32_t` | 32 | 否 | +| `t.CUInt64T` | `uint64_t` | 64 | 否 | + +### 最小宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntLeast8T` | `int_least8_t` | ≥8 | 是 | +| `t.CIntLeast16T` | `int_least16_t` | ≥16 | 是 | +| `t.CIntLeast32T` | `int_least32_t` | ≥32 | 是 | +| `t.CIntLeast64T` | `int_least64_t` | ≥64 | 是 | +| `t.CUIntLeast8T` | `uint_least8_t` | ≥8 | 否 | +| `t.CUIntLeast16T` | `uint_least16_t` | ≥16 | 否 | +| `t.CUIntLeast32T` | `uint_least32_t` | ≥32 | 否 | +| `t.CUIntLeast64T` | `uint_least64_t` | ≥64 | 否 | + +### 最快最小宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntFast8T` | `int_fast8_t` | ≥8 | 是 | +| `t.CIntFast16T` | `int_fast16_t` | ≥16 | 是 | +| `t.CIntFast32T` | `int_fast32_t` | ≥32 | 是 | +| `t.CIntFast64T` | `int_fast64_t` | ≥64 | 是 | +| `t.CUIntFast8T` | `uint_fast8_t` | ≥8 | 否 | +| `t.CUIntFast16T` | `uint_fast16_t` | ≥16 | 否 | +| `t.CUIntFast32T` | `uint_fast32_t` | ≥32 | 否 | +| `t.CUIntFast64T` | `uint_fast64_t` | ≥64 | 否 | + +### 最大宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntMaxT` | `intmax_t` | ≥64 | 是 | +| `t.CUIntMaxT` | `uintmax_t` | ≥64 | 否 | + +### stdint 短名别名 + +通过 `import stdint` 引入，这些短名是上述类型的 `t.CTypedef` 别名，便于在系统编程场景中快速使用。允许直接使用 `from stdint import *` 来直接使用。 + +#### 基本类型短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.INT` | `t.CInt` | - | +| `stdint.UINT` | `t.CUnsignedInt` | - | +| `stdint.BOOL` | `t.CInt` | 1 为真，0 为假 | +| `stdint.SHORT` | `t.CShort` | - | +| `stdint.USHORT` | `t.CUnsignedShort` | - | +| `stdint.LONG` | `t.CLong` | - | +| `stdint.ULONG` | `t.CUnsignedLong` | - | +| `stdint.LONGLONG` | `t.CLong \| t.CLong` | - | +| `stdint.ULONGLONG` | `t.CUnsignedLong \| t.CLong` | - | +| `stdint.FLOAT` | `t.CFloat` | - | +| `stdint.DOUBLE` | `t.CDouble` | - | + +#### 固定宽度短名 + +| 短名 | 等价展开 | +|------|---------| +| `stdint.INT8` | `t.CInt8T` | +| `stdint.INT16` | `t.CInt16T` | +| `stdint.INT32` | `t.CInt32T` | +| `stdint.INT64` | `t.CInt64T` | +| `stdint.UINT8` | `t.CUInt8T` | +| `stdint.UINT16` | `t.CUInt16T` | +| `stdint.UINT32` | `t.CUInt32T` | +| `stdint.UINT64` | `t.CUInt64T` | + +#### 字节与字短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.BYTE` | `t.CUnsignedChar` | 8 位无符号 | +| `stdint.WORD` | `t.CUInt16T` | 16 位无符号 | +| `stdint.DWORD` | `t.CUInt32T` | 32 位无符号 | +| `stdint.QWORD` | `t.CUInt64T` | 64 位无符号 | + +#### 字符类型短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.TCHAR` | `t.CChar` | - | +| `stdint.CHARLIST` | `str \| t.CPtr` / `list[str, None]` | - | +| `stdint.WCHAR` | `stdint.WORD` | UTF-16 代码单元 | +| `stdint.CHAR8` | `t.CChar8T` | - | +| `stdint.CHAR16` | `t.CChar16T` | - | +| `stdint.CHAR32` | `t.CChar32T` | - | + +#### 指针短名 + +| 短名 | 等价展开 | +|------|---------| +| `stdint.INTPTR` | `t.CInt \| t.CPtr` | +| `stdint.UINTPTR` | `t.CUnsignedInt \| t.CPtr` | +| `stdint.BYTEPTR` | `stdint.BYTE \| t.CPtr` | +| `stdint.SHORTPTR` | `t.CShort \| t.CPtr` | +| `stdint.USHORTPTR` | `t.CUnsignedShort \| t.CPtr` | +| `stdint.WCHARPTR` | `stdint.WORD \| t.CPtr` | +| `stdint.VOIDPTR` | `t.CVoid \| t.CPtr` | +| `stdint.INT8PTR` | `t.CInt8T \| t.CPtr` | +| `stdint.INT16PTR` | `t.CInt16T \| t.CPtr` | +| `stdint.INT32PTR` | `t.CInt32T \| t.CPtr` | +| `stdint.INT64PTR` | `t.CInt64T \| t.CPtr` | +| `stdint.UINT8PTR` | `t.CUInt8T \| t.CPtr` | +| `stdint.UINT16PTR` | `t.CUInt16T \| t.CPtr` | +| `stdint.UINT32PTR` | `t.CUInt32T \| t.CPtr` | +| `stdint.UINT64PTR` | `t.CUInt64T \| t.CPtr` | +| `stdint.CHAR8PTR` | `t.CChar8T \| t.CPtr` | +| `stdint.CHAR16PTR` | `t.CChar16T \| t.CPtr` | +| `stdint.CHAR32PTR` | `t.CChar32T \| t.CPtr` | + +#### 平台特定短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.FSIZE_t` | `stdint.DWORD` | 文件大小 | +| `stdint.LBA_t` | `stdint.DWORD` | 逻辑块地址 | + +### 平台相关类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CSizeT` | `size_t` | 64 | 否 | +| `t.CIntPtrT` | `intptr_t` | 64 | 是 | +| `t.CUIntPtrT` | `uintptr_t` | 64 | 否 | +| `t.CPtrDiffT` | `ptrdiff_t` | 64 | 是 | + +### stdint 短名别名 + +Viper 在 `stdint` 模块中提供了更简短的类型别名，通过 `t.CTypedef` 映射到对应的完整类型。使用时需 `import stdint`。 + +#### 固定宽度整数短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT8` | `t.CInt8T` | `int8_t` | +| `INT16` | `t.CInt16T` | `int16_t` | +| `INT32` | `t.CInt32T` | `int32_t` | +| `INT64` | `t.CInt64T` | `int64_t` | +| `UINT8` | `t.CUInt8T` | `uint8_t` | +| `UINT16` | `t.CUInt16T` | `uint16_t` | +| `UINT32` | `t.CUInt32T` | `uint32_t` | +| `UINT64` | `t.CUInt64T` | `uint64_t` | + +#### 固定宽度指针短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT8PTR` | `t.CInt8T \| t.CPtr` | `int8_t*` | +| `INT16PTR` | `t.CInt16T \| t.CPtr` | `int16_t*` | +| `INT32PTR` | `t.CInt32T \| t.CPtr` | `int32_t*` | +| `INT64PTR` | `t.CInt64T \| t.CPtr` | `int64_t*` | +| `UINT8PTR` | `t.CUInt8T \| t.CPtr` | `uint8_t*` | +| `UINT16PTR` | `t.CUInt16T \| t.CPtr` | `uint16_t*` | +| `UINT32PTR` | `t.CUInt32T \| t.CPtr` | `uint32_t*` | +| `UINT64PTR` | `t.CUInt64T \| t.CPtr` | `uint64_t*` | + +#### 字符类型短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `CHAR8` | `t.CChar8T` | `char8_t` | +| `CHAR16` | `t.CChar16T` | `char16_t` | +| `CHAR32` | `t.CChar32T` | `char32_t` | +| `CHAR8PTR` | `t.CChar8T \| t.CPtr` | `char8_t*` | +| `CHAR16PTR` | `t.CChar16T \| t.CPtr` | `char16_t*` | +| `CHAR32PTR` | `t.CChar32T \| t.CPtr` | `char32_t*` | + +#### 基本类型短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT` | `t.CInt` | `int` | +| `UINT` | `t.CUnsignedInt` | `unsigned int` | +| `BOOL` | `t.CInt` | `int`（0/1） | +| `SHORT` | `t.CShort` | `short` | +| `USHORT` | `t.CUnsignedShort` | `unsigned short` | +| `LONG` | `t.CLong` | `long` | +| `ULONG` | `t.CUnsignedLong` | `unsigned long` | +| `LONGLONG` | `t.CLong \| t.CLong` | `long long` | +| `ULONGLONG` | `t.CUnsignedLong \| t.CLong` | `unsigned long long` | +| `FLOAT` | `t.CFloat` | `float` | +| `DOUBLE` | `t.CDouble` | `double` | + +#### 平台 / Windows 风格短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `BYTE` | `t.CUnsignedChar` | `unsigned char`（8位） | +| `BYTEPTR` | `BYTE \| t.CPtr` | `unsigned char*` | +| `WORD` | `t.CUInt16T` | `uint16_t`（16位） | +| `DWORD` | `t.CUInt32T` | `uint32_t`（32位） | +| `QWORD` | `t.CUInt64T` | `uint64_t`（64位） | +| `INTPTR` | `t.CInt \| t.CPtr` | `int*` | +| `UINTPTR` | `t.CUnsignedInt \| t.CPtr` | `unsigned int*` | +| `SHORTPTR` | `t.CShort \| t.CPtr` | `short*` | +| `USHORTPTR` | `t.CUnsignedShort \| t.CPtr` | `unsigned short*` | +| `VOIDPTR` | `t.CVoid \| t.CPtr` | `void*` | +| `TCHAR` | `t.CChar` | `char` | +| `WCHAR` | `WORD` | `uint16_t`（UTF-16） | +| `WCHARPTR` | `WORD \| t.CPtr` | `uint16_t*`（UTF-16） | +| `CHARLIST` | `str \| t.CPtr` | `char*` | +| `FSIZE_t` | `DWORD` | `uint32_t` | +| `LBA_t` | `DWORD` | `uint32_t` | + +### 浮点类型 + +| Viper 类型 | C 等价 | 大小（位） | +|------------|--------|-----------| +| `t.CFloat` | `float` | 32 | +| `t.CDouble` | `double` | 64 | + +### 其他基本类型 + +| Viper 类型 | C 等价 | 说明 | +|------------|--------|------| +| `t.CVoid` | `void` | 空类型 | +| `t.CBool` | `bool` | 布尔类型（8位） | +| `t.CWCharT` | `wchar_t` | 宽字符（32位） | +| `t.CChar8T` | `char8_t` | UTF-8 字符（8位，无符号） | +| `t.CChar16T` | `char16_t` | UTF-16 字符（16位，无符号） | +| `t.CChar32T` | `char32_t` | UTF-32 字符（32位，无符号） | + +### Python 便捷映射类型 +| Python 类型| Viper 类型 | Viper 值 | +|--------|--------|------| +|`int`|`t.CInt`|-| +|`float`|`t.CFloat`|-| +|`str`|`t.CChar \| t.CPtr`|-| +|`bool`|`t.CBool`|-| +|`True`|`t.CBool`|`1`| +|`False`|`t.CBool`|`0`| +|`None`|`NULL`|`t.CIntPtr(0)`| + +对于 t.py 和 c.py 的引用，必须使用 import，不能使用 from ... import ...，也不能使用 as，因为这两者都是直接以字符串识别的。 + +## 类型组合 + +Viper 使用 Python 的 `|` 来组合类型修饰符（TypeUnion），其仅在左值注解中有效，其它位置均为位或语义（至少重载前是这样的），这是 Viper 最重要的语法特性之一。 + +### 指针类型 + +使用 `t.CPtr` 表示指针，通过 `|` 与基本类型组合： + +```python +x: t.CInt | t.CPtr # int* — 指向 int 的指针 +p: t.CVoid | t.CPtr # void* — 通用指针 +s: t.CChar | t.CPtr # char* — 字符串指针 +pp: t.CInt | t.CPtr | t.CPtr # int** — 指向指针的指针 +``` + +等价 C 代码： +```c +int* x; +void* p; +char* s; +int** pp; +``` + +同时，指针也可以用 `Ptr[T]` 的方法表示，但在 Viper 中不太推荐，虽然说这样做一直是类似工程的常见用法，但是如果此处 `T` 是一个多态，同时我们想要复用 Python 的高亮插件，插件无法识别 `Ptr[T]` 的属性取值，但是对于 `T | Ptr` 可以。 + +```python +x: t.CPtr[t.CInt] +p: t.CPtr[t.CVoid] +s: t.CPtr[t.CChar] +pp: t.CPtr[t.CPtr[t.CInt]] +pp2: t.CInt | t.CPtr[t.CPtr] # 同时也可以这么表示 +``` + +等价 C 代码： + +```c +int* x; +void* p; +char* s: +int** pp; +int** pp2; +``` + +### 类型限定符组合 + +```python +x: t.CConst | t.CInt # const int +p: t.CConst | t.CChar | t.CPtr # const char* +``` + +### 存储类组合 + +```python +x: t.CStatic | t.CInt # static int +f: t.CExtern | t.CInt # extern int（声明其存在于外部） +``` + +### 函数返回类型组合 + +```python +def foo() -> t.CInt | t.CExport: # 导出函数，返回 int + return 0 + +def bar() -> t.CVoid | t.CExtern: # 外部函数声明 + pass +``` + +### `t.State` 声明性标记 + +`t.State` 是**声明性标记类型**，表示"仅声明不定义"（语义等价于 `t.CExport | t.CExtern`）。主要用于 FFI 外部函数声明：函数体为 `pass`，编译器只生成函数原型（declare），不生成定义（define），由链接器在外部库中解析符号。 + +```python +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, + dwShareMode: ULONG, + lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, + dwCreationDisposition: ULONG, + dwFlagsAndAttributes: ULONG, + hTemplateFile: HANDLE) -> HANDLE | t.State: + pass +``` + +等价 C 声明： + +```c +HANDLE CreateFileA(LPCSTR, ULONG, ULONG, SECURITY_ATTRIBUTES*, ULONG, ULONG, HANDLE); +``` + +`t.State` 是 Viper FFI 机制的核心，无需任何特殊语法，仅靠返回类型注解即可声明外部函数。详见 [08-c-operations.md 的 FFI 章节](08-c-operations.md#ffi-外部函数声明) 和 [includes/w32](../includes/w32) 的 Win32 绑定。 + +## 数组类型 + +使用 Python 的 `list[ElementType, Size]` 语法表示固定大小数组（通常的，这用于栈上和 BSS 段上）： + +```python +buf: list[t.CChar, 64] # char buf[64] +ids: list[t.CInt, 10] # int ids[10] +matrix: list[t.CFloat, 16] # float matrix[16] +entries: list[idt_entry, 256] # struct idt_entry entries[256] +s: list[t.CChar, None] # 自推长度 char s[]; +``` + +等价 C 代码： +```c +char buf[64]; +int ids[10]; +float matrix[16]; +struct idt_entry entries[256]; +``` + +### 数组指针和函数指针 + +```python +lp: list[t.CInt, 12] | t.CPtr +vp: t.Callable[[t.CInt, t.CInt], t.CInt] = x # 其中 t.Callable = typing.Callable，可以直接使用。 +``` + +## 结构体类型 + +使用 `class` 定义结构体，成员通过类型注解声明： + +```python +class point: + x: t.CInt + y: t.CInt +``` + +等价 C 代码： +```c +struct point { + int x; + int y; +}; +``` + +### 带指针成员的结构体 + +```python +class node: + value: t.CInt + next: node | t.CPtr # struct node* next +``` + +### 匿名结构体成员（❌） + +> WARNING: **警告：** 由于 Viper 在声明结构体，枚举等处不需要关键字，因此 `t.Anonymouse` 是不需要使用的。此关键字不再维护，强行使用可能导致未定义行为。 + +继承 `t.Anonymous` 的类作为匿名成员嵌入： + +```python +class header(t.Anonymous): + magic: t.CUInt32T + version: t.CUInt32T + +class packet: + header # 匿名嵌入，可直接访问 packet.magic + length: t.CUInt32T +``` + +### 字节序标记 + +> WARNING: **警告：** 此关键字是实验性的。未经充分测试，不当使用可能导致未定义行为。 + +使用 `t.BigEndian` 或 `t.LittleEndian` 标记成员的字节序： + +```python +class network_packet: + length: t.CUInt16T | t.BigEndian # 大端序存储 + type: t.CUInt8T # 默认小端序 +``` + +## 联合体类型 + +继承 `t.CUnion` 定义联合体： + +```python +class value(t.CUnion): + ival: t.CInt + fval: t.CFloat + pval: t.CVoid | t.CPtr +``` + +等价 C 代码： +```c +union value { + int ival; + float fval; + void* pval; +}; +``` + +## 枚举类型 + +继承 `t.CEnum` 定义枚举： + +```python +class color(t.CEnum): + RED: t.CEnum = 0 + GREEN: t.CEnum = 1 + BLUE: t.CEnum = 2 +``` + +等价 C 代码： +```c +enum color { + RED = 0, + GREEN = 1, + BLUE = 2 +}; +``` + +枚举成员可直接使用名称： + +```python +c: t.CInt = color.RED +``` + +### REnum（标签联合体） + +`t.CEnum` 是普通枚举（整数常量集合），`t.REnum` 是**标签联合体**（tagged union / Rust enum 风格）。每个变体可携带不同的 payload，通过 `match` 模式匹配分派。适用于少变体、无通用字段、match 分派的场景（如类型系统、IR 表示）。 + +```python +class LLVMType(t.REnum): + class Int: + Bits: t.CInt + class Void: + pass + +match ty: + case LLVMType.Int: + snprintf(buf, size, "i%d", ty.Bits) + case LLVMType.Void: + string.strcpy(buf, "void") +``` + +REnum 的详细用法、与 CEnum/胖节点 AST 的对比见 [05-classes.md 的 REnum 章节](05-classes.md#renum标签联合体)。 + +## Typedef + +使用 `t.CTypedef` 创建类型别名： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] # typedef int (*irq_handler_t)(); +``` + +或通过注解赋值： + +```python +MyStruct: t.CTypedef = original_struct # typedef struct original_struct MyStruct; +``` + +更推荐通过后者的形式，前者编译器不一定能够识别到注解是对 t.CTypedef 的，还是后面的注解的，而且容易触发未定义行为。 + +## 泛型类型（PEP 695 语法） + +Viper 支持 PEP 695 泛型语法 `class Name[T]:`，定义类型参数化的类。类型参数 `T` 在类体内可作为占位类型使用，实例化时用 `Name[ConcreteType]` 特化。 + +```python +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + +lst: GSList[GNode] | t.CPtr = GSList[GNode]() +``` + +泛型类还支持**递归泛型继承**（如 `class GNode(GSListNode[GNode])`），实现强类型自引用结构，无需 `void*` + 转型。这是 C/C++/Rust 均无法如此简洁表达的能力。 + +泛型类的完整用法、递归继承、实例化示例见 [05-classes.md 的泛型类章节](05-classes.md#泛型类pep-695-语法)。 + +## 强制类型转换 +在 Viper 中的强制类型转换有些不同，强制类型转换有两种方案，都是基于 `t.CType` 的用法，符合 Python 本意。 +对于原先就继承自 `t.CType` 的类型来说，可以直接使用 call 的方法进行类型转换，比如 +```python +x: t.CChar | t.CPtr = "Hello" +t.CVoid(x, t.CPtr) +``` +其中用法为 `t.CType(x, ...)`，此处...可以是其它的 `t.CType` 或将要提到的 `Typedef`，将其视为组合类型对 `x` 进行强制转换。 + +额外补充一句，对于比较，四则运算，可以不使用强制类型转换，而是依靠隐式类型转换，但如果是严谨数学计算，推荐还是强制类型转换避免判断出错。 + +此外，若将一个不同类型的右值赋值到左值，则会隐式将右值转换为左值的类型。 + +如果你不喜欢 `t.CVoid(x, t.CPtr)` 的方法，认为这将 `void` 和 `ptr` 分开，不便于理解，那么还有一个方法，即将二者打包为一个 Typedef，即： +```python +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +``` +则如果你想，你可以直接通过 call 的方式进行强转，比如： +```python +y: VOIDPTR = VOIDPTR(x) +``` + +## 宏定义常量 + +使用 `t.CDefine` 定义编译时常量。`t.CDefine` 不是数据类型，而是编译时元指令——标记为 `t.CDefine` 的常量在 LLVM IR 中不生成任何全局变量，而是在引用处直接内联展开。详见 [01-overview.md 中 t.CDefine 深度解析](01-overview.md#tcdefine-深度解析)。 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +``` + +语义上等价于 C 的 `#define`： +```c +#define MAX_SIZE 1024 +#define PAGE_SIZE 4096 +#define FA_READ 0x01 +``` + +宏定义支持表达式： + +```python +CPU_FEATURE_FPU: t.CDefine = (1 << 0) +CPU_FEATURE_APIC: t.CDefine = (1 << 6) +``` + +宏定义也可与类型组合使用： + +```python +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +``` + +## 位域 + +使用 `t.Bit(n)` 定义位域成员： + +```python +class flags: + a: t.CUInt32T | t.Bit[1] # 此处，如果存入 2，二进制是 10，会被截断为 0 + b: t.CUInt32T | t.Bit[3] + c: t.CUInt32T | t.Bit[4] +``` + +## LLVM IR 类型简写 + +Viper 提供了 LLVM IR 级别的类型简写： + +```python +i1 # 1位整数 +i8 # 8位有符号整数 +i16 # 16位有符号整数 +i32 # 32位有符号整数 +i64 # 64位有符号整数 +u1 # 1位无符号整数 +u8 # 8位无符号整数 +u16 # 16位无符号整数 +u32 # 32位无符号整数 +u64 # 64位无符号整数 +``` + +仍然通过 `t.x` 的方式调用，比如 `t.i1` 等。 + +## `__attribute__` 属性 + +通过 `t.attr` 命名空间访问 GCC `__attribute__` 属性： + +```python +t.attr.packed() # __attribute__((packed)) +t.attr.aligned(16) # __attribute__((aligned(16))) +t.attr.section(".text.startup") # __attribute__((section(".text.startup"))) +t.attr.always_inline() # __attribute__((always_inline)) +t.attr.noinline() # __attribute__((noinline)) +t.attr.noreturn() # __attribute__((noreturn)) +t.attr.weak() # __attribute__((weak)) +t.attr.unused() # __attribute__((unused)) +t.attr.constructor() # __attribute__((constructor)) +t.attr.destructor() # __attribute__((destructor)) +t.attr.deprecated("use X") # __attribute__((deprecated("use X"))) +``` + +### LLVM 函数属性 + +通过 `t.attr.llvm` 命名空间指定 LLVM 函数属性： + +```python +t.attr.llvm.nobuiltin +t.attr.llvm.nounwind +t.attr.llvm.noredzone +t.attr.llvm.willreturn +t.attr.llvm.mustprogress +t.attr.llvm.optnone +t.attr.llvm.noinline +t.attr.llvm.alwaysinline +t.attr.llvm.readnone +t.attr.llvm.readonly +t.attr.llvm.writeonly +t.attr.llvm.inaccessiblememonly +t.attr.llvm.inaccessiblemem_or_argmemonly +``` + +在函数返回类型注解中使用： + +```python +def foo() -> t.CInt | t.attr.llvm.nobuiltin | t.attr.llvm.nounwind: + return 0 +``` diff --git a/wiki/03-variables.md b/wiki/03-variables.md new file mode 100644 index 0000000..684e8d1 --- /dev/null +++ b/wiki/03-variables.md @@ -0,0 +1,177 @@ +# 03 - 变量声明与赋值 + +Viper 的变量声明使用 Python 的类型注解语法，通过 `t` 模块指定 C 级别的类型信息。 + +## 带注解的变量声明 + +使用 `变量名: 类型` 语法声明变量： + +```python +x: t.CInt = 0 # int x = 0; +y: t.CUInt32T = 42 # uint32_t y = 42; +name: list[t.CChar, 64] # char name[64]; +ptr: t.CVoid | t.CPtr # void* ptr; +msg: t.CConst | str = "Hello" # const char* msg = "Hello"; +``` + +### 无初始值的声明 + +未赋值的变量声明会生成零初始化的变量，此处是未定义行为，即在不同环境位置可能会覆盖0，也可能不会，最好预先复制或用 memset 清除： + +```python +count: t.CInt # int count = 0; +buffer: list[t.CChar, 256] # char buffer[256] = {0}; +``` + +### 延迟赋值 + +变量可以先声明后赋值： + +```python +x: t.CInt +x = 10 +``` + +### 指针变量 + +```python +p: t.CInt | t.CPtr # int* p; +fb: t.CVoid | t.CPtr # void* fb; +str_ptr: t.CConst | t.CChar | t.CPtr # const char* str_ptr; +``` + +### 指针变量的特殊初始化 + +使用 `t.CVoid(0, t.CPtr)` 初始化指针为 NULL： + +```python +p: t.CVoid | t.CPtr = t.CVoid(0, t.CPtr) # void* p = NULL; +``` + +或使用 `None`： + +```python +p: t.CVoid | t.CPtr = None # void* p = NULL; +``` + +## 普通赋值 + +```python +x: t.CInt = 0 +x = 42 # x = 42; +``` + +## 增强赋值 + +支持所有 Python 增强赋值运算符： + +```python +x: t.CInt = 0 +x += 1 # x += 1; +x -= 1 # x -= 1; +x *= 2 # x *= 2; +x //= 3 # x /= 3; (整数除法) +x %= 5 # x %= 5; +x <<= 1 # x <<= 1; +x >>= 1 # x >>= 1; +x |= 0xFF # x |= 0xFF; +x &= 0x0F # x &= 0x0F; +x ^= 0xAA # x ^= 0xAA; +``` + +## 全局变量 + +模块顶层声明的变量即为全局变量： + +```python +kbd_tid: t.CInt = -1 # 全局 int kbd_tid = -1; +BootInfo: bootinfo | t.CPtr # 全局 struct bootinfo* BootInfo; +``` + +### 静态全局变量 + +```python +x: t.CStatic | t.CInt = 0 # static int x = 0; +``` + +### 外部声明 + +```python +x: t.CExtern | t.CInt # extern int x; +``` + +## 常量定义 + +使用 `t.CDefine` 定义编译时常量。`t.CDefine` 不是数据类型，而是编译时元指令——常量在 LLVM IR 中不生成全局变量，而是在引用处直接内联展开。详见 [01-overview.md 中 t.CDefine 深度解析](01-overview.md#tcdefine-深度解析)。 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +FA_WRITE: t.CDefine = 0x02 +``` + +## 字符串常量 + +Viper 中的字符串字面量编译为 C 的字符串常量（`const char*`）： + +```python +msg: t.CConst | str = "Hello, World!" +serial.puts(msg) +``` + +也可以直接传递字符串字面量： + +```python +serial.puts("Hello, World!") +``` + +## 类型转换 + +使用类型构造函数进行显式类型转换： + +```python +x: t.CInt = 42 +y: t.CUInt32T = t.CUInt32T(x) # (uint32_t)x +f: t.CFloat = float(x) # (float)x +i: t.CInt = int(f) # (int)f +``` + +### 指针类型转换 + +```python +ptr: t.CVoid | t.CPtr = some_ptr +int_val: t.CUInt64T = t.CUInt64T(ptr) # (uint64_t)ptr — 指针转整数 +``` + +## 结构体实例化 + +> 此处不理解参见 ([05-classes.md 中 结构体实例化与初始化](05-classes.md#结构体实例化与初始化)) + +直接使用类名作为构造函数： + +```python +info: fat32_types.fat32_fileinfo +dp: fat32_types.fat32_dirobj +``` + +带初始化的实例化： + +```python +obj: MyClass = MyClass(arg1, arg2) +``` + +使用 `c.Addr` 获取实例指针： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass(arg1, arg2)) +``` + +## delete 语句 + +`del` 语句可用于调用对象的 `__del__` 或 `__delete__` 方法： + +```python +del obj # 调用 obj.__del__() 或 obj.__delete__() +del arr[idx] # 调用 arr.__delitem__(idx) +``` diff --git a/wiki/04-functions.md b/wiki/04-functions.md new file mode 100644 index 0000000..48da6aa --- /dev/null +++ b/wiki/04-functions.md @@ -0,0 +1,254 @@ +# 04 - 函数定义与调用 + +Viper 的函数定义使用 Python 的 `def` 语法，通过类型注解指定参数类型和返回类型。 + +## 基本函数定义 + +```python +def add(a: t.CInt, b: t.CInt) -> t.CInt: + return a + b +``` + +等价 C 代码： +```c +int add(int a, int b) { + return a + b; +} +``` + +## 无返回值函数 + +```python +def print_msg(msg: t.CConst | t.CChar | t.CPtr) -> t.CVoid: + serial.puts(msg) +``` + +等价 C 代码： +```c +void print_msg(const char* msg) { + serial_puts(msg); +} +``` + +## 函数修饰符 + +通过返回类型的 `|` 组合添加函数修饰符： + +### 导出函数 + +` t.CExport` 控制函数的符号可见性——标记为 `t.CExport` 的函数不加 SHA1 前缀，全局可见。详见 [01-overview.md 中 t.CExport 深度解析](01-overview.md#tcexport-深度解析)。 + +```python +def _start() -> t.CInt | t.CExport: + return 0 +``` + +### 外部函数声明 + +使用 `c.State` 表示仅声明不定义： + +```python +def isr0() -> t.CExtern | t.CVoid | c.State: pass +def isr1() -> t.CExtern | t.CVoid | c.State: pass +``` + +等价 C 代码： +```c +extern void isr0(); +extern void isr1(); +``` + +### 内联函数 + +`t.CInline` 标记函数为内联函数，编译器在调用点直接展开函数体。详见 [01-overview.md 中 t.CInline 深度解析](01-overview.md#tcinline-深度解析)。 + +```python +def fast_add(a: t.CInt, b: t.CInt) -> t.CInt | t.CInline: + return a + b +``` + +### 宏函数 + +当函数返回类型注解包含 `t.CDefine` 时，该函数被视为宏函数，不生成 LLVM IR 函数定义，在调用处内联展开。详见 [01-overview.md 中 t.CDefine 函数](01-overview.md#tcdefine-函数)。 + +```python +def MAKE_FLAG(bit) -> t.CDefine: + return (1 << bit) +``` + +### 静态函数 + +```python +def helper() -> t.CVoid | t.CStatic: + pass +``` + +## 函数属性装饰器 + +使用 `@c.Attribute` 添加 GCC `__attribute__` 属性： + +```python +@c.Attribute(t.attr.section(".text.startup"), t.attr.aligned(16)) +def _start() -> t.CInt: + return 0 +``` + +等价 C 代码： +```c +__attribute__((section(".text.startup"), aligned(16))) +int _start() { + return 0; +} +``` + +### 常用属性组合 + +```python +@c.Attribute(t.attr.always_inline()) +def hot_path() -> t.CInt: + return 0 + +@c.Attribute(t.attr.noreturn()) +def panic(msg: t.CConst | t.CChar | t.CPtr) -> t.CVoid: + serial.puts(msg) + while True: pass + +@c.Attribute(t.attr.packed) +``` + +## LLVM 函数属性 + +在返回类型注解中通过 `t.attr.llvm` 指定 LLVM 属性： + +```python +def foo() -> t.CInt | t.attr.llvm.nobuiltin | t.attr.llvm.nounwind: + return 0 +``` + +## 多返回值（CReturn） + +使用 `@c.CReturn` 装饰器实现多返回值，通过指针参数实现： + +```python +@c.CReturn(t.CInt, t.CInt) +def divmod(a: t.CInt, b: t.CInt) -> t.CVoid: + q: t.CInt = a // b + r: t.CInt = a % b + return q, r +``` + +规则： +1. `CReturn` 中的类型数量决定返回值个数 +2. 自动为每个返回类型添加指针参数（`t.CInt` → `t.CInt | t.CPtr`） +3. 参数名自动生成为 `__ReturnValue0__`、`__ReturnValue1__` 等 +4. 调用处自动传参 `&xxx` +5. 必须使用左右值赋值，返回值不能直接传入函数。 + +## 指针参数 + +```python +def read_data(buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: + pass +``` + +等价 C 代码： +```c +int read_data(char* buf, size_t size); +``` + +### 输出参数模式 + +使用 `c.Addr` 传递变量地址作为输出参数： + +```python +res: t.CInt = fat32.opendir("/", c.Addr(dp)) +``` + +等价 C 代码： +```c +int res = fat32_opendir("/", &dp); +``` + +后续会使用隐式左值获取来减少 `c.Addr` 的直接使用。 + +## 函数指针类型 + +使用 `t.Callable` 定义函数指针类型： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] +``` + +等价 C 代码： +```c +typedef int (*irq_handler_t)(); +``` + +带参数的函数指针： + +```python +callback_t: t.Callable[[t.CInt, t.CInt], t.CVoid] +``` + +## 默认参数 + +Viper 支持函数默认参数值： + +```python +def create_window(x: t.CInt, y: t.CInt, w: t.CInt, h: t.CInt, title: t.CConst | str = "Window") -> t.CInt: + pass +``` + +## return 语句 + +```python +return 0 # return 0; +return # return; (void 函数) +return a, b # 多返回值（需配合 @c.CReturn） +``` + +## 全局变量声明 + +函数外的变量声明为全局变量，函数内通过 `global` 关键字访问： + +```python +BootInfo: bootinfo | t.CPtr + +def init() -> t.CVoid: + global BootInfo + BootInfo = saved_rdi +``` + +支持 func-in-func 嵌套函数。 + +## Call Func 函数调用 + +函数调用最好通过顺序调用来实现 + +```python +def add(x: int, y: int, z: int): + return (x + y) * z + +def main() -> int | t.CExport: + print(add(1, 2, 3)) # (1 + 2) * 3 = 9 +``` + +这也是 C 中的标准实现方法，但实际上，你也可以乱序传参。 + +```python +# 仍然用刚才的 add 举例 +def main() -> int | t.CExport: + print(add(z=3, x=1, y=2)) # (1 + 2) * 3 = 9 +``` + +乱序传参或带参传参对于结构体的初始化同样有效。同时，你也能够给函数指定默认值。 + +```python +def add(x: int, y: int, z: int = 1): + return (x + y) * z + +def main() -> int | t.CExport: + print(add(1, 2)) # (1 + 2) * 1 = 3 +``` + +在结构体中也可以指定默认值。 diff --git a/wiki/05-classes.md b/wiki/05-classes.md new file mode 100644 index 0000000..231beab --- /dev/null +++ b/wiki/05-classes.md @@ -0,0 +1,274 @@ +# 05 - 类与数据布局 + +Viper 中的 `class` 有两种用途：定义**数据布局**（结构体/联合体/枚举，本章内容），以及定义**面向对象的对象类型**（见 [06-oop.md](06-oop.md)）。 + +## 结构体定义 + +普通 `class` 定义编译为 LLVM 结构体类型： + +```python +class point: + x: t.CInt + y: t.CInt +``` + +在 C 中： +```c +struct point { + int x; + int y; +}; +``` + +在 LLVM IR 中： +```llvm +%".point" = type { i32, i32 } +``` + +### 带属性的结构体 + +```python +@c.Attribute(t.attr.packed) +class idt_entry: + base_low: t.CUInt16T + selector: t.CUInt16T + ist_attr: t.CUInt8T + type_attr: t.CUInt8T + base_middle: t.CUInt16T + base_high: t.CUInt32T + reserved1: t.CUInt32T +``` + +### 结构体数组 + +```python +idt: list[idt_entry, 256] = [] +``` + +### 结构体指针成员 + +```python +class node: + value: t.CInt + next: node | t.CPtr # 指向自身的指针（有向图链表） +``` + +### 结构体实例化与初始化 + +```python +entry: idt_entry = idt_entry() +``` + +或带构造函数： + +```python +mousepoint = Point(x=1, y=2) +``` + +乱序传参或带参传参对于结构体的初始化同样有效： + +```python +entry: idt_entry = idt_entry(selector=0x08, base_low=0x0000) +``` + +也可以给结构体成员指定默认值： + +```python +class config: + width: t.CInt = 640 + height: t.CInt = 480 + depth: t.CInt = 32 +``` + +使用 `c.Addr` 获取实例指针： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass(arg1, arg2)) +``` + +### 匿名结构体成员（❌） + +> WARNING: **警告：** 由于 Viper 在声明结构体，枚举等处不需要关键字，因此 `t.Anonymouse` 是不需要使用的。此关键字不再维护，强行使用可能导致未定义行为。 + +继承 `t.Anonymous` 的类作为匿名成员嵌入： + +```python +class header(t.Anonymous): + magic: t.CUInt32T + version: t.CUInt32T + +class packet: + header # 匿名嵌入，可直接访问 packet.magic + length: t.CUInt32T +``` + +### 字节序标记 + +> WARNING: **警告：** 此关键字是实验性的。未经充分测试，不当使用可能导致未定义行为。 + +使用 `t.BigEndian` 或 `t.LittleEndian` 标记成员的字节序： + +```python +class network_packet: + length: t.CUInt16T | t.BigEndian # 大端序存储 + type: t.CUInt8T # 默认小端序 +``` + +## 联合体 + +继承 `t.CUnion` 定义联合体： + +```python +class value(t.CUnion): + ival: t.CInt + fval: t.CFloat + pval: t.CVoid | t.CPtr +``` + +等价 C 代码： +```c +union value { + int ival; + float fval; + void* pval; +}; +``` + +## 枚举 + +继承 `t.CEnum` 定义枚举： + +```python +class color(t.CEnum): + RED: t.CEnum = 0 + GREEN: t.CEnum = 1 + BLUE: t.CEnum = 2 +``` + +等价 C 代码： +```c +enum color { + RED = 0, + GREEN = 1, + BLUE = 2 +}; +``` + +枚举成员可直接使用名称： + +```python +c: t.CInt = color.RED +``` + +## REnum（标签联合体） + +`t.CEnum` 是普通枚举（整数常量集合），`t.REnum` 是**标签联合体**（tagged union / Rust enum 风格）。每个变体可携带不同的 payload，通过 `match` 模式匹配分派。 + +### 适用场景 + +| 类型 | 变体数 | 通用字段 | 访问模式 | 适用 | +|------|--------|---------|---------|------| +| `t.CEnum` | 任意 | 无 | 整数比较 | 状态码、opcode、token 类型 | +| `t.REnum` | 少（~10） | 无 | match 分派 | 类型系统、AST 少变体场景 | +| 胖节点（如 AST） | 多（70+） | 有 | vtype + 直接属性 | AST 节点（见 [includes/ast](../includes/ast)） | + +### 定义 + +```python +class LLVMType(t.REnum): + class Int: # 变体 Int，携带 payload + Bits: t.CInt + + class Ptr: # 变体 Ptr，携带 payload + Pointee: LLVMType | t.CPtr # 自引用用指针 + + class Void: # 变体 Void，无 payload + pass +``` + +布局为 `{ i32 __tag, }`，所有变体共享最大变体的 payload 空间。 + +### match 分派 + +```python +match ty: + case LLVMType.Int: # 位置绑定 payload：ty.Bits 可用 + snprintf(buf, size, "i%d", ty.IntBits) + case LLVMType.Ptr: # ty.PtrPointee 可用 + TypePrint(buf, size, ty.PtrPointee) + case LLVMType.Void: + string.strcpy(buf, "void") +``` + +详见 [includes/llvmlite/__types.py](../includes/llvmlite/__types.py) 的 `LLVMType` 实现。 + +## Typedef + +使用 `t.CTypedef` 创建类型别名： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] # typedef int (*irq_handler_t)(); +``` + +或通过注解赋值： + +```python +MyStruct: t.CTypedef = original_struct # typedef struct original_struct MyStruct; +``` + +更推荐通过后者的形式，前者编译器不一定能够识别到注解是对 `t.CTypedef` 的，还是后面的注解的，而且容易触发未定义行为。 + +## 泛型类（PEP 695 语法） + +TransPyC 支持 PEP 695 泛型语法 `class Name[T]:`，定义类型参数化的类。 + +### 基本泛型类 + +```python +@t.NoVTable +class GSListNode[T]: + Next: T | t.CPtr + +@t.NoVTable +class GSList[T]: + Head: T | t.CPtr + Tail: T | t.CPtr + Count: t.CSizeT + def append(self, node: T): ... +``` + +### 递归泛型继承 + +泛型类可递归引用自身作为类型参数，实现强类型自引用结构（无需转型）： + +```python +@t.NoVTable +class GNode(GSListNode[GNode]): # 递归泛型继承 + value: t.CInt +# 此时 GNode.Next: GNode|CPtr —— 强类型，零转型 +``` + +这是 C（`void*` + cast）、C++（模板继承但无 `@t.NoVTable` 精细控制）、Rust（无继承，需 trait bound）均无法如此简洁表达的能力。 + +### 实例化 + +```python +lst: GSList[GNode] | t.CPtr = GSList[GNode]() +node: GNode | t.CPtr = GNode() +lst.append(node) # 类型安全：append 只接受 GNode +``` + +详见 [includes/linkedlist.py](../includes/linkedlist.py) 和 [includes/vector.py](../includes/vector.py)。 + +## 位域 + +使用 `t.Bit(n)` 定义位域成员： + +```python +class flags: + a: t.CUInt32T | t.Bit[1] + b: t.CUInt32T | t.Bit[3] + c: t.CUInt32T | t.Bit[4] +``` + +注意：如果存入超出位域宽度的值，会被截断。例如 `a` 为 1 位，存入 2（二进制 `10`）会被截断为 0。 diff --git a/wiki/06-oop.md b/wiki/06-oop.md new file mode 100644 index 0000000..cdb349f --- /dev/null +++ b/wiki/06-oop.md @@ -0,0 +1,692 @@ +# 06 - 面向对象与运算符重载 + +Viper 中的 `class` 有两种语义：**数据布局**（结构体/联合体/枚举，见 [05-classes.md](05-classes.md)）和**面向对象**。面向对象特性通过 `@t.Object` 装饰器启用，多态通过 `@t.CVTable` 装饰器启用，非多态继承（C++ 风格，无 vtable 开销）通过 `@t.NoVTable` 装饰器启用。 + +## `@t.Object` 面向对象 + +使用 `@t.Object` 装饰器启用面向对象特性，类将拥有成员方法、运算符重载、属性装饰器等 OOP 支持。编译器也会自动识别：如果结构体内有函数定义，会自动使用 `@t.Object` 行为，但在实际工程中更推荐明确标记，避免未定义行为。 + +### 基本定义 + +```python +@t.Object +class ViperKernel: + BootInfo: bootinfo | t.CPtr + VGA: vga._VGAScreenDriver | t.CPtr + VGA_drv: vga._VGAScreenDriver + + def __init__(self): + self.BootInfo = BootInfo + serial.init() + serial.puts("ViperOS VKernel Test\n") +``` + +避免在 `__init__` 中使用 `return`，虽然不会报错，但是你永远无法获得 `return` 的结果，如果需要提前结束代码流，`return` 也不失为一种办法。 + +### 成员方法 + +```python +@t.Object +class Sheet: + id: t.CInt + x: t.CInt + y: t.CInt + w: t.CInt + h: t.CInt + + def __init__(self, x: t.CInt, y: t.CInt, w: t.CInt, h: t.CInt): + self.x = x + self.y = y + self.w = w + self.h = h + + def MoveTo(self, nx: t.CInt, ny: t.CInt): + self.x = nx + self.y = ny + + def Resize(self, nw: t.CInt, nh: t.CInt): + self.w = nw + self.h = nh +``` + +### 方法调用 + +```python +sheet: Sheet = Sheet(0, 0, 640, 480) +sheet.MoveTo(100, 100) +sheet.Resize(800, 600) +``` + +### 属性装饰器 + +支持 Python 的 `@property` 和 `@xxx.setter` 装饰器： + +> `@xxx.setter` 和自定义装饰器正在试验阶段，尽可能不要使用 + +```python +@t.Object +class MyObj: + _value: t.CInt + + @property + def value(self) -> t.CInt: + return self._value + + @value.setter + def value(self, v: t.CInt): + self._value = v +``` + +### 静态方法 + +```python +@staticmethod +def _yield(): + pass +``` + +### 对象指针与解引用 + +`@t.Object` 的实例通常通过指针操作： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass()) +obj: MyClass = c.Deref(obj_ptr) +``` + +### 成员访问 + +通过 `self` 访问成员变量和方法： + +```python +def method(self) -> t.CVoid: + self.x = 10 + y: t.CInt = self.y + self.DoSomething() +``` + +### 内嵌对象 + +`@t.Object` 可以包含其他对象实例（非指针）： + +```python +@t.Object +class Sheet: + _surf_obj: gfx.Surface + _renderer_obj: gfx.Renderer + surf: gfx.Surface | t.CPtr + renderer: gfx.Renderer | t.CPtr + + def __init__(self, fb: UINT32PTR, zb: float | t.CPtr, w: t.CInt, h: t.CInt): + self._surf_obj = gfx.Surface(fb, zb, w, h) + self.surf = c.Addr(self._surf_obj) + self._renderer_obj = gfx.Renderer(self.surf) + self.renderer = c.Addr(self._renderer_obj) +``` + +### `__before_init__` 方法 + +所有类（不限 `@t.Object`/`@t.CVTable`）自动生成 `__before_init__` 方法，在 `__init__` 之前**自动填充成员默认值**，并对 `@t.CVTable` 类初始化 vtable： + +```llvm +define void @".ClassName.__before_init__"(%".ClassName"* %self) { +entry: + ; 对 @t.CVTable 类：存储 vtable 指针到 slot 0 + ; 遍历所有成员：若有默认值则 store 默认值，否则 store 零值 + ret void +} +``` + +**职责**： +- 填充类定义中声明的成员默认值（如 `x: t.CInt = 0` 中的 `0`） +- 对无默认值的成员填充类型零值 +- 对 `@t.CVTable` 类额外初始化 vtable 指针 + +**调用时机**（由编译器在构造时自动插入）： +1. `alloca` 栈分配后调用一次 +2. 若存在 `__new__` 方法且返回堆指针，对堆指针再调用一次（alloca 的默认值已随栈帧丢弃，必须重新填充） +3. `__init__` 调用前 + +**就地构造**：用户也可手动调用 `__before_init__` 实现就地构造（placement new 语义），典型用法见 `mpool.alloc_buf`： + +```python +buf: Buf | t.CPtr = pool.alloc(Buf.__sizeof__()) +buf.__before_init__() # 填充默认值 +buf.__init__(args) # 用户初始化逻辑 +``` + +跨模块导入类时，导入侧也会声明 `__before_init__`，保证跨模块构造行为一致。 + +## `@t.NoVTable` 非多态继承 + +`@t.NoVTable` 启用 C++ 风格的非多态继承（对应 `struct A : B {}` 无 virtual）。与 `@t.CVTable` 的区别： + +| 特性 | `@t.NoVTable` | `@t.CVTable` | +|------|---------------|--------------| +| vtable 指针 | ❌ 无 | ✅ 自动插入 slot 0 | +| 虚派发 | ❌ 无 | ✅ 支持 | +| 方法继承 | ✅ 生成包装函数转发 | ✅ 通过 vtable | +| 字段展平 | ✅ 父类字段前置嵌入 | ✅ 父类字段前置嵌入（vtable 之后） | +| 运行时开销 | 零 | 每次虚调用一次间接跳转 | + +### 基本定义 + +```python +@t.NoVTable +class LinkedNode: + next: "LinkedNode" | t.CPtr + prev: "LinkedNode" | t.CPtr + def append(self, node: "LinkedNode" | t.CPtr): ... + +@t.NoVTable +class GNode(LinkedNode): # 非多态继承 + value: t.CInt +``` + +子类 `GNode` 的结构体布局为 `{next, prev, value}`（父类字段前置，无 vtable 指针）。 + +### 方法继承机制 + +子类调用父类方法时，编译器通过 `_generate_inherited_method_wrappers` 自动生成包装函数：将 `self` bitcast 为父类指针后尾调用父类方法。例如 `gnode.append(child)` 会生成： + +```llvm +define void @".GNode.append"(%".GNode"* %self, ...) { +entry: + %base = bitcast %".GNode"* %self to %".LinkedNode"* + call void @".LinkedNode.append"(%".LinkedNode"* %base, ...) + ret void +} +``` + +### 跨模块继承 + +跨模块父类（如 `linkedlist.LinkedNode`）的 `@t.NoVTable` 标记会在导入时被检测并记录，子类继承时同样跳过 vtable 插入。 + +### 向下转型 + +父类指针→子类指针需显式转型（生成 bitcast IR）： + +```python +root: Node = Node() # Node 继承 LinkedNode +child: LinkedNode | t.CPtr = root.child +# child 是 LinkedNode*，需要转回 Node*： +node: Node | t.CPtr = (Node | t.CPtr)(child) +``` + +隐式转型仅向上（子→父）。 + +## `@t.CVTable` 虚函数表与多态 + +> 尽可能避免使用 `self[0]` 来获取结构体成员，其不稳定，也不要手动操作虚表。 + +使用 `@t.CVTable` 装饰器可以启用虚函数表支持。启用后，结构体第一个成员自动插入 `i8*` 类型的 vtable 指针。`@t.CVTable` 的操作必须配合 `@t.Object`，但有时编译器会自动识别是否存在结构体内函数，自动使用 `@t.Object`。 + +### 基本定义 + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 +``` + +在 LLVM IR 中，`@t.CVTable` 类的结构体类型自动插入 vtable 指针： + +```llvm +%".Base" = type { i8*, i32 } +``` + +同时，编译器会生成全局 vtable 变量： + +```llvm +@".Base_Vtable" = internal global [1 x i8*] zeroinitializer +``` + +### 继承与多态 + +虚表可以帮你使用继承语法实现多态： + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 + +@t.Object +@t.CVTable +class M1(Base): + def method(self) -> t.CInt: + return 1 +``` + +当 M1 被初始化时，VTable 会自动修改，这点和 C++ 相同，尽可能避免直接操作 VTable。 + +继承时，子类会自动继承父类的成员变量和方法。子类重写父类方法时，vtable 中对应位置的函数指针被子类方法替换，实现多态。 + +### VTable 运行时修改 + +如果需要临时构造一个额外的 `Base` 结构体，还可以使用： + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 + +def __method1(self) -> t.CInt: + return 1 + +def main() -> t.CInt | t.CExport: + b: Base = Base() + b.method = __method1 +``` + +启用 `VTable` 的表中，允许直接修改内部函数为函数指针，在底层，此操作将修改 `VTable` 来达成目的。编译器会检测当前 vtable 是否与类 vtable 相同，如果相同则创建 vtable 的副本（memcpy），避免修改影响所有实例。 + +### 虚方法调度机制 + +虚方法调用的核心流程： + +1. 从对象指针 GEP 获取第 0 个 slot（vtable 指针） +2. 加载 vtable 指针 +3. bitcast 为 `[N x i8*]*` 类型 +4. GEP 获取方法索引处的函数指针 +5. 加载函数指针 +6. bitcast 为正确的函数指针类型 +7. 间接调用 + +## 继承 + +### 结构体成员继承 + +当子类继承父类（父类在 `Gen.class_members` 中存在）时： + +- 父类的成员变量被继承到子类（排在子类成员之前） +- 父类的默认值也被继承 +- 父类的方法被继承（重命名为 `子类名.方法名`） + +```python +@t.Object +class Animal: + name: list[t.CChar, 32] + age: t.CInt + + def __init__(self, age: t.CInt): + self.age = age + + def Speak(self) -> t.CVoid: + serial.puts("...\n") + +@t.Object +class Dog(Animal): + breed: t.CInt + + def Speak(self) -> t.CVoid: + serial.puts("Woof!\n") +``` + +### 异常类继承 + +Viper 支持异常类的继承，用于 `try/except` 的异常匹配： + +```python +class MyError(Exception): + pass + +class SpecificError(MyError): + pass +``` + +详见 [09-exceptions.md](09-exceptions.md)。 + +## 运算符重载 + +`@t.Object` 类支持运算符重载。编译器在遇到运算符时，首先检查左操作数是否为结构体类型，如果是则尝试调用对应的 dunder 方法。 + +### 算术运算符重载 + +| 运算符 | Dunder 方法 | 示例 | +|--------|------------|------| +| `+` | `__add__` | `a + b` → `a.__add__(b)` | +| `-` | `__sub__` | `a - b` → `a.__sub__(b)` | +| `*` | `__mul__` | `a * b` → `a.__mul__(b)` | +| `/` | `__truediv__` | `a / b` → `a.__truediv__(b)` | +| `//` | `__floordiv__` | `a // b` → `a.__floordiv__(b)` | +| `%` | `__mod__` | `a % b` → `a.__mod__(b)` | +| `**` | `__pow__` | `a ** b` → `a.__pow__(b)` | + +示例： + +```python +@t.Object +class Vec2: + x: t.CFloat + y: t.CFloat + + def __init__(self, x: t.CFloat, y: t.CFloat): + self.x = x + self.y = y + + def __add__(self, other: Vec2) -> Vec2: + result: Vec2 = Vec2(0.0, 0.0) + result.x = self.x + other.x + result.y = self.y + other.y + return result + + def __mul__(self, scalar: t.CFloat) -> Vec2: + result: Vec2 = Vec2(0.0, 0.0) + result.x = self.x * scalar + result.y = self.y * scalar + return result +``` + +### 增量赋值运算符重载 + +| 增量运算符 | 优先 Dunder | 回退 Dunder | +|-----------|------------|------------| +| `+=` | `__iadd__` | `__add__` | +| `-=` | `__isub__` | `__sub__` | +| `*=` | `__imul__` | `__mul__` | +| `/=` | `__itruediv__` | `__truediv__` | +| `//=` | `__ifloordiv__` | `__floordiv__` | +| `%=` | `__imod__` | `__mod__` | +| `**=` | `__ipow__` | `__pow__` | + +如果未定义增量版本（如 `__iadd__`），编译器自动回退到普通版本（如 `__add__`）。 + +### 比较运算符重载 + +> **实验性**：以下比较运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。定义它们不会报错，但比较运算不会自动调用。 + +| 运算符 | Dunder 方法 | 状态 | +|--------|------------|------| +| `==` | `__eq__` | 仅类型推断识别 | +| `!=` | `__ne__` | 仅类型推断识别 | +| `<` | `__lt__` | 仅类型推断识别 | +| `<=` | `__le__` | 仅类型推断识别 | +| `>` | `__gt__` | 仅类型推断识别 | +| `>=` | `__ge__` | 仅类型推断识别 | + +### 反向运算符重载 + +> **实验性**：以下反向运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。 + +| Dunder 方法 | 预期语义 | 状态 | +|------------|---------|------| +| `__radd__` | 右操作数加法 | 仅类型推断识别 | +| `__rsub__` | 右操作数减法 | 仅类型推断识别 | +| `__rmul__` | 右操作数乘法 | 仅类型推断识别 | +| `__rtruediv__` | 右操作数真除法 | 仅类型推断识别 | +| `__rfloordiv__` | 右操作数整除 | 仅类型推断识别 | +| `__rmod__` | 右操作数取模 | 仅类型推断识别 | +| `__rpow__` | 右操作数幂运算 | 仅类型推断识别 | + +### 一元运算符重载 + +> **实验性**：以下一元运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。 + +| 运算符 | Dunder 方法 | 状态 | +|--------|------------|------| +| `-a` | `__neg__` | 仅类型推断识别 | +| `+a` | `__pos__` | 仅类型推断识别 | + +## 可调用对象 `__call__` + +```python +@t.Object +class Counter: + value: t.CInt + + def __init__(self): + self.value = 0 + + def __call__(self) -> t.CInt: + self.value += 1 + return self.value + +c: Counter = Counter() +x: t.CInt = c() # x = 1 +y: t.CInt = c() # y = 2 +``` + +## 下标访问重载 + +### `__getitem__` — 读取 + +```python +def __getitem__(self, key: t.CInt) -> t.CInt: + return self.data[key] +``` + +### `__setitem__` — 赋值 + +```python +def __setitem__(self, key: t.CInt, value: t.CInt): + self.data[key] = value +``` + +### `__delitem__` — 删除 + +```python +def __delitem__(self, key: t.CInt): + pass +``` + +## 布尔转换 `__bool__` + +```python +@t.Object +class Buffer: + size: t.CInt + data: list[t.CChar, 1024] + + def __bool__(self) -> t.CBool: + return self.size > 0 + +buf: Buffer = Buffer() +if buf: + serial.puts("buffer is not empty\n") +``` + +## 长度 `__len__` + +```python +@t.Object +class Buffer: + size: t.CInt + + def __len__(self) -> t.CInt: + return self.size + +buf: Buffer = Buffer() +n: t.CInt = len(buf) +``` + +## 字符串转换 `__str__` + +```python +@t.Object +class Point: + x: t.CInt + y: t.CInt + + def __str__(self) -> t.CConst | t.CChar | t.CPtr: + return "Point" + +p: Point = Point(1, 2) +print(p) # 调用 p.__str__() +``` + +## 绝对值 `__abs__` + +```python +@t.Object +class SignedValue: + val: t.CInt + + def __abs__(self) -> t.CInt: + if self.val < 0: + return -self.val + return self.val + +sv: SignedValue = SignedValue() +sv.val = -42 +a: t.CInt = abs(sv) # a = 42 +``` + +## 迭代器协议 + +Viper 的迭代器协议与 Python 相同，但 `__next__` 的实现方式通过 `StopIteration` 异常来结束。 + +### `__iter__` 和 `__next__` + +```python +@t.Object +class IntRange: + start: t.CInt + end: t.CInt + current: t.CInt + + def __init__(self, start: t.CInt, end: t.CInt): + self.start = start + self.end = end + self.current = start + + def __iter__(self): + return self + + def __next__(self, stop_flag) -> t.CInt: + if self.current >= self.end: + raise StopIteration + val: t.CInt = self.current + self.current += 1 + return val +``` + +### for 循环使用 + +```python +r: IntRange = IntRange(0, 10) +for i in r: + print(i) +``` + +编译器生成的等价逻辑： + +```c +IntRange r = IntRange_init(0, 10); +IntRange iter = IntRange___iter__(&r); +while (1) { + i1 stop_flag = 0; + int i = IntRange___next__(&iter, &stop_flag); + if (stop_flag == 1) break; + print_int(i); +} +``` + +## 上下文管理器 + +`with` 语句要求对象实现 `__enter__` 和 `__exit__` 方法： + +```python +@t.Object +class File: + fd: t.CInt + path: t.CConst | t.CChar | t.CPtr + + def __init__(self, path: t.CConst | t.CChar | t.CPtr, flags: t.CInt): + self.path = path + self.fd = -1 + + def __enter__(self): + self.fd = fat32.open(self.path, self.flags) + return self + + def __exit__(self): + fat32.close(self.fd) + + def read(self) -> t.CInt: + return fat32.read(self.fd) +``` + +使用方式： + +```python +with File("/test.txt", FA_READ) as f: + data: t.CInt = f.read() +``` + +编译器会自动保证 `__exit__` 在 `with` 块结束时被调用。 + +## 析构 `__del__` / `__delete__` + +```python +del obj # 调用 obj.__del__() 或 obj.__delete__() +del arr[idx] # 调用 arr.__delitem__(idx) +``` + +## Dunder 方法完整参考 + +### 已实现自动调度的 Dunder 方法 + +| Dunder 方法 | 触发语法 | 说明 | +|------------|---------|------| +| `__init__` | `ClassName(args)` | 构造函数 | +| `__before_init__` | 自动生成 | vtable 初始化，在 `__init__` 之前执行 | +| `__add__` | `a + b` | 加法 | +| `__sub__` | `a - b` | 减法 | +| `__mul__` | `a * b` | 乘法 | +| `__truediv__` | `a / b` | 真除法 | +| `__floordiv__` | `a // b` | 整除 | +| `__mod__` | `a % b` | 取模 | +| `__pow__` | `a ** b` | 幂运算 | +| `__iadd__` | `a += b` | 增量加法（回退到 `__add__`） | +| `__isub__` | `a -= b` | 增量减法（回退到 `__sub__`） | +| `__imul__` | `a *= b` | 增量乘法（回退到 `__mul__`） | +| `__itruediv__` | `a /= b` | 增量真除法（回退到 `__truediv__`） | +| `__ifloordiv__` | `a //= b` | 增量整除（回退到 `__floordiv__`） | +| `__imod__` | `a %= b` | 增量取模（回退到 `__mod__`） | +| `__ipow__` | `a **= b` | 增量幂运算（回退到 `__pow__`） | +| `__call__` | `obj(args)` | 可调用对象 | +| `__getitem__` | `obj[key]` | 下标读取 | +| `__setitem__` | `obj[key] = value` | 下标赋值 | +| `__delitem__` | `del obj[key]` | 下标删除 | +| `__bool__` | `if obj:` | 布尔转换 | +| `__len__` | `len(obj)` | 长度 | +| `__str__` | `str(obj)` / `print(obj)` | 字符串转换 | +| `__abs__` | `abs(obj)` | 绝对值 | +| `__iter__` | `for x in obj:` | 获取迭代器 | +| `__next__` | `for x in obj:` | 获取下一个值（标志位方式） | +| `__enter__` | `with obj as x:` | 上下文管理器进入 | +| `__exit__` | `with obj as x:` | 上下文管理器退出 | +| `__del__` | `del obj` | 析构 | +| `__delete__` | `del obj` | 删除描述符 | + +### 声明支持但未实现自动调度的 Dunder 方法 + +| Dunder 方法 | 预期触发语法 | 当前状态 | +|------------|------------|---------| +| `__eq__` | `a == b` | 仅类型推断识别，比较运算不自动调度 | +| `__ne__` | `a != b` | 同上 | +| `__lt__` | `a < b` | 同上 | +| `__le__` | `a <= b` | 同上 | +| `__gt__` | `a > b` | 同上 | +| `__ge__` | `a >= b` | 同上 | +| `__neg__` | `-a` | 仅类型推断识别，一元负号不自动调度 | +| `__pos__` | `+a` | 同上 | +| `__radd__` | 右操作数加法 | 仅类型推断识别 | +| `__rsub__` | 右操作数减法 | 同上 | +| `__rmul__` | 右操作数乘法 | 同上 | +| `__rtruediv__` | 右操作数真除法 | 同上 | +| `__rfloordiv__` | 右操作数整除 | 同上 | +| `__rmod__` | 右操作数取模 | 同上 | +| `__rpow__` | 右操作数幂运算 | 同上 | diff --git a/wiki/07-control-flow.md b/wiki/07-control-flow.md new file mode 100644 index 0000000..6115aee --- /dev/null +++ b/wiki/07-control-flow.md @@ -0,0 +1,256 @@ +# 07 - 控制流 + +Viper 支持 Python 风格的控制流语句，编译为 LLVM IR 的基本块和分支指令。 + +## 条件语句 + +### if / elif / else + +```python +if x > 0: + y = 1 +elif x == 0: + y = 0 +else: + y = -1 +``` + +等价 C 代码： +```c +if (x > 0) { + y = 1; +} else if (x == 0) { + y = 0; +} else { + y = -1; +} +``` + +### 条件表达式 + +```python +is_dir: t.CInt = 1 if (info.attr & AM_DIR) else 0 +``` + +等价 C 代码： +```c +int is_dir = (info.attr & AM_DIR) ? 1 : 0; +``` + +### None 检查 + +```python +if BootInfo != None: + paging.init(BootInfo.MemmapAddr, BootInfo.MemmapSize) +else: + paging.init(0, 0) +``` + +等价 C 代码： +```c +if (BootInfo != NULL) { + paging_init(BootInfo->MemmapAddr, BootInfo->MemmapSize); +} else { + paging_init(0, 0); +} +``` + +## 循环语句 + +### while 循环 + +```python +i: t.CInt = 0 +while i < 10: + buf[i] = 0 + i += 1 +``` + +等价 C 代码： +```c +int i = 0; +while (i < 10) { + buf[i] = 0; + i += 1; +} +``` + +### while-else + +```python +while i < count: + if found: + break +else: + serial.puts("not found\n") +``` + +`else` 块在循环正常结束（非 `break` 退出）时执行。 + +### 无限循环 + +```python +while True: + sched.Scheduler._yield() +``` + +等价 C 代码： +```c +while (1) { + scheduler_yield(); +} +``` + +### for 循环（range）（关于 for 循环的更多用法，参见 [06-oop.md 中 迭代器协议](06-oop.md#迭代器协议)） + +```python +for i in range(10): + buf[i] = 0 + +for i in range(5, 10): + buf[i] = 0 + +for i in range(0, 100, 2): + buf[i] = 0 +``` + +等价 C 代码： +```c +for (int i = 0; i < 10; i++) { + buf[i] = 0; +} + +for (int i = 5; i < 10; i++) { + buf[i] = 0; +} + +for (int i = 0; i < 100; i += 2) { + buf[i] = 0; +} +``` + +### for 循环（迭代器） + +如果类实现了 `__iter__` 和 `__next__` 方法，可以使用 `for ... in` 迭代： + +```python +container: Iter = Iter() +for item in container: + process(item) +``` + +### for-else + +```python +for i in range(count): + if items[i] == target: + break +else: + serial.puts("not found\n") +``` + +### break 和 continue + +```python +while True: + if done: + break + if skip: + continue + process() +``` + +## match 语句 + +Viper 支持 Python 3.10+ 的 `match` 语句，编译为 C 的 `switch` 语句： + +```python +match self.ctype: + case UI_LABEL: + self.RenderLabel(sh) + case UI_BUTTON: + self.RenderButton(sh) + case UI_PANEL: + self.RenderPanel(sh) + case _: + pass +``` + +等价 C 代码： +```c +switch (self->ctype) { + case UI_LABEL: + self_RenderLabel(self, sh); + break; + case UI_BUTTON: + self_RenderButton(self, sh); + break; + case UI_PANEL: + self_RenderPanel(self, sh); + break; + default: + break; +} +``` + +### match 值匹配 + +```python +match code: + case 0: + serial.puts("OK\n") + case 1: + serial.puts("Error\n") + case _: + serial.puts("Unknown\n") +``` + +### match 或模式 + +```python +match value: + case 1 | 2 | 3: + serial.puts("small\n") + case _: + serial.puts("other\n") +``` + +### fallthrough（无 break） + +默认每个 `case` 自动添加 `break`。如需 fallthrough，使用 `c.NoBreak`： + +```python +match value: + case 1: + do_one() + c.NoBreak + case 2: + do_two() +``` + +如果需要提前 break，需要使用 `c.Break()`，因为在 `Python` 中，`match` 分支不支持直接使用 `break`。 + +## assert 语句 +> 避免使用此语句，此语句未经充分测试，在多数情况下，编译器不知道如何展示结果。 + +```python +assert ptr != None, "null pointer" +``` + +编译为条件检查和错误报告。 + +## with 语句 + +`with` 语句用于资源管理，要求对象实现 `__enter__` 和 `__exit__` 方法： + +```python +with File("/test.txt", FA_READ) as f: + data: t.CInt = f.read() +``` + +等价 C 代码： +```c +File* f = File_enter(File_new("/test.txt", FA_READ)); +int data = File_read(f); +File_exit(f); +``` diff --git a/wiki/08-c-operations.md b/wiki/08-c-operations.md new file mode 100644 index 0000000..253a74c --- /dev/null +++ b/wiki/08-c-operations.md @@ -0,0 +1,304 @@ +# 08 - C 语言操作 + +Viper 通过 `c` 模块提供对 C 语言底层特性的访问，包括指针操作、内联汇编、预处理指令等。 + +## 指针操作 + +### 取地址 `c.Addr` + +```python +x: t.CInt = 42 +p: t.CInt | t.CPtr = c.Addr(x) # int* p = &x; +``` + +对结构体成员取地址： + +```python +res: t.CInt = fat32.opendir("/", c.Addr(dp)) # res = fat32_opendir("/", &dp); +``` + +对数组取地址： + +```python +viperlib.snprintf(c.Addr(buf), 64, "hello %d", 42) # snprintf(&buf, 64, "hello %d", 42); +``` + +### 解引用 `c.Deref` + +```python +ptr: Sheet | t.CPtr = c.Addr(sheet_obj) +obj: Sheet = c.Deref(ptr) # struct Sheet obj = *ptr; +``` + +### 内存拷贝 `c.Load` + +```python +c.Load(ptr, value) # *ptr = *value; +``` + +### 解引用赋值 `c.DerefAs` + +```python +c.DerefAs(ptr, value) # *ptr = value; +``` + + +### 赋值 `c.Set` + +```python +c.Set(target, value) # target = value; +``` + +## 内联汇编 `c.Asm` + +Viper 提供了声明式的内联汇编语法，编译为 GCC 风格的 `__asm__ __volatile__` 语句。 + +### 基本用法 + +```python +c.Asm("nop") # __asm__ __volatile__("nop"); +``` + +### 带操作数的汇编 + +使用 f-string 和 `c.AsmInp` / `c.AsmOut` 标记操作数： + +```python +saved_rdi: t.CUnsignedLong +c.Asm(f"mov {c.AsmOut(saved_rdi, t.ASM_DESCR.OUTPUT_REG)}, rdi", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RDI]) +``` + +等价 C 代码： +```c +__asm__ __volatile__( + "mov %0, rdi" + : "=r"(saved_rdi) + : + : "memory", "rdi" +); +``` + +### 输入操作数 `c.AsmInp` + +```python +msg: t.CConst | t.CChar | t.CPtr = "Hello" +c.Asm(f"mov rdi, {c.AsmInp(msg, t.ASM_DESCR.REG_ANY)}", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX]) +``` + +### 完整示例 + +```python +c.Asm(f"""mov rdi, {c.AsmInp(msg, t.ASM_DESCR.REG_ANY)} + call {c.AsmInp(log_info_fn, t.ASM_DESCR.REG_ANY)}""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX, + t.ASM_DESCR.CLOBBER_RCX, t.ASM_DESCR.CLOBBER_RDX, + t.ASM_DESCR.CLOBBER_RDI, t.ASM_DESCR.CLOBBER_RSI, + t.ASM_DESCR.CLOBBER_R8, t.ASM_DESCR.CLOBBER_R9, + t.ASM_DESCR.CLOBBER_R10, t.ASM_DESCR.CLOBBER_R11]) +``` + +### ASM_DESCR 约束字符 + +#### 操作数修饰符 + +| 常量 | 值 | 说明 | +|------|----|------| +| `MODIFIER_OUTPUT` | `=` | 输出操作数 | +| `MODIFIER_READWRITE` | `+` | 读写操作数 | +| `MODIFIER_INPUT` | `` | 输入操作数（默认） | +| `MODIFIER_GLOBAL` | `&` | 全局操作数 | + +#### 寄存器约束 + +| 常量 | 值 | 说明 | +|------|----|------| +| `REG_ANY` | `r` | 任何通用寄存器 | +| `REG_EAX` / `REG_RAX` | `a` | EAX/RAX | +| `REG_EBX` / `REG_RBX` | `b` | EBX/RBX | +| `REG_ECX` / `REG_RCX` | `c` | ECX/RCX | +| `REG_EDX` / `REG_RDX` | `d` | EDX/RDX | +| `REG_ESI` / `REG_RSI` | `S` | ESI/RSI | +| `REG_EDI` / `REG_RDI` | `D` | EDI/RDI | +| `REG_XMM` | `x` | XMM 寄存器 | + +#### 内存与立即数 + +| 常量 | 值 | 说明 | +|------|----|------| +| `MEMORY` | `m` | 内存操作数 | +| `IMMEDIATE` | `i` | 立即数 | +| `ANY` | `g` | 通用寄存器/内存/立即数 | + +#### 预定义组合约束 + +| 常量 | 值 | 说明 | +|------|----|------| +| `OUTPUT_REG` | `=r` | 输出，通用寄存器 | +| `OUTPUT_MEM` | `=m` | 输出，内存 | +| `OUTPUT_EAX` | `=a` | 输出，EAX/RAX | +| `INPUT_REG` | `r` | 输入，通用寄存器 | +| `INPUT_MEM` | `m` | 输入，内存 | +| `INPUT_IMM` | `i` | 输入，立即数 | + +#### 破坏描述符 + +| 常量 | 说明 | +|------|------| +| `CLOBBER_EAX` / `CLOBBER_RAX` | 破坏 EAX/RAX | +| `CLOBBER_EBX` / `CLOBBER_RBX` | 破坏 EBX/RBX | +| `CLOBBER_ECX` / `CLOBBER_RCX` | 破坏 ECX/RCX | +| `CLOBBER_EDX` / `CLOBBER_RDX` | 破坏 EDX/RDX | +| `CLOBBER_ESI` / `CLOBBER_RSI` | 破坏 ESI/RSI | +| `CLOBBER_EDI` / `CLOBBER_RDI` | 破坏 EDI/RDI | +| `CLOBBER_CC` | 破坏条件码（标志寄存器） | +| `CLOBBER_MEMORY` | 破坏内存 | +| `CLOBBER_R8` ~ `CLOBBER_R15` | 破坏 R8~R15 | + +## 预处理指令 + +Viper 通过 `c` 模块的函数调用实现 C 预处理指令。 + +### #define + +```python +c.CDefine("MAX_SIZE", 1024) # #define MAX_SIZE 1024 +``` + +上述方法容易引起未定义行为，至少是不便于理解，更常用的方式是使用类型注解： + +```python +MAX_SIZE: t.CDefine = 1024 # #define MAX_SIZE 1024 +``` + +### 条件编译 + +```python +c.CIfndef(HEADER_H) # #ifndef HEADER_H +c.CDefine(HEADER_H) # #define HEADER_H +c.CEndif() # #endif +``` + +```python +c.CIfdef(DEBUG) # #ifdef DEBUG +c.CEndif() # #endif +``` + +```python +c.CIf(VERSION > 2) # #if VERSION > 2 +c.CElif(VERSION > 1) # #elif VERSION > 1 +c.CElse() # #else +c.CEndif() # #endif +``` + +### #undef + +```python +c.Undef(MACRO_NAME) # #undef MACRO_NAME +``` + +### #error + +```python +c.CError("Platform not supported") # #error "Platform not supported" +``` + +### #pragma +> 此关键字已不再支持 +```python +c.CPragma("GCC diagnostic push") # #pragma GCC diagnostic push +``` + +### ## 连接符 +> 此方法可能已经不再支持 +```python +c.TokenPast("PREFIX_", "NAME") # PREFIX_ ## NAME +``` + +由于编译器设计，以及不需要像 `C` 一样做大量字符替换，对于上述所有的条件宏都远比 `C` 差，实际工程中不推荐使用条件宏，后期将完善宏的设计，引入模板等。 + +## FFI 外部函数声明 + +Viper 的 FFI（Foreign Function Interface）**无需任何特殊语法**——仅靠返回类型注解中的 `t.State` 标记即可声明外部函数。`t.State` 是声明性标记类型，表示"仅声明不定义"（语义等价于 `t.CExport | t.CExtern`），编译器只生成 `declare` 原型，不生成 `define` 函数体，由链接器在外部库中解析符号。 + +### 基本声明 + +函数体为 `pass`，返回类型用 `RetType | t.State`： + +```python +def isr0() -> t.CVoid | t.State: pass # extern void isr0(); — 来自汇编或外部 +def getchar() -> t.CInt | t.State: pass # extern int getchar(); — 来自 libc +``` + +编译为 LLVM IR： + +```llvm +declare void @"isr0"() ; 不加 SHA1 前缀（t.State 含 CExport 语义） +declare i32 @"getchar"() +``` + +### Win32 API 绑定示例 + +[includes/w32](../includes/w32) 中的 Win32 绑定全部通过 `t.State` 声明，零特殊语法： + +```python +def CreateFileA(lpFileName: LPCSTR, dwDesiredAccess: ULONG, + dwShareMode: ULONG, + lpSecurityAttributes: SECURITY_ATTRIBUTES | t.CPtr, + dwCreationDisposition: ULONG, + dwFlagsAndAttributes: ULONG, + hTemplateFile: HANDLE) -> HANDLE | t.State: + pass +``` + +等价 C 声明： + +```c +HANDLE CreateFileA(LPCSTR, ULONG, ULONG, SECURITY_ATTRIBUTES*, + ULONG, ULONG, HANDLE); +``` + +链接时通过 `project.json` 指定外部库（如 `kernel32.lib`、`user32.lib`），链接器解析这些符号。 + +### `t.State` 与 `t.CExtern` 的区别 + +| 修饰 | SHA1 前缀 | 生成内容 | 用途 | +|------|----------|---------|------| +| 无修饰 | ✅ 加前缀 | `declare` + `define` | 模块私有函数 | +| `t.CExtern` | ❌ 不加 | `declare`（引用外部） | 引用外部 C 函数 | +| `t.CExport` | ❌ 不加 | `declare` + `define` | 对外暴露 API | +| `t.State` | ❌ 不加 | 仅 `declare` | FFI 外部函数声明（= `CExport \| CExtern`） | + +### 声明外部全局变量 + +外部全局变量通过 `t.CExtern` 注解声明： + +```python +errno: t.CExtern | t.CInt # extern int errno; +``` + +详见 [02-type-system.md 的 t.State 章节](02-type-system.md#tstate-声明性标记) 和 [includes/w32](../includes/w32) 的 Win32 绑定实现。 + +## LLVM IR 内联 + +### c.LLVMIR + +直接嵌入 LLVM IR 指令来实现高效且跨平台的汇编操作： + +```python +c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) +``` + +### c.LInp / c.LOut + +标记 LLVM IR 的输入/输出操作数： + +```python +c.LInp(expr) # 输入操作数 +c.LOut(expr) # 输出操作数 +``` + +## 运算符重载 + +`@t.Object` 类支持运算符重载，详见 [06-oop.md 中 运算符重载](06-oop.md#运算符重载)。 diff --git a/wiki/09-exceptions.md b/wiki/09-exceptions.md new file mode 100644 index 0000000..7f93516 --- /dev/null +++ b/wiki/09-exceptions.md @@ -0,0 +1,186 @@ +# 09 - 异常处理 + +Viper 支持 Python 风格的 `try/except/finally` 异常处理，编译为基于 jmp_buf 和错误码的 C 级别异常处理机制。 + +## 异常处理机制 + +Viper 的异常处理不使用 C++ 风格的零开销异常，而是通过函数参数传递错误码和错误消息实现。编译器自动为可能抛出异常的函数添加 `__eh_msg_out__` 和 `__eh_code_out__` 参数。 + +## try / except + +```python +try: + result = risky_operation() +except ValueError: + serial.puts("ValueError occurred\n") +except OSError: + serial.puts("OSError occurred\n") +except Exception: + serial.puts("Unknown error\n") +``` + +### 内置异常类型 + +Viper 预定义了以下异常类型及其错误码： + +| 异常类型 | 错误码 | +|---------|--------| +| `ValueError` | 1 | +| `TypeError` | 2 | +| `RuntimeError` | 3 | +| `ZeroDivisionError` | 4 | +| `IndexError` | 5 | +| `KeyError` | 6 | +| `IOError` | 7 | +| `OSError` | 8 | +| `AssertionError` | 9 | +| `Exception` | 99 | + +### 自定义异常 + +```python +class DiskFullError(Exception): + pass + +class ReadOnlyError(DiskFullError): + pass +``` + +自定义异常会自动分配递增的错误码（从 100 开始），并支持继承匹配。 + +### 多异常捕获 + +```python +try: + result = operation() +except (ValueError, TypeError): + serial.puts("Value or Type error\n") +``` + +### 获取异常信息 + +```python +try: + result = operation() +except Exception as e: + serial.puts("Error occurred\n") +``` + +## try / finally + +```python +try: + result = operation() +finally: + cleanup() +``` + +`finally` 块无论是否发生异常都会执行。 + +## try / except / finally + +```python +try: + result = operation() +except ValueError: + handle_value_error() +finally: + cleanup() +``` + +## raise 语句 + +### 抛出异常 + +```python +raise ValueError("invalid value") +``` + +### 抛出自定义异常 + +```python +raise MyCustomError("something went wrong") +``` + +### 重新抛出异常 + +```python +raise +``` + +### StopIteration + +`StopIteration` 有特殊处理，用于迭代器协议： + +```python +raise StopIteration +``` + +## 异常处理的编译实现 + +### 函数签名变换 + +可能抛出异常的函数会自动添加错误输出参数： + +```python +def risky_operation() -> t.CInt: + if error: + raise ValueError("bad value") + return 0 +``` + +编译后等价于： + +```c +int risky_operation(char** __eh_msg_out__, int* __eh_code_out__) { + if (error) { + *__eh_code_out__ = 1; // ValueError code + *__eh_msg_out__ = "bad value"; + return 0; + } + return 0; +} +``` + +### 调用点变换 + +```python +try: + result = risky_operation() +except ValueError: + handle_error() +``` + +编译后等价于： + +```c +char* __eh_msg = NULL; +int __eh_code = 0; +int result = risky_operation(&__eh_msg, &__eh_code); +if (__eh_msg != NULL) { + if (__eh_code == 1) { // ValueError + handle_error(); + } +} +``` + +## 异常继承匹配 + +异常匹配支持继承体系。捕获父类异常时，子类异常也会被匹配： + +```python +class FileSystemError(Exception): + pass + +class FileNotFoundError(FileSystemError): + pass + +class PermissionError(FileSystemError): + pass + +try: + operation() +except FileSystemError: + # 也会捕获 FileNotFoundError 和 PermissionError + handle_fs_error() +``` diff --git a/wiki/10-imports.md b/wiki/10-imports.md new file mode 100644 index 0000000..0324803 --- /dev/null +++ b/wiki/10-imports.md @@ -0,0 +1,258 @@ +# 10 - 模块与导入系统 + +Viper 的模块系统基于 Python 的 `import` 语法，但编译时通过 TransPyC 的符号表和类型系统实现跨模块的类型解析。 + +## 导入语法 + +### 标准导入 + +```python +import t, c # Viper 内置类型和操作模块 +import asm # 汇编辅助模块 +import string # 字符串操作模块 +``` + +### 模块导入 + +```python +import bootinfo +import intr +import platform.pch as pch +import drivers.serial.uart.serial as serial +``` + +### from 导入 + +```python +from stdint import * # 导入所有 stdint 类型别名 +from drivers.video.vesafb.vga import _VGAScreenDriver +``` + +### 相对导入 + +```python +from . import submodule +from .. import parent_module +``` + +## 特殊模块 + +### `t` 模块 + +类型定义模块，提供所有 C 类型对应的 Viper 类型。每个 Viper 文件通常都需要导入： + +```python +import t +``` + +### `c` 模块 + +C 语言操作模块，提供指针操作、内联汇编、预处理指令等： + +```python +import c +``` + +### `asm` 模块 + +汇编辅助模块，提供常用汇编指令的封装： + +```python +import asm + +asm.sti() # __asm__ __volatile__("sti"); +asm.hlt() # __asm__ __volatile__("hlt"); +asm.cli() # __asm__ __volatile__("cli"); +asm.nop() # __asm__ __volatile__("nop"); +asm.BSSClean() # 清零 BSS 段 +``` + +### `string` 模块 + +字符串/内存操作模块，对应 C 标准库的 `string.h`： +> string 部分使用 x86_64 专有汇编进行加速，因此某些函数在其它架构可能需要手动重新编写。 + +```python +import string + +string.memset(c.Addr(buf), 0, 64) # memset(&buf, 0, 64); +string.memcpy(dst, src, size) # memcpy(dst, src, size); +string.memmove(dst, src, size) # memmove(dst, src, size); +string.memcmp(a, b, size) # memcmp(a, b, size); +string.strcmp(a, b) # strcmp(a, b); +string.strcpy(dst, src) # strcpy(dst, src); +``` + +### `stdint` 模块 + +提供 C `stdint.h` 中的类型别名： + +```python +from stdint import * + +# 提供以下类型别名： +# UINT8PTR, UINT16PTR, UINT32PTR, UINT64PTR +# INT8PTR, INT16PTR, INT32PTR, INT64PTR +# 等等 +``` + +### `viperlib` 模块 + +ViperOS 标准库，提供常用 C 库函数： + +```python +import viperlib + +viperlib.snprintf(c.Addr(buf), 64, "value=%d", 42) # snprintf(&buf, 64, "value=%d", 42); +``` + +### `vpsdk` 模块 + +> vpsdk 模块 专供于基于 Viper 开发的示例操作系统 ViperOS，不可用于其它平台，且需要特殊配置。 + +ViperOS 应用 SDK，提供系统调用和窗口管理： + +```python +import vpsdk.window as window +import vpsdk.process as process +import vpsdk.dynlib as dynlib +import vpsdk.syscall as syscall +``` + +### `ast` 模块（自研解析器） + +[includes/ast](../includes/ast) 是**完全自研的 Python 解析器**（词法器 + 语法器 + AST 构造，约 4176 LOC），覆盖 CPython 几乎全部语法。这是向 `TransPyV` 完全自举的关键组件——自举后编译器不再依赖 Python 运行时解析源文件。 + +```python +from ast import parse, parse_expression + +root: AST | t.CPtr = parse(src, pool) # 解析整个源文件 +expr: AST | t.CPtr = parse_expression(src, pool) # 解析单个表达式 +``` + +AST 采用**胖节点**设计：70+ 变体共用一个 struct，用 `vtype`（`ASTVType` 枚举）区分类型，通用字段直接属性访问。详见 [13-bootstrapping.md](13-bootstrapping.md) 和 [Test/AstTest](../Test/AstTest)。 + +### `llvmlite` 模块（自研 IR 生成库） + +[includes/llvmlite](../includes/llvmlite) 是**完全自研的 LLVM IR 文本生成库**（约 1500 LOC），封装类型/值/模块/函数/基本块/构造器的创建与打印。同样是自举的关键组件。 + +```python +from llvmlite import LLVMTypeCore, LLVMModuleCore + +type_core: LLVMTypeCore = LLVMTypeCore(pool) +i32: LLVMType | t.CPtr = type_core.Int32() +ptr_ty: LLVMType | t.CPtr = type_core.Ptr(i32) + +mod_core: LLVMModuleCore = LLVMModuleCore(pool) +mod: Module | t.CPtr = mod_core.NewModule("main") +func: Function | t.CPtr = mod_core.CreateFunction(mod, "main", ret_ty) +``` + +类型系统用 `t.REnum`（`LLVMType`，少变体适合 match），指令 opcode 用 `t.CEnum`（`IROp`），对象分配用普通类 + `mpool`。详见 [includes/llvmlite/README.md](../includes/llvmlite/README.md) 的选型论证。 + +### `linkedlist` 模块（NoVTable + PEP 695 泛型示范） + +[includes/linkedlist](../includes/linkedlist.py) 提供链表实现，是 `@t.NoVTable` 非多态继承和 PEP 695 递归泛型继承的典范用例： + +```python +from linkedlist import LinkedNode, GSList + +@t.NoVTable +class Node(LinkedNode): # 非多态继承，零 vtable 开销 + value: t.CInt + +lst: GSList[Node] | t.CPtr = GSList[Node]() +node: Node | t.CPtr = Node() +node.value = 42 +lst.append(node) # 类型安全：append 只接受 Node +``` + +包含 `LinkedNode`/`SListNode`（双向/单向节点）和 `GSList[T]`（泛型链表，支持递归泛型继承 `class GNode(GSListNode[GNode])`）。详见 [05-classes.md 的泛型类章节](05-classes.md#泛型类pep-695-语法) 和 [06-oop.md 的 NoVTable 章节](06-oop.md#tnovtable-非多态继承)。 + +### `w32` 模块（Win32 FFI 绑定） + +[includes/w32](../includes/w32) 提供 Win32 API 绑定，全部通过 `t.State` 的 FFI 机制声明（零特殊语法）： + +```python +from w32.win32file import CreateFileA, ReadFile, WriteFile +from w32.win32console import WriteConsole +from w32.win32memory import HeapAlloc, HeapFree +from w32.win32process import CreateProcess +from w32.win32sync import CreateMutex, WaitForSingleObject + +handle: HANDLE = CreateFileA("test.txt", GENERIC_READ, FILE_SHARE_READ, + None, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, None) +``` + +子模块：`win32base`（基础类型/常量）、`win32file`（文件 IO）、`win32console`（控制台）、`win32memory`（内存）、`win32process`（进程/线程）、`win32sync`（同步对象）、`fileio`（高层文件操作封装）。详见 [08-c-operations.md 的 FFI 章节](08-c-operations.md#ffi-外部函数声明)。 + +## 模块解析流程 + +### 编译时模块加载 + +Viper 的 `import` 是编译时操作，不涉及运行时模块加载。完整的模块解析流程与两阶段编译紧密耦合： + +1. **阶段一**：从入口文件出发，通过 `import` 语句递归发现所有可达的源文件 +2. **阶段一**：对每个源文件生成 `.pyi` 签名存根和 `.stub.ll` LLVM IR 声明 +3. **阶段二**：加载所有 `.pyi` 和 `.stub.ll`，构建共享符号表 +4. **阶段二**：翻译源文件时，通过符号表解析跨模块的类型和函数引用 +5. **阶段二**：将被引用模块的 `.stub.ll` 声明嵌入到生成的 `.ll` 文件头部 + +### 类型存根文件（.pyi） + +`.pyi` 文件是模块的声明接口，由 `PythonToStubConverter` 在阶段一生成。存根文件的内容规则： + +| 元素 | 存根表示 | +|------|---------| +| `t.CDefine` 常量 | 保留完整定义（含赋值） | +| `t.CDefine` 函数 | 保留完整函数体 | +| 普通函数 | 仅签名 + `t.State` | +| 全局变量 | 添加 `t.CExtern` | +| 类定义 | 保留成员类型注解和方法签名 | + +存根文件还自动添加宏守卫（`c.CIfndef`/`c.CEndif()`），防止重复包含。 + +### SHA1 命名空间与导入 + +每个模块的函数和结构体在 LLVM IR 中自动加上 SHA1 前缀，消除跨模块符号冲突。只有标记为 `t.CExport` 的函数保持原始名称。详见 [01-overview.md 中 SHA1 命名空间机制](01-overview.md#sha1-命名空间机制)。 + +### 符号表 + +所有模块的类型信息统一存储在 `SymbolTable` 中，包含： +- 结构体/联合体/枚举定义 +- 函数签名 +- 全局变量 +- typedef 别名 +- `t.CDefine` 常量 + +### 跨模块类型引用 + +```python +# 在 fat32_types.py 中定义 +class fat32_fileinfo: + fname: list[t.CChar, 13] + attr: t.CUInt8T + file_size: t.CUInt32T + +# 在其他模块中引用 +import fat32_types +info: fat32_types.fat32_fileinfo +``` + +跨模块引用的结构体在 LLVM IR 中使用 SHA1 前缀的类型名： +```llvm +%".fat32_fileinfo" = type { [13 x i8], i8, i32 } +``` + +## 项目 includes 配置 + +在 `project.json` 中配置模块搜索路径： + +```json +{ + "includes": ["../.."], + "source_dir": "./Kernel" +} +``` + +编译器会在 `includes` 指定的目录中搜索导入的模块。 diff --git a/wiki/11-builtins.md b/wiki/11-builtins.md new file mode 100644 index 0000000..877b367 --- /dev/null +++ b/wiki/11-builtins.md @@ -0,0 +1,204 @@ +# 11 - 内置函数与运算符 + +Viper 支持 Python 内置函数和运算符，编译为对应的 C/LLVM 操作。 + +## 内置函数 + +### print + +```python +print("Hello, World!") # 调用 puts 或 printf +print(x, y) # 多参数输出 +``` + +### len + +```python +n: t.CInt = len(array) # 获取数组长度（编译时常量） +``` + +### sizeof + +```python +s: t.CSizeT = sizeof(t.CInt) # sizeof(int) +s: t.CSizeT = obj.__sizeof__() # sizeof(obj) +``` + +### abs + +```python +a: t.CInt = abs(x) # abs(x) +``` + +### int / float / bool + +```python +i: t.CInt = int(3.14) # (int)3.14 +f: t.CFloat = float(42) # (float)42 +b: t.CBool = bool(x) # x != 0 +``` + +### min / max + +```python +m: t.CInt = min(a, b) # a < b ? a : b +m: t.CInt = max(a, b) # a > b ? a : b +``` + +### chr / ord + +```python +c: t.CChar = chr(65) # 'A' (整数转字符) +n: t.CInt = ord('A') # 65 (字符转整数) +``` + +### range + +用于 `for` 循环： + +```python +for i in range(10): pass # for (int i = 0; i < 10; i++) +for i in range(5, 10): pass # for (int i = 5; i < 10; i++) +for i in range(0, 10, 2): pass # for (int i = 0; i < 10; i += 2) +``` + +## 运算符 + +### 算术运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a + b` | `a + b` | 加法 | +| `a - b` | `a - b` | 减法 | +| `a * b` | `a * b` | 乘法 | +| `a / b` | `a / b` | 浮点除法 | +| `a // b` | `a / b` | 整数除法 | +| `a % b` | `a % b` | 取模 | +| `a ** b` | `pow(a, b)` | 幂运算 | +| `-a` | `-a` | 取负 | +| `+a` | `+a` | 正号 | + +### 位运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a & b` | `a & b` | 按位与 | +| `a \| b` | `a \| b` | 按位或 | +| `a ^ b` | `a ^ b` | 按位异或 | +| `~a` | `~a` | 按位取反 | +| `a << n` | `a << n` | 左移 | +| `a >> n` | `a >> n` | 右移 | + +> **注意**：`|` 运算符在类型注解中表示类型组合，在表达式中表示按位或。编译器根据上下文区分。 + +### 比较运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a == b` | `a == b` | 等于 | +| `a != b` | `a != b` | 不等于 | +| `a < b` | `a < b` | 小于 | +| `a > b` | `a > b` | 大于 | +| `a <= b` | `a <= b` | 小于等于 | +| `a >= b` | `a >= b` | 大于等于 | + +### 逻辑运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a and b` | `a && b` | 逻辑与 | +| `a or b` | `a \|\| b` | 逻辑或 | +| `not a` | `!a` | 逻辑非 | + +### 增强赋值运算符 + +| Viper | C 等价 | +|-------|--------| +| `a += b` | `a += b` | +| `a -= b` | `a -= b` | +| `a *= b` | `a *= b` | +| `a //= b` | `a /= b` | +| `a %= b` | `a %= b` | +| `a <<= b` | `a <<= b` | +| `a >>= b` | `a >>= b` | +| `a &= b` | `a &= b` | +| `a \|= b` | `a \|= b` | +| `a ^= b` | `a ^= b` | + +## 类型自动转换 + +Viper 在算术运算中自动进行类型提升： + +1. **整数宽度提升**：较窄的整数类型自动扩展为较宽的类型 + - 有符号类型使用符号扩展（`sext`） + - 无符号类型使用零扩展（`zext`） + +2. **整数到浮点**：整数与浮点数运算时，整数自动转换为浮点 + +3. **浮点精度提升**：`float` 与 `double` 运算时，`float` 提升为 `double` + +## 指针算术 + +指针与整数的加法支持自动计算偏移： + +```python +buf: t.CUInt32T | t.CPtr +buf[i] = value # buf[i] = value; (自动计算偏移) +``` + +## 数组下标 + +```python +buf: list[t.CChar, 64] +buf[0] = 'H' # buf[0] = 'H'; +buf[i] = value # buf[i] = value; + +info.fname[0] # info.fname[0] (结构体数组成员访问) +``` + +## 成员访问 + +```python +obj.x # obj.x (直接成员) +obj.method() # obj.method() (方法调用) +ptr.x # ptr->x (指针自动解引用) +``` + +## 字符串字面量 + +字符串字面量编译为 C 字符串常量（`const char*`），其为只读： + +```python +serial.puts("Hello") # 传递 const char* +``` + +### 字符串索引 + +```python +msg: t.CConst | str = "Hello" +serial.puts(c.Addr(msg[4])) # 访问 msg[4] 的地址，输出 'o' +``` + +## C 标准库函数 + +Viper 内置支持以下 C 标准库函数的直接调用： + +| 函数 | 签名 | +|------|------| +| `strlen(s)` | `size_t strlen(const char* s)` | +| `memset(s, c, n)` | `void* memset(void* s, int c, size_t n)` | +| `memcpy(d, s, n)` | `void* memcpy(void* d, const void* s, size_t n)` | +| `memmove(d, s, n)` | `void* memmove(void* d, const void* s, size_t n)` | +| `memcmp(a, b, n)` | `int memcmp(const void* a, const void* b, size_t n)` | +| `strcmp(a, b)` | `int strcmp(const char* a, const char* b)` | +| `strncmp(a, b, n)` | `int strncmp(const char* a, const char* b, size_t n)` | +| `strcpy(d, s)` | `char* strcpy(char* d, const char* s)` | +| `strncpy(d, s, n)` | `char* strncpy(char* d, const char* s, size_t n)` | +| `strcat(d, s)` | `char* strcat(char* d, const char* s)` | +| `puts(s)` | `int puts(const char* s)` | +| `atoi(s)` | `int atoi(const char* s)` | +| `atol(s)` | `long atol(const char* s)` | +| `abs(x)` | `int abs(int x)` | +| `labs(x)` | `long labs(long x)` | + +部分模式比如裸机没有 libc，则不能使用这些函数，即使环境满足，Viper 标准库也可能和 C 标准库冲突。 diff --git a/wiki/12-project.md b/wiki/12-project.md new file mode 100644 index 0000000..f289240 --- /dev/null +++ b/wiki/12-project.md @@ -0,0 +1,252 @@ +# 12 - 项目配置与构建 + +Viper 项目使用 `project.json` 配置文件管理编译和链接参数。 + +## project.json 结构 + +```json +{ + "name": "Kernel", + "version": "1.0.0", + "source_dir": "./Kernel", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-mtriple=x86_64-none-elf", "-relocation-model=static", "-O2"] + }, + "linker": { + "cmd": "ld.lld.exe", + "flags": [ + "-m", "elf_x86_64", + "-T", "linker.ld", + "--oformat", "elf64-x86-64", + "--strip-all" + ], + "output": "kernel.bin" + }, + "includes": ["../../includes"], + "target": { + "triple": "x86_64-none-elf", + "datalayout": "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} +``` + +## 配置项说明 + +### 基本信息字段 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `name` | string | 项目名称 | +| `version` | string | 项目版本 | +| `source_dir` | string | 源代码目录（相对于 project.json 所在目录） | +| `temp_dir` | string | 临时文件目录（存放 .pyi 存根等） | +| `output_dir` | string | 输出文件目录 | + +### 编译器配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `compiler.cmd` | string | 编译器命令（通常为 `llc`） | +| `compiler.flags` | string[] | 编译器标志 | + +常用 `llc` 标志： +- `-filetype=obj`：输出目标文件 +- `-mtriple=x86_64-none-elf`：目标三元组 +- `-relocation-model=static`：静态重定位 +- `-O2`：优化级别 + +### 链接器配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `linker.cmd` | string | 链接器命令（通常为 `ld.lld.exe`） | +| `linker.flags` | string[] | 链接器标志 | +| `linker.output` | string | 输出文件名 | + +常用 `ld.lld` 标志： +- `-m elf_x86_64`：ELF x86_64 格式 +- `-T linker.ld`：链接脚本 +- `--oformat elf64-x86-64`：输出 ELF 格式 +- `--oformat binary`：输出原始二进制（用于内核） +- `--strip-all`：去除所有符号信息 + +### 目标平台配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `target.triple` | string | LLVM 目标三元组 | +| `target.datalayout` | string | LLVM 数据布局字符串 | + +默认值： +- triple: `x86_64-none-elf` +- datalayout: `e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128` + +### 包含路径 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `includes` | string[] | 模块搜索路径列表 | + +### 编译选项 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `options.slice_level` | int | 切片优化级别（0-3） | +| `options.target` | string | 编译目标（目前仅支持 `llvm`） | +| `options.strict_mode` | bool | 严格模式 | + +## 切片优化级别 + +| 级别 | 名称 | 说明 | +|------|------|------| +| `0` | `no_optimize` | 无优化 | +| `1` | `conservative` | 保守优化 | +| `2` | `moderate` | 适度优化 | +| `3` | `aggressive` | 激进优化（默认） | +| `s` | `size` | 体积优化 | + +## 项目结构示例 + +### 内核项目 + +``` +VKernel/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── Kernel/ # 源代码目录 +│ ├── main.py # 内核入口 +│ ├── bootinfo.py +│ ├── intr/ +│ │ ├── gdt.py +│ │ ├── idt.py +│ │ └── syscall.py +│ ├── drivers/ +│ │ ├── serial/ +│ │ ├── video/ +│ │ ├── storage/ +│ │ └── ... +│ └── ... +├── temp/ # 临时文件（自动生成） +│ ├── .pyi # 签名存根（阶段一生成） +│ ├── .stub.ll # LLVM IR 声明（阶段一生成） +│ └── _sha1_map.txt # SHA1→源文件路径映射 +└── output/ # 输出文件（自动生成） + ├── kernel.bin # 内核二进制 + ├── *.deps.json # 依赖信息 + └── linker.ld # 生成的链接脚本 +``` + +### 应用项目 + +``` +HelloWorld/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── main.py # 应用入口 +├── temp/ # 临时文件 +└── output/ # 输出文件 + └── helloworld.elf # ELF 可执行文件 +``` + +### 库项目 + +``` +SerialLogger/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── serial_logger.py # 库源码 +├── temp/ # 临时文件 +└── output/ # 输出文件 +``` + +## 链接脚本 + +ViperOS 使用自定义链接脚本控制内存布局： + +```ld +ENTRY(_start) + +SECTIONS +{ + . = 0x100000; /* 加载地址 */ + + .text : { + *(.text.startup) + *(.text*) + } + + .rodata : { + *(.rodata*) + } + + .data : { + *(.data*) + } + + .bss : { + *(.bss*) + *(COMMON) + } +} +``` + +## 编译命令 + +TransPyC 编译器的基本用法（注：适用于单个文件，而不适用于工程）： + +```bash +python TransPyC.py -f InputFile -o OutputFile [options] +``` + +### 命令行参数 + +| 参数 | 说明 | +|------|------| +| `-f` | 输入文件路径 | +| `-o` | 输出文件路径 | +| `-wh` | 头文件列表 | +| `-debug` | 调试输出文件 | +| `-cc` | 编译命令 | +| `-cflags` | 编译标志 | +| `-run` | 编译后运行 | +| `-args` | 运行参数 | +| `-h` | 辅助文件（C 或 Python） | + +工程编译使用： +```bash +python Projectrans.py xxx.json +``` +即可 + +## 增量编译 + +TransPyC 支持基于 SHA1 的增量编译，与两阶段编译模型紧密耦合： + +### 阶段一增量 + +- 每个源文件按内容计算 SHA1 哈希（16位） +- 如果 `.pyi` 和 `.stub.ll` 已存在，则跳过生成（缓存命中） +- 只有源文件内容变化导致 SHA1 变化时才重新生成声明接口 +- `temp/_sha1_map.txt` 记录 SHA1 → 源文件路径的映射，格式为 `:` + +### 阶段二增量 + +- 阶段二加载 `temp/_sha1_map.txt` 重建 SHA1 映射 +- 过滤掉不在当前 SHA1 映射中的旧 `.pyi` 和 `.stub.ll` 文件 +- 共享符号表一次性构建，避免每个文件重复加载 + +### SHA1 命名空间与增量编译的协同 + +SHA1 同时服务于命名空间隔离和增量编译： +- **命名空间**：非 `t.CExport` 函数和结构体使用 SHA1 前缀，消除符号冲突 +- **增量编译**：SHA1 不变则缓存命中，SHA1 变化则重新生成 + +详见 [01-overview.md 中 SHA1 命名空间机制](01-overview.md#sha1-命名空间机制)。 diff --git a/wiki/13-bootstrapping.md b/wiki/13-bootstrapping.md new file mode 100644 index 0000000..acddd9d --- /dev/null +++ b/wiki/13-bootstrapping.md @@ -0,0 +1,252 @@ +# TransPyC 自举可行性分析 + +> **状态**：进行中 +> **创建时间**：2026-06-26 +> **最后更新**：2026-06-30 +> **结论**：完全自举可行，核心障碍已被攻克，剩余工作明确 + +--- + +## 1. 背景 + +自举（bootstrapping）指用 TransPyC 语言重写 TransPyC 编译器自身，然后用现有编译器编译，得到新的原生编译器。本文档分析自举的可行性、进展与剩余工作。 + +**当前目标**：完全自举。TransPyC 的定位是独立的系统级语言，不是"Python → 原生 的工具"。自举是语言成熟度的证明，也是脱离 Python 运行时性能限制的必经之路。 + +--- + +## 2. LOC 统计 + +### 2.1 编译器主体（lib 目录） + +| 子目录 | 文件数 | LOC | 说明 | +|--------|--------|-----|------| +| `lib/core` | 64 | 28,610 | 编译器核心 | +| ├─ `core/Handles` | 30 | 18,885 | AST 节点处理器（最大模块） | +| ├─ `core/LLVMCG` | 8 | 2,782 | LLVM 代码生成 | +| └─ `core/Translator` | 6 | 1,751 | 翻译器框架 | +| `lib/Projectrans` | 6 | 4,371 | 项目级翻译（Phase1/Phase2） | +| `lib/StubGen` | 4 | 1,187 | Stub 生成器 | +| `lib/Shell` | 4 | 630 | CLI 入口 | +| `lib/includes` | 2 | 1,609 | t.py(968) + c.py(641) 类型定义 | +| `lib/utils` | 1 | 40 | 辅助 | +| `lib/constants` | 1 | 36 | 配置 | +| **lib 总计** | **80** | **36,483** | | + +### 2.2 Top 15 文件 + +| 文件 | LOC | +|------|-----| +| `lib/core/Handles/HandlesExprCall.py` | 3,153 | +| `lib/Projectrans/Phase2Translator.py` | 2,693 | +| `lib/core/Handles/HandlesImports.py` | 1,994 | +| `lib/core/Handles/HandlesFunctions.py` | 1,612 | +| `lib/core/Handles/HandlesAssign.py` | 1,544 | +| `lib/core/Handles/HandlesExprAttr.py` | 1,168 | +| `lib/core/Handles/HandlesClassDef.py` | 1,039 | +| `lib/core/Handles/HandlesFor.py` | 997 | +| `lib/core/Handles/HandlesTypeMerge.py` | 991 | +| `lib/includes/t.py` | 968 | +| `lib/StubGen/Converter.py` | 893 | +| `lib/core/Handles/HandlesExprBuiltin.py` | 868 | +| `lib/Projectrans/DeclarationGenerator.py` | 845 | +| `lib/core/Handles/HandlesBase.py` | 756 | +| `lib/core/stub_generator.py` | 699 | + +### 2.3 运行时库（includes 目录） + +| 模块 | LOC | 说明 | +|------|-----|------| +| `includes/numpy/__init__.py` | 1,016 | ndarray n 维数组 | +| `includes/zlib/` | ~1,900 | 完整 deflate/inflate/huffman/checksum | +| `includes/ast/` | 4,176 | **自研 Python 解析器（自举组件）** | +| `includes/llvmlite/` | ~1,500 | **自研 LLVM IR 生成库（自举组件）** | +| `includes/linkedlist.py` | ~400 | 泛型链表（@t.NoVTable + PEP 695） | +| `includes/vipermath.py` | 450 | 数学库 | +| ...（30+ 文件） | ... | mpool/mbuddy/string/stdio/json/w32 等 | +| **includes 总计** | **~12,000** | | + +### 2.4 全项目总计 + +``` +全项目 .py 总计：~48,000 LOC +``` + +**自举相关核心数字：** +- 编译器主体（lib 不含 t.py/c.py）：~34,874 LOC +- 自举基础组件（includes/ast + includes/llvmlite）：~5,700 LOC ✅ 已完成 +- 运行时库（includes 其余）：~9,500 LOC + +--- + +## 3. 自举进展（已完成的核心组件） + +### 3.1 ✅ includes/ast —— Python 解析器自举 + +**状态**：完成，AstTest 通过（解析 CPython 几乎全部语法） + +| 文件 | LOC | 功能 | +|------|-----|------| +| `__tokens.py` | ~600 | TokenType/Keyword/TokOp 枚举 + 查找表 | +| `__nodes.py` | ~1,400 | AST 胖节点 + ASTVType/ASTCtx/OpKind CEnum | +| `__lexer.py` | ~800 | 词法分析器 | +| `__parser.py` | 4,176（含表格） | 递归下降语法分析器 | + +**覆盖语法**：装饰器、async/await、match（含复杂模式 `[x,*rest]`/`{'type':t,**rest}`/`Point(x,y)`/`0|1|2`）、try/except/finally/raise from、with、lambda、列表/集合/字典/生成器推导式、f-string（含 `!r`/`!s` 转换和 `:>{width}` 格式说明）、链式比较、walrus、yield from、global/nonlocal、切片、嵌套三元。 + +**设计要点**： +- 胖节点设计（所有节点共用一个 `AST` struct，用 `vtype` 区分）—— 避免 REnum 在 70+ 变体下的内存膨胀 +- 从 `mpool.MPool` 分配，零 Python 运行时依赖 +- API 与 Python `ast` 模块对齐：`parse(src, pool) -> AST` + +### 3.2 ✅ includes/llvmlite —— LLVM IR 生成库自举 + +**状态**：完成，LLvmLiteTest 18/18 通过，TransPyV 自举实验跑通 + +| 文件 | 功能 | +|------|------| +| `__types.py` | `LLVMType` REnum + 类型构造/打印 | +| `__values.py` | Value/Constant/SSA 值表示 | +| `__function.py` | Function + BasicBlock + 参数管理 | +| `__module.py` | Module 容器 + 目标三元组 + 输出 .ll | +| `__builder.py` | IRBuilder（指令发射 + SSA 命名） | + +**设计要点**： +- `LLVMType` 用 `t.REnum`（tagged union）—— 少变体、无通用字段、match 分派的理想场景 +- 指令 opcode 用 `t.CEnum`（IROp）—— 类型安全且零开销 +- IR 文本生成用 `viperlib.snprintf` 格式化，避免手写逐字符拼接 +- 对象分配用普通类 + `mpool`，不持有全局 `mbuddy` + +**验证**：[TransPyV/App/main.py](../TransPyV/App/main.py) 10 个测试覆盖类型打印、值打印、build_add、if-else、模块打印、lli 实际执行。TransPyV 程序运行时自己生成合法 IR，被 `lli` 成功执行。 + +### 3.3 ✅ 基础设施库 + +以下 includes/ 库为自举后的编译器提供数据结构和系统调用基础： + +| 库 | 自举中的作用 | +|----|-------------| +| `linkedlist.py` | AST 节点树、符号表链、IR 指令链的容器（`@t.NoVTable` + PEP 695 递归泛型继承） | +| `mpool.py` / `mbuddy.py` | 编译器内存管理（替代 Python GC） | +| `string.py` / `viperlib.py` | 字符串处理 + snprintf（替代 Python 字符串方法） | +| `w32/` | FFI 声明，自举后编译器调用 llc/clang/linker 的基础 | +| `json/` | 符号表序列化（替代 pickle） | + +--- + +## 4. 障碍分析（修正版） + +原版本文档列出的"三大致命依赖"已被逐一攻克或降级： + +### 4.1 ~~llvmlite 依赖（原：致命）~~ → ✅ 已解决 + +**原问题**：编译器通过 `llvmlite.ir` 生成 IR，TransPyC 无法调用 Python 库。 + +**解决**：`includes/llvmlite/` 用 TransPyC 自身实现了 LLVM IR 文本生成库，API 与 Python `llvmlite.ir` 对齐。采用"字符串拼 IR"路径（路径 A），不绑定 LLVM C API。18/18 测试通过，TransPyV 自举实验验证可生成可执行 IR。 + +### 4.2 ~~ast 模块依赖（原：致命）~~ → ✅ 已解决 + +**原问题**：编译器用 Python `ast` 模块解析源码，自举需自己写 Python 解析器（预计 5,000-10,000 LOC）。 + +**解决**：`includes/ast/` 4,176 LOC 实现完整 Python 解析器，AstTest 解析 CPython 凥乎全部语法通过。预言已兑现，且实际规模（4,176 LOC）低于预估下限（5,000 LOC）。 + +### 4.3 ~~动态反射 4241 处（原：致命）~~ → ⚠️ 降级为可控 + +**原问题**：`getattr`/`setattr`/`hasattr`/`isinstance`/`issubclass` 共 4,241 处，静态语言无法直接表达。 + +**修正分析**：4241 这个数字按 Python 语法 grep 统计，未区分"语义上需要反射"和"用反射偷懒"。实际分类： + +| 用法 | 数量估计 | 是否真反射 | TransPyC 解法 | +|------|----------|-----------|--------------| +| `isinstance(node, ast.Xxx)` AST 节点分派 | ~2000 | ❌ 否 | 胖节点 `node.vtype == ASTVType.Xxx` 整数比较 | +| `isinstance(x, ir.IntType)` IR 类型分派 | ~500 | ❌ 否 | REnum `match x: case LLVMType.Int:` tag 比较 | +| `getattr(t, name, None)` 字符串→类型查表 | ~35 | ❌ 否 | 注册表 `t_type_registry: dict[str, t.CInt]` | +| `CTypeInfo.__getattr__` 门面路由 | ~100 | ✅ 是 | 重构为 REnum + match 显式分派 | +| `copy.deepcopy` / `pickle` | ~10 | ✅ 是 | 已解决（`__getstate__/__setstate__`） | +| `hasattr` 防御性检查 | ~1500 | ⚠️ 多数可消除 | 显式字段声明 + None 检查 | + +**真实反射障碍约 200-500 处**，集中体现在 `CTypeInfo` 门面模式（FIELD_ROUTES 动态路由到 `_ts`/`_sm`）。重构方案：用 REnum + match 替换 `__getattr__` 路由，已在 `includes/llvmlite/__types.py` 验证此范式可行。 + +### 4.4 Python 标准库依赖（原：严重）→ ⚠️ 部分已解决 + +| Python 库 | 用途 | TransPyC 现状 | +|-----------|------|--------------| +| `ast` | 源码解析 | ✅ `includes/ast/` | +| `llvmlite.ir` | LLVM IR 生成 | ✅ `includes/llvmlite/` | +| `json` | 序列化 | ✅ `includes/json/` | +| `os`/`sys` | 文件/路径 | ⚠️ `includes/os/` + `w32/`（Windows） | +| `subprocess` | 调用 llc/clang | ❌ 待实现（w32 CreateProcess 可基础） | +| `pickle` | 符号表序列化 | ❌ 待实现（json 替代） | +| `concurrent.futures` | 并行编译 | ❌ 待实现（w32 线程可基础） | +| `re` | 正则表达式 | ❌ 待实现 | +| `traceback` | 异常栈 | ❌ 待实现 | + +--- + +## 5. 自举路径 + +### 5.1 路径 A：完全自举（当前目标，可行） + +**已完成**： +1. ✅ `includes/ast/` Python 解析器（4,176 LOC） +2. ✅ `includes/llvmlite/` LLVM IR 生成库（~1,500 LOC） +3. ✅ 基础设施库（linkedlist/mpool/mbuddy/string/json/w32） + +**剩余工作**： +1. **CTypeInfo 门面重构**（~200-500 处真反射 → REnum + match） +2. **lib/core 重写**（~20K LOC TransPyC，含 Handles/LLVMCG/Translator） +3. **lib/Projectrans 重写**（~5K LOC，Phase1/Phase2/DeclarationGenerator） +4. **lib/StubGen 重写**（~2K LOC） +5. **Python 标准库缺口补齐**（os/subprocess/re/pickle，~5K LOC） + +**预计新增代码量**：30-35K LOC TransPyC。与已完成的 includes/ast（4,176 LOC）+ includes/llvmlite（~1,500 LOC）+ 其他基础库为同一量级——已经证明过两次能做这件事。 + +### 5.2 关键技术验证点（已完成） + +以下能力已在 includes/ 库中验证，证明 TransPyC 能承载编译器复杂度： + +- **复杂泛型容器**：`linkedlist.py` 的递归泛型继承 `class GNode(GSListNode[GNode])`（C/C++/Rust 均无法如此简洁表达） +- **复杂算法**：`zlib/` 的完整 deflate/inflate/huffman（~1,900 LOC 生产级压缩） +- **n 维数组**：`numpy/__init__.py` 的 ndarray + 运算符重载 + `__new__` 堆分配 +- **平台 FFI**：`w32/` 的 Win32 API 绑定（file/memory/process/sync/console） +- **LLVM IR 生成**：`llvmlite/` 的 REnum 类型系统 + IRBuilder + ModulePrint + +--- + +## 6. 自举的意义 + +1. **性能**：当前 Python 编译器翻译 6 个文件 0.6s，全量编译 36K LOC 编译器自身时 Python 解释器开销显著。自举后编译器是原生码，编译速度数量级提升。 + +2. **部署**：当前分发需 Python 运行时 + llvmlite + ast 依赖。自举后只需一个可执行文件 + llc/clang——系统级语言编译器该有的形态。 + +3. **语言成熟度证明**：能写自己的编译器是系统级语言的"成人礼"。C、C++、Rust、Go、Zig 全都走过这条路。 + +4. **dogfooding 闭环**：自举后每次改编译器都用编译器自己编译自己，持续暴露 bug 和性能瓶颈，形成正反馈。`includes/ast` 和 `includes/llvmlite` 的开发过程已经产出了大量深层 bug 修复（REnum 变体名冲突、跨模块继承字段展平、NoVTable OOP 方法继承等）。 + +--- + +## 7. 结论 + +| 维度 | 评估 | +|------|------| +| 理论可行性 | ✅ 可行（图灵完备） | +| 实践可行性 | ✅ 完全自举可行（核心障碍已攻克） | +| 当前进展 | ⚠️ 前端（ast）+ 后端（llvmlite）+ 基础库已就位，剩中间层（lib/core）重构 | +| 剩余工作量 | ~30-35K LOC TransPyC 重写 + CTypeInfo 门面重构 | +| 推荐策略 | 路径 A：完全自举，按 lib/core → lib/Projectrans → lib/StubGen 顺序推进 | + +**核心判断**：原版本文档的"三大致命依赖"结论已全部失效——llvmlite 和 ast 已被 includes/ 下的自研库替代，4241 处反射中 90% 是工程偷懒而非语义需要。完全自举从"几乎不可能"变为"明确可行"，剩余工作是体量问题而非技术障碍。 + +--- + +## 8. 附录：原版本文档的修正说明 + +本文档（2026-06-30 版）修正了 2026-06-26 原版的以下错误判断： + +| 原版断言 | 修正 | +|---------|------| +| "完全自举几乎不可能" | 完全自举可行，核心障碍已攻克 | +| "llvmlite 依赖致命" | 已被 `includes/llvmlite/` 解决（18/18 测试） | +| "ast 模块依赖致命" | 已被 `includes/ast/` 解决（4,176 LOC，AstTest 通过） | +| "4241 处动态反射致命" | 90% 是工程偷懒（isinstance 节点分派/IR 类型分派），真实反射约 200-500 处 | +| "推荐路径 C：不自举" | 推荐路径 A：完全自举 | +| "预计自举后 60K-80K LOC" | 修正为 30-35K LOC（基础组件已 5,700 LOC 完成） | diff --git a/wiki/all.md b/wiki/all.md new file mode 100644 index 0000000..4b80528 --- /dev/null +++ b/wiki/all.md @@ -0,0 +1,3555 @@ +# 01 - 语言概述与编译流程 + +## 语言定位 + +`Viper` 是一种 **系统级编程语言**，其语法基于 `Python`，但编译目标为 `LLVM IR`，最终生成原生机器码。`Viper` 不是 `Python` 的超集或子集，也不是"`C` 的语法糖"——它是一门拥有独立类型系统、独立编译模型、独立模块体系的语言。`Viper` 选择 `Python` 语法作为表达形式，是因为 `Python` 的 `AST` 可被标准库直接解析，从而将编译器的精力集中在语义翻译而非词法/语法分析上。同时也是因为 `Python` 是人类公认的可读性最高的语言。 + +而实际上，下文中所有提到的 `C` 语言都是为了便于使用 `C` 的底层开发者理解。由于历史原因，在旧版本中我们使用 `C` 来作为中间语言，如同旧版本的 `C++` 一样，但现在 `target="c"` 的路径已经完全删除，`C` 路径永久不再受维护。 + +### 核心设计决策 + +1. **Python 语法，LLVM 语义**：源文件是完全合法的 `Python` 语法，但通过 `t` 模块的类型注解系统赋予完全不同的语义 —— `t.CInt` 不是 `Python` 的 `int`，而是 `LLVM` 的 `i32` +2. **类型注解即编译指令**：Viper 的类型注解不是可选的提示，而是编译器生成 `LLVM IR` 的决定性依据。`x: t.CInt` 生成 `i32`，`x: t.CInt | t.CPtr` 生成 `i32*`，但无注解时会自动推导，详见下文。 +3. **两阶段编译**：先从源文件提取声明接口（`.pyi` + `.stub.ll`），再使用声明接口翻译源文件为含代码的 `.ll`。这是 `Viper` 区别于简单"`Python` 转 `C`"工具的关键架构 +4. **`SHA1` 命名空间**：每个源文件按内容 `SHA1` 哈希命名，非导出函数和结构体自动加上 `SHA1` 前缀，从根本上消除跨模块的符号冲突 +5. **万物皆数据**：为便于底层开发，实际上所有的变量和类型一般并不适用于鸭子类型，但在语义层面我们会尽可能贴近鸭子类型。 + +### 与 `Python` 的关系 + +| 特性 | `Python` | `Viper` | +|------|--------|-------| +| 类型系统 | 动态类型 | 静态类型（通过 `t` 模块注解，编译时确定） | +| 内存管理 | `GC` 自动回收 | 手动管理，无 `GC`，无运行时 | +| 运行时 | 解释执行 + 字节码 | 编译为原生机器码（通过 `LLVM`） | +| 对象模型 | 万物皆对象 | 仅 `@t.Object` 类有对象语义，`@t.CVTable` 为多态，普通 `class` 仅为结构体布局 | +| 标准库 | `Python` 标准库 | `Viper` 标准库（`includes/`）+ `ViperOS SDK` | +| 空值 | `None` | `None`（编译为 `NULL`/零值） | +| 模块系统 | 运行时 `import` | 编译时解析，插入 `.stub.ll`，`SHA1` 命名空间隔离 | + +### 与 C 的关系 + +Viper 编译后的代码在底层等价于 `C` 编译后的机器码（都经过 LLVM），但 Viper 在语言层面提供了 C 所没有的能力： + +- **SHA1 命名空间**：自动消除符号冲突，无需手动管理 `static`/命名前缀 +- **类型组合语法**：`t.CConst | t.CInt | t.CPtr` 比 `const int*` 更具组合性 +- **声明式内联汇编**：`c.Asm(f"mov {c.AsmOut(x, t.ASM_DESCR.OUTPUT_REG)}, rdi")` 比裸 `__asm__` 更安全，也更具可读性。 +- **结构化预处理**：`c.CIfdef`/`c.CEndif()` 替代 `#ifdef`/`#endif`。 +- **面向对象**：`@t.Object` + `@t.CVTable` 提供 `vtable` 支持的 `OOP` +- **存根驱动的模块系统**：`.pyi` 存根文件实现跨模块类型解析，无需头文件 +- **更多内容**：额外更多的不依赖操作系统的函数和语法 + +## 两阶段编译流程 + +Viper 的编译由 `Projectrans.py` 驱动，分为两个阶段。这是 Viper 编译模型的核心，理解两阶段编译是理解 `t.CDefine`、`t.CExport`、`t.CInline` 等关键概念的前提。 + +``` +源文件 (.py) ────────────────────────────────────────────────────── + │ │ + │ ┌─────────────── 阶段一：声明提取 ───────────────┐ │ + │ │ │ │ + │ │ 1. 计算源文件 SHA1 │ │ + │ │ 2. 生成 .pyi（签名存根） │ │ + │ │ 3. 构建结构体注册表 │ │ + │ │ 4. 生成 .stub.ll（LLVM IR 声明） │ │ + │ │ │ │ + │ └────────────────────────────────────────────────┘ │ + │ │ + │ ┌─────────────── 阶段二：代码翻译 ───────────────┐ │ + │ │ │ │ + │ │ 1. 加载所有 .pyi 和 .stub.ll │ │ + │ │ 2. 构建共享符号表 │ │ + │ │ 3. 收集内联函数符号 │ │ + │ │ 4. 翻译源文件 → .ll（含代码） │ │ + │ │ 5. llc 编译 → .o（目标文件） │ │ + │ │ 6. ld.lld 链接 → 可执行文件 │ │ + │ │ │ │ + │ └────────────────────────────────────────────────┘ │ + │ │ + └──────────────────────────────────────────────────────────────┘ +``` + +### 阶段一：声明提取（Phase1Generator） + +阶段一的目标是从每个源文件提取**声明接口**，使其他模块在阶段二翻译时能够正确解析跨模块引用。 + +#### 步骤 1：可达文件发现与拓扑排序 + +从入口文件（`main.py` 或 `project.json` 指定）出发，通过 `import` 语句递归遍历，找出所有可达的 `.py` 源文件。然后对文件进行拓扑排序，确保被依赖的模块先被处理。 + +``` +main.py → import serial → import drivers.serial.uart.serial → ... +``` + +#### 步骤 2：生成 `.pyi` 签名存根 + +对每个源文件，计算其内容的 SHA1 哈希（16位），然后调用 `PythonToStubConverter` 生成签名存根文件 `.pyi`。 + +**SHA1 计算方式**： +```python +sha1 = hashlib.sha1(content.encode('utf-8')).hexdigest()[:16] +``` + +**存根文件的内容规则**： + +| 源文件元素 | 存根文件中的表示 | +|-----------|----------------| +| `t.CDefine` 常量 | 保留原样（含赋值）：`MAX_SIZE: t.CDefine = 1024` | +| `t.CDefine` 函数（返回 `t.CDefine` 的函数） | 保留完整函数体（宏展开需要） | +| 普通函数 | 仅保留签名，添加 `| c.State`：`def foo(x: t.CInt) -> t.CVoid | c.State: pass` | +| 类定义 | 保留成员类型注解和方法签名 | +| 全局变量 | 添加 `t.CExtern`：`x: t.CExtern | t.CInt` | +| `c.CIfdef`/`c.CEndif()` 等预处理指令 | 保留原样 | +| `import` 语句 | 保留原样 | + +**关键设计**：`t.CDefine` 常量和 `t.CDefine` 函数在存根中**保留完整定义**（包括赋值和函数体），因为它们是编译时常量/宏，其他模块引用时需要完整展开。普通函数只保留签名并标记 `c.State`（仅声明），因为函数体在阶段二生成。 + + +#### 步骤 3：构建结构体注册表 + +扫描所有已生成的 `.pyi` 文件，提取： +- **结构体名称集合**：所有非枚举、非异常的 `class` 定义 +- **枚举名称集合**：继承 `t.CEnum` 的类 +- **异常名称集合**：继承 `Exception` 的类 +- **结构体→SHA1 映射**：记录每个结构体定义在哪个模块中 + +这个注册表用于阶段一的声明生成和阶段二的类型解析，确保跨模块的结构体引用使用正确的 SHA1 命名空间。 + +#### 步骤 4：生成 `.stub.ll` LLVM IR 声明 + +对每个 `.pyi` 文件，由 `DeclarationGenerator` 生成对应的 LLVM IR 声明文件 `.stub.ll`。 + +**声明生成规则**： + +| 元素 | 生成的 LLVM IR | +|------|---------------| +| 普通函数 | `declare void @".foo"(i32)` — 非 CExport 函数加 SHA1 前缀 | +| CExport 函数 | `declare i32 @main()` — CExport 函数不加前缀，全局可见 | +| 结构体 | `%".ClassName" = type { i32, i32* }` — 类型名加 SHA1 前缀 | +| 全局变量 | `@varname = external global i32` | +| `t.CDefine` 常量 | **不生成声明**（编译时常量，直接内联展开） | +| `t.CTypedef` 别名 | **不生成声明**（类型别名，在类型解析时展开） | +| 枚举 | 为每个枚举成员生成 `@__config_EnumName_member = external global i32` | + +**增量编译**：如果 `.pyi` 或 `.stub.ll` 已存在，则跳过生成（缓存命中）。只有源文件内容变化导致 SHA1 变化时才重新生成。 + +### 阶段二：代码翻译（Phase2Translator） + +阶段二使用阶段一生成的声明接口，将源文件翻译为包含完整代码的 LLVM IR。 + +#### 步骤 1：加载声明接口 + +加载 `temp/` 目录中所有的 `.pyi` 和 `.stub.ll` 文件，构建： +- `sig_files`：SHA1 → `.pyi` 文件路径映射 +- `stub_files`：SHA1 → `.stub.ll` 文件路径映射 +- `sha1_map`：SHA1 → 源文件相对路径映射 + +#### 步骤 2：构建共享符号表 + +一次性构建所有文件共享的符号表数据，避免每个文件重复加载。将所有 `.pyi` 存根和 `.stub.ll` 声明中的类型信息注册到统一的 `SymbolTable` 中。 + +#### 步骤 3：收集内联函数符号 + +扫描 `includes/` 目录中被引用的模块，收集标记为 `t.CInline` 的函数。内联函数的完整 AST body 被保存，在翻译调用点时直接展开。值得注意的是，和 C 语言不同，`t.CInline` 不是优化建议，而是强制性的。 + +#### 步骤 4：翻译源文件 + +对每个源文件，使用 `TransPyC` 翻译器将 Python AST 翻译为 LLVM IR： +1. 将对应模块的 `.stub.ll` 声明嵌入到生成的 `.ll` 文件头部 +2. 所有被引用模块的 `.stub.ll` 声明也被嵌入 +3. 翻译 AST 节点为 LLVM IR 指令 +4. 输出 `.ll` 文件 + +#### 步骤 5：编译与链接 + +- `llc`：将每个 `.ll` 编译为 `.o` 目标文件 +- `ld.lld`：将所有 `.o` 链接为最终可执行文件 + +## SHA1 命名空间机制 + +SHA1 命名空间是 Viper 解决跨模块符号冲突的核心机制。 + +### 问题 + +在 C 语言中，多个模块可能定义同名函数或结构体，导致链接时符号冲突。C 的解决方案是 `static`（限制可见性）和命名前缀（手动避免冲突）。 + +### Viper 的解决方案 + +Viper 使用源文件内容的 SHA1 哈希作为命名空间前缀： + +``` +源文件 A.py（SHA1 = a1b2c3d4e5f6g7h8）中定义： + class Point: x: t.CInt; y: t.CInt + def draw(p: Point) -> t.CVoid: ... + +源文件 B.py（SHA1 = i9j0k1l2m3n4o5p6）中也定义： + class Point: x: t.CFloat; y: t.CFloat # 不同的结构体！ + def draw(p: Point) -> t.CVoid: ... + +编译后： + A.py 的结构体 → %"a1b2c3d4e5f6g7h8.Point" = type { i32, i32 } + A.py 的函数 → declare void @"a1b2c3d4e5f6g7h8.draw"(%"a1b2c3d4e5f6g7h8.Point"*) + + B.py 的结构体 → %"i9j0k1l2m3n4o5p6.Point" = type { float, float } + B.py 的函数 → declare void @"i9j0k1l2m3n4o5p6.draw"(%"i9j0k1l2m3n4o5p6.Point"*) +``` + +**没有符号冲突**——即使两个模块定义了同名类型和函数，它们的 LLVM IR 符号也是不同的。即便有两个内容完全相同的文件也不会冲突，他们会被识别为同一个文件，然后进行单次编译。 + +### SHA1 前缀的豁免：`t.CExport` + +标记为 `t.CExport` 的函数**不加 SHA1 前缀**，保持原始函数名。这是模块向外部暴露 API/ABI 的机制： + +```python +# main.py — 入口函数，必须全局可见 +def main() -> t.CInt | t.CExport: + return 0 + +# 编译后：define i32 @main() — 无 SHA1 前缀 +``` + +```python +# serial.py — 驱动接口，对外暴露 +def init() -> t.CVoid | t.CExport: + serial_puts("init\n") + +# 编译后：define void @init() — 无 SHA1 前缀，其他模块可直接调用 +``` + +### SHA1 与增量编译 + +SHA1 同时服务于增量编译： +- 源文件内容不变 → SHA1 不变 → `.pyi` 和 `.stub.ll` 缓存命中，跳过阶段一 +- 源文件内容变化 → SHA1 变化 → 重新生成声明接口 +- `temp/_sha1_map.txt` 记录 SHA1 → 源文件路径的映射，供阶段二加载 + +理论上，如果一个未经变动的模块的依赖路径上存在变动的模块，暂时的方法是将其简单做 SHA1 替换，暨将旧 SHA1 替换为新 SHA1，但不适用于签名改变的情况，不过签名改变这个未经变动的模块必然也需要改变，但是由于宏展开的存在，不能完全保证不会发生错误情况。且依赖路径的 SHA1 替换成功性未经完整测试，可能出现未定义行为。 + +## `t.CDefine` 深度解析 + +`t.CDefine` 是 Viper 中最特殊的类型——它不是数据类型，而是**编译时元指令**，控制编译器的代码生成行为。 + +### `t.CDefine` 常量 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +CPU_FEATURE_FPU: t.CDefine = (1 << 0) +``` + +**编译行为**： +1. **阶段一**：存根生成器保留 `t.CDefine` 常量的完整定义（包括赋值值），因为其他模块可能引用此常量 +2. **阶段一**：`.stub.ll` 声明生成器**跳过** `t.CDefine` 常量，不生成 `external global` 声明 +3. **阶段二**：翻译器遇到 `t.CDefine` 常量的引用时，直接内联其值，不生成任何加载指令 + +这意味着 `t.CDefine` 常量在 LLVM IR 层面**不存在**——它们在编译时被完全展开，等价于 C 的 `#define` 宏。 + +### `t.CDefine` 函数 + +当函数的返回类型注解包含 `t.CDefine` 时，该函数被编译器视为**宏函数**： + +```python +def MAKE_FLAG(bit) -> t.CDefine: + return (1 << bit) + +def GET_PAGE_ORDER(size) -> t.CDefine: + return size // PAGE_SIZE +``` + +**编译行为**： +1. **阶段一**：存根生成器保留 `t.CDefine` 函数的**完整函数体**（不像普通函数那样只保留签名） +2. **阶段二**：翻译器遇到 `t.CDefine` 函数的调用时，将调用内联展开为函数体的计算结果 + +`t.CDefine` 函数在 LLVM IR 中**不生成函数定义**——它们是纯粹的编译时宏。 + +### `t.CDefine` 与类型组合 + +`t.CDefine` 可以与具体类型组合使用，为常量指定底层类型： + +```python +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +DATA_SEG: t.CDefine | t.CUInt16T = 0x10 +``` + +这表示常量在类型检查时被视为 `t.CUInt16T`（`uint16_t`），但在代码生成时仍然内联展开。 + +## `t.CExport` 深度解析 + +`t.CExport` 控制函数的符号可见性，是 SHA1 命名空间的豁免机制。 + +### 语义 + +| 修饰 | LLVM IR 函数名 | 链接可见性 | 用途 | +|------|---------------|-----------|------| +| 无修饰 | `@.funcname` | 模块内部 | 模块私有函数 | +| `t.CExport` | `@funcname` | 全局可见 | 对外暴露的 API | +| `t.CExtern` | `@funcname`（仅声明） | 外部定义 | 引用外部函数 | + +### 使用场景 + +```python +# 入口函数 — 必须全局可见，链接器需要找到它 +def _start() -> t.CInt | t.CExport: + return kernel_main() + +# 驱动接口 — 其他模块/应用需要调用 +def init() -> t.CVoid | t.CExport: + uart_init() + +# 内部辅助函数 — 不需要全局可见，自动加 SHA1 前缀 +def _helper() -> t.CInt: + return 42 +``` + +### `t.CExport` 在存根中的表现 + +在 `.pyi` 存根文件中，`t.CExport` 函数与普通函数一样只保留签名（添加 `c.State`，表示单纯声明），但 `t.CExport` 标记被保留在返回类型注解中。阶段一的 `DeclarationGenerator` 检查 `t.CExport` 标记来决定是否添加 SHA1 前缀。 + +## `t.CInline` 深度解析 + +`t.CInline` 标记函数为内联函数，编译器在调用点直接展开函数体。 + +### 语义 + +```python +def fast_add(a: t.CInt, b: t.CInt) -> t.CInt | t.CInline: + return a + b +``` + +**编译行为**： +1. **阶段二预收集**：`_collect_inline_symbols` 扫描 `includes/` 目录中被引用的模块，收集所有 `t.CInline` 函数的 AST body +2. **翻译时展开**：遇到内联函数调用时，将函数体的 AST 直接嵌入调用点，替换参数为实际值 +3. **不生成独立函数**：内联函数不生成 LLVM IR 函数定义（除非也被非内联调用） + +### `t.CInline` 与 `t.CDefine` 函数的区别 + +| 特性 | `t.CInline` | `t.CDefine` 函数 | +|------|------------|-----------------| +| 展开时机 | 阶段二翻译时 | 阶段二翻译时 | +| 类型检查 | 完整的参数和返回类型检查 | 返回类型为宏，类型检查较弱 | +| 存根表示 | 签名 + `c.State` | 完整函数体 | +| LLVM IR | 可能生成函数定义（如果有非内联调用） | 不生成函数定义 | +| 适用场景 | 性能关键的短函数 | 编译时常量和宏计算 | + +## 目标平台 + +默认目标三元组：`x86_64-none-elf` + +数据布局：`e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128` + +## Hello World 示例 + +```python +import t, c + +def main() -> t.CInt | t.CExport: + print("Hello, ViperOS!\n") + return 0 +``` + +编译流程： + +1. **阶段一**：计算 `main.py` 的 SHA1，生成 `.pyi` 存根（`def main() -> t.CInt | t.CExport | c.State: pass`），生成 `.stub.ll` 声明（`declare i32 @main()`，因为是 CExport 不加前缀） +2. **阶段二**：翻译 `main.py` 为 `.ll`，嵌入 `print` 对应的 `puts`/`printf` 声明，生成 `define i32 @main()` 函数体 +3. **编译**：`llc` 将 `.ll` 编译为 `.o` +4. **链接**：`ld.lld` 链接为可执行文件 + + +# 02 - 类型系统 + +Viper 的类型系统通过 `t` 模块提供，所有类型注解继承 `t.CType` 。大都以 `t.C{TypeName:PascalCase}`为类型，类型注解是 Viper 的核心机制，决定了变量的内存布局、大小和对齐方式。 + +## 基本类型 + +### 整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CChar` | `char` | 8 | 是 | +| `t.CShort` | `short` | 16 | 是 | +| `t.CInt` | `int` | 32 | 是 | +| `t.CLong` | `long` | 64 | 是 | +| `t.CUnsignedChar` | `unsigned char` | 8 | 否 | +| `t.CUnsignedShort` | `unsigned short` | 16 | 否 | +| `t.CUnsigned` / `t.CUnsignedInt` | `unsigned int` | 32 | 否 | +| `t.CUnsignedLong` | `unsigned long` | 64 | 否 | +| `t.CSignedChar` | `signed char` | 8 | 是 | + +值得注意的是，t.CUnsigned 一般不单独使用，而是配合其它有符号类型，替换表示无符号，比如 `t.CUnsigned | t.CInt`，由于为了便于使用一些常用类型，则将其组合为常用的 `t.CUnginedInt` 等。 + +### 固定宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CInt8T` | `int8_t` | 8 | 是 | +| `t.CInt16T` | `int16_t` | 16 | 是 | +| `t.CInt32T` | `int32_t` | 32 | 是 | +| `t.CInt64T` | `int64_t` | 64 | 是 | +| `t.CUInt8T` | `uint8_t` | 8 | 否 | +| `t.CUInt16T` | `uint16_t` | 16 | 否 | +| `t.CUInt32T` | `uint32_t` | 32 | 否 | +| `t.CUInt64T` | `uint64_t` | 64 | 否 | + +### 最小宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntLeast8T` | `int_least8_t` | ≥8 | 是 | +| `t.CIntLeast16T` | `int_least16_t` | ≥16 | 是 | +| `t.CIntLeast32T` | `int_least32_t` | ≥32 | 是 | +| `t.CIntLeast64T` | `int_least64_t` | ≥64 | 是 | +| `t.CUIntLeast8T` | `uint_least8_t` | ≥8 | 否 | +| `t.CUIntLeast16T` | `uint_least16_t` | ≥16 | 否 | +| `t.CUIntLeast32T` | `uint_least32_t` | ≥32 | 否 | +| `t.CUIntLeast64T` | `uint_least64_t` | ≥64 | 否 | + +### 最快最小宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntFast8T` | `int_fast8_t` | ≥8 | 是 | +| `t.CIntFast16T` | `int_fast16_t` | ≥16 | 是 | +| `t.CIntFast32T` | `int_fast32_t` | ≥32 | 是 | +| `t.CIntFast64T` | `int_fast64_t` | ≥64 | 是 | +| `t.CUIntFast8T` | `uint_fast8_t` | ≥8 | 否 | +| `t.CUIntFast16T` | `uint_fast16_t` | ≥16 | 否 | +| `t.CUIntFast32T` | `uint_fast32_t` | ≥32 | 否 | +| `t.CUIntFast64T` | `uint_fast64_t` | ≥64 | 否 | + +### 最大宽度整数类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CIntMaxT` | `intmax_t` | ≥64 | 是 | +| `t.CUIntMaxT` | `uintmax_t` | ≥64 | 否 | + +### stdint 短名别名 + +通过 `import stdint` 引入，这些短名是上述类型的 `t.CTypedef` 别名，便于在系统编程场景中快速使用。允许直接使用 `from stdint import *` 来直接使用。 + +#### 基本类型短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.INT` | `t.CInt` | - | +| `stdint.UINT` | `t.CUnsignedInt` | - | +| `stdint.BOOL` | `t.CInt` | 1 为真，0 为假 | +| `stdint.SHORT` | `t.CShort` | - | +| `stdint.USHORT` | `t.CUnsignedShort` | - | +| `stdint.LONG` | `t.CLong` | - | +| `stdint.ULONG` | `t.CUnsignedLong` | - | +| `stdint.LONGLONG` | `t.CLong \| t.CLong` | - | +| `stdint.ULONGLONG` | `t.CUnsignedLong \| t.CLong` | - | +| `stdint.FLOAT` | `t.CFloat` | - | +| `stdint.DOUBLE` | `t.CDouble` | - | + +#### 固定宽度短名 + +| 短名 | 等价展开 | +|------|---------| +| `stdint.INT8` | `t.CInt8T` | +| `stdint.INT16` | `t.CInt16T` | +| `stdint.INT32` | `t.CInt32T` | +| `stdint.INT64` | `t.CInt64T` | +| `stdint.UINT8` | `t.CUInt8T` | +| `stdint.UINT16` | `t.CUInt16T` | +| `stdint.UINT32` | `t.CUInt32T` | +| `stdint.UINT64` | `t.CUInt64T` | + +#### 字节与字短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.BYTE` | `t.CUnsignedChar` | 8 位无符号 | +| `stdint.WORD` | `t.CUInt16T` | 16 位无符号 | +| `stdint.DWORD` | `t.CUInt32T` | 32 位无符号 | +| `stdint.QWORD` | `t.CUInt64T` | 64 位无符号 | + +#### 字符类型短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.TCHAR` | `t.CChar` | - | +| `stdint.CHARLIST` | `str \| t.CPtr` / `list[str, None]` | - | +| `stdint.WCHAR` | `stdint.WORD` | UTF-16 代码单元 | +| `stdint.CHAR8` | `t.CChar8T` | - | +| `stdint.CHAR16` | `t.CChar16T` | - | +| `stdint.CHAR32` | `t.CChar32T` | - | + +#### 指针短名 + +| 短名 | 等价展开 | +|------|---------| +| `stdint.INTPTR` | `t.CInt \| t.CPtr` | +| `stdint.UINTPTR` | `t.CUnsignedInt \| t.CPtr` | +| `stdint.BYTEPTR` | `stdint.BYTE \| t.CPtr` | +| `stdint.SHORTPTR` | `t.CShort \| t.CPtr` | +| `stdint.USHORTPTR` | `t.CUnsignedShort \| t.CPtr` | +| `stdint.WCHARPTR` | `stdint.WORD \| t.CPtr` | +| `stdint.VOIDPTR` | `t.CVoid \| t.CPtr` | +| `stdint.INT8PTR` | `t.CInt8T \| t.CPtr` | +| `stdint.INT16PTR` | `t.CInt16T \| t.CPtr` | +| `stdint.INT32PTR` | `t.CInt32T \| t.CPtr` | +| `stdint.INT64PTR` | `t.CInt64T \| t.CPtr` | +| `stdint.UINT8PTR` | `t.CUInt8T \| t.CPtr` | +| `stdint.UINT16PTR` | `t.CUInt16T \| t.CPtr` | +| `stdint.UINT32PTR` | `t.CUInt32T \| t.CPtr` | +| `stdint.UINT64PTR` | `t.CUInt64T \| t.CPtr` | +| `stdint.CHAR8PTR` | `t.CChar8T \| t.CPtr` | +| `stdint.CHAR16PTR` | `t.CChar16T \| t.CPtr` | +| `stdint.CHAR32PTR` | `t.CChar32T \| t.CPtr` | + +#### 平台特定短名 + +| 短名 | 等价展开 | 说明 | +|------|---------|------| +| `stdint.FSIZE_t` | `stdint.DWORD` | 文件大小 | +| `stdint.LBA_t` | `stdint.DWORD` | 逻辑块地址 | + +### 平台相关类型 + +| Viper 类型 | C 等价 | 大小（位） | 有符号 | +|------------|--------|-----------|--------| +| `t.CSizeT` | `size_t` | 64 | 否 | +| `t.CIntPtrT` | `intptr_t` | 64 | 是 | +| `t.CUIntPtrT` | `uintptr_t` | 64 | 否 | +| `t.CPtrDiffT` | `ptrdiff_t` | 64 | 是 | + +### stdint 短名别名 + +Viper 在 `stdint` 模块中提供了更简短的类型别名，通过 `t.CTypedef` 映射到对应的完整类型。使用时需 `import stdint`。 + +#### 固定宽度整数短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT8` | `t.CInt8T` | `int8_t` | +| `INT16` | `t.CInt16T` | `int16_t` | +| `INT32` | `t.CInt32T` | `int32_t` | +| `INT64` | `t.CInt64T` | `int64_t` | +| `UINT8` | `t.CUInt8T` | `uint8_t` | +| `UINT16` | `t.CUInt16T` | `uint16_t` | +| `UINT32` | `t.CUInt32T` | `uint32_t` | +| `UINT64` | `t.CUInt64T` | `uint64_t` | + +#### 固定宽度指针短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT8PTR` | `t.CInt8T \| t.CPtr` | `int8_t*` | +| `INT16PTR` | `t.CInt16T \| t.CPtr` | `int16_t*` | +| `INT32PTR` | `t.CInt32T \| t.CPtr` | `int32_t*` | +| `INT64PTR` | `t.CInt64T \| t.CPtr` | `int64_t*` | +| `UINT8PTR` | `t.CUInt8T \| t.CPtr` | `uint8_t*` | +| `UINT16PTR` | `t.CUInt16T \| t.CPtr` | `uint16_t*` | +| `UINT32PTR` | `t.CUInt32T \| t.CPtr` | `uint32_t*` | +| `UINT64PTR` | `t.CUInt64T \| t.CPtr` | `uint64_t*` | + +#### 字符类型短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `CHAR8` | `t.CChar8T` | `char8_t` | +| `CHAR16` | `t.CChar16T` | `char16_t` | +| `CHAR32` | `t.CChar32T` | `char32_t` | +| `CHAR8PTR` | `t.CChar8T \| t.CPtr` | `char8_t*` | +| `CHAR16PTR` | `t.CChar16T \| t.CPtr` | `char16_t*` | +| `CHAR32PTR` | `t.CChar32T \| t.CPtr` | `char32_t*` | + +#### 基本类型短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `INT` | `t.CInt` | `int` | +| `UINT` | `t.CUnsignedInt` | `unsigned int` | +| `BOOL` | `t.CInt` | `int`（0/1） | +| `SHORT` | `t.CShort` | `short` | +| `USHORT` | `t.CUnsignedShort` | `unsigned short` | +| `LONG` | `t.CLong` | `long` | +| `ULONG` | `t.CUnsignedLong` | `unsigned long` | +| `LONGLONG` | `t.CLong \| t.CLong` | `long long` | +| `ULONGLONG` | `t.CUnsignedLong \| t.CLong` | `unsigned long long` | +| `FLOAT` | `t.CFloat` | `float` | +| `DOUBLE` | `t.CDouble` | `double` | + +#### 平台 / Windows 风格短名 + +| 短名 | 等价 Viper 类型 | C 等价 | +|------|----------------|--------| +| `BYTE` | `t.CUnsignedChar` | `unsigned char`（8位） | +| `BYTEPTR` | `BYTE \| t.CPtr` | `unsigned char*` | +| `WORD` | `t.CUInt16T` | `uint16_t`（16位） | +| `DWORD` | `t.CUInt32T` | `uint32_t`（32位） | +| `QWORD` | `t.CUInt64T` | `uint64_t`（64位） | +| `INTPTR` | `t.CInt \| t.CPtr` | `int*` | +| `UINTPTR` | `t.CUnsignedInt \| t.CPtr` | `unsigned int*` | +| `SHORTPTR` | `t.CShort \| t.CPtr` | `short*` | +| `USHORTPTR` | `t.CUnsignedShort \| t.CPtr` | `unsigned short*` | +| `VOIDPTR` | `t.CVoid \| t.CPtr` | `void*` | +| `TCHAR` | `t.CChar` | `char` | +| `WCHAR` | `WORD` | `uint16_t`（UTF-16） | +| `WCHARPTR` | `WORD \| t.CPtr` | `uint16_t*`（UTF-16） | +| `CHARLIST` | `str \| t.CPtr` | `char*` | +| `FSIZE_t` | `DWORD` | `uint32_t` | +| `LBA_t` | `DWORD` | `uint32_t` | + +### 浮点类型 + +| Viper 类型 | C 等价 | 大小（位） | +|------------|--------|-----------| +| `t.CFloat` | `float` | 32 | +| `t.CDouble` | `double` | 64 | + +### 其他基本类型 + +| Viper 类型 | C 等价 | 说明 | +|------------|--------|------| +| `t.CVoid` | `void` | 空类型 | +| `t.CBool` | `bool` | 布尔类型（8位） | +| `t.CWCharT` | `wchar_t` | 宽字符（32位） | +| `t.CChar8T` | `char8_t` | UTF-8 字符（8位，无符号） | +| `t.CChar16T` | `char16_t` | UTF-16 字符（16位，无符号） | +| `t.CChar32T` | `char32_t` | UTF-32 字符（32位，无符号） | + +### Python 便捷映射类型 +| Python 类型| Viper 类型 | Viper 值 | +|--------|--------|------| +|`int`|`t.CInt`|-| +|`float`|`t.CFloat`|-| +|`str`|`t.CChar \| t.CPtr`|-| +|`bool`|`t.CBool`|-| +|`True`|`t.CBool`|`1`| +|`False`|`t.CBool`|`0`| +|`None`|`NULL`|`t.CIntPtr(0)`| + +对于 t.py 和 c.py 的引用，必须使用 import，不能使用 from ... import ...，也不能使用 as，因为这两者都是直接以字符串识别的。 + +## 类型组合 + +Viper 使用 Python 的 `|` 来组合类型修饰符（TypeUnion），其仅在左值注解中有效，其它位置均为位或语义（至少重载前是这样的），这是 Viper 最重要的语法特性之一。 + +### 指针类型 + +使用 `t.CPtr` 表示指针，通过 `|` 与基本类型组合： + +```python +x: t.CInt | t.CPtr # int* — 指向 int 的指针 +p: t.CVoid | t.CPtr # void* — 通用指针 +s: t.CChar | t.CPtr # char* — 字符串指针 +pp: t.CInt | t.CPtr | t.CPtr # int** — 指向指针的指针 +``` + +等价 C 代码： +```c +int* x; +void* p; +char* s; +int** pp; +``` + +同时，指针也可以用 `Ptr[T]` 的方法表示，但在 Viper 中不太推荐，虽然说这样做一直是类似工程的常见用法，但是如果此处 `T` 是一个多态，同时我们想要复用 Python 的高亮插件，插件无法识别 `Ptr[T]` 的属性取值，但是对于 `T | Ptr` 可以。 + +```python +x: t.CPtr[t.CInt] +p: t.CPtr[t.CVoid] +s: t.CPtr[t.CChar] +pp: t.CPtr[t.CPtr[t.CInt]] +pp2: t.CInt | t.CPtr[t.CPtr] # 同时也可以这么表示 +``` + +等价 C 代码： + +```c +int* x; +void* p; +char* s: +int** pp; +int** pp2; +``` + +### 类型限定符组合 + +```python +x: t.CConst | t.CInt # const int +p: t.CConst | t.CChar | t.CPtr # const char* +``` + +### 存储类组合 + +```python +x: t.CStatic | t.CInt # static int +f: t.CExtern | t.CInt # extern int（声明其存在于外部） +``` + +### 函数返回类型组合 + +```python +def foo() -> t.CInt | t.CExport: # 导出函数，返回 int + return 0 + +def bar() -> t.CVoid | t.CExtern: # 外部函数声明 + pass +``` + +## 数组类型 + +使用 Python 的 `list[ElementType, Size]` 语法表示固定大小数组（通常的，这用于栈上和 BSS 段上）： + +```python +buf: list[t.CChar, 64] # char buf[64] +ids: list[t.CInt, 10] # int ids[10] +matrix: list[t.CFloat, 16] # float matrix[16] +entries: list[idt_entry, 256] # struct idt_entry entries[256] +s: list[t.CChar, None] # 自推长度 char s[]; +``` + +等价 C 代码： +```c +char buf[64]; +int ids[10]; +float matrix[16]; +struct idt_entry entries[256]; +``` + +### 数组指针和函数指针 + +```python +lp: list[t.CInt, 12] | t.CPtr +vp: t.Callable[[t.CInt, t.CInt], t.CInt] = x # 其中 t.Callable = typing.Callable，可以直接使用。 +``` + +## 结构体类型 + +使用 `class` 定义结构体，成员通过类型注解声明： + +```python +class point: + x: t.CInt + y: t.CInt +``` + +等价 C 代码： +```c +struct point { + int x; + int y; +}; +``` + +### 带指针成员的结构体 + +```python +class node: + value: t.CInt + next: node | t.CPtr # struct node* next +``` + +### 匿名结构体成员（❌） + +> WARNING: **警告：** 由于 Viper 在声明结构体，枚举等处不需要关键字，因此 `t.Anonymouse` 是不需要使用的。此关键字不再维护，强行使用可能导致未定义行为。 + +继承 `t.Anonymous` 的类作为匿名成员嵌入： + +```python +class header(t.Anonymous): + magic: t.CUInt32T + version: t.CUInt32T + +class packet: + header # 匿名嵌入，可直接访问 packet.magic + length: t.CUInt32T +``` + +### 字节序标记 + +> WARNING: **警告：** 此关键字是实验性的。未经充分测试，不当使用可能导致未定义行为。 + +使用 `t.BigEndian` 或 `t.LittleEndian` 标记成员的字节序： + +```python +class network_packet: + length: t.CUInt16T | t.BigEndian # 大端序存储 + type: t.CUInt8T # 默认小端序 +``` + +## 联合体类型 + +继承 `t.CUnion` 定义联合体： + +```python +class value(t.CUnion): + ival: t.CInt + fval: t.CFloat + pval: t.CVoid | t.CPtr +``` + +等价 C 代码： +```c +union value { + int ival; + float fval; + void* pval; +}; +``` + +## 枚举类型 + +继承 `t.CEnum` 定义枚举： + +```python +class color(t.CEnum): + RED: t.CEnum = 0 + GREEN: t.CEnum = 1 + BLUE: t.CEnum = 2 +``` + +等价 C 代码： +```c +enum color { + RED = 0, + GREEN = 1, + BLUE = 2 +}; +``` + +枚举成员可直接使用名称： + +```python +c: t.CInt = color.RED +``` + +## Typedef + +使用 `t.CTypedef` 创建类型别名： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] # typedef int (*irq_handler_t)(); +``` + +或通过注解赋值： + +```python +MyStruct: t.CTypedef = original_struct # typedef struct original_struct MyStruct; +``` + +更推荐通过后者的形式，前者编译器不一定能够识别到注解是对 t.CTypedef 的，还是后面的注解的，而且容易触发未定义行为。 + +## 强制类型转换 +在 Viper 中的强制类型转换有些不同，强制类型转换有两种方案，都是基于 `t.CType` 的用法，符合 Python 本意。 +对于原先就继承自 `t.CType` 的类型来说，可以直接使用 call 的方法进行类型转换，比如 +```python +x: t.CChar | t.CPtr = "Hello" +t.CVoid(x, t.CPtr) +``` +其中用法为 `t.CType(x, ...)`，此处...可以是其它的 `t.CType` 或将要提到的 `Typedef`，将其视为组合类型对 `x` 进行强制转换。 + +额外补充一句，对于比较，四则运算，可以不使用强制类型转换，而是依靠隐式类型转换，但如果是严谨数学计算，推荐还是强制类型转换避免判断出错。 + +此外，若将一个不同类型的右值赋值到左值，则会隐式将右值转换为左值的类型。 + +如果你不喜欢 `t.CVoid(x, t.CPtr)` 的方法，认为这将 `void` 和 `ptr` 分开，不便于理解，那么还有一个方法，即将二者打包为一个 Typedef，即： +```python +VOIDPTR: t.CTypedef = t.CVoid | t.CPtr +``` +则如果你想，你可以直接通过 call 的方式进行强转，比如： +```python +y: VOIDPTR = VOIDPTR(x) +``` + +## 宏定义常量 + +使用 `t.CDefine` 定义编译时常量。`t.CDefine` 不是数据类型，而是编译时元指令——标记为 `t.CDefine` 的常量在 LLVM IR 中不生成任何全局变量，而是在引用处直接内联展开。详见 [01-overview.md 中 t.CDefine 深度解析](01-overview.md#tcdefine-深度解析)。 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +``` + +语义上等价于 C 的 `#define`： +```c +#define MAX_SIZE 1024 +#define PAGE_SIZE 4096 +#define FA_READ 0x01 +``` + +宏定义支持表达式： + +```python +CPU_FEATURE_FPU: t.CDefine = (1 << 0) +CPU_FEATURE_APIC: t.CDefine = (1 << 6) +``` + +宏定义也可与类型组合使用： + +```python +CODE_SEG: t.CDefine | t.CUInt16T = 0x08 +``` + +## 位域 + +使用 `t.Bit(n)` 定义位域成员： + +```python +class flags: + a: t.CUInt32T | t.Bit[1] # 此处，如果存入 2，二进制是 10，会被截断为 0 + b: t.CUInt32T | t.Bit[3] + c: t.CUInt32T | t.Bit[4] +``` + +## LLVM IR 类型简写 + +Viper 提供了 LLVM IR 级别的类型简写： + +```python +i1 # 1位整数 +i8 # 8位有符号整数 +i16 # 16位有符号整数 +i32 # 32位有符号整数 +i64 # 64位有符号整数 +u1 # 1位无符号整数 +u8 # 8位无符号整数 +u16 # 16位无符号整数 +u32 # 32位无符号整数 +u64 # 64位无符号整数 +``` + +仍然通过 `t.x` 的方式调用，比如 `t.i1` 等。 + +## `__attribute__` 属性 + +通过 `t.attr` 命名空间访问 GCC `__attribute__` 属性： + +```python +t.attr.packed() # __attribute__((packed)) +t.attr.aligned(16) # __attribute__((aligned(16))) +t.attr.section(".text.startup") # __attribute__((section(".text.startup"))) +t.attr.always_inline() # __attribute__((always_inline)) +t.attr.noinline() # __attribute__((noinline)) +t.attr.noreturn() # __attribute__((noreturn)) +t.attr.weak() # __attribute__((weak)) +t.attr.unused() # __attribute__((unused)) +t.attr.constructor() # __attribute__((constructor)) +t.attr.destructor() # __attribute__((destructor)) +t.attr.deprecated("use X") # __attribute__((deprecated("use X"))) +``` + +### LLVM 函数属性 + +通过 `t.attr.llvm` 命名空间指定 LLVM 函数属性： + +```python +t.attr.llvm.nobuiltin +t.attr.llvm.nounwind +t.attr.llvm.noredzone +t.attr.llvm.willreturn +t.attr.llvm.mustprogress +t.attr.llvm.optnone +t.attr.llvm.noinline +t.attr.llvm.alwaysinline +t.attr.llvm.readnone +t.attr.llvm.readonly +t.attr.llvm.writeonly +t.attr.llvm.inaccessiblememonly +t.attr.llvm.inaccessiblemem_or_argmemonly +``` + +在函数返回类型注解中使用： + +```python +def foo() -> t.CInt | t.attr.llvm.nobuiltin | t.attr.llvm.nounwind: + return 0 +``` + + +# 03 - 变量声明与赋值 + +Viper 的变量声明使用 Python 的类型注解语法，通过 `t` 模块指定 C 级别的类型信息。 + +## 带注解的变量声明 + +使用 `变量名: 类型` 语法声明变量： + +```python +x: t.CInt = 0 # int x = 0; +y: t.CUInt32T = 42 # uint32_t y = 42; +name: list[t.CChar, 64] # char name[64]; +ptr: t.CVoid | t.CPtr # void* ptr; +msg: t.CConst | str = "Hello" # const char* msg = "Hello"; +``` + +### 无初始值的声明 + +未赋值的变量声明会生成零初始化的变量，此处是未定义行为，即在不同环境位置可能会覆盖0，也可能不会，最好预先复制或用 memset 清除： + +```python +count: t.CInt # int count = 0; +buffer: list[t.CChar, 256] # char buffer[256] = {0}; +``` + +### 延迟赋值 + +变量可以先声明后赋值： + +```python +x: t.CInt +x = 10 +``` + +### 指针变量 + +```python +p: t.CInt | t.CPtr # int* p; +fb: t.CVoid | t.CPtr # void* fb; +str_ptr: t.CConst | t.CChar | t.CPtr # const char* str_ptr; +``` + +### 指针变量的特殊初始化 + +使用 `t.CVoid(0, t.CPtr)` 初始化指针为 NULL： + +```python +p: t.CVoid | t.CPtr = t.CVoid(0, t.CPtr) # void* p = NULL; +``` + +或使用 `None`： + +```python +p: t.CVoid | t.CPtr = None # void* p = NULL; +``` + +## 普通赋值 + +```python +x: t.CInt = 0 +x = 42 # x = 42; +``` + +## 增强赋值 + +支持所有 Python 增强赋值运算符： + +```python +x: t.CInt = 0 +x += 1 # x += 1; +x -= 1 # x -= 1; +x *= 2 # x *= 2; +x //= 3 # x /= 3; (整数除法) +x %= 5 # x %= 5; +x <<= 1 # x <<= 1; +x >>= 1 # x >>= 1; +x |= 0xFF # x |= 0xFF; +x &= 0x0F # x &= 0x0F; +x ^= 0xAA # x ^= 0xAA; +``` + +## 全局变量 + +模块顶层声明的变量即为全局变量： + +```python +kbd_tid: t.CInt = -1 # 全局 int kbd_tid = -1; +BootInfo: bootinfo | t.CPtr # 全局 struct bootinfo* BootInfo; +``` + +### 静态全局变量 + +```python +x: t.CStatic | t.CInt = 0 # static int x = 0; +``` + +### 外部声明 + +```python +x: t.CExtern | t.CInt # extern int x; +``` + +## 常量定义 + +使用 `t.CDefine` 定义编译时常量。`t.CDefine` 不是数据类型，而是编译时元指令——常量在 LLVM IR 中不生成全局变量，而是在引用处直接内联展开。详见 [01-overview.md 中 t.CDefine 深度解析](01-overview.md#tcdefine-深度解析)。 + +```python +MAX_SIZE: t.CDefine = 1024 +PAGE_SIZE: t.CDefine = 4096 +FA_READ: t.CDefine = 0x01 +FA_WRITE: t.CDefine = 0x02 +``` + +## 字符串常量 + +Viper 中的字符串字面量编译为 C 的字符串常量（`const char*`）： + +```python +msg: t.CConst | str = "Hello, World!" +serial.puts(msg) +``` + +也可以直接传递字符串字面量： + +```python +serial.puts("Hello, World!") +``` + +## 类型转换 + +使用类型构造函数进行显式类型转换： + +```python +x: t.CInt = 42 +y: t.CUInt32T = t.CUInt32T(x) # (uint32_t)x +f: t.CFloat = float(x) # (float)x +i: t.CInt = int(f) # (int)f +``` + +### 指针类型转换 + +```python +ptr: t.CVoid | t.CPtr = some_ptr +int_val: t.CUInt64T = t.CUInt64T(ptr) # (uint64_t)ptr — 指针转整数 +``` + +## 结构体实例化 + +> 此处不理解参见 ([05-classes.md 中 结构体实例化与初始化](05-classes.md#结构体实例化与初始化)) + +直接使用类名作为构造函数： + +```python +info: fat32_types.fat32_fileinfo +dp: fat32_types.fat32_dirobj +``` + +带初始化的实例化： + +```python +obj: MyClass = MyClass(arg1, arg2) +``` + +使用 `c.Addr` 获取实例指针： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass(arg1, arg2)) +``` + +## delete 语句 + +`del` 语句可用于调用对象的 `__del__` 或 `__delete__` 方法： + +```python +del obj # 调用 obj.__del__() 或 obj.__delete__() +del arr[idx] # 调用 arr.__delitem__(idx) +``` + + +# 04 - 函数定义与调用 + +Viper 的函数定义使用 Python 的 `def` 语法，通过类型注解指定参数类型和返回类型。 + +## 基本函数定义 + +```python +def add(a: t.CInt, b: t.CInt) -> t.CInt: + return a + b +``` + +等价 C 代码： +```c +int add(int a, int b) { + return a + b; +} +``` + +## 无返回值函数 + +```python +def print_msg(msg: t.CConst | t.CChar | t.CPtr) -> t.CVoid: + serial.puts(msg) +``` + +等价 C 代码： +```c +void print_msg(const char* msg) { + serial_puts(msg); +} +``` + +## 函数修饰符 + +通过返回类型的 `|` 组合添加函数修饰符： + +### 导出函数 + +` t.CExport` 控制函数的符号可见性——标记为 `t.CExport` 的函数不加 SHA1 前缀，全局可见。详见 [01-overview.md 中 t.CExport 深度解析](01-overview.md#tcexport-深度解析)。 + +```python +def _start() -> t.CInt | t.CExport: + return 0 +``` + +### 外部函数声明 + +使用 `c.State` 表示仅声明不定义： + +```python +def isr0() -> t.CExtern | t.CVoid | c.State: pass +def isr1() -> t.CExtern | t.CVoid | c.State: pass +``` + +等价 C 代码： +```c +extern void isr0(); +extern void isr1(); +``` + +### 内联函数 + +`t.CInline` 标记函数为内联函数，编译器在调用点直接展开函数体。详见 [01-overview.md 中 t.CInline 深度解析](01-overview.md#tcinline-深度解析)。 + +```python +def fast_add(a: t.CInt, b: t.CInt) -> t.CInt | t.CInline: + return a + b +``` + +### 宏函数 + +当函数返回类型注解包含 `t.CDefine` 时，该函数被视为宏函数，不生成 LLVM IR 函数定义，在调用处内联展开。详见 [01-overview.md 中 t.CDefine 函数](01-overview.md#tcdefine-函数)。 + +```python +def MAKE_FLAG(bit) -> t.CDefine: + return (1 << bit) +``` + +### 静态函数 + +```python +def helper() -> t.CVoid | t.CStatic: + pass +``` + +## 函数属性装饰器 + +使用 `@c.Attribute` 添加 GCC `__attribute__` 属性： + +```python +@c.Attribute(t.attr.section(".text.startup"), t.attr.aligned(16)) +def _start() -> t.CInt: + return 0 +``` + +等价 C 代码： +```c +__attribute__((section(".text.startup"), aligned(16))) +int _start() { + return 0; +} +``` + +### 常用属性组合 + +```python +@c.Attribute(t.attr.always_inline()) +def hot_path() -> t.CInt: + return 0 + +@c.Attribute(t.attr.noreturn()) +def panic(msg: t.CConst | t.CChar | t.CPtr) -> t.CVoid: + serial.puts(msg) + while True: pass + +@c.Attribute(t.attr.packed) +``` + +## LLVM 函数属性 + +在返回类型注解中通过 `t.attr.llvm` 指定 LLVM 属性： + +```python +def foo() -> t.CInt | t.attr.llvm.nobuiltin | t.attr.llvm.nounwind: + return 0 +``` + +## 多返回值（CReturn） + +使用 `@c.CReturn` 装饰器实现多返回值，通过指针参数实现： + +```python +@c.CReturn(t.CInt, t.CInt) +def divmod(a: t.CInt, b: t.CInt) -> t.CVoid: + q: t.CInt = a // b + r: t.CInt = a % b + return q, r +``` + +规则： +1. `CReturn` 中的类型数量决定返回值个数 +2. 自动为每个返回类型添加指针参数（`t.CInt` → `t.CInt | t.CPtr`） +3. 参数名自动生成为 `__ReturnValue0__`、`__ReturnValue1__` 等 +4. 调用处自动传参 `&xxx` +5. 必须使用左右值赋值，返回值不能直接传入函数。 + +## 指针参数 + +```python +def read_data(buf: t.CChar | t.CPtr, size: t.CSizeT) -> t.CInt: + pass +``` + +等价 C 代码： +```c +int read_data(char* buf, size_t size); +``` + +### 输出参数模式 + +使用 `c.Addr` 传递变量地址作为输出参数： + +```python +res: t.CInt = fat32.opendir("/", c.Addr(dp)) +``` + +等价 C 代码： +```c +int res = fat32_opendir("/", &dp); +``` + +后续会使用隐式左值获取来减少 `c.Addr` 的直接使用。 + +## 函数指针类型 + +使用 `t.Callable` 定义函数指针类型： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] +``` + +等价 C 代码： +```c +typedef int (*irq_handler_t)(); +``` + +带参数的函数指针： + +```python +callback_t: t.Callable[[t.CInt, t.CInt], t.CVoid] +``` + +## 默认参数 + +Viper 支持函数默认参数值： + +```python +def create_window(x: t.CInt, y: t.CInt, w: t.CInt, h: t.CInt, title: t.CConst | str = "Window") -> t.CInt: + pass +``` + +## return 语句 + +```python +return 0 # return 0; +return # return; (void 函数) +return a, b # 多返回值（需配合 @c.CReturn） +``` + +## 全局变量声明 + +函数外的变量声明为全局变量，函数内通过 `global` 关键字访问： + +```python +BootInfo: bootinfo | t.CPtr + +def init() -> t.CVoid: + global BootInfo + BootInfo = saved_rdi +``` + +支持 func-in-func 嵌套函数。 + +## Call Func 函数调用 + +函数调用最好通过顺序调用来实现 + +```python +def add(x: int, y: int, z: int): + return (x + y) * z + +def main() -> int | t.CExport: + print(add(1, 2, 3)) # (1 + 2) * 3 = 9 +``` + +这也是 C 中的标准实现方法，但实际上，你也可以乱序传参。 + +```python +# 仍然用刚才的 add 举例 +def main() -> int | t.CExport: + print(add(z=3, x=1, y=2)) # (1 + 2) * 3 = 9 +``` + +乱序传参或带参传参对于结构体的初始化同样有效。同时，你也能够给函数指定默认值。 + +```python +def add(x: int, y: int, z: int = 1): + return (x + y) * z + +def main() -> int | t.CExport: + print(add(1, 2)) # (1 + 2) * 1 = 3 +``` + +在结构体中也可以指定默认值。 + + +# 05 - 类与数据布局 + +Viper 中的 `class` 有两种用途：定义**数据布局**（结构体/联合体/枚举，本章内容），以及定义**面向对象的对象类型**（见 [06-oop.md](06-oop.md)）。 + +## 结构体定义 + +普通 `class` 定义编译为 LLVM 结构体类型： + +```python +class point: + x: t.CInt + y: t.CInt +``` + +在 C 中： +```c +struct point { + int x; + int y; +}; +``` + +在 LLVM IR 中： +```llvm +%".point" = type { i32, i32 } +``` + +### 带属性的结构体 + +```python +@c.Attribute(t.attr.packed) +class idt_entry: + base_low: t.CUInt16T + selector: t.CUInt16T + ist_attr: t.CUInt8T + type_attr: t.CUInt8T + base_middle: t.CUInt16T + base_high: t.CUInt32T + reserved1: t.CUInt32T +``` + +### 结构体数组 + +```python +idt: list[idt_entry, 256] = [] +``` + +### 结构体指针成员 + +```python +class node: + value: t.CInt + next: node | t.CPtr # 指向自身的指针（有向图链表） +``` + +### 结构体实例化与初始化 + +```python +entry: idt_entry = idt_entry() +``` + +或带构造函数： + +```python +mousepoint = Point(x=1, y=2) +``` + +乱序传参或带参传参对于结构体的初始化同样有效： + +```python +entry: idt_entry = idt_entry(selector=0x08, base_low=0x0000) +``` + +也可以给结构体成员指定默认值： + +```python +class config: + width: t.CInt = 640 + height: t.CInt = 480 + depth: t.CInt = 32 +``` + +使用 `c.Addr` 获取实例指针： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass(arg1, arg2)) +``` + +### 匿名结构体成员（❌） + +> WARNING: **警告：** 由于 Viper 在声明结构体，枚举等处不需要关键字，因此 `t.Anonymouse` 是不需要使用的。此关键字不再维护，强行使用可能导致未定义行为。 + +继承 `t.Anonymous` 的类作为匿名成员嵌入： + +```python +class header(t.Anonymous): + magic: t.CUInt32T + version: t.CUInt32T + +class packet: + header # 匿名嵌入，可直接访问 packet.magic + length: t.CUInt32T +``` + +### 字节序标记 + +> WARNING: **警告：** 此关键字是实验性的。未经充分测试，不当使用可能导致未定义行为。 + +使用 `t.BigEndian` 或 `t.LittleEndian` 标记成员的字节序： + +```python +class network_packet: + length: t.CUInt16T | t.BigEndian # 大端序存储 + type: t.CUInt8T # 默认小端序 +``` + +## 联合体 + +继承 `t.CUnion` 定义联合体： + +```python +class value(t.CUnion): + ival: t.CInt + fval: t.CFloat + pval: t.CVoid | t.CPtr +``` + +等价 C 代码： +```c +union value { + int ival; + float fval; + void* pval; +}; +``` + +## 枚举 + +继承 `t.CEnum` 定义枚举： + +```python +class color(t.CEnum): + RED: t.CEnum = 0 + GREEN: t.CEnum = 1 + BLUE: t.CEnum = 2 +``` + +等价 C 代码： +```c +enum color { + RED = 0, + GREEN = 1, + BLUE = 2 +}; +``` + +枚举成员可直接使用名称： + +```python +c: t.CInt = color.RED +``` + +## Typedef + +使用 `t.CTypedef` 创建类型别名： + +```python +irq_handler_t: t.CTypedef | t.Callable[[], t.CInt] # typedef int (*irq_handler_t)(); +``` + +或通过注解赋值： + +```python +MyStruct: t.CTypedef = original_struct # typedef struct original_struct MyStruct; +``` + +更推荐通过后者的形式，前者编译器不一定能够识别到注解是对 `t.CTypedef` 的，还是后面的注解的，而且容易触发未定义行为。 + +## 位域 + +使用 `t.Bit(n)` 定义位域成员： + +```python +class flags: + a: t.CUInt32T | t.Bit[1] + b: t.CUInt32T | t.Bit[3] + c: t.CUInt32T | t.Bit[4] +``` + +注意：如果存入超出位域宽度的值，会被截断。例如 `a` 为 1 位，存入 2（二进制 `10`）会被截断为 0。 + + +# 06 - 面向对象与运算符重载 + +Viper 中的 `class` 有两种语义：**数据布局**（结构体/联合体/枚举，见 [05-classes.md](05-classes.md)）和**面向对象**。面向对象特性通过 `@t.Object` 装饰器启用，多态通过 `@t.CVTable` 装饰器启用。 + +## `@t.Object` 面向对象 + +使用 `@t.Object` 装饰器启用面向对象特性，类将拥有成员方法、运算符重载、属性装饰器等 OOP 支持。编译器也会自动识别：如果结构体内有函数定义，会自动使用 `@t.Object` 行为，但在实际工程中更推荐明确标记，避免未定义行为。 + +### 基本定义 + +```python +@t.Object +class ViperKernel: + BootInfo: bootinfo | t.CPtr + VGA: vga._VGAScreenDriver | t.CPtr + VGA_drv: vga._VGAScreenDriver + + def __init__(self): + self.BootInfo = BootInfo + serial.init() + serial.puts("ViperOS VKernel Test\n") +``` + +避免在 `__init__` 中使用 `return`，虽然不会报错，但是你永远无法获得 `return` 的结果，如果需要提前结束代码流，`return` 也不失为一种办法。 + +### 成员方法 + +```python +@t.Object +class Sheet: + id: t.CInt + x: t.CInt + y: t.CInt + w: t.CInt + h: t.CInt + + def __init__(self, x: t.CInt, y: t.CInt, w: t.CInt, h: t.CInt): + self.x = x + self.y = y + self.w = w + self.h = h + + def MoveTo(self, nx: t.CInt, ny: t.CInt): + self.x = nx + self.y = ny + + def Resize(self, nw: t.CInt, nh: t.CInt): + self.w = nw + self.h = nh +``` + +### 方法调用 + +```python +sheet: Sheet = Sheet(0, 0, 640, 480) +sheet.MoveTo(100, 100) +sheet.Resize(800, 600) +``` + +### 属性装饰器 + +支持 Python 的 `@property` 和 `@xxx.setter` 装饰器： + +> `@xxx.setter` 和自定义装饰器正在试验阶段，尽可能不要使用 + +```python +@t.Object +class MyObj: + _value: t.CInt + + @property + def value(self) -> t.CInt: + return self._value + + @value.setter + def value(self, v: t.CInt): + self._value = v +``` + +### 静态方法 + +```python +@staticmethod +def _yield(): + pass +``` + +### 对象指针与解引用 + +`@t.Object` 的实例通常通过指针操作： + +```python +obj_ptr: MyClass | t.CPtr = c.Addr(MyClass()) +obj: MyClass = c.Deref(obj_ptr) +``` + +### 成员访问 + +通过 `self` 访问成员变量和方法： + +```python +def method(self) -> t.CVoid: + self.x = 10 + y: t.CInt = self.y + self.DoSomething() +``` + +### 内嵌对象 + +`@t.Object` 可以包含其他对象实例（非指针）： + +```python +@t.Object +class Sheet: + _surf_obj: gfx.Surface + _renderer_obj: gfx.Renderer + surf: gfx.Surface | t.CPtr + renderer: gfx.Renderer | t.CPtr + + def __init__(self, fb: UINT32PTR, zb: float | t.CPtr, w: t.CInt, h: t.CInt): + self._surf_obj = gfx.Surface(fb, zb, w, h) + self.surf = c.Addr(self._surf_obj) + self._renderer_obj = gfx.Renderer(self.surf) + self.renderer = c.Addr(self._renderer_obj) +``` + +### `__before_init__` 方法 + +`@t.Object` 或 `@t.CVTable` 类自动生成 `__before_init__` 方法声明，用于在 `__init__` 之前初始化 vtable： + +```llvm +declare void @".ClassName.__before_init__"(%".ClassName"*) +``` + +值得注意的是，`__before_init__` 并不帮忙初始化内存，而只是初始化虚表等准备工作，实际上，我们更希望用户能自己管理内存。实验性阶段内容中，我们通过内存池机制来帮助用户管理碎片化内容，包括部分小的结构体，但较大内容最好还是由用户手动处理。 + +## `@t.CVTable` 虚函数表与多态 + +> 尽可能避免使用 `self[0]` 来获取结构体成员，其不稳定，也不要手动操作虚表。 + +使用 `@t.CVTable` 装饰器可以启用虚函数表支持。启用后，结构体第一个成员自动插入 `i8*` 类型的 vtable 指针。`@t.CVTable` 的操作必须配合 `@t.Object`，但有时编译器会自动识别是否存在结构体内函数，自动使用 `@t.Object`。 + +### 基本定义 + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 +``` + +在 LLVM IR 中，`@t.CVTable` 类的结构体类型自动插入 vtable 指针： + +```llvm +%".Base" = type { i8*, i32 } +``` + +同时，编译器会生成全局 vtable 变量： + +```llvm +@".Base_Vtable" = internal global [1 x i8*] zeroinitializer +``` + +### 继承与多态 + +虚表可以帮你使用继承语法实现多态： + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 + +@t.Object +@t.CVTable +class M1(Base): + def method(self) -> t.CInt: + return 1 +``` + +当 M1 被初始化时，VTable 会自动修改，这点和 C++ 相同，尽可能避免直接操作 VTable。 + +继承时，子类会自动继承父类的成员变量和方法。子类重写父类方法时，vtable 中对应位置的函数指针被子类方法替换，实现多态。 + +### VTable 运行时修改 + +如果需要临时构造一个额外的 `Base` 结构体，还可以使用： + +```python +@t.Object +@t.CVTable +class Base: + def __init__(self): + pass + def method(self) -> t.CInt: + return 0 + +def __method1(self) -> t.CInt: + return 1 + +def main() -> t.CInt | t.CExport: + b: Base = Base() + b.method = __method1 +``` + +启用 `VTable` 的表中，允许直接修改内部函数为函数指针，在底层，此操作将修改 `VTable` 来达成目的。编译器会检测当前 vtable 是否与类 vtable 相同，如果相同则创建 vtable 的副本（memcpy），避免修改影响所有实例。 + +### 虚方法调度机制 + +虚方法调用的核心流程： + +1. 从对象指针 GEP 获取第 0 个 slot（vtable 指针） +2. 加载 vtable 指针 +3. bitcast 为 `[N x i8*]*` 类型 +4. GEP 获取方法索引处的函数指针 +5. 加载函数指针 +6. bitcast 为正确的函数指针类型 +7. 间接调用 + +## 继承 + +### 结构体成员继承 + +当子类继承父类（父类在 `Gen.class_members` 中存在）时： + +- 父类的成员变量被继承到子类（排在子类成员之前） +- 父类的默认值也被继承 +- 父类的方法被继承（重命名为 `子类名.方法名`） + +```python +@t.Object +class Animal: + name: list[t.CChar, 32] + age: t.CInt + + def __init__(self, age: t.CInt): + self.age = age + + def Speak(self) -> t.CVoid: + serial.puts("...\n") + +@t.Object +class Dog(Animal): + breed: t.CInt + + def Speak(self) -> t.CVoid: + serial.puts("Woof!\n") +``` + +### 异常类继承 + +Viper 支持异常类的继承，用于 `try/except` 的异常匹配： + +```python +class MyError(Exception): + pass + +class SpecificError(MyError): + pass +``` + +详见 [09-exceptions.md](09-exceptions.md)。 + +## 运算符重载 + +`@t.Object` 类支持运算符重载。编译器在遇到运算符时，首先检查左操作数是否为结构体类型，如果是则尝试调用对应的 dunder 方法。 + +### 算术运算符重载 + +| 运算符 | Dunder 方法 | 示例 | +|--------|------------|------| +| `+` | `__add__` | `a + b` → `a.__add__(b)` | +| `-` | `__sub__` | `a - b` → `a.__sub__(b)` | +| `*` | `__mul__` | `a * b` → `a.__mul__(b)` | +| `/` | `__truediv__` | `a / b` → `a.__truediv__(b)` | +| `//` | `__floordiv__` | `a // b` → `a.__floordiv__(b)` | +| `%` | `__mod__` | `a % b` → `a.__mod__(b)` | +| `**` | `__pow__` | `a ** b` → `a.__pow__(b)` | + +示例： + +```python +@t.Object +class Vec2: + x: t.CFloat + y: t.CFloat + + def __init__(self, x: t.CFloat, y: t.CFloat): + self.x = x + self.y = y + + def __add__(self, other: Vec2) -> Vec2: + result: Vec2 = Vec2(0.0, 0.0) + result.x = self.x + other.x + result.y = self.y + other.y + return result + + def __mul__(self, scalar: t.CFloat) -> Vec2: + result: Vec2 = Vec2(0.0, 0.0) + result.x = self.x * scalar + result.y = self.y * scalar + return result +``` + +### 增量赋值运算符重载 + +| 增量运算符 | 优先 Dunder | 回退 Dunder | +|-----------|------------|------------| +| `+=` | `__iadd__` | `__add__` | +| `-=` | `__isub__` | `__sub__` | +| `*=` | `__imul__` | `__mul__` | +| `/=` | `__itruediv__` | `__truediv__` | +| `//=` | `__ifloordiv__` | `__floordiv__` | +| `%=` | `__imod__` | `__mod__` | +| `**=` | `__ipow__` | `__pow__` | + +如果未定义增量版本（如 `__iadd__`），编译器自动回退到普通版本（如 `__add__`）。 + +### 比较运算符重载 + +> **实验性**：以下比较运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。定义它们不会报错，但比较运算不会自动调用。 + +| 运算符 | Dunder 方法 | 状态 | +|--------|------------|------| +| `==` | `__eq__` | 仅类型推断识别 | +| `!=` | `__ne__` | 仅类型推断识别 | +| `<` | `__lt__` | 仅类型推断识别 | +| `<=` | `__le__` | 仅类型推断识别 | +| `>` | `__gt__` | 仅类型推断识别 | +| `>=` | `__ge__` | 仅类型推断识别 | + +### 反向运算符重载 + +> **实验性**：以下反向运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。 + +| Dunder 方法 | 预期语义 | 状态 | +|------------|---------|------| +| `__radd__` | 右操作数加法 | 仅类型推断识别 | +| `__rsub__` | 右操作数减法 | 仅类型推断识别 | +| `__rmul__` | 右操作数乘法 | 仅类型推断识别 | +| `__rtruediv__` | 右操作数真除法 | 仅类型推断识别 | +| `__rfloordiv__` | 右操作数整除 | 仅类型推断识别 | +| `__rmod__` | 右操作数取模 | 仅类型推断识别 | +| `__rpow__` | 右操作数幂运算 | 仅类型推断识别 | + +### 一元运算符重载 + +> **实验性**：以下一元运算符 dunder 方法在编译器中被识别，但尚未实现自动调度。 + +| 运算符 | Dunder 方法 | 状态 | +|--------|------------|------| +| `-a` | `__neg__` | 仅类型推断识别 | +| `+a` | `__pos__` | 仅类型推断识别 | + +## 可调用对象 `__call__` + +```python +@t.Object +class Counter: + value: t.CInt + + def __init__(self): + self.value = 0 + + def __call__(self) -> t.CInt: + self.value += 1 + return self.value + +c: Counter = Counter() +x: t.CInt = c() # x = 1 +y: t.CInt = c() # y = 2 +``` + +## 下标访问重载 + +### `__getitem__` — 读取 + +```python +def __getitem__(self, key: t.CInt) -> t.CInt: + return self.data[key] +``` + +### `__setitem__` — 赋值 + +```python +def __setitem__(self, key: t.CInt, value: t.CInt): + self.data[key] = value +``` + +### `__delitem__` — 删除 + +```python +def __delitem__(self, key: t.CInt): + pass +``` + +## 布尔转换 `__bool__` + +```python +@t.Object +class Buffer: + size: t.CInt + data: list[t.CChar, 1024] + + def __bool__(self) -> t.CBool: + return self.size > 0 + +buf: Buffer = Buffer() +if buf: + serial.puts("buffer is not empty\n") +``` + +## 长度 `__len__` + +```python +@t.Object +class Buffer: + size: t.CInt + + def __len__(self) -> t.CInt: + return self.size + +buf: Buffer = Buffer() +n: t.CInt = len(buf) +``` + +## 字符串转换 `__str__` + +```python +@t.Object +class Point: + x: t.CInt + y: t.CInt + + def __str__(self) -> t.CConst | t.CChar | t.CPtr: + return "Point" + +p: Point = Point(1, 2) +print(p) # 调用 p.__str__() +``` + +## 绝对值 `__abs__` + +```python +@t.Object +class SignedValue: + val: t.CInt + + def __abs__(self) -> t.CInt: + if self.val < 0: + return -self.val + return self.val + +sv: SignedValue = SignedValue() +sv.val = -42 +a: t.CInt = abs(sv) # a = 42 +``` + +## 迭代器协议 + +Viper 的迭代器协议与 Python 相同，但 `__next__` 的实现方式通过 `StopIteration` 异常来结束。 + +### `__iter__` 和 `__next__` + +```python +@t.Object +class IntRange: + start: t.CInt + end: t.CInt + current: t.CInt + + def __init__(self, start: t.CInt, end: t.CInt): + self.start = start + self.end = end + self.current = start + + def __iter__(self): + return self + + def __next__(self, stop_flag) -> t.CInt: + if self.current >= self.end: + raise StopIteration + val: t.CInt = self.current + self.current += 1 + return val +``` + +### for 循环使用 + +```python +r: IntRange = IntRange(0, 10) +for i in r: + print(i) +``` + +编译器生成的等价逻辑： + +```c +IntRange r = IntRange_init(0, 10); +IntRange iter = IntRange___iter__(&r); +while (1) { + i1 stop_flag = 0; + int i = IntRange___next__(&iter, &stop_flag); + if (stop_flag == 1) break; + print_int(i); +} +``` + +## 上下文管理器 + +`with` 语句要求对象实现 `__enter__` 和 `__exit__` 方法： + +```python +@t.Object +class File: + fd: t.CInt + path: t.CConst | t.CChar | t.CPtr + + def __init__(self, path: t.CConst | t.CChar | t.CPtr, flags: t.CInt): + self.path = path + self.fd = -1 + + def __enter__(self): + self.fd = fat32.open(self.path, self.flags) + return self + + def __exit__(self): + fat32.close(self.fd) + + def read(self) -> t.CInt: + return fat32.read(self.fd) +``` + +使用方式： + +```python +with File("/test.txt", FA_READ) as f: + data: t.CInt = f.read() +``` + +编译器会自动保证 `__exit__` 在 `with` 块结束时被调用。 + +## 析构 `__del__` / `__delete__` + +```python +del obj # 调用 obj.__del__() 或 obj.__delete__() +del arr[idx] # 调用 arr.__delitem__(idx) +``` + +## Dunder 方法完整参考 + +### 已实现自动调度的 Dunder 方法 + +| Dunder 方法 | 触发语法 | 说明 | +|------------|---------|------| +| `__init__` | `ClassName(args)` | 构造函数 | +| `__before_init__` | 自动生成 | vtable 初始化，在 `__init__` 之前执行 | +| `__add__` | `a + b` | 加法 | +| `__sub__` | `a - b` | 减法 | +| `__mul__` | `a * b` | 乘法 | +| `__truediv__` | `a / b` | 真除法 | +| `__floordiv__` | `a // b` | 整除 | +| `__mod__` | `a % b` | 取模 | +| `__pow__` | `a ** b` | 幂运算 | +| `__iadd__` | `a += b` | 增量加法（回退到 `__add__`） | +| `__isub__` | `a -= b` | 增量减法（回退到 `__sub__`） | +| `__imul__` | `a *= b` | 增量乘法（回退到 `__mul__`） | +| `__itruediv__` | `a /= b` | 增量真除法（回退到 `__truediv__`） | +| `__ifloordiv__` | `a //= b` | 增量整除（回退到 `__floordiv__`） | +| `__imod__` | `a %= b` | 增量取模（回退到 `__mod__`） | +| `__ipow__` | `a **= b` | 增量幂运算（回退到 `__pow__`） | +| `__call__` | `obj(args)` | 可调用对象 | +| `__getitem__` | `obj[key]` | 下标读取 | +| `__setitem__` | `obj[key] = value` | 下标赋值 | +| `__delitem__` | `del obj[key]` | 下标删除 | +| `__bool__` | `if obj:` | 布尔转换 | +| `__len__` | `len(obj)` | 长度 | +| `__str__` | `str(obj)` / `print(obj)` | 字符串转换 | +| `__abs__` | `abs(obj)` | 绝对值 | +| `__iter__` | `for x in obj:` | 获取迭代器 | +| `__next__` | `for x in obj:` | 获取下一个值（标志位方式） | +| `__enter__` | `with obj as x:` | 上下文管理器进入 | +| `__exit__` | `with obj as x:` | 上下文管理器退出 | +| `__del__` | `del obj` | 析构 | +| `__delete__` | `del obj` | 删除描述符 | + +### 声明支持但未实现自动调度的 Dunder 方法 + +| Dunder 方法 | 预期触发语法 | 当前状态 | +|------------|------------|---------| +| `__eq__` | `a == b` | 仅类型推断识别，比较运算不自动调度 | +| `__ne__` | `a != b` | 同上 | +| `__lt__` | `a < b` | 同上 | +| `__le__` | `a <= b` | 同上 | +| `__gt__` | `a > b` | 同上 | +| `__ge__` | `a >= b` | 同上 | +| `__neg__` | `-a` | 仅类型推断识别，一元负号不自动调度 | +| `__pos__` | `+a` | 同上 | +| `__radd__` | 右操作数加法 | 仅类型推断识别 | +| `__rsub__` | 右操作数减法 | 同上 | +| `__rmul__` | 右操作数乘法 | 同上 | +| `__rtruediv__` | 右操作数真除法 | 同上 | +| `__rfloordiv__` | 右操作数整除 | 同上 | +| `__rmod__` | 右操作数取模 | 同上 | +| `__rpow__` | 右操作数幂运算 | 同上 | + + +# 07 - 控制流 + +Viper 支持 Python 风格的控制流语句，编译为 LLVM IR 的基本块和分支指令。 + +## 条件语句 + +### if / elif / else + +```python +if x > 0: + y = 1 +elif x == 0: + y = 0 +else: + y = -1 +``` + +等价 C 代码： +```c +if (x > 0) { + y = 1; +} else if (x == 0) { + y = 0; +} else { + y = -1; +} +``` + +### 条件表达式 + +```python +is_dir: t.CInt = 1 if (info.attr & AM_DIR) else 0 +``` + +等价 C 代码： +```c +int is_dir = (info.attr & AM_DIR) ? 1 : 0; +``` + +### None 检查 + +```python +if BootInfo != None: + paging.init(BootInfo.MemmapAddr, BootInfo.MemmapSize) +else: + paging.init(0, 0) +``` + +等价 C 代码： +```c +if (BootInfo != NULL) { + paging_init(BootInfo->MemmapAddr, BootInfo->MemmapSize); +} else { + paging_init(0, 0); +} +``` + +## 循环语句 + +### while 循环 + +```python +i: t.CInt = 0 +while i < 10: + buf[i] = 0 + i += 1 +``` + +等价 C 代码： +```c +int i = 0; +while (i < 10) { + buf[i] = 0; + i += 1; +} +``` + +### while-else + +```python +while i < count: + if found: + break +else: + serial.puts("not found\n") +``` + +`else` 块在循环正常结束（非 `break` 退出）时执行。 + +### 无限循环 + +```python +while True: + sched.Scheduler._yield() +``` + +等价 C 代码： +```c +while (1) { + scheduler_yield(); +} +``` + +### for 循环（range）（关于 for 循环的更多用法，参见 [06-oop.md 中 迭代器协议](06-oop.md#迭代器协议)） + +```python +for i in range(10): + buf[i] = 0 + +for i in range(5, 10): + buf[i] = 0 + +for i in range(0, 100, 2): + buf[i] = 0 +``` + +等价 C 代码： +```c +for (int i = 0; i < 10; i++) { + buf[i] = 0; +} + +for (int i = 5; i < 10; i++) { + buf[i] = 0; +} + +for (int i = 0; i < 100; i += 2) { + buf[i] = 0; +} +``` + +### for 循环（迭代器） + +如果类实现了 `__iter__` 和 `__next__` 方法，可以使用 `for ... in` 迭代： + +```python +container: Iter = Iter() +for item in container: + process(item) +``` + +### for-else + +```python +for i in range(count): + if items[i] == target: + break +else: + serial.puts("not found\n") +``` + +### break 和 continue + +```python +while True: + if done: + break + if skip: + continue + process() +``` + +## match 语句 + +Viper 支持 Python 3.10+ 的 `match` 语句，编译为 C 的 `switch` 语句： + +```python +match self.ctype: + case UI_LABEL: + self.RenderLabel(sh) + case UI_BUTTON: + self.RenderButton(sh) + case UI_PANEL: + self.RenderPanel(sh) + case _: + pass +``` + +等价 C 代码： +```c +switch (self->ctype) { + case UI_LABEL: + self_RenderLabel(self, sh); + break; + case UI_BUTTON: + self_RenderButton(self, sh); + break; + case UI_PANEL: + self_RenderPanel(self, sh); + break; + default: + break; +} +``` + +### match 值匹配 + +```python +match code: + case 0: + serial.puts("OK\n") + case 1: + serial.puts("Error\n") + case _: + serial.puts("Unknown\n") +``` + +### match 或模式 + +```python +match value: + case 1 | 2 | 3: + serial.puts("small\n") + case _: + serial.puts("other\n") +``` + +### fallthrough（无 break） + +默认每个 `case` 自动添加 `break`。如需 fallthrough，使用 `c.NoBreak`： + +```python +match value: + case 1: + do_one() + c.NoBreak + case 2: + do_two() +``` + +如果需要提前 break，需要使用 `c.Break()`，因为在 `Python` 中，`match` 分支不支持直接使用 `break`。 + +## assert 语句 +> 避免使用此语句，此语句未经充分测试，在多数情况下，编译器不知道如何展示结果。 + +```python +assert ptr != None, "null pointer" +``` + +编译为条件检查和错误报告。 + +## with 语句 + +`with` 语句用于资源管理，要求对象实现 `__enter__` 和 `__exit__` 方法： + +```python +with File("/test.txt", FA_READ) as f: + data: t.CInt = f.read() +``` + +等价 C 代码： +```c +File* f = File_enter(File_new("/test.txt", FA_READ)); +int data = File_read(f); +File_exit(f); +``` + + +# 08 - C 语言操作 + +Viper 通过 `c` 模块提供对 C 语言底层特性的访问，包括指针操作、内联汇编、预处理指令等。 + +## 指针操作 + +### 取地址 `c.Addr` + +```python +x: t.CInt = 42 +p: t.CInt | t.CPtr = c.Addr(x) # int* p = &x; +``` + +对结构体成员取地址： + +```python +res: t.CInt = fat32.opendir("/", c.Addr(dp)) # res = fat32_opendir("/", &dp); +``` + +对数组取地址： + +```python +viperlib.snprintf(c.Addr(buf), 64, "hello %d", 42) # snprintf(&buf, 64, "hello %d", 42); +``` + +### 解引用 `c.Deref` + +```python +ptr: Sheet | t.CPtr = c.Addr(sheet_obj) +obj: Sheet = c.Deref(ptr) # struct Sheet obj = *ptr; +``` + +### 内存拷贝 `c.Load` + +```python +c.Load(ptr, value) # *ptr = *value; +``` + +### 解引用赋值 `c.DerefAs` + +```python +c.DerefAs(ptr, value) # *ptr = value; +``` + + +### 赋值 `c.Set` + +```python +c.Set(target, value) # target = value; +``` + +## 内联汇编 `c.Asm` + +Viper 提供了声明式的内联汇编语法，编译为 GCC 风格的 `__asm__ __volatile__` 语句。 + +### 基本用法 + +```python +c.Asm("nop") # __asm__ __volatile__("nop"); +``` + +### 带操作数的汇编 + +使用 f-string 和 `c.AsmInp` / `c.AsmOut` 标记操作数： + +```python +saved_rdi: t.CUnsignedLong +c.Asm(f"mov {c.AsmOut(saved_rdi, t.ASM_DESCR.OUTPUT_REG)}, rdi", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RDI]) +``` + +等价 C 代码： +```c +__asm__ __volatile__( + "mov %0, rdi" + : "=r"(saved_rdi) + : + : "memory", "rdi" +); +``` + +### 输入操作数 `c.AsmInp` + +```python +msg: t.CConst | t.CChar | t.CPtr = "Hello" +c.Asm(f"mov rdi, {c.AsmInp(msg, t.ASM_DESCR.REG_ANY)}", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX]) +``` + +### 完整示例 + +```python +c.Asm(f"""mov rdi, {c.AsmInp(msg, t.ASM_DESCR.REG_ANY)} + call {c.AsmInp(log_info_fn, t.ASM_DESCR.REG_ANY)}""", + op=[t.ASM_DESCR.CLOBBER_MEMORY, t.ASM_DESCR.CLOBBER_RAX, + t.ASM_DESCR.CLOBBER_RCX, t.ASM_DESCR.CLOBBER_RDX, + t.ASM_DESCR.CLOBBER_RDI, t.ASM_DESCR.CLOBBER_RSI, + t.ASM_DESCR.CLOBBER_R8, t.ASM_DESCR.CLOBBER_R9, + t.ASM_DESCR.CLOBBER_R10, t.ASM_DESCR.CLOBBER_R11]) +``` + +### ASM_DESCR 约束字符 + +#### 操作数修饰符 + +| 常量 | 值 | 说明 | +|------|----|------| +| `MODIFIER_OUTPUT` | `=` | 输出操作数 | +| `MODIFIER_READWRITE` | `+` | 读写操作数 | +| `MODIFIER_INPUT` | `` | 输入操作数（默认） | +| `MODIFIER_GLOBAL` | `&` | 全局操作数 | + +#### 寄存器约束 + +| 常量 | 值 | 说明 | +|------|----|------| +| `REG_ANY` | `r` | 任何通用寄存器 | +| `REG_EAX` / `REG_RAX` | `a` | EAX/RAX | +| `REG_EBX` / `REG_RBX` | `b` | EBX/RBX | +| `REG_ECX` / `REG_RCX` | `c` | ECX/RCX | +| `REG_EDX` / `REG_RDX` | `d` | EDX/RDX | +| `REG_ESI` / `REG_RSI` | `S` | ESI/RSI | +| `REG_EDI` / `REG_RDI` | `D` | EDI/RDI | +| `REG_XMM` | `x` | XMM 寄存器 | + +#### 内存与立即数 + +| 常量 | 值 | 说明 | +|------|----|------| +| `MEMORY` | `m` | 内存操作数 | +| `IMMEDIATE` | `i` | 立即数 | +| `ANY` | `g` | 通用寄存器/内存/立即数 | + +#### 预定义组合约束 + +| 常量 | 值 | 说明 | +|------|----|------| +| `OUTPUT_REG` | `=r` | 输出，通用寄存器 | +| `OUTPUT_MEM` | `=m` | 输出，内存 | +| `OUTPUT_EAX` | `=a` | 输出，EAX/RAX | +| `INPUT_REG` | `r` | 输入，通用寄存器 | +| `INPUT_MEM` | `m` | 输入，内存 | +| `INPUT_IMM` | `i` | 输入，立即数 | + +#### 破坏描述符 + +| 常量 | 说明 | +|------|------| +| `CLOBBER_EAX` / `CLOBBER_RAX` | 破坏 EAX/RAX | +| `CLOBBER_EBX` / `CLOBBER_RBX` | 破坏 EBX/RBX | +| `CLOBBER_ECX` / `CLOBBER_RCX` | 破坏 ECX/RCX | +| `CLOBBER_EDX` / `CLOBBER_RDX` | 破坏 EDX/RDX | +| `CLOBBER_ESI` / `CLOBBER_RSI` | 破坏 ESI/RSI | +| `CLOBBER_EDI` / `CLOBBER_RDI` | 破坏 EDI/RDI | +| `CLOBBER_CC` | 破坏条件码（标志寄存器） | +| `CLOBBER_MEMORY` | 破坏内存 | +| `CLOBBER_R8` ~ `CLOBBER_R15` | 破坏 R8~R15 | + +## 预处理指令 + +Viper 通过 `c` 模块的函数调用实现 C 预处理指令。 + +### #define + +```python +c.CDefine("MAX_SIZE", 1024) # #define MAX_SIZE 1024 +``` + +上述方法容易引起未定义行为，至少是不便于理解，更常用的方式是使用类型注解： + +```python +MAX_SIZE: t.CDefine = 1024 # #define MAX_SIZE 1024 +``` + +### 条件编译 + +```python +c.CIfndef(HEADER_H) # #ifndef HEADER_H +c.CDefine(HEADER_H) # #define HEADER_H +c.CEndif() # #endif +``` + +```python +c.CIfdef(DEBUG) # #ifdef DEBUG +c.CEndif() # #endif +``` + +```python +c.CIf(VERSION > 2) # #if VERSION > 2 +c.CElif(VERSION > 1) # #elif VERSION > 1 +c.CElse() # #else +c.CEndif() # #endif +``` + +### #undef + +```python +c.Undef(MACRO_NAME) # #undef MACRO_NAME +``` + +### #error + +```python +c.CError("Platform not supported") # #error "Platform not supported" +``` + +### #pragma +> 此关键字已不再支持 +```python +c.CPragma("GCC diagnostic push") # #pragma GCC diagnostic push +``` + +### ## 连接符 +> 此方法可能已经不再支持 +```python +c.TokenPast("PREFIX_", "NAME") # PREFIX_ ## NAME +``` + +由于编译器设计，以及不需要像 `C` 一样做大量字符替换，对于上述所有的条件宏都远比 `C` 差，实际工程中不推荐使用条件宏，后期将完善宏的设计，引入模板等。 + +## 仅声明 `c.State` + +`c.State` 用于声明但不定义函数或变量： + +```python +def isr0() -> t.CExtern | t.CVoid | c.State: pass # extern void isr0(); +``` + +## LLVM IR 内联 + +### c.LLVMIR + +直接嵌入 LLVM IR 指令来实现高效且跨平台的汇编操作： + +```python +c.LLVMIR(f"add i32 {c.LInp(a)}, {c.LInp(b)}", t.CInt) +``` + +### c.LInp / c.LOut + +标记 LLVM IR 的输入/输出操作数： + +```python +c.LInp(expr) # 输入操作数 +c.LOut(expr) # 输出操作数 +``` + +## 运算符重载 + +`@t.Object` 类支持运算符重载，详见 [06-oop.md 中 运算符重载](06-oop.md#运算符重载)。 + + +# 09 - 异常处理 + +Viper 支持 Python 风格的 `try/except/finally` 异常处理，编译为基于 jmp_buf 和错误码的 C 级别异常处理机制。 + +## 异常处理机制 + +Viper 的异常处理不使用 C++ 风格的零开销异常，而是通过函数参数传递错误码和错误消息实现。编译器自动为可能抛出异常的函数添加 `__eh_msg_out__` 和 `__eh_code_out__` 参数。 + +## try / except + +```python +try: + result = risky_operation() +except ValueError: + serial.puts("ValueError occurred\n") +except OSError: + serial.puts("OSError occurred\n") +except Exception: + serial.puts("Unknown error\n") +``` + +### 内置异常类型 + +Viper 预定义了以下异常类型及其错误码： + +| 异常类型 | 错误码 | +|---------|--------| +| `ValueError` | 1 | +| `TypeError` | 2 | +| `RuntimeError` | 3 | +| `ZeroDivisionError` | 4 | +| `IndexError` | 5 | +| `KeyError` | 6 | +| `IOError` | 7 | +| `OSError` | 8 | +| `AssertionError` | 9 | +| `Exception` | 99 | + +### 自定义异常 + +```python +class DiskFullError(Exception): + pass + +class ReadOnlyError(DiskFullError): + pass +``` + +自定义异常会自动分配递增的错误码（从 100 开始），并支持继承匹配。 + +### 多异常捕获 + +```python +try: + result = operation() +except (ValueError, TypeError): + serial.puts("Value or Type error\n") +``` + +### 获取异常信息 + +```python +try: + result = operation() +except Exception as e: + serial.puts("Error occurred\n") +``` + +## try / finally + +```python +try: + result = operation() +finally: + cleanup() +``` + +`finally` 块无论是否发生异常都会执行。 + +## try / except / finally + +```python +try: + result = operation() +except ValueError: + handle_value_error() +finally: + cleanup() +``` + +## raise 语句 + +### 抛出异常 + +```python +raise ValueError("invalid value") +``` + +### 抛出自定义异常 + +```python +raise MyCustomError("something went wrong") +``` + +### 重新抛出异常 + +```python +raise +``` + +### StopIteration + +`StopIteration` 有特殊处理，用于迭代器协议： + +```python +raise StopIteration +``` + +## 异常处理的编译实现 + +### 函数签名变换 + +可能抛出异常的函数会自动添加错误输出参数： + +```python +def risky_operation() -> t.CInt: + if error: + raise ValueError("bad value") + return 0 +``` + +编译后等价于： + +```c +int risky_operation(char** __eh_msg_out__, int* __eh_code_out__) { + if (error) { + *__eh_code_out__ = 1; // ValueError code + *__eh_msg_out__ = "bad value"; + return 0; + } + return 0; +} +``` + +### 调用点变换 + +```python +try: + result = risky_operation() +except ValueError: + handle_error() +``` + +编译后等价于： + +```c +char* __eh_msg = NULL; +int __eh_code = 0; +int result = risky_operation(&__eh_msg, &__eh_code); +if (__eh_msg != NULL) { + if (__eh_code == 1) { // ValueError + handle_error(); + } +} +``` + +## 异常继承匹配 + +异常匹配支持继承体系。捕获父类异常时，子类异常也会被匹配： + +```python +class FileSystemError(Exception): + pass + +class FileNotFoundError(FileSystemError): + pass + +class PermissionError(FileSystemError): + pass + +try: + operation() +except FileSystemError: + # 也会捕获 FileNotFoundError 和 PermissionError + handle_fs_error() +``` + + +# 10 - 模块与导入系统 + +Viper 的模块系统基于 Python 的 `import` 语法，但编译时通过 TransPyC 的符号表和类型系统实现跨模块的类型解析。 + +## 导入语法 + +### 标准导入 + +```python +import t, c # Viper 内置类型和操作模块 +import asm # 汇编辅助模块 +import string # 字符串操作模块 +``` + +### 模块导入 + +```python +import bootinfo +import intr +import platform.pch as pch +import drivers.serial.uart.serial as serial +``` + +### from 导入 + +```python +from stdint import * # 导入所有 stdint 类型别名 +from drivers.video.vesafb.vga import _VGAScreenDriver +``` + +### 相对导入 + +```python +from . import submodule +from .. import parent_module +``` + +## 特殊模块 + +### `t` 模块 + +类型定义模块，提供所有 C 类型对应的 Viper 类型。每个 Viper 文件通常都需要导入： + +```python +import t +``` + +### `c` 模块 + +C 语言操作模块，提供指针操作、内联汇编、预处理指令等： + +```python +import c +``` + +### `asm` 模块 + +汇编辅助模块，提供常用汇编指令的封装： + +```python +import asm + +asm.sti() # __asm__ __volatile__("sti"); +asm.hlt() # __asm__ __volatile__("hlt"); +asm.cli() # __asm__ __volatile__("cli"); +asm.nop() # __asm__ __volatile__("nop"); +asm.BSSClean() # 清零 BSS 段 +``` + +### `string` 模块 + +字符串/内存操作模块，对应 C 标准库的 `string.h`： +> string 部分使用 x86_64 专有汇编进行加速，因此某些函数在其它架构可能需要手动重新编写。 + +```python +import string + +string.memset(c.Addr(buf), 0, 64) # memset(&buf, 0, 64); +string.memcpy(dst, src, size) # memcpy(dst, src, size); +string.memmove(dst, src, size) # memmove(dst, src, size); +string.memcmp(a, b, size) # memcmp(a, b, size); +string.strcmp(a, b) # strcmp(a, b); +string.strcpy(dst, src) # strcpy(dst, src); +``` + +### `stdint` 模块 + +提供 C `stdint.h` 中的类型别名： + +```python +from stdint import * + +# 提供以下类型别名： +# UINT8PTR, UINT16PTR, UINT32PTR, UINT64PTR +# INT8PTR, INT16PTR, INT32PTR, INT64PTR +# 等等 +``` + +### `viperlib` 模块 + +ViperOS 标准库，提供常用 C 库函数： + +```python +import viperlib + +viperlib.snprintf(c.Addr(buf), 64, "value=%d", 42) # snprintf(&buf, 64, "value=%d", 42); +``` + +### `vpsdk` 模块 + +> vpsdk 模块 专供于基于 Viper 开发的示例操作系统 ViperOS，不可用于其它平台，且需要特殊配置。 + +ViperOS 应用 SDK，提供系统调用和窗口管理： + +```python +import vpsdk.window as window +import vpsdk.process as process +import vpsdk.dynlib as dynlib +import vpsdk.syscall as syscall +``` + +## 模块解析流程 + +### 编译时模块加载 + +Viper 的 `import` 是编译时操作，不涉及运行时模块加载。完整的模块解析流程与两阶段编译紧密耦合： + +1. **阶段一**：从入口文件出发，通过 `import` 语句递归发现所有可达的源文件 +2. **阶段一**：对每个源文件生成 `.pyi` 签名存根和 `.stub.ll` LLVM IR 声明 +3. **阶段二**：加载所有 `.pyi` 和 `.stub.ll`，构建共享符号表 +4. **阶段二**：翻译源文件时，通过符号表解析跨模块的类型和函数引用 +5. **阶段二**：将被引用模块的 `.stub.ll` 声明嵌入到生成的 `.ll` 文件头部 + +### 类型存根文件（.pyi） + +`.pyi` 文件是模块的声明接口，由 `PythonToStubConverter` 在阶段一生成。存根文件的内容规则： + +| 元素 | 存根表示 | +|------|---------| +| `t.CDefine` 常量 | 保留完整定义（含赋值） | +| `t.CDefine` 函数 | 保留完整函数体 | +| 普通函数 | 仅签名 + `c.State` | +| 全局变量 | 添加 `t.CExtern` | +| 类定义 | 保留成员类型注解和方法签名 | + +存根文件还自动添加宏守卫（`c.CIfndef`/`c.CEndif()`），防止重复包含。 + +### SHA1 命名空间与导入 + +每个模块的函数和结构体在 LLVM IR 中自动加上 SHA1 前缀，消除跨模块符号冲突。只有标记为 `t.CExport` 的函数保持原始名称。详见 [01-overview.md 中 SHA1 命名空间机制](01-overview.md#sha1-命名空间机制)。 + +### 符号表 + +所有模块的类型信息统一存储在 `SymbolTable` 中，包含： +- 结构体/联合体/枚举定义 +- 函数签名 +- 全局变量 +- typedef 别名 +- `t.CDefine` 常量 + +### 跨模块类型引用 + +```python +# 在 fat32_types.py 中定义 +class fat32_fileinfo: + fname: list[t.CChar, 13] + attr: t.CUInt8T + file_size: t.CUInt32T + +# 在其他模块中引用 +import fat32_types +info: fat32_types.fat32_fileinfo +``` + +跨模块引用的结构体在 LLVM IR 中使用 SHA1 前缀的类型名： +```llvm +%".fat32_fileinfo" = type { [13 x i8], i8, i32 } +``` + +## 项目 includes 配置 + +在 `project.json` 中配置模块搜索路径： + +```json +{ + "includes": ["../.."], + "source_dir": "./Kernel" +} +``` + +编译器会在 `includes` 指定的目录中搜索导入的模块。 + + +# 11 - 内置函数与运算符 + +Viper 支持 Python 内置函数和运算符，编译为对应的 C/LLVM 操作。 + +## 内置函数 + +### print + +```python +print("Hello, World!") # 调用 puts 或 printf +print(x, y) # 多参数输出 +``` + +### len + +```python +n: t.CInt = len(array) # 获取数组长度（编译时常量） +``` + +### sizeof + +```python +s: t.CSizeT = sizeof(t.CInt) # sizeof(int) +s: t.CSizeT = obj.__sizeof__() # sizeof(obj) +``` + +### abs + +```python +a: t.CInt = abs(x) # abs(x) +``` + +### int / float / bool + +```python +i: t.CInt = int(3.14) # (int)3.14 +f: t.CFloat = float(42) # (float)42 +b: t.CBool = bool(x) # x != 0 +``` + +### min / max + +```python +m: t.CInt = min(a, b) # a < b ? a : b +m: t.CInt = max(a, b) # a > b ? a : b +``` + +### chr / ord + +```python +c: t.CChar = chr(65) # 'A' (整数转字符) +n: t.CInt = ord('A') # 65 (字符转整数) +``` + +### range + +用于 `for` 循环： + +```python +for i in range(10): pass # for (int i = 0; i < 10; i++) +for i in range(5, 10): pass # for (int i = 5; i < 10; i++) +for i in range(0, 10, 2): pass # for (int i = 0; i < 10; i += 2) +``` + +## 运算符 + +### 算术运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a + b` | `a + b` | 加法 | +| `a - b` | `a - b` | 减法 | +| `a * b` | `a * b` | 乘法 | +| `a / b` | `a / b` | 浮点除法 | +| `a // b` | `a / b` | 整数除法 | +| `a % b` | `a % b` | 取模 | +| `a ** b` | `pow(a, b)` | 幂运算 | +| `-a` | `-a` | 取负 | +| `+a` | `+a` | 正号 | + +### 位运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a & b` | `a & b` | 按位与 | +| `a \| b` | `a \| b` | 按位或 | +| `a ^ b` | `a ^ b` | 按位异或 | +| `~a` | `~a` | 按位取反 | +| `a << n` | `a << n` | 左移 | +| `a >> n` | `a >> n` | 右移 | + +> **注意**：`|` 运算符在类型注解中表示类型组合，在表达式中表示按位或。编译器根据上下文区分。 + +### 比较运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a == b` | `a == b` | 等于 | +| `a != b` | `a != b` | 不等于 | +| `a < b` | `a < b` | 小于 | +| `a > b` | `a > b` | 大于 | +| `a <= b` | `a <= b` | 小于等于 | +| `a >= b` | `a >= b` | 大于等于 | + +### 逻辑运算符 + +| Viper | C 等价 | 说明 | +|-------|--------|------| +| `a and b` | `a && b` | 逻辑与 | +| `a or b` | `a \|\| b` | 逻辑或 | +| `not a` | `!a` | 逻辑非 | + +### 增强赋值运算符 + +| Viper | C 等价 | +|-------|--------| +| `a += b` | `a += b` | +| `a -= b` | `a -= b` | +| `a *= b` | `a *= b` | +| `a //= b` | `a /= b` | +| `a %= b` | `a %= b` | +| `a <<= b` | `a <<= b` | +| `a >>= b` | `a >>= b` | +| `a &= b` | `a &= b` | +| `a \|= b` | `a \|= b` | +| `a ^= b` | `a ^= b` | + +## 类型自动转换 + +Viper 在算术运算中自动进行类型提升： + +1. **整数宽度提升**：较窄的整数类型自动扩展为较宽的类型 + - 有符号类型使用符号扩展（`sext`） + - 无符号类型使用零扩展（`zext`） + +2. **整数到浮点**：整数与浮点数运算时，整数自动转换为浮点 + +3. **浮点精度提升**：`float` 与 `double` 运算时，`float` 提升为 `double` + +## 指针算术 + +指针与整数的加法支持自动计算偏移： + +```python +buf: t.CUInt32T | t.CPtr +buf[i] = value # buf[i] = value; (自动计算偏移) +``` + +## 数组下标 + +```python +buf: list[t.CChar, 64] +buf[0] = 'H' # buf[0] = 'H'; +buf[i] = value # buf[i] = value; + +info.fname[0] # info.fname[0] (结构体数组成员访问) +``` + +## 成员访问 + +```python +obj.x # obj.x (直接成员) +obj.method() # obj.method() (方法调用) +ptr.x # ptr->x (指针自动解引用) +``` + +## 字符串字面量 + +字符串字面量编译为 C 字符串常量（`const char*`），其为只读： + +```python +serial.puts("Hello") # 传递 const char* +``` + +### 字符串索引 + +```python +msg: t.CConst | str = "Hello" +serial.puts(c.Addr(msg[4])) # 访问 msg[4] 的地址，输出 'o' +``` + +## C 标准库函数 + +Viper 内置支持以下 C 标准库函数的直接调用： + +| 函数 | 签名 | +|------|------| +| `strlen(s)` | `size_t strlen(const char* s)` | +| `memset(s, c, n)` | `void* memset(void* s, int c, size_t n)` | +| `memcpy(d, s, n)` | `void* memcpy(void* d, const void* s, size_t n)` | +| `memmove(d, s, n)` | `void* memmove(void* d, const void* s, size_t n)` | +| `memcmp(a, b, n)` | `int memcmp(const void* a, const void* b, size_t n)` | +| `strcmp(a, b)` | `int strcmp(const char* a, const char* b)` | +| `strncmp(a, b, n)` | `int strncmp(const char* a, const char* b, size_t n)` | +| `strcpy(d, s)` | `char* strcpy(char* d, const char* s)` | +| `strncpy(d, s, n)` | `char* strncpy(char* d, const char* s, size_t n)` | +| `strcat(d, s)` | `char* strcat(char* d, const char* s)` | +| `puts(s)` | `int puts(const char* s)` | +| `atoi(s)` | `int atoi(const char* s)` | +| `atol(s)` | `long atol(const char* s)` | +| `abs(x)` | `int abs(int x)` | +| `labs(x)` | `long labs(long x)` | + +部分模式比如裸机没有 libc，则不能使用这些函数，即使环境满足，Viper 标准库也可能和 C 标准库冲突。 + + +# 12 - 项目配置与构建 + +Viper 项目使用 `project.json` 配置文件管理编译和链接参数。 + +## project.json 结构 + +```json +{ + "name": "Kernel", + "version": "1.0.0", + "source_dir": "./Kernel", + "temp_dir": "./temp", + "output_dir": "./output", + "compiler": { + "cmd": "llc", + "flags": ["-filetype=obj", "-mtriple=x86_64-none-elf", "-relocation-model=static", "-O2"] + }, + "linker": { + "cmd": "ld.lld.exe", + "flags": [ + "-m", "elf_x86_64", + "-T", "linker.ld", + "--oformat", "elf64-x86-64", + "--strip-all" + ], + "output": "kernel.bin" + }, + "includes": ["../../includes"], + "target": { + "triple": "x86_64-none-elf", + "datalayout": "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128" + }, + "options": { + "slice_level": 3, + "target": "llvm", + "strict_mode": true + } +} +``` + +## 配置项说明 + +### 基本信息字段 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `name` | string | 项目名称 | +| `version` | string | 项目版本 | +| `source_dir` | string | 源代码目录（相对于 project.json 所在目录） | +| `temp_dir` | string | 临时文件目录（存放 .pyi 存根等） | +| `output_dir` | string | 输出文件目录 | + +### 编译器配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `compiler.cmd` | string | 编译器命令（通常为 `llc`） | +| `compiler.flags` | string[] | 编译器标志 | + +常用 `llc` 标志： +- `-filetype=obj`：输出目标文件 +- `-mtriple=x86_64-none-elf`：目标三元组 +- `-relocation-model=static`：静态重定位 +- `-O2`：优化级别 + +### 链接器配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `linker.cmd` | string | 链接器命令（通常为 `ld.lld.exe`） | +| `linker.flags` | string[] | 链接器标志 | +| `linker.output` | string | 输出文件名 | + +常用 `ld.lld` 标志： +- `-m elf_x86_64`：ELF x86_64 格式 +- `-T linker.ld`：链接脚本 +- `--oformat elf64-x86-64`：输出 ELF 格式 +- `--oformat binary`：输出原始二进制（用于内核） +- `--strip-all`：去除所有符号信息 + +### 目标平台配置 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `target.triple` | string | LLVM 目标三元组 | +| `target.datalayout` | string | LLVM 数据布局字符串 | + +默认值： +- triple: `x86_64-none-elf` +- datalayout: `e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128` + +### 包含路径 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `includes` | string[] | 模块搜索路径列表 | + +### 编译选项 + +| 字段 | 类型 | 说明 | +|------|------|------| +| `options.slice_level` | int | 切片优化级别（0-3） | +| `options.target` | string | 编译目标（目前仅支持 `llvm`） | +| `options.strict_mode` | bool | 严格模式 | + +## 切片优化级别 + +| 级别 | 名称 | 说明 | +|------|------|------| +| `0` | `no_optimize` | 无优化 | +| `1` | `conservative` | 保守优化 | +| `2` | `moderate` | 适度优化 | +| `3` | `aggressive` | 激进优化（默认） | +| `s` | `size` | 体积优化 | + +## 项目结构示例 + +### 内核项目 + +``` +VKernel/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── Kernel/ # 源代码目录 +│ ├── main.py # 内核入口 +│ ├── bootinfo.py +│ ├── intr/ +│ │ ├── gdt.py +│ │ ├── idt.py +│ │ └── syscall.py +│ ├── drivers/ +│ │ ├── serial/ +│ │ ├── video/ +│ │ ├── storage/ +│ │ └── ... +│ └── ... +├── temp/ # 临时文件（自动生成） +│ ├── .pyi # 签名存根（阶段一生成） +│ ├── .stub.ll # LLVM IR 声明（阶段一生成） +│ └── _sha1_map.txt # SHA1→源文件路径映射 +└── output/ # 输出文件（自动生成） + ├── kernel.bin # 内核二进制 + ├── *.deps.json # 依赖信息 + └── linker.ld # 生成的链接脚本 +``` + +### 应用项目 + +``` +HelloWorld/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── main.py # 应用入口 +├── temp/ # 临时文件 +└── output/ # 输出文件 + └── helloworld.elf # ELF 可执行文件 +``` + +### 库项目 + +``` +SerialLogger/ +├── project.json # 项目配置 +├── linker.ld # 链接脚本 +├── serial_logger.py # 库源码 +├── temp/ # 临时文件 +└── output/ # 输出文件 +``` + +## 链接脚本 + +ViperOS 使用自定义链接脚本控制内存布局： + +```ld +ENTRY(_start) + +SECTIONS +{ + . = 0x100000; /* 加载地址 */ + + .text : { + *(.text.startup) + *(.text*) + } + + .rodata : { + *(.rodata*) + } + + .data : { + *(.data*) + } + + .bss : { + *(.bss*) + *(COMMON) + } +} +``` + +## 编译命令 + +TransPyC 编译器的基本用法（注：适用于单个文件，而不适用于工程）： + +```bash +python TransPyC.py -f InputFile -o OutputFile [options] +``` + +### 命令行参数 + +| 参数 | 说明 | +|------|------| +| `-f` | 输入文件路径 | +| `-o` | 输出文件路径 | +| `-wh` | 头文件列表 | +| `-debug` | 调试输出文件 | +| `-cc` | 编译命令 | +| `-cflags` | 编译标志 | +| `-run` | 编译后运行 | +| `-args` | 运行参数 | +| `-h` | 辅助文件（C 或 Python） | + +工程编译使用： +```bash +python Projectrans.py xxx.json +``` +即可 + +## 增量编译 + +TransPyC 支持基于 SHA1 的增量编译，与两阶段编译模型紧密耦合： + +### 阶段一增量 + +- 每个源文件按内容计算 SHA1 哈希（16位） +- 如果 `.pyi` 和 `.stub.ll` 已存在，则跳过生成（缓存命中） +- 只有源文件内容变化导致 SHA1 变化时才重新生成声明接口 +- `temp/_sha1_map.txt` 记录 SHA1 → 源文件路径的映射，格式为 `:` + +### 阶段二增量 + +- 阶段二加载 `temp/_sha1_map.txt` 重建 SHA1 映射 +- 过滤掉不在当前 SHA1 映射中的旧 `.pyi` 和 `.stub.ll` 文件 +- 共享符号表一次性构建，避免每个文件重复加载 + +### SHA1 命名空间与增量编译的协同 + +SHA1 同时服务于命名空间隔离和增量编译： +- **命名空间**：非 `t.CExport` 函数和结构体使用 SHA1 前缀，消除符号冲突 +- **增量编译**：SHA1 不变则缓存命中，SHA1 变化则重新生成 + +详见 [01-overview.md 中 SHA1 命名空间机制](01-overview.md#sha1-命名空间机制)。 + +