Extracting function declarations from a C/C++ header file using clang

Question

I have the following recursive AST visitor implementation.

class ExampleVisitor : public clang::RecursiveASTVisitor<ExampleVisitor>{
private:
    clang::ASTContext* ast_context_; // used for getting additional AST info

public:
    explicit ExampleVisitor(clang::CompilerInstance* ci) 
      : ast_context_(&(ci->getASTContext())) // initialize private members

virtual bool VisitFunctionDecl(clang::FunctionDecl* func)
{
    numFunctions++;
    foo(func);  
    return true;
}};

The function foo prints the names of the declared functions for a given input file.

In this implementation foo prints the functions that are declared in the input file as well as dumps all function declarations from the included header files. How can I modify this code to print only the functions declared in the given input file?

Some Who Call Me Tim · Accepted Answer · 2017-01-26 05:46:46Z

4

Try using the SourceManager to determine whether the FunctionDecl is in the main file of the translation unit:

virtual bool VisitFunctionDecl(clang::FunctionDecl* func)
{
  clang::SourceManager &sm(ast_context_->getSourceManager());
  bool const inMainFile(
    sm.isInMainFile(sm.getExpansionLoc(func->getLocStart())));
  if(inMainFile){
    numFunctions++;
    foo(func);
  }
  else{
    std::cout << "Skipping '" << func->getNameAsString() 
      << "' not in main file\n";
  }  
  return true;
}};

I happened to know that there is an AST Matcher called isExpansionInMainFile. I got the code above from the source for that matcher in cfe-3.9.0.src/include/clang/ASTMatchers/ASTMatchers.h lines 209-14.

answered Jan 26, 2017 at 5:46

Some Who Call Me Tim

1,1317 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Extracting function declarations from a C/C++ header file using clang

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related