?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
add isl_ast_expr_print_macros
[isl.git] / isl_pw_templ.c
blobf7106cf6bcd2b69ebe176b6c5f0d19e7d4a6b154
1 /*
2 * Copyright 2010-2011 INRIA Saclay
3 * Copyright 2011 Sven Verdoolaege
4 * Copyright 2012-2014 Ecole Normale Superieure
5 *
6 * Use of this software is governed by the MIT license
7 *
8 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
9 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
10 * 91893 Orsay, France
11 * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
12 */
13
14 #include <isl/aff.h>
15 #include <isl_val_private.h>
16
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)
21
22 #ifdef HAS_TYPE
23 __isl_give PW *FN(PW,alloc_size)(__isl_take isl_space *dim,
24 enum isl_fold type, int n)
25 #else
26 __isl_give PW *FN(PW,alloc_size)(__isl_take isl_space *dim, int n)
27 #endif
28 {
29 isl_ctx *ctx;
30 struct PW *pw;
31
32 if (!dim)
33 return NULL;
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)));
38 if (!pw)
39 goto error;
40
41 pw->ref = 1;
42 #ifdef HAS_TYPE
43 pw->type = type;
44 #endif
45 pw->size = n;
46 pw->n = 0;
47 pw->dim = dim;
48 return pw;
49 error:
50 isl_space_free(dim);
51 return NULL;
52 }
53
54 #ifdef HAS_TYPE
55 __isl_give PW *FN(PW,ZERO)(__isl_take isl_space *dim, enum isl_fold type)
56 {
57 return FN(PW,alloc_size)(dim, type, 0);
58 }
59 #else
60 __isl_give PW *FN(PW,ZERO)(__isl_take isl_space *dim)
61 {
62 return FN(PW,alloc_size)(dim, 0);
63 }
64 #endif
65
66 __isl_give PW *FN(PW,add_piece)(__isl_take PW *pw,
67 __isl_take isl_set *set, __isl_take EL *el)
68 {
69 isl_ctx *ctx;
70 isl_space *el_dim = NULL;
71
72 if (!pw || !set || !el)
73 goto error;
74
75 if (isl_set_plain_is_empty(set) || FN(EL,EL_IS_ZERO)(el)) {
76 isl_set_free(set);
77 FN(EL,free)(el);
78 return pw;
79 }
80
81 ctx = isl_set_get_ctx(set);
82 #ifdef HAS_TYPE
83 if (pw->type != el->type)
84 isl_die(ctx, isl_error_invalid, "fold types don't match",
85 goto error);
86 #endif
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);
90
91 pw->p[pw->n].set = set;
92 pw->p[pw->n].FIELD = el;
93 pw->n++;
94
95 isl_space_free(el_dim);
96 return pw;
97 error:
98 isl_space_free(el_dim);
99 FN(PW,free)(pw);
100 isl_set_free(set);
101 FN(EL,free)(el);
102 return NULL;
103 }
104
105 #ifdef HAS_TYPE
106 __isl_give PW *FN(PW,alloc)(enum isl_fold type,
107 __isl_take isl_set *set, __isl_take EL *el)
108 #else
109 __isl_give PW *FN(PW,alloc)(__isl_take isl_set *set, __isl_take EL *el)
110 #endif
111 {
112 PW *pw;
113
114 if (!set || !el)
115 goto error;
116
117 #ifdef HAS_TYPE
118 pw = FN(PW,alloc_size)(FN(EL,get_space)(el), type, 1);
119 #else
120 pw = FN(PW,alloc_size)(FN(EL,get_space)(el), 1);
121 #endif
122
123 return FN(PW,add_piece)(pw, set, el);
124 error:
125 isl_set_free(set);
126 FN(EL,free)(el);
127 return NULL;
128 }
129
130 __isl_give PW *FN(PW,dup)(__isl_keep PW *pw)
131 {
132 int i;
133 PW *dup;
134
135 if (!pw)
136 return NULL;
137
138 #ifdef HAS_TYPE
139 dup = FN(PW,alloc_size)(isl_space_copy(pw->dim), pw->type, pw->n);
140 #else
141 dup = FN(PW,alloc_size)(isl_space_copy(pw->dim), pw->n);
142 #endif
143 if (!dup)
144 return NULL;
145
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));
149
150 return dup;
151 }
152
153 __isl_give PW *FN(PW,cow)(__isl_take PW *pw)
154 {
155 if (!pw)
156 return NULL;
157
158 if (pw->ref == 1)
159 return pw;
160 pw->ref--;
161 return FN(PW,dup)(pw);
162 }
163
164 __isl_give PW *FN(PW,copy)(__isl_keep PW *pw)
165 {
166 if (!pw)
167 return NULL;
168
169 pw->ref++;
170 return pw;
171 }
172
173 __isl_null PW *FN(PW,free)(__isl_take PW *pw)
174 {
175 int i;
176
177 if (!pw)
178 return NULL;
179 if (--pw->ref > 0)
180 return NULL;
181
182 for (i = 0; i < pw->n; ++i) {
183 isl_set_free(pw->p[i].set);
184 FN(EL,free)(pw->p[i].FIELD);
185 }
186 isl_space_free(pw->dim);
187 free(pw);
188
189 return NULL;
190 }
191
192 const char *FN(PW,get_dim_name)(__isl_keep PW *pw, enum isl_dim_type type,
193 unsigned pos)
194 {
195 return pw ? isl_space_get_dim_name(pw->dim, type, pos) : NULL;
196 }
197
198 isl_bool FN(PW,has_dim_id)(__isl_keep PW *pw, enum isl_dim_type type,
199 unsigned pos)
200 {
201 return pw ? isl_space_has_dim_id(pw->dim, type, pos) : isl_bool_error;
202 }
203
204 __isl_give isl_id *FN(PW,get_dim_id)(__isl_keep PW *pw, enum isl_dim_type type,
205 unsigned pos)
206 {
207 return pw ? isl_space_get_dim_id(pw->dim, type, pos) : NULL;
208 }
209
210 isl_bool FN(PW,has_tuple_name)(__isl_keep PW *pw, enum isl_dim_type type)
211 {
212 return pw ? isl_space_has_tuple_name(pw->dim, type) : isl_bool_error;
213 }
214
215 const char *FN(PW,get_tuple_name)(__isl_keep PW *pw, enum isl_dim_type type)
216 {
217 return pw ? isl_space_get_tuple_name(pw->dim, type) : NULL;
218 }
219
220 isl_bool FN(PW,has_tuple_id)(__isl_keep PW *pw, enum isl_dim_type type)
221 {
222 return pw ? isl_space_has_tuple_id(pw->dim, type) : isl_bool_error;
223 }
224
225 __isl_give isl_id *FN(PW,get_tuple_id)(__isl_keep PW *pw, enum isl_dim_type type)
226 {
227 return pw ? isl_space_get_tuple_id(pw->dim, type) : NULL;
228 }
229
230 isl_bool FN(PW,IS_ZERO)(__isl_keep PW *pw)
231 {
232 if (!pw)
233 return isl_bool_error;
234
235 return pw->n == 0;
236 }
237
238 #ifndef NO_REALIGN
239 __isl_give PW *FN(PW,realign_domain)(__isl_take PW *pw,
240 __isl_take isl_reordering *exp)
241 {
242 int i;
243
244 pw = FN(PW,cow)(pw);
245 if (!pw || !exp)
246 goto error;
247
248 for (i = 0; i < pw->n; ++i) {
249 pw->p[i].set = isl_set_realign(pw->p[i].set,
250 isl_reordering_copy(exp));
251 if (!pw->p[i].set)
252 goto error;
253 pw->p[i].FIELD = FN(EL,realign_domain)(pw->p[i].FIELD,
254 isl_reordering_copy(exp));
255 if (!pw->p[i].FIELD)
256 goto error;
257 }
258
259 pw = FN(PW,reset_domain_space)(pw, isl_space_copy(exp->dim));
260
261 isl_reordering_free(exp);
262 return pw;
263 error:
264 isl_reordering_free(exp);
265 FN(PW,free)(pw);
266 return NULL;
267 }
268
269 /* Align the parameters of "pw" to those of "model".
270 */
271 __isl_give PW *FN(PW,align_params)(__isl_take PW *pw, __isl_take isl_space *model)
272 {
273 isl_ctx *ctx;
274
275 if (!pw || !model)
276 goto error;
277
278 ctx = isl_space_get_ctx(model);
279 if (!isl_space_has_named_params(model))
280 isl_die(ctx, isl_error_invalid,
281 "model has unnamed parameters", goto error);
282 if (!isl_space_has_named_params(pw->dim))
283 isl_die(ctx, isl_error_invalid,
284 "input has unnamed parameters", goto error);
285 if (!isl_space_match(pw->dim, isl_dim_param, model, isl_dim_param)) {
286 isl_reordering *exp;
287
288 model = isl_space_drop_dims(model, isl_dim_in,
289 0, isl_space_dim(model, isl_dim_in));
290 model = isl_space_drop_dims(model, isl_dim_out,
291 0, isl_space_dim(model, isl_dim_out));
292 exp = isl_parameter_alignment_reordering(pw->dim, model);
293 exp = isl_reordering_extend_space(exp,
294 FN(PW,get_domain_space)(pw));
295 pw = FN(PW,realign_domain)(pw, exp);
296 }
297
298 isl_space_free(model);
299 return pw;
300 error:
301 isl_space_free(model);
302 FN(PW,free)(pw);
303 return NULL;
304 }
305
306 static __isl_give PW *FN(PW,align_params_pw_pw_and)(__isl_take PW *pw1,
307 __isl_take PW *pw2,
308 __isl_give PW *(*fn)(__isl_take PW *pw1, __isl_take PW *pw2))
309 {
310 isl_ctx *ctx;
311
312 if (!pw1 || !pw2)
313 goto error;
314 if (isl_space_match(pw1->dim, isl_dim_param, pw2->dim, isl_dim_param))
315 return fn(pw1, pw2);
316 ctx = FN(PW,get_ctx)(pw1);
317 if (!isl_space_has_named_params(pw1->dim) ||
318 !isl_space_has_named_params(pw2->dim))
319 isl_die(ctx, isl_error_invalid,
320 "unaligned unnamed parameters", goto error);
321 pw1 = FN(PW,align_params)(pw1, FN(PW,get_space)(pw2));
322 pw2 = FN(PW,align_params)(pw2, FN(PW,get_space)(pw1));
323 return fn(pw1, pw2);
324 error:
325 FN(PW,free)(pw1);
326 FN(PW,free)(pw2);
327 return NULL;
328 }
329
330 static __isl_give PW *FN(PW,align_params_pw_set_and)(__isl_take PW *pw,
331 __isl_take isl_set *set,
332 __isl_give PW *(*fn)(__isl_take PW *pw, __isl_take isl_set *set))
333 {
334 isl_ctx *ctx;
335
336 if (!pw || !set)
337 goto error;
338 if (isl_space_match(pw->dim, isl_dim_param, set->dim, isl_dim_param))
339 return fn(pw, set);
340 ctx = FN(PW,get_ctx)(pw);
341 if (!isl_space_has_named_params(pw->dim) ||
342 !isl_space_has_named_params(set->dim))
343 isl_die(ctx, isl_error_invalid,
344 "unaligned unnamed parameters", goto error);
345 pw = FN(PW,align_params)(pw, isl_set_get_space(set));
346 set = isl_set_align_params(set, FN(PW,get_space)(pw));
347 return fn(pw, set);
348 error:
349 FN(PW,free)(pw);
350 isl_set_free(set);
351 return NULL;
352 }
353 #endif
354
355 static __isl_give PW *FN(PW,union_add_aligned)(__isl_take PW *pw1,
356 __isl_take PW *pw2)
357 {
358 int i, j, n;
359 struct PW *res;
360 isl_ctx *ctx;
361 isl_set *set;
362
363 if (!pw1 || !pw2)
364 goto error;
365
366 ctx = isl_space_get_ctx(pw1->dim);
367 #ifdef HAS_TYPE
368 if (pw1->type != pw2->type)
369 isl_die(ctx, isl_error_invalid,
370 "fold types don't match", goto error);
371 #endif
372 isl_assert(ctx, isl_space_is_equal(pw1->dim, pw2->dim), goto error);
373
374 if (FN(PW,IS_ZERO)(pw1)) {
375 FN(PW,free)(pw1);
376 return pw2;
377 }
378
379 if (FN(PW,IS_ZERO)(pw2)) {
380 FN(PW,free)(pw2);
381 return pw1;
382 }
383
384 n = (pw1->n + 1) * (pw2->n + 1);
385 #ifdef HAS_TYPE
386 res = FN(PW,alloc_size)(isl_space_copy(pw1->dim), pw1->type, n);
387 #else
388 res = FN(PW,alloc_size)(isl_space_copy(pw1->dim), n);
389 #endif
390
391 for (i = 0; i < pw1->n; ++i) {
392 set = isl_set_copy(pw1->p[i].set);
393 for (j = 0; j < pw2->n; ++j) {
394 struct isl_set *common;
395 EL *sum;
396 common = isl_set_intersect(isl_set_copy(pw1->p[i].set),
397 isl_set_copy(pw2->p[j].set));
398 if (isl_set_plain_is_empty(common)) {
399 isl_set_free(common);
400 continue;
401 }
402 set = isl_set_subtract(set,
403 isl_set_copy(pw2->p[j].set));
404
405 sum = FN(EL,add_on_domain)(common,
406 FN(EL,copy)(pw1->p[i].FIELD),
407 FN(EL,copy)(pw2->p[j].FIELD));
408
409 res = FN(PW,add_piece)(res, common, sum);
410 }
411 res = FN(PW,add_piece)(res, set, FN(EL,copy)(pw1->p[i].FIELD));
412 }
413
414 for (j = 0; j < pw2->n; ++j) {
415 set = isl_set_copy(pw2->p[j].set);
416 for (i = 0; i < pw1->n; ++i)
417 set = isl_set_subtract(set,
418 isl_set_copy(pw1->p[i].set));
419 res = FN(PW,add_piece)(res, set, FN(EL,copy)(pw2->p[j].FIELD));
420 }
421
422 FN(PW,free)(pw1);
423 FN(PW,free)(pw2);
424
425 return res;
426 error:
427 FN(PW,free)(pw1);
428 FN(PW,free)(pw2);
429 return NULL;
430 }
431
432 /* Private version of "union_add". For isl_pw_qpolynomial and
433 * isl_pw_qpolynomial_fold, we prefer to simply call it "add".
434 */
435 static __isl_give PW *FN(PW,union_add_)(__isl_take PW *pw1, __isl_take PW *pw2)
436 {
437 return FN(PW,align_params_pw_pw_and)(pw1, pw2,
438 &FN(PW,union_add_aligned));
439 }
440
441 /* Make sure "pw" has room for at least "n" more pieces.
442 *
443 * If there is only one reference to pw, we extend it in place.
444 * Otherwise, we create a new PW and copy the pieces.
445 */
446 static __isl_give PW *FN(PW,grow)(__isl_take PW *pw, int n)
447 {
448 int i;
449 isl_ctx *ctx;
450 PW *res;
451
452 if (!pw)
453 return NULL;
454 if (pw->n + n <= pw->size)
455 return pw;
456 ctx = FN(PW,get_ctx)(pw);
457 n += pw->n;
458 if (pw->ref == 1) {
459 res = isl_realloc(ctx, pw, struct PW,
460 sizeof(struct PW) + (n - 1) * sizeof(S(PW,piece)));
461 if (!res)
462 return FN(PW,free)(pw);
463 res->size = n;
464 return res;
465 }
466 #ifdef HAS_TYPE
467 res = FN(PW,alloc_size)(isl_space_copy(pw->dim), pw->type, n);
468 #else
469 res = FN(PW,alloc_size)(isl_space_copy(pw->dim), n);
470 #endif
471 if (!res)
472 return FN(PW,free)(pw);
473 for (i = 0; i < pw->n; ++i)
474 res = FN(PW,add_piece)(res, isl_set_copy(pw->p[i].set),
475 FN(EL,copy)(pw->p[i].FIELD));
476 FN(PW,free)(pw);
477 return res;
478 }
479
480 static __isl_give PW *FN(PW,add_disjoint_aligned)(__isl_take PW *pw1,
481 __isl_take PW *pw2)
482 {
483 int i;
484 isl_ctx *ctx;
485
486 if (!pw1 || !pw2)
487 goto error;
488
489 if (pw1->size < pw1->n + pw2->n && pw1->n < pw2->n)
490 return FN(PW,add_disjoint_aligned)(pw2, pw1);
491
492 ctx = isl_space_get_ctx(pw1->dim);
493 #ifdef HAS_TYPE
494 if (pw1->type != pw2->type)
495 isl_die(ctx, isl_error_invalid,
496 "fold types don't match", goto error);
497 #endif
498 isl_assert(ctx, isl_space_is_equal(pw1->dim, pw2->dim), goto error);
499
500 if (FN(PW,IS_ZERO)(pw1)) {
501 FN(PW,free)(pw1);
502 return pw2;
503 }
504
505 if (FN(PW,IS_ZERO)(pw2)) {
506 FN(PW,free)(pw2);
507 return pw1;
508 }
509
510 pw1 = FN(PW,grow)(pw1, pw2->n);
511 if (!pw1)
512 goto error;
513
514 for (i = 0; i < pw2->n; ++i)
515 pw1 = FN(PW,add_piece)(pw1,
516 isl_set_copy(pw2->p[i].set),
517 FN(EL,copy)(pw2->p[i].FIELD));
518
519 FN(PW,free)(pw2);
520
521 return pw1;
522 error:
523 FN(PW,free)(pw1);
524 FN(PW,free)(pw2);
525 return NULL;
526 }
527
528 __isl_give PW *FN(PW,add_disjoint)(__isl_take PW *pw1, __isl_take PW *pw2)
529 {
530 return FN(PW,align_params_pw_pw_and)(pw1, pw2,
531 &FN(PW,add_disjoint_aligned));
532 }
533
534 /* This function is currently only used from isl_aff.c
535 */
536 static __isl_give PW *FN(PW,on_shared_domain_in)(__isl_take PW *pw1,
537 __isl_take PW *pw2, __isl_take isl_space *space,
538 __isl_give EL *(*fn)(__isl_take EL *el1, __isl_take EL *el2))
539 __attribute__ ((unused));
540
541 /* Apply "fn" to pairs of elements from pw1 and pw2 on shared domains.
542 * The result of "fn" (and therefore also of this function) lives in "space".
543 */
544 static __isl_give PW *FN(PW,on_shared_domain_in)(__isl_take PW *pw1,
545 __isl_take PW *pw2, __isl_take isl_space *space,
546 __isl_give EL *(*fn)(__isl_take EL *el1, __isl_take EL *el2))
547 {
548 int i, j, n;
549 PW *res = NULL;
550
551 if (!pw1 || !pw2)
552 goto error;
553
554 n = pw1->n * pw2->n;
555 #ifdef HAS_TYPE
556 res = FN(PW,alloc_size)(isl_space_copy(space), pw1->type, n);
557 #else
558 res = FN(PW,alloc_size)(isl_space_copy(space), n);
559 #endif
560
561 for (i = 0; i < pw1->n; ++i) {
562 for (j = 0; j < pw2->n; ++j) {
563 isl_set *common;
564 EL *res_ij;
565 int empty;
566
567 common = isl_set_intersect(
568 isl_set_copy(pw1->p[i].set),
569 isl_set_copy(pw2->p[j].set));
570 empty = isl_set_plain_is_empty(common);
571 if (empty < 0 || empty) {
572 isl_set_free(common);
573 if (empty < 0)
574 goto error;
575 continue;
576 }
577
578 res_ij = fn(FN(EL,copy)(pw1->p[i].FIELD),
579 FN(EL,copy)(pw2->p[j].FIELD));
580 res_ij = FN(EL,gist)(res_ij, isl_set_copy(common));
581
582 res = FN(PW,add_piece)(res, common, res_ij);
583 }
584 }
585
586 isl_space_free(space);
587 FN(PW,free)(pw1);
588 FN(PW,free)(pw2);
589 return res;
590 error:
591 isl_space_free(space);
592 FN(PW,free)(pw1);
593 FN(PW,free)(pw2);
594 FN(PW,free)(res);
595 return NULL;
596 }
597
598 /* This function is currently only used from isl_aff.c
599 */
600 static __isl_give PW *FN(PW,on_shared_domain)(__isl_take PW *pw1,
601 __isl_take PW *pw2,
602 __isl_give EL *(*fn)(__isl_take EL *el1, __isl_take EL *el2))
603 __attribute__ ((unused));
604
605 /* Apply "fn" to pairs of elements from pw1 and pw2 on shared domains.
606 * The result of "fn" is assumed to live in the same space as "pw1" and "pw2".
607 */
608 static __isl_give PW *FN(PW,on_shared_domain)(__isl_take PW *pw1,
609 __isl_take PW *pw2,
610 __isl_give EL *(*fn)(__isl_take EL *el1, __isl_take EL *el2))
611 {
612 isl_space *space;
613
614 if (!pw1 || !pw2)
615 goto error;
616
617 space = isl_space_copy(pw1->dim);
618 return FN(PW,on_shared_domain_in)(pw1, pw2, space, fn);
619 error:
620 FN(PW,free)(pw1);
621 FN(PW,free)(pw2);
622 return NULL;
623 }
624
625 #ifndef NO_NEG
626 __isl_give PW *FN(PW,neg)(__isl_take PW *pw)
627 {
628 int i;
629
630 if (!pw)
631 return NULL;
632
633 if (FN(PW,IS_ZERO)(pw))
634 return pw;
635
636 pw = FN(PW,cow)(pw);
637 if (!pw)
638 return NULL;
639
640 for (i = 0; i < pw->n; ++i) {
641 pw->p[i].FIELD = FN(EL,neg)(pw->p[i].FIELD);
642 if (!pw->p[i].FIELD)
643 return FN(PW,free)(pw);
644 }
645
646 return pw;
647 }
648 #endif
649
650 #ifndef NO_SUB
651 __isl_give PW *FN(PW,sub)(__isl_take PW *pw1, __isl_take PW *pw2)
652 {
653 return FN(PW,add)(pw1, FN(PW,neg)(pw2));
654 }
655 #endif
656
657 #ifndef NO_EVAL
658 __isl_give isl_val *FN(PW,eval)(__isl_take PW *pw, __isl_take isl_point *pnt)
659 {
660 int i;
661 int found = 0;
662 isl_ctx *ctx;
663 isl_space *pnt_dim = NULL;
664 isl_val *v;
665
666 if (!pw || !pnt)
667 goto error;
668 ctx = isl_point_get_ctx(pnt);
669 pnt_dim = isl_point_get_space(pnt);
670 isl_assert(ctx, isl_space_is_domain_internal(pnt_dim, pw->dim),
671 goto error);
672
673 for (i = 0; i < pw->n; ++i) {
674 found = isl_set_contains_point(pw->p[i].set, pnt);
675 if (found < 0)
676 goto error;
677 if (found)
678 break;
679 }
680 if (found)
681 v = FN(EL,eval)(FN(EL,copy)(pw->p[i].FIELD),
682 isl_point_copy(pnt));
683 else
684 v = isl_val_zero(ctx);
685 FN(PW,free)(pw);
686 isl_space_free(pnt_dim);
687 isl_point_free(pnt);
688 return v;
689 error:
690 FN(PW,free)(pw);
691 isl_space_free(pnt_dim);
692 isl_point_free(pnt);
693 return NULL;
694 }
695 #endif
696
697 /* Return the parameter domain of "pw".
698 */
699 __isl_give isl_set *FN(PW,params)(__isl_take PW *pw)
700 {
701 return isl_set_params(FN(PW,domain)(pw));
702 }
703
704 __isl_give isl_set *FN(PW,domain)(__isl_take PW *pw)
705 {
706 int i;
707 isl_set *dom;
708
709 if (!pw)
710 return NULL;
711
712 dom = isl_set_empty(FN(PW,get_domain_space)(pw));
713 for (i = 0; i < pw->n; ++i)
714 dom = isl_set_union_disjoint(dom, isl_set_copy(pw->p[i].set));
715
716 FN(PW,free)(pw);
717
718 return dom;
719 }
720
721 /* Exploit the equalities in the domain of piece "i" of "pw"
722 * to simplify the associated function.
723 * If the domain of piece "i" is empty, then remove it entirely,
724 * replacing it with the final piece.
725 */
726 static int FN(PW,exploit_equalities_and_remove_if_empty)(__isl_keep PW *pw,
727 int i)
728 {
729 isl_basic_set *aff;
730 int empty = isl_set_plain_is_empty(pw->p[i].set);
731
732 if (empty < 0)
733 return -1;
734 if (empty) {
735 isl_set_free(pw->p[i].set);
736 FN(EL,free)(pw->p[i].FIELD);
737 if (i != pw->n - 1)
738 pw->p[i] = pw->p[pw->n - 1];
739 pw->n--;
740
741 return 0;
742 }
743
744 aff = isl_set_affine_hull(isl_set_copy(pw->p[i].set));
745 pw->p[i].FIELD = FN(EL,substitute_equalities)(pw->p[i].FIELD, aff);
746 if (!pw->p[i].FIELD)
747 return -1;
748
749 return 0;
750 }
751
752 /* Convert a piecewise expression defined over a parameter domain
753 * into one that is defined over a zero-dimensional set.
754 */
755 __isl_give PW *FN(PW,from_range)(__isl_take PW *pw)
756 {
757 isl_space *space;
758
759 if (!pw)
760 return NULL;
761 if (!isl_space_is_set(pw->dim))
762 isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
763 "not living in a set space", return FN(PW,free)(pw));
764
765 space = FN(PW,get_space)(pw);
766 space = isl_space_from_range(space);
767 pw = FN(PW,reset_space)(pw, space);
768
769 return pw;
770 }
771
772 /* Fix the value of the given parameter or domain dimension of "pw"
773 * to be equal to "value".
774 */
775 __isl_give PW *FN(PW,fix_si)(__isl_take PW *pw, enum isl_dim_type type,
776 unsigned pos, int value)
777 {
778 int i;
779
780 if (!pw)
781 return NULL;
782
783 if (type == isl_dim_out)
784 isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
785 "cannot fix output dimension", return FN(PW,free)(pw));
786
787 if (pw->n == 0)
788 return pw;
789
790 if (type == isl_dim_in)
791 type = isl_dim_set;
792
793 pw = FN(PW,cow)(pw);
794 if (!pw)
795 return FN(PW,free)(pw);
796
797 for (i = pw->n - 1; i >= 0; --i) {
798 pw->p[i].set = isl_set_fix_si(pw->p[i].set, type, pos, value);
799 if (FN(PW,exploit_equalities_and_remove_if_empty)(pw, i) < 0)
800 return FN(PW,free)(pw);
801 }
802
803 return pw;
804 }
805
806 /* Restrict the domain of "pw" by combining each cell
807 * with "set" through a call to "fn", where "fn" may be
808 * isl_set_intersect, isl_set_intersect_params or isl_set_subtract.
809 */
810 static __isl_give PW *FN(PW,restrict_domain_aligned)(__isl_take PW *pw,
811 __isl_take isl_set *set,
812 __isl_give isl_set *(*fn)(__isl_take isl_set *set1,
813 __isl_take isl_set *set2))
814 {
815 int i;
816
817 if (!pw || !set)
818 goto error;
819
820 if (pw->n == 0) {
821 isl_set_free(set);
822 return pw;
823 }
824
825 pw = FN(PW,cow)(pw);
826 if (!pw)
827 goto error;
828
829 for (i = pw->n - 1; i >= 0; --i) {
830 pw->p[i].set = fn(pw->p[i].set, isl_set_copy(set));
831 if (FN(PW,exploit_equalities_and_remove_if_empty)(pw, i) < 0)
832 goto error;
833 }
834
835 isl_set_free(set);
836 return pw;
837 error:
838 isl_set_free(set);
839 FN(PW,free)(pw);
840 return NULL;
841 }
842
843 static __isl_give PW *FN(PW,intersect_domain_aligned)(__isl_take PW *pw,
844 __isl_take isl_set *set)
845 {
846 return FN(PW,restrict_domain_aligned)(pw, set, &isl_set_intersect);
847 }
848
849 __isl_give PW *FN(PW,intersect_domain)(__isl_take PW *pw,
850 __isl_take isl_set *context)
851 {
852 return FN(PW,align_params_pw_set_and)(pw, context,
853 &FN(PW,intersect_domain_aligned));
854 }
855
856 static __isl_give PW *FN(PW,intersect_params_aligned)(__isl_take PW *pw,
857 __isl_take isl_set *set)
858 {
859 return FN(PW,restrict_domain_aligned)(pw, set,
860 &isl_set_intersect_params);
861 }
862
863 /* Intersect the domain of "pw" with the parameter domain "context".
864 */
865 __isl_give PW *FN(PW,intersect_params)(__isl_take PW *pw,
866 __isl_take isl_set *context)
867 {
868 return FN(PW,align_params_pw_set_and)(pw, context,
869 &FN(PW,intersect_params_aligned));
870 }
871
872 /* Subtract "domain' from the domain of "pw", assuming their
873 * parameters have been aligned.
874 */
875 static __isl_give PW *FN(PW,subtract_domain_aligned)(__isl_take PW *pw,
876 __isl_take isl_set *domain)
877 {
878 return FN(PW,restrict_domain_aligned)(pw, domain, &isl_set_subtract);
879 }
880
881 /* Subtract "domain' from the domain of "pw".
882 */
883 __isl_give PW *FN(PW,subtract_domain)(__isl_take PW *pw,
884 __isl_take isl_set *domain)
885 {
886 return FN(PW,align_params_pw_set_and)(pw, domain,
887 &FN(PW,subtract_domain_aligned));
888 }
889
890 /* Compute the gist of "pw" with respect to the domain constraints
891 * of "context" for the case where the domain of the last element
892 * of "pw" is equal to "context".
893 * Call "fn_el" to compute the gist of this element, replace
894 * its domain by the universe and drop all other elements
895 * as their domains are necessarily disjoint from "context".
896 */
897 static __isl_give PW *FN(PW,gist_last)(__isl_take PW *pw,
898 __isl_take isl_set *context,
899 __isl_give EL *(*fn_el)(__isl_take EL *el, __isl_take isl_set *set))
900 {
901 int i;
902 isl_space *space;
903
904 for (i = 0; i < pw->n - 1; ++i) {
905 isl_set_free(pw->p[i].set);
906 FN(EL,free)(pw->p[i].FIELD);
907 }
908 pw->p[0].FIELD = pw->p[pw->n - 1].FIELD;
909 pw->p[0].set = pw->p[pw->n - 1].set;
910 pw->n = 1;
911
912 space = isl_set_get_space(context);
913 pw->p[0].FIELD = fn_el(pw->p[0].FIELD, context);
914 context = isl_set_universe(space);
915 isl_set_free(pw->p[0].set);
916 pw->p[0].set = context;
917
918 if (!pw->p[0].FIELD || !pw->p[0].set)
919 return FN(PW,free)(pw);
920
921 return pw;
922 }
923
924 /* Compute the gist of "pw" with respect to the domain constraints
925 * of "context". Call "fn_el" to compute the gist of the elements
926 * and "fn_dom" to compute the gist of the domains.
927 *
928 * If the piecewise expression is empty or the context is the universe,
929 * then nothing can be simplified.
930 */
931 static __isl_give PW *FN(PW,gist_aligned)(__isl_take PW *pw,
932 __isl_take isl_set *context,
933 __isl_give EL *(*fn_el)(__isl_take EL *el,
934 __isl_take isl_set *set),
935 __isl_give isl_set *(*fn_dom)(__isl_take isl_set *set,
936 __isl_take isl_basic_set *bset))
937 {
938 int i;
939 int is_universe;
940 isl_basic_set *hull = NULL;
941
942 if (!pw || !context)
943 goto error;
944
945 if (pw->n == 0) {
946 isl_set_free(context);
947 return pw;
948 }
949
950 is_universe = isl_set_plain_is_universe(context);
951 if (is_universe < 0)
952 goto error;
953 if (is_universe) {
954 isl_set_free(context);
955 return pw;
956 }
957
958 if (!isl_space_match(pw->dim, isl_dim_param,
959 context->dim, isl_dim_param)) {
960 pw = FN(PW,align_params)(pw, isl_set_get_space(context));
961 context = isl_set_align_params(context, FN(PW,get_space)(pw));
962 }
963
964 pw = FN(PW,cow)(pw);
965 if (!pw)
966 goto error;
967
968 if (pw->n == 1) {
969 int equal;
970
971 equal = isl_set_plain_is_equal(pw->p[0].set, context);
972 if (equal < 0)
973 goto error;
974 if (equal)
975 return FN(PW,gist_last)(pw, context, fn_el);
976 }
977
978 context = isl_set_compute_divs(context);
979 hull = isl_set_simple_hull(isl_set_copy(context));
980
981 for (i = pw->n - 1; i >= 0; --i) {
982 isl_set *set_i;
983 int empty;
984
985 if (i == pw->n - 1) {
986 int equal;
987 equal = isl_set_plain_is_equal(pw->p[i].set, context);
988 if (equal < 0)
989 goto error;
990 if (equal) {
991 isl_basic_set_free(hull);
992 return FN(PW,gist_last)(pw, context, fn_el);
993 }
994 }
995 set_i = isl_set_intersect(isl_set_copy(pw->p[i].set),
996 isl_set_copy(context));
997 empty = isl_set_plain_is_empty(set_i);
998 pw->p[i].FIELD = fn_el(pw->p[i].FIELD, set_i);
999 pw->p[i].set = fn_dom(pw->p[i].set, isl_basic_set_copy(hull));
1000 if (empty < 0 || !pw->p[i].FIELD || !pw->p[i].set)
1001 goto error;
1002 if (empty) {
1003 isl_set_free(pw->p[i].set);
1004 FN(EL,free)(pw->p[i].FIELD);
1005 if (i != pw->n - 1)
1006 pw->p[i] = pw->p[pw->n - 1];
1007 pw->n--;
1008 }
1009 }
1010
1011 isl_basic_set_free(hull);
1012 isl_set_free(context);
1013
1014 return pw;
1015 error:
1016 FN(PW,free)(pw);
1017 isl_basic_set_free(hull);
1018 isl_set_free(context);
1019 return NULL;
1020 }
1021
1022 static __isl_give PW *FN(PW,gist_domain_aligned)(__isl_take PW *pw,
1023 __isl_take isl_set *set)
1024 {
1025 return FN(PW,gist_aligned)(pw, set, &FN(EL,gist),
1026 &isl_set_gist_basic_set);
1027 }
1028
1029 __isl_give PW *FN(PW,gist)(__isl_take PW *pw, __isl_take isl_set *context)
1030 {
1031 return FN(PW,align_params_pw_set_and)(pw, context,
1032 &FN(PW,gist_domain_aligned));
1033 }
1034
1035 static __isl_give PW *FN(PW,gist_params_aligned)(__isl_take PW *pw,
1036 __isl_take isl_set *set)
1037 {
1038 return FN(PW,gist_aligned)(pw, set, &FN(EL,gist_params),
1039 &isl_set_gist_params_basic_set);
1040 }
1041
1042 __isl_give PW *FN(PW,gist_params)(__isl_take PW *pw,
1043 __isl_take isl_set *context)
1044 {
1045 return FN(PW,align_params_pw_set_and)(pw, context,
1046 &FN(PW,gist_params_aligned));
1047 }
1048
1049 __isl_give PW *FN(PW,coalesce)(__isl_take PW *pw)
1050 {
1051 int i, j;
1052
1053 if (!pw)
1054 return NULL;
1055 if (pw->n == 0)
1056 return pw;
1057
1058 for (i = pw->n - 1; i >= 0; --i) {
1059 for (j = i - 1; j >= 0; --j) {
1060 if (!FN(EL,plain_is_equal)(pw->p[i].FIELD,
1061 pw->p[j].FIELD))
1062 continue;
1063 pw->p[j].set = isl_set_union(pw->p[j].set,
1064 pw->p[i].set);
1065 FN(EL,free)(pw->p[i].FIELD);
1066 if (i != pw->n - 1)
1067 pw->p[i] = pw->p[pw->n - 1];
1068 pw->n--;
1069 break;
1070 }
1071 if (j >= 0)
1072 continue;
1073 pw->p[i].set = isl_set_coalesce(pw->p[i].set);
1074 if (!pw->p[i].set)
1075 goto error;
1076 }
1077
1078 return pw;
1079 error:
1080 FN(PW,free)(pw);
1081 return NULL;
1082 }
1083
1084 isl_ctx *FN(PW,get_ctx)(__isl_keep PW *pw)
1085 {
1086 return pw ? isl_space_get_ctx(pw->dim) : NULL;
1087 }
1088
1089 #ifndef NO_INVOLVES_DIMS
1090 isl_bool FN(PW,involves_dims)(__isl_keep PW *pw, enum isl_dim_type type,
1091 unsigned first, unsigned n)
1092 {
1093 int i;
1094 enum isl_dim_type set_type;
1095
1096 if (!pw)
1097 return isl_bool_error;
1098 if (pw->n == 0 || n == 0)
1099 return isl_bool_false;
1100
1101 set_type = type == isl_dim_in ? isl_dim_set : type;
1102
1103 for (i = 0; i < pw->n; ++i) {
1104 isl_bool involves = FN(EL,involves_dims)(pw->p[i].FIELD,
1105 type, first, n);
1106 if (involves < 0 || involves)
1107 return involves;
1108 involves = isl_set_involves_dims(pw->p[i].set,
1109 set_type, first, n);
1110 if (involves < 0 || involves)
1111 return involves;
1112 }
1113 return isl_bool_false;
1114 }
1115 #endif
1116
1117 __isl_give PW *FN(PW,set_dim_name)(__isl_take PW *pw,
1118 enum isl_dim_type type, unsigned pos, const char *s)
1119 {
1120 int i;
1121 enum isl_dim_type set_type;
1122
1123 pw = FN(PW,cow)(pw);
1124 if (!pw)
1125 return NULL;
1126
1127 set_type = type == isl_dim_in ? isl_dim_set : type;
1128
1129 pw->dim = isl_space_set_dim_name(pw->dim, type, pos, s);
1130 if (!pw->dim)
1131 goto error;
1132
1133 for (i = 0; i < pw->n; ++i) {
1134 pw->p[i].set = isl_set_set_dim_name(pw->p[i].set,
1135 set_type, pos, s);
1136 if (!pw->p[i].set)
1137 goto error;
1138 pw->p[i].FIELD = FN(EL,set_dim_name)(pw->p[i].FIELD, type, pos, s);
1139 if (!pw->p[i].FIELD)
1140 goto error;
1141 }
1142
1143 return pw;
1144 error:
1145 FN(PW,free)(pw);
1146 return NULL;
1147 }
1148
1149 #ifndef NO_DROP_DIMS
1150 __isl_give PW *FN(PW,drop_dims)(__isl_take PW *pw,
1151 enum isl_dim_type type, unsigned first, unsigned n)
1152 {
1153 int i;
1154 enum isl_dim_type set_type;
1155
1156 if (!pw)
1157 return NULL;
1158 if (n == 0 && !isl_space_get_tuple_name(pw->dim, type))
1159 return pw;
1160
1161 set_type = type == isl_dim_in ? isl_dim_set : type;
1162
1163 pw = FN(PW,cow)(pw);
1164 if (!pw)
1165 return NULL;
1166 pw->dim = isl_space_drop_dims(pw->dim, type, first, n);
1167 if (!pw->dim)
1168 goto error;
1169 for (i = 0; i < pw->n; ++i) {
1170 pw->p[i].FIELD = FN(EL,drop_dims)(pw->p[i].FIELD, type, first, n);
1171 if (!pw->p[i].FIELD)
1172 goto error;
1173 if (type == isl_dim_out)
1174 continue;
1175 pw->p[i].set = isl_set_drop(pw->p[i].set, set_type, first, n);
1176 if (!pw->p[i].set)
1177 goto error;
1178 }
1179
1180 return pw;
1181 error:
1182 FN(PW,free)(pw);
1183 return NULL;
1184 }
1185
1186 /* This function is very similar to drop_dims.
1187 * The only difference is that the cells may still involve
1188 * the specified dimensions. They are removed using
1189 * isl_set_project_out instead of isl_set_drop.
1190 */
1191 __isl_give PW *FN(PW,project_out)(__isl_take PW *pw,
1192 enum isl_dim_type type, unsigned first, unsigned n)
1193 {
1194 int i;
1195 enum isl_dim_type set_type;
1196
1197 if (!pw)
1198 return NULL;
1199 if (n == 0 && !isl_space_get_tuple_name(pw->dim, type))
1200 return pw;
1201
1202 set_type = type == isl_dim_in ? isl_dim_set : type;
1203
1204 pw = FN(PW,cow)(pw);
1205 if (!pw)
1206 return NULL;
1207 pw->dim = isl_space_drop_dims(pw->dim, type, first, n);
1208 if (!pw->dim)
1209 goto error;
1210 for (i = 0; i < pw->n; ++i) {
1211 pw->p[i].set = isl_set_project_out(pw->p[i].set,
1212 set_type, first, n);
1213 if (!pw->p[i].set)
1214 goto error;
1215 pw->p[i].FIELD = FN(EL,drop_dims)(pw->p[i].FIELD, type, first, n);
1216 if (!pw->p[i].FIELD)
1217 goto error;
1218 }
1219
1220 return pw;
1221 error:
1222 FN(PW,free)(pw);
1223 return NULL;
1224 }
1225
1226 /* Project the domain of pw onto its parameter space.
1227 */
1228 __isl_give PW *FN(PW,project_domain_on_params)(__isl_take PW *pw)
1229 {
1230 isl_space *space;
1231 unsigned n;
1232
1233 n = FN(PW,dim)(pw, isl_dim_in);
1234 pw = FN(PW,project_out)(pw, isl_dim_in, 0, n);
1235 space = FN(PW,get_domain_space)(pw);
1236 space = isl_space_params(space);
1237 pw = FN(PW,reset_domain_space)(pw, space);
1238 return pw;
1239 }
1240 #endif
1241
1242 #ifndef NO_INSERT_DIMS
1243 __isl_give PW *FN(PW,insert_dims)(__isl_take PW *pw, enum isl_dim_type type,
1244 unsigned first, unsigned n)
1245 {
1246 int i;
1247 enum isl_dim_type set_type;
1248
1249 if (!pw)
1250 return NULL;
1251 if (n == 0 && !isl_space_is_named_or_nested(pw->dim, type))
1252 return pw;
1253
1254 set_type = type == isl_dim_in ? isl_dim_set : type;
1255
1256 pw = FN(PW,cow)(pw);
1257 if (!pw)
1258 return NULL;
1259
1260 pw->dim = isl_space_insert_dims(pw->dim, type, first, n);
1261 if (!pw->dim)
1262 goto error;
1263
1264 for (i = 0; i < pw->n; ++i) {
1265 pw->p[i].set = isl_set_insert_dims(pw->p[i].set,
1266 set_type, first, n);
1267 if (!pw->p[i].set)
1268 goto error;
1269 pw->p[i].FIELD = FN(EL,insert_dims)(pw->p[i].FIELD,
1270 type, first, n);
1271 if (!pw->p[i].FIELD)
1272 goto error;
1273 }
1274
1275 return pw;
1276 error:
1277 FN(PW,free)(pw);
1278 return NULL;
1279 }
1280 #endif
1281
1282 __isl_give PW *FN(PW,fix_dim)(__isl_take PW *pw,
1283 enum isl_dim_type type, unsigned pos, isl_int v)
1284 {
1285 int i;
1286
1287 if (!pw)
1288 return NULL;
1289
1290 if (type == isl_dim_in)
1291 type = isl_dim_set;
1292
1293 pw = FN(PW,cow)(pw);
1294 if (!pw)
1295 return NULL;
1296 for (i = 0; i < pw->n; ++i) {
1297 pw->p[i].set = isl_set_fix(pw->p[i].set, type, pos, v);
1298 if (FN(PW,exploit_equalities_and_remove_if_empty)(pw, i) < 0)
1299 return FN(PW,free)(pw);
1300 }
1301
1302 return pw;
1303 }
1304
1305 /* Fix the value of the variable at position "pos" of type "type" of "pw"
1306 * to be equal to "v".
1307 */
1308 __isl_give PW *FN(PW,fix_val)(__isl_take PW *pw,
1309 enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
1310 {
1311 if (!v)
1312 return FN(PW,free)(pw);
1313 if (!isl_val_is_int(v))
1314 isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
1315 "expecting integer value", goto error);
1316
1317 pw = FN(PW,fix_dim)(pw, type, pos, v->n);
1318 isl_val_free(v);
1319
1320 return pw;
1321 error:
1322 isl_val_free(v);
1323 return FN(PW,free)(pw);
1324 }
1325
1326 unsigned FN(PW,dim)(__isl_keep PW *pw, enum isl_dim_type type)
1327 {
1328 return pw ? isl_space_dim(pw->dim, type) : 0;
1329 }
1330
1331 __isl_give PW *FN(PW,split_dims)(__isl_take PW *pw,
1332 enum isl_dim_type type, unsigned first, unsigned n)
1333 {
1334 int i;
1335
1336 if (!pw)
1337 return NULL;
1338 if (n == 0)
1339 return pw;
1340
1341 if (type == isl_dim_in)
1342 type = isl_dim_set;
1343
1344 pw = FN(PW,cow)(pw);
1345 if (!pw)
1346 return NULL;
1347 if (!pw->dim)
1348 goto error;
1349 for (i = 0; i < pw->n; ++i) {
1350 pw->p[i].set = isl_set_split_dims(pw->p[i].set, type, first, n);
1351 if (!pw->p[i].set)
1352 goto error;
1353 }
1354
1355 return pw;
1356 error:
1357 FN(PW,free)(pw);
1358 return NULL;
1359 }
1360
1361 #ifndef NO_OPT
1362 /* Compute the maximal value attained by the piecewise quasipolynomial
1363 * on its domain or zero if the domain is empty.
1364 * In the worst case, the domain is scanned completely,
1365 * so the domain is assumed to be bounded.
1366 */
1367 __isl_give isl_val *FN(PW,opt)(__isl_take PW *pw, int max)
1368 {
1369 int i;
1370 isl_val *opt;
1371
1372 if (!pw)
1373 return NULL;
1374
1375 if (pw->n == 0) {
1376 opt = isl_val_zero(FN(PW,get_ctx)(pw));
1377 FN(PW,free)(pw);
1378 return opt;
1379 }
1380
1381 opt = FN(EL,opt_on_domain)(FN(EL,copy)(pw->p[0].FIELD),
1382 isl_set_copy(pw->p[0].set), max);
1383 for (i = 1; i < pw->n; ++i) {
1384 isl_val *opt_i;
1385 opt_i = FN(EL,opt_on_domain)(FN(EL,copy)(pw->p[i].FIELD),
1386 isl_set_copy(pw->p[i].set), max);
1387 if (max)
1388 opt = isl_val_max(opt, opt_i);
1389 else
1390 opt = isl_val_min(opt, opt_i);
1391 }
1392
1393 FN(PW,free)(pw);
1394 return opt;
1395 }
1396
1397 __isl_give isl_val *FN(PW,max)(__isl_take PW *pw)
1398 {
1399 return FN(PW,opt)(pw, 1);
1400 }
1401
1402 __isl_give isl_val *FN(PW,min)(__isl_take PW *pw)
1403 {
1404 return FN(PW,opt)(pw, 0);
1405 }
1406 #endif
1407
1408 __isl_give isl_space *FN(PW,get_space)(__isl_keep PW *pw)
1409 {
1410 return pw ? isl_space_copy(pw->dim) : NULL;
1411 }
1412
1413 __isl_give isl_space *FN(PW,get_domain_space)(__isl_keep PW *pw)
1414 {
1415 return pw ? isl_space_domain(isl_space_copy(pw->dim)) : NULL;
1416 }
1417
1418 /* Return the position of the dimension of the given type and name
1419 * in "pw".
1420 * Return -1 if no such dimension can be found.
1421 */
1422 int FN(PW,find_dim_by_name)(__isl_keep PW *pw,
1423 enum isl_dim_type type, const char *name)
1424 {
1425 if (!pw)
1426 return -1;
1427 return isl_space_find_dim_by_name(pw->dim, type, name);
1428 }
1429
1430 #ifndef NO_RESET_DIM
1431 /* Reset the space of "pw". Since we don't know if the elements
1432 * represent the spaces themselves or their domains, we pass along
1433 * both when we call their reset_space_and_domain.
1434 */
1435 static __isl_give PW *FN(PW,reset_space_and_domain)(__isl_take PW *pw,
1436 __isl_take isl_space *space, __isl_take isl_space *domain)
1437 {
1438 int i;
1439
1440 pw = FN(PW,cow)(pw);
1441 if (!pw || !space || !domain)
1442 goto error;
1443
1444 for (i = 0; i < pw->n; ++i) {
1445 pw->p[i].set = isl_set_reset_space(pw->p[i].set,
1446 isl_space_copy(domain));
1447 if (!pw->p[i].set)
1448 goto error;
1449 pw->p[i].FIELD = FN(EL,reset_space_and_domain)(pw->p[i].FIELD,
1450 isl_space_copy(space), isl_space_copy(domain));
1451 if (!pw->p[i].FIELD)
1452 goto error;
1453 }
1454
1455 isl_space_free(domain);
1456
1457 isl_space_free(pw->dim);
1458 pw->dim = space;
1459
1460 return pw;
1461 error:
1462 isl_space_free(domain);
1463 isl_space_free(space);
1464 FN(PW,free)(pw);
1465 return NULL;
1466 }
1467
1468 __isl_give PW *FN(PW,reset_domain_space)(__isl_take PW *pw,
1469 __isl_take isl_space *domain)
1470 {
1471 isl_space *space;
1472
1473 space = isl_space_extend_domain_with_range(isl_space_copy(domain),
1474 FN(PW,get_space)(pw));
1475 return FN(PW,reset_space_and_domain)(pw, space, domain);
1476 }
1477
1478 __isl_give PW *FN(PW,reset_space)(__isl_take PW *pw, __isl_take isl_space *dim)
1479 {
1480 isl_space *domain;
1481
1482 domain = isl_space_domain(isl_space_copy(dim));
1483 return FN(PW,reset_space_and_domain)(pw, dim, domain);
1484 }
1485
1486 __isl_give PW *FN(PW,set_tuple_id)(__isl_take PW *pw, enum isl_dim_type type,
1487 __isl_take isl_id *id)
1488 {
1489 isl_space *space;
1490
1491 pw = FN(PW,cow)(pw);
1492 if (!pw)
1493 goto error;
1494
1495 space = FN(PW,get_space)(pw);
1496 space = isl_space_set_tuple_id(space, type, id);
1497
1498 return FN(PW,reset_space)(pw, space);
1499 error:
1500 isl_id_free(id);
1501 return FN(PW,free)(pw);
1502 }
1503
1504 /* Drop the id on the specified tuple.
1505 */
1506 __isl_give PW *FN(PW,reset_tuple_id)(__isl_take PW *pw, enum isl_dim_type type)
1507 {
1508 isl_space *space;
1509
1510 if (!pw)
1511 return NULL;
1512 if (!FN(PW,has_tuple_id)(pw, type))
1513 return pw;
1514
1515 pw = FN(PW,cow)(pw);
1516 if (!pw)
1517 return NULL;
1518
1519 space = FN(PW,get_space)(pw);
1520 space = isl_space_reset_tuple_id(space, type);
1521
1522 return FN(PW,reset_space)(pw, space);
1523 }
1524
1525 __isl_give PW *FN(PW,set_dim_id)(__isl_take PW *pw,
1526 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
1527 {
1528 pw = FN(PW,cow)(pw);
1529 if (!pw)
1530 goto error;
1531 pw->dim = isl_space_set_dim_id(pw->dim, type, pos, id);
1532 return FN(PW,reset_space)(pw, isl_space_copy(pw->dim));
1533 error:
1534 isl_id_free(id);
1535 return FN(PW,free)(pw);
1536 }
1537 #endif
1538
1539 /* Reset the user pointer on all identifiers of parameters and tuples
1540 * of the space of "pw".
1541 */
1542 __isl_give PW *FN(PW,reset_user)(__isl_take PW *pw)
1543 {
1544 isl_space *space;
1545
1546 space = FN(PW,get_space)(pw);
1547 space = isl_space_reset_user(space);
1548
1549 return FN(PW,reset_space)(pw, space);
1550 }
1551
1552 int FN(PW,has_equal_space)(__isl_keep PW *pw1, __isl_keep PW *pw2)
1553 {
1554 if (!pw1 || !pw2)
1555 return -1;
1556
1557 return isl_space_is_equal(pw1->dim, pw2->dim);
1558 }
1559
1560 #ifndef NO_MORPH
1561 __isl_give PW *FN(PW,morph_domain)(__isl_take PW *pw,
1562 __isl_take isl_morph *morph)
1563 {
1564 int i;
1565 isl_ctx *ctx;
1566
1567 if (!pw || !morph)
1568 goto error;
1569
1570 ctx = isl_space_get_ctx(pw->dim);
1571 isl_assert(ctx, isl_space_is_domain_internal(morph->dom->dim, pw->dim),
1572 goto error);
1573
1574 pw = FN(PW,cow)(pw);
1575 if (!pw)
1576 goto error;
1577 pw->dim = isl_space_extend_domain_with_range(
1578 isl_space_copy(morph->ran->dim), pw->dim);
1579 if (!pw->dim)
1580 goto error;
1581
1582 for (i = 0; i < pw->n; ++i) {
1583 pw->p[i].set = isl_morph_set(isl_morph_copy(morph), pw->p[i].set);
1584 if (!pw->p[i].set)
1585 goto error;
1586 pw->p[i].FIELD = FN(EL,morph_domain)(pw->p[i].FIELD,
1587 isl_morph_copy(morph));
1588 if (!pw->p[i].FIELD)
1589 goto error;
1590 }
1591
1592 isl_morph_free(morph);
1593
1594 return pw;
1595 error:
1596 FN(PW,free)(pw);
1597 isl_morph_free(morph);
1598 return NULL;
1599 }
1600 #endif
1601
1602 int FN(PW,n_piece)(__isl_keep PW *pw)
1603 {
1604 return pw ? pw->n : 0;
1605 }
1606
1607 isl_stat FN(PW,foreach_piece)(__isl_keep PW *pw,
1608 isl_stat (*fn)(__isl_take isl_set *set, __isl_take EL *el, void *user),
1609 void *user)
1610 {
1611 int i;
1612
1613 if (!pw)
1614 return isl_stat_error;
1615
1616 for (i = 0; i < pw->n; ++i)
1617 if (fn(isl_set_copy(pw->p[i].set),
1618 FN(EL,copy)(pw->p[i].FIELD), user) < 0)
1619 return isl_stat_error;
1620
1621 return isl_stat_ok;
1622 }
1623
1624 #ifndef NO_LIFT
1625 static int any_divs(__isl_keep isl_set *set)
1626 {
1627 int i;
1628
1629 if (!set)
1630 return -1;
1631
1632 for (i = 0; i < set->n; ++i)
1633 if (set->p[i]->n_div > 0)
1634 return 1;
1635
1636 return 0;
1637 }
1638
1639 static isl_stat foreach_lifted_subset(__isl_take isl_set *set,
1640 __isl_take EL *el,
1641 isl_stat (*fn)(__isl_take isl_set *set, __isl_take EL *el,
1642 void *user), void *user)
1643 {
1644 int i;
1645
1646 if (!set || !el)
1647 goto error;
1648
1649 for (i = 0; i < set->n; ++i) {
1650 isl_set *lift;
1651 EL *copy;
1652
1653 lift = isl_set_from_basic_set(isl_basic_set_copy(set->p[i]));
1654 lift = isl_set_lift(lift);
1655
1656 copy = FN(EL,copy)(el);
1657 copy = FN(EL,lift)(copy, isl_set_get_space(lift));
1658
1659 if (fn(lift, copy, user) < 0)
1660 goto error;
1661 }
1662
1663 isl_set_free(set);
1664 FN(EL,free)(el);
1665
1666 return isl_stat_ok;
1667 error:
1668 isl_set_free(set);
1669 FN(EL,free)(el);
1670 return isl_stat_error;
1671 }
1672
1673 isl_stat FN(PW,foreach_lifted_piece)(__isl_keep PW *pw,
1674 isl_stat (*fn)(__isl_take isl_set *set, __isl_take EL *el,
1675 void *user), void *user)
1676 {
1677 int i;
1678
1679 if (!pw)
1680 return isl_stat_error;
1681
1682 for (i = 0; i < pw->n; ++i) {
1683 isl_set *set;
1684 EL *el;
1685
1686 set = isl_set_copy(pw->p[i].set);
1687 el = FN(EL,copy)(pw->p[i].FIELD);
1688 if (!any_divs(set)) {
1689 if (fn(set, el, user) < 0)
1690 return isl_stat_error;
1691 continue;
1692 }
1693 if (foreach_lifted_subset(set, el, fn, user) < 0)
1694 return isl_stat_error;
1695 }
1696
1697 return isl_stat_ok;
1698 }
1699 #endif
1700
1701 #ifndef NO_MOVE_DIMS
1702 __isl_give PW *FN(PW,move_dims)(__isl_take PW *pw,
1703 enum isl_dim_type dst_type, unsigned dst_pos,
1704 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
1705 {
1706 int i;
1707
1708 pw = FN(PW,cow)(pw);
1709 if (!pw)
1710 return NULL;
1711
1712 pw->dim = isl_space_move_dims(pw->dim, dst_type, dst_pos, src_type, src_pos, n);
1713 if (!pw->dim)
1714 goto error;
1715
1716 for (i = 0; i < pw->n; ++i) {
1717 pw->p[i].FIELD = FN(EL,move_dims)(pw->p[i].FIELD,
1718 dst_type, dst_pos, src_type, src_pos, n);
1719 if (!pw->p[i].FIELD)
1720 goto error;
1721 }
1722
1723 if (dst_type == isl_dim_in)
1724 dst_type = isl_dim_set;
1725 if (src_type == isl_dim_in)
1726 src_type = isl_dim_set;
1727
1728 for (i = 0; i < pw->n; ++i) {
1729 pw->p[i].set = isl_set_move_dims(pw->p[i].set,
1730 dst_type, dst_pos,
1731 src_type, src_pos, n);
1732 if (!pw->p[i].set)
1733 goto error;
1734 }
1735
1736 return pw;
1737 error:
1738 FN(PW,free)(pw);
1739 return NULL;
1740 }
1741 #endif
1742
1743 __isl_give PW *FN(PW,mul_isl_int)(__isl_take PW *pw, isl_int v)
1744 {
1745 int i;
1746
1747 if (isl_int_is_one(v))
1748 return pw;
1749 if (pw && DEFAULT_IS_ZERO && isl_int_is_zero(v)) {
1750 PW *zero;
1751 isl_space *dim = FN(PW,get_space)(pw);
1752 #ifdef HAS_TYPE
1753 zero = FN(PW,ZERO)(dim, pw->type);
1754 #else
1755 zero = FN(PW,ZERO)(dim);
1756 #endif
1757 FN(PW,free)(pw);
1758 return zero;
1759 }
1760 pw = FN(PW,cow)(pw);
1761 if (!pw)
1762 return NULL;
1763 if (pw->n == 0)
1764 return pw;
1765
1766 #ifdef HAS_TYPE
1767 if (isl_int_is_neg(v))
1768 pw->type = isl_fold_type_negate(pw->type);
1769 #endif
1770 for (i = 0; i < pw->n; ++i) {
1771 pw->p[i].FIELD = FN(EL,scale)(pw->p[i].FIELD, v);
1772 if (!pw->p[i].FIELD)
1773 goto error;
1774 }
1775
1776 return pw;
1777 error:
1778 FN(PW,free)(pw);
1779 return NULL;
1780 }
1781
1782 /* Multiply the pieces of "pw" by "v" and return the result.
1783 */
1784 __isl_give PW *FN(PW,scale_val)(__isl_take PW *pw, __isl_take isl_val *v)
1785 {
1786 int i;
1787
1788 if (!pw || !v)
1789 goto error;
1790
1791 if (isl_val_is_one(v)) {
1792 isl_val_free(v);
1793 return pw;
1794 }
1795 if (pw && DEFAULT_IS_ZERO && isl_val_is_zero(v)) {
1796 PW *zero;
1797 isl_space *space = FN(PW,get_space)(pw);
1798 #ifdef HAS_TYPE
1799 zero = FN(PW,ZERO)(space, pw->type);
1800 #else
1801 zero = FN(PW,ZERO)(space);
1802 #endif
1803 FN(PW,free)(pw);
1804 isl_val_free(v);
1805 return zero;
1806 }
1807 if (pw->n == 0) {
1808 isl_val_free(v);
1809 return pw;
1810 }
1811 pw = FN(PW,cow)(pw);
1812 if (!pw)
1813 goto error;
1814
1815 #ifdef HAS_TYPE
1816 if (isl_val_is_neg(v))
1817 pw->type = isl_fold_type_negate(pw->type);
1818 #endif
1819 for (i = 0; i < pw->n; ++i) {
1820 pw->p[i].FIELD = FN(EL,scale_val)(pw->p[i].FIELD,
1821 isl_val_copy(v));
1822 if (!pw->p[i].FIELD)
1823 goto error;
1824 }
1825
1826 isl_val_free(v);
1827 return pw;
1828 error:
1829 isl_val_free(v);
1830 FN(PW,free)(pw);
1831 return NULL;
1832 }
1833
1834 /* Divide the pieces of "pw" by "v" and return the result.
1835 */
1836 __isl_give PW *FN(PW,scale_down_val)(__isl_take PW *pw, __isl_take isl_val *v)
1837 {
1838 int i;
1839
1840 if (!pw || !v)
1841 goto error;
1842
1843 if (isl_val_is_one(v)) {
1844 isl_val_free(v);
1845 return pw;
1846 }
1847
1848 if (!isl_val_is_rat(v))
1849 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1850 "expecting rational factor", goto error);
1851 if (isl_val_is_zero(v))
1852 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1853 "cannot scale down by zero", goto error);
1854
1855 if (pw->n == 0) {
1856 isl_val_free(v);
1857 return pw;
1858 }
1859 pw = FN(PW,cow)(pw);
1860 if (!pw)
1861 goto error;
1862
1863 #ifdef HAS_TYPE
1864 if (isl_val_is_neg(v))
1865 pw->type = isl_fold_type_negate(pw->type);
1866 #endif
1867 for (i = 0; i < pw->n; ++i) {
1868 pw->p[i].FIELD = FN(EL,scale_down_val)(pw->p[i].FIELD,
1869 isl_val_copy(v));
1870 if (!pw->p[i].FIELD)
1871 goto error;
1872 }
1873
1874 isl_val_free(v);
1875 return pw;
1876 error:
1877 isl_val_free(v);
1878 FN(PW,free)(pw);
1879 return NULL;
1880 }
1881
1882 __isl_give PW *FN(PW,scale)(__isl_take PW *pw, isl_int v)
1883 {
1884 return FN(PW,mul_isl_int)(pw, v);
1885 }
1886
1887 static int FN(PW,qsort_set_cmp)(const void *p1, const void *p2)
1888 {
1889 isl_set *set1 = *(isl_set * const *)p1;
1890 isl_set *set2 = *(isl_set * const *)p2;
1891
1892 return isl_set_plain_cmp(set1, set2);
1893 }
1894
1895 /* We normalize in place, but if anything goes wrong we need
1896 * to return NULL, so we need to make sure we don't change the
1897 * meaning of any possible other copies of map.
1898 */
1899 __isl_give PW *FN(PW,normalize)(__isl_take PW *pw)
1900 {
1901 int i, j;
1902 isl_set *set;
1903
1904 if (!pw)
1905 return NULL;
1906 for (i = 0; i < pw->n; ++i) {
1907 set = isl_set_normalize(isl_set_copy(pw->p[i].set));
1908 if (!set)
1909 return FN(PW,free)(pw);
1910 isl_set_free(pw->p[i].set);
1911 pw->p[i].set = set;
1912 }
1913 qsort(pw->p, pw->n, sizeof(pw->p[0]), &FN(PW,qsort_set_cmp));
1914 for (i = pw->n - 1; i >= 1; --i) {
1915 if (!isl_set_plain_is_equal(pw->p[i - 1].set, pw->p[i].set))
1916 continue;
1917 if (!FN(EL,plain_is_equal)(pw->p[i - 1].FIELD, pw->p[i].FIELD))
1918 continue;
1919 set = isl_set_union(isl_set_copy(pw->p[i - 1].set),
1920 isl_set_copy(pw->p[i].set));
1921 if (!set)
1922 return FN(PW,free)(pw);
1923 isl_set_free(pw->p[i].set);
1924 FN(EL,free)(pw->p[i].FIELD);
1925 isl_set_free(pw->p[i - 1].set);
1926 pw->p[i - 1].set = set;
1927 for (j = i + 1; j < pw->n; ++j)
1928 pw->p[j - 1] = pw->p[j];
1929 pw->n--;
1930 }
1931
1932 return pw;
1933 }
1934
1935 /* Is pw1 obviously equal to pw2?
1936 * That is, do they have obviously identical cells and obviously identical
1937 * elements on each cell?
1938 */
1939 isl_bool FN(PW,plain_is_equal)(__isl_keep PW *pw1, __isl_keep PW *pw2)
1940 {
1941 int i;
1942 isl_bool equal;
1943
1944 if (!pw1 || !pw2)
1945 return isl_bool_error;
1946
1947 if (pw1 == pw2)
1948 return isl_bool_true;
1949 if (!isl_space_is_equal(pw1->dim, pw2->dim))
1950 return isl_bool_false;
1951
1952 pw1 = FN(PW,copy)(pw1);
1953 pw2 = FN(PW,copy)(pw2);
1954 pw1 = FN(PW,normalize)(pw1);
1955 pw2 = FN(PW,normalize)(pw2);
1956 if (!pw1 || !pw2)
1957 goto error;
1958
1959 equal = pw1->n == pw2->n;
1960 for (i = 0; equal && i < pw1->n; ++i) {
1961 equal = isl_set_plain_is_equal(pw1->p[i].set, pw2->p[i].set);
1962 if (equal < 0)
1963 goto error;
1964 if (!equal)
1965 break;
1966 equal = FN(EL,plain_is_equal)(pw1->p[i].FIELD, pw2->p[i].FIELD);
1967 if (equal < 0)
1968 goto error;
1969 }
1970
1971 FN(PW,free)(pw1);
1972 FN(PW,free)(pw2);
1973 return equal;
1974 error:
1975 FN(PW,free)(pw1);
1976 FN(PW,free)(pw2);
1977 return isl_bool_error;
1978 }
1979
1980 #ifndef NO_PULLBACK
1981 static __isl_give PW *FN(PW,align_params_pw_multi_aff_and)(__isl_take PW *pw,
1982 __isl_take isl_multi_aff *ma,
1983 __isl_give PW *(*fn)(__isl_take PW *pw, __isl_take isl_multi_aff *ma))
1984 {
1985 isl_ctx *ctx;
1986 isl_space *ma_space;
1987
1988 ma_space = isl_multi_aff_get_space(ma);
1989 if (!pw || !ma || !ma_space)
1990 goto error;
1991 if (isl_space_match(pw->dim, isl_dim_param, ma_space, isl_dim_param)) {
1992 isl_space_free(ma_space);
1993 return fn(pw, ma);
1994 }
1995 ctx = FN(PW,get_ctx)(pw);
1996 if (!isl_space_has_named_params(pw->dim) ||
1997 !isl_space_has_named_params(ma_space))
1998 isl_die(ctx, isl_error_invalid,
1999 "unaligned unnamed parameters", goto error);
2000 pw = FN(PW,align_params)(pw, ma_space);
2001 ma = isl_multi_aff_align_params(ma, FN(PW,get_space)(pw));
2002 return fn(pw, ma);
2003 error:
2004 isl_space_free(ma_space);
2005 FN(PW,free)(pw);
2006 isl_multi_aff_free(ma);
2007 return NULL;
2008 }
2009
2010 static __isl_give PW *FN(PW,align_params_pw_pw_multi_aff_and)(__isl_take PW *pw,
2011 __isl_take isl_pw_multi_aff *pma,
2012 __isl_give PW *(*fn)(__isl_take PW *pw,
2013 __isl_take isl_pw_multi_aff *ma))
2014 {
2015 isl_ctx *ctx;
2016 isl_space *pma_space;
2017
2018 pma_space = isl_pw_multi_aff_get_space(pma);
2019 if (!pw || !pma || !pma_space)
2020 goto error;
2021 if (isl_space_match(pw->dim, isl_dim_param, pma_space, isl_dim_param)) {
2022 isl_space_free(pma_space);
2023 return fn(pw, pma);
2024 }
2025 ctx = FN(PW,get_ctx)(pw);
2026 if (!isl_space_has_named_params(pw->dim) ||
2027 !isl_space_has_named_params(pma_space))
2028 isl_die(ctx, isl_error_invalid,
2029 "unaligned unnamed parameters", goto error);
2030 pw = FN(PW,align_params)(pw, pma_space);
2031 pma = isl_pw_multi_aff_align_params(pma, FN(PW,get_space)(pw));
2032 return fn(pw, pma);
2033 error:
2034 isl_space_free(pma_space);
2035 FN(PW,free)(pw);
2036 isl_pw_multi_aff_free(pma);
2037 return NULL;
2038 }
2039
2040 /* Compute the pullback of "pw" by the function represented by "ma".
2041 * In other words, plug in "ma" in "pw".
2042 */
2043 static __isl_give PW *FN(PW,pullback_multi_aff_aligned)(__isl_take PW *pw,
2044 __isl_take isl_multi_aff *ma)
2045 {
2046 int i;
2047 isl_space *space = NULL;
2048
2049 ma = isl_multi_aff_align_divs(ma);
2050 pw = FN(PW,cow)(pw);
2051 if (!pw || !ma)
2052 goto error;
2053
2054 space = isl_space_join(isl_multi_aff_get_space(ma),
2055 FN(PW,get_space)(pw));
2056
2057 for (i = 0; i < pw->n; ++i) {
2058 pw->p[i].set = isl_set_preimage_multi_aff(pw->p[i].set,
2059 isl_multi_aff_copy(ma));
2060 if (!pw->p[i].set)
2061 goto error;
2062 pw->p[i].FIELD = FN(EL,pullback_multi_aff)(pw->p[i].FIELD,
2063 isl_multi_aff_copy(ma));
2064 if (!pw->p[i].FIELD)
2065 goto error;
2066 }
2067
2068 pw = FN(PW,reset_space)(pw, space);
2069 isl_multi_aff_free(ma);
2070 return pw;
2071 error:
2072 isl_space_free(space);
2073 isl_multi_aff_free(ma);
2074 FN(PW,free)(pw);
2075 return NULL;
2076 }
2077
2078 __isl_give PW *FN(PW,pullback_multi_aff)(__isl_take PW *pw,
2079 __isl_take isl_multi_aff *ma)
2080 {
2081 return FN(PW,align_params_pw_multi_aff_and)(pw, ma,
2082 &FN(PW,pullback_multi_aff_aligned));
2083 }
2084
2085 /* Compute the pullback of "pw" by the function represented by "pma".
2086 * In other words, plug in "pma" in "pw".
2087 */
2088 static __isl_give PW *FN(PW,pullback_pw_multi_aff_aligned)(__isl_take PW *pw,
2089 __isl_take isl_pw_multi_aff *pma)
2090 {
2091 int i;
2092 PW *res;
2093
2094 if (!pma)
2095 goto error;
2096
2097 if (pma->n == 0) {
2098 isl_space *space;
2099 space = isl_space_join(isl_pw_multi_aff_get_space(pma),
2100 FN(PW,get_space)(pw));
2101 isl_pw_multi_aff_free(pma);
2102 res = FN(PW,empty)(space);
2103 FN(PW,free)(pw);
2104 return res;
2105 }
2106
2107 res = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
2108 isl_multi_aff_copy(pma->p[0].maff));
2109 res = FN(PW,intersect_domain)(res, isl_set_copy(pma->p[0].set));
2110
2111 for (i = 1; i < pma->n; ++i) {
2112 PW *res_i;
2113
2114 res_i = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
2115 isl_multi_aff_copy(pma->p[i].maff));
2116 res_i = FN(PW,intersect_domain)(res_i,
2117 isl_set_copy(pma->p[i].set));
2118 res = FN(PW,add_disjoint)(res, res_i);
2119 }
2120
2121 isl_pw_multi_aff_free(pma);
2122 FN(PW,free)(pw);
2123 return res;
2124 error:
2125 isl_pw_multi_aff_free(pma);
2126 FN(PW,free)(pw);
2127 return NULL;
2128 }
2129
2130 __isl_give PW *FN(PW,pullback_pw_multi_aff)(__isl_take PW *pw,
2131 __isl_take isl_pw_multi_aff *pma)
2132 {
2133 return FN(PW,align_params_pw_pw_multi_aff_and)(pw, pma,
2134 &FN(PW,pullback_pw_multi_aff_aligned));
2135 }
2136 #endif
Integer set library