164 lines
7.1 KiB
Plaintext
164 lines
7.1 KiB
Plaintext
import t, c
|
||
from stdint import *
|
||
import ast
|
||
import llvmlite
|
||
import memhub
|
||
import string
|
||
import viperlib
|
||
import stdio
|
||
import stdlib
|
||
import lib.core.VLogger as VLogger
|
||
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
|
||
|
||
|
||
# ============================================================
|
||
# HandlesFunctions - 函数定义处理（P5-1 对象化后的委托层）
|
||
#
|
||
# P5-1 对象化重构（wiki/14-oo-refactor.md）：
|
||
# 全部函数已迁移为 lib.core.Gen.FuncGenMixin 的方法，本模块保留
|
||
# 同名模块级函数作为"一行委托"：延迟 import Gen 取全局单例
|
||
# （BuildPipeline/main 创建后 set_global_gen 注册），再调用 gen
|
||
# 的对应方法。签名与旧函数完全一致，调用方零改动。
|
||
# ============================================================
|
||
|
||
|
||
# ============================================================
|
||
# 全局 gen 单例（模块级镜像）
|
||
#
|
||
# 由 BuildPipeline/main 在 gen 创建后调用 set_global_gen 注入。
|
||
# 委托函数用本地 _g_gen（cast 成 Generator 类型），避免函数体
|
||
# import 包模块（TPC 不注册包短名）与 Generator 类/模块名歧义。
|
||
# ============================================================
|
||
_g_gen: t.CVoid | t.CPtr = None
|
||
|
||
|
||
def set_global_gen(gen: t.CVoid | t.CPtr) -> None:
|
||
"""注入全局 gen 单例（BuildPipeline/main 调用）"""
|
||
global _g_gen
|
||
_g_gen = gen
|
||
|
||
|
||
# 作用域类型常量（本地副本，避免旧编译器跨模块 CDefine 查找 bug）
|
||
SCOPE_FUNCTION: t.CDefine = 1
|
||
|
||
|
||
# ============================================================
|
||
# 委托函数：延迟 import Gen 取全局单例 → 调 gen 方法
|
||
# （gen 未注册时正常流程不会发生；有返回值函数返回默认值）
|
||
# ============================================================
|
||
|
||
|
||
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 属性字符串（委托到 gen.extract_func_attrs）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g.extract_func_attrs(pool, decorator_list, imported_modules)
|
||
return None
|
||
|
||
|
||
def _get_attr_name_from_node(node: ast.AST | t.CPtr) -> str:
|
||
"""从 AST 节点提取属性名（委托到 gen._get_attr_name_from_node）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g._get_attr_name_from_node(node)
|
||
return None
|
||
|
||
|
||
def _append_llvm_attr(attrs_buf: t.CChar | t.CPtr, attr_name: str) -> int:
|
||
"""将属性名映射到 LLVM 属性名并追加到缓冲区（委托到 gen._append_llvm_attr）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g._append_llvm_attr(attrs_buf, attr_name)
|
||
return 0
|
||
|
||
|
||
def _append_str(buf: t.CChar | t.CPtr, s: str) -> None:
|
||
"""将字符串追加到 buf 末尾（用 strlen+strcpy 模拟 strcat）（委托到 gen._append_str）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
g._append_str(buf, s)
|
||
|
||
|
||
def _mangle_name(trans: HT.Translator | t.CPtr, name: str) -> str:
|
||
"""为任意名称添加 SHA1 前缀（不做 main/export 检查）（委托到 gen._mangle_name）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g._mangle_name(trans, name)
|
||
return None
|
||
|
||
|
||
def _mangle_name_with_sha1(sha1: str, name: str) -> str:
|
||
"""用指定 SHA1 为名称添加前缀（不做 main/export 检查）（委托到 gen._mangle_name_with_sha1）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g._mangle_name_with_sha1(sha1, name)
|
||
return None
|
||
|
||
|
||
def _mangle_func_name(trans: HT.Translator | t.CPtr, name: str,
|
||
has_export: int) -> str:
|
||
"""为函数名添加 SHA1 前缀（检查 main/export）（委托到 gen._mangle_func_name）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g._mangle_func_name(trans, name, has_export)
|
||
return None
|
||
|
||
|
||
def forward_declare_functions(trans: HT.Translator | t.CPtr,
|
||
tree: ast.AST | t.CPtr) -> int:
|
||
"""预扫描所有顶层 FunctionDef，创建前向声明（委托到 gen.forward_declare_functions）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g.forward_declare_functions(trans, tree)
|
||
return 0
|
||
|
||
|
||
def translate_function_def(trans: HT.Translator | t.CPtr,
|
||
node: ast.AST | t.CPtr) -> int:
|
||
"""翻译函数定义，返回新增的变量数（通常为 0，函数定义不增加当前作用域变量）（委托到 gen.translate_function_def）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g.translate_function_def(trans, node)
|
||
return 0
|
||
|
||
|
||
def translate_nested_function_def(trans: HT.Translator | t.CPtr,
|
||
node: ast.AST | t.CPtr) -> int:
|
||
"""翻译嵌套函数定义：提升为顶层函数 + 创建闭包（委托到 gen.translate_nested_function_def）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g.translate_nested_function_def(trans, node)
|
||
return 0
|
||
|
||
|
||
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 函数并添加参数（委托到 gen.create_function）"""
|
||
import lib.core.Gen.Generator as Generator
|
||
g: Generator.Generator | t.CPtr = (Generator.Generator | t.CPtr)(t.CVoid(_g_gen, t.CPtr))
|
||
if g is not None:
|
||
return g.create_function(pool, mod, name, args_node, ret_ty)
|
||
return None
|