Change data structure to memorize computed result in ScalarEvolution
[oota-llvm.git] / include / llvm / Analysis / ScalarEvolution.h
1 //===- llvm/Analysis/ScalarEvolution.h - Scalar Evolution -------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // The ScalarEvolution class is an LLVM pass which can be used to analyze and
11 // categorize scalar expressions in loops.  It specializes in recognizing
12 // general induction variables, representing them with the abstract and opaque
13 // SCEV class.  Given this analysis, trip counts of loops and other important
14 // properties can be obtained.
15 //
16 // This analysis is primarily useful for induction variable substitution and
17 // strength reduction.
18 //
19 //===----------------------------------------------------------------------===//
20
21 #ifndef LLVM_ANALYSIS_SCALAREVOLUTION_H
22 #define LLVM_ANALYSIS_SCALAREVOLUTION_H
23
24 #include "llvm/ADT/DenseSet.h"
25 #include "llvm/ADT/FoldingSet.h"
26 #include "llvm/IR/Function.h"
27 #include "llvm/IR/Instructions.h"
28 #include "llvm/IR/Operator.h"
29 #include "llvm/Pass.h"
30 #include "llvm/Support/Allocator.h"
31 #include "llvm/Support/ConstantRange.h"
32 #include "llvm/Support/DataTypes.h"
33 #include "llvm/Support/ValueHandle.h"
34 #include <map>
35
36 namespace llvm {
37   class APInt;
38   class Constant;
39   class ConstantInt;
40   class DominatorTree;
41   class Type;
42   class ScalarEvolution;
43   class DataLayout;
44   class TargetLibraryInfo;
45   class LLVMContext;
46   class Loop;
47   class LoopInfo;
48   class Operator;
49   class SCEVUnknown;
50   class SCEV;
51   template<> struct FoldingSetTrait<SCEV>;
52
53   /// SCEV - This class represents an analyzed expression in the program.  These
54   /// are opaque objects that the client is not allowed to do much with
55   /// directly.
56   ///
57   class SCEV : public FoldingSetNode {
58     friend struct FoldingSetTrait<SCEV>;
59
60     /// FastID - A reference to an Interned FoldingSetNodeID for this node.
61     /// The ScalarEvolution's BumpPtrAllocator holds the data.
62     FoldingSetNodeIDRef FastID;
63
64     // The SCEV baseclass this node corresponds to
65     const unsigned short SCEVType;
66
67   protected:
68     /// SubclassData - This field is initialized to zero and may be used in
69     /// subclasses to store miscellaneous information.
70     unsigned short SubclassData;
71
72   private:
73     SCEV(const SCEV &) LLVM_DELETED_FUNCTION;
74     void operator=(const SCEV &) LLVM_DELETED_FUNCTION;
75
76   public:
77     /// NoWrapFlags are bitfield indices into SubclassData.
78     ///
79     /// Add and Mul expressions may have no-unsigned-wrap <NUW> or
80     /// no-signed-wrap <NSW> properties, which are derived from the IR
81     /// operator. NSW is a misnomer that we use to mean no signed overflow or
82     /// underflow.
83     ///
84     /// AddRec expression may have a no-self-wraparound <NW> property if the
85     /// result can never reach the start value. This property is independent of
86     /// the actual start value and step direction. Self-wraparound is defined
87     /// purely in terms of the recurrence's loop, step size, and
88     /// bitwidth. Formally, a recurrence with no self-wraparound satisfies:
89     /// abs(step) * max-iteration(loop) <= unsigned-max(bitwidth).
90     ///
91     /// Note that NUW and NSW are also valid properties of a recurrence, and
92     /// either implies NW. For convenience, NW will be set for a recurrence
93     /// whenever either NUW or NSW are set.
94     enum NoWrapFlags { FlagAnyWrap = 0,          // No guarantee.
95                        FlagNW      = (1 << 0),   // No self-wrap.
96                        FlagNUW     = (1 << 1),   // No unsigned wrap.
97                        FlagNSW     = (1 << 2),   // No signed wrap.
98                        NoWrapMask  = (1 << 3) -1 };
99
100     explicit SCEV(const FoldingSetNodeIDRef ID, unsigned SCEVTy) :
101       FastID(ID), SCEVType(SCEVTy), SubclassData(0) {}
102
103     unsigned getSCEVType() const { return SCEVType; }
104
105     /// getType - Return the LLVM type of this SCEV expression.
106     ///
107     Type *getType() const;
108
109     /// isZero - Return true if the expression is a constant zero.
110     ///
111     bool isZero() const;
112
113     /// isOne - Return true if the expression is a constant one.
114     ///
115     bool isOne() const;
116
117     /// isAllOnesValue - Return true if the expression is a constant
118     /// all-ones value.
119     ///
120     bool isAllOnesValue() const;
121
122     /// isNonConstantNegative - Return true if the specified scev is negated,
123     /// but not a constant.
124     bool isNonConstantNegative() const;
125
126     /// print - Print out the internal representation of this scalar to the
127     /// specified stream.  This should really only be used for debugging
128     /// purposes.
129     void print(raw_ostream &OS) const;
130
131     /// dump - This method is used for debugging.
132     ///
133     void dump() const;
134   };
135
136   // Specialize FoldingSetTrait for SCEV to avoid needing to compute
137   // temporary FoldingSetNodeID values.
138   template<> struct FoldingSetTrait<SCEV> : DefaultFoldingSetTrait<SCEV> {
139     static void Profile(const SCEV &X, FoldingSetNodeID& ID) {
140       ID = X.FastID;
141     }
142     static bool Equals(const SCEV &X, const FoldingSetNodeID &ID,
143                        unsigned IDHash, FoldingSetNodeID &TempID) {
144       return ID == X.FastID;
145     }
146     static unsigned ComputeHash(const SCEV &X, FoldingSetNodeID &TempID) {
147       return X.FastID.ComputeHash();
148     }
149   };
150
151   inline raw_ostream &operator<<(raw_ostream &OS, const SCEV &S) {
152     S.print(OS);
153     return OS;
154   }
155
156   /// SCEVCouldNotCompute - An object of this class is returned by queries that
157   /// could not be answered.  For example, if you ask for the number of
158   /// iterations of a linked-list traversal loop, you will get one of these.
159   /// None of the standard SCEV operations are valid on this class, it is just a
160   /// marker.
161   struct SCEVCouldNotCompute : public SCEV {
162     SCEVCouldNotCompute();
163
164     /// Methods for support type inquiry through isa, cast, and dyn_cast:
165     static bool classof(const SCEV *S);
166   };
167
168   /// ScalarEvolution - This class is the main scalar evolution driver.  Because
169   /// client code (intentionally) can't do much with the SCEV objects directly,
170   /// they must ask this class for services.
171   ///
172   class ScalarEvolution : public FunctionPass {
173   public:
174     /// LoopDisposition - An enum describing the relationship between a
175     /// SCEV and a loop.
176     enum LoopDisposition {
177       LoopVariant,    ///< The SCEV is loop-variant (unknown).
178       LoopInvariant,  ///< The SCEV is loop-invariant.
179       LoopComputable  ///< The SCEV varies predictably with the loop.
180     };
181
182     /// BlockDisposition - An enum describing the relationship between a
183     /// SCEV and a basic block.
184     enum BlockDisposition {
185       DoesNotDominateBlock,  ///< The SCEV does not dominate the block.
186       DominatesBlock,        ///< The SCEV dominates the block.
187       ProperlyDominatesBlock ///< The SCEV properly dominates the block.
188     };
189
190     /// Convenient NoWrapFlags manipulation that hides enum casts and is
191     /// visible in the ScalarEvolution name space.
192     static SCEV::NoWrapFlags maskFlags(SCEV::NoWrapFlags Flags, int Mask) {
193       return (SCEV::NoWrapFlags)(Flags & Mask);
194     }
195     static SCEV::NoWrapFlags setFlags(SCEV::NoWrapFlags Flags,
196                                       SCEV::NoWrapFlags OnFlags) {
197       return (SCEV::NoWrapFlags)(Flags | OnFlags);
198     }
199     static SCEV::NoWrapFlags clearFlags(SCEV::NoWrapFlags Flags,
200                                         SCEV::NoWrapFlags OffFlags) {
201       return (SCEV::NoWrapFlags)(Flags & ~OffFlags);
202     }
203
204   private:
205     /// SCEVCallbackVH - A CallbackVH to arrange for ScalarEvolution to be
206     /// notified whenever a Value is deleted.
207     class SCEVCallbackVH : public CallbackVH {
208       ScalarEvolution *SE;
209       virtual void deleted();
210       virtual void allUsesReplacedWith(Value *New);
211     public:
212       SCEVCallbackVH(Value *V, ScalarEvolution *SE = 0);
213     };
214
215     friend class SCEVCallbackVH;
216     friend class SCEVExpander;
217     friend class SCEVUnknown;
218
219     /// F - The function we are analyzing.
220     ///
221     Function *F;
222
223     /// LI - The loop information for the function we are currently analyzing.
224     ///
225     LoopInfo *LI;
226
227     /// TD - The target data information for the target we are targeting.
228     ///
229     DataLayout *TD;
230
231     /// TLI - The target library information for the target we are targeting.
232     ///
233     TargetLibraryInfo *TLI;
234
235     /// DT - The dominator tree.
236     ///
237     DominatorTree *DT;
238
239     /// CouldNotCompute - This SCEV is used to represent unknown trip
240     /// counts and things.
241     SCEVCouldNotCompute CouldNotCompute;
242
243     /// ValueExprMapType - The typedef for ValueExprMap.
244     ///
245     typedef DenseMap<SCEVCallbackVH, const SCEV *, DenseMapInfo<Value *> >
246       ValueExprMapType;
247
248     /// ValueExprMap - This is a cache of the values we have analyzed so far.
249     ///
250     ValueExprMapType ValueExprMap;
251
252     /// Mark predicate values currently being processed by isImpliedCond.
253     DenseSet<Value*> PendingLoopPredicates;
254
255     /// ExitLimit - Information about the number of loop iterations for
256     /// which a loop exit's branch condition evaluates to the not-taken path.
257     /// This is a temporary pair of exact and max expressions that are
258     /// eventually summarized in ExitNotTakenInfo and BackedgeTakenInfo.
259     struct ExitLimit {
260       const SCEV *Exact;
261       const SCEV *Max;
262
263       /*implicit*/ ExitLimit(const SCEV *E) : Exact(E), Max(E) {}
264
265       ExitLimit(const SCEV *E, const SCEV *M) : Exact(E), Max(M) {}
266
267       /// hasAnyInfo - Test whether this ExitLimit contains any computed
268       /// information, or whether it's all SCEVCouldNotCompute values.
269       bool hasAnyInfo() const {
270         return !isa<SCEVCouldNotCompute>(Exact) ||
271           !isa<SCEVCouldNotCompute>(Max);
272       }
273     };
274
275     /// ExitNotTakenInfo - Information about the number of times a particular
276     /// loop exit may be reached before exiting the loop.
277     struct ExitNotTakenInfo {
278       AssertingVH<BasicBlock> ExitingBlock;
279       const SCEV *ExactNotTaken;
280       PointerIntPair<ExitNotTakenInfo*, 1> NextExit;
281
282       ExitNotTakenInfo() : ExitingBlock(0), ExactNotTaken(0) {}
283
284       /// isCompleteList - Return true if all loop exits are computable.
285       bool isCompleteList() const {
286         return NextExit.getInt() == 0;
287       }
288
289       void setIncomplete() { NextExit.setInt(1); }
290
291       /// getNextExit - Return a pointer to the next exit's not-taken info.
292       ExitNotTakenInfo *getNextExit() const {
293         return NextExit.getPointer();
294       }
295
296       void setNextExit(ExitNotTakenInfo *ENT) { NextExit.setPointer(ENT); }
297     };
298
299     /// BackedgeTakenInfo - Information about the backedge-taken count
300     /// of a loop. This currently includes an exact count and a maximum count.
301     ///
302     class BackedgeTakenInfo {
303       /// ExitNotTaken - A list of computable exits and their not-taken counts.
304       /// Loops almost never have more than one computable exit.
305       ExitNotTakenInfo ExitNotTaken;
306
307       /// Max - An expression indicating the least maximum backedge-taken
308       /// count of the loop that is known, or a SCEVCouldNotCompute.
309       const SCEV *Max;
310
311     public:
312       BackedgeTakenInfo() : Max(0) {}
313
314       /// Initialize BackedgeTakenInfo from a list of exact exit counts.
315       BackedgeTakenInfo(
316         SmallVectorImpl< std::pair<BasicBlock *, const SCEV *> > &ExitCounts,
317         bool Complete, const SCEV *MaxCount);
318
319       /// hasAnyInfo - Test whether this BackedgeTakenInfo contains any
320       /// computed information, or whether it's all SCEVCouldNotCompute
321       /// values.
322       bool hasAnyInfo() const {
323         return ExitNotTaken.ExitingBlock || !isa<SCEVCouldNotCompute>(Max);
324       }
325
326       /// getExact - Return an expression indicating the exact backedge-taken
327       /// count of the loop if it is known, or SCEVCouldNotCompute
328       /// otherwise. This is the number of times the loop header can be
329       /// guaranteed to execute, minus one.
330       const SCEV *getExact(ScalarEvolution *SE) const;
331
332       /// getExact - Return the number of times this loop exit may fall through
333       /// to the back edge, or SCEVCouldNotCompute. The loop is guaranteed not
334       /// to exit via this block before this number of iterations, but may exit
335       /// via another block.
336       const SCEV *getExact(BasicBlock *ExitingBlock, ScalarEvolution *SE) const;
337
338       /// getMax - Get the max backedge taken count for the loop.
339       const SCEV *getMax(ScalarEvolution *SE) const;
340
341       /// Return true if any backedge taken count expressions refer to the given
342       /// subexpression.
343       bool hasOperand(const SCEV *S, ScalarEvolution *SE) const;
344
345       /// clear - Invalidate this result and free associated memory.
346       void clear();
347     };
348
349     /// BackedgeTakenCounts - Cache the backedge-taken count of the loops for
350     /// this function as they are computed.
351     DenseMap<const Loop*, BackedgeTakenInfo> BackedgeTakenCounts;
352
353     /// ConstantEvolutionLoopExitValue - This map contains entries for all of
354     /// the PHI instructions that we attempt to compute constant evolutions for.
355     /// This allows us to avoid potentially expensive recomputation of these
356     /// properties.  An instruction maps to null if we are unable to compute its
357     /// exit value.
358     DenseMap<PHINode*, Constant*> ConstantEvolutionLoopExitValue;
359
360     /// ValuesAtScopes - This map contains entries for all the expressions
361     /// that we attempt to compute getSCEVAtScope information for, which can
362     /// be expensive in extreme cases.
363     DenseMap<const SCEV *,
364              SmallVector<std::pair<const Loop *, const SCEV *>, 2> > ValuesAtScopes;
365
366     /// LoopDispositions - Memoized computeLoopDisposition results.
367     DenseMap<const SCEV *,
368              SmallVector<std::pair<const Loop *, LoopDisposition>, 2> > LoopDispositions;
369
370     /// computeLoopDisposition - Compute a LoopDisposition value.
371     LoopDisposition computeLoopDisposition(const SCEV *S, const Loop *L);
372
373     /// BlockDispositions - Memoized computeBlockDisposition results.
374     DenseMap<const SCEV *,
375              SmallVector<std::pair<const BasicBlock *, BlockDisposition>, 2> > BlockDispositions;
376
377     /// computeBlockDisposition - Compute a BlockDisposition value.
378     BlockDisposition computeBlockDisposition(const SCEV *S, const BasicBlock *BB);
379
380     /// UnsignedRanges - Memoized results from getUnsignedRange
381     DenseMap<const SCEV *, ConstantRange> UnsignedRanges;
382
383     /// SignedRanges - Memoized results from getSignedRange
384     DenseMap<const SCEV *, ConstantRange> SignedRanges;
385
386     /// setUnsignedRange - Set the memoized unsigned range for the given SCEV.
387     const ConstantRange &setUnsignedRange(const SCEV *S,
388                                           const ConstantRange &CR) {
389       std::pair<DenseMap<const SCEV *, ConstantRange>::iterator, bool> Pair =
390         UnsignedRanges.insert(std::make_pair(S, CR));
391       if (!Pair.second)
392         Pair.first->second = CR;
393       return Pair.first->second;
394     }
395
396     /// setUnsignedRange - Set the memoized signed range for the given SCEV.
397     const ConstantRange &setSignedRange(const SCEV *S,
398                                         const ConstantRange &CR) {
399       std::pair<DenseMap<const SCEV *, ConstantRange>::iterator, bool> Pair =
400         SignedRanges.insert(std::make_pair(S, CR));
401       if (!Pair.second)
402         Pair.first->second = CR;
403       return Pair.first->second;
404     }
405
406     /// createSCEV - We know that there is no SCEV for the specified value.
407     /// Analyze the expression.
408     const SCEV *createSCEV(Value *V);
409
410     /// createNodeForPHI - Provide the special handling we need to analyze PHI
411     /// SCEVs.
412     const SCEV *createNodeForPHI(PHINode *PN);
413
414     /// createNodeForGEP - Provide the special handling we need to analyze GEP
415     /// SCEVs.
416     const SCEV *createNodeForGEP(GEPOperator *GEP);
417
418     /// computeSCEVAtScope - Implementation code for getSCEVAtScope; called
419     /// at most once for each SCEV+Loop pair.
420     ///
421     const SCEV *computeSCEVAtScope(const SCEV *S, const Loop *L);
422
423     /// ForgetSymbolicValue - This looks up computed SCEV values for all
424     /// instructions that depend on the given instruction and removes them from
425     /// the ValueExprMap map if they reference SymName. This is used during PHI
426     /// resolution.
427     void ForgetSymbolicName(Instruction *I, const SCEV *SymName);
428
429     /// getBackedgeTakenInfo - Return the BackedgeTakenInfo for the given
430     /// loop, lazily computing new values if the loop hasn't been analyzed
431     /// yet.
432     const BackedgeTakenInfo &getBackedgeTakenInfo(const Loop *L);
433
434     /// ComputeBackedgeTakenCount - Compute the number of times the specified
435     /// loop will iterate.
436     BackedgeTakenInfo ComputeBackedgeTakenCount(const Loop *L);
437
438     /// ComputeExitLimit - Compute the number of times the backedge of the
439     /// specified loop will execute if it exits via the specified block.
440     ExitLimit ComputeExitLimit(const Loop *L, BasicBlock *ExitingBlock);
441
442     /// ComputeExitLimitFromCond - Compute the number of times the backedge of
443     /// the specified loop will execute if its exit condition were a conditional
444     /// branch of ExitCond, TBB, and FBB.
445     ExitLimit ComputeExitLimitFromCond(const Loop *L,
446                                        Value *ExitCond,
447                                        BasicBlock *TBB,
448                                        BasicBlock *FBB,
449                                        bool IsSubExpr);
450
451     /// ComputeExitLimitFromICmp - Compute the number of times the backedge of
452     /// the specified loop will execute if its exit condition were a conditional
453     /// branch of the ICmpInst ExitCond, TBB, and FBB.
454     ExitLimit ComputeExitLimitFromICmp(const Loop *L,
455                                        ICmpInst *ExitCond,
456                                        BasicBlock *TBB,
457                                        BasicBlock *FBB,
458                                        bool IsSubExpr);
459
460     /// ComputeLoadConstantCompareExitLimit - Given an exit condition
461     /// of 'icmp op load X, cst', try to see if we can compute the
462     /// backedge-taken count.
463     ExitLimit ComputeLoadConstantCompareExitLimit(LoadInst *LI,
464                                                   Constant *RHS,
465                                                   const Loop *L,
466                                                   ICmpInst::Predicate p);
467
468     /// ComputeExitCountExhaustively - If the loop is known to execute a
469     /// constant number of times (the condition evolves only from constants),
470     /// try to evaluate a few iterations of the loop until we get the exit
471     /// condition gets a value of ExitWhen (true or false).  If we cannot
472     /// evaluate the exit count of the loop, return CouldNotCompute.
473     const SCEV *ComputeExitCountExhaustively(const Loop *L,
474                                              Value *Cond,
475                                              bool ExitWhen);
476
477     /// HowFarToZero - Return the number of times an exit condition comparing
478     /// the specified value to zero will execute.  If not computable, return
479     /// CouldNotCompute.
480     ExitLimit HowFarToZero(const SCEV *V, const Loop *L, bool IsSubExpr);
481
482     /// HowFarToNonZero - Return the number of times an exit condition checking
483     /// the specified value for nonzero will execute.  If not computable, return
484     /// CouldNotCompute.
485     ExitLimit HowFarToNonZero(const SCEV *V, const Loop *L);
486
487     /// HowManyLessThans - Return the number of times an exit condition
488     /// containing the specified less-than comparison will execute.  If not
489     /// computable, return CouldNotCompute. isSigned specifies whether the
490     /// less-than is signed.
491     ExitLimit HowManyLessThans(const SCEV *LHS, const SCEV *RHS,
492                                const Loop *L, bool isSigned, bool IsSubExpr);
493     ExitLimit HowManyGreaterThans(const SCEV *LHS, const SCEV *RHS,
494                                   const Loop *L, bool isSigned, bool IsSubExpr);
495
496     /// getPredecessorWithUniqueSuccessorForBB - Return a predecessor of BB
497     /// (which may not be an immediate predecessor) which has exactly one
498     /// successor from which BB is reachable, or null if no such block is
499     /// found.
500     std::pair<BasicBlock *, BasicBlock *>
501     getPredecessorWithUniqueSuccessorForBB(BasicBlock *BB);
502
503     /// isImpliedCond - Test whether the condition described by Pred, LHS, and
504     /// RHS is true whenever the given FoundCondValue value evaluates to true.
505     bool isImpliedCond(ICmpInst::Predicate Pred,
506                        const SCEV *LHS, const SCEV *RHS,
507                        Value *FoundCondValue,
508                        bool Inverse);
509
510     /// isImpliedCondOperands - Test whether the condition described by Pred,
511     /// LHS, and RHS is true whenever the condition described by Pred, FoundLHS,
512     /// and FoundRHS is true.
513     bool isImpliedCondOperands(ICmpInst::Predicate Pred,
514                                const SCEV *LHS, const SCEV *RHS,
515                                const SCEV *FoundLHS, const SCEV *FoundRHS);
516
517     /// isImpliedCondOperandsHelper - Test whether the condition described by
518     /// Pred, LHS, and RHS is true whenever the condition described by Pred,
519     /// FoundLHS, and FoundRHS is true.
520     bool isImpliedCondOperandsHelper(ICmpInst::Predicate Pred,
521                                      const SCEV *LHS, const SCEV *RHS,
522                                      const SCEV *FoundLHS,
523                                      const SCEV *FoundRHS);
524
525     /// getConstantEvolutionLoopExitValue - If we know that the specified Phi is
526     /// in the header of its containing loop, we know the loop executes a
527     /// constant number of times, and the PHI node is just a recurrence
528     /// involving constants, fold it.
529     Constant *getConstantEvolutionLoopExitValue(PHINode *PN, const APInt& BEs,
530                                                 const Loop *L);
531
532     /// isKnownPredicateWithRanges - Test if the given expression is known to
533     /// satisfy the condition described by Pred and the known constant ranges
534     /// of LHS and RHS.
535     ///
536     bool isKnownPredicateWithRanges(ICmpInst::Predicate Pred,
537                                     const SCEV *LHS, const SCEV *RHS);
538
539     /// forgetMemoizedResults - Drop memoized information computed for S.
540     void forgetMemoizedResults(const SCEV *S);
541
542     /// Return false iff given SCEV contains a SCEVUnknown with NULL value-
543     /// pointer.
544     bool checkValidity(const SCEV *S) const;
545
546   public:
547     static char ID; // Pass identification, replacement for typeid
548     ScalarEvolution();
549
550     LLVMContext &getContext() const { return F->getContext(); }
551
552     /// isSCEVable - Test if values of the given type are analyzable within
553     /// the SCEV framework. This primarily includes integer types, and it
554     /// can optionally include pointer types if the ScalarEvolution class
555     /// has access to target-specific information.
556     bool isSCEVable(Type *Ty) const;
557
558     /// getTypeSizeInBits - Return the size in bits of the specified type,
559     /// for which isSCEVable must return true.
560     uint64_t getTypeSizeInBits(Type *Ty) const;
561
562     /// getEffectiveSCEVType - Return a type with the same bitwidth as
563     /// the given type and which represents how SCEV will treat the given
564     /// type, for which isSCEVable must return true. For pointer types,
565     /// this is the pointer-sized integer type.
566     Type *getEffectiveSCEVType(Type *Ty) const;
567
568     /// getSCEV - Return a SCEV expression for the full generality of the
569     /// specified expression.
570     const SCEV *getSCEV(Value *V);
571
572     const SCEV *getConstant(ConstantInt *V);
573     const SCEV *getConstant(const APInt& Val);
574     const SCEV *getConstant(Type *Ty, uint64_t V, bool isSigned = false);
575     const SCEV *getTruncateExpr(const SCEV *Op, Type *Ty);
576     const SCEV *getZeroExtendExpr(const SCEV *Op, Type *Ty);
577     const SCEV *getSignExtendExpr(const SCEV *Op, Type *Ty);
578     const SCEV *getAnyExtendExpr(const SCEV *Op, Type *Ty);
579     const SCEV *getAddExpr(SmallVectorImpl<const SCEV *> &Ops,
580                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
581     const SCEV *getAddExpr(const SCEV *LHS, const SCEV *RHS,
582                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
583       SmallVector<const SCEV *, 2> Ops;
584       Ops.push_back(LHS);
585       Ops.push_back(RHS);
586       return getAddExpr(Ops, Flags);
587     }
588     const SCEV *getAddExpr(const SCEV *Op0, const SCEV *Op1, const SCEV *Op2,
589                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
590       SmallVector<const SCEV *, 3> Ops;
591       Ops.push_back(Op0);
592       Ops.push_back(Op1);
593       Ops.push_back(Op2);
594       return getAddExpr(Ops, Flags);
595     }
596     const SCEV *getMulExpr(SmallVectorImpl<const SCEV *> &Ops,
597                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
598     const SCEV *getMulExpr(const SCEV *LHS, const SCEV *RHS,
599                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap)
600     {
601       SmallVector<const SCEV *, 2> Ops;
602       Ops.push_back(LHS);
603       Ops.push_back(RHS);
604       return getMulExpr(Ops, Flags);
605     }
606     const SCEV *getMulExpr(const SCEV *Op0, const SCEV *Op1, const SCEV *Op2,
607                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
608       SmallVector<const SCEV *, 3> Ops;
609       Ops.push_back(Op0);
610       Ops.push_back(Op1);
611       Ops.push_back(Op2);
612       return getMulExpr(Ops, Flags);
613     }
614     const SCEV *getUDivExpr(const SCEV *LHS, const SCEV *RHS);
615     const SCEV *getAddRecExpr(const SCEV *Start, const SCEV *Step,
616                               const Loop *L, SCEV::NoWrapFlags Flags);
617     const SCEV *getAddRecExpr(SmallVectorImpl<const SCEV *> &Operands,
618                               const Loop *L, SCEV::NoWrapFlags Flags);
619     const SCEV *getAddRecExpr(const SmallVectorImpl<const SCEV *> &Operands,
620                               const Loop *L, SCEV::NoWrapFlags Flags) {
621       SmallVector<const SCEV *, 4> NewOp(Operands.begin(), Operands.end());
622       return getAddRecExpr(NewOp, L, Flags);
623     }
624     const SCEV *getSMaxExpr(const SCEV *LHS, const SCEV *RHS);
625     const SCEV *getSMaxExpr(SmallVectorImpl<const SCEV *> &Operands);
626     const SCEV *getUMaxExpr(const SCEV *LHS, const SCEV *RHS);
627     const SCEV *getUMaxExpr(SmallVectorImpl<const SCEV *> &Operands);
628     const SCEV *getSMinExpr(const SCEV *LHS, const SCEV *RHS);
629     const SCEV *getUMinExpr(const SCEV *LHS, const SCEV *RHS);
630     const SCEV *getUnknown(Value *V);
631     const SCEV *getCouldNotCompute();
632
633     /// getSizeOfExpr - Return an expression for sizeof AllocTy that is type
634     /// IntTy
635     ///
636     const SCEV *getSizeOfExpr(Type *IntTy, Type *AllocTy);
637
638     /// getOffsetOfExpr - Return an expression for offsetof on the given field
639     /// with type IntTy
640     ///
641     const SCEV *getOffsetOfExpr(Type *IntTy, StructType *STy, unsigned FieldNo);
642
643     /// getNegativeSCEV - Return the SCEV object corresponding to -V.
644     ///
645     const SCEV *getNegativeSCEV(const SCEV *V);
646
647     /// getNotSCEV - Return the SCEV object corresponding to ~V.
648     ///
649     const SCEV *getNotSCEV(const SCEV *V);
650
651     /// getMinusSCEV - Return LHS-RHS.  Minus is represented in SCEV as A+B*-1.
652     const SCEV *getMinusSCEV(const SCEV *LHS, const SCEV *RHS,
653                              SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
654
655     /// getTruncateOrZeroExtend - Return a SCEV corresponding to a conversion
656     /// of the input value to the specified type.  If the type must be
657     /// extended, it is zero extended.
658     const SCEV *getTruncateOrZeroExtend(const SCEV *V, Type *Ty);
659
660     /// getTruncateOrSignExtend - Return a SCEV corresponding to a conversion
661     /// of the input value to the specified type.  If the type must be
662     /// extended, it is sign extended.
663     const SCEV *getTruncateOrSignExtend(const SCEV *V, Type *Ty);
664
665     /// getNoopOrZeroExtend - Return a SCEV corresponding to a conversion of
666     /// the input value to the specified type.  If the type must be extended,
667     /// it is zero extended.  The conversion must not be narrowing.
668     const SCEV *getNoopOrZeroExtend(const SCEV *V, Type *Ty);
669
670     /// getNoopOrSignExtend - Return a SCEV corresponding to a conversion of
671     /// the input value to the specified type.  If the type must be extended,
672     /// it is sign extended.  The conversion must not be narrowing.
673     const SCEV *getNoopOrSignExtend(const SCEV *V, Type *Ty);
674
675     /// getNoopOrAnyExtend - Return a SCEV corresponding to a conversion of
676     /// the input value to the specified type. If the type must be extended,
677     /// it is extended with unspecified bits. The conversion must not be
678     /// narrowing.
679     const SCEV *getNoopOrAnyExtend(const SCEV *V, Type *Ty);
680
681     /// getTruncateOrNoop - Return a SCEV corresponding to a conversion of the
682     /// input value to the specified type.  The conversion must not be
683     /// widening.
684     const SCEV *getTruncateOrNoop(const SCEV *V, Type *Ty);
685
686     /// getUMaxFromMismatchedTypes - Promote the operands to the wider of
687     /// the types using zero-extension, and then perform a umax operation
688     /// with them.
689     const SCEV *getUMaxFromMismatchedTypes(const SCEV *LHS,
690                                            const SCEV *RHS);
691
692     /// getUMinFromMismatchedTypes - Promote the operands to the wider of
693     /// the types using zero-extension, and then perform a umin operation
694     /// with them.
695     const SCEV *getUMinFromMismatchedTypes(const SCEV *LHS,
696                                            const SCEV *RHS);
697
698     /// getPointerBase - Transitively follow the chain of pointer-type operands
699     /// until reaching a SCEV that does not have a single pointer operand. This
700     /// returns a SCEVUnknown pointer for well-formed pointer-type expressions,
701     /// but corner cases do exist.
702     const SCEV *getPointerBase(const SCEV *V);
703
704     /// getSCEVAtScope - Return a SCEV expression for the specified value
705     /// at the specified scope in the program.  The L value specifies a loop
706     /// nest to evaluate the expression at, where null is the top-level or a
707     /// specified loop is immediately inside of the loop.
708     ///
709     /// This method can be used to compute the exit value for a variable defined
710     /// in a loop by querying what the value will hold in the parent loop.
711     ///
712     /// In the case that a relevant loop exit value cannot be computed, the
713     /// original value V is returned.
714     const SCEV *getSCEVAtScope(const SCEV *S, const Loop *L);
715
716     /// getSCEVAtScope - This is a convenience function which does
717     /// getSCEVAtScope(getSCEV(V), L).
718     const SCEV *getSCEVAtScope(Value *V, const Loop *L);
719
720     /// isLoopEntryGuardedByCond - Test whether entry to the loop is protected
721     /// by a conditional between LHS and RHS.  This is used to help avoid max
722     /// expressions in loop trip counts, and to eliminate casts.
723     bool isLoopEntryGuardedByCond(const Loop *L, ICmpInst::Predicate Pred,
724                                   const SCEV *LHS, const SCEV *RHS);
725
726     /// isLoopBackedgeGuardedByCond - Test whether the backedge of the loop is
727     /// protected by a conditional between LHS and RHS.  This is used to
728     /// to eliminate casts.
729     bool isLoopBackedgeGuardedByCond(const Loop *L, ICmpInst::Predicate Pred,
730                                      const SCEV *LHS, const SCEV *RHS);
731
732     /// getSmallConstantTripCount - Returns the maximum trip count of this loop
733     /// as a normal unsigned value. Returns 0 if the trip count is unknown or
734     /// not constant. This "trip count" assumes that control exits via
735     /// ExitingBlock. More precisely, it is the number of times that control may
736     /// reach ExitingBlock before taking the branch. For loops with multiple
737     /// exits, it may not be the number times that the loop header executes if
738     /// the loop exits prematurely via another branch.
739     unsigned getSmallConstantTripCount(Loop *L, BasicBlock *ExitingBlock);
740
741     /// getSmallConstantTripMultiple - Returns the largest constant divisor of
742     /// the trip count of this loop as a normal unsigned value, if
743     /// possible. This means that the actual trip count is always a multiple of
744     /// the returned value (don't forget the trip count could very well be zero
745     /// as well!). As explained in the comments for getSmallConstantTripCount,
746     /// this assumes that control exits the loop via ExitingBlock.
747     unsigned getSmallConstantTripMultiple(Loop *L, BasicBlock *ExitingBlock);
748
749     // getExitCount - Get the expression for the number of loop iterations for
750     // which this loop is guaranteed not to exit via ExitingBlock. Otherwise
751     // return SCEVCouldNotCompute.
752     const SCEV *getExitCount(Loop *L, BasicBlock *ExitingBlock);
753
754     /// getBackedgeTakenCount - If the specified loop has a predictable
755     /// backedge-taken count, return it, otherwise return a SCEVCouldNotCompute
756     /// object. The backedge-taken count is the number of times the loop header
757     /// will be branched to from within the loop. This is one less than the
758     /// trip count of the loop, since it doesn't count the first iteration,
759     /// when the header is branched to from outside the loop.
760     ///
761     /// Note that it is not valid to call this method on a loop without a
762     /// loop-invariant backedge-taken count (see
763     /// hasLoopInvariantBackedgeTakenCount).
764     ///
765     const SCEV *getBackedgeTakenCount(const Loop *L);
766
767     /// getMaxBackedgeTakenCount - Similar to getBackedgeTakenCount, except
768     /// return the least SCEV value that is known never to be less than the
769     /// actual backedge taken count.
770     const SCEV *getMaxBackedgeTakenCount(const Loop *L);
771
772     /// hasLoopInvariantBackedgeTakenCount - Return true if the specified loop
773     /// has an analyzable loop-invariant backedge-taken count.
774     bool hasLoopInvariantBackedgeTakenCount(const Loop *L);
775
776     /// forgetLoop - This method should be called by the client when it has
777     /// changed a loop in a way that may effect ScalarEvolution's ability to
778     /// compute a trip count, or if the loop is deleted.
779     void forgetLoop(const Loop *L);
780
781     /// forgetValue - This method should be called by the client when it has
782     /// changed a value in a way that may effect its value, or which may
783     /// disconnect it from a def-use chain linking it to a loop.
784     void forgetValue(Value *V);
785
786     /// GetMinTrailingZeros - Determine the minimum number of zero bits that S
787     /// is guaranteed to end in (at every loop iteration).  It is, at the same
788     /// time, the minimum number of times S is divisible by 2.  For example,
789     /// given {4,+,8} it returns 2.  If S is guaranteed to be 0, it returns the
790     /// bitwidth of S.
791     uint32_t GetMinTrailingZeros(const SCEV *S);
792
793     /// getUnsignedRange - Determine the unsigned range for a particular SCEV.
794     ///
795     ConstantRange getUnsignedRange(const SCEV *S);
796
797     /// getSignedRange - Determine the signed range for a particular SCEV.
798     ///
799     ConstantRange getSignedRange(const SCEV *S);
800
801     /// isKnownNegative - Test if the given expression is known to be negative.
802     ///
803     bool isKnownNegative(const SCEV *S);
804
805     /// isKnownPositive - Test if the given expression is known to be positive.
806     ///
807     bool isKnownPositive(const SCEV *S);
808
809     /// isKnownNonNegative - Test if the given expression is known to be
810     /// non-negative.
811     ///
812     bool isKnownNonNegative(const SCEV *S);
813
814     /// isKnownNonPositive - Test if the given expression is known to be
815     /// non-positive.
816     ///
817     bool isKnownNonPositive(const SCEV *S);
818
819     /// isKnownNonZero - Test if the given expression is known to be
820     /// non-zero.
821     ///
822     bool isKnownNonZero(const SCEV *S);
823
824     /// isKnownPredicate - Test if the given expression is known to satisfy
825     /// the condition described by Pred, LHS, and RHS.
826     ///
827     bool isKnownPredicate(ICmpInst::Predicate Pred,
828                           const SCEV *LHS, const SCEV *RHS);
829
830     /// SimplifyICmpOperands - Simplify LHS and RHS in a comparison with
831     /// predicate Pred. Return true iff any changes were made. If the
832     /// operands are provably equal or unequal, LHS and RHS are set to
833     /// the same value and Pred is set to either ICMP_EQ or ICMP_NE.
834     ///
835     bool SimplifyICmpOperands(ICmpInst::Predicate &Pred,
836                               const SCEV *&LHS,
837                               const SCEV *&RHS,
838                               unsigned Depth = 0);
839
840     /// getLoopDisposition - Return the "disposition" of the given SCEV with
841     /// respect to the given loop.
842     LoopDisposition getLoopDisposition(const SCEV *S, const Loop *L);
843
844     /// isLoopInvariant - Return true if the value of the given SCEV is
845     /// unchanging in the specified loop.
846     bool isLoopInvariant(const SCEV *S, const Loop *L);
847
848     /// hasComputableLoopEvolution - Return true if the given SCEV changes value
849     /// in a known way in the specified loop.  This property being true implies
850     /// that the value is variant in the loop AND that we can emit an expression
851     /// to compute the value of the expression at any particular loop iteration.
852     bool hasComputableLoopEvolution(const SCEV *S, const Loop *L);
853
854     /// getLoopDisposition - Return the "disposition" of the given SCEV with
855     /// respect to the given block.
856     BlockDisposition getBlockDisposition(const SCEV *S, const BasicBlock *BB);
857
858     /// dominates - Return true if elements that makes up the given SCEV
859     /// dominate the specified basic block.
860     bool dominates(const SCEV *S, const BasicBlock *BB);
861
862     /// properlyDominates - Return true if elements that makes up the given SCEV
863     /// properly dominate the specified basic block.
864     bool properlyDominates(const SCEV *S, const BasicBlock *BB);
865
866     /// hasOperand - Test whether the given SCEV has Op as a direct or
867     /// indirect operand.
868     bool hasOperand(const SCEV *S, const SCEV *Op) const;
869
870     virtual bool runOnFunction(Function &F);
871     virtual void releaseMemory();
872     virtual void getAnalysisUsage(AnalysisUsage &AU) const;
873     virtual void print(raw_ostream &OS, const Module* = 0) const;
874     virtual void verifyAnalysis() const;
875
876   private:
877     /// Compute the backedge taken count knowing the interval difference, the
878     /// stride and presence of the equality in the comparison.
879     const SCEV *computeBECount(const SCEV *Delta, const SCEV *Stride,
880                                bool Equality);
881
882     /// Verify if an linear IV with positive stride can overflow when in a 
883     /// less-than comparison, knowing the invariant term of the comparison,
884     /// the stride and the knowledge of NSW/NUW flags on the recurrence.
885     bool doesIVOverflowOnLT(const SCEV *RHS, const SCEV *Stride,
886                             bool IsSigned, bool NoWrap);
887
888     /// Verify if an linear IV with negative stride can overflow when in a 
889     /// greater-than comparison, knowing the invariant term of the comparison,
890     /// the stride and the knowledge of NSW/NUW flags on the recurrence.
891     bool doesIVOverflowOnGT(const SCEV *RHS, const SCEV *Stride,
892                             bool IsSigned, bool NoWrap);
893
894   private:
895     FoldingSet<SCEV> UniqueSCEVs;
896     BumpPtrAllocator SCEVAllocator;
897
898     /// FirstUnknown - The head of a linked list of all SCEVUnknown
899     /// values that have been allocated. This is used by releaseMemory
900     /// to locate them all and call their destructors.
901     SCEVUnknown *FirstUnknown;
902   };
903 }
904
905 #endif