30a7fc4af930d6473c3963716d1d21d48e9c78c6
[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   
141   LLVMContextImpl() : TheTrueVal(0), TheFalseVal(0) { }
142 };
143
144 }
145
146 #endif