1 //===-- StringTableBuilder.h - String table building utility ------*- C++ -*-=//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_MC_STRINGTABLEBUILDER_H
11 #define LLVM_MC_STRINGTABLEBUILDER_H
13 #include "llvm/ADT/SmallString.h"
14 #include "llvm/ADT/StringMap.h"
19 /// \brief Utility for building string tables with deduplicated suffixes.
20 class StringTableBuilder {
21 SmallString<256> StringTable;
22 StringMap<size_t> StringIndexMap;
25 /// \brief Add a string to the builder. Returns a StringRef to the internal
26 /// copy of s. Can only be used before the table is finalized.
27 StringRef add(StringRef s) {
28 assert(!isFinalized());
29 return StringIndexMap.insert(std::make_pair(s, 0)).first->first();
38 /// \brief Analyze the strings and build the final table. No more strings can
39 /// be added after this point.
40 void finalize(Kind kind);
42 /// \brief Retrieve the string table data. Can only be used after the table
45 assert(isFinalized());
49 /// \brief Get the offest of a string in the string table. Can only be used
50 /// after the table is finalized.
51 size_t getOffset(StringRef s) {
52 assert(isFinalized());
53 assert(StringIndexMap.count(s) && "String is not in table!");
54 return StringIndexMap[s];
61 return !StringTable.empty();
65 } // end llvm namespace