lld/COFF/TypeMerger.h

   1 //===- TypeMerger.h ---------------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLD_COFF_TYPEMERGER_H
  10 #define LLD_COFF_TYPEMERGER_H
  11
  12 #include "Config.h"
  13 #include "llvm/DebugInfo/CodeView/MergingTypeTableBuilder.h"
  14 #include "llvm/DebugInfo/CodeView/TypeHashing.h"
  15 #include "llvm/Support/Allocator.h"
  16 #include <atomic>
  17
  18 namespace lld {
  19 namespace coff {
  20
  21 using llvm::codeview::GloballyHashedType;
  22 using llvm::codeview::TypeIndex;
  23
  24 struct GHashState;
  25
  26 class TypeMerger {
  27 public:
  28   TypeMerger(llvm::BumpPtrAllocator &alloc);
  29
  30   ~TypeMerger();
  31
  32   /// Get the type table or the global type table if /DEBUG:GHASH is enabled.
  33   inline llvm::codeview::TypeCollection &getTypeTable() {
  34     assert(!config->debugGHashes);
  35     return typeTable;
  36   }
  37
  38   /// Get the ID table or the global ID table if /DEBUG:GHASH is enabled.
  39   inline llvm::codeview::TypeCollection &getIDTable() {
  40     assert(!config->debugGHashes);
  41     return idTable;
  42   }
  43
  44   /// Use global hashes to eliminate duplicate types and identify unique type
  45   /// indices in each TpiSource.
  46   void mergeTypesWithGHash();
  47
  48   /// Map from PDB function id type indexes to PDB function type indexes.
  49   /// Populated after mergeTypesWithGHash.
  50   llvm::DenseMap<TypeIndex, TypeIndex> funcIdToType;
  51
  52   /// Type records that will go into the PDB TPI stream.
  53   llvm::codeview::MergingTypeTableBuilder typeTable;
  54
  55   /// Item records that will go into the PDB IPI stream.
  56   llvm::codeview::MergingTypeTableBuilder idTable;
  57
  58   // When showSummary is enabled, these are histograms of TPI and IPI records
  59   // keyed by type index.
  60   SmallVector<uint32_t, 0> tpiCounts;
  61   SmallVector<uint32_t, 0> ipiCounts;
  62 };
  63
  64 } // namespace coff
  65 } // namespace lld
  66
  67 #endif