DwarfStringPoolEntry.h

Go to the documentation of this file.

//===- llvm/CodeGen/DwarfStringPoolEntry.h - String pool entry --*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_CODEGEN_DWARFSTRINGPOOLENTRY_H
#define LLVM_CODEGEN_DWARFSTRINGPOOLENTRY_H
 
#include "llvm/ADT/PointerUnion.h"
#include "llvm/ADT/StringMap.h"
 
namespace llvm {
 
class MCSymbol;
 
/// Data for a string pool entry.
struct DwarfStringPoolEntry {
  static constexpr unsigned NotIndexed = -1;
 
  MCSymbol *Symbol = nullptr;
  uint64_t Offset = 0;
  unsigned Index = 0;
 
  bool isIndexed() const { return Index != NotIndexed; }
};
 
/// DwarfStringPoolEntryRef: Dwarf string pool entry reference.
///
/// Dwarf string pool entry keeps string value and its data.
/// There are two variants how data are represented:
///
///   1. By value - StringMapEntry<DwarfStringPoolEntry>.
///   2. By pointer - StringMapEntry<DwarfStringPoolEntry *>.
///
/// The "By pointer" variant allows for reducing memory usage for the case
/// when string pool entry does not have data: it keeps the null pointer
/// and so no need to waste space for the full DwarfStringPoolEntry.
/// It is recommended to use "By pointer" variant if not all entries
/// of dwarf string pool have corresponding DwarfStringPoolEntry.
 
class DwarfStringPoolEntryRef {
  /// Pointer type for "By value" string entry.
  using ByValStringEntryPtr = const StringMapEntry<DwarfStringPoolEntry> *;
 
  /// Pointer type for "By pointer" string entry.
  using ByPtrStringEntryPtr = const StringMapEntry<DwarfStringPoolEntry *> *;
 
  /// Pointer to the dwarf string pool Entry.
  PointerUnion<ByValStringEntryPtr, ByPtrStringEntryPtr> MapEntry = nullptr;
 
public:
  DwarfStringPoolEntryRef() = default;
 
  /// ASSUMPTION: DwarfStringPoolEntryRef keeps pointer to \p Entry,
  /// thus specified entry mustn`t be reallocated.
  DwarfStringPoolEntryRef(const StringMapEntry<DwarfStringPoolEntry> &Entry)
      : MapEntry(&Entry) {}
 
  /// ASSUMPTION: DwarfStringPoolEntryRef keeps pointer to \p Entry,
  /// thus specified entry mustn`t be reallocated.
  DwarfStringPoolEntryRef(const StringMapEntry<DwarfStringPoolEntry *> &Entry)
      : MapEntry(&Entry) {
    assert(MapEntry.get<ByPtrStringEntryPtr>()->second != nullptr);
  }
 
  explicit operator bool() const { return !MapEntry.isNull(); }
 
  /// \returns symbol for the dwarf string.
  MCSymbol *getSymbol() const {
    assert(getEntry().Symbol && "No symbol available!");
    return getEntry().Symbol;
  }
 
  /// \returns offset for the dwarf string.
  uint64_t getOffset() const { return getEntry().Offset; }
 
  /// \returns index for the dwarf string.
  unsigned getIndex() const {
    assert(getEntry().isIndexed() && "Index is not set!");
    return getEntry().Index;
  }
 
  /// \returns string.
  StringRef getString() const {
    if (MapEntry.is<ByValStringEntryPtr>())
      return MapEntry.get<ByValStringEntryPtr>()->first();
 
    return MapEntry.get<ByPtrStringEntryPtr>()->first();
  }
 
  /// \returns the entire string pool entry for convenience.
  const DwarfStringPoolEntry &getEntry() const {
    if (MapEntry.is<ByValStringEntryPtr>())
      return MapEntry.get<ByValStringEntryPtr>()->second;
 
    return *MapEntry.get<ByPtrStringEntryPtr>()->second;
  }
 
  bool operator==(const DwarfStringPoolEntryRef &X) const {
    return MapEntry.getOpaqueValue() == X.MapEntry.getOpaqueValue();
  }
 
  bool operator!=(const DwarfStringPoolEntryRef &X) const {
    return MapEntry.getOpaqueValue() != X.MapEntry.getOpaqueValue();
  }
};
 
} // end namespace llvm
 
#endif