11 #include "isl_map_private.h"
12 #include "isl_map_piplib.h"
13 #include "isl_sample.h"
16 /* Maps dst positions to src positions */
22 static struct isl_dim_map *isl_dim_map_alloc(struct isl_ctx *ctx, unsigned len)
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));
30 dim_map->len = 1 + len;
32 for (i = 0; i < len; ++i)
33 dim_map->pos[1 + i] = -1;
37 static unsigned n(struct isl_dim *dim, enum isl_dim_type 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)
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)
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)
75 if (!dim_map || !bmap)
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)
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;
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 static unsigned basic_map_offset(struct isl_basic_map *bmap,
111 enum isl_dim_type type)
113 struct isl_dim *dim = bmap->dim;
115 case isl_dim_param: return 1;
116 case isl_dim_in: return 1 + dim->nparam;
117 case isl_dim_out: return 1 + dim->nparam + dim->n_in;
118 case isl_dim_div: return 1 + dim->nparam + dim->n_in + dim->n_out;
122 static unsigned map_offset(struct isl_map *map, enum isl_dim_type type)
124 return pos(map->dim, type);
127 unsigned isl_basic_set_dim(const struct isl_basic_set *bset,
128 enum isl_dim_type type)
130 return isl_basic_map_dim((const struct isl_basic_map*)bset, type);
133 unsigned isl_basic_set_n_dim(const struct isl_basic_set *bset)
135 return bset->dim->n_out;
138 unsigned isl_basic_set_n_param(const struct isl_basic_set *bset)
140 return bset->dim->nparam;
143 unsigned isl_basic_set_total_dim(const struct isl_basic_set *bset)
145 return isl_dim_total(bset->dim) + bset->n_div;
148 unsigned isl_set_n_dim(const struct isl_set *set)
150 return set->dim->n_out;
153 unsigned isl_set_n_param(const struct isl_set *set)
155 return set->dim->nparam;
158 unsigned isl_basic_map_n_in(const struct isl_basic_map *bmap)
160 return bmap->dim->n_in;
163 unsigned isl_basic_map_n_out(const struct isl_basic_map *bmap)
165 return bmap->dim->n_out;
168 unsigned isl_basic_map_n_param(const struct isl_basic_map *bmap)
170 return bmap->dim->nparam;
173 unsigned isl_basic_map_n_div(const struct isl_basic_map *bmap)
178 unsigned isl_basic_map_total_dim(const struct isl_basic_map *bmap)
180 return isl_dim_total(bmap->dim) + bmap->n_div;
183 unsigned isl_map_n_in(const struct isl_map *map)
185 return map->dim->n_in;
188 unsigned isl_map_n_out(const struct isl_map *map)
190 return map->dim->n_out;
193 unsigned isl_map_n_param(const struct isl_map *map)
195 return map->dim->nparam;
198 int isl_map_compatible_domain(struct isl_map *map, struct isl_set *set)
200 return map->dim->n_in == set->dim->n_out &&
201 map->dim->nparam == set->dim->nparam;
204 int isl_basic_map_compatible_domain(struct isl_basic_map *bmap,
205 struct isl_basic_set *bset)
207 return bmap->dim->n_in == bset->dim->n_out &&
208 bmap->dim->nparam == bset->dim->nparam;
211 int isl_basic_map_compatible_range(struct isl_basic_map *bmap,
212 struct isl_basic_set *bset)
214 return bmap->dim->n_out == bset->dim->n_out &&
215 bmap->dim->nparam == bset->dim->nparam;
218 static struct isl_basic_map *basic_map_init(struct isl_ctx *ctx,
219 struct isl_basic_map *bmap, unsigned extra,
220 unsigned n_eq, unsigned n_ineq)
223 size_t row_size = 1 + isl_dim_total(bmap->dim) + extra;
225 bmap->block = isl_blk_alloc(ctx, (n_eq + n_ineq) * row_size);
226 if (isl_blk_is_error(bmap->block)) {
231 bmap->eq = isl_alloc_array(ctx, isl_int *, n_eq + n_ineq);
233 isl_blk_free(ctx, bmap->block);
239 bmap->block2 = isl_blk_empty();
242 bmap->block2 = isl_blk_alloc(ctx, extra * (1 + row_size));
243 if (isl_blk_is_error(bmap->block2)) {
245 isl_blk_free(ctx, bmap->block);
250 bmap->div = isl_alloc_array(ctx, isl_int *, extra);
252 isl_blk_free(ctx, bmap->block2);
254 isl_blk_free(ctx, bmap->block);
260 for (i = 0; i < n_eq + n_ineq; ++i)
261 bmap->eq[i] = bmap->block.data + i * row_size;
263 for (i = 0; i < extra; ++i)
264 bmap->div[i] = bmap->block2.data + i * (1 + row_size);
270 bmap->c_size = n_eq + n_ineq;
271 bmap->ineq = bmap->eq + n_eq;
280 isl_basic_map_free(bmap);
284 struct isl_basic_set *isl_basic_set_alloc(struct isl_ctx *ctx,
285 unsigned nparam, unsigned dim, unsigned extra,
286 unsigned n_eq, unsigned n_ineq)
288 struct isl_basic_map *bmap;
289 bmap = isl_basic_map_alloc(ctx, nparam, 0, dim, extra, n_eq, n_ineq);
290 return (struct isl_basic_set *)bmap;
293 struct isl_basic_set *isl_basic_set_alloc_dim(struct isl_dim *dim,
294 unsigned extra, unsigned n_eq, unsigned n_ineq)
296 struct isl_basic_map *bmap;
299 isl_assert(dim->ctx, dim->n_in == 0, return NULL);
300 bmap = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
301 return (struct isl_basic_set *)bmap;
304 struct isl_basic_map *isl_basic_map_alloc_dim(struct isl_dim *dim,
305 unsigned extra, unsigned n_eq, unsigned n_ineq)
307 struct isl_basic_map *bmap;
311 bmap = isl_alloc_type(dim->ctx, struct isl_basic_map);
316 return basic_map_init(dim->ctx, bmap, extra, n_eq, n_ineq);
322 struct isl_basic_map *isl_basic_map_alloc(struct isl_ctx *ctx,
323 unsigned nparam, unsigned in, unsigned out, unsigned extra,
324 unsigned n_eq, unsigned n_ineq)
326 struct isl_basic_map *bmap;
329 dim = isl_dim_alloc(ctx, nparam, in, out);
333 bmap = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
337 static void dup_constraints(
338 struct isl_basic_map *dst, struct isl_basic_map *src)
341 unsigned total = isl_basic_map_total_dim(src);
343 for (i = 0; i < src->n_eq; ++i) {
344 int j = isl_basic_map_alloc_equality(dst);
345 isl_seq_cpy(dst->eq[j], src->eq[i], 1+total);
348 for (i = 0; i < src->n_ineq; ++i) {
349 int j = isl_basic_map_alloc_inequality(dst);
350 isl_seq_cpy(dst->ineq[j], src->ineq[i], 1+total);
353 for (i = 0; i < src->n_div; ++i) {
354 int j = isl_basic_map_alloc_div(dst);
355 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total);
357 F_SET(dst, ISL_BASIC_SET_FINAL);
360 struct isl_basic_map *isl_basic_map_dup(struct isl_basic_map *bmap)
362 struct isl_basic_map *dup;
366 dup = isl_basic_map_alloc_dim(isl_dim_copy(bmap->dim),
367 bmap->n_div, bmap->n_eq, bmap->n_ineq);
370 dup->flags = bmap->flags;
371 dup_constraints(dup, bmap);
372 dup->sample = isl_vec_copy(bmap->ctx, bmap->sample);
376 struct isl_basic_set *isl_basic_set_dup(struct isl_basic_set *bset)
378 struct isl_basic_map *dup;
380 dup = isl_basic_map_dup((struct isl_basic_map *)bset);
381 return (struct isl_basic_set *)dup;
384 struct isl_basic_set *isl_basic_set_copy(struct isl_basic_set *bset)
389 if (F_ISSET(bset, ISL_BASIC_SET_FINAL)) {
393 return isl_basic_set_dup(bset);
396 struct isl_set *isl_set_copy(struct isl_set *set)
405 struct isl_basic_map *isl_basic_map_copy(struct isl_basic_map *bmap)
410 if (F_ISSET(bmap, ISL_BASIC_SET_FINAL)) {
414 return isl_basic_map_dup(bmap);
417 struct isl_map *isl_map_copy(struct isl_map *map)
426 void isl_basic_map_free(struct isl_basic_map *bmap)
434 isl_ctx_deref(bmap->ctx);
436 isl_blk_free(bmap->ctx, bmap->block2);
438 isl_blk_free(bmap->ctx, bmap->block);
439 isl_vec_free(bmap->ctx, bmap->sample);
440 isl_dim_free(bmap->dim);
444 void isl_basic_set_free(struct isl_basic_set *bset)
446 isl_basic_map_free((struct isl_basic_map *)bset);
449 int isl_basic_map_alloc_equality(struct isl_basic_map *bmap)
455 isl_assert(ctx, bmap->n_eq + bmap->n_ineq < bmap->c_size, return -1);
456 isl_assert(ctx, bmap->eq + bmap->n_eq <= bmap->ineq, return -1);
457 if (bmap->eq + bmap->n_eq == bmap->ineq) {
459 int j = isl_basic_map_alloc_inequality(bmap);
463 bmap->ineq[0] = bmap->ineq[j];
468 isl_seq_clr(bmap->eq[bmap->n_eq] +
469 1 + isl_basic_map_total_dim(bmap),
470 bmap->extra - bmap->n_div);
471 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
475 int isl_basic_set_alloc_equality(struct isl_basic_set *bset)
477 return isl_basic_map_alloc_equality((struct isl_basic_map *)bset);
480 int isl_basic_map_free_equality(struct isl_basic_map *bmap, unsigned n)
484 isl_assert(bmap->ctx, n <= bmap->n_eq, return -1);
489 int isl_basic_map_drop_equality(struct isl_basic_map *bmap, unsigned pos)
494 isl_assert(bmap->ctx, pos < bmap->n_eq, return -1);
496 if (pos != bmap->n_eq - 1) {
498 bmap->eq[pos] = bmap->eq[bmap->n_eq - 1];
499 bmap->eq[bmap->n_eq - 1] = t;
505 int isl_basic_set_drop_equality(struct isl_basic_set *bset, unsigned pos)
507 return isl_basic_map_drop_equality((struct isl_basic_map *)bset, pos);
510 void isl_basic_map_inequality_to_equality(
511 struct isl_basic_map *bmap, unsigned pos)
516 bmap->ineq[pos] = bmap->ineq[0];
517 bmap->ineq[0] = bmap->eq[bmap->n_eq];
518 bmap->eq[bmap->n_eq] = t;
522 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
523 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
526 int isl_basic_map_alloc_inequality(struct isl_basic_map *bmap)
532 isl_assert(ctx, (bmap->ineq - bmap->eq) + bmap->n_ineq < bmap->c_size,
534 F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
535 F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
536 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
537 isl_seq_clr(bmap->ineq[bmap->n_ineq] +
538 1 + isl_basic_map_total_dim(bmap),
539 bmap->extra - bmap->n_div);
540 return bmap->n_ineq++;
543 int isl_basic_set_alloc_inequality(struct isl_basic_set *bset)
545 return isl_basic_map_alloc_inequality((struct isl_basic_map *)bset);
548 int isl_basic_map_free_inequality(struct isl_basic_map *bmap, unsigned n)
552 isl_assert(bmap->ctx, n <= bmap->n_ineq, return -1);
557 int isl_basic_set_free_inequality(struct isl_basic_set *bset, unsigned n)
559 return isl_basic_map_free_inequality((struct isl_basic_map *)bset, n);
562 int isl_basic_map_drop_inequality(struct isl_basic_map *bmap, unsigned pos)
567 isl_assert(bmap->ctx, pos < bmap->n_ineq, return -1);
569 if (pos != bmap->n_ineq - 1) {
571 bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
572 bmap->ineq[bmap->n_ineq - 1] = t;
573 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
579 int isl_basic_set_drop_inequality(struct isl_basic_set *bset, unsigned pos)
581 return isl_basic_map_drop_inequality((struct isl_basic_map *)bset, pos);
584 int isl_basic_map_alloc_div(struct isl_basic_map *bmap)
588 isl_assert(bmap->ctx, bmap->n_div < bmap->extra, return -1);
589 isl_seq_clr(bmap->div[bmap->n_div] +
590 1 + 1 + isl_basic_map_total_dim(bmap),
591 bmap->extra - bmap->n_div);
592 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
593 return bmap->n_div++;
596 int isl_basic_map_free_div(struct isl_basic_map *bmap, unsigned n)
600 isl_assert(bmap->ctx, n <= bmap->n_div, return -1);
605 /* Copy constraint from src to dst, putting the vars of src at offset
606 * dim_off in dst and the divs of src at offset div_off in dst.
607 * If both sets are actually map, then dim_off applies to the input
610 static void copy_constraint(struct isl_basic_map *dst_map, isl_int *dst,
611 struct isl_basic_map *src_map, isl_int *src,
612 unsigned in_off, unsigned out_off, unsigned div_off)
614 unsigned src_nparam = isl_basic_map_n_param(src_map);
615 unsigned dst_nparam = isl_basic_map_n_param(dst_map);
616 unsigned src_in = isl_basic_map_n_in(src_map);
617 unsigned dst_in = isl_basic_map_n_in(dst_map);
618 unsigned src_out = isl_basic_map_n_out(src_map);
619 unsigned dst_out = isl_basic_map_n_out(dst_map);
620 isl_int_set(dst[0], src[0]);
621 isl_seq_cpy(dst+1, src+1, isl_min(dst_nparam, src_nparam));
622 if (dst_nparam > src_nparam)
623 isl_seq_clr(dst+1+src_nparam,
624 dst_nparam - src_nparam);
625 isl_seq_clr(dst+1+dst_nparam, in_off);
626 isl_seq_cpy(dst+1+dst_nparam+in_off,
628 isl_min(dst_in-in_off, src_in));
629 if (dst_in-in_off > src_in)
630 isl_seq_clr(dst+1+dst_nparam+in_off+src_in,
631 dst_in - in_off - src_in);
632 isl_seq_clr(dst+1+dst_nparam+dst_in, out_off);
633 isl_seq_cpy(dst+1+dst_nparam+dst_in+out_off,
634 src+1+src_nparam+src_in,
635 isl_min(dst_out-out_off, src_out));
636 if (dst_out-out_off > src_out)
637 isl_seq_clr(dst+1+dst_nparam+dst_in+out_off+src_out,
638 dst_out - out_off - src_out);
639 isl_seq_clr(dst+1+dst_nparam+dst_in+dst_out, div_off);
640 isl_seq_cpy(dst+1+dst_nparam+dst_in+dst_out+div_off,
641 src+1+src_nparam+src_in+src_out,
642 isl_min(dst_map->extra-div_off, src_map->n_div));
643 if (dst_map->n_div-div_off > src_map->n_div)
644 isl_seq_clr(dst+1+dst_nparam+dst_in+dst_out+
645 div_off+src_map->n_div,
646 dst_map->n_div - div_off - src_map->n_div);
649 static void copy_div(struct isl_basic_map *dst_map, isl_int *dst,
650 struct isl_basic_map *src_map, isl_int *src,
651 unsigned in_off, unsigned out_off, unsigned div_off)
653 isl_int_set(dst[0], src[0]);
654 copy_constraint(dst_map, dst+1, src_map, src+1, in_off, out_off, div_off);
657 static struct isl_basic_map *add_constraints(struct isl_basic_map *bmap1,
658 struct isl_basic_map *bmap2, unsigned i_pos, unsigned o_pos)
663 if (!bmap1 || !bmap2)
666 div_off = bmap1->n_div;
668 for (i = 0; i < bmap2->n_eq; ++i) {
669 int i1 = isl_basic_map_alloc_equality(bmap1);
672 copy_constraint(bmap1, bmap1->eq[i1], bmap2, bmap2->eq[i],
673 i_pos, o_pos, div_off);
676 for (i = 0; i < bmap2->n_ineq; ++i) {
677 int i1 = isl_basic_map_alloc_inequality(bmap1);
680 copy_constraint(bmap1, bmap1->ineq[i1], bmap2, bmap2->ineq[i],
681 i_pos, o_pos, div_off);
684 for (i = 0; i < bmap2->n_div; ++i) {
685 int i1 = isl_basic_map_alloc_div(bmap1);
688 copy_div(bmap1, bmap1->div[i1], bmap2, bmap2->div[i],
689 i_pos, o_pos, div_off);
692 isl_basic_map_free(bmap2);
697 isl_basic_map_free(bmap1);
698 isl_basic_map_free(bmap2);
702 static void copy_constraint_dim_map(isl_int *dst, isl_int *src,
703 struct isl_dim_map *dim_map)
707 for (i = 0; i < dim_map->len; ++i) {
708 if (dim_map->pos[i] < 0)
709 isl_int_set_si(dst[i], 0);
711 isl_int_set(dst[i], src[dim_map->pos[i]]);
715 static void copy_div_dim_map(isl_int *dst, isl_int *src,
716 struct isl_dim_map *dim_map)
718 isl_int_set(dst[0], src[0]);
719 copy_constraint_dim_map(dst+1, src+1, dim_map);
722 static struct isl_basic_map *add_constraints_dim_map(struct isl_basic_map *dst,
723 struct isl_basic_map *src, struct isl_dim_map *dim_map)
727 if (!src || !dst || !dim_map)
730 for (i = 0; i < src->n_eq; ++i) {
731 int i1 = isl_basic_map_alloc_equality(dst);
734 copy_constraint_dim_map(dst->eq[i1], src->eq[i], dim_map);
737 for (i = 0; i < src->n_ineq; ++i) {
738 int i1 = isl_basic_map_alloc_inequality(dst);
741 copy_constraint_dim_map(dst->ineq[i1], src->ineq[i], dim_map);
744 for (i = 0; i < src->n_div; ++i) {
745 int i1 = isl_basic_map_alloc_div(dst);
748 copy_div_dim_map(dst->div[i1], src->div[i], dim_map);
752 isl_basic_map_free(src);
757 isl_basic_map_free(src);
758 isl_basic_map_free(dst);
762 struct isl_basic_set *isl_basic_set_add_constraints(struct isl_basic_set *bset1,
763 struct isl_basic_set *bset2, unsigned pos)
765 return (struct isl_basic_set *)
766 add_constraints((struct isl_basic_map *)bset1,
767 (struct isl_basic_map *)bset2, 0, pos);
770 struct isl_basic_map *isl_basic_map_extend_dim(struct isl_basic_map *base,
771 struct isl_dim *dim, unsigned extra,
772 unsigned n_eq, unsigned n_ineq)
774 struct isl_basic_map *ext;
781 base = isl_basic_map_cow(base);
785 dims_ok = isl_dim_equal(base->dim, dim) &&
786 base->extra >= base->n_div + extra;
788 if (dims_ok && n_eq == 0 && n_ineq == 0) {
793 isl_assert(base->ctx, base->dim->nparam <= dim->nparam, goto error);
794 isl_assert(base->ctx, base->dim->n_in <= dim->n_in, goto error);
795 isl_assert(base->ctx, base->dim->n_out <= dim->n_out, goto error);
796 extra += base->extra;
798 n_ineq += base->n_ineq;
800 ext = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
806 ext = add_constraints(ext, base, 0, 0);
809 F_CLR(ext, ISL_BASIC_SET_FINAL);
816 isl_basic_map_free(base);
820 struct isl_basic_map *isl_basic_map_extend_constraints(
821 struct isl_basic_map *base, unsigned n_eq, unsigned n_ineq)
825 return isl_basic_map_extend_dim(base, isl_dim_copy(base->dim),
829 struct isl_basic_map *isl_basic_map_extend(struct isl_basic_map *base,
830 unsigned nparam, unsigned n_in, unsigned n_out, unsigned extra,
831 unsigned n_eq, unsigned n_ineq)
833 struct isl_basic_map *bmap;
838 dim = isl_dim_alloc(base->ctx, nparam, n_in, n_out);
842 bmap = isl_basic_map_extend_dim(base, dim, extra, n_eq, n_ineq);
846 struct isl_basic_set *isl_basic_set_extend(struct isl_basic_set *base,
847 unsigned nparam, unsigned dim, unsigned extra,
848 unsigned n_eq, unsigned n_ineq)
850 return (struct isl_basic_set *)
851 isl_basic_map_extend((struct isl_basic_map *)base,
852 nparam, 0, dim, extra, n_eq, n_ineq);
855 struct isl_basic_set *isl_basic_set_extend_constraints(
856 struct isl_basic_set *base, unsigned n_eq, unsigned n_ineq)
858 return (struct isl_basic_set *)
859 isl_basic_map_extend_constraints((struct isl_basic_map *)base,
863 struct isl_basic_set *isl_basic_set_cow(struct isl_basic_set *bset)
865 return (struct isl_basic_set *)
866 isl_basic_map_cow((struct isl_basic_map *)bset);
869 struct isl_basic_map *isl_basic_map_cow(struct isl_basic_map *bmap)
876 bmap = isl_basic_map_dup(bmap);
878 F_CLR(bmap, ISL_BASIC_SET_FINAL);
882 struct isl_set *isl_set_cow(struct isl_set *set)
890 return isl_set_dup(set);
893 struct isl_map *isl_map_cow(struct isl_map *map)
901 return isl_map_dup(map);
904 static void swap_vars(struct isl_blk blk, isl_int *a,
905 unsigned a_len, unsigned b_len)
907 isl_seq_cpy(blk.data, a+a_len, b_len);
908 isl_seq_cpy(blk.data+b_len, a, a_len);
909 isl_seq_cpy(a, blk.data, b_len+a_len);
912 struct isl_basic_set *isl_basic_set_swap_vars(
913 struct isl_basic_set *bset, unsigned n)
923 nparam = isl_basic_set_n_param(bset);
924 dim = isl_basic_set_n_dim(bset);
925 isl_assert(bset->ctx, n <= dim, goto error);
930 bset = isl_basic_set_cow(bset);
934 blk = isl_blk_alloc(bset->ctx, dim);
935 if (isl_blk_is_error(blk))
938 for (i = 0; i < bset->n_eq; ++i)
940 bset->eq[i]+1+nparam, n, dim - n);
942 for (i = 0; i < bset->n_ineq; ++i)
944 bset->ineq[i]+1+nparam, n, dim - n);
946 for (i = 0; i < bset->n_div; ++i)
948 bset->div[i]+1+1+nparam, n, dim - n);
950 isl_blk_free(bset->ctx, blk);
952 F_CLR(bset, ISL_BASIC_SET_NORMALIZED);
956 isl_basic_set_free(bset);
960 struct isl_set *isl_set_swap_vars(struct isl_set *set, unsigned n)
963 set = isl_set_cow(set);
967 for (i = 0; i < set->n; ++i) {
968 set->p[i] = isl_basic_set_swap_vars(set->p[i], n);
974 F_CLR(set, ISL_SET_NORMALIZED);
978 struct isl_basic_map *isl_basic_map_set_to_empty(struct isl_basic_map *bmap)
984 total = isl_basic_map_total_dim(bmap);
985 isl_basic_map_free_div(bmap, bmap->n_div);
986 isl_basic_map_free_inequality(bmap, bmap->n_ineq);
988 isl_basic_map_free_equality(bmap, bmap->n_eq-1);
990 isl_basic_map_alloc_equality(bmap);
994 isl_int_set_si(bmap->eq[i][0], 1);
995 isl_seq_clr(bmap->eq[i]+1, total);
996 F_SET(bmap, ISL_BASIC_MAP_EMPTY);
997 return isl_basic_map_finalize(bmap);
999 isl_basic_map_free(bmap);
1003 struct isl_basic_set *isl_basic_set_set_to_empty(struct isl_basic_set *bset)
1005 return (struct isl_basic_set *)
1006 isl_basic_map_set_to_empty((struct isl_basic_map *)bset);
1009 static void swap_div(struct isl_basic_map *bmap, int a, int b)
1012 unsigned off = isl_dim_total(bmap->dim);
1013 isl_int *t = bmap->div[a];
1014 bmap->div[a] = bmap->div[b];
1017 for (i = 0; i < bmap->n_eq; ++i)
1018 isl_int_swap(bmap->eq[i][1+off+a], bmap->eq[i][1+off+b]);
1020 for (i = 0; i < bmap->n_ineq; ++i)
1021 isl_int_swap(bmap->ineq[i][1+off+a], bmap->ineq[i][1+off+b]);
1023 for (i = 0; i < bmap->n_div; ++i)
1024 isl_int_swap(bmap->div[i][1+1+off+a], bmap->div[i][1+1+off+b]);
1025 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
1028 /* Eliminate the specified n dimensions starting at first from the
1029 * constraints using Fourier-Motzkin, The dimensions themselves
1032 struct isl_set *isl_set_eliminate_dims(struct isl_set *set,
1033 unsigned first, unsigned n)
1043 set = isl_set_cow(set);
1046 isl_assert(set->ctx, first+n <= isl_set_n_dim(set), goto error);
1047 nparam = isl_set_n_param(set);
1049 for (i = 0; i < set->n; ++i) {
1050 set->p[i] = isl_basic_set_eliminate_vars(set->p[i],
1061 /* Project out n dimensions starting at first using Fourier-Motzkin */
1062 struct isl_set *isl_set_remove_dims(struct isl_set *set,
1063 unsigned first, unsigned n)
1065 set = isl_set_eliminate_dims(set, first, n);
1066 set = isl_set_drop_dims(set, first, n);
1070 struct isl_basic_set *isl_basic_set_remove_divs(struct isl_basic_set *bset)
1072 bset = isl_basic_set_eliminate_vars(bset, isl_dim_total(bset->dim),
1080 struct isl_set *isl_set_remove_divs(struct isl_set *set)
1089 set = isl_set_cow(set);
1093 for (i = 0; i < set->n; ++i) {
1094 set->p[i] = isl_basic_set_remove_divs(set->p[i]);
1104 /* Project out n inputs starting at first using Fourier-Motzkin */
1105 struct isl_map *isl_map_remove_inputs(struct isl_map *map,
1106 unsigned first, unsigned n)
1114 map = isl_map_cow(map);
1117 nparam = isl_map_n_param(map);
1118 isl_assert(map->ctx, first+n <= isl_map_n_in(map), goto error);
1120 for (i = 0; i < map->n; ++i) {
1121 map->p[i] = isl_basic_map_eliminate_vars(map->p[i],
1126 map = isl_map_drop_inputs(map, first, n);
1133 /* Project out n dimensions starting at first using Fourier-Motzkin */
1134 struct isl_basic_set *isl_basic_set_remove_dims(struct isl_basic_set *bset,
1135 unsigned first, unsigned n)
1137 unsigned nparam = isl_basic_set_n_param(bset);
1138 bset = isl_basic_set_eliminate_vars(bset, nparam + first, n);
1139 bset = isl_basic_set_drop_dims(bset, first, n);
1143 static void dump_term(struct isl_basic_map *bmap,
1144 isl_int c, int pos, FILE *out)
1147 unsigned in = isl_basic_map_n_in(bmap);
1148 unsigned dim = in + isl_basic_map_n_out(bmap);
1149 unsigned nparam = isl_basic_map_n_param(bmap);
1151 isl_int_print(out, c, 0);
1153 if (!isl_int_is_one(c))
1154 isl_int_print(out, c, 0);
1155 if (pos < 1 + nparam) {
1156 name = isl_dim_get_name(bmap->dim,
1157 isl_dim_param, pos - 1);
1159 fprintf(out, "%s", name);
1161 fprintf(out, "p%d", pos - 1);
1162 } else if (pos < 1 + nparam + in)
1163 fprintf(out, "i%d", pos - 1 - nparam);
1164 else if (pos < 1 + nparam + dim)
1165 fprintf(out, "o%d", pos - 1 - nparam - in);
1167 fprintf(out, "e%d", pos - 1 - nparam - dim);
1171 static void dump_constraint_sign(struct isl_basic_map *bmap, isl_int *c,
1172 int sign, FILE *out)
1176 unsigned len = 1 + isl_basic_map_total_dim(bmap);
1180 for (i = 0, first = 1; i < len; ++i) {
1181 if (isl_int_sgn(c[i]) * sign <= 0)
1184 fprintf(out, " + ");
1186 isl_int_abs(v, c[i]);
1187 dump_term(bmap, v, i, out);
1194 static void dump_constraint(struct isl_basic_map *bmap, isl_int *c,
1195 const char *op, FILE *out, int indent)
1199 fprintf(out, "%*s", indent, "");
1201 dump_constraint_sign(bmap, c, 1, out);
1202 fprintf(out, " %s ", op);
1203 dump_constraint_sign(bmap, c, -1, out);
1207 for (i = bmap->n_div; i < bmap->extra; ++i) {
1208 if (isl_int_is_zero(c[1+isl_dim_total(bmap->dim)+i]))
1210 fprintf(out, "%*s", indent, "");
1211 fprintf(out, "ERROR: unused div coefficient not zero\n");
1216 static void dump_constraints(struct isl_basic_map *bmap,
1217 isl_int **c, unsigned n,
1218 const char *op, FILE *out, int indent)
1222 for (i = 0; i < n; ++i)
1223 dump_constraint(bmap, c[i], op, out, indent);
1226 static void dump_affine(struct isl_basic_map *bmap, isl_int *exp, FILE *out)
1230 unsigned total = isl_basic_map_total_dim(bmap);
1232 for (j = 0; j < 1 + total; ++j) {
1233 if (isl_int_is_zero(exp[j]))
1235 if (!first && isl_int_is_pos(exp[j]))
1237 dump_term(bmap, exp[j], j, out);
1242 static void dump(struct isl_basic_map *bmap, FILE *out, int indent)
1246 dump_constraints(bmap, bmap->eq, bmap->n_eq, "=", out, indent);
1247 dump_constraints(bmap, bmap->ineq, bmap->n_ineq, ">=", out, indent);
1249 for (i = 0; i < bmap->n_div; ++i) {
1250 fprintf(out, "%*s", indent, "");
1251 fprintf(out, "e%d = [(", i);
1252 dump_affine(bmap, bmap->div[i]+1, out);
1254 isl_int_print(out, bmap->div[i][0], 0);
1255 fprintf(out, "]\n");
1259 void isl_basic_set_dump(struct isl_basic_set *bset, FILE *out, int indent)
1262 fprintf(out, "null basic set\n");
1266 fprintf(out, "%*s", indent, "");
1267 fprintf(out, "ref: %d, nparam: %d, dim: %d, extra: %d, flags: %x\n",
1268 bset->ref, bset->dim->nparam, bset->dim->n_out,
1269 bset->extra, bset->flags);
1270 dump((struct isl_basic_map *)bset, out, indent);
1273 void isl_basic_map_dump(struct isl_basic_map *bmap, FILE *out, int indent)
1276 fprintf(out, "null basic map\n");
1280 fprintf(out, "%*s", indent, "");
1281 fprintf(out, "ref: %d, nparam: %d, in: %d, out: %d, extra: %d, "
1282 "flags: %x, n_name: %d\n",
1284 bmap->dim->nparam, bmap->dim->n_in, bmap->dim->n_out,
1285 bmap->extra, bmap->flags, bmap->dim->n_name);
1286 dump(bmap, out, indent);
1289 int isl_inequality_negate(struct isl_basic_map *bmap, unsigned pos)
1294 total = isl_basic_map_total_dim(bmap);
1295 isl_assert(bmap->ctx, pos < bmap->n_ineq, return -1);
1296 isl_seq_neg(bmap->ineq[pos], bmap->ineq[pos], 1 + total);
1297 isl_int_sub_ui(bmap->ineq[pos][0], bmap->ineq[pos][0], 1);
1298 F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED);
1302 struct isl_set *isl_set_alloc_dim(struct isl_dim *dim, int n, unsigned flags)
1304 struct isl_set *set;
1308 isl_assert(dim->ctx, dim->n_in == 0, return NULL);
1309 isl_assert(dim->ctx, n >= 0, return NULL);
1310 set = isl_alloc(dim->ctx, struct isl_set,
1311 sizeof(struct isl_set) +
1312 n * sizeof(struct isl_basic_set *));
1316 set->ctx = dim->ctx;
1317 isl_ctx_ref(set->ctx);
1329 struct isl_set *isl_set_alloc(struct isl_ctx *ctx,
1330 unsigned nparam, unsigned dim, int n, unsigned flags)
1332 struct isl_set *set;
1333 struct isl_dim *dims;
1335 dims = isl_dim_alloc(ctx, nparam, 0, dim);
1339 set = isl_set_alloc_dim(dims, n, flags);
1343 struct isl_set *isl_set_dup(struct isl_set *set)
1346 struct isl_set *dup;
1351 dup = isl_set_alloc_dim(isl_dim_copy(set->dim), set->n, set->flags);
1354 for (i = 0; i < set->n; ++i)
1355 dup = isl_set_add(dup, isl_basic_set_copy(set->p[i]));
1359 struct isl_set *isl_set_from_basic_set(struct isl_basic_set *bset)
1361 struct isl_set *set;
1366 set = isl_set_alloc_dim(isl_dim_copy(bset->dim), 1, ISL_MAP_DISJOINT);
1368 isl_basic_set_free(bset);
1371 return isl_set_add(set, bset);
1374 struct isl_map *isl_map_from_basic_map(struct isl_basic_map *bmap)
1376 struct isl_map *map;
1381 map = isl_map_alloc_dim(isl_dim_copy(bmap->dim), 1, ISL_MAP_DISJOINT);
1383 isl_basic_map_free(bmap);
1386 return isl_map_add(map, bmap);
1389 struct isl_set *isl_set_add(struct isl_set *set, struct isl_basic_set *bset)
1393 isl_assert(set->ctx, isl_dim_equal(set->dim, bset->dim), goto error);
1394 isl_assert(set->ctx, set->n < set->size, goto error);
1395 set->p[set->n] = bset;
1402 isl_basic_set_free(bset);
1406 void isl_set_free(struct isl_set *set)
1416 isl_ctx_deref(set->ctx);
1417 for (i = 0; i < set->n; ++i)
1418 isl_basic_set_free(set->p[i]);
1419 isl_dim_free(set->dim);
1423 void isl_set_dump(struct isl_set *set, FILE *out, int indent)
1428 fprintf(out, "null set\n");
1432 fprintf(out, "%*s", indent, "");
1433 fprintf(out, "ref: %d, n: %d, nparam: %d, dim: %d, flags: %x\n",
1434 set->ref, set->n, set->dim->nparam, set->dim->n_out,
1436 for (i = 0; i < set->n; ++i) {
1437 fprintf(out, "%*s", indent, "");
1438 fprintf(out, "basic set %d:\n", i);
1439 isl_basic_set_dump(set->p[i], out, indent+4);
1443 void isl_map_dump(struct isl_map *map, FILE *out, int indent)
1448 fprintf(out, "null map\n");
1452 fprintf(out, "%*s", indent, "");
1453 fprintf(out, "ref: %d, n: %d, nparam: %d, in: %d, out: %d, "
1454 "flags: %x, n_name: %d\n",
1455 map->ref, map->n, map->dim->nparam, map->dim->n_in,
1456 map->dim->n_out, map->flags, map->dim->n_name);
1457 for (i = 0; i < map->n; ++i) {
1458 fprintf(out, "%*s", indent, "");
1459 fprintf(out, "basic map %d:\n", i);
1460 isl_basic_map_dump(map->p[i], out, indent+4);
1464 struct isl_basic_map *isl_basic_map_intersect_domain(
1465 struct isl_basic_map *bmap, struct isl_basic_set *bset)
1467 struct isl_basic_map *bmap_domain;
1468 struct isl_dim *dim;
1473 isl_assert(set->ctx, isl_basic_map_compatible_domain(bmap, bset),
1476 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1477 bset->n_div, bset->n_eq, bset->n_ineq);
1480 dim = isl_dim_reverse(isl_dim_copy(bset->dim));
1481 bmap_domain = isl_basic_map_from_basic_set(bset, dim);
1482 bmap = add_constraints(bmap, bmap_domain, 0, 0);
1484 bmap = isl_basic_map_simplify(bmap);
1485 return isl_basic_map_finalize(bmap);
1487 isl_basic_map_free(bmap);
1488 isl_basic_set_free(bset);
1492 struct isl_basic_map *isl_basic_map_intersect_range(
1493 struct isl_basic_map *bmap, struct isl_basic_set *bset)
1495 struct isl_basic_map *bmap_range;
1500 isl_assert(bset->ctx, isl_basic_map_compatible_range(bmap, bset),
1503 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1504 bset->n_div, bset->n_eq, bset->n_ineq);
1507 bmap_range = isl_basic_map_from_basic_set(bset, isl_dim_copy(bset->dim));
1508 bmap = add_constraints(bmap, bmap_range, 0, 0);
1510 bmap = isl_basic_map_simplify(bmap);
1511 return isl_basic_map_finalize(bmap);
1513 isl_basic_map_free(bmap);
1514 isl_basic_set_free(bset);
1518 struct isl_basic_map *isl_basic_map_intersect(
1519 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1521 if (!bmap1 || !bmap2)
1524 isl_assert(bmap1->ctx, isl_dim_match(bmap1->dim, isl_dim_param,
1525 bmap2->dim, isl_dim_param), goto error);
1526 if (isl_dim_total(bmap1->dim) ==
1527 isl_dim_size(bmap1->dim, isl_dim_param) &&
1528 isl_dim_total(bmap2->dim) !=
1529 isl_dim_size(bmap2->dim, isl_dim_param))
1530 return isl_basic_map_intersect(bmap2, bmap1);
1532 if (isl_dim_total(bmap2->dim) !=
1533 isl_dim_size(bmap2->dim, isl_dim_param))
1534 isl_assert(bmap1->ctx,
1535 isl_dim_equal(bmap1->dim, bmap2->dim), goto error);
1537 bmap1 = isl_basic_map_extend_dim(bmap1, isl_dim_copy(bmap1->dim),
1538 bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
1541 bmap1 = add_constraints(bmap1, bmap2, 0, 0);
1543 bmap1 = isl_basic_map_simplify(bmap1);
1544 return isl_basic_map_finalize(bmap1);
1546 isl_basic_map_free(bmap1);
1547 isl_basic_map_free(bmap2);
1551 struct isl_basic_set *isl_basic_set_intersect(
1552 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
1554 return (struct isl_basic_set *)
1555 isl_basic_map_intersect(
1556 (struct isl_basic_map *)bset1,
1557 (struct isl_basic_map *)bset2);
1560 struct isl_map *isl_map_intersect(struct isl_map *map1, struct isl_map *map2)
1563 struct isl_map *result;
1569 isl_assert(map1->ctx, isl_dim_match(map1->dim, isl_dim_param,
1570 map2->dim, isl_dim_param), goto error);
1571 if (isl_dim_total(map1->dim) ==
1572 isl_dim_size(map1->dim, isl_dim_param) &&
1573 isl_dim_total(map2->dim) != isl_dim_size(map2->dim, isl_dim_param))
1574 return isl_map_intersect(map2, map1);
1576 if (isl_dim_total(map2->dim) != isl_dim_size(map2->dim, isl_dim_param))
1577 isl_assert(map1->ctx,
1578 isl_dim_equal(map1->dim, map2->dim), goto error);
1580 if (F_ISSET(map1, ISL_MAP_DISJOINT) &&
1581 F_ISSET(map2, ISL_MAP_DISJOINT))
1582 FL_SET(flags, ISL_MAP_DISJOINT);
1584 result = isl_map_alloc_dim(isl_dim_copy(map1->dim),
1585 map1->n * map2->n, flags);
1588 for (i = 0; i < map1->n; ++i)
1589 for (j = 0; j < map2->n; ++j) {
1590 struct isl_basic_map *part;
1591 part = isl_basic_map_intersect(
1592 isl_basic_map_copy(map1->p[i]),
1593 isl_basic_map_copy(map2->p[j]));
1594 if (isl_basic_map_is_empty(part))
1595 isl_basic_map_free(part);
1597 result = isl_map_add(result, part);
1610 struct isl_set *isl_set_intersect(struct isl_set *set1, struct isl_set *set2)
1612 return (struct isl_set *)
1613 isl_map_intersect((struct isl_map *)set1,
1614 (struct isl_map *)set2);
1617 struct isl_basic_map *isl_basic_map_reverse(struct isl_basic_map *bmap)
1619 struct isl_dim *dim;
1620 struct isl_basic_set *bset;
1625 bmap = isl_basic_map_cow(bmap);
1628 dim = isl_dim_reverse(isl_dim_copy(bmap->dim));
1629 in = isl_basic_map_n_in(bmap);
1630 bset = isl_basic_set_from_basic_map(bmap);
1631 bset = isl_basic_set_swap_vars(bset, in);
1632 return isl_basic_map_from_basic_set(bset, dim);
1635 /* Turn final n dimensions into existentially quantified variables.
1637 struct isl_basic_set *isl_basic_set_project_out(
1638 struct isl_basic_set *bset, unsigned n, unsigned flags)
1648 isl_assert(bset->ctx, n <= isl_basic_set_n_dim(bset), goto error);
1653 bset = isl_basic_set_cow(bset);
1655 row_size = 1 + isl_dim_total(bset->dim) + bset->extra;
1656 old = bset->block2.data;
1657 bset->block2 = isl_blk_extend(bset->ctx, bset->block2,
1658 (bset->extra + n) * (1 + row_size));
1659 if (!bset->block2.data)
1661 new_div = isl_alloc_array(ctx, isl_int *, bset->extra + n);
1664 for (i = 0; i < n; ++i) {
1665 new_div[i] = bset->block2.data +
1666 (bset->extra + i) * (1 + row_size);
1667 isl_seq_clr(new_div[i], 1 + row_size);
1669 for (i = 0; i < bset->extra; ++i)
1670 new_div[n + i] = bset->block2.data + (bset->div[i] - old);
1672 bset->div = new_div;
1675 bset->dim = isl_dim_drop_outputs(bset->dim,
1676 isl_basic_set_n_dim(bset) - n, n);
1679 bset = isl_basic_set_simplify(bset);
1680 return isl_basic_set_finalize(bset);
1682 isl_basic_set_free(bset);
1686 struct isl_basic_map *add_divs(struct isl_basic_map *bmap, unsigned n)
1690 for (i = 0; i < n; ++i) {
1691 j = isl_basic_map_alloc_div(bmap);
1694 isl_seq_clr(bmap->div[j], 1+1+isl_basic_map_total_dim(bmap));
1698 isl_basic_map_free(bmap);
1702 struct isl_basic_map *isl_basic_map_apply_range(
1703 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1705 struct isl_dim *dim_result = NULL;
1706 struct isl_basic_map *bmap;
1707 unsigned n_in, n_out, n, nparam, total, pos;
1708 struct isl_dim_map *dim_map1, *dim_map2;
1710 if (!bmap1 || !bmap2)
1713 dim_result = isl_dim_join(isl_dim_copy(bmap1->dim),
1714 isl_dim_copy(bmap2->dim));
1716 n_in = isl_basic_map_n_in(bmap1);
1717 n_out = isl_basic_map_n_out(bmap2);
1718 n = isl_basic_map_n_out(bmap1);
1719 nparam = isl_basic_map_n_param(bmap1);
1721 total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + n;
1722 dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
1723 dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
1724 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
1725 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
1726 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
1727 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_in);
1728 isl_dim_map_div(dim_map1, bmap1, pos += n_out);
1729 isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
1730 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
1731 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
1733 bmap = isl_basic_map_alloc_dim(dim_result,
1734 bmap1->n_div + bmap2->n_div + n,
1735 bmap1->n_eq + bmap2->n_eq,
1736 bmap1->n_ineq + bmap2->n_ineq);
1737 bmap = add_constraints_dim_map(bmap, bmap1, dim_map1);
1738 bmap = add_constraints_dim_map(bmap, bmap2, dim_map2);
1739 bmap = add_divs(bmap, n);
1740 bmap = isl_basic_map_simplify(bmap);
1741 return isl_basic_map_finalize(bmap);
1743 isl_basic_map_free(bmap1);
1744 isl_basic_map_free(bmap2);
1748 struct isl_basic_set *isl_basic_set_apply(
1749 struct isl_basic_set *bset, struct isl_basic_map *bmap)
1754 isl_assert(set->ctx, isl_basic_map_compatible_domain(bmap, bset),
1757 return (struct isl_basic_set *)
1758 isl_basic_map_apply_range((struct isl_basic_map *)bset, bmap);
1760 isl_basic_set_free(bset);
1761 isl_basic_map_free(bmap);
1765 struct isl_basic_map *isl_basic_map_apply_domain(
1766 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1768 if (!bmap1 || !bmap2)
1772 isl_basic_map_n_in(bmap1) == isl_basic_map_n_in(bmap2), goto error);
1774 isl_basic_map_n_param(bmap1) == isl_basic_map_n_param(bmap2),
1777 bmap1 = isl_basic_map_reverse(bmap1);
1778 bmap1 = isl_basic_map_apply_range(bmap1, bmap2);
1779 return isl_basic_map_reverse(bmap1);
1781 isl_basic_map_free(bmap1);
1782 isl_basic_map_free(bmap2);
1786 /* Given two basic maps A -> f(A) and B -> g(B), construct a basic map
1787 * A \cap B -> f(A) + f(B)
1789 struct isl_basic_map *isl_basic_map_sum(
1790 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
1792 unsigned n_in, n_out, nparam, total, pos;
1793 struct isl_basic_map *bmap = NULL;
1794 struct isl_dim_map *dim_map1, *dim_map2;
1797 if (!bmap1 || !bmap2)
1800 isl_assert(bmap1->ctx, isl_dim_equal(bmap1->dim, bmap2->dim),
1803 nparam = isl_basic_map_n_param(bmap1);
1804 n_in = isl_basic_map_n_in(bmap1);
1805 n_out = isl_basic_map_n_out(bmap1);
1807 total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + 2 * n_out;
1808 dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
1809 dim_map2 = isl_dim_map_alloc(bmap2->ctx, total);
1810 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
1811 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos);
1812 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
1813 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
1814 isl_dim_map_div(dim_map1, bmap1, pos += n_in + n_out);
1815 isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
1816 isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
1817 isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_out);
1819 bmap = isl_basic_map_alloc_dim(isl_dim_copy(bmap1->dim),
1820 bmap1->n_div + bmap2->n_div + 2 * n_out,
1821 bmap1->n_eq + bmap2->n_eq + n_out,
1822 bmap1->n_ineq + bmap2->n_ineq);
1823 for (i = 0; i < n_out; ++i) {
1824 int j = isl_basic_map_alloc_equality(bmap);
1827 isl_seq_clr(bmap->eq[j], 1+total);
1828 isl_int_set_si(bmap->eq[j][1+nparam+n_in+i], -1);
1829 isl_int_set_si(bmap->eq[j][1+pos+i], 1);
1830 isl_int_set_si(bmap->eq[j][1+pos-n_out+i], 1);
1832 bmap = add_constraints_dim_map(bmap, bmap1, dim_map1);
1833 bmap = add_constraints_dim_map(bmap, bmap2, dim_map2);
1834 bmap = add_divs(bmap, 2 * n_out);
1836 bmap = isl_basic_map_simplify(bmap);
1837 return isl_basic_map_finalize(bmap);
1839 isl_basic_map_free(bmap);
1840 isl_basic_map_free(bmap1);
1841 isl_basic_map_free(bmap2);
1845 /* Given a basic map A -> f(A), construct A -> -f(A).
1847 struct isl_basic_map *isl_basic_map_neg(struct isl_basic_map *bmap)
1852 bmap = isl_basic_map_cow(bmap);
1856 n = isl_basic_map_dim(bmap, isl_dim_out);
1857 off = basic_map_offset(bmap, isl_dim_out);
1858 for (i = 0; i < bmap->n_eq; ++i)
1859 for (j = 0; j < n; ++j)
1860 isl_int_neg(bmap->eq[i][off+j], bmap->eq[i][off+j]);
1861 for (i = 0; i < bmap->n_ineq; ++i)
1862 for (j = 0; j < n; ++j)
1863 isl_int_neg(bmap->ineq[i][off+j], bmap->ineq[i][off+j]);
1864 for (i = 0; i < bmap->n_div; ++i)
1865 for (j = 0; j < n; ++j)
1866 isl_int_neg(bmap->div[i][1+off+j], bmap->div[i][1+off+j]);
1867 return isl_basic_map_finalize(bmap);
1870 /* Given a basic map A -> f(A) and an integer d, construct a basic map
1871 * A -> floor(f(A)/d).
1873 struct isl_basic_map *isl_basic_map_floordiv(struct isl_basic_map *bmap,
1876 unsigned n_in, n_out, nparam, total, pos;
1877 struct isl_basic_map *result = NULL;
1878 struct isl_dim_map *dim_map;
1884 nparam = isl_basic_map_n_param(bmap);
1885 n_in = isl_basic_map_n_in(bmap);
1886 n_out = isl_basic_map_n_out(bmap);
1888 total = nparam + n_in + n_out + bmap->n_div + n_out;
1889 dim_map = isl_dim_map_alloc(bmap->ctx, total);
1890 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_param, pos = 0);
1891 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_in, pos += nparam);
1892 isl_dim_map_div(dim_map, bmap, pos += n_in + n_out);
1893 isl_dim_map_dim(dim_map, bmap->dim, isl_dim_out, pos += bmap->n_div);
1895 result = isl_basic_map_alloc_dim(isl_dim_copy(bmap->dim),
1896 bmap->n_div + n_out,
1897 bmap->n_eq, bmap->n_ineq + 2 * n_out);
1898 result = add_constraints_dim_map(result, bmap, dim_map);
1899 result = add_divs(result, n_out);
1900 for (i = 0; i < n_out; ++i) {
1902 j = isl_basic_map_alloc_inequality(result);
1905 isl_seq_clr(result->ineq[j], 1+total);
1906 isl_int_neg(result->ineq[j][1+nparam+n_in+i], d);
1907 isl_int_set_si(result->ineq[j][1+pos+i], 1);
1908 j = isl_basic_map_alloc_inequality(result);
1911 isl_seq_clr(result->ineq[j], 1+total);
1912 isl_int_set(result->ineq[j][1+nparam+n_in+i], d);
1913 isl_int_set_si(result->ineq[j][1+pos+i], -1);
1914 isl_int_sub_ui(result->ineq[j][0], d, 1);
1917 result = isl_basic_map_simplify(result);
1918 return isl_basic_map_finalize(result);
1920 isl_basic_map_free(result);
1924 static struct isl_basic_map *var_equal(struct isl_ctx *ctx,
1925 struct isl_basic_map *bmap, unsigned pos)
1931 i = isl_basic_map_alloc_equality(bmap);
1934 nparam = isl_basic_map_n_param(bmap);
1935 n_in = isl_basic_map_n_in(bmap);
1936 isl_seq_clr(bmap->eq[i], 1 + isl_basic_map_total_dim(bmap));
1937 isl_int_set_si(bmap->eq[i][1+nparam+pos], -1);
1938 isl_int_set_si(bmap->eq[i][1+nparam+n_in+pos], 1);
1939 return isl_basic_map_finalize(bmap);
1941 isl_basic_map_free(bmap);
1945 static struct isl_basic_map *var_more(struct isl_ctx *ctx,
1946 struct isl_basic_map *bmap, unsigned pos)
1952 i = isl_basic_map_alloc_inequality(bmap);
1955 nparam = isl_basic_map_n_param(bmap);
1956 n_in = isl_basic_map_n_in(bmap);
1957 isl_seq_clr(bmap->ineq[i], 1 + isl_basic_map_total_dim(bmap));
1958 isl_int_set_si(bmap->ineq[i][0], -1);
1959 isl_int_set_si(bmap->ineq[i][1+nparam+pos], -1);
1960 isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], 1);
1961 return isl_basic_map_finalize(bmap);
1963 isl_basic_map_free(bmap);
1967 static struct isl_basic_map *var_less(struct isl_ctx *ctx,
1968 struct isl_basic_map *bmap, unsigned pos)
1974 i = isl_basic_map_alloc_inequality(bmap);
1977 nparam = isl_basic_map_n_param(bmap);
1978 n_in = isl_basic_map_n_in(bmap);
1979 isl_seq_clr(bmap->ineq[i], 1 + isl_basic_map_total_dim(bmap));
1980 isl_int_set_si(bmap->ineq[i][0], -1);
1981 isl_int_set_si(bmap->ineq[i][1+nparam+pos], 1);
1982 isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], -1);
1983 return isl_basic_map_finalize(bmap);
1985 isl_basic_map_free(bmap);
1989 struct isl_basic_map *isl_basic_map_equal(struct isl_ctx *ctx,
1990 unsigned nparam, unsigned in, unsigned out, unsigned n_equal)
1993 struct isl_basic_map *bmap;
1994 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, n_equal, 0);
1997 for (i = 0; i < n_equal && bmap; ++i)
1998 bmap = var_equal(ctx, bmap, i);
1999 return isl_basic_map_finalize(bmap);
2002 struct isl_basic_map *isl_basic_map_less_at(struct isl_ctx *ctx,
2003 unsigned nparam, unsigned in, unsigned out, unsigned pos)
2006 struct isl_basic_map *bmap;
2007 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, pos, 1);
2010 for (i = 0; i < pos && bmap; ++i)
2011 bmap = var_equal(ctx, bmap, i);
2013 bmap = var_less(ctx, bmap, pos);
2014 return isl_basic_map_finalize(bmap);
2017 struct isl_basic_map *isl_basic_map_more_at(struct isl_ctx *ctx,
2018 unsigned nparam, unsigned in, unsigned out, unsigned pos)
2021 struct isl_basic_map *bmap;
2022 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, pos, 1);
2025 for (i = 0; i < pos && bmap; ++i)
2026 bmap = var_equal(ctx, bmap, i);
2028 bmap = var_more(ctx, bmap, pos);
2029 return isl_basic_map_finalize(bmap);
2032 struct isl_basic_map *isl_basic_map_from_basic_set(
2033 struct isl_basic_set *bset, struct isl_dim *dim)
2035 struct isl_basic_map *bmap;
2037 bset = isl_basic_set_cow(bset);
2041 isl_assert(bset->ctx, isl_dim_compatible(bset->dim, dim), goto error);
2042 isl_dim_free(bset->dim);
2043 bmap = (struct isl_basic_map *) bset;
2045 return isl_basic_map_finalize(bmap);
2047 isl_basic_set_free(bset);
2052 struct isl_basic_set *isl_basic_set_from_basic_map(struct isl_basic_map *bmap)
2056 if (bmap->dim->n_in == 0)
2057 return (struct isl_basic_set *)bmap;
2058 bmap = isl_basic_map_cow(bmap);
2061 bmap->dim = isl_dim_cow(bmap->dim);
2064 bmap->dim->n_out += bmap->dim->n_in;
2065 bmap->dim->n_in = 0;
2066 bmap = isl_basic_map_finalize(bmap);
2067 return (struct isl_basic_set *)bmap;
2069 isl_basic_map_free(bmap);
2073 /* For a div d = floor(f/m), add the constraints
2076 * -(f-(n-1)) + m d >= 0
2078 * Note that the second constraint is the negation of
2082 static int add_div_constraints(struct isl_basic_map *bmap, unsigned div)
2085 unsigned total = isl_basic_map_total_dim(bmap);
2086 unsigned div_pos = 1 + total - bmap->n_div + div;
2088 i = isl_basic_map_alloc_inequality(bmap);
2091 isl_seq_cpy(bmap->ineq[i], bmap->div[div]+1, 1+total);
2092 isl_int_neg(bmap->ineq[i][div_pos], bmap->div[div][0]);
2094 j = isl_basic_map_alloc_inequality(bmap);
2097 isl_seq_neg(bmap->ineq[j], bmap->ineq[i], 1 + total);
2098 isl_int_add(bmap->ineq[j][0], bmap->ineq[j][0], bmap->ineq[j][div_pos]);
2099 isl_int_sub_ui(bmap->ineq[j][0], bmap->ineq[j][0], 1);
2103 struct isl_basic_set *isl_basic_map_underlying_set(
2104 struct isl_basic_map *bmap)
2108 if (bmap->dim->nparam == 0 && bmap->dim->n_in == 0 && bmap->n_div == 0)
2109 return (struct isl_basic_set *)bmap;
2110 bmap = isl_basic_map_cow(bmap);
2113 bmap->dim = isl_dim_underlying(bmap->dim, bmap->n_div);
2116 bmap->extra -= bmap->n_div;
2118 bmap = isl_basic_map_finalize(bmap);
2119 return (struct isl_basic_set *)bmap;
2124 struct isl_basic_map *isl_basic_map_overlying_set(
2125 struct isl_basic_set *bset, struct isl_basic_map *like)
2127 struct isl_basic_map *bmap;
2128 struct isl_ctx *ctx;
2135 isl_assert(ctx, bset->n_div == 0, goto error);
2136 isl_assert(ctx, isl_basic_set_n_param(bset) == 0, goto error);
2137 isl_assert(ctx, bset->dim->n_out == isl_basic_map_total_dim(like),
2139 if (isl_dim_equal(bset->dim, like->dim) && like->n_div == 0) {
2140 isl_basic_map_free(like);
2141 return (struct isl_basic_map *)bset;
2143 bset = isl_basic_set_cow(bset);
2146 total = bset->dim->n_out + bset->extra;
2147 bmap = (struct isl_basic_map *)bset;
2148 isl_dim_free(bmap->dim);
2149 bmap->dim = isl_dim_copy(like->dim);
2152 bmap->n_div = like->n_div;
2153 bmap->extra += like->n_div;
2156 ltotal = total - bmap->extra + like->extra;
2159 bmap->block2 = isl_blk_extend(ctx, bmap->block2,
2160 bmap->extra * (1 + 1 + total));
2161 if (isl_blk_is_error(bmap->block2))
2163 bmap->div = isl_realloc_array(ctx, bmap->div, isl_int *,
2167 for (i = 0; i < bmap->extra; ++i)
2168 bmap->div[i] = bmap->block2.data + i * (1 + 1 + total);
2169 for (i = 0; i < like->n_div; ++i) {
2170 isl_seq_cpy(bmap->div[i], like->div[i], 1 + 1 + ltotal);
2171 isl_seq_clr(bmap->div[i]+1+1+ltotal, total - ltotal);
2173 bmap = isl_basic_map_extend_constraints(bmap,
2174 0, 2 * like->n_div);
2175 for (i = 0; i < like->n_div; ++i) {
2176 if (isl_int_is_zero(bmap->div[i][0]))
2178 if (add_div_constraints(bmap, i) < 0)
2182 isl_basic_map_free(like);
2183 bmap = isl_basic_map_simplify(bmap);
2184 bmap = isl_basic_map_finalize(bmap);
2187 isl_basic_map_free(like);
2188 isl_basic_set_free(bset);
2192 struct isl_basic_set *isl_basic_set_from_underlying_set(
2193 struct isl_basic_set *bset, struct isl_basic_set *like)
2195 return (struct isl_basic_set *)
2196 isl_basic_map_overlying_set(bset, (struct isl_basic_map *)like);
2199 struct isl_set *isl_set_from_underlying_set(
2200 struct isl_set *set, struct isl_basic_set *like)
2206 isl_assert(set->ctx, set->dim->n_out == isl_basic_set_total_dim(like),
2208 if (isl_dim_equal(set->dim, like->dim) && like->n_div == 0) {
2209 isl_basic_set_free(like);
2212 set = isl_set_cow(set);
2215 for (i = 0; i < set->n; ++i) {
2216 set->p[i] = isl_basic_set_from_underlying_set(set->p[i],
2217 isl_basic_set_copy(like));
2221 isl_dim_free(set->dim);
2222 set->dim = isl_dim_copy(like->dim);
2225 isl_basic_set_free(like);
2228 isl_basic_set_free(like);
2233 struct isl_set *isl_map_underlying_set(struct isl_map *map)
2237 map = isl_map_cow(map);
2240 map->dim = isl_dim_cow(map->dim);
2244 for (i = 1; i < map->n; ++i)
2245 isl_assert(map->ctx, map->p[0]->n_div == map->p[i]->n_div,
2247 for (i = 0; i < map->n; ++i) {
2248 map->p[i] = (struct isl_basic_map *)
2249 isl_basic_map_underlying_set(map->p[i]);
2254 map->dim = isl_dim_underlying(map->dim, 0);
2256 isl_dim_free(map->dim);
2257 map->dim = isl_dim_copy(map->p[0]->dim);
2261 return (struct isl_set *)map;
2267 struct isl_set *isl_set_to_underlying_set(struct isl_set *set)
2269 return (struct isl_set *)isl_map_underlying_set((struct isl_map *)set);
2272 struct isl_basic_set *isl_basic_map_domain(struct isl_basic_map *bmap)
2274 struct isl_basic_set *domain;
2278 n_out = isl_basic_map_n_out(bmap);
2279 domain = isl_basic_set_from_basic_map(bmap);
2280 return isl_basic_set_project_out(domain, n_out, 0);
2283 struct isl_basic_set *isl_basic_map_range(struct isl_basic_map *bmap)
2285 return isl_basic_map_domain(isl_basic_map_reverse(bmap));
2288 struct isl_set *isl_map_range(struct isl_map *map)
2291 struct isl_set *set;
2295 map = isl_map_cow(map);
2299 set = (struct isl_set *) map;
2300 if (set->dim->n_in != 0) {
2301 set->dim = isl_dim_drop_inputs(set->dim, 0, set->dim->n_in);
2305 for (i = 0; i < map->n; ++i) {
2306 set->p[i] = isl_basic_map_range(map->p[i]);
2310 F_CLR(set, ISL_MAP_DISJOINT);
2311 F_CLR(set, ISL_SET_NORMALIZED);
2318 struct isl_map *isl_map_from_set(struct isl_set *set, struct isl_dim *dim)
2321 struct isl_map *map = NULL;
2323 set = isl_set_cow(set);
2326 isl_assert(set->ctx, isl_dim_compatible(set->dim, dim), goto error);
2327 map = (struct isl_map *)set;
2328 for (i = 0; i < set->n; ++i) {
2329 map->p[i] = isl_basic_map_from_basic_set(
2330 set->p[i], isl_dim_copy(dim));
2334 isl_dim_free(map->dim);
2343 struct isl_set *isl_set_from_map(struct isl_map *map)
2346 struct isl_set *set = NULL;
2350 map = isl_map_cow(map);
2353 map->dim = isl_dim_cow(map->dim);
2356 map->dim->n_out += map->dim->n_in;
2358 set = (struct isl_set *)map;
2359 for (i = 0; i < map->n; ++i) {
2360 set->p[i] = isl_basic_set_from_basic_map(map->p[i]);
2370 struct isl_map *isl_map_alloc_dim(struct isl_dim *dim, int n, unsigned flags)
2372 struct isl_map *map;
2376 isl_assert(dim->ctx, n >= 0, return NULL);
2377 map = isl_alloc(dim->ctx, struct isl_map,
2378 sizeof(struct isl_map) +
2379 n * sizeof(struct isl_basic_map *));
2383 map->ctx = dim->ctx;
2384 isl_ctx_ref(map->ctx);
2396 struct isl_map *isl_map_alloc(struct isl_ctx *ctx,
2397 unsigned nparam, unsigned in, unsigned out, int n,
2400 struct isl_map *map;
2401 struct isl_dim *dims;
2403 dims = isl_dim_alloc(ctx, nparam, in, out);
2407 map = isl_map_alloc_dim(dims, n, flags);
2411 struct isl_basic_map *isl_basic_map_empty(struct isl_ctx *ctx,
2412 unsigned nparam, unsigned in, unsigned out)
2414 struct isl_basic_map *bmap;
2415 bmap = isl_basic_map_alloc(ctx, nparam, in, out, 0, 1, 0);
2416 bmap = isl_basic_map_set_to_empty(bmap);
2420 struct isl_basic_set *isl_basic_set_empty(struct isl_dim *dim)
2422 struct isl_basic_set *bset;
2423 bset = isl_basic_set_alloc_dim(dim, 0, 1, 0);
2424 bset = isl_basic_set_set_to_empty(bset);
2428 struct isl_basic_map *isl_basic_map_empty_like(struct isl_basic_map *model)
2430 struct isl_basic_map *bmap;
2433 bmap = isl_basic_map_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2434 bmap = isl_basic_map_set_to_empty(bmap);
2438 struct isl_basic_map *isl_basic_map_empty_like_map(struct isl_map *model)
2440 struct isl_basic_map *bmap;
2443 bmap = isl_basic_map_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2444 bmap = isl_basic_map_set_to_empty(bmap);
2448 struct isl_basic_set *isl_basic_set_empty_like(struct isl_basic_set *model)
2450 struct isl_basic_set *bset;
2453 bset = isl_basic_set_alloc_dim(isl_dim_copy(model->dim), 0, 1, 0);
2454 bset = isl_basic_set_set_to_empty(bset);
2458 struct isl_basic_map *isl_basic_map_universe(struct isl_dim *dim)
2460 struct isl_basic_map *bmap;
2461 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
2465 struct isl_basic_set *isl_basic_set_universe(struct isl_dim *dim)
2467 struct isl_basic_set *bset;
2468 bset = isl_basic_set_alloc_dim(dim, 0, 0, 0);
2472 struct isl_basic_set *isl_basic_set_universe_like(struct isl_basic_set *model)
2476 return isl_basic_set_alloc_dim(isl_dim_copy(model->dim), 0, 0, 0);
2479 struct isl_map *isl_map_empty(struct isl_ctx *ctx,
2480 unsigned nparam, unsigned in, unsigned out)
2482 return isl_map_alloc(ctx, nparam, in, out, 0, ISL_MAP_DISJOINT);
2485 struct isl_map *isl_map_empty_like_basic_map(struct isl_basic_map *model)
2489 return isl_map_alloc_dim(isl_dim_copy(model->dim), 0, ISL_MAP_DISJOINT);
2492 struct isl_set *isl_set_empty(struct isl_dim *dim)
2494 return isl_set_alloc_dim(dim, 0, ISL_MAP_DISJOINT);
2497 struct isl_set *isl_set_empty_like(struct isl_set *model)
2501 return isl_set_empty(isl_dim_copy(model->dim));
2504 struct isl_set *isl_set_universe(struct isl_dim *dim)
2506 struct isl_set *set;
2509 set = isl_set_alloc_dim(isl_dim_copy(dim), 1, ISL_MAP_DISJOINT);
2510 set = isl_set_add(set, isl_basic_set_universe(dim));
2514 struct isl_map *isl_map_dup(struct isl_map *map)
2517 struct isl_map *dup;
2521 dup = isl_map_alloc_dim(isl_dim_copy(map->dim), map->n, map->flags);
2522 for (i = 0; i < map->n; ++i)
2523 dup = isl_map_add(dup, isl_basic_map_copy(map->p[i]));
2527 struct isl_map *isl_map_add(struct isl_map *map, struct isl_basic_map *bmap)
2531 isl_assert(map->ctx, isl_dim_equal(map->dim, bmap->dim), goto error);
2532 isl_assert(map->ctx, map->n < map->size, goto error);
2533 map->p[map->n] = bmap;
2535 F_CLR(map, ISL_MAP_NORMALIZED);
2541 isl_basic_map_free(bmap);
2545 void isl_map_free(struct isl_map *map)
2555 isl_ctx_deref(map->ctx);
2556 for (i = 0; i < map->n; ++i)
2557 isl_basic_map_free(map->p[i]);
2558 isl_dim_free(map->dim);
2562 struct isl_map *isl_map_extend(struct isl_map *base,
2563 unsigned nparam, unsigned n_in, unsigned n_out)
2567 base = isl_map_cow(base);
2571 base->dim = isl_dim_extend(base->dim, nparam, n_in, n_out);
2574 for (i = 0; i < base->n; ++i) {
2575 base->p[i] = isl_basic_map_extend_dim(base->p[i],
2576 isl_dim_copy(base->dim), 0, 0, 0);
2586 struct isl_set *isl_set_extend(struct isl_set *base,
2587 unsigned nparam, unsigned dim)
2589 return (struct isl_set *)isl_map_extend((struct isl_map *)base,
2593 static struct isl_basic_map *isl_basic_map_fix_pos(struct isl_basic_map *bmap,
2594 unsigned pos, int value)
2598 bmap = isl_basic_map_cow(bmap);
2599 bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
2600 j = isl_basic_map_alloc_equality(bmap);
2603 isl_seq_clr(bmap->eq[j], 1 + isl_basic_map_total_dim(bmap));
2604 isl_int_set_si(bmap->eq[j][pos], -1);
2605 isl_int_set_si(bmap->eq[j][0], value);
2606 bmap = isl_basic_map_simplify(bmap);
2607 return isl_basic_map_finalize(bmap);
2609 isl_basic_map_free(bmap);
2613 struct isl_basic_map *isl_basic_map_fix_si(struct isl_basic_map *bmap,
2614 enum isl_dim_type type, unsigned pos, int value)
2618 isl_assert(bmap->ctx, pos < isl_basic_map_dim(bmap, type), goto error);
2619 return isl_basic_map_fix_pos(bmap, basic_map_offset(bmap, type) + pos,
2622 isl_basic_map_free(bmap);
2626 struct isl_basic_set *isl_basic_set_fix_si(struct isl_basic_set *bset,
2627 enum isl_dim_type type, unsigned pos, int value)
2629 return (struct isl_basic_set *)
2630 isl_basic_map_fix_si((struct isl_basic_map *)bset,
2634 struct isl_basic_map *isl_basic_map_fix_input_si(struct isl_basic_map *bmap,
2635 unsigned input, int value)
2637 return isl_basic_map_fix_si(bmap, isl_dim_in, input, value);
2640 struct isl_basic_set *isl_basic_set_fix_dim_si(struct isl_basic_set *bset,
2641 unsigned dim, int value)
2643 return (struct isl_basic_set *)
2644 isl_basic_map_fix_si((struct isl_basic_map *)bset,
2645 isl_dim_set, dim, value);
2648 struct isl_map *isl_map_fix_si(struct isl_map *map,
2649 enum isl_dim_type type, unsigned pos, int value)
2653 map = isl_map_cow(map);
2657 isl_assert(ctx, pos < isl_map_dim(map, type), goto error);
2658 for (i = 0; i < map->n; ++i) {
2659 map->p[i] = isl_basic_map_fix_si(map->p[i], type, pos, value);
2663 F_CLR(map, ISL_MAP_NORMALIZED);
2670 struct isl_map *isl_map_fix_input_si(struct isl_map *map,
2671 unsigned input, int value)
2673 return isl_map_fix_si(map, isl_dim_in, input, value);
2676 struct isl_set *isl_set_fix_dim_si(struct isl_set *set, unsigned dim, int value)
2678 return (struct isl_set *)
2679 isl_map_fix_si((struct isl_map *)set, isl_dim_set, dim, value);
2682 struct isl_basic_set *isl_basic_set_lower_bound_dim(struct isl_basic_set *bset,
2683 unsigned dim, isl_int value)
2688 bset = isl_basic_set_cow(bset);
2689 bset = isl_basic_set_extend_constraints(bset, 0, 1);
2690 j = isl_basic_set_alloc_inequality(bset);
2693 isl_seq_clr(bset->ineq[j], 1 + isl_basic_set_total_dim(bset));
2694 isl_int_set_si(bset->ineq[j][1 + isl_basic_set_n_param(bset) + dim], 1);
2695 isl_int_neg(bset->ineq[j][0], value);
2696 bset = isl_basic_set_simplify(bset);
2697 return isl_basic_set_finalize(bset);
2699 isl_basic_set_free(bset);
2703 struct isl_set *isl_set_lower_bound_dim(struct isl_set *set, unsigned dim,
2708 set = isl_set_cow(set);
2712 isl_assert(set->ctx, dim < isl_set_n_dim(set), goto error);
2713 for (i = 0; i < set->n; ++i) {
2714 set->p[i] = isl_basic_set_lower_bound_dim(set->p[i], dim, value);
2724 struct isl_map *isl_map_reverse(struct isl_map *map)
2729 map = isl_map_cow(map);
2733 map->dim = isl_dim_reverse(map->dim);
2736 for (i = 0; i < map->n; ++i) {
2737 map->p[i] = isl_basic_map_reverse(map->p[i]);
2741 F_CLR(map, ISL_MAP_NORMALIZED);
2748 struct isl_map *isl_basic_map_lexmax(
2749 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2750 struct isl_set **empty)
2752 return isl_pip_basic_map_lexmax(bmap, dom, empty);
2755 struct isl_map *isl_basic_map_lexmin(
2756 struct isl_basic_map *bmap, struct isl_basic_set *dom,
2757 struct isl_set **empty)
2759 return isl_pip_basic_map_lexmin(bmap, dom, empty);
2762 struct isl_set *isl_basic_set_lexmin(struct isl_basic_set *bset)
2764 struct isl_basic_map *bmap = NULL;
2765 struct isl_basic_set *dom = NULL;
2766 struct isl_map *min;
2767 struct isl_dim *param_dim;
2771 bmap = isl_basic_map_from_basic_set(bset, isl_dim_copy(bset->dim));
2774 param_dim = isl_dim_domain(isl_dim_copy(bmap->dim));
2775 dom = isl_basic_set_universe(param_dim);
2778 min = isl_basic_map_lexmin(bmap, dom, NULL);
2779 return isl_map_range(min);
2781 isl_basic_map_free(bmap);
2785 struct isl_map *isl_basic_map_compute_divs(struct isl_basic_map *bmap)
2792 off = isl_dim_total(bmap->dim);
2793 for (i = 0; i < bmap->n_div; ++i) {
2794 if (isl_int_is_zero(bmap->div[i][0]))
2795 return isl_pip_basic_map_compute_divs(bmap);
2796 isl_assert(bmap->ctx, isl_int_is_zero(bmap->div[i][1+1+off+i]),
2799 return isl_map_from_basic_map(bmap);
2801 isl_basic_map_free(bmap);
2805 struct isl_map *isl_map_compute_divs(struct isl_map *map)
2808 struct isl_map *res;
2814 res = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[0]));
2815 for (i = 1 ; i < map->n; ++i) {
2817 r2 = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[i]));
2818 if (F_ISSET(map, ISL_MAP_DISJOINT))
2819 res = isl_map_union_disjoint(res, r2);
2821 res = isl_map_union(res, r2);
2828 struct isl_set *isl_basic_set_compute_divs(struct isl_basic_set *bset)
2830 return (struct isl_set *)
2831 isl_basic_map_compute_divs((struct isl_basic_map *)bset);
2834 struct isl_set *isl_set_compute_divs(struct isl_set *set)
2836 return (struct isl_set *)
2837 isl_map_compute_divs((struct isl_map *)set);
2840 struct isl_set *isl_map_domain(struct isl_map *map)
2843 struct isl_set *set;
2848 map = isl_map_cow(map);
2852 set = (struct isl_set *)map;
2853 set->dim = isl_dim_domain(set->dim);
2856 for (i = 0; i < map->n; ++i) {
2857 set->p[i] = isl_basic_map_domain(map->p[i]);
2861 F_CLR(set, ISL_MAP_DISJOINT);
2862 F_CLR(set, ISL_SET_NORMALIZED);
2869 struct isl_map *isl_map_union_disjoint(
2870 struct isl_map *map1, struct isl_map *map2)
2874 struct isl_map *map = NULL;
2888 isl_assert(map1->ctx, isl_dim_equal(map1->dim, map2->dim), goto error);
2890 if (F_ISSET(map1, ISL_MAP_DISJOINT) &&
2891 F_ISSET(map2, ISL_MAP_DISJOINT))
2892 FL_SET(flags, ISL_MAP_DISJOINT);
2894 map = isl_map_alloc_dim(isl_dim_copy(map1->dim),
2895 map1->n + map2->n, flags);
2898 for (i = 0; i < map1->n; ++i) {
2899 map = isl_map_add(map,
2900 isl_basic_map_copy(map1->p[i]));
2904 for (i = 0; i < map2->n; ++i) {
2905 map = isl_map_add(map,
2906 isl_basic_map_copy(map2->p[i]));
2920 struct isl_map *isl_map_union(struct isl_map *map1, struct isl_map *map2)
2922 map1 = isl_map_union_disjoint(map1, map2);
2926 F_CLR(map1, ISL_MAP_DISJOINT);
2930 struct isl_set *isl_set_union_disjoint(
2931 struct isl_set *set1, struct isl_set *set2)
2933 return (struct isl_set *)
2934 isl_map_union_disjoint(
2935 (struct isl_map *)set1, (struct isl_map *)set2);
2938 struct isl_set *isl_set_union(struct isl_set *set1, struct isl_set *set2)
2940 return (struct isl_set *)
2941 isl_map_union((struct isl_map *)set1, (struct isl_map *)set2);
2944 struct isl_map *isl_map_intersect_range(
2945 struct isl_map *map, struct isl_set *set)
2948 struct isl_map *result;
2954 if (F_ISSET(map, ISL_MAP_DISJOINT) &&
2955 F_ISSET(set, ISL_MAP_DISJOINT))
2956 FL_SET(flags, ISL_MAP_DISJOINT);
2958 result = isl_map_alloc_dim(isl_dim_copy(map->dim),
2959 map->n * set->n, flags);
2962 for (i = 0; i < map->n; ++i)
2963 for (j = 0; j < set->n; ++j) {
2964 result = isl_map_add(result,
2965 isl_basic_map_intersect_range(
2966 isl_basic_map_copy(map->p[i]),
2967 isl_basic_set_copy(set->p[j])));
2980 struct isl_map *isl_map_intersect_domain(
2981 struct isl_map *map, struct isl_set *set)
2983 return isl_map_reverse(
2984 isl_map_intersect_range(isl_map_reverse(map), set));
2987 struct isl_map *isl_map_apply_domain(
2988 struct isl_map *map1, struct isl_map *map2)
2992 map1 = isl_map_reverse(map1);
2993 map1 = isl_map_apply_range(map1, map2);
2994 return isl_map_reverse(map1);
3001 struct isl_map *isl_map_apply_range(
3002 struct isl_map *map1, struct isl_map *map2)
3004 struct isl_dim *dim_result;
3005 struct isl_map *result;
3014 dim_result = isl_dim_join(isl_dim_copy(map1->dim),
3015 isl_dim_copy(map2->dim));
3017 result = isl_map_alloc_dim(dim_result, map1->n * map2->n, 0);
3020 for (i = 0; i < map1->n; ++i)
3021 for (j = 0; j < map2->n; ++j) {
3022 result = isl_map_add(result,
3023 isl_basic_map_apply_range(
3024 isl_basic_map_copy(map1->p[i]),
3025 isl_basic_map_copy(map2->p[j])));
3031 if (result && result->n <= 1)
3032 F_SET(result, ISL_MAP_DISJOINT);
3041 * returns range - domain
3043 struct isl_basic_set *isl_basic_map_deltas(struct isl_basic_map *bmap)
3045 struct isl_basic_set *bset;
3052 dim = isl_basic_map_n_in(bmap);
3053 nparam = isl_basic_map_n_param(bmap);
3054 isl_assert(bmap->ctx, dim == isl_basic_map_n_out(bmap), goto error);
3055 bset = isl_basic_set_from_basic_map(bmap);
3056 bset = isl_basic_set_extend(bset, nparam, 3*dim, 0, dim, 0);
3057 bset = isl_basic_set_swap_vars(bset, 2*dim);
3058 for (i = 0; i < dim; ++i) {
3059 int j = isl_basic_map_alloc_equality(
3060 (struct isl_basic_map *)bset);
3063 isl_seq_clr(bset->eq[j], 1 + isl_basic_set_total_dim(bset));
3064 isl_int_set_si(bset->eq[j][1+nparam+i], 1);
3065 isl_int_set_si(bset->eq[j][1+nparam+dim+i], 1);
3066 isl_int_set_si(bset->eq[j][1+nparam+2*dim+i], -1);
3068 return isl_basic_set_project_out(bset, 2*dim, 0);
3070 isl_basic_map_free(bmap);
3075 * returns range - domain
3077 struct isl_set *isl_map_deltas(struct isl_map *map)
3080 struct isl_set *result;
3085 isl_assert(map->ctx, isl_map_n_in(map) == isl_map_n_out(map), goto error);
3086 result = isl_set_alloc(map->ctx, isl_map_n_param(map),
3087 isl_map_n_in(map), map->n, map->flags);
3090 for (i = 0; i < map->n; ++i)
3091 result = isl_set_add(result,
3092 isl_basic_map_deltas(isl_basic_map_copy(map->p[i])));
3100 static struct isl_basic_map *basic_map_identity(struct isl_dim *dims)
3102 struct isl_basic_map *bmap;
3110 nparam = dims->nparam;
3112 bmap = isl_basic_map_alloc_dim(dims, 0, dim, 0);
3116 for (i = 0; i < dim; ++i) {
3117 int j = isl_basic_map_alloc_equality(bmap);
3120 isl_seq_clr(bmap->eq[j], 1 + isl_basic_map_total_dim(bmap));
3121 isl_int_set_si(bmap->eq[j][1+nparam+i], 1);
3122 isl_int_set_si(bmap->eq[j][1+nparam+dim+i], -1);
3124 return isl_basic_map_finalize(bmap);
3126 isl_basic_map_free(bmap);
3130 struct isl_basic_map *isl_basic_map_identity(struct isl_dim *set_dim)
3132 struct isl_dim *dim = isl_dim_map(set_dim);
3135 return basic_map_identity(dim);
3138 struct isl_basic_map *isl_basic_map_identity_like(struct isl_basic_map *model)
3140 if (!model || !model->dim)
3142 isl_assert(model->ctx,
3143 model->dim->n_in == model->dim->n_out, return NULL);
3144 return basic_map_identity(isl_dim_copy(model->dim));
3147 static struct isl_map *map_identity(struct isl_dim *dim)
3149 struct isl_map *map = isl_map_alloc_dim(dim, 1, ISL_MAP_DISJOINT);
3150 return isl_map_add(map, basic_map_identity(isl_dim_copy(dim)));
3153 struct isl_map *isl_map_identity(struct isl_dim *set_dim)
3155 struct isl_dim *dim = isl_dim_map(set_dim);
3158 return map_identity(dim);
3161 struct isl_map *isl_map_identity_like(struct isl_basic_map *model)
3163 if (!model || !model->dim)
3165 isl_assert(model->ctx,
3166 model->dim->n_in == model->dim->n_out, return NULL);
3167 return map_identity(isl_dim_copy(model->dim));
3170 int isl_set_is_equal(struct isl_set *set1, struct isl_set *set2)
3172 return isl_map_is_equal((struct isl_map *)set1, (struct isl_map *)set2);
3175 int isl_set_is_subset(struct isl_set *set1, struct isl_set *set2)
3177 return isl_map_is_subset(
3178 (struct isl_map *)set1, (struct isl_map *)set2);
3181 int isl_basic_map_is_subset(
3182 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3185 struct isl_map *map1;
3186 struct isl_map *map2;
3188 if (!bmap1 || !bmap2)
3191 map1 = isl_map_from_basic_map(isl_basic_map_copy(bmap1));
3192 map2 = isl_map_from_basic_map(isl_basic_map_copy(bmap2));
3194 is_subset = isl_map_is_subset(map1, map2);
3202 int isl_basic_map_is_equal(
3203 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3207 if (!bmap1 || !bmap2)
3209 is_subset = isl_basic_map_is_subset(bmap1, bmap2);
3212 is_subset = isl_basic_map_is_subset(bmap2, bmap1);
3216 int isl_basic_set_is_equal(
3217 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
3219 return isl_basic_map_is_equal(
3220 (struct isl_basic_map *)bset1, (struct isl_basic_map *)bset2);
3223 int isl_map_is_empty(struct isl_map *map)
3230 for (i = 0; i < map->n; ++i) {
3231 is_empty = isl_basic_map_is_empty(map->p[i]);
3240 int isl_set_is_empty(struct isl_set *set)
3242 return isl_map_is_empty((struct isl_map *)set);
3245 int isl_map_is_subset(struct isl_map *map1, struct isl_map *map2)
3249 struct isl_map *diff;
3254 if (isl_map_is_empty(map1))
3257 if (isl_map_is_empty(map2))
3260 diff = isl_map_subtract(isl_map_copy(map1), isl_map_copy(map2));
3264 is_subset = isl_map_is_empty(diff);
3270 int isl_map_is_equal(struct isl_map *map1, struct isl_map *map2)
3276 is_subset = isl_map_is_subset(map1, map2);
3279 is_subset = isl_map_is_subset(map2, map1);
3283 int isl_basic_map_is_strict_subset(
3284 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3288 if (!bmap1 || !bmap2)
3290 is_subset = isl_basic_map_is_subset(bmap1, bmap2);
3293 is_subset = isl_basic_map_is_subset(bmap2, bmap1);
3294 if (is_subset == -1)
3299 static int basic_map_contains(struct isl_basic_map *bmap, struct isl_vec *vec)
3305 total = 1 + isl_basic_map_total_dim(bmap);
3306 if (total != vec->size)
3311 for (i = 0; i < bmap->n_eq; ++i) {
3312 isl_seq_inner_product(vec->block.data, bmap->eq[i], total, &s);
3313 if (!isl_int_is_zero(s)) {
3319 for (i = 0; i < bmap->n_ineq; ++i) {
3320 isl_seq_inner_product(vec->block.data, bmap->ineq[i], total, &s);
3321 if (isl_int_is_neg(s)) {
3332 int isl_basic_map_is_universe(struct isl_basic_map *bmap)
3336 return bmap->n_eq == 0 && bmap->n_ineq == 0;
3339 int isl_basic_map_is_empty(struct isl_basic_map *bmap)
3341 struct isl_basic_set *bset = NULL;
3342 struct isl_vec *sample = NULL;
3349 if (F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
3352 if (F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL)) {
3353 struct isl_basic_map *copy = isl_basic_map_copy(bmap);
3354 copy = isl_basic_map_convex_hull(copy);
3355 empty = F_ISSET(copy, ISL_BASIC_MAP_EMPTY);
3356 isl_basic_map_free(copy);
3360 total = 1 + isl_basic_map_total_dim(bmap);
3361 if (bmap->sample && bmap->sample->size == total) {
3362 int contains = basic_map_contains(bmap, bmap->sample);
3368 bset = isl_basic_map_underlying_set(isl_basic_map_copy(bmap));
3371 sample = isl_basic_set_sample(bset);
3374 empty = sample->size == 0;
3376 isl_vec_free(bmap->ctx, bmap->sample);
3377 bmap->sample = sample;
3382 int isl_basic_set_is_empty(struct isl_basic_set *bset)
3384 return isl_basic_map_is_empty((struct isl_basic_map *)bset);
3387 struct isl_map *isl_basic_map_union(
3388 struct isl_basic_map *bmap1, struct isl_basic_map *bmap2)
3390 struct isl_map *map;
3391 if (!bmap1 || !bmap2)
3394 isl_assert(map1->ctx, isl_dim_equal(bmap1->dim, bmap2->dim), goto error);
3396 map = isl_map_alloc_dim(isl_dim_copy(bmap1->dim), 2, 0);
3399 map = isl_map_add(map, bmap1);
3400 map = isl_map_add(map, bmap2);
3403 isl_basic_map_free(bmap1);
3404 isl_basic_map_free(bmap2);
3408 struct isl_set *isl_basic_set_union(
3409 struct isl_basic_set *bset1, struct isl_basic_set *bset2)
3411 return (struct isl_set *)isl_basic_map_union(
3412 (struct isl_basic_map *)bset1,
3413 (struct isl_basic_map *)bset2);
3416 /* Order divs such that any div only depends on previous divs */
3417 static struct isl_basic_map *order_divs(struct isl_basic_map *bmap)
3420 unsigned off = isl_dim_total(bmap->dim);
3422 for (i = 0; i < bmap->n_div; ++i) {
3424 pos = isl_seq_first_non_zero(bmap->div[i]+1+1+off+i,
3428 swap_div(bmap, i, pos);
3434 /* Look for a div in dst that corresponds to the div "div" in src.
3435 * The divs before "div" in src and dst are assumed to be the same.
3437 * Returns -1 if no corresponding div was found and the position
3438 * of the corresponding div in dst otherwise.
3440 static int find_div(struct isl_basic_map *dst,
3441 struct isl_basic_map *src, unsigned div)
3445 unsigned total = isl_dim_total(src->dim);
3447 isl_assert(dst->ctx, div <= dst->n_div, return -1);
3448 for (i = div; i < dst->n_div; ++i)
3449 if (isl_seq_eq(dst->div[i], src->div[div], 1+1+total+div) &&
3450 isl_seq_first_non_zero(dst->div[i]+1+1+total+div,
3451 dst->n_div - div) == -1)
3456 struct isl_basic_map *isl_basic_map_align_divs(
3457 struct isl_basic_map *dst, struct isl_basic_map *src)
3460 unsigned total = isl_dim_total(src->dim);
3465 if (src->n_div == 0)
3468 for (i = 0; i < src->n_div; ++i)
3469 isl_assert(src->ctx, !isl_int_is_zero(src->div[i][0]), goto error);
3471 src = order_divs(src);
3472 dst = isl_basic_map_extend_dim(dst, isl_dim_copy(dst->dim),
3473 src->n_div, 0, 2 * src->n_div);
3476 for (i = 0; i < src->n_div; ++i) {
3477 int j = find_div(dst, src, i);
3479 j = isl_basic_map_alloc_div(dst);
3482 isl_seq_cpy(dst->div[j], src->div[i], 1+1+total+i);
3483 isl_seq_clr(dst->div[j]+1+1+total+i, dst->n_div - i);
3484 if (add_div_constraints(dst, j) < 0)
3488 swap_div(dst, i, j);
3492 isl_basic_map_free(dst);
3496 struct isl_basic_set *isl_basic_set_align_divs(
3497 struct isl_basic_set *dst, struct isl_basic_set *src)
3499 return (struct isl_basic_set *)isl_basic_map_align_divs(
3500 (struct isl_basic_map *)dst, (struct isl_basic_map *)src);
3503 struct isl_map *isl_map_align_divs(struct isl_map *map)
3507 map = isl_map_compute_divs(map);
3508 map = isl_map_cow(map);
3512 for (i = 1; i < map->n; ++i)
3513 map->p[0] = isl_basic_map_align_divs(map->p[0], map->p[i]);
3514 for (i = 1; i < map->n; ++i)
3515 map->p[i] = isl_basic_map_align_divs(map->p[i], map->p[0]);
3517 F_CLR(map, ISL_MAP_NORMALIZED);
3521 static struct isl_map *add_cut_constraint(struct isl_map *dst,
3522 struct isl_basic_map *src, isl_int *c,
3523 unsigned len, int oppose)
3525 struct isl_basic_map *copy = NULL;
3530 copy = isl_basic_map_copy(src);
3531 copy = isl_basic_map_cow(copy);
3534 copy = isl_basic_map_extend_constraints(copy, 0, 1);
3535 k = isl_basic_map_alloc_inequality(copy);
3539 isl_seq_neg(copy->ineq[k], c, len);
3541 isl_seq_cpy(copy->ineq[k], c, len);
3542 total = 1 + isl_basic_map_total_dim(copy);
3543 isl_seq_clr(copy->ineq[k]+len, total - len);
3544 isl_inequality_negate(copy, k);
3545 copy = isl_basic_map_simplify(copy);
3546 copy = isl_basic_map_finalize(copy);
3547 is_empty = isl_basic_map_is_empty(copy);
3551 dst = isl_map_add(dst, copy);
3553 isl_basic_map_free(copy);
3556 isl_basic_map_free(copy);
3561 static struct isl_map *subtract(struct isl_map *map, struct isl_basic_map *bmap)
3565 struct isl_map *rest = NULL;
3567 unsigned total = isl_basic_map_total_dim(bmap);
3574 if (F_ISSET(map, ISL_MAP_DISJOINT))
3575 FL_SET(flags, ISL_MAP_DISJOINT);
3577 max = map->n * (2 * bmap->n_eq + bmap->n_ineq);
3578 rest = isl_map_alloc_dim(isl_dim_copy(map->dim), max, flags);
3582 for (i = 0; i < map->n; ++i) {
3583 map->p[i] = isl_basic_map_align_divs(map->p[i], bmap);
3588 for (j = 0; j < map->n; ++j)
3589 map->p[j] = isl_basic_map_cow(map->p[j]);
3591 for (i = 0; i < bmap->n_eq; ++i) {
3592 for (j = 0; j < map->n; ++j) {
3593 rest = add_cut_constraint(rest,
3594 map->p[j], bmap->eq[i], 1+total, 0);
3598 rest = add_cut_constraint(rest,
3599 map->p[j], bmap->eq[i], 1+total, 1);
3603 map->p[j] = isl_basic_map_extend_constraints(map->p[j],
3607 k = isl_basic_map_alloc_equality(map->p[j]);
3610 isl_seq_cpy(map->p[j]->eq[k], bmap->eq[i], 1+total);
3611 isl_seq_clr(map->p[j]->eq[k]+1+total,
3612 map->p[j]->n_div - bmap->n_div);
3616 for (i = 0; i < bmap->n_ineq; ++i) {
3617 for (j = 0; j < map->n; ++j) {
3618 rest = add_cut_constraint(rest,
3619 map->p[j], bmap->ineq[i], 1+total, 0);
3623 map->p[j] = isl_basic_map_extend_constraints(map->p[j],
3627 k = isl_basic_map_alloc_inequality(map->p[j]);
3630 isl_seq_cpy(map->p[j]->ineq[k], bmap->ineq[i], 1+total);
3631 isl_seq_clr(map->p[j]->ineq[k]+1+total,
3632 map->p[j]->n_div - bmap->n_div);
3644 struct isl_map *isl_map_subtract(struct isl_map *map1, struct isl_map *map2)
3650 isl_assert(map1->ctx, isl_dim_equal(map1->dim, map2->dim), goto error);
3652 if (isl_map_is_empty(map2)) {
3657 map1 = isl_map_compute_divs(map1);
3658 map2 = isl_map_compute_divs(map2);
3662 for (i = 0; map1 && i < map2->n; ++i)
3663 map1 = subtract(map1, map2->p[i]);
3673 struct isl_set *isl_set_subtract(struct isl_set *set1, struct isl_set *set2)
3675 return (struct isl_set *)
3677 (struct isl_map *)set1, (struct isl_map *)set2);
3680 struct isl_set *isl_set_apply(struct isl_set *set, struct isl_map *map)
3684 isl_assert(set->ctx, isl_map_compatible_domain(map, set), goto error);
3685 map = isl_map_intersect_domain(map, set);
3686 set = isl_map_range(map);
3694 /* There is no need to cow as removing empty parts doesn't change
3695 * the meaning of the set.
3697 struct isl_map *isl_map_remove_empty_parts(struct isl_map *map)
3704 for (i = map->n-1; i >= 0; --i) {
3705 if (!F_ISSET(map->p[i], ISL_BASIC_MAP_EMPTY))
3707 isl_basic_map_free(map->p[i]);
3708 if (i != map->n-1) {
3709 F_CLR(map, ISL_MAP_NORMALIZED);
3710 map->p[i] = map->p[map->n-1];
3718 struct isl_set *isl_set_remove_empty_parts(struct isl_set *set)
3720 return (struct isl_set *)
3721 isl_map_remove_empty_parts((struct isl_map *)set);
3724 struct isl_basic_set *isl_set_copy_basic_set(struct isl_set *set)
3726 struct isl_basic_set *bset;
3727 if (!set || set->n == 0)
3729 bset = set->p[set->n-1];
3730 isl_assert(set->ctx, F_ISSET(bset, ISL_BASIC_SET_FINAL), return NULL);
3731 return isl_basic_set_copy(bset);
3734 struct isl_set *isl_set_drop_basic_set(struct isl_set *set,
3735 struct isl_basic_set *bset)
3741 for (i = set->n-1; i >= 0; --i) {
3742 if (set->p[i] != bset)
3744 set = isl_set_cow(set);
3747 isl_basic_set_free(set->p[i]);
3748 if (i != set->n-1) {
3749 F_CLR(set, ISL_SET_NORMALIZED);
3750 set->p[i] = set->p[set->n-1];
3755 isl_basic_set_free(bset);
3759 isl_basic_set_free(bset);
3763 /* Given two _disjoint_ basic sets bset1 and bset2, check whether
3764 * for any common value of the parameters and dimensions preceding dim
3765 * in both basic sets, the values of dimension pos in bset1 are
3766 * smaller or larger than those in bset2.
3769 * 1 if bset1 follows bset2
3770 * -1 if bset1 precedes bset2
3771 * 0 if bset1 and bset2 are incomparable
3772 * -2 if some error occurred.
3774 int isl_basic_set_compare_at(struct isl_basic_set *bset1,
3775 struct isl_basic_set *bset2, int pos)
3777 struct isl_dim *dims;
3778 struct isl_basic_map *bmap1 = NULL;
3779 struct isl_basic_map *bmap2 = NULL;
3780 struct isl_ctx *ctx;
3781 struct isl_vec *obj;
3784 unsigned dim1, dim2;
3786 enum isl_lp_result res;
3789 if (!bset1 || !bset2)
3792 nparam = isl_basic_set_n_param(bset1);
3793 dim1 = isl_basic_set_n_dim(bset1);
3794 dim2 = isl_basic_set_n_dim(bset2);
3795 dims = isl_dim_alloc(bset1->ctx, nparam, pos, dim1 - pos);
3796 bmap1 = isl_basic_map_from_basic_set(isl_basic_set_copy(bset1), dims);
3797 dims = isl_dim_alloc(bset2->ctx, nparam, pos, dim2 - pos);
3798 bmap2 = isl_basic_map_from_basic_set(isl_basic_set_copy(bset2), dims);
3799 if (!bmap1 || !bmap2)
3801 bmap1 = isl_basic_map_extend(bmap1, nparam,
3802 pos, (dim1 - pos) + (dim2 - pos),
3803 bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
3804 bmap1 = add_constraints(bmap1, bmap2, 0, dim1 - pos);
3807 total = isl_basic_map_total_dim(bmap1);
3809 obj = isl_vec_alloc(ctx, total);
3810 isl_seq_clr(obj->block.data, total);
3811 isl_int_set_si(obj->block.data[nparam+pos], 1);
3812 isl_int_set_si(obj->block.data[nparam+pos+(dim1-pos)], -1);
3817 res = isl_solve_lp(bmap1, 0, obj->block.data, ctx->one, &num, &den);
3818 if (res == isl_lp_empty)
3820 else if (res == isl_lp_ok && isl_int_is_pos(num))
3822 else if ((res == isl_lp_ok && isl_int_is_neg(num)) ||
3823 res == isl_lp_unbounded)
3829 isl_basic_map_free(bmap1);
3830 isl_vec_free(ctx, obj);
3833 isl_basic_map_free(bmap1);
3834 isl_basic_map_free(bmap2);
3838 static int isl_basic_map_fast_has_fixed_var(struct isl_basic_map *bmap,
3839 unsigned pos, isl_int *val)
3847 total = isl_basic_map_total_dim(bmap);
3848 for (i = 0, d = total-1; i < bmap->n_eq && d+1 > pos; ++i) {
3849 for (; d+1 > pos; --d)
3850 if (!isl_int_is_zero(bmap->eq[i][1+d]))
3854 if (isl_seq_first_non_zero(bmap->eq[i]+1, d) != -1)
3856 if (isl_seq_first_non_zero(bmap->eq[i]+1+d+1, total-d-1) != -1)
3858 if (!isl_int_is_one(bmap->eq[i][1+d]))
3861 isl_int_neg(*val, bmap->eq[i][0]);
3867 static int isl_map_fast_has_fixed_var(struct isl_map *map,
3868 unsigned pos, isl_int *val)
3880 return isl_basic_map_fast_has_fixed_var(map->p[0], pos, val);
3883 fixed = isl_basic_map_fast_has_fixed_var(map->p[0], pos, &v);
3884 for (i = 1; fixed == 1 && i < map->n; ++i) {
3885 fixed = isl_basic_map_fast_has_fixed_var(map->p[i], pos, &tmp);
3886 if (fixed == 1 && isl_int_ne(tmp, v))
3890 isl_int_set(*val, v);
3896 static int isl_set_fast_has_fixed_var(struct isl_set *set, unsigned pos,
3899 return isl_map_fast_has_fixed_var((struct isl_map *)set, pos, val);
3902 int isl_basic_map_fast_is_fixed(struct isl_basic_map *bmap,
3903 enum isl_dim_type type, unsigned pos, isl_int *val)
3905 if (pos >= isl_basic_map_dim(bmap, type))
3907 return isl_basic_map_fast_has_fixed_var(bmap,
3908 basic_map_offset(bmap, type) - 1 + pos, val);
3911 /* Check if dimension dim has fixed value and if so and if val is not NULL,
3912 * then return this fixed value in *val.
3914 int isl_set_fast_dim_is_fixed(struct isl_set *set, unsigned dim, isl_int *val)
3916 return isl_set_fast_has_fixed_var(set, isl_set_n_param(set) + dim, val);
3919 /* Check if input variable in has fixed value and if so and if val is not NULL,
3920 * then return this fixed value in *val.
3922 int isl_map_fast_input_is_fixed(struct isl_map *map, unsigned in, isl_int *val)
3924 return isl_map_fast_has_fixed_var(map, isl_map_n_param(map) + in, val);
3927 /* Check if dimension dim has an (obvious) fixed lower bound and if so
3928 * and if val is not NULL, then return this lower bound in *val.
3930 int isl_basic_set_fast_dim_has_fixed_lower_bound(struct isl_basic_set *bset,
3931 unsigned dim, isl_int *val)
3933 int i, i_eq = -1, i_ineq = -1;
3940 total = isl_basic_set_total_dim(bset);
3941 nparam = isl_basic_set_n_param(bset);
3942 for (i = 0; i < bset->n_eq; ++i) {
3943 if (isl_int_is_zero(bset->eq[i][1+nparam+dim]))
3949 for (i = 0; i < bset->n_ineq; ++i) {
3950 if (!isl_int_is_pos(bset->ineq[i][1+nparam+dim]))
3952 if (i_eq != -1 || i_ineq != -1)
3956 if (i_eq == -1 && i_ineq == -1)
3958 c = i_eq != -1 ? bset->eq[i_eq] : bset->ineq[i_ineq];
3959 /* The coefficient should always be one due to normalization. */
3960 if (!isl_int_is_one(c[1+nparam+dim]))
3962 if (isl_seq_first_non_zero(c+1, nparam+dim) != -1)
3964 if (isl_seq_first_non_zero(c+1+nparam+dim+1,
3965 total - nparam - dim - 1) != -1)
3968 isl_int_neg(*val, c[0]);
3972 int isl_set_fast_dim_has_fixed_lower_bound(struct isl_set *set,
3973 unsigned dim, isl_int *val)
3985 return isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[0],
3989 fixed = isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[0],
3991 for (i = 1; fixed == 1 && i < set->n; ++i) {
3992 fixed = isl_basic_set_fast_dim_has_fixed_lower_bound(set->p[i],
3994 if (fixed == 1 && isl_int_ne(tmp, v))
3998 isl_int_set(*val, v);
4009 static int qsort_constraint_cmp(const void *p1, const void *p2)
4011 const struct constraint *c1 = (const struct constraint *)p1;
4012 const struct constraint *c2 = (const struct constraint *)p2;
4013 unsigned size = isl_min(c1->size, c2->size);
4014 return isl_seq_cmp(c1->c, c2->c, size);
4017 static struct isl_basic_map *isl_basic_map_sort_constraints(
4018 struct isl_basic_map *bmap)
4021 struct constraint *c;
4026 total = isl_basic_map_total_dim(bmap);
4027 c = isl_alloc_array(bmap->ctx, struct constraint, bmap->n_ineq);
4030 for (i = 0; i < bmap->n_ineq; ++i) {
4032 c[i].c = bmap->ineq[i];
4034 qsort(c, bmap->n_ineq, sizeof(struct constraint), qsort_constraint_cmp);
4035 for (i = 0; i < bmap->n_ineq; ++i)
4036 bmap->ineq[i] = c[i].c;
4040 isl_basic_map_free(bmap);
4044 struct isl_basic_map *isl_basic_map_normalize(struct isl_basic_map *bmap)
4048 if (F_ISSET(bmap, ISL_BASIC_MAP_NORMALIZED))
4050 bmap = isl_basic_map_convex_hull(bmap);
4051 bmap = isl_basic_map_sort_constraints(bmap);
4052 F_SET(bmap, ISL_BASIC_MAP_NORMALIZED);
4056 struct isl_basic_set *isl_basic_set_normalize(struct isl_basic_set *bset)
4058 return (struct isl_basic_set *)isl_basic_map_normalize(
4059 (struct isl_basic_map *)bset);
4062 static int isl_basic_map_fast_cmp(const struct isl_basic_map *bmap1,
4063 const struct isl_basic_map *bmap2)
4070 if (isl_basic_map_n_param(bmap1) != isl_basic_map_n_param(bmap2))
4071 return isl_basic_map_n_param(bmap1) - isl_basic_map_n_param(bmap2);
4072 if (isl_basic_map_n_in(bmap1) != isl_basic_map_n_in(bmap2))
4073 return isl_basic_map_n_out(bmap1) - isl_basic_map_n_out(bmap2);
4074 if (isl_basic_map_n_out(bmap1) != isl_basic_map_n_out(bmap2))
4075 return isl_basic_map_n_out(bmap1) - isl_basic_map_n_out(bmap2);
4076 if (F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY) &&
4077 F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
4079 if (F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY))
4081 if (F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
4083 if (bmap1->n_eq != bmap2->n_eq)
4084 return bmap1->n_eq - bmap2->n_eq;
4085 if (bmap1->n_ineq != bmap2->n_ineq)
4086 return bmap1->n_ineq - bmap2->n_ineq;
4087 if (bmap1->n_div != bmap2->n_div)
4088 return bmap1->n_div - bmap2->n_div;
4089 total = isl_basic_map_total_dim(bmap1);
4090 for (i = 0; i < bmap1->n_eq; ++i) {
4091 cmp = isl_seq_cmp(bmap1->eq[i], bmap2->eq[i], 1+total);
4095 for (i = 0; i < bmap1->n_ineq; ++i) {
4096 cmp = isl_seq_cmp(bmap1->ineq[i], bmap2->ineq[i], 1+total);
4100 for (i = 0; i < bmap1->n_div; ++i) {
4101 cmp = isl_seq_cmp(bmap1->div[i], bmap2->div[i], 1+1+total);
4108 static int isl_basic_map_fast_is_equal(struct isl_basic_map *bmap1,
4109 struct isl_basic_map *bmap2)
4111 return isl_basic_map_fast_cmp(bmap1, bmap2) == 0;
4114 static int qsort_bmap_cmp(const void *p1, const void *p2)
4116 const struct isl_basic_map *bmap1 = *(const struct isl_basic_map **)p1;
4117 const struct isl_basic_map *bmap2 = *(const struct isl_basic_map **)p2;
4119 return isl_basic_map_fast_cmp(bmap1, bmap2);
4122 /* We normalize in place, but if anything goes wrong we need
4123 * to return NULL, so we need to make sure we don't change the
4124 * meaning of any possible other copies of map.
4126 struct isl_map *isl_map_normalize(struct isl_map *map)
4129 struct isl_basic_map *bmap;
4133 if (F_ISSET(map, ISL_MAP_NORMALIZED))
4135 for (i = 0; i < map->n; ++i) {
4136 bmap = isl_basic_map_normalize(isl_basic_map_copy(map->p[i]));
4139 isl_basic_map_free(map->p[i]);
4142 qsort(map->p, map->n, sizeof(struct isl_basic_map *), qsort_bmap_cmp);
4143 F_SET(map, ISL_MAP_NORMALIZED);
4144 map = isl_map_remove_empty_parts(map);
4147 for (i = map->n - 1; i >= 1; --i) {
4148 if (!isl_basic_map_fast_is_equal(map->p[i-1], map->p[i]))
4150 isl_basic_map_free(map->p[i-1]);
4151 for (j = i; j < map->n; ++j)
4152 map->p[j-1] = map->p[j];
4162 struct isl_set *isl_set_normalize(struct isl_set *set)
4164 return (struct isl_set *)isl_map_normalize((struct isl_map *)set);
4167 int isl_map_fast_is_equal(struct isl_map *map1, struct isl_map *map2)
4177 if (!isl_dim_equal(map1->dim, map2->dim))
4180 map1 = isl_map_copy(map1);
4181 map2 = isl_map_copy(map2);
4182 map1 = isl_map_normalize(map1);
4183 map2 = isl_map_normalize(map2);
4186 equal = map1->n == map2->n;
4187 for (i = 0; equal && i < map1->n; ++i) {
4188 equal = isl_basic_map_fast_is_equal(map1->p[i], map2->p[i]);
4201 int isl_set_fast_is_equal(struct isl_set *set1, struct isl_set *set2)
4203 return isl_map_fast_is_equal((struct isl_map *)set1,
4204 (struct isl_map *)set2);
4207 /* Return an interval that ranges from min to max (inclusive)
4209 struct isl_basic_set *isl_basic_set_interval(struct isl_ctx *ctx,
4210 isl_int min, isl_int max)
4213 struct isl_basic_set *bset = NULL;
4215 bset = isl_basic_set_alloc(ctx, 0, 1, 0, 0, 2);
4219 k = isl_basic_set_alloc_inequality(bset);
4222 isl_int_set_si(bset->ineq[k][1], 1);
4223 isl_int_neg(bset->ineq[k][0], min);
4225 k = isl_basic_set_alloc_inequality(bset);
4228 isl_int_set_si(bset->ineq[k][1], -1);
4229 isl_int_set(bset->ineq[k][0], max);
4233 isl_basic_set_free(bset);
4237 /* Return the Cartesian product of the basic sets in list (in the given order).
4239 struct isl_basic_set *isl_basic_set_product(struct isl_basic_set_list *list)
4247 struct isl_basic_set *product = NULL;
4251 isl_assert(list->ctx, list->n > 0, goto error);
4252 isl_assert(list->ctx, list->p[0], goto error);
4253 nparam = isl_basic_set_n_param(list->p[0]);
4254 dim = isl_basic_set_n_dim(list->p[0]);
4255 extra = list->p[0]->n_div;
4256 n_eq = list->p[0]->n_eq;
4257 n_ineq = list->p[0]->n_ineq;
4258 for (i = 1; i < list->n; ++i) {
4259 isl_assert(list->ctx, list->p[i], goto error);
4260 isl_assert(list->ctx,
4261 nparam == isl_basic_set_n_param(list->p[i]), goto error);
4262 dim += isl_basic_set_n_dim(list->p[i]);
4263 extra += list->p[i]->n_div;
4264 n_eq += list->p[i]->n_eq;
4265 n_ineq += list->p[i]->n_ineq;
4267 product = isl_basic_set_alloc(list->ctx, nparam, dim, extra,
4272 for (i = 0; i < list->n; ++i) {
4273 isl_basic_set_add_constraints(product,
4274 isl_basic_set_copy(list->p[i]), dim);
4275 dim += isl_basic_set_n_dim(list->p[i]);
4277 isl_basic_set_list_free(list);
4280 isl_basic_set_free(product);
4281 isl_basic_set_list_free(list);
4285 uint32_t isl_basic_set_get_hash(struct isl_basic_set *bset)
4293 bset = isl_basic_set_copy(bset);
4294 bset = isl_basic_set_normalize(bset);
4297 total = isl_basic_set_total_dim(bset);
4298 isl_hash_byte(hash, bset->n_eq & 0xFF);
4299 for (i = 0; i < bset->n_eq; ++i) {
4301 c_hash = isl_seq_get_hash(bset->eq[i], 1 + total);
4302 isl_hash_hash(hash, c_hash);
4304 isl_hash_byte(hash, bset->n_ineq & 0xFF);
4305 for (i = 0; i < bset->n_ineq; ++i) {
4307 c_hash = isl_seq_get_hash(bset->ineq[i], 1 + total);
4308 isl_hash_hash(hash, c_hash);
4310 isl_hash_byte(hash, bset->n_div & 0xFF);
4311 for (i = 0; i < bset->n_div; ++i) {
4313 if (isl_int_is_zero(bset->div[i][0]))
4315 isl_hash_byte(hash, i & 0xFF);
4316 c_hash = isl_seq_get_hash(bset->div[i], 1 + 1 + total);
4317 isl_hash_hash(hash, c_hash);
4319 isl_basic_set_free(bset);
4323 uint32_t isl_set_get_hash(struct isl_set *set)
4330 set = isl_set_copy(set);
4331 set = isl_set_normalize(set);
4335 hash = isl_hash_init();
4336 for (i = 0; i < set->n; ++i) {
4338 bset_hash = isl_basic_set_get_hash(set->p[i]);
4339 isl_hash_hash(hash, bset_hash);
4347 /* Check if the value for dimension dim is completely determined
4348 * by the values of the other parameters and variables.
4349 * That is, check if dimension dim is involved in an equality.
4351 int isl_basic_set_dim_is_unique(struct isl_basic_set *bset, unsigned dim)
4358 nparam = isl_basic_set_n_param(bset);
4359 for (i = 0; i < bset->n_eq; ++i)
4360 if (!isl_int_is_zero(bset->eq[i][1 + nparam + dim]))
4365 /* Check if the value for dimension dim is completely determined
4366 * by the values of the other parameters and variables.
4367 * That is, check if dimension dim is involved in an equality
4368 * for each of the subsets.
4370 int isl_set_dim_is_unique(struct isl_set *set, unsigned dim)
4376 for (i = 0; i < set->n; ++i) {
4378 unique = isl_basic_set_dim_is_unique(set->p[i], dim);