Use CallbackVH, instead of WeakVH, to hold MDNode elements.
[oota-llvm.git] / lib / VMCore / LLVMContextImpl.h
1 //===-- LLVMContextImpl.h - The LLVMContextImpl opaque class --------------===//
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 declares LLVMContextImpl, the opaque implementation 
11 //  of LLVMContext.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef LLVM_LLVMCONTEXT_IMPL_H
16 #define LLVM_LLVMCONTEXT_IMPL_H
17
18 #include "ConstantsContext.h"
19 #include "LeaksContext.h"
20 #include "TypesContext.h"
21 #include "llvm/LLVMContext.h"
22 #include "llvm/Metadata.h"
23 #include "llvm/Constants.h"
24 #include "llvm/DerivedTypes.h"
25 #include "llvm/System/Mutex.h"
26 #include "llvm/System/RWMutex.h"
27 #include "llvm/Assembly/Writer.h"
28 #include "llvm/ADT/APFloat.h"
29 #include "llvm/ADT/APInt.h"
30 #include "llvm/ADT/DenseMap.h"
31 #include "llvm/ADT/FoldingSet.h"
32 #include "llvm/ADT/StringMap.h"
33 #include <vector>
34
35 namespace llvm {
36
37 class ConstantInt;
38 class ConstantFP;
39 class MDString;
40 class MDNode;
41 class LLVMContext;
42 class Type;
43 class Value;
44
45 struct DenseMapAPIntKeyInfo {
46   struct KeyTy {
47     APInt val;
48     const Type* type;
49     KeyTy(const APInt& V, const Type* Ty) : val(V), type(Ty) {}
50     KeyTy(const KeyTy& that) : val(that.val), type(that.type) {}
51     bool operator==(const KeyTy& that) const {
52       return type == that.type && this->val == that.val;
53     }
54     bool operator!=(const KeyTy& that) const {
55       return !this->operator==(that);
56     }
57   };
58   static inline KeyTy getEmptyKey() { return KeyTy(APInt(1,0), 0); }
59   static inline KeyTy getTombstoneKey() { return KeyTy(APInt(1,1), 0); }
60   static unsigned getHashValue(const KeyTy &Key) {
61     return DenseMapInfo<void*>::getHashValue(Key.type) ^ 
62       Key.val.getHashValue();
63   }
64   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
65     return LHS == RHS;
66   }
67   static bool isPod() { return false; }
68 };
69
70 struct DenseMapAPFloatKeyInfo {
71   struct KeyTy {
72     APFloat val;
73     KeyTy(const APFloat& V) : val(V){}
74     KeyTy(const KeyTy& that) : val(that.val) {}
75     bool operator==(const KeyTy& that) const {
76       return this->val.bitwiseIsEqual(that.val);
77     }
78     bool operator!=(const KeyTy& that) const {
79       return !this->operator==(that);
80     }
81   };
82   static inline KeyTy getEmptyKey() { 
83     return KeyTy(APFloat(APFloat::Bogus,1));
84   }
85   static inline KeyTy getTombstoneKey() { 
86     return KeyTy(APFloat(APFloat::Bogus,2)); 
87   }
88   static unsigned getHashValue(const KeyTy &Key) {
89     return Key.val.getHashValue();
90   }
91   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
92     return LHS == RHS;
93   }
94   static bool isPod() { return false; }
95 };
96
97 class LLVMContextImpl {
98 public:
99   sys::SmartRWMutex<true> ConstantsLock;
100   typedef DenseMap<DenseMapAPIntKeyInfo::KeyTy, ConstantInt*, 
101                          DenseMapAPIntKeyInfo> IntMapTy;
102   IntMapTy IntConstants;
103   
104   typedef DenseMap<DenseMapAPFloatKeyInfo::KeyTy, ConstantFP*, 
105                          DenseMapAPFloatKeyInfo> FPMapTy;
106   FPMapTy FPConstants;
107   
108   StringMap<MDString*> MDStringCache;
109   
110   FoldingSet<MDNode> MDNodeSet;
111   
112   ValueMap<char, Type, ConstantAggregateZero> AggZeroConstants;
113
114   SmallPtrSet<const MDNode *, 8> MDNodes;
115
116   typedef ValueMap<std::vector<Constant*>, ArrayType, 
117     ConstantArray, true /*largekey*/> ArrayConstantsTy;
118   ArrayConstantsTy ArrayConstants;
119   
120   typedef ValueMap<std::vector<Constant*>, StructType,
121                    ConstantStruct, true /*largekey*/> StructConstantsTy;
122   StructConstantsTy StructConstants;
123   
124   typedef ValueMap<std::vector<Constant*>, VectorType,
125                    ConstantVector> VectorConstantsTy;
126   VectorConstantsTy VectorConstants;
127   
128   ValueMap<char, PointerType, ConstantPointerNull> NullPtrConstants;
129   
130   ValueMap<char, Type, UndefValue> UndefValueConstants;
131   
132   ValueMap<ExprMapKeyType, Type, ConstantExpr> ExprConstants;
133   
134   ConstantInt *TheTrueVal;
135   ConstantInt *TheFalseVal;
136   
137   // Lock used for guarding access to the leak detector
138   sys::SmartMutex<true> LLVMObjectsLock;
139   LeakDetectorImpl<Value> LLVMObjects;
140   
141   // Lock used for guarding access to the type maps.
142   sys::SmartMutex<true> TypeMapLock;
143   
144   // Recursive lock used for guarding access to AbstractTypeUsers.
145   // NOTE: The true template parameter means this will no-op when we're not in
146   // multithreaded mode.
147   sys::SmartMutex<true> AbstractTypeUsersLock;
148
149   // Basic type instances.
150   const Type VoidTy;
151   const Type LabelTy;
152   const Type FloatTy;
153   const Type DoubleTy;
154   const Type MetadataTy;
155   const Type X86_FP80Ty;
156   const Type FP128Ty;
157   const Type PPC_FP128Ty;
158   const IntegerType Int1Ty;
159   const IntegerType Int8Ty;
160   const IntegerType Int16Ty;
161   const IntegerType Int32Ty;
162   const IntegerType Int64Ty;
163
164   // Concrete/Abstract TypeDescriptions - We lazily calculate type descriptions
165   // for types as they are needed.  Because resolution of types must invalidate
166   // all of the abstract type descriptions, we keep them in a seperate map to 
167   // make this easy.
168   TypePrinting ConcreteTypeDescriptions;
169   TypePrinting AbstractTypeDescriptions;
170   
171   TypeMap<ArrayValType, ArrayType> ArrayTypes;
172   TypeMap<VectorValType, VectorType> VectorTypes;
173   TypeMap<PointerValType, PointerType> PointerTypes;
174   TypeMap<FunctionValType, FunctionType> FunctionTypes;
175   TypeMap<StructValType, StructType> StructTypes;
176   TypeMap<IntegerValType, IntegerType> IntegerTypes;
177
178   /// ValueHandles - This map keeps track of all of the value handles that are
179   /// watching a Value*.  The Value::HasValueHandle bit is used to know
180   // whether or not a value has an entry in this map.
181   typedef DenseMap<Value*, ValueHandleBase*> ValueHandlesTy;
182   ValueHandlesTy ValueHandles;
183   
184   LLVMContextImpl(LLVMContext &C) : TheTrueVal(0), TheFalseVal(0),
185     VoidTy(C, Type::VoidTyID),
186     LabelTy(C, Type::LabelTyID),
187     FloatTy(C, Type::FloatTyID),
188     DoubleTy(C, Type::DoubleTyID),
189     MetadataTy(C, Type::MetadataTyID),
190     X86_FP80Ty(C, Type::X86_FP80TyID),
191     FP128Ty(C, Type::FP128TyID),
192     PPC_FP128Ty(C, Type::PPC_FP128TyID),
193     Int1Ty(C, 1),
194     Int8Ty(C, 8),
195     Int16Ty(C, 16),
196     Int32Ty(C, 32),
197     Int64Ty(C, 64) { }
198
199   ~LLVMContextImpl()
200   {
201     ExprConstants.freeConstants();
202     ArrayConstants.freeConstants();
203     StructConstants.freeConstants();
204     VectorConstants.freeConstants();
205     AggZeroConstants.freeConstants();
206     NullPtrConstants.freeConstants();
207     UndefValueConstants.freeConstants();
208     for (IntMapTy::iterator I=IntConstants.begin(), E=IntConstants.end(); 
209          I != E; ++I) {
210       if (I->second->use_empty())
211         delete I->second;
212     }
213     for (FPMapTy::iterator I=FPConstants.begin(), E=FPConstants.end(); 
214          I != E; ++I) {
215       if (I->second->use_empty())
216         delete I->second;
217     }
218   }
219 };
220
221 }
222
223 #endif