isl_access_info_compute_flow: align parameters of input access relations
[isl.git] / isl_space.c
blobb6e6b5a17d77c8b51aa0c6771798f27b66e34ce0
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 __isl_give isl_space *isl_space_range_product(__isl_take isl_space *left,
873 __isl_take isl_space *right)
875 isl_space *dom, *ran1, *ran2, *nest;
877 if (!left || !right)
878 goto error;
880 isl_assert(left->ctx, match(left, isl_dim_param, right, isl_dim_param),
881 goto error);
882 if (!isl_space_tuple_match(left, isl_dim_in, right, isl_dim_in))
883 isl_die(left->ctx, isl_error_invalid,
884 "domains need to match", goto error);
886 dom = isl_space_domain(isl_space_copy(left));
888 ran1 = isl_space_range(left);
889 ran2 = isl_space_range(right);
890 nest = isl_space_wrap(isl_space_join(isl_space_reverse(ran1), ran2));
892 return isl_space_join(isl_space_reverse(dom), nest);
893 error:
894 isl_space_free(left);
895 isl_space_free(right);
896 return NULL;
899 __isl_give isl_space *isl_space_map_from_set(__isl_take isl_space *dim)
901 isl_id **ids = NULL;
903 if (!dim)
904 return NULL;
905 isl_assert(dim->ctx, dim->n_in == 0, goto error);
906 if (dim->n_out == 0 && !isl_space_is_named_or_nested(dim, isl_dim_out))
907 return dim;
908 dim = isl_space_cow(dim);
909 if (!dim)
910 return NULL;
911 if (dim->ids) {
912 ids = isl_calloc_array(dim->ctx, isl_id *,
913 dim->nparam + dim->n_out + dim->n_out);
914 if (!ids)
915 goto error;
916 get_ids(dim, isl_dim_param, 0, dim->nparam, ids);
917 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->nparam);
919 dim->n_in = dim->n_out;
920 if (ids) {
921 free(dim->ids);
922 dim->ids = ids;
923 dim->n_id = dim->nparam + dim->n_out + dim->n_out;
924 dim = copy_ids(dim, isl_dim_out, 0, dim, isl_dim_in);
926 isl_id_free(dim->tuple_id[0]);
927 dim->tuple_id[0] = isl_id_copy(dim->tuple_id[1]);
928 isl_space_free(dim->nested[0]);
929 dim->nested[0] = isl_space_copy(dim->nested[1]);
930 return dim;
931 error:
932 isl_space_free(dim);
933 return NULL;
936 static __isl_give isl_space *set_ids(__isl_take isl_space *dim,
937 enum isl_dim_type type,
938 unsigned first, unsigned n, __isl_take isl_id **ids)
940 int i;
942 for (i = 0; i < n ; ++i)
943 dim = set_id(dim, type, first + i, ids[i]);
945 return dim;
948 __isl_give isl_space *isl_space_reverse(__isl_take isl_space *dim)
950 unsigned t;
951 isl_space *nested;
952 isl_id **ids = NULL;
953 isl_id *id;
955 if (!dim)
956 return NULL;
957 if (match(dim, isl_dim_in, dim, isl_dim_out))
958 return dim;
960 dim = isl_space_cow(dim);
961 if (!dim)
962 return NULL;
964 id = dim->tuple_id[0];
965 dim->tuple_id[0] = dim->tuple_id[1];
966 dim->tuple_id[1] = id;
968 nested = dim->nested[0];
969 dim->nested[0] = dim->nested[1];
970 dim->nested[1] = nested;
972 if (dim->ids) {
973 ids = isl_alloc_array(dim->ctx, isl_id *,
974 dim->n_in + dim->n_out);
975 if (!ids)
976 goto error;
977 get_ids(dim, isl_dim_in, 0, dim->n_in, ids);
978 get_ids(dim, isl_dim_out, 0, dim->n_out, ids + dim->n_in);
981 t = dim->n_in;
982 dim->n_in = dim->n_out;
983 dim->n_out = t;
985 if (dim->ids) {
986 dim = set_ids(dim, isl_dim_out, 0, dim->n_out, ids);
987 dim = set_ids(dim, isl_dim_in, 0, dim->n_in, ids + dim->n_out);
988 free(ids);
991 return dim;
992 error:
993 free(ids);
994 isl_space_free(dim);
995 return NULL;
998 __isl_give isl_space *isl_space_drop_dims(__isl_take isl_space *dim,
999 enum isl_dim_type type, unsigned first, unsigned num)
1001 int i;
1003 if (!dim)
1004 return NULL;
1006 if (num == 0)
1007 return isl_space_reset(dim, type);
1009 if (!valid_dim_type(type))
1010 isl_die(dim->ctx, isl_error_invalid,
1011 "cannot drop dimensions of specified type", goto error);
1013 isl_assert(dim->ctx, first + num <= n(dim, type), goto error);
1014 dim = isl_space_cow(dim);
1015 if (!dim)
1016 goto error;
1017 if (dim->ids) {
1018 dim = extend_ids(dim);
1019 if (!dim)
1020 goto error;
1021 for (i = 0; i < num; ++i)
1022 isl_id_free(get_id(dim, type, first + i));
1023 for (i = first+num; i < n(dim, type); ++i)
1024 set_id(dim, type, i - num, get_id(dim, type, i));
1025 switch (type) {
1026 case isl_dim_param:
1027 get_ids(dim, isl_dim_in, 0, dim->n_in,
1028 dim->ids + offset(dim, isl_dim_in) - num);
1029 case isl_dim_in:
1030 get_ids(dim, isl_dim_out, 0, dim->n_out,
1031 dim->ids + offset(dim, isl_dim_out) - num);
1032 default:
1035 dim->n_id -= num;
1037 switch (type) {
1038 case isl_dim_param: dim->nparam -= num; break;
1039 case isl_dim_in: dim->n_in -= num; break;
1040 case isl_dim_out: dim->n_out -= num; break;
1041 default: ;
1043 dim = isl_space_reset(dim, type);
1044 if (type == isl_dim_param) {
1045 if (dim && dim->nested[0] &&
1046 !(dim->nested[0] = isl_space_drop_dims(dim->nested[0],
1047 isl_dim_param, first, num)))
1048 goto error;
1049 if (dim && dim->nested[1] &&
1050 !(dim->nested[1] = isl_space_drop_dims(dim->nested[1],
1051 isl_dim_param, first, num)))
1052 goto error;
1054 return dim;
1055 error:
1056 isl_space_free(dim);
1057 return NULL;
1060 __isl_give isl_space *isl_space_drop_inputs(__isl_take isl_space *dim,
1061 unsigned first, unsigned n)
1063 if (!dim)
1064 return NULL;
1065 return isl_space_drop_dims(dim, isl_dim_in, first, n);
1068 __isl_give isl_space *isl_space_drop_outputs(__isl_take isl_space *dim,
1069 unsigned first, unsigned n)
1071 if (!dim)
1072 return NULL;
1073 return isl_space_drop_dims(dim, isl_dim_out, first, n);
1076 __isl_give isl_space *isl_space_domain(__isl_take isl_space *dim)
1078 if (!dim)
1079 return NULL;
1080 dim = isl_space_drop_outputs(dim, 0, dim->n_out);
1081 return isl_space_reverse(dim);
1084 __isl_give isl_space *isl_space_from_domain(__isl_take isl_space *dim)
1086 return isl_space_reverse(dim);
1089 __isl_give isl_space *isl_space_range(__isl_take isl_space *dim)
1091 if (!dim)
1092 return NULL;
1093 return isl_space_drop_inputs(dim, 0, dim->n_in);
1096 __isl_give isl_space *isl_space_from_range(__isl_take isl_space *dim)
1098 return dim;
1101 __isl_give isl_space *isl_space_params(__isl_take isl_space *space)
1103 space = isl_space_drop_dims(space,
1104 isl_dim_in, 0, isl_space_dim(space, isl_dim_in));
1105 space = isl_space_drop_dims(space,
1106 isl_dim_out, 0, isl_space_dim(space, isl_dim_out));
1107 return space;
1110 __isl_give isl_space *isl_space_as_set_space(__isl_take isl_space *dim)
1112 dim = isl_space_cow(dim);
1113 if (!dim)
1114 return NULL;
1116 dim->n_out += dim->n_in;
1117 dim->n_in = 0;
1118 dim = isl_space_reset(dim, isl_dim_in);
1119 dim = isl_space_reset(dim, isl_dim_out);
1121 return dim;
1124 __isl_give isl_space *isl_space_underlying(__isl_take isl_space *dim,
1125 unsigned n_div)
1127 int i;
1129 if (!dim)
1130 return NULL;
1131 if (n_div == 0 &&
1132 dim->nparam == 0 && dim->n_in == 0 && dim->n_id == 0)
1133 return isl_space_reset(isl_space_reset(dim, isl_dim_in), isl_dim_out);
1134 dim = isl_space_cow(dim);
1135 if (!dim)
1136 return NULL;
1137 dim->n_out += dim->nparam + dim->n_in + n_div;
1138 dim->nparam = 0;
1139 dim->n_in = 0;
1141 for (i = 0; i < dim->n_id; ++i)
1142 isl_id_free(get_id(dim, isl_dim_out, i));
1143 dim->n_id = 0;
1144 dim = isl_space_reset(dim, isl_dim_in);
1145 dim = isl_space_reset(dim, isl_dim_out);
1147 return dim;
1150 int isl_space_is_equal(__isl_keep isl_space *dim1, __isl_keep isl_space *dim2)
1152 if (!dim1 || !dim2)
1153 return -1;
1154 return match(dim1, isl_dim_param, dim2, isl_dim_param) &&
1155 isl_space_tuple_match(dim1, isl_dim_in, dim2, isl_dim_in) &&
1156 isl_space_tuple_match(dim1, isl_dim_out, dim2, isl_dim_out);
1159 int isl_space_compatible(__isl_keep isl_space *dim1,
1160 __isl_keep isl_space *dim2)
1162 return dim1->nparam == dim2->nparam &&
1163 dim1->n_in + dim1->n_out == dim2->n_in + dim2->n_out;
1166 static uint32_t isl_hash_dim(uint32_t hash, __isl_keep isl_space *dim)
1168 int i;
1169 isl_id *id;
1171 if (!dim)
1172 return hash;
1174 hash = isl_hash_builtin(hash, dim->nparam);
1175 hash = isl_hash_builtin(hash, dim->n_in);
1176 hash = isl_hash_builtin(hash, dim->n_out);
1178 for (i = 0; i < dim->nparam; ++i) {
1179 id = get_id(dim, isl_dim_param, i);
1180 hash = isl_hash_id(hash, id);
1183 id = tuple_id(dim, isl_dim_in);
1184 hash = isl_hash_id(hash, id);
1185 id = tuple_id(dim, isl_dim_out);
1186 hash = isl_hash_id(hash, id);
1188 hash = isl_hash_dim(hash, dim->nested[0]);
1189 hash = isl_hash_dim(hash, dim->nested[1]);
1191 return hash;
1194 uint32_t isl_space_get_hash(__isl_keep isl_space *dim)
1196 uint32_t hash;
1198 if (!dim)
1199 return 0;
1201 hash = isl_hash_init();
1202 hash = isl_hash_dim(hash, dim);
1204 return hash;
1207 int isl_space_is_wrapping(__isl_keep isl_space *dim)
1209 if (!dim)
1210 return -1;
1212 if (dim->n_in != 0 || dim->tuple_id[0] || dim->nested[0])
1213 return 0;
1215 return dim->nested[1] != NULL;
1218 __isl_give isl_space *isl_space_wrap(__isl_take isl_space *dim)
1220 isl_space *wrap;
1222 if (!dim)
1223 return NULL;
1225 wrap = isl_space_alloc(dim->ctx, dim->nparam, 0, dim->n_in + dim->n_out);
1227 wrap = copy_ids(wrap, isl_dim_param, 0, dim, isl_dim_param);
1228 wrap = copy_ids(wrap, isl_dim_set, 0, dim, isl_dim_in);
1229 wrap = copy_ids(wrap, isl_dim_set, dim->n_in, dim, isl_dim_out);
1231 if (!wrap)
1232 goto error;
1234 wrap->nested[1] = dim;
1236 return wrap;
1237 error:
1238 isl_space_free(dim);
1239 return NULL;
1242 __isl_give isl_space *isl_space_unwrap(__isl_take isl_space *dim)
1244 isl_space *unwrap;
1246 if (!dim)
1247 return NULL;
1249 if (!isl_space_is_wrapping(dim))
1250 isl_die(dim->ctx, isl_error_invalid, "not a wrapping dim",
1251 goto error);
1253 unwrap = isl_space_copy(dim->nested[1]);
1254 isl_space_free(dim);
1256 return unwrap;
1257 error:
1258 isl_space_free(dim);
1259 return NULL;
1262 int isl_space_is_named_or_nested(__isl_keep isl_space *dim, enum isl_dim_type type)
1264 if (type != isl_dim_in && type != isl_dim_out)
1265 return 0;
1266 if (!dim)
1267 return -1;
1268 if (dim->tuple_id[type - isl_dim_in])
1269 return 1;
1270 if (dim->nested[type - isl_dim_in])
1271 return 1;
1272 return 0;
1275 int isl_space_may_be_set(__isl_keep isl_space *dim)
1277 if (!dim)
1278 return -1;
1279 if (isl_space_dim(dim, isl_dim_in) != 0)
1280 return 0;
1281 if (isl_space_is_named_or_nested(dim, isl_dim_in))
1282 return 0;
1283 return 1;
1286 __isl_give isl_space *isl_space_reset(__isl_take isl_space *dim,
1287 enum isl_dim_type type)
1289 if (!isl_space_is_named_or_nested(dim, type))
1290 return dim;
1292 dim = isl_space_cow(dim);
1293 if (!dim)
1294 return NULL;
1296 isl_id_free(dim->tuple_id[type - isl_dim_in]);
1297 dim->tuple_id[type - isl_dim_in] = NULL;
1298 isl_space_free(dim->nested[type - isl_dim_in]);
1299 dim->nested[type - isl_dim_in] = NULL;
1301 return dim;
1304 __isl_give isl_space *isl_space_flatten(__isl_take isl_space *dim)
1306 if (!dim)
1307 return NULL;
1308 if (!dim->nested[0] && !dim->nested[1])
1309 return dim;
1311 if (dim->nested[0])
1312 dim = isl_space_reset(dim, isl_dim_in);
1313 if (dim && dim->nested[1])
1314 dim = isl_space_reset(dim, isl_dim_out);
1316 return dim;
1319 __isl_give isl_space *isl_space_flatten_range(__isl_take isl_space *dim)
1321 if (!dim)
1322 return NULL;
1323 if (!dim->nested[1])
1324 return dim;
1326 return isl_space_reset(dim, isl_dim_out);
1329 /* Replace the dimensions of the given type of dst by those of src.
1331 __isl_give isl_space *isl_space_replace(__isl_take isl_space *dst,
1332 enum isl_dim_type type, __isl_keep isl_space *src)
1334 dst = isl_space_cow(dst);
1336 if (!dst || !src)
1337 goto error;
1339 dst = isl_space_drop_dims(dst, type, 0, isl_space_dim(dst, type));
1340 dst = isl_space_add_dims(dst, type, isl_space_dim(src, type));
1341 dst = copy_ids(dst, type, 0, src, type);
1343 if (dst && type == isl_dim_param) {
1344 int i;
1345 for (i = 0; i <= 1; ++i) {
1346 if (!dst->nested[i])
1347 continue;
1348 dst->nested[i] = isl_space_replace(dst->nested[i],
1349 type, src);
1350 if (!dst->nested[i])
1351 goto error;
1355 return dst;
1356 error:
1357 isl_space_free(dst);
1358 return NULL;
1361 /* Given a dimension specification "dim" of a set, create a dimension
1362 * specification for the lift of the set. In particular, the result
1363 * is of the form [dim -> local[..]], with n_local variables in the
1364 * range of the wrapped map.
1366 __isl_give isl_space *isl_space_lift(__isl_take isl_space *dim, unsigned n_local)
1368 isl_space *local_dim;
1370 if (!dim)
1371 return NULL;
1373 local_dim = isl_space_dup(dim);
1374 local_dim = isl_space_drop_dims(local_dim, isl_dim_set, 0, dim->n_out);
1375 local_dim = isl_space_add_dims(local_dim, isl_dim_set, n_local);
1376 local_dim = isl_space_set_tuple_name(local_dim, isl_dim_set, "local");
1377 dim = isl_space_join(isl_space_from_domain(dim),
1378 isl_space_from_range(local_dim));
1379 dim = isl_space_wrap(dim);
1380 dim = isl_space_set_tuple_name(dim, isl_dim_set, "lifted");
1382 return dim;
1385 int isl_space_can_zip(__isl_keep isl_space *dim)
1387 if (!dim)
1388 return -1;
1390 return dim->nested[0] && dim->nested[1];
1393 __isl_give isl_space *isl_space_zip(__isl_take isl_space *dim)
1395 isl_space *dom, *ran;
1396 isl_space *dom_dom, *dom_ran, *ran_dom, *ran_ran;
1398 if (!isl_space_can_zip(dim))
1399 isl_die(dim->ctx, isl_error_invalid, "dim cannot be zipped",
1400 goto error);
1402 if (!dim)
1403 return 0;
1404 dom = isl_space_unwrap(isl_space_domain(isl_space_copy(dim)));
1405 ran = isl_space_unwrap(isl_space_range(dim));
1406 dom_dom = isl_space_domain(isl_space_copy(dom));
1407 dom_ran = isl_space_range(dom);
1408 ran_dom = isl_space_domain(isl_space_copy(ran));
1409 ran_ran = isl_space_range(ran);
1410 dom = isl_space_join(isl_space_from_domain(dom_dom),
1411 isl_space_from_range(ran_dom));
1412 ran = isl_space_join(isl_space_from_domain(dom_ran),
1413 isl_space_from_range(ran_ran));
1414 return isl_space_join(isl_space_from_domain(isl_space_wrap(dom)),
1415 isl_space_from_range(isl_space_wrap(ran)));
1416 error:
1417 isl_space_free(dim);
1418 return NULL;
1421 int isl_space_has_named_params(__isl_keep isl_space *dim)
1423 int i;
1424 unsigned off;
1426 if (!dim)
1427 return -1;
1428 if (dim->nparam == 0)
1429 return 1;
1430 off = isl_space_offset(dim, isl_dim_param);
1431 if (off + dim->nparam > dim->n_id)
1432 return 0;
1433 for (i = 0; i < dim->nparam; ++i)
1434 if (!dim->ids[off + i])
1435 return 0;
1436 return 1;
1439 /* Align the initial parameters of dim1 to match the order in dim2.
1441 __isl_give isl_space *isl_space_align_params(__isl_take isl_space *dim1,
1442 __isl_take isl_space *dim2)
1444 isl_reordering *exp;
1446 if (!isl_space_has_named_params(dim1) || !isl_space_has_named_params(dim2))
1447 isl_die(isl_space_get_ctx(dim1), isl_error_invalid,
1448 "parameter alignment requires named parameters",
1449 goto error);
1451 dim2 = isl_space_params(dim2);
1452 exp = isl_parameter_alignment_reordering(dim1, dim2);
1453 exp = isl_reordering_extend_space(exp, dim1);
1454 isl_space_free(dim2);
1455 if (!exp)
1456 return NULL;
1457 dim1 = isl_space_copy(exp->dim);
1458 isl_reordering_free(exp);
1459 return dim1;
1460 error:
1461 isl_space_free(dim1);
1462 isl_space_free(dim2);
1463 return NULL;