Added method "HandleTranslationUnit" to ASTConsumer. This is called by ParseAST...

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@53042 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/clang/AST/ASTConsumer.h b/include/clang/AST/ASTConsumer.h
index c08f97e640944fc5b2028f46d04dd35a3218a541..b9513be240bc6c453a33701c09326a4cc46a2ab9 100644 (file)
--- a/include/clang/AST/ASTConsumer.h
+++ b/include/clang/AST/ASTConsumer.h
@@ -37,7 +37,10 @@ public:
   /// HandleTopLevelDecl - Handle the specified top-level declaration.  This is
   ///  called by HandleTopLevelDeclaration to process every top-level Decl*.
   virtual void HandleTopLevelDecl(Decl *D) {}
-    
+  
+  /// HandleTranslationUnit - This method is called when the ASTs for entire
+  ///  translation unit have been parsed.
+  virtual void HandleTranslationUnit(TranslationUnit& TU) {}    
   
   /// HandleTopLevelDeclaration - Handle the specified top-level declaration.
   ///  This is called only for Decl* that are the head of a chain of

diff --git a/lib/Sema/ParseAST.cpp b/lib/Sema/ParseAST.cpp
index dce577bc4c21a9cb6e67c74fe7bec5be7573dc47..94e0185f03f92c8bc42057ec321fb8efd0043721 100644 (file)
--- a/lib/Sema/ParseAST.cpp
+++ b/lib/Sema/ParseAST.cpp
@@ -63,6 +63,8 @@ void clang::ParseAST(Preprocessor &PP, ASTConsumer *Consumer, bool PrintStats) {
       Consumer->HandleTopLevelDecl(D);
     }
   };
+  
+  Consumer->HandleTranslationUnit(TU);
 
   if (PrintStats) {
     fprintf(stderr, "\nSTATISTICS:\n");