2017-04-01 1 views
2

J'utilise LibTooling pour faire une analyse. Je sais comment traverser l'AST et insérer du texte quelque part. Par exemple,Clang utilisant LibTooling Rewriter pour générer un nouveau fichier?

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

Maintenant, je me demande s'il existe un moyen d'enregistrer le code? Parce que après l'analyse, je peux seulement lire le résultat sur le terminal et ce n'est pas assez pour moi.

Répondre

3

il devrait être fait dans votre objet ASTFrontEndAction, dans la fonction 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()); 
    } 

Dans ce sortie.txt cas est le fichier de sortie que vous vouliez.

1

mywriter.overwriteChangedFiles();