Free the constants that have no uses in ~LLVMContext.
[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/Constants.h"
23 #include "llvm/DerivedTypes.h"
24 #include "llvm/System/Mutex.h"
25 #include "llvm/System/RWMutex.h"
26 #include "llvm/Assembly/Writer.h"
27 #include "llvm/ADT/APFloat.h"
28 #include "llvm/ADT/APInt.h"
29 #include "llvm/ADT/DenseMap.h"
30 #include "llvm/ADT/FoldingSet.h"
31 #include "llvm/ADT/StringMap.h"
32 #include <vector>
33
34 namespace llvm {
35
36 class ConstantInt;
37 class ConstantFP;
38 class MDString;
39 class MDNode;
40 class LLVMContext;
41 class Type;
42 class Value;
43
44 struct DenseMapAPIntKeyInfo {
45   struct KeyTy {
46     APInt val;
47     const Type* type;
48     KeyTy(const APInt& V, const Type* Ty) : val(V), type(Ty) {}
49     KeyTy(const KeyTy& that) : val(that.val), type(that.type) {}
50     bool operator==(const KeyTy& that) const {
51       return type == that.type && this->val == that.val;
52     }
53     bool operator!=(const KeyTy& that) const {
54       return !this->operator==(that);
55     }
56   };
57   static inline KeyTy getEmptyKey() { return KeyTy(APInt(1,0), 0); }
58   static inline KeyTy getTombstoneKey() { return KeyTy(APInt(1,1), 0); }
59   static unsigned getHashValue(const KeyTy &Key) {
60     return DenseMapInfo<void*>::getHashValue(Key.type) ^ 
61       Key.val.getHashValue();
62   }
63   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
64     return LHS == RHS;
65   }
66   static bool isPod() { return false; }
67 };
68
69 struct DenseMapAPFloatKeyInfo {
70   struct KeyTy {
71     APFloat val;
72     KeyTy(const APFloat& V) : val(V){}
73     KeyTy(const KeyTy& that) : val(that.val) {}
74     bool operator==(const KeyTy& that) const {
75       return this->val.bitwiseIsEqual(that.val);
76     }
77     bool operator!=(const KeyTy& that) const {
78       return !this->operator==(that);
79     }
80   };
81   static inline KeyTy getEmptyKey() { 
82     return KeyTy(APFloat(APFloat::Bogus,1));
83   }
84   static inline KeyTy getTombstoneKey() { 
85     return KeyTy(APFloat(APFloat::Bogus,2)); 
86   }
87   static unsigned getHashValue(const KeyTy &Key) {
88     return Key.val.getHashValue();
89   }
90   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
91     return LHS == RHS;
92   }
93   static bool isPod() { return false; }
94 };
95
96 class LLVMContextImpl {
97 public:
98   sys::SmartRWMutex<true> ConstantsLock;
99   typedef DenseMap<DenseMapAPIntKeyInfo::KeyTy, ConstantInt*, 
100                          DenseMapAPIntKeyInfo> IntMapTy;
101   IntMapTy IntConstants;
102   
103   typedef DenseMap<DenseMapAPFloatKeyInfo::KeyTy, ConstantFP*, 
104                          DenseMapAPFloatKeyInfo> FPMapTy;
105   FPMapTy FPConstants;
106   
107   StringMap<MDString*> MDStringCache;
108   
109   ValueMap<char, Type, ConstantAggregateZero> AggZeroConstants;
110
111   typedef ValueMap<std::vector<Value*>, Type, MDNode, true /*largekey*/> 
112   MDNodeMapTy;
113
114   MDNodeMapTy 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
206     AggZeroConstants.freeConstants();
207     NullPtrConstants.freeConstants();
208     UndefValueConstants.freeConstants();
209     for (IntMapTy::iterator I=IntConstants.begin(), E=IntConstants.end(); 
210          I != E; ++I) {
211       if (I->second->use_empty())
212         delete I->second;
213     }
214     for (FPMapTy::iterator I=FPConstants.begin(), E=FPConstants.end(); 
215          I != E; ++I) {
216       if (I->second->use_empty())
217         delete I->second;
218     }
219   }
220 };
221
222 }
223
224 #endif