diff options
author | Pirama Arumuga Nainar <pirama@google.com> | 2015-07-01 12:25:36 -0700 |
---|---|---|
committer | Pirama Arumuga Nainar <pirama@google.com> | 2015-07-01 12:25:36 -0700 |
commit | b6d6993e6e6d3daf4d9876794254d20a134e37c2 (patch) | |
tree | 56c1af986f9588ac13a9f88597ce0738a574c6b9 /examples | |
parent | e428a2cde8a5a5aba99db38a7d7f6a0b7ee823d2 (diff) | |
download | clang-b6d6993e6e6d3daf4d9876794254d20a134e37c2.tar.gz |
Update aosp/master clang for rebase to r239765
Change-Id: I0393bcc952590a7226af8c4b58534a8ee5fd2d99
Diffstat (limited to 'examples')
-rw-r--r-- | examples/PrintFunctionNames/PrintFunctionNames.cpp | 55 |
1 files changed, 53 insertions, 2 deletions
diff --git a/examples/PrintFunctionNames/PrintFunctionNames.cpp b/examples/PrintFunctionNames/PrintFunctionNames.cpp index 39443f597f..9f8f6e3f05 100644 --- a/examples/PrintFunctionNames/PrintFunctionNames.cpp +++ b/examples/PrintFunctionNames/PrintFunctionNames.cpp @@ -15,14 +15,23 @@ #include "clang/Frontend/FrontendPluginRegistry.h" #include "clang/AST/AST.h" #include "clang/AST/ASTConsumer.h" +#include "clang/AST/RecursiveASTVisitor.h" #include "clang/Frontend/CompilerInstance.h" +#include "clang/Sema/Sema.h" #include "llvm/Support/raw_ostream.h" using namespace clang; namespace { class PrintFunctionsConsumer : public ASTConsumer { + CompilerInstance &Instance; + std::set<std::string> ParsedTemplates; + public: + PrintFunctionsConsumer(CompilerInstance &Instance, + std::set<std::string> ParsedTemplates) + : Instance(Instance), ParsedTemplates(ParsedTemplates) {} + bool HandleTopLevelDecl(DeclGroupRef DG) override { for (DeclGroupRef::iterator i = DG.begin(), e = DG.end(); i != e; ++i) { const Decl *D = *i; @@ -32,13 +41,47 @@ public: return true; } + + void HandleTranslationUnit(ASTContext& context) override { + if (!Instance.getLangOpts().DelayedTemplateParsing) + return; + + // This demonstrates how to force instantiation of some templates in + // -fdelayed-template-parsing mode. (Note: Doing this unconditionally for + // all templates is similar to not using -fdelayed-template-parsig in the + // first place.) + // The advantage of doing this in HandleTranslationUnit() is that all + // codegen (when using -add-plugin) is completely finished and this can't + // affect the compiler output. + struct Visitor : public RecursiveASTVisitor<Visitor> { + const std::set<std::string> &ParsedTemplates; + Visitor(const std::set<std::string> &ParsedTemplates) + : ParsedTemplates(ParsedTemplates) {} + bool VisitFunctionDecl(FunctionDecl *FD) { + if (FD->isLateTemplateParsed() && + ParsedTemplates.count(FD->getNameAsString())) + LateParsedDecls.insert(FD); + return true; + } + + std::set<FunctionDecl*> LateParsedDecls; + } v(ParsedTemplates); + v.TraverseDecl(context.getTranslationUnitDecl()); + clang::Sema &sema = Instance.getSema(); + for (const FunctionDecl *FD : v.LateParsedDecls) { + clang::LateParsedTemplate* LPT = sema.LateParsedTemplateMap.lookup(FD); + sema.LateTemplateParser(sema.OpaqueParser, *LPT); + llvm::errs() << "late-parsed-decl: \"" << FD->getNameAsString() << "\"\n"; + } + } }; class PrintFunctionNamesAction : public PluginASTAction { + std::set<std::string> ParsedTemplates; protected: std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, llvm::StringRef) override { - return llvm::make_unique<PrintFunctionsConsumer>(); + return llvm::make_unique<PrintFunctionsConsumer>(CI, ParsedTemplates); } bool ParseArgs(const CompilerInstance &CI, @@ -47,12 +90,20 @@ protected: llvm::errs() << "PrintFunctionNames arg = " << args[i] << "\n"; // Example error handling. + DiagnosticsEngine &D = CI.getDiagnostics(); if (args[i] == "-an-error") { - DiagnosticsEngine &D = CI.getDiagnostics(); unsigned DiagID = D.getCustomDiagID(DiagnosticsEngine::Error, "invalid argument '%0'"); D.Report(DiagID) << args[i]; return false; + } else if (args[i] == "-parse-template") { + if (i + 1 >= e) { + D.Report(D.getCustomDiagID(DiagnosticsEngine::Error, + "missing -parse-template argument")); + return false; + } + ++i; + ParsedTemplates.insert(args[i]); } } if (!args.empty() && args[0] == "help") |