lib/Index/DeclReferenceMap.cpp

   1 //===--- DeclReferenceMap.cpp - Map Decls to their references -------------===//
   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 //  DeclReferenceMap creates a mapping from Decls to the ASTLocations that
  11 //  reference them.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "clang/Index/DeclReferenceMap.h"
  16 #include "clang/Index/ASTLocation.h"
  17 #include "ASTVisitor.h"
  18 using namespace clang;
  19 using namespace idx;
  20
  21 namespace {
  22
  23 class RefMapper : public ASTVisitor<RefMapper> {
  24   DeclReferenceMap::MapTy &Map;
  25
  26 public:
  27   RefMapper(DeclReferenceMap::MapTy &map) : Map(map) { }
  28
  29   void VisitDeclRefExpr(DeclRefExpr *Node);
  30   void VisitMemberExpr(MemberExpr *Node);
  31   void VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node);
  32
  33   void VisitTypedefTypeLoc(TypedefTypeLoc TL);
  34   void VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL);
  35 };
  36
  37 } // anonymous namespace
  38
  39 //===----------------------------------------------------------------------===//
  40 // RefMapper Implementation
  41 //===----------------------------------------------------------------------===//
  42
  43 void RefMapper::VisitDeclRefExpr(DeclRefExpr *Node) {
  44   NamedDecl *PrimD = cast<NamedDecl>(Node->getDecl()->getCanonicalDecl());
  45   Map.insert(std::make_pair(PrimD, ASTLocation(CurrentDecl, Node)));
  46 }
  47
  48 void RefMapper::VisitMemberExpr(MemberExpr *Node) {
  49   NamedDecl *PrimD = cast<NamedDecl>(Node->getMemberDecl()->getCanonicalDecl());
  50   Map.insert(std::make_pair(PrimD, ASTLocation(CurrentDecl, Node)));
  51 }
  52
  53 void RefMapper::VisitObjCIvarRefExpr(ObjCIvarRefExpr *Node) {
  54   Map.insert(std::make_pair(Node->getDecl(), ASTLocation(CurrentDecl, Node)));
  55 }
  56
  57 void RefMapper::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
  58   NamedDecl *ND = TL.getTypedefDecl();
  59   Map.insert(std::make_pair(ND, ASTLocation(CurrentDecl, ND, TL.getNameLoc())));
  60 }
  61
  62 void RefMapper::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
  63   NamedDecl *ND = TL.getIFaceDecl();
  64   Map.insert(std::make_pair(ND, ASTLocation(CurrentDecl, ND, TL.getNameLoc())));
  65 }
  66
  67 //===----------------------------------------------------------------------===//
  68 // DeclReferenceMap Implementation
  69 //===----------------------------------------------------------------------===//
  70
  71 DeclReferenceMap::DeclReferenceMap(ASTContext &Ctx) {
  72   RefMapper(Map).Visit(Ctx.getTranslationUnitDecl());
  73 }
  74
  75 DeclReferenceMap::astlocation_iterator
  76 DeclReferenceMap::refs_begin(NamedDecl *D) const {
  77   NamedDecl *Prim = cast<NamedDecl>(D->getCanonicalDecl());
  78   return astlocation_iterator(Map.lower_bound(Prim));
  79 }
  80
  81 DeclReferenceMap::astlocation_iterator
  82 DeclReferenceMap::refs_end(NamedDecl *D) const {
  83   NamedDecl *Prim = cast<NamedDecl>(D->getCanonicalDecl());
  84   return astlocation_iterator(Map.upper_bound(Prim));
  85 }
  86
  87 bool DeclReferenceMap::refs_empty(NamedDecl *D) const {
  88   NamedDecl *Prim = cast<NamedDecl>(D->getCanonicalDecl());
  89   return refs_begin(Prim) == refs_end(Prim);
  90 }