Home | History | Annotate | Download | only in AST
      1 //===-- ASTUnresolvedSet.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 provides an UnresolvedSet-like class, whose contents are
     11 //  allocated using the allocator associated with an ASTContext.
     12 //
     13 //===----------------------------------------------------------------------===//
     14 
     15 #ifndef LLVM_CLANG_AST_ASTUNRESOLVEDSET_H
     16 #define LLVM_CLANG_AST_ASTUNRESOLVEDSET_H
     17 
     18 #include "clang/AST/ASTVector.h"
     19 #include "clang/AST/UnresolvedSet.h"
     20 
     21 namespace clang {
     22 
     23 /// \brief An UnresolvedSet-like class which uses the ASTContext's allocator.
     24 class ASTUnresolvedSet {
     25   typedef ASTVector<DeclAccessPair> DeclsTy;
     26   DeclsTy Decls;
     27 
     28   ASTUnresolvedSet(const ASTUnresolvedSet &) LLVM_DELETED_FUNCTION;
     29   void operator=(const ASTUnresolvedSet &) LLVM_DELETED_FUNCTION;
     30 
     31 public:
     32   ASTUnresolvedSet() {}
     33   ASTUnresolvedSet(ASTContext &C, unsigned N) : Decls(C, N) {}
     34 
     35   typedef UnresolvedSetIterator iterator;
     36   typedef UnresolvedSetIterator const_iterator;
     37 
     38   iterator begin() { return iterator(Decls.begin()); }
     39   iterator end() { return iterator(Decls.end()); }
     40 
     41   const_iterator begin() const { return const_iterator(Decls.begin()); }
     42   const_iterator end() const { return const_iterator(Decls.end()); }
     43 
     44   void addDecl(ASTContext &C, NamedDecl *D, AccessSpecifier AS) {
     45     Decls.push_back(DeclAccessPair::make(D, AS), C);
     46   }
     47 
     48   /// Replaces the given declaration with the new one, once.
     49   ///
     50   /// \return true if the set changed
     51   bool replace(const NamedDecl* Old, NamedDecl *New, AccessSpecifier AS) {
     52     for (DeclsTy::iterator I = Decls.begin(), E = Decls.end(); I != E; ++I) {
     53       if (I->getDecl() == Old) {
     54         I->set(New, AS);
     55         return true;
     56       }
     57     }
     58     return false;
     59   }
     60 
     61   void erase(unsigned I) {
     62     Decls[I] = Decls.back();
     63     Decls.pop_back();
     64   }
     65 
     66   void clear() { Decls.clear(); }
     67 
     68   bool empty() const { return Decls.empty(); }
     69   unsigned size() const { return Decls.size(); }
     70 
     71   void reserve(ASTContext &C, unsigned N) {
     72     Decls.reserve(C, N);
     73   }
     74 
     75   void append(ASTContext &C, iterator I, iterator E) {
     76     Decls.append(C, I.ir, E.ir);
     77   }
     78 
     79   DeclAccessPair &operator[](unsigned I) { return Decls[I]; }
     80   const DeclAccessPair &operator[](unsigned I) const { return Decls[I]; }
     81 };
     82 
     83 } // namespace clang
     84 
     85 #endif
     86