blob: 85a89de145a1643d5c4f4d9e955c992ec02c7977 [file] [log] [blame]
Chris Lattner2f7c9632001-06-06 20:29:01 +00001//===- ConstantHandling.cpp - Implement ConstantHandling.h ----------------===//
2//
3// This file implements the various intrinsic operations, on constant values.
4//
5//===----------------------------------------------------------------------===//
6
Chris Lattner65b529f2002-04-08 20:18:09 +00007#include "llvm/ConstantHandling.h"
Chris Lattner78a0d422002-05-07 20:44:59 +00008#include "llvm/iPHINode.h"
Chris Lattner8acf3462003-05-27 19:16:07 +00009#include "llvm/InstrTypes.h"
Chris Lattner1f0049c2003-04-17 19:24:18 +000010#include "llvm/DerivedTypes.h"
Chris Lattner0a144ad2002-05-03 21:41:07 +000011#include <cmath>
Chris Lattnerd42d4922001-06-30 04:36:40 +000012
Chris Lattner61607ee2001-09-09 21:01:20 +000013AnnotationID ConstRules::AID(AnnotationManager::getID("opt::ConstRules",
14 &ConstRules::find));
15
Chris Lattner9f307732002-05-06 17:54:27 +000016// ConstantFoldInstruction - Attempt to constant fold the specified instruction.
17// If successful, the constant result is returned, if not, null is returned.
18//
19Constant *ConstantFoldInstruction(Instruction *I) {
Chris Lattner78a0d422002-05-07 20:44:59 +000020 if (PHINode *PN = dyn_cast<PHINode>(I)) {
21 if (PN->getNumIncomingValues() == 0)
22 return Constant::getNullValue(PN->getType());
23
24 Constant *Result = dyn_cast<Constant>(PN->getIncomingValue(0));
25 if (Result == 0) return 0;
26
27 // Handle PHI nodes specially here...
28 for (unsigned i = 1, e = PN->getNumIncomingValues(); i != e; ++i)
29 if (PN->getIncomingValue(i) != Result)
30 return 0; // Not all the same incoming constants...
31
32 // If we reach here, all incoming values are the same constant.
33 return Result;
34 }
35
Chris Lattner9f307732002-05-06 17:54:27 +000036 Constant *Op0 = 0;
37 Constant *Op1 = 0;
38
39 if (I->getNumOperands() != 0) { // Get first operand if it's a constant...
40 Op0 = dyn_cast<Constant>(I->getOperand(0));
41 if (Op0 == 0) return 0; // Not a constant?, can't fold
42
43 if (I->getNumOperands() != 1) { // Get second operand if it's a constant...
44 Op1 = dyn_cast<Constant>(I->getOperand(1));
45 if (Op1 == 0) return 0; // Not a constant?, can't fold
46 }
47 }
48
Chris Lattner8acf3462003-05-27 19:16:07 +000049 if (isa<BinaryOperator>(I))
50 return ConstantExpr::get(I->getOpcode(), Op0, Op1);
51
Chris Lattner9f307732002-05-06 17:54:27 +000052 switch (I->getOpcode()) {
53 case Instruction::Cast:
Chris Lattner8acf3462003-05-27 19:16:07 +000054 return ConstantExpr::getCast(Op0, I->getType());
55 case Instruction::Shl:
56 case Instruction::Shr:
57 return ConstantExpr::getShift(I->getOpcode(), Op0, Op1);
Chris Lattner1f0049c2003-04-17 19:24:18 +000058 case Instruction::GetElementPtr: {
59 std::vector<Constant*> IdxList;
60 IdxList.reserve(I->getNumOperands()-1);
61 if (Op1) IdxList.push_back(Op1);
62 for (unsigned i = 2, e = I->getNumOperands(); i != e; ++i)
63 if (Constant *C = dyn_cast<Constant>(I->getOperand(i)))
64 IdxList.push_back(C);
65 else
66 return 0; // Non-constant operand
Chris Lattner8acf3462003-05-27 19:16:07 +000067 return ConstantExpr::getGetElementPtr(Op0, IdxList);
Chris Lattner1f0049c2003-04-17 19:24:18 +000068 }
Chris Lattner9f307732002-05-06 17:54:27 +000069 default:
70 return 0;
71 }
72}
73
Chris Lattner1f0049c2003-04-17 19:24:18 +000074static unsigned getSize(const Type *Ty) {
75 unsigned S = Ty->getPrimitiveSize();
76 return S ? S : 8; // Treat pointers at 8 bytes
77}
78
Chris Lattner9f307732002-05-06 17:54:27 +000079Constant *ConstantFoldCastInstruction(const Constant *V, const Type *DestTy) {
Chris Lattner1f0049c2003-04-17 19:24:18 +000080 if (V->getType() == DestTy) return (Constant*)V;
81
82 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(V))
83 if (CE->getOpcode() == Instruction::Cast) {
Chris Lattner55ed6562003-05-14 17:51:05 +000084 Constant *Op = const_cast<Constant*>(CE->getOperand(0));
Chris Lattner1f0049c2003-04-17 19:24:18 +000085 // Try to not produce a cast of a cast, which is almost always redundant.
86 if (!Op->getType()->isFloatingPoint() &&
87 !CE->getType()->isFloatingPoint() &&
88 !DestTy->getType()->isFloatingPoint()) {
89 unsigned S1 = getSize(Op->getType()), S2 = getSize(CE->getType());
90 unsigned S3 = getSize(DestTy);
91 if (Op->getType() == DestTy && S3 >= S2)
92 return Op;
93 if (S1 >= S2 && S2 >= S3)
94 return ConstantExpr::getCast(Op, DestTy);
95 if (S1 <= S2 && S2 >= S3 && S1 <= S3)
96 return ConstantExpr::getCast(Op, DestTy);
97 }
98 }
99
Chris Lattnere17c1c5a2003-04-25 02:52:06 +0000100 return ConstRules::get(*V, *V)->castTo(V, DestTy);
Chris Lattner9f307732002-05-06 17:54:27 +0000101}
102
Chris Lattner9f307732002-05-06 17:54:27 +0000103Constant *ConstantFoldBinaryInstruction(unsigned Opcode, const Constant *V1,
104 const Constant *V2) {
105 switch (Opcode) {
106 case Instruction::Add: return *V1 + *V2;
107 case Instruction::Sub: return *V1 - *V2;
108 case Instruction::Mul: return *V1 * *V2;
109 case Instruction::Div: return *V1 / *V2;
110 case Instruction::Rem: return *V1 % *V2;
Chris Lattnere87f65e2002-07-30 16:24:28 +0000111 case Instruction::And: return *V1 & *V2;
112 case Instruction::Or: return *V1 | *V2;
113 case Instruction::Xor: return *V1 ^ *V2;
Chris Lattner9f307732002-05-06 17:54:27 +0000114
115 case Instruction::SetEQ: return *V1 == *V2;
116 case Instruction::SetNE: return *V1 != *V2;
117 case Instruction::SetLE: return *V1 <= *V2;
118 case Instruction::SetGE: return *V1 >= *V2;
119 case Instruction::SetLT: return *V1 < *V2;
120 case Instruction::SetGT: return *V1 > *V2;
121 }
122 return 0;
123}
124
125Constant *ConstantFoldShiftInstruction(unsigned Opcode, const Constant *V1,
126 const Constant *V2) {
127 switch (Opcode) {
128 case Instruction::Shl: return *V1 << *V2;
129 case Instruction::Shr: return *V1 >> *V2;
130 default: return 0;
131 }
132}
133
Chris Lattner1f0049c2003-04-17 19:24:18 +0000134Constant *ConstantFoldGetElementPtr(const Constant *C,
135 const std::vector<Constant*> &IdxList) {
136 if (IdxList.size() == 0 ||
137 (IdxList.size() == 1 && IdxList[0]->isNullValue()))
138 return const_cast<Constant*>(C);
139
140 // If C is null and all idx's are null, return null of the right type.
141
142 // FIXME: Implement folding of GEP constant exprs the same as instcombine does
Chris Lattnerc2ceed12003-05-13 21:50:52 +0000143
Chris Lattner4ede64e2003-06-26 05:22:45 +0000144 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
Chris Lattnerddab2392003-08-20 16:11:27 +0000145 // Combine Indices - If the source pointer to this getelementptr instruction
146 // is a getelementptr instruction, combine the indices of the two
147 // getelementptr instructions into a single instruction.
Chris Lattner4ede64e2003-06-26 05:22:45 +0000148 //
Chris Lattnerddab2392003-08-20 16:11:27 +0000149 if (CE->getOpcode() == Instruction::GetElementPtr) {
150 if (CE->getOperand(CE->getNumOperands()-1)->getType() == Type::LongTy) {
Chris Lattner4ede64e2003-06-26 05:22:45 +0000151 std::vector<Constant*> NewIndices;
152 NewIndices.reserve(IdxList.size() + CE->getNumOperands());
Chris Lattnerddab2392003-08-20 16:11:27 +0000153 for (unsigned i = 1, e = CE->getNumOperands()-1; i != e; ++i)
Chris Lattner4ede64e2003-06-26 05:22:45 +0000154 NewIndices.push_back(cast<Constant>(CE->getOperand(i)));
Chris Lattnerddab2392003-08-20 16:11:27 +0000155
156 // Add the last index of the source with the first index of the new GEP.
157 Constant *Combined =
158 ConstantExpr::get(Instruction::Add, IdxList[0],
159 CE->getOperand(CE->getNumOperands()-1));
160
161 NewIndices.push_back(Combined);
Chris Lattner4ede64e2003-06-26 05:22:45 +0000162 NewIndices.insert(NewIndices.end(), IdxList.begin()+1, IdxList.end());
163 return ConstantExpr::getGetElementPtr(CE->getOperand(0), NewIndices);
164 }
Chris Lattnerddab2392003-08-20 16:11:27 +0000165 }
Chris Lattner4ede64e2003-06-26 05:22:45 +0000166
167 // Implement folding of:
168 // int* getelementptr ([2 x int]* cast ([3 x int]* %X to [2 x int]*),
169 // long 0, long 0)
170 // To: int* getelementptr ([3 x int]* %X, long 0, long 0)
171 //
Chris Lattner69f6af12003-05-14 02:47:13 +0000172 if (CE->getOpcode() == Instruction::Cast && IdxList.size() > 1 &&
173 IdxList[0]->isNullValue())
Chris Lattnerc2ceed12003-05-13 21:50:52 +0000174 if (const PointerType *SPT =
175 dyn_cast<PointerType>(CE->getOperand(0)->getType()))
176 if (const ArrayType *SAT = dyn_cast<ArrayType>(SPT->getElementType()))
177 if (const ArrayType *CAT =
178 dyn_cast<ArrayType>(cast<PointerType>(C->getType())->getElementType()))
179 if (CAT->getElementType() == SAT->getElementType())
180 return ConstantExpr::getGetElementPtr(
Chris Lattner55ed6562003-05-14 17:51:05 +0000181 (Constant*)CE->getOperand(0), IdxList);
Chris Lattner4ede64e2003-06-26 05:22:45 +0000182 }
Chris Lattner1f0049c2003-04-17 19:24:18 +0000183 return 0;
184}
185
Chris Lattner9f307732002-05-06 17:54:27 +0000186
Chris Lattner2f7c9632001-06-06 20:29:01 +0000187//===----------------------------------------------------------------------===//
188// TemplateRules Class
189//===----------------------------------------------------------------------===//
190//
191// TemplateRules - Implement a subclass of ConstRules that provides all
192// operations as noops. All other rules classes inherit from this class so
193// that if functionality is needed in the future, it can simply be added here
194// and to ConstRules without changing anything else...
195//
196// This class also provides subclasses with typesafe implementations of methods
197// so that don't have to do type casting.
198//
199template<class ArgType, class SubClassName>
200class TemplateRules : public ConstRules {
201
202 //===--------------------------------------------------------------------===//
203 // Redirecting functions that cast to the appropriate types
204 //===--------------------------------------------------------------------===//
205
Chris Lattnere87f65e2002-07-30 16:24:28 +0000206 virtual Constant *add(const Constant *V1, const Constant *V2) const {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000207 return SubClassName::Add((const ArgType *)V1, (const ArgType *)V2);
208 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000209 virtual Constant *sub(const Constant *V1, const Constant *V2) const {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000210 return SubClassName::Sub((const ArgType *)V1, (const ArgType *)V2);
211 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000212 virtual Constant *mul(const Constant *V1, const Constant *V2) const {
Chris Lattner4f6031f2001-07-20 19:15:36 +0000213 return SubClassName::Mul((const ArgType *)V1, (const ArgType *)V2);
214 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000215 virtual Constant *div(const Constant *V1, const Constant *V2) const {
Chris Lattneraf259a72002-04-07 08:10:14 +0000216 return SubClassName::Div((const ArgType *)V1, (const ArgType *)V2);
217 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000218 virtual Constant *rem(const Constant *V1, const Constant *V2) const {
Chris Lattner0a144ad2002-05-03 21:41:07 +0000219 return SubClassName::Rem((const ArgType *)V1, (const ArgType *)V2);
220 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000221 virtual Constant *op_and(const Constant *V1, const Constant *V2) const {
222 return SubClassName::And((const ArgType *)V1, (const ArgType *)V2);
223 }
224 virtual Constant *op_or(const Constant *V1, const Constant *V2) const {
225 return SubClassName::Or((const ArgType *)V1, (const ArgType *)V2);
226 }
227 virtual Constant *op_xor(const Constant *V1, const Constant *V2) const {
228 return SubClassName::Xor((const ArgType *)V1, (const ArgType *)V2);
229 }
230 virtual Constant *shl(const Constant *V1, const Constant *V2) const {
Chris Lattner6670d862002-05-06 03:00:54 +0000231 return SubClassName::Shl((const ArgType *)V1, (const ArgType *)V2);
232 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000233 virtual Constant *shr(const Constant *V1, const Constant *V2) const {
Chris Lattner6670d862002-05-06 03:00:54 +0000234 return SubClassName::Shr((const ArgType *)V1, (const ArgType *)V2);
235 }
Chris Lattner4f6031f2001-07-20 19:15:36 +0000236
Chris Lattner3462ae32001-12-03 22:26:30 +0000237 virtual ConstantBool *lessthan(const Constant *V1,
238 const Constant *V2) const {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000239 return SubClassName::LessThan((const ArgType *)V1, (const ArgType *)V2);
240 }
241
Chris Lattner55406842001-07-21 19:10:49 +0000242 // Casting operators. ick
Chris Lattner3462ae32001-12-03 22:26:30 +0000243 virtual ConstantBool *castToBool(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000244 return SubClassName::CastToBool((const ArgType*)V);
245 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000246 virtual ConstantSInt *castToSByte(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000247 return SubClassName::CastToSByte((const ArgType*)V);
248 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000249 virtual ConstantUInt *castToUByte(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000250 return SubClassName::CastToUByte((const ArgType*)V);
251 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000252 virtual ConstantSInt *castToShort(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000253 return SubClassName::CastToShort((const ArgType*)V);
254 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000255 virtual ConstantUInt *castToUShort(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000256 return SubClassName::CastToUShort((const ArgType*)V);
257 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000258 virtual ConstantSInt *castToInt(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000259 return SubClassName::CastToInt((const ArgType*)V);
260 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000261 virtual ConstantUInt *castToUInt(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000262 return SubClassName::CastToUInt((const ArgType*)V);
263 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000264 virtual ConstantSInt *castToLong(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000265 return SubClassName::CastToLong((const ArgType*)V);
266 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000267 virtual ConstantUInt *castToULong(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000268 return SubClassName::CastToULong((const ArgType*)V);
269 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000270 virtual ConstantFP *castToFloat(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000271 return SubClassName::CastToFloat((const ArgType*)V);
272 }
Chris Lattner3462ae32001-12-03 22:26:30 +0000273 virtual ConstantFP *castToDouble(const Constant *V) const {
Chris Lattner55406842001-07-21 19:10:49 +0000274 return SubClassName::CastToDouble((const ArgType*)V);
275 }
Chris Lattner1f0049c2003-04-17 19:24:18 +0000276 virtual Constant *castToPointer(const Constant *V,
277 const PointerType *Ty) const {
Chris Lattner977f0042001-11-01 05:55:13 +0000278 return SubClassName::CastToPointer((const ArgType*)V, Ty);
279 }
Chris Lattner55406842001-07-21 19:10:49 +0000280
Chris Lattner2f7c9632001-06-06 20:29:01 +0000281 //===--------------------------------------------------------------------===//
282 // Default "noop" implementations
283 //===--------------------------------------------------------------------===//
284
Chris Lattnere87f65e2002-07-30 16:24:28 +0000285 static Constant *Add(const ArgType *V1, const ArgType *V2) { return 0; }
286 static Constant *Sub(const ArgType *V1, const ArgType *V2) { return 0; }
287 static Constant *Mul(const ArgType *V1, const ArgType *V2) { return 0; }
288 static Constant *Div(const ArgType *V1, const ArgType *V2) { return 0; }
289 static Constant *Rem(const ArgType *V1, const ArgType *V2) { return 0; }
290 static Constant *And(const ArgType *V1, const ArgType *V2) { return 0; }
291 static Constant *Or (const ArgType *V1, const ArgType *V2) { return 0; }
292 static Constant *Xor(const ArgType *V1, const ArgType *V2) { return 0; }
293 static Constant *Shl(const ArgType *V1, const ArgType *V2) { return 0; }
294 static Constant *Shr(const ArgType *V1, const ArgType *V2) { return 0; }
295 static ConstantBool *LessThan(const ArgType *V1, const ArgType *V2) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000296 return 0;
297 }
Chris Lattner55406842001-07-21 19:10:49 +0000298
299 // Casting operators. ick
Chris Lattnere87f65e2002-07-30 16:24:28 +0000300 static ConstantBool *CastToBool (const Constant *V) { return 0; }
301 static ConstantSInt *CastToSByte (const Constant *V) { return 0; }
302 static ConstantUInt *CastToUByte (const Constant *V) { return 0; }
303 static ConstantSInt *CastToShort (const Constant *V) { return 0; }
304 static ConstantUInt *CastToUShort(const Constant *V) { return 0; }
305 static ConstantSInt *CastToInt (const Constant *V) { return 0; }
306 static ConstantUInt *CastToUInt (const Constant *V) { return 0; }
307 static ConstantSInt *CastToLong (const Constant *V) { return 0; }
308 static ConstantUInt *CastToULong (const Constant *V) { return 0; }
309 static ConstantFP *CastToFloat (const Constant *V) { return 0; }
310 static ConstantFP *CastToDouble(const Constant *V) { return 0; }
Chris Lattner1f0049c2003-04-17 19:24:18 +0000311 static Constant *CastToPointer(const Constant *,
312 const PointerType *) {return 0;}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000313};
314
315
316
317//===----------------------------------------------------------------------===//
318// EmptyRules Class
319//===----------------------------------------------------------------------===//
320//
321// EmptyRules provides a concrete base class of ConstRules that does nothing
322//
Chris Lattner3462ae32001-12-03 22:26:30 +0000323struct EmptyRules : public TemplateRules<Constant, EmptyRules> {
Chris Lattner61607ee2001-09-09 21:01:20 +0000324};
Chris Lattner2f7c9632001-06-06 20:29:01 +0000325
326
327
328//===----------------------------------------------------------------------===//
329// BoolRules Class
330//===----------------------------------------------------------------------===//
331//
332// BoolRules provides a concrete base class of ConstRules for the 'bool' type.
333//
Chris Lattner3462ae32001-12-03 22:26:30 +0000334struct BoolRules : public TemplateRules<ConstantBool, BoolRules> {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000335
Chris Lattner07507a42002-09-03 20:09:49 +0000336 static ConstantBool *LessThan(const ConstantBool *V1, const ConstantBool *V2){
337 return ConstantBool::get(V1->getValue() < V2->getValue());
338 }
339
Chris Lattnere87f65e2002-07-30 16:24:28 +0000340 static Constant *And(const ConstantBool *V1, const ConstantBool *V2) {
341 return ConstantBool::get(V1->getValue() & V2->getValue());
342 }
343
344 static Constant *Or(const ConstantBool *V1, const ConstantBool *V2) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000345 return ConstantBool::get(V1->getValue() | V2->getValue());
Chris Lattner2f7c9632001-06-06 20:29:01 +0000346 }
347
Chris Lattnere87f65e2002-07-30 16:24:28 +0000348 static Constant *Xor(const ConstantBool *V1, const ConstantBool *V2) {
349 return ConstantBool::get(V1->getValue() ^ V2->getValue());
Chris Lattner2f7c9632001-06-06 20:29:01 +0000350 }
Chris Lattnercea4d8c2003-08-13 15:52:25 +0000351
352 // Casting operators. ick
353#define DEF_CAST(TYPE, CLASS, CTYPE) \
354 static CLASS *CastTo##TYPE (const ConstantBool *V) { \
355 return CLASS::get(Type::TYPE##Ty, (CTYPE)(bool)V->getValue()); \
356 }
357
358 DEF_CAST(Bool , ConstantBool, bool)
359 DEF_CAST(SByte , ConstantSInt, signed char)
360 DEF_CAST(UByte , ConstantUInt, unsigned char)
361 DEF_CAST(Short , ConstantSInt, signed short)
362 DEF_CAST(UShort, ConstantUInt, unsigned short)
363 DEF_CAST(Int , ConstantSInt, signed int)
364 DEF_CAST(UInt , ConstantUInt, unsigned int)
365 DEF_CAST(Long , ConstantSInt, int64_t)
366 DEF_CAST(ULong , ConstantUInt, uint64_t)
367 DEF_CAST(Float , ConstantFP , float)
368 DEF_CAST(Double, ConstantFP , double)
369#undef DEF_CAST
Chris Lattner61607ee2001-09-09 21:01:20 +0000370};
Chris Lattner2f7c9632001-06-06 20:29:01 +0000371
372
373//===----------------------------------------------------------------------===//
Chris Lattner977f0042001-11-01 05:55:13 +0000374// PointerRules Class
375//===----------------------------------------------------------------------===//
376//
377// PointerRules provides a concrete base class of ConstRules for pointer types
378//
Chris Lattner3462ae32001-12-03 22:26:30 +0000379struct PointerRules : public TemplateRules<ConstantPointer, PointerRules> {
Chris Lattnere87f65e2002-07-30 16:24:28 +0000380 static ConstantBool *CastToBool (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000381 if (V->isNullValue()) return ConstantBool::False;
Chris Lattner977f0042001-11-01 05:55:13 +0000382 return 0; // Can't const prop other types of pointers
383 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000384 static ConstantSInt *CastToSByte (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000385 if (V->isNullValue()) return ConstantSInt::get(Type::SByteTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000386 return 0; // Can't const prop other types of pointers
387 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000388 static ConstantUInt *CastToUByte (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000389 if (V->isNullValue()) return ConstantUInt::get(Type::UByteTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000390 return 0; // Can't const prop other types of pointers
391 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000392 static ConstantSInt *CastToShort (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000393 if (V->isNullValue()) return ConstantSInt::get(Type::ShortTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000394 return 0; // Can't const prop other types of pointers
395 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000396 static ConstantUInt *CastToUShort(const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000397 if (V->isNullValue()) return ConstantUInt::get(Type::UShortTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000398 return 0; // Can't const prop other types of pointers
399 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000400 static ConstantSInt *CastToInt (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000401 if (V->isNullValue()) return ConstantSInt::get(Type::IntTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000402 return 0; // Can't const prop other types of pointers
403 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000404 static ConstantUInt *CastToUInt (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000405 if (V->isNullValue()) return ConstantUInt::get(Type::UIntTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000406 return 0; // Can't const prop other types of pointers
407 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000408 static ConstantSInt *CastToLong (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000409 if (V->isNullValue()) return ConstantSInt::get(Type::LongTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000410 return 0; // Can't const prop other types of pointers
411 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000412 static ConstantUInt *CastToULong (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000413 if (V->isNullValue()) return ConstantUInt::get(Type::ULongTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000414 return 0; // Can't const prop other types of pointers
415 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000416 static ConstantFP *CastToFloat (const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000417 if (V->isNullValue()) return ConstantFP::get(Type::FloatTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000418 return 0; // Can't const prop other types of pointers
419 }
Chris Lattnere87f65e2002-07-30 16:24:28 +0000420 static ConstantFP *CastToDouble(const Constant *V) {
Chris Lattner3462ae32001-12-03 22:26:30 +0000421 if (V->isNullValue()) return ConstantFP::get(Type::DoubleTy, 0);
Chris Lattner977f0042001-11-01 05:55:13 +0000422 return 0; // Can't const prop other types of pointers
423 }
424
Chris Lattner1f0049c2003-04-17 19:24:18 +0000425 static Constant *CastToPointer(const ConstantPointer *V,
426 const PointerType *PTy) {
Chris Lattner62af86e2002-05-03 20:09:52 +0000427 if (V->getType() == PTy)
428 return const_cast<ConstantPointer*>(V); // Allow cast %PTy %ptr to %PTy
Chris Lattner977f0042001-11-01 05:55:13 +0000429 if (V->isNullValue())
Chris Lattner3462ae32001-12-03 22:26:30 +0000430 return ConstantPointerNull::get(PTy);
Chris Lattner977f0042001-11-01 05:55:13 +0000431 return 0; // Can't const prop other types of pointers
432 }
433};
434
435
436//===----------------------------------------------------------------------===//
Chris Lattner2f7c9632001-06-06 20:29:01 +0000437// DirectRules Class
438//===----------------------------------------------------------------------===//
439//
440// DirectRules provides a concrete base classes of ConstRules for a variety of
441// different types. This allows the C++ compiler to automatically generate our
442// constant handling operations in a typesafe and accurate manner.
443//
Chris Lattner0a144ad2002-05-03 21:41:07 +0000444template<class ConstantClass, class BuiltinType, Type **Ty, class SuperClass>
445struct DirectRules : public TemplateRules<ConstantClass, SuperClass> {
Chris Lattnere87f65e2002-07-30 16:24:28 +0000446 static Constant *Add(const ConstantClass *V1, const ConstantClass *V2) {
447 BuiltinType R = (BuiltinType)V1->getValue() + (BuiltinType)V2->getValue();
448 return ConstantClass::get(*Ty, R);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000449 }
450
Chris Lattnere87f65e2002-07-30 16:24:28 +0000451 static Constant *Sub(const ConstantClass *V1, const ConstantClass *V2) {
452 BuiltinType R = (BuiltinType)V1->getValue() - (BuiltinType)V2->getValue();
453 return ConstantClass::get(*Ty, R);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000454 }
455
Chris Lattnere87f65e2002-07-30 16:24:28 +0000456 static Constant *Mul(const ConstantClass *V1, const ConstantClass *V2) {
457 BuiltinType R = (BuiltinType)V1->getValue() * (BuiltinType)V2->getValue();
458 return ConstantClass::get(*Ty, R);
Chris Lattner4f6031f2001-07-20 19:15:36 +0000459 }
460
Chris Lattnere87f65e2002-07-30 16:24:28 +0000461 static Constant *Div(const ConstantClass *V1, const ConstantClass *V2) {
Chris Lattner0a144ad2002-05-03 21:41:07 +0000462 if (V2->isNullValue()) return 0;
Chris Lattnere87f65e2002-07-30 16:24:28 +0000463 BuiltinType R = (BuiltinType)V1->getValue() / (BuiltinType)V2->getValue();
464 return ConstantClass::get(*Ty, R);
Chris Lattneraf259a72002-04-07 08:10:14 +0000465 }
466
Chris Lattnere87f65e2002-07-30 16:24:28 +0000467 static ConstantBool *LessThan(const ConstantClass *V1,
468 const ConstantClass *V2) {
469 bool R = (BuiltinType)V1->getValue() < (BuiltinType)V2->getValue();
470 return ConstantBool::get(R);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000471 }
Chris Lattner55406842001-07-21 19:10:49 +0000472
Chris Lattner1f0049c2003-04-17 19:24:18 +0000473 static Constant *CastToPointer(const ConstantClass *V,
474 const PointerType *PTy) {
Chris Lattner977f0042001-11-01 05:55:13 +0000475 if (V->isNullValue()) // Is it a FP or Integral null value?
Chris Lattner3462ae32001-12-03 22:26:30 +0000476 return ConstantPointerNull::get(PTy);
Chris Lattner977f0042001-11-01 05:55:13 +0000477 return 0; // Can't const prop other types of pointers
478 }
479
Chris Lattner55406842001-07-21 19:10:49 +0000480 // Casting operators. ick
481#define DEF_CAST(TYPE, CLASS, CTYPE) \
Chris Lattnere87f65e2002-07-30 16:24:28 +0000482 static CLASS *CastTo##TYPE (const ConstantClass *V) { \
Chris Lattnerbbb22962001-09-07 16:40:34 +0000483 return CLASS::get(Type::TYPE##Ty, (CTYPE)(BuiltinType)V->getValue()); \
Chris Lattner55406842001-07-21 19:10:49 +0000484 }
485
Chris Lattner3462ae32001-12-03 22:26:30 +0000486 DEF_CAST(Bool , ConstantBool, bool)
487 DEF_CAST(SByte , ConstantSInt, signed char)
488 DEF_CAST(UByte , ConstantUInt, unsigned char)
489 DEF_CAST(Short , ConstantSInt, signed short)
490 DEF_CAST(UShort, ConstantUInt, unsigned short)
491 DEF_CAST(Int , ConstantSInt, signed int)
492 DEF_CAST(UInt , ConstantUInt, unsigned int)
493 DEF_CAST(Long , ConstantSInt, int64_t)
494 DEF_CAST(ULong , ConstantUInt, uint64_t)
495 DEF_CAST(Float , ConstantFP , float)
496 DEF_CAST(Double, ConstantFP , double)
Chris Lattner55406842001-07-21 19:10:49 +0000497#undef DEF_CAST
Chris Lattner2f7c9632001-06-06 20:29:01 +0000498};
499
Chris Lattner62af86e2002-05-03 20:09:52 +0000500
501//===----------------------------------------------------------------------===//
502// DirectIntRules Class
503//===----------------------------------------------------------------------===//
504//
505// DirectIntRules provides implementations of functions that are valid on
506// integer types, but not all types in general.
507//
508template <class ConstantClass, class BuiltinType, Type **Ty>
Chris Lattner0a144ad2002-05-03 21:41:07 +0000509struct DirectIntRules
510 : public DirectRules<ConstantClass, BuiltinType, Ty,
511 DirectIntRules<ConstantClass, BuiltinType, Ty> > {
Chris Lattner0a144ad2002-05-03 21:41:07 +0000512
Chris Lattner268916262003-05-12 15:26:25 +0000513 static Constant *Div(const ConstantClass *V1, const ConstantClass *V2) {
514 if (V2->isNullValue()) return 0;
515 if (V2->isAllOnesValue() && // MIN_INT / -1
516 (BuiltinType)V1->getValue() == -(BuiltinType)V1->getValue())
517 return 0;
518 BuiltinType R = (BuiltinType)V1->getValue() / (BuiltinType)V2->getValue();
519 return ConstantClass::get(*Ty, R);
520 }
521
Chris Lattnere87f65e2002-07-30 16:24:28 +0000522 static Constant *Rem(const ConstantClass *V1,
523 const ConstantClass *V2) {
Chris Lattner268916262003-05-12 15:26:25 +0000524 if (V2->isNullValue()) return 0; // X / 0
525 if (V2->isAllOnesValue() && // MIN_INT / -1
526 (BuiltinType)V1->getValue() == -(BuiltinType)V1->getValue())
527 return 0;
Chris Lattnere87f65e2002-07-30 16:24:28 +0000528 BuiltinType R = (BuiltinType)V1->getValue() % (BuiltinType)V2->getValue();
529 return ConstantClass::get(*Ty, R);
Chris Lattner0a144ad2002-05-03 21:41:07 +0000530 }
Chris Lattner6670d862002-05-06 03:00:54 +0000531
Chris Lattnere87f65e2002-07-30 16:24:28 +0000532 static Constant *And(const ConstantClass *V1, const ConstantClass *V2) {
533 BuiltinType R = (BuiltinType)V1->getValue() & (BuiltinType)V2->getValue();
534 return ConstantClass::get(*Ty, R);
535 }
536 static Constant *Or(const ConstantClass *V1, const ConstantClass *V2) {
537 BuiltinType R = (BuiltinType)V1->getValue() | (BuiltinType)V2->getValue();
538 return ConstantClass::get(*Ty, R);
539 }
540 static Constant *Xor(const ConstantClass *V1, const ConstantClass *V2) {
541 BuiltinType R = (BuiltinType)V1->getValue() ^ (BuiltinType)V2->getValue();
542 return ConstantClass::get(*Ty, R);
Chris Lattner6670d862002-05-06 03:00:54 +0000543 }
544
Chris Lattnere87f65e2002-07-30 16:24:28 +0000545 static Constant *Shl(const ConstantClass *V1, const ConstantClass *V2) {
546 BuiltinType R = (BuiltinType)V1->getValue() << (BuiltinType)V2->getValue();
547 return ConstantClass::get(*Ty, R);
548 }
549
550 static Constant *Shr(const ConstantClass *V1, const ConstantClass *V2) {
551 BuiltinType R = (BuiltinType)V1->getValue() >> (BuiltinType)V2->getValue();
552 return ConstantClass::get(*Ty, R);
Chris Lattner6670d862002-05-06 03:00:54 +0000553 }
Chris Lattner0a144ad2002-05-03 21:41:07 +0000554};
555
556
557//===----------------------------------------------------------------------===//
558// DirectFPRules Class
559//===----------------------------------------------------------------------===//
560//
561// DirectFPRules provides implementations of functions that are valid on
562// floating point types, but not all types in general.
563//
564template <class ConstantClass, class BuiltinType, Type **Ty>
565struct DirectFPRules
566 : public DirectRules<ConstantClass, BuiltinType, Ty,
567 DirectFPRules<ConstantClass, BuiltinType, Ty> > {
Chris Lattnere87f65e2002-07-30 16:24:28 +0000568 static Constant *Rem(const ConstantClass *V1, const ConstantClass *V2) {
Chris Lattner0a144ad2002-05-03 21:41:07 +0000569 if (V2->isNullValue()) return 0;
570 BuiltinType Result = std::fmod((BuiltinType)V1->getValue(),
571 (BuiltinType)V2->getValue());
572 return ConstantClass::get(*Ty, Result);
573 }
Chris Lattner62af86e2002-05-03 20:09:52 +0000574};
575
Chris Lattner2f7c9632001-06-06 20:29:01 +0000576//===----------------------------------------------------------------------===//
577// DirectRules Subclasses
578//===----------------------------------------------------------------------===//
579//
580// Given the DirectRules class we can now implement lots of types with little
581// code. Thank goodness C++ compilers are great at stomping out layers of
582// templates... can you imagine having to do this all by hand? (/me is lazy :)
583//
Chris Lattner2f7c9632001-06-06 20:29:01 +0000584
585// ConstRules::find - Return the constant rules that take care of the specified
Chris Lattner61607ee2001-09-09 21:01:20 +0000586// type.
Chris Lattner2f7c9632001-06-06 20:29:01 +0000587//
Chris Lattner61607ee2001-09-09 21:01:20 +0000588Annotation *ConstRules::find(AnnotationID AID, const Annotable *TyA, void *) {
589 assert(AID == ConstRules::AID && "Bad annotation for factory!");
Chris Lattner8f191122001-10-01 18:26:53 +0000590 const Type *Ty = cast<Type>((const Value*)TyA);
Chris Lattner61607ee2001-09-09 21:01:20 +0000591
Chris Lattner2f7c9632001-06-06 20:29:01 +0000592 switch (Ty->getPrimitiveID()) {
Chris Lattner977f0042001-11-01 05:55:13 +0000593 case Type::BoolTyID: return new BoolRules();
594 case Type::PointerTyID: return new PointerRules();
Chris Lattner61607ee2001-09-09 21:01:20 +0000595 case Type::SByteTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000596 return new DirectIntRules<ConstantSInt, signed char , &Type::SByteTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000597 case Type::UByteTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000598 return new DirectIntRules<ConstantUInt, unsigned char , &Type::UByteTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000599 case Type::ShortTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000600 return new DirectIntRules<ConstantSInt, signed short, &Type::ShortTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000601 case Type::UShortTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000602 return new DirectIntRules<ConstantUInt, unsigned short, &Type::UShortTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000603 case Type::IntTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000604 return new DirectIntRules<ConstantSInt, signed int , &Type::IntTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000605 case Type::UIntTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000606 return new DirectIntRules<ConstantUInt, unsigned int , &Type::UIntTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000607 case Type::LongTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000608 return new DirectIntRules<ConstantSInt, int64_t , &Type::LongTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000609 case Type::ULongTyID:
Chris Lattner62af86e2002-05-03 20:09:52 +0000610 return new DirectIntRules<ConstantUInt, uint64_t , &Type::ULongTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000611 case Type::FloatTyID:
Chris Lattner0a144ad2002-05-03 21:41:07 +0000612 return new DirectFPRules<ConstantFP , float , &Type::FloatTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000613 case Type::DoubleTyID:
Chris Lattner0a144ad2002-05-03 21:41:07 +0000614 return new DirectFPRules<ConstantFP , double , &Type::DoubleTy>();
Chris Lattner61607ee2001-09-09 21:01:20 +0000615 default:
616 return new EmptyRules();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000617 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000618}
Chris Lattnere17c1c5a2003-04-25 02:52:06 +0000619
620ConstRules *ConstRules::getConstantExprRules() {
621 static EmptyRules CERules;
622 return &CERules;
623}