include/llvm/MC/StringTableBuilder.h

   1 //===-- StringTableBuilder.h - String table building utility ------*- C++ -*-=//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #ifndef LLVM_MC_STRINGTABLE_BUILDER_H
  11 #define LLVM_MC_STRINGTABLE_BUILDER_H
  12
  13 #include "llvm/ADT/SmallString.h"
  14 #include "llvm/ADT/StringMap.h"
  15 #include <cassert>
  16
  17 namespace llvm {
  18
  19 /// \brief Utility for building string tables with deduplicated suffixes.
  20 class StringTableBuilder {
  21   SmallString<256> StringTable;
  22   StringMap<size_t> StringIndexMap;
  23
  24 public:
  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.GetOrCreateValue(s, 0).getKey();
  30   }
  31
  32   /// \brief Analyze the strings and build the final table. No more strings can
  33   /// be added after this point.
  34   void finalize();
  35
  36   /// \brief Retrieve the string table data. Can only be used after the table
  37   /// is finalized.
  38   StringRef data() {
  39     assert(isFinalized());
  40     return StringTable;
  41   }
  42
  43   /// \brief Get the offest of a string in the string table. Can only be used
  44   /// after the table is finalized.
  45   size_t getOffset(StringRef s) {
  46     assert(isFinalized());
  47     assert(StringIndexMap.count(s) && "String is not in table!");
  48     return StringIndexMap[s];
  49   }
  50
  51 private:
  52   bool isFinalized() {
  53     return !StringTable.empty();
  54   }
  55 };
  56
  57 } // end llvm namespace
  58
  59 #endif