Home | History | Annotate | Download | only in AST
      1 //===-- DeclLookups.h - Low-level interface to all names in a DC-*- 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 defines DeclContext::all_lookups_iterator.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #ifndef LLVM_CLANG_AST_DECLLOOKUPS_H
     15 #define LLVM_CLANG_AST_DECLLOOKUPS_H
     16 
     17 #include "clang/AST/ASTContext.h"
     18 #include "clang/AST/DeclBase.h"
     19 #include "clang/AST/DeclContextInternals.h"
     20 #include "clang/AST/DeclarationName.h"
     21 
     22 namespace clang {
     23 
     24 /// all_lookups_iterator - An iterator that provides a view over the results
     25 /// of looking up every possible name.
     26 class DeclContext::all_lookups_iterator {
     27   StoredDeclsMap::iterator It, End;
     28 public:
     29   typedef lookup_result             value_type;
     30   typedef lookup_result             reference;
     31   typedef lookup_result             pointer;
     32   typedef std::forward_iterator_tag iterator_category;
     33   typedef std::ptrdiff_t            difference_type;
     34 
     35   all_lookups_iterator() {}
     36   all_lookups_iterator(StoredDeclsMap::iterator It,
     37                        StoredDeclsMap::iterator End)
     38       : It(It), End(End) {}
     39 
     40   DeclarationName getLookupName() const { return It->first; }
     41 
     42   reference operator*() const { return It->second.getLookupResult(); }
     43   pointer operator->() const { return It->second.getLookupResult(); }
     44 
     45   all_lookups_iterator& operator++() {
     46     // Filter out using directives. They don't belong as results from name
     47     // lookup anyways, except as an implementation detail. Users of the API
     48     // should not expect to get them (or worse, rely on it).
     49     do {
     50       ++It;
     51     } while (It != End &&
     52              It->first == DeclarationName::getUsingDirectiveName());
     53 
     54     return *this;
     55   }
     56 
     57   all_lookups_iterator operator++(int) {
     58     all_lookups_iterator tmp(*this);
     59     ++(*this);
     60     return tmp;
     61   }
     62 
     63   friend bool operator==(all_lookups_iterator x, all_lookups_iterator y) {
     64     return x.It == y.It;
     65   }
     66   friend bool operator!=(all_lookups_iterator x, all_lookups_iterator y) {
     67     return x.It != y.It;
     68   }
     69 };
     70 
     71 inline DeclContext::all_lookups_iterator DeclContext::lookups_begin() const {
     72   DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext();
     73   if (Primary->hasExternalVisibleStorage())
     74     getParentASTContext().getExternalSource()->completeVisibleDeclsMap(Primary);
     75   if (StoredDeclsMap *Map = Primary->buildLookup())
     76     return all_lookups_iterator(Map->begin(), Map->end());
     77   return all_lookups_iterator();
     78 }
     79 
     80 inline DeclContext::all_lookups_iterator DeclContext::lookups_end() const {
     81   DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext();
     82   if (Primary->hasExternalVisibleStorage())
     83     getParentASTContext().getExternalSource()->completeVisibleDeclsMap(Primary);
     84   if (StoredDeclsMap *Map = Primary->buildLookup())
     85     return all_lookups_iterator(Map->end(), Map->end());
     86   return all_lookups_iterator();
     87 }
     88 
     89 inline
     90 DeclContext::all_lookups_iterator DeclContext::noload_lookups_begin() const {
     91   DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext();
     92   if (StoredDeclsMap *Map = Primary->getLookupPtr())
     93     return all_lookups_iterator(Map->begin(), Map->end());
     94   return all_lookups_iterator();
     95 }
     96 
     97 inline
     98 DeclContext::all_lookups_iterator DeclContext::noload_lookups_end() const {
     99   DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext();
    100   if (StoredDeclsMap *Map = Primary->getLookupPtr())
    101     return all_lookups_iterator(Map->end(), Map->end());
    102   return all_lookups_iterator();
    103 }
    104 
    105 } // end namespace clang
    106 
    107 #endif
    108