Insert interception point onStartOfTranslationUnit.
Often users of the ASTMatchers want to add tasks that are done once per translation unit, for example, cleaning up caches. Combined with the interception point for the end of source file one can add to the factory creation, this covers the cases we've seen users need. llvm-svn: 167271
This commit is contained in:
parent
7dcadc6dbe
commit
bd0e2b7111
|
@ -85,7 +85,14 @@ public:
|
|||
class MatchCallback {
|
||||
public:
|
||||
virtual ~MatchCallback();
|
||||
|
||||
/// \brief Called on every match by the \c MatchFinder.
|
||||
virtual void run(const MatchResult &Result) = 0;
|
||||
|
||||
/// \brief Called at the start of each translation unit.
|
||||
///
|
||||
/// Optionally override to do per translation unit tasks.
|
||||
virtual void onStartOfTranslationUnit() {}
|
||||
};
|
||||
|
||||
/// \brief Called when parsing is finished. Intended for testing only.
|
||||
|
|
|
@ -315,6 +315,15 @@ public:
|
|||
ActiveASTContext(NULL) {
|
||||
}
|
||||
|
||||
void onStartOfTranslationUnit() {
|
||||
for (std::vector<std::pair<const internal::DynTypedMatcher*,
|
||||
MatchCallback*> >::const_iterator
|
||||
I = MatcherCallbackPairs->begin(), E = MatcherCallbackPairs->end();
|
||||
I != E; ++I) {
|
||||
I->second->onStartOfTranslationUnit();
|
||||
}
|
||||
}
|
||||
|
||||
void set_active_ast_context(ASTContext *NewActiveASTContext) {
|
||||
ActiveASTContext = NewActiveASTContext;
|
||||
}
|
||||
|
@ -649,6 +658,7 @@ private:
|
|||
ParsingDone->run();
|
||||
}
|
||||
Visitor.set_active_ast_context(&Context);
|
||||
Visitor.onStartOfTranslationUnit();
|
||||
Visitor.TraverseDecl(Context.getTranslationUnitDecl());
|
||||
Visitor.set_active_ast_context(NULL);
|
||||
}
|
||||
|
|
|
@ -3366,5 +3366,26 @@ TEST(MatchFinder, CanMatchStatementsRecursively) {
|
|||
new VerifyRecursiveMatch<clang::Stmt>("if", declStmt())));
|
||||
}
|
||||
|
||||
class VerifyStartOfTranslationUnit : public MatchFinder::MatchCallback {
|
||||
public:
|
||||
VerifyStartOfTranslationUnit() : Called(false) {}
|
||||
virtual void run(const MatchFinder::MatchResult &Result) {
|
||||
EXPECT_TRUE(Called);
|
||||
}
|
||||
virtual void onStartOfTranslationUnit() {
|
||||
Called = true;
|
||||
}
|
||||
bool Called;
|
||||
};
|
||||
|
||||
TEST(MatchFinder, InterceptsStartOfTranslationUnit) {
|
||||
MatchFinder Finder;
|
||||
VerifyStartOfTranslationUnit VerifyCallback;
|
||||
Finder.addMatcher(decl(), &VerifyCallback);
|
||||
OwningPtr<FrontendActionFactory> Factory(newFrontendActionFactory(&Finder));
|
||||
ASSERT_TRUE(tooling::runToolOnCode(Factory->create(), "int x;"));
|
||||
EXPECT_TRUE(VerifyCallback.Called);
|
||||
}
|
||||
|
||||
} // end namespace ast_matchers
|
||||
} // end namespace clang
|
||||
|
|
Loading…
Reference in New Issue