llvm-6502/lib/CodeGen/AsmPrinter/DwarfStringPool.cpp
David Blaikie 1d4f28c6bc Remove StringMap::GetOrCreateValue in favor of StringMap::insert
Having two ways to do this doesn't seem terribly helpful and
consistently using the insert version (which we already has) seems like
it'll make the code easier to understand to anyone working with standard
data structures. (I also updated many references to the Entry's
key and value to use first() and second instead of getKey{Data,Length,}
and get/setValue - for similar consistency)

Also removes the GetOrCreateValue functions so there's less surface area
to StringMap to fix/improve/change/accommodate move semantics, etc.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@222319 91177308-0d34-0410-b5e6-96231b3b80d8
2014-11-19 05:49:42 +00:00

71 lines
2.2 KiB
C++

//===-- llvm/CodeGen/DwarfStringPool.cpp - Dwarf Debug Framework ----------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "DwarfStringPool.h"
#include "llvm/MC/MCStreamer.h"
using namespace llvm;
static std::pair<MCSymbol *, unsigned> &
getEntry(AsmPrinter &Asm,
StringMap<std::pair<MCSymbol *, unsigned>, BumpPtrAllocator &> &Pool,
StringRef Prefix, StringRef Str) {
std::pair<MCSymbol *, unsigned> &Entry = Pool[Str];
if (!Entry.first) {
Entry.second = Pool.size() - 1;
Entry.first = Asm.GetTempSymbol(Prefix, Entry.second);
}
return Entry;
}
MCSymbol *DwarfStringPool::getSymbol(AsmPrinter &Asm, StringRef Str) {
return getEntry(Asm, Pool, Prefix, Str).first;
}
unsigned DwarfStringPool::getIndex(AsmPrinter &Asm, StringRef Str) {
return getEntry(Asm, Pool, Prefix, Str).second;
}
void DwarfStringPool::emit(AsmPrinter &Asm, const MCSection *StrSection,
const MCSection *OffsetSection) {
if (Pool.empty())
return;
// Start the dwarf str section.
Asm.OutStreamer.SwitchSection(StrSection);
// Get all of the string pool entries and put them in an array by their ID so
// we can sort them.
SmallVector<const StringMapEntry<std::pair<MCSymbol *, unsigned>> *, 64>
Entries(Pool.size());
for (const auto &E : Pool)
Entries[E.getValue().second] = &E;
for (const auto &Entry : Entries) {
// Emit a label for reference from debug information entries.
Asm.OutStreamer.EmitLabel(Entry->getValue().first);
// Emit the string itself with a terminating null byte.
Asm.OutStreamer.EmitBytes(
StringRef(Entry->getKeyData(), Entry->getKeyLength() + 1));
}
// If we've got an offset section go ahead and emit that now as well.
if (OffsetSection) {
Asm.OutStreamer.SwitchSection(OffsetSection);
unsigned offset = 0;
unsigned size = 4; // FIXME: DWARF64 is 8.
for (const auto &Entry : Entries) {
Asm.OutStreamer.EmitIntValue(offset, size);
offset += Entry->getKeyLength() + 1;
}
}
}