?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
add isl_multi_pw_aff_intersect_domain
[isl.git] / isl_multi_templ.c
blob1184f9170fe788d1b3b784c7327d76094231ba11
1 /*
2 * Copyright 2011 Sven Verdoolaege
3 * Copyright 2012-2013 Ecole Normale Superieure
4 *
5 * Use of this software is governed by the MIT license
6 *
7 * Written by Sven Verdoolaege,
8 * Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
9 */
10
11 #include <isl_space_private.h>
12 #include <isl/set.h>
13 #include <isl_reordering.h>
14
15 #define xCAT(A,B) A ## B
16 #define CAT(A,B) xCAT(A,B)
17 #undef EL
18 #define EL CAT(isl_,BASE)
19 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
20 #define FN(TYPE,NAME) xFN(TYPE,NAME)
21 #define xMULTI(BASE) isl_multi_ ## BASE
22 #define MULTI(BASE) xMULTI(BASE)
23 #define MULTI_NAME(BASE) "isl_multi_" #BASE
24 #define xLIST(EL) EL ## _list
25 #define LIST(EL) xLIST(EL)
26
27 isl_ctx *FN(MULTI(BASE),get_ctx)(__isl_keep MULTI(BASE) *multi)
28 {
29 return multi ? isl_space_get_ctx(multi->space) : NULL;
30 }
31
32 __isl_give isl_space *FN(MULTI(BASE),get_space)(__isl_keep MULTI(BASE) *multi)
33 {
34 return multi ? isl_space_copy(multi->space) : NULL;
35 }
36
37 __isl_give isl_space *FN(MULTI(BASE),get_domain_space)(
38 __isl_keep MULTI(BASE) *multi)
39 {
40 return multi ? isl_space_domain(isl_space_copy(multi->space)) : NULL;
41 }
42
43 __isl_give MULTI(BASE) *FN(MULTI(BASE),alloc)(__isl_take isl_space *space)
44 {
45 isl_ctx *ctx;
46 int n;
47 MULTI(BASE) *multi;
48
49 if (!space)
50 return NULL;
51
52 ctx = isl_space_get_ctx(space);
53 n = isl_space_dim(space, isl_dim_out);
54 multi = isl_calloc(ctx, MULTI(BASE),
55 sizeof(MULTI(BASE)) + (n - 1) * sizeof(struct EL *));
56 if (!multi)
57 goto error;
58
59 multi->space = space;
60 multi->n = n;
61 multi->ref = 1;
62 return multi;
63 error:
64 isl_space_free(space);
65 return NULL;
66 }
67
68 __isl_give MULTI(BASE) *FN(MULTI(BASE),dup)(__isl_keep MULTI(BASE) *multi)
69 {
70 int i;
71 MULTI(BASE) *dup;
72
73 if (!multi)
74 return NULL;
75
76 dup = FN(MULTI(BASE),alloc)(isl_space_copy(multi->space));
77 if (!dup)
78 return NULL;
79
80 for (i = 0; i < multi->n; ++i)
81 dup = FN(FN(MULTI(BASE),set),BASE)(dup, i,
82 FN(EL,copy)(multi->p[i]));
83
84 return dup;
85 }
86
87 __isl_give MULTI(BASE) *FN(MULTI(BASE),cow)(__isl_take MULTI(BASE) *multi)
88 {
89 if (!multi)
90 return NULL;
91
92 if (multi->ref == 1)
93 return multi;
94
95 multi->ref--;
96 return FN(MULTI(BASE),dup)(multi);
97 }
98
99 __isl_give MULTI(BASE) *FN(MULTI(BASE),copy)(__isl_keep MULTI(BASE) *multi)
100 {
101 if (!multi)
102 return NULL;
103
104 multi->ref++;
105 return multi;
106 }
107
108 void *FN(MULTI(BASE),free)(__isl_take MULTI(BASE) *multi)
109 {
110 int i;
111
112 if (!multi)
113 return NULL;
114
115 if (--multi->ref > 0)
116 return NULL;
117
118 isl_space_free(multi->space);
119 for (i = 0; i < multi->n; ++i)
120 FN(EL,free)(multi->p[i]);
121 free(multi);
122
123 return NULL;
124 }
125
126 __isl_give MULTI(BASE) *FN(MULTI(BASE),insert_dims)(
127 __isl_take MULTI(BASE) *multi,
128 enum isl_dim_type type, unsigned first, unsigned n)
129 {
130 int i;
131
132 if (!multi)
133 return NULL;
134 if (type == isl_dim_out)
135 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
136 "cannot insert output/set dimensions",
137 return FN(MULTI(BASE),free)(multi));
138 if (n == 0 && !isl_space_is_named_or_nested(multi->space, type))
139 return multi;
140
141 multi = FN(MULTI(BASE),cow)(multi);
142 if (!multi)
143 return NULL;
144
145 multi->space = isl_space_insert_dims(multi->space, type, first, n);
146 if (!multi->space)
147 return FN(MULTI(BASE),free)(multi);
148
149 for (i = 0; i < multi->n; ++i) {
150 multi->p[i] = FN(EL,insert_dims)(multi->p[i], type, first, n);
151 if (!multi->p[i])
152 return FN(MULTI(BASE),free)(multi);
153 }
154
155 return multi;
156 }
157
158 __isl_give MULTI(BASE) *FN(MULTI(BASE),add_dims)(__isl_take MULTI(BASE) *multi,
159 enum isl_dim_type type, unsigned n)
160 {
161 unsigned pos;
162
163 pos = FN(MULTI(BASE),dim)(multi, type);
164
165 return FN(MULTI(BASE),insert_dims)(multi, type, pos, n);
166 }
167
168 unsigned FN(MULTI(BASE),dim)(__isl_keep MULTI(BASE) *multi,
169 enum isl_dim_type type)
170 {
171 return multi ? isl_space_dim(multi->space, type) : 0;
172 }
173
174 /* Return the position of the first dimension of "type" with id "id".
175 * Return -1 if there is no such dimension.
176 */
177 int FN(MULTI(BASE),find_dim_by_id)(__isl_keep MULTI(BASE) *multi,
178 enum isl_dim_type type, __isl_keep isl_id *id)
179 {
180 if (!multi)
181 return -1;
182 return isl_space_find_dim_by_id(multi->space, type, id);
183 }
184
185 /* Return the id of the given dimension.
186 */
187 __isl_give isl_id *FN(MULTI(BASE),get_dim_id)(__isl_keep MULTI(BASE) *multi,
188 enum isl_dim_type type, unsigned pos)
189 {
190 return multi ? isl_space_get_dim_id(multi->space, type, pos) : NULL;
191 }
192
193 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_name)(
194 __isl_take MULTI(BASE) *multi,
195 enum isl_dim_type type, unsigned pos, const char *s)
196 {
197 int i;
198
199 multi = FN(MULTI(BASE),cow)(multi);
200 if (!multi)
201 return NULL;
202
203 multi->space = isl_space_set_dim_name(multi->space, type, pos, s);
204 if (!multi->space)
205 return FN(MULTI(BASE),free)(multi);
206
207 if (type == isl_dim_out)
208 return multi;
209 for (i = 0; i < multi->n; ++i) {
210 multi->p[i] = FN(EL,set_dim_name)(multi->p[i], type, pos, s);
211 if (!multi->p[i])
212 return FN(MULTI(BASE),free)(multi);
213 }
214
215 return multi;
216 }
217
218 const char *FN(MULTI(BASE),get_tuple_name)(__isl_keep MULTI(BASE) *multi,
219 enum isl_dim_type type)
220 {
221 return multi ? isl_space_get_tuple_name(multi->space, type) : NULL;
222 }
223
224 /* Does the specified tuple have an id?
225 */
226 int FN(MULTI(BASE),has_tuple_id)(__isl_keep MULTI(BASE) *multi,
227 enum isl_dim_type type)
228 {
229 return multi ? isl_space_has_tuple_id(multi->space, type) : -1;
230 }
231
232 /* Return the id of the specified tuple.
233 */
234 __isl_give isl_id *FN(MULTI(BASE),get_tuple_id)(__isl_keep MULTI(BASE) *multi,
235 enum isl_dim_type type)
236 {
237 return multi ? isl_space_get_tuple_id(multi->space, type) : NULL;
238 }
239
240 __isl_give EL *FN(FN(MULTI(BASE),get),BASE)(__isl_keep MULTI(BASE) *multi,
241 int pos)
242 {
243 isl_ctx *ctx;
244
245 if (!multi)
246 return NULL;
247 ctx = FN(MULTI(BASE),get_ctx)(multi);
248 if (pos < 0 || pos >= multi->n)
249 isl_die(ctx, isl_error_invalid,
250 "index out of bounds", return NULL);
251 return FN(EL,copy)(multi->p[pos]);
252 }
253
254 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),set),BASE)(
255 __isl_take MULTI(BASE) *multi, int pos, __isl_take EL *el)
256 {
257 isl_space *multi_space = NULL;
258 isl_space *el_space = NULL;
259 int match;
260
261 multi = FN(MULTI(BASE),cow)(multi);
262 if (!multi || !el)
263 goto error;
264
265 multi_space = FN(MULTI(BASE),get_space)(multi);
266 match = FN(EL,matching_params)(el, multi_space);
267 if (match < 0)
268 goto error;
269 if (!match) {
270 multi = FN(MULTI(BASE),align_params)(multi,
271 FN(EL,get_space)(el));
272 isl_space_free(multi_space);
273 multi_space = FN(MULTI(BASE),get_space)(multi);
274 el = FN(EL,align_params)(el, isl_space_copy(multi_space));
275 }
276 if (FN(EL,check_match_domain_space)(el, multi_space) < 0)
277 goto error;
278
279 if (pos < 0 || pos >= multi->n)
280 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
281 "index out of bounds", goto error);
282
283 FN(EL,free)(multi->p[pos]);
284 multi->p[pos] = el;
285
286 isl_space_free(multi_space);
287 isl_space_free(el_space);
288
289 return multi;
290 error:
291 FN(MULTI(BASE),free)(multi);
292 FN(EL,free)(el);
293 isl_space_free(multi_space);
294 isl_space_free(el_space);
295 return NULL;
296 }
297
298 /* Reset the space of "multi". This function is called from isl_pw_templ.c
299 * and doesn't know if the space of an element object is represented
300 * directly or through its domain. It therefore passes along both,
301 * which we pass along to the element function since we don't how
302 * that is represented either.
303 */
304 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space_and_domain)(
305 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space,
306 __isl_take isl_space *domain)
307 {
308 int i;
309
310 multi = FN(MULTI(BASE),cow)(multi);
311 if (!multi || !space || !domain)
312 goto error;
313
314 for (i = 0; i < multi->n; ++i) {
315 multi->p[i] = FN(EL,reset_domain_space)(multi->p[i],
316 isl_space_copy(domain));
317 if (!multi->p[i])
318 goto error;
319 }
320 isl_space_free(domain);
321 isl_space_free(multi->space);
322 multi->space = space;
323
324 return multi;
325 error:
326 isl_space_free(domain);
327 isl_space_free(space);
328 FN(MULTI(BASE),free)(multi);
329 return NULL;
330 }
331
332 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_domain_space)(
333 __isl_take MULTI(BASE) *multi, __isl_take isl_space *domain)
334 {
335 isl_space *space;
336
337 space = isl_space_extend_domain_with_range(isl_space_copy(domain),
338 isl_space_copy(multi->space));
339 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
340 }
341
342 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space)(
343 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space)
344 {
345 isl_space *domain;
346
347 domain = isl_space_domain(isl_space_copy(space));
348 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
349 }
350
351 /* Set the id of the given dimension of "multi" to "id".
352 */
353 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_id)(
354 __isl_take MULTI(BASE) *multi,
355 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
356 {
357 isl_space *space;
358
359 multi = FN(MULTI(BASE),cow)(multi);
360 if (!multi || !id)
361 goto error;
362
363 space = FN(MULTI(BASE),get_space)(multi);
364 space = isl_space_set_dim_id(space, type, pos, id);
365
366 return FN(MULTI(BASE),reset_space)(multi, space);
367 error:
368 isl_id_free(id);
369 FN(MULTI(BASE),free)(multi);
370 return NULL;
371 }
372
373 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_tuple_name)(
374 __isl_keep MULTI(BASE) *multi, enum isl_dim_type type,
375 const char *s)
376 {
377 isl_space *space;
378
379 multi = FN(MULTI(BASE),cow)(multi);
380 if (!multi)
381 return NULL;
382
383 space = FN(MULTI(BASE),get_space)(multi);
384 space = isl_space_set_tuple_name(space, type, s);
385
386 return FN(MULTI(BASE),reset_space)(multi, space);
387 }
388
389 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_tuple_id)(
390 __isl_take MULTI(BASE) *multi, enum isl_dim_type type,
391 __isl_take isl_id *id)
392 {
393 isl_space *space;
394
395 multi = FN(MULTI(BASE),cow)(multi);
396 if (!multi)
397 return isl_id_free(id);
398
399 space = FN(MULTI(BASE),get_space)(multi);
400 space = isl_space_set_tuple_id(space, type, id);
401
402 return FN(MULTI(BASE),reset_space)(multi, space);
403 }
404
405 /* Drop the id on the specified tuple.
406 */
407 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_tuple_id)(
408 __isl_take MULTI(BASE) *multi, enum isl_dim_type type)
409 {
410 isl_space *space;
411
412 if (!multi)
413 return NULL;
414 if (!FN(MULTI(BASE),has_tuple_id)(multi, type))
415 return multi;
416
417 multi = FN(MULTI(BASE),cow)(multi);
418 if (!multi)
419 return NULL;
420
421 space = FN(MULTI(BASE),get_space)(multi);
422 space = isl_space_reset_tuple_id(space, type);
423
424 return FN(MULTI(BASE),reset_space)(multi, space);
425 }
426
427 __isl_give MULTI(BASE) *FN(MULTI(BASE),realign_domain)(
428 __isl_take MULTI(BASE) *multi, __isl_take isl_reordering *exp)
429 {
430 int i;
431
432 multi = FN(MULTI(BASE),cow)(multi);
433 if (!multi || !exp)
434 goto error;
435
436 for (i = 0; i < multi->n; ++i) {
437 multi->p[i] = FN(EL,realign_domain)(multi->p[i],
438 isl_reordering_copy(exp));
439 if (!multi->p[i])
440 goto error;
441 }
442
443 multi = FN(MULTI(BASE),reset_domain_space)(multi,
444 isl_space_copy(exp->dim));
445
446 isl_reordering_free(exp);
447 return multi;
448 error:
449 isl_reordering_free(exp);
450 FN(MULTI(BASE),free)(multi);
451 return NULL;
452 }
453
454 /* Align the parameters of "multi" to those of "model".
455 */
456 __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params)(
457 __isl_take MULTI(BASE) *multi, __isl_take isl_space *model)
458 {
459 isl_ctx *ctx;
460
461 if (!multi || !model)
462 goto error;
463
464 ctx = isl_space_get_ctx(model);
465 if (!isl_space_has_named_params(model))
466 isl_die(ctx, isl_error_invalid,
467 "model has unnamed parameters", goto error);
468 if (!isl_space_has_named_params(multi->space))
469 isl_die(ctx, isl_error_invalid,
470 "input has unnamed parameters", goto error);
471 if (!isl_space_match(multi->space, isl_dim_param,
472 model, isl_dim_param)) {
473 isl_reordering *exp;
474
475 model = isl_space_params(model);
476 exp = isl_parameter_alignment_reordering(multi->space, model);
477 exp = isl_reordering_extend_space(exp,
478 FN(MULTI(BASE),get_domain_space)(multi));
479 multi = FN(MULTI(BASE),realign_domain)(multi, exp);
480 }
481
482 isl_space_free(model);
483 return multi;
484 error:
485 isl_space_free(model);
486 FN(MULTI(BASE),free)(multi);
487 return NULL;
488 }
489
490 #if !defined(NO_GIST) || !defined(NO_INTERSECT_DOMAIN)
491 static __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params_multi_set_and)(
492 __isl_take MULTI(BASE) *multi, __isl_take isl_set *set,
493 __isl_give MULTI(BASE) *(*fn)(__isl_take MULTI(BASE) *multi,
494 __isl_take isl_set *set))
495 {
496 isl_ctx *ctx;
497
498 if (!multi || !set)
499 goto error;
500 if (isl_space_match(multi->space, isl_dim_param,
501 set->dim, isl_dim_param))
502 return fn(multi, set);
503 ctx = FN(MULTI(BASE),get_ctx)(multi);
504 if (!isl_space_has_named_params(multi->space) ||
505 !isl_space_has_named_params(set->dim))
506 isl_die(ctx, isl_error_invalid,
507 "unaligned unnamed parameters", goto error);
508 multi = FN(MULTI(BASE),align_params)(multi, isl_set_get_space(set));
509 set = isl_set_align_params(set, FN(MULTI(BASE),get_space)(multi));
510 return fn(multi, set);
511 error:
512 FN(MULTI(BASE),free)(multi);
513 isl_set_free(set);
514 return NULL;
515 }
516 #endif
517
518 #ifndef NO_GIST
519 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist_aligned)(
520 __isl_take MULTI(BASE) *multi, __isl_take isl_set *context)
521 {
522 int i;
523
524 multi = FN(MULTI(BASE),cow)(multi);
525 if (!multi || !context)
526 goto error;
527
528 for (i = 0; i < multi->n; ++i) {
529 multi->p[i] = FN(EL,gist)(multi->p[i], isl_set_copy(context));
530 if (!multi->p[i])
531 goto error;
532 }
533
534 isl_set_free(context);
535 return multi;
536 error:
537 isl_set_free(context);
538 FN(MULTI(BASE),free)(multi);
539 return NULL;
540 }
541
542 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist)(__isl_take MULTI(BASE) *multi,
543 __isl_take isl_set *context)
544 {
545 return FN(MULTI(BASE),align_params_multi_set_and)(multi, context,
546 &FN(MULTI(BASE),gist_aligned));
547 }
548
549 __isl_give MULTI(BASE) *FN(MULTI(BASE),gist_params)(
550 __isl_take MULTI(BASE) *multi, __isl_take isl_set *context)
551 {
552 isl_space *space = FN(MULTI(BASE),get_domain_space)(multi);
553 isl_set *dom_context = isl_set_universe(space);
554 dom_context = isl_set_intersect_params(dom_context, context);
555 return FN(MULTI(BASE),gist)(multi, dom_context);
556 }
557 #endif
558
559 #ifndef NO_INTERSECT_DOMAIN
560 /* Intersect the domain of "multi" with "domain".
561 *
562 * The parameters of "multi" and "domain" are assumed to have been aligned.
563 */
564 __isl_give MULTI(BASE) *FN(MULTI(BASE),intersect_domain_aligned)(
565 __isl_take MULTI(BASE) *multi, __isl_take isl_set *domain)
566 {
567 int i;
568
569 multi = FN(MULTI(BASE),cow)(multi);
570 if (!multi || !domain)
571 goto error;
572
573 for (i = 0; i < multi->n; ++i) {
574 multi->p[i] = FN(EL,intersect_domain)(multi->p[i],
575 isl_set_copy(domain));
576 if (!multi->p[i])
577 goto error;
578 }
579
580 isl_set_free(domain);
581 return multi;
582 error:
583 isl_set_free(domain);
584 FN(MULTI(BASE),free)(multi);
585 return NULL;
586 }
587
588 /* Intersect the domain of "multi" with "domain".
589 */
590 __isl_give MULTI(BASE) *FN(MULTI(BASE),intersect_domain)(
591 __isl_take MULTI(BASE) *multi, __isl_take isl_set *domain)
592 {
593 return FN(MULTI(BASE),align_params_multi_set_and)(multi, domain,
594 &FN(MULTI(BASE),intersect_domain_aligned));
595 }
596 #endif
597
598 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),LIST(BASE))(
599 __isl_take isl_space *space, __isl_take LIST(EL) *list)
600 {
601 int i;
602 int n;
603 isl_ctx *ctx;
604 MULTI(BASE) *multi;
605
606 if (!space || !list)
607 goto error;
608
609 ctx = isl_space_get_ctx(space);
610 n = FN(FN(LIST(EL),n),BASE)(list);
611 if (n != isl_space_dim(space, isl_dim_out))
612 isl_die(ctx, isl_error_invalid,
613 "invalid number of elements in list", goto error);
614
615 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
616 for (i = 0; i < n; ++i) {
617 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
618 FN(FN(LIST(EL),get),BASE)(list, i));
619 }
620
621 isl_space_free(space);
622 FN(LIST(EL),free)(list);
623 return multi;
624 error:
625 isl_space_free(space);
626 FN(LIST(EL),free)(list);
627 return NULL;
628 }
629
630 #ifndef NO_IDENTITY
631 /* Create a multi expression in the given space that maps each
632 * input dimension to the corresponding output dimension.
633 */
634 __isl_give MULTI(BASE) *FN(MULTI(BASE),identity)(__isl_take isl_space *space)
635 {
636 int i, n;
637 isl_local_space *ls;
638 MULTI(BASE) *multi;
639
640 if (!space)
641 return NULL;
642
643 if (isl_space_is_set(space))
644 isl_die(isl_space_get_ctx(space), isl_error_invalid,
645 "expecting map space", goto error);
646
647 n = isl_space_dim(space, isl_dim_out);
648 if (n != isl_space_dim(space, isl_dim_in))
649 isl_die(isl_space_get_ctx(space), isl_error_invalid,
650 "number of input and output dimensions needs to be "
651 "the same", goto error);
652
653 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
654
655 if (!n) {
656 isl_space_free(space);
657 return multi;
658 }
659
660 space = isl_space_domain(space);
661 ls = isl_local_space_from_space(space);
662
663 for (i = 0; i < n; ++i) {
664 EL *el;
665 el = FN(EL,var_on_domain)(isl_local_space_copy(ls),
666 isl_dim_set, i);
667 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i, el);
668 }
669
670 isl_local_space_free(ls);
671
672 return multi;
673 error:
674 isl_space_free(space);
675 return NULL;
676 }
677 #endif
678
679 /* Construct a multi expression in the given space with value zero in
680 * each of the output dimensions.
681 */
682 __isl_give MULTI(BASE) *FN(MULTI(BASE),zero)(__isl_take isl_space *space)
683 {
684 int n;
685 MULTI(BASE) *multi;
686
687 if (!space)
688 return NULL;
689
690 n = isl_space_dim(space , isl_dim_out);
691 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
692
693 if (!n)
694 isl_space_free(space);
695 else {
696 int i;
697 isl_local_space *ls;
698 EL *el;
699
700 space = isl_space_domain(space);
701 ls = isl_local_space_from_space(space);
702 el = FN(EL,zero_on_domain)(ls);
703
704 for (i = 0; i < n; ++i)
705 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
706 FN(EL,copy)(el));
707
708 FN(EL,free)(el);
709 }
710
711 return multi;
712 }
713
714 #ifndef NO_FROM_BASE
715 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),BASE)(__isl_take EL *el)
716 {
717 MULTI(BASE) *multi;
718
719 multi = FN(MULTI(BASE),alloc)(FN(EL,get_space)(el));
720 multi = FN(FN(MULTI(BASE),set),BASE)(multi, 0, el);
721
722 return multi;
723 }
724 #endif
725
726 __isl_give MULTI(BASE) *FN(MULTI(BASE),drop_dims)(
727 __isl_take MULTI(BASE) *multi,
728 enum isl_dim_type type, unsigned first, unsigned n)
729 {
730 int i;
731 unsigned dim;
732
733 multi = FN(MULTI(BASE),cow)(multi);
734 if (!multi)
735 return NULL;
736
737 dim = FN(MULTI(BASE),dim)(multi, type);
738 if (first + n > dim || first + n < first)
739 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
740 "index out of bounds",
741 return FN(MULTI(BASE),cow)(multi));
742
743 multi->space = isl_space_drop_dims(multi->space, type, first, n);
744 if (!multi->space)
745 return FN(MULTI(BASE),cow)(multi);
746
747 if (type == isl_dim_out) {
748 for (i = 0; i < n; ++i)
749 FN(EL,free)(multi->p[first + i]);
750 for (i = first; i + n < multi->n; ++i)
751 multi->p[i] = multi->p[i + n];
752 multi->n -= n;
753
754 return multi;
755 }
756
757 for (i = 0; i < multi->n; ++i) {
758 multi->p[i] = FN(EL,drop_dims)(multi->p[i], type, first, n);
759 if (!multi->p[i])
760 return FN(MULTI(BASE),cow)(multi);
761 }
762
763 return multi;
764 }
765
766 /* Align the parameters of "multi1" and "multi2" (if needed) and call "fn".
767 */
768 static __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params_multi_multi_and)(
769 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
770 __isl_give MULTI(BASE) *(*fn)(__isl_take MULTI(BASE) *multi1,
771 __isl_take MULTI(BASE) *multi2))
772 {
773 isl_ctx *ctx;
774
775 if (!multi1 || !multi2)
776 goto error;
777 if (isl_space_match(multi1->space, isl_dim_param,
778 multi2->space, isl_dim_param))
779 return fn(multi1, multi2);
780 ctx = FN(MULTI(BASE),get_ctx)(multi1);
781 if (!isl_space_has_named_params(multi1->space) ||
782 !isl_space_has_named_params(multi2->space))
783 isl_die(ctx, isl_error_invalid,
784 "unaligned unnamed parameters", goto error);
785 multi1 = FN(MULTI(BASE),align_params)(multi1,
786 FN(MULTI(BASE),get_space)(multi2));
787 multi2 = FN(MULTI(BASE),align_params)(multi2,
788 FN(MULTI(BASE),get_space)(multi1));
789 return fn(multi1, multi2);
790 error:
791 FN(MULTI(BASE),free)(multi1);
792 FN(MULTI(BASE),free)(multi2);
793 return NULL;
794 }
795
796 /* Given two MULTI(BASE)s A -> B and C -> D,
797 * construct a MULTI(BASE) (A * C) -> (B, D).
798 *
799 * The parameters are assumed to have been aligned.
800 */
801 static __isl_give MULTI(BASE) *FN(MULTI(BASE),range_product_aligned)(
802 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
803 {
804 int i, n1, n2;
805 EL *el;
806 isl_space *space;
807 MULTI(BASE) *res;
808
809 if (!multi1 || !multi2)
810 goto error;
811
812 space = isl_space_range_product(FN(MULTI(BASE),get_space)(multi1),
813 FN(MULTI(BASE),get_space)(multi2));
814 res = FN(MULTI(BASE),alloc)(space);
815
816 n1 = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
817 n2 = FN(MULTI(BASE),dim)(multi2, isl_dim_out);
818
819 for (i = 0; i < n1; ++i) {
820 el = FN(FN(MULTI(BASE),get),BASE)(multi1, i);
821 res = FN(FN(MULTI(BASE),set),BASE)(res, i, el);
822 }
823
824 for (i = 0; i < n2; ++i) {
825 el = FN(FN(MULTI(BASE),get),BASE)(multi2, i);
826 res = FN(FN(MULTI(BASE),set),BASE)(res, n1 + i, el);
827 }
828
829 FN(MULTI(BASE),free)(multi1);
830 FN(MULTI(BASE),free)(multi2);
831 return res;
832 error:
833 FN(MULTI(BASE),free)(multi1);
834 FN(MULTI(BASE),free)(multi2);
835 return NULL;
836 }
837
838 /* Given two MULTI(BASE)s A -> B and C -> D,
839 * construct a MULTI(BASE) (A * C) -> (B, D).
840 */
841 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_product)(
842 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
843 {
844 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
845 &FN(MULTI(BASE),range_product_aligned));
846 }
847
848 /* Given two MULTI(BASE)s A -> B and C -> D,
849 * construct a MULTI(BASE) [A -> C] -> [B -> D].
850 *
851 * The parameters are assumed to have been aligned.
852 */
853 __isl_give MULTI(BASE) *FN(MULTI(BASE),product_aligned)(
854 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
855 {
856 int i;
857 EL *el;
858 isl_space *space;
859 MULTI(BASE) *res;
860 int in1, in2, out1, out2;
861
862 in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
863 in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
864 out1 = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
865 out2 = FN(MULTI(BASE),dim)(multi2, isl_dim_out);
866 space = isl_space_product(FN(MULTI(BASE),get_space)(multi1),
867 FN(MULTI(BASE),get_space)(multi2));
868 res = FN(MULTI(BASE),alloc)(isl_space_copy(space));
869 space = isl_space_domain(space);
870
871 for (i = 0; i < out1; ++i) {
872 el = FN(FN(MULTI(BASE),get),BASE)(multi1, i);
873 el = FN(EL,insert_dims)(el, isl_dim_in, in1, in2);
874 el = FN(EL,reset_domain_space)(el, isl_space_copy(space));
875 res = FN(FN(MULTI(BASE),set),BASE)(res, i, el);
876 }
877
878 for (i = 0; i < out2; ++i) {
879 el = FN(FN(MULTI(BASE),get),BASE)(multi2, i);
880 el = FN(EL,insert_dims)(el, isl_dim_in, 0, in1);
881 el = FN(EL,reset_domain_space)(el, isl_space_copy(space));
882 res = FN(FN(MULTI(BASE),set),BASE)(res, out1 + i, el);
883 }
884
885 isl_space_free(space);
886 FN(MULTI(BASE),free)(multi1);
887 FN(MULTI(BASE),free)(multi2);
888 return res;
889 }
890
891 /* Given two MULTI(BASE)s A -> B and C -> D,
892 * construct a MULTI(BASE) [A -> C] -> [B -> D].
893 */
894 __isl_give MULTI(BASE) *FN(MULTI(BASE),product)(
895 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
896 {
897 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
898 &FN(MULTI(BASE),product_aligned));
899 }
900
901 __isl_give MULTI(BASE) *FN(MULTI(BASE),flatten_range)(
902 __isl_take MULTI(BASE) *multi)
903 {
904 if (!multi)
905 return NULL;
906
907 if (!multi->space->nested[1])
908 return multi;
909
910 multi = FN(MULTI(BASE),cow)(multi);
911 if (!multi)
912 return NULL;
913
914 multi->space = isl_space_flatten_range(multi->space);
915 if (!multi->space)
916 return FN(MULTI(BASE),free)(multi);
917
918 return multi;
919 }
920
921 /* Given two MULTI(BASE)s A -> B and C -> D,
922 * construct a MULTI(BASE) (A * C) -> [B -> D].
923 */
924 __isl_give MULTI(BASE) *FN(MULTI(BASE),flat_range_product)(
925 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
926 {
927 MULTI(BASE) *multi;
928
929 multi = FN(MULTI(BASE),range_product)(multi1, multi2);
930 multi = FN(MULTI(BASE),flatten_range)(multi);
931 return multi;
932 }
933
934 /* Given two multi expressions, "multi1"
935 *
936 * [A] -> [B1 B2]
937 *
938 * where B2 starts at position "pos", and "multi2"
939 *
940 * [A] -> [D]
941 *
942 * return the multi expression
943 *
944 * [A] -> [B1 D B2]
945 */
946 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_splice)(
947 __isl_take MULTI(BASE) *multi1, unsigned pos,
948 __isl_take MULTI(BASE) *multi2)
949 {
950 MULTI(BASE) *res;
951 unsigned dim;
952
953 if (!multi1 || !multi2)
954 goto error;
955
956 dim = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
957 if (pos > dim)
958 isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
959 "index out of bounds", goto error);
960
961 res = FN(MULTI(BASE),copy)(multi1);
962 res = FN(MULTI(BASE),drop_dims)(res, isl_dim_out, pos, dim - pos);
963 multi1 = FN(MULTI(BASE),drop_dims)(multi1, isl_dim_out, 0, pos);
964
965 res = FN(MULTI(BASE),flat_range_product)(res, multi2);
966 res = FN(MULTI(BASE),flat_range_product)(res, multi1);
967
968 return res;
969 error:
970 FN(MULTI(BASE),free)(multi1);
971 FN(MULTI(BASE),free)(multi2);
972 return NULL;
973 }
974
975 /* Given two multi expressions, "multi1"
976 *
977 * [A1 A2] -> [B1 B2]
978 *
979 * where A2 starts at position "in_pos" and B2 starts at position "out_pos",
980 * and "multi2"
981 *
982 * [C] -> [D]
983 *
984 * return the multi expression
985 *
986 * [A1 C A2] -> [B1 D B2]
987 *
988 * We first insert input dimensions to obtain
989 *
990 * [A1 C A2] -> [B1 B2]
991 *
992 * and
993 *
994 * [A1 C A2] -> [D]
995 *
996 * and then apply range_splice.
997 */
998 __isl_give MULTI(BASE) *FN(MULTI(BASE),splice)(
999 __isl_take MULTI(BASE) *multi1, unsigned in_pos, unsigned out_pos,
1000 __isl_take MULTI(BASE) *multi2)
1001 {
1002 unsigned n_in1;
1003 unsigned n_in2;
1004
1005 if (!multi1 || !multi2)
1006 goto error;
1007
1008 n_in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
1009 if (in_pos > n_in1)
1010 isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
1011 "index out of bounds", goto error);
1012
1013 n_in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
1014
1015 multi1 = FN(MULTI(BASE),insert_dims)(multi1, isl_dim_in, in_pos, n_in2);
1016 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, n_in2,
1017 n_in1 - in_pos);
1018 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, 0, in_pos);
1019
1020 return FN(MULTI(BASE),range_splice)(multi1, out_pos, multi2);
1021 error:
1022 FN(MULTI(BASE),free)(multi1);
1023 FN(MULTI(BASE),free)(multi2);
1024 return NULL;
1025 }
1026
1027 /* This function is currently only used from isl_aff.c
1028 */
1029 static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
1030 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
1031 __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
1032 __attribute__ ((unused));
1033
1034 /* Pairwise perform "fn" to the elements of "multi1" and "multi2" and
1035 * return the result.
1036 */
1037 static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
1038 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
1039 __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
1040 {
1041 int i;
1042 isl_ctx *ctx;
1043
1044 multi1 = FN(MULTI(BASE),cow)(multi1);
1045 if (!multi1 || !multi2)
1046 goto error;
1047
1048 ctx = FN(MULTI(BASE),get_ctx)(multi1);
1049 if (!isl_space_is_equal(multi1->space, multi2->space))
1050 isl_die(ctx, isl_error_invalid,
1051 "spaces don't match", goto error);
1052
1053 for (i = 0; i < multi1->n; ++i) {
1054 multi1->p[i] = fn(multi1->p[i], FN(EL,copy)(multi2->p[i]));
1055 if (!multi1->p[i])
1056 goto error;
1057 }
1058
1059 FN(MULTI(BASE),free)(multi2);
1060 return multi1;
1061 error:
1062 FN(MULTI(BASE),free)(multi1);
1063 FN(MULTI(BASE),free)(multi2);
1064 return NULL;
1065 }
1066
1067 /* Multiply the elements of "multi" by "v" and return the result.
1068 */
1069 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_val)(__isl_take MULTI(BASE) *multi,
1070 __isl_take isl_val *v)
1071 {
1072 int i;
1073
1074 if (!multi || !v)
1075 goto error;
1076
1077 if (isl_val_is_one(v)) {
1078 isl_val_free(v);
1079 return multi;
1080 }
1081
1082 if (!isl_val_is_rat(v))
1083 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1084 "expecting rational factor", goto error);
1085
1086 multi = FN(MULTI(BASE),cow)(multi);
1087 if (!multi)
1088 return NULL;
1089
1090 for (i = 0; i < multi->n; ++i) {
1091 multi->p[i] = FN(EL,scale_val)(multi->p[i], isl_val_copy(v));
1092 if (!multi->p[i])
1093 goto error;
1094 }
1095
1096 isl_val_free(v);
1097 return multi;
1098 error:
1099 isl_val_free(v);
1100 return FN(MULTI(BASE),free)(multi);
1101 }
1102
1103 /* Multiply the elements of "multi" by the corresponding element of "mv"
1104 * and return the result.
1105 */
1106 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_multi_val)(
1107 __isl_take MULTI(BASE) *multi, __isl_take isl_multi_val *mv)
1108 {
1109 int i;
1110
1111 if (!multi || !mv)
1112 goto error;
1113
1114 if (!isl_space_tuple_match(multi->space, isl_dim_out,
1115 mv->space, isl_dim_set))
1116 isl_die(isl_multi_val_get_ctx(mv), isl_error_invalid,
1117 "spaces don't match", goto error);
1118
1119 multi = FN(MULTI(BASE),cow)(multi);
1120 if (!multi)
1121 return NULL;
1122
1123 for (i = 0; i < multi->n; ++i) {
1124 isl_val *v;
1125
1126 v = isl_multi_val_get_val(mv, i);
1127 multi->p[i] = FN(EL,scale_val)(multi->p[i], v);
1128 if (!multi->p[i])
1129 goto error;
1130 }
1131
1132 isl_multi_val_free(mv);
1133 return multi;
1134 error:
1135 isl_multi_val_free(mv);
1136 return FN(MULTI(BASE),free)(multi);
1137 }
1138
1139 /* Divide the elements of "multi" by the corresponding element of "mv"
1140 * and return the result.
1141 */
1142 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_down_multi_val)(
1143 __isl_take MULTI(BASE) *multi, __isl_take isl_multi_val *mv)
1144 {
1145 int i;
1146
1147 if (!multi || !mv)
1148 goto error;
1149
1150 if (!isl_space_tuple_match(multi->space, isl_dim_out,
1151 mv->space, isl_dim_set))
1152 isl_die(isl_multi_val_get_ctx(mv), isl_error_invalid,
1153 "spaces don't match", goto error);
1154
1155 multi = FN(MULTI(BASE),cow)(multi);
1156 if (!multi)
1157 return NULL;
1158
1159 for (i = 0; i < multi->n; ++i) {
1160 isl_val *v;
1161
1162 v = isl_multi_val_get_val(mv, i);
1163 multi->p[i] = FN(EL,scale_down_val)(multi->p[i], v);
1164 if (!multi->p[i])
1165 goto error;
1166 }
1167
1168 isl_multi_val_free(mv);
1169 return multi;
1170 error:
1171 isl_multi_val_free(mv);
1172 return FN(MULTI(BASE),free)(multi);
1173 }
1174
1175 #ifndef NO_MOVE_DIMS
1176 /* Move the "n" dimensions of "src_type" starting at "src_pos" of "multi"
1177 * to dimensions of "dst_type" at "dst_pos".
1178 *
1179 * We only support moving input dimensions to parameters and vice versa.
1180 */
1181 __isl_give MULTI(BASE) *FN(MULTI(BASE),move_dims)(__isl_take MULTI(BASE) *multi,
1182 enum isl_dim_type dst_type, unsigned dst_pos,
1183 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
1184 {
1185 int i;
1186
1187 if (!multi)
1188 return NULL;
1189
1190 if (n == 0 &&
1191 !isl_space_is_named_or_nested(multi->space, src_type) &&
1192 !isl_space_is_named_or_nested(multi->space, dst_type))
1193 return multi;
1194
1195 if (dst_type == isl_dim_out || src_type == isl_dim_out)
1196 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1197 "cannot move output/set dimension",
1198 return FN(MULTI(BASE),free)(multi));
1199 if (dst_type == isl_dim_div || src_type == isl_dim_div)
1200 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1201 "cannot move divs",
1202 return FN(MULTI(BASE),free)(multi));
1203 if (src_pos + n > isl_space_dim(multi->space, src_type))
1204 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1205 "range out of bounds",
1206 return FN(MULTI(BASE),free)(multi));
1207 if (dst_type == src_type)
1208 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_unsupported,
1209 "moving dims within the same type not supported",
1210 return FN(MULTI(BASE),free)(multi));
1211
1212 multi = FN(MULTI(BASE),cow)(multi);
1213 if (!multi)
1214 return NULL;
1215
1216 multi->space = isl_space_move_dims(multi->space, dst_type, dst_pos,
1217 src_type, src_pos, n);
1218 if (!multi->space)
1219 return FN(MULTI(BASE),free)(multi);
1220
1221 for (i = 0; i < multi->n; ++i) {
1222 multi->p[i] = FN(EL,move_dims)(multi->p[i], dst_type, dst_pos,
1223 src_type, src_pos, n);
1224 if (!multi->p[i])
1225 return FN(MULTI(BASE),free)(multi);
1226 }
1227
1228 return multi;
1229 }
1230 #endif
1231
1232 /* Convert a multiple expression defined over a parameter domain
1233 * into one that is defined over a zero-dimensional set.
1234 */
1235 __isl_give MULTI(BASE) *FN(MULTI(BASE),from_range)(
1236 __isl_take MULTI(BASE) *multi)
1237 {
1238 isl_space *space;
1239
1240 if (!multi)
1241 return NULL;
1242 if (!isl_space_is_set(multi->space))
1243 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1244 "not living in a set space",
1245 return FN(MULTI(BASE),free)(multi));
1246
1247 space = FN(MULTI(BASE),get_space)(multi);
1248 space = isl_space_from_range(space);
1249 multi = FN(MULTI(BASE),reset_space)(multi, space);
1250
1251 return multi;
1252 }
1253
1254 /* Are "multi1" and "multi2" obviously equal?
1255 */
1256 int FN(MULTI(BASE),plain_is_equal)(__isl_keep MULTI(BASE) *multi1,
1257 __isl_keep MULTI(BASE) *multi2)
1258 {
1259 int i;
1260 int equal;
1261
1262 if (!multi1 || !multi2)
1263 return -1;
1264 if (multi1->n != multi2->n)
1265 return 0;
1266 equal = isl_space_is_equal(multi1->space, multi2->space);
1267 if (equal < 0 || !equal)
1268 return equal;
1269
1270 for (i = 0; i < multi1->n; ++i) {
1271 equal = FN(EL,plain_is_equal)(multi1->p[i], multi2->p[i]);
1272 if (equal < 0 || !equal)
1273 return equal;
1274 }
1275
1276 return 1;
1277 }
Integer set library