Home | History | Annotate | Line # | Download | only in ADT 
      1 //===- StringSet.h - An efficient set built on StringMap --------*- C++ -*-===//
      2 //
      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
      6 //
      7 //===----------------------------------------------------------------------===//
      8 //
      9 //  StringSet - A set-like wrapper for the StringMap.
     10 //
     11 //===----------------------------------------------------------------------===//
     12 
     13 #ifndef LLVM_ADT_STRINGSET_H
     14 #define LLVM_ADT_STRINGSET_H
     15 
     16 #include "llvm/ADT/StringMap.h"
     17 
     18 namespace llvm {
     19 
     20 /// StringSet - A wrapper for StringMap that provides set-like functionality.
     21 template <class AllocatorTy = MallocAllocator>
     22 class StringSet : public StringMap<NoneType, AllocatorTy> {
     23   using Base = StringMap<NoneType, AllocatorTy>;
     24 
     25 public:
     26   StringSet() = default;
     27   StringSet(std::initializer_list<StringRef> initializer) {
     28     for (StringRef str : initializer)
     29       insert(str);
     30   }
     31   explicit StringSet(AllocatorTy a) : Base(a) {}
     32 
     33   std::pair<typename Base::iterator, bool> insert(StringRef key) {
     34     return Base::try_emplace(key);
     35   }
     36 
     37   template <typename InputIt>
     38   void insert(const InputIt &begin, const InputIt &end) {
     39     for (auto it = begin; it != end; ++it)
     40       insert(*it);
     41   }
     42 
     43   template <typename ValueTy>
     44   std::pair<typename Base::iterator, bool>
     45   insert(const StringMapEntry<ValueTy> &mapEntry) {
     46     return insert(mapEntry.getKey());
     47   }
     48 
     49   /// Check if the set contains the given \c key.
     50   bool contains(StringRef key) const { return Base::FindKey(key) != -1; }
     51 };
     52 
     53 } // end namespace llvm
     54 
     55 #endif // LLVM_ADT_STRINGSET_H
     56