blob: b22fd1795eab6aed9915d25f369da314c21a00dd [file] [log] [blame]
Chris Lattnere6794492002-08-12 21:17:25 +00001//===- InstructionCombining.cpp - Combine multiple instructions -----------===//
John Criswell482202a2003-10-20 19:43:21 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
Chris Lattnerca081252001-12-14 16:52:21 +00009//
10// InstructionCombining - Combine instructions to form fewer, simple
Chris Lattner99f48c62002-09-02 04:59:56 +000011// instructions. This pass does not modify the CFG This pass is where algebraic
12// simplification happens.
Chris Lattnerca081252001-12-14 16:52:21 +000013//
14// This pass combines things like:
Chris Lattnerdd1a86d2004-05-04 15:19:33 +000015// %Y = add int %X, 1
16// %Z = add int %Y, 1
Chris Lattnerca081252001-12-14 16:52:21 +000017// into:
Chris Lattnerdd1a86d2004-05-04 15:19:33 +000018// %Z = add int %X, 2
Chris Lattnerca081252001-12-14 16:52:21 +000019//
20// This is a simple worklist driven algorithm.
21//
Chris Lattner216c7b82003-09-10 05:29:43 +000022// This pass guarantees that the following canonicalizations are performed on
Chris Lattnerbfb1d032003-07-23 21:41:57 +000023// the program:
24// 1. If a binary operator has a constant operand, it is moved to the RHS
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +000025// 2. Bitwise operators with constant operands are always grouped so that
26// shifts are performed first, then or's, then and's, then xor's.
Chris Lattnerbfb1d032003-07-23 21:41:57 +000027// 3. SetCC instructions are converted from <,>,<=,>= to ==,!= if possible
28// 4. All SetCC instructions on boolean values are replaced with logical ops
Chris Lattnerede3fe02003-08-13 04:18:28 +000029// 5. add X, X is represented as (X*2) => (X << 1)
30// 6. Multiplies with a power-of-two constant argument are transformed into
31// shifts.
Chris Lattnerbfb1d032003-07-23 21:41:57 +000032// N. This list is incomplete
33//
Chris Lattnerca081252001-12-14 16:52:21 +000034//===----------------------------------------------------------------------===//
35
Chris Lattner7d2a5392004-03-13 23:54:27 +000036#define DEBUG_TYPE "instcombine"
Chris Lattnerb4cfa7f2002-05-07 20:03:00 +000037#include "llvm/Transforms/Scalar.h"
Chris Lattner471bd762003-05-22 19:07:21 +000038#include "llvm/Instructions.h"
Chris Lattner51ea1272004-02-28 05:22:00 +000039#include "llvm/Intrinsics.h"
Chris Lattner04805fa2002-02-26 21:46:54 +000040#include "llvm/Pass.h"
Chris Lattner34428442003-05-27 16:40:51 +000041#include "llvm/Constants.h"
Chris Lattner1085bdf2002-11-04 16:18:53 +000042#include "llvm/DerivedTypes.h"
Chris Lattner0f1d8a32003-06-26 05:06:25 +000043#include "llvm/GlobalVariable.h"
Chris Lattnerf4ad1652003-11-02 05:57:39 +000044#include "llvm/Target/TargetData.h"
45#include "llvm/Transforms/Utils/BasicBlockUtils.h"
46#include "llvm/Transforms/Utils/Local.h"
Chris Lattner69193f92004-04-05 01:30:19 +000047#include "llvm/Support/CallSite.h"
48#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattner60a65912002-02-12 21:07:25 +000049#include "llvm/Support/InstIterator.h"
Chris Lattner260ab202002-04-18 17:39:14 +000050#include "llvm/Support/InstVisitor.h"
Chris Lattner7d2a5392004-03-13 23:54:27 +000051#include "Support/Debug.h"
Chris Lattnerbf3a0992002-10-01 22:38:41 +000052#include "Support/Statistic.h"
Chris Lattner053c0932002-05-14 15:24:07 +000053#include <algorithm>
Chris Lattner8427bff2003-12-07 01:24:23 +000054using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000055
Chris Lattner260ab202002-04-18 17:39:14 +000056namespace {
Chris Lattnerbf3a0992002-10-01 22:38:41 +000057 Statistic<> NumCombined ("instcombine", "Number of insts combined");
58 Statistic<> NumConstProp("instcombine", "Number of constant folds");
59 Statistic<> NumDeadInst ("instcombine", "Number of dead inst eliminated");
60
Chris Lattnerc8e66542002-04-27 06:56:12 +000061 class InstCombiner : public FunctionPass,
Chris Lattner260ab202002-04-18 17:39:14 +000062 public InstVisitor<InstCombiner, Instruction*> {
63 // Worklist of all of the instructions that need to be simplified.
64 std::vector<Instruction*> WorkList;
Chris Lattnerf4ad1652003-11-02 05:57:39 +000065 TargetData *TD;
Chris Lattner260ab202002-04-18 17:39:14 +000066
Chris Lattner51ea1272004-02-28 05:22:00 +000067 /// AddUsersToWorkList - When an instruction is simplified, add all users of
68 /// the instruction to the work lists because they might get more simplified
69 /// now.
70 ///
71 void AddUsersToWorkList(Instruction &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +000072 for (Value::use_iterator UI = I.use_begin(), UE = I.use_end();
Chris Lattner260ab202002-04-18 17:39:14 +000073 UI != UE; ++UI)
74 WorkList.push_back(cast<Instruction>(*UI));
75 }
76
Chris Lattner51ea1272004-02-28 05:22:00 +000077 /// AddUsesToWorkList - When an instruction is simplified, add operands to
78 /// the work lists because they might get more simplified now.
79 ///
80 void AddUsesToWorkList(Instruction &I) {
81 for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i)
82 if (Instruction *Op = dyn_cast<Instruction>(I.getOperand(i)))
83 WorkList.push_back(Op);
84 }
85
Chris Lattner99f48c62002-09-02 04:59:56 +000086 // removeFromWorkList - remove all instances of I from the worklist.
87 void removeFromWorkList(Instruction *I);
Chris Lattner260ab202002-04-18 17:39:14 +000088 public:
Chris Lattner113f4f42002-06-25 16:13:24 +000089 virtual bool runOnFunction(Function &F);
Chris Lattner260ab202002-04-18 17:39:14 +000090
Chris Lattnerf12cc842002-04-28 21:27:06 +000091 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Chris Lattnerf4ad1652003-11-02 05:57:39 +000092 AU.addRequired<TargetData>();
Chris Lattner820d9712002-10-21 20:00:28 +000093 AU.setPreservesCFG();
Chris Lattnerf12cc842002-04-28 21:27:06 +000094 }
95
Chris Lattner69193f92004-04-05 01:30:19 +000096 TargetData &getTargetData() const { return *TD; }
97
Chris Lattner260ab202002-04-18 17:39:14 +000098 // Visitation implementation - Implement instruction combining for different
99 // instruction types. The semantics are as follows:
100 // Return Value:
101 // null - No change was made
Chris Lattnere6794492002-08-12 21:17:25 +0000102 // I - Change was made, I is still valid, I may be dead though
Chris Lattner260ab202002-04-18 17:39:14 +0000103 // otherwise - Change was made, replace I with returned instruction
104 //
Chris Lattner113f4f42002-06-25 16:13:24 +0000105 Instruction *visitAdd(BinaryOperator &I);
106 Instruction *visitSub(BinaryOperator &I);
107 Instruction *visitMul(BinaryOperator &I);
108 Instruction *visitDiv(BinaryOperator &I);
109 Instruction *visitRem(BinaryOperator &I);
110 Instruction *visitAnd(BinaryOperator &I);
111 Instruction *visitOr (BinaryOperator &I);
112 Instruction *visitXor(BinaryOperator &I);
113 Instruction *visitSetCondInst(BinaryOperator &I);
Chris Lattnere8d6c602003-03-10 19:16:08 +0000114 Instruction *visitShiftInst(ShiftInst &I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000115 Instruction *visitCastInst(CastInst &CI);
Chris Lattnerb909e8b2004-03-12 05:52:32 +0000116 Instruction *visitSelectInst(SelectInst &CI);
Chris Lattner970c33a2003-06-19 17:00:31 +0000117 Instruction *visitCallInst(CallInst &CI);
118 Instruction *visitInvokeInst(InvokeInst &II);
Chris Lattner113f4f42002-06-25 16:13:24 +0000119 Instruction *visitPHINode(PHINode &PN);
120 Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP);
Chris Lattner1085bdf2002-11-04 16:18:53 +0000121 Instruction *visitAllocationInst(AllocationInst &AI);
Chris Lattner8427bff2003-12-07 01:24:23 +0000122 Instruction *visitFreeInst(FreeInst &FI);
Chris Lattner0f1d8a32003-06-26 05:06:25 +0000123 Instruction *visitLoadInst(LoadInst &LI);
Chris Lattner9eef8a72003-06-04 04:46:00 +0000124 Instruction *visitBranchInst(BranchInst &BI);
Chris Lattner260ab202002-04-18 17:39:14 +0000125
126 // visitInstruction - Specify what to return for unhandled instructions...
Chris Lattner113f4f42002-06-25 16:13:24 +0000127 Instruction *visitInstruction(Instruction &I) { return 0; }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000128
Chris Lattner970c33a2003-06-19 17:00:31 +0000129 private:
Chris Lattneraec3d942003-10-07 22:32:43 +0000130 Instruction *visitCallSite(CallSite CS);
Chris Lattner970c33a2003-06-19 17:00:31 +0000131 bool transformConstExprCastCall(CallSite CS);
132
Chris Lattner69193f92004-04-05 01:30:19 +0000133 public:
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000134 // InsertNewInstBefore - insert an instruction New before instruction Old
135 // in the program. Add the new instruction to the worklist.
136 //
Chris Lattnere79e8542004-02-23 06:38:22 +0000137 Value *InsertNewInstBefore(Instruction *New, Instruction &Old) {
Chris Lattner65217ff2002-08-23 18:32:43 +0000138 assert(New && New->getParent() == 0 &&
139 "New instruction already inserted into a basic block!");
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000140 BasicBlock *BB = Old.getParent();
141 BB->getInstList().insert(&Old, New); // Insert inst
142 WorkList.push_back(New); // Add to worklist
Chris Lattnere79e8542004-02-23 06:38:22 +0000143 return New;
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000144 }
145
146 // ReplaceInstUsesWith - This method is to be used when an instruction is
147 // found to be dead, replacable with another preexisting expression. Here
148 // we add all uses of I to the worklist, replace all uses of I with the new
149 // value, then return I, so that the inst combiner will know that I was
150 // modified.
151 //
152 Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) {
Chris Lattner51ea1272004-02-28 05:22:00 +0000153 AddUsersToWorkList(I); // Add all modified instrs to worklist
Chris Lattner8953b902004-04-05 02:10:19 +0000154 if (&I != V) {
155 I.replaceAllUsesWith(V);
156 return &I;
157 } else {
158 // If we are replacing the instruction with itself, this must be in a
159 // segment of unreachable code, so just clobber the instruction.
160 I.replaceAllUsesWith(Constant::getNullValue(I.getType()));
161 return &I;
162 }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000163 }
Chris Lattner51ea1272004-02-28 05:22:00 +0000164
165 // EraseInstFromFunction - When dealing with an instruction that has side
166 // effects or produces a void value, we can't rely on DCE to delete the
167 // instruction. Instead, visit methods should return the value returned by
168 // this function.
169 Instruction *EraseInstFromFunction(Instruction &I) {
170 assert(I.use_empty() && "Cannot erase instruction that is used!");
171 AddUsesToWorkList(I);
172 removeFromWorkList(&I);
173 I.getParent()->getInstList().erase(&I);
174 return 0; // Don't do anything with FI
175 }
176
177
Chris Lattner3ac7c262003-08-13 20:16:26 +0000178 private:
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000179 /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
180 /// InsertBefore instruction. This is specialized a bit to avoid inserting
181 /// casts that are known to not do anything...
182 ///
183 Value *InsertOperandCastBefore(Value *V, const Type *DestTy,
184 Instruction *InsertBefore);
185
Chris Lattner7fb29e12003-03-11 00:12:48 +0000186 // SimplifyCommutative - This performs a few simplifications for commutative
187 // operators...
188 bool SimplifyCommutative(BinaryOperator &I);
Chris Lattnerba1cb382003-09-19 17:17:26 +0000189
190 Instruction *OptAndOp(Instruction *Op, ConstantIntegral *OpRHS,
191 ConstantIntegral *AndRHS, BinaryOperator &TheAnd);
Chris Lattner260ab202002-04-18 17:39:14 +0000192 };
Chris Lattnerb28b6802002-07-23 18:06:35 +0000193
Chris Lattnerc8b70922002-07-26 21:12:46 +0000194 RegisterOpt<InstCombiner> X("instcombine", "Combine redundant instructions");
Chris Lattner260ab202002-04-18 17:39:14 +0000195}
196
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000197// getComplexity: Assign a complexity or rank value to LLVM Values...
198// 0 -> Constant, 1 -> Other, 2 -> Argument, 2 -> Unary, 3 -> OtherInst
199static unsigned getComplexity(Value *V) {
200 if (isa<Instruction>(V)) {
201 if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V))
202 return 2;
203 return 3;
204 }
205 if (isa<Argument>(V)) return 2;
206 return isa<Constant>(V) ? 0 : 1;
207}
Chris Lattner260ab202002-04-18 17:39:14 +0000208
Chris Lattner7fb29e12003-03-11 00:12:48 +0000209// isOnlyUse - Return true if this instruction will be deleted if we stop using
210// it.
211static bool isOnlyUse(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000212 return V->hasOneUse() || isa<Constant>(V);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000213}
214
Chris Lattnere79e8542004-02-23 06:38:22 +0000215// getSignedIntegralType - Given an unsigned integral type, return the signed
216// version of it that has the same size.
217static const Type *getSignedIntegralType(const Type *Ty) {
218 switch (Ty->getPrimitiveID()) {
219 default: assert(0 && "Invalid unsigned integer type!"); abort();
220 case Type::UByteTyID: return Type::SByteTy;
221 case Type::UShortTyID: return Type::ShortTy;
222 case Type::UIntTyID: return Type::IntTy;
223 case Type::ULongTyID: return Type::LongTy;
224 }
225}
226
Chris Lattner92295c52004-03-12 23:53:13 +0000227// getUnsignedIntegralType - Given an signed integral type, return the unsigned
228// version of it that has the same size.
229static const Type *getUnsignedIntegralType(const Type *Ty) {
230 switch (Ty->getPrimitiveID()) {
231 default: assert(0 && "Invalid signed integer type!"); abort();
232 case Type::SByteTyID: return Type::UByteTy;
233 case Type::ShortTyID: return Type::UShortTy;
234 case Type::IntTyID: return Type::UIntTy;
235 case Type::LongTyID: return Type::ULongTy;
236 }
237}
238
Chris Lattnere79e8542004-02-23 06:38:22 +0000239// getPromotedType - Return the specified type promoted as it would be to pass
240// though a va_arg area...
241static const Type *getPromotedType(const Type *Ty) {
242 switch (Ty->getPrimitiveID()) {
243 case Type::SByteTyID:
244 case Type::ShortTyID: return Type::IntTy;
245 case Type::UByteTyID:
246 case Type::UShortTyID: return Type::UIntTy;
247 case Type::FloatTyID: return Type::DoubleTy;
248 default: return Ty;
249 }
250}
251
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000252// SimplifyCommutative - This performs a few simplifications for commutative
253// operators:
Chris Lattner260ab202002-04-18 17:39:14 +0000254//
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000255// 1. Order operands such that they are listed from right (least complex) to
256// left (most complex). This puts constants before unary operators before
257// binary operators.
258//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000259// 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2))
260// 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000261//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000262bool InstCombiner::SimplifyCommutative(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000263 bool Changed = false;
264 if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1)))
265 Changed = !I.swapOperands();
266
267 if (!I.isAssociative()) return Changed;
268 Instruction::BinaryOps Opcode = I.getOpcode();
Chris Lattner7fb29e12003-03-11 00:12:48 +0000269 if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0)))
270 if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) {
271 if (isa<Constant>(I.getOperand(1))) {
Chris Lattner34428442003-05-27 16:40:51 +0000272 Constant *Folded = ConstantExpr::get(I.getOpcode(),
273 cast<Constant>(I.getOperand(1)),
274 cast<Constant>(Op->getOperand(1)));
Chris Lattner7fb29e12003-03-11 00:12:48 +0000275 I.setOperand(0, Op->getOperand(0));
276 I.setOperand(1, Folded);
277 return true;
278 } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1)))
279 if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) &&
280 isOnlyUse(Op) && isOnlyUse(Op1)) {
281 Constant *C1 = cast<Constant>(Op->getOperand(1));
282 Constant *C2 = cast<Constant>(Op1->getOperand(1));
283
284 // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattner34428442003-05-27 16:40:51 +0000285 Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000286 Instruction *New = BinaryOperator::create(Opcode, Op->getOperand(0),
287 Op1->getOperand(0),
288 Op1->getName(), &I);
289 WorkList.push_back(New);
290 I.setOperand(0, New);
291 I.setOperand(1, Folded);
292 return true;
293 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000294 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000295 return Changed;
Chris Lattner260ab202002-04-18 17:39:14 +0000296}
Chris Lattnerca081252001-12-14 16:52:21 +0000297
Chris Lattnerbb74e222003-03-10 23:06:50 +0000298// dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction
299// if the LHS is a constant zero (which is the 'negate' form).
Chris Lattner9fa53de2002-05-06 16:49:18 +0000300//
Chris Lattnerbb74e222003-03-10 23:06:50 +0000301static inline Value *dyn_castNegVal(Value *V) {
302 if (BinaryOperator::isNeg(V))
303 return BinaryOperator::getNegArgument(cast<BinaryOperator>(V));
304
Chris Lattner9244df62003-04-30 22:19:10 +0000305 // Constants can be considered to be negated values if they can be folded...
306 if (Constant *C = dyn_cast<Constant>(V))
Chris Lattner34428442003-05-27 16:40:51 +0000307 return ConstantExpr::get(Instruction::Sub,
308 Constant::getNullValue(V->getType()), C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000309 return 0;
Chris Lattner9fa53de2002-05-06 16:49:18 +0000310}
311
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000312static Constant *NotConstant(Constant *C) {
313 return ConstantExpr::get(Instruction::Xor, C,
314 ConstantIntegral::getAllOnesValue(C->getType()));
315}
316
Chris Lattnerbb74e222003-03-10 23:06:50 +0000317static inline Value *dyn_castNotVal(Value *V) {
318 if (BinaryOperator::isNot(V))
319 return BinaryOperator::getNotArgument(cast<BinaryOperator>(V));
320
321 // Constants can be considered to be not'ed values...
Chris Lattnerdd65d862003-04-30 22:34:06 +0000322 if (ConstantIntegral *C = dyn_cast<ConstantIntegral>(V))
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000323 return NotConstant(C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000324 return 0;
325}
326
Chris Lattner7fb29e12003-03-11 00:12:48 +0000327// dyn_castFoldableMul - If this value is a multiply that can be folded into
328// other computations (because it has a constant operand), return the
329// non-constant operand of the multiply.
330//
331static inline Value *dyn_castFoldableMul(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000332 if (V->hasOneUse() && V->getType()->isInteger())
Chris Lattner7fb29e12003-03-11 00:12:48 +0000333 if (Instruction *I = dyn_cast<Instruction>(V))
334 if (I->getOpcode() == Instruction::Mul)
335 if (isa<Constant>(I->getOperand(1)))
336 return I->getOperand(0);
337 return 0;
Chris Lattner3082c5a2003-02-18 19:28:33 +0000338}
Chris Lattner31ae8632002-08-14 17:51:49 +0000339
Chris Lattner7fb29e12003-03-11 00:12:48 +0000340// dyn_castMaskingAnd - If this value is an And instruction masking a value with
341// a constant, return the constant being anded with.
342//
Chris Lattner01d56392003-08-12 19:17:27 +0000343template<class ValueType>
344static inline Constant *dyn_castMaskingAnd(ValueType *V) {
Chris Lattner7fb29e12003-03-11 00:12:48 +0000345 if (Instruction *I = dyn_cast<Instruction>(V))
346 if (I->getOpcode() == Instruction::And)
347 return dyn_cast<Constant>(I->getOperand(1));
348
349 // If this is a constant, it acts just like we were masking with it.
350 return dyn_cast<Constant>(V);
351}
Chris Lattner3082c5a2003-02-18 19:28:33 +0000352
353// Log2 - Calculate the log base 2 for the specified value if it is exactly a
354// power of 2.
355static unsigned Log2(uint64_t Val) {
356 assert(Val > 1 && "Values 0 and 1 should be handled elsewhere!");
357 unsigned Count = 0;
358 while (Val != 1) {
359 if (Val & 1) return 0; // Multiple bits set?
360 Val >>= 1;
361 ++Count;
362 }
363 return Count;
Chris Lattner31ae8632002-08-14 17:51:49 +0000364}
365
Chris Lattnerb8b97502003-08-13 19:01:45 +0000366
367/// AssociativeOpt - Perform an optimization on an associative operator. This
368/// function is designed to check a chain of associative operators for a
369/// potential to apply a certain optimization. Since the optimization may be
370/// applicable if the expression was reassociated, this checks the chain, then
371/// reassociates the expression as necessary to expose the optimization
372/// opportunity. This makes use of a special Functor, which must define
373/// 'shouldApply' and 'apply' methods.
374///
375template<typename Functor>
376Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) {
377 unsigned Opcode = Root.getOpcode();
378 Value *LHS = Root.getOperand(0);
379
380 // Quick check, see if the immediate LHS matches...
381 if (F.shouldApply(LHS))
382 return F.apply(Root);
383
384 // Otherwise, if the LHS is not of the same opcode as the root, return.
385 Instruction *LHSI = dyn_cast<Instruction>(LHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000386 while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) {
Chris Lattnerb8b97502003-08-13 19:01:45 +0000387 // Should we apply this transform to the RHS?
388 bool ShouldApply = F.shouldApply(LHSI->getOperand(1));
389
390 // If not to the RHS, check to see if we should apply to the LHS...
391 if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) {
392 cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS
393 ShouldApply = true;
394 }
395
396 // If the functor wants to apply the optimization to the RHS of LHSI,
397 // reassociate the expression from ((? op A) op B) to (? op (A op B))
398 if (ShouldApply) {
399 BasicBlock *BB = Root.getParent();
Chris Lattnerb8b97502003-08-13 19:01:45 +0000400
401 // Now all of the instructions are in the current basic block, go ahead
402 // and perform the reassociation.
403 Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0));
404
405 // First move the selected RHS to the LHS of the root...
406 Root.setOperand(0, LHSI->getOperand(1));
407
408 // Make what used to be the LHS of the root be the user of the root...
409 Value *ExtraOperand = TmpLHSI->getOperand(1);
Chris Lattner284d3b02004-04-16 18:08:07 +0000410 if (&Root == TmpLHSI) {
Chris Lattner8953b902004-04-05 02:10:19 +0000411 Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType()));
412 return 0;
413 }
Chris Lattner284d3b02004-04-16 18:08:07 +0000414 Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI
Chris Lattnerb8b97502003-08-13 19:01:45 +0000415 TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root
Chris Lattner284d3b02004-04-16 18:08:07 +0000416 TmpLHSI->getParent()->getInstList().remove(TmpLHSI);
417 BasicBlock::iterator ARI = &Root; ++ARI;
418 BB->getInstList().insert(ARI, TmpLHSI); // Move TmpLHSI to after Root
419 ARI = Root;
Chris Lattnerb8b97502003-08-13 19:01:45 +0000420
421 // Now propagate the ExtraOperand down the chain of instructions until we
422 // get to LHSI.
423 while (TmpLHSI != LHSI) {
424 Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0));
Chris Lattner284d3b02004-04-16 18:08:07 +0000425 // Move the instruction to immediately before the chain we are
426 // constructing to avoid breaking dominance properties.
427 NextLHSI->getParent()->getInstList().remove(NextLHSI);
428 BB->getInstList().insert(ARI, NextLHSI);
429 ARI = NextLHSI;
430
Chris Lattnerb8b97502003-08-13 19:01:45 +0000431 Value *NextOp = NextLHSI->getOperand(1);
432 NextLHSI->setOperand(1, ExtraOperand);
433 TmpLHSI = NextLHSI;
434 ExtraOperand = NextOp;
435 }
436
437 // Now that the instructions are reassociated, have the functor perform
438 // the transformation...
439 return F.apply(Root);
440 }
441
442 LHSI = dyn_cast<Instruction>(LHSI->getOperand(0));
443 }
444 return 0;
445}
446
447
448// AddRHS - Implements: X + X --> X << 1
449struct AddRHS {
450 Value *RHS;
451 AddRHS(Value *rhs) : RHS(rhs) {}
452 bool shouldApply(Value *LHS) const { return LHS == RHS; }
453 Instruction *apply(BinaryOperator &Add) const {
454 return new ShiftInst(Instruction::Shl, Add.getOperand(0),
455 ConstantInt::get(Type::UByteTy, 1));
456 }
457};
458
459// AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2)
460// iff C1&C2 == 0
461struct AddMaskingAnd {
462 Constant *C2;
463 AddMaskingAnd(Constant *c) : C2(c) {}
464 bool shouldApply(Value *LHS) const {
465 if (Constant *C1 = dyn_castMaskingAnd(LHS))
466 return ConstantExpr::get(Instruction::And, C1, C2)->isNullValue();
467 return false;
468 }
469 Instruction *apply(BinaryOperator &Add) const {
470 return BinaryOperator::create(Instruction::Or, Add.getOperand(0),
471 Add.getOperand(1));
472 }
473};
474
Chris Lattner183b3362004-04-09 19:05:30 +0000475static Value *FoldOperationIntoSelectOperand(Instruction &BI, Value *SO,
476 InstCombiner *IC) {
477 // Figure out if the constant is the left or the right argument.
478 bool ConstIsRHS = isa<Constant>(BI.getOperand(1));
479 Constant *ConstOperand = cast<Constant>(BI.getOperand(ConstIsRHS));
Chris Lattnerb8b97502003-08-13 19:01:45 +0000480
Chris Lattner183b3362004-04-09 19:05:30 +0000481 if (Constant *SOC = dyn_cast<Constant>(SO)) {
482 if (ConstIsRHS)
483 return ConstantExpr::get(BI.getOpcode(), SOC, ConstOperand);
484 return ConstantExpr::get(BI.getOpcode(), ConstOperand, SOC);
485 }
486
487 Value *Op0 = SO, *Op1 = ConstOperand;
488 if (!ConstIsRHS)
489 std::swap(Op0, Op1);
490 Instruction *New;
491 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&BI))
492 New = BinaryOperator::create(BO->getOpcode(), Op0, Op1);
493 else if (ShiftInst *SI = dyn_cast<ShiftInst>(&BI))
494 New = new ShiftInst(SI->getOpcode(), Op0, Op1);
Chris Lattnerf9d96652004-04-10 19:15:56 +0000495 else {
Chris Lattner183b3362004-04-09 19:05:30 +0000496 assert(0 && "Unknown binary instruction type!");
Chris Lattnerf9d96652004-04-10 19:15:56 +0000497 abort();
498 }
Chris Lattner183b3362004-04-09 19:05:30 +0000499 return IC->InsertNewInstBefore(New, BI);
500}
501
502// FoldBinOpIntoSelect - Given an instruction with a select as one operand and a
503// constant as the other operand, try to fold the binary operator into the
504// select arguments.
505static Instruction *FoldBinOpIntoSelect(Instruction &BI, SelectInst *SI,
506 InstCombiner *IC) {
507 // Don't modify shared select instructions
508 if (!SI->hasOneUse()) return 0;
509 Value *TV = SI->getOperand(1);
510 Value *FV = SI->getOperand(2);
511
512 if (isa<Constant>(TV) || isa<Constant>(FV)) {
513 Value *SelectTrueVal = FoldOperationIntoSelectOperand(BI, TV, IC);
514 Value *SelectFalseVal = FoldOperationIntoSelectOperand(BI, FV, IC);
515
516 return new SelectInst(SI->getCondition(), SelectTrueVal,
517 SelectFalseVal);
518 }
519 return 0;
520}
Chris Lattnerb8b97502003-08-13 19:01:45 +0000521
Chris Lattner113f4f42002-06-25 16:13:24 +0000522Instruction *InstCombiner::visitAdd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000523 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000524 Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000525
Chris Lattnercf4a9962004-04-10 22:01:55 +0000526 if (Constant *RHSC = dyn_cast<Constant>(RHS)) {
527 // X + 0 --> X
528 if (!I.getType()->isFloatingPoint() && // -0 + +0 = +0, so it's not a noop
529 RHSC->isNullValue())
530 return ReplaceInstUsesWith(I, LHS);
531
532 // X + (signbit) --> X ^ signbit
533 if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) {
534 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
535 uint64_t Val = CI->getRawValue() & (1ULL << NumBits)-1;
536 if (Val == (1ULL << NumBits-1))
537 return BinaryOperator::create(Instruction::Xor, LHS, RHS);
538 }
539 }
Chris Lattner9fa53de2002-05-06 16:49:18 +0000540
Chris Lattnerb8b97502003-08-13 19:01:45 +0000541 // X + X --> X << 1
542 if (I.getType()->isInteger())
543 if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result;
Chris Lattnerede3fe02003-08-13 04:18:28 +0000544
Chris Lattner147e9752002-05-08 22:46:53 +0000545 // -A + B --> B - A
Chris Lattnerbb74e222003-03-10 23:06:50 +0000546 if (Value *V = dyn_castNegVal(LHS))
Chris Lattner147e9752002-05-08 22:46:53 +0000547 return BinaryOperator::create(Instruction::Sub, RHS, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000548
549 // A + -B --> A - B
Chris Lattnerbb74e222003-03-10 23:06:50 +0000550 if (!isa<Constant>(RHS))
551 if (Value *V = dyn_castNegVal(RHS))
552 return BinaryOperator::create(Instruction::Sub, LHS, V);
Chris Lattner260ab202002-04-18 17:39:14 +0000553
Chris Lattner57c8d992003-02-18 19:57:07 +0000554 // X*C + X --> X * (C+1)
555 if (dyn_castFoldableMul(LHS) == RHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000556 Constant *CP1 =
557 ConstantExpr::get(Instruction::Add,
558 cast<Constant>(cast<Instruction>(LHS)->getOperand(1)),
559 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000560 return BinaryOperator::create(Instruction::Mul, RHS, CP1);
561 }
562
563 // X + X*C --> X * (C+1)
564 if (dyn_castFoldableMul(RHS) == LHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000565 Constant *CP1 =
566 ConstantExpr::get(Instruction::Add,
567 cast<Constant>(cast<Instruction>(RHS)->getOperand(1)),
568 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000569 return BinaryOperator::create(Instruction::Mul, LHS, CP1);
570 }
571
Chris Lattnerb8b97502003-08-13 19:01:45 +0000572 // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0
573 if (Constant *C2 = dyn_castMaskingAnd(RHS))
574 if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) return R;
Chris Lattner7fb29e12003-03-11 00:12:48 +0000575
Chris Lattnerb9cde762003-10-02 15:11:26 +0000576 if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) {
577 if (Instruction *ILHS = dyn_cast<Instruction>(LHS)) {
578 switch (ILHS->getOpcode()) {
579 case Instruction::Xor:
580 // ~X + C --> (C-1) - X
581 if (ConstantInt *XorRHS = dyn_cast<ConstantInt>(ILHS->getOperand(1)))
582 if (XorRHS->isAllOnesValue())
583 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000584 ConstantExpr::get(Instruction::Sub,
585 CRHS, ConstantInt::get(I.getType(), 1)),
Chris Lattnerb9cde762003-10-02 15:11:26 +0000586 ILHS->getOperand(0));
587 break;
Chris Lattner183b3362004-04-09 19:05:30 +0000588 case Instruction::Select:
589 // Try to fold constant add into select arguments.
590 if (Instruction *R = FoldBinOpIntoSelect(I,cast<SelectInst>(ILHS),this))
591 return R;
592
Chris Lattnerb9cde762003-10-02 15:11:26 +0000593 default: break;
594 }
595 }
596 }
597
Chris Lattner113f4f42002-06-25 16:13:24 +0000598 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000599}
600
Chris Lattnerbdb0ce02003-07-22 21:46:59 +0000601// isSignBit - Return true if the value represented by the constant only has the
602// highest order bit set.
603static bool isSignBit(ConstantInt *CI) {
604 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
605 return (CI->getRawValue() & ~(-1LL << NumBits)) == (1ULL << (NumBits-1));
606}
607
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000608static unsigned getTypeSizeInBits(const Type *Ty) {
609 return Ty == Type::BoolTy ? 1 : Ty->getPrimitiveSize()*8;
610}
611
Chris Lattner022167f2004-03-13 00:11:49 +0000612/// RemoveNoopCast - Strip off nonconverting casts from the value.
613///
614static Value *RemoveNoopCast(Value *V) {
615 if (CastInst *CI = dyn_cast<CastInst>(V)) {
616 const Type *CTy = CI->getType();
617 const Type *OpTy = CI->getOperand(0)->getType();
618 if (CTy->isInteger() && OpTy->isInteger()) {
619 if (CTy->getPrimitiveSize() == OpTy->getPrimitiveSize())
620 return RemoveNoopCast(CI->getOperand(0));
621 } else if (isa<PointerType>(CTy) && isa<PointerType>(OpTy))
622 return RemoveNoopCast(CI->getOperand(0));
623 }
624 return V;
625}
626
Chris Lattner113f4f42002-06-25 16:13:24 +0000627Instruction *InstCombiner::visitSub(BinaryOperator &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000628 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000629
Chris Lattnere6794492002-08-12 21:17:25 +0000630 if (Op0 == Op1) // sub X, X -> 0
631 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner260ab202002-04-18 17:39:14 +0000632
Chris Lattnere6794492002-08-12 21:17:25 +0000633 // If this is a 'B = x-(-A)', change to B = x+A...
Chris Lattnerbb74e222003-03-10 23:06:50 +0000634 if (Value *V = dyn_castNegVal(Op1))
Chris Lattner147e9752002-05-08 22:46:53 +0000635 return BinaryOperator::create(Instruction::Add, Op0, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000636
Chris Lattner8f2f5982003-11-05 01:06:05 +0000637 if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) {
638 // Replace (-1 - A) with (~A)...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000639 if (C->isAllOnesValue())
640 return BinaryOperator::createNot(Op1);
Chris Lattnerad3c4952002-05-09 01:29:19 +0000641
Chris Lattner8f2f5982003-11-05 01:06:05 +0000642 // C - ~X == X + (1+C)
643 if (BinaryOperator::isNot(Op1))
644 return BinaryOperator::create(Instruction::Add,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000645 BinaryOperator::getNotArgument(cast<BinaryOperator>(Op1)),
646 ConstantExpr::get(Instruction::Add, C,
647 ConstantInt::get(I.getType(), 1)));
Chris Lattner92295c52004-03-12 23:53:13 +0000648 // -((uint)X >> 31) -> ((int)X >> 31)
649 // -((int)X >> 31) -> ((uint)X >> 31)
Chris Lattner022167f2004-03-13 00:11:49 +0000650 if (C->isNullValue()) {
651 Value *NoopCastedRHS = RemoveNoopCast(Op1);
652 if (ShiftInst *SI = dyn_cast<ShiftInst>(NoopCastedRHS))
Chris Lattner92295c52004-03-12 23:53:13 +0000653 if (SI->getOpcode() == Instruction::Shr)
654 if (ConstantUInt *CU = dyn_cast<ConstantUInt>(SI->getOperand(1))) {
655 const Type *NewTy;
Chris Lattner022167f2004-03-13 00:11:49 +0000656 if (SI->getType()->isSigned())
657 NewTy = getUnsignedIntegralType(SI->getType());
Chris Lattner92295c52004-03-12 23:53:13 +0000658 else
Chris Lattner022167f2004-03-13 00:11:49 +0000659 NewTy = getSignedIntegralType(SI->getType());
Chris Lattner92295c52004-03-12 23:53:13 +0000660 // Check to see if we are shifting out everything but the sign bit.
Chris Lattner022167f2004-03-13 00:11:49 +0000661 if (CU->getValue() == SI->getType()->getPrimitiveSize()*8-1) {
Chris Lattner92295c52004-03-12 23:53:13 +0000662 // Ok, the transformation is safe. Insert a cast of the incoming
663 // value, then the new shift, then the new cast.
664 Instruction *FirstCast = new CastInst(SI->getOperand(0), NewTy,
665 SI->getOperand(0)->getName());
666 Value *InV = InsertNewInstBefore(FirstCast, I);
667 Instruction *NewShift = new ShiftInst(Instruction::Shr, FirstCast,
668 CU, SI->getName());
Chris Lattner022167f2004-03-13 00:11:49 +0000669 if (NewShift->getType() == I.getType())
670 return NewShift;
671 else {
672 InV = InsertNewInstBefore(NewShift, I);
673 return new CastInst(NewShift, I.getType());
674 }
Chris Lattner92295c52004-03-12 23:53:13 +0000675 }
676 }
Chris Lattner022167f2004-03-13 00:11:49 +0000677 }
Chris Lattner183b3362004-04-09 19:05:30 +0000678
679 // Try to fold constant sub into select arguments.
680 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
681 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
682 return R;
Chris Lattner8f2f5982003-11-05 01:06:05 +0000683 }
684
Chris Lattner3082c5a2003-02-18 19:28:33 +0000685 if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1))
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000686 if (Op1I->hasOneUse()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000687 // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression
688 // is not used by anyone else...
689 //
Chris Lattnerc2f0aa52004-02-02 20:09:56 +0000690 if (Op1I->getOpcode() == Instruction::Sub &&
691 !Op1I->getType()->isFloatingPoint()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000692 // Swap the two operands of the subexpr...
693 Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1);
694 Op1I->setOperand(0, IIOp1);
695 Op1I->setOperand(1, IIOp0);
696
697 // Create the new top level add instruction...
698 return BinaryOperator::create(Instruction::Add, Op0, Op1);
699 }
700
701 // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)...
702 //
703 if (Op1I->getOpcode() == Instruction::And &&
704 (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) {
705 Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0);
706
Chris Lattner396dbfe2004-06-09 05:08:07 +0000707 Value *NewNot =
708 InsertNewInstBefore(BinaryOperator::createNot(OtherOp, "B.not"), I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000709 return BinaryOperator::create(Instruction::And, Op0, NewNot);
710 }
Chris Lattner57c8d992003-02-18 19:57:07 +0000711
712 // X - X*C --> X * (1-C)
713 if (dyn_castFoldableMul(Op1I) == Op0) {
Chris Lattner34428442003-05-27 16:40:51 +0000714 Constant *CP1 =
715 ConstantExpr::get(Instruction::Sub,
716 ConstantInt::get(I.getType(), 1),
717 cast<Constant>(cast<Instruction>(Op1)->getOperand(1)));
Chris Lattner57c8d992003-02-18 19:57:07 +0000718 assert(CP1 && "Couldn't constant fold 1-C?");
719 return BinaryOperator::create(Instruction::Mul, Op0, CP1);
720 }
Chris Lattnerad3c4952002-05-09 01:29:19 +0000721 }
Chris Lattner3082c5a2003-02-18 19:28:33 +0000722
Chris Lattner57c8d992003-02-18 19:57:07 +0000723 // X*C - X --> X * (C-1)
724 if (dyn_castFoldableMul(Op0) == Op1) {
Chris Lattner34428442003-05-27 16:40:51 +0000725 Constant *CP1 =
726 ConstantExpr::get(Instruction::Sub,
727 cast<Constant>(cast<Instruction>(Op0)->getOperand(1)),
728 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000729 assert(CP1 && "Couldn't constant fold C - 1?");
730 return BinaryOperator::create(Instruction::Mul, Op1, CP1);
731 }
732
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000733 return 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000734}
735
Chris Lattnere79e8542004-02-23 06:38:22 +0000736/// isSignBitCheck - Given an exploded setcc instruction, return true if it is
737/// really just returns true if the most significant (sign) bit is set.
738static bool isSignBitCheck(unsigned Opcode, Value *LHS, ConstantInt *RHS) {
739 if (RHS->getType()->isSigned()) {
740 // True if source is LHS < 0 or LHS <= -1
741 return Opcode == Instruction::SetLT && RHS->isNullValue() ||
742 Opcode == Instruction::SetLE && RHS->isAllOnesValue();
743 } else {
744 ConstantUInt *RHSC = cast<ConstantUInt>(RHS);
745 // True if source is LHS > 127 or LHS >= 128, where the constants depend on
746 // the size of the integer type.
747 if (Opcode == Instruction::SetGE)
748 return RHSC->getValue() == 1ULL<<(RHS->getType()->getPrimitiveSize()*8-1);
749 if (Opcode == Instruction::SetGT)
750 return RHSC->getValue() ==
751 (1ULL << (RHS->getType()->getPrimitiveSize()*8-1))-1;
752 }
753 return false;
754}
755
Chris Lattner113f4f42002-06-25 16:13:24 +0000756Instruction *InstCombiner::visitMul(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000757 bool Changed = SimplifyCommutative(I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000758 Value *Op0 = I.getOperand(0);
Chris Lattner260ab202002-04-18 17:39:14 +0000759
Chris Lattnere6794492002-08-12 21:17:25 +0000760 // Simplify mul instructions with a constant RHS...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000761 if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) {
762 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerede3fe02003-08-13 04:18:28 +0000763
764 // ((X << C1)*C2) == (X * (C2 << C1))
765 if (ShiftInst *SI = dyn_cast<ShiftInst>(Op0))
766 if (SI->getOpcode() == Instruction::Shl)
767 if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1)))
768 return BinaryOperator::create(Instruction::Mul, SI->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000769 ConstantExpr::get(Instruction::Shl, CI, ShOp));
770
Chris Lattnercce81be2003-09-11 22:24:54 +0000771 if (CI->isNullValue())
772 return ReplaceInstUsesWith(I, Op1); // X * 0 == 0
773 if (CI->equalsInt(1)) // X * 1 == X
774 return ReplaceInstUsesWith(I, Op0);
775 if (CI->isAllOnesValue()) // X * -1 == 0 - X
Chris Lattner35236d82003-06-25 17:09:20 +0000776 return BinaryOperator::createNeg(Op0, I.getName());
Chris Lattner31ba1292002-04-29 22:24:47 +0000777
Chris Lattnercce81be2003-09-11 22:24:54 +0000778 int64_t Val = (int64_t)cast<ConstantInt>(CI)->getRawValue();
Chris Lattner3082c5a2003-02-18 19:28:33 +0000779 if (uint64_t C = Log2(Val)) // Replace X*(2^C) with X << C
780 return new ShiftInst(Instruction::Shl, Op0,
781 ConstantUInt::get(Type::UByteTy, C));
782 } else {
783 ConstantFP *Op1F = cast<ConstantFP>(Op1);
784 if (Op1F->isNullValue())
785 return ReplaceInstUsesWith(I, Op1);
Chris Lattner31ba1292002-04-29 22:24:47 +0000786
Chris Lattner3082c5a2003-02-18 19:28:33 +0000787 // "In IEEE floating point, x*1 is not equivalent to x for nans. However,
788 // ANSI says we can drop signals, so we can do this anyway." (from GCC)
789 if (Op1F->getValue() == 1.0)
790 return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0'
791 }
Chris Lattner183b3362004-04-09 19:05:30 +0000792
793 // Try to fold constant mul into select arguments.
794 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
795 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
796 return R;
Chris Lattner260ab202002-04-18 17:39:14 +0000797 }
798
Chris Lattner934a64cf2003-03-10 23:23:04 +0000799 if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y
800 if (Value *Op1v = dyn_castNegVal(I.getOperand(1)))
801 return BinaryOperator::create(Instruction::Mul, Op0v, Op1v);
802
Chris Lattner2635b522004-02-23 05:39:21 +0000803 // If one of the operands of the multiply is a cast from a boolean value, then
804 // we know the bool is either zero or one, so this is a 'masking' multiply.
805 // See if we can simplify things based on how the boolean was originally
806 // formed.
807 CastInst *BoolCast = 0;
808 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(0)))
809 if (CI->getOperand(0)->getType() == Type::BoolTy)
810 BoolCast = CI;
811 if (!BoolCast)
812 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(1)))
813 if (CI->getOperand(0)->getType() == Type::BoolTy)
814 BoolCast = CI;
815 if (BoolCast) {
816 if (SetCondInst *SCI = dyn_cast<SetCondInst>(BoolCast->getOperand(0))) {
817 Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1);
818 const Type *SCOpTy = SCIOp0->getType();
819
Chris Lattnere79e8542004-02-23 06:38:22 +0000820 // If the setcc is true iff the sign bit of X is set, then convert this
821 // multiply into a shift/and combination.
822 if (isa<ConstantInt>(SCIOp1) &&
823 isSignBitCheck(SCI->getOpcode(), SCIOp0, cast<ConstantInt>(SCIOp1))) {
Chris Lattner2635b522004-02-23 05:39:21 +0000824 // Shift the X value right to turn it into "all signbits".
825 Constant *Amt = ConstantUInt::get(Type::UByteTy,
826 SCOpTy->getPrimitiveSize()*8-1);
Chris Lattnere79e8542004-02-23 06:38:22 +0000827 if (SCIOp0->getType()->isUnsigned()) {
828 const Type *NewTy = getSignedIntegralType(SCIOp0->getType());
829 SCIOp0 = InsertNewInstBefore(new CastInst(SCIOp0, NewTy,
830 SCIOp0->getName()), I);
831 }
832
833 Value *V =
834 InsertNewInstBefore(new ShiftInst(Instruction::Shr, SCIOp0, Amt,
835 BoolCast->getOperand(0)->getName()+
836 ".mask"), I);
Chris Lattner2635b522004-02-23 05:39:21 +0000837
838 // If the multiply type is not the same as the source type, sign extend
839 // or truncate to the multiply type.
840 if (I.getType() != V->getType())
Chris Lattnere79e8542004-02-23 06:38:22 +0000841 V = InsertNewInstBefore(new CastInst(V, I.getType(), V->getName()),I);
Chris Lattner2635b522004-02-23 05:39:21 +0000842
843 Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0;
844 return BinaryOperator::create(Instruction::And, V, OtherOp);
845 }
846 }
847 }
848
Chris Lattner113f4f42002-06-25 16:13:24 +0000849 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000850}
851
Chris Lattner113f4f42002-06-25 16:13:24 +0000852Instruction *InstCombiner::visitDiv(BinaryOperator &I) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000853 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
Chris Lattnere20c3342004-04-26 14:01:59 +0000854 // div X, 1 == X
Chris Lattnere6794492002-08-12 21:17:25 +0000855 if (RHS->equalsInt(1))
856 return ReplaceInstUsesWith(I, I.getOperand(0));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000857
Chris Lattnere20c3342004-04-26 14:01:59 +0000858 // div X, -1 == -X
859 if (RHS->isAllOnesValue())
860 return BinaryOperator::createNeg(I.getOperand(0));
861
Chris Lattner3082c5a2003-02-18 19:28:33 +0000862 // Check to see if this is an unsigned division with an exact power of 2,
863 // if so, convert to a right shift.
864 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
865 if (uint64_t Val = C->getValue()) // Don't break X / 0
866 if (uint64_t C = Log2(Val))
867 return new ShiftInst(Instruction::Shr, I.getOperand(0),
868 ConstantUInt::get(Type::UByteTy, C));
869 }
870
871 // 0 / X == 0, we don't need to preserve faults!
872 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
873 if (LHS->equalsInt(0))
874 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
875
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000876 return 0;
877}
878
879
Chris Lattner113f4f42002-06-25 16:13:24 +0000880Instruction *InstCombiner::visitRem(BinaryOperator &I) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000881 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
882 if (RHS->equalsInt(1)) // X % 1 == 0
883 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner56b50512004-03-26 16:11:24 +0000884 if (RHS->isAllOnesValue()) // X % -1 == 0
885 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000886
887 // Check to see if this is an unsigned remainder with an exact power of 2,
888 // if so, convert to a bitwise and.
889 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
890 if (uint64_t Val = C->getValue()) // Don't break X % 0 (divide by zero)
Chris Lattnerd9e58132004-05-07 15:35:56 +0000891 if (!(Val & (Val-1))) // Power of 2
Chris Lattner3082c5a2003-02-18 19:28:33 +0000892 return BinaryOperator::create(Instruction::And, I.getOperand(0),
893 ConstantUInt::get(I.getType(), Val-1));
894 }
895
896 // 0 % X == 0, we don't need to preserve faults!
897 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
898 if (LHS->equalsInt(0))
Chris Lattnere6794492002-08-12 21:17:25 +0000899 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
900
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000901 return 0;
902}
903
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000904// isMaxValueMinusOne - return true if this is Max-1
Chris Lattnere6794492002-08-12 21:17:25 +0000905static bool isMaxValueMinusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000906 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C)) {
907 // Calculate -1 casted to the right type...
908 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
909 uint64_t Val = ~0ULL; // All ones
910 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
911 return CU->getValue() == Val-1;
912 }
913
914 const ConstantSInt *CS = cast<ConstantSInt>(C);
915
916 // Calculate 0111111111..11111
917 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
918 int64_t Val = INT64_MAX; // All ones
919 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
920 return CS->getValue() == Val-1;
921}
922
923// isMinValuePlusOne - return true if this is Min+1
Chris Lattnere6794492002-08-12 21:17:25 +0000924static bool isMinValuePlusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000925 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C))
926 return CU->getValue() == 1;
927
928 const ConstantSInt *CS = cast<ConstantSInt>(C);
929
930 // Calculate 1111111111000000000000
931 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
932 int64_t Val = -1; // All ones
933 Val <<= TypeBits-1; // Shift over to the right spot
934 return CS->getValue() == Val+1;
935}
936
Chris Lattner35167c32004-06-09 07:59:58 +0000937// isOneBitSet - Return true if there is exactly one bit set in the specified
938// constant.
939static bool isOneBitSet(const ConstantInt *CI) {
940 uint64_t V = CI->getRawValue();
941 return V && (V & (V-1)) == 0;
942}
943
Chris Lattner3ac7c262003-08-13 20:16:26 +0000944/// getSetCondCode - Encode a setcc opcode into a three bit mask. These bits
945/// are carefully arranged to allow folding of expressions such as:
946///
947/// (A < B) | (A > B) --> (A != B)
948///
949/// Bit value '4' represents that the comparison is true if A > B, bit value '2'
950/// represents that the comparison is true if A == B, and bit value '1' is true
951/// if A < B.
952///
953static unsigned getSetCondCode(const SetCondInst *SCI) {
954 switch (SCI->getOpcode()) {
955 // False -> 0
956 case Instruction::SetGT: return 1;
957 case Instruction::SetEQ: return 2;
958 case Instruction::SetGE: return 3;
959 case Instruction::SetLT: return 4;
960 case Instruction::SetNE: return 5;
961 case Instruction::SetLE: return 6;
962 // True -> 7
963 default:
964 assert(0 && "Invalid SetCC opcode!");
965 return 0;
966 }
967}
968
969/// getSetCCValue - This is the complement of getSetCondCode, which turns an
970/// opcode and two operands into either a constant true or false, or a brand new
971/// SetCC instruction.
972static Value *getSetCCValue(unsigned Opcode, Value *LHS, Value *RHS) {
973 switch (Opcode) {
974 case 0: return ConstantBool::False;
975 case 1: return new SetCondInst(Instruction::SetGT, LHS, RHS);
976 case 2: return new SetCondInst(Instruction::SetEQ, LHS, RHS);
977 case 3: return new SetCondInst(Instruction::SetGE, LHS, RHS);
978 case 4: return new SetCondInst(Instruction::SetLT, LHS, RHS);
979 case 5: return new SetCondInst(Instruction::SetNE, LHS, RHS);
980 case 6: return new SetCondInst(Instruction::SetLE, LHS, RHS);
981 case 7: return ConstantBool::True;
982 default: assert(0 && "Illegal SetCCCode!"); return 0;
983 }
984}
985
986// FoldSetCCLogical - Implements (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
987struct FoldSetCCLogical {
988 InstCombiner &IC;
989 Value *LHS, *RHS;
990 FoldSetCCLogical(InstCombiner &ic, SetCondInst *SCI)
991 : IC(ic), LHS(SCI->getOperand(0)), RHS(SCI->getOperand(1)) {}
992 bool shouldApply(Value *V) const {
993 if (SetCondInst *SCI = dyn_cast<SetCondInst>(V))
994 return (SCI->getOperand(0) == LHS && SCI->getOperand(1) == RHS ||
995 SCI->getOperand(0) == RHS && SCI->getOperand(1) == LHS);
996 return false;
997 }
998 Instruction *apply(BinaryOperator &Log) const {
999 SetCondInst *SCI = cast<SetCondInst>(Log.getOperand(0));
1000 if (SCI->getOperand(0) != LHS) {
1001 assert(SCI->getOperand(1) == LHS);
1002 SCI->swapOperands(); // Swap the LHS and RHS of the SetCC
1003 }
1004
1005 unsigned LHSCode = getSetCondCode(SCI);
1006 unsigned RHSCode = getSetCondCode(cast<SetCondInst>(Log.getOperand(1)));
1007 unsigned Code;
1008 switch (Log.getOpcode()) {
1009 case Instruction::And: Code = LHSCode & RHSCode; break;
1010 case Instruction::Or: Code = LHSCode | RHSCode; break;
1011 case Instruction::Xor: Code = LHSCode ^ RHSCode; break;
Chris Lattner2caaaba2003-09-22 20:33:34 +00001012 default: assert(0 && "Illegal logical opcode!"); return 0;
Chris Lattner3ac7c262003-08-13 20:16:26 +00001013 }
1014
1015 Value *RV = getSetCCValue(Code, LHS, RHS);
1016 if (Instruction *I = dyn_cast<Instruction>(RV))
1017 return I;
1018 // Otherwise, it's a constant boolean value...
1019 return IC.ReplaceInstUsesWith(Log, RV);
1020 }
1021};
1022
1023
Chris Lattnerba1cb382003-09-19 17:17:26 +00001024// OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where
1025// the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is
1026// guaranteed to be either a shift instruction or a binary operator.
1027Instruction *InstCombiner::OptAndOp(Instruction *Op,
1028 ConstantIntegral *OpRHS,
1029 ConstantIntegral *AndRHS,
1030 BinaryOperator &TheAnd) {
1031 Value *X = Op->getOperand(0);
Chris Lattnerfcf21a72004-01-12 19:47:05 +00001032 Constant *Together = 0;
1033 if (!isa<ShiftInst>(Op))
1034 Together = ConstantExpr::get(Instruction::And, AndRHS, OpRHS);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001035
Chris Lattnerba1cb382003-09-19 17:17:26 +00001036 switch (Op->getOpcode()) {
1037 case Instruction::Xor:
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001038 if (Together->isNullValue()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001039 // (X ^ C1) & C2 --> (X & C2) iff (C1&C2) == 0
1040 return BinaryOperator::create(Instruction::And, X, AndRHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001041 } else if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001042 // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2)
1043 std::string OpName = Op->getName(); Op->setName("");
1044 Instruction *And = BinaryOperator::create(Instruction::And,
1045 X, AndRHS, OpName);
1046 InsertNewInstBefore(And, TheAnd);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001047 return BinaryOperator::create(Instruction::Xor, And, Together);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001048 }
1049 break;
1050 case Instruction::Or:
1051 // (X | C1) & C2 --> X & C2 iff C1 & C1 == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001052 if (Together->isNullValue())
Chris Lattnerba1cb382003-09-19 17:17:26 +00001053 return BinaryOperator::create(Instruction::And, X, AndRHS);
1054 else {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001055 if (Together == AndRHS) // (X | C) & C --> C
1056 return ReplaceInstUsesWith(TheAnd, AndRHS);
1057
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001058 if (Op->hasOneUse() && Together != OpRHS) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001059 // (X | C1) & C2 --> (X | (C1&C2)) & C2
1060 std::string Op0Name = Op->getName(); Op->setName("");
1061 Instruction *Or = BinaryOperator::create(Instruction::Or, X,
1062 Together, Op0Name);
1063 InsertNewInstBefore(Or, TheAnd);
1064 return BinaryOperator::create(Instruction::And, Or, AndRHS);
1065 }
1066 }
1067 break;
1068 case Instruction::Add:
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001069 if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001070 // Adding a one to a single bit bit-field should be turned into an XOR
1071 // of the bit. First thing to check is to see if this AND is with a
1072 // single bit constant.
Chris Lattner35167c32004-06-09 07:59:58 +00001073 uint64_t AndRHSV = cast<ConstantInt>(AndRHS)->getRawValue();
Chris Lattnerba1cb382003-09-19 17:17:26 +00001074
1075 // Clear bits that are not part of the constant.
1076 AndRHSV &= (1ULL << AndRHS->getType()->getPrimitiveSize()*8)-1;
1077
1078 // If there is only one bit set...
Chris Lattner35167c32004-06-09 07:59:58 +00001079 if (isOneBitSet(cast<ConstantInt>(AndRHS))) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001080 // Ok, at this point, we know that we are masking the result of the
1081 // ADD down to exactly one bit. If the constant we are adding has
1082 // no bits set below this bit, then we can eliminate the ADD.
Chris Lattner35167c32004-06-09 07:59:58 +00001083 uint64_t AddRHS = cast<ConstantInt>(OpRHS)->getRawValue();
Chris Lattnerba1cb382003-09-19 17:17:26 +00001084
1085 // Check to see if any bits below the one bit set in AndRHSV are set.
1086 if ((AddRHS & (AndRHSV-1)) == 0) {
1087 // If not, the only thing that can effect the output of the AND is
1088 // the bit specified by AndRHSV. If that bit is set, the effect of
1089 // the XOR is to toggle the bit. If it is clear, then the ADD has
1090 // no effect.
1091 if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop
1092 TheAnd.setOperand(0, X);
1093 return &TheAnd;
1094 } else {
1095 std::string Name = Op->getName(); Op->setName("");
1096 // Pull the XOR out of the AND.
1097 Instruction *NewAnd =
1098 BinaryOperator::create(Instruction::And, X, AndRHS, Name);
1099 InsertNewInstBefore(NewAnd, TheAnd);
1100 return BinaryOperator::create(Instruction::Xor, NewAnd, AndRHS);
1101 }
1102 }
1103 }
1104 }
1105 break;
Chris Lattner2da29172003-09-19 19:05:02 +00001106
1107 case Instruction::Shl: {
1108 // We know that the AND will not produce any of the bits shifted in, so if
1109 // the anded constant includes them, clear them now!
1110 //
1111 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001112 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
1113 ConstantExpr::get(Instruction::Shl, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001114 if (CI != AndRHS) {
1115 TheAnd.setOperand(1, CI);
1116 return &TheAnd;
1117 }
1118 break;
1119 }
1120 case Instruction::Shr:
1121 // We know that the AND will not produce any of the bits shifted in, so if
1122 // the anded constant includes them, clear them now! This only applies to
1123 // unsigned shifts, because a signed shr may bring in set bits!
1124 //
1125 if (AndRHS->getType()->isUnsigned()) {
1126 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001127 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
1128 ConstantExpr::get(Instruction::Shr, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001129 if (CI != AndRHS) {
1130 TheAnd.setOperand(1, CI);
1131 return &TheAnd;
1132 }
1133 }
1134 break;
Chris Lattnerba1cb382003-09-19 17:17:26 +00001135 }
1136 return 0;
1137}
1138
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001139
Chris Lattner113f4f42002-06-25 16:13:24 +00001140Instruction *InstCombiner::visitAnd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001141 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001142 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001143
1144 // and X, X = X and X, 0 == 0
Chris Lattnere6794492002-08-12 21:17:25 +00001145 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1146 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001147
1148 // and X, -1 == X
Chris Lattner49b47ae2003-07-23 17:57:01 +00001149 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001150 if (RHS->isAllOnesValue())
1151 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001152
Chris Lattnerba1cb382003-09-19 17:17:26 +00001153 // Optimize a variety of ((val OP C1) & C2) combinations...
1154 if (isa<BinaryOperator>(Op0) || isa<ShiftInst>(Op0)) {
1155 Instruction *Op0I = cast<Instruction>(Op0);
Chris Lattner33217db2003-07-23 19:36:21 +00001156 Value *X = Op0I->getOperand(0);
Chris Lattner16464b32003-07-23 19:25:52 +00001157 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnerba1cb382003-09-19 17:17:26 +00001158 if (Instruction *Res = OptAndOp(Op0I, Op0CI, RHS, I))
1159 return Res;
Chris Lattner33217db2003-07-23 19:36:21 +00001160 }
Chris Lattner183b3362004-04-09 19:05:30 +00001161
1162 // Try to fold constant and into select arguments.
1163 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1164 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1165 return R;
Chris Lattner49b47ae2003-07-23 17:57:01 +00001166 }
1167
Chris Lattnerbb74e222003-03-10 23:06:50 +00001168 Value *Op0NotVal = dyn_castNotVal(Op0);
1169 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001170
1171 // (~A & ~B) == (~(A | B)) - Demorgan's Law
Chris Lattnerbb74e222003-03-10 23:06:50 +00001172 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattner3082c5a2003-02-18 19:28:33 +00001173 Instruction *Or = BinaryOperator::create(Instruction::Or, Op0NotVal,
Chris Lattner49b47ae2003-07-23 17:57:01 +00001174 Op1NotVal,I.getName()+".demorgan");
1175 InsertNewInstBefore(Or, I);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001176 return BinaryOperator::createNot(Or);
1177 }
1178
1179 if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0
1180 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner65217ff2002-08-23 18:32:43 +00001181
Chris Lattner3ac7c262003-08-13 20:16:26 +00001182 // (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
1183 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1184 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1185 return R;
1186
Chris Lattner113f4f42002-06-25 16:13:24 +00001187 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001188}
1189
1190
1191
Chris Lattner113f4f42002-06-25 16:13:24 +00001192Instruction *InstCombiner::visitOr(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001193 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001194 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001195
1196 // or X, X = X or X, 0 == X
Chris Lattnere6794492002-08-12 21:17:25 +00001197 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1198 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001199
1200 // or X, -1 == -1
Chris Lattner8f0d1562003-07-23 18:29:44 +00001201 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001202 if (RHS->isAllOnesValue())
1203 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001204
Chris Lattner8f0d1562003-07-23 18:29:44 +00001205 if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) {
1206 // (X & C1) | C2 --> (X | C2) & (C1|C2)
1207 if (Op0I->getOpcode() == Instruction::And && isOnlyUse(Op0))
1208 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1209 std::string Op0Name = Op0I->getName(); Op0I->setName("");
1210 Instruction *Or = BinaryOperator::create(Instruction::Or,
1211 Op0I->getOperand(0), RHS,
1212 Op0Name);
1213 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001214 return BinaryOperator::create(Instruction::And, Or,
1215 ConstantExpr::get(Instruction::Or, RHS, Op0CI));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001216 }
1217
1218 // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2)
1219 if (Op0I->getOpcode() == Instruction::Xor && isOnlyUse(Op0))
1220 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1221 std::string Op0Name = Op0I->getName(); Op0I->setName("");
1222 Instruction *Or = BinaryOperator::create(Instruction::Or,
1223 Op0I->getOperand(0), RHS,
1224 Op0Name);
1225 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001226 return BinaryOperator::create(Instruction::Xor, Or,
1227 ConstantExpr::get(Instruction::And, Op0CI,
1228 NotConstant(RHS)));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001229 }
1230 }
Chris Lattner183b3362004-04-09 19:05:30 +00001231
1232 // Try to fold constant and into select arguments.
1233 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1234 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1235 return R;
Chris Lattner8f0d1562003-07-23 18:29:44 +00001236 }
1237
Chris Lattner812aab72003-08-12 19:11:07 +00001238 // (A & C1)|(A & C2) == A & (C1|C2)
Chris Lattner01d56392003-08-12 19:17:27 +00001239 if (Instruction *LHS = dyn_cast<BinaryOperator>(Op0))
1240 if (Instruction *RHS = dyn_cast<BinaryOperator>(Op1))
1241 if (LHS->getOperand(0) == RHS->getOperand(0))
1242 if (Constant *C0 = dyn_castMaskingAnd(LHS))
1243 if (Constant *C1 = dyn_castMaskingAnd(RHS))
1244 return BinaryOperator::create(Instruction::And, LHS->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001245 ConstantExpr::get(Instruction::Or, C0, C1));
Chris Lattner812aab72003-08-12 19:11:07 +00001246
Chris Lattner3e327a42003-03-10 23:13:59 +00001247 Value *Op0NotVal = dyn_castNotVal(Op0);
1248 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001249
Chris Lattner3e327a42003-03-10 23:13:59 +00001250 if (Op1 == Op0NotVal) // ~A | A == -1
1251 return ReplaceInstUsesWith(I,
1252 ConstantIntegral::getAllOnesValue(I.getType()));
1253
1254 if (Op0 == Op1NotVal) // A | ~A == -1
1255 return ReplaceInstUsesWith(I,
1256 ConstantIntegral::getAllOnesValue(I.getType()));
1257
1258 // (~A | ~B) == (~(A & B)) - Demorgan's Law
1259 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattner396dbfe2004-06-09 05:08:07 +00001260 Value *And = InsertNewInstBefore(
1261 BinaryOperator::create(Instruction::And, Op0NotVal,
1262 Op1NotVal,I.getName()+".demorgan"), I);
Chris Lattner3e327a42003-03-10 23:13:59 +00001263 return BinaryOperator::createNot(And);
1264 }
Chris Lattner3082c5a2003-02-18 19:28:33 +00001265
Chris Lattner3ac7c262003-08-13 20:16:26 +00001266 // (setcc1 A, B) | (setcc2 A, B) --> (setcc3 A, B)
1267 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1268 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1269 return R;
1270
Chris Lattner113f4f42002-06-25 16:13:24 +00001271 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001272}
1273
Chris Lattnerc2076352004-02-16 01:20:27 +00001274// XorSelf - Implements: X ^ X --> 0
1275struct XorSelf {
1276 Value *RHS;
1277 XorSelf(Value *rhs) : RHS(rhs) {}
1278 bool shouldApply(Value *LHS) const { return LHS == RHS; }
1279 Instruction *apply(BinaryOperator &Xor) const {
1280 return &Xor;
1281 }
1282};
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001283
1284
Chris Lattner113f4f42002-06-25 16:13:24 +00001285Instruction *InstCombiner::visitXor(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001286 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001287 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001288
Chris Lattnerc2076352004-02-16 01:20:27 +00001289 // xor X, X = 0, even if X is nested in a sequence of Xor's.
1290 if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) {
1291 assert(Result == &I && "AssociativeOpt didn't work?");
Chris Lattnere6794492002-08-12 21:17:25 +00001292 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattnerc2076352004-02-16 01:20:27 +00001293 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001294
Chris Lattner97638592003-07-23 21:37:07 +00001295 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001296 // xor X, 0 == X
Chris Lattner97638592003-07-23 21:37:07 +00001297 if (RHS->isNullValue())
Chris Lattnere6794492002-08-12 21:17:25 +00001298 return ReplaceInstUsesWith(I, Op0);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001299
Chris Lattner97638592003-07-23 21:37:07 +00001300 if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) {
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001301 // xor (setcc A, B), true = not (setcc A, B) = setncc A, B
Chris Lattner97638592003-07-23 21:37:07 +00001302 if (SetCondInst *SCI = dyn_cast<SetCondInst>(Op0I))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001303 if (RHS == ConstantBool::True && SCI->hasOneUse())
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001304 return new SetCondInst(SCI->getInverseCondition(),
1305 SCI->getOperand(0), SCI->getOperand(1));
Chris Lattnere5806662003-11-04 23:50:51 +00001306
Chris Lattner8f2f5982003-11-05 01:06:05 +00001307 // ~(c-X) == X-c-1 == X+(-c-1)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001308 if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue())
1309 if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) {
1310 Constant *NegOp0I0C = ConstantExpr::get(Instruction::Sub,
1311 Constant::getNullValue(Op0I0C->getType()), Op0I0C);
1312 Constant *ConstantRHS = ConstantExpr::get(Instruction::Sub, NegOp0I0C,
1313 ConstantInt::get(I.getType(), 1));
1314 return BinaryOperator::create(Instruction::Add, Op0I->getOperand(1),
1315 ConstantRHS);
1316 }
Chris Lattner97638592003-07-23 21:37:07 +00001317
1318 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnere5806662003-11-04 23:50:51 +00001319 switch (Op0I->getOpcode()) {
1320 case Instruction::Add:
Chris Lattner0f68fa62003-11-04 23:37:10 +00001321 // ~(X-c) --> (-c-1)-X
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001322 if (RHS->isAllOnesValue()) {
1323 Constant *NegOp0CI = ConstantExpr::get(Instruction::Sub,
1324 Constant::getNullValue(Op0CI->getType()), Op0CI);
Chris Lattner0f68fa62003-11-04 23:37:10 +00001325 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001326 ConstantExpr::get(Instruction::Sub, NegOp0CI,
1327 ConstantInt::get(I.getType(), 1)),
Chris Lattner0f68fa62003-11-04 23:37:10 +00001328 Op0I->getOperand(0));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001329 }
Chris Lattnere5806662003-11-04 23:50:51 +00001330 break;
1331 case Instruction::And:
Chris Lattner97638592003-07-23 21:37:07 +00001332 // (X & C1) ^ C2 --> (X & C1) | C2 iff (C1&C2) == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001333 if (ConstantExpr::get(Instruction::And, RHS, Op0CI)->isNullValue())
Chris Lattner97638592003-07-23 21:37:07 +00001334 return BinaryOperator::create(Instruction::Or, Op0, RHS);
Chris Lattnere5806662003-11-04 23:50:51 +00001335 break;
1336 case Instruction::Or:
Chris Lattner97638592003-07-23 21:37:07 +00001337 // (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001338 if (ConstantExpr::get(Instruction::And, RHS, Op0CI) == RHS)
1339 return BinaryOperator::create(Instruction::And, Op0,
1340 NotConstant(RHS));
Chris Lattnere5806662003-11-04 23:50:51 +00001341 break;
1342 default: break;
Chris Lattner97638592003-07-23 21:37:07 +00001343 }
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001344 }
Chris Lattner183b3362004-04-09 19:05:30 +00001345
1346 // Try to fold constant and into select arguments.
1347 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1348 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1349 return R;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001350 }
1351
Chris Lattnerbb74e222003-03-10 23:06:50 +00001352 if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001353 if (X == Op1)
1354 return ReplaceInstUsesWith(I,
1355 ConstantIntegral::getAllOnesValue(I.getType()));
1356
Chris Lattnerbb74e222003-03-10 23:06:50 +00001357 if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001358 if (X == Op0)
1359 return ReplaceInstUsesWith(I,
1360 ConstantIntegral::getAllOnesValue(I.getType()));
1361
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001362 if (Instruction *Op1I = dyn_cast<Instruction>(Op1))
Chris Lattnerb36d9082004-02-16 03:54:20 +00001363 if (Op1I->getOpcode() == Instruction::Or) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001364 if (Op1I->getOperand(0) == Op0) { // B^(B|A) == (A|B)^B
1365 cast<BinaryOperator>(Op1I)->swapOperands();
1366 I.swapOperands();
1367 std::swap(Op0, Op1);
1368 } else if (Op1I->getOperand(1) == Op0) { // B^(A|B) == (A|B)^B
1369 I.swapOperands();
1370 std::swap(Op0, Op1);
Chris Lattnerb36d9082004-02-16 03:54:20 +00001371 }
1372 } else if (Op1I->getOpcode() == Instruction::Xor) {
1373 if (Op0 == Op1I->getOperand(0)) // A^(A^B) == B
1374 return ReplaceInstUsesWith(I, Op1I->getOperand(1));
1375 else if (Op0 == Op1I->getOperand(1)) // A^(B^A) == B
1376 return ReplaceInstUsesWith(I, Op1I->getOperand(0));
1377 }
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001378
1379 if (Instruction *Op0I = dyn_cast<Instruction>(Op0))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001380 if (Op0I->getOpcode() == Instruction::Or && Op0I->hasOneUse()) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001381 if (Op0I->getOperand(0) == Op1) // (B|A)^B == (A|B)^B
1382 cast<BinaryOperator>(Op0I)->swapOperands();
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001383 if (Op0I->getOperand(1) == Op1) { // (A|B)^B == A & ~B
Chris Lattner396dbfe2004-06-09 05:08:07 +00001384 Value *NotB = InsertNewInstBefore(BinaryOperator::createNot(Op1,
1385 Op1->getName()+".not"), I);
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001386 return BinaryOperator::create(Instruction::And, Op0I->getOperand(0),
1387 NotB);
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001388 }
Chris Lattnerb36d9082004-02-16 03:54:20 +00001389 } else if (Op0I->getOpcode() == Instruction::Xor) {
1390 if (Op1 == Op0I->getOperand(0)) // (A^B)^A == B
1391 return ReplaceInstUsesWith(I, Op0I->getOperand(1));
1392 else if (Op1 == Op0I->getOperand(1)) // (B^A)^A == B
1393 return ReplaceInstUsesWith(I, Op0I->getOperand(0));
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001394 }
1395
Chris Lattner7fb29e12003-03-11 00:12:48 +00001396 // (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1^C2 == 0
1397 if (Constant *C1 = dyn_castMaskingAnd(Op0))
1398 if (Constant *C2 = dyn_castMaskingAnd(Op1))
Chris Lattner34428442003-05-27 16:40:51 +00001399 if (ConstantExpr::get(Instruction::And, C1, C2)->isNullValue())
Chris Lattner7fb29e12003-03-11 00:12:48 +00001400 return BinaryOperator::create(Instruction::Or, Op0, Op1);
1401
Chris Lattner3ac7c262003-08-13 20:16:26 +00001402 // (setcc1 A, B) ^ (setcc2 A, B) --> (setcc3 A, B)
1403 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1404 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1405 return R;
1406
Chris Lattner113f4f42002-06-25 16:13:24 +00001407 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001408}
1409
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001410// AddOne, SubOne - Add or subtract a constant one from an integer constant...
1411static Constant *AddOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001412 Constant *Result = ConstantExpr::get(Instruction::Add, C,
1413 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001414 assert(Result && "Constant folding integer addition failed!");
1415 return Result;
1416}
1417static Constant *SubOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001418 Constant *Result = ConstantExpr::get(Instruction::Sub, C,
1419 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001420 assert(Result && "Constant folding integer addition failed!");
1421 return Result;
1422}
1423
Chris Lattner1fc23f32002-05-09 20:11:54 +00001424// isTrueWhenEqual - Return true if the specified setcondinst instruction is
1425// true when both operands are equal...
1426//
Chris Lattner113f4f42002-06-25 16:13:24 +00001427static bool isTrueWhenEqual(Instruction &I) {
1428 return I.getOpcode() == Instruction::SetEQ ||
1429 I.getOpcode() == Instruction::SetGE ||
1430 I.getOpcode() == Instruction::SetLE;
Chris Lattner1fc23f32002-05-09 20:11:54 +00001431}
1432
Chris Lattner113f4f42002-06-25 16:13:24 +00001433Instruction *InstCombiner::visitSetCondInst(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001434 bool Changed = SimplifyCommutative(I);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001435 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
1436 const Type *Ty = Op0->getType();
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001437
1438 // setcc X, X
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001439 if (Op0 == Op1)
1440 return ReplaceInstUsesWith(I, ConstantBool::get(isTrueWhenEqual(I)));
Chris Lattner1fc23f32002-05-09 20:11:54 +00001441
Chris Lattnerd07283a2003-08-13 05:38:46 +00001442 // setcc <global/alloca*>, 0 - Global/Stack value addresses are never null!
1443 if (isa<ConstantPointerNull>(Op1) &&
1444 (isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0)))
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001445 return ReplaceInstUsesWith(I, ConstantBool::get(!isTrueWhenEqual(I)));
1446
Chris Lattnerd07283a2003-08-13 05:38:46 +00001447
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001448 // setcc's with boolean values can always be turned into bitwise operations
1449 if (Ty == Type::BoolTy) {
1450 // If this is <, >, or !=, we can change this into a simple xor instruction
1451 if (!isTrueWhenEqual(I))
Chris Lattner16930792003-11-03 04:25:02 +00001452 return BinaryOperator::create(Instruction::Xor, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001453
1454 // Otherwise we need to make a temporary intermediate instruction and insert
1455 // it into the instruction stream. This is what we are after:
1456 //
1457 // seteq bool %A, %B -> ~(A^B)
1458 // setle bool %A, %B -> ~A | B
1459 // setge bool %A, %B -> A | ~B
1460 //
1461 if (I.getOpcode() == Instruction::SetEQ) { // seteq case
1462 Instruction *Xor = BinaryOperator::create(Instruction::Xor, Op0, Op1,
1463 I.getName()+"tmp");
1464 InsertNewInstBefore(Xor, I);
Chris Lattner16930792003-11-03 04:25:02 +00001465 return BinaryOperator::createNot(Xor);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001466 }
1467
1468 // Handle the setXe cases...
1469 assert(I.getOpcode() == Instruction::SetGE ||
1470 I.getOpcode() == Instruction::SetLE);
1471
1472 if (I.getOpcode() == Instruction::SetGE)
1473 std::swap(Op0, Op1); // Change setge -> setle
1474
1475 // Now we just have the SetLE case.
Chris Lattner31ae8632002-08-14 17:51:49 +00001476 Instruction *Not = BinaryOperator::createNot(Op0, I.getName()+"tmp");
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001477 InsertNewInstBefore(Not, I);
Chris Lattner16930792003-11-03 04:25:02 +00001478 return BinaryOperator::create(Instruction::Or, Not, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001479 }
1480
Chris Lattner2dd01742004-06-09 04:24:29 +00001481 // See if we are doing a comparison between a constant and an instruction that
1482 // can be folded into the comparison.
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001483 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnere1e10e12004-05-25 06:32:08 +00001484 if (Instruction *LHSI = dyn_cast<Instruction>(Op0))
Chris Lattner2dd01742004-06-09 04:24:29 +00001485 if (LHSI->hasOneUse())
Chris Lattner35167c32004-06-09 07:59:58 +00001486 switch (LHSI->getOpcode()) {
1487 case Instruction::And:
1488 if (isa<ConstantInt>(LHSI->getOperand(1))) {
1489
Chris Lattnere1e10e12004-05-25 06:32:08 +00001490
Chris Lattner35167c32004-06-09 07:59:58 +00001491 // If this is: (X >> C1) & C2 != C3 (where any shift and any compare
1492 // could exist), turn it into (X & (C2 << C1)) != (C3 << C1). This
1493 // happens a LOT in code produced by the C front-end, for bitfield
1494 // access.
1495 if (LHSI->getOperand(0)->hasOneUse())
1496 if (ShiftInst *Shift = dyn_cast<ShiftInst>(LHSI->getOperand(0)))
1497 if (ConstantUInt *ShAmt =
1498 dyn_cast<ConstantUInt>(Shift->getOperand(1))) {
1499 ConstantInt *AndCST = cast<ConstantInt>(LHSI->getOperand(1));
1500
1501 // We can fold this as long as we can't shift unknown bits
1502 // into the mask. This can only happen with signed shift
1503 // rights, as they sign-extend.
1504 const Type *Ty = Shift->getType();
1505 if (Shift->getOpcode() != Instruction::Shr ||
1506 Shift->getType()->isUnsigned() ||
1507 // To test for the bad case of the signed shr, see if any
1508 // of the bits shifted in could be tested after the mask.
1509 ConstantExpr::getAnd(ConstantExpr::getShl(ConstantInt::getAllOnesValue(Ty), ConstantUInt::get(Type::UByteTy, Ty->getPrimitiveSize()*8-ShAmt->getValue())), AndCST)->isNullValue()) {
1510 unsigned ShiftOp = Shift->getOpcode() == Instruction::Shl
1511 ? Instruction::Shr : Instruction::Shl;
1512 I.setOperand(1, ConstantExpr::get(ShiftOp, CI, ShAmt));
1513 LHSI->setOperand(1,ConstantExpr::get(ShiftOp,AndCST,ShAmt));
1514 LHSI->setOperand(0, Shift->getOperand(0));
1515 WorkList.push_back(Shift); // Shift is probably dead.
1516 AddUsesToWorkList(I);
1517 return &I;
1518 }
Chris Lattner2dd01742004-06-09 04:24:29 +00001519 }
Chris Lattner35167c32004-06-09 07:59:58 +00001520 }
1521 break;
1522 case Instruction::Select:
Chris Lattner2dd01742004-06-09 04:24:29 +00001523 // If either operand of the select is a constant, we can fold the
1524 // comparison into the select arms, which will cause one to be
1525 // constant folded and the select turned into a bitwise or.
1526 Value *Op1 = 0, *Op2 = 0;
Chris Lattner35167c32004-06-09 07:59:58 +00001527 if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) {
Chris Lattner2dd01742004-06-09 04:24:29 +00001528 // Fold the known value into the constant operand.
1529 Op1 = ConstantExpr::get(I.getOpcode(), C, CI);
1530 // Insert a new SetCC of the other select operand.
1531 Op2 = InsertNewInstBefore(new SetCondInst(I.getOpcode(),
Chris Lattner35167c32004-06-09 07:59:58 +00001532 LHSI->getOperand(2), CI,
Chris Lattner2dd01742004-06-09 04:24:29 +00001533 I.getName()), I);
Chris Lattner35167c32004-06-09 07:59:58 +00001534 } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) {
Chris Lattner2dd01742004-06-09 04:24:29 +00001535 // Fold the known value into the constant operand.
1536 Op2 = ConstantExpr::get(I.getOpcode(), C, CI);
1537 // Insert a new SetCC of the other select operand.
1538 Op1 = InsertNewInstBefore(new SetCondInst(I.getOpcode(),
Chris Lattner35167c32004-06-09 07:59:58 +00001539 LHSI->getOperand(1), CI,
Chris Lattner2dd01742004-06-09 04:24:29 +00001540 I.getName()), I);
1541 }
1542
1543 if (Op1)
Chris Lattner35167c32004-06-09 07:59:58 +00001544 return new SelectInst(LHSI->getOperand(0), Op1, Op2);
1545 break;
Chris Lattner2dd01742004-06-09 04:24:29 +00001546 }
Chris Lattnere1e10e12004-05-25 06:32:08 +00001547
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001548 // Simplify seteq and setne instructions...
1549 if (I.getOpcode() == Instruction::SetEQ ||
1550 I.getOpcode() == Instruction::SetNE) {
1551 bool isSetNE = I.getOpcode() == Instruction::SetNE;
1552
Chris Lattnercfbce7c2003-07-23 17:26:36 +00001553 // If the first operand is (and|or|xor) with a constant, and the second
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001554 // operand is a constant, simplify a bit.
Chris Lattnerc992add2003-08-13 05:33:12 +00001555 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) {
1556 switch (BO->getOpcode()) {
1557 case Instruction::Add:
1558 if (CI->isNullValue()) {
1559 // Replace ((add A, B) != 0) with (A != -B) if A or B is
1560 // efficiently invertible, or if the add has just this one use.
1561 Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1);
1562 if (Value *NegVal = dyn_castNegVal(BOp1))
1563 return new SetCondInst(I.getOpcode(), BOp0, NegVal);
1564 else if (Value *NegVal = dyn_castNegVal(BOp0))
1565 return new SetCondInst(I.getOpcode(), NegVal, BOp1);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001566 else if (BO->hasOneUse()) {
Chris Lattnerc992add2003-08-13 05:33:12 +00001567 Instruction *Neg = BinaryOperator::createNeg(BOp1, BO->getName());
1568 BO->setName("");
1569 InsertNewInstBefore(Neg, I);
1570 return new SetCondInst(I.getOpcode(), BOp0, Neg);
1571 }
1572 }
1573 break;
1574 case Instruction::Xor:
1575 // For the xor case, we can xor two constants together, eliminating
1576 // the explicit xor.
1577 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1)))
1578 return BinaryOperator::create(I.getOpcode(), BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001579 ConstantExpr::get(Instruction::Xor, CI, BOC));
Chris Lattnerc992add2003-08-13 05:33:12 +00001580
1581 // FALLTHROUGH
1582 case Instruction::Sub:
1583 // Replace (([sub|xor] A, B) != 0) with (A != B)
1584 if (CI->isNullValue())
1585 return new SetCondInst(I.getOpcode(), BO->getOperand(0),
1586 BO->getOperand(1));
1587 break;
1588
1589 case Instruction::Or:
1590 // If bits are being or'd in that are not present in the constant we
1591 // are comparing against, then the comparison could never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001592 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) {
1593 Constant *NotCI = NotConstant(CI);
1594 if (!ConstantExpr::get(Instruction::And, BOC, NotCI)->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001595 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001596 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001597 break;
1598
1599 case Instruction::And:
1600 if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001601 // If bits are being compared against that are and'd out, then the
1602 // comparison can never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001603 if (!ConstantExpr::get(Instruction::And, CI,
1604 NotConstant(BOC))->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001605 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc992add2003-08-13 05:33:12 +00001606
Chris Lattner35167c32004-06-09 07:59:58 +00001607 // If we have ((X & C) == C), turn it into ((X & C) != 0).
1608 if (CI == BOC) {
1609 // Don't infinite loop if C is null and the & isn't folded yet.
1610 if (CI->isNullValue())
1611 return ReplaceInstUsesWith(I, ConstantBool::get(!isSetNE));
1612 return new SetCondInst(isSetNE ? Instruction::SetEQ :
1613 Instruction::SetNE, Op0,
1614 Constant::getNullValue(CI->getType()));
1615 }
1616
Chris Lattnerc992add2003-08-13 05:33:12 +00001617 // Replace (and X, (1 << size(X)-1) != 0) with x < 0, converting X
1618 // to be a signed value as appropriate.
1619 if (isSignBit(BOC)) {
1620 Value *X = BO->getOperand(0);
1621 // If 'X' is not signed, insert a cast now...
1622 if (!BOC->getType()->isSigned()) {
Chris Lattnere79e8542004-02-23 06:38:22 +00001623 const Type *DestTy = getSignedIntegralType(BOC->getType());
Chris Lattnerc992add2003-08-13 05:33:12 +00001624 CastInst *NewCI = new CastInst(X,DestTy,X->getName()+".signed");
1625 InsertNewInstBefore(NewCI, I);
1626 X = NewCI;
1627 }
1628 return new SetCondInst(isSetNE ? Instruction::SetLT :
1629 Instruction::SetGE, X,
1630 Constant::getNullValue(X->getType()));
1631 }
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001632 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001633 default: break;
1634 }
1635 }
Chris Lattner2b55ea32004-02-23 07:16:20 +00001636 } else { // Not a SetEQ/SetNE
1637 // If the LHS is a cast from an integral value of the same size,
1638 if (CastInst *Cast = dyn_cast<CastInst>(Op0)) {
1639 Value *CastOp = Cast->getOperand(0);
1640 const Type *SrcTy = CastOp->getType();
1641 unsigned SrcTySize = SrcTy->getPrimitiveSize();
1642 if (SrcTy != Cast->getType() && SrcTy->isInteger() &&
1643 SrcTySize == Cast->getType()->getPrimitiveSize()) {
1644 assert((SrcTy->isSigned() ^ Cast->getType()->isSigned()) &&
1645 "Source and destination signednesses should differ!");
1646 if (Cast->getType()->isSigned()) {
1647 // If this is a signed comparison, check for comparisons in the
1648 // vicinity of zero.
1649 if (I.getOpcode() == Instruction::SetLT && CI->isNullValue())
1650 // X < 0 => x > 127
1651 return BinaryOperator::create(Instruction::SetGT, CastOp,
1652 ConstantUInt::get(SrcTy, (1ULL << (SrcTySize*8-1))-1));
1653 else if (I.getOpcode() == Instruction::SetGT &&
1654 cast<ConstantSInt>(CI)->getValue() == -1)
1655 // X > -1 => x < 128
Chris Lattnerc40b9d72004-02-23 21:46:42 +00001656 return BinaryOperator::create(Instruction::SetLT, CastOp,
Chris Lattner2b55ea32004-02-23 07:16:20 +00001657 ConstantUInt::get(SrcTy, 1ULL << (SrcTySize*8-1)));
1658 } else {
1659 ConstantUInt *CUI = cast<ConstantUInt>(CI);
1660 if (I.getOpcode() == Instruction::SetLT &&
1661 CUI->getValue() == 1ULL << (SrcTySize*8-1))
1662 // X < 128 => X > -1
1663 return BinaryOperator::create(Instruction::SetGT, CastOp,
1664 ConstantSInt::get(SrcTy, -1));
1665 else if (I.getOpcode() == Instruction::SetGT &&
1666 CUI->getValue() == (1ULL << (SrcTySize*8-1))-1)
1667 // X > 127 => X < 0
1668 return BinaryOperator::create(Instruction::SetLT, CastOp,
1669 Constant::getNullValue(SrcTy));
1670 }
1671 }
1672 }
Chris Lattnere967b342003-06-04 05:10:11 +00001673 }
Chris Lattner791ac1a2003-06-01 03:35:25 +00001674
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001675 // Check to see if we are comparing against the minimum or maximum value...
Chris Lattnere6794492002-08-12 21:17:25 +00001676 if (CI->isMinValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001677 if (I.getOpcode() == Instruction::SetLT) // A < MIN -> FALSE
1678 return ReplaceInstUsesWith(I, ConstantBool::False);
1679 if (I.getOpcode() == Instruction::SetGE) // A >= MIN -> TRUE
1680 return ReplaceInstUsesWith(I, ConstantBool::True);
1681 if (I.getOpcode() == Instruction::SetLE) // A <= MIN -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001682 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001683 if (I.getOpcode() == Instruction::SetGT) // A > MIN -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001684 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001685
Chris Lattnere6794492002-08-12 21:17:25 +00001686 } else if (CI->isMaxValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001687 if (I.getOpcode() == Instruction::SetGT) // A > MAX -> FALSE
1688 return ReplaceInstUsesWith(I, ConstantBool::False);
1689 if (I.getOpcode() == Instruction::SetLE) // A <= MAX -> TRUE
1690 return ReplaceInstUsesWith(I, ConstantBool::True);
1691 if (I.getOpcode() == Instruction::SetGE) // A >= MAX -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001692 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001693 if (I.getOpcode() == Instruction::SetLT) // A < MAX -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001694 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001695
1696 // Comparing against a value really close to min or max?
1697 } else if (isMinValuePlusOne(CI)) {
1698 if (I.getOpcode() == Instruction::SetLT) // A < MIN+1 -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001699 return BinaryOperator::create(Instruction::SetEQ, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001700 if (I.getOpcode() == Instruction::SetGE) // A >= MIN-1 -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001701 return BinaryOperator::create(Instruction::SetNE, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001702
1703 } else if (isMaxValueMinusOne(CI)) {
1704 if (I.getOpcode() == Instruction::SetGT) // A > MAX-1 -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001705 return BinaryOperator::create(Instruction::SetEQ, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001706 if (I.getOpcode() == Instruction::SetLE) // A <= MAX-1 -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001707 return BinaryOperator::create(Instruction::SetNE, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001708 }
Chris Lattner59611142004-02-23 05:47:48 +00001709
1710 // If we still have a setle or setge instruction, turn it into the
1711 // appropriate setlt or setgt instruction. Since the border cases have
1712 // already been handled above, this requires little checking.
1713 //
1714 if (I.getOpcode() == Instruction::SetLE)
1715 return BinaryOperator::create(Instruction::SetLT, Op0, AddOne(CI));
1716 if (I.getOpcode() == Instruction::SetGE)
1717 return BinaryOperator::create(Instruction::SetGT, Op0, SubOne(CI));
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001718 }
1719
Chris Lattner16930792003-11-03 04:25:02 +00001720 // Test to see if the operands of the setcc are casted versions of other
1721 // values. If the cast can be stripped off both arguments, we do so now.
Chris Lattner6444c372003-11-03 05:17:03 +00001722 if (CastInst *CI = dyn_cast<CastInst>(Op0)) {
1723 Value *CastOp0 = CI->getOperand(0);
1724 if (CastOp0->getType()->isLosslesslyConvertibleTo(CI->getType()) &&
Chris Lattner7d2a5392004-03-13 23:54:27 +00001725 (isa<Constant>(Op1) || isa<CastInst>(Op1)) &&
Chris Lattner16930792003-11-03 04:25:02 +00001726 (I.getOpcode() == Instruction::SetEQ ||
1727 I.getOpcode() == Instruction::SetNE)) {
1728 // We keep moving the cast from the left operand over to the right
1729 // operand, where it can often be eliminated completely.
Chris Lattner6444c372003-11-03 05:17:03 +00001730 Op0 = CastOp0;
Chris Lattner16930792003-11-03 04:25:02 +00001731
1732 // If operand #1 is a cast instruction, see if we can eliminate it as
1733 // well.
Chris Lattner6444c372003-11-03 05:17:03 +00001734 if (CastInst *CI2 = dyn_cast<CastInst>(Op1))
1735 if (CI2->getOperand(0)->getType()->isLosslesslyConvertibleTo(
Chris Lattner16930792003-11-03 04:25:02 +00001736 Op0->getType()))
Chris Lattner6444c372003-11-03 05:17:03 +00001737 Op1 = CI2->getOperand(0);
Chris Lattner16930792003-11-03 04:25:02 +00001738
1739 // If Op1 is a constant, we can fold the cast into the constant.
1740 if (Op1->getType() != Op0->getType())
1741 if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
1742 Op1 = ConstantExpr::getCast(Op1C, Op0->getType());
1743 } else {
1744 // Otherwise, cast the RHS right before the setcc
1745 Op1 = new CastInst(Op1, Op0->getType(), Op1->getName());
1746 InsertNewInstBefore(cast<Instruction>(Op1), I);
1747 }
1748 return BinaryOperator::create(I.getOpcode(), Op0, Op1);
1749 }
1750
Chris Lattner6444c372003-11-03 05:17:03 +00001751 // Handle the special case of: setcc (cast bool to X), <cst>
1752 // This comes up when you have code like
1753 // int X = A < B;
1754 // if (X) ...
1755 // For generality, we handle any zero-extension of any operand comparison
1756 // with a constant.
1757 if (ConstantInt *ConstantRHS = dyn_cast<ConstantInt>(Op1)) {
1758 const Type *SrcTy = CastOp0->getType();
1759 const Type *DestTy = Op0->getType();
1760 if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
1761 (SrcTy->isUnsigned() || SrcTy == Type::BoolTy)) {
1762 // Ok, we have an expansion of operand 0 into a new type. Get the
1763 // constant value, masink off bits which are not set in the RHS. These
1764 // could be set if the destination value is signed.
1765 uint64_t ConstVal = ConstantRHS->getRawValue();
1766 ConstVal &= (1ULL << DestTy->getPrimitiveSize()*8)-1;
1767
1768 // If the constant we are comparing it with has high bits set, which
1769 // don't exist in the original value, the values could never be equal,
1770 // because the source would be zero extended.
1771 unsigned SrcBits =
1772 SrcTy == Type::BoolTy ? 1 : SrcTy->getPrimitiveSize()*8;
Chris Lattner7c94d112003-11-05 17:31:36 +00001773 bool HasSignBit = ConstVal & (1ULL << (DestTy->getPrimitiveSize()*8-1));
1774 if (ConstVal & ~((1ULL << SrcBits)-1)) {
Chris Lattner6444c372003-11-03 05:17:03 +00001775 switch (I.getOpcode()) {
1776 default: assert(0 && "Unknown comparison type!");
1777 case Instruction::SetEQ:
1778 return ReplaceInstUsesWith(I, ConstantBool::False);
1779 case Instruction::SetNE:
1780 return ReplaceInstUsesWith(I, ConstantBool::True);
1781 case Instruction::SetLT:
1782 case Instruction::SetLE:
1783 if (DestTy->isSigned() && HasSignBit)
1784 return ReplaceInstUsesWith(I, ConstantBool::False);
1785 return ReplaceInstUsesWith(I, ConstantBool::True);
1786 case Instruction::SetGT:
1787 case Instruction::SetGE:
1788 if (DestTy->isSigned() && HasSignBit)
1789 return ReplaceInstUsesWith(I, ConstantBool::True);
1790 return ReplaceInstUsesWith(I, ConstantBool::False);
1791 }
1792 }
1793
1794 // Otherwise, we can replace the setcc with a setcc of the smaller
1795 // operand value.
1796 Op1 = ConstantExpr::getCast(cast<Constant>(Op1), SrcTy);
1797 return BinaryOperator::create(I.getOpcode(), CastOp0, Op1);
1798 }
1799 }
1800 }
Chris Lattner113f4f42002-06-25 16:13:24 +00001801 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001802}
1803
1804
1805
Chris Lattnere8d6c602003-03-10 19:16:08 +00001806Instruction *InstCombiner::visitShiftInst(ShiftInst &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +00001807 assert(I.getOperand(1)->getType() == Type::UByteTy);
1808 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001809 bool isLeftShift = I.getOpcode() == Instruction::Shl;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001810
1811 // shl X, 0 == X and shr X, 0 == X
1812 // shl 0, X == 0 and shr 0, X == 0
1813 if (Op1 == Constant::getNullValue(Type::UByteTy) ||
Chris Lattnere6794492002-08-12 21:17:25 +00001814 Op0 == Constant::getNullValue(Op0->getType()))
1815 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001816
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001817 // shr int -1, X = -1 (for any arithmetic shift rights of ~0)
1818 if (!isLeftShift)
1819 if (ConstantSInt *CSI = dyn_cast<ConstantSInt>(Op0))
1820 if (CSI->isAllOnesValue())
1821 return ReplaceInstUsesWith(I, CSI);
1822
Chris Lattner183b3362004-04-09 19:05:30 +00001823 // Try to fold constant and into select arguments.
1824 if (isa<Constant>(Op0))
1825 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
1826 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1827 return R;
1828
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001829 if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(Op1)) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001830 // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr
1831 // of a signed value.
1832 //
Chris Lattnere8d6c602003-03-10 19:16:08 +00001833 unsigned TypeBits = Op0->getType()->getPrimitiveSize()*8;
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001834 if (CUI->getValue() >= TypeBits) {
1835 if (!Op0->getType()->isSigned() || isLeftShift)
1836 return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType()));
1837 else {
1838 I.setOperand(1, ConstantUInt::get(Type::UByteTy, TypeBits-1));
1839 return &I;
1840 }
1841 }
Chris Lattner55f3d942002-09-10 23:04:09 +00001842
Chris Lattnerede3fe02003-08-13 04:18:28 +00001843 // ((X*C1) << C2) == (X * (C1 << C2))
1844 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0))
1845 if (BO->getOpcode() == Instruction::Mul && isLeftShift)
1846 if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1)))
1847 return BinaryOperator::create(Instruction::Mul, BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001848 ConstantExpr::get(Instruction::Shl, BOOp, CUI));
Chris Lattnerede3fe02003-08-13 04:18:28 +00001849
Chris Lattner183b3362004-04-09 19:05:30 +00001850 // Try to fold constant and into select arguments.
1851 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1852 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1853 return R;
Chris Lattnerede3fe02003-08-13 04:18:28 +00001854
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001855 // If the operand is an bitwise operator with a constant RHS, and the
1856 // shift is the only use, we can pull it out of the shift.
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001857 if (Op0->hasOneUse())
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001858 if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0))
1859 if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) {
1860 bool isValid = true; // Valid only for And, Or, Xor
1861 bool highBitSet = false; // Transform if high bit of constant set?
1862
1863 switch (Op0BO->getOpcode()) {
1864 default: isValid = false; break; // Do not perform transform!
1865 case Instruction::Or:
1866 case Instruction::Xor:
1867 highBitSet = false;
1868 break;
1869 case Instruction::And:
1870 highBitSet = true;
1871 break;
1872 }
1873
1874 // If this is a signed shift right, and the high bit is modified
1875 // by the logical operation, do not perform the transformation.
1876 // The highBitSet boolean indicates the value of the high bit of
1877 // the constant which would cause it to be modified for this
1878 // operation.
1879 //
1880 if (isValid && !isLeftShift && !I.getType()->isUnsigned()) {
1881 uint64_t Val = Op0C->getRawValue();
1882 isValid = ((Val & (1 << (TypeBits-1))) != 0) == highBitSet;
1883 }
1884
1885 if (isValid) {
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001886 Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, CUI);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001887
1888 Instruction *NewShift =
1889 new ShiftInst(I.getOpcode(), Op0BO->getOperand(0), CUI,
1890 Op0BO->getName());
1891 Op0BO->setName("");
1892 InsertNewInstBefore(NewShift, I);
1893
1894 return BinaryOperator::create(Op0BO->getOpcode(), NewShift,
1895 NewRHS);
1896 }
1897 }
1898
Chris Lattner3204d4e2003-07-24 17:52:58 +00001899 // If this is a shift of a shift, see if we can fold the two together...
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001900 if (ShiftInst *Op0SI = dyn_cast<ShiftInst>(Op0))
Chris Lattnerab780df2003-07-24 18:38:56 +00001901 if (ConstantUInt *ShiftAmt1C =
1902 dyn_cast<ConstantUInt>(Op0SI->getOperand(1))) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001903 unsigned ShiftAmt1 = ShiftAmt1C->getValue();
1904 unsigned ShiftAmt2 = CUI->getValue();
1905
1906 // Check for (A << c1) << c2 and (A >> c1) >> c2
1907 if (I.getOpcode() == Op0SI->getOpcode()) {
1908 unsigned Amt = ShiftAmt1+ShiftAmt2; // Fold into one big shift...
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001909 if (Op0->getType()->getPrimitiveSize()*8 < Amt)
1910 Amt = Op0->getType()->getPrimitiveSize()*8;
Chris Lattner3204d4e2003-07-24 17:52:58 +00001911 return new ShiftInst(I.getOpcode(), Op0SI->getOperand(0),
1912 ConstantUInt::get(Type::UByteTy, Amt));
1913 }
1914
Chris Lattnerab780df2003-07-24 18:38:56 +00001915 // Check for (A << c1) >> c2 or visaversa. If we are dealing with
1916 // signed types, we can only support the (A >> c1) << c2 configuration,
1917 // because it can not turn an arbitrary bit of A into a sign bit.
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001918 if (I.getType()->isUnsigned() || isLeftShift) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001919 // Calculate bitmask for what gets shifted off the edge...
1920 Constant *C = ConstantIntegral::getAllOnesValue(I.getType());
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001921 if (isLeftShift)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001922 C = ConstantExpr::get(Instruction::Shl, C, ShiftAmt1C);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001923 else
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001924 C = ConstantExpr::get(Instruction::Shr, C, ShiftAmt1C);
Chris Lattner3204d4e2003-07-24 17:52:58 +00001925
1926 Instruction *Mask =
1927 BinaryOperator::create(Instruction::And, Op0SI->getOperand(0),
1928 C, Op0SI->getOperand(0)->getName()+".mask");
1929 InsertNewInstBefore(Mask, I);
1930
1931 // Figure out what flavor of shift we should use...
1932 if (ShiftAmt1 == ShiftAmt2)
1933 return ReplaceInstUsesWith(I, Mask); // (A << c) >> c === A & c2
1934 else if (ShiftAmt1 < ShiftAmt2) {
1935 return new ShiftInst(I.getOpcode(), Mask,
1936 ConstantUInt::get(Type::UByteTy, ShiftAmt2-ShiftAmt1));
1937 } else {
1938 return new ShiftInst(Op0SI->getOpcode(), Mask,
1939 ConstantUInt::get(Type::UByteTy, ShiftAmt1-ShiftAmt2));
1940 }
1941 }
1942 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001943 }
Chris Lattner2e0fb392002-10-08 16:16:40 +00001944
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001945 return 0;
1946}
1947
1948
Chris Lattner48a44f72002-05-02 17:06:02 +00001949// isEliminableCastOfCast - Return true if it is valid to eliminate the CI
1950// instruction.
1951//
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001952static inline bool isEliminableCastOfCast(const Type *SrcTy, const Type *MidTy,
1953 const Type *DstTy) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001954
Chris Lattner650b6da2002-08-02 20:00:25 +00001955 // It is legal to eliminate the instruction if casting A->B->A if the sizes
1956 // are identical and the bits don't get reinterpreted (for example
Chris Lattner0bb75912002-08-14 23:21:10 +00001957 // int->float->int would not be allowed)
Misha Brukmane5838c42003-05-20 18:45:36 +00001958 if (SrcTy == DstTy && SrcTy->isLosslesslyConvertibleTo(MidTy))
Chris Lattner650b6da2002-08-02 20:00:25 +00001959 return true;
Chris Lattner48a44f72002-05-02 17:06:02 +00001960
1961 // Allow free casting and conversion of sizes as long as the sign doesn't
1962 // change...
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001963 if (SrcTy->isIntegral() && MidTy->isIntegral() && DstTy->isIntegral()) {
Chris Lattner650b6da2002-08-02 20:00:25 +00001964 unsigned SrcSize = SrcTy->getPrimitiveSize();
1965 unsigned MidSize = MidTy->getPrimitiveSize();
1966 unsigned DstSize = DstTy->getPrimitiveSize();
Chris Lattner650b6da2002-08-02 20:00:25 +00001967
Chris Lattner3732aca2002-08-15 16:15:25 +00001968 // Cases where we are monotonically decreasing the size of the type are
1969 // always ok, regardless of what sign changes are going on.
1970 //
Chris Lattner0bb75912002-08-14 23:21:10 +00001971 if (SrcSize >= MidSize && MidSize >= DstSize)
Chris Lattner650b6da2002-08-02 20:00:25 +00001972 return true;
Chris Lattner3732aca2002-08-15 16:15:25 +00001973
Chris Lattner555518c2002-09-23 23:39:43 +00001974 // Cases where the source and destination type are the same, but the middle
1975 // type is bigger are noops.
1976 //
1977 if (SrcSize == DstSize && MidSize > SrcSize)
1978 return true;
1979
Chris Lattner3732aca2002-08-15 16:15:25 +00001980 // If we are monotonically growing, things are more complex.
1981 //
1982 if (SrcSize <= MidSize && MidSize <= DstSize) {
1983 // We have eight combinations of signedness to worry about. Here's the
1984 // table:
1985 static const int SignTable[8] = {
1986 // CODE, SrcSigned, MidSigned, DstSigned, Comment
1987 1, // U U U Always ok
1988 1, // U U S Always ok
1989 3, // U S U Ok iff SrcSize != MidSize
1990 3, // U S S Ok iff SrcSize != MidSize
1991 0, // S U U Never ok
1992 2, // S U S Ok iff MidSize == DstSize
1993 1, // S S U Always ok
1994 1, // S S S Always ok
1995 };
1996
1997 // Choose an action based on the current entry of the signtable that this
1998 // cast of cast refers to...
1999 unsigned Row = SrcTy->isSigned()*4+MidTy->isSigned()*2+DstTy->isSigned();
2000 switch (SignTable[Row]) {
2001 case 0: return false; // Never ok
2002 case 1: return true; // Always ok
2003 case 2: return MidSize == DstSize; // Ok iff MidSize == DstSize
2004 case 3: // Ok iff SrcSize != MidSize
2005 return SrcSize != MidSize || SrcTy == Type::BoolTy;
2006 default: assert(0 && "Bad entry in sign table!");
2007 }
Chris Lattner3732aca2002-08-15 16:15:25 +00002008 }
Chris Lattner650b6da2002-08-02 20:00:25 +00002009 }
Chris Lattner48a44f72002-05-02 17:06:02 +00002010
2011 // Otherwise, we cannot succeed. Specifically we do not want to allow things
2012 // like: short -> ushort -> uint, because this can create wrong results if
2013 // the input short is negative!
2014 //
2015 return false;
2016}
2017
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002018static bool ValueRequiresCast(const Value *V, const Type *Ty) {
2019 if (V->getType() == Ty || isa<Constant>(V)) return false;
2020 if (const CastInst *CI = dyn_cast<CastInst>(V))
2021 if (isEliminableCastOfCast(CI->getOperand(0)->getType(), CI->getType(), Ty))
2022 return false;
2023 return true;
2024}
2025
2026/// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
2027/// InsertBefore instruction. This is specialized a bit to avoid inserting
2028/// casts that are known to not do anything...
2029///
2030Value *InstCombiner::InsertOperandCastBefore(Value *V, const Type *DestTy,
2031 Instruction *InsertBefore) {
2032 if (V->getType() == DestTy) return V;
2033 if (Constant *C = dyn_cast<Constant>(V))
2034 return ConstantExpr::getCast(C, DestTy);
2035
2036 CastInst *CI = new CastInst(V, DestTy, V->getName());
2037 InsertNewInstBefore(CI, *InsertBefore);
2038 return CI;
2039}
Chris Lattner48a44f72002-05-02 17:06:02 +00002040
2041// CastInst simplification
Chris Lattner260ab202002-04-18 17:39:14 +00002042//
Chris Lattner113f4f42002-06-25 16:13:24 +00002043Instruction *InstCombiner::visitCastInst(CastInst &CI) {
Chris Lattner55d4bda2003-06-23 21:59:52 +00002044 Value *Src = CI.getOperand(0);
2045
Chris Lattner48a44f72002-05-02 17:06:02 +00002046 // If the user is casting a value to the same type, eliminate this cast
2047 // instruction...
Chris Lattner55d4bda2003-06-23 21:59:52 +00002048 if (CI.getType() == Src->getType())
2049 return ReplaceInstUsesWith(CI, Src);
Chris Lattner48a44f72002-05-02 17:06:02 +00002050
Chris Lattner48a44f72002-05-02 17:06:02 +00002051 // If casting the result of another cast instruction, try to eliminate this
2052 // one!
2053 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002054 if (CastInst *CSrc = dyn_cast<CastInst>(Src)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002055 if (isEliminableCastOfCast(CSrc->getOperand(0)->getType(),
2056 CSrc->getType(), CI.getType())) {
Chris Lattner48a44f72002-05-02 17:06:02 +00002057 // This instruction now refers directly to the cast's src operand. This
2058 // has a good chance of making CSrc dead.
Chris Lattner113f4f42002-06-25 16:13:24 +00002059 CI.setOperand(0, CSrc->getOperand(0));
2060 return &CI;
Chris Lattner48a44f72002-05-02 17:06:02 +00002061 }
2062
Chris Lattner650b6da2002-08-02 20:00:25 +00002063 // If this is an A->B->A cast, and we are dealing with integral types, try
2064 // to convert this into a logical 'and' instruction.
2065 //
2066 if (CSrc->getOperand(0)->getType() == CI.getType() &&
Chris Lattnerb0b412e2002-09-03 01:08:28 +00002067 CI.getType()->isInteger() && CSrc->getType()->isInteger() &&
Chris Lattner650b6da2002-08-02 20:00:25 +00002068 CI.getType()->isUnsigned() && CSrc->getType()->isUnsigned() &&
2069 CSrc->getType()->getPrimitiveSize() < CI.getType()->getPrimitiveSize()){
2070 assert(CSrc->getType() != Type::ULongTy &&
2071 "Cannot have type bigger than ulong!");
Chris Lattner196897c2003-05-26 23:41:32 +00002072 uint64_t AndValue = (1ULL << CSrc->getType()->getPrimitiveSize()*8)-1;
Chris Lattner650b6da2002-08-02 20:00:25 +00002073 Constant *AndOp = ConstantUInt::get(CI.getType(), AndValue);
2074 return BinaryOperator::create(Instruction::And, CSrc->getOperand(0),
2075 AndOp);
2076 }
2077 }
2078
Chris Lattner03841652004-05-25 04:29:21 +00002079 // If this is a cast to bool, turn it into the appropriate setne instruction.
2080 if (CI.getType() == Type::BoolTy)
2081 return BinaryOperator::create(Instruction::SetNE, CI.getOperand(0),
2082 Constant::getNullValue(CI.getOperand(0)->getType()));
2083
Chris Lattnerd0d51602003-06-21 23:12:02 +00002084 // If casting the result of a getelementptr instruction with no offset, turn
2085 // this into a cast of the original pointer!
2086 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002087 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
Chris Lattnerd0d51602003-06-21 23:12:02 +00002088 bool AllZeroOperands = true;
2089 for (unsigned i = 1, e = GEP->getNumOperands(); i != e; ++i)
2090 if (!isa<Constant>(GEP->getOperand(i)) ||
2091 !cast<Constant>(GEP->getOperand(i))->isNullValue()) {
2092 AllZeroOperands = false;
2093 break;
2094 }
2095 if (AllZeroOperands) {
2096 CI.setOperand(0, GEP->getOperand(0));
2097 return &CI;
2098 }
2099 }
2100
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002101 // If we are casting a malloc or alloca to a pointer to a type of the same
2102 // size, rewrite the allocation instruction to allocate the "right" type.
2103 //
2104 if (AllocationInst *AI = dyn_cast<AllocationInst>(Src))
Chris Lattnerd4d987d2003-11-02 06:54:48 +00002105 if (AI->hasOneUse() && !AI->isArrayAllocation())
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002106 if (const PointerType *PTy = dyn_cast<PointerType>(CI.getType())) {
2107 // Get the type really allocated and the type casted to...
2108 const Type *AllocElTy = AI->getAllocatedType();
2109 unsigned AllocElTySize = TD->getTypeSize(AllocElTy);
2110 const Type *CastElTy = PTy->getElementType();
2111 unsigned CastElTySize = TD->getTypeSize(CastElTy);
Chris Lattner7c94d112003-11-05 17:31:36 +00002112
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002113 // If the allocation is for an even multiple of the cast type size
Chris Lattneraf789322003-11-03 01:29:41 +00002114 if (CastElTySize && (AllocElTySize % CastElTySize == 0)) {
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002115 Value *Amt = ConstantUInt::get(Type::UIntTy,
2116 AllocElTySize/CastElTySize);
2117 std::string Name = AI->getName(); AI->setName("");
2118 AllocationInst *New;
2119 if (isa<MallocInst>(AI))
2120 New = new MallocInst(CastElTy, Amt, Name);
2121 else
2122 New = new AllocaInst(CastElTy, Amt, Name);
Chris Lattner652064e2004-04-30 04:37:52 +00002123 InsertNewInstBefore(New, *AI);
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002124 return ReplaceInstUsesWith(CI, New);
2125 }
2126 }
2127
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002128 // If the source value is an instruction with only this use, we can attempt to
2129 // propagate the cast into the instruction. Also, only handle integral types
2130 // for now.
2131 if (Instruction *SrcI = dyn_cast<Instruction>(Src))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00002132 if (SrcI->hasOneUse() && Src->getType()->isIntegral() &&
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002133 CI.getType()->isInteger()) { // Don't mess with casts to bool here
2134 const Type *DestTy = CI.getType();
2135 unsigned SrcBitSize = getTypeSizeInBits(Src->getType());
2136 unsigned DestBitSize = getTypeSizeInBits(DestTy);
2137
2138 Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0;
2139 Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0;
2140
2141 switch (SrcI->getOpcode()) {
2142 case Instruction::Add:
2143 case Instruction::Mul:
2144 case Instruction::And:
2145 case Instruction::Or:
2146 case Instruction::Xor:
2147 // If we are discarding information, or just changing the sign, rewrite.
2148 if (DestBitSize <= SrcBitSize && DestBitSize != 1) {
2149 // Don't insert two casts if they cannot be eliminated. We allow two
2150 // casts to be inserted if the sizes are the same. This could only be
2151 // converting signedness, which is a noop.
2152 if (DestBitSize == SrcBitSize || !ValueRequiresCast(Op1, DestTy) ||
2153 !ValueRequiresCast(Op0, DestTy)) {
2154 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2155 Value *Op1c = InsertOperandCastBefore(Op1, DestTy, SrcI);
2156 return BinaryOperator::create(cast<BinaryOperator>(SrcI)
2157 ->getOpcode(), Op0c, Op1c);
2158 }
2159 }
2160 break;
2161 case Instruction::Shl:
2162 // Allow changing the sign of the source operand. Do not allow changing
2163 // the size of the shift, UNLESS the shift amount is a constant. We
2164 // mush not change variable sized shifts to a smaller size, because it
2165 // is undefined to shift more bits out than exist in the value.
2166 if (DestBitSize == SrcBitSize ||
2167 (DestBitSize < SrcBitSize && isa<Constant>(Op1))) {
2168 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2169 return new ShiftInst(Instruction::Shl, Op0c, Op1);
2170 }
2171 break;
2172 }
2173 }
2174
Chris Lattner260ab202002-04-18 17:39:14 +00002175 return 0;
Chris Lattnerca081252001-12-14 16:52:21 +00002176}
2177
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002178/// GetSelectFoldableOperands - We want to turn code that looks like this:
2179/// %C = or %A, %B
2180/// %D = select %cond, %C, %A
2181/// into:
2182/// %C = select %cond, %B, 0
2183/// %D = or %A, %C
2184///
2185/// Assuming that the specified instruction is an operand to the select, return
2186/// a bitmask indicating which operands of this instruction are foldable if they
2187/// equal the other incoming value of the select.
2188///
2189static unsigned GetSelectFoldableOperands(Instruction *I) {
2190 switch (I->getOpcode()) {
2191 case Instruction::Add:
2192 case Instruction::Mul:
2193 case Instruction::And:
2194 case Instruction::Or:
2195 case Instruction::Xor:
2196 return 3; // Can fold through either operand.
2197 case Instruction::Sub: // Can only fold on the amount subtracted.
2198 case Instruction::Shl: // Can only fold on the shift amount.
2199 case Instruction::Shr:
2200 return 1;
2201 default:
2202 return 0; // Cannot fold
2203 }
2204}
2205
2206/// GetSelectFoldableConstant - For the same transformation as the previous
2207/// function, return the identity constant that goes into the select.
2208static Constant *GetSelectFoldableConstant(Instruction *I) {
2209 switch (I->getOpcode()) {
2210 default: assert(0 && "This cannot happen!"); abort();
2211 case Instruction::Add:
2212 case Instruction::Sub:
2213 case Instruction::Or:
2214 case Instruction::Xor:
2215 return Constant::getNullValue(I->getType());
2216 case Instruction::Shl:
2217 case Instruction::Shr:
2218 return Constant::getNullValue(Type::UByteTy);
2219 case Instruction::And:
2220 return ConstantInt::getAllOnesValue(I->getType());
2221 case Instruction::Mul:
2222 return ConstantInt::get(I->getType(), 1);
2223 }
2224}
2225
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002226Instruction *InstCombiner::visitSelectInst(SelectInst &SI) {
Chris Lattner533bc492004-03-30 19:37:13 +00002227 Value *CondVal = SI.getCondition();
2228 Value *TrueVal = SI.getTrueValue();
2229 Value *FalseVal = SI.getFalseValue();
2230
2231 // select true, X, Y -> X
2232 // select false, X, Y -> Y
2233 if (ConstantBool *C = dyn_cast<ConstantBool>(CondVal))
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002234 if (C == ConstantBool::True)
Chris Lattner533bc492004-03-30 19:37:13 +00002235 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002236 else {
2237 assert(C == ConstantBool::False);
Chris Lattner533bc492004-03-30 19:37:13 +00002238 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002239 }
Chris Lattner533bc492004-03-30 19:37:13 +00002240
2241 // select C, X, X -> X
2242 if (TrueVal == FalseVal)
2243 return ReplaceInstUsesWith(SI, TrueVal);
2244
Chris Lattner1c631e82004-04-08 04:43:23 +00002245 if (SI.getType() == Type::BoolTy)
2246 if (ConstantBool *C = dyn_cast<ConstantBool>(TrueVal)) {
2247 if (C == ConstantBool::True) {
2248 // Change: A = select B, true, C --> A = or B, C
2249 return BinaryOperator::create(Instruction::Or, CondVal, FalseVal);
2250 } else {
2251 // Change: A = select B, false, C --> A = and !B, C
2252 Value *NotCond =
2253 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2254 "not."+CondVal->getName()), SI);
2255 return BinaryOperator::create(Instruction::And, NotCond, FalseVal);
2256 }
2257 } else if (ConstantBool *C = dyn_cast<ConstantBool>(FalseVal)) {
2258 if (C == ConstantBool::False) {
2259 // Change: A = select B, C, false --> A = and B, C
2260 return BinaryOperator::create(Instruction::And, CondVal, TrueVal);
2261 } else {
2262 // Change: A = select B, C, true --> A = or !B, C
2263 Value *NotCond =
2264 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2265 "not."+CondVal->getName()), SI);
2266 return BinaryOperator::create(Instruction::Or, NotCond, TrueVal);
2267 }
2268 }
2269
Chris Lattner183b3362004-04-09 19:05:30 +00002270 // Selecting between two integer constants?
2271 if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal))
2272 if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) {
2273 // select C, 1, 0 -> cast C to int
2274 if (FalseValC->isNullValue() && TrueValC->getRawValue() == 1) {
2275 return new CastInst(CondVal, SI.getType());
2276 } else if (TrueValC->isNullValue() && FalseValC->getRawValue() == 1) {
2277 // select C, 0, 1 -> cast !C to int
2278 Value *NotCond =
2279 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
Chris Lattnercf7baf32004-04-09 18:19:44 +00002280 "not."+CondVal->getName()), SI);
Chris Lattner183b3362004-04-09 19:05:30 +00002281 return new CastInst(NotCond, SI.getType());
Chris Lattnercf7baf32004-04-09 18:19:44 +00002282 }
Chris Lattner35167c32004-06-09 07:59:58 +00002283
2284 // If one of the constants is zero (we know they can't both be) and we
2285 // have a setcc instruction with zero, and we have an 'and' with the
2286 // non-constant value, eliminate this whole mess. This corresponds to
2287 // cases like this: ((X & 27) ? 27 : 0)
2288 if (TrueValC->isNullValue() || FalseValC->isNullValue())
2289 if (Instruction *IC = dyn_cast<Instruction>(SI.getCondition()))
2290 if ((IC->getOpcode() == Instruction::SetEQ ||
2291 IC->getOpcode() == Instruction::SetNE) &&
2292 isa<ConstantInt>(IC->getOperand(1)) &&
2293 cast<Constant>(IC->getOperand(1))->isNullValue())
2294 if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0)))
2295 if (ICA->getOpcode() == Instruction::And &&
2296 isa<ConstantInt>(ICA->getOperand(1)) &&
2297 (ICA->getOperand(1) == TrueValC ||
2298 ICA->getOperand(1) == FalseValC) &&
2299 isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) {
2300 // Okay, now we know that everything is set up, we just don't
2301 // know whether we have a setne or seteq and whether the true or
2302 // false val is the zero.
2303 bool ShouldNotVal = !TrueValC->isNullValue();
2304 ShouldNotVal ^= IC->getOpcode() == Instruction::SetNE;
2305 Value *V = ICA;
2306 if (ShouldNotVal)
2307 V = InsertNewInstBefore(BinaryOperator::create(
2308 Instruction::Xor, V, ICA->getOperand(1)), SI);
2309 return ReplaceInstUsesWith(SI, V);
2310 }
Chris Lattner533bc492004-03-30 19:37:13 +00002311 }
Chris Lattner623fba12004-04-10 22:21:27 +00002312
2313 // See if we are selecting two values based on a comparison of the two values.
2314 if (SetCondInst *SCI = dyn_cast<SetCondInst>(CondVal)) {
2315 if (SCI->getOperand(0) == TrueVal && SCI->getOperand(1) == FalseVal) {
2316 // Transform (X == Y) ? X : Y -> Y
2317 if (SCI->getOpcode() == Instruction::SetEQ)
2318 return ReplaceInstUsesWith(SI, FalseVal);
2319 // Transform (X != Y) ? X : Y -> X
2320 if (SCI->getOpcode() == Instruction::SetNE)
2321 return ReplaceInstUsesWith(SI, TrueVal);
2322 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2323
2324 } else if (SCI->getOperand(0) == FalseVal && SCI->getOperand(1) == TrueVal){
2325 // Transform (X == Y) ? Y : X -> X
2326 if (SCI->getOpcode() == Instruction::SetEQ)
Chris Lattner24cf0202004-04-11 01:39:19 +00002327 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002328 // Transform (X != Y) ? Y : X -> Y
2329 if (SCI->getOpcode() == Instruction::SetNE)
Chris Lattner24cf0202004-04-11 01:39:19 +00002330 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002331 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2332 }
2333 }
Chris Lattner1c631e82004-04-08 04:43:23 +00002334
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002335 // See if we can fold the select into one of our operands.
2336 if (SI.getType()->isInteger()) {
2337 // See the comment above GetSelectFoldableOperands for a description of the
2338 // transformation we are doing here.
2339 if (Instruction *TVI = dyn_cast<Instruction>(TrueVal))
2340 if (TVI->hasOneUse() && TVI->getNumOperands() == 2 &&
2341 !isa<Constant>(FalseVal))
2342 if (unsigned SFO = GetSelectFoldableOperands(TVI)) {
2343 unsigned OpToFold = 0;
2344 if ((SFO & 1) && FalseVal == TVI->getOperand(0)) {
2345 OpToFold = 1;
2346 } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) {
2347 OpToFold = 2;
2348 }
2349
2350 if (OpToFold) {
2351 Constant *C = GetSelectFoldableConstant(TVI);
2352 std::string Name = TVI->getName(); TVI->setName("");
2353 Instruction *NewSel =
2354 new SelectInst(SI.getCondition(), TVI->getOperand(2-OpToFold), C,
2355 Name);
2356 InsertNewInstBefore(NewSel, SI);
2357 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI))
2358 return BinaryOperator::create(BO->getOpcode(), FalseVal, NewSel);
2359 else if (ShiftInst *SI = dyn_cast<ShiftInst>(TVI))
2360 return new ShiftInst(SI->getOpcode(), FalseVal, NewSel);
2361 else {
2362 assert(0 && "Unknown instruction!!");
2363 }
2364 }
2365 }
2366
2367 if (Instruction *FVI = dyn_cast<Instruction>(FalseVal))
2368 if (FVI->hasOneUse() && FVI->getNumOperands() == 2 &&
2369 !isa<Constant>(TrueVal))
2370 if (unsigned SFO = GetSelectFoldableOperands(FVI)) {
2371 unsigned OpToFold = 0;
2372 if ((SFO & 1) && TrueVal == FVI->getOperand(0)) {
2373 OpToFold = 1;
2374 } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) {
2375 OpToFold = 2;
2376 }
2377
2378 if (OpToFold) {
2379 Constant *C = GetSelectFoldableConstant(FVI);
2380 std::string Name = FVI->getName(); FVI->setName("");
2381 Instruction *NewSel =
2382 new SelectInst(SI.getCondition(), C, FVI->getOperand(2-OpToFold),
2383 Name);
2384 InsertNewInstBefore(NewSel, SI);
2385 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI))
2386 return BinaryOperator::create(BO->getOpcode(), TrueVal, NewSel);
2387 else if (ShiftInst *SI = dyn_cast<ShiftInst>(FVI))
2388 return new ShiftInst(SI->getOpcode(), TrueVal, NewSel);
2389 else {
2390 assert(0 && "Unknown instruction!!");
2391 }
2392 }
2393 }
2394 }
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002395 return 0;
2396}
2397
2398
Chris Lattner970c33a2003-06-19 17:00:31 +00002399// CallInst simplification
2400//
2401Instruction *InstCombiner::visitCallInst(CallInst &CI) {
Chris Lattner51ea1272004-02-28 05:22:00 +00002402 // Intrinsics cannot occur in an invoke, so handle them here instead of in
2403 // visitCallSite.
2404 if (Function *F = CI.getCalledFunction())
2405 switch (F->getIntrinsicID()) {
2406 case Intrinsic::memmove:
2407 case Intrinsic::memcpy:
2408 case Intrinsic::memset:
2409 // memmove/cpy/set of zero bytes is a noop.
2410 if (Constant *NumBytes = dyn_cast<Constant>(CI.getOperand(3))) {
2411 if (NumBytes->isNullValue())
2412 return EraseInstFromFunction(CI);
2413 }
2414 break;
2415 default:
2416 break;
2417 }
2418
Chris Lattneraec3d942003-10-07 22:32:43 +00002419 return visitCallSite(&CI);
Chris Lattner970c33a2003-06-19 17:00:31 +00002420}
2421
2422// InvokeInst simplification
2423//
2424Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) {
Chris Lattneraec3d942003-10-07 22:32:43 +00002425 return visitCallSite(&II);
Chris Lattner970c33a2003-06-19 17:00:31 +00002426}
2427
Chris Lattneraec3d942003-10-07 22:32:43 +00002428// visitCallSite - Improvements for call and invoke instructions.
2429//
2430Instruction *InstCombiner::visitCallSite(CallSite CS) {
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002431 bool Changed = false;
2432
2433 // If the callee is a constexpr cast of a function, attempt to move the cast
2434 // to the arguments of the call/invoke.
Chris Lattneraec3d942003-10-07 22:32:43 +00002435 if (transformConstExprCastCall(CS)) return 0;
2436
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002437 Value *Callee = CS.getCalledValue();
2438 const PointerType *PTy = cast<PointerType>(Callee->getType());
2439 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
2440 if (FTy->isVarArg()) {
2441 // See if we can optimize any arguments passed through the varargs area of
2442 // the call.
2443 for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(),
2444 E = CS.arg_end(); I != E; ++I)
2445 if (CastInst *CI = dyn_cast<CastInst>(*I)) {
2446 // If this cast does not effect the value passed through the varargs
2447 // area, we can eliminate the use of the cast.
2448 Value *Op = CI->getOperand(0);
2449 if (CI->getType()->isLosslesslyConvertibleTo(Op->getType())) {
2450 *I = Op;
2451 Changed = true;
2452 }
2453 }
2454 }
Chris Lattneraec3d942003-10-07 22:32:43 +00002455
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002456 return Changed ? CS.getInstruction() : 0;
Chris Lattneraec3d942003-10-07 22:32:43 +00002457}
2458
Chris Lattner970c33a2003-06-19 17:00:31 +00002459// transformConstExprCastCall - If the callee is a constexpr cast of a function,
2460// attempt to move the cast to the arguments of the call/invoke.
2461//
2462bool InstCombiner::transformConstExprCastCall(CallSite CS) {
2463 if (!isa<ConstantExpr>(CS.getCalledValue())) return false;
2464 ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue());
2465 if (CE->getOpcode() != Instruction::Cast ||
2466 !isa<ConstantPointerRef>(CE->getOperand(0)))
2467 return false;
2468 ConstantPointerRef *CPR = cast<ConstantPointerRef>(CE->getOperand(0));
2469 if (!isa<Function>(CPR->getValue())) return false;
2470 Function *Callee = cast<Function>(CPR->getValue());
2471 Instruction *Caller = CS.getInstruction();
2472
2473 // Okay, this is a cast from a function to a different type. Unless doing so
2474 // would cause a type conversion of one of our arguments, change this call to
2475 // be a direct call with arguments casted to the appropriate types.
2476 //
2477 const FunctionType *FT = Callee->getFunctionType();
2478 const Type *OldRetTy = Caller->getType();
2479
Chris Lattner1f7942f2004-01-14 06:06:08 +00002480 // Check to see if we are changing the return type...
2481 if (OldRetTy != FT->getReturnType()) {
2482 if (Callee->isExternal() &&
2483 !OldRetTy->isLosslesslyConvertibleTo(FT->getReturnType()) &&
2484 !Caller->use_empty())
2485 return false; // Cannot transform this return value...
2486
2487 // If the callsite is an invoke instruction, and the return value is used by
2488 // a PHI node in a successor, we cannot change the return type of the call
2489 // because there is no place to put the cast instruction (without breaking
2490 // the critical edge). Bail out in this case.
2491 if (!Caller->use_empty())
2492 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller))
2493 for (Value::use_iterator UI = II->use_begin(), E = II->use_end();
2494 UI != E; ++UI)
2495 if (PHINode *PN = dyn_cast<PHINode>(*UI))
2496 if (PN->getParent() == II->getNormalDest() ||
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002497 PN->getParent() == II->getUnwindDest())
Chris Lattner1f7942f2004-01-14 06:06:08 +00002498 return false;
2499 }
Chris Lattner970c33a2003-06-19 17:00:31 +00002500
2501 unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin());
2502 unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs);
2503
2504 CallSite::arg_iterator AI = CS.arg_begin();
2505 for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) {
2506 const Type *ParamTy = FT->getParamType(i);
2507 bool isConvertible = (*AI)->getType()->isLosslesslyConvertibleTo(ParamTy);
2508 if (Callee->isExternal() && !isConvertible) return false;
2509 }
2510
2511 if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() &&
2512 Callee->isExternal())
2513 return false; // Do not delete arguments unless we have a function body...
2514
2515 // Okay, we decided that this is a safe thing to do: go ahead and start
2516 // inserting cast instructions as necessary...
2517 std::vector<Value*> Args;
2518 Args.reserve(NumActualArgs);
2519
2520 AI = CS.arg_begin();
2521 for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) {
2522 const Type *ParamTy = FT->getParamType(i);
2523 if ((*AI)->getType() == ParamTy) {
2524 Args.push_back(*AI);
2525 } else {
Chris Lattner1c631e82004-04-08 04:43:23 +00002526 Args.push_back(InsertNewInstBefore(new CastInst(*AI, ParamTy, "tmp"),
2527 *Caller));
Chris Lattner970c33a2003-06-19 17:00:31 +00002528 }
2529 }
2530
2531 // If the function takes more arguments than the call was taking, add them
2532 // now...
2533 for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i)
2534 Args.push_back(Constant::getNullValue(FT->getParamType(i)));
2535
2536 // If we are removing arguments to the function, emit an obnoxious warning...
2537 if (FT->getNumParams() < NumActualArgs)
2538 if (!FT->isVarArg()) {
2539 std::cerr << "WARNING: While resolving call to function '"
2540 << Callee->getName() << "' arguments were dropped!\n";
2541 } else {
2542 // Add all of the arguments in their promoted form to the arg list...
2543 for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) {
2544 const Type *PTy = getPromotedType((*AI)->getType());
2545 if (PTy != (*AI)->getType()) {
2546 // Must promote to pass through va_arg area!
2547 Instruction *Cast = new CastInst(*AI, PTy, "tmp");
2548 InsertNewInstBefore(Cast, *Caller);
2549 Args.push_back(Cast);
2550 } else {
2551 Args.push_back(*AI);
2552 }
2553 }
2554 }
2555
2556 if (FT->getReturnType() == Type::VoidTy)
2557 Caller->setName(""); // Void type should not have a name...
2558
2559 Instruction *NC;
2560 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002561 NC = new InvokeInst(Callee, II->getNormalDest(), II->getUnwindDest(),
Chris Lattner970c33a2003-06-19 17:00:31 +00002562 Args, Caller->getName(), Caller);
2563 } else {
2564 NC = new CallInst(Callee, Args, Caller->getName(), Caller);
2565 }
2566
2567 // Insert a cast of the return type as necessary...
2568 Value *NV = NC;
2569 if (Caller->getType() != NV->getType() && !Caller->use_empty()) {
2570 if (NV->getType() != Type::VoidTy) {
2571 NV = NC = new CastInst(NC, Caller->getType(), "tmp");
Chris Lattner686767f2003-10-30 00:46:41 +00002572
2573 // If this is an invoke instruction, we should insert it after the first
2574 // non-phi, instruction in the normal successor block.
2575 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
2576 BasicBlock::iterator I = II->getNormalDest()->begin();
2577 while (isa<PHINode>(I)) ++I;
2578 InsertNewInstBefore(NC, *I);
2579 } else {
2580 // Otherwise, it's a call, just insert cast right after the call instr
2581 InsertNewInstBefore(NC, *Caller);
2582 }
Chris Lattner51ea1272004-02-28 05:22:00 +00002583 AddUsersToWorkList(*Caller);
Chris Lattner970c33a2003-06-19 17:00:31 +00002584 } else {
2585 NV = Constant::getNullValue(Caller->getType());
2586 }
2587 }
2588
2589 if (Caller->getType() != Type::VoidTy && !Caller->use_empty())
2590 Caller->replaceAllUsesWith(NV);
2591 Caller->getParent()->getInstList().erase(Caller);
2592 removeFromWorkList(Caller);
2593 return true;
2594}
2595
2596
Chris Lattner48a44f72002-05-02 17:06:02 +00002597
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002598// PHINode simplification
2599//
Chris Lattner113f4f42002-06-25 16:13:24 +00002600Instruction *InstCombiner::visitPHINode(PHINode &PN) {
Chris Lattner91daeb52003-12-19 05:58:40 +00002601 if (Value *V = hasConstantValue(&PN))
2602 return ReplaceInstUsesWith(PN, V);
Chris Lattner4db2d222004-02-16 05:07:08 +00002603
2604 // If the only user of this instruction is a cast instruction, and all of the
2605 // incoming values are constants, change this PHI to merge together the casted
2606 // constants.
2607 if (PN.hasOneUse())
2608 if (CastInst *CI = dyn_cast<CastInst>(PN.use_back()))
2609 if (CI->getType() != PN.getType()) { // noop casts will be folded
2610 bool AllConstant = true;
2611 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i)
2612 if (!isa<Constant>(PN.getIncomingValue(i))) {
2613 AllConstant = false;
2614 break;
2615 }
2616 if (AllConstant) {
2617 // Make a new PHI with all casted values.
2618 PHINode *New = new PHINode(CI->getType(), PN.getName(), &PN);
2619 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
2620 Constant *OldArg = cast<Constant>(PN.getIncomingValue(i));
2621 New->addIncoming(ConstantExpr::getCast(OldArg, New->getType()),
2622 PN.getIncomingBlock(i));
2623 }
2624
2625 // Update the cast instruction.
2626 CI->setOperand(0, New);
2627 WorkList.push_back(CI); // revisit the cast instruction to fold.
2628 WorkList.push_back(New); // Make sure to revisit the new Phi
2629 return &PN; // PN is now dead!
2630 }
2631 }
Chris Lattner91daeb52003-12-19 05:58:40 +00002632 return 0;
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002633}
2634
Chris Lattner69193f92004-04-05 01:30:19 +00002635static Value *InsertSignExtendToPtrTy(Value *V, const Type *DTy,
2636 Instruction *InsertPoint,
2637 InstCombiner *IC) {
2638 unsigned PS = IC->getTargetData().getPointerSize();
2639 const Type *VTy = V->getType();
2640 Instruction *Cast;
2641 if (!VTy->isSigned() && VTy->getPrimitiveSize() < PS)
2642 // We must insert a cast to ensure we sign-extend.
2643 V = IC->InsertNewInstBefore(new CastInst(V, VTy->getSignedVersion(),
2644 V->getName()), *InsertPoint);
2645 return IC->InsertNewInstBefore(new CastInst(V, DTy, V->getName()),
2646 *InsertPoint);
2647}
2648
Chris Lattner48a44f72002-05-02 17:06:02 +00002649
Chris Lattner113f4f42002-06-25 16:13:24 +00002650Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002651 Value *PtrOp = GEP.getOperand(0);
Chris Lattner471bd762003-05-22 19:07:21 +00002652 // Is it 'getelementptr %P, long 0' or 'getelementptr %P'
Chris Lattner113f4f42002-06-25 16:13:24 +00002653 // If so, eliminate the noop.
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002654 if (GEP.getNumOperands() == 1)
Chris Lattner5f667a62004-05-07 22:09:22 +00002655 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002656
2657 bool HasZeroPointerIndex = false;
2658 if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1)))
2659 HasZeroPointerIndex = C->isNullValue();
2660
2661 if (GEP.getNumOperands() == 2 && HasZeroPointerIndex)
Chris Lattner5f667a62004-05-07 22:09:22 +00002662 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner48a44f72002-05-02 17:06:02 +00002663
Chris Lattner69193f92004-04-05 01:30:19 +00002664 // Eliminate unneeded casts for indices.
2665 bool MadeChange = false;
Chris Lattner2b2412d2004-04-07 18:38:20 +00002666 gep_type_iterator GTI = gep_type_begin(GEP);
2667 for (unsigned i = 1, e = GEP.getNumOperands(); i != e; ++i, ++GTI)
2668 if (isa<SequentialType>(*GTI)) {
2669 if (CastInst *CI = dyn_cast<CastInst>(GEP.getOperand(i))) {
2670 Value *Src = CI->getOperand(0);
2671 const Type *SrcTy = Src->getType();
2672 const Type *DestTy = CI->getType();
2673 if (Src->getType()->isInteger()) {
2674 if (SrcTy->getPrimitiveSize() == DestTy->getPrimitiveSize()) {
2675 // We can always eliminate a cast from ulong or long to the other.
2676 // We can always eliminate a cast from uint to int or the other on
2677 // 32-bit pointer platforms.
2678 if (DestTy->getPrimitiveSize() >= TD->getPointerSize()) {
2679 MadeChange = true;
2680 GEP.setOperand(i, Src);
2681 }
2682 } else if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
2683 SrcTy->getPrimitiveSize() == 4) {
2684 // We can always eliminate a cast from int to [u]long. We can
2685 // eliminate a cast from uint to [u]long iff the target is a 32-bit
2686 // pointer target.
2687 if (SrcTy->isSigned() ||
2688 SrcTy->getPrimitiveSize() >= TD->getPointerSize()) {
2689 MadeChange = true;
2690 GEP.setOperand(i, Src);
2691 }
Chris Lattner69193f92004-04-05 01:30:19 +00002692 }
2693 }
2694 }
Chris Lattner2b2412d2004-04-07 18:38:20 +00002695 // If we are using a wider index than needed for this platform, shrink it
2696 // to what we need. If the incoming value needs a cast instruction,
2697 // insert it. This explicit cast can make subsequent optimizations more
2698 // obvious.
2699 Value *Op = GEP.getOperand(i);
2700 if (Op->getType()->getPrimitiveSize() > TD->getPointerSize())
Chris Lattner1e9ac1a2004-04-17 18:16:10 +00002701 if (Constant *C = dyn_cast<Constant>(Op)) {
2702 GEP.setOperand(i, ConstantExpr::getCast(C, TD->getIntPtrType()));
2703 MadeChange = true;
2704 } else {
Chris Lattner2b2412d2004-04-07 18:38:20 +00002705 Op = InsertNewInstBefore(new CastInst(Op, TD->getIntPtrType(),
2706 Op->getName()), GEP);
2707 GEP.setOperand(i, Op);
2708 MadeChange = true;
2709 }
Chris Lattner69193f92004-04-05 01:30:19 +00002710 }
2711 if (MadeChange) return &GEP;
2712
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002713 // Combine Indices - If the source pointer to this getelementptr instruction
2714 // is a getelementptr instruction, combine the indices of the two
2715 // getelementptr instructions into a single instruction.
2716 //
Chris Lattner57c67b02004-03-25 22:59:29 +00002717 std::vector<Value*> SrcGEPOperands;
Chris Lattner5f667a62004-05-07 22:09:22 +00002718 if (GetElementPtrInst *Src = dyn_cast<GetElementPtrInst>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002719 SrcGEPOperands.assign(Src->op_begin(), Src->op_end());
Chris Lattner5f667a62004-05-07 22:09:22 +00002720 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002721 if (CE->getOpcode() == Instruction::GetElementPtr)
2722 SrcGEPOperands.assign(CE->op_begin(), CE->op_end());
2723 }
2724
2725 if (!SrcGEPOperands.empty()) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002726 // Note that if our source is a gep chain itself that we wait for that
2727 // chain to be resolved before we perform this transformation. This
2728 // avoids us creating a TON of code in some cases.
2729 //
2730 if (isa<GetElementPtrInst>(SrcGEPOperands[0]) &&
2731 cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2)
2732 return 0; // Wait until our source is folded to completion.
2733
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002734 std::vector<Value *> Indices;
Chris Lattner5f667a62004-05-07 22:09:22 +00002735
2736 // Find out whether the last index in the source GEP is a sequential idx.
2737 bool EndsWithSequential = false;
2738 for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)),
2739 E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I)
Chris Lattner8ec5f882004-05-08 22:41:42 +00002740 EndsWithSequential = !isa<StructType>(*I);
Chris Lattnerca081252001-12-14 16:52:21 +00002741
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002742 // Can we combine the two pointer arithmetics offsets?
Chris Lattner5f667a62004-05-07 22:09:22 +00002743 if (EndsWithSequential) {
Chris Lattner235af562003-03-05 22:33:14 +00002744 // Replace: gep (gep %P, long B), long A, ...
2745 // With: T = long A+B; gep %P, T, ...
2746 //
Chris Lattner5f667a62004-05-07 22:09:22 +00002747 Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1);
Chris Lattner69193f92004-04-05 01:30:19 +00002748 if (SO1 == Constant::getNullValue(SO1->getType())) {
2749 Sum = GO1;
2750 } else if (GO1 == Constant::getNullValue(GO1->getType())) {
2751 Sum = SO1;
2752 } else {
2753 // If they aren't the same type, convert both to an integer of the
2754 // target's pointer size.
2755 if (SO1->getType() != GO1->getType()) {
2756 if (Constant *SO1C = dyn_cast<Constant>(SO1)) {
2757 SO1 = ConstantExpr::getCast(SO1C, GO1->getType());
2758 } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) {
2759 GO1 = ConstantExpr::getCast(GO1C, SO1->getType());
2760 } else {
2761 unsigned PS = TD->getPointerSize();
2762 Instruction *Cast;
2763 if (SO1->getType()->getPrimitiveSize() == PS) {
2764 // Convert GO1 to SO1's type.
2765 GO1 = InsertSignExtendToPtrTy(GO1, SO1->getType(), &GEP, this);
2766
2767 } else if (GO1->getType()->getPrimitiveSize() == PS) {
2768 // Convert SO1 to GO1's type.
2769 SO1 = InsertSignExtendToPtrTy(SO1, GO1->getType(), &GEP, this);
2770 } else {
2771 const Type *PT = TD->getIntPtrType();
2772 SO1 = InsertSignExtendToPtrTy(SO1, PT, &GEP, this);
2773 GO1 = InsertSignExtendToPtrTy(GO1, PT, &GEP, this);
2774 }
2775 }
2776 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002777 if (isa<Constant>(SO1) && isa<Constant>(GO1))
2778 Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1));
2779 else {
2780 Sum = BinaryOperator::create(Instruction::Add, SO1, GO1,
2781 PtrOp->getName()+".sum", &GEP);
2782 WorkList.push_back(cast<Instruction>(Sum));
2783 }
Chris Lattner69193f92004-04-05 01:30:19 +00002784 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002785
2786 // Recycle the GEP we already have if possible.
2787 if (SrcGEPOperands.size() == 2) {
2788 GEP.setOperand(0, SrcGEPOperands[0]);
2789 GEP.setOperand(1, Sum);
2790 return &GEP;
2791 } else {
2792 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2793 SrcGEPOperands.end()-1);
2794 Indices.push_back(Sum);
2795 Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end());
2796 }
Chris Lattner69193f92004-04-05 01:30:19 +00002797 } else if (isa<Constant>(*GEP.idx_begin()) &&
2798 cast<Constant>(*GEP.idx_begin())->isNullValue() &&
Chris Lattner57c67b02004-03-25 22:59:29 +00002799 SrcGEPOperands.size() != 1) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002800 // Otherwise we can do the fold if the first index of the GEP is a zero
Chris Lattner57c67b02004-03-25 22:59:29 +00002801 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2802 SrcGEPOperands.end());
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002803 Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end());
2804 }
2805
2806 if (!Indices.empty())
Chris Lattner57c67b02004-03-25 22:59:29 +00002807 return new GetElementPtrInst(SrcGEPOperands[0], Indices, GEP.getName());
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002808
Chris Lattner5f667a62004-05-07 22:09:22 +00002809 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) {
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002810 // GEP of global variable. If all of the indices for this GEP are
2811 // constants, we can promote this to a constexpr instead of an instruction.
2812
2813 // Scan for nonconstants...
2814 std::vector<Constant*> Indices;
2815 User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end();
2816 for (; I != E && isa<Constant>(*I); ++I)
2817 Indices.push_back(cast<Constant>(*I));
2818
2819 if (I == E) { // If they are all constants...
Chris Lattner46b3d302003-04-16 22:40:51 +00002820 Constant *CE =
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002821 ConstantExpr::getGetElementPtr(ConstantPointerRef::get(GV), Indices);
2822
2823 // Replace all uses of the GEP with the new constexpr...
2824 return ReplaceInstUsesWith(GEP, CE);
2825 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002826 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002827 if (CE->getOpcode() == Instruction::Cast) {
2828 if (HasZeroPointerIndex) {
2829 // transform: GEP (cast [10 x ubyte]* X to [0 x ubyte]*), long 0, ...
2830 // into : GEP [10 x ubyte]* X, long 0, ...
2831 //
2832 // This occurs when the program declares an array extern like "int X[];"
2833 //
2834 Constant *X = CE->getOperand(0);
2835 const PointerType *CPTy = cast<PointerType>(CE->getType());
2836 if (const PointerType *XTy = dyn_cast<PointerType>(X->getType()))
2837 if (const ArrayType *XATy =
2838 dyn_cast<ArrayType>(XTy->getElementType()))
2839 if (const ArrayType *CATy =
2840 dyn_cast<ArrayType>(CPTy->getElementType()))
2841 if (CATy->getElementType() == XATy->getElementType()) {
2842 // At this point, we know that the cast source type is a pointer
2843 // to an array of the same type as the destination pointer
2844 // array. Because the array type is never stepped over (there
2845 // is a leading zero) we can fold the cast into this GEP.
2846 GEP.setOperand(0, X);
2847 return &GEP;
2848 }
2849 }
2850 }
Chris Lattnerca081252001-12-14 16:52:21 +00002851 }
2852
Chris Lattnerca081252001-12-14 16:52:21 +00002853 return 0;
2854}
2855
Chris Lattner1085bdf2002-11-04 16:18:53 +00002856Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) {
2857 // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1
2858 if (AI.isArrayAllocation()) // Check C != 1
2859 if (const ConstantUInt *C = dyn_cast<ConstantUInt>(AI.getArraySize())) {
2860 const Type *NewTy = ArrayType::get(AI.getAllocatedType(), C->getValue());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002861 AllocationInst *New = 0;
Chris Lattner1085bdf2002-11-04 16:18:53 +00002862
2863 // Create and insert the replacement instruction...
2864 if (isa<MallocInst>(AI))
Chris Lattnerabb77c92004-03-19 06:08:10 +00002865 New = new MallocInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002866 else {
2867 assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!");
Chris Lattnerabb77c92004-03-19 06:08:10 +00002868 New = new AllocaInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002869 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002870
2871 InsertNewInstBefore(New, AI);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002872
2873 // Scan to the end of the allocation instructions, to skip over a block of
2874 // allocas if possible...
2875 //
2876 BasicBlock::iterator It = New;
2877 while (isa<AllocationInst>(*It)) ++It;
2878
2879 // Now that I is pointing to the first non-allocation-inst in the block,
2880 // insert our getelementptr instruction...
2881 //
Chris Lattner69193f92004-04-05 01:30:19 +00002882 std::vector<Value*> Idx(2, Constant::getNullValue(Type::IntTy));
Chris Lattner1085bdf2002-11-04 16:18:53 +00002883 Value *V = new GetElementPtrInst(New, Idx, New->getName()+".sub", It);
2884
2885 // Now make everything use the getelementptr instead of the original
2886 // allocation.
Chris Lattnerabb77c92004-03-19 06:08:10 +00002887 return ReplaceInstUsesWith(AI, V);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002888 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002889
2890 // If alloca'ing a zero byte object, replace the alloca with a null pointer.
2891 // Note that we only do this for alloca's, because malloc should allocate and
2892 // return a unique pointer, even for a zero byte allocation.
2893 if (isa<AllocaInst>(AI) && TD->getTypeSize(AI.getAllocatedType()) == 0)
2894 return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType()));
2895
Chris Lattner1085bdf2002-11-04 16:18:53 +00002896 return 0;
2897}
2898
Chris Lattner8427bff2003-12-07 01:24:23 +00002899Instruction *InstCombiner::visitFreeInst(FreeInst &FI) {
2900 Value *Op = FI.getOperand(0);
2901
2902 // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X
2903 if (CastInst *CI = dyn_cast<CastInst>(Op))
2904 if (isa<PointerType>(CI->getOperand(0)->getType())) {
2905 FI.setOperand(0, CI->getOperand(0));
2906 return &FI;
2907 }
2908
Chris Lattnerf3a36602004-02-28 04:57:37 +00002909 // If we have 'free null' delete the instruction. This can happen in stl code
2910 // when lots of inlining happens.
Chris Lattner51ea1272004-02-28 05:22:00 +00002911 if (isa<ConstantPointerNull>(Op))
2912 return EraseInstFromFunction(FI);
Chris Lattnerf3a36602004-02-28 04:57:37 +00002913
Chris Lattner8427bff2003-12-07 01:24:23 +00002914 return 0;
2915}
2916
2917
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002918/// GetGEPGlobalInitializer - Given a constant, and a getelementptr
2919/// constantexpr, return the constant value being addressed by the constant
2920/// expression, or null if something is funny.
2921///
2922static Constant *GetGEPGlobalInitializer(Constant *C, ConstantExpr *CE) {
Chris Lattner69193f92004-04-05 01:30:19 +00002923 if (CE->getOperand(1) != Constant::getNullValue(CE->getOperand(1)->getType()))
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002924 return 0; // Do not allow stepping over the value!
2925
2926 // Loop over all of the operands, tracking down which value we are
2927 // addressing...
Chris Lattnered79d8a2004-05-27 17:30:27 +00002928 gep_type_iterator I = gep_type_begin(CE), E = gep_type_end(CE);
2929 for (++I; I != E; ++I)
2930 if (const StructType *STy = dyn_cast<StructType>(*I)) {
2931 ConstantUInt *CU = cast<ConstantUInt>(I.getOperand());
2932 assert(CU->getValue() < STy->getNumElements() &&
2933 "Struct index out of range!");
2934 if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) {
2935 C = cast<Constant>(CS->getValues()[CU->getValue()]);
2936 } else if (isa<ConstantAggregateZero>(C)) {
2937 C = Constant::getNullValue(STy->getElementType(CU->getValue()));
2938 } else {
2939 return 0;
2940 }
2941 } else if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand())) {
2942 const ArrayType *ATy = cast<ArrayType>(*I);
2943 if ((uint64_t)CI->getRawValue() >= ATy->getNumElements()) return 0;
2944 if (ConstantArray *CA = dyn_cast<ConstantArray>(C))
2945 C = cast<Constant>(CA->getValues()[CI->getRawValue()]);
2946 else if (isa<ConstantAggregateZero>(C))
2947 C = Constant::getNullValue(ATy->getElementType());
2948 else
2949 return 0;
2950 } else {
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002951 return 0;
Chris Lattnered79d8a2004-05-27 17:30:27 +00002952 }
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002953 return C;
2954}
2955
2956Instruction *InstCombiner::visitLoadInst(LoadInst &LI) {
2957 Value *Op = LI.getOperand(0);
Chris Lattner7e8af382004-01-12 04:13:56 +00002958 if (LI.isVolatile()) return 0;
2959
Chris Lattner6679e462004-04-14 03:28:36 +00002960 if (Constant *C = dyn_cast<Constant>(Op))
2961 if (C->isNullValue()) // load null -> 0
2962 return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType()));
2963 else if (ConstantPointerRef *CPR = dyn_cast<ConstantPointerRef>(C))
2964 Op = CPR->getValue();
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002965
2966 // Instcombine load (constant global) into the value loaded...
2967 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002968 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002969 return ReplaceInstUsesWith(LI, GV->getInitializer());
2970
2971 // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded...
2972 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op))
2973 if (CE->getOpcode() == Instruction::GetElementPtr)
2974 if (ConstantPointerRef *G=dyn_cast<ConstantPointerRef>(CE->getOperand(0)))
2975 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(G->getValue()))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002976 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002977 if (Constant *V = GetGEPGlobalInitializer(GV->getInitializer(), CE))
2978 return ReplaceInstUsesWith(LI, V);
Chris Lattnere228ee52004-04-08 20:39:49 +00002979
2980 // load (cast X) --> cast (load X) iff safe
2981 if (CastInst *CI = dyn_cast<CastInst>(Op)) {
2982 const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType();
2983 if (const PointerType *SrcTy =
2984 dyn_cast<PointerType>(CI->getOperand(0)->getType())) {
2985 const Type *SrcPTy = SrcTy->getElementType();
Chris Lattnerd9e58132004-05-07 15:35:56 +00002986 if (SrcPTy->isSized() && DestPTy->isSized() &&
2987 TD->getTypeSize(SrcPTy) == TD->getTypeSize(DestPTy) &&
Chris Lattnere228ee52004-04-08 20:39:49 +00002988 (SrcPTy->isInteger() || isa<PointerType>(SrcPTy)) &&
2989 (DestPTy->isInteger() || isa<PointerType>(DestPTy))) {
2990 // Okay, we are casting from one integer or pointer type to another of
2991 // the same size. Instead of casting the pointer before the load, cast
2992 // the result of the loaded value.
2993 Value *NewLoad = InsertNewInstBefore(new LoadInst(CI->getOperand(0),
2994 CI->getName()), LI);
2995 // Now cast the result of the load.
2996 return new CastInst(NewLoad, LI.getType());
2997 }
2998 }
2999 }
3000
Chris Lattner0f1d8a32003-06-26 05:06:25 +00003001 return 0;
3002}
3003
3004
Chris Lattner9eef8a72003-06-04 04:46:00 +00003005Instruction *InstCombiner::visitBranchInst(BranchInst &BI) {
3006 // Change br (not X), label True, label False to: br X, label False, True
Chris Lattner4f7acca2004-02-27 06:27:46 +00003007 if (BI.isConditional() && !isa<Constant>(BI.getCondition())) {
Chris Lattnere967b342003-06-04 05:10:11 +00003008 if (Value *V = dyn_castNotVal(BI.getCondition())) {
3009 BasicBlock *TrueDest = BI.getSuccessor(0);
3010 BasicBlock *FalseDest = BI.getSuccessor(1);
3011 // Swap Destinations and condition...
3012 BI.setCondition(V);
3013 BI.setSuccessor(0, FalseDest);
3014 BI.setSuccessor(1, TrueDest);
3015 return &BI;
Chris Lattner4f7acca2004-02-27 06:27:46 +00003016 } else if (SetCondInst *I = dyn_cast<SetCondInst>(BI.getCondition())) {
3017 // Cannonicalize setne -> seteq
3018 if ((I->getOpcode() == Instruction::SetNE ||
3019 I->getOpcode() == Instruction::SetLE ||
3020 I->getOpcode() == Instruction::SetGE) && I->hasOneUse()) {
3021 std::string Name = I->getName(); I->setName("");
3022 Instruction::BinaryOps NewOpcode =
3023 SetCondInst::getInverseCondition(I->getOpcode());
3024 Value *NewSCC = BinaryOperator::create(NewOpcode, I->getOperand(0),
3025 I->getOperand(1), Name, I);
3026 BasicBlock *TrueDest = BI.getSuccessor(0);
3027 BasicBlock *FalseDest = BI.getSuccessor(1);
3028 // Swap Destinations and condition...
3029 BI.setCondition(NewSCC);
3030 BI.setSuccessor(0, FalseDest);
3031 BI.setSuccessor(1, TrueDest);
3032 removeFromWorkList(I);
3033 I->getParent()->getInstList().erase(I);
3034 WorkList.push_back(cast<Instruction>(NewSCC));
3035 return &BI;
3036 }
Chris Lattnere967b342003-06-04 05:10:11 +00003037 }
Chris Lattner4f7acca2004-02-27 06:27:46 +00003038 }
Chris Lattner9eef8a72003-06-04 04:46:00 +00003039 return 0;
3040}
Chris Lattner1085bdf2002-11-04 16:18:53 +00003041
Chris Lattnerca081252001-12-14 16:52:21 +00003042
Chris Lattner99f48c62002-09-02 04:59:56 +00003043void InstCombiner::removeFromWorkList(Instruction *I) {
3044 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), I),
3045 WorkList.end());
3046}
3047
Chris Lattner113f4f42002-06-25 16:13:24 +00003048bool InstCombiner::runOnFunction(Function &F) {
Chris Lattner260ab202002-04-18 17:39:14 +00003049 bool Changed = false;
Chris Lattnerf4ad1652003-11-02 05:57:39 +00003050 TD = &getAnalysis<TargetData>();
Chris Lattnerca081252001-12-14 16:52:21 +00003051
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003052 for (inst_iterator i = inst_begin(F), e = inst_end(F); i != e; ++i)
3053 WorkList.push_back(&*i);
Chris Lattner2d3a7a62004-04-27 15:13:33 +00003054
Chris Lattnerca081252001-12-14 16:52:21 +00003055
3056 while (!WorkList.empty()) {
3057 Instruction *I = WorkList.back(); // Get an instruction from the worklist
3058 WorkList.pop_back();
3059
Misha Brukman632df282002-10-29 23:06:16 +00003060 // Check to see if we can DCE or ConstantPropagate the instruction...
Chris Lattner99f48c62002-09-02 04:59:56 +00003061 // Check to see if we can DIE the instruction...
3062 if (isInstructionTriviallyDead(I)) {
3063 // Add operands to the worklist...
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003064 if (I->getNumOperands() < 4)
Chris Lattner51ea1272004-02-28 05:22:00 +00003065 AddUsesToWorkList(*I);
Chris Lattner99f48c62002-09-02 04:59:56 +00003066 ++NumDeadInst;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003067
3068 I->getParent()->getInstList().erase(I);
3069 removeFromWorkList(I);
3070 continue;
3071 }
Chris Lattner99f48c62002-09-02 04:59:56 +00003072
Misha Brukman632df282002-10-29 23:06:16 +00003073 // Instruction isn't dead, see if we can constant propagate it...
Chris Lattner99f48c62002-09-02 04:59:56 +00003074 if (Constant *C = ConstantFoldInstruction(I)) {
3075 // Add operands to the worklist...
Chris Lattner51ea1272004-02-28 05:22:00 +00003076 AddUsesToWorkList(*I);
Chris Lattnerc6509f42002-12-05 22:41:53 +00003077 ReplaceInstUsesWith(*I, C);
3078
Chris Lattner99f48c62002-09-02 04:59:56 +00003079 ++NumConstProp;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003080 I->getParent()->getInstList().erase(I);
Chris Lattner800aaaf2003-10-07 15:17:02 +00003081 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003082 continue;
Chris Lattner99f48c62002-09-02 04:59:56 +00003083 }
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003084
Chris Lattner57c67b02004-03-25 22:59:29 +00003085 // Check to see if any of the operands of this instruction are a
3086 // ConstantPointerRef. Since they sneak in all over the place and inhibit
3087 // optimization, we want to strip them out unconditionally!
3088 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3089 if (ConstantPointerRef *CPR =
3090 dyn_cast<ConstantPointerRef>(I->getOperand(i))) {
3091 I->setOperand(i, CPR->getValue());
3092 Changed = true;
3093 }
3094
Chris Lattnerca081252001-12-14 16:52:21 +00003095 // Now that we have an instruction, try combining it to simplify it...
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003096 if (Instruction *Result = visit(*I)) {
Chris Lattner0b18c1d2002-05-10 15:38:35 +00003097 ++NumCombined;
Chris Lattner260ab202002-04-18 17:39:14 +00003098 // Should we replace the old instruction with a new one?
Chris Lattner053c0932002-05-14 15:24:07 +00003099 if (Result != I) {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003100 DEBUG(std::cerr << "IC: Old = " << *I
3101 << " New = " << *Result);
3102
Chris Lattner396dbfe2004-06-09 05:08:07 +00003103 // Everything uses the new instruction now.
3104 I->replaceAllUsesWith(Result);
3105
3106 // Push the new instruction and any users onto the worklist.
3107 WorkList.push_back(Result);
3108 AddUsersToWorkList(*Result);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003109
3110 // Move the name to the new instruction first...
3111 std::string OldName = I->getName(); I->setName("");
Chris Lattner950fc782003-10-07 22:58:41 +00003112 Result->setName(OldName);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003113
3114 // Insert the new instruction into the basic block...
3115 BasicBlock *InstParent = I->getParent();
3116 InstParent->getInstList().insert(I, Result);
3117
Chris Lattner63d75af2004-05-01 23:27:23 +00003118 // Make sure that we reprocess all operands now that we reduced their
3119 // use counts.
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003120 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3121 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3122 WorkList.push_back(OpI);
3123
Chris Lattner396dbfe2004-06-09 05:08:07 +00003124 // Instructions can end up on the worklist more than once. Make sure
3125 // we do not process an instruction that has been deleted.
3126 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003127
3128 // Erase the old instruction.
3129 InstParent->getInstList().erase(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00003130 } else {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003131 DEBUG(std::cerr << "IC: MOD = " << *I);
3132
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003133 // If the instruction was modified, it's possible that it is now dead.
3134 // if so, remove it.
Chris Lattner63d75af2004-05-01 23:27:23 +00003135 if (isInstructionTriviallyDead(I)) {
3136 // Make sure we process all operands now that we are reducing their
3137 // use counts.
3138 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3139 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3140 WorkList.push_back(OpI);
3141
3142 // Instructions may end up in the worklist more than once. Erase all
3143 // occurrances of this instruction.
Chris Lattner99f48c62002-09-02 04:59:56 +00003144 removeFromWorkList(I);
Chris Lattner63d75af2004-05-01 23:27:23 +00003145 I->getParent()->getInstList().erase(I);
Chris Lattner396dbfe2004-06-09 05:08:07 +00003146 } else {
3147 WorkList.push_back(Result);
3148 AddUsersToWorkList(*Result);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003149 }
Chris Lattner053c0932002-05-14 15:24:07 +00003150 }
Chris Lattner260ab202002-04-18 17:39:14 +00003151 Changed = true;
Chris Lattnerca081252001-12-14 16:52:21 +00003152 }
3153 }
3154
Chris Lattner260ab202002-04-18 17:39:14 +00003155 return Changed;
Chris Lattner04805fa2002-02-26 21:46:54 +00003156}
3157
Chris Lattner8427bff2003-12-07 01:24:23 +00003158Pass *llvm::createInstructionCombiningPass() {
Chris Lattner260ab202002-04-18 17:39:14 +00003159 return new InstCombiner();
Chris Lattner04805fa2002-02-26 21:46:54 +00003160}
Brian Gaeke960707c2003-11-11 22:41:34 +00003161