1 //===- ASTUnresolvedSet.h - Unresolved sets of declarations -----*- C++ -*-===//
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 //===----------------------------------------------------------------------===//
9 // This file provides an UnresolvedSet-like class, whose contents are
10 // allocated using the allocator associated with an ASTContext.
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_CLANG_AST_ASTUNRESOLVEDSET_H
15 #define LLVM_CLANG_AST_ASTUNRESOLVEDSET_H
17 #include "clang/AST/ASTVector.h"
18 #include "clang/AST/DeclAccessPair.h"
19 #include "clang/AST/UnresolvedSet.h"
20 #include "clang/Basic/Specifiers.h"
28 /// An UnresolvedSet-like class which uses the ASTContext's allocator.
29 class ASTUnresolvedSet {
30 friend class LazyASTUnresolvedSet;
32 struct DeclsTy : ASTVector<DeclAccessPair> {
34 DeclsTy(ASTContext &C, unsigned N) : ASTVector<DeclAccessPair>(C, N) {}
36 bool isLazy() const { return getTag(); }
37 void setLazy(bool Lazy) { setTag(Lazy); }
43 ASTUnresolvedSet() = default;
44 ASTUnresolvedSet(ASTContext &C, unsigned N) : Decls(C, N) {}
46 using iterator = UnresolvedSetIterator;
47 using const_iterator = UnresolvedSetIterator;
49 iterator begin() { return iterator(Decls.begin()); }
50 iterator end() { return iterator(Decls.end()); }
52 const_iterator begin() const { return const_iterator(Decls.begin()); }
53 const_iterator end() const { return const_iterator(Decls.end()); }
55 void addDecl(ASTContext &C, NamedDecl *D, AccessSpecifier AS) {
56 Decls.push_back(DeclAccessPair::make(D, AS), C);
59 /// Replaces the given declaration with the new one, once.
61 /// \return true if the set changed
62 bool replace(const NamedDecl *Old, NamedDecl *New, AccessSpecifier AS) {
63 for (DeclsTy::iterator I = Decls.begin(), E = Decls.end(); I != E; ++I) {
64 if (I->getDecl() == Old) {
72 void erase(unsigned I) { Decls[I] = Decls.pop_back_val(); }
74 void clear() { Decls.clear(); }
76 bool empty() const { return Decls.empty(); }
77 unsigned size() const { return Decls.size(); }
79 void reserve(ASTContext &C, unsigned N) {
83 void append(ASTContext &C, iterator I, iterator E) {
84 Decls.append(C, I.I, E.I);
87 DeclAccessPair &operator[](unsigned I) { return Decls[I]; }
88 const DeclAccessPair &operator[](unsigned I) const { return Decls[I]; }
91 /// An UnresolvedSet-like class that might not have been loaded from the
92 /// external AST source yet.
93 class LazyASTUnresolvedSet {
94 mutable ASTUnresolvedSet Impl;
96 void getFromExternalSource(ASTContext &C) const;
99 ASTUnresolvedSet &get(ASTContext &C) const {
100 if (Impl.Decls.isLazy())
101 getFromExternalSource(C);
105 void reserve(ASTContext &C, unsigned N) { Impl.reserve(C, N); }
107 void addLazyDecl(ASTContext &C, uintptr_t ID, AccessSpecifier AS) {
108 assert(Impl.empty() || Impl.Decls.isLazy());
109 Impl.Decls.setLazy(true);
110 Impl.addDecl(C, reinterpret_cast<NamedDecl *>(ID << 2), AS);
116 #endif // LLVM_CLANG_AST_ASTUNRESOLVEDSET_H