lib/DebugInfo/CodeView/TypeTableCollection.cpp

   1 //===- TypeTableCollection.cpp -------------------------------- *- 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 #include "llvm/DebugInfo/CodeView/TypeTableCollection.h"
  10
  11 #include "llvm/DebugInfo/CodeView/CVTypeVisitor.h"
  12 #include "llvm/DebugInfo/CodeView/RecordName.h"
  13 #include "llvm/Support/BinaryStreamReader.h"
  14
  15 using namespace llvm;
  16 using namespace llvm::codeview;
  17
  18 TypeTableCollection::TypeTableCollection(ArrayRef<ArrayRef<uint8_t>> Records)
  19     : NameStorage(Allocator), Records(Records) {
  20   Names.resize(Records.size());
  21 }
  22
  23 Optional<TypeIndex> TypeTableCollection::getFirst() {
  24   if (empty())
  25     return None;
  26   return TypeIndex::fromArrayIndex(0);
  27 }
  28
  29 Optional<TypeIndex> TypeTableCollection::getNext(TypeIndex Prev) {
  30   assert(contains(Prev));
  31   ++Prev;
  32   if (Prev.toArrayIndex() == size())
  33     return None;
  34   return Prev;
  35 }
  36
  37 CVType TypeTableCollection::getType(TypeIndex Index) {
  38   assert(Index.toArrayIndex() < Records.size());
  39   return CVType(Records[Index.toArrayIndex()]);
  40 }
  41
  42 StringRef TypeTableCollection::getTypeName(TypeIndex Index) {
  43   if (Index.isNoneType() || Index.isSimple())
  44     return TypeIndex::simpleTypeName(Index);
  45
  46   uint32_t I = Index.toArrayIndex();
  47   if (Names[I].data() == nullptr) {
  48     StringRef Result = NameStorage.save(computeTypeName(*this, Index));
  49     Names[I] = Result;
  50   }
  51   return Names[I];
  52 }
  53
  54 bool TypeTableCollection::contains(TypeIndex Index) {
  55   return Index.toArrayIndex() <= size();
  56 }
  57
  58 uint32_t TypeTableCollection::size() { return Records.size(); }
  59
  60 uint32_t TypeTableCollection::capacity() { return Records.size(); }