You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
2.0 KiB
57 lines
2.0 KiB
# -*- coding: utf-8 -*-
|
|
import clang.cindex
|
|
from clang.cindex import CursorKind
|
|
|
|
|
|
class AST_Parser:
|
|
def __init__(self,library_file=None):
|
|
if not clang.cindex.Config.loaded:
|
|
print("Using libclang from: %s", library_file)
|
|
clang.cindex.Config.set_library_file(library_file)
|
|
|
|
def parse(self, filename, content):
|
|
index = clang.cindex.Index.create()
|
|
arguments = ["-x", "c"]
|
|
options = clang.cindex.TranslationUnit.PARSE_SKIP_FUNCTION_BODIES
|
|
content = [(filename, content)]
|
|
translation_unit = index.parse(filename, unsaved_files=content, args=arguments, options=options)
|
|
ret = self._parse(translation_unit.cursor, filename)
|
|
return ret
|
|
|
|
def _get_children_filelocal(self, cursor, path):
|
|
return [c for c in cursor.get_children() if c.location.file and c.location.file.name == path]
|
|
|
|
def _parse(self, cursor, path):
|
|
item = {}
|
|
excluded_cursortypes = [CursorKind.INTEGER_LITERAL]
|
|
if not cursor.kind in excluded_cursortypes:
|
|
if not str(cursor.kind) == "":
|
|
item["kind"] = str(cursor.kind)
|
|
|
|
if not cursor.spelling == "":
|
|
item["name"] = cursor.spelling
|
|
|
|
if not cursor.displayname == "":
|
|
item["displayname"] = cursor.displayname
|
|
|
|
if not cursor.type.spelling == "":
|
|
item["type"] = cursor.type.spelling
|
|
|
|
if not cursor.result_type.spelling == "":
|
|
item["result_type"] = cursor.result_type.spelling
|
|
|
|
if cursor.kind == CursorKind.ENUM_CONSTANT_DECL:
|
|
item["value"] = cursor.enum_value
|
|
|
|
child_cursors = self._get_children_filelocal(cursor, path)
|
|
if len(child_cursors) > 0:
|
|
child_arr = []
|
|
for child_cursor in child_cursors:
|
|
child_result = self._parse(child_cursor, path)
|
|
if child_result:
|
|
child_arr.append(child_result)
|
|
|
|
if child_arr:
|
|
item["children"] = child_arr
|
|
|
|
return item
|
|
|