include/llvm/Support/TargetFolder.h

   1 //====-- llvm/Support/TargetFolder.h - Constant folding helper -*- 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 defines the TargetFolder class, a helper for IRBuilder.
  11 // It provides IRBuilder with a set of methods for creating constants with
  12 // target dependent folding, in addition to the same target-independent
  13 // folding that the ConstantFolder class provides.  For general constant
  14 // creation and folding, use ConstantExpr and the routines in
  15 // llvm/Analysis/ConstantFolding.h.
  16 //
  17 //===----------------------------------------------------------------------===//
  18
  19 #ifndef LLVM_SUPPORT_TARGETFOLDER_H
  20 #define LLVM_SUPPORT_TARGETFOLDER_H
  21
  22 #include "llvm/Constants.h"
  23 #include "llvm/InstrTypes.h"
  24 #include "llvm/Analysis/ConstantFolding.h"
  25
  26 namespace llvm {
  27
  28 class TargetData;
  29
  30 /// TargetFolder - Create constants with target dependent folding.
  31 class TargetFolder {
  32   const TargetData *TD;
  33
  34   /// Fold - Fold the constant using target specific information.
  35   Constant *Fold(Constant *C) const {
  36     if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C))
  37       if (Constant *CF = ConstantFoldConstantExpression(CE, TD))
  38         return CF;
  39     return C;
  40   }
  41
  42 public:
  43   explicit TargetFolder(const TargetData *TheTD) : TD(TheTD) {}
  44
  45   //===--------------------------------------------------------------------===//
  46   // Binary Operators
  47   //===--------------------------------------------------------------------===//
  48
  49   Constant *CreateAdd(Constant *LHS, Constant *RHS) const {
  50     return Fold(ConstantExpr::getAdd(LHS, RHS));
  51   }
  52   Constant *CreateNSWAdd(Constant *LHS, Constant *RHS) const {
  53     return Fold(ConstantExpr::getNSWAdd(LHS, RHS));
  54   }
  55   Constant *CreateNUWAdd(Constant *LHS, Constant *RHS) const {
  56     return Fold(ConstantExpr::getNUWAdd(LHS, RHS));
  57   }
  58   Constant *CreateFAdd(Constant *LHS, Constant *RHS) const {
  59     return Fold(ConstantExpr::getFAdd(LHS, RHS));
  60   }
  61   Constant *CreateSub(Constant *LHS, Constant *RHS) const {
  62     return Fold(ConstantExpr::getSub(LHS, RHS));
  63   }
  64   Constant *CreateNSWSub(Constant *LHS, Constant *RHS) const {
  65     return Fold(ConstantExpr::getNSWSub(LHS, RHS));
  66   }
  67   Constant *CreateNUWSub(Constant *LHS, Constant *RHS) const {
  68     return Fold(ConstantExpr::getNUWSub(LHS, RHS));
  69   }
  70   Constant *CreateFSub(Constant *LHS, Constant *RHS) const {
  71     return Fold(ConstantExpr::getFSub(LHS, RHS));
  72   }
  73   Constant *CreateMul(Constant *LHS, Constant *RHS) const {
  74     return Fold(ConstantExpr::getMul(LHS, RHS));
  75   }
  76   Constant *CreateNSWMul(Constant *LHS, Constant *RHS) const {
  77     return Fold(ConstantExpr::getNSWMul(LHS, RHS));
  78   }
  79   Constant *CreateNUWMul(Constant *LHS, Constant *RHS) const {
  80     return Fold(ConstantExpr::getNUWMul(LHS, RHS));
  81   }
  82   Constant *CreateFMul(Constant *LHS, Constant *RHS) const {
  83     return Fold(ConstantExpr::getFMul(LHS, RHS));
  84   }
  85   Constant *CreateUDiv(Constant *LHS, Constant *RHS) const {
  86     return Fold(ConstantExpr::getUDiv(LHS, RHS));
  87   }
  88   Constant *CreateExactUDiv(Constant *LHS, Constant *RHS) const {
  89     return Fold(ConstantExpr::getExactUDiv(LHS, RHS));
  90   }
  91   Constant *CreateSDiv(Constant *LHS, Constant *RHS) const {
  92     return Fold(ConstantExpr::getSDiv(LHS, RHS));
  93   }
  94   Constant *CreateExactSDiv(Constant *LHS, Constant *RHS) const {
  95     return Fold(ConstantExpr::getExactSDiv(LHS, RHS));
  96   }
  97   Constant *CreateFDiv(Constant *LHS, Constant *RHS) const {
  98     return Fold(ConstantExpr::getFDiv(LHS, RHS));
  99   }
 100   Constant *CreateURem(Constant *LHS, Constant *RHS) const {
 101     return Fold(ConstantExpr::getURem(LHS, RHS));
 102   }
 103   Constant *CreateSRem(Constant *LHS, Constant *RHS) const {
 104     return Fold(ConstantExpr::getSRem(LHS, RHS));
 105   }
 106   Constant *CreateFRem(Constant *LHS, Constant *RHS) const {
 107     return Fold(ConstantExpr::getFRem(LHS, RHS));
 108   }
 109   Constant *CreateShl(Constant *LHS, Constant *RHS) const {
 110     return Fold(ConstantExpr::getShl(LHS, RHS));
 111   }
 112   Constant *CreateLShr(Constant *LHS, Constant *RHS) const {
 113     return Fold(ConstantExpr::getLShr(LHS, RHS));
 114   }
 115   Constant *CreateAShr(Constant *LHS, Constant *RHS) const {
 116     return Fold(ConstantExpr::getAShr(LHS, RHS));
 117   }
 118   Constant *CreateAnd(Constant *LHS, Constant *RHS) const {
 119     return Fold(ConstantExpr::getAnd(LHS, RHS));
 120   }
 121   Constant *CreateOr(Constant *LHS, Constant *RHS) const {
 122     return Fold(ConstantExpr::getOr(LHS, RHS));
 123   }
 124   Constant *CreateXor(Constant *LHS, Constant *RHS) const {
 125     return Fold(ConstantExpr::getXor(LHS, RHS));
 126   }
 127
 128   Constant *CreateBinOp(Instruction::BinaryOps Opc,
 129                         Constant *LHS, Constant *RHS) const {
 130     return Fold(ConstantExpr::get(Opc, LHS, RHS));
 131   }
 132
 133   //===--------------------------------------------------------------------===//
 134   // Unary Operators
 135   //===--------------------------------------------------------------------===//
 136
 137   Constant *CreateNeg(Constant *C) const {
 138     return Fold(ConstantExpr::getNeg(C));
 139   }
 140   Constant *CreateNSWNeg(Constant *C) const {
 141     return Fold(ConstantExpr::getNSWNeg(C));
 142   }
 143   Constant *CreateNUWNeg(Constant *C) const {
 144     return Fold(ConstantExpr::getNUWNeg(C));
 145   }
 146   Constant *CreateFNeg(Constant *C) const {
 147     return Fold(ConstantExpr::getFNeg(C));
 148   }
 149   Constant *CreateNot(Constant *C) const {
 150     return Fold(ConstantExpr::getNot(C));
 151   }
 152
 153   //===--------------------------------------------------------------------===//
 154   // Memory Instructions
 155   //===--------------------------------------------------------------------===//
 156
 157   Constant *CreateGetElementPtr(Constant *C, Constant* const *IdxList,
 158                                 unsigned NumIdx) const {
 159     return Fold(ConstantExpr::getGetElementPtr(C, IdxList, NumIdx));
 160   }
 161   Constant *CreateGetElementPtr(Constant *C, Value* const *IdxList,
 162                                 unsigned NumIdx) const {
 163     return Fold(ConstantExpr::getGetElementPtr(C, IdxList, NumIdx));
 164   }
 165
 166   Constant *CreateInBoundsGetElementPtr(Constant *C, Constant* const *IdxList,
 167                                         unsigned NumIdx) const {
 168     return Fold(ConstantExpr::getInBoundsGetElementPtr(C, IdxList, NumIdx));
 169   }
 170   Constant *CreateInBoundsGetElementPtr(Constant *C, Value* const *IdxList,
 171                                         unsigned NumIdx) const {
 172     return Fold(ConstantExpr::getInBoundsGetElementPtr(C, IdxList, NumIdx));
 173   }
 174
 175   //===--------------------------------------------------------------------===//
 176   // Cast/Conversion Operators
 177   //===--------------------------------------------------------------------===//
 178
 179   Constant *CreateCast(Instruction::CastOps Op, Constant *C,
 180                        const Type *DestTy) const {
 181     if (C->getType() == DestTy)
 182       return C; // avoid calling Fold
 183     return Fold(ConstantExpr::getCast(Op, C, DestTy));
 184   }
 185   Constant *CreateIntCast(Constant *C, const Type *DestTy,
 186                           bool isSigned) const {
 187     if (C->getType() == DestTy)
 188       return C; // avoid calling Fold
 189     return Fold(ConstantExpr::getIntegerCast(C, DestTy, isSigned));
 190   }
 191   Constant *CreatePointerCast(Constant *C, const Type *DestTy) const {
 192     return ConstantExpr::getPointerCast(C, DestTy);
 193   }
 194   Constant *CreateBitCast(Constant *C, const Type *DestTy) const {
 195     return CreateCast(Instruction::BitCast, C, DestTy);
 196   }
 197   Constant *CreateIntToPtr(Constant *C, const Type *DestTy) const {
 198     return CreateCast(Instruction::IntToPtr, C, DestTy);
 199   }
 200   Constant *CreatePtrToInt(Constant *C, const Type *DestTy) const {
 201     return CreateCast(Instruction::PtrToInt, C, DestTy);
 202   }
 203   Constant *CreateZExtOrBitCast(Constant *C, const Type *DestTy) const {
 204     if (C->getType() == DestTy)
 205       return C; // avoid calling Fold
 206     return Fold(ConstantExpr::getZExtOrBitCast(C, DestTy));
 207   }
 208   Constant *CreateSExtOrBitCast(Constant *C, const Type *DestTy) const {
 209     if (C->getType() == DestTy)
 210       return C; // avoid calling Fold
 211     return Fold(ConstantExpr::getSExtOrBitCast(C, DestTy));
 212   }
 213   Constant *CreateTruncOrBitCast(Constant *C, const Type *DestTy) const {
 214     if (C->getType() == DestTy)
 215       return C; // avoid calling Fold
 216     return Fold(ConstantExpr::getTruncOrBitCast(C, DestTy));
 217   }
 218
 219   //===--------------------------------------------------------------------===//
 220   // Compare Instructions
 221   //===--------------------------------------------------------------------===//
 222
 223   Constant *CreateICmp(CmpInst::Predicate P, Constant *LHS,
 224                        Constant *RHS) const {
 225     return Fold(ConstantExpr::getCompare(P, LHS, RHS));
 226   }
 227   Constant *CreateFCmp(CmpInst::Predicate P, Constant *LHS,
 228                        Constant *RHS) const {
 229     return Fold(ConstantExpr::getCompare(P, LHS, RHS));
 230   }
 231
 232   //===--------------------------------------------------------------------===//
 233   // Other Instructions
 234   //===--------------------------------------------------------------------===//
 235
 236   Constant *CreateSelect(Constant *C, Constant *True, Constant *False) const {
 237     return Fold(ConstantExpr::getSelect(C, True, False));
 238   }
 239
 240   Constant *CreateExtractElement(Constant *Vec, Constant *Idx) const {
 241     return Fold(ConstantExpr::getExtractElement(Vec, Idx));
 242   }
 243
 244   Constant *CreateInsertElement(Constant *Vec, Constant *NewElt,
 245                                 Constant *Idx) const {
 246     return Fold(ConstantExpr::getInsertElement(Vec, NewElt, Idx));
 247   }
 248
 249   Constant *CreateShuffleVector(Constant *V1, Constant *V2,
 250                                 Constant *Mask) const {
 251     return Fold(ConstantExpr::getShuffleVector(V1, V2, Mask));
 252   }
 253
 254   Constant *CreateExtractValue(Constant *Agg, const unsigned *IdxList,
 255                                unsigned NumIdx) const {
 256     return Fold(ConstantExpr::getExtractValue(Agg, IdxList, NumIdx));
 257   }
 258
 259   Constant *CreateInsertValue(Constant *Agg, Constant *Val,
 260                               const unsigned *IdxList, unsigned NumIdx) const {
 261     return Fold(ConstantExpr::getInsertValue(Agg, Val, IdxList, NumIdx));
 262   }
 263 };
 264
 265 }
 266
 267 #endif