?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
isl_schedule_read.c: read_band: avoid double free on error path
[isl.git] / isl_space.c
bloba1cb8ef681715516752453f1179f6a23895fefaf
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
4 * Copyright 2013-2014 Ecole Normale Superieure
5 *
6 * Use of this software is governed by the MIT license
7 *
8 * Written by Sven Verdoolaege, K.U.Leuven, Departement
9 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
10 * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
11 * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
12 * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
13 */
14
15 #include <stdlib.h>
16 #include <string.h>
17 #include <isl_space_private.h>
18 #include <isl_id_private.h>
19 #include <isl_reordering.h>
20
21 isl_ctx *isl_space_get_ctx(__isl_keep isl_space *dim)
22 {
23 return dim ? dim->ctx : NULL;
24 }
25
26 __isl_give isl_space *isl_space_alloc(isl_ctx *ctx,
27 unsigned nparam, unsigned n_in, unsigned n_out)
28 {
29 isl_space *dim;
30
31 dim = isl_alloc_type(ctx, struct isl_space);
32 if (!dim)
33 return NULL;
34
35 dim->ctx = ctx;
36 isl_ctx_ref(ctx);
37 dim->ref = 1;
38 dim->nparam = nparam;
39 dim->n_in = n_in;
40 dim->n_out = n_out;
41
42 dim->tuple_id[0] = NULL;
43 dim->tuple_id[1] = NULL;
44
45 dim->nested[0] = NULL;
46 dim->nested[1] = NULL;
47
48 dim->n_id = 0;
49 dim->ids = NULL;
50
51 return dim;
52 }
53
54 /* Mark the space as being that of a set, by setting the domain tuple
55 * to isl_id_none.
56 */
57 static __isl_give isl_space *mark_as_set(__isl_take isl_space *space)
58 {
59 space = isl_space_cow(space);
60 if (!space)
61 return NULL;
62 space = isl_space_set_tuple_id(space, isl_dim_in, &isl_id_none);
63 return space;
64 }
65
66 /* Is the space that of a set?
67 */
68 isl_bool isl_space_is_set(__isl_keep isl_space *space)
69 {
70 if (!space)
71 return isl_bool_error;
72 if (space->n_in != 0 || space->nested[0])
73 return isl_bool_false;
74 if (space->tuple_id[0] != &isl_id_none)
75 return isl_bool_false;
76 return isl_bool_true;
77 }
78
79 /* Is the given space that of a map?
80 */
81 isl_bool isl_space_is_map(__isl_keep isl_space *space)
82 {
83 if (!space)
84 return isl_bool_error;
85 return space->tuple_id[0] != &isl_id_none &&
86 space->tuple_id[1] != &isl_id_none;
87 }
88
89 __isl_give isl_space *isl_space_set_alloc(isl_ctx *ctx,
90 unsigned nparam, unsigned dim)
91 {
92 isl_space *space;
93 space = isl_space_alloc(ctx, nparam, 0, dim);
94 space = mark_as_set(space);
95 return space;
96 }
97
98 /* Mark the space as being that of a parameter domain, by setting
99 * both tuples to isl_id_none.
100 */
101 static __isl_give isl_space *mark_as_params(isl_space *space)
102 {
103 if (!space)
104 return NULL;
105 space = isl_space_set_tuple_id(space, isl_dim_in, &isl_id_none);
106 space = isl_space_set_tuple_id(space, isl_dim_out, &isl_id_none);
107 return space;
108 }
109
110 /* Is the space that of a parameter domain?
111 */
112 isl_bool isl_space_is_params(__isl_keep isl_space *space)
113 {
114 if (!space)
115 return isl_bool_error;
116 if (space->n_in != 0 || space->nested[0] ||
117 space->n_out != 0 || space->nested[1])
118 return isl_bool_false;
119 if (space->tuple_id[0] != &isl_id_none)
120 return isl_bool_false;
121 if (space->tuple_id[1] != &isl_id_none)
122 return isl_bool_false;
123 return isl_bool_true;
124 }
125
126 /* Create a space for a parameter domain.
127 */
128 __isl_give isl_space *isl_space_params_alloc(isl_ctx *ctx, unsigned nparam)
129 {
130 isl_space *space;
131 space = isl_space_alloc(ctx, nparam, 0, 0);
132 space = mark_as_params(space);
133 return space;
134 }
135
136 static unsigned global_pos(__isl_keep isl_space *dim,
137 enum isl_dim_type type, unsigned pos)
138 {
139 struct isl_ctx *ctx = dim->ctx;
140
141 switch (type) {
142 case isl_dim_param:
143 isl_assert(ctx, pos < dim->nparam,
144 return isl_space_dim(dim, isl_dim_all));
145 return pos;
146 case isl_dim_in:
147 isl_assert(ctx, pos < dim->n_in,
148 return isl_space_dim(dim, isl_dim_all));
149 return pos + dim->nparam;
150 case isl_dim_out:
151 isl_assert(ctx, pos < dim->n_out,
152 return isl_space_dim(dim, isl_dim_all));
153 return pos + dim->nparam + dim->n_in;
154 default:
155 isl_assert(ctx, 0, return isl_space_dim(dim, isl_dim_all));
156 }
157 return isl_space_dim(dim, isl_dim_all);
158 }
159
160 /* Extend length of ids array to the total number of dimensions.
161 */
162 static __isl_give isl_space *extend_ids(__isl_take isl_space *dim)
163 {
164 isl_id **ids;
165 int i;
166
167 if (isl_space_dim(dim, isl_dim_all) <= dim->n_id)
168 return dim;
169
170 if (!dim->ids) {
171 dim->ids = isl_calloc_array(dim->ctx,
172 isl_id *, isl_space_dim(dim, isl_dim_all));
173 if (!dim->ids)
174 goto error;
175 } else {
176 ids = isl_realloc_array(dim->ctx, dim->ids,
177 isl_id *, isl_space_dim(dim, isl_dim_all));
178 if (!ids)
179 goto error;
180 dim->ids = ids;
181 for (i = dim->n_id; i < isl_space_dim(dim, isl_dim_all); ++i)
182 dim->ids[i] = NULL;
183 }
184
185 dim->n_id = isl_space_dim(dim, isl_dim_all);
186
187 return dim;
188 error:
189 isl_space_free(dim);
190 return NULL;
191 }
192
193 static __isl_give isl_space *set_id(__isl_take isl_space *dim,
194 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
195 {
196 dim = isl_space_cow(dim);
197
198 if (!dim)
199 goto error;
200
201 pos = global_pos(dim, type, pos);
202 if (pos == isl_space_dim(dim, isl_dim_all))
203 goto error;
204
205 if (pos >= dim->n_id) {
206 if (!id)
207 return dim;
208 dim = extend_ids(dim);
209 if (!dim)
210 goto error;
211 }
212
213 dim->ids[pos] = id;
214
215 return dim;
216 error:
217 isl_id_free(id);
218 isl_space_free(dim);
219 return NULL;
220 }
221
222 static __isl_keep isl_id *get_id(__isl_keep isl_space *dim,
223 enum isl_dim_type type, unsigned pos)
224 {
225 if (!dim)
226 return NULL;
227
228 pos = global_pos(dim, type, pos);
229 if (pos == isl_space_dim(dim, isl_dim_all))
230 return NULL;
231 if (pos >= dim->n_id)
232 return NULL;
233 return dim->ids[pos];
234 }
235
236 static unsigned offset(__isl_keep isl_space *dim, enum isl_dim_type type)
237 {
238 switch (type) {
239 case isl_dim_param: return 0;
240 case isl_dim_in: return dim->nparam;
241 case isl_dim_out: return dim->nparam + dim->n_in;
242 default: return 0;
243 }
244 }
245
246 static unsigned n(__isl_keep isl_space *dim, enum isl_dim_type type)
247 {
248 switch (type) {
249 case isl_dim_param: return dim->nparam;
250 case isl_dim_in: return dim->n_in;
251 case isl_dim_out: return dim->n_out;
252 case isl_dim_all: return dim->nparam + dim->n_in + dim->n_out;
253 default: return 0;
254 }
255 }
256
257 unsigned isl_space_dim(__isl_keep isl_space *dim, enum isl_dim_type type)
258 {
259 if (!dim)
260 return 0;
261 return n(dim, type);
262 }
263
264 unsigned isl_space_offset(__isl_keep isl_space *dim, enum isl_dim_type type)
265 {
266 if (!dim)
267 return 0;
268 return offset(dim, type);
269 }
270
271 static __isl_give isl_space *copy_ids(__isl_take isl_space *dst,
272 enum isl_dim_type dst_type, unsigned offset, __isl_keep isl_space *src,
273 enum isl_dim_type src_type)
274 {
275 int i;
276 isl_id *id;
277
278 if (!dst)
279 return NULL;
280
281 for (i = 0; i < n(src, src_type); ++i) {
282 id = get_id(src, src_type, i);
283 if (!id)
284 continue;
285 dst = set_id(dst, dst_type, offset + i, isl_id_copy(id));
286 if (!dst)
287 return NULL;
288 }
289 return dst;
290 }
291
292 __isl_take isl_space *isl_space_dup(__isl_keep isl_space *dim)
293 {
294 isl_space *dup;
295 if (!dim)
296 return NULL;
297 dup = isl_space_alloc(dim->ctx, dim->nparam, dim->n_in, dim->n_out);
298 if (!dup)
299 return NULL;
300 if (dim->tuple_id[0] &&
301 !(dup->tuple_id[0] = isl_id_copy(dim->tuple_id[0])))
302 goto error;
303 if (dim->tuple_id[1] &&
304 !(dup->tuple_id[1] = isl_id_copy(dim->tuple_id[1])))
305 goto error;
306 if (dim->nested[0] && !(dup->nested[0] = isl_space_copy(dim->nested[0])))
307 goto error;
308 if (dim->nested[1] && !(dup->nested[1] = isl_space_copy(dim->nested[1])))
309 goto error;
310 if (!dim->ids)
311 return dup;
312 dup = copy_ids(dup, isl_dim_param, 0, dim, isl_dim_param);
313 dup = copy_ids(dup, isl_dim_in, 0, dim, isl_dim_in);
314 dup = copy_ids(dup, isl_dim_out, 0, dim, isl_dim_out);
315 return dup;
316 error:
317 isl_space_free(dup);
318 return NULL;
319 }
320
321 __isl_give isl_space *isl_space_cow(__isl_take isl_space *dim)
322 {
323 if (!dim)
324 return NULL;
325
326 if (dim->ref == 1)
327 return dim;
328 dim->ref--;
329 return isl_space_dup(dim);
330 }
331
332 __isl_give isl_space *isl_space_copy(__isl_keep isl_space *dim)
333 {
334 if (!dim)
335 return NULL;
336
337 dim->ref++;
338 return dim;
339 }
340
341 __isl_null isl_space *isl_space_free(__isl_take isl_space *space)
342 {
343 int i;
344
345 if (!space)
346 return NULL;
347
348 if (--space->ref > 0)
349 return NULL;
350
351 isl_id_free(space->tuple_id[0]);
352 isl_id_free(space->tuple_id[1]);
353
354 isl_space_free(space->nested[0]);
355 isl_space_free(space->nested[1]);
356
357 for (i = 0; i < space->n_id; ++i)
358 isl_id_free(space->ids[i]);
359 free(space->ids);
360 isl_ctx_deref(space->ctx);
361
362 free(space);
363
364 return NULL;
365 }
366
367 /* Check if "s" is a valid dimension or tuple name.
368 * We currently only forbid names that look like a number.
369 *
370 * s is assumed to be non-NULL.
371 */
372 static int name_ok(isl_ctx *ctx, const char *s)
373 {
374 char *p;
375 long dummy;
376
377 dummy = strtol(s, &p, 0);
378 if (p != s)
379 isl_die(ctx, isl_error_invalid, "name looks like a number",
380 return 0);
381
382 return 1;
383 }
384
385 /* Is it possible for the given dimension type to have a tuple id?
386 */
387 static int space_can_have_id(__isl_keep isl_space *space,
388 enum isl_dim_type type)
389 {
390 if (!space)
391 return 0;
392 if (isl_space_is_params(space))
393 isl_die(space->ctx, isl_error_invalid,
394 "parameter spaces don't have tuple ids", return 0);
395 if (isl_space_is_set(space) && type != isl_dim_set)
396 isl_die(space->ctx, isl_error_invalid,
397 "set spaces can only have a set id", return 0);
398 if (type != isl_dim_in && type != isl_dim_out)
399 isl_die(space->ctx, isl_error_invalid,
400 "only input, output and set tuples can have ids",
401 return 0);
402
403 return 1;
404 }
405
406 /* Does the tuple have an id?
407 */
408 isl_bool isl_space_has_tuple_id(__isl_keep isl_space *dim,
409 enum isl_dim_type type)
410 {
411 if (!space_can_have_id(dim, type))
412 return isl_bool_error;
413 return dim->tuple_id[type - isl_dim_in] != NULL;
414 }
415
416 __isl_give isl_id *isl_space_get_tuple_id(__isl_keep isl_space *dim,
417 enum isl_dim_type type)
418 {
419 int has_id;
420
421 if (!dim)
422 return NULL;
423 has_id = isl_space_has_tuple_id(dim, type);
424 if (has_id < 0)
425 return NULL;
426 if (!has_id)
427 isl_die(dim->ctx, isl_error_invalid,
428 "tuple has no id", return NULL);
429 return isl_id_copy(dim->tuple_id[type - isl_dim_in]);
430 }
431
432 __isl_give isl_space *isl_space_set_tuple_id(__isl_take isl_space *dim,
433 enum isl_dim_type type, __isl_take isl_id *id)
434 {
435 dim = isl_space_cow(dim);
436 if (!dim || !id)
437 goto error;
438 if (type != isl_dim_in && type != isl_dim_out)
439 isl_die(dim->ctx, isl_error_invalid,
440 "only input, output and set tuples can have names",
441 goto error);
442
443 isl_id_free(dim->tuple_id[type - isl_dim_in]);
444 dim->tuple_id[type - isl_dim_in] = id;
445
446 return dim;
447 error:
448 isl_id_free(id);
449 isl_space_free(dim);
450 return NULL;
451 }
452
453 __isl_give isl_space *isl_space_reset_tuple_id(__isl_take isl_space *dim,
454 enum isl_dim_type type)
455 {
456 dim = isl_space_cow(dim);
457 if (!dim)
458 return NULL;
459 if (type != isl_dim_in && type != isl_dim_out)
460 isl_die(dim->ctx, isl_error_invalid,
461 "only input, output and set tuples can have names",
462 goto error);
463
464 isl_id_free(dim->tuple_id[type - isl_dim_in]);
465 dim->tuple_id[type - isl_dim_in] = NULL;
466
467 return dim;
468 error:
469 isl_space_free(dim);
470 return NULL;
471 }
472
473 /* Set the id of the given dimension of "space" to "id".
474 * If the dimension already has an id, then it is replaced.
475 * If the dimension is a parameter, then we need to change it
476 * in the nested spaces (if any) as well.
477 */
478 __isl_give isl_space *isl_space_set_dim_id(__isl_take isl_space *space,
479 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
480 {
481 space = isl_space_cow(space);
482 if (!space || !id)
483 goto error;
484
485 if (type == isl_dim_param) {
486 int i;
487
488 for (i = 0; i < 2; ++i) {
489 if (!space->nested[i])
490 continue;
491 space->nested[i] =
492 isl_space_set_dim_id(space->nested[i],
493 type, pos, isl_id_copy(id));
494 if (!space->nested[i])
495 goto error;
496 }
497 }
498
499 isl_id_free(get_id(space, type, pos));
500 return set_id(space, type, pos, id);
501 error:
502 isl_id_free(id);
503 isl_space_free(space);
504 return NULL;
505 }
506
507 /* Reset the id of the given dimension of "space".
508 * If the dimension already has an id, then it is removed.
509 * If the dimension is a parameter, then we need to reset it
510 * in the nested spaces (if any) as well.
511 */
512 __isl_give isl_space *isl_space_reset_dim_id(__isl_take isl_space *space,
513 enum isl_dim_type type, unsigned pos)
514 {
515 space = isl_space_cow(space);
516 if (!space)
517 goto error;
518
519 if (type == isl_dim_param) {
520 int i;
521
522 for (i = 0; i < 2; ++i) {
523 if (!space->nested[i])
524 continue;
525 space->nested[i] =
526 isl_space_reset_dim_id(space->nested[i],
527 type, pos);
528 if (!space->nested[i])
529 goto error;
530 }
531 }
532
533 isl_id_free(get_id(space, type, pos));
534 return set_id(space, type, pos, NULL);
535 error:
536 isl_space_free(space);
537 return NULL;
538 }
539
540 isl_bool isl_space_has_dim_id(__isl_keep isl_space *dim,
541 enum isl_dim_type type, unsigned pos)
542 {
543 if (!dim)
544 return isl_bool_error;
545 return get_id(dim, type, pos) != NULL;
546 }
547
548 __isl_give isl_id *isl_space_get_dim_id(__isl_keep isl_space *dim,
549 enum isl_dim_type type, unsigned pos)
550 {
551 if (!dim)
552 return NULL;
553 if (!get_id(dim, type, pos))
554 isl_die(dim->ctx, isl_error_invalid,
555 "dim has no id", return NULL);
556 return isl_id_copy(get_id(dim, type, pos));
557 }
558
559 __isl_give isl_space *isl_space_set_tuple_name(__isl_take isl_space *dim,
560 enum isl_dim_type type, const char *s)
561 {
562 isl_id *id;
563
564 if (!dim)
565 return NULL;
566
567 if (!s)
568 return isl_space_reset_tuple_id(dim, type);
569
570 if (!name_ok(dim->ctx, s))
571 goto error;
572
573 id = isl_id_alloc(dim->ctx, s, NULL);
574 return isl_space_set_tuple_id(dim, type, id);
575 error:
576 isl_space_free(dim);
577 return NULL;
578 }
579
580 /* Does the tuple have a name?
581 */
582 isl_bool isl_space_has_tuple_name(__isl_keep isl_space *space,
583 enum isl_dim_type type)
584 {
585 isl_id *id;
586
587 if (!space_can_have_id(space, type))
588 return isl_bool_error;
589 id = space->tuple_id[type - isl_dim_in];
590 return id && id->name;
591 }
592
593 const char *isl_space_get_tuple_name(__isl_keep isl_space *dim,
594 enum isl_dim_type type)
595 {
596 isl_id *id;
597 if (!dim)
598 return NULL;
599 if (type != isl_dim_in && type != isl_dim_out)
600 return NULL;
601 id = dim->tuple_id[type - isl_dim_in];
602 return id ? id->name : NULL;
603 }
604
605 __isl_give isl_space *isl_space_set_dim_name(__isl_take isl_space *dim,
606 enum isl_dim_type type, unsigned pos,
607 const char *s)
608 {
609 isl_id *id;
610
611 if (!dim)
612 return NULL;
613 if (!s)
614 return isl_space_reset_dim_id(dim, type, pos);
615 if (!name_ok(dim->ctx, s))
616 goto error;
617 id = isl_id_alloc(dim->ctx, s, NULL);
618 return isl_space_set_dim_id(dim, type, pos, id);
619 error:
620 isl_space_free(dim);
621 return NULL;
622 }
623
624 /* Does the given dimension have a name?
625 */
626 isl_bool isl_space_has_dim_name(__isl_keep isl_space *space,
627 enum isl_dim_type type, unsigned pos)
628 {
629 isl_id *id;
630
631 if (!space)
632 return isl_bool_error;
633 id = get_id(space, type, pos);
634 return id && id->name;
635 }
636
637 __isl_keep const char *isl_space_get_dim_name(__isl_keep isl_space *dim,
638 enum isl_dim_type type, unsigned pos)
639 {
640 isl_id *id = get_id(dim, type, pos);
641 return id ? id->name : NULL;
642 }
643
644 int isl_space_find_dim_by_id(__isl_keep isl_space *dim, enum isl_dim_type type,
645 __isl_keep isl_id *id)
646 {
647 int i;
648 int offset;
649 int n;
650
651 if (!dim || !id)
652 return -1;
653
654 offset = isl_space_offset(dim, type);
655 n = isl_space_dim(dim, type);
656 for (i = 0; i < n && offset + i < dim->n_id; ++i)
657 if (dim->ids[offset + i] == id)
658 return i;
659
660 return -1;
661 }
662
663 int isl_space_find_dim_by_name(__isl_keep isl_space *space,
664 enum isl_dim_type type, const char *name)
665 {
666 int i;
667 int offset;
668 int n;
669
670 if (!space || !name)
671 return -1;
672
673 offset = isl_space_offset(space, type);
674 n = isl_space_dim(space, type);
675 for (i = 0; i < n && offset + i < space->n_id; ++i) {
676 isl_id *id = get_id(space, type, i);
677 if (id && id->name && !strcmp(id->name, name))
678 return i;
679 }
680
681 return -1;
682 }
683
684 /* Reset the user pointer on all identifiers of parameters and tuples
685 * of "space".
686 */
687 __isl_give isl_space *isl_space_reset_user(__isl_take isl_space *space)
688 {
689 int i;
690 isl_ctx *ctx;
691 isl_id *id;
692 const char *name;
693
694 if (!space)
695 return NULL;
696
697 ctx = isl_space_get_ctx(space);
698
699 for (i = 0; i < space->nparam && i < space->n_id; ++i) {
700 if (!isl_id_get_user(space->ids[i]))
701 continue;
702 space = isl_space_cow(space);
703 if (!space)
704 return NULL;
705 name = isl_id_get_name(space->ids[i]);
706 id = isl_id_alloc(ctx, name, NULL);
707 isl_id_free(space->ids[i]);
708 space->ids[i] = id;
709 if (!id)
710 return isl_space_free(space);
711 }
712
713 for (i = 0; i < 2; ++i) {
714 if (!space->tuple_id[i])
715 continue;
716 if (!isl_id_get_user(space->tuple_id[i]))
717 continue;
718 space = isl_space_cow(space);
719 if (!space)
720 return NULL;
721 name = isl_id_get_name(space->tuple_id[i]);
722 id = isl_id_alloc(ctx, name, NULL);
723 isl_id_free(space->tuple_id[i]);
724 space->tuple_id[i] = id;
725 if (!id)
726 return isl_space_free(space);
727 }
728
729 for (i = 0; i < 2; ++i) {
730 if (!space->nested[i])
731 continue;
732 space = isl_space_cow(space);
733 if (!space)
734 return NULL;
735 space->nested[i] = isl_space_reset_user(space->nested[i]);
736 if (!space->nested[i])
737 return isl_space_free(space);
738 }
739
740 return space;
741 }
742
743 static __isl_keep isl_id *tuple_id(__isl_keep isl_space *dim,
744 enum isl_dim_type type)
745 {
746 if (!dim)
747 return NULL;
748 if (type == isl_dim_in)
749 return dim->tuple_id[0];
750 if (type == isl_dim_out)
751 return dim->tuple_id[1];
752 return NULL;
753 }
754
755 static __isl_keep isl_space *nested(__isl_keep isl_space *dim,
756 enum isl_dim_type type)
757 {
758 if (!dim)
759 return NULL;
760 if (type == isl_dim_in)
761 return dim->nested[0];
762 if (type == isl_dim_out)
763 return dim->nested[1];
764 return NULL;
765 }
766
767 /* Are the two spaces the same, apart from positions and names of parameters?
768 */
769 isl_bool isl_space_has_equal_tuples(__isl_keep isl_space *space1,
770 __isl_keep isl_space *space2)
771 {
772 if (!space1 || !space2)
773 return isl_bool_error;
774 if (space1 == space2)
775 return isl_bool_true;
776 return isl_space_tuple_is_equal(space1, isl_dim_in,
777 space2, isl_dim_in) &&
778 isl_space_tuple_is_equal(space1, isl_dim_out,
779 space2, isl_dim_out);
780 }
781
782 /* Check if the tuple of type "type1" of "space1" is the same as
783 * the tuple of type "type2" of "space2".
784 *
785 * That is, check if the tuples have the same identifier, the same dimension
786 * and the same internal structure.
787 * The identifiers of the dimensions inside the tuples do not affect the result.
788 *
789 * Note that this function only checks the tuples themselves.
790 * If nested tuples are involved, then we need to be careful not
791 * to have result affected by possibly differing parameters
792 * in those nested tuples.
793 */
794 isl_bool isl_space_tuple_is_equal(__isl_keep isl_space *space1,
795 enum isl_dim_type type1, __isl_keep isl_space *space2,
796 enum isl_dim_type type2)
797 {
798 isl_id *id1, *id2;
799 isl_space *nested1, *nested2;
800
801 if (!space1 || !space2)
802 return isl_bool_error;
803
804 if (space1 == space2 && type1 == type2)
805 return isl_bool_true;
806
807 if (n(space1, type1) != n(space2, type2))
808 return isl_bool_false;
809 id1 = tuple_id(space1, type1);
810 id2 = tuple_id(space2, type2);
811 if (!id1 ^ !id2)
812 return isl_bool_false;
813 if (id1 && id1 != id2)
814 return isl_bool_false;
815 nested1 = nested(space1, type1);
816 nested2 = nested(space2, type2);
817 if (!nested1 ^ !nested2)
818 return isl_bool_false;
819 if (nested1 && !isl_space_has_equal_tuples(nested1, nested2))
820 return isl_bool_false;
821 return isl_bool_true;
822 }
823
824 /* This is the old, undocumented, name for isl_space_tuple_is_equal.
825 * It will be removed at some point.
826 */
827 int isl_space_tuple_match(__isl_keep isl_space *space1, enum isl_dim_type type1,
828 __isl_keep isl_space *space2, enum isl_dim_type type2)
829 {
830 return isl_space_tuple_is_equal(space1, type1, space2, type2);
831 }
832
833 static int match(__isl_keep isl_space *space1, enum isl_dim_type type1,
834 __isl_keep isl_space *space2, enum isl_dim_type type2)
835 {
836 int i;
837
838 if (space1 == space2 && type1 == type2)
839 return 1;
840
841 if (!isl_space_tuple_is_equal(space1, type1, space2, type2))
842 return 0;
843
844 if (!space1->ids && !space2->ids)
845 return 1;
846
847 for (i = 0; i < n(space1, type1); ++i) {
848 if (get_id(space1, type1, i) != get_id(space2, type2, i))
849 return 0;
850 }
851 return 1;
852 }
853
854 isl_bool isl_space_match(__isl_keep isl_space *space1, enum isl_dim_type type1,
855 __isl_keep isl_space *space2, enum isl_dim_type type2)
856 {
857 if (!space1 || !space2)
858 return isl_bool_error;
859
860 return match(space1, type1, space2, type2);
861 }
862
863 static void get_ids(__isl_keep isl_space *dim, enum isl_dim_type type,
864 unsigned first, unsigned n, __isl_keep isl_id **ids)
865 {
866 int i;
867
868 for (i = 0; i < n ; ++i)
869 ids[i] = get_id(dim, type, first + i);
870 }
871
872 __isl_give isl_space *isl_space_extend(__isl_take isl_space *space,
873 unsigned nparam, unsigned n_in, unsigned n_out)
874 {
875 isl_id **ids = NULL;
876
877 if (!space)
878 return NULL;
879 if (space->nparam == nparam &&
880 space->n_in == n_in && space->n_out == n_out)
881 return space;
882
883 isl_assert(space->ctx, space->nparam <= nparam, goto error);
884 isl_assert(space->ctx, space->n_in <= n_in, goto error);
885 isl_assert(space->ctx, space->n_out <= n_out, goto error);
886
887 space = isl_space_cow(space);
888 if (!space)
889 goto error;
890
891 if (space->ids) {
892 unsigned n;
893 n = nparam + n_in + n_out;
894 if (n < nparam || n < n_in || n < n_out)
895 isl_die(isl_space_get_ctx(space), isl_error_invalid,
896 "overflow in total number of dimensions",
897 goto error);
898 ids = isl_calloc_array(space->ctx, isl_id *, n);
899 if (!ids)
900 goto error;
901 get_ids(space, isl_dim_param, 0, space->nparam, ids);
902 get_ids(space, isl_dim_in, 0, space->n_in, ids + nparam);
903 get_ids(space, isl_dim_out, 0, space->n_out,
904 ids + nparam + n_in);
905 free(space->ids);
906 space->ids = ids;
907 space->n_id = nparam + n_in + n_out;
908 }
909 space->nparam = nparam;
910 space->n_in = n_in;
911 space->n_out = n_out;
912
913 return space;
914 error:
915 free(ids);
916 isl_space_free(space);
917 return NULL;
918 }
919
920 __isl_give isl_space *isl_space_add_dims(__isl_take isl_space *dim,
921 enum isl_dim_type type, unsigned n)
922 {
923 dim = isl_space_reset(dim, type);
924 if (!dim)
925 return NULL;
926 switch (type) {
927 case isl_dim_param:
928 dim = isl_space_extend(dim,
929 dim->nparam + n, dim->n_in, dim->n_out);
930 if (dim && dim->nested[0] &&
931 !(dim->nested[0] = isl_space_add_dims(dim->nested[0],
932 isl_dim_param, n)))
933 goto error;
934 if (dim && dim->nested[1] &&
935 !(dim->nested[1] = isl_space_add_dims(dim->nested[1],
936 isl_dim_param, n)))
937 goto error;
938 return dim;
939 case isl_dim_in:
940 return isl_space_extend(dim,
941 dim->nparam, dim->n_in + n, dim->n_out);
942 case isl_dim_out:
943 return isl_space_extend(dim,
944 dim->nparam, dim->n_in, dim->n_out + n);
945 default:
946 isl_die(dim->ctx, isl_error_invalid,
947 "cannot add dimensions of specified type", goto error);
948 }
949 error:
950 isl_space_free(dim);
951 return NULL;
952 }
953
954 static int valid_dim_type(enum isl_dim_type type)
955 {
956 switch (type) {
957 case isl_dim_param:
958 case isl_dim_in:
959 case isl_dim_out:
960 return 1;
961 default:
962 return 0;
963 }
964 }
965
966 /* Insert "n" dimensions of type "type" at position "pos".
967 * If we are inserting parameters, then they are also inserted in
968 * any nested spaces.
969 */
970 __isl_give isl_space *isl_space_insert_dims(__isl_take isl_space *dim,
971 enum isl_dim_type type, unsigned pos, unsigned n)
972 {
973 isl_id **ids = NULL;
974
975 if (!dim)
976 return NULL;
977 if (n == 0)
978 return isl_space_reset(dim, type);
979
980 if (!valid_dim_type(type))
981 isl_die(dim->ctx, isl_error_invalid,
982 "cannot insert dimensions of specified type",
983 goto error);
984
985 isl_assert(dim->ctx, pos <= isl_space_dim(dim, type), goto error);
986
987 dim = isl_space_cow(dim);
988 if (!dim)
989 return NULL;
990
991 if (dim->ids) {
992 enum isl_dim_type t, o = isl_dim_param;
993 int off;
994 int s[3];
995 ids = isl_calloc_array(dim->ctx, isl_id *,
996 dim->nparam + dim->n_in + dim->n_out + n);
997 if (!ids)
998 goto error;
999 off = 0;
1000 s[isl_dim_param - o] = dim->nparam;
1001 s[isl_dim_in - o] = dim->n_in;
1002 s[isl_dim_out - o] = dim->n_out;
1003 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
1004 if (t != type) {
1005 get_ids(dim, t, 0, s[t - o], ids + off);
1006 off += s[t - o];
1007 } else {
1008 get_ids(dim, t, 0, pos, ids + off);
1009 off += pos + n;
1010 get_ids(dim, t, pos, s[t - o] - pos, ids + off);
1011 off += s[t - o] - pos;
1012 }
1013 }
1014 free(dim->ids);
1015 dim->ids = ids;
1016 dim->n_id = dim->nparam + dim->n_in + dim->n_out + n;
1017 }
1018 switch (type) {
1019 case isl_dim_param: dim->nparam += n; break;
1020 case isl_dim_in: dim->n_in += n; break;
1021 case isl_dim_out: dim->n_out += n; break;
1022 default: ;
1023 }
1024 dim = isl_space_reset(dim, type);
1025
1026 if (type == isl_dim_param) {
1027 if (dim && dim->nested[0] &&
1028 !(dim->nested[0] = isl_space_insert_dims(dim->nested[0],
1029 isl_dim_param, pos, n)))
1030 goto error;
1031 if (dim && dim->nested[1] &&
1032 !(dim->nested[1] = isl_space_insert_dims(dim->nested[1],
1033 isl_dim_param, pos, n)))
1034 goto error;
1035 }
1036
1037 return dim;
1038 error:
1039 isl_space_free(dim);
1040 return NULL;
1041 }
1042
1043 __isl_give isl_space *isl_space_move_dims(__isl_take isl_space *dim,
1044 enum isl_dim_type dst_type, unsigned dst_pos,
1045 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
1046 {
1047 int i;
1048
1049 if (!dim)
1050 return NULL;
1051 if (n == 0) {
1052 dim = isl_space_reset(dim, src_type);
1053 return isl_space_reset(dim, dst_type);
1054 }
1055
1056 isl_assert(dim->ctx, src_pos + n <= isl_space_dim(dim, src_type),
1057 goto error);
1058
1059 if (dst_type == src_type && dst_pos == src_pos)
1060 return dim;
1061
1062 isl_assert(dim->ctx, dst_type != src_type, goto error);
1063
1064 dim = isl_space_reset(dim, src_type);
1065 dim = isl_space_reset(dim, dst_type);
1066
1067 dim = isl_space_cow(dim);
1068 if (!dim)
1069 return NULL;
1070
1071 if (dim->ids) {
1072 isl_id **ids;
1073 enum isl_dim_type t, o = isl_dim_param;
1074 int off;
1075 int s[3];
1076 ids = isl_calloc_array(dim->ctx, isl_id *,
1077 dim->nparam + dim->n_in + dim->n_out);
1078 if (!ids)
1079 goto error;
1080 off = 0;
1081 s[isl_dim_param - o] = dim->nparam;
1082 s[isl_dim_in - o] = dim->n_in;
1083 s[isl_dim_out - o] = dim->n_out;
1084 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
1085 if (t == dst_type) {
1086 get_ids(dim, t, 0, dst_pos, ids + off);
1087 off += dst_pos;
1088 get_ids(dim, src_type, src_pos, n, ids + off);
1089 off += n;
1090 get_ids(dim, t, dst_pos, s[t - o] - dst_pos,
1091 ids + off);
1092 off += s[t - o] - dst_pos;
1093 } else if (t == src_type) {
1094 get_ids(dim, t, 0, src_pos, ids + off);
1095 off += src_pos;
1096 get_ids(dim, t, src_pos + n,
1097 s[t - o] - src_pos - n, ids + off);
1098 off += s[t - o] - src_pos - n;
1099 } else {
1100 get_ids(dim, t, 0, s[t - o], ids + off);
1101 off += s[t - o];
1102 }
1103 }
1104 free(dim->ids);
1105 dim->ids = ids;
1106 dim->n_id = dim->nparam + dim->n_in + dim->n_out;
1107 }
1108
1109 switch (dst_type) {
1110 case isl_dim_param: dim->nparam += n; break;
1111 case isl_dim_in: dim->n_in += n; break;
1112 case isl_dim_out: dim->n_out += n; break;
1113 default: ;
1114 }
1115
1116 switch (src_type) {
1117 case isl_dim_param: dim->nparam -= n; break;
1118 case isl_dim_in: dim->n_in -= n; break;
1119 case isl_dim_out: dim->n_out -= n; break;
1120 default: ;
1121 }
1122
1123 if (dst_type != isl_dim_param && src_type != isl_dim_param)
1124 return dim;
1125
1126 for (i = 0; i < 2; ++i) {
1127 if (!dim->nested[i])
1128 continue;
1129 dim->nested[i] = isl_space_replace(dim->nested[i],
1130 isl_dim_param, dim);
1131 if (!dim->nested[i])
1132 goto error;
1133 }
1134
1135 return dim;
1136 error:
1137 isl_space_free(dim);
1138 return NULL;
1139 }
1140
1141 __isl_give isl_space *isl_space_join(__isl_take isl_space *left,
1142 __isl_take isl_space *right)
1143 {
1144 isl_space *dim;
1145
1146 if (!left || !right)
1147 goto error;
1148
1149 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
1150 goto error);
1151 isl_assert(left->ctx,
1152 isl_space_tuple_is_equal(left, isl_dim_out, right, isl_dim_in),
1153 goto error);
1154
1155 dim = isl_space_alloc(left->ctx, left->nparam, left->n_in, right->n_out);
1156 if (!dim)
1157 goto error;
1158
1159 dim = copy_ids(dim, isl_dim_param, 0, left, isl_dim_param);
1160 dim = copy_ids(dim, isl_dim_in, 0, left, isl_dim_in);
1161 dim = copy_ids(dim, isl_dim_out, 0, right, isl_dim_out);
1162
1163 if (dim && left->tuple_id[0] &&
1164 !(dim->tuple_id[0] = isl_id_copy(left->tuple_id[0])))
1165 goto error;
1166 if (dim && right->tuple_id[1] &&
1167 !(dim->tuple_id[1] = isl_id_copy(right->tuple_id[1])))
1168 goto error;
1169 if (dim && left->nested[0] &&
1170 !(dim->nested[0] = isl_space_copy(left->nested[0])))
1171 goto error;
1172 if (dim && right->nested[1] &&
1173 !(dim->nested[1] = isl_space_copy(right->nested[1])))
1174 goto error;
1175
1176 isl_space_free(left);
1177 isl_space_free(right);
1178
1179 return dim;
1180 error:
1181 isl_space_free(left);
1182 isl_space_free(right);
1183 return NULL;
1184 }
1185
1186 /* Given two map spaces { A -> C } and { B -> D }, construct the space
1187 * { [A -> B] -> [C -> D] }.
1188 * Given two set spaces { A } and { B }, construct the space { [A -> B] }.
1189 */
1190 __isl_give isl_space *isl_space_product(__isl_take isl_space *left,
1191 __isl_take isl_space *right)
1192 {
1193 isl_space *dom1, *dom2, *nest1, *nest2;
1194 int is_set;
1195
1196 if (!left || !right)
1197 goto error;
1198
1199 is_set = isl_space_is_set(left);
1200 if (is_set != isl_space_is_set(right))
1201 isl_die(isl_space_get_ctx(left), isl_error_invalid,
1202 "expecting either two set spaces or two map spaces",
1203 goto error);
1204 if (is_set)
1205 return isl_space_range_product(left, right);
1206
1207 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
1208 goto error);
1209
1210 dom1 = isl_space_domain(isl_space_copy(left));
1211 dom2 = isl_space_domain(isl_space_copy(right));
1212 nest1 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
1213
1214 dom1 = isl_space_range(left);
1215 dom2 = isl_space_range(right);
1216 nest2 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
1217
1218 return isl_space_join(isl_space_reverse(nest1), nest2);
1219 error:
1220 isl_space_free(left);
1221 isl_space_free(right);
1222 return NULL;
1223 }
1224
1225 /* Given two spaces { A -> C } and { B -> C }, construct the space
1226 * { [A -> B] -> C }
1227 */
1228 __isl_give isl_space *isl_space_domain_product(__isl_take isl_space *left,
1229 __isl_take isl_space *right)
1230 {
1231 isl_space *ran, *dom1, *dom2, *nest;
1232
1233 if (!left || !right)
1234 goto error;
1235
1236 if (!match(left, isl_dim_param, right, isl_dim_param))
1237 isl_die(left->ctx, isl_error_invalid,
1238 "parameters need to match", goto error);
1239 if (!isl_space_tuple_is_equal(left, isl_dim_out, right, isl_dim_out))
1240 isl_die(left->ctx, isl_error_invalid,
1241 "ranges need to match", goto error);
1242
1243 ran = isl_space_range(isl_space_copy(left));
1244
1245 dom1 = isl_space_domain(left);
1246 dom2 = isl_space_domain(right);
1247 nest = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
1248
1249 return isl_space_join(isl_space_reverse(nest), ran);
1250 error:
1251 isl_space_free(left);
1252 isl_space_free(right);
1253 return NULL;
1254 }
1255
1256 __isl_give isl_space *isl_space_range_product(__isl_take isl_space *left,
1257 __isl_take isl_space *right)
1258 {
1259 isl_space *dom, *ran1, *ran2, *nest;
1260
1261 if (!left || !right)
1262 goto error;
1263
1264 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
1265 goto error);
1266 if (!isl_space_tuple_is_equal(left, isl_dim_in, right, isl_dim_in))
1267 isl_die(left->ctx, isl_error_invalid,
1268 "domains need to match", goto error);
1269
1270 dom = isl_space_domain(isl_space_copy(left));
1271
1272 ran1 = isl_space_range(left);
1273 ran2 = isl_space_range(right);
1274 nest = isl_space_wrap(isl_space_join(isl_space_reverse(ran1), ran2));
1275
1276 return isl_space_join(isl_space_reverse(dom), nest);
1277 error:
1278 isl_space_free(left);
1279 isl_space_free(right);
1280 return NULL;
1281 }
1282
1283 /* Given a space of the form [A -> B] -> [C -> D], return the space A -> C.
1284 */
1285 __isl_give isl_space *isl_space_factor_domain(__isl_take isl_space *space)
1286 {
1287 space = isl_space_domain_factor_domain(space);
1288 space = isl_space_range_factor_domain(space);
1289 return space;
1290 }
1291
1292 /* Given a space of the form [A -> B] -> C, return the space A -> C.
1293 */
1294 __isl_give isl_space *isl_space_domain_factor_domain(
1295 __isl_take isl_space *space)
1296 {
1297 isl_space *nested;
1298 isl_space *domain;
1299
1300 if (!space)
1301 return NULL;
1302 if (!isl_space_domain_is_wrapping(space))
1303 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1304 "domain not a product", return isl_space_free(space));
1305
1306 nested = space->nested[0];
1307 domain = isl_space_copy(space);
1308 domain = isl_space_drop_dims(domain, isl_dim_in,
1309 nested->n_in, nested->n_out);
1310 if (!domain)
1311 return isl_space_free(space);
1312 if (nested->tuple_id[0]) {
1313 domain->tuple_id[0] = isl_id_copy(nested->tuple_id[0]);
1314 if (!domain->tuple_id[0])
1315 goto error;
1316 }
1317 if (nested->nested[0]) {
1318 domain->nested[0] = isl_space_copy(nested->nested[0]);
1319 if (!domain->nested[0])
1320 goto error;
1321 }
1322
1323 isl_space_free(space);
1324 return domain;
1325 error:
1326 isl_space_free(space);
1327 isl_space_free(domain);
1328 return NULL;
1329 }
1330
1331 /* Given a space of the form [A -> B] -> C, return the space B -> C.
1332 */
1333 __isl_give isl_space *isl_space_domain_factor_range(
1334 __isl_take isl_space *space)
1335 {
1336 isl_space *nested;
1337 isl_space *range;
1338
1339 if (!space)
1340 return NULL;
1341 if (!isl_space_domain_is_wrapping(space))
1342 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1343 "domain not a product", return isl_space_free(space));
1344
1345 nested = space->nested[0];
1346 range = isl_space_copy(space);
1347 range = isl_space_drop_dims(range, isl_dim_in, 0, nested->n_in);
1348 if (!range)
1349 return isl_space_free(space);
1350 if (nested->tuple_id[1]) {
1351 range->tuple_id[0] = isl_id_copy(nested->tuple_id[1]);
1352 if (!range->tuple_id[0])
1353 goto error;
1354 }
1355 if (nested->nested[1]) {
1356 range->nested[0] = isl_space_copy(nested->nested[1]);
1357 if (!range->nested[0])
1358 goto error;
1359 }
1360
1361 isl_space_free(space);
1362 return range;
1363 error:
1364 isl_space_free(space);
1365 isl_space_free(range);
1366 return NULL;
1367 }
1368
1369 /* Given a space of the form A -> [B -> C], return the space A -> B.
1370 */
1371 __isl_give isl_space *isl_space_range_factor_domain(
1372 __isl_take isl_space *space)
1373 {
1374 isl_space *nested;
1375 isl_space *domain;
1376
1377 if (!space)
1378 return NULL;
1379 if (!isl_space_range_is_wrapping(space))
1380 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1381 "range not a product", return isl_space_free(space));
1382
1383 nested = space->nested[1];
1384 domain = isl_space_copy(space);
1385 domain = isl_space_drop_dims(domain, isl_dim_out,
1386 nested->n_in, nested->n_out);
1387 if (!domain)
1388 return isl_space_free(space);
1389 if (nested->tuple_id[0]) {
1390 domain->tuple_id[1] = isl_id_copy(nested->tuple_id[0]);
1391 if (!domain->tuple_id[1])
1392 goto error;
1393 }
1394 if (nested->nested[0]) {
1395 domain->nested[1] = isl_space_copy(nested->nested[0]);
1396 if (!domain->nested[1])
1397 goto error;
1398 }
1399
1400 isl_space_free(space);
1401 return domain;
1402 error:
1403 isl_space_free(space);
1404 isl_space_free(domain);
1405 return NULL;
1406 }
1407
1408 /* Internal function that selects the range of the map that is
1409 * embedded in either a set space or the range of a map space.
1410 * In particular, given a space of the form [A -> B], return the space B.
1411 * Given a space of the form A -> [B -> C], return the space A -> C.
1412 */
1413 static __isl_give isl_space *range_factor_range(__isl_take isl_space *space)
1414 {
1415 isl_space *nested;
1416 isl_space *range;
1417
1418 if (!space)
1419 return NULL;
1420
1421 nested = space->nested[1];
1422 range = isl_space_copy(space);
1423 range = isl_space_drop_dims(range, isl_dim_out, 0, nested->n_in);
1424 if (!range)
1425 return isl_space_free(space);
1426 if (nested->tuple_id[1]) {
1427 range->tuple_id[1] = isl_id_copy(nested->tuple_id[1]);
1428 if (!range->tuple_id[1])
1429 goto error;
1430 }
1431 if (nested->nested[1]) {
1432 range->nested[1] = isl_space_copy(nested->nested[1]);
1433 if (!range->nested[1])
1434 goto error;
1435 }
1436
1437 isl_space_free(space);
1438 return range;
1439 error:
1440 isl_space_free(space);
1441 isl_space_free(range);
1442 return NULL;
1443 }
1444
1445 /* Given a space of the form A -> [B -> C], return the space A -> C.
1446 */
1447 __isl_give isl_space *isl_space_range_factor_range(
1448 __isl_take isl_space *space)
1449 {
1450 if (!space)
1451 return NULL;
1452 if (!isl_space_range_is_wrapping(space))
1453 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1454 "range not a product", return isl_space_free(space));
1455
1456 return range_factor_range(space);
1457 }
1458
1459 /* Given a space of the form [A -> B], return the space B.
1460 */
1461 static __isl_give isl_space *set_factor_range(__isl_take isl_space *space)
1462 {
1463 if (!space)
1464 return NULL;
1465 if (!isl_space_is_wrapping(space))
1466 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1467 "not a product", return isl_space_free(space));
1468
1469 return range_factor_range(space);
1470 }
1471
1472 /* Given a space of the form [A -> B] -> [C -> D], return the space B -> D.
1473 * Given a space of the form [A -> B], return the space B.
1474 */
1475 __isl_give isl_space *isl_space_factor_range(__isl_take isl_space *space)
1476 {
1477 if (!space)
1478 return NULL;
1479 if (isl_space_is_set(space))
1480 return set_factor_range(space);
1481 space = isl_space_domain_factor_range(space);
1482 space = isl_space_range_factor_range(space);
1483 return space;
1484 }
1485
1486 __isl_give isl_space *isl_space_map_from_set(__isl_take isl_space *dim)
1487 {
1488 isl_ctx *ctx;
1489 isl_id **ids = NULL;
1490
1491 if (!dim)
1492 return NULL;
1493 ctx = isl_space_get_ctx(dim);
1494 if (!isl_space_is_set(dim))
1495 isl_die(ctx, isl_error_invalid, "not a set space", goto error);
1496 dim = isl_space_cow(dim);
1497 if (!dim)
1498 return NULL;
1499 if (dim->ids) {
1500 ids = isl_calloc_array(dim->ctx, isl_id *,
1501 dim->nparam + dim->n_out + dim->n_out);
1502 if (!ids)
1503 goto error;
1504 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
1505 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->nparam);
1506 }
1507 dim->n_in = dim->n_out;
1508 if (ids) {
1509 free(dim->ids);
1510 dim->ids = ids;
1511 dim->n_id = dim->nparam + dim->n_out + dim->n_out;
1512 dim = copy_ids(dim, isl_dim_out, 0, dim, isl_dim_in);
1513 }
1514 isl_id_free(dim->tuple_id[0]);
1515 dim->tuple_id[0] = isl_id_copy(dim->tuple_id[1]);
1516 isl_space_free(dim->nested[0]);
1517 dim->nested[0] = isl_space_copy(dim->nested[1]);
1518 return dim;
1519 error:
1520 isl_space_free(dim);
1521 return NULL;
1522 }
1523
1524 __isl_give isl_space *isl_space_map_from_domain_and_range(
1525 __isl_take isl_space *domain, __isl_take isl_space *range)
1526 {
1527 if (!domain || !range)
1528 goto error;
1529 if (!isl_space_is_set(domain))
1530 isl_die(isl_space_get_ctx(domain), isl_error_invalid,
1531 "domain is not a set space", goto error);
1532 if (!isl_space_is_set(range))
1533 isl_die(isl_space_get_ctx(range), isl_error_invalid,
1534 "range is not a set space", goto error);
1535 return isl_space_join(isl_space_reverse(domain), range);
1536 error:
1537 isl_space_free(domain);
1538 isl_space_free(range);
1539 return NULL;
1540 }
1541
1542 static __isl_give isl_space *set_ids(__isl_take isl_space *dim,
1543 enum isl_dim_type type,
1544 unsigned first, unsigned n, __isl_take isl_id **ids)
1545 {
1546 int i;
1547
1548 for (i = 0; i < n ; ++i)
1549 dim = set_id(dim, type, first + i, ids[i]);
1550
1551 return dim;
1552 }
1553
1554 __isl_give isl_space *isl_space_reverse(__isl_take isl_space *dim)
1555 {
1556 unsigned t;
1557 isl_space *nested;
1558 isl_id **ids = NULL;
1559 isl_id *id;
1560
1561 if (!dim)
1562 return NULL;
1563 if (match(dim, isl_dim_in, dim, isl_dim_out))
1564 return dim;
1565
1566 dim = isl_space_cow(dim);
1567 if (!dim)
1568 return NULL;
1569
1570 id = dim->tuple_id[0];
1571 dim->tuple_id[0] = dim->tuple_id[1];
1572 dim->tuple_id[1] = id;
1573
1574 nested = dim->nested[0];
1575 dim->nested[0] = dim->nested[1];
1576 dim->nested[1] = nested;
1577
1578 if (dim->ids) {
1579 int n_id = dim->n_in + dim->n_out;
1580 ids = isl_alloc_array(dim->ctx, isl_id *, n_id);
1581 if (n_id && !ids)
1582 goto error;
1583 get_ids(dim, isl_dim_in, 0, dim->n_in, ids);
1584 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->n_in);
1585 }
1586
1587 t = dim->n_in;
1588 dim->n_in = dim->n_out;
1589 dim->n_out = t;
1590
1591 if (dim->ids) {
1592 dim = set_ids(dim, isl_dim_out, 0, dim->n_out, ids);
1593 dim = set_ids(dim, isl_dim_in, 0, dim->n_in, ids + dim->n_out);
1594 free(ids);
1595 }
1596
1597 return dim;
1598 error:
1599 free(ids);
1600 isl_space_free(dim);
1601 return NULL;
1602 }
1603
1604 __isl_give isl_space *isl_space_drop_dims(__isl_take isl_space *dim,
1605 enum isl_dim_type type, unsigned first, unsigned num)
1606 {
1607 int i;
1608
1609 if (!dim)
1610 return NULL;
1611
1612 if (num == 0)
1613 return isl_space_reset(dim, type);
1614
1615 if (!valid_dim_type(type))
1616 isl_die(dim->ctx, isl_error_invalid,
1617 "cannot drop dimensions of specified type", goto error);
1618
1619 if (first + num > n(dim, type) || first + num < first)
1620 isl_die(isl_space_get_ctx(dim), isl_error_invalid,
1621 "index out of bounds", return isl_space_free(dim));
1622 dim = isl_space_cow(dim);
1623 if (!dim)
1624 goto error;
1625 if (dim->ids) {
1626 dim = extend_ids(dim);
1627 if (!dim)
1628 goto error;
1629 for (i = 0; i < num; ++i)
1630 isl_id_free(get_id(dim, type, first + i));
1631 for (i = first+num; i < n(dim, type); ++i)
1632 set_id(dim, type, i - num, get_id(dim, type, i));
1633 switch (type) {
1634 case isl_dim_param:
1635 get_ids(dim, isl_dim_in, 0, dim->n_in,
1636 dim->ids + offset(dim, isl_dim_in) - num);
1637 case isl_dim_in:
1638 get_ids(dim, isl_dim_out, 0, dim->n_out,
1639 dim->ids + offset(dim, isl_dim_out) - num);
1640 default:
1641 ;
1642 }
1643 dim->n_id -= num;
1644 }
1645 switch (type) {
1646 case isl_dim_param: dim->nparam -= num; break;
1647 case isl_dim_in: dim->n_in -= num; break;
1648 case isl_dim_out: dim->n_out -= num; break;
1649 default: ;
1650 }
1651 dim = isl_space_reset(dim, type);
1652 if (type == isl_dim_param) {
1653 if (dim && dim->nested[0] &&
1654 !(dim->nested[0] = isl_space_drop_dims(dim->nested[0],
1655 isl_dim_param, first, num)))
1656 goto error;
1657 if (dim && dim->nested[1] &&
1658 !(dim->nested[1] = isl_space_drop_dims(dim->nested[1],
1659 isl_dim_param, first, num)))
1660 goto error;
1661 }
1662 return dim;
1663 error:
1664 isl_space_free(dim);
1665 return NULL;
1666 }
1667
1668 __isl_give isl_space *isl_space_drop_inputs(__isl_take isl_space *dim,
1669 unsigned first, unsigned n)
1670 {
1671 if (!dim)
1672 return NULL;
1673 return isl_space_drop_dims(dim, isl_dim_in, first, n);
1674 }
1675
1676 __isl_give isl_space *isl_space_drop_outputs(__isl_take isl_space *dim,
1677 unsigned first, unsigned n)
1678 {
1679 if (!dim)
1680 return NULL;
1681 return isl_space_drop_dims(dim, isl_dim_out, first, n);
1682 }
1683
1684 __isl_give isl_space *isl_space_domain(__isl_take isl_space *dim)
1685 {
1686 if (!dim)
1687 return NULL;
1688 dim = isl_space_drop_outputs(dim, 0, dim->n_out);
1689 dim = isl_space_reverse(dim);
1690 dim = mark_as_set(dim);
1691 return dim;
1692 }
1693
1694 __isl_give isl_space *isl_space_from_domain(__isl_take isl_space *dim)
1695 {
1696 if (!dim)
1697 return NULL;
1698 if (!isl_space_is_set(dim))
1699 isl_die(isl_space_get_ctx(dim), isl_error_invalid,
1700 "not a set space", goto error);
1701 dim = isl_space_reverse(dim);
1702 dim = isl_space_reset(dim, isl_dim_out);
1703 return dim;
1704 error:
1705 isl_space_free(dim);
1706 return NULL;
1707 }
1708
1709 __isl_give isl_space *isl_space_range(__isl_take isl_space *dim)
1710 {
1711 if (!dim)
1712 return NULL;
1713 dim = isl_space_drop_inputs(dim, 0, dim->n_in);
1714 dim = mark_as_set(dim);
1715 return dim;
1716 }
1717
1718 __isl_give isl_space *isl_space_from_range(__isl_take isl_space *dim)
1719 {
1720 if (!dim)
1721 return NULL;
1722 if (!isl_space_is_set(dim))
1723 isl_die(isl_space_get_ctx(dim), isl_error_invalid,
1724 "not a set space", goto error);
1725 return isl_space_reset(dim, isl_dim_in);
1726 error:
1727 isl_space_free(dim);
1728 return NULL;
1729 }
1730
1731 /* Given a map space A -> B, return the map space [A -> B] -> A.
1732 */
1733 __isl_give isl_space *isl_space_domain_map(__isl_take isl_space *space)
1734 {
1735 isl_space *domain;
1736
1737 domain = isl_space_from_range(isl_space_domain(isl_space_copy(space)));
1738 space = isl_space_from_domain(isl_space_wrap(space));
1739 space = isl_space_join(space, domain);
1740
1741 return space;
1742 }
1743
1744 /* Given a map space A -> B, return the map space [A -> B] -> B.
1745 */
1746 __isl_give isl_space *isl_space_range_map(__isl_take isl_space *space)
1747 {
1748 isl_space *range;
1749
1750 range = isl_space_from_range(isl_space_range(isl_space_copy(space)));
1751 space = isl_space_from_domain(isl_space_wrap(space));
1752 space = isl_space_join(space, range);
1753
1754 return space;
1755 }
1756
1757 __isl_give isl_space *isl_space_params(__isl_take isl_space *space)
1758 {
1759 if (isl_space_is_params(space))
1760 return space;
1761 space = isl_space_drop_dims(space,
1762 isl_dim_in, 0, isl_space_dim(space, isl_dim_in));
1763 space = isl_space_drop_dims(space,
1764 isl_dim_out, 0, isl_space_dim(space, isl_dim_out));
1765 space = mark_as_params(space);
1766 return space;
1767 }
1768
1769 __isl_give isl_space *isl_space_set_from_params(__isl_take isl_space *space)
1770 {
1771 if (!space)
1772 return NULL;
1773 if (!isl_space_is_params(space))
1774 isl_die(isl_space_get_ctx(space), isl_error_invalid,
1775 "not a parameter space", goto error);
1776 return isl_space_reset(space, isl_dim_set);
1777 error:
1778 isl_space_free(space);
1779 return NULL;
1780 }
1781
1782 __isl_give isl_space *isl_space_as_set_space(__isl_take isl_space *dim)
1783 {
1784 dim = isl_space_cow(dim);
1785 if (!dim)
1786 return NULL;
1787
1788 dim->n_out += dim->n_in;
1789 dim->n_in = 0;
1790 dim = isl_space_reset(dim, isl_dim_in);
1791 dim = isl_space_reset(dim, isl_dim_out);
1792
1793 return dim;
1794 }
1795
1796 __isl_give isl_space *isl_space_underlying(__isl_take isl_space *dim,
1797 unsigned n_div)
1798 {
1799 int i;
1800
1801 if (!dim)
1802 return NULL;
1803 if (n_div == 0 &&
1804 dim->nparam == 0 && dim->n_in == 0 && dim->n_id == 0)
1805 return isl_space_reset(isl_space_reset(dim, isl_dim_in), isl_dim_out);
1806 dim = isl_space_cow(dim);
1807 if (!dim)
1808 return NULL;
1809 dim->n_out += dim->nparam + dim->n_in + n_div;
1810 dim->nparam = 0;
1811 dim->n_in = 0;
1812
1813 for (i = 0; i < dim->n_id; ++i)
1814 isl_id_free(get_id(dim, isl_dim_out, i));
1815 dim->n_id = 0;
1816 dim = isl_space_reset(dim, isl_dim_in);
1817 dim = isl_space_reset(dim, isl_dim_out);
1818
1819 return dim;
1820 }
1821
1822 /* Are the two spaces the same, including positions and names of parameters?
1823 */
1824 isl_bool isl_space_is_equal(__isl_keep isl_space *dim1,
1825 __isl_keep isl_space *dim2)
1826 {
1827 if (!dim1 || !dim2)
1828 return isl_bool_error;
1829 if (dim1 == dim2)
1830 return isl_bool_true;
1831 return match(dim1, isl_dim_param, dim2, isl_dim_param) &&
1832 isl_space_tuple_is_equal(dim1, isl_dim_in, dim2, isl_dim_in) &&
1833 isl_space_tuple_is_equal(dim1, isl_dim_out, dim2, isl_dim_out);
1834 }
1835
1836 /* Is space1 equal to the domain of space2?
1837 *
1838 * In the internal version we also allow space2 to be the space of a set,
1839 * provided space1 is a parameter space.
1840 */
1841 isl_bool isl_space_is_domain_internal(__isl_keep isl_space *space1,
1842 __isl_keep isl_space *space2)
1843 {
1844 if (!space1 || !space2)
1845 return isl_bool_error;
1846 if (!isl_space_is_set(space1))
1847 return isl_bool_false;
1848 return match(space1, isl_dim_param, space2, isl_dim_param) &&
1849 isl_space_tuple_is_equal(space1, isl_dim_set,
1850 space2, isl_dim_in);
1851 }
1852
1853 /* Is space1 equal to the domain of space2?
1854 */
1855 isl_bool isl_space_is_domain(__isl_keep isl_space *space1,
1856 __isl_keep isl_space *space2)
1857 {
1858 if (!space2)
1859 return isl_bool_error;
1860 if (!isl_space_is_map(space2))
1861 return isl_bool_false;
1862 return isl_space_is_domain_internal(space1, space2);
1863 }
1864
1865 /* Is space1 equal to the range of space2?
1866 *
1867 * In the internal version, space2 is allowed to be the space of a set,
1868 * in which case it should be equal to space1.
1869 */
1870 isl_bool isl_space_is_range_internal(__isl_keep isl_space *space1,
1871 __isl_keep isl_space *space2)
1872 {
1873 if (!space1 || !space2)
1874 return isl_bool_error;
1875 if (!isl_space_is_set(space1))
1876 return isl_bool_false;
1877 return match(space1, isl_dim_param, space2, isl_dim_param) &&
1878 isl_space_tuple_is_equal(space1, isl_dim_set,
1879 space2, isl_dim_out);
1880 }
1881
1882 /* Is space1 equal to the range of space2?
1883 */
1884 isl_bool isl_space_is_range(__isl_keep isl_space *space1,
1885 __isl_keep isl_space *space2)
1886 {
1887 if (!space2)
1888 return isl_bool_error;
1889 if (!isl_space_is_map(space2))
1890 return isl_bool_false;
1891 return isl_space_is_range_internal(space1, space2);
1892 }
1893
1894 int isl_space_compatible_internal(__isl_keep isl_space *dim1,
1895 __isl_keep isl_space *dim2)
1896 {
1897 return dim1->nparam == dim2->nparam &&
1898 dim1->n_in + dim1->n_out == dim2->n_in + dim2->n_out;
1899 }
1900
1901 int isl_space_compatible(__isl_keep isl_space *space1,
1902 __isl_keep isl_space *space2)
1903 {
1904 return isl_space_compatible_internal(space1, space2);
1905 }
1906
1907 /* Update "hash" by hashing in "space".
1908 * Changes in this function should be reflected in isl_hash_space_domain.
1909 */
1910 static uint32_t isl_hash_space(uint32_t hash, __isl_keep isl_space *space)
1911 {
1912 int i;
1913 isl_id *id;
1914
1915 if (!space)
1916 return hash;
1917
1918 isl_hash_byte(hash, space->nparam % 256);
1919 isl_hash_byte(hash, space->n_in % 256);
1920 isl_hash_byte(hash, space->n_out % 256);
1921
1922 for (i = 0; i < space->nparam; ++i) {
1923 id = get_id(space, isl_dim_param, i);
1924 hash = isl_hash_id(hash, id);
1925 }
1926
1927 id = tuple_id(space, isl_dim_in);
1928 hash = isl_hash_id(hash, id);
1929 id = tuple_id(space, isl_dim_out);
1930 hash = isl_hash_id(hash, id);
1931
1932 hash = isl_hash_space(hash, space->nested[0]);
1933 hash = isl_hash_space(hash, space->nested[1]);
1934
1935 return hash;
1936 }
1937
1938 /* Update "hash" by hashing in the domain of "space".
1939 * The result of this function is equal to the result of applying
1940 * isl_hash_space to the domain of "space".
1941 */
1942 static uint32_t isl_hash_space_domain(uint32_t hash,
1943 __isl_keep isl_space *space)
1944 {
1945 int i;
1946 isl_id *id;
1947
1948 if (!space)
1949 return hash;
1950
1951 isl_hash_byte(hash, space->nparam % 256);
1952 isl_hash_byte(hash, 0);
1953 isl_hash_byte(hash, space->n_in % 256);
1954
1955 for (i = 0; i < space->nparam; ++i) {
1956 id = get_id(space, isl_dim_param, i);
1957 hash = isl_hash_id(hash, id);
1958 }
1959
1960 hash = isl_hash_id(hash, &isl_id_none);
1961 id = tuple_id(space, isl_dim_in);
1962 hash = isl_hash_id(hash, id);
1963
1964 hash = isl_hash_space(hash, space->nested[0]);
1965
1966 return hash;
1967 }
1968
1969 uint32_t isl_space_get_hash(__isl_keep isl_space *dim)
1970 {
1971 uint32_t hash;
1972
1973 if (!dim)
1974 return 0;
1975
1976 hash = isl_hash_init();
1977 hash = isl_hash_space(hash, dim);
1978
1979 return hash;
1980 }
1981
1982 /* Return the hash value of the domain of "space".
1983 * That is, isl_space_get_domain_hash(space) is equal to
1984 * isl_space_get_hash(isl_space_domain(space)).
1985 */
1986 uint32_t isl_space_get_domain_hash(__isl_keep isl_space *space)
1987 {
1988 uint32_t hash;
1989
1990 if (!space)
1991 return 0;
1992
1993 hash = isl_hash_init();
1994 hash = isl_hash_space_domain(hash, space);
1995
1996 return hash;
1997 }
1998
1999 isl_bool isl_space_is_wrapping(__isl_keep isl_space *dim)
2000 {
2001 if (!dim)
2002 return isl_bool_error;
2003
2004 if (!isl_space_is_set(dim))
2005 return isl_bool_false;
2006
2007 return dim->nested[1] != NULL;
2008 }
2009
2010 /* Is "space" the space of a map where the domain is a wrapped map space?
2011 */
2012 isl_bool isl_space_domain_is_wrapping(__isl_keep isl_space *space)
2013 {
2014 if (!space)
2015 return isl_bool_error;
2016
2017 if (isl_space_is_set(space))
2018 return isl_bool_false;
2019
2020 return space->nested[0] != NULL;
2021 }
2022
2023 /* Is "space" the space of a map where the range is a wrapped map space?
2024 */
2025 isl_bool isl_space_range_is_wrapping(__isl_keep isl_space *space)
2026 {
2027 if (!space)
2028 return isl_bool_error;
2029
2030 if (isl_space_is_set(space))
2031 return isl_bool_false;
2032
2033 return space->nested[1] != NULL;
2034 }
2035
2036 __isl_give isl_space *isl_space_wrap(__isl_take isl_space *dim)
2037 {
2038 isl_space *wrap;
2039
2040 if (!dim)
2041 return NULL;
2042
2043 wrap = isl_space_set_alloc(dim->ctx,
2044 dim->nparam, dim->n_in + dim->n_out);
2045
2046 wrap = copy_ids(wrap, isl_dim_param, 0, dim, isl_dim_param);
2047 wrap = copy_ids(wrap, isl_dim_set, 0, dim, isl_dim_in);
2048 wrap = copy_ids(wrap, isl_dim_set, dim->n_in, dim, isl_dim_out);
2049
2050 if (!wrap)
2051 goto error;
2052
2053 wrap->nested[1] = dim;
2054
2055 return wrap;
2056 error:
2057 isl_space_free(dim);
2058 return NULL;
2059 }
2060
2061 __isl_give isl_space *isl_space_unwrap(__isl_take isl_space *dim)
2062 {
2063 isl_space *unwrap;
2064
2065 if (!dim)
2066 return NULL;
2067
2068 if (!isl_space_is_wrapping(dim))
2069 isl_die(dim->ctx, isl_error_invalid, "not a wrapping space",
2070 goto error);
2071
2072 unwrap = isl_space_copy(dim->nested[1]);
2073 isl_space_free(dim);
2074
2075 return unwrap;
2076 error:
2077 isl_space_free(dim);
2078 return NULL;
2079 }
2080
2081 int isl_space_is_named_or_nested(__isl_keep isl_space *dim, enum isl_dim_type type)
2082 {
2083 if (type != isl_dim_in && type != isl_dim_out)
2084 return 0;
2085 if (!dim)
2086 return -1;
2087 if (dim->tuple_id[type - isl_dim_in])
2088 return 1;
2089 if (dim->nested[type - isl_dim_in])
2090 return 1;
2091 return 0;
2092 }
2093
2094 int isl_space_may_be_set(__isl_keep isl_space *dim)
2095 {
2096 if (!dim)
2097 return -1;
2098 if (isl_space_is_set(dim))
2099 return 1;
2100 if (isl_space_dim(dim, isl_dim_in) != 0)
2101 return 0;
2102 if (isl_space_is_named_or_nested(dim, isl_dim_in))
2103 return 0;
2104 return 1;
2105 }
2106
2107 __isl_give isl_space *isl_space_reset(__isl_take isl_space *dim,
2108 enum isl_dim_type type)
2109 {
2110 if (!isl_space_is_named_or_nested(dim, type))
2111 return dim;
2112
2113 dim = isl_space_cow(dim);
2114 if (!dim)
2115 return NULL;
2116
2117 isl_id_free(dim->tuple_id[type - isl_dim_in]);
2118 dim->tuple_id[type - isl_dim_in] = NULL;
2119 isl_space_free(dim->nested[type - isl_dim_in]);
2120 dim->nested[type - isl_dim_in] = NULL;
2121
2122 return dim;
2123 }
2124
2125 __isl_give isl_space *isl_space_flatten(__isl_take isl_space *dim)
2126 {
2127 if (!dim)
2128 return NULL;
2129 if (!dim->nested[0] && !dim->nested[1])
2130 return dim;
2131
2132 if (dim->nested[0])
2133 dim = isl_space_reset(dim, isl_dim_in);
2134 if (dim && dim->nested[1])
2135 dim = isl_space_reset(dim, isl_dim_out);
2136
2137 return dim;
2138 }
2139
2140 __isl_give isl_space *isl_space_flatten_domain(__isl_take isl_space *dim)
2141 {
2142 if (!dim)
2143 return NULL;
2144 if (!dim->nested[0])
2145 return dim;
2146
2147 return isl_space_reset(dim, isl_dim_in);
2148 }
2149
2150 __isl_give isl_space *isl_space_flatten_range(__isl_take isl_space *dim)
2151 {
2152 if (!dim)
2153 return NULL;
2154 if (!dim->nested[1])
2155 return dim;
2156
2157 return isl_space_reset(dim, isl_dim_out);
2158 }
2159
2160 /* Replace the dimensions of the given type of dst by those of src.
2161 */
2162 __isl_give isl_space *isl_space_replace(__isl_take isl_space *dst,
2163 enum isl_dim_type type, __isl_keep isl_space *src)
2164 {
2165 dst = isl_space_cow(dst);
2166
2167 if (!dst || !src)
2168 goto error;
2169
2170 dst = isl_space_drop_dims(dst, type, 0, isl_space_dim(dst, type));
2171 dst = isl_space_add_dims(dst, type, isl_space_dim(src, type));
2172 dst = copy_ids(dst, type, 0, src, type);
2173
2174 if (dst && type == isl_dim_param) {
2175 int i;
2176 for (i = 0; i <= 1; ++i) {
2177 if (!dst->nested[i])
2178 continue;
2179 dst->nested[i] = isl_space_replace(dst->nested[i],
2180 type, src);
2181 if (!dst->nested[i])
2182 goto error;
2183 }
2184 }
2185
2186 return dst;
2187 error:
2188 isl_space_free(dst);
2189 return NULL;
2190 }
2191
2192 /* Given a dimension specification "dim" of a set, create a dimension
2193 * specification for the lift of the set. In particular, the result
2194 * is of the form [dim -> local[..]], with n_local variables in the
2195 * range of the wrapped map.
2196 */
2197 __isl_give isl_space *isl_space_lift(__isl_take isl_space *dim, unsigned n_local)
2198 {
2199 isl_space *local_dim;
2200
2201 if (!dim)
2202 return NULL;
2203
2204 local_dim = isl_space_dup(dim);
2205 local_dim = isl_space_drop_dims(local_dim, isl_dim_set, 0, dim->n_out);
2206 local_dim = isl_space_add_dims(local_dim, isl_dim_set, n_local);
2207 local_dim = isl_space_set_tuple_name(local_dim, isl_dim_set, "local");
2208 dim = isl_space_join(isl_space_from_domain(dim),
2209 isl_space_from_range(local_dim));
2210 dim = isl_space_wrap(dim);
2211 dim = isl_space_set_tuple_name(dim, isl_dim_set, "lifted");
2212
2213 return dim;
2214 }
2215
2216 isl_bool isl_space_can_zip(__isl_keep isl_space *dim)
2217 {
2218 if (!dim)
2219 return isl_bool_error;
2220
2221 return dim->nested[0] && dim->nested[1];
2222 }
2223
2224 __isl_give isl_space *isl_space_zip(__isl_take isl_space *dim)
2225 {
2226 isl_space *dom, *ran;
2227 isl_space *dom_dom, *dom_ran, *ran_dom, *ran_ran;
2228
2229 if (!isl_space_can_zip(dim))
2230 isl_die(dim->ctx, isl_error_invalid, "dim cannot be zipped",
2231 goto error);
2232
2233 if (!dim)
2234 return NULL;
2235 dom = isl_space_unwrap(isl_space_domain(isl_space_copy(dim)));
2236 ran = isl_space_unwrap(isl_space_range(dim));
2237 dom_dom = isl_space_domain(isl_space_copy(dom));
2238 dom_ran = isl_space_range(dom);
2239 ran_dom = isl_space_domain(isl_space_copy(ran));
2240 ran_ran = isl_space_range(ran);
2241 dom = isl_space_join(isl_space_from_domain(dom_dom),
2242 isl_space_from_range(ran_dom));
2243 ran = isl_space_join(isl_space_from_domain(dom_ran),
2244 isl_space_from_range(ran_ran));
2245 return isl_space_join(isl_space_from_domain(isl_space_wrap(dom)),
2246 isl_space_from_range(isl_space_wrap(ran)));
2247 error:
2248 isl_space_free(dim);
2249 return NULL;
2250 }
2251
2252 /* Can we apply isl_space_curry to "space"?
2253 * That is, does it have a nested relation in its domain?
2254 */
2255 isl_bool isl_space_can_curry(__isl_keep isl_space *space)
2256 {
2257 if (!space)
2258 return isl_bool_error;
2259
2260 return !!space->nested[0];
2261 }
2262
2263 /* Given a space (A -> B) -> C, return the corresponding space
2264 * A -> (B -> C).
2265 */
2266 __isl_give isl_space *isl_space_curry(__isl_take isl_space *space)
2267 {
2268 isl_space *dom, *ran;
2269 isl_space *dom_dom, *dom_ran;
2270
2271 if (!space)
2272 return NULL;
2273
2274 if (!isl_space_can_curry(space))
2275 isl_die(space->ctx, isl_error_invalid,
2276 "space cannot be curried", goto error);
2277
2278 dom = isl_space_unwrap(isl_space_domain(isl_space_copy(space)));
2279 ran = isl_space_range(space);
2280 dom_dom = isl_space_domain(isl_space_copy(dom));
2281 dom_ran = isl_space_range(dom);
2282 ran = isl_space_join(isl_space_from_domain(dom_ran),
2283 isl_space_from_range(ran));
2284 return isl_space_join(isl_space_from_domain(dom_dom),
2285 isl_space_from_range(isl_space_wrap(ran)));
2286 error:
2287 isl_space_free(space);
2288 return NULL;
2289 }
2290
2291 /* Can isl_space_range_curry be applied to "space"?
2292 * That is, does it have a nested relation in its range,
2293 * the domain of which is itself a nested relation?
2294 */
2295 isl_bool isl_space_can_range_curry(__isl_keep isl_space *space)
2296 {
2297 isl_bool can;
2298
2299 if (!space)
2300 return isl_bool_error;
2301 can = isl_space_range_is_wrapping(space);
2302 if (can < 0 || !can)
2303 return can;
2304 return isl_space_can_curry(space->nested[1]);
2305 }
2306
2307 /* Given a space A -> ((B -> C) -> D), return the corresponding space
2308 * A -> (B -> (C -> D)).
2309 */
2310 __isl_give isl_space *isl_space_range_curry(__isl_take isl_space *space)
2311 {
2312 if (!space)
2313 return NULL;
2314
2315 if (!isl_space_can_range_curry(space))
2316 isl_die(isl_space_get_ctx(space), isl_error_invalid,
2317 "space range cannot be curried",
2318 return isl_space_free(space));
2319
2320 space = isl_space_cow(space);
2321 if (!space)
2322 return NULL;
2323 space->nested[1] = isl_space_curry(space->nested[1]);
2324 if (!space->nested[1])
2325 return isl_space_free(space);
2326
2327 return space;
2328 }
2329
2330 /* Can we apply isl_space_uncurry to "space"?
2331 * That is, does it have a nested relation in its range?
2332 */
2333 isl_bool isl_space_can_uncurry(__isl_keep isl_space *space)
2334 {
2335 if (!space)
2336 return isl_bool_error;
2337
2338 return !!space->nested[1];
2339 }
2340
2341 /* Given a space A -> (B -> C), return the corresponding space
2342 * (A -> B) -> C.
2343 */
2344 __isl_give isl_space *isl_space_uncurry(__isl_take isl_space *space)
2345 {
2346 isl_space *dom, *ran;
2347 isl_space *ran_dom, *ran_ran;
2348
2349 if (!space)
2350 return NULL;
2351
2352 if (!isl_space_can_uncurry(space))
2353 isl_die(space->ctx, isl_error_invalid,
2354 "space cannot be uncurried",
2355 return isl_space_free(space));
2356
2357 dom = isl_space_domain(isl_space_copy(space));
2358 ran = isl_space_unwrap(isl_space_range(space));
2359 ran_dom = isl_space_domain(isl_space_copy(ran));
2360 ran_ran = isl_space_range(ran);
2361 dom = isl_space_join(isl_space_from_domain(dom),
2362 isl_space_from_range(ran_dom));
2363 return isl_space_join(isl_space_from_domain(isl_space_wrap(dom)),
2364 isl_space_from_range(ran_ran));
2365 }
2366
2367 int isl_space_has_named_params(__isl_keep isl_space *dim)
2368 {
2369 int i;
2370 unsigned off;
2371
2372 if (!dim)
2373 return -1;
2374 if (dim->nparam == 0)
2375 return 1;
2376 off = isl_space_offset(dim, isl_dim_param);
2377 if (off + dim->nparam > dim->n_id)
2378 return 0;
2379 for (i = 0; i < dim->nparam; ++i)
2380 if (!dim->ids[off + i])
2381 return 0;
2382 return 1;
2383 }
2384
2385 /* Align the initial parameters of dim1 to match the order in dim2.
2386 */
2387 __isl_give isl_space *isl_space_align_params(__isl_take isl_space *dim1,
2388 __isl_take isl_space *dim2)
2389 {
2390 isl_reordering *exp;
2391
2392 if (!isl_space_has_named_params(dim1) || !isl_space_has_named_params(dim2))
2393 isl_die(isl_space_get_ctx(dim1), isl_error_invalid,
2394 "parameter alignment requires named parameters",
2395 goto error);
2396
2397 dim2 = isl_space_params(dim2);
2398 exp = isl_parameter_alignment_reordering(dim1, dim2);
2399 exp = isl_reordering_extend_space(exp, dim1);
2400 isl_space_free(dim2);
2401 if (!exp)
2402 return NULL;
2403 dim1 = isl_space_copy(exp->dim);
2404 isl_reordering_free(exp);
2405 return dim1;
2406 error:
2407 isl_space_free(dim1);
2408 isl_space_free(dim2);
2409 return NULL;
2410 }
2411
2412 /* Given the space of set (domain), construct a space for a map
2413 * with as domain the given space and as range the range of "model".
2414 */
2415 __isl_give isl_space *isl_space_extend_domain_with_range(
2416 __isl_take isl_space *space, __isl_take isl_space *model)
2417 {
2418 if (!model)
2419 goto error;
2420
2421 space = isl_space_from_domain(space);
2422 space = isl_space_add_dims(space, isl_dim_out,
2423 isl_space_dim(model, isl_dim_out));
2424 if (isl_space_has_tuple_id(model, isl_dim_out))
2425 space = isl_space_set_tuple_id(space, isl_dim_out,
2426 isl_space_get_tuple_id(model, isl_dim_out));
2427 if (!space)
2428 goto error;
2429 if (model->nested[1]) {
2430 isl_space *nested = isl_space_copy(model->nested[1]);
2431 int n_nested, n_space;
2432 nested = isl_space_align_params(nested, isl_space_copy(space));
2433 n_nested = isl_space_dim(nested, isl_dim_param);
2434 n_space = isl_space_dim(space, isl_dim_param);
2435 if (n_nested > n_space)
2436 nested = isl_space_drop_dims(nested, isl_dim_param,
2437 n_space, n_nested - n_space);
2438 if (!nested)
2439 goto error;
2440 space->nested[1] = nested;
2441 }
2442 isl_space_free(model);
2443 return space;
2444 error:
2445 isl_space_free(model);
2446 isl_space_free(space);
2447 return NULL;
2448 }
2449
2450 /* Compare the "type" dimensions of two isl_spaces.
2451 *
2452 * The order is fairly arbitrary.
2453 */
2454 static int isl_space_cmp_type(__isl_keep isl_space *space1,
2455 __isl_keep isl_space *space2, enum isl_dim_type type)
2456 {
2457 int cmp;
2458 isl_space *nested1, *nested2;
2459
2460 if (isl_space_dim(space1, type) != isl_space_dim(space2, type))
2461 return isl_space_dim(space1, type) -
2462 isl_space_dim(space2, type);
2463
2464 cmp = isl_id_cmp(tuple_id(space1, type), tuple_id(space2, type));
2465 if (cmp != 0)
2466 return cmp;
2467
2468 nested1 = nested(space1, type);
2469 nested2 = nested(space2, type);
2470 if (!nested1 != !nested2)
2471 return !nested1 - !nested2;
2472
2473 if (nested1)
2474 return isl_space_cmp(nested1, nested2);
2475
2476 return 0;
2477 }
2478
2479 /* Compare two isl_spaces.
2480 *
2481 * The order is fairly arbitrary.
2482 */
2483 int isl_space_cmp(__isl_keep isl_space *space1, __isl_keep isl_space *space2)
2484 {
2485 int i;
2486 int cmp;
2487
2488 if (space1 == space2)
2489 return 0;
2490 if (!space1)
2491 return -1;
2492 if (!space2)
2493 return 1;
2494
2495 cmp = isl_space_cmp_type(space1, space2, isl_dim_param);
2496 if (cmp != 0)
2497 return cmp;
2498 cmp = isl_space_cmp_type(space1, space2, isl_dim_in);
2499 if (cmp != 0)
2500 return cmp;
2501 cmp = isl_space_cmp_type(space1, space2, isl_dim_out);
2502 if (cmp != 0)
2503 return cmp;
2504
2505 if (!space1->ids && !space2->ids)
2506 return 0;
2507
2508 for (i = 0; i < n(space1, isl_dim_param); ++i) {
2509 cmp = isl_id_cmp(get_id(space1, isl_dim_param, i),
2510 get_id(space2, isl_dim_param, i));
2511 if (cmp != 0)
2512 return cmp;
2513 }
2514
2515 return 0;
2516 }
Integer set library