Clang libtooling:如何打印编译器宏定义

#include <iostream>
#include <string>
#include <vector>
#include "clang/AST/ASTConsumer.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/Decl.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/Frontend/CompilerInstance.h"
#include "clang/Frontend/FrontendAction.h"
#include "clang/Frontend/FrontendActions.h"
#include "clang/Tooling/CommonOptionsParser.h"
#include "clang/Tooling/Tooling.h"

using namespace clang;
using namespace clang::tooling;

class MacroPrinter : public RecursiveASTVisitor<MacroPrinter> {
public:
  explicit MacroPrinter(ASTContext *Context) : Context(Context) {}

  bool VisitVarDecl(VarDecl *VD) {
    if (VD->hasAttrs()) {
      for (const auto *Attr : VD->attrs()) {
        if (const auto *MacroAttr = dyn_cast<MacroAttribute>(Attr)) {
          std::cout << "Macro definition: " << MacroAttr->getMacroName()
                    << std::endl;
        }
      }
    }
    return true;
  }

private:
  ASTContext *Context;
};

class MacroPrinterConsumer : public clang::ASTConsumer {
public:
  explicit MacroPrinterConsumer(ASTContext *Context)
      : Visitor(Context) {}

  void HandleTranslationUnit(clang::ASTContext &Context) override {
    Visitor.TraverseDecl(Context.getTranslationUnitDecl());
  }

private:
  MacroPrinter Visitor;
};

class MacroPrinterAction : public clang::ASTFrontendAction {
public:
  std::unique_ptr<clang::ASTConsumer>
  CreateASTConsumer(clang::CompilerInstance &Compiler,
                    llvm::StringRef InFile) override {
    return std::make_unique<MacroPrinterConsumer>(&Compiler.getASTContext());
  }
};

int main(int argc, const char **argv) {
  CommonOptionsParser OptionsParser(argc, argv);
  ClangTool Tool(OptionsParser.getCompilations(),
                 OptionsParser.getSourcePathList());
  return Tool.run(newFrontendActionFactory<MacroPrinterAction>().get());
}

clang++ -std=c++11 -I <path_to_clang_include> -I <path_to_llvm_include> -L <path_to_clang_lib> -lclangTooling -lclangASTMatchers -lclangAST -lclangFrontend -lclangSerialization -lclangDriver -lclangParse -lclangSema -lclangAnalysis -lclangEdit -lclangAST -lclangLex -lclangBasic -o macro_printer macro_printer.cpp
./macro_printer <source_file.cpp>