contrib/llvm/tools/clang/include/clang/Analysis/Visitors/CFGRecStmtDeclVisitor.h

   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/AST/Decl.h"
  21 #include "clang/AST/DeclCXX.h"
  22 #include "clang/AST/DeclObjC.h"
  23 #include "clang/Analysis/Visitors/CFGRecStmtVisitor.h"
  24
  25 #define DISPATCH_CASE(CLASS)                                  \
  26 case Decl::CLASS:                                             \
  27 static_cast<ImplClass*>(this)->Visit##CLASS##Decl(            \
  28                                static_cast<CLASS##Decl*>(D)); \
  29 break;
  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
  36 namespace clang {
  37 template <typename ImplClass>
  38 class CFGRecStmtDeclVisitor : public CFGRecStmtVisitor<ImplClass> {
  39 public:
  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(TypeAlias)
  67         DISPATCH_CASE(Record)    // FIXME: Refine.  VisitStructDecl?
  68         DISPATCH_CASE(CXXRecord)
  69         DISPATCH_CASE(Enum)
  70         DISPATCH_CASE(Field)
  71         DISPATCH_CASE(UsingDirective)
  72         DISPATCH_CASE(Using)
  73         DISPATCH_CASE(NamespaceAlias)
  74       default:
  75         llvm_unreachable("Subtype of ScopedDecl not handled.");
  76     }
  77   }
  78
  79   DEFAULT_DISPATCH(Var)
  80   DEFAULT_DISPATCH(Function)
  81   DEFAULT_DISPATCH(CXXMethod)
  82   DEFAULT_DISPATCH_VARDECL(ParmVar)
  83   DEFAULT_DISPATCH(ImplicitParam)
  84   DEFAULT_DISPATCH(EnumConstant)
  85   DEFAULT_DISPATCH(Typedef)
  86   DEFAULT_DISPATCH(TypeAlias)
  87   DEFAULT_DISPATCH(Record)
  88   DEFAULT_DISPATCH(Enum)
  89   DEFAULT_DISPATCH(Field)
  90   DEFAULT_DISPATCH(ObjCInterface)
  91   DEFAULT_DISPATCH(ObjCMethod)
  92   DEFAULT_DISPATCH(ObjCProtocol)
  93   DEFAULT_DISPATCH(ObjCCategory)
  94   DEFAULT_DISPATCH(UsingDirective)
  95   DEFAULT_DISPATCH(Using)
  96   DEFAULT_DISPATCH(NamespaceAlias)
  97
  98   void VisitCXXRecordDecl(CXXRecordDecl *D) {
  99     static_cast<ImplClass*>(this)->VisitRecordDecl(D);
 100   }
 101 };
 102
 103 } // end namespace clang
 104
 105 #undef DISPATCH_CASE
 106 #undef DEFAULT_DISPATCH
 107 #endif