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