1 //===-- UnresolvedSet.h - Unresolved sets of declarations ------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines the UnresolvedSet class, which is used to store
11 // collections of declarations in the AST.
13 //===----------------------------------------------------------------------===//
15 #ifndef LLVM_CLANG_AST_UNRESOLVEDSET_H
16 #define LLVM_CLANG_AST_UNRESOLVEDSET_H
18 #include "clang/AST/DeclAccessPair.h"
19 #include "clang/Basic/LLVM.h"
20 #include "llvm/ADT/ArrayRef.h"
21 #include "llvm/ADT/SmallVector.h"
22 #include "llvm/ADT/iterator.h"
26 /// The iterator over UnresolvedSets. Serves as both the const and
27 /// non-const iterator.
28 class UnresolvedSetIterator : public llvm::iterator_adaptor_base<
29 UnresolvedSetIterator, DeclAccessPair *,
30 std::random_access_iterator_tag, NamedDecl *,
31 std::ptrdiff_t, NamedDecl *, NamedDecl *> {
32 friend class UnresolvedSetImpl;
33 friend class ASTUnresolvedSet;
34 friend class OverloadExpr;
36 explicit UnresolvedSetIterator(DeclAccessPair *Iter)
37 : iterator_adaptor_base(Iter) {}
38 explicit UnresolvedSetIterator(const DeclAccessPair *Iter)
39 : iterator_adaptor_base(const_cast<DeclAccessPair *>(Iter)) {}
42 UnresolvedSetIterator() {}
44 NamedDecl *getDecl() const { return I->getDecl(); }
45 void setDecl(NamedDecl *ND) const { return I->setDecl(ND); }
46 AccessSpecifier getAccess() const { return I->getAccess(); }
47 void setAccess(AccessSpecifier AS) { I->setAccess(AS); }
48 const DeclAccessPair &getPair() const { return *I; }
50 NamedDecl *operator*() const { return getDecl(); }
51 NamedDecl *operator->() const { return **this; }
54 /// \brief A set of unresolved declarations.
55 class UnresolvedSetImpl {
56 typedef SmallVectorImpl<DeclAccessPair> DeclsTy;
58 // Don't allow direct construction, and only permit subclassing by
61 template <unsigned N> friend class UnresolvedSet;
62 UnresolvedSetImpl() = default;
63 UnresolvedSetImpl(const UnresolvedSetImpl &) = default;
64 UnresolvedSetImpl &operator=(const UnresolvedSetImpl &) = default;
66 // FIXME: Switch these to "= default" once MSVC supports generating move ops
67 UnresolvedSetImpl(UnresolvedSetImpl &&) {}
68 UnresolvedSetImpl &operator=(UnresolvedSetImpl &&) { return *this; }
71 // We don't currently support assignment through this iterator, so we might
72 // as well use the same implementation twice.
73 typedef UnresolvedSetIterator iterator;
74 typedef UnresolvedSetIterator const_iterator;
76 iterator begin() { return iterator(decls().begin()); }
77 iterator end() { return iterator(decls().end()); }
79 const_iterator begin() const { return const_iterator(decls().begin()); }
80 const_iterator end() const { return const_iterator(decls().end()); }
82 void addDecl(NamedDecl *D) {
86 void addDecl(NamedDecl *D, AccessSpecifier AS) {
87 decls().push_back(DeclAccessPair::make(D, AS));
90 /// Replaces the given declaration with the new one, once.
92 /// \return true if the set changed
93 bool replace(const NamedDecl* Old, NamedDecl *New) {
94 for (DeclsTy::iterator I = decls().begin(), E = decls().end(); I != E; ++I)
95 if (I->getDecl() == Old)
96 return (I->setDecl(New), true);
100 /// Replaces the declaration at the given iterator with the new one,
101 /// preserving the original access bits.
102 void replace(iterator I, NamedDecl *New) { I.I->setDecl(New); }
104 void replace(iterator I, NamedDecl *New, AccessSpecifier AS) {
108 void erase(unsigned I) { decls()[I] = decls().pop_back_val(); }
110 void erase(iterator I) { *I.I = decls().pop_back_val(); }
112 void setAccess(iterator I, AccessSpecifier AS) { I.I->setAccess(AS); }
114 void clear() { decls().clear(); }
115 void set_size(unsigned N) { decls().set_size(N); }
117 bool empty() const { return decls().empty(); }
118 unsigned size() const { return decls().size(); }
120 void append(iterator I, iterator E) { decls().append(I.I, E.I); }
122 DeclAccessPair &operator[](unsigned I) { return decls()[I]; }
123 const DeclAccessPair &operator[](unsigned I) const { return decls()[I]; }
126 // These work because the only permitted subclass is UnresolvedSetImpl
129 return *reinterpret_cast<DeclsTy*>(this);
131 const DeclsTy &decls() const {
132 return *reinterpret_cast<const DeclsTy*>(this);
136 /// \brief A set of unresolved declarations.
137 template <unsigned InlineCapacity> class UnresolvedSet :
138 public UnresolvedSetImpl {
139 SmallVector<DeclAccessPair, InlineCapacity> Decls;