2006-08-06 Paolo Carlini <pcarlini@suse.de>
[official-gcc.git] / gcc / tree-chrec.c
blobf36fc9b0291848f38f5ddf9399c7f9a39481431f
1 /* Chains of recurrences.
2 Copyright (C) 2003, 2004, 2005, 2006 Free Software Foundation, Inc.
3 Contributed by Sebastian Pop <pop@cri.ensmp.fr>
5 This file is part of GCC.
7 GCC is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 2, or (at your option) any later
10 version.
12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
15 for more details.
17 You should have received a copy of the GNU General Public License
18 along with GCC; see the file COPYING. If not, write to the Free
19 Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA
20 02110-1301, USA. */
22 /* This file implements operations on chains of recurrences. Chains
23 of recurrences are used for modeling evolution functions of scalar
24 variables.
27 #include "config.h"
28 #include "system.h"
29 #include "coretypes.h"
30 #include "tm.h"
31 #include "ggc.h"
32 #include "tree.h"
33 #include "real.h"
34 #include "diagnostic.h"
35 #include "cfgloop.h"
36 #include "tree-flow.h"
37 #include "tree-chrec.h"
38 #include "tree-pass.h"
39 #include "params.h"
40 #include "tree-scalar-evolution.h"
44 /* Extended folder for chrecs. */
46 /* Determines whether CST is not a constant evolution. */
48 static inline bool
49 is_not_constant_evolution (tree cst)
51 return (TREE_CODE (cst) == POLYNOMIAL_CHREC);
54 /* Fold CODE for a polynomial function and a constant. */
56 static inline tree
57 chrec_fold_poly_cst (enum tree_code code,
58 tree type,
59 tree poly,
60 tree cst)
62 gcc_assert (poly);
63 gcc_assert (cst);
64 gcc_assert (TREE_CODE (poly) == POLYNOMIAL_CHREC);
65 gcc_assert (!is_not_constant_evolution (cst));
66 gcc_assert (type == chrec_type (poly));
68 switch (code)
70 case PLUS_EXPR:
71 return build_polynomial_chrec
72 (CHREC_VARIABLE (poly),
73 chrec_fold_plus (type, CHREC_LEFT (poly), cst),
74 CHREC_RIGHT (poly));
76 case MINUS_EXPR:
77 return build_polynomial_chrec
78 (CHREC_VARIABLE (poly),
79 chrec_fold_minus (type, CHREC_LEFT (poly), cst),
80 CHREC_RIGHT (poly));
82 case MULT_EXPR:
83 return build_polynomial_chrec
84 (CHREC_VARIABLE (poly),
85 chrec_fold_multiply (type, CHREC_LEFT (poly), cst),
86 chrec_fold_multiply (type, CHREC_RIGHT (poly), cst));
88 default:
89 return chrec_dont_know;
93 /* Fold the addition of two polynomial functions. */
95 static inline tree
96 chrec_fold_plus_poly_poly (enum tree_code code,
97 tree type,
98 tree poly0,
99 tree poly1)
101 tree left, right;
103 gcc_assert (poly0);
104 gcc_assert (poly1);
105 gcc_assert (TREE_CODE (poly0) == POLYNOMIAL_CHREC);
106 gcc_assert (TREE_CODE (poly1) == POLYNOMIAL_CHREC);
107 gcc_assert (chrec_type (poly0) == chrec_type (poly1));
108 gcc_assert (type == chrec_type (poly0));
111 {a, +, b}_1 + {c, +, d}_2 -> {{a, +, b}_1 + c, +, d}_2,
112 {a, +, b}_2 + {c, +, d}_1 -> {{c, +, d}_1 + a, +, b}_2,
113 {a, +, b}_x + {c, +, d}_x -> {a+c, +, b+d}_x. */
114 if (CHREC_VARIABLE (poly0) < CHREC_VARIABLE (poly1))
116 if (code == PLUS_EXPR)
117 return build_polynomial_chrec
118 (CHREC_VARIABLE (poly1),
119 chrec_fold_plus (type, poly0, CHREC_LEFT (poly1)),
120 CHREC_RIGHT (poly1));
121 else
122 return build_polynomial_chrec
123 (CHREC_VARIABLE (poly1),
124 chrec_fold_minus (type, poly0, CHREC_LEFT (poly1)),
125 chrec_fold_multiply (type, CHREC_RIGHT (poly1),
126 SCALAR_FLOAT_TYPE_P (type)
127 ? build_real (type, dconstm1)
128 : build_int_cst_type (type, -1)));
131 if (CHREC_VARIABLE (poly0) > CHREC_VARIABLE (poly1))
133 if (code == PLUS_EXPR)
134 return build_polynomial_chrec
135 (CHREC_VARIABLE (poly0),
136 chrec_fold_plus (type, CHREC_LEFT (poly0), poly1),
137 CHREC_RIGHT (poly0));
138 else
139 return build_polynomial_chrec
140 (CHREC_VARIABLE (poly0),
141 chrec_fold_minus (type, CHREC_LEFT (poly0), poly1),
142 CHREC_RIGHT (poly0));
145 if (code == PLUS_EXPR)
147 left = chrec_fold_plus
148 (type, CHREC_LEFT (poly0), CHREC_LEFT (poly1));
149 right = chrec_fold_plus
150 (type, CHREC_RIGHT (poly0), CHREC_RIGHT (poly1));
152 else
154 left = chrec_fold_minus
155 (type, CHREC_LEFT (poly0), CHREC_LEFT (poly1));
156 right = chrec_fold_minus
157 (type, CHREC_RIGHT (poly0), CHREC_RIGHT (poly1));
160 if (chrec_zerop (right))
161 return left;
162 else
163 return build_polynomial_chrec
164 (CHREC_VARIABLE (poly0), left, right);
169 /* Fold the multiplication of two polynomial functions. */
171 static inline tree
172 chrec_fold_multiply_poly_poly (tree type,
173 tree poly0,
174 tree poly1)
176 tree t0, t1, t2;
177 int var;
179 gcc_assert (poly0);
180 gcc_assert (poly1);
181 gcc_assert (TREE_CODE (poly0) == POLYNOMIAL_CHREC);
182 gcc_assert (TREE_CODE (poly1) == POLYNOMIAL_CHREC);
183 gcc_assert (chrec_type (poly0) == chrec_type (poly1));
184 gcc_assert (type == chrec_type (poly0));
186 /* {a, +, b}_1 * {c, +, d}_2 -> {c*{a, +, b}_1, +, d}_2,
187 {a, +, b}_2 * {c, +, d}_1 -> {a*{c, +, d}_1, +, b}_2,
188 {a, +, b}_x * {c, +, d}_x -> {a*c, +, a*d + b*c + b*d, +, 2*b*d}_x. */
189 if (CHREC_VARIABLE (poly0) < CHREC_VARIABLE (poly1))
190 /* poly0 is a constant wrt. poly1. */
191 return build_polynomial_chrec
192 (CHREC_VARIABLE (poly1),
193 chrec_fold_multiply (type, CHREC_LEFT (poly1), poly0),
194 CHREC_RIGHT (poly1));
196 if (CHREC_VARIABLE (poly1) < CHREC_VARIABLE (poly0))
197 /* poly1 is a constant wrt. poly0. */
198 return build_polynomial_chrec
199 (CHREC_VARIABLE (poly0),
200 chrec_fold_multiply (type, CHREC_LEFT (poly0), poly1),
201 CHREC_RIGHT (poly0));
203 /* poly0 and poly1 are two polynomials in the same variable,
204 {a, +, b}_x * {c, +, d}_x -> {a*c, +, a*d + b*c + b*d, +, 2*b*d}_x. */
206 /* "a*c". */
207 t0 = chrec_fold_multiply (type, CHREC_LEFT (poly0), CHREC_LEFT (poly1));
209 /* "a*d + b*c + b*d". */
210 t1 = chrec_fold_multiply (type, CHREC_LEFT (poly0), CHREC_RIGHT (poly1));
211 t1 = chrec_fold_plus (type, t1, chrec_fold_multiply (type,
212 CHREC_RIGHT (poly0),
213 CHREC_LEFT (poly1)));
214 t1 = chrec_fold_plus (type, t1, chrec_fold_multiply (type,
215 CHREC_RIGHT (poly0),
216 CHREC_RIGHT (poly1)));
217 /* "2*b*d". */
218 t2 = chrec_fold_multiply (type, CHREC_RIGHT (poly0), CHREC_RIGHT (poly1));
219 t2 = chrec_fold_multiply (type, SCALAR_FLOAT_TYPE_P (type)
220 ? build_real (type, dconst2)
221 : build_int_cst (type, 2), t2);
223 var = CHREC_VARIABLE (poly0);
224 return build_polynomial_chrec (var, t0,
225 build_polynomial_chrec (var, t1, t2));
228 /* When the operands are automatically_generated_chrec_p, the fold has
229 to respect the semantics of the operands. */
231 static inline tree
232 chrec_fold_automatically_generated_operands (tree op0,
233 tree op1)
235 if (op0 == chrec_dont_know
236 || op1 == chrec_dont_know)
237 return chrec_dont_know;
239 if (op0 == chrec_known
240 || op1 == chrec_known)
241 return chrec_known;
243 if (op0 == chrec_not_analyzed_yet
244 || op1 == chrec_not_analyzed_yet)
245 return chrec_not_analyzed_yet;
247 /* The default case produces a safe result. */
248 return chrec_dont_know;
251 /* Fold the addition of two chrecs. */
253 static tree
254 chrec_fold_plus_1 (enum tree_code code, tree type,
255 tree op0, tree op1)
257 if (automatically_generated_chrec_p (op0)
258 || automatically_generated_chrec_p (op1))
259 return chrec_fold_automatically_generated_operands (op0, op1);
261 switch (TREE_CODE (op0))
263 case POLYNOMIAL_CHREC:
264 switch (TREE_CODE (op1))
266 case POLYNOMIAL_CHREC:
267 return chrec_fold_plus_poly_poly (code, type, op0, op1);
269 default:
270 if (code == PLUS_EXPR)
271 return build_polynomial_chrec
272 (CHREC_VARIABLE (op0),
273 chrec_fold_plus (type, CHREC_LEFT (op0), op1),
274 CHREC_RIGHT (op0));
275 else
276 return build_polynomial_chrec
277 (CHREC_VARIABLE (op0),
278 chrec_fold_minus (type, CHREC_LEFT (op0), op1),
279 CHREC_RIGHT (op0));
282 default:
283 switch (TREE_CODE (op1))
285 case POLYNOMIAL_CHREC:
286 if (code == PLUS_EXPR)
287 return build_polynomial_chrec
288 (CHREC_VARIABLE (op1),
289 chrec_fold_plus (type, op0, CHREC_LEFT (op1)),
290 CHREC_RIGHT (op1));
291 else
292 return build_polynomial_chrec
293 (CHREC_VARIABLE (op1),
294 chrec_fold_minus (type, op0, CHREC_LEFT (op1)),
295 chrec_fold_multiply (type, CHREC_RIGHT (op1),
296 SCALAR_FLOAT_TYPE_P (type)
297 ? build_real (type, dconstm1)
298 : build_int_cst_type (type, -1)));
300 default:
302 int size = 0;
303 if ((tree_contains_chrecs (op0, &size)
304 || tree_contains_chrecs (op1, &size))
305 && size < PARAM_VALUE (PARAM_SCEV_MAX_EXPR_SIZE))
306 return build2 (code, type, op0, op1);
307 else if (size < PARAM_VALUE (PARAM_SCEV_MAX_EXPR_SIZE))
308 return fold_build2 (code, type,
309 fold_convert (type, op0),
310 fold_convert (type, op1));
311 else
312 return chrec_dont_know;
318 /* Fold the addition of two chrecs. */
320 tree
321 chrec_fold_plus (tree type,
322 tree op0,
323 tree op1)
325 if (automatically_generated_chrec_p (op0)
326 || automatically_generated_chrec_p (op1))
327 return chrec_fold_automatically_generated_operands (op0, op1);
329 if (integer_zerop (op0))
330 return op1;
331 if (integer_zerop (op1))
332 return op0;
334 return chrec_fold_plus_1 (PLUS_EXPR, type, op0, op1);
337 /* Fold the subtraction of two chrecs. */
339 tree
340 chrec_fold_minus (tree type,
341 tree op0,
342 tree op1)
344 if (automatically_generated_chrec_p (op0)
345 || automatically_generated_chrec_p (op1))
346 return chrec_fold_automatically_generated_operands (op0, op1);
348 if (integer_zerop (op1))
349 return op0;
351 return chrec_fold_plus_1 (MINUS_EXPR, type, op0, op1);
354 /* Fold the multiplication of two chrecs. */
356 tree
357 chrec_fold_multiply (tree type,
358 tree op0,
359 tree op1)
361 if (automatically_generated_chrec_p (op0)
362 || automatically_generated_chrec_p (op1))
363 return chrec_fold_automatically_generated_operands (op0, op1);
365 switch (TREE_CODE (op0))
367 case POLYNOMIAL_CHREC:
368 switch (TREE_CODE (op1))
370 case POLYNOMIAL_CHREC:
371 return chrec_fold_multiply_poly_poly (type, op0, op1);
373 default:
374 if (integer_onep (op1))
375 return op0;
376 if (integer_zerop (op1))
377 return build_int_cst (type, 0);
379 return build_polynomial_chrec
380 (CHREC_VARIABLE (op0),
381 chrec_fold_multiply (type, CHREC_LEFT (op0), op1),
382 chrec_fold_multiply (type, CHREC_RIGHT (op0), op1));
385 default:
386 if (integer_onep (op0))
387 return op1;
389 if (integer_zerop (op0))
390 return build_int_cst (type, 0);
392 switch (TREE_CODE (op1))
394 case POLYNOMIAL_CHREC:
395 return build_polynomial_chrec
396 (CHREC_VARIABLE (op1),
397 chrec_fold_multiply (type, CHREC_LEFT (op1), op0),
398 chrec_fold_multiply (type, CHREC_RIGHT (op1), op0));
400 default:
401 if (integer_onep (op1))
402 return op0;
403 if (integer_zerop (op1))
404 return build_int_cst (type, 0);
405 return fold_build2 (MULT_EXPR, type, op0, op1);
412 /* Operations. */
414 /* Evaluate the binomial coefficient. Return NULL_TREE if the intermediate
415 calculation overflows, otherwise return C(n,k) with type TYPE. */
417 static tree
418 tree_fold_binomial (tree type, tree n, unsigned int k)
420 unsigned HOST_WIDE_INT lidx, lnum, ldenom, lres, ldum;
421 HOST_WIDE_INT hidx, hnum, hdenom, hres, hdum;
422 unsigned int i;
423 tree res;
425 /* Handle the most frequent cases. */
426 if (k == 0)
427 return build_int_cst (type, 1);
428 if (k == 1)
429 return fold_convert (type, n);
431 /* Check that k <= n. */
432 if (TREE_INT_CST_HIGH (n) == 0
433 && TREE_INT_CST_LOW (n) < k)
434 return NULL_TREE;
436 /* Numerator = n. */
437 lnum = TREE_INT_CST_LOW (n);
438 hnum = TREE_INT_CST_HIGH (n);
440 /* Denominator = 2. */
441 ldenom = 2;
442 hdenom = 0;
444 /* Index = Numerator-1. */
445 if (lnum == 0)
447 hidx = hnum - 1;
448 lidx = ~ (unsigned HOST_WIDE_INT) 0;
450 else
452 hidx = hnum;
453 lidx = lnum - 1;
456 /* Numerator = Numerator*Index = n*(n-1). */
457 if (mul_double (lnum, hnum, lidx, hidx, &lnum, &hnum))
458 return NULL_TREE;
460 for (i = 3; i <= k; i++)
462 /* Index--. */
463 if (lidx == 0)
465 hidx--;
466 lidx = ~ (unsigned HOST_WIDE_INT) 0;
468 else
469 lidx--;
471 /* Numerator *= Index. */
472 if (mul_double (lnum, hnum, lidx, hidx, &lnum, &hnum))
473 return NULL_TREE;
475 /* Denominator *= i. */
476 mul_double (ldenom, hdenom, i, 0, &ldenom, &hdenom);
479 /* Result = Numerator / Denominator. */
480 div_and_round_double (EXACT_DIV_EXPR, 1, lnum, hnum, ldenom, hdenom,
481 &lres, &hres, &ldum, &hdum);
483 res = build_int_cst_wide (type, lres, hres);
484 return int_fits_type_p (res, type) ? res : NULL_TREE;
487 /* Helper function. Use the Newton's interpolating formula for
488 evaluating the value of the evolution function. */
490 static tree
491 chrec_evaluate (unsigned var, tree chrec, tree n, unsigned int k)
493 tree arg0, arg1, binomial_n_k;
494 tree type = TREE_TYPE (chrec);
496 while (TREE_CODE (chrec) == POLYNOMIAL_CHREC
497 && CHREC_VARIABLE (chrec) > var)
498 chrec = CHREC_LEFT (chrec);
500 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC
501 && CHREC_VARIABLE (chrec) == var)
503 arg0 = chrec_evaluate (var, CHREC_RIGHT (chrec), n, k + 1);
504 if (arg0 == chrec_dont_know)
505 return chrec_dont_know;
506 binomial_n_k = tree_fold_binomial (type, n, k);
507 if (!binomial_n_k)
508 return chrec_dont_know;
509 arg1 = fold_build2 (MULT_EXPR, type,
510 CHREC_LEFT (chrec), binomial_n_k);
511 return chrec_fold_plus (type, arg0, arg1);
514 binomial_n_k = tree_fold_binomial (type, n, k);
515 if (!binomial_n_k)
516 return chrec_dont_know;
518 return fold_build2 (MULT_EXPR, type, chrec, binomial_n_k);
521 /* Evaluates "CHREC (X)" when the varying variable is VAR.
522 Example: Given the following parameters,
524 var = 1
525 chrec = {3, +, 4}_1
526 x = 10
528 The result is given by the Newton's interpolating formula:
529 3 * \binom{10}{0} + 4 * \binom{10}{1}.
532 tree
533 chrec_apply (unsigned var,
534 tree chrec,
535 tree x)
537 tree type = chrec_type (chrec);
538 tree res = chrec_dont_know;
540 if (automatically_generated_chrec_p (chrec)
541 || automatically_generated_chrec_p (x)
543 /* When the symbols are defined in an outer loop, it is possible
544 to symbolically compute the apply, since the symbols are
545 constants with respect to the varying loop. */
546 || chrec_contains_symbols_defined_in_loop (chrec, var))
547 return chrec_dont_know;
549 if (dump_file && (dump_flags & TDF_DETAILS))
550 fprintf (dump_file, "(chrec_apply \n");
552 if (TREE_CODE (x) == INTEGER_CST && SCALAR_FLOAT_TYPE_P (type))
553 x = build_real_from_int_cst (type, x);
555 if (evolution_function_is_affine_p (chrec))
557 /* "{a, +, b} (x)" -> "a + b*x". */
558 x = chrec_convert (type, x, NULL_TREE);
559 res = chrec_fold_multiply (type, CHREC_RIGHT (chrec), x);
560 if (!integer_zerop (CHREC_LEFT (chrec)))
561 res = chrec_fold_plus (type, CHREC_LEFT (chrec), res);
564 else if (TREE_CODE (chrec) != POLYNOMIAL_CHREC)
565 res = chrec;
567 else if (TREE_CODE (x) == INTEGER_CST
568 && tree_int_cst_sgn (x) == 1)
569 /* testsuite/.../ssa-chrec-38.c. */
570 res = chrec_evaluate (var, chrec, x, 0);
571 else
572 res = chrec_dont_know;
574 if (dump_file && (dump_flags & TDF_DETAILS))
576 fprintf (dump_file, " (varying_loop = %d\n", var);
577 fprintf (dump_file, ")\n (chrec = ");
578 print_generic_expr (dump_file, chrec, 0);
579 fprintf (dump_file, ")\n (x = ");
580 print_generic_expr (dump_file, x, 0);
581 fprintf (dump_file, ")\n (res = ");
582 print_generic_expr (dump_file, res, 0);
583 fprintf (dump_file, "))\n");
586 return res;
589 /* Replaces the initial condition in CHREC with INIT_COND. */
591 tree
592 chrec_replace_initial_condition (tree chrec,
593 tree init_cond)
595 if (automatically_generated_chrec_p (chrec))
596 return chrec;
598 gcc_assert (chrec_type (chrec) == chrec_type (init_cond));
600 switch (TREE_CODE (chrec))
602 case POLYNOMIAL_CHREC:
603 return build_polynomial_chrec
604 (CHREC_VARIABLE (chrec),
605 chrec_replace_initial_condition (CHREC_LEFT (chrec), init_cond),
606 CHREC_RIGHT (chrec));
608 default:
609 return init_cond;
613 /* Returns the initial condition of a given CHREC. */
615 tree
616 initial_condition (tree chrec)
618 if (automatically_generated_chrec_p (chrec))
619 return chrec;
621 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC)
622 return initial_condition (CHREC_LEFT (chrec));
623 else
624 return chrec;
627 /* Returns a univariate function that represents the evolution in
628 LOOP_NUM. Mask the evolution of any other loop. */
630 tree
631 hide_evolution_in_other_loops_than_loop (tree chrec,
632 unsigned loop_num)
634 if (automatically_generated_chrec_p (chrec))
635 return chrec;
637 switch (TREE_CODE (chrec))
639 case POLYNOMIAL_CHREC:
640 if (CHREC_VARIABLE (chrec) == loop_num)
641 return build_polynomial_chrec
642 (loop_num,
643 hide_evolution_in_other_loops_than_loop (CHREC_LEFT (chrec),
644 loop_num),
645 CHREC_RIGHT (chrec));
647 else if (CHREC_VARIABLE (chrec) < loop_num)
648 /* There is no evolution in this loop. */
649 return initial_condition (chrec);
651 else
652 return hide_evolution_in_other_loops_than_loop (CHREC_LEFT (chrec),
653 loop_num);
655 default:
656 return chrec;
660 /* Returns the evolution part of CHREC in LOOP_NUM when RIGHT is
661 true, otherwise returns the initial condition in LOOP_NUM. */
663 static tree
664 chrec_component_in_loop_num (tree chrec,
665 unsigned loop_num,
666 bool right)
668 tree component;
670 if (automatically_generated_chrec_p (chrec))
671 return chrec;
673 switch (TREE_CODE (chrec))
675 case POLYNOMIAL_CHREC:
676 if (CHREC_VARIABLE (chrec) == loop_num)
678 if (right)
679 component = CHREC_RIGHT (chrec);
680 else
681 component = CHREC_LEFT (chrec);
683 if (TREE_CODE (CHREC_LEFT (chrec)) != POLYNOMIAL_CHREC
684 || CHREC_VARIABLE (CHREC_LEFT (chrec)) != CHREC_VARIABLE (chrec))
685 return component;
687 else
688 return build_polynomial_chrec
689 (loop_num,
690 chrec_component_in_loop_num (CHREC_LEFT (chrec),
691 loop_num,
692 right),
693 component);
696 else if (CHREC_VARIABLE (chrec) < loop_num)
697 /* There is no evolution part in this loop. */
698 return NULL_TREE;
700 else
701 return chrec_component_in_loop_num (CHREC_LEFT (chrec),
702 loop_num,
703 right);
705 default:
706 if (right)
707 return NULL_TREE;
708 else
709 return chrec;
713 /* Returns the evolution part in LOOP_NUM. Example: the call
714 evolution_part_in_loop_num ({{0, +, 1}_1, +, 2}_1, 1) returns
715 {1, +, 2}_1 */
717 tree
718 evolution_part_in_loop_num (tree chrec,
719 unsigned loop_num)
721 return chrec_component_in_loop_num (chrec, loop_num, true);
724 /* Returns the initial condition in LOOP_NUM. Example: the call
725 initial_condition_in_loop_num ({{0, +, 1}_1, +, 2}_2, 2) returns
726 {0, +, 1}_1 */
728 tree
729 initial_condition_in_loop_num (tree chrec,
730 unsigned loop_num)
732 return chrec_component_in_loop_num (chrec, loop_num, false);
735 /* Set or reset the evolution of CHREC to NEW_EVOL in loop LOOP_NUM.
736 This function is essentially used for setting the evolution to
737 chrec_dont_know, for example after having determined that it is
738 impossible to say how many times a loop will execute. */
740 tree
741 reset_evolution_in_loop (unsigned loop_num,
742 tree chrec,
743 tree new_evol)
745 gcc_assert (chrec_type (chrec) == chrec_type (new_evol));
747 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC
748 && CHREC_VARIABLE (chrec) > loop_num)
750 tree left = reset_evolution_in_loop (loop_num, CHREC_LEFT (chrec),
751 new_evol);
752 tree right = reset_evolution_in_loop (loop_num, CHREC_RIGHT (chrec),
753 new_evol);
754 return build3 (POLYNOMIAL_CHREC, TREE_TYPE (left),
755 build_int_cst (NULL_TREE, CHREC_VARIABLE (chrec)),
756 left, right);
759 while (TREE_CODE (chrec) == POLYNOMIAL_CHREC
760 && CHREC_VARIABLE (chrec) == loop_num)
761 chrec = CHREC_LEFT (chrec);
763 return build_polynomial_chrec (loop_num, chrec, new_evol);
766 /* Merges two evolution functions that were found by following two
767 alternate paths of a conditional expression. */
769 tree
770 chrec_merge (tree chrec1,
771 tree chrec2)
773 if (chrec1 == chrec_dont_know
774 || chrec2 == chrec_dont_know)
775 return chrec_dont_know;
777 if (chrec1 == chrec_known
778 || chrec2 == chrec_known)
779 return chrec_known;
781 if (chrec1 == chrec_not_analyzed_yet)
782 return chrec2;
783 if (chrec2 == chrec_not_analyzed_yet)
784 return chrec1;
786 if (eq_evolutions_p (chrec1, chrec2))
787 return chrec1;
789 return chrec_dont_know;
794 /* Observers. */
796 /* Helper function for is_multivariate_chrec. */
798 static bool
799 is_multivariate_chrec_rec (tree chrec, unsigned int rec_var)
801 if (chrec == NULL_TREE)
802 return false;
804 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC)
806 if (CHREC_VARIABLE (chrec) != rec_var)
807 return true;
808 else
809 return (is_multivariate_chrec_rec (CHREC_LEFT (chrec), rec_var)
810 || is_multivariate_chrec_rec (CHREC_RIGHT (chrec), rec_var));
812 else
813 return false;
816 /* Determine whether the given chrec is multivariate or not. */
818 bool
819 is_multivariate_chrec (tree chrec)
821 if (chrec == NULL_TREE)
822 return false;
824 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC)
825 return (is_multivariate_chrec_rec (CHREC_LEFT (chrec),
826 CHREC_VARIABLE (chrec))
827 || is_multivariate_chrec_rec (CHREC_RIGHT (chrec),
828 CHREC_VARIABLE (chrec)));
829 else
830 return false;
833 /* Determines whether the chrec contains symbolic names or not. */
835 bool
836 chrec_contains_symbols (tree chrec)
838 if (chrec == NULL_TREE)
839 return false;
841 if (TREE_CODE (chrec) == SSA_NAME
842 || TREE_CODE (chrec) == VAR_DECL
843 || TREE_CODE (chrec) == PARM_DECL
844 || TREE_CODE (chrec) == FUNCTION_DECL
845 || TREE_CODE (chrec) == LABEL_DECL
846 || TREE_CODE (chrec) == RESULT_DECL
847 || TREE_CODE (chrec) == FIELD_DECL)
848 return true;
850 switch (TREE_CODE_LENGTH (TREE_CODE (chrec)))
852 case 3:
853 if (chrec_contains_symbols (TREE_OPERAND (chrec, 2)))
854 return true;
856 case 2:
857 if (chrec_contains_symbols (TREE_OPERAND (chrec, 1)))
858 return true;
860 case 1:
861 if (chrec_contains_symbols (TREE_OPERAND (chrec, 0)))
862 return true;
864 default:
865 return false;
869 /* Determines whether the chrec contains undetermined coefficients. */
871 bool
872 chrec_contains_undetermined (tree chrec)
874 if (chrec == chrec_dont_know
875 || chrec == chrec_not_analyzed_yet
876 || chrec == NULL_TREE)
877 return true;
879 switch (TREE_CODE_LENGTH (TREE_CODE (chrec)))
881 case 3:
882 if (chrec_contains_undetermined (TREE_OPERAND (chrec, 2)))
883 return true;
885 case 2:
886 if (chrec_contains_undetermined (TREE_OPERAND (chrec, 1)))
887 return true;
889 case 1:
890 if (chrec_contains_undetermined (TREE_OPERAND (chrec, 0)))
891 return true;
893 default:
894 return false;
898 /* Determines whether the tree EXPR contains chrecs, and increment
899 SIZE if it is not a NULL pointer by an estimation of the depth of
900 the tree. */
902 bool
903 tree_contains_chrecs (tree expr, int *size)
905 if (expr == NULL_TREE)
906 return false;
908 if (size)
909 (*size)++;
911 if (tree_is_chrec (expr))
912 return true;
914 switch (TREE_CODE_LENGTH (TREE_CODE (expr)))
916 case 3:
917 if (tree_contains_chrecs (TREE_OPERAND (expr, 2), size))
918 return true;
920 case 2:
921 if (tree_contains_chrecs (TREE_OPERAND (expr, 1), size))
922 return true;
924 case 1:
925 if (tree_contains_chrecs (TREE_OPERAND (expr, 0), size))
926 return true;
928 default:
929 return false;
933 /* Recursive helper function. */
935 static bool
936 evolution_function_is_invariant_rec_p (tree chrec, int loopnum)
938 if (evolution_function_is_constant_p (chrec))
939 return true;
941 if (TREE_CODE (chrec) == SSA_NAME
942 && expr_invariant_in_loop_p (current_loops->parray[loopnum],
943 chrec))
944 return true;
946 if (TREE_CODE (chrec) == POLYNOMIAL_CHREC)
948 if (CHREC_VARIABLE (chrec) == (unsigned) loopnum
949 || !evolution_function_is_invariant_rec_p (CHREC_RIGHT (chrec),
950 loopnum)
951 || !evolution_function_is_invariant_rec_p (CHREC_LEFT (chrec),
952 loopnum))
953 return false;
954 return true;
957 switch (TREE_CODE_LENGTH (TREE_CODE (chrec)))
959 case 2:
960 if (!evolution_function_is_invariant_rec_p (TREE_OPERAND (chrec, 1),
961 loopnum))
962 return false;
964 case 1:
965 if (!evolution_function_is_invariant_rec_p (TREE_OPERAND (chrec, 0),
966 loopnum))
967 return false;
968 return true;
970 default:
971 return false;
974 return false;
977 /* Return true if CHREC is invariant in loop LOOPNUM, false otherwise. */
979 bool
980 evolution_function_is_invariant_p (tree chrec, int loopnum)
982 if (evolution_function_is_constant_p (chrec))
983 return true;
985 if (current_loops != NULL)
986 return evolution_function_is_invariant_rec_p (chrec, loopnum);
988 return false;
991 /* Determine whether the given tree is an affine multivariate
992 evolution. */
994 bool
995 evolution_function_is_affine_multivariate_p (tree chrec)
997 if (chrec == NULL_TREE)
998 return false;
1000 switch (TREE_CODE (chrec))
1002 case POLYNOMIAL_CHREC:
1003 if (evolution_function_is_constant_p (CHREC_LEFT (chrec)))
1005 if (evolution_function_is_constant_p (CHREC_RIGHT (chrec)))
1006 return true;
1007 else
1009 if (TREE_CODE (CHREC_RIGHT (chrec)) == POLYNOMIAL_CHREC
1010 && CHREC_VARIABLE (CHREC_RIGHT (chrec))
1011 != CHREC_VARIABLE (chrec)
1012 && evolution_function_is_affine_multivariate_p
1013 (CHREC_RIGHT (chrec)))
1014 return true;
1015 else
1016 return false;
1019 else
1021 if (evolution_function_is_constant_p (CHREC_RIGHT (chrec))
1022 && TREE_CODE (CHREC_LEFT (chrec)) == POLYNOMIAL_CHREC
1023 && CHREC_VARIABLE (CHREC_LEFT (chrec)) != CHREC_VARIABLE (chrec)
1024 && evolution_function_is_affine_multivariate_p
1025 (CHREC_LEFT (chrec)))
1026 return true;
1027 else
1028 return false;
1031 default:
1032 return false;
1036 /* Determine whether the given tree is a function in zero or one
1037 variables. */
1039 bool
1040 evolution_function_is_univariate_p (tree chrec)
1042 if (chrec == NULL_TREE)
1043 return true;
1045 switch (TREE_CODE (chrec))
1047 case POLYNOMIAL_CHREC:
1048 switch (TREE_CODE (CHREC_LEFT (chrec)))
1050 case POLYNOMIAL_CHREC:
1051 if (CHREC_VARIABLE (chrec) != CHREC_VARIABLE (CHREC_LEFT (chrec)))
1052 return false;
1053 if (!evolution_function_is_univariate_p (CHREC_LEFT (chrec)))
1054 return false;
1055 break;
1057 default:
1058 break;
1061 switch (TREE_CODE (CHREC_RIGHT (chrec)))
1063 case POLYNOMIAL_CHREC:
1064 if (CHREC_VARIABLE (chrec) != CHREC_VARIABLE (CHREC_RIGHT (chrec)))
1065 return false;
1066 if (!evolution_function_is_univariate_p (CHREC_RIGHT (chrec)))
1067 return false;
1068 break;
1070 default:
1071 break;
1074 default:
1075 return true;
1079 /* Returns the number of variables of CHREC. Example: the call
1080 nb_vars_in_chrec ({{0, +, 1}_5, +, 2}_6) returns 2. */
1082 unsigned
1083 nb_vars_in_chrec (tree chrec)
1085 if (chrec == NULL_TREE)
1086 return 0;
1088 switch (TREE_CODE (chrec))
1090 case POLYNOMIAL_CHREC:
1091 return 1 + nb_vars_in_chrec
1092 (initial_condition_in_loop_num (chrec, CHREC_VARIABLE (chrec)));
1094 default:
1095 return 0;
1099 /* Returns true if TYPE is a type in that we cannot directly perform
1100 arithmetics, even though it is a scalar type. */
1102 static bool
1103 avoid_arithmetics_in_type_p (tree type)
1105 /* Ada frontend uses subtypes -- an arithmetic cannot be directly performed
1106 in the subtype, but a base type must be used, and the result then can
1107 be casted to the subtype. */
1108 if (TREE_CODE (type) == INTEGER_TYPE && TREE_TYPE (type) != NULL_TREE)
1109 return true;
1111 return false;
1114 static tree chrec_convert_1 (tree, tree, tree, bool);
1116 /* Converts BASE and STEP of affine scev to TYPE. LOOP is the loop whose iv
1117 the scev corresponds to. AT_STMT is the statement at that the scev is
1118 evaluated. USE_OVERFLOW_SEMANTICS is true if this function should assume that
1119 the rules for overflow of the given language apply (e.g., that signed
1120 arithmetics in C does not overflow) -- i.e., to use them to avoid unnecessary
1121 tests, but also to enforce that the result follows them. Returns true if the
1122 conversion succeeded, false otherwise. */
1124 bool
1125 convert_affine_scev (struct loop *loop, tree type,
1126 tree *base, tree *step, tree at_stmt,
1127 bool use_overflow_semantics)
1129 tree ct = TREE_TYPE (*step);
1130 bool enforce_overflow_semantics;
1131 bool must_check_src_overflow, must_check_rslt_overflow;
1132 tree new_base, new_step;
1134 /* If we cannot perform arithmetic in TYPE, avoid creating an scev. */
1135 if (avoid_arithmetics_in_type_p (type))
1136 return false;
1138 /* In general,
1139 (TYPE) (BASE + STEP * i) = (TYPE) BASE + (TYPE -- sign extend) STEP * i,
1140 but we must check some assumptions.
1142 1) If [BASE, +, STEP] wraps, the equation is not valid when precision
1143 of CT is smaller than the precision of TYPE. For example, when we
1144 cast unsigned char [254, +, 1] to unsigned, the values on left side
1145 are 254, 255, 0, 1, ..., but those on the right side are
1146 254, 255, 256, 257, ...
1147 2) In case that we must also preserve the fact that signed ivs do not
1148 overflow, we must additionally check that the new iv does not wrap.
1149 For example, unsigned char [125, +, 1] casted to signed char could
1150 become a wrapping variable with values 125, 126, 127, -128, -127, ...,
1151 which would confuse optimizers that assume that this does not
1152 happen. */
1153 must_check_src_overflow = TYPE_PRECISION (ct) < TYPE_PRECISION (type);
1155 enforce_overflow_semantics = (use_overflow_semantics
1156 && nowrap_type_p (type));
1157 if (enforce_overflow_semantics)
1159 /* We can avoid checking whether the result overflows in the following
1160 cases:
1162 -- must_check_src_overflow is true, and the range of TYPE is superset
1163 of the range of CT -- i.e., in all cases except if CT signed and
1164 TYPE unsigned.
1165 -- both CT and TYPE have the same precision and signedness. */
1166 if (must_check_src_overflow)
1168 if (TYPE_UNSIGNED (type) && !TYPE_UNSIGNED (ct))
1169 must_check_rslt_overflow = true;
1170 else
1171 must_check_rslt_overflow = false;
1173 else if (TYPE_UNSIGNED (ct) == TYPE_UNSIGNED (type)
1174 && TYPE_PRECISION (ct) == TYPE_PRECISION (type))
1175 must_check_rslt_overflow = false;
1176 else
1177 must_check_rslt_overflow = true;
1179 else
1180 must_check_rslt_overflow = false;
1182 if (must_check_src_overflow
1183 && scev_probably_wraps_p (*base, *step, at_stmt, loop,
1184 use_overflow_semantics))
1185 return false;
1187 new_base = chrec_convert_1 (type, *base, at_stmt,
1188 use_overflow_semantics);
1189 /* The step must be sign extended, regardless of the signedness
1190 of CT and TYPE. This only needs to be handled specially when
1191 CT is unsigned -- to avoid e.g. unsigned char [100, +, 255]
1192 (with values 100, 99, 98, ...) from becoming signed or unsigned
1193 [100, +, 255] with values 100, 355, ...; the sign-extension is
1194 performed by default when CT is signed. */
1195 new_step = *step;
1196 if (TYPE_PRECISION (type) > TYPE_PRECISION (ct) && TYPE_UNSIGNED (ct))
1197 new_step = chrec_convert_1 (signed_type_for (ct), new_step, at_stmt,
1198 use_overflow_semantics);
1199 new_step = chrec_convert_1 (type, new_step, at_stmt, use_overflow_semantics);
1201 if (automatically_generated_chrec_p (new_base)
1202 || automatically_generated_chrec_p (new_step))
1203 return false;
1205 if (must_check_rslt_overflow
1206 /* Note that in this case we cannot use the fact that signed variables
1207 do not overflow, as this is what we are verifying for the new iv. */
1208 && scev_probably_wraps_p (new_base, new_step, at_stmt, loop, false))
1209 return false;
1211 *base = new_base;
1212 *step = new_step;
1213 return true;
1217 /* Convert CHREC to TYPE. When the analyzer knows the context in
1218 which the CHREC is built, it sets AT_STMT to the statement that
1219 contains the definition of the analyzed variable, otherwise the
1220 conversion is less accurate: the information is used for
1221 determining a more accurate estimation of the number of iterations.
1222 By default AT_STMT could be safely set to NULL_TREE.
1224 The following rule is always true: TREE_TYPE (chrec) ==
1225 TREE_TYPE (CHREC_LEFT (chrec)) == TREE_TYPE (CHREC_RIGHT (chrec)).
1226 An example of what could happen when adding two chrecs and the type
1227 of the CHREC_RIGHT is different than CHREC_LEFT is:
1229 {(uint) 0, +, (uchar) 10} +
1230 {(uint) 0, +, (uchar) 250}
1232 that would produce a wrong result if CHREC_RIGHT is not (uint):
1234 {(uint) 0, +, (uchar) 4}
1236 instead of
1238 {(uint) 0, +, (uint) 260}
1241 tree
1242 chrec_convert (tree type, tree chrec, tree at_stmt)
1244 return chrec_convert_1 (type, chrec, at_stmt, true);
1247 /* Convert CHREC to TYPE. When the analyzer knows the context in
1248 which the CHREC is built, it sets AT_STMT to the statement that
1249 contains the definition of the analyzed variable, otherwise the
1250 conversion is less accurate: the information is used for
1251 determining a more accurate estimation of the number of iterations.
1252 By default AT_STMT could be safely set to NULL_TREE.
1254 USE_OVERFLOW_SEMANTICS is true if this function should assume that
1255 the rules for overflow of the given language apply (e.g., that signed
1256 arithmetics in C does not overflow) -- i.e., to use them to avoid unnecessary
1257 tests, but also to enforce that the result follows them. */
1259 static tree
1260 chrec_convert_1 (tree type, tree chrec, tree at_stmt,
1261 bool use_overflow_semantics)
1263 tree ct, res;
1264 tree base, step;
1265 struct loop *loop;
1267 if (automatically_generated_chrec_p (chrec))
1268 return chrec;
1270 ct = chrec_type (chrec);
1271 if (ct == type)
1272 return chrec;
1274 if (!evolution_function_is_affine_p (chrec))
1275 goto keep_cast;
1277 loop = current_loops->parray[CHREC_VARIABLE (chrec)];
1278 base = CHREC_LEFT (chrec);
1279 step = CHREC_RIGHT (chrec);
1281 if (convert_affine_scev (loop, type, &base, &step, at_stmt,
1282 use_overflow_semantics))
1283 return build_polynomial_chrec (loop->num, base, step);
1285 /* If we cannot propagate the cast inside the chrec, just keep the cast. */
1286 keep_cast:
1287 res = fold_convert (type, chrec);
1289 /* Don't propagate overflows. */
1290 if (CONSTANT_CLASS_P (res))
1292 TREE_CONSTANT_OVERFLOW (res) = 0;
1293 TREE_OVERFLOW (res) = 0;
1296 /* But reject constants that don't fit in their type after conversion.
1297 This can happen if TYPE_MIN_VALUE or TYPE_MAX_VALUE are not the
1298 natural values associated with TYPE_PRECISION and TYPE_UNSIGNED,
1299 and can cause problems later when computing niters of loops. Note
1300 that we don't do the check before converting because we don't want
1301 to reject conversions of negative chrecs to unsigned types. */
1302 if (TREE_CODE (res) == INTEGER_CST
1303 && TREE_CODE (type) == INTEGER_TYPE
1304 && !int_fits_type_p (res, type))
1305 res = chrec_dont_know;
1307 return res;
1310 /* Convert CHREC to TYPE, without regard to signed overflows. Returns the new
1311 chrec if something else than what chrec_convert would do happens, NULL_TREE
1312 otherwise. */
1314 tree
1315 chrec_convert_aggressive (tree type, tree chrec)
1317 tree inner_type, left, right, lc, rc;
1319 if (automatically_generated_chrec_p (chrec)
1320 || TREE_CODE (chrec) != POLYNOMIAL_CHREC)
1321 return NULL_TREE;
1323 inner_type = TREE_TYPE (chrec);
1324 if (TYPE_PRECISION (type) > TYPE_PRECISION (inner_type))
1325 return NULL_TREE;
1327 /* If we cannot perform arithmetic in TYPE, avoid creating an scev. */
1328 if (avoid_arithmetics_in_type_p (type))
1329 return NULL_TREE;
1331 left = CHREC_LEFT (chrec);
1332 right = CHREC_RIGHT (chrec);
1333 lc = chrec_convert_aggressive (type, left);
1334 if (!lc)
1335 lc = chrec_convert (type, left, NULL_TREE);
1336 rc = chrec_convert_aggressive (type, right);
1337 if (!rc)
1338 rc = chrec_convert (type, right, NULL_TREE);
1340 return build_polynomial_chrec (CHREC_VARIABLE (chrec), lc, rc);
1343 /* Returns true when CHREC0 == CHREC1. */
1345 bool
1346 eq_evolutions_p (tree chrec0,
1347 tree chrec1)
1349 if (chrec0 == NULL_TREE
1350 || chrec1 == NULL_TREE
1351 || TREE_CODE (chrec0) != TREE_CODE (chrec1))
1352 return false;
1354 if (chrec0 == chrec1)
1355 return true;
1357 switch (TREE_CODE (chrec0))
1359 case INTEGER_CST:
1360 return operand_equal_p (chrec0, chrec1, 0);
1362 case POLYNOMIAL_CHREC:
1363 return (CHREC_VARIABLE (chrec0) == CHREC_VARIABLE (chrec1)
1364 && eq_evolutions_p (CHREC_LEFT (chrec0), CHREC_LEFT (chrec1))
1365 && eq_evolutions_p (CHREC_RIGHT (chrec0), CHREC_RIGHT (chrec1)));
1366 default:
1367 return false;
1371 /* Returns EV_GROWS if CHREC grows (assuming that it does not overflow),
1372 EV_DECREASES if it decreases, and EV_UNKNOWN if we cannot determine
1373 which of these cases happens. */
1375 enum ev_direction
1376 scev_direction (tree chrec)
1378 tree step;
1380 if (!evolution_function_is_affine_p (chrec))
1381 return EV_DIR_UNKNOWN;
1383 step = CHREC_RIGHT (chrec);
1384 if (TREE_CODE (step) != INTEGER_CST)
1385 return EV_DIR_UNKNOWN;
1387 if (tree_int_cst_sign_bit (step))
1388 return EV_DIR_DECREASES;
1389 else
1390 return EV_DIR_GROWS;