#include "llvm/Instruction.h"
#include "llvm/Instructions.h"
#include "llvm/Operator.h"
-#include "llvm/MDNode.h"
+#include "llvm/Metadata.h"
#include "llvm/Module.h"
#include "llvm/ValueSymbolTable.h"
#include "llvm/TypeSymbolTable.h"
/// TheMDNode - The MDNode for which we are holding slot numbers.
const MDNode *TheMDNode;
+ /// TheNamedMDNode - The MDNode for which we are holding slot numbers.
+ const NamedMDNode *TheNamedMDNode;
+
/// mMap - The TypePlanes map for the module level data.
ValueMap mMap;
unsigned mNext;
explicit SlotTracker(const Function *F);
/// Construct from a mdnode.
explicit SlotTracker(const MDNode *N);
+ /// Construct from a named mdnode.
+ explicit SlotTracker(const NamedMDNode *N);
/// Return the slot number of the specified value in it's type
/// plane. If something is not in the SlotTracker, return -1.
/// Add all MDNode operands.
void processMDNode();
+ /// Add all MDNode operands.
+ void processNamedMDNode();
+
SlotTracker(const SlotTracker &); // DO NOT IMPLEMENT
void operator=(const SlotTracker &); // DO NOT IMPLEMENT
};
// to be added to the slot table.
SlotTracker::SlotTracker(const Module *M)
: TheModule(M), TheFunction(0), FunctionProcessed(false), TheMDNode(0),
- mNext(0), fNext(0), mdnNext(0) {
+ TheNamedMDNode(0), mNext(0), fNext(0), mdnNext(0) {
}
// Function level constructor. Causes the contents of the Module and the one
// function provided to be added to the slot table.
SlotTracker::SlotTracker(const Function *F)
: TheModule(F ? F->getParent() : 0), TheFunction(F), FunctionProcessed(false),
- TheMDNode(0), mNext(0), fNext(0), mdnNext(0) {
+ TheMDNode(0), TheNamedMDNode(0), mNext(0), fNext(0), mdnNext(0) {
}
// Constructor to handle single MDNode.
SlotTracker::SlotTracker(const MDNode *C)
: TheModule(0), TheFunction(0), FunctionProcessed(false), TheMDNode(C),
- mNext(0), fNext(0), mdnNext(0) {
+ TheNamedMDNode(0), mNext(0), fNext(0), mdnNext(0) {
+}
+
+// Constructor to handle single NamedMDNode.
+SlotTracker::SlotTracker(const NamedMDNode *N)
+ : TheModule(0), TheFunction(0), FunctionProcessed(false), TheMDNode(0),
+ TheNamedMDNode(N), mNext(0), fNext(0), mdnNext(0) {
}
inline void SlotTracker::initialize() {
if (TheMDNode)
processMDNode();
+
+ if (TheNamedMDNode)
+ processNamedMDNode();
}
// Iterate through all the global variables, functions, and global
}
}
+ // Add metadata used by named metadata.
+ for (Module::const_named_metadata_iterator
+ I = TheModule->named_metadata_begin(),
+ E = TheModule->named_metadata_end(); I != E; ++I) {
+ const NamedMDNode *NMD = I;
+ for (unsigned i = 0, e = NMD->getNumElements(); i != e; ++i) {
+ MDNode *MD = dyn_cast_or_null<MDNode>(NMD->getElement(i));
+ if (MD)
+ CreateMetadataSlot(MD);
+ }
+ }
+
// Add all the unnamed functions to the table.
for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
I != E; ++I)
ST_DEBUG("end processMDNode!\n");
}
+/// processNamedMDNode - Process TheNamedMDNode.
+void SlotTracker::processNamedMDNode() {
+ ST_DEBUG("begin processNamedMDNode!\n");
+ mdnNext = 0;
+ for (unsigned i = 0, e = TheNamedMDNode->getNumElements(); i != e; ++i) {
+ MDNode *MD = dyn_cast_or_null<MDNode>(TheNamedMDNode->getElement(i));
+ if (MD)
+ CreateMetadataSlot(MD);
+ }
+ TheNamedMDNode = 0;
+ ST_DEBUG("end processNamedMDNode!\n");
+}
+
/// Clean up after incorporating a function. This is the only way to get out of
/// the function incorporation state that affects get*Slot/Create*Slot. Function
/// incorporation state is indicated by TheFunction != 0.
for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
printFunction(I);
+ // Output named metadata.
+ for (Module::const_named_metadata_iterator I = M->named_metadata_begin(),
+ E = M->named_metadata_end(); I != E; ++I) {
+ const NamedMDNode *NMD = I;
+ Out << "!" << NMD->getName() << " = !{";
+ for (unsigned i = 0, e = NMD->getNumElements(); i != e; ++i) {
+ if (i) Out << ", ";
+ MDNode *MD = dyn_cast_or_null<MDNode>(NMD->getElement(i));
+ Out << '!' << Machine.getMetadataSlot(MD);
+ }
+ Out << "}\n";
+ }
+
+ // Output metadata.
WriteMDNodes(Out, TypePrinter, Machine);
}
TypePrinting TypePrinter;
SlotTable.initialize();
WriteMDNodes(OS, TypePrinter, SlotTable);
+ } else if (const NamedMDNode *N = dyn_cast<NamedMDNode>(this)) {
+ SlotTracker SlotTable(N);
+ TypePrinting TypePrinter;
+ SlotTable.initialize();
+ OS << "!" << N->getName() << " = !{";
+ for (unsigned i = 0, e = N->getNumElements(); i != e; ++i) {
+ if (i) OS << ", ";
+ MDNode *MD = dyn_cast_or_null<MDNode>(N->getElement(i));
+ if (MD)
+ OS << '!' << SlotTable.getMetadataSlot(MD);
+ else
+ OS << "null";
+ }
+ OS << "}\n";
+ WriteMDNodes(OS, TypePrinter, SlotTable);
} else if (const Constant *C = dyn_cast<Constant>(this)) {
TypePrinting TypePrinter;
TypePrinter.print(C->getType(), OS);