contrib/llvm/include/llvm/DebugInfo/PDB/Native/NamedStreamMap.h

   1 //===- NamedStreamMap.h - PDB Named Stream Map ------------------*- 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_DEBUGINFO_PDB_NATIVE_NAMEDSTREAMMAP_H
  11 #define LLVM_DEBUGINFO_PDB_NATIVE_NAMEDSTREAMMAP_H
  12
  13 #include "llvm/ADT/Optional.h"
  14 #include "llvm/ADT/StringMap.h"
  15 #include "llvm/ADT/StringRef.h"
  16 #include "llvm/ADT/iterator_range.h"
  17 #include "llvm/DebugInfo/PDB/Native/HashTable.h"
  18 #include "llvm/Support/Error.h"
  19 #include <cstdint>
  20
  21 namespace llvm {
  22
  23 class BinaryStreamReader;
  24 class BinaryStreamWriter;
  25
  26 namespace pdb {
  27
  28 class NamedStreamMap {
  29   friend class NamedStreamMapBuilder;
  30
  31   struct FinalizationInfo {
  32     uint32_t StringDataBytes = 0;
  33     uint32_t SerializedLength = 0;
  34   };
  35
  36 public:
  37   NamedStreamMap();
  38
  39   Error load(BinaryStreamReader &Stream);
  40   Error commit(BinaryStreamWriter &Writer) const;
  41   uint32_t finalize();
  42
  43   uint32_t size() const;
  44   bool get(StringRef Stream, uint32_t &StreamNo) const;
  45   void set(StringRef Stream, uint32_t StreamNo);
  46   void remove(StringRef Stream);
  47   const StringMap<uint32_t> &getStringMap() const { return Mapping; }
  48   iterator_range<StringMapConstIterator<uint32_t>> entries() const;
  49
  50 private:
  51   Optional<FinalizationInfo> FinalizedInfo;
  52   HashTable FinalizedHashTable;
  53   StringMap<uint32_t> Mapping;
  54 };
  55
  56 } // end namespace pdb
  57
  58 } // end namespace llvm
  59
  60 #endif // LLVM_DEBUGINFO_PDB_NATIVE_NAMEDSTREAMMAP_H