8 #include "isl_map_private.h"
9 #include "isl_map_piplib.h"
10 #include "isl_sample.h"
13 static struct isl_basic_map *basic_map_init(struct isl_ctx *ctx,
14 struct isl_basic_map *bmap,
15 unsigned nparam, unsigned n_in, unsigned n_out, unsigned extra,
16 unsigned n_eq, unsigned n_ineq)
19 size_t row_size = 1 + nparam + n_in + n_out + extra;
21 bmap->block = isl_blk_alloc(ctx, (n_eq + n_ineq) * row_size);
22 if (isl_blk_is_error(bmap->block)) {
27 bmap->eq = isl_alloc_array(ctx, isl_int *, n_eq + n_ineq);
29 isl_blk_free(ctx, bmap->block);
35 bmap->block2 = isl_blk_empty();
38 bmap->block2 = isl_blk_alloc(ctx, extra * (1 + row_size));
39 if (isl_blk_is_error(bmap->block2)) {
41 isl_blk_free(ctx, bmap->block);
46 bmap->div = isl_alloc_array(ctx, isl_int *, extra);
48 isl_blk_free(ctx, bmap->block2);
50 isl_blk_free(ctx, bmap->block);
56 for (i = 0; i < n_eq + n_ineq; ++i)
57 bmap->eq[i] = bmap->block.data + i * row_size;
59 for (i = 0; i < extra; ++i)
60 bmap->div[i] = bmap->block2.data + i * (1 + row_size);
64 bmap->c_size = n_eq + n_ineq;
65 bmap->ineq = bmap->eq + n_eq;
66 bmap->nparam = nparam;
78 struct isl_basic_set *isl_basic_set_alloc(struct isl_ctx *ctx,
79 unsigned nparam, unsigned dim, unsigned extra,
80 unsigned n_eq, unsigned n_ineq)
82 struct isl_basic_map *bmap;
83 bmap = isl_basic_map_alloc(ctx, nparam, 0, dim, extra, n_eq, n_ineq);
84 return (struct isl_basic_set *)bmap;
87 struct isl_basic_map *isl_basic_map_alloc(struct isl_ctx *ctx,
88 unsigned nparam, unsigned in, unsigned out, unsigned extra,
89 unsigned n_eq, unsigned n_ineq)
91 struct isl_basic_map *bmap;
93 bmap = isl_alloc_type(ctx, struct isl_basic_map);
97 return basic_map_init(ctx, bmap,
98 nparam, in, out, extra, n_eq, n_ineq);
101 static void dup_constraints(struct isl_ctx *ctx,
102 struct isl_basic_map *dst, struct isl_basic_map *src)
105 unsigned total = src->nparam + src->n_in + src->n_out + src->n_div;
107 for (i = 0; i < src->n_eq; ++i) {
108 int j = isl_basic_map_alloc_equality(ctx, dst);
109 isl_seq_cpy(dst->eq[j], src->eq[i], 1+total);
112 for (i = 0; i < src->n_ineq; ++i) {
113 int j = isl_basic_map_alloc_inequality(ctx, dst);
114 isl_seq_cpy(dst->ineq[j], src->ineq[i], 1+total);
117 for (i = 0; i < src->n_div; ++i) {
118 int j = isl_basic_map_alloc_div(ctx, dst);
119 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total);
121 F_SET(dst, ISL_BASIC_SET_FINAL);
124 struct isl_basic_map *isl_basic_map_dup(struct isl_ctx *ctx,
125 struct isl_basic_map *bmap)
127 struct isl_basic_map *dup;
129 dup = isl_basic_map_alloc(ctx, bmap->nparam,
130 bmap->n_in, bmap->n_out,
131 bmap->n_div, bmap->n_eq, bmap->n_ineq);
134 dup->flags = bmap->flags;
135 dup_constraints(ctx, dup, bmap);
136 dup->sample = isl_vec_copy(ctx, bmap->sample);
140 struct isl_basic_set *isl_basic_set_dup(struct isl_ctx *ctx,
141 struct isl_basic_set *bset)
143 struct isl_basic_map *dup;
145 dup = isl_basic_map_dup(ctx, (struct isl_basic_map *)bset);
146 return (struct isl_basic_set *)dup;
149 struct isl_basic_set *isl_basic_set_copy(struct isl_ctx *ctx,
150 struct isl_basic_set *bset)
155 if (F_ISSET(bset, ISL_BASIC_SET_FINAL)) {
159 return isl_basic_set_dup(ctx, bset);
162 struct isl_set *isl_set_copy(struct isl_ctx *ctx, struct isl_set *set)
171 struct isl_basic_map *isl_basic_map_copy(struct isl_ctx *ctx,
172 struct isl_basic_map *bmap)
177 if (F_ISSET(bmap, ISL_BASIC_SET_FINAL)) {
181 return isl_basic_map_dup(ctx, bmap);
184 struct isl_map *isl_map_copy(struct isl_ctx *ctx, struct isl_map *map)
193 void isl_basic_map_free(struct isl_ctx *ctx, struct isl_basic_map *bmap)
202 isl_blk_free(ctx, bmap->block2);
204 isl_blk_free(ctx, bmap->block);
205 isl_vec_free(ctx, bmap->sample);
209 void isl_basic_set_free(struct isl_ctx *ctx, struct isl_basic_set *bset)
211 isl_basic_map_free(ctx, (struct isl_basic_map *)bset);
214 int isl_basic_map_alloc_equality(struct isl_ctx *ctx,
215 struct isl_basic_map *bmap)
219 isl_assert(ctx, bmap->n_eq + bmap->n_ineq < bmap->c_size, return -1);
220 isl_assert(ctx, bmap->eq + bmap->n_eq <= bmap->ineq, return -1);
221 if (bmap->eq + bmap->n_eq == bmap->ineq) {
223 int j = isl_basic_map_alloc_inequality(ctx, bmap);
227 bmap->ineq[0] = bmap->ineq[j];
235 int isl_basic_set_alloc_equality(struct isl_ctx *ctx,
236 struct isl_basic_set *bset)
238 return isl_basic_map_alloc_equality(ctx, (struct isl_basic_map *)bset);
241 int isl_basic_map_free_equality(struct isl_ctx *ctx,
242 struct isl_basic_map *bmap, unsigned n)
244 isl_assert(ctx, n <= bmap->n_eq, return -1);
249 int isl_basic_map_drop_equality(struct isl_ctx *ctx,
250 struct isl_basic_map *bmap, unsigned pos)
253 isl_assert(ctx, pos < bmap->n_eq, return -1);
255 if (pos != bmap->n_eq - 1) {
257 bmap->eq[pos] = bmap->eq[bmap->n_eq - 1];
258 bmap->eq[bmap->n_eq - 1] = t;
264 int isl_basic_set_drop_equality(struct isl_ctx *ctx,
265 struct isl_basic_set *bset, unsigned pos)
267 return isl_basic_map_drop_equality(ctx,
268 (struct isl_basic_map *)bset, pos);
271 void isl_basic_map_inequality_to_equality(struct isl_ctx *ctx,
272 struct isl_basic_map *bmap, unsigned pos)
277 bmap->ineq[pos] = bmap->ineq[0];
278 bmap->ineq[0] = bmap->eq[bmap->n_eq];
279 bmap->eq[bmap->n_eq] = t;
285 int isl_basic_map_alloc_inequality(struct isl_ctx *ctx,
286 struct isl_basic_map *bmap)
288 isl_assert(ctx, (bmap->ineq - bmap->eq) + bmap->n_ineq < bmap->c_size,
290 F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
291 F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
292 return bmap->n_ineq++;
295 int isl_basic_set_alloc_inequality(struct isl_ctx *ctx,
296 struct isl_basic_set *bset)
298 return isl_basic_map_alloc_inequality(ctx, (struct isl_basic_map *)bset);
301 int isl_basic_map_free_inequality(struct isl_ctx *ctx,
302 struct isl_basic_map *bmap, unsigned n)
304 isl_assert(ctx, n <= bmap->n_ineq, return -1);
309 int isl_basic_set_free_inequality(struct isl_ctx *ctx,
310 struct isl_basic_set *bset, unsigned n)
312 return isl_basic_map_free_inequality(ctx,
313 (struct isl_basic_map *)bset, n);
316 int isl_basic_map_drop_inequality(struct isl_ctx *ctx,
317 struct isl_basic_map *bmap, unsigned pos)
320 isl_assert(ctx, pos < bmap->n_ineq, return -1);
322 if (pos != bmap->n_ineq - 1) {
324 bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
325 bmap->ineq[bmap->n_ineq - 1] = t;
331 int isl_basic_set_drop_inequality(struct isl_ctx *ctx,
332 struct isl_basic_set *bset, unsigned pos)
334 return isl_basic_map_drop_inequality(ctx,
335 (struct isl_basic_map *)bset, pos);
338 int isl_basic_map_alloc_div(struct isl_ctx *ctx,
339 struct isl_basic_map *bmap)
341 isl_assert(ctx, bmap->n_div < bmap->extra, return -1);
342 return bmap->n_div++;
345 int isl_basic_map_free_div(struct isl_ctx *ctx,
346 struct isl_basic_map *bmap, unsigned n)
348 isl_assert(ctx, n <= bmap->n_div, return -1);
353 /* Copy constraint from src to dst, putting the vars of src at offset
354 * dim_off in dst and the divs of src at offset div_off in dst.
355 * If both sets are actually map, then dim_off applies to the input
358 static void copy_constraint(struct isl_basic_map *dst_map, isl_int *dst,
359 struct isl_basic_map *src_map, isl_int *src,
360 unsigned in_off, unsigned out_off, unsigned div_off)
362 isl_int_set(dst[0], src[0]);
363 isl_seq_cpy(dst+1, src+1, isl_min(dst_map->nparam, src_map->nparam));
364 if (dst_map->nparam > src_map->nparam)
365 isl_seq_clr(dst+1+src_map->nparam,
366 dst_map->nparam - src_map->nparam);
367 isl_seq_clr(dst+1+dst_map->nparam, in_off);
368 isl_seq_cpy(dst+1+dst_map->nparam+in_off,
369 src+1+src_map->nparam,
370 isl_min(dst_map->n_in-in_off, src_map->n_in));
371 if (dst_map->n_in-in_off > src_map->n_in)
372 isl_seq_clr(dst+1+dst_map->nparam+in_off+src_map->n_in,
373 dst_map->n_in - in_off - src_map->n_in);
374 isl_seq_clr(dst+1+dst_map->nparam+dst_map->n_in, out_off);
375 isl_seq_cpy(dst+1+dst_map->nparam+dst_map->n_in+out_off,
376 src+1+src_map->nparam+src_map->n_in,
377 isl_min(dst_map->n_out-out_off, src_map->n_out));
378 if (dst_map->n_out-out_off > src_map->n_out)
379 isl_seq_clr(dst+1+dst_map->nparam+dst_map->n_in+out_off+src_map->n_out,
380 dst_map->n_out - out_off - src_map->n_out);
381 isl_seq_clr(dst+1+dst_map->nparam+dst_map->n_in+dst_map->n_out, div_off);
382 isl_seq_cpy(dst+1+dst_map->nparam+dst_map->n_in+dst_map->n_out+div_off,
383 src+1+src_map->nparam+src_map->n_in+src_map->n_out,
384 isl_min(dst_map->extra-div_off, src_map->extra));
385 if (dst_map->extra-div_off > src_map->extra)
386 isl_seq_clr(dst+1+dst_map->nparam+dst_map->n_in+dst_map->n_out+
387 div_off+src_map->extra,
388 dst_map->extra - div_off - src_map->extra);
391 static void copy_div(struct isl_basic_map *dst_map, isl_int *dst,
392 struct isl_basic_map *src_map, isl_int *src,
393 unsigned in_off, unsigned out_off, unsigned div_off)
395 isl_int_set(dst[0], src[0]);
396 copy_constraint(dst_map, dst+1, src_map, src+1, in_off, out_off, div_off);
399 static struct isl_basic_map *add_constraints(struct isl_ctx *ctx,
400 struct isl_basic_map *bmap1,
401 struct isl_basic_map *bmap2, unsigned i_pos, unsigned o_pos)
404 unsigned div_off = bmap1->n_div;
406 for (i = 0; i < bmap2->n_eq; ++i) {
407 int i1 = isl_basic_map_alloc_equality(ctx, bmap1);
410 copy_constraint(bmap1, bmap1->eq[i1], bmap2, bmap2->eq[i],
411 i_pos, o_pos, div_off);
414 for (i = 0; i < bmap2->n_ineq; ++i) {
415 int i1 = isl_basic_map_alloc_inequality(ctx, bmap1);
418 copy_constraint(bmap1, bmap1->ineq[i1], bmap2, bmap2->ineq[i],
419 i_pos, o_pos, div_off);
422 for (i = 0; i < bmap2->n_div; ++i) {
423 int i1 = isl_basic_map_alloc_div(ctx, bmap1);
426 copy_div(bmap1, bmap1->div[i1], bmap2, bmap2->div[i],
427 i_pos, o_pos, div_off);
430 isl_basic_map_free(ctx, bmap2);
435 isl_basic_map_free(ctx, bmap1);
436 isl_basic_map_free(ctx, bmap2);
440 struct isl_basic_map *isl_basic_map_extend(struct isl_ctx *ctx,
441 struct isl_basic_map *base,
442 unsigned nparam, unsigned n_in, unsigned n_out, unsigned extra,
443 unsigned n_eq, unsigned n_ineq)
445 struct isl_basic_map *ext;
449 base = isl_basic_map_cow(ctx, base);
453 dims_ok = base && base->nparam == nparam &&
454 base->n_in == n_in && base->n_out == n_out &&
455 base->extra >= base->n_div + extra;
457 if (dims_ok && n_eq == 0 && n_ineq == 0)
461 isl_assert(ctx, base->nparam <= nparam, goto error);
462 isl_assert(ctx, base->n_in <= n_in, goto error);
463 isl_assert(ctx, base->n_out <= n_out, goto error);
464 extra += base->extra;
466 n_ineq += base->n_ineq;
469 ext = isl_basic_map_alloc(ctx, nparam, n_in, n_out,
470 extra, n_eq, n_ineq);
477 ext = add_constraints(ctx, ext, base, 0, 0);
480 F_CLR(ext, ISL_BASIC_SET_FINAL);
486 isl_basic_map_free(ctx, base);
490 struct isl_basic_set *isl_basic_set_extend(struct isl_ctx *ctx,
491 struct isl_basic_set *base,
492 unsigned nparam, unsigned dim, unsigned extra,
493 unsigned n_eq, unsigned n_ineq)
495 return (struct isl_basic_set *)
496 isl_basic_map_extend(ctx, (struct isl_basic_map *)base,
497 nparam, 0, dim, extra, n_eq, n_ineq);
500 struct isl_basic_set *isl_basic_set_cow(struct isl_ctx *ctx,
501 struct isl_basic_set *bset)
503 return (struct isl_basic_set *)
504 isl_basic_map_cow(ctx, (struct isl_basic_map *)bset);
507 struct isl_basic_map *isl_basic_map_cow(struct isl_ctx *ctx,
508 struct isl_basic_map *bmap)
515 bmap = isl_basic_map_dup(ctx, bmap);
517 F_CLR(bmap, ISL_BASIC_SET_FINAL);
521 struct isl_set *isl_set_cow(struct isl_ctx *ctx, struct isl_set *set)
529 return isl_set_dup(ctx, set);
532 struct isl_map *isl_map_cow(struct isl_ctx *ctx, struct isl_map *map)
540 return isl_map_dup(ctx, map);
543 static void swap_vars(struct isl_blk blk, isl_int *a,
544 unsigned a_len, unsigned b_len)
546 isl_seq_cpy(blk.data, a+a_len, b_len);
547 isl_seq_cpy(blk.data+b_len, a, a_len);
548 isl_seq_cpy(a, blk.data, b_len+a_len);
551 struct isl_basic_set *isl_basic_set_swap_vars(struct isl_ctx *ctx,
552 struct isl_basic_set *bset, unsigned n)
560 isl_assert(ctx, n <= bset->dim, goto error);
565 bset = isl_basic_set_cow(ctx, bset);
569 blk = isl_blk_alloc(ctx, bset->dim);
570 if (isl_blk_is_error(blk))
573 for (i = 0; i < bset->n_eq; ++i)
575 bset->eq[i]+1+bset->nparam, n, bset->dim - n);
577 for (i = 0; i < bset->n_ineq; ++i)
579 bset->ineq[i]+1+bset->nparam, n, bset->dim - n);
581 for (i = 0; i < bset->n_div; ++i)
583 bset->div[i]+1+1+bset->nparam, n, bset->dim - n);
585 isl_blk_free(ctx, blk);
590 isl_basic_set_free(ctx, bset);
594 struct isl_set *isl_set_swap_vars(struct isl_ctx *ctx,
595 struct isl_set *set, unsigned n)
598 set = isl_set_cow(ctx, set);
602 for (i = 0; i < set->n; ++i) {
603 set->p[i] = isl_basic_set_swap_vars(ctx, set->p[i], n);
605 isl_set_free(ctx, set);
612 static void constraint_drop_vars(isl_int *c, unsigned n, unsigned rem)
614 isl_seq_cpy(c, c + n, rem);
615 isl_seq_clr(c + rem, n);
618 struct isl_basic_set *isl_basic_set_drop_vars(struct isl_ctx *ctx,
619 struct isl_basic_set *bset, unsigned first, unsigned n)
626 isl_assert(ctx, first + n <= bset->dim, goto error);
631 bset = isl_basic_set_cow(ctx, bset);
635 for (i = 0; i < bset->n_eq; ++i)
636 constraint_drop_vars(bset->eq[i]+1+bset->nparam+first, n,
637 (bset->dim-first-n)+bset->extra);
639 for (i = 0; i < bset->n_ineq; ++i)
640 constraint_drop_vars(bset->ineq[i]+1+bset->nparam+first, n,
641 (bset->dim-first-n)+bset->extra);
643 for (i = 0; i < bset->n_div; ++i)
644 constraint_drop_vars(bset->div[i]+1+1+bset->nparam+first, n,
645 (bset->dim-first-n)+bset->extra);
650 return isl_basic_set_finalize(ctx, bset);
652 isl_basic_set_free(ctx, bset);
656 struct isl_set *isl_set_drop_vars(struct isl_ctx *ctx,
657 struct isl_set *set, unsigned first, unsigned n)
664 isl_assert(ctx, first + n <= set->dim, goto error);
668 set = isl_set_cow(ctx, set);
672 for (i = 0; i < set->n; ++i) {
673 set->p[i] = isl_basic_set_drop_vars(ctx, set->p[i], first, n);
681 isl_set_free(ctx, set);
686 * We don't cow, as the div is assumed to be redundant.
688 static struct isl_basic_map *isl_basic_map_drop_div(struct isl_ctx *ctx,
689 struct isl_basic_map *bmap, unsigned div)
697 pos = 1 + bmap->nparam + bmap->n_in + bmap->n_out + div;
699 isl_assert(ctx, div < bmap->n_div, goto error);
701 for (i = 0; i < bmap->n_eq; ++i)
702 constraint_drop_vars(bmap->eq[i]+pos, 1, bmap->extra-div-1);
704 for (i = 0; i < bmap->n_ineq; ++i) {
705 if (!isl_int_is_zero(bmap->ineq[i][pos])) {
706 isl_basic_map_drop_inequality(ctx, bmap, i);
710 constraint_drop_vars(bmap->ineq[i]+pos, 1, bmap->extra-div-1);
713 for (i = 0; i < bmap->n_div; ++i)
714 constraint_drop_vars(bmap->div[i]+1+pos, 1, bmap->extra-div-1);
716 if (div != bmap->n_div - 1) {
718 isl_int *t = bmap->div[div];
720 for (j = div; j < bmap->n_div - 1; ++j)
721 bmap->div[j] = bmap->div[j+1];
723 bmap->div[bmap->n_div - 1] = t;
725 isl_basic_map_free_div(ctx, bmap, 1);
729 isl_basic_map_free(ctx, bmap);
733 static void eliminate_div(struct isl_ctx *ctx, struct isl_basic_map *bmap,
734 isl_int *eq, unsigned div)
737 unsigned pos = 1 + bmap->nparam + bmap->n_in + bmap->n_out + div;
739 len = 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
741 for (i = 0; i < bmap->n_eq; ++i)
742 if (bmap->eq[i] != eq)
743 isl_seq_elim(bmap->eq[i], eq, pos, len, NULL);
745 for (i = 0; i < bmap->n_ineq; ++i)
746 isl_seq_elim(bmap->ineq[i], eq, pos, len, NULL);
748 /* We need to be careful about circular definitions,
749 * so for now we just remove the definitions of other divs that
750 * depend on this div and (possibly) recompute them later.
752 for (i = 0; i < bmap->n_div; ++i)
753 if (!isl_int_is_zero(bmap->div[i][0]) &&
754 !isl_int_is_zero(bmap->div[i][1 + pos]))
755 isl_seq_clr(bmap->div[i], 1 + len);
757 isl_basic_map_drop_div(ctx, bmap, div);
760 struct isl_basic_map *isl_basic_map_set_to_empty(
761 struct isl_ctx *ctx, struct isl_basic_map *bmap)
767 total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra;
768 isl_basic_map_free_div(ctx, bmap, bmap->n_div);
769 isl_basic_map_free_inequality(ctx, bmap, bmap->n_ineq);
771 isl_basic_map_free_equality(ctx, bmap, bmap->n_eq-1);
773 isl_basic_map_alloc_equality(ctx, bmap);
777 isl_int_set_si(bmap->eq[i][0], 1);
778 isl_seq_clr(bmap->eq[i]+1, total);
779 F_SET(bmap, ISL_BASIC_MAP_EMPTY);
780 return isl_basic_map_finalize(ctx, bmap);
782 isl_basic_map_free(ctx, bmap);
786 struct isl_basic_set *isl_basic_set_set_to_empty(
787 struct isl_ctx *ctx, struct isl_basic_set *bset)
789 return (struct isl_basic_set *)
790 isl_basic_map_set_to_empty(ctx, (struct isl_basic_map *)bset);
793 static void swap_equality(struct isl_basic_map *bmap, int a, int b)
795 isl_int *t = bmap->eq[a];
796 bmap->eq[a] = bmap->eq[b];
800 static void swap_inequality(struct isl_basic_map *bmap, int a, int b)
802 isl_int *t = bmap->ineq[a];
803 bmap->ineq[a] = bmap->ineq[b];
807 static void swap_div(struct isl_basic_map *bmap, int a, int b)
810 unsigned off = bmap->nparam + bmap->n_in + bmap->n_out;
811 isl_int *t = bmap->div[a];
812 bmap->div[a] = bmap->div[b];
815 for (i = 0; i < bmap->n_eq; ++i)
816 isl_int_swap(bmap->eq[i][1+off+a], bmap->eq[i][1+off+b]);
818 for (i = 0; i < bmap->n_ineq; ++i)
819 isl_int_swap(bmap->ineq[i][1+off+a], bmap->ineq[i][1+off+b]);
821 for (i = 0; i < bmap->n_div; ++i)
822 isl_int_swap(bmap->div[i][1+1+off+a], bmap->div[i][1+1+off+b]);
825 struct isl_basic_map *isl_basic_map_gauss(struct isl_ctx *ctx,
826 struct isl_basic_map *bmap, int *progress)
837 total_var = bmap->nparam + bmap->n_in + bmap->n_out;
838 total = total_var + bmap->n_div;
840 last_var = total - 1;
841 for (done = 0; done < bmap->n_eq; ++done) {
842 for (; last_var >= 0; --last_var) {
843 for (k = done; k < bmap->n_eq; ++k)
844 if (!isl_int_is_zero(bmap->eq[k][1+last_var]))
852 swap_equality(bmap, k, done);
853 if (isl_int_is_neg(bmap->eq[done][1+last_var]))
854 isl_seq_neg(bmap->eq[done], bmap->eq[done], 1+total);
856 for (k = 0; k < bmap->n_eq; ++k) {
859 if (isl_int_is_zero(bmap->eq[k][1+last_var]))
863 isl_seq_elim(bmap->eq[k], bmap->eq[done],
864 1+last_var, 1+total, NULL);
867 for (k = 0; k < bmap->n_ineq; ++k) {
868 if (isl_int_is_zero(bmap->ineq[k][1+last_var]))
872 isl_seq_elim(bmap->ineq[k], bmap->eq[done],
873 1+last_var, 1+total, NULL);
876 for (k = 0; k < bmap->n_div; ++k) {
877 if (isl_int_is_zero(bmap->div[k][0]))
879 if (isl_int_is_zero(bmap->div[k][1+1+last_var]))
883 isl_seq_elim(bmap->div[k]+1, bmap->eq[done],
884 1+last_var, 1+total, &bmap->div[k][0]);
887 if (last_var >= total_var &&
888 isl_int_is_zero(bmap->div[last_var - total_var][0])) {
889 unsigned div = last_var - total_var;
890 isl_seq_neg(bmap->div[div]+1, bmap->eq[done], 1+total);
891 isl_int_set_si(bmap->div[div][1+1+last_var], 0);
892 isl_int_set(bmap->div[div][0],
893 bmap->eq[done][1+last_var]);
896 if (done == bmap->n_eq)
898 for (k = done; k < bmap->n_eq; ++k) {
899 if (isl_int_is_zero(bmap->eq[k][0]))
901 return isl_basic_map_set_to_empty(ctx, bmap);
903 isl_basic_map_free_equality(ctx, bmap, bmap->n_eq-done);
907 struct isl_basic_set *isl_basic_set_gauss(struct isl_ctx *ctx,
908 struct isl_basic_set *bset, int *progress)
910 return (struct isl_basic_set*)isl_basic_map_gauss(ctx,
911 (struct isl_basic_map *)bset, progress);
914 static unsigned int round_up(unsigned int v)
925 static int hash_index(int *index, unsigned int size, int bits,
926 struct isl_basic_map *bmap, int k)
929 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
930 u_int32_t hash = isl_seq_hash(bmap->ineq[k]+1, total, bits);
931 for (h = hash; index[h]; h = (h+1) % size)
932 if (k != index[h]-1 &&
933 isl_seq_eq(bmap->ineq[k]+1,
934 bmap->ineq[index[h]-1]+1, total))
939 static struct isl_basic_map *remove_duplicate_constraints(
941 struct isl_basic_map *bmap, int *progress)
947 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
950 if (bmap->n_ineq <= 1)
953 size = round_up(4 * (bmap->n_ineq+1) / 3 - 1);
954 bits = ffs(size) - 1;
955 index = isl_alloc_array(ctx, int, size);
956 memset(index, 0, size * sizeof(int));
960 index[isl_seq_hash(bmap->ineq[0]+1, total, bits)] = 1;
961 for (k = 1; k < bmap->n_ineq; ++k) {
962 h = hash_index(index, size, bits, bmap, k);
969 if (isl_int_lt(bmap->ineq[k][0], bmap->ineq[l][0]))
970 swap_inequality(bmap, k, l);
971 isl_basic_map_drop_inequality(ctx, bmap, k);
975 for (k = 0; k < bmap->n_ineq-1; ++k) {
976 isl_seq_neg(bmap->ineq[k]+1, bmap->ineq[k]+1, total);
977 h = hash_index(index, size, bits, bmap, k);
978 isl_seq_neg(bmap->ineq[k]+1, bmap->ineq[k]+1, total);
982 isl_int_add(sum, bmap->ineq[k][0], bmap->ineq[l][0]);
983 if (isl_int_is_pos(sum))
985 if (isl_int_is_zero(sum)) {
986 /* We need to break out of the loop after these
987 * changes since the contents of the hash
988 * will no longer be valid.
989 * Plus, we probably we want to regauss first.
991 isl_basic_map_drop_inequality(ctx, bmap, l);
992 isl_basic_map_inequality_to_equality(ctx, bmap, k);
994 bmap = isl_basic_map_set_to_empty(ctx, bmap);
1003 static struct isl_basic_map *remove_duplicate_divs(struct isl_ctx *ctx,
1004 struct isl_basic_map *bmap, int *progress)
1011 unsigned total_var = bmap->nparam + bmap->n_in + bmap->n_out;
1012 unsigned total = total_var + bmap->n_div;
1014 if (bmap->n_div <= 1)
1017 for (k = bmap->n_div - 1; k >= 0; --k)
1018 if (!isl_int_is_zero(bmap->div[k][0]))
1023 size = round_up(4 * bmap->n_div / 3 - 1);
1024 bits = ffs(size) - 1;
1025 index = isl_alloc_array(ctx, int, size);
1026 memset(index, 0, size * sizeof(int));
1029 eq = isl_blk_alloc(ctx, 1+total);
1030 if (isl_blk_is_error(eq))
1033 isl_seq_clr(eq.data, 1+total);
1034 index[isl_seq_hash(bmap->div[k], 2+total, bits)] = k + 1;
1035 for (--k; k >= 0; --k) {
1038 if (isl_int_is_zero(bmap->div[k][0]))
1041 hash = isl_seq_hash(bmap->div[k], 2+total, bits);
1042 for (h = hash; index[h]; h = (h+1) % size)
1043 if (isl_seq_eq(bmap->div[k],
1044 bmap->div[index[h]-1], 2+total))
1049 isl_int_set_si(eq.data[1+total_var+k], -1);
1050 isl_int_set_si(eq.data[1+total_var+l], 1);
1051 eliminate_div(ctx, bmap, eq.data, l);
1052 isl_int_set_si(eq.data[1+total_var+k], 0);
1053 isl_int_set_si(eq.data[1+total_var+l], 0);
1058 isl_blk_free(ctx, eq);
1064 /* Elimininate divs based on equalities
1066 static struct isl_basic_map *eliminate_divs_eq(
1067 struct isl_ctx *ctx, struct isl_basic_map *bmap,
1078 off = 1 + bmap->nparam + bmap->n_in + bmap->n_out;
1080 for (d = bmap->n_div - 1; d >= 0 ; --d) {
1081 for (i = 0; i < bmap->n_eq; ++i) {
1082 if (!isl_int_is_one(bmap->eq[i][off + d]) &&
1083 !isl_int_is_negone(bmap->eq[i][off + d]))
1087 eliminate_div(ctx, bmap, bmap->eq[i], d);
1088 isl_basic_map_drop_equality(ctx, bmap, i);
1093 return eliminate_divs_eq(ctx, bmap, progress);
1097 static struct isl_basic_map *normalize_constraints(
1098 struct isl_ctx *ctx, struct isl_basic_map *bmap)
1102 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
1105 for (i = bmap->n_eq - 1; i >= 0; --i) {
1106 isl_seq_gcd(bmap->eq[i]+1, total, &gcd);
1107 if (isl_int_is_zero(gcd)) {
1108 if (!isl_int_is_zero(bmap->eq[i][0])) {
1109 bmap = isl_basic_map_set_to_empty(ctx, bmap);
1112 isl_basic_map_drop_equality(ctx, bmap, i);
1115 if (F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL))
1116 isl_int_gcd(gcd, gcd, bmap->eq[i][0]);
1117 if (isl_int_is_one(gcd))
1119 if (!isl_int_is_divisible_by(bmap->eq[i][0], gcd)) {
1120 bmap = isl_basic_map_set_to_empty(ctx, bmap);
1123 isl_seq_scale_down(bmap->eq[i], bmap->eq[i], gcd, 1+total);
1126 for (i = bmap->n_ineq - 1; i >= 0; --i) {
1127 isl_seq_gcd(bmap->ineq[i]+1, total, &gcd);
1128 if (isl_int_is_zero(gcd)) {
1129 if (isl_int_is_neg(bmap->ineq[i][0])) {
1130 bmap = isl_basic_map_set_to_empty(ctx, bmap);
1133 isl_basic_map_drop_inequality(ctx, bmap, i);
1136 if (F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL))
1137 isl_int_gcd(gcd, gcd, bmap->ineq[i][0]);
1138 if (isl_int_is_one(gcd))
1140 isl_int_fdiv_q(bmap->ineq[i][0], bmap->ineq[i][0], gcd);
1141 isl_seq_scale_down(bmap->ineq[i]+1, bmap->ineq[i]+1, gcd, total);
1149 /* Eliminate the specified variables from the constraints using
1150 * Fourier-Motzkin. The variables themselves are not removed.
1152 struct isl_basic_map *isl_basic_map_eliminate_vars(struct isl_ctx *ctx,
1153 struct isl_basic_map *bmap, int pos, unsigned n)
1162 total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
1163 extra = bmap->extra - bmap->n_div;
1165 bmap = isl_basic_map_cow(ctx, bmap);
1166 bmap = isl_basic_map_gauss(ctx, bmap, NULL);
1167 for (d = pos + n - 1; d >= pos; --d) {
1168 int n_lower, n_upper;
1171 for (i = 0; i < bmap->n_eq; ++i) {
1172 if (isl_int_is_zero(bmap->eq[i][1+d]))
1174 isl_basic_map_drop_equality(ctx, bmap, i);
1181 for (i = 0; i < bmap->n_ineq; ++i) {
1182 if (isl_int_is_pos(bmap->ineq[i][1+d]))
1184 else if (isl_int_is_neg(bmap->ineq[i][1+d]))
1187 bmap = isl_basic_map_extend(ctx, bmap,
1188 bmap->nparam, bmap->n_in, bmap->n_out, 0,
1189 0, n_lower * n_upper);
1190 for (i = bmap->n_ineq - 1; i >= 0; --i) {
1192 if (isl_int_is_zero(bmap->ineq[i][1+d]))
1195 for (j = 0; j < i; ++j) {
1196 if (isl_int_is_zero(bmap->ineq[j][1+d]))
1199 if (isl_int_sgn(bmap->ineq[i][1+d]) ==
1200 isl_int_sgn(bmap->ineq[j][1+d]))
1202 k = isl_basic_map_alloc_inequality(ctx, bmap);
1205 isl_seq_cpy(bmap->ineq[k], bmap->ineq[i],
1207 isl_seq_clr(bmap->ineq[k]+1+total, extra);
1208 isl_seq_elim(bmap->ineq[k], bmap->ineq[j],
1209 1+d, 1+total, NULL);
1211 isl_basic_map_drop_inequality(ctx, bmap, i);
1214 if (n_lower > 0 && n_upper > 0) {
1215 bmap = normalize_constraints(ctx, bmap);
1216 bmap = isl_basic_map_gauss(ctx, bmap, NULL);
1217 if (F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
1223 isl_basic_map_free(ctx, bmap);
1227 struct isl_basic_set *isl_basic_set_eliminate_vars(struct isl_ctx *ctx,
1228 struct isl_basic_set *bset, unsigned pos, unsigned n)
1230 return (struct isl_basic_set *)isl_basic_map_eliminate_vars(ctx,
1231 (struct isl_basic_map *)bset, pos, n);
1234 /* Elimininate divs based on inequalities
1236 static struct isl_basic_map *eliminate_divs_ineq(
1237 struct isl_ctx *ctx, struct isl_basic_map *bmap,
1247 off = 1 + bmap->nparam + bmap->n_in + bmap->n_out;
1249 for (d = bmap->n_div - 1; d >= 0 ; --d) {
1250 for (i = 0; i < bmap->n_eq; ++i)
1251 if (!isl_int_is_zero(bmap->eq[i][off + d]))
1255 for (i = 0; i < bmap->n_ineq; ++i)
1256 if (isl_int_abs_gt(bmap->ineq[i][off + d], ctx->one))
1258 if (i < bmap->n_ineq)
1261 bmap = isl_basic_map_eliminate_vars(ctx, bmap, (off-1)+d, 1);
1262 if (F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
1264 bmap = isl_basic_map_drop_div(ctx, bmap, d);
1271 struct isl_basic_map *isl_basic_map_simplify(
1272 struct isl_ctx *ctx, struct isl_basic_map *bmap)
1279 bmap = normalize_constraints(ctx, bmap);
1280 bmap = eliminate_divs_eq(ctx, bmap, &progress);
1281 bmap = eliminate_divs_ineq(ctx, bmap, &progress);
1282 bmap = isl_basic_map_gauss(ctx, bmap, &progress);
1283 bmap = remove_duplicate_divs(ctx, bmap, &progress);
1284 bmap = remove_duplicate_constraints(ctx, bmap, &progress);
1289 struct isl_basic_set *isl_basic_set_simplify(
1290 struct isl_ctx *ctx, struct isl_basic_set *bset)
1292 return (struct isl_basic_set *)
1293 isl_basic_map_simplify(ctx,
1294 (struct isl_basic_map *)bset);
1297 static void dump_term(struct isl_basic_map *bmap,
1298 isl_int c, int pos, FILE *out)
1300 unsigned in = bmap->n_in;
1301 unsigned dim = bmap->n_in + bmap->n_out;
1303 isl_int_print(out, c);
1305 if (!isl_int_is_one(c))
1306 isl_int_print(out, c);
1307 if (pos < 1 + bmap->nparam)
1308 fprintf(out, "p%d", pos - 1);
1309 else if (pos < 1 + bmap->nparam + in)
1310 fprintf(out, "i%d", pos - 1 - bmap->nparam);
1311 else if (pos < 1 + bmap->nparam + dim)
1312 fprintf(out, "o%d", pos - 1 - bmap->nparam - in);
1314 fprintf(out, "e%d", pos - 1 - bmap->nparam - dim);
1318 static void dump_constraint_sign(struct isl_basic_map *bmap, isl_int *c,
1319 int sign, FILE *out)
1323 unsigned len = 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
1327 for (i = 0, first = 1; i < len; ++i) {
1328 if (isl_int_sgn(c[i]) * sign <= 0)
1331 fprintf(out, " + ");
1333 isl_int_abs(v, c[i]);
1334 dump_term(bmap, v, i, out);
1341 static void dump_constraint(struct isl_basic_map *bmap, isl_int *c,
1342 const char *op, FILE *out, int indent)
1346 fprintf(out, "%*s", indent, "");
1348 dump_constraint_sign(bmap, c, 1, out);
1349 fprintf(out, " %s ", op);
1350 dump_constraint_sign(bmap, c, -1, out);
1354 for (i = bmap->n_div; i < bmap->extra; ++i) {
1355 if (isl_int_is_zero(c[1+bmap->nparam+bmap->n_in+bmap->n_out+i]))
1357 fprintf(out, "%*s", indent, "");
1358 fprintf(out, "ERROR: unused div coefficient not zero\n");
1362 static void dump_constraints(struct isl_basic_map *bmap,
1363 isl_int **c, unsigned n,
1364 const char *op, FILE *out, int indent)
1368 for (i = 0; i < n; ++i)
1369 dump_constraint(bmap, c[i], op, out, indent);
1372 static void dump_affine(struct isl_basic_map *bmap, isl_int *exp, FILE *out)
1376 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
1378 for (j = 0; j < 1 + total; ++j) {
1379 if (isl_int_is_zero(exp[j]))
1381 if (!first && isl_int_is_pos(exp[j]))
1383 dump_term(bmap, exp[j], j, out);
1388 static void dump(struct isl_basic_map *bmap, FILE *out, int indent)
1392 dump_constraints(bmap, bmap->eq, bmap->n_eq, "=", out, indent);
1393 dump_constraints(bmap, bmap->ineq, bmap->n_ineq, ">=", out, indent);
1395 for (i = 0; i < bmap->n_div; ++i) {
1396 fprintf(out, "%*s", indent, "");
1397 fprintf(out, "e%d = [(", i);
1398 dump_affine(bmap, bmap->div[i]+1, out);
1400 isl_int_print(out, bmap->div[i][0]);
1401 fprintf(out, "]\n");
1405 void isl_basic_set_dump(struct isl_ctx *ctx, struct isl_basic_set *bset,
1406 FILE *out, int indent)
1409 fprintf(out, "null basic set\n");
1413 fprintf(out, "%*s", indent, "");
1414 fprintf(out, "nparam: %d, dim: %d, extra: %d\n",
1415 bset->nparam, bset->dim, bset->extra);
1416 dump((struct isl_basic_map *)bset, out, indent);
1419 void isl_basic_map_dump(struct isl_ctx *ctx, struct isl_basic_map *bmap,
1420 FILE *out, int indent)
1423 fprintf(out, "null basic map\n");
1427 fprintf(out, "%*s", indent, "");
1428 fprintf(out, "ref: %d, nparam: %d, in: %d, out: %d, extra: %d, flags: %x\n",
1430 bmap->nparam, bmap->n_in, bmap->n_out, bmap->extra, bmap->flags);
1431 dump(bmap, out, indent);
1434 int isl_inequality_negate(struct isl_ctx *ctx,
1435 struct isl_basic_map *bmap, unsigned pos)
1437 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
1438 isl_assert(ctx, pos < bmap->n_ineq, return -1);
1439 isl_seq_neg(bmap->ineq[pos], bmap->ineq[pos], 1 + total);
1440 isl_int_sub_ui(bmap->ineq[pos][0], bmap->ineq[pos][0], 1);
1444 struct isl_set *isl_set_alloc(struct isl_ctx *ctx,
1445 unsigned nparam, unsigned dim, int n, unsigned flags)
1447 struct isl_set *set;
1449 set = isl_alloc(ctx, struct isl_set,
1450 sizeof(struct isl_set) +
1451 n * sizeof(struct isl_basic_set *));
1458 set->nparam = nparam;
1465 struct isl_set *isl_set_dup(struct isl_ctx *ctx, struct isl_set *set)
1468 struct isl_set *dup;
1470 dup = isl_set_alloc(ctx, set->nparam, set->dim, set->n, set->flags);
1473 for (i = 0; i < set->n; ++i)
1474 dup = isl_set_add(ctx, dup,
1475 isl_basic_set_dup(ctx, set->p[i]));
1479 struct isl_set *isl_set_from_basic_set(struct isl_ctx *ctx,
1480 struct isl_basic_set *bset)
1482 struct isl_set *set;
1484 set = isl_set_alloc(ctx, bset->nparam, bset->dim, 1, ISL_MAP_DISJOINT);
1486 isl_basic_set_free(ctx, bset);
1489 return isl_set_add(ctx, set, bset);
1492 struct isl_map *isl_map_from_basic_map(struct isl_ctx *ctx,
1493 struct isl_basic_map *bmap)
1495 struct isl_map *map;
1497 map = isl_map_alloc(ctx, bmap->nparam, bmap->n_in, bmap->n_out, 1,
1500 isl_basic_map_free(ctx, bmap);
1503 return isl_map_add(ctx, map, bmap);
1506 struct isl_set *isl_set_add(struct isl_ctx *ctx, struct isl_set *set,
1507 struct isl_basic_set *bset)
1511 isl_assert(ctx, set->nparam == bset->nparam, goto error);
1512 isl_assert(ctx, set->dim == bset->dim, goto error);
1513 isl_assert(ctx, set->n < set->size, goto error);
1514 set->p[set->n] = bset;
1519 isl_set_free(ctx, set);
1521 isl_basic_set_free(ctx, bset);
1525 void isl_set_free(struct isl_ctx *ctx, struct isl_set *set)
1535 for (i = 0; i < set->n; ++i)
1536 isl_basic_set_free(ctx, set->p[i]);
1540 void isl_set_dump(struct isl_ctx *ctx, struct isl_set *set, FILE *out,
1546 fprintf(out, "null set\n");
1550 fprintf(out, "%*s", indent, "");
1551 fprintf(out, "ref: %d, n: %d\n", set->ref, set->n);
1552 for (i = 0; i < set->n; ++i) {
1553 fprintf(out, "%*s", indent, "");
1554 fprintf(out, "basic set %d:\n", i);
1555 isl_basic_set_dump(ctx, set->p[i], out, indent+4);
1559 void isl_map_dump(struct isl_ctx *ctx, struct isl_map *map, FILE *out,
1565 fprintf(out, "null map\n");
1569 fprintf(out, "%*s", indent, "");
1570 fprintf(out, "ref: %d, n: %d, nparam: %d, in: %d, out: %d\n",
1571 map->ref, map->n, map->nparam, map->n_in, map->n_out);
1572 for (i = 0; i < map->n; ++i) {
1573 fprintf(out, "%*s", indent, "");
1574 fprintf(out, "basic map %d:\n", i);
1575 isl_basic_map_dump(ctx, map->p[i], out, indent+4);
1579 struct isl_basic_map *isl_basic_map_intersect_domain(
1580 struct isl_ctx *ctx, struct isl_basic_map *bmap,
1581 struct isl_basic_set *bset)
1583 struct isl_basic_map *bmap_domain;
1588 isl_assert(ctx, bset->dim == bmap->n_in, goto error);
1589 isl_assert(ctx, bset->nparam == bmap->nparam, goto error);
1591 bmap = isl_basic_map_extend(ctx, bmap,
1592 bset->nparam, bmap->n_in, bmap->n_out,
1593 bset->n_div, bset->n_eq, bset->n_ineq);
1596 bmap_domain = isl_basic_map_from_basic_set(ctx, bset,
1598 bmap = add_constraints(ctx, bmap, bmap_domain, 0, 0);
1600 bmap = isl_basic_map_simplify(ctx, bmap);
1601 return isl_basic_map_finalize(ctx, bmap);
1603 isl_basic_map_free(ctx, bmap);
1604 isl_basic_set_free(ctx, bset);
1608 struct isl_basic_map *isl_basic_map_intersect_range(
1609 struct isl_ctx *ctx, struct isl_basic_map *bmap,
1610 struct isl_basic_set *bset)
1612 struct isl_basic_map *bmap_range;
1617 isl_assert(ctx, bset->dim == bmap->n_out, goto error);
1618 isl_assert(ctx, bset->nparam == bmap->nparam, goto error);
1620 bmap = isl_basic_map_extend(ctx, bmap,
1621 bset->nparam, bmap->n_in, bmap->n_out,
1622 bset->n_div, bset->n_eq, bset->n_ineq);
1625 bmap_range = isl_basic_map_from_basic_set(ctx, bset,
1627 bmap = add_constraints(ctx, bmap, bmap_range, 0, 0);
1629 bmap = isl_basic_map_simplify(ctx, bmap);
1630 return isl_basic_map_finalize(ctx, bmap);
1632 isl_basic_map_free(ctx, bmap);
1633 isl_basic_set_free(ctx, bset);
1637 struct isl_basic_map *isl_basic_map_intersect(
1638 struct isl_ctx *ctx, struct isl_basic_map *bmap1,
1639 struct isl_basic_map *bmap2)
1641 if (!bmap1 || !bmap2)
1644 isl_assert(ctx, bmap1->nparam == bmap2->nparam, goto error);
1645 isl_assert(ctx, bmap1->n_in == bmap2->n_in, goto error);
1646 isl_assert(ctx, bmap1->n_out == bmap2->n_out, goto error);
1648 bmap1 = isl_basic_map_extend(ctx, bmap1,
1649 bmap1->nparam, bmap1->n_in, bmap1->n_out,
1650 bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
1653 bmap1 = add_constraints(ctx, bmap1, bmap2, 0, 0);
1655 bmap1 = isl_basic_map_simplify(ctx, bmap1);
1656 return isl_basic_map_finalize(ctx, bmap1);
1658 isl_basic_map_free(ctx, bmap1);
1659 isl_basic_map_free(ctx, bmap2);
1663 struct isl_basic_set *isl_basic_set_intersect(
1664 struct isl_ctx *ctx, struct isl_basic_set *bset1,
1665 struct isl_basic_set *bset2)
1667 return (struct isl_basic_set *)
1668 isl_basic_map_intersect(ctx,
1669 (struct isl_basic_map *)bset1,
1670 (struct isl_basic_map *)bset2);
1673 struct isl_map *isl_map_intersect(struct isl_ctx *ctx, struct isl_map *map1,
1674 struct isl_map *map2)
1677 struct isl_map *result;
1683 if (F_ISSET(map1, ISL_MAP_DISJOINT) &&
1684 F_ISSET(map2, ISL_MAP_DISJOINT))
1685 FL_SET(flags, ISL_MAP_DISJOINT);
1687 result = isl_map_alloc(ctx, map1->nparam, map1->n_in, map1->n_out,
1688 map1->n * map2->n, flags);
1691 for (i = 0; i < map1->n; ++i)
1692 for (j = 0; j < map2->n; ++j) {
1693 struct isl_basic_map *part;
1694 part = isl_basic_map_intersect(ctx,
1695 isl_basic_map_copy(ctx, map1->p[i]),
1696 isl_basic_map_copy(ctx, map2->p[j]));
1697 if (isl_basic_map_is_empty(ctx, part))
1698 isl_basic_map_free(ctx, part);
1700 result = isl_map_add(ctx, result, part);
1704 isl_map_free(ctx, map1);
1705 isl_map_free(ctx, map2);
1708 isl_map_free(ctx, map1);
1709 isl_map_free(ctx, map2);
1713 struct isl_basic_map *isl_basic_map_reverse(struct isl_ctx *ctx,
1714 struct isl_basic_map *bmap)
1716 struct isl_basic_set *bset;
1721 bmap = isl_basic_map_cow(ctx, bmap);
1725 bset = isl_basic_set_from_basic_map(ctx, bmap);
1726 bset = isl_basic_set_swap_vars(ctx, bset, in);
1727 return isl_basic_map_from_basic_set(ctx, bset, bset->dim-in, in);
1730 /* Turn final n dimensions into existentially quantified variables.
1732 struct isl_basic_set *isl_basic_set_project_out(
1733 struct isl_ctx *ctx, struct isl_basic_set *bset,
1734 unsigned n, unsigned flags)
1744 isl_assert(ctx, n <= bset->dim, goto error);
1749 bset = isl_basic_set_cow(ctx, bset);
1751 row_size = 1 + bset->nparam + bset->dim + bset->extra;
1752 old = bset->block2.data;
1753 bset->block2 = isl_blk_extend(ctx, bset->block2,
1754 (bset->extra + n) * (1 + row_size));
1755 if (!bset->block2.data)
1757 new_div = isl_alloc_array(ctx, isl_int *, bset->extra + n);
1760 for (i = 0; i < n; ++i) {
1761 new_div[i] = bset->block2.data +
1762 (bset->extra + i) * (1 + row_size);
1763 isl_seq_clr(new_div[i], 1 + row_size);
1765 for (i = 0; i < bset->extra; ++i)
1766 new_div[n + i] = bset->block2.data + (bset->div[i] - old);
1768 bset->div = new_div;
1772 bset = isl_basic_set_simplify(ctx, bset);
1773 return isl_basic_set_finalize(ctx, bset);
1775 isl_basic_set_free(ctx, bset);
1779 struct isl_basic_map *isl_basic_map_apply_range(
1780 struct isl_ctx *ctx, struct isl_basic_map *bmap1,
1781 struct isl_basic_map *bmap2)
1783 struct isl_basic_set *bset;
1784 unsigned n_in, n_out;
1786 if (!bmap1 || !bmap2)
1789 isl_assert(ctx, bmap1->n_out == bmap2->n_in, goto error);
1790 isl_assert(ctx, bmap1->nparam == bmap2->nparam, goto error);
1793 n_out = bmap2->n_out;
1795 bmap2 = isl_basic_map_reverse(ctx, bmap2);
1798 bmap1 = isl_basic_map_extend(ctx, bmap1, bmap1->nparam,
1799 bmap1->n_in + bmap2->n_in, bmap1->n_out,
1800 bmap2->extra, bmap2->n_eq, bmap2->n_ineq);
1803 bmap1 = add_constraints(ctx, bmap1, bmap2, bmap1->n_in - bmap2->n_in, 0);
1804 bmap1 = isl_basic_map_simplify(ctx, bmap1);
1805 bset = isl_basic_set_from_basic_map(ctx, bmap1);
1806 bset = isl_basic_set_project_out(ctx, bset,
1807 bset->dim - (n_in + n_out), 0);
1808 return isl_basic_map_from_basic_set(ctx, bset, n_in, n_out);
1810 isl_basic_map_free(ctx, bmap1);
1811 isl_basic_map_free(ctx, bmap2);
1815 struct isl_basic_map *isl_basic_map_apply_domain(
1816 struct isl_ctx *ctx, struct isl_basic_map *bmap1,
1817 struct isl_basic_map *bmap2)
1819 if (!bmap1 || !bmap2)
1822 isl_assert(ctx, bmap1->n_in == bmap2->n_in, goto error);
1823 isl_assert(ctx, bmap1->nparam == bmap2->nparam, goto error);
1825 bmap1 = isl_basic_map_reverse(ctx, bmap1);
1826 bmap1 = isl_basic_map_apply_range(ctx, bmap1, bmap2);
1827 return isl_basic_map_reverse(ctx, bmap1);
1829 isl_basic_map_free(ctx, bmap1);
1830 isl_basic_map_free(ctx, bmap2);
1834 static struct isl_basic_map *var_equal(struct isl_ctx *ctx,
1835 struct isl_basic_map *bmap, unsigned pos)
1838 i = isl_basic_map_alloc_equality(ctx, bmap);
1841 isl_seq_clr(bmap->eq[i],
1842 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra);
1843 isl_int_set_si(bmap->eq[i][1+bmap->nparam+pos], -1);
1844 isl_int_set_si(bmap->eq[i][1+bmap->nparam+bmap->n_in+pos], 1);
1845 return isl_basic_map_finalize(ctx, bmap);
1847 isl_basic_map_free(ctx, bmap);
1851 static struct isl_basic_map *var_more(struct isl_ctx *ctx,
1852 struct isl_basic_map *bmap, unsigned pos)
1855 i = isl_basic_map_alloc_inequality(ctx, bmap);
1858 isl_seq_clr(bmap->ineq[i],
1859 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra);
1860 isl_int_set_si(bmap->ineq[i][0], -1);
1861 isl_int_set_si(bmap->ineq[i][1+bmap->nparam+pos], -1);
1862 isl_int_set_si(bmap->ineq[i][1+bmap->nparam+bmap->n_in+pos], 1);
1863 return isl_basic_map_finalize(ctx, bmap);
1865 isl_basic_map_free(ctx, bmap);
1869 static struct isl_basic_map *var_less(struct isl_ctx *ctx,
1870 struct isl_basic_map *bmap, unsigned pos)
1873 i = isl_basic_map_alloc_inequality(ctx, bmap);
1876 isl_seq_clr(bmap->ineq[i],
1877 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra);
1878 isl_int_set_si(bmap->ineq[i][0], -1);
1879 isl_int_set_si(bmap->ineq[i][1+bmap->nparam+pos], 1);
1880 isl_int_set_si(bmap->ineq[i][1+bmap->nparam+bmap->n_in+pos], -1);
1881 return isl_basic_map_finalize(ctx, bmap);
1883 isl_basic_map_free(ctx, bmap);
1887 struct isl_basic_map *isl_basic_map_equal(struct isl_ctx *ctx,
1888 unsigned nparam, unsigned in, unsigned out, unsigned n_equal)
1891 struct isl_basic_map *bmap;
1892 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, n_equal, 0);
1895 for (i = 0; i < n_equal && bmap; ++i)
1896 bmap = var_equal(ctx, bmap, i);
1897 return isl_basic_map_finalize(ctx, bmap);
1900 struct isl_basic_map *isl_basic_map_less_at(struct isl_ctx *ctx,
1901 unsigned nparam, unsigned in, unsigned out, unsigned pos)
1904 struct isl_basic_map *bmap;
1905 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, pos, 1);
1908 for (i = 0; i < pos && bmap; ++i)
1909 bmap = var_equal(ctx, bmap, i);
1911 bmap = var_less(ctx, bmap, pos);
1912 return isl_basic_map_finalize(ctx, bmap);
1915 struct isl_basic_map *isl_basic_map_more_at(struct isl_ctx *ctx,
1916 unsigned nparam, unsigned in, unsigned out, unsigned pos)
1919 struct isl_basic_map *bmap;
1920 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, pos, 1);
1923 for (i = 0; i < pos && bmap; ++i)
1924 bmap = var_equal(ctx, bmap, i);
1926 bmap = var_more(ctx, bmap, pos);
1927 return isl_basic_map_finalize(ctx, bmap);
1930 struct isl_basic_map *isl_basic_map_from_basic_set(
1931 struct isl_ctx *ctx, struct isl_basic_set *bset,
1932 unsigned n_in, unsigned n_out)
1934 struct isl_basic_map *bmap;
1936 bset = isl_basic_set_cow(ctx, bset);
1940 isl_assert(ctx, bset->dim == n_in + n_out, goto error);
1941 bmap = (struct isl_basic_map *) bset;
1943 bmap->n_out = n_out;
1944 return isl_basic_map_finalize(ctx, bmap);
1946 isl_basic_set_free(ctx, bset);
1950 struct isl_basic_set *isl_basic_set_from_basic_map(
1951 struct isl_ctx *ctx, struct isl_basic_map *bmap)
1955 if (bmap->n_in == 0)
1956 return (struct isl_basic_set *)bmap;
1957 bmap = isl_basic_map_cow(ctx, bmap);
1960 bmap->n_out += bmap->n_in;
1962 bmap = isl_basic_map_finalize(ctx, bmap);
1963 return (struct isl_basic_set *)bmap;
1968 /* For a div d = floor(f/m), add the constraints
1971 * -(f-(n-1)) + m d >= 0
1973 * Note that the second constraint is the negation of
1977 static int add_div_constraints(struct isl_ctx *ctx,
1978 struct isl_basic_map *bmap, unsigned div)
1981 unsigned div_pos = 1 + bmap->nparam + bmap->n_in + bmap->n_out + div;
1982 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra;
1984 i = isl_basic_map_alloc_inequality(ctx, bmap);
1987 isl_seq_cpy(bmap->ineq[i], bmap->div[div]+1, 1+total);
1988 isl_int_neg(bmap->ineq[i][div_pos], bmap->div[div][0]);
1990 j = isl_basic_map_alloc_inequality(ctx, bmap);
1993 isl_seq_neg(bmap->ineq[j], bmap->ineq[i], 1 + total);
1994 isl_int_add(bmap->ineq[j][0], bmap->ineq[j][0], bmap->ineq[j][div_pos]);
1995 isl_int_sub_ui(bmap->ineq[j][0], bmap->ineq[j][0], 1);
1999 static struct isl_basic_set *isl_basic_map_underlying_set(
2000 struct isl_ctx *ctx, struct isl_basic_map *bmap)
2004 if (bmap->nparam == 0 && bmap->n_in == 0 && bmap->n_div == 0)
2005 return (struct isl_basic_set *)bmap;
2006 bmap = isl_basic_map_cow(ctx, bmap);
2009 bmap->n_out += bmap->nparam + bmap->n_in + bmap->n_div;
2012 bmap->extra -= bmap->n_div;
2014 bmap = isl_basic_map_finalize(ctx, bmap);
2015 return (struct isl_basic_set *)bmap;
2020 struct isl_basic_map *isl_basic_map_overlying_set(
2021 struct isl_ctx *ctx, struct isl_basic_set *bset,
2022 struct isl_basic_map *like)
2024 struct isl_basic_map *bmap;
2030 isl_assert(ctx, bset->dim ==
2031 like->nparam + like->n_in + like->n_out + like->n_div,
2033 if (like->nparam == 0 && like->n_in == 0 && like->n_div == 0) {
2034 isl_basic_map_free(ctx, like);
2035 return (struct isl_basic_map *)bset;
2037 bset = isl_basic_set_cow(ctx, bset);
2038 total = bset->dim + bset->extra;
2041 bmap = (struct isl_basic_map *)bset;
2042 bmap->nparam = like->nparam;
2043 bmap->n_in = like->n_in;
2044 bmap->n_out = like->n_out;
2045 bmap->extra += like->n_div;
2048 ltotal = total - bmap->extra + like->extra;
2051 bmap->block2 = isl_blk_extend(ctx, bmap->block2,
2052 bmap->extra * (1 + 1 + total));
2053 if (isl_blk_is_error(bmap->block2))
2055 bmap->div = isl_realloc_array(ctx, bmap->div, isl_int *,
2059 bmap = isl_basic_map_extend(ctx, bmap, bmap->nparam,
2060 bmap->n_in, bmap->n_out, 0, 0, 2 * like->n_div);
2061 for (i = 0; i < like->n_div; ++i) {
2062 k = isl_basic_map_alloc_div(ctx, bmap);
2065 isl_seq_cpy(bmap->div[k], like->div[i], 1 + 1 + ltotal);
2066 isl_seq_clr(bmap->div[k]+1+1+ltotal, total - ltotal);
2067 if (add_div_constraints(ctx, bmap, k) < 0)
2071 isl_basic_map_free(ctx, like);
2072 bmap = isl_basic_map_finalize(ctx, bmap);
2075 isl_basic_map_free(ctx, like);
2076 isl_basic_set_free(ctx, bset);
2080 struct isl_set *isl_map_underlying_set(struct isl_ctx *ctx, struct isl_map *map)
2084 map = isl_map_align_divs(ctx, map);
2085 map = isl_map_cow(ctx, map);
2089 for (i = 0; i < map->n; ++i) {
2090 map->p[i] = (struct isl_basic_map *)
2091 isl_basic_map_underlying_set(ctx, map->p[i]);
2096 map->n_out += map->nparam + map->n_in;
2098 map->n_out = map->p[0]->n_out;
2101 return (struct isl_set *)map;
2103 isl_map_free(ctx, map);
2107 struct isl_basic_set *isl_basic_map_domain(struct isl_ctx *ctx,
2108 struct isl_basic_map *bmap)
2110 struct isl_basic_set *domain;
2114 n_out = bmap->n_out;
2115 domain = isl_basic_set_from_basic_map(ctx, bmap);
2116 return isl_basic_set_project_out(ctx, domain, n_out, 0);
2119 struct isl_basic_set *isl_basic_map_range(struct isl_ctx *ctx,
2120 struct isl_basic_map *bmap)
2122 return isl_basic_map_domain(ctx,
2123 isl_basic_map_reverse(ctx, bmap));
2126 struct isl_set *isl_map_range(struct isl_ctx *ctx, struct isl_map *map)
2129 struct isl_set *set;
2133 map = isl_map_cow(ctx, map);
2137 set = (struct isl_set *) map;
2139 for (i = 0; i < map->n; ++i) {
2140 set->p[i] = isl_basic_map_range(ctx, map->p[i]);
2144 F_CLR(set, ISL_MAP_DISJOINT);
2147 isl_map_free(ctx, map);
2151 struct isl_map *isl_map_from_set(
2152 struct isl_ctx *ctx, struct isl_set *set,
2153 unsigned n_in, unsigned n_out)
2156 struct isl_map *map = NULL;
2160 isl_assert(ctx, set->dim == n_in + n_out, goto error);
2161 set = isl_set_cow(ctx, set);
2164 map = (struct isl_map *)set;
2165 for (i = 0; i < set->n; ++i) {
2166 map->p[i] = isl_basic_map_from_basic_set(ctx,
2167 set->p[i], n_in, n_out);
2175 isl_set_free(ctx, set);
2179 struct isl_map *isl_map_alloc(struct isl_ctx *ctx,
2180 unsigned nparam, unsigned in, unsigned out, int n,
2183 struct isl_map *map;
2185 map = isl_alloc(ctx, struct isl_map,
2186 sizeof(struct isl_map) +
2187 n * sizeof(struct isl_basic_map *));
2194 map->nparam = nparam;
2201 struct isl_basic_map *isl_basic_map_empty(struct isl_ctx *ctx,
2202 unsigned nparam, unsigned in, unsigned out)
2204 struct isl_basic_map *bmap;
2205 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, 1, 0);
2206 bmap = isl_basic_map_set_to_empty(ctx, bmap);
2210 struct isl_basic_map *isl_basic_map_universe(struct isl_ctx *ctx,
2211 unsigned nparam, unsigned in, unsigned out)
2213 struct isl_basic_map *bmap;
2214 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, 0, 0);
2218 struct isl_basic_set *isl_basic_set_universe(struct isl_ctx *ctx,
2219 unsigned nparam, unsigned dim)
2221 struct isl_basic_set *bset;
2222 bset = isl_basic_set_alloc(ctx, nparam, dim, 0, 0, 0);
2226 struct isl_map *isl_map_empty(struct isl_ctx *ctx,
2227 unsigned nparam, unsigned in, unsigned out)
2229 return isl_map_alloc(ctx, nparam, in, out, 0, ISL_MAP_DISJOINT);
2232 struct isl_set *isl_set_empty(struct isl_ctx *ctx,
2233 unsigned nparam, unsigned dim)
2235 return isl_set_alloc(ctx, nparam, dim, 0, ISL_MAP_DISJOINT);
2238 struct isl_map *isl_map_dup(struct isl_ctx *ctx, struct isl_map *map)
2241 struct isl_map *dup;
2243 dup = isl_map_alloc(ctx, map->nparam, map->n_in, map->n_out, map->n,
2245 for (i = 0; i < map->n; ++i)
2246 dup = isl_map_add(ctx, dup,
2247 isl_basic_map_dup(ctx, map->p[i]));
2251 struct isl_map *isl_map_add(struct isl_ctx *ctx, struct isl_map *map,
2252 struct isl_basic_map *bmap)
2256 isl_assert(ctx, map->nparam == bmap->nparam, goto error);
2257 isl_assert(ctx, map->n_in == bmap->n_in, goto error);
2258 isl_assert(ctx, map->n_out == bmap->n_out, goto error);
2259 isl_assert(ctx, map->n < map->size, goto error);
2260 map->p[map->n] = bmap;
2265 isl_map_free(ctx, map);
2267 isl_basic_map_free(ctx, bmap);
2271 void isl_map_free(struct isl_ctx *ctx, struct isl_map *map)
2281 for (i = 0; i < map->n; ++i)
2282 isl_basic_map_free(ctx, map->p[i]);
2286 struct isl_map *isl_map_extend(struct isl_ctx *ctx, struct isl_map *base,
2287 unsigned nparam, unsigned n_in, unsigned n_out)
2291 base = isl_map_cow(ctx, base);
2295 isl_assert(ctx, base->nparam <= nparam, goto error);
2296 isl_assert(ctx, base->n_in <= n_in, goto error);
2297 isl_assert(ctx, base->n_out <= n_out, goto error);
2298 base->nparam = nparam;
2300 base->n_out = n_out;
2301 for (i = 0; i < base->n; ++i) {
2302 base->p[i] = isl_basic_map_extend(ctx, base->p[i],
2303 nparam, n_in, n_out, 0, 0, 0);
2309 isl_map_free(ctx, base);
2313 struct isl_basic_map *isl_basic_map_fix_input_si(struct isl_ctx *ctx,
2314 struct isl_basic_map *bmap,
2315 unsigned input, int value)
2319 bmap = isl_basic_map_cow(ctx, bmap);
2322 isl_assert(ctx, input < bmap->n_in, goto error);
2324 bmap = isl_basic_map_extend(ctx, bmap,
2325 bmap->nparam, bmap->n_in, bmap->n_out, 0, 1, 0);
2326 j = isl_basic_map_alloc_equality(ctx, bmap);
2329 isl_seq_clr(bmap->eq[j],
2330 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra);
2331 isl_int_set_si(bmap->eq[j][1+bmap->nparam+input], -1);
2332 isl_int_set_si(bmap->eq[j][0], value);
2333 bmap = isl_basic_map_simplify(ctx, bmap);
2334 return isl_basic_map_finalize(ctx, bmap);
2336 isl_basic_map_free(ctx, bmap);
2340 struct isl_map *isl_map_fix_input_si(struct isl_ctx *ctx, struct isl_map *map,
2341 unsigned input, int value)
2345 map = isl_map_cow(ctx, map);
2349 isl_assert(ctx, input < map->n_in, goto error);
2350 for (i = 0; i < map->n; ++i) {
2351 map->p[i] = isl_basic_map_fix_input_si(ctx, map->p[i],
2358 isl_map_free(ctx, map);
2362 struct isl_map *isl_map_reverse(struct isl_ctx *ctx, struct isl_map *map)
2367 map = isl_map_cow(ctx, map);
2372 map->n_in = map->n_out;
2374 for (i = 0; i < map->n; ++i) {
2375 map->p[i] = isl_basic_map_reverse(ctx, map->p[i]);
2381 isl_map_free(ctx, map);
2385 struct isl_map *isl_basic_map_lexmax(struct isl_ctx *ctx,
2386 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2387 struct isl_set **empty)
2389 return isl_pip_basic_map_lexmax(ctx, bmap, dom, empty);
2392 struct isl_map *isl_basic_map_lexmin(struct isl_ctx *ctx,
2393 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2394 struct isl_set **empty)
2396 return isl_pip_basic_map_lexmin(ctx, bmap, dom, empty);
2399 struct isl_set *isl_basic_set_lexmin(struct isl_ctx *ctx,
2400 struct isl_basic_set *bset)
2402 struct isl_basic_map *bmap = NULL;
2403 struct isl_basic_set *dom = NULL;
2404 struct isl_map *min;
2408 bmap = isl_basic_map_from_basic_set(ctx, bset, 0, bset->dim);
2411 dom = isl_basic_set_alloc(ctx, bmap->nparam, 0, 0, 0, 0);
2414 min = isl_basic_map_lexmin(ctx, bmap, dom, NULL);
2415 return isl_map_range(ctx, min);
2417 isl_basic_map_free(ctx, bmap);
2421 struct isl_map *isl_basic_map_compute_divs(struct isl_ctx *ctx,
2422 struct isl_basic_map *bmap)
2426 if (bmap->n_div == 0)
2427 return isl_map_from_basic_map(ctx, bmap);
2428 return isl_pip_basic_map_compute_divs(ctx, bmap);
2431 struct isl_map *isl_map_compute_divs(struct isl_ctx *ctx, struct isl_map *map)
2434 struct isl_map *res;
2440 res = isl_basic_map_compute_divs(ctx,
2441 isl_basic_map_copy(ctx, map->p[0]));
2442 for (i = 1 ; i < map->n; ++i) {
2444 r2 = isl_basic_map_compute_divs(ctx,
2445 isl_basic_map_copy(ctx, map->p[i]));
2446 if (F_ISSET(map, ISL_MAP_DISJOINT))
2447 res = isl_map_union_disjoint(ctx, res, r2);
2449 res = isl_map_union(ctx, res, r2);
2451 isl_map_free(ctx, map);
2456 struct isl_set *isl_basic_set_compute_divs(struct isl_ctx *ctx,
2457 struct isl_basic_set *bset)
2459 return (struct isl_set *)
2460 isl_basic_map_compute_divs(ctx,
2461 (struct isl_basic_map *)bset);
2464 struct isl_set *isl_map_domain(struct isl_ctx *ctx, struct isl_map *map)
2467 struct isl_set *set;
2472 map = isl_map_cow(ctx, map);
2476 set = (struct isl_set *)map;
2477 set->dim = map->n_in;
2479 for (i = 0; i < map->n; ++i) {
2480 set->p[i] = isl_basic_map_domain(ctx, map->p[i]);
2484 F_CLR(set, ISL_MAP_DISJOINT);
2487 isl_map_free(ctx, map);
2491 struct isl_map *isl_map_union_disjoint(struct isl_ctx *ctx,
2492 struct isl_map *map1, struct isl_map *map2)
2496 struct isl_map *map = NULL;
2502 isl_map_free(ctx, map1);
2506 isl_map_free(ctx, map2);
2510 isl_assert(ctx, map1->nparam == map2->nparam, goto error);
2511 isl_assert(ctx, map1->n_in == map2->n_in, goto error);
2512 isl_assert(ctx, map1->n_out == map2->n_out, goto error);
2514 if (F_ISSET(map1, ISL_MAP_DISJOINT) &&
2515 F_ISSET(map2, ISL_MAP_DISJOINT))
2516 FL_SET(flags, ISL_MAP_DISJOINT);
2518 map = isl_map_alloc(ctx, map1->nparam, map1->n_in, map1->n_out,
2519 map1->n + map2->n, flags);
2522 for (i = 0; i < map1->n; ++i) {
2523 map = isl_map_add(ctx, map,
2524 isl_basic_map_copy(ctx, map1->p[i]));
2528 for (i = 0; i < map2->n; ++i) {
2529 map = isl_map_add(ctx, map,
2530 isl_basic_map_copy(ctx, map2->p[i]));
2534 isl_map_free(ctx, map1);
2535 isl_map_free(ctx, map2);
2538 isl_map_free(ctx, map);
2539 isl_map_free(ctx, map1);
2540 isl_map_free(ctx, map2);
2544 struct isl_map *isl_map_union(struct isl_ctx *ctx,
2545 struct isl_map *map1, struct isl_map *map2)
2547 map1 = isl_map_union_disjoint(ctx, map1, map2);
2551 F_CLR(map1, ISL_MAP_DISJOINT);
2555 struct isl_set *isl_set_union_disjoint(struct isl_ctx *ctx,
2556 struct isl_set *set1, struct isl_set *set2)
2558 return (struct isl_set *)
2559 isl_map_union_disjoint(ctx,
2560 (struct isl_map *)set1, (struct isl_map *)set2);
2563 struct isl_set *isl_set_union(struct isl_ctx *ctx,
2564 struct isl_set *set1, struct isl_set *set2)
2566 return (struct isl_set *)
2568 (struct isl_map *)set1, (struct isl_map *)set2);
2571 struct isl_map *isl_map_intersect_range(
2572 struct isl_ctx *ctx, struct isl_map *map,
2573 struct isl_set *set)
2576 struct isl_map *result;
2582 if (F_ISSET(map, ISL_MAP_DISJOINT) &&
2583 F_ISSET(set, ISL_MAP_DISJOINT))
2584 FL_SET(flags, ISL_MAP_DISJOINT);
2586 result = isl_map_alloc(ctx, map->nparam, map->n_in, map->n_out,
2587 map->n * set->n, flags);
2590 for (i = 0; i < map->n; ++i)
2591 for (j = 0; j < set->n; ++j) {
2592 result = isl_map_add(ctx, result,
2593 isl_basic_map_intersect_range(ctx,
2594 isl_basic_map_copy(ctx, map->p[i]),
2595 isl_basic_set_copy(ctx, set->p[j])));
2599 isl_map_free(ctx, map);
2600 isl_set_free(ctx, set);
2603 isl_map_free(ctx, map);
2604 isl_set_free(ctx, set);
2608 struct isl_map *isl_map_intersect_domain(
2609 struct isl_ctx *ctx, struct isl_map *map,
2610 struct isl_set *set)
2612 return isl_map_reverse(ctx,
2613 isl_map_intersect_range(ctx, isl_map_reverse(ctx, map), set));
2616 struct isl_map *isl_map_apply_domain(
2617 struct isl_ctx *ctx, struct isl_map *map1,
2618 struct isl_map *map2)
2622 map1 = isl_map_reverse(ctx, map1);
2623 map1 = isl_map_apply_range(ctx, map1, map2);
2624 return isl_map_reverse(ctx, map1);
2626 isl_map_free(ctx, map1);
2627 isl_map_free(ctx, map2);
2631 struct isl_map *isl_map_apply_range(
2632 struct isl_ctx *ctx, struct isl_map *map1,
2633 struct isl_map *map2)
2635 struct isl_map *result;
2641 isl_assert(ctx, map1->nparam == map2->nparam, goto error);
2642 isl_assert(ctx, map1->n_out == map2->n_in, goto error);
2644 result = isl_map_alloc(ctx, map1->nparam, map1->n_in, map2->n_out,
2645 map1->n * map2->n, 0);
2648 for (i = 0; i < map1->n; ++i)
2649 for (j = 0; j < map2->n; ++j) {
2650 result = isl_map_add(ctx, result,
2651 isl_basic_map_apply_range(ctx,
2652 isl_basic_map_copy(ctx, map1->p[i]),
2653 isl_basic_map_copy(ctx, map2->p[j])));
2657 isl_map_free(ctx, map1);
2658 isl_map_free(ctx, map2);
2661 isl_map_free(ctx, map1);
2662 isl_map_free(ctx, map2);
2667 * returns range - domain
2669 struct isl_basic_set *isl_basic_map_deltas(struct isl_ctx *ctx,
2670 struct isl_basic_map *bmap)
2672 struct isl_basic_set *bset;
2678 isl_assert(ctx, bmap->n_in == bmap->n_out, goto error);
2680 bset = isl_basic_set_from_basic_map(ctx, bmap);
2681 bset = isl_basic_set_extend(ctx, bset, bset->nparam, 3*dim, 0,
2683 bset = isl_basic_set_swap_vars(ctx, bset, 2*dim);
2684 for (i = 0; i < dim; ++i) {
2685 int j = isl_basic_map_alloc_equality(ctx,
2686 (struct isl_basic_map *)bset);
2689 isl_seq_clr(bset->eq[j],
2690 1 + bset->nparam + bset->dim + bset->extra);
2691 isl_int_set_si(bset->eq[j][1+bset->nparam+i], 1);
2692 isl_int_set_si(bset->eq[j][1+bset->nparam+dim+i], 1);
2693 isl_int_set_si(bset->eq[j][1+bset->nparam+2*dim+i], -1);
2695 return isl_basic_set_project_out(ctx, bset, 2*dim, 0);
2697 isl_basic_map_free(ctx, bmap);
2701 /* If the only constraints a div d=floor(f/m)
2702 * appears in are its two defining constraints
2705 * -(f - (m - 1)) + m d >= 0
2707 * then it can safely be removed.
2709 static int div_is_redundant(struct isl_ctx *ctx, struct isl_basic_map *bmap,
2713 unsigned pos = 1 + bmap->nparam + bmap->n_in + bmap->n_out + div;
2715 for (i = 0; i < bmap->n_eq; ++i)
2716 if (!isl_int_is_zero(bmap->eq[i][pos]))
2719 for (i = 0; i < bmap->n_ineq; ++i) {
2720 if (isl_int_is_zero(bmap->ineq[i][pos]))
2722 if (isl_int_eq(bmap->ineq[i][pos], bmap->div[div][0])) {
2724 isl_int_sub(bmap->div[div][1],
2725 bmap->div[div][1], bmap->div[div][0]);
2726 isl_int_add_ui(bmap->div[div][1], bmap->div[div][1], 1);
2727 neg = isl_seq_is_neg(bmap->ineq[i], bmap->div[div]+1, pos);
2728 isl_int_sub_ui(bmap->div[div][1], bmap->div[div][1], 1);
2729 isl_int_add(bmap->div[div][1],
2730 bmap->div[div][1], bmap->div[div][0]);
2733 if (isl_seq_first_non_zero(bmap->ineq[i]+pos+1,
2734 bmap->n_div-div-1) != -1)
2736 } else if (isl_int_abs_eq(bmap->ineq[i][pos], bmap->div[div][0])) {
2737 if (!isl_seq_eq(bmap->ineq[i], bmap->div[div]+1, pos))
2739 if (isl_seq_first_non_zero(bmap->ineq[i]+pos+1,
2740 bmap->n_div-div-1) != -1)
2746 for (i = 0; i < bmap->n_div; ++i)
2747 if (!isl_int_is_zero(bmap->div[i][1+pos]))
2754 * Remove divs that don't occur in any of the constraints or other divs.
2755 * These can arise when dropping some of the variables in a quast
2756 * returned by piplib.
2758 static struct isl_basic_map *remove_redundant_divs(struct isl_ctx *ctx,
2759 struct isl_basic_map *bmap)
2763 for (i = bmap->n_div-1; i >= 0; --i) {
2764 if (!div_is_redundant(ctx, bmap, i))
2766 bmap = isl_basic_map_drop_div(ctx, bmap, i);
2771 struct isl_basic_map *isl_basic_map_finalize(struct isl_ctx *ctx,
2772 struct isl_basic_map *bmap)
2774 bmap = remove_redundant_divs(ctx, bmap);
2777 F_SET(bmap, ISL_BASIC_SET_FINAL);
2781 struct isl_basic_set *isl_basic_set_finalize(struct isl_ctx *ctx,
2782 struct isl_basic_set *bset)
2784 return (struct isl_basic_set *)
2785 isl_basic_map_finalize(ctx,
2786 (struct isl_basic_map *)bset);
2789 struct isl_set *isl_set_finalize(struct isl_ctx *ctx, struct isl_set *set)
2795 for (i = 0; i < set->n; ++i) {
2796 set->p[i] = isl_basic_set_finalize(ctx, set->p[i]);
2802 isl_set_free(ctx, set);
2806 struct isl_map *isl_map_finalize(struct isl_ctx *ctx, struct isl_map *map)
2812 for (i = 0; i < map->n; ++i) {
2813 map->p[i] = isl_basic_map_finalize(ctx, map->p[i]);
2819 isl_map_free(ctx, map);
2823 struct isl_basic_map *isl_basic_map_identity(struct isl_ctx *ctx,
2824 unsigned nparam, unsigned dim)
2826 struct isl_basic_map *bmap;
2829 bmap = isl_basic_map_alloc(ctx, nparam, dim, dim, 0, dim, 0);
2833 for (i = 0; i < dim; ++i) {
2834 int j = isl_basic_map_alloc_equality(ctx, bmap);
2837 isl_seq_clr(bmap->eq[j],
2838 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->extra);
2839 isl_int_set_si(bmap->eq[j][1+nparam+i], 1);
2840 isl_int_set_si(bmap->eq[j][1+nparam+dim+i], -1);
2842 return isl_basic_map_finalize(ctx, bmap);
2844 isl_basic_map_free(ctx, bmap);
2848 struct isl_map *isl_map_identity(struct isl_ctx *ctx,
2849 unsigned nparam, unsigned dim)
2851 struct isl_map *map = isl_map_alloc(ctx, nparam, dim, dim, 1,
2855 map = isl_map_add(ctx, map,
2856 isl_basic_map_identity(ctx, nparam, dim));
2859 isl_map_free(ctx, map);
2863 int isl_set_is_equal(struct isl_ctx *ctx,
2864 struct isl_set *set1, struct isl_set *set2)
2866 return isl_map_is_equal(ctx,
2867 (struct isl_map *)set1, (struct isl_map *)set2);
2870 int isl_set_is_subset(struct isl_ctx *ctx,
2871 struct isl_set *set1, struct isl_set *set2)
2873 return isl_map_is_subset(ctx,
2874 (struct isl_map *)set1, (struct isl_map *)set2);
2877 int isl_basic_map_is_subset(struct isl_ctx *ctx,
2878 struct isl_basic_map *bmap1,
2879 struct isl_basic_map *bmap2)
2882 struct isl_map *map1;
2883 struct isl_map *map2;
2885 if (!bmap1 || !bmap2)
2888 map1 = isl_map_from_basic_map(ctx,
2889 isl_basic_map_copy(ctx, bmap1));
2890 map2 = isl_map_from_basic_map(ctx,
2891 isl_basic_map_copy(ctx, bmap2));
2893 is_subset = isl_map_is_subset(ctx, map1, map2);
2895 isl_map_free(ctx, map1);
2896 isl_map_free(ctx, map2);
2901 int isl_basic_map_is_equal(struct isl_ctx *ctx,
2902 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
2906 if (!bmap1 || !bmap2)
2908 is_subset = isl_basic_map_is_subset(ctx, bmap1, bmap2);
2911 is_subset = isl_basic_map_is_subset(ctx, bmap2, bmap1);
2915 int isl_basic_set_is_equal(struct isl_ctx *ctx,
2916 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
2918 return isl_basic_map_is_equal(ctx,
2919 (struct isl_basic_map *)bset1, (struct isl_basic_map *)bset2);
2922 int isl_map_is_empty(struct isl_ctx *ctx, struct isl_map *map)
2929 for (i = 0; i < map->n; ++i) {
2930 is_empty = isl_basic_map_is_empty(ctx, map->p[i]);
2939 int isl_set_is_empty(struct isl_ctx *ctx, struct isl_set *set)
2941 return isl_map_is_empty(ctx, (struct isl_map *)set);
2944 int isl_map_is_subset(struct isl_ctx *ctx, struct isl_map *map1,
2945 struct isl_map *map2)
2949 struct isl_map *diff;
2954 if (isl_map_is_empty(ctx, map1))
2957 if (isl_map_is_empty(ctx, map2))
2960 diff = isl_map_subtract(ctx, isl_map_copy(ctx, map1),
2961 isl_map_copy(ctx, map2));
2965 is_subset = isl_map_is_empty(ctx, diff);
2966 isl_map_free(ctx, diff);
2971 int isl_map_is_equal(struct isl_ctx *ctx,
2972 struct isl_map *map1, struct isl_map *map2)
2978 is_subset = isl_map_is_subset(ctx, map1, map2);
2981 is_subset = isl_map_is_subset(ctx, map2, map1);
2985 int isl_basic_map_is_strict_subset(struct isl_ctx *ctx,
2986 struct isl_basic_map *bmap1,
2987 struct isl_basic_map *bmap2)
2991 if (!bmap1 || !bmap2)
2993 is_subset = isl_basic_map_is_subset(ctx, bmap1, bmap2);
2996 is_subset = isl_basic_map_is_subset(ctx, bmap2, bmap1);
2997 if (is_subset == -1)
3002 static int basic_map_contains(struct isl_ctx *ctx, struct isl_basic_map *bmap,
3003 struct isl_vec *vec)
3009 total = 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
3010 if (total != vec->size)
3015 for (i = 0; i < bmap->n_eq; ++i) {
3016 isl_seq_inner_product(vec->block.data, bmap->eq[i], total, &s);
3017 if (!isl_int_is_zero(s)) {
3023 for (i = 0; i < bmap->n_ineq; ++i) {
3024 isl_seq_inner_product(vec->block.data, bmap->ineq[i], total, &s);
3025 if (isl_int_is_neg(s)) {
3036 int isl_basic_map_is_universe(struct isl_ctx *ctx,
3037 struct isl_basic_map *bmap)
3041 return bmap->n_eq == 0 && bmap->n_ineq == 0;
3044 int isl_basic_map_is_empty(struct isl_ctx *ctx,
3045 struct isl_basic_map *bmap)
3047 struct isl_basic_set *bset = NULL;
3048 struct isl_vec *sample = NULL;
3055 if (F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
3058 total = 1 + bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
3059 if (bmap->sample && bmap->sample->size == total) {
3060 int contains = basic_map_contains(ctx, bmap, bmap->sample);
3066 bset = isl_basic_map_underlying_set(ctx,
3067 isl_basic_map_copy(ctx, bmap));
3070 sample = isl_basic_set_sample(ctx, bset);
3073 empty = sample->size == 0;
3075 isl_vec_free(ctx, bmap->sample);
3076 bmap->sample = sample;
3081 struct isl_map *isl_basic_map_union(
3082 struct isl_ctx *ctx, struct isl_basic_map *bmap1,
3083 struct isl_basic_map *bmap2)
3085 struct isl_map *map;
3086 if (!bmap1 || !bmap2)
3089 isl_assert(ctx, bmap1->nparam == bmap2->nparam, goto error);
3090 isl_assert(ctx, bmap1->n_in == bmap2->n_in, goto error);
3091 isl_assert(ctx, bmap1->n_out == bmap2->n_out, goto error);
3093 map = isl_map_alloc(ctx, bmap1->nparam,
3094 bmap1->n_in, bmap1->n_out, 2, 0);
3097 map = isl_map_add(ctx, map, bmap1);
3098 map = isl_map_add(ctx, map, bmap2);
3101 isl_basic_map_free(ctx, bmap1);
3102 isl_basic_map_free(ctx, bmap2);
3106 struct isl_set *isl_basic_set_union(
3107 struct isl_ctx *ctx, struct isl_basic_set *bset1,
3108 struct isl_basic_set *bset2)
3110 return (struct isl_set *)isl_basic_map_union(ctx,
3111 (struct isl_basic_map *)bset1,
3112 (struct isl_basic_map *)bset2);
3115 /* Order divs such that any div only depends on previous divs */
3116 static struct isl_basic_map *order_divs(struct isl_ctx *ctx,
3117 struct isl_basic_map *bmap)
3120 unsigned off = bmap->nparam + bmap->n_in + bmap->n_out;
3122 for (i = 0; i < bmap->n_div; ++i) {
3124 pos = isl_seq_first_non_zero(bmap->div[i]+1+1+off+i,
3128 swap_div(bmap, i, pos);
3134 static int find_div(struct isl_basic_map *dst,
3135 struct isl_basic_map *src, unsigned div)
3139 unsigned total = src->nparam + src->n_in + src->n_out + src->n_div;
3141 for (i = 0; i < dst->n_div; ++i)
3142 if (isl_seq_eq(dst->div[i], src->div[div], 1+1+total) &&
3143 isl_seq_first_non_zero(dst->div[i]+1+1+total,
3144 dst->n_div - src->n_div) == -1)
3149 struct isl_basic_map *isl_basic_map_align_divs(struct isl_ctx *ctx,
3150 struct isl_basic_map *dst, struct isl_basic_map *src)
3153 unsigned total = src->nparam + src->n_in + src->n_out + src->n_div;
3155 if (src->n_div == 0)
3158 src = order_divs(ctx, src);
3159 dst = isl_basic_map_extend(ctx, dst, dst->nparam, dst->n_in,
3160 dst->n_out, src->n_div, 0, 2 * src->n_div);
3163 for (i = 0; i < src->n_div; ++i) {
3164 int j = find_div(dst, src, i);
3166 j = isl_basic_map_alloc_div(ctx, dst);
3169 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total);
3170 isl_seq_clr(dst->div[j]+1+1+total,
3171 dst->extra - src->n_div);
3172 if (add_div_constraints(ctx, dst, j) < 0)
3176 swap_div(dst, i, j);
3180 isl_basic_map_free(ctx, dst);
3184 struct isl_map *isl_map_align_divs(struct isl_ctx *ctx, struct isl_map *map)
3188 map = isl_map_compute_divs(ctx, map);
3189 map = isl_map_cow(ctx, map);
3193 for (i = 1; i < map->n; ++i)
3194 map->p[0] = isl_basic_map_align_divs(ctx, map->p[0], map->p[i]);
3195 for (i = 1; i < map->n; ++i)
3196 map->p[i] = isl_basic_map_align_divs(ctx, map->p[i], map->p[0]);
3201 static struct isl_map *add_cut_constraint(struct isl_ctx *ctx,
3202 struct isl_map *dst,
3203 struct isl_basic_map *src, isl_int *c,
3204 unsigned len, int oppose)
3206 struct isl_basic_map *copy = NULL;
3211 copy = isl_basic_map_copy(ctx, src);
3212 copy = isl_basic_map_cow(ctx, copy);
3215 copy = isl_basic_map_extend(ctx, copy,
3216 copy->nparam, copy->n_in, copy->n_out, 0, 0, 1);
3217 k = isl_basic_map_alloc_inequality(ctx, copy);
3221 isl_seq_neg(copy->ineq[k], c, len);
3223 isl_seq_cpy(copy->ineq[k], c, len);
3224 total = 1 + copy->nparam + copy->n_in + copy->n_out + copy->extra;
3225 isl_seq_clr(copy->ineq[k]+len, total - len);
3226 isl_inequality_negate(ctx, copy, k);
3227 copy = isl_basic_map_simplify(ctx, copy);
3228 copy = isl_basic_map_finalize(ctx, copy);
3229 is_empty = isl_basic_map_is_empty(ctx, copy);
3233 dst = isl_map_add(ctx, dst, copy);
3235 isl_basic_map_free(ctx, copy);
3238 isl_basic_map_free(ctx, copy);
3239 isl_map_free(ctx, dst);
3243 static struct isl_map *subtract(struct isl_ctx *ctx, struct isl_map *map,
3244 struct isl_basic_map *bmap)
3248 struct isl_map *rest = NULL;
3250 unsigned total = bmap->nparam + bmap->n_in + bmap->n_out + bmap->n_div;
3257 if (F_ISSET(map, ISL_MAP_DISJOINT))
3258 FL_SET(flags, ISL_MAP_DISJOINT);
3260 max = map->n * (2 * bmap->n_eq + bmap->n_ineq);
3261 rest = isl_map_alloc(ctx, map->nparam, map->n_in, map->n_out,
3266 for (i = 0; i < map->n; ++i) {
3267 map->p[i] = isl_basic_map_align_divs(ctx, map->p[i], bmap);
3272 for (j = 0; j < map->n; ++j)
3273 map->p[j] = isl_basic_map_cow(ctx, map->p[j]);
3275 for (i = 0; i < bmap->n_eq; ++i) {
3276 for (j = 0; j < map->n; ++j) {
3277 rest = add_cut_constraint(ctx, rest,
3278 map->p[j], bmap->eq[i], 1+total, 0);
3282 rest = add_cut_constraint(ctx, rest,
3283 map->p[j], bmap->eq[i], 1+total, 1);
3287 map->p[j] = isl_basic_map_extend(ctx, map->p[j],
3288 map->p[j]->nparam, map->p[j]->n_in,
3289 map->p[j]->n_out, 0, 1, 0);
3292 k = isl_basic_map_alloc_equality(ctx, map->p[j]);
3295 isl_seq_cpy(map->p[j]->eq[k], bmap->eq[i], 1+total);
3296 isl_seq_clr(map->p[j]->eq[k]+1+total,
3297 map->p[j]->extra - bmap->n_div);
3301 for (i = 0; i < bmap->n_ineq; ++i) {
3302 for (j = 0; j < map->n; ++j) {
3303 rest = add_cut_constraint(ctx, rest,
3304 map->p[j], bmap->ineq[i], 1+total, 0);
3308 map->p[j] = isl_basic_map_extend(ctx, map->p[j],
3309 map->p[j]->nparam, map->p[j]->n_in,
3310 map->p[j]->n_out, 0, 0, 1);
3313 k = isl_basic_map_alloc_inequality(ctx, map->p[j]);
3316 isl_seq_cpy(map->p[j]->ineq[k], bmap->ineq[i], 1+total);
3317 isl_seq_clr(map->p[j]->ineq[k]+1+total,
3318 map->p[j]->extra - bmap->n_div);
3322 isl_map_free(ctx, map);
3325 isl_map_free(ctx, map);
3326 isl_map_free(ctx, rest);
3330 struct isl_map *isl_map_subtract(struct isl_ctx *ctx, struct isl_map *map1,
3331 struct isl_map *map2)
3337 isl_assert(ctx, map1->nparam == map2->nparam, goto error);
3338 isl_assert(ctx, map1->n_in == map2->n_in, goto error);
3339 isl_assert(ctx, map1->n_out == map2->n_out, goto error);
3341 if (isl_map_is_empty(ctx, map2)) {
3342 isl_map_free(ctx, map2);
3346 map1 = isl_map_compute_divs(ctx, map1);
3347 map2 = isl_map_compute_divs(ctx, map2);
3351 for (i = 0; map1 && i < map2->n; ++i)
3352 map1 = subtract(ctx, map1, map2->p[i]);
3354 isl_map_free(ctx, map2);
3357 isl_map_free(ctx, map1);
3358 isl_map_free(ctx, map2);
3362 struct isl_set *isl_set_subtract(struct isl_ctx *ctx, struct isl_set *set1,
3363 struct isl_set *set2)
3365 return (struct isl_set *)
3366 isl_map_subtract(ctx,
3367 (struct isl_map *)set1, (struct isl_map *)set2);
3370 struct isl_set *isl_set_apply(struct isl_ctx *ctx, struct isl_set *set,
3371 struct isl_map *map)
3373 isl_assert(ctx, set->dim == map->n_in, goto error);
3374 map = isl_map_intersect_domain(ctx, map, set);
3375 set = isl_map_range(ctx, map);
3378 isl_set_free(ctx, set);
3379 isl_map_free(ctx, map);
3383 /* There is no need to cow as removing empty parts doesn't change
3384 * the meaning of the set.
3386 struct isl_map *isl_map_remove_empty_parts(struct isl_ctx *ctx,
3387 struct isl_map *map)
3394 for (i = map->n-1; i >= 0; --i) {
3395 if (!F_ISSET(map->p[i], ISL_BASIC_MAP_EMPTY))
3397 isl_basic_map_free(ctx, map->p[i]);
3399 map->p[i] = map->p[map->n-1];
3406 struct isl_set *isl_set_remove_empty_parts(struct isl_ctx *ctx,
3407 struct isl_set *set)
3409 return (struct isl_set *)
3410 isl_map_remove_empty_parts(ctx, (struct isl_map *)set);