diff options
Diffstat (limited to 'clang-r353983e/include/llvm/ADT/StringSet.h')
| -rw-r--r-- | clang-r353983e/include/llvm/ADT/StringSet.h | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/clang-r353983e/include/llvm/ADT/StringSet.h b/clang-r353983e/include/llvm/ADT/StringSet.h new file mode 100644 index 00000000..fcf9519b --- /dev/null +++ b/clang-r353983e/include/llvm/ADT/StringSet.h @@ -0,0 +1,51 @@ +//===- StringSet.h - The LLVM Compiler Driver -------------------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// StringSet - A set-like wrapper for the StringMap. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_ADT_STRINGSET_H +#define LLVM_ADT_STRINGSET_H + +#include "llvm/ADT/StringMap.h" +#include "llvm/ADT/StringRef.h" +#include "llvm/Support/Allocator.h" +#include <cassert> +#include <initializer_list> +#include <utility> + +namespace llvm { + + /// StringSet - A wrapper for StringMap that provides set-like functionality. + template <class AllocatorTy = MallocAllocator> + class StringSet : public StringMap<char, AllocatorTy> { + using base = StringMap<char, AllocatorTy>; + + public: + StringSet() = default; + StringSet(std::initializer_list<StringRef> S) { + for (StringRef X : S) + insert(X); + } + + std::pair<typename base::iterator, bool> insert(StringRef Key) { + assert(!Key.empty()); + return base::insert(std::make_pair(Key, '\0')); + } + + template <typename InputIt> + void insert(const InputIt &Begin, const InputIt &End) { + for (auto It = Begin; It != End; ++It) + base::insert(std::make_pair(*It, '\0')); + } + }; + +} // end namespace llvm + +#endif // LLVM_ADT_STRINGSET_H |
