def visitor(ast_node, ast_parent_node, parser):
ast_node_location = ast_node.location
ast_node_tunit_spelling = ast_node.translation_unit.spelling
if (ast_node_location.file and ast_node_location.file.name == ast_node_tunit_spelling): # we are not interested in symbols which got into this TU via includes
id = parser.get_ast_node_id(ast_node)
usr = ast_node.referenced.get_usr() if ast_node.referenced else ast_node.get_usr()
line = int(parser.get_ast_node_line(ast_node))
column = int(parser.get_ast_node_column(ast_node))
if id in [
ASTNodeId.getClassId(), ASTNodeId.getStructId(), ASTNodeId.getEnumId(), ASTNodeId.getEnumValueId(), # handle user-defined types
ASTNodeId.getUnionId(), ASTNodeId.getTypedefId(), ASTNodeId.getUsingDeclarationId(),
ASTNodeId.getFunctionId(), ASTNodeId.getMethodId(), # handle functions and methods
ASTNodeId.getLocalVariableId(), ASTNodeId.getFunctionParameterId(), ASTNodeId.getFieldId(), # handle local/function variables and member variables
ASTNodeId.getMacroDefinitionId(), ASTNodeId.getMacroInstantiationId() # handle macros
]:
symbol_db.insert_single(
get_basename(root_directory, ast_node_tunit_spelling),
line,
column,
usr,
extract_cursor_context(ast_node_tunit_spelling, line),
ast_node.referenced._kind_id if ast_node.referenced else ast_node._kind_id,
ast_node.is_definition()
)
else:
pass
return ChildVisitResult.RECURSE.value # If we are positioned in TU of interest, then we'll traverse through all descendants
return ChildVisitResult.CONTINUE.value # Otherwise, we'll skip to the next sibling
def to_ast_node_id(kind):
if (kind == clang.cindex.CursorKind.NAMESPACE):
return ASTNodeId.getNamespaceId()
if (kind in [
clang.cindex.CursorKind.CLASS_DECL,
clang.cindex.CursorKind.CLASS_TEMPLATE,
clang.cindex.CursorKind.CLASS_TEMPLATE_PARTIAL_SPECIALIZATION
]):
return ASTNodeId.getClassId()
if (kind == clang.cindex.CursorKind.STRUCT_DECL):
return ASTNodeId.getStructId()
if (kind == clang.cindex.CursorKind.ENUM_DECL):
return ASTNodeId.getEnumId()
if (kind == clang.cindex.CursorKind.ENUM_CONSTANT_DECL):
return ASTNodeId.getEnumValueId()
if (kind == clang.cindex.CursorKind.UNION_DECL):
return ASTNodeId.getUnionId()
if (kind == clang.cindex.CursorKind.FIELD_DECL):
return ASTNodeId.getFieldId()
if (kind == clang.cindex.CursorKind.VAR_DECL):
return ASTNodeId.getLocalVariableId()
if (kind in [
clang.cindex.CursorKind.FUNCTION_DECL,
clang.cindex.CursorKind.FUNCTION_TEMPLATE
]):
return ASTNodeId.getFunctionId()
if (kind in [
clang.cindex.CursorKind.CXX_METHOD,
clang.cindex.CursorKind.CONSTRUCTOR,
clang.cindex.CursorKind.DESTRUCTOR
]):
return ASTNodeId.getMethodId()
if (kind == clang.cindex.CursorKind.PARM_DECL):
return ASTNodeId.getFunctionParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_TYPE_PARAMETER):
return ASTNodeId.getTemplateTypeParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_NON_TYPE_PARAMETER):
return ASTNodeId.getTemplateNonTypeParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_TEMPLATE_PARAMETER):
return ASTNodeId.getTemplateTemplateParameterId()
if (kind == clang.cindex.CursorKind.MACRO_DEFINITION):
return ASTNodeId.getMacroDefinitionId()
if (kind == clang.cindex.CursorKind.MACRO_INSTANTIATION):
return ASTNodeId.getMacroInstantiationId()
if (kind in [
clang.cindex.CursorKind.TYPEDEF_DECL,
clang.cindex.CursorKind.TYPE_ALIAS_DECL
]):
return ASTNodeId.getTypedefId()
if (kind == clang.cindex.CursorKind.NAMESPACE_ALIAS):
return ASTNodeId.getNamespaceAliasId()
if (kind == clang.cindex.CursorKind.USING_DIRECTIVE):
return ASTNodeId.getUsingDirectiveId()
if (kind == clang.cindex.CursorKind.USING_DECLARATION):
return ASTNodeId.getUsingDeclarationId()
return ASTNodeId.getUnsupportedId()
def to_ast_node_id(kind):
if (kind == clang.cindex.CursorKind.NAMESPACE):
return ASTNodeId.getNamespaceId()
if (kind in [clang.cindex.CursorKind.CLASS_DECL, clang.cindex.CursorKind.CLASS_TEMPLATE, clang.cindex.CursorKind.CLASS_TEMPLATE_PARTIAL_SPECIALIZATION]):
return ASTNodeId.getClassId()
if (kind == clang.cindex.CursorKind.STRUCT_DECL):
return ASTNodeId.getStructId()
if (kind == clang.cindex.CursorKind.ENUM_DECL):
return ASTNodeId.getEnumId()
if (kind == clang.cindex.CursorKind.ENUM_CONSTANT_DECL):
return ASTNodeId.getEnumValueId()
if (kind == clang.cindex.CursorKind.UNION_DECL):
return ASTNodeId.getUnionId()
if (kind == clang.cindex.CursorKind.FIELD_DECL):
return ASTNodeId.getFieldId()
if (kind == clang.cindex.CursorKind.VAR_DECL):
return ASTNodeId.getLocalVariableId()
if (kind in [clang.cindex.CursorKind.FUNCTION_DECL, clang.cindex.CursorKind.FUNCTION_TEMPLATE]):
return ASTNodeId.getFunctionId()
if (kind in [clang.cindex.CursorKind.CXX_METHOD, clang.cindex.CursorKind.CONSTRUCTOR, clang.cindex.CursorKind.DESTRUCTOR]):
return ASTNodeId.getMethodId()
if (kind == clang.cindex.CursorKind.PARM_DECL):
return ASTNodeId.getFunctionParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_TYPE_PARAMETER):
return ASTNodeId.getTemplateTypeParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_NON_TYPE_PARAMETER):
return ASTNodeId.getTemplateNonTypeParameterId()
if (kind == clang.cindex.CursorKind.TEMPLATE_TEMPLATE_PARAMETER):
return ASTNodeId.getTemplateTemplateParameterId()
if (kind == clang.cindex.CursorKind.MACRO_DEFINITION):
return ASTNodeId.getMacroDefinitionId()
if (kind == clang.cindex.CursorKind.MACRO_INSTANTIATION):
return ASTNodeId.getMacroInstantiationId()
if (kind in [clang.cindex.CursorKind.TYPEDEF_DECL, clang.cindex.CursorKind.TYPE_ALIAS_DECL]):
return ASTNodeId.getTypedefId()
if (kind == clang.cindex.CursorKind.NAMESPACE_ALIAS):
return ASTNodeId.getNamespaceAliasId()
if (kind == clang.cindex.CursorKind.USING_DIRECTIVE):
return ASTNodeId.getUsingDirectiveId()
if (kind == clang.cindex.CursorKind.USING_DECLARATION):
return ASTNodeId.getUsingDeclarationId()
return ASTNodeId.getUnsupportedId()
def __tag_id_to_vim_syntax_group(tag_identifier):
if tag_identifier == ASTNodeId.getNamespaceId():
return "yavideCppNamespace"
if tag_identifier == ASTNodeId.getNamespaceAliasId():
return "yavideCppNamespaceAlias"
if tag_identifier == ASTNodeId.getClassId():
return "yavideCppClass"
if tag_identifier == ASTNodeId.getStructId():
return "yavideCppStructure"
if tag_identifier == ASTNodeId.getEnumId():
return "yavideCppEnum"
if tag_identifier == ASTNodeId.getEnumValueId():
return "yavideCppEnumValue"
if tag_identifier == ASTNodeId.getUnionId():
return "yavideCppUnion"
if tag_identifier == ASTNodeId.getFieldId():
return "yavideCppField"
if tag_identifier == ASTNodeId.getLocalVariableId():
return "yavideCppLocalVariable"
if tag_identifier == ASTNodeId.getFunctionId():
return "yavideCppFunction"
if tag_identifier == ASTNodeId.getMethodId():
return "yavideCppMethod"
if tag_identifier == ASTNodeId.getFunctionParameterId():
return "yavideCppFunctionParameter"
if tag_identifier == ASTNodeId.getTemplateTypeParameterId():
return "yavideCppTemplateTypeParameter"
if tag_identifier == ASTNodeId.getTemplateNonTypeParameterId():
return "yavideCppTemplateNonTypeParameter"
if tag_identifier == ASTNodeId.getTemplateTemplateParameterId():
return "yavideCppTemplateTemplateParameter"
if tag_identifier == ASTNodeId.getMacroDefinitionId():
return "yavideCppMacroDefinition"
if tag_identifier == ASTNodeId.getMacroInstantiationId():
return "yavideCppMacroInstantiation"
if tag_identifier == ASTNodeId.getTypedefId():
return "yavideCppTypedef"
if tag_identifier == ASTNodeId.getUsingDirectiveId():
return "yavideCppUsingDirective"
if tag_identifier == ASTNodeId.getUsingDeclarationId():
return "yavideCppUsingDeclaration"
def visitor(ast_node, ast_parent_node, parser):
ast_node_location = ast_node.location
ast_node_tunit_spelling = ast_node.translation_unit.spelling
if (
ast_node_location.file
and ast_node_location.file.name == ast_node_tunit_spelling
): # we are not interested in symbols which got into this TU via includes
id = parser.get_ast_node_id(ast_node)
usr = ast_node.referenced.get_usr(
) if ast_node.referenced else ast_node.get_usr()
line = int(parser.get_ast_node_line(ast_node))
column = int(parser.get_ast_node_column(ast_node))
if id in [
ASTNodeId.getClassId(),
ASTNodeId.getStructId(),
ASTNodeId.getEnumId(),
ASTNodeId.getEnumValueId(), # handle user-defined types
ASTNodeId.getUnionId(),
ASTNodeId.getTypedefId(),
ASTNodeId.getUsingDeclarationId(),
ASTNodeId.getFunctionId(),
ASTNodeId.getMethodId(), # handle functions and methods
ASTNodeId.getLocalVariableId(),
ASTNodeId.getFunctionParameterId(),
ASTNodeId.getFieldId(
), # handle local/function variables and member variables
ASTNodeId.getMacroDefinitionId(),
ASTNodeId.getMacroInstantiationId() # handle macros
]:
symbol_db.insert_single(
get_basename(root_directory,
ast_node_tunit_spelling), line, column, usr,
extract_cursor_context(ast_node_tunit_spelling, line),
ast_node.referenced._kind_id
if ast_node.referenced else ast_node._kind_id,
ast_node.is_definition())
else:
pass
return ChildVisitResult.RECURSE.value # If we are positioned in TU of interest, then we'll traverse through all descendants
return ChildVisitResult.CONTINUE.value # Otherwise, we'll skip to the next sibling