source/isl/domain.c: fix typo in comment
[cloog.git] / source / polylib / constraintset.c
blob26cde7dd1e141f15c8f81506c55f243e4661a7d9
2 /**-------------------------------------------------------------------**
3 ** CLooG **
4 **-------------------------------------------------------------------**
5 ** constraintset.c **
6 **-------------------------------------------------------------------**
7 ** First version: april 17th 2005 **
8 **-------------------------------------------------------------------**/
11 /******************************************************************************
12 * CLooG : the Chunky Loop Generator (experimental) *
13 ******************************************************************************
14 * *
15 * Copyright (C) 2005 Cedric Bastoul *
16 * *
17 * This is free software; you can redistribute it and/or modify it under the *
18 * terms of the GNU General Public License as published by the Free Software *
19 * Foundation; either version 2 of the License, or (at your option) any later *
20 * version. *
21 * *
22 * This software is distributed in the hope that it will be useful, but *
23 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY *
24 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License *
25 * for more details. *
26 * *
27 * You should have received a copy of the GNU General Public License along *
28 * with software; if not, write to the Free Software Foundation, Inc., *
29 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *
30 * *
31 * CLooG, the Chunky Loop Generator *
32 * Written by Cedric Bastoul, Cedric.Bastoul@inria.fr *
33 * *
34 ******************************************************************************/
35 /* CAUTION: the english used for comments is probably the worst you ever read,
36 * please feel free to correct and improve it !
40 # include <stdlib.h>
41 # include <stdio.h>
42 # include <ctype.h>
43 #include <cloog/polylib/cloog.h>
46 #define ALLOC(type) (type*)malloc(sizeof(type))
47 #define ALLOCN(type,n) (type*)malloc((n)*sizeof(type))
50 /******************************************************************************
51 * PolyLib interface *
52 ******************************************************************************/
55 /**
56 * CLooG makes an intensive use of matrix operations and the PolyLib do
57 * the job. Here are the interfaces to all the PolyLib calls (CLooG uses 18
58 * PolyLib functions), with or without some adaptations. If another matrix
59 * library can be used, only these functions have to be changed.
63 /**
64 * cloog_polylib_matrix_print function:
65 * This function prints the content of a Matrix structure (matrix) into a
66 * file (foo, possibly stdout).
68 void cloog_polylib_matrix_print(FILE * foo, Matrix * matrix)
69 { Matrix_Print(foo,P_VALUE_FMT,matrix) ;
73 /**
74 * cloog_polylib_matrix_free function:
75 * This function frees the allocated memory for a Matrix structure
76 * (matrix).
78 void cloog_polylib_matrix_free(Matrix * matrix)
80 Matrix_Free(matrix) ;
83 void cloog_constraint_set_free(CloogConstraintSet *constraints)
85 cloog_polylib_matrix_free(constraints);
88 int cloog_constraint_set_contains_level(CloogConstraintSet *constraints,
89 int level, int nb_parameters)
91 return constraints->NbColumns - 2 - nb_parameters >= level;
94 /* Check if the variable at position level is defined by an
95 * equality. If so, return the row number. Otherwise, return -1.
97 * If there is an equality, we can print it directly -no ambiguity-.
98 * PolyLib can give more than one equality, we use just the first one
99 * (this is a PolyLib problem, but all equalities are equivalent).
101 CloogConstraint cloog_constraint_set_defining_equality(CloogConstraintSet *matrix, int level)
103 CloogConstraint constraint;
104 int i;
106 constraint.set = matrix;
107 for (i = 0; i < matrix->NbRows; i++)
108 if (value_zero_p(matrix->p[i][0]) &&
109 value_notzero_p(matrix->p[i][level])) {
110 constraint.line = &matrix->p[i];
111 return constraint;
113 return cloog_constraint_invalid();
116 /* Check if two vectors are eachothers opposite.
117 * Return 1 if they are, 0 if they are not.
119 static int Vector_Opposite(Value *p1, Value *p2, unsigned len)
121 int i;
123 for (i = 0; i < len; ++i) {
124 if (value_abs_ne(p1[i], p2[i]))
125 return 0;
126 if (value_zero_p(p1[i]))
127 continue;
128 if (value_eq(p1[i], p2[i]))
129 return 0;
131 return 1;
134 /* Check if the variable (e) at position level is defined by a
135 * pair of inequalities
136 * <a, i> + -m e + <b, p> + k1 >= 0
137 * <-a, i> + m e + <-b, p> + k2 >= 0
138 * with 0 <= k1 + k2 < m
139 * If so return the row number of the upper bound and set *lower
140 * to the row number of the lower bound. If not, return -1.
142 * If the variable at position level occurs in any other constraint,
143 * then we currently return -1. The modulo guard that we would generate
144 * would still be correct, but we would also need to generate
145 * guards corresponding to the other constraints, and this has not
146 * been implemented yet.
148 CloogConstraint cloog_constraint_set_defining_inequalities(CloogConstraintSet *matrix,
149 int level, CloogConstraint *lower, int nb_par)
151 int i, j, k;
152 Value m;
153 unsigned len = matrix->NbColumns - 2;
154 unsigned nb_iter = len - nb_par;
155 CloogConstraint constraint;
157 constraint.set = matrix;
158 lower->set = matrix;
159 for (i = 0; i < matrix->NbRows; i++) {
160 if (value_zero_p(matrix->p[i][level]))
161 continue;
162 if (value_zero_p(matrix->p[i][0]))
163 return cloog_constraint_invalid();
164 if (value_one_p(matrix->p[i][level]))
165 return cloog_constraint_invalid();
166 if (value_mone_p(matrix->p[i][level]))
167 return cloog_constraint_invalid();
168 if (First_Non_Zero(matrix->p[i]+level+1,
169 (1+nb_iter)-(level+1)) != -1)
170 return cloog_constraint_invalid();
171 for (j = i+1; j < matrix->NbRows; ++j) {
172 if (value_zero_p(matrix->p[j][level]))
173 continue;
174 if (value_zero_p(matrix->p[j][0]))
175 return cloog_constraint_invalid();
176 if (value_one_p(matrix->p[j][level]))
177 return cloog_constraint_invalid();
178 if (value_mone_p(matrix->p[j][level]))
179 return cloog_constraint_invalid();
180 if (First_Non_Zero(matrix->p[j]+level+1,
181 (1+nb_iter)-(level+1)) != -1)
182 return cloog_constraint_invalid();
184 value_init(m);
185 value_addto(m, matrix->p[i][1+len], matrix->p[j][1+len]);
186 if (value_neg_p(m) ||
187 value_abs_ge(m, matrix->p[i][level])) {
188 value_clear(m);
189 return cloog_constraint_invalid();
191 value_clear(m);
193 if (!Vector_Opposite(matrix->p[i]+1, matrix->p[j]+1,
194 len))
195 return cloog_constraint_invalid();
196 for (k = j+1; k < matrix->NbRows; ++k)
197 if (value_notzero_p(matrix->p[k][level]))
198 return cloog_constraint_invalid();
199 if (value_pos_p(matrix->p[i][level])) {
200 lower->line = &matrix->p[i];
201 constraint.line = &matrix->p[j];
202 } else {
203 lower->line = &matrix->p[j];
204 constraint.line = &matrix->p[i];
206 return constraint;
209 return cloog_constraint_invalid();
212 int cloog_constraint_set_total_dimension(CloogConstraintSet *constraints)
214 return constraints->NbColumns - 2;
217 int cloog_constraint_set_n_iterators(CloogConstraintSet *constraint, int nb_par)
219 return cloog_constraint_set_total_dimension(constraint) - nb_par;
222 int cloog_equal_total_dimension(CloogEqualities *equal)
224 return cloog_constraint_set_total_dimension(equal->constraints);
227 int cloog_constraint_total_dimension(CloogConstraint constraint)
229 return cloog_constraint_set_total_dimension(constraint.set);
233 * cloog_polylib_matrix_alloc function:
234 * This function allocates the memory space for a Matrix structure having
235 * nb_rows rows and nb_columns columns, it set its elements to 0.
237 Matrix * cloog_polylib_matrix_alloc(unsigned nb_rows, unsigned nb_columns)
239 return Matrix_Alloc(nb_rows,nb_columns) ;
244 * cloog_polylib_matrix_matrix function:
245 * This function converts a PolyLib Matrix to a Matrix structure.
247 Matrix * cloog_polylib_matrix_matrix(Matrix *matrix)
249 return matrix;
253 /******************************************************************************
254 * Structure display function *
255 ******************************************************************************/
259 * cloog_loop_print_structure function:
260 * Displays a Matrix structure (matrix) into a file (file, possibly stdout)
261 * in a way that trends to be understandable without falling in a deep
262 * depression or, for the lucky ones, getting a headache... It includes an
263 * indentation level (level) in order to work with others print_structure
264 * functions. Written by Olivier Chorier, Luc Marchaud, Pierre Martin and
265 * Romain Tartiere.
266 * - April 24th 2005: Initial version.
267 * - June 2nd 2005: (Ced) Extraction from cloog_loop_print_structure and
268 * integration in matrix.c.
269 * - June 22rd 2005: Adaptation for GMP.
271 void cloog_polylib_matrix_print_structure(FILE * file, Matrix * matrix, int level)
272 { int i, j ;
274 /* Display the matrix. */
275 for (i=0; i<matrix->NbRows; i++)
276 { for(j=0; j<=level; j++)
277 fprintf(file,"|\t") ;
279 fprintf(file,"[ ") ;
281 for (j=0; j<matrix->NbColumns; j++)
282 { value_print(file,P_VALUE_FMT,matrix->p[i][j]) ;
283 fprintf(file," ") ;
286 fprintf(file,"]\n") ;
291 /******************************************************************************
292 * Reading function *
293 ******************************************************************************/
297 * cloog_polylib_matrix_read function:
298 * Adaptation from the PolyLib. This function reads a matrix into a file (foo,
299 * posibly stdin) and returns a pointer this matrix.
300 * October 18th 2001: first version.
301 * - April 17th 2005: this function moved from domain.c to here.
302 * - June 21rd 2005: Adaptation for GMP (based on S. Verdoolaege's version of
303 * CLooG 0.12.1).
305 Matrix * cloog_polylib_matrix_read(FILE * foo)
306 { unsigned NbRows, NbColumns ;
307 int i, j, n ;
308 char *c, s[MAX_STRING], str[MAX_STRING] ;
309 Matrix * matrix ;
310 Value * p ;
312 while (fgets(s,MAX_STRING,foo) == 0) ;
313 while ((*s=='#' || *s=='\n') || (sscanf(s," %u %u",&NbRows,&NbColumns)<2))
314 fgets(s,MAX_STRING,foo) ;
316 matrix = cloog_polylib_matrix_alloc(NbRows,NbColumns) ;
318 p = matrix->p_Init ;
319 for (i=0;i<matrix->NbRows;i++)
320 { do
321 { c = fgets(s,MAX_STRING,foo) ;
322 while ((c != NULL) && isspace(*c) && (*c != '\n'))
323 c++ ;
325 while (c != NULL && (*c == '#' || *c == '\n'));
327 if (c == NULL)
328 cloog_die("not enough rows.\n");
329 for (j=0;j<matrix->NbColumns;j++)
330 { if (c == NULL || *c == '#' || *c == '\n')
331 cloog_die("not enough columns.\n");
332 /* NdCed : Dans le n ca met strlen(str). */
333 if (sscanf(c,"%s%n",str,&n) == 0)
334 cloog_die("not enough rows.\n");
335 value_read(*(p++),str) ;
336 c += n ;
340 return(matrix) ;
344 /******************************************************************************
345 * Equalities spreading functions *
346 ******************************************************************************/
349 /* Equalities are stored inside a Matrix data structure called "equal".
350 * This matrix has (nb_scattering + nb_iterators + 1) rows (i.e. total
351 * dimensions + 1, the "+ 1" is because a statement can be included inside an
352 * external loop without iteration domain), and (nb_scattering + nb_iterators +
353 * nb_parameters + 2) columns (all unknowns plus the scalar plus the equality
354 * type). The ith row corresponds to the equality "= 0" for the ith dimension
355 * iterator. The first column gives the equality type (0: no equality, then
356 * EQTYPE_* -see pprint.h-). At each recursion of pprint, if an equality for
357 * the current level is found, the corresponding row is updated. Then the
358 * equality if it exists is used to simplify expressions (e.g. if we have
359 * "i+1" while we know that "i=2", we simplify it in "3"). At the end of
360 * the pprint call, the corresponding row is reset to zero.
363 CloogEqualities *cloog_equal_alloc(int n, int nb_levels,
364 int nb_parameters)
366 int i;
367 CloogEqualities *equal = ALLOC(CloogEqualities);
369 equal->constraints = cloog_polylib_matrix_alloc(n, nb_levels + nb_parameters + 1);
370 equal->types = ALLOCN(int, n);
371 for (i = 0; i < n; ++i)
372 equal->types[i] = EQTYPE_NONE;
373 return equal;
376 void cloog_equal_free(CloogEqualities *equal)
378 cloog_polylib_matrix_free(equal->constraints);
379 free(equal->types);
380 free(equal);
383 int cloog_equal_count(CloogEqualities *equal)
385 return equal->constraints->NbRows;
388 CloogConstraintSet *cloog_equal_constraints(CloogEqualities *equal)
390 return equal->constraints;
395 * cloog_constraint_equal_type function :
396 * This function returns the type of the equality in the constraint (line) of
397 * (constraints) for the element (level). An equality is 'constant' iff all
398 * other factors are null except the constant one. It is a 'pure item' iff
399 * it is equal or opposite to a single variable or parameter.
400 * Otherwise it is an 'affine expression'.
401 * For instance:
402 * i = -13 is constant, i = j, j = -M are pure items,
403 * j = 2*M, i = j+1, 2*j = M are affine expressions.
405 * - constraints is the matrix of constraints,
406 * - level is the column number in equal of the element which is 'equal to',
408 * - July 3rd 2002: first version, called pprint_equal_isconstant.
409 * - July 6th 2002: adaptation for the 3 types.
410 * - June 15th 2005: (debug) expr = domain->Constraint[line] was evaluated
411 * before checking if line != ONE_TIME_LOOP. Since
412 * ONE_TIME_LOOP is -1, an invalid read was possible.
413 * - October 19th 2005: Removal of the once-time-loop specific processing.
415 static int cloog_constraint_equal_type(CloogConstraint constraint, int level)
417 int i, one=0 ;
418 Value * expr ;
420 expr = *constraint.line;
422 if (value_notone_p(expr[level]) && value_notmone_p(expr[level]))
423 return EQTYPE_EXAFFINE;
425 /* There is only one non null factor, and it must be +1 or -1 for
426 * iterators or parameters.
428 for (i=1;i<=constraint.set->NbColumns-2;i++)
429 if (value_notzero_p(expr[i]) && (i != level))
430 { if ((value_notone_p(expr[i]) && value_notmone_p(expr[i])) || (one != 0))
431 return EQTYPE_EXAFFINE ;
432 else
433 one = 1 ;
435 /* if the constant factor is non null, it must be alone. */
436 if (one != 0)
437 { if (value_notzero_p(expr[constraint.set->NbColumns-1]))
438 return EQTYPE_EXAFFINE ;
440 else
441 return EQTYPE_CONSTANT ;
443 return EQTYPE_PUREITEM ;
447 int cloog_equal_type(CloogEqualities *equal, int level)
449 return equal->types[level-1];
454 * cloog_equal_update function:
455 * this function updates a matrix of equalities where each row corresponds to
456 * the equality "=0" of an affine expression such that the entry at column
457 * "row" (="level") is not zero. This matrix is upper-triangular, except the
458 * row number "level-1" which has to be updated for the matrix to be triangular.
459 * This function achieves the processing.
460 * - equal is the matrix to be updated,
461 * - level gives the row that has to be updated (it is actually row "level-1"),
462 * - nb_par is the number of parameters of the program.
464 * - September 20th 2005: first version.
466 static void cloog_equal_update(CloogEqualities *equal, int level, int nb_par)
467 { int i, j ;
468 Value gcd, factor_level, factor_outer, temp_level, temp_outer ;
470 value_init(gcd);
471 value_init(temp_level);
472 value_init(temp_outer);
473 value_init(factor_level);
474 value_init(factor_outer);
476 /* For each previous level, */
477 for (i=level-2;i>=0;i--)
478 { /* if the corresponding iterator is inside the current equality and is equal
479 * to something,
481 if (value_notzero_p(equal->constraints->p[level-1][i+1]) && equal->types[i])
482 { /* Compute the Greatest Common Divisor. */
483 Gcd(equal->constraints->p[level-1][i+1],
484 equal->constraints->p[i][i+1], &gcd);
486 /* Compute the factors to apply to each row vector element. */
487 value_division(factor_level, equal->constraints->p[i][i+1], gcd);
488 value_division(factor_outer, equal->constraints->p[level-1][i+1], gcd);
490 /* Now update the row 'level'. */
491 /* - the iterators, up to level, */
492 for (j=1;j<=level;j++)
493 { value_multiply(temp_level, factor_level,
494 equal->constraints->p[level-1][j]);
495 value_multiply(temp_outer, factor_outer, equal->constraints->p[i][j]);
496 value_substract(equal->constraints->p[level-1][j], temp_level, temp_outer);
498 /* - between last useful iterator (level) and the first parameter, the
499 * matrix is sparse (full of zeroes), we just do nothing there.
500 * - the parameters and the scalar.
502 for (j=0;j<nb_par+1;j++)
503 { value_multiply(temp_level,factor_level,
504 equal->constraints->p[level-1]
505 [equal->constraints->NbColumns-j-1]);
506 value_multiply(temp_outer,factor_outer,
507 equal->constraints->p[i][equal->constraints->NbColumns-j-1]);
508 value_substract(equal->constraints->p[level-1]
509 [equal->constraints->NbColumns-j-1],
510 temp_level,temp_outer) ;
515 /* Normalize (divide by GCD of all elements) the updated equality. */
516 Vector_Normalize(&(equal->constraints->p[level-1][1]),
517 equal->constraints->NbColumns-1);
519 value_clear(gcd);
520 value_clear(temp_level);
521 value_clear(temp_outer);
522 value_clear(factor_level);
523 value_clear(factor_outer);
528 * cloog_equal_add function:
529 * This function updates the row (level-1) of the equality matrix (equal) with
530 * the row that corresponds to the row (line) of the matrix (matrix).
531 * - equal is the matrix of equalities,
532 * - matrix is the matrix of constraints,
533 * - level is the column number in matrix of the element which is 'equal to',
534 * - line is the line number in matrix of the constraint we want to study,
535 * - the infos structure gives the user all options on code printing and more.
537 * - July 2nd 2002: first version.
538 * - October 19th 2005: Addition of the once-time-loop specific processing.
540 void cloog_equal_add(CloogEqualities *equal, CloogConstraintSet *matrix,
541 int level, CloogConstraint line, int nb_par)
543 int j;
544 CloogConstraint i;
546 /* If we are in the case of a loop running once, this means that the equality
547 * comes from an inequality. Here we find this inequality.
549 if (!cloog_constraint_is_valid(line))
550 { for (i = cloog_constraint_first(matrix);
551 cloog_constraint_is_valid(i); i = cloog_constraint_next(i))
552 if ((value_notzero_p(i.line[0][0]))&& (value_notzero_p(i.line[0][level])))
553 { line = i ;
555 /* Since in once-time-loops, equalities derive from inequalities, we
556 * may have to offset the values. For instance if we have 2i>=3, the
557 * equality is in fact i=2. This may happen when the level coefficient is
558 * not 1 or -1 and the scalar value is not zero. In any other case (e.g.,
559 * if the inequality is an expression including outer loop counters or
560 * parameters) the once time loop would not have been detected
561 * because of floord and ceild functions.
563 if (value_ne_si(i.line[0][level],1) &&
564 value_ne_si(i.line[0][level],-1) &&
565 value_notzero_p(i.line[0][matrix->NbColumns-1])) {
566 cloog_int_t denominator;
568 cloog_int_init(denominator);
569 cloog_int_abs(denominator, i.line[0][level]);
570 cloog_int_fdiv_q(i.line[0][matrix->NbColumns-1],
571 i.line[0][matrix->NbColumns-1], denominator);
572 cloog_int_set_si(i.line[0][level], cloog_int_sgn(i.line[0][level]));
573 cloog_int_clear(denominator);
576 break ;
579 assert(cloog_constraint_is_valid(line));
581 /* We update the line of equal corresponding to level:
582 * - the first element gives the equality type,
584 equal->types[level-1] = cloog_constraint_equal_type(line, level);
585 /* - the other elements corresponding to the equality itself
586 * (the iterators up to level, then the parameters and the scalar).
588 for (j=1;j<=level;j++)
589 value_assign(equal->constraints->p[level-1][j], line.line[0][j]);
590 for (j = 0; j < nb_par + 1; j++)
591 value_assign(equal->constraints->p[level-1][equal->constraints->NbColumns-j-1],
592 line.line[0][line.set->NbColumns-j-1]);
594 cloog_equal_update(equal, level, nb_par);
599 * cloog_equal_del function :
600 * This function reset the equality corresponding to the iterator (level)
601 * in the equality matrix (equal).
602 * - July 2nd 2002: first version.
604 void cloog_equal_del(CloogEqualities *equal, int level)
606 equal->types[level-1] = EQTYPE_NONE;
611 /******************************************************************************
612 * Processing functions *
613 ******************************************************************************/
616 * Function cloog_constraint_set_normalize:
617 * This function will modify the constraint system in such a way that when
618 * there is an equality depending on the element at level 'level', there are
619 * no more (in)equalities depending on this element. For instance, try
620 * test/valilache.cloog with options -f 8 -l 9, with and without the call
621 * to this function. At a given moment, for the level L we will have
622 * 32*P=L && L>=1 (P is a lower level), this constraint system cannot be
623 * translated directly into a source code. Thus, we normalize the domain to
624 * remove L from the inequalities. In our example, this leads to
625 * 32*P=L && 32*P>=1, that can be transated to the code
626 * if (P>=1) { L=32*P ; ... }. This function solves the DaeGon Kim bug.
627 * WARNING: Remember that if there is another call to Polylib after a call to
628 * this function, we have to recall this function.
629 * -June 16th 2005: first version (adaptation from URGent June-7th-2005 by
630 * N. Vasilache).
631 * - June 21rd 2005: Adaptation for GMP.
632 * - November 4th 2005: Complete rewriting, simpler and faster. It is no more an
633 * adaptation from URGent.
635 void cloog_constraint_set_normalize(CloogConstraintSet *matrix, int level)
636 { int ref, i, j ;
637 Value factor_i, factor_ref, temp_i, temp_ref, gcd ;
639 if (matrix == NULL)
640 return ;
642 /* Don't "normalize" the constant term. */
643 if (level == matrix->NbColumns-1)
644 return;
646 /* Let us find an equality for the current level that can be propagated. */
647 for (ref=0;ref<matrix->NbRows;ref++)
648 if (value_zero_p(matrix->p[ref][0]) && value_notzero_p(matrix->p[ref][level]))
649 { value_init(gcd);
650 value_init(temp_i);
651 value_init(temp_ref);
652 value_init(factor_i);
653 value_init(factor_ref);
655 /* Row "ref" is the reference equality, now let us find a row to simplify.*/
656 for (i=ref+1;i<matrix->NbRows;i++)
657 if (value_notzero_p(matrix->p[i][level]))
658 { /* Now let us set to 0 the "level" coefficient of row "j" using "ref".
659 * First we compute the factors to apply to each row vector element.
661 Gcd(matrix->p[ref][level],matrix->p[i][level],&gcd) ;
662 value_division(factor_i,matrix->p[ref][level],gcd) ;
663 value_division(factor_ref,matrix->p[i][level],gcd) ;
665 /* Maybe we are simplifying an inequality: factor_i must not be <0. */
666 if (value_neg_p(factor_i))
667 { value_absolute(factor_i,factor_i) ;
668 value_oppose(factor_ref,factor_ref) ;
671 /* Now update the vector. */
672 for (j=1;j<matrix->NbColumns;j++)
673 { value_multiply(temp_i,factor_i,matrix->p[i][j]) ;
674 value_multiply(temp_ref,factor_ref,matrix->p[ref][j]) ;
675 value_substract(matrix->p[i][j],temp_i,temp_ref) ;
678 /* Normalize (divide by GCD of all elements) the updated vector. */
679 Vector_Normalize(&(matrix->p[i][1]),matrix->NbColumns-1) ;
682 value_clear(gcd);
683 value_clear(temp_i);
684 value_clear(temp_ref);
685 value_clear(factor_i);
686 value_clear(factor_ref);
687 break ;
694 * cloog_constraint_set_copy function:
695 * this functions builds and returns a "hard copy" (not a pointer copy) of a
696 * Matrix data structure.
697 * - October 26th 2005: first version.
699 CloogConstraintSet *cloog_constraint_set_copy(CloogConstraintSet *matrix)
700 { int i, j ;
701 Matrix * copy ;
703 copy = cloog_polylib_matrix_alloc(matrix->NbRows,matrix->NbColumns) ;
705 for (i=0;i<matrix->NbRows;i++)
706 for (j=0;j<matrix->NbColumns;j++)
707 value_assign(copy->p[i][j],matrix->p[i][j]) ;
709 return copy ;
714 * cloog_polylib_matrix_vector_copy function:
715 * this function rutuns a hard copy of the vector "vector" of length "length"
716 * provided as input.
717 * - November 3rd 2005: first version.
719 static Value *cloog_polylib_matrix_vector_copy(Value *vector, int length)
720 { int i ;
721 Value * copy ;
723 /* We allocate the memory space for the new vector, and we fill it with
724 * the original coefficients.
726 copy = (Value *)malloc(length * sizeof(Value)) ;
727 for (i=0;i<length;i++) {
728 value_init(copy[i]);
729 value_assign(copy[i],vector[i]);
732 return copy ;
737 * cloog_polylib_matrix_vector_free function:
738 * this function clears the elements of a vector "vector" of length "length",
739 * then frees the vector itself this is useful for the GMP version of CLooG
740 * which has to clear all elements.
741 * - October 29th 2005: first version.
743 static void cloog_polylib_matrix_vector_free(Value * vector, int length)
744 { int i ;
746 for (i=0;i<length;i++)
747 value_clear(vector[i]);
748 free(vector) ;
753 * cloog_equal_vector_simplify function:
754 * this function simplify an affine expression with its coefficients in
755 * "vector" of length "length" thanks to an equality matrix "equal" that gives
756 * for some elements of the affine expression an equality with other elements,
757 * preferably constants. For instance, if the vector contains i+j+3 and the
758 * equality matrix gives i=n and j=2, the vector is simplified to n+3 and is
759 * returned in a new vector.
760 * - vector is the array of affine expression coefficients
761 * - equal is the matrix of equalities,
762 * - length is the vector length,
763 * - level is a level we don't want to simplify (-1 if none),
764 * - nb_par is the number of parameters of the program.
766 * - September 20th 2005: first version.
767 * - November 2nd 2005: (debug) we are simplifying inequalities, thus we are
768 * not allowed to multiply the vector by a negative
769 * constant.Problem found after a report of Michael
770 * Classen.
772 Value *cloog_equal_vector_simplify(CloogEqualities *equal, Value *vector,
773 int length, int level, int nb_par)
774 { int i, j ;
775 Value gcd, factor_vector, factor_equal, temp_vector, temp_equal, * simplified;
777 simplified = cloog_polylib_matrix_vector_copy(vector,length) ;
779 value_init(gcd);
780 value_init(temp_vector);
781 value_init(temp_equal);
782 value_init(factor_vector);
783 value_init(factor_equal);
785 /* For each non-null coefficient in the vector, */
786 for (i=length-nb_par-2;i>0;i--)
787 if (i != level)
788 { /* if the coefficient in not null, and there exists a useful equality */
789 if ((value_notzero_p(simplified[i])) && equal->types[i-1])
790 { /* Compute the Greatest Common Divisor. */
791 Gcd(simplified[i], equal->constraints->p[i-1][i], &gcd);
793 /* Compute the factors to apply to each row vector element. */
794 value_division(factor_vector, equal->constraints->p[i-1][i], gcd);
795 value_division(factor_equal,simplified[i],gcd) ;
797 /* We are simplifying an inequality: factor_vector must not be <0. */
798 if (value_neg_p(factor_vector))
799 { value_absolute(factor_vector,factor_vector) ;
800 value_oppose(factor_equal,factor_equal) ;
803 /* Now update the vector. */
804 /* - the iterators, up to the current level, */
805 for (j=1;j<=length-nb_par-2;j++)
806 { value_multiply(temp_vector,factor_vector,simplified[j]) ;
807 value_multiply(temp_equal, factor_equal, equal->constraints->p[i-1][j]);
808 value_substract(simplified[j],temp_vector,temp_equal) ;
810 /* - between last useful iterator (i) and the first parameter, the equal
811 * matrix is sparse (full of zeroes), we just do nothing there.
812 * - the parameters and the scalar.
814 for (j=0;j<nb_par+1;j++)
815 { value_multiply(temp_vector,factor_vector,simplified[length-1-j]) ;
816 value_multiply(temp_equal,factor_equal,
817 equal->constraints->p[i-1][equal->constraints->NbColumns-j-1]);
818 value_substract(simplified[length-1-j],temp_vector,temp_equal) ;
823 /* Normalize (divide by GCD of all elements) the updated vector. */
824 Vector_Normalize(&(simplified[1]),length-1) ;
826 value_clear(gcd);
827 value_clear(temp_vector);
828 value_clear(temp_equal);
829 value_clear(factor_vector);
830 value_clear(factor_equal);
832 return simplified ;
837 * cloog_constraint_set_simplify function:
838 * this function simplify all constraints inside the matrix "matrix" thanks to
839 * an equality matrix "equal" that gives for some elements of the affine
840 * constraint an equality with other elements, preferably constants.
841 * For instance, if a row of the matrix contains i+j+3>=0 and the equality
842 * matrix gives i=n and j=2, the constraint is simplified to n+3>=0. The
843 * simplified constraints are returned back inside a new simplified matrix.
844 * - matrix is the set of constraints to simplify,
845 * - equal is the matrix of equalities,
846 * - level is a level we don't want to simplify (-1 if none),
847 * - nb_par is the number of parameters of the program.
849 * - November 4th 2005: first version.
851 CloogConstraintSet *cloog_constraint_set_simplify(CloogConstraintSet *matrix,
852 CloogEqualities *equal, int level, int nb_par)
853 { int i, j, k ;
854 Value * vector ;
855 Matrix * simplified ;
857 if (matrix == NULL)
858 return NULL ;
860 /* The simplified matrix is such that each row has been simplified thanks
861 * tho the "equal" matrix. We allocate the memory for the simplified matrix,
862 * then for each row of the original matrix, we compute the simplified
863 * vector and we copy its content into the according simplified row.
865 simplified = cloog_polylib_matrix_alloc(matrix->NbRows,matrix->NbColumns) ;
866 for (i=0;i<matrix->NbRows;i++)
867 { vector = cloog_equal_vector_simplify(equal, matrix->p[i],
868 matrix->NbColumns, level, nb_par);
869 for (j=0;j<matrix->NbColumns;j++)
870 value_assign(simplified->p[i][j],vector[j]) ;
872 cloog_polylib_matrix_vector_free(vector,matrix->NbColumns) ;
875 /* After simplification, it may happen that few constraints are the same,
876 * we remove them here by replacing them with 0=0 constraints.
878 for (i=0;i<simplified->NbRows;i++)
879 for (j=i+1;j<simplified->NbRows;j++)
880 { for (k=0;k<simplified->NbColumns;k++)
881 if (value_ne(simplified->p[i][k],simplified->p[j][k]))
882 break ;
884 if (k == matrix->NbColumns)
885 { for (k=0;k<matrix->NbColumns;k++)
886 value_set_si(simplified->p[j][k],0) ;
890 return simplified ;
895 * Return clast_expr corresponding to the variable "level" (1 based) in
896 * the given constraint.
898 struct clast_expr *cloog_constraint_variable_expr(CloogConstraint constraint,
899 int level, CloogNames *names)
901 int total_dim, nb_iter;
902 const char *name;
904 total_dim = cloog_constraint_total_dimension(constraint);
905 nb_iter = total_dim - names->nb_parameters;
907 if (level <= nb_iter)
908 name = cloog_names_name_at_level(names, level);
909 else
910 name = names->parameters[level - (nb_iter+1)] ;
912 return &new_clast_name(name)->expr;
917 * Return true if constraint c involves variable v (zero-based).
919 int cloog_constraint_involves(CloogConstraint constraint, int v)
921 return value_notzero_p(constraint.line[0][1+v]);
924 int cloog_constraint_is_lower_bound(CloogConstraint constraint, int v)
926 return value_pos_p(constraint.line[0][1+v]);
929 int cloog_constraint_is_upper_bound(CloogConstraint constraint, int v)
931 return value_neg_p(constraint.line[0][1+v]);
934 int cloog_constraint_is_equality(CloogConstraint constraint)
936 return value_zero_p(constraint.line[0][0]);
939 void cloog_constraint_clear(CloogConstraint constraint)
941 int k;
943 for (k = 1; k <= constraint.set->NbColumns - 2; k++)
944 value_set_si(constraint.line[0][k], 0);
947 void cloog_constraint_coefficient_get(CloogConstraint constraint,
948 int var, Value *val)
950 value_assign(*val, constraint.line[0][1+var]);
953 void cloog_constraint_coefficient_set(CloogConstraint constraint,
954 int var, Value val)
956 value_assign(constraint.line[0][1+var], val);
959 void cloog_constraint_constant_get(CloogConstraint constraint, cloog_int_t *val)
961 value_assign(*val, constraint.line[0][constraint.set->NbColumns-1]);
965 * Copy the coefficient of constraint c into dst in PolyLib order,
966 * i.e., first the coefficients of the variables, then the coefficients
967 * of the parameters and finally the constant.
969 void cloog_constraint_copy_coefficients(CloogConstraint constraint,
970 cloog_int_t *dst)
972 Vector_Copy(constraint.line[0]+1, dst, constraint.set->NbColumns-1);
975 CloogConstraint cloog_constraint_invalid(void)
977 CloogConstraint c;
978 c.set = NULL;
979 c.line = NULL;
980 return c;
983 int cloog_constraint_is_valid(CloogConstraint constraint)
985 return constraint.set != NULL && constraint.line != NULL;
989 * Create a CloogConstraintSet containing enough information to perform
990 * a reduction on the upper equality (in this case lower is an invalid
991 * CloogConstraint) or the pair of inequalities upper and lower
992 * from within insert_modulo_guard.
993 * In the PolyLib backend, we return a CloogConstraintSet containting only
994 * the upper bound. The reduction will not change the stride so there
995 * will be no need to recompute the bound on the modulo expression.
997 CloogConstraintSet *cloog_constraint_set_for_reduction(CloogConstraint upper,
998 CloogConstraint lower)
1000 CloogConstraintSet *set;
1002 set = cloog_polylib_matrix_alloc(1, upper.set->NbColumns);
1003 Vector_Copy(upper.line[0], set->p[0], set->NbColumns);
1004 return set;
1008 /* Computes x, y and g such that g = gcd(a,b) and a*x+b*y = g */
1009 static void Euclid(cloog_int_t a, cloog_int_t b,
1010 cloog_int_t *x, cloog_int_t *y, cloog_int_t *g)
1012 cloog_int_t c, d, e, f, tmp;
1014 cloog_int_init(c);
1015 cloog_int_init(d);
1016 cloog_int_init(e);
1017 cloog_int_init(f);
1018 cloog_int_init(tmp);
1019 cloog_int_abs(c, a);
1020 cloog_int_abs(d, b);
1021 cloog_int_set_si(e, 1);
1022 cloog_int_set_si(f, 0);
1023 while (cloog_int_is_pos(d)) {
1024 cloog_int_tdiv_q(tmp, c, d);
1025 cloog_int_mul(tmp, tmp, f);
1026 cloog_int_sub(e, e, tmp);
1027 cloog_int_tdiv_q(tmp, c, d);
1028 cloog_int_mul(tmp, tmp, d);
1029 cloog_int_sub(c, c, tmp);
1030 cloog_int_swap(c, d);
1031 cloog_int_swap(e, f);
1033 cloog_int_set(*g, c);
1034 if (cloog_int_is_zero(a))
1035 cloog_int_set_si(*x, 0);
1036 else if (cloog_int_is_pos(a))
1037 cloog_int_set(*x, e);
1038 else cloog_int_neg(*x, e);
1039 if (cloog_int_is_zero(b))
1040 cloog_int_set_si(*y, 0);
1041 else {
1042 cloog_int_mul(tmp, a, *x);
1043 cloog_int_sub(tmp, c, tmp);
1044 cloog_int_divexact(*y, tmp, b);
1046 cloog_int_clear(c);
1047 cloog_int_clear(d);
1048 cloog_int_clear(e);
1049 cloog_int_clear(f);
1050 cloog_int_clear(tmp);
1054 * Reduce the modulo guard expressed by "contraints" using equalities
1055 * found in outer nesting levels (stored in "equal").
1056 * The modulo guard may be an equality or a pair of inequalities.
1057 * In case of a pair of inequalities, "constraints" only contains the
1058 * upper bound and *bound contains the bound on the
1059 * corresponding modulo expression. The bound is left untouched by
1060 * this function.
1062 CloogConstraintSet *cloog_constraint_set_reduce(CloogConstraintSet *constraints,
1063 int level, CloogEqualities *equal, int nb_par, cloog_int_t *bound)
1065 int i, j, k, len, len2, nb_iter;
1066 struct cloog_vec *line_vector2;
1067 cloog_int_t *line, *line2, val, x, y, g;
1069 len = constraints->NbColumns;
1070 len2 = cloog_equal_total_dimension(equal) + 2;
1071 nb_iter = len - 2 - nb_par;
1073 cloog_int_init(val);
1074 cloog_int_init(x);
1075 cloog_int_init(y);
1076 cloog_int_init(g);
1078 line_vector2 = cloog_vec_alloc(len2);
1079 line2 = line_vector2->p;
1081 line = constraints->p[0];
1082 if (cloog_int_is_pos(line[level]))
1083 cloog_seq_neg(line+1, line+1, len-1);
1084 cloog_int_neg(line[level], line[level]);
1085 assert(cloog_int_is_pos(line[level]));
1087 for (i = nb_iter; i >= 1; --i) {
1088 if (i == level)
1089 continue;
1090 cloog_int_fdiv_r(line[i], line[i], line[level]);
1091 if (cloog_int_is_zero(line[i]))
1092 continue;
1094 /* Look for an earlier variable that is also a multiple of line[level]
1095 * and check whether we can use the corresponding affine expression
1096 * to "reduce" the modulo guard, where reduction means that we eliminate
1097 * a variable, possibly at the expense of introducing other variables
1098 * with smaller index.
1100 for (j = level-1; j >= 0; --j) {
1101 CloogConstraint equal_constraint;
1102 if (cloog_equal_type(equal, j+1) != EQTYPE_EXAFFINE)
1103 continue;
1104 equal_constraint = cloog_equal_constraint(equal, j);
1105 cloog_constraint_coefficient_get(equal_constraint, j, &val);
1106 if (!cloog_int_is_divisible_by(val, line[level])) {
1107 cloog_constraint_release(equal_constraint);
1108 continue;
1110 cloog_constraint_coefficient_get(equal_constraint, i-1, &val);
1111 if (cloog_int_is_divisible_by(val, line[level])) {
1112 cloog_constraint_release(equal_constraint);
1113 continue;
1115 for (k = j; k > i; --k) {
1116 cloog_constraint_coefficient_get(equal_constraint, k-1, &val);
1117 if (cloog_int_is_zero(val))
1118 continue;
1119 if (!cloog_int_is_divisible_by(val, line[level]))
1120 break;
1122 if (k > i) {
1123 cloog_constraint_release(equal_constraint);
1124 continue;
1126 cloog_constraint_coefficient_get(equal_constraint, i-1, &val);
1127 Euclid(val, line[level], &x, &y, &g);
1128 if (!cloog_int_is_divisible_by(val, line[i])) {
1129 cloog_constraint_release(equal_constraint);
1130 continue;
1132 cloog_int_divexact(val, line[i], g);
1133 cloog_int_neg(val, val);
1134 cloog_int_mul(val, val, x);
1135 cloog_int_set_si(y, 1);
1136 /* Add (equal->p[j][i])^{-1} * line[i] times the equality */
1137 cloog_constraint_copy_coefficients(equal_constraint, line2+1);
1138 cloog_seq_combine(line+1, y, line+1, val, line2+1, i);
1139 cloog_seq_combine(line+len-nb_par-1, y, line+len-nb_par-1,
1140 val, line2+len2-nb_par-1, nb_par+1);
1141 cloog_constraint_release(equal_constraint);
1142 break;
1146 cloog_vec_free(line_vector2);
1148 cloog_int_clear(val);
1149 cloog_int_clear(x);
1150 cloog_int_clear(y);
1151 cloog_int_clear(g);
1153 /* Make sure the line is not inverted again in the calling function. */
1154 cloog_int_neg(line[level], line[level]);
1156 return constraints;
1159 CloogConstraint cloog_constraint_first(CloogConstraintSet *constraints)
1161 CloogConstraint c;
1162 if (constraints->NbRows == 0)
1163 return cloog_constraint_invalid();
1164 c.set = constraints;
1165 c.line = &constraints->p[0];
1166 return c;
1169 CloogConstraint cloog_constraint_next(CloogConstraint constraint)
1171 CloogConstraint c = constraint;
1172 c.line++;
1173 if (c.line == c.set->p + c.set->NbRows) {
1174 c.line = NULL;
1175 c.set = NULL;
1177 return c;
1180 CloogConstraint cloog_constraint_copy(CloogConstraint constraint)
1182 return constraint;
1185 void cloog_constraint_release(CloogConstraint constraint)
1189 CloogConstraint cloog_equal_constraint(CloogEqualities *equal, int j)
1191 CloogConstraint c;
1192 c.set = equal->constraints;
1193 c.line = &equal->constraints->p[j];
1194 return c;