1 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
2 #define FN(TYPE,NAME) xFN(TYPE,NAME)
3 #define xS(TYPE,NAME) struct TYPE ## _ ## NAME
4 #define S(TYPE,NAME) xS(TYPE,NAME)
7 static __isl_give PW
*FN(PW
,alloc_
)(__isl_take isl_dim
*dim
,
8 enum isl_fold type
, int n
)
10 static __isl_give PW
*FN(PW
,alloc_
)(__isl_take isl_dim
*dim
, int n
)
18 ctx
= isl_dim_get_ctx(dim
);
19 isl_assert(ctx
, n
>= 0, goto error
);
20 pw
= isl_alloc(ctx
, struct PW
,
21 sizeof(struct PW
) + (n
- 1) * sizeof(S(PW
,piece
)));
39 __isl_give PW
*FN(PW
,ZERO
)(__isl_take isl_dim
*dim
, enum isl_fold type
)
41 return FN(PW
,alloc_
)(dim
, type
, 0);
44 __isl_give PW
*FN(PW
,ZERO
)(__isl_take isl_dim
*dim
)
46 return FN(PW
,alloc_
)(dim
, 0);
50 __isl_give PW
*FN(PW
,add_piece
)(__isl_take PW
*pw
,
51 __isl_take isl_set
*set
, __isl_take EL
*el
)
54 isl_dim
*el_dim
= NULL
;
56 if (!pw
|| !set
|| !el
)
59 if (isl_set_plain_is_empty(set
) || FN(EL
,EL_IS_ZERO
)(el
)) {
65 ctx
= isl_set_get_ctx(set
);
67 if (pw
->type
!= el
->type
)
68 isl_die(ctx
, isl_error_invalid
, "fold types don't match",
71 el_dim
= FN(EL
,get_dim(el
));
72 isl_assert(ctx
, isl_dim_equal(pw
->dim
, el_dim
), goto error
);
73 isl_assert(ctx
, pw
->n
< pw
->size
, goto error
);
75 pw
->p
[pw
->n
].set
= set
;
76 pw
->p
[pw
->n
].FIELD
= el
;
90 __isl_give PW
*FN(PW
,alloc
)(enum isl_fold type
,
91 __isl_take isl_set
*set
, __isl_take EL
*el
)
93 __isl_give PW
*FN(PW
,alloc
)(__isl_take isl_set
*set
, __isl_take EL
*el
)
102 pw
= FN(PW
,alloc_
)(isl_set_get_dim(set
), type
, 1);
104 pw
= FN(PW
,alloc_
)(isl_set_get_dim(set
), 1);
107 return FN(PW
,add_piece
)(pw
, set
, el
);
114 __isl_give PW
*FN(PW
,dup
)(__isl_keep PW
*pw
)
123 dup
= FN(PW
,alloc_
)(isl_dim_copy(pw
->dim
), pw
->type
, pw
->n
);
125 dup
= FN(PW
,alloc_
)(isl_dim_copy(pw
->dim
), pw
->n
);
130 for (i
= 0; i
< pw
->n
; ++i
)
131 dup
= FN(PW
,add_piece
)(dup
, isl_set_copy(pw
->p
[i
].set
),
132 FN(EL
,copy
)(pw
->p
[i
].FIELD
));
137 __isl_give PW
*FN(PW
,cow
)(__isl_take PW
*pw
)
145 return FN(PW
,dup
)(pw
);
148 __isl_give PW
*FN(PW
,copy
)(__isl_keep PW
*pw
)
157 void *FN(PW
,free
)(__isl_take PW
*pw
)
166 for (i
= 0; i
< pw
->n
; ++i
) {
167 isl_set_free(pw
->p
[i
].set
);
168 FN(EL
,free
)(pw
->p
[i
].FIELD
);
170 isl_dim_free(pw
->dim
);
176 int FN(PW
,IS_ZERO
)(__isl_keep PW
*pw
)
184 __isl_give PW
*FN(PW
,add
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
194 ctx
= isl_dim_get_ctx(pw1
->dim
);
196 if (pw1
->type
!= pw2
->type
)
197 isl_die(ctx
, isl_error_invalid
,
198 "fold types don't match", goto error
);
200 isl_assert(ctx
, isl_dim_equal(pw1
->dim
, pw2
->dim
), goto error
);
202 if (FN(PW
,IS_ZERO
)(pw1
)) {
207 if (FN(PW
,IS_ZERO
)(pw2
)) {
212 n
= (pw1
->n
+ 1) * (pw2
->n
+ 1);
214 res
= FN(PW
,alloc_
)(isl_dim_copy(pw1
->dim
), pw1
->type
, n
);
216 res
= FN(PW
,alloc_
)(isl_dim_copy(pw1
->dim
), n
);
219 for (i
= 0; i
< pw1
->n
; ++i
) {
220 set
= isl_set_copy(pw1
->p
[i
].set
);
221 for (j
= 0; j
< pw2
->n
; ++j
) {
222 struct isl_set
*common
;
224 set
= isl_set_subtract(set
,
225 isl_set_copy(pw2
->p
[j
].set
));
226 common
= isl_set_intersect(isl_set_copy(pw1
->p
[i
].set
),
227 isl_set_copy(pw2
->p
[j
].set
));
228 if (isl_set_plain_is_empty(common
)) {
229 isl_set_free(common
);
233 sum
= FN(EL
,add_on_domain
)(common
,
234 FN(EL
,copy
)(pw1
->p
[i
].FIELD
),
235 FN(EL
,copy
)(pw2
->p
[j
].FIELD
));
237 res
= FN(PW
,add_piece
)(res
, common
, sum
);
239 res
= FN(PW
,add_piece
)(res
, set
, FN(EL
,copy
)(pw1
->p
[i
].FIELD
));
242 for (j
= 0; j
< pw2
->n
; ++j
) {
243 set
= isl_set_copy(pw2
->p
[j
].set
);
244 for (i
= 0; i
< pw1
->n
; ++i
)
245 set
= isl_set_subtract(set
,
246 isl_set_copy(pw1
->p
[i
].set
));
247 res
= FN(PW
,add_piece
)(res
, set
, FN(EL
,copy
)(pw2
->p
[j
].FIELD
));
260 __isl_give PW
*FN(PW
,add_disjoint
)(__isl_take PW
*pw1
, __isl_take PW
*pw2
)
269 ctx
= isl_dim_get_ctx(pw1
->dim
);
271 if (pw1
->type
!= pw2
->type
)
272 isl_die(ctx
, isl_error_invalid
,
273 "fold types don't match", goto error
);
275 isl_assert(ctx
, isl_dim_equal(pw1
->dim
, pw2
->dim
), goto error
);
277 if (FN(PW
,IS_ZERO
)(pw1
)) {
282 if (FN(PW
,IS_ZERO
)(pw2
)) {
288 res
= FN(PW
,alloc_
)(isl_dim_copy(pw1
->dim
), pw1
->type
, pw1
->n
+ pw2
->n
);
290 res
= FN(PW
,alloc_
)(isl_dim_copy(pw1
->dim
), pw1
->n
+ pw2
->n
);
293 for (i
= 0; i
< pw1
->n
; ++i
)
294 res
= FN(PW
,add_piece
)(res
,
295 isl_set_copy(pw1
->p
[i
].set
),
296 FN(EL
,copy
)(pw1
->p
[i
].FIELD
));
298 for (i
= 0; i
< pw2
->n
; ++i
)
299 res
= FN(PW
,add_piece
)(res
,
300 isl_set_copy(pw2
->p
[i
].set
),
301 FN(EL
,copy
)(pw2
->p
[i
].FIELD
));
314 __isl_give PW
*FN(PW
,neg
)(__isl_take PW
*pw
)
321 if (FN(PW
,IS_ZERO
)(pw
))
328 for (i
= 0; i
< pw
->n
; ++i
) {
329 pw
->p
[i
].FIELD
= FN(EL
,neg
)(pw
->p
[i
].FIELD
);
331 return FN(PW
,free
)(pw
);
339 __isl_give isl_qpolynomial
*FN(PW
,eval
)(__isl_take PW
*pw
,
340 __isl_take isl_point
*pnt
)
345 isl_dim
*pnt_dim
= NULL
;
350 ctx
= isl_point_get_ctx(pnt
);
351 pnt_dim
= isl_point_get_dim(pnt
);
352 isl_assert(ctx
, isl_dim_equal(pnt_dim
, pw
->dim
), goto error
);
354 for (i
= 0; i
< pw
->n
; ++i
) {
355 found
= isl_set_contains_point(pw
->p
[i
].set
, pnt
);
362 qp
= FN(EL
,eval
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
363 isl_point_copy(pnt
));
365 qp
= isl_qpolynomial_zero(isl_dim_copy(pw
->dim
));
367 isl_dim_free(pnt_dim
);
372 isl_dim_free(pnt_dim
);
378 __isl_give isl_set
*FN(PW
,domain
)(__isl_take PW
*pw
)
386 dom
= isl_set_empty(isl_dim_copy(pw
->dim
));
387 for (i
= 0; i
< pw
->n
; ++i
)
388 dom
= isl_set_union_disjoint(dom
, isl_set_copy(pw
->p
[i
].set
));
395 __isl_give PW
*FN(PW
,intersect_domain
)(__isl_take PW
*pw
, __isl_take isl_set
*set
)
411 for (i
= pw
->n
- 1; i
>= 0; --i
) {
413 pw
->p
[i
].set
= isl_set_intersect(pw
->p
[i
].set
, isl_set_copy(set
));
416 aff
= isl_set_affine_hull(isl_set_copy(pw
->p
[i
].set
));
417 pw
->p
[i
].FIELD
= FN(EL
,substitute_equalities
)(pw
->p
[i
].FIELD
,
419 if (isl_set_plain_is_empty(pw
->p
[i
].set
)) {
420 isl_set_free(pw
->p
[i
].set
);
421 FN(EL
,free
)(pw
->p
[i
].FIELD
);
423 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
436 __isl_give PW
*FN(PW
,gist
)(__isl_take PW
*pw
, __isl_take isl_set
*context
)
439 isl_basic_set
*hull
= NULL
;
445 isl_set_free(context
);
449 context
= isl_set_compute_divs(context
);
450 hull
= isl_set_simple_hull(isl_set_copy(context
));
456 for (i
= pw
->n
- 1; i
>= 0; --i
) {
457 pw
->p
[i
].set
= isl_set_intersect(pw
->p
[i
].set
,
458 isl_set_copy(context
));
461 pw
->p
[i
].FIELD
= FN(EL
,gist
)(pw
->p
[i
].FIELD
,
462 isl_set_copy(pw
->p
[i
].set
));
463 pw
->p
[i
].set
= isl_set_gist_basic_set(pw
->p
[i
].set
,
464 isl_basic_set_copy(hull
));
467 if (isl_set_plain_is_empty(pw
->p
[i
].set
)) {
468 isl_set_free(pw
->p
[i
].set
);
469 FN(EL
,free
)(pw
->p
[i
].FIELD
);
471 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
476 isl_basic_set_free(hull
);
477 isl_set_free(context
);
482 isl_basic_set_free(hull
);
483 isl_set_free(context
);
487 __isl_give PW
*FN(PW
,coalesce
)(__isl_take PW
*pw
)
496 for (i
= pw
->n
- 1; i
>= 0; --i
) {
497 for (j
= i
- 1; j
>= 0; --j
) {
498 if (!FN(EL
,plain_is_equal
)(pw
->p
[i
].FIELD
,
501 pw
->p
[j
].set
= isl_set_union(pw
->p
[j
].set
,
503 FN(EL
,free
)(pw
->p
[i
].FIELD
);
505 pw
->p
[i
] = pw
->p
[pw
->n
- 1];
511 pw
->p
[i
].set
= isl_set_coalesce(pw
->p
[i
].set
);
522 isl_ctx
*FN(PW
,get_ctx
)(__isl_keep PW
*pw
)
524 return pw
? isl_dim_get_ctx(pw
->dim
) : NULL
;
527 #ifndef NO_INVOLVES_DIMS
528 int FN(PW
,involves_dims
)(__isl_keep PW
*pw
, enum isl_dim_type type
,
529 unsigned first
, unsigned n
)
535 if (pw
->n
== 0 || n
== 0)
537 for (i
= 0; i
< pw
->n
; ++i
) {
538 int involves
= FN(EL
,involves_dims
)(pw
->p
[i
].FIELD
,
540 if (involves
< 0 || involves
)
542 involves
= isl_set_involves_dims(pw
->p
[i
].set
, type
, first
, n
);
543 if (involves
< 0 || involves
)
550 __isl_give PW
*FN(PW
,set_dim_name
)(__isl_take PW
*pw
,
551 enum isl_dim_type type
, unsigned pos
, const char *s
)
559 pw
->dim
= isl_dim_set_name(pw
->dim
, type
, pos
, s
);
563 for (i
= 0; i
< pw
->n
; ++i
) {
564 pw
->p
[i
].set
= isl_set_set_dim_name(pw
->p
[i
].set
, type
, pos
, s
);
567 pw
->p
[i
].FIELD
= FN(EL
,set_dim_name
)(pw
->p
[i
].FIELD
, type
, pos
, s
);
579 __isl_give PW
*FN(PW
,drop_dims
)(__isl_take PW
*pw
,
580 enum isl_dim_type type
, unsigned first
, unsigned n
)
586 if (n
== 0 && !isl_dim_get_tuple_name(pw
->dim
, type
))
592 pw
->dim
= isl_dim_drop(pw
->dim
, type
, first
, n
);
595 for (i
= 0; i
< pw
->n
; ++i
) {
596 pw
->p
[i
].set
= isl_set_drop(pw
->p
[i
].set
, type
, first
, n
);
599 pw
->p
[i
].FIELD
= FN(EL
,drop_dims
)(pw
->p
[i
].FIELD
, type
, first
, n
);
611 #ifndef NO_INSERT_DIMS
612 __isl_give PW
*FN(PW
,insert_dims
)(__isl_take PW
*pw
, enum isl_dim_type type
,
613 unsigned first
, unsigned n
)
619 if (n
== 0 && !isl_dim_is_named_or_nested(pw
->dim
, type
))
626 pw
->dim
= isl_dim_insert(pw
->dim
, type
, first
, n
);
630 for (i
= 0; i
< pw
->n
; ++i
) {
631 pw
->p
[i
].set
= isl_set_insert(pw
->p
[i
].set
, type
, first
, n
);
634 pw
->p
[i
].FIELD
= FN(EL
,insert_dims
)(pw
->p
[i
].FIELD
,
647 __isl_give PW
*FN(PW
,fix_dim
)(__isl_take PW
*pw
,
648 enum isl_dim_type type
, unsigned pos
, isl_int v
)
658 for (i
= 0; i
< pw
->n
; ++i
) {
659 pw
->p
[i
].set
= isl_set_fix(pw
->p
[i
].set
, type
, pos
, v
);
670 unsigned FN(PW
,dim
)(__isl_keep PW
*pw
, enum isl_dim_type type
)
672 return pw
? isl_dim_size(pw
->dim
, type
) : 0;
675 __isl_give PW
*FN(PW
,split_dims
)(__isl_take PW
*pw
,
676 enum isl_dim_type type
, unsigned first
, unsigned n
)
690 for (i
= 0; i
< pw
->n
; ++i
) {
691 pw
->p
[i
].set
= isl_set_split_dims(pw
->p
[i
].set
, type
, first
, n
);
703 /* Compute the maximal value attained by the piecewise quasipolynomial
704 * on its domain or zero if the domain is empty.
705 * In the worst case, the domain is scanned completely,
706 * so the domain is assumed to be bounded.
708 __isl_give isl_qpolynomial
*FN(PW
,opt
)(__isl_take PW
*pw
, int max
)
711 isl_qpolynomial
*opt
;
717 isl_dim
*dim
= isl_dim_copy(pw
->dim
);
719 return isl_qpolynomial_zero(dim
);
722 opt
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[0].FIELD
),
723 isl_set_copy(pw
->p
[0].set
), max
);
724 for (i
= 1; i
< pw
->n
; ++i
) {
725 isl_qpolynomial
*opt_i
;
726 opt_i
= FN(EL
,opt_on_domain
)(FN(EL
,copy
)(pw
->p
[i
].FIELD
),
727 isl_set_copy(pw
->p
[i
].set
), max
);
729 opt
= isl_qpolynomial_max_cst(opt
, opt_i
);
731 opt
= isl_qpolynomial_min_cst(opt
, opt_i
);
738 __isl_give isl_qpolynomial
*FN(PW
,max
)(__isl_take PW
*pw
)
740 return FN(PW
,opt
)(pw
, 1);
743 __isl_give isl_qpolynomial
*FN(PW
,min
)(__isl_take PW
*pw
)
745 return FN(PW
,opt
)(pw
, 0);
749 __isl_give isl_dim
*FN(PW
,get_dim
)(__isl_keep PW
*pw
)
751 return pw
? isl_dim_copy(pw
->dim
) : NULL
;
755 __isl_give PW
*FN(PW
,reset_dim
)(__isl_take PW
*pw
, __isl_take isl_dim
*dim
)
763 for (i
= 0; i
< pw
->n
; ++i
) {
764 pw
->p
[i
].set
= isl_set_reset_dim(pw
->p
[i
].set
,
768 pw
->p
[i
].FIELD
= FN(EL
,reset_dim
)(pw
->p
[i
].FIELD
,
773 isl_dim_free(pw
->dim
);
784 int FN(PW
,has_equal_dim
)(__isl_keep PW
*pw1
, __isl_keep PW
*pw2
)
789 return isl_dim_equal(pw1
->dim
, pw2
->dim
);
793 __isl_give PW
*FN(PW
,morph
)(__isl_take PW
*pw
, __isl_take isl_morph
*morph
)
801 ctx
= isl_dim_get_ctx(pw
->dim
);
802 isl_assert(ctx
, isl_dim_equal(pw
->dim
, morph
->dom
->dim
),
808 isl_dim_free(pw
->dim
);
809 pw
->dim
= isl_dim_copy(morph
->ran
->dim
);
813 for (i
= 0; i
< pw
->n
; ++i
) {
814 pw
->p
[i
].set
= isl_morph_set(isl_morph_copy(morph
), pw
->p
[i
].set
);
817 pw
->p
[i
].FIELD
= FN(EL
,morph
)(pw
->p
[i
].FIELD
,
818 isl_morph_copy(morph
));
823 isl_morph_free(morph
);
828 isl_morph_free(morph
);
833 int FN(PW
,foreach_piece
)(__isl_keep PW
*pw
,
834 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
, void *user
),
842 for (i
= 0; i
< pw
->n
; ++i
)
843 if (fn(isl_set_copy(pw
->p
[i
].set
),
844 FN(EL
,copy
)(pw
->p
[i
].FIELD
), user
) < 0)
851 static int any_divs(__isl_keep isl_set
*set
)
858 for (i
= 0; i
< set
->n
; ++i
)
859 if (set
->p
[i
]->n_div
> 0)
865 static int foreach_lifted_subset(__isl_take isl_set
*set
, __isl_take EL
*el
,
866 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
867 void *user
), void *user
)
874 for (i
= 0; i
< set
->n
; ++i
) {
878 lift
= isl_set_from_basic_set(isl_basic_set_copy(set
->p
[i
]));
879 lift
= isl_set_lift(lift
);
881 copy
= FN(EL
,copy
)(el
);
882 copy
= FN(EL
,lift
)(copy
, isl_set_get_dim(lift
));
884 if (fn(lift
, copy
, user
) < 0)
898 int FN(PW
,foreach_lifted_piece
)(__isl_keep PW
*pw
,
899 int (*fn
)(__isl_take isl_set
*set
, __isl_take EL
*el
,
900 void *user
), void *user
)
907 for (i
= 0; i
< pw
->n
; ++i
) {
911 set
= isl_set_copy(pw
->p
[i
].set
);
912 el
= FN(EL
,copy
)(pw
->p
[i
].FIELD
);
913 if (!any_divs(set
)) {
914 if (fn(set
, el
, user
) < 0)
918 if (foreach_lifted_subset(set
, el
, fn
, user
) < 0)
927 __isl_give PW
*FN(PW
,move_dims
)(__isl_take PW
*pw
,
928 enum isl_dim_type dst_type
, unsigned dst_pos
,
929 enum isl_dim_type src_type
, unsigned src_pos
, unsigned n
)
937 pw
->dim
= isl_dim_move(pw
->dim
, dst_type
, dst_pos
, src_type
, src_pos
, n
);
941 for (i
= 0; i
< pw
->n
; ++i
) {
942 pw
->p
[i
].set
= isl_set_move_dims(pw
->p
[i
].set
,
944 src_type
, src_pos
, n
);
947 pw
->p
[i
].FIELD
= FN(EL
,move_dims
)(pw
->p
[i
].FIELD
,
948 dst_type
, dst_pos
, src_type
, src_pos
, n
);
961 __isl_give PW
*FN(PW
,realign
)(__isl_take PW
*pw
, __isl_take isl_reordering
*exp
)
969 for (i
= 0; i
< pw
->n
; ++i
) {
970 pw
->p
[i
].set
= isl_set_realign(pw
->p
[i
].set
,
971 isl_reordering_copy(exp
));
974 pw
->p
[i
].FIELD
= FN(EL
,realign
)(pw
->p
[i
].FIELD
,
975 isl_reordering_copy(exp
));
980 pw
= FN(PW
,reset_dim
)(pw
, isl_dim_copy(exp
->dim
));
982 isl_reordering_free(exp
);
985 isl_reordering_free(exp
);
990 /* Align the parameters of "pw" to those of "model".
992 __isl_give PW
*FN(PW
,align_params
)(__isl_take PW
*pw
, __isl_take isl_dim
*model
)
999 ctx
= isl_dim_get_ctx(model
);
1000 if (!isl_dim_has_named_params(model
))
1001 isl_die(ctx
, isl_error_invalid
,
1002 "model has unnamed parameters", goto error
);
1003 if (!isl_dim_has_named_params(pw
->dim
))
1004 isl_die(ctx
, isl_error_invalid
,
1005 "input has unnamed parameters", goto error
);
1006 if (!isl_dim_match(pw
->dim
, isl_dim_param
, model
, isl_dim_param
)) {
1007 isl_reordering
*exp
;
1009 model
= isl_dim_drop(model
, isl_dim_in
,
1010 0, isl_dim_size(model
, isl_dim_in
));
1011 model
= isl_dim_drop(model
, isl_dim_out
,
1012 0, isl_dim_size(model
, isl_dim_out
));
1013 exp
= isl_parameter_alignment_reordering(pw
->dim
, model
);
1014 exp
= isl_reordering_extend_dim(exp
, FN(PW
,get_dim
)(pw
));
1015 pw
= FN(PW
,realign
)(pw
, exp
);
1018 isl_dim_free(model
);
1021 isl_dim_free(model
);
1027 __isl_give PW
*FN(PW
,mul_isl_int
)(__isl_take PW
*pw
, isl_int v
)
1031 if (isl_int_is_one(v
))
1033 if (pw
&& isl_int_is_zero(v
)) {
1035 isl_dim
*dim
= FN(PW
,get_dim
)(pw
);
1037 zero
= FN(PW
,ZERO
)(dim
, pw
->type
);
1039 zero
= FN(PW
,ZERO
)(dim
);
1044 pw
= FN(PW
,cow
)(pw
);
1051 if (isl_int_is_neg(v
))
1052 pw
->type
= isl_fold_type_negate(pw
->type
);
1054 for (i
= 0; i
< pw
->n
; ++i
) {
1055 pw
->p
[i
].FIELD
= FN(EL
,scale
)(pw
->p
[i
].FIELD
, v
);
1056 if (!pw
->p
[i
].FIELD
)