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/DeclBase.h" 18 #include "clang/AST/DeclContextInternals.h" 19 #include "clang/AST/DeclarationName.h" 20 21 namespace clang { 22 23 /// all_lookups_iterator - An iterator that provides a view over the results 24 /// of looking up every possible name. 25 class DeclContext::all_lookups_iterator { 26 StoredDeclsMap::iterator It, End; 27 public: 28 typedef lookup_result value_type; 29 typedef lookup_result reference; 30 typedef lookup_result pointer; 31 typedef std::forward_iterator_tag iterator_category; 32 typedef std::ptrdiff_t difference_type; 33 34 all_lookups_iterator() {} 35 all_lookups_iterator(StoredDeclsMap::iterator It, 36 StoredDeclsMap::iterator End) 37 : It(It), End(End) {} 38 39 reference operator*() const { return It->second.getLookupResult(); } 40 pointer operator->() const { return It->second.getLookupResult(); } 41 42 all_lookups_iterator& operator++() { 43 // Filter out using directives. They don't belong as results from name 44 // lookup anyways, except as an implementation detail. Users of the API 45 // should not expect to get them (or worse, rely on it). 46 do { 47 ++It; 48 } while (It != End && 49 It->first == DeclarationName::getUsingDirectiveName()); 50 51 return *this; 52 } 53 54 all_lookups_iterator operator++(int) { 55 all_lookups_iterator tmp(*this); 56 ++(*this); 57 return tmp; 58 } 59 60 friend bool operator==(all_lookups_iterator x, all_lookups_iterator y) { 61 return x.It == y.It; 62 } 63 friend bool operator!=(all_lookups_iterator x, all_lookups_iterator y) { 64 return x.It != y.It; 65 } 66 }; 67 68 DeclContext::all_lookups_iterator DeclContext::lookups_begin() const { 69 DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext(); 70 if (hasExternalVisibleStorage()) 71 getParentASTContext().getExternalSource()->completeVisibleDeclsMap(Primary); 72 if (StoredDeclsMap *Map = Primary->buildLookup()) 73 return all_lookups_iterator(Map->begin(), Map->end()); 74 return all_lookups_iterator(); 75 } 76 77 DeclContext::all_lookups_iterator DeclContext::lookups_end() const { 78 DeclContext *Primary = const_cast<DeclContext*>(this)->getPrimaryContext(); 79 if (hasExternalVisibleStorage()) 80 getParentASTContext().getExternalSource()->completeVisibleDeclsMap(Primary); 81 if (StoredDeclsMap *Map = Primary->buildLookup()) 82 return all_lookups_iterator(Map->end(), Map->end()); 83 return all_lookups_iterator(); 84 } 85 86 } // end namespace clang 87 88 #endif 89