1 files changed, 64 insertions, 0 deletions
diff --git a/include/clang/Analysis/Visitors/CFGVarDeclVisitor.h b/include/clang/Analysis/Visitors/CFGVarDeclVisitor.h
new file mode 100644
index 0000000..2510123
--- /dev/null
+++ b/include/clang/Analysis/Visitors/CFGVarDeclVisitor.h
@@ -0,0 +1,64 @@
+//==- CFGVarDeclVisitor - Generic visitor of VarDecls in a CFG --*- C++ --*-==//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the template class CFGVarDeclVisitor, which provides
+// a generic way to visit all the VarDecl's in a CFG.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_CLANG_ANALYSIS_CFG_VARDECL_VISITOR_H
+#define LLVM_CLANG_ANALYSIS_CFG_VARDECL_VISITOR_H
+
+#include "clang/Analysis/Visitors/CFGStmtVisitor.h"
+#include "clang/AST/Decl.h"
+#include "clang/AST/Stmt.h"
+#include "clang/AST/CFG.h"
+
+namespace clang {
+
+template <typename ImplClass>
+class CFGVarDeclVisitor : public CFGStmtVisitor<ImplClass> {
+  const CFG& cfg;
+public:  
+  CFGVarDeclVisitor(const CFG& c) : cfg(c) {}
+  
+  void VisitStmt(Stmt* S) {
+    static_cast<ImplClass*>(this)->VisitChildren(S);
+  }
+  
+  void VisitDeclRefExpr(DeclRefExpr* DR) {
+    static_cast<ImplClass*>(this)->VisitDeclChain(DR->getDecl());
+  }
+  
+  void VisitDeclStmt(DeclStmt* DS) {
+    static_cast<ImplClass*>(this)->VisitDeclChain(DS->getDecl());
+  }
+  
+  void VisitDeclChain(ScopedDecl* D) {  
+    for (; D != NULL ; D = D->getNextDeclarator())
+      static_cast<ImplClass*>(this)->VisitScopedDecl(D);
+  }
+  
+  void VisitScopedDecl(ScopedDecl* D) {
+    if (VarDecl* V = dyn_cast<VarDecl>(D))
+      static_cast<ImplClass*>(this)->VisitVarDecl(V);
+  }
+  
+  void VisitVarDecl(VarDecl* D) {}
+  
+  void VisitAllDecls() {
+    for (CFG::const_iterator BI = cfg.begin(), BE = cfg.end(); BI != BE; ++BI)
+      for (CFGBlock::const_iterator SI=BI->begin(),SE = BI->end();SI != SE;++SI)
+        static_cast<ImplClass*>(this)->BlockStmt_Visit(const_cast<Stmt*>(*SI));    
+  }
+};
+
+} // end namespace clang
+
+#endif