diff options
Diffstat (limited to 'gcc/tree-chrec.c')
| -rw-r--r-- | gcc/tree-chrec.c | 60 |
1 files changed, 48 insertions, 12 deletions
diff --git a/gcc/tree-chrec.c b/gcc/tree-chrec.c index 47419c2c3d7..2d48093c0e2 100644 --- a/gcc/tree-chrec.c +++ b/gcc/tree-chrec.c @@ -583,14 +583,16 @@ hide_evolution_in_other_loops_than_loop (tree chrec, } } -/* Returns the evolution part in LOOP_NUM. Example: the call - get_evolution_in_loop (1, {{0, +, 1}_1, +, 2}_1) returns - {1, +, 2}_1 */ +/* Returns the evolution part of CHREC in LOOP_NUM when RIGHT is + true, otherwise returns the initial condition in LOOP_NUM. */ -tree -evolution_part_in_loop_num (tree chrec, - unsigned loop_num) +static tree +chrec_component_in_loop_num (tree chrec, + unsigned loop_num, + bool right) { + tree component; + if (automatically_generated_chrec_p (chrec)) return chrec; @@ -599,15 +601,22 @@ evolution_part_in_loop_num (tree chrec, case POLYNOMIAL_CHREC: if (CHREC_VARIABLE (chrec) == loop_num) { + if (right) + component = CHREC_RIGHT (chrec); + else + component = CHREC_LEFT (chrec); + if (TREE_CODE (CHREC_LEFT (chrec)) != POLYNOMIAL_CHREC || CHREC_VARIABLE (CHREC_LEFT (chrec)) != CHREC_VARIABLE (chrec)) - return CHREC_RIGHT (chrec); + return component; else return build_polynomial_chrec (loop_num, - evolution_part_in_loop_num (CHREC_LEFT (chrec), loop_num), - CHREC_RIGHT (chrec)); + chrec_component_in_loop_num (CHREC_LEFT (chrec), + loop_num, + right), + component); } else if (CHREC_VARIABLE (chrec) < loop_num) @@ -615,13 +624,40 @@ evolution_part_in_loop_num (tree chrec, return NULL_TREE; else - return evolution_part_in_loop_num (CHREC_LEFT (chrec), loop_num); + return chrec_component_in_loop_num (CHREC_LEFT (chrec), + loop_num, + right); - default: - return NULL_TREE; + default: + if (right) + return NULL_TREE; + else + return chrec; } } +/* Returns the evolution part in LOOP_NUM. Example: the call + evolution_part_in_loop_num (1, {{0, +, 1}_1, +, 2}_1) returns + {1, +, 2}_1 */ + +tree +evolution_part_in_loop_num (tree chrec, + unsigned loop_num) +{ + return chrec_component_in_loop_num (chrec, loop_num, true); +} + +/* Returns the initial condition in LOOP_NUM. Example: the call + initial_condition_in_loop_num ({{0, +, 1}_1, +, 2}_2, 1) returns + {0, +, 1}_1 */ + +tree +initial_condition_in_loop_num (tree chrec, + unsigned loop_num) +{ + return chrec_component_in_loop_num (chrec, loop_num, false); +} + /* Set or reset the evolution of CHREC to NEW_EVOL in loop LOOP_NUM. This function is essentially used for setting the evolution to chrec_dont_know, for example after having determined that it is |