Add locking around the attributes list.
[oota-llvm.git] / lib / VMCore / LLVMContextImpl.h
1 //===----------------- LLVMContextImpl.h - Implementation ------*- 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 "TypesContext.h"
20 #include "llvm/LLVMContext.h"
21 #include "llvm/Constants.h"
22 #include "llvm/DerivedTypes.h"
23 #include "llvm/System/RWMutex.h"
24 #include "llvm/ADT/APFloat.h"
25 #include "llvm/ADT/APInt.h"
26 #include "llvm/ADT/DenseMap.h"
27 #include "llvm/ADT/FoldingSet.h"
28 #include "llvm/ADT/StringMap.h"
29 #include <vector>
30
31 namespace llvm {
32
33 class ConstantInt;
34 class ConstantFP;
35 class MDString;
36 class MDNode;
37 class LLVMContext;
38 class Type;
39 class Value;
40
41 struct DenseMapAPIntKeyInfo {
42   struct KeyTy {
43     APInt val;
44     const Type* type;
45     KeyTy(const APInt& V, const Type* Ty) : val(V), type(Ty) {}
46     KeyTy(const KeyTy& that) : val(that.val), type(that.type) {}
47     bool operator==(const KeyTy& that) const {
48       return type == that.type && this->val == that.val;
49     }
50     bool operator!=(const KeyTy& that) const {
51       return !this->operator==(that);
52     }
53   };
54   static inline KeyTy getEmptyKey() { return KeyTy(APInt(1,0), 0); }
55   static inline KeyTy getTombstoneKey() { return KeyTy(APInt(1,1), 0); }
56   static unsigned getHashValue(const KeyTy &Key) {
57     return DenseMapInfo<void*>::getHashValue(Key.type) ^ 
58       Key.val.getHashValue();
59   }
60   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
61     return LHS == RHS;
62   }
63   static bool isPod() { return false; }
64 };
65
66 struct DenseMapAPFloatKeyInfo {
67   struct KeyTy {
68     APFloat val;
69     KeyTy(const APFloat& V) : val(V){}
70     KeyTy(const KeyTy& that) : val(that.val) {}
71     bool operator==(const KeyTy& that) const {
72       return this->val.bitwiseIsEqual(that.val);
73     }
74     bool operator!=(const KeyTy& that) const {
75       return !this->operator==(that);
76     }
77   };
78   static inline KeyTy getEmptyKey() { 
79     return KeyTy(APFloat(APFloat::Bogus,1));
80   }
81   static inline KeyTy getTombstoneKey() { 
82     return KeyTy(APFloat(APFloat::Bogus,2)); 
83   }
84   static unsigned getHashValue(const KeyTy &Key) {
85     return Key.val.getHashValue();
86   }
87   static bool isEqual(const KeyTy &LHS, const KeyTy &RHS) {
88     return LHS == RHS;
89   }
90   static bool isPod() { return false; }
91 };
92
93 class LLVMContextImpl {
94 public:
95   sys::SmartRWMutex<true> ConstantsLock;
96   
97   typedef DenseMap<DenseMapAPIntKeyInfo::KeyTy, ConstantInt*, 
98                          DenseMapAPIntKeyInfo> IntMapTy;
99   IntMapTy IntConstants;
100   
101   typedef DenseMap<DenseMapAPFloatKeyInfo::KeyTy, ConstantFP*, 
102                          DenseMapAPFloatKeyInfo> FPMapTy;
103   FPMapTy FPConstants;
104   
105   StringMap<MDString*> MDStringCache;
106   
107   ValueMap<char, Type, ConstantAggregateZero> AggZeroConstants;
108
109   typedef ValueMap<std::vector<Value*>, Type, MDNode, true /*largekey*/> 
110   MDNodeMapTy;
111
112   MDNodeMapTy MDNodes;
113   
114   typedef ValueMap<std::vector<Constant*>, ArrayType, 
115     ConstantArray, true /*largekey*/> ArrayConstantsTy;
116   ArrayConstantsTy ArrayConstants;
117   
118   typedef ValueMap<std::vector<Constant*>, StructType,
119                    ConstantStruct, true /*largekey*/> StructConstantsTy;
120   StructConstantsTy StructConstants;
121   
122   typedef ValueMap<std::vector<Constant*>, VectorType,
123                    ConstantVector> VectorConstantsTy;
124   VectorConstantsTy VectorConstants;
125   
126   ValueMap<char, PointerType, ConstantPointerNull> NullPtrConstants;
127   
128   ValueMap<char, Type, UndefValue> UndefValueConstants;
129   
130   ValueMap<ExprMapKeyType, Type, ConstantExpr> ExprConstants;
131   
132   ConstantInt *TheTrueVal;
133   ConstantInt *TheFalseVal;
134   
135   TypeMap<ArrayValType, ArrayType> ArrayTypes;
136   TypeMap<VectorValType, VectorType> VectorTypes;
137   TypeMap<PointerValType, PointerType> PointerTypes;
138   TypeMap<FunctionValType, FunctionType> FunctionTypes;
139   TypeMap<StructValType, StructType> StructTypes;
140   TypeMap<IntegerValType, IntegerType> IntegerTypes;
141   
142   const Type *VoidTy;
143   const Type *LabelTy;
144   const Type *FloatTy;
145   const Type *DoubleTy;
146   const Type *MetadataTy;
147   const Type *X86_FP80Ty;
148   const Type *FP128Ty;
149   const Type *PPC_FP128Ty;
150   
151   const IntegerType *Int1Ty;
152   const IntegerType *Int8Ty;
153   const IntegerType *Int16Ty;
154   const IntegerType *Int32Ty;
155   const IntegerType *Int64Ty;
156   
157   LLVMContextImpl(LLVMContext &C) : TheTrueVal(0), TheFalseVal(0),
158     VoidTy(new Type(C, Type::VoidTyID)),
159     LabelTy(new Type(C, Type::LabelTyID)),
160     FloatTy(new Type(C, Type::FloatTyID)),
161     DoubleTy(new Type(C, Type::DoubleTyID)),
162     MetadataTy(new Type(C, Type::MetadataTyID)),
163     X86_FP80Ty(new Type(C, Type::X86_FP80TyID)),
164     FP128Ty(new Type(C, Type::FP128TyID)),
165     PPC_FP128Ty(new Type(C, Type::PPC_FP128TyID)),
166     Int1Ty(new IntegerType(C, 1)),
167     Int8Ty(new IntegerType(C, 8)),
168     Int16Ty(new IntegerType(C, 16)),
169     Int32Ty(new IntegerType(C, 32)),
170     Int64Ty(new IntegerType(C, 64)) { }
171   
172   ~LLVMContextImpl() {
173     // In principle, we should delete the member types here.  However,
174     // this causes destruction order issues with the types in the TypeMaps.
175     // For now, just leak this, which is at least not a regression from the
176     // previous behavior, though still undesirable.
177 #if 0
178     delete VoidTy;
179     delete LabelTy;
180     delete FloatTy;
181     delete DoubleTy;
182     delete MetadataTy;
183     delete X86_FP80Ty;
184     delete FP128Ty;
185     delete PPC_FP128Ty;
186     
187     delete Int1Ty;
188     delete Int8Ty;
189     delete Int16Ty;
190     delete Int32Ty;
191     delete Int64Ty;
192 #endif
193   }
194 };
195
196 }
197
198 #endif