parse AST with Clang-example-白红宇

parse AST with Clang-example

阅读量：6965 次

发布时间：2019-06-27

本文共 4952 字，大约阅读时间需要 16 分钟。

#include 
    
     #include 
     
      #include 
      
       #include 
       
        #include "clang/AST/ASTConsumer.h"#include "clang/AST/RecursiveASTVisitor.h"#include "clang/Basic/Diagnostic.h"#include "clang/Basic/FileManager.h"#include "clang/Basic/SourceManager.h"#include "clang/Basic/TargetOptions.h"#include "clang/Basic/TargetInfo.h"#include "clang/Frontend/CompilerInstance.h"#include "clang/Lex/Preprocessor.h"#include "clang/Parse/ParseAST.h"#include "clang/Rewrite/Core/Rewriter.h"#include "clang/Rewrite/Frontend/Rewriters.h"#include "llvm/Support/Host.h"#include "llvm/Support/raw_ostream.h"using namespace clang;using namespace std;// By implementing RecursiveASTVisitor, we can specify which AST nodes// we're interested in by overriding relevant methods.class MyASTVisitor : public RecursiveASTVisitor
        
         {public:    MyASTVisitor(Rewriter &R)        : TheRewriter(R)    {}    bool VisitStmt(Stmt *s) {        // Only care about If statements.        if (isa
         
          (s)) {            IfStmt *IfStatement = cast
          
           (s);            Stmt *Then = IfStatement->getThen();            TheRewriter.InsertText(Then->getLocStart(),                                   "// the 'if' part\n",                                   true, true);            Stmt *Else = IfStatement->getElse();            if (Else)                TheRewriter.InsertText(Else->getLocStart(),                                       "// the 'else' part\n",                                       true, true);        }        return true;    }    bool VisitFunctionDecl(FunctionDecl *f) {        // Only function definitions (with bodies), not declarations.        if (f->hasBody()) {            Stmt *FuncBody = f->getBody();            // Type name as string           // QualType QT = f->getResultType();            QualType QT = f->getCallResultType();            string TypeStr = QT.getAsString();            // Function name            DeclarationName DeclName = f->getNameInfo().getName();            string FuncName = DeclName.getAsString();            // Add comment before            stringstream SSBefore;            SSBefore << "// Begin function " << FuncName << " returning "                     << TypeStr << "\n";            SourceLocation ST = f->getSourceRange().getBegin();            TheRewriter.InsertText(ST, SSBefore.str(), true, true);            // And after            stringstream SSAfter;            SSAfter << "\n// End function " << FuncName << "\n";            ST = FuncBody->getLocEnd().getLocWithOffset(1);            TheRewriter.InsertText(ST, SSAfter.str(), true, true);        }        return true;    }private:    void AddBraces(Stmt *s);    Rewriter &TheRewriter;};// Implementation of the ASTConsumer interface for reading an AST produced// by the Clang parser.class MyASTConsumer : public ASTConsumer{public:    MyASTConsumer(Rewriter &R)        : Visitor(R)    {}    // Override the method that gets called for each parsed top-level    // declaration.    virtual bool HandleTopLevelDecl(DeclGroupRef DR) {        for (DeclGroupRef::iterator b = DR.begin(), e = DR.end();             b != e; ++b)            // Traverse the declaration using our AST visitor.            Visitor.TraverseDecl(*b);        return true;    }private:    MyASTVisitor Visitor;};int main(int argc, char *argv[]){    if (argc != 2) {        llvm::errs() << "Usage: rewritersample 
           
            \n";        return 1;    }    // CompilerInstance will hold the instance of the Clang compiler for us,    // managing the various objects needed to run the compiler.    CompilerInstance TheCompInst;    //TheCompInst.createDiagnostics(0, 0);    TheCompInst.createDiagnostics();    // Initialize target info with the default triple for our platform.  /*    TargetOptions TO;    TO.Triple = llvm::sys::getDefaultTargetTriple();    TargetInfo *TI = TargetInfo::CreateTargetInfo( TheCompInst.getDiagnostics(), TO);    */     auto TO=std::make_shared
            
             ();     (*TO).Triple=llvm::sys::getDefaultTargetTriple();    TargetInfo *TI = TargetInfo::CreateTargetInfo( TheCompInst.getDiagnostics(), TO);    TheCompInst.setTarget(TI);    TheCompInst.createFileManager();    FileManager &FileMgr = TheCompInst.getFileManager();    TheCompInst.createSourceManager(FileMgr);    SourceManager &SourceMgr = TheCompInst.getSourceManager();   // TheCompInst.createPreprocessor();    ///TU_Complete   TU_Prefix  TU_Module    TheCompInst.createPreprocessor(TU_Complete);    TheCompInst.createASTContext();    // A Rewriter helps us manage the code rewriting task.    Rewriter TheRewriter;    TheRewriter.setSourceMgr(SourceMgr, TheCompInst.getLangOpts());    // Set the main file handled by the source manager to the input file.    const FileEntry *FileIn = FileMgr.getFile(argv[1]);   // SourceMgr.createMainFileID(FileIn);    ///因为原来的createMainFileID去掉啦    SourceMgr.setMainFileID(SourceMgr.createFileID(FileIn, SourceLocation(), SrcMgr::C_User));    TheCompInst.getDiagnosticClient().BeginSourceFile(        TheCompInst.getLangOpts(),        &TheCompInst.getPreprocessor());    // Create an AST consumer instance which is going to get called by    // ParseAST.    MyASTConsumer TheConsumer(TheRewriter);    // Parse the file to AST, registering our consumer as the AST consumer.    ParseAST(TheCompInst.getPreprocessor(), &TheConsumer,             TheCompInst.getASTContext());    // At this point the rewriter's buffer should be full with the rewritten    // file contents.    const RewriteBuffer *RewriteBuf =        TheRewriter.getRewriteBufferFor(SourceMgr.getMainFileID());    llvm::outs() << string(RewriteBuf->begin(), RewriteBuf->end());    return 0;}