• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 
19 namespace llvm {
20 
21   /// StringSet - A wrapper for StringMap that provides set-like functionality.
22   template <class AllocatorTy = llvm::MallocAllocator>
23   class StringSet : public llvm::StringMap<char, AllocatorTy> {
24     typedef llvm::StringMap<char, AllocatorTy> base;
25   public:
26     StringSet() = default;
StringSet(std::initializer_list<StringRef> S)27     StringSet(std::initializer_list<StringRef> S) {
28       for (StringRef X : S)
29         insert(X);
30     }
31 
insert(StringRef Key)32     std::pair<typename base::iterator, bool> insert(StringRef Key) {
33       assert(!Key.empty());
34       return base::insert(std::make_pair(Key, '\0'));
35     }
36   };
37 }
38 
39 #endif // LLVM_ADT_STRINGSET_H
40