Do not hardcode metadata names.

llvm-svn: 83010
This commit is contained in:
Devang Patel 2009-09-28 20:56:00 +00:00
parent 3c46142f12
commit 9e2af6ed0b

View File

@ -1266,6 +1266,7 @@ class AssemblyWriter {
TypePrinting TypePrinter;
AssemblyAnnotationWriter *AnnotationWriter;
std::vector<const Type*> NumberedTypes;
DenseMap<unsigned, const char *> MDNames;
public:
inline AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
@ -1273,6 +1274,14 @@ public:
AssemblyAnnotationWriter *AAW)
: Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
AddModuleTypesToPrinter(TypePrinter, NumberedTypes, M);
// FIXME: Provide MDPrinter
Metadata &TheMetadata = M->getContext().getMetadata();
const StringMap<unsigned> *Names = TheMetadata.getHandlerNames();
for (StringMapConstIterator<unsigned> I = Names->begin(),
E = Names->end(); I != E; ++I) {
const StringMapEntry<unsigned> &Entry = *I;
MDNames[I->second] = Entry.getKeyData();
}
}
void write(const Module *M) { printModule(M); }
@ -1991,11 +2000,16 @@ void AssemblyWriter::printInstruction(const Instruction &I) {
Out << ", align " << cast<StoreInst>(I).getAlignment();
}
// Print DebugInfo
// Print Metadata info
Metadata &TheMetadata = I.getContext().getMetadata();
unsigned MDDbgKind = TheMetadata.getMDKind("dbg");
if (const MDNode *Dbg = TheMetadata.getMD(MDDbgKind, &I))
Out << ", dbg !" << Machine.getMetadataSlot(Dbg);
const Metadata::MDMapTy *MDMap = TheMetadata.getMDs(&I);
if (MDMap)
for (Metadata::MDMapTy::const_iterator MI = MDMap->begin(),
ME = MDMap->end(); MI != ME; ++MI)
if (const MDNode *MD = dyn_cast_or_null<MDNode>(MI->second))
Out << ", " << MDNames[MI->first]
<< " !" << Machine.getMetadataSlot(MD);
printInfoComment(I);
}