You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
import os
|
|
import json
|
|
from clang.cindex import Index, CursorKind
|
|
|
|
def load_compile_flags(compile_commands_path, cpp_stub_path):
|
|
with open(compile_commands_path) as f:
|
|
compile_commands = json.load(f)
|
|
|
|
for entry in compile_commands:
|
|
if os.path.abspath(entry['file']) == os.path.abspath(cpp_stub_path):
|
|
cmd = entry['arguments'] if 'arguments' in entry else entry['command'].split()
|
|
if "-c" in cmd:
|
|
del cmd[cmd.index("-c") + 1]
|
|
del cmd[cmd.index("-c")]
|
|
return [arg for arg in cmd if not arg.endswith('g++') and not arg.endswith('clang++')]
|
|
return []
|
|
|
|
def has_annotation(cursor, annotation_text):
|
|
return any(
|
|
c.kind == CursorKind.ANNOTATE_ATTR and annotation_text in c.spelling
|
|
for c in cursor.get_children()
|
|
)
|
|
|
|
def main():
|
|
cpp_stub_path = os.path.abspath("src/ICounter.cpp")
|
|
compile_commands_path = os.path.abspath("build/compile_commands.json")
|
|
|
|
args = load_compile_flags(compile_commands_path, cpp_stub_path)
|
|
|
|
print(args)
|
|
index = Index.create()
|
|
tu = index.parse(cpp_stub_path, args=args, options=0)
|
|
|
|
for cursor in tu.cursor.walk_preorder():
|
|
if cursor.kind == CursorKind.CLASS_DECL and has_annotation(cursor, "export"):
|
|
print(f"Found exported class: {cursor.spelling}")
|
|
for c in cursor.get_children():
|
|
if c.kind == CursorKind.CXX_METHOD:
|
|
is_virtual = c.is_virtual_method()
|
|
print(f" method: {c.spelling} (virtual={is_virtual})")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|