将 EnumName/value/IsSigned 加入 FIELD_ROUTES，消除所有冗余 hasattr/getattr

2026-06-18 19:09:50 +08:00
parent f99666420b
commit d7b98cc9c5
33 changed files with 1722 additions and 1200 deletions
--- a/lib/core/SymbolUtils.py
+++ b/lib/core/SymbolUtils.py
@@ -0,0 +1,56 @@
+"""共享工具函数：符号表处理中使用的 AST 分析工具"""
+from __future__ import annotations
+import ast
+from lib.includes import t
+
+
+def IsTModuleType(TypeName: str) -> bool:
+    """检测类型名称是否是 t 模块中的类型（CType 或其他特殊类型）"""
+    if TypeName in ('t', 'State', 'Bit', 'Anonymous', 'Postdefinition'):
+        return True
+    TypeClass = getattr(t, TypeName, None)
+    if TypeClass and isinstance(TypeClass, type):
+        if issubclass(TypeClass, t.CType):
+            return True
+    FallbackClass = getattr(t, f'_{TypeName}', None)
+    if FallbackClass and isinstance(FallbackClass, type):
+        if issubclass(FallbackClass, t.CType):
+            return True
+    return False
+
+
+def AnnotationContainsTType(node, TypeName: str) -> bool:
+    """检测 AST 节点中是否包含指定名称的 t 模块类型"""
+    if isinstance(node, ast.Attribute):
+        if (hasattr(node.value, 'id') and node.value.id == 't' 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 CheckAnnotationHasCInline(annotation_node) -> bool:
+    """检测注解节点中是否包含 CInline"""
+    if annotation_node is None:
+        return False
+    if isinstance(annotation_node, ast.Attribute):
+        if hasattr(annotation_node.value, 'id') and annotation_node.value.id == 't' 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