1 //===- StringSet.h - The LLVM Compiler Driver -------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open
6 // Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 //  StringSet - A set-like wrapper for the StringMap.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_ADT_STRINGSET_H
15 #define LLVM_ADT_STRINGSET_H
16 
17 #include "llvm/ADT/StringMap.h"
18 #include "llvm/ADT/StringRef.h"
19 #include "llvm/Support/Allocator.h"
20 #include <cassert>
21 #include <initializer_list>
22 #include <utility>
23 
24 namespace llvm {
25 
26   /// StringSet - A wrapper for StringMap that provides set-like functionality.
27   template <class AllocatorTy = MallocAllocator>
28   class StringSet : public StringMap<char, AllocatorTy> {
29     using base = StringMap<char, AllocatorTy>;
30 
31   public:
32     StringSet() = default;
StringSet(std::initializer_list<StringRef> S)33     StringSet(std::initializer_list<StringRef> S) {
34       for (StringRef X : S)
35         insert(X);
36     }
37 
insert(StringRef Key)38     std::pair<typename base::iterator, bool> insert(StringRef Key) {
39       assert(!Key.empty());
40       return base::insert(std::make_pair(Key, '\0'));
41     }
42 
43     template <typename InputIt>
insert(const InputIt & Begin,const InputIt & End)44     void insert(const InputIt &Begin, const InputIt &End) {
45       for (auto It = Begin; It != End; ++It)
46         base::insert(std::make_pair(*It, '\0'));
47     }
48   };
49 
50 } // end namespace llvm
51 
52 #endif // LLVM_ADT_STRINGSET_H
53