add isl_map_flat_domain_product
[isl.git] / isl_space.c
blob6a9e9d74b514bcd91042bf205343ddc5bfb8def1
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
5 * Use of this software is governed by the GNU LGPLv2.1 license
7 * Written by Sven Verdoolaege, K.U.Leuven, Departement
8 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
9 * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
10 * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
13 #include <stdlib.h>
14 #include <isl_space_private.h>
15 #include <isl_id_private.h>
16 #include <isl_reordering.h>
18 isl_ctx *isl_space_get_ctx(__isl_keep isl_space *dim)
20 return dim ? dim->ctx : NULL;
23 __isl_give isl_space *isl_space_alloc(isl_ctx *ctx,
24 unsigned nparam, unsigned n_in, unsigned n_out)
26 isl_space *dim;
28 dim = isl_alloc_type(ctx, struct isl_space);
29 if (!dim)
30 return NULL;
32 dim->ctx = ctx;
33 isl_ctx_ref(ctx);
34 dim->ref = 1;
35 dim->nparam = nparam;
36 dim->n_in = n_in;
37 dim->n_out = n_out;
39 dim->tuple_id[0] = NULL;
40 dim->tuple_id[1] = NULL;
42 dim->nested[0] = NULL;
43 dim->nested[1] = NULL;
45 dim->n_id = 0;
46 dim->ids = NULL;
48 return dim;
51 __isl_give isl_space *isl_space_set_alloc(isl_ctx *ctx,
52 unsigned nparam, unsigned dim)
54 return isl_space_alloc(ctx, nparam, 0, dim);
57 static unsigned global_pos(__isl_keep isl_space *dim,
58 enum isl_dim_type type, unsigned pos)
60 struct isl_ctx *ctx = dim->ctx;
62 switch (type) {
63 case isl_dim_param:
64 isl_assert(ctx, pos < dim->nparam,
65 return isl_space_dim(dim, isl_dim_all));
66 return pos;
67 case isl_dim_in:
68 isl_assert(ctx, pos < dim->n_in,
69 return isl_space_dim(dim, isl_dim_all));
70 return pos + dim->nparam;
71 case isl_dim_out:
72 isl_assert(ctx, pos < dim->n_out,
73 return isl_space_dim(dim, isl_dim_all));
74 return pos + dim->nparam + dim->n_in;
75 default:
76 isl_assert(ctx, 0, return isl_space_dim(dim, isl_dim_all));
78 return isl_space_dim(dim, isl_dim_all);
81 /* Extend length of ids array to the total number of dimensions.
83 static __isl_give isl_space *extend_ids(__isl_take isl_space *dim)
85 isl_id **ids;
86 int i;
88 if (isl_space_dim(dim, isl_dim_all) <= dim->n_id)
89 return dim;
91 if (!dim->ids) {
92 dim->ids = isl_calloc_array(dim->ctx,
93 isl_id *, isl_space_dim(dim, isl_dim_all));
94 if (!dim->ids)
95 goto error;
96 } else {
97 ids = isl_realloc_array(dim->ctx, dim->ids,
98 isl_id *, isl_space_dim(dim, isl_dim_all));
99 if (!ids)
100 goto error;
101 dim->ids = ids;
102 for (i = dim->n_id; i < isl_space_dim(dim, isl_dim_all); ++i)
103 dim->ids[i] = NULL;
106 dim->n_id = isl_space_dim(dim, isl_dim_all);
108 return dim;
109 error:
110 isl_space_free(dim);
111 return NULL;
114 static __isl_give isl_space *set_id(__isl_take isl_space *dim,
115 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
117 dim = isl_space_cow(dim);
119 if (!dim)
120 goto error;
122 pos = global_pos(dim, type, pos);
123 if (pos == isl_space_dim(dim, isl_dim_all))
124 goto error;
126 if (pos >= dim->n_id) {
127 if (!id)
128 return dim;
129 dim = extend_ids(dim);
130 if (!dim)
131 goto error;
134 dim->ids[pos] = id;
136 return dim;
137 error:
138 isl_id_free(id);
139 isl_space_free(dim);
140 return NULL;
143 static __isl_keep isl_id *get_id(__isl_keep isl_space *dim,
144 enum isl_dim_type type, unsigned pos)
146 if (!dim)
147 return NULL;
149 pos = global_pos(dim, type, pos);
150 if (pos == isl_space_dim(dim, isl_dim_all))
151 return NULL;
152 if (pos >= dim->n_id)
153 return NULL;
154 return dim->ids[pos];
157 static unsigned offset(__isl_keep isl_space *dim, enum isl_dim_type type)
159 switch (type) {
160 case isl_dim_param: return 0;
161 case isl_dim_in: return dim->nparam;
162 case isl_dim_out: return dim->nparam + dim->n_in;
163 default: return 0;
167 static unsigned n(__isl_keep isl_space *dim, enum isl_dim_type type)
169 switch (type) {
170 case isl_dim_param: return dim->nparam;
171 case isl_dim_in: return dim->n_in;
172 case isl_dim_out: return dim->n_out;
173 case isl_dim_all: return dim->nparam + dim->n_in + dim->n_out;
174 default: return 0;
178 unsigned isl_space_dim(__isl_keep isl_space *dim, enum isl_dim_type type)
180 if (!dim)
181 return 0;
182 return n(dim, type);
185 unsigned isl_space_offset(__isl_keep isl_space *dim, enum isl_dim_type type)
187 if (!dim)
188 return 0;
189 return offset(dim, type);
192 static __isl_give isl_space *copy_ids(__isl_take isl_space *dst,
193 enum isl_dim_type dst_type, unsigned offset, __isl_keep isl_space *src,
194 enum isl_dim_type src_type)
196 int i;
197 isl_id *id;
199 if (!dst)
200 return NULL;
202 for (i = 0; i < n(src, src_type); ++i) {
203 id = get_id(src, src_type, i);
204 if (!id)
205 continue;
206 dst = set_id(dst, dst_type, offset + i, isl_id_copy(id));
207 if (!dst)
208 return NULL;
210 return dst;
213 __isl_take isl_space *isl_space_dup(__isl_keep isl_space *dim)
215 isl_space *dup;
216 if (!dim)
217 return NULL;
218 dup = isl_space_alloc(dim->ctx, dim->nparam, dim->n_in, dim->n_out);
219 if (dim->tuple_id[0] &&
220 !(dup->tuple_id[0] = isl_id_copy(dim->tuple_id[0])))
221 goto error;
222 if (dim->tuple_id[1] &&
223 !(dup->tuple_id[1] = isl_id_copy(dim->tuple_id[1])))
224 goto error;
225 if (dim->nested[0] && !(dup->nested[0] = isl_space_copy(dim->nested[0])))
226 goto error;
227 if (dim->nested[1] && !(dup->nested[1] = isl_space_copy(dim->nested[1])))
228 goto error;
229 if (!dim->ids)
230 return dup;
231 dup = copy_ids(dup, isl_dim_param, 0, dim, isl_dim_param);
232 dup = copy_ids(dup, isl_dim_in, 0, dim, isl_dim_in);
233 dup = copy_ids(dup, isl_dim_out, 0, dim, isl_dim_out);
234 return dup;
235 error:
236 isl_space_free(dup);
237 return NULL;
240 __isl_give isl_space *isl_space_cow(__isl_take isl_space *dim)
242 if (!dim)
243 return NULL;
245 if (dim->ref == 1)
246 return dim;
247 dim->ref--;
248 return isl_space_dup(dim);
251 __isl_give isl_space *isl_space_copy(__isl_keep isl_space *dim)
253 if (!dim)
254 return NULL;
256 dim->ref++;
257 return dim;
260 void isl_space_free(__isl_take isl_space *dim)
262 int i;
264 if (!dim)
265 return;
267 if (--dim->ref > 0)
268 return;
270 isl_id_free(dim->tuple_id[0]);
271 isl_id_free(dim->tuple_id[1]);
273 isl_space_free(dim->nested[0]);
274 isl_space_free(dim->nested[1]);
276 for (i = 0; i < dim->n_id; ++i)
277 isl_id_free(dim->ids[i]);
278 free(dim->ids);
279 isl_ctx_deref(dim->ctx);
281 free(dim);
284 static int name_ok(isl_ctx *ctx, const char *s)
286 char *p;
287 long dummy;
289 dummy = strtol(s, &p, 0);
290 if (p != s)
291 isl_die(ctx, isl_error_invalid, "name looks like a number",
292 return 0);
294 return 1;
297 int isl_space_has_tuple_id(__isl_keep isl_space *dim, enum isl_dim_type type)
299 if (!dim)
300 return -1;
301 if (type != isl_dim_in && type != isl_dim_out)
302 isl_die(dim->ctx, isl_error_invalid,
303 "only input, output and set tuples can have ids",
304 return -1);
305 return dim->tuple_id[type - isl_dim_in] != NULL;
308 __isl_give isl_id *isl_space_get_tuple_id(__isl_keep isl_space *dim,
309 enum isl_dim_type type)
311 if (!dim)
312 return NULL;
313 if (type != isl_dim_in && type != isl_dim_out)
314 isl_die(dim->ctx, isl_error_invalid,
315 "only input, output and set tuples can have ids",
316 return NULL);
317 if (!dim->tuple_id[type - isl_dim_in])
318 isl_die(dim->ctx, isl_error_invalid,
319 "tuple has no id", return NULL);
320 return isl_id_copy(dim->tuple_id[type - isl_dim_in]);
323 __isl_give isl_space *isl_space_set_tuple_id(__isl_take isl_space *dim,
324 enum isl_dim_type type, __isl_take isl_id *id)
326 dim = isl_space_cow(dim);
327 if (!dim || !id)
328 goto error;
329 if (type != isl_dim_in && type != isl_dim_out)
330 isl_die(dim->ctx, isl_error_invalid,
331 "only input, output and set tuples can have names",
332 goto error);
334 isl_id_free(dim->tuple_id[type - isl_dim_in]);
335 dim->tuple_id[type - isl_dim_in] = id;
337 return dim;
338 error:
339 isl_id_free(id);
340 isl_space_free(dim);
341 return NULL;
344 __isl_give isl_space *isl_space_reset_tuple_id(__isl_take isl_space *dim,
345 enum isl_dim_type type)
347 dim = isl_space_cow(dim);
348 if (!dim)
349 return NULL;
350 if (type != isl_dim_in && type != isl_dim_out)
351 isl_die(dim->ctx, isl_error_invalid,
352 "only input, output and set tuples can have names",
353 goto error);
355 isl_id_free(dim->tuple_id[type - isl_dim_in]);
356 dim->tuple_id[type - isl_dim_in] = NULL;
358 return dim;
359 error:
360 isl_space_free(dim);
361 return NULL;
364 __isl_give isl_space *isl_space_set_dim_id(__isl_take isl_space *dim,
365 enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
367 dim = isl_space_cow(dim);
368 if (!dim || !id)
369 goto error;
370 isl_id_free(get_id(dim, type, pos));
371 return set_id(dim, type, pos, id);
372 error:
373 isl_id_free(id);
374 isl_space_free(dim);
375 return NULL;
378 int isl_space_has_dim_id(__isl_keep isl_space *dim,
379 enum isl_dim_type type, unsigned pos)
381 if (!dim)
382 return -1;
383 return get_id(dim, type, pos) != NULL;
386 __isl_give isl_id *isl_space_get_dim_id(__isl_keep isl_space *dim,
387 enum isl_dim_type type, unsigned pos)
389 if (!dim)
390 return NULL;
391 if (!get_id(dim, type, pos))
392 isl_die(dim->ctx, isl_error_invalid,
393 "dim has no id", return NULL);
394 return isl_id_copy(get_id(dim, type, pos));
397 __isl_give isl_space *isl_space_set_tuple_name(__isl_take isl_space *dim,
398 enum isl_dim_type type, const char *s)
400 isl_id *id;
402 if (!dim)
403 return NULL;
405 if (!s)
406 return isl_space_reset_tuple_id(dim, type);
408 if (!name_ok(dim->ctx, s))
409 goto error;
411 id = isl_id_alloc(dim->ctx, s, NULL);
412 return isl_space_set_tuple_id(dim, type, id);
413 error:
414 isl_space_free(dim);
415 return NULL;
418 const char *isl_space_get_tuple_name(__isl_keep isl_space *dim,
419 enum isl_dim_type type)
421 isl_id *id;
422 if (!dim)
423 return NULL;
424 if (type != isl_dim_in && type != isl_dim_out)
425 return NULL;
426 id = dim->tuple_id[type - isl_dim_in];
427 return id ? id->name : NULL;
430 __isl_give isl_space *isl_space_set_dim_name(__isl_take isl_space *dim,
431 enum isl_dim_type type, unsigned pos,
432 const char *s)
434 isl_id *id;
436 if (!dim)
437 return NULL;
438 if (!name_ok(dim->ctx, s))
439 goto error;
440 id = isl_id_alloc(dim->ctx, s, NULL);
441 return isl_space_set_dim_id(dim, type, pos, id);
442 error:
443 isl_space_free(dim);
444 return NULL;
447 __isl_keep const char *isl_space_get_dim_name(__isl_keep isl_space *dim,
448 enum isl_dim_type type, unsigned pos)
450 isl_id *id = get_id(dim, type, pos);
451 return id ? id->name : NULL;
454 int isl_space_find_dim_by_id(__isl_keep isl_space *dim, enum isl_dim_type type,
455 __isl_keep isl_id *id)
457 int i;
458 int offset;
459 int n;
461 if (!dim || !id)
462 return -1;
464 offset = isl_space_offset(dim, type);
465 n = isl_space_dim(dim, type);
466 for (i = 0; i < n && offset + i < dim->n_id; ++i)
467 if (dim->ids[offset + i] == id)
468 return i;
470 return -1;
473 static __isl_keep isl_id *tuple_id(__isl_keep isl_space *dim,
474 enum isl_dim_type type)
476 if (!dim)
477 return NULL;
478 if (type == isl_dim_in)
479 return dim->tuple_id[0];
480 if (type == isl_dim_out)
481 return dim->tuple_id[1];
482 return NULL;
485 static __isl_keep isl_space *nested(__isl_keep isl_space *dim,
486 enum isl_dim_type type)
488 if (!dim)
489 return NULL;
490 if (type == isl_dim_in)
491 return dim->nested[0];
492 if (type == isl_dim_out)
493 return dim->nested[1];
494 return NULL;
497 int isl_space_tuple_match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
498 __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
500 isl_id *id1, *id2;
501 isl_space *nested1, *nested2;
503 if (!dim1 || !dim2)
504 return -1;
506 if (n(dim1, dim1_type) != n(dim2, dim2_type))
507 return 0;
508 id1 = tuple_id(dim1, dim1_type);
509 id2 = tuple_id(dim2, dim2_type);
510 if (!id1 ^ !id2)
511 return 0;
512 if (id1 && id1 != id2)
513 return 0;
514 nested1 = nested(dim1, dim1_type);
515 nested2 = nested(dim2, dim2_type);
516 if (!nested1 ^ !nested2)
517 return 0;
518 if (nested1 && !isl_space_is_equal(nested1, nested2))
519 return 0;
520 return 1;
523 static int match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
524 __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
526 int i;
528 if (!isl_space_tuple_match(dim1, dim1_type, dim2, dim2_type))
529 return 0;
531 if (!dim1->ids && !dim2->ids)
532 return 1;
534 for (i = 0; i < n(dim1, dim1_type); ++i) {
535 if (get_id(dim1, dim1_type, i) != get_id(dim2, dim2_type, i))
536 return 0;
538 return 1;
541 int isl_space_match(__isl_keep isl_space *dim1, enum isl_dim_type dim1_type,
542 __isl_keep isl_space *dim2, enum isl_dim_type dim2_type)
544 return match(dim1, dim1_type, dim2, dim2_type);
547 static void get_ids(__isl_keep isl_space *dim, enum isl_dim_type type,
548 unsigned first, unsigned n, __isl_keep isl_id **ids)
550 int i;
552 for (i = 0; i < n ; ++i)
553 ids[i] = get_id(dim, type, first + i);
556 __isl_give isl_space *isl_space_extend(__isl_take isl_space *dim,
557 unsigned nparam, unsigned n_in, unsigned n_out)
559 isl_id **ids = NULL;
561 if (!dim)
562 return NULL;
563 if (dim->nparam == nparam && dim->n_in == n_in && dim->n_out == n_out)
564 return dim;
566 isl_assert(dim->ctx, dim->nparam <= nparam, goto error);
567 isl_assert(dim->ctx, dim->n_in <= n_in, goto error);
568 isl_assert(dim->ctx, dim->n_out <= n_out, goto error);
570 dim = isl_space_cow(dim);
572 if (dim->ids) {
573 ids = isl_calloc_array(dim->ctx, isl_id *,
574 nparam + n_in + n_out);
575 if (!ids)
576 goto error;
577 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
578 get_ids(dim, isl_dim_in, 0, dim->n_in, ids + nparam);
579 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + nparam + n_in);
580 free(dim->ids);
581 dim->ids = ids;
582 dim->n_id = nparam + n_in + n_out;
584 dim->nparam = nparam;
585 dim->n_in = n_in;
586 dim->n_out = n_out;
588 return dim;
589 error:
590 free(ids);
591 isl_space_free(dim);
592 return NULL;
595 __isl_give isl_space *isl_space_add_dims(__isl_take isl_space *dim,
596 enum isl_dim_type type, unsigned n)
598 if (!dim)
599 return NULL;
600 dim = isl_space_reset(dim, type);
601 switch (type) {
602 case isl_dim_param:
603 dim = isl_space_extend(dim,
604 dim->nparam + n, dim->n_in, dim->n_out);
605 if (dim && dim->nested[0] &&
606 !(dim->nested[0] = isl_space_add_dims(dim->nested[0],
607 isl_dim_param, n)))
608 goto error;
609 if (dim && dim->nested[1] &&
610 !(dim->nested[1] = isl_space_add_dims(dim->nested[1],
611 isl_dim_param, n)))
612 goto error;
613 return dim;
614 case isl_dim_in:
615 return isl_space_extend(dim,
616 dim->nparam, dim->n_in + n, dim->n_out);
617 case isl_dim_out:
618 return isl_space_extend(dim,
619 dim->nparam, dim->n_in, dim->n_out + n);
620 default:
621 isl_die(dim->ctx, isl_error_invalid,
622 "cannot add dimensions of specified type", goto error);
624 error:
625 isl_space_free(dim);
626 return NULL;
629 static int valid_dim_type(enum isl_dim_type type)
631 switch (type) {
632 case isl_dim_param:
633 case isl_dim_in:
634 case isl_dim_out:
635 return 1;
636 default:
637 return 0;
641 __isl_give isl_space *isl_space_insert_dims(__isl_take isl_space *dim,
642 enum isl_dim_type type, unsigned pos, unsigned n)
644 isl_id **ids = NULL;
646 if (!dim)
647 return NULL;
648 if (n == 0)
649 return isl_space_reset(dim, type);
651 if (!valid_dim_type(type))
652 isl_die(dim->ctx, isl_error_invalid,
653 "cannot insert dimensions of specified type",
654 goto error);
656 isl_assert(dim->ctx, pos <= isl_space_dim(dim, type), goto error);
658 dim = isl_space_cow(dim);
659 if (!dim)
660 return NULL;
662 if (dim->ids) {
663 enum isl_dim_type t;
664 int off;
665 int s[3];
666 int *size = s - isl_dim_param;
667 ids = isl_calloc_array(dim->ctx, isl_id *,
668 dim->nparam + dim->n_in + dim->n_out + n);
669 if (!ids)
670 goto error;
671 off = 0;
672 size[isl_dim_param] = dim->nparam;
673 size[isl_dim_in] = dim->n_in;
674 size[isl_dim_out] = dim->n_out;
675 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
676 if (t != type) {
677 get_ids(dim, t, 0, size[t], ids + off);
678 off += size[t];
679 } else {
680 get_ids(dim, t, 0, pos, ids + off);
681 off += pos + n;
682 get_ids(dim, t, pos, size[t] - pos, ids + off);
683 off += size[t] - pos;
686 free(dim->ids);
687 dim->ids = ids;
688 dim->n_id = dim->nparam + dim->n_in + dim->n_out + n;
690 switch (type) {
691 case isl_dim_param: dim->nparam += n; break;
692 case isl_dim_in: dim->n_in += n; break;
693 case isl_dim_out: dim->n_out += n; break;
694 default: ;
696 dim = isl_space_reset(dim, type);
698 return dim;
699 error:
700 isl_space_free(dim);
701 return NULL;
704 __isl_give isl_space *isl_space_move_dims(__isl_take isl_space *dim,
705 enum isl_dim_type dst_type, unsigned dst_pos,
706 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
708 int i;
710 if (!dim)
711 return NULL;
712 if (n == 0)
713 return dim;
715 isl_assert(dim->ctx, src_pos + n <= isl_space_dim(dim, src_type),
716 goto error);
718 if (dst_type == src_type && dst_pos == src_pos)
719 return dim;
721 isl_assert(dim->ctx, dst_type != src_type, goto error);
723 dim = isl_space_reset(dim, src_type);
724 dim = isl_space_reset(dim, dst_type);
726 dim = isl_space_cow(dim);
727 if (!dim)
728 return NULL;
730 if (dim->ids) {
731 isl_id **ids;
732 enum isl_dim_type t;
733 int off;
734 int s[3];
735 int *size = s - isl_dim_param;
736 ids = isl_calloc_array(dim->ctx, isl_id *,
737 dim->nparam + dim->n_in + dim->n_out);
738 if (!ids)
739 goto error;
740 off = 0;
741 size[isl_dim_param] = dim->nparam;
742 size[isl_dim_in] = dim->n_in;
743 size[isl_dim_out] = dim->n_out;
744 for (t = isl_dim_param; t <= isl_dim_out; ++t) {
745 if (t == dst_type) {
746 get_ids(dim, t, 0, dst_pos, ids + off);
747 off += dst_pos;
748 get_ids(dim, src_type, src_pos, n, ids + off);
749 off += n;
750 get_ids(dim, t, dst_pos, size[t] - dst_pos,
751 ids + off);
752 off += size[t] - dst_pos;
753 } else if (t == src_type) {
754 get_ids(dim, t, 0, src_pos, ids + off);
755 off += src_pos;
756 get_ids(dim, t, src_pos + n,
757 size[t] - src_pos - n, ids + off);
758 off += size[t] - src_pos - n;
759 } else {
760 get_ids(dim, t, 0, size[t], ids + off);
761 off += size[t];
764 free(dim->ids);
765 dim->ids = ids;
766 dim->n_id = dim->nparam + dim->n_in + dim->n_out;
769 switch (dst_type) {
770 case isl_dim_param: dim->nparam += n; break;
771 case isl_dim_in: dim->n_in += n; break;
772 case isl_dim_out: dim->n_out += n; break;
773 default: ;
776 switch (src_type) {
777 case isl_dim_param: dim->nparam -= n; break;
778 case isl_dim_in: dim->n_in -= n; break;
779 case isl_dim_out: dim->n_out -= n; break;
780 default: ;
783 if (dst_type != isl_dim_param && src_type != isl_dim_param)
784 return dim;
786 for (i = 0; i < 2; ++i) {
787 if (!dim->nested[i])
788 continue;
789 dim->nested[i] = isl_space_replace(dim->nested[i],
790 isl_dim_param, dim);
791 if (!dim->nested[i])
792 goto error;
795 return dim;
796 error:
797 isl_space_free(dim);
798 return NULL;
801 __isl_give isl_space *isl_space_join(__isl_take isl_space *left,
802 __isl_take isl_space *right)
804 isl_space *dim;
806 if (!left || !right)
807 goto error;
809 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
810 goto error);
811 isl_assert(left->ctx,
812 isl_space_tuple_match(left, isl_dim_out, right, isl_dim_in),
813 goto error);
815 dim = isl_space_alloc(left->ctx, left->nparam, left->n_in, right->n_out);
816 if (!dim)
817 goto error;
819 dim = copy_ids(dim, isl_dim_param, 0, left, isl_dim_param);
820 dim = copy_ids(dim, isl_dim_in, 0, left, isl_dim_in);
821 dim = copy_ids(dim, isl_dim_out, 0, right, isl_dim_out);
823 if (dim && left->tuple_id[0] &&
824 !(dim->tuple_id[0] = isl_id_copy(left->tuple_id[0])))
825 goto error;
826 if (dim && right->tuple_id[1] &&
827 !(dim->tuple_id[1] = isl_id_copy(right->tuple_id[1])))
828 goto error;
829 if (dim && left->nested[0] &&
830 !(dim->nested[0] = isl_space_copy(left->nested[0])))
831 goto error;
832 if (dim && right->nested[1] &&
833 !(dim->nested[1] = isl_space_copy(right->nested[1])))
834 goto error;
836 isl_space_free(left);
837 isl_space_free(right);
839 return dim;
840 error:
841 isl_space_free(left);
842 isl_space_free(right);
843 return NULL;
846 __isl_give isl_space *isl_space_product(__isl_take isl_space *left,
847 __isl_take isl_space *right)
849 isl_space *dom1, *dom2, *nest1, *nest2;
851 if (!left || !right)
852 goto error;
854 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
855 goto error);
857 dom1 = isl_space_domain(isl_space_copy(left));
858 dom2 = isl_space_domain(isl_space_copy(right));
859 nest1 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
861 dom1 = isl_space_range(left);
862 dom2 = isl_space_range(right);
863 nest2 = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
865 return isl_space_join(isl_space_reverse(nest1), nest2);
866 error:
867 isl_space_free(left);
868 isl_space_free(right);
869 return NULL;
872 /* Given two spaces { A -> C } and { B -> C }, construct the space
873 * { [A -> B] -> C }
875 __isl_give isl_space *isl_space_domain_product(__isl_take isl_space *left,
876 __isl_take isl_space *right)
878 isl_space *ran, *dom1, *dom2, *nest;
880 if (!left || !right)
881 goto error;
883 if (!match(left, isl_dim_param, right, isl_dim_param))
884 isl_die(left->ctx, isl_error_invalid,
885 "parameters need to match", goto error);
886 if (!isl_space_tuple_match(left, isl_dim_out, right, isl_dim_out))
887 isl_die(left->ctx, isl_error_invalid,
888 "ranges need to match", goto error);
890 ran = isl_space_range(isl_space_copy(left));
892 dom1 = isl_space_domain(left);
893 dom2 = isl_space_domain(right);
894 nest = isl_space_wrap(isl_space_join(isl_space_reverse(dom1), dom2));
896 return isl_space_join(isl_space_reverse(nest), ran);
897 error:
898 isl_space_free(left);
899 isl_space_free(right);
900 return NULL;
903 __isl_give isl_space *isl_space_range_product(__isl_take isl_space *left,
904 __isl_take isl_space *right)
906 isl_space *dom, *ran1, *ran2, *nest;
908 if (!left || !right)
909 goto error;
911 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
912 goto error);
913 if (!isl_space_tuple_match(left, isl_dim_in, right, isl_dim_in))
914 isl_die(left->ctx, isl_error_invalid,
915 "domains need to match", goto error);
917 dom = isl_space_domain(isl_space_copy(left));
919 ran1 = isl_space_range(left);
920 ran2 = isl_space_range(right);
921 nest = isl_space_wrap(isl_space_join(isl_space_reverse(ran1), ran2));
923 return isl_space_join(isl_space_reverse(dom), nest);
924 error:
925 isl_space_free(left);
926 isl_space_free(right);
927 return NULL;
930 __isl_give isl_space *isl_space_map_from_set(__isl_take isl_space *dim)
932 isl_id **ids = NULL;
934 if (!dim)
935 return NULL;
936 isl_assert(dim->ctx, dim->n_in == 0, goto error);
937 if (dim->n_out == 0 && !isl_space_is_named_or_nested(dim, isl_dim_out))
938 return dim;
939 dim = isl_space_cow(dim);
940 if (!dim)
941 return NULL;
942 if (dim->ids) {
943 ids = isl_calloc_array(dim->ctx, isl_id *,
944 dim->nparam + dim->n_out + dim->n_out);
945 if (!ids)
946 goto error;
947 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
948 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->nparam);
950 dim->n_in = dim->n_out;
951 if (ids) {
952 free(dim->ids);
953 dim->ids = ids;
954 dim->n_id = dim->nparam + dim->n_out + dim->n_out;
955 dim = copy_ids(dim, isl_dim_out, 0, dim, isl_dim_in);
957 isl_id_free(dim->tuple_id[0]);
958 dim->tuple_id[0] = isl_id_copy(dim->tuple_id[1]);
959 isl_space_free(dim->nested[0]);
960 dim->nested[0] = isl_space_copy(dim->nested[1]);
961 return dim;
962 error:
963 isl_space_free(dim);
964 return NULL;
967 static __isl_give isl_space *set_ids(__isl_take isl_space *dim,
968 enum isl_dim_type type,
969 unsigned first, unsigned n, __isl_take isl_id **ids)
971 int i;
973 for (i = 0; i < n ; ++i)
974 dim = set_id(dim, type, first + i, ids[i]);
976 return dim;
979 __isl_give isl_space *isl_space_reverse(__isl_take isl_space *dim)
981 unsigned t;
982 isl_space *nested;
983 isl_id **ids = NULL;
984 isl_id *id;
986 if (!dim)
987 return NULL;
988 if (match(dim, isl_dim_in, dim, isl_dim_out))
989 return dim;
991 dim = isl_space_cow(dim);
992 if (!dim)
993 return NULL;
995 id = dim->tuple_id[0];
996 dim->tuple_id[0] = dim->tuple_id[1];
997 dim->tuple_id[1] = id;
999 nested = dim->nested[0];
1000 dim->nested[0] = dim->nested[1];
1001 dim->nested[1] = nested;
1003 if (dim->ids) {
1004 ids = isl_alloc_array(dim->ctx, isl_id *,
1005 dim->n_in + dim->n_out);
1006 if (!ids)
1007 goto error;
1008 get_ids(dim, isl_dim_in, 0, dim->n_in, ids);
1009 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->n_in);
1012 t = dim->n_in;
1013 dim->n_in = dim->n_out;
1014 dim->n_out = t;
1016 if (dim->ids) {
1017 dim = set_ids(dim, isl_dim_out, 0, dim->n_out, ids);
1018 dim = set_ids(dim, isl_dim_in, 0, dim->n_in, ids + dim->n_out);
1019 free(ids);
1022 return dim;
1023 error:
1024 free(ids);
1025 isl_space_free(dim);
1026 return NULL;
1029 __isl_give isl_space *isl_space_drop_dims(__isl_take isl_space *dim,
1030 enum isl_dim_type type, unsigned first, unsigned num)
1032 int i;
1034 if (!dim)
1035 return NULL;
1037 if (num == 0)
1038 return isl_space_reset(dim, type);
1040 if (!valid_dim_type(type))
1041 isl_die(dim->ctx, isl_error_invalid,
1042 "cannot drop dimensions of specified type", goto error);
1044 isl_assert(dim->ctx, first + num <= n(dim, type), goto error);
1045 dim = isl_space_cow(dim);
1046 if (!dim)
1047 goto error;
1048 if (dim->ids) {
1049 dim = extend_ids(dim);
1050 if (!dim)
1051 goto error;
1052 for (i = 0; i < num; ++i)
1053 isl_id_free(get_id(dim, type, first + i));
1054 for (i = first+num; i < n(dim, type); ++i)
1055 set_id(dim, type, i - num, get_id(dim, type, i));
1056 switch (type) {
1057 case isl_dim_param:
1058 get_ids(dim, isl_dim_in, 0, dim->n_in,
1059 dim->ids + offset(dim, isl_dim_in) - num);
1060 case isl_dim_in:
1061 get_ids(dim, isl_dim_out, 0, dim->n_out,
1062 dim->ids + offset(dim, isl_dim_out) - num);
1063 default:
1066 dim->n_id -= num;
1068 switch (type) {
1069 case isl_dim_param: dim->nparam -= num; break;
1070 case isl_dim_in: dim->n_in -= num; break;
1071 case isl_dim_out: dim->n_out -= num; break;
1072 default: ;
1074 dim = isl_space_reset(dim, type);
1075 if (type == isl_dim_param) {
1076 if (dim && dim->nested[0] &&
1077 !(dim->nested[0] = isl_space_drop_dims(dim->nested[0],
1078 isl_dim_param, first, num)))
1079 goto error;
1080 if (dim && dim->nested[1] &&
1081 !(dim->nested[1] = isl_space_drop_dims(dim->nested[1],
1082 isl_dim_param, first, num)))
1083 goto error;
1085 return dim;
1086 error:
1087 isl_space_free(dim);
1088 return NULL;
1091 __isl_give isl_space *isl_space_drop_inputs(__isl_take isl_space *dim,
1092 unsigned first, unsigned n)
1094 if (!dim)
1095 return NULL;
1096 return isl_space_drop_dims(dim, isl_dim_in, first, n);
1099 __isl_give isl_space *isl_space_drop_outputs(__isl_take isl_space *dim,
1100 unsigned first, unsigned n)
1102 if (!dim)
1103 return NULL;
1104 return isl_space_drop_dims(dim, isl_dim_out, first, n);
1107 __isl_give isl_space *isl_space_domain(__isl_take isl_space *dim)
1109 if (!dim)
1110 return NULL;
1111 dim = isl_space_drop_outputs(dim, 0, dim->n_out);
1112 return isl_space_reverse(dim);
1115 __isl_give isl_space *isl_space_from_domain(__isl_take isl_space *dim)
1117 return isl_space_reverse(dim);
1120 __isl_give isl_space *isl_space_range(__isl_take isl_space *dim)
1122 if (!dim)
1123 return NULL;
1124 return isl_space_drop_inputs(dim, 0, dim->n_in);
1127 __isl_give isl_space *isl_space_from_range(__isl_take isl_space *dim)
1129 return dim;
1132 __isl_give isl_space *isl_space_params(__isl_take isl_space *space)
1134 space = isl_space_drop_dims(space,
1135 isl_dim_in, 0, isl_space_dim(space, isl_dim_in));
1136 space = isl_space_drop_dims(space,
1137 isl_dim_out, 0, isl_space_dim(space, isl_dim_out));
1138 return space;
1141 __isl_give isl_space *isl_space_as_set_space(__isl_take isl_space *dim)
1143 dim = isl_space_cow(dim);
1144 if (!dim)
1145 return NULL;
1147 dim->n_out += dim->n_in;
1148 dim->n_in = 0;
1149 dim = isl_space_reset(dim, isl_dim_in);
1150 dim = isl_space_reset(dim, isl_dim_out);
1152 return dim;
1155 __isl_give isl_space *isl_space_underlying(__isl_take isl_space *dim,
1156 unsigned n_div)
1158 int i;
1160 if (!dim)
1161 return NULL;
1162 if (n_div == 0 &&
1163 dim->nparam == 0 && dim->n_in == 0 && dim->n_id == 0)
1164 return isl_space_reset(isl_space_reset(dim, isl_dim_in), isl_dim_out);
1165 dim = isl_space_cow(dim);
1166 if (!dim)
1167 return NULL;
1168 dim->n_out += dim->nparam + dim->n_in + n_div;
1169 dim->nparam = 0;
1170 dim->n_in = 0;
1172 for (i = 0; i < dim->n_id; ++i)
1173 isl_id_free(get_id(dim, isl_dim_out, i));
1174 dim->n_id = 0;
1175 dim = isl_space_reset(dim, isl_dim_in);
1176 dim = isl_space_reset(dim, isl_dim_out);
1178 return dim;
1181 int isl_space_is_equal(__isl_keep isl_space *dim1, __isl_keep isl_space *dim2)
1183 if (!dim1 || !dim2)
1184 return -1;
1185 return match(dim1, isl_dim_param, dim2, isl_dim_param) &&
1186 isl_space_tuple_match(dim1, isl_dim_in, dim2, isl_dim_in) &&
1187 isl_space_tuple_match(dim1, isl_dim_out, dim2, isl_dim_out);
1190 int isl_space_compatible(__isl_keep isl_space *dim1,
1191 __isl_keep isl_space *dim2)
1193 return dim1->nparam == dim2->nparam &&
1194 dim1->n_in + dim1->n_out == dim2->n_in + dim2->n_out;
1197 static uint32_t isl_hash_dim(uint32_t hash, __isl_keep isl_space *dim)
1199 int i;
1200 isl_id *id;
1202 if (!dim)
1203 return hash;
1205 hash = isl_hash_builtin(hash, dim->nparam);
1206 hash = isl_hash_builtin(hash, dim->n_in);
1207 hash = isl_hash_builtin(hash, dim->n_out);
1209 for (i = 0; i < dim->nparam; ++i) {
1210 id = get_id(dim, isl_dim_param, i);
1211 hash = isl_hash_id(hash, id);
1214 id = tuple_id(dim, isl_dim_in);
1215 hash = isl_hash_id(hash, id);
1216 id = tuple_id(dim, isl_dim_out);
1217 hash = isl_hash_id(hash, id);
1219 hash = isl_hash_dim(hash, dim->nested[0]);
1220 hash = isl_hash_dim(hash, dim->nested[1]);
1222 return hash;
1225 uint32_t isl_space_get_hash(__isl_keep isl_space *dim)
1227 uint32_t hash;
1229 if (!dim)
1230 return 0;
1232 hash = isl_hash_init();
1233 hash = isl_hash_dim(hash, dim);
1235 return hash;
1238 int isl_space_is_wrapping(__isl_keep isl_space *dim)
1240 if (!dim)
1241 return -1;
1243 if (dim->n_in != 0 || dim->tuple_id[0] || dim->nested[0])
1244 return 0;
1246 return dim->nested[1] != NULL;
1249 __isl_give isl_space *isl_space_wrap(__isl_take isl_space *dim)
1251 isl_space *wrap;
1253 if (!dim)
1254 return NULL;
1256 wrap = isl_space_alloc(dim->ctx, dim->nparam, 0, dim->n_in + dim->n_out);
1258 wrap = copy_ids(wrap, isl_dim_param, 0, dim, isl_dim_param);
1259 wrap = copy_ids(wrap, isl_dim_set, 0, dim, isl_dim_in);
1260 wrap = copy_ids(wrap, isl_dim_set, dim->n_in, dim, isl_dim_out);
1262 if (!wrap)
1263 goto error;
1265 wrap->nested[1] = dim;
1267 return wrap;
1268 error:
1269 isl_space_free(dim);
1270 return NULL;
1273 __isl_give isl_space *isl_space_unwrap(__isl_take isl_space *dim)
1275 isl_space *unwrap;
1277 if (!dim)
1278 return NULL;
1280 if (!isl_space_is_wrapping(dim))
1281 isl_die(dim->ctx, isl_error_invalid, "not a wrapping dim",
1282 goto error);
1284 unwrap = isl_space_copy(dim->nested[1]);
1285 isl_space_free(dim);
1287 return unwrap;
1288 error:
1289 isl_space_free(dim);
1290 return NULL;
1293 int isl_space_is_named_or_nested(__isl_keep isl_space *dim, enum isl_dim_type type)
1295 if (type != isl_dim_in && type != isl_dim_out)
1296 return 0;
1297 if (!dim)
1298 return -1;
1299 if (dim->tuple_id[type - isl_dim_in])
1300 return 1;
1301 if (dim->nested[type - isl_dim_in])
1302 return 1;
1303 return 0;
1306 int isl_space_may_be_set(__isl_keep isl_space *dim)
1308 if (!dim)
1309 return -1;
1310 if (isl_space_dim(dim, isl_dim_in) != 0)
1311 return 0;
1312 if (isl_space_is_named_or_nested(dim, isl_dim_in))
1313 return 0;
1314 return 1;
1317 __isl_give isl_space *isl_space_reset(__isl_take isl_space *dim,
1318 enum isl_dim_type type)
1320 if (!isl_space_is_named_or_nested(dim, type))
1321 return dim;
1323 dim = isl_space_cow(dim);
1324 if (!dim)
1325 return NULL;
1327 isl_id_free(dim->tuple_id[type - isl_dim_in]);
1328 dim->tuple_id[type - isl_dim_in] = NULL;
1329 isl_space_free(dim->nested[type - isl_dim_in]);
1330 dim->nested[type - isl_dim_in] = NULL;
1332 return dim;
1335 __isl_give isl_space *isl_space_flatten(__isl_take isl_space *dim)
1337 if (!dim)
1338 return NULL;
1339 if (!dim->nested[0] && !dim->nested[1])
1340 return dim;
1342 if (dim->nested[0])
1343 dim = isl_space_reset(dim, isl_dim_in);
1344 if (dim && dim->nested[1])
1345 dim = isl_space_reset(dim, isl_dim_out);
1347 return dim;
1350 __isl_give isl_space *isl_space_flatten_domain(__isl_take isl_space *dim)
1352 if (!dim)
1353 return NULL;
1354 if (!dim->nested[0])
1355 return dim;
1357 return isl_space_reset(dim, isl_dim_in);
1360 __isl_give isl_space *isl_space_flatten_range(__isl_take isl_space *dim)
1362 if (!dim)
1363 return NULL;
1364 if (!dim->nested[1])
1365 return dim;
1367 return isl_space_reset(dim, isl_dim_out);
1370 /* Replace the dimensions of the given type of dst by those of src.
1372 __isl_give isl_space *isl_space_replace(__isl_take isl_space *dst,
1373 enum isl_dim_type type, __isl_keep isl_space *src)
1375 dst = isl_space_cow(dst);
1377 if (!dst || !src)
1378 goto error;
1380 dst = isl_space_drop_dims(dst, type, 0, isl_space_dim(dst, type));
1381 dst = isl_space_add_dims(dst, type, isl_space_dim(src, type));
1382 dst = copy_ids(dst, type, 0, src, type);
1384 if (dst && type == isl_dim_param) {
1385 int i;
1386 for (i = 0; i <= 1; ++i) {
1387 if (!dst->nested[i])
1388 continue;
1389 dst->nested[i] = isl_space_replace(dst->nested[i],
1390 type, src);
1391 if (!dst->nested[i])
1392 goto error;
1396 return dst;
1397 error:
1398 isl_space_free(dst);
1399 return NULL;
1402 /* Given a dimension specification "dim" of a set, create a dimension
1403 * specification for the lift of the set. In particular, the result
1404 * is of the form [dim -> local[..]], with n_local variables in the
1405 * range of the wrapped map.
1407 __isl_give isl_space *isl_space_lift(__isl_take isl_space *dim, unsigned n_local)
1409 isl_space *local_dim;
1411 if (!dim)
1412 return NULL;
1414 local_dim = isl_space_dup(dim);
1415 local_dim = isl_space_drop_dims(local_dim, isl_dim_set, 0, dim->n_out);
1416 local_dim = isl_space_add_dims(local_dim, isl_dim_set, n_local);
1417 local_dim = isl_space_set_tuple_name(local_dim, isl_dim_set, "local");
1418 dim = isl_space_join(isl_space_from_domain(dim),
1419 isl_space_from_range(local_dim));
1420 dim = isl_space_wrap(dim);
1421 dim = isl_space_set_tuple_name(dim, isl_dim_set, "lifted");
1423 return dim;
1426 int isl_space_can_zip(__isl_keep isl_space *dim)
1428 if (!dim)
1429 return -1;
1431 return dim->nested[0] && dim->nested[1];
1434 __isl_give isl_space *isl_space_zip(__isl_take isl_space *dim)
1436 isl_space *dom, *ran;
1437 isl_space *dom_dom, *dom_ran, *ran_dom, *ran_ran;
1439 if (!isl_space_can_zip(dim))
1440 isl_die(dim->ctx, isl_error_invalid, "dim cannot be zipped",
1441 goto error);
1443 if (!dim)
1444 return 0;
1445 dom = isl_space_unwrap(isl_space_domain(isl_space_copy(dim)));
1446 ran = isl_space_unwrap(isl_space_range(dim));
1447 dom_dom = isl_space_domain(isl_space_copy(dom));
1448 dom_ran = isl_space_range(dom);
1449 ran_dom = isl_space_domain(isl_space_copy(ran));
1450 ran_ran = isl_space_range(ran);
1451 dom = isl_space_join(isl_space_from_domain(dom_dom),
1452 isl_space_from_range(ran_dom));
1453 ran = isl_space_join(isl_space_from_domain(dom_ran),
1454 isl_space_from_range(ran_ran));
1455 return isl_space_join(isl_space_from_domain(isl_space_wrap(dom)),
1456 isl_space_from_range(isl_space_wrap(ran)));
1457 error:
1458 isl_space_free(dim);
1459 return NULL;
1462 int isl_space_has_named_params(__isl_keep isl_space *dim)
1464 int i;
1465 unsigned off;
1467 if (!dim)
1468 return -1;
1469 if (dim->nparam == 0)
1470 return 1;
1471 off = isl_space_offset(dim, isl_dim_param);
1472 if (off + dim->nparam > dim->n_id)
1473 return 0;
1474 for (i = 0; i < dim->nparam; ++i)
1475 if (!dim->ids[off + i])
1476 return 0;
1477 return 1;
1480 /* Align the initial parameters of dim1 to match the order in dim2.
1482 __isl_give isl_space *isl_space_align_params(__isl_take isl_space *dim1,
1483 __isl_take isl_space *dim2)
1485 isl_reordering *exp;
1487 if (!isl_space_has_named_params(dim1) || !isl_space_has_named_params(dim2))
1488 isl_die(isl_space_get_ctx(dim1), isl_error_invalid,
1489 "parameter alignment requires named parameters",
1490 goto error);
1492 dim2 = isl_space_params(dim2);
1493 exp = isl_parameter_alignment_reordering(dim1, dim2);
1494 exp = isl_reordering_extend_space(exp, dim1);
1495 isl_space_free(dim2);
1496 if (!exp)
1497 return NULL;
1498 dim1 = isl_space_copy(exp->dim);
1499 isl_reordering_free(exp);
1500 return dim1;
1501 error:
1502 isl_space_free(dim1);
1503 isl_space_free(dim2);
1504 return NULL;