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