2 * Copyright 2010-2011 INRIA Saclay
3 * Copyright 2011 Sven Verdoolaege
4 * Copyright 2012-2013 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
);
648 __isl_give PW
*FN(PW
,sub
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
650 return FN(PW
,add
)(pw1
, FN(PW
,neg
)(pw2
));
655 __isl_give isl_val
*FN(PW
,eval
)(__isl_take PW
*pw
, __isl_take isl_point
*pnt
)
660 isl_space
*pnt_dim
= NULL
;
665 ctx
= isl_point_get_ctx(pnt
);
666 pnt_dim
= isl_point_get_space(pnt
);
667 isl_assert(ctx
, isl_space_is_domain_internal(pnt_dim
, pw
->dim
),
670 for (i
= 0; i
< pw
->n
; ++i
) {
671 found
= isl_set_contains_point(pw
->p
[i
].set
, pnt
);
678 v
= FN(EL
,eval
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
679 isl_point_copy(pnt
));
681 v
= isl_val_zero(ctx
);
683 isl_space_free(pnt_dim
);
688 isl_space_free(pnt_dim
);
694 /* Return the parameter domain of "pw".
696 __isl_give isl_set
*FN(PW
,params
)(__isl_take PW
*pw
)
698 return isl_set_params(FN(PW
,domain
)(pw
));
701 __isl_give isl_set
*FN(PW
,domain
)(__isl_take PW
*pw
)
709 dom
= isl_set_empty(FN(PW
,get_domain_space
)(pw
));
710 for (i
= 0; i
< pw
->n
; ++i
)
711 dom
= isl_set_union_disjoint(dom
, isl_set_copy(pw
->p
[i
].set
));
718 /* Exploit the equalities in the domain of piece "i" of "pw"
719 * to simplify the associated function.
720 * If the domain of piece "i" is empty, then remove it entirely,
721 * replacing it with the final piece.
723 static int FN(PW
,exploit_equalities_and_remove_if_empty
)(__isl_keep PW
*pw
,
727 int empty
= isl_set_plain_is_empty(pw
->p
[i
].set
);
732 isl_set_free(pw
->p
[i
].set
);
733 FN(EL
,free
)(pw
->p
[i
].FIELD
);
735 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
741 aff
= isl_set_affine_hull(isl_set_copy(pw
->p
[i
].set
));
742 pw
->p
[i
].FIELD
= FN(EL
,substitute_equalities
)(pw
->p
[i
].FIELD
, aff
);
749 /* Convert a piecewise expression defined over a parameter domain
750 * into one that is defined over a zero-dimensional set.
752 __isl_give PW
*FN(PW
,from_range
)(__isl_take PW
*pw
)
758 if (!isl_space_is_set(pw
->dim
))
759 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
760 "not living in a set space", return FN(PW
,free
)(pw
));
762 space
= FN(PW
,get_space
)(pw
);
763 space
= isl_space_from_range(space
);
764 pw
= FN(PW
,reset_space
)(pw
, space
);
769 /* Fix the value of the given parameter or domain dimension of "pw"
770 * to be equal to "value".
772 __isl_give PW
*FN(PW
,fix_si
)(__isl_take PW
*pw
, enum isl_dim_type type
,
773 unsigned pos
, int value
)
780 if (type
== isl_dim_out
)
781 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
782 "cannot fix output dimension", return FN(PW
,free
)(pw
));
787 if (type
== isl_dim_in
)
792 return FN(PW
,free
)(pw
);
794 for (i
= pw
->n
- 1; i
>= 0; --i
) {
795 pw
->p
[i
].set
= isl_set_fix_si(pw
->p
[i
].set
, type
, pos
, value
);
796 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
797 return FN(PW
,free
)(pw
);
803 /* Restrict the domain of "pw" by combining each cell
804 * with "set" through a call to "fn", where "fn" may be
805 * isl_set_intersect or isl_set_intersect_params.
807 static __isl_give PW
*FN(PW
,intersect_aligned
)(__isl_take PW
*pw
,
808 __isl_take isl_set
*set
,
809 __isl_give isl_set
*(*fn
)(__isl_take isl_set
*set1
,
810 __isl_take isl_set
*set2
))
826 for (i
= pw
->n
- 1; i
>= 0; --i
) {
827 pw
->p
[i
].set
= fn(pw
->p
[i
].set
, isl_set_copy(set
));
828 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
840 static __isl_give PW
*FN(PW
,intersect_domain_aligned
)(__isl_take PW
*pw
,
841 __isl_take isl_set
*set
)
843 return FN(PW
,intersect_aligned
)(pw
, set
, &isl_set_intersect
);
846 __isl_give PW
*FN(PW
,intersect_domain
)(__isl_take PW
*pw
,
847 __isl_take isl_set
*context
)
849 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
850 &FN(PW
,intersect_domain_aligned
));
853 static __isl_give PW
*FN(PW
,intersect_params_aligned
)(__isl_take PW
*pw
,
854 __isl_take isl_set
*set
)
856 return FN(PW
,intersect_aligned
)(pw
, set
, &isl_set_intersect_params
);
859 /* Intersect the domain of "pw" with the parameter domain "context".
861 __isl_give PW
*FN(PW
,intersect_params
)(__isl_take PW
*pw
,
862 __isl_take isl_set
*context
)
864 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
865 &FN(PW
,intersect_params_aligned
));
868 static __isl_give PW
*FN(PW
,gist_aligned
)(__isl_take PW
*pw
,
869 __isl_take isl_set
*context
,
870 __isl_give EL
*(*fn_el
)(__isl_take EL
*el
,
871 __isl_take isl_set
*set
),
872 __isl_give isl_set
*(*fn_dom
)(__isl_take isl_set
*set
,
873 __isl_take isl_basic_set
*bset
))
876 isl_basic_set
*hull
= NULL
;
882 isl_set_free(context
);
886 if (!isl_space_match(pw
->dim
, isl_dim_param
,
887 context
->dim
, isl_dim_param
)) {
888 pw
= FN(PW
,align_params
)(pw
, isl_set_get_space(context
));
889 context
= isl_set_align_params(context
, FN(PW
,get_space
)(pw
));
892 context
= isl_set_compute_divs(context
);
893 hull
= isl_set_simple_hull(isl_set_copy(context
));
899 for (i
= pw
->n
- 1; i
>= 0; --i
) {
903 set_i
= isl_set_intersect(isl_set_copy(pw
->p
[i
].set
),
904 isl_set_copy(context
));
905 empty
= isl_set_plain_is_empty(set_i
);
906 pw
->p
[i
].FIELD
= fn_el(pw
->p
[i
].FIELD
, set_i
);
907 pw
->p
[i
].set
= fn_dom(pw
->p
[i
].set
, isl_basic_set_copy(hull
));
908 if (!pw
->p
[i
].FIELD
|| !pw
->p
[i
].set
)
911 isl_set_free(pw
->p
[i
].set
);
912 FN(EL
,free
)(pw
->p
[i
].FIELD
);
914 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
919 isl_basic_set_free(hull
);
920 isl_set_free(context
);
925 isl_basic_set_free(hull
);
926 isl_set_free(context
);
930 static __isl_give PW
*FN(PW
,gist_domain_aligned
)(__isl_take PW
*pw
,
931 __isl_take isl_set
*set
)
933 return FN(PW
,gist_aligned
)(pw
, set
, &FN(EL
,gist
),
934 &isl_set_gist_basic_set
);
937 __isl_give PW
*FN(PW
,gist
)(__isl_take PW
*pw
, __isl_take isl_set
*context
)
939 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
940 &FN(PW
,gist_domain_aligned
));
943 static __isl_give PW
*FN(PW
,gist_params_aligned
)(__isl_take PW
*pw
,
944 __isl_take isl_set
*set
)
946 return FN(PW
,gist_aligned
)(pw
, set
, &FN(EL
,gist_params
),
947 &isl_set_gist_params_basic_set
);
950 __isl_give PW
*FN(PW
,gist_params
)(__isl_take PW
*pw
,
951 __isl_take isl_set
*context
)
953 return FN(PW
,align_params_pw_set_and
)(pw
, context
,
954 &FN(PW
,gist_params_aligned
));
957 __isl_give PW
*FN(PW
,coalesce
)(__isl_take PW
*pw
)
966 for (i
= pw
->n
- 1; i
>= 0; --i
) {
967 for (j
= i
- 1; j
>= 0; --j
) {
968 if (!FN(EL
,plain_is_equal
)(pw
->p
[i
].FIELD
,
971 pw
->p
[j
].set
= isl_set_union(pw
->p
[j
].set
,
973 FN(EL
,free
)(pw
->p
[i
].FIELD
);
975 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
981 pw
->p
[i
].set
= isl_set_coalesce(pw
->p
[i
].set
);
992 isl_ctx
*FN(PW
,get_ctx
)(__isl_keep PW
*pw
)
994 return pw
? isl_space_get_ctx(pw
->dim
) : NULL
;
997 #ifndef NO_INVOLVES_DIMS
998 int FN(PW
,involves_dims
)(__isl_keep PW
*pw
, enum isl_dim_type type
,
999 unsigned first
, unsigned n
)
1002 enum isl_dim_type set_type
;
1006 if (pw
->n
== 0 || n
== 0)
1009 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1011 for (i
= 0; i
< pw
->n
; ++i
) {
1012 int involves
= FN(EL
,involves_dims
)(pw
->p
[i
].FIELD
,
1014 if (involves
< 0 || involves
)
1016 involves
= isl_set_involves_dims(pw
->p
[i
].set
,
1017 set_type
, first
, n
);
1018 if (involves
< 0 || involves
)
1025 __isl_give PW
*FN(PW
,set_dim_name
)(__isl_take PW
*pw
,
1026 enum isl_dim_type type
, unsigned pos
, const char *s
)
1029 enum isl_dim_type set_type
;
1031 pw
= FN(PW
,cow
)(pw
);
1035 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1037 pw
->dim
= isl_space_set_dim_name(pw
->dim
, type
, pos
, s
);
1041 for (i
= 0; i
< pw
->n
; ++i
) {
1042 pw
->p
[i
].set
= isl_set_set_dim_name(pw
->p
[i
].set
,
1046 pw
->p
[i
].FIELD
= FN(EL
,set_dim_name
)(pw
->p
[i
].FIELD
, type
, pos
, s
);
1047 if (!pw
->p
[i
].FIELD
)
1057 #ifndef NO_DROP_DIMS
1058 __isl_give PW
*FN(PW
,drop_dims
)(__isl_take PW
*pw
,
1059 enum isl_dim_type type
, unsigned first
, unsigned n
)
1062 enum isl_dim_type set_type
;
1066 if (n
== 0 && !isl_space_get_tuple_name(pw
->dim
, type
))
1069 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1071 pw
= FN(PW
,cow
)(pw
);
1074 pw
->dim
= isl_space_drop_dims(pw
->dim
, type
, first
, n
);
1077 for (i
= 0; i
< pw
->n
; ++i
) {
1078 pw
->p
[i
].FIELD
= FN(EL
,drop_dims
)(pw
->p
[i
].FIELD
, type
, first
, n
);
1079 if (!pw
->p
[i
].FIELD
)
1081 if (type
== isl_dim_out
)
1083 pw
->p
[i
].set
= isl_set_drop(pw
->p
[i
].set
, set_type
, first
, n
);
1094 /* This function is very similar to drop_dims.
1095 * The only difference is that the cells may still involve
1096 * the specified dimensions. They are removed using
1097 * isl_set_project_out instead of isl_set_drop.
1099 __isl_give PW
*FN(PW
,project_out
)(__isl_take PW
*pw
,
1100 enum isl_dim_type type
, unsigned first
, unsigned n
)
1103 enum isl_dim_type set_type
;
1107 if (n
== 0 && !isl_space_get_tuple_name(pw
->dim
, type
))
1110 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1112 pw
= FN(PW
,cow
)(pw
);
1115 pw
->dim
= isl_space_drop_dims(pw
->dim
, type
, first
, n
);
1118 for (i
= 0; i
< pw
->n
; ++i
) {
1119 pw
->p
[i
].set
= isl_set_project_out(pw
->p
[i
].set
,
1120 set_type
, first
, n
);
1123 pw
->p
[i
].FIELD
= FN(EL
,drop_dims
)(pw
->p
[i
].FIELD
, type
, first
, n
);
1124 if (!pw
->p
[i
].FIELD
)
1134 /* Project the domain of pw onto its parameter space.
1136 __isl_give PW
*FN(PW
,project_domain_on_params
)(__isl_take PW
*pw
)
1141 n
= FN(PW
,dim
)(pw
, isl_dim_in
);
1142 pw
= FN(PW
,project_out
)(pw
, isl_dim_in
, 0, n
);
1143 space
= FN(PW
,get_domain_space
)(pw
);
1144 space
= isl_space_params(space
);
1145 pw
= FN(PW
,reset_domain_space
)(pw
, space
);
1150 #ifndef NO_INSERT_DIMS
1151 __isl_give PW
*FN(PW
,insert_dims
)(__isl_take PW
*pw
, enum isl_dim_type type
,
1152 unsigned first
, unsigned n
)
1155 enum isl_dim_type set_type
;
1159 if (n
== 0 && !isl_space_is_named_or_nested(pw
->dim
, type
))
1162 set_type
= type
== isl_dim_in
? isl_dim_set
: type
;
1164 pw
= FN(PW
,cow
)(pw
);
1168 pw
->dim
= isl_space_insert_dims(pw
->dim
, type
, first
, n
);
1172 for (i
= 0; i
< pw
->n
; ++i
) {
1173 pw
->p
[i
].set
= isl_set_insert_dims(pw
->p
[i
].set
,
1174 set_type
, first
, n
);
1177 pw
->p
[i
].FIELD
= FN(EL
,insert_dims
)(pw
->p
[i
].FIELD
,
1179 if (!pw
->p
[i
].FIELD
)
1190 __isl_give PW
*FN(PW
,fix_dim
)(__isl_take PW
*pw
,
1191 enum isl_dim_type type
, unsigned pos
, isl_int v
)
1198 if (type
== isl_dim_in
)
1201 pw
= FN(PW
,cow
)(pw
);
1204 for (i
= 0; i
< pw
->n
; ++i
) {
1205 pw
->p
[i
].set
= isl_set_fix(pw
->p
[i
].set
, type
, pos
, v
);
1206 if (FN(PW
,exploit_equalities_and_remove_if_empty
)(pw
, i
) < 0)
1207 return FN(PW
,free
)(pw
);
1213 /* Fix the value of the variable at position "pos" of type "type" of "pw"
1214 * to be equal to "v".
1216 __isl_give PW
*FN(PW
,fix_val
)(__isl_take PW
*pw
,
1217 enum isl_dim_type type
, unsigned pos
, __isl_take isl_val
*v
)
1220 return FN(PW
,free
)(pw
);
1221 if (!isl_val_is_int(v
))
1222 isl_die(FN(PW
,get_ctx
)(pw
), isl_error_invalid
,
1223 "expecting integer value", goto error
);
1225 pw
= FN(PW
,fix_dim
)(pw
, type
, pos
, v
->n
);
1231 return FN(PW
,free
)(pw
);
1234 unsigned FN(PW
,dim
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
1236 return pw
? isl_space_dim(pw
->dim
, type
) : 0;
1239 __isl_give PW
*FN(PW
,split_dims
)(__isl_take PW
*pw
,
1240 enum isl_dim_type type
, unsigned first
, unsigned n
)
1249 if (type
== isl_dim_in
)
1252 pw
= FN(PW
,cow
)(pw
);
1257 for (i
= 0; i
< pw
->n
; ++i
) {
1258 pw
->p
[i
].set
= isl_set_split_dims(pw
->p
[i
].set
, type
, first
, n
);
1270 /* Compute the maximal value attained by the piecewise quasipolynomial
1271 * on its domain or zero if the domain is empty.
1272 * In the worst case, the domain is scanned completely,
1273 * so the domain is assumed to be bounded.
1275 __isl_give isl_val
*FN(PW
,opt
)(__isl_take PW
*pw
, int max
)
1284 opt
= isl_val_zero(FN(PW
,get_ctx
)(pw
));
1289 opt
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[0].FIELD
),
1290 isl_set_copy(pw
->p
[0].set
), max
);
1291 for (i
= 1; i
< pw
->n
; ++i
) {
1293 opt_i
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
1294 isl_set_copy(pw
->p
[i
].set
), max
);
1296 opt
= isl_val_max(opt
, opt_i
);
1298 opt
= isl_val_min(opt
, opt_i
);
1305 __isl_give isl_val
*FN(PW
,max
)(__isl_take PW
*pw
)
1307 return FN(PW
,opt
)(pw
, 1);
1310 __isl_give isl_val
*FN(PW
,min
)(__isl_take PW
*pw
)
1312 return FN(PW
,opt
)(pw
, 0);
1316 __isl_give isl_space
*FN(PW
,get_space
)(__isl_keep PW
*pw
)
1318 return pw
? isl_space_copy(pw
->dim
) : NULL
;
1321 __isl_give isl_space
*FN(PW
,get_domain_space
)(__isl_keep PW
*pw
)
1323 return pw
? isl_space_domain(isl_space_copy(pw
->dim
)) : NULL
;
1326 #ifndef NO_RESET_DIM
1327 /* Reset the space of "pw". Since we don't know if the elements
1328 * represent the spaces themselves or their domains, we pass along
1329 * both when we call their reset_space_and_domain.
1331 static __isl_give PW
*FN(PW
,reset_space_and_domain
)(__isl_take PW
*pw
,
1332 __isl_take isl_space
*space
, __isl_take isl_space
*domain
)
1336 pw
= FN(PW
,cow
)(pw
);
1337 if (!pw
|| !space
|| !domain
)
1340 for (i
= 0; i
< pw
->n
; ++i
) {
1341 pw
->p
[i
].set
= isl_set_reset_space(pw
->p
[i
].set
,
1342 isl_space_copy(domain
));
1345 pw
->p
[i
].FIELD
= FN(EL
,reset_space_and_domain
)(pw
->p
[i
].FIELD
,
1346 isl_space_copy(space
), isl_space_copy(domain
));
1347 if (!pw
->p
[i
].FIELD
)
1351 isl_space_free(domain
);
1353 isl_space_free(pw
->dim
);
1358 isl_space_free(domain
);
1359 isl_space_free(space
);
1364 __isl_give PW
*FN(PW
,reset_domain_space
)(__isl_take PW
*pw
,
1365 __isl_take isl_space
*domain
)
1369 space
= isl_space_extend_domain_with_range(isl_space_copy(domain
),
1370 FN(PW
,get_space
)(pw
));
1371 return FN(PW
,reset_space_and_domain
)(pw
, space
, domain
);
1374 __isl_give PW
*FN(PW
,reset_space
)(__isl_take PW
*pw
, __isl_take isl_space
*dim
)
1378 domain
= isl_space_domain(isl_space_copy(dim
));
1379 return FN(PW
,reset_space_and_domain
)(pw
, dim
, domain
);
1382 __isl_give PW
*FN(PW
,set_tuple_id
)(__isl_take PW
*pw
, enum isl_dim_type type
,
1383 __isl_take isl_id
*id
)
1387 pw
= FN(PW
,cow
)(pw
);
1391 space
= FN(PW
,get_space
)(pw
);
1392 space
= isl_space_set_tuple_id(space
, type
, id
);
1394 return FN(PW
,reset_space
)(pw
, space
);
1397 return FN(PW
,free
)(pw
);
1400 /* Drop the id on the specified tuple.
1402 __isl_give PW
*FN(PW
,reset_tuple_id
)(__isl_take PW
*pw
, enum isl_dim_type type
)
1408 if (!FN(PW
,has_tuple_id
)(pw
, type
))
1411 pw
= FN(PW
,cow
)(pw
);
1415 space
= FN(PW
,get_space
)(pw
);
1416 space
= isl_space_reset_tuple_id(space
, type
);
1418 return FN(PW
,reset_space
)(pw
, space
);
1421 __isl_give PW
*FN(PW
,set_dim_id
)(__isl_take PW
*pw
,
1422 enum isl_dim_type type
, unsigned pos
, __isl_take isl_id
*id
)
1424 pw
= FN(PW
,cow
)(pw
);
1427 pw
->dim
= isl_space_set_dim_id(pw
->dim
, type
, pos
, id
);
1428 return FN(PW
,reset_space
)(pw
, isl_space_copy(pw
->dim
));
1431 return FN(PW
,free
)(pw
);
1435 int FN(PW
,has_equal_space
)(__isl_keep PW
*pw1
, __isl_keep PW
*pw2
)
1440 return isl_space_is_equal(pw1
->dim
, pw2
->dim
);
1444 __isl_give PW
*FN(PW
,morph_domain
)(__isl_take PW
*pw
,
1445 __isl_take isl_morph
*morph
)
1453 ctx
= isl_space_get_ctx(pw
->dim
);
1454 isl_assert(ctx
, isl_space_is_domain_internal(morph
->dom
->dim
, pw
->dim
),
1457 pw
= FN(PW
,cow
)(pw
);
1460 pw
->dim
= isl_space_extend_domain_with_range(
1461 isl_space_copy(morph
->ran
->dim
), pw
->dim
);
1465 for (i
= 0; i
< pw
->n
; ++i
) {
1466 pw
->p
[i
].set
= isl_morph_set(isl_morph_copy(morph
), pw
->p
[i
].set
);
1469 pw
->p
[i
].FIELD
= FN(EL
,morph_domain
)(pw
->p
[i
].FIELD
,
1470 isl_morph_copy(morph
));
1471 if (!pw
->p
[i
].FIELD
)
1475 isl_morph_free(morph
);
1480 isl_morph_free(morph
);
1485 int FN(PW
,n_piece
)(__isl_keep PW
*pw
)
1487 return pw
? pw
->n
: 0;
1490 int FN(PW
,foreach_piece
)(__isl_keep PW
*pw
,
1491 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
, void *user
),
1499 for (i
= 0; i
< pw
->n
; ++i
)
1500 if (fn(isl_set_copy(pw
->p
[i
].set
),
1501 FN(EL
,copy
)(pw
->p
[i
].FIELD
), user
) < 0)
1508 static int any_divs(__isl_keep isl_set
*set
)
1515 for (i
= 0; i
< set
->n
; ++i
)
1516 if (set
->p
[i
]->n_div
> 0)
1522 static int foreach_lifted_subset(__isl_take isl_set
*set
, __isl_take EL
*el
,
1523 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
1524 void *user
), void *user
)
1531 for (i
= 0; i
< set
->n
; ++i
) {
1535 lift
= isl_set_from_basic_set(isl_basic_set_copy(set
->p
[i
]));
1536 lift
= isl_set_lift(lift
);
1538 copy
= FN(EL
,copy
)(el
);
1539 copy
= FN(EL
,lift
)(copy
, isl_set_get_space(lift
));
1541 if (fn(lift
, copy
, user
) < 0)
1555 int FN(PW
,foreach_lifted_piece
)(__isl_keep PW
*pw
,
1556 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
1557 void *user
), void *user
)
1564 for (i
= 0; i
< pw
->n
; ++i
) {
1568 set
= isl_set_copy(pw
->p
[i
].set
);
1569 el
= FN(EL
,copy
)(pw
->p
[i
].FIELD
);
1570 if (!any_divs(set
)) {
1571 if (fn(set
, el
, user
) < 0)
1575 if (foreach_lifted_subset(set
, el
, fn
, user
) < 0)
1583 #ifndef NO_MOVE_DIMS
1584 __isl_give PW
*FN(PW
,move_dims
)(__isl_take PW
*pw
,
1585 enum isl_dim_type dst_type
, unsigned dst_pos
,
1586 enum isl_dim_type src_type
, unsigned src_pos
, unsigned n
)
1590 pw
= FN(PW
,cow
)(pw
);
1594 pw
->dim
= isl_space_move_dims(pw
->dim
, dst_type
, dst_pos
, src_type
, src_pos
, n
);
1598 for (i
= 0; i
< pw
->n
; ++i
) {
1599 pw
->p
[i
].FIELD
= FN(EL
,move_dims
)(pw
->p
[i
].FIELD
,
1600 dst_type
, dst_pos
, src_type
, src_pos
, n
);
1601 if (!pw
->p
[i
].FIELD
)
1605 if (dst_type
== isl_dim_in
)
1606 dst_type
= isl_dim_set
;
1607 if (src_type
== isl_dim_in
)
1608 src_type
= isl_dim_set
;
1610 for (i
= 0; i
< pw
->n
; ++i
) {
1611 pw
->p
[i
].set
= isl_set_move_dims(pw
->p
[i
].set
,
1613 src_type
, src_pos
, n
);
1625 __isl_give PW
*FN(PW
,mul_isl_int
)(__isl_take PW
*pw
, isl_int v
)
1629 if (isl_int_is_one(v
))
1631 if (pw
&& DEFAULT_IS_ZERO
&& isl_int_is_zero(v
)) {
1633 isl_space
*dim
= FN(PW
,get_space
)(pw
);
1635 zero
= FN(PW
,ZERO
)(dim
, pw
->type
);
1637 zero
= FN(PW
,ZERO
)(dim
);
1642 pw
= FN(PW
,cow
)(pw
);
1649 if (isl_int_is_neg(v
))
1650 pw
->type
= isl_fold_type_negate(pw
->type
);
1652 for (i
= 0; i
< pw
->n
; ++i
) {
1653 pw
->p
[i
].FIELD
= FN(EL
,scale
)(pw
->p
[i
].FIELD
, v
);
1654 if (!pw
->p
[i
].FIELD
)
1664 /* Multiply the pieces of "pw" by "v" and return the result.
1666 __isl_give PW
*FN(PW
,scale_val
)(__isl_take PW
*pw
, __isl_take isl_val
*v
)
1673 if (isl_val_is_one(v
)) {
1677 if (pw
&& DEFAULT_IS_ZERO
&& isl_val_is_zero(v
)) {
1679 isl_space
*space
= FN(PW
,get_space
)(pw
);
1681 zero
= FN(PW
,ZERO
)(space
, pw
->type
);
1683 zero
= FN(PW
,ZERO
)(space
);
1693 pw
= FN(PW
,cow
)(pw
);
1698 if (isl_val_is_neg(v
))
1699 pw
->type
= isl_fold_type_negate(pw
->type
);
1701 for (i
= 0; i
< pw
->n
; ++i
) {
1702 pw
->p
[i
].FIELD
= FN(EL
,scale_val
)(pw
->p
[i
].FIELD
,
1704 if (!pw
->p
[i
].FIELD
)
1716 __isl_give PW
*FN(PW
,scale
)(__isl_take PW
*pw
, isl_int v
)
1718 return FN(PW
,mul_isl_int
)(pw
, v
);
1721 static int FN(PW
,qsort_set_cmp
)(const void *p1
, const void *p2
)
1723 isl_set
*set1
= *(isl_set
* const *)p1
;
1724 isl_set
*set2
= *(isl_set
* const *)p2
;
1726 return isl_set_plain_cmp(set1
, set2
);
1729 /* We normalize in place, but if anything goes wrong we need
1730 * to return NULL, so we need to make sure we don't change the
1731 * meaning of any possible other copies of map.
1733 __isl_give PW
*FN(PW
,normalize
)(__isl_take PW
*pw
)
1740 for (i
= 0; i
< pw
->n
; ++i
) {
1741 set
= isl_set_normalize(isl_set_copy(pw
->p
[i
].set
));
1743 return FN(PW
,free
)(pw
);
1744 isl_set_free(pw
->p
[i
].set
);
1747 qsort(pw
->p
, pw
->n
, sizeof(pw
->p
[0]), &FN(PW
,qsort_set_cmp
));
1748 for (i
= pw
->n
- 1; i
>= 1; --i
) {
1749 if (!isl_set_plain_is_equal(pw
->p
[i
- 1].set
, pw
->p
[i
].set
))
1751 if (!FN(EL
,plain_is_equal
)(pw
->p
[i
- 1].FIELD
, pw
->p
[i
].FIELD
))
1753 set
= isl_set_union(isl_set_copy(pw
->p
[i
- 1].set
),
1754 isl_set_copy(pw
->p
[i
].set
));
1756 return FN(PW
,free
)(pw
);
1757 isl_set_free(pw
->p
[i
].set
);
1758 FN(EL
,free
)(pw
->p
[i
].FIELD
);
1759 isl_set_free(pw
->p
[i
- 1].set
);
1760 pw
->p
[i
- 1].set
= set
;
1761 for (j
= i
+ 1; j
< pw
->n
; ++j
)
1762 pw
->p
[j
- 1] = pw
->p
[j
];
1769 /* Is pw1 obviously equal to pw2?
1770 * That is, do they have obviously identical cells and obviously identical
1771 * elements on each cell?
1773 int FN(PW
,plain_is_equal
)(__isl_keep PW
*pw1
, __isl_keep PW
*pw2
)
1783 if (!isl_space_is_equal(pw1
->dim
, pw2
->dim
))
1786 pw1
= FN(PW
,copy
)(pw1
);
1787 pw2
= FN(PW
,copy
)(pw2
);
1788 pw1
= FN(PW
,normalize
)(pw1
);
1789 pw2
= FN(PW
,normalize
)(pw2
);
1793 equal
= pw1
->n
== pw2
->n
;
1794 for (i
= 0; equal
&& i
< pw1
->n
; ++i
) {
1795 equal
= isl_set_plain_is_equal(pw1
->p
[i
].set
, pw2
->p
[i
].set
);
1800 equal
= FN(EL
,plain_is_equal
)(pw1
->p
[i
].FIELD
, pw2
->p
[i
].FIELD
);
1815 static __isl_give PW
*FN(PW
,align_params_pw_multi_aff_and
)(__isl_take PW
*pw
,
1816 __isl_take isl_multi_aff
*ma
,
1817 __isl_give PW
*(*fn
)(__isl_take PW
*pw
, __isl_take isl_multi_aff
*ma
))
1820 isl_space
*ma_space
;
1822 ma_space
= isl_multi_aff_get_space(ma
);
1823 if (!pw
|| !ma
|| !ma_space
)
1825 if (isl_space_match(pw
->dim
, isl_dim_param
, ma_space
, isl_dim_param
)) {
1826 isl_space_free(ma_space
);
1829 ctx
= FN(PW
,get_ctx
)(pw
);
1830 if (!isl_space_has_named_params(pw
->dim
) ||
1831 !isl_space_has_named_params(ma_space
))
1832 isl_die(ctx
, isl_error_invalid
,
1833 "unaligned unnamed parameters", goto error
);
1834 pw
= FN(PW
,align_params
)(pw
, ma_space
);
1835 ma
= isl_multi_aff_align_params(ma
, FN(PW
,get_space
)(pw
));
1838 isl_space_free(ma_space
);
1840 isl_multi_aff_free(ma
);
1844 static __isl_give PW
*FN(PW
,align_params_pw_pw_multi_aff_and
)(__isl_take PW
*pw
,
1845 __isl_take isl_pw_multi_aff
*pma
,
1846 __isl_give PW
*(*fn
)(__isl_take PW
*pw
,
1847 __isl_take isl_pw_multi_aff
*ma
))
1850 isl_space
*pma_space
;
1852 pma_space
= isl_pw_multi_aff_get_space(pma
);
1853 if (!pw
|| !pma
|| !pma_space
)
1855 if (isl_space_match(pw
->dim
, isl_dim_param
, pma_space
, isl_dim_param
)) {
1856 isl_space_free(pma_space
);
1859 ctx
= FN(PW
,get_ctx
)(pw
);
1860 if (!isl_space_has_named_params(pw
->dim
) ||
1861 !isl_space_has_named_params(pma_space
))
1862 isl_die(ctx
, isl_error_invalid
,
1863 "unaligned unnamed parameters", goto error
);
1864 pw
= FN(PW
,align_params
)(pw
, pma_space
);
1865 pma
= isl_pw_multi_aff_align_params(pma
, FN(PW
,get_space
)(pw
));
1868 isl_space_free(pma_space
);
1870 isl_pw_multi_aff_free(pma
);
1874 /* Compute the pullback of "pw" by the function represented by "ma".
1875 * In other words, plug in "ma" in "pw".
1877 static __isl_give PW
*FN(PW
,pullback_multi_aff_aligned
)(__isl_take PW
*pw
,
1878 __isl_take isl_multi_aff
*ma
)
1881 isl_space
*space
= NULL
;
1883 ma
= isl_multi_aff_align_divs(ma
);
1884 pw
= FN(PW
,cow
)(pw
);
1888 space
= isl_space_join(isl_multi_aff_get_space(ma
),
1889 FN(PW
,get_space
)(pw
));
1891 for (i
= 0; i
< pw
->n
; ++i
) {
1892 pw
->p
[i
].set
= isl_set_preimage_multi_aff(pw
->p
[i
].set
,
1893 isl_multi_aff_copy(ma
));
1896 pw
->p
[i
].FIELD
= FN(EL
,pullback_multi_aff
)(pw
->p
[i
].FIELD
,
1897 isl_multi_aff_copy(ma
));
1898 if (!pw
->p
[i
].FIELD
)
1902 pw
= FN(PW
,reset_space
)(pw
, space
);
1903 isl_multi_aff_free(ma
);
1906 isl_space_free(space
);
1907 isl_multi_aff_free(ma
);
1912 __isl_give PW
*FN(PW
,pullback_multi_aff
)(__isl_take PW
*pw
,
1913 __isl_take isl_multi_aff
*ma
)
1915 return FN(PW
,align_params_pw_multi_aff_and
)(pw
, ma
,
1916 &FN(PW
,pullback_multi_aff_aligned
));
1919 /* Compute the pullback of "pw" by the function represented by "pma".
1920 * In other words, plug in "pma" in "pw".
1922 static __isl_give PW
*FN(PW
,pullback_pw_multi_aff_aligned
)(__isl_take PW
*pw
,
1923 __isl_take isl_pw_multi_aff
*pma
)
1933 space
= isl_space_join(isl_pw_multi_aff_get_space(pma
),
1934 FN(PW
,get_space
)(pw
));
1935 isl_pw_multi_aff_free(pma
);
1936 res
= FN(PW
,empty
)(space
);
1941 res
= FN(PW
,pullback_multi_aff
)(FN(PW
,copy
)(pw
),
1942 isl_multi_aff_copy(pma
->p
[0].maff
));
1943 res
= FN(PW
,intersect_domain
)(res
, isl_set_copy(pma
->p
[0].set
));
1945 for (i
= 1; i
< pma
->n
; ++i
) {
1948 res_i
= FN(PW
,pullback_multi_aff
)(FN(PW
,copy
)(pw
),
1949 isl_multi_aff_copy(pma
->p
[i
].maff
));
1950 res_i
= FN(PW
,intersect_domain
)(res_i
,
1951 isl_set_copy(pma
->p
[i
].set
));
1952 res
= FN(PW
,add_disjoint
)(res
, res_i
);
1955 isl_pw_multi_aff_free(pma
);
1959 isl_pw_multi_aff_free(pma
);
1964 __isl_give PW
*FN(PW
,pullback_pw_multi_aff
)(__isl_take PW
*pw
,
1965 __isl_take isl_pw_multi_aff
*pma
)
1967 return FN(PW
,align_params_pw_pw_multi_aff_and
)(pw
, pma
,
1968 &FN(PW
,pullback_pw_multi_aff_aligned
));