1//= CFGRecStmtDeclVisitor - Recursive visitor of CFG stmts/decls -*- C++ --*-=// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file implements the template class CFGRecStmtDeclVisitor, which extends 11// CFGRecStmtVisitor by implementing (typed) visitation of decls. 12// 13// FIXME: This may not be fully complete. We currently explore only subtypes 14// of ScopedDecl. 15//===----------------------------------------------------------------------===// 16 17#ifndef LLVM_CLANG_ANALYSIS_CFG_REC_STMT_DECL_VISITOR_H 18#define LLVM_CLANG_ANALYSIS_CFG_REC_STMT_DECL_VISITOR_H 19 20#include "clang/Analysis/Visitors/CFGRecStmtVisitor.h" 21#include "clang/AST/Decl.h" 22#include "clang/AST/DeclObjC.h" 23#include "clang/AST/DeclCXX.h" 24 25#define DISPATCH_CASE(CLASS) \ 26case Decl::CLASS: \ 27static_cast<ImplClass*>(this)->Visit##CLASS##Decl( \ 28 static_cast<CLASS##Decl*>(D)); \ 29break; 30 31#define DEFAULT_DISPATCH(CLASS) void Visit##CLASS##Decl(CLASS##Decl *D) {} 32#define DEFAULT_DISPATCH_VARDECL(CLASS) void Visit##CLASS##Decl(CLASS##Decl *D)\ 33 { static_cast<ImplClass*>(this)->VisitVarDecl(D); } 34 35 36namespace clang { 37template <typename ImplClass> 38class CFGRecStmtDeclVisitor : public CFGRecStmtVisitor<ImplClass> { 39public: 40 41 void VisitDeclRefExpr(DeclRefExpr *DR) { 42 static_cast<ImplClass*>(this)->VisitDecl(DR->getDecl()); 43 } 44 45 void VisitDeclStmt(DeclStmt *DS) { 46 for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end(); 47 DI != DE; ++DI) { 48 Decl *D = *DI; 49 static_cast<ImplClass*>(this)->VisitDecl(D); 50 // Visit the initializer. 51 if (VarDecl *VD = dyn_cast<VarDecl>(D)) 52 if (Expr *I = VD->getInit()) 53 static_cast<ImplClass*>(this)->Visit(I); 54 } 55 } 56 57 void VisitDecl(Decl *D) { 58 switch (D->getKind()) { 59 DISPATCH_CASE(Function) 60 DISPATCH_CASE(CXXMethod) 61 DISPATCH_CASE(Var) 62 DISPATCH_CASE(ParmVar) // FIXME: (same) 63 DISPATCH_CASE(ImplicitParam) 64 DISPATCH_CASE(EnumConstant) 65 DISPATCH_CASE(Typedef) 66 DISPATCH_CASE(Record) // FIXME: Refine. VisitStructDecl? 67 DISPATCH_CASE(CXXRecord) 68 DISPATCH_CASE(Enum) 69 DISPATCH_CASE(Field) 70 DISPATCH_CASE(UsingDirective) 71 DISPATCH_CASE(Using) 72 DISPATCH_CASE(NamespaceAlias) 73 default: 74 llvm_unreachable("Subtype of ScopedDecl not handled."); 75 } 76 } 77 78 DEFAULT_DISPATCH(Var) 79 DEFAULT_DISPATCH(Function) 80 DEFAULT_DISPATCH(CXXMethod) 81 DEFAULT_DISPATCH_VARDECL(ParmVar) 82 DEFAULT_DISPATCH(ImplicitParam) 83 DEFAULT_DISPATCH(EnumConstant) 84 DEFAULT_DISPATCH(Typedef) 85 DEFAULT_DISPATCH(Record) 86 DEFAULT_DISPATCH(Enum) 87 DEFAULT_DISPATCH(Field) 88 DEFAULT_DISPATCH(ObjCInterface) 89 DEFAULT_DISPATCH(ObjCMethod) 90 DEFAULT_DISPATCH(ObjCProtocol) 91 DEFAULT_DISPATCH(ObjCCategory) 92 DEFAULT_DISPATCH(UsingDirective) 93 DEFAULT_DISPATCH(Using) 94 DEFAULT_DISPATCH(NamespaceAlias) 95 96 void VisitCXXRecordDecl(CXXRecordDecl *D) { 97 static_cast<ImplClass*>(this)->VisitRecordDecl(D); 98 } 99}; 100 101} // end namespace clang 102 103#undef DISPATCH_CASE 104#undef DEFAULT_DISPATCH 105#endif 106