Remove MDNode from ValueMap when MDNode is destroyed.
[oota-llvm.git] / lib / VMCore / Metadata.cpp
1 //===-- Metadata.cpp - Implement Metadata classes -------------------------===//
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 // This file implements the Metadata classes.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "LLVMContextImpl.h"
15 #include "llvm/Metadata.h"
16 #include "llvm/LLVMContext.h"
17 #include "llvm/Module.h"
18 #include "SymbolTableListTraitsImpl.h"
19 using namespace llvm;
20
21 //===----------------------------------------------------------------------===//
22 //MetadataBase implementation
23 //
24
25 /// resizeOperands - Metadata keeps track of other metadata uses using 
26 /// OperandList. Resize this list to hold anticipated number of metadata
27 /// operands.
28 void MetadataBase::resizeOperands(unsigned NumOps) {
29   unsigned e = getNumOperands();
30   if (NumOps == 0) {
31     NumOps = e*2;
32     if (NumOps < 2) NumOps = 2;  
33   } else if (NumOps > NumOperands) {
34     // No resize needed.
35     if (ReservedSpace >= NumOps) return;
36   } else if (NumOps == NumOperands) {
37     if (ReservedSpace == NumOps) return;
38   } else {
39     return;
40   }
41
42   ReservedSpace = NumOps;
43   Use *OldOps = OperandList;
44   Use *NewOps = allocHungoffUses(NumOps);
45   std::copy(OldOps, OldOps + e, NewOps);
46   OperandList = NewOps;
47   if (OldOps) Use::zap(OldOps, OldOps + e, true);
48 }
49 //===----------------------------------------------------------------------===//
50 //MDString implementation
51 //
52 MDString *MDString::get(LLVMContext &Context, const StringRef &Str) {
53   LLVMContextImpl *pImpl = Context.pImpl;
54   sys::SmartScopedWriter<true> Writer(pImpl->ConstantsLock);
55   StringMapEntry<MDString *> &Entry = 
56     pImpl->MDStringCache.GetOrCreateValue(Str);
57   MDString *&S = Entry.getValue();
58   if (!S) S = new MDString(Entry.getKeyData(),
59                            Entry.getKeyLength());
60
61   return S;
62 }
63
64 //===----------------------------------------------------------------------===//
65 //MDNode implementation
66 //
67 MDNode::MDNode(Value*const* Vals, unsigned NumVals)
68   : MetadataBase(Type::MetadataTy, Value::MDNodeVal) {
69   NumOperands = 0;
70   resizeOperands(NumVals);
71   for (unsigned i = 0; i != NumVals; ++i) {
72     // Only record metadata uses.
73     if (MetadataBase *MB = dyn_cast_or_null<MetadataBase>(Vals[i]))
74       OperandList[NumOperands++] = MB;
75     Node.push_back(WeakVH(Vals[i]));
76   }
77 }
78
79 MDNode *MDNode::get(LLVMContext &Context, Value*const* Vals, unsigned NumVals) {
80   LLVMContextImpl *pImpl = Context.pImpl;
81   std::vector<Value*> V;
82   V.reserve(NumVals);
83   for (unsigned i = 0; i < NumVals; ++i)
84     V.push_back(Vals[i]);
85   
86   return pImpl->MDNodes.getOrCreate(Type::MetadataTy, V);
87 }
88
89 /// dropAllReferences - Remove all uses and clear node vector.
90 void MDNode::dropAllReferences() {
91   User::dropAllReferences();
92   Node.clear();
93 }
94
95 static std::vector<Value*> getValType(MDNode *N) {
96   std::vector<Value*> Elements;
97   Elements.reserve(N->getNumElements());
98   for (unsigned i = 0, e = N->getNumElements(); i != e; ++i)
99     Elements.push_back(N->getElement(i));
100   return Elements;
101 }
102
103 MDNode::~MDNode() {
104   dropAllReferences();
105   getType()->getContext().pImpl->MDNodes.remove(this);
106 }
107 //===----------------------------------------------------------------------===//
108 //NamedMDNode implementation
109 //
110 NamedMDNode::NamedMDNode(const Twine &N, MetadataBase*const* MDs, 
111                          unsigned NumMDs, Module *ParentModule)
112   : MetadataBase(Type::MetadataTy, Value::NamedMDNodeVal), Parent(0) {
113   setName(N);
114   NumOperands = 0;
115   resizeOperands(NumMDs);
116
117   for (unsigned i = 0; i != NumMDs; ++i) {
118     if (MDs[i])
119       OperandList[NumOperands++] = MDs[i];
120     Node.push_back(WeakMetadataVH(MDs[i]));
121   }
122   if (ParentModule)
123     ParentModule->getNamedMDList().push_back(this);
124 }
125
126 /// eraseFromParent - Drop all references and remove the node from parent
127 /// module.
128 void NamedMDNode::eraseFromParent() {
129   getParent()->getNamedMDList().erase(this);
130 }
131
132 /// dropAllReferences - Remove all uses and clear node vector.
133 void NamedMDNode::dropAllReferences() {
134   User::dropAllReferences();
135   Node.clear();
136 }
137
138 NamedMDNode::~NamedMDNode() {
139   dropAllReferences();
140 }