blob: a5fcbc92b91fde75bf7d9fe866a25583007414b4 [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:
15// %Y = add int 1, %X
16// %Z = add int 1, %Y
17// into:
18// %Z = add int 2, %X
19//
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 Lattnerb4cfa7f2002-05-07 20:03:00 +000036#include "llvm/Transforms/Scalar.h"
Chris Lattner471bd762003-05-22 19:07:21 +000037#include "llvm/Instructions.h"
Chris Lattner04805fa2002-02-26 21:46:54 +000038#include "llvm/Pass.h"
Chris Lattner34428442003-05-27 16:40:51 +000039#include "llvm/Constants.h"
Chris Lattner1085bdf2002-11-04 16:18:53 +000040#include "llvm/DerivedTypes.h"
Chris Lattner0f1d8a32003-06-26 05:06:25 +000041#include "llvm/GlobalVariable.h"
Chris Lattnerf4ad1652003-11-02 05:57:39 +000042#include "llvm/Target/TargetData.h"
43#include "llvm/Transforms/Utils/BasicBlockUtils.h"
44#include "llvm/Transforms/Utils/Local.h"
Chris Lattner60a65912002-02-12 21:07:25 +000045#include "llvm/Support/InstIterator.h"
Chris Lattner260ab202002-04-18 17:39:14 +000046#include "llvm/Support/InstVisitor.h"
Chris Lattner970c33a2003-06-19 17:00:31 +000047#include "llvm/Support/CallSite.h"
Chris Lattnerbf3a0992002-10-01 22:38:41 +000048#include "Support/Statistic.h"
Chris Lattner053c0932002-05-14 15:24:07 +000049#include <algorithm>
Chris Lattner8427bff2003-12-07 01:24:23 +000050using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000051
Chris Lattner260ab202002-04-18 17:39:14 +000052namespace {
Chris Lattnerbf3a0992002-10-01 22:38:41 +000053 Statistic<> NumCombined ("instcombine", "Number of insts combined");
54 Statistic<> NumConstProp("instcombine", "Number of constant folds");
55 Statistic<> NumDeadInst ("instcombine", "Number of dead inst eliminated");
56
Chris Lattnerc8e66542002-04-27 06:56:12 +000057 class InstCombiner : public FunctionPass,
Chris Lattner260ab202002-04-18 17:39:14 +000058 public InstVisitor<InstCombiner, Instruction*> {
59 // Worklist of all of the instructions that need to be simplified.
60 std::vector<Instruction*> WorkList;
Chris Lattnerf4ad1652003-11-02 05:57:39 +000061 TargetData *TD;
Chris Lattner260ab202002-04-18 17:39:14 +000062
Chris Lattner113f4f42002-06-25 16:13:24 +000063 void AddUsesToWorkList(Instruction &I) {
Chris Lattner260ab202002-04-18 17:39:14 +000064 // The instruction was simplified, add all users of the instruction to
65 // the work lists because they might get more simplified now...
66 //
Chris Lattner113f4f42002-06-25 16:13:24 +000067 for (Value::use_iterator UI = I.use_begin(), UE = I.use_end();
Chris Lattner260ab202002-04-18 17:39:14 +000068 UI != UE; ++UI)
69 WorkList.push_back(cast<Instruction>(*UI));
70 }
71
Chris Lattner99f48c62002-09-02 04:59:56 +000072 // removeFromWorkList - remove all instances of I from the worklist.
73 void removeFromWorkList(Instruction *I);
Chris Lattner260ab202002-04-18 17:39:14 +000074 public:
Chris Lattner113f4f42002-06-25 16:13:24 +000075 virtual bool runOnFunction(Function &F);
Chris Lattner260ab202002-04-18 17:39:14 +000076
Chris Lattnerf12cc842002-04-28 21:27:06 +000077 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Chris Lattnerf4ad1652003-11-02 05:57:39 +000078 AU.addRequired<TargetData>();
Chris Lattner820d9712002-10-21 20:00:28 +000079 AU.setPreservesCFG();
Chris Lattnerf12cc842002-04-28 21:27:06 +000080 }
81
Chris Lattner260ab202002-04-18 17:39:14 +000082 // Visitation implementation - Implement instruction combining for different
83 // instruction types. The semantics are as follows:
84 // Return Value:
85 // null - No change was made
Chris Lattnere6794492002-08-12 21:17:25 +000086 // I - Change was made, I is still valid, I may be dead though
Chris Lattner260ab202002-04-18 17:39:14 +000087 // otherwise - Change was made, replace I with returned instruction
88 //
Chris Lattner113f4f42002-06-25 16:13:24 +000089 Instruction *visitAdd(BinaryOperator &I);
90 Instruction *visitSub(BinaryOperator &I);
91 Instruction *visitMul(BinaryOperator &I);
92 Instruction *visitDiv(BinaryOperator &I);
93 Instruction *visitRem(BinaryOperator &I);
94 Instruction *visitAnd(BinaryOperator &I);
95 Instruction *visitOr (BinaryOperator &I);
96 Instruction *visitXor(BinaryOperator &I);
97 Instruction *visitSetCondInst(BinaryOperator &I);
Chris Lattnere8d6c602003-03-10 19:16:08 +000098 Instruction *visitShiftInst(ShiftInst &I);
Chris Lattner113f4f42002-06-25 16:13:24 +000099 Instruction *visitCastInst(CastInst &CI);
Chris Lattner970c33a2003-06-19 17:00:31 +0000100 Instruction *visitCallInst(CallInst &CI);
101 Instruction *visitInvokeInst(InvokeInst &II);
Chris Lattner113f4f42002-06-25 16:13:24 +0000102 Instruction *visitPHINode(PHINode &PN);
103 Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP);
Chris Lattner1085bdf2002-11-04 16:18:53 +0000104 Instruction *visitAllocationInst(AllocationInst &AI);
Chris Lattner8427bff2003-12-07 01:24:23 +0000105 Instruction *visitFreeInst(FreeInst &FI);
Chris Lattner0f1d8a32003-06-26 05:06:25 +0000106 Instruction *visitLoadInst(LoadInst &LI);
Chris Lattner9eef8a72003-06-04 04:46:00 +0000107 Instruction *visitBranchInst(BranchInst &BI);
Chris Lattner260ab202002-04-18 17:39:14 +0000108
109 // visitInstruction - Specify what to return for unhandled instructions...
Chris Lattner113f4f42002-06-25 16:13:24 +0000110 Instruction *visitInstruction(Instruction &I) { return 0; }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000111
Chris Lattner970c33a2003-06-19 17:00:31 +0000112 private:
Chris Lattneraec3d942003-10-07 22:32:43 +0000113 Instruction *visitCallSite(CallSite CS);
Chris Lattner970c33a2003-06-19 17:00:31 +0000114 bool transformConstExprCastCall(CallSite CS);
115
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000116 // InsertNewInstBefore - insert an instruction New before instruction Old
117 // in the program. Add the new instruction to the worklist.
118 //
119 void InsertNewInstBefore(Instruction *New, Instruction &Old) {
Chris Lattner65217ff2002-08-23 18:32:43 +0000120 assert(New && New->getParent() == 0 &&
121 "New instruction already inserted into a basic block!");
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000122 BasicBlock *BB = Old.getParent();
123 BB->getInstList().insert(&Old, New); // Insert inst
124 WorkList.push_back(New); // Add to worklist
125 }
126
Chris Lattner3ac7c262003-08-13 20:16:26 +0000127 public:
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000128 // ReplaceInstUsesWith - This method is to be used when an instruction is
129 // found to be dead, replacable with another preexisting expression. Here
130 // we add all uses of I to the worklist, replace all uses of I with the new
131 // value, then return I, so that the inst combiner will know that I was
132 // modified.
133 //
134 Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) {
135 AddUsesToWorkList(I); // Add all modified instrs to worklist
136 I.replaceAllUsesWith(V);
137 return &I;
138 }
Chris Lattner3ac7c262003-08-13 20:16:26 +0000139 private:
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000140 /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
141 /// InsertBefore instruction. This is specialized a bit to avoid inserting
142 /// casts that are known to not do anything...
143 ///
144 Value *InsertOperandCastBefore(Value *V, const Type *DestTy,
145 Instruction *InsertBefore);
146
Chris Lattner7fb29e12003-03-11 00:12:48 +0000147 // SimplifyCommutative - This performs a few simplifications for commutative
148 // operators...
149 bool SimplifyCommutative(BinaryOperator &I);
Chris Lattnerba1cb382003-09-19 17:17:26 +0000150
151 Instruction *OptAndOp(Instruction *Op, ConstantIntegral *OpRHS,
152 ConstantIntegral *AndRHS, BinaryOperator &TheAnd);
Chris Lattner260ab202002-04-18 17:39:14 +0000153 };
Chris Lattnerb28b6802002-07-23 18:06:35 +0000154
Chris Lattnerc8b70922002-07-26 21:12:46 +0000155 RegisterOpt<InstCombiner> X("instcombine", "Combine redundant instructions");
Chris Lattner260ab202002-04-18 17:39:14 +0000156}
157
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000158// getComplexity: Assign a complexity or rank value to LLVM Values...
159// 0 -> Constant, 1 -> Other, 2 -> Argument, 2 -> Unary, 3 -> OtherInst
160static unsigned getComplexity(Value *V) {
161 if (isa<Instruction>(V)) {
162 if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V))
163 return 2;
164 return 3;
165 }
166 if (isa<Argument>(V)) return 2;
167 return isa<Constant>(V) ? 0 : 1;
168}
Chris Lattner260ab202002-04-18 17:39:14 +0000169
Chris Lattner7fb29e12003-03-11 00:12:48 +0000170// isOnlyUse - Return true if this instruction will be deleted if we stop using
171// it.
172static bool isOnlyUse(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000173 return V->hasOneUse() || isa<Constant>(V);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000174}
175
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000176// SimplifyCommutative - This performs a few simplifications for commutative
177// operators:
Chris Lattner260ab202002-04-18 17:39:14 +0000178//
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000179// 1. Order operands such that they are listed from right (least complex) to
180// left (most complex). This puts constants before unary operators before
181// binary operators.
182//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000183// 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2))
184// 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000185//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000186bool InstCombiner::SimplifyCommutative(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000187 bool Changed = false;
188 if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1)))
189 Changed = !I.swapOperands();
190
191 if (!I.isAssociative()) return Changed;
192 Instruction::BinaryOps Opcode = I.getOpcode();
Chris Lattner7fb29e12003-03-11 00:12:48 +0000193 if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0)))
194 if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) {
195 if (isa<Constant>(I.getOperand(1))) {
Chris Lattner34428442003-05-27 16:40:51 +0000196 Constant *Folded = ConstantExpr::get(I.getOpcode(),
197 cast<Constant>(I.getOperand(1)),
198 cast<Constant>(Op->getOperand(1)));
Chris Lattner7fb29e12003-03-11 00:12:48 +0000199 I.setOperand(0, Op->getOperand(0));
200 I.setOperand(1, Folded);
201 return true;
202 } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1)))
203 if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) &&
204 isOnlyUse(Op) && isOnlyUse(Op1)) {
205 Constant *C1 = cast<Constant>(Op->getOperand(1));
206 Constant *C2 = cast<Constant>(Op1->getOperand(1));
207
208 // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattner34428442003-05-27 16:40:51 +0000209 Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000210 Instruction *New = BinaryOperator::create(Opcode, Op->getOperand(0),
211 Op1->getOperand(0),
212 Op1->getName(), &I);
213 WorkList.push_back(New);
214 I.setOperand(0, New);
215 I.setOperand(1, Folded);
216 return true;
217 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000218 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000219 return Changed;
Chris Lattner260ab202002-04-18 17:39:14 +0000220}
Chris Lattnerca081252001-12-14 16:52:21 +0000221
Chris Lattnerbb74e222003-03-10 23:06:50 +0000222// dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction
223// if the LHS is a constant zero (which is the 'negate' form).
Chris Lattner9fa53de2002-05-06 16:49:18 +0000224//
Chris Lattnerbb74e222003-03-10 23:06:50 +0000225static inline Value *dyn_castNegVal(Value *V) {
226 if (BinaryOperator::isNeg(V))
227 return BinaryOperator::getNegArgument(cast<BinaryOperator>(V));
228
Chris Lattner9244df62003-04-30 22:19:10 +0000229 // Constants can be considered to be negated values if they can be folded...
230 if (Constant *C = dyn_cast<Constant>(V))
Chris Lattner34428442003-05-27 16:40:51 +0000231 return ConstantExpr::get(Instruction::Sub,
232 Constant::getNullValue(V->getType()), C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000233 return 0;
Chris Lattner9fa53de2002-05-06 16:49:18 +0000234}
235
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000236static Constant *NotConstant(Constant *C) {
237 return ConstantExpr::get(Instruction::Xor, C,
238 ConstantIntegral::getAllOnesValue(C->getType()));
239}
240
Chris Lattnerbb74e222003-03-10 23:06:50 +0000241static inline Value *dyn_castNotVal(Value *V) {
242 if (BinaryOperator::isNot(V))
243 return BinaryOperator::getNotArgument(cast<BinaryOperator>(V));
244
245 // Constants can be considered to be not'ed values...
Chris Lattnerdd65d862003-04-30 22:34:06 +0000246 if (ConstantIntegral *C = dyn_cast<ConstantIntegral>(V))
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000247 return NotConstant(C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000248 return 0;
249}
250
Chris Lattner7fb29e12003-03-11 00:12:48 +0000251// dyn_castFoldableMul - If this value is a multiply that can be folded into
252// other computations (because it has a constant operand), return the
253// non-constant operand of the multiply.
254//
255static inline Value *dyn_castFoldableMul(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000256 if (V->hasOneUse() && V->getType()->isInteger())
Chris Lattner7fb29e12003-03-11 00:12:48 +0000257 if (Instruction *I = dyn_cast<Instruction>(V))
258 if (I->getOpcode() == Instruction::Mul)
259 if (isa<Constant>(I->getOperand(1)))
260 return I->getOperand(0);
261 return 0;
Chris Lattner3082c5a2003-02-18 19:28:33 +0000262}
Chris Lattner31ae8632002-08-14 17:51:49 +0000263
Chris Lattner7fb29e12003-03-11 00:12:48 +0000264// dyn_castMaskingAnd - If this value is an And instruction masking a value with
265// a constant, return the constant being anded with.
266//
Chris Lattner01d56392003-08-12 19:17:27 +0000267template<class ValueType>
268static inline Constant *dyn_castMaskingAnd(ValueType *V) {
Chris Lattner7fb29e12003-03-11 00:12:48 +0000269 if (Instruction *I = dyn_cast<Instruction>(V))
270 if (I->getOpcode() == Instruction::And)
271 return dyn_cast<Constant>(I->getOperand(1));
272
273 // If this is a constant, it acts just like we were masking with it.
274 return dyn_cast<Constant>(V);
275}
Chris Lattner3082c5a2003-02-18 19:28:33 +0000276
277// Log2 - Calculate the log base 2 for the specified value if it is exactly a
278// power of 2.
279static unsigned Log2(uint64_t Val) {
280 assert(Val > 1 && "Values 0 and 1 should be handled elsewhere!");
281 unsigned Count = 0;
282 while (Val != 1) {
283 if (Val & 1) return 0; // Multiple bits set?
284 Val >>= 1;
285 ++Count;
286 }
287 return Count;
Chris Lattner31ae8632002-08-14 17:51:49 +0000288}
289
Chris Lattnerb8b97502003-08-13 19:01:45 +0000290
291/// AssociativeOpt - Perform an optimization on an associative operator. This
292/// function is designed to check a chain of associative operators for a
293/// potential to apply a certain optimization. Since the optimization may be
294/// applicable if the expression was reassociated, this checks the chain, then
295/// reassociates the expression as necessary to expose the optimization
296/// opportunity. This makes use of a special Functor, which must define
297/// 'shouldApply' and 'apply' methods.
298///
299template<typename Functor>
300Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) {
301 unsigned Opcode = Root.getOpcode();
302 Value *LHS = Root.getOperand(0);
303
304 // Quick check, see if the immediate LHS matches...
305 if (F.shouldApply(LHS))
306 return F.apply(Root);
307
308 // Otherwise, if the LHS is not of the same opcode as the root, return.
309 Instruction *LHSI = dyn_cast<Instruction>(LHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000310 while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) {
Chris Lattnerb8b97502003-08-13 19:01:45 +0000311 // Should we apply this transform to the RHS?
312 bool ShouldApply = F.shouldApply(LHSI->getOperand(1));
313
314 // If not to the RHS, check to see if we should apply to the LHS...
315 if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) {
316 cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS
317 ShouldApply = true;
318 }
319
320 // If the functor wants to apply the optimization to the RHS of LHSI,
321 // reassociate the expression from ((? op A) op B) to (? op (A op B))
322 if (ShouldApply) {
323 BasicBlock *BB = Root.getParent();
324 // All of the instructions have a single use and have no side-effects,
325 // because of this, we can pull them all into the current basic block.
326 if (LHSI->getParent() != BB) {
327 // Move all of the instructions from root to LHSI into the current
328 // block.
329 Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0));
330 Instruction *LastUse = &Root;
331 while (TmpLHSI->getParent() == BB) {
332 LastUse = TmpLHSI;
333 TmpLHSI = cast<Instruction>(TmpLHSI->getOperand(0));
334 }
335
336 // Loop over all of the instructions in other blocks, moving them into
337 // the current one.
338 Value *TmpLHS = TmpLHSI;
339 do {
340 TmpLHSI = cast<Instruction>(TmpLHS);
341 // Remove from current block...
342 TmpLHSI->getParent()->getInstList().remove(TmpLHSI);
343 // Insert before the last instruction...
344 BB->getInstList().insert(LastUse, TmpLHSI);
345 TmpLHS = TmpLHSI->getOperand(0);
346 } while (TmpLHSI != LHSI);
347 }
348
349 // Now all of the instructions are in the current basic block, go ahead
350 // and perform the reassociation.
351 Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0));
352
353 // First move the selected RHS to the LHS of the root...
354 Root.setOperand(0, LHSI->getOperand(1));
355
356 // Make what used to be the LHS of the root be the user of the root...
357 Value *ExtraOperand = TmpLHSI->getOperand(1);
358 Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI
359 TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root
360 BB->getInstList().remove(&Root); // Remove root from the BB
361 BB->getInstList().insert(TmpLHSI, &Root); // Insert root before TmpLHSI
362
363 // Now propagate the ExtraOperand down the chain of instructions until we
364 // get to LHSI.
365 while (TmpLHSI != LHSI) {
366 Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0));
367 Value *NextOp = NextLHSI->getOperand(1);
368 NextLHSI->setOperand(1, ExtraOperand);
369 TmpLHSI = NextLHSI;
370 ExtraOperand = NextOp;
371 }
372
373 // Now that the instructions are reassociated, have the functor perform
374 // the transformation...
375 return F.apply(Root);
376 }
377
378 LHSI = dyn_cast<Instruction>(LHSI->getOperand(0));
379 }
380 return 0;
381}
382
383
384// AddRHS - Implements: X + X --> X << 1
385struct AddRHS {
386 Value *RHS;
387 AddRHS(Value *rhs) : RHS(rhs) {}
388 bool shouldApply(Value *LHS) const { return LHS == RHS; }
389 Instruction *apply(BinaryOperator &Add) const {
390 return new ShiftInst(Instruction::Shl, Add.getOperand(0),
391 ConstantInt::get(Type::UByteTy, 1));
392 }
393};
394
395// AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2)
396// iff C1&C2 == 0
397struct AddMaskingAnd {
398 Constant *C2;
399 AddMaskingAnd(Constant *c) : C2(c) {}
400 bool shouldApply(Value *LHS) const {
401 if (Constant *C1 = dyn_castMaskingAnd(LHS))
402 return ConstantExpr::get(Instruction::And, C1, C2)->isNullValue();
403 return false;
404 }
405 Instruction *apply(BinaryOperator &Add) const {
406 return BinaryOperator::create(Instruction::Or, Add.getOperand(0),
407 Add.getOperand(1));
408 }
409};
410
411
412
Chris Lattner113f4f42002-06-25 16:13:24 +0000413Instruction *InstCombiner::visitAdd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000414 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000415 Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000416
Chris Lattnerb8b97502003-08-13 19:01:45 +0000417 // X + 0 --> X
Chris Lattnere6794492002-08-12 21:17:25 +0000418 if (RHS == Constant::getNullValue(I.getType()))
419 return ReplaceInstUsesWith(I, LHS);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000420
Chris Lattnerb8b97502003-08-13 19:01:45 +0000421 // X + X --> X << 1
422 if (I.getType()->isInteger())
423 if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result;
Chris Lattnerede3fe02003-08-13 04:18:28 +0000424
Chris Lattner147e9752002-05-08 22:46:53 +0000425 // -A + B --> B - A
Chris Lattnerbb74e222003-03-10 23:06:50 +0000426 if (Value *V = dyn_castNegVal(LHS))
Chris Lattner147e9752002-05-08 22:46:53 +0000427 return BinaryOperator::create(Instruction::Sub, RHS, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000428
429 // A + -B --> A - B
Chris Lattnerbb74e222003-03-10 23:06:50 +0000430 if (!isa<Constant>(RHS))
431 if (Value *V = dyn_castNegVal(RHS))
432 return BinaryOperator::create(Instruction::Sub, LHS, V);
Chris Lattner260ab202002-04-18 17:39:14 +0000433
Chris Lattner57c8d992003-02-18 19:57:07 +0000434 // X*C + X --> X * (C+1)
435 if (dyn_castFoldableMul(LHS) == RHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000436 Constant *CP1 =
437 ConstantExpr::get(Instruction::Add,
438 cast<Constant>(cast<Instruction>(LHS)->getOperand(1)),
439 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000440 return BinaryOperator::create(Instruction::Mul, RHS, CP1);
441 }
442
443 // X + X*C --> X * (C+1)
444 if (dyn_castFoldableMul(RHS) == LHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000445 Constant *CP1 =
446 ConstantExpr::get(Instruction::Add,
447 cast<Constant>(cast<Instruction>(RHS)->getOperand(1)),
448 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000449 return BinaryOperator::create(Instruction::Mul, LHS, CP1);
450 }
451
Chris Lattnerb8b97502003-08-13 19:01:45 +0000452 // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0
453 if (Constant *C2 = dyn_castMaskingAnd(RHS))
454 if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) return R;
Chris Lattner7fb29e12003-03-11 00:12:48 +0000455
Chris Lattnerb9cde762003-10-02 15:11:26 +0000456 if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) {
457 if (Instruction *ILHS = dyn_cast<Instruction>(LHS)) {
458 switch (ILHS->getOpcode()) {
459 case Instruction::Xor:
460 // ~X + C --> (C-1) - X
461 if (ConstantInt *XorRHS = dyn_cast<ConstantInt>(ILHS->getOperand(1)))
462 if (XorRHS->isAllOnesValue())
463 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000464 ConstantExpr::get(Instruction::Sub,
465 CRHS, ConstantInt::get(I.getType(), 1)),
Chris Lattnerb9cde762003-10-02 15:11:26 +0000466 ILHS->getOperand(0));
467 break;
468 default: break;
469 }
470 }
471 }
472
Chris Lattner113f4f42002-06-25 16:13:24 +0000473 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000474}
475
Chris Lattnerbdb0ce02003-07-22 21:46:59 +0000476// isSignBit - Return true if the value represented by the constant only has the
477// highest order bit set.
478static bool isSignBit(ConstantInt *CI) {
479 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
480 return (CI->getRawValue() & ~(-1LL << NumBits)) == (1ULL << (NumBits-1));
481}
482
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000483static unsigned getTypeSizeInBits(const Type *Ty) {
484 return Ty == Type::BoolTy ? 1 : Ty->getPrimitiveSize()*8;
485}
486
Chris Lattner113f4f42002-06-25 16:13:24 +0000487Instruction *InstCombiner::visitSub(BinaryOperator &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000488 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000489
Chris Lattnere6794492002-08-12 21:17:25 +0000490 if (Op0 == Op1) // sub X, X -> 0
491 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner260ab202002-04-18 17:39:14 +0000492
Chris Lattnere6794492002-08-12 21:17:25 +0000493 // If this is a 'B = x-(-A)', change to B = x+A...
Chris Lattnerbb74e222003-03-10 23:06:50 +0000494 if (Value *V = dyn_castNegVal(Op1))
Chris Lattner147e9752002-05-08 22:46:53 +0000495 return BinaryOperator::create(Instruction::Add, Op0, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000496
Chris Lattner8f2f5982003-11-05 01:06:05 +0000497 if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) {
498 // Replace (-1 - A) with (~A)...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000499 if (C->isAllOnesValue())
500 return BinaryOperator::createNot(Op1);
Chris Lattnerad3c4952002-05-09 01:29:19 +0000501
Chris Lattner8f2f5982003-11-05 01:06:05 +0000502 // C - ~X == X + (1+C)
503 if (BinaryOperator::isNot(Op1))
504 return BinaryOperator::create(Instruction::Add,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000505 BinaryOperator::getNotArgument(cast<BinaryOperator>(Op1)),
506 ConstantExpr::get(Instruction::Add, C,
507 ConstantInt::get(I.getType(), 1)));
Chris Lattner8f2f5982003-11-05 01:06:05 +0000508 }
509
Chris Lattner3082c5a2003-02-18 19:28:33 +0000510 if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1))
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000511 if (Op1I->hasOneUse()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000512 // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression
513 // is not used by anyone else...
514 //
Chris Lattnerc2f0aa52004-02-02 20:09:56 +0000515 if (Op1I->getOpcode() == Instruction::Sub &&
516 !Op1I->getType()->isFloatingPoint()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000517 // Swap the two operands of the subexpr...
518 Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1);
519 Op1I->setOperand(0, IIOp1);
520 Op1I->setOperand(1, IIOp0);
521
522 // Create the new top level add instruction...
523 return BinaryOperator::create(Instruction::Add, Op0, Op1);
524 }
525
526 // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)...
527 //
528 if (Op1I->getOpcode() == Instruction::And &&
529 (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) {
530 Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0);
531
532 Instruction *NewNot = BinaryOperator::createNot(OtherOp, "B.not", &I);
533 return BinaryOperator::create(Instruction::And, Op0, NewNot);
534 }
Chris Lattner57c8d992003-02-18 19:57:07 +0000535
536 // X - X*C --> X * (1-C)
537 if (dyn_castFoldableMul(Op1I) == Op0) {
Chris Lattner34428442003-05-27 16:40:51 +0000538 Constant *CP1 =
539 ConstantExpr::get(Instruction::Sub,
540 ConstantInt::get(I.getType(), 1),
541 cast<Constant>(cast<Instruction>(Op1)->getOperand(1)));
Chris Lattner57c8d992003-02-18 19:57:07 +0000542 assert(CP1 && "Couldn't constant fold 1-C?");
543 return BinaryOperator::create(Instruction::Mul, Op0, CP1);
544 }
Chris Lattnerad3c4952002-05-09 01:29:19 +0000545 }
Chris Lattner3082c5a2003-02-18 19:28:33 +0000546
Chris Lattner57c8d992003-02-18 19:57:07 +0000547 // X*C - X --> X * (C-1)
548 if (dyn_castFoldableMul(Op0) == Op1) {
Chris Lattner34428442003-05-27 16:40:51 +0000549 Constant *CP1 =
550 ConstantExpr::get(Instruction::Sub,
551 cast<Constant>(cast<Instruction>(Op0)->getOperand(1)),
552 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000553 assert(CP1 && "Couldn't constant fold C - 1?");
554 return BinaryOperator::create(Instruction::Mul, Op1, CP1);
555 }
556
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000557 return 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000558}
559
Chris Lattner113f4f42002-06-25 16:13:24 +0000560Instruction *InstCombiner::visitMul(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000561 bool Changed = SimplifyCommutative(I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000562 Value *Op0 = I.getOperand(0);
Chris Lattner260ab202002-04-18 17:39:14 +0000563
Chris Lattnere6794492002-08-12 21:17:25 +0000564 // Simplify mul instructions with a constant RHS...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000565 if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) {
566 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerede3fe02003-08-13 04:18:28 +0000567
568 // ((X << C1)*C2) == (X * (C2 << C1))
569 if (ShiftInst *SI = dyn_cast<ShiftInst>(Op0))
570 if (SI->getOpcode() == Instruction::Shl)
571 if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1)))
572 return BinaryOperator::create(Instruction::Mul, SI->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000573 ConstantExpr::get(Instruction::Shl, CI, ShOp));
574
Chris Lattnercce81be2003-09-11 22:24:54 +0000575 if (CI->isNullValue())
576 return ReplaceInstUsesWith(I, Op1); // X * 0 == 0
577 if (CI->equalsInt(1)) // X * 1 == X
578 return ReplaceInstUsesWith(I, Op0);
579 if (CI->isAllOnesValue()) // X * -1 == 0 - X
Chris Lattner35236d82003-06-25 17:09:20 +0000580 return BinaryOperator::createNeg(Op0, I.getName());
Chris Lattner31ba1292002-04-29 22:24:47 +0000581
Chris Lattnercce81be2003-09-11 22:24:54 +0000582 int64_t Val = (int64_t)cast<ConstantInt>(CI)->getRawValue();
Chris Lattner3082c5a2003-02-18 19:28:33 +0000583 if (uint64_t C = Log2(Val)) // Replace X*(2^C) with X << C
584 return new ShiftInst(Instruction::Shl, Op0,
585 ConstantUInt::get(Type::UByteTy, C));
586 } else {
587 ConstantFP *Op1F = cast<ConstantFP>(Op1);
588 if (Op1F->isNullValue())
589 return ReplaceInstUsesWith(I, Op1);
Chris Lattner31ba1292002-04-29 22:24:47 +0000590
Chris Lattner3082c5a2003-02-18 19:28:33 +0000591 // "In IEEE floating point, x*1 is not equivalent to x for nans. However,
592 // ANSI says we can drop signals, so we can do this anyway." (from GCC)
593 if (Op1F->getValue() == 1.0)
594 return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0'
595 }
Chris Lattner260ab202002-04-18 17:39:14 +0000596 }
597
Chris Lattner934a64cf2003-03-10 23:23:04 +0000598 if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y
599 if (Value *Op1v = dyn_castNegVal(I.getOperand(1)))
600 return BinaryOperator::create(Instruction::Mul, Op0v, Op1v);
601
Chris Lattner2635b522004-02-23 05:39:21 +0000602 // If one of the operands of the multiply is a cast from a boolean value, then
603 // we know the bool is either zero or one, so this is a 'masking' multiply.
604 // See if we can simplify things based on how the boolean was originally
605 // formed.
606 CastInst *BoolCast = 0;
607 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(0)))
608 if (CI->getOperand(0)->getType() == Type::BoolTy)
609 BoolCast = CI;
610 if (!BoolCast)
611 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(1)))
612 if (CI->getOperand(0)->getType() == Type::BoolTy)
613 BoolCast = CI;
614 if (BoolCast) {
615 if (SetCondInst *SCI = dyn_cast<SetCondInst>(BoolCast->getOperand(0))) {
616 Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1);
617 const Type *SCOpTy = SCIOp0->getType();
618
Chris Lattnerc8a10c42004-02-23 06:00:11 +0000619 // If the source is X < 0 or X <= -1, and X is a signed integer type,
620 // convert this multiply into a shift/and combination.
621 if (SCOpTy->isSigned() && isa<ConstantInt>(SCIOp1) &&
622 ((SCI->getOpcode() == Instruction::SetLT &&
623 cast<Constant>(SCIOp1)->isNullValue()) ||
624 (SCI->getOpcode() == Instruction::SetLE &&
625 cast<ConstantInt>(SCIOp1)->isAllOnesValue()))) {
Chris Lattner2635b522004-02-23 05:39:21 +0000626 // Shift the X value right to turn it into "all signbits".
627 Constant *Amt = ConstantUInt::get(Type::UByteTy,
628 SCOpTy->getPrimitiveSize()*8-1);
629 Value *V = new ShiftInst(Instruction::Shr, SCIOp0, Amt,
630 BoolCast->getName()+".mask", &I);
631
632 // If the multiply type is not the same as the source type, sign extend
633 // or truncate to the multiply type.
634 if (I.getType() != V->getType())
635 V = new CastInst(V, I.getType(), V->getName(), &I);
636
637 Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0;
638 return BinaryOperator::create(Instruction::And, V, OtherOp);
639 }
640 }
641 }
642
Chris Lattner113f4f42002-06-25 16:13:24 +0000643 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000644}
645
Chris Lattner113f4f42002-06-25 16:13:24 +0000646Instruction *InstCombiner::visitDiv(BinaryOperator &I) {
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000647 // div X, 1 == X
Chris Lattner3082c5a2003-02-18 19:28:33 +0000648 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
Chris Lattnere6794492002-08-12 21:17:25 +0000649 if (RHS->equalsInt(1))
650 return ReplaceInstUsesWith(I, I.getOperand(0));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000651
652 // Check to see if this is an unsigned division with an exact power of 2,
653 // if so, convert to a right shift.
654 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
655 if (uint64_t Val = C->getValue()) // Don't break X / 0
656 if (uint64_t C = Log2(Val))
657 return new ShiftInst(Instruction::Shr, I.getOperand(0),
658 ConstantUInt::get(Type::UByteTy, C));
659 }
660
661 // 0 / X == 0, we don't need to preserve faults!
662 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
663 if (LHS->equalsInt(0))
664 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
665
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000666 return 0;
667}
668
669
Chris Lattner113f4f42002-06-25 16:13:24 +0000670Instruction *InstCombiner::visitRem(BinaryOperator &I) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000671 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
672 if (RHS->equalsInt(1)) // X % 1 == 0
673 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
674
675 // Check to see if this is an unsigned remainder with an exact power of 2,
676 // if so, convert to a bitwise and.
677 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
678 if (uint64_t Val = C->getValue()) // Don't break X % 0 (divide by zero)
679 if (Log2(Val))
680 return BinaryOperator::create(Instruction::And, I.getOperand(0),
681 ConstantUInt::get(I.getType(), Val-1));
682 }
683
684 // 0 % X == 0, we don't need to preserve faults!
685 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
686 if (LHS->equalsInt(0))
Chris Lattnere6794492002-08-12 21:17:25 +0000687 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
688
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000689 return 0;
690}
691
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000692// isMaxValueMinusOne - return true if this is Max-1
Chris Lattnere6794492002-08-12 21:17:25 +0000693static bool isMaxValueMinusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000694 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C)) {
695 // Calculate -1 casted to the right type...
696 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
697 uint64_t Val = ~0ULL; // All ones
698 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
699 return CU->getValue() == Val-1;
700 }
701
702 const ConstantSInt *CS = cast<ConstantSInt>(C);
703
704 // Calculate 0111111111..11111
705 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
706 int64_t Val = INT64_MAX; // All ones
707 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
708 return CS->getValue() == Val-1;
709}
710
711// isMinValuePlusOne - return true if this is Min+1
Chris Lattnere6794492002-08-12 21:17:25 +0000712static bool isMinValuePlusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000713 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C))
714 return CU->getValue() == 1;
715
716 const ConstantSInt *CS = cast<ConstantSInt>(C);
717
718 // Calculate 1111111111000000000000
719 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
720 int64_t Val = -1; // All ones
721 Val <<= TypeBits-1; // Shift over to the right spot
722 return CS->getValue() == Val+1;
723}
724
Chris Lattner3ac7c262003-08-13 20:16:26 +0000725/// getSetCondCode - Encode a setcc opcode into a three bit mask. These bits
726/// are carefully arranged to allow folding of expressions such as:
727///
728/// (A < B) | (A > B) --> (A != B)
729///
730/// Bit value '4' represents that the comparison is true if A > B, bit value '2'
731/// represents that the comparison is true if A == B, and bit value '1' is true
732/// if A < B.
733///
734static unsigned getSetCondCode(const SetCondInst *SCI) {
735 switch (SCI->getOpcode()) {
736 // False -> 0
737 case Instruction::SetGT: return 1;
738 case Instruction::SetEQ: return 2;
739 case Instruction::SetGE: return 3;
740 case Instruction::SetLT: return 4;
741 case Instruction::SetNE: return 5;
742 case Instruction::SetLE: return 6;
743 // True -> 7
744 default:
745 assert(0 && "Invalid SetCC opcode!");
746 return 0;
747 }
748}
749
750/// getSetCCValue - This is the complement of getSetCondCode, which turns an
751/// opcode and two operands into either a constant true or false, or a brand new
752/// SetCC instruction.
753static Value *getSetCCValue(unsigned Opcode, Value *LHS, Value *RHS) {
754 switch (Opcode) {
755 case 0: return ConstantBool::False;
756 case 1: return new SetCondInst(Instruction::SetGT, LHS, RHS);
757 case 2: return new SetCondInst(Instruction::SetEQ, LHS, RHS);
758 case 3: return new SetCondInst(Instruction::SetGE, LHS, RHS);
759 case 4: return new SetCondInst(Instruction::SetLT, LHS, RHS);
760 case 5: return new SetCondInst(Instruction::SetNE, LHS, RHS);
761 case 6: return new SetCondInst(Instruction::SetLE, LHS, RHS);
762 case 7: return ConstantBool::True;
763 default: assert(0 && "Illegal SetCCCode!"); return 0;
764 }
765}
766
767// FoldSetCCLogical - Implements (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
768struct FoldSetCCLogical {
769 InstCombiner &IC;
770 Value *LHS, *RHS;
771 FoldSetCCLogical(InstCombiner &ic, SetCondInst *SCI)
772 : IC(ic), LHS(SCI->getOperand(0)), RHS(SCI->getOperand(1)) {}
773 bool shouldApply(Value *V) const {
774 if (SetCondInst *SCI = dyn_cast<SetCondInst>(V))
775 return (SCI->getOperand(0) == LHS && SCI->getOperand(1) == RHS ||
776 SCI->getOperand(0) == RHS && SCI->getOperand(1) == LHS);
777 return false;
778 }
779 Instruction *apply(BinaryOperator &Log) const {
780 SetCondInst *SCI = cast<SetCondInst>(Log.getOperand(0));
781 if (SCI->getOperand(0) != LHS) {
782 assert(SCI->getOperand(1) == LHS);
783 SCI->swapOperands(); // Swap the LHS and RHS of the SetCC
784 }
785
786 unsigned LHSCode = getSetCondCode(SCI);
787 unsigned RHSCode = getSetCondCode(cast<SetCondInst>(Log.getOperand(1)));
788 unsigned Code;
789 switch (Log.getOpcode()) {
790 case Instruction::And: Code = LHSCode & RHSCode; break;
791 case Instruction::Or: Code = LHSCode | RHSCode; break;
792 case Instruction::Xor: Code = LHSCode ^ RHSCode; break;
Chris Lattner2caaaba2003-09-22 20:33:34 +0000793 default: assert(0 && "Illegal logical opcode!"); return 0;
Chris Lattner3ac7c262003-08-13 20:16:26 +0000794 }
795
796 Value *RV = getSetCCValue(Code, LHS, RHS);
797 if (Instruction *I = dyn_cast<Instruction>(RV))
798 return I;
799 // Otherwise, it's a constant boolean value...
800 return IC.ReplaceInstUsesWith(Log, RV);
801 }
802};
803
804
Chris Lattnerba1cb382003-09-19 17:17:26 +0000805// OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where
806// the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is
807// guaranteed to be either a shift instruction or a binary operator.
808Instruction *InstCombiner::OptAndOp(Instruction *Op,
809 ConstantIntegral *OpRHS,
810 ConstantIntegral *AndRHS,
811 BinaryOperator &TheAnd) {
812 Value *X = Op->getOperand(0);
Chris Lattnerfcf21a72004-01-12 19:47:05 +0000813 Constant *Together = 0;
814 if (!isa<ShiftInst>(Op))
815 Together = ConstantExpr::get(Instruction::And, AndRHS, OpRHS);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000816
Chris Lattnerba1cb382003-09-19 17:17:26 +0000817 switch (Op->getOpcode()) {
818 case Instruction::Xor:
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000819 if (Together->isNullValue()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +0000820 // (X ^ C1) & C2 --> (X & C2) iff (C1&C2) == 0
821 return BinaryOperator::create(Instruction::And, X, AndRHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000822 } else if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +0000823 // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2)
824 std::string OpName = Op->getName(); Op->setName("");
825 Instruction *And = BinaryOperator::create(Instruction::And,
826 X, AndRHS, OpName);
827 InsertNewInstBefore(And, TheAnd);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000828 return BinaryOperator::create(Instruction::Xor, And, Together);
Chris Lattnerba1cb382003-09-19 17:17:26 +0000829 }
830 break;
831 case Instruction::Or:
832 // (X | C1) & C2 --> X & C2 iff C1 & C1 == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000833 if (Together->isNullValue())
Chris Lattnerba1cb382003-09-19 17:17:26 +0000834 return BinaryOperator::create(Instruction::And, X, AndRHS);
835 else {
Chris Lattnerba1cb382003-09-19 17:17:26 +0000836 if (Together == AndRHS) // (X | C) & C --> C
837 return ReplaceInstUsesWith(TheAnd, AndRHS);
838
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000839 if (Op->hasOneUse() && Together != OpRHS) {
Chris Lattnerba1cb382003-09-19 17:17:26 +0000840 // (X | C1) & C2 --> (X | (C1&C2)) & C2
841 std::string Op0Name = Op->getName(); Op->setName("");
842 Instruction *Or = BinaryOperator::create(Instruction::Or, X,
843 Together, Op0Name);
844 InsertNewInstBefore(Or, TheAnd);
845 return BinaryOperator::create(Instruction::And, Or, AndRHS);
846 }
847 }
848 break;
849 case Instruction::Add:
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000850 if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +0000851 // Adding a one to a single bit bit-field should be turned into an XOR
852 // of the bit. First thing to check is to see if this AND is with a
853 // single bit constant.
854 unsigned long long AndRHSV = cast<ConstantInt>(AndRHS)->getRawValue();
855
856 // Clear bits that are not part of the constant.
857 AndRHSV &= (1ULL << AndRHS->getType()->getPrimitiveSize()*8)-1;
858
859 // If there is only one bit set...
860 if ((AndRHSV & (AndRHSV-1)) == 0) {
861 // Ok, at this point, we know that we are masking the result of the
862 // ADD down to exactly one bit. If the constant we are adding has
863 // no bits set below this bit, then we can eliminate the ADD.
864 unsigned long long AddRHS = cast<ConstantInt>(OpRHS)->getRawValue();
865
866 // Check to see if any bits below the one bit set in AndRHSV are set.
867 if ((AddRHS & (AndRHSV-1)) == 0) {
868 // If not, the only thing that can effect the output of the AND is
869 // the bit specified by AndRHSV. If that bit is set, the effect of
870 // the XOR is to toggle the bit. If it is clear, then the ADD has
871 // no effect.
872 if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop
873 TheAnd.setOperand(0, X);
874 return &TheAnd;
875 } else {
876 std::string Name = Op->getName(); Op->setName("");
877 // Pull the XOR out of the AND.
878 Instruction *NewAnd =
879 BinaryOperator::create(Instruction::And, X, AndRHS, Name);
880 InsertNewInstBefore(NewAnd, TheAnd);
881 return BinaryOperator::create(Instruction::Xor, NewAnd, AndRHS);
882 }
883 }
884 }
885 }
886 break;
Chris Lattner2da29172003-09-19 19:05:02 +0000887
888 case Instruction::Shl: {
889 // We know that the AND will not produce any of the bits shifted in, so if
890 // the anded constant includes them, clear them now!
891 //
892 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000893 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
894 ConstantExpr::get(Instruction::Shl, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +0000895 if (CI != AndRHS) {
896 TheAnd.setOperand(1, CI);
897 return &TheAnd;
898 }
899 break;
900 }
901 case Instruction::Shr:
902 // We know that the AND will not produce any of the bits shifted in, so if
903 // the anded constant includes them, clear them now! This only applies to
904 // unsigned shifts, because a signed shr may bring in set bits!
905 //
906 if (AndRHS->getType()->isUnsigned()) {
907 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000908 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
909 ConstantExpr::get(Instruction::Shr, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +0000910 if (CI != AndRHS) {
911 TheAnd.setOperand(1, CI);
912 return &TheAnd;
913 }
914 }
915 break;
Chris Lattnerba1cb382003-09-19 17:17:26 +0000916 }
917 return 0;
918}
919
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000920
Chris Lattner113f4f42002-06-25 16:13:24 +0000921Instruction *InstCombiner::visitAnd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000922 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000923 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000924
925 // and X, X = X and X, 0 == 0
Chris Lattnere6794492002-08-12 21:17:25 +0000926 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
927 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000928
929 // and X, -1 == X
Chris Lattner49b47ae2003-07-23 17:57:01 +0000930 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +0000931 if (RHS->isAllOnesValue())
932 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000933
Chris Lattnerba1cb382003-09-19 17:17:26 +0000934 // Optimize a variety of ((val OP C1) & C2) combinations...
935 if (isa<BinaryOperator>(Op0) || isa<ShiftInst>(Op0)) {
936 Instruction *Op0I = cast<Instruction>(Op0);
Chris Lattner33217db2003-07-23 19:36:21 +0000937 Value *X = Op0I->getOperand(0);
Chris Lattner16464b32003-07-23 19:25:52 +0000938 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnerba1cb382003-09-19 17:17:26 +0000939 if (Instruction *Res = OptAndOp(Op0I, Op0CI, RHS, I))
940 return Res;
Chris Lattner33217db2003-07-23 19:36:21 +0000941 }
Chris Lattner49b47ae2003-07-23 17:57:01 +0000942 }
943
Chris Lattnerbb74e222003-03-10 23:06:50 +0000944 Value *Op0NotVal = dyn_castNotVal(Op0);
945 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000946
947 // (~A & ~B) == (~(A | B)) - Demorgan's Law
Chris Lattnerbb74e222003-03-10 23:06:50 +0000948 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000949 Instruction *Or = BinaryOperator::create(Instruction::Or, Op0NotVal,
Chris Lattner49b47ae2003-07-23 17:57:01 +0000950 Op1NotVal,I.getName()+".demorgan");
951 InsertNewInstBefore(Or, I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000952 return BinaryOperator::createNot(Or);
953 }
954
955 if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0
956 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner65217ff2002-08-23 18:32:43 +0000957
Chris Lattner3ac7c262003-08-13 20:16:26 +0000958 // (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
959 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
960 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
961 return R;
962
Chris Lattner113f4f42002-06-25 16:13:24 +0000963 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000964}
965
966
967
Chris Lattner113f4f42002-06-25 16:13:24 +0000968Instruction *InstCombiner::visitOr(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000969 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000970 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000971
972 // or X, X = X or X, 0 == X
Chris Lattnere6794492002-08-12 21:17:25 +0000973 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
974 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000975
976 // or X, -1 == -1
Chris Lattner8f0d1562003-07-23 18:29:44 +0000977 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +0000978 if (RHS->isAllOnesValue())
979 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000980
Chris Lattner8f0d1562003-07-23 18:29:44 +0000981 if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) {
982 // (X & C1) | C2 --> (X | C2) & (C1|C2)
983 if (Op0I->getOpcode() == Instruction::And && isOnlyUse(Op0))
984 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
985 std::string Op0Name = Op0I->getName(); Op0I->setName("");
986 Instruction *Or = BinaryOperator::create(Instruction::Or,
987 Op0I->getOperand(0), RHS,
988 Op0Name);
989 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000990 return BinaryOperator::create(Instruction::And, Or,
991 ConstantExpr::get(Instruction::Or, RHS, Op0CI));
Chris Lattner8f0d1562003-07-23 18:29:44 +0000992 }
993
994 // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2)
995 if (Op0I->getOpcode() == Instruction::Xor && isOnlyUse(Op0))
996 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
997 std::string Op0Name = Op0I->getName(); Op0I->setName("");
998 Instruction *Or = BinaryOperator::create(Instruction::Or,
999 Op0I->getOperand(0), RHS,
1000 Op0Name);
1001 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001002 return BinaryOperator::create(Instruction::Xor, Or,
1003 ConstantExpr::get(Instruction::And, Op0CI,
1004 NotConstant(RHS)));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001005 }
1006 }
1007 }
1008
Chris Lattner812aab72003-08-12 19:11:07 +00001009 // (A & C1)|(A & C2) == A & (C1|C2)
Chris Lattner01d56392003-08-12 19:17:27 +00001010 if (Instruction *LHS = dyn_cast<BinaryOperator>(Op0))
1011 if (Instruction *RHS = dyn_cast<BinaryOperator>(Op1))
1012 if (LHS->getOperand(0) == RHS->getOperand(0))
1013 if (Constant *C0 = dyn_castMaskingAnd(LHS))
1014 if (Constant *C1 = dyn_castMaskingAnd(RHS))
1015 return BinaryOperator::create(Instruction::And, LHS->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001016 ConstantExpr::get(Instruction::Or, C0, C1));
Chris Lattner812aab72003-08-12 19:11:07 +00001017
Chris Lattner3e327a42003-03-10 23:13:59 +00001018 Value *Op0NotVal = dyn_castNotVal(Op0);
1019 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001020
Chris Lattner3e327a42003-03-10 23:13:59 +00001021 if (Op1 == Op0NotVal) // ~A | A == -1
1022 return ReplaceInstUsesWith(I,
1023 ConstantIntegral::getAllOnesValue(I.getType()));
1024
1025 if (Op0 == Op1NotVal) // A | ~A == -1
1026 return ReplaceInstUsesWith(I,
1027 ConstantIntegral::getAllOnesValue(I.getType()));
1028
1029 // (~A | ~B) == (~(A & B)) - Demorgan's Law
1030 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
1031 Instruction *And = BinaryOperator::create(Instruction::And, Op0NotVal,
1032 Op1NotVal,I.getName()+".demorgan",
1033 &I);
1034 WorkList.push_back(And);
1035 return BinaryOperator::createNot(And);
1036 }
Chris Lattner3082c5a2003-02-18 19:28:33 +00001037
Chris Lattner3ac7c262003-08-13 20:16:26 +00001038 // (setcc1 A, B) | (setcc2 A, B) --> (setcc3 A, B)
1039 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1040 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1041 return R;
1042
Chris Lattner113f4f42002-06-25 16:13:24 +00001043 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001044}
1045
Chris Lattnerc2076352004-02-16 01:20:27 +00001046// XorSelf - Implements: X ^ X --> 0
1047struct XorSelf {
1048 Value *RHS;
1049 XorSelf(Value *rhs) : RHS(rhs) {}
1050 bool shouldApply(Value *LHS) const { return LHS == RHS; }
1051 Instruction *apply(BinaryOperator &Xor) const {
1052 return &Xor;
1053 }
1054};
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001055
1056
Chris Lattner113f4f42002-06-25 16:13:24 +00001057Instruction *InstCombiner::visitXor(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001058 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001059 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001060
Chris Lattnerc2076352004-02-16 01:20:27 +00001061 // xor X, X = 0, even if X is nested in a sequence of Xor's.
1062 if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) {
1063 assert(Result == &I && "AssociativeOpt didn't work?");
Chris Lattnere6794492002-08-12 21:17:25 +00001064 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattnerc2076352004-02-16 01:20:27 +00001065 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001066
Chris Lattner97638592003-07-23 21:37:07 +00001067 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001068 // xor X, 0 == X
Chris Lattner97638592003-07-23 21:37:07 +00001069 if (RHS->isNullValue())
Chris Lattnere6794492002-08-12 21:17:25 +00001070 return ReplaceInstUsesWith(I, Op0);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001071
Chris Lattner97638592003-07-23 21:37:07 +00001072 if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) {
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001073 // xor (setcc A, B), true = not (setcc A, B) = setncc A, B
Chris Lattner97638592003-07-23 21:37:07 +00001074 if (SetCondInst *SCI = dyn_cast<SetCondInst>(Op0I))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001075 if (RHS == ConstantBool::True && SCI->hasOneUse())
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001076 return new SetCondInst(SCI->getInverseCondition(),
1077 SCI->getOperand(0), SCI->getOperand(1));
Chris Lattnere5806662003-11-04 23:50:51 +00001078
Chris Lattner8f2f5982003-11-05 01:06:05 +00001079 // ~(c-X) == X-c-1 == X+(-c-1)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001080 if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue())
1081 if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) {
1082 Constant *NegOp0I0C = ConstantExpr::get(Instruction::Sub,
1083 Constant::getNullValue(Op0I0C->getType()), Op0I0C);
1084 Constant *ConstantRHS = ConstantExpr::get(Instruction::Sub, NegOp0I0C,
1085 ConstantInt::get(I.getType(), 1));
1086 return BinaryOperator::create(Instruction::Add, Op0I->getOperand(1),
1087 ConstantRHS);
1088 }
Chris Lattner97638592003-07-23 21:37:07 +00001089
1090 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnere5806662003-11-04 23:50:51 +00001091 switch (Op0I->getOpcode()) {
1092 case Instruction::Add:
Chris Lattner0f68fa62003-11-04 23:37:10 +00001093 // ~(X-c) --> (-c-1)-X
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001094 if (RHS->isAllOnesValue()) {
1095 Constant *NegOp0CI = ConstantExpr::get(Instruction::Sub,
1096 Constant::getNullValue(Op0CI->getType()), Op0CI);
Chris Lattner0f68fa62003-11-04 23:37:10 +00001097 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001098 ConstantExpr::get(Instruction::Sub, NegOp0CI,
1099 ConstantInt::get(I.getType(), 1)),
Chris Lattner0f68fa62003-11-04 23:37:10 +00001100 Op0I->getOperand(0));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001101 }
Chris Lattnere5806662003-11-04 23:50:51 +00001102 break;
1103 case Instruction::And:
Chris Lattner97638592003-07-23 21:37:07 +00001104 // (X & C1) ^ C2 --> (X & C1) | C2 iff (C1&C2) == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001105 if (ConstantExpr::get(Instruction::And, RHS, Op0CI)->isNullValue())
Chris Lattner97638592003-07-23 21:37:07 +00001106 return BinaryOperator::create(Instruction::Or, Op0, RHS);
Chris Lattnere5806662003-11-04 23:50:51 +00001107 break;
1108 case Instruction::Or:
Chris Lattner97638592003-07-23 21:37:07 +00001109 // (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001110 if (ConstantExpr::get(Instruction::And, RHS, Op0CI) == RHS)
1111 return BinaryOperator::create(Instruction::And, Op0,
1112 NotConstant(RHS));
Chris Lattnere5806662003-11-04 23:50:51 +00001113 break;
1114 default: break;
Chris Lattner97638592003-07-23 21:37:07 +00001115 }
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001116 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001117 }
1118
Chris Lattnerbb74e222003-03-10 23:06:50 +00001119 if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001120 if (X == Op1)
1121 return ReplaceInstUsesWith(I,
1122 ConstantIntegral::getAllOnesValue(I.getType()));
1123
Chris Lattnerbb74e222003-03-10 23:06:50 +00001124 if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001125 if (X == Op0)
1126 return ReplaceInstUsesWith(I,
1127 ConstantIntegral::getAllOnesValue(I.getType()));
1128
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001129 if (Instruction *Op1I = dyn_cast<Instruction>(Op1))
Chris Lattnerb36d9082004-02-16 03:54:20 +00001130 if (Op1I->getOpcode() == Instruction::Or) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001131 if (Op1I->getOperand(0) == Op0) { // B^(B|A) == (A|B)^B
1132 cast<BinaryOperator>(Op1I)->swapOperands();
1133 I.swapOperands();
1134 std::swap(Op0, Op1);
1135 } else if (Op1I->getOperand(1) == Op0) { // B^(A|B) == (A|B)^B
1136 I.swapOperands();
1137 std::swap(Op0, Op1);
Chris Lattnerb36d9082004-02-16 03:54:20 +00001138 }
1139 } else if (Op1I->getOpcode() == Instruction::Xor) {
1140 if (Op0 == Op1I->getOperand(0)) // A^(A^B) == B
1141 return ReplaceInstUsesWith(I, Op1I->getOperand(1));
1142 else if (Op0 == Op1I->getOperand(1)) // A^(B^A) == B
1143 return ReplaceInstUsesWith(I, Op1I->getOperand(0));
1144 }
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001145
1146 if (Instruction *Op0I = dyn_cast<Instruction>(Op0))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001147 if (Op0I->getOpcode() == Instruction::Or && Op0I->hasOneUse()) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001148 if (Op0I->getOperand(0) == Op1) // (B|A)^B == (A|B)^B
1149 cast<BinaryOperator>(Op0I)->swapOperands();
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001150 if (Op0I->getOperand(1) == Op1) { // (A|B)^B == A & ~B
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001151 Value *NotB = BinaryOperator::createNot(Op1, Op1->getName()+".not", &I);
1152 WorkList.push_back(cast<Instruction>(NotB));
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001153 return BinaryOperator::create(Instruction::And, Op0I->getOperand(0),
1154 NotB);
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001155 }
Chris Lattnerb36d9082004-02-16 03:54:20 +00001156 } else if (Op0I->getOpcode() == Instruction::Xor) {
1157 if (Op1 == Op0I->getOperand(0)) // (A^B)^A == B
1158 return ReplaceInstUsesWith(I, Op0I->getOperand(1));
1159 else if (Op1 == Op0I->getOperand(1)) // (B^A)^A == B
1160 return ReplaceInstUsesWith(I, Op0I->getOperand(0));
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001161 }
1162
Chris Lattner7fb29e12003-03-11 00:12:48 +00001163 // (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1^C2 == 0
1164 if (Constant *C1 = dyn_castMaskingAnd(Op0))
1165 if (Constant *C2 = dyn_castMaskingAnd(Op1))
Chris Lattner34428442003-05-27 16:40:51 +00001166 if (ConstantExpr::get(Instruction::And, C1, C2)->isNullValue())
Chris Lattner7fb29e12003-03-11 00:12:48 +00001167 return BinaryOperator::create(Instruction::Or, Op0, Op1);
1168
Chris Lattner3ac7c262003-08-13 20:16:26 +00001169 // (setcc1 A, B) ^ (setcc2 A, B) --> (setcc3 A, B)
1170 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1171 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1172 return R;
1173
Chris Lattner113f4f42002-06-25 16:13:24 +00001174 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001175}
1176
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001177// AddOne, SubOne - Add or subtract a constant one from an integer constant...
1178static Constant *AddOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001179 Constant *Result = ConstantExpr::get(Instruction::Add, C,
1180 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001181 assert(Result && "Constant folding integer addition failed!");
1182 return Result;
1183}
1184static Constant *SubOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001185 Constant *Result = ConstantExpr::get(Instruction::Sub, C,
1186 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001187 assert(Result && "Constant folding integer addition failed!");
1188 return Result;
1189}
1190
Chris Lattner1fc23f32002-05-09 20:11:54 +00001191// isTrueWhenEqual - Return true if the specified setcondinst instruction is
1192// true when both operands are equal...
1193//
Chris Lattner113f4f42002-06-25 16:13:24 +00001194static bool isTrueWhenEqual(Instruction &I) {
1195 return I.getOpcode() == Instruction::SetEQ ||
1196 I.getOpcode() == Instruction::SetGE ||
1197 I.getOpcode() == Instruction::SetLE;
Chris Lattner1fc23f32002-05-09 20:11:54 +00001198}
1199
Chris Lattner113f4f42002-06-25 16:13:24 +00001200Instruction *InstCombiner::visitSetCondInst(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001201 bool Changed = SimplifyCommutative(I);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001202 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
1203 const Type *Ty = Op0->getType();
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001204
1205 // setcc X, X
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001206 if (Op0 == Op1)
1207 return ReplaceInstUsesWith(I, ConstantBool::get(isTrueWhenEqual(I)));
Chris Lattner1fc23f32002-05-09 20:11:54 +00001208
Chris Lattnerd07283a2003-08-13 05:38:46 +00001209 // setcc <global/alloca*>, 0 - Global/Stack value addresses are never null!
1210 if (isa<ConstantPointerNull>(Op1) &&
1211 (isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0)))
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001212 return ReplaceInstUsesWith(I, ConstantBool::get(!isTrueWhenEqual(I)));
1213
Chris Lattnerd07283a2003-08-13 05:38:46 +00001214
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001215 // setcc's with boolean values can always be turned into bitwise operations
1216 if (Ty == Type::BoolTy) {
1217 // If this is <, >, or !=, we can change this into a simple xor instruction
1218 if (!isTrueWhenEqual(I))
Chris Lattner16930792003-11-03 04:25:02 +00001219 return BinaryOperator::create(Instruction::Xor, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001220
1221 // Otherwise we need to make a temporary intermediate instruction and insert
1222 // it into the instruction stream. This is what we are after:
1223 //
1224 // seteq bool %A, %B -> ~(A^B)
1225 // setle bool %A, %B -> ~A | B
1226 // setge bool %A, %B -> A | ~B
1227 //
1228 if (I.getOpcode() == Instruction::SetEQ) { // seteq case
1229 Instruction *Xor = BinaryOperator::create(Instruction::Xor, Op0, Op1,
1230 I.getName()+"tmp");
1231 InsertNewInstBefore(Xor, I);
Chris Lattner16930792003-11-03 04:25:02 +00001232 return BinaryOperator::createNot(Xor);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001233 }
1234
1235 // Handle the setXe cases...
1236 assert(I.getOpcode() == Instruction::SetGE ||
1237 I.getOpcode() == Instruction::SetLE);
1238
1239 if (I.getOpcode() == Instruction::SetGE)
1240 std::swap(Op0, Op1); // Change setge -> setle
1241
1242 // Now we just have the SetLE case.
Chris Lattner31ae8632002-08-14 17:51:49 +00001243 Instruction *Not = BinaryOperator::createNot(Op0, I.getName()+"tmp");
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001244 InsertNewInstBefore(Not, I);
Chris Lattner16930792003-11-03 04:25:02 +00001245 return BinaryOperator::create(Instruction::Or, Not, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001246 }
1247
1248 // Check to see if we are doing one of many comparisons against constant
1249 // integers at the end of their ranges...
1250 //
1251 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001252 // Simplify seteq and setne instructions...
1253 if (I.getOpcode() == Instruction::SetEQ ||
1254 I.getOpcode() == Instruction::SetNE) {
1255 bool isSetNE = I.getOpcode() == Instruction::SetNE;
1256
Chris Lattnercfbce7c2003-07-23 17:26:36 +00001257 // If the first operand is (and|or|xor) with a constant, and the second
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001258 // operand is a constant, simplify a bit.
Chris Lattnerc992add2003-08-13 05:33:12 +00001259 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) {
1260 switch (BO->getOpcode()) {
1261 case Instruction::Add:
1262 if (CI->isNullValue()) {
1263 // Replace ((add A, B) != 0) with (A != -B) if A or B is
1264 // efficiently invertible, or if the add has just this one use.
1265 Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1);
1266 if (Value *NegVal = dyn_castNegVal(BOp1))
1267 return new SetCondInst(I.getOpcode(), BOp0, NegVal);
1268 else if (Value *NegVal = dyn_castNegVal(BOp0))
1269 return new SetCondInst(I.getOpcode(), NegVal, BOp1);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001270 else if (BO->hasOneUse()) {
Chris Lattnerc992add2003-08-13 05:33:12 +00001271 Instruction *Neg = BinaryOperator::createNeg(BOp1, BO->getName());
1272 BO->setName("");
1273 InsertNewInstBefore(Neg, I);
1274 return new SetCondInst(I.getOpcode(), BOp0, Neg);
1275 }
1276 }
1277 break;
1278 case Instruction::Xor:
1279 // For the xor case, we can xor two constants together, eliminating
1280 // the explicit xor.
1281 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1)))
1282 return BinaryOperator::create(I.getOpcode(), BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001283 ConstantExpr::get(Instruction::Xor, CI, BOC));
Chris Lattnerc992add2003-08-13 05:33:12 +00001284
1285 // FALLTHROUGH
1286 case Instruction::Sub:
1287 // Replace (([sub|xor] A, B) != 0) with (A != B)
1288 if (CI->isNullValue())
1289 return new SetCondInst(I.getOpcode(), BO->getOperand(0),
1290 BO->getOperand(1));
1291 break;
1292
1293 case Instruction::Or:
1294 // If bits are being or'd in that are not present in the constant we
1295 // are comparing against, then the comparison could never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001296 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) {
1297 Constant *NotCI = NotConstant(CI);
1298 if (!ConstantExpr::get(Instruction::And, BOC, NotCI)->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001299 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001300 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001301 break;
1302
1303 case Instruction::And:
1304 if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001305 // If bits are being compared against that are and'd out, then the
1306 // comparison can never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001307 if (!ConstantExpr::get(Instruction::And, CI,
1308 NotConstant(BOC))->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001309 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc992add2003-08-13 05:33:12 +00001310
1311 // Replace (and X, (1 << size(X)-1) != 0) with x < 0, converting X
1312 // to be a signed value as appropriate.
1313 if (isSignBit(BOC)) {
1314 Value *X = BO->getOperand(0);
1315 // If 'X' is not signed, insert a cast now...
1316 if (!BOC->getType()->isSigned()) {
1317 const Type *DestTy;
1318 switch (BOC->getType()->getPrimitiveID()) {
1319 case Type::UByteTyID: DestTy = Type::SByteTy; break;
1320 case Type::UShortTyID: DestTy = Type::ShortTy; break;
1321 case Type::UIntTyID: DestTy = Type::IntTy; break;
1322 case Type::ULongTyID: DestTy = Type::LongTy; break;
1323 default: assert(0 && "Invalid unsigned integer type!"); abort();
1324 }
1325 CastInst *NewCI = new CastInst(X,DestTy,X->getName()+".signed");
1326 InsertNewInstBefore(NewCI, I);
1327 X = NewCI;
1328 }
1329 return new SetCondInst(isSetNE ? Instruction::SetLT :
1330 Instruction::SetGE, X,
1331 Constant::getNullValue(X->getType()));
1332 }
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001333 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001334 default: break;
1335 }
1336 }
Chris Lattnere967b342003-06-04 05:10:11 +00001337 }
Chris Lattner791ac1a2003-06-01 03:35:25 +00001338
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001339 // Check to see if we are comparing against the minimum or maximum value...
Chris Lattnere6794492002-08-12 21:17:25 +00001340 if (CI->isMinValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001341 if (I.getOpcode() == Instruction::SetLT) // A < MIN -> FALSE
1342 return ReplaceInstUsesWith(I, ConstantBool::False);
1343 if (I.getOpcode() == Instruction::SetGE) // A >= MIN -> TRUE
1344 return ReplaceInstUsesWith(I, ConstantBool::True);
1345 if (I.getOpcode() == Instruction::SetLE) // A <= MIN -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001346 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001347 if (I.getOpcode() == Instruction::SetGT) // A > MIN -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001348 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001349
Chris Lattnere6794492002-08-12 21:17:25 +00001350 } else if (CI->isMaxValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001351 if (I.getOpcode() == Instruction::SetGT) // A > MAX -> FALSE
1352 return ReplaceInstUsesWith(I, ConstantBool::False);
1353 if (I.getOpcode() == Instruction::SetLE) // A <= MAX -> TRUE
1354 return ReplaceInstUsesWith(I, ConstantBool::True);
1355 if (I.getOpcode() == Instruction::SetGE) // A >= MAX -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001356 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001357 if (I.getOpcode() == Instruction::SetLT) // A < MAX -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001358 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001359
1360 // Comparing against a value really close to min or max?
1361 } else if (isMinValuePlusOne(CI)) {
1362 if (I.getOpcode() == Instruction::SetLT) // A < MIN+1 -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001363 return BinaryOperator::create(Instruction::SetEQ, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001364 if (I.getOpcode() == Instruction::SetGE) // A >= MIN-1 -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001365 return BinaryOperator::create(Instruction::SetNE, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001366
1367 } else if (isMaxValueMinusOne(CI)) {
1368 if (I.getOpcode() == Instruction::SetGT) // A > MAX-1 -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001369 return BinaryOperator::create(Instruction::SetEQ, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001370 if (I.getOpcode() == Instruction::SetLE) // A <= MAX-1 -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001371 return BinaryOperator::create(Instruction::SetNE, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001372 }
Chris Lattner59611142004-02-23 05:47:48 +00001373
1374 // If we still have a setle or setge instruction, turn it into the
1375 // appropriate setlt or setgt instruction. Since the border cases have
1376 // already been handled above, this requires little checking.
1377 //
1378 if (I.getOpcode() == Instruction::SetLE)
1379 return BinaryOperator::create(Instruction::SetLT, Op0, AddOne(CI));
1380 if (I.getOpcode() == Instruction::SetGE)
1381 return BinaryOperator::create(Instruction::SetGT, Op0, SubOne(CI));
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001382 }
1383
Chris Lattner16930792003-11-03 04:25:02 +00001384 // Test to see if the operands of the setcc are casted versions of other
1385 // values. If the cast can be stripped off both arguments, we do so now.
Chris Lattner6444c372003-11-03 05:17:03 +00001386 if (CastInst *CI = dyn_cast<CastInst>(Op0)) {
1387 Value *CastOp0 = CI->getOperand(0);
1388 if (CastOp0->getType()->isLosslesslyConvertibleTo(CI->getType()) &&
Chris Lattner16930792003-11-03 04:25:02 +00001389 !isa<Argument>(Op1) &&
1390 (I.getOpcode() == Instruction::SetEQ ||
1391 I.getOpcode() == Instruction::SetNE)) {
1392 // We keep moving the cast from the left operand over to the right
1393 // operand, where it can often be eliminated completely.
Chris Lattner6444c372003-11-03 05:17:03 +00001394 Op0 = CastOp0;
Chris Lattner16930792003-11-03 04:25:02 +00001395
1396 // If operand #1 is a cast instruction, see if we can eliminate it as
1397 // well.
Chris Lattner6444c372003-11-03 05:17:03 +00001398 if (CastInst *CI2 = dyn_cast<CastInst>(Op1))
1399 if (CI2->getOperand(0)->getType()->isLosslesslyConvertibleTo(
Chris Lattner16930792003-11-03 04:25:02 +00001400 Op0->getType()))
Chris Lattner6444c372003-11-03 05:17:03 +00001401 Op1 = CI2->getOperand(0);
Chris Lattner16930792003-11-03 04:25:02 +00001402
1403 // If Op1 is a constant, we can fold the cast into the constant.
1404 if (Op1->getType() != Op0->getType())
1405 if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
1406 Op1 = ConstantExpr::getCast(Op1C, Op0->getType());
1407 } else {
1408 // Otherwise, cast the RHS right before the setcc
1409 Op1 = new CastInst(Op1, Op0->getType(), Op1->getName());
1410 InsertNewInstBefore(cast<Instruction>(Op1), I);
1411 }
1412 return BinaryOperator::create(I.getOpcode(), Op0, Op1);
1413 }
1414
Chris Lattner6444c372003-11-03 05:17:03 +00001415 // Handle the special case of: setcc (cast bool to X), <cst>
1416 // This comes up when you have code like
1417 // int X = A < B;
1418 // if (X) ...
1419 // For generality, we handle any zero-extension of any operand comparison
1420 // with a constant.
1421 if (ConstantInt *ConstantRHS = dyn_cast<ConstantInt>(Op1)) {
1422 const Type *SrcTy = CastOp0->getType();
1423 const Type *DestTy = Op0->getType();
1424 if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
1425 (SrcTy->isUnsigned() || SrcTy == Type::BoolTy)) {
1426 // Ok, we have an expansion of operand 0 into a new type. Get the
1427 // constant value, masink off bits which are not set in the RHS. These
1428 // could be set if the destination value is signed.
1429 uint64_t ConstVal = ConstantRHS->getRawValue();
1430 ConstVal &= (1ULL << DestTy->getPrimitiveSize()*8)-1;
1431
1432 // If the constant we are comparing it with has high bits set, which
1433 // don't exist in the original value, the values could never be equal,
1434 // because the source would be zero extended.
1435 unsigned SrcBits =
1436 SrcTy == Type::BoolTy ? 1 : SrcTy->getPrimitiveSize()*8;
Chris Lattner7c94d112003-11-05 17:31:36 +00001437 bool HasSignBit = ConstVal & (1ULL << (DestTy->getPrimitiveSize()*8-1));
1438 if (ConstVal & ~((1ULL << SrcBits)-1)) {
Chris Lattner6444c372003-11-03 05:17:03 +00001439 switch (I.getOpcode()) {
1440 default: assert(0 && "Unknown comparison type!");
1441 case Instruction::SetEQ:
1442 return ReplaceInstUsesWith(I, ConstantBool::False);
1443 case Instruction::SetNE:
1444 return ReplaceInstUsesWith(I, ConstantBool::True);
1445 case Instruction::SetLT:
1446 case Instruction::SetLE:
1447 if (DestTy->isSigned() && HasSignBit)
1448 return ReplaceInstUsesWith(I, ConstantBool::False);
1449 return ReplaceInstUsesWith(I, ConstantBool::True);
1450 case Instruction::SetGT:
1451 case Instruction::SetGE:
1452 if (DestTy->isSigned() && HasSignBit)
1453 return ReplaceInstUsesWith(I, ConstantBool::True);
1454 return ReplaceInstUsesWith(I, ConstantBool::False);
1455 }
1456 }
1457
1458 // Otherwise, we can replace the setcc with a setcc of the smaller
1459 // operand value.
1460 Op1 = ConstantExpr::getCast(cast<Constant>(Op1), SrcTy);
1461 return BinaryOperator::create(I.getOpcode(), CastOp0, Op1);
1462 }
1463 }
1464 }
Chris Lattner113f4f42002-06-25 16:13:24 +00001465 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001466}
1467
1468
1469
Chris Lattnere8d6c602003-03-10 19:16:08 +00001470Instruction *InstCombiner::visitShiftInst(ShiftInst &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +00001471 assert(I.getOperand(1)->getType() == Type::UByteTy);
1472 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001473 bool isLeftShift = I.getOpcode() == Instruction::Shl;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001474
1475 // shl X, 0 == X and shr X, 0 == X
1476 // shl 0, X == 0 and shr 0, X == 0
1477 if (Op1 == Constant::getNullValue(Type::UByteTy) ||
Chris Lattnere6794492002-08-12 21:17:25 +00001478 Op0 == Constant::getNullValue(Op0->getType()))
1479 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001480
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001481 // shr int -1, X = -1 (for any arithmetic shift rights of ~0)
1482 if (!isLeftShift)
1483 if (ConstantSInt *CSI = dyn_cast<ConstantSInt>(Op0))
1484 if (CSI->isAllOnesValue())
1485 return ReplaceInstUsesWith(I, CSI);
1486
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001487 if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(Op1)) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001488 // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr
1489 // of a signed value.
1490 //
Chris Lattnere8d6c602003-03-10 19:16:08 +00001491 unsigned TypeBits = Op0->getType()->getPrimitiveSize()*8;
1492 if (CUI->getValue() >= TypeBits &&
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001493 (!Op0->getType()->isSigned() || isLeftShift))
Chris Lattnere8d6c602003-03-10 19:16:08 +00001494 return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType()));
Chris Lattner55f3d942002-09-10 23:04:09 +00001495
Chris Lattnerede3fe02003-08-13 04:18:28 +00001496 // ((X*C1) << C2) == (X * (C1 << C2))
1497 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0))
1498 if (BO->getOpcode() == Instruction::Mul && isLeftShift)
1499 if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1)))
1500 return BinaryOperator::create(Instruction::Mul, BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001501 ConstantExpr::get(Instruction::Shl, BOOp, CUI));
Chris Lattnerede3fe02003-08-13 04:18:28 +00001502
1503
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001504 // If the operand is an bitwise operator with a constant RHS, and the
1505 // shift is the only use, we can pull it out of the shift.
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001506 if (Op0->hasOneUse())
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001507 if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0))
1508 if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) {
1509 bool isValid = true; // Valid only for And, Or, Xor
1510 bool highBitSet = false; // Transform if high bit of constant set?
1511
1512 switch (Op0BO->getOpcode()) {
1513 default: isValid = false; break; // Do not perform transform!
1514 case Instruction::Or:
1515 case Instruction::Xor:
1516 highBitSet = false;
1517 break;
1518 case Instruction::And:
1519 highBitSet = true;
1520 break;
1521 }
1522
1523 // If this is a signed shift right, and the high bit is modified
1524 // by the logical operation, do not perform the transformation.
1525 // The highBitSet boolean indicates the value of the high bit of
1526 // the constant which would cause it to be modified for this
1527 // operation.
1528 //
1529 if (isValid && !isLeftShift && !I.getType()->isUnsigned()) {
1530 uint64_t Val = Op0C->getRawValue();
1531 isValid = ((Val & (1 << (TypeBits-1))) != 0) == highBitSet;
1532 }
1533
1534 if (isValid) {
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001535 Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, CUI);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001536
1537 Instruction *NewShift =
1538 new ShiftInst(I.getOpcode(), Op0BO->getOperand(0), CUI,
1539 Op0BO->getName());
1540 Op0BO->setName("");
1541 InsertNewInstBefore(NewShift, I);
1542
1543 return BinaryOperator::create(Op0BO->getOpcode(), NewShift,
1544 NewRHS);
1545 }
1546 }
1547
Chris Lattner3204d4e2003-07-24 17:52:58 +00001548 // If this is a shift of a shift, see if we can fold the two together...
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001549 if (ShiftInst *Op0SI = dyn_cast<ShiftInst>(Op0))
Chris Lattnerab780df2003-07-24 18:38:56 +00001550 if (ConstantUInt *ShiftAmt1C =
1551 dyn_cast<ConstantUInt>(Op0SI->getOperand(1))) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001552 unsigned ShiftAmt1 = ShiftAmt1C->getValue();
1553 unsigned ShiftAmt2 = CUI->getValue();
1554
1555 // Check for (A << c1) << c2 and (A >> c1) >> c2
1556 if (I.getOpcode() == Op0SI->getOpcode()) {
1557 unsigned Amt = ShiftAmt1+ShiftAmt2; // Fold into one big shift...
1558 return new ShiftInst(I.getOpcode(), Op0SI->getOperand(0),
1559 ConstantUInt::get(Type::UByteTy, Amt));
1560 }
1561
Chris Lattnerab780df2003-07-24 18:38:56 +00001562 // Check for (A << c1) >> c2 or visaversa. If we are dealing with
1563 // signed types, we can only support the (A >> c1) << c2 configuration,
1564 // because it can not turn an arbitrary bit of A into a sign bit.
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001565 if (I.getType()->isUnsigned() || isLeftShift) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001566 // Calculate bitmask for what gets shifted off the edge...
1567 Constant *C = ConstantIntegral::getAllOnesValue(I.getType());
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001568 if (isLeftShift)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001569 C = ConstantExpr::get(Instruction::Shl, C, ShiftAmt1C);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001570 else
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001571 C = ConstantExpr::get(Instruction::Shr, C, ShiftAmt1C);
Chris Lattner3204d4e2003-07-24 17:52:58 +00001572
1573 Instruction *Mask =
1574 BinaryOperator::create(Instruction::And, Op0SI->getOperand(0),
1575 C, Op0SI->getOperand(0)->getName()+".mask");
1576 InsertNewInstBefore(Mask, I);
1577
1578 // Figure out what flavor of shift we should use...
1579 if (ShiftAmt1 == ShiftAmt2)
1580 return ReplaceInstUsesWith(I, Mask); // (A << c) >> c === A & c2
1581 else if (ShiftAmt1 < ShiftAmt2) {
1582 return new ShiftInst(I.getOpcode(), Mask,
1583 ConstantUInt::get(Type::UByteTy, ShiftAmt2-ShiftAmt1));
1584 } else {
1585 return new ShiftInst(Op0SI->getOpcode(), Mask,
1586 ConstantUInt::get(Type::UByteTy, ShiftAmt1-ShiftAmt2));
1587 }
1588 }
1589 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001590 }
Chris Lattner2e0fb392002-10-08 16:16:40 +00001591
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001592 return 0;
1593}
1594
1595
Chris Lattner48a44f72002-05-02 17:06:02 +00001596// isEliminableCastOfCast - Return true if it is valid to eliminate the CI
1597// instruction.
1598//
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001599static inline bool isEliminableCastOfCast(const Type *SrcTy, const Type *MidTy,
1600 const Type *DstTy) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001601
Chris Lattner650b6da2002-08-02 20:00:25 +00001602 // It is legal to eliminate the instruction if casting A->B->A if the sizes
1603 // are identical and the bits don't get reinterpreted (for example
Chris Lattner0bb75912002-08-14 23:21:10 +00001604 // int->float->int would not be allowed)
Misha Brukmane5838c42003-05-20 18:45:36 +00001605 if (SrcTy == DstTy && SrcTy->isLosslesslyConvertibleTo(MidTy))
Chris Lattner650b6da2002-08-02 20:00:25 +00001606 return true;
Chris Lattner48a44f72002-05-02 17:06:02 +00001607
1608 // Allow free casting and conversion of sizes as long as the sign doesn't
1609 // change...
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001610 if (SrcTy->isIntegral() && MidTy->isIntegral() && DstTy->isIntegral()) {
Chris Lattner650b6da2002-08-02 20:00:25 +00001611 unsigned SrcSize = SrcTy->getPrimitiveSize();
1612 unsigned MidSize = MidTy->getPrimitiveSize();
1613 unsigned DstSize = DstTy->getPrimitiveSize();
Chris Lattner650b6da2002-08-02 20:00:25 +00001614
Chris Lattner3732aca2002-08-15 16:15:25 +00001615 // Cases where we are monotonically decreasing the size of the type are
1616 // always ok, regardless of what sign changes are going on.
1617 //
Chris Lattner0bb75912002-08-14 23:21:10 +00001618 if (SrcSize >= MidSize && MidSize >= DstSize)
Chris Lattner650b6da2002-08-02 20:00:25 +00001619 return true;
Chris Lattner3732aca2002-08-15 16:15:25 +00001620
Chris Lattner555518c2002-09-23 23:39:43 +00001621 // Cases where the source and destination type are the same, but the middle
1622 // type is bigger are noops.
1623 //
1624 if (SrcSize == DstSize && MidSize > SrcSize)
1625 return true;
1626
Chris Lattner3732aca2002-08-15 16:15:25 +00001627 // If we are monotonically growing, things are more complex.
1628 //
1629 if (SrcSize <= MidSize && MidSize <= DstSize) {
1630 // We have eight combinations of signedness to worry about. Here's the
1631 // table:
1632 static const int SignTable[8] = {
1633 // CODE, SrcSigned, MidSigned, DstSigned, Comment
1634 1, // U U U Always ok
1635 1, // U U S Always ok
1636 3, // U S U Ok iff SrcSize != MidSize
1637 3, // U S S Ok iff SrcSize != MidSize
1638 0, // S U U Never ok
1639 2, // S U S Ok iff MidSize == DstSize
1640 1, // S S U Always ok
1641 1, // S S S Always ok
1642 };
1643
1644 // Choose an action based on the current entry of the signtable that this
1645 // cast of cast refers to...
1646 unsigned Row = SrcTy->isSigned()*4+MidTy->isSigned()*2+DstTy->isSigned();
1647 switch (SignTable[Row]) {
1648 case 0: return false; // Never ok
1649 case 1: return true; // Always ok
1650 case 2: return MidSize == DstSize; // Ok iff MidSize == DstSize
1651 case 3: // Ok iff SrcSize != MidSize
1652 return SrcSize != MidSize || SrcTy == Type::BoolTy;
1653 default: assert(0 && "Bad entry in sign table!");
1654 }
Chris Lattner3732aca2002-08-15 16:15:25 +00001655 }
Chris Lattner650b6da2002-08-02 20:00:25 +00001656 }
Chris Lattner48a44f72002-05-02 17:06:02 +00001657
1658 // Otherwise, we cannot succeed. Specifically we do not want to allow things
1659 // like: short -> ushort -> uint, because this can create wrong results if
1660 // the input short is negative!
1661 //
1662 return false;
1663}
1664
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001665static bool ValueRequiresCast(const Value *V, const Type *Ty) {
1666 if (V->getType() == Ty || isa<Constant>(V)) return false;
1667 if (const CastInst *CI = dyn_cast<CastInst>(V))
1668 if (isEliminableCastOfCast(CI->getOperand(0)->getType(), CI->getType(), Ty))
1669 return false;
1670 return true;
1671}
1672
1673/// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
1674/// InsertBefore instruction. This is specialized a bit to avoid inserting
1675/// casts that are known to not do anything...
1676///
1677Value *InstCombiner::InsertOperandCastBefore(Value *V, const Type *DestTy,
1678 Instruction *InsertBefore) {
1679 if (V->getType() == DestTy) return V;
1680 if (Constant *C = dyn_cast<Constant>(V))
1681 return ConstantExpr::getCast(C, DestTy);
1682
1683 CastInst *CI = new CastInst(V, DestTy, V->getName());
1684 InsertNewInstBefore(CI, *InsertBefore);
1685 return CI;
1686}
Chris Lattner48a44f72002-05-02 17:06:02 +00001687
1688// CastInst simplification
Chris Lattner260ab202002-04-18 17:39:14 +00001689//
Chris Lattner113f4f42002-06-25 16:13:24 +00001690Instruction *InstCombiner::visitCastInst(CastInst &CI) {
Chris Lattner55d4bda2003-06-23 21:59:52 +00001691 Value *Src = CI.getOperand(0);
1692
Chris Lattner48a44f72002-05-02 17:06:02 +00001693 // If the user is casting a value to the same type, eliminate this cast
1694 // instruction...
Chris Lattner55d4bda2003-06-23 21:59:52 +00001695 if (CI.getType() == Src->getType())
1696 return ReplaceInstUsesWith(CI, Src);
Chris Lattner48a44f72002-05-02 17:06:02 +00001697
Chris Lattner48a44f72002-05-02 17:06:02 +00001698 // If casting the result of another cast instruction, try to eliminate this
1699 // one!
1700 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00001701 if (CastInst *CSrc = dyn_cast<CastInst>(Src)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001702 if (isEliminableCastOfCast(CSrc->getOperand(0)->getType(),
1703 CSrc->getType(), CI.getType())) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001704 // This instruction now refers directly to the cast's src operand. This
1705 // has a good chance of making CSrc dead.
Chris Lattner113f4f42002-06-25 16:13:24 +00001706 CI.setOperand(0, CSrc->getOperand(0));
1707 return &CI;
Chris Lattner48a44f72002-05-02 17:06:02 +00001708 }
1709
Chris Lattner650b6da2002-08-02 20:00:25 +00001710 // If this is an A->B->A cast, and we are dealing with integral types, try
1711 // to convert this into a logical 'and' instruction.
1712 //
1713 if (CSrc->getOperand(0)->getType() == CI.getType() &&
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001714 CI.getType()->isInteger() && CSrc->getType()->isInteger() &&
Chris Lattner650b6da2002-08-02 20:00:25 +00001715 CI.getType()->isUnsigned() && CSrc->getType()->isUnsigned() &&
1716 CSrc->getType()->getPrimitiveSize() < CI.getType()->getPrimitiveSize()){
1717 assert(CSrc->getType() != Type::ULongTy &&
1718 "Cannot have type bigger than ulong!");
Chris Lattner196897c2003-05-26 23:41:32 +00001719 uint64_t AndValue = (1ULL << CSrc->getType()->getPrimitiveSize()*8)-1;
Chris Lattner650b6da2002-08-02 20:00:25 +00001720 Constant *AndOp = ConstantUInt::get(CI.getType(), AndValue);
1721 return BinaryOperator::create(Instruction::And, CSrc->getOperand(0),
1722 AndOp);
1723 }
1724 }
1725
Chris Lattnerd0d51602003-06-21 23:12:02 +00001726 // If casting the result of a getelementptr instruction with no offset, turn
1727 // this into a cast of the original pointer!
1728 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00001729 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
Chris Lattnerd0d51602003-06-21 23:12:02 +00001730 bool AllZeroOperands = true;
1731 for (unsigned i = 1, e = GEP->getNumOperands(); i != e; ++i)
1732 if (!isa<Constant>(GEP->getOperand(i)) ||
1733 !cast<Constant>(GEP->getOperand(i))->isNullValue()) {
1734 AllZeroOperands = false;
1735 break;
1736 }
1737 if (AllZeroOperands) {
1738 CI.setOperand(0, GEP->getOperand(0));
1739 return &CI;
1740 }
1741 }
1742
Chris Lattnerf4ad1652003-11-02 05:57:39 +00001743 // If we are casting a malloc or alloca to a pointer to a type of the same
1744 // size, rewrite the allocation instruction to allocate the "right" type.
1745 //
1746 if (AllocationInst *AI = dyn_cast<AllocationInst>(Src))
Chris Lattnerd4d987d2003-11-02 06:54:48 +00001747 if (AI->hasOneUse() && !AI->isArrayAllocation())
Chris Lattnerf4ad1652003-11-02 05:57:39 +00001748 if (const PointerType *PTy = dyn_cast<PointerType>(CI.getType())) {
1749 // Get the type really allocated and the type casted to...
1750 const Type *AllocElTy = AI->getAllocatedType();
1751 unsigned AllocElTySize = TD->getTypeSize(AllocElTy);
1752 const Type *CastElTy = PTy->getElementType();
1753 unsigned CastElTySize = TD->getTypeSize(CastElTy);
Chris Lattner7c94d112003-11-05 17:31:36 +00001754
Chris Lattnerf4ad1652003-11-02 05:57:39 +00001755 // If the allocation is for an even multiple of the cast type size
Chris Lattneraf789322003-11-03 01:29:41 +00001756 if (CastElTySize && (AllocElTySize % CastElTySize == 0)) {
Chris Lattnerf4ad1652003-11-02 05:57:39 +00001757 Value *Amt = ConstantUInt::get(Type::UIntTy,
1758 AllocElTySize/CastElTySize);
1759 std::string Name = AI->getName(); AI->setName("");
1760 AllocationInst *New;
1761 if (isa<MallocInst>(AI))
1762 New = new MallocInst(CastElTy, Amt, Name);
1763 else
1764 New = new AllocaInst(CastElTy, Amt, Name);
1765 InsertNewInstBefore(New, CI);
1766 return ReplaceInstUsesWith(CI, New);
1767 }
1768 }
1769
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001770 // If the source value is an instruction with only this use, we can attempt to
1771 // propagate the cast into the instruction. Also, only handle integral types
1772 // for now.
1773 if (Instruction *SrcI = dyn_cast<Instruction>(Src))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001774 if (SrcI->hasOneUse() && Src->getType()->isIntegral() &&
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001775 CI.getType()->isInteger()) { // Don't mess with casts to bool here
1776 const Type *DestTy = CI.getType();
1777 unsigned SrcBitSize = getTypeSizeInBits(Src->getType());
1778 unsigned DestBitSize = getTypeSizeInBits(DestTy);
1779
1780 Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0;
1781 Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0;
1782
1783 switch (SrcI->getOpcode()) {
1784 case Instruction::Add:
1785 case Instruction::Mul:
1786 case Instruction::And:
1787 case Instruction::Or:
1788 case Instruction::Xor:
1789 // If we are discarding information, or just changing the sign, rewrite.
1790 if (DestBitSize <= SrcBitSize && DestBitSize != 1) {
1791 // Don't insert two casts if they cannot be eliminated. We allow two
1792 // casts to be inserted if the sizes are the same. This could only be
1793 // converting signedness, which is a noop.
1794 if (DestBitSize == SrcBitSize || !ValueRequiresCast(Op1, DestTy) ||
1795 !ValueRequiresCast(Op0, DestTy)) {
1796 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
1797 Value *Op1c = InsertOperandCastBefore(Op1, DestTy, SrcI);
1798 return BinaryOperator::create(cast<BinaryOperator>(SrcI)
1799 ->getOpcode(), Op0c, Op1c);
1800 }
1801 }
1802 break;
1803 case Instruction::Shl:
1804 // Allow changing the sign of the source operand. Do not allow changing
1805 // the size of the shift, UNLESS the shift amount is a constant. We
1806 // mush not change variable sized shifts to a smaller size, because it
1807 // is undefined to shift more bits out than exist in the value.
1808 if (DestBitSize == SrcBitSize ||
1809 (DestBitSize < SrcBitSize && isa<Constant>(Op1))) {
1810 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
1811 return new ShiftInst(Instruction::Shl, Op0c, Op1);
1812 }
1813 break;
1814 }
1815 }
1816
Chris Lattner260ab202002-04-18 17:39:14 +00001817 return 0;
Chris Lattnerca081252001-12-14 16:52:21 +00001818}
1819
Chris Lattner970c33a2003-06-19 17:00:31 +00001820// CallInst simplification
1821//
1822Instruction *InstCombiner::visitCallInst(CallInst &CI) {
Chris Lattneraec3d942003-10-07 22:32:43 +00001823 return visitCallSite(&CI);
Chris Lattner970c33a2003-06-19 17:00:31 +00001824}
1825
1826// InvokeInst simplification
1827//
1828Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) {
Chris Lattneraec3d942003-10-07 22:32:43 +00001829 return visitCallSite(&II);
Chris Lattner970c33a2003-06-19 17:00:31 +00001830}
1831
1832// getPromotedType - Return the specified type promoted as it would be to pass
1833// though a va_arg area...
1834static const Type *getPromotedType(const Type *Ty) {
1835 switch (Ty->getPrimitiveID()) {
1836 case Type::SByteTyID:
1837 case Type::ShortTyID: return Type::IntTy;
1838 case Type::UByteTyID:
1839 case Type::UShortTyID: return Type::UIntTy;
1840 case Type::FloatTyID: return Type::DoubleTy;
1841 default: return Ty;
1842 }
1843}
1844
Chris Lattneraec3d942003-10-07 22:32:43 +00001845// visitCallSite - Improvements for call and invoke instructions.
1846//
1847Instruction *InstCombiner::visitCallSite(CallSite CS) {
Chris Lattner75b4d1d2003-10-07 22:54:13 +00001848 bool Changed = false;
1849
1850 // If the callee is a constexpr cast of a function, attempt to move the cast
1851 // to the arguments of the call/invoke.
Chris Lattneraec3d942003-10-07 22:32:43 +00001852 if (transformConstExprCastCall(CS)) return 0;
1853
Chris Lattner75b4d1d2003-10-07 22:54:13 +00001854 Value *Callee = CS.getCalledValue();
1855 const PointerType *PTy = cast<PointerType>(Callee->getType());
1856 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1857 if (FTy->isVarArg()) {
1858 // See if we can optimize any arguments passed through the varargs area of
1859 // the call.
1860 for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(),
1861 E = CS.arg_end(); I != E; ++I)
1862 if (CastInst *CI = dyn_cast<CastInst>(*I)) {
1863 // If this cast does not effect the value passed through the varargs
1864 // area, we can eliminate the use of the cast.
1865 Value *Op = CI->getOperand(0);
1866 if (CI->getType()->isLosslesslyConvertibleTo(Op->getType())) {
1867 *I = Op;
1868 Changed = true;
1869 }
1870 }
1871 }
Chris Lattneraec3d942003-10-07 22:32:43 +00001872
Chris Lattner75b4d1d2003-10-07 22:54:13 +00001873 return Changed ? CS.getInstruction() : 0;
Chris Lattneraec3d942003-10-07 22:32:43 +00001874}
1875
Chris Lattner970c33a2003-06-19 17:00:31 +00001876// transformConstExprCastCall - If the callee is a constexpr cast of a function,
1877// attempt to move the cast to the arguments of the call/invoke.
1878//
1879bool InstCombiner::transformConstExprCastCall(CallSite CS) {
1880 if (!isa<ConstantExpr>(CS.getCalledValue())) return false;
1881 ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue());
1882 if (CE->getOpcode() != Instruction::Cast ||
1883 !isa<ConstantPointerRef>(CE->getOperand(0)))
1884 return false;
1885 ConstantPointerRef *CPR = cast<ConstantPointerRef>(CE->getOperand(0));
1886 if (!isa<Function>(CPR->getValue())) return false;
1887 Function *Callee = cast<Function>(CPR->getValue());
1888 Instruction *Caller = CS.getInstruction();
1889
1890 // Okay, this is a cast from a function to a different type. Unless doing so
1891 // would cause a type conversion of one of our arguments, change this call to
1892 // be a direct call with arguments casted to the appropriate types.
1893 //
1894 const FunctionType *FT = Callee->getFunctionType();
1895 const Type *OldRetTy = Caller->getType();
1896
Chris Lattner1f7942f2004-01-14 06:06:08 +00001897 // Check to see if we are changing the return type...
1898 if (OldRetTy != FT->getReturnType()) {
1899 if (Callee->isExternal() &&
1900 !OldRetTy->isLosslesslyConvertibleTo(FT->getReturnType()) &&
1901 !Caller->use_empty())
1902 return false; // Cannot transform this return value...
1903
1904 // If the callsite is an invoke instruction, and the return value is used by
1905 // a PHI node in a successor, we cannot change the return type of the call
1906 // because there is no place to put the cast instruction (without breaking
1907 // the critical edge). Bail out in this case.
1908 if (!Caller->use_empty())
1909 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller))
1910 for (Value::use_iterator UI = II->use_begin(), E = II->use_end();
1911 UI != E; ++UI)
1912 if (PHINode *PN = dyn_cast<PHINode>(*UI))
1913 if (PN->getParent() == II->getNormalDest() ||
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001914 PN->getParent() == II->getUnwindDest())
Chris Lattner1f7942f2004-01-14 06:06:08 +00001915 return false;
1916 }
Chris Lattner970c33a2003-06-19 17:00:31 +00001917
1918 unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin());
1919 unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs);
1920
1921 CallSite::arg_iterator AI = CS.arg_begin();
1922 for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) {
1923 const Type *ParamTy = FT->getParamType(i);
1924 bool isConvertible = (*AI)->getType()->isLosslesslyConvertibleTo(ParamTy);
1925 if (Callee->isExternal() && !isConvertible) return false;
1926 }
1927
1928 if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() &&
1929 Callee->isExternal())
1930 return false; // Do not delete arguments unless we have a function body...
1931
1932 // Okay, we decided that this is a safe thing to do: go ahead and start
1933 // inserting cast instructions as necessary...
1934 std::vector<Value*> Args;
1935 Args.reserve(NumActualArgs);
1936
1937 AI = CS.arg_begin();
1938 for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) {
1939 const Type *ParamTy = FT->getParamType(i);
1940 if ((*AI)->getType() == ParamTy) {
1941 Args.push_back(*AI);
1942 } else {
1943 Instruction *Cast = new CastInst(*AI, ParamTy, "tmp");
1944 InsertNewInstBefore(Cast, *Caller);
1945 Args.push_back(Cast);
1946 }
1947 }
1948
1949 // If the function takes more arguments than the call was taking, add them
1950 // now...
1951 for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i)
1952 Args.push_back(Constant::getNullValue(FT->getParamType(i)));
1953
1954 // If we are removing arguments to the function, emit an obnoxious warning...
1955 if (FT->getNumParams() < NumActualArgs)
1956 if (!FT->isVarArg()) {
1957 std::cerr << "WARNING: While resolving call to function '"
1958 << Callee->getName() << "' arguments were dropped!\n";
1959 } else {
1960 // Add all of the arguments in their promoted form to the arg list...
1961 for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) {
1962 const Type *PTy = getPromotedType((*AI)->getType());
1963 if (PTy != (*AI)->getType()) {
1964 // Must promote to pass through va_arg area!
1965 Instruction *Cast = new CastInst(*AI, PTy, "tmp");
1966 InsertNewInstBefore(Cast, *Caller);
1967 Args.push_back(Cast);
1968 } else {
1969 Args.push_back(*AI);
1970 }
1971 }
1972 }
1973
1974 if (FT->getReturnType() == Type::VoidTy)
1975 Caller->setName(""); // Void type should not have a name...
1976
1977 Instruction *NC;
1978 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001979 NC = new InvokeInst(Callee, II->getNormalDest(), II->getUnwindDest(),
Chris Lattner970c33a2003-06-19 17:00:31 +00001980 Args, Caller->getName(), Caller);
1981 } else {
1982 NC = new CallInst(Callee, Args, Caller->getName(), Caller);
1983 }
1984
1985 // Insert a cast of the return type as necessary...
1986 Value *NV = NC;
1987 if (Caller->getType() != NV->getType() && !Caller->use_empty()) {
1988 if (NV->getType() != Type::VoidTy) {
1989 NV = NC = new CastInst(NC, Caller->getType(), "tmp");
Chris Lattner686767f2003-10-30 00:46:41 +00001990
1991 // If this is an invoke instruction, we should insert it after the first
1992 // non-phi, instruction in the normal successor block.
1993 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
1994 BasicBlock::iterator I = II->getNormalDest()->begin();
1995 while (isa<PHINode>(I)) ++I;
1996 InsertNewInstBefore(NC, *I);
1997 } else {
1998 // Otherwise, it's a call, just insert cast right after the call instr
1999 InsertNewInstBefore(NC, *Caller);
2000 }
Chris Lattner970c33a2003-06-19 17:00:31 +00002001 AddUsesToWorkList(*Caller);
2002 } else {
2003 NV = Constant::getNullValue(Caller->getType());
2004 }
2005 }
2006
2007 if (Caller->getType() != Type::VoidTy && !Caller->use_empty())
2008 Caller->replaceAllUsesWith(NV);
2009 Caller->getParent()->getInstList().erase(Caller);
2010 removeFromWorkList(Caller);
2011 return true;
2012}
2013
2014
Chris Lattner48a44f72002-05-02 17:06:02 +00002015
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002016// PHINode simplification
2017//
Chris Lattner113f4f42002-06-25 16:13:24 +00002018Instruction *InstCombiner::visitPHINode(PHINode &PN) {
Chris Lattner91daeb52003-12-19 05:58:40 +00002019 if (Value *V = hasConstantValue(&PN))
2020 return ReplaceInstUsesWith(PN, V);
Chris Lattner4db2d222004-02-16 05:07:08 +00002021
2022 // If the only user of this instruction is a cast instruction, and all of the
2023 // incoming values are constants, change this PHI to merge together the casted
2024 // constants.
2025 if (PN.hasOneUse())
2026 if (CastInst *CI = dyn_cast<CastInst>(PN.use_back()))
2027 if (CI->getType() != PN.getType()) { // noop casts will be folded
2028 bool AllConstant = true;
2029 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i)
2030 if (!isa<Constant>(PN.getIncomingValue(i))) {
2031 AllConstant = false;
2032 break;
2033 }
2034 if (AllConstant) {
2035 // Make a new PHI with all casted values.
2036 PHINode *New = new PHINode(CI->getType(), PN.getName(), &PN);
2037 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
2038 Constant *OldArg = cast<Constant>(PN.getIncomingValue(i));
2039 New->addIncoming(ConstantExpr::getCast(OldArg, New->getType()),
2040 PN.getIncomingBlock(i));
2041 }
2042
2043 // Update the cast instruction.
2044 CI->setOperand(0, New);
2045 WorkList.push_back(CI); // revisit the cast instruction to fold.
2046 WorkList.push_back(New); // Make sure to revisit the new Phi
2047 return &PN; // PN is now dead!
2048 }
2049 }
Chris Lattner91daeb52003-12-19 05:58:40 +00002050 return 0;
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002051}
2052
Chris Lattner48a44f72002-05-02 17:06:02 +00002053
Chris Lattner113f4f42002-06-25 16:13:24 +00002054Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) {
Chris Lattner471bd762003-05-22 19:07:21 +00002055 // Is it 'getelementptr %P, long 0' or 'getelementptr %P'
Chris Lattner113f4f42002-06-25 16:13:24 +00002056 // If so, eliminate the noop.
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002057 if (GEP.getNumOperands() == 1)
2058 return ReplaceInstUsesWith(GEP, GEP.getOperand(0));
2059
2060 bool HasZeroPointerIndex = false;
2061 if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1)))
2062 HasZeroPointerIndex = C->isNullValue();
2063
2064 if (GEP.getNumOperands() == 2 && HasZeroPointerIndex)
Chris Lattnere6794492002-08-12 21:17:25 +00002065 return ReplaceInstUsesWith(GEP, GEP.getOperand(0));
Chris Lattner48a44f72002-05-02 17:06:02 +00002066
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002067 // Combine Indices - If the source pointer to this getelementptr instruction
2068 // is a getelementptr instruction, combine the indices of the two
2069 // getelementptr instructions into a single instruction.
2070 //
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002071 if (GetElementPtrInst *Src = dyn_cast<GetElementPtrInst>(GEP.getOperand(0))) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002072 std::vector<Value *> Indices;
Chris Lattnerca081252001-12-14 16:52:21 +00002073
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002074 // Can we combine the two pointer arithmetics offsets?
Chris Lattner471bd762003-05-22 19:07:21 +00002075 if (Src->getNumOperands() == 2 && isa<Constant>(Src->getOperand(1)) &&
2076 isa<Constant>(GEP.getOperand(1))) {
Chris Lattner235af562003-03-05 22:33:14 +00002077 // Replace: gep (gep %P, long C1), long C2, ...
2078 // With: gep %P, long (C1+C2), ...
Chris Lattner34428442003-05-27 16:40:51 +00002079 Value *Sum = ConstantExpr::get(Instruction::Add,
2080 cast<Constant>(Src->getOperand(1)),
2081 cast<Constant>(GEP.getOperand(1)));
Chris Lattner235af562003-03-05 22:33:14 +00002082 assert(Sum && "Constant folding of longs failed!?");
2083 GEP.setOperand(0, Src->getOperand(0));
2084 GEP.setOperand(1, Sum);
2085 AddUsesToWorkList(*Src); // Reduce use count of Src
2086 return &GEP;
Chris Lattner471bd762003-05-22 19:07:21 +00002087 } else if (Src->getNumOperands() == 2) {
Chris Lattner235af562003-03-05 22:33:14 +00002088 // Replace: gep (gep %P, long B), long A, ...
2089 // With: T = long A+B; gep %P, T, ...
2090 //
2091 Value *Sum = BinaryOperator::create(Instruction::Add, Src->getOperand(1),
2092 GEP.getOperand(1),
2093 Src->getName()+".sum", &GEP);
2094 GEP.setOperand(0, Src->getOperand(0));
2095 GEP.setOperand(1, Sum);
2096 WorkList.push_back(cast<Instruction>(Sum));
2097 return &GEP;
Chris Lattner5d606a02002-11-04 16:43:32 +00002098 } else if (*GEP.idx_begin() == Constant::getNullValue(Type::LongTy) &&
Chris Lattnera8339e32002-09-17 21:05:42 +00002099 Src->getNumOperands() != 1) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002100 // Otherwise we can do the fold if the first index of the GEP is a zero
2101 Indices.insert(Indices.end(), Src->idx_begin(), Src->idx_end());
2102 Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end());
Chris Lattner5d606a02002-11-04 16:43:32 +00002103 } else if (Src->getOperand(Src->getNumOperands()-1) ==
2104 Constant::getNullValue(Type::LongTy)) {
2105 // If the src gep ends with a constant array index, merge this get into
2106 // it, even if we have a non-zero array index.
2107 Indices.insert(Indices.end(), Src->idx_begin(), Src->idx_end()-1);
2108 Indices.insert(Indices.end(), GEP.idx_begin(), GEP.idx_end());
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002109 }
2110
2111 if (!Indices.empty())
2112 return new GetElementPtrInst(Src->getOperand(0), Indices, GEP.getName());
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002113
2114 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(GEP.getOperand(0))) {
2115 // GEP of global variable. If all of the indices for this GEP are
2116 // constants, we can promote this to a constexpr instead of an instruction.
2117
2118 // Scan for nonconstants...
2119 std::vector<Constant*> Indices;
2120 User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end();
2121 for (; I != E && isa<Constant>(*I); ++I)
2122 Indices.push_back(cast<Constant>(*I));
2123
2124 if (I == E) { // If they are all constants...
Chris Lattner46b3d302003-04-16 22:40:51 +00002125 Constant *CE =
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002126 ConstantExpr::getGetElementPtr(ConstantPointerRef::get(GV), Indices);
2127
2128 // Replace all uses of the GEP with the new constexpr...
2129 return ReplaceInstUsesWith(GEP, CE);
2130 }
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002131 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(GEP.getOperand(0))) {
2132 if (CE->getOpcode() == Instruction::Cast) {
2133 if (HasZeroPointerIndex) {
2134 // transform: GEP (cast [10 x ubyte]* X to [0 x ubyte]*), long 0, ...
2135 // into : GEP [10 x ubyte]* X, long 0, ...
2136 //
2137 // This occurs when the program declares an array extern like "int X[];"
2138 //
2139 Constant *X = CE->getOperand(0);
2140 const PointerType *CPTy = cast<PointerType>(CE->getType());
2141 if (const PointerType *XTy = dyn_cast<PointerType>(X->getType()))
2142 if (const ArrayType *XATy =
2143 dyn_cast<ArrayType>(XTy->getElementType()))
2144 if (const ArrayType *CATy =
2145 dyn_cast<ArrayType>(CPTy->getElementType()))
2146 if (CATy->getElementType() == XATy->getElementType()) {
2147 // At this point, we know that the cast source type is a pointer
2148 // to an array of the same type as the destination pointer
2149 // array. Because the array type is never stepped over (there
2150 // is a leading zero) we can fold the cast into this GEP.
2151 GEP.setOperand(0, X);
2152 return &GEP;
2153 }
2154 }
2155 }
Chris Lattnerca081252001-12-14 16:52:21 +00002156 }
2157
Chris Lattnerca081252001-12-14 16:52:21 +00002158 return 0;
2159}
2160
Chris Lattner1085bdf2002-11-04 16:18:53 +00002161Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) {
2162 // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1
2163 if (AI.isArrayAllocation()) // Check C != 1
2164 if (const ConstantUInt *C = dyn_cast<ConstantUInt>(AI.getArraySize())) {
2165 const Type *NewTy = ArrayType::get(AI.getAllocatedType(), C->getValue());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002166 AllocationInst *New = 0;
Chris Lattner1085bdf2002-11-04 16:18:53 +00002167
2168 // Create and insert the replacement instruction...
2169 if (isa<MallocInst>(AI))
2170 New = new MallocInst(NewTy, 0, AI.getName(), &AI);
Chris Lattnera2620ac2002-11-09 00:49:43 +00002171 else {
2172 assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!");
Chris Lattner1085bdf2002-11-04 16:18:53 +00002173 New = new AllocaInst(NewTy, 0, AI.getName(), &AI);
Chris Lattnera2620ac2002-11-09 00:49:43 +00002174 }
Chris Lattner1085bdf2002-11-04 16:18:53 +00002175
2176 // Scan to the end of the allocation instructions, to skip over a block of
2177 // allocas if possible...
2178 //
2179 BasicBlock::iterator It = New;
2180 while (isa<AllocationInst>(*It)) ++It;
2181
2182 // Now that I is pointing to the first non-allocation-inst in the block,
2183 // insert our getelementptr instruction...
2184 //
2185 std::vector<Value*> Idx(2, Constant::getNullValue(Type::LongTy));
2186 Value *V = new GetElementPtrInst(New, Idx, New->getName()+".sub", It);
2187
2188 // Now make everything use the getelementptr instead of the original
2189 // allocation.
2190 ReplaceInstUsesWith(AI, V);
2191 return &AI;
2192 }
2193 return 0;
2194}
2195
Chris Lattner8427bff2003-12-07 01:24:23 +00002196Instruction *InstCombiner::visitFreeInst(FreeInst &FI) {
2197 Value *Op = FI.getOperand(0);
2198
2199 // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X
2200 if (CastInst *CI = dyn_cast<CastInst>(Op))
2201 if (isa<PointerType>(CI->getOperand(0)->getType())) {
2202 FI.setOperand(0, CI->getOperand(0));
2203 return &FI;
2204 }
2205
2206 return 0;
2207}
2208
2209
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002210/// GetGEPGlobalInitializer - Given a constant, and a getelementptr
2211/// constantexpr, return the constant value being addressed by the constant
2212/// expression, or null if something is funny.
2213///
2214static Constant *GetGEPGlobalInitializer(Constant *C, ConstantExpr *CE) {
2215 if (CE->getOperand(1) != Constant::getNullValue(Type::LongTy))
2216 return 0; // Do not allow stepping over the value!
2217
2218 // Loop over all of the operands, tracking down which value we are
2219 // addressing...
2220 for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i)
2221 if (ConstantUInt *CU = dyn_cast<ConstantUInt>(CE->getOperand(i))) {
Chris Lattner76b2ff42004-02-15 05:55:15 +00002222 ConstantStruct *CS = dyn_cast<ConstantStruct>(C);
2223 if (CS == 0) return 0;
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002224 if (CU->getValue() >= CS->getValues().size()) return 0;
2225 C = cast<Constant>(CS->getValues()[CU->getValue()]);
2226 } else if (ConstantSInt *CS = dyn_cast<ConstantSInt>(CE->getOperand(i))) {
Chris Lattner76b2ff42004-02-15 05:55:15 +00002227 ConstantArray *CA = dyn_cast<ConstantArray>(C);
2228 if (CA == 0) return 0;
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002229 if ((uint64_t)CS->getValue() >= CA->getValues().size()) return 0;
2230 C = cast<Constant>(CA->getValues()[CS->getValue()]);
2231 } else
2232 return 0;
2233 return C;
2234}
2235
2236Instruction *InstCombiner::visitLoadInst(LoadInst &LI) {
2237 Value *Op = LI.getOperand(0);
Chris Lattner7e8af382004-01-12 04:13:56 +00002238 if (LI.isVolatile()) return 0;
2239
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002240 if (ConstantPointerRef *CPR = dyn_cast<ConstantPointerRef>(Op))
2241 Op = CPR->getValue();
2242
2243 // Instcombine load (constant global) into the value loaded...
2244 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002245 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002246 return ReplaceInstUsesWith(LI, GV->getInitializer());
2247
2248 // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded...
2249 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op))
2250 if (CE->getOpcode() == Instruction::GetElementPtr)
2251 if (ConstantPointerRef *G=dyn_cast<ConstantPointerRef>(CE->getOperand(0)))
2252 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(G->getValue()))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002253 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002254 if (Constant *V = GetGEPGlobalInitializer(GV->getInitializer(), CE))
2255 return ReplaceInstUsesWith(LI, V);
2256 return 0;
2257}
2258
2259
Chris Lattner9eef8a72003-06-04 04:46:00 +00002260Instruction *InstCombiner::visitBranchInst(BranchInst &BI) {
2261 // Change br (not X), label True, label False to: br X, label False, True
Chris Lattner45789ac2003-06-05 20:12:51 +00002262 if (BI.isConditional() && !isa<Constant>(BI.getCondition()))
Chris Lattnere967b342003-06-04 05:10:11 +00002263 if (Value *V = dyn_castNotVal(BI.getCondition())) {
2264 BasicBlock *TrueDest = BI.getSuccessor(0);
2265 BasicBlock *FalseDest = BI.getSuccessor(1);
2266 // Swap Destinations and condition...
2267 BI.setCondition(V);
2268 BI.setSuccessor(0, FalseDest);
2269 BI.setSuccessor(1, TrueDest);
2270 return &BI;
2271 }
Chris Lattner9eef8a72003-06-04 04:46:00 +00002272 return 0;
2273}
Chris Lattner1085bdf2002-11-04 16:18:53 +00002274
Chris Lattnerca081252001-12-14 16:52:21 +00002275
Chris Lattner99f48c62002-09-02 04:59:56 +00002276void InstCombiner::removeFromWorkList(Instruction *I) {
2277 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), I),
2278 WorkList.end());
2279}
2280
Chris Lattner113f4f42002-06-25 16:13:24 +00002281bool InstCombiner::runOnFunction(Function &F) {
Chris Lattner260ab202002-04-18 17:39:14 +00002282 bool Changed = false;
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002283 TD = &getAnalysis<TargetData>();
Chris Lattnerca081252001-12-14 16:52:21 +00002284
Chris Lattner260ab202002-04-18 17:39:14 +00002285 WorkList.insert(WorkList.end(), inst_begin(F), inst_end(F));
Chris Lattnerca081252001-12-14 16:52:21 +00002286
2287 while (!WorkList.empty()) {
2288 Instruction *I = WorkList.back(); // Get an instruction from the worklist
2289 WorkList.pop_back();
2290
Misha Brukman632df282002-10-29 23:06:16 +00002291 // Check to see if we can DCE or ConstantPropagate the instruction...
Chris Lattner99f48c62002-09-02 04:59:56 +00002292 // Check to see if we can DIE the instruction...
2293 if (isInstructionTriviallyDead(I)) {
2294 // Add operands to the worklist...
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002295 if (I->getNumOperands() < 4)
2296 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
2297 if (Instruction *Op = dyn_cast<Instruction>(I->getOperand(i)))
2298 WorkList.push_back(Op);
Chris Lattner99f48c62002-09-02 04:59:56 +00002299 ++NumDeadInst;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002300
2301 I->getParent()->getInstList().erase(I);
2302 removeFromWorkList(I);
2303 continue;
2304 }
Chris Lattner99f48c62002-09-02 04:59:56 +00002305
Misha Brukman632df282002-10-29 23:06:16 +00002306 // Instruction isn't dead, see if we can constant propagate it...
Chris Lattner99f48c62002-09-02 04:59:56 +00002307 if (Constant *C = ConstantFoldInstruction(I)) {
2308 // Add operands to the worklist...
2309 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
2310 if (Instruction *Op = dyn_cast<Instruction>(I->getOperand(i)))
2311 WorkList.push_back(Op);
Chris Lattnerc6509f42002-12-05 22:41:53 +00002312 ReplaceInstUsesWith(*I, C);
2313
Chris Lattner99f48c62002-09-02 04:59:56 +00002314 ++NumConstProp;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002315 I->getParent()->getInstList().erase(I);
Chris Lattner800aaaf2003-10-07 15:17:02 +00002316 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002317 continue;
Chris Lattner99f48c62002-09-02 04:59:56 +00002318 }
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002319
Chris Lattnerca081252001-12-14 16:52:21 +00002320 // Now that we have an instruction, try combining it to simplify it...
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002321 if (Instruction *Result = visit(*I)) {
Chris Lattner0b18c1d2002-05-10 15:38:35 +00002322 ++NumCombined;
Chris Lattner260ab202002-04-18 17:39:14 +00002323 // Should we replace the old instruction with a new one?
Chris Lattner053c0932002-05-14 15:24:07 +00002324 if (Result != I) {
2325 // Instructions can end up on the worklist more than once. Make sure
2326 // we do not process an instruction that has been deleted.
Chris Lattner99f48c62002-09-02 04:59:56 +00002327 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002328
2329 // Move the name to the new instruction first...
2330 std::string OldName = I->getName(); I->setName("");
Chris Lattner950fc782003-10-07 22:58:41 +00002331 Result->setName(OldName);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002332
2333 // Insert the new instruction into the basic block...
2334 BasicBlock *InstParent = I->getParent();
2335 InstParent->getInstList().insert(I, Result);
2336
2337 // Everything uses the new instruction now...
2338 I->replaceAllUsesWith(Result);
2339
2340 // Erase the old instruction.
2341 InstParent->getInstList().erase(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00002342 } else {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002343 BasicBlock::iterator II = I;
2344
2345 // If the instruction was modified, it's possible that it is now dead.
2346 // if so, remove it.
2347 if (dceInstruction(II)) {
2348 // Instructions may end up in the worklist more than once. Erase them
2349 // all.
Chris Lattner99f48c62002-09-02 04:59:56 +00002350 removeFromWorkList(I);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002351 Result = 0;
2352 }
Chris Lattner053c0932002-05-14 15:24:07 +00002353 }
Chris Lattner260ab202002-04-18 17:39:14 +00002354
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002355 if (Result) {
2356 WorkList.push_back(Result);
2357 AddUsesToWorkList(*Result);
2358 }
Chris Lattner260ab202002-04-18 17:39:14 +00002359 Changed = true;
Chris Lattnerca081252001-12-14 16:52:21 +00002360 }
2361 }
2362
Chris Lattner260ab202002-04-18 17:39:14 +00002363 return Changed;
Chris Lattner04805fa2002-02-26 21:46:54 +00002364}
2365
Chris Lattner8427bff2003-12-07 01:24:23 +00002366Pass *llvm::createInstructionCombiningPass() {
Chris Lattner260ab202002-04-18 17:39:14 +00002367 return new InstCombiner();
Chris Lattner04805fa2002-02-26 21:46:54 +00002368}
Brian Gaeke960707c2003-11-11 22:41:34 +00002369