?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
add isl_aff_domain_factor_domain
[isl.git] / isl_multi_templ.c
blob2f0c1a8f4f3c5f4d269d4e6d097d3bfbb768bfc9
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/id.h>
12 #include <isl_space_private.h>
13 #include <isl_val_private.h>
14 #include <isl/set.h>
15 #include <isl_reordering.h>
16
17 #include <isl_multi_macro.h>
18
19 #define MULTI_NAME(BASE) "isl_multi_" #BASE
20 #define xLIST(EL) EL ## _list
21 #define LIST(EL) xLIST(EL)
22
23 isl_ctx *FN(MULTI(BASE),get_ctx)(__isl_keep MULTI(BASE) *multi)
24 {
25 return multi ? isl_space_get_ctx(multi->space) : NULL;
26 }
27
28 __isl_give isl_space *FN(MULTI(BASE),get_space)(__isl_keep MULTI(BASE) *multi)
29 {
30 return multi ? isl_space_copy(multi->space) : NULL;
31 }
32
33 /* Return the position of the dimension of the given type and name
34 * in "multi".
35 * Return -1 if no such dimension can be found.
36 */
37 int FN(MULTI(BASE),find_dim_by_name)(__isl_keep MULTI(BASE) *multi,
38 enum isl_dim_type type, const char *name)
39 {
40 if (!multi)
41 return -1;
42 return isl_space_find_dim_by_name(multi->space, type, name);
43 }
44
45 __isl_give isl_space *FN(MULTI(BASE),get_domain_space)(
46 __isl_keep MULTI(BASE) *multi)
47 {
48 return multi ? isl_space_domain(isl_space_copy(multi->space)) : NULL;
49 }
50
51 /* Allocate a multi expression living in "space".
52 *
53 * If the number of base expressions is zero, then make sure
54 * there is enough room in the structure for the explicit domain,
55 * in case the type supports such an explicit domain.
56 */
57 __isl_give MULTI(BASE) *FN(MULTI(BASE),alloc)(__isl_take isl_space *space)
58 {
59 isl_ctx *ctx;
60 int n;
61 MULTI(BASE) *multi;
62
63 if (!space)
64 return NULL;
65
66 ctx = isl_space_get_ctx(space);
67 n = isl_space_dim(space, isl_dim_out);
68 if (n > 0)
69 multi = isl_calloc(ctx, MULTI(BASE),
70 sizeof(MULTI(BASE)) + (n - 1) * sizeof(struct EL *));
71 else
72 multi = isl_calloc(ctx, MULTI(BASE), sizeof(MULTI(BASE)));
73 if (!multi)
74 goto error;
75
76 multi->space = space;
77 multi->n = n;
78 multi->ref = 1;
79 if (FN(MULTI(BASE),has_explicit_domain)(multi))
80 multi = FN(MULTI(BASE),init_explicit_domain)(multi);
81 return multi;
82 error:
83 isl_space_free(space);
84 return NULL;
85 }
86
87 __isl_give MULTI(BASE) *FN(MULTI(BASE),dup)(__isl_keep MULTI(BASE) *multi)
88 {
89 int i;
90 MULTI(BASE) *dup;
91
92 if (!multi)
93 return NULL;
94
95 dup = FN(MULTI(BASE),alloc)(isl_space_copy(multi->space));
96 if (!dup)
97 return NULL;
98
99 for (i = 0; i < multi->n; ++i)
100 dup = FN(FN(MULTI(BASE),set),BASE)(dup, i,
101 FN(EL,copy)(multi->u.p[i]));
102 if (FN(MULTI(BASE),has_explicit_domain)(multi))
103 dup = FN(MULTI(BASE),copy_explicit_domain)(dup, multi);
104
105 return dup;
106 }
107
108 __isl_give MULTI(BASE) *FN(MULTI(BASE),cow)(__isl_take MULTI(BASE) *multi)
109 {
110 if (!multi)
111 return NULL;
112
113 if (multi->ref == 1)
114 return multi;
115
116 multi->ref--;
117 return FN(MULTI(BASE),dup)(multi);
118 }
119
120 __isl_give MULTI(BASE) *FN(MULTI(BASE),copy)(__isl_keep MULTI(BASE) *multi)
121 {
122 if (!multi)
123 return NULL;
124
125 multi->ref++;
126 return multi;
127 }
128
129 __isl_null MULTI(BASE) *FN(MULTI(BASE),free)(__isl_take MULTI(BASE) *multi)
130 {
131 int i;
132
133 if (!multi)
134 return NULL;
135
136 if (--multi->ref > 0)
137 return NULL;
138
139 isl_space_free(multi->space);
140 for (i = 0; i < multi->n; ++i)
141 FN(EL,free)(multi->u.p[i]);
142 if (FN(MULTI(BASE),has_explicit_domain)(multi))
143 FN(MULTI(BASE),free_explicit_domain)(multi);
144 free(multi);
145
146 return NULL;
147 }
148
149 unsigned FN(MULTI(BASE),dim)(__isl_keep MULTI(BASE) *multi,
150 enum isl_dim_type type)
151 {
152 return multi ? isl_space_dim(multi->space, type) : 0;
153 }
154
155 /* Return the position of the first dimension of "type" with id "id".
156 * Return -1 if there is no such dimension.
157 */
158 int FN(MULTI(BASE),find_dim_by_id)(__isl_keep MULTI(BASE) *multi,
159 enum isl_dim_type type, __isl_keep isl_id *id)
160 {
161 if (!multi)
162 return -1;
163 return isl_space_find_dim_by_id(multi->space, type, id);
164 }
165
166 /* Return the id of the given dimension.
167 */
168 __isl_give isl_id *FN(MULTI(BASE),get_dim_id)(__isl_keep MULTI(BASE) *multi,
169 enum isl_dim_type type, unsigned pos)
170 {
171 return multi ? isl_space_get_dim_id(multi->space, type, pos) : NULL;
172 }
173
174 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_name)(
175 __isl_take MULTI(BASE) *multi,
176 enum isl_dim_type type, unsigned pos, const char *s)
177 {
178 int i;
179
180 multi = FN(MULTI(BASE),cow)(multi);
181 if (!multi)
182 return NULL;
183
184 multi->space = isl_space_set_dim_name(multi->space, type, pos, s);
185 if (!multi->space)
186 return FN(MULTI(BASE),free)(multi);
187
188 if (type == isl_dim_out)
189 return multi;
190 for (i = 0; i < multi->n; ++i) {
191 multi->u.p[i] = FN(EL,set_dim_name)(multi->u.p[i],
192 type, pos, s);
193 if (!multi->u.p[i])
194 return FN(MULTI(BASE),free)(multi);
195 }
196
197 return multi;
198 }
199
200 const char *FN(MULTI(BASE),get_tuple_name)(__isl_keep MULTI(BASE) *multi,
201 enum isl_dim_type type)
202 {
203 return multi ? isl_space_get_tuple_name(multi->space, type) : NULL;
204 }
205
206 /* Does the specified tuple have an id?
207 */
208 isl_bool FN(MULTI(BASE),has_tuple_id)(__isl_keep MULTI(BASE) *multi,
209 enum isl_dim_type type)
210 {
211 if (!multi)
212 return isl_bool_error;
213 return isl_space_has_tuple_id(multi->space, type);
214 }
215
216 /* Return the id of the specified tuple.
217 */
218 __isl_give isl_id *FN(MULTI(BASE),get_tuple_id)(__isl_keep MULTI(BASE) *multi,
219 enum isl_dim_type type)
220 {
221 return multi ? isl_space_get_tuple_id(multi->space, type) : NULL;
222 }
223
224 __isl_give EL *FN(FN(MULTI(BASE),get),BASE)(__isl_keep MULTI(BASE) *multi,
225 int pos)
226 {
227 isl_ctx *ctx;
228
229 if (!multi)
230 return NULL;
231 ctx = FN(MULTI(BASE),get_ctx)(multi);
232 if (pos < 0 || pos >= multi->n)
233 isl_die(ctx, isl_error_invalid,
234 "index out of bounds", return NULL);
235 return FN(EL,copy)(multi->u.p[pos]);
236 }
237
238 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),set),BASE)(
239 __isl_take MULTI(BASE) *multi, int pos, __isl_take EL *el)
240 {
241 isl_space *multi_space = NULL;
242 isl_space *el_space = NULL;
243 isl_bool match;
244
245 multi = FN(MULTI(BASE),cow)(multi);
246 if (!multi || !el)
247 goto error;
248
249 multi_space = FN(MULTI(BASE),get_space)(multi);
250 match = FN(EL,matching_params)(el, multi_space);
251 if (match < 0)
252 goto error;
253 if (!match) {
254 multi = FN(MULTI(BASE),align_params)(multi,
255 FN(EL,get_space)(el));
256 isl_space_free(multi_space);
257 multi_space = FN(MULTI(BASE),get_space)(multi);
258 el = FN(EL,align_params)(el, isl_space_copy(multi_space));
259 }
260 if (FN(EL,check_match_domain_space)(el, multi_space) < 0)
261 goto error;
262
263 if (pos < 0 || pos >= multi->n)
264 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
265 "index out of bounds", goto error);
266
267 FN(EL,free)(multi->u.p[pos]);
268 multi->u.p[pos] = el;
269
270 isl_space_free(multi_space);
271 isl_space_free(el_space);
272
273 return multi;
274 error:
275 FN(MULTI(BASE),free)(multi);
276 FN(EL,free)(el);
277 isl_space_free(multi_space);
278 isl_space_free(el_space);
279 return NULL;
280 }
281
282 /* Reset the space of "multi". This function is called from isl_pw_templ.c
283 * and doesn't know if the space of an element object is represented
284 * directly or through its domain. It therefore passes along both,
285 * which we pass along to the element function since we don't know how
286 * that is represented either.
287 *
288 * If "multi" has an explicit domain, then the caller is expected
289 * to make sure that any modification that would change the dimensions
290 * of the explicit domain has bee applied before this function is called.
291 */
292 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space_and_domain)(
293 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space,
294 __isl_take isl_space *domain)
295 {
296 int i;
297
298 multi = FN(MULTI(BASE),cow)(multi);
299 if (!multi || !space || !domain)
300 goto error;
301
302 for (i = 0; i < multi->n; ++i) {
303 multi->u.p[i] = FN(EL,reset_domain_space)(multi->u.p[i],
304 isl_space_copy(domain));
305 if (!multi->u.p[i])
306 goto error;
307 }
308 if (FN(MULTI(BASE),has_explicit_domain)(multi)) {
309 multi = FN(MULTI(BASE),reset_explicit_domain_space)(multi,
310 isl_space_copy(domain));
311 if (!multi)
312 goto error;
313 }
314 isl_space_free(domain);
315 isl_space_free(multi->space);
316 multi->space = space;
317
318 return multi;
319 error:
320 isl_space_free(domain);
321 isl_space_free(space);
322 FN(MULTI(BASE),free)(multi);
323 return NULL;
324 }
325
326 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_domain_space)(
327 __isl_take MULTI(BASE) *multi, __isl_take isl_space *domain)
328 {
329 isl_space *space;
330
331 space = isl_space_extend_domain_with_range(isl_space_copy(domain),
332 isl_space_copy(multi->space));
333 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
334 }
335
336 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_space)(
337 __isl_take MULTI(BASE) *multi, __isl_take isl_space *space)
338 {
339 isl_space *domain;
340
341 domain = isl_space_domain(isl_space_copy(space));
342 return FN(MULTI(BASE),reset_space_and_domain)(multi, space, domain);
343 }
344
345 /* Set the id of the given dimension of "multi" to "id".
346 */
347 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_dim_id)(
348 __isl_take MULTI(BASE) *multi,
349 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
350 {
351 isl_space *space;
352
353 multi = FN(MULTI(BASE),cow)(multi);
354 if (!multi || !id)
355 goto error;
356
357 space = FN(MULTI(BASE),get_space)(multi);
358 space = isl_space_set_dim_id(space, type, pos, id);
359
360 return FN(MULTI(BASE),reset_space)(multi, space);
361 error:
362 isl_id_free(id);
363 FN(MULTI(BASE),free)(multi);
364 return NULL;
365 }
366
367 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_tuple_name)(
368 __isl_keep MULTI(BASE) *multi, enum isl_dim_type type,
369 const char *s)
370 {
371 isl_space *space;
372
373 multi = FN(MULTI(BASE),cow)(multi);
374 if (!multi)
375 return NULL;
376
377 space = FN(MULTI(BASE),get_space)(multi);
378 space = isl_space_set_tuple_name(space, type, s);
379
380 return FN(MULTI(BASE),reset_space)(multi, space);
381 }
382
383 __isl_give MULTI(BASE) *FN(MULTI(BASE),set_tuple_id)(
384 __isl_take MULTI(BASE) *multi, enum isl_dim_type type,
385 __isl_take isl_id *id)
386 {
387 isl_space *space;
388
389 multi = FN(MULTI(BASE),cow)(multi);
390 if (!multi)
391 goto error;
392
393 space = FN(MULTI(BASE),get_space)(multi);
394 space = isl_space_set_tuple_id(space, type, id);
395
396 return FN(MULTI(BASE),reset_space)(multi, space);
397 error:
398 isl_id_free(id);
399 return NULL;
400 }
401
402 /* Drop the id on the specified tuple.
403 */
404 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_tuple_id)(
405 __isl_take MULTI(BASE) *multi, enum isl_dim_type type)
406 {
407 isl_space *space;
408
409 if (!multi)
410 return NULL;
411 if (!FN(MULTI(BASE),has_tuple_id)(multi, type))
412 return multi;
413
414 multi = FN(MULTI(BASE),cow)(multi);
415 if (!multi)
416 return NULL;
417
418 space = FN(MULTI(BASE),get_space)(multi);
419 space = isl_space_reset_tuple_id(space, type);
420
421 return FN(MULTI(BASE),reset_space)(multi, space);
422 }
423
424 /* Reset the user pointer on all identifiers of parameters and tuples
425 * of the space of "multi".
426 */
427 __isl_give MULTI(BASE) *FN(MULTI(BASE),reset_user)(
428 __isl_take MULTI(BASE) *multi)
429 {
430 isl_space *space;
431
432 space = FN(MULTI(BASE),get_space)(multi);
433 space = isl_space_reset_user(space);
434
435 return FN(MULTI(BASE),reset_space)(multi, space);
436 }
437
438 __isl_give MULTI(BASE) *FN(MULTI(BASE),realign_domain)(
439 __isl_take MULTI(BASE) *multi, __isl_take isl_reordering *exp)
440 {
441 int i;
442
443 multi = FN(MULTI(BASE),cow)(multi);
444 if (!multi || !exp)
445 goto error;
446
447 for (i = 0; i < multi->n; ++i) {
448 multi->u.p[i] = FN(EL,realign_domain)(multi->u.p[i],
449 isl_reordering_copy(exp));
450 if (!multi->u.p[i])
451 goto error;
452 }
453
454 multi = FN(MULTI(BASE),reset_domain_space)(multi,
455 isl_space_copy(exp->dim));
456
457 isl_reordering_free(exp);
458 return multi;
459 error:
460 isl_reordering_free(exp);
461 FN(MULTI(BASE),free)(multi);
462 return NULL;
463 }
464
465 /* Align the parameters of "multi" to those of "model".
466 *
467 * If "multi" has an explicit domain, then align the parameters
468 * of the domain first.
469 */
470 __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params)(
471 __isl_take MULTI(BASE) *multi, __isl_take isl_space *model)
472 {
473 isl_ctx *ctx;
474 isl_bool equal_params;
475 isl_reordering *exp;
476
477 if (!multi || !model)
478 goto error;
479
480 equal_params = isl_space_has_equal_params(multi->space, model);
481 if (equal_params < 0)
482 goto error;
483 if (equal_params) {
484 isl_space_free(model);
485 return multi;
486 }
487
488 ctx = isl_space_get_ctx(model);
489 if (!isl_space_has_named_params(model))
490 isl_die(ctx, isl_error_invalid,
491 "model has unnamed parameters", goto error);
492 if (!isl_space_has_named_params(multi->space))
493 isl_die(ctx, isl_error_invalid,
494 "input has unnamed parameters", goto error);
495
496 if (FN(MULTI(BASE),has_explicit_domain)(multi)) {
497 multi = FN(MULTI(BASE),align_explicit_domain_params)(multi,
498 isl_space_copy(model));
499 if (!multi)
500 goto error;
501 }
502 model = isl_space_params(model);
503 exp = isl_parameter_alignment_reordering(multi->space, model);
504 exp = isl_reordering_extend_space(exp,
505 FN(MULTI(BASE),get_domain_space)(multi));
506 multi = FN(MULTI(BASE),realign_domain)(multi, exp);
507
508 isl_space_free(model);
509 return multi;
510 error:
511 isl_space_free(model);
512 FN(MULTI(BASE),free)(multi);
513 return NULL;
514 }
515
516 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),LIST(BASE))(
517 __isl_take isl_space *space, __isl_take LIST(EL) *list)
518 {
519 int i;
520 int n;
521 isl_ctx *ctx;
522 MULTI(BASE) *multi;
523
524 if (!space || !list)
525 goto error;
526
527 ctx = isl_space_get_ctx(space);
528 n = FN(FN(LIST(EL),n),BASE)(list);
529 if (n != isl_space_dim(space, isl_dim_out))
530 isl_die(ctx, isl_error_invalid,
531 "invalid number of elements in list", goto error);
532
533 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
534 for (i = 0; i < n; ++i) {
535 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
536 FN(FN(LIST(EL),get),BASE)(list, i));
537 }
538
539 isl_space_free(space);
540 FN(LIST(EL),free)(list);
541 return multi;
542 error:
543 isl_space_free(space);
544 FN(LIST(EL),free)(list);
545 return NULL;
546 }
547
548 #ifndef NO_IDENTITY
549 /* Create a multi expression in the given space that maps each
550 * input dimension to the corresponding output dimension.
551 */
552 __isl_give MULTI(BASE) *FN(MULTI(BASE),identity)(__isl_take isl_space *space)
553 {
554 int i, n;
555 isl_local_space *ls;
556 MULTI(BASE) *multi;
557
558 if (!space)
559 return NULL;
560
561 if (isl_space_is_set(space))
562 isl_die(isl_space_get_ctx(space), isl_error_invalid,
563 "expecting map space", goto error);
564
565 n = isl_space_dim(space, isl_dim_out);
566 if (n != isl_space_dim(space, isl_dim_in))
567 isl_die(isl_space_get_ctx(space), isl_error_invalid,
568 "number of input and output dimensions needs to be "
569 "the same", goto error);
570
571 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
572
573 if (!n) {
574 isl_space_free(space);
575 return multi;
576 }
577
578 space = isl_space_domain(space);
579 ls = isl_local_space_from_space(space);
580
581 for (i = 0; i < n; ++i) {
582 EL *el;
583 el = FN(EL,var_on_domain)(isl_local_space_copy(ls),
584 isl_dim_set, i);
585 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i, el);
586 }
587
588 isl_local_space_free(ls);
589
590 return multi;
591 error:
592 isl_space_free(space);
593 return NULL;
594 }
595 #endif
596
597 #ifndef NO_ZERO
598 /* Construct a multi expression in the given space with value zero in
599 * each of the output dimensions.
600 */
601 __isl_give MULTI(BASE) *FN(MULTI(BASE),zero)(__isl_take isl_space *space)
602 {
603 int n;
604 MULTI(BASE) *multi;
605
606 if (!space)
607 return NULL;
608
609 n = isl_space_dim(space , isl_dim_out);
610 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));
611
612 if (!n)
613 isl_space_free(space);
614 else {
615 int i;
616 isl_local_space *ls;
617 EL *el;
618
619 space = isl_space_domain(space);
620 ls = isl_local_space_from_space(space);
621 el = FN(EL,zero_on_domain)(ls);
622
623 for (i = 0; i < n; ++i)
624 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i,
625 FN(EL,copy)(el));
626
627 FN(EL,free)(el);
628 }
629
630 return multi;
631 }
632 #endif
633
634 #ifndef NO_FROM_BASE
635 /* Create a multiple expression with a single output/set dimension
636 * equal to "el".
637 * For most multiple expression types, the base type has a single
638 * output/set dimension and the space of the result is therefore
639 * the same as the space of the input.
640 * In the case of isl_multi_union_pw_aff, however, the base type
641 * lives in a parameter space and we therefore need to add
642 * a single set dimension.
643 */
644 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),BASE)(__isl_take EL *el)
645 {
646 isl_space *space;
647 MULTI(BASE) *multi;
648
649 space = FN(EL,get_space(el));
650 if (isl_space_is_params(space)) {
651 space = isl_space_set_from_params(space);
652 space = isl_space_add_dims(space, isl_dim_set, 1);
653 }
654 multi = FN(MULTI(BASE),alloc)(space);
655 multi = FN(FN(MULTI(BASE),set),BASE)(multi, 0, el);
656
657 return multi;
658 }
659 #endif
660
661 __isl_give MULTI(BASE) *FN(MULTI(BASE),drop_dims)(
662 __isl_take MULTI(BASE) *multi,
663 enum isl_dim_type type, unsigned first, unsigned n)
664 {
665 int i;
666 unsigned dim;
667
668 multi = FN(MULTI(BASE),cow)(multi);
669 if (!multi)
670 return NULL;
671
672 dim = FN(MULTI(BASE),dim)(multi, type);
673 if (first + n > dim || first + n < first)
674 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
675 "index out of bounds",
676 return FN(MULTI(BASE),free)(multi));
677
678 multi->space = isl_space_drop_dims(multi->space, type, first, n);
679 if (!multi->space)
680 return FN(MULTI(BASE),free)(multi);
681
682 if (type == isl_dim_out) {
683 for (i = 0; i < n; ++i)
684 FN(EL,free)(multi->u.p[first + i]);
685 for (i = first; i + n < multi->n; ++i)
686 multi->u.p[i] = multi->u.p[i + n];
687 multi->n -= n;
688 if (n > 0 && FN(MULTI(BASE),has_explicit_domain)(multi))
689 multi = FN(MULTI(BASE),init_explicit_domain)(multi);
690
691 return multi;
692 }
693
694 if (FN(MULTI(BASE),has_explicit_domain)(multi))
695 multi = FN(MULTI(BASE),drop_explicit_domain_dims)(multi,
696 type, first, n);
697 if (!multi)
698 return NULL;
699
700 for (i = 0; i < multi->n; ++i) {
701 multi->u.p[i] = FN(EL,drop_dims)(multi->u.p[i], type, first, n);
702 if (!multi->u.p[i])
703 return FN(MULTI(BASE),free)(multi);
704 }
705
706 return multi;
707 }
708
709 /* Align the parameters of "multi1" and "multi2" (if needed) and call "fn".
710 */
711 static __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params_multi_multi_and)(
712 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
713 __isl_give MULTI(BASE) *(*fn)(__isl_take MULTI(BASE) *multi1,
714 __isl_take MULTI(BASE) *multi2))
715 {
716 isl_ctx *ctx;
717 isl_bool equal_params;
718
719 if (!multi1 || !multi2)
720 goto error;
721 equal_params = isl_space_has_equal_params(multi1->space, multi2->space);
722 if (equal_params < 0)
723 goto error;
724 if (equal_params)
725 return fn(multi1, multi2);
726 ctx = FN(MULTI(BASE),get_ctx)(multi1);
727 if (!isl_space_has_named_params(multi1->space) ||
728 !isl_space_has_named_params(multi2->space))
729 isl_die(ctx, isl_error_invalid,
730 "unaligned unnamed parameters", goto error);
731 multi1 = FN(MULTI(BASE),align_params)(multi1,
732 FN(MULTI(BASE),get_space)(multi2));
733 multi2 = FN(MULTI(BASE),align_params)(multi2,
734 FN(MULTI(BASE),get_space)(multi1));
735 return fn(multi1, multi2);
736 error:
737 FN(MULTI(BASE),free)(multi1);
738 FN(MULTI(BASE),free)(multi2);
739 return NULL;
740 }
741
742 /* Given two MULTI(BASE)s A -> B and C -> D,
743 * construct a MULTI(BASE) (A * C) -> [B -> D].
744 *
745 * The parameters are assumed to have been aligned.
746 *
747 * If "multi1" and/or "multi2" has an explicit domain, then
748 * intersect the domain of the result with these explicit domains.
749 */
750 static __isl_give MULTI(BASE) *FN(MULTI(BASE),range_product_aligned)(
751 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
752 {
753 int i, n1, n2;
754 EL *el;
755 isl_space *space;
756 MULTI(BASE) *res;
757
758 if (!multi1 || !multi2)
759 goto error;
760
761 space = isl_space_range_product(FN(MULTI(BASE),get_space)(multi1),
762 FN(MULTI(BASE),get_space)(multi2));
763 res = FN(MULTI(BASE),alloc)(space);
764
765 n1 = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
766 n2 = FN(MULTI(BASE),dim)(multi2, isl_dim_out);
767
768 for (i = 0; i < n1; ++i) {
769 el = FN(FN(MULTI(BASE),get),BASE)(multi1, i);
770 res = FN(FN(MULTI(BASE),set),BASE)(res, i, el);
771 }
772
773 for (i = 0; i < n2; ++i) {
774 el = FN(FN(MULTI(BASE),get),BASE)(multi2, i);
775 res = FN(FN(MULTI(BASE),set),BASE)(res, n1 + i, el);
776 }
777
778 if (FN(MULTI(BASE),has_explicit_domain)(multi1))
779 res = FN(MULTI(BASE),intersect_explicit_domain)(res, multi1);
780 if (FN(MULTI(BASE),has_explicit_domain)(multi2))
781 res = FN(MULTI(BASE),intersect_explicit_domain)(res, multi2);
782
783 FN(MULTI(BASE),free)(multi1);
784 FN(MULTI(BASE),free)(multi2);
785 return res;
786 error:
787 FN(MULTI(BASE),free)(multi1);
788 FN(MULTI(BASE),free)(multi2);
789 return NULL;
790 }
791
792 /* Given two MULTI(BASE)s A -> B and C -> D,
793 * construct a MULTI(BASE) (A * C) -> [B -> D].
794 */
795 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_product)(
796 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
797 {
798 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
799 &FN(MULTI(BASE),range_product_aligned));
800 }
801
802 /* Is the range of "multi" a wrapped relation?
803 */
804 isl_bool FN(MULTI(BASE),range_is_wrapping)(__isl_keep MULTI(BASE) *multi)
805 {
806 if (!multi)
807 return isl_bool_error;
808 return isl_space_range_is_wrapping(multi->space);
809 }
810
811 /* Given a function A -> [B -> C], extract the function A -> B.
812 */
813 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_factor_domain)(
814 __isl_take MULTI(BASE) *multi)
815 {
816 isl_space *space;
817 int total, keep;
818
819 if (!multi)
820 return NULL;
821 if (!isl_space_range_is_wrapping(multi->space))
822 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
823 "range is not a product",
824 return FN(MULTI(BASE),free)(multi));
825
826 space = FN(MULTI(BASE),get_space)(multi);
827 total = isl_space_dim(space, isl_dim_out);
828 space = isl_space_range_factor_domain(space);
829 keep = isl_space_dim(space, isl_dim_out);
830 multi = FN(MULTI(BASE),drop_dims)(multi,
831 isl_dim_out, keep, total - keep);
832 multi = FN(MULTI(BASE),reset_space)(multi, space);
833
834 return multi;
835 }
836
837 /* Given a function A -> [B -> C], extract the function A -> C.
838 */
839 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_factor_range)(
840 __isl_take MULTI(BASE) *multi)
841 {
842 isl_space *space;
843 int total, keep;
844
845 if (!multi)
846 return NULL;
847 if (!isl_space_range_is_wrapping(multi->space))
848 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
849 "range is not a product",
850 return FN(MULTI(BASE),free)(multi));
851
852 space = FN(MULTI(BASE),get_space)(multi);
853 total = isl_space_dim(space, isl_dim_out);
854 space = isl_space_range_factor_range(space);
855 keep = isl_space_dim(space, isl_dim_out);
856 multi = FN(MULTI(BASE),drop_dims)(multi, isl_dim_out, 0, total - keep);
857 multi = FN(MULTI(BASE),reset_space)(multi, space);
858
859 return multi;
860 }
861
862 /* Given a function [B -> C], extract the function C.
863 */
864 __isl_give MULTI(BASE) *FN(MULTI(BASE),factor_range)(
865 __isl_take MULTI(BASE) *multi)
866 {
867 isl_space *space;
868 int total, keep;
869
870 if (!multi)
871 return NULL;
872 if (!isl_space_is_wrapping(multi->space))
873 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
874 "not a product", return FN(MULTI(BASE),free)(multi));
875
876 space = FN(MULTI(BASE),get_space)(multi);
877 total = isl_space_dim(space, isl_dim_out);
878 space = isl_space_factor_range(space);
879 keep = isl_space_dim(space, isl_dim_out);
880 multi = FN(MULTI(BASE),drop_dims)(multi, isl_dim_out, 0, total - keep);
881 multi = FN(MULTI(BASE),reset_space)(multi, space);
882
883 return multi;
884 }
885
886 #ifndef NO_PRODUCT
887 /* Given two MULTI(BASE)s A -> B and C -> D,
888 * construct a MULTI(BASE) [A -> C] -> [B -> D].
889 *
890 * The parameters are assumed to have been aligned.
891 *
892 * If "multi1" and/or "multi2" has an explicit domain, then
893 * intersect the domain of the result with these explicit domains.
894 */
895 __isl_give MULTI(BASE) *FN(MULTI(BASE),product_aligned)(
896 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
897 {
898 int i;
899 EL *el;
900 isl_space *space;
901 MULTI(BASE) *res;
902 int in1, in2, out1, out2;
903
904 in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
905 in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
906 out1 = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
907 out2 = FN(MULTI(BASE),dim)(multi2, isl_dim_out);
908 space = isl_space_product(FN(MULTI(BASE),get_space)(multi1),
909 FN(MULTI(BASE),get_space)(multi2));
910 res = FN(MULTI(BASE),alloc)(isl_space_copy(space));
911 space = isl_space_domain(space);
912
913 for (i = 0; i < out1; ++i) {
914 el = FN(FN(MULTI(BASE),get),BASE)(multi1, i);
915 el = FN(EL,insert_dims)(el, isl_dim_in, in1, in2);
916 el = FN(EL,reset_domain_space)(el, isl_space_copy(space));
917 res = FN(FN(MULTI(BASE),set),BASE)(res, i, el);
918 }
919
920 for (i = 0; i < out2; ++i) {
921 el = FN(FN(MULTI(BASE),get),BASE)(multi2, i);
922 el = FN(EL,insert_dims)(el, isl_dim_in, 0, in1);
923 el = FN(EL,reset_domain_space)(el, isl_space_copy(space));
924 res = FN(FN(MULTI(BASE),set),BASE)(res, out1 + i, el);
925 }
926
927 if (FN(MULTI(BASE),has_explicit_domain)(multi1) ||
928 FN(MULTI(BASE),has_explicit_domain)(multi2))
929 res = FN(MULTI(BASE),intersect_explicit_domain_product)(res,
930 multi1, multi2);
931
932 isl_space_free(space);
933 FN(MULTI(BASE),free)(multi1);
934 FN(MULTI(BASE),free)(multi2);
935 return res;
936 }
937
938 /* Given two MULTI(BASE)s A -> B and C -> D,
939 * construct a MULTI(BASE) [A -> C] -> [B -> D].
940 */
941 __isl_give MULTI(BASE) *FN(MULTI(BASE),product)(
942 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
943 {
944 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
945 &FN(MULTI(BASE),product_aligned));
946 }
947 #endif
948
949 __isl_give MULTI(BASE) *FN(MULTI(BASE),flatten_range)(
950 __isl_take MULTI(BASE) *multi)
951 {
952 if (!multi)
953 return NULL;
954
955 if (!multi->space->nested[1])
956 return multi;
957
958 multi = FN(MULTI(BASE),cow)(multi);
959 if (!multi)
960 return NULL;
961
962 multi->space = isl_space_flatten_range(multi->space);
963 if (!multi->space)
964 return FN(MULTI(BASE),free)(multi);
965
966 return multi;
967 }
968
969 /* Given two MULTI(BASE)s A -> B and C -> D,
970 * construct a MULTI(BASE) (A * C) -> (B, D).
971 */
972 __isl_give MULTI(BASE) *FN(MULTI(BASE),flat_range_product)(
973 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
974 {
975 MULTI(BASE) *multi;
976
977 multi = FN(MULTI(BASE),range_product)(multi1, multi2);
978 multi = FN(MULTI(BASE),flatten_range)(multi);
979 return multi;
980 }
981
982 /* Given two multi expressions, "multi1"
983 *
984 * [A] -> [B1 B2]
985 *
986 * where B2 starts at position "pos", and "multi2"
987 *
988 * [A] -> [D]
989 *
990 * return the multi expression
991 *
992 * [A] -> [B1 D B2]
993 */
994 __isl_give MULTI(BASE) *FN(MULTI(BASE),range_splice)(
995 __isl_take MULTI(BASE) *multi1, unsigned pos,
996 __isl_take MULTI(BASE) *multi2)
997 {
998 MULTI(BASE) *res;
999 unsigned dim;
1000
1001 if (!multi1 || !multi2)
1002 goto error;
1003
1004 dim = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
1005 if (pos > dim)
1006 isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
1007 "index out of bounds", goto error);
1008
1009 res = FN(MULTI(BASE),copy)(multi1);
1010 res = FN(MULTI(BASE),drop_dims)(res, isl_dim_out, pos, dim - pos);
1011 multi1 = FN(MULTI(BASE),drop_dims)(multi1, isl_dim_out, 0, pos);
1012
1013 res = FN(MULTI(BASE),flat_range_product)(res, multi2);
1014 res = FN(MULTI(BASE),flat_range_product)(res, multi1);
1015
1016 return res;
1017 error:
1018 FN(MULTI(BASE),free)(multi1);
1019 FN(MULTI(BASE),free)(multi2);
1020 return NULL;
1021 }
1022
1023 #ifndef NO_SPLICE
1024 /* Given two multi expressions, "multi1"
1025 *
1026 * [A1 A2] -> [B1 B2]
1027 *
1028 * where A2 starts at position "in_pos" and B2 starts at position "out_pos",
1029 * and "multi2"
1030 *
1031 * [C] -> [D]
1032 *
1033 * return the multi expression
1034 *
1035 * [A1 C A2] -> [B1 D B2]
1036 *
1037 * We first insert input dimensions to obtain
1038 *
1039 * [A1 C A2] -> [B1 B2]
1040 *
1041 * and
1042 *
1043 * [A1 C A2] -> [D]
1044 *
1045 * and then apply range_splice.
1046 */
1047 __isl_give MULTI(BASE) *FN(MULTI(BASE),splice)(
1048 __isl_take MULTI(BASE) *multi1, unsigned in_pos, unsigned out_pos,
1049 __isl_take MULTI(BASE) *multi2)
1050 {
1051 unsigned n_in1;
1052 unsigned n_in2;
1053
1054 if (!multi1 || !multi2)
1055 goto error;
1056
1057 n_in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
1058 if (in_pos > n_in1)
1059 isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
1060 "index out of bounds", goto error);
1061
1062 n_in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
1063
1064 multi1 = FN(MULTI(BASE),insert_dims)(multi1, isl_dim_in, in_pos, n_in2);
1065 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, n_in2,
1066 n_in1 - in_pos);
1067 multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, 0, in_pos);
1068
1069 return FN(MULTI(BASE),range_splice)(multi1, out_pos, multi2);
1070 error:
1071 FN(MULTI(BASE),free)(multi1);
1072 FN(MULTI(BASE),free)(multi2);
1073 return NULL;
1074 }
1075 #endif
1076
1077 /* Check that "multi1" and "multi2" live in the same space,
1078 * reporting an error if they do not.
1079 */
1080 static isl_stat FN(MULTI(BASE),check_equal_space)(
1081 __isl_keep MULTI(BASE) *multi1, __isl_keep MULTI(BASE) *multi2)
1082 {
1083 isl_bool equal;
1084
1085 if (!multi1 || !multi2)
1086 return isl_stat_error;
1087
1088 equal = isl_space_is_equal(multi1->space, multi2->space);
1089 if (equal < 0)
1090 return isl_stat_error;
1091 if (!equal)
1092 isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
1093 "spaces don't match", return isl_stat_error);
1094
1095 return isl_stat_ok;
1096 }
1097
1098 /* This function is currently only used from isl_aff.c
1099 */
1100 static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
1101 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
1102 __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
1103 __attribute__ ((unused));
1104
1105 /* Pairwise perform "fn" to the elements of "multi1" and "multi2" and
1106 * return the result.
1107 *
1108 * If "multi2" has an explicit domain, then
1109 * intersect the domain of the result with this explicit domain.
1110 */
1111 static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
1112 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
1113 __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
1114 {
1115 int i;
1116
1117 multi1 = FN(MULTI(BASE),cow)(multi1);
1118 if (FN(MULTI(BASE),check_equal_space)(multi1, multi2) < 0)
1119 goto error;
1120
1121 for (i = 0; i < multi1->n; ++i) {
1122 multi1->u.p[i] = fn(multi1->u.p[i],
1123 FN(EL,copy)(multi2->u.p[i]));
1124 if (!multi1->u.p[i])
1125 goto error;
1126 }
1127
1128 if (FN(MULTI(BASE),has_explicit_domain)(multi2))
1129 multi1 = FN(MULTI(BASE),intersect_explicit_domain)(multi1,
1130 multi2);
1131
1132 FN(MULTI(BASE),free)(multi2);
1133 return multi1;
1134 error:
1135 FN(MULTI(BASE),free)(multi1);
1136 FN(MULTI(BASE),free)(multi2);
1137 return NULL;
1138 }
1139
1140 /* Add "multi2" from "multi1" and return the result.
1141 *
1142 * The parameters of "multi1" and "multi2" are assumed to have been aligned.
1143 */
1144 static __isl_give MULTI(BASE) *FN(MULTI(BASE),add_aligned)(
1145 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
1146 {
1147 return FN(MULTI(BASE),bin_op)(multi1, multi2, &FN(EL,add));
1148 }
1149
1150 /* Add "multi2" from "multi1" and return the result.
1151 */
1152 __isl_give MULTI(BASE) *FN(MULTI(BASE),add)(__isl_take MULTI(BASE) *multi1,
1153 __isl_take MULTI(BASE) *multi2)
1154 {
1155 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
1156 &FN(MULTI(BASE),add_aligned));
1157 }
1158
1159 /* Subtract "multi2" from "multi1" and return the result.
1160 *
1161 * The parameters of "multi1" and "multi2" are assumed to have been aligned.
1162 */
1163 static __isl_give MULTI(BASE) *FN(MULTI(BASE),sub_aligned)(
1164 __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2)
1165 {
1166 return FN(MULTI(BASE),bin_op)(multi1, multi2, &FN(EL,sub));
1167 }
1168
1169 /* Subtract "multi2" from "multi1" and return the result.
1170 */
1171 __isl_give MULTI(BASE) *FN(MULTI(BASE),sub)(__isl_take MULTI(BASE) *multi1,
1172 __isl_take MULTI(BASE) *multi2)
1173 {
1174 return FN(MULTI(BASE),align_params_multi_multi_and)(multi1, multi2,
1175 &FN(MULTI(BASE),sub_aligned));
1176 }
1177
1178 /* Multiply the elements of "multi" by "v" and return the result.
1179 */
1180 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_val)(__isl_take MULTI(BASE) *multi,
1181 __isl_take isl_val *v)
1182 {
1183 int i;
1184
1185 if (!multi || !v)
1186 goto error;
1187
1188 if (isl_val_is_one(v)) {
1189 isl_val_free(v);
1190 return multi;
1191 }
1192
1193 if (!isl_val_is_rat(v))
1194 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1195 "expecting rational factor", goto error);
1196
1197 multi = FN(MULTI(BASE),cow)(multi);
1198 if (!multi)
1199 return NULL;
1200
1201 for (i = 0; i < multi->n; ++i) {
1202 multi->u.p[i] = FN(EL,scale_val)(multi->u.p[i],
1203 isl_val_copy(v));
1204 if (!multi->u.p[i])
1205 goto error;
1206 }
1207
1208 isl_val_free(v);
1209 return multi;
1210 error:
1211 isl_val_free(v);
1212 return FN(MULTI(BASE),free)(multi);
1213 }
1214
1215 /* Divide the elements of "multi" by "v" and return the result.
1216 */
1217 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_down_val)(
1218 __isl_take MULTI(BASE) *multi, __isl_take isl_val *v)
1219 {
1220 int i;
1221
1222 if (!multi || !v)
1223 goto error;
1224
1225 if (isl_val_is_one(v)) {
1226 isl_val_free(v);
1227 return multi;
1228 }
1229
1230 if (!isl_val_is_rat(v))
1231 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1232 "expecting rational factor", goto error);
1233 if (isl_val_is_zero(v))
1234 isl_die(isl_val_get_ctx(v), isl_error_invalid,
1235 "cannot scale down by zero", goto error);
1236
1237 multi = FN(MULTI(BASE),cow)(multi);
1238 if (!multi)
1239 return NULL;
1240
1241 for (i = 0; i < multi->n; ++i) {
1242 multi->u.p[i] = FN(EL,scale_down_val)(multi->u.p[i],
1243 isl_val_copy(v));
1244 if (!multi->u.p[i])
1245 goto error;
1246 }
1247
1248 isl_val_free(v);
1249 return multi;
1250 error:
1251 isl_val_free(v);
1252 return FN(MULTI(BASE),free)(multi);
1253 }
1254
1255 /* Multiply the elements of "multi" by the corresponding element of "mv"
1256 * and return the result.
1257 */
1258 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_multi_val)(
1259 __isl_take MULTI(BASE) *multi, __isl_take isl_multi_val *mv)
1260 {
1261 int i;
1262
1263 if (!multi || !mv)
1264 goto error;
1265
1266 if (!isl_space_tuple_is_equal(multi->space, isl_dim_out,
1267 mv->space, isl_dim_set))
1268 isl_die(isl_multi_val_get_ctx(mv), isl_error_invalid,
1269 "spaces don't match", goto error);
1270
1271 multi = FN(MULTI(BASE),cow)(multi);
1272 if (!multi)
1273 goto error;
1274
1275 for (i = 0; i < multi->n; ++i) {
1276 isl_val *v;
1277
1278 v = isl_multi_val_get_val(mv, i);
1279 multi->u.p[i] = FN(EL,scale_val)(multi->u.p[i], v);
1280 if (!multi->u.p[i])
1281 goto error;
1282 }
1283
1284 isl_multi_val_free(mv);
1285 return multi;
1286 error:
1287 isl_multi_val_free(mv);
1288 return FN(MULTI(BASE),free)(multi);
1289 }
1290
1291 /* Divide the elements of "multi" by the corresponding element of "mv"
1292 * and return the result.
1293 */
1294 __isl_give MULTI(BASE) *FN(MULTI(BASE),scale_down_multi_val)(
1295 __isl_take MULTI(BASE) *multi, __isl_take isl_multi_val *mv)
1296 {
1297 int i;
1298
1299 if (!multi || !mv)
1300 goto error;
1301
1302 if (!isl_space_tuple_is_equal(multi->space, isl_dim_out,
1303 mv->space, isl_dim_set))
1304 isl_die(isl_multi_val_get_ctx(mv), isl_error_invalid,
1305 "spaces don't match", goto error);
1306
1307 multi = FN(MULTI(BASE),cow)(multi);
1308 if (!multi)
1309 return NULL;
1310
1311 for (i = 0; i < multi->n; ++i) {
1312 isl_val *v;
1313
1314 v = isl_multi_val_get_val(mv, i);
1315 multi->u.p[i] = FN(EL,scale_down_val)(multi->u.p[i], v);
1316 if (!multi->u.p[i])
1317 goto error;
1318 }
1319
1320 isl_multi_val_free(mv);
1321 return multi;
1322 error:
1323 isl_multi_val_free(mv);
1324 return FN(MULTI(BASE),free)(multi);
1325 }
1326
1327 /* Compute the residues of the elements of "multi" modulo
1328 * the corresponding element of "mv" and return the result.
1329 */
1330 __isl_give MULTI(BASE) *FN(MULTI(BASE),mod_multi_val)(
1331 __isl_take MULTI(BASE) *multi, __isl_take isl_multi_val *mv)
1332 {
1333 int i;
1334
1335 if (!multi || !mv)
1336 goto error;
1337
1338 if (!isl_space_tuple_is_equal(multi->space, isl_dim_out,
1339 mv->space, isl_dim_set))
1340 isl_die(isl_multi_val_get_ctx(mv), isl_error_invalid,
1341 "spaces don't match", goto error);
1342
1343 multi = FN(MULTI(BASE),cow)(multi);
1344 if (!multi)
1345 goto error;
1346
1347 for (i = 0; i < multi->n; ++i) {
1348 isl_val *v;
1349
1350 v = isl_multi_val_get_val(mv, i);
1351 multi->u.p[i] = FN(EL,mod_val)(multi->u.p[i], v);
1352 if (!multi->u.p[i])
1353 goto error;
1354 }
1355
1356 isl_multi_val_free(mv);
1357 return multi;
1358 error:
1359 isl_multi_val_free(mv);
1360 return FN(MULTI(BASE),free)(multi);
1361 }
1362
1363 #ifndef NO_MOVE_DIMS
1364 /* Move the "n" dimensions of "src_type" starting at "src_pos" of "multi"
1365 * to dimensions of "dst_type" at "dst_pos".
1366 *
1367 * We only support moving input dimensions to parameters and vice versa.
1368 */
1369 __isl_give MULTI(BASE) *FN(MULTI(BASE),move_dims)(__isl_take MULTI(BASE) *multi,
1370 enum isl_dim_type dst_type, unsigned dst_pos,
1371 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
1372 {
1373 int i;
1374
1375 if (!multi)
1376 return NULL;
1377
1378 if (n == 0 &&
1379 !isl_space_is_named_or_nested(multi->space, src_type) &&
1380 !isl_space_is_named_or_nested(multi->space, dst_type))
1381 return multi;
1382
1383 if (dst_type == isl_dim_out || src_type == isl_dim_out)
1384 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1385 "cannot move output/set dimension",
1386 return FN(MULTI(BASE),free)(multi));
1387 if (dst_type == isl_dim_div || src_type == isl_dim_div)
1388 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1389 "cannot move divs",
1390 return FN(MULTI(BASE),free)(multi));
1391 if (src_pos + n > isl_space_dim(multi->space, src_type))
1392 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1393 "range out of bounds",
1394 return FN(MULTI(BASE),free)(multi));
1395 if (dst_type == src_type)
1396 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_unsupported,
1397 "moving dims within the same type not supported",
1398 return FN(MULTI(BASE),free)(multi));
1399
1400 multi = FN(MULTI(BASE),cow)(multi);
1401 if (!multi)
1402 return NULL;
1403
1404 multi->space = isl_space_move_dims(multi->space, dst_type, dst_pos,
1405 src_type, src_pos, n);
1406 if (!multi->space)
1407 return FN(MULTI(BASE),free)(multi);
1408 if (FN(MULTI(BASE),has_explicit_domain)(multi))
1409 multi = FN(MULTI(BASE),move_explicit_domain_dims)(multi,
1410 dst_type, dst_pos, src_type, src_pos, n);
1411 if (!multi)
1412 return NULL;
1413
1414 for (i = 0; i < multi->n; ++i) {
1415 multi->u.p[i] = FN(EL,move_dims)(multi->u.p[i],
1416 dst_type, dst_pos,
1417 src_type, src_pos, n);
1418 if (!multi->u.p[i])
1419 return FN(MULTI(BASE),free)(multi);
1420 }
1421
1422 return multi;
1423 }
1424 #endif
1425
1426 /* Convert a multiple expression defined over a parameter domain
1427 * into one that is defined over a zero-dimensional set.
1428 */
1429 __isl_give MULTI(BASE) *FN(MULTI(BASE),from_range)(
1430 __isl_take MULTI(BASE) *multi)
1431 {
1432 isl_space *space;
1433
1434 if (!multi)
1435 return NULL;
1436 if (!isl_space_is_set(multi->space))
1437 isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
1438 "not living in a set space",
1439 return FN(MULTI(BASE),free)(multi));
1440
1441 space = FN(MULTI(BASE),get_space)(multi);
1442 space = isl_space_from_range(space);
1443 multi = FN(MULTI(BASE),reset_space)(multi, space);
1444
1445 return multi;
1446 }
1447
1448 /* Are "multi1" and "multi2" obviously equal?
1449 */
1450 isl_bool FN(MULTI(BASE),plain_is_equal)(__isl_keep MULTI(BASE) *multi1,
1451 __isl_keep MULTI(BASE) *multi2)
1452 {
1453 int i;
1454 isl_bool equal;
1455
1456 if (!multi1 || !multi2)
1457 return isl_bool_error;
1458 if (multi1->n != multi2->n)
1459 return isl_bool_false;
1460 equal = isl_space_is_equal(multi1->space, multi2->space);
1461 if (equal < 0 || !equal)
1462 return equal;
1463
1464 for (i = 0; i < multi1->n; ++i) {
1465 equal = FN(EL,plain_is_equal)(multi1->u.p[i], multi2->u.p[i]);
1466 if (equal < 0 || !equal)
1467 return equal;
1468 }
1469
1470 if (FN(MULTI(BASE),has_explicit_domain)(multi1) ||
1471 FN(MULTI(BASE),has_explicit_domain)(multi2)) {
1472 equal = FN(MULTI(BASE),equal_explicit_domain)(multi1, multi2);
1473 if (equal < 0 || !equal)
1474 return equal;
1475 }
1476
1477 return isl_bool_true;
1478 }
1479
1480 /* Does "multi" involve any NaNs?
1481 */
1482 isl_bool FN(MULTI(BASE),involves_nan)(__isl_keep MULTI(BASE) *multi)
1483 {
1484 int i;
1485
1486 if (!multi)
1487 return isl_bool_error;
1488 if (multi->n == 0)
1489 return isl_bool_false;
1490
1491 for (i = 0; i < multi->n; ++i) {
1492 isl_bool has_nan = FN(EL,involves_nan)(multi->u.p[i]);
1493 if (has_nan < 0 || has_nan)
1494 return has_nan;
1495 }
1496
1497 return isl_bool_false;
1498 }
1499
1500 #ifndef NO_DOMAIN
1501 /* Return the shared domain of the elements of "multi".
1502 *
1503 * If "multi" has an explicit domain, then return this domain.
1504 */
1505 __isl_give isl_set *FN(MULTI(BASE),domain)(__isl_take MULTI(BASE) *multi)
1506 {
1507 int i;
1508 isl_set *dom;
1509
1510 if (!multi)
1511 return NULL;
1512
1513 if (FN(MULTI(BASE),has_explicit_domain)(multi)) {
1514 dom = FN(MULTI(BASE),get_explicit_domain)(multi);
1515 FN(MULTI(BASE),free)(multi);
1516 return dom;
1517 }
1518
1519 dom = isl_set_universe(FN(MULTI(BASE),get_domain_space)(multi));
1520 for (i = 0; i < multi->n; ++i) {
1521 isl_set *dom_i;
1522
1523 dom_i = FN(EL,domain)(FN(FN(MULTI(BASE),get),BASE)(multi, i));
1524 dom = isl_set_intersect(dom, dom_i);
1525 }
1526
1527 FN(MULTI(BASE),free)(multi);
1528 return dom;
1529 }
1530 #endif
1531
1532 #ifndef NO_NEG
1533 /* Return the opposite of "multi".
1534 */
1535 __isl_give MULTI(BASE) *FN(MULTI(BASE),neg)(__isl_take MULTI(BASE) *multi)
1536 {
1537 int i;
1538
1539 multi = FN(MULTI(BASE),cow)(multi);
1540 if (!multi)
1541 return NULL;
1542
1543 for (i = 0; i < multi->n; ++i) {
1544 multi->u.p[i] = FN(EL,neg)(multi->u.p[i]);
1545 if (!multi->u.p[i])
1546 return FN(MULTI(BASE),free)(multi);
1547 }
1548
1549 return multi;
1550 }
1551 #endif
Integer set library