isl_map_gist: handle some special cases
[isl.git] / isl_map.c
blobf0ee231a2dd47ccd69f8406fed89053a1cafd764
1 #include <string.h>
2 #include <strings.h>
3 #include "isl_ctx.h"
4 #include "isl_blk.h"
5 #include "isl_dim.h"
6 #include "isl_list.h"
7 #include "isl_lp.h"
8 #include "isl_seq.h"
9 #include "isl_set.h"
10 #include "isl_map.h"
11 #include "isl_map_private.h"
12 #include "isl_map_piplib.h"
13 #include "isl_sample.h"
14 #include "isl_vec.h"
16 /* Maps dst positions to src positions */
17 struct isl_dim_map {
18 unsigned len;
19 int pos[1];
22 static struct isl_dim_map *isl_dim_map_alloc(struct isl_ctx *ctx, unsigned len)
24 int i;
25 struct isl_dim_map *dim_map;
26 dim_map = isl_alloc(ctx, struct isl_dim_map,
27 sizeof(struct isl_dim_map) + len * sizeof(int));
28 if (!dim_map)
29 return NULL;
30 dim_map->len = 1 + len;
31 dim_map->pos[0] = 0;
32 for (i = 0; i < len; ++i)
33 dim_map->pos[1 + i] = -1;
34 return dim_map;
37 static unsigned n(struct isl_dim *dim, enum isl_dim_type type)
39 switch (type) {
40 case isl_dim_param: return dim->nparam;
41 case isl_dim_in: return dim->n_in;
42 case isl_dim_out: return dim->n_out;
46 static unsigned pos(struct isl_dim *dim, enum isl_dim_type type)
48 switch (type) {
49 case isl_dim_param: return 1;
50 case isl_dim_in: return 1 + dim->nparam;
51 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
55 static void isl_dim_map_dim(struct isl_dim_map *dim_map, struct isl_dim *dim,
56 enum isl_dim_type type, unsigned dst_pos)
58 int i;
59 unsigned src_pos;
61 if (!dim_map || !dim)
62 return;
64 src_pos = pos(dim, type);
65 for (i = 0; i < n(dim, type); ++i)
66 dim_map->pos[1 + dst_pos + i] = src_pos + i;
69 static void isl_dim_map_div(struct isl_dim_map *dim_map,
70 struct isl_basic_map *bmap, unsigned dst_pos)
72 int i;
73 unsigned src_pos;
75 if (!dim_map || !bmap)
76 return;
78 src_pos = 1 + isl_dim_total(bmap->dim);
79 for (i = 0; i < bmap->n_div; ++i)
80 dim_map->pos[1 + dst_pos + i] = src_pos + i;
83 static void isl_dim_map_dump(struct isl_dim_map *dim_map)
85 int i;
87 for (i = 0; i < dim_map->len; ++i)
88 fprintf(stderr, "%d -> %d; ", i, dim_map->pos[i]);
89 fprintf(stderr, "\n");
92 unsigned isl_basic_map_dim(const struct isl_basic_map *bmap,
93 enum isl_dim_type type)
95 struct isl_dim *dim = bmap->dim;
96 switch (type) {
97 case isl_dim_param:
98 case isl_dim_in:
99 case isl_dim_out: return isl_dim_size(bmap->dim, type);
100 case isl_dim_div: return bmap->n_div;
101 case isl_dim_all: return isl_basic_map_total_dim(bmap);
105 unsigned isl_map_dim(const struct isl_map *map, enum isl_dim_type type)
107 return n(map->dim, type);
110 unsigned isl_set_dim(const struct isl_set *set, enum isl_dim_type type)
112 return n(set->dim, type);
115 unsigned isl_basic_map_offset(struct isl_basic_map *bmap,
116 enum isl_dim_type type)
118 struct isl_dim *dim = bmap->dim;
119 switch (type) {
120 case isl_dim_param: return 1;
121 case isl_dim_in: return 1 + dim->nparam;
122 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
123 case isl_dim_div: return 1 + dim->nparam + dim->n_in + dim->n_out;
127 static unsigned map_offset(struct isl_map *map, enum isl_dim_type type)
129 return pos(map->dim, type);
132 unsigned isl_basic_set_dim(const struct isl_basic_set *bset,
133 enum isl_dim_type type)
135 return isl_basic_map_dim((const struct isl_basic_map*)bset, type);
138 unsigned isl_basic_set_n_dim(const struct isl_basic_set *bset)
140 return bset->dim->n_out;
143 unsigned isl_basic_set_n_param(const struct isl_basic_set *bset)
145 return bset->dim->nparam;
148 unsigned isl_basic_set_total_dim(const struct isl_basic_set *bset)
150 return isl_dim_total(bset->dim) + bset->n_div;
153 unsigned isl_set_n_dim(const struct isl_set *set)
155 return set->dim->n_out;
158 unsigned isl_set_n_param(const struct isl_set *set)
160 return set->dim->nparam;
163 unsigned isl_basic_map_n_in(const struct isl_basic_map *bmap)
165 return bmap->dim->n_in;
168 unsigned isl_basic_map_n_out(const struct isl_basic_map *bmap)
170 return bmap->dim->n_out;
173 unsigned isl_basic_map_n_param(const struct isl_basic_map *bmap)
175 return bmap->dim->nparam;
178 unsigned isl_basic_map_n_div(const struct isl_basic_map *bmap)
180 return bmap->n_div;
183 unsigned isl_basic_map_total_dim(const struct isl_basic_map *bmap)
185 return isl_dim_total(bmap->dim) + bmap->n_div;
188 unsigned isl_map_n_in(const struct isl_map *map)
190 return map->dim->n_in;
193 unsigned isl_map_n_out(const struct isl_map *map)
195 return map->dim->n_out;
198 unsigned isl_map_n_param(const struct isl_map *map)
200 return map->dim->nparam;
203 int isl_map_compatible_domain(struct isl_map *map, struct isl_set *set)
205 return map->dim->n_in == set->dim->n_out &&
206 map->dim->nparam == set->dim->nparam;
209 int isl_basic_map_compatible_domain(struct isl_basic_map *bmap,
210 struct isl_basic_set *bset)
212 return bmap->dim->n_in == bset->dim->n_out &&
213 bmap->dim->nparam == bset->dim->nparam;
216 int isl_basic_map_compatible_range(struct isl_basic_map *bmap,
217 struct isl_basic_set *bset)
219 return bmap->dim->n_out == bset->dim->n_out &&
220 bmap->dim->nparam == bset->dim->nparam;
223 static struct isl_basic_map *basic_map_init(struct isl_ctx *ctx,
224 struct isl_basic_map *bmap, unsigned extra,
225 unsigned n_eq, unsigned n_ineq)
227 int i;
228 size_t row_size = 1 + isl_dim_total(bmap->dim) + extra;
230 bmap->block = isl_blk_alloc(ctx, (n_ineq + n_eq) * row_size);
231 if (isl_blk_is_error(bmap->block)) {
232 free(bmap);
233 return NULL;
236 bmap->ineq = isl_alloc_array(ctx, isl_int *, n_ineq + n_eq);
237 if (!bmap->ineq) {
238 isl_blk_free(ctx, bmap->block);
239 free(bmap);
240 return NULL;
243 if (extra == 0) {
244 bmap->block2 = isl_blk_empty();
245 bmap->div = NULL;
246 } else {
247 bmap->block2 = isl_blk_alloc(ctx, extra * (1 + row_size));
248 if (isl_blk_is_error(bmap->block2)) {
249 free(bmap->ineq);
250 isl_blk_free(ctx, bmap->block);
251 free(bmap);
252 return NULL;
255 bmap->div = isl_alloc_array(ctx, isl_int *, extra);
256 if (!bmap->div) {
257 isl_blk_free(ctx, bmap->block2);
258 free(bmap->ineq);
259 isl_blk_free(ctx, bmap->block);
260 free(bmap);
261 return NULL;
265 for (i = 0; i < n_ineq + n_eq; ++i)
266 bmap->ineq[i] = bmap->block.data + i * row_size;
268 for (i = 0; i < extra; ++i)
269 bmap->div[i] = bmap->block2.data + i * (1 + row_size);
271 bmap->ctx = ctx;
272 isl_ctx_ref(ctx);
273 bmap->ref = 1;
274 bmap->flags = 0;
275 bmap->c_size = n_eq + n_ineq;
276 bmap->eq = bmap->ineq + n_ineq;
277 bmap->extra = extra;
278 bmap->n_eq = 0;
279 bmap->n_ineq = 0;
280 bmap->n_div = 0;
281 bmap->sample = NULL;
283 return bmap;
284 error:
285 isl_basic_map_free(bmap);
286 return NULL;
289 struct isl_basic_set *isl_basic_set_alloc(struct isl_ctx *ctx,
290 unsigned nparam, unsigned dim, unsigned extra,
291 unsigned n_eq, unsigned n_ineq)
293 struct isl_basic_map *bmap;
294 bmap = isl_basic_map_alloc(ctx, nparam, 0, dim, extra, n_eq, n_ineq);
295 return (struct isl_basic_set *)bmap;
298 struct isl_basic_set *isl_basic_set_alloc_dim(struct isl_dim *dim,
299 unsigned extra, unsigned n_eq, unsigned n_ineq)
301 struct isl_basic_map *bmap;
302 if (!dim)
303 return NULL;
304 isl_assert(dim->ctx, dim->n_in == 0, return NULL);
305 bmap = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
306 return (struct isl_basic_set *)bmap;
309 struct isl_basic_map *isl_basic_map_alloc_dim(struct isl_dim *dim,
310 unsigned extra, unsigned n_eq, unsigned n_ineq)
312 struct isl_basic_map *bmap;
314 if (!dim)
315 return NULL;
316 bmap = isl_alloc_type(dim->ctx, struct isl_basic_map);
317 if (!bmap)
318 goto error;
319 bmap->dim = dim;
321 return basic_map_init(dim->ctx, bmap, extra, n_eq, n_ineq);
322 error:
323 isl_dim_free(dim);
324 return NULL;
327 struct isl_basic_map *isl_basic_map_alloc(struct isl_ctx *ctx,
328 unsigned nparam, unsigned in, unsigned out, unsigned extra,
329 unsigned n_eq, unsigned n_ineq)
331 struct isl_basic_map *bmap;
332 struct isl_dim *dim;
334 dim = isl_dim_alloc(ctx, nparam, in, out);
335 if (!dim)
336 return NULL;
338 bmap = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
339 return bmap;
342 static void dup_constraints(
343 struct isl_basic_map *dst, struct isl_basic_map *src)
345 int i;
346 unsigned total = isl_basic_map_total_dim(src);
348 for (i = 0; i < src->n_eq; ++i) {
349 int j = isl_basic_map_alloc_equality(dst);
350 isl_seq_cpy(dst->eq[j], src->eq[i], 1+total);
353 for (i = 0; i < src->n_ineq; ++i) {
354 int j = isl_basic_map_alloc_inequality(dst);
355 isl_seq_cpy(dst->ineq[j], src->ineq[i], 1+total);
358 for (i = 0; i < src->n_div; ++i) {
359 int j = isl_basic_map_alloc_div(dst);
360 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total);
362 ISL_F_SET(dst, ISL_BASIC_SET_FINAL);
365 struct isl_basic_map *isl_basic_map_dup(struct isl_basic_map *bmap)
367 struct isl_basic_map *dup;
369 if (!bmap)
370 return NULL;
371 dup = isl_basic_map_alloc_dim(isl_dim_copy(bmap->dim),
372 bmap->n_div, bmap->n_eq, bmap->n_ineq);
373 if (!dup)
374 return NULL;
375 dup_constraints(dup, bmap);
376 dup->flags = bmap->flags;
377 dup->sample = isl_vec_copy(bmap->ctx, bmap->sample);
378 return dup;
381 struct isl_basic_set *isl_basic_set_dup(struct isl_basic_set *bset)
383 struct isl_basic_map *dup;
385 dup = isl_basic_map_dup((struct isl_basic_map *)bset);
386 return (struct isl_basic_set *)dup;
389 struct isl_basic_set *isl_basic_set_copy(struct isl_basic_set *bset)
391 if (!bset)
392 return NULL;
394 if (ISL_F_ISSET(bset, ISL_BASIC_SET_FINAL)) {
395 bset->ref++;
396 return bset;
398 return isl_basic_set_dup(bset);
401 struct isl_set *isl_set_copy(struct isl_set *set)
403 if (!set)
404 return NULL;
406 set->ref++;
407 return set;
410 struct isl_basic_map *isl_basic_map_copy(struct isl_basic_map *bmap)
412 if (!bmap)
413 return NULL;
415 if (ISL_F_ISSET(bmap, ISL_BASIC_SET_FINAL)) {
416 bmap->ref++;
417 return bmap;
419 return isl_basic_map_dup(bmap);
422 struct isl_map *isl_map_copy(struct isl_map *map)
424 if (!map)
425 return NULL;
427 map->ref++;
428 return map;
431 void isl_basic_map_free(struct isl_basic_map *bmap)
433 if (!bmap)
434 return;
436 if (--bmap->ref > 0)
437 return;
439 isl_ctx_deref(bmap->ctx);
440 free(bmap->div);
441 isl_blk_free(bmap->ctx, bmap->block2);
442 free(bmap->ineq);
443 isl_blk_free(bmap->ctx, bmap->block);
444 isl_vec_free(bmap->ctx, bmap->sample);
445 isl_dim_free(bmap->dim);
446 free(bmap);
449 void isl_basic_set_free(struct isl_basic_set *bset)
451 isl_basic_map_free((struct isl_basic_map *)bset);
454 static int room_for_con(struct isl_basic_map *bmap, unsigned n)
456 return bmap->n_eq + bmap->n_ineq + n <= bmap->c_size;
459 int isl_basic_map_alloc_equality(struct isl_basic_map *bmap)
461 struct isl_ctx *ctx;
462 if (!bmap)
463 return -1;
464 ctx = bmap->ctx;
465 isl_assert(ctx, room_for_con(bmap, 1), return -1);
466 isl_assert(ctx, (bmap->eq - bmap->ineq) + bmap->n_eq <= bmap->c_size,
467 return -1);
468 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
469 ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
470 ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
471 ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
472 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
473 if ((bmap->eq - bmap->ineq) + bmap->n_eq == bmap->c_size) {
474 isl_int *t;
475 int j = isl_basic_map_alloc_inequality(bmap);
476 if (j < 0)
477 return -1;
478 t = bmap->ineq[j];
479 bmap->ineq[j] = bmap->ineq[bmap->n_ineq - 1];
480 bmap->ineq[bmap->n_ineq - 1] = bmap->eq[-1];
481 bmap->eq[-1] = t;
482 bmap->n_eq++;
483 bmap->n_ineq--;
484 bmap->eq--;
485 return 0;
487 isl_seq_clr(bmap->eq[bmap->n_eq] + 1 + isl_basic_map_total_dim(bmap),
488 bmap->extra - bmap->n_div);
489 return bmap->n_eq++;
492 int isl_basic_set_alloc_equality(struct isl_basic_set *bset)
494 return isl_basic_map_alloc_equality((struct isl_basic_map *)bset);
497 int isl_basic_map_free_equality(struct isl_basic_map *bmap, unsigned n)
499 if (!bmap)
500 return -1;
501 isl_assert(bmap->ctx, n <= bmap->n_eq, return -1);
502 bmap->n_eq -= n;
503 return 0;
506 int isl_basic_set_free_equality(struct isl_basic_set *bset, unsigned n)
508 return isl_basic_map_free_equality((struct isl_basic_map *)bset, n);
511 int isl_basic_map_drop_equality(struct isl_basic_map *bmap, unsigned pos)
513 isl_int *t;
514 if (!bmap)
515 return -1;
516 isl_assert(bmap->ctx, pos < bmap->n_eq, return -1);
518 if (pos != bmap->n_eq - 1) {
519 t = bmap->eq[pos];
520 bmap->eq[pos] = bmap->eq[bmap->n_eq - 1];
521 bmap->eq[bmap->n_eq - 1] = t;
523 bmap->n_eq--;
524 return 0;
527 int isl_basic_set_drop_equality(struct isl_basic_set *bset, unsigned pos)
529 return isl_basic_map_drop_equality((struct isl_basic_map *)bset, pos);
532 void isl_basic_map_inequality_to_equality(
533 struct isl_basic_map *bmap, unsigned pos)
535 isl_int *t;
537 t = bmap->ineq[pos];
538 bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
539 bmap->ineq[bmap->n_ineq - 1] = bmap->eq[-1];
540 bmap->eq[-1] = t;
541 bmap->n_eq++;
542 bmap->n_ineq--;
543 bmap->eq--;
544 ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
545 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
546 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
547 ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
550 static int room_for_ineq(struct isl_basic_map *bmap, unsigned n)
552 return bmap->n_ineq + n <= bmap->eq - bmap->ineq;
555 int isl_basic_map_alloc_inequality(struct isl_basic_map *bmap)
557 struct isl_ctx *ctx;
558 if (!bmap)
559 return -1;
560 ctx = bmap->ctx;
561 isl_assert(ctx, room_for_ineq(bmap, 1), return -1);
562 ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
563 ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
564 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
565 ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
566 isl_seq_clr(bmap->ineq[bmap->n_ineq] +
567 1 + isl_basic_map_total_dim(bmap),
568 bmap->extra - bmap->n_div);
569 return bmap->n_ineq++;
572 int isl_basic_set_alloc_inequality(struct isl_basic_set *bset)
574 return isl_basic_map_alloc_inequality((struct isl_basic_map *)bset);
577 int isl_basic_map_free_inequality(struct isl_basic_map *bmap, unsigned n)
579 if (!bmap)
580 return -1;
581 isl_assert(bmap->ctx, n <= bmap->n_ineq, return -1);
582 bmap->n_ineq -= n;
583 return 0;
586 int isl_basic_set_free_inequality(struct isl_basic_set *bset, unsigned n)
588 return isl_basic_map_free_inequality((struct isl_basic_map *)bset, n);
591 int isl_basic_map_drop_inequality(struct isl_basic_map *bmap, unsigned pos)
593 isl_int *t;
594 if (!bmap)
595 return -1;
596 isl_assert(bmap->ctx, pos < bmap->n_ineq, return -1);
598 if (pos != bmap->n_ineq - 1) {
599 t = bmap->ineq[pos];
600 bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
601 bmap->ineq[bmap->n_ineq - 1] = t;
602 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
604 bmap->n_ineq--;
605 return 0;
608 int isl_basic_set_drop_inequality(struct isl_basic_set *bset, unsigned pos)
610 return isl_basic_map_drop_inequality((struct isl_basic_map *)bset, pos);
613 int isl_basic_map_alloc_div(struct isl_basic_map *bmap)
615 if (!bmap)
616 return -1;
617 isl_assert(bmap->ctx, bmap->n_div < bmap->extra, return -1);
618 isl_seq_clr(bmap->div[bmap->n_div] +
619 1 + 1 + isl_basic_map_total_dim(bmap),
620 bmap->extra - bmap->n_div);
621 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
622 return bmap->n_div++;
625 int isl_basic_set_alloc_div(struct isl_basic_set *bset)
627 return isl_basic_map_alloc_div((struct isl_basic_map *)bset);
630 int isl_basic_map_free_div(struct isl_basic_map *bmap, unsigned n)
632 if (!bmap)
633 return -1;
634 isl_assert(bmap->ctx, n <= bmap->n_div, return -1);
635 bmap->n_div -= n;
636 return 0;
639 /* Copy constraint from src to dst, putting the vars of src at offset
640 * dim_off in dst and the divs of src at offset div_off in dst.
641 * If both sets are actually map, then dim_off applies to the input
642 * variables.
644 static void copy_constraint(struct isl_basic_map *dst_map, isl_int *dst,
645 struct isl_basic_map *src_map, isl_int *src,
646 unsigned in_off, unsigned out_off, unsigned div_off)
648 unsigned src_nparam = isl_basic_map_n_param(src_map);
649 unsigned dst_nparam = isl_basic_map_n_param(dst_map);
650 unsigned src_in = isl_basic_map_n_in(src_map);
651 unsigned dst_in = isl_basic_map_n_in(dst_map);
652 unsigned src_out = isl_basic_map_n_out(src_map);
653 unsigned dst_out = isl_basic_map_n_out(dst_map);
654 isl_int_set(dst[0], src[0]);
655 isl_seq_cpy(dst+1, src+1, isl_min(dst_nparam, src_nparam));
656 if (dst_nparam > src_nparam)
657 isl_seq_clr(dst+1+src_nparam,
658 dst_nparam - src_nparam);
659 isl_seq_clr(dst+1+dst_nparam, in_off);
660 isl_seq_cpy(dst+1+dst_nparam+in_off,
661 src+1+src_nparam,
662 isl_min(dst_in-in_off, src_in));
663 if (dst_in-in_off > src_in)
664 isl_seq_clr(dst+1+dst_nparam+in_off+src_in,
665 dst_in - in_off - src_in);
666 isl_seq_clr(dst+1+dst_nparam+dst_in, out_off);
667 isl_seq_cpy(dst+1+dst_nparam+dst_in+out_off,
668 src+1+src_nparam+src_in,
669 isl_min(dst_out-out_off, src_out));
670 if (dst_out-out_off > src_out)
671 isl_seq_clr(dst+1+dst_nparam+dst_in+out_off+src_out,
672 dst_out - out_off - src_out);
673 isl_seq_clr(dst+1+dst_nparam+dst_in+dst_out, div_off);
674 isl_seq_cpy(dst+1+dst_nparam+dst_in+dst_out+div_off,
675 src+1+src_nparam+src_in+src_out,
676 isl_min(dst_map->extra-div_off, src_map->n_div));
677 if (dst_map->n_div-div_off > src_map->n_div)
678 isl_seq_clr(dst+1+dst_nparam+dst_in+dst_out+
679 div_off+src_map->n_div,
680 dst_map->n_div - div_off - src_map->n_div);
683 static void copy_div(struct isl_basic_map *dst_map, isl_int *dst,
684 struct isl_basic_map *src_map, isl_int *src,
685 unsigned in_off, unsigned out_off, unsigned div_off)
687 isl_int_set(dst[0], src[0]);
688 copy_constraint(dst_map, dst+1, src_map, src+1, in_off, out_off, div_off);
691 static struct isl_basic_map *add_constraints(struct isl_basic_map *bmap1,
692 struct isl_basic_map *bmap2, unsigned i_pos, unsigned o_pos)
694 int i;
695 unsigned div_off;
697 if (!bmap1 || !bmap2)
698 goto error;
700 div_off = bmap1->n_div;
702 for (i = 0; i < bmap2->n_eq; ++i) {
703 int i1 = isl_basic_map_alloc_equality(bmap1);
704 if (i1 < 0)
705 goto error;
706 copy_constraint(bmap1, bmap1->eq[i1], bmap2, bmap2->eq[i],
707 i_pos, o_pos, div_off);
710 for (i = 0; i < bmap2->n_ineq; ++i) {
711 int i1 = isl_basic_map_alloc_inequality(bmap1);
712 if (i1 < 0)
713 goto error;
714 copy_constraint(bmap1, bmap1->ineq[i1], bmap2, bmap2->ineq[i],
715 i_pos, o_pos, div_off);
718 for (i = 0; i < bmap2->n_div; ++i) {
719 int i1 = isl_basic_map_alloc_div(bmap1);
720 if (i1 < 0)
721 goto error;
722 copy_div(bmap1, bmap1->div[i1], bmap2, bmap2->div[i],
723 i_pos, o_pos, div_off);
726 isl_basic_map_free(bmap2);
728 return bmap1;
730 error:
731 isl_basic_map_free(bmap1);
732 isl_basic_map_free(bmap2);
733 return NULL;
736 static void copy_constraint_dim_map(isl_int *dst, isl_int *src,
737 struct isl_dim_map *dim_map)
739 int i;
741 for (i = 0; i < dim_map->len; ++i) {
742 if (dim_map->pos[i] < 0)
743 isl_int_set_si(dst[i], 0);
744 else
745 isl_int_set(dst[i], src[dim_map->pos[i]]);
749 static void copy_div_dim_map(isl_int *dst, isl_int *src,
750 struct isl_dim_map *dim_map)
752 isl_int_set(dst[0], src[0]);
753 copy_constraint_dim_map(dst+1, src+1, dim_map);
756 static struct isl_basic_map *add_constraints_dim_map(struct isl_basic_map *dst,
757 struct isl_basic_map *src, struct isl_dim_map *dim_map)
759 int i;
761 if (!src || !dst || !dim_map)
762 goto error;
764 for (i = 0; i < src->n_eq; ++i) {
765 int i1 = isl_basic_map_alloc_equality(dst);
766 if (i1 < 0)
767 goto error;
768 copy_constraint_dim_map(dst->eq[i1], src->eq[i], dim_map);
771 for (i = 0; i < src->n_ineq; ++i) {
772 int i1 = isl_basic_map_alloc_inequality(dst);
773 if (i1 < 0)
774 goto error;
775 copy_constraint_dim_map(dst->ineq[i1], src->ineq[i], dim_map);
778 for (i = 0; i < src->n_div; ++i) {
779 int i1 = isl_basic_map_alloc_div(dst);
780 if (i1 < 0)
781 goto error;
782 copy_div_dim_map(dst->div[i1], src->div[i], dim_map);
785 free(dim_map);
786 isl_basic_map_free(src);
788 return dst;
789 error:
790 free(dim_map);
791 isl_basic_map_free(src);
792 isl_basic_map_free(dst);
793 return NULL;
796 struct isl_basic_set *isl_basic_set_add_constraints(struct isl_basic_set *bset1,
797 struct isl_basic_set *bset2, unsigned pos)
799 return (struct isl_basic_set *)
800 add_constraints((struct isl_basic_map *)bset1,
801 (struct isl_basic_map *)bset2, 0, pos);
804 struct isl_basic_map *isl_basic_map_extend_dim(struct isl_basic_map *base,
805 struct isl_dim *dim, unsigned extra,
806 unsigned n_eq, unsigned n_ineq)
808 struct isl_basic_map *ext;
809 unsigned flags;
810 int dims_ok;
812 if (!dim)
813 goto error;
815 if (!base)
816 goto error;
818 dims_ok = isl_dim_equal(base->dim, dim) &&
819 base->extra >= base->n_div + extra;
821 if (dims_ok && room_for_con(base, n_eq + n_ineq) &&
822 room_for_ineq(base, n_ineq)) {
823 isl_dim_free(dim);
824 return base;
827 isl_assert(base->ctx, base->dim->nparam <= dim->nparam, goto error);
828 isl_assert(base->ctx, base->dim->n_in <= dim->n_in, goto error);
829 isl_assert(base->ctx, base->dim->n_out <= dim->n_out, goto error);
830 extra += base->extra;
831 n_eq += base->n_eq;
832 n_ineq += base->n_ineq;
834 ext = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
835 dim = NULL;
836 if (!ext)
837 goto error;
839 flags = base->flags;
840 ext = add_constraints(ext, base, 0, 0);
841 if (ext) {
842 ext->flags = flags;
843 ISL_F_CLR(ext, ISL_BASIC_SET_FINAL);
846 return ext;
848 error:
849 isl_dim_free(dim);
850 isl_basic_map_free(base);
851 return NULL;
854 struct isl_basic_set *isl_basic_set_extend_dim(struct isl_basic_set *base,
855 struct isl_dim *dim, unsigned extra,
856 unsigned n_eq, unsigned n_ineq)
858 return (struct isl_basic_set *)
859 isl_basic_map_extend_dim((struct isl_basic_map *)base, dim,
860 extra, n_eq, n_ineq);
863 struct isl_basic_map *isl_basic_map_extend_constraints(
864 struct isl_basic_map *base, unsigned n_eq, unsigned n_ineq)
866 if (!base)
867 return NULL;
868 return isl_basic_map_extend_dim(base, isl_dim_copy(base->dim),
869 0, n_eq, n_ineq);
872 struct isl_basic_map *isl_basic_map_extend(struct isl_basic_map *base,
873 unsigned nparam, unsigned n_in, unsigned n_out, unsigned extra,
874 unsigned n_eq, unsigned n_ineq)
876 struct isl_basic_map *bmap;
877 struct isl_dim *dim;
879 if (!base)
880 return NULL;
881 dim = isl_dim_alloc(base->ctx, nparam, n_in, n_out);
882 if (!dim)
883 return NULL;
885 bmap = isl_basic_map_extend_dim(base, dim, extra, n_eq, n_ineq);
886 return bmap;
889 struct isl_basic_set *isl_basic_set_extend(struct isl_basic_set *base,
890 unsigned nparam, unsigned dim, unsigned extra,
891 unsigned n_eq, unsigned n_ineq)
893 return (struct isl_basic_set *)
894 isl_basic_map_extend((struct isl_basic_map *)base,
895 nparam, 0, dim, extra, n_eq, n_ineq);
898 struct isl_basic_set *isl_basic_set_extend_constraints(
899 struct isl_basic_set *base, unsigned n_eq, unsigned n_ineq)
901 return (struct isl_basic_set *)
902 isl_basic_map_extend_constraints((struct isl_basic_map *)base,
903 n_eq, n_ineq);
906 struct isl_basic_set *isl_basic_set_cow(struct isl_basic_set *bset)
908 return (struct isl_basic_set *)
909 isl_basic_map_cow((struct isl_basic_map *)bset);
912 struct isl_basic_map *isl_basic_map_cow(struct isl_basic_map *bmap)
914 if (!bmap)
915 return NULL;
917 if (bmap->ref > 1) {
918 bmap->ref--;
919 bmap = isl_basic_map_dup(bmap);
921 ISL_F_CLR(bmap, ISL_BASIC_SET_FINAL);
922 return bmap;
925 struct isl_set *isl_set_cow(struct isl_set *set)
927 if (!set)
928 return NULL;
930 if (set->ref == 1)
931 return set;
932 set->ref--;
933 return isl_set_dup(set);
936 struct isl_map *isl_map_cow(struct isl_map *map)
938 if (!map)
939 return NULL;
941 if (map->ref == 1)
942 return map;
943 map->ref--;
944 return isl_map_dup(map);
947 static void swap_vars(struct isl_blk blk, isl_int *a,
948 unsigned a_len, unsigned b_len)
950 isl_seq_cpy(blk.data, a+a_len, b_len);
951 isl_seq_cpy(blk.data+b_len, a, a_len);
952 isl_seq_cpy(a, blk.data, b_len+a_len);
955 struct isl_basic_set *isl_basic_set_swap_vars(
956 struct isl_basic_set *bset, unsigned n)
958 int i;
959 struct isl_blk blk;
960 unsigned dim;
961 unsigned nparam;
963 if (!bset)
964 goto error;
966 nparam = isl_basic_set_n_param(bset);
967 dim = isl_basic_set_n_dim(bset);
968 isl_assert(bset->ctx, n <= dim, goto error);
970 if (n == dim)
971 return bset;
973 bset = isl_basic_set_cow(bset);
974 if (!bset)
975 return NULL;
977 blk = isl_blk_alloc(bset->ctx, dim);
978 if (isl_blk_is_error(blk))
979 goto error;
981 for (i = 0; i < bset->n_eq; ++i)
982 swap_vars(blk,
983 bset->eq[i]+1+nparam, n, dim - n);
985 for (i = 0; i < bset->n_ineq; ++i)
986 swap_vars(blk,
987 bset->ineq[i]+1+nparam, n, dim - n);
989 for (i = 0; i < bset->n_div; ++i)
990 swap_vars(blk,
991 bset->div[i]+1+1+nparam, n, dim - n);
993 isl_blk_free(bset->ctx, blk);
995 ISL_F_CLR(bset, ISL_BASIC_SET_NORMALIZED);
996 return bset;
998 error:
999 isl_basic_set_free(bset);
1000 return NULL;
1003 struct isl_set *isl_set_swap_vars(struct isl_set *set, unsigned n)
1005 int i;
1006 set = isl_set_cow(set);
1007 if (!set)
1008 return NULL;
1010 for (i = 0; i < set->n; ++i) {
1011 set->p[i] = isl_basic_set_swap_vars(set->p[i], n);
1012 if (!set->p[i]) {
1013 isl_set_free(set);
1014 return NULL;
1017 ISL_F_CLR(set, ISL_SET_NORMALIZED);
1018 return set;
1021 struct isl_basic_map *isl_basic_map_set_to_empty(struct isl_basic_map *bmap)
1023 int i = 0;
1024 unsigned total;
1025 if (!bmap)
1026 goto error;
1027 total = isl_basic_map_total_dim(bmap);
1028 isl_basic_map_free_div(bmap, bmap->n_div);
1029 isl_basic_map_free_inequality(bmap, bmap->n_ineq);
1030 if (bmap->n_eq > 0)
1031 isl_basic_map_free_equality(bmap, bmap->n_eq-1);
1032 else {
1033 isl_basic_map_alloc_equality(bmap);
1034 if (i < 0)
1035 goto error;
1037 isl_int_set_si(bmap->eq[i][0], 1);
1038 isl_seq_clr(bmap->eq[i]+1, total);
1039 ISL_F_SET(bmap, ISL_BASIC_MAP_EMPTY);
1040 return isl_basic_map_finalize(bmap);
1041 error:
1042 isl_basic_map_free(bmap);
1043 return NULL;
1046 struct isl_basic_set *isl_basic_set_set_to_empty(struct isl_basic_set *bset)
1048 return (struct isl_basic_set *)
1049 isl_basic_map_set_to_empty((struct isl_basic_map *)bset);
1052 static void swap_div(struct isl_basic_map *bmap, int a, int b)
1054 int i;
1055 unsigned off = isl_dim_total(bmap->dim);
1056 isl_int *t = bmap->div[a];
1057 bmap->div[a] = bmap->div[b];
1058 bmap->div[b] = t;
1060 for (i = 0; i < bmap->n_eq; ++i)
1061 isl_int_swap(bmap->eq[i][1+off+a], bmap->eq[i][1+off+b]);
1063 for (i = 0; i < bmap->n_ineq; ++i)
1064 isl_int_swap(bmap->ineq[i][1+off+a], bmap->ineq[i][1+off+b]);
1066 for (i = 0; i < bmap->n_div; ++i)
1067 isl_int_swap(bmap->div[i][1+1+off+a], bmap->div[i][1+1+off+b]);
1068 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
1071 /* Eliminate the specified n dimensions starting at first from the
1072 * constraints using Fourier-Motzkin, The dimensions themselves
1073 * are not removed.
1075 struct isl_set *isl_set_eliminate_dims(struct isl_set *set,
1076 unsigned first, unsigned n)
1078 int i;
1079 unsigned nparam;
1081 if (!set)
1082 return NULL;
1083 if (n == 0)
1084 return set;
1086 set = isl_set_cow(set);
1087 if (!set)
1088 return NULL;
1089 isl_assert(set->ctx, first+n <= isl_set_n_dim(set), goto error);
1090 nparam = isl_set_n_param(set);
1092 for (i = 0; i < set->n; ++i) {
1093 set->p[i] = isl_basic_set_eliminate_vars(set->p[i],
1094 nparam + first, n);
1095 if (!set->p[i])
1096 goto error;
1098 return set;
1099 error:
1100 isl_set_free(set);
1101 return NULL;
1104 /* Project out n dimensions starting at first using Fourier-Motzkin */
1105 struct isl_set *isl_set_remove_dims(struct isl_set *set,
1106 unsigned first, unsigned n)
1108 set = isl_set_eliminate_dims(set, first, n);
1109 set = isl_set_drop_dims(set, first, n);
1110 return set;
1113 struct isl_basic_set *isl_basic_set_remove_divs(struct isl_basic_set *bset)
1115 bset = isl_basic_set_eliminate_vars(bset, isl_dim_total(bset->dim),
1116 bset->n_div);
1117 if (!bset)
1118 return NULL;
1119 bset->n_div = 0;
1120 return bset;
1123 struct isl_set *isl_set_remove_divs(struct isl_set *set)
1125 int i;
1127 if (!set)
1128 return NULL;
1129 if (set->n == 0)
1130 return set;
1132 set = isl_set_cow(set);
1133 if (!set)
1134 return NULL;
1136 for (i = 0; i < set->n; ++i) {
1137 set->p[i] = isl_basic_set_remove_divs(set->p[i]);
1138 if (!set->p[i])
1139 goto error;
1141 return set;
1142 error:
1143 isl_set_free(set);
1144 return NULL;
1147 struct isl_basic_map *isl_basic_map_remove(struct isl_basic_map *bmap,
1148 enum isl_dim_type type, unsigned first, unsigned n)
1150 if (!bmap)
1151 return NULL;
1152 isl_assert(bmap->ctx, first + n <= isl_basic_map_dim(bmap, type),
1153 goto error);
1154 if (n == 0)
1155 return bmap;
1156 bmap = isl_basic_map_eliminate_vars(bmap,
1157 isl_basic_map_offset(bmap, type) - 1 + first, n);
1158 bmap = isl_basic_map_drop(bmap, type, first, n);
1159 return bmap;
1160 error:
1161 isl_basic_map_free(bmap);
1162 return NULL;
1165 struct isl_map *isl_map_remove(struct isl_map *map,
1166 enum isl_dim_type type, unsigned first, unsigned n)
1168 int i;
1169 unsigned nparam;
1171 if (n == 0)
1172 return map;
1174 map = isl_map_cow(map);
1175 if (!map)
1176 return NULL;
1177 isl_assert(map->ctx, first + n <= isl_map_dim(map, type), goto error);
1179 for (i = 0; i < map->n; ++i) {
1180 map->p[i] = isl_basic_map_eliminate_vars(map->p[i],
1181 isl_basic_map_offset(map->p[i], type) - 1 + first, n);
1182 if (!map->p[i])
1183 goto error;
1185 map = isl_map_drop(map, type, first, n);
1186 return map;
1187 error:
1188 isl_map_free(map);
1189 return NULL;
1192 /* Project out n inputs starting at first using Fourier-Motzkin */
1193 struct isl_map *isl_map_remove_inputs(struct isl_map *map,
1194 unsigned first, unsigned n)
1196 return isl_map_remove(map, isl_dim_in, first, n);
1199 /* Project out n dimensions starting at first using Fourier-Motzkin */
1200 struct isl_basic_set *isl_basic_set_remove_dims(struct isl_basic_set *bset,
1201 unsigned first, unsigned n)
1203 unsigned nparam = isl_basic_set_n_param(bset);
1204 bset = isl_basic_set_eliminate_vars(bset, nparam + first, n);
1205 bset = isl_basic_set_drop_dims(bset, first, n);
1206 return bset;
1209 static void dump_term(struct isl_basic_map *bmap,
1210 isl_int c, int pos, FILE *out)
1212 const char *name;
1213 unsigned in = isl_basic_map_n_in(bmap);
1214 unsigned dim = in + isl_basic_map_n_out(bmap);
1215 unsigned nparam = isl_basic_map_n_param(bmap);
1216 if (!pos)
1217 isl_int_print(out, c, 0);
1218 else {
1219 if (!isl_int_is_one(c))
1220 isl_int_print(out, c, 0);
1221 if (pos < 1 + nparam) {
1222 name = isl_dim_get_name(bmap->dim,
1223 isl_dim_param, pos - 1);
1224 if (name)
1225 fprintf(out, "%s", name);
1226 else
1227 fprintf(out, "p%d", pos - 1);
1228 } else if (pos < 1 + nparam + in)
1229 fprintf(out, "i%d", pos - 1 - nparam);
1230 else if (pos < 1 + nparam + dim)
1231 fprintf(out, "o%d", pos - 1 - nparam - in);
1232 else
1233 fprintf(out, "e%d", pos - 1 - nparam - dim);
1237 static void dump_constraint_sign(struct isl_basic_map *bmap, isl_int *c,
1238 int sign, FILE *out)
1240 int i;
1241 int first;
1242 unsigned len = 1 + isl_basic_map_total_dim(bmap);
1243 isl_int v;
1245 isl_int_init(v);
1246 for (i = 0, first = 1; i < len; ++i) {
1247 if (isl_int_sgn(c[i]) * sign <= 0)
1248 continue;
1249 if (!first)
1250 fprintf(out, " + ");
1251 first = 0;
1252 isl_int_abs(v, c[i]);
1253 dump_term(bmap, v, i, out);
1255 isl_int_clear(v);
1256 if (first)
1257 fprintf(out, "0");
1260 static void dump_constraint(struct isl_basic_map *bmap, isl_int *c,
1261 const char *op, FILE *out, int indent)
1263 int i;
1265 fprintf(out, "%*s", indent, "");
1267 dump_constraint_sign(bmap, c, 1, out);
1268 fprintf(out, " %s ", op);
1269 dump_constraint_sign(bmap, c, -1, out);
1271 fprintf(out, "\n");
1273 for (i = bmap->n_div; i < bmap->extra; ++i) {
1274 if (isl_int_is_zero(c[1+isl_dim_total(bmap->dim)+i]))
1275 continue;
1276 fprintf(out, "%*s", indent, "");
1277 fprintf(out, "ERROR: unused div coefficient not zero\n");
1278 abort();
1282 static void dump_constraints(struct isl_basic_map *bmap,
1283 isl_int **c, unsigned n,
1284 const char *op, FILE *out, int indent)
1286 int i;
1288 for (i = 0; i < n; ++i)
1289 dump_constraint(bmap, c[i], op, out, indent);
1292 static void dump_affine(struct isl_basic_map *bmap, isl_int *exp, FILE *out)
1294 int j;
1295 int first = 1;
1296 unsigned total = isl_basic_map_total_dim(bmap);
1298 for (j = 0; j < 1 + total; ++j) {
1299 if (isl_int_is_zero(exp[j]))
1300 continue;
1301 if (!first && isl_int_is_pos(exp[j]))
1302 fprintf(out, "+");
1303 dump_term(bmap, exp[j], j, out);
1304 first = 0;
1308 static void dump(struct isl_basic_map *bmap, FILE *out, int indent)
1310 int i;
1312 dump_constraints(bmap, bmap->eq, bmap->n_eq, "=", out, indent);
1313 dump_constraints(bmap, bmap->ineq, bmap->n_ineq, ">=", out, indent);
1315 for (i = 0; i < bmap->n_div; ++i) {
1316 fprintf(out, "%*s", indent, "");
1317 fprintf(out, "e%d = [(", i);
1318 dump_affine(bmap, bmap->div[i]+1, out);
1319 fprintf(out, ")/");
1320 isl_int_print(out, bmap->div[i][0], 0);
1321 fprintf(out, "]\n");
1325 void isl_basic_set_dump(struct isl_basic_set *bset, FILE *out, int indent)
1327 if (!bset) {
1328 fprintf(out, "null basic set\n");
1329 return;
1332 fprintf(out, "%*s", indent, "");
1333 fprintf(out, "ref: %d, nparam: %d, dim: %d, extra: %d, flags: %x\n",
1334 bset->ref, bset->dim->nparam, bset->dim->n_out,
1335 bset->extra, bset->flags);
1336 dump((struct isl_basic_map *)bset, out, indent);
1339 void isl_basic_map_dump(struct isl_basic_map *bmap, FILE *out, int indent)
1341 if (!bmap) {
1342 fprintf(out, "null basic map\n");
1343 return;
1346 fprintf(out, "%*s", indent, "");
1347 fprintf(out, "ref: %d, nparam: %d, in: %d, out: %d, extra: %d, "
1348 "flags: %x, n_name: %d\n",
1349 bmap->ref,
1350 bmap->dim->nparam, bmap->dim->n_in, bmap->dim->n_out,
1351 bmap->extra, bmap->flags, bmap->dim->n_name);
1352 dump(bmap, out, indent);
1355 int isl_inequality_negate(struct isl_basic_map *bmap, unsigned pos)
1357 unsigned total;
1358 if (!bmap)
1359 return -1;
1360 total = isl_basic_map_total_dim(bmap);
1361 isl_assert(bmap->ctx, pos < bmap->n_ineq, return -1);
1362 isl_seq_neg(bmap->ineq[pos], bmap->ineq[pos], 1 + total);
1363 isl_int_sub_ui(bmap->ineq[pos][0], bmap->ineq[pos][0], 1);
1364 ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
1365 return 0;
1368 struct isl_set *isl_set_alloc_dim(struct isl_dim *dim, int n, unsigned flags)
1370 struct isl_set *set;
1372 if (!dim)
1373 return NULL;
1374 isl_assert(dim->ctx, dim->n_in == 0, return NULL);
1375 isl_assert(dim->ctx, n >= 0, return NULL);
1376 set = isl_alloc(dim->ctx, struct isl_set,
1377 sizeof(struct isl_set) +
1378 n * sizeof(struct isl_basic_set *));
1379 if (!set)
1380 goto error;
1382 set->ctx = dim->ctx;
1383 isl_ctx_ref(set->ctx);
1384 set->ref = 1;
1385 set->size = n;
1386 set->n = 0;
1387 set->dim = dim;
1388 set->flags = flags;
1389 return set;
1390 error:
1391 isl_dim_free(dim);
1392 return NULL;
1395 struct isl_set *isl_set_alloc(struct isl_ctx *ctx,
1396 unsigned nparam, unsigned dim, int n, unsigned flags)
1398 struct isl_set *set;
1399 struct isl_dim *dims;
1401 dims = isl_dim_alloc(ctx, nparam, 0, dim);
1402 if (!dims)
1403 return NULL;
1405 set = isl_set_alloc_dim(dims, n, flags);
1406 return set;
1409 struct isl_set *isl_set_dup(struct isl_set *set)
1411 int i;
1412 struct isl_set *dup;
1414 if (!set)
1415 return NULL;
1417 dup = isl_set_alloc_dim(isl_dim_copy(set->dim), set->n, set->flags);
1418 if (!dup)
1419 return NULL;
1420 for (i = 0; i < set->n; ++i)
1421 dup = isl_set_add(dup, isl_basic_set_copy(set->p[i]));
1422 return dup;
1425 struct isl_set *isl_set_from_basic_set(struct isl_basic_set *bset)
1427 struct isl_set *set;
1429 if (!bset)
1430 return NULL;
1432 set = isl_set_alloc_dim(isl_dim_copy(bset->dim), 1, ISL_MAP_DISJOINT);
1433 if (!set) {
1434 isl_basic_set_free(bset);
1435 return NULL;
1437 return isl_set_add(set, bset);
1440 struct isl_map *isl_map_from_basic_map(struct isl_basic_map *bmap)
1442 struct isl_map *map;
1444 if (!bmap)
1445 return NULL;
1447 map = isl_map_alloc_dim(isl_dim_copy(bmap->dim), 1, ISL_MAP_DISJOINT);
1448 if (!map) {
1449 isl_basic_map_free(bmap);
1450 return NULL;
1452 return isl_map_add(map, bmap);
1455 struct isl_set *isl_set_add(struct isl_set *set, struct isl_basic_set *bset)
1457 if (!bset || !set)
1458 goto error;
1459 isl_assert(set->ctx, isl_dim_equal(set->dim, bset->dim), goto error);
1460 isl_assert(set->ctx, set->n < set->size, goto error);
1461 set->p[set->n] = bset;
1462 set->n++;
1463 return set;
1464 error:
1465 if (set)
1466 isl_set_free(set);
1467 if (bset)
1468 isl_basic_set_free(bset);
1469 return NULL;
1472 void isl_set_free(struct isl_set *set)
1474 int i;
1476 if (!set)
1477 return;
1479 if (--set->ref > 0)
1480 return;
1482 isl_ctx_deref(set->ctx);
1483 for (i = 0; i < set->n; ++i)
1484 isl_basic_set_free(set->p[i]);
1485 isl_dim_free(set->dim);
1486 free(set);
1489 void isl_set_dump(struct isl_set *set, FILE *out, int indent)
1491 int i;
1493 if (!set) {
1494 fprintf(out, "null set\n");
1495 return;
1498 fprintf(out, "%*s", indent, "");
1499 fprintf(out, "ref: %d, n: %d, nparam: %d, dim: %d, flags: %x\n",
1500 set->ref, set->n, set->dim->nparam, set->dim->n_out,
1501 set->flags);
1502 for (i = 0; i < set->n; ++i) {
1503 fprintf(out, "%*s", indent, "");
1504 fprintf(out, "basic set %d:\n", i);
1505 isl_basic_set_dump(set->p[i], out, indent+4);
1509 void isl_map_dump(struct isl_map *map, FILE *out, int indent)
1511 int i;
1513 if (!map) {
1514 fprintf(out, "null map\n");
1515 return;
1518 fprintf(out, "%*s", indent, "");
1519 fprintf(out, "ref: %d, n: %d, nparam: %d, in: %d, out: %d, "
1520 "flags: %x, n_name: %d\n",
1521 map->ref, map->n, map->dim->nparam, map->dim->n_in,
1522 map->dim->n_out, map->flags, map->dim->n_name);
1523 for (i = 0; i < map->n; ++i) {
1524 fprintf(out, "%*s", indent, "");
1525 fprintf(out, "basic map %d:\n", i);
1526 isl_basic_map_dump(map->p[i], out, indent+4);
1530 struct isl_basic_map *isl_basic_map_intersect_domain(
1531 struct isl_basic_map *bmap, struct isl_basic_set *bset)
1533 struct isl_basic_map *bmap_domain;
1534 struct isl_dim *dim;
1536 if (!bmap || !bset)
1537 goto error;
1539 isl_assert(bset->ctx, isl_dim_match(bmap->dim, isl_dim_param,
1540 bset->dim, isl_dim_param), goto error);
1542 if (isl_dim_size(bset->dim, isl_dim_set) != 0)
1543 isl_assert(bset->ctx,
1544 isl_basic_map_compatible_domain(bmap, bset), goto error);
1546 bmap = isl_basic_map_cow(bmap);
1547 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1548 bset->n_div, bset->n_eq, bset->n_ineq);
1549 if (!bmap)
1550 goto error;
1551 dim = isl_dim_reverse(isl_dim_copy(bset->dim));
1552 bmap_domain = isl_basic_map_from_basic_set(bset, dim);
1553 bmap = add_constraints(bmap, bmap_domain, 0, 0);
1555 bmap = isl_basic_map_simplify(bmap);
1556 return isl_basic_map_finalize(bmap);
1557 error:
1558 isl_basic_map_free(bmap);
1559 isl_basic_set_free(bset);
1560 return NULL;
1563 struct isl_basic_map *isl_basic_map_intersect_range(
1564 struct isl_basic_map *bmap, struct isl_basic_set *bset)
1566 struct isl_basic_map *bmap_range;
1568 if (!bmap || !bset)
1569 goto error;
1571 isl_assert(bset->ctx, isl_dim_match(bmap->dim, isl_dim_param,
1572 bset->dim, isl_dim_param), goto error);
1574 if (isl_dim_size(bset->dim, isl_dim_set) != 0)
1575 isl_assert(bset->ctx,
1576 isl_basic_map_compatible_range(bmap, bset), goto error);
1578 bmap = isl_basic_map_cow(bmap);
1579 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1580 bset->n_div, bset->n_eq, bset->n_ineq);
1581 if (!bmap)
1582 goto error;
1583 bmap_range = isl_basic_map_from_basic_set(bset, isl_dim_copy(bset->dim));
1584 bmap = add_constraints(bmap, bmap_range, 0, 0);
1586 bmap = isl_basic_map_simplify(bmap);
1587 return isl_basic_map_finalize(bmap);
1588 error:
1589 isl_basic_map_free(bmap);
1590 isl_basic_set_free(bset);
1591 return NULL;
1594 static int basic_map_contains(struct isl_basic_map *bmap, struct isl_vec *vec)
1596 int i;
1597 unsigned total;
1598 isl_int s;
1600 total = 1 + isl_basic_map_total_dim(bmap);
1601 if (total != vec->size)
1602 return -1;
1604 isl_int_init(s);
1606 for (i = 0; i < bmap->n_eq; ++i) {
1607 isl_seq_inner_product(vec->block.data, bmap->eq[i], total, &s);
1608 if (!isl_int_is_zero(s)) {
1609 isl_int_clear(s);
1610 return 0;
1614 for (i = 0; i < bmap->n_ineq; ++i) {
1615 isl_seq_inner_product(vec->block.data, bmap->ineq[i], total, &s);
1616 if (isl_int_is_neg(s)) {
1617 isl_int_clear(s);
1618 return 0;
1622 isl_int_clear(s);
1624 return 1;
1627 int isl_basic_set_contains(struct isl_basic_set *bset, struct isl_vec *vec)
1629 return basic_map_contains((struct isl_basic_map *)bset, vec);
1632 struct isl_basic_map *isl_basic_map_intersect(
1633 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1635 struct isl_vec *sample = NULL;
1637 if (!bmap1 || !bmap2)
1638 goto error;
1640 isl_assert(bmap1->ctx, isl_dim_match(bmap1->dim, isl_dim_param,
1641 bmap2->dim, isl_dim_param), goto error);
1642 if (isl_dim_total(bmap1->dim) ==
1643 isl_dim_size(bmap1->dim, isl_dim_param) &&
1644 isl_dim_total(bmap2->dim) !=
1645 isl_dim_size(bmap2->dim, isl_dim_param))
1646 return isl_basic_map_intersect(bmap2, bmap1);
1648 if (isl_dim_total(bmap2->dim) !=
1649 isl_dim_size(bmap2->dim, isl_dim_param))
1650 isl_assert(bmap1->ctx,
1651 isl_dim_equal(bmap1->dim, bmap2->dim), goto error);
1653 if (bmap1->sample &&
1654 basic_map_contains(bmap1, bmap1->sample) > 0 &&
1655 basic_map_contains(bmap2, bmap1->sample) > 0)
1656 sample = isl_vec_copy(bmap1->ctx, bmap1->sample);
1657 else if (bmap2->sample &&
1658 basic_map_contains(bmap1, bmap2->sample) > 0 &&
1659 basic_map_contains(bmap2, bmap2->sample) > 0)
1660 sample = isl_vec_copy(bmap2->ctx, bmap2->sample);
1662 bmap1 = isl_basic_map_cow(bmap1);
1663 bmap1 = isl_basic_map_extend_dim(bmap1, isl_dim_copy(bmap1->dim),
1664 bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
1665 if (!bmap1)
1666 goto error;
1667 bmap1 = add_constraints(bmap1, bmap2, 0, 0);
1669 if (sample) {
1670 isl_vec_free(bmap1->ctx, bmap1->sample);
1671 bmap1->sample = sample;
1674 bmap1 = isl_basic_map_simplify(bmap1);
1675 return isl_basic_map_finalize(bmap1);
1676 error:
1677 if (sample)
1678 isl_vec_free(bmap1->ctx, sample);
1679 isl_basic_map_free(bmap1);
1680 isl_basic_map_free(bmap2);
1681 return NULL;
1684 struct isl_basic_set *isl_basic_set_intersect(
1685 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
1687 return (struct isl_basic_set *)
1688 isl_basic_map_intersect(
1689 (struct isl_basic_map *)bset1,
1690 (struct isl_basic_map *)bset2);
1693 struct isl_map *isl_map_intersect(struct isl_map *map1, struct isl_map *map2)
1695 unsigned flags = 0;
1696 struct isl_map *result;
1697 int i, j;
1699 if (!map1 || !map2)
1700 goto error;
1702 isl_assert(map1->ctx, isl_dim_match(map1->dim, isl_dim_param,
1703 map2->dim, isl_dim_param), goto error);
1704 if (isl_dim_total(map1->dim) ==
1705 isl_dim_size(map1->dim, isl_dim_param) &&
1706 isl_dim_total(map2->dim) != isl_dim_size(map2->dim, isl_dim_param))
1707 return isl_map_intersect(map2, map1);
1709 if (isl_dim_total(map2->dim) != isl_dim_size(map2->dim, isl_dim_param))
1710 isl_assert(map1->ctx,
1711 isl_dim_equal(map1->dim, map2->dim), goto error);
1713 if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
1714 ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
1715 ISL_FL_SET(flags, ISL_MAP_DISJOINT);
1717 result = isl_map_alloc_dim(isl_dim_copy(map1->dim),
1718 map1->n * map2->n, flags);
1719 if (!result)
1720 goto error;
1721 for (i = 0; i < map1->n; ++i)
1722 for (j = 0; j < map2->n; ++j) {
1723 struct isl_basic_map *part;
1724 part = isl_basic_map_intersect(
1725 isl_basic_map_copy(map1->p[i]),
1726 isl_basic_map_copy(map2->p[j]));
1727 if (isl_basic_map_is_empty(part))
1728 isl_basic_map_free(part);
1729 else
1730 result = isl_map_add(result, part);
1731 if (!result)
1732 goto error;
1734 isl_map_free(map1);
1735 isl_map_free(map2);
1736 return result;
1737 error:
1738 isl_map_free(map1);
1739 isl_map_free(map2);
1740 return NULL;
1743 struct isl_set *isl_set_intersect(struct isl_set *set1, struct isl_set *set2)
1745 return (struct isl_set *)
1746 isl_map_intersect((struct isl_map *)set1,
1747 (struct isl_map *)set2);
1750 struct isl_basic_map *isl_basic_map_reverse(struct isl_basic_map *bmap)
1752 struct isl_dim *dim;
1753 struct isl_basic_set *bset;
1754 unsigned in;
1756 if (!bmap)
1757 return NULL;
1758 bmap = isl_basic_map_cow(bmap);
1759 if (!bmap)
1760 return NULL;
1761 dim = isl_dim_reverse(isl_dim_copy(bmap->dim));
1762 in = isl_basic_map_n_in(bmap);
1763 bset = isl_basic_set_from_basic_map(bmap);
1764 bset = isl_basic_set_swap_vars(bset, in);
1765 return isl_basic_map_from_basic_set(bset, dim);
1768 /* Turn final n dimensions into existentially quantified variables.
1770 struct isl_basic_set *isl_basic_set_project_out(
1771 struct isl_basic_set *bset, unsigned n, unsigned flags)
1773 int i;
1774 size_t row_size;
1775 isl_int **new_div;
1776 isl_int *old;
1778 if (!bset)
1779 return NULL;
1781 isl_assert(bset->ctx, n <= isl_basic_set_n_dim(bset), goto error);
1783 if (n == 0)
1784 return bset;
1786 bset = isl_basic_set_cow(bset);
1788 row_size = 1 + isl_dim_total(bset->dim) + bset->extra;
1789 old = bset->block2.data;
1790 bset->block2 = isl_blk_extend(bset->ctx, bset->block2,
1791 (bset->extra + n) * (1 + row_size));
1792 if (!bset->block2.data)
1793 goto error;
1794 new_div = isl_alloc_array(ctx, isl_int *, bset->extra + n);
1795 if (!new_div)
1796 goto error;
1797 for (i = 0; i < n; ++i) {
1798 new_div[i] = bset->block2.data +
1799 (bset->extra + i) * (1 + row_size);
1800 isl_seq_clr(new_div[i], 1 + row_size);
1802 for (i = 0; i < bset->extra; ++i)
1803 new_div[n + i] = bset->block2.data + (bset->div[i] - old);
1804 free(bset->div);
1805 bset->div = new_div;
1806 bset->n_div += n;
1807 bset->extra += n;
1808 bset->dim = isl_dim_drop_outputs(bset->dim,
1809 isl_basic_set_n_dim(bset) - n, n);
1810 if (!bset->dim)
1811 goto error;
1812 bset = isl_basic_set_simplify(bset);
1813 return isl_basic_set_finalize(bset);
1814 error:
1815 isl_basic_set_free(bset);
1816 return NULL;
1819 struct isl_basic_map *add_divs(struct isl_basic_map *bmap, unsigned n)
1821 int i, j;
1823 for (i = 0; i < n; ++i) {
1824 j = isl_basic_map_alloc_div(bmap);
1825 if (j < 0)
1826 goto error;
1827 isl_seq_clr(bmap->div[j], 1+1+isl_basic_map_total_dim(bmap));
1829 return bmap;
1830 error:
1831 isl_basic_map_free(bmap);
1832 return NULL;
1835 struct isl_basic_map *isl_basic_map_apply_range(
1836 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1838 struct isl_dim *dim_result = NULL;
1839 struct isl_basic_map *bmap;
1840 unsigned n_in, n_out, n, nparam, total, pos;
1841 struct isl_dim_map *dim_map1, *dim_map2;
1843 if (!bmap1 || !bmap2)
1844 goto error;
1846 dim_result = isl_dim_join(isl_dim_copy(bmap1->dim),
1847 isl_dim_copy(bmap2->dim));
1849 n_in = isl_basic_map_n_in(bmap1);
1850 n_out = isl_basic_map_n_out(bmap2);
1851 n = isl_basic_map_n_out(bmap1);
1852 nparam = isl_basic_map_n_param(bmap1);
1854 total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + n;
1855 dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
1856 dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
1857 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
1858 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
1859 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
1860 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_in);
1861 isl_dim_map_div(dim_map1, bmap1, pos += n_out);
1862 isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
1863 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
1864 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
1866 bmap = isl_basic_map_alloc_dim(dim_result,
1867 bmap1->n_div + bmap2->n_div + n,
1868 bmap1->n_eq + bmap2->n_eq,
1869 bmap1->n_ineq + bmap2->n_ineq);
1870 bmap = add_constraints_dim_map(bmap, bmap1, dim_map1);
1871 bmap = add_constraints_dim_map(bmap, bmap2, dim_map2);
1872 bmap = add_divs(bmap, n);
1873 bmap = isl_basic_map_simplify(bmap);
1874 return isl_basic_map_finalize(bmap);
1875 error:
1876 isl_basic_map_free(bmap1);
1877 isl_basic_map_free(bmap2);
1878 return NULL;
1881 struct isl_basic_set *isl_basic_set_apply(
1882 struct isl_basic_set *bset, struct isl_basic_map *bmap)
1884 if (!bset || !bmap)
1885 goto error;
1887 isl_assert(set->ctx, isl_basic_map_compatible_domain(bmap, bset),
1888 goto error);
1890 return (struct isl_basic_set *)
1891 isl_basic_map_apply_range((struct isl_basic_map *)bset, bmap);
1892 error:
1893 isl_basic_set_free(bset);
1894 isl_basic_map_free(bmap);
1895 return NULL;
1898 struct isl_basic_map *isl_basic_map_apply_domain(
1899 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1901 if (!bmap1 || !bmap2)
1902 goto error;
1904 isl_assert(ctx,
1905 isl_basic_map_n_in(bmap1) == isl_basic_map_n_in(bmap2), goto error);
1906 isl_assert(ctx,
1907 isl_basic_map_n_param(bmap1) == isl_basic_map_n_param(bmap2),
1908 goto error);
1910 bmap1 = isl_basic_map_reverse(bmap1);
1911 bmap1 = isl_basic_map_apply_range(bmap1, bmap2);
1912 return isl_basic_map_reverse(bmap1);
1913 error:
1914 isl_basic_map_free(bmap1);
1915 isl_basic_map_free(bmap2);
1916 return NULL;
1919 /* Given two basic maps A -> f(A) and B -> g(B), construct a basic map
1920 * A \cap B -> f(A) + f(B)
1922 struct isl_basic_map *isl_basic_map_sum(
1923 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1925 unsigned n_in, n_out, nparam, total, pos;
1926 struct isl_basic_map *bmap = NULL;
1927 struct isl_dim_map *dim_map1, *dim_map2;
1928 int i;
1930 if (!bmap1 || !bmap2)
1931 goto error;
1933 isl_assert(bmap1->ctx, isl_dim_equal(bmap1->dim, bmap2->dim),
1934 goto error);
1936 nparam = isl_basic_map_n_param(bmap1);
1937 n_in = isl_basic_map_n_in(bmap1);
1938 n_out = isl_basic_map_n_out(bmap1);
1940 total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + 2 * n_out;
1941 dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
1942 dim_map2 = isl_dim_map_alloc(bmap2->ctx, total);
1943 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
1944 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos);
1945 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
1946 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
1947 isl_dim_map_div(dim_map1, bmap1, pos += n_in + n_out);
1948 isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
1949 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
1950 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_out);
1952 bmap = isl_basic_map_alloc_dim(isl_dim_copy(bmap1->dim),
1953 bmap1->n_div + bmap2->n_div + 2 * n_out,
1954 bmap1->n_eq + bmap2->n_eq + n_out,
1955 bmap1->n_ineq + bmap2->n_ineq);
1956 for (i = 0; i < n_out; ++i) {
1957 int j = isl_basic_map_alloc_equality(bmap);
1958 if (j < 0)
1959 goto error;
1960 isl_seq_clr(bmap->eq[j], 1+total);
1961 isl_int_set_si(bmap->eq[j][1+nparam+n_in+i], -1);
1962 isl_int_set_si(bmap->eq[j][1+pos+i], 1);
1963 isl_int_set_si(bmap->eq[j][1+pos-n_out+i], 1);
1965 bmap = add_constraints_dim_map(bmap, bmap1, dim_map1);
1966 bmap = add_constraints_dim_map(bmap, bmap2, dim_map2);
1967 bmap = add_divs(bmap, 2 * n_out);
1969 bmap = isl_basic_map_simplify(bmap);
1970 return isl_basic_map_finalize(bmap);
1971 error:
1972 isl_basic_map_free(bmap);
1973 isl_basic_map_free(bmap1);
1974 isl_basic_map_free(bmap2);
1975 return NULL;
1978 /* Given a basic map A -> f(A), construct A -> -f(A).
1980 struct isl_basic_map *isl_basic_map_neg(struct isl_basic_map *bmap)
1982 int i, j;
1983 unsigned off, n;
1985 bmap = isl_basic_map_cow(bmap);
1986 if (!bmap)
1987 return NULL;
1989 n = isl_basic_map_dim(bmap, isl_dim_out);
1990 off = isl_basic_map_offset(bmap, isl_dim_out);
1991 for (i = 0; i < bmap->n_eq; ++i)
1992 for (j = 0; j < n; ++j)
1993 isl_int_neg(bmap->eq[i][off+j], bmap->eq[i][off+j]);
1994 for (i = 0; i < bmap->n_ineq; ++i)
1995 for (j = 0; j < n; ++j)
1996 isl_int_neg(bmap->ineq[i][off+j], bmap->ineq[i][off+j]);
1997 for (i = 0; i < bmap->n_div; ++i)
1998 for (j = 0; j < n; ++j)
1999 isl_int_neg(bmap->div[i][1+off+j], bmap->div[i][1+off+j]);
2000 return isl_basic_map_finalize(bmap);
2003 /* Given a basic map A -> f(A) and an integer d, construct a basic map
2004 * A -> floor(f(A)/d).
2006 struct isl_basic_map *isl_basic_map_floordiv(struct isl_basic_map *bmap,
2007 isl_int d)
2009 unsigned n_in, n_out, nparam, total, pos;
2010 struct isl_basic_map *result = NULL;
2011 struct isl_dim_map *dim_map;
2012 int i;
2014 if (!bmap)
2015 return NULL;
2017 nparam = isl_basic_map_n_param(bmap);
2018 n_in = isl_basic_map_n_in(bmap);
2019 n_out = isl_basic_map_n_out(bmap);
2021 total = nparam + n_in + n_out + bmap->n_div + n_out;
2022 dim_map = isl_dim_map_alloc(bmap->ctx, total);
2023 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_param, pos = 0);
2024 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_in, pos += nparam);
2025 isl_dim_map_div(dim_map, bmap, pos += n_in + n_out);
2026 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_out, pos += bmap->n_div);
2028 result = isl_basic_map_alloc_dim(isl_dim_copy(bmap->dim),
2029 bmap->n_div + n_out,
2030 bmap->n_eq, bmap->n_ineq + 2 * n_out);
2031 result = add_constraints_dim_map(result, bmap, dim_map);
2032 result = add_divs(result, n_out);
2033 for (i = 0; i < n_out; ++i) {
2034 int j;
2035 j = isl_basic_map_alloc_inequality(result);
2036 if (j < 0)
2037 goto error;
2038 isl_seq_clr(result->ineq[j], 1+total);
2039 isl_int_neg(result->ineq[j][1+nparam+n_in+i], d);
2040 isl_int_set_si(result->ineq[j][1+pos+i], 1);
2041 j = isl_basic_map_alloc_inequality(result);
2042 if (j < 0)
2043 goto error;
2044 isl_seq_clr(result->ineq[j], 1+total);
2045 isl_int_set(result->ineq[j][1+nparam+n_in+i], d);
2046 isl_int_set_si(result->ineq[j][1+pos+i], -1);
2047 isl_int_sub_ui(result->ineq[j][0], d, 1);
2050 result = isl_basic_map_simplify(result);
2051 return isl_basic_map_finalize(result);
2052 error:
2053 isl_basic_map_free(result);
2054 return NULL;
2057 static struct isl_basic_map *var_equal(struct isl_basic_map *bmap, unsigned pos)
2059 int i;
2060 unsigned nparam;
2061 unsigned n_in;
2063 i = isl_basic_map_alloc_equality(bmap);
2064 if (i < 0)
2065 goto error;
2066 nparam = isl_basic_map_n_param(bmap);
2067 n_in = isl_basic_map_n_in(bmap);
2068 isl_seq_clr(bmap->eq[i], 1 + isl_basic_map_total_dim(bmap));
2069 isl_int_set_si(bmap->eq[i][1+nparam+pos], -1);
2070 isl_int_set_si(bmap->eq[i][1+nparam+n_in+pos], 1);
2071 return isl_basic_map_finalize(bmap);
2072 error:
2073 isl_basic_map_free(bmap);
2074 return NULL;
2077 static struct isl_basic_map *var_more(struct isl_basic_map *bmap, unsigned pos)
2079 int i;
2080 unsigned nparam;
2081 unsigned n_in;
2083 i = isl_basic_map_alloc_inequality(bmap);
2084 if (i < 0)
2085 goto error;
2086 nparam = isl_basic_map_n_param(bmap);
2087 n_in = isl_basic_map_n_in(bmap);
2088 isl_seq_clr(bmap->ineq[i], 1 + isl_basic_map_total_dim(bmap));
2089 isl_int_set_si(bmap->ineq[i][0], -1);
2090 isl_int_set_si(bmap->ineq[i][1+nparam+pos], -1);
2091 isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], 1);
2092 return isl_basic_map_finalize(bmap);
2093 error:
2094 isl_basic_map_free(bmap);
2095 return NULL;
2098 static struct isl_basic_map *var_less(struct isl_basic_map *bmap, unsigned pos)
2100 int i;
2101 unsigned nparam;
2102 unsigned n_in;
2104 i = isl_basic_map_alloc_inequality(bmap);
2105 if (i < 0)
2106 goto error;
2107 nparam = isl_basic_map_n_param(bmap);
2108 n_in = isl_basic_map_n_in(bmap);
2109 isl_seq_clr(bmap->ineq[i], 1 + isl_basic_map_total_dim(bmap));
2110 isl_int_set_si(bmap->ineq[i][0], -1);
2111 isl_int_set_si(bmap->ineq[i][1+nparam+pos], 1);
2112 isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], -1);
2113 return isl_basic_map_finalize(bmap);
2114 error:
2115 isl_basic_map_free(bmap);
2116 return NULL;
2119 struct isl_basic_map *isl_basic_map_equal(struct isl_dim *dim, unsigned n_equal)
2121 int i;
2122 struct isl_basic_map *bmap;
2123 bmap = isl_basic_map_alloc_dim(dim, 0, n_equal, 0);
2124 if (!bmap)
2125 return NULL;
2126 for (i = 0; i < n_equal && bmap; ++i)
2127 bmap = var_equal(bmap, i);
2128 return isl_basic_map_finalize(bmap);
2131 struct isl_basic_map *isl_basic_map_less_at(struct isl_dim *dim, unsigned pos)
2133 int i;
2134 struct isl_basic_map *bmap;
2135 bmap = isl_basic_map_alloc_dim(dim, 0, pos, 1);
2136 if (!bmap)
2137 return NULL;
2138 for (i = 0; i < pos && bmap; ++i)
2139 bmap = var_equal(bmap, i);
2140 if (bmap)
2141 bmap = var_less(bmap, pos);
2142 return isl_basic_map_finalize(bmap);
2145 struct isl_basic_map *isl_basic_map_more_at(struct isl_dim *dim, unsigned pos)
2147 int i;
2148 struct isl_basic_map *bmap;
2149 bmap = isl_basic_map_alloc_dim(dim, 0, pos, 1);
2150 if (!bmap)
2151 return NULL;
2152 for (i = 0; i < pos && bmap; ++i)
2153 bmap = var_equal(bmap, i);
2154 if (bmap)
2155 bmap = var_more(bmap, pos);
2156 return isl_basic_map_finalize(bmap);
2159 struct isl_basic_map *isl_basic_map_from_basic_set(
2160 struct isl_basic_set *bset, struct isl_dim *dim)
2162 struct isl_basic_map *bmap;
2164 bset = isl_basic_set_cow(bset);
2165 if (!bset || !dim)
2166 goto error;
2168 isl_assert(bset->ctx, isl_dim_compatible(bset->dim, dim), goto error);
2169 isl_dim_free(bset->dim);
2170 bmap = (struct isl_basic_map *) bset;
2171 bmap->dim = dim;
2172 return isl_basic_map_finalize(bmap);
2173 error:
2174 isl_basic_set_free(bset);
2175 isl_dim_free(dim);
2176 return NULL;
2179 struct isl_basic_set *isl_basic_set_from_basic_map(struct isl_basic_map *bmap)
2181 if (!bmap)
2182 goto error;
2183 if (bmap->dim->n_in == 0)
2184 return (struct isl_basic_set *)bmap;
2185 bmap = isl_basic_map_cow(bmap);
2186 if (!bmap)
2187 goto error;
2188 bmap->dim = isl_dim_cow(bmap->dim);
2189 if (!bmap->dim)
2190 goto error;
2191 bmap->dim->n_out += bmap->dim->n_in;
2192 bmap->dim->n_in = 0;
2193 bmap = isl_basic_map_finalize(bmap);
2194 return (struct isl_basic_set *)bmap;
2195 error:
2196 isl_basic_map_free(bmap);
2197 return NULL;
2200 /* For a div d = floor(f/m), add the constraints
2202 * f - m d >= 0
2203 * -(f-(n-1)) + m d >= 0
2205 * Note that the second constraint is the negation of
2207 * f - m d >= n
2209 static int add_div_constraints(struct isl_basic_map *bmap, unsigned div)
2211 int i, j;
2212 unsigned total = isl_basic_map_total_dim(bmap);
2213 unsigned div_pos = 1 + total - bmap->n_div + div;
2215 i = isl_basic_map_alloc_inequality(bmap);
2216 if (i < 0)
2217 return -1;
2218 isl_seq_cpy(bmap->ineq[i], bmap->div[div]+1, 1+total);
2219 isl_int_neg(bmap->ineq[i][div_pos], bmap->div[div][0]);
2221 j = isl_basic_map_alloc_inequality(bmap);
2222 if (j < 0)
2223 return -1;
2224 isl_seq_neg(bmap->ineq[j], bmap->ineq[i], 1 + total);
2225 isl_int_add(bmap->ineq[j][0], bmap->ineq[j][0], bmap->ineq[j][div_pos]);
2226 isl_int_sub_ui(bmap->ineq[j][0], bmap->ineq[j][0], 1);
2227 return j;
2230 struct isl_basic_set *isl_basic_map_underlying_set(
2231 struct isl_basic_map *bmap)
2233 if (!bmap)
2234 goto error;
2235 if (bmap->dim->nparam == 0 && bmap->dim->n_in == 0 && bmap->n_div == 0)
2236 return (struct isl_basic_set *)bmap;
2237 bmap = isl_basic_map_cow(bmap);
2238 if (!bmap)
2239 goto error;
2240 bmap->dim = isl_dim_underlying(bmap->dim, bmap->n_div);
2241 if (!bmap->dim)
2242 goto error;
2243 bmap->extra -= bmap->n_div;
2244 bmap->n_div = 0;
2245 bmap = isl_basic_map_finalize(bmap);
2246 return (struct isl_basic_set *)bmap;
2247 error:
2248 return NULL;
2251 struct isl_basic_map *isl_basic_map_overlying_set(
2252 struct isl_basic_set *bset, struct isl_basic_map *like)
2254 struct isl_basic_map *bmap;
2255 struct isl_ctx *ctx;
2256 unsigned total;
2257 int i;
2259 if (!bset || !like)
2260 goto error;
2261 ctx = bset->ctx;
2262 isl_assert(ctx, bset->n_div == 0, goto error);
2263 isl_assert(ctx, isl_basic_set_n_param(bset) == 0, goto error);
2264 isl_assert(ctx, bset->dim->n_out == isl_basic_map_total_dim(like),
2265 goto error);
2266 if (isl_dim_equal(bset->dim, like->dim) && like->n_div == 0) {
2267 isl_basic_map_free(like);
2268 return (struct isl_basic_map *)bset;
2270 bset = isl_basic_set_cow(bset);
2271 if (!bset)
2272 goto error;
2273 total = bset->dim->n_out + bset->extra;
2274 bmap = (struct isl_basic_map *)bset;
2275 isl_dim_free(bmap->dim);
2276 bmap->dim = isl_dim_copy(like->dim);
2277 if (!bmap->dim)
2278 goto error;
2279 bmap->n_div = like->n_div;
2280 bmap->extra += like->n_div;
2281 if (bmap->extra) {
2282 unsigned ltotal;
2283 ltotal = total - bmap->extra + like->extra;
2284 if (ltotal > total)
2285 ltotal = total;
2286 bmap->block2 = isl_blk_extend(ctx, bmap->block2,
2287 bmap->extra * (1 + 1 + total));
2288 if (isl_blk_is_error(bmap->block2))
2289 goto error;
2290 bmap->div = isl_realloc_array(ctx, bmap->div, isl_int *,
2291 bmap->extra);
2292 if (!bmap->div)
2293 goto error;
2294 for (i = 0; i < bmap->extra; ++i)
2295 bmap->div[i] = bmap->block2.data + i * (1 + 1 + total);
2296 for (i = 0; i < like->n_div; ++i) {
2297 isl_seq_cpy(bmap->div[i], like->div[i], 1 + 1 + ltotal);
2298 isl_seq_clr(bmap->div[i]+1+1+ltotal, total - ltotal);
2300 bmap = isl_basic_map_extend_constraints(bmap,
2301 0, 2 * like->n_div);
2302 for (i = 0; i < like->n_div; ++i) {
2303 if (isl_int_is_zero(bmap->div[i][0]))
2304 continue;
2305 if (add_div_constraints(bmap, i) < 0)
2306 goto error;
2309 isl_basic_map_free(like);
2310 bmap = isl_basic_map_simplify(bmap);
2311 bmap = isl_basic_map_finalize(bmap);
2312 return bmap;
2313 error:
2314 isl_basic_map_free(like);
2315 isl_basic_set_free(bset);
2316 return NULL;
2319 struct isl_basic_set *isl_basic_set_from_underlying_set(
2320 struct isl_basic_set *bset, struct isl_basic_set *like)
2322 return (struct isl_basic_set *)
2323 isl_basic_map_overlying_set(bset, (struct isl_basic_map *)like);
2326 struct isl_set *isl_set_from_underlying_set(
2327 struct isl_set *set, struct isl_basic_set *like)
2329 int i;
2331 if (!set || !like)
2332 goto error;
2333 isl_assert(set->ctx, set->dim->n_out == isl_basic_set_total_dim(like),
2334 goto error);
2335 if (isl_dim_equal(set->dim, like->dim) && like->n_div == 0) {
2336 isl_basic_set_free(like);
2337 return set;
2339 set = isl_set_cow(set);
2340 if (!set)
2341 goto error;
2342 for (i = 0; i < set->n; ++i) {
2343 set->p[i] = isl_basic_set_from_underlying_set(set->p[i],
2344 isl_basic_set_copy(like));
2345 if (!set->p[i])
2346 goto error;
2348 isl_dim_free(set->dim);
2349 set->dim = isl_dim_copy(like->dim);
2350 if (!set->dim)
2351 goto error;
2352 isl_basic_set_free(like);
2353 return set;
2354 error:
2355 isl_basic_set_free(like);
2356 isl_set_free(set);
2357 return NULL;
2360 struct isl_set *isl_map_underlying_set(struct isl_map *map)
2362 int i;
2364 map = isl_map_cow(map);
2365 if (!map)
2366 return NULL;
2367 map->dim = isl_dim_cow(map->dim);
2368 if (!map->dim)
2369 goto error;
2371 for (i = 1; i < map->n; ++i)
2372 isl_assert(map->ctx, map->p[0]->n_div == map->p[i]->n_div,
2373 goto error);
2374 for (i = 0; i < map->n; ++i) {
2375 map->p[i] = (struct isl_basic_map *)
2376 isl_basic_map_underlying_set(map->p[i]);
2377 if (!map->p[i])
2378 goto error;
2380 if (map->n == 0)
2381 map->dim = isl_dim_underlying(map->dim, 0);
2382 else {
2383 isl_dim_free(map->dim);
2384 map->dim = isl_dim_copy(map->p[0]->dim);
2386 if (!map->dim)
2387 goto error;
2388 return (struct isl_set *)map;
2389 error:
2390 isl_map_free(map);
2391 return NULL;
2394 struct isl_set *isl_set_to_underlying_set(struct isl_set *set)
2396 return (struct isl_set *)isl_map_underlying_set((struct isl_map *)set);
2399 struct isl_basic_set *isl_basic_map_domain(struct isl_basic_map *bmap)
2401 struct isl_basic_set *domain;
2402 unsigned n_out;
2403 if (!bmap)
2404 return NULL;
2405 n_out = isl_basic_map_n_out(bmap);
2406 domain = isl_basic_set_from_basic_map(bmap);
2407 return isl_basic_set_project_out(domain, n_out, 0);
2410 struct isl_basic_set *isl_basic_map_range(struct isl_basic_map *bmap)
2412 return isl_basic_map_domain(isl_basic_map_reverse(bmap));
2415 struct isl_set *isl_map_range(struct isl_map *map)
2417 int i;
2418 struct isl_set *set;
2420 if (!map)
2421 goto error;
2422 map = isl_map_cow(map);
2423 if (!map)
2424 goto error;
2426 set = (struct isl_set *) map;
2427 if (set->dim->n_in != 0) {
2428 set->dim = isl_dim_drop_inputs(set->dim, 0, set->dim->n_in);
2429 if (!set->dim)
2430 goto error;
2432 for (i = 0; i < map->n; ++i) {
2433 set->p[i] = isl_basic_map_range(map->p[i]);
2434 if (!set->p[i])
2435 goto error;
2437 ISL_F_CLR(set, ISL_MAP_DISJOINT);
2438 ISL_F_CLR(set, ISL_SET_NORMALIZED);
2439 return set;
2440 error:
2441 isl_map_free(map);
2442 return NULL;
2445 struct isl_map *isl_map_from_set(struct isl_set *set, struct isl_dim *dim)
2447 int i;
2448 struct isl_map *map = NULL;
2450 set = isl_set_cow(set);
2451 if (!set || !dim)
2452 goto error;
2453 isl_assert(set->ctx, isl_dim_compatible(set->dim, dim), goto error);
2454 map = (struct isl_map *)set;
2455 for (i = 0; i < set->n; ++i) {
2456 map->p[i] = isl_basic_map_from_basic_set(
2457 set->p[i], isl_dim_copy(dim));
2458 if (!map->p[i])
2459 goto error;
2461 isl_dim_free(map->dim);
2462 map->dim = dim;
2463 return map;
2464 error:
2465 isl_dim_free(dim);
2466 isl_set_free(set);
2467 return NULL;
2470 struct isl_map *isl_map_from_range(struct isl_set *set)
2472 return (struct isl_map *)set;
2475 struct isl_set *isl_set_from_map(struct isl_map *map)
2477 int i;
2478 struct isl_set *set = NULL;
2480 if (!map)
2481 return NULL;
2482 map = isl_map_cow(map);
2483 if (!map)
2484 return NULL;
2485 map->dim = isl_dim_cow(map->dim);
2486 if (!map->dim)
2487 goto error;
2488 map->dim->n_out += map->dim->n_in;
2489 map->dim->n_in = 0;
2490 set = (struct isl_set *)map;
2491 for (i = 0; i < map->n; ++i) {
2492 set->p[i] = isl_basic_set_from_basic_map(map->p[i]);
2493 if (!set->p[i])
2494 goto error;
2496 return set;
2497 error:
2498 isl_map_free(map);
2499 return NULL;
2502 struct isl_map *isl_map_alloc_dim(struct isl_dim *dim, int n, unsigned flags)
2504 struct isl_map *map;
2506 if (!dim)
2507 return NULL;
2508 isl_assert(dim->ctx, n >= 0, return NULL);
2509 map = isl_alloc(dim->ctx, struct isl_map,
2510 sizeof(struct isl_map) +
2511 n * sizeof(struct isl_basic_map *));
2512 if (!map)
2513 goto error;
2515 map->ctx = dim->ctx;
2516 isl_ctx_ref(map->ctx);
2517 map->ref = 1;
2518 map->size = n;
2519 map->n = 0;
2520 map->dim = dim;
2521 map->flags = flags;
2522 return map;
2523 error:
2524 isl_dim_free(dim);
2525 return NULL;
2528 struct isl_map *isl_map_alloc(struct isl_ctx *ctx,
2529 unsigned nparam, unsigned in, unsigned out, int n,
2530 unsigned flags)
2532 struct isl_map *map;
2533 struct isl_dim *dims;
2535 dims = isl_dim_alloc(ctx, nparam, in, out);
2536 if (!dims)
2537 return NULL;
2539 map = isl_map_alloc_dim(dims, n, flags);
2540 return map;
2543 struct isl_basic_map *isl_basic_map_empty(struct isl_ctx *ctx,
2544 unsigned nparam, unsigned in, unsigned out)
2546 struct isl_basic_map *bmap;
2547 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, 1, 0);
2548 bmap = isl_basic_map_set_to_empty(bmap);
2549 return bmap;
2552 struct isl_basic_set *isl_basic_set_empty(struct isl_dim *dim)
2554 struct isl_basic_set *bset;
2555 bset = isl_basic_set_alloc_dim(dim, 0, 1, 0);
2556 bset = isl_basic_set_set_to_empty(bset);
2557 return bset;
2560 struct isl_basic_map *isl_basic_map_empty_like(struct isl_basic_map *model)
2562 struct isl_basic_map *bmap;
2563 if (!model)
2564 return NULL;
2565 bmap = isl_basic_map_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2566 bmap = isl_basic_map_set_to_empty(bmap);
2567 return bmap;
2570 struct isl_basic_map *isl_basic_map_empty_like_map(struct isl_map *model)
2572 struct isl_basic_map *bmap;
2573 if (!model)
2574 return NULL;
2575 bmap = isl_basic_map_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2576 bmap = isl_basic_map_set_to_empty(bmap);
2577 return bmap;
2580 struct isl_basic_set *isl_basic_set_empty_like(struct isl_basic_set *model)
2582 struct isl_basic_set *bset;
2583 if (!model)
2584 return NULL;
2585 bset = isl_basic_set_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2586 bset = isl_basic_set_set_to_empty(bset);
2587 return bset;
2590 struct isl_basic_map *isl_basic_map_universe(struct isl_dim *dim)
2592 struct isl_basic_map *bmap;
2593 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
2594 return bmap;
2597 struct isl_basic_set *isl_basic_set_universe(struct isl_dim *dim)
2599 struct isl_basic_set *bset;
2600 bset = isl_basic_set_alloc_dim(dim, 0, 0, 0);
2601 return bset;
2604 struct isl_basic_set *isl_basic_set_universe_like(struct isl_basic_set *model)
2606 if (!model)
2607 return NULL;
2608 return isl_basic_set_alloc_dim(isl_dim_copy(model->dim), 0, 0, 0);
2611 struct isl_map *isl_map_empty(struct isl_dim *dim)
2613 return isl_map_alloc_dim(dim, 0, ISL_MAP_DISJOINT);
2616 struct isl_map *isl_map_empty_like(struct isl_map *model)
2618 if (!model)
2619 return NULL;
2620 return isl_map_alloc_dim(isl_dim_copy(model->dim), 0, ISL_MAP_DISJOINT);
2623 struct isl_map *isl_map_empty_like_basic_map(struct isl_basic_map *model)
2625 if (!model)
2626 return NULL;
2627 return isl_map_alloc_dim(isl_dim_copy(model->dim), 0, ISL_MAP_DISJOINT);
2630 struct isl_set *isl_set_empty(struct isl_dim *dim)
2632 return isl_set_alloc_dim(dim, 0, ISL_MAP_DISJOINT);
2635 struct isl_set *isl_set_empty_like(struct isl_set *model)
2637 if (!model)
2638 return NULL;
2639 return isl_set_empty(isl_dim_copy(model->dim));
2642 struct isl_map *isl_map_universe(struct isl_dim *dim)
2644 struct isl_map *map;
2645 if (!dim)
2646 return NULL;
2647 map = isl_map_alloc_dim(isl_dim_copy(dim), 1, ISL_MAP_DISJOINT);
2648 map = isl_map_add(map, isl_basic_map_universe(dim));
2649 return map;
2652 struct isl_set *isl_set_universe(struct isl_dim *dim)
2654 struct isl_set *set;
2655 if (!dim)
2656 return NULL;
2657 set = isl_set_alloc_dim(isl_dim_copy(dim), 1, ISL_MAP_DISJOINT);
2658 set = isl_set_add(set, isl_basic_set_universe(dim));
2659 return set;
2662 struct isl_map *isl_map_dup(struct isl_map *map)
2664 int i;
2665 struct isl_map *dup;
2667 if (!map)
2668 return NULL;
2669 dup = isl_map_alloc_dim(isl_dim_copy(map->dim), map->n, map->flags);
2670 for (i = 0; i < map->n; ++i)
2671 dup = isl_map_add(dup, isl_basic_map_copy(map->p[i]));
2672 return dup;
2675 struct isl_map *isl_map_add(struct isl_map *map, struct isl_basic_map *bmap)
2677 if (!bmap || !map)
2678 goto error;
2679 isl_assert(map->ctx, isl_dim_equal(map->dim, bmap->dim), goto error);
2680 isl_assert(map->ctx, map->n < map->size, goto error);
2681 map->p[map->n] = bmap;
2682 map->n++;
2683 ISL_F_CLR(map, ISL_MAP_NORMALIZED);
2684 return map;
2685 error:
2686 if (map)
2687 isl_map_free(map);
2688 if (bmap)
2689 isl_basic_map_free(bmap);
2690 return NULL;
2693 void isl_map_free(struct isl_map *map)
2695 int i;
2697 if (!map)
2698 return;
2700 if (--map->ref > 0)
2701 return;
2703 isl_ctx_deref(map->ctx);
2704 for (i = 0; i < map->n; ++i)
2705 isl_basic_map_free(map->p[i]);
2706 isl_dim_free(map->dim);
2707 free(map);
2710 struct isl_map *isl_map_extend(struct isl_map *base,
2711 unsigned nparam, unsigned n_in, unsigned n_out)
2713 int i;
2715 base = isl_map_cow(base);
2716 if (!base)
2717 return NULL;
2719 base->dim = isl_dim_extend(base->dim, nparam, n_in, n_out);
2720 if (!base->dim)
2721 goto error;
2722 for (i = 0; i < base->n; ++i) {
2723 base->p[i] = isl_basic_map_extend_dim(base->p[i],
2724 isl_dim_copy(base->dim), 0, 0, 0);
2725 if (!base->p[i])
2726 goto error;
2728 return base;
2729 error:
2730 isl_map_free(base);
2731 return NULL;
2734 struct isl_set *isl_set_extend(struct isl_set *base,
2735 unsigned nparam, unsigned dim)
2737 return (struct isl_set *)isl_map_extend((struct isl_map *)base,
2738 nparam, 0, dim);
2741 static struct isl_basic_map *isl_basic_map_fix_pos(struct isl_basic_map *bmap,
2742 unsigned pos, int value)
2744 int j;
2746 bmap = isl_basic_map_cow(bmap);
2747 bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
2748 j = isl_basic_map_alloc_equality(bmap);
2749 if (j < 0)
2750 goto error;
2751 isl_seq_clr(bmap->eq[j], 1 + isl_basic_map_total_dim(bmap));
2752 isl_int_set_si(bmap->eq[j][pos], -1);
2753 isl_int_set_si(bmap->eq[j][0], value);
2754 bmap = isl_basic_map_simplify(bmap);
2755 return isl_basic_map_finalize(bmap);
2756 error:
2757 isl_basic_map_free(bmap);
2758 return NULL;
2761 struct isl_basic_map *isl_basic_map_fix_si(struct isl_basic_map *bmap,
2762 enum isl_dim_type type, unsigned pos, int value)
2764 if (!bmap)
2765 return NULL;
2766 isl_assert(bmap->ctx, pos < isl_basic_map_dim(bmap, type), goto error);
2767 return isl_basic_map_fix_pos(bmap, isl_basic_map_offset(bmap, type) + pos,
2768 value);
2769 error:
2770 isl_basic_map_free(bmap);
2771 return NULL;
2774 struct isl_basic_set *isl_basic_set_fix_si(struct isl_basic_set *bset,
2775 enum isl_dim_type type, unsigned pos, int value)
2777 return (struct isl_basic_set *)
2778 isl_basic_map_fix_si((struct isl_basic_map *)bset,
2779 type, pos, value);
2782 struct isl_basic_map *isl_basic_map_fix_input_si(struct isl_basic_map *bmap,
2783 unsigned input, int value)
2785 return isl_basic_map_fix_si(bmap, isl_dim_in, input, value);
2788 struct isl_basic_set *isl_basic_set_fix_dim_si(struct isl_basic_set *bset,
2789 unsigned dim, int value)
2791 return (struct isl_basic_set *)
2792 isl_basic_map_fix_si((struct isl_basic_map *)bset,
2793 isl_dim_set, dim, value);
2796 struct isl_map *isl_map_fix_si(struct isl_map *map,
2797 enum isl_dim_type type, unsigned pos, int value)
2799 int i;
2801 map = isl_map_cow(map);
2802 if (!map)
2803 return NULL;
2805 isl_assert(ctx, pos < isl_map_dim(map, type), goto error);
2806 for (i = 0; i < map->n; ++i) {
2807 map->p[i] = isl_basic_map_fix_si(map->p[i], type, pos, value);
2808 if (!map->p[i])
2809 goto error;
2811 ISL_F_CLR(map, ISL_MAP_NORMALIZED);
2812 return map;
2813 error:
2814 isl_map_free(map);
2815 return NULL;
2818 struct isl_map *isl_map_fix_input_si(struct isl_map *map,
2819 unsigned input, int value)
2821 return isl_map_fix_si(map, isl_dim_in, input, value);
2824 struct isl_set *isl_set_fix_dim_si(struct isl_set *set, unsigned dim, int value)
2826 return (struct isl_set *)
2827 isl_map_fix_si((struct isl_map *)set, isl_dim_set, dim, value);
2830 struct isl_basic_set *isl_basic_set_lower_bound_dim(struct isl_basic_set *bset,
2831 unsigned dim, isl_int value)
2833 int j;
2834 unsigned nparam;
2836 bset = isl_basic_set_cow(bset);
2837 bset = isl_basic_set_extend_constraints(bset, 0, 1);
2838 j = isl_basic_set_alloc_inequality(bset);
2839 if (j < 0)
2840 goto error;
2841 isl_seq_clr(bset->ineq[j], 1 + isl_basic_set_total_dim(bset));
2842 isl_int_set_si(bset->ineq[j][1 + isl_basic_set_n_param(bset) + dim], 1);
2843 isl_int_neg(bset->ineq[j][0], value);
2844 bset = isl_basic_set_simplify(bset);
2845 return isl_basic_set_finalize(bset);
2846 error:
2847 isl_basic_set_free(bset);
2848 return NULL;
2851 struct isl_set *isl_set_lower_bound_dim(struct isl_set *set, unsigned dim,
2852 isl_int value)
2854 int i;
2856 set = isl_set_cow(set);
2857 if (!set)
2858 return NULL;
2860 isl_assert(set->ctx, dim < isl_set_n_dim(set), goto error);
2861 for (i = 0; i < set->n; ++i) {
2862 set->p[i] = isl_basic_set_lower_bound_dim(set->p[i], dim, value);
2863 if (!set->p[i])
2864 goto error;
2866 return set;
2867 error:
2868 isl_set_free(set);
2869 return NULL;
2872 struct isl_map *isl_map_reverse(struct isl_map *map)
2874 int i;
2875 unsigned t;
2877 map = isl_map_cow(map);
2878 if (!map)
2879 return NULL;
2881 map->dim = isl_dim_reverse(map->dim);
2882 if (!map->dim)
2883 goto error;
2884 for (i = 0; i < map->n; ++i) {
2885 map->p[i] = isl_basic_map_reverse(map->p[i]);
2886 if (!map->p[i])
2887 goto error;
2889 ISL_F_CLR(map, ISL_MAP_NORMALIZED);
2890 return map;
2891 error:
2892 isl_map_free(map);
2893 return NULL;
2896 struct isl_map *isl_basic_map_lexmax(
2897 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2898 struct isl_set **empty)
2900 return isl_pip_basic_map_lexmax(bmap, dom, empty);
2903 struct isl_map *isl_basic_map_lexmin(
2904 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2905 struct isl_set **empty)
2907 return isl_pip_basic_map_lexmin(bmap, dom, empty);
2910 struct isl_set *isl_basic_set_lexmin(struct isl_basic_set *bset)
2912 struct isl_basic_map *bmap = NULL;
2913 struct isl_basic_set *dom = NULL;
2914 struct isl_map *min;
2915 struct isl_dim *param_dim;
2917 if (!bset)
2918 goto error;
2919 bmap = isl_basic_map_from_basic_set(bset, isl_dim_copy(bset->dim));
2920 if (!bmap)
2921 goto error;
2922 param_dim = isl_dim_domain(isl_dim_copy(bmap->dim));
2923 dom = isl_basic_set_universe(param_dim);
2924 if (!dom)
2925 goto error;
2926 min = isl_basic_map_lexmin(bmap, dom, NULL);
2927 return isl_map_range(min);
2928 error:
2929 isl_basic_map_free(bmap);
2930 return NULL;
2933 struct isl_map *isl_basic_map_compute_divs(struct isl_basic_map *bmap)
2935 int i;
2936 unsigned off;
2938 if (!bmap)
2939 return NULL;
2940 off = isl_dim_total(bmap->dim);
2941 for (i = 0; i < bmap->n_div; ++i) {
2942 if (isl_int_is_zero(bmap->div[i][0]))
2943 return isl_pip_basic_map_compute_divs(bmap);
2944 isl_assert(bmap->ctx, isl_int_is_zero(bmap->div[i][1+1+off+i]),
2945 goto error);
2947 return isl_map_from_basic_map(bmap);
2948 error:
2949 isl_basic_map_free(bmap);
2950 return NULL;
2953 struct isl_map *isl_map_compute_divs(struct isl_map *map)
2955 int i;
2956 struct isl_map *res;
2958 if (!map)
2959 return NULL;
2960 if (map->n == 0)
2961 return map;
2962 res = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[0]));
2963 for (i = 1 ; i < map->n; ++i) {
2964 struct isl_map *r2;
2965 r2 = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[i]));
2966 if (ISL_F_ISSET(map, ISL_MAP_DISJOINT))
2967 res = isl_map_union_disjoint(res, r2);
2968 else
2969 res = isl_map_union(res, r2);
2971 isl_map_free(map);
2973 return res;
2976 struct isl_set *isl_basic_set_compute_divs(struct isl_basic_set *bset)
2978 return (struct isl_set *)
2979 isl_basic_map_compute_divs((struct isl_basic_map *)bset);
2982 struct isl_set *isl_set_compute_divs(struct isl_set *set)
2984 return (struct isl_set *)
2985 isl_map_compute_divs((struct isl_map *)set);
2988 struct isl_set *isl_map_domain(struct isl_map *map)
2990 int i;
2991 struct isl_set *set;
2993 if (!map)
2994 goto error;
2996 map = isl_map_cow(map);
2997 if (!map)
2998 return NULL;
3000 set = (struct isl_set *)map;
3001 set->dim = isl_dim_domain(set->dim);
3002 if (!set->dim)
3003 goto error;
3004 for (i = 0; i < map->n; ++i) {
3005 set->p[i] = isl_basic_map_domain(map->p[i]);
3006 if (!set->p[i])
3007 goto error;
3009 ISL_F_CLR(set, ISL_MAP_DISJOINT);
3010 ISL_F_CLR(set, ISL_SET_NORMALIZED);
3011 return set;
3012 error:
3013 isl_map_free(map);
3014 return NULL;
3017 struct isl_map *isl_map_union_disjoint(
3018 struct isl_map *map1, struct isl_map *map2)
3020 int i;
3021 unsigned flags = 0;
3022 struct isl_map *map = NULL;
3024 if (!map1 || !map2)
3025 goto error;
3027 if (map1->n == 0) {
3028 isl_map_free(map1);
3029 return map2;
3031 if (map2->n == 0) {
3032 isl_map_free(map2);
3033 return map1;
3036 isl_assert(map1->ctx, isl_dim_equal(map1->dim, map2->dim), goto error);
3038 if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
3039 ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
3040 ISL_FL_SET(flags, ISL_MAP_DISJOINT);
3042 map = isl_map_alloc_dim(isl_dim_copy(map1->dim),
3043 map1->n + map2->n, flags);
3044 if (!map)
3045 goto error;
3046 for (i = 0; i < map1->n; ++i) {
3047 map = isl_map_add(map,
3048 isl_basic_map_copy(map1->p[i]));
3049 if (!map)
3050 goto error;
3052 for (i = 0; i < map2->n; ++i) {
3053 map = isl_map_add(map,
3054 isl_basic_map_copy(map2->p[i]));
3055 if (!map)
3056 goto error;
3058 isl_map_free(map1);
3059 isl_map_free(map2);
3060 return map;
3061 error:
3062 isl_map_free(map);
3063 isl_map_free(map1);
3064 isl_map_free(map2);
3065 return NULL;
3068 struct isl_map *isl_map_union(struct isl_map *map1, struct isl_map *map2)
3070 map1 = isl_map_union_disjoint(map1, map2);
3071 if (!map1)
3072 return NULL;
3073 if (map1->n > 1)
3074 ISL_F_CLR(map1, ISL_MAP_DISJOINT);
3075 return map1;
3078 struct isl_set *isl_set_union_disjoint(
3079 struct isl_set *set1, struct isl_set *set2)
3081 return (struct isl_set *)
3082 isl_map_union_disjoint(
3083 (struct isl_map *)set1, (struct isl_map *)set2);
3086 struct isl_set *isl_set_union(struct isl_set *set1, struct isl_set *set2)
3088 return (struct isl_set *)
3089 isl_map_union((struct isl_map *)set1, (struct isl_map *)set2);
3092 struct isl_map *isl_map_intersect_range(
3093 struct isl_map *map, struct isl_set *set)
3095 unsigned flags = 0;
3096 struct isl_map *result;
3097 int i, j;
3099 if (!map || !set)
3100 goto error;
3102 if (ISL_F_ISSET(map, ISL_MAP_DISJOINT) &&
3103 ISL_F_ISSET(set, ISL_MAP_DISJOINT))
3104 ISL_FL_SET(flags, ISL_MAP_DISJOINT);
3106 result = isl_map_alloc_dim(isl_dim_copy(map->dim),
3107 map->n * set->n, flags);
3108 if (!result)
3109 goto error;
3110 for (i = 0; i < map->n; ++i)
3111 for (j = 0; j < set->n; ++j) {
3112 result = isl_map_add(result,
3113 isl_basic_map_intersect_range(
3114 isl_basic_map_copy(map->p[i]),
3115 isl_basic_set_copy(set->p[j])));
3116 if (!result)
3117 goto error;
3119 isl_map_free(map);
3120 isl_set_free(set);
3121 return result;
3122 error:
3123 isl_map_free(map);
3124 isl_set_free(set);
3125 return NULL;
3128 struct isl_map *isl_map_intersect_domain(
3129 struct isl_map *map, struct isl_set *set)
3131 return isl_map_reverse(
3132 isl_map_intersect_range(isl_map_reverse(map), set));
3135 struct isl_map *isl_map_apply_domain(
3136 struct isl_map *map1, struct isl_map *map2)
3138 if (!map1 || !map2)
3139 goto error;
3140 map1 = isl_map_reverse(map1);
3141 map1 = isl_map_apply_range(map1, map2);
3142 return isl_map_reverse(map1);
3143 error:
3144 isl_map_free(map1);
3145 isl_map_free(map2);
3146 return NULL;
3149 struct isl_map *isl_map_apply_range(
3150 struct isl_map *map1, struct isl_map *map2)
3152 struct isl_dim *dim_result;
3153 struct isl_map *result;
3154 int i, j;
3155 unsigned nparam;
3156 unsigned n_in;
3157 unsigned n_out;
3159 if (!map1 || !map2)
3160 goto error;
3162 dim_result = isl_dim_join(isl_dim_copy(map1->dim),
3163 isl_dim_copy(map2->dim));
3165 result = isl_map_alloc_dim(dim_result, map1->n * map2->n, 0);
3166 if (!result)
3167 goto error;
3168 for (i = 0; i < map1->n; ++i)
3169 for (j = 0; j < map2->n; ++j) {
3170 result = isl_map_add(result,
3171 isl_basic_map_apply_range(
3172 isl_basic_map_copy(map1->p[i]),
3173 isl_basic_map_copy(map2->p[j])));
3174 if (!result)
3175 goto error;
3177 isl_map_free(map1);
3178 isl_map_free(map2);
3179 if (result && result->n <= 1)
3180 ISL_F_SET(result, ISL_MAP_DISJOINT);
3181 return result;
3182 error:
3183 isl_map_free(map1);
3184 isl_map_free(map2);
3185 return NULL;
3189 * returns range - domain
3191 struct isl_basic_set *isl_basic_map_deltas(struct isl_basic_map *bmap)
3193 struct isl_basic_set *bset;
3194 unsigned dim;
3195 unsigned nparam;
3196 int i;
3198 if (!bmap)
3199 goto error;
3200 dim = isl_basic_map_n_in(bmap);
3201 nparam = isl_basic_map_n_param(bmap);
3202 isl_assert(bmap->ctx, dim == isl_basic_map_n_out(bmap), goto error);
3203 bset = isl_basic_set_from_basic_map(bmap);
3204 bset = isl_basic_set_cow(bset);
3205 bset = isl_basic_set_extend(bset, nparam, 3*dim, 0, dim, 0);
3206 bset = isl_basic_set_swap_vars(bset, 2*dim);
3207 for (i = 0; i < dim; ++i) {
3208 int j = isl_basic_map_alloc_equality(
3209 (struct isl_basic_map *)bset);
3210 if (j < 0)
3211 goto error;
3212 isl_seq_clr(bset->eq[j], 1 + isl_basic_set_total_dim(bset));
3213 isl_int_set_si(bset->eq[j][1+nparam+i], 1);
3214 isl_int_set_si(bset->eq[j][1+nparam+dim+i], 1);
3215 isl_int_set_si(bset->eq[j][1+nparam+2*dim+i], -1);
3217 return isl_basic_set_project_out(bset, 2*dim, 0);
3218 error:
3219 isl_basic_map_free(bmap);
3220 return NULL;
3224 * returns range - domain
3226 struct isl_set *isl_map_deltas(struct isl_map *map)
3228 int i;
3229 struct isl_set *result;
3231 if (!map)
3232 return NULL;
3234 isl_assert(map->ctx, isl_map_n_in(map) == isl_map_n_out(map), goto error);
3235 result = isl_set_alloc(map->ctx, isl_map_n_param(map),
3236 isl_map_n_in(map), map->n, map->flags);
3237 if (!result)
3238 goto error;
3239 for (i = 0; i < map->n; ++i)
3240 result = isl_set_add(result,
3241 isl_basic_map_deltas(isl_basic_map_copy(map->p[i])));
3242 isl_map_free(map);
3243 return result;
3244 error:
3245 isl_map_free(map);
3246 return NULL;
3249 static struct isl_basic_map *basic_map_identity(struct isl_dim *dims)
3251 struct isl_basic_map *bmap;
3252 unsigned nparam;
3253 unsigned dim;
3254 int i;
3256 if (!dims)
3257 return NULL;
3259 nparam = dims->nparam;
3260 dim = dims->n_out;
3261 bmap = isl_basic_map_alloc_dim(dims, 0, dim, 0);
3262 if (!bmap)
3263 goto error;
3265 for (i = 0; i < dim; ++i) {
3266 int j = isl_basic_map_alloc_equality(bmap);
3267 if (j < 0)
3268 goto error;
3269 isl_seq_clr(bmap->eq[j], 1 + isl_basic_map_total_dim(bmap));
3270 isl_int_set_si(bmap->eq[j][1+nparam+i], 1);
3271 isl_int_set_si(bmap->eq[j][1+nparam+dim+i], -1);
3273 return isl_basic_map_finalize(bmap);
3274 error:
3275 isl_basic_map_free(bmap);
3276 return NULL;
3279 struct isl_basic_map *isl_basic_map_identity(struct isl_dim *set_dim)
3281 struct isl_dim *dim = isl_dim_map(set_dim);
3282 if (!dim)
3283 return NULL;
3284 return basic_map_identity(dim);
3287 struct isl_basic_map *isl_basic_map_identity_like(struct isl_basic_map *model)
3289 if (!model || !model->dim)
3290 return NULL;
3291 isl_assert(model->ctx,
3292 model->dim->n_in == model->dim->n_out, return NULL);
3293 return basic_map_identity(isl_dim_copy(model->dim));
3296 static struct isl_map *map_identity(struct isl_dim *dim)
3298 struct isl_map *map = isl_map_alloc_dim(dim, 1, ISL_MAP_DISJOINT);
3299 return isl_map_add(map, basic_map_identity(isl_dim_copy(dim)));
3302 struct isl_map *isl_map_identity(struct isl_dim *set_dim)
3304 struct isl_dim *dim = isl_dim_map(set_dim);
3305 if (!dim)
3306 return NULL;
3307 return map_identity(dim);
3310 struct isl_map *isl_map_identity_like(struct isl_basic_map *model)
3312 if (!model || !model->dim)
3313 return NULL;
3314 isl_assert(model->ctx,
3315 model->dim->n_in == model->dim->n_out, return NULL);
3316 return map_identity(isl_dim_copy(model->dim));
3319 int isl_set_is_equal(struct isl_set *set1, struct isl_set *set2)
3321 return isl_map_is_equal((struct isl_map *)set1, (struct isl_map *)set2);
3324 int isl_set_is_subset(struct isl_set *set1, struct isl_set *set2)
3326 return isl_map_is_subset(
3327 (struct isl_map *)set1, (struct isl_map *)set2);
3330 int isl_basic_map_is_subset(
3331 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3333 int is_subset;
3334 struct isl_map *map1;
3335 struct isl_map *map2;
3337 if (!bmap1 || !bmap2)
3338 return -1;
3340 map1 = isl_map_from_basic_map(isl_basic_map_copy(bmap1));
3341 map2 = isl_map_from_basic_map(isl_basic_map_copy(bmap2));
3343 is_subset = isl_map_is_subset(map1, map2);
3345 isl_map_free(map1);
3346 isl_map_free(map2);
3348 return is_subset;
3351 int isl_basic_map_is_equal(
3352 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3354 int is_subset;
3356 if (!bmap1 || !bmap2)
3357 return -1;
3358 is_subset = isl_basic_map_is_subset(bmap1, bmap2);
3359 if (is_subset != 1)
3360 return is_subset;
3361 is_subset = isl_basic_map_is_subset(bmap2, bmap1);
3362 return is_subset;
3365 int isl_basic_set_is_equal(
3366 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
3368 return isl_basic_map_is_equal(
3369 (struct isl_basic_map *)bset1, (struct isl_basic_map *)bset2);
3372 int isl_map_is_empty(struct isl_map *map)
3374 int i;
3375 int is_empty;
3377 if (!map)
3378 return -1;
3379 for (i = 0; i < map->n; ++i) {
3380 is_empty = isl_basic_map_is_empty(map->p[i]);
3381 if (is_empty < 0)
3382 return -1;
3383 if (!is_empty)
3384 return 0;
3386 return 1;
3389 int isl_map_fast_is_empty(struct isl_map *map)
3391 return map->n == 0;
3394 int isl_set_is_empty(struct isl_set *set)
3396 return isl_map_is_empty((struct isl_map *)set);
3399 int isl_map_is_subset(struct isl_map *map1, struct isl_map *map2)
3401 int i;
3402 int is_subset = 0;
3403 struct isl_map *diff;
3405 if (!map1 || !map2)
3406 return -1;
3408 if (isl_map_is_empty(map1))
3409 return 1;
3411 if (isl_map_is_empty(map2))
3412 return 0;
3414 diff = isl_map_subtract(isl_map_copy(map1), isl_map_copy(map2));
3415 if (!diff)
3416 return -1;
3418 is_subset = isl_map_is_empty(diff);
3419 isl_map_free(diff);
3421 return is_subset;
3424 int isl_map_is_equal(struct isl_map *map1, struct isl_map *map2)
3426 int is_subset;
3428 if (!map1 || !map2)
3429 return -1;
3430 is_subset = isl_map_is_subset(map1, map2);
3431 if (is_subset != 1)
3432 return is_subset;
3433 is_subset = isl_map_is_subset(map2, map1);
3434 return is_subset;
3437 int isl_basic_map_is_strict_subset(
3438 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3440 int is_subset;
3442 if (!bmap1 || !bmap2)
3443 return -1;
3444 is_subset = isl_basic_map_is_subset(bmap1, bmap2);
3445 if (is_subset != 1)
3446 return is_subset;
3447 is_subset = isl_basic_map_is_subset(bmap2, bmap1);
3448 if (is_subset == -1)
3449 return is_subset;
3450 return !is_subset;
3453 int isl_basic_map_is_universe(struct isl_basic_map *bmap)
3455 if (!bmap)
3456 return -1;
3457 return bmap->n_eq == 0 && bmap->n_ineq == 0;
3460 int isl_basic_map_is_empty(struct isl_basic_map *bmap)
3462 struct isl_basic_set *bset = NULL;
3463 struct isl_vec *sample = NULL;
3464 int empty;
3465 unsigned total;
3467 if (!bmap)
3468 return -1;
3470 if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
3471 return 1;
3473 if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL)) {
3474 struct isl_basic_map *copy = isl_basic_map_copy(bmap);
3475 copy = isl_basic_map_convex_hull(copy);
3476 empty = ISL_F_ISSET(copy, ISL_BASIC_MAP_EMPTY);
3477 isl_basic_map_free(copy);
3478 return empty;
3481 total = 1 + isl_basic_map_total_dim(bmap);
3482 if (bmap->sample && bmap->sample->size == total) {
3483 int contains = basic_map_contains(bmap, bmap->sample);
3484 if (contains < 0)
3485 return -1;
3486 if (contains)
3487 return 0;
3489 bset = isl_basic_map_underlying_set(isl_basic_map_copy(bmap));
3490 if (!bset)
3491 return -1;
3492 sample = isl_basic_set_sample(bset);
3493 if (!sample)
3494 return -1;
3495 empty = sample->size == 0;
3496 if (bmap->sample)
3497 isl_vec_free(bmap->ctx, bmap->sample);
3498 bmap->sample = sample;
3500 return empty;
3503 int isl_basic_map_fast_is_empty(struct isl_basic_map *bmap)
3505 if (!bmap)
3506 return -1;
3507 return ISL_F_ISSET(bmap, ISL_BASIC_MAP_EMPTY);
3510 int isl_basic_set_is_empty(struct isl_basic_set *bset)
3512 return isl_basic_map_is_empty((struct isl_basic_map *)bset);
3515 struct isl_map *isl_basic_map_union(
3516 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3518 struct isl_map *map;
3519 if (!bmap1 || !bmap2)
3520 return NULL;
3522 isl_assert(map1->ctx, isl_dim_equal(bmap1->dim, bmap2->dim), goto error);
3524 map = isl_map_alloc_dim(isl_dim_copy(bmap1->dim), 2, 0);
3525 if (!map)
3526 goto error;
3527 map = isl_map_add(map, bmap1);
3528 map = isl_map_add(map, bmap2);
3529 return map;
3530 error:
3531 isl_basic_map_free(bmap1);
3532 isl_basic_map_free(bmap2);
3533 return NULL;
3536 struct isl_set *isl_basic_set_union(
3537 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
3539 return (struct isl_set *)isl_basic_map_union(
3540 (struct isl_basic_map *)bset1,
3541 (struct isl_basic_map *)bset2);
3544 /* Order divs such that any div only depends on previous divs */
3545 static struct isl_basic_map *order_divs(struct isl_basic_map *bmap)
3547 int i;
3548 unsigned off = isl_dim_total(bmap->dim);
3550 for (i = 0; i < bmap->n_div; ++i) {
3551 int pos;
3552 pos = isl_seq_first_non_zero(bmap->div[i]+1+1+off+i,
3553 bmap->n_div-i);
3554 if (pos == -1)
3555 continue;
3556 swap_div(bmap, i, pos);
3557 --i;
3559 return bmap;
3562 /* Look for a div in dst that corresponds to the div "div" in src.
3563 * The divs before "div" in src and dst are assumed to be the same.
3565 * Returns -1 if no corresponding div was found and the position
3566 * of the corresponding div in dst otherwise.
3568 static int find_div(struct isl_basic_map *dst,
3569 struct isl_basic_map *src, unsigned div)
3571 int i;
3573 unsigned total = isl_dim_total(src->dim);
3575 isl_assert(dst->ctx, div <= dst->n_div, return -1);
3576 for (i = div; i < dst->n_div; ++i)
3577 if (isl_seq_eq(dst->div[i], src->div[div], 1+1+total+div) &&
3578 isl_seq_first_non_zero(dst->div[i]+1+1+total+div,
3579 dst->n_div - div) == -1)
3580 return i;
3581 return -1;
3584 struct isl_basic_map *isl_basic_map_align_divs(
3585 struct isl_basic_map *dst, struct isl_basic_map *src)
3587 int i;
3588 unsigned total = isl_dim_total(src->dim);
3590 if (!dst || !src)
3591 goto error;
3593 if (src->n_div == 0)
3594 return dst;
3596 for (i = 0; i < src->n_div; ++i)
3597 isl_assert(src->ctx, !isl_int_is_zero(src->div[i][0]), goto error);
3599 src = order_divs(src);
3600 dst = isl_basic_map_cow(dst);
3601 dst = isl_basic_map_extend_dim(dst, isl_dim_copy(dst->dim),
3602 src->n_div, 0, 2 * src->n_div);
3603 if (!dst)
3604 return NULL;
3605 for (i = 0; i < src->n_div; ++i) {
3606 int j = find_div(dst, src, i);
3607 if (j < 0) {
3608 j = isl_basic_map_alloc_div(dst);
3609 if (j < 0)
3610 goto error;
3611 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total+i);
3612 isl_seq_clr(dst->div[j]+1+1+total+i, dst->n_div - i);
3613 if (add_div_constraints(dst, j) < 0)
3614 goto error;
3616 if (j != i)
3617 swap_div(dst, i, j);
3619 return dst;
3620 error:
3621 isl_basic_map_free(dst);
3622 return NULL;
3625 struct isl_basic_set *isl_basic_set_align_divs(
3626 struct isl_basic_set *dst, struct isl_basic_set *src)
3628 return (struct isl_basic_set *)isl_basic_map_align_divs(
3629 (struct isl_basic_map *)dst, (struct isl_basic_map *)src);
3632 struct isl_map *isl_map_align_divs(struct isl_map *map)
3634 int i;
3636 map = isl_map_compute_divs(map);
3637 map = isl_map_cow(map);
3638 if (!map)
3639 return NULL;
3641 for (i = 1; i < map->n; ++i)
3642 map->p[0] = isl_basic_map_align_divs(map->p[0], map->p[i]);
3643 for (i = 1; i < map->n; ++i)
3644 map->p[i] = isl_basic_map_align_divs(map->p[i], map->p[0]);
3646 ISL_F_CLR(map, ISL_MAP_NORMALIZED);
3647 return map;
3650 struct isl_set *isl_set_align_divs(struct isl_set *set)
3652 return (struct isl_set *)isl_map_align_divs((struct isl_map *)set);
3655 static struct isl_map *add_cut_constraint(struct isl_map *dst,
3656 struct isl_basic_map *src, isl_int *c,
3657 unsigned len, int oppose)
3659 struct isl_basic_map *copy = NULL;
3660 int is_empty;
3661 int k;
3662 unsigned total;
3664 copy = isl_basic_map_copy(src);
3665 copy = isl_basic_map_cow(copy);
3666 if (!copy)
3667 goto error;
3668 copy = isl_basic_map_extend_constraints(copy, 0, 1);
3669 k = isl_basic_map_alloc_inequality(copy);
3670 if (k < 0)
3671 goto error;
3672 if (oppose)
3673 isl_seq_neg(copy->ineq[k], c, len);
3674 else
3675 isl_seq_cpy(copy->ineq[k], c, len);
3676 total = 1 + isl_basic_map_total_dim(copy);
3677 isl_seq_clr(copy->ineq[k]+len, total - len);
3678 isl_inequality_negate(copy, k);
3679 copy = isl_basic_map_simplify(copy);
3680 copy = isl_basic_map_finalize(copy);
3681 is_empty = isl_basic_map_is_empty(copy);
3682 if (is_empty < 0)
3683 goto error;
3684 if (!is_empty)
3685 dst = isl_map_add(dst, copy);
3686 else
3687 isl_basic_map_free(copy);
3688 return dst;
3689 error:
3690 isl_basic_map_free(copy);
3691 isl_map_free(dst);
3692 return NULL;
3695 static struct isl_map *subtract(struct isl_map *map, struct isl_basic_map *bmap)
3697 int i, j, k;
3698 unsigned flags = 0;
3699 struct isl_map *rest = NULL;
3700 unsigned max;
3701 unsigned total = isl_basic_map_total_dim(bmap);
3703 assert(bmap);
3705 if (!map)
3706 goto error;
3708 if (ISL_F_ISSET(map, ISL_MAP_DISJOINT))
3709 ISL_FL_SET(flags, ISL_MAP_DISJOINT);
3711 max = map->n * (2 * bmap->n_eq + bmap->n_ineq);
3712 rest = isl_map_alloc_dim(isl_dim_copy(map->dim), max, flags);
3713 if (!rest)
3714 goto error;
3716 for (i = 0; i < map->n; ++i) {
3717 map->p[i] = isl_basic_map_align_divs(map->p[i], bmap);
3718 if (!map->p[i])
3719 goto error;
3722 for (j = 0; j < map->n; ++j)
3723 map->p[j] = isl_basic_map_cow(map->p[j]);
3725 for (i = 0; i < bmap->n_eq; ++i) {
3726 for (j = 0; j < map->n; ++j) {
3727 rest = add_cut_constraint(rest,
3728 map->p[j], bmap->eq[i], 1+total, 0);
3729 if (!rest)
3730 goto error;
3732 rest = add_cut_constraint(rest,
3733 map->p[j], bmap->eq[i], 1+total, 1);
3734 if (!rest)
3735 goto error;
3737 map->p[j] = isl_basic_map_extend_constraints(map->p[j],
3738 1, 0);
3739 if (!map->p[j])
3740 goto error;
3741 k = isl_basic_map_alloc_equality(map->p[j]);
3742 if (k < 0)
3743 goto error;
3744 isl_seq_cpy(map->p[j]->eq[k], bmap->eq[i], 1+total);
3745 isl_seq_clr(map->p[j]->eq[k]+1+total,
3746 map->p[j]->n_div - bmap->n_div);
3750 for (i = 0; i < bmap->n_ineq; ++i) {
3751 for (j = 0; j < map->n; ++j) {
3752 rest = add_cut_constraint(rest,
3753 map->p[j], bmap->ineq[i], 1+total, 0);
3754 if (!rest)
3755 goto error;
3757 map->p[j] = isl_basic_map_extend_constraints(map->p[j],
3758 0, 1);
3759 if (!map->p[j])
3760 goto error;
3761 k = isl_basic_map_alloc_inequality(map->p[j]);
3762 if (k < 0)
3763 goto error;
3764 isl_seq_cpy(map->p[j]->ineq[k], bmap->ineq[i], 1+total);
3765 isl_seq_clr(map->p[j]->ineq[k]+1+total,
3766 map->p[j]->n_div - bmap->n_div);
3770 isl_map_free(map);
3771 return rest;
3772 error:
3773 isl_map_free(map);
3774 isl_map_free(rest);
3775 return NULL;
3778 struct isl_map *isl_map_subtract(struct isl_map *map1, struct isl_map *map2)
3780 int i;
3781 if (!map1 || !map2)
3782 goto error;
3784 isl_assert(map1->ctx, isl_dim_equal(map1->dim, map2->dim), goto error);
3786 if (isl_map_is_empty(map2)) {
3787 isl_map_free(map2);
3788 return map1;
3791 map1 = isl_map_compute_divs(map1);
3792 map2 = isl_map_compute_divs(map2);
3793 if (!map1 || !map2)
3794 goto error;
3796 for (i = 0; map1 && i < map2->n; ++i)
3797 map1 = subtract(map1, map2->p[i]);
3799 isl_map_free(map2);
3800 return map1;
3801 error:
3802 isl_map_free(map1);
3803 isl_map_free(map2);
3804 return NULL;
3807 struct isl_set *isl_set_subtract(struct isl_set *set1, struct isl_set *set2)
3809 return (struct isl_set *)
3810 isl_map_subtract(
3811 (struct isl_map *)set1, (struct isl_map *)set2);
3814 struct isl_set *isl_set_apply(struct isl_set *set, struct isl_map *map)
3816 if (!set || !map)
3817 goto error;
3818 isl_assert(set->ctx, isl_map_compatible_domain(map, set), goto error);
3819 map = isl_map_intersect_domain(map, set);
3820 set = isl_map_range(map);
3821 return set;
3822 error:
3823 isl_set_free(set);
3824 isl_map_free(map);
3825 return NULL;
3828 /* There is no need to cow as removing empty parts doesn't change
3829 * the meaning of the set.
3831 struct isl_map *isl_map_remove_empty_parts(struct isl_map *map)
3833 int i;
3835 if (!map)
3836 return NULL;
3838 for (i = map->n-1; i >= 0; --i) {
3839 if (!ISL_F_ISSET(map->p[i], ISL_BASIC_MAP_EMPTY))
3840 continue;
3841 isl_basic_map_free(map->p[i]);
3842 if (i != map->n-1) {
3843 ISL_F_CLR(map, ISL_MAP_NORMALIZED);
3844 map->p[i] = map->p[map->n-1];
3846 map->n--;
3849 return map;
3852 struct isl_set *isl_set_remove_empty_parts(struct isl_set *set)
3854 return (struct isl_set *)
3855 isl_map_remove_empty_parts((struct isl_map *)set);
3858 struct isl_basic_map *isl_map_copy_basic_map(struct isl_map *map)
3860 struct isl_basic_map *bmap;
3861 if (!map || map->n == 0)
3862 return NULL;
3863 bmap = map->p[map->n-1];
3864 isl_assert(map->ctx, ISL_F_ISSET(bmap, ISL_BASIC_SET_FINAL), return NULL);
3865 return isl_basic_map_copy(bmap);
3868 struct isl_basic_set *isl_set_copy_basic_set(struct isl_set *set)
3870 (struct isl_basic_set *)isl_map_copy_basic_map((struct isl_map *)set);
3873 struct isl_map *isl_map_drop_basic_map(struct isl_map *map,
3874 struct isl_basic_map *bmap)
3876 int i;
3878 if (!map || !bmap)
3879 goto error;
3880 for (i = map->n-1; i >= 0; --i) {
3881 if (map->p[i] != bmap)
3882 continue;
3883 map = isl_map_cow(map);
3884 if (!map)
3885 goto error;
3886 isl_basic_map_free(map->p[i]);
3887 if (i != map->n-1) {
3888 ISL_F_CLR(map, ISL_SET_NORMALIZED);
3889 map->p[i] = map->p[map->n-1];
3891 map->n--;
3892 return map;
3894 isl_basic_map_free(bmap);
3895 return map;
3896 error:
3897 isl_map_free(map);
3898 isl_basic_map_free(bmap);
3899 return NULL;
3902 struct isl_set *isl_set_drop_basic_set(struct isl_set *set,
3903 struct isl_basic_set *bset)
3905 (struct isl_set *)isl_map_drop_basic_map((struct isl_map *)set,
3906 (struct isl_basic_map *)bset);
3909 /* Given two _disjoint_ basic sets bset1 and bset2, check whether
3910 * for any common value of the parameters and dimensions preceding dim
3911 * in both basic sets, the values of dimension pos in bset1 are
3912 * smaller or larger than those in bset2.
3914 * Returns
3915 * 1 if bset1 follows bset2
3916 * -1 if bset1 precedes bset2
3917 * 0 if bset1 and bset2 are incomparable
3918 * -2 if some error occurred.
3920 int isl_basic_set_compare_at(struct isl_basic_set *bset1,
3921 struct isl_basic_set *bset2, int pos)
3923 struct isl_dim *dims;
3924 struct isl_basic_map *bmap1 = NULL;
3925 struct isl_basic_map *bmap2 = NULL;
3926 struct isl_ctx *ctx;
3927 struct isl_vec *obj;
3928 unsigned total;
3929 unsigned nparam;
3930 unsigned dim1, dim2;
3931 isl_int num, den;
3932 enum isl_lp_result res;
3933 int cmp;
3935 if (!bset1 || !bset2)
3936 return -2;
3938 nparam = isl_basic_set_n_param(bset1);
3939 dim1 = isl_basic_set_n_dim(bset1);
3940 dim2 = isl_basic_set_n_dim(bset2);
3941 dims = isl_dim_alloc(bset1->ctx, nparam, pos, dim1 - pos);
3942 bmap1 = isl_basic_map_from_basic_set(isl_basic_set_copy(bset1), dims);
3943 dims = isl_dim_alloc(bset2->ctx, nparam, pos, dim2 - pos);
3944 bmap2 = isl_basic_map_from_basic_set(isl_basic_set_copy(bset2), dims);
3945 if (!bmap1 || !bmap2)
3946 goto error;
3947 bmap1 = isl_basic_map_cow(bmap1);
3948 bmap1 = isl_basic_map_extend(bmap1, nparam,
3949 pos, (dim1 - pos) + (dim2 - pos),
3950 bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
3951 bmap1 = add_constraints(bmap1, bmap2, 0, dim1 - pos);
3952 if (!bmap1)
3953 goto error;
3954 total = isl_basic_map_total_dim(bmap1);
3955 ctx = bmap1->ctx;
3956 obj = isl_vec_alloc(ctx, 1 + total);
3957 isl_seq_clr(obj->block.data, 1 + total);
3958 isl_int_set_si(obj->block.data[1+nparam+pos], 1);
3959 isl_int_set_si(obj->block.data[1+nparam+pos+(dim1-pos)], -1);
3960 if (!obj)
3961 goto error;
3962 isl_int_init(num);
3963 isl_int_init(den);
3964 res = isl_solve_lp(bmap1, 0, obj->block.data, ctx->one, &num, &den);
3965 if (res == isl_lp_empty)
3966 cmp = 0;
3967 else if (res == isl_lp_ok && isl_int_is_pos(num))
3968 cmp = 1;
3969 else if ((res == isl_lp_ok && isl_int_is_neg(num)) ||
3970 res == isl_lp_unbounded)
3971 cmp = -1;
3972 else
3973 cmp = -2;
3974 isl_int_clear(num);
3975 isl_int_clear(den);
3976 isl_basic_map_free(bmap1);
3977 isl_vec_free(ctx, obj);
3978 return cmp;
3979 error:
3980 isl_basic_map_free(bmap1);
3981 isl_basic_map_free(bmap2);
3982 return -2;
3985 static int isl_basic_map_fast_has_fixed_var(struct isl_basic_map *bmap,
3986 unsigned pos, isl_int *val)
3988 int i;
3989 int d;
3990 unsigned total;
3992 if (!bmap)
3993 return -1;
3994 total = isl_basic_map_total_dim(bmap);
3995 for (i = 0, d = total-1; i < bmap->n_eq && d+1 > pos; ++i) {
3996 for (; d+1 > pos; --d)
3997 if (!isl_int_is_zero(bmap->eq[i][1+d]))
3998 break;
3999 if (d != pos)
4000 continue;
4001 if (isl_seq_first_non_zero(bmap->eq[i]+1, d) != -1)
4002 return 0;
4003 if (isl_seq_first_non_zero(bmap->eq[i]+1+d+1, total-d-1) != -1)
4004 return 0;
4005 if (!isl_int_is_one(bmap->eq[i][1+d]))
4006 return 0;
4007 if (val)
4008 isl_int_neg(*val, bmap->eq[i][0]);
4009 return 1;
4011 return 0;
4014 static int isl_map_fast_has_fixed_var(struct isl_map *map,
4015 unsigned pos, isl_int *val)
4017 int i;
4018 isl_int v;
4019 isl_int tmp;
4020 int fixed;
4022 if (!map)
4023 return -1;
4024 if (map->n == 0)
4025 return 0;
4026 if (map->n == 1)
4027 return isl_basic_map_fast_has_fixed_var(map->p[0], pos, val);
4028 isl_int_init(v);
4029 isl_int_init(tmp);
4030 fixed = isl_basic_map_fast_has_fixed_var(map->p[0], pos, &v);
4031 for (i = 1; fixed == 1 && i < map->n; ++i) {
4032 fixed = isl_basic_map_fast_has_fixed_var(map->p[i], pos, &tmp);
4033 if (fixed == 1 && isl_int_ne(tmp, v))
4034 fixed = 0;
4036 if (val)
4037 isl_int_set(*val, v);
4038 isl_int_clear(tmp);
4039 isl_int_clear(v);
4040 return fixed;
4043 static int isl_set_fast_has_fixed_var(struct isl_set *set, unsigned pos,
4044 isl_int *val)
4046 return isl_map_fast_has_fixed_var((struct isl_map *)set, pos, val);
4049 int isl_basic_map_fast_is_fixed(struct isl_basic_map *bmap,
4050 enum isl_dim_type type, unsigned pos, isl_int *val)
4052 if (pos >= isl_basic_map_dim(bmap, type))
4053 return -1;
4054 return isl_basic_map_fast_has_fixed_var(bmap,
4055 isl_basic_map_offset(bmap, type) - 1 + pos, val);
4058 /* Check if dimension dim has fixed value and if so and if val is not NULL,
4059 * then return this fixed value in *val.
4061 int isl_set_fast_dim_is_fixed(struct isl_set *set, unsigned dim, isl_int *val)
4063 return isl_set_fast_has_fixed_var(set, isl_set_n_param(set) + dim, val);
4066 /* Check if input variable in has fixed value and if so and if val is not NULL,
4067 * then return this fixed value in *val.
4069 int isl_map_fast_input_is_fixed(struct isl_map *map, unsigned in, isl_int *val)
4071 return isl_map_fast_has_fixed_var(map, isl_map_n_param(map) + in, val);
4074 /* Check if dimension dim has an (obvious) fixed lower bound and if so
4075 * and if val is not NULL, then return this lower bound in *val.
4077 int isl_basic_set_fast_dim_has_fixed_lower_bound(struct isl_basic_set *bset,
4078 unsigned dim, isl_int *val)
4080 int i, i_eq = -1, i_ineq = -1;
4081 isl_int *c;
4082 unsigned total;
4083 unsigned nparam;
4085 if (!bset)
4086 return -1;
4087 total = isl_basic_set_total_dim(bset);
4088 nparam = isl_basic_set_n_param(bset);
4089 for (i = 0; i < bset->n_eq; ++i) {
4090 if (isl_int_is_zero(bset->eq[i][1+nparam+dim]))
4091 continue;
4092 if (i_eq != -1)
4093 return 0;
4094 i_eq = i;
4096 for (i = 0; i < bset->n_ineq; ++i) {
4097 if (!isl_int_is_pos(bset->ineq[i][1+nparam+dim]))
4098 continue;
4099 if (i_eq != -1 || i_ineq != -1)
4100 return 0;
4101 i_ineq = i;
4103 if (i_eq == -1 && i_ineq == -1)
4104 return 0;
4105 c = i_eq != -1 ? bset->eq[i_eq] : bset->ineq[i_ineq];
4106 /* The coefficient should always be one due to normalization. */
4107 if (!isl_int_is_one(c[1+nparam+dim]))
4108 return 0;
4109 if (isl_seq_first_non_zero(c+1, nparam+dim) != -1)
4110 return 0;
4111 if (isl_seq_first_non_zero(c+1+nparam+dim+1,
4112 total - nparam - dim - 1) != -1)
4113 return 0;
4114 if (val)
4115 isl_int_neg(*val, c[0]);
4116 return 1;
4119 int isl_set_fast_dim_has_fixed_lower_bound(struct isl_set *set,
4120 unsigned dim, isl_int *val)
4122 int i;
4123 isl_int v;
4124 isl_int tmp;
4125 int fixed;
4127 if (!set)
4128 return -1;
4129 if (set->n == 0)
4130 return 0;
4131 if (set->n == 1)
4132 return isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[0],
4133 dim, val);
4134 isl_int_init(v);
4135 isl_int_init(tmp);
4136 fixed = isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[0],
4137 dim, &v);
4138 for (i = 1; fixed == 1 && i < set->n; ++i) {
4139 fixed = isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[i],
4140 dim, &tmp);
4141 if (fixed == 1 && isl_int_ne(tmp, v))
4142 fixed = 0;
4144 if (val)
4145 isl_int_set(*val, v);
4146 isl_int_clear(tmp);
4147 isl_int_clear(v);
4148 return fixed;
4151 struct constraint {
4152 unsigned size;
4153 isl_int *c;
4156 static int qsort_constraint_cmp(const void *p1, const void *p2)
4158 const struct constraint *c1 = (const struct constraint *)p1;
4159 const struct constraint *c2 = (const struct constraint *)p2;
4160 unsigned size = isl_min(c1->size, c2->size);
4161 return isl_seq_cmp(c1->c, c2->c, size);
4164 static struct isl_basic_map *isl_basic_map_sort_constraints(
4165 struct isl_basic_map *bmap)
4167 int i;
4168 struct constraint *c;
4169 unsigned total;
4171 if (!bmap)
4172 return NULL;
4173 total = isl_basic_map_total_dim(bmap);
4174 c = isl_alloc_array(bmap->ctx, struct constraint, bmap->n_ineq);
4175 if (!c)
4176 goto error;
4177 for (i = 0; i < bmap->n_ineq; ++i) {
4178 c[i].size = total;
4179 c[i].c = bmap->ineq[i];
4181 qsort(c, bmap->n_ineq, sizeof(struct constraint), qsort_constraint_cmp);
4182 for (i = 0; i < bmap->n_ineq; ++i)
4183 bmap->ineq[i] = c[i].c;
4184 free(c);
4185 return bmap;
4186 error:
4187 isl_basic_map_free(bmap);
4188 return NULL;
4191 struct isl_basic_map *isl_basic_map_normalize(struct isl_basic_map *bmap)
4193 if (!bmap)
4194 return NULL;
4195 if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_NORMALIZED))
4196 return bmap;
4197 bmap = isl_basic_map_convex_hull(bmap);
4198 bmap = isl_basic_map_sort_constraints(bmap);
4199 ISL_F_SET(bmap, ISL_BASIC_MAP_NORMALIZED);
4200 return bmap;
4203 struct isl_basic_set *isl_basic_set_normalize(struct isl_basic_set *bset)
4205 return (struct isl_basic_set *)isl_basic_map_normalize(
4206 (struct isl_basic_map *)bset);
4209 static int isl_basic_map_fast_cmp(const struct isl_basic_map *bmap1,
4210 const struct isl_basic_map *bmap2)
4212 int i, cmp;
4213 unsigned total;
4215 if (bmap1 == bmap2)
4216 return 0;
4217 if (isl_basic_map_n_param(bmap1) != isl_basic_map_n_param(bmap2))
4218 return isl_basic_map_n_param(bmap1) - isl_basic_map_n_param(bmap2);
4219 if (isl_basic_map_n_in(bmap1) != isl_basic_map_n_in(bmap2))
4220 return isl_basic_map_n_out(bmap1) - isl_basic_map_n_out(bmap2);
4221 if (isl_basic_map_n_out(bmap1) != isl_basic_map_n_out(bmap2))
4222 return isl_basic_map_n_out(bmap1) - isl_basic_map_n_out(bmap2);
4223 if (ISL_F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY) &&
4224 ISL_F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
4225 return 0;
4226 if (ISL_F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY))
4227 return 1;
4228 if (ISL_F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
4229 return -1;
4230 if (bmap1->n_eq != bmap2->n_eq)
4231 return bmap1->n_eq - bmap2->n_eq;
4232 if (bmap1->n_ineq != bmap2->n_ineq)
4233 return bmap1->n_ineq - bmap2->n_ineq;
4234 if (bmap1->n_div != bmap2->n_div)
4235 return bmap1->n_div - bmap2->n_div;
4236 total = isl_basic_map_total_dim(bmap1);
4237 for (i = 0; i < bmap1->n_eq; ++i) {
4238 cmp = isl_seq_cmp(bmap1->eq[i], bmap2->eq[i], 1+total);
4239 if (cmp)
4240 return cmp;
4242 for (i = 0; i < bmap1->n_ineq; ++i) {
4243 cmp = isl_seq_cmp(bmap1->ineq[i], bmap2->ineq[i], 1+total);
4244 if (cmp)
4245 return cmp;
4247 for (i = 0; i < bmap1->n_div; ++i) {
4248 cmp = isl_seq_cmp(bmap1->div[i], bmap2->div[i], 1+1+total);
4249 if (cmp)
4250 return cmp;
4252 return 0;
4255 static int isl_basic_map_fast_is_equal(struct isl_basic_map *bmap1,
4256 struct isl_basic_map *bmap2)
4258 return isl_basic_map_fast_cmp(bmap1, bmap2) == 0;
4261 static int qsort_bmap_cmp(const void *p1, const void *p2)
4263 const struct isl_basic_map *bmap1 = *(const struct isl_basic_map **)p1;
4264 const struct isl_basic_map *bmap2 = *(const struct isl_basic_map **)p2;
4266 return isl_basic_map_fast_cmp(bmap1, bmap2);
4269 /* We normalize in place, but if anything goes wrong we need
4270 * to return NULL, so we need to make sure we don't change the
4271 * meaning of any possible other copies of map.
4273 struct isl_map *isl_map_normalize(struct isl_map *map)
4275 int i, j;
4276 struct isl_basic_map *bmap;
4278 if (!map)
4279 return NULL;
4280 if (ISL_F_ISSET(map, ISL_MAP_NORMALIZED))
4281 return map;
4282 for (i = 0; i < map->n; ++i) {
4283 bmap = isl_basic_map_normalize(isl_basic_map_copy(map->p[i]));
4284 if (!bmap)
4285 goto error;
4286 isl_basic_map_free(map->p[i]);
4287 map->p[i] = bmap;
4289 qsort(map->p, map->n, sizeof(struct isl_basic_map *), qsort_bmap_cmp);
4290 ISL_F_SET(map, ISL_MAP_NORMALIZED);
4291 map = isl_map_remove_empty_parts(map);
4292 if (!map)
4293 return NULL;
4294 for (i = map->n - 1; i >= 1; --i) {
4295 if (!isl_basic_map_fast_is_equal(map->p[i-1], map->p[i]))
4296 continue;
4297 isl_basic_map_free(map->p[i-1]);
4298 for (j = i; j < map->n; ++j)
4299 map->p[j-1] = map->p[j];
4300 map->n--;
4302 return map;
4303 error:
4304 isl_map_free(map);
4305 return NULL;
4309 struct isl_set *isl_set_normalize(struct isl_set *set)
4311 return (struct isl_set *)isl_map_normalize((struct isl_map *)set);
4314 int isl_map_fast_is_equal(struct isl_map *map1, struct isl_map *map2)
4316 int i;
4317 int equal;
4319 if (!map1 || !map2)
4320 return -1;
4322 if (map1 == map2)
4323 return 1;
4324 if (!isl_dim_equal(map1->dim, map2->dim))
4325 return 0;
4327 map1 = isl_map_copy(map1);
4328 map2 = isl_map_copy(map2);
4329 map1 = isl_map_normalize(map1);
4330 map2 = isl_map_normalize(map2);
4331 if (!map1 || !map2)
4332 goto error;
4333 equal = map1->n == map2->n;
4334 for (i = 0; equal && i < map1->n; ++i) {
4335 equal = isl_basic_map_fast_is_equal(map1->p[i], map2->p[i]);
4336 if (equal < 0)
4337 goto error;
4339 isl_map_free(map1);
4340 isl_map_free(map2);
4341 return equal;
4342 error:
4343 isl_map_free(map1);
4344 isl_map_free(map2);
4345 return -1;
4348 int isl_set_fast_is_equal(struct isl_set *set1, struct isl_set *set2)
4350 return isl_map_fast_is_equal((struct isl_map *)set1,
4351 (struct isl_map *)set2);
4354 /* Return an interval that ranges from min to max (inclusive)
4356 struct isl_basic_set *isl_basic_set_interval(struct isl_ctx *ctx,
4357 isl_int min, isl_int max)
4359 int k;
4360 struct isl_basic_set *bset = NULL;
4362 bset = isl_basic_set_alloc(ctx, 0, 1, 0, 0, 2);
4363 if (!bset)
4364 goto error;
4366 k = isl_basic_set_alloc_inequality(bset);
4367 if (k < 0)
4368 goto error;
4369 isl_int_set_si(bset->ineq[k][1], 1);
4370 isl_int_neg(bset->ineq[k][0], min);
4372 k = isl_basic_set_alloc_inequality(bset);
4373 if (k < 0)
4374 goto error;
4375 isl_int_set_si(bset->ineq[k][1], -1);
4376 isl_int_set(bset->ineq[k][0], max);
4378 return bset;
4379 error:
4380 isl_basic_set_free(bset);
4381 return NULL;
4384 /* Return the Cartesian product of the basic sets in list (in the given order).
4386 struct isl_basic_set *isl_basic_set_product(struct isl_basic_set_list *list)
4388 int i;
4389 unsigned dim;
4390 unsigned nparam;
4391 unsigned extra;
4392 unsigned n_eq;
4393 unsigned n_ineq;
4394 struct isl_basic_set *product = NULL;
4396 if (!list)
4397 goto error;
4398 isl_assert(list->ctx, list->n > 0, goto error);
4399 isl_assert(list->ctx, list->p[0], goto error);
4400 nparam = isl_basic_set_n_param(list->p[0]);
4401 dim = isl_basic_set_n_dim(list->p[0]);
4402 extra = list->p[0]->n_div;
4403 n_eq = list->p[0]->n_eq;
4404 n_ineq = list->p[0]->n_ineq;
4405 for (i = 1; i < list->n; ++i) {
4406 isl_assert(list->ctx, list->p[i], goto error);
4407 isl_assert(list->ctx,
4408 nparam == isl_basic_set_n_param(list->p[i]), goto error);
4409 dim += isl_basic_set_n_dim(list->p[i]);
4410 extra += list->p[i]->n_div;
4411 n_eq += list->p[i]->n_eq;
4412 n_ineq += list->p[i]->n_ineq;
4414 product = isl_basic_set_alloc(list->ctx, nparam, dim, extra,
4415 n_eq, n_ineq);
4416 if (!product)
4417 goto error;
4418 dim = 0;
4419 for (i = 0; i < list->n; ++i) {
4420 isl_basic_set_add_constraints(product,
4421 isl_basic_set_copy(list->p[i]), dim);
4422 dim += isl_basic_set_n_dim(list->p[i]);
4424 isl_basic_set_list_free(list);
4425 return product;
4426 error:
4427 isl_basic_set_free(product);
4428 isl_basic_set_list_free(list);
4429 return NULL;
4432 struct isl_basic_map *isl_basic_map_product(
4433 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
4435 struct isl_dim *dim_result = NULL;
4436 struct isl_basic_map *bmap;
4437 unsigned in1, in2, out1, out2, nparam, total, pos;
4438 struct isl_dim_map *dim_map1, *dim_map2;
4440 if (!bmap1 || !bmap2)
4441 goto error;
4443 isl_assert(map1->ctx, isl_dim_match(bmap1->dim, isl_dim_param,
4444 bmap2->dim, isl_dim_param), goto error);
4445 dim_result = isl_dim_product(isl_dim_copy(bmap1->dim),
4446 isl_dim_copy(bmap2->dim));
4448 in1 = isl_basic_map_n_in(bmap1);
4449 in2 = isl_basic_map_n_in(bmap2);
4450 out1 = isl_basic_map_n_out(bmap1);
4451 out2 = isl_basic_map_n_out(bmap2);
4452 nparam = isl_basic_map_n_param(bmap1);
4454 total = nparam + in1 + in2 + out1 + out2 + bmap1->n_div + bmap2->n_div;
4455 dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
4456 dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
4457 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
4458 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
4459 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
4460 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos += in1);
4461 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += in2);
4462 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += out1);
4463 isl_dim_map_div(dim_map1, bmap1, pos += out2);
4464 isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
4466 bmap = isl_basic_map_alloc_dim(dim_result,
4467 bmap1->n_div + bmap2->n_div,
4468 bmap1->n_eq + bmap2->n_eq,
4469 bmap1->n_ineq + bmap2->n_ineq);
4470 bmap = add_constraints_dim_map(bmap, bmap1, dim_map1);
4471 bmap = add_constraints_dim_map(bmap, bmap2, dim_map2);
4472 bmap = isl_basic_map_simplify(bmap);
4473 return isl_basic_map_finalize(bmap);
4474 error:
4475 isl_basic_map_free(bmap1);
4476 isl_basic_map_free(bmap2);
4477 return NULL;
4480 /* Given two maps A -> B and C -> D, construct a map (A, C) -> (B, D)
4482 struct isl_map *isl_map_product(struct isl_map *map1, struct isl_map *map2)
4484 unsigned flags = 0;
4485 struct isl_map *result;
4486 int i, j;
4488 if (!map1 || !map2)
4489 goto error;
4491 isl_assert(map1->ctx, isl_dim_match(map1->dim, isl_dim_param,
4492 map2->dim, isl_dim_param), goto error);
4494 if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
4495 ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
4496 ISL_FL_SET(flags, ISL_MAP_DISJOINT);
4498 result = isl_map_alloc_dim(isl_dim_product(isl_dim_copy(map1->dim),
4499 isl_dim_copy(map2->dim)),
4500 map1->n * map2->n, flags);
4501 if (!result)
4502 goto error;
4503 for (i = 0; i < map1->n; ++i)
4504 for (j = 0; j < map2->n; ++j) {
4505 struct isl_basic_map *part;
4506 part = isl_basic_map_product(
4507 isl_basic_map_copy(map1->p[i]),
4508 isl_basic_map_copy(map2->p[j]));
4509 if (isl_basic_map_is_empty(part))
4510 isl_basic_map_free(part);
4511 else
4512 result = isl_map_add(result, part);
4513 if (!result)
4514 goto error;
4516 isl_map_free(map1);
4517 isl_map_free(map2);
4518 return result;
4519 error:
4520 isl_map_free(map1);
4521 isl_map_free(map2);
4522 return NULL;
4525 uint32_t isl_basic_set_get_hash(struct isl_basic_set *bset)
4527 int i;
4528 uint32_t hash;
4529 unsigned total;
4531 if (!bset)
4532 return 0;
4533 bset = isl_basic_set_copy(bset);
4534 bset = isl_basic_set_normalize(bset);
4535 if (!bset)
4536 return 0;
4537 total = isl_basic_set_total_dim(bset);
4538 isl_hash_byte(hash, bset->n_eq & 0xFF);
4539 for (i = 0; i < bset->n_eq; ++i) {
4540 uint32_t c_hash;
4541 c_hash = isl_seq_get_hash(bset->eq[i], 1 + total);
4542 isl_hash_hash(hash, c_hash);
4544 isl_hash_byte(hash, bset->n_ineq & 0xFF);
4545 for (i = 0; i < bset->n_ineq; ++i) {
4546 uint32_t c_hash;
4547 c_hash = isl_seq_get_hash(bset->ineq[i], 1 + total);
4548 isl_hash_hash(hash, c_hash);
4550 isl_hash_byte(hash, bset->n_div & 0xFF);
4551 for (i = 0; i < bset->n_div; ++i) {
4552 uint32_t c_hash;
4553 if (isl_int_is_zero(bset->div[i][0]))
4554 continue;
4555 isl_hash_byte(hash, i & 0xFF);
4556 c_hash = isl_seq_get_hash(bset->div[i], 1 + 1 + total);
4557 isl_hash_hash(hash, c_hash);
4559 isl_basic_set_free(bset);
4560 return hash;
4563 uint32_t isl_set_get_hash(struct isl_set *set)
4565 int i;
4566 uint32_t hash;
4568 if (!set)
4569 return 0;
4570 set = isl_set_copy(set);
4571 set = isl_set_normalize(set);
4572 if (!set)
4573 return 0;
4575 hash = isl_hash_init();
4576 for (i = 0; i < set->n; ++i) {
4577 uint32_t bset_hash;
4578 bset_hash = isl_basic_set_get_hash(set->p[i]);
4579 isl_hash_hash(hash, bset_hash);
4582 isl_set_free(set);
4584 return hash;
4587 /* Check if the value for dimension dim is completely determined
4588 * by the values of the other parameters and variables.
4589 * That is, check if dimension dim is involved in an equality.
4591 int isl_basic_set_dim_is_unique(struct isl_basic_set *bset, unsigned dim)
4593 int i;
4594 unsigned nparam;
4596 if (!bset)
4597 return -1;
4598 nparam = isl_basic_set_n_param(bset);
4599 for (i = 0; i < bset->n_eq; ++i)
4600 if (!isl_int_is_zero(bset->eq[i][1 + nparam + dim]))
4601 return 1;
4602 return 0;
4605 /* Check if the value for dimension dim is completely determined
4606 * by the values of the other parameters and variables.
4607 * That is, check if dimension dim is involved in an equality
4608 * for each of the subsets.
4610 int isl_set_dim_is_unique(struct isl_set *set, unsigned dim)
4612 int i;
4614 if (!set)
4615 return -1;
4616 for (i = 0; i < set->n; ++i) {
4617 int unique;
4618 unique = isl_basic_set_dim_is_unique(set->p[i], dim);
4619 if (unique != 1)
4620 return unique;
4622 return 1;