include/llvm/iMemory.h

   1 //===-- llvm/iMemory.h - Memory Operator node definitions --------*- C++ -*--=//
   2 //
   3 // This file contains the declarations of all of the memory related operators.
   4 // This includes: malloc, free, alloca, load, store, getfield, putfield
   5 //
   6 //===----------------------------------------------------------------------===//
   7
   8 #ifndef LLVM_IMEMORY_H
   9 #define LLVM_IMEMORY_H
  10
  11 #include "llvm/Instruction.h"
  12 #include "llvm/DerivedTypes.h"
  13
  14 //===----------------------------------------------------------------------===//
  15 //                             AllocationInst Class
  16 //===----------------------------------------------------------------------===//
  17 //
  18 // AllocationInst - This class is the common base class of MallocInst and
  19 // AllocaInst.
  20 //
  21 class AllocationInst : public Instruction {
  22 public:
  23   AllocationInst(const Type *Ty, Value *ArraySize, unsigned iTy,
  24                  const string &Name = "")
  25     : Instruction(Ty, iTy, Name) {
  26     assert(Ty->isPointerType() && "Can't allocate a non pointer type!");
  27
  28     if (ArraySize) {
  29       // Make sure they didn't try to specify a size for !(unsized array) type
  30       assert((getType()->getValueType()->isArrayType() &&
  31               ((const ArrayType*)getType()->getValueType())->isUnsized()) &&
  32           "Trying to allocate something other than unsized array, with size!");
  33
  34       Operands.reserve(1);
  35       Operands.push_back(Use(ArraySize, this));
  36     }
  37   }
  38
  39   // getType - Overload to return most specific pointer type...
  40   inline const PointerType *getType() const {
  41     return (const PointerType*)Instruction::getType();
  42   }
  43
  44   virtual Instruction *clone() const = 0;
  45 };
  46
  47
  48 //===----------------------------------------------------------------------===//
  49 //                                MallocInst Class
  50 //===----------------------------------------------------------------------===//
  51
  52 class MallocInst : public AllocationInst {
  53 public:
  54   MallocInst(const Type *Ty, Value *ArraySize = 0, const string &Name = "")
  55     : AllocationInst(Ty, ArraySize, Malloc, Name) {}
  56
  57   virtual Instruction *clone() const {
  58     return new MallocInst(getType(), Operands.size() ? Operands[1] : 0);
  59   }
  60
  61   virtual const char *getOpcodeName() const { return "malloc"; }
  62 };
  63
  64
  65 //===----------------------------------------------------------------------===//
  66 //                                AllocaInst Class
  67 //===----------------------------------------------------------------------===//
  68
  69 class AllocaInst : public AllocationInst {
  70 public:
  71   AllocaInst(const Type *Ty, Value *ArraySize = 0, const string &Name = "")
  72     : AllocationInst(Ty, ArraySize, Alloca, Name) {}
  73
  74   virtual Instruction *clone() const {
  75     return new AllocaInst(getType(), Operands.size() ? Operands[1] : 0);
  76   }
  77
  78   virtual const char *getOpcodeName() const { return "alloca"; }
  79 };
  80
  81
  82 //===----------------------------------------------------------------------===//
  83 //                                 FreeInst Class
  84 //===----------------------------------------------------------------------===//
  85
  86 class FreeInst : public Instruction {
  87 public:
  88   FreeInst(Value *Ptr, const string &Name = "")
  89     : Instruction(Type::VoidTy, Free, Name) {
  90     assert(Ptr->getType()->isPointerType() && "Can't free nonpointer!");
  91     Operands.reserve(1);
  92     Operands.push_back(Use(Ptr, this));
  93   }
  94
  95   virtual Instruction *clone() const { return new FreeInst(Operands[0]); }
  96
  97   virtual const char *getOpcodeName() const { return "free"; }
  98 };
  99
 100
 101 //===----------------------------------------------------------------------===//
 102 //                              MemAccessInst Class
 103 //===----------------------------------------------------------------------===//
 104 //
 105 // MemAccessInst - Common base class of LoadInst, StoreInst, and
 106 // GetElementPtrInst...
 107 //
 108 class MemAccessInst : public Instruction {
 109 protected:
 110   inline MemAccessInst(const Type *Ty, unsigned Opcode, const string &Nam = "")
 111     : Instruction(Ty, Opcode, Nam) {}
 112 public:
 113   // getIndexedType - Returns the type of the element that would be loaded with
 114   // a load instruction with the specified parameters.
 115   //
 116   // A null type is returned if the indices are invalid for the specified
 117   // pointer type.
 118   //
 119   static const Type *getIndexedType(const Type *Ptr,
 120                                     const vector<ConstPoolVal*> &Indices,
 121                                     bool AllowStructLeaf = false);
 122 };
 123
 124
 125 //===----------------------------------------------------------------------===//
 126 //                                LoadInst Class
 127 //===----------------------------------------------------------------------===//
 128
 129 class LoadInst : public MemAccessInst {
 130   LoadInst(const LoadInst &LI) : MemAccessInst(LI.getType(), Load) {
 131     Operands.reserve(LI.Operands.size());
 132     for (unsigned i = 0, E = LI.Operands.size(); i != E; ++i)
 133       Operands.push_back(Use(LI.Operands[i], this));
 134   }
 135 public:
 136   LoadInst(Value *Ptr, const vector<ConstPoolVal*> &Idx,
 137            const string &Name = "");
 138   virtual Instruction *clone() const { return new LoadInst(*this); }
 139   virtual const char *getOpcodeName() const { return "load"; }
 140 };
 141
 142
 143 //===----------------------------------------------------------------------===//
 144 //                                StoreInst Class
 145 //===----------------------------------------------------------------------===//
 146
 147 class StoreInst : public MemAccessInst {
 148   StoreInst(const StoreInst &SI) : MemAccessInst(SI.getType(), Store) {
 149     Operands.reserve(SI.Operands.size());
 150     for (unsigned i = 0, E = SI.Operands.size(); i != E; ++i)
 151       Operands.push_back(Use(SI.Operands[i], this));
 152   }
 153 public:
 154   StoreInst(Value *Val, Value *Ptr, const vector<ConstPoolVal*> &Idx,
 155             const string &Name = "");
 156   virtual Instruction *clone() const { return new StoreInst(*this); }
 157   virtual const char *getOpcodeName() const { return "store"; }
 158 };
 159
 160
 161 //===----------------------------------------------------------------------===//
 162 //                             GetElementPtrInst Class
 163 //===----------------------------------------------------------------------===//
 164
 165 class GetElementPtrInst : public MemAccessInst {
 166   GetElementPtrInst(const GetElementPtrInst &EPI)
 167     : MemAccessInst(EPI.getType(), GetElementPtr) {
 168     Operands.reserve(EPI.Operands.size());
 169     for (unsigned i = 0, E = EPI.Operands.size(); i != E; ++i)
 170       Operands.push_back(Use(EPI.Operands[i], this));
 171   }
 172 public:
 173   GetElementPtrInst(Value *Ptr, const vector<ConstPoolVal*> &Idx,
 174                     const string &Name = "");
 175   virtual Instruction *clone() const { return new GetElementPtrInst(*this); }
 176   virtual const char *getOpcodeName() const { return "getelementptr"; }
 177 };
 178
 179 #endif // LLVM_IMEMORY_H