2 * Copyright 2010-2011 INRIA Saclay
3 * Copyright 2011 Sven Verdoolaege
4 * Copyright 2012-2014 Ecole Normale Superieure
6 * Use of this software is governed by the MIT license
8 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
9 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
11 * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
15 #include <isl_val_private.h>
17 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
18 #define FN(TYPE,NAME) xFN(TYPE,NAME)
19 #define xS(TYPE,NAME) struct TYPE ## _ ## NAME
20 #define S(TYPE,NAME) xS(TYPE,NAME)
23 __isl_give PW
*FN(PW
,alloc_size
)(__isl_take isl_space
*dim
,
24 enum isl_fold type
, int n
)
26 __isl_give PW
*FN(PW
,alloc_size
)(__isl_take isl_space
*dim
, int n
)
34 ctx
= isl_space_get_ctx(dim
);
35 isl_assert(ctx
, n
>= 0, goto error
);
36 pw
= isl_alloc(ctx
, struct PW
,
37 sizeof(struct PW
) + (n
- 1) * sizeof(S(PW
,piece
)));
55 __isl_give PW
*FN(PW
,ZERO
)(__isl_take isl_space
*dim
, enum isl_fold type
)
57 return FN(PW
,alloc_size
)(dim
, type
, 0);
60 __isl_give PW
*FN(PW
,ZERO
)(__isl_take isl_space
*dim
)
62 return FN(PW
,alloc_size
)(dim
, 0);
66 __isl_give PW
*FN(PW
,add_piece
)(__isl_take PW
*pw
,
67 __isl_take isl_set
*set
, __isl_take EL
*el
)
70 isl_space
*el_dim
= NULL
;
72 if (!pw
|| !set
|| !el
)
75 if (isl_set_plain_is_empty(set
) || FN(EL
,EL_IS_ZERO
)(el
)) {
81 ctx
= isl_set_get_ctx(set
);
83 if (pw
->type
!= el
->type
)
84 isl_die(ctx
, isl_error_invalid
, "fold types don't match",
87 el_dim
= FN(EL
,get_space(el
));
88 isl_assert(ctx
, isl_space_is_equal(pw
->dim
, el_dim
), goto error
);
89 isl_assert(ctx
, pw
->n
< pw
->size
, goto error
);
91 pw
->p
[pw
->n
].set
= set
;
92 pw
->p
[pw
->n
].FIELD
= el
;
95 isl_space_free(el_dim
);
98 isl_space_free(el_dim
);
106 __isl_give PW
*FN(PW
,alloc
)(enum isl_fold type
,
107 __isl_take isl_set
*set
, __isl_take EL
*el
)
109 __isl_give PW
*FN(PW
,alloc
)(__isl_take isl_set
*set
, __isl_take EL
*el
)
118 pw
= FN(PW
,alloc_size
)(FN(EL
,get_space
)(el
), type
, 1);
120 pw
= FN(PW
,alloc_size
)(FN(EL
,get_space
)(el
), 1);
123 return FN(PW
,add_piece
)(pw
, set
, el
);
130 __isl_give PW
*FN(PW
,dup
)(__isl_keep PW
*pw
)
139 dup
= FN(PW
,alloc_size
)(isl_space_copy(pw
->dim
), pw
->type
, pw
->n
);
141 dup
= FN(PW
,alloc_size
)(isl_space_copy(pw
->dim
), pw
->n
);
146 for (i
= 0; i
< pw
->n
; ++i
)
147 dup
= FN(PW
,add_piece
)(dup
, isl_set_copy(pw
->p
[i
].set
),
148 FN(EL
,copy
)(pw
->p
[i
].FIELD
));
153 __isl_give PW
*FN(PW
,cow
)(__isl_take PW
*pw
)
161 return FN(PW
,dup
)(pw
);
164 __isl_give PW
*FN(PW
,copy
)(__isl_keep PW
*pw
)
173 __isl_null PW
*FN(PW
,free
)(__isl_take PW
*pw
)
182 for (i
= 0; i
< pw
->n
; ++i
) {
183 isl_set_free(pw
->p
[i
].set
);
184 FN(EL
,free
)(pw
->p
[i
].FIELD
);
186 isl_space_free(pw
->dim
);
192 const char *FN(PW
,get_dim_name
)(__isl_keep PW
*pw
, enum isl_dim_type type
,
195 return pw
? isl_space_get_dim_name(pw
->dim
, type
, pos
) : NULL
;
198 int FN(PW
,has_dim_id
)(__isl_keep PW
*pw
, enum isl_dim_type type
, unsigned pos
)
200 return pw
? isl_space_has_dim_id(pw
->dim
, type
, pos
) : -1;
203 __isl_give isl_id
*FN(PW
,get_dim_id
)(__isl_keep PW
*pw
, enum isl_dim_type type
,
206 return pw
? isl_space_get_dim_id(pw
->dim
, type
, pos
) : NULL
;
209 int FN(PW
,has_tuple_name
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
211 return pw
? isl_space_has_tuple_name(pw
->dim
, type
) : -1;
214 const char *FN(PW
,get_tuple_name
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
216 return pw
? isl_space_get_tuple_name(pw
->dim
, type
) : NULL
;
219 int FN(PW
,has_tuple_id
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
221 return pw
? isl_space_has_tuple_id(pw
->dim
, type
) : -1;
224 __isl_give isl_id
*FN(PW
,get_tuple_id
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
226 return pw
? isl_space_get_tuple_id(pw
->dim
, type
) : NULL
;
229 int FN(PW
,IS_ZERO
)(__isl_keep PW
*pw
)
238 __isl_give PW
*FN(PW
,realign_domain
)(__isl_take PW
*pw
,
239 __isl_take isl_reordering
*exp
)
247 for (i
= 0; i
< pw
->n
; ++i
) {
248 pw
->p
[i
].set
= isl_set_realign(pw
->p
[i
].set
,
249 isl_reordering_copy(exp
));
252 pw
->p
[i
].FIELD
= FN(EL
,realign_domain
)(pw
->p
[i
].FIELD
,
253 isl_reordering_copy(exp
));
258 pw
= FN(PW
,reset_domain_space
)(pw
, isl_space_copy(exp
->dim
));
260 isl_reordering_free(exp
);
263 isl_reordering_free(exp
);
268 /* Align the parameters of "pw" to those of "model".
270 __isl_give PW
*FN(PW
,align_params
)(__isl_take PW
*pw
, __isl_take isl_space
*model
)
277 ctx
= isl_space_get_ctx(model
);
278 if (!isl_space_has_named_params(model
))
279 isl_die(ctx
, isl_error_invalid
,
280 "model has unnamed parameters", goto error
);
281 if (!isl_space_has_named_params(pw
->dim
))
282 isl_die(ctx
, isl_error_invalid
,
283 "input has unnamed parameters", goto error
);
284 if (!isl_space_match(pw
->dim
, isl_dim_param
, model
, isl_dim_param
)) {
287 model
= isl_space_drop_dims(model
, isl_dim_in
,
288 0, isl_space_dim(model
, isl_dim_in
));
289 model
= isl_space_drop_dims(model
, isl_dim_out
,
290 0, isl_space_dim(model
, isl_dim_out
));
291 exp
= isl_parameter_alignment_reordering(pw
->dim
, model
);
292 exp
= isl_reordering_extend_space(exp
,
293 FN(PW
,get_domain_space
)(pw
));
294 pw
= FN(PW
,realign_domain
)(pw
, exp
);
297 isl_space_free(model
);
300 isl_space_free(model
);
305 static __isl_give PW
*FN(PW
,align_params_pw_pw_and
)(__isl_take PW
*pw1
,
307 __isl_give PW
*(*fn
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
))
313 if (isl_space_match(pw1
->dim
, isl_dim_param
, pw2
->dim
, isl_dim_param
))
315 ctx
= FN(PW
,get_ctx
)(pw1
);
316 if (!isl_space_has_named_params(pw1
->dim
) ||
317 !isl_space_has_named_params(pw2
->dim
))
318 isl_die(ctx
, isl_error_invalid
,
319 "unaligned unnamed parameters", goto error
);
320 pw1
= FN(PW
,align_params
)(pw1
, FN(PW
,get_space
)(pw2
));
321 pw2
= FN(PW
,align_params
)(pw2
, FN(PW
,get_space
)(pw1
));
329 static __isl_give PW
*FN(PW
,align_params_pw_set_and
)(__isl_take PW
*pw
,
330 __isl_take isl_set
*set
,
331 __isl_give PW
*(*fn
)(__isl_take PW
*pw
, __isl_take isl_set
*set
))
337 if (isl_space_match(pw
->dim
, isl_dim_param
, set
->dim
, isl_dim_param
))
339 ctx
= FN(PW
,get_ctx
)(pw
);
340 if (!isl_space_has_named_params(pw
->dim
) ||
341 !isl_space_has_named_params(set
->dim
))
342 isl_die(ctx
, isl_error_invalid
,
343 "unaligned unnamed parameters", goto error
);
344 pw
= FN(PW
,align_params
)(pw
, isl_set_get_space(set
));
345 set
= isl_set_align_params(set
, FN(PW
,get_space
)(pw
));
354 static __isl_give PW
*FN(PW
,union_add_aligned
)(__isl_take PW
*pw1
,
365 ctx
= isl_space_get_ctx(pw1
->dim
);
367 if (pw1
->type
!= pw2
->type
)
368 isl_die(ctx
, isl_error_invalid
,
369 "fold types don't match", goto error
);
371 isl_assert(ctx
, isl_space_is_equal(pw1
->dim
, pw2
->dim
), goto error
);
373 if (FN(PW
,IS_ZERO
)(pw1
)) {
378 if (FN(PW
,IS_ZERO
)(pw2
)) {
383 n
= (pw1
->n
+ 1) * (pw2
->n
+ 1);
385 res
= FN(PW
,alloc_size
)(isl_space_copy(pw1
->dim
), pw1
->type
, n
);
387 res
= FN(PW
,alloc_size
)(isl_space_copy(pw1
->dim
), n
);
390 for (i
= 0; i
< pw1
->n
; ++i
) {
391 set
= isl_set_copy(pw1
->p
[i
].set
);
392 for (j
= 0; j
< pw2
->n
; ++j
) {
393 struct isl_set
*common
;
395 common
= isl_set_intersect(isl_set_copy(pw1
->p
[i
].set
),
396 isl_set_copy(pw2
->p
[j
].set
));
397 if (isl_set_plain_is_empty(common
)) {
398 isl_set_free(common
);
401 set
= isl_set_subtract(set
,
402 isl_set_copy(pw2
->p
[j
].set
));
404 sum
= FN(EL
,add_on_domain
)(common
,
405 FN(EL
,copy
)(pw1
->p
[i
].FIELD
),
406 FN(EL
,copy
)(pw2
->p
[j
].FIELD
));
408 res
= FN(PW
,add_piece
)(res
, common
, sum
);
410 res
= FN(PW
,add_piece
)(res
, set
, FN(EL
,copy
)(pw1
->p
[i
].FIELD
));
413 for (j
= 0; j
< pw2
->n
; ++j
) {
414 set
= isl_set_copy(pw2
->p
[j
].set
);
415 for (i
= 0; i
< pw1
->n
; ++i
)
416 set
= isl_set_subtract(set
,
417 isl_set_copy(pw1
->p
[i
].set
));
418 res
= FN(PW
,add_piece
)(res
, set
, FN(EL
,copy
)(pw2
->p
[j
].FIELD
));
431 /* Private version of "union_add". For isl_pw_qpolynomial and
432 * isl_pw_qpolynomial_fold, we prefer to simply call it "add".
434 static __isl_give PW
*FN(PW
,union_add_
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
436 return FN(PW
,align_params_pw_pw_and
)(pw1
, pw2
,
437 &FN(PW
,union_add_aligned
));
440 /* Make sure "pw" has room for at least "n" more pieces.
442 * If there is only one reference to pw, we extend it in place.
443 * Otherwise, we create a new PW and copy the pieces.
445 static __isl_give PW
*FN(PW
,grow
)(__isl_take PW
*pw
, int n
)
453 if (pw
->n
+ n
<= pw
->size
)
455 ctx
= FN(PW
,get_ctx
)(pw
);
458 res
= isl_realloc(ctx
, pw
, struct PW
,
459 sizeof(struct PW
) + (n
- 1) * sizeof(S(PW
,piece
)));
461 return FN(PW
,free
)(pw
);
466 res
= FN(PW
,alloc_size
)(isl_space_copy(pw
->dim
), pw
->type
, n
);
468 res
= FN(PW
,alloc_size
)(isl_space_copy(pw
->dim
), n
);
471 return FN(PW
,free
)(pw
);
472 for (i
= 0; i
< pw
->n
; ++i
)
473 res
= FN(PW
,add_piece
)(res
, isl_set_copy(pw
->p
[i
].set
),
474 FN(EL
,copy
)(pw
->p
[i
].FIELD
));
479 static __isl_give PW
*FN(PW
,add_disjoint_aligned
)(__isl_take PW
*pw1
,
488 if (pw1
->size
< pw1
->n
+ pw2
->n
&& pw1
->n
< pw2
->n
)
489 return FN(PW
,add_disjoint_aligned
)(pw2
, pw1
);
491 ctx
= isl_space_get_ctx(pw1
->dim
);
493 if (pw1
->type
!= pw2
->type
)
494 isl_die(ctx
, isl_error_invalid
,
495 "fold types don't match", goto error
);
497 isl_assert(ctx
, isl_space_is_equal(pw1
->dim
, pw2
->dim
), goto error
);
499 if (FN(PW
,IS_ZERO
)(pw1
)) {
504 if (FN(PW
,IS_ZERO
)(pw2
)) {
509 pw1
= FN(PW
,grow
)(pw1
, pw2
->n
);
513 for (i
= 0; i
< pw2
->n
; ++i
)
514 pw1
= FN(PW
,add_piece
)(pw1
,
515 isl_set_copy(pw2
->p
[i
].set
),
516 FN(EL
,copy
)(pw2
->p
[i
].FIELD
));
527 __isl_give PW
*FN(PW
,add_disjoint
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
529 return FN(PW
,align_params_pw_pw_and
)(pw1
, pw2
,
530 &FN(PW
,add_disjoint_aligned
));
533 /* This function is currently only used from isl_aff.c
535 static __isl_give PW
*FN(PW
,on_shared_domain_in
)(__isl_take PW
*pw1
,
536 __isl_take PW
*pw2
, __isl_take isl_space
*space
,
537 __isl_give EL
*(*fn
)(__isl_take EL
*el1
, __isl_take EL
*el2
))
538 __attribute__ ((unused
));
540 /* Apply "fn" to pairs of elements from pw1 and pw2 on shared domains.
541 * The result of "fn" (and therefore also of this function) lives in "space".
543 static __isl_give PW
*FN(PW
,on_shared_domain_in
)(__isl_take PW
*pw1
,
544 __isl_take PW
*pw2
, __isl_take isl_space
*space
,
545 __isl_give EL
*(*fn
)(__isl_take EL
*el1
, __isl_take EL
*el2
))
555 res
= FN(PW
,alloc_size
)(isl_space_copy(space
), pw1
->type
, n
);
557 res
= FN(PW
,alloc_size
)(isl_space_copy(space
), n
);
560 for (i
= 0; i
< pw1
->n
; ++i
) {
561 for (j
= 0; j
< pw2
->n
; ++j
) {
566 common
= isl_set_intersect(
567 isl_set_copy(pw1
->p
[i
].set
),
568 isl_set_copy(pw2
->p
[j
].set
));
569 empty
= isl_set_plain_is_empty(common
);
570 if (empty
< 0 || empty
) {
571 isl_set_free(common
);
577 res_ij
= fn(FN(EL
,copy
)(pw1
->p
[i
].FIELD
),
578 FN(EL
,copy
)(pw2
->p
[j
].FIELD
));
579 res_ij
= FN(EL
,gist
)(res_ij
, isl_set_copy(common
));
581 res
= FN(PW
,add_piece
)(res
, common
, res_ij
);
585 isl_space_free(space
);
590 isl_space_free(space
);
597 /* This function is currently only used from isl_aff.c
599 static __isl_give PW
*FN(PW
,on_shared_domain
)(__isl_take PW
*pw1
,
601 __isl_give EL
*(*fn
)(__isl_take EL
*el1
, __isl_take EL
*el2
))
602 __attribute__ ((unused
));
604 /* Apply "fn" to pairs of elements from pw1 and pw2 on shared domains.
605 * The result of "fn" is assumed to live in the same space as "pw1" and "pw2".
607 static __isl_give PW
*FN(PW
,on_shared_domain
)(__isl_take PW
*pw1
,
609 __isl_give EL
*(*fn
)(__isl_take EL
*el1
, __isl_take EL
*el2
))
616 space
= isl_space_copy(pw1
->dim
);
617 return FN(PW
,on_shared_domain_in
)(pw1
, pw2
, space
, fn
);
625 __isl_give PW
*FN(PW
,neg
)(__isl_take PW
*pw
)
632 if (FN(PW
,IS_ZERO
)(pw
))
639 for (i
= 0; i
< pw
->n
; ++i
) {
640 pw
->p
[i
].FIELD
= FN(EL
,neg
)(pw
->p
[i
].FIELD
);
642 return FN(PW
,free
)(pw
);
650 __isl_give PW
*FN(PW
,sub
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
652 return FN(PW
,add
)(pw1
, FN(PW
,neg
)(pw2
));
657 __isl_give isl_val
*FN(PW
,eval
)(__isl_take PW
*pw
, __isl_take isl_point
*pnt
)
662 isl_space
*pnt_dim
= NULL
;
667 ctx
= isl_point_get_ctx(pnt
);
668 pnt_dim
= isl_point_get_space(pnt
);
669 isl_assert(ctx
, isl_space_is_domain_internal(pnt_dim
, pw
->dim
),
672 for (i
= 0; i
< pw
->n
; ++i
) {
673 found
= isl_set_contains_point(pw
->p
[i
].set
, pnt
);
680 v
= FN(EL
,eval
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
681 isl_point_copy(pnt
));
683 v
= isl_val_zero(ctx
);
685 isl_space_free(pnt_dim
);
690 isl_space_free(pnt_dim
);
696 /* Return the parameter domain of "pw".
698 __isl_give isl_set
*FN(PW
,params
)(__isl_take PW
*pw
)
700 return isl_set_params(FN(PW
,domain
)(pw
));
703 __isl_give isl_set
*FN(PW
,domain
)(__isl_take PW
*pw
)
711 dom
= isl_set_empty(FN(PW
,get_domain_space
)(pw
));
712 for (i
= 0; i
< pw
->n
; ++i
)
713 dom
= isl_set_union_disjoint(dom
, isl_set_copy(pw
->p
[i
].set
));
720 /* Exploit the equalities in the domain of piece "i" of "pw"
721 * to simplify the associated function.
722 * If the domain of piece "i" is empty, then remove it entirely,
723 * replacing it with the final piece.
725 static int FN(PW
,exploit_equalities_and_remove_if_empty
)(__isl_keep PW
*pw
,
729 int empty
= isl_set_plain_is_empty(pw
->p
[i
].set
);
734 isl_set_free(pw
->p
[i
].set
);
735 FN(EL
,free
)(pw
->p
[i
].FIELD
);
737 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
743 aff
= isl_set_affine_hull(isl_set_copy(pw
->p
[i
].set
));
744 pw
->p
[i
].FIELD
= FN(EL
,substitute_equalities
)(pw
->p
[i
].FIELD
, aff
);
751 /* Convert a piecewise expression defined over a parameter domain
752 * into one that is defined over a zero-dimensional set.
754 __isl_give PW
*FN(PW
,from_range
)(__isl_take PW
*pw
)
760 if (!isl_space_is_set(pw
->dim
))
761 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
762 "not living in a set space", return FN(PW
,free
)(pw
));
764 space
= FN(PW
,get_space
)(pw
);
765 space
= isl_space_from_range(space
);
766 pw
= FN(PW
,reset_space
)(pw
, space
);
771 /* Fix the value of the given parameter or domain dimension of "pw"
772 * to be equal to "value".
774 __isl_give PW
*FN(PW
,fix_si
)(__isl_take PW
*pw
, enum isl_dim_type type
,
775 unsigned pos
, int value
)
782 if (type
== isl_dim_out
)
783 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
784 "cannot fix output dimension", return FN(PW
,free
)(pw
));
789 if (type
== isl_dim_in
)
794 return FN(PW
,free
)(pw
);
796 for (i
= pw
->n
- 1; i
>= 0; --i
) {
797 pw
->p
[i
].set
= isl_set_fix_si(pw
->p
[i
].set
, type
, pos
, value
);
798 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
799 return FN(PW
,free
)(pw
);
805 /* Restrict the domain of "pw" by combining each cell
806 * with "set" through a call to "fn", where "fn" may be
807 * isl_set_intersect, isl_set_intersect_params or isl_set_subtract.
809 static __isl_give PW
*FN(PW
,restrict_domain_aligned
)(__isl_take PW
*pw
,
810 __isl_take isl_set
*set
,
811 __isl_give isl_set
*(*fn
)(__isl_take isl_set
*set1
,
812 __isl_take isl_set
*set2
))
828 for (i
= pw
->n
- 1; i
>= 0; --i
) {
829 pw
->p
[i
].set
= fn(pw
->p
[i
].set
, isl_set_copy(set
));
830 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
842 static __isl_give PW
*FN(PW
,intersect_domain_aligned
)(__isl_take PW
*pw
,
843 __isl_take isl_set
*set
)
845 return FN(PW
,restrict_domain_aligned
)(pw
, set
, &isl_set_intersect
);
848 __isl_give PW
*FN(PW
,intersect_domain
)(__isl_take PW
*pw
,
849 __isl_take isl_set
*context
)
851 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
852 &FN(PW
,intersect_domain_aligned
));
855 static __isl_give PW
*FN(PW
,intersect_params_aligned
)(__isl_take PW
*pw
,
856 __isl_take isl_set
*set
)
858 return FN(PW
,restrict_domain_aligned
)(pw
, set
,
859 &isl_set_intersect_params
);
862 /* Intersect the domain of "pw" with the parameter domain "context".
864 __isl_give PW
*FN(PW
,intersect_params
)(__isl_take PW
*pw
,
865 __isl_take isl_set
*context
)
867 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
868 &FN(PW
,intersect_params_aligned
));
871 /* Subtract "domain' from the domain of "pw", assuming their
872 * parameters have been aligned.
874 static __isl_give PW
*FN(PW
,subtract_domain_aligned
)(__isl_take PW
*pw
,
875 __isl_take isl_set
*domain
)
877 return FN(PW
,restrict_domain_aligned
)(pw
, domain
, &isl_set_subtract
);
880 /* Subtract "domain' from the domain of "pw".
882 __isl_give PW
*FN(PW
,subtract_domain
)(__isl_take PW
*pw
,
883 __isl_take isl_set
*domain
)
885 return FN(PW
,align_params_pw_set_and
)(pw
, domain
,
886 &FN(PW
,subtract_domain_aligned
));
889 /* Compute the gist of "pw" with respect to the domain constraints
890 * of "context" for the case where the domain of the last element
891 * of "pw" is equal to "context".
892 * Call "fn_el" to compute the gist of this element, replace
893 * its domain by the universe and drop all other elements
894 * as their domains are necessarily disjoint from "context".
896 static __isl_give PW
*FN(PW
,gist_last
)(__isl_take PW
*pw
,
897 __isl_take isl_set
*context
,
898 __isl_give EL
*(*fn_el
)(__isl_take EL
*el
, __isl_take isl_set
*set
))
903 for (i
= 0; i
< pw
->n
- 1; ++i
) {
904 isl_set_free(pw
->p
[i
].set
);
905 FN(EL
,free
)(pw
->p
[i
].FIELD
);
907 pw
->p
[0].FIELD
= pw
->p
[pw
->n
- 1].FIELD
;
908 pw
->p
[0].set
= pw
->p
[pw
->n
- 1].set
;
911 space
= isl_set_get_space(context
);
912 pw
->p
[0].FIELD
= fn_el(pw
->p
[0].FIELD
, context
);
913 context
= isl_set_universe(space
);
914 isl_set_free(pw
->p
[0].set
);
915 pw
->p
[0].set
= context
;
917 if (!pw
->p
[0].FIELD
|| !pw
->p
[0].set
)
918 return FN(PW
,free
)(pw
);
923 /* Compute the gist of "pw" with respect to the domain constraints
924 * of "context". Call "fn_el" to compute the gist of the elements
925 * and "fn_dom" to compute the gist of the domains.
927 * If the piecewise expression is empty or the context is the universe,
928 * then nothing can be simplified.
930 static __isl_give PW
*FN(PW
,gist_aligned
)(__isl_take PW
*pw
,
931 __isl_take isl_set
*context
,
932 __isl_give EL
*(*fn_el
)(__isl_take EL
*el
,
933 __isl_take isl_set
*set
),
934 __isl_give isl_set
*(*fn_dom
)(__isl_take isl_set
*set
,
935 __isl_take isl_basic_set
*bset
))
939 isl_basic_set
*hull
= NULL
;
945 isl_set_free(context
);
949 is_universe
= isl_set_plain_is_universe(context
);
953 isl_set_free(context
);
957 if (!isl_space_match(pw
->dim
, isl_dim_param
,
958 context
->dim
, isl_dim_param
)) {
959 pw
= FN(PW
,align_params
)(pw
, isl_set_get_space(context
));
960 context
= isl_set_align_params(context
, FN(PW
,get_space
)(pw
));
970 equal
= isl_set_plain_is_equal(pw
->p
[0].set
, context
);
974 return FN(PW
,gist_last
)(pw
, context
, fn_el
);
977 context
= isl_set_compute_divs(context
);
978 hull
= isl_set_simple_hull(isl_set_copy(context
));
980 for (i
= pw
->n
- 1; i
>= 0; --i
) {
984 if (i
== pw
->n
- 1) {
986 equal
= isl_set_plain_is_equal(pw
->p
[i
].set
, context
);
990 isl_basic_set_free(hull
);
991 return FN(PW
,gist_last
)(pw
, context
, fn_el
);
994 set_i
= isl_set_intersect(isl_set_copy(pw
->p
[i
].set
),
995 isl_set_copy(context
));
996 empty
= isl_set_plain_is_empty(set_i
);
997 pw
->p
[i
].FIELD
= fn_el(pw
->p
[i
].FIELD
, set_i
);
998 pw
->p
[i
].set
= fn_dom(pw
->p
[i
].set
, isl_basic_set_copy(hull
));
999 if (empty
< 0 || !pw
->p
[i
].FIELD
|| !pw
->p
[i
].set
)
1002 isl_set_free(pw
->p
[i
].set
);
1003 FN(EL
,free
)(pw
->p
[i
].FIELD
);
1005 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
1010 isl_basic_set_free(hull
);
1011 isl_set_free(context
);
1016 isl_basic_set_free(hull
);
1017 isl_set_free(context
);
1021 static __isl_give PW
*FN(PW
,gist_domain_aligned
)(__isl_take PW
*pw
,
1022 __isl_take isl_set
*set
)
1024 return FN(PW
,gist_aligned
)(pw
, set
, &FN(EL
,gist
),
1025 &isl_set_gist_basic_set
);
1028 __isl_give PW
*FN(PW
,gist
)(__isl_take PW
*pw
, __isl_take isl_set
*context
)
1030 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
1031 &FN(PW
,gist_domain_aligned
));
1034 static __isl_give PW
*FN(PW
,gist_params_aligned
)(__isl_take PW
*pw
,
1035 __isl_take isl_set
*set
)
1037 return FN(PW
,gist_aligned
)(pw
, set
, &FN(EL
,gist_params
),
1038 &isl_set_gist_params_basic_set
);
1041 __isl_give PW
*FN(PW
,gist_params
)(__isl_take PW
*pw
,
1042 __isl_take isl_set
*context
)
1044 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
1045 &FN(PW
,gist_params_aligned
));
1048 __isl_give PW
*FN(PW
,coalesce
)(__isl_take PW
*pw
)
1057 for (i
= pw
->n
- 1; i
>= 0; --i
) {
1058 for (j
= i
- 1; j
>= 0; --j
) {
1059 if (!FN(EL
,plain_is_equal
)(pw
->p
[i
].FIELD
,
1062 pw
->p
[j
].set
= isl_set_union(pw
->p
[j
].set
,
1064 FN(EL
,free
)(pw
->p
[i
].FIELD
);
1066 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
1072 pw
->p
[i
].set
= isl_set_coalesce(pw
->p
[i
].set
);
1083 isl_ctx
*FN(PW
,get_ctx
)(__isl_keep PW
*pw
)
1085 return pw
? isl_space_get_ctx(pw
->dim
) : NULL
;
1088 #ifndef NO_INVOLVES_DIMS
1089 int FN(PW
,involves_dims
)(__isl_keep PW
*pw
, enum isl_dim_type type
,
1090 unsigned first
, unsigned n
)
1093 enum isl_dim_type set_type
;
1097 if (pw
->n
== 0 || n
== 0)
1100 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1102 for (i
= 0; i
< pw
->n
; ++i
) {
1103 int involves
= FN(EL
,involves_dims
)(pw
->p
[i
].FIELD
,
1105 if (involves
< 0 || involves
)
1107 involves
= isl_set_involves_dims(pw
->p
[i
].set
,
1108 set_type
, first
, n
);
1109 if (involves
< 0 || involves
)
1116 __isl_give PW
*FN(PW
,set_dim_name
)(__isl_take PW
*pw
,
1117 enum isl_dim_type type
, unsigned pos
, const char *s
)
1120 enum isl_dim_type set_type
;
1122 pw
= FN(PW
,cow
)(pw
);
1126 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1128 pw
->dim
= isl_space_set_dim_name(pw
->dim
, type
, pos
, s
);
1132 for (i
= 0; i
< pw
->n
; ++i
) {
1133 pw
->p
[i
].set
= isl_set_set_dim_name(pw
->p
[i
].set
,
1137 pw
->p
[i
].FIELD
= FN(EL
,set_dim_name
)(pw
->p
[i
].FIELD
, type
, pos
, s
);
1138 if (!pw
->p
[i
].FIELD
)
1148 #ifndef NO_DROP_DIMS
1149 __isl_give PW
*FN(PW
,drop_dims
)(__isl_take PW
*pw
,
1150 enum isl_dim_type type
, unsigned first
, unsigned n
)
1153 enum isl_dim_type set_type
;
1157 if (n
== 0 && !isl_space_get_tuple_name(pw
->dim
, type
))
1160 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1162 pw
= FN(PW
,cow
)(pw
);
1165 pw
->dim
= isl_space_drop_dims(pw
->dim
, type
, first
, n
);
1168 for (i
= 0; i
< pw
->n
; ++i
) {
1169 pw
->p
[i
].FIELD
= FN(EL
,drop_dims
)(pw
->p
[i
].FIELD
, type
, first
, n
);
1170 if (!pw
->p
[i
].FIELD
)
1172 if (type
== isl_dim_out
)
1174 pw
->p
[i
].set
= isl_set_drop(pw
->p
[i
].set
, set_type
, first
, n
);
1185 /* This function is very similar to drop_dims.
1186 * The only difference is that the cells may still involve
1187 * the specified dimensions. They are removed using
1188 * isl_set_project_out instead of isl_set_drop.
1190 __isl_give PW
*FN(PW
,project_out
)(__isl_take PW
*pw
,
1191 enum isl_dim_type type
, unsigned first
, unsigned n
)
1194 enum isl_dim_type set_type
;
1198 if (n
== 0 && !isl_space_get_tuple_name(pw
->dim
, type
))
1201 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1203 pw
= FN(PW
,cow
)(pw
);
1206 pw
->dim
= isl_space_drop_dims(pw
->dim
, type
, first
, n
);
1209 for (i
= 0; i
< pw
->n
; ++i
) {
1210 pw
->p
[i
].set
= isl_set_project_out(pw
->p
[i
].set
,
1211 set_type
, first
, n
);
1214 pw
->p
[i
].FIELD
= FN(EL
,drop_dims
)(pw
->p
[i
].FIELD
, type
, first
, n
);
1215 if (!pw
->p
[i
].FIELD
)
1225 /* Project the domain of pw onto its parameter space.
1227 __isl_give PW
*FN(PW
,project_domain_on_params
)(__isl_take PW
*pw
)
1232 n
= FN(PW
,dim
)(pw
, isl_dim_in
);
1233 pw
= FN(PW
,project_out
)(pw
, isl_dim_in
, 0, n
);
1234 space
= FN(PW
,get_domain_space
)(pw
);
1235 space
= isl_space_params(space
);
1236 pw
= FN(PW
,reset_domain_space
)(pw
, space
);
1241 #ifndef NO_INSERT_DIMS
1242 __isl_give PW
*FN(PW
,insert_dims
)(__isl_take PW
*pw
, enum isl_dim_type type
,
1243 unsigned first
, unsigned n
)
1246 enum isl_dim_type set_type
;
1250 if (n
== 0 && !isl_space_is_named_or_nested(pw
->dim
, type
))
1253 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1255 pw
= FN(PW
,cow
)(pw
);
1259 pw
->dim
= isl_space_insert_dims(pw
->dim
, type
, first
, n
);
1263 for (i
= 0; i
< pw
->n
; ++i
) {
1264 pw
->p
[i
].set
= isl_set_insert_dims(pw
->p
[i
].set
,
1265 set_type
, first
, n
);
1268 pw
->p
[i
].FIELD
= FN(EL
,insert_dims
)(pw
->p
[i
].FIELD
,
1270 if (!pw
->p
[i
].FIELD
)
1281 __isl_give PW
*FN(PW
,fix_dim
)(__isl_take PW
*pw
,
1282 enum isl_dim_type type
, unsigned pos
, isl_int v
)
1289 if (type
== isl_dim_in
)
1292 pw
= FN(PW
,cow
)(pw
);
1295 for (i
= 0; i
< pw
->n
; ++i
) {
1296 pw
->p
[i
].set
= isl_set_fix(pw
->p
[i
].set
, type
, pos
, v
);
1297 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
1298 return FN(PW
,free
)(pw
);
1304 /* Fix the value of the variable at position "pos" of type "type" of "pw"
1305 * to be equal to "v".
1307 __isl_give PW
*FN(PW
,fix_val
)(__isl_take PW
*pw
,
1308 enum isl_dim_type type
, unsigned pos
, __isl_take isl_val
*v
)
1311 return FN(PW
,free
)(pw
);
1312 if (!isl_val_is_int(v
))
1313 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
1314 "expecting integer value", goto error
);
1316 pw
= FN(PW
,fix_dim
)(pw
, type
, pos
, v
->n
);
1322 return FN(PW
,free
)(pw
);
1325 unsigned FN(PW
,dim
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
1327 return pw
? isl_space_dim(pw
->dim
, type
) : 0;
1330 __isl_give PW
*FN(PW
,split_dims
)(__isl_take PW
*pw
,
1331 enum isl_dim_type type
, unsigned first
, unsigned n
)
1340 if (type
== isl_dim_in
)
1343 pw
= FN(PW
,cow
)(pw
);
1348 for (i
= 0; i
< pw
->n
; ++i
) {
1349 pw
->p
[i
].set
= isl_set_split_dims(pw
->p
[i
].set
, type
, first
, n
);
1361 /* Compute the maximal value attained by the piecewise quasipolynomial
1362 * on its domain or zero if the domain is empty.
1363 * In the worst case, the domain is scanned completely,
1364 * so the domain is assumed to be bounded.
1366 __isl_give isl_val
*FN(PW
,opt
)(__isl_take PW
*pw
, int max
)
1375 opt
= isl_val_zero(FN(PW
,get_ctx
)(pw
));
1380 opt
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[0].FIELD
),
1381 isl_set_copy(pw
->p
[0].set
), max
);
1382 for (i
= 1; i
< pw
->n
; ++i
) {
1384 opt_i
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
1385 isl_set_copy(pw
->p
[i
].set
), max
);
1387 opt
= isl_val_max(opt
, opt_i
);
1389 opt
= isl_val_min(opt
, opt_i
);
1396 __isl_give isl_val
*FN(PW
,max
)(__isl_take PW
*pw
)
1398 return FN(PW
,opt
)(pw
, 1);
1401 __isl_give isl_val
*FN(PW
,min
)(__isl_take PW
*pw
)
1403 return FN(PW
,opt
)(pw
, 0);
1407 __isl_give isl_space
*FN(PW
,get_space
)(__isl_keep PW
*pw
)
1409 return pw
? isl_space_copy(pw
->dim
) : NULL
;
1412 __isl_give isl_space
*FN(PW
,get_domain_space
)(__isl_keep PW
*pw
)
1414 return pw
? isl_space_domain(isl_space_copy(pw
->dim
)) : NULL
;
1417 /* Return the position of the dimension of the given type and name
1419 * Return -1 if no such dimension can be found.
1421 int FN(PW
,find_dim_by_name
)(__isl_keep PW
*pw
,
1422 enum isl_dim_type type
, const char *name
)
1426 return isl_space_find_dim_by_name(pw
->dim
, type
, name
);
1429 #ifndef NO_RESET_DIM
1430 /* Reset the space of "pw". Since we don't know if the elements
1431 * represent the spaces themselves or their domains, we pass along
1432 * both when we call their reset_space_and_domain.
1434 static __isl_give PW
*FN(PW
,reset_space_and_domain
)(__isl_take PW
*pw
,
1435 __isl_take isl_space
*space
, __isl_take isl_space
*domain
)
1439 pw
= FN(PW
,cow
)(pw
);
1440 if (!pw
|| !space
|| !domain
)
1443 for (i
= 0; i
< pw
->n
; ++i
) {
1444 pw
->p
[i
].set
= isl_set_reset_space(pw
->p
[i
].set
,
1445 isl_space_copy(domain
));
1448 pw
->p
[i
].FIELD
= FN(EL
,reset_space_and_domain
)(pw
->p
[i
].FIELD
,
1449 isl_space_copy(space
), isl_space_copy(domain
));
1450 if (!pw
->p
[i
].FIELD
)
1454 isl_space_free(domain
);
1456 isl_space_free(pw
->dim
);
1461 isl_space_free(domain
);
1462 isl_space_free(space
);
1467 __isl_give PW
*FN(PW
,reset_domain_space
)(__isl_take PW
*pw
,
1468 __isl_take isl_space
*domain
)
1472 space
= isl_space_extend_domain_with_range(isl_space_copy(domain
),
1473 FN(PW
,get_space
)(pw
));
1474 return FN(PW
,reset_space_and_domain
)(pw
, space
, domain
);
1477 __isl_give PW
*FN(PW
,reset_space
)(__isl_take PW
*pw
, __isl_take isl_space
*dim
)
1481 domain
= isl_space_domain(isl_space_copy(dim
));
1482 return FN(PW
,reset_space_and_domain
)(pw
, dim
, domain
);
1485 __isl_give PW
*FN(PW
,set_tuple_id
)(__isl_take PW
*pw
, enum isl_dim_type type
,
1486 __isl_take isl_id
*id
)
1490 pw
= FN(PW
,cow
)(pw
);
1494 space
= FN(PW
,get_space
)(pw
);
1495 space
= isl_space_set_tuple_id(space
, type
, id
);
1497 return FN(PW
,reset_space
)(pw
, space
);
1500 return FN(PW
,free
)(pw
);
1503 /* Drop the id on the specified tuple.
1505 __isl_give PW
*FN(PW
,reset_tuple_id
)(__isl_take PW
*pw
, enum isl_dim_type type
)
1511 if (!FN(PW
,has_tuple_id
)(pw
, type
))
1514 pw
= FN(PW
,cow
)(pw
);
1518 space
= FN(PW
,get_space
)(pw
);
1519 space
= isl_space_reset_tuple_id(space
, type
);
1521 return FN(PW
,reset_space
)(pw
, space
);
1524 __isl_give PW
*FN(PW
,set_dim_id
)(__isl_take PW
*pw
,
1525 enum isl_dim_type type
, unsigned pos
, __isl_take isl_id
*id
)
1527 pw
= FN(PW
,cow
)(pw
);
1530 pw
->dim
= isl_space_set_dim_id(pw
->dim
, type
, pos
, id
);
1531 return FN(PW
,reset_space
)(pw
, isl_space_copy(pw
->dim
));
1534 return FN(PW
,free
)(pw
);
1538 /* Reset the user pointer on all identifiers of parameters and tuples
1539 * of the space of "pw".
1541 __isl_give PW
*FN(PW
,reset_user
)(__isl_take PW
*pw
)
1545 space
= FN(PW
,get_space
)(pw
);
1546 space
= isl_space_reset_user(space
);
1548 return FN(PW
,reset_space
)(pw
, space
);
1551 int FN(PW
,has_equal_space
)(__isl_keep PW
*pw1
, __isl_keep PW
*pw2
)
1556 return isl_space_is_equal(pw1
->dim
, pw2
->dim
);
1560 __isl_give PW
*FN(PW
,morph_domain
)(__isl_take PW
*pw
,
1561 __isl_take isl_morph
*morph
)
1569 ctx
= isl_space_get_ctx(pw
->dim
);
1570 isl_assert(ctx
, isl_space_is_domain_internal(morph
->dom
->dim
, pw
->dim
),
1573 pw
= FN(PW
,cow
)(pw
);
1576 pw
->dim
= isl_space_extend_domain_with_range(
1577 isl_space_copy(morph
->ran
->dim
), pw
->dim
);
1581 for (i
= 0; i
< pw
->n
; ++i
) {
1582 pw
->p
[i
].set
= isl_morph_set(isl_morph_copy(morph
), pw
->p
[i
].set
);
1585 pw
->p
[i
].FIELD
= FN(EL
,morph_domain
)(pw
->p
[i
].FIELD
,
1586 isl_morph_copy(morph
));
1587 if (!pw
->p
[i
].FIELD
)
1591 isl_morph_free(morph
);
1596 isl_morph_free(morph
);
1601 int FN(PW
,n_piece
)(__isl_keep PW
*pw
)
1603 return pw
? pw
->n
: 0;
1606 int FN(PW
,foreach_piece
)(__isl_keep PW
*pw
,
1607 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
, void *user
),
1615 for (i
= 0; i
< pw
->n
; ++i
)
1616 if (fn(isl_set_copy(pw
->p
[i
].set
),
1617 FN(EL
,copy
)(pw
->p
[i
].FIELD
), user
) < 0)
1624 static int any_divs(__isl_keep isl_set
*set
)
1631 for (i
= 0; i
< set
->n
; ++i
)
1632 if (set
->p
[i
]->n_div
> 0)
1638 static int foreach_lifted_subset(__isl_take isl_set
*set
, __isl_take EL
*el
,
1639 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
1640 void *user
), void *user
)
1647 for (i
= 0; i
< set
->n
; ++i
) {
1651 lift
= isl_set_from_basic_set(isl_basic_set_copy(set
->p
[i
]));
1652 lift
= isl_set_lift(lift
);
1654 copy
= FN(EL
,copy
)(el
);
1655 copy
= FN(EL
,lift
)(copy
, isl_set_get_space(lift
));
1657 if (fn(lift
, copy
, user
) < 0)
1671 int FN(PW
,foreach_lifted_piece
)(__isl_keep PW
*pw
,
1672 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
1673 void *user
), void *user
)
1680 for (i
= 0; i
< pw
->n
; ++i
) {
1684 set
= isl_set_copy(pw
->p
[i
].set
);
1685 el
= FN(EL
,copy
)(pw
->p
[i
].FIELD
);
1686 if (!any_divs(set
)) {
1687 if (fn(set
, el
, user
) < 0)
1691 if (foreach_lifted_subset(set
, el
, fn
, user
) < 0)
1699 #ifndef NO_MOVE_DIMS
1700 __isl_give PW
*FN(PW
,move_dims
)(__isl_take PW
*pw
,
1701 enum isl_dim_type dst_type
, unsigned dst_pos
,
1702 enum isl_dim_type src_type
, unsigned src_pos
, unsigned n
)
1706 pw
= FN(PW
,cow
)(pw
);
1710 pw
->dim
= isl_space_move_dims(pw
->dim
, dst_type
, dst_pos
, src_type
, src_pos
, n
);
1714 for (i
= 0; i
< pw
->n
; ++i
) {
1715 pw
->p
[i
].FIELD
= FN(EL
,move_dims
)(pw
->p
[i
].FIELD
,
1716 dst_type
, dst_pos
, src_type
, src_pos
, n
);
1717 if (!pw
->p
[i
].FIELD
)
1721 if (dst_type
== isl_dim_in
)
1722 dst_type
= isl_dim_set
;
1723 if (src_type
== isl_dim_in
)
1724 src_type
= isl_dim_set
;
1726 for (i
= 0; i
< pw
->n
; ++i
) {
1727 pw
->p
[i
].set
= isl_set_move_dims(pw
->p
[i
].set
,
1729 src_type
, src_pos
, n
);
1741 __isl_give PW
*FN(PW
,mul_isl_int
)(__isl_take PW
*pw
, isl_int v
)
1745 if (isl_int_is_one(v
))
1747 if (pw
&& DEFAULT_IS_ZERO
&& isl_int_is_zero(v
)) {
1749 isl_space
*dim
= FN(PW
,get_space
)(pw
);
1751 zero
= FN(PW
,ZERO
)(dim
, pw
->type
);
1753 zero
= FN(PW
,ZERO
)(dim
);
1758 pw
= FN(PW
,cow
)(pw
);
1765 if (isl_int_is_neg(v
))
1766 pw
->type
= isl_fold_type_negate(pw
->type
);
1768 for (i
= 0; i
< pw
->n
; ++i
) {
1769 pw
->p
[i
].FIELD
= FN(EL
,scale
)(pw
->p
[i
].FIELD
, v
);
1770 if (!pw
->p
[i
].FIELD
)
1780 /* Multiply the pieces of "pw" by "v" and return the result.
1782 __isl_give PW
*FN(PW
,scale_val
)(__isl_take PW
*pw
, __isl_take isl_val
*v
)
1789 if (isl_val_is_one(v
)) {
1793 if (pw
&& DEFAULT_IS_ZERO
&& isl_val_is_zero(v
)) {
1795 isl_space
*space
= FN(PW
,get_space
)(pw
);
1797 zero
= FN(PW
,ZERO
)(space
, pw
->type
);
1799 zero
= FN(PW
,ZERO
)(space
);
1809 pw
= FN(PW
,cow
)(pw
);
1814 if (isl_val_is_neg(v
))
1815 pw
->type
= isl_fold_type_negate(pw
->type
);
1817 for (i
= 0; i
< pw
->n
; ++i
) {
1818 pw
->p
[i
].FIELD
= FN(EL
,scale_val
)(pw
->p
[i
].FIELD
,
1820 if (!pw
->p
[i
].FIELD
)
1832 /* Divide the pieces of "pw" by "v" and return the result.
1834 __isl_give PW
*FN(PW
,scale_down_val
)(__isl_take PW
*pw
, __isl_take isl_val
*v
)
1841 if (isl_val_is_one(v
)) {
1846 if (!isl_val_is_rat(v
))
1847 isl_die(isl_val_get_ctx(v
), isl_error_invalid
,
1848 "expecting rational factor", goto error
);
1849 if (isl_val_is_zero(v
))
1850 isl_die(isl_val_get_ctx(v
), isl_error_invalid
,
1851 "cannot scale down by zero", goto error
);
1857 pw
= FN(PW
,cow
)(pw
);
1862 if (isl_val_is_neg(v
))
1863 pw
->type
= isl_fold_type_negate(pw
->type
);
1865 for (i
= 0; i
< pw
->n
; ++i
) {
1866 pw
->p
[i
].FIELD
= FN(EL
,scale_down_val
)(pw
->p
[i
].FIELD
,
1868 if (!pw
->p
[i
].FIELD
)
1880 __isl_give PW
*FN(PW
,scale
)(__isl_take PW
*pw
, isl_int v
)
1882 return FN(PW
,mul_isl_int
)(pw
, v
);
1885 static int FN(PW
,qsort_set_cmp
)(const void *p1
, const void *p2
)
1887 isl_set
*set1
= *(isl_set
* const *)p1
;
1888 isl_set
*set2
= *(isl_set
* const *)p2
;
1890 return isl_set_plain_cmp(set1
, set2
);
1893 /* We normalize in place, but if anything goes wrong we need
1894 * to return NULL, so we need to make sure we don't change the
1895 * meaning of any possible other copies of map.
1897 __isl_give PW
*FN(PW
,normalize
)(__isl_take PW
*pw
)
1904 for (i
= 0; i
< pw
->n
; ++i
) {
1905 set
= isl_set_normalize(isl_set_copy(pw
->p
[i
].set
));
1907 return FN(PW
,free
)(pw
);
1908 isl_set_free(pw
->p
[i
].set
);
1911 qsort(pw
->p
, pw
->n
, sizeof(pw
->p
[0]), &FN(PW
,qsort_set_cmp
));
1912 for (i
= pw
->n
- 1; i
>= 1; --i
) {
1913 if (!isl_set_plain_is_equal(pw
->p
[i
- 1].set
, pw
->p
[i
].set
))
1915 if (!FN(EL
,plain_is_equal
)(pw
->p
[i
- 1].FIELD
, pw
->p
[i
].FIELD
))
1917 set
= isl_set_union(isl_set_copy(pw
->p
[i
- 1].set
),
1918 isl_set_copy(pw
->p
[i
].set
));
1920 return FN(PW
,free
)(pw
);
1921 isl_set_free(pw
->p
[i
].set
);
1922 FN(EL
,free
)(pw
->p
[i
].FIELD
);
1923 isl_set_free(pw
->p
[i
- 1].set
);
1924 pw
->p
[i
- 1].set
= set
;
1925 for (j
= i
+ 1; j
< pw
->n
; ++j
)
1926 pw
->p
[j
- 1] = pw
->p
[j
];
1933 /* Is pw1 obviously equal to pw2?
1934 * That is, do they have obviously identical cells and obviously identical
1935 * elements on each cell?
1937 int FN(PW
,plain_is_equal
)(__isl_keep PW
*pw1
, __isl_keep PW
*pw2
)
1947 if (!isl_space_is_equal(pw1
->dim
, pw2
->dim
))
1950 pw1
= FN(PW
,copy
)(pw1
);
1951 pw2
= FN(PW
,copy
)(pw2
);
1952 pw1
= FN(PW
,normalize
)(pw1
);
1953 pw2
= FN(PW
,normalize
)(pw2
);
1957 equal
= pw1
->n
== pw2
->n
;
1958 for (i
= 0; equal
&& i
< pw1
->n
; ++i
) {
1959 equal
= isl_set_plain_is_equal(pw1
->p
[i
].set
, pw2
->p
[i
].set
);
1964 equal
= FN(EL
,plain_is_equal
)(pw1
->p
[i
].FIELD
, pw2
->p
[i
].FIELD
);
1979 static __isl_give PW
*FN(PW
,align_params_pw_multi_aff_and
)(__isl_take PW
*pw
,
1980 __isl_take isl_multi_aff
*ma
,
1981 __isl_give PW
*(*fn
)(__isl_take PW
*pw
, __isl_take isl_multi_aff
*ma
))
1984 isl_space
*ma_space
;
1986 ma_space
= isl_multi_aff_get_space(ma
);
1987 if (!pw
|| !ma
|| !ma_space
)
1989 if (isl_space_match(pw
->dim
, isl_dim_param
, ma_space
, isl_dim_param
)) {
1990 isl_space_free(ma_space
);
1993 ctx
= FN(PW
,get_ctx
)(pw
);
1994 if (!isl_space_has_named_params(pw
->dim
) ||
1995 !isl_space_has_named_params(ma_space
))
1996 isl_die(ctx
, isl_error_invalid
,
1997 "unaligned unnamed parameters", goto error
);
1998 pw
= FN(PW
,align_params
)(pw
, ma_space
);
1999 ma
= isl_multi_aff_align_params(ma
, FN(PW
,get_space
)(pw
));
2002 isl_space_free(ma_space
);
2004 isl_multi_aff_free(ma
);
2008 static __isl_give PW
*FN(PW
,align_params_pw_pw_multi_aff_and
)(__isl_take PW
*pw
,
2009 __isl_take isl_pw_multi_aff
*pma
,
2010 __isl_give PW
*(*fn
)(__isl_take PW
*pw
,
2011 __isl_take isl_pw_multi_aff
*ma
))
2014 isl_space
*pma_space
;
2016 pma_space
= isl_pw_multi_aff_get_space(pma
);
2017 if (!pw
|| !pma
|| !pma_space
)
2019 if (isl_space_match(pw
->dim
, isl_dim_param
, pma_space
, isl_dim_param
)) {
2020 isl_space_free(pma_space
);
2023 ctx
= FN(PW
,get_ctx
)(pw
);
2024 if (!isl_space_has_named_params(pw
->dim
) ||
2025 !isl_space_has_named_params(pma_space
))
2026 isl_die(ctx
, isl_error_invalid
,
2027 "unaligned unnamed parameters", goto error
);
2028 pw
= FN(PW
,align_params
)(pw
, pma_space
);
2029 pma
= isl_pw_multi_aff_align_params(pma
, FN(PW
,get_space
)(pw
));
2032 isl_space_free(pma_space
);
2034 isl_pw_multi_aff_free(pma
);
2038 /* Compute the pullback of "pw" by the function represented by "ma".
2039 * In other words, plug in "ma" in "pw".
2041 static __isl_give PW
*FN(PW
,pullback_multi_aff_aligned
)(__isl_take PW
*pw
,
2042 __isl_take isl_multi_aff
*ma
)
2045 isl_space
*space
= NULL
;
2047 ma
= isl_multi_aff_align_divs(ma
);
2048 pw
= FN(PW
,cow
)(pw
);
2052 space
= isl_space_join(isl_multi_aff_get_space(ma
),
2053 FN(PW
,get_space
)(pw
));
2055 for (i
= 0; i
< pw
->n
; ++i
) {
2056 pw
->p
[i
].set
= isl_set_preimage_multi_aff(pw
->p
[i
].set
,
2057 isl_multi_aff_copy(ma
));
2060 pw
->p
[i
].FIELD
= FN(EL
,pullback_multi_aff
)(pw
->p
[i
].FIELD
,
2061 isl_multi_aff_copy(ma
));
2062 if (!pw
->p
[i
].FIELD
)
2066 pw
= FN(PW
,reset_space
)(pw
, space
);
2067 isl_multi_aff_free(ma
);
2070 isl_space_free(space
);
2071 isl_multi_aff_free(ma
);
2076 __isl_give PW
*FN(PW
,pullback_multi_aff
)(__isl_take PW
*pw
,
2077 __isl_take isl_multi_aff
*ma
)
2079 return FN(PW
,align_params_pw_multi_aff_and
)(pw
, ma
,
2080 &FN(PW
,pullback_multi_aff_aligned
));
2083 /* Compute the pullback of "pw" by the function represented by "pma".
2084 * In other words, plug in "pma" in "pw".
2086 static __isl_give PW
*FN(PW
,pullback_pw_multi_aff_aligned
)(__isl_take PW
*pw
,
2087 __isl_take isl_pw_multi_aff
*pma
)
2097 space
= isl_space_join(isl_pw_multi_aff_get_space(pma
),
2098 FN(PW
,get_space
)(pw
));
2099 isl_pw_multi_aff_free(pma
);
2100 res
= FN(PW
,empty
)(space
);
2105 res
= FN(PW
,pullback_multi_aff
)(FN(PW
,copy
)(pw
),
2106 isl_multi_aff_copy(pma
->p
[0].maff
));
2107 res
= FN(PW
,intersect_domain
)(res
, isl_set_copy(pma
->p
[0].set
));
2109 for (i
= 1; i
< pma
->n
; ++i
) {
2112 res_i
= FN(PW
,pullback_multi_aff
)(FN(PW
,copy
)(pw
),
2113 isl_multi_aff_copy(pma
->p
[i
].maff
));
2114 res_i
= FN(PW
,intersect_domain
)(res_i
,
2115 isl_set_copy(pma
->p
[i
].set
));
2116 res
= FN(PW
,add_disjoint
)(res
, res_i
);
2119 isl_pw_multi_aff_free(pma
);
2123 isl_pw_multi_aff_free(pma
);
2128 __isl_give PW
*FN(PW
,pullback_pw_multi_aff
)(__isl_take PW
*pw
,
2129 __isl_take isl_pw_multi_aff
*pma
)
2131 return FN(PW
,align_params_pw_pw_multi_aff_and
)(pw
, pma
,
2132 &FN(PW
,pullback_pw_multi_aff_aligned
));