2017-04-01 8 views
2

저는 LibTooling을 사용하여 분석을하고 있습니다. 나는 AST를 횡단하고 어딘가에 어떤 텍스트를 삽입하는 법을 안다. 예 :LibTooling Rewriter를 사용하여 새 파일을 생성 하시겠습니까?

이제 코드를 저장할 방법이 있는지 궁금합니다. (날씨는 원본 파일에 저장하거나 새 파일을 생성합니다.)

분석 후에는 터미널에서만 결과를 읽을 수 있기 때문에 충분하지 않습니다.

답변

3

은이 경우의 경우 output.txt에서 EndSourceFileAction() 함수

// For each source file provided to the tool, a new FrontendAction is created. 
class MyFrontendAction : public ASTFrontendAction { 
public: 
    MyFrontendAction() {} 
    void EndSourceFileAction() override { 
    SourceManager &SM = TheRewriter.getSourceMgr(); 
    llvm::errs() << "** EndSourceFileAction for: " 
       << SM.getFileEntryForID(SM.getMainFileID())->getName() << "\n"; 

    // Now emit the rewritten buffer. 
    // TheRewriter.getEditBuffer(SM.getMainFileID()).write(llvm::outs()); --> this will output to screen as what you got. 
    std::error_code error_code; 
    llvm::raw_fd_ostream outFile("output.txt", error_code, llvm::sys::fs::F_None); 
    TheRewriter.getEditBuffer(SM.getMainFileID()).write(outFile); // --> this will write the result to outFile 
    outFile.close(); 
    } 
//as normal, make sure it matches your ASTConsumer constructor 
    std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, 
               StringRef file) override { 

    llvm::errs() << "** Creating AST consumer for: " << file << "\n"; 
    TheRewriter.setSourceMgr(CI.getSourceManager(), CI.getLangOpts()); 
    return llvm::make_unique<MyASTConsumer>(TheRewriter,&CI.getASTContext()); 
    } 

내에서, 당신의 ASTFrontEndAction 객체에서 당신이 원하는 출력 파일입니다 수행해야합니다.

1

mywriter.overwriteChangedFiles();