Home | History | Annotate | Download | only in AsmPrinter
      1 //===-- llvm/CodeGen/DwarfStringPool.h - Dwarf Debug Framework -*- 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 CODEGEN_ASMPRINTER_STRINGPOOL_H__
     11 #define CODEGEN_ASMPRINTER_STRINGPOOL_H__
     12 
     13 #include "llvm/ADT/StringMap.h"
     14 #include "llvm/CodeGen/AsmPrinter.h"
     15 #include "llvm/Support/Allocator.h"
     16 
     17 #include <utility>
     18 
     19 namespace llvm {
     20 
     21 class MCSymbol;
     22 class MCSection;
     23 class StringRef;
     24 
     25 // Collection of strings for this unit and assorted symbols.
     26 // A String->Symbol mapping of strings used by indirect
     27 // references.
     28 class DwarfStringPool {
     29   StringMap<std::pair<MCSymbol *, unsigned>, BumpPtrAllocator &> Pool;
     30   StringRef Prefix;
     31   MCSymbol *SectionSymbol;
     32 
     33 public:
     34   DwarfStringPool(BumpPtrAllocator &A, AsmPrinter &Asm, StringRef Prefix)
     35       : Pool(A), Prefix(Prefix), SectionSymbol(Asm.GetTempSymbol(Prefix)) {}
     36 
     37   void emit(AsmPrinter &Asm, const MCSection *StrSection,
     38             const MCSection *OffsetSection = nullptr,
     39             const MCSymbol *StrSecSym = nullptr);
     40 
     41   /// \brief Returns the entry into the start of the pool.
     42   MCSymbol *getSectionSymbol();
     43 
     44   /// \brief Returns an entry into the string pool with the given
     45   /// string text.
     46   MCSymbol *getSymbol(AsmPrinter &Asm, StringRef Str);
     47 
     48   /// \brief Returns the index into the string pool with the given
     49   /// string text.
     50   unsigned getIndex(AsmPrinter &Asm, StringRef Str);
     51 
     52   bool empty() const { return Pool.empty(); }
     53 };
     54 }
     55 #endif
     56