Introduce a recursive AST visitor that makes it trivial to recursively

walk an entire AST, including all of the types, declarations,
statements, and expressions, and allowing one to easily override the
behavior of the walk at any particular node kind.

llvm-svn: 103308

1 files changed, 4 insertions, 1 deletions

diff --git a/clang/lib/Frontend/BoostConAction.cpp b/clang/lib/Frontend/BoostConAction.cpp
index 505d5afe3e7..6995969fdf9 100644
--- a/clang/lib/Frontend/BoostConAction.cpp
+++ b/clang/lib/Frontend/BoostConAction.cpp
@@ -8,11 +8,13 @@
 //===----------------------------------------------------------------------===//
 #include "clang/Frontend/FrontendActions.h"
 #include "clang/AST/ASTConsumer.h"
+#include "clang/AST/RecursiveASTVisitor.h"
 #include <cstdio>
 using namespace clang;
 
 namespace {
-  class BoostConASTConsumer : public ASTConsumer {
+  class BoostConASTConsumer : public ASTConsumer, 
+                              public RecursiveASTVisitor<BoostConASTConsumer> {
   public:
     /// HandleTranslationUnit - This method is called when the ASTs for entire
     /// translation unit have been parsed.
@@ -27,4 +29,5 @@ ASTConsumer *BoostConAction::CreateASTConsumer(CompilerInstance &CI,
 
 void BoostConASTConsumer::HandleTranslationUnit(ASTContext &Ctx) {
   fprintf(stderr, "Welcome to BoostCon!\n");
+  Visit(Ctx.getTranslationUnitDecl());
 }