Question

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

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

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

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

Answer 1

它应该在你的ASTFrontEndAction对象中，在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());
  }

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

Answer 2

mywriter.overwriteChangedFiles（）;

Clang使用LibTooling Rewriter生成新文件？

2 个答案: