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   LLVMContext::InlineAsmDiagHandlerTy InlineAsmDiagHandler;
 123   void *InlineAsmDiagContext;
 124
 125   typedef DenseMap<DenseMapAPIntKeyInfo::KeyTy, ConstantInt*,
 126                          DenseMapAPIntKeyInfo> IntMapTy;
 127   IntMapTy IntConstants;
 128
 129   typedef DenseMap<DenseMapAPFloatKeyInfo::KeyTy, ConstantFP*,
 130                          DenseMapAPFloatKeyInfo> FPMapTy;
 131   FPMapTy FPConstants;
 132
 133   StringMap<MDString*> MDStringCache;
 134
 135   FoldingSet<MDNode> MDNodeSet;
 136   // MDNodes may be uniqued or not uniqued.  When they're not uniqued, they
 137   // aren't in the MDNodeSet, but they're still shared between objects, so no
 138   // one object can destroy them.  This set allows us to at least destroy them
 139   // on Context destruction.
 140   SmallPtrSet<MDNode*, 1> NonUniquedMDNodes;
 141
 142   ConstantUniqueMap<char, Type, ConstantAggregateZero> AggZeroConstants;
 143
 144   typedef ConstantUniqueMap<std::vector<Constant*>, ArrayType,
 145     ConstantArray, true /*largekey*/> ArrayConstantsTy;
 146   ArrayConstantsTy ArrayConstants;
 147
 148   typedef ConstantUniqueMap<std::vector<Constant*>, StructType,
 149     ConstantStruct, true /*largekey*/> StructConstantsTy;
 150   StructConstantsTy StructConstants;
 151
 152   typedef ConstantUniqueMap<std::vector<Constant*>, VectorType,
 153                             ConstantVector> VectorConstantsTy;
 154   VectorConstantsTy VectorConstants;
 155
 156   ConstantUniqueMap<char, PointerType, ConstantPointerNull> NullPtrConstants;
 157   ConstantUniqueMap<char, Type, UndefValue> UndefValueConstants;
 158
 159   DenseMap<std::pair<Function*, BasicBlock*> , BlockAddress*> BlockAddresses;
 160   ConstantUniqueMap<ExprMapKeyType, Type, ConstantExpr> ExprConstants;
 161
 162   ConstantUniqueMap<InlineAsmKeyType, PointerType, InlineAsm> InlineAsms;
 163
 164   ConstantInt *TheTrueVal;
 165   ConstantInt *TheFalseVal;
 166
 167   LeakDetectorImpl<Value> LLVMObjects;
 168
 169   // Basic type instances.
 170   const Type VoidTy;
 171   const Type LabelTy;
 172   const Type FloatTy;
 173   const Type DoubleTy;
 174   const Type MetadataTy;
 175   const Type X86_FP80Ty;
 176   const Type FP128Ty;
 177   const Type PPC_FP128Ty;
 178   const Type X86_MMXTy;
 179   const IntegerType Int1Ty;
 180   const IntegerType Int8Ty;
 181   const IntegerType Int16Ty;
 182   const IntegerType Int32Ty;
 183   const IntegerType Int64Ty;
 184
 185   // Concrete/Abstract TypeDescriptions - We lazily calculate type descriptions
 186   // for types as they are needed.  Because resolution of types must invalidate
 187   // all of the abstract type descriptions, we keep them in a seperate map to
 188   // make this easy.
 189   TypePrinting ConcreteTypeDescriptions;
 190   TypePrinting AbstractTypeDescriptions;
 191
 192   TypeMap<ArrayValType, ArrayType> ArrayTypes;
 193   TypeMap<VectorValType, VectorType> VectorTypes;
 194   TypeMap<PointerValType, PointerType> PointerTypes;
 195   TypeMap<FunctionValType, FunctionType> FunctionTypes;
 196   TypeMap<StructValType, StructType> StructTypes;
 197   TypeMap<IntegerValType, IntegerType> IntegerTypes;
 198
 199   // Opaque types are not structurally uniqued, so don't use TypeMap.
 200   typedef SmallPtrSet<const OpaqueType*, 8> OpaqueTypesTy;
 201   OpaqueTypesTy OpaqueTypes;
 202
 203   /// Used as an abstract type that will never be resolved.
 204   OpaqueType *const AlwaysOpaqueTy;
 205
 206
 207   /// ValueHandles - This map keeps track of all of the value handles that are
 208   /// watching a Value*.  The Value::HasValueHandle bit is used to know
 209   // whether or not a value has an entry in this map.
 210   typedef DenseMap<Value*, ValueHandleBase*> ValueHandlesTy;
 211   ValueHandlesTy ValueHandles;
 212
 213   /// CustomMDKindNames - Map to hold the metadata string to ID mapping.
 214   StringMap<unsigned> CustomMDKindNames;
 215
 216   typedef std::pair<unsigned, TrackingVH<MDNode> > MDPairTy;
 217   typedef SmallVector<MDPairTy, 2> MDMapTy;
 218
 219   /// MetadataStore - Collection of per-instruction metadata used in this
 220   /// context.
 221   DenseMap<const Instruction *, MDMapTy> MetadataStore;
 222
 223   /// ScopeRecordIdx - This is the index in ScopeRecords for an MDNode scope
 224   /// entry with no "inlined at" element.
 225   DenseMap<MDNode*, int> ScopeRecordIdx;
 226
 227   /// ScopeRecords - These are the actual mdnodes (in a value handle) for an
 228   /// index.  The ValueHandle ensures that ScopeRecordIdx stays up to date if
 229   /// the MDNode is RAUW'd.
 230   std::vector<DebugRecVH> ScopeRecords;
 231
 232   /// ScopeInlinedAtIdx - This is the index in ScopeInlinedAtRecords for an
 233   /// scope/inlined-at pair.
 234   DenseMap<std::pair<MDNode*, MDNode*>, int> ScopeInlinedAtIdx;
 235
 236   /// ScopeInlinedAtRecords - These are the actual mdnodes (in value handles)
 237   /// for an index.  The ValueHandle ensures that ScopeINlinedAtIdx stays up
 238   /// to date.
 239   std::vector<std::pair<DebugRecVH, DebugRecVH> > ScopeInlinedAtRecords;
 240
 241   int getOrAddScopeRecordIdxEntry(MDNode *N, int ExistingIdx);
 242   int getOrAddScopeInlinedAtIdxEntry(MDNode *Scope, MDNode *IA,int ExistingIdx);
 243
 244   LLVMContextImpl(LLVMContext &C);
 245   ~LLVMContextImpl();
 246 };
 247
 248 }
 249
 250 #endif