1//===- llvm/CodeGen/DwarfStringPool.h - Dwarf Debug Framework ---*- 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 LLVM_LIB_CODEGEN_ASMPRINTER_DWARFSTRINGPOOL_H 10#define LLVM_LIB_CODEGEN_ASMPRINTER_DWARFSTRINGPOOL_H 11 12#include "llvm/ADT/StringMap.h" 13#include "llvm/ADT/StringRef.h" 14#include "llvm/CodeGen/DwarfStringPoolEntry.h" 15#include "llvm/Support/Allocator.h" 16 17namespace llvm { 18 19class AsmPrinter; 20class MCSection; 21class MCSymbol; 22 23// Collection of strings for this unit and assorted symbols. 24// A String->Symbol mapping of strings used by indirect 25// references. 26class DwarfStringPool { 27 using EntryTy = DwarfStringPoolEntry; 28 29 StringMap<EntryTy, BumpPtrAllocator &> Pool; 30 StringRef Prefix; 31 unsigned NumBytes = 0; 32 unsigned NumIndexedStrings = 0; 33 bool ShouldCreateSymbols; 34 35 StringMapEntry<EntryTy> &getEntryImpl(AsmPrinter &Asm, StringRef Str); 36 37public: 38 using EntryRef = DwarfStringPoolEntryRef; 39 40 DwarfStringPool(BumpPtrAllocator &A, AsmPrinter &Asm, StringRef Prefix); 41 42 void emitStringOffsetsTableHeader(AsmPrinter &Asm, MCSection *OffsetSection, 43 MCSymbol *StartSym); 44 45 void emit(AsmPrinter &Asm, MCSection *StrSection, 46 MCSection *OffsetSection = nullptr, 47 bool UseRelativeOffsets = false); 48 49 bool empty() const { return Pool.empty(); } 50 51 unsigned size() const { return Pool.size(); } 52 53 unsigned getNumIndexedStrings() const { return NumIndexedStrings; } 54 55 /// Get a reference to an entry in the string pool. 56 EntryRef getEntry(AsmPrinter &Asm, StringRef Str); 57 58 /// Same as getEntry, except that you can use EntryRef::getIndex to obtain a 59 /// unique ID of this entry (e.g., for use in indexed forms like 60 /// DW_FORM_strx). 61 EntryRef getIndexedEntry(AsmPrinter &Asm, StringRef Str); 62}; 63 64} // end namespace llvm 65 66#endif // LLVM_LIB_CODEGEN_ASMPRINTER_DWARFSTRINGPOOL_H 67