6df804ab084430179698994fa96267fe55076538
[oota-llvm.git] / lib / VMCore / LLVMContextImpl.h
1 //===-- LLVMContextImpl.h - The LLVMContextImpl opaque class ----*- C++ -*-===//
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/Metadata.h"
25 #include "llvm/Assembly/Writer.h"
26 #include "llvm/Support/ValueHandle.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/SmallPtrSet.h"
32 #include "llvm/ADT/StringMap.h"
33 #include <vector>
34
35 namespace llvm {
36
37 class ConstantInt;
38 class ConstantFP;
39 class LLVMContext;
40 class Type;
41 class Value;
42
43 struct DenseMapAPIntKeyInfo {
44   struct KeyTy {
45     APInt val;
46     const Type* type;
47     KeyTy(const APInt& V, const Type* Ty) : val(V), type(Ty) {}
48     KeyTy(const KeyTy& that) : val(that.val), type(that.type) {}
49     bool operator==(const KeyTy& that) const {
50       return type == that.type && this->val == that.val;
51     }
52     bool operator!=(const KeyTy& that) const {
53       return !this->operator==(that);
54     }
55   };
56   static inline KeyTy getEmptyKey() { return KeyTy(APInt(1,0), 0); }
57   static inline KeyTy getTombstoneKey() { return KeyTy(APInt(1,1), 0); }
58   static unsigned getHashValue(const KeyTy &Key) {
59     return DenseMapInfo<void*>::getHashValue(Key.type) ^ 
60       Key.val.getHashValue();
61   }
62   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
63     return LHS == RHS;
64   }
65 };
66
67 struct DenseMapAPFloatKeyInfo {
68   struct KeyTy {
69     APFloat val;
70     KeyTy(const APFloat& V) : val(V){}
71     KeyTy(const KeyTy& that) : val(that.val) {}
72     bool operator==(const KeyTy& that) const {
73       return this->val.bitwiseIsEqual(that.val);
74     }
75     bool operator!=(const KeyTy& that) const {
76       return !this->operator==(that);
77     }
78   };
79   static inline KeyTy getEmptyKey() { 
80     return KeyTy(APFloat(APFloat::Bogus,1));
81   }
82   static inline KeyTy getTombstoneKey() { 
83     return KeyTy(APFloat(APFloat::Bogus,2)); 
84   }
85   static unsigned getHashValue(const KeyTy &Key) {
86     return Key.val.getHashValue();
87   }
88   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
89     return LHS == RHS;
90   }
91 };
92
93 /// DebugRecVH - This is a CallbackVH used to keep the Scope -> index maps
94 /// up to date as MDNodes mutate.  This class is implemented in DebugLoc.cpp.
95 class DebugRecVH : public CallbackVH {
96   /// Ctx - This is the LLVM Context being referenced.
97   LLVMContextImpl *Ctx;
98   
99   /// Idx - The index into either ScopeRecordIdx or ScopeInlinedAtRecords that
100   /// this reference lives in.  If this is zero, then it represents a
101   /// non-canonical entry that has no DenseMap value.  This can happen due to
102   /// RAUW.
103   int Idx;
104 public:
105   DebugRecVH(MDNode *n, LLVMContextImpl *ctx, int idx)
106     : CallbackVH(n), Ctx(ctx), Idx(idx) {}
107   
108   MDNode *get() const {
109     return cast_or_null<MDNode>(getValPtr());
110   }
111   
112   virtual void deleted();
113   virtual void allUsesReplacedWith(Value *VNew);
114 };
115   
116 class LLVMContextImpl {
117 public:
118   /// OwnedModules - The set of modules instantiated in this context, and which
119   /// will be automatically deleted if this context is deleted.
120   SmallPtrSet<Module*, 4> OwnedModules;
121   
122   void *InlineAsmDiagHandler, *InlineAsmDiagContext;
123   
124   typedef DenseMap<DenseMapAPIntKeyInfo::KeyTy, ConstantInt*, 
125                          DenseMapAPIntKeyInfo> IntMapTy;
126   IntMapTy IntConstants;
127   
128   typedef DenseMap<DenseMapAPFloatKeyInfo::KeyTy, ConstantFP*, 
129                          DenseMapAPFloatKeyInfo> FPMapTy;
130   FPMapTy FPConstants;
131   
132   StringMap<MDString*> MDStringCache;
133   
134   FoldingSet<MDNode> MDNodeSet;
135   // MDNodes may be uniqued or not uniqued.  When they're not uniqued, they
136   // aren't in the MDNodeSet, but they're still shared between objects, so no
137   // one object can destroy them.  This set allows us to at least destroy them
138   // on Context destruction.
139   SmallPtrSet<MDNode*, 1> NonUniquedMDNodes;
140   
141   ConstantUniqueMap<char, Type, ConstantAggregateZero> AggZeroConstants;
142
143   typedef ConstantUniqueMap<std::vector<Constant*>, ArrayType,
144     ConstantArray, true /*largekey*/> ArrayConstantsTy;
145   ArrayConstantsTy ArrayConstants;
146   
147   typedef ConstantUniqueMap<std::vector<Constant*>, StructType,
148     ConstantStruct, true /*largekey*/> StructConstantsTy;
149   StructConstantsTy StructConstants;
150   
151   typedef ConstantUniqueMap<std::vector<Constant*>, VectorType,
152                             ConstantVector> VectorConstantsTy;
153   VectorConstantsTy VectorConstants;
154   
155   ConstantUniqueMap<char, PointerType, ConstantPointerNull> NullPtrConstants;
156   ConstantUniqueMap<char, Type, UndefValue> UndefValueConstants;
157   
158   DenseMap<std::pair<Function*, BasicBlock*> , BlockAddress*> BlockAddresses;
159   ConstantUniqueMap<ExprMapKeyType, Type, ConstantExpr> ExprConstants;
160
161   ConstantUniqueMap<InlineAsmKeyType, PointerType, InlineAsm> InlineAsms;
162   
163   ConstantInt *TheTrueVal;
164   ConstantInt *TheFalseVal;
165   
166   LeakDetectorImpl<Value> LLVMObjects;
167   
168   // Basic type instances.
169   const Type VoidTy;
170   const Type LabelTy;
171   const Type FloatTy;
172   const Type DoubleTy;
173   const Type MetadataTy;
174   const Type X86_FP80Ty;
175   const Type FP128Ty;
176   const Type PPC_FP128Ty;
177   const IntegerType Int1Ty;
178   const IntegerType Int8Ty;
179   const IntegerType Int16Ty;
180   const IntegerType Int32Ty;
181   const IntegerType Int64Ty;
182
183   // Concrete/Abstract TypeDescriptions - We lazily calculate type descriptions
184   // for types as they are needed.  Because resolution of types must invalidate
185   // all of the abstract type descriptions, we keep them in a seperate map to 
186   // make this easy.
187   TypePrinting ConcreteTypeDescriptions;
188   TypePrinting AbstractTypeDescriptions;
189   
190   TypeMap<ArrayValType, ArrayType> ArrayTypes;
191   TypeMap<VectorValType, VectorType> VectorTypes;
192   TypeMap<PointerValType, PointerType> PointerTypes;
193   TypeMap<FunctionValType, FunctionType> FunctionTypes;
194   TypeMap<StructValType, StructType> StructTypes;
195   TypeMap<IntegerValType, IntegerType> IntegerTypes;
196
197   // Opaque types are not structurally uniqued, so don't use TypeMap.
198   typedef SmallPtrSet<const OpaqueType*, 8> OpaqueTypesTy;
199   OpaqueTypesTy OpaqueTypes;
200
201   /// Used as an abstract type that will never be resolved.
202   OpaqueType *const AlwaysOpaqueTy;
203
204
205   /// ValueHandles - This map keeps track of all of the value handles that are
206   /// watching a Value*.  The Value::HasValueHandle bit is used to know
207   // whether or not a value has an entry in this map.
208   typedef DenseMap<Value*, ValueHandleBase*> ValueHandlesTy;
209   ValueHandlesTy ValueHandles;
210   
211   /// CustomMDKindNames - Map to hold the metadata string to ID mapping.
212   StringMap<unsigned> CustomMDKindNames;
213   
214   typedef std::pair<unsigned, TrackingVH<MDNode> > MDPairTy;
215   typedef SmallVector<MDPairTy, 2> MDMapTy;
216
217   /// MetadataStore - Collection of per-instruction metadata used in this
218   /// context.
219   DenseMap<const Instruction *, MDMapTy> MetadataStore;
220   
221   /// ScopeRecordIdx - This is the index in ScopeRecords for an MDNode scope
222   /// entry with no "inlined at" element.
223   DenseMap<MDNode*, int> ScopeRecordIdx;
224   
225   /// ScopeRecords - These are the actual mdnodes (in a value handle) for an
226   /// index.  The ValueHandle ensures that ScopeRecordIdx stays up to date if
227   /// the MDNode is RAUW'd.
228   std::vector<DebugRecVH> ScopeRecords;
229   
230   /// ScopeInlinedAtIdx - This is the index in ScopeInlinedAtRecords for an
231   /// scope/inlined-at pair.
232   DenseMap<std::pair<MDNode*, MDNode*>, int> ScopeInlinedAtIdx;
233   
234   /// ScopeInlinedAtRecords - These are the actual mdnodes (in value handles)
235   /// for an index.  The ValueHandle ensures that ScopeINlinedAtIdx stays up
236   /// to date.
237   std::vector<std::pair<DebugRecVH, DebugRecVH> > ScopeInlinedAtRecords;
238   
239   int getOrAddScopeRecordIdxEntry(MDNode *N, int ExistingIdx);
240   int getOrAddScopeInlinedAtIdxEntry(MDNode *Scope, MDNode *IA,int ExistingIdx);
241   
242   LLVMContextImpl(LLVMContext &C);
243   ~LLVMContextImpl();
244 };
245
246 }
247
248 #endif