69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
"""Pass 3: SymbolInserter — 将解析后的符号插入 SymbolTable（主命名空间）"""
|
||
from __future__ import annotations
|
||
from typing import TYPE_CHECKING, List
|
||
if TYPE_CHECKING:
|
||
from lib.core.SymbolTable import SymbolTable
|
||
|
||
from lib.core.SymbolData import ModuleSymbols
|
||
|
||
|
||
class SymbolInserter:
|
||
"""将解析后的符号插入 SymbolTable 的主命名空间（无前缀）"""
|
||
|
||
def __init__(self, symbol_table: SymbolTable):
|
||
self._symtab = symbol_table
|
||
|
||
def insert(self, module_symbols: ModuleSymbols) -> List[str]:
|
||
"""插入所有符号到主命名空间，返回已加载的符号名列表"""
|
||
loaded = []
|
||
file_path = module_symbols.file_path
|
||
|
||
# 插入类符号
|
||
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 cls.type_kind == 'enum':
|
||
for member_name, _member_value, member_lineno in cls.enum_members:
|
||
self._symtab._InsertEnumMemberSymbol(member_name, cls.name, member_lineno, file_path)
|
||
loaded.append(member_name)
|
||
class_member_name = f"{cls.name}.{member_name}"
|
||
self._symtab._InsertEnumMemberSymbol(class_member_name, cls.name, member_lineno, file_path)
|
||
loaded.append(class_member_name)
|
||
|
||
# 插入 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)
|
||
|
||
# 插入函数符号
|
||
for func in module_symbols.functions:
|
||
self._symtab._InsertFuncSymbol(
|
||
func.name, func.ret_type, func.param_types,
|
||
func.lineno, file_path, func.is_variadic, func.is_inline
|
||
)
|
||
loaded.append(func.name)
|
||
|
||
# 插入 define 符号
|
||
for define in module_symbols.defines:
|
||
self._symtab._InsertDefineSymbol(
|
||
define.name, define.value, define.lineno, file_path
|
||
)
|
||
loaded.append(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
|