MDNodes that refer to an instruction are local to a function; in that case, explicitly keep track of the function they are local to

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@91497 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Victor Hernandez 2009-12-16 02:52:09 +00:00
parent bbc71b2904
commit 57c0f20601
2 changed files with 16 additions and 5 deletions

View File

@ -111,13 +111,16 @@ class MDNode : public MetadataBase, public FoldingSetNode {
ElementVH *Node;
unsigned NodeSize;
Function *LocalFunction;
protected:
explicit MDNode(LLVMContext &C, Value *const *Vals, unsigned NumVals);
explicit MDNode(LLVMContext &C, Value *const *Vals, unsigned NumVals,
Function *LocalFunction = NULL);
public:
// Constructors and destructors.
static MDNode *get(LLVMContext &Context,
Value *const *Vals, unsigned NumVals);
Value *const *Vals, unsigned NumVals,
Function *LocalFunction = NULL);
/// ~MDNode - Destroy MDNode.
~MDNode();
@ -131,6 +134,9 @@ public:
/// getNumElements - Return number of MDNode elements.
unsigned getNumElements() const { return NodeSize; }
/// isFunctionLocal - Return whether MDNode is local to a function.
bool isFunctionLocal() const { return LocalFunction; }
/// Profile - calculate a unique identifier for this MDNode to collapse
/// duplicates
void Profile(FoldingSetNodeID &ID) const;

View File

@ -49,13 +49,15 @@ MDString *MDString::get(LLVMContext &Context, const char *Str) {
//===----------------------------------------------------------------------===//
// MDNode implementation.
//
MDNode::MDNode(LLVMContext &C, Value *const *Vals, unsigned NumVals)
MDNode::MDNode(LLVMContext &C, Value *const *Vals, unsigned NumVals,
Function *LocalFunction)
: MetadataBase(Type::getMetadataTy(C), Value::MDNodeVal) {
NodeSize = NumVals;
Node = new ElementVH[NodeSize];
ElementVH *Ptr = Node;
for (unsigned i = 0; i != NumVals; ++i)
*Ptr++ = ElementVH(Vals[i], this);
LocalFunction = LocalFunction;
}
void MDNode::Profile(FoldingSetNodeID &ID) const {
@ -63,17 +65,20 @@ void MDNode::Profile(FoldingSetNodeID &ID) const {
ID.AddPointer(getElement(i));
}
MDNode *MDNode::get(LLVMContext &Context, Value*const* Vals, unsigned NumVals) {
MDNode *MDNode::get(LLVMContext &Context, Value*const* Vals, unsigned NumVals,
Function *LocalFunction) {
LLVMContextImpl *pImpl = Context.pImpl;
FoldingSetNodeID ID;
for (unsigned i = 0; i != NumVals; ++i)
ID.AddPointer(Vals[i]);
if (LocalFunction)
ID.AddPointer(LocalFunction);
void *InsertPoint;
MDNode *N = pImpl->MDNodeSet.FindNodeOrInsertPos(ID, InsertPoint);
if (!N) {
// InsertPoint will have been set by the FindNodeOrInsertPos call.
N = new MDNode(Context, Vals, NumVals);
N = new MDNode(Context, Vals, NumVals, LocalFunction);
pImpl->MDNodeSet.InsertNode(N, InsertPoint);
}
return N;