c++ - Clang 使用 LibTooling Rewriter 生成新文件？

Question

我正在使用 LibTooling 进行一些分析。我知道如何遍历 AST 并将一些文本插入某处。例如，

Rewriter mywriter;
mywriter.InsertTextAfter(func->getLocEnd(),"Hello");

现在我想知道是否有任何方法可以保存代码？（天气保存到原始文件或生成新文件）

因为分析后，我只能在终端上读取结果，这对我来说还不够。

Answer 1

它应该在 EndSourceFileAction() 函数中的 ASTFrontEndAction 对象中完成

// 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());
  }

在这种情况下 output.txt 是您想要的输出文件。

Answer 2

mywriter.overwriteChangedFiles();