StringSet.h revision 245431
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 19namespace llvm { 20 21 /// StringSet - A wrapper for StringMap that provides set-like 22 /// functionality. Only insert() and count() methods are used by my 23 /// code. 24 template <class AllocatorTy = llvm::MallocAllocator> 25 class StringSet : public llvm::StringMap<char, AllocatorTy> { 26 typedef llvm::StringMap<char, AllocatorTy> base; 27 public: 28 bool insert(StringRef InLang) { 29 assert(!InLang.empty()); 30 const char *KeyStart = InLang.data(); 31 const char *KeyEnd = KeyStart + InLang.size(); 32 llvm::StringMapEntry<char> *Entry = llvm::StringMapEntry<char>:: 33 Create(KeyStart, KeyEnd, base::getAllocator(), '+'); 34 if (!base::insert(Entry)) { 35 Entry->Destroy(base::getAllocator()); 36 return false; 37 } 38 return true; 39 } 40 }; 41} 42 43#endif // LLVM_ADT_STRINGSET_H 44