1//===- UnresolvedSet.h - Unresolved sets of declarations --------*- 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 the UnresolvedSet class, which is used to store
11// collections of declarations in the AST.
12//
13//===----------------------------------------------------------------------===//
14
15#ifndef LLVM_CLANG_AST_UNRESOLVEDSET_H
16#define LLVM_CLANG_AST_UNRESOLVEDSET_H
17
18#include "clang/AST/DeclAccessPair.h"
19#include "clang/Basic/LLVM.h"
20#include "clang/Basic/Specifiers.h"
21#include "llvm/ADT/SmallVector.h"
22#include "llvm/ADT/iterator.h"
23#include <cstddef>
24#include <iterator>
25
26namespace clang {
27
28class NamedDecl;
29
30/// The iterator over UnresolvedSets. Serves as both the const and
31/// non-const iterator.
32class UnresolvedSetIterator : public llvm::iterator_adaptor_base<
33 UnresolvedSetIterator, DeclAccessPair *,
34 std::random_access_iterator_tag, NamedDecl *,
35 std::ptrdiff_t, NamedDecl *, NamedDecl *> {
36 friend class ASTUnresolvedSet;
37 friend class OverloadExpr;
38 friend class UnresolvedSetImpl;
39
40 explicit UnresolvedSetIterator(DeclAccessPair *Iter)
41 : iterator_adaptor_base(Iter) {}
42 explicit UnresolvedSetIterator(const DeclAccessPair *Iter)
43 : iterator_adaptor_base(const_cast<DeclAccessPair *>(Iter)) {}
44
45public:
46 // Work around a bug in MSVC 2013 where explicitly default constructed
47 // temporaries with defaulted ctors are not zero initialized.
48 UnresolvedSetIterator() : iterator_adaptor_base(nullptr) {}
49
50 NamedDecl *getDecl() const { return I->getDecl(); }
51 void setDecl(NamedDecl *ND) const { return I->setDecl(ND); }
52 AccessSpecifier getAccess() const { return I->getAccess(); }
53 void setAccess(AccessSpecifier AS) { I->setAccess(AS); }
54 const DeclAccessPair &getPair() const { return *I; }
55
56 NamedDecl *operator*() const { return getDecl(); }
57 NamedDecl *operator->() const { return **this; }
58};
59
60/// A set of unresolved declarations.
61class UnresolvedSetImpl {
62 using DeclsTy = SmallVectorImpl<DeclAccessPair>;
63
64 // Don't allow direct construction, and only permit subclassing by
65 // UnresolvedSet.
66private:
67 template <unsigned N> friend class UnresolvedSet;
68
69 UnresolvedSetImpl() = default;
70 UnresolvedSetImpl(const UnresolvedSetImpl &) = default;
71 UnresolvedSetImpl &operator=(const UnresolvedSetImpl &) = default;
72
73 // FIXME: Switch these to "= default" once MSVC supports generating move ops
74 UnresolvedSetImpl(UnresolvedSetImpl &&) {}
75 UnresolvedSetImpl &operator=(UnresolvedSetImpl &&) { return *this; }
76
77public:
78 // We don't currently support assignment through this iterator, so we might
79 // as well use the same implementation twice.
80 using iterator = UnresolvedSetIterator;
81 using const_iterator = UnresolvedSetIterator;
82
83 iterator begin() { return iterator(decls().begin()); }
84 iterator end() { return iterator(decls().end()); }
85
86 const_iterator begin() const { return const_iterator(decls().begin()); }
87 const_iterator end() const { return const_iterator(decls().end()); }
88
89 void addDecl(NamedDecl *D) {
90 addDecl(D, AS_none);
91 }
92
93 void addDecl(NamedDecl *D, AccessSpecifier AS) {
94 decls().push_back(DeclAccessPair::make(D, AS));
95 }
96
97 /// Replaces the given declaration with the new one, once.
98 ///
99 /// \return true if the set changed
100 bool replace(const NamedDecl* Old, NamedDecl *New) {
101 for (DeclsTy::iterator I = decls().begin(), E = decls().end(); I != E; ++I)
102 if (I->getDecl() == Old)
103 return (I->setDecl(New), true);
104 return false;
105 }
106
107 /// Replaces the declaration at the given iterator with the new one,
108 /// preserving the original access bits.
109 void replace(iterator I, NamedDecl *New) { I.I->setDecl(New); }
110
111 void replace(iterator I, NamedDecl *New, AccessSpecifier AS) {
112 I.I->set(New, AS);
113 }
114
115 void erase(unsigned I) { decls()[I] = decls().pop_back_val(); }
116
117 void erase(iterator I) { *I.I = decls().pop_back_val(); }
118
119 void setAccess(iterator I, AccessSpecifier AS) { I.I->setAccess(AS); }
120
121 void clear() { decls().clear(); }
122 void set_size(unsigned N) { decls().set_size(N); }
123
124 bool empty() const { return decls().empty(); }
125 unsigned size() const { return decls().size(); }
126
127 void append(iterator I, iterator E) { decls().append(I.I, E.I); }
128
129 DeclAccessPair &operator[](unsigned I) { return decls()[I]; }
130 const DeclAccessPair &operator[](unsigned I) const { return decls()[I]; }
131
132private:
133 // These work because the only permitted subclass is UnresolvedSetImpl
134
135 DeclsTy &decls() {
136 return *reinterpret_cast<DeclsTy*>(this);
137 }
138 const DeclsTy &decls() const {
139 return *reinterpret_cast<const DeclsTy*>(this);
140 }
141};
142
143/// A set of unresolved declarations.
144template <unsigned InlineCapacity> class UnresolvedSet :
145 public UnresolvedSetImpl {
146 SmallVector<DeclAccessPair, InlineCapacity> Decls;
147};
148
149
150} // namespace clang
151
152#endif // LLVM_CLANG_AST_UNRESOLVEDSET_H
153